Source file src/google.golang.org/protobuf/internal/filedesc/desc_lazy.go

Documentation: google.golang.org/protobuf/internal/filedesc

     1  // Copyright 2019 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package filedesc
     6  
     7  import (
     8  	"reflect"
     9  	"sync"
    10  
    11  	"google.golang.org/protobuf/encoding/protowire"
    12  	"google.golang.org/protobuf/internal/descopts"
    13  	"google.golang.org/protobuf/internal/genid"
    14  	"google.golang.org/protobuf/internal/strs"
    15  	"google.golang.org/protobuf/proto"
    16  	"google.golang.org/protobuf/reflect/protoreflect"
    17  )
    18  
    19  func (fd *File) lazyRawInit() {
    20  	fd.unmarshalFull(fd.builder.RawDescriptor)
    21  	fd.resolveMessages()
    22  	fd.resolveExtensions()
    23  	fd.resolveServices()
    24  }
    25  
    26  func (file *File) resolveMessages() {
    27  	var depIdx int32
    28  	for i := range file.allMessages {
    29  		md := &file.allMessages[i]
    30  
    31  		// Resolve message field dependencies.
    32  		for j := range md.L2.Fields.List {
    33  			fd := &md.L2.Fields.List[j]
    34  
    35  			// Weak fields are resolved upon actual use.
    36  			if fd.L1.IsWeak {
    37  				continue
    38  			}
    39  
    40  			// Resolve message field dependency.
    41  			switch fd.L1.Kind {
    42  			case protoreflect.EnumKind:
    43  				fd.L1.Enum = file.resolveEnumDependency(fd.L1.Enum, listFieldDeps, depIdx)
    44  				depIdx++
    45  			case protoreflect.MessageKind, protoreflect.GroupKind:
    46  				fd.L1.Message = file.resolveMessageDependency(fd.L1.Message, listFieldDeps, depIdx)
    47  				depIdx++
    48  			}
    49  
    50  			// Default is resolved here since it depends on Enum being resolved.
    51  			if v := fd.L1.Default.val; v.IsValid() {
    52  				fd.L1.Default = unmarshalDefault(v.Bytes(), fd.L1.Kind, file, fd.L1.Enum)
    53  			}
    54  		}
    55  	}
    56  }
    57  
    58  func (file *File) resolveExtensions() {
    59  	var depIdx int32
    60  	for i := range file.allExtensions {
    61  		xd := &file.allExtensions[i]
    62  
    63  		// Resolve extension field dependency.
    64  		switch xd.L1.Kind {
    65  		case protoreflect.EnumKind:
    66  			xd.L2.Enum = file.resolveEnumDependency(xd.L2.Enum, listExtDeps, depIdx)
    67  			depIdx++
    68  		case protoreflect.MessageKind, protoreflect.GroupKind:
    69  			xd.L2.Message = file.resolveMessageDependency(xd.L2.Message, listExtDeps, depIdx)
    70  			depIdx++
    71  		}
    72  
    73  		// Default is resolved here since it depends on Enum being resolved.
    74  		if v := xd.L2.Default.val; v.IsValid() {
    75  			xd.L2.Default = unmarshalDefault(v.Bytes(), xd.L1.Kind, file, xd.L2.Enum)
    76  		}
    77  	}
    78  }
    79  
    80  func (file *File) resolveServices() {
    81  	var depIdx int32
    82  	for i := range file.allServices {
    83  		sd := &file.allServices[i]
    84  
    85  		// Resolve method dependencies.
    86  		for j := range sd.L2.Methods.List {
    87  			md := &sd.L2.Methods.List[j]
    88  			md.L1.Input = file.resolveMessageDependency(md.L1.Input, listMethInDeps, depIdx)
    89  			md.L1.Output = file.resolveMessageDependency(md.L1.Output, listMethOutDeps, depIdx)
    90  			depIdx++
    91  		}
    92  	}
    93  }
    94  
    95  func (file *File) resolveEnumDependency(ed protoreflect.EnumDescriptor, i, j int32) protoreflect.EnumDescriptor {
    96  	r := file.builder.FileRegistry
    97  	if r, ok := r.(resolverByIndex); ok {
    98  		if ed2 := r.FindEnumByIndex(i, j, file.allEnums, file.allMessages); ed2 != nil {
    99  			return ed2
   100  		}
   101  	}
   102  	for i := range file.allEnums {
   103  		if ed2 := &file.allEnums[i]; ed2.L0.FullName == ed.FullName() {
   104  			return ed2
   105  		}
   106  	}
   107  	if d, _ := r.FindDescriptorByName(ed.FullName()); d != nil {
   108  		return d.(protoreflect.EnumDescriptor)
   109  	}
   110  	return ed
   111  }
   112  
   113  func (file *File) resolveMessageDependency(md protoreflect.MessageDescriptor, i, j int32) protoreflect.MessageDescriptor {
   114  	r := file.builder.FileRegistry
   115  	if r, ok := r.(resolverByIndex); ok {
   116  		if md2 := r.FindMessageByIndex(i, j, file.allEnums, file.allMessages); md2 != nil {
   117  			return md2
   118  		}
   119  	}
   120  	for i := range file.allMessages {
   121  		if md2 := &file.allMessages[i]; md2.L0.FullName == md.FullName() {
   122  			return md2
   123  		}
   124  	}
   125  	if d, _ := r.FindDescriptorByName(md.FullName()); d != nil {
   126  		return d.(protoreflect.MessageDescriptor)
   127  	}
   128  	return md
   129  }
   130  
   131  func (fd *File) unmarshalFull(b []byte) {
   132  	sb := getBuilder()
   133  	defer putBuilder(sb)
   134  
   135  	var enumIdx, messageIdx, extensionIdx, serviceIdx int
   136  	var rawOptions []byte
   137  	fd.L2 = new(FileL2)
   138  	for len(b) > 0 {
   139  		num, typ, n := protowire.ConsumeTag(b)
   140  		b = b[n:]
   141  		switch typ {
   142  		case protowire.VarintType:
   143  			v, m := protowire.ConsumeVarint(b)
   144  			b = b[m:]
   145  			switch num {
   146  			case genid.FileDescriptorProto_PublicDependency_field_number:
   147  				fd.L2.Imports[v].IsPublic = true
   148  			case genid.FileDescriptorProto_WeakDependency_field_number:
   149  				fd.L2.Imports[v].IsWeak = true
   150  			}
   151  		case protowire.BytesType:
   152  			v, m := protowire.ConsumeBytes(b)
   153  			b = b[m:]
   154  			switch num {
   155  			case genid.FileDescriptorProto_Dependency_field_number:
   156  				path := sb.MakeString(v)
   157  				imp, _ := fd.builder.FileRegistry.FindFileByPath(path)
   158  				if imp == nil {
   159  					imp = PlaceholderFile(path)
   160  				}
   161  				fd.L2.Imports = append(fd.L2.Imports, protoreflect.FileImport{FileDescriptor: imp})
   162  			case genid.FileDescriptorProto_EnumType_field_number:
   163  				fd.L1.Enums.List[enumIdx].unmarshalFull(v, sb)
   164  				enumIdx++
   165  			case genid.FileDescriptorProto_MessageType_field_number:
   166  				fd.L1.Messages.List[messageIdx].unmarshalFull(v, sb)
   167  				messageIdx++
   168  			case genid.FileDescriptorProto_Extension_field_number:
   169  				fd.L1.Extensions.List[extensionIdx].unmarshalFull(v, sb)
   170  				extensionIdx++
   171  			case genid.FileDescriptorProto_Service_field_number:
   172  				fd.L1.Services.List[serviceIdx].unmarshalFull(v, sb)
   173  				serviceIdx++
   174  			case genid.FileDescriptorProto_Options_field_number:
   175  				rawOptions = appendOptions(rawOptions, v)
   176  			}
   177  		default:
   178  			m := protowire.ConsumeFieldValue(num, typ, b)
   179  			b = b[m:]
   180  		}
   181  	}
   182  	fd.L2.Options = fd.builder.optionsUnmarshaler(&descopts.File, rawOptions)
   183  }
   184  
   185  func (ed *Enum) unmarshalFull(b []byte, sb *strs.Builder) {
   186  	var rawValues [][]byte
   187  	var rawOptions []byte
   188  	if !ed.L1.eagerValues {
   189  		ed.L2 = new(EnumL2)
   190  	}
   191  	for len(b) > 0 {
   192  		num, typ, n := protowire.ConsumeTag(b)
   193  		b = b[n:]
   194  		switch typ {
   195  		case protowire.BytesType:
   196  			v, m := protowire.ConsumeBytes(b)
   197  			b = b[m:]
   198  			switch num {
   199  			case genid.EnumDescriptorProto_Value_field_number:
   200  				rawValues = append(rawValues, v)
   201  			case genid.EnumDescriptorProto_ReservedName_field_number:
   202  				ed.L2.ReservedNames.List = append(ed.L2.ReservedNames.List, protoreflect.Name(sb.MakeString(v)))
   203  			case genid.EnumDescriptorProto_ReservedRange_field_number:
   204  				ed.L2.ReservedRanges.List = append(ed.L2.ReservedRanges.List, unmarshalEnumReservedRange(v))
   205  			case genid.EnumDescriptorProto_Options_field_number:
   206  				rawOptions = appendOptions(rawOptions, v)
   207  			}
   208  		default:
   209  			m := protowire.ConsumeFieldValue(num, typ, b)
   210  			b = b[m:]
   211  		}
   212  	}
   213  	if !ed.L1.eagerValues && len(rawValues) > 0 {
   214  		ed.L2.Values.List = make([]EnumValue, len(rawValues))
   215  		for i, b := range rawValues {
   216  			ed.L2.Values.List[i].unmarshalFull(b, sb, ed.L0.ParentFile, ed, i)
   217  		}
   218  	}
   219  	ed.L2.Options = ed.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Enum, rawOptions)
   220  }
   221  
   222  func unmarshalEnumReservedRange(b []byte) (r [2]protoreflect.EnumNumber) {
   223  	for len(b) > 0 {
   224  		num, typ, n := protowire.ConsumeTag(b)
   225  		b = b[n:]
   226  		switch typ {
   227  		case protowire.VarintType:
   228  			v, m := protowire.ConsumeVarint(b)
   229  			b = b[m:]
   230  			switch num {
   231  			case genid.EnumDescriptorProto_EnumReservedRange_Start_field_number:
   232  				r[0] = protoreflect.EnumNumber(v)
   233  			case genid.EnumDescriptorProto_EnumReservedRange_End_field_number:
   234  				r[1] = protoreflect.EnumNumber(v)
   235  			}
   236  		default:
   237  			m := protowire.ConsumeFieldValue(num, typ, b)
   238  			b = b[m:]
   239  		}
   240  	}
   241  	return r
   242  }
   243  
   244  func (vd *EnumValue) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd protoreflect.Descriptor, i int) {
   245  	vd.L0.ParentFile = pf
   246  	vd.L0.Parent = pd
   247  	vd.L0.Index = i
   248  
   249  	var rawOptions []byte
   250  	for len(b) > 0 {
   251  		num, typ, n := protowire.ConsumeTag(b)
   252  		b = b[n:]
   253  		switch typ {
   254  		case protowire.VarintType:
   255  			v, m := protowire.ConsumeVarint(b)
   256  			b = b[m:]
   257  			switch num {
   258  			case genid.EnumValueDescriptorProto_Number_field_number:
   259  				vd.L1.Number = protoreflect.EnumNumber(v)
   260  			}
   261  		case protowire.BytesType:
   262  			v, m := protowire.ConsumeBytes(b)
   263  			b = b[m:]
   264  			switch num {
   265  			case genid.EnumValueDescriptorProto_Name_field_number:
   266  				// NOTE: Enum values are in the same scope as the enum parent.
   267  				vd.L0.FullName = appendFullName(sb, pd.Parent().FullName(), v)
   268  			case genid.EnumValueDescriptorProto_Options_field_number:
   269  				rawOptions = appendOptions(rawOptions, v)
   270  			}
   271  		default:
   272  			m := protowire.ConsumeFieldValue(num, typ, b)
   273  			b = b[m:]
   274  		}
   275  	}
   276  	vd.L1.Options = pf.builder.optionsUnmarshaler(&descopts.EnumValue, rawOptions)
   277  }
   278  
   279  func (md *Message) unmarshalFull(b []byte, sb *strs.Builder) {
   280  	var rawFields, rawOneofs [][]byte
   281  	var enumIdx, messageIdx, extensionIdx int
   282  	var rawOptions []byte
   283  	md.L2 = new(MessageL2)
   284  	for len(b) > 0 {
   285  		num, typ, n := protowire.ConsumeTag(b)
   286  		b = b[n:]
   287  		switch typ {
   288  		case protowire.BytesType:
   289  			v, m := protowire.ConsumeBytes(b)
   290  			b = b[m:]
   291  			switch num {
   292  			case genid.DescriptorProto_Field_field_number:
   293  				rawFields = append(rawFields, v)
   294  			case genid.DescriptorProto_OneofDecl_field_number:
   295  				rawOneofs = append(rawOneofs, v)
   296  			case genid.DescriptorProto_ReservedName_field_number:
   297  				md.L2.ReservedNames.List = append(md.L2.ReservedNames.List, protoreflect.Name(sb.MakeString(v)))
   298  			case genid.DescriptorProto_ReservedRange_field_number:
   299  				md.L2.ReservedRanges.List = append(md.L2.ReservedRanges.List, unmarshalMessageReservedRange(v))
   300  			case genid.DescriptorProto_ExtensionRange_field_number:
   301  				r, rawOptions := unmarshalMessageExtensionRange(v)
   302  				opts := md.L0.ParentFile.builder.optionsUnmarshaler(&descopts.ExtensionRange, rawOptions)
   303  				md.L2.ExtensionRanges.List = append(md.L2.ExtensionRanges.List, r)
   304  				md.L2.ExtensionRangeOptions = append(md.L2.ExtensionRangeOptions, opts)
   305  			case genid.DescriptorProto_EnumType_field_number:
   306  				md.L1.Enums.List[enumIdx].unmarshalFull(v, sb)
   307  				enumIdx++
   308  			case genid.DescriptorProto_NestedType_field_number:
   309  				md.L1.Messages.List[messageIdx].unmarshalFull(v, sb)
   310  				messageIdx++
   311  			case genid.DescriptorProto_Extension_field_number:
   312  				md.L1.Extensions.List[extensionIdx].unmarshalFull(v, sb)
   313  				extensionIdx++
   314  			case genid.DescriptorProto_Options_field_number:
   315  				md.unmarshalOptions(v)
   316  				rawOptions = appendOptions(rawOptions, v)
   317  			}
   318  		default:
   319  			m := protowire.ConsumeFieldValue(num, typ, b)
   320  			b = b[m:]
   321  		}
   322  	}
   323  	if len(rawFields) > 0 || len(rawOneofs) > 0 {
   324  		md.L2.Fields.List = make([]Field, len(rawFields))
   325  		md.L2.Oneofs.List = make([]Oneof, len(rawOneofs))
   326  		for i, b := range rawFields {
   327  			fd := &md.L2.Fields.List[i]
   328  			fd.unmarshalFull(b, sb, md.L0.ParentFile, md, i)
   329  			if fd.L1.Cardinality == protoreflect.Required {
   330  				md.L2.RequiredNumbers.List = append(md.L2.RequiredNumbers.List, fd.L1.Number)
   331  			}
   332  		}
   333  		for i, b := range rawOneofs {
   334  			od := &md.L2.Oneofs.List[i]
   335  			od.unmarshalFull(b, sb, md.L0.ParentFile, md, i)
   336  		}
   337  	}
   338  	md.L2.Options = md.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Message, rawOptions)
   339  }
   340  
   341  func (md *Message) unmarshalOptions(b []byte) {
   342  	for len(b) > 0 {
   343  		num, typ, n := protowire.ConsumeTag(b)
   344  		b = b[n:]
   345  		switch typ {
   346  		case protowire.VarintType:
   347  			v, m := protowire.ConsumeVarint(b)
   348  			b = b[m:]
   349  			switch num {
   350  			case genid.MessageOptions_MapEntry_field_number:
   351  				md.L1.IsMapEntry = protowire.DecodeBool(v)
   352  			case genid.MessageOptions_MessageSetWireFormat_field_number:
   353  				md.L1.IsMessageSet = protowire.DecodeBool(v)
   354  			}
   355  		default:
   356  			m := protowire.ConsumeFieldValue(num, typ, b)
   357  			b = b[m:]
   358  		}
   359  	}
   360  }
   361  
   362  func unmarshalMessageReservedRange(b []byte) (r [2]protoreflect.FieldNumber) {
   363  	for len(b) > 0 {
   364  		num, typ, n := protowire.ConsumeTag(b)
   365  		b = b[n:]
   366  		switch typ {
   367  		case protowire.VarintType:
   368  			v, m := protowire.ConsumeVarint(b)
   369  			b = b[m:]
   370  			switch num {
   371  			case genid.DescriptorProto_ReservedRange_Start_field_number:
   372  				r[0] = protoreflect.FieldNumber(v)
   373  			case genid.DescriptorProto_ReservedRange_End_field_number:
   374  				r[1] = protoreflect.FieldNumber(v)
   375  			}
   376  		default:
   377  			m := protowire.ConsumeFieldValue(num, typ, b)
   378  			b = b[m:]
   379  		}
   380  	}
   381  	return r
   382  }
   383  
   384  func unmarshalMessageExtensionRange(b []byte) (r [2]protoreflect.FieldNumber, rawOptions []byte) {
   385  	for len(b) > 0 {
   386  		num, typ, n := protowire.ConsumeTag(b)
   387  		b = b[n:]
   388  		switch typ {
   389  		case protowire.VarintType:
   390  			v, m := protowire.ConsumeVarint(b)
   391  			b = b[m:]
   392  			switch num {
   393  			case genid.DescriptorProto_ExtensionRange_Start_field_number:
   394  				r[0] = protoreflect.FieldNumber(v)
   395  			case genid.DescriptorProto_ExtensionRange_End_field_number:
   396  				r[1] = protoreflect.FieldNumber(v)
   397  			}
   398  		case protowire.BytesType:
   399  			v, m := protowire.ConsumeBytes(b)
   400  			b = b[m:]
   401  			switch num {
   402  			case genid.DescriptorProto_ExtensionRange_Options_field_number:
   403  				rawOptions = appendOptions(rawOptions, v)
   404  			}
   405  		default:
   406  			m := protowire.ConsumeFieldValue(num, typ, b)
   407  			b = b[m:]
   408  		}
   409  	}
   410  	return r, rawOptions
   411  }
   412  
   413  func (fd *Field) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd protoreflect.Descriptor, i int) {
   414  	fd.L0.ParentFile = pf
   415  	fd.L0.Parent = pd
   416  	fd.L0.Index = i
   417  	fd.L1.EditionFeatures = featuresFromParentDesc(fd.Parent())
   418  
   419  	var rawTypeName []byte
   420  	var rawOptions []byte
   421  	for len(b) > 0 {
   422  		num, typ, n := protowire.ConsumeTag(b)
   423  		b = b[n:]
   424  		switch typ {
   425  		case protowire.VarintType:
   426  			v, m := protowire.ConsumeVarint(b)
   427  			b = b[m:]
   428  			switch num {
   429  			case genid.FieldDescriptorProto_Number_field_number:
   430  				fd.L1.Number = protoreflect.FieldNumber(v)
   431  			case genid.FieldDescriptorProto_Label_field_number:
   432  				fd.L1.Cardinality = protoreflect.Cardinality(v)
   433  			case genid.FieldDescriptorProto_Type_field_number:
   434  				fd.L1.Kind = protoreflect.Kind(v)
   435  			case genid.FieldDescriptorProto_OneofIndex_field_number:
   436  				// In Message.unmarshalFull, we allocate slices for both
   437  				// the field and oneof descriptors before unmarshaling either
   438  				// of them. This ensures pointers to slice elements are stable.
   439  				od := &pd.(*Message).L2.Oneofs.List[v]
   440  				od.L1.Fields.List = append(od.L1.Fields.List, fd)
   441  				if fd.L1.ContainingOneof != nil {
   442  					panic("oneof type already set")
   443  				}
   444  				fd.L1.ContainingOneof = od
   445  			case genid.FieldDescriptorProto_Proto3Optional_field_number:
   446  				fd.L1.IsProto3Optional = protowire.DecodeBool(v)
   447  			}
   448  		case protowire.BytesType:
   449  			v, m := protowire.ConsumeBytes(b)
   450  			b = b[m:]
   451  			switch num {
   452  			case genid.FieldDescriptorProto_Name_field_number:
   453  				fd.L0.FullName = appendFullName(sb, pd.FullName(), v)
   454  			case genid.FieldDescriptorProto_JsonName_field_number:
   455  				fd.L1.StringName.InitJSON(sb.MakeString(v))
   456  			case genid.FieldDescriptorProto_DefaultValue_field_number:
   457  				fd.L1.Default.val = protoreflect.ValueOfBytes(v) // temporarily store as bytes; later resolved in resolveMessages
   458  			case genid.FieldDescriptorProto_TypeName_field_number:
   459  				rawTypeName = v
   460  			case genid.FieldDescriptorProto_Options_field_number:
   461  				fd.unmarshalOptions(v)
   462  				rawOptions = appendOptions(rawOptions, v)
   463  			}
   464  		default:
   465  			m := protowire.ConsumeFieldValue(num, typ, b)
   466  			b = b[m:]
   467  		}
   468  	}
   469  	if fd.L1.Kind == protoreflect.MessageKind && fd.L1.EditionFeatures.IsDelimitedEncoded {
   470  		fd.L1.Kind = protoreflect.GroupKind
   471  	}
   472  	if fd.L1.EditionFeatures.IsLegacyRequired {
   473  		fd.L1.Cardinality = protoreflect.Required
   474  	}
   475  	if rawTypeName != nil {
   476  		name := makeFullName(sb, rawTypeName)
   477  		switch fd.L1.Kind {
   478  		case protoreflect.EnumKind:
   479  			fd.L1.Enum = PlaceholderEnum(name)
   480  		case protoreflect.MessageKind, protoreflect.GroupKind:
   481  			fd.L1.Message = PlaceholderMessage(name)
   482  		}
   483  	}
   484  	fd.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Field, rawOptions)
   485  }
   486  
   487  func (fd *Field) unmarshalOptions(b []byte) {
   488  	const FieldOptions_EnforceUTF8 = 13
   489  
   490  	for len(b) > 0 {
   491  		num, typ, n := protowire.ConsumeTag(b)
   492  		b = b[n:]
   493  		switch typ {
   494  		case protowire.VarintType:
   495  			v, m := protowire.ConsumeVarint(b)
   496  			b = b[m:]
   497  			switch num {
   498  			case genid.FieldOptions_Packed_field_number:
   499  				fd.L1.EditionFeatures.IsPacked = protowire.DecodeBool(v)
   500  			case genid.FieldOptions_Weak_field_number:
   501  				fd.L1.IsWeak = protowire.DecodeBool(v)
   502  			case FieldOptions_EnforceUTF8:
   503  				fd.L1.EditionFeatures.IsUTF8Validated = protowire.DecodeBool(v)
   504  			}
   505  		case protowire.BytesType:
   506  			v, m := protowire.ConsumeBytes(b)
   507  			b = b[m:]
   508  			switch num {
   509  			case genid.FieldOptions_Features_field_number:
   510  				fd.L1.EditionFeatures = unmarshalFeatureSet(v, fd.L1.EditionFeatures)
   511  			}
   512  		default:
   513  			m := protowire.ConsumeFieldValue(num, typ, b)
   514  			b = b[m:]
   515  		}
   516  	}
   517  }
   518  
   519  func (od *Oneof) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd protoreflect.Descriptor, i int) {
   520  	od.L0.ParentFile = pf
   521  	od.L0.Parent = pd
   522  	od.L0.Index = i
   523  
   524  	var rawOptions []byte
   525  	for len(b) > 0 {
   526  		num, typ, n := protowire.ConsumeTag(b)
   527  		b = b[n:]
   528  		switch typ {
   529  		case protowire.BytesType:
   530  			v, m := protowire.ConsumeBytes(b)
   531  			b = b[m:]
   532  			switch num {
   533  			case genid.OneofDescriptorProto_Name_field_number:
   534  				od.L0.FullName = appendFullName(sb, pd.FullName(), v)
   535  			case genid.OneofDescriptorProto_Options_field_number:
   536  				rawOptions = appendOptions(rawOptions, v)
   537  			}
   538  		default:
   539  			m := protowire.ConsumeFieldValue(num, typ, b)
   540  			b = b[m:]
   541  		}
   542  	}
   543  	od.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Oneof, rawOptions)
   544  }
   545  
   546  func (xd *Extension) unmarshalFull(b []byte, sb *strs.Builder) {
   547  	var rawTypeName []byte
   548  	var rawOptions []byte
   549  	xd.L2 = new(ExtensionL2)
   550  	for len(b) > 0 {
   551  		num, typ, n := protowire.ConsumeTag(b)
   552  		b = b[n:]
   553  		switch typ {
   554  		case protowire.VarintType:
   555  			v, m := protowire.ConsumeVarint(b)
   556  			b = b[m:]
   557  			switch num {
   558  			case genid.FieldDescriptorProto_Proto3Optional_field_number:
   559  				xd.L2.IsProto3Optional = protowire.DecodeBool(v)
   560  			}
   561  		case protowire.BytesType:
   562  			v, m := protowire.ConsumeBytes(b)
   563  			b = b[m:]
   564  			switch num {
   565  			case genid.FieldDescriptorProto_JsonName_field_number:
   566  				xd.L2.StringName.InitJSON(sb.MakeString(v))
   567  			case genid.FieldDescriptorProto_DefaultValue_field_number:
   568  				xd.L2.Default.val = protoreflect.ValueOfBytes(v) // temporarily store as bytes; later resolved in resolveExtensions
   569  			case genid.FieldDescriptorProto_TypeName_field_number:
   570  				rawTypeName = v
   571  			case genid.FieldDescriptorProto_Options_field_number:
   572  				rawOptions = appendOptions(rawOptions, v)
   573  			}
   574  		default:
   575  			m := protowire.ConsumeFieldValue(num, typ, b)
   576  			b = b[m:]
   577  		}
   578  	}
   579  	if rawTypeName != nil {
   580  		name := makeFullName(sb, rawTypeName)
   581  		switch xd.L1.Kind {
   582  		case protoreflect.EnumKind:
   583  			xd.L2.Enum = PlaceholderEnum(name)
   584  		case protoreflect.MessageKind, protoreflect.GroupKind:
   585  			xd.L2.Message = PlaceholderMessage(name)
   586  		}
   587  	}
   588  	xd.L2.Options = xd.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Field, rawOptions)
   589  }
   590  
   591  func (sd *Service) unmarshalFull(b []byte, sb *strs.Builder) {
   592  	var rawMethods [][]byte
   593  	var rawOptions []byte
   594  	sd.L2 = new(ServiceL2)
   595  	for len(b) > 0 {
   596  		num, typ, n := protowire.ConsumeTag(b)
   597  		b = b[n:]
   598  		switch typ {
   599  		case protowire.BytesType:
   600  			v, m := protowire.ConsumeBytes(b)
   601  			b = b[m:]
   602  			switch num {
   603  			case genid.ServiceDescriptorProto_Method_field_number:
   604  				rawMethods = append(rawMethods, v)
   605  			case genid.ServiceDescriptorProto_Options_field_number:
   606  				rawOptions = appendOptions(rawOptions, v)
   607  			}
   608  		default:
   609  			m := protowire.ConsumeFieldValue(num, typ, b)
   610  			b = b[m:]
   611  		}
   612  	}
   613  	if len(rawMethods) > 0 {
   614  		sd.L2.Methods.List = make([]Method, len(rawMethods))
   615  		for i, b := range rawMethods {
   616  			sd.L2.Methods.List[i].unmarshalFull(b, sb, sd.L0.ParentFile, sd, i)
   617  		}
   618  	}
   619  	sd.L2.Options = sd.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Service, rawOptions)
   620  }
   621  
   622  func (md *Method) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd protoreflect.Descriptor, i int) {
   623  	md.L0.ParentFile = pf
   624  	md.L0.Parent = pd
   625  	md.L0.Index = i
   626  
   627  	var rawOptions []byte
   628  	for len(b) > 0 {
   629  		num, typ, n := protowire.ConsumeTag(b)
   630  		b = b[n:]
   631  		switch typ {
   632  		case protowire.VarintType:
   633  			v, m := protowire.ConsumeVarint(b)
   634  			b = b[m:]
   635  			switch num {
   636  			case genid.MethodDescriptorProto_ClientStreaming_field_number:
   637  				md.L1.IsStreamingClient = protowire.DecodeBool(v)
   638  			case genid.MethodDescriptorProto_ServerStreaming_field_number:
   639  				md.L1.IsStreamingServer = protowire.DecodeBool(v)
   640  			}
   641  		case protowire.BytesType:
   642  			v, m := protowire.ConsumeBytes(b)
   643  			b = b[m:]
   644  			switch num {
   645  			case genid.MethodDescriptorProto_Name_field_number:
   646  				md.L0.FullName = appendFullName(sb, pd.FullName(), v)
   647  			case genid.MethodDescriptorProto_InputType_field_number:
   648  				md.L1.Input = PlaceholderMessage(makeFullName(sb, v))
   649  			case genid.MethodDescriptorProto_OutputType_field_number:
   650  				md.L1.Output = PlaceholderMessage(makeFullName(sb, v))
   651  			case genid.MethodDescriptorProto_Options_field_number:
   652  				rawOptions = appendOptions(rawOptions, v)
   653  			}
   654  		default:
   655  			m := protowire.ConsumeFieldValue(num, typ, b)
   656  			b = b[m:]
   657  		}
   658  	}
   659  	md.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Method, rawOptions)
   660  }
   661  
   662  // appendOptions appends src to dst, where the returned slice is never nil.
   663  // This is necessary to distinguish between empty and unpopulated options.
   664  func appendOptions(dst, src []byte) []byte {
   665  	if dst == nil {
   666  		dst = []byte{}
   667  	}
   668  	return append(dst, src...)
   669  }
   670  
   671  // optionsUnmarshaler constructs a lazy unmarshal function for an options message.
   672  //
   673  // The type of message to unmarshal to is passed as a pointer since the
   674  // vars in descopts may not yet be populated at the time this function is called.
   675  func (db *Builder) optionsUnmarshaler(p *protoreflect.ProtoMessage, b []byte) func() protoreflect.ProtoMessage {
   676  	if b == nil {
   677  		return nil
   678  	}
   679  	var opts protoreflect.ProtoMessage
   680  	var once sync.Once
   681  	return func() protoreflect.ProtoMessage {
   682  		once.Do(func() {
   683  			if *p == nil {
   684  				panic("Descriptor.Options called without importing the descriptor package")
   685  			}
   686  			opts = reflect.New(reflect.TypeOf(*p).Elem()).Interface().(protoreflect.ProtoMessage)
   687  			if err := (proto.UnmarshalOptions{
   688  				AllowPartial: true,
   689  				Resolver:     db.TypeResolver,
   690  			}).Unmarshal(b, opts); err != nil {
   691  				panic(err)
   692  			}
   693  		})
   694  		return opts
   695  	}
   696  }
   697  

View as plain text