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 proto 6 7import ( 8 "google.golang.org/protobuf/encoding/protowire" 9 "google.golang.org/protobuf/internal/encoding/messageset" 10 "google.golang.org/protobuf/internal/errors" 11 "google.golang.org/protobuf/internal/flags" 12 "google.golang.org/protobuf/internal/order" 13 "google.golang.org/protobuf/reflect/protoreflect" 14 "google.golang.org/protobuf/reflect/protoregistry" 15) 16 17func (o MarshalOptions) sizeMessageSet(m protoreflect.Message) (size int) { 18 m.Range(func(fd protoreflect.FieldDescriptor, v protoreflect.Value) bool { 19 size += messageset.SizeField(fd.Number()) 20 size += protowire.SizeTag(messageset.FieldMessage) 21 size += protowire.SizeBytes(o.size(v.Message())) 22 return true 23 }) 24 size += messageset.SizeUnknown(m.GetUnknown()) 25 return size 26} 27 28func (o MarshalOptions) marshalMessageSet(b []byte, m protoreflect.Message) ([]byte, error) { 29 if !flags.ProtoLegacy { 30 return b, errors.New("no support for message_set_wire_format") 31 } 32 fieldOrder := order.AnyFieldOrder 33 if o.Deterministic { 34 fieldOrder = order.NumberFieldOrder 35 } 36 var err error 37 order.RangeFields(m, fieldOrder, func(fd protoreflect.FieldDescriptor, v protoreflect.Value) bool { 38 b, err = o.marshalMessageSetField(b, fd, v) 39 return err == nil 40 }) 41 if err != nil { 42 return b, err 43 } 44 return messageset.AppendUnknown(b, m.GetUnknown()) 45} 46 47func (o MarshalOptions) marshalMessageSetField(b []byte, fd protoreflect.FieldDescriptor, value protoreflect.Value) ([]byte, error) { 48 b = messageset.AppendFieldStart(b, fd.Number()) 49 b = protowire.AppendTag(b, messageset.FieldMessage, protowire.BytesType) 50 b = protowire.AppendVarint(b, uint64(o.Size(value.Message().Interface()))) 51 b, err := o.marshalMessage(b, value.Message()) 52 if err != nil { 53 return b, err 54 } 55 b = messageset.AppendFieldEnd(b) 56 return b, nil 57} 58 59func (o UnmarshalOptions) unmarshalMessageSet(b []byte, m protoreflect.Message) error { 60 if !flags.ProtoLegacy { 61 return errors.New("no support for message_set_wire_format") 62 } 63 return messageset.Unmarshal(b, false, func(num protowire.Number, v []byte) error { 64 err := o.unmarshalMessageSetField(m, num, v) 65 if err == errUnknown { 66 unknown := m.GetUnknown() 67 unknown = protowire.AppendTag(unknown, num, protowire.BytesType) 68 unknown = protowire.AppendBytes(unknown, v) 69 m.SetUnknown(unknown) 70 return nil 71 } 72 return err 73 }) 74} 75 76func (o UnmarshalOptions) unmarshalMessageSetField(m protoreflect.Message, num protowire.Number, v []byte) error { 77 md := m.Descriptor() 78 if !md.ExtensionRanges().Has(num) { 79 return errUnknown 80 } 81 xt, err := o.Resolver.FindExtensionByNumber(md.FullName(), num) 82 if err == protoregistry.NotFound { 83 return errUnknown 84 } 85 if err != nil { 86 return errors.New("%v: unable to resolve extension %v: %v", md.FullName(), num, err) 87 } 88 xd := xt.TypeDescriptor() 89 if err := o.unmarshalMessage(v, m.Mutable(xd).Message()); err != nil { 90 return err 91 } 92 return nil 93} 94