create
This commit is contained in:
84
Pods/gRPC-Core/third_party/upb/upb/base/descriptor_constants.h
generated
vendored
Normal file
84
Pods/gRPC-Core/third_party/upb/upb/base/descriptor_constants.h
generated
vendored
Normal file
@@ -0,0 +1,84 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_BASE_DESCRIPTOR_CONSTANTS_H_
|
||||
#define UPB_BASE_DESCRIPTOR_CONSTANTS_H_
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// The types a field can have. Note that this list is not identical to the
|
||||
// types defined in descriptor.proto, which gives INT32 and SINT32 separate
|
||||
// types (we distinguish the two with the "integer encoding" enum below).
|
||||
// This enum is an internal convenience only and has no meaning outside of upb.
|
||||
typedef enum {
|
||||
kUpb_CType_Bool = 1,
|
||||
kUpb_CType_Float = 2,
|
||||
kUpb_CType_Int32 = 3,
|
||||
kUpb_CType_UInt32 = 4,
|
||||
kUpb_CType_Enum = 5, // Enum values are int32. TODO: rename
|
||||
kUpb_CType_Message = 6,
|
||||
kUpb_CType_Double = 7,
|
||||
kUpb_CType_Int64 = 8,
|
||||
kUpb_CType_UInt64 = 9,
|
||||
kUpb_CType_String = 10,
|
||||
kUpb_CType_Bytes = 11
|
||||
} upb_CType;
|
||||
|
||||
// The repeated-ness of each field; this matches descriptor.proto.
|
||||
typedef enum {
|
||||
kUpb_Label_Optional = 1,
|
||||
kUpb_Label_Required = 2,
|
||||
kUpb_Label_Repeated = 3
|
||||
} upb_Label;
|
||||
|
||||
// Descriptor types, as defined in descriptor.proto.
|
||||
typedef enum {
|
||||
kUpb_FieldType_Double = 1,
|
||||
kUpb_FieldType_Float = 2,
|
||||
kUpb_FieldType_Int64 = 3,
|
||||
kUpb_FieldType_UInt64 = 4,
|
||||
kUpb_FieldType_Int32 = 5,
|
||||
kUpb_FieldType_Fixed64 = 6,
|
||||
kUpb_FieldType_Fixed32 = 7,
|
||||
kUpb_FieldType_Bool = 8,
|
||||
kUpb_FieldType_String = 9,
|
||||
kUpb_FieldType_Group = 10,
|
||||
kUpb_FieldType_Message = 11,
|
||||
kUpb_FieldType_Bytes = 12,
|
||||
kUpb_FieldType_UInt32 = 13,
|
||||
kUpb_FieldType_Enum = 14,
|
||||
kUpb_FieldType_SFixed32 = 15,
|
||||
kUpb_FieldType_SFixed64 = 16,
|
||||
kUpb_FieldType_SInt32 = 17,
|
||||
kUpb_FieldType_SInt64 = 18,
|
||||
} upb_FieldType;
|
||||
|
||||
#define kUpb_FieldType_SizeOf 19
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_INLINE bool upb_FieldType_IsPackable(upb_FieldType type) {
|
||||
// clang-format off
|
||||
const unsigned kUnpackableTypes =
|
||||
(1 << kUpb_FieldType_String) |
|
||||
(1 << kUpb_FieldType_Bytes) |
|
||||
(1 << kUpb_FieldType_Message) |
|
||||
(1 << kUpb_FieldType_Group);
|
||||
// clang-format on
|
||||
return (1 << type) & ~kUnpackableTypes;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_BASE_DESCRIPTOR_CONSTANTS_H_ */
|
||||
38
Pods/gRPC-Core/third_party/upb/upb/base/internal/log2.h
generated
vendored
Normal file
38
Pods/gRPC-Core/third_party/upb/upb/base/internal/log2.h
generated
vendored
Normal file
@@ -0,0 +1,38 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
///
|
||||
// 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
|
||||
|
||||
#ifndef UPB_BASE_INTERNAL_LOG2_H_
|
||||
#define UPB_BASE_INTERNAL_LOG2_H_
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_INLINE int upb_Log2Ceiling(int x) {
|
||||
if (x <= 1) return 0;
|
||||
#ifdef __GNUC__
|
||||
return 32 - __builtin_clz(x - 1);
|
||||
#else
|
||||
int lg2 = 0;
|
||||
while ((1 << lg2) < x) lg2++;
|
||||
return lg2;
|
||||
#endif
|
||||
}
|
||||
|
||||
UPB_INLINE int upb_Log2CeilingSize(int x) { return 1 << upb_Log2Ceiling(x); }
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_BASE_INTERNAL_LOG2_H_ */
|
||||
61
Pods/gRPC-Core/third_party/upb/upb/base/status.c
generated
vendored
Normal file
61
Pods/gRPC-Core/third_party/upb/upb/base/status.c
generated
vendored
Normal file
@@ -0,0 +1,61 @@
|
||||
// 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/base/status.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <float.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
void upb_Status_Clear(upb_Status* status) {
|
||||
if (!status) return;
|
||||
status->ok = true;
|
||||
status->msg[0] = '\0';
|
||||
}
|
||||
|
||||
bool upb_Status_IsOk(const upb_Status* status) { return status->ok; }
|
||||
|
||||
const char* upb_Status_ErrorMessage(const upb_Status* status) {
|
||||
return status->msg;
|
||||
}
|
||||
|
||||
void upb_Status_SetErrorMessage(upb_Status* status, const char* msg) {
|
||||
if (!status) return;
|
||||
status->ok = false;
|
||||
strncpy(status->msg, msg, _kUpb_Status_MaxMessage - 1);
|
||||
status->msg[_kUpb_Status_MaxMessage - 1] = '\0';
|
||||
}
|
||||
|
||||
void upb_Status_SetErrorFormat(upb_Status* status, const char* fmt, ...) {
|
||||
va_list args;
|
||||
va_start(args, fmt);
|
||||
upb_Status_VSetErrorFormat(status, fmt, args);
|
||||
va_end(args);
|
||||
}
|
||||
|
||||
void upb_Status_VSetErrorFormat(upb_Status* status, const char* fmt,
|
||||
va_list args) {
|
||||
if (!status) return;
|
||||
status->ok = false;
|
||||
vsnprintf(status->msg, sizeof(status->msg), fmt, args);
|
||||
status->msg[_kUpb_Status_MaxMessage - 1] = '\0';
|
||||
}
|
||||
|
||||
void upb_Status_VAppendErrorFormat(upb_Status* status, const char* fmt,
|
||||
va_list args) {
|
||||
size_t len;
|
||||
if (!status) return;
|
||||
status->ok = false;
|
||||
len = strlen(status->msg);
|
||||
vsnprintf(status->msg + len, sizeof(status->msg) - len, fmt, args);
|
||||
status->msg[_kUpb_Status_MaxMessage - 1] = '\0';
|
||||
}
|
||||
46
Pods/gRPC-Core/third_party/upb/upb/base/status.h
generated
vendored
Normal file
46
Pods/gRPC-Core/third_party/upb/upb/base/status.h
generated
vendored
Normal file
@@ -0,0 +1,46 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_BASE_STATUS_H_
|
||||
#define UPB_BASE_STATUS_H_
|
||||
|
||||
#include <stdarg.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#define _kUpb_Status_MaxMessage 127
|
||||
|
||||
typedef struct {
|
||||
bool ok;
|
||||
char msg[_kUpb_Status_MaxMessage]; // Error message; NULL-terminated.
|
||||
} upb_Status;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_API const char* upb_Status_ErrorMessage(const upb_Status* status);
|
||||
UPB_API bool upb_Status_IsOk(const upb_Status* status);
|
||||
|
||||
// These are no-op if |status| is NULL.
|
||||
UPB_API void upb_Status_Clear(upb_Status* status);
|
||||
void upb_Status_SetErrorMessage(upb_Status* status, const char* msg);
|
||||
void upb_Status_SetErrorFormat(upb_Status* status, const char* fmt, ...)
|
||||
UPB_PRINTF(2, 3);
|
||||
void upb_Status_VSetErrorFormat(upb_Status* status, const char* fmt,
|
||||
va_list args) UPB_PRINTF(2, 0);
|
||||
void upb_Status_VAppendErrorFormat(upb_Status* status, const char* fmt,
|
||||
va_list args) UPB_PRINTF(2, 0);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_BASE_STATUS_H_ */
|
||||
50
Pods/gRPC-Core/third_party/upb/upb/base/status.hpp
generated
vendored
Normal file
50
Pods/gRPC-Core/third_party/upb/upb/base/status.hpp
generated
vendored
Normal file
@@ -0,0 +1,50 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_BASE_STATUS_HPP_
|
||||
#define UPB_BASE_STATUS_HPP_
|
||||
|
||||
#include "upb/base/status.h"
|
||||
|
||||
namespace upb {
|
||||
|
||||
class Status {
|
||||
public:
|
||||
Status() { upb_Status_Clear(&status_); }
|
||||
|
||||
upb_Status* ptr() { return &status_; }
|
||||
|
||||
// Returns true if there is no error.
|
||||
bool ok() const { return upb_Status_IsOk(&status_); }
|
||||
|
||||
// Guaranteed to be NULL-terminated.
|
||||
const char* error_message() const {
|
||||
return upb_Status_ErrorMessage(&status_);
|
||||
}
|
||||
|
||||
// The error message will be truncated if it is longer than
|
||||
// _kUpb_Status_MaxMessage-4.
|
||||
void SetErrorMessage(const char* msg) {
|
||||
upb_Status_SetErrorMessage(&status_, msg);
|
||||
}
|
||||
void SetFormattedErrorMessage(const char* fmt, ...) {
|
||||
va_list args;
|
||||
va_start(args, fmt);
|
||||
upb_Status_VSetErrorFormat(&status_, fmt, args);
|
||||
va_end(args);
|
||||
}
|
||||
|
||||
// Resets the status to a successful state with no message.
|
||||
void Clear() { upb_Status_Clear(&status_); }
|
||||
|
||||
private:
|
||||
upb_Status status_;
|
||||
};
|
||||
|
||||
} // namespace upb
|
||||
|
||||
#endif // UPB_BASE_STATUS_HPP_
|
||||
58
Pods/gRPC-Core/third_party/upb/upb/base/string_view.h
generated
vendored
Normal file
58
Pods/gRPC-Core/third_party/upb/upb/base/string_view.h
generated
vendored
Normal file
@@ -0,0 +1,58 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// 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
|
||||
#ifndef UPB_BASE_STRING_VIEW_H_
|
||||
#define UPB_BASE_STRING_VIEW_H_
|
||||
|
||||
#include <string.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#define UPB_STRINGVIEW_INIT(ptr, len) \
|
||||
{ ptr, len }
|
||||
|
||||
#define UPB_STRINGVIEW_FORMAT "%.*s"
|
||||
#define UPB_STRINGVIEW_ARGS(view) (int)(view).size, (view).data
|
||||
|
||||
// LINT.IfChange(struct_definition)
|
||||
typedef struct {
|
||||
const char* data;
|
||||
size_t size;
|
||||
} upb_StringView;
|
||||
// LINT.ThenChange(
|
||||
// GoogleInternalName0,
|
||||
// //depot/google3/third_party/upb/bits/golang/accessor.go:map_go_string
|
||||
// )
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_API_INLINE upb_StringView upb_StringView_FromDataAndSize(const char* data,
|
||||
size_t size) {
|
||||
upb_StringView ret;
|
||||
ret.data = data;
|
||||
ret.size = size;
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_INLINE upb_StringView upb_StringView_FromString(const char* data) {
|
||||
return upb_StringView_FromDataAndSize(data, strlen(data));
|
||||
}
|
||||
|
||||
UPB_INLINE bool upb_StringView_IsEqual(upb_StringView a, upb_StringView b) {
|
||||
return a.size == b.size && memcmp(a.data, b.data, a.size) == 0;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_BASE_STRING_VIEW_H_ */
|
||||
33
Pods/gRPC-Core/third_party/upb/upb/generated_code_support.h
generated
vendored
Normal file
33
Pods/gRPC-Core/third_party/upb/upb/generated_code_support.h
generated
vendored
Normal file
@@ -0,0 +1,33 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_GENERATED_CODE_SUPPORT_H_
|
||||
#define UPB_GENERATED_CODE_SUPPORT_H_
|
||||
|
||||
// IWYU pragma: begin_exports
|
||||
#include "upb/message/accessors.h"
|
||||
#include "upb/message/array.h"
|
||||
#include "upb/message/internal/accessors.h"
|
||||
#include "upb/message/internal/array.h"
|
||||
#include "upb/message/internal/extension.h"
|
||||
#include "upb/message/internal/message.h"
|
||||
#include "upb/message/map_gencode_util.h"
|
||||
#include "upb/message/message.h"
|
||||
#include "upb/mini_descriptor/decode.h"
|
||||
#include "upb/mini_table/enum.h"
|
||||
#include "upb/mini_table/extension.h"
|
||||
#include "upb/mini_table/extension_registry.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
#include "upb/mini_table/file.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
#include "upb/mini_table/sub.h"
|
||||
#include "upb/wire/decode.h"
|
||||
#include "upb/wire/decode_fast.h"
|
||||
#include "upb/wire/encode.h"
|
||||
// IWYU pragma: end_exports
|
||||
|
||||
#endif // UPB_GENERATED_CODE_SUPPORT_H_
|
||||
836
Pods/gRPC-Core/third_party/upb/upb/hash/common.c
generated
vendored
Normal file
836
Pods/gRPC-Core/third_party/upb/upb/hash/common.c
generated
vendored
Normal file
@@ -0,0 +1,836 @@
|
||||
// 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
|
||||
|
||||
/*
|
||||
* upb_table Implementation
|
||||
*
|
||||
* Implementation is heavily inspired by Lua's ltable.c.
|
||||
*/
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/base/internal/log2.h"
|
||||
#include "upb/hash/int_table.h"
|
||||
#include "upb/hash/str_table.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#define UPB_MAXARRSIZE 16 // 2**16 = 64k.
|
||||
|
||||
// From Chromium.
|
||||
#define ARRAY_SIZE(x) \
|
||||
((sizeof(x) / sizeof(0 [x])) / ((size_t)(!(sizeof(x) % sizeof(0 [x])))))
|
||||
|
||||
static const double MAX_LOAD = 0.85;
|
||||
|
||||
/* The minimum utilization of the array part of a mixed hash/array table. This
|
||||
* is a speed/memory-usage tradeoff (though it's not straightforward because of
|
||||
* cache effects). The lower this is, the more memory we'll use. */
|
||||
static const double MIN_DENSITY = 0.1;
|
||||
|
||||
static bool is_pow2(uint64_t v) { return v == 0 || (v & (v - 1)) == 0; }
|
||||
|
||||
static upb_value _upb_value_val(uint64_t val) {
|
||||
upb_value ret;
|
||||
_upb_value_setval(&ret, val);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int log2ceil(uint64_t v) {
|
||||
int ret = 0;
|
||||
bool pow2 = is_pow2(v);
|
||||
while (v >>= 1) ret++;
|
||||
ret = pow2 ? ret : ret + 1; // Ceiling.
|
||||
return UPB_MIN(UPB_MAXARRSIZE, ret);
|
||||
}
|
||||
|
||||
/* A type to represent the lookup key of either a strtable or an inttable. */
|
||||
typedef union {
|
||||
uintptr_t num;
|
||||
struct {
|
||||
const char* str;
|
||||
size_t len;
|
||||
} str;
|
||||
} lookupkey_t;
|
||||
|
||||
static lookupkey_t strkey2(const char* str, size_t len) {
|
||||
lookupkey_t k;
|
||||
k.str.str = str;
|
||||
k.str.len = len;
|
||||
return k;
|
||||
}
|
||||
|
||||
static lookupkey_t intkey(uintptr_t key) {
|
||||
lookupkey_t k;
|
||||
k.num = key;
|
||||
return k;
|
||||
}
|
||||
|
||||
typedef uint32_t hashfunc_t(upb_tabkey key);
|
||||
typedef bool eqlfunc_t(upb_tabkey k1, lookupkey_t k2);
|
||||
|
||||
/* Base table (shared code) ***************************************************/
|
||||
|
||||
static uint32_t upb_inthash(uintptr_t key) { return (uint32_t)key; }
|
||||
|
||||
static const upb_tabent* upb_getentry(const upb_table* t, uint32_t hash) {
|
||||
return t->entries + (hash & t->mask);
|
||||
}
|
||||
|
||||
static bool upb_arrhas(upb_tabval key) { return key.val != (uint64_t)-1; }
|
||||
|
||||
static bool isfull(upb_table* t) { return t->count == t->max_count; }
|
||||
|
||||
static bool init(upb_table* t, uint8_t size_lg2, upb_Arena* a) {
|
||||
size_t bytes;
|
||||
|
||||
t->count = 0;
|
||||
t->size_lg2 = size_lg2;
|
||||
t->mask = upb_table_size(t) ? upb_table_size(t) - 1 : 0;
|
||||
t->max_count = upb_table_size(t) * MAX_LOAD;
|
||||
bytes = upb_table_size(t) * sizeof(upb_tabent);
|
||||
if (bytes > 0) {
|
||||
t->entries = upb_Arena_Malloc(a, bytes);
|
||||
if (!t->entries) return false;
|
||||
memset(t->entries, 0, bytes);
|
||||
} else {
|
||||
t->entries = NULL;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static upb_tabent* emptyent(upb_table* t, upb_tabent* e) {
|
||||
upb_tabent* begin = t->entries;
|
||||
upb_tabent* end = begin + upb_table_size(t);
|
||||
for (e = e + 1; e < end; e++) {
|
||||
if (upb_tabent_isempty(e)) return e;
|
||||
}
|
||||
for (e = begin; e < end; e++) {
|
||||
if (upb_tabent_isempty(e)) return e;
|
||||
}
|
||||
UPB_ASSERT(false);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static upb_tabent* getentry_mutable(upb_table* t, uint32_t hash) {
|
||||
return (upb_tabent*)upb_getentry(t, hash);
|
||||
}
|
||||
|
||||
static const upb_tabent* findentry(const upb_table* t, lookupkey_t key,
|
||||
uint32_t hash, eqlfunc_t* eql) {
|
||||
const upb_tabent* e;
|
||||
|
||||
if (t->size_lg2 == 0) return NULL;
|
||||
e = upb_getentry(t, hash);
|
||||
if (upb_tabent_isempty(e)) return NULL;
|
||||
while (1) {
|
||||
if (eql(e->key, key)) return e;
|
||||
if ((e = e->next) == NULL) return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static upb_tabent* findentry_mutable(upb_table* t, lookupkey_t key,
|
||||
uint32_t hash, eqlfunc_t* eql) {
|
||||
return (upb_tabent*)findentry(t, key, hash, eql);
|
||||
}
|
||||
|
||||
static bool lookup(const upb_table* t, lookupkey_t key, upb_value* v,
|
||||
uint32_t hash, eqlfunc_t* eql) {
|
||||
const upb_tabent* e = findentry(t, key, hash, eql);
|
||||
if (e) {
|
||||
if (v) {
|
||||
_upb_value_setval(v, e->val.val);
|
||||
}
|
||||
return true;
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/* The given key must not already exist in the table. */
|
||||
static void insert(upb_table* t, lookupkey_t key, upb_tabkey tabkey,
|
||||
upb_value val, uint32_t hash, hashfunc_t* hashfunc,
|
||||
eqlfunc_t* eql) {
|
||||
upb_tabent* mainpos_e;
|
||||
upb_tabent* our_e;
|
||||
|
||||
UPB_ASSERT(findentry(t, key, hash, eql) == NULL);
|
||||
|
||||
t->count++;
|
||||
mainpos_e = getentry_mutable(t, hash);
|
||||
our_e = mainpos_e;
|
||||
|
||||
if (upb_tabent_isempty(mainpos_e)) {
|
||||
/* Our main position is empty; use it. */
|
||||
our_e->next = NULL;
|
||||
} else {
|
||||
/* Collision. */
|
||||
upb_tabent* new_e = emptyent(t, mainpos_e);
|
||||
/* Head of collider's chain. */
|
||||
upb_tabent* chain = getentry_mutable(t, hashfunc(mainpos_e->key));
|
||||
if (chain == mainpos_e) {
|
||||
/* Existing ent is in its main position (it has the same hash as us, and
|
||||
* is the head of our chain). Insert to new ent and append to this chain.
|
||||
*/
|
||||
new_e->next = mainpos_e->next;
|
||||
mainpos_e->next = new_e;
|
||||
our_e = new_e;
|
||||
} else {
|
||||
/* Existing ent is not in its main position (it is a node in some other
|
||||
* chain). This implies that no existing ent in the table has our hash.
|
||||
* Evict it (updating its chain) and use its ent for head of our chain. */
|
||||
*new_e = *mainpos_e; /* copies next. */
|
||||
while (chain->next != mainpos_e) {
|
||||
chain = (upb_tabent*)chain->next;
|
||||
UPB_ASSERT(chain);
|
||||
}
|
||||
chain->next = new_e;
|
||||
our_e = mainpos_e;
|
||||
our_e->next = NULL;
|
||||
}
|
||||
}
|
||||
our_e->key = tabkey;
|
||||
our_e->val.val = val.val;
|
||||
UPB_ASSERT(findentry(t, key, hash, eql) == our_e);
|
||||
}
|
||||
|
||||
static bool rm(upb_table* t, lookupkey_t key, upb_value* val,
|
||||
upb_tabkey* removed, uint32_t hash, eqlfunc_t* eql) {
|
||||
upb_tabent* chain = getentry_mutable(t, hash);
|
||||
if (upb_tabent_isempty(chain)) return false;
|
||||
if (eql(chain->key, key)) {
|
||||
/* Element to remove is at the head of its chain. */
|
||||
t->count--;
|
||||
if (val) _upb_value_setval(val, chain->val.val);
|
||||
if (removed) *removed = chain->key;
|
||||
if (chain->next) {
|
||||
upb_tabent* move = (upb_tabent*)chain->next;
|
||||
*chain = *move;
|
||||
move->key = 0; /* Make the slot empty. */
|
||||
} else {
|
||||
chain->key = 0; /* Make the slot empty. */
|
||||
}
|
||||
return true;
|
||||
} else {
|
||||
/* Element to remove is either in a non-head position or not in the
|
||||
* table. */
|
||||
while (chain->next && !eql(chain->next->key, key)) {
|
||||
chain = (upb_tabent*)chain->next;
|
||||
}
|
||||
if (chain->next) {
|
||||
/* Found element to remove. */
|
||||
upb_tabent* rm = (upb_tabent*)chain->next;
|
||||
t->count--;
|
||||
if (val) _upb_value_setval(val, chain->next->val.val);
|
||||
if (removed) *removed = rm->key;
|
||||
rm->key = 0; /* Make the slot empty. */
|
||||
chain->next = rm->next;
|
||||
return true;
|
||||
} else {
|
||||
/* Element to remove is not in the table. */
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static size_t next(const upb_table* t, size_t i) {
|
||||
do {
|
||||
if (++i >= upb_table_size(t)) return SIZE_MAX - 1; /* Distinct from -1. */
|
||||
} while (upb_tabent_isempty(&t->entries[i]));
|
||||
|
||||
return i;
|
||||
}
|
||||
|
||||
static size_t begin(const upb_table* t) { return next(t, -1); }
|
||||
|
||||
/* upb_strtable ***************************************************************/
|
||||
|
||||
/* A simple "subclass" of upb_table that only adds a hash function for strings.
|
||||
*/
|
||||
|
||||
static upb_tabkey strcopy(lookupkey_t k2, upb_Arena* a) {
|
||||
uint32_t len = (uint32_t)k2.str.len;
|
||||
char* str = upb_Arena_Malloc(a, k2.str.len + sizeof(uint32_t) + 1);
|
||||
if (str == NULL) return 0;
|
||||
memcpy(str, &len, sizeof(uint32_t));
|
||||
if (k2.str.len) memcpy(str + sizeof(uint32_t), k2.str.str, k2.str.len);
|
||||
str[sizeof(uint32_t) + k2.str.len] = '\0';
|
||||
return (uintptr_t)str;
|
||||
}
|
||||
|
||||
/* Adapted from ABSL's wyhash. */
|
||||
|
||||
static uint64_t UnalignedLoad64(const void* p) {
|
||||
uint64_t val;
|
||||
memcpy(&val, p, 8);
|
||||
return val;
|
||||
}
|
||||
|
||||
static uint32_t UnalignedLoad32(const void* p) {
|
||||
uint32_t val;
|
||||
memcpy(&val, p, 4);
|
||||
return val;
|
||||
}
|
||||
|
||||
#if defined(_MSC_VER) && defined(_M_X64)
|
||||
#include <intrin.h>
|
||||
#endif
|
||||
|
||||
/* Computes a * b, returning the low 64 bits of the result and storing the high
|
||||
* 64 bits in |*high|. */
|
||||
static uint64_t upb_umul128(uint64_t v0, uint64_t v1, uint64_t* out_high) {
|
||||
#ifdef __SIZEOF_INT128__
|
||||
__uint128_t p = v0;
|
||||
p *= v1;
|
||||
*out_high = (uint64_t)(p >> 64);
|
||||
return (uint64_t)p;
|
||||
#elif defined(_MSC_VER) && defined(_M_X64)
|
||||
return _umul128(v0, v1, out_high);
|
||||
#else
|
||||
uint64_t a32 = v0 >> 32;
|
||||
uint64_t a00 = v0 & 0xffffffff;
|
||||
uint64_t b32 = v1 >> 32;
|
||||
uint64_t b00 = v1 & 0xffffffff;
|
||||
uint64_t high = a32 * b32;
|
||||
uint64_t low = a00 * b00;
|
||||
uint64_t mid1 = a32 * b00;
|
||||
uint64_t mid2 = a00 * b32;
|
||||
low += (mid1 << 32) + (mid2 << 32);
|
||||
// Omit carry bit, for mixing we do not care about exact numerical precision.
|
||||
high += (mid1 >> 32) + (mid2 >> 32);
|
||||
*out_high = high;
|
||||
return low;
|
||||
#endif
|
||||
}
|
||||
|
||||
static uint64_t WyhashMix(uint64_t v0, uint64_t v1) {
|
||||
uint64_t high;
|
||||
uint64_t low = upb_umul128(v0, v1, &high);
|
||||
return low ^ high;
|
||||
}
|
||||
|
||||
static uint64_t Wyhash(const void* data, size_t len, uint64_t seed,
|
||||
const uint64_t salt[]) {
|
||||
const uint8_t* ptr = (const uint8_t*)data;
|
||||
uint64_t starting_length = (uint64_t)len;
|
||||
uint64_t current_state = seed ^ salt[0];
|
||||
|
||||
if (len > 64) {
|
||||
// If we have more than 64 bytes, we're going to handle chunks of 64
|
||||
// bytes at a time. We're going to build up two separate hash states
|
||||
// which we will then hash together.
|
||||
uint64_t duplicated_state = current_state;
|
||||
|
||||
do {
|
||||
uint64_t a = UnalignedLoad64(ptr);
|
||||
uint64_t b = UnalignedLoad64(ptr + 8);
|
||||
uint64_t c = UnalignedLoad64(ptr + 16);
|
||||
uint64_t d = UnalignedLoad64(ptr + 24);
|
||||
uint64_t e = UnalignedLoad64(ptr + 32);
|
||||
uint64_t f = UnalignedLoad64(ptr + 40);
|
||||
uint64_t g = UnalignedLoad64(ptr + 48);
|
||||
uint64_t h = UnalignedLoad64(ptr + 56);
|
||||
|
||||
uint64_t cs0 = WyhashMix(a ^ salt[1], b ^ current_state);
|
||||
uint64_t cs1 = WyhashMix(c ^ salt[2], d ^ current_state);
|
||||
current_state = (cs0 ^ cs1);
|
||||
|
||||
uint64_t ds0 = WyhashMix(e ^ salt[3], f ^ duplicated_state);
|
||||
uint64_t ds1 = WyhashMix(g ^ salt[4], h ^ duplicated_state);
|
||||
duplicated_state = (ds0 ^ ds1);
|
||||
|
||||
ptr += 64;
|
||||
len -= 64;
|
||||
} while (len > 64);
|
||||
|
||||
current_state = current_state ^ duplicated_state;
|
||||
}
|
||||
|
||||
// We now have a data `ptr` with at most 64 bytes and the current state
|
||||
// of the hashing state machine stored in current_state.
|
||||
while (len > 16) {
|
||||
uint64_t a = UnalignedLoad64(ptr);
|
||||
uint64_t b = UnalignedLoad64(ptr + 8);
|
||||
|
||||
current_state = WyhashMix(a ^ salt[1], b ^ current_state);
|
||||
|
||||
ptr += 16;
|
||||
len -= 16;
|
||||
}
|
||||
|
||||
// We now have a data `ptr` with at most 16 bytes.
|
||||
uint64_t a = 0;
|
||||
uint64_t b = 0;
|
||||
if (len > 8) {
|
||||
// When we have at least 9 and at most 16 bytes, set A to the first 64
|
||||
// bits of the input and B to the last 64 bits of the input. Yes, they will
|
||||
// overlap in the middle if we are working with less than the full 16
|
||||
// bytes.
|
||||
a = UnalignedLoad64(ptr);
|
||||
b = UnalignedLoad64(ptr + len - 8);
|
||||
} else if (len > 3) {
|
||||
// If we have at least 4 and at most 8 bytes, set A to the first 32
|
||||
// bits and B to the last 32 bits.
|
||||
a = UnalignedLoad32(ptr);
|
||||
b = UnalignedLoad32(ptr + len - 4);
|
||||
} else if (len > 0) {
|
||||
// If we have at least 1 and at most 3 bytes, read all of the provided
|
||||
// bits into A, with some adjustments.
|
||||
a = ((ptr[0] << 16) | (ptr[len >> 1] << 8) | ptr[len - 1]);
|
||||
b = 0;
|
||||
} else {
|
||||
a = 0;
|
||||
b = 0;
|
||||
}
|
||||
|
||||
uint64_t w = WyhashMix(a ^ salt[1], b ^ current_state);
|
||||
uint64_t z = salt[1] ^ starting_length;
|
||||
return WyhashMix(w, z);
|
||||
}
|
||||
|
||||
const uint64_t kWyhashSalt[5] = {
|
||||
0x243F6A8885A308D3ULL, 0x13198A2E03707344ULL, 0xA4093822299F31D0ULL,
|
||||
0x082EFA98EC4E6C89ULL, 0x452821E638D01377ULL,
|
||||
};
|
||||
|
||||
uint32_t _upb_Hash(const void* p, size_t n, uint64_t seed) {
|
||||
return Wyhash(p, n, seed, kWyhashSalt);
|
||||
}
|
||||
|
||||
static uint32_t _upb_Hash_NoSeed(const char* p, size_t n) {
|
||||
return _upb_Hash(p, n, 0);
|
||||
}
|
||||
|
||||
static uint32_t strhash(upb_tabkey key) {
|
||||
uint32_t len;
|
||||
char* str = upb_tabstr(key, &len);
|
||||
return _upb_Hash_NoSeed(str, len);
|
||||
}
|
||||
|
||||
static bool streql(upb_tabkey k1, lookupkey_t k2) {
|
||||
uint32_t len;
|
||||
char* str = upb_tabstr(k1, &len);
|
||||
return len == k2.str.len && (len == 0 || memcmp(str, k2.str.str, len) == 0);
|
||||
}
|
||||
|
||||
bool upb_strtable_init(upb_strtable* t, size_t expected_size, upb_Arena* a) {
|
||||
// Multiply by approximate reciprocal of MAX_LOAD (0.85), with pow2
|
||||
// denominator.
|
||||
size_t need_entries = (expected_size + 1) * 1204 / 1024;
|
||||
UPB_ASSERT(need_entries >= expected_size * 0.85);
|
||||
int size_lg2 = upb_Log2Ceiling(need_entries);
|
||||
return init(&t->t, size_lg2, a);
|
||||
}
|
||||
|
||||
void upb_strtable_clear(upb_strtable* t) {
|
||||
size_t bytes = upb_table_size(&t->t) * sizeof(upb_tabent);
|
||||
t->t.count = 0;
|
||||
memset((char*)t->t.entries, 0, bytes);
|
||||
}
|
||||
|
||||
bool upb_strtable_resize(upb_strtable* t, size_t size_lg2, upb_Arena* a) {
|
||||
upb_strtable new_table;
|
||||
if (!init(&new_table.t, size_lg2, a)) return false;
|
||||
|
||||
intptr_t iter = UPB_STRTABLE_BEGIN;
|
||||
upb_StringView key;
|
||||
upb_value val;
|
||||
while (upb_strtable_next2(t, &key, &val, &iter)) {
|
||||
upb_strtable_insert(&new_table, key.data, key.size, val, a);
|
||||
}
|
||||
*t = new_table;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_strtable_insert(upb_strtable* t, const char* k, size_t len,
|
||||
upb_value v, upb_Arena* a) {
|
||||
lookupkey_t key;
|
||||
upb_tabkey tabkey;
|
||||
uint32_t hash;
|
||||
|
||||
if (isfull(&t->t)) {
|
||||
/* Need to resize. New table of double the size, add old elements to it. */
|
||||
if (!upb_strtable_resize(t, t->t.size_lg2 + 1, a)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
key = strkey2(k, len);
|
||||
tabkey = strcopy(key, a);
|
||||
if (tabkey == 0) return false;
|
||||
|
||||
hash = _upb_Hash_NoSeed(key.str.str, key.str.len);
|
||||
insert(&t->t, key, tabkey, v, hash, &strhash, &streql);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_strtable_lookup2(const upb_strtable* t, const char* key, size_t len,
|
||||
upb_value* v) {
|
||||
uint32_t hash = _upb_Hash_NoSeed(key, len);
|
||||
return lookup(&t->t, strkey2(key, len), v, hash, &streql);
|
||||
}
|
||||
|
||||
bool upb_strtable_remove2(upb_strtable* t, const char* key, size_t len,
|
||||
upb_value* val) {
|
||||
uint32_t hash = _upb_Hash_NoSeed(key, len);
|
||||
upb_tabkey tabkey;
|
||||
return rm(&t->t, strkey2(key, len), val, &tabkey, hash, &streql);
|
||||
}
|
||||
|
||||
/* Iteration */
|
||||
|
||||
void upb_strtable_begin(upb_strtable_iter* i, const upb_strtable* t) {
|
||||
i->t = t;
|
||||
i->index = begin(&t->t);
|
||||
}
|
||||
|
||||
void upb_strtable_next(upb_strtable_iter* i) {
|
||||
i->index = next(&i->t->t, i->index);
|
||||
}
|
||||
|
||||
bool upb_strtable_done(const upb_strtable_iter* i) {
|
||||
if (!i->t) return true;
|
||||
return i->index >= upb_table_size(&i->t->t) ||
|
||||
upb_tabent_isempty(str_tabent(i));
|
||||
}
|
||||
|
||||
upb_StringView upb_strtable_iter_key(const upb_strtable_iter* i) {
|
||||
upb_StringView key;
|
||||
uint32_t len;
|
||||
UPB_ASSERT(!upb_strtable_done(i));
|
||||
key.data = upb_tabstr(str_tabent(i)->key, &len);
|
||||
key.size = len;
|
||||
return key;
|
||||
}
|
||||
|
||||
upb_value upb_strtable_iter_value(const upb_strtable_iter* i) {
|
||||
UPB_ASSERT(!upb_strtable_done(i));
|
||||
return _upb_value_val(str_tabent(i)->val.val);
|
||||
}
|
||||
|
||||
void upb_strtable_iter_setdone(upb_strtable_iter* i) {
|
||||
i->t = NULL;
|
||||
i->index = SIZE_MAX;
|
||||
}
|
||||
|
||||
bool upb_strtable_iter_isequal(const upb_strtable_iter* i1,
|
||||
const upb_strtable_iter* i2) {
|
||||
if (upb_strtable_done(i1) && upb_strtable_done(i2)) return true;
|
||||
return i1->t == i2->t && i1->index == i2->index;
|
||||
}
|
||||
|
||||
/* upb_inttable ***************************************************************/
|
||||
|
||||
/* For inttables we use a hybrid structure where small keys are kept in an
|
||||
* array and large keys are put in the hash table. */
|
||||
|
||||
static uint32_t inthash(upb_tabkey key) { return upb_inthash(key); }
|
||||
|
||||
static bool inteql(upb_tabkey k1, lookupkey_t k2) { return k1 == k2.num; }
|
||||
|
||||
static upb_tabval* mutable_array(upb_inttable* t) {
|
||||
return (upb_tabval*)t->array;
|
||||
}
|
||||
|
||||
static upb_tabval* inttable_val(upb_inttable* t, uintptr_t key) {
|
||||
if (key < t->array_size) {
|
||||
return upb_arrhas(t->array[key]) ? &(mutable_array(t)[key]) : NULL;
|
||||
} else {
|
||||
upb_tabent* e =
|
||||
findentry_mutable(&t->t, intkey(key), upb_inthash(key), &inteql);
|
||||
return e ? &e->val : NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static const upb_tabval* inttable_val_const(const upb_inttable* t,
|
||||
uintptr_t key) {
|
||||
return inttable_val((upb_inttable*)t, key);
|
||||
}
|
||||
|
||||
size_t upb_inttable_count(const upb_inttable* t) {
|
||||
return t->t.count + t->array_count;
|
||||
}
|
||||
|
||||
static void check(upb_inttable* t) {
|
||||
UPB_UNUSED(t);
|
||||
#if defined(UPB_DEBUG_TABLE) && !defined(NDEBUG)
|
||||
{
|
||||
// This check is very expensive (makes inserts/deletes O(N)).
|
||||
size_t count = 0;
|
||||
intptr_t iter = UPB_INTTABLE_BEGIN;
|
||||
uintptr_t key;
|
||||
upb_value val;
|
||||
while (upb_inttable_next(t, &key, &val, &iter)) {
|
||||
UPB_ASSERT(upb_inttable_lookup(t, key, NULL));
|
||||
}
|
||||
UPB_ASSERT(count == upb_inttable_count(t));
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
bool upb_inttable_sizedinit(upb_inttable* t, size_t asize, int hsize_lg2,
|
||||
upb_Arena* a) {
|
||||
size_t array_bytes;
|
||||
|
||||
if (!init(&t->t, hsize_lg2, a)) return false;
|
||||
/* Always make the array part at least 1 long, so that we know key 0
|
||||
* won't be in the hash part, which simplifies things. */
|
||||
t->array_size = UPB_MAX(1, asize);
|
||||
t->array_count = 0;
|
||||
array_bytes = t->array_size * sizeof(upb_value);
|
||||
t->array = upb_Arena_Malloc(a, array_bytes);
|
||||
if (!t->array) {
|
||||
return false;
|
||||
}
|
||||
memset(mutable_array(t), 0xff, array_bytes);
|
||||
check(t);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_inttable_init(upb_inttable* t, upb_Arena* a) {
|
||||
return upb_inttable_sizedinit(t, 0, 4, a);
|
||||
}
|
||||
|
||||
bool upb_inttable_insert(upb_inttable* t, uintptr_t key, upb_value val,
|
||||
upb_Arena* a) {
|
||||
upb_tabval tabval;
|
||||
tabval.val = val.val;
|
||||
UPB_ASSERT(
|
||||
upb_arrhas(tabval)); /* This will reject (uint64_t)-1. Fix this. */
|
||||
|
||||
if (key < t->array_size) {
|
||||
UPB_ASSERT(!upb_arrhas(t->array[key]));
|
||||
t->array_count++;
|
||||
mutable_array(t)[key].val = val.val;
|
||||
} else {
|
||||
if (isfull(&t->t)) {
|
||||
/* Need to resize the hash part, but we re-use the array part. */
|
||||
size_t i;
|
||||
upb_table new_table;
|
||||
|
||||
if (!init(&new_table, t->t.size_lg2 + 1, a)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
for (i = begin(&t->t); i < upb_table_size(&t->t); i = next(&t->t, i)) {
|
||||
const upb_tabent* e = &t->t.entries[i];
|
||||
uint32_t hash;
|
||||
upb_value v;
|
||||
|
||||
_upb_value_setval(&v, e->val.val);
|
||||
hash = upb_inthash(e->key);
|
||||
insert(&new_table, intkey(e->key), e->key, v, hash, &inthash, &inteql);
|
||||
}
|
||||
|
||||
UPB_ASSERT(t->t.count == new_table.count);
|
||||
|
||||
t->t = new_table;
|
||||
}
|
||||
insert(&t->t, intkey(key), key, val, upb_inthash(key), &inthash, &inteql);
|
||||
}
|
||||
check(t);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_inttable_lookup(const upb_inttable* t, uintptr_t key, upb_value* v) {
|
||||
const upb_tabval* table_v = inttable_val_const(t, key);
|
||||
if (!table_v) return false;
|
||||
if (v) _upb_value_setval(v, table_v->val);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_inttable_replace(upb_inttable* t, uintptr_t key, upb_value val) {
|
||||
upb_tabval* table_v = inttable_val(t, key);
|
||||
if (!table_v) return false;
|
||||
table_v->val = val.val;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_inttable_remove(upb_inttable* t, uintptr_t key, upb_value* val) {
|
||||
bool success;
|
||||
if (key < t->array_size) {
|
||||
if (upb_arrhas(t->array[key])) {
|
||||
upb_tabval empty = UPB_TABVALUE_EMPTY_INIT;
|
||||
t->array_count--;
|
||||
if (val) {
|
||||
_upb_value_setval(val, t->array[key].val);
|
||||
}
|
||||
mutable_array(t)[key] = empty;
|
||||
success = true;
|
||||
} else {
|
||||
success = false;
|
||||
}
|
||||
} else {
|
||||
success = rm(&t->t, intkey(key), val, NULL, upb_inthash(key), &inteql);
|
||||
}
|
||||
check(t);
|
||||
return success;
|
||||
}
|
||||
|
||||
void upb_inttable_compact(upb_inttable* t, upb_Arena* a) {
|
||||
/* A power-of-two histogram of the table keys. */
|
||||
size_t counts[UPB_MAXARRSIZE + 1] = {0};
|
||||
|
||||
/* The max key in each bucket. */
|
||||
uintptr_t max[UPB_MAXARRSIZE + 1] = {0};
|
||||
|
||||
{
|
||||
intptr_t iter = UPB_INTTABLE_BEGIN;
|
||||
uintptr_t key;
|
||||
upb_value val;
|
||||
while (upb_inttable_next(t, &key, &val, &iter)) {
|
||||
int bucket = log2ceil(key);
|
||||
max[bucket] = UPB_MAX(max[bucket], key);
|
||||
counts[bucket]++;
|
||||
}
|
||||
}
|
||||
|
||||
/* Find the largest power of two that satisfies the MIN_DENSITY
|
||||
* definition (while actually having some keys). */
|
||||
size_t arr_count = upb_inttable_count(t);
|
||||
int size_lg2;
|
||||
upb_inttable new_t;
|
||||
|
||||
for (size_lg2 = ARRAY_SIZE(counts) - 1; size_lg2 > 0; size_lg2--) {
|
||||
if (counts[size_lg2] == 0) {
|
||||
/* We can halve again without losing any entries. */
|
||||
continue;
|
||||
} else if (arr_count >= (1 << size_lg2) * MIN_DENSITY) {
|
||||
break;
|
||||
}
|
||||
|
||||
arr_count -= counts[size_lg2];
|
||||
}
|
||||
|
||||
UPB_ASSERT(arr_count <= upb_inttable_count(t));
|
||||
|
||||
{
|
||||
/* Insert all elements into new, perfectly-sized table. */
|
||||
size_t arr_size = max[size_lg2] + 1; /* +1 so arr[max] will fit. */
|
||||
size_t hash_count = upb_inttable_count(t) - arr_count;
|
||||
size_t hash_size = hash_count ? (hash_count / MAX_LOAD) + 1 : 0;
|
||||
int hashsize_lg2 = log2ceil(hash_size);
|
||||
|
||||
upb_inttable_sizedinit(&new_t, arr_size, hashsize_lg2, a);
|
||||
|
||||
{
|
||||
intptr_t iter = UPB_INTTABLE_BEGIN;
|
||||
uintptr_t key;
|
||||
upb_value val;
|
||||
while (upb_inttable_next(t, &key, &val, &iter)) {
|
||||
upb_inttable_insert(&new_t, key, val, a);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_ASSERT(new_t.array_size == arr_size);
|
||||
UPB_ASSERT(new_t.t.size_lg2 == hashsize_lg2);
|
||||
}
|
||||
*t = new_t;
|
||||
}
|
||||
|
||||
// Iteration.
|
||||
|
||||
bool upb_inttable_next(const upb_inttable* t, uintptr_t* key, upb_value* val,
|
||||
intptr_t* iter) {
|
||||
intptr_t i = *iter;
|
||||
if ((size_t)(i + 1) <= t->array_size) {
|
||||
while ((size_t)++i < t->array_size) {
|
||||
upb_tabval ent = t->array[i];
|
||||
if (upb_arrhas(ent)) {
|
||||
*key = i;
|
||||
*val = _upb_value_val(ent.val);
|
||||
*iter = i;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
i--; // Back up to exactly one position before the start of the table.
|
||||
}
|
||||
|
||||
size_t tab_idx = next(&t->t, i - t->array_size);
|
||||
if (tab_idx < upb_table_size(&t->t)) {
|
||||
upb_tabent* ent = &t->t.entries[tab_idx];
|
||||
*key = ent->key;
|
||||
*val = _upb_value_val(ent->val.val);
|
||||
*iter = tab_idx + t->array_size;
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
void upb_inttable_removeiter(upb_inttable* t, intptr_t* iter) {
|
||||
intptr_t i = *iter;
|
||||
if ((size_t)i < t->array_size) {
|
||||
t->array_count--;
|
||||
mutable_array(t)[i].val = -1;
|
||||
} else {
|
||||
upb_tabent* ent = &t->t.entries[i - t->array_size];
|
||||
upb_tabent* prev = NULL;
|
||||
|
||||
// Linear search, not great.
|
||||
upb_tabent* end = &t->t.entries[upb_table_size(&t->t)];
|
||||
for (upb_tabent* e = t->t.entries; e != end; e++) {
|
||||
if (e->next == ent) {
|
||||
prev = e;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (prev) {
|
||||
prev->next = ent->next;
|
||||
}
|
||||
|
||||
t->t.count--;
|
||||
ent->key = 0;
|
||||
ent->next = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
bool upb_strtable_next2(const upb_strtable* t, upb_StringView* key,
|
||||
upb_value* val, intptr_t* iter) {
|
||||
size_t tab_idx = next(&t->t, *iter);
|
||||
if (tab_idx < upb_table_size(&t->t)) {
|
||||
upb_tabent* ent = &t->t.entries[tab_idx];
|
||||
uint32_t len;
|
||||
key->data = upb_tabstr(ent->key, &len);
|
||||
key->size = len;
|
||||
*val = _upb_value_val(ent->val.val);
|
||||
*iter = tab_idx;
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
void upb_strtable_removeiter(upb_strtable* t, intptr_t* iter) {
|
||||
intptr_t i = *iter;
|
||||
upb_tabent* ent = &t->t.entries[i];
|
||||
upb_tabent* prev = NULL;
|
||||
|
||||
// Linear search, not great.
|
||||
upb_tabent* end = &t->t.entries[upb_table_size(&t->t)];
|
||||
for (upb_tabent* e = t->t.entries; e != end; e++) {
|
||||
if (e->next == ent) {
|
||||
prev = e;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (prev) {
|
||||
prev->next = ent->next;
|
||||
}
|
||||
|
||||
t->t.count--;
|
||||
ent->key = 0;
|
||||
ent->next = NULL;
|
||||
}
|
||||
|
||||
void upb_strtable_setentryvalue(upb_strtable* t, intptr_t iter, upb_value v) {
|
||||
upb_tabent* ent = &t->t.entries[iter];
|
||||
ent->val.val = v.val;
|
||||
}
|
||||
174
Pods/gRPC-Core/third_party/upb/upb/hash/common.h
generated
vendored
Normal file
174
Pods/gRPC-Core/third_party/upb/upb/hash/common.h
generated
vendored
Normal file
@@ -0,0 +1,174 @@
|
||||
// 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
|
||||
|
||||
/*
|
||||
* upb_table
|
||||
*
|
||||
* This header is INTERNAL-ONLY! Its interfaces are not public or stable!
|
||||
* This file defines very fast int->upb_value (inttable) and string->upb_value
|
||||
* (strtable) hash tables.
|
||||
*
|
||||
* The table uses chained scatter with Brent's variation (inspired by the Lua
|
||||
* implementation of hash tables). The hash function for strings is Austin
|
||||
* Appleby's "MurmurHash."
|
||||
*
|
||||
* The inttable uses uintptr_t as its key, which guarantees it can be used to
|
||||
* store pointers or integers of at least 32 bits (upb isn't really useful on
|
||||
* systems where sizeof(void*) < 4).
|
||||
*
|
||||
* The table must be homogeneous (all values of the same type). In debug
|
||||
* mode, we check this on insert and lookup.
|
||||
*/
|
||||
|
||||
#ifndef UPB_HASH_COMMON_H_
|
||||
#define UPB_HASH_COMMON_H_
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/mem/arena.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
/* upb_value ******************************************************************/
|
||||
|
||||
typedef struct {
|
||||
uint64_t val;
|
||||
} upb_value;
|
||||
|
||||
UPB_INLINE void _upb_value_setval(upb_value* v, uint64_t val) { v->val = val; }
|
||||
|
||||
/* For each value ctype, define the following set of functions:
|
||||
*
|
||||
* // Get/set an int32 from a upb_value.
|
||||
* int32_t upb_value_getint32(upb_value val);
|
||||
* void upb_value_setint32(upb_value *val, int32_t cval);
|
||||
*
|
||||
* // Construct a new upb_value from an int32.
|
||||
* upb_value upb_value_int32(int32_t val); */
|
||||
#define FUNCS(name, membername, type_t, converter) \
|
||||
UPB_INLINE void upb_value_set##name(upb_value* val, type_t cval) { \
|
||||
val->val = (converter)cval; \
|
||||
} \
|
||||
UPB_INLINE upb_value upb_value_##name(type_t val) { \
|
||||
upb_value ret; \
|
||||
upb_value_set##name(&ret, val); \
|
||||
return ret; \
|
||||
} \
|
||||
UPB_INLINE type_t upb_value_get##name(upb_value val) { \
|
||||
return (type_t)(converter)val.val; \
|
||||
}
|
||||
|
||||
FUNCS(int32, int32, int32_t, int32_t)
|
||||
FUNCS(int64, int64, int64_t, int64_t)
|
||||
FUNCS(uint32, uint32, uint32_t, uint32_t)
|
||||
FUNCS(uint64, uint64, uint64_t, uint64_t)
|
||||
FUNCS(bool, _bool, bool, bool)
|
||||
FUNCS(cstr, cstr, char*, uintptr_t)
|
||||
FUNCS(uintptr, uptr, uintptr_t, uintptr_t)
|
||||
FUNCS(ptr, ptr, void*, uintptr_t)
|
||||
FUNCS(constptr, constptr, const void*, uintptr_t)
|
||||
|
||||
#undef FUNCS
|
||||
|
||||
UPB_INLINE void upb_value_setfloat(upb_value* val, float cval) {
|
||||
memcpy(&val->val, &cval, sizeof(cval));
|
||||
}
|
||||
|
||||
UPB_INLINE void upb_value_setdouble(upb_value* val, double cval) {
|
||||
memcpy(&val->val, &cval, sizeof(cval));
|
||||
}
|
||||
|
||||
UPB_INLINE upb_value upb_value_float(float cval) {
|
||||
upb_value ret;
|
||||
upb_value_setfloat(&ret, cval);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_INLINE upb_value upb_value_double(double cval) {
|
||||
upb_value ret;
|
||||
upb_value_setdouble(&ret, cval);
|
||||
return ret;
|
||||
}
|
||||
|
||||
/* upb_tabkey *****************************************************************/
|
||||
|
||||
/* Either:
|
||||
* 1. an actual integer key, or
|
||||
* 2. a pointer to a string prefixed by its uint32_t length, owned by us.
|
||||
*
|
||||
* ...depending on whether this is a string table or an int table. We would
|
||||
* make this a union of those two types, but C89 doesn't support statically
|
||||
* initializing a non-first union member. */
|
||||
typedef uintptr_t upb_tabkey;
|
||||
|
||||
UPB_INLINE char* upb_tabstr(upb_tabkey key, uint32_t* len) {
|
||||
char* mem = (char*)key;
|
||||
if (len) memcpy(len, mem, sizeof(*len));
|
||||
return mem + sizeof(*len);
|
||||
}
|
||||
|
||||
UPB_INLINE upb_StringView upb_tabstrview(upb_tabkey key) {
|
||||
upb_StringView ret;
|
||||
uint32_t len;
|
||||
ret.data = upb_tabstr(key, &len);
|
||||
ret.size = len;
|
||||
return ret;
|
||||
}
|
||||
|
||||
/* upb_tabval *****************************************************************/
|
||||
|
||||
typedef struct upb_tabval {
|
||||
uint64_t val;
|
||||
} upb_tabval;
|
||||
|
||||
#define UPB_TABVALUE_EMPTY_INIT \
|
||||
{ -1 }
|
||||
|
||||
/* upb_table ******************************************************************/
|
||||
|
||||
typedef struct _upb_tabent {
|
||||
upb_tabkey key;
|
||||
upb_tabval val;
|
||||
|
||||
/* Internal chaining. This is const so we can create static initializers for
|
||||
* tables. We cast away const sometimes, but *only* when the containing
|
||||
* upb_table is known to be non-const. This requires a bit of care, but
|
||||
* the subtlety is confined to table.c. */
|
||||
const struct _upb_tabent* next;
|
||||
} upb_tabent;
|
||||
|
||||
typedef struct {
|
||||
size_t count; /* Number of entries in the hash part. */
|
||||
uint32_t mask; /* Mask to turn hash value -> bucket. */
|
||||
uint32_t max_count; /* Max count before we hit our load limit. */
|
||||
uint8_t size_lg2; /* Size of the hashtable part is 2^size_lg2 entries. */
|
||||
upb_tabent* entries;
|
||||
} upb_table;
|
||||
|
||||
UPB_INLINE size_t upb_table_size(const upb_table* t) {
|
||||
return t->size_lg2 ? 1 << t->size_lg2 : 0;
|
||||
}
|
||||
|
||||
// Internal-only functions, in .h file only out of necessity.
|
||||
|
||||
UPB_INLINE bool upb_tabent_isempty(const upb_tabent* e) { return e->key == 0; }
|
||||
|
||||
uint32_t _upb_Hash(const void* p, size_t n, uint64_t seed);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_HASH_COMMON_H_ */
|
||||
82
Pods/gRPC-Core/third_party/upb/upb/hash/int_table.h
generated
vendored
Normal file
82
Pods/gRPC-Core/third_party/upb/upb/hash/int_table.h
generated
vendored
Normal file
@@ -0,0 +1,82 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_HASH_INT_TABLE_H_
|
||||
#define UPB_HASH_INT_TABLE_H_
|
||||
|
||||
#include "upb/hash/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct {
|
||||
upb_table t; // For entries that don't fit in the array part.
|
||||
const upb_tabval* array; // Array part of the table. See const note above.
|
||||
size_t array_size; // Array part size.
|
||||
size_t array_count; // Array part number of elements.
|
||||
} upb_inttable;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Initialize a table. If memory allocation failed, false is returned and
|
||||
// the table is uninitialized.
|
||||
bool upb_inttable_init(upb_inttable* table, upb_Arena* a);
|
||||
|
||||
// Returns the number of values in the table.
|
||||
size_t upb_inttable_count(const upb_inttable* t);
|
||||
|
||||
// Inserts the given key into the hashtable with the given value.
|
||||
// The key must not already exist in the hash table.
|
||||
// The value must not be UINTPTR_MAX.
|
||||
//
|
||||
// If a table resize was required but memory allocation failed, false is
|
||||
// returned and the table is unchanged.
|
||||
bool upb_inttable_insert(upb_inttable* t, uintptr_t key, upb_value val,
|
||||
upb_Arena* a);
|
||||
|
||||
// Looks up key in this table, returning "true" if the key was found.
|
||||
// If v is non-NULL, copies the value for this key into *v.
|
||||
bool upb_inttable_lookup(const upb_inttable* t, uintptr_t key, upb_value* v);
|
||||
|
||||
// Removes an item from the table. Returns true if the remove was successful,
|
||||
// and stores the removed item in *val if non-NULL.
|
||||
bool upb_inttable_remove(upb_inttable* t, uintptr_t key, upb_value* val);
|
||||
|
||||
// Updates an existing entry in an inttable.
|
||||
// If the entry does not exist, returns false and does nothing.
|
||||
// Unlike insert/remove, this does not invalidate iterators.
|
||||
bool upb_inttable_replace(upb_inttable* t, uintptr_t key, upb_value val);
|
||||
|
||||
// Optimizes the table for the current set of entries, for both memory use and
|
||||
// lookup time. Client should call this after all entries have been inserted;
|
||||
// inserting more entries is legal, but will likely require a table resize.
|
||||
void upb_inttable_compact(upb_inttable* t, upb_Arena* a);
|
||||
|
||||
// Iteration over inttable:
|
||||
//
|
||||
// intptr_t iter = UPB_INTTABLE_BEGIN;
|
||||
// uintptr_t key;
|
||||
// upb_value val;
|
||||
// while (upb_inttable_next(t, &key, &val, &iter)) {
|
||||
// // ...
|
||||
// }
|
||||
|
||||
#define UPB_INTTABLE_BEGIN -1
|
||||
|
||||
bool upb_inttable_next(const upb_inttable* t, uintptr_t* key, upb_value* val,
|
||||
intptr_t* iter);
|
||||
void upb_inttable_removeiter(upb_inttable* t, intptr_t* iter);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_HASH_INT_TABLE_H_ */
|
||||
142
Pods/gRPC-Core/third_party/upb/upb/hash/str_table.h
generated
vendored
Normal file
142
Pods/gRPC-Core/third_party/upb/upb/hash/str_table.h
generated
vendored
Normal file
@@ -0,0 +1,142 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_HASH_STR_TABLE_H_
|
||||
#define UPB_HASH_STR_TABLE_H_
|
||||
|
||||
#include "upb/hash/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct {
|
||||
upb_table t;
|
||||
} upb_strtable;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Initialize a table. If memory allocation failed, false is returned and
|
||||
// the table is uninitialized.
|
||||
bool upb_strtable_init(upb_strtable* table, size_t expected_size, upb_Arena* a);
|
||||
|
||||
// Returns the number of values in the table.
|
||||
UPB_INLINE size_t upb_strtable_count(const upb_strtable* t) {
|
||||
return t->t.count;
|
||||
}
|
||||
|
||||
void upb_strtable_clear(upb_strtable* t);
|
||||
|
||||
// Inserts the given key into the hashtable with the given value.
|
||||
// The key must not already exist in the hash table. The key is not required
|
||||
// to be NULL-terminated, and the table will make an internal copy of the key.
|
||||
//
|
||||
// If a table resize was required but memory allocation failed, false is
|
||||
// returned and the table is unchanged. */
|
||||
bool upb_strtable_insert(upb_strtable* t, const char* key, size_t len,
|
||||
upb_value val, upb_Arena* a);
|
||||
|
||||
// Looks up key in this table, returning "true" if the key was found.
|
||||
// If v is non-NULL, copies the value for this key into *v.
|
||||
bool upb_strtable_lookup2(const upb_strtable* t, const char* key, size_t len,
|
||||
upb_value* v);
|
||||
|
||||
// For NULL-terminated strings.
|
||||
UPB_INLINE bool upb_strtable_lookup(const upb_strtable* t, const char* key,
|
||||
upb_value* v) {
|
||||
return upb_strtable_lookup2(t, key, strlen(key), v);
|
||||
}
|
||||
|
||||
// Removes an item from the table. Returns true if the remove was successful,
|
||||
// and stores the removed item in *val if non-NULL.
|
||||
bool upb_strtable_remove2(upb_strtable* t, const char* key, size_t len,
|
||||
upb_value* val);
|
||||
|
||||
UPB_INLINE bool upb_strtable_remove(upb_strtable* t, const char* key,
|
||||
upb_value* v) {
|
||||
return upb_strtable_remove2(t, key, strlen(key), v);
|
||||
}
|
||||
|
||||
// Exposed for testing only.
|
||||
bool upb_strtable_resize(upb_strtable* t, size_t size_lg2, upb_Arena* a);
|
||||
|
||||
/* Iteration over strtable:
|
||||
*
|
||||
* intptr_t iter = UPB_STRTABLE_BEGIN;
|
||||
* upb_StringView key;
|
||||
* upb_value val;
|
||||
* while (upb_strtable_next2(t, &key, &val, &iter)) {
|
||||
* // ...
|
||||
* }
|
||||
*/
|
||||
|
||||
#define UPB_STRTABLE_BEGIN -1
|
||||
|
||||
bool upb_strtable_next2(const upb_strtable* t, upb_StringView* key,
|
||||
upb_value* val, intptr_t* iter);
|
||||
void upb_strtable_removeiter(upb_strtable* t, intptr_t* iter);
|
||||
void upb_strtable_setentryvalue(upb_strtable* t, intptr_t iter, upb_value v);
|
||||
|
||||
/* DEPRECATED iterators, slated for removal.
|
||||
*
|
||||
* Iterators for string tables. We are subject to some kind of unusual
|
||||
* design constraints:
|
||||
*
|
||||
* For high-level languages:
|
||||
* - we must be able to guarantee that we don't crash or corrupt memory even if
|
||||
* the program accesses an invalidated iterator.
|
||||
*
|
||||
* For C++11 range-based for:
|
||||
* - iterators must be copyable
|
||||
* - iterators must be comparable
|
||||
* - it must be possible to construct an "end" value.
|
||||
*
|
||||
* Iteration order is undefined.
|
||||
*
|
||||
* Modifying the table invalidates iterators. upb_{str,int}table_done() is
|
||||
* guaranteed to work even on an invalidated iterator, as long as the table it
|
||||
* is iterating over has not been freed. Calling next() or accessing data from
|
||||
* an invalidated iterator yields unspecified elements from the table, but it is
|
||||
* guaranteed not to crash and to return real table elements (except when done()
|
||||
* is true). */
|
||||
/* upb_strtable_iter **********************************************************/
|
||||
|
||||
/* upb_strtable_iter i;
|
||||
* upb_strtable_begin(&i, t);
|
||||
* for(; !upb_strtable_done(&i); upb_strtable_next(&i)) {
|
||||
* const char *key = upb_strtable_iter_key(&i);
|
||||
* const upb_value val = upb_strtable_iter_value(&i);
|
||||
* // ...
|
||||
* }
|
||||
*/
|
||||
|
||||
typedef struct {
|
||||
const upb_strtable* t;
|
||||
size_t index;
|
||||
} upb_strtable_iter;
|
||||
|
||||
UPB_INLINE const upb_tabent* str_tabent(const upb_strtable_iter* i) {
|
||||
return &i->t->t.entries[i->index];
|
||||
}
|
||||
|
||||
void upb_strtable_begin(upb_strtable_iter* i, const upb_strtable* t);
|
||||
void upb_strtable_next(upb_strtable_iter* i);
|
||||
bool upb_strtable_done(const upb_strtable_iter* i);
|
||||
upb_StringView upb_strtable_iter_key(const upb_strtable_iter* i);
|
||||
upb_value upb_strtable_iter_value(const upb_strtable_iter* i);
|
||||
void upb_strtable_iter_setdone(upb_strtable_iter* i);
|
||||
bool upb_strtable_iter_isequal(const upb_strtable_iter* i1,
|
||||
const upb_strtable_iter* i2);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_HASH_STR_TABLE_H_ */
|
||||
1480
Pods/gRPC-Core/third_party/upb/upb/json/decode.c
generated
vendored
Normal file
1480
Pods/gRPC-Core/third_party/upb/upb/json/decode.c
generated
vendored
Normal file
@@ -0,0 +1,1480 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#include "upb/json/decode.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <float.h>
|
||||
#include <inttypes.h>
|
||||
#include <limits.h>
|
||||
#include <math.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/lex/atoi.h"
|
||||
#include "upb/lex/unicode.h"
|
||||
#include "upb/message/map.h"
|
||||
#include "upb/reflection/message.h"
|
||||
#include "upb/wire/encode.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct {
|
||||
const char *ptr, *end;
|
||||
upb_Arena* arena; /* TODO: should we have a tmp arena for tmp data? */
|
||||
const upb_DefPool* symtab;
|
||||
int depth;
|
||||
upb_Status* status;
|
||||
jmp_buf err;
|
||||
int line;
|
||||
const char* line_begin;
|
||||
bool is_first;
|
||||
int options;
|
||||
const upb_FieldDef* debug_field;
|
||||
} jsondec;
|
||||
|
||||
enum { JD_OBJECT, JD_ARRAY, JD_STRING, JD_NUMBER, JD_TRUE, JD_FALSE, JD_NULL };
|
||||
|
||||
/* Forward declarations of mutually-recursive functions. */
|
||||
static void jsondec_wellknown(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m);
|
||||
static upb_MessageValue jsondec_value(jsondec* d, const upb_FieldDef* f);
|
||||
static void jsondec_wellknownvalue(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m);
|
||||
static void jsondec_object(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m);
|
||||
|
||||
static bool jsondec_streql(upb_StringView str, const char* lit) {
|
||||
return str.size == strlen(lit) && memcmp(str.data, lit, str.size) == 0;
|
||||
}
|
||||
|
||||
static bool jsondec_isnullvalue(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_CType(f) == kUpb_CType_Enum &&
|
||||
strcmp(upb_EnumDef_FullName(upb_FieldDef_EnumSubDef(f)),
|
||||
"google.protobuf.NullValue") == 0;
|
||||
}
|
||||
|
||||
static bool jsondec_isvalue(const upb_FieldDef* f) {
|
||||
return (upb_FieldDef_CType(f) == kUpb_CType_Message &&
|
||||
upb_MessageDef_WellKnownType(upb_FieldDef_MessageSubDef(f)) ==
|
||||
kUpb_WellKnown_Value) ||
|
||||
jsondec_isnullvalue(f);
|
||||
}
|
||||
|
||||
UPB_NORETURN static void jsondec_err(jsondec* d, const char* msg) {
|
||||
upb_Status_SetErrorFormat(d->status, "Error parsing JSON @%d:%d: %s", d->line,
|
||||
(int)(d->ptr - d->line_begin), msg);
|
||||
UPB_LONGJMP(d->err, 1);
|
||||
}
|
||||
|
||||
UPB_PRINTF(2, 3)
|
||||
UPB_NORETURN static void jsondec_errf(jsondec* d, const char* fmt, ...) {
|
||||
va_list argp;
|
||||
upb_Status_SetErrorFormat(d->status, "Error parsing JSON @%d:%d: ", d->line,
|
||||
(int)(d->ptr - d->line_begin));
|
||||
va_start(argp, fmt);
|
||||
upb_Status_VAppendErrorFormat(d->status, fmt, argp);
|
||||
va_end(argp);
|
||||
UPB_LONGJMP(d->err, 1);
|
||||
}
|
||||
|
||||
static void jsondec_skipws(jsondec* d) {
|
||||
while (d->ptr != d->end) {
|
||||
switch (*d->ptr) {
|
||||
case '\n':
|
||||
d->line++;
|
||||
d->line_begin = d->ptr;
|
||||
/* Fallthrough. */
|
||||
case '\r':
|
||||
case '\t':
|
||||
case ' ':
|
||||
d->ptr++;
|
||||
break;
|
||||
default:
|
||||
return;
|
||||
}
|
||||
}
|
||||
jsondec_err(d, "Unexpected EOF");
|
||||
}
|
||||
|
||||
static bool jsondec_tryparsech(jsondec* d, char ch) {
|
||||
if (d->ptr == d->end || *d->ptr != ch) return false;
|
||||
d->ptr++;
|
||||
return true;
|
||||
}
|
||||
|
||||
static void jsondec_parselit(jsondec* d, const char* lit) {
|
||||
size_t avail = d->end - d->ptr;
|
||||
size_t len = strlen(lit);
|
||||
if (avail < len || memcmp(d->ptr, lit, len) != 0) {
|
||||
jsondec_errf(d, "Expected: '%s'", lit);
|
||||
}
|
||||
d->ptr += len;
|
||||
}
|
||||
|
||||
static void jsondec_wsch(jsondec* d, char ch) {
|
||||
jsondec_skipws(d);
|
||||
if (!jsondec_tryparsech(d, ch)) {
|
||||
jsondec_errf(d, "Expected: '%c'", ch);
|
||||
}
|
||||
}
|
||||
|
||||
static void jsondec_true(jsondec* d) { jsondec_parselit(d, "true"); }
|
||||
static void jsondec_false(jsondec* d) { jsondec_parselit(d, "false"); }
|
||||
static void jsondec_null(jsondec* d) { jsondec_parselit(d, "null"); }
|
||||
|
||||
static void jsondec_entrysep(jsondec* d) {
|
||||
jsondec_skipws(d);
|
||||
jsondec_parselit(d, ":");
|
||||
}
|
||||
|
||||
static int jsondec_rawpeek(jsondec* d) {
|
||||
switch (*d->ptr) {
|
||||
case '{':
|
||||
return JD_OBJECT;
|
||||
case '[':
|
||||
return JD_ARRAY;
|
||||
case '"':
|
||||
return JD_STRING;
|
||||
case '-':
|
||||
case '0':
|
||||
case '1':
|
||||
case '2':
|
||||
case '3':
|
||||
case '4':
|
||||
case '5':
|
||||
case '6':
|
||||
case '7':
|
||||
case '8':
|
||||
case '9':
|
||||
return JD_NUMBER;
|
||||
case 't':
|
||||
return JD_TRUE;
|
||||
case 'f':
|
||||
return JD_FALSE;
|
||||
case 'n':
|
||||
return JD_NULL;
|
||||
default:
|
||||
jsondec_errf(d, "Unexpected character: '%c'", *d->ptr);
|
||||
}
|
||||
}
|
||||
|
||||
/* JSON object/array **********************************************************/
|
||||
|
||||
/* These are used like so:
|
||||
*
|
||||
* jsondec_objstart(d);
|
||||
* while (jsondec_objnext(d)) {
|
||||
* ...
|
||||
* }
|
||||
* jsondec_objend(d) */
|
||||
|
||||
static int jsondec_peek(jsondec* d) {
|
||||
jsondec_skipws(d);
|
||||
return jsondec_rawpeek(d);
|
||||
}
|
||||
|
||||
static void jsondec_push(jsondec* d) {
|
||||
if (--d->depth < 0) {
|
||||
jsondec_err(d, "Recursion limit exceeded");
|
||||
}
|
||||
d->is_first = true;
|
||||
}
|
||||
|
||||
static bool jsondec_seqnext(jsondec* d, char end_ch) {
|
||||
bool is_first = d->is_first;
|
||||
d->is_first = false;
|
||||
jsondec_skipws(d);
|
||||
if (*d->ptr == end_ch) return false;
|
||||
if (!is_first) jsondec_parselit(d, ",");
|
||||
return true;
|
||||
}
|
||||
|
||||
static void jsondec_arrstart(jsondec* d) {
|
||||
jsondec_push(d);
|
||||
jsondec_wsch(d, '[');
|
||||
}
|
||||
|
||||
static void jsondec_arrend(jsondec* d) {
|
||||
d->depth++;
|
||||
jsondec_wsch(d, ']');
|
||||
}
|
||||
|
||||
static bool jsondec_arrnext(jsondec* d) { return jsondec_seqnext(d, ']'); }
|
||||
|
||||
static void jsondec_objstart(jsondec* d) {
|
||||
jsondec_push(d);
|
||||
jsondec_wsch(d, '{');
|
||||
}
|
||||
|
||||
static void jsondec_objend(jsondec* d) {
|
||||
d->depth++;
|
||||
jsondec_wsch(d, '}');
|
||||
}
|
||||
|
||||
static bool jsondec_objnext(jsondec* d) {
|
||||
if (!jsondec_seqnext(d, '}')) return false;
|
||||
if (jsondec_peek(d) != JD_STRING) {
|
||||
jsondec_err(d, "Object must start with string");
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/* JSON number ****************************************************************/
|
||||
|
||||
static bool jsondec_tryskipdigits(jsondec* d) {
|
||||
const char* start = d->ptr;
|
||||
|
||||
while (d->ptr < d->end) {
|
||||
if (*d->ptr < '0' || *d->ptr > '9') {
|
||||
break;
|
||||
}
|
||||
d->ptr++;
|
||||
}
|
||||
|
||||
return d->ptr != start;
|
||||
}
|
||||
|
||||
static void jsondec_skipdigits(jsondec* d) {
|
||||
if (!jsondec_tryskipdigits(d)) {
|
||||
jsondec_err(d, "Expected one or more digits");
|
||||
}
|
||||
}
|
||||
|
||||
static double jsondec_number(jsondec* d) {
|
||||
const char* start = d->ptr;
|
||||
|
||||
assert(jsondec_rawpeek(d) == JD_NUMBER);
|
||||
|
||||
/* Skip over the syntax of a number, as specified by JSON. */
|
||||
if (*d->ptr == '-') d->ptr++;
|
||||
|
||||
if (jsondec_tryparsech(d, '0')) {
|
||||
if (jsondec_tryskipdigits(d)) {
|
||||
jsondec_err(d, "number cannot have leading zero");
|
||||
}
|
||||
} else {
|
||||
jsondec_skipdigits(d);
|
||||
}
|
||||
|
||||
if (d->ptr == d->end) goto parse;
|
||||
if (jsondec_tryparsech(d, '.')) {
|
||||
jsondec_skipdigits(d);
|
||||
}
|
||||
if (d->ptr == d->end) goto parse;
|
||||
|
||||
if (*d->ptr == 'e' || *d->ptr == 'E') {
|
||||
d->ptr++;
|
||||
if (d->ptr == d->end) {
|
||||
jsondec_err(d, "Unexpected EOF in number");
|
||||
}
|
||||
if (*d->ptr == '+' || *d->ptr == '-') {
|
||||
d->ptr++;
|
||||
}
|
||||
jsondec_skipdigits(d);
|
||||
}
|
||||
|
||||
parse:
|
||||
/* Having verified the syntax of a JSON number, use strtod() to parse
|
||||
* (strtod() accepts a superset of JSON syntax). */
|
||||
errno = 0;
|
||||
{
|
||||
char* end;
|
||||
double val = strtod(start, &end);
|
||||
assert(end == d->ptr);
|
||||
|
||||
/* Currently the min/max-val conformance tests fail if we check this. Does
|
||||
* this mean the conformance tests are wrong or strtod() is wrong, or
|
||||
* something else? Investigate further. */
|
||||
/*
|
||||
if (errno == ERANGE) {
|
||||
jsondec_err(d, "Number out of range");
|
||||
}
|
||||
*/
|
||||
|
||||
if (val > DBL_MAX || val < -DBL_MAX) {
|
||||
jsondec_err(d, "Number out of range");
|
||||
}
|
||||
|
||||
return val;
|
||||
}
|
||||
}
|
||||
|
||||
/* JSON string ****************************************************************/
|
||||
|
||||
static char jsondec_escape(jsondec* d) {
|
||||
switch (*d->ptr++) {
|
||||
case '"':
|
||||
return '\"';
|
||||
case '\\':
|
||||
return '\\';
|
||||
case '/':
|
||||
return '/';
|
||||
case 'b':
|
||||
return '\b';
|
||||
case 'f':
|
||||
return '\f';
|
||||
case 'n':
|
||||
return '\n';
|
||||
case 'r':
|
||||
return '\r';
|
||||
case 't':
|
||||
return '\t';
|
||||
default:
|
||||
jsondec_err(d, "Invalid escape char");
|
||||
}
|
||||
}
|
||||
|
||||
static uint32_t jsondec_codepoint(jsondec* d) {
|
||||
uint32_t cp = 0;
|
||||
const char* end;
|
||||
|
||||
if (d->end - d->ptr < 4) {
|
||||
jsondec_err(d, "EOF inside string");
|
||||
}
|
||||
|
||||
end = d->ptr + 4;
|
||||
while (d->ptr < end) {
|
||||
char ch = *d->ptr++;
|
||||
if (ch >= '0' && ch <= '9') {
|
||||
ch -= '0';
|
||||
} else if (ch >= 'a' && ch <= 'f') {
|
||||
ch = ch - 'a' + 10;
|
||||
} else if (ch >= 'A' && ch <= 'F') {
|
||||
ch = ch - 'A' + 10;
|
||||
} else {
|
||||
jsondec_err(d, "Invalid hex digit");
|
||||
}
|
||||
cp = (cp << 4) | ch;
|
||||
}
|
||||
|
||||
return cp;
|
||||
}
|
||||
|
||||
/* Parses a \uXXXX unicode escape (possibly a surrogate pair). */
|
||||
static size_t jsondec_unicode(jsondec* d, char* out) {
|
||||
uint32_t cp = jsondec_codepoint(d);
|
||||
if (upb_Unicode_IsHigh(cp)) {
|
||||
/* Surrogate pair: two 16-bit codepoints become a 32-bit codepoint. */
|
||||
jsondec_parselit(d, "\\u");
|
||||
uint32_t low = jsondec_codepoint(d);
|
||||
if (!upb_Unicode_IsLow(low)) jsondec_err(d, "Invalid low surrogate");
|
||||
cp = upb_Unicode_FromPair(cp, low);
|
||||
} else if (upb_Unicode_IsLow(cp)) {
|
||||
jsondec_err(d, "Unpaired low surrogate");
|
||||
}
|
||||
|
||||
/* Write to UTF-8 */
|
||||
int bytes = upb_Unicode_ToUTF8(cp, out);
|
||||
if (bytes == 0) jsondec_err(d, "Invalid codepoint");
|
||||
return bytes;
|
||||
}
|
||||
|
||||
static void jsondec_resize(jsondec* d, char** buf, char** end, char** buf_end) {
|
||||
size_t oldsize = *buf_end - *buf;
|
||||
size_t len = *end - *buf;
|
||||
size_t size = UPB_MAX(8, 2 * oldsize);
|
||||
|
||||
*buf = upb_Arena_Realloc(d->arena, *buf, len, size);
|
||||
if (!*buf) jsondec_err(d, "Out of memory");
|
||||
|
||||
*end = *buf + len;
|
||||
*buf_end = *buf + size;
|
||||
}
|
||||
|
||||
static upb_StringView jsondec_string(jsondec* d) {
|
||||
char* buf = NULL;
|
||||
char* end = NULL;
|
||||
char* buf_end = NULL;
|
||||
|
||||
jsondec_skipws(d);
|
||||
|
||||
if (*d->ptr++ != '"') {
|
||||
jsondec_err(d, "Expected string");
|
||||
}
|
||||
|
||||
while (d->ptr < d->end) {
|
||||
char ch = *d->ptr++;
|
||||
|
||||
if (end == buf_end) {
|
||||
jsondec_resize(d, &buf, &end, &buf_end);
|
||||
}
|
||||
|
||||
switch (ch) {
|
||||
case '"': {
|
||||
upb_StringView ret;
|
||||
ret.data = buf;
|
||||
ret.size = end - buf;
|
||||
*end = '\0'; /* Needed for possible strtod(). */
|
||||
return ret;
|
||||
}
|
||||
case '\\':
|
||||
if (d->ptr == d->end) goto eof;
|
||||
if (*d->ptr == 'u') {
|
||||
d->ptr++;
|
||||
if (buf_end - end < 4) {
|
||||
/* Allow space for maximum-sized codepoint (4 bytes). */
|
||||
jsondec_resize(d, &buf, &end, &buf_end);
|
||||
}
|
||||
end += jsondec_unicode(d, end);
|
||||
} else {
|
||||
*end++ = jsondec_escape(d);
|
||||
}
|
||||
break;
|
||||
default:
|
||||
if ((unsigned char)*d->ptr < 0x20) {
|
||||
jsondec_err(d, "Invalid char in JSON string");
|
||||
}
|
||||
*end++ = ch;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
eof:
|
||||
jsondec_err(d, "EOF inside string");
|
||||
}
|
||||
|
||||
static void jsondec_skipval(jsondec* d) {
|
||||
switch (jsondec_peek(d)) {
|
||||
case JD_OBJECT:
|
||||
jsondec_objstart(d);
|
||||
while (jsondec_objnext(d)) {
|
||||
jsondec_string(d);
|
||||
jsondec_entrysep(d);
|
||||
jsondec_skipval(d);
|
||||
}
|
||||
jsondec_objend(d);
|
||||
break;
|
||||
case JD_ARRAY:
|
||||
jsondec_arrstart(d);
|
||||
while (jsondec_arrnext(d)) {
|
||||
jsondec_skipval(d);
|
||||
}
|
||||
jsondec_arrend(d);
|
||||
break;
|
||||
case JD_TRUE:
|
||||
jsondec_true(d);
|
||||
break;
|
||||
case JD_FALSE:
|
||||
jsondec_false(d);
|
||||
break;
|
||||
case JD_NULL:
|
||||
jsondec_null(d);
|
||||
break;
|
||||
case JD_STRING:
|
||||
jsondec_string(d);
|
||||
break;
|
||||
case JD_NUMBER:
|
||||
jsondec_number(d);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/* Base64 decoding for bytes fields. ******************************************/
|
||||
|
||||
static unsigned int jsondec_base64_tablelookup(const char ch) {
|
||||
/* Table includes the normal base64 chars plus the URL-safe variant. */
|
||||
const signed char table[256] = {
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, 62 /*+*/, -1, 62 /*-*/, -1, 63 /*/ */, 52 /*0*/,
|
||||
53 /*1*/, 54 /*2*/, 55 /*3*/, 56 /*4*/, 57 /*5*/, 58 /*6*/, 59 /*7*/,
|
||||
60 /*8*/, 61 /*9*/, -1, -1, -1, -1, -1,
|
||||
-1, -1, 0 /*A*/, 1 /*B*/, 2 /*C*/, 3 /*D*/, 4 /*E*/,
|
||||
5 /*F*/, 6 /*G*/, 07 /*H*/, 8 /*I*/, 9 /*J*/, 10 /*K*/, 11 /*L*/,
|
||||
12 /*M*/, 13 /*N*/, 14 /*O*/, 15 /*P*/, 16 /*Q*/, 17 /*R*/, 18 /*S*/,
|
||||
19 /*T*/, 20 /*U*/, 21 /*V*/, 22 /*W*/, 23 /*X*/, 24 /*Y*/, 25 /*Z*/,
|
||||
-1, -1, -1, -1, 63 /*_*/, -1, 26 /*a*/,
|
||||
27 /*b*/, 28 /*c*/, 29 /*d*/, 30 /*e*/, 31 /*f*/, 32 /*g*/, 33 /*h*/,
|
||||
34 /*i*/, 35 /*j*/, 36 /*k*/, 37 /*l*/, 38 /*m*/, 39 /*n*/, 40 /*o*/,
|
||||
41 /*p*/, 42 /*q*/, 43 /*r*/, 44 /*s*/, 45 /*t*/, 46 /*u*/, 47 /*v*/,
|
||||
48 /*w*/, 49 /*x*/, 50 /*y*/, 51 /*z*/, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1, -1, -1, -1,
|
||||
-1, -1, -1, -1};
|
||||
|
||||
/* Sign-extend return value so high bit will be set on any unexpected char. */
|
||||
return table[(unsigned)ch];
|
||||
}
|
||||
|
||||
static char* jsondec_partialbase64(jsondec* d, const char* ptr, const char* end,
|
||||
char* out) {
|
||||
int32_t val = -1;
|
||||
|
||||
switch (end - ptr) {
|
||||
case 2:
|
||||
val = jsondec_base64_tablelookup(ptr[0]) << 18 |
|
||||
jsondec_base64_tablelookup(ptr[1]) << 12;
|
||||
out[0] = val >> 16;
|
||||
out += 1;
|
||||
break;
|
||||
case 3:
|
||||
val = jsondec_base64_tablelookup(ptr[0]) << 18 |
|
||||
jsondec_base64_tablelookup(ptr[1]) << 12 |
|
||||
jsondec_base64_tablelookup(ptr[2]) << 6;
|
||||
out[0] = val >> 16;
|
||||
out[1] = (val >> 8) & 0xff;
|
||||
out += 2;
|
||||
break;
|
||||
}
|
||||
|
||||
if (val < 0) {
|
||||
jsondec_err(d, "Corrupt base64");
|
||||
}
|
||||
|
||||
return out;
|
||||
}
|
||||
|
||||
static size_t jsondec_base64(jsondec* d, upb_StringView str) {
|
||||
/* We decode in place. This is safe because this is a new buffer (not
|
||||
* aliasing the input) and because base64 decoding shrinks 4 bytes into 3. */
|
||||
char* out = (char*)str.data;
|
||||
const char* ptr = str.data;
|
||||
const char* end = ptr + str.size;
|
||||
const char* end4 = ptr + (str.size & -4); /* Round down to multiple of 4. */
|
||||
|
||||
for (; ptr < end4; ptr += 4, out += 3) {
|
||||
int val = jsondec_base64_tablelookup(ptr[0]) << 18 |
|
||||
jsondec_base64_tablelookup(ptr[1]) << 12 |
|
||||
jsondec_base64_tablelookup(ptr[2]) << 6 |
|
||||
jsondec_base64_tablelookup(ptr[3]) << 0;
|
||||
|
||||
if (val < 0) {
|
||||
/* Junk chars or padding. Remove trailing padding, if any. */
|
||||
if (end - ptr == 4 && ptr[3] == '=') {
|
||||
if (ptr[2] == '=') {
|
||||
end -= 2;
|
||||
} else {
|
||||
end -= 1;
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
out[0] = val >> 16;
|
||||
out[1] = (val >> 8) & 0xff;
|
||||
out[2] = val & 0xff;
|
||||
}
|
||||
|
||||
if (ptr < end) {
|
||||
/* Process remaining chars. We do not require padding. */
|
||||
out = jsondec_partialbase64(d, ptr, end, out);
|
||||
}
|
||||
|
||||
return out - str.data;
|
||||
}
|
||||
|
||||
/* Low-level integer parsing **************************************************/
|
||||
|
||||
static const char* jsondec_buftouint64(jsondec* d, const char* ptr,
|
||||
const char* end, uint64_t* val) {
|
||||
const char* out = upb_BufToUint64(ptr, end, val);
|
||||
if (!out) jsondec_err(d, "Integer overflow");
|
||||
return out;
|
||||
}
|
||||
|
||||
static const char* jsondec_buftoint64(jsondec* d, const char* ptr,
|
||||
const char* end, int64_t* val,
|
||||
bool* is_neg) {
|
||||
const char* out = upb_BufToInt64(ptr, end, val, is_neg);
|
||||
if (!out) jsondec_err(d, "Integer overflow");
|
||||
return out;
|
||||
}
|
||||
|
||||
static uint64_t jsondec_strtouint64(jsondec* d, upb_StringView str) {
|
||||
const char* end = str.data + str.size;
|
||||
uint64_t ret;
|
||||
if (jsondec_buftouint64(d, str.data, end, &ret) != end) {
|
||||
jsondec_err(d, "Non-number characters in quoted integer");
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int64_t jsondec_strtoint64(jsondec* d, upb_StringView str) {
|
||||
const char* end = str.data + str.size;
|
||||
int64_t ret;
|
||||
if (jsondec_buftoint64(d, str.data, end, &ret, NULL) != end) {
|
||||
jsondec_err(d, "Non-number characters in quoted integer");
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
/* Primitive value types ******************************************************/
|
||||
|
||||
/* Parse INT32 or INT64 value. */
|
||||
static upb_MessageValue jsondec_int(jsondec* d, const upb_FieldDef* f) {
|
||||
upb_MessageValue val;
|
||||
|
||||
switch (jsondec_peek(d)) {
|
||||
case JD_NUMBER: {
|
||||
double dbl = jsondec_number(d);
|
||||
if (dbl > 9223372036854774784.0 || dbl < -9223372036854775808.0) {
|
||||
jsondec_err(d, "JSON number is out of range.");
|
||||
}
|
||||
val.int64_val = dbl; /* must be guarded, overflow here is UB */
|
||||
if (val.int64_val != dbl) {
|
||||
jsondec_errf(d, "JSON number was not integral (%f != %" PRId64 ")", dbl,
|
||||
val.int64_val);
|
||||
}
|
||||
break;
|
||||
}
|
||||
case JD_STRING: {
|
||||
upb_StringView str = jsondec_string(d);
|
||||
val.int64_val = jsondec_strtoint64(d, str);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
jsondec_err(d, "Expected number or string");
|
||||
}
|
||||
|
||||
if (upb_FieldDef_CType(f) == kUpb_CType_Int32 ||
|
||||
upb_FieldDef_CType(f) == kUpb_CType_Enum) {
|
||||
if (val.int64_val > INT32_MAX || val.int64_val < INT32_MIN) {
|
||||
jsondec_err(d, "Integer out of range.");
|
||||
}
|
||||
val.int32_val = (int32_t)val.int64_val;
|
||||
}
|
||||
|
||||
return val;
|
||||
}
|
||||
|
||||
/* Parse UINT32 or UINT64 value. */
|
||||
static upb_MessageValue jsondec_uint(jsondec* d, const upb_FieldDef* f) {
|
||||
upb_MessageValue val = {0};
|
||||
|
||||
switch (jsondec_peek(d)) {
|
||||
case JD_NUMBER: {
|
||||
double dbl = jsondec_number(d);
|
||||
if (dbl > 18446744073709549568.0 || dbl < 0) {
|
||||
jsondec_err(d, "JSON number is out of range.");
|
||||
}
|
||||
val.uint64_val = dbl; /* must be guarded, overflow here is UB */
|
||||
if (val.uint64_val != dbl) {
|
||||
jsondec_errf(d, "JSON number was not integral (%f != %" PRIu64 ")", dbl,
|
||||
val.uint64_val);
|
||||
}
|
||||
break;
|
||||
}
|
||||
case JD_STRING: {
|
||||
upb_StringView str = jsondec_string(d);
|
||||
val.uint64_val = jsondec_strtouint64(d, str);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
jsondec_err(d, "Expected number or string");
|
||||
}
|
||||
|
||||
if (upb_FieldDef_CType(f) == kUpb_CType_UInt32) {
|
||||
if (val.uint64_val > UINT32_MAX) {
|
||||
jsondec_err(d, "Integer out of range.");
|
||||
}
|
||||
val.uint32_val = (uint32_t)val.uint64_val;
|
||||
}
|
||||
|
||||
return val;
|
||||
}
|
||||
|
||||
/* Parse DOUBLE or FLOAT value. */
|
||||
static upb_MessageValue jsondec_double(jsondec* d, const upb_FieldDef* f) {
|
||||
upb_StringView str;
|
||||
upb_MessageValue val = {0};
|
||||
|
||||
switch (jsondec_peek(d)) {
|
||||
case JD_NUMBER:
|
||||
val.double_val = jsondec_number(d);
|
||||
break;
|
||||
case JD_STRING:
|
||||
str = jsondec_string(d);
|
||||
if (jsondec_streql(str, "NaN")) {
|
||||
val.double_val = NAN;
|
||||
} else if (jsondec_streql(str, "Infinity")) {
|
||||
val.double_val = INFINITY;
|
||||
} else if (jsondec_streql(str, "-Infinity")) {
|
||||
val.double_val = -INFINITY;
|
||||
} else {
|
||||
val.double_val = strtod(str.data, NULL);
|
||||
}
|
||||
break;
|
||||
default:
|
||||
jsondec_err(d, "Expected number or string");
|
||||
}
|
||||
|
||||
if (upb_FieldDef_CType(f) == kUpb_CType_Float) {
|
||||
float f = val.double_val;
|
||||
if (val.double_val != INFINITY && val.double_val != -INFINITY) {
|
||||
if (f == INFINITY || f == -INFINITY) jsondec_err(d, "Float out of range");
|
||||
}
|
||||
val.float_val = f;
|
||||
}
|
||||
|
||||
return val;
|
||||
}
|
||||
|
||||
/* Parse STRING or BYTES value. */
|
||||
static upb_MessageValue jsondec_strfield(jsondec* d, const upb_FieldDef* f) {
|
||||
upb_MessageValue val;
|
||||
val.str_val = jsondec_string(d);
|
||||
if (upb_FieldDef_CType(f) == kUpb_CType_Bytes) {
|
||||
val.str_val.size = jsondec_base64(d, val.str_val);
|
||||
}
|
||||
return val;
|
||||
}
|
||||
|
||||
static upb_MessageValue jsondec_enum(jsondec* d, const upb_FieldDef* f) {
|
||||
switch (jsondec_peek(d)) {
|
||||
case JD_STRING: {
|
||||
upb_StringView str = jsondec_string(d);
|
||||
const upb_EnumDef* e = upb_FieldDef_EnumSubDef(f);
|
||||
const upb_EnumValueDef* ev =
|
||||
upb_EnumDef_FindValueByNameWithSize(e, str.data, str.size);
|
||||
upb_MessageValue val;
|
||||
if (ev) {
|
||||
val.int32_val = upb_EnumValueDef_Number(ev);
|
||||
} else {
|
||||
if (d->options & upb_JsonDecode_IgnoreUnknown) {
|
||||
val.int32_val = 0;
|
||||
} else {
|
||||
jsondec_errf(d, "Unknown enumerator: '" UPB_STRINGVIEW_FORMAT "'",
|
||||
UPB_STRINGVIEW_ARGS(str));
|
||||
}
|
||||
}
|
||||
return val;
|
||||
}
|
||||
case JD_NULL: {
|
||||
if (jsondec_isnullvalue(f)) {
|
||||
upb_MessageValue val;
|
||||
jsondec_null(d);
|
||||
val.int32_val = 0;
|
||||
return val;
|
||||
}
|
||||
}
|
||||
/* Fallthrough. */
|
||||
default:
|
||||
return jsondec_int(d, f);
|
||||
}
|
||||
}
|
||||
|
||||
static upb_MessageValue jsondec_bool(jsondec* d, const upb_FieldDef* f) {
|
||||
bool is_map_key = upb_FieldDef_Number(f) == 1 &&
|
||||
upb_MessageDef_IsMapEntry(upb_FieldDef_ContainingType(f));
|
||||
upb_MessageValue val;
|
||||
|
||||
if (is_map_key) {
|
||||
upb_StringView str = jsondec_string(d);
|
||||
if (jsondec_streql(str, "true")) {
|
||||
val.bool_val = true;
|
||||
} else if (jsondec_streql(str, "false")) {
|
||||
val.bool_val = false;
|
||||
} else {
|
||||
jsondec_err(d, "Invalid boolean map key");
|
||||
}
|
||||
} else {
|
||||
switch (jsondec_peek(d)) {
|
||||
case JD_TRUE:
|
||||
val.bool_val = true;
|
||||
jsondec_true(d);
|
||||
break;
|
||||
case JD_FALSE:
|
||||
val.bool_val = false;
|
||||
jsondec_false(d);
|
||||
break;
|
||||
default:
|
||||
jsondec_err(d, "Expected true or false");
|
||||
}
|
||||
}
|
||||
|
||||
return val;
|
||||
}
|
||||
|
||||
/* Composite types (array/message/map) ****************************************/
|
||||
|
||||
static void jsondec_array(jsondec* d, upb_Message* msg, const upb_FieldDef* f) {
|
||||
upb_Array* arr = upb_Message_Mutable(msg, f, d->arena).array;
|
||||
|
||||
jsondec_arrstart(d);
|
||||
while (jsondec_arrnext(d)) {
|
||||
upb_MessageValue elem = jsondec_value(d, f);
|
||||
upb_Array_Append(arr, elem, d->arena);
|
||||
}
|
||||
jsondec_arrend(d);
|
||||
}
|
||||
|
||||
static void jsondec_map(jsondec* d, upb_Message* msg, const upb_FieldDef* f) {
|
||||
upb_Map* map = upb_Message_Mutable(msg, f, d->arena).map;
|
||||
const upb_MessageDef* entry = upb_FieldDef_MessageSubDef(f);
|
||||
const upb_FieldDef* key_f = upb_MessageDef_FindFieldByNumber(entry, 1);
|
||||
const upb_FieldDef* val_f = upb_MessageDef_FindFieldByNumber(entry, 2);
|
||||
|
||||
jsondec_objstart(d);
|
||||
while (jsondec_objnext(d)) {
|
||||
upb_MessageValue key, val;
|
||||
key = jsondec_value(d, key_f);
|
||||
jsondec_entrysep(d);
|
||||
val = jsondec_value(d, val_f);
|
||||
upb_Map_Set(map, key, val, d->arena);
|
||||
}
|
||||
jsondec_objend(d);
|
||||
}
|
||||
|
||||
static void jsondec_tomsg(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
if (upb_MessageDef_WellKnownType(m) == kUpb_WellKnown_Unspecified) {
|
||||
jsondec_object(d, msg, m);
|
||||
} else {
|
||||
jsondec_wellknown(d, msg, m);
|
||||
}
|
||||
}
|
||||
|
||||
static upb_MessageValue jsondec_msg(jsondec* d, const upb_FieldDef* f) {
|
||||
const upb_MessageDef* m = upb_FieldDef_MessageSubDef(f);
|
||||
const upb_MiniTable* layout = upb_MessageDef_MiniTable(m);
|
||||
upb_Message* msg = upb_Message_New(layout, d->arena);
|
||||
upb_MessageValue val;
|
||||
|
||||
jsondec_tomsg(d, msg, m);
|
||||
val.msg_val = msg;
|
||||
return val;
|
||||
}
|
||||
|
||||
static void jsondec_field(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
upb_StringView name;
|
||||
const upb_FieldDef* f;
|
||||
const upb_FieldDef* preserved;
|
||||
|
||||
name = jsondec_string(d);
|
||||
jsondec_entrysep(d);
|
||||
|
||||
if (name.size >= 2 && name.data[0] == '[' &&
|
||||
name.data[name.size - 1] == ']') {
|
||||
f = upb_DefPool_FindExtensionByNameWithSize(d->symtab, name.data + 1,
|
||||
name.size - 2);
|
||||
if (f && upb_FieldDef_ContainingType(f) != m) {
|
||||
jsondec_errf(
|
||||
d, "Extension %s extends message %s, but was seen in message %s",
|
||||
upb_FieldDef_FullName(f),
|
||||
upb_MessageDef_FullName(upb_FieldDef_ContainingType(f)),
|
||||
upb_MessageDef_FullName(m));
|
||||
}
|
||||
} else {
|
||||
f = upb_MessageDef_FindByJsonNameWithSize(m, name.data, name.size);
|
||||
}
|
||||
|
||||
if (!f) {
|
||||
if ((d->options & upb_JsonDecode_IgnoreUnknown) == 0) {
|
||||
jsondec_errf(d, "No such field: " UPB_STRINGVIEW_FORMAT,
|
||||
UPB_STRINGVIEW_ARGS(name));
|
||||
}
|
||||
jsondec_skipval(d);
|
||||
return;
|
||||
}
|
||||
|
||||
if (jsondec_peek(d) == JD_NULL && !jsondec_isvalue(f)) {
|
||||
/* JSON "null" indicates a default value, so no need to set anything. */
|
||||
jsondec_null(d);
|
||||
return;
|
||||
}
|
||||
|
||||
if (upb_FieldDef_RealContainingOneof(f) &&
|
||||
upb_Message_WhichOneof(msg, upb_FieldDef_ContainingOneof(f))) {
|
||||
jsondec_err(d, "More than one field for this oneof.");
|
||||
}
|
||||
|
||||
preserved = d->debug_field;
|
||||
d->debug_field = f;
|
||||
|
||||
if (upb_FieldDef_IsMap(f)) {
|
||||
jsondec_map(d, msg, f);
|
||||
} else if (upb_FieldDef_IsRepeated(f)) {
|
||||
jsondec_array(d, msg, f);
|
||||
} else if (upb_FieldDef_IsSubMessage(f)) {
|
||||
upb_Message* submsg = upb_Message_Mutable(msg, f, d->arena).msg;
|
||||
const upb_MessageDef* subm = upb_FieldDef_MessageSubDef(f);
|
||||
jsondec_tomsg(d, submsg, subm);
|
||||
} else {
|
||||
upb_MessageValue val = jsondec_value(d, f);
|
||||
upb_Message_SetFieldByDef(msg, f, val, d->arena);
|
||||
}
|
||||
|
||||
d->debug_field = preserved;
|
||||
}
|
||||
|
||||
static void jsondec_object(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
jsondec_objstart(d);
|
||||
while (jsondec_objnext(d)) {
|
||||
jsondec_field(d, msg, m);
|
||||
}
|
||||
jsondec_objend(d);
|
||||
}
|
||||
|
||||
static upb_MessageValue jsondec_value(jsondec* d, const upb_FieldDef* f) {
|
||||
switch (upb_FieldDef_CType(f)) {
|
||||
case kUpb_CType_Bool:
|
||||
return jsondec_bool(d, f);
|
||||
case kUpb_CType_Float:
|
||||
case kUpb_CType_Double:
|
||||
return jsondec_double(d, f);
|
||||
case kUpb_CType_UInt32:
|
||||
case kUpb_CType_UInt64:
|
||||
return jsondec_uint(d, f);
|
||||
case kUpb_CType_Int32:
|
||||
case kUpb_CType_Int64:
|
||||
return jsondec_int(d, f);
|
||||
case kUpb_CType_String:
|
||||
case kUpb_CType_Bytes:
|
||||
return jsondec_strfield(d, f);
|
||||
case kUpb_CType_Enum:
|
||||
return jsondec_enum(d, f);
|
||||
case kUpb_CType_Message:
|
||||
return jsondec_msg(d, f);
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
}
|
||||
|
||||
/* Well-known types ***********************************************************/
|
||||
|
||||
static int jsondec_tsdigits(jsondec* d, const char** ptr, size_t digits,
|
||||
const char* after) {
|
||||
uint64_t val;
|
||||
const char* p = *ptr;
|
||||
const char* end = p + digits;
|
||||
size_t after_len = after ? strlen(after) : 0;
|
||||
|
||||
UPB_ASSERT(digits <= 9); /* int can't overflow. */
|
||||
|
||||
if (jsondec_buftouint64(d, p, end, &val) != end ||
|
||||
(after_len && memcmp(end, after, after_len) != 0)) {
|
||||
jsondec_err(d, "Malformed timestamp");
|
||||
}
|
||||
|
||||
UPB_ASSERT(val < INT_MAX);
|
||||
|
||||
*ptr = end + after_len;
|
||||
return (int)val;
|
||||
}
|
||||
|
||||
static int jsondec_nanos(jsondec* d, const char** ptr, const char* end) {
|
||||
uint64_t nanos = 0;
|
||||
const char* p = *ptr;
|
||||
|
||||
if (p != end && *p == '.') {
|
||||
const char* nano_end = jsondec_buftouint64(d, p + 1, end, &nanos);
|
||||
int digits = (int)(nano_end - p - 1);
|
||||
int exp_lg10 = 9 - digits;
|
||||
if (digits > 9) {
|
||||
jsondec_err(d, "Too many digits for partial seconds");
|
||||
}
|
||||
while (exp_lg10--) nanos *= 10;
|
||||
*ptr = nano_end;
|
||||
}
|
||||
|
||||
UPB_ASSERT(nanos < INT_MAX);
|
||||
|
||||
return (int)nanos;
|
||||
}
|
||||
|
||||
/* jsondec_epochdays(1970, 1, 1) == 1970-01-01 == 0. */
|
||||
int jsondec_epochdays(int y, int m, int d) {
|
||||
const uint32_t year_base = 4800; /* Before min year, multiple of 400. */
|
||||
const uint32_t m_adj = m - 3; /* March-based month. */
|
||||
const uint32_t carry = m_adj > (uint32_t)m ? 1 : 0;
|
||||
const uint32_t adjust = carry ? 12 : 0;
|
||||
const uint32_t y_adj = y + year_base - carry;
|
||||
const uint32_t month_days = ((m_adj + adjust) * 62719 + 769) / 2048;
|
||||
const uint32_t leap_days = y_adj / 4 - y_adj / 100 + y_adj / 400;
|
||||
return y_adj * 365 + leap_days + month_days + (d - 1) - 2472632;
|
||||
}
|
||||
|
||||
static int64_t jsondec_unixtime(int y, int m, int d, int h, int min, int s) {
|
||||
return (int64_t)jsondec_epochdays(y, m, d) * 86400 + h * 3600 + min * 60 + s;
|
||||
}
|
||||
|
||||
static void jsondec_timestamp(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
upb_MessageValue seconds;
|
||||
upb_MessageValue nanos;
|
||||
upb_StringView str = jsondec_string(d);
|
||||
const char* ptr = str.data;
|
||||
const char* end = ptr + str.size;
|
||||
|
||||
if (str.size < 20) goto malformed;
|
||||
|
||||
{
|
||||
/* 1972-01-01T01:00:00 */
|
||||
int year = jsondec_tsdigits(d, &ptr, 4, "-");
|
||||
int mon = jsondec_tsdigits(d, &ptr, 2, "-");
|
||||
int day = jsondec_tsdigits(d, &ptr, 2, "T");
|
||||
int hour = jsondec_tsdigits(d, &ptr, 2, ":");
|
||||
int min = jsondec_tsdigits(d, &ptr, 2, ":");
|
||||
int sec = jsondec_tsdigits(d, &ptr, 2, NULL);
|
||||
|
||||
seconds.int64_val = jsondec_unixtime(year, mon, day, hour, min, sec);
|
||||
}
|
||||
|
||||
nanos.int32_val = jsondec_nanos(d, &ptr, end);
|
||||
|
||||
{
|
||||
/* [+-]08:00 or Z */
|
||||
int ofs_hour = 0;
|
||||
int ofs_min = 0;
|
||||
bool neg = false;
|
||||
|
||||
if (ptr == end) goto malformed;
|
||||
|
||||
switch (*ptr++) {
|
||||
case '-':
|
||||
neg = true;
|
||||
/* fallthrough */
|
||||
case '+':
|
||||
if ((end - ptr) != 5) goto malformed;
|
||||
ofs_hour = jsondec_tsdigits(d, &ptr, 2, ":");
|
||||
ofs_min = jsondec_tsdigits(d, &ptr, 2, NULL);
|
||||
ofs_min = ((ofs_hour * 60) + ofs_min) * 60;
|
||||
seconds.int64_val += (neg ? ofs_min : -ofs_min);
|
||||
break;
|
||||
case 'Z':
|
||||
if (ptr != end) goto malformed;
|
||||
break;
|
||||
default:
|
||||
goto malformed;
|
||||
}
|
||||
}
|
||||
|
||||
if (seconds.int64_val < -62135596800) {
|
||||
jsondec_err(d, "Timestamp out of range");
|
||||
}
|
||||
|
||||
upb_Message_SetFieldByDef(msg, upb_MessageDef_FindFieldByNumber(m, 1),
|
||||
seconds, d->arena);
|
||||
upb_Message_SetFieldByDef(msg, upb_MessageDef_FindFieldByNumber(m, 2), nanos,
|
||||
d->arena);
|
||||
return;
|
||||
|
||||
malformed:
|
||||
jsondec_err(d, "Malformed timestamp");
|
||||
}
|
||||
|
||||
static void jsondec_duration(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
upb_MessageValue seconds;
|
||||
upb_MessageValue nanos;
|
||||
upb_StringView str = jsondec_string(d);
|
||||
const char* ptr = str.data;
|
||||
const char* end = ptr + str.size;
|
||||
const int64_t max = (uint64_t)3652500 * 86400;
|
||||
bool neg = false;
|
||||
|
||||
/* "3.000000001s", "3s", etc. */
|
||||
ptr = jsondec_buftoint64(d, ptr, end, &seconds.int64_val, &neg);
|
||||
nanos.int32_val = jsondec_nanos(d, &ptr, end);
|
||||
|
||||
if (end - ptr != 1 || *ptr != 's') {
|
||||
jsondec_err(d, "Malformed duration");
|
||||
}
|
||||
|
||||
if (seconds.int64_val < -max || seconds.int64_val > max) {
|
||||
jsondec_err(d, "Duration out of range");
|
||||
}
|
||||
|
||||
if (neg) {
|
||||
nanos.int32_val = -nanos.int32_val;
|
||||
}
|
||||
|
||||
upb_Message_SetFieldByDef(msg, upb_MessageDef_FindFieldByNumber(m, 1),
|
||||
seconds, d->arena);
|
||||
upb_Message_SetFieldByDef(msg, upb_MessageDef_FindFieldByNumber(m, 2), nanos,
|
||||
d->arena);
|
||||
}
|
||||
|
||||
static void jsondec_listvalue(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* values_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_MessageDef* value_m = upb_FieldDef_MessageSubDef(values_f);
|
||||
const upb_MiniTable* value_layout = upb_MessageDef_MiniTable(value_m);
|
||||
upb_Array* values = upb_Message_Mutable(msg, values_f, d->arena).array;
|
||||
|
||||
jsondec_arrstart(d);
|
||||
while (jsondec_arrnext(d)) {
|
||||
upb_Message* value_msg = upb_Message_New(value_layout, d->arena);
|
||||
upb_MessageValue value;
|
||||
value.msg_val = value_msg;
|
||||
upb_Array_Append(values, value, d->arena);
|
||||
jsondec_wellknownvalue(d, value_msg, value_m);
|
||||
}
|
||||
jsondec_arrend(d);
|
||||
}
|
||||
|
||||
static void jsondec_struct(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* fields_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_MessageDef* entry_m = upb_FieldDef_MessageSubDef(fields_f);
|
||||
const upb_FieldDef* value_f = upb_MessageDef_FindFieldByNumber(entry_m, 2);
|
||||
const upb_MessageDef* value_m = upb_FieldDef_MessageSubDef(value_f);
|
||||
const upb_MiniTable* value_layout = upb_MessageDef_MiniTable(value_m);
|
||||
upb_Map* fields = upb_Message_Mutable(msg, fields_f, d->arena).map;
|
||||
|
||||
jsondec_objstart(d);
|
||||
while (jsondec_objnext(d)) {
|
||||
upb_MessageValue key, value;
|
||||
upb_Message* value_msg = upb_Message_New(value_layout, d->arena);
|
||||
key.str_val = jsondec_string(d);
|
||||
value.msg_val = value_msg;
|
||||
upb_Map_Set(fields, key, value, d->arena);
|
||||
jsondec_entrysep(d);
|
||||
jsondec_wellknownvalue(d, value_msg, value_m);
|
||||
}
|
||||
jsondec_objend(d);
|
||||
}
|
||||
|
||||
static void jsondec_wellknownvalue(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
upb_MessageValue val;
|
||||
const upb_FieldDef* f;
|
||||
upb_Message* submsg;
|
||||
|
||||
switch (jsondec_peek(d)) {
|
||||
case JD_NUMBER:
|
||||
/* double number_value = 2; */
|
||||
f = upb_MessageDef_FindFieldByNumber(m, 2);
|
||||
val.double_val = jsondec_number(d);
|
||||
break;
|
||||
case JD_STRING:
|
||||
/* string string_value = 3; */
|
||||
f = upb_MessageDef_FindFieldByNumber(m, 3);
|
||||
val.str_val = jsondec_string(d);
|
||||
break;
|
||||
case JD_FALSE:
|
||||
/* bool bool_value = 4; */
|
||||
f = upb_MessageDef_FindFieldByNumber(m, 4);
|
||||
val.bool_val = false;
|
||||
jsondec_false(d);
|
||||
break;
|
||||
case JD_TRUE:
|
||||
/* bool bool_value = 4; */
|
||||
f = upb_MessageDef_FindFieldByNumber(m, 4);
|
||||
val.bool_val = true;
|
||||
jsondec_true(d);
|
||||
break;
|
||||
case JD_NULL:
|
||||
/* NullValue null_value = 1; */
|
||||
f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
val.int32_val = 0;
|
||||
jsondec_null(d);
|
||||
break;
|
||||
/* Note: these cases return, because upb_Message_Mutable() is enough. */
|
||||
case JD_OBJECT:
|
||||
/* Struct struct_value = 5; */
|
||||
f = upb_MessageDef_FindFieldByNumber(m, 5);
|
||||
submsg = upb_Message_Mutable(msg, f, d->arena).msg;
|
||||
jsondec_struct(d, submsg, upb_FieldDef_MessageSubDef(f));
|
||||
return;
|
||||
case JD_ARRAY:
|
||||
/* ListValue list_value = 6; */
|
||||
f = upb_MessageDef_FindFieldByNumber(m, 6);
|
||||
submsg = upb_Message_Mutable(msg, f, d->arena).msg;
|
||||
jsondec_listvalue(d, submsg, upb_FieldDef_MessageSubDef(f));
|
||||
return;
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
upb_Message_SetFieldByDef(msg, f, val, d->arena);
|
||||
}
|
||||
|
||||
static upb_StringView jsondec_mask(jsondec* d, const char* buf,
|
||||
const char* end) {
|
||||
/* FieldMask fields grow due to inserted '_' characters, so we can't do the
|
||||
* transform in place. */
|
||||
const char* ptr = buf;
|
||||
upb_StringView ret;
|
||||
char* out;
|
||||
|
||||
ret.size = end - ptr;
|
||||
while (ptr < end) {
|
||||
ret.size += (*ptr >= 'A' && *ptr <= 'Z');
|
||||
ptr++;
|
||||
}
|
||||
|
||||
out = upb_Arena_Malloc(d->arena, ret.size);
|
||||
ptr = buf;
|
||||
ret.data = out;
|
||||
|
||||
while (ptr < end) {
|
||||
char ch = *ptr++;
|
||||
if (ch >= 'A' && ch <= 'Z') {
|
||||
*out++ = '_';
|
||||
*out++ = ch + 32;
|
||||
} else if (ch == '_') {
|
||||
jsondec_err(d, "field mask may not contain '_'");
|
||||
} else {
|
||||
*out++ = ch;
|
||||
}
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static void jsondec_fieldmask(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
/* repeated string paths = 1; */
|
||||
const upb_FieldDef* paths_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
upb_Array* arr = upb_Message_Mutable(msg, paths_f, d->arena).array;
|
||||
upb_StringView str = jsondec_string(d);
|
||||
const char* ptr = str.data;
|
||||
const char* end = ptr + str.size;
|
||||
upb_MessageValue val;
|
||||
|
||||
while (ptr < end) {
|
||||
const char* elem_end = memchr(ptr, ',', end - ptr);
|
||||
if (elem_end) {
|
||||
val.str_val = jsondec_mask(d, ptr, elem_end);
|
||||
ptr = elem_end + 1;
|
||||
} else {
|
||||
val.str_val = jsondec_mask(d, ptr, end);
|
||||
ptr = end;
|
||||
}
|
||||
upb_Array_Append(arr, val, d->arena);
|
||||
}
|
||||
}
|
||||
|
||||
static void jsondec_anyfield(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
if (upb_MessageDef_WellKnownType(m) == kUpb_WellKnown_Unspecified) {
|
||||
/* For regular types: {"@type": "[user type]", "f1": <V1>, "f2": <V2>}
|
||||
* where f1, f2, etc. are the normal fields of this type. */
|
||||
jsondec_field(d, msg, m);
|
||||
} else {
|
||||
/* For well-known types: {"@type": "[well-known type]", "value": <X>}
|
||||
* where <X> is whatever encoding the WKT normally uses. */
|
||||
upb_StringView str = jsondec_string(d);
|
||||
jsondec_entrysep(d);
|
||||
if (!jsondec_streql(str, "value")) {
|
||||
jsondec_err(d, "Key for well-known type must be 'value'");
|
||||
}
|
||||
jsondec_wellknown(d, msg, m);
|
||||
}
|
||||
}
|
||||
|
||||
static const upb_MessageDef* jsondec_typeurl(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* type_url_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_MessageDef* type_m;
|
||||
upb_StringView type_url = jsondec_string(d);
|
||||
const char* end = type_url.data + type_url.size;
|
||||
const char* ptr = end;
|
||||
upb_MessageValue val;
|
||||
|
||||
val.str_val = type_url;
|
||||
upb_Message_SetFieldByDef(msg, type_url_f, val, d->arena);
|
||||
|
||||
/* Find message name after the last '/' */
|
||||
while (ptr > type_url.data && *--ptr != '/') {
|
||||
}
|
||||
|
||||
if (ptr == type_url.data || ptr == end) {
|
||||
jsondec_err(d, "Type url must have at least one '/' and non-empty host");
|
||||
}
|
||||
|
||||
ptr++;
|
||||
type_m = upb_DefPool_FindMessageByNameWithSize(d->symtab, ptr, end - ptr);
|
||||
|
||||
if (!type_m) {
|
||||
jsondec_err(d, "Type was not found");
|
||||
}
|
||||
|
||||
return type_m;
|
||||
}
|
||||
|
||||
static void jsondec_any(jsondec* d, upb_Message* msg, const upb_MessageDef* m) {
|
||||
/* string type_url = 1;
|
||||
* bytes value = 2; */
|
||||
const upb_FieldDef* value_f = upb_MessageDef_FindFieldByNumber(m, 2);
|
||||
upb_Message* any_msg;
|
||||
const upb_MessageDef* any_m = NULL;
|
||||
const char* pre_type_data = NULL;
|
||||
const char* pre_type_end = NULL;
|
||||
upb_MessageValue encoded;
|
||||
|
||||
jsondec_objstart(d);
|
||||
|
||||
/* Scan looking for "@type", which is not necessarily first. */
|
||||
while (!any_m && jsondec_objnext(d)) {
|
||||
const char* start = d->ptr;
|
||||
upb_StringView name = jsondec_string(d);
|
||||
jsondec_entrysep(d);
|
||||
if (jsondec_streql(name, "@type")) {
|
||||
any_m = jsondec_typeurl(d, msg, m);
|
||||
if (pre_type_data) {
|
||||
pre_type_end = start;
|
||||
while (*pre_type_end != ',') pre_type_end--;
|
||||
}
|
||||
} else {
|
||||
if (!pre_type_data) pre_type_data = start;
|
||||
jsondec_skipval(d);
|
||||
}
|
||||
}
|
||||
|
||||
if (!any_m) {
|
||||
jsondec_err(d, "Any object didn't contain a '@type' field");
|
||||
}
|
||||
|
||||
const upb_MiniTable* any_layout = upb_MessageDef_MiniTable(any_m);
|
||||
any_msg = upb_Message_New(any_layout, d->arena);
|
||||
|
||||
if (pre_type_data) {
|
||||
size_t len = pre_type_end - pre_type_data + 1;
|
||||
char* tmp = upb_Arena_Malloc(d->arena, len);
|
||||
const char* saved_ptr = d->ptr;
|
||||
const char* saved_end = d->end;
|
||||
memcpy(tmp, pre_type_data, len - 1);
|
||||
tmp[len - 1] = '}';
|
||||
d->ptr = tmp;
|
||||
d->end = tmp + len;
|
||||
d->is_first = true;
|
||||
while (jsondec_objnext(d)) {
|
||||
jsondec_anyfield(d, any_msg, any_m);
|
||||
}
|
||||
d->ptr = saved_ptr;
|
||||
d->end = saved_end;
|
||||
}
|
||||
|
||||
while (jsondec_objnext(d)) {
|
||||
jsondec_anyfield(d, any_msg, any_m);
|
||||
}
|
||||
|
||||
jsondec_objend(d);
|
||||
|
||||
upb_EncodeStatus status =
|
||||
upb_Encode(any_msg, upb_MessageDef_MiniTable(any_m), 0, d->arena,
|
||||
(char**)&encoded.str_val.data, &encoded.str_val.size);
|
||||
// TODO: We should fail gracefully here on a bad return status.
|
||||
UPB_ASSERT(status == kUpb_EncodeStatus_Ok);
|
||||
upb_Message_SetFieldByDef(msg, value_f, encoded, d->arena);
|
||||
}
|
||||
|
||||
static void jsondec_wrapper(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* value_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
upb_MessageValue val = jsondec_value(d, value_f);
|
||||
upb_Message_SetFieldByDef(msg, value_f, val, d->arena);
|
||||
}
|
||||
|
||||
static void jsondec_wellknown(jsondec* d, upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
switch (upb_MessageDef_WellKnownType(m)) {
|
||||
case kUpb_WellKnown_Any:
|
||||
jsondec_any(d, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_FieldMask:
|
||||
jsondec_fieldmask(d, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Duration:
|
||||
jsondec_duration(d, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Timestamp:
|
||||
jsondec_timestamp(d, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Value:
|
||||
jsondec_wellknownvalue(d, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_ListValue:
|
||||
jsondec_listvalue(d, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Struct:
|
||||
jsondec_struct(d, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_DoubleValue:
|
||||
case kUpb_WellKnown_FloatValue:
|
||||
case kUpb_WellKnown_Int64Value:
|
||||
case kUpb_WellKnown_UInt64Value:
|
||||
case kUpb_WellKnown_Int32Value:
|
||||
case kUpb_WellKnown_UInt32Value:
|
||||
case kUpb_WellKnown_StringValue:
|
||||
case kUpb_WellKnown_BytesValue:
|
||||
case kUpb_WellKnown_BoolValue:
|
||||
jsondec_wrapper(d, msg, m);
|
||||
break;
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
}
|
||||
|
||||
static bool upb_JsonDecoder_Decode(jsondec* const d, upb_Message* const msg,
|
||||
const upb_MessageDef* const m) {
|
||||
if (UPB_SETJMP(d->err)) return false;
|
||||
|
||||
jsondec_tomsg(d, msg, m);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_JsonDecode(const char* buf, size_t size, upb_Message* msg,
|
||||
const upb_MessageDef* m, const upb_DefPool* symtab,
|
||||
int options, upb_Arena* arena, upb_Status* status) {
|
||||
jsondec d;
|
||||
|
||||
if (size == 0) return true;
|
||||
|
||||
d.ptr = buf;
|
||||
d.end = buf + size;
|
||||
d.arena = arena;
|
||||
d.symtab = symtab;
|
||||
d.status = status;
|
||||
d.options = options;
|
||||
d.depth = 64;
|
||||
d.line = 1;
|
||||
d.line_begin = d.ptr;
|
||||
d.debug_field = NULL;
|
||||
d.is_first = false;
|
||||
|
||||
return upb_JsonDecoder_Decode(&d, msg, m);
|
||||
}
|
||||
55
Pods/gRPC-Core/third_party/upb/upb/json/decode.h
generated
vendored
Normal file
55
Pods/gRPC-Core/third_party/upb/upb/json/decode.h
generated
vendored
Normal file
@@ -0,0 +1,55 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef UPB_JSON_DECODE_H_
|
||||
#define UPB_JSON_DECODE_H_
|
||||
|
||||
#include "upb/reflection/def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
enum { upb_JsonDecode_IgnoreUnknown = 1 };
|
||||
|
||||
UPB_API bool upb_JsonDecode(const char* buf, size_t size, upb_Message* msg,
|
||||
const upb_MessageDef* m, const upb_DefPool* symtab,
|
||||
int options, upb_Arena* arena, upb_Status* status);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_JSONDECODE_H_ */
|
||||
807
Pods/gRPC-Core/third_party/upb/upb/json/encode.c
generated
vendored
Normal file
807
Pods/gRPC-Core/third_party/upb/upb/json/encode.c
generated
vendored
Normal file
@@ -0,0 +1,807 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#include "upb/json/encode.h"
|
||||
|
||||
#include <ctype.h>
|
||||
#include <float.h>
|
||||
#include <inttypes.h>
|
||||
#include <math.h>
|
||||
#include <stdarg.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/lex/round_trip.h"
|
||||
#include "upb/message/map.h"
|
||||
#include "upb/port/vsnprintf_compat.h"
|
||||
#include "upb/reflection/message.h"
|
||||
#include "upb/wire/decode.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct {
|
||||
char *buf, *ptr, *end;
|
||||
size_t overflow;
|
||||
int indent_depth;
|
||||
int options;
|
||||
const upb_DefPool* ext_pool;
|
||||
jmp_buf err;
|
||||
upb_Status* status;
|
||||
upb_Arena* arena;
|
||||
} jsonenc;
|
||||
|
||||
static void jsonenc_msg(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m);
|
||||
static void jsonenc_scalar(jsonenc* e, upb_MessageValue val,
|
||||
const upb_FieldDef* f);
|
||||
static void jsonenc_msgfield(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m);
|
||||
static void jsonenc_msgfields(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m, bool first);
|
||||
static void jsonenc_value(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m);
|
||||
|
||||
UPB_NORETURN static void jsonenc_err(jsonenc* e, const char* msg) {
|
||||
upb_Status_SetErrorMessage(e->status, msg);
|
||||
longjmp(e->err, 1);
|
||||
}
|
||||
|
||||
UPB_PRINTF(2, 3)
|
||||
UPB_NORETURN static void jsonenc_errf(jsonenc* e, const char* fmt, ...) {
|
||||
va_list argp;
|
||||
va_start(argp, fmt);
|
||||
upb_Status_VSetErrorFormat(e->status, fmt, argp);
|
||||
va_end(argp);
|
||||
longjmp(e->err, 1);
|
||||
}
|
||||
|
||||
static upb_Arena* jsonenc_arena(jsonenc* e) {
|
||||
/* Create lazily, since it's only needed for Any */
|
||||
if (!e->arena) {
|
||||
e->arena = upb_Arena_New();
|
||||
}
|
||||
return e->arena;
|
||||
}
|
||||
|
||||
static void jsonenc_putbytes(jsonenc* e, const void* data, size_t len) {
|
||||
size_t have = e->end - e->ptr;
|
||||
if (UPB_LIKELY(have >= len)) {
|
||||
memcpy(e->ptr, data, len);
|
||||
e->ptr += len;
|
||||
} else {
|
||||
if (have) {
|
||||
memcpy(e->ptr, data, have);
|
||||
e->ptr += have;
|
||||
}
|
||||
e->overflow += (len - have);
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_putstr(jsonenc* e, const char* str) {
|
||||
jsonenc_putbytes(e, str, strlen(str));
|
||||
}
|
||||
|
||||
UPB_PRINTF(2, 3)
|
||||
static void jsonenc_printf(jsonenc* e, const char* fmt, ...) {
|
||||
size_t n;
|
||||
size_t have = e->end - e->ptr;
|
||||
va_list args;
|
||||
|
||||
va_start(args, fmt);
|
||||
n = _upb_vsnprintf(e->ptr, have, fmt, args);
|
||||
va_end(args);
|
||||
|
||||
if (UPB_LIKELY(have > n)) {
|
||||
e->ptr += n;
|
||||
} else {
|
||||
e->ptr = UPB_PTRADD(e->ptr, have);
|
||||
e->overflow += (n - have);
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_nanos(jsonenc* e, int32_t nanos) {
|
||||
int digits = 9;
|
||||
|
||||
if (nanos == 0) return;
|
||||
if (nanos < 0 || nanos >= 1000000000) {
|
||||
jsonenc_err(e, "error formatting timestamp as JSON: invalid nanos");
|
||||
}
|
||||
|
||||
while (nanos % 1000 == 0) {
|
||||
nanos /= 1000;
|
||||
digits -= 3;
|
||||
}
|
||||
|
||||
jsonenc_printf(e, ".%.*" PRId32, digits, nanos);
|
||||
}
|
||||
|
||||
static void jsonenc_timestamp(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* seconds_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_FieldDef* nanos_f = upb_MessageDef_FindFieldByNumber(m, 2);
|
||||
int64_t seconds = upb_Message_GetFieldByDef(msg, seconds_f).int64_val;
|
||||
int32_t nanos = upb_Message_GetFieldByDef(msg, nanos_f).int32_val;
|
||||
int L, N, I, J, K, hour, min, sec;
|
||||
|
||||
if (seconds < -62135596800) {
|
||||
jsonenc_err(e,
|
||||
"error formatting timestamp as JSON: minimum acceptable value "
|
||||
"is 0001-01-01T00:00:00Z");
|
||||
} else if (seconds > 253402300799) {
|
||||
jsonenc_err(e,
|
||||
"error formatting timestamp as JSON: maximum acceptable value "
|
||||
"is 9999-12-31T23:59:59Z");
|
||||
}
|
||||
|
||||
/* Julian Day -> Y/M/D, Algorithm from:
|
||||
* Fliegel, H. F., and Van Flandern, T. C., "A Machine Algorithm for
|
||||
* Processing Calendar Dates," Communications of the Association of
|
||||
* Computing Machines, vol. 11 (1968), p. 657. */
|
||||
seconds += 62135596800; // Ensure seconds is positive.
|
||||
L = (int)(seconds / 86400) - 719162 + 68569 + 2440588;
|
||||
N = 4 * L / 146097;
|
||||
L = L - (146097 * N + 3) / 4;
|
||||
I = 4000 * (L + 1) / 1461001;
|
||||
L = L - 1461 * I / 4 + 31;
|
||||
J = 80 * L / 2447;
|
||||
K = L - 2447 * J / 80;
|
||||
L = J / 11;
|
||||
J = J + 2 - 12 * L;
|
||||
I = 100 * (N - 49) + I + L;
|
||||
|
||||
sec = seconds % 60;
|
||||
min = (seconds / 60) % 60;
|
||||
hour = (seconds / 3600) % 24;
|
||||
|
||||
jsonenc_printf(e, "\"%04d-%02d-%02dT%02d:%02d:%02d", I, J, K, hour, min, sec);
|
||||
jsonenc_nanos(e, nanos);
|
||||
jsonenc_putstr(e, "Z\"");
|
||||
}
|
||||
|
||||
static void jsonenc_duration(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* seconds_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_FieldDef* nanos_f = upb_MessageDef_FindFieldByNumber(m, 2);
|
||||
int64_t seconds = upb_Message_GetFieldByDef(msg, seconds_f).int64_val;
|
||||
int32_t nanos = upb_Message_GetFieldByDef(msg, nanos_f).int32_val;
|
||||
bool negative = false;
|
||||
|
||||
if (seconds > 315576000000 || seconds < -315576000000 ||
|
||||
(seconds != 0 && nanos != 0 && (seconds < 0) != (nanos < 0))) {
|
||||
jsonenc_err(e, "bad duration");
|
||||
}
|
||||
|
||||
if (seconds < 0) {
|
||||
negative = true;
|
||||
seconds = -seconds;
|
||||
}
|
||||
if (nanos < 0) {
|
||||
negative = true;
|
||||
nanos = -nanos;
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "\"");
|
||||
if (negative) {
|
||||
jsonenc_putstr(e, "-");
|
||||
}
|
||||
jsonenc_printf(e, "%" PRId64, seconds);
|
||||
jsonenc_nanos(e, nanos);
|
||||
jsonenc_putstr(e, "s\"");
|
||||
}
|
||||
|
||||
static void jsonenc_enum(int32_t val, const upb_FieldDef* f, jsonenc* e) {
|
||||
const upb_EnumDef* e_def = upb_FieldDef_EnumSubDef(f);
|
||||
|
||||
if (strcmp(upb_EnumDef_FullName(e_def), "google.protobuf.NullValue") == 0) {
|
||||
jsonenc_putstr(e, "null");
|
||||
} else {
|
||||
const upb_EnumValueDef* ev =
|
||||
(e->options & upb_JsonEncode_FormatEnumsAsIntegers)
|
||||
? NULL
|
||||
: upb_EnumDef_FindValueByNumber(e_def, val);
|
||||
|
||||
if (ev) {
|
||||
jsonenc_printf(e, "\"%s\"", upb_EnumValueDef_Name(ev));
|
||||
} else {
|
||||
jsonenc_printf(e, "%" PRId32, val);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_bytes(jsonenc* e, upb_StringView str) {
|
||||
/* This is the regular base64, not the "web-safe" version. */
|
||||
static const char base64[] =
|
||||
"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
||||
const unsigned char* ptr = (unsigned char*)str.data;
|
||||
const unsigned char* end = UPB_PTRADD(ptr, str.size);
|
||||
char buf[4];
|
||||
|
||||
jsonenc_putstr(e, "\"");
|
||||
|
||||
while (end - ptr >= 3) {
|
||||
buf[0] = base64[ptr[0] >> 2];
|
||||
buf[1] = base64[((ptr[0] & 0x3) << 4) | (ptr[1] >> 4)];
|
||||
buf[2] = base64[((ptr[1] & 0xf) << 2) | (ptr[2] >> 6)];
|
||||
buf[3] = base64[ptr[2] & 0x3f];
|
||||
jsonenc_putbytes(e, buf, 4);
|
||||
ptr += 3;
|
||||
}
|
||||
|
||||
switch (end - ptr) {
|
||||
case 2:
|
||||
buf[0] = base64[ptr[0] >> 2];
|
||||
buf[1] = base64[((ptr[0] & 0x3) << 4) | (ptr[1] >> 4)];
|
||||
buf[2] = base64[(ptr[1] & 0xf) << 2];
|
||||
buf[3] = '=';
|
||||
jsonenc_putbytes(e, buf, 4);
|
||||
break;
|
||||
case 1:
|
||||
buf[0] = base64[ptr[0] >> 2];
|
||||
buf[1] = base64[((ptr[0] & 0x3) << 4)];
|
||||
buf[2] = '=';
|
||||
buf[3] = '=';
|
||||
jsonenc_putbytes(e, buf, 4);
|
||||
break;
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "\"");
|
||||
}
|
||||
|
||||
static void jsonenc_stringbody(jsonenc* e, upb_StringView str) {
|
||||
const char* ptr = str.data;
|
||||
const char* end = UPB_PTRADD(ptr, str.size);
|
||||
|
||||
while (ptr < end) {
|
||||
switch (*ptr) {
|
||||
case '\n':
|
||||
jsonenc_putstr(e, "\\n");
|
||||
break;
|
||||
case '\r':
|
||||
jsonenc_putstr(e, "\\r");
|
||||
break;
|
||||
case '\t':
|
||||
jsonenc_putstr(e, "\\t");
|
||||
break;
|
||||
case '\"':
|
||||
jsonenc_putstr(e, "\\\"");
|
||||
break;
|
||||
case '\f':
|
||||
jsonenc_putstr(e, "\\f");
|
||||
break;
|
||||
case '\b':
|
||||
jsonenc_putstr(e, "\\b");
|
||||
break;
|
||||
case '\\':
|
||||
jsonenc_putstr(e, "\\\\");
|
||||
break;
|
||||
default:
|
||||
if ((uint8_t)*ptr < 0x20) {
|
||||
jsonenc_printf(e, "\\u%04x", (int)(uint8_t)*ptr);
|
||||
} else {
|
||||
/* This could be a non-ASCII byte. We rely on the string being valid
|
||||
* UTF-8. */
|
||||
jsonenc_putbytes(e, ptr, 1);
|
||||
}
|
||||
break;
|
||||
}
|
||||
ptr++;
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_string(jsonenc* e, upb_StringView str) {
|
||||
jsonenc_putstr(e, "\"");
|
||||
jsonenc_stringbody(e, str);
|
||||
jsonenc_putstr(e, "\"");
|
||||
}
|
||||
|
||||
static bool upb_JsonEncode_HandleSpecialDoubles(jsonenc* e, double val) {
|
||||
if (val == INFINITY) {
|
||||
jsonenc_putstr(e, "\"Infinity\"");
|
||||
} else if (val == -INFINITY) {
|
||||
jsonenc_putstr(e, "\"-Infinity\"");
|
||||
} else if (val != val) {
|
||||
jsonenc_putstr(e, "\"NaN\"");
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static void upb_JsonEncode_Double(jsonenc* e, double val) {
|
||||
if (upb_JsonEncode_HandleSpecialDoubles(e, val)) return;
|
||||
char buf[32];
|
||||
_upb_EncodeRoundTripDouble(val, buf, sizeof(buf));
|
||||
jsonenc_putstr(e, buf);
|
||||
}
|
||||
|
||||
static void upb_JsonEncode_Float(jsonenc* e, float val) {
|
||||
if (upb_JsonEncode_HandleSpecialDoubles(e, val)) return;
|
||||
char buf[32];
|
||||
_upb_EncodeRoundTripFloat(val, buf, sizeof(buf));
|
||||
jsonenc_putstr(e, buf);
|
||||
}
|
||||
|
||||
static void jsonenc_wrapper(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* val_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
upb_MessageValue val = upb_Message_GetFieldByDef(msg, val_f);
|
||||
jsonenc_scalar(e, val, val_f);
|
||||
}
|
||||
|
||||
static const upb_MessageDef* jsonenc_getanymsg(jsonenc* e,
|
||||
upb_StringView type_url) {
|
||||
/* Find last '/', if any. */
|
||||
const char* end = type_url.data + type_url.size;
|
||||
const char* ptr = end;
|
||||
const upb_MessageDef* ret;
|
||||
|
||||
if (!e->ext_pool) {
|
||||
jsonenc_err(e, "Tried to encode Any, but no symtab was provided");
|
||||
}
|
||||
|
||||
if (type_url.size == 0) goto badurl;
|
||||
|
||||
while (true) {
|
||||
if (--ptr == type_url.data) {
|
||||
/* Type URL must contain at least one '/', with host before. */
|
||||
goto badurl;
|
||||
}
|
||||
if (*ptr == '/') {
|
||||
ptr++;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
ret = upb_DefPool_FindMessageByNameWithSize(e->ext_pool, ptr, end - ptr);
|
||||
|
||||
if (!ret) {
|
||||
jsonenc_errf(e, "Couldn't find Any type: %.*s", (int)(end - ptr), ptr);
|
||||
}
|
||||
|
||||
return ret;
|
||||
|
||||
badurl:
|
||||
jsonenc_errf(e, "Bad type URL: " UPB_STRINGVIEW_FORMAT,
|
||||
UPB_STRINGVIEW_ARGS(type_url));
|
||||
}
|
||||
|
||||
static void jsonenc_any(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* type_url_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_FieldDef* value_f = upb_MessageDef_FindFieldByNumber(m, 2);
|
||||
upb_StringView type_url = upb_Message_GetFieldByDef(msg, type_url_f).str_val;
|
||||
upb_StringView value = upb_Message_GetFieldByDef(msg, value_f).str_val;
|
||||
const upb_MessageDef* any_m = jsonenc_getanymsg(e, type_url);
|
||||
const upb_MiniTable* any_layout = upb_MessageDef_MiniTable(any_m);
|
||||
upb_Arena* arena = jsonenc_arena(e);
|
||||
upb_Message* any = upb_Message_New(any_layout, arena);
|
||||
|
||||
if (upb_Decode(value.data, value.size, any, any_layout, NULL, 0, arena) !=
|
||||
kUpb_DecodeStatus_Ok) {
|
||||
jsonenc_err(e, "Error decoding message in Any");
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "{\"@type\":");
|
||||
jsonenc_string(e, type_url);
|
||||
|
||||
if (upb_MessageDef_WellKnownType(any_m) == kUpb_WellKnown_Unspecified) {
|
||||
/* Regular messages: {"@type": "...","foo": 1, "bar": 2} */
|
||||
jsonenc_msgfields(e, any, any_m, false);
|
||||
} else {
|
||||
/* Well-known type: {"@type": "...","value": <well-known encoding>} */
|
||||
jsonenc_putstr(e, ",\"value\":");
|
||||
jsonenc_msgfield(e, any, any_m);
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "}");
|
||||
}
|
||||
|
||||
static void jsonenc_putsep(jsonenc* e, const char* str, bool* first) {
|
||||
if (*first) {
|
||||
*first = false;
|
||||
} else {
|
||||
jsonenc_putstr(e, str);
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_fieldpath(jsonenc* e, upb_StringView path) {
|
||||
const char* ptr = path.data;
|
||||
const char* end = ptr + path.size;
|
||||
|
||||
while (ptr < end) {
|
||||
char ch = *ptr;
|
||||
|
||||
if (ch >= 'A' && ch <= 'Z') {
|
||||
jsonenc_err(e, "Field mask element may not have upper-case letter.");
|
||||
} else if (ch == '_') {
|
||||
if (ptr == end - 1 || *(ptr + 1) < 'a' || *(ptr + 1) > 'z') {
|
||||
jsonenc_err(e, "Underscore must be followed by a lowercase letter.");
|
||||
}
|
||||
ch = *++ptr - 32;
|
||||
}
|
||||
|
||||
jsonenc_putbytes(e, &ch, 1);
|
||||
ptr++;
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_fieldmask(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* paths_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_Array* paths = upb_Message_GetFieldByDef(msg, paths_f).array_val;
|
||||
bool first = true;
|
||||
size_t i, n = 0;
|
||||
|
||||
if (paths) n = upb_Array_Size(paths);
|
||||
|
||||
jsonenc_putstr(e, "\"");
|
||||
|
||||
for (i = 0; i < n; i++) {
|
||||
jsonenc_putsep(e, ",", &first);
|
||||
jsonenc_fieldpath(e, upb_Array_Get(paths, i).str_val);
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "\"");
|
||||
}
|
||||
|
||||
static void jsonenc_struct(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
jsonenc_putstr(e, "{");
|
||||
|
||||
const upb_FieldDef* fields_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_Map* fields = upb_Message_GetFieldByDef(msg, fields_f).map_val;
|
||||
|
||||
if (fields) {
|
||||
const upb_MessageDef* entry_m = upb_FieldDef_MessageSubDef(fields_f);
|
||||
const upb_FieldDef* value_f = upb_MessageDef_FindFieldByNumber(entry_m, 2);
|
||||
|
||||
size_t iter = kUpb_Map_Begin;
|
||||
bool first = true;
|
||||
|
||||
upb_MessageValue key, val;
|
||||
while (upb_Map_Next(fields, &key, &val, &iter)) {
|
||||
jsonenc_putsep(e, ",", &first);
|
||||
jsonenc_string(e, key.str_val);
|
||||
jsonenc_putstr(e, ":");
|
||||
jsonenc_value(e, val.msg_val, upb_FieldDef_MessageSubDef(value_f));
|
||||
}
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "}");
|
||||
}
|
||||
|
||||
static void jsonenc_listvalue(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
const upb_FieldDef* values_f = upb_MessageDef_FindFieldByNumber(m, 1);
|
||||
const upb_MessageDef* values_m = upb_FieldDef_MessageSubDef(values_f);
|
||||
const upb_Array* values = upb_Message_GetFieldByDef(msg, values_f).array_val;
|
||||
size_t i;
|
||||
bool first = true;
|
||||
|
||||
jsonenc_putstr(e, "[");
|
||||
|
||||
if (values) {
|
||||
const size_t size = upb_Array_Size(values);
|
||||
for (i = 0; i < size; i++) {
|
||||
upb_MessageValue elem = upb_Array_Get(values, i);
|
||||
|
||||
jsonenc_putsep(e, ",", &first);
|
||||
jsonenc_value(e, elem.msg_val, values_m);
|
||||
}
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "]");
|
||||
}
|
||||
|
||||
static void jsonenc_value(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
/* TODO: do we want a reflection method to get oneof case? */
|
||||
size_t iter = kUpb_Message_Begin;
|
||||
const upb_FieldDef* f;
|
||||
upb_MessageValue val;
|
||||
|
||||
if (!upb_Message_Next(msg, m, NULL, &f, &val, &iter)) {
|
||||
jsonenc_err(e, "No value set in Value proto");
|
||||
}
|
||||
|
||||
switch (upb_FieldDef_Number(f)) {
|
||||
case 1:
|
||||
jsonenc_putstr(e, "null");
|
||||
break;
|
||||
case 2:
|
||||
if (upb_JsonEncode_HandleSpecialDoubles(e, val.double_val)) {
|
||||
jsonenc_err(
|
||||
e,
|
||||
"google.protobuf.Value cannot encode double values for "
|
||||
"infinity or nan, because they would be parsed as a string");
|
||||
}
|
||||
upb_JsonEncode_Double(e, val.double_val);
|
||||
break;
|
||||
case 3:
|
||||
jsonenc_string(e, val.str_val);
|
||||
break;
|
||||
case 4:
|
||||
jsonenc_putstr(e, val.bool_val ? "true" : "false");
|
||||
break;
|
||||
case 5:
|
||||
jsonenc_struct(e, val.msg_val, upb_FieldDef_MessageSubDef(f));
|
||||
break;
|
||||
case 6:
|
||||
jsonenc_listvalue(e, val.msg_val, upb_FieldDef_MessageSubDef(f));
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_msgfield(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
switch (upb_MessageDef_WellKnownType(m)) {
|
||||
case kUpb_WellKnown_Unspecified:
|
||||
jsonenc_msg(e, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Any:
|
||||
jsonenc_any(e, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_FieldMask:
|
||||
jsonenc_fieldmask(e, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Duration:
|
||||
jsonenc_duration(e, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Timestamp:
|
||||
jsonenc_timestamp(e, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_DoubleValue:
|
||||
case kUpb_WellKnown_FloatValue:
|
||||
case kUpb_WellKnown_Int64Value:
|
||||
case kUpb_WellKnown_UInt64Value:
|
||||
case kUpb_WellKnown_Int32Value:
|
||||
case kUpb_WellKnown_UInt32Value:
|
||||
case kUpb_WellKnown_StringValue:
|
||||
case kUpb_WellKnown_BytesValue:
|
||||
case kUpb_WellKnown_BoolValue:
|
||||
jsonenc_wrapper(e, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Value:
|
||||
jsonenc_value(e, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_ListValue:
|
||||
jsonenc_listvalue(e, msg, m);
|
||||
break;
|
||||
case kUpb_WellKnown_Struct:
|
||||
jsonenc_struct(e, msg, m);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_scalar(jsonenc* e, upb_MessageValue val,
|
||||
const upb_FieldDef* f) {
|
||||
switch (upb_FieldDef_CType(f)) {
|
||||
case kUpb_CType_Bool:
|
||||
jsonenc_putstr(e, val.bool_val ? "true" : "false");
|
||||
break;
|
||||
case kUpb_CType_Float:
|
||||
upb_JsonEncode_Float(e, val.float_val);
|
||||
break;
|
||||
case kUpb_CType_Double:
|
||||
upb_JsonEncode_Double(e, val.double_val);
|
||||
break;
|
||||
case kUpb_CType_Int32:
|
||||
jsonenc_printf(e, "%" PRId32, val.int32_val);
|
||||
break;
|
||||
case kUpb_CType_UInt32:
|
||||
jsonenc_printf(e, "%" PRIu32, val.uint32_val);
|
||||
break;
|
||||
case kUpb_CType_Int64:
|
||||
jsonenc_printf(e, "\"%" PRId64 "\"", val.int64_val);
|
||||
break;
|
||||
case kUpb_CType_UInt64:
|
||||
jsonenc_printf(e, "\"%" PRIu64 "\"", val.uint64_val);
|
||||
break;
|
||||
case kUpb_CType_String:
|
||||
jsonenc_string(e, val.str_val);
|
||||
break;
|
||||
case kUpb_CType_Bytes:
|
||||
jsonenc_bytes(e, val.str_val);
|
||||
break;
|
||||
case kUpb_CType_Enum:
|
||||
jsonenc_enum(val.int32_val, f, e);
|
||||
break;
|
||||
case kUpb_CType_Message:
|
||||
jsonenc_msgfield(e, val.msg_val, upb_FieldDef_MessageSubDef(f));
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_mapkey(jsonenc* e, upb_MessageValue val,
|
||||
const upb_FieldDef* f) {
|
||||
jsonenc_putstr(e, "\"");
|
||||
|
||||
switch (upb_FieldDef_CType(f)) {
|
||||
case kUpb_CType_Bool:
|
||||
jsonenc_putstr(e, val.bool_val ? "true" : "false");
|
||||
break;
|
||||
case kUpb_CType_Int32:
|
||||
jsonenc_printf(e, "%" PRId32, val.int32_val);
|
||||
break;
|
||||
case kUpb_CType_UInt32:
|
||||
jsonenc_printf(e, "%" PRIu32, val.uint32_val);
|
||||
break;
|
||||
case kUpb_CType_Int64:
|
||||
jsonenc_printf(e, "%" PRId64, val.int64_val);
|
||||
break;
|
||||
case kUpb_CType_UInt64:
|
||||
jsonenc_printf(e, "%" PRIu64, val.uint64_val);
|
||||
break;
|
||||
case kUpb_CType_String:
|
||||
jsonenc_stringbody(e, val.str_val);
|
||||
break;
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "\":");
|
||||
}
|
||||
|
||||
static void jsonenc_array(jsonenc* e, const upb_Array* arr,
|
||||
const upb_FieldDef* f) {
|
||||
size_t i;
|
||||
size_t size = arr ? upb_Array_Size(arr) : 0;
|
||||
bool first = true;
|
||||
|
||||
jsonenc_putstr(e, "[");
|
||||
|
||||
for (i = 0; i < size; i++) {
|
||||
jsonenc_putsep(e, ",", &first);
|
||||
jsonenc_scalar(e, upb_Array_Get(arr, i), f);
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "]");
|
||||
}
|
||||
|
||||
static void jsonenc_map(jsonenc* e, const upb_Map* map, const upb_FieldDef* f) {
|
||||
jsonenc_putstr(e, "{");
|
||||
|
||||
const upb_MessageDef* entry = upb_FieldDef_MessageSubDef(f);
|
||||
const upb_FieldDef* key_f = upb_MessageDef_FindFieldByNumber(entry, 1);
|
||||
const upb_FieldDef* val_f = upb_MessageDef_FindFieldByNumber(entry, 2);
|
||||
|
||||
if (map) {
|
||||
size_t iter = kUpb_Map_Begin;
|
||||
bool first = true;
|
||||
|
||||
upb_MessageValue key, val;
|
||||
while (upb_Map_Next(map, &key, &val, &iter)) {
|
||||
jsonenc_putsep(e, ",", &first);
|
||||
jsonenc_mapkey(e, key, key_f);
|
||||
jsonenc_scalar(e, val, val_f);
|
||||
}
|
||||
}
|
||||
|
||||
jsonenc_putstr(e, "}");
|
||||
}
|
||||
|
||||
static void jsonenc_fieldval(jsonenc* e, const upb_FieldDef* f,
|
||||
upb_MessageValue val, bool* first) {
|
||||
const char* name;
|
||||
|
||||
jsonenc_putsep(e, ",", first);
|
||||
|
||||
if (upb_FieldDef_IsExtension(f)) {
|
||||
// TODO: For MessageSet, I would have expected this to print the message
|
||||
// name here, but Python doesn't appear to do this. We should do more
|
||||
// research here about what various implementations do.
|
||||
jsonenc_printf(e, "\"[%s]\":", upb_FieldDef_FullName(f));
|
||||
} else {
|
||||
if (e->options & upb_JsonEncode_UseProtoNames) {
|
||||
name = upb_FieldDef_Name(f);
|
||||
} else {
|
||||
name = upb_FieldDef_JsonName(f);
|
||||
}
|
||||
jsonenc_printf(e, "\"%s\":", name);
|
||||
}
|
||||
|
||||
if (upb_FieldDef_IsMap(f)) {
|
||||
jsonenc_map(e, val.map_val, f);
|
||||
} else if (upb_FieldDef_IsRepeated(f)) {
|
||||
jsonenc_array(e, val.array_val, f);
|
||||
} else {
|
||||
jsonenc_scalar(e, val, f);
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_msgfields(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m, bool first) {
|
||||
upb_MessageValue val;
|
||||
const upb_FieldDef* f;
|
||||
|
||||
if (e->options & upb_JsonEncode_EmitDefaults) {
|
||||
/* Iterate over all fields. */
|
||||
int i = 0;
|
||||
int n = upb_MessageDef_FieldCount(m);
|
||||
for (i = 0; i < n; i++) {
|
||||
f = upb_MessageDef_Field(m, i);
|
||||
if (!upb_FieldDef_HasPresence(f) || upb_Message_HasFieldByDef(msg, f)) {
|
||||
jsonenc_fieldval(e, f, upb_Message_GetFieldByDef(msg, f), &first);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
/* Iterate over non-empty fields. */
|
||||
size_t iter = kUpb_Message_Begin;
|
||||
while (upb_Message_Next(msg, m, e->ext_pool, &f, &val, &iter)) {
|
||||
jsonenc_fieldval(e, f, val, &first);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void jsonenc_msg(jsonenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
jsonenc_putstr(e, "{");
|
||||
jsonenc_msgfields(e, msg, m, true);
|
||||
jsonenc_putstr(e, "}");
|
||||
}
|
||||
|
||||
static size_t jsonenc_nullz(jsonenc* e, size_t size) {
|
||||
size_t ret = e->ptr - e->buf + e->overflow;
|
||||
|
||||
if (size > 0) {
|
||||
if (e->ptr == e->end) e->ptr--;
|
||||
*e->ptr = '\0';
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static size_t upb_JsonEncoder_Encode(jsonenc* const e,
|
||||
const upb_Message* const msg,
|
||||
const upb_MessageDef* const m,
|
||||
const size_t size) {
|
||||
if (UPB_SETJMP(e->err) != 0) return -1;
|
||||
|
||||
jsonenc_msgfield(e, msg, m);
|
||||
if (e->arena) upb_Arena_Free(e->arena);
|
||||
return jsonenc_nullz(e, size);
|
||||
}
|
||||
|
||||
size_t upb_JsonEncode(const upb_Message* msg, const upb_MessageDef* m,
|
||||
const upb_DefPool* ext_pool, int options, char* buf,
|
||||
size_t size, upb_Status* status) {
|
||||
jsonenc e;
|
||||
|
||||
e.buf = buf;
|
||||
e.ptr = buf;
|
||||
e.end = UPB_PTRADD(buf, size);
|
||||
e.overflow = 0;
|
||||
e.options = options;
|
||||
e.ext_pool = ext_pool;
|
||||
e.status = status;
|
||||
e.arena = NULL;
|
||||
|
||||
return upb_JsonEncoder_Encode(&e, msg, m, size);
|
||||
}
|
||||
73
Pods/gRPC-Core/third_party/upb/upb/json/encode.h
generated
vendored
Normal file
73
Pods/gRPC-Core/third_party/upb/upb/json/encode.h
generated
vendored
Normal file
@@ -0,0 +1,73 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef UPB_JSON_ENCODE_H_
|
||||
#define UPB_JSON_ENCODE_H_
|
||||
|
||||
#include "upb/reflection/def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
enum {
|
||||
/* When set, emits 0/default values. TODO: proto3 only? */
|
||||
upb_JsonEncode_EmitDefaults = 1 << 0,
|
||||
|
||||
/* When set, use normal (snake_case) field names instead of JSON (camelCase)
|
||||
names. */
|
||||
upb_JsonEncode_UseProtoNames = 1 << 1,
|
||||
|
||||
/* When set, emits enums as their integer values instead of as their names. */
|
||||
upb_JsonEncode_FormatEnumsAsIntegers = 1 << 2
|
||||
};
|
||||
|
||||
/* Encodes the given |msg| to JSON format. The message's reflection is given in
|
||||
* |m|. The DefPool in |ext_pool| is used to find extensions (if NULL,
|
||||
* extensions will not be printed).
|
||||
*
|
||||
* Output is placed in the given buffer, and always NULL-terminated. The output
|
||||
* size (excluding NULL) is returned. This means that a return value >= |size|
|
||||
* implies that the output was truncated. (These are the same semantics as
|
||||
* snprintf()). */
|
||||
UPB_API size_t upb_JsonEncode(const upb_Message* msg, const upb_MessageDef* m,
|
||||
const upb_DefPool* ext_pool, int options,
|
||||
char* buf, size_t size, upb_Status* status);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_JSONENCODE_H_ */
|
||||
48
Pods/gRPC-Core/third_party/upb/upb/lex/atoi.c
generated
vendored
Normal file
48
Pods/gRPC-Core/third_party/upb/upb/lex/atoi.c
generated
vendored
Normal file
@@ -0,0 +1,48 @@
|
||||
// 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/lex/atoi.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
const char* upb_BufToUint64(const char* ptr, const char* end, uint64_t* val) {
|
||||
uint64_t u64 = 0;
|
||||
while (ptr < end) {
|
||||
unsigned ch = *ptr - '0';
|
||||
if (ch >= 10) break;
|
||||
if (u64 > UINT64_MAX / 10 || u64 * 10 > UINT64_MAX - ch) {
|
||||
return NULL; // integer overflow
|
||||
}
|
||||
u64 *= 10;
|
||||
u64 += ch;
|
||||
ptr++;
|
||||
}
|
||||
|
||||
*val = u64;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
const char* upb_BufToInt64(const char* ptr, const char* end, int64_t* val,
|
||||
bool* is_neg) {
|
||||
bool neg = false;
|
||||
uint64_t u64;
|
||||
|
||||
if (ptr != end && *ptr == '-') {
|
||||
ptr++;
|
||||
neg = true;
|
||||
}
|
||||
|
||||
ptr = upb_BufToUint64(ptr, end, &u64);
|
||||
if (!ptr || u64 > (uint64_t)INT64_MAX + neg) {
|
||||
return NULL; // integer overflow
|
||||
}
|
||||
|
||||
*val = neg ? -u64 : u64;
|
||||
if (is_neg) *is_neg = neg;
|
||||
return ptr;
|
||||
}
|
||||
35
Pods/gRPC-Core/third_party/upb/upb/lex/atoi.h
generated
vendored
Normal file
35
Pods/gRPC-Core/third_party/upb/upb/lex/atoi.h
generated
vendored
Normal file
@@ -0,0 +1,35 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_LEX_ATOI_H_
|
||||
#define UPB_LEX_ATOI_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// We use these hand-written routines instead of strto[u]l() because the "long
|
||||
// long" variants aren't in c89. Also our version allows setting a ptr limit.
|
||||
// Return the new position of the pointer after parsing the int, or NULL on
|
||||
// integer overflow.
|
||||
|
||||
const char* upb_BufToUint64(const char* ptr, const char* end, uint64_t* val);
|
||||
const char* upb_BufToInt64(const char* ptr, const char* end, int64_t* val,
|
||||
bool* is_neg);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_LEX_ATOI_H_ */
|
||||
47
Pods/gRPC-Core/third_party/upb/upb/lex/round_trip.c
generated
vendored
Normal file
47
Pods/gRPC-Core/third_party/upb/upb/lex/round_trip.c
generated
vendored
Normal file
@@ -0,0 +1,47 @@
|
||||
// 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/lex/round_trip.h"
|
||||
|
||||
#include <float.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
/* Miscellaneous utilities ****************************************************/
|
||||
|
||||
static void upb_FixLocale(char* p) {
|
||||
/* printf() is dependent on locales; sadly there is no easy and portable way
|
||||
* to avoid this. This little post-processing step will translate 1,2 -> 1.2
|
||||
* since JSON needs the latter. Arguably a hack, but it is simple and the
|
||||
* alternatives are far more complicated, platform-dependent, and/or larger
|
||||
* in code size. */
|
||||
for (; *p; p++) {
|
||||
if (*p == ',') *p = '.';
|
||||
}
|
||||
}
|
||||
|
||||
void _upb_EncodeRoundTripDouble(double val, char* buf, size_t size) {
|
||||
assert(size >= kUpb_RoundTripBufferSize);
|
||||
snprintf(buf, size, "%.*g", DBL_DIG, val);
|
||||
if (strtod(buf, NULL) != val) {
|
||||
snprintf(buf, size, "%.*g", DBL_DIG + 2, val);
|
||||
assert(strtod(buf, NULL) == val);
|
||||
}
|
||||
upb_FixLocale(buf);
|
||||
}
|
||||
|
||||
void _upb_EncodeRoundTripFloat(float val, char* buf, size_t size) {
|
||||
assert(size >= kUpb_RoundTripBufferSize);
|
||||
snprintf(buf, size, "%.*g", FLT_DIG, val);
|
||||
if (strtof(buf, NULL) != val) {
|
||||
snprintf(buf, size, "%.*g", FLT_DIG + 3, val);
|
||||
assert(strtof(buf, NULL) == val);
|
||||
}
|
||||
upb_FixLocale(buf);
|
||||
}
|
||||
35
Pods/gRPC-Core/third_party/upb/upb/lex/round_trip.h
generated
vendored
Normal file
35
Pods/gRPC-Core/third_party/upb/upb/lex/round_trip.h
generated
vendored
Normal file
@@ -0,0 +1,35 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_LEX_ROUND_TRIP_H_
|
||||
#define UPB_LEX_ROUND_TRIP_H_
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// Encodes a float or double that is round-trippable, but as short as possible.
|
||||
// These routines are not fully optimal (not guaranteed to be shortest), but are
|
||||
// short-ish and match the implementation that has been used in protobuf since
|
||||
// the beginning.
|
||||
|
||||
// The given buffer size must be at least kUpb_RoundTripBufferSize.
|
||||
enum { kUpb_RoundTripBufferSize = 32 };
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
void _upb_EncodeRoundTripDouble(double val, char* buf, size_t size);
|
||||
void _upb_EncodeRoundTripFloat(float val, char* buf, size_t size);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_LEX_ROUND_TRIP_H_ */
|
||||
77
Pods/gRPC-Core/third_party/upb/upb/lex/strtod.c
generated
vendored
Normal file
77
Pods/gRPC-Core/third_party/upb/upb/lex/strtod.c
generated
vendored
Normal file
@@ -0,0 +1,77 @@
|
||||
// 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/lex/strtod.h"
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// Determine the locale-specific radix character by calling sprintf() to print
|
||||
// the number 1.5, then stripping off the digits. As far as I can tell, this
|
||||
// is the only portable, thread-safe way to get the C library to divulge the
|
||||
// locale's radix character. No, localeconv() is NOT thread-safe.
|
||||
|
||||
static int GetLocaleRadix(char *data, size_t capacity) {
|
||||
char temp[16];
|
||||
const int size = snprintf(temp, sizeof(temp), "%.1f", 1.5);
|
||||
UPB_ASSERT(temp[0] == '1');
|
||||
UPB_ASSERT(temp[size - 1] == '5');
|
||||
UPB_ASSERT(size < capacity);
|
||||
temp[size - 1] = '\0';
|
||||
strcpy(data, temp + 1);
|
||||
return size - 2;
|
||||
}
|
||||
|
||||
// Populates a string identical to *input except that the character pointed to
|
||||
// by pos (which should be '.') is replaced with the locale-specific radix.
|
||||
|
||||
static void LocalizeRadix(const char *input, const char *pos, char *output) {
|
||||
const int len1 = pos - input;
|
||||
|
||||
char radix[8];
|
||||
const int len2 = GetLocaleRadix(radix, sizeof(radix));
|
||||
|
||||
memcpy(output, input, len1);
|
||||
memcpy(output + len1, radix, len2);
|
||||
strcpy(output + len1 + len2, input + len1 + 1);
|
||||
}
|
||||
|
||||
double _upb_NoLocaleStrtod(const char *str, char **endptr) {
|
||||
// We cannot simply set the locale to "C" temporarily with setlocale()
|
||||
// as this is not thread-safe. Instead, we try to parse in the current
|
||||
// locale first. If parsing stops at a '.' character, then this is a
|
||||
// pretty good hint that we're actually in some other locale in which
|
||||
// '.' is not the radix character.
|
||||
|
||||
char *temp_endptr;
|
||||
double result = strtod(str, &temp_endptr);
|
||||
if (endptr != NULL) *endptr = temp_endptr;
|
||||
if (*temp_endptr != '.') return result;
|
||||
|
||||
// Parsing halted on a '.'. Perhaps we're in a different locale? Let's
|
||||
// try to replace the '.' with a locale-specific radix character and
|
||||
// try again.
|
||||
|
||||
char localized[80];
|
||||
LocalizeRadix(str, temp_endptr, localized);
|
||||
char *localized_endptr;
|
||||
result = strtod(localized, &localized_endptr);
|
||||
if ((localized_endptr - &localized[0]) > (temp_endptr - str)) {
|
||||
// This attempt got further, so replacing the decimal must have helped.
|
||||
// Update endptr to point at the right location.
|
||||
if (endptr != NULL) {
|
||||
// size_diff is non-zero if the localized radix has multiple bytes.
|
||||
int size_diff = strlen(localized) - strlen(str);
|
||||
*endptr = (char *)str + (localized_endptr - &localized[0] - size_diff);
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
26
Pods/gRPC-Core/third_party/upb/upb/lex/strtod.h
generated
vendored
Normal file
26
Pods/gRPC-Core/third_party/upb/upb/lex/strtod.h
generated
vendored
Normal file
@@ -0,0 +1,26 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_LEX_STRTOD_H_
|
||||
#define UPB_LEX_STRTOD_H_
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
double _upb_NoLocaleStrtod(const char *str, char **endptr);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_LEX_STRTOD_H_ */
|
||||
37
Pods/gRPC-Core/third_party/upb/upb/lex/unicode.c
generated
vendored
Normal file
37
Pods/gRPC-Core/third_party/upb/upb/lex/unicode.c
generated
vendored
Normal file
@@ -0,0 +1,37 @@
|
||||
// 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/lex/unicode.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
int upb_Unicode_ToUTF8(uint32_t cp, char* out) {
|
||||
if (cp <= 0x7f) {
|
||||
out[0] = cp;
|
||||
return 1;
|
||||
}
|
||||
if (cp <= 0x07ff) {
|
||||
out[0] = (cp >> 6) | 0xc0;
|
||||
out[1] = (cp & 0x3f) | 0x80;
|
||||
return 2;
|
||||
}
|
||||
if (cp <= 0xffff) {
|
||||
out[0] = (cp >> 12) | 0xe0;
|
||||
out[1] = ((cp >> 6) & 0x3f) | 0x80;
|
||||
out[2] = (cp & 0x3f) | 0x80;
|
||||
return 3;
|
||||
}
|
||||
if (cp <= 0x10ffff) {
|
||||
out[0] = (cp >> 18) | 0xf0;
|
||||
out[1] = ((cp >> 12) & 0x3f) | 0x80;
|
||||
out[2] = ((cp >> 6) & 0x3f) | 0x80;
|
||||
out[3] = (cp & 0x3f) | 0x80;
|
||||
return 4;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
59
Pods/gRPC-Core/third_party/upb/upb/lex/unicode.h
generated
vendored
Normal file
59
Pods/gRPC-Core/third_party/upb/upb/lex/unicode.h
generated
vendored
Normal file
@@ -0,0 +1,59 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_LEX_UNICODE_H_
|
||||
#define UPB_LEX_UNICODE_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Returns true iff a codepoint is the value for a high surrogate.
|
||||
UPB_INLINE bool upb_Unicode_IsHigh(uint32_t cp) {
|
||||
return (cp >= 0xd800 && cp <= 0xdbff);
|
||||
}
|
||||
|
||||
// Returns true iff a codepoint is the value for a low surrogate.
|
||||
UPB_INLINE bool upb_Unicode_IsLow(uint32_t cp) {
|
||||
return (cp >= 0xdc00 && cp <= 0xdfff);
|
||||
}
|
||||
|
||||
// Returns the high 16-bit surrogate value for a supplementary codepoint.
|
||||
// Does not sanity-check the input.
|
||||
UPB_INLINE uint16_t upb_Unicode_ToHigh(uint32_t cp) {
|
||||
return (cp >> 10) + 0xd7c0;
|
||||
}
|
||||
|
||||
// Returns the low 16-bit surrogate value for a supplementary codepoint.
|
||||
// Does not sanity-check the input.
|
||||
UPB_INLINE uint16_t upb_Unicode_ToLow(uint32_t cp) {
|
||||
return (cp & 0x3ff) | 0xdc00;
|
||||
}
|
||||
|
||||
// Returns the 32-bit value corresponding to a pair of 16-bit surrogates.
|
||||
// Does not sanity-check the input.
|
||||
UPB_INLINE uint32_t upb_Unicode_FromPair(uint32_t high, uint32_t low) {
|
||||
return ((high & 0x3ff) << 10) + (low & 0x3ff) + 0x10000;
|
||||
}
|
||||
|
||||
// Outputs a codepoint as UTF8.
|
||||
// Returns the number of bytes written (1-4 on success, 0 on error).
|
||||
// Does not sanity-check the input. Specifically does not check for surrogates.
|
||||
int upb_Unicode_ToUTF8(uint32_t cp, char* out);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_LEX_UNICODE_H_ */
|
||||
27
Pods/gRPC-Core/third_party/upb/upb/mem/alloc.c
generated
vendored
Normal file
27
Pods/gRPC-Core/third_party/upb/upb/mem/alloc.c
generated
vendored
Normal file
@@ -0,0 +1,27 @@
|
||||
// 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/mem/alloc.h"
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
static void* upb_global_allocfunc(upb_alloc* alloc, void* ptr, size_t oldsize,
|
||||
size_t size) {
|
||||
UPB_UNUSED(alloc);
|
||||
UPB_UNUSED(oldsize);
|
||||
if (size == 0) {
|
||||
free(ptr);
|
||||
return NULL;
|
||||
} else {
|
||||
return realloc(ptr, size);
|
||||
}
|
||||
}
|
||||
|
||||
upb_alloc upb_alloc_global = {&upb_global_allocfunc};
|
||||
78
Pods/gRPC-Core/third_party/upb/upb/mem/alloc.h
generated
vendored
Normal file
78
Pods/gRPC-Core/third_party/upb/upb/mem/alloc.h
generated
vendored
Normal file
@@ -0,0 +1,78 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MEM_ALLOC_H_
|
||||
#define UPB_MEM_ALLOC_H_
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
typedef struct upb_alloc upb_alloc;
|
||||
|
||||
/* A combined `malloc()`/`free()` function.
|
||||
* If `size` is 0 then the function acts like `free()`, otherwise it acts like
|
||||
* `realloc()`. Only `oldsize` bytes from a previous allocation are
|
||||
* preserved. */
|
||||
typedef void* upb_alloc_func(upb_alloc* alloc, void* ptr, size_t oldsize,
|
||||
size_t size);
|
||||
|
||||
/* A upb_alloc is a possibly-stateful allocator object.
|
||||
*
|
||||
* It could either be an arena allocator (which doesn't require individual
|
||||
* `free()` calls) or a regular `malloc()` (which does). The client must
|
||||
* therefore free memory unless it knows that the allocator is an arena
|
||||
* allocator. */
|
||||
struct upb_alloc {
|
||||
upb_alloc_func* func;
|
||||
};
|
||||
|
||||
UPB_INLINE void* upb_malloc(upb_alloc* alloc, size_t size) {
|
||||
UPB_ASSERT(alloc);
|
||||
return alloc->func(alloc, NULL, 0, size);
|
||||
}
|
||||
|
||||
UPB_INLINE void* upb_realloc(upb_alloc* alloc, void* ptr, size_t oldsize,
|
||||
size_t size) {
|
||||
UPB_ASSERT(alloc);
|
||||
return alloc->func(alloc, ptr, oldsize, size);
|
||||
}
|
||||
|
||||
UPB_INLINE void upb_free(upb_alloc* alloc, void* ptr) {
|
||||
UPB_ASSERT(alloc);
|
||||
alloc->func(alloc, ptr, 0, 0);
|
||||
}
|
||||
|
||||
// The global allocator used by upb. Uses the standard malloc()/free().
|
||||
|
||||
extern upb_alloc upb_alloc_global;
|
||||
|
||||
/* Functions that hard-code the global malloc.
|
||||
*
|
||||
* We still get benefit because we can put custom logic into our global
|
||||
* allocator, like injecting out-of-memory faults in debug/testing builds. */
|
||||
|
||||
UPB_INLINE void* upb_gmalloc(size_t size) {
|
||||
return upb_malloc(&upb_alloc_global, size);
|
||||
}
|
||||
|
||||
UPB_INLINE void* upb_grealloc(void* ptr, size_t oldsize, size_t size) {
|
||||
return upb_realloc(&upb_alloc_global, ptr, oldsize, size);
|
||||
}
|
||||
|
||||
UPB_INLINE void upb_gfree(void* ptr) { upb_free(&upb_alloc_global, ptr); }
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MEM_ALLOC_H_ */
|
||||
368
Pods/gRPC-Core/third_party/upb/upb/mem/arena.c
generated
vendored
Normal file
368
Pods/gRPC-Core/third_party/upb/upb/mem/arena.c
generated
vendored
Normal file
@@ -0,0 +1,368 @@
|
||||
// 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/mem/internal/arena.h"
|
||||
|
||||
#include "upb/port/atomic.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct _upb_MemBlock {
|
||||
// Atomic only for the benefit of SpaceAllocated().
|
||||
UPB_ATOMIC(_upb_MemBlock*) next;
|
||||
uint32_t size;
|
||||
// Data follows.
|
||||
};
|
||||
|
||||
static const size_t memblock_reserve =
|
||||
UPB_ALIGN_UP(sizeof(_upb_MemBlock), UPB_MALLOC_ALIGN);
|
||||
|
||||
typedef struct _upb_ArenaRoot {
|
||||
upb_Arena* root;
|
||||
uintptr_t tagged_count;
|
||||
} _upb_ArenaRoot;
|
||||
|
||||
static _upb_ArenaRoot _upb_Arena_FindRoot(upb_Arena* a) {
|
||||
uintptr_t poc = upb_Atomic_Load(&a->parent_or_count, memory_order_acquire);
|
||||
while (_upb_Arena_IsTaggedPointer(poc)) {
|
||||
upb_Arena* next = _upb_Arena_PointerFromTagged(poc);
|
||||
UPB_ASSERT(a != next);
|
||||
uintptr_t next_poc =
|
||||
upb_Atomic_Load(&next->parent_or_count, memory_order_acquire);
|
||||
|
||||
if (_upb_Arena_IsTaggedPointer(next_poc)) {
|
||||
// To keep complexity down, we lazily collapse levels of the tree. This
|
||||
// keeps it flat in the final case, but doesn't cost much incrementally.
|
||||
//
|
||||
// Path splitting keeps time complexity down, see:
|
||||
// https://en.wikipedia.org/wiki/Disjoint-set_data_structure
|
||||
//
|
||||
// We can safely use a relaxed atomic here because all threads doing this
|
||||
// will converge on the same value and we don't need memory orderings to
|
||||
// be visible.
|
||||
//
|
||||
// This is true because:
|
||||
// - If no fuses occur, this will eventually become the root.
|
||||
// - If fuses are actively occurring, the root may change, but the
|
||||
// invariant is that `parent_or_count` merely points to *a* parent.
|
||||
//
|
||||
// In other words, it is moving towards "the" root, and that root may move
|
||||
// further away over time, but the path towards that root will continue to
|
||||
// be valid and the creation of the path carries all the memory orderings
|
||||
// required.
|
||||
UPB_ASSERT(a != _upb_Arena_PointerFromTagged(next_poc));
|
||||
upb_Atomic_Store(&a->parent_or_count, next_poc, memory_order_relaxed);
|
||||
}
|
||||
a = next;
|
||||
poc = next_poc;
|
||||
}
|
||||
return (_upb_ArenaRoot){.root = a, .tagged_count = poc};
|
||||
}
|
||||
|
||||
size_t upb_Arena_SpaceAllocated(upb_Arena* arena) {
|
||||
arena = _upb_Arena_FindRoot(arena).root;
|
||||
size_t memsize = 0;
|
||||
|
||||
while (arena != NULL) {
|
||||
_upb_MemBlock* block =
|
||||
upb_Atomic_Load(&arena->blocks, memory_order_relaxed);
|
||||
while (block != NULL) {
|
||||
memsize += sizeof(_upb_MemBlock) + block->size;
|
||||
block = upb_Atomic_Load(&block->next, memory_order_relaxed);
|
||||
}
|
||||
arena = upb_Atomic_Load(&arena->next, memory_order_relaxed);
|
||||
}
|
||||
|
||||
return memsize;
|
||||
}
|
||||
|
||||
uint32_t upb_Arena_DebugRefCount(upb_Arena* a) {
|
||||
// These loads could probably be relaxed, but given that this is debug-only,
|
||||
// it's not worth introducing a new variant for it.
|
||||
uintptr_t poc = upb_Atomic_Load(&a->parent_or_count, memory_order_acquire);
|
||||
while (_upb_Arena_IsTaggedPointer(poc)) {
|
||||
a = _upb_Arena_PointerFromTagged(poc);
|
||||
poc = upb_Atomic_Load(&a->parent_or_count, memory_order_acquire);
|
||||
}
|
||||
return _upb_Arena_RefCountFromTagged(poc);
|
||||
}
|
||||
|
||||
static void upb_Arena_AddBlock(upb_Arena* a, void* ptr, size_t size) {
|
||||
_upb_MemBlock* block = ptr;
|
||||
|
||||
// Insert into linked list.
|
||||
block->size = (uint32_t)size;
|
||||
upb_Atomic_Init(&block->next, a->blocks);
|
||||
upb_Atomic_Store(&a->blocks, block, memory_order_release);
|
||||
|
||||
a->head.ptr = UPB_PTR_AT(block, memblock_reserve, char);
|
||||
a->head.end = UPB_PTR_AT(block, size, char);
|
||||
|
||||
UPB_POISON_MEMORY_REGION(a->head.ptr, a->head.end - a->head.ptr);
|
||||
}
|
||||
|
||||
static bool upb_Arena_AllocBlock(upb_Arena* a, size_t size) {
|
||||
if (!a->block_alloc) return false;
|
||||
_upb_MemBlock* last_block = upb_Atomic_Load(&a->blocks, memory_order_acquire);
|
||||
size_t last_size = last_block != NULL ? last_block->size : 128;
|
||||
size_t block_size = UPB_MAX(size, last_size * 2) + memblock_reserve;
|
||||
_upb_MemBlock* block = upb_malloc(upb_Arena_BlockAlloc(a), block_size);
|
||||
|
||||
if (!block) return false;
|
||||
upb_Arena_AddBlock(a, block, block_size);
|
||||
return true;
|
||||
}
|
||||
|
||||
void* _upb_Arena_SlowMalloc(upb_Arena* a, size_t size) {
|
||||
if (!upb_Arena_AllocBlock(a, size)) return NULL; /* Out of memory. */
|
||||
UPB_ASSERT(_upb_ArenaHas(a) >= size);
|
||||
return upb_Arena_Malloc(a, size);
|
||||
}
|
||||
|
||||
/* Public Arena API ***********************************************************/
|
||||
|
||||
static upb_Arena* upb_Arena_InitSlow(upb_alloc* alloc) {
|
||||
const size_t first_block_overhead = sizeof(upb_Arena) + memblock_reserve;
|
||||
upb_Arena* a;
|
||||
|
||||
/* We need to malloc the initial block. */
|
||||
char* mem;
|
||||
size_t n = first_block_overhead + 256;
|
||||
if (!alloc || !(mem = upb_malloc(alloc, n))) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
a = UPB_PTR_AT(mem, n - sizeof(*a), upb_Arena);
|
||||
n -= sizeof(*a);
|
||||
|
||||
a->block_alloc = upb_Arena_MakeBlockAlloc(alloc, 0);
|
||||
upb_Atomic_Init(&a->parent_or_count, _upb_Arena_TaggedFromRefcount(1));
|
||||
upb_Atomic_Init(&a->next, NULL);
|
||||
upb_Atomic_Init(&a->tail, a);
|
||||
upb_Atomic_Init(&a->blocks, NULL);
|
||||
|
||||
upb_Arena_AddBlock(a, mem, n);
|
||||
|
||||
return a;
|
||||
}
|
||||
|
||||
upb_Arena* upb_Arena_Init(void* mem, size_t n, upb_alloc* alloc) {
|
||||
upb_Arena* a;
|
||||
|
||||
if (n) {
|
||||
/* Align initial pointer up so that we return properly-aligned pointers. */
|
||||
void* aligned = (void*)UPB_ALIGN_UP((uintptr_t)mem, UPB_MALLOC_ALIGN);
|
||||
size_t delta = (uintptr_t)aligned - (uintptr_t)mem;
|
||||
n = delta <= n ? n - delta : 0;
|
||||
mem = aligned;
|
||||
}
|
||||
|
||||
/* Round block size down to alignof(*a) since we will allocate the arena
|
||||
* itself at the end. */
|
||||
n = UPB_ALIGN_DOWN(n, UPB_ALIGN_OF(upb_Arena));
|
||||
|
||||
if (UPB_UNLIKELY(n < sizeof(upb_Arena))) {
|
||||
return upb_Arena_InitSlow(alloc);
|
||||
}
|
||||
|
||||
a = UPB_PTR_AT(mem, n - sizeof(*a), upb_Arena);
|
||||
|
||||
upb_Atomic_Init(&a->parent_or_count, _upb_Arena_TaggedFromRefcount(1));
|
||||
upb_Atomic_Init(&a->next, NULL);
|
||||
upb_Atomic_Init(&a->tail, a);
|
||||
upb_Atomic_Init(&a->blocks, NULL);
|
||||
a->block_alloc = upb_Arena_MakeBlockAlloc(alloc, 1);
|
||||
a->head.ptr = mem;
|
||||
a->head.end = UPB_PTR_AT(mem, n - sizeof(*a), char);
|
||||
|
||||
return a;
|
||||
}
|
||||
|
||||
static void arena_dofree(upb_Arena* a) {
|
||||
UPB_ASSERT(_upb_Arena_RefCountFromTagged(a->parent_or_count) == 1);
|
||||
|
||||
while (a != NULL) {
|
||||
// Load first since arena itself is likely from one of its blocks.
|
||||
upb_Arena* next_arena =
|
||||
(upb_Arena*)upb_Atomic_Load(&a->next, memory_order_acquire);
|
||||
upb_alloc* block_alloc = upb_Arena_BlockAlloc(a);
|
||||
_upb_MemBlock* block = upb_Atomic_Load(&a->blocks, memory_order_acquire);
|
||||
while (block != NULL) {
|
||||
// Load first since we are deleting block.
|
||||
_upb_MemBlock* next_block =
|
||||
upb_Atomic_Load(&block->next, memory_order_acquire);
|
||||
upb_free(block_alloc, block);
|
||||
block = next_block;
|
||||
}
|
||||
a = next_arena;
|
||||
}
|
||||
}
|
||||
|
||||
void upb_Arena_Free(upb_Arena* a) {
|
||||
uintptr_t poc = upb_Atomic_Load(&a->parent_or_count, memory_order_acquire);
|
||||
retry:
|
||||
while (_upb_Arena_IsTaggedPointer(poc)) {
|
||||
a = _upb_Arena_PointerFromTagged(poc);
|
||||
poc = upb_Atomic_Load(&a->parent_or_count, memory_order_acquire);
|
||||
}
|
||||
|
||||
// compare_exchange or fetch_sub are RMW operations, which are more
|
||||
// expensive then direct loads. As an optimization, we only do RMW ops
|
||||
// when we need to update things for other threads to see.
|
||||
if (poc == _upb_Arena_TaggedFromRefcount(1)) {
|
||||
arena_dofree(a);
|
||||
return;
|
||||
}
|
||||
|
||||
if (upb_Atomic_CompareExchangeWeak(
|
||||
&a->parent_or_count, &poc,
|
||||
_upb_Arena_TaggedFromRefcount(_upb_Arena_RefCountFromTagged(poc) - 1),
|
||||
memory_order_release, memory_order_acquire)) {
|
||||
// We were >1 and we decremented it successfully, so we are done.
|
||||
return;
|
||||
}
|
||||
|
||||
// We failed our update, so someone has done something, retry the whole
|
||||
// process, but the failed exchange reloaded `poc` for us.
|
||||
goto retry;
|
||||
}
|
||||
|
||||
static void _upb_Arena_DoFuseArenaLists(upb_Arena* const parent,
|
||||
upb_Arena* child) {
|
||||
upb_Arena* parent_tail = upb_Atomic_Load(&parent->tail, memory_order_relaxed);
|
||||
do {
|
||||
// Our tail might be stale, but it will always converge to the true tail.
|
||||
upb_Arena* parent_tail_next =
|
||||
upb_Atomic_Load(&parent_tail->next, memory_order_relaxed);
|
||||
while (parent_tail_next != NULL) {
|
||||
parent_tail = parent_tail_next;
|
||||
parent_tail_next =
|
||||
upb_Atomic_Load(&parent_tail->next, memory_order_relaxed);
|
||||
}
|
||||
|
||||
upb_Arena* displaced =
|
||||
upb_Atomic_Exchange(&parent_tail->next, child, memory_order_relaxed);
|
||||
parent_tail = upb_Atomic_Load(&child->tail, memory_order_relaxed);
|
||||
|
||||
// If we displaced something that got installed racily, we can simply
|
||||
// reinstall it on our new tail.
|
||||
child = displaced;
|
||||
} while (child != NULL);
|
||||
|
||||
upb_Atomic_Store(&parent->tail, parent_tail, memory_order_relaxed);
|
||||
}
|
||||
|
||||
static upb_Arena* _upb_Arena_DoFuse(upb_Arena* a1, upb_Arena* a2,
|
||||
uintptr_t* ref_delta) {
|
||||
// `parent_or_count` has two disctint modes
|
||||
// - parent pointer mode
|
||||
// - refcount mode
|
||||
//
|
||||
// In parent pointer mode, it may change what pointer it refers to in the
|
||||
// tree, but it will always approach a root. Any operation that walks the
|
||||
// tree to the root may collapse levels of the tree concurrently.
|
||||
_upb_ArenaRoot r1 = _upb_Arena_FindRoot(a1);
|
||||
_upb_ArenaRoot r2 = _upb_Arena_FindRoot(a2);
|
||||
|
||||
if (r1.root == r2.root) return r1.root; // Already fused.
|
||||
|
||||
// Avoid cycles by always fusing into the root with the lower address.
|
||||
if ((uintptr_t)r1.root > (uintptr_t)r2.root) {
|
||||
_upb_ArenaRoot tmp = r1;
|
||||
r1 = r2;
|
||||
r2 = tmp;
|
||||
}
|
||||
|
||||
// The moment we install `r1` as the parent for `r2` all racing frees may
|
||||
// immediately begin decrementing `r1`'s refcount (including pending
|
||||
// increments to that refcount and their frees!). We need to add `r2`'s refs
|
||||
// now, so that `r1` can withstand any unrefs that come from r2.
|
||||
//
|
||||
// Note that while it is possible for `r2`'s refcount to increase
|
||||
// asynchronously, we will not actually do the reparenting operation below
|
||||
// unless `r2`'s refcount is unchanged from when we read it.
|
||||
//
|
||||
// Note that we may have done this previously, either to this node or a
|
||||
// different node, during a previous and failed DoFuse() attempt. But we will
|
||||
// not lose track of these refs because we always add them to our overall
|
||||
// delta.
|
||||
uintptr_t r2_untagged_count = r2.tagged_count & ~1;
|
||||
uintptr_t with_r2_refs = r1.tagged_count + r2_untagged_count;
|
||||
if (!upb_Atomic_CompareExchangeStrong(
|
||||
&r1.root->parent_or_count, &r1.tagged_count, with_r2_refs,
|
||||
memory_order_release, memory_order_acquire)) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// Perform the actual fuse by removing the refs from `r2` and swapping in the
|
||||
// parent pointer.
|
||||
if (!upb_Atomic_CompareExchangeStrong(
|
||||
&r2.root->parent_or_count, &r2.tagged_count,
|
||||
_upb_Arena_TaggedFromPointer(r1.root), memory_order_release,
|
||||
memory_order_acquire)) {
|
||||
// We'll need to remove the excess refs we added to r1 previously.
|
||||
*ref_delta += r2_untagged_count;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// Now that the fuse has been performed (and can no longer fail) we need to
|
||||
// append `r2` to `r1`'s linked list.
|
||||
_upb_Arena_DoFuseArenaLists(r1.root, r2.root);
|
||||
return r1.root;
|
||||
}
|
||||
|
||||
static bool _upb_Arena_FixupRefs(upb_Arena* new_root, uintptr_t ref_delta) {
|
||||
if (ref_delta == 0) return true; // No fixup required.
|
||||
uintptr_t poc =
|
||||
upb_Atomic_Load(&new_root->parent_or_count, memory_order_relaxed);
|
||||
if (_upb_Arena_IsTaggedPointer(poc)) return false;
|
||||
uintptr_t with_refs = poc - ref_delta;
|
||||
UPB_ASSERT(!_upb_Arena_IsTaggedPointer(with_refs));
|
||||
return upb_Atomic_CompareExchangeStrong(&new_root->parent_or_count, &poc,
|
||||
with_refs, memory_order_relaxed,
|
||||
memory_order_relaxed);
|
||||
}
|
||||
|
||||
bool upb_Arena_Fuse(upb_Arena* a1, upb_Arena* a2) {
|
||||
if (a1 == a2) return true; // trivial fuse
|
||||
|
||||
// Do not fuse initial blocks since we cannot lifetime extend them.
|
||||
// Any other fuse scenario is allowed.
|
||||
if (upb_Arena_HasInitialBlock(a1) || upb_Arena_HasInitialBlock(a2)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// The number of refs we ultimately need to transfer to the new root.
|
||||
uintptr_t ref_delta = 0;
|
||||
while (true) {
|
||||
upb_Arena* new_root = _upb_Arena_DoFuse(a1, a2, &ref_delta);
|
||||
if (new_root != NULL && _upb_Arena_FixupRefs(new_root, ref_delta)) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void upb_Arena_IncRefFor(upb_Arena* arena, const void* owner) {
|
||||
_upb_ArenaRoot r;
|
||||
retry:
|
||||
r = _upb_Arena_FindRoot(arena);
|
||||
if (upb_Atomic_CompareExchangeWeak(
|
||||
&r.root->parent_or_count, &r.tagged_count,
|
||||
_upb_Arena_TaggedFromRefcount(
|
||||
_upb_Arena_RefCountFromTagged(r.tagged_count) + 1),
|
||||
memory_order_release, memory_order_acquire)) {
|
||||
// We incremented it successfully, so we are done.
|
||||
return;
|
||||
}
|
||||
// We failed update due to parent switching on the arena.
|
||||
goto retry;
|
||||
}
|
||||
|
||||
void upb_Arena_DecRefFor(upb_Arena* arena, const void* owner) {
|
||||
upb_Arena_Free(arena);
|
||||
}
|
||||
132
Pods/gRPC-Core/third_party/upb/upb/mem/arena.h
generated
vendored
Normal file
132
Pods/gRPC-Core/third_party/upb/upb/mem/arena.h
generated
vendored
Normal file
@@ -0,0 +1,132 @@
|
||||
// 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
|
||||
|
||||
/* upb_Arena is a specific allocator implementation that uses arena allocation.
|
||||
* The user provides an allocator that will be used to allocate the underlying
|
||||
* arena blocks. Arenas by nature do not require the individual allocations
|
||||
* to be freed. However the Arena does allow users to register cleanup
|
||||
* functions that will run when the arena is destroyed.
|
||||
*
|
||||
* A upb_Arena is *not* thread-safe.
|
||||
*
|
||||
* You could write a thread-safe arena allocator that satisfies the
|
||||
* upb_alloc interface, but it would not be as efficient for the
|
||||
* single-threaded case. */
|
||||
|
||||
#ifndef UPB_MEM_ARENA_H_
|
||||
#define UPB_MEM_ARENA_H_
|
||||
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/mem/alloc.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct upb_Arena upb_Arena;
|
||||
|
||||
typedef struct {
|
||||
char *ptr, *end;
|
||||
} _upb_ArenaHead;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Creates an arena from the given initial block (if any -- n may be 0).
|
||||
// Additional blocks will be allocated from |alloc|. If |alloc| is NULL, this
|
||||
// is a fixed-size arena and cannot grow.
|
||||
UPB_API upb_Arena* upb_Arena_Init(void* mem, size_t n, upb_alloc* alloc);
|
||||
|
||||
UPB_API void upb_Arena_Free(upb_Arena* a);
|
||||
UPB_API bool upb_Arena_Fuse(upb_Arena* a, upb_Arena* b);
|
||||
|
||||
void upb_Arena_IncRefFor(upb_Arena* arena, const void* owner);
|
||||
void upb_Arena_DecRefFor(upb_Arena* arena, const void* owner);
|
||||
|
||||
void* _upb_Arena_SlowMalloc(upb_Arena* a, size_t size);
|
||||
size_t upb_Arena_SpaceAllocated(upb_Arena* arena);
|
||||
uint32_t upb_Arena_DebugRefCount(upb_Arena* arena);
|
||||
|
||||
UPB_INLINE size_t _upb_ArenaHas(upb_Arena* a) {
|
||||
_upb_ArenaHead* h = (_upb_ArenaHead*)a;
|
||||
return (size_t)(h->end - h->ptr);
|
||||
}
|
||||
|
||||
UPB_API_INLINE void* upb_Arena_Malloc(upb_Arena* a, size_t size) {
|
||||
size = UPB_ALIGN_MALLOC(size);
|
||||
size_t span = size + UPB_ASAN_GUARD_SIZE;
|
||||
if (UPB_UNLIKELY(_upb_ArenaHas(a) < span)) {
|
||||
return _upb_Arena_SlowMalloc(a, size);
|
||||
}
|
||||
|
||||
// We have enough space to do a fast malloc.
|
||||
_upb_ArenaHead* h = (_upb_ArenaHead*)a;
|
||||
void* ret = h->ptr;
|
||||
UPB_ASSERT(UPB_ALIGN_MALLOC((uintptr_t)ret) == (uintptr_t)ret);
|
||||
UPB_ASSERT(UPB_ALIGN_MALLOC(size) == size);
|
||||
UPB_UNPOISON_MEMORY_REGION(ret, size);
|
||||
|
||||
h->ptr += span;
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
// Shrinks the last alloc from arena.
|
||||
// REQUIRES: (ptr, oldsize) was the last malloc/realloc from this arena.
|
||||
// We could also add a upb_Arena_TryShrinkLast() which is simply a no-op if
|
||||
// this was not the last alloc.
|
||||
UPB_API_INLINE void upb_Arena_ShrinkLast(upb_Arena* a, void* ptr,
|
||||
size_t oldsize, size_t size) {
|
||||
_upb_ArenaHead* h = (_upb_ArenaHead*)a;
|
||||
oldsize = UPB_ALIGN_MALLOC(oldsize);
|
||||
size = UPB_ALIGN_MALLOC(size);
|
||||
// Must be the last alloc.
|
||||
UPB_ASSERT((char*)ptr + oldsize == h->ptr - UPB_ASAN_GUARD_SIZE);
|
||||
UPB_ASSERT(size <= oldsize);
|
||||
h->ptr = (char*)ptr + size;
|
||||
}
|
||||
|
||||
UPB_API_INLINE void* upb_Arena_Realloc(upb_Arena* a, void* ptr, size_t oldsize,
|
||||
size_t size) {
|
||||
_upb_ArenaHead* h = (_upb_ArenaHead*)a;
|
||||
oldsize = UPB_ALIGN_MALLOC(oldsize);
|
||||
size = UPB_ALIGN_MALLOC(size);
|
||||
bool is_most_recent_alloc = (uintptr_t)ptr + oldsize == (uintptr_t)h->ptr;
|
||||
|
||||
if (is_most_recent_alloc) {
|
||||
ptrdiff_t diff = size - oldsize;
|
||||
if ((ptrdiff_t)_upb_ArenaHas(a) >= diff) {
|
||||
h->ptr += diff;
|
||||
return ptr;
|
||||
}
|
||||
} else if (size <= oldsize) {
|
||||
return ptr;
|
||||
}
|
||||
|
||||
void* ret = upb_Arena_Malloc(a, size);
|
||||
|
||||
if (ret && oldsize > 0) {
|
||||
memcpy(ret, ptr, UPB_MIN(oldsize, size));
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_Arena* upb_Arena_New(void) {
|
||||
return upb_Arena_Init(NULL, 0, &upb_alloc_global);
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MEM_ARENA_H_ */
|
||||
54
Pods/gRPC-Core/third_party/upb/upb/mem/arena.hpp
generated
vendored
Normal file
54
Pods/gRPC-Core/third_party/upb/upb/mem/arena.hpp
generated
vendored
Normal file
@@ -0,0 +1,54 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MEM_ARENA_HPP_
|
||||
#define UPB_MEM_ARENA_HPP_
|
||||
|
||||
#include <memory>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
|
||||
namespace upb {
|
||||
|
||||
class Arena {
|
||||
public:
|
||||
// A simple arena with no initial memory block and the default allocator.
|
||||
Arena() : ptr_(upb_Arena_New(), upb_Arena_Free) {}
|
||||
Arena(char* initial_block, size_t size)
|
||||
: ptr_(upb_Arena_Init(initial_block, size, &upb_alloc_global),
|
||||
upb_Arena_Free) {}
|
||||
|
||||
upb_Arena* ptr() const { return ptr_.get(); }
|
||||
|
||||
void Fuse(Arena& other) { upb_Arena_Fuse(ptr(), other.ptr()); }
|
||||
|
||||
protected:
|
||||
std::unique_ptr<upb_Arena, decltype(&upb_Arena_Free)> ptr_;
|
||||
};
|
||||
|
||||
// InlinedArena seeds the arenas with a predefined amount of memory. No
|
||||
// heap memory will be allocated until the initial block is exceeded.
|
||||
template <int N>
|
||||
class InlinedArena : public Arena {
|
||||
public:
|
||||
InlinedArena() : Arena(initial_block_, N) {}
|
||||
~InlinedArena() {
|
||||
// Explicitly destroy the arena now so that it does not outlive
|
||||
// initial_block_.
|
||||
ptr_.reset();
|
||||
}
|
||||
|
||||
private:
|
||||
InlinedArena(const InlinedArena*) = delete;
|
||||
InlinedArena& operator=(const InlinedArena*) = delete;
|
||||
|
||||
char initial_block_[N];
|
||||
};
|
||||
|
||||
} // namespace upb
|
||||
|
||||
#endif // UPB_MEM_ARENA_HPP_
|
||||
94
Pods/gRPC-Core/third_party/upb/upb/mem/internal/arena.h
generated
vendored
Normal file
94
Pods/gRPC-Core/third_party/upb/upb/mem/internal/arena.h
generated
vendored
Normal file
@@ -0,0 +1,94 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MEM_INTERNAL_ARENA_H_
|
||||
#define UPB_MEM_INTERNAL_ARENA_H_
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct _upb_MemBlock _upb_MemBlock;
|
||||
|
||||
struct upb_Arena {
|
||||
_upb_ArenaHead head;
|
||||
|
||||
// upb_alloc* together with a low bit which signals if there is an initial
|
||||
// block.
|
||||
uintptr_t block_alloc;
|
||||
|
||||
// When multiple arenas are fused together, each arena points to a parent
|
||||
// arena (root points to itself). The root tracks how many live arenas
|
||||
// reference it.
|
||||
|
||||
// The low bit is tagged:
|
||||
// 0: pointer to parent
|
||||
// 1: count, left shifted by one
|
||||
UPB_ATOMIC(uintptr_t) parent_or_count;
|
||||
|
||||
// All nodes that are fused together are in a singly-linked list.
|
||||
UPB_ATOMIC(upb_Arena*) next; // NULL at end of list.
|
||||
|
||||
// The last element of the linked list. This is present only as an
|
||||
// optimization, so that we do not have to iterate over all members for every
|
||||
// fuse. Only significant for an arena root. In other cases it is ignored.
|
||||
UPB_ATOMIC(upb_Arena*) tail; // == self when no other list members.
|
||||
|
||||
// Linked list of blocks to free/cleanup. Atomic only for the benefit of
|
||||
// upb_Arena_SpaceAllocated().
|
||||
UPB_ATOMIC(_upb_MemBlock*) blocks;
|
||||
};
|
||||
|
||||
UPB_INLINE bool _upb_Arena_IsTaggedRefcount(uintptr_t parent_or_count) {
|
||||
return (parent_or_count & 1) == 1;
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_Arena_IsTaggedPointer(uintptr_t parent_or_count) {
|
||||
return (parent_or_count & 1) == 0;
|
||||
}
|
||||
|
||||
UPB_INLINE uintptr_t _upb_Arena_RefCountFromTagged(uintptr_t parent_or_count) {
|
||||
UPB_ASSERT(_upb_Arena_IsTaggedRefcount(parent_or_count));
|
||||
return parent_or_count >> 1;
|
||||
}
|
||||
|
||||
UPB_INLINE uintptr_t _upb_Arena_TaggedFromRefcount(uintptr_t refcount) {
|
||||
uintptr_t parent_or_count = (refcount << 1) | 1;
|
||||
UPB_ASSERT(_upb_Arena_IsTaggedRefcount(parent_or_count));
|
||||
return parent_or_count;
|
||||
}
|
||||
|
||||
UPB_INLINE upb_Arena* _upb_Arena_PointerFromTagged(uintptr_t parent_or_count) {
|
||||
UPB_ASSERT(_upb_Arena_IsTaggedPointer(parent_or_count));
|
||||
return (upb_Arena*)parent_or_count;
|
||||
}
|
||||
|
||||
UPB_INLINE uintptr_t _upb_Arena_TaggedFromPointer(upb_Arena* a) {
|
||||
uintptr_t parent_or_count = (uintptr_t)a;
|
||||
UPB_ASSERT(_upb_Arena_IsTaggedPointer(parent_or_count));
|
||||
return parent_or_count;
|
||||
}
|
||||
|
||||
UPB_INLINE upb_alloc* upb_Arena_BlockAlloc(upb_Arena* arena) {
|
||||
return (upb_alloc*)(arena->block_alloc & ~0x1);
|
||||
}
|
||||
|
||||
UPB_INLINE uintptr_t upb_Arena_MakeBlockAlloc(upb_alloc* alloc,
|
||||
bool has_initial) {
|
||||
uintptr_t alloc_uint = (uintptr_t)alloc;
|
||||
UPB_ASSERT((alloc_uint & 1) == 0);
|
||||
return alloc_uint | (has_initial ? 1 : 0);
|
||||
}
|
||||
|
||||
UPB_INLINE bool upb_Arena_HasInitialBlock(upb_Arena* arena) {
|
||||
return arena->block_alloc & 0x1;
|
||||
}
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MEM_INTERNAL_ARENA_H_ */
|
||||
71
Pods/gRPC-Core/third_party/upb/upb/message/accessors.c
generated
vendored
Normal file
71
Pods/gRPC-Core/third_party/upb/upb/message/accessors.c
generated
vendored
Normal file
@@ -0,0 +1,71 @@
|
||||
// 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 "upb/message/array.h"
|
||||
#include "upb/message/internal/array.h"
|
||||
#include "upb/message/map.h"
|
||||
#include "upb/message/message.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
#include "upb/wire/decode.h"
|
||||
#include "upb/wire/encode.h"
|
||||
#include "upb/wire/eps_copy_input_stream.h"
|
||||
#include "upb/wire/reader.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
upb_MapInsertStatus upb_Message_InsertMapEntry(upb_Map* map,
|
||||
const upb_MiniTable* mini_table,
|
||||
const upb_MiniTableField* field,
|
||||
upb_Message* map_entry_message,
|
||||
upb_Arena* arena) {
|
||||
const upb_MiniTable* map_entry_mini_table =
|
||||
mini_table->subs[field->UPB_PRIVATE(submsg_index)].submsg;
|
||||
UPB_ASSERT(map_entry_mini_table);
|
||||
UPB_ASSERT(map_entry_mini_table->field_count == 2);
|
||||
const upb_MiniTableField* map_entry_key_field =
|
||||
&map_entry_mini_table->fields[0];
|
||||
const upb_MiniTableField* map_entry_value_field =
|
||||
&map_entry_mini_table->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_MessageValue map_entry_value;
|
||||
_upb_Message_GetField(map_entry_message, map_entry_key_field, &default_val,
|
||||
&map_entry_key);
|
||||
_upb_Message_GetField(map_entry_message, map_entry_value_field, &default_val,
|
||||
&map_entry_value);
|
||||
return upb_Map_Insert(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;
|
||||
}
|
||||
380
Pods/gRPC-Core/third_party/upb/upb/message/accessors.h
generated
vendored
Normal file
380
Pods/gRPC-Core/third_party/upb/upb/message/accessors.h
generated
vendored
Normal file
@@ -0,0 +1,380 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MESSAGE_ACCESSORS_H_
|
||||
#define UPB_MESSAGE_ACCESSORS_H_
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/message/array.h"
|
||||
#include "upb/message/internal/accessors.h"
|
||||
#include "upb/message/internal/array.h"
|
||||
#include "upb/message/internal/map.h"
|
||||
#include "upb/message/internal/message.h"
|
||||
#include "upb/message/map.h"
|
||||
#include "upb/mini_table/enum.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_API_INLINE void upb_Message_ClearField(upb_Message* msg,
|
||||
const upb_MiniTableField* field) {
|
||||
if (upb_MiniTableField_IsExtension(field)) {
|
||||
const upb_MiniTableExtension* ext = (const upb_MiniTableExtension*)field;
|
||||
_upb_Message_ClearExtensionField(msg, ext);
|
||||
} else {
|
||||
_upb_Message_ClearNonExtensionField(msg, field);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_API_INLINE void upb_Message_Clear(upb_Message* msg,
|
||||
const upb_MiniTable* l) {
|
||||
// Note: Can't use UPB_PTR_AT() here because we are doing pointer subtraction.
|
||||
char* mem = (char*)msg - sizeof(upb_Message_Internal);
|
||||
memset(mem, 0, upb_msg_sizeof(l));
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_HasField(const upb_Message* msg,
|
||||
const upb_MiniTableField* field) {
|
||||
if (upb_MiniTableField_IsExtension(field)) {
|
||||
const upb_MiniTableExtension* ext = (const upb_MiniTableExtension*)field;
|
||||
return _upb_Message_HasExtensionField(msg, ext);
|
||||
} else {
|
||||
return _upb_Message_HasNonExtensionField(msg, field);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_API_INLINE uint32_t upb_Message_WhichOneofFieldNumber(
|
||||
const upb_Message* message, const upb_MiniTableField* oneof_field) {
|
||||
UPB_ASSUME(_upb_MiniTableField_InOneOf(oneof_field));
|
||||
return _upb_getoneofcase_field(message, oneof_field);
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_GetBool(const upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
bool default_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Bool);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_1Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
bool ret;
|
||||
_upb_Message_GetField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_SetBool(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
bool value, upb_Arena* a) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Bool);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_1Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
return _upb_Message_SetField(msg, field, &value, a);
|
||||
}
|
||||
|
||||
UPB_API_INLINE int32_t upb_Message_GetInt32(const upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
int32_t default_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Int32 ||
|
||||
upb_MiniTableField_CType(field) == kUpb_CType_Enum);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_4Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
int32_t ret;
|
||||
_upb_Message_GetField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_SetInt32(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
int32_t value, upb_Arena* a) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Int32 ||
|
||||
upb_MiniTableField_CType(field) == kUpb_CType_Enum);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_4Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
return _upb_Message_SetField(msg, field, &value, a);
|
||||
}
|
||||
|
||||
UPB_API_INLINE uint32_t upb_Message_GetUInt32(const upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
uint32_t default_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_UInt32);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_4Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
uint32_t ret;
|
||||
_upb_Message_GetField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_SetUInt32(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
uint32_t value, upb_Arena* a) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_UInt32);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_4Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
return _upb_Message_SetField(msg, field, &value, a);
|
||||
}
|
||||
|
||||
UPB_API_INLINE void upb_Message_SetClosedEnum(
|
||||
upb_Message* msg, const upb_MiniTable* msg_mini_table,
|
||||
const upb_MiniTableField* field, int32_t value) {
|
||||
UPB_ASSERT(upb_MiniTableField_IsClosedEnum(field));
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_4Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
UPB_ASSERT(upb_MiniTableEnum_CheckValue(
|
||||
upb_MiniTable_GetSubEnumTable(msg_mini_table, field), value));
|
||||
_upb_Message_SetNonExtensionField(msg, field, &value);
|
||||
}
|
||||
|
||||
UPB_API_INLINE int64_t upb_Message_GetInt64(const upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
uint64_t default_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Int64);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_8Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
int64_t ret;
|
||||
_upb_Message_GetField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_SetInt64(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
int64_t value, upb_Arena* a) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Int64);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_8Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
return _upb_Message_SetField(msg, field, &value, a);
|
||||
}
|
||||
|
||||
UPB_API_INLINE uint64_t upb_Message_GetUInt64(const upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
uint64_t default_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_UInt64);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_8Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
uint64_t ret;
|
||||
_upb_Message_GetField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_SetUInt64(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
uint64_t value, upb_Arena* a) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_UInt64);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_8Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
return _upb_Message_SetField(msg, field, &value, a);
|
||||
}
|
||||
|
||||
UPB_API_INLINE float upb_Message_GetFloat(const upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
float default_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Float);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_4Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
float ret;
|
||||
_upb_Message_GetField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_SetFloat(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
float value, upb_Arena* a) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Float);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_4Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
return _upb_Message_SetField(msg, field, &value, a);
|
||||
}
|
||||
|
||||
UPB_API_INLINE double upb_Message_GetDouble(const upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
double default_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Double);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_8Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
double ret;
|
||||
_upb_Message_GetField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_SetDouble(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
double value, upb_Arena* a) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Double);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_8Byte);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
return _upb_Message_SetField(msg, field, &value, a);
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_StringView
|
||||
upb_Message_GetString(const upb_Message* msg, const upb_MiniTableField* field,
|
||||
upb_StringView def_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_String ||
|
||||
upb_MiniTableField_CType(field) == kUpb_CType_Bytes);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_StringView);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
upb_StringView ret;
|
||||
_upb_Message_GetField(msg, field, &def_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_Message_SetString(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
upb_StringView value, upb_Arena* a) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_String ||
|
||||
upb_MiniTableField_CType(field) == kUpb_CType_Bytes);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_StringView);
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
return _upb_Message_SetField(msg, field, &value, a);
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_TaggedMessagePtr upb_Message_GetTaggedMessagePtr(
|
||||
const upb_Message* msg, const upb_MiniTableField* field,
|
||||
upb_Message* default_val) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Message);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) ==
|
||||
UPB_SIZE(kUpb_FieldRep_4Byte, kUpb_FieldRep_8Byte));
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
upb_TaggedMessagePtr tagged;
|
||||
_upb_Message_GetNonExtensionField(msg, field, &default_val, &tagged);
|
||||
return tagged;
|
||||
}
|
||||
|
||||
UPB_API_INLINE const upb_Message* upb_Message_GetMessage(
|
||||
const upb_Message* msg, const upb_MiniTableField* field,
|
||||
upb_Message* default_val) {
|
||||
upb_TaggedMessagePtr tagged =
|
||||
upb_Message_GetTaggedMessagePtr(msg, field, default_val);
|
||||
return upb_TaggedMessagePtr_GetNonEmptyMessage(tagged);
|
||||
}
|
||||
|
||||
// For internal use only; users cannot set tagged messages because only the
|
||||
// parser and the message copier are allowed to directly create an empty
|
||||
// message.
|
||||
UPB_API_INLINE void _upb_Message_SetTaggedMessagePtr(
|
||||
upb_Message* msg, const upb_MiniTable* mini_table,
|
||||
const upb_MiniTableField* field, upb_TaggedMessagePtr sub_message) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Message);
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) ==
|
||||
UPB_SIZE(kUpb_FieldRep_4Byte, kUpb_FieldRep_8Byte));
|
||||
UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
UPB_ASSERT(mini_table->subs[field->UPB_PRIVATE(submsg_index)].submsg);
|
||||
_upb_Message_SetNonExtensionField(msg, field, &sub_message);
|
||||
}
|
||||
|
||||
UPB_API_INLINE void upb_Message_SetMessage(upb_Message* msg,
|
||||
const upb_MiniTable* mini_table,
|
||||
const upb_MiniTableField* field,
|
||||
upb_Message* sub_message) {
|
||||
_upb_Message_SetTaggedMessagePtr(
|
||||
msg, mini_table, field, _upb_TaggedMessagePtr_Pack(sub_message, false));
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_Message* upb_Message_GetOrCreateMutableMessage(
|
||||
upb_Message* msg, const upb_MiniTable* mini_table,
|
||||
const upb_MiniTableField* field, upb_Arena* arena) {
|
||||
UPB_ASSERT(arena);
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Message);
|
||||
upb_Message* sub_message = *UPB_PTR_AT(msg, field->offset, upb_Message*);
|
||||
if (!sub_message) {
|
||||
const upb_MiniTable* sub_mini_table =
|
||||
mini_table->subs[field->UPB_PRIVATE(submsg_index)].submsg;
|
||||
UPB_ASSERT(sub_mini_table);
|
||||
sub_message = _upb_Message_New(sub_mini_table, arena);
|
||||
*UPB_PTR_AT(msg, field->offset, upb_Message*) = sub_message;
|
||||
_upb_Message_SetPresence(msg, field);
|
||||
}
|
||||
return sub_message;
|
||||
}
|
||||
|
||||
UPB_API_INLINE const upb_Array* upb_Message_GetArray(
|
||||
const upb_Message* msg, const upb_MiniTableField* field) {
|
||||
_upb_MiniTableField_CheckIsArray(field);
|
||||
upb_Array* ret;
|
||||
const upb_Array* default_val = NULL;
|
||||
_upb_Message_GetNonExtensionField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_Array* upb_Message_GetMutableArray(
|
||||
upb_Message* msg, const upb_MiniTableField* field) {
|
||||
_upb_MiniTableField_CheckIsArray(field);
|
||||
return (upb_Array*)upb_Message_GetArray(msg, field);
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_Array* upb_Message_GetOrCreateMutableArray(
|
||||
upb_Message* msg, const upb_MiniTableField* field, upb_Arena* arena) {
|
||||
UPB_ASSERT(arena);
|
||||
_upb_MiniTableField_CheckIsArray(field);
|
||||
upb_Array* array = upb_Message_GetMutableArray(msg, field);
|
||||
if (!array) {
|
||||
array = _upb_Array_New(arena, 4, _upb_MiniTable_ElementSizeLg2(field));
|
||||
// Check again due to: https://godbolt.org/z/7WfaoKG1r
|
||||
_upb_MiniTableField_CheckIsArray(field);
|
||||
_upb_Message_SetField(msg, field, &array, arena);
|
||||
}
|
||||
return array;
|
||||
}
|
||||
|
||||
UPB_API_INLINE void* upb_Message_ResizeArrayUninitialized(
|
||||
upb_Message* msg, const upb_MiniTableField* field, size_t size,
|
||||
upb_Arena* arena) {
|
||||
_upb_MiniTableField_CheckIsArray(field);
|
||||
upb_Array* arr = upb_Message_GetOrCreateMutableArray(msg, field, arena);
|
||||
if (!arr || !_upb_Array_ResizeUninitialized(arr, size, arena)) return NULL;
|
||||
return _upb_array_ptr(arr);
|
||||
}
|
||||
|
||||
UPB_API_INLINE const upb_Map* upb_Message_GetMap(
|
||||
const upb_Message* msg, const upb_MiniTableField* field) {
|
||||
_upb_MiniTableField_CheckIsMap(field);
|
||||
_upb_Message_AssertMapIsUntagged(msg, field);
|
||||
upb_Map* ret;
|
||||
const upb_Map* default_val = NULL;
|
||||
_upb_Message_GetNonExtensionField(msg, field, &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_Map* upb_Message_GetMutableMap(
|
||||
upb_Message* msg, const upb_MiniTableField* field) {
|
||||
return (upb_Map*)upb_Message_GetMap(msg, field);
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_Map* upb_Message_GetOrCreateMutableMap(
|
||||
upb_Message* msg, const upb_MiniTable* map_entry_mini_table,
|
||||
const upb_MiniTableField* field, upb_Arena* arena) {
|
||||
UPB_ASSUME(upb_MiniTableField_CType(field) == kUpb_CType_Message);
|
||||
const upb_MiniTableField* map_entry_key_field =
|
||||
&map_entry_mini_table->fields[0];
|
||||
const upb_MiniTableField* map_entry_value_field =
|
||||
&map_entry_mini_table->fields[1];
|
||||
return _upb_Message_GetOrCreateMutableMap(
|
||||
msg, field,
|
||||
_upb_Map_CTypeSize(upb_MiniTableField_CType(map_entry_key_field)),
|
||||
_upb_Map_CTypeSize(upb_MiniTableField_CType(map_entry_value_field)),
|
||||
arena);
|
||||
}
|
||||
|
||||
// Updates a map entry given an entry message.
|
||||
upb_MapInsertStatus upb_Message_InsertMapEntry(upb_Map* map,
|
||||
const upb_MiniTable* mini_table,
|
||||
const upb_MiniTableField* field,
|
||||
upb_Message* map_entry_message,
|
||||
upb_Arena* arena);
|
||||
|
||||
// Compares two messages by serializing them and calling memcmp().
|
||||
bool upb_Message_IsExactlyEqual(const upb_Message* m1, const upb_Message* m2,
|
||||
const upb_MiniTable* layout);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_MESSAGE_ACCESSORS_H_
|
||||
154
Pods/gRPC-Core/third_party/upb/upb/message/array.c
generated
vendored
Normal file
154
Pods/gRPC-Core/third_party/upb/upb/message/array.c
generated
vendored
Normal file
@@ -0,0 +1,154 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#include "upb/message/internal/array.h"
|
||||
|
||||
#include <string.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
const char _upb_Array_CTypeSizeLg2Table[] = {
|
||||
[kUpb_CType_Bool] = 0,
|
||||
[kUpb_CType_Float] = 2,
|
||||
[kUpb_CType_Int32] = 2,
|
||||
[kUpb_CType_UInt32] = 2,
|
||||
[kUpb_CType_Enum] = 2,
|
||||
[kUpb_CType_Message] = UPB_SIZE(2, 3),
|
||||
[kUpb_CType_Double] = 3,
|
||||
[kUpb_CType_Int64] = 3,
|
||||
[kUpb_CType_UInt64] = 3,
|
||||
[kUpb_CType_String] = UPB_SIZE(3, 4),
|
||||
[kUpb_CType_Bytes] = UPB_SIZE(3, 4),
|
||||
};
|
||||
|
||||
upb_Array* upb_Array_New(upb_Arena* a, upb_CType type) {
|
||||
return _upb_Array_New(a, 4, _upb_Array_CTypeSizeLg2(type));
|
||||
}
|
||||
|
||||
const void* upb_Array_DataPtr(const upb_Array* arr) {
|
||||
return _upb_array_ptr((upb_Array*)arr);
|
||||
}
|
||||
|
||||
void* upb_Array_MutableDataPtr(upb_Array* arr) { return _upb_array_ptr(arr); }
|
||||
|
||||
size_t upb_Array_Size(const upb_Array* arr) { return arr->size; }
|
||||
|
||||
upb_MessageValue upb_Array_Get(const upb_Array* arr, size_t i) {
|
||||
upb_MessageValue ret;
|
||||
const char* data = _upb_array_constptr(arr);
|
||||
int lg2 = arr->data & 7;
|
||||
UPB_ASSERT(i < arr->size);
|
||||
memcpy(&ret, data + (i << lg2), 1 << lg2);
|
||||
return ret;
|
||||
}
|
||||
|
||||
void upb_Array_Set(upb_Array* arr, size_t i, upb_MessageValue val) {
|
||||
char* data = _upb_array_ptr(arr);
|
||||
int lg2 = arr->data & 7;
|
||||
UPB_ASSERT(i < arr->size);
|
||||
memcpy(data + (i << lg2), &val, 1 << lg2);
|
||||
}
|
||||
|
||||
bool upb_Array_Append(upb_Array* arr, upb_MessageValue val, upb_Arena* arena) {
|
||||
UPB_ASSERT(arena);
|
||||
if (!upb_Array_Resize(arr, arr->size + 1, arena)) {
|
||||
return false;
|
||||
}
|
||||
upb_Array_Set(arr, arr->size - 1, val);
|
||||
return true;
|
||||
}
|
||||
|
||||
void upb_Array_Move(upb_Array* arr, size_t dst_idx, size_t src_idx,
|
||||
size_t count) {
|
||||
const int lg2 = arr->data & 7;
|
||||
char* data = _upb_array_ptr(arr);
|
||||
memmove(&data[dst_idx << lg2], &data[src_idx << lg2], count << lg2);
|
||||
}
|
||||
|
||||
bool upb_Array_Insert(upb_Array* arr, size_t i, size_t count,
|
||||
upb_Arena* arena) {
|
||||
UPB_ASSERT(arena);
|
||||
UPB_ASSERT(i <= arr->size);
|
||||
UPB_ASSERT(count + arr->size >= count);
|
||||
const size_t oldsize = arr->size;
|
||||
if (!upb_Array_Resize(arr, arr->size + count, arena)) {
|
||||
return false;
|
||||
}
|
||||
upb_Array_Move(arr, i + count, i, oldsize - i);
|
||||
return true;
|
||||
}
|
||||
|
||||
/*
|
||||
* i end arr->size
|
||||
* |------------|XXXXXXXX|--------|
|
||||
*/
|
||||
void upb_Array_Delete(upb_Array* arr, size_t i, size_t count) {
|
||||
const size_t end = i + count;
|
||||
UPB_ASSERT(i <= end);
|
||||
UPB_ASSERT(end <= arr->size);
|
||||
upb_Array_Move(arr, i, end, arr->size - end);
|
||||
arr->size -= count;
|
||||
}
|
||||
|
||||
bool upb_Array_Resize(upb_Array* arr, size_t size, upb_Arena* arena) {
|
||||
const size_t oldsize = arr->size;
|
||||
if (UPB_UNLIKELY(!_upb_Array_ResizeUninitialized(arr, size, arena))) {
|
||||
return false;
|
||||
}
|
||||
const size_t newsize = arr->size;
|
||||
if (newsize > oldsize) {
|
||||
const int lg2 = arr->data & 7;
|
||||
char* data = _upb_array_ptr(arr);
|
||||
memset(data + (oldsize << lg2), 0, (newsize - oldsize) << lg2);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
// EVERYTHING BELOW THIS LINE IS INTERNAL - DO NOT USE /////////////////////////
|
||||
|
||||
bool _upb_array_realloc(upb_Array* arr, size_t min_capacity, upb_Arena* arena) {
|
||||
size_t new_capacity = UPB_MAX(arr->capacity, 4);
|
||||
int elem_size_lg2 = arr->data & 7;
|
||||
size_t old_bytes = arr->capacity << elem_size_lg2;
|
||||
size_t new_bytes;
|
||||
void* ptr = _upb_array_ptr(arr);
|
||||
|
||||
// Log2 ceiling of size.
|
||||
while (new_capacity < min_capacity) new_capacity *= 2;
|
||||
|
||||
new_bytes = new_capacity << elem_size_lg2;
|
||||
ptr = upb_Arena_Realloc(arena, ptr, old_bytes, new_bytes);
|
||||
if (!ptr) return false;
|
||||
|
||||
arr->data = _upb_tag_arrptr(ptr, elem_size_lg2);
|
||||
arr->capacity = new_capacity;
|
||||
return true;
|
||||
}
|
||||
98
Pods/gRPC-Core/third_party/upb/upb/message/array.h
generated
vendored
Normal file
98
Pods/gRPC-Core/third_party/upb/upb/message/array.h
generated
vendored
Normal file
@@ -0,0 +1,98 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef UPB_MESSAGE_ARRAY_H_
|
||||
#define UPB_MESSAGE_ARRAY_H_
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/value.h" // IWYU pragma: export
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct upb_Array upb_Array;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Creates a new array on the given arena that holds elements of this type.
|
||||
UPB_API upb_Array* upb_Array_New(upb_Arena* a, upb_CType type);
|
||||
|
||||
// Returns the number of elements in the array.
|
||||
UPB_API size_t upb_Array_Size(const upb_Array* arr);
|
||||
|
||||
// Returns the given element, which must be within the array's current size.
|
||||
UPB_API upb_MessageValue upb_Array_Get(const upb_Array* arr, size_t i);
|
||||
|
||||
// Sets the given element, which must be within the array's current size.
|
||||
UPB_API void upb_Array_Set(upb_Array* arr, size_t i, upb_MessageValue val);
|
||||
|
||||
// Appends an element to the array. Returns false on allocation failure.
|
||||
UPB_API bool upb_Array_Append(upb_Array* array, upb_MessageValue val,
|
||||
upb_Arena* arena);
|
||||
|
||||
// Moves elements within the array using memmove().
|
||||
// Like memmove(), the source and destination elements may be overlapping.
|
||||
UPB_API void upb_Array_Move(upb_Array* array, size_t dst_idx, size_t src_idx,
|
||||
size_t count);
|
||||
|
||||
// Inserts one or more empty elements into the array.
|
||||
// Existing elements are shifted right.
|
||||
// The new elements have undefined state and must be set with `upb_Array_Set()`.
|
||||
// REQUIRES: `i <= upb_Array_Size(arr)`
|
||||
UPB_API bool upb_Array_Insert(upb_Array* array, size_t i, size_t count,
|
||||
upb_Arena* arena);
|
||||
|
||||
// Deletes one or more elements from the array.
|
||||
// Existing elements are shifted left.
|
||||
// REQUIRES: `i + count <= upb_Array_Size(arr)`
|
||||
UPB_API void upb_Array_Delete(upb_Array* array, size_t i, size_t count);
|
||||
|
||||
// Changes the size of a vector. New elements are initialized to NULL/0.
|
||||
// Returns false on allocation failure.
|
||||
UPB_API bool upb_Array_Resize(upb_Array* array, size_t size, upb_Arena* arena);
|
||||
|
||||
// Returns pointer to array data.
|
||||
UPB_API const void* upb_Array_DataPtr(const upb_Array* arr);
|
||||
|
||||
// Returns mutable pointer to array data.
|
||||
UPB_API void* upb_Array_MutableDataPtr(upb_Array* arr);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MESSAGE_ARRAY_H_ */
|
||||
367
Pods/gRPC-Core/third_party/upb/upb/message/internal/accessors.h
generated
vendored
Normal file
367
Pods/gRPC-Core/third_party/upb/upb/message/internal/accessors.h
generated
vendored
Normal file
@@ -0,0 +1,367 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MESSAGE_INTERNAL_ACCESSORS_H_
|
||||
#define UPB_MESSAGE_INTERNAL_ACCESSORS_H_
|
||||
|
||||
#include "upb/message/internal/extension.h"
|
||||
#include "upb/message/internal/map.h"
|
||||
#include "upb/message/internal/message.h"
|
||||
#include "upb/mini_table/internal/field.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#if defined(__GNUC__) && !defined(__clang__)
|
||||
// GCC raises incorrect warnings in these functions. It thinks that we are
|
||||
// overrunning buffers, but we carefully write the functions in this file to
|
||||
// guarantee that this is impossible. GCC gets this wrong due it its failure
|
||||
// to perform constant propagation as we expect:
|
||||
// - https://gcc.gnu.org/bugzilla/show_bug.cgi?id=108217
|
||||
// - https://gcc.gnu.org/bugzilla/show_bug.cgi?id=108226
|
||||
//
|
||||
// Unfortunately this also indicates that GCC is not optimizing away the
|
||||
// switch() in cases where it should be, compromising the performance.
|
||||
#pragma GCC diagnostic push
|
||||
#pragma GCC diagnostic ignored "-Warray-bounds"
|
||||
#pragma GCC diagnostic ignored "-Wstringop-overflow"
|
||||
#if __GNUC__ >= 11
|
||||
#pragma GCC diagnostic ignored "-Wstringop-overread"
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// LINT.IfChange(presence_logic)
|
||||
|
||||
// Hasbit access ///////////////////////////////////////////////////////////////
|
||||
|
||||
UPB_INLINE size_t _upb_hasbit_ofs(size_t idx) { return idx / 8; }
|
||||
|
||||
UPB_INLINE char _upb_hasbit_mask(size_t idx) { return 1 << (idx % 8); }
|
||||
|
||||
UPB_INLINE bool _upb_hasbit(const upb_Message* msg, size_t idx) {
|
||||
return (*UPB_PTR_AT(msg, _upb_hasbit_ofs(idx), const char) &
|
||||
_upb_hasbit_mask(idx)) != 0;
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_sethas(const upb_Message* msg, size_t idx) {
|
||||
(*UPB_PTR_AT(msg, _upb_hasbit_ofs(idx), char)) |= _upb_hasbit_mask(idx);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_clearhas(const upb_Message* msg, size_t idx) {
|
||||
(*UPB_PTR_AT(msg, _upb_hasbit_ofs(idx), char)) &= ~_upb_hasbit_mask(idx);
|
||||
}
|
||||
|
||||
UPB_INLINE size_t _upb_Message_Hasidx(const upb_MiniTableField* f) {
|
||||
UPB_ASSERT(f->presence > 0);
|
||||
return f->presence;
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_hasbit_field(const upb_Message* msg,
|
||||
const upb_MiniTableField* f) {
|
||||
return _upb_hasbit(msg, _upb_Message_Hasidx(f));
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_sethas_field(const upb_Message* msg,
|
||||
const upb_MiniTableField* f) {
|
||||
_upb_sethas(msg, _upb_Message_Hasidx(f));
|
||||
}
|
||||
|
||||
// Oneof case access ///////////////////////////////////////////////////////////
|
||||
|
||||
UPB_INLINE size_t _upb_oneofcase_ofs(const upb_MiniTableField* f) {
|
||||
UPB_ASSERT(f->presence < 0);
|
||||
return ~(ptrdiff_t)f->presence;
|
||||
}
|
||||
|
||||
UPB_INLINE uint32_t* _upb_oneofcase_field(upb_Message* msg,
|
||||
const upb_MiniTableField* f) {
|
||||
return UPB_PTR_AT(msg, _upb_oneofcase_ofs(f), uint32_t);
|
||||
}
|
||||
|
||||
UPB_INLINE uint32_t _upb_getoneofcase_field(const upb_Message* msg,
|
||||
const upb_MiniTableField* f) {
|
||||
return *_upb_oneofcase_field((upb_Message*)msg, f);
|
||||
}
|
||||
|
||||
// LINT.ThenChange(GoogleInternalName2)
|
||||
|
||||
UPB_INLINE bool _upb_MiniTableField_InOneOf(const upb_MiniTableField* field) {
|
||||
return field->presence < 0;
|
||||
}
|
||||
|
||||
UPB_INLINE void* _upb_MiniTableField_GetPtr(upb_Message* msg,
|
||||
const upb_MiniTableField* field) {
|
||||
return (char*)msg + field->offset;
|
||||
}
|
||||
|
||||
UPB_INLINE const void* _upb_MiniTableField_GetConstPtr(
|
||||
const upb_Message* msg, const upb_MiniTableField* field) {
|
||||
return (char*)msg + field->offset;
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_Message_SetPresence(upb_Message* msg,
|
||||
const upb_MiniTableField* field) {
|
||||
if (field->presence > 0) {
|
||||
_upb_sethas_field(msg, field);
|
||||
} else if (_upb_MiniTableField_InOneOf(field)) {
|
||||
*_upb_oneofcase_field(msg, field) = field->number;
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_MiniTable_ValueIsNonZero(const void* default_val,
|
||||
const upb_MiniTableField* field) {
|
||||
char zero[16] = {0};
|
||||
switch (_upb_MiniTableField_GetRep(field)) {
|
||||
case kUpb_FieldRep_1Byte:
|
||||
return memcmp(&zero, default_val, 1) != 0;
|
||||
case kUpb_FieldRep_4Byte:
|
||||
return memcmp(&zero, default_val, 4) != 0;
|
||||
case kUpb_FieldRep_8Byte:
|
||||
return memcmp(&zero, default_val, 8) != 0;
|
||||
case kUpb_FieldRep_StringView: {
|
||||
const upb_StringView* sv = (const upb_StringView*)default_val;
|
||||
return sv->size != 0;
|
||||
}
|
||||
}
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_MiniTable_CopyFieldData(void* to, const void* from,
|
||||
const upb_MiniTableField* field) {
|
||||
switch (_upb_MiniTableField_GetRep(field)) {
|
||||
case kUpb_FieldRep_1Byte:
|
||||
memcpy(to, from, 1);
|
||||
return;
|
||||
case kUpb_FieldRep_4Byte:
|
||||
memcpy(to, from, 4);
|
||||
return;
|
||||
case kUpb_FieldRep_8Byte:
|
||||
memcpy(to, from, 8);
|
||||
return;
|
||||
case kUpb_FieldRep_StringView: {
|
||||
memcpy(to, from, sizeof(upb_StringView));
|
||||
return;
|
||||
}
|
||||
}
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
UPB_INLINE size_t
|
||||
_upb_MiniTable_ElementSizeLg2(const upb_MiniTableField* field) {
|
||||
const unsigned char table[] = {
|
||||
0,
|
||||
3, // kUpb_FieldType_Double = 1,
|
||||
2, // kUpb_FieldType_Float = 2,
|
||||
3, // kUpb_FieldType_Int64 = 3,
|
||||
3, // kUpb_FieldType_UInt64 = 4,
|
||||
2, // kUpb_FieldType_Int32 = 5,
|
||||
3, // kUpb_FieldType_Fixed64 = 6,
|
||||
2, // kUpb_FieldType_Fixed32 = 7,
|
||||
0, // kUpb_FieldType_Bool = 8,
|
||||
UPB_SIZE(3, 4), // kUpb_FieldType_String = 9,
|
||||
UPB_SIZE(2, 3), // kUpb_FieldType_Group = 10,
|
||||
UPB_SIZE(2, 3), // kUpb_FieldType_Message = 11,
|
||||
UPB_SIZE(3, 4), // kUpb_FieldType_Bytes = 12,
|
||||
2, // kUpb_FieldType_UInt32 = 13,
|
||||
2, // kUpb_FieldType_Enum = 14,
|
||||
2, // kUpb_FieldType_SFixed32 = 15,
|
||||
3, // kUpb_FieldType_SFixed64 = 16,
|
||||
2, // kUpb_FieldType_SInt32 = 17,
|
||||
3, // kUpb_FieldType_SInt64 = 18,
|
||||
};
|
||||
return table[field->UPB_PRIVATE(descriptortype)];
|
||||
}
|
||||
|
||||
// Here we define universal getter/setter functions for message fields.
|
||||
// These look very branchy and inefficient, but as long as the MiniTableField
|
||||
// values are known at compile time, all the branches are optimized away and
|
||||
// we are left with ideal code. This can happen either through through
|
||||
// literals or UPB_ASSUME():
|
||||
//
|
||||
// // Via struct literals.
|
||||
// bool FooMessage_set_bool_field(const upb_Message* msg, bool val) {
|
||||
// const upb_MiniTableField field = {1, 0, 0, /* etc... */};
|
||||
// // All value in "field" are compile-time known.
|
||||
// _upb_Message_SetNonExtensionField(msg, &field, &value);
|
||||
// }
|
||||
//
|
||||
// // Via UPB_ASSUME().
|
||||
// UPB_INLINE bool upb_Message_SetBool(upb_Message* msg,
|
||||
// const upb_MiniTableField* field,
|
||||
// bool value, upb_Arena* a) {
|
||||
// UPB_ASSUME(field->UPB_PRIVATE(descriptortype) == kUpb_FieldType_Bool);
|
||||
// UPB_ASSUME(!upb_IsRepeatedOrMap(field));
|
||||
// UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_1Byte);
|
||||
// _upb_Message_SetField(msg, field, &value, a);
|
||||
// }
|
||||
//
|
||||
// As a result, we can use these universal getters/setters for *all* message
|
||||
// accessors: generated code, MiniTable accessors, and reflection. The only
|
||||
// exception is the binary encoder/decoder, which need to be a bit more clever
|
||||
// about how they read/write the message data, for efficiency.
|
||||
//
|
||||
// These functions work on both extensions and non-extensions. If the field
|
||||
// of a setter is known to be a non-extension, the arena may be NULL and the
|
||||
// returned bool value may be ignored since it will always succeed.
|
||||
|
||||
UPB_INLINE bool _upb_Message_HasExtensionField(
|
||||
const upb_Message* msg, const upb_MiniTableExtension* ext) {
|
||||
UPB_ASSERT(upb_MiniTableField_HasPresence(&ext->field));
|
||||
return _upb_Message_Getext(msg, ext) != NULL;
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_Message_HasNonExtensionField(
|
||||
const upb_Message* msg, const upb_MiniTableField* field) {
|
||||
UPB_ASSERT(upb_MiniTableField_HasPresence(field));
|
||||
UPB_ASSUME(!upb_MiniTableField_IsExtension(field));
|
||||
if (_upb_MiniTableField_InOneOf(field)) {
|
||||
return _upb_getoneofcase_field(msg, field) == field->number;
|
||||
} else {
|
||||
return _upb_hasbit_field(msg, field);
|
||||
}
|
||||
}
|
||||
|
||||
static UPB_FORCEINLINE void _upb_Message_GetNonExtensionField(
|
||||
const upb_Message* msg, const upb_MiniTableField* field,
|
||||
const void* default_val, void* val) {
|
||||
UPB_ASSUME(!upb_MiniTableField_IsExtension(field));
|
||||
if ((_upb_MiniTableField_InOneOf(field) ||
|
||||
_upb_MiniTable_ValueIsNonZero(default_val, field)) &&
|
||||
!_upb_Message_HasNonExtensionField(msg, field)) {
|
||||
_upb_MiniTable_CopyFieldData(val, default_val, field);
|
||||
return;
|
||||
}
|
||||
_upb_MiniTable_CopyFieldData(val, _upb_MiniTableField_GetConstPtr(msg, field),
|
||||
field);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_Message_GetExtensionField(
|
||||
const upb_Message* msg, const upb_MiniTableExtension* mt_ext,
|
||||
const void* default_val, void* val) {
|
||||
UPB_ASSUME(upb_MiniTableField_IsExtension(&mt_ext->field));
|
||||
const upb_Message_Extension* ext = _upb_Message_Getext(msg, mt_ext);
|
||||
if (ext) {
|
||||
_upb_MiniTable_CopyFieldData(val, &ext->data, &mt_ext->field);
|
||||
} else {
|
||||
_upb_MiniTable_CopyFieldData(val, default_val, &mt_ext->field);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_Message_GetField(const upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
const void* default_val, void* val) {
|
||||
if (upb_MiniTableField_IsExtension(field)) {
|
||||
_upb_Message_GetExtensionField(msg, (upb_MiniTableExtension*)field,
|
||||
default_val, val);
|
||||
} else {
|
||||
_upb_Message_GetNonExtensionField(msg, field, default_val, val);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_Message_SetNonExtensionField(
|
||||
upb_Message* msg, const upb_MiniTableField* field, const void* val) {
|
||||
UPB_ASSUME(!upb_MiniTableField_IsExtension(field));
|
||||
_upb_Message_SetPresence(msg, field);
|
||||
_upb_MiniTable_CopyFieldData(_upb_MiniTableField_GetPtr(msg, field), val,
|
||||
field);
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_Message_SetExtensionField(
|
||||
upb_Message* msg, const upb_MiniTableExtension* mt_ext, const void* val,
|
||||
upb_Arena* a) {
|
||||
UPB_ASSERT(a);
|
||||
upb_Message_Extension* ext =
|
||||
_upb_Message_GetOrCreateExtension(msg, mt_ext, a);
|
||||
if (!ext) return false;
|
||||
_upb_MiniTable_CopyFieldData(&ext->data, val, &mt_ext->field);
|
||||
return true;
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_Message_SetField(upb_Message* msg,
|
||||
const upb_MiniTableField* field,
|
||||
const void* val, upb_Arena* a) {
|
||||
if (upb_MiniTableField_IsExtension(field)) {
|
||||
const upb_MiniTableExtension* ext = (const upb_MiniTableExtension*)field;
|
||||
return _upb_Message_SetExtensionField(msg, ext, val, a);
|
||||
} else {
|
||||
_upb_Message_SetNonExtensionField(msg, field, val);
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_Message_ClearExtensionField(
|
||||
upb_Message* msg, const upb_MiniTableExtension* ext_l) {
|
||||
upb_Message_Internal* in = upb_Message_Getinternal(msg);
|
||||
if (!in->internal) return;
|
||||
const upb_Message_Extension* base =
|
||||
UPB_PTR_AT(in->internal, in->internal->ext_begin, upb_Message_Extension);
|
||||
upb_Message_Extension* ext =
|
||||
(upb_Message_Extension*)_upb_Message_Getext(msg, ext_l);
|
||||
if (ext) {
|
||||
*ext = *base;
|
||||
in->internal->ext_begin += sizeof(upb_Message_Extension);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_Message_ClearNonExtensionField(
|
||||
upb_Message* msg, const upb_MiniTableField* field) {
|
||||
if (field->presence > 0) {
|
||||
_upb_clearhas(msg, _upb_Message_Hasidx(field));
|
||||
} else if (_upb_MiniTableField_InOneOf(field)) {
|
||||
uint32_t* oneof_case = _upb_oneofcase_field(msg, field);
|
||||
if (*oneof_case != field->number) return;
|
||||
*oneof_case = 0;
|
||||
}
|
||||
const char zeros[16] = {0};
|
||||
_upb_MiniTable_CopyFieldData(_upb_MiniTableField_GetPtr(msg, field), zeros,
|
||||
field);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_Message_AssertMapIsUntagged(
|
||||
const upb_Message* msg, const upb_MiniTableField* field) {
|
||||
UPB_UNUSED(msg);
|
||||
_upb_MiniTableField_CheckIsMap(field);
|
||||
#ifndef NDEBUG
|
||||
upb_TaggedMessagePtr default_val = 0;
|
||||
upb_TaggedMessagePtr tagged;
|
||||
_upb_Message_GetNonExtensionField(msg, field, &default_val, &tagged);
|
||||
UPB_ASSERT(!upb_TaggedMessagePtr_IsEmpty(tagged));
|
||||
#endif
|
||||
}
|
||||
|
||||
UPB_INLINE upb_Map* _upb_Message_GetOrCreateMutableMap(
|
||||
upb_Message* msg, const upb_MiniTableField* field, size_t key_size,
|
||||
size_t val_size, upb_Arena* arena) {
|
||||
_upb_MiniTableField_CheckIsMap(field);
|
||||
_upb_Message_AssertMapIsUntagged(msg, field);
|
||||
upb_Map* map = NULL;
|
||||
upb_Map* default_map_value = NULL;
|
||||
_upb_Message_GetNonExtensionField(msg, field, &default_map_value, &map);
|
||||
if (!map) {
|
||||
map = _upb_Map_New(arena, key_size, val_size);
|
||||
// Check again due to: https://godbolt.org/z/7WfaoKG1r
|
||||
_upb_MiniTableField_CheckIsMap(field);
|
||||
_upb_Message_SetNonExtensionField(msg, field, &map);
|
||||
}
|
||||
return map;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#if defined(__GNUC__) && !defined(__clang__)
|
||||
#pragma GCC diagnostic pop
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_MESSAGE_INTERNAL_ACCESSORS_H_
|
||||
138
Pods/gRPC-Core/third_party/upb/upb/message/internal/array.h
generated
vendored
Normal file
138
Pods/gRPC-Core/third_party/upb/upb/message/internal/array.h
generated
vendored
Normal file
@@ -0,0 +1,138 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef UPB_MESSAGE_INTERNAL_ARRAY_H_
|
||||
#define UPB_MESSAGE_INTERNAL_ARRAY_H_
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/message/array.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// LINT.IfChange(struct_definition)
|
||||
// Our internal representation for repeated fields.
|
||||
struct upb_Array {
|
||||
uintptr_t data; /* Tagged ptr: low 3 bits of ptr are lg2(elem size). */
|
||||
size_t size; /* The number of elements in the array. */
|
||||
size_t capacity; /* Allocated storage. Measured in elements. */
|
||||
};
|
||||
// LINT.ThenChange(GoogleInternalName1)
|
||||
|
||||
UPB_INLINE size_t _upb_Array_ElementSizeLg2(const upb_Array* arr) {
|
||||
size_t ret = arr->data & 7;
|
||||
UPB_ASSERT(ret <= 4);
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_INLINE const void* _upb_array_constptr(const upb_Array* arr) {
|
||||
_upb_Array_ElementSizeLg2(arr); // Check assertion.
|
||||
return (void*)(arr->data & ~(uintptr_t)7);
|
||||
}
|
||||
|
||||
UPB_INLINE uintptr_t _upb_array_tagptr(void* ptr, int elem_size_lg2) {
|
||||
UPB_ASSERT(elem_size_lg2 <= 4);
|
||||
return (uintptr_t)ptr | elem_size_lg2;
|
||||
}
|
||||
|
||||
UPB_INLINE void* _upb_array_ptr(upb_Array* arr) {
|
||||
return (void*)_upb_array_constptr(arr);
|
||||
}
|
||||
|
||||
UPB_INLINE uintptr_t _upb_tag_arrptr(void* ptr, int elem_size_lg2) {
|
||||
UPB_ASSERT(elem_size_lg2 <= 4);
|
||||
UPB_ASSERT(((uintptr_t)ptr & 7) == 0);
|
||||
return (uintptr_t)ptr | (unsigned)elem_size_lg2;
|
||||
}
|
||||
|
||||
extern const char _upb_Array_CTypeSizeLg2Table[];
|
||||
|
||||
UPB_INLINE size_t _upb_Array_CTypeSizeLg2(upb_CType ctype) {
|
||||
return _upb_Array_CTypeSizeLg2Table[ctype];
|
||||
}
|
||||
|
||||
UPB_INLINE upb_Array* _upb_Array_New(upb_Arena* a, size_t init_capacity,
|
||||
int elem_size_lg2) {
|
||||
UPB_ASSERT(elem_size_lg2 <= 4);
|
||||
const size_t arr_size = UPB_ALIGN_UP(sizeof(upb_Array), UPB_MALLOC_ALIGN);
|
||||
const size_t bytes = arr_size + (init_capacity << elem_size_lg2);
|
||||
upb_Array* arr = (upb_Array*)upb_Arena_Malloc(a, bytes);
|
||||
if (!arr) return NULL;
|
||||
arr->data = _upb_tag_arrptr(UPB_PTR_AT(arr, arr_size, void), elem_size_lg2);
|
||||
arr->size = 0;
|
||||
arr->capacity = init_capacity;
|
||||
return arr;
|
||||
}
|
||||
|
||||
// Resizes the capacity of the array to be at least min_size.
|
||||
bool _upb_array_realloc(upb_Array* arr, size_t min_size, upb_Arena* arena);
|
||||
|
||||
UPB_INLINE bool _upb_array_reserve(upb_Array* arr, size_t size,
|
||||
upb_Arena* arena) {
|
||||
if (arr->capacity < size) return _upb_array_realloc(arr, size, arena);
|
||||
return true;
|
||||
}
|
||||
|
||||
// Resize without initializing new elements.
|
||||
UPB_INLINE bool _upb_Array_ResizeUninitialized(upb_Array* arr, size_t size,
|
||||
upb_Arena* arena) {
|
||||
UPB_ASSERT(size <= arr->size || arena); // Allow NULL arena when shrinking.
|
||||
if (!_upb_array_reserve(arr, size, arena)) return false;
|
||||
arr->size = size;
|
||||
return true;
|
||||
}
|
||||
|
||||
// This function is intended for situations where elem_size is compile-time
|
||||
// constant or a known expression of the form (1 << lg2), so that the expression
|
||||
// i*elem_size does not result in an actual multiplication.
|
||||
UPB_INLINE void _upb_Array_Set(upb_Array* arr, size_t i, const void* data,
|
||||
size_t elem_size) {
|
||||
UPB_ASSERT(i < arr->size);
|
||||
UPB_ASSERT(elem_size == 1U << _upb_Array_ElementSizeLg2(arr));
|
||||
char* arr_data = (char*)_upb_array_ptr(arr);
|
||||
memcpy(arr_data + (i * elem_size), data, elem_size);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_array_detach(const void* msg, size_t ofs) {
|
||||
*UPB_PTR_AT(msg, ofs, upb_Array*) = NULL;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MESSAGE_INTERNAL_ARRAY_H_ */
|
||||
63
Pods/gRPC-Core/third_party/upb/upb/message/internal/extension.h
generated
vendored
Normal file
63
Pods/gRPC-Core/third_party/upb/upb/message/internal/extension.h
generated
vendored
Normal file
@@ -0,0 +1,63 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MESSAGE_INTERNAL_EXTENSION_H_
|
||||
#define UPB_MESSAGE_INTERNAL_EXTENSION_H_
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/message.h"
|
||||
#include "upb/mini_table/extension.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// The internal representation of an extension is self-describing: it contains
|
||||
// enough information that we can serialize it to binary format without needing
|
||||
// to look it up in a upb_ExtensionRegistry.
|
||||
//
|
||||
// This representation allocates 16 bytes to data on 64-bit platforms.
|
||||
// This is rather wasteful for scalars (in the extreme case of bool,
|
||||
// it wastes 15 bytes). We accept this because we expect messages to be
|
||||
// the most common extension type.
|
||||
typedef struct {
|
||||
const upb_MiniTableExtension* ext;
|
||||
union {
|
||||
upb_StringView str;
|
||||
void* ptr;
|
||||
char scalar_data[8];
|
||||
} data;
|
||||
} upb_Message_Extension;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Adds the given extension data to the given message.
|
||||
// |ext| is copied into the message instance.
|
||||
// This logically replaces any previously-added extension with this number.
|
||||
upb_Message_Extension* _upb_Message_GetOrCreateExtension(
|
||||
upb_Message* msg, const upb_MiniTableExtension* ext, upb_Arena* arena);
|
||||
|
||||
// Returns an array of extensions for this message.
|
||||
// Note: the array is ordered in reverse relative to the order of creation.
|
||||
const upb_Message_Extension* _upb_Message_Getexts(const upb_Message* msg,
|
||||
size_t* count);
|
||||
|
||||
// Returns an extension for the given field number, or NULL if no extension
|
||||
// exists for this field number.
|
||||
const upb_Message_Extension* _upb_Message_Getext(
|
||||
const upb_Message* msg, const upb_MiniTableExtension* ext);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MESSAGE_INTERNAL_EXTENSION_H_ */
|
||||
173
Pods/gRPC-Core/third_party/upb/upb/message/internal/map.h
generated
vendored
Normal file
173
Pods/gRPC-Core/third_party/upb/upb/message/internal/map.h
generated
vendored
Normal file
@@ -0,0 +1,173 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
// EVERYTHING BELOW THIS LINE IS INTERNAL - DO NOT USE /////////////////////////
|
||||
|
||||
#ifndef UPB_COLLECTIONS_INTERNAL_MAP_H_
|
||||
#define UPB_COLLECTIONS_INTERNAL_MAP_H_
|
||||
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/hash/str_table.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/map.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_Map {
|
||||
// Size of key and val, based on the map type.
|
||||
// Strings are represented as '0' because they must be handled specially.
|
||||
char key_size;
|
||||
char val_size;
|
||||
|
||||
upb_strtable table;
|
||||
};
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Converting between internal table representation and user values.
|
||||
//
|
||||
// _upb_map_tokey() and _upb_map_fromkey() are inverses.
|
||||
// _upb_map_tovalue() and _upb_map_fromvalue() are inverses.
|
||||
//
|
||||
// These functions account for the fact that strings are treated differently
|
||||
// from other types when stored in a map.
|
||||
|
||||
UPB_INLINE upb_StringView _upb_map_tokey(const void* key, size_t size) {
|
||||
if (size == UPB_MAPTYPE_STRING) {
|
||||
return *(upb_StringView*)key;
|
||||
} else {
|
||||
return upb_StringView_FromDataAndSize((const char*)key, size);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_map_fromkey(upb_StringView key, void* out, size_t size) {
|
||||
if (size == UPB_MAPTYPE_STRING) {
|
||||
memcpy(out, &key, sizeof(key));
|
||||
} else {
|
||||
memcpy(out, key.data, size);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_map_tovalue(const void* val, size_t size,
|
||||
upb_value* msgval, upb_Arena* a) {
|
||||
if (size == UPB_MAPTYPE_STRING) {
|
||||
upb_StringView* strp = (upb_StringView*)upb_Arena_Malloc(a, sizeof(*strp));
|
||||
if (!strp) return false;
|
||||
*strp = *(upb_StringView*)val;
|
||||
*msgval = upb_value_ptr(strp);
|
||||
} else {
|
||||
memcpy(msgval, val, size);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_map_fromvalue(upb_value val, void* out, size_t size) {
|
||||
if (size == UPB_MAPTYPE_STRING) {
|
||||
const upb_StringView* strp = (const upb_StringView*)upb_value_getptr(val);
|
||||
memcpy(out, strp, sizeof(upb_StringView));
|
||||
} else {
|
||||
memcpy(out, &val, size);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE void* _upb_map_next(const upb_Map* map, size_t* iter) {
|
||||
upb_strtable_iter it;
|
||||
it.t = &map->table;
|
||||
it.index = *iter;
|
||||
upb_strtable_next(&it);
|
||||
*iter = it.index;
|
||||
if (upb_strtable_done(&it)) return NULL;
|
||||
return (void*)str_tabent(&it);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_Map_Clear(upb_Map* map) {
|
||||
upb_strtable_clear(&map->table);
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_Map_Delete(upb_Map* map, const void* key, size_t key_size,
|
||||
upb_value* val) {
|
||||
upb_StringView k = _upb_map_tokey(key, key_size);
|
||||
return upb_strtable_remove2(&map->table, k.data, k.size, val);
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_Map_Get(const upb_Map* map, const void* key,
|
||||
size_t key_size, void* val, size_t val_size) {
|
||||
upb_value tabval;
|
||||
upb_StringView k = _upb_map_tokey(key, key_size);
|
||||
bool ret = upb_strtable_lookup2(&map->table, k.data, k.size, &tabval);
|
||||
if (ret && val) {
|
||||
_upb_map_fromvalue(tabval, val, val_size);
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_INLINE upb_MapInsertStatus _upb_Map_Insert(upb_Map* map, const void* key,
|
||||
size_t key_size, void* val,
|
||||
size_t val_size, upb_Arena* a) {
|
||||
upb_StringView strkey = _upb_map_tokey(key, key_size);
|
||||
upb_value tabval = {0};
|
||||
if (!_upb_map_tovalue(val, val_size, &tabval, a)) {
|
||||
return kUpb_MapInsertStatus_OutOfMemory;
|
||||
}
|
||||
|
||||
// TODO: add overwrite operation to minimize number of lookups.
|
||||
bool removed =
|
||||
upb_strtable_remove2(&map->table, strkey.data, strkey.size, NULL);
|
||||
if (!upb_strtable_insert(&map->table, strkey.data, strkey.size, tabval, a)) {
|
||||
return kUpb_MapInsertStatus_OutOfMemory;
|
||||
}
|
||||
return removed ? kUpb_MapInsertStatus_Replaced
|
||||
: kUpb_MapInsertStatus_Inserted;
|
||||
}
|
||||
|
||||
UPB_INLINE size_t _upb_Map_Size(const upb_Map* map) {
|
||||
return map->table.t.count;
|
||||
}
|
||||
|
||||
// Strings/bytes are special-cased in maps.
|
||||
extern char _upb_Map_CTypeSizeTable[12];
|
||||
|
||||
UPB_INLINE size_t _upb_Map_CTypeSize(upb_CType ctype) {
|
||||
return _upb_Map_CTypeSizeTable[ctype];
|
||||
}
|
||||
|
||||
// Creates a new map on the given arena with this key/value type.
|
||||
upb_Map* _upb_Map_New(upb_Arena* a, size_t key_size, size_t value_size);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_COLLECTIONS_INTERNAL_MAP_H_ */
|
||||
66
Pods/gRPC-Core/third_party/upb/upb/message/internal/map_entry.h
generated
vendored
Normal file
66
Pods/gRPC-Core/third_party/upb/upb/message/internal/map_entry.h
generated
vendored
Normal file
@@ -0,0 +1,66 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef UPB_COLLECTIONS_INTERNAL_MAP_ENTRY_H_
|
||||
#define UPB_COLLECTIONS_INTERNAL_MAP_ENTRY_H_
|
||||
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/hash/common.h"
|
||||
#include "upb/message/internal/types.h"
|
||||
|
||||
// Map entries aren't actually stored for map fields, they are only used during
|
||||
// parsing. For parsing, it helps a lot if all map entry messages have the same
|
||||
// layout. The layout code in mini_table/decode.c will ensure that all map
|
||||
// entries have this layout.
|
||||
//
|
||||
// Note that users can and do create map entries directly, which will also use
|
||||
// this layout.
|
||||
//
|
||||
// NOTE: sync with wire/decode.c.
|
||||
typedef struct {
|
||||
// We only need 2 hasbits max, but due to alignment we'll use 8 bytes here,
|
||||
// and the uint64_t helps make this clear.
|
||||
uint64_t hasbits;
|
||||
union {
|
||||
upb_StringView str; // For str/bytes.
|
||||
upb_value val; // For all other types.
|
||||
} k;
|
||||
union {
|
||||
upb_StringView str; // For str/bytes.
|
||||
upb_value val; // For all other types.
|
||||
} v;
|
||||
} upb_MapEntryData;
|
||||
|
||||
typedef struct {
|
||||
upb_Message_Internal internal;
|
||||
upb_MapEntryData data;
|
||||
} upb_MapEntry;
|
||||
|
||||
#endif // UPB_COLLECTIONS_INTERNAL_MAP_ENTRY_H_
|
||||
112
Pods/gRPC-Core/third_party/upb/upb/message/internal/map_sorter.h
generated
vendored
Normal file
112
Pods/gRPC-Core/third_party/upb/upb/message/internal/map_sorter.h
generated
vendored
Normal file
@@ -0,0 +1,112 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
// EVERYTHING BELOW THIS LINE IS INTERNAL - DO NOT USE /////////////////////////
|
||||
|
||||
#ifndef UPB_COLLECTIONS_INTERNAL_MAP_SORTER_H_
|
||||
#define UPB_COLLECTIONS_INTERNAL_MAP_SORTER_H_
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "upb/message/internal/extension.h"
|
||||
#include "upb/message/internal/map.h"
|
||||
#include "upb/message/internal/map_entry.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// _upb_mapsorter sorts maps and provides ordered iteration over the entries.
|
||||
// Since maps can be recursive (map values can be messages which contain other
|
||||
// maps), _upb_mapsorter can contain a stack of maps.
|
||||
|
||||
typedef struct {
|
||||
void const** entries;
|
||||
int size;
|
||||
int cap;
|
||||
} _upb_mapsorter;
|
||||
|
||||
typedef struct {
|
||||
int start;
|
||||
int pos;
|
||||
int end;
|
||||
} _upb_sortedmap;
|
||||
|
||||
UPB_INLINE void _upb_mapsorter_init(_upb_mapsorter* s) {
|
||||
s->entries = NULL;
|
||||
s->size = 0;
|
||||
s->cap = 0;
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_mapsorter_destroy(_upb_mapsorter* s) {
|
||||
if (s->entries) free(s->entries);
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_sortedmap_next(_upb_mapsorter* s, const upb_Map* map,
|
||||
_upb_sortedmap* sorted, upb_MapEntry* ent) {
|
||||
if (sorted->pos == sorted->end) return false;
|
||||
const upb_tabent* tabent = (const upb_tabent*)s->entries[sorted->pos++];
|
||||
upb_StringView key = upb_tabstrview(tabent->key);
|
||||
_upb_map_fromkey(key, &ent->data.k, map->key_size);
|
||||
upb_value val = {tabent->val.val};
|
||||
_upb_map_fromvalue(val, &ent->data.v, map->val_size);
|
||||
return true;
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_sortedmap_nextext(_upb_mapsorter* s,
|
||||
_upb_sortedmap* sorted,
|
||||
const upb_Message_Extension** ext) {
|
||||
if (sorted->pos == sorted->end) return false;
|
||||
*ext = (const upb_Message_Extension*)s->entries[sorted->pos++];
|
||||
return true;
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_mapsorter_popmap(_upb_mapsorter* s,
|
||||
_upb_sortedmap* sorted) {
|
||||
s->size = sorted->start;
|
||||
}
|
||||
|
||||
bool _upb_mapsorter_pushmap(_upb_mapsorter* s, upb_FieldType key_type,
|
||||
const upb_Map* map, _upb_sortedmap* sorted);
|
||||
|
||||
bool _upb_mapsorter_pushexts(_upb_mapsorter* s,
|
||||
const upb_Message_Extension* exts, size_t count,
|
||||
_upb_sortedmap* sorted);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_COLLECTIONS_INTERNAL_MAP_SORTER_H_ */
|
||||
106
Pods/gRPC-Core/third_party/upb/upb/message/internal/message.h
generated
vendored
Normal file
106
Pods/gRPC-Core/third_party/upb/upb/message/internal/message.h
generated
vendored
Normal file
@@ -0,0 +1,106 @@
|
||||
// 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
|
||||
|
||||
/*
|
||||
** Our memory representation for parsing tables and messages themselves.
|
||||
** Functions in this file are used by generated code and possibly reflection.
|
||||
**
|
||||
** The definitions in this file are internal to upb.
|
||||
**/
|
||||
|
||||
#ifndef UPB_MESSAGE_INTERNAL_H_
|
||||
#define UPB_MESSAGE_INTERNAL_H_
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/message/internal/extension.h"
|
||||
#include "upb/message/internal/types.h"
|
||||
#include "upb/message/message.h"
|
||||
#include "upb/mini_table/extension.h"
|
||||
#include "upb/mini_table/extension_registry.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
extern const float kUpb_FltInfinity;
|
||||
extern const double kUpb_Infinity;
|
||||
extern const double kUpb_NaN;
|
||||
|
||||
/* Internal members of a upb_Message that track unknown fields and/or
|
||||
* extensions. We can change this without breaking binary compatibility. We put
|
||||
* these before the user's data. The user's upb_Message* points after the
|
||||
* upb_Message_Internal. */
|
||||
|
||||
struct upb_Message_InternalData {
|
||||
/* Total size of this structure, including the data that follows.
|
||||
* Must be aligned to 8, which is alignof(upb_Message_Extension) */
|
||||
uint32_t size;
|
||||
|
||||
/* Offsets relative to the beginning of this structure.
|
||||
*
|
||||
* Unknown data grows forward from the beginning to unknown_end.
|
||||
* Extension data grows backward from size to ext_begin.
|
||||
* When the two meet, we're out of data and have to realloc.
|
||||
*
|
||||
* If we imagine that the final member of this struct is:
|
||||
* char data[size - overhead]; // overhead =
|
||||
* sizeof(upb_Message_InternalData)
|
||||
*
|
||||
* Then we have:
|
||||
* unknown data: data[0 .. (unknown_end - overhead)]
|
||||
* extensions data: data[(ext_begin - overhead) .. (size - overhead)] */
|
||||
uint32_t unknown_end;
|
||||
uint32_t ext_begin;
|
||||
/* Data follows, as if there were an array:
|
||||
* char data[size - sizeof(upb_Message_InternalData)]; */
|
||||
};
|
||||
|
||||
/* Maps upb_CType -> memory size. */
|
||||
extern char _upb_CTypeo_size[12];
|
||||
|
||||
UPB_INLINE size_t upb_msg_sizeof(const upb_MiniTable* t) {
|
||||
return t->size + sizeof(upb_Message_Internal);
|
||||
}
|
||||
|
||||
// Inline version upb_Message_New(), for internal use.
|
||||
UPB_INLINE upb_Message* _upb_Message_New(const upb_MiniTable* mini_table,
|
||||
upb_Arena* arena) {
|
||||
size_t size = upb_msg_sizeof(mini_table);
|
||||
void* mem = upb_Arena_Malloc(arena, size + sizeof(upb_Message_Internal));
|
||||
if (UPB_UNLIKELY(!mem)) return NULL;
|
||||
upb_Message* msg = UPB_PTR_AT(mem, sizeof(upb_Message_Internal), upb_Message);
|
||||
memset(mem, 0, size);
|
||||
return msg;
|
||||
}
|
||||
|
||||
UPB_INLINE upb_Message_Internal* upb_Message_Getinternal(
|
||||
const upb_Message* msg) {
|
||||
ptrdiff_t size = sizeof(upb_Message_Internal);
|
||||
return (upb_Message_Internal*)((char*)msg - size);
|
||||
}
|
||||
|
||||
// Discards the unknown fields for this message only.
|
||||
void _upb_Message_DiscardUnknown_shallow(upb_Message* msg);
|
||||
|
||||
// Adds unknown data (serialized protobuf data) to the given message.
|
||||
// The data is copied into the message instance.
|
||||
bool _upb_Message_AddUnknown(upb_Message* msg, const char* data, size_t len,
|
||||
upb_Arena* arena);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MESSAGE_INTERNAL_H_ */
|
||||
23
Pods/gRPC-Core/third_party/upb/upb/message/internal/types.h
generated
vendored
Normal file
23
Pods/gRPC-Core/third_party/upb/upb/message/internal/types.h
generated
vendored
Normal file
@@ -0,0 +1,23 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_INTERNAL_TYPES_H_
|
||||
#define UPB_MINI_TABLE_INTERNAL_TYPES_H_
|
||||
|
||||
typedef struct upb_Message_InternalData upb_Message_InternalData;
|
||||
|
||||
typedef struct {
|
||||
union {
|
||||
upb_Message_InternalData* internal;
|
||||
|
||||
// Force 8-byte alignment, since the data members may contain members that
|
||||
// require 8-byte alignment.
|
||||
double d;
|
||||
};
|
||||
} upb_Message_Internal;
|
||||
|
||||
#endif // UPB_MINI_TABLE_INTERNAL_TYPES_H_
|
||||
145
Pods/gRPC-Core/third_party/upb/upb/message/map.c
generated
vendored
Normal file
145
Pods/gRPC-Core/third_party/upb/upb/message/map.c
generated
vendored
Normal file
@@ -0,0 +1,145 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#include "upb/message/map.h"
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/internal/map.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// Strings/bytes are special-cased in maps.
|
||||
char _upb_Map_CTypeSizeTable[12] = {
|
||||
[kUpb_CType_Bool] = 1,
|
||||
[kUpb_CType_Float] = 4,
|
||||
[kUpb_CType_Int32] = 4,
|
||||
[kUpb_CType_UInt32] = 4,
|
||||
[kUpb_CType_Enum] = 4,
|
||||
[kUpb_CType_Message] = sizeof(void*),
|
||||
[kUpb_CType_Double] = 8,
|
||||
[kUpb_CType_Int64] = 8,
|
||||
[kUpb_CType_UInt64] = 8,
|
||||
[kUpb_CType_String] = UPB_MAPTYPE_STRING,
|
||||
[kUpb_CType_Bytes] = UPB_MAPTYPE_STRING,
|
||||
};
|
||||
|
||||
upb_Map* upb_Map_New(upb_Arena* a, upb_CType key_type, upb_CType value_type) {
|
||||
return _upb_Map_New(a, _upb_Map_CTypeSize(key_type),
|
||||
_upb_Map_CTypeSize(value_type));
|
||||
}
|
||||
|
||||
size_t upb_Map_Size(const upb_Map* map) { return _upb_Map_Size(map); }
|
||||
|
||||
bool upb_Map_Get(const upb_Map* map, upb_MessageValue key,
|
||||
upb_MessageValue* val) {
|
||||
return _upb_Map_Get(map, &key, map->key_size, val, map->val_size);
|
||||
}
|
||||
|
||||
void upb_Map_Clear(upb_Map* map) { _upb_Map_Clear(map); }
|
||||
|
||||
upb_MapInsertStatus upb_Map_Insert(upb_Map* map, upb_MessageValue key,
|
||||
upb_MessageValue val, upb_Arena* arena) {
|
||||
UPB_ASSERT(arena);
|
||||
return (upb_MapInsertStatus)_upb_Map_Insert(map, &key, map->key_size, &val,
|
||||
map->val_size, arena);
|
||||
}
|
||||
|
||||
bool upb_Map_Delete(upb_Map* map, upb_MessageValue key, upb_MessageValue* val) {
|
||||
upb_value v;
|
||||
const bool removed = _upb_Map_Delete(map, &key, map->key_size, &v);
|
||||
if (val) _upb_map_fromvalue(v, val, map->val_size);
|
||||
return removed;
|
||||
}
|
||||
|
||||
bool upb_Map_Next(const upb_Map* map, upb_MessageValue* key,
|
||||
upb_MessageValue* val, size_t* iter) {
|
||||
upb_StringView k;
|
||||
upb_value v;
|
||||
const bool ok = upb_strtable_next2(&map->table, &k, &v, (intptr_t*)iter);
|
||||
if (ok) {
|
||||
_upb_map_fromkey(k, key, map->key_size);
|
||||
_upb_map_fromvalue(v, val, map->val_size);
|
||||
}
|
||||
return ok;
|
||||
}
|
||||
|
||||
UPB_API void upb_Map_SetEntryValue(upb_Map* map, size_t iter,
|
||||
upb_MessageValue val) {
|
||||
upb_value v;
|
||||
_upb_map_tovalue(&val, map->val_size, &v, NULL);
|
||||
upb_strtable_setentryvalue(&map->table, iter, v);
|
||||
}
|
||||
|
||||
bool upb_MapIterator_Next(const upb_Map* map, size_t* iter) {
|
||||
return _upb_map_next(map, iter);
|
||||
}
|
||||
|
||||
bool upb_MapIterator_Done(const upb_Map* map, size_t iter) {
|
||||
upb_strtable_iter i;
|
||||
UPB_ASSERT(iter != kUpb_Map_Begin);
|
||||
i.t = &map->table;
|
||||
i.index = iter;
|
||||
return upb_strtable_done(&i);
|
||||
}
|
||||
|
||||
// Returns the key and value for this entry of the map.
|
||||
upb_MessageValue upb_MapIterator_Key(const upb_Map* map, size_t iter) {
|
||||
upb_strtable_iter i;
|
||||
upb_MessageValue ret;
|
||||
i.t = &map->table;
|
||||
i.index = iter;
|
||||
_upb_map_fromkey(upb_strtable_iter_key(&i), &ret, map->key_size);
|
||||
return ret;
|
||||
}
|
||||
|
||||
upb_MessageValue upb_MapIterator_Value(const upb_Map* map, size_t iter) {
|
||||
upb_strtable_iter i;
|
||||
upb_MessageValue ret;
|
||||
i.t = &map->table;
|
||||
i.index = iter;
|
||||
_upb_map_fromvalue(upb_strtable_iter_value(&i), &ret, map->val_size);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// EVERYTHING BELOW THIS LINE IS INTERNAL - DO NOT USE /////////////////////////
|
||||
|
||||
upb_Map* _upb_Map_New(upb_Arena* a, size_t key_size, size_t value_size) {
|
||||
upb_Map* map = upb_Arena_Malloc(a, sizeof(upb_Map));
|
||||
if (!map) return NULL;
|
||||
|
||||
upb_strtable_init(&map->table, 4, a);
|
||||
map->key_size = key_size;
|
||||
map->val_size = value_size;
|
||||
|
||||
return map;
|
||||
}
|
||||
147
Pods/gRPC-Core/third_party/upb/upb/message/map.h
generated
vendored
Normal file
147
Pods/gRPC-Core/third_party/upb/upb/message/map.h
generated
vendored
Normal file
@@ -0,0 +1,147 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef UPB_MESSAGE_MAP_H_
|
||||
#define UPB_MESSAGE_MAP_H_
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/value.h" // IWYU pragma: export
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct upb_Map upb_Map;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Creates a new map on the given arena with the given key/value size.
|
||||
UPB_API upb_Map* upb_Map_New(upb_Arena* a, upb_CType key_type,
|
||||
upb_CType value_type);
|
||||
|
||||
// Returns the number of entries in the map.
|
||||
UPB_API size_t upb_Map_Size(const upb_Map* map);
|
||||
|
||||
// Stores a value for the given key into |*val| (or the zero value if the key is
|
||||
// not present). Returns whether the key was present. The |val| pointer may be
|
||||
// NULL, in which case the function tests whether the given key is present.
|
||||
UPB_API bool upb_Map_Get(const upb_Map* map, upb_MessageValue key,
|
||||
upb_MessageValue* val);
|
||||
|
||||
// Removes all entries in the map.
|
||||
UPB_API void upb_Map_Clear(upb_Map* map);
|
||||
|
||||
typedef enum {
|
||||
kUpb_MapInsertStatus_Inserted = 0,
|
||||
kUpb_MapInsertStatus_Replaced = 1,
|
||||
kUpb_MapInsertStatus_OutOfMemory = 2,
|
||||
} upb_MapInsertStatus;
|
||||
|
||||
// Sets the given key to the given value, returning whether the key was inserted
|
||||
// or replaced. If the key was inserted, then any existing iterators will be
|
||||
// invalidated.
|
||||
UPB_API upb_MapInsertStatus upb_Map_Insert(upb_Map* map, upb_MessageValue key,
|
||||
upb_MessageValue val,
|
||||
upb_Arena* arena);
|
||||
|
||||
// Sets the given key to the given value. Returns false if memory allocation
|
||||
// failed. If the key is newly inserted, then any existing iterators will be
|
||||
// invalidated.
|
||||
UPB_API_INLINE bool upb_Map_Set(upb_Map* map, upb_MessageValue key,
|
||||
upb_MessageValue val, upb_Arena* arena) {
|
||||
return upb_Map_Insert(map, key, val, arena) !=
|
||||
kUpb_MapInsertStatus_OutOfMemory;
|
||||
}
|
||||
|
||||
// Deletes this key from the table. Returns true if the key was present.
|
||||
// If present and |val| is non-NULL, stores the deleted value.
|
||||
UPB_API bool upb_Map_Delete(upb_Map* map, upb_MessageValue key,
|
||||
upb_MessageValue* val);
|
||||
|
||||
// (DEPRECATED and going away soon. Do not use.)
|
||||
UPB_INLINE bool upb_Map_Delete2(upb_Map* map, upb_MessageValue key,
|
||||
upb_MessageValue* val) {
|
||||
return upb_Map_Delete(map, key, val);
|
||||
}
|
||||
|
||||
// Map iteration:
|
||||
//
|
||||
// size_t iter = kUpb_Map_Begin;
|
||||
// upb_MessageValue key, val;
|
||||
// while (upb_Map_Next(map, &key, &val, &iter)) {
|
||||
// ...
|
||||
// }
|
||||
|
||||
#define kUpb_Map_Begin ((size_t)-1)
|
||||
|
||||
// Advances to the next entry. Returns false if no more entries are present.
|
||||
// Otherwise returns true and populates both *key and *value.
|
||||
UPB_API bool upb_Map_Next(const upb_Map* map, upb_MessageValue* key,
|
||||
upb_MessageValue* val, size_t* iter);
|
||||
|
||||
// Sets the value for the entry pointed to by iter.
|
||||
// WARNING: this does not currently work for string values!
|
||||
UPB_API void upb_Map_SetEntryValue(upb_Map* map, size_t iter,
|
||||
upb_MessageValue val);
|
||||
|
||||
// DEPRECATED iterator, slated for removal.
|
||||
|
||||
/* Map iteration:
|
||||
*
|
||||
* size_t iter = kUpb_Map_Begin;
|
||||
* while (upb_MapIterator_Next(map, &iter)) {
|
||||
* upb_MessageValue key = upb_MapIterator_Key(map, iter);
|
||||
* upb_MessageValue val = upb_MapIterator_Value(map, iter);
|
||||
* }
|
||||
*/
|
||||
|
||||
// Advances to the next entry. Returns false if no more entries are present.
|
||||
UPB_API bool upb_MapIterator_Next(const upb_Map* map, size_t* iter);
|
||||
|
||||
// Returns true if the iterator still points to a valid entry, or false if the
|
||||
// iterator is past the last element. It is an error to call this function with
|
||||
// kUpb_Map_Begin (you must call next() at least once first).
|
||||
UPB_API bool upb_MapIterator_Done(const upb_Map* map, size_t iter);
|
||||
|
||||
// Returns the key and value for this entry of the map.
|
||||
UPB_API upb_MessageValue upb_MapIterator_Key(const upb_Map* map, size_t iter);
|
||||
UPB_API upb_MessageValue upb_MapIterator_Value(const upb_Map* map, size_t iter);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MESSAGE_MAP_H_ */
|
||||
81
Pods/gRPC-Core/third_party/upb/upb/message/map_gencode_util.h
generated
vendored
Normal file
81
Pods/gRPC-Core/third_party/upb/upb/message/map_gencode_util.h
generated
vendored
Normal file
@@ -0,0 +1,81 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
// These functions are only used by generated code.
|
||||
|
||||
#ifndef UPB_MESSAGE_MAP_GENCODE_UTIL_H_
|
||||
#define UPB_MESSAGE_MAP_GENCODE_UTIL_H_
|
||||
|
||||
#include "upb/message/internal/map.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Message map operations, these get the map from the message first.
|
||||
|
||||
UPB_INLINE void _upb_msg_map_key(const void* msg, void* key, size_t size) {
|
||||
const upb_tabent* ent = (const upb_tabent*)msg;
|
||||
uint32_t u32len;
|
||||
upb_StringView k;
|
||||
k.data = upb_tabstr(ent->key, &u32len);
|
||||
k.size = u32len;
|
||||
_upb_map_fromkey(k, key, size);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_msg_map_value(const void* msg, void* val, size_t size) {
|
||||
const upb_tabent* ent = (const upb_tabent*)msg;
|
||||
upb_value v = {ent->val.val};
|
||||
_upb_map_fromvalue(v, val, size);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_msg_map_set_value(void* msg, const void* val,
|
||||
size_t size) {
|
||||
upb_tabent* ent = (upb_tabent*)msg;
|
||||
// This is like _upb_map_tovalue() except the entry already exists
|
||||
// so we can reuse the allocated upb_StringView for string fields.
|
||||
if (size == UPB_MAPTYPE_STRING) {
|
||||
upb_StringView* strp = (upb_StringView*)(uintptr_t)ent->val.val;
|
||||
memcpy(strp, val, sizeof(*strp));
|
||||
} else {
|
||||
memcpy(&ent->val.val, val, size);
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MESSAGE_MAP_GENCODE_UTIL_H_ */
|
||||
170
Pods/gRPC-Core/third_party/upb/upb/message/map_sorter.c
generated
vendored
Normal file
170
Pods/gRPC-Core/third_party/upb/upb/message/map_sorter.c
generated
vendored
Normal file
@@ -0,0 +1,170 @@
|
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#include "upb/message/internal/map_sorter.h"
|
||||
|
||||
#include "upb/base/internal/log2.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
static void _upb_mapsorter_getkeys(const void* _a, const void* _b, void* a_key,
|
||||
void* b_key, size_t size) {
|
||||
const upb_tabent* const* a = _a;
|
||||
const upb_tabent* const* b = _b;
|
||||
upb_StringView a_tabkey = upb_tabstrview((*a)->key);
|
||||
upb_StringView b_tabkey = upb_tabstrview((*b)->key);
|
||||
_upb_map_fromkey(a_tabkey, a_key, size);
|
||||
_upb_map_fromkey(b_tabkey, b_key, size);
|
||||
}
|
||||
|
||||
static int _upb_mapsorter_cmpi64(const void* _a, const void* _b) {
|
||||
int64_t a, b;
|
||||
_upb_mapsorter_getkeys(_a, _b, &a, &b, 8);
|
||||
return a < b ? -1 : a > b;
|
||||
}
|
||||
|
||||
static int _upb_mapsorter_cmpu64(const void* _a, const void* _b) {
|
||||
uint64_t a, b;
|
||||
_upb_mapsorter_getkeys(_a, _b, &a, &b, 8);
|
||||
return a < b ? -1 : a > b;
|
||||
}
|
||||
|
||||
static int _upb_mapsorter_cmpi32(const void* _a, const void* _b) {
|
||||
int32_t a, b;
|
||||
_upb_mapsorter_getkeys(_a, _b, &a, &b, 4);
|
||||
return a < b ? -1 : a > b;
|
||||
}
|
||||
|
||||
static int _upb_mapsorter_cmpu32(const void* _a, const void* _b) {
|
||||
uint32_t a, b;
|
||||
_upb_mapsorter_getkeys(_a, _b, &a, &b, 4);
|
||||
return a < b ? -1 : a > b;
|
||||
}
|
||||
|
||||
static int _upb_mapsorter_cmpbool(const void* _a, const void* _b) {
|
||||
bool a, b;
|
||||
_upb_mapsorter_getkeys(_a, _b, &a, &b, 1);
|
||||
return a < b ? -1 : a > b;
|
||||
}
|
||||
|
||||
static int _upb_mapsorter_cmpstr(const void* _a, const void* _b) {
|
||||
upb_StringView a, b;
|
||||
_upb_mapsorter_getkeys(_a, _b, &a, &b, UPB_MAPTYPE_STRING);
|
||||
size_t common_size = UPB_MIN(a.size, b.size);
|
||||
int cmp = memcmp(a.data, b.data, common_size);
|
||||
if (cmp) return -cmp;
|
||||
return a.size < b.size ? -1 : a.size > b.size;
|
||||
}
|
||||
|
||||
static int (*const compar[kUpb_FieldType_SizeOf])(const void*, const void*) = {
|
||||
[kUpb_FieldType_Int64] = _upb_mapsorter_cmpi64,
|
||||
[kUpb_FieldType_SFixed64] = _upb_mapsorter_cmpi64,
|
||||
[kUpb_FieldType_SInt64] = _upb_mapsorter_cmpi64,
|
||||
|
||||
[kUpb_FieldType_UInt64] = _upb_mapsorter_cmpu64,
|
||||
[kUpb_FieldType_Fixed64] = _upb_mapsorter_cmpu64,
|
||||
|
||||
[kUpb_FieldType_Int32] = _upb_mapsorter_cmpi32,
|
||||
[kUpb_FieldType_SInt32] = _upb_mapsorter_cmpi32,
|
||||
[kUpb_FieldType_SFixed32] = _upb_mapsorter_cmpi32,
|
||||
[kUpb_FieldType_Enum] = _upb_mapsorter_cmpi32,
|
||||
|
||||
[kUpb_FieldType_UInt32] = _upb_mapsorter_cmpu32,
|
||||
[kUpb_FieldType_Fixed32] = _upb_mapsorter_cmpu32,
|
||||
|
||||
[kUpb_FieldType_Bool] = _upb_mapsorter_cmpbool,
|
||||
|
||||
[kUpb_FieldType_String] = _upb_mapsorter_cmpstr,
|
||||
[kUpb_FieldType_Bytes] = _upb_mapsorter_cmpstr,
|
||||
};
|
||||
|
||||
static bool _upb_mapsorter_resize(_upb_mapsorter* s, _upb_sortedmap* sorted,
|
||||
int size) {
|
||||
sorted->start = s->size;
|
||||
sorted->pos = sorted->start;
|
||||
sorted->end = sorted->start + size;
|
||||
|
||||
if (sorted->end > s->cap) {
|
||||
s->cap = upb_Log2CeilingSize(sorted->end);
|
||||
s->entries = realloc(s->entries, s->cap * sizeof(*s->entries));
|
||||
if (!s->entries) return false;
|
||||
}
|
||||
|
||||
s->size = sorted->end;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool _upb_mapsorter_pushmap(_upb_mapsorter* s, upb_FieldType key_type,
|
||||
const upb_Map* map, _upb_sortedmap* sorted) {
|
||||
int map_size = _upb_Map_Size(map);
|
||||
|
||||
if (!_upb_mapsorter_resize(s, sorted, map_size)) return false;
|
||||
|
||||
// Copy non-empty entries from the table to s->entries.
|
||||
const void** dst = &s->entries[sorted->start];
|
||||
const upb_tabent* src = map->table.t.entries;
|
||||
const upb_tabent* end = src + upb_table_size(&map->table.t);
|
||||
for (; src < end; src++) {
|
||||
if (!upb_tabent_isempty(src)) {
|
||||
*dst = src;
|
||||
dst++;
|
||||
}
|
||||
}
|
||||
UPB_ASSERT(dst == &s->entries[sorted->end]);
|
||||
|
||||
// Sort entries according to the key type.
|
||||
qsort(&s->entries[sorted->start], map_size, sizeof(*s->entries),
|
||||
compar[key_type]);
|
||||
return true;
|
||||
}
|
||||
|
||||
static int _upb_mapsorter_cmpext(const void* _a, const void* _b) {
|
||||
const upb_Message_Extension* const* a = _a;
|
||||
const upb_Message_Extension* const* b = _b;
|
||||
uint32_t a_num = (*a)->ext->field.number;
|
||||
uint32_t b_num = (*b)->ext->field.number;
|
||||
assert(a_num != b_num);
|
||||
return a_num < b_num ? -1 : 1;
|
||||
}
|
||||
|
||||
bool _upb_mapsorter_pushexts(_upb_mapsorter* s,
|
||||
const upb_Message_Extension* exts, size_t count,
|
||||
_upb_sortedmap* sorted) {
|
||||
if (!_upb_mapsorter_resize(s, sorted, count)) return false;
|
||||
|
||||
for (size_t i = 0; i < count; i++) {
|
||||
s->entries[sorted->start + i] = &exts[i];
|
||||
}
|
||||
|
||||
qsort(&s->entries[sorted->start], count, sizeof(*s->entries),
|
||||
_upb_mapsorter_cmpext);
|
||||
return true;
|
||||
}
|
||||
154
Pods/gRPC-Core/third_party/upb/upb/message/message.c
generated
vendored
Normal file
154
Pods/gRPC-Core/third_party/upb/upb/message/message.c
generated
vendored
Normal file
@@ -0,0 +1,154 @@
|
||||
// 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/message.h"
|
||||
|
||||
#include <math.h>
|
||||
|
||||
#include "upb/base/internal/log2.h"
|
||||
#include "upb/message/internal/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
const float kUpb_FltInfinity = INFINITY;
|
||||
const double kUpb_Infinity = INFINITY;
|
||||
const double kUpb_NaN = NAN;
|
||||
|
||||
static const size_t overhead = sizeof(upb_Message_InternalData);
|
||||
|
||||
upb_Message* upb_Message_New(const upb_MiniTable* mini_table,
|
||||
upb_Arena* arena) {
|
||||
return _upb_Message_New(mini_table, arena);
|
||||
}
|
||||
|
||||
static bool realloc_internal(upb_Message* msg, size_t need, upb_Arena* arena) {
|
||||
upb_Message_Internal* in = upb_Message_Getinternal(msg);
|
||||
if (!in->internal) {
|
||||
/* No internal data, allocate from scratch. */
|
||||
size_t size = UPB_MAX(128, upb_Log2CeilingSize(need + overhead));
|
||||
upb_Message_InternalData* internal = upb_Arena_Malloc(arena, size);
|
||||
if (!internal) return false;
|
||||
internal->size = size;
|
||||
internal->unknown_end = overhead;
|
||||
internal->ext_begin = size;
|
||||
in->internal = internal;
|
||||
} else if (in->internal->ext_begin - in->internal->unknown_end < need) {
|
||||
/* Internal data is too small, reallocate. */
|
||||
size_t new_size = upb_Log2CeilingSize(in->internal->size + need);
|
||||
size_t ext_bytes = in->internal->size - in->internal->ext_begin;
|
||||
size_t new_ext_begin = new_size - ext_bytes;
|
||||
upb_Message_InternalData* internal =
|
||||
upb_Arena_Realloc(arena, in->internal, in->internal->size, new_size);
|
||||
if (!internal) return false;
|
||||
if (ext_bytes) {
|
||||
/* Need to move extension data to the end. */
|
||||
char* ptr = (char*)internal;
|
||||
memmove(ptr + new_ext_begin, ptr + internal->ext_begin, ext_bytes);
|
||||
}
|
||||
internal->ext_begin = new_ext_begin;
|
||||
internal->size = new_size;
|
||||
in->internal = internal;
|
||||
}
|
||||
UPB_ASSERT(in->internal->ext_begin - in->internal->unknown_end >= need);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool _upb_Message_AddUnknown(upb_Message* msg, const char* data, size_t len,
|
||||
upb_Arena* arena) {
|
||||
if (!realloc_internal(msg, len, arena)) return false;
|
||||
upb_Message_Internal* in = upb_Message_Getinternal(msg);
|
||||
memcpy(UPB_PTR_AT(in->internal, in->internal->unknown_end, char), data, len);
|
||||
in->internal->unknown_end += len;
|
||||
return true;
|
||||
}
|
||||
|
||||
void _upb_Message_DiscardUnknown_shallow(upb_Message* msg) {
|
||||
upb_Message_Internal* in = upb_Message_Getinternal(msg);
|
||||
if (in->internal) {
|
||||
in->internal->unknown_end = overhead;
|
||||
}
|
||||
}
|
||||
|
||||
const char* upb_Message_GetUnknown(const upb_Message* msg, size_t* len) {
|
||||
const upb_Message_Internal* in = upb_Message_Getinternal(msg);
|
||||
if (in->internal) {
|
||||
*len = in->internal->unknown_end - overhead;
|
||||
return (char*)(in->internal + 1);
|
||||
} else {
|
||||
*len = 0;
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
void upb_Message_DeleteUnknown(upb_Message* msg, const char* data, size_t len) {
|
||||
upb_Message_Internal* in = upb_Message_Getinternal(msg);
|
||||
const char* internal_unknown_end =
|
||||
UPB_PTR_AT(in->internal, in->internal->unknown_end, char);
|
||||
#ifndef NDEBUG
|
||||
size_t full_unknown_size;
|
||||
const char* full_unknown = upb_Message_GetUnknown(msg, &full_unknown_size);
|
||||
UPB_ASSERT((uintptr_t)data >= (uintptr_t)full_unknown);
|
||||
UPB_ASSERT((uintptr_t)data < (uintptr_t)(full_unknown + full_unknown_size));
|
||||
UPB_ASSERT((uintptr_t)(data + len) > (uintptr_t)data);
|
||||
UPB_ASSERT((uintptr_t)(data + len) <= (uintptr_t)internal_unknown_end);
|
||||
#endif
|
||||
if ((data + len) != internal_unknown_end) {
|
||||
memmove((char*)data, data + len, internal_unknown_end - data - len);
|
||||
}
|
||||
in->internal->unknown_end -= len;
|
||||
}
|
||||
|
||||
const upb_Message_Extension* _upb_Message_Getexts(const upb_Message* msg,
|
||||
size_t* count) {
|
||||
const upb_Message_Internal* in = upb_Message_Getinternal(msg);
|
||||
if (in->internal) {
|
||||
*count = (in->internal->size - in->internal->ext_begin) /
|
||||
sizeof(upb_Message_Extension);
|
||||
return UPB_PTR_AT(in->internal, in->internal->ext_begin, void);
|
||||
} else {
|
||||
*count = 0;
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
const upb_Message_Extension* _upb_Message_Getext(
|
||||
const upb_Message* msg, const upb_MiniTableExtension* e) {
|
||||
size_t n;
|
||||
const upb_Message_Extension* ext = _upb_Message_Getexts(msg, &n);
|
||||
|
||||
/* For now we use linear search exclusively to find extensions. If this
|
||||
* becomes an issue due to messages with lots of extensions, we can introduce
|
||||
* a table of some sort. */
|
||||
for (size_t i = 0; i < n; i++) {
|
||||
if (ext[i].ext == e) {
|
||||
return &ext[i];
|
||||
}
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
upb_Message_Extension* _upb_Message_GetOrCreateExtension(
|
||||
upb_Message* msg, const upb_MiniTableExtension* e, upb_Arena* arena) {
|
||||
upb_Message_Extension* ext =
|
||||
(upb_Message_Extension*)_upb_Message_Getext(msg, e);
|
||||
if (ext) return ext;
|
||||
if (!realloc_internal(msg, sizeof(upb_Message_Extension), arena)) return NULL;
|
||||
upb_Message_Internal* in = upb_Message_Getinternal(msg);
|
||||
in->internal->ext_begin -= sizeof(upb_Message_Extension);
|
||||
ext = UPB_PTR_AT(in->internal, in->internal->ext_begin, void);
|
||||
memset(ext, 0, sizeof(upb_Message_Extension));
|
||||
ext->ext = e;
|
||||
return ext;
|
||||
}
|
||||
|
||||
size_t upb_Message_ExtensionCount(const upb_Message* msg) {
|
||||
size_t count;
|
||||
_upb_Message_Getexts(msg, &count);
|
||||
return count;
|
||||
}
|
||||
52
Pods/gRPC-Core/third_party/upb/upb/message/message.h
generated
vendored
Normal file
52
Pods/gRPC-Core/third_party/upb/upb/message/message.h
generated
vendored
Normal file
@@ -0,0 +1,52 @@
|
||||
// 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
|
||||
|
||||
// Public APIs for message operations that do not depend on the schema.
|
||||
//
|
||||
// MiniTable-based accessors live in accessors.h.
|
||||
|
||||
#ifndef UPB_MESSAGE_MESSAGE_H_
|
||||
#define UPB_MESSAGE_MESSAGE_H_
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/types.h" // IWYU pragma: export
|
||||
#include "upb/mini_table/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Creates a new message with the given mini_table on the given arena.
|
||||
UPB_API upb_Message* upb_Message_New(const upb_MiniTable* mini_table,
|
||||
upb_Arena* arena);
|
||||
|
||||
// Adds unknown data (serialized protobuf data) to the given message.
|
||||
// The data is copied into the message instance.
|
||||
void upb_Message_AddUnknown(upb_Message* msg, const char* data, size_t len,
|
||||
upb_Arena* arena);
|
||||
|
||||
// Returns a reference to the message's unknown data.
|
||||
const char* upb_Message_GetUnknown(const upb_Message* msg, size_t* len);
|
||||
|
||||
// Removes partial unknown data from message.
|
||||
void upb_Message_DeleteUnknown(upb_Message* msg, const char* data, size_t len);
|
||||
|
||||
// Returns the number of extensions present in this message.
|
||||
size_t upb_Message_ExtensionCount(const upb_Message* msg);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MESSAGE_MESSAGE_H_ */
|
||||
69
Pods/gRPC-Core/third_party/upb/upb/message/tagged_ptr.h
generated
vendored
Normal file
69
Pods/gRPC-Core/third_party/upb/upb/message/tagged_ptr.h
generated
vendored
Normal file
@@ -0,0 +1,69 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_TYPES_H_
|
||||
#define UPB_MINI_TABLE_TYPES_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#include "upb/message/types.h" // IWYU pragma: export
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// When a upb_Message* is stored in a message, array, or map, it is stored in a
|
||||
// tagged form. If the tag bit is set, the referenced upb_Message is of type
|
||||
// _kUpb_MiniTable_Empty (a sentinel message type with no fields) instead of
|
||||
// that field's true message type. This forms the basis of what we call
|
||||
// "dynamic tree shaking."
|
||||
//
|
||||
// See the documentation for kUpb_DecodeOption_ExperimentalAllowUnlinked for
|
||||
// more information.
|
||||
typedef uintptr_t upb_TaggedMessagePtr;
|
||||
|
||||
// Internal-only because empty messages cannot be created by the user.
|
||||
UPB_INLINE upb_TaggedMessagePtr _upb_TaggedMessagePtr_Pack(upb_Message* ptr,
|
||||
bool empty) {
|
||||
UPB_ASSERT(((uintptr_t)ptr & 1) == 0);
|
||||
return (uintptr_t)ptr | (empty ? 1 : 0);
|
||||
}
|
||||
|
||||
// Users who enable unlinked sub-messages must use this to test whether a
|
||||
// message is empty before accessing it. If a message is empty, it must be
|
||||
// first promoted using the interfaces in message/promote.h.
|
||||
UPB_INLINE bool upb_TaggedMessagePtr_IsEmpty(upb_TaggedMessagePtr ptr) {
|
||||
return ptr & 1;
|
||||
}
|
||||
|
||||
UPB_INLINE upb_Message* _upb_TaggedMessagePtr_GetMessage(
|
||||
upb_TaggedMessagePtr ptr) {
|
||||
return (upb_Message*)(ptr & ~(uintptr_t)1);
|
||||
}
|
||||
|
||||
UPB_INLINE upb_Message* upb_TaggedMessagePtr_GetNonEmptyMessage(
|
||||
upb_TaggedMessagePtr ptr) {
|
||||
UPB_ASSERT(!upb_TaggedMessagePtr_IsEmpty(ptr));
|
||||
return _upb_TaggedMessagePtr_GetMessage(ptr);
|
||||
}
|
||||
|
||||
UPB_INLINE upb_Message* _upb_TaggedMessagePtr_GetEmptyMessage(
|
||||
upb_TaggedMessagePtr ptr) {
|
||||
UPB_ASSERT(upb_TaggedMessagePtr_IsEmpty(ptr));
|
||||
return _upb_TaggedMessagePtr_GetMessage(ptr);
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_TYPES_H_ */
|
||||
15
Pods/gRPC-Core/third_party/upb/upb/message/types.h
generated
vendored
Normal file
15
Pods/gRPC-Core/third_party/upb/upb/message/types.h
generated
vendored
Normal file
@@ -0,0 +1,15 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MESSAGE_TYPES_H_
|
||||
#define UPB_MESSAGE_TYPES_H_
|
||||
|
||||
// This typedef is in a leaf header to resolve a circular dependency between
|
||||
// messages and mini tables.
|
||||
typedef void upb_Message;
|
||||
|
||||
#endif /* UPB_MESSAGE_TYPES_H_ */
|
||||
46
Pods/gRPC-Core/third_party/upb/upb/message/value.h
generated
vendored
Normal file
46
Pods/gRPC-Core/third_party/upb/upb/message/value.h
generated
vendored
Normal file
@@ -0,0 +1,46 @@
|
||||
// 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
|
||||
|
||||
// Users should include array.h or map.h instead.
|
||||
// IWYU pragma: private, include "upb/message/array.h"
|
||||
|
||||
#ifndef UPB_MESSAGE_VALUE_H_
|
||||
#define UPB_MESSAGE_VALUE_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/message/tagged_ptr.h"
|
||||
#include "upb/message/types.h"
|
||||
|
||||
typedef union {
|
||||
bool bool_val;
|
||||
float float_val;
|
||||
double double_val;
|
||||
int32_t int32_val;
|
||||
int64_t int64_val;
|
||||
uint32_t uint32_val;
|
||||
uint64_t uint64_val;
|
||||
const struct upb_Array* array_val;
|
||||
const struct upb_Map* map_val;
|
||||
const upb_Message* msg_val;
|
||||
upb_StringView str_val;
|
||||
|
||||
// EXPERIMENTAL: A tagged upb_Message*. Users must use this instead of
|
||||
// msg_val if unlinked sub-messages may possibly be in use. See the
|
||||
// documentation in kUpb_DecodeOption_ExperimentalAllowUnlinked for more
|
||||
// information.
|
||||
upb_TaggedMessagePtr tagged_msg_val;
|
||||
} upb_MessageValue;
|
||||
|
||||
typedef union {
|
||||
struct upb_Array* array;
|
||||
struct upb_Map* map;
|
||||
upb_Message* msg;
|
||||
} upb_MutableMessageValue;
|
||||
|
||||
#endif /* UPB_MESSAGE_VALUE_H_ */
|
||||
86
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/BUILD
generated
vendored
Normal file
86
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/BUILD
generated
vendored
Normal file
@@ -0,0 +1,86 @@
|
||||
load(
|
||||
"//bazel:build_defs.bzl",
|
||||
"UPB_DEFAULT_COPTS",
|
||||
"UPB_DEFAULT_CPPOPTS",
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "mini_descriptor",
|
||||
srcs = [
|
||||
"build_enum.c",
|
||||
"decode.c",
|
||||
"link.c",
|
||||
],
|
||||
hdrs = [
|
||||
"build_enum.h",
|
||||
"decode.h",
|
||||
"link.h",
|
||||
],
|
||||
copts = UPB_DEFAULT_COPTS,
|
||||
visibility = ["//visibility:public"],
|
||||
deps = [
|
||||
":internal",
|
||||
"//upb:base",
|
||||
"//upb:mem",
|
||||
"//upb:mini_table",
|
||||
"//upb:mini_table_internal",
|
||||
"//upb:port",
|
||||
],
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "internal",
|
||||
srcs = [
|
||||
"internal/base92.c",
|
||||
"internal/encode.c",
|
||||
],
|
||||
hdrs = [
|
||||
"internal/base92.h",
|
||||
"internal/decoder.h",
|
||||
"internal/encode.h",
|
||||
"internal/encode.hpp",
|
||||
"internal/modifiers.h",
|
||||
"internal/wire_constants.h",
|
||||
],
|
||||
copts = UPB_DEFAULT_COPTS,
|
||||
visibility = ["//visibility:public"],
|
||||
deps = [
|
||||
"//upb:base",
|
||||
"//upb:base_internal",
|
||||
"//upb:port",
|
||||
],
|
||||
)
|
||||
|
||||
cc_test(
|
||||
name = "encode_test",
|
||||
srcs = ["internal/encode_test.cc"],
|
||||
copts = UPB_DEFAULT_CPPOPTS,
|
||||
deps = [
|
||||
":internal",
|
||||
":mini_descriptor",
|
||||
"//:protobuf",
|
||||
"@com_google_googletest//:gtest_main",
|
||||
"//upb:base",
|
||||
"//upb:mem",
|
||||
"//upb:message_accessors_internal",
|
||||
"//upb:mini_table",
|
||||
"//upb:wire",
|
||||
"@com_google_absl//absl/container:flat_hash_set",
|
||||
],
|
||||
)
|
||||
|
||||
# begin:github_only
|
||||
filegroup(
|
||||
name = "source_files",
|
||||
srcs = glob(
|
||||
[
|
||||
"**/*.c",
|
||||
"**/*.h",
|
||||
],
|
||||
),
|
||||
visibility = [
|
||||
"//upb/cmake:__pkg__",
|
||||
"//python/dist:__pkg__",
|
||||
]
|
||||
)
|
||||
# end:github_only
|
||||
130
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/build_enum.c
generated
vendored
Normal file
130
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/build_enum.c
generated
vendored
Normal file
@@ -0,0 +1,130 @@
|
||||
// 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/mini_descriptor/build_enum.h"
|
||||
|
||||
#include "upb/mini_descriptor/internal/decoder.h"
|
||||
#include "upb/mini_descriptor/internal/wire_constants.h"
|
||||
#include "upb/mini_table/internal/enum.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct {
|
||||
upb_MdDecoder base;
|
||||
upb_Arena* arena;
|
||||
upb_MiniTableEnum* enum_table;
|
||||
uint32_t enum_value_count;
|
||||
uint32_t enum_data_count;
|
||||
uint32_t enum_data_capacity;
|
||||
} upb_MdEnumDecoder;
|
||||
|
||||
static size_t upb_MiniTableEnum_Size(size_t count) {
|
||||
return sizeof(upb_MiniTableEnum) + count * sizeof(uint32_t);
|
||||
}
|
||||
|
||||
static upb_MiniTableEnum* _upb_MiniTable_AddEnumDataMember(upb_MdEnumDecoder* d,
|
||||
uint32_t val) {
|
||||
if (d->enum_data_count == d->enum_data_capacity) {
|
||||
size_t old_sz = upb_MiniTableEnum_Size(d->enum_data_capacity);
|
||||
d->enum_data_capacity = UPB_MAX(2, d->enum_data_capacity * 2);
|
||||
size_t new_sz = upb_MiniTableEnum_Size(d->enum_data_capacity);
|
||||
d->enum_table = upb_Arena_Realloc(d->arena, d->enum_table, old_sz, new_sz);
|
||||
upb_MdDecoder_CheckOutOfMemory(&d->base, d->enum_table);
|
||||
}
|
||||
d->enum_table->data[d->enum_data_count++] = val;
|
||||
return d->enum_table;
|
||||
}
|
||||
|
||||
static void upb_MiniTableEnum_BuildValue(upb_MdEnumDecoder* d, uint32_t val) {
|
||||
upb_MiniTableEnum* table = d->enum_table;
|
||||
d->enum_value_count++;
|
||||
if (table->value_count || (val > 512 && d->enum_value_count < val / 32)) {
|
||||
if (table->value_count == 0) {
|
||||
assert(d->enum_data_count == table->mask_limit / 32);
|
||||
}
|
||||
table = _upb_MiniTable_AddEnumDataMember(d, val);
|
||||
table->value_count++;
|
||||
} else {
|
||||
uint32_t new_mask_limit = ((val / 32) + 1) * 32;
|
||||
while (table->mask_limit < new_mask_limit) {
|
||||
table = _upb_MiniTable_AddEnumDataMember(d, 0);
|
||||
table->mask_limit += 32;
|
||||
}
|
||||
table->data[val / 32] |= 1ULL << (val % 32);
|
||||
}
|
||||
}
|
||||
|
||||
static upb_MiniTableEnum* upb_MtDecoder_DoBuildMiniTableEnum(
|
||||
upb_MdEnumDecoder* d, const char* data, size_t len) {
|
||||
// If the string is non-empty then it must begin with a version tag.
|
||||
if (len) {
|
||||
if (*data != kUpb_EncodedVersion_EnumV1) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Invalid enum version: %c", *data);
|
||||
}
|
||||
data++;
|
||||
len--;
|
||||
}
|
||||
|
||||
upb_MdDecoder_CheckOutOfMemory(&d->base, d->enum_table);
|
||||
|
||||
// Guarantee at least 64 bits of mask without checking mask size.
|
||||
d->enum_table->mask_limit = 64;
|
||||
d->enum_table = _upb_MiniTable_AddEnumDataMember(d, 0);
|
||||
d->enum_table = _upb_MiniTable_AddEnumDataMember(d, 0);
|
||||
|
||||
d->enum_table->value_count = 0;
|
||||
|
||||
const char* ptr = data;
|
||||
uint32_t base = 0;
|
||||
|
||||
while (ptr < d->base.end) {
|
||||
char ch = *ptr++;
|
||||
if (ch <= kUpb_EncodedValue_MaxEnumMask) {
|
||||
uint32_t mask = _upb_FromBase92(ch);
|
||||
for (int i = 0; i < 5; i++, base++, mask >>= 1) {
|
||||
if (mask & 1) upb_MiniTableEnum_BuildValue(d, base);
|
||||
}
|
||||
} else if (kUpb_EncodedValue_MinSkip <= ch &&
|
||||
ch <= kUpb_EncodedValue_MaxSkip) {
|
||||
uint32_t skip;
|
||||
ptr = upb_MdDecoder_DecodeBase92Varint(&d->base, ptr, ch,
|
||||
kUpb_EncodedValue_MinSkip,
|
||||
kUpb_EncodedValue_MaxSkip, &skip);
|
||||
base += skip;
|
||||
} else {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Unexpected character: %c", ch);
|
||||
}
|
||||
}
|
||||
|
||||
return d->enum_table;
|
||||
}
|
||||
|
||||
static upb_MiniTableEnum* upb_MtDecoder_BuildMiniTableEnum(
|
||||
upb_MdEnumDecoder* const decoder, const char* const data, size_t const len) {
|
||||
if (UPB_SETJMP(decoder->base.err) != 0) return NULL;
|
||||
return upb_MtDecoder_DoBuildMiniTableEnum(decoder, data, len);
|
||||
}
|
||||
|
||||
upb_MiniTableEnum* upb_MiniDescriptor_BuildEnum(const char* data, size_t len,
|
||||
upb_Arena* arena,
|
||||
upb_Status* status) {
|
||||
upb_MdEnumDecoder decoder = {
|
||||
.base =
|
||||
{
|
||||
.end = UPB_PTRADD(data, len),
|
||||
.status = status,
|
||||
},
|
||||
.arena = arena,
|
||||
.enum_table = upb_Arena_Malloc(arena, upb_MiniTableEnum_Size(2)),
|
||||
.enum_value_count = 0,
|
||||
.enum_data_count = 0,
|
||||
.enum_data_capacity = 1,
|
||||
};
|
||||
|
||||
return upb_MtDecoder_BuildMiniTableEnum(&decoder, data, len);
|
||||
}
|
||||
42
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/build_enum.h
generated
vendored
Normal file
42
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/build_enum.h
generated
vendored
Normal file
@@ -0,0 +1,42 @@
|
||||
// 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
|
||||
#ifndef UPB_MINI_DESCRIPTOR_BUILD_ENUM_H_
|
||||
#define UPB_MINI_DESCRIPTOR_BUILD_ENUM_H_
|
||||
|
||||
#include "upb/base/status.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mini_table/enum.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Builds a upb_MiniTableEnum from an enum MiniDescriptor. The MiniDescriptor
|
||||
// must be for an enum, not a message.
|
||||
UPB_API upb_MiniTableEnum* upb_MiniDescriptor_BuildEnum(const char* data,
|
||||
size_t len,
|
||||
upb_Arena* arena,
|
||||
upb_Status* status);
|
||||
|
||||
// TODO: Deprecated name; update callers.
|
||||
UPB_API_INLINE upb_MiniTableEnum* upb_MiniTableEnum_Build(const char* data,
|
||||
size_t len,
|
||||
upb_Arena* arena,
|
||||
upb_Status* status) {
|
||||
return upb_MiniDescriptor_BuildEnum(data, len, arena, status);
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_MINI_DESCRIPTOR_BUILD_ENUM_H_
|
||||
870
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/decode.c
generated
vendored
Normal file
870
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/decode.c
generated
vendored
Normal file
@@ -0,0 +1,870 @@
|
||||
// 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/mini_descriptor/decode.h"
|
||||
|
||||
#include <inttypes.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mini_descriptor/internal/base92.h"
|
||||
#include "upb/mini_descriptor/internal/decoder.h"
|
||||
#include "upb/mini_descriptor/internal/modifiers.h"
|
||||
#include "upb/mini_descriptor/internal/wire_constants.h"
|
||||
#include "upb/mini_table/internal/field.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// Note: we sort by this number when calculating layout order.
|
||||
typedef enum {
|
||||
kUpb_LayoutItemType_OneofCase, // Oneof case.
|
||||
kUpb_LayoutItemType_OneofField, // Oneof field data.
|
||||
kUpb_LayoutItemType_Field, // Non-oneof field data.
|
||||
|
||||
kUpb_LayoutItemType_Max = kUpb_LayoutItemType_Field,
|
||||
} upb_LayoutItemType;
|
||||
|
||||
#define kUpb_LayoutItem_IndexSentinel ((uint16_t)-1)
|
||||
|
||||
typedef struct {
|
||||
// Index of the corresponding field. When this is a oneof field, the field's
|
||||
// offset will be the index of the next field in a linked list.
|
||||
uint16_t field_index;
|
||||
uint16_t offset;
|
||||
upb_FieldRep rep;
|
||||
upb_LayoutItemType type;
|
||||
} upb_LayoutItem;
|
||||
|
||||
typedef struct {
|
||||
upb_LayoutItem* data;
|
||||
size_t size;
|
||||
size_t capacity;
|
||||
} upb_LayoutItemVector;
|
||||
|
||||
typedef struct {
|
||||
upb_MdDecoder base;
|
||||
upb_MiniTable* table;
|
||||
upb_MiniTableField* fields;
|
||||
upb_MiniTablePlatform platform;
|
||||
upb_LayoutItemVector vec;
|
||||
upb_Arena* arena;
|
||||
} upb_MtDecoder;
|
||||
|
||||
// In each field's offset, we temporarily store a presence classifier:
|
||||
enum PresenceClass {
|
||||
kNoPresence = 0,
|
||||
kHasbitPresence = 1,
|
||||
kRequiredPresence = 2,
|
||||
kOneofBase = 3,
|
||||
// Negative values refer to a specific oneof with that number. Positive
|
||||
// values >= kOneofBase indicate that this field is in a oneof, and specify
|
||||
// the next field in this oneof's linked list.
|
||||
};
|
||||
|
||||
static bool upb_MtDecoder_FieldIsPackable(upb_MiniTableField* field) {
|
||||
return (field->mode & kUpb_FieldMode_Array) &&
|
||||
upb_FieldType_IsPackable(field->UPB_PRIVATE(descriptortype));
|
||||
}
|
||||
|
||||
typedef struct {
|
||||
uint16_t submsg_count;
|
||||
uint16_t subenum_count;
|
||||
} upb_SubCounts;
|
||||
|
||||
static void upb_MiniTable_SetTypeAndSub(upb_MiniTableField* field,
|
||||
upb_FieldType type,
|
||||
upb_SubCounts* sub_counts,
|
||||
uint64_t msg_modifiers,
|
||||
bool is_proto3_enum) {
|
||||
if (is_proto3_enum) {
|
||||
UPB_ASSERT(type == kUpb_FieldType_Enum);
|
||||
type = kUpb_FieldType_Int32;
|
||||
field->mode |= kUpb_LabelFlags_IsAlternate;
|
||||
} else if (type == kUpb_FieldType_String &&
|
||||
!(msg_modifiers & kUpb_MessageModifier_ValidateUtf8)) {
|
||||
type = kUpb_FieldType_Bytes;
|
||||
field->mode |= kUpb_LabelFlags_IsAlternate;
|
||||
}
|
||||
|
||||
field->UPB_PRIVATE(descriptortype) = type;
|
||||
|
||||
if (upb_MtDecoder_FieldIsPackable(field) &&
|
||||
(msg_modifiers & kUpb_MessageModifier_DefaultIsPacked)) {
|
||||
field->mode |= kUpb_LabelFlags_IsPacked;
|
||||
}
|
||||
|
||||
if (type == kUpb_FieldType_Message || type == kUpb_FieldType_Group) {
|
||||
field->UPB_PRIVATE(submsg_index) = sub_counts->submsg_count++;
|
||||
} else if (type == kUpb_FieldType_Enum) {
|
||||
// We will need to update this later once we know the total number of
|
||||
// submsg fields.
|
||||
field->UPB_PRIVATE(submsg_index) = sub_counts->subenum_count++;
|
||||
} else {
|
||||
field->UPB_PRIVATE(submsg_index) = kUpb_NoSub;
|
||||
}
|
||||
}
|
||||
|
||||
static const char kUpb_EncodedToType[] = {
|
||||
[kUpb_EncodedType_Double] = kUpb_FieldType_Double,
|
||||
[kUpb_EncodedType_Float] = kUpb_FieldType_Float,
|
||||
[kUpb_EncodedType_Int64] = kUpb_FieldType_Int64,
|
||||
[kUpb_EncodedType_UInt64] = kUpb_FieldType_UInt64,
|
||||
[kUpb_EncodedType_Int32] = kUpb_FieldType_Int32,
|
||||
[kUpb_EncodedType_Fixed64] = kUpb_FieldType_Fixed64,
|
||||
[kUpb_EncodedType_Fixed32] = kUpb_FieldType_Fixed32,
|
||||
[kUpb_EncodedType_Bool] = kUpb_FieldType_Bool,
|
||||
[kUpb_EncodedType_String] = kUpb_FieldType_String,
|
||||
[kUpb_EncodedType_Group] = kUpb_FieldType_Group,
|
||||
[kUpb_EncodedType_Message] = kUpb_FieldType_Message,
|
||||
[kUpb_EncodedType_Bytes] = kUpb_FieldType_Bytes,
|
||||
[kUpb_EncodedType_UInt32] = kUpb_FieldType_UInt32,
|
||||
[kUpb_EncodedType_OpenEnum] = kUpb_FieldType_Enum,
|
||||
[kUpb_EncodedType_SFixed32] = kUpb_FieldType_SFixed32,
|
||||
[kUpb_EncodedType_SFixed64] = kUpb_FieldType_SFixed64,
|
||||
[kUpb_EncodedType_SInt32] = kUpb_FieldType_SInt32,
|
||||
[kUpb_EncodedType_SInt64] = kUpb_FieldType_SInt64,
|
||||
[kUpb_EncodedType_ClosedEnum] = kUpb_FieldType_Enum,
|
||||
};
|
||||
|
||||
static void upb_MiniTable_SetField(upb_MtDecoder* d, uint8_t ch,
|
||||
upb_MiniTableField* field,
|
||||
uint64_t msg_modifiers,
|
||||
upb_SubCounts* sub_counts) {
|
||||
static const char kUpb_EncodedToFieldRep[] = {
|
||||
[kUpb_EncodedType_Double] = kUpb_FieldRep_8Byte,
|
||||
[kUpb_EncodedType_Float] = kUpb_FieldRep_4Byte,
|
||||
[kUpb_EncodedType_Int64] = kUpb_FieldRep_8Byte,
|
||||
[kUpb_EncodedType_UInt64] = kUpb_FieldRep_8Byte,
|
||||
[kUpb_EncodedType_Int32] = kUpb_FieldRep_4Byte,
|
||||
[kUpb_EncodedType_Fixed64] = kUpb_FieldRep_8Byte,
|
||||
[kUpb_EncodedType_Fixed32] = kUpb_FieldRep_4Byte,
|
||||
[kUpb_EncodedType_Bool] = kUpb_FieldRep_1Byte,
|
||||
[kUpb_EncodedType_String] = kUpb_FieldRep_StringView,
|
||||
[kUpb_EncodedType_Bytes] = kUpb_FieldRep_StringView,
|
||||
[kUpb_EncodedType_UInt32] = kUpb_FieldRep_4Byte,
|
||||
[kUpb_EncodedType_OpenEnum] = kUpb_FieldRep_4Byte,
|
||||
[kUpb_EncodedType_SFixed32] = kUpb_FieldRep_4Byte,
|
||||
[kUpb_EncodedType_SFixed64] = kUpb_FieldRep_8Byte,
|
||||
[kUpb_EncodedType_SInt32] = kUpb_FieldRep_4Byte,
|
||||
[kUpb_EncodedType_SInt64] = kUpb_FieldRep_8Byte,
|
||||
[kUpb_EncodedType_ClosedEnum] = kUpb_FieldRep_4Byte,
|
||||
};
|
||||
|
||||
char pointer_rep = d->platform == kUpb_MiniTablePlatform_32Bit
|
||||
? kUpb_FieldRep_4Byte
|
||||
: kUpb_FieldRep_8Byte;
|
||||
|
||||
int8_t type = _upb_FromBase92(ch);
|
||||
if (ch >= _upb_ToBase92(kUpb_EncodedType_RepeatedBase)) {
|
||||
type -= kUpb_EncodedType_RepeatedBase;
|
||||
field->mode = kUpb_FieldMode_Array;
|
||||
field->mode |= pointer_rep << kUpb_FieldRep_Shift;
|
||||
field->offset = kNoPresence;
|
||||
} else {
|
||||
field->mode = kUpb_FieldMode_Scalar;
|
||||
field->offset = kHasbitPresence;
|
||||
if (type == kUpb_EncodedType_Group || type == kUpb_EncodedType_Message) {
|
||||
field->mode |= pointer_rep << kUpb_FieldRep_Shift;
|
||||
} else if ((unsigned long)type >= sizeof(kUpb_EncodedToFieldRep)) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Invalid field type: %d", (int)type);
|
||||
} else {
|
||||
field->mode |= kUpb_EncodedToFieldRep[type] << kUpb_FieldRep_Shift;
|
||||
}
|
||||
}
|
||||
if ((unsigned long)type >= sizeof(kUpb_EncodedToType)) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Invalid field type: %d", (int)type);
|
||||
}
|
||||
upb_MiniTable_SetTypeAndSub(field, kUpb_EncodedToType[type], sub_counts,
|
||||
msg_modifiers, type == kUpb_EncodedType_OpenEnum);
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_ModifyField(upb_MtDecoder* d,
|
||||
uint32_t message_modifiers,
|
||||
uint32_t field_modifiers,
|
||||
upb_MiniTableField* field) {
|
||||
if (field_modifiers & kUpb_EncodedFieldModifier_FlipPacked) {
|
||||
if (!upb_MtDecoder_FieldIsPackable(field)) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base,
|
||||
"Cannot flip packed on unpackable field %" PRIu32,
|
||||
field->number);
|
||||
}
|
||||
field->mode ^= kUpb_LabelFlags_IsPacked;
|
||||
}
|
||||
|
||||
if (field_modifiers & kUpb_EncodedFieldModifier_FlipValidateUtf8) {
|
||||
if (field->UPB_PRIVATE(descriptortype) != kUpb_FieldType_Bytes ||
|
||||
!(field->mode & kUpb_LabelFlags_IsAlternate)) {
|
||||
upb_MdDecoder_ErrorJmp(
|
||||
&d->base,
|
||||
"Cannot flip ValidateUtf8 on field %" PRIu32 ", type=%d, mode=%d",
|
||||
field->number, (int)field->UPB_PRIVATE(descriptortype),
|
||||
(int)field->mode);
|
||||
}
|
||||
field->UPB_PRIVATE(descriptortype) = kUpb_FieldType_String;
|
||||
field->mode &= ~kUpb_LabelFlags_IsAlternate;
|
||||
}
|
||||
|
||||
bool singular = field_modifiers & kUpb_EncodedFieldModifier_IsProto3Singular;
|
||||
bool required = field_modifiers & kUpb_EncodedFieldModifier_IsRequired;
|
||||
|
||||
// Validate.
|
||||
if ((singular || required) && field->offset != kHasbitPresence) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base,
|
||||
"Invalid modifier(s) for repeated field %" PRIu32,
|
||||
field->number);
|
||||
}
|
||||
if (singular && required) {
|
||||
upb_MdDecoder_ErrorJmp(
|
||||
&d->base, "Field %" PRIu32 " cannot be both singular and required",
|
||||
field->number);
|
||||
}
|
||||
|
||||
if (singular) field->offset = kNoPresence;
|
||||
if (required) {
|
||||
field->offset = kRequiredPresence;
|
||||
}
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_PushItem(upb_MtDecoder* d, upb_LayoutItem item) {
|
||||
if (d->vec.size == d->vec.capacity) {
|
||||
size_t new_cap = UPB_MAX(8, d->vec.size * 2);
|
||||
d->vec.data = realloc(d->vec.data, new_cap * sizeof(*d->vec.data));
|
||||
upb_MdDecoder_CheckOutOfMemory(&d->base, d->vec.data);
|
||||
d->vec.capacity = new_cap;
|
||||
}
|
||||
d->vec.data[d->vec.size++] = item;
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_PushOneof(upb_MtDecoder* d, upb_LayoutItem item) {
|
||||
if (item.field_index == kUpb_LayoutItem_IndexSentinel) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Empty oneof");
|
||||
}
|
||||
item.field_index -= kOneofBase;
|
||||
|
||||
// Push oneof data.
|
||||
item.type = kUpb_LayoutItemType_OneofField;
|
||||
upb_MtDecoder_PushItem(d, item);
|
||||
|
||||
// Push oneof case.
|
||||
item.rep = kUpb_FieldRep_4Byte; // Field Number.
|
||||
item.type = kUpb_LayoutItemType_OneofCase;
|
||||
upb_MtDecoder_PushItem(d, item);
|
||||
}
|
||||
|
||||
size_t upb_MtDecoder_SizeOfRep(upb_FieldRep rep,
|
||||
upb_MiniTablePlatform platform) {
|
||||
static const uint8_t kRepToSize32[] = {
|
||||
[kUpb_FieldRep_1Byte] = 1,
|
||||
[kUpb_FieldRep_4Byte] = 4,
|
||||
[kUpb_FieldRep_StringView] = 8,
|
||||
[kUpb_FieldRep_8Byte] = 8,
|
||||
};
|
||||
static const uint8_t kRepToSize64[] = {
|
||||
[kUpb_FieldRep_1Byte] = 1,
|
||||
[kUpb_FieldRep_4Byte] = 4,
|
||||
[kUpb_FieldRep_StringView] = 16,
|
||||
[kUpb_FieldRep_8Byte] = 8,
|
||||
};
|
||||
UPB_ASSERT(sizeof(upb_StringView) ==
|
||||
UPB_SIZE(kRepToSize32, kRepToSize64)[kUpb_FieldRep_StringView]);
|
||||
return platform == kUpb_MiniTablePlatform_32Bit ? kRepToSize32[rep]
|
||||
: kRepToSize64[rep];
|
||||
}
|
||||
|
||||
size_t upb_MtDecoder_AlignOfRep(upb_FieldRep rep,
|
||||
upb_MiniTablePlatform platform) {
|
||||
static const uint8_t kRepToAlign32[] = {
|
||||
[kUpb_FieldRep_1Byte] = 1,
|
||||
[kUpb_FieldRep_4Byte] = 4,
|
||||
[kUpb_FieldRep_StringView] = 4,
|
||||
[kUpb_FieldRep_8Byte] = 8,
|
||||
};
|
||||
static const uint8_t kRepToAlign64[] = {
|
||||
[kUpb_FieldRep_1Byte] = 1,
|
||||
[kUpb_FieldRep_4Byte] = 4,
|
||||
[kUpb_FieldRep_StringView] = 8,
|
||||
[kUpb_FieldRep_8Byte] = 8,
|
||||
};
|
||||
UPB_ASSERT(UPB_ALIGN_OF(upb_StringView) ==
|
||||
UPB_SIZE(kRepToAlign32, kRepToAlign64)[kUpb_FieldRep_StringView]);
|
||||
return platform == kUpb_MiniTablePlatform_32Bit ? kRepToAlign32[rep]
|
||||
: kRepToAlign64[rep];
|
||||
}
|
||||
|
||||
static const char* upb_MtDecoder_DecodeOneofField(upb_MtDecoder* d,
|
||||
const char* ptr,
|
||||
char first_ch,
|
||||
upb_LayoutItem* item) {
|
||||
uint32_t field_num;
|
||||
ptr = upb_MdDecoder_DecodeBase92Varint(
|
||||
&d->base, ptr, first_ch, kUpb_EncodedValue_MinOneofField,
|
||||
kUpb_EncodedValue_MaxOneofField, &field_num);
|
||||
upb_MiniTableField* f =
|
||||
(void*)upb_MiniTable_FindFieldByNumber(d->table, field_num);
|
||||
|
||||
if (!f) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base,
|
||||
"Couldn't add field number %" PRIu32
|
||||
" to oneof, no such field number.",
|
||||
field_num);
|
||||
}
|
||||
if (f->offset != kHasbitPresence) {
|
||||
upb_MdDecoder_ErrorJmp(
|
||||
&d->base,
|
||||
"Cannot add repeated, required, or singular field %" PRIu32
|
||||
" to oneof.",
|
||||
field_num);
|
||||
}
|
||||
|
||||
// Oneof storage must be large enough to accommodate the largest member.
|
||||
int rep = f->mode >> kUpb_FieldRep_Shift;
|
||||
if (upb_MtDecoder_SizeOfRep(rep, d->platform) >
|
||||
upb_MtDecoder_SizeOfRep(item->rep, d->platform)) {
|
||||
item->rep = rep;
|
||||
}
|
||||
// Prepend this field to the linked list.
|
||||
f->offset = item->field_index;
|
||||
item->field_index = (f - d->fields) + kOneofBase;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static const char* upb_MtDecoder_DecodeOneofs(upb_MtDecoder* d,
|
||||
const char* ptr) {
|
||||
upb_LayoutItem item = {.rep = 0,
|
||||
.field_index = kUpb_LayoutItem_IndexSentinel};
|
||||
while (ptr < d->base.end) {
|
||||
char ch = *ptr++;
|
||||
if (ch == kUpb_EncodedValue_FieldSeparator) {
|
||||
// Field separator, no action needed.
|
||||
} else if (ch == kUpb_EncodedValue_OneofSeparator) {
|
||||
// End of oneof.
|
||||
upb_MtDecoder_PushOneof(d, item);
|
||||
item.field_index = kUpb_LayoutItem_IndexSentinel; // Move to next oneof.
|
||||
} else {
|
||||
ptr = upb_MtDecoder_DecodeOneofField(d, ptr, ch, &item);
|
||||
}
|
||||
}
|
||||
|
||||
// Push final oneof.
|
||||
upb_MtDecoder_PushOneof(d, item);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static const char* upb_MtDecoder_ParseModifier(upb_MtDecoder* d,
|
||||
const char* ptr, char first_ch,
|
||||
upb_MiniTableField* last_field,
|
||||
uint64_t* msg_modifiers) {
|
||||
uint32_t mod;
|
||||
ptr = upb_MdDecoder_DecodeBase92Varint(&d->base, ptr, first_ch,
|
||||
kUpb_EncodedValue_MinModifier,
|
||||
kUpb_EncodedValue_MaxModifier, &mod);
|
||||
if (last_field) {
|
||||
upb_MtDecoder_ModifyField(d, *msg_modifiers, mod, last_field);
|
||||
} else {
|
||||
if (!d->table) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base,
|
||||
"Extensions cannot have message modifiers");
|
||||
}
|
||||
*msg_modifiers = mod;
|
||||
}
|
||||
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_AllocateSubs(upb_MtDecoder* d,
|
||||
upb_SubCounts sub_counts) {
|
||||
uint32_t total_count = sub_counts.submsg_count + sub_counts.subenum_count;
|
||||
size_t subs_bytes = sizeof(*d->table->subs) * total_count;
|
||||
upb_MiniTableSub* subs = upb_Arena_Malloc(d->arena, subs_bytes);
|
||||
upb_MdDecoder_CheckOutOfMemory(&d->base, subs);
|
||||
uint32_t i = 0;
|
||||
for (; i < sub_counts.submsg_count; i++) {
|
||||
subs[i].submsg = &_kUpb_MiniTable_Empty;
|
||||
}
|
||||
if (sub_counts.subenum_count) {
|
||||
upb_MiniTableField* f = d->fields;
|
||||
upb_MiniTableField* end_f = f + d->table->field_count;
|
||||
for (; f < end_f; f++) {
|
||||
if (f->UPB_PRIVATE(descriptortype) == kUpb_FieldType_Enum) {
|
||||
f->UPB_PRIVATE(submsg_index) += sub_counts.submsg_count;
|
||||
}
|
||||
}
|
||||
for (; i < sub_counts.submsg_count + sub_counts.subenum_count; i++) {
|
||||
subs[i].subenum = NULL;
|
||||
}
|
||||
}
|
||||
d->table->subs = subs;
|
||||
}
|
||||
|
||||
static const char* upb_MtDecoder_Parse(upb_MtDecoder* d, const char* ptr,
|
||||
size_t len, void* fields,
|
||||
size_t field_size, uint16_t* field_count,
|
||||
upb_SubCounts* sub_counts) {
|
||||
uint64_t msg_modifiers = 0;
|
||||
uint32_t last_field_number = 0;
|
||||
upb_MiniTableField* last_field = NULL;
|
||||
bool need_dense_below = d->table != NULL;
|
||||
|
||||
d->base.end = UPB_PTRADD(ptr, len);
|
||||
|
||||
while (ptr < d->base.end) {
|
||||
char ch = *ptr++;
|
||||
if (ch <= kUpb_EncodedValue_MaxField) {
|
||||
if (!d->table && last_field) {
|
||||
// For extensions, consume only a single field and then return.
|
||||
return --ptr;
|
||||
}
|
||||
upb_MiniTableField* field = fields;
|
||||
*field_count += 1;
|
||||
fields = (char*)fields + field_size;
|
||||
field->number = ++last_field_number;
|
||||
last_field = field;
|
||||
upb_MiniTable_SetField(d, ch, field, msg_modifiers, sub_counts);
|
||||
} else if (kUpb_EncodedValue_MinModifier <= ch &&
|
||||
ch <= kUpb_EncodedValue_MaxModifier) {
|
||||
ptr = upb_MtDecoder_ParseModifier(d, ptr, ch, last_field, &msg_modifiers);
|
||||
if (msg_modifiers & kUpb_MessageModifier_IsExtendable) {
|
||||
d->table->ext |= kUpb_ExtMode_Extendable;
|
||||
}
|
||||
} else if (ch == kUpb_EncodedValue_End) {
|
||||
if (!d->table) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Extensions cannot have oneofs.");
|
||||
}
|
||||
ptr = upb_MtDecoder_DecodeOneofs(d, ptr);
|
||||
} else if (kUpb_EncodedValue_MinSkip <= ch &&
|
||||
ch <= kUpb_EncodedValue_MaxSkip) {
|
||||
if (need_dense_below) {
|
||||
d->table->dense_below = d->table->field_count;
|
||||
need_dense_below = false;
|
||||
}
|
||||
uint32_t skip;
|
||||
ptr = upb_MdDecoder_DecodeBase92Varint(&d->base, ptr, ch,
|
||||
kUpb_EncodedValue_MinSkip,
|
||||
kUpb_EncodedValue_MaxSkip, &skip);
|
||||
last_field_number += skip;
|
||||
last_field_number--; // Next field seen will increment.
|
||||
} else {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Invalid char: %c", ch);
|
||||
}
|
||||
}
|
||||
|
||||
if (need_dense_below) {
|
||||
d->table->dense_below = d->table->field_count;
|
||||
}
|
||||
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_ParseMessage(upb_MtDecoder* d, const char* data,
|
||||
size_t len) {
|
||||
// Buffer length is an upper bound on the number of fields. We will return
|
||||
// what we don't use.
|
||||
d->fields = upb_Arena_Malloc(d->arena, sizeof(*d->fields) * len);
|
||||
upb_MdDecoder_CheckOutOfMemory(&d->base, d->fields);
|
||||
|
||||
upb_SubCounts sub_counts = {0, 0};
|
||||
d->table->field_count = 0;
|
||||
d->table->fields = d->fields;
|
||||
upb_MtDecoder_Parse(d, data, len, d->fields, sizeof(*d->fields),
|
||||
&d->table->field_count, &sub_counts);
|
||||
|
||||
upb_Arena_ShrinkLast(d->arena, d->fields, sizeof(*d->fields) * len,
|
||||
sizeof(*d->fields) * d->table->field_count);
|
||||
d->table->fields = d->fields;
|
||||
upb_MtDecoder_AllocateSubs(d, sub_counts);
|
||||
}
|
||||
|
||||
int upb_MtDecoder_CompareFields(const void* _a, const void* _b) {
|
||||
const upb_LayoutItem* a = _a;
|
||||
const upb_LayoutItem* b = _b;
|
||||
// Currently we just sort by:
|
||||
// 1. rep (smallest fields first)
|
||||
// 2. type (oneof cases first)
|
||||
// 2. field_index (smallest numbers first)
|
||||
// The main goal of this is to reduce space lost to padding.
|
||||
// Later we may have more subtle reasons to prefer a different ordering.
|
||||
const int rep_bits = upb_Log2Ceiling(kUpb_FieldRep_Max);
|
||||
const int type_bits = upb_Log2Ceiling(kUpb_LayoutItemType_Max);
|
||||
const int idx_bits = (sizeof(a->field_index) * 8);
|
||||
UPB_ASSERT(idx_bits + rep_bits + type_bits < 32);
|
||||
#define UPB_COMBINE(rep, ty, idx) (((rep << type_bits) | ty) << idx_bits) | idx
|
||||
uint32_t a_packed = UPB_COMBINE(a->rep, a->type, a->field_index);
|
||||
uint32_t b_packed = UPB_COMBINE(b->rep, b->type, b->field_index);
|
||||
assert(a_packed != b_packed);
|
||||
#undef UPB_COMBINE
|
||||
return a_packed < b_packed ? -1 : 1;
|
||||
}
|
||||
|
||||
static bool upb_MtDecoder_SortLayoutItems(upb_MtDecoder* d) {
|
||||
// Add items for all non-oneof fields (oneofs were already added).
|
||||
int n = d->table->field_count;
|
||||
for (int i = 0; i < n; i++) {
|
||||
upb_MiniTableField* f = &d->fields[i];
|
||||
if (f->offset >= kOneofBase) continue;
|
||||
upb_LayoutItem item = {.field_index = i,
|
||||
.rep = f->mode >> kUpb_FieldRep_Shift,
|
||||
.type = kUpb_LayoutItemType_Field};
|
||||
upb_MtDecoder_PushItem(d, item);
|
||||
}
|
||||
|
||||
if (d->vec.size) {
|
||||
qsort(d->vec.data, d->vec.size, sizeof(*d->vec.data),
|
||||
upb_MtDecoder_CompareFields);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
static size_t upb_MiniTable_DivideRoundUp(size_t n, size_t d) {
|
||||
return (n + d - 1) / d;
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_AssignHasbits(upb_MtDecoder* d) {
|
||||
upb_MiniTable* ret = d->table;
|
||||
int n = ret->field_count;
|
||||
int last_hasbit = 0; // 0 cannot be used.
|
||||
|
||||
// First assign required fields, which must have the lowest hasbits.
|
||||
for (int i = 0; i < n; i++) {
|
||||
upb_MiniTableField* field = (upb_MiniTableField*)&ret->fields[i];
|
||||
if (field->offset == kRequiredPresence) {
|
||||
field->presence = ++last_hasbit;
|
||||
} else if (field->offset == kNoPresence) {
|
||||
field->presence = 0;
|
||||
}
|
||||
}
|
||||
ret->required_count = last_hasbit;
|
||||
|
||||
if (ret->required_count > 63) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Too many required fields");
|
||||
}
|
||||
|
||||
// Next assign non-required hasbit fields.
|
||||
for (int i = 0; i < n; i++) {
|
||||
upb_MiniTableField* field = (upb_MiniTableField*)&ret->fields[i];
|
||||
if (field->offset == kHasbitPresence) {
|
||||
field->presence = ++last_hasbit;
|
||||
}
|
||||
}
|
||||
|
||||
ret->size = last_hasbit ? upb_MiniTable_DivideRoundUp(last_hasbit + 1, 8) : 0;
|
||||
}
|
||||
|
||||
size_t upb_MtDecoder_Place(upb_MtDecoder* d, upb_FieldRep rep) {
|
||||
size_t size = upb_MtDecoder_SizeOfRep(rep, d->platform);
|
||||
size_t align = upb_MtDecoder_AlignOfRep(rep, d->platform);
|
||||
size_t ret = UPB_ALIGN_UP(d->table->size, align);
|
||||
static const size_t max = UINT16_MAX;
|
||||
size_t new_size = ret + size;
|
||||
if (new_size > max) {
|
||||
upb_MdDecoder_ErrorJmp(
|
||||
&d->base, "Message size exceeded maximum size of %zu bytes", max);
|
||||
}
|
||||
d->table->size = new_size;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_AssignOffsets(upb_MtDecoder* d) {
|
||||
upb_LayoutItem* end = UPB_PTRADD(d->vec.data, d->vec.size);
|
||||
|
||||
// Compute offsets.
|
||||
for (upb_LayoutItem* item = d->vec.data; item < end; item++) {
|
||||
item->offset = upb_MtDecoder_Place(d, item->rep);
|
||||
}
|
||||
|
||||
// Assign oneof case offsets. We must do these first, since assigning
|
||||
// actual offsets will overwrite the links of the linked list.
|
||||
for (upb_LayoutItem* item = d->vec.data; item < end; item++) {
|
||||
if (item->type != kUpb_LayoutItemType_OneofCase) continue;
|
||||
upb_MiniTableField* f = &d->fields[item->field_index];
|
||||
while (true) {
|
||||
f->presence = ~item->offset;
|
||||
if (f->offset == kUpb_LayoutItem_IndexSentinel) break;
|
||||
UPB_ASSERT(f->offset - kOneofBase < d->table->field_count);
|
||||
f = &d->fields[f->offset - kOneofBase];
|
||||
}
|
||||
}
|
||||
|
||||
// Assign offsets.
|
||||
for (upb_LayoutItem* item = d->vec.data; item < end; item++) {
|
||||
upb_MiniTableField* f = &d->fields[item->field_index];
|
||||
switch (item->type) {
|
||||
case kUpb_LayoutItemType_OneofField:
|
||||
while (true) {
|
||||
uint16_t next_offset = f->offset;
|
||||
f->offset = item->offset;
|
||||
if (next_offset == kUpb_LayoutItem_IndexSentinel) break;
|
||||
f = &d->fields[next_offset - kOneofBase];
|
||||
}
|
||||
break;
|
||||
case kUpb_LayoutItemType_Field:
|
||||
f->offset = item->offset;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// The fasttable parser (supported on 64-bit only) depends on this being a
|
||||
// multiple of 8 in order to satisfy UPB_MALLOC_ALIGN, which is also 8.
|
||||
//
|
||||
// On 32-bit we could potentially make this smaller, but there is no
|
||||
// compelling reason to optimize this right now.
|
||||
d->table->size = UPB_ALIGN_UP(d->table->size, 8);
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_ValidateEntryField(upb_MtDecoder* d,
|
||||
const upb_MiniTableField* f,
|
||||
uint32_t expected_num) {
|
||||
const char* name = expected_num == 1 ? "key" : "val";
|
||||
if (f->number != expected_num) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base,
|
||||
"map %s did not have expected number (%d vs %d)",
|
||||
name, expected_num, (int)f->number);
|
||||
}
|
||||
|
||||
if (upb_IsRepeatedOrMap(f)) {
|
||||
upb_MdDecoder_ErrorJmp(
|
||||
&d->base, "map %s cannot be repeated or map, or be in oneof", name);
|
||||
}
|
||||
|
||||
uint32_t not_ok_types;
|
||||
if (expected_num == 1) {
|
||||
not_ok_types = (1 << kUpb_FieldType_Float) | (1 << kUpb_FieldType_Double) |
|
||||
(1 << kUpb_FieldType_Message) | (1 << kUpb_FieldType_Group) |
|
||||
(1 << kUpb_FieldType_Bytes) | (1 << kUpb_FieldType_Enum);
|
||||
} else {
|
||||
not_ok_types = 1 << kUpb_FieldType_Group;
|
||||
}
|
||||
|
||||
if ((1 << upb_MiniTableField_Type(f)) & not_ok_types) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "map %s cannot have type %d", name,
|
||||
(int)f->UPB_PRIVATE(descriptortype));
|
||||
}
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_ParseMap(upb_MtDecoder* d, const char* data,
|
||||
size_t len) {
|
||||
upb_MtDecoder_ParseMessage(d, data, len);
|
||||
upb_MtDecoder_AssignHasbits(d);
|
||||
|
||||
if (UPB_UNLIKELY(d->table->field_count != 2)) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "%hu fields in map",
|
||||
d->table->field_count);
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
upb_LayoutItem* end = UPB_PTRADD(d->vec.data, d->vec.size);
|
||||
for (upb_LayoutItem* item = d->vec.data; item < end; item++) {
|
||||
if (item->type == kUpb_LayoutItemType_OneofCase) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Map entry cannot have oneof");
|
||||
}
|
||||
}
|
||||
|
||||
upb_MtDecoder_ValidateEntryField(d, &d->table->fields[0], 1);
|
||||
upb_MtDecoder_ValidateEntryField(d, &d->table->fields[1], 2);
|
||||
|
||||
// Map entries have a pre-determined layout, regardless of types.
|
||||
// NOTE: sync with mini_table/message_internal.h.
|
||||
const size_t kv_size = d->platform == kUpb_MiniTablePlatform_32Bit ? 8 : 16;
|
||||
const size_t hasbit_size = 8;
|
||||
d->fields[0].offset = hasbit_size;
|
||||
d->fields[1].offset = hasbit_size + kv_size;
|
||||
d->table->size = UPB_ALIGN_UP(hasbit_size + kv_size + kv_size, 8);
|
||||
|
||||
// Map entries have a special bit set to signal it's a map entry, used in
|
||||
// upb_MiniTable_SetSubMessage() below.
|
||||
d->table->ext |= kUpb_ExtMode_IsMapEntry;
|
||||
}
|
||||
|
||||
static void upb_MtDecoder_ParseMessageSet(upb_MtDecoder* d, const char* data,
|
||||
size_t len) {
|
||||
if (len > 0) {
|
||||
upb_MdDecoder_ErrorJmp(&d->base, "Invalid message set encode length: %zu",
|
||||
len);
|
||||
}
|
||||
|
||||
upb_MiniTable* ret = d->table;
|
||||
ret->size = 0;
|
||||
ret->field_count = 0;
|
||||
ret->ext = kUpb_ExtMode_IsMessageSet;
|
||||
ret->dense_below = 0;
|
||||
ret->table_mask = -1;
|
||||
ret->required_count = 0;
|
||||
}
|
||||
|
||||
static upb_MiniTable* upb_MtDecoder_DoBuildMiniTableWithBuf(
|
||||
upb_MtDecoder* decoder, const char* data, size_t len, void** buf,
|
||||
size_t* buf_size) {
|
||||
upb_MdDecoder_CheckOutOfMemory(&decoder->base, decoder->table);
|
||||
|
||||
decoder->table->size = 0;
|
||||
decoder->table->field_count = 0;
|
||||
decoder->table->ext = kUpb_ExtMode_NonExtendable;
|
||||
decoder->table->dense_below = 0;
|
||||
decoder->table->table_mask = -1;
|
||||
decoder->table->required_count = 0;
|
||||
|
||||
// Strip off and verify the version tag.
|
||||
if (!len--) goto done;
|
||||
const char vers = *data++;
|
||||
|
||||
switch (vers) {
|
||||
case kUpb_EncodedVersion_MapV1:
|
||||
upb_MtDecoder_ParseMap(decoder, data, len);
|
||||
break;
|
||||
|
||||
case kUpb_EncodedVersion_MessageV1:
|
||||
upb_MtDecoder_ParseMessage(decoder, data, len);
|
||||
upb_MtDecoder_AssignHasbits(decoder);
|
||||
upb_MtDecoder_SortLayoutItems(decoder);
|
||||
upb_MtDecoder_AssignOffsets(decoder);
|
||||
break;
|
||||
|
||||
case kUpb_EncodedVersion_MessageSetV1:
|
||||
upb_MtDecoder_ParseMessageSet(decoder, data, len);
|
||||
break;
|
||||
|
||||
default:
|
||||
upb_MdDecoder_ErrorJmp(&decoder->base, "Invalid message version: %c",
|
||||
vers);
|
||||
}
|
||||
|
||||
done:
|
||||
*buf = decoder->vec.data;
|
||||
*buf_size = decoder->vec.capacity * sizeof(*decoder->vec.data);
|
||||
return decoder->table;
|
||||
}
|
||||
|
||||
static upb_MiniTable* upb_MtDecoder_BuildMiniTableWithBuf(
|
||||
upb_MtDecoder* const decoder, const char* const data, const size_t len,
|
||||
void** const buf, size_t* const buf_size) {
|
||||
if (UPB_SETJMP(decoder->base.err) != 0) {
|
||||
*buf = decoder->vec.data;
|
||||
*buf_size = decoder->vec.capacity * sizeof(*decoder->vec.data);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return upb_MtDecoder_DoBuildMiniTableWithBuf(decoder, data, len, buf,
|
||||
buf_size);
|
||||
}
|
||||
|
||||
upb_MiniTable* upb_MiniTable_BuildWithBuf(const char* data, size_t len,
|
||||
upb_MiniTablePlatform platform,
|
||||
upb_Arena* arena, void** buf,
|
||||
size_t* buf_size,
|
||||
upb_Status* status) {
|
||||
upb_MtDecoder decoder = {
|
||||
.base = {.status = status},
|
||||
.platform = platform,
|
||||
.vec =
|
||||
{
|
||||
.data = *buf,
|
||||
.capacity = *buf_size / sizeof(*decoder.vec.data),
|
||||
.size = 0,
|
||||
},
|
||||
.arena = arena,
|
||||
.table = upb_Arena_Malloc(arena, sizeof(*decoder.table)),
|
||||
};
|
||||
|
||||
return upb_MtDecoder_BuildMiniTableWithBuf(&decoder, data, len, buf,
|
||||
buf_size);
|
||||
}
|
||||
|
||||
static const char* upb_MtDecoder_DoBuildMiniTableExtension(
|
||||
upb_MtDecoder* decoder, const char* data, size_t len,
|
||||
upb_MiniTableExtension* ext, const upb_MiniTable* extendee,
|
||||
upb_MiniTableSub sub) {
|
||||
// If the string is non-empty then it must begin with a version tag.
|
||||
if (len) {
|
||||
if (*data != kUpb_EncodedVersion_ExtensionV1) {
|
||||
upb_MdDecoder_ErrorJmp(&decoder->base, "Invalid ext version: %c", *data);
|
||||
}
|
||||
data++;
|
||||
len--;
|
||||
}
|
||||
|
||||
uint16_t count = 0;
|
||||
upb_SubCounts sub_counts = {0, 0};
|
||||
const char* ret = upb_MtDecoder_Parse(decoder, data, len, ext, sizeof(*ext),
|
||||
&count, &sub_counts);
|
||||
if (!ret || count != 1) return NULL;
|
||||
|
||||
upb_MiniTableField* f = &ext->field;
|
||||
|
||||
f->mode |= kUpb_LabelFlags_IsExtension;
|
||||
f->offset = 0;
|
||||
f->presence = 0;
|
||||
|
||||
if (extendee->ext & kUpb_ExtMode_IsMessageSet) {
|
||||
// Extensions of MessageSet must be messages.
|
||||
if (!upb_IsSubMessage(f)) return NULL;
|
||||
|
||||
// Extensions of MessageSet must be non-repeating.
|
||||
if ((f->mode & kUpb_FieldMode_Mask) == kUpb_FieldMode_Array) return NULL;
|
||||
}
|
||||
|
||||
ext->extendee = extendee;
|
||||
ext->sub = sub;
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static const char* upb_MtDecoder_BuildMiniTableExtension(
|
||||
upb_MtDecoder* const decoder, const char* const data, const size_t len,
|
||||
upb_MiniTableExtension* const ext, const upb_MiniTable* const extendee,
|
||||
const upb_MiniTableSub sub) {
|
||||
if (UPB_SETJMP(decoder->base.err) != 0) return NULL;
|
||||
return upb_MtDecoder_DoBuildMiniTableExtension(decoder, data, len, ext,
|
||||
extendee, sub);
|
||||
}
|
||||
|
||||
const char* _upb_MiniTableExtension_Init(const char* data, size_t len,
|
||||
upb_MiniTableExtension* ext,
|
||||
const upb_MiniTable* extendee,
|
||||
upb_MiniTableSub sub,
|
||||
upb_MiniTablePlatform platform,
|
||||
upb_Status* status) {
|
||||
upb_MtDecoder decoder = {
|
||||
.base = {.status = status},
|
||||
.arena = NULL,
|
||||
.table = NULL,
|
||||
.platform = platform,
|
||||
};
|
||||
|
||||
return upb_MtDecoder_BuildMiniTableExtension(&decoder, data, len, ext,
|
||||
extendee, sub);
|
||||
}
|
||||
|
||||
upb_MiniTableExtension* _upb_MiniTableExtension_Build(
|
||||
const char* data, size_t len, const upb_MiniTable* extendee,
|
||||
upb_MiniTableSub sub, upb_MiniTablePlatform platform, upb_Arena* arena,
|
||||
upb_Status* status) {
|
||||
upb_MiniTableExtension* ext =
|
||||
upb_Arena_Malloc(arena, sizeof(upb_MiniTableExtension));
|
||||
if (UPB_UNLIKELY(!ext)) return NULL;
|
||||
|
||||
const char* ptr = _upb_MiniTableExtension_Init(data, len, ext, extendee, sub,
|
||||
platform, status);
|
||||
if (UPB_UNLIKELY(!ptr)) return NULL;
|
||||
|
||||
return ext;
|
||||
}
|
||||
|
||||
upb_MiniTable* _upb_MiniTable_Build(const char* data, size_t len,
|
||||
upb_MiniTablePlatform platform,
|
||||
upb_Arena* arena, upb_Status* status) {
|
||||
void* buf = NULL;
|
||||
size_t size = 0;
|
||||
upb_MiniTable* ret = upb_MiniTable_BuildWithBuf(data, len, platform, arena,
|
||||
&buf, &size, status);
|
||||
free(buf);
|
||||
return ret;
|
||||
}
|
||||
120
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/decode.h
generated
vendored
Normal file
120
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/decode.h
generated
vendored
Normal file
@@ -0,0 +1,120 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_DECODE_H_
|
||||
#define UPB_MINI_TABLE_DECODE_H_
|
||||
|
||||
#include "upb/base/status.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mini_table/extension.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
#include "upb/mini_table/sub.h"
|
||||
|
||||
// Export the newer headers, for legacy users. New users should include the
|
||||
// more specific headers directly.
|
||||
// IWYU pragma: begin_exports
|
||||
#include "upb/mini_descriptor/build_enum.h"
|
||||
#include "upb/mini_descriptor/link.h"
|
||||
// IWYU pragma: end_exports
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef enum {
|
||||
kUpb_MiniTablePlatform_32Bit,
|
||||
kUpb_MiniTablePlatform_64Bit,
|
||||
kUpb_MiniTablePlatform_Native =
|
||||
UPB_SIZE(kUpb_MiniTablePlatform_32Bit, kUpb_MiniTablePlatform_64Bit),
|
||||
} upb_MiniTablePlatform;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Builds a mini table from the data encoded in the buffer [data, len]. If any
|
||||
// errors occur, returns NULL and sets a status message. In the success case,
|
||||
// the caller must call upb_MiniTable_SetSub*() for all message or proto2 enum
|
||||
// fields to link the table to the appropriate sub-tables.
|
||||
upb_MiniTable* _upb_MiniTable_Build(const char* data, size_t len,
|
||||
upb_MiniTablePlatform platform,
|
||||
upb_Arena* arena, upb_Status* status);
|
||||
|
||||
UPB_API_INLINE upb_MiniTable* upb_MiniTable_Build(const char* data, size_t len,
|
||||
upb_Arena* arena,
|
||||
upb_Status* status) {
|
||||
return _upb_MiniTable_Build(data, len, kUpb_MiniTablePlatform_Native, arena,
|
||||
status);
|
||||
}
|
||||
|
||||
// Initializes a MiniTableExtension buffer that has already been allocated.
|
||||
// This is needed by upb_FileDef and upb_MessageDef, which allocate all of the
|
||||
// extensions together in a single contiguous array.
|
||||
const char* _upb_MiniTableExtension_Init(const char* data, size_t len,
|
||||
upb_MiniTableExtension* ext,
|
||||
const upb_MiniTable* extendee,
|
||||
upb_MiniTableSub sub,
|
||||
upb_MiniTablePlatform platform,
|
||||
upb_Status* status);
|
||||
|
||||
UPB_API_INLINE const char* upb_MiniTableExtension_Init(
|
||||
const char* data, size_t len, upb_MiniTableExtension* ext,
|
||||
const upb_MiniTable* extendee, upb_MiniTableSub sub, upb_Status* status) {
|
||||
return _upb_MiniTableExtension_Init(data, len, ext, extendee, sub,
|
||||
kUpb_MiniTablePlatform_Native, status);
|
||||
}
|
||||
|
||||
UPB_API upb_MiniTableExtension* _upb_MiniTableExtension_Build(
|
||||
const char* data, size_t len, const upb_MiniTable* extendee,
|
||||
upb_MiniTableSub sub, upb_MiniTablePlatform platform, upb_Arena* arena,
|
||||
upb_Status* status);
|
||||
|
||||
UPB_API_INLINE upb_MiniTableExtension* upb_MiniTableExtension_Build(
|
||||
const char* data, size_t len, const upb_MiniTable* extendee,
|
||||
upb_Arena* arena, upb_Status* status) {
|
||||
upb_MiniTableSub sub;
|
||||
sub.submsg = NULL;
|
||||
return _upb_MiniTableExtension_Build(
|
||||
data, len, extendee, sub, kUpb_MiniTablePlatform_Native, arena, status);
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_MiniTableExtension* upb_MiniTableExtension_BuildMessage(
|
||||
const char* data, size_t len, const upb_MiniTable* extendee,
|
||||
upb_MiniTable* submsg, upb_Arena* arena, upb_Status* status) {
|
||||
upb_MiniTableSub sub;
|
||||
sub.submsg = submsg;
|
||||
return _upb_MiniTableExtension_Build(
|
||||
data, len, extendee, sub, kUpb_MiniTablePlatform_Native, arena, status);
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_MiniTableExtension* upb_MiniTableExtension_BuildEnum(
|
||||
const char* data, size_t len, const upb_MiniTable* extendee,
|
||||
upb_MiniTableEnum* subenum, upb_Arena* arena, upb_Status* status) {
|
||||
upb_MiniTableSub sub;
|
||||
sub.subenum = subenum;
|
||||
return _upb_MiniTableExtension_Build(
|
||||
data, len, extendee, sub, kUpb_MiniTablePlatform_Native, arena, status);
|
||||
}
|
||||
|
||||
// Like upb_MiniTable_Build(), but the user provides a buffer of layout data so
|
||||
// it can be reused from call to call, avoiding repeated realloc()/free().
|
||||
//
|
||||
// The caller owns `*buf` both before and after the call, and must free() it
|
||||
// when it is no longer in use. The function will realloc() `*buf` as
|
||||
// necessary, updating `*size` accordingly.
|
||||
upb_MiniTable* upb_MiniTable_BuildWithBuf(const char* data, size_t len,
|
||||
upb_MiniTablePlatform platform,
|
||||
upb_Arena* arena, void** buf,
|
||||
size_t* buf_size, upb_Status* status);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_DECODE_H_ */
|
||||
26
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/base92.c
generated
vendored
Normal file
26
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/base92.c
generated
vendored
Normal file
@@ -0,0 +1,26 @@
|
||||
// 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/mini_descriptor/internal/base92.h"
|
||||
|
||||
const char _kUpb_ToBase92[] = {
|
||||
' ', '!', '#', '$', '%', '&', '(', ')', '*', '+', ',', '-', '.', '/',
|
||||
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', ':', ';', '<', '=',
|
||||
'>', '?', '@', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K',
|
||||
'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y',
|
||||
'Z', '[', ']', '^', '_', '`', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h',
|
||||
'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v',
|
||||
'w', 'x', 'y', 'z', '{', '|', '}', '~',
|
||||
};
|
||||
|
||||
const int8_t _kUpb_FromBase92[] = {
|
||||
0, 1, -1, 2, 3, 4, 5, -1, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
|
||||
17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35,
|
||||
36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54,
|
||||
55, 56, 57, -1, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72,
|
||||
73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91,
|
||||
};
|
||||
63
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/base92.h
generated
vendored
Normal file
63
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/base92.h
generated
vendored
Normal file
@@ -0,0 +1,63 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_DESCRIPTOR_INTERNAL_BASE92_H_
|
||||
#define UPB_MINI_DESCRIPTOR_INTERNAL_BASE92_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#include "upb/base/internal/log2.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_INLINE char _upb_ToBase92(int8_t ch) {
|
||||
extern const char _kUpb_ToBase92[];
|
||||
UPB_ASSERT(0 <= ch && ch < 92);
|
||||
return _kUpb_ToBase92[ch];
|
||||
}
|
||||
|
||||
UPB_INLINE char _upb_FromBase92(uint8_t ch) {
|
||||
extern const int8_t _kUpb_FromBase92[];
|
||||
if (' ' > ch || ch > '~') return -1;
|
||||
return _kUpb_FromBase92[ch - ' '];
|
||||
}
|
||||
|
||||
UPB_INLINE const char* _upb_Base92_DecodeVarint(const char* ptr,
|
||||
const char* end, char first_ch,
|
||||
uint8_t min, uint8_t max,
|
||||
uint32_t* out_val) {
|
||||
uint32_t val = 0;
|
||||
uint32_t shift = 0;
|
||||
const int bits_per_char =
|
||||
upb_Log2Ceiling(_upb_FromBase92(max) - _upb_FromBase92(min));
|
||||
char ch = first_ch;
|
||||
while (1) {
|
||||
uint32_t bits = _upb_FromBase92(ch) - _upb_FromBase92(min);
|
||||
val |= bits << shift;
|
||||
if (ptr == end || *ptr < min || max < *ptr) {
|
||||
*out_val = val;
|
||||
UPB_ASSUME(ptr != NULL);
|
||||
return ptr;
|
||||
}
|
||||
ch = *ptr++;
|
||||
shift += bits_per_char;
|
||||
if (shift >= 32) return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_MINI_DESCRIPTOR_INTERNAL_BASE92_H_
|
||||
53
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/decoder.h
generated
vendored
Normal file
53
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/decoder.h
generated
vendored
Normal file
@@ -0,0 +1,53 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_DESCRIPTOR_INTERNAL_DECODER_H_
|
||||
#define UPB_MINI_DESCRIPTOR_INTERNAL_DECODER_H_
|
||||
|
||||
#include "upb/base/status.h"
|
||||
#include "upb/mini_descriptor/internal/base92.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// upb_MdDecoder: used internally for decoding MiniDescriptors for messages,
|
||||
// extensions, and enums.
|
||||
typedef struct {
|
||||
const char* end;
|
||||
upb_Status* status;
|
||||
jmp_buf err;
|
||||
} upb_MdDecoder;
|
||||
|
||||
UPB_PRINTF(2, 3)
|
||||
UPB_NORETURN UPB_INLINE void upb_MdDecoder_ErrorJmp(upb_MdDecoder* d,
|
||||
const char* fmt, ...) {
|
||||
if (d->status) {
|
||||
va_list argp;
|
||||
upb_Status_SetErrorMessage(d->status, "Error building mini table: ");
|
||||
va_start(argp, fmt);
|
||||
upb_Status_VAppendErrorFormat(d->status, fmt, argp);
|
||||
va_end(argp);
|
||||
}
|
||||
UPB_LONGJMP(d->err, 1);
|
||||
}
|
||||
|
||||
UPB_INLINE void upb_MdDecoder_CheckOutOfMemory(upb_MdDecoder* d,
|
||||
const void* ptr) {
|
||||
if (!ptr) upb_MdDecoder_ErrorJmp(d, "Out of memory");
|
||||
}
|
||||
|
||||
UPB_INLINE const char* upb_MdDecoder_DecodeBase92Varint(
|
||||
upb_MdDecoder* d, const char* ptr, char first_ch, uint8_t min, uint8_t max,
|
||||
uint32_t* out_val) {
|
||||
ptr = _upb_Base92_DecodeVarint(ptr, d->end, first_ch, min, max, out_val);
|
||||
if (!ptr) upb_MdDecoder_ErrorJmp(d, "Overlong varint");
|
||||
return ptr;
|
||||
}
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_MINI_DESCRIPTOR_INTERNAL_DECODER_H_
|
||||
323
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/encode.c
generated
vendored
Normal file
323
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/encode.c
generated
vendored
Normal file
@@ -0,0 +1,323 @@
|
||||
// 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/mini_descriptor/internal/encode.h"
|
||||
|
||||
#include <assert.h>
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
|
||||
#include "upb/base/internal/log2.h"
|
||||
#include "upb/mini_descriptor/internal/base92.h"
|
||||
#include "upb/mini_descriptor/internal/modifiers.h"
|
||||
#include "upb/mini_descriptor/internal/wire_constants.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct {
|
||||
uint64_t present_values_mask;
|
||||
uint32_t last_written_value;
|
||||
} upb_MtDataEncoderInternal_EnumState;
|
||||
|
||||
typedef struct {
|
||||
uint64_t msg_modifiers;
|
||||
uint32_t last_field_num;
|
||||
enum {
|
||||
kUpb_OneofState_NotStarted,
|
||||
kUpb_OneofState_StartedOneof,
|
||||
kUpb_OneofState_EmittedOneofField,
|
||||
} oneof_state;
|
||||
} upb_MtDataEncoderInternal_MsgState;
|
||||
|
||||
typedef struct {
|
||||
char* buf_start; // Only for checking kUpb_MtDataEncoder_MinSize.
|
||||
union {
|
||||
upb_MtDataEncoderInternal_EnumState enum_state;
|
||||
upb_MtDataEncoderInternal_MsgState msg_state;
|
||||
} state;
|
||||
} upb_MtDataEncoderInternal;
|
||||
|
||||
static upb_MtDataEncoderInternal* upb_MtDataEncoder_GetInternal(
|
||||
upb_MtDataEncoder* e, char* buf_start) {
|
||||
UPB_ASSERT(sizeof(upb_MtDataEncoderInternal) <= sizeof(e->internal));
|
||||
upb_MtDataEncoderInternal* ret = (upb_MtDataEncoderInternal*)e->internal;
|
||||
ret->buf_start = buf_start;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static char* upb_MtDataEncoder_PutRaw(upb_MtDataEncoder* e, char* ptr,
|
||||
char ch) {
|
||||
upb_MtDataEncoderInternal* in = (upb_MtDataEncoderInternal*)e->internal;
|
||||
UPB_ASSERT(ptr - in->buf_start < kUpb_MtDataEncoder_MinSize);
|
||||
if (ptr == e->end) return NULL;
|
||||
*ptr++ = ch;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static char* upb_MtDataEncoder_Put(upb_MtDataEncoder* e, char* ptr, char ch) {
|
||||
return upb_MtDataEncoder_PutRaw(e, ptr, _upb_ToBase92(ch));
|
||||
}
|
||||
|
||||
static char* upb_MtDataEncoder_PutBase92Varint(upb_MtDataEncoder* e, char* ptr,
|
||||
uint32_t val, int min, int max) {
|
||||
int shift = upb_Log2Ceiling(_upb_FromBase92(max) - _upb_FromBase92(min) + 1);
|
||||
UPB_ASSERT(shift <= 6);
|
||||
uint32_t mask = (1 << shift) - 1;
|
||||
do {
|
||||
uint32_t bits = val & mask;
|
||||
ptr = upb_MtDataEncoder_Put(e, ptr, bits + _upb_FromBase92(min));
|
||||
if (!ptr) return NULL;
|
||||
val >>= shift;
|
||||
} while (val);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_PutModifier(upb_MtDataEncoder* e, char* ptr,
|
||||
uint64_t mod) {
|
||||
if (mod) {
|
||||
ptr = upb_MtDataEncoder_PutBase92Varint(e, ptr, mod,
|
||||
kUpb_EncodedValue_MinModifier,
|
||||
kUpb_EncodedValue_MaxModifier);
|
||||
}
|
||||
return ptr;
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_EncodeExtension(upb_MtDataEncoder* e, char* ptr,
|
||||
upb_FieldType type, uint32_t field_num,
|
||||
uint64_t field_mod) {
|
||||
upb_MtDataEncoderInternal* in = upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
in->state.msg_state.msg_modifiers = 0;
|
||||
in->state.msg_state.last_field_num = 0;
|
||||
in->state.msg_state.oneof_state = kUpb_OneofState_NotStarted;
|
||||
|
||||
ptr = upb_MtDataEncoder_PutRaw(e, ptr, kUpb_EncodedVersion_ExtensionV1);
|
||||
if (!ptr) return NULL;
|
||||
|
||||
return upb_MtDataEncoder_PutField(e, ptr, type, field_num, field_mod);
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_EncodeMap(upb_MtDataEncoder* e, char* ptr,
|
||||
upb_FieldType key_type,
|
||||
upb_FieldType value_type, uint64_t key_mod,
|
||||
uint64_t value_mod) {
|
||||
upb_MtDataEncoderInternal* in = upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
in->state.msg_state.msg_modifiers = 0;
|
||||
in->state.msg_state.last_field_num = 0;
|
||||
in->state.msg_state.oneof_state = kUpb_OneofState_NotStarted;
|
||||
|
||||
ptr = upb_MtDataEncoder_PutRaw(e, ptr, kUpb_EncodedVersion_MapV1);
|
||||
if (!ptr) return NULL;
|
||||
|
||||
ptr = upb_MtDataEncoder_PutField(e, ptr, key_type, 1, key_mod);
|
||||
if (!ptr) return NULL;
|
||||
|
||||
return upb_MtDataEncoder_PutField(e, ptr, value_type, 2, value_mod);
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_EncodeMessageSet(upb_MtDataEncoder* e, char* ptr) {
|
||||
(void)upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
return upb_MtDataEncoder_PutRaw(e, ptr, kUpb_EncodedVersion_MessageSetV1);
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_StartMessage(upb_MtDataEncoder* e, char* ptr,
|
||||
uint64_t msg_mod) {
|
||||
upb_MtDataEncoderInternal* in = upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
in->state.msg_state.msg_modifiers = msg_mod;
|
||||
in->state.msg_state.last_field_num = 0;
|
||||
in->state.msg_state.oneof_state = kUpb_OneofState_NotStarted;
|
||||
|
||||
ptr = upb_MtDataEncoder_PutRaw(e, ptr, kUpb_EncodedVersion_MessageV1);
|
||||
if (!ptr) return NULL;
|
||||
|
||||
return upb_MtDataEncoder_PutModifier(e, ptr, msg_mod);
|
||||
}
|
||||
|
||||
static char* _upb_MtDataEncoder_MaybePutFieldSkip(upb_MtDataEncoder* e,
|
||||
char* ptr,
|
||||
uint32_t field_num) {
|
||||
upb_MtDataEncoderInternal* in = (upb_MtDataEncoderInternal*)e->internal;
|
||||
if (field_num <= in->state.msg_state.last_field_num) return NULL;
|
||||
if (in->state.msg_state.last_field_num + 1 != field_num) {
|
||||
// Put skip.
|
||||
UPB_ASSERT(field_num > in->state.msg_state.last_field_num);
|
||||
uint32_t skip = field_num - in->state.msg_state.last_field_num;
|
||||
ptr = upb_MtDataEncoder_PutBase92Varint(
|
||||
e, ptr, skip, kUpb_EncodedValue_MinSkip, kUpb_EncodedValue_MaxSkip);
|
||||
if (!ptr) return NULL;
|
||||
}
|
||||
in->state.msg_state.last_field_num = field_num;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static char* _upb_MtDataEncoder_PutFieldType(upb_MtDataEncoder* e, char* ptr,
|
||||
upb_FieldType type,
|
||||
uint64_t field_mod) {
|
||||
static const char kUpb_TypeToEncoded[] = {
|
||||
[kUpb_FieldType_Double] = kUpb_EncodedType_Double,
|
||||
[kUpb_FieldType_Float] = kUpb_EncodedType_Float,
|
||||
[kUpb_FieldType_Int64] = kUpb_EncodedType_Int64,
|
||||
[kUpb_FieldType_UInt64] = kUpb_EncodedType_UInt64,
|
||||
[kUpb_FieldType_Int32] = kUpb_EncodedType_Int32,
|
||||
[kUpb_FieldType_Fixed64] = kUpb_EncodedType_Fixed64,
|
||||
[kUpb_FieldType_Fixed32] = kUpb_EncodedType_Fixed32,
|
||||
[kUpb_FieldType_Bool] = kUpb_EncodedType_Bool,
|
||||
[kUpb_FieldType_String] = kUpb_EncodedType_String,
|
||||
[kUpb_FieldType_Group] = kUpb_EncodedType_Group,
|
||||
[kUpb_FieldType_Message] = kUpb_EncodedType_Message,
|
||||
[kUpb_FieldType_Bytes] = kUpb_EncodedType_Bytes,
|
||||
[kUpb_FieldType_UInt32] = kUpb_EncodedType_UInt32,
|
||||
[kUpb_FieldType_Enum] = kUpb_EncodedType_OpenEnum,
|
||||
[kUpb_FieldType_SFixed32] = kUpb_EncodedType_SFixed32,
|
||||
[kUpb_FieldType_SFixed64] = kUpb_EncodedType_SFixed64,
|
||||
[kUpb_FieldType_SInt32] = kUpb_EncodedType_SInt32,
|
||||
[kUpb_FieldType_SInt64] = kUpb_EncodedType_SInt64,
|
||||
};
|
||||
|
||||
int encoded_type = kUpb_TypeToEncoded[type];
|
||||
|
||||
if (field_mod & kUpb_FieldModifier_IsClosedEnum) {
|
||||
UPB_ASSERT(type == kUpb_FieldType_Enum);
|
||||
encoded_type = kUpb_EncodedType_ClosedEnum;
|
||||
}
|
||||
|
||||
if (field_mod & kUpb_FieldModifier_IsRepeated) {
|
||||
// Repeated fields shift the type number up (unlike other modifiers which
|
||||
// are bit flags).
|
||||
encoded_type += kUpb_EncodedType_RepeatedBase;
|
||||
}
|
||||
|
||||
return upb_MtDataEncoder_Put(e, ptr, encoded_type);
|
||||
}
|
||||
|
||||
static char* _upb_MtDataEncoder_MaybePutModifiers(upb_MtDataEncoder* e,
|
||||
char* ptr, upb_FieldType type,
|
||||
uint64_t field_mod) {
|
||||
upb_MtDataEncoderInternal* in = (upb_MtDataEncoderInternal*)e->internal;
|
||||
uint32_t encoded_modifiers = 0;
|
||||
if ((field_mod & kUpb_FieldModifier_IsRepeated) &&
|
||||
upb_FieldType_IsPackable(type)) {
|
||||
bool field_is_packed = field_mod & kUpb_FieldModifier_IsPacked;
|
||||
bool default_is_packed = in->state.msg_state.msg_modifiers &
|
||||
kUpb_MessageModifier_DefaultIsPacked;
|
||||
if (field_is_packed != default_is_packed) {
|
||||
encoded_modifiers |= kUpb_EncodedFieldModifier_FlipPacked;
|
||||
}
|
||||
}
|
||||
|
||||
if (type == kUpb_FieldType_String) {
|
||||
bool field_validates_utf8 = field_mod & kUpb_FieldModifier_ValidateUtf8;
|
||||
bool message_validates_utf8 =
|
||||
in->state.msg_state.msg_modifiers & kUpb_MessageModifier_ValidateUtf8;
|
||||
if (field_validates_utf8 != message_validates_utf8) {
|
||||
// Old binaries do not recognize the field modifier. We need the failure
|
||||
// mode to be too lax rather than too strict. Our caller should have
|
||||
// handled this (see _upb_MessageDef_ValidateUtf8()).
|
||||
assert(!message_validates_utf8);
|
||||
encoded_modifiers |= kUpb_EncodedFieldModifier_FlipValidateUtf8;
|
||||
}
|
||||
}
|
||||
|
||||
if (field_mod & kUpb_FieldModifier_IsProto3Singular) {
|
||||
encoded_modifiers |= kUpb_EncodedFieldModifier_IsProto3Singular;
|
||||
}
|
||||
|
||||
if (field_mod & kUpb_FieldModifier_IsRequired) {
|
||||
encoded_modifiers |= kUpb_EncodedFieldModifier_IsRequired;
|
||||
}
|
||||
|
||||
return upb_MtDataEncoder_PutModifier(e, ptr, encoded_modifiers);
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_PutField(upb_MtDataEncoder* e, char* ptr,
|
||||
upb_FieldType type, uint32_t field_num,
|
||||
uint64_t field_mod) {
|
||||
upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
|
||||
ptr = _upb_MtDataEncoder_MaybePutFieldSkip(e, ptr, field_num);
|
||||
if (!ptr) return NULL;
|
||||
|
||||
ptr = _upb_MtDataEncoder_PutFieldType(e, ptr, type, field_mod);
|
||||
if (!ptr) return NULL;
|
||||
|
||||
return _upb_MtDataEncoder_MaybePutModifiers(e, ptr, type, field_mod);
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_StartOneof(upb_MtDataEncoder* e, char* ptr) {
|
||||
upb_MtDataEncoderInternal* in = upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
if (in->state.msg_state.oneof_state == kUpb_OneofState_NotStarted) {
|
||||
ptr = upb_MtDataEncoder_Put(e, ptr, _upb_FromBase92(kUpb_EncodedValue_End));
|
||||
} else {
|
||||
ptr = upb_MtDataEncoder_Put(
|
||||
e, ptr, _upb_FromBase92(kUpb_EncodedValue_OneofSeparator));
|
||||
}
|
||||
in->state.msg_state.oneof_state = kUpb_OneofState_StartedOneof;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_PutOneofField(upb_MtDataEncoder* e, char* ptr,
|
||||
uint32_t field_num) {
|
||||
upb_MtDataEncoderInternal* in = upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
if (in->state.msg_state.oneof_state == kUpb_OneofState_EmittedOneofField) {
|
||||
ptr = upb_MtDataEncoder_Put(
|
||||
e, ptr, _upb_FromBase92(kUpb_EncodedValue_FieldSeparator));
|
||||
if (!ptr) return NULL;
|
||||
}
|
||||
ptr = upb_MtDataEncoder_PutBase92Varint(e, ptr, field_num, _upb_ToBase92(0),
|
||||
_upb_ToBase92(63));
|
||||
in->state.msg_state.oneof_state = kUpb_OneofState_EmittedOneofField;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_StartEnum(upb_MtDataEncoder* e, char* ptr) {
|
||||
upb_MtDataEncoderInternal* in = upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
in->state.enum_state.present_values_mask = 0;
|
||||
in->state.enum_state.last_written_value = 0;
|
||||
|
||||
return upb_MtDataEncoder_PutRaw(e, ptr, kUpb_EncodedVersion_EnumV1);
|
||||
}
|
||||
|
||||
static char* upb_MtDataEncoder_FlushDenseEnumMask(upb_MtDataEncoder* e,
|
||||
char* ptr) {
|
||||
upb_MtDataEncoderInternal* in = (upb_MtDataEncoderInternal*)e->internal;
|
||||
ptr = upb_MtDataEncoder_Put(e, ptr, in->state.enum_state.present_values_mask);
|
||||
in->state.enum_state.present_values_mask = 0;
|
||||
in->state.enum_state.last_written_value += 5;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_PutEnumValue(upb_MtDataEncoder* e, char* ptr,
|
||||
uint32_t val) {
|
||||
// TODO: optimize this encoding.
|
||||
upb_MtDataEncoderInternal* in = upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
UPB_ASSERT(val >= in->state.enum_state.last_written_value);
|
||||
uint32_t delta = val - in->state.enum_state.last_written_value;
|
||||
if (delta >= 5 && in->state.enum_state.present_values_mask) {
|
||||
ptr = upb_MtDataEncoder_FlushDenseEnumMask(e, ptr);
|
||||
if (!ptr) {
|
||||
return NULL;
|
||||
}
|
||||
delta -= 5;
|
||||
}
|
||||
|
||||
if (delta >= 5) {
|
||||
ptr = upb_MtDataEncoder_PutBase92Varint(
|
||||
e, ptr, delta, kUpb_EncodedValue_MinSkip, kUpb_EncodedValue_MaxSkip);
|
||||
in->state.enum_state.last_written_value += delta;
|
||||
delta = 0;
|
||||
}
|
||||
|
||||
UPB_ASSERT((in->state.enum_state.present_values_mask >> delta) == 0);
|
||||
in->state.enum_state.present_values_mask |= 1ULL << delta;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
char* upb_MtDataEncoder_EndEnum(upb_MtDataEncoder* e, char* ptr) {
|
||||
upb_MtDataEncoderInternal* in = upb_MtDataEncoder_GetInternal(e, ptr);
|
||||
if (!in->state.enum_state.present_values_mask) return ptr;
|
||||
return upb_MtDataEncoder_FlushDenseEnumMask(e, ptr);
|
||||
}
|
||||
92
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/encode.h
generated
vendored
Normal file
92
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/encode.h
generated
vendored
Normal file
@@ -0,0 +1,92 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_DESCRIPTOR_INTERNAL_ENCODE_H_
|
||||
#define UPB_MINI_DESCRIPTOR_INTERNAL_ENCODE_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// If the input buffer has at least this many bytes available, the encoder call
|
||||
// is guaranteed to succeed (as long as field number order is maintained).
|
||||
#define kUpb_MtDataEncoder_MinSize 16
|
||||
|
||||
typedef struct {
|
||||
char* end; // Limit of the buffer passed as a parameter.
|
||||
// Aliased to internal-only members in .cc.
|
||||
char internal[32];
|
||||
} upb_MtDataEncoder;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Encodes field/oneof information for a given message. The sequence of calls
|
||||
// should look like:
|
||||
//
|
||||
// upb_MtDataEncoder e;
|
||||
// char buf[256];
|
||||
// char* ptr = buf;
|
||||
// e.end = ptr + sizeof(buf);
|
||||
// unit64_t msg_mod = ...; // bitwise & of kUpb_MessageModifiers or zero
|
||||
// ptr = upb_MtDataEncoder_StartMessage(&e, ptr, msg_mod);
|
||||
// // Fields *must* be in field number order.
|
||||
// ptr = upb_MtDataEncoder_PutField(&e, ptr, ...);
|
||||
// ptr = upb_MtDataEncoder_PutField(&e, ptr, ...);
|
||||
// ptr = upb_MtDataEncoder_PutField(&e, ptr, ...);
|
||||
//
|
||||
// // If oneofs are present. Oneofs must be encoded after regular fields.
|
||||
// ptr = upb_MiniTable_StartOneof(&e, ptr)
|
||||
// ptr = upb_MiniTable_PutOneofField(&e, ptr, ...);
|
||||
// ptr = upb_MiniTable_PutOneofField(&e, ptr, ...);
|
||||
//
|
||||
// ptr = upb_MiniTable_StartOneof(&e, ptr);
|
||||
// ptr = upb_MiniTable_PutOneofField(&e, ptr, ...);
|
||||
// ptr = upb_MiniTable_PutOneofField(&e, ptr, ...);
|
||||
//
|
||||
// Oneofs must be encoded after all regular fields.
|
||||
char* upb_MtDataEncoder_StartMessage(upb_MtDataEncoder* e, char* ptr,
|
||||
uint64_t msg_mod);
|
||||
char* upb_MtDataEncoder_PutField(upb_MtDataEncoder* e, char* ptr,
|
||||
upb_FieldType type, uint32_t field_num,
|
||||
uint64_t field_mod);
|
||||
char* upb_MtDataEncoder_StartOneof(upb_MtDataEncoder* e, char* ptr);
|
||||
char* upb_MtDataEncoder_PutOneofField(upb_MtDataEncoder* e, char* ptr,
|
||||
uint32_t field_num);
|
||||
|
||||
// Encodes the set of values for a given enum. The values must be given in
|
||||
// order (after casting to uint32_t), and repeats are not allowed.
|
||||
char* upb_MtDataEncoder_StartEnum(upb_MtDataEncoder* e, char* ptr);
|
||||
char* upb_MtDataEncoder_PutEnumValue(upb_MtDataEncoder* e, char* ptr,
|
||||
uint32_t val);
|
||||
char* upb_MtDataEncoder_EndEnum(upb_MtDataEncoder* e, char* ptr);
|
||||
|
||||
// Encodes an entire mini descriptor for an extension.
|
||||
char* upb_MtDataEncoder_EncodeExtension(upb_MtDataEncoder* e, char* ptr,
|
||||
upb_FieldType type, uint32_t field_num,
|
||||
uint64_t field_mod);
|
||||
|
||||
// Encodes an entire mini descriptor for a map.
|
||||
char* upb_MtDataEncoder_EncodeMap(upb_MtDataEncoder* e, char* ptr,
|
||||
upb_FieldType key_type,
|
||||
upb_FieldType value_type, uint64_t key_mod,
|
||||
uint64_t value_mod);
|
||||
|
||||
// Encodes an entire mini descriptor for a message set.
|
||||
char* upb_MtDataEncoder_EncodeMessageSet(upb_MtDataEncoder* e, char* ptr);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_DESCRIPTOR_INTERNAL_ENCODE_H_ */
|
||||
118
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/encode.hpp
generated
vendored
Normal file
118
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/encode.hpp
generated
vendored
Normal file
@@ -0,0 +1,118 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_ENCODE_INTERNAL_HPP_
|
||||
#define UPB_MINI_TABLE_ENCODE_INTERNAL_HPP_
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "upb/base/internal/log2.h"
|
||||
#include "upb/mini_descriptor/internal/encode.h"
|
||||
|
||||
namespace upb {
|
||||
|
||||
class MtDataEncoder {
|
||||
public:
|
||||
MtDataEncoder() : appender_(&encoder_) {}
|
||||
|
||||
bool StartMessage(uint64_t msg_mod) {
|
||||
return appender_([this, msg_mod](char* buf) {
|
||||
return upb_MtDataEncoder_StartMessage(&encoder_, buf, msg_mod);
|
||||
});
|
||||
}
|
||||
|
||||
bool PutField(upb_FieldType type, uint32_t field_num, uint64_t field_mod) {
|
||||
return appender_([this, type, field_num, field_mod](char* buf) {
|
||||
return upb_MtDataEncoder_PutField(&encoder_, buf, type, field_num,
|
||||
field_mod);
|
||||
});
|
||||
}
|
||||
|
||||
bool StartOneof() {
|
||||
return appender_([this](char* buf) {
|
||||
return upb_MtDataEncoder_StartOneof(&encoder_, buf);
|
||||
});
|
||||
}
|
||||
|
||||
bool PutOneofField(uint32_t field_num) {
|
||||
return appender_([this, field_num](char* buf) {
|
||||
return upb_MtDataEncoder_PutOneofField(&encoder_, buf, field_num);
|
||||
});
|
||||
}
|
||||
|
||||
bool StartEnum() {
|
||||
return appender_([this](char* buf) {
|
||||
return upb_MtDataEncoder_StartEnum(&encoder_, buf);
|
||||
});
|
||||
}
|
||||
|
||||
bool PutEnumValue(uint32_t enum_value) {
|
||||
return appender_([this, enum_value](char* buf) {
|
||||
return upb_MtDataEncoder_PutEnumValue(&encoder_, buf, enum_value);
|
||||
});
|
||||
}
|
||||
|
||||
bool EndEnum() {
|
||||
return appender_([this](char* buf) {
|
||||
return upb_MtDataEncoder_EndEnum(&encoder_, buf);
|
||||
});
|
||||
}
|
||||
|
||||
bool EncodeExtension(upb_FieldType type, uint32_t field_num,
|
||||
uint64_t field_mod) {
|
||||
return appender_([this, type, field_num, field_mod](char* buf) {
|
||||
return upb_MtDataEncoder_EncodeExtension(&encoder_, buf, type, field_num,
|
||||
field_mod);
|
||||
});
|
||||
}
|
||||
|
||||
bool EncodeMap(upb_FieldType key_type, upb_FieldType val_type,
|
||||
uint64_t key_mod, uint64_t val_mod) {
|
||||
return appender_([this, key_type, val_type, key_mod, val_mod](char* buf) {
|
||||
return upb_MtDataEncoder_EncodeMap(&encoder_, buf, key_type, val_type,
|
||||
key_mod, val_mod);
|
||||
});
|
||||
}
|
||||
|
||||
bool EncodeMessageSet() {
|
||||
return appender_([this](char* buf) {
|
||||
return upb_MtDataEncoder_EncodeMessageSet(&encoder_, buf);
|
||||
});
|
||||
}
|
||||
|
||||
const std::string& data() const { return appender_.data(); }
|
||||
|
||||
private:
|
||||
class StringAppender {
|
||||
public:
|
||||
StringAppender(upb_MtDataEncoder* e) { e->end = buf_ + sizeof(buf_); }
|
||||
|
||||
template <class T>
|
||||
bool operator()(T&& func) {
|
||||
char* end = func(buf_);
|
||||
if (!end) return false;
|
||||
// C++ does not guarantee that string has doubling growth behavior, but
|
||||
// we need it to avoid O(n^2).
|
||||
str_.reserve(upb_Log2CeilingSize(str_.size() + (end - buf_)));
|
||||
str_.append(buf_, end - buf_);
|
||||
return true;
|
||||
}
|
||||
|
||||
const std::string& data() const { return str_; }
|
||||
|
||||
private:
|
||||
char buf_[kUpb_MtDataEncoder_MinSize];
|
||||
std::string str_;
|
||||
};
|
||||
|
||||
upb_MtDataEncoder encoder_;
|
||||
StringAppender appender_;
|
||||
};
|
||||
|
||||
} // namespace upb
|
||||
|
||||
#endif /* UPB_MINI_TABLE_ENCODE_INTERNAL_HPP_ */
|
||||
32
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/modifiers.h
generated
vendored
Normal file
32
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/modifiers.h
generated
vendored
Normal file
@@ -0,0 +1,32 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_DESCRIPTOR_INTERNAL_MODIFIERS_H_
|
||||
#define UPB_MINI_DESCRIPTOR_INTERNAL_MODIFIERS_H_
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef enum {
|
||||
kUpb_FieldModifier_IsRepeated = 1 << 0,
|
||||
kUpb_FieldModifier_IsPacked = 1 << 1,
|
||||
kUpb_FieldModifier_IsClosedEnum = 1 << 2,
|
||||
kUpb_FieldModifier_IsProto3Singular = 1 << 3,
|
||||
kUpb_FieldModifier_IsRequired = 1 << 4,
|
||||
kUpb_FieldModifier_ValidateUtf8 = 1 << 5,
|
||||
} kUpb_FieldModifier;
|
||||
|
||||
// These modifiers are also used on the wire.
|
||||
typedef enum {
|
||||
kUpb_MessageModifier_ValidateUtf8 = 1 << 0,
|
||||
kUpb_MessageModifier_DefaultIsPacked = 1 << 1,
|
||||
kUpb_MessageModifier_IsExtendable = 1 << 2,
|
||||
} kUpb_MessageModifier;
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_MINI_DESCRIPTOR_INTERNAL_MODIFIERS_H_
|
||||
72
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/wire_constants.h
generated
vendored
Normal file
72
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/internal/wire_constants.h
generated
vendored
Normal file
@@ -0,0 +1,72 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_DESCRIPTOR_INTERNAL_WIRE_CONSTANTS_H_
|
||||
#define UPB_MINI_DESCRIPTOR_INTERNAL_WIRE_CONSTANTS_H_
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef enum {
|
||||
kUpb_EncodedType_Double = 0,
|
||||
kUpb_EncodedType_Float = 1,
|
||||
kUpb_EncodedType_Fixed32 = 2,
|
||||
kUpb_EncodedType_Fixed64 = 3,
|
||||
kUpb_EncodedType_SFixed32 = 4,
|
||||
kUpb_EncodedType_SFixed64 = 5,
|
||||
kUpb_EncodedType_Int32 = 6,
|
||||
kUpb_EncodedType_UInt32 = 7,
|
||||
kUpb_EncodedType_SInt32 = 8,
|
||||
kUpb_EncodedType_Int64 = 9,
|
||||
kUpb_EncodedType_UInt64 = 10,
|
||||
kUpb_EncodedType_SInt64 = 11,
|
||||
kUpb_EncodedType_OpenEnum = 12,
|
||||
kUpb_EncodedType_Bool = 13,
|
||||
kUpb_EncodedType_Bytes = 14,
|
||||
kUpb_EncodedType_String = 15,
|
||||
kUpb_EncodedType_Group = 16,
|
||||
kUpb_EncodedType_Message = 17,
|
||||
kUpb_EncodedType_ClosedEnum = 18,
|
||||
|
||||
kUpb_EncodedType_RepeatedBase = 20,
|
||||
} upb_EncodedType;
|
||||
|
||||
typedef enum {
|
||||
kUpb_EncodedFieldModifier_FlipPacked = 1 << 0,
|
||||
kUpb_EncodedFieldModifier_IsRequired = 1 << 1,
|
||||
kUpb_EncodedFieldModifier_IsProto3Singular = 1 << 2,
|
||||
kUpb_EncodedFieldModifier_FlipValidateUtf8 = 1 << 3,
|
||||
} upb_EncodedFieldModifier;
|
||||
|
||||
enum {
|
||||
kUpb_EncodedValue_MinField = ' ',
|
||||
kUpb_EncodedValue_MaxField = 'I',
|
||||
kUpb_EncodedValue_MinModifier = 'L',
|
||||
kUpb_EncodedValue_MaxModifier = '[',
|
||||
kUpb_EncodedValue_End = '^',
|
||||
kUpb_EncodedValue_MinSkip = '_',
|
||||
kUpb_EncodedValue_MaxSkip = '~',
|
||||
kUpb_EncodedValue_OneofSeparator = '~',
|
||||
kUpb_EncodedValue_FieldSeparator = '|',
|
||||
kUpb_EncodedValue_MinOneofField = ' ',
|
||||
kUpb_EncodedValue_MaxOneofField = 'b',
|
||||
kUpb_EncodedValue_MaxEnumMask = 'A',
|
||||
};
|
||||
|
||||
enum {
|
||||
kUpb_EncodedVersion_EnumV1 = '!',
|
||||
kUpb_EncodedVersion_ExtensionV1 = '#',
|
||||
kUpb_EncodedVersion_MapV1 = '%',
|
||||
kUpb_EncodedVersion_MessageV1 = '$',
|
||||
kUpb_EncodedVersion_MessageSetV1 = '&',
|
||||
};
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_MINI_DESCRIPTOR_INTERNAL_WIRE_CONSTANTS_H_
|
||||
122
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/link.c
generated
vendored
Normal file
122
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/link.c
generated
vendored
Normal file
@@ -0,0 +1,122 @@
|
||||
// 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/mini_descriptor/link.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
bool upb_MiniTable_SetSubMessage(upb_MiniTable* table,
|
||||
upb_MiniTableField* field,
|
||||
const upb_MiniTable* sub) {
|
||||
UPB_ASSERT((uintptr_t)table->fields <= (uintptr_t)field &&
|
||||
(uintptr_t)field <
|
||||
(uintptr_t)(table->fields + table->field_count));
|
||||
UPB_ASSERT(sub);
|
||||
|
||||
const bool sub_is_map = sub->ext & kUpb_ExtMode_IsMapEntry;
|
||||
|
||||
switch (field->UPB_PRIVATE(descriptortype)) {
|
||||
case kUpb_FieldType_Message:
|
||||
if (sub_is_map) {
|
||||
const bool table_is_map = table->ext & kUpb_ExtMode_IsMapEntry;
|
||||
if (UPB_UNLIKELY(table_is_map)) return false;
|
||||
|
||||
field->mode = (field->mode & ~kUpb_FieldMode_Mask) | kUpb_FieldMode_Map;
|
||||
}
|
||||
break;
|
||||
|
||||
case kUpb_FieldType_Group:
|
||||
if (UPB_UNLIKELY(sub_is_map)) return false;
|
||||
break;
|
||||
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
|
||||
upb_MiniTableSub* table_sub =
|
||||
(void*)&table->subs[field->UPB_PRIVATE(submsg_index)];
|
||||
// TODO: Add this assert back once YouTube is updated to not call
|
||||
// this function repeatedly.
|
||||
// UPB_ASSERT(table_sub->submsg == &_kUpb_MiniTable_Empty);
|
||||
table_sub->submsg = sub;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_MiniTable_SetSubEnum(upb_MiniTable* table, upb_MiniTableField* field,
|
||||
const upb_MiniTableEnum* sub) {
|
||||
UPB_ASSERT((uintptr_t)table->fields <= (uintptr_t)field &&
|
||||
(uintptr_t)field <
|
||||
(uintptr_t)(table->fields + table->field_count));
|
||||
UPB_ASSERT(sub);
|
||||
|
||||
upb_MiniTableSub* table_sub =
|
||||
(void*)&table->subs[field->UPB_PRIVATE(submsg_index)];
|
||||
table_sub->subenum = sub;
|
||||
return true;
|
||||
}
|
||||
|
||||
uint32_t upb_MiniTable_GetSubList(const upb_MiniTable* mt,
|
||||
const upb_MiniTableField** subs) {
|
||||
uint32_t msg_count = 0;
|
||||
uint32_t enum_count = 0;
|
||||
|
||||
for (int i = 0; i < mt->field_count; i++) {
|
||||
const upb_MiniTableField* f = &mt->fields[i];
|
||||
if (upb_MiniTableField_CType(f) == kUpb_CType_Message) {
|
||||
*subs = f;
|
||||
++subs;
|
||||
msg_count++;
|
||||
}
|
||||
}
|
||||
|
||||
for (int i = 0; i < mt->field_count; i++) {
|
||||
const upb_MiniTableField* f = &mt->fields[i];
|
||||
if (upb_MiniTableField_CType(f) == kUpb_CType_Enum) {
|
||||
*subs = f;
|
||||
++subs;
|
||||
enum_count++;
|
||||
}
|
||||
}
|
||||
|
||||
return (msg_count << 16) | enum_count;
|
||||
}
|
||||
|
||||
// The list of sub_tables and sub_enums must exactly match the number and order
|
||||
// of sub-message fields and sub-enum fields given by upb_MiniTable_GetSubList()
|
||||
// above.
|
||||
bool upb_MiniTable_Link(upb_MiniTable* mt, const upb_MiniTable** sub_tables,
|
||||
size_t sub_table_count,
|
||||
const upb_MiniTableEnum** sub_enums,
|
||||
size_t sub_enum_count) {
|
||||
uint32_t msg_count = 0;
|
||||
uint32_t enum_count = 0;
|
||||
|
||||
for (int i = 0; i < mt->field_count; i++) {
|
||||
upb_MiniTableField* f = (upb_MiniTableField*)&mt->fields[i];
|
||||
if (upb_MiniTableField_CType(f) == kUpb_CType_Message) {
|
||||
const upb_MiniTable* sub = sub_tables[msg_count++];
|
||||
if (msg_count > sub_table_count) return false;
|
||||
if (sub != NULL) {
|
||||
if (!upb_MiniTable_SetSubMessage(mt, f, sub)) return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (int i = 0; i < mt->field_count; i++) {
|
||||
upb_MiniTableField* f = (upb_MiniTableField*)&mt->fields[i];
|
||||
if (upb_MiniTableField_IsClosedEnum(f)) {
|
||||
const upb_MiniTableEnum* sub = sub_enums[enum_count++];
|
||||
if (enum_count > sub_enum_count) return false;
|
||||
if (sub != NULL) {
|
||||
if (!upb_MiniTable_SetSubEnum(mt, f, sub)) return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
84
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/link.h
generated
vendored
Normal file
84
Pods/gRPC-Core/third_party/upb/upb/mini_descriptor/link.h
generated
vendored
Normal file
@@ -0,0 +1,84 @@
|
||||
// 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
|
||||
|
||||
// Functions for linking MiniTables together once they are built from a
|
||||
// MiniDescriptor.
|
||||
//
|
||||
// These functions have names like upb_MiniTable_Link() because they operate on
|
||||
// MiniTables. We put them here, rather than in the mini_table/ directory,
|
||||
// because they are only needed when building MiniTables from MiniDescriptors.
|
||||
// The interfaces in mini_table/ assume that MiniTables are immutable.
|
||||
|
||||
#ifndef UPB_MINI_DESCRIPTOR_LINK_H_
|
||||
#define UPB_MINI_DESCRIPTOR_LINK_H_
|
||||
|
||||
#include "upb/base/status.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mini_table/extension.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
#include "upb/mini_table/sub.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Links a sub-message field to a MiniTable for that sub-message. If a
|
||||
// sub-message field is not linked, it will be treated as an unknown field
|
||||
// during parsing, and setting the field will not be allowed. It is possible
|
||||
// to link the message field later, at which point it will no longer be treated
|
||||
// as unknown. However there is no synchronization for this operation, which
|
||||
// means parallel mutation requires external synchronization.
|
||||
// Returns success/failure.
|
||||
UPB_API bool upb_MiniTable_SetSubMessage(upb_MiniTable* table,
|
||||
upb_MiniTableField* field,
|
||||
const upb_MiniTable* sub);
|
||||
|
||||
// Links an enum field to a MiniTable for that enum.
|
||||
// All enum fields must be linked prior to parsing.
|
||||
// Returns success/failure.
|
||||
UPB_API bool upb_MiniTable_SetSubEnum(upb_MiniTable* table,
|
||||
upb_MiniTableField* field,
|
||||
const upb_MiniTableEnum* sub);
|
||||
|
||||
// Returns a list of fields that require linking at runtime, to connect the
|
||||
// MiniTable to its sub-messages and sub-enums. The list of fields will be
|
||||
// written to the `subs` array, which must have been allocated by the caller
|
||||
// and must be large enough to hold a list of all fields in the message.
|
||||
//
|
||||
// The order of the fields returned by this function is significant: it matches
|
||||
// the order expected by upb_MiniTable_Link() below.
|
||||
//
|
||||
// The return value packs the sub-message count and sub-enum count into a single
|
||||
// integer like so:
|
||||
// return (msg_count << 16) | enum_count;
|
||||
UPB_API uint32_t upb_MiniTable_GetSubList(const upb_MiniTable* mt,
|
||||
const upb_MiniTableField** subs);
|
||||
|
||||
// Links a message to its sub-messages and sub-enums. The caller must pass
|
||||
// arrays of sub-tables and sub-enums, in the same length and order as is
|
||||
// returned by upb_MiniTable_GetSubList() above. However, individual elements
|
||||
// of the sub_tables may be NULL if those sub-messages were tree shaken.
|
||||
//
|
||||
// Returns false if either array is too short, or if any of the tables fails
|
||||
// to link.
|
||||
UPB_API bool upb_MiniTable_Link(upb_MiniTable* mt,
|
||||
const upb_MiniTable** sub_tables,
|
||||
size_t sub_table_count,
|
||||
const upb_MiniTableEnum** sub_enums,
|
||||
size_t sub_enum_count);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_MINI_DESCRIPTOR_LINK_H_
|
||||
38
Pods/gRPC-Core/third_party/upb/upb/mini_table/enum.h
generated
vendored
Normal file
38
Pods/gRPC-Core/third_party/upb/upb/mini_table/enum.h
generated
vendored
Normal file
@@ -0,0 +1,38 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_ENUM_H_
|
||||
#define UPB_MINI_TABLE_ENUM_H_
|
||||
|
||||
#include "upb/mini_table/internal/enum.h"
|
||||
|
||||
// Must be last
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct upb_MiniTableEnum upb_MiniTableEnum;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Validates enum value against range defined by enum mini table.
|
||||
UPB_INLINE bool upb_MiniTableEnum_CheckValue(const struct upb_MiniTableEnum* e,
|
||||
uint32_t val) {
|
||||
_kUpb_FastEnumCheck_Status status = _upb_MiniTable_CheckEnumValueFast(e, val);
|
||||
if (UPB_UNLIKELY(status == _kUpb_FastEnumCheck_CannotCheckFast)) {
|
||||
return _upb_MiniTable_CheckEnumValueSlow(e, val);
|
||||
}
|
||||
return status == _kUpb_FastEnumCheck_ValueIsInEnum ? true : false;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_ENUM_H_ */
|
||||
15
Pods/gRPC-Core/third_party/upb/upb/mini_table/extension.h
generated
vendored
Normal file
15
Pods/gRPC-Core/third_party/upb/upb/mini_table/extension.h
generated
vendored
Normal file
@@ -0,0 +1,15 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_EXTENSION_H_
|
||||
#define UPB_MINI_TABLE_EXTENSION_H_
|
||||
|
||||
#include "upb/mini_table/internal/extension.h"
|
||||
|
||||
typedef struct upb_MiniTableExtension upb_MiniTableExtension;
|
||||
|
||||
#endif /* UPB_MINI_TABLE_EXTENSION_H_ */
|
||||
76
Pods/gRPC-Core/third_party/upb/upb/mini_table/extension_registry.c
generated
vendored
Normal file
76
Pods/gRPC-Core/third_party/upb/upb/mini_table/extension_registry.c
generated
vendored
Normal file
@@ -0,0 +1,76 @@
|
||||
// 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/mini_table/extension_registry.h"
|
||||
|
||||
#include "upb/hash/str_table.h"
|
||||
#include "upb/mini_table/extension.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#define EXTREG_KEY_SIZE (sizeof(upb_MiniTable*) + sizeof(uint32_t))
|
||||
|
||||
struct upb_ExtensionRegistry {
|
||||
upb_Arena* arena;
|
||||
upb_strtable exts; // Key is upb_MiniTable* concatenated with fieldnum.
|
||||
};
|
||||
|
||||
static void extreg_key(char* buf, const upb_MiniTable* l, uint32_t fieldnum) {
|
||||
memcpy(buf, &l, sizeof(l));
|
||||
memcpy(buf + sizeof(l), &fieldnum, sizeof(fieldnum));
|
||||
}
|
||||
|
||||
upb_ExtensionRegistry* upb_ExtensionRegistry_New(upb_Arena* arena) {
|
||||
upb_ExtensionRegistry* r = upb_Arena_Malloc(arena, sizeof(*r));
|
||||
if (!r) return NULL;
|
||||
r->arena = arena;
|
||||
if (!upb_strtable_init(&r->exts, 8, arena)) return NULL;
|
||||
return r;
|
||||
}
|
||||
|
||||
UPB_API bool upb_ExtensionRegistry_Add(upb_ExtensionRegistry* r,
|
||||
const upb_MiniTableExtension* e) {
|
||||
char buf[EXTREG_KEY_SIZE];
|
||||
extreg_key(buf, e->extendee, e->field.number);
|
||||
if (upb_strtable_lookup2(&r->exts, buf, EXTREG_KEY_SIZE, NULL)) return false;
|
||||
return upb_strtable_insert(&r->exts, buf, EXTREG_KEY_SIZE,
|
||||
upb_value_constptr(e), r->arena);
|
||||
}
|
||||
|
||||
bool upb_ExtensionRegistry_AddArray(upb_ExtensionRegistry* r,
|
||||
const upb_MiniTableExtension** e,
|
||||
size_t count) {
|
||||
const upb_MiniTableExtension** start = e;
|
||||
const upb_MiniTableExtension** end = UPB_PTRADD(e, count);
|
||||
for (; e < end; e++) {
|
||||
if (!upb_ExtensionRegistry_Add(r, *e)) goto failure;
|
||||
}
|
||||
return true;
|
||||
|
||||
failure:
|
||||
// Back out the entries previously added.
|
||||
for (end = e, e = start; e < end; e++) {
|
||||
const upb_MiniTableExtension* ext = *e;
|
||||
char buf[EXTREG_KEY_SIZE];
|
||||
extreg_key(buf, ext->extendee, ext->field.number);
|
||||
upb_strtable_remove2(&r->exts, buf, EXTREG_KEY_SIZE, NULL);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
const upb_MiniTableExtension* upb_ExtensionRegistry_Lookup(
|
||||
const upb_ExtensionRegistry* r, const upb_MiniTable* t, uint32_t num) {
|
||||
char buf[EXTREG_KEY_SIZE];
|
||||
upb_value v;
|
||||
extreg_key(buf, t, num);
|
||||
if (upb_strtable_lookup2(&r->exts, buf, EXTREG_KEY_SIZE, &v)) {
|
||||
return upb_value_getconstptr(v);
|
||||
} else {
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
85
Pods/gRPC-Core/third_party/upb/upb/mini_table/extension_registry.h
generated
vendored
Normal file
85
Pods/gRPC-Core/third_party/upb/upb/mini_table/extension_registry.h
generated
vendored
Normal file
@@ -0,0 +1,85 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_EXTENSION_REGISTRY_H_
|
||||
#define UPB_MINI_TABLE_EXTENSION_REGISTRY_H_
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mini_table/extension.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
/* Extension registry: a dynamic data structure that stores a map of:
|
||||
* (upb_MiniTable, number) -> extension info
|
||||
*
|
||||
* upb_decode() uses upb_ExtensionRegistry to look up extensions while parsing
|
||||
* binary format.
|
||||
*
|
||||
* upb_ExtensionRegistry is part of the mini-table (msglayout) family of
|
||||
* objects. Like all mini-table objects, it is suitable for reflection-less
|
||||
* builds that do not want to expose names into the binary.
|
||||
*
|
||||
* Unlike most mini-table types, upb_ExtensionRegistry requires dynamic memory
|
||||
* allocation and dynamic initialization:
|
||||
* * If reflection is being used, then upb_DefPool will construct an appropriate
|
||||
* upb_ExtensionRegistry automatically.
|
||||
* * For a mini-table only build, the user must manually construct the
|
||||
* upb_ExtensionRegistry and populate it with all of the extensions the user
|
||||
* cares about.
|
||||
* * A third alternative is to manually unpack relevant extensions after the
|
||||
* main parse is complete, similar to how Any works. This is perhaps the
|
||||
* nicest solution from the perspective of reducing dependencies, avoiding
|
||||
* dynamic memory allocation, and avoiding the need to parse uninteresting
|
||||
* extensions. The downsides are:
|
||||
* (1) parse errors are not caught during the main parse
|
||||
* (2) the CPU hit of parsing comes during access, which could cause an
|
||||
* undesirable stutter in application performance.
|
||||
*
|
||||
* Users cannot directly get or put into this map. Users can only add the
|
||||
* extensions from a generated module and pass the extension registry to the
|
||||
* binary decoder.
|
||||
*
|
||||
* A upb_DefPool provides a upb_ExtensionRegistry, so any users who use
|
||||
* reflection do not need to populate a upb_ExtensionRegistry directly.
|
||||
*/
|
||||
|
||||
typedef struct upb_ExtensionRegistry upb_ExtensionRegistry;
|
||||
|
||||
// Creates a upb_ExtensionRegistry in the given arena.
|
||||
// The arena must outlive any use of the extreg.
|
||||
UPB_API upb_ExtensionRegistry* upb_ExtensionRegistry_New(upb_Arena* arena);
|
||||
|
||||
UPB_API bool upb_ExtensionRegistry_Add(upb_ExtensionRegistry* r,
|
||||
const upb_MiniTableExtension* e);
|
||||
|
||||
// Adds the given extension info for the array |e| of size |count| into the
|
||||
// registry. If there are any errors, the entire array is backed out.
|
||||
// The extensions must outlive the registry.
|
||||
// Possible errors include OOM or an extension number that already exists.
|
||||
// TODO: There is currently no way to know the exact reason for failure.
|
||||
bool upb_ExtensionRegistry_AddArray(upb_ExtensionRegistry* r,
|
||||
const upb_MiniTableExtension** e,
|
||||
size_t count);
|
||||
|
||||
// Looks up the extension (if any) defined for message type |t| and field
|
||||
// number |num|. Returns the extension if found, otherwise NULL.
|
||||
UPB_API const upb_MiniTableExtension* upb_ExtensionRegistry_Lookup(
|
||||
const upb_ExtensionRegistry* r, const upb_MiniTable* t, uint32_t num);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_EXTENSION_REGISTRY_H_ */
|
||||
98
Pods/gRPC-Core/third_party/upb/upb/mini_table/field.h
generated
vendored
Normal file
98
Pods/gRPC-Core/third_party/upb/upb/mini_table/field.h
generated
vendored
Normal file
@@ -0,0 +1,98 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_FIELD_H_
|
||||
#define UPB_MINI_TABLE_FIELD_H_
|
||||
|
||||
#include "upb/mini_table/internal/field.h"
|
||||
#include "upb/mini_table/internal/message.h"
|
||||
#include "upb/mini_table/internal/sub.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
typedef struct upb_MiniTableField upb_MiniTableField;
|
||||
|
||||
UPB_API_INLINE upb_FieldType
|
||||
upb_MiniTableField_Type(const upb_MiniTableField* field) {
|
||||
if (field->mode & kUpb_LabelFlags_IsAlternate) {
|
||||
if (field->UPB_PRIVATE(descriptortype) == kUpb_FieldType_Int32) {
|
||||
return kUpb_FieldType_Enum;
|
||||
} else if (field->UPB_PRIVATE(descriptortype) == kUpb_FieldType_Bytes) {
|
||||
return kUpb_FieldType_String;
|
||||
} else {
|
||||
UPB_ASSERT(false);
|
||||
}
|
||||
}
|
||||
return (upb_FieldType)field->UPB_PRIVATE(descriptortype);
|
||||
}
|
||||
|
||||
UPB_API_INLINE upb_CType upb_MiniTableField_CType(const upb_MiniTableField* f) {
|
||||
switch (upb_MiniTableField_Type(f)) {
|
||||
case kUpb_FieldType_Double:
|
||||
return kUpb_CType_Double;
|
||||
case kUpb_FieldType_Float:
|
||||
return kUpb_CType_Float;
|
||||
case kUpb_FieldType_Int64:
|
||||
case kUpb_FieldType_SInt64:
|
||||
case kUpb_FieldType_SFixed64:
|
||||
return kUpb_CType_Int64;
|
||||
case kUpb_FieldType_Int32:
|
||||
case kUpb_FieldType_SFixed32:
|
||||
case kUpb_FieldType_SInt32:
|
||||
return kUpb_CType_Int32;
|
||||
case kUpb_FieldType_UInt64:
|
||||
case kUpb_FieldType_Fixed64:
|
||||
return kUpb_CType_UInt64;
|
||||
case kUpb_FieldType_UInt32:
|
||||
case kUpb_FieldType_Fixed32:
|
||||
return kUpb_CType_UInt32;
|
||||
case kUpb_FieldType_Enum:
|
||||
return kUpb_CType_Enum;
|
||||
case kUpb_FieldType_Bool:
|
||||
return kUpb_CType_Bool;
|
||||
case kUpb_FieldType_String:
|
||||
return kUpb_CType_String;
|
||||
case kUpb_FieldType_Bytes:
|
||||
return kUpb_CType_Bytes;
|
||||
case kUpb_FieldType_Group:
|
||||
case kUpb_FieldType_Message:
|
||||
return kUpb_CType_Message;
|
||||
}
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_MiniTableField_IsExtension(
|
||||
const upb_MiniTableField* field) {
|
||||
return field->mode & kUpb_LabelFlags_IsExtension;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_MiniTableField_IsClosedEnum(
|
||||
const upb_MiniTableField* field) {
|
||||
return field->UPB_PRIVATE(descriptortype) == kUpb_FieldType_Enum;
|
||||
}
|
||||
|
||||
UPB_API_INLINE bool upb_MiniTableField_HasPresence(
|
||||
const upb_MiniTableField* field) {
|
||||
if (upb_MiniTableField_IsExtension(field)) {
|
||||
return !upb_IsRepeatedOrMap(field);
|
||||
} else {
|
||||
return field->presence != 0;
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_FIELD_H_ */
|
||||
15
Pods/gRPC-Core/third_party/upb/upb/mini_table/file.h
generated
vendored
Normal file
15
Pods/gRPC-Core/third_party/upb/upb/mini_table/file.h
generated
vendored
Normal file
@@ -0,0 +1,15 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_FILE_H_
|
||||
#define UPB_MINI_TABLE_FILE_H_
|
||||
|
||||
#include "upb/mini_table/internal/file.h"
|
||||
|
||||
typedef struct upb_MiniTableFile upb_MiniTableFile;
|
||||
|
||||
#endif /* UPB_MINI_TABLE_FILE_H_ */
|
||||
58
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/enum.h
generated
vendored
Normal file
58
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/enum.h
generated
vendored
Normal file
@@ -0,0 +1,58 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_INTERNAL_ENUM_H_
|
||||
#define UPB_MINI_TABLE_INTERNAL_ENUM_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_MiniTableEnum {
|
||||
uint32_t mask_limit; // Limit enum value that can be tested with mask.
|
||||
uint32_t value_count; // Number of values after the bitfield.
|
||||
uint32_t data[]; // Bitmask + enumerated values follow.
|
||||
};
|
||||
|
||||
typedef enum {
|
||||
_kUpb_FastEnumCheck_ValueIsInEnum = 0,
|
||||
_kUpb_FastEnumCheck_ValueIsNotInEnum = 1,
|
||||
_kUpb_FastEnumCheck_CannotCheckFast = 2,
|
||||
} _kUpb_FastEnumCheck_Status;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_INLINE _kUpb_FastEnumCheck_Status _upb_MiniTable_CheckEnumValueFast(
|
||||
const struct upb_MiniTableEnum* e, uint32_t val) {
|
||||
if (UPB_UNLIKELY(val >= 64)) return _kUpb_FastEnumCheck_CannotCheckFast;
|
||||
uint64_t mask = e->data[0] | ((uint64_t)e->data[1] << 32);
|
||||
return (mask & (1ULL << val)) ? _kUpb_FastEnumCheck_ValueIsInEnum
|
||||
: _kUpb_FastEnumCheck_ValueIsNotInEnum;
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_MiniTable_CheckEnumValueSlow(
|
||||
const struct upb_MiniTableEnum* e, uint32_t val) {
|
||||
if (val < e->mask_limit) return e->data[val / 32] & (1ULL << (val % 32));
|
||||
// OPT: binary search long lists?
|
||||
const uint32_t* start = &e->data[e->mask_limit / 32];
|
||||
const uint32_t* limit = &e->data[(e->mask_limit / 32) + e->value_count];
|
||||
for (const uint32_t* p = start; p < limit; p++) {
|
||||
if (*p == val) return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_INTERNAL_ENUM_H_ */
|
||||
27
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/extension.h
generated
vendored
Normal file
27
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/extension.h
generated
vendored
Normal file
@@ -0,0 +1,27 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_INTERNAL_EXTENSION_H_
|
||||
#define UPB_MINI_TABLE_INTERNAL_EXTENSION_H_
|
||||
|
||||
#include "upb/mini_table/internal/field.h"
|
||||
#include "upb/mini_table/internal/sub.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_MiniTableExtension {
|
||||
// Do not move this field. We need to be able to alias pointers.
|
||||
struct upb_MiniTableField field;
|
||||
|
||||
const struct upb_MiniTable* extendee;
|
||||
union upb_MiniTableSub sub; // NULL unless submessage or proto2 enum
|
||||
};
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_INTERNAL_EXTENSION_H_ */
|
||||
114
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/field.h
generated
vendored
Normal file
114
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/field.h
generated
vendored
Normal file
@@ -0,0 +1,114 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_INTERNAL_FIELD_H_
|
||||
#define UPB_MINI_TABLE_INTERNAL_FIELD_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_MiniTableField {
|
||||
uint32_t number;
|
||||
uint16_t offset;
|
||||
int16_t presence; // If >0, hasbit_index. If <0, ~oneof_index
|
||||
|
||||
// Indexes into `upb_MiniTable.subs`
|
||||
// Will be set to `kUpb_NoSub` if `descriptortype` != MESSAGE/GROUP/ENUM
|
||||
uint16_t UPB_PRIVATE(submsg_index);
|
||||
|
||||
uint8_t UPB_PRIVATE(descriptortype);
|
||||
|
||||
// upb_FieldMode | upb_LabelFlags | (upb_FieldRep << kUpb_FieldRep_Shift)
|
||||
uint8_t mode;
|
||||
};
|
||||
|
||||
#define kUpb_NoSub ((uint16_t)-1)
|
||||
|
||||
typedef enum {
|
||||
kUpb_FieldMode_Map = 0,
|
||||
kUpb_FieldMode_Array = 1,
|
||||
kUpb_FieldMode_Scalar = 2,
|
||||
} upb_FieldMode;
|
||||
|
||||
// Mask to isolate the upb_FieldMode from field.mode.
|
||||
#define kUpb_FieldMode_Mask 3
|
||||
|
||||
// Extra flags on the mode field.
|
||||
typedef enum {
|
||||
kUpb_LabelFlags_IsPacked = 4,
|
||||
kUpb_LabelFlags_IsExtension = 8,
|
||||
// Indicates that this descriptor type is an "alternate type":
|
||||
// - for Int32, this indicates that the actual type is Enum (but was
|
||||
// rewritten to Int32 because it is an open enum that requires no check).
|
||||
// - for Bytes, this indicates that the actual type is String (but does
|
||||
// not require any UTF-8 check).
|
||||
kUpb_LabelFlags_IsAlternate = 16,
|
||||
} upb_LabelFlags;
|
||||
|
||||
// Note: we sort by this number when calculating layout order.
|
||||
typedef enum {
|
||||
kUpb_FieldRep_1Byte = 0,
|
||||
kUpb_FieldRep_4Byte = 1,
|
||||
kUpb_FieldRep_StringView = 2,
|
||||
kUpb_FieldRep_8Byte = 3,
|
||||
|
||||
kUpb_FieldRep_NativePointer =
|
||||
UPB_SIZE(kUpb_FieldRep_4Byte, kUpb_FieldRep_8Byte),
|
||||
kUpb_FieldRep_Max = kUpb_FieldRep_8Byte,
|
||||
} upb_FieldRep;
|
||||
|
||||
#define kUpb_FieldRep_Shift 6
|
||||
|
||||
UPB_INLINE upb_FieldRep
|
||||
_upb_MiniTableField_GetRep(const struct upb_MiniTableField* field) {
|
||||
return (upb_FieldRep)(field->mode >> kUpb_FieldRep_Shift);
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_INLINE upb_FieldMode
|
||||
upb_FieldMode_Get(const struct upb_MiniTableField* field) {
|
||||
return (upb_FieldMode)(field->mode & 3);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_MiniTableField_CheckIsArray(
|
||||
const struct upb_MiniTableField* field) {
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_NativePointer);
|
||||
UPB_ASSUME(upb_FieldMode_Get(field) == kUpb_FieldMode_Array);
|
||||
UPB_ASSUME(field->presence == 0);
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_MiniTableField_CheckIsMap(
|
||||
const struct upb_MiniTableField* field) {
|
||||
UPB_ASSUME(_upb_MiniTableField_GetRep(field) == kUpb_FieldRep_NativePointer);
|
||||
UPB_ASSUME(upb_FieldMode_Get(field) == kUpb_FieldMode_Map);
|
||||
UPB_ASSUME(field->presence == 0);
|
||||
}
|
||||
|
||||
UPB_INLINE bool upb_IsRepeatedOrMap(const struct upb_MiniTableField* field) {
|
||||
// This works because upb_FieldMode has no value 3.
|
||||
return !(field->mode & kUpb_FieldMode_Scalar);
|
||||
}
|
||||
|
||||
UPB_INLINE bool upb_IsSubMessage(const struct upb_MiniTableField* field) {
|
||||
return field->UPB_PRIVATE(descriptortype) == kUpb_FieldType_Message ||
|
||||
field->UPB_PRIVATE(descriptortype) == kUpb_FieldType_Group;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_INTERNAL_FIELD_H_ */
|
||||
29
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/file.h
generated
vendored
Normal file
29
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/file.h
generated
vendored
Normal file
@@ -0,0 +1,29 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_INTERNAL_FILE_H_
|
||||
#define UPB_MINI_TABLE_INTERNAL_FILE_H_
|
||||
|
||||
#include "upb/mini_table/internal/enum.h"
|
||||
#include "upb/mini_table/internal/extension.h"
|
||||
#include "upb/mini_table/internal/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_MiniTableFile {
|
||||
const struct upb_MiniTable** msgs;
|
||||
const struct upb_MiniTableEnum** enums;
|
||||
const struct upb_MiniTableExtension** exts;
|
||||
int msg_count;
|
||||
int enum_count;
|
||||
int ext_count;
|
||||
};
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_INTERNAL_FILE_H_ */
|
||||
19
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/message.c
generated
vendored
Normal file
19
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/message.c
generated
vendored
Normal file
@@ -0,0 +1,19 @@
|
||||
// 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/mini_table/internal/message.h"
|
||||
|
||||
const struct upb_MiniTable _kUpb_MiniTable_Empty = {
|
||||
.subs = NULL,
|
||||
.fields = NULL,
|
||||
.size = 0,
|
||||
.field_count = 0,
|
||||
.ext = kUpb_ExtMode_NonExtendable,
|
||||
.dense_below = 0,
|
||||
.table_mask = -1,
|
||||
.required_count = 0,
|
||||
};
|
||||
88
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/message.h
generated
vendored
Normal file
88
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/message.h
generated
vendored
Normal file
@@ -0,0 +1,88 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_INTERNAL_MESSAGE_H_
|
||||
#define UPB_MINI_TABLE_INTERNAL_MESSAGE_H_
|
||||
|
||||
#include "upb/message/types.h"
|
||||
#include "upb/mini_table/internal/field.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_Decoder;
|
||||
typedef const char* _upb_FieldParser(struct upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg, intptr_t table,
|
||||
uint64_t hasbits, uint64_t data);
|
||||
typedef struct {
|
||||
uint64_t field_data;
|
||||
_upb_FieldParser* field_parser;
|
||||
} _upb_FastTable_Entry;
|
||||
|
||||
typedef enum {
|
||||
kUpb_ExtMode_NonExtendable = 0, // Non-extendable message.
|
||||
kUpb_ExtMode_Extendable = 1, // Normal extendable message.
|
||||
kUpb_ExtMode_IsMessageSet = 2, // MessageSet message.
|
||||
kUpb_ExtMode_IsMessageSet_ITEM =
|
||||
3, // MessageSet item (temporary only, see decode.c)
|
||||
|
||||
// During table building we steal a bit to indicate that the message is a map
|
||||
// entry. *Only* used during table building!
|
||||
kUpb_ExtMode_IsMapEntry = 4,
|
||||
} upb_ExtMode;
|
||||
|
||||
union upb_MiniTableSub;
|
||||
|
||||
// upb_MiniTable represents the memory layout of a given upb_MessageDef.
|
||||
// The members are public so generated code can initialize them,
|
||||
// but users MUST NOT directly read or write any of its members.
|
||||
struct upb_MiniTable {
|
||||
const union upb_MiniTableSub* subs;
|
||||
const struct upb_MiniTableField* fields;
|
||||
|
||||
// Must be aligned to sizeof(void*). Doesn't include internal members like
|
||||
// unknown fields, extension dict, pointer to msglayout, etc.
|
||||
uint16_t size;
|
||||
|
||||
uint16_t field_count;
|
||||
uint8_t ext; // upb_ExtMode, declared as uint8_t so sizeof(ext) == 1
|
||||
uint8_t dense_below;
|
||||
uint8_t table_mask;
|
||||
uint8_t required_count; // Required fields have the lowest hasbits.
|
||||
|
||||
// To statically initialize the tables of variable length, we need a flexible
|
||||
// array member, and we need to compile in gnu99 mode (constant initialization
|
||||
// of flexible array members is a GNU extension, not in C99 unfortunately.
|
||||
_upb_FastTable_Entry fasttable[];
|
||||
};
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// A MiniTable for an empty message, used for unlinked sub-messages.
|
||||
extern const struct upb_MiniTable _kUpb_MiniTable_Empty;
|
||||
|
||||
// Computes a bitmask in which the |l->required_count| lowest bits are set,
|
||||
// except that we skip the lowest bit (because upb never uses hasbit 0).
|
||||
//
|
||||
// Sample output:
|
||||
// requiredmask(1) => 0b10 (0x2)
|
||||
// requiredmask(5) => 0b111110 (0x3e)
|
||||
UPB_INLINE uint64_t upb_MiniTable_requiredmask(const struct upb_MiniTable* l) {
|
||||
int n = l->required_count;
|
||||
assert(0 < n && n <= 63);
|
||||
return ((1ULL << n) - 1) << 1;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_INTERNAL_MESSAGE_H_ */
|
||||
18
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/sub.h
generated
vendored
Normal file
18
Pods/gRPC-Core/third_party/upb/upb/mini_table/internal/sub.h
generated
vendored
Normal file
@@ -0,0 +1,18 @@
|
||||
// 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
|
||||
#ifndef UPB_MINI_TABLE_INTERNAL_SUB_H_
|
||||
#define UPB_MINI_TABLE_INTERNAL_SUB_H_
|
||||
|
||||
#include "upb/mini_table/internal/enum.h"
|
||||
#include "upb/mini_table/internal/message.h"
|
||||
|
||||
union upb_MiniTableSub {
|
||||
const struct upb_MiniTable* submsg;
|
||||
const struct upb_MiniTableEnum* subenum;
|
||||
};
|
||||
|
||||
#endif /* UPB_MINI_TABLE_INTERNAL_SUB_H_ */
|
||||
77
Pods/gRPC-Core/third_party/upb/upb/mini_table/message.c
generated
vendored
Normal file
77
Pods/gRPC-Core/third_party/upb/upb/mini_table/message.c
generated
vendored
Normal file
@@ -0,0 +1,77 @@
|
||||
// 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/mini_table/message.h"
|
||||
|
||||
#include <inttypes.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mini_table/internal/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
const upb_MiniTableField* upb_MiniTable_FindFieldByNumber(
|
||||
const upb_MiniTable* t, uint32_t number) {
|
||||
const size_t i = ((size_t)number) - 1; // 0 wraps to SIZE_MAX
|
||||
|
||||
// Ideal case: index into dense fields
|
||||
if (i < t->dense_below) {
|
||||
UPB_ASSERT(t->fields[i].number == number);
|
||||
return &t->fields[i];
|
||||
}
|
||||
|
||||
// Slow case: binary search
|
||||
int lo = t->dense_below;
|
||||
int hi = t->field_count - 1;
|
||||
while (lo <= hi) {
|
||||
int mid = (lo + hi) / 2;
|
||||
uint32_t num = t->fields[mid].number;
|
||||
if (num < number) {
|
||||
lo = mid + 1;
|
||||
continue;
|
||||
}
|
||||
if (num > number) {
|
||||
hi = mid - 1;
|
||||
continue;
|
||||
}
|
||||
return &t->fields[mid];
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static bool upb_MiniTable_Is_Oneof(const upb_MiniTableField* f) {
|
||||
return f->presence < 0;
|
||||
}
|
||||
|
||||
const upb_MiniTableField* upb_MiniTable_GetOneof(const upb_MiniTable* m,
|
||||
const upb_MiniTableField* f) {
|
||||
if (UPB_UNLIKELY(!upb_MiniTable_Is_Oneof(f))) {
|
||||
return NULL;
|
||||
}
|
||||
const upb_MiniTableField* ptr = &m->fields[0];
|
||||
const upb_MiniTableField* end = &m->fields[m->field_count];
|
||||
while (++ptr < end) {
|
||||
if (ptr->presence == (*f).presence) {
|
||||
return ptr;
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
bool upb_MiniTable_NextOneofField(const upb_MiniTable* m,
|
||||
const upb_MiniTableField** f) {
|
||||
const upb_MiniTableField* ptr = *f;
|
||||
const upb_MiniTableField* end = &m->fields[m->field_count];
|
||||
while (++ptr < end) {
|
||||
if (ptr->presence == (*f)->presence) {
|
||||
*f = ptr;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
85
Pods/gRPC-Core/third_party/upb/upb/mini_table/message.h
generated
vendored
Normal file
85
Pods/gRPC-Core/third_party/upb/upb/mini_table/message.h
generated
vendored
Normal file
@@ -0,0 +1,85 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_MESSAGE_H_
|
||||
#define UPB_MINI_TABLE_MESSAGE_H_
|
||||
|
||||
#include "upb/mini_table/enum.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
#include "upb/mini_table/internal/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
typedef struct upb_MiniTable upb_MiniTable;
|
||||
|
||||
UPB_API const upb_MiniTableField* upb_MiniTable_FindFieldByNumber(
|
||||
const upb_MiniTable* table, uint32_t number);
|
||||
|
||||
UPB_API_INLINE const upb_MiniTableField* upb_MiniTable_GetFieldByIndex(
|
||||
const upb_MiniTable* t, uint32_t index) {
|
||||
return &t->fields[index];
|
||||
}
|
||||
|
||||
// Returns the MiniTable for this message field. If the field is unlinked,
|
||||
// returns NULL.
|
||||
UPB_API_INLINE const upb_MiniTable* upb_MiniTable_GetSubMessageTable(
|
||||
const upb_MiniTable* mini_table, const upb_MiniTableField* field) {
|
||||
UPB_ASSERT(upb_MiniTableField_CType(field) == kUpb_CType_Message);
|
||||
const upb_MiniTable* ret =
|
||||
mini_table->subs[field->UPB_PRIVATE(submsg_index)].submsg;
|
||||
UPB_ASSUME(ret);
|
||||
return ret == &_kUpb_MiniTable_Empty ? NULL : ret;
|
||||
}
|
||||
|
||||
// Returns the MiniTableEnum for this enum field. If the field is unlinked,
|
||||
// returns NULL.
|
||||
UPB_API_INLINE const upb_MiniTableEnum* upb_MiniTable_GetSubEnumTable(
|
||||
const upb_MiniTable* mini_table, const upb_MiniTableField* field) {
|
||||
UPB_ASSERT(upb_MiniTableField_CType(field) == kUpb_CType_Enum);
|
||||
return mini_table->subs[field->UPB_PRIVATE(submsg_index)].subenum;
|
||||
}
|
||||
|
||||
// Returns true if this MiniTable field is linked to a MiniTable for the
|
||||
// sub-message.
|
||||
UPB_API_INLINE bool upb_MiniTable_MessageFieldIsLinked(
|
||||
const upb_MiniTable* mini_table, const upb_MiniTableField* field) {
|
||||
return upb_MiniTable_GetSubMessageTable(mini_table, field) != NULL;
|
||||
}
|
||||
|
||||
// If this field is in a oneof, returns the first field in the oneof.
|
||||
//
|
||||
// Otherwise returns NULL.
|
||||
//
|
||||
// Usage:
|
||||
// const upb_MiniTableField* field = upb_MiniTable_GetOneof(m, f);
|
||||
// do {
|
||||
// ..
|
||||
// } while (upb_MiniTable_NextOneofField(m, &field);
|
||||
//
|
||||
const upb_MiniTableField* upb_MiniTable_GetOneof(const upb_MiniTable* m,
|
||||
const upb_MiniTableField* f);
|
||||
|
||||
// Iterates to the next field in the oneof. If this is the last field in the
|
||||
// oneof, returns false. The ordering of fields in the oneof is not
|
||||
// guaranteed.
|
||||
// REQUIRES: |f| is the field initialized by upb_MiniTable_GetOneof and updated
|
||||
// by prior upb_MiniTable_NextOneofField calls.
|
||||
bool upb_MiniTable_NextOneofField(const upb_MiniTable* m,
|
||||
const upb_MiniTableField** f);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_MINI_TABLE_MESSAGE_H_ */
|
||||
15
Pods/gRPC-Core/third_party/upb/upb/mini_table/sub.h
generated
vendored
Normal file
15
Pods/gRPC-Core/third_party/upb/upb/mini_table/sub.h
generated
vendored
Normal file
@@ -0,0 +1,15 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_MINI_TABLE_SUB_H_
|
||||
#define UPB_MINI_TABLE_SUB_H_
|
||||
|
||||
#include "upb/mini_table/internal/sub.h"
|
||||
|
||||
typedef union upb_MiniTableSub upb_MiniTableSub;
|
||||
|
||||
#endif /* UPB_MINI_TABLE_INTERNAL_SUB_H_ */
|
||||
83
Pods/gRPC-Core/third_party/upb/upb/port/atomic.h
generated
vendored
Normal file
83
Pods/gRPC-Core/third_party/upb/upb/port/atomic.h
generated
vendored
Normal file
@@ -0,0 +1,83 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_PORT_ATOMIC_H_
|
||||
#define UPB_PORT_ATOMIC_H_
|
||||
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef UPB_USE_C11_ATOMICS
|
||||
|
||||
// IWYU pragma: begin_exports
|
||||
#include <stdatomic.h>
|
||||
#include <stdbool.h>
|
||||
// IWYU pragma: end_exports
|
||||
|
||||
#define upb_Atomic_Init(addr, val) atomic_init(addr, val)
|
||||
#define upb_Atomic_Load(addr, order) atomic_load_explicit(addr, order)
|
||||
#define upb_Atomic_Store(addr, val, order) \
|
||||
atomic_store_explicit(addr, val, order)
|
||||
#define upb_Atomic_Add(addr, val, order) \
|
||||
atomic_fetch_add_explicit(addr, val, order)
|
||||
#define upb_Atomic_Sub(addr, val, order) \
|
||||
atomic_fetch_sub_explicit(addr, val, order)
|
||||
#define upb_Atomic_Exchange(addr, val, order) \
|
||||
atomic_exchange_explicit(addr, val, order)
|
||||
#define upb_Atomic_CompareExchangeStrong(addr, expected, desired, \
|
||||
success_order, failure_order) \
|
||||
atomic_compare_exchange_strong_explicit(addr, expected, desired, \
|
||||
success_order, failure_order)
|
||||
#define upb_Atomic_CompareExchangeWeak(addr, expected, desired, success_order, \
|
||||
failure_order) \
|
||||
atomic_compare_exchange_weak_explicit(addr, expected, desired, \
|
||||
success_order, failure_order)
|
||||
|
||||
#else // !UPB_USE_C11_ATOMICS
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#define upb_Atomic_Init(addr, val) (*addr = val)
|
||||
#define upb_Atomic_Load(addr, order) (*addr)
|
||||
#define upb_Atomic_Store(addr, val, order) (*(addr) = val)
|
||||
#define upb_Atomic_Add(addr, val, order) (*(addr) += val)
|
||||
#define upb_Atomic_Sub(addr, val, order) (*(addr) -= val)
|
||||
|
||||
UPB_INLINE void* _upb_NonAtomic_Exchange(void* addr, void* value) {
|
||||
void* old;
|
||||
memcpy(&old, addr, sizeof(value));
|
||||
memcpy(addr, &value, sizeof(value));
|
||||
return old;
|
||||
}
|
||||
|
||||
#define upb_Atomic_Exchange(addr, val, order) _upb_NonAtomic_Exchange(addr, val)
|
||||
|
||||
// `addr` and `expected` are logically double pointers.
|
||||
UPB_INLINE bool _upb_NonAtomic_CompareExchangeStrongP(void* addr,
|
||||
void* expected,
|
||||
void* desired) {
|
||||
if (memcmp(addr, expected, sizeof(desired)) == 0) {
|
||||
memcpy(addr, &desired, sizeof(desired));
|
||||
return true;
|
||||
} else {
|
||||
memcpy(expected, addr, sizeof(desired));
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
#define upb_Atomic_CompareExchangeStrong(addr, expected, desired, \
|
||||
success_order, failure_order) \
|
||||
_upb_NonAtomic_CompareExchangeStrongP((void*)addr, (void*)expected, \
|
||||
(void*)desired)
|
||||
#define upb_Atomic_CompareExchangeWeak(addr, expected, desired, success_order, \
|
||||
failure_order) \
|
||||
upb_Atomic_CompareExchangeStrong(addr, expected, desired, 0, 0)
|
||||
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_PORT_ATOMIC_H_
|
||||
328
Pods/gRPC-Core/third_party/upb/upb/port/def.inc
generated
vendored
Normal file
328
Pods/gRPC-Core/third_party/upb/upb/port/def.inc
generated
vendored
Normal file
@@ -0,0 +1,328 @@
|
||||
// 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
|
||||
|
||||
/*
|
||||
* This is where we define internal portability macros used across upb.
|
||||
*
|
||||
* All of these macros are undef'd in undef.inc to avoid leaking them to users.
|
||||
*
|
||||
* The correct usage is:
|
||||
*
|
||||
* #include "upb/foobar.h"
|
||||
* #include "upb/baz.h"
|
||||
*
|
||||
* // MUST be last included header.
|
||||
* #include "upb/port/def.inc"
|
||||
*
|
||||
* // Code for this file.
|
||||
* // <...>
|
||||
*
|
||||
* // Can be omitted for .c files, required for .h.
|
||||
* #include "upb/port/undef.inc"
|
||||
*
|
||||
* This file is private and must not be included by users!
|
||||
*/
|
||||
|
||||
#if !((defined(__STDC_VERSION__) && __STDC_VERSION__ >= 199901L) || \
|
||||
(defined(__cplusplus) && __cplusplus >= 201402L) || \
|
||||
(defined(_MSC_VER) && _MSC_VER >= 1900))
|
||||
#error upb requires C99 or C++14 or MSVC >= 2015.
|
||||
#endif
|
||||
|
||||
// Portable check for GCC minimum version:
|
||||
// https://gcc.gnu.org/onlinedocs/cpp/Common-Predefined-Macros.html
|
||||
#if defined(__GNUC__) && defined(__GNUC_MINOR__) && defined(__GNUC_PATCHLEVEL__)
|
||||
#define UPB_GNUC_MIN(x, y) \
|
||||
(__GNUC__ > (x) || __GNUC__ == (x) && __GNUC_MINOR__ >= (y))
|
||||
#else
|
||||
#define UPB_GNUC_MIN(x, y) 0
|
||||
#endif
|
||||
|
||||
#include <assert.h>
|
||||
#include <setjmp.h>
|
||||
#include <stdbool.h>
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
|
||||
#ifndef UINTPTR_MAX
|
||||
Error, UINTPTR_MAX is undefined
|
||||
#endif
|
||||
|
||||
#if UINTPTR_MAX == 0xffffffff
|
||||
#define UPB_SIZE(size32, size64) size32
|
||||
#else
|
||||
#define UPB_SIZE(size32, size64) size64
|
||||
#endif
|
||||
|
||||
/* If we always read/write as a consistent type to each address, this shouldn't
|
||||
* violate aliasing.
|
||||
*/
|
||||
#define UPB_PTR_AT(msg, ofs, type) ((type*)((char*)(msg) + (ofs)))
|
||||
|
||||
#define UPB_MAPTYPE_STRING 0
|
||||
|
||||
// UPB_EXPORT: always generate a public symbol.
|
||||
#if defined(__GNUC__) || defined(__clang__)
|
||||
#define UPB_EXPORT __attribute__((visibility("default"))) __attribute__((used))
|
||||
#else
|
||||
#define UPB_EXPORT
|
||||
#endif
|
||||
|
||||
// UPB_INLINE: inline if possible, emit standalone code if required.
|
||||
#ifdef __cplusplus
|
||||
#define UPB_INLINE inline
|
||||
#elif defined (__GNUC__) || defined(__clang__)
|
||||
#define UPB_INLINE static __inline__
|
||||
#else
|
||||
#define UPB_INLINE static
|
||||
#endif
|
||||
|
||||
#ifdef UPB_BUILD_API
|
||||
#define UPB_API UPB_EXPORT
|
||||
#define UPB_API_INLINE UPB_EXPORT
|
||||
#else
|
||||
#define UPB_API
|
||||
#define UPB_API_INLINE UPB_INLINE
|
||||
#endif
|
||||
|
||||
#define UPB_MALLOC_ALIGN 8
|
||||
#define UPB_ALIGN_UP(size, align) (((size) + (align) - 1) / (align) * (align))
|
||||
#define UPB_ALIGN_DOWN(size, align) ((size) / (align) * (align))
|
||||
#define UPB_ALIGN_MALLOC(size) UPB_ALIGN_UP(size, UPB_MALLOC_ALIGN)
|
||||
#ifdef __clang__
|
||||
#define UPB_ALIGN_OF(type) _Alignof(type)
|
||||
#else
|
||||
#define UPB_ALIGN_OF(type) offsetof (struct { char c; type member; }, member)
|
||||
#endif
|
||||
|
||||
// Hints to the compiler about likely/unlikely branches.
|
||||
#if defined (__GNUC__) || defined(__clang__)
|
||||
#define UPB_LIKELY(x) __builtin_expect((bool)(x), 1)
|
||||
#define UPB_UNLIKELY(x) __builtin_expect((bool)(x), 0)
|
||||
#else
|
||||
#define UPB_LIKELY(x) (x)
|
||||
#define UPB_UNLIKELY(x) (x)
|
||||
#endif
|
||||
|
||||
// Macros for function attributes on compilers that support them.
|
||||
#ifdef __GNUC__
|
||||
#define UPB_FORCEINLINE __inline__ __attribute__((always_inline))
|
||||
#define UPB_NOINLINE __attribute__((noinline))
|
||||
#define UPB_NORETURN __attribute__((__noreturn__))
|
||||
#define UPB_PRINTF(str, first_vararg) __attribute__((format (printf, str, first_vararg)))
|
||||
#elif defined(_MSC_VER)
|
||||
#define UPB_NOINLINE
|
||||
#define UPB_FORCEINLINE
|
||||
#define UPB_NORETURN __declspec(noreturn)
|
||||
#define UPB_PRINTF(str, first_vararg)
|
||||
#else /* !defined(__GNUC__) */
|
||||
#define UPB_FORCEINLINE
|
||||
#define UPB_NOINLINE
|
||||
#define UPB_NORETURN
|
||||
#define UPB_PRINTF(str, first_vararg)
|
||||
#endif
|
||||
|
||||
#define UPB_MAX(x, y) ((x) > (y) ? (x) : (y))
|
||||
#define UPB_MIN(x, y) ((x) < (y) ? (x) : (y))
|
||||
|
||||
#define UPB_UNUSED(var) (void)var
|
||||
|
||||
// UPB_ASSUME(): in release mode, we tell the compiler to assume this is true.
|
||||
#ifdef NDEBUG
|
||||
#ifdef __GNUC__
|
||||
#define UPB_ASSUME(expr) if (!(expr)) __builtin_unreachable()
|
||||
#elif defined _MSC_VER
|
||||
#define UPB_ASSUME(expr) if (!(expr)) __assume(0)
|
||||
#else
|
||||
#define UPB_ASSUME(expr) do {} while (false && (expr))
|
||||
#endif
|
||||
#else
|
||||
#define UPB_ASSUME(expr) assert(expr)
|
||||
#endif
|
||||
|
||||
/* UPB_ASSERT(): in release mode, we use the expression without letting it be
|
||||
* evaluated. This prevents "unused variable" warnings. */
|
||||
#ifdef NDEBUG
|
||||
#define UPB_ASSERT(expr) do {} while (false && (expr))
|
||||
#else
|
||||
#define UPB_ASSERT(expr) assert(expr)
|
||||
#endif
|
||||
|
||||
#if defined(__GNUC__) || defined(__clang__)
|
||||
#define UPB_UNREACHABLE() do { assert(0); __builtin_unreachable(); } while(0)
|
||||
#elif defined(_MSC_VER)
|
||||
#define UPB_UNREACHABLE() \
|
||||
do { \
|
||||
assert(0); \
|
||||
__assume(0); \
|
||||
} while (0)
|
||||
#else
|
||||
#define UPB_UNREACHABLE() do { assert(0); } while(0)
|
||||
#endif
|
||||
|
||||
/* UPB_SETJMP() / UPB_LONGJMP(): avoid setting/restoring signal mask. */
|
||||
#ifdef __APPLE__
|
||||
#define UPB_SETJMP(buf) _setjmp(buf)
|
||||
#define UPB_LONGJMP(buf, val) _longjmp(buf, val)
|
||||
#else
|
||||
#define UPB_SETJMP(buf) setjmp(buf)
|
||||
#define UPB_LONGJMP(buf, val) longjmp(buf, val)
|
||||
#endif
|
||||
|
||||
#ifdef __GNUC__
|
||||
#define UPB_USE_C11_ATOMICS
|
||||
#define UPB_ATOMIC(T) _Atomic(T)
|
||||
#else
|
||||
#define UPB_ATOMIC(T) T
|
||||
#endif
|
||||
|
||||
/* UPB_PTRADD(ptr, ofs): add pointer while avoiding "NULL + 0" UB */
|
||||
#define UPB_PTRADD(ptr, ofs) ((ofs) ? (ptr) + (ofs) : (ptr))
|
||||
|
||||
#define UPB_PRIVATE(x) x##_dont_copy_me__upb_internal_use_only
|
||||
|
||||
/* Configure whether fasttable is switched on or not. *************************/
|
||||
|
||||
#ifdef __has_attribute
|
||||
#define UPB_HAS_ATTRIBUTE(x) __has_attribute(x)
|
||||
#else
|
||||
#define UPB_HAS_ATTRIBUTE(x) 0
|
||||
#endif
|
||||
|
||||
#if UPB_HAS_ATTRIBUTE(musttail)
|
||||
#define UPB_MUSTTAIL __attribute__((musttail))
|
||||
#else
|
||||
#define UPB_MUSTTAIL
|
||||
#endif
|
||||
|
||||
#undef UPB_HAS_ATTRIBUTE
|
||||
|
||||
/* This check is not fully robust: it does not require that we have "musttail"
|
||||
* support available. We need tail calls to avoid consuming arbitrary amounts
|
||||
* of stack space.
|
||||
*
|
||||
* GCC/Clang can mostly be trusted to generate tail calls as long as
|
||||
* optimization is enabled, but, debug builds will not generate tail calls
|
||||
* unless "musttail" is available.
|
||||
*
|
||||
* We should probably either:
|
||||
* 1. require that the compiler supports musttail.
|
||||
* 2. add some fallback code for when musttail isn't available (ie. return
|
||||
* instead of tail calling). This is safe and portable, but this comes at
|
||||
* a CPU cost.
|
||||
*/
|
||||
#if (defined(__x86_64__) || defined(__aarch64__)) && defined(__GNUC__)
|
||||
#define UPB_FASTTABLE_SUPPORTED 1
|
||||
#else
|
||||
#define UPB_FASTTABLE_SUPPORTED 0
|
||||
#endif
|
||||
|
||||
/* define UPB_ENABLE_FASTTABLE to force fast table support.
|
||||
* This is useful when we want to ensure we are really getting fasttable,
|
||||
* for example for testing or benchmarking. */
|
||||
#if defined(UPB_ENABLE_FASTTABLE)
|
||||
#if !UPB_FASTTABLE_SUPPORTED
|
||||
#error fasttable is x86-64/ARM64 only and requires GCC or Clang.
|
||||
#endif
|
||||
#define UPB_FASTTABLE 1
|
||||
/* Define UPB_TRY_ENABLE_FASTTABLE to use fasttable if possible.
|
||||
* This is useful for releasing code that might be used on multiple platforms,
|
||||
* for example the PHP or Ruby C extensions. */
|
||||
#elif defined(UPB_TRY_ENABLE_FASTTABLE)
|
||||
#define UPB_FASTTABLE UPB_FASTTABLE_SUPPORTED
|
||||
#else
|
||||
#define UPB_FASTTABLE 0
|
||||
#endif
|
||||
|
||||
/* UPB_FASTTABLE_INIT() allows protos compiled for fasttable to gracefully
|
||||
* degrade to non-fasttable if the runtime or platform do not support it. */
|
||||
#if !UPB_FASTTABLE
|
||||
#define UPB_FASTTABLE_INIT(...)
|
||||
#define UPB_FASTTABLE_MASK(mask) -1
|
||||
#else
|
||||
#define UPB_FASTTABLE_INIT(...) __VA_ARGS__
|
||||
#define UPB_FASTTABLE_MASK(mask) mask
|
||||
#endif
|
||||
|
||||
#undef UPB_FASTTABLE_SUPPORTED
|
||||
|
||||
/* ASAN poisoning (for arena).
|
||||
* If using UPB from an interpreted language like Ruby, a build of the
|
||||
* interpreter compiled with ASAN enabled must be used in order to get sane and
|
||||
* expected behavior.
|
||||
*/
|
||||
|
||||
/* Due to preprocessor limitations, the conditional logic for setting
|
||||
* UPN_CLANG_ASAN below cannot be consolidated into a portable one-liner.
|
||||
* See https://gcc.gnu.org/onlinedocs/cpp/_005f_005fhas_005fattribute.html.
|
||||
*/
|
||||
#if defined(__has_feature)
|
||||
#if __has_feature(address_sanitizer)
|
||||
#define UPB_CLANG_ASAN 1
|
||||
#else
|
||||
#define UPB_CLANG_ASAN 0
|
||||
#endif
|
||||
#else
|
||||
#define UPB_CLANG_ASAN 0
|
||||
#endif
|
||||
|
||||
#if defined(__SANITIZE_ADDRESS__) || UPB_CLANG_ASAN
|
||||
#define UPB_ASAN 1
|
||||
#define UPB_ASAN_GUARD_SIZE 32
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
void __asan_poison_memory_region(void const volatile *addr, size_t size);
|
||||
void __asan_unpoison_memory_region(void const volatile *addr, size_t size);
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
#define UPB_POISON_MEMORY_REGION(addr, size) \
|
||||
__asan_poison_memory_region((addr), (size))
|
||||
#define UPB_UNPOISON_MEMORY_REGION(addr, size) \
|
||||
__asan_unpoison_memory_region((addr), (size))
|
||||
#else
|
||||
#define UPB_ASAN 0
|
||||
#define UPB_ASAN_GUARD_SIZE 0
|
||||
#define UPB_POISON_MEMORY_REGION(addr, size) \
|
||||
((void)(addr), (void)(size))
|
||||
#define UPB_UNPOISON_MEMORY_REGION(addr, size) \
|
||||
((void)(addr), (void)(size))
|
||||
#endif
|
||||
|
||||
/* Disable proto2 arena behavior (TEMPORARY) **********************************/
|
||||
|
||||
#ifdef UPB_DISABLE_PROTO2_ENUM_CHECKING
|
||||
#define UPB_TREAT_PROTO2_ENUMS_LIKE_PROTO3 1
|
||||
#else
|
||||
#define UPB_TREAT_PROTO2_ENUMS_LIKE_PROTO3 0
|
||||
#endif
|
||||
|
||||
#if defined(__cplusplus)
|
||||
#if defined(__clang__) || UPB_GNUC_MIN(6, 0)
|
||||
// https://gcc.gnu.org/gcc-6/changes.html
|
||||
#if __cplusplus >= 201402L
|
||||
#define UPB_DEPRECATED [[deprecated]]
|
||||
#else
|
||||
#define UPB_DEPRECATED __attribute__((deprecated))
|
||||
#endif
|
||||
#else
|
||||
#define UPB_DEPRECATED
|
||||
#endif
|
||||
#else
|
||||
#define UPB_DEPRECATED
|
||||
#endif
|
||||
|
||||
// begin:google_only
|
||||
// #define UPB_IS_GOOGLE3
|
||||
// end:google_only
|
||||
|
||||
#if defined(UPB_IS_GOOGLE3) && !defined(UPB_BOOTSTRAP_STAGE0)
|
||||
#define UPB_DESC(sym) proto2_##sym
|
||||
#else
|
||||
#define UPB_DESC(sym) google_protobuf_##sym
|
||||
#endif
|
||||
55
Pods/gRPC-Core/third_party/upb/upb/port/undef.inc
generated
vendored
Normal file
55
Pods/gRPC-Core/third_party/upb/upb/port/undef.inc
generated
vendored
Normal file
@@ -0,0 +1,55 @@
|
||||
// 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
|
||||
|
||||
// This should #undef all macros #defined in def.inc
|
||||
|
||||
#undef UPB_SIZE
|
||||
#undef UPB_PTR_AT
|
||||
#undef UPB_MAPTYPE_STRING
|
||||
#undef UPB_EXPORT
|
||||
#undef UPB_INLINE
|
||||
#undef UPB_API
|
||||
#undef UPB_API_INLINE
|
||||
#undef UPB_ALIGN_UP
|
||||
#undef UPB_ALIGN_DOWN
|
||||
#undef UPB_ALIGN_MALLOC
|
||||
#undef UPB_ALIGN_OF
|
||||
#undef UPB_MALLOC_ALIGN
|
||||
#undef UPB_LIKELY
|
||||
#undef UPB_UNLIKELY
|
||||
#undef UPB_FORCEINLINE
|
||||
#undef UPB_NOINLINE
|
||||
#undef UPB_NORETURN
|
||||
#undef UPB_PRINTF
|
||||
#undef UPB_MAX
|
||||
#undef UPB_MIN
|
||||
#undef UPB_UNUSED
|
||||
#undef UPB_ASSUME
|
||||
#undef UPB_ASSERT
|
||||
#undef UPB_UNREACHABLE
|
||||
#undef UPB_SETJMP
|
||||
#undef UPB_LONGJMP
|
||||
#undef UPB_PTRADD
|
||||
#undef UPB_MUSTTAIL
|
||||
#undef UPB_FASTTABLE_SUPPORTED
|
||||
#undef UPB_FASTTABLE_MASK
|
||||
#undef UPB_FASTTABLE
|
||||
#undef UPB_FASTTABLE_INIT
|
||||
#undef UPB_POISON_MEMORY_REGION
|
||||
#undef UPB_UNPOISON_MEMORY_REGION
|
||||
#undef UPB_ASAN
|
||||
#undef UPB_ASAN_GUARD_SIZE
|
||||
#undef UPB_CLANG_ASAN
|
||||
#undef UPB_TREAT_PROTO2_ENUMS_LIKE_PROTO3
|
||||
#undef UPB_DEPRECATED
|
||||
#undef UPB_GNUC_MIN
|
||||
#undef UPB_DESCRIPTOR_UPB_H_FILENAME
|
||||
#undef UPB_DESC
|
||||
#undef UPB_IS_GOOGLE3
|
||||
#undef UPB_ATOMIC
|
||||
#undef UPB_USE_C11_ATOMICS
|
||||
#undef UPB_PRIVATE
|
||||
30
Pods/gRPC-Core/third_party/upb/upb/port/vsnprintf_compat.h
generated
vendored
Normal file
30
Pods/gRPC-Core/third_party/upb/upb/port/vsnprintf_compat.h
generated
vendored
Normal file
@@ -0,0 +1,30 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_PORT_VSNPRINTF_COMPAT_H_
|
||||
#define UPB_PORT_VSNPRINTF_COMPAT_H_
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
UPB_INLINE int _upb_vsnprintf(char* buf, size_t size, const char* fmt,
|
||||
va_list ap) {
|
||||
#if defined(__MINGW64__) || defined(__MINGW32__) || defined(_MSC_VER)
|
||||
// The msvc runtime has a non-conforming vsnprintf() that requires the
|
||||
// following compatibility code to become conformant.
|
||||
int n = -1;
|
||||
if (size != 0) n = _vsnprintf_s(buf, size, _TRUNCATE, fmt, ap);
|
||||
if (n == -1) n = _vscprintf(fmt, ap);
|
||||
return n;
|
||||
#else
|
||||
return vsnprintf(buf, size, fmt, ap);
|
||||
#endif
|
||||
}
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_PORT_VSNPRINTF_COMPAT_H_
|
||||
51
Pods/gRPC-Core/third_party/upb/upb/reflection/common.h
generated
vendored
Normal file
51
Pods/gRPC-Core/third_party/upb/upb/reflection/common.h
generated
vendored
Normal file
@@ -0,0 +1,51 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
// Declarations common to all public def types.
|
||||
|
||||
#ifndef UPB_REFLECTION_COMMON_H_
|
||||
#define UPB_REFLECTION_COMMON_H_
|
||||
|
||||
// begin:google_only
|
||||
// #ifndef UPB_BOOTSTRAP_STAGE0
|
||||
// #include "net/proto2/proto/descriptor.upb.h"
|
||||
// #else
|
||||
// #include "google/protobuf/descriptor.upb.h"
|
||||
// #endif
|
||||
// end:google_only
|
||||
|
||||
// begin:github_only
|
||||
#include "google/protobuf/descriptor.upb.h"
|
||||
// end:github_only
|
||||
|
||||
typedef enum {
|
||||
kUpb_Syntax_Proto2 = 2,
|
||||
kUpb_Syntax_Proto3 = 3,
|
||||
kUpb_Syntax_Editions = 99
|
||||
} upb_Syntax;
|
||||
|
||||
// Forward declarations for circular references.
|
||||
typedef struct upb_DefPool upb_DefPool;
|
||||
typedef struct upb_EnumDef upb_EnumDef;
|
||||
typedef struct upb_EnumReservedRange upb_EnumReservedRange;
|
||||
typedef struct upb_EnumValueDef upb_EnumValueDef;
|
||||
typedef struct upb_ExtensionRange upb_ExtensionRange;
|
||||
typedef struct upb_FieldDef upb_FieldDef;
|
||||
typedef struct upb_FileDef upb_FileDef;
|
||||
typedef struct upb_MessageDef upb_MessageDef;
|
||||
typedef struct upb_MessageReservedRange upb_MessageReservedRange;
|
||||
typedef struct upb_MethodDef upb_MethodDef;
|
||||
typedef struct upb_OneofDef upb_OneofDef;
|
||||
typedef struct upb_ServiceDef upb_ServiceDef;
|
||||
|
||||
// EVERYTHING BELOW THIS LINE IS INTERNAL - DO NOT USE /////////////////////////
|
||||
|
||||
typedef struct upb_DefBuilder upb_DefBuilder;
|
||||
|
||||
#endif /* UPB_REFLECTION_COMMON_H_ */
|
||||
24
Pods/gRPC-Core/third_party/upb/upb/reflection/def.h
generated
vendored
Normal file
24
Pods/gRPC-Core/third_party/upb/upb/reflection/def.h
generated
vendored
Normal file
@@ -0,0 +1,24 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_DEF_H_
|
||||
#define UPB_REFLECTION_DEF_H_
|
||||
|
||||
// IWYU pragma: begin_exports
|
||||
#include "upb/reflection/def_pool.h"
|
||||
#include "upb/reflection/enum_def.h"
|
||||
#include "upb/reflection/enum_value_def.h"
|
||||
#include "upb/reflection/extension_range.h"
|
||||
#include "upb/reflection/field_def.h"
|
||||
#include "upb/reflection/file_def.h"
|
||||
#include "upb/reflection/message_def.h"
|
||||
#include "upb/reflection/method_def.h"
|
||||
#include "upb/reflection/oneof_def.h"
|
||||
#include "upb/reflection/service_def.h"
|
||||
// IWYU pragma: end_exports
|
||||
|
||||
#endif /* UPB_REFLECTION_DEF_H_ */
|
||||
605
Pods/gRPC-Core/third_party/upb/upb/reflection/def.hpp
generated
vendored
Normal file
605
Pods/gRPC-Core/third_party/upb/upb/reflection/def.hpp
generated
vendored
Normal file
@@ -0,0 +1,605 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_DEF_HPP_
|
||||
#define UPB_REFLECTION_DEF_HPP_
|
||||
|
||||
#include <cstring>
|
||||
#include <memory>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "upb/base/status.hpp"
|
||||
#include "upb/mem/arena.hpp"
|
||||
#include "upb/reflection/def.h"
|
||||
#include "upb/reflection/internal/def_pool.h"
|
||||
#include "upb/reflection/internal/enum_def.h"
|
||||
#include "upb/reflection/message.h"
|
||||
|
||||
// Must be last
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
namespace upb {
|
||||
|
||||
typedef upb_MessageValue MessageValue;
|
||||
|
||||
class EnumDefPtr;
|
||||
class FileDefPtr;
|
||||
class MessageDefPtr;
|
||||
class OneofDefPtr;
|
||||
|
||||
// A upb::FieldDefPtr describes a single field in a message. It is most often
|
||||
// found as a part of a upb_MessageDef, but can also stand alone to represent
|
||||
// an extension.
|
||||
class FieldDefPtr {
|
||||
public:
|
||||
FieldDefPtr() : ptr_(nullptr) {}
|
||||
explicit FieldDefPtr(const upb_FieldDef* ptr) : ptr_(ptr) {}
|
||||
|
||||
const upb_FieldDef* ptr() const { return ptr_; }
|
||||
|
||||
typedef upb_FieldType Type;
|
||||
typedef upb_CType CType;
|
||||
typedef upb_Label Label;
|
||||
|
||||
FileDefPtr file() const;
|
||||
const char* full_name() const { return upb_FieldDef_FullName(ptr_); }
|
||||
|
||||
const upb_MiniTableField* mini_table() const {
|
||||
return upb_FieldDef_MiniTable(ptr_);
|
||||
}
|
||||
|
||||
const UPB_DESC(FieldOptions) * options() const {
|
||||
return upb_FieldDef_Options(ptr_);
|
||||
}
|
||||
|
||||
Type type() const { return upb_FieldDef_Type(ptr_); }
|
||||
CType ctype() const { return upb_FieldDef_CType(ptr_); }
|
||||
Label label() const { return upb_FieldDef_Label(ptr_); }
|
||||
const char* name() const { return upb_FieldDef_Name(ptr_); }
|
||||
const char* json_name() const { return upb_FieldDef_JsonName(ptr_); }
|
||||
uint32_t number() const { return upb_FieldDef_Number(ptr_); }
|
||||
bool is_extension() const { return upb_FieldDef_IsExtension(ptr_); }
|
||||
bool is_required() const { return upb_FieldDef_IsRequired(ptr_); }
|
||||
bool has_presence() const { return upb_FieldDef_HasPresence(ptr_); }
|
||||
|
||||
// For non-string, non-submessage fields, this indicates whether binary
|
||||
// protobufs are encoded in packed or non-packed format.
|
||||
//
|
||||
// Note: this accessor reflects the fact that "packed" has different defaults
|
||||
// depending on whether the proto is proto2 or proto3.
|
||||
bool packed() const { return upb_FieldDef_IsPacked(ptr_); }
|
||||
|
||||
// An integer that can be used as an index into an array of fields for
|
||||
// whatever message this field belongs to. Guaranteed to be less than
|
||||
// f->containing_type()->field_count(). May only be accessed once the def has
|
||||
// been finalized.
|
||||
uint32_t index() const { return upb_FieldDef_Index(ptr_); }
|
||||
|
||||
// The MessageDef to which this field belongs (for extensions, the extended
|
||||
// message).
|
||||
MessageDefPtr containing_type() const;
|
||||
|
||||
// For extensions, the message the extension is declared inside, or NULL if
|
||||
// none.
|
||||
MessageDefPtr extension_scope() const;
|
||||
|
||||
// The OneofDef to which this field belongs, or NULL if this field is not part
|
||||
// of a oneof.
|
||||
OneofDefPtr containing_oneof() const;
|
||||
OneofDefPtr real_containing_oneof() const;
|
||||
|
||||
// Convenient field type tests.
|
||||
bool IsSubMessage() const { return upb_FieldDef_IsSubMessage(ptr_); }
|
||||
bool IsString() const { return upb_FieldDef_IsString(ptr_); }
|
||||
bool IsSequence() const { return upb_FieldDef_IsRepeated(ptr_); }
|
||||
bool IsPrimitive() const { return upb_FieldDef_IsPrimitive(ptr_); }
|
||||
bool IsMap() const { return upb_FieldDef_IsMap(ptr_); }
|
||||
|
||||
MessageValue default_value() const { return upb_FieldDef_Default(ptr_); }
|
||||
|
||||
// Returns the enum or submessage def for this field, if any. The field's
|
||||
// type must match (ie. you may only call enum_subdef() for fields where
|
||||
// type() == kUpb_CType_Enum).
|
||||
EnumDefPtr enum_subdef() const;
|
||||
MessageDefPtr message_type() const;
|
||||
|
||||
explicit operator bool() const { return ptr_ != nullptr; }
|
||||
|
||||
friend bool operator==(FieldDefPtr lhs, FieldDefPtr rhs) {
|
||||
return lhs.ptr_ == rhs.ptr_;
|
||||
}
|
||||
|
||||
friend bool operator!=(FieldDefPtr lhs, FieldDefPtr rhs) {
|
||||
return !(lhs == rhs);
|
||||
}
|
||||
|
||||
private:
|
||||
const upb_FieldDef* ptr_;
|
||||
};
|
||||
|
||||
// Class that represents a oneof.
|
||||
class OneofDefPtr {
|
||||
public:
|
||||
OneofDefPtr() : ptr_(nullptr) {}
|
||||
explicit OneofDefPtr(const upb_OneofDef* ptr) : ptr_(ptr) {}
|
||||
|
||||
const upb_OneofDef* ptr() const { return ptr_; }
|
||||
explicit operator bool() const { return ptr_ != nullptr; }
|
||||
|
||||
const UPB_DESC(OneofOptions) * options() const {
|
||||
return upb_OneofDef_Options(ptr_);
|
||||
}
|
||||
|
||||
// Returns the MessageDef that contains this OneofDef.
|
||||
MessageDefPtr containing_type() const;
|
||||
|
||||
// Returns the name of this oneof.
|
||||
const char* name() const { return upb_OneofDef_Name(ptr_); }
|
||||
const char* full_name() const { return upb_OneofDef_FullName(ptr_); }
|
||||
|
||||
// Returns the number of fields in the oneof.
|
||||
int field_count() const { return upb_OneofDef_FieldCount(ptr_); }
|
||||
FieldDefPtr field(int i) const {
|
||||
return FieldDefPtr(upb_OneofDef_Field(ptr_, i));
|
||||
}
|
||||
|
||||
// Looks up by name.
|
||||
FieldDefPtr FindFieldByName(const char* name, size_t len) const {
|
||||
return FieldDefPtr(upb_OneofDef_LookupNameWithSize(ptr_, name, len));
|
||||
}
|
||||
FieldDefPtr FindFieldByName(const char* name) const {
|
||||
return FieldDefPtr(upb_OneofDef_LookupName(ptr_, name));
|
||||
}
|
||||
|
||||
template <class T>
|
||||
FieldDefPtr FindFieldByName(const T& str) const {
|
||||
return FindFieldByName(str.c_str(), str.size());
|
||||
}
|
||||
|
||||
// Looks up by tag number.
|
||||
FieldDefPtr FindFieldByNumber(uint32_t num) const {
|
||||
return FieldDefPtr(upb_OneofDef_LookupNumber(ptr_, num));
|
||||
}
|
||||
|
||||
private:
|
||||
const upb_OneofDef* ptr_;
|
||||
};
|
||||
|
||||
// Structure that describes a single .proto message type.
|
||||
class MessageDefPtr {
|
||||
public:
|
||||
MessageDefPtr() : ptr_(nullptr) {}
|
||||
explicit MessageDefPtr(const upb_MessageDef* ptr) : ptr_(ptr) {}
|
||||
|
||||
const UPB_DESC(MessageOptions) * options() const {
|
||||
return upb_MessageDef_Options(ptr_);
|
||||
}
|
||||
|
||||
std::string MiniDescriptorEncode() const {
|
||||
upb::Arena arena;
|
||||
upb_StringView md;
|
||||
upb_MessageDef_MiniDescriptorEncode(ptr_, arena.ptr(), &md);
|
||||
return std::string(md.data, md.size);
|
||||
}
|
||||
|
||||
const upb_MessageDef* ptr() const { return ptr_; }
|
||||
|
||||
FileDefPtr file() const;
|
||||
|
||||
const char* full_name() const { return upb_MessageDef_FullName(ptr_); }
|
||||
const char* name() const { return upb_MessageDef_Name(ptr_); }
|
||||
|
||||
const upb_MiniTable* mini_table() const {
|
||||
return upb_MessageDef_MiniTable(ptr_);
|
||||
}
|
||||
|
||||
// The number of fields that belong to the MessageDef.
|
||||
int field_count() const { return upb_MessageDef_FieldCount(ptr_); }
|
||||
FieldDefPtr field(int i) const {
|
||||
return FieldDefPtr(upb_MessageDef_Field(ptr_, i));
|
||||
}
|
||||
|
||||
// The number of oneofs that belong to the MessageDef.
|
||||
int oneof_count() const { return upb_MessageDef_OneofCount(ptr_); }
|
||||
int real_oneof_count() const { return upb_MessageDef_RealOneofCount(ptr_); }
|
||||
OneofDefPtr oneof(int i) const {
|
||||
return OneofDefPtr(upb_MessageDef_Oneof(ptr_, i));
|
||||
}
|
||||
|
||||
int enum_type_count() const { return upb_MessageDef_NestedEnumCount(ptr_); }
|
||||
EnumDefPtr enum_type(int i) const;
|
||||
|
||||
int nested_message_count() const {
|
||||
return upb_MessageDef_NestedMessageCount(ptr_);
|
||||
}
|
||||
MessageDefPtr nested_message(int i) const {
|
||||
return MessageDefPtr(upb_MessageDef_NestedMessage(ptr_, i));
|
||||
}
|
||||
|
||||
int nested_extension_count() const {
|
||||
return upb_MessageDef_NestedExtensionCount(ptr_);
|
||||
}
|
||||
FieldDefPtr nested_extension(int i) const {
|
||||
return FieldDefPtr(upb_MessageDef_NestedExtension(ptr_, i));
|
||||
}
|
||||
|
||||
int extension_range_count() const {
|
||||
return upb_MessageDef_ExtensionRangeCount(ptr_);
|
||||
}
|
||||
|
||||
upb_Syntax syntax() const { return upb_MessageDef_Syntax(ptr_); }
|
||||
|
||||
// These return null pointers if the field is not found.
|
||||
FieldDefPtr FindFieldByNumber(uint32_t number) const {
|
||||
return FieldDefPtr(upb_MessageDef_FindFieldByNumber(ptr_, number));
|
||||
}
|
||||
FieldDefPtr FindFieldByName(const char* name, size_t len) const {
|
||||
return FieldDefPtr(upb_MessageDef_FindFieldByNameWithSize(ptr_, name, len));
|
||||
}
|
||||
FieldDefPtr FindFieldByName(const char* name) const {
|
||||
return FieldDefPtr(upb_MessageDef_FindFieldByName(ptr_, name));
|
||||
}
|
||||
|
||||
template <class T>
|
||||
FieldDefPtr FindFieldByName(const T& str) const {
|
||||
return FindFieldByName(str.c_str(), str.size());
|
||||
}
|
||||
|
||||
OneofDefPtr FindOneofByName(const char* name, size_t len) const {
|
||||
return OneofDefPtr(upb_MessageDef_FindOneofByNameWithSize(ptr_, name, len));
|
||||
}
|
||||
|
||||
OneofDefPtr FindOneofByName(const char* name) const {
|
||||
return OneofDefPtr(upb_MessageDef_FindOneofByName(ptr_, name));
|
||||
}
|
||||
|
||||
template <class T>
|
||||
OneofDefPtr FindOneofByName(const T& str) const {
|
||||
return FindOneofByName(str.c_str(), str.size());
|
||||
}
|
||||
|
||||
// Is this message a map entry?
|
||||
bool mapentry() const { return upb_MessageDef_IsMapEntry(ptr_); }
|
||||
|
||||
FieldDefPtr map_key() const {
|
||||
if (!mapentry()) return FieldDefPtr();
|
||||
return FieldDefPtr(upb_MessageDef_Field(ptr_, 0));
|
||||
}
|
||||
|
||||
FieldDefPtr map_value() const {
|
||||
if (!mapentry()) return FieldDefPtr();
|
||||
return FieldDefPtr(upb_MessageDef_Field(ptr_, 1));
|
||||
}
|
||||
|
||||
// Return the type of well known type message. kUpb_WellKnown_Unspecified for
|
||||
// non-well-known message.
|
||||
upb_WellKnown wellknowntype() const {
|
||||
return upb_MessageDef_WellKnownType(ptr_);
|
||||
}
|
||||
|
||||
explicit operator bool() const { return ptr_ != nullptr; }
|
||||
|
||||
friend bool operator==(MessageDefPtr lhs, MessageDefPtr rhs) {
|
||||
return lhs.ptr_ == rhs.ptr_;
|
||||
}
|
||||
|
||||
friend bool operator!=(MessageDefPtr lhs, MessageDefPtr rhs) {
|
||||
return !(lhs == rhs);
|
||||
}
|
||||
|
||||
private:
|
||||
class FieldIter {
|
||||
public:
|
||||
explicit FieldIter(const upb_MessageDef* m, int i) : m_(m), i_(i) {}
|
||||
void operator++() { i_++; }
|
||||
|
||||
FieldDefPtr operator*() {
|
||||
return FieldDefPtr(upb_MessageDef_Field(m_, i_));
|
||||
}
|
||||
|
||||
friend bool operator==(FieldIter lhs, FieldIter rhs) {
|
||||
return lhs.i_ == rhs.i_;
|
||||
}
|
||||
|
||||
friend bool operator!=(FieldIter lhs, FieldIter rhs) {
|
||||
return !(lhs == rhs);
|
||||
}
|
||||
|
||||
private:
|
||||
const upb_MessageDef* m_;
|
||||
int i_;
|
||||
};
|
||||
|
||||
class FieldAccessor {
|
||||
public:
|
||||
explicit FieldAccessor(const upb_MessageDef* md) : md_(md) {}
|
||||
FieldIter begin() { return FieldIter(md_, 0); }
|
||||
FieldIter end() { return FieldIter(md_, upb_MessageDef_FieldCount(md_)); }
|
||||
|
||||
private:
|
||||
const upb_MessageDef* md_;
|
||||
};
|
||||
|
||||
class OneofIter {
|
||||
public:
|
||||
explicit OneofIter(const upb_MessageDef* m, int i) : m_(m), i_(i) {}
|
||||
void operator++() { i_++; }
|
||||
|
||||
OneofDefPtr operator*() {
|
||||
return OneofDefPtr(upb_MessageDef_Oneof(m_, i_));
|
||||
}
|
||||
|
||||
friend bool operator==(OneofIter lhs, OneofIter rhs) {
|
||||
return lhs.i_ == rhs.i_;
|
||||
}
|
||||
|
||||
friend bool operator!=(OneofIter lhs, OneofIter rhs) {
|
||||
return !(lhs == rhs);
|
||||
}
|
||||
|
||||
private:
|
||||
const upb_MessageDef* m_;
|
||||
int i_;
|
||||
};
|
||||
|
||||
class OneofAccessor {
|
||||
public:
|
||||
explicit OneofAccessor(const upb_MessageDef* md) : md_(md) {}
|
||||
OneofIter begin() { return OneofIter(md_, 0); }
|
||||
OneofIter end() { return OneofIter(md_, upb_MessageDef_OneofCount(md_)); }
|
||||
|
||||
private:
|
||||
const upb_MessageDef* md_;
|
||||
};
|
||||
|
||||
public:
|
||||
FieldAccessor fields() const { return FieldAccessor(ptr()); }
|
||||
OneofAccessor oneofs() const { return OneofAccessor(ptr()); }
|
||||
|
||||
private:
|
||||
const upb_MessageDef* ptr_;
|
||||
};
|
||||
|
||||
class EnumValDefPtr {
|
||||
public:
|
||||
EnumValDefPtr() : ptr_(nullptr) {}
|
||||
explicit EnumValDefPtr(const upb_EnumValueDef* ptr) : ptr_(ptr) {}
|
||||
|
||||
const UPB_DESC(EnumValueOptions) * options() const {
|
||||
return upb_EnumValueDef_Options(ptr_);
|
||||
}
|
||||
|
||||
int32_t number() const { return upb_EnumValueDef_Number(ptr_); }
|
||||
const char* full_name() const { return upb_EnumValueDef_FullName(ptr_); }
|
||||
const char* name() const { return upb_EnumValueDef_Name(ptr_); }
|
||||
|
||||
private:
|
||||
const upb_EnumValueDef* ptr_;
|
||||
};
|
||||
|
||||
class EnumDefPtr {
|
||||
public:
|
||||
EnumDefPtr() : ptr_(nullptr) {}
|
||||
explicit EnumDefPtr(const upb_EnumDef* ptr) : ptr_(ptr) {}
|
||||
|
||||
const UPB_DESC(EnumOptions) * options() const {
|
||||
return upb_EnumDef_Options(ptr_);
|
||||
}
|
||||
|
||||
const upb_MiniTableEnum* mini_table() const {
|
||||
return _upb_EnumDef_MiniTable(ptr_);
|
||||
}
|
||||
|
||||
std::string MiniDescriptorEncode() const {
|
||||
upb::Arena arena;
|
||||
upb_StringView md;
|
||||
upb_EnumDef_MiniDescriptorEncode(ptr_, arena.ptr(), &md);
|
||||
return std::string(md.data, md.size);
|
||||
}
|
||||
|
||||
const upb_EnumDef* ptr() const { return ptr_; }
|
||||
explicit operator bool() const { return ptr_ != nullptr; }
|
||||
|
||||
const char* full_name() const { return upb_EnumDef_FullName(ptr_); }
|
||||
const char* name() const { return upb_EnumDef_Name(ptr_); }
|
||||
bool is_closed() const { return upb_EnumDef_IsClosed(ptr_); }
|
||||
|
||||
// The value that is used as the default when no field default is specified.
|
||||
// If not set explicitly, the first value that was added will be used.
|
||||
// The default value must be a member of the enum.
|
||||
// Requires that value_count() > 0.
|
||||
int32_t default_value() const { return upb_EnumDef_Default(ptr_); }
|
||||
|
||||
// Returns the number of values currently defined in the enum. Note that
|
||||
// multiple names can refer to the same number, so this may be greater than
|
||||
// the total number of unique numbers.
|
||||
int value_count() const { return upb_EnumDef_ValueCount(ptr_); }
|
||||
EnumValDefPtr value(int i) const {
|
||||
return EnumValDefPtr(upb_EnumDef_Value(ptr_, i));
|
||||
}
|
||||
|
||||
// Lookups from name to integer, returning true if found.
|
||||
EnumValDefPtr FindValueByName(const char* name) const {
|
||||
return EnumValDefPtr(upb_EnumDef_FindValueByName(ptr_, name));
|
||||
}
|
||||
|
||||
// Finds the name corresponding to the given number, or NULL if none was
|
||||
// found. If more than one name corresponds to this number, returns the
|
||||
// first one that was added.
|
||||
EnumValDefPtr FindValueByNumber(int32_t num) const {
|
||||
return EnumValDefPtr(upb_EnumDef_FindValueByNumber(ptr_, num));
|
||||
}
|
||||
|
||||
private:
|
||||
const upb_EnumDef* ptr_;
|
||||
};
|
||||
|
||||
// Class that represents a .proto file with some things defined in it.
|
||||
//
|
||||
// Many users won't care about FileDefs, but they are necessary if you want to
|
||||
// read the values of file-level options.
|
||||
class FileDefPtr {
|
||||
public:
|
||||
explicit FileDefPtr(const upb_FileDef* ptr) : ptr_(ptr) {}
|
||||
|
||||
const UPB_DESC(FileOptions) * options() const {
|
||||
return upb_FileDef_Options(ptr_);
|
||||
}
|
||||
|
||||
const upb_FileDef* ptr() const { return ptr_; }
|
||||
|
||||
// Get/set name of the file (eg. "foo/bar.proto").
|
||||
const char* name() const { return upb_FileDef_Name(ptr_); }
|
||||
|
||||
// Package name for definitions inside the file (eg. "foo.bar").
|
||||
const char* package() const { return upb_FileDef_Package(ptr_); }
|
||||
|
||||
// Syntax for the file. Defaults to proto2.
|
||||
upb_Syntax syntax() const { return upb_FileDef_Syntax(ptr_); }
|
||||
|
||||
// Get the list of dependencies from the file. These are returned in the
|
||||
// order that they were added to the FileDefPtr.
|
||||
int dependency_count() const { return upb_FileDef_DependencyCount(ptr_); }
|
||||
FileDefPtr dependency(int index) const {
|
||||
return FileDefPtr(upb_FileDef_Dependency(ptr_, index));
|
||||
}
|
||||
|
||||
int public_dependency_count() const {
|
||||
return upb_FileDef_PublicDependencyCount(ptr_);
|
||||
}
|
||||
FileDefPtr public_dependency(int index) const {
|
||||
return FileDefPtr(upb_FileDef_PublicDependency(ptr_, index));
|
||||
}
|
||||
|
||||
int toplevel_enum_count() const {
|
||||
return upb_FileDef_TopLevelEnumCount(ptr_);
|
||||
}
|
||||
EnumDefPtr toplevel_enum(int index) const {
|
||||
return EnumDefPtr(upb_FileDef_TopLevelEnum(ptr_, index));
|
||||
}
|
||||
|
||||
int toplevel_message_count() const {
|
||||
return upb_FileDef_TopLevelMessageCount(ptr_);
|
||||
}
|
||||
MessageDefPtr toplevel_message(int index) const {
|
||||
return MessageDefPtr(upb_FileDef_TopLevelMessage(ptr_, index));
|
||||
}
|
||||
|
||||
int toplevel_extension_count() const {
|
||||
return upb_FileDef_TopLevelExtensionCount(ptr_);
|
||||
}
|
||||
FieldDefPtr toplevel_extension(int index) const {
|
||||
return FieldDefPtr(upb_FileDef_TopLevelExtension(ptr_, index));
|
||||
}
|
||||
|
||||
explicit operator bool() const { return ptr_ != nullptr; }
|
||||
|
||||
friend bool operator==(FileDefPtr lhs, FileDefPtr rhs) {
|
||||
return lhs.ptr_ == rhs.ptr_;
|
||||
}
|
||||
|
||||
friend bool operator!=(FileDefPtr lhs, FileDefPtr rhs) {
|
||||
return !(lhs == rhs);
|
||||
}
|
||||
|
||||
private:
|
||||
const upb_FileDef* ptr_;
|
||||
};
|
||||
|
||||
// Non-const methods in upb::DefPool are NOT thread-safe.
|
||||
class DefPool {
|
||||
public:
|
||||
DefPool() : ptr_(upb_DefPool_New(), upb_DefPool_Free) {}
|
||||
explicit DefPool(upb_DefPool* s) : ptr_(s, upb_DefPool_Free) {}
|
||||
|
||||
const upb_DefPool* ptr() const { return ptr_.get(); }
|
||||
upb_DefPool* ptr() { return ptr_.get(); }
|
||||
|
||||
// Finds an entry in the symbol table with this exact name. If not found,
|
||||
// returns NULL.
|
||||
MessageDefPtr FindMessageByName(const char* sym) const {
|
||||
return MessageDefPtr(upb_DefPool_FindMessageByName(ptr_.get(), sym));
|
||||
}
|
||||
|
||||
EnumDefPtr FindEnumByName(const char* sym) const {
|
||||
return EnumDefPtr(upb_DefPool_FindEnumByName(ptr_.get(), sym));
|
||||
}
|
||||
|
||||
FileDefPtr FindFileByName(const char* name) const {
|
||||
return FileDefPtr(upb_DefPool_FindFileByName(ptr_.get(), name));
|
||||
}
|
||||
|
||||
FieldDefPtr FindExtensionByName(const char* name) const {
|
||||
return FieldDefPtr(upb_DefPool_FindExtensionByName(ptr_.get(), name));
|
||||
}
|
||||
|
||||
void _SetPlatform(upb_MiniTablePlatform platform) {
|
||||
_upb_DefPool_SetPlatform(ptr_.get(), platform);
|
||||
}
|
||||
|
||||
// TODO: iteration?
|
||||
|
||||
// Adds the given serialized FileDescriptorProto to the pool.
|
||||
FileDefPtr AddFile(const UPB_DESC(FileDescriptorProto) * file_proto,
|
||||
Status* status) {
|
||||
return FileDefPtr(
|
||||
upb_DefPool_AddFile(ptr_.get(), file_proto, status->ptr()));
|
||||
}
|
||||
|
||||
private:
|
||||
std::unique_ptr<upb_DefPool, decltype(&upb_DefPool_Free)> ptr_;
|
||||
};
|
||||
|
||||
// TODO: This typedef is deprecated. Delete it.
|
||||
using SymbolTable = DefPool;
|
||||
|
||||
inline FileDefPtr FieldDefPtr::file() const {
|
||||
return FileDefPtr(upb_FieldDef_File(ptr_));
|
||||
}
|
||||
|
||||
inline FileDefPtr MessageDefPtr::file() const {
|
||||
return FileDefPtr(upb_MessageDef_File(ptr_));
|
||||
}
|
||||
|
||||
inline EnumDefPtr MessageDefPtr::enum_type(int i) const {
|
||||
return EnumDefPtr(upb_MessageDef_NestedEnum(ptr_, i));
|
||||
}
|
||||
|
||||
inline MessageDefPtr FieldDefPtr::message_type() const {
|
||||
return MessageDefPtr(upb_FieldDef_MessageSubDef(ptr_));
|
||||
}
|
||||
|
||||
inline MessageDefPtr FieldDefPtr::containing_type() const {
|
||||
return MessageDefPtr(upb_FieldDef_ContainingType(ptr_));
|
||||
}
|
||||
|
||||
inline MessageDefPtr FieldDefPtr::extension_scope() const {
|
||||
return MessageDefPtr(upb_FieldDef_ExtensionScope(ptr_));
|
||||
}
|
||||
|
||||
inline MessageDefPtr OneofDefPtr::containing_type() const {
|
||||
return MessageDefPtr(upb_OneofDef_ContainingType(ptr_));
|
||||
}
|
||||
|
||||
inline OneofDefPtr FieldDefPtr::containing_oneof() const {
|
||||
return OneofDefPtr(upb_FieldDef_ContainingOneof(ptr_));
|
||||
}
|
||||
|
||||
inline OneofDefPtr FieldDefPtr::real_containing_oneof() const {
|
||||
return OneofDefPtr(upb_FieldDef_RealContainingOneof(ptr_));
|
||||
}
|
||||
|
||||
inline EnumDefPtr FieldDefPtr::enum_subdef() const {
|
||||
return EnumDefPtr(upb_FieldDef_EnumSubDef(ptr_));
|
||||
}
|
||||
|
||||
} // namespace upb
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_REFLECTION_DEF_HPP_
|
||||
443
Pods/gRPC-Core/third_party/upb/upb/reflection/def_pool.c
generated
vendored
Normal file
443
Pods/gRPC-Core/third_party/upb/upb/reflection/def_pool.c
generated
vendored
Normal file
@@ -0,0 +1,443 @@
|
||||
// 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/reflection/internal/def_pool.h"
|
||||
|
||||
#include "upb/hash/int_table.h"
|
||||
#include "upb/hash/str_table.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/enum_def.h"
|
||||
#include "upb/reflection/internal/enum_value_def.h"
|
||||
#include "upb/reflection/internal/field_def.h"
|
||||
#include "upb/reflection/internal/file_def.h"
|
||||
#include "upb/reflection/internal/message_def.h"
|
||||
#include "upb/reflection/internal/service_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_DefPool {
|
||||
upb_Arena* arena;
|
||||
upb_strtable syms; // full_name -> packed def ptr
|
||||
upb_strtable files; // file_name -> (upb_FileDef*)
|
||||
upb_inttable exts; // (upb_MiniTableExtension*) -> (upb_FieldDef*)
|
||||
upb_ExtensionRegistry* extreg;
|
||||
upb_MiniTablePlatform platform;
|
||||
void* scratch_data;
|
||||
size_t scratch_size;
|
||||
size_t bytes_loaded;
|
||||
};
|
||||
|
||||
void upb_DefPool_Free(upb_DefPool* s) {
|
||||
upb_Arena_Free(s->arena);
|
||||
upb_gfree(s->scratch_data);
|
||||
upb_gfree(s);
|
||||
}
|
||||
|
||||
upb_DefPool* upb_DefPool_New(void) {
|
||||
upb_DefPool* s = upb_gmalloc(sizeof(*s));
|
||||
if (!s) return NULL;
|
||||
|
||||
s->arena = upb_Arena_New();
|
||||
s->bytes_loaded = 0;
|
||||
|
||||
s->scratch_size = 240;
|
||||
s->scratch_data = upb_gmalloc(s->scratch_size);
|
||||
if (!s->scratch_data) goto err;
|
||||
|
||||
if (!upb_strtable_init(&s->syms, 32, s->arena)) goto err;
|
||||
if (!upb_strtable_init(&s->files, 4, s->arena)) goto err;
|
||||
if (!upb_inttable_init(&s->exts, s->arena)) goto err;
|
||||
|
||||
s->extreg = upb_ExtensionRegistry_New(s->arena);
|
||||
if (!s->extreg) goto err;
|
||||
|
||||
s->platform = kUpb_MiniTablePlatform_Native;
|
||||
|
||||
return s;
|
||||
|
||||
err:
|
||||
upb_DefPool_Free(s);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
bool _upb_DefPool_InsertExt(upb_DefPool* s, const upb_MiniTableExtension* ext,
|
||||
const upb_FieldDef* f) {
|
||||
return upb_inttable_insert(&s->exts, (uintptr_t)ext, upb_value_constptr(f),
|
||||
s->arena);
|
||||
}
|
||||
|
||||
bool _upb_DefPool_InsertSym(upb_DefPool* s, upb_StringView sym, upb_value v,
|
||||
upb_Status* status) {
|
||||
// TODO: table should support an operation "tryinsert" to avoid the double
|
||||
// lookup.
|
||||
if (upb_strtable_lookup2(&s->syms, sym.data, sym.size, NULL)) {
|
||||
upb_Status_SetErrorFormat(status, "duplicate symbol '%s'", sym.data);
|
||||
return false;
|
||||
}
|
||||
if (!upb_strtable_insert(&s->syms, sym.data, sym.size, v, s->arena)) {
|
||||
upb_Status_SetErrorMessage(status, "out of memory");
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static const void* _upb_DefPool_Unpack(const upb_DefPool* s, const char* sym,
|
||||
size_t size, upb_deftype_t type) {
|
||||
upb_value v;
|
||||
return upb_strtable_lookup2(&s->syms, sym, size, &v)
|
||||
? _upb_DefType_Unpack(v, type)
|
||||
: NULL;
|
||||
}
|
||||
|
||||
bool _upb_DefPool_LookupSym(const upb_DefPool* s, const char* sym, size_t size,
|
||||
upb_value* v) {
|
||||
return upb_strtable_lookup2(&s->syms, sym, size, v);
|
||||
}
|
||||
|
||||
upb_ExtensionRegistry* _upb_DefPool_ExtReg(const upb_DefPool* s) {
|
||||
return s->extreg;
|
||||
}
|
||||
|
||||
void** _upb_DefPool_ScratchData(const upb_DefPool* s) {
|
||||
return (void**)&s->scratch_data;
|
||||
}
|
||||
|
||||
size_t* _upb_DefPool_ScratchSize(const upb_DefPool* s) {
|
||||
return (size_t*)&s->scratch_size;
|
||||
}
|
||||
|
||||
void _upb_DefPool_SetPlatform(upb_DefPool* s, upb_MiniTablePlatform platform) {
|
||||
assert(upb_strtable_count(&s->files) == 0);
|
||||
s->platform = platform;
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_DefPool_FindMessageByName(const upb_DefPool* s,
|
||||
const char* sym) {
|
||||
return _upb_DefPool_Unpack(s, sym, strlen(sym), UPB_DEFTYPE_MSG);
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_DefPool_FindMessageByNameWithSize(
|
||||
const upb_DefPool* s, const char* sym, size_t len) {
|
||||
return _upb_DefPool_Unpack(s, sym, len, UPB_DEFTYPE_MSG);
|
||||
}
|
||||
|
||||
const upb_EnumDef* upb_DefPool_FindEnumByName(const upb_DefPool* s,
|
||||
const char* sym) {
|
||||
return _upb_DefPool_Unpack(s, sym, strlen(sym), UPB_DEFTYPE_ENUM);
|
||||
}
|
||||
|
||||
const upb_EnumValueDef* upb_DefPool_FindEnumByNameval(const upb_DefPool* s,
|
||||
const char* sym) {
|
||||
return _upb_DefPool_Unpack(s, sym, strlen(sym), UPB_DEFTYPE_ENUMVAL);
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_DefPool_FindFileByName(const upb_DefPool* s,
|
||||
const char* name) {
|
||||
upb_value v;
|
||||
return upb_strtable_lookup(&s->files, name, &v) ? upb_value_getconstptr(v)
|
||||
: NULL;
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_DefPool_FindFileByNameWithSize(const upb_DefPool* s,
|
||||
const char* name,
|
||||
size_t len) {
|
||||
upb_value v;
|
||||
return upb_strtable_lookup2(&s->files, name, len, &v)
|
||||
? upb_value_getconstptr(v)
|
||||
: NULL;
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_DefPool_FindExtensionByNameWithSize(
|
||||
const upb_DefPool* s, const char* name, size_t size) {
|
||||
upb_value v;
|
||||
if (!upb_strtable_lookup2(&s->syms, name, size, &v)) return NULL;
|
||||
|
||||
switch (_upb_DefType_Type(v)) {
|
||||
case UPB_DEFTYPE_FIELD:
|
||||
return _upb_DefType_Unpack(v, UPB_DEFTYPE_FIELD);
|
||||
case UPB_DEFTYPE_MSG: {
|
||||
const upb_MessageDef* m = _upb_DefType_Unpack(v, UPB_DEFTYPE_MSG);
|
||||
return _upb_MessageDef_InMessageSet(m)
|
||||
? upb_MessageDef_NestedExtension(m, 0)
|
||||
: NULL;
|
||||
}
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_DefPool_FindExtensionByName(const upb_DefPool* s,
|
||||
const char* sym) {
|
||||
return upb_DefPool_FindExtensionByNameWithSize(s, sym, strlen(sym));
|
||||
}
|
||||
|
||||
const upb_ServiceDef* upb_DefPool_FindServiceByName(const upb_DefPool* s,
|
||||
const char* name) {
|
||||
return _upb_DefPool_Unpack(s, name, strlen(name), UPB_DEFTYPE_SERVICE);
|
||||
}
|
||||
|
||||
const upb_ServiceDef* upb_DefPool_FindServiceByNameWithSize(
|
||||
const upb_DefPool* s, const char* name, size_t size) {
|
||||
return _upb_DefPool_Unpack(s, name, size, UPB_DEFTYPE_SERVICE);
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_DefPool_FindFileContainingSymbol(const upb_DefPool* s,
|
||||
const char* name) {
|
||||
upb_value v;
|
||||
// TODO: non-extension fields and oneofs.
|
||||
if (upb_strtable_lookup(&s->syms, name, &v)) {
|
||||
switch (_upb_DefType_Type(v)) {
|
||||
case UPB_DEFTYPE_EXT: {
|
||||
const upb_FieldDef* f = _upb_DefType_Unpack(v, UPB_DEFTYPE_EXT);
|
||||
return upb_FieldDef_File(f);
|
||||
}
|
||||
case UPB_DEFTYPE_MSG: {
|
||||
const upb_MessageDef* m = _upb_DefType_Unpack(v, UPB_DEFTYPE_MSG);
|
||||
return upb_MessageDef_File(m);
|
||||
}
|
||||
case UPB_DEFTYPE_ENUM: {
|
||||
const upb_EnumDef* e = _upb_DefType_Unpack(v, UPB_DEFTYPE_ENUM);
|
||||
return upb_EnumDef_File(e);
|
||||
}
|
||||
case UPB_DEFTYPE_ENUMVAL: {
|
||||
const upb_EnumValueDef* ev =
|
||||
_upb_DefType_Unpack(v, UPB_DEFTYPE_ENUMVAL);
|
||||
return upb_EnumDef_File(upb_EnumValueDef_Enum(ev));
|
||||
}
|
||||
case UPB_DEFTYPE_SERVICE: {
|
||||
const upb_ServiceDef* service =
|
||||
_upb_DefType_Unpack(v, UPB_DEFTYPE_SERVICE);
|
||||
return upb_ServiceDef_File(service);
|
||||
}
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
}
|
||||
|
||||
const char* last_dot = strrchr(name, '.');
|
||||
if (last_dot) {
|
||||
const upb_MessageDef* parent =
|
||||
upb_DefPool_FindMessageByNameWithSize(s, name, last_dot - name);
|
||||
if (parent) {
|
||||
const char* shortname = last_dot + 1;
|
||||
if (upb_MessageDef_FindByNameWithSize(parent, shortname,
|
||||
strlen(shortname), NULL, NULL)) {
|
||||
return upb_MessageDef_File(parent);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static void remove_filedef(upb_DefPool* s, upb_FileDef* file) {
|
||||
intptr_t iter = UPB_INTTABLE_BEGIN;
|
||||
upb_StringView key;
|
||||
upb_value val;
|
||||
while (upb_strtable_next2(&s->syms, &key, &val, &iter)) {
|
||||
const upb_FileDef* f;
|
||||
switch (_upb_DefType_Type(val)) {
|
||||
case UPB_DEFTYPE_EXT:
|
||||
f = upb_FieldDef_File(_upb_DefType_Unpack(val, UPB_DEFTYPE_EXT));
|
||||
break;
|
||||
case UPB_DEFTYPE_MSG:
|
||||
f = upb_MessageDef_File(_upb_DefType_Unpack(val, UPB_DEFTYPE_MSG));
|
||||
break;
|
||||
case UPB_DEFTYPE_ENUM:
|
||||
f = upb_EnumDef_File(_upb_DefType_Unpack(val, UPB_DEFTYPE_ENUM));
|
||||
break;
|
||||
case UPB_DEFTYPE_ENUMVAL:
|
||||
f = upb_EnumDef_File(upb_EnumValueDef_Enum(
|
||||
_upb_DefType_Unpack(val, UPB_DEFTYPE_ENUMVAL)));
|
||||
break;
|
||||
case UPB_DEFTYPE_SERVICE:
|
||||
f = upb_ServiceDef_File(_upb_DefType_Unpack(val, UPB_DEFTYPE_SERVICE));
|
||||
break;
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
if (f == file) upb_strtable_removeiter(&s->syms, &iter);
|
||||
}
|
||||
}
|
||||
|
||||
static const upb_FileDef* upb_DefBuilder_AddFileToPool(
|
||||
upb_DefBuilder* const builder, upb_DefPool* const s,
|
||||
const UPB_DESC(FileDescriptorProto) * const file_proto,
|
||||
const upb_StringView name, upb_Status* const status) {
|
||||
if (UPB_SETJMP(builder->err) != 0) {
|
||||
UPB_ASSERT(!upb_Status_IsOk(status));
|
||||
if (builder->file) {
|
||||
remove_filedef(s, builder->file);
|
||||
builder->file = NULL;
|
||||
}
|
||||
} else if (!builder->arena || !builder->tmp_arena) {
|
||||
_upb_DefBuilder_OomErr(builder);
|
||||
} else {
|
||||
_upb_FileDef_Create(builder, file_proto);
|
||||
upb_strtable_insert(&s->files, name.data, name.size,
|
||||
upb_value_constptr(builder->file), builder->arena);
|
||||
UPB_ASSERT(upb_Status_IsOk(status));
|
||||
upb_Arena_Fuse(s->arena, builder->arena);
|
||||
}
|
||||
|
||||
if (builder->arena) upb_Arena_Free(builder->arena);
|
||||
if (builder->tmp_arena) upb_Arena_Free(builder->tmp_arena);
|
||||
return builder->file;
|
||||
}
|
||||
|
||||
static const upb_FileDef* _upb_DefPool_AddFile(
|
||||
upb_DefPool* s, const UPB_DESC(FileDescriptorProto) * file_proto,
|
||||
const upb_MiniTableFile* layout, upb_Status* status) {
|
||||
const upb_StringView name = UPB_DESC(FileDescriptorProto_name)(file_proto);
|
||||
|
||||
// Determine whether we already know about this file.
|
||||
{
|
||||
upb_value v;
|
||||
if (upb_strtable_lookup2(&s->files, name.data, name.size, &v)) {
|
||||
upb_Status_SetErrorFormat(status,
|
||||
"duplicate file name " UPB_STRINGVIEW_FORMAT,
|
||||
UPB_STRINGVIEW_ARGS(name));
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
upb_DefBuilder ctx = {
|
||||
.symtab = s,
|
||||
.layout = layout,
|
||||
.platform = s->platform,
|
||||
.msg_count = 0,
|
||||
.enum_count = 0,
|
||||
.ext_count = 0,
|
||||
.status = status,
|
||||
.file = NULL,
|
||||
.arena = upb_Arena_New(),
|
||||
.tmp_arena = upb_Arena_New(),
|
||||
};
|
||||
|
||||
return upb_DefBuilder_AddFileToPool(&ctx, s, file_proto, name, status);
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_DefPool_AddFile(upb_DefPool* s,
|
||||
const UPB_DESC(FileDescriptorProto) *
|
||||
file_proto,
|
||||
upb_Status* status) {
|
||||
return _upb_DefPool_AddFile(s, file_proto, NULL, status);
|
||||
}
|
||||
|
||||
bool _upb_DefPool_LoadDefInitEx(upb_DefPool* s, const _upb_DefPool_Init* init,
|
||||
bool rebuild_minitable) {
|
||||
/* Since this function should never fail (it would indicate a bug in upb) we
|
||||
* print errors to stderr instead of returning error status to the user. */
|
||||
_upb_DefPool_Init** deps = init->deps;
|
||||
UPB_DESC(FileDescriptorProto) * file;
|
||||
upb_Arena* arena;
|
||||
upb_Status status;
|
||||
|
||||
upb_Status_Clear(&status);
|
||||
|
||||
if (upb_DefPool_FindFileByName(s, init->filename)) {
|
||||
return true;
|
||||
}
|
||||
|
||||
arena = upb_Arena_New();
|
||||
|
||||
for (; *deps; deps++) {
|
||||
if (!_upb_DefPool_LoadDefInitEx(s, *deps, rebuild_minitable)) goto err;
|
||||
}
|
||||
|
||||
file = UPB_DESC(FileDescriptorProto_parse_ex)(
|
||||
init->descriptor.data, init->descriptor.size, NULL,
|
||||
kUpb_DecodeOption_AliasString, arena);
|
||||
s->bytes_loaded += init->descriptor.size;
|
||||
|
||||
if (!file) {
|
||||
upb_Status_SetErrorFormat(
|
||||
&status,
|
||||
"Failed to parse compiled-in descriptor for file '%s'. This should "
|
||||
"never happen.",
|
||||
init->filename);
|
||||
goto err;
|
||||
}
|
||||
|
||||
const upb_MiniTableFile* mt = rebuild_minitable ? NULL : init->layout;
|
||||
if (!_upb_DefPool_AddFile(s, file, mt, &status)) {
|
||||
goto err;
|
||||
}
|
||||
|
||||
upb_Arena_Free(arena);
|
||||
return true;
|
||||
|
||||
err:
|
||||
fprintf(stderr,
|
||||
"Error loading compiled-in descriptor for file '%s' (this should "
|
||||
"never happen): %s\n",
|
||||
init->filename, upb_Status_ErrorMessage(&status));
|
||||
upb_Arena_Free(arena);
|
||||
return false;
|
||||
}
|
||||
|
||||
size_t _upb_DefPool_BytesLoaded(const upb_DefPool* s) {
|
||||
return s->bytes_loaded;
|
||||
}
|
||||
|
||||
upb_Arena* _upb_DefPool_Arena(const upb_DefPool* s) { return s->arena; }
|
||||
|
||||
const upb_FieldDef* upb_DefPool_FindExtensionByMiniTable(
|
||||
const upb_DefPool* s, const upb_MiniTableExtension* ext) {
|
||||
upb_value v;
|
||||
bool ok = upb_inttable_lookup(&s->exts, (uintptr_t)ext, &v);
|
||||
UPB_ASSERT(ok);
|
||||
return upb_value_getconstptr(v);
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_DefPool_FindExtensionByNumber(const upb_DefPool* s,
|
||||
const upb_MessageDef* m,
|
||||
int32_t fieldnum) {
|
||||
const upb_MiniTable* t = upb_MessageDef_MiniTable(m);
|
||||
const upb_MiniTableExtension* ext =
|
||||
upb_ExtensionRegistry_Lookup(s->extreg, t, fieldnum);
|
||||
return ext ? upb_DefPool_FindExtensionByMiniTable(s, ext) : NULL;
|
||||
}
|
||||
|
||||
const upb_ExtensionRegistry* upb_DefPool_ExtensionRegistry(
|
||||
const upb_DefPool* s) {
|
||||
return s->extreg;
|
||||
}
|
||||
|
||||
const upb_FieldDef** upb_DefPool_GetAllExtensions(const upb_DefPool* s,
|
||||
const upb_MessageDef* m,
|
||||
size_t* count) {
|
||||
size_t n = 0;
|
||||
intptr_t iter = UPB_INTTABLE_BEGIN;
|
||||
uintptr_t key;
|
||||
upb_value val;
|
||||
// This is O(all exts) instead of O(exts for m). If we need this to be
|
||||
// efficient we may need to make extreg into a two-level table, or have a
|
||||
// second per-message index.
|
||||
while (upb_inttable_next(&s->exts, &key, &val, &iter)) {
|
||||
const upb_FieldDef* f = upb_value_getconstptr(val);
|
||||
if (upb_FieldDef_ContainingType(f) == m) n++;
|
||||
}
|
||||
const upb_FieldDef** exts = malloc(n * sizeof(*exts));
|
||||
iter = UPB_INTTABLE_BEGIN;
|
||||
size_t i = 0;
|
||||
while (upb_inttable_next(&s->exts, &key, &val, &iter)) {
|
||||
const upb_FieldDef* f = upb_value_getconstptr(val);
|
||||
if (upb_FieldDef_ContainingType(f) == m) exts[i++] = f;
|
||||
}
|
||||
*count = n;
|
||||
return exts;
|
||||
}
|
||||
|
||||
bool _upb_DefPool_LoadDefInit(upb_DefPool* s, const _upb_DefPool_Init* init) {
|
||||
return _upb_DefPool_LoadDefInitEx(s, init, false);
|
||||
}
|
||||
87
Pods/gRPC-Core/third_party/upb/upb/reflection/def_pool.h
generated
vendored
Normal file
87
Pods/gRPC-Core/third_party/upb/upb/reflection/def_pool.h
generated
vendored
Normal file
@@ -0,0 +1,87 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_DEF_POOL_H_
|
||||
#define UPB_REFLECTION_DEF_POOL_H_
|
||||
|
||||
#include "upb/base/status.h"
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/reflection/common.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_API void upb_DefPool_Free(upb_DefPool* s);
|
||||
|
||||
UPB_API upb_DefPool* upb_DefPool_New(void);
|
||||
|
||||
UPB_API const upb_MessageDef* upb_DefPool_FindMessageByName(
|
||||
const upb_DefPool* s, const char* sym);
|
||||
|
||||
const upb_MessageDef* upb_DefPool_FindMessageByNameWithSize(
|
||||
const upb_DefPool* s, const char* sym, size_t len);
|
||||
|
||||
UPB_API const upb_EnumDef* upb_DefPool_FindEnumByName(const upb_DefPool* s,
|
||||
const char* sym);
|
||||
|
||||
const upb_EnumValueDef* upb_DefPool_FindEnumByNameval(const upb_DefPool* s,
|
||||
const char* sym);
|
||||
|
||||
const upb_FileDef* upb_DefPool_FindFileByName(const upb_DefPool* s,
|
||||
const char* name);
|
||||
|
||||
const upb_FileDef* upb_DefPool_FindFileByNameWithSize(const upb_DefPool* s,
|
||||
const char* name,
|
||||
size_t len);
|
||||
|
||||
const upb_FieldDef* upb_DefPool_FindExtensionByMiniTable(
|
||||
const upb_DefPool* s, const upb_MiniTableExtension* ext);
|
||||
|
||||
UPB_API const upb_FieldDef* upb_DefPool_FindExtensionByName(const upb_DefPool* s,
|
||||
const char* sym);
|
||||
|
||||
const upb_FieldDef* upb_DefPool_FindExtensionByNameWithSize(
|
||||
const upb_DefPool* s, const char* name, size_t size);
|
||||
|
||||
const upb_FieldDef* upb_DefPool_FindExtensionByNumber(const upb_DefPool* s,
|
||||
const upb_MessageDef* m,
|
||||
int32_t fieldnum);
|
||||
|
||||
const upb_ServiceDef* upb_DefPool_FindServiceByName(const upb_DefPool* s,
|
||||
const char* name);
|
||||
|
||||
const upb_ServiceDef* upb_DefPool_FindServiceByNameWithSize(
|
||||
const upb_DefPool* s, const char* name, size_t size);
|
||||
|
||||
const upb_FileDef* upb_DefPool_FindFileContainingSymbol(const upb_DefPool* s,
|
||||
const char* name);
|
||||
|
||||
UPB_API const upb_FileDef* upb_DefPool_AddFile(
|
||||
upb_DefPool* s, const UPB_DESC(FileDescriptorProto) * file_proto,
|
||||
upb_Status* status);
|
||||
|
||||
UPB_API const upb_ExtensionRegistry* upb_DefPool_ExtensionRegistry(
|
||||
const upb_DefPool* s);
|
||||
|
||||
const upb_FieldDef** upb_DefPool_GetAllExtensions(const upb_DefPool* s,
|
||||
const upb_MessageDef* m,
|
||||
size_t* count);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_DEF_POOL_H_ */
|
||||
30
Pods/gRPC-Core/third_party/upb/upb/reflection/def_type.c
generated
vendored
Normal file
30
Pods/gRPC-Core/third_party/upb/upb/reflection/def_type.c
generated
vendored
Normal file
@@ -0,0 +1,30 @@
|
||||
// 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/reflection/def_type.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
upb_deftype_t _upb_DefType_Type(upb_value v) {
|
||||
const uintptr_t num = (uintptr_t)upb_value_getconstptr(v);
|
||||
return num & UPB_DEFTYPE_MASK;
|
||||
}
|
||||
|
||||
upb_value _upb_DefType_Pack(const void* ptr, upb_deftype_t type) {
|
||||
uintptr_t num = (uintptr_t)ptr;
|
||||
UPB_ASSERT((num & UPB_DEFTYPE_MASK) == 0);
|
||||
num |= type;
|
||||
return upb_value_constptr((const void*)num);
|
||||
}
|
||||
|
||||
const void* _upb_DefType_Unpack(upb_value v, upb_deftype_t type) {
|
||||
uintptr_t num = (uintptr_t)upb_value_getconstptr(v);
|
||||
return (num & UPB_DEFTYPE_MASK) == type
|
||||
? (const void*)(num & ~UPB_DEFTYPE_MASK)
|
||||
: NULL;
|
||||
}
|
||||
60
Pods/gRPC-Core/third_party/upb/upb/reflection/def_type.h
generated
vendored
Normal file
60
Pods/gRPC-Core/third_party/upb/upb/reflection/def_type.h
generated
vendored
Normal file
@@ -0,0 +1,60 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_DEF_TYPE_H_
|
||||
#define UPB_REFLECTION_DEF_TYPE_H_
|
||||
|
||||
#include "upb/hash/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// Inside a symtab we store tagged pointers to specific def types.
|
||||
typedef enum {
|
||||
UPB_DEFTYPE_MASK = 7,
|
||||
|
||||
// Only inside symtab table.
|
||||
UPB_DEFTYPE_EXT = 0,
|
||||
UPB_DEFTYPE_MSG = 1,
|
||||
UPB_DEFTYPE_ENUM = 2,
|
||||
UPB_DEFTYPE_ENUMVAL = 3,
|
||||
UPB_DEFTYPE_SERVICE = 4,
|
||||
|
||||
// Only inside message table.
|
||||
UPB_DEFTYPE_FIELD = 0,
|
||||
UPB_DEFTYPE_ONEOF = 1,
|
||||
} upb_deftype_t;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Our 3-bit pointer tagging requires all pointers to be multiples of 8.
|
||||
// The arena will always yield 8-byte-aligned addresses, however we put
|
||||
// the defs into arrays. For each element in the array to be 8-byte-aligned,
|
||||
// the sizes of each def type must also be a multiple of 8.
|
||||
//
|
||||
// If any of these asserts fail, we need to add or remove padding on 32-bit
|
||||
// machines (64-bit machines will have 8-byte alignment already due to
|
||||
// pointers, which all of these structs have).
|
||||
UPB_INLINE void _upb_DefType_CheckPadding(size_t size) {
|
||||
UPB_ASSERT((size & UPB_DEFTYPE_MASK) == 0);
|
||||
}
|
||||
|
||||
upb_deftype_t _upb_DefType_Type(upb_value v);
|
||||
|
||||
upb_value _upb_DefType_Pack(const void* ptr, upb_deftype_t type);
|
||||
|
||||
const void* _upb_DefType_Unpack(upb_value v, upb_deftype_t type);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_DEF_TYPE_H_ */
|
||||
33
Pods/gRPC-Core/third_party/upb/upb/reflection/desc_state.c
generated
vendored
Normal file
33
Pods/gRPC-Core/third_party/upb/upb/reflection/desc_state.c
generated
vendored
Normal file
@@ -0,0 +1,33 @@
|
||||
// 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/reflection/internal/desc_state.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
bool _upb_DescState_Grow(upb_DescState* d, upb_Arena* a) {
|
||||
const size_t oldbufsize = d->bufsize;
|
||||
const int used = d->ptr - d->buf;
|
||||
|
||||
if (!d->buf) {
|
||||
d->buf = upb_Arena_Malloc(a, d->bufsize);
|
||||
if (!d->buf) return false;
|
||||
d->ptr = d->buf;
|
||||
d->e.end = d->buf + d->bufsize;
|
||||
}
|
||||
|
||||
if (oldbufsize - used < kUpb_MtDataEncoder_MinSize) {
|
||||
d->bufsize *= 2;
|
||||
d->buf = upb_Arena_Realloc(a, d->buf, oldbufsize, d->bufsize);
|
||||
if (!d->buf) return false;
|
||||
d->ptr = d->buf + used;
|
||||
d->e.end = d->buf + d->bufsize;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
292
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_def.c
generated
vendored
Normal file
292
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_def.c
generated
vendored
Normal file
@@ -0,0 +1,292 @@
|
||||
// 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/reflection/internal/enum_def.h"
|
||||
|
||||
#include "upb/hash/int_table.h"
|
||||
#include "upb/hash/str_table.h"
|
||||
#include "upb/mini_descriptor/decode.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/desc_state.h"
|
||||
#include "upb/reflection/internal/enum_reserved_range.h"
|
||||
#include "upb/reflection/internal/enum_value_def.h"
|
||||
#include "upb/reflection/internal/file_def.h"
|
||||
#include "upb/reflection/internal/message_def.h"
|
||||
#include "upb/reflection/internal/strdup2.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_EnumDef {
|
||||
const UPB_DESC(EnumOptions) * opts;
|
||||
const upb_MiniTableEnum* layout; // Only for proto2.
|
||||
const upb_FileDef* file;
|
||||
const upb_MessageDef* containing_type; // Could be merged with "file".
|
||||
const char* full_name;
|
||||
upb_strtable ntoi;
|
||||
upb_inttable iton;
|
||||
const upb_EnumValueDef* values;
|
||||
const upb_EnumReservedRange* res_ranges;
|
||||
const upb_StringView* res_names;
|
||||
int value_count;
|
||||
int res_range_count;
|
||||
int res_name_count;
|
||||
int32_t defaultval;
|
||||
bool is_closed;
|
||||
bool is_sorted; // Whether all of the values are defined in ascending order.
|
||||
};
|
||||
|
||||
upb_EnumDef* _upb_EnumDef_At(const upb_EnumDef* e, int i) {
|
||||
return (upb_EnumDef*)&e[i];
|
||||
}
|
||||
|
||||
const upb_MiniTableEnum* _upb_EnumDef_MiniTable(const upb_EnumDef* e) {
|
||||
return e->layout;
|
||||
}
|
||||
|
||||
bool _upb_EnumDef_Insert(upb_EnumDef* e, upb_EnumValueDef* v, upb_Arena* a) {
|
||||
const char* name = upb_EnumValueDef_Name(v);
|
||||
const upb_value val = upb_value_constptr(v);
|
||||
bool ok = upb_strtable_insert(&e->ntoi, name, strlen(name), val, a);
|
||||
if (!ok) return false;
|
||||
|
||||
// Multiple enumerators can have the same number, first one wins.
|
||||
const int number = upb_EnumValueDef_Number(v);
|
||||
if (!upb_inttable_lookup(&e->iton, number, NULL)) {
|
||||
return upb_inttable_insert(&e->iton, number, val, a);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
const UPB_DESC(EnumOptions) * upb_EnumDef_Options(const upb_EnumDef* e) {
|
||||
return e->opts;
|
||||
}
|
||||
|
||||
bool upb_EnumDef_HasOptions(const upb_EnumDef* e) {
|
||||
return e->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
const char* upb_EnumDef_FullName(const upb_EnumDef* e) { return e->full_name; }
|
||||
|
||||
const char* upb_EnumDef_Name(const upb_EnumDef* e) {
|
||||
return _upb_DefBuilder_FullToShort(e->full_name);
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_EnumDef_File(const upb_EnumDef* e) { return e->file; }
|
||||
|
||||
const upb_MessageDef* upb_EnumDef_ContainingType(const upb_EnumDef* e) {
|
||||
return e->containing_type;
|
||||
}
|
||||
|
||||
int32_t upb_EnumDef_Default(const upb_EnumDef* e) {
|
||||
UPB_ASSERT(upb_EnumDef_FindValueByNumber(e, e->defaultval));
|
||||
return e->defaultval;
|
||||
}
|
||||
|
||||
int upb_EnumDef_ReservedRangeCount(const upb_EnumDef* e) {
|
||||
return e->res_range_count;
|
||||
}
|
||||
|
||||
const upb_EnumReservedRange* upb_EnumDef_ReservedRange(const upb_EnumDef* e,
|
||||
int i) {
|
||||
UPB_ASSERT(0 <= i && i < e->res_range_count);
|
||||
return _upb_EnumReservedRange_At(e->res_ranges, i);
|
||||
}
|
||||
|
||||
int upb_EnumDef_ReservedNameCount(const upb_EnumDef* e) {
|
||||
return e->res_name_count;
|
||||
}
|
||||
|
||||
upb_StringView upb_EnumDef_ReservedName(const upb_EnumDef* e, int i) {
|
||||
UPB_ASSERT(0 <= i && i < e->res_name_count);
|
||||
return e->res_names[i];
|
||||
}
|
||||
|
||||
int upb_EnumDef_ValueCount(const upb_EnumDef* e) { return e->value_count; }
|
||||
|
||||
const upb_EnumValueDef* upb_EnumDef_FindValueByName(const upb_EnumDef* e,
|
||||
const char* name) {
|
||||
return upb_EnumDef_FindValueByNameWithSize(e, name, strlen(name));
|
||||
}
|
||||
|
||||
const upb_EnumValueDef* upb_EnumDef_FindValueByNameWithSize(
|
||||
const upb_EnumDef* e, const char* name, size_t size) {
|
||||
upb_value v;
|
||||
return upb_strtable_lookup2(&e->ntoi, name, size, &v)
|
||||
? upb_value_getconstptr(v)
|
||||
: NULL;
|
||||
}
|
||||
|
||||
const upb_EnumValueDef* upb_EnumDef_FindValueByNumber(const upb_EnumDef* e,
|
||||
int32_t num) {
|
||||
upb_value v;
|
||||
return upb_inttable_lookup(&e->iton, num, &v) ? upb_value_getconstptr(v)
|
||||
: NULL;
|
||||
}
|
||||
|
||||
bool upb_EnumDef_CheckNumber(const upb_EnumDef* e, int32_t num) {
|
||||
// We could use upb_EnumDef_FindValueByNumber(e, num) != NULL, but we expect
|
||||
// this to be faster (especially for small numbers).
|
||||
return upb_MiniTableEnum_CheckValue(e->layout, num);
|
||||
}
|
||||
|
||||
const upb_EnumValueDef* upb_EnumDef_Value(const upb_EnumDef* e, int i) {
|
||||
UPB_ASSERT(0 <= i && i < e->value_count);
|
||||
return _upb_EnumValueDef_At(e->values, i);
|
||||
}
|
||||
|
||||
bool upb_EnumDef_IsClosed(const upb_EnumDef* e) { return e->is_closed; }
|
||||
|
||||
bool upb_EnumDef_MiniDescriptorEncode(const upb_EnumDef* e, upb_Arena* a,
|
||||
upb_StringView* out) {
|
||||
upb_DescState s;
|
||||
_upb_DescState_Init(&s);
|
||||
|
||||
const upb_EnumValueDef** sorted = NULL;
|
||||
if (!e->is_sorted) {
|
||||
sorted = _upb_EnumValueDefs_Sorted(e->values, e->value_count, a);
|
||||
if (!sorted) return false;
|
||||
}
|
||||
|
||||
if (!_upb_DescState_Grow(&s, a)) return false;
|
||||
s.ptr = upb_MtDataEncoder_StartEnum(&s.e, s.ptr);
|
||||
|
||||
// Duplicate values are allowed but we only encode each value once.
|
||||
uint32_t previous = 0;
|
||||
|
||||
for (int i = 0; i < e->value_count; i++) {
|
||||
const uint32_t current =
|
||||
upb_EnumValueDef_Number(sorted ? sorted[i] : upb_EnumDef_Value(e, i));
|
||||
if (i != 0 && previous == current) continue;
|
||||
|
||||
if (!_upb_DescState_Grow(&s, a)) return false;
|
||||
s.ptr = upb_MtDataEncoder_PutEnumValue(&s.e, s.ptr, current);
|
||||
previous = current;
|
||||
}
|
||||
|
||||
if (!_upb_DescState_Grow(&s, a)) return false;
|
||||
s.ptr = upb_MtDataEncoder_EndEnum(&s.e, s.ptr);
|
||||
|
||||
// There will always be room for this '\0' in the encoder buffer because
|
||||
// kUpb_MtDataEncoder_MinSize is overkill for upb_MtDataEncoder_EndEnum().
|
||||
UPB_ASSERT(s.ptr < s.buf + s.bufsize);
|
||||
*s.ptr = '\0';
|
||||
|
||||
out->data = s.buf;
|
||||
out->size = s.ptr - s.buf;
|
||||
return true;
|
||||
}
|
||||
|
||||
static upb_MiniTableEnum* create_enumlayout(upb_DefBuilder* ctx,
|
||||
const upb_EnumDef* e) {
|
||||
upb_StringView sv;
|
||||
bool ok = upb_EnumDef_MiniDescriptorEncode(e, ctx->tmp_arena, &sv);
|
||||
if (!ok) _upb_DefBuilder_Errf(ctx, "OOM while building enum MiniDescriptor");
|
||||
|
||||
upb_Status status;
|
||||
upb_MiniTableEnum* layout =
|
||||
upb_MiniTableEnum_Build(sv.data, sv.size, ctx->arena, &status);
|
||||
if (!layout)
|
||||
_upb_DefBuilder_Errf(ctx, "Error building enum MiniTable: %s", status.msg);
|
||||
return layout;
|
||||
}
|
||||
|
||||
static upb_StringView* _upb_EnumReservedNames_New(
|
||||
upb_DefBuilder* ctx, int n, const upb_StringView* protos) {
|
||||
upb_StringView* sv = _upb_DefBuilder_Alloc(ctx, sizeof(upb_StringView) * n);
|
||||
for (int i = 0; i < n; i++) {
|
||||
sv[i].data =
|
||||
upb_strdup2(protos[i].data, protos[i].size, _upb_DefBuilder_Arena(ctx));
|
||||
sv[i].size = protos[i].size;
|
||||
}
|
||||
return sv;
|
||||
}
|
||||
|
||||
static void create_enumdef(upb_DefBuilder* ctx, const char* prefix,
|
||||
const UPB_DESC(EnumDescriptorProto) * enum_proto,
|
||||
upb_EnumDef* e) {
|
||||
const UPB_DESC(EnumValueDescriptorProto)* const* values;
|
||||
const UPB_DESC(EnumDescriptorProto_EnumReservedRange)* const* res_ranges;
|
||||
const upb_StringView* res_names;
|
||||
upb_StringView name;
|
||||
size_t n_value, n_res_range, n_res_name;
|
||||
|
||||
// Must happen before _upb_DefBuilder_Add()
|
||||
e->file = _upb_DefBuilder_File(ctx);
|
||||
|
||||
name = UPB_DESC(EnumDescriptorProto_name)(enum_proto);
|
||||
|
||||
e->full_name = _upb_DefBuilder_MakeFullName(ctx, prefix, name);
|
||||
_upb_DefBuilder_Add(ctx, e->full_name,
|
||||
_upb_DefType_Pack(e, UPB_DEFTYPE_ENUM));
|
||||
|
||||
e->is_closed = (!UPB_TREAT_PROTO2_ENUMS_LIKE_PROTO3) &&
|
||||
(upb_FileDef_Syntax(e->file) == kUpb_Syntax_Proto2);
|
||||
|
||||
values = UPB_DESC(EnumDescriptorProto_value)(enum_proto, &n_value);
|
||||
|
||||
bool ok = upb_strtable_init(&e->ntoi, n_value, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
ok = upb_inttable_init(&e->iton, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
e->defaultval = 0;
|
||||
e->value_count = n_value;
|
||||
e->values =
|
||||
_upb_EnumValueDefs_New(ctx, prefix, n_value, values, e, &e->is_sorted);
|
||||
|
||||
if (n_value == 0) {
|
||||
_upb_DefBuilder_Errf(ctx, "enums must contain at least one value (%s)",
|
||||
e->full_name);
|
||||
}
|
||||
|
||||
res_ranges =
|
||||
UPB_DESC(EnumDescriptorProto_reserved_range)(enum_proto, &n_res_range);
|
||||
e->res_range_count = n_res_range;
|
||||
e->res_ranges = _upb_EnumReservedRanges_New(ctx, n_res_range, res_ranges, e);
|
||||
|
||||
res_names =
|
||||
UPB_DESC(EnumDescriptorProto_reserved_name)(enum_proto, &n_res_name);
|
||||
e->res_name_count = n_res_name;
|
||||
e->res_names = _upb_EnumReservedNames_New(ctx, n_res_name, res_names);
|
||||
|
||||
UPB_DEF_SET_OPTIONS(e->opts, EnumDescriptorProto, EnumOptions, enum_proto);
|
||||
|
||||
upb_inttable_compact(&e->iton, ctx->arena);
|
||||
|
||||
if (e->is_closed) {
|
||||
if (ctx->layout) {
|
||||
UPB_ASSERT(ctx->enum_count < ctx->layout->enum_count);
|
||||
e->layout = ctx->layout->enums[ctx->enum_count++];
|
||||
} else {
|
||||
e->layout = create_enumlayout(ctx, e);
|
||||
}
|
||||
} else {
|
||||
e->layout = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
upb_EnumDef* _upb_EnumDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(EnumDescriptorProto) * const* protos,
|
||||
const upb_MessageDef* containing_type) {
|
||||
_upb_DefType_CheckPadding(sizeof(upb_EnumDef));
|
||||
|
||||
// If a containing type is defined then get the full name from that.
|
||||
// Otherwise use the package name from the file def.
|
||||
const char* name = containing_type ? upb_MessageDef_FullName(containing_type)
|
||||
: _upb_FileDef_RawPackage(ctx->file);
|
||||
|
||||
upb_EnumDef* e = _upb_DefBuilder_Alloc(ctx, sizeof(upb_EnumDef) * n);
|
||||
for (int i = 0; i < n; i++) {
|
||||
create_enumdef(ctx, name, protos[i], &e[i]);
|
||||
e[i].containing_type = containing_type;
|
||||
}
|
||||
return e;
|
||||
}
|
||||
60
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_def.h
generated
vendored
Normal file
60
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_def.h
generated
vendored
Normal file
@@ -0,0 +1,60 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_ENUM_DEF_H_
|
||||
#define UPB_REFLECTION_ENUM_DEF_H_
|
||||
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
bool upb_EnumDef_CheckNumber(const upb_EnumDef* e, int32_t num);
|
||||
const upb_MessageDef* upb_EnumDef_ContainingType(const upb_EnumDef* e);
|
||||
int32_t upb_EnumDef_Default(const upb_EnumDef* e);
|
||||
UPB_API const upb_FileDef* upb_EnumDef_File(const upb_EnumDef* e);
|
||||
const upb_EnumValueDef* upb_EnumDef_FindValueByName(const upb_EnumDef* e,
|
||||
const char* name);
|
||||
UPB_API const upb_EnumValueDef* upb_EnumDef_FindValueByNameWithSize(
|
||||
const upb_EnumDef* e, const char* name, size_t size);
|
||||
UPB_API const upb_EnumValueDef* upb_EnumDef_FindValueByNumber(
|
||||
const upb_EnumDef* e, int32_t num);
|
||||
UPB_API const char* upb_EnumDef_FullName(const upb_EnumDef* e);
|
||||
bool upb_EnumDef_HasOptions(const upb_EnumDef* e);
|
||||
bool upb_EnumDef_IsClosed(const upb_EnumDef* e);
|
||||
|
||||
// Creates a mini descriptor string for an enum, returns true on success.
|
||||
bool upb_EnumDef_MiniDescriptorEncode(const upb_EnumDef* e, upb_Arena* a,
|
||||
upb_StringView* out);
|
||||
|
||||
const char* upb_EnumDef_Name(const upb_EnumDef* e);
|
||||
const UPB_DESC(EnumOptions) * upb_EnumDef_Options(const upb_EnumDef* e);
|
||||
|
||||
upb_StringView upb_EnumDef_ReservedName(const upb_EnumDef* e, int i);
|
||||
int upb_EnumDef_ReservedNameCount(const upb_EnumDef* e);
|
||||
|
||||
const upb_EnumReservedRange* upb_EnumDef_ReservedRange(const upb_EnumDef* e,
|
||||
int i);
|
||||
int upb_EnumDef_ReservedRangeCount(const upb_EnumDef* e);
|
||||
|
||||
UPB_API const upb_EnumValueDef* upb_EnumDef_Value(const upb_EnumDef* e, int i);
|
||||
UPB_API int upb_EnumDef_ValueCount(const upb_EnumDef* e);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_ENUM_DEF_H_ */
|
||||
63
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_reserved_range.c
generated
vendored
Normal file
63
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_reserved_range.c
generated
vendored
Normal file
@@ -0,0 +1,63 @@
|
||||
// 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/reflection/internal/enum_reserved_range.h"
|
||||
|
||||
#include "upb/reflection/enum_def.h"
|
||||
#include "upb/reflection/field_def.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_EnumReservedRange {
|
||||
int32_t start;
|
||||
int32_t end;
|
||||
};
|
||||
|
||||
upb_EnumReservedRange* _upb_EnumReservedRange_At(const upb_EnumReservedRange* r,
|
||||
int i) {
|
||||
return (upb_EnumReservedRange*)&r[i];
|
||||
}
|
||||
|
||||
int32_t upb_EnumReservedRange_Start(const upb_EnumReservedRange* r) {
|
||||
return r->start;
|
||||
}
|
||||
int32_t upb_EnumReservedRange_End(const upb_EnumReservedRange* r) {
|
||||
return r->end;
|
||||
}
|
||||
|
||||
upb_EnumReservedRange* _upb_EnumReservedRanges_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(EnumDescriptorProto_EnumReservedRange) * const* protos,
|
||||
const upb_EnumDef* e) {
|
||||
upb_EnumReservedRange* r =
|
||||
_upb_DefBuilder_Alloc(ctx, sizeof(upb_EnumReservedRange) * n);
|
||||
|
||||
for (int i = 0; i < n; i++) {
|
||||
const int32_t start =
|
||||
UPB_DESC(EnumDescriptorProto_EnumReservedRange_start)(protos[i]);
|
||||
const int32_t end =
|
||||
UPB_DESC(EnumDescriptorProto_EnumReservedRange_end)(protos[i]);
|
||||
|
||||
// A full validation would also check that each range is disjoint, and that
|
||||
// none of the fields overlap with the extension ranges, but we are just
|
||||
// sanity checking here.
|
||||
|
||||
// Note: Not a typo! Unlike extension ranges and message reserved ranges,
|
||||
// the end value of an enum reserved range is *inclusive*!
|
||||
if (end < start) {
|
||||
_upb_DefBuilder_Errf(ctx, "Reserved range (%d, %d) is invalid, enum=%s\n",
|
||||
(int)start, (int)end, upb_EnumDef_FullName(e));
|
||||
}
|
||||
|
||||
r[i].start = start;
|
||||
r[i].end = end;
|
||||
}
|
||||
|
||||
return r;
|
||||
}
|
||||
31
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_reserved_range.h
generated
vendored
Normal file
31
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_reserved_range.h
generated
vendored
Normal file
@@ -0,0 +1,31 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_ENUM_RESERVED_RANGE_H_
|
||||
#define UPB_REFLECTION_ENUM_RESERVED_RANGE_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
int32_t upb_EnumReservedRange_Start(const upb_EnumReservedRange* r);
|
||||
int32_t upb_EnumReservedRange_End(const upb_EnumReservedRange* r);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_ENUM_RESERVED_RANGE_H_ */
|
||||
125
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_value_def.c
generated
vendored
Normal file
125
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_value_def.c
generated
vendored
Normal file
@@ -0,0 +1,125 @@
|
||||
// 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/reflection/internal/enum_value_def.h"
|
||||
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/enum_def.h"
|
||||
#include "upb/reflection/internal/file_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_EnumValueDef {
|
||||
const UPB_DESC(EnumValueOptions) * opts;
|
||||
const upb_EnumDef* parent;
|
||||
const char* full_name;
|
||||
int32_t number;
|
||||
};
|
||||
|
||||
upb_EnumValueDef* _upb_EnumValueDef_At(const upb_EnumValueDef* v, int i) {
|
||||
return (upb_EnumValueDef*)&v[i];
|
||||
}
|
||||
|
||||
static int _upb_EnumValueDef_Compare(const void* p1, const void* p2) {
|
||||
const uint32_t v1 = (*(const upb_EnumValueDef**)p1)->number;
|
||||
const uint32_t v2 = (*(const upb_EnumValueDef**)p2)->number;
|
||||
return (v1 < v2) ? -1 : (v1 > v2);
|
||||
}
|
||||
|
||||
const upb_EnumValueDef** _upb_EnumValueDefs_Sorted(const upb_EnumValueDef* v,
|
||||
int n, upb_Arena* a) {
|
||||
// TODO: Try to replace this arena alloc with a persistent scratch buffer.
|
||||
upb_EnumValueDef** out =
|
||||
(upb_EnumValueDef**)upb_Arena_Malloc(a, n * sizeof(void*));
|
||||
if (!out) return NULL;
|
||||
|
||||
for (int i = 0; i < n; i++) {
|
||||
out[i] = (upb_EnumValueDef*)&v[i];
|
||||
}
|
||||
qsort(out, n, sizeof(void*), _upb_EnumValueDef_Compare);
|
||||
|
||||
return (const upb_EnumValueDef**)out;
|
||||
}
|
||||
|
||||
const UPB_DESC(EnumValueOptions) *
|
||||
upb_EnumValueDef_Options(const upb_EnumValueDef* v) {
|
||||
return v->opts;
|
||||
}
|
||||
|
||||
bool upb_EnumValueDef_HasOptions(const upb_EnumValueDef* v) {
|
||||
return v->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
const upb_EnumDef* upb_EnumValueDef_Enum(const upb_EnumValueDef* v) {
|
||||
return v->parent;
|
||||
}
|
||||
|
||||
const char* upb_EnumValueDef_FullName(const upb_EnumValueDef* v) {
|
||||
return v->full_name;
|
||||
}
|
||||
|
||||
const char* upb_EnumValueDef_Name(const upb_EnumValueDef* v) {
|
||||
return _upb_DefBuilder_FullToShort(v->full_name);
|
||||
}
|
||||
|
||||
int32_t upb_EnumValueDef_Number(const upb_EnumValueDef* v) { return v->number; }
|
||||
|
||||
uint32_t upb_EnumValueDef_Index(const upb_EnumValueDef* v) {
|
||||
// Compute index in our parent's array.
|
||||
return v - upb_EnumDef_Value(v->parent, 0);
|
||||
}
|
||||
|
||||
static void create_enumvaldef(upb_DefBuilder* ctx, const char* prefix,
|
||||
const UPB_DESC(EnumValueDescriptorProto) *
|
||||
val_proto,
|
||||
upb_EnumDef* e, upb_EnumValueDef* v) {
|
||||
upb_StringView name = UPB_DESC(EnumValueDescriptorProto_name)(val_proto);
|
||||
|
||||
v->parent = e; // Must happen prior to _upb_DefBuilder_Add()
|
||||
v->full_name = _upb_DefBuilder_MakeFullName(ctx, prefix, name);
|
||||
v->number = UPB_DESC(EnumValueDescriptorProto_number)(val_proto);
|
||||
_upb_DefBuilder_Add(ctx, v->full_name,
|
||||
_upb_DefType_Pack(v, UPB_DEFTYPE_ENUMVAL));
|
||||
|
||||
UPB_DEF_SET_OPTIONS(v->opts, EnumValueDescriptorProto, EnumValueOptions,
|
||||
val_proto);
|
||||
|
||||
bool ok = _upb_EnumDef_Insert(e, v, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
}
|
||||
|
||||
// Allocate and initialize an array of |n| enum value defs owned by |e|.
|
||||
upb_EnumValueDef* _upb_EnumValueDefs_New(
|
||||
upb_DefBuilder* ctx, const char* prefix, int n,
|
||||
const UPB_DESC(EnumValueDescriptorProto) * const* protos, upb_EnumDef* e,
|
||||
bool* is_sorted) {
|
||||
_upb_DefType_CheckPadding(sizeof(upb_EnumValueDef));
|
||||
|
||||
upb_EnumValueDef* v =
|
||||
_upb_DefBuilder_Alloc(ctx, sizeof(upb_EnumValueDef) * n);
|
||||
|
||||
*is_sorted = true;
|
||||
uint32_t previous = 0;
|
||||
for (int i = 0; i < n; i++) {
|
||||
create_enumvaldef(ctx, prefix, protos[i], e, &v[i]);
|
||||
|
||||
const uint32_t current = v[i].number;
|
||||
if (previous > current) *is_sorted = false;
|
||||
previous = current;
|
||||
}
|
||||
|
||||
if (upb_FileDef_Syntax(ctx->file) == kUpb_Syntax_Proto3 && n > 0 &&
|
||||
v[0].number != 0) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"for proto3, the first enum value must be zero (%s)",
|
||||
upb_EnumDef_FullName(e));
|
||||
}
|
||||
|
||||
return v;
|
||||
}
|
||||
37
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_value_def.h
generated
vendored
Normal file
37
Pods/gRPC-Core/third_party/upb/upb/reflection/enum_value_def.h
generated
vendored
Normal file
@@ -0,0 +1,37 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_ENUM_VALUE_DEF_H_
|
||||
#define UPB_REFLECTION_ENUM_VALUE_DEF_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
const upb_EnumDef* upb_EnumValueDef_Enum(const upb_EnumValueDef* v);
|
||||
const char* upb_EnumValueDef_FullName(const upb_EnumValueDef* v);
|
||||
bool upb_EnumValueDef_HasOptions(const upb_EnumValueDef* v);
|
||||
uint32_t upb_EnumValueDef_Index(const upb_EnumValueDef* v);
|
||||
UPB_API const char* upb_EnumValueDef_Name(const upb_EnumValueDef* v);
|
||||
UPB_API int32_t upb_EnumValueDef_Number(const upb_EnumValueDef* v);
|
||||
const UPB_DESC(EnumValueOptions) *
|
||||
upb_EnumValueDef_Options(const upb_EnumValueDef* v);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_ENUM_VALUE_DEF_H_ */
|
||||
74
Pods/gRPC-Core/third_party/upb/upb/reflection/extension_range.c
generated
vendored
Normal file
74
Pods/gRPC-Core/third_party/upb/upb/reflection/extension_range.c
generated
vendored
Normal file
@@ -0,0 +1,74 @@
|
||||
// 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/reflection/internal/extension_range.h"
|
||||
|
||||
#include "upb/reflection/field_def.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/message_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_ExtensionRange {
|
||||
const UPB_DESC(ExtensionRangeOptions) * opts;
|
||||
int32_t start;
|
||||
int32_t end;
|
||||
};
|
||||
|
||||
upb_ExtensionRange* _upb_ExtensionRange_At(const upb_ExtensionRange* r, int i) {
|
||||
return (upb_ExtensionRange*)&r[i];
|
||||
}
|
||||
|
||||
const UPB_DESC(ExtensionRangeOptions) *
|
||||
upb_ExtensionRange_Options(const upb_ExtensionRange* r) {
|
||||
return r->opts;
|
||||
}
|
||||
|
||||
bool upb_ExtensionRange_HasOptions(const upb_ExtensionRange* r) {
|
||||
return r->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
int32_t upb_ExtensionRange_Start(const upb_ExtensionRange* r) {
|
||||
return r->start;
|
||||
}
|
||||
|
||||
int32_t upb_ExtensionRange_End(const upb_ExtensionRange* r) { return r->end; }
|
||||
|
||||
upb_ExtensionRange* _upb_ExtensionRanges_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(DescriptorProto_ExtensionRange) * const* protos,
|
||||
const upb_MessageDef* m) {
|
||||
upb_ExtensionRange* r =
|
||||
_upb_DefBuilder_Alloc(ctx, sizeof(upb_ExtensionRange) * n);
|
||||
|
||||
for (int i = 0; i < n; i++) {
|
||||
const int32_t start =
|
||||
UPB_DESC(DescriptorProto_ExtensionRange_start)(protos[i]);
|
||||
const int32_t end = UPB_DESC(DescriptorProto_ExtensionRange_end)(protos[i]);
|
||||
const int32_t max = UPB_DESC(MessageOptions_message_set_wire_format)(
|
||||
upb_MessageDef_Options(m))
|
||||
? INT32_MAX
|
||||
: kUpb_MaxFieldNumber + 1;
|
||||
|
||||
// A full validation would also check that each range is disjoint, and that
|
||||
// none of the fields overlap with the extension ranges, but we are just
|
||||
// sanity checking here.
|
||||
if (start < 1 || end <= start || end > max) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"Extension range (%d, %d) is invalid, message=%s\n",
|
||||
(int)start, (int)end, upb_MessageDef_FullName(m));
|
||||
}
|
||||
|
||||
r[i].start = start;
|
||||
r[i].end = end;
|
||||
UPB_DEF_SET_OPTIONS(r[i].opts, DescriptorProto_ExtensionRange,
|
||||
ExtensionRangeOptions, protos[i]);
|
||||
}
|
||||
|
||||
return r;
|
||||
}
|
||||
35
Pods/gRPC-Core/third_party/upb/upb/reflection/extension_range.h
generated
vendored
Normal file
35
Pods/gRPC-Core/third_party/upb/upb/reflection/extension_range.h
generated
vendored
Normal file
@@ -0,0 +1,35 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_EXTENSION_RANGE_H_
|
||||
#define UPB_REFLECTION_EXTENSION_RANGE_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
int32_t upb_ExtensionRange_Start(const upb_ExtensionRange* r);
|
||||
int32_t upb_ExtensionRange_End(const upb_ExtensionRange* r);
|
||||
|
||||
bool upb_ExtensionRange_HasOptions(const upb_ExtensionRange* r);
|
||||
const UPB_DESC(ExtensionRangeOptions) *
|
||||
upb_ExtensionRange_Options(const upb_ExtensionRange* r);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_EXTENSION_RANGE_H_ */
|
||||
948
Pods/gRPC-Core/third_party/upb/upb/reflection/field_def.c
generated
vendored
Normal file
948
Pods/gRPC-Core/third_party/upb/upb/reflection/field_def.c
generated
vendored
Normal file
@@ -0,0 +1,948 @@
|
||||
// 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/reflection/internal/field_def.h"
|
||||
|
||||
#include <ctype.h>
|
||||
#include <errno.h>
|
||||
#include <stdbool.h>
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/mini_descriptor/decode.h"
|
||||
#include "upb/mini_descriptor/internal/modifiers.h"
|
||||
#include "upb/reflection/def.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/desc_state.h"
|
||||
#include "upb/reflection/internal/enum_def.h"
|
||||
#include "upb/reflection/internal/file_def.h"
|
||||
#include "upb/reflection/internal/message_def.h"
|
||||
#include "upb/reflection/internal/oneof_def.h"
|
||||
#include "upb/reflection/internal/strdup2.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#define UPB_FIELD_TYPE_UNSPECIFIED 0
|
||||
|
||||
typedef struct {
|
||||
size_t len;
|
||||
char str[1]; // Null-terminated string data follows.
|
||||
} str_t;
|
||||
|
||||
struct upb_FieldDef {
|
||||
const UPB_DESC(FieldOptions) * opts;
|
||||
const upb_FileDef* file;
|
||||
const upb_MessageDef* msgdef;
|
||||
const char* full_name;
|
||||
const char* json_name;
|
||||
union {
|
||||
int64_t sint;
|
||||
uint64_t uint;
|
||||
double dbl;
|
||||
float flt;
|
||||
bool boolean;
|
||||
str_t* str;
|
||||
void* msg; // Always NULL.
|
||||
} defaultval;
|
||||
union {
|
||||
const upb_OneofDef* oneof;
|
||||
const upb_MessageDef* extension_scope;
|
||||
} scope;
|
||||
union {
|
||||
const upb_MessageDef* msgdef;
|
||||
const upb_EnumDef* enumdef;
|
||||
const UPB_DESC(FieldDescriptorProto) * unresolved;
|
||||
} sub;
|
||||
uint32_t number_;
|
||||
uint16_t index_;
|
||||
uint16_t layout_index; // Index into msgdef->layout->fields or file->exts
|
||||
bool has_default;
|
||||
bool has_json_name;
|
||||
bool has_presence;
|
||||
bool is_extension;
|
||||
bool is_packed;
|
||||
bool is_proto3_optional;
|
||||
upb_FieldType type_;
|
||||
upb_Label label_;
|
||||
#if UINTPTR_MAX == 0xffffffff
|
||||
uint32_t padding; // Increase size to a multiple of 8.
|
||||
#endif
|
||||
};
|
||||
|
||||
upb_FieldDef* _upb_FieldDef_At(const upb_FieldDef* f, int i) {
|
||||
return (upb_FieldDef*)&f[i];
|
||||
}
|
||||
|
||||
const UPB_DESC(FieldOptions) * upb_FieldDef_Options(const upb_FieldDef* f) {
|
||||
return f->opts;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_HasOptions(const upb_FieldDef* f) {
|
||||
return f->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
const char* upb_FieldDef_FullName(const upb_FieldDef* f) {
|
||||
return f->full_name;
|
||||
}
|
||||
|
||||
upb_CType upb_FieldDef_CType(const upb_FieldDef* f) {
|
||||
switch (f->type_) {
|
||||
case kUpb_FieldType_Double:
|
||||
return kUpb_CType_Double;
|
||||
case kUpb_FieldType_Float:
|
||||
return kUpb_CType_Float;
|
||||
case kUpb_FieldType_Int64:
|
||||
case kUpb_FieldType_SInt64:
|
||||
case kUpb_FieldType_SFixed64:
|
||||
return kUpb_CType_Int64;
|
||||
case kUpb_FieldType_Int32:
|
||||
case kUpb_FieldType_SFixed32:
|
||||
case kUpb_FieldType_SInt32:
|
||||
return kUpb_CType_Int32;
|
||||
case kUpb_FieldType_UInt64:
|
||||
case kUpb_FieldType_Fixed64:
|
||||
return kUpb_CType_UInt64;
|
||||
case kUpb_FieldType_UInt32:
|
||||
case kUpb_FieldType_Fixed32:
|
||||
return kUpb_CType_UInt32;
|
||||
case kUpb_FieldType_Enum:
|
||||
return kUpb_CType_Enum;
|
||||
case kUpb_FieldType_Bool:
|
||||
return kUpb_CType_Bool;
|
||||
case kUpb_FieldType_String:
|
||||
return kUpb_CType_String;
|
||||
case kUpb_FieldType_Bytes:
|
||||
return kUpb_CType_Bytes;
|
||||
case kUpb_FieldType_Group:
|
||||
case kUpb_FieldType_Message:
|
||||
return kUpb_CType_Message;
|
||||
}
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
upb_FieldType upb_FieldDef_Type(const upb_FieldDef* f) { return f->type_; }
|
||||
|
||||
uint32_t upb_FieldDef_Index(const upb_FieldDef* f) { return f->index_; }
|
||||
|
||||
upb_Label upb_FieldDef_Label(const upb_FieldDef* f) { return f->label_; }
|
||||
|
||||
uint32_t upb_FieldDef_Number(const upb_FieldDef* f) { return f->number_; }
|
||||
|
||||
bool upb_FieldDef_IsExtension(const upb_FieldDef* f) { return f->is_extension; }
|
||||
|
||||
bool _upb_FieldDef_IsPackable(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_IsRepeated(f) && upb_FieldDef_IsPrimitive(f);
|
||||
}
|
||||
|
||||
bool upb_FieldDef_IsPacked(const upb_FieldDef* f) {
|
||||
return _upb_FieldDef_IsPackable(f) && f->is_packed;
|
||||
}
|
||||
|
||||
const char* upb_FieldDef_Name(const upb_FieldDef* f) {
|
||||
return _upb_DefBuilder_FullToShort(f->full_name);
|
||||
}
|
||||
|
||||
const char* upb_FieldDef_JsonName(const upb_FieldDef* f) {
|
||||
return f->json_name;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_HasJsonName(const upb_FieldDef* f) {
|
||||
return f->has_json_name;
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_FieldDef_File(const upb_FieldDef* f) { return f->file; }
|
||||
|
||||
const upb_MessageDef* upb_FieldDef_ContainingType(const upb_FieldDef* f) {
|
||||
return f->msgdef;
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_FieldDef_ExtensionScope(const upb_FieldDef* f) {
|
||||
return f->is_extension ? f->scope.extension_scope : NULL;
|
||||
}
|
||||
|
||||
const upb_OneofDef* upb_FieldDef_ContainingOneof(const upb_FieldDef* f) {
|
||||
return f->is_extension ? NULL : f->scope.oneof;
|
||||
}
|
||||
|
||||
const upb_OneofDef* upb_FieldDef_RealContainingOneof(const upb_FieldDef* f) {
|
||||
const upb_OneofDef* oneof = upb_FieldDef_ContainingOneof(f);
|
||||
if (!oneof || upb_OneofDef_IsSynthetic(oneof)) return NULL;
|
||||
return oneof;
|
||||
}
|
||||
|
||||
upb_MessageValue upb_FieldDef_Default(const upb_FieldDef* f) {
|
||||
upb_MessageValue ret;
|
||||
|
||||
if (upb_FieldDef_IsRepeated(f) || upb_FieldDef_IsSubMessage(f)) {
|
||||
return (upb_MessageValue){.msg_val = NULL};
|
||||
}
|
||||
|
||||
switch (upb_FieldDef_CType(f)) {
|
||||
case kUpb_CType_Bool:
|
||||
return (upb_MessageValue){.bool_val = f->defaultval.boolean};
|
||||
case kUpb_CType_Int64:
|
||||
return (upb_MessageValue){.int64_val = f->defaultval.sint};
|
||||
case kUpb_CType_UInt64:
|
||||
return (upb_MessageValue){.uint64_val = f->defaultval.uint};
|
||||
case kUpb_CType_Enum:
|
||||
case kUpb_CType_Int32:
|
||||
return (upb_MessageValue){.int32_val = (int32_t)f->defaultval.sint};
|
||||
case kUpb_CType_UInt32:
|
||||
return (upb_MessageValue){.uint32_val = (uint32_t)f->defaultval.uint};
|
||||
case kUpb_CType_Float:
|
||||
return (upb_MessageValue){.float_val = f->defaultval.flt};
|
||||
case kUpb_CType_Double:
|
||||
return (upb_MessageValue){.double_val = f->defaultval.dbl};
|
||||
case kUpb_CType_String:
|
||||
case kUpb_CType_Bytes: {
|
||||
str_t* str = f->defaultval.str;
|
||||
if (str) {
|
||||
return (upb_MessageValue){
|
||||
.str_val = (upb_StringView){.data = str->str, .size = str->len}};
|
||||
} else {
|
||||
return (upb_MessageValue){
|
||||
.str_val = (upb_StringView){.data = NULL, .size = 0}};
|
||||
}
|
||||
}
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_FieldDef_MessageSubDef(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_CType(f) == kUpb_CType_Message ? f->sub.msgdef : NULL;
|
||||
}
|
||||
|
||||
const upb_EnumDef* upb_FieldDef_EnumSubDef(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_CType(f) == kUpb_CType_Enum ? f->sub.enumdef : NULL;
|
||||
}
|
||||
|
||||
const upb_MiniTableField* upb_FieldDef_MiniTable(const upb_FieldDef* f) {
|
||||
if (upb_FieldDef_IsExtension(f)) {
|
||||
const upb_FileDef* file = upb_FieldDef_File(f);
|
||||
return (upb_MiniTableField*)_upb_FileDef_ExtensionMiniTable(
|
||||
file, f->layout_index);
|
||||
} else {
|
||||
const upb_MiniTable* layout = upb_MessageDef_MiniTable(f->msgdef);
|
||||
return &layout->fields[f->layout_index];
|
||||
}
|
||||
}
|
||||
|
||||
const upb_MiniTableExtension* _upb_FieldDef_ExtensionMiniTable(
|
||||
const upb_FieldDef* f) {
|
||||
UPB_ASSERT(upb_FieldDef_IsExtension(f));
|
||||
const upb_FileDef* file = upb_FieldDef_File(f);
|
||||
return _upb_FileDef_ExtensionMiniTable(file, f->layout_index);
|
||||
}
|
||||
|
||||
bool _upb_FieldDef_IsClosedEnum(const upb_FieldDef* f) {
|
||||
if (f->type_ != kUpb_FieldType_Enum) return false;
|
||||
return upb_EnumDef_IsClosed(f->sub.enumdef);
|
||||
}
|
||||
|
||||
bool _upb_FieldDef_IsProto3Optional(const upb_FieldDef* f) {
|
||||
return f->is_proto3_optional;
|
||||
}
|
||||
|
||||
int _upb_FieldDef_LayoutIndex(const upb_FieldDef* f) { return f->layout_index; }
|
||||
|
||||
// begin:google_only
|
||||
// static bool _upb_FieldDef_EnforceUtf8Option(const upb_FieldDef* f) {
|
||||
// #if defined(UPB_BOOTSTRAP_STAGE0)
|
||||
// return true;
|
||||
// #else
|
||||
// return UPB_DESC(FieldOptions_enforce_utf8)(f->opts);
|
||||
// #endif
|
||||
// }
|
||||
// end:google_only
|
||||
|
||||
// begin:github_only
|
||||
static bool _upb_FieldDef_EnforceUtf8Option(const upb_FieldDef* f) {
|
||||
return true;
|
||||
}
|
||||
// end:github_only
|
||||
|
||||
bool _upb_FieldDef_ValidateUtf8(const upb_FieldDef* f) {
|
||||
if (upb_FieldDef_Type(f) != kUpb_FieldType_String) return false;
|
||||
return upb_FileDef_Syntax(upb_FieldDef_File(f)) == kUpb_Syntax_Proto3
|
||||
? _upb_FieldDef_EnforceUtf8Option(f)
|
||||
: false;
|
||||
}
|
||||
|
||||
uint64_t _upb_FieldDef_Modifiers(const upb_FieldDef* f) {
|
||||
uint64_t out = upb_FieldDef_IsPacked(f) ? kUpb_FieldModifier_IsPacked : 0;
|
||||
|
||||
switch (f->label_) {
|
||||
case kUpb_Label_Optional:
|
||||
if (!upb_FieldDef_HasPresence(f)) {
|
||||
out |= kUpb_FieldModifier_IsProto3Singular;
|
||||
}
|
||||
break;
|
||||
case kUpb_Label_Repeated:
|
||||
out |= kUpb_FieldModifier_IsRepeated;
|
||||
break;
|
||||
case kUpb_Label_Required:
|
||||
out |= kUpb_FieldModifier_IsRequired;
|
||||
break;
|
||||
}
|
||||
|
||||
if (_upb_FieldDef_IsClosedEnum(f)) {
|
||||
out |= kUpb_FieldModifier_IsClosedEnum;
|
||||
}
|
||||
|
||||
if (_upb_FieldDef_ValidateUtf8(f)) {
|
||||
out |= kUpb_FieldModifier_ValidateUtf8;
|
||||
}
|
||||
|
||||
return out;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_HasDefault(const upb_FieldDef* f) { return f->has_default; }
|
||||
bool upb_FieldDef_HasPresence(const upb_FieldDef* f) { return f->has_presence; }
|
||||
|
||||
bool upb_FieldDef_HasSubDef(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_IsSubMessage(f) ||
|
||||
upb_FieldDef_CType(f) == kUpb_CType_Enum;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_IsMap(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_IsRepeated(f) && upb_FieldDef_IsSubMessage(f) &&
|
||||
upb_MessageDef_IsMapEntry(upb_FieldDef_MessageSubDef(f));
|
||||
}
|
||||
|
||||
bool upb_FieldDef_IsOptional(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_Label(f) == kUpb_Label_Optional;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_IsPrimitive(const upb_FieldDef* f) {
|
||||
return !upb_FieldDef_IsString(f) && !upb_FieldDef_IsSubMessage(f);
|
||||
}
|
||||
|
||||
bool upb_FieldDef_IsRepeated(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_Label(f) == kUpb_Label_Repeated;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_IsRequired(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_Label(f) == kUpb_Label_Required;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_IsString(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_CType(f) == kUpb_CType_String ||
|
||||
upb_FieldDef_CType(f) == kUpb_CType_Bytes;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_IsSubMessage(const upb_FieldDef* f) {
|
||||
return upb_FieldDef_CType(f) == kUpb_CType_Message;
|
||||
}
|
||||
|
||||
static bool between(int32_t x, int32_t low, int32_t high) {
|
||||
return x >= low && x <= high;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_checklabel(int32_t label) { return between(label, 1, 3); }
|
||||
bool upb_FieldDef_checktype(int32_t type) { return between(type, 1, 11); }
|
||||
bool upb_FieldDef_checkintfmt(int32_t fmt) { return between(fmt, 1, 3); }
|
||||
|
||||
bool upb_FieldDef_checkdescriptortype(int32_t type) {
|
||||
return between(type, 1, 18);
|
||||
}
|
||||
|
||||
static bool streql2(const char* a, size_t n, const char* b) {
|
||||
return n == strlen(b) && memcmp(a, b, n) == 0;
|
||||
}
|
||||
|
||||
// Implement the transformation as described in the spec:
|
||||
// 1. upper case all letters after an underscore.
|
||||
// 2. remove all underscores.
|
||||
static char* make_json_name(const char* name, size_t size, upb_Arena* a) {
|
||||
char* out = upb_Arena_Malloc(a, size + 1); // +1 is to add a trailing '\0'
|
||||
if (out == NULL) return NULL;
|
||||
|
||||
bool ucase_next = false;
|
||||
char* des = out;
|
||||
for (size_t i = 0; i < size; i++) {
|
||||
if (name[i] == '_') {
|
||||
ucase_next = true;
|
||||
} else {
|
||||
*des++ = ucase_next ? toupper(name[i]) : name[i];
|
||||
ucase_next = false;
|
||||
}
|
||||
}
|
||||
*des++ = '\0';
|
||||
return out;
|
||||
}
|
||||
|
||||
static str_t* newstr(upb_DefBuilder* ctx, const char* data, size_t len) {
|
||||
str_t* ret = _upb_DefBuilder_Alloc(ctx, sizeof(*ret) + len);
|
||||
if (!ret) _upb_DefBuilder_OomErr(ctx);
|
||||
ret->len = len;
|
||||
if (len) memcpy(ret->str, data, len);
|
||||
ret->str[len] = '\0';
|
||||
return ret;
|
||||
}
|
||||
|
||||
static str_t* unescape(upb_DefBuilder* ctx, const upb_FieldDef* f,
|
||||
const char* data, size_t len) {
|
||||
// Size here is an upper bound; escape sequences could ultimately shrink it.
|
||||
str_t* ret = _upb_DefBuilder_Alloc(ctx, sizeof(*ret) + len);
|
||||
char* dst = &ret->str[0];
|
||||
const char* src = data;
|
||||
const char* end = data + len;
|
||||
|
||||
while (src < end) {
|
||||
if (*src == '\\') {
|
||||
src++;
|
||||
*dst++ = _upb_DefBuilder_ParseEscape(ctx, f, &src, end);
|
||||
} else {
|
||||
*dst++ = *src++;
|
||||
}
|
||||
}
|
||||
|
||||
ret->len = dst - &ret->str[0];
|
||||
return ret;
|
||||
}
|
||||
|
||||
static void parse_default(upb_DefBuilder* ctx, const char* str, size_t len,
|
||||
upb_FieldDef* f) {
|
||||
char* end;
|
||||
char nullz[64];
|
||||
errno = 0;
|
||||
|
||||
switch (upb_FieldDef_CType(f)) {
|
||||
case kUpb_CType_Int32:
|
||||
case kUpb_CType_Int64:
|
||||
case kUpb_CType_UInt32:
|
||||
case kUpb_CType_UInt64:
|
||||
case kUpb_CType_Double:
|
||||
case kUpb_CType_Float:
|
||||
// Standard C number parsing functions expect null-terminated strings.
|
||||
if (len >= sizeof(nullz) - 1) {
|
||||
_upb_DefBuilder_Errf(ctx, "Default too long: %.*s", (int)len, str);
|
||||
}
|
||||
memcpy(nullz, str, len);
|
||||
nullz[len] = '\0';
|
||||
str = nullz;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
switch (upb_FieldDef_CType(f)) {
|
||||
case kUpb_CType_Int32: {
|
||||
long val = strtol(str, &end, 0);
|
||||
if (val > INT32_MAX || val < INT32_MIN || errno == ERANGE || *end) {
|
||||
goto invalid;
|
||||
}
|
||||
f->defaultval.sint = val;
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_Enum: {
|
||||
const upb_EnumDef* e = f->sub.enumdef;
|
||||
const upb_EnumValueDef* ev =
|
||||
upb_EnumDef_FindValueByNameWithSize(e, str, len);
|
||||
if (!ev) {
|
||||
goto invalid;
|
||||
}
|
||||
f->defaultval.sint = upb_EnumValueDef_Number(ev);
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_Int64: {
|
||||
long long val = strtoll(str, &end, 0);
|
||||
if (val > INT64_MAX || val < INT64_MIN || errno == ERANGE || *end) {
|
||||
goto invalid;
|
||||
}
|
||||
f->defaultval.sint = val;
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_UInt32: {
|
||||
unsigned long val = strtoul(str, &end, 0);
|
||||
if (val > UINT32_MAX || errno == ERANGE || *end) {
|
||||
goto invalid;
|
||||
}
|
||||
f->defaultval.uint = val;
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_UInt64: {
|
||||
unsigned long long val = strtoull(str, &end, 0);
|
||||
if (val > UINT64_MAX || errno == ERANGE || *end) {
|
||||
goto invalid;
|
||||
}
|
||||
f->defaultval.uint = val;
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_Double: {
|
||||
double val = strtod(str, &end);
|
||||
if (errno == ERANGE || *end) {
|
||||
goto invalid;
|
||||
}
|
||||
f->defaultval.dbl = val;
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_Float: {
|
||||
float val = strtof(str, &end);
|
||||
if (errno == ERANGE || *end) {
|
||||
goto invalid;
|
||||
}
|
||||
f->defaultval.flt = val;
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_Bool: {
|
||||
if (streql2(str, len, "false")) {
|
||||
f->defaultval.boolean = false;
|
||||
} else if (streql2(str, len, "true")) {
|
||||
f->defaultval.boolean = true;
|
||||
} else {
|
||||
goto invalid;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_String:
|
||||
f->defaultval.str = newstr(ctx, str, len);
|
||||
break;
|
||||
case kUpb_CType_Bytes:
|
||||
f->defaultval.str = unescape(ctx, f, str, len);
|
||||
break;
|
||||
case kUpb_CType_Message:
|
||||
/* Should not have a default value. */
|
||||
_upb_DefBuilder_Errf(ctx, "Message should not have a default (%s)",
|
||||
upb_FieldDef_FullName(f));
|
||||
}
|
||||
|
||||
return;
|
||||
|
||||
invalid:
|
||||
_upb_DefBuilder_Errf(ctx, "Invalid default '%.*s' for field %s of type %d",
|
||||
(int)len, str, upb_FieldDef_FullName(f),
|
||||
(int)upb_FieldDef_Type(f));
|
||||
}
|
||||
|
||||
static void set_default_default(upb_DefBuilder* ctx, upb_FieldDef* f) {
|
||||
switch (upb_FieldDef_CType(f)) {
|
||||
case kUpb_CType_Int32:
|
||||
case kUpb_CType_Int64:
|
||||
f->defaultval.sint = 0;
|
||||
break;
|
||||
case kUpb_CType_UInt64:
|
||||
case kUpb_CType_UInt32:
|
||||
f->defaultval.uint = 0;
|
||||
break;
|
||||
case kUpb_CType_Double:
|
||||
case kUpb_CType_Float:
|
||||
f->defaultval.dbl = 0;
|
||||
break;
|
||||
case kUpb_CType_String:
|
||||
case kUpb_CType_Bytes:
|
||||
f->defaultval.str = newstr(ctx, NULL, 0);
|
||||
break;
|
||||
case kUpb_CType_Bool:
|
||||
f->defaultval.boolean = false;
|
||||
break;
|
||||
case kUpb_CType_Enum: {
|
||||
const upb_EnumValueDef* v = upb_EnumDef_Value(f->sub.enumdef, 0);
|
||||
f->defaultval.sint = upb_EnumValueDef_Number(v);
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_Message:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void _upb_FieldDef_Create(upb_DefBuilder* ctx, const char* prefix,
|
||||
const UPB_DESC(FieldDescriptorProto) *
|
||||
field_proto,
|
||||
upb_MessageDef* m, upb_FieldDef* f) {
|
||||
// Must happen before _upb_DefBuilder_Add()
|
||||
f->file = _upb_DefBuilder_File(ctx);
|
||||
|
||||
if (!UPB_DESC(FieldDescriptorProto_has_name)(field_proto)) {
|
||||
_upb_DefBuilder_Errf(ctx, "field has no name");
|
||||
}
|
||||
|
||||
const upb_StringView name = UPB_DESC(FieldDescriptorProto_name)(field_proto);
|
||||
|
||||
f->full_name = _upb_DefBuilder_MakeFullName(ctx, prefix, name);
|
||||
f->label_ = (int)UPB_DESC(FieldDescriptorProto_label)(field_proto);
|
||||
f->number_ = UPB_DESC(FieldDescriptorProto_number)(field_proto);
|
||||
f->is_proto3_optional =
|
||||
UPB_DESC(FieldDescriptorProto_proto3_optional)(field_proto);
|
||||
f->msgdef = m;
|
||||
f->scope.oneof = NULL;
|
||||
|
||||
f->has_json_name = UPB_DESC(FieldDescriptorProto_has_json_name)(field_proto);
|
||||
if (f->has_json_name) {
|
||||
const upb_StringView sv =
|
||||
UPB_DESC(FieldDescriptorProto_json_name)(field_proto);
|
||||
f->json_name = upb_strdup2(sv.data, sv.size, ctx->arena);
|
||||
} else {
|
||||
f->json_name = make_json_name(name.data, name.size, ctx->arena);
|
||||
}
|
||||
if (!f->json_name) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
const bool has_type = UPB_DESC(FieldDescriptorProto_has_type)(field_proto);
|
||||
const bool has_type_name =
|
||||
UPB_DESC(FieldDescriptorProto_has_type_name)(field_proto);
|
||||
|
||||
f->type_ = (int)UPB_DESC(FieldDescriptorProto_type)(field_proto);
|
||||
|
||||
if (has_type) {
|
||||
switch (f->type_) {
|
||||
case kUpb_FieldType_Message:
|
||||
case kUpb_FieldType_Group:
|
||||
case kUpb_FieldType_Enum:
|
||||
if (!has_type_name) {
|
||||
_upb_DefBuilder_Errf(ctx, "field of type %d requires type name (%s)",
|
||||
(int)f->type_, f->full_name);
|
||||
}
|
||||
break;
|
||||
default:
|
||||
if (has_type_name) {
|
||||
_upb_DefBuilder_Errf(
|
||||
ctx, "invalid type for field with type_name set (%s, %d)",
|
||||
f->full_name, (int)f->type_);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!has_type && has_type_name) {
|
||||
f->type_ =
|
||||
UPB_FIELD_TYPE_UNSPECIFIED; // We'll assign this in resolve_subdef()
|
||||
} else {
|
||||
if (f->type_ < kUpb_FieldType_Double || f->type_ > kUpb_FieldType_SInt64) {
|
||||
_upb_DefBuilder_Errf(ctx, "invalid type for field %s (%d)", f->full_name,
|
||||
f->type_);
|
||||
}
|
||||
}
|
||||
|
||||
if (f->label_ < kUpb_Label_Optional || f->label_ > kUpb_Label_Repeated) {
|
||||
_upb_DefBuilder_Errf(ctx, "invalid label for field %s (%d)", f->full_name,
|
||||
f->label_);
|
||||
}
|
||||
|
||||
/* We can't resolve the subdef or (in the case of extensions) the containing
|
||||
* message yet, because it may not have been defined yet. We stash a pointer
|
||||
* to the field_proto until later when we can properly resolve it. */
|
||||
f->sub.unresolved = field_proto;
|
||||
|
||||
if (f->label_ == kUpb_Label_Required &&
|
||||
upb_FileDef_Syntax(f->file) == kUpb_Syntax_Proto3) {
|
||||
_upb_DefBuilder_Errf(ctx, "proto3 fields cannot be required (%s)",
|
||||
f->full_name);
|
||||
}
|
||||
|
||||
if (UPB_DESC(FieldDescriptorProto_has_oneof_index)(field_proto)) {
|
||||
int oneof_index = UPB_DESC(FieldDescriptorProto_oneof_index)(field_proto);
|
||||
|
||||
if (upb_FieldDef_Label(f) != kUpb_Label_Optional) {
|
||||
_upb_DefBuilder_Errf(ctx, "fields in oneof must have OPTIONAL label (%s)",
|
||||
f->full_name);
|
||||
}
|
||||
|
||||
if (!m) {
|
||||
_upb_DefBuilder_Errf(ctx, "oneof field (%s) has no containing msg",
|
||||
f->full_name);
|
||||
}
|
||||
|
||||
if (oneof_index >= upb_MessageDef_OneofCount(m)) {
|
||||
_upb_DefBuilder_Errf(ctx, "oneof_index out of range (%s)", f->full_name);
|
||||
}
|
||||
|
||||
upb_OneofDef* oneof = (upb_OneofDef*)upb_MessageDef_Oneof(m, oneof_index);
|
||||
f->scope.oneof = oneof;
|
||||
|
||||
_upb_OneofDef_Insert(ctx, oneof, f, name.data, name.size);
|
||||
}
|
||||
|
||||
UPB_DEF_SET_OPTIONS(f->opts, FieldDescriptorProto, FieldOptions, field_proto);
|
||||
|
||||
if (UPB_DESC(FieldOptions_has_packed)(f->opts)) {
|
||||
f->is_packed = UPB_DESC(FieldOptions_packed)(f->opts);
|
||||
} else {
|
||||
f->is_packed = upb_FileDef_Syntax(f->file) == kUpb_Syntax_Proto3;
|
||||
}
|
||||
|
||||
f->has_presence =
|
||||
(!upb_FieldDef_IsRepeated(f)) &&
|
||||
(f->type_ == kUpb_FieldType_Message || f->type_ == kUpb_FieldType_Group ||
|
||||
upb_FieldDef_ContainingOneof(f) ||
|
||||
(upb_FileDef_Syntax(f->file) == kUpb_Syntax_Proto2));
|
||||
}
|
||||
|
||||
static void _upb_FieldDef_CreateExt(upb_DefBuilder* ctx, const char* prefix,
|
||||
const UPB_DESC(FieldDescriptorProto) *
|
||||
field_proto,
|
||||
upb_MessageDef* m, upb_FieldDef* f) {
|
||||
f->is_extension = true;
|
||||
_upb_FieldDef_Create(ctx, prefix, field_proto, m, f);
|
||||
|
||||
if (UPB_DESC(FieldDescriptorProto_has_oneof_index)(field_proto)) {
|
||||
_upb_DefBuilder_Errf(ctx, "oneof_index provided for extension field (%s)",
|
||||
f->full_name);
|
||||
}
|
||||
|
||||
f->scope.extension_scope = m;
|
||||
_upb_DefBuilder_Add(ctx, f->full_name, _upb_DefType_Pack(f, UPB_DEFTYPE_EXT));
|
||||
f->layout_index = ctx->ext_count++;
|
||||
|
||||
if (ctx->layout) {
|
||||
UPB_ASSERT(_upb_FieldDef_ExtensionMiniTable(f)->field.number == f->number_);
|
||||
}
|
||||
}
|
||||
|
||||
static void _upb_FieldDef_CreateNotExt(upb_DefBuilder* ctx, const char* prefix,
|
||||
const UPB_DESC(FieldDescriptorProto) *
|
||||
field_proto,
|
||||
upb_MessageDef* m, upb_FieldDef* f) {
|
||||
f->is_extension = false;
|
||||
_upb_FieldDef_Create(ctx, prefix, field_proto, m, f);
|
||||
|
||||
if (!UPB_DESC(FieldDescriptorProto_has_oneof_index)(field_proto)) {
|
||||
if (f->is_proto3_optional) {
|
||||
_upb_DefBuilder_Errf(
|
||||
ctx,
|
||||
"non-extension field (%s) with proto3_optional was not in a oneof",
|
||||
f->full_name);
|
||||
}
|
||||
}
|
||||
|
||||
_upb_MessageDef_InsertField(ctx, m, f);
|
||||
}
|
||||
|
||||
upb_FieldDef* _upb_Extensions_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(FieldDescriptorProto) * const* protos, const char* prefix,
|
||||
upb_MessageDef* m) {
|
||||
_upb_DefType_CheckPadding(sizeof(upb_FieldDef));
|
||||
upb_FieldDef* defs =
|
||||
(upb_FieldDef*)_upb_DefBuilder_Alloc(ctx, sizeof(upb_FieldDef) * n);
|
||||
|
||||
for (int i = 0; i < n; i++) {
|
||||
upb_FieldDef* f = &defs[i];
|
||||
|
||||
_upb_FieldDef_CreateExt(ctx, prefix, protos[i], m, f);
|
||||
f->index_ = i;
|
||||
}
|
||||
|
||||
return defs;
|
||||
}
|
||||
|
||||
upb_FieldDef* _upb_FieldDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(FieldDescriptorProto) * const* protos, const char* prefix,
|
||||
upb_MessageDef* m, bool* is_sorted) {
|
||||
_upb_DefType_CheckPadding(sizeof(upb_FieldDef));
|
||||
upb_FieldDef* defs =
|
||||
(upb_FieldDef*)_upb_DefBuilder_Alloc(ctx, sizeof(upb_FieldDef) * n);
|
||||
|
||||
uint32_t previous = 0;
|
||||
for (int i = 0; i < n; i++) {
|
||||
upb_FieldDef* f = &defs[i];
|
||||
|
||||
_upb_FieldDef_CreateNotExt(ctx, prefix, protos[i], m, f);
|
||||
f->index_ = i;
|
||||
if (!ctx->layout) {
|
||||
// Speculate that the def fields are sorted. We will always sort the
|
||||
// MiniTable fields, so if defs are sorted then indices will match.
|
||||
//
|
||||
// If this is incorrect, we will overwrite later.
|
||||
f->layout_index = i;
|
||||
}
|
||||
|
||||
const uint32_t current = f->number_;
|
||||
if (previous > current) *is_sorted = false;
|
||||
previous = current;
|
||||
}
|
||||
|
||||
return defs;
|
||||
}
|
||||
|
||||
static void resolve_subdef(upb_DefBuilder* ctx, const char* prefix,
|
||||
upb_FieldDef* f) {
|
||||
const UPB_DESC(FieldDescriptorProto)* field_proto = f->sub.unresolved;
|
||||
upb_StringView name = UPB_DESC(FieldDescriptorProto_type_name)(field_proto);
|
||||
bool has_name = UPB_DESC(FieldDescriptorProto_has_type_name)(field_proto);
|
||||
switch ((int)f->type_) {
|
||||
case UPB_FIELD_TYPE_UNSPECIFIED: {
|
||||
// Type was not specified and must be inferred.
|
||||
UPB_ASSERT(has_name);
|
||||
upb_deftype_t type;
|
||||
const void* def =
|
||||
_upb_DefBuilder_ResolveAny(ctx, f->full_name, prefix, name, &type);
|
||||
switch (type) {
|
||||
case UPB_DEFTYPE_ENUM:
|
||||
f->sub.enumdef = def;
|
||||
f->type_ = kUpb_FieldType_Enum;
|
||||
break;
|
||||
case UPB_DEFTYPE_MSG:
|
||||
f->sub.msgdef = def;
|
||||
f->type_ = kUpb_FieldType_Message; // It appears there is no way of
|
||||
// this being a group.
|
||||
f->has_presence = !upb_FieldDef_IsRepeated(f);
|
||||
break;
|
||||
default:
|
||||
_upb_DefBuilder_Errf(ctx, "Couldn't resolve type name for field %s",
|
||||
f->full_name);
|
||||
}
|
||||
break;
|
||||
}
|
||||
case kUpb_FieldType_Message:
|
||||
case kUpb_FieldType_Group:
|
||||
UPB_ASSERT(has_name);
|
||||
f->sub.msgdef = _upb_DefBuilder_Resolve(ctx, f->full_name, prefix, name,
|
||||
UPB_DEFTYPE_MSG);
|
||||
break;
|
||||
case kUpb_FieldType_Enum:
|
||||
UPB_ASSERT(has_name);
|
||||
f->sub.enumdef = _upb_DefBuilder_Resolve(ctx, f->full_name, prefix, name,
|
||||
UPB_DEFTYPE_ENUM);
|
||||
break;
|
||||
default:
|
||||
// No resolution necessary.
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static int _upb_FieldDef_Compare(const void* p1, const void* p2) {
|
||||
const uint32_t v1 = (*(upb_FieldDef**)p1)->number_;
|
||||
const uint32_t v2 = (*(upb_FieldDef**)p2)->number_;
|
||||
return (v1 < v2) ? -1 : (v1 > v2);
|
||||
}
|
||||
|
||||
// _upb_FieldDefs_Sorted() is mostly a pure function of its inputs, but has one
|
||||
// critical side effect that we depend on: it sets layout_index appropriately
|
||||
// for non-sorted lists of fields.
|
||||
const upb_FieldDef** _upb_FieldDefs_Sorted(const upb_FieldDef* f, int n,
|
||||
upb_Arena* a) {
|
||||
// TODO: Replace this arena alloc with a persistent scratch buffer.
|
||||
upb_FieldDef** out = (upb_FieldDef**)upb_Arena_Malloc(a, n * sizeof(void*));
|
||||
if (!out) return NULL;
|
||||
|
||||
for (int i = 0; i < n; i++) {
|
||||
out[i] = (upb_FieldDef*)&f[i];
|
||||
}
|
||||
qsort(out, n, sizeof(void*), _upb_FieldDef_Compare);
|
||||
|
||||
for (int i = 0; i < n; i++) {
|
||||
out[i]->layout_index = i;
|
||||
}
|
||||
return (const upb_FieldDef**)out;
|
||||
}
|
||||
|
||||
bool upb_FieldDef_MiniDescriptorEncode(const upb_FieldDef* f, upb_Arena* a,
|
||||
upb_StringView* out) {
|
||||
UPB_ASSERT(f->is_extension);
|
||||
|
||||
upb_DescState s;
|
||||
_upb_DescState_Init(&s);
|
||||
|
||||
const int number = upb_FieldDef_Number(f);
|
||||
const uint64_t modifiers = _upb_FieldDef_Modifiers(f);
|
||||
|
||||
if (!_upb_DescState_Grow(&s, a)) return false;
|
||||
s.ptr = upb_MtDataEncoder_EncodeExtension(&s.e, s.ptr, f->type_, number,
|
||||
modifiers);
|
||||
*s.ptr = '\0';
|
||||
|
||||
out->data = s.buf;
|
||||
out->size = s.ptr - s.buf;
|
||||
return true;
|
||||
}
|
||||
|
||||
static void resolve_extension(upb_DefBuilder* ctx, const char* prefix,
|
||||
upb_FieldDef* f,
|
||||
const UPB_DESC(FieldDescriptorProto) *
|
||||
field_proto) {
|
||||
if (!UPB_DESC(FieldDescriptorProto_has_extendee)(field_proto)) {
|
||||
_upb_DefBuilder_Errf(ctx, "extension for field '%s' had no extendee",
|
||||
f->full_name);
|
||||
}
|
||||
|
||||
upb_StringView name = UPB_DESC(FieldDescriptorProto_extendee)(field_proto);
|
||||
const upb_MessageDef* m =
|
||||
_upb_DefBuilder_Resolve(ctx, f->full_name, prefix, name, UPB_DEFTYPE_MSG);
|
||||
f->msgdef = m;
|
||||
|
||||
if (!_upb_MessageDef_IsValidExtensionNumber(m, f->number_)) {
|
||||
_upb_DefBuilder_Errf(
|
||||
ctx,
|
||||
"field number %u in extension %s has no extension range in message %s",
|
||||
(unsigned)f->number_, f->full_name, upb_MessageDef_FullName(m));
|
||||
}
|
||||
}
|
||||
|
||||
void _upb_FieldDef_BuildMiniTableExtension(upb_DefBuilder* ctx,
|
||||
const upb_FieldDef* f) {
|
||||
const upb_MiniTableExtension* ext = _upb_FieldDef_ExtensionMiniTable(f);
|
||||
|
||||
if (ctx->layout) {
|
||||
UPB_ASSERT(upb_FieldDef_Number(f) == ext->field.number);
|
||||
} else {
|
||||
upb_StringView desc;
|
||||
if (!upb_FieldDef_MiniDescriptorEncode(f, ctx->tmp_arena, &desc)) {
|
||||
_upb_DefBuilder_OomErr(ctx);
|
||||
}
|
||||
|
||||
upb_MiniTableExtension* mut_ext = (upb_MiniTableExtension*)ext;
|
||||
upb_MiniTableSub sub = {NULL};
|
||||
if (upb_FieldDef_IsSubMessage(f)) {
|
||||
sub.submsg = upb_MessageDef_MiniTable(f->sub.msgdef);
|
||||
} else if (_upb_FieldDef_IsClosedEnum(f)) {
|
||||
sub.subenum = _upb_EnumDef_MiniTable(f->sub.enumdef);
|
||||
}
|
||||
bool ok2 = upb_MiniTableExtension_Init(desc.data, desc.size, mut_ext,
|
||||
upb_MessageDef_MiniTable(f->msgdef),
|
||||
sub, ctx->status);
|
||||
if (!ok2) _upb_DefBuilder_Errf(ctx, "Could not build extension mini table");
|
||||
}
|
||||
|
||||
bool ok = _upb_DefPool_InsertExt(ctx->symtab, ext, f);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
}
|
||||
|
||||
static void resolve_default(upb_DefBuilder* ctx, upb_FieldDef* f,
|
||||
const UPB_DESC(FieldDescriptorProto) *
|
||||
field_proto) {
|
||||
// Have to delay resolving of the default value until now because of the enum
|
||||
// case, since enum defaults are specified with a label.
|
||||
if (UPB_DESC(FieldDescriptorProto_has_default_value)(field_proto)) {
|
||||
upb_StringView defaultval =
|
||||
UPB_DESC(FieldDescriptorProto_default_value)(field_proto);
|
||||
|
||||
if (upb_FileDef_Syntax(f->file) == kUpb_Syntax_Proto3) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"proto3 fields cannot have explicit defaults (%s)",
|
||||
f->full_name);
|
||||
}
|
||||
|
||||
if (upb_FieldDef_IsSubMessage(f)) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"message fields cannot have explicit defaults (%s)",
|
||||
f->full_name);
|
||||
}
|
||||
|
||||
parse_default(ctx, defaultval.data, defaultval.size, f);
|
||||
f->has_default = true;
|
||||
} else {
|
||||
set_default_default(ctx, f);
|
||||
f->has_default = false;
|
||||
}
|
||||
}
|
||||
|
||||
void _upb_FieldDef_Resolve(upb_DefBuilder* ctx, const char* prefix,
|
||||
upb_FieldDef* f) {
|
||||
// We have to stash this away since resolve_subdef() may overwrite it.
|
||||
const UPB_DESC(FieldDescriptorProto)* field_proto = f->sub.unresolved;
|
||||
|
||||
resolve_subdef(ctx, prefix, f);
|
||||
resolve_default(ctx, f, field_proto);
|
||||
|
||||
if (f->is_extension) {
|
||||
resolve_extension(ctx, prefix, f, field_proto);
|
||||
}
|
||||
}
|
||||
74
Pods/gRPC-Core/third_party/upb/upb/reflection/field_def.h
generated
vendored
Normal file
74
Pods/gRPC-Core/third_party/upb/upb/reflection/field_def.h
generated
vendored
Normal file
@@ -0,0 +1,74 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_FIELD_DEF_H_
|
||||
#define UPB_REFLECTION_FIELD_DEF_H_
|
||||
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// Maximum field number allowed for FieldDefs.
|
||||
// This is an inherent limit of the protobuf wire format.
|
||||
#define kUpb_MaxFieldNumber ((1 << 29) - 1)
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
const upb_OneofDef* upb_FieldDef_ContainingOneof(const upb_FieldDef* f);
|
||||
UPB_API const upb_MessageDef* upb_FieldDef_ContainingType(
|
||||
const upb_FieldDef* f);
|
||||
UPB_API upb_CType upb_FieldDef_CType(const upb_FieldDef* f);
|
||||
UPB_API upb_MessageValue upb_FieldDef_Default(const upb_FieldDef* f);
|
||||
UPB_API const upb_EnumDef* upb_FieldDef_EnumSubDef(const upb_FieldDef* f);
|
||||
const upb_MessageDef* upb_FieldDef_ExtensionScope(const upb_FieldDef* f);
|
||||
UPB_API const upb_FileDef* upb_FieldDef_File(const upb_FieldDef* f);
|
||||
const char* upb_FieldDef_FullName(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_HasDefault(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_HasJsonName(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_HasOptions(const upb_FieldDef* f);
|
||||
UPB_API bool upb_FieldDef_HasPresence(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_HasSubDef(const upb_FieldDef* f);
|
||||
uint32_t upb_FieldDef_Index(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_IsExtension(const upb_FieldDef* f);
|
||||
UPB_API bool upb_FieldDef_IsMap(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_IsOptional(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_IsPacked(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_IsPrimitive(const upb_FieldDef* f);
|
||||
UPB_API bool upb_FieldDef_IsRepeated(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_IsRequired(const upb_FieldDef* f);
|
||||
bool upb_FieldDef_IsString(const upb_FieldDef* f);
|
||||
UPB_API bool upb_FieldDef_IsSubMessage(const upb_FieldDef* f);
|
||||
UPB_API const char* upb_FieldDef_JsonName(const upb_FieldDef* f);
|
||||
UPB_API upb_Label upb_FieldDef_Label(const upb_FieldDef* f);
|
||||
UPB_API const upb_MessageDef* upb_FieldDef_MessageSubDef(const upb_FieldDef* f);
|
||||
bool _upb_FieldDef_ValidateUtf8(const upb_FieldDef* f);
|
||||
|
||||
// Creates a mini descriptor string for a field, returns true on success.
|
||||
bool upb_FieldDef_MiniDescriptorEncode(const upb_FieldDef* f, upb_Arena* a,
|
||||
upb_StringView* out);
|
||||
|
||||
const upb_MiniTableField* upb_FieldDef_MiniTable(const upb_FieldDef* f);
|
||||
UPB_API const char* upb_FieldDef_Name(const upb_FieldDef* f);
|
||||
UPB_API uint32_t upb_FieldDef_Number(const upb_FieldDef* f);
|
||||
const UPB_DESC(FieldOptions) * upb_FieldDef_Options(const upb_FieldDef* f);
|
||||
UPB_API const upb_OneofDef* upb_FieldDef_RealContainingOneof(
|
||||
const upb_FieldDef* f);
|
||||
UPB_API upb_FieldType upb_FieldDef_Type(const upb_FieldDef* f);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_FIELD_DEF_H_ */
|
||||
345
Pods/gRPC-Core/third_party/upb/upb/reflection/file_def.c
generated
vendored
Normal file
345
Pods/gRPC-Core/third_party/upb/upb/reflection/file_def.c
generated
vendored
Normal file
@@ -0,0 +1,345 @@
|
||||
// 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/reflection/internal/file_def.h"
|
||||
|
||||
#include "upb/reflection/def_pool.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/enum_def.h"
|
||||
#include "upb/reflection/internal/field_def.h"
|
||||
#include "upb/reflection/internal/message_def.h"
|
||||
#include "upb/reflection/internal/service_def.h"
|
||||
#include "upb/reflection/internal/strdup2.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_FileDef {
|
||||
const UPB_DESC(FileOptions) * opts;
|
||||
const char* name;
|
||||
const char* package;
|
||||
UPB_DESC(Edition) edition;
|
||||
|
||||
const upb_FileDef** deps;
|
||||
const int32_t* public_deps;
|
||||
const int32_t* weak_deps;
|
||||
const upb_MessageDef* top_lvl_msgs;
|
||||
const upb_EnumDef* top_lvl_enums;
|
||||
const upb_FieldDef* top_lvl_exts;
|
||||
const upb_ServiceDef* services;
|
||||
const upb_MiniTableExtension** ext_layouts;
|
||||
const upb_DefPool* symtab;
|
||||
|
||||
int dep_count;
|
||||
int public_dep_count;
|
||||
int weak_dep_count;
|
||||
int top_lvl_msg_count;
|
||||
int top_lvl_enum_count;
|
||||
int top_lvl_ext_count;
|
||||
int service_count;
|
||||
int ext_count; // All exts in the file.
|
||||
upb_Syntax syntax;
|
||||
};
|
||||
|
||||
const UPB_DESC(FileOptions) * upb_FileDef_Options(const upb_FileDef* f) {
|
||||
return f->opts;
|
||||
}
|
||||
|
||||
bool upb_FileDef_HasOptions(const upb_FileDef* f) {
|
||||
return f->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
const char* upb_FileDef_Name(const upb_FileDef* f) { return f->name; }
|
||||
|
||||
const char* upb_FileDef_Package(const upb_FileDef* f) {
|
||||
return f->package ? f->package : "";
|
||||
}
|
||||
|
||||
UPB_DESC(Edition) upb_FileDef_Edition(const upb_FileDef* f) {
|
||||
return f->edition;
|
||||
}
|
||||
|
||||
const char* _upb_FileDef_RawPackage(const upb_FileDef* f) { return f->package; }
|
||||
|
||||
upb_Syntax upb_FileDef_Syntax(const upb_FileDef* f) { return f->syntax; }
|
||||
|
||||
int upb_FileDef_TopLevelMessageCount(const upb_FileDef* f) {
|
||||
return f->top_lvl_msg_count;
|
||||
}
|
||||
|
||||
int upb_FileDef_DependencyCount(const upb_FileDef* f) { return f->dep_count; }
|
||||
|
||||
int upb_FileDef_PublicDependencyCount(const upb_FileDef* f) {
|
||||
return f->public_dep_count;
|
||||
}
|
||||
|
||||
int upb_FileDef_WeakDependencyCount(const upb_FileDef* f) {
|
||||
return f->weak_dep_count;
|
||||
}
|
||||
|
||||
const int32_t* _upb_FileDef_PublicDependencyIndexes(const upb_FileDef* f) {
|
||||
return f->public_deps;
|
||||
}
|
||||
|
||||
const int32_t* _upb_FileDef_WeakDependencyIndexes(const upb_FileDef* f) {
|
||||
return f->weak_deps;
|
||||
}
|
||||
|
||||
int upb_FileDef_TopLevelEnumCount(const upb_FileDef* f) {
|
||||
return f->top_lvl_enum_count;
|
||||
}
|
||||
|
||||
int upb_FileDef_TopLevelExtensionCount(const upb_FileDef* f) {
|
||||
return f->top_lvl_ext_count;
|
||||
}
|
||||
|
||||
int upb_FileDef_ServiceCount(const upb_FileDef* f) { return f->service_count; }
|
||||
|
||||
const upb_FileDef* upb_FileDef_Dependency(const upb_FileDef* f, int i) {
|
||||
UPB_ASSERT(0 <= i && i < f->dep_count);
|
||||
return f->deps[i];
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_FileDef_PublicDependency(const upb_FileDef* f, int i) {
|
||||
UPB_ASSERT(0 <= i && i < f->public_dep_count);
|
||||
return f->deps[f->public_deps[i]];
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_FileDef_WeakDependency(const upb_FileDef* f, int i) {
|
||||
UPB_ASSERT(0 <= i && i < f->public_dep_count);
|
||||
return f->deps[f->weak_deps[i]];
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_FileDef_TopLevelMessage(const upb_FileDef* f, int i) {
|
||||
UPB_ASSERT(0 <= i && i < f->top_lvl_msg_count);
|
||||
return _upb_MessageDef_At(f->top_lvl_msgs, i);
|
||||
}
|
||||
|
||||
const upb_EnumDef* upb_FileDef_TopLevelEnum(const upb_FileDef* f, int i) {
|
||||
UPB_ASSERT(0 <= i && i < f->top_lvl_enum_count);
|
||||
return _upb_EnumDef_At(f->top_lvl_enums, i);
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_FileDef_TopLevelExtension(const upb_FileDef* f, int i) {
|
||||
UPB_ASSERT(0 <= i && i < f->top_lvl_ext_count);
|
||||
return _upb_FieldDef_At(f->top_lvl_exts, i);
|
||||
}
|
||||
|
||||
const upb_ServiceDef* upb_FileDef_Service(const upb_FileDef* f, int i) {
|
||||
UPB_ASSERT(0 <= i && i < f->service_count);
|
||||
return _upb_ServiceDef_At(f->services, i);
|
||||
}
|
||||
|
||||
const upb_DefPool* upb_FileDef_Pool(const upb_FileDef* f) { return f->symtab; }
|
||||
|
||||
const upb_MiniTableExtension* _upb_FileDef_ExtensionMiniTable(
|
||||
const upb_FileDef* f, int i) {
|
||||
return f->ext_layouts[i];
|
||||
}
|
||||
|
||||
static char* strviewdup(upb_DefBuilder* ctx, upb_StringView view) {
|
||||
char* ret = upb_strdup2(view.data, view.size, _upb_DefBuilder_Arena(ctx));
|
||||
if (!ret) _upb_DefBuilder_OomErr(ctx);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static bool streql_view(upb_StringView view, const char* b) {
|
||||
return view.size == strlen(b) && memcmp(view.data, b, view.size) == 0;
|
||||
}
|
||||
|
||||
static int count_exts_in_msg(const UPB_DESC(DescriptorProto) * msg_proto) {
|
||||
size_t n;
|
||||
UPB_DESC(DescriptorProto_extension)(msg_proto, &n);
|
||||
int ext_count = n;
|
||||
|
||||
const UPB_DESC(DescriptorProto)* const* nested_msgs =
|
||||
UPB_DESC(DescriptorProto_nested_type)(msg_proto, &n);
|
||||
for (size_t i = 0; i < n; i++) {
|
||||
ext_count += count_exts_in_msg(nested_msgs[i]);
|
||||
}
|
||||
|
||||
return ext_count;
|
||||
}
|
||||
|
||||
// Allocate and initialize one file def, and add it to the context object.
|
||||
void _upb_FileDef_Create(upb_DefBuilder* ctx,
|
||||
const UPB_DESC(FileDescriptorProto) * file_proto) {
|
||||
upb_FileDef* file = _upb_DefBuilder_Alloc(ctx, sizeof(upb_FileDef));
|
||||
ctx->file = file;
|
||||
|
||||
const UPB_DESC(DescriptorProto)* const* msgs;
|
||||
const UPB_DESC(EnumDescriptorProto)* const* enums;
|
||||
const UPB_DESC(FieldDescriptorProto)* const* exts;
|
||||
const UPB_DESC(ServiceDescriptorProto)* const* services;
|
||||
const upb_StringView* strs;
|
||||
const int32_t* public_deps;
|
||||
const int32_t* weak_deps;
|
||||
size_t n;
|
||||
|
||||
file->symtab = ctx->symtab;
|
||||
|
||||
// Count all extensions in the file, to build a flat array of layouts.
|
||||
UPB_DESC(FileDescriptorProto_extension)(file_proto, &n);
|
||||
int ext_count = n;
|
||||
msgs = UPB_DESC(FileDescriptorProto_message_type)(file_proto, &n);
|
||||
for (size_t i = 0; i < n; i++) {
|
||||
ext_count += count_exts_in_msg(msgs[i]);
|
||||
}
|
||||
file->ext_count = ext_count;
|
||||
|
||||
if (ctx->layout) {
|
||||
// We are using the ext layouts that were passed in.
|
||||
file->ext_layouts = ctx->layout->exts;
|
||||
if (ctx->layout->ext_count != file->ext_count) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"Extension count did not match layout (%d vs %d)",
|
||||
ctx->layout->ext_count, file->ext_count);
|
||||
}
|
||||
} else {
|
||||
// We are building ext layouts from scratch.
|
||||
file->ext_layouts = _upb_DefBuilder_Alloc(
|
||||
ctx, sizeof(*file->ext_layouts) * file->ext_count);
|
||||
upb_MiniTableExtension* ext =
|
||||
_upb_DefBuilder_Alloc(ctx, sizeof(*ext) * file->ext_count);
|
||||
for (int i = 0; i < file->ext_count; i++) {
|
||||
file->ext_layouts[i] = &ext[i];
|
||||
}
|
||||
}
|
||||
|
||||
upb_StringView name = UPB_DESC(FileDescriptorProto_name)(file_proto);
|
||||
file->name = strviewdup(ctx, name);
|
||||
if (strlen(file->name) != name.size) {
|
||||
_upb_DefBuilder_Errf(ctx, "File name contained embedded NULL");
|
||||
}
|
||||
|
||||
upb_StringView package = UPB_DESC(FileDescriptorProto_package)(file_proto);
|
||||
|
||||
if (package.size) {
|
||||
_upb_DefBuilder_CheckIdentFull(ctx, package);
|
||||
file->package = strviewdup(ctx, package);
|
||||
} else {
|
||||
file->package = NULL;
|
||||
}
|
||||
|
||||
// TODO: How should we validate this?
|
||||
file->edition = UPB_DESC(FileDescriptorProto_edition)(file_proto);
|
||||
|
||||
if (UPB_DESC(FileDescriptorProto_has_syntax)(file_proto)) {
|
||||
upb_StringView syntax = UPB_DESC(FileDescriptorProto_syntax)(file_proto);
|
||||
|
||||
if (streql_view(syntax, "proto2")) {
|
||||
file->syntax = kUpb_Syntax_Proto2;
|
||||
} else if (streql_view(syntax, "proto3")) {
|
||||
file->syntax = kUpb_Syntax_Proto3;
|
||||
} else if (streql_view(syntax, "editions")) {
|
||||
file->syntax = kUpb_Syntax_Editions;
|
||||
} else {
|
||||
_upb_DefBuilder_Errf(ctx, "Invalid syntax '" UPB_STRINGVIEW_FORMAT "'",
|
||||
UPB_STRINGVIEW_ARGS(syntax));
|
||||
}
|
||||
} else {
|
||||
file->syntax = kUpb_Syntax_Proto2;
|
||||
}
|
||||
|
||||
// Read options.
|
||||
UPB_DEF_SET_OPTIONS(file->opts, FileDescriptorProto, FileOptions, file_proto);
|
||||
|
||||
// Verify dependencies.
|
||||
strs = UPB_DESC(FileDescriptorProto_dependency)(file_proto, &n);
|
||||
file->dep_count = n;
|
||||
file->deps = _upb_DefBuilder_Alloc(ctx, sizeof(*file->deps) * n);
|
||||
|
||||
for (size_t i = 0; i < n; i++) {
|
||||
upb_StringView str = strs[i];
|
||||
file->deps[i] =
|
||||
upb_DefPool_FindFileByNameWithSize(ctx->symtab, str.data, str.size);
|
||||
if (!file->deps[i]) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"Depends on file '" UPB_STRINGVIEW_FORMAT
|
||||
"', but it has not been loaded",
|
||||
UPB_STRINGVIEW_ARGS(str));
|
||||
}
|
||||
}
|
||||
|
||||
public_deps = UPB_DESC(FileDescriptorProto_public_dependency)(file_proto, &n);
|
||||
file->public_dep_count = n;
|
||||
file->public_deps =
|
||||
_upb_DefBuilder_Alloc(ctx, sizeof(*file->public_deps) * n);
|
||||
int32_t* mutable_public_deps = (int32_t*)file->public_deps;
|
||||
for (size_t i = 0; i < n; i++) {
|
||||
if (public_deps[i] >= file->dep_count) {
|
||||
_upb_DefBuilder_Errf(ctx, "public_dep %d is out of range",
|
||||
(int)public_deps[i]);
|
||||
}
|
||||
mutable_public_deps[i] = public_deps[i];
|
||||
}
|
||||
|
||||
weak_deps = UPB_DESC(FileDescriptorProto_weak_dependency)(file_proto, &n);
|
||||
file->weak_dep_count = n;
|
||||
file->weak_deps = _upb_DefBuilder_Alloc(ctx, sizeof(*file->weak_deps) * n);
|
||||
int32_t* mutable_weak_deps = (int32_t*)file->weak_deps;
|
||||
for (size_t i = 0; i < n; i++) {
|
||||
if (weak_deps[i] >= file->dep_count) {
|
||||
_upb_DefBuilder_Errf(ctx, "weak_dep %d is out of range",
|
||||
(int)weak_deps[i]);
|
||||
}
|
||||
mutable_weak_deps[i] = weak_deps[i];
|
||||
}
|
||||
|
||||
// Create enums.
|
||||
enums = UPB_DESC(FileDescriptorProto_enum_type)(file_proto, &n);
|
||||
file->top_lvl_enum_count = n;
|
||||
file->top_lvl_enums = _upb_EnumDefs_New(ctx, n, enums, NULL);
|
||||
|
||||
// Create extensions.
|
||||
exts = UPB_DESC(FileDescriptorProto_extension)(file_proto, &n);
|
||||
file->top_lvl_ext_count = n;
|
||||
file->top_lvl_exts = _upb_Extensions_New(ctx, n, exts, file->package, NULL);
|
||||
|
||||
// Create messages.
|
||||
msgs = UPB_DESC(FileDescriptorProto_message_type)(file_proto, &n);
|
||||
file->top_lvl_msg_count = n;
|
||||
file->top_lvl_msgs = _upb_MessageDefs_New(ctx, n, msgs, NULL);
|
||||
|
||||
// Create services.
|
||||
services = UPB_DESC(FileDescriptorProto_service)(file_proto, &n);
|
||||
file->service_count = n;
|
||||
file->services = _upb_ServiceDefs_New(ctx, n, services);
|
||||
|
||||
// Now that all names are in the table, build layouts and resolve refs.
|
||||
|
||||
for (int i = 0; i < file->top_lvl_msg_count; i++) {
|
||||
upb_MessageDef* m = (upb_MessageDef*)upb_FileDef_TopLevelMessage(file, i);
|
||||
_upb_MessageDef_Resolve(ctx, m);
|
||||
}
|
||||
|
||||
for (int i = 0; i < file->top_lvl_ext_count; i++) {
|
||||
upb_FieldDef* f = (upb_FieldDef*)upb_FileDef_TopLevelExtension(file, i);
|
||||
_upb_FieldDef_Resolve(ctx, file->package, f);
|
||||
}
|
||||
|
||||
for (int i = 0; i < file->top_lvl_msg_count; i++) {
|
||||
upb_MessageDef* m = (upb_MessageDef*)upb_FileDef_TopLevelMessage(file, i);
|
||||
_upb_MessageDef_CreateMiniTable(ctx, (upb_MessageDef*)m);
|
||||
}
|
||||
|
||||
for (int i = 0; i < file->top_lvl_ext_count; i++) {
|
||||
upb_FieldDef* f = (upb_FieldDef*)upb_FileDef_TopLevelExtension(file, i);
|
||||
_upb_FieldDef_BuildMiniTableExtension(ctx, f);
|
||||
}
|
||||
|
||||
for (int i = 0; i < file->top_lvl_msg_count; i++) {
|
||||
upb_MessageDef* m = (upb_MessageDef*)upb_FileDef_TopLevelMessage(file, i);
|
||||
_upb_MessageDef_LinkMiniTable(ctx, m);
|
||||
}
|
||||
|
||||
if (file->ext_count) {
|
||||
bool ok = upb_ExtensionRegistry_AddArray(
|
||||
_upb_DefPool_ExtReg(ctx->symtab), file->ext_layouts, file->ext_count);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
}
|
||||
}
|
||||
57
Pods/gRPC-Core/third_party/upb/upb/reflection/file_def.h
generated
vendored
Normal file
57
Pods/gRPC-Core/third_party/upb/upb/reflection/file_def.h
generated
vendored
Normal file
@@ -0,0 +1,57 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_FILE_DEF_H_
|
||||
#define UPB_REFLECTION_FILE_DEF_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
const upb_FileDef* upb_FileDef_Dependency(const upb_FileDef* f, int i);
|
||||
int upb_FileDef_DependencyCount(const upb_FileDef* f);
|
||||
bool upb_FileDef_HasOptions(const upb_FileDef* f);
|
||||
UPB_API const char* upb_FileDef_Name(const upb_FileDef* f);
|
||||
const UPB_DESC(FileOptions) * upb_FileDef_Options(const upb_FileDef* f);
|
||||
const char* upb_FileDef_Package(const upb_FileDef* f);
|
||||
UPB_DESC(Edition) upb_FileDef_Edition(const upb_FileDef* f);
|
||||
UPB_API const upb_DefPool* upb_FileDef_Pool(const upb_FileDef* f);
|
||||
|
||||
const upb_FileDef* upb_FileDef_PublicDependency(const upb_FileDef* f, int i);
|
||||
int upb_FileDef_PublicDependencyCount(const upb_FileDef* f);
|
||||
|
||||
const upb_ServiceDef* upb_FileDef_Service(const upb_FileDef* f, int i);
|
||||
int upb_FileDef_ServiceCount(const upb_FileDef* f);
|
||||
|
||||
UPB_API upb_Syntax upb_FileDef_Syntax(const upb_FileDef* f);
|
||||
|
||||
const upb_EnumDef* upb_FileDef_TopLevelEnum(const upb_FileDef* f, int i);
|
||||
int upb_FileDef_TopLevelEnumCount(const upb_FileDef* f);
|
||||
|
||||
const upb_FieldDef* upb_FileDef_TopLevelExtension(const upb_FileDef* f, int i);
|
||||
int upb_FileDef_TopLevelExtensionCount(const upb_FileDef* f);
|
||||
|
||||
const upb_MessageDef* upb_FileDef_TopLevelMessage(const upb_FileDef* f, int i);
|
||||
int upb_FileDef_TopLevelMessageCount(const upb_FileDef* f);
|
||||
|
||||
const upb_FileDef* upb_FileDef_WeakDependency(const upb_FileDef* f, int i);
|
||||
int upb_FileDef_WeakDependencyCount(const upb_FileDef* f);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_FILE_DEF_H_ */
|
||||
339
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/def_builder.c
generated
vendored
Normal file
339
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/def_builder.c
generated
vendored
Normal file
@@ -0,0 +1,339 @@
|
||||
// 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/reflection/internal/def_builder.h"
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/reflection/def_pool.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/field_def.h"
|
||||
#include "upb/reflection/internal/strdup2.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
/* The upb core does not generally have a concept of default instances. However
|
||||
* for descriptor options we make an exception since the max size is known and
|
||||
* modest (<200 bytes). All types can share a default instance since it is
|
||||
* initialized to zeroes.
|
||||
*
|
||||
* We have to allocate an extra pointer for upb's internal metadata. */
|
||||
static const char opt_default_buf[_UPB_MAXOPT_SIZE + sizeof(void*)] = {0};
|
||||
const char* kUpbDefOptDefault = &opt_default_buf[sizeof(void*)];
|
||||
|
||||
const char* _upb_DefBuilder_FullToShort(const char* fullname) {
|
||||
const char* p;
|
||||
|
||||
if (fullname == NULL) {
|
||||
return NULL;
|
||||
} else if ((p = strrchr(fullname, '.')) == NULL) {
|
||||
/* No '.' in the name, return the full string. */
|
||||
return fullname;
|
||||
} else {
|
||||
/* Return one past the last '.'. */
|
||||
return p + 1;
|
||||
}
|
||||
}
|
||||
|
||||
void _upb_DefBuilder_FailJmp(upb_DefBuilder* ctx) { UPB_LONGJMP(ctx->err, 1); }
|
||||
|
||||
void _upb_DefBuilder_Errf(upb_DefBuilder* ctx, const char* fmt, ...) {
|
||||
va_list argp;
|
||||
va_start(argp, fmt);
|
||||
upb_Status_VSetErrorFormat(ctx->status, fmt, argp);
|
||||
va_end(argp);
|
||||
_upb_DefBuilder_FailJmp(ctx);
|
||||
}
|
||||
|
||||
void _upb_DefBuilder_OomErr(upb_DefBuilder* ctx) {
|
||||
upb_Status_SetErrorMessage(ctx->status, "out of memory");
|
||||
_upb_DefBuilder_FailJmp(ctx);
|
||||
}
|
||||
|
||||
// Verify a relative identifier string. The loop is branchless for speed.
|
||||
static void _upb_DefBuilder_CheckIdentNotFull(upb_DefBuilder* ctx,
|
||||
upb_StringView name) {
|
||||
bool good = name.size > 0;
|
||||
|
||||
for (size_t i = 0; i < name.size; i++) {
|
||||
const char c = name.data[i];
|
||||
const char d = c | 0x20; // force lowercase
|
||||
const bool is_alpha = (('a' <= d) & (d <= 'z')) | (c == '_');
|
||||
const bool is_numer = ('0' <= c) & (c <= '9') & (i != 0);
|
||||
|
||||
good &= is_alpha | is_numer;
|
||||
}
|
||||
|
||||
if (!good) _upb_DefBuilder_CheckIdentSlow(ctx, name, false);
|
||||
}
|
||||
|
||||
const char* _upb_DefBuilder_MakeFullName(upb_DefBuilder* ctx,
|
||||
const char* prefix,
|
||||
upb_StringView name) {
|
||||
_upb_DefBuilder_CheckIdentNotFull(ctx, name);
|
||||
if (prefix) {
|
||||
// ret = prefix + '.' + name;
|
||||
size_t n = strlen(prefix);
|
||||
char* ret = _upb_DefBuilder_Alloc(ctx, n + name.size + 2);
|
||||
strcpy(ret, prefix);
|
||||
ret[n] = '.';
|
||||
memcpy(&ret[n + 1], name.data, name.size);
|
||||
ret[n + 1 + name.size] = '\0';
|
||||
return ret;
|
||||
} else {
|
||||
char* ret = upb_strdup2(name.data, name.size, ctx->arena);
|
||||
if (!ret) _upb_DefBuilder_OomErr(ctx);
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
|
||||
static bool remove_component(char* base, size_t* len) {
|
||||
if (*len == 0) return false;
|
||||
|
||||
for (size_t i = *len - 1; i > 0; i--) {
|
||||
if (base[i] == '.') {
|
||||
*len = i;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
*len = 0;
|
||||
return true;
|
||||
}
|
||||
|
||||
const void* _upb_DefBuilder_ResolveAny(upb_DefBuilder* ctx,
|
||||
const char* from_name_dbg,
|
||||
const char* base, upb_StringView sym,
|
||||
upb_deftype_t* type) {
|
||||
if (sym.size == 0) goto notfound;
|
||||
upb_value v;
|
||||
if (sym.data[0] == '.') {
|
||||
/* Symbols starting with '.' are absolute, so we do a single lookup.
|
||||
* Slice to omit the leading '.' */
|
||||
if (!_upb_DefPool_LookupSym(ctx->symtab, sym.data + 1, sym.size - 1, &v)) {
|
||||
goto notfound;
|
||||
}
|
||||
} else {
|
||||
/* Remove components from base until we find an entry or run out. */
|
||||
size_t baselen = base ? strlen(base) : 0;
|
||||
char* tmp = malloc(sym.size + baselen + 1);
|
||||
while (1) {
|
||||
char* p = tmp;
|
||||
if (baselen) {
|
||||
memcpy(p, base, baselen);
|
||||
p[baselen] = '.';
|
||||
p += baselen + 1;
|
||||
}
|
||||
memcpy(p, sym.data, sym.size);
|
||||
p += sym.size;
|
||||
if (_upb_DefPool_LookupSym(ctx->symtab, tmp, p - tmp, &v)) {
|
||||
break;
|
||||
}
|
||||
if (!remove_component(tmp, &baselen)) {
|
||||
free(tmp);
|
||||
goto notfound;
|
||||
}
|
||||
}
|
||||
free(tmp);
|
||||
}
|
||||
|
||||
*type = _upb_DefType_Type(v);
|
||||
return _upb_DefType_Unpack(v, *type);
|
||||
|
||||
notfound:
|
||||
_upb_DefBuilder_Errf(ctx, "couldn't resolve name '" UPB_STRINGVIEW_FORMAT "'",
|
||||
UPB_STRINGVIEW_ARGS(sym));
|
||||
}
|
||||
|
||||
const void* _upb_DefBuilder_Resolve(upb_DefBuilder* ctx,
|
||||
const char* from_name_dbg, const char* base,
|
||||
upb_StringView sym, upb_deftype_t type) {
|
||||
upb_deftype_t found_type;
|
||||
const void* ret =
|
||||
_upb_DefBuilder_ResolveAny(ctx, from_name_dbg, base, sym, &found_type);
|
||||
if (ret && found_type != type) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"type mismatch when resolving %s: couldn't find "
|
||||
"name " UPB_STRINGVIEW_FORMAT " with type=%d",
|
||||
from_name_dbg, UPB_STRINGVIEW_ARGS(sym), (int)type);
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
// Per ASCII this will lower-case a letter. If the result is a letter, the
|
||||
// input was definitely a letter. If the output is not a letter, this may
|
||||
// have transformed the character unpredictably.
|
||||
static char upb_ascii_lower(char ch) { return ch | 0x20; }
|
||||
|
||||
// isalpha() etc. from <ctype.h> are locale-dependent, which we don't want.
|
||||
static bool upb_isbetween(uint8_t c, uint8_t low, uint8_t high) {
|
||||
return low <= c && c <= high;
|
||||
}
|
||||
|
||||
static bool upb_isletter(char c) {
|
||||
char lower = upb_ascii_lower(c);
|
||||
return upb_isbetween(lower, 'a', 'z') || c == '_';
|
||||
}
|
||||
|
||||
static bool upb_isalphanum(char c) {
|
||||
return upb_isletter(c) || upb_isbetween(c, '0', '9');
|
||||
}
|
||||
|
||||
static bool TryGetChar(const char** src, const char* end, char* ch) {
|
||||
if (*src == end) return false;
|
||||
*ch = **src;
|
||||
*src += 1;
|
||||
return true;
|
||||
}
|
||||
|
||||
static int TryGetHexDigit(const char** src, const char* end) {
|
||||
char ch;
|
||||
if (!TryGetChar(src, end, &ch)) return -1;
|
||||
if ('0' <= ch && ch <= '9') {
|
||||
return ch - '0';
|
||||
}
|
||||
ch = upb_ascii_lower(ch);
|
||||
if ('a' <= ch && ch <= 'f') {
|
||||
return ch - 'a' + 0xa;
|
||||
}
|
||||
*src -= 1; // Char wasn't actually a hex digit.
|
||||
return -1;
|
||||
}
|
||||
|
||||
static char upb_DefBuilder_ParseHexEscape(upb_DefBuilder* ctx,
|
||||
const upb_FieldDef* f,
|
||||
const char** src, const char* end) {
|
||||
int hex_digit = TryGetHexDigit(src, end);
|
||||
if (hex_digit < 0) {
|
||||
_upb_DefBuilder_Errf(
|
||||
ctx, "\\x must be followed by at least one hex digit (field='%s')",
|
||||
upb_FieldDef_FullName(f));
|
||||
return 0;
|
||||
}
|
||||
unsigned int ret = hex_digit;
|
||||
while ((hex_digit = TryGetHexDigit(src, end)) >= 0) {
|
||||
ret = (ret << 4) | hex_digit;
|
||||
}
|
||||
if (ret > 0xff) {
|
||||
_upb_DefBuilder_Errf(ctx, "Value of hex escape in field %s exceeds 8 bits",
|
||||
upb_FieldDef_FullName(f));
|
||||
return 0;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
static char TryGetOctalDigit(const char** src, const char* end) {
|
||||
char ch;
|
||||
if (!TryGetChar(src, end, &ch)) return -1;
|
||||
if ('0' <= ch && ch <= '7') {
|
||||
return ch - '0';
|
||||
}
|
||||
*src -= 1; // Char wasn't actually an octal digit.
|
||||
return -1;
|
||||
}
|
||||
|
||||
static char upb_DefBuilder_ParseOctalEscape(upb_DefBuilder* ctx,
|
||||
const upb_FieldDef* f,
|
||||
const char** src, const char* end) {
|
||||
char ch = 0;
|
||||
for (int i = 0; i < 3; i++) {
|
||||
char digit;
|
||||
if ((digit = TryGetOctalDigit(src, end)) >= 0) {
|
||||
ch = (ch << 3) | digit;
|
||||
}
|
||||
}
|
||||
return ch;
|
||||
}
|
||||
|
||||
char _upb_DefBuilder_ParseEscape(upb_DefBuilder* ctx, const upb_FieldDef* f,
|
||||
const char** src, const char* end) {
|
||||
char ch;
|
||||
if (!TryGetChar(src, end, &ch)) {
|
||||
_upb_DefBuilder_Errf(ctx, "unterminated escape sequence in field %s",
|
||||
upb_FieldDef_FullName(f));
|
||||
return 0;
|
||||
}
|
||||
switch (ch) {
|
||||
case 'a':
|
||||
return '\a';
|
||||
case 'b':
|
||||
return '\b';
|
||||
case 'f':
|
||||
return '\f';
|
||||
case 'n':
|
||||
return '\n';
|
||||
case 'r':
|
||||
return '\r';
|
||||
case 't':
|
||||
return '\t';
|
||||
case 'v':
|
||||
return '\v';
|
||||
case '\\':
|
||||
return '\\';
|
||||
case '\'':
|
||||
return '\'';
|
||||
case '\"':
|
||||
return '\"';
|
||||
case '?':
|
||||
return '\?';
|
||||
case 'x':
|
||||
case 'X':
|
||||
return upb_DefBuilder_ParseHexEscape(ctx, f, src, end);
|
||||
case '0':
|
||||
case '1':
|
||||
case '2':
|
||||
case '3':
|
||||
case '4':
|
||||
case '5':
|
||||
case '6':
|
||||
case '7':
|
||||
*src -= 1;
|
||||
return upb_DefBuilder_ParseOctalEscape(ctx, f, src, end);
|
||||
}
|
||||
_upb_DefBuilder_Errf(ctx, "Unknown escape sequence: \\%c", ch);
|
||||
}
|
||||
|
||||
void _upb_DefBuilder_CheckIdentSlow(upb_DefBuilder* ctx, upb_StringView name,
|
||||
bool full) {
|
||||
const char* str = name.data;
|
||||
const size_t len = name.size;
|
||||
bool start = true;
|
||||
for (size_t i = 0; i < len; i++) {
|
||||
const char c = str[i];
|
||||
if (c == '.') {
|
||||
if (start || !full) {
|
||||
_upb_DefBuilder_Errf(
|
||||
ctx, "invalid name: unexpected '.' (" UPB_STRINGVIEW_FORMAT ")",
|
||||
UPB_STRINGVIEW_ARGS(name));
|
||||
}
|
||||
start = true;
|
||||
} else if (start) {
|
||||
if (!upb_isletter(c)) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"invalid name: path components must start with a "
|
||||
"letter (" UPB_STRINGVIEW_FORMAT ")",
|
||||
UPB_STRINGVIEW_ARGS(name));
|
||||
}
|
||||
start = false;
|
||||
} else if (!upb_isalphanum(c)) {
|
||||
_upb_DefBuilder_Errf(
|
||||
ctx,
|
||||
"invalid name: non-alphanumeric character (" UPB_STRINGVIEW_FORMAT
|
||||
")",
|
||||
UPB_STRINGVIEW_ARGS(name));
|
||||
}
|
||||
}
|
||||
if (start) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"invalid name: empty part (" UPB_STRINGVIEW_FORMAT ")",
|
||||
UPB_STRINGVIEW_ARGS(name));
|
||||
}
|
||||
|
||||
// We should never reach this point.
|
||||
UPB_ASSERT(false);
|
||||
}
|
||||
137
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/def_builder.h
generated
vendored
Normal file
137
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/def_builder.h
generated
vendored
Normal file
@@ -0,0 +1,137 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_DEF_BUILDER_INTERNAL_H_
|
||||
#define UPB_REFLECTION_DEF_BUILDER_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_pool.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// We want to copy the options verbatim into the destination options proto.
|
||||
// We use serialize+parse as our deep copy.
|
||||
#define UPB_DEF_SET_OPTIONS(target, desc_type, options_type, proto) \
|
||||
if (UPB_DESC(desc_type##_has_options)(proto)) { \
|
||||
size_t size; \
|
||||
char* pb = UPB_DESC(options_type##_serialize)( \
|
||||
UPB_DESC(desc_type##_options)(proto), ctx->tmp_arena, &size); \
|
||||
if (!pb) _upb_DefBuilder_OomErr(ctx); \
|
||||
target = \
|
||||
UPB_DESC(options_type##_parse)(pb, size, _upb_DefBuilder_Arena(ctx)); \
|
||||
if (!target) _upb_DefBuilder_OomErr(ctx); \
|
||||
} else { \
|
||||
target = (const UPB_DESC(options_type)*)kUpbDefOptDefault; \
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
struct upb_DefBuilder {
|
||||
upb_DefPool* symtab;
|
||||
upb_FileDef* file; // File we are building.
|
||||
upb_Arena* arena; // Allocate defs here.
|
||||
upb_Arena* tmp_arena; // For temporary allocations.
|
||||
upb_Status* status; // Record errors here.
|
||||
const upb_MiniTableFile* layout; // NULL if we should build layouts.
|
||||
upb_MiniTablePlatform platform; // Platform we are targeting.
|
||||
int enum_count; // Count of enums built so far.
|
||||
int msg_count; // Count of messages built so far.
|
||||
int ext_count; // Count of extensions built so far.
|
||||
jmp_buf err; // longjmp() on error.
|
||||
};
|
||||
|
||||
extern const char* kUpbDefOptDefault;
|
||||
|
||||
// ctx->status has already been set elsewhere so just fail/longjmp()
|
||||
UPB_NORETURN void _upb_DefBuilder_FailJmp(upb_DefBuilder* ctx);
|
||||
|
||||
UPB_NORETURN void _upb_DefBuilder_Errf(upb_DefBuilder* ctx, const char* fmt,
|
||||
...) UPB_PRINTF(2, 3);
|
||||
UPB_NORETURN void _upb_DefBuilder_OomErr(upb_DefBuilder* ctx);
|
||||
|
||||
const char* _upb_DefBuilder_MakeFullName(upb_DefBuilder* ctx,
|
||||
const char* prefix,
|
||||
upb_StringView name);
|
||||
|
||||
// Given a symbol and the base symbol inside which it is defined,
|
||||
// find the symbol's definition.
|
||||
const void* _upb_DefBuilder_ResolveAny(upb_DefBuilder* ctx,
|
||||
const char* from_name_dbg,
|
||||
const char* base, upb_StringView sym,
|
||||
upb_deftype_t* type);
|
||||
|
||||
const void* _upb_DefBuilder_Resolve(upb_DefBuilder* ctx,
|
||||
const char* from_name_dbg, const char* base,
|
||||
upb_StringView sym, upb_deftype_t type);
|
||||
|
||||
char _upb_DefBuilder_ParseEscape(upb_DefBuilder* ctx, const upb_FieldDef* f,
|
||||
const char** src, const char* end);
|
||||
|
||||
const char* _upb_DefBuilder_FullToShort(const char* fullname);
|
||||
|
||||
UPB_INLINE void* _upb_DefBuilder_Alloc(upb_DefBuilder* ctx, size_t bytes) {
|
||||
if (bytes == 0) return NULL;
|
||||
void* ret = upb_Arena_Malloc(ctx->arena, bytes);
|
||||
if (!ret) _upb_DefBuilder_OomErr(ctx);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// Adds a symbol |v| to the symtab, which must be a def pointer previously
|
||||
// packed with pack_def(). The def's pointer to upb_FileDef* must be set before
|
||||
// adding, so we know which entries to remove if building this file fails.
|
||||
UPB_INLINE void _upb_DefBuilder_Add(upb_DefBuilder* ctx, const char* name,
|
||||
upb_value v) {
|
||||
upb_StringView sym = {.data = name, .size = strlen(name)};
|
||||
bool ok = _upb_DefPool_InsertSym(ctx->symtab, sym, v, ctx->status);
|
||||
if (!ok) _upb_DefBuilder_FailJmp(ctx);
|
||||
}
|
||||
|
||||
UPB_INLINE upb_Arena* _upb_DefBuilder_Arena(const upb_DefBuilder* ctx) {
|
||||
return ctx->arena;
|
||||
}
|
||||
|
||||
UPB_INLINE upb_FileDef* _upb_DefBuilder_File(const upb_DefBuilder* ctx) {
|
||||
return ctx->file;
|
||||
}
|
||||
|
||||
// This version of CheckIdent() is only called by other, faster versions after
|
||||
// they detect a parsing error.
|
||||
void _upb_DefBuilder_CheckIdentSlow(upb_DefBuilder* ctx, upb_StringView name,
|
||||
bool full);
|
||||
|
||||
// Verify a full identifier string. This is slightly more complicated than
|
||||
// verifying a relative identifier string because we must track '.' chars.
|
||||
UPB_INLINE void _upb_DefBuilder_CheckIdentFull(upb_DefBuilder* ctx,
|
||||
upb_StringView name) {
|
||||
bool good = name.size > 0;
|
||||
bool start = true;
|
||||
|
||||
for (size_t i = 0; i < name.size; i++) {
|
||||
const char c = name.data[i];
|
||||
const char d = c | 0x20; // force lowercase
|
||||
const bool is_alpha = (('a' <= d) & (d <= 'z')) | (c == '_');
|
||||
const bool is_numer = ('0' <= c) & (c <= '9') & !start;
|
||||
const bool is_dot = (c == '.') & !start;
|
||||
|
||||
good &= is_alpha | is_numer | is_dot;
|
||||
start = is_dot;
|
||||
}
|
||||
|
||||
if (!good) _upb_DefBuilder_CheckIdentSlow(ctx, name, true);
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_DEF_BUILDER_INTERNAL_H_ */
|
||||
57
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/def_pool.h
generated
vendored
Normal file
57
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/def_pool.h
generated
vendored
Normal file
@@ -0,0 +1,57 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_DEF_POOL_INTERNAL_H_
|
||||
#define UPB_REFLECTION_DEF_POOL_INTERNAL_H_
|
||||
|
||||
#include "upb/mini_descriptor/decode.h"
|
||||
#include "upb/reflection/def_pool.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_Arena* _upb_DefPool_Arena(const upb_DefPool* s);
|
||||
size_t _upb_DefPool_BytesLoaded(const upb_DefPool* s);
|
||||
upb_ExtensionRegistry* _upb_DefPool_ExtReg(const upb_DefPool* s);
|
||||
|
||||
bool _upb_DefPool_InsertExt(upb_DefPool* s, const upb_MiniTableExtension* ext,
|
||||
const upb_FieldDef* f);
|
||||
bool _upb_DefPool_InsertSym(upb_DefPool* s, upb_StringView sym, upb_value v,
|
||||
upb_Status* status);
|
||||
bool _upb_DefPool_LookupSym(const upb_DefPool* s, const char* sym, size_t size,
|
||||
upb_value* v);
|
||||
|
||||
void** _upb_DefPool_ScratchData(const upb_DefPool* s);
|
||||
size_t* _upb_DefPool_ScratchSize(const upb_DefPool* s);
|
||||
void _upb_DefPool_SetPlatform(upb_DefPool* s, upb_MiniTablePlatform platform);
|
||||
|
||||
// For generated code only: loads a generated descriptor.
|
||||
typedef struct _upb_DefPool_Init {
|
||||
struct _upb_DefPool_Init** deps; // Dependencies of this file.
|
||||
const upb_MiniTableFile* layout;
|
||||
const char* filename;
|
||||
upb_StringView descriptor; // Serialized descriptor.
|
||||
} _upb_DefPool_Init;
|
||||
|
||||
bool _upb_DefPool_LoadDefInit(upb_DefPool* s, const _upb_DefPool_Init* init);
|
||||
|
||||
// Should only be directly called by tests. This variant lets us suppress
|
||||
// the use of compiled-in tables, forcing a rebuild of the tables at runtime.
|
||||
bool _upb_DefPool_LoadDefInitEx(upb_DefPool* s, const _upb_DefPool_Init* init,
|
||||
bool rebuild_minitable);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_DEF_POOL_INTERNAL_H_ */
|
||||
44
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/desc_state.h
generated
vendored
Normal file
44
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/desc_state.h
generated
vendored
Normal file
@@ -0,0 +1,44 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_DESC_STATE_INTERNAL_H_
|
||||
#define UPB_REFLECTION_DESC_STATE_INTERNAL_H_
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mini_descriptor/internal/encode.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// Manages the storage for mini descriptor strings as they are being encoded.
|
||||
// TODO: Move some of this state directly into the encoder, maybe.
|
||||
typedef struct {
|
||||
upb_MtDataEncoder e;
|
||||
size_t bufsize;
|
||||
char* buf;
|
||||
char* ptr;
|
||||
} upb_DescState;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_INLINE void _upb_DescState_Init(upb_DescState* d) {
|
||||
d->bufsize = kUpb_MtDataEncoder_MinSize * 2;
|
||||
d->buf = NULL;
|
||||
d->ptr = NULL;
|
||||
}
|
||||
|
||||
bool _upb_DescState_Grow(upb_DescState* d, upb_Arena* a);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_DESC_STATE_INTERNAL_H_ */
|
||||
36
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/enum_def.h
generated
vendored
Normal file
36
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/enum_def.h
generated
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_ENUM_DEF_INTERNAL_H_
|
||||
#define UPB_REFLECTION_ENUM_DEF_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/enum_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_EnumDef* _upb_EnumDef_At(const upb_EnumDef* e, int i);
|
||||
bool _upb_EnumDef_Insert(upb_EnumDef* e, upb_EnumValueDef* v, upb_Arena* a);
|
||||
const upb_MiniTableEnum* _upb_EnumDef_MiniTable(const upb_EnumDef* e);
|
||||
|
||||
// Allocate and initialize an array of |n| enum defs.
|
||||
upb_EnumDef* _upb_EnumDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(EnumDescriptorProto) * const* protos,
|
||||
const upb_MessageDef* containing_type);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_ENUM_DEF_INTERNAL_H_ */
|
||||
35
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/enum_reserved_range.h
generated
vendored
Normal file
35
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/enum_reserved_range.h
generated
vendored
Normal file
@@ -0,0 +1,35 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_ENUM_RESERVED_RANGE_INTERNAL_H_
|
||||
#define UPB_REFLECTION_ENUM_RESERVED_RANGE_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/enum_reserved_range.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_EnumReservedRange* _upb_EnumReservedRange_At(const upb_EnumReservedRange* r,
|
||||
int i);
|
||||
|
||||
// Allocate and initialize an array of |n| reserved ranges owned by |e|.
|
||||
upb_EnumReservedRange* _upb_EnumReservedRanges_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(EnumDescriptorProto_EnumReservedRange) * const* protos,
|
||||
const upb_EnumDef* e);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_ENUM_RESERVED_RANGE_INTERNAL_H_ */
|
||||
37
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/enum_value_def.h
generated
vendored
Normal file
37
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/enum_value_def.h
generated
vendored
Normal file
@@ -0,0 +1,37 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_ENUM_VALUE_DEF_INTERNAL_H_
|
||||
#define UPB_REFLECTION_ENUM_VALUE_DEF_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/enum_value_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_EnumValueDef* _upb_EnumValueDef_At(const upb_EnumValueDef* v, int i);
|
||||
|
||||
// Allocate and initialize an array of |n| enum value defs owned by |e|.
|
||||
upb_EnumValueDef* _upb_EnumValueDefs_New(
|
||||
upb_DefBuilder* ctx, const char* prefix, int n,
|
||||
const UPB_DESC(EnumValueDescriptorProto) * const* protos, upb_EnumDef* e,
|
||||
bool* is_sorted);
|
||||
|
||||
const upb_EnumValueDef** _upb_EnumValueDefs_Sorted(const upb_EnumValueDef* v,
|
||||
int n, upb_Arena* a);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_ENUM_VALUE_DEF_INTERNAL_H_ */
|
||||
34
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/extension_range.h
generated
vendored
Normal file
34
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/extension_range.h
generated
vendored
Normal file
@@ -0,0 +1,34 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_EXTENSION_RANGE_INTERNAL_H_
|
||||
#define UPB_REFLECTION_EXTENSION_RANGE_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/extension_range.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_ExtensionRange* _upb_ExtensionRange_At(const upb_ExtensionRange* r, int i);
|
||||
|
||||
// Allocate and initialize an array of |n| extension ranges owned by |m|.
|
||||
upb_ExtensionRange* _upb_ExtensionRanges_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(DescriptorProto_ExtensionRange) * const* protos,
|
||||
const upb_MessageDef* m);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_EXTENSION_RANGE_INTERNAL_H_ */
|
||||
56
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/field_def.h
generated
vendored
Normal file
56
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/field_def.h
generated
vendored
Normal file
@@ -0,0 +1,56 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_FIELD_DEF_INTERNAL_H_
|
||||
#define UPB_REFLECTION_FIELD_DEF_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/field_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_FieldDef* _upb_FieldDef_At(const upb_FieldDef* f, int i);
|
||||
|
||||
const upb_MiniTableExtension* _upb_FieldDef_ExtensionMiniTable(
|
||||
const upb_FieldDef* f);
|
||||
bool _upb_FieldDef_IsClosedEnum(const upb_FieldDef* f);
|
||||
bool _upb_FieldDef_IsProto3Optional(const upb_FieldDef* f);
|
||||
int _upb_FieldDef_LayoutIndex(const upb_FieldDef* f);
|
||||
uint64_t _upb_FieldDef_Modifiers(const upb_FieldDef* f);
|
||||
void _upb_FieldDef_Resolve(upb_DefBuilder* ctx, const char* prefix,
|
||||
upb_FieldDef* f);
|
||||
void _upb_FieldDef_BuildMiniTableExtension(upb_DefBuilder* ctx,
|
||||
const upb_FieldDef* f);
|
||||
|
||||
// Allocate and initialize an array of |n| extensions (field defs).
|
||||
upb_FieldDef* _upb_Extensions_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(FieldDescriptorProto) * const* protos, const char* prefix,
|
||||
upb_MessageDef* m);
|
||||
|
||||
// Allocate and initialize an array of |n| field defs.
|
||||
upb_FieldDef* _upb_FieldDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(FieldDescriptorProto) * const* protos, const char* prefix,
|
||||
upb_MessageDef* m, bool* is_sorted);
|
||||
|
||||
// Allocate and return a list of pointers to the |n| field defs in |ff|,
|
||||
// sorted by field number.
|
||||
const upb_FieldDef** _upb_FieldDefs_Sorted(const upb_FieldDef* f, int n,
|
||||
upb_Arena* a);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_FIELD_DEF_INTERNAL_H_ */
|
||||
37
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/file_def.h
generated
vendored
Normal file
37
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/file_def.h
generated
vendored
Normal file
@@ -0,0 +1,37 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_FILE_DEF_INTERNAL_H_
|
||||
#define UPB_REFLECTION_FILE_DEF_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/file_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
const upb_MiniTableExtension* _upb_FileDef_ExtensionMiniTable(
|
||||
const upb_FileDef* f, int i);
|
||||
const int32_t* _upb_FileDef_PublicDependencyIndexes(const upb_FileDef* f);
|
||||
const int32_t* _upb_FileDef_WeakDependencyIndexes(const upb_FileDef* f);
|
||||
|
||||
// upb_FileDef_Package() returns "" if f->package is NULL, this does not.
|
||||
const char* _upb_FileDef_RawPackage(const upb_FileDef* f);
|
||||
|
||||
void _upb_FileDef_Create(upb_DefBuilder* ctx,
|
||||
const UPB_DESC(FileDescriptorProto) * file_proto);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_FILE_DEF_INTERNAL_H_ */
|
||||
43
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/message_def.h
generated
vendored
Normal file
43
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/message_def.h
generated
vendored
Normal file
@@ -0,0 +1,43 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_MESSAGE_DEF_INTERNAL_H_
|
||||
#define UPB_REFLECTION_MESSAGE_DEF_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/message_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_MessageDef* _upb_MessageDef_At(const upb_MessageDef* m, int i);
|
||||
bool _upb_MessageDef_InMessageSet(const upb_MessageDef* m);
|
||||
bool _upb_MessageDef_Insert(upb_MessageDef* m, const char* name, size_t size,
|
||||
upb_value v, upb_Arena* a);
|
||||
void _upb_MessageDef_InsertField(upb_DefBuilder* ctx, upb_MessageDef* m,
|
||||
const upb_FieldDef* f);
|
||||
bool _upb_MessageDef_IsValidExtensionNumber(const upb_MessageDef* m, int n);
|
||||
void _upb_MessageDef_CreateMiniTable(upb_DefBuilder* ctx, upb_MessageDef* m);
|
||||
void _upb_MessageDef_LinkMiniTable(upb_DefBuilder* ctx,
|
||||
const upb_MessageDef* m);
|
||||
void _upb_MessageDef_Resolve(upb_DefBuilder* ctx, upb_MessageDef* m);
|
||||
|
||||
// Allocate and initialize an array of |n| message defs.
|
||||
upb_MessageDef* _upb_MessageDefs_New(
|
||||
upb_DefBuilder* ctx, int n, const UPB_DESC(DescriptorProto) * const* protos,
|
||||
const upb_MessageDef* containing_type);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_MESSAGE_DEF_INTERNAL_H_ */
|
||||
35
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/message_reserved_range.h
generated
vendored
Normal file
35
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/message_reserved_range.h
generated
vendored
Normal file
@@ -0,0 +1,35 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_MESSAGE_RESERVED_RANGE_INTERNAL_H_
|
||||
#define UPB_REFLECTION_MESSAGE_RESERVED_RANGE_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/message_reserved_range.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_MessageReservedRange* _upb_MessageReservedRange_At(
|
||||
const upb_MessageReservedRange* r, int i);
|
||||
|
||||
// Allocate and initialize an array of |n| reserved ranges owned by |m|.
|
||||
upb_MessageReservedRange* _upb_MessageReservedRanges_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(DescriptorProto_ReservedRange) * const* protos,
|
||||
const upb_MessageDef* m);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_MESSAGE_RESERVED_RANGE_INTERNAL_H_ */
|
||||
33
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/method_def.h
generated
vendored
Normal file
33
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/method_def.h
generated
vendored
Normal file
@@ -0,0 +1,33 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_METHOD_DEF_INTERNAL_H_
|
||||
#define UPB_REFLECTION_METHOD_DEF_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/method_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_MethodDef* _upb_MethodDef_At(const upb_MethodDef* m, int i);
|
||||
|
||||
// Allocate and initialize an array of |n| method defs owned by |s|.
|
||||
upb_MethodDef* _upb_MethodDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(MethodDescriptorProto) * const* protos, upb_ServiceDef* s);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_METHOD_DEF_INTERNAL_H_ */
|
||||
37
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/oneof_def.h
generated
vendored
Normal file
37
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/oneof_def.h
generated
vendored
Normal file
@@ -0,0 +1,37 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_ONEOF_DEF_INTERNAL_H_
|
||||
#define UPB_REFLECTION_ONEOF_DEF_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/oneof_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_OneofDef* _upb_OneofDef_At(const upb_OneofDef* o, int i);
|
||||
void _upb_OneofDef_Insert(upb_DefBuilder* ctx, upb_OneofDef* o,
|
||||
const upb_FieldDef* f, const char* name, size_t size);
|
||||
|
||||
// Allocate and initialize an array of |n| oneof defs owned by |m|.
|
||||
upb_OneofDef* _upb_OneofDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(OneofDescriptorProto) * const* protos, upb_MessageDef* m);
|
||||
|
||||
size_t _upb_OneofDefs_Finalize(upb_DefBuilder* ctx, upb_MessageDef* m);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_ONEOF_DEF_INTERNAL_H_ */
|
||||
33
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/service_def.h
generated
vendored
Normal file
33
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/service_def.h
generated
vendored
Normal file
@@ -0,0 +1,33 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_SERVICE_DEF_INTERNAL_H_
|
||||
#define UPB_REFLECTION_SERVICE_DEF_INTERNAL_H_
|
||||
|
||||
#include "upb/reflection/service_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
upb_ServiceDef* _upb_ServiceDef_At(const upb_ServiceDef* s, int i);
|
||||
|
||||
// Allocate and initialize an array of |n| service defs.
|
||||
upb_ServiceDef* _upb_ServiceDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(ServiceDescriptorProto) * const* protos);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_SERVICE_DEF_INTERNAL_H_ */
|
||||
33
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/strdup2.c
generated
vendored
Normal file
33
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/strdup2.c
generated
vendored
Normal file
@@ -0,0 +1,33 @@
|
||||
// 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/reflection/internal/strdup2.h"
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
char* upb_strdup2(const char* s, size_t len, upb_Arena* a) {
|
||||
size_t n;
|
||||
char* p;
|
||||
|
||||
// Prevent overflow errors.
|
||||
if (len == SIZE_MAX) return NULL;
|
||||
|
||||
// Always null-terminate, even if binary data; but don't rely on the input to
|
||||
// have a null-terminating byte since it may be a raw binary buffer.
|
||||
n = len + 1;
|
||||
p = upb_Arena_Malloc(a, n);
|
||||
if (p) {
|
||||
if (len != 0) memcpy(p, s, len);
|
||||
p[len] = 0;
|
||||
}
|
||||
return p;
|
||||
}
|
||||
32
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/strdup2.h
generated
vendored
Normal file
32
Pods/gRPC-Core/third_party/upb/upb/reflection/internal/strdup2.h
generated
vendored
Normal file
@@ -0,0 +1,32 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_INTERNAL_STRDUP2_H_
|
||||
#define UPB_REFLECTION_INTERNAL_STRDUP2_H_
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Variant that works with a length-delimited rather than NULL-delimited string,
|
||||
// as supported by strtable.
|
||||
char* upb_strdup2(const char* s, size_t len, upb_Arena* a);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_INTERNAL_STRDUP2_H_ */
|
||||
213
Pods/gRPC-Core/third_party/upb/upb/reflection/message.c
generated
vendored
Normal file
213
Pods/gRPC-Core/third_party/upb/upb/reflection/message.c
generated
vendored
Normal file
@@ -0,0 +1,213 @@
|
||||
// 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/reflection/message.h"
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/hash/common.h"
|
||||
#include "upb/message/accessors.h"
|
||||
#include "upb/message/map.h"
|
||||
#include "upb/message/message.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
#include "upb/reflection/def.h"
|
||||
#include "upb/reflection/def_pool.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/field_def.h"
|
||||
#include "upb/reflection/message_def.h"
|
||||
#include "upb/reflection/oneof_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
bool upb_Message_HasFieldByDef(const upb_Message* msg, const upb_FieldDef* f) {
|
||||
UPB_ASSERT(upb_FieldDef_HasPresence(f));
|
||||
return upb_Message_HasField(msg, upb_FieldDef_MiniTable(f));
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_Message_WhichOneof(const upb_Message* msg,
|
||||
const upb_OneofDef* o) {
|
||||
const upb_FieldDef* f = upb_OneofDef_Field(o, 0);
|
||||
if (upb_OneofDef_IsSynthetic(o)) {
|
||||
UPB_ASSERT(upb_OneofDef_FieldCount(o) == 1);
|
||||
return upb_Message_HasFieldByDef(msg, f) ? f : NULL;
|
||||
} else {
|
||||
const upb_MiniTableField* field = upb_FieldDef_MiniTable(f);
|
||||
uint32_t oneof_case = upb_Message_WhichOneofFieldNumber(msg, field);
|
||||
f = oneof_case ? upb_OneofDef_LookupNumber(o, oneof_case) : NULL;
|
||||
UPB_ASSERT((f != NULL) == (oneof_case != 0));
|
||||
return f;
|
||||
}
|
||||
}
|
||||
|
||||
upb_MessageValue upb_Message_GetFieldByDef(const upb_Message* msg,
|
||||
const upb_FieldDef* f) {
|
||||
upb_MessageValue default_val = upb_FieldDef_Default(f);
|
||||
upb_MessageValue ret;
|
||||
_upb_Message_GetField(msg, upb_FieldDef_MiniTable(f), &default_val, &ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
upb_MutableMessageValue upb_Message_Mutable(upb_Message* msg,
|
||||
const upb_FieldDef* f,
|
||||
upb_Arena* a) {
|
||||
UPB_ASSERT(upb_FieldDef_IsSubMessage(f) || upb_FieldDef_IsRepeated(f));
|
||||
if (upb_FieldDef_HasPresence(f) && !upb_Message_HasFieldByDef(msg, f)) {
|
||||
// We need to skip the upb_Message_GetFieldByDef() call in this case.
|
||||
goto make;
|
||||
}
|
||||
|
||||
upb_MessageValue val = upb_Message_GetFieldByDef(msg, f);
|
||||
if (val.array_val) {
|
||||
return (upb_MutableMessageValue){.array = (upb_Array*)val.array_val};
|
||||
}
|
||||
|
||||
upb_MutableMessageValue ret;
|
||||
make:
|
||||
if (!a) return (upb_MutableMessageValue){.array = NULL};
|
||||
if (upb_FieldDef_IsMap(f)) {
|
||||
const upb_MessageDef* entry = upb_FieldDef_MessageSubDef(f);
|
||||
const upb_FieldDef* key =
|
||||
upb_MessageDef_FindFieldByNumber(entry, kUpb_MapEntry_KeyFieldNumber);
|
||||
const upb_FieldDef* value =
|
||||
upb_MessageDef_FindFieldByNumber(entry, kUpb_MapEntry_ValueFieldNumber);
|
||||
ret.map =
|
||||
upb_Map_New(a, upb_FieldDef_CType(key), upb_FieldDef_CType(value));
|
||||
} else if (upb_FieldDef_IsRepeated(f)) {
|
||||
ret.array = upb_Array_New(a, upb_FieldDef_CType(f));
|
||||
} else {
|
||||
UPB_ASSERT(upb_FieldDef_IsSubMessage(f));
|
||||
const upb_MessageDef* m = upb_FieldDef_MessageSubDef(f);
|
||||
ret.msg = upb_Message_New(upb_MessageDef_MiniTable(m), a);
|
||||
}
|
||||
|
||||
val.array_val = ret.array;
|
||||
upb_Message_SetFieldByDef(msg, f, val, a);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
bool upb_Message_SetFieldByDef(upb_Message* msg, const upb_FieldDef* f,
|
||||
upb_MessageValue val, upb_Arena* a) {
|
||||
return _upb_Message_SetField(msg, upb_FieldDef_MiniTable(f), &val, a);
|
||||
}
|
||||
|
||||
void upb_Message_ClearFieldByDef(upb_Message* msg, const upb_FieldDef* f) {
|
||||
upb_Message_ClearField(msg, upb_FieldDef_MiniTable(f));
|
||||
}
|
||||
|
||||
void upb_Message_ClearByDef(upb_Message* msg, const upb_MessageDef* m) {
|
||||
upb_Message_Clear(msg, upb_MessageDef_MiniTable(m));
|
||||
}
|
||||
|
||||
bool upb_Message_Next(const upb_Message* msg, const upb_MessageDef* m,
|
||||
const upb_DefPool* ext_pool, const upb_FieldDef** out_f,
|
||||
upb_MessageValue* out_val, size_t* iter) {
|
||||
size_t i = *iter;
|
||||
size_t n = upb_MessageDef_FieldCount(m);
|
||||
UPB_UNUSED(ext_pool);
|
||||
|
||||
// Iterate over normal fields, returning the first one that is set.
|
||||
while (++i < n) {
|
||||
const upb_FieldDef* f = upb_MessageDef_Field(m, i);
|
||||
const upb_MiniTableField* field = upb_FieldDef_MiniTable(f);
|
||||
upb_MessageValue val = upb_Message_GetFieldByDef(msg, f);
|
||||
|
||||
// Skip field if unset or empty.
|
||||
if (upb_MiniTableField_HasPresence(field)) {
|
||||
if (!upb_Message_HasFieldByDef(msg, f)) continue;
|
||||
} else {
|
||||
switch (upb_FieldMode_Get(field)) {
|
||||
case kUpb_FieldMode_Map:
|
||||
if (!val.map_val || upb_Map_Size(val.map_val) == 0) continue;
|
||||
break;
|
||||
case kUpb_FieldMode_Array:
|
||||
if (!val.array_val || upb_Array_Size(val.array_val) == 0) continue;
|
||||
break;
|
||||
case kUpb_FieldMode_Scalar:
|
||||
if (!_upb_MiniTable_ValueIsNonZero(&val, field)) continue;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
*out_val = val;
|
||||
*out_f = f;
|
||||
*iter = i;
|
||||
return true;
|
||||
}
|
||||
|
||||
if (ext_pool) {
|
||||
// Return any extensions that are set.
|
||||
size_t count;
|
||||
const upb_Message_Extension* ext = _upb_Message_Getexts(msg, &count);
|
||||
if (i - n < count) {
|
||||
ext += count - 1 - (i - n);
|
||||
memcpy(out_val, &ext->data, sizeof(*out_val));
|
||||
*out_f = upb_DefPool_FindExtensionByMiniTable(ext_pool, ext->ext);
|
||||
*iter = i;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
*iter = i;
|
||||
return false;
|
||||
}
|
||||
|
||||
bool _upb_Message_DiscardUnknown(upb_Message* msg, const upb_MessageDef* m,
|
||||
int depth) {
|
||||
size_t iter = kUpb_Message_Begin;
|
||||
const upb_FieldDef* f;
|
||||
upb_MessageValue val;
|
||||
bool ret = true;
|
||||
|
||||
if (--depth == 0) return false;
|
||||
|
||||
_upb_Message_DiscardUnknown_shallow(msg);
|
||||
|
||||
while (upb_Message_Next(msg, m, NULL /*ext_pool*/, &f, &val, &iter)) {
|
||||
const upb_MessageDef* subm = upb_FieldDef_MessageSubDef(f);
|
||||
if (!subm) continue;
|
||||
if (upb_FieldDef_IsMap(f)) {
|
||||
const upb_FieldDef* val_f = upb_MessageDef_FindFieldByNumber(subm, 2);
|
||||
const upb_MessageDef* val_m = upb_FieldDef_MessageSubDef(val_f);
|
||||
upb_Map* map = (upb_Map*)val.map_val;
|
||||
size_t iter = kUpb_Map_Begin;
|
||||
|
||||
if (!val_m) continue;
|
||||
|
||||
upb_MessageValue map_key, map_val;
|
||||
while (upb_Map_Next(map, &map_key, &map_val, &iter)) {
|
||||
if (!_upb_Message_DiscardUnknown((upb_Message*)map_val.msg_val, val_m,
|
||||
depth)) {
|
||||
ret = false;
|
||||
}
|
||||
}
|
||||
} else if (upb_FieldDef_IsRepeated(f)) {
|
||||
const upb_Array* arr = val.array_val;
|
||||
size_t i, n = upb_Array_Size(arr);
|
||||
for (i = 0; i < n; i++) {
|
||||
upb_MessageValue elem = upb_Array_Get(arr, i);
|
||||
if (!_upb_Message_DiscardUnknown((upb_Message*)elem.msg_val, subm,
|
||||
depth)) {
|
||||
ret = false;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if (!_upb_Message_DiscardUnknown((upb_Message*)val.msg_val, subm,
|
||||
depth)) {
|
||||
ret = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
bool upb_Message_DiscardUnknown(upb_Message* msg, const upb_MessageDef* m,
|
||||
int maxdepth) {
|
||||
return _upb_Message_DiscardUnknown(msg, m, maxdepth);
|
||||
}
|
||||
89
Pods/gRPC-Core/third_party/upb/upb/reflection/message.h
generated
vendored
Normal file
89
Pods/gRPC-Core/third_party/upb/upb/reflection/message.h
generated
vendored
Normal file
@@ -0,0 +1,89 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_MESSAGE_H_
|
||||
#define UPB_REFLECTION_MESSAGE_H_
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/map.h"
|
||||
#include "upb/message/types.h"
|
||||
#include "upb/message/value.h" // IWYU pragma: export
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// Returns a mutable pointer to a map, array, or submessage value. If the given
|
||||
// arena is non-NULL this will construct a new object if it was not previously
|
||||
// present. May not be called for primitive fields.
|
||||
UPB_API upb_MutableMessageValue upb_Message_Mutable(upb_Message* msg,
|
||||
const upb_FieldDef* f,
|
||||
upb_Arena* a);
|
||||
|
||||
// Returns the field that is set in the oneof, or NULL if none are set.
|
||||
UPB_API const upb_FieldDef* upb_Message_WhichOneof(const upb_Message* msg,
|
||||
const upb_OneofDef* o);
|
||||
|
||||
// Clear all data and unknown fields.
|
||||
void upb_Message_ClearByDef(upb_Message* msg, const upb_MessageDef* m);
|
||||
|
||||
// Clears any field presence and sets the value back to its default.
|
||||
UPB_API void upb_Message_ClearFieldByDef(upb_Message* msg,
|
||||
const upb_FieldDef* f);
|
||||
|
||||
// May only be called for fields where upb_FieldDef_HasPresence(f) == true.
|
||||
UPB_API bool upb_Message_HasFieldByDef(const upb_Message* msg,
|
||||
const upb_FieldDef* f);
|
||||
|
||||
// Returns the value in the message associated with this field def.
|
||||
UPB_API upb_MessageValue upb_Message_GetFieldByDef(const upb_Message* msg,
|
||||
const upb_FieldDef* f);
|
||||
|
||||
// Sets the given field to the given value. For a msg/array/map/string, the
|
||||
// caller must ensure that the target data outlives |msg| (by living either in
|
||||
// the same arena or a different arena that outlives it).
|
||||
//
|
||||
// Returns false if allocation fails.
|
||||
UPB_API bool upb_Message_SetFieldByDef(upb_Message* msg, const upb_FieldDef* f,
|
||||
upb_MessageValue val, upb_Arena* a);
|
||||
|
||||
// Iterate over present fields.
|
||||
//
|
||||
// size_t iter = kUpb_Message_Begin;
|
||||
// const upb_FieldDef *f;
|
||||
// upb_MessageValue val;
|
||||
// while (upb_Message_Next(msg, m, ext_pool, &f, &val, &iter)) {
|
||||
// process_field(f, val);
|
||||
// }
|
||||
//
|
||||
// If ext_pool is NULL, no extensions will be returned. If the given symtab
|
||||
// returns extensions that don't match what is in this message, those extensions
|
||||
// will be skipped.
|
||||
|
||||
#define kUpb_Message_Begin -1
|
||||
|
||||
bool upb_Message_Next(const upb_Message* msg, const upb_MessageDef* m,
|
||||
const upb_DefPool* ext_pool, const upb_FieldDef** f,
|
||||
upb_MessageValue* val, size_t* iter);
|
||||
|
||||
// Clears all unknown field data from this message and all submessages.
|
||||
UPB_API bool upb_Message_DiscardUnknown(upb_Message* msg,
|
||||
const upb_MessageDef* m, int maxdepth);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_MESSAGE_H_ */
|
||||
19
Pods/gRPC-Core/third_party/upb/upb/reflection/message.hpp
generated
vendored
Normal file
19
Pods/gRPC-Core/third_party/upb/upb/reflection/message.hpp
generated
vendored
Normal file
@@ -0,0 +1,19 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_REFLECTION_MESSAGE_HPP_
|
||||
#define UPB_REFLECTION_MESSAGE_HPP_
|
||||
|
||||
#include "upb/reflection/message.h"
|
||||
|
||||
namespace upb {
|
||||
|
||||
typedef upb_MessageValue MessageValue;
|
||||
|
||||
} // namespace upb
|
||||
|
||||
#endif // UPB_REFLECTION_MESSAGE_HPP_
|
||||
736
Pods/gRPC-Core/third_party/upb/upb/reflection/message_def.c
generated
vendored
Normal file
736
Pods/gRPC-Core/third_party/upb/upb/reflection/message_def.c
generated
vendored
Normal file
@@ -0,0 +1,736 @@
|
||||
// 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/reflection/internal/message_def.h"
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/hash/int_table.h"
|
||||
#include "upb/hash/str_table.h"
|
||||
#include "upb/mini_descriptor/decode.h"
|
||||
#include "upb/mini_descriptor/internal/modifiers.h"
|
||||
#include "upb/reflection/def.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/desc_state.h"
|
||||
#include "upb/reflection/internal/enum_def.h"
|
||||
#include "upb/reflection/internal/extension_range.h"
|
||||
#include "upb/reflection/internal/field_def.h"
|
||||
#include "upb/reflection/internal/file_def.h"
|
||||
#include "upb/reflection/internal/message_reserved_range.h"
|
||||
#include "upb/reflection/internal/oneof_def.h"
|
||||
#include "upb/reflection/internal/strdup2.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_MessageDef {
|
||||
const UPB_DESC(MessageOptions) * opts;
|
||||
const upb_MiniTable* layout;
|
||||
const upb_FileDef* file;
|
||||
const upb_MessageDef* containing_type;
|
||||
const char* full_name;
|
||||
|
||||
// Tables for looking up fields by number and name.
|
||||
upb_inttable itof;
|
||||
upb_strtable ntof;
|
||||
|
||||
// Looking up fields by json name.
|
||||
upb_strtable jtof;
|
||||
|
||||
/* All nested defs.
|
||||
* MEM: We could save some space here by putting nested defs in a contiguous
|
||||
* region and calculating counts from offsets or vice-versa. */
|
||||
const upb_FieldDef* fields;
|
||||
const upb_OneofDef* oneofs;
|
||||
const upb_ExtensionRange* ext_ranges;
|
||||
const upb_StringView* res_names;
|
||||
const upb_MessageDef* nested_msgs;
|
||||
const upb_MessageReservedRange* res_ranges;
|
||||
const upb_EnumDef* nested_enums;
|
||||
const upb_FieldDef* nested_exts;
|
||||
|
||||
// TODO: These counters don't need anywhere near 32 bits.
|
||||
int field_count;
|
||||
int real_oneof_count;
|
||||
int oneof_count;
|
||||
int ext_range_count;
|
||||
int res_range_count;
|
||||
int res_name_count;
|
||||
int nested_msg_count;
|
||||
int nested_enum_count;
|
||||
int nested_ext_count;
|
||||
bool in_message_set;
|
||||
bool is_sorted;
|
||||
upb_WellKnown well_known_type;
|
||||
};
|
||||
|
||||
static void assign_msg_wellknowntype(upb_MessageDef* m) {
|
||||
const char* name = m->full_name;
|
||||
if (name == NULL) {
|
||||
m->well_known_type = kUpb_WellKnown_Unspecified;
|
||||
return;
|
||||
}
|
||||
if (!strcmp(name, "google.protobuf.Any")) {
|
||||
m->well_known_type = kUpb_WellKnown_Any;
|
||||
} else if (!strcmp(name, "google.protobuf.FieldMask")) {
|
||||
m->well_known_type = kUpb_WellKnown_FieldMask;
|
||||
} else if (!strcmp(name, "google.protobuf.Duration")) {
|
||||
m->well_known_type = kUpb_WellKnown_Duration;
|
||||
} else if (!strcmp(name, "google.protobuf.Timestamp")) {
|
||||
m->well_known_type = kUpb_WellKnown_Timestamp;
|
||||
} else if (!strcmp(name, "google.protobuf.DoubleValue")) {
|
||||
m->well_known_type = kUpb_WellKnown_DoubleValue;
|
||||
} else if (!strcmp(name, "google.protobuf.FloatValue")) {
|
||||
m->well_known_type = kUpb_WellKnown_FloatValue;
|
||||
} else if (!strcmp(name, "google.protobuf.Int64Value")) {
|
||||
m->well_known_type = kUpb_WellKnown_Int64Value;
|
||||
} else if (!strcmp(name, "google.protobuf.UInt64Value")) {
|
||||
m->well_known_type = kUpb_WellKnown_UInt64Value;
|
||||
} else if (!strcmp(name, "google.protobuf.Int32Value")) {
|
||||
m->well_known_type = kUpb_WellKnown_Int32Value;
|
||||
} else if (!strcmp(name, "google.protobuf.UInt32Value")) {
|
||||
m->well_known_type = kUpb_WellKnown_UInt32Value;
|
||||
} else if (!strcmp(name, "google.protobuf.BoolValue")) {
|
||||
m->well_known_type = kUpb_WellKnown_BoolValue;
|
||||
} else if (!strcmp(name, "google.protobuf.StringValue")) {
|
||||
m->well_known_type = kUpb_WellKnown_StringValue;
|
||||
} else if (!strcmp(name, "google.protobuf.BytesValue")) {
|
||||
m->well_known_type = kUpb_WellKnown_BytesValue;
|
||||
} else if (!strcmp(name, "google.protobuf.Value")) {
|
||||
m->well_known_type = kUpb_WellKnown_Value;
|
||||
} else if (!strcmp(name, "google.protobuf.ListValue")) {
|
||||
m->well_known_type = kUpb_WellKnown_ListValue;
|
||||
} else if (!strcmp(name, "google.protobuf.Struct")) {
|
||||
m->well_known_type = kUpb_WellKnown_Struct;
|
||||
} else {
|
||||
m->well_known_type = kUpb_WellKnown_Unspecified;
|
||||
}
|
||||
}
|
||||
|
||||
upb_MessageDef* _upb_MessageDef_At(const upb_MessageDef* m, int i) {
|
||||
return (upb_MessageDef*)&m[i];
|
||||
}
|
||||
|
||||
bool _upb_MessageDef_IsValidExtensionNumber(const upb_MessageDef* m, int n) {
|
||||
for (int i = 0; i < m->ext_range_count; i++) {
|
||||
const upb_ExtensionRange* r = upb_MessageDef_ExtensionRange(m, i);
|
||||
if (upb_ExtensionRange_Start(r) <= n && n < upb_ExtensionRange_End(r)) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
const UPB_DESC(MessageOptions) *
|
||||
upb_MessageDef_Options(const upb_MessageDef* m) {
|
||||
return m->opts;
|
||||
}
|
||||
|
||||
bool upb_MessageDef_HasOptions(const upb_MessageDef* m) {
|
||||
return m->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
const char* upb_MessageDef_FullName(const upb_MessageDef* m) {
|
||||
return m->full_name;
|
||||
}
|
||||
|
||||
const upb_FileDef* upb_MessageDef_File(const upb_MessageDef* m) {
|
||||
return m->file;
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_MessageDef_ContainingType(const upb_MessageDef* m) {
|
||||
return m->containing_type;
|
||||
}
|
||||
|
||||
const char* upb_MessageDef_Name(const upb_MessageDef* m) {
|
||||
return _upb_DefBuilder_FullToShort(m->full_name);
|
||||
}
|
||||
|
||||
upb_Syntax upb_MessageDef_Syntax(const upb_MessageDef* m) {
|
||||
return upb_FileDef_Syntax(m->file);
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_MessageDef_FindFieldByNumber(const upb_MessageDef* m,
|
||||
uint32_t i) {
|
||||
upb_value val;
|
||||
return upb_inttable_lookup(&m->itof, i, &val) ? upb_value_getconstptr(val)
|
||||
: NULL;
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_MessageDef_FindFieldByNameWithSize(
|
||||
const upb_MessageDef* m, const char* name, size_t size) {
|
||||
upb_value val;
|
||||
|
||||
if (!upb_strtable_lookup2(&m->ntof, name, size, &val)) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return _upb_DefType_Unpack(val, UPB_DEFTYPE_FIELD);
|
||||
}
|
||||
|
||||
const upb_OneofDef* upb_MessageDef_FindOneofByNameWithSize(
|
||||
const upb_MessageDef* m, const char* name, size_t size) {
|
||||
upb_value val;
|
||||
|
||||
if (!upb_strtable_lookup2(&m->ntof, name, size, &val)) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return _upb_DefType_Unpack(val, UPB_DEFTYPE_ONEOF);
|
||||
}
|
||||
|
||||
bool _upb_MessageDef_Insert(upb_MessageDef* m, const char* name, size_t len,
|
||||
upb_value v, upb_Arena* a) {
|
||||
return upb_strtable_insert(&m->ntof, name, len, v, a);
|
||||
}
|
||||
|
||||
bool upb_MessageDef_FindByNameWithSize(const upb_MessageDef* m,
|
||||
const char* name, size_t len,
|
||||
const upb_FieldDef** out_f,
|
||||
const upb_OneofDef** out_o) {
|
||||
upb_value val;
|
||||
|
||||
if (!upb_strtable_lookup2(&m->ntof, name, len, &val)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
const upb_FieldDef* f = _upb_DefType_Unpack(val, UPB_DEFTYPE_FIELD);
|
||||
const upb_OneofDef* o = _upb_DefType_Unpack(val, UPB_DEFTYPE_ONEOF);
|
||||
if (out_f) *out_f = f;
|
||||
if (out_o) *out_o = o;
|
||||
return f || o; /* False if this was a JSON name. */
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_MessageDef_FindByJsonNameWithSize(
|
||||
const upb_MessageDef* m, const char* name, size_t size) {
|
||||
upb_value val;
|
||||
|
||||
if (upb_strtable_lookup2(&m->jtof, name, size, &val)) {
|
||||
return upb_value_getconstptr(val);
|
||||
}
|
||||
|
||||
if (!upb_strtable_lookup2(&m->ntof, name, size, &val)) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return _upb_DefType_Unpack(val, UPB_DEFTYPE_FIELD);
|
||||
}
|
||||
|
||||
int upb_MessageDef_ExtensionRangeCount(const upb_MessageDef* m) {
|
||||
return m->ext_range_count;
|
||||
}
|
||||
|
||||
int upb_MessageDef_ReservedRangeCount(const upb_MessageDef* m) {
|
||||
return m->res_range_count;
|
||||
}
|
||||
|
||||
int upb_MessageDef_ReservedNameCount(const upb_MessageDef* m) {
|
||||
return m->res_name_count;
|
||||
}
|
||||
|
||||
int upb_MessageDef_FieldCount(const upb_MessageDef* m) {
|
||||
return m->field_count;
|
||||
}
|
||||
|
||||
int upb_MessageDef_OneofCount(const upb_MessageDef* m) {
|
||||
return m->oneof_count;
|
||||
}
|
||||
|
||||
int upb_MessageDef_RealOneofCount(const upb_MessageDef* m) {
|
||||
return m->real_oneof_count;
|
||||
}
|
||||
|
||||
int upb_MessageDef_NestedMessageCount(const upb_MessageDef* m) {
|
||||
return m->nested_msg_count;
|
||||
}
|
||||
|
||||
int upb_MessageDef_NestedEnumCount(const upb_MessageDef* m) {
|
||||
return m->nested_enum_count;
|
||||
}
|
||||
|
||||
int upb_MessageDef_NestedExtensionCount(const upb_MessageDef* m) {
|
||||
return m->nested_ext_count;
|
||||
}
|
||||
|
||||
const upb_MiniTable* upb_MessageDef_MiniTable(const upb_MessageDef* m) {
|
||||
return m->layout;
|
||||
}
|
||||
|
||||
const upb_ExtensionRange* upb_MessageDef_ExtensionRange(const upb_MessageDef* m,
|
||||
int i) {
|
||||
UPB_ASSERT(0 <= i && i < m->ext_range_count);
|
||||
return _upb_ExtensionRange_At(m->ext_ranges, i);
|
||||
}
|
||||
|
||||
const upb_MessageReservedRange* upb_MessageDef_ReservedRange(
|
||||
const upb_MessageDef* m, int i) {
|
||||
UPB_ASSERT(0 <= i && i < m->res_range_count);
|
||||
return _upb_MessageReservedRange_At(m->res_ranges, i);
|
||||
}
|
||||
|
||||
upb_StringView upb_MessageDef_ReservedName(const upb_MessageDef* m, int i) {
|
||||
UPB_ASSERT(0 <= i && i < m->res_name_count);
|
||||
return m->res_names[i];
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_MessageDef_Field(const upb_MessageDef* m, int i) {
|
||||
UPB_ASSERT(0 <= i && i < m->field_count);
|
||||
return _upb_FieldDef_At(m->fields, i);
|
||||
}
|
||||
|
||||
const upb_OneofDef* upb_MessageDef_Oneof(const upb_MessageDef* m, int i) {
|
||||
UPB_ASSERT(0 <= i && i < m->oneof_count);
|
||||
return _upb_OneofDef_At(m->oneofs, i);
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_MessageDef_NestedMessage(const upb_MessageDef* m,
|
||||
int i) {
|
||||
UPB_ASSERT(0 <= i && i < m->nested_msg_count);
|
||||
return &m->nested_msgs[i];
|
||||
}
|
||||
|
||||
const upb_EnumDef* upb_MessageDef_NestedEnum(const upb_MessageDef* m, int i) {
|
||||
UPB_ASSERT(0 <= i && i < m->nested_enum_count);
|
||||
return _upb_EnumDef_At(m->nested_enums, i);
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_MessageDef_NestedExtension(const upb_MessageDef* m,
|
||||
int i) {
|
||||
UPB_ASSERT(0 <= i && i < m->nested_ext_count);
|
||||
return _upb_FieldDef_At(m->nested_exts, i);
|
||||
}
|
||||
|
||||
upb_WellKnown upb_MessageDef_WellKnownType(const upb_MessageDef* m) {
|
||||
return m->well_known_type;
|
||||
}
|
||||
|
||||
bool _upb_MessageDef_InMessageSet(const upb_MessageDef* m) {
|
||||
return m->in_message_set;
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_MessageDef_FindFieldByName(const upb_MessageDef* m,
|
||||
const char* name) {
|
||||
return upb_MessageDef_FindFieldByNameWithSize(m, name, strlen(name));
|
||||
}
|
||||
|
||||
const upb_OneofDef* upb_MessageDef_FindOneofByName(const upb_MessageDef* m,
|
||||
const char* name) {
|
||||
return upb_MessageDef_FindOneofByNameWithSize(m, name, strlen(name));
|
||||
}
|
||||
|
||||
bool upb_MessageDef_IsMapEntry(const upb_MessageDef* m) {
|
||||
return UPB_DESC(MessageOptions_map_entry)(m->opts);
|
||||
}
|
||||
|
||||
bool upb_MessageDef_IsMessageSet(const upb_MessageDef* m) {
|
||||
return UPB_DESC(MessageOptions_message_set_wire_format)(m->opts);
|
||||
}
|
||||
|
||||
static upb_MiniTable* _upb_MessageDef_MakeMiniTable(upb_DefBuilder* ctx,
|
||||
const upb_MessageDef* m) {
|
||||
upb_StringView desc;
|
||||
// Note: this will assign layout_index for fields, so upb_FieldDef_MiniTable()
|
||||
// is safe to call only after this call.
|
||||
bool ok = upb_MessageDef_MiniDescriptorEncode(m, ctx->tmp_arena, &desc);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
void** scratch_data = _upb_DefPool_ScratchData(ctx->symtab);
|
||||
size_t* scratch_size = _upb_DefPool_ScratchSize(ctx->symtab);
|
||||
upb_MiniTable* ret = upb_MiniTable_BuildWithBuf(
|
||||
desc.data, desc.size, ctx->platform, ctx->arena, scratch_data,
|
||||
scratch_size, ctx->status);
|
||||
if (!ret) _upb_DefBuilder_FailJmp(ctx);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void _upb_MessageDef_Resolve(upb_DefBuilder* ctx, upb_MessageDef* m) {
|
||||
for (int i = 0; i < m->field_count; i++) {
|
||||
upb_FieldDef* f = (upb_FieldDef*)upb_MessageDef_Field(m, i);
|
||||
_upb_FieldDef_Resolve(ctx, m->full_name, f);
|
||||
}
|
||||
|
||||
m->in_message_set = false;
|
||||
for (int i = 0; i < upb_MessageDef_NestedExtensionCount(m); i++) {
|
||||
upb_FieldDef* ext = (upb_FieldDef*)upb_MessageDef_NestedExtension(m, i);
|
||||
_upb_FieldDef_Resolve(ctx, m->full_name, ext);
|
||||
if (upb_FieldDef_Type(ext) == kUpb_FieldType_Message &&
|
||||
upb_FieldDef_Label(ext) == kUpb_Label_Optional &&
|
||||
upb_FieldDef_MessageSubDef(ext) == m &&
|
||||
UPB_DESC(MessageOptions_message_set_wire_format)(
|
||||
upb_MessageDef_Options(upb_FieldDef_ContainingType(ext)))) {
|
||||
m->in_message_set = true;
|
||||
}
|
||||
}
|
||||
|
||||
for (int i = 0; i < upb_MessageDef_NestedMessageCount(m); i++) {
|
||||
upb_MessageDef* n = (upb_MessageDef*)upb_MessageDef_NestedMessage(m, i);
|
||||
_upb_MessageDef_Resolve(ctx, n);
|
||||
}
|
||||
}
|
||||
|
||||
void _upb_MessageDef_InsertField(upb_DefBuilder* ctx, upb_MessageDef* m,
|
||||
const upb_FieldDef* f) {
|
||||
const int32_t field_number = upb_FieldDef_Number(f);
|
||||
|
||||
if (field_number <= 0 || field_number > kUpb_MaxFieldNumber) {
|
||||
_upb_DefBuilder_Errf(ctx, "invalid field number (%u)", field_number);
|
||||
}
|
||||
|
||||
const char* json_name = upb_FieldDef_JsonName(f);
|
||||
const char* shortname = upb_FieldDef_Name(f);
|
||||
const size_t shortnamelen = strlen(shortname);
|
||||
|
||||
upb_value v = upb_value_constptr(f);
|
||||
|
||||
upb_value existing_v;
|
||||
if (upb_strtable_lookup(&m->ntof, shortname, &existing_v)) {
|
||||
_upb_DefBuilder_Errf(ctx, "duplicate field name (%s)", shortname);
|
||||
}
|
||||
|
||||
const upb_value field_v = _upb_DefType_Pack(f, UPB_DEFTYPE_FIELD);
|
||||
bool ok =
|
||||
_upb_MessageDef_Insert(m, shortname, shortnamelen, field_v, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
// TODO: Once editions is supported this should turn into a
|
||||
// check on LEGACY_BEST_EFFORT
|
||||
if (strcmp(shortname, json_name) != 0 &&
|
||||
upb_FileDef_Syntax(m->file) == kUpb_Syntax_Proto3 &&
|
||||
upb_strtable_lookup(&m->ntof, json_name, &v)) {
|
||||
_upb_DefBuilder_Errf(
|
||||
ctx, "duplicate json_name for (%s) with original field name (%s)",
|
||||
shortname, json_name);
|
||||
}
|
||||
|
||||
if (upb_strtable_lookup(&m->jtof, json_name, &v)) {
|
||||
_upb_DefBuilder_Errf(ctx, "duplicate json_name (%s)", json_name);
|
||||
}
|
||||
|
||||
const size_t json_size = strlen(json_name);
|
||||
ok = upb_strtable_insert(&m->jtof, json_name, json_size,
|
||||
upb_value_constptr(f), ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
if (upb_inttable_lookup(&m->itof, field_number, NULL)) {
|
||||
_upb_DefBuilder_Errf(ctx, "duplicate field number (%u)", field_number);
|
||||
}
|
||||
|
||||
ok = upb_inttable_insert(&m->itof, field_number, v, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
}
|
||||
|
||||
void _upb_MessageDef_CreateMiniTable(upb_DefBuilder* ctx, upb_MessageDef* m) {
|
||||
if (ctx->layout == NULL) {
|
||||
m->layout = _upb_MessageDef_MakeMiniTable(ctx, m);
|
||||
} else {
|
||||
UPB_ASSERT(ctx->msg_count < ctx->layout->msg_count);
|
||||
m->layout = ctx->layout->msgs[ctx->msg_count++];
|
||||
UPB_ASSERT(m->field_count == m->layout->field_count);
|
||||
|
||||
// We don't need the result of this call, but it will assign layout_index
|
||||
// for all the fields in O(n lg n) time.
|
||||
_upb_FieldDefs_Sorted(m->fields, m->field_count, ctx->tmp_arena);
|
||||
}
|
||||
|
||||
for (int i = 0; i < m->nested_msg_count; i++) {
|
||||
upb_MessageDef* nested =
|
||||
(upb_MessageDef*)upb_MessageDef_NestedMessage(m, i);
|
||||
_upb_MessageDef_CreateMiniTable(ctx, nested);
|
||||
}
|
||||
}
|
||||
|
||||
void _upb_MessageDef_LinkMiniTable(upb_DefBuilder* ctx,
|
||||
const upb_MessageDef* m) {
|
||||
for (int i = 0; i < upb_MessageDef_NestedExtensionCount(m); i++) {
|
||||
const upb_FieldDef* ext = upb_MessageDef_NestedExtension(m, i);
|
||||
_upb_FieldDef_BuildMiniTableExtension(ctx, ext);
|
||||
}
|
||||
|
||||
for (int i = 0; i < m->nested_msg_count; i++) {
|
||||
_upb_MessageDef_LinkMiniTable(ctx, upb_MessageDef_NestedMessage(m, i));
|
||||
}
|
||||
|
||||
if (ctx->layout) return;
|
||||
|
||||
for (int i = 0; i < m->field_count; i++) {
|
||||
const upb_FieldDef* f = upb_MessageDef_Field(m, i);
|
||||
const upb_MessageDef* sub_m = upb_FieldDef_MessageSubDef(f);
|
||||
const upb_EnumDef* sub_e = upb_FieldDef_EnumSubDef(f);
|
||||
const int layout_index = _upb_FieldDef_LayoutIndex(f);
|
||||
upb_MiniTable* mt = (upb_MiniTable*)upb_MessageDef_MiniTable(m);
|
||||
|
||||
UPB_ASSERT(layout_index < m->field_count);
|
||||
upb_MiniTableField* mt_f =
|
||||
(upb_MiniTableField*)&m->layout->fields[layout_index];
|
||||
if (sub_m) {
|
||||
if (!mt->subs) {
|
||||
_upb_DefBuilder_Errf(ctx, "unexpected submsg for (%s)", m->full_name);
|
||||
}
|
||||
UPB_ASSERT(mt_f);
|
||||
UPB_ASSERT(sub_m->layout);
|
||||
if (UPB_UNLIKELY(!upb_MiniTable_SetSubMessage(mt, mt_f, sub_m->layout))) {
|
||||
_upb_DefBuilder_Errf(ctx, "invalid submsg for (%s)", m->full_name);
|
||||
}
|
||||
} else if (_upb_FieldDef_IsClosedEnum(f)) {
|
||||
const upb_MiniTableEnum* mt_e = _upb_EnumDef_MiniTable(sub_e);
|
||||
if (UPB_UNLIKELY(!upb_MiniTable_SetSubEnum(mt, mt_f, mt_e))) {
|
||||
_upb_DefBuilder_Errf(ctx, "invalid subenum for (%s)", m->full_name);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#ifndef NDEBUG
|
||||
for (int i = 0; i < m->field_count; i++) {
|
||||
const upb_FieldDef* f = upb_MessageDef_Field(m, i);
|
||||
const int layout_index = _upb_FieldDef_LayoutIndex(f);
|
||||
UPB_ASSERT(layout_index < m->layout->field_count);
|
||||
const upb_MiniTableField* mt_f = &m->layout->fields[layout_index];
|
||||
UPB_ASSERT(upb_FieldDef_Type(f) == upb_MiniTableField_Type(mt_f));
|
||||
UPB_ASSERT(upb_FieldDef_CType(f) == upb_MiniTableField_CType(mt_f));
|
||||
UPB_ASSERT(upb_FieldDef_HasPresence(f) ==
|
||||
upb_MiniTableField_HasPresence(mt_f));
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
static bool _upb_MessageDef_ValidateUtf8(const upb_MessageDef* m) {
|
||||
bool has_string = false;
|
||||
for (int i = 0; i < m->field_count; i++) {
|
||||
const upb_FieldDef* f = upb_MessageDef_Field(m, i);
|
||||
// Old binaries do not recognize the field-level "FlipValidateUtf8" wire
|
||||
// modifier, so we do not actually have field-level control for old
|
||||
// binaries. Given this, we judge that the better failure mode is to be
|
||||
// more lax than intended, rather than more strict. To achieve this, we
|
||||
// only mark the message with the ValidateUtf8 modifier if *all* fields
|
||||
// validate UTF-8.
|
||||
if (!_upb_FieldDef_ValidateUtf8(f)) return false;
|
||||
if (upb_FieldDef_Type(f) == kUpb_FieldType_String) has_string = true;
|
||||
}
|
||||
return has_string;
|
||||
}
|
||||
|
||||
static uint64_t _upb_MessageDef_Modifiers(const upb_MessageDef* m) {
|
||||
uint64_t out = 0;
|
||||
|
||||
if (upb_FileDef_Syntax(m->file) == kUpb_Syntax_Proto3) {
|
||||
out |= kUpb_MessageModifier_DefaultIsPacked;
|
||||
}
|
||||
|
||||
if (_upb_MessageDef_ValidateUtf8(m)) {
|
||||
out |= kUpb_MessageModifier_ValidateUtf8;
|
||||
}
|
||||
|
||||
if (m->ext_range_count) {
|
||||
out |= kUpb_MessageModifier_IsExtendable;
|
||||
}
|
||||
|
||||
return out;
|
||||
}
|
||||
|
||||
static bool _upb_MessageDef_EncodeMap(upb_DescState* s, const upb_MessageDef* m,
|
||||
upb_Arena* a) {
|
||||
if (m->field_count != 2) return false;
|
||||
|
||||
const upb_FieldDef* key_field = upb_MessageDef_Field(m, 0);
|
||||
const upb_FieldDef* val_field = upb_MessageDef_Field(m, 1);
|
||||
if (key_field == NULL || val_field == NULL) return false;
|
||||
|
||||
UPB_ASSERT(_upb_FieldDef_LayoutIndex(key_field) == 0);
|
||||
UPB_ASSERT(_upb_FieldDef_LayoutIndex(val_field) == 1);
|
||||
|
||||
s->ptr = upb_MtDataEncoder_EncodeMap(
|
||||
&s->e, s->ptr, upb_FieldDef_Type(key_field), upb_FieldDef_Type(val_field),
|
||||
_upb_FieldDef_Modifiers(key_field), _upb_FieldDef_Modifiers(val_field));
|
||||
return true;
|
||||
}
|
||||
|
||||
static bool _upb_MessageDef_EncodeMessage(upb_DescState* s,
|
||||
const upb_MessageDef* m,
|
||||
upb_Arena* a) {
|
||||
const upb_FieldDef** sorted = NULL;
|
||||
if (!m->is_sorted) {
|
||||
sorted = _upb_FieldDefs_Sorted(m->fields, m->field_count, a);
|
||||
if (!sorted) return false;
|
||||
}
|
||||
|
||||
s->ptr = upb_MtDataEncoder_StartMessage(&s->e, s->ptr,
|
||||
_upb_MessageDef_Modifiers(m));
|
||||
|
||||
for (int i = 0; i < m->field_count; i++) {
|
||||
const upb_FieldDef* f = sorted ? sorted[i] : upb_MessageDef_Field(m, i);
|
||||
const upb_FieldType type = upb_FieldDef_Type(f);
|
||||
const int number = upb_FieldDef_Number(f);
|
||||
const uint64_t modifiers = _upb_FieldDef_Modifiers(f);
|
||||
|
||||
if (!_upb_DescState_Grow(s, a)) return false;
|
||||
s->ptr = upb_MtDataEncoder_PutField(&s->e, s->ptr, type, number, modifiers);
|
||||
}
|
||||
|
||||
for (int i = 0; i < m->real_oneof_count; i++) {
|
||||
if (!_upb_DescState_Grow(s, a)) return false;
|
||||
s->ptr = upb_MtDataEncoder_StartOneof(&s->e, s->ptr);
|
||||
|
||||
const upb_OneofDef* o = upb_MessageDef_Oneof(m, i);
|
||||
const int field_count = upb_OneofDef_FieldCount(o);
|
||||
for (int j = 0; j < field_count; j++) {
|
||||
const int number = upb_FieldDef_Number(upb_OneofDef_Field(o, j));
|
||||
|
||||
if (!_upb_DescState_Grow(s, a)) return false;
|
||||
s->ptr = upb_MtDataEncoder_PutOneofField(&s->e, s->ptr, number);
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
static bool _upb_MessageDef_EncodeMessageSet(upb_DescState* s,
|
||||
const upb_MessageDef* m,
|
||||
upb_Arena* a) {
|
||||
s->ptr = upb_MtDataEncoder_EncodeMessageSet(&s->e, s->ptr);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool upb_MessageDef_MiniDescriptorEncode(const upb_MessageDef* m, upb_Arena* a,
|
||||
upb_StringView* out) {
|
||||
upb_DescState s;
|
||||
_upb_DescState_Init(&s);
|
||||
|
||||
if (!_upb_DescState_Grow(&s, a)) return false;
|
||||
|
||||
if (upb_MessageDef_IsMapEntry(m)) {
|
||||
if (!_upb_MessageDef_EncodeMap(&s, m, a)) return false;
|
||||
} else if (UPB_DESC(MessageOptions_message_set_wire_format)(m->opts)) {
|
||||
if (!_upb_MessageDef_EncodeMessageSet(&s, m, a)) return false;
|
||||
} else {
|
||||
if (!_upb_MessageDef_EncodeMessage(&s, m, a)) return false;
|
||||
}
|
||||
|
||||
if (!_upb_DescState_Grow(&s, a)) return false;
|
||||
*s.ptr = '\0';
|
||||
|
||||
out->data = s.buf;
|
||||
out->size = s.ptr - s.buf;
|
||||
return true;
|
||||
}
|
||||
|
||||
static upb_StringView* _upb_ReservedNames_New(upb_DefBuilder* ctx, int n,
|
||||
const upb_StringView* protos) {
|
||||
upb_StringView* sv = _upb_DefBuilder_Alloc(ctx, sizeof(upb_StringView) * n);
|
||||
for (int i = 0; i < n; i++) {
|
||||
sv[i].data =
|
||||
upb_strdup2(protos[i].data, protos[i].size, _upb_DefBuilder_Arena(ctx));
|
||||
sv[i].size = protos[i].size;
|
||||
}
|
||||
return sv;
|
||||
}
|
||||
|
||||
static void create_msgdef(upb_DefBuilder* ctx, const char* prefix,
|
||||
const UPB_DESC(DescriptorProto) * msg_proto,
|
||||
const upb_MessageDef* containing_type,
|
||||
upb_MessageDef* m) {
|
||||
const UPB_DESC(OneofDescriptorProto)* const* oneofs;
|
||||
const UPB_DESC(FieldDescriptorProto)* const* fields;
|
||||
const UPB_DESC(DescriptorProto_ExtensionRange)* const* ext_ranges;
|
||||
const UPB_DESC(DescriptorProto_ReservedRange)* const* res_ranges;
|
||||
const upb_StringView* res_names;
|
||||
size_t n_oneof, n_field, n_enum, n_ext, n_msg;
|
||||
size_t n_ext_range, n_res_range, n_res_name;
|
||||
upb_StringView name;
|
||||
|
||||
// Must happen before _upb_DefBuilder_Add()
|
||||
m->file = _upb_DefBuilder_File(ctx);
|
||||
|
||||
m->containing_type = containing_type;
|
||||
m->is_sorted = true;
|
||||
|
||||
name = UPB_DESC(DescriptorProto_name)(msg_proto);
|
||||
|
||||
m->full_name = _upb_DefBuilder_MakeFullName(ctx, prefix, name);
|
||||
_upb_DefBuilder_Add(ctx, m->full_name, _upb_DefType_Pack(m, UPB_DEFTYPE_MSG));
|
||||
|
||||
oneofs = UPB_DESC(DescriptorProto_oneof_decl)(msg_proto, &n_oneof);
|
||||
fields = UPB_DESC(DescriptorProto_field)(msg_proto, &n_field);
|
||||
ext_ranges =
|
||||
UPB_DESC(DescriptorProto_extension_range)(msg_proto, &n_ext_range);
|
||||
res_ranges =
|
||||
UPB_DESC(DescriptorProto_reserved_range)(msg_proto, &n_res_range);
|
||||
res_names = UPB_DESC(DescriptorProto_reserved_name)(msg_proto, &n_res_name);
|
||||
|
||||
bool ok = upb_inttable_init(&m->itof, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
ok = upb_strtable_init(&m->ntof, n_oneof + n_field, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
ok = upb_strtable_init(&m->jtof, n_field, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
UPB_DEF_SET_OPTIONS(m->opts, DescriptorProto, MessageOptions, msg_proto);
|
||||
|
||||
m->oneof_count = n_oneof;
|
||||
m->oneofs = _upb_OneofDefs_New(ctx, n_oneof, oneofs, m);
|
||||
|
||||
m->field_count = n_field;
|
||||
m->fields =
|
||||
_upb_FieldDefs_New(ctx, n_field, fields, m->full_name, m, &m->is_sorted);
|
||||
|
||||
// Message Sets may not contain fields.
|
||||
if (UPB_UNLIKELY(UPB_DESC(MessageOptions_message_set_wire_format)(m->opts))) {
|
||||
if (UPB_UNLIKELY(n_field > 0)) {
|
||||
_upb_DefBuilder_Errf(ctx, "invalid message set (%s)", m->full_name);
|
||||
}
|
||||
}
|
||||
|
||||
m->ext_range_count = n_ext_range;
|
||||
m->ext_ranges = _upb_ExtensionRanges_New(ctx, n_ext_range, ext_ranges, m);
|
||||
|
||||
m->res_range_count = n_res_range;
|
||||
m->res_ranges =
|
||||
_upb_MessageReservedRanges_New(ctx, n_res_range, res_ranges, m);
|
||||
|
||||
m->res_name_count = n_res_name;
|
||||
m->res_names = _upb_ReservedNames_New(ctx, n_res_name, res_names);
|
||||
|
||||
const size_t synthetic_count = _upb_OneofDefs_Finalize(ctx, m);
|
||||
m->real_oneof_count = m->oneof_count - synthetic_count;
|
||||
|
||||
assign_msg_wellknowntype(m);
|
||||
upb_inttable_compact(&m->itof, ctx->arena);
|
||||
|
||||
const UPB_DESC(EnumDescriptorProto)* const* enums =
|
||||
UPB_DESC(DescriptorProto_enum_type)(msg_proto, &n_enum);
|
||||
m->nested_enum_count = n_enum;
|
||||
m->nested_enums = _upb_EnumDefs_New(ctx, n_enum, enums, m);
|
||||
|
||||
const UPB_DESC(FieldDescriptorProto)* const* exts =
|
||||
UPB_DESC(DescriptorProto_extension)(msg_proto, &n_ext);
|
||||
m->nested_ext_count = n_ext;
|
||||
m->nested_exts = _upb_Extensions_New(ctx, n_ext, exts, m->full_name, m);
|
||||
|
||||
const UPB_DESC(DescriptorProto)* const* msgs =
|
||||
UPB_DESC(DescriptorProto_nested_type)(msg_proto, &n_msg);
|
||||
m->nested_msg_count = n_msg;
|
||||
m->nested_msgs = _upb_MessageDefs_New(ctx, n_msg, msgs, m);
|
||||
}
|
||||
|
||||
// Allocate and initialize an array of |n| message defs.
|
||||
upb_MessageDef* _upb_MessageDefs_New(
|
||||
upb_DefBuilder* ctx, int n, const UPB_DESC(DescriptorProto) * const* protos,
|
||||
const upb_MessageDef* containing_type) {
|
||||
_upb_DefType_CheckPadding(sizeof(upb_MessageDef));
|
||||
|
||||
const char* name = containing_type ? containing_type->full_name
|
||||
: _upb_FileDef_RawPackage(ctx->file);
|
||||
|
||||
upb_MessageDef* m = _upb_DefBuilder_Alloc(ctx, sizeof(upb_MessageDef) * n);
|
||||
for (int i = 0; i < n; i++) {
|
||||
create_msgdef(ctx, name, protos[i], containing_type, &m[i]);
|
||||
}
|
||||
return m;
|
||||
}
|
||||
156
Pods/gRPC-Core/third_party/upb/upb/reflection/message_def.h
generated
vendored
Normal file
156
Pods/gRPC-Core/third_party/upb/upb/reflection/message_def.h
generated
vendored
Normal file
@@ -0,0 +1,156 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_MESSAGE_DEF_H_
|
||||
#define UPB_REFLECTION_MESSAGE_DEF_H_
|
||||
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// Well-known field tag numbers for map-entry messages.
|
||||
#define kUpb_MapEntry_KeyFieldNumber 1
|
||||
#define kUpb_MapEntry_ValueFieldNumber 2
|
||||
|
||||
// Well-known field tag numbers for Any messages.
|
||||
#define kUpb_Any_TypeFieldNumber 1
|
||||
#define kUpb_Any_ValueFieldNumber 2
|
||||
|
||||
// Well-known field tag numbers for duration messages.
|
||||
#define kUpb_Duration_SecondsFieldNumber 1
|
||||
#define kUpb_Duration_NanosFieldNumber 2
|
||||
|
||||
// Well-known field tag numbers for timestamp messages.
|
||||
#define kUpb_Timestamp_SecondsFieldNumber 1
|
||||
#define kUpb_Timestamp_NanosFieldNumber 2
|
||||
|
||||
// All the different kind of well known type messages. For simplicity of check,
|
||||
// number wrappers and string wrappers are grouped together. Make sure the
|
||||
// order and number of these groups are not changed.
|
||||
typedef enum {
|
||||
kUpb_WellKnown_Unspecified,
|
||||
kUpb_WellKnown_Any,
|
||||
kUpb_WellKnown_FieldMask,
|
||||
kUpb_WellKnown_Duration,
|
||||
kUpb_WellKnown_Timestamp,
|
||||
|
||||
// number wrappers
|
||||
kUpb_WellKnown_DoubleValue,
|
||||
kUpb_WellKnown_FloatValue,
|
||||
kUpb_WellKnown_Int64Value,
|
||||
kUpb_WellKnown_UInt64Value,
|
||||
kUpb_WellKnown_Int32Value,
|
||||
kUpb_WellKnown_UInt32Value,
|
||||
|
||||
// string wrappers
|
||||
kUpb_WellKnown_StringValue,
|
||||
kUpb_WellKnown_BytesValue,
|
||||
kUpb_WellKnown_BoolValue,
|
||||
kUpb_WellKnown_Value,
|
||||
kUpb_WellKnown_ListValue,
|
||||
kUpb_WellKnown_Struct,
|
||||
} upb_WellKnown;
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
const upb_MessageDef* upb_MessageDef_ContainingType(const upb_MessageDef* m);
|
||||
|
||||
const upb_ExtensionRange* upb_MessageDef_ExtensionRange(const upb_MessageDef* m,
|
||||
int i);
|
||||
int upb_MessageDef_ExtensionRangeCount(const upb_MessageDef* m);
|
||||
|
||||
UPB_API const upb_FieldDef* upb_MessageDef_Field(const upb_MessageDef* m,
|
||||
int i);
|
||||
UPB_API int upb_MessageDef_FieldCount(const upb_MessageDef* m);
|
||||
|
||||
UPB_API const upb_FileDef* upb_MessageDef_File(const upb_MessageDef* m);
|
||||
|
||||
// Returns a field by either JSON name or regular proto name.
|
||||
const upb_FieldDef* upb_MessageDef_FindByJsonNameWithSize(
|
||||
const upb_MessageDef* m, const char* name, size_t size);
|
||||
UPB_INLINE const upb_FieldDef* upb_MessageDef_FindByJsonName(
|
||||
const upb_MessageDef* m, const char* name) {
|
||||
return upb_MessageDef_FindByJsonNameWithSize(m, name, strlen(name));
|
||||
}
|
||||
|
||||
// Lookup of either field or oneof by name. Returns whether either was found.
|
||||
// If the return is true, then the found def will be set, and the non-found
|
||||
// one set to NULL.
|
||||
UPB_API bool upb_MessageDef_FindByNameWithSize(const upb_MessageDef* m,
|
||||
const char* name, size_t size,
|
||||
const upb_FieldDef** f,
|
||||
const upb_OneofDef** o);
|
||||
UPB_INLINE bool upb_MessageDef_FindByName(const upb_MessageDef* m,
|
||||
const char* name,
|
||||
const upb_FieldDef** f,
|
||||
const upb_OneofDef** o) {
|
||||
return upb_MessageDef_FindByNameWithSize(m, name, strlen(name), f, o);
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_MessageDef_FindFieldByName(const upb_MessageDef* m,
|
||||
const char* name);
|
||||
UPB_API const upb_FieldDef* upb_MessageDef_FindFieldByNameWithSize(
|
||||
const upb_MessageDef* m, const char* name, size_t size);
|
||||
UPB_API const upb_FieldDef* upb_MessageDef_FindFieldByNumber(
|
||||
const upb_MessageDef* m, uint32_t i);
|
||||
const upb_OneofDef* upb_MessageDef_FindOneofByName(const upb_MessageDef* m,
|
||||
const char* name);
|
||||
UPB_API const upb_OneofDef* upb_MessageDef_FindOneofByNameWithSize(
|
||||
const upb_MessageDef* m, const char* name, size_t size);
|
||||
UPB_API const char* upb_MessageDef_FullName(const upb_MessageDef* m);
|
||||
bool upb_MessageDef_HasOptions(const upb_MessageDef* m);
|
||||
bool upb_MessageDef_IsMapEntry(const upb_MessageDef* m);
|
||||
bool upb_MessageDef_IsMessageSet(const upb_MessageDef* m);
|
||||
|
||||
// Creates a mini descriptor string for a message, returns true on success.
|
||||
bool upb_MessageDef_MiniDescriptorEncode(const upb_MessageDef* m, upb_Arena* a,
|
||||
upb_StringView* out);
|
||||
|
||||
UPB_API const upb_MiniTable* upb_MessageDef_MiniTable(const upb_MessageDef* m);
|
||||
const char* upb_MessageDef_Name(const upb_MessageDef* m);
|
||||
|
||||
const upb_EnumDef* upb_MessageDef_NestedEnum(const upb_MessageDef* m, int i);
|
||||
const upb_FieldDef* upb_MessageDef_NestedExtension(const upb_MessageDef* m,
|
||||
int i);
|
||||
const upb_MessageDef* upb_MessageDef_NestedMessage(const upb_MessageDef* m,
|
||||
int i);
|
||||
|
||||
int upb_MessageDef_NestedEnumCount(const upb_MessageDef* m);
|
||||
int upb_MessageDef_NestedExtensionCount(const upb_MessageDef* m);
|
||||
int upb_MessageDef_NestedMessageCount(const upb_MessageDef* m);
|
||||
|
||||
UPB_API const upb_OneofDef* upb_MessageDef_Oneof(const upb_MessageDef* m,
|
||||
int i);
|
||||
UPB_API int upb_MessageDef_OneofCount(const upb_MessageDef* m);
|
||||
int upb_MessageDef_RealOneofCount(const upb_MessageDef* m);
|
||||
|
||||
const UPB_DESC(MessageOptions) *
|
||||
upb_MessageDef_Options(const upb_MessageDef* m);
|
||||
|
||||
upb_StringView upb_MessageDef_ReservedName(const upb_MessageDef* m, int i);
|
||||
int upb_MessageDef_ReservedNameCount(const upb_MessageDef* m);
|
||||
|
||||
const upb_MessageReservedRange* upb_MessageDef_ReservedRange(
|
||||
const upb_MessageDef* m, int i);
|
||||
int upb_MessageDef_ReservedRangeCount(const upb_MessageDef* m);
|
||||
|
||||
UPB_API upb_Syntax upb_MessageDef_Syntax(const upb_MessageDef* m);
|
||||
UPB_API upb_WellKnown upb_MessageDef_WellKnownType(const upb_MessageDef* m);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_MESSAGE_DEF_H_ */
|
||||
61
Pods/gRPC-Core/third_party/upb/upb/reflection/message_reserved_range.c
generated
vendored
Normal file
61
Pods/gRPC-Core/third_party/upb/upb/reflection/message_reserved_range.c
generated
vendored
Normal file
@@ -0,0 +1,61 @@
|
||||
// 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/reflection/enum_def.h"
|
||||
#include "upb/reflection/field_def.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/extension_range.h"
|
||||
#include "upb/reflection/message_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_MessageReservedRange {
|
||||
int32_t start;
|
||||
int32_t end;
|
||||
};
|
||||
|
||||
upb_MessageReservedRange* _upb_MessageReservedRange_At(
|
||||
const upb_MessageReservedRange* r, int i) {
|
||||
return (upb_MessageReservedRange*)&r[i];
|
||||
}
|
||||
|
||||
int32_t upb_MessageReservedRange_Start(const upb_MessageReservedRange* r) {
|
||||
return r->start;
|
||||
}
|
||||
int32_t upb_MessageReservedRange_End(const upb_MessageReservedRange* r) {
|
||||
return r->end;
|
||||
}
|
||||
|
||||
upb_MessageReservedRange* _upb_MessageReservedRanges_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(DescriptorProto_ReservedRange) * const* protos,
|
||||
const upb_MessageDef* m) {
|
||||
upb_MessageReservedRange* r =
|
||||
_upb_DefBuilder_Alloc(ctx, sizeof(upb_MessageReservedRange) * n);
|
||||
|
||||
for (int i = 0; i < n; i++) {
|
||||
const int32_t start =
|
||||
UPB_DESC(DescriptorProto_ReservedRange_start)(protos[i]);
|
||||
const int32_t end = UPB_DESC(DescriptorProto_ReservedRange_end)(protos[i]);
|
||||
const int32_t max = kUpb_MaxFieldNumber + 1;
|
||||
|
||||
// A full validation would also check that each range is disjoint, and that
|
||||
// none of the fields overlap with the extension ranges, but we are just
|
||||
// sanity checking here.
|
||||
if (start < 1 || end <= start || end > max) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"Reserved range (%d, %d) is invalid, message=%s\n",
|
||||
(int)start, (int)end, upb_MessageDef_FullName(m));
|
||||
}
|
||||
|
||||
r[i].start = start;
|
||||
r[i].end = end;
|
||||
}
|
||||
|
||||
return r;
|
||||
}
|
||||
31
Pods/gRPC-Core/third_party/upb/upb/reflection/message_reserved_range.h
generated
vendored
Normal file
31
Pods/gRPC-Core/third_party/upb/upb/reflection/message_reserved_range.h
generated
vendored
Normal file
@@ -0,0 +1,31 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_MESSAGE_RESERVED_RANGE_H_
|
||||
#define UPB_REFLECTION_MESSAGE_RESERVED_RANGE_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
int32_t upb_MessageReservedRange_Start(const upb_MessageReservedRange* r);
|
||||
int32_t upb_MessageReservedRange_End(const upb_MessageReservedRange* r);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_MESSAGE_RESERVED_RANGE_H_ */
|
||||
105
Pods/gRPC-Core/third_party/upb/upb/reflection/method_def.c
generated
vendored
Normal file
105
Pods/gRPC-Core/third_party/upb/upb/reflection/method_def.c
generated
vendored
Normal file
@@ -0,0 +1,105 @@
|
||||
// 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/reflection/internal/method_def.h"
|
||||
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/service_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_MethodDef {
|
||||
const UPB_DESC(MethodOptions) * opts;
|
||||
upb_ServiceDef* service;
|
||||
const char* full_name;
|
||||
const upb_MessageDef* input_type;
|
||||
const upb_MessageDef* output_type;
|
||||
int index;
|
||||
bool client_streaming;
|
||||
bool server_streaming;
|
||||
};
|
||||
|
||||
upb_MethodDef* _upb_MethodDef_At(const upb_MethodDef* m, int i) {
|
||||
return (upb_MethodDef*)&m[i];
|
||||
}
|
||||
|
||||
const upb_ServiceDef* upb_MethodDef_Service(const upb_MethodDef* m) {
|
||||
return m->service;
|
||||
}
|
||||
|
||||
const UPB_DESC(MethodOptions) * upb_MethodDef_Options(const upb_MethodDef* m) {
|
||||
return m->opts;
|
||||
}
|
||||
|
||||
bool upb_MethodDef_HasOptions(const upb_MethodDef* m) {
|
||||
return m->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
const char* upb_MethodDef_FullName(const upb_MethodDef* m) {
|
||||
return m->full_name;
|
||||
}
|
||||
|
||||
const char* upb_MethodDef_Name(const upb_MethodDef* m) {
|
||||
return _upb_DefBuilder_FullToShort(m->full_name);
|
||||
}
|
||||
|
||||
int upb_MethodDef_Index(const upb_MethodDef* m) { return m->index; }
|
||||
|
||||
const upb_MessageDef* upb_MethodDef_InputType(const upb_MethodDef* m) {
|
||||
return m->input_type;
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_MethodDef_OutputType(const upb_MethodDef* m) {
|
||||
return m->output_type;
|
||||
}
|
||||
|
||||
bool upb_MethodDef_ClientStreaming(const upb_MethodDef* m) {
|
||||
return m->client_streaming;
|
||||
}
|
||||
|
||||
bool upb_MethodDef_ServerStreaming(const upb_MethodDef* m) {
|
||||
return m->server_streaming;
|
||||
}
|
||||
|
||||
static void create_method(upb_DefBuilder* ctx,
|
||||
const UPB_DESC(MethodDescriptorProto) * method_proto,
|
||||
upb_ServiceDef* s, upb_MethodDef* m) {
|
||||
upb_StringView name = UPB_DESC(MethodDescriptorProto_name)(method_proto);
|
||||
|
||||
m->service = s;
|
||||
m->full_name =
|
||||
_upb_DefBuilder_MakeFullName(ctx, upb_ServiceDef_FullName(s), name);
|
||||
m->client_streaming =
|
||||
UPB_DESC(MethodDescriptorProto_client_streaming)(method_proto);
|
||||
m->server_streaming =
|
||||
UPB_DESC(MethodDescriptorProto_server_streaming)(method_proto);
|
||||
m->input_type = _upb_DefBuilder_Resolve(
|
||||
ctx, m->full_name, m->full_name,
|
||||
UPB_DESC(MethodDescriptorProto_input_type)(method_proto),
|
||||
UPB_DEFTYPE_MSG);
|
||||
m->output_type = _upb_DefBuilder_Resolve(
|
||||
ctx, m->full_name, m->full_name,
|
||||
UPB_DESC(MethodDescriptorProto_output_type)(method_proto),
|
||||
UPB_DEFTYPE_MSG);
|
||||
|
||||
UPB_DEF_SET_OPTIONS(m->opts, MethodDescriptorProto, MethodOptions,
|
||||
method_proto);
|
||||
}
|
||||
|
||||
// Allocate and initialize an array of |n| method defs belonging to |s|.
|
||||
upb_MethodDef* _upb_MethodDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(MethodDescriptorProto) * const* protos, upb_ServiceDef* s) {
|
||||
upb_MethodDef* m = _upb_DefBuilder_Alloc(ctx, sizeof(upb_MethodDef) * n);
|
||||
for (int i = 0; i < n; i++) {
|
||||
create_method(ctx, protos[i], s, &m[i]);
|
||||
m[i].index = i;
|
||||
}
|
||||
return m;
|
||||
}
|
||||
39
Pods/gRPC-Core/third_party/upb/upb/reflection/method_def.h
generated
vendored
Normal file
39
Pods/gRPC-Core/third_party/upb/upb/reflection/method_def.h
generated
vendored
Normal file
@@ -0,0 +1,39 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_METHOD_DEF_H_
|
||||
#define UPB_REFLECTION_METHOD_DEF_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
bool upb_MethodDef_ClientStreaming(const upb_MethodDef* m);
|
||||
const char* upb_MethodDef_FullName(const upb_MethodDef* m);
|
||||
bool upb_MethodDef_HasOptions(const upb_MethodDef* m);
|
||||
int upb_MethodDef_Index(const upb_MethodDef* m);
|
||||
const upb_MessageDef* upb_MethodDef_InputType(const upb_MethodDef* m);
|
||||
const char* upb_MethodDef_Name(const upb_MethodDef* m);
|
||||
const UPB_DESC(MethodOptions) * upb_MethodDef_Options(const upb_MethodDef* m);
|
||||
const upb_MessageDef* upb_MethodDef_OutputType(const upb_MethodDef* m);
|
||||
bool upb_MethodDef_ServerStreaming(const upb_MethodDef* m);
|
||||
const upb_ServiceDef* upb_MethodDef_Service(const upb_MethodDef* m);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_METHOD_DEF_H_ */
|
||||
208
Pods/gRPC-Core/third_party/upb/upb/reflection/oneof_def.c
generated
vendored
Normal file
208
Pods/gRPC-Core/third_party/upb/upb/reflection/oneof_def.c
generated
vendored
Normal file
@@ -0,0 +1,208 @@
|
||||
// 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/reflection/internal/oneof_def.h"
|
||||
|
||||
#include <ctype.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/hash/int_table.h"
|
||||
#include "upb/hash/str_table.h"
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/field_def.h"
|
||||
#include "upb/reflection/internal/message_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_OneofDef {
|
||||
const UPB_DESC(OneofOptions) * opts;
|
||||
const upb_MessageDef* parent;
|
||||
const char* full_name;
|
||||
int field_count;
|
||||
bool synthetic;
|
||||
const upb_FieldDef** fields;
|
||||
upb_strtable ntof; // lookup a field by name
|
||||
upb_inttable itof; // lookup a field by number (index)
|
||||
#if UINTPTR_MAX == 0xffffffff
|
||||
uint32_t padding; // Increase size to a multiple of 8.
|
||||
#endif
|
||||
};
|
||||
|
||||
upb_OneofDef* _upb_OneofDef_At(const upb_OneofDef* o, int i) {
|
||||
return (upb_OneofDef*)&o[i];
|
||||
}
|
||||
|
||||
const UPB_DESC(OneofOptions) * upb_OneofDef_Options(const upb_OneofDef* o) {
|
||||
return o->opts;
|
||||
}
|
||||
|
||||
bool upb_OneofDef_HasOptions(const upb_OneofDef* o) {
|
||||
return o->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
const char* upb_OneofDef_FullName(const upb_OneofDef* o) {
|
||||
return o->full_name;
|
||||
}
|
||||
|
||||
const char* upb_OneofDef_Name(const upb_OneofDef* o) {
|
||||
return _upb_DefBuilder_FullToShort(o->full_name);
|
||||
}
|
||||
|
||||
const upb_MessageDef* upb_OneofDef_ContainingType(const upb_OneofDef* o) {
|
||||
return o->parent;
|
||||
}
|
||||
|
||||
int upb_OneofDef_FieldCount(const upb_OneofDef* o) { return o->field_count; }
|
||||
|
||||
const upb_FieldDef* upb_OneofDef_Field(const upb_OneofDef* o, int i) {
|
||||
UPB_ASSERT(i < o->field_count);
|
||||
return o->fields[i];
|
||||
}
|
||||
|
||||
int upb_OneofDef_numfields(const upb_OneofDef* o) { return o->field_count; }
|
||||
|
||||
uint32_t upb_OneofDef_Index(const upb_OneofDef* o) {
|
||||
// Compute index in our parent's array.
|
||||
return o - upb_MessageDef_Oneof(o->parent, 0);
|
||||
}
|
||||
|
||||
bool upb_OneofDef_IsSynthetic(const upb_OneofDef* o) { return o->synthetic; }
|
||||
|
||||
const upb_FieldDef* upb_OneofDef_LookupNameWithSize(const upb_OneofDef* o,
|
||||
const char* name,
|
||||
size_t size) {
|
||||
upb_value val;
|
||||
return upb_strtable_lookup2(&o->ntof, name, size, &val)
|
||||
? upb_value_getptr(val)
|
||||
: NULL;
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_OneofDef_LookupName(const upb_OneofDef* o,
|
||||
const char* name) {
|
||||
return upb_OneofDef_LookupNameWithSize(o, name, strlen(name));
|
||||
}
|
||||
|
||||
const upb_FieldDef* upb_OneofDef_LookupNumber(const upb_OneofDef* o,
|
||||
uint32_t num) {
|
||||
upb_value val;
|
||||
return upb_inttable_lookup(&o->itof, num, &val) ? upb_value_getptr(val)
|
||||
: NULL;
|
||||
}
|
||||
|
||||
void _upb_OneofDef_Insert(upb_DefBuilder* ctx, upb_OneofDef* o,
|
||||
const upb_FieldDef* f, const char* name,
|
||||
size_t size) {
|
||||
o->field_count++;
|
||||
if (_upb_FieldDef_IsProto3Optional(f)) o->synthetic = true;
|
||||
|
||||
const int number = upb_FieldDef_Number(f);
|
||||
const upb_value v = upb_value_constptr(f);
|
||||
|
||||
// TODO: This lookup is unfortunate because we also perform it when
|
||||
// inserting into the message's table. Unfortunately that step occurs after
|
||||
// this one and moving things around could be tricky so let's leave it for
|
||||
// a future refactoring.
|
||||
const bool number_exists = upb_inttable_lookup(&o->itof, number, NULL);
|
||||
if (UPB_UNLIKELY(number_exists)) {
|
||||
_upb_DefBuilder_Errf(ctx, "oneof fields have the same number (%d)", number);
|
||||
}
|
||||
|
||||
// TODO: More redundant work happening here.
|
||||
const bool name_exists = upb_strtable_lookup2(&o->ntof, name, size, NULL);
|
||||
if (UPB_UNLIKELY(name_exists)) {
|
||||
_upb_DefBuilder_Errf(ctx, "oneof fields have the same name (%.*s)",
|
||||
(int)size, name);
|
||||
}
|
||||
|
||||
const bool ok = upb_inttable_insert(&o->itof, number, v, ctx->arena) &&
|
||||
upb_strtable_insert(&o->ntof, name, size, v, ctx->arena);
|
||||
if (UPB_UNLIKELY(!ok)) {
|
||||
_upb_DefBuilder_OomErr(ctx);
|
||||
}
|
||||
}
|
||||
|
||||
// Returns the synthetic count.
|
||||
size_t _upb_OneofDefs_Finalize(upb_DefBuilder* ctx, upb_MessageDef* m) {
|
||||
int synthetic_count = 0;
|
||||
|
||||
for (int i = 0; i < upb_MessageDef_OneofCount(m); i++) {
|
||||
upb_OneofDef* o = (upb_OneofDef*)upb_MessageDef_Oneof(m, i);
|
||||
|
||||
if (o->synthetic && o->field_count != 1) {
|
||||
_upb_DefBuilder_Errf(ctx,
|
||||
"Synthetic oneofs must have one field, not %d: %s",
|
||||
o->field_count, upb_OneofDef_Name(o));
|
||||
}
|
||||
|
||||
if (o->synthetic) {
|
||||
synthetic_count++;
|
||||
} else if (synthetic_count != 0) {
|
||||
_upb_DefBuilder_Errf(
|
||||
ctx, "Synthetic oneofs must be after all other oneofs: %s",
|
||||
upb_OneofDef_Name(o));
|
||||
}
|
||||
|
||||
o->fields =
|
||||
_upb_DefBuilder_Alloc(ctx, sizeof(upb_FieldDef*) * o->field_count);
|
||||
o->field_count = 0;
|
||||
}
|
||||
|
||||
for (int i = 0; i < upb_MessageDef_FieldCount(m); i++) {
|
||||
const upb_FieldDef* f = upb_MessageDef_Field(m, i);
|
||||
upb_OneofDef* o = (upb_OneofDef*)upb_FieldDef_ContainingOneof(f);
|
||||
if (o) {
|
||||
o->fields[o->field_count++] = f;
|
||||
}
|
||||
}
|
||||
|
||||
return synthetic_count;
|
||||
}
|
||||
|
||||
static void create_oneofdef(upb_DefBuilder* ctx, upb_MessageDef* m,
|
||||
const UPB_DESC(OneofDescriptorProto) * oneof_proto,
|
||||
const upb_OneofDef* _o) {
|
||||
upb_OneofDef* o = (upb_OneofDef*)_o;
|
||||
upb_StringView name = UPB_DESC(OneofDescriptorProto_name)(oneof_proto);
|
||||
|
||||
o->parent = m;
|
||||
o->full_name =
|
||||
_upb_DefBuilder_MakeFullName(ctx, upb_MessageDef_FullName(m), name);
|
||||
o->field_count = 0;
|
||||
o->synthetic = false;
|
||||
|
||||
UPB_DEF_SET_OPTIONS(o->opts, OneofDescriptorProto, OneofOptions, oneof_proto);
|
||||
|
||||
if (upb_MessageDef_FindByNameWithSize(m, name.data, name.size, NULL, NULL)) {
|
||||
_upb_DefBuilder_Errf(ctx, "duplicate oneof name (%s)", o->full_name);
|
||||
}
|
||||
|
||||
upb_value v = _upb_DefType_Pack(o, UPB_DEFTYPE_ONEOF);
|
||||
bool ok = _upb_MessageDef_Insert(m, name.data, name.size, v, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
ok = upb_inttable_init(&o->itof, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
|
||||
ok = upb_strtable_init(&o->ntof, 4, ctx->arena);
|
||||
if (!ok) _upb_DefBuilder_OomErr(ctx);
|
||||
}
|
||||
|
||||
// Allocate and initialize an array of |n| oneof defs.
|
||||
upb_OneofDef* _upb_OneofDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(OneofDescriptorProto) * const* protos, upb_MessageDef* m) {
|
||||
_upb_DefType_CheckPadding(sizeof(upb_OneofDef));
|
||||
|
||||
upb_OneofDef* o = _upb_DefBuilder_Alloc(ctx, sizeof(upb_OneofDef) * n);
|
||||
for (int i = 0; i < n; i++) {
|
||||
create_oneofdef(ctx, m, protos[i], &o[i]);
|
||||
}
|
||||
return o;
|
||||
}
|
||||
47
Pods/gRPC-Core/third_party/upb/upb/reflection/oneof_def.h
generated
vendored
Normal file
47
Pods/gRPC-Core/third_party/upb/upb/reflection/oneof_def.h
generated
vendored
Normal file
@@ -0,0 +1,47 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_ONEOF_DEF_H_
|
||||
#define UPB_REFLECTION_ONEOF_DEF_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_API const upb_MessageDef* upb_OneofDef_ContainingType(
|
||||
const upb_OneofDef* o);
|
||||
UPB_API const upb_FieldDef* upb_OneofDef_Field(const upb_OneofDef* o, int i);
|
||||
UPB_API int upb_OneofDef_FieldCount(const upb_OneofDef* o);
|
||||
const char* upb_OneofDef_FullName(const upb_OneofDef* o);
|
||||
bool upb_OneofDef_HasOptions(const upb_OneofDef* o);
|
||||
uint32_t upb_OneofDef_Index(const upb_OneofDef* o);
|
||||
bool upb_OneofDef_IsSynthetic(const upb_OneofDef* o);
|
||||
const upb_FieldDef* upb_OneofDef_LookupName(const upb_OneofDef* o,
|
||||
const char* name);
|
||||
const upb_FieldDef* upb_OneofDef_LookupNameWithSize(const upb_OneofDef* o,
|
||||
const char* name,
|
||||
size_t size);
|
||||
const upb_FieldDef* upb_OneofDef_LookupNumber(const upb_OneofDef* o,
|
||||
uint32_t num);
|
||||
UPB_API const char* upb_OneofDef_Name(const upb_OneofDef* o);
|
||||
int upb_OneofDef_numfields(const upb_OneofDef* o);
|
||||
const UPB_DESC(OneofOptions) * upb_OneofDef_Options(const upb_OneofDef* o);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_ONEOF_DEF_H_ */
|
||||
109
Pods/gRPC-Core/third_party/upb/upb/reflection/service_def.c
generated
vendored
Normal file
109
Pods/gRPC-Core/third_party/upb/upb/reflection/service_def.c
generated
vendored
Normal file
@@ -0,0 +1,109 @@
|
||||
// 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/reflection/internal/service_def.h"
|
||||
|
||||
#include "upb/reflection/def_type.h"
|
||||
#include "upb/reflection/internal/def_builder.h"
|
||||
#include "upb/reflection/internal/file_def.h"
|
||||
#include "upb/reflection/internal/method_def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
struct upb_ServiceDef {
|
||||
const UPB_DESC(ServiceOptions) * opts;
|
||||
const upb_FileDef* file;
|
||||
const char* full_name;
|
||||
upb_MethodDef* methods;
|
||||
int method_count;
|
||||
int index;
|
||||
};
|
||||
|
||||
upb_ServiceDef* _upb_ServiceDef_At(const upb_ServiceDef* s, int index) {
|
||||
return (upb_ServiceDef*)&s[index];
|
||||
}
|
||||
|
||||
const UPB_DESC(ServiceOptions) *
|
||||
upb_ServiceDef_Options(const upb_ServiceDef* s) {
|
||||
return s->opts;
|
||||
}
|
||||
|
||||
bool upb_ServiceDef_HasOptions(const upb_ServiceDef* s) {
|
||||
return s->opts != (void*)kUpbDefOptDefault;
|
||||
}
|
||||
|
||||
const char* upb_ServiceDef_FullName(const upb_ServiceDef* s) {
|
||||
return s->full_name;
|
||||
}
|
||||
|
||||
const char* upb_ServiceDef_Name(const upb_ServiceDef* s) {
|
||||
return _upb_DefBuilder_FullToShort(s->full_name);
|
||||
}
|
||||
|
||||
int upb_ServiceDef_Index(const upb_ServiceDef* s) { return s->index; }
|
||||
|
||||
const upb_FileDef* upb_ServiceDef_File(const upb_ServiceDef* s) {
|
||||
return s->file;
|
||||
}
|
||||
|
||||
int upb_ServiceDef_MethodCount(const upb_ServiceDef* s) {
|
||||
return s->method_count;
|
||||
}
|
||||
|
||||
const upb_MethodDef* upb_ServiceDef_Method(const upb_ServiceDef* s, int i) {
|
||||
return (i < 0 || i >= s->method_count) ? NULL
|
||||
: _upb_MethodDef_At(s->methods, i);
|
||||
}
|
||||
|
||||
const upb_MethodDef* upb_ServiceDef_FindMethodByName(const upb_ServiceDef* s,
|
||||
const char* name) {
|
||||
for (int i = 0; i < s->method_count; i++) {
|
||||
const upb_MethodDef* m = _upb_MethodDef_At(s->methods, i);
|
||||
if (strcmp(name, upb_MethodDef_Name(m)) == 0) {
|
||||
return m;
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static void create_service(upb_DefBuilder* ctx,
|
||||
const UPB_DESC(ServiceDescriptorProto) * svc_proto,
|
||||
upb_ServiceDef* s) {
|
||||
upb_StringView name;
|
||||
size_t n;
|
||||
|
||||
// Must happen before _upb_DefBuilder_Add()
|
||||
s->file = _upb_DefBuilder_File(ctx);
|
||||
|
||||
name = UPB_DESC(ServiceDescriptorProto_name)(svc_proto);
|
||||
const char* package = _upb_FileDef_RawPackage(s->file);
|
||||
s->full_name = _upb_DefBuilder_MakeFullName(ctx, package, name);
|
||||
_upb_DefBuilder_Add(ctx, s->full_name,
|
||||
_upb_DefType_Pack(s, UPB_DEFTYPE_SERVICE));
|
||||
|
||||
const UPB_DESC(MethodDescriptorProto)* const* methods =
|
||||
UPB_DESC(ServiceDescriptorProto_method)(svc_proto, &n);
|
||||
s->method_count = n;
|
||||
s->methods = _upb_MethodDefs_New(ctx, n, methods, s);
|
||||
|
||||
UPB_DEF_SET_OPTIONS(s->opts, ServiceDescriptorProto, ServiceOptions,
|
||||
svc_proto);
|
||||
}
|
||||
|
||||
upb_ServiceDef* _upb_ServiceDefs_New(
|
||||
upb_DefBuilder* ctx, int n,
|
||||
const UPB_DESC(ServiceDescriptorProto) * const* protos) {
|
||||
_upb_DefType_CheckPadding(sizeof(upb_ServiceDef));
|
||||
|
||||
upb_ServiceDef* s = _upb_DefBuilder_Alloc(ctx, sizeof(upb_ServiceDef) * n);
|
||||
for (int i = 0; i < n; i++) {
|
||||
create_service(ctx, protos[i], &s[i]);
|
||||
s[i].index = i;
|
||||
}
|
||||
return s;
|
||||
}
|
||||
40
Pods/gRPC-Core/third_party/upb/upb/reflection/service_def.h
generated
vendored
Normal file
40
Pods/gRPC-Core/third_party/upb/upb/reflection/service_def.h
generated
vendored
Normal file
@@ -0,0 +1,40 @@
|
||||
// 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
|
||||
|
||||
// IWYU pragma: private, include "upb/reflection/def.h"
|
||||
|
||||
#ifndef UPB_REFLECTION_SERVICE_DEF_H_
|
||||
#define UPB_REFLECTION_SERVICE_DEF_H_
|
||||
|
||||
#include "upb/reflection/common.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
const upb_FileDef* upb_ServiceDef_File(const upb_ServiceDef* s);
|
||||
const upb_MethodDef* upb_ServiceDef_FindMethodByName(const upb_ServiceDef* s,
|
||||
const char* name);
|
||||
const char* upb_ServiceDef_FullName(const upb_ServiceDef* s);
|
||||
bool upb_ServiceDef_HasOptions(const upb_ServiceDef* s);
|
||||
int upb_ServiceDef_Index(const upb_ServiceDef* s);
|
||||
const upb_MethodDef* upb_ServiceDef_Method(const upb_ServiceDef* s, int i);
|
||||
int upb_ServiceDef_MethodCount(const upb_ServiceDef* s);
|
||||
const char* upb_ServiceDef_Name(const upb_ServiceDef* s);
|
||||
const UPB_DESC(ServiceOptions) *
|
||||
upb_ServiceDef_Options(const upb_ServiceDef* s);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_REFLECTION_SERVICE_DEF_H_ */
|
||||
457
Pods/gRPC-Core/third_party/upb/upb/text/encode.c
generated
vendored
Normal file
457
Pods/gRPC-Core/third_party/upb/upb/text/encode.c
generated
vendored
Normal file
@@ -0,0 +1,457 @@
|
||||
// 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/text/encode.h"
|
||||
|
||||
#include <ctype.h>
|
||||
#include <float.h>
|
||||
#include <inttypes.h>
|
||||
#include <stdarg.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/lex/round_trip.h"
|
||||
#include "upb/message/internal/map_sorter.h"
|
||||
#include "upb/message/map.h"
|
||||
#include "upb/port/vsnprintf_compat.h"
|
||||
#include "upb/reflection/message.h"
|
||||
#include "upb/wire/eps_copy_input_stream.h"
|
||||
#include "upb/wire/reader.h"
|
||||
#include "upb/wire/types.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
typedef struct {
|
||||
char *buf, *ptr, *end;
|
||||
size_t overflow;
|
||||
int indent_depth;
|
||||
int options;
|
||||
const upb_DefPool* ext_pool;
|
||||
_upb_mapsorter sorter;
|
||||
} txtenc;
|
||||
|
||||
static void txtenc_msg(txtenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m);
|
||||
|
||||
static void txtenc_putbytes(txtenc* e, const void* data, size_t len) {
|
||||
size_t have = e->end - e->ptr;
|
||||
if (UPB_LIKELY(have >= len)) {
|
||||
memcpy(e->ptr, data, len);
|
||||
e->ptr += len;
|
||||
} else {
|
||||
if (have) {
|
||||
memcpy(e->ptr, data, have);
|
||||
e->ptr += have;
|
||||
}
|
||||
e->overflow += (len - have);
|
||||
}
|
||||
}
|
||||
|
||||
static void txtenc_putstr(txtenc* e, const char* str) {
|
||||
txtenc_putbytes(e, str, strlen(str));
|
||||
}
|
||||
|
||||
static void txtenc_printf(txtenc* e, const char* fmt, ...) {
|
||||
size_t n;
|
||||
size_t have = e->end - e->ptr;
|
||||
va_list args;
|
||||
|
||||
va_start(args, fmt);
|
||||
n = _upb_vsnprintf(e->ptr, have, fmt, args);
|
||||
va_end(args);
|
||||
|
||||
if (UPB_LIKELY(have > n)) {
|
||||
e->ptr += n;
|
||||
} else {
|
||||
e->ptr = UPB_PTRADD(e->ptr, have);
|
||||
e->overflow += (n - have);
|
||||
}
|
||||
}
|
||||
|
||||
static void txtenc_indent(txtenc* e) {
|
||||
if ((e->options & UPB_TXTENC_SINGLELINE) == 0) {
|
||||
int i = e->indent_depth;
|
||||
while (i-- > 0) {
|
||||
txtenc_putstr(e, " ");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void txtenc_endfield(txtenc* e) {
|
||||
if (e->options & UPB_TXTENC_SINGLELINE) {
|
||||
txtenc_putstr(e, " ");
|
||||
} else {
|
||||
txtenc_putstr(e, "\n");
|
||||
}
|
||||
}
|
||||
|
||||
static void txtenc_enum(int32_t val, const upb_FieldDef* f, txtenc* e) {
|
||||
const upb_EnumDef* e_def = upb_FieldDef_EnumSubDef(f);
|
||||
const upb_EnumValueDef* ev = upb_EnumDef_FindValueByNumber(e_def, val);
|
||||
|
||||
if (ev) {
|
||||
txtenc_printf(e, "%s", upb_EnumValueDef_Name(ev));
|
||||
} else {
|
||||
txtenc_printf(e, "%" PRId32, val);
|
||||
}
|
||||
}
|
||||
|
||||
static void txtenc_string(txtenc* e, upb_StringView str, bool bytes) {
|
||||
const char* ptr = str.data;
|
||||
const char* end = ptr + str.size;
|
||||
txtenc_putstr(e, "\"");
|
||||
|
||||
while (ptr < end) {
|
||||
switch (*ptr) {
|
||||
case '\n':
|
||||
txtenc_putstr(e, "\\n");
|
||||
break;
|
||||
case '\r':
|
||||
txtenc_putstr(e, "\\r");
|
||||
break;
|
||||
case '\t':
|
||||
txtenc_putstr(e, "\\t");
|
||||
break;
|
||||
case '\"':
|
||||
txtenc_putstr(e, "\\\"");
|
||||
break;
|
||||
case '\'':
|
||||
txtenc_putstr(e, "\\'");
|
||||
break;
|
||||
case '\\':
|
||||
txtenc_putstr(e, "\\\\");
|
||||
break;
|
||||
default:
|
||||
if ((bytes || (uint8_t)*ptr < 0x80) && !isprint(*ptr)) {
|
||||
txtenc_printf(e, "\\%03o", (int)(uint8_t)*ptr);
|
||||
} else {
|
||||
txtenc_putbytes(e, ptr, 1);
|
||||
}
|
||||
break;
|
||||
}
|
||||
ptr++;
|
||||
}
|
||||
|
||||
txtenc_putstr(e, "\"");
|
||||
}
|
||||
|
||||
static void txtenc_field(txtenc* e, upb_MessageValue val,
|
||||
const upb_FieldDef* f) {
|
||||
txtenc_indent(e);
|
||||
const upb_CType type = upb_FieldDef_CType(f);
|
||||
const bool is_ext = upb_FieldDef_IsExtension(f);
|
||||
const char* full = upb_FieldDef_FullName(f);
|
||||
const char* name = upb_FieldDef_Name(f);
|
||||
|
||||
if (type == kUpb_CType_Message) {
|
||||
if (is_ext) {
|
||||
txtenc_printf(e, "[%s] {", full);
|
||||
} else {
|
||||
txtenc_printf(e, "%s {", name);
|
||||
}
|
||||
txtenc_endfield(e);
|
||||
e->indent_depth++;
|
||||
txtenc_msg(e, val.msg_val, upb_FieldDef_MessageSubDef(f));
|
||||
e->indent_depth--;
|
||||
txtenc_indent(e);
|
||||
txtenc_putstr(e, "}");
|
||||
txtenc_endfield(e);
|
||||
return;
|
||||
}
|
||||
|
||||
if (is_ext) {
|
||||
txtenc_printf(e, "[%s]: ", full);
|
||||
} else {
|
||||
txtenc_printf(e, "%s: ", name);
|
||||
}
|
||||
|
||||
switch (type) {
|
||||
case kUpb_CType_Bool:
|
||||
txtenc_putstr(e, val.bool_val ? "true" : "false");
|
||||
break;
|
||||
case kUpb_CType_Float: {
|
||||
char buf[32];
|
||||
_upb_EncodeRoundTripFloat(val.float_val, buf, sizeof(buf));
|
||||
txtenc_putstr(e, buf);
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_Double: {
|
||||
char buf[32];
|
||||
_upb_EncodeRoundTripDouble(val.double_val, buf, sizeof(buf));
|
||||
txtenc_putstr(e, buf);
|
||||
break;
|
||||
}
|
||||
case kUpb_CType_Int32:
|
||||
txtenc_printf(e, "%" PRId32, val.int32_val);
|
||||
break;
|
||||
case kUpb_CType_UInt32:
|
||||
txtenc_printf(e, "%" PRIu32, val.uint32_val);
|
||||
break;
|
||||
case kUpb_CType_Int64:
|
||||
txtenc_printf(e, "%" PRId64, val.int64_val);
|
||||
break;
|
||||
case kUpb_CType_UInt64:
|
||||
txtenc_printf(e, "%" PRIu64, val.uint64_val);
|
||||
break;
|
||||
case kUpb_CType_String:
|
||||
txtenc_string(e, val.str_val, false);
|
||||
break;
|
||||
case kUpb_CType_Bytes:
|
||||
txtenc_string(e, val.str_val, true);
|
||||
break;
|
||||
case kUpb_CType_Enum:
|
||||
txtenc_enum(val.int32_val, f, e);
|
||||
break;
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
txtenc_endfield(e);
|
||||
}
|
||||
|
||||
/*
|
||||
* Arrays print as simple repeated elements, eg.
|
||||
*
|
||||
* foo_field: 1
|
||||
* foo_field: 2
|
||||
* foo_field: 3
|
||||
*/
|
||||
static void txtenc_array(txtenc* e, const upb_Array* arr,
|
||||
const upb_FieldDef* f) {
|
||||
size_t i;
|
||||
size_t size = upb_Array_Size(arr);
|
||||
|
||||
for (i = 0; i < size; i++) {
|
||||
txtenc_field(e, upb_Array_Get(arr, i), f);
|
||||
}
|
||||
}
|
||||
|
||||
static void txtenc_mapentry(txtenc* e, upb_MessageValue key,
|
||||
upb_MessageValue val, const upb_FieldDef* f) {
|
||||
const upb_MessageDef* entry = upb_FieldDef_MessageSubDef(f);
|
||||
const upb_FieldDef* key_f = upb_MessageDef_Field(entry, 0);
|
||||
const upb_FieldDef* val_f = upb_MessageDef_Field(entry, 1);
|
||||
txtenc_indent(e);
|
||||
txtenc_printf(e, "%s {", upb_FieldDef_Name(f));
|
||||
txtenc_endfield(e);
|
||||
e->indent_depth++;
|
||||
|
||||
txtenc_field(e, key, key_f);
|
||||
txtenc_field(e, val, val_f);
|
||||
|
||||
e->indent_depth--;
|
||||
txtenc_indent(e);
|
||||
txtenc_putstr(e, "}");
|
||||
txtenc_endfield(e);
|
||||
}
|
||||
|
||||
/*
|
||||
* Maps print as messages of key/value, etc.
|
||||
*
|
||||
* foo_map: {
|
||||
* key: "abc"
|
||||
* value: 123
|
||||
* }
|
||||
* foo_map: {
|
||||
* key: "def"
|
||||
* value: 456
|
||||
* }
|
||||
*/
|
||||
static void txtenc_map(txtenc* e, const upb_Map* map, const upb_FieldDef* f) {
|
||||
if (e->options & UPB_TXTENC_NOSORT) {
|
||||
size_t iter = kUpb_Map_Begin;
|
||||
upb_MessageValue key, val;
|
||||
while (upb_Map_Next(map, &key, &val, &iter)) {
|
||||
txtenc_mapentry(e, key, val, f);
|
||||
}
|
||||
} else {
|
||||
const upb_MessageDef* entry = upb_FieldDef_MessageSubDef(f);
|
||||
const upb_FieldDef* key_f = upb_MessageDef_Field(entry, 0);
|
||||
_upb_sortedmap sorted;
|
||||
upb_MapEntry ent;
|
||||
|
||||
_upb_mapsorter_pushmap(&e->sorter, upb_FieldDef_Type(key_f), map, &sorted);
|
||||
while (_upb_sortedmap_next(&e->sorter, map, &sorted, &ent)) {
|
||||
upb_MessageValue key, val;
|
||||
memcpy(&key, &ent.data.k, sizeof(key));
|
||||
memcpy(&val, &ent.data.v, sizeof(val));
|
||||
txtenc_mapentry(e, key, val, f);
|
||||
}
|
||||
_upb_mapsorter_popmap(&e->sorter, &sorted);
|
||||
}
|
||||
}
|
||||
|
||||
#define CHK(x) \
|
||||
do { \
|
||||
if (!(x)) { \
|
||||
return false; \
|
||||
} \
|
||||
} while (0)
|
||||
|
||||
/*
|
||||
* Unknown fields are printed by number.
|
||||
*
|
||||
* 1001: 123
|
||||
* 1002: "hello"
|
||||
* 1006: 0xdeadbeef
|
||||
* 1003: {
|
||||
* 1: 111
|
||||
* }
|
||||
*/
|
||||
static const char* txtenc_unknown(txtenc* e, const char* ptr,
|
||||
upb_EpsCopyInputStream* stream,
|
||||
int groupnum) {
|
||||
// We are guaranteed that the unknown data is valid wire format, and will not
|
||||
// contain tag zero.
|
||||
uint32_t end_group = groupnum > 0
|
||||
? ((groupnum << kUpb_WireReader_WireTypeBits) |
|
||||
kUpb_WireType_EndGroup)
|
||||
: 0;
|
||||
|
||||
while (!upb_EpsCopyInputStream_IsDone(stream, &ptr)) {
|
||||
uint32_t tag;
|
||||
CHK(ptr = upb_WireReader_ReadTag(ptr, &tag));
|
||||
if (tag == end_group) return ptr;
|
||||
|
||||
txtenc_indent(e);
|
||||
txtenc_printf(e, "%d: ", (int)upb_WireReader_GetFieldNumber(tag));
|
||||
|
||||
switch (upb_WireReader_GetWireType(tag)) {
|
||||
case kUpb_WireType_Varint: {
|
||||
uint64_t val;
|
||||
CHK(ptr = upb_WireReader_ReadVarint(ptr, &val));
|
||||
txtenc_printf(e, "%" PRIu64, val);
|
||||
break;
|
||||
}
|
||||
case kUpb_WireType_32Bit: {
|
||||
uint32_t val;
|
||||
ptr = upb_WireReader_ReadFixed32(ptr, &val);
|
||||
txtenc_printf(e, "0x%08" PRIu32, val);
|
||||
break;
|
||||
}
|
||||
case kUpb_WireType_64Bit: {
|
||||
uint64_t val;
|
||||
ptr = upb_WireReader_ReadFixed64(ptr, &val);
|
||||
txtenc_printf(e, "0x%016" PRIu64, val);
|
||||
break;
|
||||
}
|
||||
case kUpb_WireType_Delimited: {
|
||||
int size;
|
||||
char* start = e->ptr;
|
||||
size_t start_overflow = e->overflow;
|
||||
CHK(ptr = upb_WireReader_ReadSize(ptr, &size));
|
||||
CHK(upb_EpsCopyInputStream_CheckDataSizeAvailable(stream, ptr, size));
|
||||
|
||||
// Speculatively try to parse as message.
|
||||
txtenc_putstr(e, "{");
|
||||
txtenc_endfield(e);
|
||||
|
||||
// EpsCopyInputStream can't back up, so create a sub-stream for the
|
||||
// speculative parse.
|
||||
upb_EpsCopyInputStream sub_stream;
|
||||
const char* sub_ptr = upb_EpsCopyInputStream_GetAliasedPtr(stream, ptr);
|
||||
upb_EpsCopyInputStream_Init(&sub_stream, &sub_ptr, size, true);
|
||||
|
||||
e->indent_depth++;
|
||||
if (txtenc_unknown(e, sub_ptr, &sub_stream, -1)) {
|
||||
ptr = upb_EpsCopyInputStream_Skip(stream, ptr, size);
|
||||
e->indent_depth--;
|
||||
txtenc_indent(e);
|
||||
txtenc_putstr(e, "}");
|
||||
} else {
|
||||
// Didn't work out, print as raw bytes.
|
||||
e->indent_depth--;
|
||||
e->ptr = start;
|
||||
e->overflow = start_overflow;
|
||||
const char* str = ptr;
|
||||
ptr = upb_EpsCopyInputStream_ReadString(stream, &str, size, NULL);
|
||||
assert(ptr);
|
||||
txtenc_string(e, (upb_StringView){.data = str, .size = size}, true);
|
||||
}
|
||||
break;
|
||||
}
|
||||
case kUpb_WireType_StartGroup:
|
||||
txtenc_putstr(e, "{");
|
||||
txtenc_endfield(e);
|
||||
e->indent_depth++;
|
||||
CHK(ptr = txtenc_unknown(e, ptr, stream,
|
||||
upb_WireReader_GetFieldNumber(tag)));
|
||||
e->indent_depth--;
|
||||
txtenc_indent(e);
|
||||
txtenc_putstr(e, "}");
|
||||
break;
|
||||
default:
|
||||
return NULL;
|
||||
}
|
||||
txtenc_endfield(e);
|
||||
}
|
||||
|
||||
return end_group == 0 && !upb_EpsCopyInputStream_IsError(stream) ? ptr : NULL;
|
||||
}
|
||||
|
||||
#undef CHK
|
||||
|
||||
static void txtenc_msg(txtenc* e, const upb_Message* msg,
|
||||
const upb_MessageDef* m) {
|
||||
size_t iter = kUpb_Message_Begin;
|
||||
const upb_FieldDef* f;
|
||||
upb_MessageValue val;
|
||||
|
||||
while (upb_Message_Next(msg, m, e->ext_pool, &f, &val, &iter)) {
|
||||
if (upb_FieldDef_IsMap(f)) {
|
||||
txtenc_map(e, val.map_val, f);
|
||||
} else if (upb_FieldDef_IsRepeated(f)) {
|
||||
txtenc_array(e, val.array_val, f);
|
||||
} else {
|
||||
txtenc_field(e, val, f);
|
||||
}
|
||||
}
|
||||
|
||||
if ((e->options & UPB_TXTENC_SKIPUNKNOWN) == 0) {
|
||||
size_t size;
|
||||
const char* ptr = upb_Message_GetUnknown(msg, &size);
|
||||
if (size != 0) {
|
||||
char* start = e->ptr;
|
||||
upb_EpsCopyInputStream stream;
|
||||
upb_EpsCopyInputStream_Init(&stream, &ptr, size, true);
|
||||
if (!txtenc_unknown(e, ptr, &stream, -1)) {
|
||||
/* Unknown failed to parse, back up and don't print it at all. */
|
||||
e->ptr = start;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
size_t txtenc_nullz(txtenc* e, size_t size) {
|
||||
size_t ret = e->ptr - e->buf + e->overflow;
|
||||
|
||||
if (size > 0) {
|
||||
if (e->ptr == e->end) e->ptr--;
|
||||
*e->ptr = '\0';
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
size_t upb_TextEncode(const upb_Message* msg, const upb_MessageDef* m,
|
||||
const upb_DefPool* ext_pool, int options, char* buf,
|
||||
size_t size) {
|
||||
txtenc e;
|
||||
|
||||
e.buf = buf;
|
||||
e.ptr = buf;
|
||||
e.end = UPB_PTRADD(buf, size);
|
||||
e.overflow = 0;
|
||||
e.indent_depth = 0;
|
||||
e.options = options;
|
||||
e.ext_pool = ext_pool;
|
||||
_upb_mapsorter_init(&e.sorter);
|
||||
|
||||
txtenc_msg(&e, msg, m);
|
||||
_upb_mapsorter_destroy(&e.sorter);
|
||||
return txtenc_nullz(&e, size);
|
||||
}
|
||||
49
Pods/gRPC-Core/third_party/upb/upb/text/encode.h
generated
vendored
Normal file
49
Pods/gRPC-Core/third_party/upb/upb/text/encode.h
generated
vendored
Normal file
@@ -0,0 +1,49 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_TEXT_ENCODE_H_
|
||||
#define UPB_TEXT_ENCODE_H_
|
||||
|
||||
#include "upb/reflection/def.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
enum {
|
||||
// When set, prints everything on a single line.
|
||||
UPB_TXTENC_SINGLELINE = 1,
|
||||
|
||||
// When set, unknown fields are not printed.
|
||||
UPB_TXTENC_SKIPUNKNOWN = 2,
|
||||
|
||||
// When set, maps are *not* sorted (this avoids allocating tmp mem).
|
||||
UPB_TXTENC_NOSORT = 4
|
||||
};
|
||||
|
||||
/* Encodes the given |msg| to text format. The message's reflection is given in
|
||||
* |m|. The symtab in |symtab| is used to find extensions (if NULL, extensions
|
||||
* will not be printed).
|
||||
*
|
||||
* Output is placed in the given buffer, and always NULL-terminated. The output
|
||||
* size (excluding NULL) is returned. This means that a return value >= |size|
|
||||
* implies that the output was truncated. (These are the same semantics as
|
||||
* snprintf()). */
|
||||
size_t upb_TextEncode(const upb_Message* msg, const upb_MessageDef* m,
|
||||
const upb_DefPool* ext_pool, int options, char* buf,
|
||||
size_t size);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_TEXT_ENCODE_H_ */
|
||||
1381
Pods/gRPC-Core/third_party/upb/upb/wire/decode.c
generated
vendored
Normal file
1381
Pods/gRPC-Core/third_party/upb/upb/wire/decode.c
generated
vendored
Normal file
@@ -0,0 +1,1381 @@
|
||||
// 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/wire/decode.h"
|
||||
|
||||
#include <assert.h>
|
||||
#include <stdbool.h>
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/hash/common.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mem/internal/arena.h"
|
||||
#include "upb/message/array.h"
|
||||
#include "upb/message/internal/accessors.h"
|
||||
#include "upb/message/internal/array.h"
|
||||
#include "upb/message/internal/extension.h"
|
||||
#include "upb/message/internal/map.h"
|
||||
#include "upb/message/internal/map_entry.h"
|
||||
#include "upb/message/internal/message.h"
|
||||
#include "upb/message/map.h"
|
||||
#include "upb/message/message.h"
|
||||
#include "upb/message/tagged_ptr.h"
|
||||
#include "upb/mini_table/enum.h"
|
||||
#include "upb/mini_table/extension.h"
|
||||
#include "upb/mini_table/extension_registry.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
#include "upb/mini_table/internal/enum.h"
|
||||
#include "upb/mini_table/internal/field.h"
|
||||
#include "upb/mini_table/internal/message.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
#include "upb/mini_table/sub.h"
|
||||
#include "upb/port/atomic.h"
|
||||
#include "upb/wire/encode.h"
|
||||
#include "upb/wire/eps_copy_input_stream.h"
|
||||
#include "upb/wire/internal/constants.h"
|
||||
#include "upb/wire/internal/decode.h"
|
||||
#include "upb/wire/internal/swap.h"
|
||||
#include "upb/wire/reader.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
// A few fake field types for our tables.
|
||||
enum {
|
||||
kUpb_FakeFieldType_FieldNotFound = 0,
|
||||
kUpb_FakeFieldType_MessageSetItem = 19,
|
||||
};
|
||||
|
||||
// DecodeOp: an action to be performed for a wire-type/field-type combination.
|
||||
enum {
|
||||
// Special ops: we don't write data to regular fields for these.
|
||||
kUpb_DecodeOp_UnknownField = -1,
|
||||
kUpb_DecodeOp_MessageSetItem = -2,
|
||||
|
||||
// Scalar-only ops.
|
||||
kUpb_DecodeOp_Scalar1Byte = 0,
|
||||
kUpb_DecodeOp_Scalar4Byte = 2,
|
||||
kUpb_DecodeOp_Scalar8Byte = 3,
|
||||
kUpb_DecodeOp_Enum = 1,
|
||||
|
||||
// Scalar/repeated ops.
|
||||
kUpb_DecodeOp_String = 4,
|
||||
kUpb_DecodeOp_Bytes = 5,
|
||||
kUpb_DecodeOp_SubMessage = 6,
|
||||
|
||||
// Repeated-only ops (also see macros below).
|
||||
kUpb_DecodeOp_PackedEnum = 13,
|
||||
};
|
||||
|
||||
// For packed fields it is helpful to be able to recover the lg2 of the data
|
||||
// size from the op.
|
||||
#define OP_FIXPCK_LG2(n) (n + 5) /* n in [2, 3] => op in [7, 8] */
|
||||
#define OP_VARPCK_LG2(n) (n + 9) /* n in [0, 2, 3] => op in [9, 11, 12] */
|
||||
|
||||
typedef union {
|
||||
bool bool_val;
|
||||
uint32_t uint32_val;
|
||||
uint64_t uint64_val;
|
||||
uint32_t size;
|
||||
} wireval;
|
||||
|
||||
static const char* _upb_Decoder_DecodeMessage(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg,
|
||||
const upb_MiniTable* layout);
|
||||
|
||||
UPB_NORETURN static void* _upb_Decoder_ErrorJmp(upb_Decoder* d,
|
||||
upb_DecodeStatus status) {
|
||||
UPB_ASSERT(status != kUpb_DecodeStatus_Ok);
|
||||
d->status = status;
|
||||
UPB_LONGJMP(d->err, 1);
|
||||
}
|
||||
|
||||
const char* _upb_FastDecoder_ErrorJmp(upb_Decoder* d, int status) {
|
||||
UPB_ASSERT(status != kUpb_DecodeStatus_Ok);
|
||||
d->status = status;
|
||||
UPB_LONGJMP(d->err, 1);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static void _upb_Decoder_VerifyUtf8(upb_Decoder* d, const char* buf, int len) {
|
||||
if (!_upb_Decoder_VerifyUtf8Inline(buf, len)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_BadUtf8);
|
||||
}
|
||||
}
|
||||
|
||||
static bool _upb_Decoder_Reserve(upb_Decoder* d, upb_Array* arr, size_t elem) {
|
||||
bool need_realloc = arr->capacity - arr->size < elem;
|
||||
if (need_realloc && !_upb_array_realloc(arr, arr->size + elem, &d->arena)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
return need_realloc;
|
||||
}
|
||||
|
||||
typedef struct {
|
||||
const char* ptr;
|
||||
uint64_t val;
|
||||
} _upb_DecodeLongVarintReturn;
|
||||
|
||||
UPB_NOINLINE
|
||||
static _upb_DecodeLongVarintReturn _upb_Decoder_DecodeLongVarint(
|
||||
const char* ptr, uint64_t val) {
|
||||
_upb_DecodeLongVarintReturn ret = {NULL, 0};
|
||||
uint64_t byte;
|
||||
int i;
|
||||
for (i = 1; i < 10; i++) {
|
||||
byte = (uint8_t)ptr[i];
|
||||
val += (byte - 1) << (i * 7);
|
||||
if (!(byte & 0x80)) {
|
||||
ret.ptr = ptr + i + 1;
|
||||
ret.val = val;
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeVarint(upb_Decoder* d, const char* ptr,
|
||||
uint64_t* val) {
|
||||
uint64_t byte = (uint8_t)*ptr;
|
||||
if (UPB_LIKELY((byte & 0x80) == 0)) {
|
||||
*val = byte;
|
||||
return ptr + 1;
|
||||
} else {
|
||||
_upb_DecodeLongVarintReturn res = _upb_Decoder_DecodeLongVarint(ptr, byte);
|
||||
if (!res.ptr) _upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
*val = res.val;
|
||||
return res.ptr;
|
||||
}
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeTag(upb_Decoder* d, const char* ptr,
|
||||
uint32_t* val) {
|
||||
uint64_t byte = (uint8_t)*ptr;
|
||||
if (UPB_LIKELY((byte & 0x80) == 0)) {
|
||||
*val = byte;
|
||||
return ptr + 1;
|
||||
} else {
|
||||
const char* start = ptr;
|
||||
_upb_DecodeLongVarintReturn res = _upb_Decoder_DecodeLongVarint(ptr, byte);
|
||||
if (!res.ptr || res.ptr - start > 5 || res.val > UINT32_MAX) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
}
|
||||
*val = res.val;
|
||||
return res.ptr;
|
||||
}
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* upb_Decoder_DecodeSize(upb_Decoder* d, const char* ptr,
|
||||
uint32_t* size) {
|
||||
uint64_t size64;
|
||||
ptr = _upb_Decoder_DecodeVarint(d, ptr, &size64);
|
||||
if (size64 >= INT32_MAX ||
|
||||
!upb_EpsCopyInputStream_CheckSize(&d->input, ptr, (int)size64)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
}
|
||||
*size = size64;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static void _upb_Decoder_MungeInt32(wireval* val) {
|
||||
if (!_upb_IsLittleEndian()) {
|
||||
/* The next stage will memcpy(dst, &val, 4) */
|
||||
val->uint32_val = val->uint64_val;
|
||||
}
|
||||
}
|
||||
|
||||
static void _upb_Decoder_Munge(int type, wireval* val) {
|
||||
switch (type) {
|
||||
case kUpb_FieldType_Bool:
|
||||
val->bool_val = val->uint64_val != 0;
|
||||
break;
|
||||
case kUpb_FieldType_SInt32: {
|
||||
uint32_t n = val->uint64_val;
|
||||
val->uint32_val = (n >> 1) ^ -(int32_t)(n & 1);
|
||||
break;
|
||||
}
|
||||
case kUpb_FieldType_SInt64: {
|
||||
uint64_t n = val->uint64_val;
|
||||
val->uint64_val = (n >> 1) ^ -(int64_t)(n & 1);
|
||||
break;
|
||||
}
|
||||
case kUpb_FieldType_Int32:
|
||||
case kUpb_FieldType_UInt32:
|
||||
case kUpb_FieldType_Enum:
|
||||
_upb_Decoder_MungeInt32(val);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static upb_Message* _upb_Decoder_NewSubMessage(upb_Decoder* d,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* field,
|
||||
upb_TaggedMessagePtr* target) {
|
||||
const upb_MiniTable* subl = subs[field->UPB_PRIVATE(submsg_index)].submsg;
|
||||
UPB_ASSERT(subl);
|
||||
upb_Message* msg = _upb_Message_New(subl, &d->arena);
|
||||
if (!msg) _upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
|
||||
// Extensions should not be unlinked. A message extension should not be
|
||||
// registered until its sub-message type is available to be linked.
|
||||
bool is_empty = subl == &_kUpb_MiniTable_Empty;
|
||||
bool is_extension = field->mode & kUpb_LabelFlags_IsExtension;
|
||||
UPB_ASSERT(!(is_empty && is_extension));
|
||||
|
||||
if (is_empty && !(d->options & kUpb_DecodeOption_ExperimentalAllowUnlinked)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_UnlinkedSubMessage);
|
||||
}
|
||||
|
||||
upb_TaggedMessagePtr tagged = _upb_TaggedMessagePtr_Pack(msg, is_empty);
|
||||
memcpy(target, &tagged, sizeof(tagged));
|
||||
return msg;
|
||||
}
|
||||
|
||||
static upb_Message* _upb_Decoder_ReuseSubMessage(
|
||||
upb_Decoder* d, const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* field, upb_TaggedMessagePtr* target) {
|
||||
upb_TaggedMessagePtr tagged = *target;
|
||||
const upb_MiniTable* subl = subs[field->UPB_PRIVATE(submsg_index)].submsg;
|
||||
UPB_ASSERT(subl);
|
||||
if (!upb_TaggedMessagePtr_IsEmpty(tagged) || subl == &_kUpb_MiniTable_Empty) {
|
||||
return _upb_TaggedMessagePtr_GetMessage(tagged);
|
||||
}
|
||||
|
||||
// We found an empty message from a previous parse that was performed before
|
||||
// this field was linked. But it is linked now, so we want to allocate a new
|
||||
// message of the correct type and promote data into it before continuing.
|
||||
upb_Message* existing = _upb_TaggedMessagePtr_GetEmptyMessage(tagged);
|
||||
upb_Message* promoted = _upb_Decoder_NewSubMessage(d, subs, field, target);
|
||||
size_t size;
|
||||
const char* unknown = upb_Message_GetUnknown(existing, &size);
|
||||
upb_DecodeStatus status = upb_Decode(unknown, size, promoted, subl, d->extreg,
|
||||
d->options, &d->arena);
|
||||
if (status != kUpb_DecodeStatus_Ok) _upb_Decoder_ErrorJmp(d, status);
|
||||
return promoted;
|
||||
}
|
||||
|
||||
static const char* _upb_Decoder_ReadString(upb_Decoder* d, const char* ptr,
|
||||
int size, upb_StringView* str) {
|
||||
const char* str_ptr = ptr;
|
||||
ptr = upb_EpsCopyInputStream_ReadString(&d->input, &str_ptr, size, &d->arena);
|
||||
if (!ptr) _upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
str->data = str_ptr;
|
||||
str->size = size;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_RecurseSubMessage(upb_Decoder* d,
|
||||
const char* ptr,
|
||||
upb_Message* submsg,
|
||||
const upb_MiniTable* subl,
|
||||
uint32_t expected_end_group) {
|
||||
if (--d->depth < 0) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_MaxDepthExceeded);
|
||||
}
|
||||
ptr = _upb_Decoder_DecodeMessage(d, ptr, submsg, subl);
|
||||
d->depth++;
|
||||
if (d->end_group != expected_end_group) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
}
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeSubMessage(
|
||||
upb_Decoder* d, const char* ptr, upb_Message* submsg,
|
||||
const upb_MiniTableSub* subs, const upb_MiniTableField* field, int size) {
|
||||
int saved_delta = upb_EpsCopyInputStream_PushLimit(&d->input, ptr, size);
|
||||
const upb_MiniTable* subl = subs[field->UPB_PRIVATE(submsg_index)].submsg;
|
||||
UPB_ASSERT(subl);
|
||||
ptr = _upb_Decoder_RecurseSubMessage(d, ptr, submsg, subl, DECODE_NOGROUP);
|
||||
upb_EpsCopyInputStream_PopLimit(&d->input, ptr, saved_delta);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeGroup(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* submsg,
|
||||
const upb_MiniTable* subl,
|
||||
uint32_t number) {
|
||||
if (_upb_Decoder_IsDone(d, &ptr)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
}
|
||||
ptr = _upb_Decoder_RecurseSubMessage(d, ptr, submsg, subl, number);
|
||||
d->end_group = DECODE_NOGROUP;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeUnknownGroup(upb_Decoder* d,
|
||||
const char* ptr,
|
||||
uint32_t number) {
|
||||
return _upb_Decoder_DecodeGroup(d, ptr, NULL, NULL, number);
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeKnownGroup(
|
||||
upb_Decoder* d, const char* ptr, upb_Message* submsg,
|
||||
const upb_MiniTableSub* subs, const upb_MiniTableField* field) {
|
||||
const upb_MiniTable* subl = subs[field->UPB_PRIVATE(submsg_index)].submsg;
|
||||
UPB_ASSERT(subl);
|
||||
return _upb_Decoder_DecodeGroup(d, ptr, submsg, subl, field->number);
|
||||
}
|
||||
|
||||
static char* upb_Decoder_EncodeVarint32(uint32_t val, char* ptr) {
|
||||
do {
|
||||
uint8_t byte = val & 0x7fU;
|
||||
val >>= 7;
|
||||
if (val) byte |= 0x80U;
|
||||
*(ptr++) = byte;
|
||||
} while (val);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static void _upb_Decoder_AddUnknownVarints(upb_Decoder* d, upb_Message* msg,
|
||||
uint32_t val1, uint32_t val2) {
|
||||
char buf[20];
|
||||
char* end = buf;
|
||||
end = upb_Decoder_EncodeVarint32(val1, end);
|
||||
end = upb_Decoder_EncodeVarint32(val2, end);
|
||||
|
||||
if (!_upb_Message_AddUnknown(msg, buf, end - buf, &d->arena)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
static bool _upb_Decoder_CheckEnumSlow(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg,
|
||||
const upb_MiniTableEnum* e,
|
||||
const upb_MiniTableField* field,
|
||||
uint32_t v) {
|
||||
if (_upb_MiniTable_CheckEnumValueSlow(e, v)) return true;
|
||||
|
||||
// Unrecognized enum goes into unknown fields.
|
||||
// For packed fields the tag could be arbitrarily far in the past, so we
|
||||
// just re-encode the tag and value here.
|
||||
uint32_t tag = ((uint32_t)field->number << 3) | kUpb_WireType_Varint;
|
||||
upb_Message* unknown_msg =
|
||||
field->mode & kUpb_LabelFlags_IsExtension ? d->unknown_msg : msg;
|
||||
_upb_Decoder_AddUnknownVarints(d, unknown_msg, tag, v);
|
||||
return false;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static bool _upb_Decoder_CheckEnum(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg, const upb_MiniTableEnum* e,
|
||||
const upb_MiniTableField* field,
|
||||
wireval* val) {
|
||||
uint32_t v = val->uint32_val;
|
||||
|
||||
_kUpb_FastEnumCheck_Status status = _upb_MiniTable_CheckEnumValueFast(e, v);
|
||||
if (UPB_LIKELY(status == _kUpb_FastEnumCheck_ValueIsInEnum)) return true;
|
||||
return _upb_Decoder_CheckEnumSlow(d, ptr, msg, e, field, v);
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
static const char* _upb_Decoder_DecodeEnumArray(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg,
|
||||
upb_Array* arr,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* field,
|
||||
wireval* val) {
|
||||
const upb_MiniTableEnum* e = subs[field->UPB_PRIVATE(submsg_index)].subenum;
|
||||
if (!_upb_Decoder_CheckEnum(d, ptr, msg, e, field, val)) return ptr;
|
||||
void* mem = UPB_PTR_AT(_upb_array_ptr(arr), arr->size * 4, void);
|
||||
arr->size++;
|
||||
memcpy(mem, val, 4);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeFixedPacked(
|
||||
upb_Decoder* d, const char* ptr, upb_Array* arr, wireval* val,
|
||||
const upb_MiniTableField* field, int lg2) {
|
||||
int mask = (1 << lg2) - 1;
|
||||
size_t count = val->size >> lg2;
|
||||
if ((val->size & mask) != 0) {
|
||||
// Length isn't a round multiple of elem size.
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
}
|
||||
_upb_Decoder_Reserve(d, arr, count);
|
||||
void* mem = UPB_PTR_AT(_upb_array_ptr(arr), arr->size << lg2, void);
|
||||
arr->size += count;
|
||||
// Note: if/when the decoder supports multi-buffer input, we will need to
|
||||
// handle buffer seams here.
|
||||
if (_upb_IsLittleEndian()) {
|
||||
ptr = upb_EpsCopyInputStream_Copy(&d->input, ptr, mem, val->size);
|
||||
} else {
|
||||
int delta = upb_EpsCopyInputStream_PushLimit(&d->input, ptr, val->size);
|
||||
char* dst = mem;
|
||||
while (!_upb_Decoder_IsDone(d, &ptr)) {
|
||||
if (lg2 == 2) {
|
||||
ptr = upb_WireReader_ReadFixed32(ptr, dst);
|
||||
dst += 4;
|
||||
} else {
|
||||
UPB_ASSERT(lg2 == 3);
|
||||
ptr = upb_WireReader_ReadFixed64(ptr, dst);
|
||||
dst += 8;
|
||||
}
|
||||
}
|
||||
upb_EpsCopyInputStream_PopLimit(&d->input, ptr, delta);
|
||||
}
|
||||
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeVarintPacked(
|
||||
upb_Decoder* d, const char* ptr, upb_Array* arr, wireval* val,
|
||||
const upb_MiniTableField* field, int lg2) {
|
||||
int scale = 1 << lg2;
|
||||
int saved_limit = upb_EpsCopyInputStream_PushLimit(&d->input, ptr, val->size);
|
||||
char* out = UPB_PTR_AT(_upb_array_ptr(arr), arr->size << lg2, void);
|
||||
while (!_upb_Decoder_IsDone(d, &ptr)) {
|
||||
wireval elem;
|
||||
ptr = _upb_Decoder_DecodeVarint(d, ptr, &elem.uint64_val);
|
||||
_upb_Decoder_Munge(field->UPB_PRIVATE(descriptortype), &elem);
|
||||
if (_upb_Decoder_Reserve(d, arr, 1)) {
|
||||
out = UPB_PTR_AT(_upb_array_ptr(arr), arr->size << lg2, void);
|
||||
}
|
||||
arr->size++;
|
||||
memcpy(out, &elem, scale);
|
||||
out += scale;
|
||||
}
|
||||
upb_EpsCopyInputStream_PopLimit(&d->input, ptr, saved_limit);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
static const char* _upb_Decoder_DecodeEnumPacked(
|
||||
upb_Decoder* d, const char* ptr, upb_Message* msg, upb_Array* arr,
|
||||
const upb_MiniTableSub* subs, const upb_MiniTableField* field,
|
||||
wireval* val) {
|
||||
const upb_MiniTableEnum* e = subs[field->UPB_PRIVATE(submsg_index)].subenum;
|
||||
int saved_limit = upb_EpsCopyInputStream_PushLimit(&d->input, ptr, val->size);
|
||||
char* out = UPB_PTR_AT(_upb_array_ptr(arr), arr->size * 4, void);
|
||||
while (!_upb_Decoder_IsDone(d, &ptr)) {
|
||||
wireval elem;
|
||||
ptr = _upb_Decoder_DecodeVarint(d, ptr, &elem.uint64_val);
|
||||
_upb_Decoder_MungeInt32(&elem);
|
||||
if (!_upb_Decoder_CheckEnum(d, ptr, msg, e, field, &elem)) {
|
||||
continue;
|
||||
}
|
||||
if (_upb_Decoder_Reserve(d, arr, 1)) {
|
||||
out = UPB_PTR_AT(_upb_array_ptr(arr), arr->size * 4, void);
|
||||
}
|
||||
arr->size++;
|
||||
memcpy(out, &elem, 4);
|
||||
out += 4;
|
||||
}
|
||||
upb_EpsCopyInputStream_PopLimit(&d->input, ptr, saved_limit);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
upb_Array* _upb_Decoder_CreateArray(upb_Decoder* d,
|
||||
const upb_MiniTableField* field) {
|
||||
/* Maps descriptor type -> elem_size_lg2. */
|
||||
static const uint8_t kElemSizeLg2[] = {
|
||||
[0] = -1, // invalid descriptor type
|
||||
[kUpb_FieldType_Double] = 3,
|
||||
[kUpb_FieldType_Float] = 2,
|
||||
[kUpb_FieldType_Int64] = 3,
|
||||
[kUpb_FieldType_UInt64] = 3,
|
||||
[kUpb_FieldType_Int32] = 2,
|
||||
[kUpb_FieldType_Fixed64] = 3,
|
||||
[kUpb_FieldType_Fixed32] = 2,
|
||||
[kUpb_FieldType_Bool] = 0,
|
||||
[kUpb_FieldType_String] = UPB_SIZE(3, 4),
|
||||
[kUpb_FieldType_Group] = UPB_SIZE(2, 3),
|
||||
[kUpb_FieldType_Message] = UPB_SIZE(2, 3),
|
||||
[kUpb_FieldType_Bytes] = UPB_SIZE(3, 4),
|
||||
[kUpb_FieldType_UInt32] = 2,
|
||||
[kUpb_FieldType_Enum] = 2,
|
||||
[kUpb_FieldType_SFixed32] = 2,
|
||||
[kUpb_FieldType_SFixed64] = 3,
|
||||
[kUpb_FieldType_SInt32] = 2,
|
||||
[kUpb_FieldType_SInt64] = 3,
|
||||
};
|
||||
|
||||
size_t lg2 = kElemSizeLg2[field->UPB_PRIVATE(descriptortype)];
|
||||
upb_Array* ret = _upb_Array_New(&d->arena, 4, lg2);
|
||||
if (!ret) _upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static const char* _upb_Decoder_DecodeToArray(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* field,
|
||||
wireval* val, int op) {
|
||||
upb_Array** arrp = UPB_PTR_AT(msg, field->offset, void);
|
||||
upb_Array* arr = *arrp;
|
||||
void* mem;
|
||||
|
||||
if (arr) {
|
||||
_upb_Decoder_Reserve(d, arr, 1);
|
||||
} else {
|
||||
arr = _upb_Decoder_CreateArray(d, field);
|
||||
*arrp = arr;
|
||||
}
|
||||
|
||||
switch (op) {
|
||||
case kUpb_DecodeOp_Scalar1Byte:
|
||||
case kUpb_DecodeOp_Scalar4Byte:
|
||||
case kUpb_DecodeOp_Scalar8Byte:
|
||||
/* Append scalar value. */
|
||||
mem = UPB_PTR_AT(_upb_array_ptr(arr), arr->size << op, void);
|
||||
arr->size++;
|
||||
memcpy(mem, val, 1 << op);
|
||||
return ptr;
|
||||
case kUpb_DecodeOp_String:
|
||||
_upb_Decoder_VerifyUtf8(d, ptr, val->size);
|
||||
/* Fallthrough. */
|
||||
case kUpb_DecodeOp_Bytes: {
|
||||
/* Append bytes. */
|
||||
upb_StringView* str = (upb_StringView*)_upb_array_ptr(arr) + arr->size;
|
||||
arr->size++;
|
||||
return _upb_Decoder_ReadString(d, ptr, val->size, str);
|
||||
}
|
||||
case kUpb_DecodeOp_SubMessage: {
|
||||
/* Append submessage / group. */
|
||||
upb_TaggedMessagePtr* target = UPB_PTR_AT(
|
||||
_upb_array_ptr(arr), arr->size * sizeof(void*), upb_TaggedMessagePtr);
|
||||
upb_Message* submsg = _upb_Decoder_NewSubMessage(d, subs, field, target);
|
||||
arr->size++;
|
||||
if (UPB_UNLIKELY(field->UPB_PRIVATE(descriptortype) ==
|
||||
kUpb_FieldType_Group)) {
|
||||
return _upb_Decoder_DecodeKnownGroup(d, ptr, submsg, subs, field);
|
||||
} else {
|
||||
return _upb_Decoder_DecodeSubMessage(d, ptr, submsg, subs, field,
|
||||
val->size);
|
||||
}
|
||||
}
|
||||
case OP_FIXPCK_LG2(2):
|
||||
case OP_FIXPCK_LG2(3):
|
||||
return _upb_Decoder_DecodeFixedPacked(d, ptr, arr, val, field,
|
||||
op - OP_FIXPCK_LG2(0));
|
||||
case OP_VARPCK_LG2(0):
|
||||
case OP_VARPCK_LG2(2):
|
||||
case OP_VARPCK_LG2(3):
|
||||
return _upb_Decoder_DecodeVarintPacked(d, ptr, arr, val, field,
|
||||
op - OP_VARPCK_LG2(0));
|
||||
case kUpb_DecodeOp_Enum:
|
||||
return _upb_Decoder_DecodeEnumArray(d, ptr, msg, arr, subs, field, val);
|
||||
case kUpb_DecodeOp_PackedEnum:
|
||||
return _upb_Decoder_DecodeEnumPacked(d, ptr, msg, arr, subs, field, val);
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
}
|
||||
|
||||
upb_Map* _upb_Decoder_CreateMap(upb_Decoder* d, const upb_MiniTable* entry) {
|
||||
/* Maps descriptor type -> upb map size. */
|
||||
static const uint8_t kSizeInMap[] = {
|
||||
[0] = -1, // invalid descriptor type */
|
||||
[kUpb_FieldType_Double] = 8,
|
||||
[kUpb_FieldType_Float] = 4,
|
||||
[kUpb_FieldType_Int64] = 8,
|
||||
[kUpb_FieldType_UInt64] = 8,
|
||||
[kUpb_FieldType_Int32] = 4,
|
||||
[kUpb_FieldType_Fixed64] = 8,
|
||||
[kUpb_FieldType_Fixed32] = 4,
|
||||
[kUpb_FieldType_Bool] = 1,
|
||||
[kUpb_FieldType_String] = UPB_MAPTYPE_STRING,
|
||||
[kUpb_FieldType_Group] = sizeof(void*),
|
||||
[kUpb_FieldType_Message] = sizeof(void*),
|
||||
[kUpb_FieldType_Bytes] = UPB_MAPTYPE_STRING,
|
||||
[kUpb_FieldType_UInt32] = 4,
|
||||
[kUpb_FieldType_Enum] = 4,
|
||||
[kUpb_FieldType_SFixed32] = 4,
|
||||
[kUpb_FieldType_SFixed64] = 8,
|
||||
[kUpb_FieldType_SInt32] = 4,
|
||||
[kUpb_FieldType_SInt64] = 8,
|
||||
};
|
||||
|
||||
const upb_MiniTableField* key_field = &entry->fields[0];
|
||||
const upb_MiniTableField* val_field = &entry->fields[1];
|
||||
char key_size = kSizeInMap[key_field->UPB_PRIVATE(descriptortype)];
|
||||
char val_size = kSizeInMap[val_field->UPB_PRIVATE(descriptortype)];
|
||||
UPB_ASSERT(key_field->offset == offsetof(upb_MapEntryData, k));
|
||||
UPB_ASSERT(val_field->offset == offsetof(upb_MapEntryData, v));
|
||||
upb_Map* ret = _upb_Map_New(&d->arena, key_size, val_size);
|
||||
if (!ret) _upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static const char* _upb_Decoder_DecodeToMap(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* field,
|
||||
wireval* val) {
|
||||
upb_Map** map_p = UPB_PTR_AT(msg, field->offset, upb_Map*);
|
||||
upb_Map* map = *map_p;
|
||||
upb_MapEntry ent;
|
||||
UPB_ASSERT(upb_MiniTableField_Type(field) == kUpb_FieldType_Message);
|
||||
const upb_MiniTable* entry = subs[field->UPB_PRIVATE(submsg_index)].submsg;
|
||||
|
||||
UPB_ASSERT(entry);
|
||||
UPB_ASSERT(entry->field_count == 2);
|
||||
UPB_ASSERT(!upb_IsRepeatedOrMap(&entry->fields[0]));
|
||||
UPB_ASSERT(!upb_IsRepeatedOrMap(&entry->fields[1]));
|
||||
|
||||
if (!map) {
|
||||
map = _upb_Decoder_CreateMap(d, entry);
|
||||
*map_p = map;
|
||||
}
|
||||
|
||||
// Parse map entry.
|
||||
memset(&ent, 0, sizeof(ent));
|
||||
|
||||
if (entry->fields[1].UPB_PRIVATE(descriptortype) == kUpb_FieldType_Message ||
|
||||
entry->fields[1].UPB_PRIVATE(descriptortype) == kUpb_FieldType_Group) {
|
||||
// Create proactively to handle the case where it doesn't appear.
|
||||
upb_TaggedMessagePtr msg;
|
||||
_upb_Decoder_NewSubMessage(d, entry->subs, &entry->fields[1], &msg);
|
||||
ent.data.v.val = upb_value_uintptr(msg);
|
||||
}
|
||||
|
||||
ptr =
|
||||
_upb_Decoder_DecodeSubMessage(d, ptr, &ent.data, subs, field, val->size);
|
||||
// check if ent had any unknown fields
|
||||
size_t size;
|
||||
upb_Message_GetUnknown(&ent.data, &size);
|
||||
if (size != 0) {
|
||||
char* buf;
|
||||
size_t size;
|
||||
uint32_t tag = ((uint32_t)field->number << 3) | kUpb_WireType_Delimited;
|
||||
upb_EncodeStatus status =
|
||||
upb_Encode(&ent.data, entry, 0, &d->arena, &buf, &size);
|
||||
if (status != kUpb_EncodeStatus_Ok) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
_upb_Decoder_AddUnknownVarints(d, msg, tag, size);
|
||||
if (!_upb_Message_AddUnknown(msg, buf, size, &d->arena)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
} else {
|
||||
if (_upb_Map_Insert(map, &ent.data.k, map->key_size, &ent.data.v,
|
||||
map->val_size,
|
||||
&d->arena) == kUpb_MapInsertStatus_OutOfMemory) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
}
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static const char* _upb_Decoder_DecodeToSubMessage(
|
||||
upb_Decoder* d, const char* ptr, upb_Message* msg,
|
||||
const upb_MiniTableSub* subs, const upb_MiniTableField* field, wireval* val,
|
||||
int op) {
|
||||
void* mem = UPB_PTR_AT(msg, field->offset, void);
|
||||
int type = field->UPB_PRIVATE(descriptortype);
|
||||
|
||||
if (UPB_UNLIKELY(op == kUpb_DecodeOp_Enum) &&
|
||||
!_upb_Decoder_CheckEnum(d, ptr, msg,
|
||||
subs[field->UPB_PRIVATE(submsg_index)].subenum,
|
||||
field, val)) {
|
||||
return ptr;
|
||||
}
|
||||
|
||||
/* Set presence if necessary. */
|
||||
if (field->presence > 0) {
|
||||
_upb_sethas_field(msg, field);
|
||||
} else if (field->presence < 0) {
|
||||
/* Oneof case */
|
||||
uint32_t* oneof_case = _upb_oneofcase_field(msg, field);
|
||||
if (op == kUpb_DecodeOp_SubMessage && *oneof_case != field->number) {
|
||||
memset(mem, 0, sizeof(void*));
|
||||
}
|
||||
*oneof_case = field->number;
|
||||
}
|
||||
|
||||
/* Store into message. */
|
||||
switch (op) {
|
||||
case kUpb_DecodeOp_SubMessage: {
|
||||
upb_TaggedMessagePtr* submsgp = mem;
|
||||
upb_Message* submsg;
|
||||
if (*submsgp) {
|
||||
submsg = _upb_Decoder_ReuseSubMessage(d, subs, field, submsgp);
|
||||
} else {
|
||||
submsg = _upb_Decoder_NewSubMessage(d, subs, field, submsgp);
|
||||
}
|
||||
if (UPB_UNLIKELY(type == kUpb_FieldType_Group)) {
|
||||
ptr = _upb_Decoder_DecodeKnownGroup(d, ptr, submsg, subs, field);
|
||||
} else {
|
||||
ptr = _upb_Decoder_DecodeSubMessage(d, ptr, submsg, subs, field,
|
||||
val->size);
|
||||
}
|
||||
break;
|
||||
}
|
||||
case kUpb_DecodeOp_String:
|
||||
_upb_Decoder_VerifyUtf8(d, ptr, val->size);
|
||||
/* Fallthrough. */
|
||||
case kUpb_DecodeOp_Bytes:
|
||||
return _upb_Decoder_ReadString(d, ptr, val->size, mem);
|
||||
case kUpb_DecodeOp_Scalar8Byte:
|
||||
memcpy(mem, val, 8);
|
||||
break;
|
||||
case kUpb_DecodeOp_Enum:
|
||||
case kUpb_DecodeOp_Scalar4Byte:
|
||||
memcpy(mem, val, 4);
|
||||
break;
|
||||
case kUpb_DecodeOp_Scalar1Byte:
|
||||
memcpy(mem, val, 1);
|
||||
break;
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
const char* _upb_Decoder_CheckRequired(upb_Decoder* d, const char* ptr,
|
||||
const upb_Message* msg,
|
||||
const upb_MiniTable* l) {
|
||||
UPB_ASSERT(l->required_count);
|
||||
if (UPB_LIKELY((d->options & kUpb_DecodeOption_CheckRequired) == 0)) {
|
||||
return ptr;
|
||||
}
|
||||
uint64_t msg_head;
|
||||
memcpy(&msg_head, msg, 8);
|
||||
msg_head = _upb_BigEndian_Swap64(msg_head);
|
||||
if (upb_MiniTable_requiredmask(l) & ~msg_head) {
|
||||
d->missing_required = true;
|
||||
}
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static bool _upb_Decoder_TryFastDispatch(upb_Decoder* d, const char** ptr,
|
||||
upb_Message* msg,
|
||||
const upb_MiniTable* layout) {
|
||||
#if UPB_FASTTABLE
|
||||
if (layout && layout->table_mask != (unsigned char)-1) {
|
||||
uint16_t tag = _upb_FastDecoder_LoadTag(*ptr);
|
||||
intptr_t table = decode_totable(layout);
|
||||
*ptr = _upb_FastDecoder_TagDispatch(d, *ptr, msg, table, 0, tag);
|
||||
return true;
|
||||
}
|
||||
#endif
|
||||
return false;
|
||||
}
|
||||
|
||||
static const char* upb_Decoder_SkipField(upb_Decoder* d, const char* ptr,
|
||||
uint32_t tag) {
|
||||
int field_number = tag >> 3;
|
||||
int wire_type = tag & 7;
|
||||
switch (wire_type) {
|
||||
case kUpb_WireType_Varint: {
|
||||
uint64_t val;
|
||||
return _upb_Decoder_DecodeVarint(d, ptr, &val);
|
||||
}
|
||||
case kUpb_WireType_64Bit:
|
||||
return ptr + 8;
|
||||
case kUpb_WireType_32Bit:
|
||||
return ptr + 4;
|
||||
case kUpb_WireType_Delimited: {
|
||||
uint32_t size;
|
||||
ptr = upb_Decoder_DecodeSize(d, ptr, &size);
|
||||
return ptr + size;
|
||||
}
|
||||
case kUpb_WireType_StartGroup:
|
||||
return _upb_Decoder_DecodeUnknownGroup(d, ptr, field_number);
|
||||
default:
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
}
|
||||
}
|
||||
|
||||
enum {
|
||||
kStartItemTag = ((kUpb_MsgSet_Item << 3) | kUpb_WireType_StartGroup),
|
||||
kEndItemTag = ((kUpb_MsgSet_Item << 3) | kUpb_WireType_EndGroup),
|
||||
kTypeIdTag = ((kUpb_MsgSet_TypeId << 3) | kUpb_WireType_Varint),
|
||||
kMessageTag = ((kUpb_MsgSet_Message << 3) | kUpb_WireType_Delimited),
|
||||
};
|
||||
|
||||
static void upb_Decoder_AddKnownMessageSetItem(
|
||||
upb_Decoder* d, upb_Message* msg, const upb_MiniTableExtension* item_mt,
|
||||
const char* data, uint32_t size) {
|
||||
upb_Message_Extension* ext =
|
||||
_upb_Message_GetOrCreateExtension(msg, item_mt, &d->arena);
|
||||
if (UPB_UNLIKELY(!ext)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
upb_Message* submsg = _upb_Decoder_NewSubMessage(
|
||||
d, &ext->ext->sub, &ext->ext->field, (upb_TaggedMessagePtr*)&ext->data);
|
||||
upb_DecodeStatus status = upb_Decode(data, size, submsg, item_mt->sub.submsg,
|
||||
d->extreg, d->options, &d->arena);
|
||||
if (status != kUpb_DecodeStatus_Ok) _upb_Decoder_ErrorJmp(d, status);
|
||||
}
|
||||
|
||||
static void upb_Decoder_AddUnknownMessageSetItem(upb_Decoder* d,
|
||||
upb_Message* msg,
|
||||
uint32_t type_id,
|
||||
const char* message_data,
|
||||
uint32_t message_size) {
|
||||
char buf[60];
|
||||
char* ptr = buf;
|
||||
ptr = upb_Decoder_EncodeVarint32(kStartItemTag, ptr);
|
||||
ptr = upb_Decoder_EncodeVarint32(kTypeIdTag, ptr);
|
||||
ptr = upb_Decoder_EncodeVarint32(type_id, ptr);
|
||||
ptr = upb_Decoder_EncodeVarint32(kMessageTag, ptr);
|
||||
ptr = upb_Decoder_EncodeVarint32(message_size, ptr);
|
||||
char* split = ptr;
|
||||
|
||||
ptr = upb_Decoder_EncodeVarint32(kEndItemTag, ptr);
|
||||
char* end = ptr;
|
||||
|
||||
if (!_upb_Message_AddUnknown(msg, buf, split - buf, &d->arena) ||
|
||||
!_upb_Message_AddUnknown(msg, message_data, message_size, &d->arena) ||
|
||||
!_upb_Message_AddUnknown(msg, split, end - split, &d->arena)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
}
|
||||
|
||||
static void upb_Decoder_AddMessageSetItem(upb_Decoder* d, upb_Message* msg,
|
||||
const upb_MiniTable* t,
|
||||
uint32_t type_id, const char* data,
|
||||
uint32_t size) {
|
||||
const upb_MiniTableExtension* item_mt =
|
||||
upb_ExtensionRegistry_Lookup(d->extreg, t, type_id);
|
||||
if (item_mt) {
|
||||
upb_Decoder_AddKnownMessageSetItem(d, msg, item_mt, data, size);
|
||||
} else {
|
||||
upb_Decoder_AddUnknownMessageSetItem(d, msg, type_id, data, size);
|
||||
}
|
||||
}
|
||||
|
||||
static const char* upb_Decoder_DecodeMessageSetItem(
|
||||
upb_Decoder* d, const char* ptr, upb_Message* msg,
|
||||
const upb_MiniTable* layout) {
|
||||
uint32_t type_id = 0;
|
||||
upb_StringView preserved = {NULL, 0};
|
||||
typedef enum {
|
||||
kUpb_HaveId = 1 << 0,
|
||||
kUpb_HavePayload = 1 << 1,
|
||||
} StateMask;
|
||||
StateMask state_mask = 0;
|
||||
while (!_upb_Decoder_IsDone(d, &ptr)) {
|
||||
uint32_t tag;
|
||||
ptr = _upb_Decoder_DecodeTag(d, ptr, &tag);
|
||||
switch (tag) {
|
||||
case kEndItemTag:
|
||||
return ptr;
|
||||
case kTypeIdTag: {
|
||||
uint64_t tmp;
|
||||
ptr = _upb_Decoder_DecodeVarint(d, ptr, &tmp);
|
||||
if (state_mask & kUpb_HaveId) break; // Ignore dup.
|
||||
state_mask |= kUpb_HaveId;
|
||||
type_id = tmp;
|
||||
if (state_mask & kUpb_HavePayload) {
|
||||
upb_Decoder_AddMessageSetItem(d, msg, layout, type_id, preserved.data,
|
||||
preserved.size);
|
||||
}
|
||||
break;
|
||||
}
|
||||
case kMessageTag: {
|
||||
uint32_t size;
|
||||
ptr = upb_Decoder_DecodeSize(d, ptr, &size);
|
||||
const char* data = ptr;
|
||||
ptr += size;
|
||||
if (state_mask & kUpb_HavePayload) break; // Ignore dup.
|
||||
state_mask |= kUpb_HavePayload;
|
||||
if (state_mask & kUpb_HaveId) {
|
||||
upb_Decoder_AddMessageSetItem(d, msg, layout, type_id, data, size);
|
||||
} else {
|
||||
// Out of order, we must preserve the payload.
|
||||
preserved.data = data;
|
||||
preserved.size = size;
|
||||
}
|
||||
break;
|
||||
}
|
||||
default:
|
||||
// We do not preserve unexpected fields inside a message set item.
|
||||
ptr = upb_Decoder_SkipField(d, ptr, tag);
|
||||
break;
|
||||
}
|
||||
}
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
}
|
||||
|
||||
static const upb_MiniTableField* _upb_Decoder_FindField(upb_Decoder* d,
|
||||
const upb_MiniTable* t,
|
||||
uint32_t field_number,
|
||||
int* last_field_index) {
|
||||
static upb_MiniTableField none = {
|
||||
0, 0, 0, 0, kUpb_FakeFieldType_FieldNotFound, 0};
|
||||
if (t == NULL) return &none;
|
||||
|
||||
size_t idx = ((size_t)field_number) - 1; // 0 wraps to SIZE_MAX
|
||||
if (idx < t->dense_below) {
|
||||
/* Fastest case: index into dense fields. */
|
||||
goto found;
|
||||
}
|
||||
|
||||
if (t->dense_below < t->field_count) {
|
||||
/* Linear search non-dense fields. Resume scanning from last_field_index
|
||||
* since fields are usually in order. */
|
||||
size_t last = *last_field_index;
|
||||
for (idx = last; idx < t->field_count; idx++) {
|
||||
if (t->fields[idx].number == field_number) {
|
||||
goto found;
|
||||
}
|
||||
}
|
||||
|
||||
for (idx = t->dense_below; idx < last; idx++) {
|
||||
if (t->fields[idx].number == field_number) {
|
||||
goto found;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (d->extreg) {
|
||||
switch (t->ext) {
|
||||
case kUpb_ExtMode_Extendable: {
|
||||
const upb_MiniTableExtension* ext =
|
||||
upb_ExtensionRegistry_Lookup(d->extreg, t, field_number);
|
||||
if (ext) return &ext->field;
|
||||
break;
|
||||
}
|
||||
case kUpb_ExtMode_IsMessageSet:
|
||||
if (field_number == kUpb_MsgSet_Item) {
|
||||
static upb_MiniTableField item = {
|
||||
0, 0, 0, 0, kUpb_FakeFieldType_MessageSetItem, 0};
|
||||
return &item;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return &none; /* Unknown field. */
|
||||
|
||||
found:
|
||||
UPB_ASSERT(t->fields[idx].number == field_number);
|
||||
*last_field_index = idx;
|
||||
return &t->fields[idx];
|
||||
}
|
||||
|
||||
int _upb_Decoder_GetVarintOp(const upb_MiniTableField* field) {
|
||||
static const int8_t kVarintOps[] = {
|
||||
[kUpb_FakeFieldType_FieldNotFound] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Double] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Float] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Int64] = kUpb_DecodeOp_Scalar8Byte,
|
||||
[kUpb_FieldType_UInt64] = kUpb_DecodeOp_Scalar8Byte,
|
||||
[kUpb_FieldType_Int32] = kUpb_DecodeOp_Scalar4Byte,
|
||||
[kUpb_FieldType_Fixed64] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Fixed32] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Bool] = kUpb_DecodeOp_Scalar1Byte,
|
||||
[kUpb_FieldType_String] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Group] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Message] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Bytes] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_UInt32] = kUpb_DecodeOp_Scalar4Byte,
|
||||
[kUpb_FieldType_Enum] = kUpb_DecodeOp_Enum,
|
||||
[kUpb_FieldType_SFixed32] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_SFixed64] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_SInt32] = kUpb_DecodeOp_Scalar4Byte,
|
||||
[kUpb_FieldType_SInt64] = kUpb_DecodeOp_Scalar8Byte,
|
||||
[kUpb_FakeFieldType_MessageSetItem] = kUpb_DecodeOp_UnknownField,
|
||||
};
|
||||
|
||||
return kVarintOps[field->UPB_PRIVATE(descriptortype)];
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static void _upb_Decoder_CheckUnlinked(upb_Decoder* d, const upb_MiniTable* mt,
|
||||
const upb_MiniTableField* field,
|
||||
int* op) {
|
||||
// If sub-message is not linked, treat as unknown.
|
||||
if (field->mode & kUpb_LabelFlags_IsExtension) return;
|
||||
const upb_MiniTableSub* sub = &mt->subs[field->UPB_PRIVATE(submsg_index)];
|
||||
if ((d->options & kUpb_DecodeOption_ExperimentalAllowUnlinked) ||
|
||||
sub->submsg != &_kUpb_MiniTable_Empty) {
|
||||
return;
|
||||
}
|
||||
#ifndef NDEBUG
|
||||
const upb_MiniTableField* oneof = upb_MiniTable_GetOneof(mt, field);
|
||||
if (oneof) {
|
||||
// All other members of the oneof must be message fields that are also
|
||||
// unlinked.
|
||||
do {
|
||||
UPB_ASSERT(upb_MiniTableField_CType(oneof) == kUpb_CType_Message);
|
||||
const upb_MiniTableSub* oneof_sub =
|
||||
&mt->subs[oneof->UPB_PRIVATE(submsg_index)];
|
||||
UPB_ASSERT(!oneof_sub);
|
||||
} while (upb_MiniTable_NextOneofField(mt, &oneof));
|
||||
}
|
||||
#endif // NDEBUG
|
||||
*op = kUpb_DecodeOp_UnknownField;
|
||||
}
|
||||
|
||||
int _upb_Decoder_GetDelimitedOp(upb_Decoder* d, const upb_MiniTable* mt,
|
||||
const upb_MiniTableField* field) {
|
||||
enum { kRepeatedBase = 19 };
|
||||
|
||||
static const int8_t kDelimitedOps[] = {
|
||||
/* For non-repeated field type. */
|
||||
[kUpb_FakeFieldType_FieldNotFound] =
|
||||
kUpb_DecodeOp_UnknownField, // Field not found.
|
||||
[kUpb_FieldType_Double] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Float] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Int64] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_UInt64] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Int32] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Fixed64] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Fixed32] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Bool] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_String] = kUpb_DecodeOp_String,
|
||||
[kUpb_FieldType_Group] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Message] = kUpb_DecodeOp_SubMessage,
|
||||
[kUpb_FieldType_Bytes] = kUpb_DecodeOp_Bytes,
|
||||
[kUpb_FieldType_UInt32] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_Enum] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_SFixed32] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_SFixed64] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_SInt32] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FieldType_SInt64] = kUpb_DecodeOp_UnknownField,
|
||||
[kUpb_FakeFieldType_MessageSetItem] = kUpb_DecodeOp_UnknownField,
|
||||
// For repeated field type. */
|
||||
[kRepeatedBase + kUpb_FieldType_Double] = OP_FIXPCK_LG2(3),
|
||||
[kRepeatedBase + kUpb_FieldType_Float] = OP_FIXPCK_LG2(2),
|
||||
[kRepeatedBase + kUpb_FieldType_Int64] = OP_VARPCK_LG2(3),
|
||||
[kRepeatedBase + kUpb_FieldType_UInt64] = OP_VARPCK_LG2(3),
|
||||
[kRepeatedBase + kUpb_FieldType_Int32] = OP_VARPCK_LG2(2),
|
||||
[kRepeatedBase + kUpb_FieldType_Fixed64] = OP_FIXPCK_LG2(3),
|
||||
[kRepeatedBase + kUpb_FieldType_Fixed32] = OP_FIXPCK_LG2(2),
|
||||
[kRepeatedBase + kUpb_FieldType_Bool] = OP_VARPCK_LG2(0),
|
||||
[kRepeatedBase + kUpb_FieldType_String] = kUpb_DecodeOp_String,
|
||||
[kRepeatedBase + kUpb_FieldType_Group] = kUpb_DecodeOp_SubMessage,
|
||||
[kRepeatedBase + kUpb_FieldType_Message] = kUpb_DecodeOp_SubMessage,
|
||||
[kRepeatedBase + kUpb_FieldType_Bytes] = kUpb_DecodeOp_Bytes,
|
||||
[kRepeatedBase + kUpb_FieldType_UInt32] = OP_VARPCK_LG2(2),
|
||||
[kRepeatedBase + kUpb_FieldType_Enum] = kUpb_DecodeOp_PackedEnum,
|
||||
[kRepeatedBase + kUpb_FieldType_SFixed32] = OP_FIXPCK_LG2(2),
|
||||
[kRepeatedBase + kUpb_FieldType_SFixed64] = OP_FIXPCK_LG2(3),
|
||||
[kRepeatedBase + kUpb_FieldType_SInt32] = OP_VARPCK_LG2(2),
|
||||
[kRepeatedBase + kUpb_FieldType_SInt64] = OP_VARPCK_LG2(3),
|
||||
// Omitting kUpb_FakeFieldType_MessageSetItem, because we never emit a
|
||||
// repeated msgset type
|
||||
};
|
||||
|
||||
int ndx = field->UPB_PRIVATE(descriptortype);
|
||||
if (upb_FieldMode_Get(field) == kUpb_FieldMode_Array) ndx += kRepeatedBase;
|
||||
int op = kDelimitedOps[ndx];
|
||||
|
||||
if (op == kUpb_DecodeOp_SubMessage) {
|
||||
_upb_Decoder_CheckUnlinked(d, mt, field, &op);
|
||||
}
|
||||
|
||||
return op;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeWireValue(upb_Decoder* d, const char* ptr,
|
||||
const upb_MiniTable* mt,
|
||||
const upb_MiniTableField* field,
|
||||
int wire_type, wireval* val,
|
||||
int* op) {
|
||||
static const unsigned kFixed32OkMask = (1 << kUpb_FieldType_Float) |
|
||||
(1 << kUpb_FieldType_Fixed32) |
|
||||
(1 << kUpb_FieldType_SFixed32);
|
||||
|
||||
static const unsigned kFixed64OkMask = (1 << kUpb_FieldType_Double) |
|
||||
(1 << kUpb_FieldType_Fixed64) |
|
||||
(1 << kUpb_FieldType_SFixed64);
|
||||
|
||||
switch (wire_type) {
|
||||
case kUpb_WireType_Varint:
|
||||
ptr = _upb_Decoder_DecodeVarint(d, ptr, &val->uint64_val);
|
||||
*op = _upb_Decoder_GetVarintOp(field);
|
||||
_upb_Decoder_Munge(field->UPB_PRIVATE(descriptortype), val);
|
||||
return ptr;
|
||||
case kUpb_WireType_32Bit:
|
||||
*op = kUpb_DecodeOp_Scalar4Byte;
|
||||
if (((1 << field->UPB_PRIVATE(descriptortype)) & kFixed32OkMask) == 0) {
|
||||
*op = kUpb_DecodeOp_UnknownField;
|
||||
}
|
||||
return upb_WireReader_ReadFixed32(ptr, &val->uint32_val);
|
||||
case kUpb_WireType_64Bit:
|
||||
*op = kUpb_DecodeOp_Scalar8Byte;
|
||||
if (((1 << field->UPB_PRIVATE(descriptortype)) & kFixed64OkMask) == 0) {
|
||||
*op = kUpb_DecodeOp_UnknownField;
|
||||
}
|
||||
return upb_WireReader_ReadFixed64(ptr, &val->uint64_val);
|
||||
case kUpb_WireType_Delimited:
|
||||
ptr = upb_Decoder_DecodeSize(d, ptr, &val->size);
|
||||
*op = _upb_Decoder_GetDelimitedOp(d, mt, field);
|
||||
return ptr;
|
||||
case kUpb_WireType_StartGroup:
|
||||
val->uint32_val = field->number;
|
||||
if (field->UPB_PRIVATE(descriptortype) == kUpb_FieldType_Group) {
|
||||
*op = kUpb_DecodeOp_SubMessage;
|
||||
_upb_Decoder_CheckUnlinked(d, mt, field, op);
|
||||
} else if (field->UPB_PRIVATE(descriptortype) ==
|
||||
kUpb_FakeFieldType_MessageSetItem) {
|
||||
*op = kUpb_DecodeOp_MessageSetItem;
|
||||
} else {
|
||||
*op = kUpb_DecodeOp_UnknownField;
|
||||
}
|
||||
return ptr;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* _upb_Decoder_DecodeKnownField(
|
||||
upb_Decoder* d, const char* ptr, upb_Message* msg,
|
||||
const upb_MiniTable* layout, const upb_MiniTableField* field, int op,
|
||||
wireval* val) {
|
||||
const upb_MiniTableSub* subs = layout->subs;
|
||||
uint8_t mode = field->mode;
|
||||
|
||||
if (UPB_UNLIKELY(mode & kUpb_LabelFlags_IsExtension)) {
|
||||
const upb_MiniTableExtension* ext_layout =
|
||||
(const upb_MiniTableExtension*)field;
|
||||
upb_Message_Extension* ext =
|
||||
_upb_Message_GetOrCreateExtension(msg, ext_layout, &d->arena);
|
||||
if (UPB_UNLIKELY(!ext)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
d->unknown_msg = msg;
|
||||
msg = &ext->data;
|
||||
subs = &ext->ext->sub;
|
||||
}
|
||||
|
||||
switch (mode & kUpb_FieldMode_Mask) {
|
||||
case kUpb_FieldMode_Array:
|
||||
return _upb_Decoder_DecodeToArray(d, ptr, msg, subs, field, val, op);
|
||||
case kUpb_FieldMode_Map:
|
||||
return _upb_Decoder_DecodeToMap(d, ptr, msg, subs, field, val);
|
||||
case kUpb_FieldMode_Scalar:
|
||||
return _upb_Decoder_DecodeToSubMessage(d, ptr, msg, subs, field, val, op);
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
}
|
||||
|
||||
static const char* _upb_Decoder_ReverseSkipVarint(const char* ptr,
|
||||
uint32_t val) {
|
||||
uint32_t seen = 0;
|
||||
do {
|
||||
ptr--;
|
||||
seen <<= 7;
|
||||
seen |= *ptr & 0x7f;
|
||||
} while (seen != val);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static const char* _upb_Decoder_DecodeUnknownField(upb_Decoder* d,
|
||||
const char* ptr,
|
||||
upb_Message* msg,
|
||||
int field_number,
|
||||
int wire_type, wireval val) {
|
||||
if (field_number == 0) _upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
|
||||
// Since unknown fields are the uncommon case, we do a little extra work here
|
||||
// to walk backwards through the buffer to find the field start. This frees
|
||||
// up a register in the fast paths (when the field is known), which leads to
|
||||
// significant speedups in benchmarks.
|
||||
const char* start = ptr;
|
||||
|
||||
if (wire_type == kUpb_WireType_Delimited) ptr += val.size;
|
||||
if (msg) {
|
||||
switch (wire_type) {
|
||||
case kUpb_WireType_Varint:
|
||||
case kUpb_WireType_Delimited:
|
||||
start--;
|
||||
while (start[-1] & 0x80) start--;
|
||||
break;
|
||||
case kUpb_WireType_32Bit:
|
||||
start -= 4;
|
||||
break;
|
||||
case kUpb_WireType_64Bit:
|
||||
start -= 8;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
assert(start == d->debug_valstart);
|
||||
uint32_t tag = ((uint32_t)field_number << 3) | wire_type;
|
||||
start = _upb_Decoder_ReverseSkipVarint(start, tag);
|
||||
assert(start == d->debug_tagstart);
|
||||
|
||||
if (wire_type == kUpb_WireType_StartGroup) {
|
||||
d->unknown = start;
|
||||
d->unknown_msg = msg;
|
||||
ptr = _upb_Decoder_DecodeUnknownGroup(d, ptr, field_number);
|
||||
start = d->unknown;
|
||||
d->unknown = NULL;
|
||||
}
|
||||
if (!_upb_Message_AddUnknown(msg, start, ptr - start, &d->arena)) {
|
||||
_upb_Decoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
} else if (wire_type == kUpb_WireType_StartGroup) {
|
||||
ptr = _upb_Decoder_DecodeUnknownGroup(d, ptr, field_number);
|
||||
}
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
static const char* _upb_Decoder_DecodeMessage(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg,
|
||||
const upb_MiniTable* layout) {
|
||||
int last_field_index = 0;
|
||||
|
||||
#if UPB_FASTTABLE
|
||||
// The first time we want to skip fast dispatch, because we may have just been
|
||||
// invoked by the fast parser to handle a case that it bailed on.
|
||||
if (!_upb_Decoder_IsDone(d, &ptr)) goto nofast;
|
||||
#endif
|
||||
|
||||
while (!_upb_Decoder_IsDone(d, &ptr)) {
|
||||
uint32_t tag;
|
||||
const upb_MiniTableField* field;
|
||||
int field_number;
|
||||
int wire_type;
|
||||
wireval val;
|
||||
int op;
|
||||
|
||||
if (_upb_Decoder_TryFastDispatch(d, &ptr, msg, layout)) break;
|
||||
|
||||
#if UPB_FASTTABLE
|
||||
nofast:
|
||||
#endif
|
||||
|
||||
#ifndef NDEBUG
|
||||
d->debug_tagstart = ptr;
|
||||
#endif
|
||||
|
||||
UPB_ASSERT(ptr < d->input.limit_ptr);
|
||||
ptr = _upb_Decoder_DecodeTag(d, ptr, &tag);
|
||||
field_number = tag >> 3;
|
||||
wire_type = tag & 7;
|
||||
|
||||
#ifndef NDEBUG
|
||||
d->debug_valstart = ptr;
|
||||
#endif
|
||||
|
||||
if (wire_type == kUpb_WireType_EndGroup) {
|
||||
d->end_group = field_number;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
field = _upb_Decoder_FindField(d, layout, field_number, &last_field_index);
|
||||
ptr = _upb_Decoder_DecodeWireValue(d, ptr, layout, field, wire_type, &val,
|
||||
&op);
|
||||
|
||||
if (op >= 0) {
|
||||
ptr = _upb_Decoder_DecodeKnownField(d, ptr, msg, layout, field, op, &val);
|
||||
} else {
|
||||
switch (op) {
|
||||
case kUpb_DecodeOp_UnknownField:
|
||||
ptr = _upb_Decoder_DecodeUnknownField(d, ptr, msg, field_number,
|
||||
wire_type, val);
|
||||
break;
|
||||
case kUpb_DecodeOp_MessageSetItem:
|
||||
ptr = upb_Decoder_DecodeMessageSetItem(d, ptr, msg, layout);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return UPB_UNLIKELY(layout && layout->required_count)
|
||||
? _upb_Decoder_CheckRequired(d, ptr, msg, layout)
|
||||
: ptr;
|
||||
}
|
||||
|
||||
const char* _upb_FastDecoder_DecodeGeneric(struct upb_Decoder* d,
|
||||
const char* ptr, upb_Message* msg,
|
||||
intptr_t table, uint64_t hasbits,
|
||||
uint64_t data) {
|
||||
(void)data;
|
||||
*(uint32_t*)msg |= hasbits;
|
||||
return _upb_Decoder_DecodeMessage(d, ptr, msg, decode_totablep(table));
|
||||
}
|
||||
|
||||
static upb_DecodeStatus _upb_Decoder_DecodeTop(struct upb_Decoder* d,
|
||||
const char* buf, void* msg,
|
||||
const upb_MiniTable* l) {
|
||||
if (!_upb_Decoder_TryFastDispatch(d, &buf, msg, l)) {
|
||||
_upb_Decoder_DecodeMessage(d, buf, msg, l);
|
||||
}
|
||||
if (d->end_group != DECODE_NOGROUP) return kUpb_DecodeStatus_Malformed;
|
||||
if (d->missing_required) return kUpb_DecodeStatus_MissingRequired;
|
||||
return kUpb_DecodeStatus_Ok;
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
const char* _upb_Decoder_IsDoneFallback(upb_EpsCopyInputStream* e,
|
||||
const char* ptr, int overrun) {
|
||||
return _upb_EpsCopyInputStream_IsDoneFallbackInline(
|
||||
e, ptr, overrun, _upb_Decoder_BufferFlipCallback);
|
||||
}
|
||||
|
||||
static upb_DecodeStatus upb_Decoder_Decode(upb_Decoder* const decoder,
|
||||
const char* const buf,
|
||||
void* const msg,
|
||||
const upb_MiniTable* const l,
|
||||
upb_Arena* const arena) {
|
||||
if (UPB_SETJMP(decoder->err) == 0) {
|
||||
decoder->status = _upb_Decoder_DecodeTop(decoder, buf, msg, l);
|
||||
} else {
|
||||
UPB_ASSERT(decoder->status != kUpb_DecodeStatus_Ok);
|
||||
}
|
||||
|
||||
_upb_MemBlock* blocks =
|
||||
upb_Atomic_Load(&decoder->arena.blocks, memory_order_relaxed);
|
||||
arena->head = decoder->arena.head;
|
||||
upb_Atomic_Store(&arena->blocks, blocks, memory_order_relaxed);
|
||||
return decoder->status;
|
||||
}
|
||||
|
||||
upb_DecodeStatus upb_Decode(const char* buf, size_t size, void* msg,
|
||||
const upb_MiniTable* l,
|
||||
const upb_ExtensionRegistry* extreg, int options,
|
||||
upb_Arena* arena) {
|
||||
upb_Decoder decoder;
|
||||
unsigned depth = (unsigned)options >> 16;
|
||||
|
||||
upb_EpsCopyInputStream_Init(&decoder.input, &buf, size,
|
||||
options & kUpb_DecodeOption_AliasString);
|
||||
|
||||
decoder.extreg = extreg;
|
||||
decoder.unknown = NULL;
|
||||
decoder.depth = depth ? depth : kUpb_WireFormat_DefaultDepthLimit;
|
||||
decoder.end_group = DECODE_NOGROUP;
|
||||
decoder.options = (uint16_t)options;
|
||||
decoder.missing_required = false;
|
||||
decoder.status = kUpb_DecodeStatus_Ok;
|
||||
|
||||
// Violating the encapsulation of the arena for performance reasons.
|
||||
// This is a temporary arena that we swap into and swap out of when we are
|
||||
// done. The temporary arena only needs to be able to handle allocation,
|
||||
// not fuse or free, so it does not need many of the members to be initialized
|
||||
// (particularly parent_or_count).
|
||||
_upb_MemBlock* blocks = upb_Atomic_Load(&arena->blocks, memory_order_relaxed);
|
||||
decoder.arena.head = arena->head;
|
||||
decoder.arena.block_alloc = arena->block_alloc;
|
||||
upb_Atomic_Init(&decoder.arena.blocks, blocks);
|
||||
|
||||
return upb_Decoder_Decode(&decoder, buf, msg, l, arena);
|
||||
}
|
||||
|
||||
#undef OP_FIXPCK_LG2
|
||||
#undef OP_VARPCK_LG2
|
||||
132
Pods/gRPC-Core/third_party/upb/upb/wire/decode.h
generated
vendored
Normal file
132
Pods/gRPC-Core/third_party/upb/upb/wire/decode.h
generated
vendored
Normal file
@@ -0,0 +1,132 @@
|
||||
// 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
|
||||
|
||||
// upb_decode: parsing into a upb_Message using a upb_MiniTable.
|
||||
|
||||
#ifndef UPB_WIRE_DECODE_H_
|
||||
#define UPB_WIRE_DECODE_H_
|
||||
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/message.h"
|
||||
#include "upb/mini_table/extension_registry.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
enum {
|
||||
/* If set, strings will alias the input buffer instead of copying into the
|
||||
* arena. */
|
||||
kUpb_DecodeOption_AliasString = 1,
|
||||
|
||||
/* If set, the parse will return failure if any message is missing any
|
||||
* required fields when the message data ends. The parse will still continue,
|
||||
* and the failure will only be reported at the end.
|
||||
*
|
||||
* IMPORTANT CAVEATS:
|
||||
*
|
||||
* 1. This can throw a false positive failure if an incomplete message is seen
|
||||
* on the wire but is later completed when the sub-message occurs again.
|
||||
* For this reason, a second pass is required to verify a failure, to be
|
||||
* truly robust.
|
||||
*
|
||||
* 2. This can return a false success if you are decoding into a message that
|
||||
* already has some sub-message fields present. If the sub-message does
|
||||
* not occur in the binary payload, we will never visit it and discover the
|
||||
* incomplete sub-message. For this reason, this check is only useful for
|
||||
* implemting ParseFromString() semantics. For MergeFromString(), a
|
||||
* post-parse validation step will always be necessary. */
|
||||
kUpb_DecodeOption_CheckRequired = 2,
|
||||
|
||||
/* EXPERIMENTAL:
|
||||
*
|
||||
* If set, the parser will allow parsing of sub-message fields that were not
|
||||
* previously linked using upb_MiniTable_SetSubMessage(). The data will be
|
||||
* parsed into an internal "empty" message type that cannot be accessed
|
||||
* directly, but can be later promoted into the true message type if the
|
||||
* sub-message fields are linked at a later time.
|
||||
*
|
||||
* Users should set this option if they intend to perform dynamic tree shaking
|
||||
* and promoting using the interfaces in message/promote.h. If this option is
|
||||
* enabled, it is important that the resulting messages are only accessed by
|
||||
* code that is aware of promotion rules:
|
||||
*
|
||||
* 1. Message pointers in upb_Message, upb_Array, and upb_Map are represented
|
||||
* by a tagged pointer upb_TaggedMessagePointer. The tag indicates whether
|
||||
* the message uses the internal "empty" type.
|
||||
*
|
||||
* 2. Any code *reading* these message pointers must test whether the "empty"
|
||||
* tag bit is set, using the interfaces in mini_table/types.h. However
|
||||
* writing of message pointers should always use plain upb_Message*, since
|
||||
* users are not allowed to create "empty" messages.
|
||||
*
|
||||
* 3. It is always safe to test whether a field is present or test the array
|
||||
* length; these interfaces will reflect that empty messages are present,
|
||||
* even though their data cannot be accessed without promoting first.
|
||||
*
|
||||
* 4. If a message pointer is indeed tagged as empty, the message may not be
|
||||
* accessed directly, only promoted through the interfaces in
|
||||
* message/promote.h.
|
||||
*
|
||||
* 5. Tagged/empty messages may never be created by the user. They may only
|
||||
* be created by the parser or the message-copying logic in message/copy.h.
|
||||
*/
|
||||
kUpb_DecodeOption_ExperimentalAllowUnlinked = 4,
|
||||
};
|
||||
|
||||
UPB_INLINE uint32_t upb_DecodeOptions_MaxDepth(uint16_t depth) {
|
||||
return (uint32_t)depth << 16;
|
||||
}
|
||||
|
||||
UPB_INLINE uint16_t upb_DecodeOptions_GetMaxDepth(uint32_t options) {
|
||||
return options >> 16;
|
||||
}
|
||||
|
||||
// Enforce an upper bound on recursion depth.
|
||||
UPB_INLINE int upb_Decode_LimitDepth(uint32_t decode_options, uint32_t limit) {
|
||||
uint32_t max_depth = upb_DecodeOptions_GetMaxDepth(decode_options);
|
||||
if (max_depth > limit) max_depth = limit;
|
||||
return upb_DecodeOptions_MaxDepth(max_depth) | (decode_options & 0xffff);
|
||||
}
|
||||
|
||||
typedef enum {
|
||||
kUpb_DecodeStatus_Ok = 0,
|
||||
kUpb_DecodeStatus_Malformed = 1, // Wire format was corrupt
|
||||
kUpb_DecodeStatus_OutOfMemory = 2, // Arena alloc failed
|
||||
kUpb_DecodeStatus_BadUtf8 = 3, // String field had bad UTF-8
|
||||
kUpb_DecodeStatus_MaxDepthExceeded =
|
||||
4, // Exceeded upb_DecodeOptions_MaxDepth
|
||||
|
||||
// kUpb_DecodeOption_CheckRequired failed (see above), but the parse otherwise
|
||||
// succeeded.
|
||||
kUpb_DecodeStatus_MissingRequired = 5,
|
||||
|
||||
// Unlinked sub-message field was present, but
|
||||
// kUpb_DecodeOptions_ExperimentalAllowUnlinked was not specified in the list
|
||||
// of options.
|
||||
kUpb_DecodeStatus_UnlinkedSubMessage = 6,
|
||||
} upb_DecodeStatus;
|
||||
|
||||
UPB_API upb_DecodeStatus upb_Decode(const char* buf, size_t size,
|
||||
upb_Message* msg, const upb_MiniTable* l,
|
||||
const upb_ExtensionRegistry* extreg,
|
||||
int options, upb_Arena* arena);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_WIRE_DECODE_H_ */
|
||||
996
Pods/gRPC-Core/third_party/upb/upb/wire/decode_fast.c
generated
vendored
Normal file
996
Pods/gRPC-Core/third_party/upb/upb/wire/decode_fast.c
generated
vendored
Normal file
@@ -0,0 +1,996 @@
|
||||
// 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
|
||||
|
||||
// Fast decoder: ~3x the speed of decode.c, but requires x86-64/ARM64.
|
||||
// Also the table size grows by 2x.
|
||||
//
|
||||
// Could potentially be ported to other 64-bit archs that pass at least six
|
||||
// arguments in registers and have 8 unused high bits in pointers.
|
||||
//
|
||||
// The overall design is to create specialized functions for every possible
|
||||
// field type (eg. oneof boolean field with a 1 byte tag) and then dispatch
|
||||
// to the specialized function as quickly as possible.
|
||||
|
||||
#include "upb/wire/decode_fast.h"
|
||||
|
||||
#include "upb/message/array.h"
|
||||
#include "upb/message/internal/array.h"
|
||||
#include "upb/message/internal/types.h"
|
||||
#include "upb/wire/internal/decode.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#if UPB_FASTTABLE
|
||||
|
||||
// The standard set of arguments passed to each parsing function.
|
||||
// Thanks to x86-64 calling conventions, these will stay in registers.
|
||||
#define UPB_PARSE_PARAMS \
|
||||
upb_Decoder *d, const char *ptr, upb_Message *msg, intptr_t table, \
|
||||
uint64_t hasbits, uint64_t data
|
||||
|
||||
#define UPB_PARSE_ARGS d, ptr, msg, table, hasbits, data
|
||||
|
||||
#define RETURN_GENERIC(m) \
|
||||
/* Uncomment either of these for debugging purposes. */ \
|
||||
/* fprintf(stderr, m); */ \
|
||||
/*__builtin_trap(); */ \
|
||||
return _upb_FastDecoder_DecodeGeneric(d, ptr, msg, table, hasbits, 0);
|
||||
|
||||
typedef enum {
|
||||
CARD_s = 0, /* Singular (optional, non-repeated) */
|
||||
CARD_o = 1, /* Oneof */
|
||||
CARD_r = 2, /* Repeated */
|
||||
CARD_p = 3 /* Packed Repeated */
|
||||
} upb_card;
|
||||
|
||||
UPB_NOINLINE
|
||||
static const char* fastdecode_isdonefallback(UPB_PARSE_PARAMS) {
|
||||
int overrun = data;
|
||||
ptr = _upb_EpsCopyInputStream_IsDoneFallbackInline(
|
||||
&d->input, ptr, overrun, _upb_Decoder_BufferFlipCallback);
|
||||
data = _upb_FastDecoder_LoadTag(ptr);
|
||||
UPB_MUSTTAIL return _upb_FastDecoder_TagDispatch(UPB_PARSE_ARGS);
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* fastdecode_dispatch(UPB_PARSE_PARAMS) {
|
||||
int overrun;
|
||||
switch (upb_EpsCopyInputStream_IsDoneStatus(&d->input, ptr, &overrun)) {
|
||||
case kUpb_IsDoneStatus_Done:
|
||||
*(uint32_t*)msg |= hasbits; // Sync hasbits.
|
||||
const upb_MiniTable* l = decode_totablep(table);
|
||||
return UPB_UNLIKELY(l->required_count)
|
||||
? _upb_Decoder_CheckRequired(d, ptr, msg, l)
|
||||
: ptr;
|
||||
case kUpb_IsDoneStatus_NotDone:
|
||||
break;
|
||||
case kUpb_IsDoneStatus_NeedFallback:
|
||||
data = overrun;
|
||||
UPB_MUSTTAIL return fastdecode_isdonefallback(UPB_PARSE_ARGS);
|
||||
}
|
||||
|
||||
// Read two bytes of tag data (for a one-byte tag, the high byte is junk).
|
||||
data = _upb_FastDecoder_LoadTag(ptr);
|
||||
UPB_MUSTTAIL return _upb_FastDecoder_TagDispatch(UPB_PARSE_ARGS);
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static bool fastdecode_checktag(uint16_t data, int tagbytes) {
|
||||
if (tagbytes == 1) {
|
||||
return (data & 0xff) == 0;
|
||||
} else {
|
||||
return data == 0;
|
||||
}
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* fastdecode_longsize(const char* ptr, int* size) {
|
||||
int i;
|
||||
UPB_ASSERT(*size & 0x80);
|
||||
*size &= 0xff;
|
||||
for (i = 0; i < 3; i++) {
|
||||
ptr++;
|
||||
size_t byte = (uint8_t)ptr[-1];
|
||||
*size += (byte - 1) << (7 + 7 * i);
|
||||
if (UPB_LIKELY((byte & 0x80) == 0)) return ptr;
|
||||
}
|
||||
ptr++;
|
||||
size_t byte = (uint8_t)ptr[-1];
|
||||
// len is limited by 2gb not 4gb, hence 8 and not 16 as normally expected
|
||||
// for a 32 bit varint.
|
||||
if (UPB_UNLIKELY(byte >= 8)) return NULL;
|
||||
*size += (byte - 1) << 28;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* fastdecode_delimited(
|
||||
upb_Decoder* d, const char* ptr,
|
||||
upb_EpsCopyInputStream_ParseDelimitedFunc* func, void* ctx) {
|
||||
ptr++;
|
||||
|
||||
// Sign-extend so varint greater than one byte becomes negative, causing
|
||||
// fast delimited parse to fail.
|
||||
int len = (int8_t)ptr[-1];
|
||||
|
||||
if (!upb_EpsCopyInputStream_TryParseDelimitedFast(&d->input, &ptr, len, func,
|
||||
ctx)) {
|
||||
// Slow case: Sub-message is >=128 bytes and/or exceeds the current buffer.
|
||||
// If it exceeds the buffer limit, limit/limit_ptr will change during
|
||||
// sub-message parsing, so we need to preserve delta, not limit.
|
||||
if (UPB_UNLIKELY(len & 0x80)) {
|
||||
// Size varint >1 byte (length >= 128).
|
||||
ptr = fastdecode_longsize(ptr, &len);
|
||||
if (!ptr) {
|
||||
// Corrupt wire format: size exceeded INT_MAX.
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
if (!upb_EpsCopyInputStream_CheckSize(&d->input, ptr, len)) {
|
||||
// Corrupt wire format: invalid limit.
|
||||
return NULL;
|
||||
}
|
||||
int delta = upb_EpsCopyInputStream_PushLimit(&d->input, ptr, len);
|
||||
ptr = func(&d->input, ptr, ctx);
|
||||
upb_EpsCopyInputStream_PopLimit(&d->input, ptr, delta);
|
||||
}
|
||||
return ptr;
|
||||
}
|
||||
|
||||
/* singular, oneof, repeated field handling ***********************************/
|
||||
|
||||
typedef struct {
|
||||
upb_Array* arr;
|
||||
void* end;
|
||||
} fastdecode_arr;
|
||||
|
||||
typedef enum {
|
||||
FD_NEXT_ATLIMIT,
|
||||
FD_NEXT_SAMEFIELD,
|
||||
FD_NEXT_OTHERFIELD
|
||||
} fastdecode_next;
|
||||
|
||||
typedef struct {
|
||||
void* dst;
|
||||
fastdecode_next next;
|
||||
uint32_t tag;
|
||||
} fastdecode_nextret;
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static void* fastdecode_resizearr(upb_Decoder* d, void* dst,
|
||||
fastdecode_arr* farr, int valbytes) {
|
||||
if (UPB_UNLIKELY(dst == farr->end)) {
|
||||
size_t old_size = farr->arr->capacity;
|
||||
size_t old_bytes = old_size * valbytes;
|
||||
size_t new_size = old_size * 2;
|
||||
size_t new_bytes = new_size * valbytes;
|
||||
char* old_ptr = _upb_array_ptr(farr->arr);
|
||||
char* new_ptr = upb_Arena_Realloc(&d->arena, old_ptr, old_bytes, new_bytes);
|
||||
uint8_t elem_size_lg2 = __builtin_ctz(valbytes);
|
||||
farr->arr->capacity = new_size;
|
||||
farr->arr->data = _upb_array_tagptr(new_ptr, elem_size_lg2);
|
||||
dst = (void*)(new_ptr + (old_size * valbytes));
|
||||
farr->end = (void*)(new_ptr + (new_size * valbytes));
|
||||
}
|
||||
return dst;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static bool fastdecode_tagmatch(uint32_t tag, uint64_t data, int tagbytes) {
|
||||
if (tagbytes == 1) {
|
||||
return (uint8_t)tag == (uint8_t)data;
|
||||
} else {
|
||||
return (uint16_t)tag == (uint16_t)data;
|
||||
}
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static void fastdecode_commitarr(void* dst, fastdecode_arr* farr,
|
||||
int valbytes) {
|
||||
farr->arr->size =
|
||||
(size_t)((char*)dst - (char*)_upb_array_ptr(farr->arr)) / valbytes;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static fastdecode_nextret fastdecode_nextrepeated(upb_Decoder* d, void* dst,
|
||||
const char** ptr,
|
||||
fastdecode_arr* farr,
|
||||
uint64_t data, int tagbytes,
|
||||
int valbytes) {
|
||||
fastdecode_nextret ret;
|
||||
dst = (char*)dst + valbytes;
|
||||
|
||||
if (UPB_LIKELY(!_upb_Decoder_IsDone(d, ptr))) {
|
||||
ret.tag = _upb_FastDecoder_LoadTag(*ptr);
|
||||
if (fastdecode_tagmatch(ret.tag, data, tagbytes)) {
|
||||
ret.next = FD_NEXT_SAMEFIELD;
|
||||
} else {
|
||||
fastdecode_commitarr(dst, farr, valbytes);
|
||||
ret.next = FD_NEXT_OTHERFIELD;
|
||||
}
|
||||
} else {
|
||||
fastdecode_commitarr(dst, farr, valbytes);
|
||||
ret.next = FD_NEXT_ATLIMIT;
|
||||
}
|
||||
|
||||
ret.dst = dst;
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static void* fastdecode_fieldmem(upb_Message* msg, uint64_t data) {
|
||||
size_t ofs = data >> 48;
|
||||
return (char*)msg + ofs;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static void* fastdecode_getfield(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg, uint64_t* data,
|
||||
uint64_t* hasbits, fastdecode_arr* farr,
|
||||
int valbytes, upb_card card) {
|
||||
switch (card) {
|
||||
case CARD_s: {
|
||||
uint8_t hasbit_index = *data >> 24;
|
||||
// Set hasbit and return pointer to scalar field.
|
||||
*hasbits |= 1ull << hasbit_index;
|
||||
return fastdecode_fieldmem(msg, *data);
|
||||
}
|
||||
case CARD_o: {
|
||||
uint16_t case_ofs = *data >> 32;
|
||||
uint32_t* oneof_case = UPB_PTR_AT(msg, case_ofs, uint32_t);
|
||||
uint8_t field_number = *data >> 24;
|
||||
*oneof_case = field_number;
|
||||
return fastdecode_fieldmem(msg, *data);
|
||||
}
|
||||
case CARD_r: {
|
||||
// Get pointer to upb_Array and allocate/expand if necessary.
|
||||
uint8_t elem_size_lg2 = __builtin_ctz(valbytes);
|
||||
upb_Array** arr_p = fastdecode_fieldmem(msg, *data);
|
||||
char* begin;
|
||||
*(uint32_t*)msg |= *hasbits;
|
||||
*hasbits = 0;
|
||||
if (UPB_LIKELY(!*arr_p)) {
|
||||
farr->arr = _upb_Array_New(&d->arena, 8, elem_size_lg2);
|
||||
*arr_p = farr->arr;
|
||||
} else {
|
||||
farr->arr = *arr_p;
|
||||
}
|
||||
begin = _upb_array_ptr(farr->arr);
|
||||
farr->end = begin + (farr->arr->capacity * valbytes);
|
||||
*data = _upb_FastDecoder_LoadTag(ptr);
|
||||
return begin + (farr->arr->size * valbytes);
|
||||
}
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static bool fastdecode_flippacked(uint64_t* data, int tagbytes) {
|
||||
*data ^= (0x2 ^ 0x0); // Patch data to match packed wiretype.
|
||||
return fastdecode_checktag(*data, tagbytes);
|
||||
}
|
||||
|
||||
#define FASTDECODE_CHECKPACKED(tagbytes, card, func) \
|
||||
if (UPB_UNLIKELY(!fastdecode_checktag(data, tagbytes))) { \
|
||||
if (card == CARD_r && fastdecode_flippacked(&data, tagbytes)) { \
|
||||
UPB_MUSTTAIL return func(UPB_PARSE_ARGS); \
|
||||
} \
|
||||
RETURN_GENERIC("packed check tag mismatch\n"); \
|
||||
}
|
||||
|
||||
/* varint fields **************************************************************/
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static uint64_t fastdecode_munge(uint64_t val, int valbytes, bool zigzag) {
|
||||
if (valbytes == 1) {
|
||||
return val != 0;
|
||||
} else if (zigzag) {
|
||||
if (valbytes == 4) {
|
||||
uint32_t n = val;
|
||||
return (n >> 1) ^ -(int32_t)(n & 1);
|
||||
} else if (valbytes == 8) {
|
||||
return (val >> 1) ^ -(int64_t)(val & 1);
|
||||
}
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
return val;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* fastdecode_varint64(const char* ptr, uint64_t* val) {
|
||||
ptr++;
|
||||
*val = (uint8_t)ptr[-1];
|
||||
if (UPB_UNLIKELY(*val & 0x80)) {
|
||||
int i;
|
||||
for (i = 0; i < 8; i++) {
|
||||
ptr++;
|
||||
uint64_t byte = (uint8_t)ptr[-1];
|
||||
*val += (byte - 1) << (7 + 7 * i);
|
||||
if (UPB_LIKELY((byte & 0x80) == 0)) goto done;
|
||||
}
|
||||
ptr++;
|
||||
uint64_t byte = (uint8_t)ptr[-1];
|
||||
if (byte > 1) {
|
||||
return NULL;
|
||||
}
|
||||
*val += (byte - 1) << 63;
|
||||
}
|
||||
done:
|
||||
UPB_ASSUME(ptr != NULL);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
#define FASTDECODE_UNPACKEDVARINT(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, card, zigzag, packed) \
|
||||
uint64_t val; \
|
||||
void* dst; \
|
||||
fastdecode_arr farr; \
|
||||
\
|
||||
FASTDECODE_CHECKPACKED(tagbytes, card, packed); \
|
||||
\
|
||||
dst = fastdecode_getfield(d, ptr, msg, &data, &hasbits, &farr, valbytes, \
|
||||
card); \
|
||||
if (card == CARD_r) { \
|
||||
if (UPB_UNLIKELY(!dst)) { \
|
||||
RETURN_GENERIC("need array resize\n"); \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
again: \
|
||||
if (card == CARD_r) { \
|
||||
dst = fastdecode_resizearr(d, dst, &farr, valbytes); \
|
||||
} \
|
||||
\
|
||||
ptr += tagbytes; \
|
||||
ptr = fastdecode_varint64(ptr, &val); \
|
||||
if (ptr == NULL) _upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed); \
|
||||
val = fastdecode_munge(val, valbytes, zigzag); \
|
||||
memcpy(dst, &val, valbytes); \
|
||||
\
|
||||
if (card == CARD_r) { \
|
||||
fastdecode_nextret ret = fastdecode_nextrepeated( \
|
||||
d, dst, &ptr, &farr, data, tagbytes, valbytes); \
|
||||
switch (ret.next) { \
|
||||
case FD_NEXT_SAMEFIELD: \
|
||||
dst = ret.dst; \
|
||||
goto again; \
|
||||
case FD_NEXT_OTHERFIELD: \
|
||||
data = ret.tag; \
|
||||
UPB_MUSTTAIL return _upb_FastDecoder_TagDispatch(UPB_PARSE_ARGS); \
|
||||
case FD_NEXT_ATLIMIT: \
|
||||
return ptr; \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(UPB_PARSE_ARGS);
|
||||
|
||||
typedef struct {
|
||||
uint8_t valbytes;
|
||||
bool zigzag;
|
||||
void* dst;
|
||||
fastdecode_arr farr;
|
||||
} fastdecode_varintdata;
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* fastdecode_topackedvarint(upb_EpsCopyInputStream* e,
|
||||
const char* ptr, void* ctx) {
|
||||
upb_Decoder* d = (upb_Decoder*)e;
|
||||
fastdecode_varintdata* data = ctx;
|
||||
void* dst = data->dst;
|
||||
uint64_t val;
|
||||
|
||||
while (!_upb_Decoder_IsDone(d, &ptr)) {
|
||||
dst = fastdecode_resizearr(d, dst, &data->farr, data->valbytes);
|
||||
ptr = fastdecode_varint64(ptr, &val);
|
||||
if (ptr == NULL) return NULL;
|
||||
val = fastdecode_munge(val, data->valbytes, data->zigzag);
|
||||
memcpy(dst, &val, data->valbytes);
|
||||
dst = (char*)dst + data->valbytes;
|
||||
}
|
||||
|
||||
fastdecode_commitarr(dst, &data->farr, data->valbytes);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
#define FASTDECODE_PACKEDVARINT(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, zigzag, unpacked) \
|
||||
fastdecode_varintdata ctx = {valbytes, zigzag}; \
|
||||
\
|
||||
FASTDECODE_CHECKPACKED(tagbytes, CARD_r, unpacked); \
|
||||
\
|
||||
ctx.dst = fastdecode_getfield(d, ptr, msg, &data, &hasbits, &ctx.farr, \
|
||||
valbytes, CARD_r); \
|
||||
if (UPB_UNLIKELY(!ctx.dst)) { \
|
||||
RETURN_GENERIC("need array resize\n"); \
|
||||
} \
|
||||
\
|
||||
ptr += tagbytes; \
|
||||
ptr = fastdecode_delimited(d, ptr, &fastdecode_topackedvarint, &ctx); \
|
||||
\
|
||||
if (UPB_UNLIKELY(ptr == NULL)) { \
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed); \
|
||||
} \
|
||||
\
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(d, ptr, msg, table, hasbits, 0);
|
||||
|
||||
#define FASTDECODE_VARINT(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, card, zigzag, unpacked, packed) \
|
||||
if (card == CARD_p) { \
|
||||
FASTDECODE_PACKEDVARINT(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, zigzag, unpacked); \
|
||||
} else { \
|
||||
FASTDECODE_UNPACKEDVARINT(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, card, zigzag, packed); \
|
||||
}
|
||||
|
||||
#define z_ZZ true
|
||||
#define b_ZZ false
|
||||
#define v_ZZ false
|
||||
|
||||
/* Generate all combinations:
|
||||
* {s,o,r,p} x {b1,v4,z4,v8,z8} x {1bt,2bt} */
|
||||
|
||||
#define F(card, type, valbytes, tagbytes) \
|
||||
UPB_NOINLINE \
|
||||
const char* upb_p##card##type##valbytes##_##tagbytes##bt(UPB_PARSE_PARAMS) { \
|
||||
FASTDECODE_VARINT(d, ptr, msg, table, hasbits, data, tagbytes, valbytes, \
|
||||
CARD_##card, type##_ZZ, \
|
||||
upb_pr##type##valbytes##_##tagbytes##bt, \
|
||||
upb_pp##type##valbytes##_##tagbytes##bt); \
|
||||
}
|
||||
|
||||
#define TYPES(card, tagbytes) \
|
||||
F(card, b, 1, tagbytes) \
|
||||
F(card, v, 4, tagbytes) \
|
||||
F(card, v, 8, tagbytes) \
|
||||
F(card, z, 4, tagbytes) \
|
||||
F(card, z, 8, tagbytes)
|
||||
|
||||
#define TAGBYTES(card) \
|
||||
TYPES(card, 1) \
|
||||
TYPES(card, 2)
|
||||
|
||||
TAGBYTES(s)
|
||||
TAGBYTES(o)
|
||||
TAGBYTES(r)
|
||||
TAGBYTES(p)
|
||||
|
||||
#undef z_ZZ
|
||||
#undef b_ZZ
|
||||
#undef v_ZZ
|
||||
#undef o_ONEOF
|
||||
#undef s_ONEOF
|
||||
#undef r_ONEOF
|
||||
#undef F
|
||||
#undef TYPES
|
||||
#undef TAGBYTES
|
||||
#undef FASTDECODE_UNPACKEDVARINT
|
||||
#undef FASTDECODE_PACKEDVARINT
|
||||
#undef FASTDECODE_VARINT
|
||||
|
||||
/* fixed fields ***************************************************************/
|
||||
|
||||
#define FASTDECODE_UNPACKEDFIXED(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, card, packed) \
|
||||
void* dst; \
|
||||
fastdecode_arr farr; \
|
||||
\
|
||||
FASTDECODE_CHECKPACKED(tagbytes, card, packed) \
|
||||
\
|
||||
dst = fastdecode_getfield(d, ptr, msg, &data, &hasbits, &farr, valbytes, \
|
||||
card); \
|
||||
if (card == CARD_r) { \
|
||||
if (UPB_UNLIKELY(!dst)) { \
|
||||
RETURN_GENERIC("couldn't allocate array in arena\n"); \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
again: \
|
||||
if (card == CARD_r) { \
|
||||
dst = fastdecode_resizearr(d, dst, &farr, valbytes); \
|
||||
} \
|
||||
\
|
||||
ptr += tagbytes; \
|
||||
memcpy(dst, ptr, valbytes); \
|
||||
ptr += valbytes; \
|
||||
\
|
||||
if (card == CARD_r) { \
|
||||
fastdecode_nextret ret = fastdecode_nextrepeated( \
|
||||
d, dst, &ptr, &farr, data, tagbytes, valbytes); \
|
||||
switch (ret.next) { \
|
||||
case FD_NEXT_SAMEFIELD: \
|
||||
dst = ret.dst; \
|
||||
goto again; \
|
||||
case FD_NEXT_OTHERFIELD: \
|
||||
data = ret.tag; \
|
||||
UPB_MUSTTAIL return _upb_FastDecoder_TagDispatch(UPB_PARSE_ARGS); \
|
||||
case FD_NEXT_ATLIMIT: \
|
||||
return ptr; \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(UPB_PARSE_ARGS);
|
||||
|
||||
#define FASTDECODE_PACKEDFIXED(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, unpacked) \
|
||||
FASTDECODE_CHECKPACKED(tagbytes, CARD_r, unpacked) \
|
||||
\
|
||||
ptr += tagbytes; \
|
||||
int size = (uint8_t)ptr[0]; \
|
||||
ptr++; \
|
||||
if (size & 0x80) { \
|
||||
ptr = fastdecode_longsize(ptr, &size); \
|
||||
} \
|
||||
\
|
||||
if (UPB_UNLIKELY(!upb_EpsCopyInputStream_CheckDataSizeAvailable( \
|
||||
&d->input, ptr, size) || \
|
||||
(size % valbytes) != 0)) { \
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed); \
|
||||
} \
|
||||
\
|
||||
upb_Array** arr_p = fastdecode_fieldmem(msg, data); \
|
||||
upb_Array* arr = *arr_p; \
|
||||
uint8_t elem_size_lg2 = __builtin_ctz(valbytes); \
|
||||
int elems = size / valbytes; \
|
||||
\
|
||||
if (UPB_LIKELY(!arr)) { \
|
||||
*arr_p = arr = _upb_Array_New(&d->arena, elems, elem_size_lg2); \
|
||||
if (!arr) { \
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed); \
|
||||
} \
|
||||
} else { \
|
||||
_upb_Array_ResizeUninitialized(arr, elems, &d->arena); \
|
||||
} \
|
||||
\
|
||||
char* dst = _upb_array_ptr(arr); \
|
||||
memcpy(dst, ptr, size); \
|
||||
arr->size = elems; \
|
||||
\
|
||||
ptr += size; \
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(UPB_PARSE_ARGS);
|
||||
|
||||
#define FASTDECODE_FIXED(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, card, unpacked, packed) \
|
||||
if (card == CARD_p) { \
|
||||
FASTDECODE_PACKEDFIXED(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, unpacked); \
|
||||
} else { \
|
||||
FASTDECODE_UNPACKEDFIXED(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
valbytes, card, packed); \
|
||||
}
|
||||
|
||||
/* Generate all combinations:
|
||||
* {s,o,r,p} x {f4,f8} x {1bt,2bt} */
|
||||
|
||||
#define F(card, valbytes, tagbytes) \
|
||||
UPB_NOINLINE \
|
||||
const char* upb_p##card##f##valbytes##_##tagbytes##bt(UPB_PARSE_PARAMS) { \
|
||||
FASTDECODE_FIXED(d, ptr, msg, table, hasbits, data, tagbytes, valbytes, \
|
||||
CARD_##card, upb_ppf##valbytes##_##tagbytes##bt, \
|
||||
upb_prf##valbytes##_##tagbytes##bt); \
|
||||
}
|
||||
|
||||
#define TYPES(card, tagbytes) \
|
||||
F(card, 4, tagbytes) \
|
||||
F(card, 8, tagbytes)
|
||||
|
||||
#define TAGBYTES(card) \
|
||||
TYPES(card, 1) \
|
||||
TYPES(card, 2)
|
||||
|
||||
TAGBYTES(s)
|
||||
TAGBYTES(o)
|
||||
TAGBYTES(r)
|
||||
TAGBYTES(p)
|
||||
|
||||
#undef F
|
||||
#undef TYPES
|
||||
#undef TAGBYTES
|
||||
#undef FASTDECODE_UNPACKEDFIXED
|
||||
#undef FASTDECODE_PACKEDFIXED
|
||||
|
||||
/* string fields **************************************************************/
|
||||
|
||||
typedef const char* fastdecode_copystr_func(struct upb_Decoder* d,
|
||||
const char* ptr, upb_Message* msg,
|
||||
const upb_MiniTable* table,
|
||||
uint64_t hasbits,
|
||||
upb_StringView* dst);
|
||||
|
||||
UPB_NOINLINE
|
||||
static const char* fastdecode_verifyutf8(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg, intptr_t table,
|
||||
uint64_t hasbits, uint64_t data) {
|
||||
upb_StringView* dst = (upb_StringView*)data;
|
||||
if (!_upb_Decoder_VerifyUtf8Inline(dst->data, dst->size)) {
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_BadUtf8);
|
||||
}
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(UPB_PARSE_ARGS);
|
||||
}
|
||||
|
||||
#define FASTDECODE_LONGSTRING(d, ptr, msg, table, hasbits, dst, validate_utf8) \
|
||||
int size = (uint8_t)ptr[0]; /* Could plumb through hasbits. */ \
|
||||
ptr++; \
|
||||
if (size & 0x80) { \
|
||||
ptr = fastdecode_longsize(ptr, &size); \
|
||||
} \
|
||||
\
|
||||
if (UPB_UNLIKELY(!upb_EpsCopyInputStream_CheckSize(&d->input, ptr, size))) { \
|
||||
dst->size = 0; \
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed); \
|
||||
} \
|
||||
\
|
||||
const char* s_ptr = ptr; \
|
||||
ptr = upb_EpsCopyInputStream_ReadString(&d->input, &s_ptr, size, &d->arena); \
|
||||
if (!ptr) _upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory); \
|
||||
dst->data = s_ptr; \
|
||||
dst->size = size; \
|
||||
\
|
||||
if (validate_utf8) { \
|
||||
data = (uint64_t)dst; \
|
||||
UPB_MUSTTAIL return fastdecode_verifyutf8(UPB_PARSE_ARGS); \
|
||||
} else { \
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(UPB_PARSE_ARGS); \
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
static const char* fastdecode_longstring_utf8(struct upb_Decoder* d,
|
||||
const char* ptr, upb_Message* msg,
|
||||
intptr_t table, uint64_t hasbits,
|
||||
uint64_t data) {
|
||||
upb_StringView* dst = (upb_StringView*)data;
|
||||
FASTDECODE_LONGSTRING(d, ptr, msg, table, hasbits, dst, true);
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
static const char* fastdecode_longstring_noutf8(
|
||||
struct upb_Decoder* d, const char* ptr, upb_Message* msg, intptr_t table,
|
||||
uint64_t hasbits, uint64_t data) {
|
||||
upb_StringView* dst = (upb_StringView*)data;
|
||||
FASTDECODE_LONGSTRING(d, ptr, msg, table, hasbits, dst, false);
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static void fastdecode_docopy(upb_Decoder* d, const char* ptr, uint32_t size,
|
||||
int copy, char* data, upb_StringView* dst) {
|
||||
d->arena.head.ptr += copy;
|
||||
dst->data = data;
|
||||
UPB_UNPOISON_MEMORY_REGION(data, copy);
|
||||
memcpy(data, ptr, copy);
|
||||
UPB_POISON_MEMORY_REGION(data + size, copy - size);
|
||||
}
|
||||
|
||||
#define FASTDECODE_COPYSTRING(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
card, validate_utf8) \
|
||||
upb_StringView* dst; \
|
||||
fastdecode_arr farr; \
|
||||
int64_t size; \
|
||||
size_t arena_has; \
|
||||
size_t common_has; \
|
||||
char* buf; \
|
||||
\
|
||||
UPB_ASSERT(!upb_EpsCopyInputStream_AliasingAvailable(&d->input, ptr, 0)); \
|
||||
UPB_ASSERT(fastdecode_checktag(data, tagbytes)); \
|
||||
\
|
||||
dst = fastdecode_getfield(d, ptr, msg, &data, &hasbits, &farr, \
|
||||
sizeof(upb_StringView), card); \
|
||||
\
|
||||
again: \
|
||||
if (card == CARD_r) { \
|
||||
dst = fastdecode_resizearr(d, dst, &farr, sizeof(upb_StringView)); \
|
||||
} \
|
||||
\
|
||||
size = (uint8_t)ptr[tagbytes]; \
|
||||
ptr += tagbytes + 1; \
|
||||
dst->size = size; \
|
||||
\
|
||||
buf = d->arena.head.ptr; \
|
||||
arena_has = _upb_ArenaHas(&d->arena); \
|
||||
common_has = UPB_MIN(arena_has, \
|
||||
upb_EpsCopyInputStream_BytesAvailable(&d->input, ptr)); \
|
||||
\
|
||||
if (UPB_LIKELY(size <= 15 - tagbytes)) { \
|
||||
if (arena_has < 16) goto longstr; \
|
||||
d->arena.head.ptr += 16; \
|
||||
memcpy(buf, ptr - tagbytes - 1, 16); \
|
||||
dst->data = buf + tagbytes + 1; \
|
||||
} else if (UPB_LIKELY(size <= 32)) { \
|
||||
if (UPB_UNLIKELY(common_has < 32)) goto longstr; \
|
||||
fastdecode_docopy(d, ptr, size, 32, buf, dst); \
|
||||
} else if (UPB_LIKELY(size <= 64)) { \
|
||||
if (UPB_UNLIKELY(common_has < 64)) goto longstr; \
|
||||
fastdecode_docopy(d, ptr, size, 64, buf, dst); \
|
||||
} else if (UPB_LIKELY(size < 128)) { \
|
||||
if (UPB_UNLIKELY(common_has < 128)) goto longstr; \
|
||||
fastdecode_docopy(d, ptr, size, 128, buf, dst); \
|
||||
} else { \
|
||||
goto longstr; \
|
||||
} \
|
||||
\
|
||||
ptr += size; \
|
||||
\
|
||||
if (card == CARD_r) { \
|
||||
if (validate_utf8 && \
|
||||
!_upb_Decoder_VerifyUtf8Inline(dst->data, dst->size)) { \
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_BadUtf8); \
|
||||
} \
|
||||
fastdecode_nextret ret = fastdecode_nextrepeated( \
|
||||
d, dst, &ptr, &farr, data, tagbytes, sizeof(upb_StringView)); \
|
||||
switch (ret.next) { \
|
||||
case FD_NEXT_SAMEFIELD: \
|
||||
dst = ret.dst; \
|
||||
goto again; \
|
||||
case FD_NEXT_OTHERFIELD: \
|
||||
data = ret.tag; \
|
||||
UPB_MUSTTAIL return _upb_FastDecoder_TagDispatch(UPB_PARSE_ARGS); \
|
||||
case FD_NEXT_ATLIMIT: \
|
||||
return ptr; \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
if (card != CARD_r && validate_utf8) { \
|
||||
data = (uint64_t)dst; \
|
||||
UPB_MUSTTAIL return fastdecode_verifyutf8(UPB_PARSE_ARGS); \
|
||||
} \
|
||||
\
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(UPB_PARSE_ARGS); \
|
||||
\
|
||||
longstr: \
|
||||
if (card == CARD_r) { \
|
||||
fastdecode_commitarr(dst + 1, &farr, sizeof(upb_StringView)); \
|
||||
} \
|
||||
ptr--; \
|
||||
if (validate_utf8) { \
|
||||
UPB_MUSTTAIL return fastdecode_longstring_utf8(d, ptr, msg, table, \
|
||||
hasbits, (uint64_t)dst); \
|
||||
} else { \
|
||||
UPB_MUSTTAIL return fastdecode_longstring_noutf8(d, ptr, msg, table, \
|
||||
hasbits, (uint64_t)dst); \
|
||||
}
|
||||
|
||||
#define FASTDECODE_STRING(d, ptr, msg, table, hasbits, data, tagbytes, card, \
|
||||
copyfunc, validate_utf8) \
|
||||
upb_StringView* dst; \
|
||||
fastdecode_arr farr; \
|
||||
int64_t size; \
|
||||
\
|
||||
if (UPB_UNLIKELY(!fastdecode_checktag(data, tagbytes))) { \
|
||||
RETURN_GENERIC("string field tag mismatch\n"); \
|
||||
} \
|
||||
\
|
||||
if (UPB_UNLIKELY( \
|
||||
!upb_EpsCopyInputStream_AliasingAvailable(&d->input, ptr, 0))) { \
|
||||
UPB_MUSTTAIL return copyfunc(UPB_PARSE_ARGS); \
|
||||
} \
|
||||
\
|
||||
dst = fastdecode_getfield(d, ptr, msg, &data, &hasbits, &farr, \
|
||||
sizeof(upb_StringView), card); \
|
||||
\
|
||||
again: \
|
||||
if (card == CARD_r) { \
|
||||
dst = fastdecode_resizearr(d, dst, &farr, sizeof(upb_StringView)); \
|
||||
} \
|
||||
\
|
||||
size = (int8_t)ptr[tagbytes]; \
|
||||
ptr += tagbytes + 1; \
|
||||
\
|
||||
if (UPB_UNLIKELY( \
|
||||
!upb_EpsCopyInputStream_AliasingAvailable(&d->input, ptr, size))) { \
|
||||
ptr--; \
|
||||
if (validate_utf8) { \
|
||||
return fastdecode_longstring_utf8(d, ptr, msg, table, hasbits, \
|
||||
(uint64_t)dst); \
|
||||
} else { \
|
||||
return fastdecode_longstring_noutf8(d, ptr, msg, table, hasbits, \
|
||||
(uint64_t)dst); \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
dst->data = ptr; \
|
||||
dst->size = size; \
|
||||
ptr = upb_EpsCopyInputStream_ReadStringAliased(&d->input, &dst->data, \
|
||||
dst->size); \
|
||||
\
|
||||
if (card == CARD_r) { \
|
||||
if (validate_utf8 && \
|
||||
!_upb_Decoder_VerifyUtf8Inline(dst->data, dst->size)) { \
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_BadUtf8); \
|
||||
} \
|
||||
fastdecode_nextret ret = fastdecode_nextrepeated( \
|
||||
d, dst, &ptr, &farr, data, tagbytes, sizeof(upb_StringView)); \
|
||||
switch (ret.next) { \
|
||||
case FD_NEXT_SAMEFIELD: \
|
||||
dst = ret.dst; \
|
||||
goto again; \
|
||||
case FD_NEXT_OTHERFIELD: \
|
||||
data = ret.tag; \
|
||||
UPB_MUSTTAIL return _upb_FastDecoder_TagDispatch(UPB_PARSE_ARGS); \
|
||||
case FD_NEXT_ATLIMIT: \
|
||||
return ptr; \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
if (card != CARD_r && validate_utf8) { \
|
||||
data = (uint64_t)dst; \
|
||||
UPB_MUSTTAIL return fastdecode_verifyutf8(UPB_PARSE_ARGS); \
|
||||
} \
|
||||
\
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(UPB_PARSE_ARGS);
|
||||
|
||||
/* Generate all combinations:
|
||||
* {p,c} x {s,o,r} x {s, b} x {1bt,2bt} */
|
||||
|
||||
#define s_VALIDATE true
|
||||
#define b_VALIDATE false
|
||||
|
||||
#define F(card, tagbytes, type) \
|
||||
UPB_NOINLINE \
|
||||
const char* upb_c##card##type##_##tagbytes##bt(UPB_PARSE_PARAMS) { \
|
||||
FASTDECODE_COPYSTRING(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
CARD_##card, type##_VALIDATE); \
|
||||
} \
|
||||
const char* upb_p##card##type##_##tagbytes##bt(UPB_PARSE_PARAMS) { \
|
||||
FASTDECODE_STRING(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
CARD_##card, upb_c##card##type##_##tagbytes##bt, \
|
||||
type##_VALIDATE); \
|
||||
}
|
||||
|
||||
#define UTF8(card, tagbytes) \
|
||||
F(card, tagbytes, s) \
|
||||
F(card, tagbytes, b)
|
||||
|
||||
#define TAGBYTES(card) \
|
||||
UTF8(card, 1) \
|
||||
UTF8(card, 2)
|
||||
|
||||
TAGBYTES(s)
|
||||
TAGBYTES(o)
|
||||
TAGBYTES(r)
|
||||
|
||||
#undef s_VALIDATE
|
||||
#undef b_VALIDATE
|
||||
#undef F
|
||||
#undef TAGBYTES
|
||||
#undef FASTDECODE_LONGSTRING
|
||||
#undef FASTDECODE_COPYSTRING
|
||||
#undef FASTDECODE_STRING
|
||||
|
||||
/* message fields *************************************************************/
|
||||
|
||||
UPB_INLINE
|
||||
upb_Message* decode_newmsg_ceil(upb_Decoder* d, const upb_MiniTable* l,
|
||||
int msg_ceil_bytes) {
|
||||
size_t size = l->size + sizeof(upb_Message_Internal);
|
||||
char* msg_data;
|
||||
if (UPB_LIKELY(msg_ceil_bytes > 0 &&
|
||||
_upb_ArenaHas(&d->arena) >= msg_ceil_bytes)) {
|
||||
UPB_ASSERT(size <= (size_t)msg_ceil_bytes);
|
||||
msg_data = d->arena.head.ptr;
|
||||
d->arena.head.ptr += size;
|
||||
UPB_UNPOISON_MEMORY_REGION(msg_data, msg_ceil_bytes);
|
||||
memset(msg_data, 0, msg_ceil_bytes);
|
||||
UPB_POISON_MEMORY_REGION(msg_data + size, msg_ceil_bytes - size);
|
||||
} else {
|
||||
msg_data = (char*)upb_Arena_Malloc(&d->arena, size);
|
||||
memset(msg_data, 0, size);
|
||||
}
|
||||
return msg_data + sizeof(upb_Message_Internal);
|
||||
}
|
||||
|
||||
typedef struct {
|
||||
intptr_t table;
|
||||
upb_Message* msg;
|
||||
} fastdecode_submsgdata;
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static const char* fastdecode_tosubmsg(upb_EpsCopyInputStream* e,
|
||||
const char* ptr, void* ctx) {
|
||||
upb_Decoder* d = (upb_Decoder*)e;
|
||||
fastdecode_submsgdata* submsg = ctx;
|
||||
ptr = fastdecode_dispatch(d, ptr, submsg->msg, submsg->table, 0, 0);
|
||||
UPB_ASSUME(ptr != NULL);
|
||||
return ptr;
|
||||
}
|
||||
|
||||
#define FASTDECODE_SUBMSG(d, ptr, msg, table, hasbits, data, tagbytes, \
|
||||
msg_ceil_bytes, card) \
|
||||
\
|
||||
if (UPB_UNLIKELY(!fastdecode_checktag(data, tagbytes))) { \
|
||||
RETURN_GENERIC("submessage field tag mismatch\n"); \
|
||||
} \
|
||||
\
|
||||
if (--d->depth == 0) { \
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_MaxDepthExceeded); \
|
||||
} \
|
||||
\
|
||||
upb_Message** dst; \
|
||||
uint32_t submsg_idx = (data >> 16) & 0xff; \
|
||||
const upb_MiniTable* tablep = decode_totablep(table); \
|
||||
const upb_MiniTable* subtablep = tablep->subs[submsg_idx].submsg; \
|
||||
fastdecode_submsgdata submsg = {decode_totable(subtablep)}; \
|
||||
fastdecode_arr farr; \
|
||||
\
|
||||
if (subtablep->table_mask == (uint8_t)-1) { \
|
||||
RETURN_GENERIC("submessage doesn't have fast tables."); \
|
||||
} \
|
||||
\
|
||||
dst = fastdecode_getfield(d, ptr, msg, &data, &hasbits, &farr, \
|
||||
sizeof(upb_Message*), card); \
|
||||
\
|
||||
if (card == CARD_s) { \
|
||||
*(uint32_t*)msg |= hasbits; \
|
||||
hasbits = 0; \
|
||||
} \
|
||||
\
|
||||
again: \
|
||||
if (card == CARD_r) { \
|
||||
dst = fastdecode_resizearr(d, dst, &farr, sizeof(upb_Message*)); \
|
||||
} \
|
||||
\
|
||||
submsg.msg = *dst; \
|
||||
\
|
||||
if (card == CARD_r || UPB_LIKELY(!submsg.msg)) { \
|
||||
*dst = submsg.msg = decode_newmsg_ceil(d, subtablep, msg_ceil_bytes); \
|
||||
} \
|
||||
\
|
||||
ptr += tagbytes; \
|
||||
ptr = fastdecode_delimited(d, ptr, fastdecode_tosubmsg, &submsg); \
|
||||
\
|
||||
if (UPB_UNLIKELY(ptr == NULL || d->end_group != DECODE_NOGROUP)) { \
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed); \
|
||||
} \
|
||||
\
|
||||
if (card == CARD_r) { \
|
||||
fastdecode_nextret ret = fastdecode_nextrepeated( \
|
||||
d, dst, &ptr, &farr, data, tagbytes, sizeof(upb_Message*)); \
|
||||
switch (ret.next) { \
|
||||
case FD_NEXT_SAMEFIELD: \
|
||||
dst = ret.dst; \
|
||||
goto again; \
|
||||
case FD_NEXT_OTHERFIELD: \
|
||||
d->depth++; \
|
||||
data = ret.tag; \
|
||||
UPB_MUSTTAIL return _upb_FastDecoder_TagDispatch(UPB_PARSE_ARGS); \
|
||||
case FD_NEXT_ATLIMIT: \
|
||||
d->depth++; \
|
||||
return ptr; \
|
||||
} \
|
||||
} \
|
||||
\
|
||||
d->depth++; \
|
||||
UPB_MUSTTAIL return fastdecode_dispatch(UPB_PARSE_ARGS);
|
||||
|
||||
#define F(card, tagbytes, size_ceil, ceil_arg) \
|
||||
const char* upb_p##card##m_##tagbytes##bt_max##size_ceil##b( \
|
||||
UPB_PARSE_PARAMS) { \
|
||||
FASTDECODE_SUBMSG(d, ptr, msg, table, hasbits, data, tagbytes, ceil_arg, \
|
||||
CARD_##card); \
|
||||
}
|
||||
|
||||
#define SIZES(card, tagbytes) \
|
||||
F(card, tagbytes, 64, 64) \
|
||||
F(card, tagbytes, 128, 128) \
|
||||
F(card, tagbytes, 192, 192) \
|
||||
F(card, tagbytes, 256, 256) \
|
||||
F(card, tagbytes, max, -1)
|
||||
|
||||
#define TAGBYTES(card) \
|
||||
SIZES(card, 1) \
|
||||
SIZES(card, 2)
|
||||
|
||||
TAGBYTES(s)
|
||||
TAGBYTES(o)
|
||||
TAGBYTES(r)
|
||||
|
||||
#undef TAGBYTES
|
||||
#undef SIZES
|
||||
#undef F
|
||||
#undef FASTDECODE_SUBMSG
|
||||
|
||||
#endif /* UPB_FASTTABLE */
|
||||
147
Pods/gRPC-Core/third_party/upb/upb/wire/decode_fast.h
generated
vendored
Normal file
147
Pods/gRPC-Core/third_party/upb/upb/wire/decode_fast.h
generated
vendored
Normal file
@@ -0,0 +1,147 @@
|
||||
// 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
|
||||
|
||||
// These are the specialized field parser functions for the fast parser.
|
||||
// Generated tables will refer to these by name.
|
||||
//
|
||||
// The function names are encoded with names like:
|
||||
//
|
||||
// // 123 4
|
||||
// upb_pss_1bt(); // Parse singular string, 1 byte tag.
|
||||
//
|
||||
// In position 1:
|
||||
// - 'p' for parse, most function use this
|
||||
// - 'c' for copy, for when we are copying strings instead of aliasing
|
||||
//
|
||||
// In position 2 (cardinality):
|
||||
// - 's' for singular, with or without hasbit
|
||||
// - 'o' for oneof
|
||||
// - 'r' for non-packed repeated
|
||||
// - 'p' for packed repeated
|
||||
//
|
||||
// In position 3 (type):
|
||||
// - 'b1' for bool
|
||||
// - 'v4' for 4-byte varint
|
||||
// - 'v8' for 8-byte varint
|
||||
// - 'z4' for zig-zag-encoded 4-byte varint
|
||||
// - 'z8' for zig-zag-encoded 8-byte varint
|
||||
// - 'f4' for 4-byte fixed
|
||||
// - 'f8' for 8-byte fixed
|
||||
// - 'm' for sub-message
|
||||
// - 's' for string (validate UTF-8)
|
||||
// - 'b' for bytes
|
||||
//
|
||||
// In position 4 (tag length):
|
||||
// - '1' for one-byte tags (field numbers 1-15)
|
||||
// - '2' for two-byte tags (field numbers 16-2048)
|
||||
|
||||
#ifndef UPB_WIRE_DECODE_FAST_H_
|
||||
#define UPB_WIRE_DECODE_FAST_H_
|
||||
|
||||
#include "upb/message/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
struct upb_Decoder;
|
||||
|
||||
// The fallback, generic parsing function that can handle any field type.
|
||||
// This just uses the regular (non-fast) parser to parse a single field.
|
||||
const char* _upb_FastDecoder_DecodeGeneric(struct upb_Decoder* d,
|
||||
const char* ptr, upb_Message* msg,
|
||||
intptr_t table, uint64_t hasbits,
|
||||
uint64_t data);
|
||||
|
||||
#define UPB_PARSE_PARAMS \
|
||||
struct upb_Decoder *d, const char *ptr, upb_Message *msg, intptr_t table, \
|
||||
uint64_t hasbits, uint64_t data
|
||||
|
||||
/* primitive fields ***********************************************************/
|
||||
|
||||
#define F(card, type, valbytes, tagbytes) \
|
||||
const char* upb_p##card##type##valbytes##_##tagbytes##bt(UPB_PARSE_PARAMS);
|
||||
|
||||
#define TYPES(card, tagbytes) \
|
||||
F(card, b, 1, tagbytes) \
|
||||
F(card, v, 4, tagbytes) \
|
||||
F(card, v, 8, tagbytes) \
|
||||
F(card, z, 4, tagbytes) \
|
||||
F(card, z, 8, tagbytes) \
|
||||
F(card, f, 4, tagbytes) \
|
||||
F(card, f, 8, tagbytes)
|
||||
|
||||
#define TAGBYTES(card) \
|
||||
TYPES(card, 1) \
|
||||
TYPES(card, 2)
|
||||
|
||||
TAGBYTES(s)
|
||||
TAGBYTES(o)
|
||||
TAGBYTES(r)
|
||||
TAGBYTES(p)
|
||||
|
||||
#undef F
|
||||
#undef TYPES
|
||||
#undef TAGBYTES
|
||||
|
||||
/* string fields **************************************************************/
|
||||
|
||||
#define F(card, tagbytes, type) \
|
||||
const char* upb_p##card##type##_##tagbytes##bt(UPB_PARSE_PARAMS); \
|
||||
const char* upb_c##card##type##_##tagbytes##bt(UPB_PARSE_PARAMS);
|
||||
|
||||
#define UTF8(card, tagbytes) \
|
||||
F(card, tagbytes, s) \
|
||||
F(card, tagbytes, b)
|
||||
|
||||
#define TAGBYTES(card) \
|
||||
UTF8(card, 1) \
|
||||
UTF8(card, 2)
|
||||
|
||||
TAGBYTES(s)
|
||||
TAGBYTES(o)
|
||||
TAGBYTES(r)
|
||||
|
||||
#undef F
|
||||
#undef TAGBYTES
|
||||
|
||||
/* sub-message fields *********************************************************/
|
||||
|
||||
#define F(card, tagbytes, size_ceil, ceil_arg) \
|
||||
const char* upb_p##card##m_##tagbytes##bt_max##size_ceil##b(UPB_PARSE_PARAMS);
|
||||
|
||||
#define SIZES(card, tagbytes) \
|
||||
F(card, tagbytes, 64, 64) \
|
||||
F(card, tagbytes, 128, 128) \
|
||||
F(card, tagbytes, 192, 192) \
|
||||
F(card, tagbytes, 256, 256) \
|
||||
F(card, tagbytes, max, -1)
|
||||
|
||||
#define TAGBYTES(card) \
|
||||
SIZES(card, 1) \
|
||||
SIZES(card, 2)
|
||||
|
||||
TAGBYTES(s)
|
||||
TAGBYTES(o)
|
||||
TAGBYTES(r)
|
||||
|
||||
#undef TAGBYTES
|
||||
#undef SIZES
|
||||
#undef F
|
||||
|
||||
#undef UPB_PARSE_PARAMS
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_WIRE_DECODE_FAST_H_ */
|
||||
638
Pods/gRPC-Core/third_party/upb/upb/wire/encode.c
generated
vendored
Normal file
638
Pods/gRPC-Core/third_party/upb/upb/wire/encode.c
generated
vendored
Normal file
@@ -0,0 +1,638 @@
|
||||
// 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
|
||||
|
||||
// We encode backwards, to avoid pre-computing lengths (one-pass encode).
|
||||
|
||||
#include "upb/wire/encode.h"
|
||||
|
||||
#include <setjmp.h>
|
||||
#include <stdbool.h>
|
||||
#include <stdint.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/base/descriptor_constants.h"
|
||||
#include "upb/base/string_view.h"
|
||||
#include "upb/hash/common.h"
|
||||
#include "upb/hash/str_table.h"
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/message/array.h"
|
||||
#include "upb/message/internal/accessors.h"
|
||||
#include "upb/message/internal/array.h"
|
||||
#include "upb/message/internal/extension.h"
|
||||
#include "upb/message/internal/map.h"
|
||||
#include "upb/message/internal/map_entry.h"
|
||||
#include "upb/message/internal/map_sorter.h"
|
||||
#include "upb/message/message.h"
|
||||
#include "upb/message/tagged_ptr.h"
|
||||
#include "upb/mini_table/field.h"
|
||||
#include "upb/mini_table/internal/field.h"
|
||||
#include "upb/mini_table/internal/message.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
#include "upb/mini_table/sub.h"
|
||||
#include "upb/wire/internal/constants.h"
|
||||
#include "upb/wire/internal/swap.h"
|
||||
#include "upb/wire/types.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#define UPB_PB_VARINT_MAX_LEN 10
|
||||
|
||||
UPB_NOINLINE
|
||||
static size_t encode_varint64(uint64_t val, char* buf) {
|
||||
size_t i = 0;
|
||||
do {
|
||||
uint8_t byte = val & 0x7fU;
|
||||
val >>= 7;
|
||||
if (val) byte |= 0x80U;
|
||||
buf[i++] = byte;
|
||||
} while (val);
|
||||
return i;
|
||||
}
|
||||
|
||||
static uint32_t encode_zz32(int32_t n) {
|
||||
return ((uint32_t)n << 1) ^ (n >> 31);
|
||||
}
|
||||
static uint64_t encode_zz64(int64_t n) {
|
||||
return ((uint64_t)n << 1) ^ (n >> 63);
|
||||
}
|
||||
|
||||
typedef struct {
|
||||
upb_EncodeStatus status;
|
||||
jmp_buf err;
|
||||
upb_Arena* arena;
|
||||
char *buf, *ptr, *limit;
|
||||
int options;
|
||||
int depth;
|
||||
_upb_mapsorter sorter;
|
||||
} upb_encstate;
|
||||
|
||||
static size_t upb_roundup_pow2(size_t bytes) {
|
||||
size_t ret = 128;
|
||||
while (ret < bytes) {
|
||||
ret *= 2;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
UPB_NORETURN static void encode_err(upb_encstate* e, upb_EncodeStatus s) {
|
||||
UPB_ASSERT(s != kUpb_EncodeStatus_Ok);
|
||||
e->status = s;
|
||||
UPB_LONGJMP(e->err, 1);
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
static void encode_growbuffer(upb_encstate* e, size_t bytes) {
|
||||
size_t old_size = e->limit - e->buf;
|
||||
size_t new_size = upb_roundup_pow2(bytes + (e->limit - e->ptr));
|
||||
char* new_buf = upb_Arena_Realloc(e->arena, e->buf, old_size, new_size);
|
||||
|
||||
if (!new_buf) encode_err(e, kUpb_EncodeStatus_OutOfMemory);
|
||||
|
||||
// We want previous data at the end, realloc() put it at the beginning.
|
||||
// TODO: This is somewhat inefficient since we are copying twice.
|
||||
// Maybe create a realloc() that copies to the end of the new buffer?
|
||||
if (old_size > 0) {
|
||||
memmove(new_buf + new_size - old_size, e->buf, old_size);
|
||||
}
|
||||
|
||||
e->ptr = new_buf + new_size - (e->limit - e->ptr);
|
||||
e->limit = new_buf + new_size;
|
||||
e->buf = new_buf;
|
||||
|
||||
e->ptr -= bytes;
|
||||
}
|
||||
|
||||
/* Call to ensure that at least "bytes" bytes are available for writing at
|
||||
* e->ptr. Returns false if the bytes could not be allocated. */
|
||||
UPB_FORCEINLINE
|
||||
static void encode_reserve(upb_encstate* e, size_t bytes) {
|
||||
if ((size_t)(e->ptr - e->buf) < bytes) {
|
||||
encode_growbuffer(e, bytes);
|
||||
return;
|
||||
}
|
||||
|
||||
e->ptr -= bytes;
|
||||
}
|
||||
|
||||
/* Writes the given bytes to the buffer, handling reserve/advance. */
|
||||
static void encode_bytes(upb_encstate* e, const void* data, size_t len) {
|
||||
if (len == 0) return; /* memcpy() with zero size is UB */
|
||||
encode_reserve(e, len);
|
||||
memcpy(e->ptr, data, len);
|
||||
}
|
||||
|
||||
static void encode_fixed64(upb_encstate* e, uint64_t val) {
|
||||
val = _upb_BigEndian_Swap64(val);
|
||||
encode_bytes(e, &val, sizeof(uint64_t));
|
||||
}
|
||||
|
||||
static void encode_fixed32(upb_encstate* e, uint32_t val) {
|
||||
val = _upb_BigEndian_Swap32(val);
|
||||
encode_bytes(e, &val, sizeof(uint32_t));
|
||||
}
|
||||
|
||||
UPB_NOINLINE
|
||||
static void encode_longvarint(upb_encstate* e, uint64_t val) {
|
||||
size_t len;
|
||||
char* start;
|
||||
|
||||
encode_reserve(e, UPB_PB_VARINT_MAX_LEN);
|
||||
len = encode_varint64(val, e->ptr);
|
||||
start = e->ptr + UPB_PB_VARINT_MAX_LEN - len;
|
||||
memmove(start, e->ptr, len);
|
||||
e->ptr = start;
|
||||
}
|
||||
|
||||
UPB_FORCEINLINE
|
||||
static void encode_varint(upb_encstate* e, uint64_t val) {
|
||||
if (val < 128 && e->ptr != e->buf) {
|
||||
--e->ptr;
|
||||
*e->ptr = val;
|
||||
} else {
|
||||
encode_longvarint(e, val);
|
||||
}
|
||||
}
|
||||
|
||||
static void encode_double(upb_encstate* e, double d) {
|
||||
uint64_t u64;
|
||||
UPB_ASSERT(sizeof(double) == sizeof(uint64_t));
|
||||
memcpy(&u64, &d, sizeof(uint64_t));
|
||||
encode_fixed64(e, u64);
|
||||
}
|
||||
|
||||
static void encode_float(upb_encstate* e, float d) {
|
||||
uint32_t u32;
|
||||
UPB_ASSERT(sizeof(float) == sizeof(uint32_t));
|
||||
memcpy(&u32, &d, sizeof(uint32_t));
|
||||
encode_fixed32(e, u32);
|
||||
}
|
||||
|
||||
static void encode_tag(upb_encstate* e, uint32_t field_number,
|
||||
uint8_t wire_type) {
|
||||
encode_varint(e, (field_number << 3) | wire_type);
|
||||
}
|
||||
|
||||
static void encode_fixedarray(upb_encstate* e, const upb_Array* arr,
|
||||
size_t elem_size, uint32_t tag) {
|
||||
size_t bytes = arr->size * elem_size;
|
||||
const char* data = _upb_array_constptr(arr);
|
||||
const char* ptr = data + bytes - elem_size;
|
||||
|
||||
if (tag || !_upb_IsLittleEndian()) {
|
||||
while (true) {
|
||||
if (elem_size == 4) {
|
||||
uint32_t val;
|
||||
memcpy(&val, ptr, sizeof(val));
|
||||
val = _upb_BigEndian_Swap32(val);
|
||||
encode_bytes(e, &val, elem_size);
|
||||
} else {
|
||||
UPB_ASSERT(elem_size == 8);
|
||||
uint64_t val;
|
||||
memcpy(&val, ptr, sizeof(val));
|
||||
val = _upb_BigEndian_Swap64(val);
|
||||
encode_bytes(e, &val, elem_size);
|
||||
}
|
||||
|
||||
if (tag) encode_varint(e, tag);
|
||||
if (ptr == data) break;
|
||||
ptr -= elem_size;
|
||||
}
|
||||
} else {
|
||||
encode_bytes(e, data, bytes);
|
||||
}
|
||||
}
|
||||
|
||||
static void encode_message(upb_encstate* e, const upb_Message* msg,
|
||||
const upb_MiniTable* m, size_t* size);
|
||||
|
||||
static void encode_TaggedMessagePtr(upb_encstate* e,
|
||||
upb_TaggedMessagePtr tagged,
|
||||
const upb_MiniTable* m, size_t* size) {
|
||||
if (upb_TaggedMessagePtr_IsEmpty(tagged)) {
|
||||
m = &_kUpb_MiniTable_Empty;
|
||||
}
|
||||
encode_message(e, _upb_TaggedMessagePtr_GetMessage(tagged), m, size);
|
||||
}
|
||||
|
||||
static void encode_scalar(upb_encstate* e, const void* _field_mem,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* f) {
|
||||
const char* field_mem = _field_mem;
|
||||
int wire_type;
|
||||
|
||||
#define CASE(ctype, type, wtype, encodeval) \
|
||||
{ \
|
||||
ctype val = *(ctype*)field_mem; \
|
||||
encode_##type(e, encodeval); \
|
||||
wire_type = wtype; \
|
||||
break; \
|
||||
}
|
||||
|
||||
switch (f->UPB_PRIVATE(descriptortype)) {
|
||||
case kUpb_FieldType_Double:
|
||||
CASE(double, double, kUpb_WireType_64Bit, val);
|
||||
case kUpb_FieldType_Float:
|
||||
CASE(float, float, kUpb_WireType_32Bit, val);
|
||||
case kUpb_FieldType_Int64:
|
||||
case kUpb_FieldType_UInt64:
|
||||
CASE(uint64_t, varint, kUpb_WireType_Varint, val);
|
||||
case kUpb_FieldType_UInt32:
|
||||
CASE(uint32_t, varint, kUpb_WireType_Varint, val);
|
||||
case kUpb_FieldType_Int32:
|
||||
case kUpb_FieldType_Enum:
|
||||
CASE(int32_t, varint, kUpb_WireType_Varint, (int64_t)val);
|
||||
case kUpb_FieldType_SFixed64:
|
||||
case kUpb_FieldType_Fixed64:
|
||||
CASE(uint64_t, fixed64, kUpb_WireType_64Bit, val);
|
||||
case kUpb_FieldType_Fixed32:
|
||||
case kUpb_FieldType_SFixed32:
|
||||
CASE(uint32_t, fixed32, kUpb_WireType_32Bit, val);
|
||||
case kUpb_FieldType_Bool:
|
||||
CASE(bool, varint, kUpb_WireType_Varint, val);
|
||||
case kUpb_FieldType_SInt32:
|
||||
CASE(int32_t, varint, kUpb_WireType_Varint, encode_zz32(val));
|
||||
case kUpb_FieldType_SInt64:
|
||||
CASE(int64_t, varint, kUpb_WireType_Varint, encode_zz64(val));
|
||||
case kUpb_FieldType_String:
|
||||
case kUpb_FieldType_Bytes: {
|
||||
upb_StringView view = *(upb_StringView*)field_mem;
|
||||
encode_bytes(e, view.data, view.size);
|
||||
encode_varint(e, view.size);
|
||||
wire_type = kUpb_WireType_Delimited;
|
||||
break;
|
||||
}
|
||||
case kUpb_FieldType_Group: {
|
||||
size_t size;
|
||||
upb_TaggedMessagePtr submsg = *(upb_TaggedMessagePtr*)field_mem;
|
||||
const upb_MiniTable* subm = subs[f->UPB_PRIVATE(submsg_index)].submsg;
|
||||
if (submsg == 0) {
|
||||
return;
|
||||
}
|
||||
if (--e->depth == 0) encode_err(e, kUpb_EncodeStatus_MaxDepthExceeded);
|
||||
encode_tag(e, f->number, kUpb_WireType_EndGroup);
|
||||
encode_TaggedMessagePtr(e, submsg, subm, &size);
|
||||
wire_type = kUpb_WireType_StartGroup;
|
||||
e->depth++;
|
||||
break;
|
||||
}
|
||||
case kUpb_FieldType_Message: {
|
||||
size_t size;
|
||||
upb_TaggedMessagePtr submsg = *(upb_TaggedMessagePtr*)field_mem;
|
||||
const upb_MiniTable* subm = subs[f->UPB_PRIVATE(submsg_index)].submsg;
|
||||
if (submsg == 0) {
|
||||
return;
|
||||
}
|
||||
if (--e->depth == 0) encode_err(e, kUpb_EncodeStatus_MaxDepthExceeded);
|
||||
encode_TaggedMessagePtr(e, submsg, subm, &size);
|
||||
encode_varint(e, size);
|
||||
wire_type = kUpb_WireType_Delimited;
|
||||
e->depth++;
|
||||
break;
|
||||
}
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
#undef CASE
|
||||
|
||||
encode_tag(e, f->number, wire_type);
|
||||
}
|
||||
|
||||
static void encode_array(upb_encstate* e, const upb_Message* msg,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* f) {
|
||||
const upb_Array* arr = *UPB_PTR_AT(msg, f->offset, upb_Array*);
|
||||
bool packed = f->mode & kUpb_LabelFlags_IsPacked;
|
||||
size_t pre_len = e->limit - e->ptr;
|
||||
|
||||
if (arr == NULL || arr->size == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
#define VARINT_CASE(ctype, encode) \
|
||||
{ \
|
||||
const ctype* start = _upb_array_constptr(arr); \
|
||||
const ctype* ptr = start + arr->size; \
|
||||
uint32_t tag = packed ? 0 : (f->number << 3) | kUpb_WireType_Varint; \
|
||||
do { \
|
||||
ptr--; \
|
||||
encode_varint(e, encode); \
|
||||
if (tag) encode_varint(e, tag); \
|
||||
} while (ptr != start); \
|
||||
} \
|
||||
break;
|
||||
|
||||
#define TAG(wire_type) (packed ? 0 : (f->number << 3 | wire_type))
|
||||
|
||||
switch (f->UPB_PRIVATE(descriptortype)) {
|
||||
case kUpb_FieldType_Double:
|
||||
encode_fixedarray(e, arr, sizeof(double), TAG(kUpb_WireType_64Bit));
|
||||
break;
|
||||
case kUpb_FieldType_Float:
|
||||
encode_fixedarray(e, arr, sizeof(float), TAG(kUpb_WireType_32Bit));
|
||||
break;
|
||||
case kUpb_FieldType_SFixed64:
|
||||
case kUpb_FieldType_Fixed64:
|
||||
encode_fixedarray(e, arr, sizeof(uint64_t), TAG(kUpb_WireType_64Bit));
|
||||
break;
|
||||
case kUpb_FieldType_Fixed32:
|
||||
case kUpb_FieldType_SFixed32:
|
||||
encode_fixedarray(e, arr, sizeof(uint32_t), TAG(kUpb_WireType_32Bit));
|
||||
break;
|
||||
case kUpb_FieldType_Int64:
|
||||
case kUpb_FieldType_UInt64:
|
||||
VARINT_CASE(uint64_t, *ptr);
|
||||
case kUpb_FieldType_UInt32:
|
||||
VARINT_CASE(uint32_t, *ptr);
|
||||
case kUpb_FieldType_Int32:
|
||||
case kUpb_FieldType_Enum:
|
||||
VARINT_CASE(int32_t, (int64_t)*ptr);
|
||||
case kUpb_FieldType_Bool:
|
||||
VARINT_CASE(bool, *ptr);
|
||||
case kUpb_FieldType_SInt32:
|
||||
VARINT_CASE(int32_t, encode_zz32(*ptr));
|
||||
case kUpb_FieldType_SInt64:
|
||||
VARINT_CASE(int64_t, encode_zz64(*ptr));
|
||||
case kUpb_FieldType_String:
|
||||
case kUpb_FieldType_Bytes: {
|
||||
const upb_StringView* start = _upb_array_constptr(arr);
|
||||
const upb_StringView* ptr = start + arr->size;
|
||||
do {
|
||||
ptr--;
|
||||
encode_bytes(e, ptr->data, ptr->size);
|
||||
encode_varint(e, ptr->size);
|
||||
encode_tag(e, f->number, kUpb_WireType_Delimited);
|
||||
} while (ptr != start);
|
||||
return;
|
||||
}
|
||||
case kUpb_FieldType_Group: {
|
||||
const upb_TaggedMessagePtr* start = _upb_array_constptr(arr);
|
||||
const upb_TaggedMessagePtr* ptr = start + arr->size;
|
||||
const upb_MiniTable* subm = subs[f->UPB_PRIVATE(submsg_index)].submsg;
|
||||
if (--e->depth == 0) encode_err(e, kUpb_EncodeStatus_MaxDepthExceeded);
|
||||
do {
|
||||
size_t size;
|
||||
ptr--;
|
||||
encode_tag(e, f->number, kUpb_WireType_EndGroup);
|
||||
encode_TaggedMessagePtr(e, *ptr, subm, &size);
|
||||
encode_tag(e, f->number, kUpb_WireType_StartGroup);
|
||||
} while (ptr != start);
|
||||
e->depth++;
|
||||
return;
|
||||
}
|
||||
case kUpb_FieldType_Message: {
|
||||
const upb_TaggedMessagePtr* start = _upb_array_constptr(arr);
|
||||
const upb_TaggedMessagePtr* ptr = start + arr->size;
|
||||
const upb_MiniTable* subm = subs[f->UPB_PRIVATE(submsg_index)].submsg;
|
||||
if (--e->depth == 0) encode_err(e, kUpb_EncodeStatus_MaxDepthExceeded);
|
||||
do {
|
||||
size_t size;
|
||||
ptr--;
|
||||
encode_TaggedMessagePtr(e, *ptr, subm, &size);
|
||||
encode_varint(e, size);
|
||||
encode_tag(e, f->number, kUpb_WireType_Delimited);
|
||||
} while (ptr != start);
|
||||
e->depth++;
|
||||
return;
|
||||
}
|
||||
}
|
||||
#undef VARINT_CASE
|
||||
|
||||
if (packed) {
|
||||
encode_varint(e, e->limit - e->ptr - pre_len);
|
||||
encode_tag(e, f->number, kUpb_WireType_Delimited);
|
||||
}
|
||||
}
|
||||
|
||||
static void encode_mapentry(upb_encstate* e, uint32_t number,
|
||||
const upb_MiniTable* layout,
|
||||
const upb_MapEntry* ent) {
|
||||
const upb_MiniTableField* key_field = &layout->fields[0];
|
||||
const upb_MiniTableField* val_field = &layout->fields[1];
|
||||
size_t pre_len = e->limit - e->ptr;
|
||||
size_t size;
|
||||
encode_scalar(e, &ent->data.v, layout->subs, val_field);
|
||||
encode_scalar(e, &ent->data.k, layout->subs, key_field);
|
||||
size = (e->limit - e->ptr) - pre_len;
|
||||
encode_varint(e, size);
|
||||
encode_tag(e, number, kUpb_WireType_Delimited);
|
||||
}
|
||||
|
||||
static void encode_map(upb_encstate* e, const upb_Message* msg,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* f) {
|
||||
const upb_Map* map = *UPB_PTR_AT(msg, f->offset, const upb_Map*);
|
||||
const upb_MiniTable* layout = subs[f->UPB_PRIVATE(submsg_index)].submsg;
|
||||
UPB_ASSERT(layout->field_count == 2);
|
||||
|
||||
if (map == NULL) return;
|
||||
|
||||
if (e->options & kUpb_EncodeOption_Deterministic) {
|
||||
_upb_sortedmap sorted;
|
||||
_upb_mapsorter_pushmap(&e->sorter,
|
||||
layout->fields[0].UPB_PRIVATE(descriptortype), map,
|
||||
&sorted);
|
||||
upb_MapEntry ent;
|
||||
while (_upb_sortedmap_next(&e->sorter, map, &sorted, &ent)) {
|
||||
encode_mapentry(e, f->number, layout, &ent);
|
||||
}
|
||||
_upb_mapsorter_popmap(&e->sorter, &sorted);
|
||||
} else {
|
||||
intptr_t iter = UPB_STRTABLE_BEGIN;
|
||||
upb_StringView key;
|
||||
upb_value val;
|
||||
while (upb_strtable_next2(&map->table, &key, &val, &iter)) {
|
||||
upb_MapEntry ent;
|
||||
_upb_map_fromkey(key, &ent.data.k, map->key_size);
|
||||
_upb_map_fromvalue(val, &ent.data.v, map->val_size);
|
||||
encode_mapentry(e, f->number, layout, &ent);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static bool encode_shouldencode(upb_encstate* e, const upb_Message* msg,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* f) {
|
||||
if (f->presence == 0) {
|
||||
/* Proto3 presence or map/array. */
|
||||
const void* mem = UPB_PTR_AT(msg, f->offset, void);
|
||||
switch (_upb_MiniTableField_GetRep(f)) {
|
||||
case kUpb_FieldRep_1Byte: {
|
||||
char ch;
|
||||
memcpy(&ch, mem, 1);
|
||||
return ch != 0;
|
||||
}
|
||||
case kUpb_FieldRep_4Byte: {
|
||||
uint32_t u32;
|
||||
memcpy(&u32, mem, 4);
|
||||
return u32 != 0;
|
||||
}
|
||||
case kUpb_FieldRep_8Byte: {
|
||||
uint64_t u64;
|
||||
memcpy(&u64, mem, 8);
|
||||
return u64 != 0;
|
||||
}
|
||||
case kUpb_FieldRep_StringView: {
|
||||
const upb_StringView* str = (const upb_StringView*)mem;
|
||||
return str->size != 0;
|
||||
}
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
} else if (f->presence > 0) {
|
||||
/* Proto2 presence: hasbit. */
|
||||
return _upb_hasbit_field(msg, f);
|
||||
} else {
|
||||
/* Field is in a oneof. */
|
||||
return _upb_getoneofcase_field(msg, f) == f->number;
|
||||
}
|
||||
}
|
||||
|
||||
static void encode_field(upb_encstate* e, const upb_Message* msg,
|
||||
const upb_MiniTableSub* subs,
|
||||
const upb_MiniTableField* field) {
|
||||
switch (upb_FieldMode_Get(field)) {
|
||||
case kUpb_FieldMode_Array:
|
||||
encode_array(e, msg, subs, field);
|
||||
break;
|
||||
case kUpb_FieldMode_Map:
|
||||
encode_map(e, msg, subs, field);
|
||||
break;
|
||||
case kUpb_FieldMode_Scalar:
|
||||
encode_scalar(e, UPB_PTR_AT(msg, field->offset, void), subs, field);
|
||||
break;
|
||||
default:
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
}
|
||||
|
||||
static void encode_msgset_item(upb_encstate* e,
|
||||
const upb_Message_Extension* ext) {
|
||||
size_t size;
|
||||
encode_tag(e, kUpb_MsgSet_Item, kUpb_WireType_EndGroup);
|
||||
encode_message(e, ext->data.ptr, ext->ext->sub.submsg, &size);
|
||||
encode_varint(e, size);
|
||||
encode_tag(e, kUpb_MsgSet_Message, kUpb_WireType_Delimited);
|
||||
encode_varint(e, ext->ext->field.number);
|
||||
encode_tag(e, kUpb_MsgSet_TypeId, kUpb_WireType_Varint);
|
||||
encode_tag(e, kUpb_MsgSet_Item, kUpb_WireType_StartGroup);
|
||||
}
|
||||
|
||||
static void encode_ext(upb_encstate* e, const upb_Message_Extension* ext,
|
||||
bool is_message_set) {
|
||||
if (UPB_UNLIKELY(is_message_set)) {
|
||||
encode_msgset_item(e, ext);
|
||||
} else {
|
||||
encode_field(e, &ext->data, &ext->ext->sub, &ext->ext->field);
|
||||
}
|
||||
}
|
||||
|
||||
static void encode_message(upb_encstate* e, const upb_Message* msg,
|
||||
const upb_MiniTable* m, size_t* size) {
|
||||
size_t pre_len = e->limit - e->ptr;
|
||||
|
||||
if ((e->options & kUpb_EncodeOption_CheckRequired) && m->required_count) {
|
||||
uint64_t msg_head;
|
||||
memcpy(&msg_head, msg, 8);
|
||||
msg_head = _upb_BigEndian_Swap64(msg_head);
|
||||
if (upb_MiniTable_requiredmask(m) & ~msg_head) {
|
||||
encode_err(e, kUpb_EncodeStatus_MissingRequired);
|
||||
}
|
||||
}
|
||||
|
||||
if ((e->options & kUpb_EncodeOption_SkipUnknown) == 0) {
|
||||
size_t unknown_size;
|
||||
const char* unknown = upb_Message_GetUnknown(msg, &unknown_size);
|
||||
|
||||
if (unknown) {
|
||||
encode_bytes(e, unknown, unknown_size);
|
||||
}
|
||||
}
|
||||
|
||||
if (m->ext != kUpb_ExtMode_NonExtendable) {
|
||||
/* Encode all extensions together. Unlike C++, we do not attempt to keep
|
||||
* these in field number order relative to normal fields or even to each
|
||||
* other. */
|
||||
size_t ext_count;
|
||||
const upb_Message_Extension* ext = _upb_Message_Getexts(msg, &ext_count);
|
||||
if (ext_count) {
|
||||
if (e->options & kUpb_EncodeOption_Deterministic) {
|
||||
_upb_sortedmap sorted;
|
||||
_upb_mapsorter_pushexts(&e->sorter, ext, ext_count, &sorted);
|
||||
while (_upb_sortedmap_nextext(&e->sorter, &sorted, &ext)) {
|
||||
encode_ext(e, ext, m->ext == kUpb_ExtMode_IsMessageSet);
|
||||
}
|
||||
_upb_mapsorter_popmap(&e->sorter, &sorted);
|
||||
} else {
|
||||
const upb_Message_Extension* end = ext + ext_count;
|
||||
for (; ext != end; ext++) {
|
||||
encode_ext(e, ext, m->ext == kUpb_ExtMode_IsMessageSet);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (m->field_count) {
|
||||
const upb_MiniTableField* f = &m->fields[m->field_count];
|
||||
const upb_MiniTableField* first = &m->fields[0];
|
||||
while (f != first) {
|
||||
f--;
|
||||
if (encode_shouldencode(e, msg, m->subs, f)) {
|
||||
encode_field(e, msg, m->subs, f);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
*size = (e->limit - e->ptr) - pre_len;
|
||||
}
|
||||
|
||||
static upb_EncodeStatus upb_Encoder_Encode(upb_encstate* const encoder,
|
||||
const void* const msg,
|
||||
const upb_MiniTable* const l,
|
||||
char** const buf,
|
||||
size_t* const size) {
|
||||
// Unfortunately we must continue to perform hackery here because there are
|
||||
// code paths which blindly copy the returned pointer without bothering to
|
||||
// check for errors until much later (b/235839510). So we still set *buf to
|
||||
// NULL on error and we still set it to non-NULL on a successful empty result.
|
||||
if (UPB_SETJMP(encoder->err) == 0) {
|
||||
encode_message(encoder, msg, l, size);
|
||||
*size = encoder->limit - encoder->ptr;
|
||||
if (*size == 0) {
|
||||
static char ch;
|
||||
*buf = &ch;
|
||||
} else {
|
||||
UPB_ASSERT(encoder->ptr);
|
||||
*buf = encoder->ptr;
|
||||
}
|
||||
} else {
|
||||
UPB_ASSERT(encoder->status != kUpb_EncodeStatus_Ok);
|
||||
*buf = NULL;
|
||||
*size = 0;
|
||||
}
|
||||
|
||||
_upb_mapsorter_destroy(&encoder->sorter);
|
||||
return encoder->status;
|
||||
}
|
||||
|
||||
upb_EncodeStatus upb_Encode(const void* msg, const upb_MiniTable* l,
|
||||
int options, upb_Arena* arena, char** buf,
|
||||
size_t* size) {
|
||||
upb_encstate e;
|
||||
unsigned depth = (unsigned)options >> 16;
|
||||
|
||||
e.status = kUpb_EncodeStatus_Ok;
|
||||
e.arena = arena;
|
||||
e.buf = NULL;
|
||||
e.limit = NULL;
|
||||
e.ptr = NULL;
|
||||
e.depth = depth ? depth : kUpb_WireFormat_DefaultDepthLimit;
|
||||
e.options = options;
|
||||
_upb_mapsorter_init(&e.sorter);
|
||||
|
||||
return upb_Encoder_Encode(&e, msg, l, buf, size);
|
||||
}
|
||||
76
Pods/gRPC-Core/third_party/upb/upb/wire/encode.h
generated
vendored
Normal file
76
Pods/gRPC-Core/third_party/upb/upb/wire/encode.h
generated
vendored
Normal file
@@ -0,0 +1,76 @@
|
||||
// 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
|
||||
|
||||
// upb_Encode: parsing from a upb_Message using a upb_MiniTable.
|
||||
|
||||
#ifndef UPB_WIRE_ENCODE_H_
|
||||
#define UPB_WIRE_ENCODE_H_
|
||||
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
#include "upb/mini_table/message.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
enum {
|
||||
/* If set, the results of serializing will be deterministic across all
|
||||
* instances of this binary. There are no guarantees across different
|
||||
* binary builds.
|
||||
*
|
||||
* If your proto contains maps, the encoder will need to malloc()/free()
|
||||
* memory during encode. */
|
||||
kUpb_EncodeOption_Deterministic = 1,
|
||||
|
||||
// When set, unknown fields are not printed.
|
||||
kUpb_EncodeOption_SkipUnknown = 2,
|
||||
|
||||
// When set, the encode will fail if any required fields are missing.
|
||||
kUpb_EncodeOption_CheckRequired = 4,
|
||||
};
|
||||
|
||||
typedef enum {
|
||||
kUpb_EncodeStatus_Ok = 0,
|
||||
kUpb_EncodeStatus_OutOfMemory = 1, // Arena alloc failed
|
||||
kUpb_EncodeStatus_MaxDepthExceeded = 2,
|
||||
|
||||
// kUpb_EncodeOption_CheckRequired failed but the parse otherwise succeeded.
|
||||
kUpb_EncodeStatus_MissingRequired = 3,
|
||||
} upb_EncodeStatus;
|
||||
|
||||
UPB_INLINE uint32_t upb_EncodeOptions_MaxDepth(uint16_t depth) {
|
||||
return (uint32_t)depth << 16;
|
||||
}
|
||||
|
||||
UPB_INLINE uint16_t upb_EncodeOptions_GetMaxDepth(uint32_t options) {
|
||||
return options >> 16;
|
||||
}
|
||||
|
||||
// Enforce an upper bound on recursion depth.
|
||||
UPB_INLINE int upb_Encode_LimitDepth(uint32_t encode_options, uint32_t limit) {
|
||||
uint32_t max_depth = upb_EncodeOptions_GetMaxDepth(encode_options);
|
||||
if (max_depth > limit) max_depth = limit;
|
||||
return upb_EncodeOptions_MaxDepth(max_depth) | (encode_options & 0xffff);
|
||||
}
|
||||
|
||||
UPB_API upb_EncodeStatus upb_Encode(const void* msg, const upb_MiniTable* l,
|
||||
int options, upb_Arena* arena, char** buf,
|
||||
size_t* size);
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_WIRE_ENCODE_H_ */
|
||||
19
Pods/gRPC-Core/third_party/upb/upb/wire/eps_copy_input_stream.c
generated
vendored
Normal file
19
Pods/gRPC-Core/third_party/upb/upb/wire/eps_copy_input_stream.c
generated
vendored
Normal file
@@ -0,0 +1,19 @@
|
||||
// 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/wire/eps_copy_input_stream.h"
|
||||
|
||||
static const char* _upb_EpsCopyInputStream_NoOpCallback(
|
||||
upb_EpsCopyInputStream* e, const char* old_end, const char* new_start) {
|
||||
return new_start;
|
||||
}
|
||||
|
||||
const char* _upb_EpsCopyInputStream_IsDoneFallbackNoCallback(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, int overrun) {
|
||||
return _upb_EpsCopyInputStream_IsDoneFallbackInline(
|
||||
e, ptr, overrun, _upb_EpsCopyInputStream_NoOpCallback);
|
||||
}
|
||||
405
Pods/gRPC-Core/third_party/upb/upb/wire/eps_copy_input_stream.h
generated
vendored
Normal file
405
Pods/gRPC-Core/third_party/upb/upb/wire/eps_copy_input_stream.h
generated
vendored
Normal file
@@ -0,0 +1,405 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_WIRE_EPS_COPY_INPUT_STREAM_H_
|
||||
#define UPB_WIRE_EPS_COPY_INPUT_STREAM_H_
|
||||
|
||||
#include <string.h>
|
||||
|
||||
#include "upb/mem/arena.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// The maximum number of bytes a single protobuf field can take up in the
|
||||
// wire format. We only want to do one bounds check per field, so the input
|
||||
// stream guarantees that after upb_EpsCopyInputStream_IsDone() is called,
|
||||
// the decoder can read this many bytes without performing another bounds
|
||||
// check. The stream will copy into a patch buffer as necessary to guarantee
|
||||
// this invariant.
|
||||
#define kUpb_EpsCopyInputStream_SlopBytes 16
|
||||
|
||||
enum {
|
||||
kUpb_EpsCopyInputStream_NoAliasing = 0,
|
||||
kUpb_EpsCopyInputStream_OnPatch = 1,
|
||||
kUpb_EpsCopyInputStream_NoDelta = 2
|
||||
};
|
||||
|
||||
typedef struct {
|
||||
const char* end; // Can read up to SlopBytes bytes beyond this.
|
||||
const char* limit_ptr; // For bounds checks, = end + UPB_MIN(limit, 0)
|
||||
uintptr_t aliasing;
|
||||
int limit; // Submessage limit relative to end
|
||||
bool error; // To distinguish between EOF and error.
|
||||
char patch[kUpb_EpsCopyInputStream_SlopBytes * 2];
|
||||
} upb_EpsCopyInputStream;
|
||||
|
||||
// Returns true if the stream is in the error state. A stream enters the error
|
||||
// state when the user reads past a limit (caught in IsDone()) or the
|
||||
// ZeroCopyInputStream returns an error.
|
||||
UPB_INLINE bool upb_EpsCopyInputStream_IsError(upb_EpsCopyInputStream* e) {
|
||||
return e->error;
|
||||
}
|
||||
|
||||
typedef const char* upb_EpsCopyInputStream_BufferFlipCallback(
|
||||
upb_EpsCopyInputStream* e, const char* old_end, const char* new_start);
|
||||
|
||||
typedef const char* upb_EpsCopyInputStream_IsDoneFallbackFunc(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, int overrun);
|
||||
|
||||
// Initializes a upb_EpsCopyInputStream using the contents of the buffer
|
||||
// [*ptr, size]. Updates `*ptr` as necessary to guarantee that at least
|
||||
// kUpb_EpsCopyInputStream_SlopBytes are available to read.
|
||||
UPB_INLINE void upb_EpsCopyInputStream_Init(upb_EpsCopyInputStream* e,
|
||||
const char** ptr, size_t size,
|
||||
bool enable_aliasing) {
|
||||
if (size <= kUpb_EpsCopyInputStream_SlopBytes) {
|
||||
memset(&e->patch, 0, 32);
|
||||
if (size) memcpy(&e->patch, *ptr, size);
|
||||
e->aliasing = enable_aliasing ? (uintptr_t)*ptr - (uintptr_t)e->patch
|
||||
: kUpb_EpsCopyInputStream_NoAliasing;
|
||||
*ptr = e->patch;
|
||||
e->end = *ptr + size;
|
||||
e->limit = 0;
|
||||
} else {
|
||||
e->end = *ptr + size - kUpb_EpsCopyInputStream_SlopBytes;
|
||||
e->limit = kUpb_EpsCopyInputStream_SlopBytes;
|
||||
e->aliasing = enable_aliasing ? kUpb_EpsCopyInputStream_NoDelta
|
||||
: kUpb_EpsCopyInputStream_NoAliasing;
|
||||
}
|
||||
e->limit_ptr = e->end;
|
||||
e->error = false;
|
||||
}
|
||||
|
||||
typedef enum {
|
||||
// The current stream position is at a limit.
|
||||
kUpb_IsDoneStatus_Done,
|
||||
|
||||
// The current stream position is not at a limit.
|
||||
kUpb_IsDoneStatus_NotDone,
|
||||
|
||||
// The current stream position is not at a limit, and the stream needs to
|
||||
// be flipped to a new buffer before more data can be read.
|
||||
kUpb_IsDoneStatus_NeedFallback,
|
||||
} upb_IsDoneStatus;
|
||||
|
||||
// Returns the status of the current stream position. This is a low-level
|
||||
// function, it is simpler to call upb_EpsCopyInputStream_IsDone() if possible.
|
||||
UPB_INLINE upb_IsDoneStatus upb_EpsCopyInputStream_IsDoneStatus(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, int* overrun) {
|
||||
*overrun = ptr - e->end;
|
||||
if (UPB_LIKELY(ptr < e->limit_ptr)) {
|
||||
return kUpb_IsDoneStatus_NotDone;
|
||||
} else if (UPB_LIKELY(*overrun == e->limit)) {
|
||||
return kUpb_IsDoneStatus_Done;
|
||||
} else {
|
||||
return kUpb_IsDoneStatus_NeedFallback;
|
||||
}
|
||||
}
|
||||
|
||||
// Returns true if the stream has hit a limit, either the current delimited
|
||||
// limit or the overall end-of-stream. As a side effect, this function may flip
|
||||
// the pointer to a new buffer if there are less than
|
||||
// kUpb_EpsCopyInputStream_SlopBytes of data to be read in the current buffer.
|
||||
//
|
||||
// Postcondition: if the function returns false, there are at least
|
||||
// kUpb_EpsCopyInputStream_SlopBytes of data available to read at *ptr.
|
||||
UPB_INLINE bool upb_EpsCopyInputStream_IsDoneWithCallback(
|
||||
upb_EpsCopyInputStream* e, const char** ptr,
|
||||
upb_EpsCopyInputStream_IsDoneFallbackFunc* func) {
|
||||
int overrun;
|
||||
switch (upb_EpsCopyInputStream_IsDoneStatus(e, *ptr, &overrun)) {
|
||||
case kUpb_IsDoneStatus_Done:
|
||||
return true;
|
||||
case kUpb_IsDoneStatus_NotDone:
|
||||
return false;
|
||||
case kUpb_IsDoneStatus_NeedFallback:
|
||||
*ptr = func(e, *ptr, overrun);
|
||||
return *ptr == NULL;
|
||||
}
|
||||
UPB_UNREACHABLE();
|
||||
}
|
||||
|
||||
const char* _upb_EpsCopyInputStream_IsDoneFallbackNoCallback(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, int overrun);
|
||||
|
||||
// A simpler version of IsDoneWithCallback() that does not support a buffer flip
|
||||
// callback. Useful in cases where we do not need to insert custom logic at
|
||||
// every buffer flip.
|
||||
//
|
||||
// If this returns true, the user must call upb_EpsCopyInputStream_IsError()
|
||||
// to distinguish between EOF and error.
|
||||
UPB_INLINE bool upb_EpsCopyInputStream_IsDone(upb_EpsCopyInputStream* e,
|
||||
const char** ptr) {
|
||||
return upb_EpsCopyInputStream_IsDoneWithCallback(
|
||||
e, ptr, _upb_EpsCopyInputStream_IsDoneFallbackNoCallback);
|
||||
}
|
||||
|
||||
// Returns the total number of bytes that are safe to read from the current
|
||||
// buffer without reading uninitialized or unallocated memory.
|
||||
//
|
||||
// Note that this check does not respect any semantic limits on the stream,
|
||||
// either limits from PushLimit() or the overall stream end, so some of these
|
||||
// bytes may have unpredictable, nonsense values in them. The guarantee is only
|
||||
// that the bytes are valid to read from the perspective of the C language
|
||||
// (ie. you can read without triggering UBSAN or ASAN).
|
||||
UPB_INLINE size_t upb_EpsCopyInputStream_BytesAvailable(
|
||||
upb_EpsCopyInputStream* e, const char* ptr) {
|
||||
return (e->end - ptr) + kUpb_EpsCopyInputStream_SlopBytes;
|
||||
}
|
||||
|
||||
// Returns true if the given delimited field size is valid (it does not extend
|
||||
// beyond any previously-pushed limits). `ptr` should point to the beginning
|
||||
// of the field data, after the delimited size.
|
||||
//
|
||||
// Note that this does *not* guarantee that all of the data for this field is in
|
||||
// the current buffer.
|
||||
UPB_INLINE bool upb_EpsCopyInputStream_CheckSize(
|
||||
const upb_EpsCopyInputStream* e, const char* ptr, int size) {
|
||||
UPB_ASSERT(size >= 0);
|
||||
return ptr - e->end + size <= e->limit;
|
||||
}
|
||||
|
||||
UPB_INLINE bool _upb_EpsCopyInputStream_CheckSizeAvailable(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, int size, bool submessage) {
|
||||
// This is one extra branch compared to the more normal:
|
||||
// return (size_t)(end - ptr) < size;
|
||||
// However it is one less computation if we are just about to use "ptr + len":
|
||||
// https://godbolt.org/z/35YGPz
|
||||
// In microbenchmarks this shows a small improvement.
|
||||
uintptr_t uptr = (uintptr_t)ptr;
|
||||
uintptr_t uend = (uintptr_t)e->limit_ptr;
|
||||
uintptr_t res = uptr + (size_t)size;
|
||||
if (!submessage) uend += kUpb_EpsCopyInputStream_SlopBytes;
|
||||
// NOTE: this check depends on having a linear address space. This is not
|
||||
// technically guaranteed by uintptr_t.
|
||||
bool ret = res >= uptr && res <= uend;
|
||||
if (size < 0) UPB_ASSERT(!ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// Returns true if the given delimited field size is valid (it does not extend
|
||||
// beyond any previously-pushed limited) *and* all of the data for this field is
|
||||
// available to be read in the current buffer.
|
||||
//
|
||||
// If the size is negative, this function will always return false. This
|
||||
// property can be useful in some cases.
|
||||
UPB_INLINE bool upb_EpsCopyInputStream_CheckDataSizeAvailable(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, int size) {
|
||||
return _upb_EpsCopyInputStream_CheckSizeAvailable(e, ptr, size, false);
|
||||
}
|
||||
|
||||
// Returns true if the given sub-message size is valid (it does not extend
|
||||
// beyond any previously-pushed limited) *and* all of the data for this
|
||||
// sub-message is available to be parsed in the current buffer.
|
||||
//
|
||||
// This implies that all fields from the sub-message can be parsed from the
|
||||
// current buffer while maintaining the invariant that we always have at least
|
||||
// kUpb_EpsCopyInputStream_SlopBytes of data available past the beginning of
|
||||
// any individual field start.
|
||||
//
|
||||
// If the size is negative, this function will always return false. This
|
||||
// property can be useful in some cases.
|
||||
UPB_INLINE bool upb_EpsCopyInputStream_CheckSubMessageSizeAvailable(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, int size) {
|
||||
return _upb_EpsCopyInputStream_CheckSizeAvailable(e, ptr, size, true);
|
||||
}
|
||||
|
||||
// Returns true if aliasing_enabled=true was passed to
|
||||
// upb_EpsCopyInputStream_Init() when this stream was initialized.
|
||||
UPB_INLINE bool upb_EpsCopyInputStream_AliasingEnabled(
|
||||
upb_EpsCopyInputStream* e) {
|
||||
return e->aliasing != kUpb_EpsCopyInputStream_NoAliasing;
|
||||
}
|
||||
|
||||
// Returns true if aliasing_enabled=true was passed to
|
||||
// upb_EpsCopyInputStream_Init() when this stream was initialized *and* we can
|
||||
// alias into the region [ptr, size] in an input buffer.
|
||||
UPB_INLINE bool upb_EpsCopyInputStream_AliasingAvailable(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, size_t size) {
|
||||
// When EpsCopyInputStream supports streaming, this will need to become a
|
||||
// runtime check.
|
||||
return upb_EpsCopyInputStream_CheckDataSizeAvailable(e, ptr, size) &&
|
||||
e->aliasing >= kUpb_EpsCopyInputStream_NoDelta;
|
||||
}
|
||||
|
||||
// Returns a pointer into an input buffer that corresponds to the parsing
|
||||
// pointer `ptr`. The returned pointer may be the same as `ptr`, but also may
|
||||
// be different if we are currently parsing out of the patch buffer.
|
||||
//
|
||||
// REQUIRES: Aliasing must be available for the given pointer. If the input is a
|
||||
// flat buffer and aliasing is enabled, then aliasing will always be available.
|
||||
UPB_INLINE const char* upb_EpsCopyInputStream_GetAliasedPtr(
|
||||
upb_EpsCopyInputStream* e, const char* ptr) {
|
||||
UPB_ASSUME(upb_EpsCopyInputStream_AliasingAvailable(e, ptr, 0));
|
||||
uintptr_t delta =
|
||||
e->aliasing == kUpb_EpsCopyInputStream_NoDelta ? 0 : e->aliasing;
|
||||
return (const char*)((uintptr_t)ptr + delta);
|
||||
}
|
||||
|
||||
// Reads string data from the input, aliasing into the input buffer instead of
|
||||
// copying. The parsing pointer is passed in `*ptr`, and will be updated if
|
||||
// necessary to point to the actual input buffer. Returns the new parsing
|
||||
// pointer, which will be advanced past the string data.
|
||||
//
|
||||
// REQUIRES: Aliasing must be available for this data region (test with
|
||||
// upb_EpsCopyInputStream_AliasingAvailable().
|
||||
UPB_INLINE const char* upb_EpsCopyInputStream_ReadStringAliased(
|
||||
upb_EpsCopyInputStream* e, const char** ptr, size_t size) {
|
||||
UPB_ASSUME(upb_EpsCopyInputStream_AliasingAvailable(e, *ptr, size));
|
||||
const char* ret = *ptr + size;
|
||||
*ptr = upb_EpsCopyInputStream_GetAliasedPtr(e, *ptr);
|
||||
UPB_ASSUME(ret != NULL);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// Skips `size` bytes of data from the input and returns a pointer past the end.
|
||||
// Returns NULL on end of stream or error.
|
||||
UPB_INLINE const char* upb_EpsCopyInputStream_Skip(upb_EpsCopyInputStream* e,
|
||||
const char* ptr, int size) {
|
||||
if (!upb_EpsCopyInputStream_CheckDataSizeAvailable(e, ptr, size)) return NULL;
|
||||
return ptr + size;
|
||||
}
|
||||
|
||||
// Copies `size` bytes of data from the input `ptr` into the buffer `to`, and
|
||||
// returns a pointer past the end. Returns NULL on end of stream or error.
|
||||
UPB_INLINE const char* upb_EpsCopyInputStream_Copy(upb_EpsCopyInputStream* e,
|
||||
const char* ptr, void* to,
|
||||
int size) {
|
||||
if (!upb_EpsCopyInputStream_CheckDataSizeAvailable(e, ptr, size)) return NULL;
|
||||
memcpy(to, ptr, size);
|
||||
return ptr + size;
|
||||
}
|
||||
|
||||
// Reads string data from the stream and advances the pointer accordingly.
|
||||
// If aliasing was enabled when the stream was initialized, then the returned
|
||||
// pointer will point into the input buffer if possible, otherwise new data
|
||||
// will be allocated from arena and copied into. We may be forced to copy even
|
||||
// if aliasing was enabled if the input data spans input buffers.
|
||||
//
|
||||
// Returns NULL if memory allocation failed, or we reached a premature EOF.
|
||||
UPB_INLINE const char* upb_EpsCopyInputStream_ReadString(
|
||||
upb_EpsCopyInputStream* e, const char** ptr, size_t size,
|
||||
upb_Arena* arena) {
|
||||
if (upb_EpsCopyInputStream_AliasingAvailable(e, *ptr, size)) {
|
||||
return upb_EpsCopyInputStream_ReadStringAliased(e, ptr, size);
|
||||
} else {
|
||||
// We need to allocate and copy.
|
||||
if (!upb_EpsCopyInputStream_CheckDataSizeAvailable(e, *ptr, size)) {
|
||||
return NULL;
|
||||
}
|
||||
UPB_ASSERT(arena);
|
||||
char* data = (char*)upb_Arena_Malloc(arena, size);
|
||||
if (!data) return NULL;
|
||||
const char* ret = upb_EpsCopyInputStream_Copy(e, *ptr, data, size);
|
||||
*ptr = data;
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
|
||||
UPB_INLINE void _upb_EpsCopyInputStream_CheckLimit(upb_EpsCopyInputStream* e) {
|
||||
UPB_ASSERT(e->limit_ptr == e->end + UPB_MIN(0, e->limit));
|
||||
}
|
||||
|
||||
// Pushes a limit onto the stack of limits for the current stream. The limit
|
||||
// will extend for `size` bytes beyond the position in `ptr`. Future calls to
|
||||
// upb_EpsCopyInputStream_IsDone() will return `true` when the stream position
|
||||
// reaches this limit.
|
||||
//
|
||||
// Returns a delta that the caller must store and supply to PopLimit() below.
|
||||
UPB_INLINE int upb_EpsCopyInputStream_PushLimit(upb_EpsCopyInputStream* e,
|
||||
const char* ptr, int size) {
|
||||
int limit = size + (int)(ptr - e->end);
|
||||
int delta = e->limit - limit;
|
||||
_upb_EpsCopyInputStream_CheckLimit(e);
|
||||
UPB_ASSERT(limit <= e->limit);
|
||||
e->limit = limit;
|
||||
e->limit_ptr = e->end + UPB_MIN(0, limit);
|
||||
_upb_EpsCopyInputStream_CheckLimit(e);
|
||||
return delta;
|
||||
}
|
||||
|
||||
// Pops the last limit that was pushed on this stream. This may only be called
|
||||
// once IsDone() returns true. The user must pass the delta that was returned
|
||||
// from PushLimit().
|
||||
UPB_INLINE void upb_EpsCopyInputStream_PopLimit(upb_EpsCopyInputStream* e,
|
||||
const char* ptr,
|
||||
int saved_delta) {
|
||||
UPB_ASSERT(ptr - e->end == e->limit);
|
||||
_upb_EpsCopyInputStream_CheckLimit(e);
|
||||
e->limit += saved_delta;
|
||||
e->limit_ptr = e->end + UPB_MIN(0, e->limit);
|
||||
_upb_EpsCopyInputStream_CheckLimit(e);
|
||||
}
|
||||
|
||||
UPB_INLINE const char* _upb_EpsCopyInputStream_IsDoneFallbackInline(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, int overrun,
|
||||
upb_EpsCopyInputStream_BufferFlipCallback* callback) {
|
||||
if (overrun < e->limit) {
|
||||
// Need to copy remaining data into patch buffer.
|
||||
UPB_ASSERT(overrun < kUpb_EpsCopyInputStream_SlopBytes);
|
||||
const char* old_end = ptr;
|
||||
const char* new_start = &e->patch[0] + overrun;
|
||||
memset(e->patch + kUpb_EpsCopyInputStream_SlopBytes, 0,
|
||||
kUpb_EpsCopyInputStream_SlopBytes);
|
||||
memcpy(e->patch, e->end, kUpb_EpsCopyInputStream_SlopBytes);
|
||||
ptr = new_start;
|
||||
e->end = &e->patch[kUpb_EpsCopyInputStream_SlopBytes];
|
||||
e->limit -= kUpb_EpsCopyInputStream_SlopBytes;
|
||||
e->limit_ptr = e->end + e->limit;
|
||||
UPB_ASSERT(ptr < e->limit_ptr);
|
||||
if (e->aliasing != kUpb_EpsCopyInputStream_NoAliasing) {
|
||||
e->aliasing = (uintptr_t)old_end - (uintptr_t)new_start;
|
||||
}
|
||||
return callback(e, old_end, new_start);
|
||||
} else {
|
||||
UPB_ASSERT(overrun > e->limit);
|
||||
e->error = true;
|
||||
return callback(e, NULL, NULL);
|
||||
}
|
||||
}
|
||||
|
||||
typedef const char* upb_EpsCopyInputStream_ParseDelimitedFunc(
|
||||
upb_EpsCopyInputStream* e, const char* ptr, void* ctx);
|
||||
|
||||
// Tries to perform a fast-path handling of the given delimited message data.
|
||||
// If the sub-message beginning at `*ptr` and extending for `len` is short and
|
||||
// fits within this buffer, calls `func` with `ctx` as a parameter, where the
|
||||
// pushing and popping of limits is handled automatically and with lower cost
|
||||
// than the normal PushLimit()/PopLimit() sequence.
|
||||
static UPB_FORCEINLINE bool upb_EpsCopyInputStream_TryParseDelimitedFast(
|
||||
upb_EpsCopyInputStream* e, const char** ptr, int len,
|
||||
upb_EpsCopyInputStream_ParseDelimitedFunc* func, void* ctx) {
|
||||
if (!upb_EpsCopyInputStream_CheckSubMessageSizeAvailable(e, *ptr, len)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// Fast case: Sub-message is <128 bytes and fits in the current buffer.
|
||||
// This means we can preserve limit/limit_ptr verbatim.
|
||||
const char* saved_limit_ptr = e->limit_ptr;
|
||||
int saved_limit = e->limit;
|
||||
e->limit_ptr = *ptr + len;
|
||||
e->limit = e->limit_ptr - e->end;
|
||||
UPB_ASSERT(e->limit_ptr == e->end + UPB_MIN(0, e->limit));
|
||||
*ptr = func(e, *ptr, ctx);
|
||||
e->limit_ptr = saved_limit_ptr;
|
||||
e->limit = saved_limit;
|
||||
UPB_ASSERT(e->limit_ptr == e->end + UPB_MIN(0, e->limit));
|
||||
return true;
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_WIRE_EPS_COPY_INPUT_STREAM_H_
|
||||
27
Pods/gRPC-Core/third_party/upb/upb/wire/internal/constants.h
generated
vendored
Normal file
27
Pods/gRPC-Core/third_party/upb/upb/wire/internal/constants.h
generated
vendored
Normal file
@@ -0,0 +1,27 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_WIRE_INTERNAL_CONSTANTS_H_
|
||||
#define UPB_WIRE_INTERNAL_CONSTANTS_H_
|
||||
|
||||
#define kUpb_WireFormat_DefaultDepthLimit 100
|
||||
|
||||
// MessageSet wire format is:
|
||||
// message MessageSet {
|
||||
// repeated group Item = 1 {
|
||||
// required int32 type_id = 2;
|
||||
// required bytes message = 3;
|
||||
// }
|
||||
// }
|
||||
|
||||
enum {
|
||||
kUpb_MsgSet_Item = 1,
|
||||
kUpb_MsgSet_TypeId = 2,
|
||||
kUpb_MsgSet_Message = 3,
|
||||
};
|
||||
|
||||
#endif /* UPB_WIRE_INTERNAL_CONSTANTS_H_ */
|
||||
143
Pods/gRPC-Core/third_party/upb/upb/wire/internal/decode.h
generated
vendored
Normal file
143
Pods/gRPC-Core/third_party/upb/upb/wire/internal/decode.h
generated
vendored
Normal file
@@ -0,0 +1,143 @@
|
||||
// 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
|
||||
|
||||
/*
|
||||
* Internal implementation details of the decoder that are shared between
|
||||
* decode.c and decode_fast.c.
|
||||
*/
|
||||
|
||||
#ifndef UPB_WIRE_INTERNAL_DECODE_H_
|
||||
#define UPB_WIRE_INTERNAL_DECODE_H_
|
||||
|
||||
#include "upb/mem/internal/arena.h"
|
||||
#include "upb/message/internal/message.h"
|
||||
#include "upb/wire/decode.h"
|
||||
#include "upb/wire/eps_copy_input_stream.h"
|
||||
#include "utf8_range.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#define DECODE_NOGROUP (uint32_t) - 1
|
||||
|
||||
typedef struct upb_Decoder {
|
||||
upb_EpsCopyInputStream input;
|
||||
const upb_ExtensionRegistry* extreg;
|
||||
const char* unknown; // Start of unknown data, preserve at buffer flip
|
||||
upb_Message* unknown_msg; // Pointer to preserve data to
|
||||
int depth; // Tracks recursion depth to bound stack usage.
|
||||
uint32_t end_group; // field number of END_GROUP tag, else DECODE_NOGROUP.
|
||||
uint16_t options;
|
||||
bool missing_required;
|
||||
upb_Arena arena;
|
||||
upb_DecodeStatus status;
|
||||
jmp_buf err;
|
||||
|
||||
#ifndef NDEBUG
|
||||
const char* debug_tagstart;
|
||||
const char* debug_valstart;
|
||||
#endif
|
||||
} upb_Decoder;
|
||||
|
||||
/* Error function that will abort decoding with longjmp(). We can't declare this
|
||||
* UPB_NORETURN, even though it is appropriate, because if we do then compilers
|
||||
* will "helpfully" refuse to tailcall to it
|
||||
* (see: https://stackoverflow.com/a/55657013), which will defeat a major goal
|
||||
* of our optimizations. That is also why we must declare it in a separate file,
|
||||
* otherwise the compiler will see that it calls longjmp() and deduce that it is
|
||||
* noreturn. */
|
||||
const char* _upb_FastDecoder_ErrorJmp(upb_Decoder* d, int status);
|
||||
|
||||
extern const uint8_t upb_utf8_offsets[];
|
||||
|
||||
UPB_INLINE
|
||||
bool _upb_Decoder_VerifyUtf8Inline(const char* ptr, int len) {
|
||||
const char* end = ptr + len;
|
||||
|
||||
// Check 8 bytes at a time for any non-ASCII char.
|
||||
while (end - ptr >= 8) {
|
||||
uint64_t data;
|
||||
memcpy(&data, ptr, 8);
|
||||
if (data & 0x8080808080808080) goto non_ascii;
|
||||
ptr += 8;
|
||||
}
|
||||
|
||||
// Check one byte at a time for non-ASCII.
|
||||
while (ptr < end) {
|
||||
if (*ptr & 0x80) goto non_ascii;
|
||||
ptr++;
|
||||
}
|
||||
|
||||
return true;
|
||||
|
||||
non_ascii:
|
||||
return utf8_range2((const unsigned char*)ptr, end - ptr) == 0;
|
||||
}
|
||||
|
||||
const char* _upb_Decoder_CheckRequired(upb_Decoder* d, const char* ptr,
|
||||
const upb_Message* msg,
|
||||
const upb_MiniTable* l);
|
||||
|
||||
/* x86-64 pointers always have the high 16 bits matching. So we can shift
|
||||
* left 8 and right 8 without loss of information. */
|
||||
UPB_INLINE intptr_t decode_totable(const upb_MiniTable* tablep) {
|
||||
return ((intptr_t)tablep << 8) | tablep->table_mask;
|
||||
}
|
||||
|
||||
UPB_INLINE const upb_MiniTable* decode_totablep(intptr_t table) {
|
||||
return (const upb_MiniTable*)(table >> 8);
|
||||
}
|
||||
|
||||
const char* _upb_Decoder_IsDoneFallback(upb_EpsCopyInputStream* e,
|
||||
const char* ptr, int overrun);
|
||||
|
||||
UPB_INLINE bool _upb_Decoder_IsDone(upb_Decoder* d, const char** ptr) {
|
||||
return upb_EpsCopyInputStream_IsDoneWithCallback(
|
||||
&d->input, ptr, &_upb_Decoder_IsDoneFallback);
|
||||
}
|
||||
|
||||
UPB_INLINE const char* _upb_Decoder_BufferFlipCallback(
|
||||
upb_EpsCopyInputStream* e, const char* old_end, const char* new_start) {
|
||||
upb_Decoder* d = (upb_Decoder*)e;
|
||||
if (!old_end) _upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_Malformed);
|
||||
|
||||
if (d->unknown) {
|
||||
if (!_upb_Message_AddUnknown(d->unknown_msg, d->unknown,
|
||||
old_end - d->unknown, &d->arena)) {
|
||||
_upb_FastDecoder_ErrorJmp(d, kUpb_DecodeStatus_OutOfMemory);
|
||||
}
|
||||
d->unknown = new_start;
|
||||
}
|
||||
return new_start;
|
||||
}
|
||||
|
||||
#if UPB_FASTTABLE
|
||||
UPB_INLINE
|
||||
const char* _upb_FastDecoder_TagDispatch(upb_Decoder* d, const char* ptr,
|
||||
upb_Message* msg, intptr_t table,
|
||||
uint64_t hasbits, uint64_t tag) {
|
||||
const upb_MiniTable* table_p = decode_totablep(table);
|
||||
uint8_t mask = table;
|
||||
uint64_t data;
|
||||
size_t idx = tag & mask;
|
||||
UPB_ASSUME((idx & 7) == 0);
|
||||
idx >>= 3;
|
||||
data = table_p->fasttable[idx].field_data ^ tag;
|
||||
UPB_MUSTTAIL return table_p->fasttable[idx].field_parser(d, ptr, msg, table,
|
||||
hasbits, data);
|
||||
}
|
||||
#endif
|
||||
|
||||
UPB_INLINE uint32_t _upb_FastDecoder_LoadTag(const char* ptr) {
|
||||
uint16_t tag;
|
||||
memcpy(&tag, ptr, 2);
|
||||
return tag;
|
||||
}
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_WIRE_INTERNAL_DECODE_H_ */
|
||||
45
Pods/gRPC-Core/third_party/upb/upb/wire/internal/swap.h
generated
vendored
Normal file
45
Pods/gRPC-Core/third_party/upb/upb/wire/internal/swap.h
generated
vendored
Normal file
@@ -0,0 +1,45 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_WIRE_INTERNAL_SWAP_H_
|
||||
#define UPB_WIRE_INTERNAL_SWAP_H_
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
UPB_INLINE bool _upb_IsLittleEndian(void) {
|
||||
int x = 1;
|
||||
return *(char*)&x == 1;
|
||||
}
|
||||
|
||||
UPB_INLINE uint32_t _upb_BigEndian_Swap32(uint32_t val) {
|
||||
if (_upb_IsLittleEndian()) return val;
|
||||
|
||||
return ((val & 0xff) << 24) | ((val & 0xff00) << 8) |
|
||||
((val & 0xff0000) >> 8) | ((val & 0xff000000) >> 24);
|
||||
}
|
||||
|
||||
UPB_INLINE uint64_t _upb_BigEndian_Swap64(uint64_t val) {
|
||||
if (_upb_IsLittleEndian()) return val;
|
||||
|
||||
return ((uint64_t)_upb_BigEndian_Swap32((uint32_t)val) << 32) |
|
||||
_upb_BigEndian_Swap32((uint32_t)(val >> 32));
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif /* UPB_WIRE_INTERNAL_SWAP_H_ */
|
||||
47
Pods/gRPC-Core/third_party/upb/upb/wire/reader.c
generated
vendored
Normal file
47
Pods/gRPC-Core/third_party/upb/upb/wire/reader.c
generated
vendored
Normal file
@@ -0,0 +1,47 @@
|
||||
// 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/wire/reader.h"
|
||||
|
||||
#include "upb/wire/eps_copy_input_stream.h"
|
||||
#include "upb/wire/types.h"
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
UPB_NOINLINE _upb_WireReader_ReadLongVarintRet
|
||||
_upb_WireReader_ReadLongVarint(const char* ptr, uint64_t val) {
|
||||
_upb_WireReader_ReadLongVarintRet ret = {NULL, 0};
|
||||
uint64_t byte;
|
||||
int i;
|
||||
for (i = 1; i < 10; i++) {
|
||||
byte = (uint8_t)ptr[i];
|
||||
val += (byte - 1) << (i * 7);
|
||||
if (!(byte & 0x80)) {
|
||||
ret.ptr = ptr + i + 1;
|
||||
ret.val = val;
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
const char* _upb_WireReader_SkipGroup(const char* ptr, uint32_t tag,
|
||||
int depth_limit,
|
||||
upb_EpsCopyInputStream* stream) {
|
||||
if (--depth_limit == 0) return NULL;
|
||||
uint32_t end_group_tag = (tag & ~7ULL) | kUpb_WireType_EndGroup;
|
||||
while (!upb_EpsCopyInputStream_IsDone(stream, &ptr)) {
|
||||
uint32_t tag;
|
||||
ptr = upb_WireReader_ReadTag(ptr, &tag);
|
||||
if (!ptr) return NULL;
|
||||
if (tag == end_group_tag) return ptr;
|
||||
ptr = _upb_WireReader_SkipValue(ptr, tag, depth_limit, stream);
|
||||
if (!ptr) return NULL;
|
||||
}
|
||||
return ptr;
|
||||
}
|
||||
207
Pods/gRPC-Core/third_party/upb/upb/wire/reader.h
generated
vendored
Normal file
207
Pods/gRPC-Core/third_party/upb/upb/wire/reader.h
generated
vendored
Normal file
@@ -0,0 +1,207 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_WIRE_READER_H_
|
||||
#define UPB_WIRE_READER_H_
|
||||
|
||||
#include "upb/wire/eps_copy_input_stream.h"
|
||||
#include "upb/wire/internal/swap.h"
|
||||
#include "upb/wire/types.h" // IWYU pragma: export
|
||||
|
||||
// Must be last.
|
||||
#include "upb/port/def.inc"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// The upb_WireReader interface is suitable for general-purpose parsing of
|
||||
// protobuf binary wire format. It is designed to be used along with
|
||||
// upb_EpsCopyInputStream for buffering, and all parsing routines in this file
|
||||
// assume that at least kUpb_EpsCopyInputStream_SlopBytes worth of data is
|
||||
// available to read without any bounds checks.
|
||||
|
||||
#define kUpb_WireReader_WireTypeMask 7
|
||||
#define kUpb_WireReader_WireTypeBits 3
|
||||
|
||||
typedef struct {
|
||||
const char* ptr;
|
||||
uint64_t val;
|
||||
} _upb_WireReader_ReadLongVarintRet;
|
||||
|
||||
_upb_WireReader_ReadLongVarintRet _upb_WireReader_ReadLongVarint(
|
||||
const char* ptr, uint64_t val);
|
||||
|
||||
static UPB_FORCEINLINE const char* _upb_WireReader_ReadVarint(const char* ptr,
|
||||
uint64_t* val,
|
||||
int maxlen,
|
||||
uint64_t maxval) {
|
||||
uint64_t byte = (uint8_t)*ptr;
|
||||
if (UPB_LIKELY((byte & 0x80) == 0)) {
|
||||
*val = (uint32_t)byte;
|
||||
return ptr + 1;
|
||||
}
|
||||
const char* start = ptr;
|
||||
_upb_WireReader_ReadLongVarintRet res =
|
||||
_upb_WireReader_ReadLongVarint(ptr, byte);
|
||||
if (!res.ptr || (maxlen < 10 && res.ptr - start > maxlen) ||
|
||||
res.val > maxval) {
|
||||
return NULL; // Malformed.
|
||||
}
|
||||
*val = res.val;
|
||||
return res.ptr;
|
||||
}
|
||||
|
||||
// Parses a tag into `tag`, and returns a pointer past the end of the tag, or
|
||||
// NULL if there was an error in the tag data.
|
||||
//
|
||||
// REQUIRES: there must be at least 10 bytes of data available at `ptr`.
|
||||
// Bounds checks must be performed before calling this function, preferably
|
||||
// by calling upb_EpsCopyInputStream_IsDone().
|
||||
static UPB_FORCEINLINE const char* upb_WireReader_ReadTag(const char* ptr,
|
||||
uint32_t* tag) {
|
||||
uint64_t val;
|
||||
ptr = _upb_WireReader_ReadVarint(ptr, &val, 5, UINT32_MAX);
|
||||
if (!ptr) return NULL;
|
||||
*tag = val;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
// Given a tag, returns the field number.
|
||||
UPB_INLINE uint32_t upb_WireReader_GetFieldNumber(uint32_t tag) {
|
||||
return tag >> kUpb_WireReader_WireTypeBits;
|
||||
}
|
||||
|
||||
// Given a tag, returns the wire type.
|
||||
UPB_INLINE uint8_t upb_WireReader_GetWireType(uint32_t tag) {
|
||||
return tag & kUpb_WireReader_WireTypeMask;
|
||||
}
|
||||
|
||||
UPB_INLINE const char* upb_WireReader_ReadVarint(const char* ptr,
|
||||
uint64_t* val) {
|
||||
return _upb_WireReader_ReadVarint(ptr, val, 10, UINT64_MAX);
|
||||
}
|
||||
|
||||
// Skips data for a varint, returning a pointer past the end of the varint, or
|
||||
// NULL if there was an error in the varint data.
|
||||
//
|
||||
// REQUIRES: there must be at least 10 bytes of data available at `ptr`.
|
||||
// Bounds checks must be performed before calling this function, preferably
|
||||
// by calling upb_EpsCopyInputStream_IsDone().
|
||||
UPB_INLINE const char* upb_WireReader_SkipVarint(const char* ptr) {
|
||||
uint64_t val;
|
||||
return upb_WireReader_ReadVarint(ptr, &val);
|
||||
}
|
||||
|
||||
// Reads a varint indicating the size of a delimited field into `size`, or
|
||||
// NULL if there was an error in the varint data.
|
||||
//
|
||||
// REQUIRES: there must be at least 10 bytes of data available at `ptr`.
|
||||
// Bounds checks must be performed before calling this function, preferably
|
||||
// by calling upb_EpsCopyInputStream_IsDone().
|
||||
UPB_INLINE const char* upb_WireReader_ReadSize(const char* ptr, int* size) {
|
||||
uint64_t size64;
|
||||
ptr = upb_WireReader_ReadVarint(ptr, &size64);
|
||||
if (!ptr || size64 >= INT32_MAX) return NULL;
|
||||
*size = size64;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
// Reads a fixed32 field, performing byte swapping if necessary.
|
||||
//
|
||||
// REQUIRES: there must be at least 4 bytes of data available at `ptr`.
|
||||
// Bounds checks must be performed before calling this function, preferably
|
||||
// by calling upb_EpsCopyInputStream_IsDone().
|
||||
UPB_INLINE const char* upb_WireReader_ReadFixed32(const char* ptr, void* val) {
|
||||
uint32_t uval;
|
||||
memcpy(&uval, ptr, 4);
|
||||
uval = _upb_BigEndian_Swap32(uval);
|
||||
memcpy(val, &uval, 4);
|
||||
return ptr + 4;
|
||||
}
|
||||
|
||||
// Reads a fixed64 field, performing byte swapping if necessary.
|
||||
//
|
||||
// REQUIRES: there must be at least 4 bytes of data available at `ptr`.
|
||||
// Bounds checks must be performed before calling this function, preferably
|
||||
// by calling upb_EpsCopyInputStream_IsDone().
|
||||
UPB_INLINE const char* upb_WireReader_ReadFixed64(const char* ptr, void* val) {
|
||||
uint64_t uval;
|
||||
memcpy(&uval, ptr, 8);
|
||||
uval = _upb_BigEndian_Swap64(uval);
|
||||
memcpy(val, &uval, 8);
|
||||
return ptr + 8;
|
||||
}
|
||||
|
||||
const char* _upb_WireReader_SkipGroup(const char* ptr, uint32_t tag,
|
||||
int depth_limit,
|
||||
upb_EpsCopyInputStream* stream);
|
||||
|
||||
// Skips data for a group, returning a pointer past the end of the group, or
|
||||
// NULL if there was an error parsing the group. The `tag` argument should be
|
||||
// the start group tag that begins the group. The `depth_limit` argument
|
||||
// indicates how many levels of recursion the group is allowed to have before
|
||||
// reporting a parse error (this limit exists to protect against stack
|
||||
// overflow).
|
||||
//
|
||||
// TODO: evaluate how the depth_limit should be specified. Do users need
|
||||
// control over this?
|
||||
UPB_INLINE const char* upb_WireReader_SkipGroup(
|
||||
const char* ptr, uint32_t tag, upb_EpsCopyInputStream* stream) {
|
||||
return _upb_WireReader_SkipGroup(ptr, tag, 100, stream);
|
||||
}
|
||||
|
||||
UPB_INLINE const char* _upb_WireReader_SkipValue(
|
||||
const char* ptr, uint32_t tag, int depth_limit,
|
||||
upb_EpsCopyInputStream* stream) {
|
||||
switch (upb_WireReader_GetWireType(tag)) {
|
||||
case kUpb_WireType_Varint:
|
||||
return upb_WireReader_SkipVarint(ptr);
|
||||
case kUpb_WireType_32Bit:
|
||||
return ptr + 4;
|
||||
case kUpb_WireType_64Bit:
|
||||
return ptr + 8;
|
||||
case kUpb_WireType_Delimited: {
|
||||
int size;
|
||||
ptr = upb_WireReader_ReadSize(ptr, &size);
|
||||
if (!ptr) return NULL;
|
||||
ptr += size;
|
||||
return ptr;
|
||||
}
|
||||
case kUpb_WireType_StartGroup:
|
||||
return _upb_WireReader_SkipGroup(ptr, tag, depth_limit, stream);
|
||||
case kUpb_WireType_EndGroup:
|
||||
return NULL; // Should be handled before now.
|
||||
default:
|
||||
return NULL; // Unknown wire type.
|
||||
}
|
||||
}
|
||||
|
||||
// Skips data for a wire value of any type, returning a pointer past the end of
|
||||
// the data, or NULL if there was an error parsing the group. The `tag` argument
|
||||
// should be the tag that was just parsed. The `depth_limit` argument indicates
|
||||
// how many levels of recursion a group is allowed to have before reporting a
|
||||
// parse error (this limit exists to protect against stack overflow).
|
||||
//
|
||||
// REQUIRES: there must be at least 10 bytes of data available at `ptr`.
|
||||
// Bounds checks must be performed before calling this function, preferably
|
||||
// by calling upb_EpsCopyInputStream_IsDone().
|
||||
//
|
||||
// TODO: evaluate how the depth_limit should be specified. Do users need
|
||||
// control over this?
|
||||
UPB_INLINE const char* upb_WireReader_SkipValue(
|
||||
const char* ptr, uint32_t tag, upb_EpsCopyInputStream* stream) {
|
||||
return _upb_WireReader_SkipValue(ptr, tag, 100, stream);
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
} /* extern "C" */
|
||||
#endif
|
||||
|
||||
#include "upb/port/undef.inc"
|
||||
|
||||
#endif // UPB_WIRE_READER_H_
|
||||
21
Pods/gRPC-Core/third_party/upb/upb/wire/types.h
generated
vendored
Normal file
21
Pods/gRPC-Core/third_party/upb/upb/wire/types.h
generated
vendored
Normal file
@@ -0,0 +1,21 @@
|
||||
// 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
|
||||
|
||||
#ifndef UPB_WIRE_TYPES_H_
|
||||
#define UPB_WIRE_TYPES_H_
|
||||
|
||||
// A list of types as they are encoded on the wire.
|
||||
typedef enum {
|
||||
kUpb_WireType_Varint = 0,
|
||||
kUpb_WireType_64Bit = 1,
|
||||
kUpb_WireType_Delimited = 2,
|
||||
kUpb_WireType_StartGroup = 3,
|
||||
kUpb_WireType_EndGroup = 4,
|
||||
kUpb_WireType_32Bit = 5
|
||||
} upb_WireType;
|
||||
|
||||
#endif /* UPB_WIRE_TYPES_H_ */
|
||||
Reference in New Issue
Block a user