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