mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2025-02-06 00:40:02 +00:00
internal/impl: handle some dynamic legacy messages
When creating a MessageDescriptor for a legacy message with a Descriptor method, we call that method on the type's zero value to get the message's DescriptorProto. Some existing dynamic message types have a Descriptor method which panics in this case. Catch the panic and continue as if the Descriptor method wasn't present. Change-Id: I98d4625d6917cc1ec25737e5670a443f5d02a404 Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/206637 Reviewed-by: Joe Tsai <thebrokentoaster@gmail.com>
This commit is contained in:
parent
ce3384cd34
commit
1605775be0
@ -93,7 +93,20 @@ func legacyLoadMessageDesc(t reflect.Type, name pref.FullName) pref.MessageDescr
|
||||
if !ok {
|
||||
return aberrantLoadMessageDesc(t, name)
|
||||
}
|
||||
b, idxs := mdV1.Descriptor()
|
||||
|
||||
// If this is a dynamic message type where there isn't a 1-1 mapping between
|
||||
// Go and protobuf types, calling the Descriptor method on the zero value of
|
||||
// the message type isn't likely to work. If it panics, swallow the panic and
|
||||
// continue as if the Descriptor method wasn't present.
|
||||
b, idxs := func() ([]byte, []int) {
|
||||
defer func() {
|
||||
recover()
|
||||
}()
|
||||
return mdV1.Descriptor()
|
||||
}()
|
||||
if b == nil {
|
||||
return aberrantLoadMessageDesc(t, name)
|
||||
}
|
||||
|
||||
md := legacyLoadFileDesc(b).Messages().Get(idxs[0])
|
||||
for _, i := range idxs[1:] {
|
||||
|
@ -79,3 +79,24 @@ func TestLegacyUnmarshalMethod(t *testing.T) {
|
||||
t.Fatalf("proto.Unmarshal(selfMarshaler{}): Marshal method not called")
|
||||
}
|
||||
}
|
||||
|
||||
type descPanicSelfMarshaler struct{}
|
||||
|
||||
const descPanicSelfMarshalerBytes = "bytes"
|
||||
|
||||
func (m descPanicSelfMarshaler) Reset() {}
|
||||
func (m descPanicSelfMarshaler) ProtoMessage() {}
|
||||
func (m descPanicSelfMarshaler) Descriptor() ([]byte, []int) { panic("Descriptor method panics") }
|
||||
func (m descPanicSelfMarshaler) String() string { return "descPanicSelfMarshaler{}" }
|
||||
func (m descPanicSelfMarshaler) Marshal() ([]byte, error) {
|
||||
return []byte(descPanicSelfMarshalerBytes), nil
|
||||
}
|
||||
|
||||
func TestSelfMarshalerDescriptorPanics(t *testing.T) {
|
||||
m := descPanicSelfMarshaler{}
|
||||
got, err := proto.Marshal(impl.Export{}.MessageOf(m).Interface())
|
||||
want := []byte(descPanicSelfMarshalerBytes)
|
||||
if err != nil || !bytes.Equal(got, want) {
|
||||
t.Fatalf("proto.Marshal(%v) = %v, %v; want %v, nil", m, got, err, want)
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user