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/accessors.h"
9
10 #include <string.h>
11
12 #include "upb/mem/arena.h"
13 #include "upb/message/array.h"
14 #include "upb/message/map.h"
15 #include "upb/message/message.h"
16 #include "upb/mini_table/field.h"
17 #include "upb/mini_table/message.h"
18 #include "upb/mini_table/sub.h"
19
20 // Must be last.
21 #include "upb/port/def.inc"
22
upb_Message_SetMapEntry(upb_Map * map,const upb_MiniTable * mini_table,const upb_MiniTableField * f,upb_Message * map_entry_message,upb_Arena * arena)23 bool upb_Message_SetMapEntry(upb_Map* map, const upb_MiniTable* mini_table,
24 const upb_MiniTableField* f,
25 upb_Message* map_entry_message, upb_Arena* arena) {
26 // TODO: use a variant of upb_MiniTable_GetSubMessageTable() here.
27 const upb_MiniTable* map_entry_mini_table = upb_MiniTableSub_Message(
28 mini_table->UPB_PRIVATE(subs)[f->UPB_PRIVATE(submsg_index)]);
29 UPB_ASSERT(map_entry_mini_table);
30 UPB_ASSERT(map_entry_mini_table->UPB_PRIVATE(field_count) == 2);
31 const upb_MiniTableField* map_entry_key_field =
32 &map_entry_mini_table->UPB_PRIVATE(fields)[0];
33 const upb_MiniTableField* map_entry_value_field =
34 &map_entry_mini_table->UPB_PRIVATE(fields)[1];
35 // Map key/value cannot have explicit defaults,
36 // hence assuming a zero default is valid.
37 upb_MessageValue default_val;
38 memset(&default_val, 0, sizeof(upb_MessageValue));
39 upb_MessageValue map_entry_key =
40 upb_Message_GetField(map_entry_message, map_entry_key_field, default_val);
41 upb_MessageValue map_entry_value = upb_Message_GetField(
42 map_entry_message, map_entry_value_field, default_val);
43 return upb_Map_Set(map, map_entry_key, map_entry_value, arena);
44 }
45