1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2023 Google LLC. All rights reserved.
3 //
4 // Use of this source code is governed by a BSD-style
5 // license that can be found in the LICENSE file or at
6 // https://developers.google.com/open-source/licenses/bsd
7
8 #include "upb/reflection/internal/extension_range.h"
9
10 #include <stdint.h>
11
12 #include "upb/reflection/extension_range.h"
13 #include "upb/reflection/field_def.h"
14 #include "upb/reflection/internal/def_builder.h"
15 #include "upb/reflection/message_def.h"
16
17 // Must be last.
18 #include "upb/port/def.inc"
19
20 struct upb_ExtensionRange {
21 const UPB_DESC(ExtensionRangeOptions*) opts;
22 const UPB_DESC(FeatureSet*) resolved_features;
23 int32_t start;
24 int32_t end;
25 };
26
_upb_ExtensionRange_At(const upb_ExtensionRange * r,int i)27 upb_ExtensionRange* _upb_ExtensionRange_At(const upb_ExtensionRange* r, int i) {
28 return (upb_ExtensionRange*)&r[i];
29 }
30
UPB_DESC(ExtensionRangeOptions)31 const UPB_DESC(ExtensionRangeOptions) *
32 upb_ExtensionRange_Options(const upb_ExtensionRange* r) {
33 return r->opts;
34 }
35
upb_ExtensionRange_HasOptions(const upb_ExtensionRange * r)36 bool upb_ExtensionRange_HasOptions(const upb_ExtensionRange* r) {
37 return r->opts != (void*)kUpbDefOptDefault;
38 }
39
upb_ExtensionRange_Start(const upb_ExtensionRange * r)40 int32_t upb_ExtensionRange_Start(const upb_ExtensionRange* r) {
41 return r->start;
42 }
43
upb_ExtensionRange_End(const upb_ExtensionRange * r)44 int32_t upb_ExtensionRange_End(const upb_ExtensionRange* r) { return r->end; }
45
_upb_ExtensionRanges_New(upb_DefBuilder * ctx,int n,const UPB_DESC (DescriptorProto_ExtensionRange *)const * protos,const UPB_DESC (FeatureSet *)parent_features,const upb_MessageDef * m)46 upb_ExtensionRange* _upb_ExtensionRanges_New(
47 upb_DefBuilder* ctx, int n,
48 const UPB_DESC(DescriptorProto_ExtensionRange*) const* protos,
49 const UPB_DESC(FeatureSet*) parent_features, const upb_MessageDef* m) {
50 upb_ExtensionRange* r =
51 _upb_DefBuilder_Alloc(ctx, sizeof(upb_ExtensionRange) * n);
52
53 for (int i = 0; i < n; i++) {
54 UPB_DEF_SET_OPTIONS(r[i].opts, DescriptorProto_ExtensionRange,
55 ExtensionRangeOptions, protos[i]);
56 r[i].resolved_features = _upb_DefBuilder_ResolveFeatures(
57 ctx, parent_features,
58 UPB_DESC(ExtensionRangeOptions_features)(r[i].opts));
59
60 const int32_t start =
61 UPB_DESC(DescriptorProto_ExtensionRange_start)(protos[i]);
62 const int32_t end = UPB_DESC(DescriptorProto_ExtensionRange_end)(protos[i]);
63 const int32_t max = UPB_DESC(MessageOptions_message_set_wire_format)(
64 upb_MessageDef_Options(m))
65 ? INT32_MAX
66 : kUpb_MaxFieldNumber + 1;
67
68 // A full validation would also check that each range is disjoint, and that
69 // none of the fields overlap with the extension ranges, but we are just
70 // sanity checking here.
71 if (start < 1 || end <= start || end > max) {
72 _upb_DefBuilder_Errf(ctx,
73 "Extension range (%d, %d) is invalid, message=%s\n",
74 (int)start, (int)end, upb_MessageDef_FullName(m));
75 }
76
77 r[i].start = start;
78 r[i].end = end;
79 }
80
81 return r;
82 }
83