Protocol Buffers - Google's data interchange format (grpc依赖)
https://developers.google.com/protocol-buffers/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
2.7 KiB
69 lines
2.7 KiB
// Protocol Buffers - Google's data interchange format |
|
// Copyright 2023 Google LLC. All rights reserved. |
|
// |
|
// Use of this source code is governed by a BSD-style |
|
// license that can be found in the LICENSE file or at |
|
// https://developers.google.com/open-source/licenses/bsd |
|
|
|
#include "upb/message/accessors.h" |
|
|
|
#include <string.h> |
|
|
|
#include "upb/mem/arena.h" |
|
#include "upb/message/array.h" |
|
#include "upb/message/map.h" |
|
#include "upb/message/message.h" |
|
#include "upb/mini_table/field.h" |
|
#include "upb/mini_table/message.h" |
|
#include "upb/mini_table/sub.h" |
|
#include "upb/wire/encode.h" |
|
|
|
// Must be last. |
|
#include "upb/port/def.inc" |
|
|
|
bool upb_Message_SetMapEntry(upb_Map* map, const upb_MiniTable* mini_table, |
|
const upb_MiniTableField* f, |
|
upb_Message* map_entry_message, upb_Arena* arena) { |
|
// TODO: use a variant of upb_MiniTable_GetSubMessageTable() here. |
|
const upb_MiniTable* map_entry_mini_table = upb_MiniTableSub_Message( |
|
mini_table->UPB_PRIVATE(subs)[f->UPB_PRIVATE(submsg_index)]); |
|
UPB_ASSERT(map_entry_mini_table); |
|
UPB_ASSERT(map_entry_mini_table->UPB_PRIVATE(field_count) == 2); |
|
const upb_MiniTableField* map_entry_key_field = |
|
&map_entry_mini_table->UPB_PRIVATE(fields)[0]; |
|
const upb_MiniTableField* map_entry_value_field = |
|
&map_entry_mini_table->UPB_PRIVATE(fields)[1]; |
|
// Map key/value cannot have explicit defaults, |
|
// hence assuming a zero default is valid. |
|
upb_MessageValue default_val; |
|
memset(&default_val, 0, sizeof(upb_MessageValue)); |
|
upb_MessageValue map_entry_key = |
|
upb_Message_GetField(map_entry_message, map_entry_key_field, default_val); |
|
upb_MessageValue map_entry_value = upb_Message_GetField( |
|
map_entry_message, map_entry_value_field, default_val); |
|
return upb_Map_Set(map, map_entry_key, map_entry_value, arena); |
|
} |
|
|
|
bool upb_Message_IsExactlyEqual(const upb_Message* m1, const upb_Message* m2, |
|
const upb_MiniTable* layout) { |
|
if (m1 == m2) return true; |
|
|
|
int opts = kUpb_EncodeOption_SkipUnknown | kUpb_EncodeOption_Deterministic; |
|
upb_Arena* a = upb_Arena_New(); |
|
|
|
// Compare deterministically serialized payloads with no unknown fields. |
|
size_t size1, size2; |
|
char *data1, *data2; |
|
upb_EncodeStatus status1 = upb_Encode(m1, layout, opts, a, &data1, &size1); |
|
upb_EncodeStatus status2 = upb_Encode(m2, layout, opts, a, &data2, &size2); |
|
|
|
if (status1 != kUpb_EncodeStatus_Ok || status2 != kUpb_EncodeStatus_Ok) { |
|
// TODO: How should we fail here? (In Ruby we throw an exception.) |
|
upb_Arena_Free(a); |
|
return false; |
|
} |
|
|
|
const bool ret = (size1 == size2) && (memcmp(data1, data2, size1) == 0); |
|
upb_Arena_Free(a); |
|
return ret; |
|
}
|
|
|