mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2024-12-28 18:25:46 +00:00
d0b074956d
Move the test inputs for the wire marshaler and unmarshaler out of decode_test.go and into a new file. Consolidate some tests for invalid messages (UTF-8 validation failures, field numbers out of range) into a single list of invalid messages. Break out the no-enforce-utf8 test into a separate file, since it is both complicated and conditional on legacy support. Change-Id: Ide80fa9d3aec2b6d42a57e6f9265358aa5e661a7 Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/211557 Reviewed-by: Joe Tsai <joetsai@google.com>
221 lines
7.1 KiB
Go
221 lines
7.1 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_test
|
|
|
|
import (
|
|
"google.golang.org/protobuf/internal/encoding/pack"
|
|
"google.golang.org/protobuf/internal/encoding/wire"
|
|
"google.golang.org/protobuf/internal/flags"
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
messagesetpb "google.golang.org/protobuf/internal/testprotos/messageset/messagesetpb"
|
|
msetextpb "google.golang.org/protobuf/internal/testprotos/messageset/msetextpb"
|
|
)
|
|
|
|
func init() {
|
|
if flags.ProtoLegacy {
|
|
testValidMessages = append(testValidMessages, messageSetTestProtos...)
|
|
}
|
|
}
|
|
|
|
var messageSetTestProtos = []testProto{
|
|
{
|
|
desc: "MessageSet type_id before message content",
|
|
decodeTo: []proto.Message{func() proto.Message {
|
|
m := &messagesetpb.MessageSetContainer{MessageSet: &messagesetpb.MessageSet{}}
|
|
proto.SetExtension(m.MessageSet, msetextpb.E_Ext1_MessageSetExtension, &msetextpb.Ext1{
|
|
Ext1Field1: proto.Int32(10),
|
|
})
|
|
return m
|
|
}()},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1000),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(10),
|
|
}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}),
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet type_id after message content",
|
|
decodeTo: []proto.Message{func() proto.Message {
|
|
m := &messagesetpb.MessageSetContainer{MessageSet: &messagesetpb.MessageSet{}}
|
|
proto.SetExtension(m.MessageSet, msetextpb.E_Ext1_MessageSetExtension, &msetextpb.Ext1{
|
|
Ext1Field1: proto.Int32(10),
|
|
})
|
|
return m
|
|
}()},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(10),
|
|
}),
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1000),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}),
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet does not preserve unknown field",
|
|
decodeTo: []proto.Message{build(
|
|
&messagesetpb.MessageSet{},
|
|
extend(msetextpb.E_Ext1_MessageSetExtension, &msetextpb.Ext1{
|
|
Ext1Field1: proto.Int32(10),
|
|
}),
|
|
)},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1000),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(10),
|
|
}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
// Unknown field
|
|
pack.Tag{4, pack.VarintType}, pack.Varint(30),
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet with unknown type_id",
|
|
decodeTo: []proto.Message{build(
|
|
&messagesetpb.MessageSet{},
|
|
unknown(pack.Message{
|
|
pack.Tag{1002, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(10),
|
|
}),
|
|
}.Marshal()),
|
|
)},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1002),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(10),
|
|
}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet merges repeated message fields in item",
|
|
decodeTo: []proto.Message{build(
|
|
&messagesetpb.MessageSet{},
|
|
extend(msetextpb.E_Ext1_MessageSetExtension, &msetextpb.Ext1{
|
|
Ext1Field1: proto.Int32(10),
|
|
Ext1Field2: proto.Int32(20),
|
|
}),
|
|
)},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1000),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(10),
|
|
}),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(20),
|
|
}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet merges message fields in repeated items",
|
|
decodeTo: []proto.Message{build(
|
|
&messagesetpb.MessageSet{},
|
|
extend(msetextpb.E_Ext1_MessageSetExtension, &msetextpb.Ext1{
|
|
Ext1Field1: proto.Int32(10),
|
|
Ext1Field2: proto.Int32(20),
|
|
}),
|
|
extend(msetextpb.E_Ext2_MessageSetExtension, &msetextpb.Ext2{
|
|
Ext2Field1: proto.Int32(30),
|
|
}),
|
|
)},
|
|
wire: pack.Message{
|
|
// Ext1, field1
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1000),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(10),
|
|
}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
// Ext2, field1
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1001),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(30),
|
|
}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
// Ext2, field2
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1000),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(20),
|
|
}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet with missing type_id",
|
|
decodeTo: []proto.Message{build(
|
|
&messagesetpb.MessageSet{},
|
|
)},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.VarintType}, pack.Varint(10),
|
|
}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet with missing message",
|
|
decodeTo: []proto.Message{build(
|
|
&messagesetpb.MessageSet{},
|
|
extend(msetextpb.E_Ext1_MessageSetExtension, &msetextpb.Ext1{}),
|
|
)},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(1000),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet with type id out of valid field number range",
|
|
decodeTo: []proto.Message{func() proto.Message {
|
|
m := &messagesetpb.MessageSetContainer{MessageSet: &messagesetpb.MessageSet{}}
|
|
proto.SetExtension(m.MessageSet, msetextpb.E_ExtLargeNumber_MessageSetExtension, &msetextpb.ExtLargeNumber{})
|
|
return m
|
|
}()},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(wire.MaxValidNumber + 1),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}),
|
|
}.Marshal(),
|
|
},
|
|
{
|
|
desc: "MessageSet with unknown type id out of valid field number range",
|
|
decodeTo: []proto.Message{func() proto.Message {
|
|
m := &messagesetpb.MessageSetContainer{MessageSet: &messagesetpb.MessageSet{}}
|
|
m.MessageSet.ProtoReflect().SetUnknown(
|
|
pack.Message{
|
|
pack.Tag{wire.MaxValidNumber + 2, pack.BytesType}, pack.LengthPrefix{},
|
|
}.Marshal(),
|
|
)
|
|
return m
|
|
}()},
|
|
wire: pack.Message{
|
|
pack.Tag{1, pack.BytesType}, pack.LengthPrefix(pack.Message{
|
|
pack.Tag{1, pack.StartGroupType},
|
|
pack.Tag{2, pack.VarintType}, pack.Varint(wire.MaxValidNumber + 2),
|
|
pack.Tag{3, pack.BytesType}, pack.LengthPrefix(pack.Message{}),
|
|
pack.Tag{1, pack.EndGroupType},
|
|
}),
|
|
}.Marshal(),
|
|
},
|
|
}
|