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.
76 lines
2.5 KiB
76 lines
2.5 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/internal/message.h" |
|
|
|
#include <math.h> |
|
#include <string.h> |
|
|
|
#include "upb/base/internal/log2.h" |
|
#include "upb/mem/arena.h" |
|
#include "upb/message/internal/types.h" |
|
|
|
// Must be last. |
|
#include "upb/port/def.inc" |
|
|
|
const float kUpb_FltInfinity = INFINITY; |
|
const double kUpb_Infinity = INFINITY; |
|
const double kUpb_NaN = NAN; |
|
|
|
bool UPB_PRIVATE(_upb_Message_Realloc)(struct upb_Message* msg, size_t need, |
|
upb_Arena* a) { |
|
const size_t overhead = sizeof(upb_Message_Internal); |
|
|
|
upb_Message_Internal* in = UPB_PRIVATE(_upb_Message_GetInternal)(msg); |
|
if (!in) { |
|
// No internal data, allocate from scratch. |
|
size_t size = UPB_MAX(128, upb_Log2CeilingSize(need + overhead)); |
|
in = upb_Arena_Malloc(a, size); |
|
if (!in) return false; |
|
|
|
in->size = size; |
|
in->unknown_end = overhead; |
|
in->ext_begin = size; |
|
UPB_PRIVATE(_upb_Message_SetInternal)(msg, in); |
|
} else if (in->ext_begin - in->unknown_end < need) { |
|
// Internal data is too small, reallocate. |
|
size_t new_size = upb_Log2CeilingSize(in->size + need); |
|
size_t ext_bytes = in->size - in->ext_begin; |
|
size_t new_ext_begin = new_size - ext_bytes; |
|
in = upb_Arena_Realloc(a, in, in->size, new_size); |
|
if (!in) return false; |
|
|
|
if (ext_bytes) { |
|
// Need to move extension data to the end. |
|
char* ptr = (char*)in; |
|
memmove(ptr + new_ext_begin, ptr + in->ext_begin, ext_bytes); |
|
} |
|
in->ext_begin = new_ext_begin; |
|
in->size = new_size; |
|
UPB_PRIVATE(_upb_Message_SetInternal)(msg, in); |
|
} |
|
|
|
UPB_ASSERT(in->ext_begin - in->unknown_end >= need); |
|
return true; |
|
} |
|
|
|
#if UPB_TRACING_ENABLED |
|
static void (*_new_message_trace_handler)(const upb_MiniTable*, |
|
const upb_Arena*); |
|
|
|
void UPB_PRIVATE(upb_Message_SetNewMessageTraceHandler)( |
|
void (*new_message_trace_handler)(const upb_MiniTable*, const upb_Arena*)) { |
|
_new_message_trace_handler = new_message_trace_handler; |
|
} |
|
|
|
void UPB_PRIVATE(upb_Message_LogNewMessage)(const upb_MiniTable* mini_table, |
|
const upb_Arena* arena) { |
|
if (_new_message_trace_handler) { |
|
_new_message_trace_handler(mini_table, arena); |
|
} |
|
} |
|
#endif
|
|
|