blob: c43b3e25e0487c00f6a47df4e6031db1c777d2f8 [file] [log] [blame]
Joe Tsaid8881392019-06-06 13:01:53 -07001// 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
5package filedesc
6
7import (
8 "reflect"
Joe Tsai8a4c3d12019-07-06 18:08:55 -07009 "sync"
Joe Tsaid8881392019-06-06 13:01:53 -070010
11 "google.golang.org/protobuf/internal/descopts"
12 "google.golang.org/protobuf/internal/encoding/wire"
13 "google.golang.org/protobuf/internal/fieldnum"
Joe Tsai97a87392019-07-07 01:49:59 -070014 "google.golang.org/protobuf/internal/strs"
Joe Tsaid8881392019-06-06 13:01:53 -070015 "google.golang.org/protobuf/proto"
16 pref "google.golang.org/protobuf/reflect/protoreflect"
17)
18
19func (fd *File) lazyRawInit() {
20 fd.unmarshalFull(fd.builder.RawDescriptor)
21 fd.resolveMessages()
22 fd.resolveExtensions()
23 fd.resolveServices()
24}
25
26func (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
Joe Tsaif98dd982019-08-05 13:41:37 -070035 // Weak fields are resolved upon actual use.
Joe Tsaid8881392019-06-06 13:01:53 -070036 if fd.L1.IsWeak {
Joe Tsaid8881392019-06-06 13:01:53 -070037 continue
38 }
39
40 // Resolve message field dependency.
41 switch fd.L1.Kind {
42 case pref.EnumKind:
43 fd.L1.Enum = file.resolveEnumDependency(fd.L1.Enum, listFieldDeps, depIdx)
44 depIdx++
45 case pref.MessageKind, pref.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
58func (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 pref.EnumKind:
66 xd.L2.Enum = file.resolveEnumDependency(xd.L2.Enum, listExtDeps, depIdx)
67 depIdx++
68 case pref.MessageKind, pref.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
80func (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
95func (file *File) resolveEnumDependency(ed pref.EnumDescriptor, i, j int32) pref.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 }
Joe Tsaie407ee12019-06-24 16:19:06 -0700107 if d, _ := r.FindDescriptorByName(ed.FullName()); d != nil {
108 return d.(pref.EnumDescriptor)
Joe Tsaid8881392019-06-06 13:01:53 -0700109 }
110 return ed
111}
112
113func (file *File) resolveMessageDependency(md pref.MessageDescriptor, i, j int32) pref.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 }
Joe Tsaie407ee12019-06-24 16:19:06 -0700125 if d, _ := r.FindDescriptorByName(md.FullName()); d != nil {
126 return d.(pref.MessageDescriptor)
Joe Tsaid8881392019-06-06 13:01:53 -0700127 }
128 return md
129}
130
131func (fd *File) unmarshalFull(b []byte) {
Joe Tsai97a87392019-07-07 01:49:59 -0700132 sb := getBuilder()
133 defer putBuilder(sb)
Joe Tsaid8881392019-06-06 13:01:53 -0700134
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 := wire.ConsumeTag(b)
140 b = b[n:]
141 switch typ {
142 case wire.VarintType:
143 v, m := wire.ConsumeVarint(b)
144 b = b[m:]
145 switch num {
146 case fieldnum.FileDescriptorProto_PublicDependency:
147 fd.L2.Imports[v].IsPublic = true
148 case fieldnum.FileDescriptorProto_WeakDependency:
149 fd.L2.Imports[v].IsWeak = true
150 }
151 case wire.BytesType:
152 v, m := wire.ConsumeBytes(b)
153 b = b[m:]
154 switch num {
155 case fieldnum.FileDescriptorProto_Dependency:
Joe Tsai97a87392019-07-07 01:49:59 -0700156 path := sb.MakeString(v)
Joe Tsaid8881392019-06-06 13:01:53 -0700157 imp, _ := fd.builder.FileRegistry.FindFileByPath(path)
158 if imp == nil {
159 imp = PlaceholderFile(path)
160 }
161 fd.L2.Imports = append(fd.L2.Imports, pref.FileImport{FileDescriptor: imp})
162 case fieldnum.FileDescriptorProto_EnumType:
Joe Tsai97a87392019-07-07 01:49:59 -0700163 fd.L1.Enums.List[enumIdx].unmarshalFull(v, sb)
Joe Tsaid8881392019-06-06 13:01:53 -0700164 enumIdx++
165 case fieldnum.FileDescriptorProto_MessageType:
Joe Tsai97a87392019-07-07 01:49:59 -0700166 fd.L1.Messages.List[messageIdx].unmarshalFull(v, sb)
Joe Tsaid8881392019-06-06 13:01:53 -0700167 messageIdx++
168 case fieldnum.FileDescriptorProto_Extension:
Joe Tsai97a87392019-07-07 01:49:59 -0700169 fd.L1.Extensions.List[extensionIdx].unmarshalFull(v, sb)
Joe Tsaid8881392019-06-06 13:01:53 -0700170 extensionIdx++
171 case fieldnum.FileDescriptorProto_Service:
Joe Tsai97a87392019-07-07 01:49:59 -0700172 fd.L1.Services.List[serviceIdx].unmarshalFull(v, sb)
Joe Tsaid8881392019-06-06 13:01:53 -0700173 serviceIdx++
174 case fieldnum.FileDescriptorProto_Options:
175 rawOptions = appendOptions(rawOptions, v)
176 }
177 default:
178 m := wire.ConsumeFieldValue(num, typ, b)
179 b = b[m:]
180 }
181 }
Damien Neil4e173a52019-08-02 13:16:57 -0700182 fd.L2.Options = fd.builder.optionsUnmarshaler(&descopts.File, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700183}
184
Joe Tsai97a87392019-07-07 01:49:59 -0700185func (ed *Enum) unmarshalFull(b []byte, sb *strs.Builder) {
Joe Tsaid8881392019-06-06 13:01:53 -0700186 var rawValues [][]byte
187 var rawOptions []byte
Joe Tsai42413972019-06-21 14:25:16 -0700188 if !ed.L1.eagerValues {
189 ed.L2 = new(EnumL2)
190 }
Joe Tsaid8881392019-06-06 13:01:53 -0700191 for len(b) > 0 {
192 num, typ, n := wire.ConsumeTag(b)
193 b = b[n:]
194 switch typ {
195 case wire.BytesType:
196 v, m := wire.ConsumeBytes(b)
197 b = b[m:]
198 switch num {
199 case fieldnum.EnumDescriptorProto_Value:
200 rawValues = append(rawValues, v)
201 case fieldnum.EnumDescriptorProto_ReservedName:
Joe Tsai97a87392019-07-07 01:49:59 -0700202 ed.L2.ReservedNames.List = append(ed.L2.ReservedNames.List, pref.Name(sb.MakeString(v)))
Joe Tsaid8881392019-06-06 13:01:53 -0700203 case fieldnum.EnumDescriptorProto_ReservedRange:
204 ed.L2.ReservedRanges.List = append(ed.L2.ReservedRanges.List, unmarshalEnumReservedRange(v))
205 case fieldnum.EnumDescriptorProto_Options:
206 rawOptions = appendOptions(rawOptions, v)
207 }
208 default:
209 m := wire.ConsumeFieldValue(num, typ, b)
210 b = b[m:]
211 }
212 }
Joe Tsai42413972019-06-21 14:25:16 -0700213 if !ed.L1.eagerValues && len(rawValues) > 0 {
Joe Tsaid8881392019-06-06 13:01:53 -0700214 ed.L2.Values.List = make([]EnumValue, len(rawValues))
215 for i, b := range rawValues {
Joe Tsai97a87392019-07-07 01:49:59 -0700216 ed.L2.Values.List[i].unmarshalFull(b, sb, ed.L0.ParentFile, ed, i)
Joe Tsaid8881392019-06-06 13:01:53 -0700217 }
218 }
Damien Neil4e173a52019-08-02 13:16:57 -0700219 ed.L2.Options = ed.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Enum, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700220}
221
222func unmarshalEnumReservedRange(b []byte) (r [2]pref.EnumNumber) {
223 for len(b) > 0 {
224 num, typ, n := wire.ConsumeTag(b)
225 b = b[n:]
226 switch typ {
227 case wire.VarintType:
228 v, m := wire.ConsumeVarint(b)
229 b = b[m:]
230 switch num {
231 case fieldnum.EnumDescriptorProto_EnumReservedRange_Start:
232 r[0] = pref.EnumNumber(v)
233 case fieldnum.EnumDescriptorProto_EnumReservedRange_End:
234 r[1] = pref.EnumNumber(v)
235 }
236 default:
237 m := wire.ConsumeFieldValue(num, typ, b)
238 b = b[m:]
239 }
240 }
241 return r
242}
243
Joe Tsai97a87392019-07-07 01:49:59 -0700244func (vd *EnumValue) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd pref.Descriptor, i int) {
Joe Tsaid8881392019-06-06 13:01:53 -0700245 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 := wire.ConsumeTag(b)
252 b = b[n:]
253 switch typ {
254 case wire.VarintType:
255 v, m := wire.ConsumeVarint(b)
256 b = b[m:]
257 switch num {
258 case fieldnum.EnumValueDescriptorProto_Number:
259 vd.L1.Number = pref.EnumNumber(v)
260 }
261 case wire.BytesType:
262 v, m := wire.ConsumeBytes(b)
263 b = b[m:]
264 switch num {
265 case fieldnum.EnumValueDescriptorProto_Name:
266 // NOTE: Enum values are in the same scope as the enum parent.
Joe Tsai97a87392019-07-07 01:49:59 -0700267 vd.L0.FullName = appendFullName(sb, pd.Parent().FullName(), v)
Joe Tsaid8881392019-06-06 13:01:53 -0700268 case fieldnum.EnumValueDescriptorProto_Options:
269 rawOptions = appendOptions(rawOptions, v)
270 }
271 default:
272 m := wire.ConsumeFieldValue(num, typ, b)
273 b = b[m:]
274 }
275 }
Damien Neil4e173a52019-08-02 13:16:57 -0700276 vd.L1.Options = pf.builder.optionsUnmarshaler(&descopts.EnumValue, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700277}
278
Joe Tsai97a87392019-07-07 01:49:59 -0700279func (md *Message) unmarshalFull(b []byte, sb *strs.Builder) {
Joe Tsaid8881392019-06-06 13:01:53 -0700280 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 := wire.ConsumeTag(b)
286 b = b[n:]
287 switch typ {
288 case wire.BytesType:
289 v, m := wire.ConsumeBytes(b)
290 b = b[m:]
291 switch num {
292 case fieldnum.DescriptorProto_Field:
293 rawFields = append(rawFields, v)
294 case fieldnum.DescriptorProto_OneofDecl:
295 rawOneofs = append(rawOneofs, v)
296 case fieldnum.DescriptorProto_ReservedName:
Joe Tsai97a87392019-07-07 01:49:59 -0700297 md.L2.ReservedNames.List = append(md.L2.ReservedNames.List, pref.Name(sb.MakeString(v)))
Joe Tsaid8881392019-06-06 13:01:53 -0700298 case fieldnum.DescriptorProto_ReservedRange:
299 md.L2.ReservedRanges.List = append(md.L2.ReservedRanges.List, unmarshalMessageReservedRange(v))
300 case fieldnum.DescriptorProto_ExtensionRange:
301 r, rawOptions := unmarshalMessageExtensionRange(v)
Damien Neil4e173a52019-08-02 13:16:57 -0700302 opts := md.L0.ParentFile.builder.optionsUnmarshaler(&descopts.ExtensionRange, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700303 md.L2.ExtensionRanges.List = append(md.L2.ExtensionRanges.List, r)
304 md.L2.ExtensionRangeOptions = append(md.L2.ExtensionRangeOptions, opts)
305 case fieldnum.DescriptorProto_EnumType:
Joe Tsai97a87392019-07-07 01:49:59 -0700306 md.L1.Enums.List[enumIdx].unmarshalFull(v, sb)
Joe Tsaid8881392019-06-06 13:01:53 -0700307 enumIdx++
308 case fieldnum.DescriptorProto_NestedType:
Joe Tsai97a87392019-07-07 01:49:59 -0700309 md.L1.Messages.List[messageIdx].unmarshalFull(v, sb)
Joe Tsaid8881392019-06-06 13:01:53 -0700310 messageIdx++
311 case fieldnum.DescriptorProto_Extension:
Joe Tsai97a87392019-07-07 01:49:59 -0700312 md.L1.Extensions.List[extensionIdx].unmarshalFull(v, sb)
Joe Tsaid8881392019-06-06 13:01:53 -0700313 extensionIdx++
314 case fieldnum.DescriptorProto_Options:
315 md.unmarshalOptions(v)
316 rawOptions = appendOptions(rawOptions, v)
317 }
318 default:
319 m := wire.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]
Joe Tsai97a87392019-07-07 01:49:59 -0700328 fd.unmarshalFull(b, sb, md.L0.ParentFile, md, i)
Joe Tsaid8881392019-06-06 13:01:53 -0700329 if fd.L1.Cardinality == pref.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]
Joe Tsai97a87392019-07-07 01:49:59 -0700335 od.unmarshalFull(b, sb, md.L0.ParentFile, md, i)
Joe Tsaid8881392019-06-06 13:01:53 -0700336 }
337 }
Damien Neil4e173a52019-08-02 13:16:57 -0700338 md.L2.Options = md.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Message, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700339}
340
341func (md *Message) unmarshalOptions(b []byte) {
342 for len(b) > 0 {
343 num, typ, n := wire.ConsumeTag(b)
344 b = b[n:]
345 switch typ {
346 case wire.VarintType:
347 v, m := wire.ConsumeVarint(b)
348 b = b[m:]
349 switch num {
350 case fieldnum.MessageOptions_MapEntry:
Damien Neile9187322019-11-07 15:30:44 -0800351 md.L1.IsMapEntry = wire.DecodeBool(v)
Joe Tsaid8881392019-06-06 13:01:53 -0700352 case fieldnum.MessageOptions_MessageSetWireFormat:
Damien Neile9187322019-11-07 15:30:44 -0800353 md.L1.IsMessageSet = wire.DecodeBool(v)
Joe Tsaid8881392019-06-06 13:01:53 -0700354 }
355 default:
356 m := wire.ConsumeFieldValue(num, typ, b)
357 b = b[m:]
358 }
359 }
360}
361
362func unmarshalMessageReservedRange(b []byte) (r [2]pref.FieldNumber) {
363 for len(b) > 0 {
364 num, typ, n := wire.ConsumeTag(b)
365 b = b[n:]
366 switch typ {
367 case wire.VarintType:
368 v, m := wire.ConsumeVarint(b)
369 b = b[m:]
370 switch num {
371 case fieldnum.DescriptorProto_ReservedRange_Start:
372 r[0] = pref.FieldNumber(v)
373 case fieldnum.DescriptorProto_ReservedRange_End:
374 r[1] = pref.FieldNumber(v)
375 }
376 default:
377 m := wire.ConsumeFieldValue(num, typ, b)
378 b = b[m:]
379 }
380 }
381 return r
382}
383
384func unmarshalMessageExtensionRange(b []byte) (r [2]pref.FieldNumber, rawOptions []byte) {
385 for len(b) > 0 {
386 num, typ, n := wire.ConsumeTag(b)
387 b = b[n:]
388 switch typ {
389 case wire.VarintType:
390 v, m := wire.ConsumeVarint(b)
391 b = b[m:]
392 switch num {
393 case fieldnum.DescriptorProto_ExtensionRange_Start:
394 r[0] = pref.FieldNumber(v)
395 case fieldnum.DescriptorProto_ExtensionRange_End:
396 r[1] = pref.FieldNumber(v)
397 }
398 case wire.BytesType:
399 v, m := wire.ConsumeBytes(b)
400 b = b[m:]
401 switch num {
402 case fieldnum.DescriptorProto_ExtensionRange_Options:
403 rawOptions = appendOptions(rawOptions, v)
404 }
405 default:
406 m := wire.ConsumeFieldValue(num, typ, b)
407 b = b[m:]
408 }
409 }
410 return r, rawOptions
411}
412
Joe Tsai97a87392019-07-07 01:49:59 -0700413func (fd *Field) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd pref.Descriptor, i int) {
Joe Tsaid8881392019-06-06 13:01:53 -0700414 fd.L0.ParentFile = pf
415 fd.L0.Parent = pd
416 fd.L0.Index = i
417
418 var rawTypeName []byte
419 var rawOptions []byte
420 for len(b) > 0 {
421 num, typ, n := wire.ConsumeTag(b)
422 b = b[n:]
423 switch typ {
424 case wire.VarintType:
425 v, m := wire.ConsumeVarint(b)
426 b = b[m:]
427 switch num {
428 case fieldnum.FieldDescriptorProto_Number:
429 fd.L1.Number = pref.FieldNumber(v)
430 case fieldnum.FieldDescriptorProto_Label:
431 fd.L1.Cardinality = pref.Cardinality(v)
432 case fieldnum.FieldDescriptorProto_Type:
433 fd.L1.Kind = pref.Kind(v)
434 case fieldnum.FieldDescriptorProto_OneofIndex:
435 // In Message.unmarshalFull, we allocate slices for both
436 // the field and oneof descriptors before unmarshaling either
437 // of them. This ensures pointers to slice elements are stable.
438 od := &pd.(*Message).L2.Oneofs.List[v]
439 od.L1.Fields.List = append(od.L1.Fields.List, fd)
440 if fd.L1.ContainingOneof != nil {
441 panic("oneof type already set")
442 }
443 fd.L1.ContainingOneof = od
444 }
445 case wire.BytesType:
446 v, m := wire.ConsumeBytes(b)
447 b = b[m:]
448 switch num {
449 case fieldnum.FieldDescriptorProto_Name:
Joe Tsai97a87392019-07-07 01:49:59 -0700450 fd.L0.FullName = appendFullName(sb, pd.FullName(), v)
Joe Tsaid8881392019-06-06 13:01:53 -0700451 case fieldnum.FieldDescriptorProto_JsonName:
Joe Tsai6c26a042020-01-03 15:53:47 -0800452 fd.L1.JSONName.Init(sb.MakeString(v))
Joe Tsaid8881392019-06-06 13:01:53 -0700453 case fieldnum.FieldDescriptorProto_DefaultValue:
Joe Tsai84177c92019-09-17 13:38:48 -0700454 fd.L1.Default.val = pref.ValueOfBytes(v) // temporarily store as bytes; later resolved in resolveMessages
Joe Tsaid8881392019-06-06 13:01:53 -0700455 case fieldnum.FieldDescriptorProto_TypeName:
456 rawTypeName = v
457 case fieldnum.FieldDescriptorProto_Options:
458 fd.unmarshalOptions(v)
459 rawOptions = appendOptions(rawOptions, v)
460 }
461 default:
462 m := wire.ConsumeFieldValue(num, typ, b)
463 b = b[m:]
464 }
465 }
466 if rawTypeName != nil {
Joe Tsai97a87392019-07-07 01:49:59 -0700467 name := makeFullName(sb, rawTypeName)
Joe Tsaid8881392019-06-06 13:01:53 -0700468 switch fd.L1.Kind {
469 case pref.EnumKind:
470 fd.L1.Enum = PlaceholderEnum(name)
471 case pref.MessageKind, pref.GroupKind:
472 fd.L1.Message = PlaceholderMessage(name)
473 }
474 }
Damien Neil4e173a52019-08-02 13:16:57 -0700475 fd.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Field, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700476}
477
478func (fd *Field) unmarshalOptions(b []byte) {
Joe Tsaic51e2e02019-07-13 00:44:41 -0700479 const FieldOptions_EnforceUTF8 = 13
480
Joe Tsaid8881392019-06-06 13:01:53 -0700481 for len(b) > 0 {
482 num, typ, n := wire.ConsumeTag(b)
483 b = b[n:]
484 switch typ {
485 case wire.VarintType:
486 v, m := wire.ConsumeVarint(b)
487 b = b[m:]
488 switch num {
489 case fieldnum.FieldOptions_Packed:
490 fd.L1.HasPacked = true
491 fd.L1.IsPacked = wire.DecodeBool(v)
492 case fieldnum.FieldOptions_Weak:
493 fd.L1.IsWeak = wire.DecodeBool(v)
Joe Tsaic51e2e02019-07-13 00:44:41 -0700494 case FieldOptions_EnforceUTF8:
495 fd.L1.HasEnforceUTF8 = true
Damien Neil45f14b42019-07-31 11:29:40 -0700496 fd.L1.EnforceUTF8 = wire.DecodeBool(v)
Joe Tsaid8881392019-06-06 13:01:53 -0700497 }
498 default:
499 m := wire.ConsumeFieldValue(num, typ, b)
500 b = b[m:]
501 }
502 }
503}
504
Joe Tsai97a87392019-07-07 01:49:59 -0700505func (od *Oneof) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd pref.Descriptor, i int) {
Joe Tsaid8881392019-06-06 13:01:53 -0700506 od.L0.ParentFile = pf
507 od.L0.Parent = pd
508 od.L0.Index = i
509
510 var rawOptions []byte
511 for len(b) > 0 {
512 num, typ, n := wire.ConsumeTag(b)
513 b = b[n:]
514 switch typ {
515 case wire.BytesType:
516 v, m := wire.ConsumeBytes(b)
517 b = b[m:]
518 switch num {
519 case fieldnum.OneofDescriptorProto_Name:
Joe Tsai97a87392019-07-07 01:49:59 -0700520 od.L0.FullName = appendFullName(sb, pd.FullName(), v)
Joe Tsaid8881392019-06-06 13:01:53 -0700521 case fieldnum.OneofDescriptorProto_Options:
522 rawOptions = appendOptions(rawOptions, v)
523 }
524 default:
525 m := wire.ConsumeFieldValue(num, typ, b)
526 b = b[m:]
527 }
528 }
Damien Neil4e173a52019-08-02 13:16:57 -0700529 od.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Oneof, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700530}
531
Joe Tsai97a87392019-07-07 01:49:59 -0700532func (xd *Extension) unmarshalFull(b []byte, sb *strs.Builder) {
Joe Tsaid8881392019-06-06 13:01:53 -0700533 var rawTypeName []byte
534 var rawOptions []byte
535 xd.L2 = new(ExtensionL2)
536 for len(b) > 0 {
537 num, typ, n := wire.ConsumeTag(b)
538 b = b[n:]
539 switch typ {
Joe Tsaid8881392019-06-06 13:01:53 -0700540 case wire.BytesType:
541 v, m := wire.ConsumeBytes(b)
542 b = b[m:]
543 switch num {
544 case fieldnum.FieldDescriptorProto_JsonName:
Joe Tsai6c26a042020-01-03 15:53:47 -0800545 xd.L2.JSONName.Init(sb.MakeString(v))
Joe Tsaid8881392019-06-06 13:01:53 -0700546 case fieldnum.FieldDescriptorProto_DefaultValue:
Joe Tsai84177c92019-09-17 13:38:48 -0700547 xd.L2.Default.val = pref.ValueOfBytes(v) // temporarily store as bytes; later resolved in resolveExtensions
Joe Tsaid8881392019-06-06 13:01:53 -0700548 case fieldnum.FieldDescriptorProto_TypeName:
549 rawTypeName = v
550 case fieldnum.FieldDescriptorProto_Options:
551 xd.unmarshalOptions(v)
552 rawOptions = appendOptions(rawOptions, v)
553 }
554 default:
555 m := wire.ConsumeFieldValue(num, typ, b)
556 b = b[m:]
557 }
558 }
559 if rawTypeName != nil {
Joe Tsai97a87392019-07-07 01:49:59 -0700560 name := makeFullName(sb, rawTypeName)
Joe Tsaid8881392019-06-06 13:01:53 -0700561 switch xd.L1.Kind {
562 case pref.EnumKind:
563 xd.L2.Enum = PlaceholderEnum(name)
564 case pref.MessageKind, pref.GroupKind:
565 xd.L2.Message = PlaceholderMessage(name)
566 }
567 }
Damien Neil4e173a52019-08-02 13:16:57 -0700568 xd.L2.Options = xd.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Field, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700569}
570
571func (xd *Extension) unmarshalOptions(b []byte) {
572 for len(b) > 0 {
573 num, typ, n := wire.ConsumeTag(b)
574 b = b[n:]
575 switch typ {
576 case wire.VarintType:
577 v, m := wire.ConsumeVarint(b)
578 b = b[m:]
579 switch num {
580 case fieldnum.FieldOptions_Packed:
581 xd.L2.IsPacked = wire.DecodeBool(v)
582 }
583 default:
584 m := wire.ConsumeFieldValue(num, typ, b)
585 b = b[m:]
586 }
587 }
588}
589
Joe Tsai97a87392019-07-07 01:49:59 -0700590func (sd *Service) unmarshalFull(b []byte, sb *strs.Builder) {
Joe Tsaid8881392019-06-06 13:01:53 -0700591 var rawMethods [][]byte
592 var rawOptions []byte
593 sd.L2 = new(ServiceL2)
594 for len(b) > 0 {
595 num, typ, n := wire.ConsumeTag(b)
596 b = b[n:]
597 switch typ {
598 case wire.BytesType:
599 v, m := wire.ConsumeBytes(b)
600 b = b[m:]
601 switch num {
602 case fieldnum.ServiceDescriptorProto_Method:
603 rawMethods = append(rawMethods, v)
604 case fieldnum.ServiceDescriptorProto_Options:
605 rawOptions = appendOptions(rawOptions, v)
606 }
607 default:
608 m := wire.ConsumeFieldValue(num, typ, b)
609 b = b[m:]
610 }
611 }
612 if len(rawMethods) > 0 {
613 sd.L2.Methods.List = make([]Method, len(rawMethods))
614 for i, b := range rawMethods {
Joe Tsai97a87392019-07-07 01:49:59 -0700615 sd.L2.Methods.List[i].unmarshalFull(b, sb, sd.L0.ParentFile, sd, i)
Joe Tsaid8881392019-06-06 13:01:53 -0700616 }
617 }
Damien Neil4e173a52019-08-02 13:16:57 -0700618 sd.L2.Options = sd.L0.ParentFile.builder.optionsUnmarshaler(&descopts.Service, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700619}
620
Joe Tsai97a87392019-07-07 01:49:59 -0700621func (md *Method) unmarshalFull(b []byte, sb *strs.Builder, pf *File, pd pref.Descriptor, i int) {
Joe Tsaid8881392019-06-06 13:01:53 -0700622 md.L0.ParentFile = pf
623 md.L0.Parent = pd
624 md.L0.Index = i
625
626 var rawOptions []byte
627 for len(b) > 0 {
628 num, typ, n := wire.ConsumeTag(b)
629 b = b[n:]
630 switch typ {
631 case wire.VarintType:
632 v, m := wire.ConsumeVarint(b)
633 b = b[m:]
634 switch num {
635 case fieldnum.MethodDescriptorProto_ClientStreaming:
636 md.L1.IsStreamingClient = wire.DecodeBool(v)
637 case fieldnum.MethodDescriptorProto_ServerStreaming:
638 md.L1.IsStreamingServer = wire.DecodeBool(v)
639 }
640 case wire.BytesType:
641 v, m := wire.ConsumeBytes(b)
642 b = b[m:]
643 switch num {
644 case fieldnum.MethodDescriptorProto_Name:
Joe Tsai97a87392019-07-07 01:49:59 -0700645 md.L0.FullName = appendFullName(sb, pd.FullName(), v)
Joe Tsaid8881392019-06-06 13:01:53 -0700646 case fieldnum.MethodDescriptorProto_InputType:
Joe Tsai97a87392019-07-07 01:49:59 -0700647 md.L1.Input = PlaceholderMessage(makeFullName(sb, v))
Joe Tsaid8881392019-06-06 13:01:53 -0700648 case fieldnum.MethodDescriptorProto_OutputType:
Joe Tsai97a87392019-07-07 01:49:59 -0700649 md.L1.Output = PlaceholderMessage(makeFullName(sb, v))
Joe Tsaid8881392019-06-06 13:01:53 -0700650 case fieldnum.MethodDescriptorProto_Options:
651 rawOptions = appendOptions(rawOptions, v)
652 }
653 default:
654 m := wire.ConsumeFieldValue(num, typ, b)
655 b = b[m:]
656 }
657 }
Damien Neil4e173a52019-08-02 13:16:57 -0700658 md.L1.Options = pf.builder.optionsUnmarshaler(&descopts.Method, rawOptions)
Joe Tsaid8881392019-06-06 13:01:53 -0700659}
660
661// appendOptions appends src to dst, where the returned slice is never nil.
662// This is necessary to distinguish between empty and unpopulated options.
663func appendOptions(dst, src []byte) []byte {
664 if dst == nil {
665 dst = []byte{}
666 }
667 return append(dst, src...)
668}
669
Damien Neil4e173a52019-08-02 13:16:57 -0700670// optionsUnmarshaler constructs a lazy unmarshal function for an options message.
671//
672// The type of message to unmarshal to is passed as a pointer since the
673// vars in descopts may not yet be populated at the time this function is called.
Joe Tsai52ec1752019-08-05 15:49:29 -0700674func (db *Builder) optionsUnmarshaler(p *pref.ProtoMessage, b []byte) func() pref.ProtoMessage {
Joe Tsaid8881392019-06-06 13:01:53 -0700675 if b == nil {
676 return nil
677 }
Joe Tsai8a4c3d12019-07-06 18:08:55 -0700678 var opts pref.ProtoMessage
679 var once sync.Once
Joe Tsaid8881392019-06-06 13:01:53 -0700680 return func() pref.ProtoMessage {
Joe Tsai8a4c3d12019-07-06 18:08:55 -0700681 once.Do(func() {
Damien Neil4e173a52019-08-02 13:16:57 -0700682 if *p == nil {
683 panic("Descriptor.Options called without importing the descriptor package")
684 }
685 opts = reflect.New(reflect.TypeOf(*p).Elem()).Interface().(pref.ProtoMessage)
Joe Tsai8a4c3d12019-07-06 18:08:55 -0700686 if err := (proto.UnmarshalOptions{
687 AllowPartial: true,
688 Resolver: db.TypeResolver,
689 }).Unmarshal(b, opts); err != nil {
690 panic(err)
691 }
692 })
693 return opts
Joe Tsaid8881392019-06-06 13:01:53 -0700694 }
695}