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/internal/message.h"
9
10 #include <math.h>
11 #include <string.h>
12
13 #include "upb/base/internal/log2.h"
14 #include "upb/mem/arena.h"
15 #include "upb/message/internal/types.h"
16
17 // Must be last.
18 #include "upb/port/def.inc"
19
20 const float kUpb_FltInfinity = (float)(1.0 / 0.0);
21 const double kUpb_Infinity = 1.0 / 0.0;
22 const double kUpb_NaN = 0.0 / 0.0;
23
UPB_PRIVATE(_upb_Message_Realloc)24 bool UPB_PRIVATE(_upb_Message_Realloc)(struct upb_Message* msg, size_t need,
25 upb_Arena* a) {
26 UPB_ASSERT(!upb_Message_IsFrozen(msg));
27 const size_t overhead = sizeof(upb_Message_Internal);
28
29 upb_Message_Internal* in = UPB_PRIVATE(_upb_Message_GetInternal)(msg);
30 if (!in) {
31 // No internal data, allocate from scratch.
32 size_t size = UPB_MAX(128, upb_Log2CeilingSize(need + overhead));
33 in = upb_Arena_Malloc(a, size);
34 if (!in) return false;
35
36 in->size = size;
37 in->unknown_end = overhead;
38 in->ext_begin = size;
39 UPB_PRIVATE(_upb_Message_SetInternal)(msg, in);
40 } else if (in->ext_begin - in->unknown_end < need) {
41 // Internal data is too small, reallocate.
42 size_t new_size = upb_Log2CeilingSize(in->size + need);
43 size_t ext_bytes = in->size - in->ext_begin;
44 size_t new_ext_begin = new_size - ext_bytes;
45 in = upb_Arena_Realloc(a, in, in->size, new_size);
46 if (!in) return false;
47
48 if (ext_bytes) {
49 // Need to move extension data to the end.
50 char* ptr = (char*)in;
51 memmove(ptr + new_ext_begin, ptr + in->ext_begin, ext_bytes);
52 }
53 in->ext_begin = new_ext_begin;
54 in->size = new_size;
55 UPB_PRIVATE(_upb_Message_SetInternal)(msg, in);
56 }
57
58 UPB_ASSERT(in->ext_begin - in->unknown_end >= need);
59 return true;
60 }
61
62 #if UPB_TRACING_ENABLED
63 static void (*_message_trace_handler)(const upb_MiniTable*, const upb_Arena*);
64
upb_Message_LogNewMessage(const upb_MiniTable * m,const upb_Arena * arena)65 void upb_Message_LogNewMessage(const upb_MiniTable* m, const upb_Arena* arena) {
66 if (_message_trace_handler) {
67 _message_trace_handler(m, arena);
68 }
69 }
70
upb_Message_SetNewMessageTraceHandler(void (* handler)(const upb_MiniTable *,const upb_Arena *))71 void upb_Message_SetNewMessageTraceHandler(void (*handler)(const upb_MiniTable*,
72 const upb_Arena*)) {
73 _message_trace_handler = handler;
74 }
75 #endif // UPB_TRACING_ENABLED
76