1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2024 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/message/internal/iterator.h" // IWYU pragma: keep
9
10 #include <stddef.h>
11
12 #include "upb/message/accessors.h"
13 #include "upb/message/array.h"
14 #include "upb/message/internal/accessors.h"
15 #include "upb/message/internal/extension.h"
16 #include "upb/message/map.h"
17 #include "upb/message/message.h"
18 #include "upb/mini_table/extension.h"
19 #include "upb/mini_table/field.h"
20 #include "upb/mini_table/message.h"
21
22 // Must be last.
23 #include "upb/port/def.inc"
24
UPB_PRIVATE(_upb_Message_NextBaseField)25 bool UPB_PRIVATE(_upb_Message_NextBaseField)(const upb_Message* msg,
26 const upb_MiniTable* m,
27 const upb_MiniTableField** out_f,
28 upb_MessageValue* out_v,
29 size_t* iter) {
30 const size_t count = upb_MiniTable_FieldCount(m);
31 size_t i = *iter;
32
33 while (++i < count) {
34 const upb_MiniTableField* f = upb_MiniTable_GetFieldByIndex(m, i);
35 const void* src = UPB_PRIVATE(_upb_Message_DataPtr)(msg, f);
36
37 upb_MessageValue val;
38 UPB_PRIVATE(_upb_MiniTableField_DataCopy)(f, &val, src);
39
40 // Skip field if unset or empty.
41 if (upb_MiniTableField_HasPresence(f)) {
42 if (!upb_Message_HasBaseField(msg, f)) continue;
43 } else {
44 if (UPB_PRIVATE(_upb_MiniTableField_DataIsZero)(f, src)) continue;
45
46 if (upb_MiniTableField_IsArray(f)) {
47 if (upb_Array_Size(val.array_val) == 0) continue;
48 } else if (upb_MiniTableField_IsMap(f)) {
49 if (upb_Map_Size(val.map_val) == 0) continue;
50 }
51 }
52
53 *out_f = f;
54 *out_v = val;
55 *iter = i;
56 return true;
57 }
58
59 return false;
60 }
61
UPB_PRIVATE(_upb_Message_NextExtension)62 bool UPB_PRIVATE(_upb_Message_NextExtension)(
63 const upb_Message* msg, const upb_MiniTable* m,
64 const upb_MiniTableExtension** out_e, upb_MessageValue* out_v,
65 size_t* iter) {
66 size_t count;
67 const upb_Extension* exts = UPB_PRIVATE(_upb_Message_Getexts)(msg, &count);
68 size_t i = *iter;
69
70 if (++i < count) {
71 *out_e = exts[i].ext;
72 *out_v = exts[i].data;
73 *iter = i;
74 return true;
75 }
76
77 return false;
78 }