mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2024-12-27 06:21:25 +00:00
94bb78c93b
There currently is no risk of producing invalid wire format, but that will change with subsequent changes regarding lazy decoding. We have been running this change in production for about a month, without ever triggering the check (until lazy decoding is involved). related to golang/protobuf#1609 Change-Id: I3c5c956aee2fa81f99dea03ed2a977a1547081fc Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/579595 Auto-Submit: Michael Stapelberg <stapelberg@google.com> Reviewed-by: Lasse Folger <lassefolger@google.com> LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com>
99 lines
3.2 KiB
Go
99 lines
3.2 KiB
Go
// Copyright 2019 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package proto
|
|
|
|
import (
|
|
"google.golang.org/protobuf/encoding/protowire"
|
|
"google.golang.org/protobuf/internal/encoding/messageset"
|
|
"google.golang.org/protobuf/internal/errors"
|
|
"google.golang.org/protobuf/internal/flags"
|
|
"google.golang.org/protobuf/internal/order"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
"google.golang.org/protobuf/reflect/protoregistry"
|
|
)
|
|
|
|
func (o MarshalOptions) sizeMessageSet(m protoreflect.Message) (size int) {
|
|
m.Range(func(fd protoreflect.FieldDescriptor, v protoreflect.Value) bool {
|
|
size += messageset.SizeField(fd.Number())
|
|
size += protowire.SizeTag(messageset.FieldMessage)
|
|
size += protowire.SizeBytes(o.size(v.Message()))
|
|
return true
|
|
})
|
|
size += messageset.SizeUnknown(m.GetUnknown())
|
|
return size
|
|
}
|
|
|
|
func (o MarshalOptions) marshalMessageSet(b []byte, m protoreflect.Message) ([]byte, error) {
|
|
if !flags.ProtoLegacy {
|
|
return b, errors.New("no support for message_set_wire_format")
|
|
}
|
|
fieldOrder := order.AnyFieldOrder
|
|
if o.Deterministic {
|
|
fieldOrder = order.NumberFieldOrder
|
|
}
|
|
var err error
|
|
order.RangeFields(m, fieldOrder, func(fd protoreflect.FieldDescriptor, v protoreflect.Value) bool {
|
|
b, err = o.marshalMessageSetField(b, fd, v)
|
|
return err == nil
|
|
})
|
|
if err != nil {
|
|
return b, err
|
|
}
|
|
return messageset.AppendUnknown(b, m.GetUnknown())
|
|
}
|
|
|
|
func (o MarshalOptions) marshalMessageSetField(b []byte, fd protoreflect.FieldDescriptor, value protoreflect.Value) ([]byte, error) {
|
|
b = messageset.AppendFieldStart(b, fd.Number())
|
|
b = protowire.AppendTag(b, messageset.FieldMessage, protowire.BytesType)
|
|
calculatedSize := o.Size(value.Message().Interface())
|
|
b = protowire.AppendVarint(b, uint64(calculatedSize))
|
|
before := len(b)
|
|
b, err := o.marshalMessage(b, value.Message())
|
|
if err != nil {
|
|
return b, err
|
|
}
|
|
if measuredSize := len(b) - before; calculatedSize != measuredSize {
|
|
return nil, errors.MismatchedSizeCalculation(calculatedSize, measuredSize)
|
|
}
|
|
b = messageset.AppendFieldEnd(b)
|
|
return b, nil
|
|
}
|
|
|
|
func (o UnmarshalOptions) unmarshalMessageSet(b []byte, m protoreflect.Message) error {
|
|
if !flags.ProtoLegacy {
|
|
return errors.New("no support for message_set_wire_format")
|
|
}
|
|
return messageset.Unmarshal(b, false, func(num protowire.Number, v []byte) error {
|
|
err := o.unmarshalMessageSetField(m, num, v)
|
|
if err == errUnknown {
|
|
unknown := m.GetUnknown()
|
|
unknown = protowire.AppendTag(unknown, num, protowire.BytesType)
|
|
unknown = protowire.AppendBytes(unknown, v)
|
|
m.SetUnknown(unknown)
|
|
return nil
|
|
}
|
|
return err
|
|
})
|
|
}
|
|
|
|
func (o UnmarshalOptions) unmarshalMessageSetField(m protoreflect.Message, num protowire.Number, v []byte) error {
|
|
md := m.Descriptor()
|
|
if !md.ExtensionRanges().Has(num) {
|
|
return errUnknown
|
|
}
|
|
xt, err := o.Resolver.FindExtensionByNumber(md.FullName(), num)
|
|
if err == protoregistry.NotFound {
|
|
return errUnknown
|
|
}
|
|
if err != nil {
|
|
return errors.New("%v: unable to resolve extension %v: %v", md.FullName(), num, err)
|
|
}
|
|
xd := xt.TypeDescriptor()
|
|
if err := o.unmarshalMessage(v, m.Mutable(xd).Message()); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|