1// Copyright 2019 The Go Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style 3// license that can be found in the LICENSE file. 4 5package impl 6 7import ( 8 "sort" 9 10 "google.golang.org/protobuf/encoding/protowire" 11 "google.golang.org/protobuf/internal/encoding/messageset" 12 "google.golang.org/protobuf/internal/errors" 13 "google.golang.org/protobuf/internal/flags" 14) 15 16func sizeMessageSet(mi *MessageInfo, p pointer, opts marshalOptions) (size int) { 17 if !flags.ProtoLegacy { 18 return 0 19 } 20 21 ext := *p.Apply(mi.extensionOffset).Extensions() 22 for _, x := range ext { 23 xi := getExtensionFieldInfo(x.Type()) 24 if xi.funcs.size == nil { 25 continue 26 } 27 num, _ := protowire.DecodeTag(xi.wiretag) 28 size += messageset.SizeField(num) 29 size += xi.funcs.size(x.Value(), protowire.SizeTag(messageset.FieldMessage), opts) 30 } 31 32 if u := mi.getUnknownBytes(p); u != nil { 33 size += messageset.SizeUnknown(*u) 34 } 35 36 return size 37} 38 39func marshalMessageSet(mi *MessageInfo, b []byte, p pointer, opts marshalOptions) ([]byte, error) { 40 if !flags.ProtoLegacy { 41 return b, errors.New("no support for message_set_wire_format") 42 } 43 44 ext := *p.Apply(mi.extensionOffset).Extensions() 45 switch len(ext) { 46 case 0: 47 case 1: 48 // Fast-path for one extension: Don't bother sorting the keys. 49 for _, x := range ext { 50 var err error 51 b, err = marshalMessageSetField(mi, b, x, opts) 52 if err != nil { 53 return b, err 54 } 55 } 56 default: 57 // Sort the keys to provide a deterministic encoding. 58 // Not sure this is required, but the old code does it. 59 keys := make([]int, 0, len(ext)) 60 for k := range ext { 61 keys = append(keys, int(k)) 62 } 63 sort.Ints(keys) 64 for _, k := range keys { 65 var err error 66 b, err = marshalMessageSetField(mi, b, ext[int32(k)], opts) 67 if err != nil { 68 return b, err 69 } 70 } 71 } 72 73 if u := mi.getUnknownBytes(p); u != nil { 74 var err error 75 b, err = messageset.AppendUnknown(b, *u) 76 if err != nil { 77 return b, err 78 } 79 } 80 81 return b, nil 82} 83 84func marshalMessageSetField(mi *MessageInfo, b []byte, x ExtensionField, opts marshalOptions) ([]byte, error) { 85 xi := getExtensionFieldInfo(x.Type()) 86 num, _ := protowire.DecodeTag(xi.wiretag) 87 b = messageset.AppendFieldStart(b, num) 88 b, err := xi.funcs.marshal(b, x.Value(), protowire.EncodeTag(messageset.FieldMessage, protowire.BytesType), opts) 89 if err != nil { 90 return b, err 91 } 92 b = messageset.AppendFieldEnd(b) 93 return b, nil 94} 95 96func unmarshalMessageSet(mi *MessageInfo, b []byte, p pointer, opts unmarshalOptions) (out unmarshalOutput, err error) { 97 if !flags.ProtoLegacy { 98 return out, errors.New("no support for message_set_wire_format") 99 } 100 101 ep := p.Apply(mi.extensionOffset).Extensions() 102 if *ep == nil { 103 *ep = make(map[int32]ExtensionField) 104 } 105 ext := *ep 106 initialized := true 107 err = messageset.Unmarshal(b, true, func(num protowire.Number, v []byte) error { 108 o, err := mi.unmarshalExtension(v, num, protowire.BytesType, ext, opts) 109 if err == errUnknown { 110 u := mi.mutableUnknownBytes(p) 111 *u = protowire.AppendTag(*u, num, protowire.BytesType) 112 *u = append(*u, v...) 113 return nil 114 } 115 if !o.initialized { 116 initialized = false 117 } 118 return err 119 }) 120 out.n = len(b) 121 out.initialized = initialized 122 return out, err 123} 124