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:
Damien Neil 2019-11-11 16:30:04 -08:00
parent ce3384cd34
commit 1605775be0
2 changed files with 35 additions and 1 deletions

View File

@ -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:] {

View File

@ -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)
}
}