mirror/protobuf-go
1
0
Fork 0
mirror of https://github.com/protocolbuffers/protobuf-go.git synced 2025-03-02 10:13:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

internal/impl: fix unmarshal of group containing their own field number

Browse Source 
The fast-path unmarshal was getting confused when parsing a group
containing a field with a number the same as the group's own field
number. Separate the handling of EndGroup tags.

Change-Id: I637702b42c94a26102e693ee29a55e80b37d7f28
Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/214737
Reviewed-by: Joe Tsai <joetsai@google.com>
This commit is contained in:
Damien Neil 2020-01-14 11:12:21 -08:00
parent 1c8015fff5
commit 2ae60936c2
4 changed files with 834 additions and 803 deletions

10
internal/impl/decode.go

@ -90,6 +90,13 @@ func (mi *MessageInfo) unmarshalPointer(b []byte, p pointer, groupTag wire.Numbe
}
b = b[n:]
if wtyp == wire.EndGroupType {
if num != groupTag {
return 0, errors.New("mismatching end group marker")
}
return start - len(b), nil
}
var f *coderFieldInfo
if int(num) < len(mi.denseCoderFields) {
f = mi.denseCoderFields[num]
@ -103,9 +110,6 @@ func (mi *MessageInfo) unmarshalPointer(b []byte, p pointer, groupTag wire.Numbe
break
}
n, err = f.funcs.unmarshal(b, p.Apply(f.offset), wtyp, opts)
case num == groupTag && wtyp == wire.EndGroupType:
// End of group.
return start - len(b), nil
default:
// Possible extension.
if exts == nil && mi.extensionOffset.IsValid() {

1618
internal/testprotos/test/test.pb.go

File diff suppressed because it is too large Load Diff

2
internal/testprotos/test/test.proto

@ -44,6 +44,7 @@ message TestAllTypes {
optional group OptionalGroup = 16 {
optional int32 a = 17;
optional NestedMessage optional_nested_message = 1000;
optional int32 same_field_number = 16;
}
optional NestedMessage optional_nested_message = 18;
optional ForeignMessage optional_foreign_message = 19;
@ -190,6 +191,7 @@ extend TestAllExtensions {
optional group OptionalGroup_extension = 16 {
optional int32 a = 17;
optional int32 same_field_number = 16;
optional TestAllTypes.NestedMessage optional_nested_message = 1000;
}

7
proto/testmessages_test.go

@ -158,17 +158,20 @@ var testValidMessages = []testProto{
desc: "groups",
decodeTo: []proto.Message{&testpb.TestAllTypes{
Optionalgroup: &testpb.TestAllTypes_OptionalGroup{
A: proto.Int32(1017),
A: proto.Int32(1017),
SameFieldNumber: proto.Int32(1016),
},
}, build(
&testpb.TestAllExtensions{},
extend(testpb.E_OptionalgroupExtension, &testpb.OptionalGroupExtension{
A: proto.Int32(1017),
A: proto.Int32(1017),
SameFieldNumber: proto.Int32(1016),
}),
)},
wire: pack.Message{
pack.Tag{16, pack.StartGroupType},
pack.Tag{17, pack.VarintType}, pack.Varint(1017),
pack.Tag{16, pack.VarintType}, pack.Varint(1016),
pack.Tag{16, pack.EndGroupType},
}.Marshal(),
},