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/message/map.h"
9
10 #include <stdint.h>
11 #include <string.h>
12
13 #include "upb/base/descriptor_constants.h"
14 #include "upb/base/string_view.h"
15 #include "upb/hash/common.h"
16 #include "upb/hash/str_table.h"
17 #include "upb/mem/arena.h"
18 #include "upb/message/internal/map.h"
19 #include "upb/message/map.h"
20 #include "upb/message/message.h"
21 #include "upb/message/value.h"
22 #include "upb/mini_table/field.h"
23 #include "upb/mini_table/message.h"
24
25 // Must be last.
26 #include "upb/port/def.inc"
27
28 // Strings/bytes are special-cased in maps.
29 char _upb_Map_CTypeSizeTable[12] = {
30 [kUpb_CType_Bool] = 1,
31 [kUpb_CType_Float] = 4,
32 [kUpb_CType_Int32] = 4,
33 [kUpb_CType_UInt32] = 4,
34 [kUpb_CType_Enum] = 4,
35 [kUpb_CType_Message] = sizeof(void*),
36 [kUpb_CType_Double] = 8,
37 [kUpb_CType_Int64] = 8,
38 [kUpb_CType_UInt64] = 8,
39 [kUpb_CType_String] = UPB_MAPTYPE_STRING,
40 [kUpb_CType_Bytes] = UPB_MAPTYPE_STRING,
41 };
42
upb_Map_New(upb_Arena * a,upb_CType key_type,upb_CType value_type)43 upb_Map* upb_Map_New(upb_Arena* a, upb_CType key_type, upb_CType value_type) {
44 return _upb_Map_New(a, _upb_Map_CTypeSize(key_type),
45 _upb_Map_CTypeSize(value_type));
46 }
47
upb_Map_Size(const upb_Map * map)48 size_t upb_Map_Size(const upb_Map* map) { return _upb_Map_Size(map); }
49
upb_Map_Get(const upb_Map * map,upb_MessageValue key,upb_MessageValue * val)50 bool upb_Map_Get(const upb_Map* map, upb_MessageValue key,
51 upb_MessageValue* val) {
52 return _upb_Map_Get(map, &key, map->key_size, val, map->val_size);
53 }
54
upb_Map_Clear(upb_Map * map)55 void upb_Map_Clear(upb_Map* map) { _upb_Map_Clear(map); }
56
upb_Map_Insert(upb_Map * map,upb_MessageValue key,upb_MessageValue val,upb_Arena * arena)57 upb_MapInsertStatus upb_Map_Insert(upb_Map* map, upb_MessageValue key,
58 upb_MessageValue val, upb_Arena* arena) {
59 UPB_ASSERT(arena);
60 return (upb_MapInsertStatus)_upb_Map_Insert(map, &key, map->key_size, &val,
61 map->val_size, arena);
62 }
63
upb_Map_Delete(upb_Map * map,upb_MessageValue key,upb_MessageValue * val)64 bool upb_Map_Delete(upb_Map* map, upb_MessageValue key, upb_MessageValue* val) {
65 upb_value v;
66 const bool removed = _upb_Map_Delete(map, &key, map->key_size, &v);
67 if (val) _upb_map_fromvalue(v, val, map->val_size);
68 return removed;
69 }
70
upb_Map_Next(const upb_Map * map,upb_MessageValue * key,upb_MessageValue * val,size_t * iter)71 bool upb_Map_Next(const upb_Map* map, upb_MessageValue* key,
72 upb_MessageValue* val, size_t* iter) {
73 upb_StringView k;
74 upb_value v;
75 const bool ok = upb_strtable_next2(&map->table, &k, &v, (intptr_t*)iter);
76 if (ok) {
77 _upb_map_fromkey(k, key, map->key_size);
78 _upb_map_fromvalue(v, val, map->val_size);
79 }
80 return ok;
81 }
82
upb_Map_SetEntryValue(upb_Map * map,size_t iter,upb_MessageValue val)83 UPB_API void upb_Map_SetEntryValue(upb_Map* map, size_t iter,
84 upb_MessageValue val) {
85 upb_value v;
86 _upb_map_tovalue(&val, map->val_size, &v, NULL);
87 upb_strtable_setentryvalue(&map->table, iter, v);
88 }
89
upb_MapIterator_Next(const upb_Map * map,size_t * iter)90 bool upb_MapIterator_Next(const upb_Map* map, size_t* iter) {
91 return _upb_map_next(map, iter);
92 }
93
upb_MapIterator_Done(const upb_Map * map,size_t iter)94 bool upb_MapIterator_Done(const upb_Map* map, size_t iter) {
95 upb_strtable_iter i;
96 UPB_ASSERT(iter != kUpb_Map_Begin);
97 i.t = &map->table;
98 i.index = iter;
99 return upb_strtable_done(&i);
100 }
101
102 // Returns the key and value for this entry of the map.
upb_MapIterator_Key(const upb_Map * map,size_t iter)103 upb_MessageValue upb_MapIterator_Key(const upb_Map* map, size_t iter) {
104 upb_strtable_iter i;
105 upb_MessageValue ret;
106 i.t = &map->table;
107 i.index = iter;
108 _upb_map_fromkey(upb_strtable_iter_key(&i), &ret, map->key_size);
109 return ret;
110 }
111
upb_MapIterator_Value(const upb_Map * map,size_t iter)112 upb_MessageValue upb_MapIterator_Value(const upb_Map* map, size_t iter) {
113 upb_strtable_iter i;
114 upb_MessageValue ret;
115 i.t = &map->table;
116 i.index = iter;
117 _upb_map_fromvalue(upb_strtable_iter_value(&i), &ret, map->val_size);
118 return ret;
119 }
120
upb_Map_Freeze(upb_Map * map,const upb_MiniTable * m)121 void upb_Map_Freeze(upb_Map* map, const upb_MiniTable* m) {
122 if (upb_Map_IsFrozen(map)) return;
123 UPB_PRIVATE(_upb_Map_ShallowFreeze)(map);
124
125 if (m) {
126 size_t iter = kUpb_Map_Begin;
127 upb_MessageValue key, val;
128
129 while (upb_Map_Next(map, &key, &val, &iter)) {
130 upb_Message_Freeze((upb_Message*)val.msg_val, m);
131 }
132 }
133 }
134
135 // EVERYTHING BELOW THIS LINE IS INTERNAL - DO NOT USE /////////////////////////
136
_upb_Map_New(upb_Arena * a,size_t key_size,size_t value_size)137 upb_Map* _upb_Map_New(upb_Arena* a, size_t key_size, size_t value_size) {
138 upb_Map* map = upb_Arena_Malloc(a, sizeof(upb_Map));
139 if (!map) return NULL;
140
141 upb_strtable_init(&map->table, 4, a);
142 map->key_size = key_size;
143 map->val_size = value_size;
144 map->UPB_PRIVATE(is_frozen) = false;
145
146 return map;
147 }
148