| /* |
| * Copyright (c) 2009-2022, Google LLC |
| * All rights reserved. |
| * |
| * 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 Google LLC 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_MINI_TABLE_ACCESSORS_H_ |
| #define UPB_MINI_TABLE_ACCESSORS_H_ |
| |
| #include "upb/array.h" |
| #include "upb/internal/mini_table_accessors.h" |
| #include "upb/mini_table.h" |
| |
| // Must be last. |
| #include "upb/port_def.inc" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| bool upb_MiniTable_HasField(const upb_Message* msg, |
| const upb_MiniTable_Field* field); |
| |
| void upb_MiniTable_ClearField(upb_Message* msg, |
| const upb_MiniTable_Field* field); |
| |
| UPB_INLINE bool upb_MiniTable_GetBool(const upb_Message* msg, |
| const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Bool); |
| return *UPB_PTR_AT(msg, field->offset, bool); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetBool(upb_Message* msg, |
| const upb_MiniTable_Field* field, |
| bool value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Bool); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, bool) = value; |
| } |
| |
| UPB_INLINE int32_t upb_MiniTable_GetInt32(const upb_Message* msg, |
| const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Int32 || |
| field->descriptortype == kUpb_FieldType_SInt32 || |
| field->descriptortype == kUpb_FieldType_SFixed32 || |
| field->descriptortype == kUpb_FieldType_Enum); |
| return *UPB_PTR_AT(msg, field->offset, int32_t); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetInt32(upb_Message* msg, |
| const upb_MiniTable_Field* field, |
| int32_t value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Int32 || |
| field->descriptortype == kUpb_FieldType_SInt32 || |
| field->descriptortype == kUpb_FieldType_SFixed32); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, int32_t) = value; |
| } |
| |
| UPB_INLINE uint32_t upb_MiniTable_GetUInt32(const upb_Message* msg, |
| const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_UInt32 || |
| field->descriptortype == kUpb_FieldType_Fixed32); |
| return *UPB_PTR_AT(msg, field->offset, uint32_t); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetUInt32(upb_Message* msg, |
| const upb_MiniTable_Field* field, |
| uint32_t value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_UInt32 || |
| field->descriptortype == kUpb_FieldType_Fixed32); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, uint32_t) = value; |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetEnumProto2(upb_Message* msg, |
| const upb_MiniTable* msg_mini_table, |
| const upb_MiniTable_Field* field, |
| int32_t value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Enum); |
| UPB_ASSERT(upb_MiniTable_Enum_CheckValue( |
| upb_MiniTable_GetSubEnumTable(msg_mini_table, field), value)); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, int32_t) = value; |
| } |
| |
| UPB_INLINE int64_t upb_MiniTable_GetInt64(const upb_Message* msg, |
| const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Int64 || |
| field->descriptortype == kUpb_FieldType_SInt64 || |
| field->descriptortype == kUpb_FieldType_SFixed64); |
| return *UPB_PTR_AT(msg, field->offset, int64_t); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetInt64(upb_Message* msg, |
| const upb_MiniTable_Field* field, |
| int64_t value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Int64 || |
| field->descriptortype == kUpb_FieldType_SInt64 || |
| field->descriptortype == kUpb_FieldType_SFixed64); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, int64_t) = value; |
| } |
| |
| UPB_INLINE uint64_t upb_MiniTable_GetUInt64(const upb_Message* msg, |
| const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_UInt64 || |
| field->descriptortype == kUpb_FieldType_Fixed64); |
| return *UPB_PTR_AT(msg, field->offset, uint64_t); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetUInt64(upb_Message* msg, |
| const upb_MiniTable_Field* field, |
| uint64_t value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_UInt64 || |
| field->descriptortype == kUpb_FieldType_Fixed64); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, uint64_t) = value; |
| } |
| |
| UPB_INLINE float upb_MiniTable_GetFloat(const upb_Message* msg, |
| const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Float); |
| return *UPB_PTR_AT(msg, field->offset, float); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetFloat(upb_Message* msg, |
| const upb_MiniTable_Field* field, |
| float value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Float); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, float) = value; |
| } |
| |
| UPB_INLINE double upb_MiniTable_GetDouble(const upb_Message* msg, |
| const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Double); |
| return *UPB_PTR_AT(msg, field->offset, double); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetDouble(upb_Message* msg, |
| const upb_MiniTable_Field* field, |
| double value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Double); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, double) = value; |
| } |
| |
| UPB_INLINE upb_StringView upb_MiniTable_GetString( |
| const upb_Message* msg, const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Bytes || |
| field->descriptortype == kUpb_FieldType_String); |
| return *UPB_PTR_AT(msg, field->offset, upb_StringView); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetString(upb_Message* msg, |
| const upb_MiniTable_Field* field, |
| upb_StringView value) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Bytes || |
| field->descriptortype == kUpb_FieldType_String); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, upb_StringView) = value; |
| } |
| |
| UPB_INLINE const upb_Message* upb_MiniTable_GetMessage( |
| const upb_Message* msg, const upb_MiniTable_Field* field) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Message || |
| field->descriptortype == kUpb_FieldType_Group); |
| return *UPB_PTR_AT(msg, field->offset, const upb_Message*); |
| } |
| |
| UPB_INLINE void upb_MiniTable_SetMessage(upb_Message* msg, |
| const upb_MiniTable* mini_table, |
| const upb_MiniTable_Field* field, |
| upb_Message* sub_message) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Message || |
| field->descriptortype == kUpb_FieldType_Group); |
| UPB_ASSERT(mini_table->subs[field->submsg_index].submsg); |
| _upb_MiniTable_SetPresence(msg, field); |
| *UPB_PTR_AT(msg, field->offset, const upb_Message*) = sub_message; |
| } |
| |
| UPB_INLINE upb_Message* upb_MiniTable_GetMutableMessage( |
| upb_Message* msg, const upb_MiniTable* mini_table, |
| const upb_MiniTable_Field* field, upb_Arena* arena) { |
| UPB_ASSERT(field->descriptortype == kUpb_FieldType_Message || |
| field->descriptortype == kUpb_FieldType_Group); |
| 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->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_MiniTable_SetPresence(msg, field); |
| } |
| return sub_message; |
| } |
| |
| UPB_INLINE const upb_Array* upb_MiniTable_GetArray( |
| const upb_Message* msg, const upb_MiniTable_Field* field) { |
| return (const upb_Array*)*UPB_PTR_AT(msg, field->offset, upb_Array*); |
| } |
| |
| UPB_INLINE upb_Array* upb_MiniTable_GetMutableArray( |
| upb_Message* msg, const upb_MiniTable_Field* field) { |
| return (upb_Array*)*UPB_PTR_AT(msg, field->offset, upb_Array*); |
| } |
| |
| void* upb_MiniTable_ResizeArray(upb_Message* msg, |
| const upb_MiniTable_Field* field, size_t len, |
| upb_Arena* arena); |
| typedef enum { |
| kUpb_GetExtension_Ok, |
| kUpb_GetExtension_NotPresent, |
| kUpb_GetExtension_ParseError, |
| kUpb_GetExtension_OutOfMemory, |
| } upb_GetExtension_Status; |
| |
| typedef enum { |
| kUpb_GetExtensionAsBytes_Ok, |
| kUpb_GetExtensionAsBytes_NotPresent, |
| kUpb_GetExtensionAsBytes_EncodeError, |
| } upb_GetExtensionAsBytes_Status; |
| |
| // Returns a message extension or promotes an unknown field to |
| // an extension. |
| // |
| // TODO(ferhat): Only supports extension fields that are messages, |
| // expand support to include non-message types. |
| upb_GetExtension_Status upb_MiniTable_GetOrPromoteExtension( |
| upb_Message* msg, const upb_MiniTable_Extension* ext_table, |
| int decode_options, upb_Arena* arena, |
| const upb_Message_Extension** extension); |
| |
| // Returns a message extension or unknown field matching the extension |
| // data as bytes. |
| // |
| // If an extension has already been decoded it will be re-encoded |
| // to bytes. |
| upb_GetExtensionAsBytes_Status upb_MiniTable_GetExtensionAsBytes( |
| const upb_Message* msg, const upb_MiniTable_Extension* ext_table, |
| int encode_options, upb_Arena* arena, const char** extension_data, |
| size_t* len); |
| |
| typedef enum { |
| kUpb_FindUnknown_Ok, |
| kUpb_FindUnknown_NotPresent, |
| kUpb_FindUnknown_ParseError, |
| } upb_FindUnknown_Status; |
| |
| typedef struct { |
| upb_FindUnknown_Status status; |
| // Start of unknown field data in message arena. |
| const char* ptr; |
| // Size of unknown field data. |
| size_t len; |
| } upb_FindUnknownRet; |
| |
| // Finds first occurrence of unknown data by tag id in message. |
| upb_FindUnknownRet upb_MiniTable_FindUnknown(const upb_Message* msg, |
| uint32_t field_number); |
| |
| typedef enum { |
| kUpb_UnknownToMessage_Ok, |
| kUpb_UnknownToMessage_ParseError, |
| kUpb_UnknownToMessage_OutOfMemory, |
| kUpb_UnknownToMessage_NotFound, |
| } upb_UnknownToMessage_Status; |
| |
| typedef struct { |
| upb_UnknownToMessage_Status status; |
| upb_Message* message; |
| } upb_UnknownToMessageRet; |
| |
| // Promotes unknown data inside message to a upb_Message parsing the unknown. |
| // |
| // The unknown data is removed from message after field value is set |
| // using upb_MiniTable_SetMessage. |
| upb_UnknownToMessageRet upb_MiniTable_PromoteUnknownToMessage( |
| upb_Message* msg, const upb_MiniTable* mini_table, |
| const upb_MiniTable_Field* field, const upb_MiniTable* sub_mini_table, |
| int decode_options, upb_Arena* arena); |
| |
| // Promotes all unknown data that matches field tag id to repeated messages |
| // in upb_Array. |
| // |
| // The unknown data is removed from message after upb_Array is populated. |
| // Since repeated messages can't be packed we remove each unknown that |
| // contains the target tag id. |
| upb_UnknownToMessage_Status upb_MiniTable_PromoteUnknownToMessageArray( |
| upb_Message* msg, const upb_MiniTable_Field* field, |
| const upb_MiniTable* mini_table, int decode_options, upb_Arena* arena); |
| |
| #ifdef __cplusplus |
| } /* extern "C" */ |
| #endif |
| |
| #include "upb/port_undef.inc" |
| |
| #endif // UPB_MINI_TABLE_ACCESSORS_H_ |