审查视图

vendor/google.golang.org/protobuf/internal/filedesc/desc_lazy.go 20.8 KB
tangxvhui authored
1 2 3 4 5 6 7 8 9 10 11 12
// 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 filedesc

import (
	"reflect"
	"sync"

	"google.golang.org/protobuf/encoding/protowire"
	"google.golang.org/protobuf/internal/descopts"
唐旭辉 authored
13
	"google.golang.org/protobuf/internal/genid"
tangxvhui authored
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
	"google.golang.org/protobuf/internal/strs"
	"google.golang.org/protobuf/proto"
	pref "google.golang.org/protobuf/reflect/protoreflect"
)

func (fd *File) lazyRawInit() {
	fd.unmarshalFull(fd.builder.RawDescriptor)
	fd.resolveMessages()
	fd.resolveExtensions()
	fd.resolveServices()
}

func (file *File) resolveMessages() {
	var depIdx int32
	for i := range file.allMessages {
		md := &file.allMessages[i]

		// Resolve message field dependencies.
		for j := range md.L2.Fields.List {
			fd := &md.L2.Fields.List[j]

			// Weak fields are resolved upon actual use.
			if fd.L1.IsWeak {
				continue
			}

			// Resolve message field dependency.
			switch fd.L1.Kind {
			case pref.EnumKind:
				fd.L1.Enum = file.resolveEnumDependency(fd.L1.Enum, listFieldDeps, depIdx)
				depIdx++
			case pref.MessageKind, pref.GroupKind:
				fd.L1.Message = file.resolveMessageDependency(fd.L1.Message, listFieldDeps, depIdx)
				depIdx++
			}

			// Default is resolved here since it depends on Enum being resolved.
			if v := fd.L1.Default.val; v.IsValid() {
				fd.L1.Default = unmarshalDefault(v.Bytes(), fd.L1.Kind, file, fd.L1.Enum)
			}
		}
	}
}

func (file *File) resolveExtensions() {
	var depIdx int32
	for i := range file.allExtensions {
		xd := &file.allExtensions[i]

		// Resolve extension field dependency.
		switch xd.L1.Kind {
		case pref.EnumKind:
			xd.L2.Enum = file.resolveEnumDependency(xd.L2.Enum, listExtDeps, depIdx)
			depIdx++
		case pref.MessageKind, pref.GroupKind:
			xd.L2.Message = file.resolveMessageDependency(xd.L2.Message, listExtDeps, depIdx)
			depIdx++
		}

		// Default is resolved here since it depends on Enum being resolved.
		if v := xd.L2.Default.val; v.IsValid() {
			xd.L2.Default = unmarshalDefault(v.Bytes(), xd.L1.Kind, file, xd.L2.Enum)
		}
	}
}

func (file *File) resolveServices() {
	var depIdx int32
	for i := range file.allServices {
		sd := &file.allServices[i]

		// Resolve method dependencies.
		for j := range sd.L2.Methods.List {
			md := &sd.L2.Methods.List[j]
			md.L1.Input = file.resolveMessageDependency(md.L1.Input, listMethInDeps, depIdx)
			md.L1.Output = file.resolveMessageDependency(md.L1.Output, listMethOutDeps, depIdx)
			depIdx++
		}
	}
}

func (file *File) resolveEnumDependency(ed pref.EnumDescriptor, i, j int32) pref.EnumDescriptor {
	r := file.builder.FileRegistry
	if r, ok := r.(resolverByIndex); ok {
		if ed2 := r.FindEnumByIndex(i, j, file.allEnums, file.allMessages); ed2 != nil {
			return ed2
		}
	}
	for i := range file.allEnums {
		if ed2 := &file.allEnums[i]; ed2.L0.FullName == ed.FullName() {
			return ed2
		}
	}
	if d, _ := r.FindDescriptorByName(ed.FullName()); d != nil {
		return d.(pref.EnumDescriptor)
	}
	return ed
}

func (file *File) resolveMessageDependency(md pref.MessageDescriptor, i, j int32) pref.MessageDescriptor {
	r := file.builder.FileRegistry
	if r, ok := r.(resolverByIndex); ok {
		if md2 := r.FindMessageByIndex(i, j, file.allEnums, file.allMessages); md2 != nil {
			return md2
		}
	}
	for i := range file.allMessages {
		if md2 := &file.allMessages[i]; md2.L0.FullName == md.FullName() {
			return md2
		}
	}
	if d, _ := r.FindDescriptorByName(md.FullName()); d != nil {
		return d.(pref.MessageDescriptor)
	}
	return md
}

func (fd *File) unmarshalFull(b []byte) {
	sb := getBuilder()
	defer putBuilder(sb)

	var enumIdx, messageIdx, extensionIdx, serviceIdx int
	var rawOptions []byte
	fd.L2 = new(FileL2)
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
146
			case genid.FileDescriptorProto_PublicDependency_field_number:
tangxvhui authored
147
				fd.L2.Imports[v].IsPublic = true
唐旭辉 authored
148
			case genid.FileDescriptorProto_WeakDependency_field_number:
tangxvhui authored
149 150 151 152 153 154
				fd.L2.Imports[v].IsWeak = true
			}
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
155
			case genid.FileDescriptorProto_Dependency_field_number:
tangxvhui authored
156 157 158 159 160 161
				path := sb.MakeString(v)
				imp, _ := fd.builder.FileRegistry.FindFileByPath(path)
				if imp == nil {
					imp = PlaceholderFile(path)
				}
				fd.L2.Imports = append(fd.L2.Imports, pref.FileImport{FileDescriptor: imp})
唐旭辉 authored
162
			case genid.FileDescriptorProto_EnumType_field_number:
tangxvhui authored
163 164
				fd.L1.Enums.List[enumIdx].unmarshalFull(v, sb)
				enumIdx++
唐旭辉 authored
165
			case genid.FileDescriptorProto_MessageType_field_number:
tangxvhui authored
166 167
				fd.L1.Messages.List[messageIdx].unmarshalFull(v, sb)
				messageIdx++
唐旭辉 authored
168
			case genid.FileDescriptorProto_Extension_field_number:
tangxvhui authored
169 170
				fd.L1.Extensions.List[extensionIdx].unmarshalFull(v, sb)
				extensionIdx++
唐旭辉 authored
171
			case genid.FileDescriptorProto_Service_field_number:
tangxvhui authored
172 173
				fd.L1.Services.List[serviceIdx].unmarshalFull(v, sb)
				serviceIdx++
唐旭辉 authored
174
			case genid.FileDescriptorProto_Options_field_number:
tangxvhui authored
175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	fd.L2.Options = fd.builder.optionsUnmarshaler(&descopts.File, rawOptions)
}

func (ed *Enum) unmarshalFull(b []byte, sb *strs.Builder) {
	var rawValues [][]byte
	var rawOptions []byte
	if !ed.L1.eagerValues {
		ed.L2 = new(EnumL2)
	}
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
199
			case genid.EnumDescriptorProto_Value_field_number:
tangxvhui authored
200
				rawValues = append(rawValues, v)
唐旭辉 authored
201
			case genid.EnumDescriptorProto_ReservedName_field_number:
tangxvhui authored
202
				ed.L2.ReservedNames.List = append(ed.L2.ReservedNames.List, pref.Name(sb.MakeString(v)))
唐旭辉 authored
203
			case genid.EnumDescriptorProto_ReservedRange_field_number:
tangxvhui authored
204
				ed.L2.ReservedRanges.List = append(ed.L2.ReservedRanges.List, unmarshalEnumReservedRange(v))
唐旭辉 authored
205
			case genid.EnumDescriptorProto_Options_field_number:
tangxvhui authored
206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	if !ed.L1.eagerValues && len(rawValues) > 0 {
		ed.L2.Values.List = make([]EnumValue, len(rawValues))
		for i, b := range rawValues {
			ed.L2.Values.List[i].unmarshalFull(b, sb, ed.L0.ParentFile, ed, i)
		}
	}
	ed.L2.Options = ed.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Enum, rawOptions)
}

func unmarshalEnumReservedRange(b []byte) (r [2]pref.EnumNumber) {
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
231
			case genid.EnumDescriptorProto_EnumReservedRange_Start_field_number:
tangxvhui authored
232
				r[0] = pref.EnumNumber(v)
唐旭辉 authored
233
			case genid.EnumDescriptorProto_EnumReservedRange_End_field_number:
tangxvhui authored
234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257
				r[1] = pref.EnumNumber(v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	return r
}

func (vd *EnumValue) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd pref.Descriptor, i int) {
	vd.L0.ParentFile = pf
	vd.L0.Parent = pd
	vd.L0.Index = i

	var rawOptions []byte
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
258
			case genid.EnumValueDescriptorProto_Number_field_number:
tangxvhui authored
259 260 261 262 263 264
				vd.L1.Number = pref.EnumNumber(v)
			}
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
265
			case genid.EnumValueDescriptorProto_Name_field_number:
tangxvhui authored
266 267
				// NOTE: Enum values are in the same scope as the enum parent.
				vd.L0.FullName = appendFullName(sb, pd.Parent().FullName(), v)
唐旭辉 authored
268
			case genid.EnumValueDescriptorProto_Options_field_number:
tangxvhui authored
269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	vd.L1.Options = pf.builder.optionsUnmarshaler(&descopts.EnumValue, rawOptions)
}

func (md *Message) unmarshalFull(b []byte, sb *strs.Builder) {
	var rawFields, rawOneofs [][]byte
	var enumIdx, messageIdx, extensionIdx int
	var rawOptions []byte
	md.L2 = new(MessageL2)
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
292
			case genid.DescriptorProto_Field_field_number:
tangxvhui authored
293
				rawFields = append(rawFields, v)
唐旭辉 authored
294
			case genid.DescriptorProto_OneofDecl_field_number:
tangxvhui authored
295
				rawOneofs = append(rawOneofs, v)
唐旭辉 authored
296
			case genid.DescriptorProto_ReservedName_field_number:
tangxvhui authored
297
				md.L2.ReservedNames.List = append(md.L2.ReservedNames.List, pref.Name(sb.MakeString(v)))
唐旭辉 authored
298
			case genid.DescriptorProto_ReservedRange_field_number:
tangxvhui authored
299
				md.L2.ReservedRanges.List = append(md.L2.ReservedRanges.List, unmarshalMessageReservedRange(v))
唐旭辉 authored
300
			case genid.DescriptorProto_ExtensionRange_field_number:
tangxvhui authored
301 302 303 304
				r, rawOptions := unmarshalMessageExtensionRange(v)
				opts := md.L0.ParentFile.builder.optionsUnmarshaler(&descopts.ExtensionRange, rawOptions)
				md.L2.ExtensionRanges.List = append(md.L2.ExtensionRanges.List, r)
				md.L2.ExtensionRangeOptions = append(md.L2.ExtensionRangeOptions, opts)
唐旭辉 authored
305
			case genid.DescriptorProto_EnumType_field_number:
tangxvhui authored
306 307
				md.L1.Enums.List[enumIdx].unmarshalFull(v, sb)
				enumIdx++
唐旭辉 authored
308
			case genid.DescriptorProto_NestedType_field_number:
tangxvhui authored
309 310
				md.L1.Messages.List[messageIdx].unmarshalFull(v, sb)
				messageIdx++
唐旭辉 authored
311
			case genid.DescriptorProto_Extension_field_number:
tangxvhui authored
312 313
				md.L1.Extensions.List[extensionIdx].unmarshalFull(v, sb)
				extensionIdx++
唐旭辉 authored
314
			case genid.DescriptorProto_Options_field_number:
tangxvhui authored
315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349
				md.unmarshalOptions(v)
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	if len(rawFields) > 0 || len(rawOneofs) > 0 {
		md.L2.Fields.List = make([]Field, len(rawFields))
		md.L2.Oneofs.List = make([]Oneof, len(rawOneofs))
		for i, b := range rawFields {
			fd := &md.L2.Fields.List[i]
			fd.unmarshalFull(b, sb, md.L0.ParentFile, md, i)
			if fd.L1.Cardinality == pref.Required {
				md.L2.RequiredNumbers.List = append(md.L2.RequiredNumbers.List, fd.L1.Number)
			}
		}
		for i, b := range rawOneofs {
			od := &md.L2.Oneofs.List[i]
			od.unmarshalFull(b, sb, md.L0.ParentFile, md, i)
		}
	}
	md.L2.Options = md.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Message, rawOptions)
}

func (md *Message) unmarshalOptions(b []byte) {
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
350
			case genid.MessageOptions_MapEntry_field_number:
tangxvhui authored
351
				md.L1.IsMapEntry = protowire.DecodeBool(v)
唐旭辉 authored
352
			case genid.MessageOptions_MessageSetWireFormat_field_number:
tangxvhui authored
353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370
				md.L1.IsMessageSet = protowire.DecodeBool(v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
}

func unmarshalMessageReservedRange(b []byte) (r [2]pref.FieldNumber) {
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
371
			case genid.DescriptorProto_ReservedRange_Start_field_number:
tangxvhui authored
372
				r[0] = pref.FieldNumber(v)
唐旭辉 authored
373
			case genid.DescriptorProto_ReservedRange_End_field_number:
tangxvhui authored
374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392
				r[1] = pref.FieldNumber(v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	return r
}

func unmarshalMessageExtensionRange(b []byte) (r [2]pref.FieldNumber, rawOptions []byte) {
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
393
			case genid.DescriptorProto_ExtensionRange_Start_field_number:
tangxvhui authored
394
				r[0] = pref.FieldNumber(v)
唐旭辉 authored
395
			case genid.DescriptorProto_ExtensionRange_End_field_number:
tangxvhui authored
396 397 398 399 400 401
				r[1] = pref.FieldNumber(v)
			}
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
402
			case genid.DescriptorProto_ExtensionRange_Options_field_number:
tangxvhui authored
403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	return r, rawOptions
}

func (fd *Field) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd pref.Descriptor, i int) {
	fd.L0.ParentFile = pf
	fd.L0.Parent = pd
	fd.L0.Index = i

	var rawTypeName []byte
	var rawOptions []byte
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
428
			case genid.FieldDescriptorProto_Number_field_number:
tangxvhui authored
429
				fd.L1.Number = pref.FieldNumber(v)
唐旭辉 authored
430
			case genid.FieldDescriptorProto_Label_field_number:
tangxvhui authored
431
				fd.L1.Cardinality = pref.Cardinality(v)
唐旭辉 authored
432
			case genid.FieldDescriptorProto_Type_field_number:
tangxvhui authored
433
				fd.L1.Kind = pref.Kind(v)
唐旭辉 authored
434
			case genid.FieldDescriptorProto_OneofIndex_field_number:
tangxvhui authored
435 436 437 438 439 440 441 442 443
				// In Message.unmarshalFull, we allocate slices for both
				// the field and oneof descriptors before unmarshaling either
				// of them. This ensures pointers to slice elements are stable.
				od := &pd.(*Message).L2.Oneofs.List[v]
				od.L1.Fields.List = append(od.L1.Fields.List, fd)
				if fd.L1.ContainingOneof != nil {
					panic("oneof type already set")
				}
				fd.L1.ContainingOneof = od
唐旭辉 authored
444
			case genid.FieldDescriptorProto_Proto3Optional_field_number:
tangxvhui authored
445 446 447 448 449 450
				fd.L1.IsProto3Optional = protowire.DecodeBool(v)
			}
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
451
			case genid.FieldDescriptorProto_Name_field_number:
tangxvhui authored
452
				fd.L0.FullName = appendFullName(sb, pd.FullName(), v)
唐旭辉 authored
453
			case genid.FieldDescriptorProto_JsonName_field_number:
tangxvhui authored
454
				fd.L1.JSONName.Init(sb.MakeString(v))
唐旭辉 authored
455
			case genid.FieldDescriptorProto_DefaultValue_field_number:
tangxvhui authored
456
				fd.L1.Default.val = pref.ValueOfBytes(v) // temporarily store as bytes; later resolved in resolveMessages
唐旭辉 authored
457
			case genid.FieldDescriptorProto_TypeName_field_number:
tangxvhui authored
458
				rawTypeName = v
唐旭辉 authored
459
			case genid.FieldDescriptorProto_Options_field_number:
tangxvhui authored
460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490
				fd.unmarshalOptions(v)
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	if rawTypeName != nil {
		name := makeFullName(sb, rawTypeName)
		switch fd.L1.Kind {
		case pref.EnumKind:
			fd.L1.Enum = PlaceholderEnum(name)
		case pref.MessageKind, pref.GroupKind:
			fd.L1.Message = PlaceholderMessage(name)
		}
	}
	fd.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Field, rawOptions)
}

func (fd *Field) unmarshalOptions(b []byte) {
	const FieldOptions_EnforceUTF8 = 13

	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
491
			case genid.FieldOptions_Packed_field_number:
tangxvhui authored
492 493
				fd.L1.HasPacked = true
				fd.L1.IsPacked = protowire.DecodeBool(v)
唐旭辉 authored
494
			case genid.FieldOptions_Weak_field_number:
tangxvhui authored
495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520
				fd.L1.IsWeak = protowire.DecodeBool(v)
			case FieldOptions_EnforceUTF8:
				fd.L1.HasEnforceUTF8 = true
				fd.L1.EnforceUTF8 = protowire.DecodeBool(v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
}

func (od *Oneof) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd pref.Descriptor, i int) {
	od.L0.ParentFile = pf
	od.L0.Parent = pd
	od.L0.Index = i

	var rawOptions []byte
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
521
			case genid.OneofDescriptorProto_Name_field_number:
tangxvhui authored
522
				od.L0.FullName = appendFullName(sb, pd.FullName(), v)
唐旭辉 authored
523
			case genid.OneofDescriptorProto_Options_field_number:
tangxvhui authored
524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	od.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Oneof, rawOptions)
}

func (xd *Extension) unmarshalFull(b []byte, sb *strs.Builder) {
	var rawTypeName []byte
	var rawOptions []byte
	xd.L2 = new(ExtensionL2)
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
546
			case genid.FieldDescriptorProto_Proto3Optional_field_number:
tangxvhui authored
547 548 549 550 551 552
				xd.L2.IsProto3Optional = protowire.DecodeBool(v)
			}
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
553
			case genid.FieldDescriptorProto_JsonName_field_number:
tangxvhui authored
554
				xd.L2.JSONName.Init(sb.MakeString(v))
唐旭辉 authored
555
			case genid.FieldDescriptorProto_DefaultValue_field_number:
tangxvhui authored
556
				xd.L2.Default.val = pref.ValueOfBytes(v) // temporarily store as bytes; later resolved in resolveExtensions
唐旭辉 authored
557
			case genid.FieldDescriptorProto_TypeName_field_number:
tangxvhui authored
558
				rawTypeName = v
唐旭辉 authored
559
			case genid.FieldDescriptorProto_Options_field_number:
tangxvhui authored
560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588
				xd.unmarshalOptions(v)
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	if rawTypeName != nil {
		name := makeFullName(sb, rawTypeName)
		switch xd.L1.Kind {
		case pref.EnumKind:
			xd.L2.Enum = PlaceholderEnum(name)
		case pref.MessageKind, pref.GroupKind:
			xd.L2.Message = PlaceholderMessage(name)
		}
	}
	xd.L2.Options = xd.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Field, rawOptions)
}

func (xd *Extension) unmarshalOptions(b []byte) {
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
589
			case genid.FieldOptions_Packed_field_number:
tangxvhui authored
590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610
				xd.L2.IsPacked = protowire.DecodeBool(v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
}

func (sd *Service) unmarshalFull(b []byte, sb *strs.Builder) {
	var rawMethods [][]byte
	var rawOptions []byte
	sd.L2 = new(ServiceL2)
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
611
			case genid.ServiceDescriptorProto_Method_field_number:
tangxvhui authored
612
				rawMethods = append(rawMethods, v)
唐旭辉 authored
613
			case genid.ServiceDescriptorProto_Options_field_number:
tangxvhui authored
614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	if len(rawMethods) > 0 {
		sd.L2.Methods.List = make([]Method, len(rawMethods))
		for i, b := range rawMethods {
			sd.L2.Methods.List[i].unmarshalFull(b, sb, sd.L0.ParentFile, sd, i)
		}
	}
	sd.L2.Options = sd.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Service, rawOptions)
}

func (md *Method) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd pref.Descriptor, i int) {
	md.L0.ParentFile = pf
	md.L0.Parent = pd
	md.L0.Index = i

	var rawOptions []byte
	for len(b) > 0 {
		num, typ, n := protowire.ConsumeTag(b)
		b = b[n:]
		switch typ {
		case protowire.VarintType:
			v, m := protowire.ConsumeVarint(b)
			b = b[m:]
			switch num {
唐旭辉 authored
644
			case genid.MethodDescriptorProto_ClientStreaming_field_number:
tangxvhui authored
645
				md.L1.IsStreamingClient = protowire.DecodeBool(v)
唐旭辉 authored
646
			case genid.MethodDescriptorProto_ServerStreaming_field_number:
tangxvhui authored
647 648 649 650 651 652
				md.L1.IsStreamingServer = protowire.DecodeBool(v)
			}
		case protowire.BytesType:
			v, m := protowire.ConsumeBytes(b)
			b = b[m:]
			switch num {
唐旭辉 authored
653
			case genid.MethodDescriptorProto_Name_field_number:
tangxvhui authored
654
				md.L0.FullName = appendFullName(sb, pd.FullName(), v)
唐旭辉 authored
655
			case genid.MethodDescriptorProto_InputType_field_number:
tangxvhui authored
656
				md.L1.Input = PlaceholderMessage(makeFullName(sb, v))
唐旭辉 authored
657
			case genid.MethodDescriptorProto_OutputType_field_number:
tangxvhui authored
658
				md.L1.Output = PlaceholderMessage(makeFullName(sb, v))
唐旭辉 authored
659
			case genid.MethodDescriptorProto_Options_field_number:
tangxvhui authored
660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704
				rawOptions = appendOptions(rawOptions, v)
			}
		default:
			m := protowire.ConsumeFieldValue(num, typ, b)
			b = b[m:]
		}
	}
	md.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Method, rawOptions)
}

// appendOptions appends src to dst, where the returned slice is never nil.
// This is necessary to distinguish between empty and unpopulated options.
func appendOptions(dst, src []byte) []byte {
	if dst == nil {
		dst = []byte{}
	}
	return append(dst, src...)
}

// optionsUnmarshaler constructs a lazy unmarshal function for an options message.
//
// The type of message to unmarshal to is passed as a pointer since the
// vars in descopts may not yet be populated at the time this function is called.
func (db *Builder) optionsUnmarshaler(p *pref.ProtoMessage, b []byte) func() pref.ProtoMessage {
	if b == nil {
		return nil
	}
	var opts pref.ProtoMessage
	var once sync.Once
	return func() pref.ProtoMessage {
		once.Do(func() {
			if *p == nil {
				panic("Descriptor.Options called without importing the descriptor package")
			}
			opts = reflect.New(reflect.TypeOf(*p).Elem()).Interface().(pref.ProtoMessage)
			if err := (proto.UnmarshalOptions{
				AllowPartial: true,
				Resolver:     db.TypeResolver,
			}).Unmarshal(b, opts); err != nil {
				panic(err)
			}
		})
		return opts
	}
}