mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 22:30:22 -04:00
1350 lines
40 KiB
Go
1350 lines
40 KiB
Go
![]() |
// Protocol Buffers for Go with Gadgets
|
||
|
//
|
||
|
// Copyright (c) 2013, The GoGo Authors. All rights reserved.
|
||
|
// http://github.com/gogo/protobuf
|
||
|
//
|
||
|
// Redistribution and use in source and binary forms, with or without
|
||
|
// modification, are permitted provided that the following conditions are
|
||
|
// met:
|
||
|
//
|
||
|
// * Redistributions of source code must retain the above copyright
|
||
|
// notice, this list of conditions and the following disclaimer.
|
||
|
// * Redistributions in binary form must reproduce the above
|
||
|
// copyright notice, this list of conditions and the following disclaimer
|
||
|
// in the documentation and/or other materials provided with the
|
||
|
// distribution.
|
||
|
//
|
||
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
||
|
/*
|
||
|
The unmarshal plugin generates a Unmarshal method for each message.
|
||
|
The `Unmarshal([]byte) error` method results in the fact that the message
|
||
|
implements the Unmarshaler interface.
|
||
|
The allows proto.Unmarshal to be faster by calling the generated Unmarshal method rather than using reflect.
|
||
|
|
||
|
If is enabled by the following extensions:
|
||
|
|
||
|
- unmarshaler
|
||
|
- unmarshaler_all
|
||
|
|
||
|
Or the following extensions:
|
||
|
|
||
|
- unsafe_unmarshaler
|
||
|
- unsafe_unmarshaler_all
|
||
|
|
||
|
That is if you want to use the unsafe package in your generated code.
|
||
|
The speed up using the unsafe package is not very significant.
|
||
|
|
||
|
The generation of unmarshalling tests are enabled using one of the following extensions:
|
||
|
|
||
|
- testgen
|
||
|
- testgen_all
|
||
|
|
||
|
And benchmarks given it is enabled using one of the following extensions:
|
||
|
|
||
|
- benchgen
|
||
|
- benchgen_all
|
||
|
|
||
|
Let us look at:
|
||
|
|
||
|
github.com/gogo/protobuf/test/example/example.proto
|
||
|
|
||
|
Btw all the output can be seen at:
|
||
|
|
||
|
github.com/gogo/protobuf/test/example/*
|
||
|
|
||
|
The following message:
|
||
|
|
||
|
option (gogoproto.unmarshaler_all) = true;
|
||
|
|
||
|
message B {
|
||
|
option (gogoproto.description) = true;
|
||
|
optional A A = 1 [(gogoproto.nullable) = false, (gogoproto.embed) = true];
|
||
|
repeated bytes G = 2 [(gogoproto.customtype) = "github.com/gogo/protobuf/test/custom.Uint128", (gogoproto.nullable) = false];
|
||
|
}
|
||
|
|
||
|
given to the unmarshal plugin, will generate the following code:
|
||
|
|
||
|
func (m *B) Unmarshal(dAtA []byte) error {
|
||
|
l := len(dAtA)
|
||
|
iNdEx := 0
|
||
|
for iNdEx < l {
|
||
|
var wire uint64
|
||
|
for shift := uint(0); ; shift += 7 {
|
||
|
if iNdEx >= l {
|
||
|
return io.ErrUnexpectedEOF
|
||
|
}
|
||
|
b := dAtA[iNdEx]
|
||
|
iNdEx++
|
||
|
wire |= (uint64(b) & 0x7F) << shift
|
||
|
if b < 0x80 {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
fieldNum := int32(wire >> 3)
|
||
|
wireType := int(wire & 0x7)
|
||
|
switch fieldNum {
|
||
|
case 1:
|
||
|
if wireType != 2 {
|
||
|
return proto.ErrWrongType
|
||
|
}
|
||
|
var msglen int
|
||
|
for shift := uint(0); ; shift += 7 {
|
||
|
if iNdEx >= l {
|
||
|
return io.ErrUnexpectedEOF
|
||
|
}
|
||
|
b := dAtA[iNdEx]
|
||
|
iNdEx++
|
||
|
msglen |= (int(b) & 0x7F) << shift
|
||
|
if b < 0x80 {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
postIndex := iNdEx + msglen
|
||
|
if postIndex > l {
|
||
|
return io.ErrUnexpectedEOF
|
||
|
}
|
||
|
if err := m.A.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
iNdEx = postIndex
|
||
|
case 2:
|
||
|
if wireType != 2 {
|
||
|
return proto.ErrWrongType
|
||
|
}
|
||
|
var byteLen int
|
||
|
for shift := uint(0); ; shift += 7 {
|
||
|
if iNdEx >= l {
|
||
|
return io.ErrUnexpectedEOF
|
||
|
}
|
||
|
b := dAtA[iNdEx]
|
||
|
iNdEx++
|
||
|
byteLen |= (int(b) & 0x7F) << shift
|
||
|
if b < 0x80 {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
postIndex := iNdEx + byteLen
|
||
|
if postIndex > l {
|
||
|
return io.ErrUnexpectedEOF
|
||
|
}
|
||
|
m.G = append(m.G, github_com_gogo_protobuf_test_custom.Uint128{})
|
||
|
if err := m.G[len(m.G)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
iNdEx = postIndex
|
||
|
default:
|
||
|
var sizeOfWire int
|
||
|
for {
|
||
|
sizeOfWire++
|
||
|
wire >>= 7
|
||
|
if wire == 0 {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
iNdEx -= sizeOfWire
|
||
|
skippy, err := skip(dAtA[iNdEx:])
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if (iNdEx + skippy) > l {
|
||
|
return io.ErrUnexpectedEOF
|
||
|
}
|
||
|
m.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)
|
||
|
iNdEx += skippy
|
||
|
}
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
Remember when using this code to call proto.Unmarshal.
|
||
|
This will call m.Reset and invoke the generated Unmarshal method for you.
|
||
|
If you call m.Unmarshal without m.Reset you could be merging protocol buffers.
|
||
|
|
||
|
*/
|
||
|
package unmarshal
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"strconv"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/gogo/protobuf/gogoproto"
|
||
|
"github.com/gogo/protobuf/proto"
|
||
|
descriptor "github.com/gogo/protobuf/protoc-gen-gogo/descriptor"
|
||
|
"github.com/gogo/protobuf/protoc-gen-gogo/generator"
|
||
|
)
|
||
|
|
||
|
type unmarshal struct {
|
||
|
*generator.Generator
|
||
|
generator.PluginImports
|
||
|
atleastOne bool
|
||
|
ioPkg generator.Single
|
||
|
mathPkg generator.Single
|
||
|
typesPkg generator.Single
|
||
|
binaryPkg generator.Single
|
||
|
localName string
|
||
|
}
|
||
|
|
||
|
func NewUnmarshal() *unmarshal {
|
||
|
return &unmarshal{}
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) Name() string {
|
||
|
return "unmarshal"
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) Init(g *generator.Generator) {
|
||
|
p.Generator = g
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) decodeVarint(varName string, typName string) {
|
||
|
p.P(`for shift := uint(0); ; shift += 7 {`)
|
||
|
p.In()
|
||
|
p.P(`if shift >= 64 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrIntOverflow` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`if iNdEx >= l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`b := dAtA[iNdEx]`)
|
||
|
p.P(`iNdEx++`)
|
||
|
p.P(varName, ` |= (`, typName, `(b) & 0x7F) << shift`)
|
||
|
p.P(`if b < 0x80 {`)
|
||
|
p.In()
|
||
|
p.P(`break`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) decodeFixed32(varName string, typeName string) {
|
||
|
p.P(`if (iNdEx+4) > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(varName, ` = `, typeName, `(`, p.binaryPkg.Use(), `.LittleEndian.Uint32(dAtA[iNdEx:]))`)
|
||
|
p.P(`iNdEx += 4`)
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) decodeFixed64(varName string, typeName string) {
|
||
|
p.P(`if (iNdEx+8) > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(varName, ` = `, typeName, `(`, p.binaryPkg.Use(), `.LittleEndian.Uint64(dAtA[iNdEx:]))`)
|
||
|
p.P(`iNdEx += 8`)
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) declareMapField(varName string, nullable bool, customType bool, field *descriptor.FieldDescriptorProto) {
|
||
|
switch field.GetType() {
|
||
|
case descriptor.FieldDescriptorProto_TYPE_DOUBLE:
|
||
|
p.P(`var `, varName, ` float64`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FLOAT:
|
||
|
p.P(`var `, varName, ` float32`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_INT64:
|
||
|
p.P(`var `, varName, ` int64`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_UINT64:
|
||
|
p.P(`var `, varName, ` uint64`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_INT32:
|
||
|
p.P(`var `, varName, ` int32`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FIXED64:
|
||
|
p.P(`var `, varName, ` uint64`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FIXED32:
|
||
|
p.P(`var `, varName, ` uint32`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_BOOL:
|
||
|
p.P(`var `, varName, ` bool`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_STRING:
|
||
|
cast, _ := p.GoType(nil, field)
|
||
|
cast = strings.Replace(cast, "*", "", 1)
|
||
|
p.P(`var `, varName, ` `, cast)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_MESSAGE:
|
||
|
if gogoproto.IsStdTime(field) {
|
||
|
p.P(varName, ` := new(time.Time)`)
|
||
|
} else if gogoproto.IsStdDuration(field) {
|
||
|
p.P(varName, ` := new(time.Duration)`)
|
||
|
} else {
|
||
|
desc := p.ObjectNamed(field.GetTypeName())
|
||
|
msgname := p.TypeName(desc)
|
||
|
if nullable {
|
||
|
p.P(`var `, varName, ` *`, msgname)
|
||
|
} else {
|
||
|
p.P(varName, ` := &`, msgname, `{}`)
|
||
|
}
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_BYTES:
|
||
|
if customType {
|
||
|
_, ctyp, err := generator.GetCustomType(field)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
p.P(`var `, varName, `1 `, ctyp)
|
||
|
p.P(`var `, varName, ` = &`, varName, `1`)
|
||
|
} else {
|
||
|
p.P(varName, ` := []byte{}`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_UINT32:
|
||
|
p.P(`var `, varName, ` uint32`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_ENUM:
|
||
|
typName := p.TypeName(p.ObjectNamed(field.GetTypeName()))
|
||
|
p.P(`var `, varName, ` `, typName)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SFIXED32:
|
||
|
p.P(`var `, varName, ` int32`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SFIXED64:
|
||
|
p.P(`var `, varName, ` int64`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SINT32:
|
||
|
p.P(`var `, varName, ` int32`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SINT64:
|
||
|
p.P(`var `, varName, ` int64`)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) mapField(varName string, customType bool, field *descriptor.FieldDescriptorProto) {
|
||
|
switch field.GetType() {
|
||
|
case descriptor.FieldDescriptorProto_TYPE_DOUBLE:
|
||
|
p.P(`var `, varName, `temp uint64`)
|
||
|
p.decodeFixed64(varName+"temp", "uint64")
|
||
|
p.P(varName, ` = `, p.mathPkg.Use(), `.Float64frombits(`, varName, `temp)`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FLOAT:
|
||
|
p.P(`var `, varName, `temp uint32`)
|
||
|
p.decodeFixed32(varName+"temp", "uint32")
|
||
|
p.P(varName, ` = `, p.mathPkg.Use(), `.Float32frombits(`, varName, `temp)`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_INT64:
|
||
|
p.decodeVarint(varName, "int64")
|
||
|
case descriptor.FieldDescriptorProto_TYPE_UINT64:
|
||
|
p.decodeVarint(varName, "uint64")
|
||
|
case descriptor.FieldDescriptorProto_TYPE_INT32:
|
||
|
p.decodeVarint(varName, "int32")
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FIXED64:
|
||
|
p.decodeFixed64(varName, "uint64")
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FIXED32:
|
||
|
p.decodeFixed32(varName, "uint32")
|
||
|
case descriptor.FieldDescriptorProto_TYPE_BOOL:
|
||
|
p.P(`var `, varName, `temp int`)
|
||
|
p.decodeVarint(varName+"temp", "int")
|
||
|
p.P(varName, ` = bool(`, varName, `temp != 0)`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_STRING:
|
||
|
p.P(`var stringLen`, varName, ` uint64`)
|
||
|
p.decodeVarint("stringLen"+varName, "uint64")
|
||
|
p.P(`intStringLen`, varName, ` := int(stringLen`, varName, `)`)
|
||
|
p.P(`if intStringLen`, varName, ` < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`postStringIndex`, varName, ` := iNdEx + intStringLen`, varName)
|
||
|
p.P(`if postStringIndex`, varName, ` > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
cast, _ := p.GoType(nil, field)
|
||
|
cast = strings.Replace(cast, "*", "", 1)
|
||
|
p.P(varName, ` = `, cast, `(dAtA[iNdEx:postStringIndex`, varName, `])`)
|
||
|
p.P(`iNdEx = postStringIndex`, varName)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_MESSAGE:
|
||
|
p.P(`var mapmsglen int`)
|
||
|
p.decodeVarint("mapmsglen", "int")
|
||
|
p.P(`if mapmsglen < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`postmsgIndex := iNdEx + mapmsglen`)
|
||
|
p.P(`if mapmsglen < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`if postmsgIndex > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
buf := `dAtA[iNdEx:postmsgIndex]`
|
||
|
if gogoproto.IsStdTime(field) {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdTimeUnmarshal(`, varName, `, `, buf, `); err != nil {`)
|
||
|
} else if gogoproto.IsStdDuration(field) {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdDurationUnmarshal(`, varName, `, `, buf, `); err != nil {`)
|
||
|
} else {
|
||
|
desc := p.ObjectNamed(field.GetTypeName())
|
||
|
msgname := p.TypeName(desc)
|
||
|
p.P(varName, ` = &`, msgname, `{}`)
|
||
|
p.P(`if err := `, varName, `.Unmarshal(`, buf, `); err != nil {`)
|
||
|
}
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`iNdEx = postmsgIndex`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_BYTES:
|
||
|
p.P(`var mapbyteLen uint64`)
|
||
|
p.decodeVarint("mapbyteLen", "uint64")
|
||
|
p.P(`intMapbyteLen := int(mapbyteLen)`)
|
||
|
p.P(`if intMapbyteLen < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`postbytesIndex := iNdEx + intMapbyteLen`)
|
||
|
p.P(`if postbytesIndex > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
if customType {
|
||
|
p.P(`if err := `, varName, `.Unmarshal(dAtA[iNdEx:postbytesIndex]); err != nil {`)
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
} else {
|
||
|
p.P(varName, ` = make([]byte, mapbyteLen)`)
|
||
|
p.P(`copy(`, varName, `, dAtA[iNdEx:postbytesIndex])`)
|
||
|
}
|
||
|
p.P(`iNdEx = postbytesIndex`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_UINT32:
|
||
|
p.decodeVarint(varName, "uint32")
|
||
|
case descriptor.FieldDescriptorProto_TYPE_ENUM:
|
||
|
typName := p.TypeName(p.ObjectNamed(field.GetTypeName()))
|
||
|
p.decodeVarint(varName, typName)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SFIXED32:
|
||
|
p.decodeFixed32(varName, "int32")
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SFIXED64:
|
||
|
p.decodeFixed64(varName, "int64")
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SINT32:
|
||
|
p.P(`var `, varName, `temp int32`)
|
||
|
p.decodeVarint(varName+"temp", "int32")
|
||
|
p.P(varName, `temp = int32((uint32(`, varName, `temp) >> 1) ^ uint32(((`, varName, `temp&1)<<31)>>31))`)
|
||
|
p.P(varName, ` = int32(`, varName, `temp)`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SINT64:
|
||
|
p.P(`var `, varName, `temp uint64`)
|
||
|
p.decodeVarint(varName+"temp", "uint64")
|
||
|
p.P(varName, `temp = (`, varName, `temp >> 1) ^ uint64((int64(`, varName, `temp&1)<<63)>>63)`)
|
||
|
p.P(varName, ` = int64(`, varName, `temp)`)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) noStarOrSliceType(msg *generator.Descriptor, field *descriptor.FieldDescriptorProto) string {
|
||
|
typ, _ := p.GoType(msg, field)
|
||
|
if typ[0] == '*' {
|
||
|
return typ[1:]
|
||
|
}
|
||
|
if typ[0] == '[' && typ[1] == ']' {
|
||
|
return typ[2:]
|
||
|
}
|
||
|
return typ
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) field(file *generator.FileDescriptor, msg *generator.Descriptor, field *descriptor.FieldDescriptorProto, fieldname string, proto3 bool) {
|
||
|
repeated := field.IsRepeated()
|
||
|
nullable := gogoproto.IsNullable(field)
|
||
|
typ := p.noStarOrSliceType(msg, field)
|
||
|
oneof := field.OneofIndex != nil
|
||
|
switch *field.Type {
|
||
|
case descriptor.FieldDescriptorProto_TYPE_DOUBLE:
|
||
|
p.P(`var v uint64`)
|
||
|
p.decodeFixed64("v", "uint64")
|
||
|
if oneof {
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{`, typ, "(", p.mathPkg.Use(), `.Float64frombits(v))}`)
|
||
|
} else if repeated {
|
||
|
p.P(`v2 := `, typ, "(", p.mathPkg.Use(), `.Float64frombits(v))`)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v2)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = `, typ, "(", p.mathPkg.Use(), `.Float64frombits(v))`)
|
||
|
} else {
|
||
|
p.P(`v2 := `, typ, "(", p.mathPkg.Use(), `.Float64frombits(v))`)
|
||
|
p.P(`m.`, fieldname, ` = &v2`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FLOAT:
|
||
|
p.P(`var v uint32`)
|
||
|
p.decodeFixed32("v", "uint32")
|
||
|
if oneof {
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{`, typ, "(", p.mathPkg.Use(), `.Float32frombits(v))}`)
|
||
|
} else if repeated {
|
||
|
p.P(`v2 := `, typ, "(", p.mathPkg.Use(), `.Float32frombits(v))`)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v2)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = `, typ, "(", p.mathPkg.Use(), `.Float32frombits(v))`)
|
||
|
} else {
|
||
|
p.P(`v2 := `, typ, "(", p.mathPkg.Use(), `.Float32frombits(v))`)
|
||
|
p.P(`m.`, fieldname, ` = &v2`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_INT64:
|
||
|
if oneof {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeVarint("m."+fieldname, typ)
|
||
|
} else {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_UINT64:
|
||
|
if oneof {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeVarint("m."+fieldname, typ)
|
||
|
} else {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_INT32:
|
||
|
if oneof {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeVarint("m."+fieldname, typ)
|
||
|
} else {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FIXED64:
|
||
|
if oneof {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed64("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed64("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeFixed64("m."+fieldname, typ)
|
||
|
} else {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed64("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_FIXED32:
|
||
|
if oneof {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed32("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed32("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeFixed32("m."+fieldname, typ)
|
||
|
} else {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed32("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_BOOL:
|
||
|
p.P(`var v int`)
|
||
|
p.decodeVarint("v", "int")
|
||
|
if oneof {
|
||
|
p.P(`b := `, typ, `(v != 0)`)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{b}`)
|
||
|
} else if repeated {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, `, typ, `(v != 0))`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = `, typ, `(v != 0)`)
|
||
|
} else {
|
||
|
p.P(`b := `, typ, `(v != 0)`)
|
||
|
p.P(`m.`, fieldname, ` = &b`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_STRING:
|
||
|
p.P(`var stringLen uint64`)
|
||
|
p.decodeVarint("stringLen", "uint64")
|
||
|
p.P(`intStringLen := int(stringLen)`)
|
||
|
p.P(`if intStringLen < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`postIndex := iNdEx + intStringLen`)
|
||
|
p.P(`if postIndex > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
if oneof {
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{`, typ, `(dAtA[iNdEx:postIndex])}`)
|
||
|
} else if repeated {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, `, typ, `(dAtA[iNdEx:postIndex]))`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = `, typ, `(dAtA[iNdEx:postIndex])`)
|
||
|
} else {
|
||
|
p.P(`s := `, typ, `(dAtA[iNdEx:postIndex])`)
|
||
|
p.P(`m.`, fieldname, ` = &s`)
|
||
|
}
|
||
|
p.P(`iNdEx = postIndex`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_GROUP:
|
||
|
panic(fmt.Errorf("unmarshaler does not support group %v", fieldname))
|
||
|
case descriptor.FieldDescriptorProto_TYPE_MESSAGE:
|
||
|
desc := p.ObjectNamed(field.GetTypeName())
|
||
|
msgname := p.TypeName(desc)
|
||
|
p.P(`var msglen int`)
|
||
|
p.decodeVarint("msglen", "int")
|
||
|
p.P(`if msglen < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`postIndex := iNdEx + msglen`)
|
||
|
p.P(`if postIndex > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
if oneof {
|
||
|
buf := `dAtA[iNdEx:postIndex]`
|
||
|
if gogoproto.IsStdTime(field) {
|
||
|
if nullable {
|
||
|
p.P(`v := new(time.Time)`)
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdTimeUnmarshal(v, `, buf, `); err != nil {`)
|
||
|
} else {
|
||
|
p.P(`v := time.Time{}`)
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdTimeUnmarshal(&v, `, buf, `); err != nil {`)
|
||
|
}
|
||
|
} else if gogoproto.IsStdDuration(field) {
|
||
|
if nullable {
|
||
|
p.P(`v := new(time.Duration)`)
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdDurationUnmarshal(v, `, buf, `); err != nil {`)
|
||
|
} else {
|
||
|
p.P(`v := time.Duration(0)`)
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdDurationUnmarshal(&v, `, buf, `); err != nil {`)
|
||
|
}
|
||
|
} else {
|
||
|
p.P(`v := &`, msgname, `{}`)
|
||
|
p.P(`if err := v.Unmarshal(`, buf, `); err != nil {`)
|
||
|
}
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if p.IsMap(field) {
|
||
|
m := p.GoMapType(nil, field)
|
||
|
|
||
|
keygoTyp, _ := p.GoType(nil, m.KeyField)
|
||
|
keygoAliasTyp, _ := p.GoType(nil, m.KeyAliasField)
|
||
|
// keys may not be pointers
|
||
|
keygoTyp = strings.Replace(keygoTyp, "*", "", 1)
|
||
|
keygoAliasTyp = strings.Replace(keygoAliasTyp, "*", "", 1)
|
||
|
|
||
|
valuegoTyp, _ := p.GoType(nil, m.ValueField)
|
||
|
valuegoAliasTyp, _ := p.GoType(nil, m.ValueAliasField)
|
||
|
|
||
|
// if the map type is an alias and key or values are aliases (type Foo map[Bar]Baz),
|
||
|
// we need to explicitly record their use here.
|
||
|
if gogoproto.IsCastKey(field) {
|
||
|
p.RecordTypeUse(m.KeyAliasField.GetTypeName())
|
||
|
}
|
||
|
if gogoproto.IsCastValue(field) {
|
||
|
p.RecordTypeUse(m.ValueAliasField.GetTypeName())
|
||
|
}
|
||
|
|
||
|
nullable, valuegoTyp, valuegoAliasTyp = generator.GoMapValueTypes(field, m.ValueField, valuegoTyp, valuegoAliasTyp)
|
||
|
if gogoproto.IsStdTime(field) || gogoproto.IsStdDuration(field) {
|
||
|
valuegoTyp = valuegoAliasTyp
|
||
|
}
|
||
|
|
||
|
p.P(`if m.`, fieldname, ` == nil {`)
|
||
|
p.In()
|
||
|
p.P(`m.`, fieldname, ` = make(`, m.GoType, `)`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
|
||
|
p.declareMapField("mapkey", false, false, m.KeyAliasField)
|
||
|
p.declareMapField("mapvalue", nullable, gogoproto.IsCustomType(field), m.ValueAliasField)
|
||
|
p.P(`for iNdEx < postIndex {`)
|
||
|
p.In()
|
||
|
|
||
|
p.P(`entryPreIndex := iNdEx`)
|
||
|
p.P(`var wire uint64`)
|
||
|
p.decodeVarint("wire", "uint64")
|
||
|
p.P(`fieldNum := int32(wire >> 3)`)
|
||
|
|
||
|
p.P(`if fieldNum == 1 {`)
|
||
|
p.In()
|
||
|
p.mapField("mapkey", false, m.KeyAliasField)
|
||
|
p.Out()
|
||
|
p.P(`} else if fieldNum == 2 {`)
|
||
|
p.In()
|
||
|
p.mapField("mapvalue", gogoproto.IsCustomType(field), m.ValueAliasField)
|
||
|
p.Out()
|
||
|
p.P(`} else {`)
|
||
|
p.In()
|
||
|
p.P(`iNdEx = entryPreIndex`)
|
||
|
p.P(`skippy, err := skip`, p.localName, `(dAtA[iNdEx:])`)
|
||
|
p.P(`if err != nil {`)
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`if skippy < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength`, p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`if (iNdEx + skippy) > postIndex {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`iNdEx += skippy`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
|
||
|
s := `m.` + fieldname
|
||
|
if keygoTyp == keygoAliasTyp {
|
||
|
s += `[mapkey]`
|
||
|
} else {
|
||
|
s += `[` + keygoAliasTyp + `(mapkey)]`
|
||
|
}
|
||
|
|
||
|
v := `mapvalue`
|
||
|
if (m.ValueField.IsMessage() || gogoproto.IsCustomType(field)) && !nullable {
|
||
|
v = `*` + v
|
||
|
}
|
||
|
if valuegoTyp != valuegoAliasTyp {
|
||
|
v = `((` + valuegoAliasTyp + `)(` + v + `))`
|
||
|
}
|
||
|
|
||
|
p.P(s, ` = `, v)
|
||
|
} else if repeated {
|
||
|
if gogoproto.IsStdTime(field) {
|
||
|
if nullable {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, new(time.Time))`)
|
||
|
} else {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, time.Time{})`)
|
||
|
}
|
||
|
} else if gogoproto.IsStdDuration(field) {
|
||
|
if nullable {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, new(time.Duration))`)
|
||
|
} else {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, time.Duration(0))`)
|
||
|
}
|
||
|
} else if nullable && !gogoproto.IsCustomType(field) {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, &`, msgname, `{})`)
|
||
|
} else {
|
||
|
goType, _ := p.GoType(nil, field)
|
||
|
// remove the slice from the type, i.e. []*T -> *T
|
||
|
goType = goType[2:]
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, `, goType, `{})`)
|
||
|
}
|
||
|
varName := `m.` + fieldname + `[len(m.` + fieldname + `)-1]`
|
||
|
buf := `dAtA[iNdEx:postIndex]`
|
||
|
if gogoproto.IsStdTime(field) {
|
||
|
if nullable {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdTimeUnmarshal(`, varName, `,`, buf, `); err != nil {`)
|
||
|
} else {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdTimeUnmarshal(&(`, varName, `),`, buf, `); err != nil {`)
|
||
|
}
|
||
|
} else if gogoproto.IsStdDuration(field) {
|
||
|
if nullable {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdDurationUnmarshal(`, varName, `,`, buf, `); err != nil {`)
|
||
|
} else {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdDurationUnmarshal(&(`, varName, `),`, buf, `); err != nil {`)
|
||
|
}
|
||
|
} else {
|
||
|
p.P(`if err := `, varName, `.Unmarshal(`, buf, `); err != nil {`)
|
||
|
}
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
} else if nullable {
|
||
|
p.P(`if m.`, fieldname, ` == nil {`)
|
||
|
p.In()
|
||
|
if gogoproto.IsStdTime(field) {
|
||
|
p.P(`m.`, fieldname, ` = new(time.Time)`)
|
||
|
} else if gogoproto.IsStdDuration(field) {
|
||
|
p.P(`m.`, fieldname, ` = new(time.Duration)`)
|
||
|
} else {
|
||
|
goType, _ := p.GoType(nil, field)
|
||
|
// remove the star from the type
|
||
|
p.P(`m.`, fieldname, ` = &`, goType[1:], `{}`)
|
||
|
}
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
if gogoproto.IsStdTime(field) {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdTimeUnmarshal(m.`, fieldname, `, dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
} else if gogoproto.IsStdDuration(field) {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdDurationUnmarshal(m.`, fieldname, `, dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
} else {
|
||
|
p.P(`if err := m.`, fieldname, `.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
}
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
} else {
|
||
|
if gogoproto.IsStdTime(field) {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdTimeUnmarshal(&m.`, fieldname, `, dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
} else if gogoproto.IsStdDuration(field) {
|
||
|
p.P(`if err := `, p.typesPkg.Use(), `.StdDurationUnmarshal(&m.`, fieldname, `, dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
} else {
|
||
|
p.P(`if err := m.`, fieldname, `.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
}
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
}
|
||
|
p.P(`iNdEx = postIndex`)
|
||
|
|
||
|
case descriptor.FieldDescriptorProto_TYPE_BYTES:
|
||
|
p.P(`var byteLen int`)
|
||
|
p.decodeVarint("byteLen", "int")
|
||
|
p.P(`if byteLen < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`postIndex := iNdEx + byteLen`)
|
||
|
p.P(`if postIndex > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
if !gogoproto.IsCustomType(field) {
|
||
|
if oneof {
|
||
|
p.P(`v := make([]byte, postIndex-iNdEx)`)
|
||
|
p.P(`copy(v, dAtA[iNdEx:postIndex])`)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, make([]byte, postIndex-iNdEx))`)
|
||
|
p.P(`copy(m.`, fieldname, `[len(m.`, fieldname, `)-1], dAtA[iNdEx:postIndex])`)
|
||
|
} else {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `[:0] , dAtA[iNdEx:postIndex]...)`)
|
||
|
p.P(`if m.`, fieldname, ` == nil {`)
|
||
|
p.In()
|
||
|
p.P(`m.`, fieldname, ` = []byte{}`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
}
|
||
|
} else {
|
||
|
_, ctyp, err := generator.GetCustomType(field)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
if oneof {
|
||
|
p.P(`var vv `, ctyp)
|
||
|
p.P(`v := &vv`)
|
||
|
p.P(`if err := v.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{*v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, ctyp)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
p.P(`if err := m.`, fieldname, `[len(m.`, fieldname, `)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
} else if nullable {
|
||
|
p.P(`var v `, ctyp)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
p.P(`if err := m.`, fieldname, `.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
} else {
|
||
|
p.P(`if err := m.`, fieldname, `.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {`)
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
}
|
||
|
}
|
||
|
p.P(`iNdEx = postIndex`)
|
||
|
case descriptor.FieldDescriptorProto_TYPE_UINT32:
|
||
|
if oneof {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeVarint("m."+fieldname, typ)
|
||
|
} else {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_ENUM:
|
||
|
typName := p.TypeName(p.ObjectNamed(field.GetTypeName()))
|
||
|
if oneof {
|
||
|
p.P(`var v `, typName)
|
||
|
p.decodeVarint("v", typName)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typName)
|
||
|
p.decodeVarint("v", typName)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeVarint("m."+fieldname, typName)
|
||
|
} else {
|
||
|
p.P(`var v `, typName)
|
||
|
p.decodeVarint("v", typName)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SFIXED32:
|
||
|
if oneof {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed32("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed32("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeFixed32("m."+fieldname, typ)
|
||
|
} else {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed32("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SFIXED64:
|
||
|
if oneof {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed64("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed64("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = 0`)
|
||
|
p.decodeFixed64("m."+fieldname, typ)
|
||
|
} else {
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeFixed64("v", typ)
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SINT32:
|
||
|
p.P(`var v `, typ)
|
||
|
p.decodeVarint("v", typ)
|
||
|
p.P(`v = `, typ, `((uint32(v) >> 1) ^ uint32(((v&1)<<31)>>31))`)
|
||
|
if oneof {
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{v}`)
|
||
|
} else if repeated {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = v`)
|
||
|
} else {
|
||
|
p.P(`m.`, fieldname, ` = &v`)
|
||
|
}
|
||
|
case descriptor.FieldDescriptorProto_TYPE_SINT64:
|
||
|
p.P(`var v uint64`)
|
||
|
p.decodeVarint("v", "uint64")
|
||
|
p.P(`v = (v >> 1) ^ uint64((int64(v&1)<<63)>>63)`)
|
||
|
if oneof {
|
||
|
p.P(`m.`, fieldname, ` = &`, p.OneOfTypeName(msg, field), `{`, typ, `(v)}`)
|
||
|
} else if repeated {
|
||
|
p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, `, typ, `(v))`)
|
||
|
} else if proto3 || !nullable {
|
||
|
p.P(`m.`, fieldname, ` = `, typ, `(v)`)
|
||
|
} else {
|
||
|
p.P(`v2 := `, typ, `(v)`)
|
||
|
p.P(`m.`, fieldname, ` = &v2`)
|
||
|
}
|
||
|
default:
|
||
|
panic("not implemented")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (p *unmarshal) Generate(file *generator.FileDescriptor) {
|
||
|
proto3 := gogoproto.IsProto3(file.FileDescriptorProto)
|
||
|
p.PluginImports = generator.NewPluginImports(p.Generator)
|
||
|
p.atleastOne = false
|
||
|
p.localName = generator.FileName(file)
|
||
|
|
||
|
p.ioPkg = p.NewImport("io")
|
||
|
p.mathPkg = p.NewImport("math")
|
||
|
p.typesPkg = p.NewImport("github.com/gogo/protobuf/types")
|
||
|
p.binaryPkg = p.NewImport("encoding/binary")
|
||
|
fmtPkg := p.NewImport("fmt")
|
||
|
protoPkg := p.NewImport("github.com/gogo/protobuf/proto")
|
||
|
if !gogoproto.ImportsGoGoProto(file.FileDescriptorProto) {
|
||
|
protoPkg = p.NewImport("github.com/golang/protobuf/proto")
|
||
|
}
|
||
|
|
||
|
for _, message := range file.Messages() {
|
||
|
ccTypeName := generator.CamelCaseSlice(message.TypeName())
|
||
|
if !gogoproto.IsUnmarshaler(file.FileDescriptorProto, message.DescriptorProto) &&
|
||
|
!gogoproto.IsUnsafeUnmarshaler(file.FileDescriptorProto, message.DescriptorProto) {
|
||
|
continue
|
||
|
}
|
||
|
if message.DescriptorProto.GetOptions().GetMapEntry() {
|
||
|
continue
|
||
|
}
|
||
|
p.atleastOne = true
|
||
|
|
||
|
// build a map required field_id -> bitmask offset
|
||
|
rfMap := make(map[int32]uint)
|
||
|
rfNextId := uint(0)
|
||
|
for _, field := range message.Field {
|
||
|
if field.IsRequired() {
|
||
|
rfMap[field.GetNumber()] = rfNextId
|
||
|
rfNextId++
|
||
|
}
|
||
|
}
|
||
|
rfCount := len(rfMap)
|
||
|
|
||
|
p.P(`func (m *`, ccTypeName, `) Unmarshal(dAtA []byte) error {`)
|
||
|
p.In()
|
||
|
if rfCount > 0 {
|
||
|
p.P(`var hasFields [`, strconv.Itoa(1+(rfCount-1)/64), `]uint64`)
|
||
|
}
|
||
|
p.P(`l := len(dAtA)`)
|
||
|
p.P(`iNdEx := 0`)
|
||
|
p.P(`for iNdEx < l {`)
|
||
|
p.In()
|
||
|
p.P(`preIndex := iNdEx`)
|
||
|
p.P(`var wire uint64`)
|
||
|
p.decodeVarint("wire", "uint64")
|
||
|
p.P(`fieldNum := int32(wire >> 3)`)
|
||
|
if len(message.Field) > 0 || !message.IsGroup() {
|
||
|
p.P(`wireType := int(wire & 0x7)`)
|
||
|
}
|
||
|
if !message.IsGroup() {
|
||
|
p.P(`if wireType == `, strconv.Itoa(proto.WireEndGroup), ` {`)
|
||
|
p.In()
|
||
|
p.P(`return `, fmtPkg.Use(), `.Errorf("proto: `+message.GetName()+`: wiretype end group for non-group")`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
}
|
||
|
p.P(`if fieldNum <= 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return `, fmtPkg.Use(), `.Errorf("proto: `+message.GetName()+`: illegal tag %d (wire type %d)", fieldNum, wire)`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`switch fieldNum {`)
|
||
|
p.In()
|
||
|
for _, field := range message.Field {
|
||
|
fieldname := p.GetFieldName(message, field)
|
||
|
errFieldname := fieldname
|
||
|
if field.OneofIndex != nil {
|
||
|
errFieldname = p.GetOneOfFieldName(message, field)
|
||
|
}
|
||
|
possiblyPacked := field.IsScalar() && field.IsRepeated()
|
||
|
p.P(`case `, strconv.Itoa(int(field.GetNumber())), `:`)
|
||
|
p.In()
|
||
|
wireType := field.WireType()
|
||
|
if possiblyPacked {
|
||
|
p.P(`if wireType == `, strconv.Itoa(wireType), `{`)
|
||
|
p.In()
|
||
|
p.field(file, message, field, fieldname, false)
|
||
|
p.Out()
|
||
|
p.P(`} else if wireType == `, strconv.Itoa(proto.WireBytes), `{`)
|
||
|
p.In()
|
||
|
p.P(`var packedLen int`)
|
||
|
p.decodeVarint("packedLen", "int")
|
||
|
p.P(`if packedLen < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength` + p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`postIndex := iNdEx + packedLen`)
|
||
|
p.P(`if postIndex > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`for iNdEx < postIndex {`)
|
||
|
p.In()
|
||
|
p.field(file, message, field, fieldname, false)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.Out()
|
||
|
p.P(`} else {`)
|
||
|
p.In()
|
||
|
p.P(`return ` + fmtPkg.Use() + `.Errorf("proto: wrong wireType = %d for field ` + errFieldname + `", wireType)`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
} else {
|
||
|
p.P(`if wireType != `, strconv.Itoa(wireType), `{`)
|
||
|
p.In()
|
||
|
p.P(`return ` + fmtPkg.Use() + `.Errorf("proto: wrong wireType = %d for field ` + errFieldname + `", wireType)`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.field(file, message, field, fieldname, proto3)
|
||
|
}
|
||
|
|
||
|
if field.IsRequired() {
|
||
|
fieldBit, ok := rfMap[field.GetNumber()]
|
||
|
if !ok {
|
||
|
panic("field is required, but no bit registered")
|
||
|
}
|
||
|
p.P(`hasFields[`, strconv.Itoa(int(fieldBit/64)), `] |= uint64(`, fmt.Sprintf("0x%08x", 1<<(fieldBit%64)), `)`)
|
||
|
}
|
||
|
}
|
||
|
p.Out()
|
||
|
p.P(`default:`)
|
||
|
p.In()
|
||
|
if message.DescriptorProto.HasExtension() {
|
||
|
c := []string{}
|
||
|
for _, erange := range message.GetExtensionRange() {
|
||
|
c = append(c, `((fieldNum >= `+strconv.Itoa(int(erange.GetStart()))+") && (fieldNum<"+strconv.Itoa(int(erange.GetEnd()))+`))`)
|
||
|
}
|
||
|
p.P(`if `, strings.Join(c, "||"), `{`)
|
||
|
p.In()
|
||
|
p.P(`var sizeOfWire int`)
|
||
|
p.P(`for {`)
|
||
|
p.In()
|
||
|
p.P(`sizeOfWire++`)
|
||
|
p.P(`wire >>= 7`)
|
||
|
p.P(`if wire == 0 {`)
|
||
|
p.In()
|
||
|
p.P(`break`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`iNdEx-=sizeOfWire`)
|
||
|
p.P(`skippy, err := skip`, p.localName+`(dAtA[iNdEx:])`)
|
||
|
p.P(`if err != nil {`)
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`if skippy < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength`, p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`if (iNdEx + skippy) > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(protoPkg.Use(), `.AppendExtension(m, int32(fieldNum), dAtA[iNdEx:iNdEx+skippy])`)
|
||
|
p.P(`iNdEx += skippy`)
|
||
|
p.Out()
|
||
|
p.P(`} else {`)
|
||
|
p.In()
|
||
|
}
|
||
|
p.P(`iNdEx=preIndex`)
|
||
|
p.P(`skippy, err := skip`, p.localName, `(dAtA[iNdEx:])`)
|
||
|
p.P(`if err != nil {`)
|
||
|
p.In()
|
||
|
p.P(`return err`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`if skippy < 0 {`)
|
||
|
p.In()
|
||
|
p.P(`return ErrInvalidLength`, p.localName)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`if (iNdEx + skippy) > l {`)
|
||
|
p.In()
|
||
|
p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
if gogoproto.HasUnrecognized(file.FileDescriptorProto, message.DescriptorProto) {
|
||
|
p.P(`m.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)`)
|
||
|
}
|
||
|
p.P(`iNdEx += skippy`)
|
||
|
p.Out()
|
||
|
if message.DescriptorProto.HasExtension() {
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
}
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
|
||
|
for _, field := range message.Field {
|
||
|
if !field.IsRequired() {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
fieldBit, ok := rfMap[field.GetNumber()]
|
||
|
if !ok {
|
||
|
panic("field is required, but no bit registered")
|
||
|
}
|
||
|
|
||
|
p.P(`if hasFields[`, strconv.Itoa(int(fieldBit/64)), `] & uint64(`, fmt.Sprintf("0x%08x", 1<<(fieldBit%64)), `) == 0 {`)
|
||
|
p.In()
|
||
|
if !gogoproto.ImportsGoGoProto(file.FileDescriptorProto) {
|
||
|
p.P(`return new(`, protoPkg.Use(), `.RequiredNotSetError)`)
|
||
|
} else {
|
||
|
p.P(`return `, protoPkg.Use(), `.NewRequiredNotSetError("`, field.GetName(), `")`)
|
||
|
}
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
}
|
||
|
p.P()
|
||
|
p.P(`if iNdEx > l {`)
|
||
|
p.In()
|
||
|
p.P(`return ` + p.ioPkg.Use() + `.ErrUnexpectedEOF`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
p.P(`return nil`)
|
||
|
p.Out()
|
||
|
p.P(`}`)
|
||
|
}
|
||
|
if !p.atleastOne {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
p.P(`func skip` + p.localName + `(dAtA []byte) (n int, err error) {
|
||
|
l := len(dAtA)
|
||
|
iNdEx := 0
|
||
|
for iNdEx < l {
|
||
|
var wire uint64
|
||
|
for shift := uint(0); ; shift += 7 {
|
||
|
if shift >= 64 {
|
||
|
return 0, ErrIntOverflow` + p.localName + `
|
||
|
}
|
||
|
if iNdEx >= l {
|
||
|
return 0, ` + p.ioPkg.Use() + `.ErrUnexpectedEOF
|
||
|
}
|
||
|
b := dAtA[iNdEx]
|
||
|
iNdEx++
|
||
|
wire |= (uint64(b) & 0x7F) << shift
|
||
|
if b < 0x80 {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
wireType := int(wire & 0x7)
|
||
|
switch wireType {
|
||
|
case 0:
|
||
|
for shift := uint(0); ; shift += 7 {
|
||
|
if shift >= 64 {
|
||
|
return 0, ErrIntOverflow` + p.localName + `
|
||
|
}
|
||
|
if iNdEx >= l {
|
||
|
return 0, ` + p.ioPkg.Use() + `.ErrUnexpectedEOF
|
||
|
}
|
||
|
iNdEx++
|
||
|
if dAtA[iNdEx-1] < 0x80 {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
return iNdEx, nil
|
||
|
case 1:
|
||
|
iNdEx += 8
|
||
|
return iNdEx, nil
|
||
|
case 2:
|
||
|
var length int
|
||
|
for shift := uint(0); ; shift += 7 {
|
||
|
if shift >= 64 {
|
||
|
return 0, ErrIntOverflow` + p.localName + `
|
||
|
}
|
||
|
if iNdEx >= l {
|
||
|
return 0, ` + p.ioPkg.Use() + `.ErrUnexpectedEOF
|
||
|
}
|
||
|
b := dAtA[iNdEx]
|
||
|
iNdEx++
|
||
|
length |= (int(b) & 0x7F) << shift
|
||
|
if b < 0x80 {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
iNdEx += length
|
||
|
if length < 0 {
|
||
|
return 0, ErrInvalidLength` + p.localName + `
|
||
|
}
|
||
|
return iNdEx, nil
|
||
|
case 3:
|
||
|
for {
|
||
|
var innerWire uint64
|
||
|
var start int = iNdEx
|
||
|
for shift := uint(0); ; shift += 7 {
|
||
|
if shift >= 64 {
|
||
|
return 0, ErrIntOverflow` + p.localName + `
|
||
|
}
|
||
|
if iNdEx >= l {
|
||
|
return 0, ` + p.ioPkg.Use() + `.ErrUnexpectedEOF
|
||
|
}
|
||
|
b := dAtA[iNdEx]
|
||
|
iNdEx++
|
||
|
innerWire |= (uint64(b) & 0x7F) << shift
|
||
|
if b < 0x80 {
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
innerWireType := int(innerWire & 0x7)
|
||
|
if innerWireType == 4 {
|
||
|
break
|
||
|
}
|
||
|
next, err := skip` + p.localName + `(dAtA[start:])
|
||
|
if err != nil {
|
||
|
return 0, err
|
||
|
}
|
||
|
iNdEx = start + next
|
||
|
}
|
||
|
return iNdEx, nil
|
||
|
case 4:
|
||
|
return iNdEx, nil
|
||
|
case 5:
|
||
|
iNdEx += 4
|
||
|
return iNdEx, nil
|
||
|
default:
|
||
|
return 0, ` + fmtPkg.Use() + `.Errorf("proto: illegal wireType %d", wireType)
|
||
|
}
|
||
|
}
|
||
|
panic("unreachable")
|
||
|
}
|
||
|
|
||
|
var (
|
||
|
ErrInvalidLength` + p.localName + ` = ` + fmtPkg.Use() + `.Errorf("proto: negative length found during unmarshaling")
|
||
|
ErrIntOverflow` + p.localName + ` = ` + fmtPkg.Use() + `.Errorf("proto: integer overflow")
|
||
|
)
|
||
|
`)
|
||
|
}
|
||
|
|
||
|
func init() {
|
||
|
generator.RegisterPlugin(NewUnmarshal())
|
||
|
}
|