mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
655 lines
24 KiB
C++
655 lines
24 KiB
C++
// Protocol Buffers - Google's data interchange format
|
|
// Copyright 2008 Google Inc. 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 Inc. 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 GOOGLE_PROTOBUF_MAP_ENTRY_LITE_H__
|
|
#define GOOGLE_PROTOBUF_MAP_ENTRY_LITE_H__
|
|
|
|
#include <assert.h>
|
|
#include <string>
|
|
|
|
#include <thirdparty/protobuf/stubs/casts.h>
|
|
#include <thirdparty/protobuf/parse_context.h>
|
|
#include <thirdparty/protobuf/io/coded_stream.h>
|
|
#include <thirdparty/protobuf/arena.h>
|
|
#include <thirdparty/protobuf/arenastring.h>
|
|
#include <thirdparty/protobuf/generated_message_util.h>
|
|
#include <thirdparty/protobuf/map.h>
|
|
#include <thirdparty/protobuf/map_type_handler.h>
|
|
#include <thirdparty/protobuf/port.h>
|
|
#include <thirdparty/protobuf/wire_format_lite.h>
|
|
|
|
#include <thirdparty/protobuf/port_def.inc>
|
|
#ifdef SWIG
|
|
#error "You cannot SWIG proto headers"
|
|
#endif
|
|
|
|
namespace google {
|
|
namespace protobuf {
|
|
namespace internal {
|
|
template <typename Derived, typename Key, typename Value,
|
|
WireFormatLite::FieldType kKeyFieldType,
|
|
WireFormatLite::FieldType kValueFieldType>
|
|
class MapEntry;
|
|
template <typename Derived, typename Key, typename Value,
|
|
WireFormatLite::FieldType kKeyFieldType,
|
|
WireFormatLite::FieldType kValueFieldType>
|
|
class MapFieldLite;
|
|
} // namespace internal
|
|
} // namespace protobuf
|
|
} // namespace google
|
|
|
|
namespace google {
|
|
namespace protobuf {
|
|
namespace internal {
|
|
|
|
// MoveHelper::Move is used to set *dest. It copies *src, or moves it (in
|
|
// the C++11 sense), or swaps it. *src is left in a sane state for
|
|
// subsequent destruction, but shouldn't be used for anything.
|
|
template <bool is_enum, bool is_message, bool is_stringlike, typename T>
|
|
struct MoveHelper { // primitives
|
|
static void Move(T* src, T* dest) { *dest = *src; }
|
|
};
|
|
|
|
template <bool is_message, bool is_stringlike, typename T>
|
|
struct MoveHelper<true, is_message, is_stringlike, T> { // enums
|
|
static void Move(T* src, T* dest) { *dest = *src; }
|
|
// T is an enum here, so allow conversions to and from int.
|
|
static void Move(T* src, int* dest) { *dest = static_cast<int>(*src); }
|
|
static void Move(int* src, T* dest) { *dest = static_cast<T>(*src); }
|
|
};
|
|
|
|
template <bool is_stringlike, typename T>
|
|
struct MoveHelper<false, true, is_stringlike, T> { // messages
|
|
static void Move(T* src, T* dest) { dest->Swap(src); }
|
|
};
|
|
|
|
template <typename T>
|
|
struct MoveHelper<false, false, true, T> { // strings and similar
|
|
static void Move(T* src, T* dest) {
|
|
*dest = std::move(*src);
|
|
}
|
|
};
|
|
|
|
// Functions for operating on a map entry. Does not contain any representation
|
|
// (this class is not intended to be instantiated).
|
|
template <typename Key, typename Value, WireFormatLite::FieldType kKeyFieldType,
|
|
WireFormatLite::FieldType kValueFieldType>
|
|
struct MapEntryFuncs {
|
|
typedef MapTypeHandler<kKeyFieldType, Key> KeyTypeHandler;
|
|
typedef MapTypeHandler<kValueFieldType, Value> ValueTypeHandler;
|
|
static const int kKeyFieldNumber = 1;
|
|
static const int kValueFieldNumber = 2;
|
|
|
|
static uint8_t* InternalSerialize(int field_number, const Key& key,
|
|
const Value& value, uint8_t* ptr,
|
|
io::EpsCopyOutputStream* stream) {
|
|
ptr = stream->EnsureSpace(ptr);
|
|
ptr = WireFormatLite::WriteTagToArray(
|
|
field_number, WireFormatLite::WIRETYPE_LENGTH_DELIMITED, ptr);
|
|
ptr = io::CodedOutputStream::WriteVarint32ToArray(GetCachedSize(key, value),
|
|
ptr);
|
|
|
|
ptr = KeyTypeHandler::Write(kKeyFieldNumber, key, ptr, stream);
|
|
return ValueTypeHandler::Write(kValueFieldNumber, value, ptr, stream);
|
|
}
|
|
|
|
static size_t ByteSizeLong(const Key& key, const Value& value) {
|
|
// Tags for key and value will both be one byte (field numbers 1 and 2).
|
|
size_t inner_length =
|
|
2 + KeyTypeHandler::ByteSize(key) + ValueTypeHandler::ByteSize(value);
|
|
return inner_length + io::CodedOutputStream::VarintSize32(
|
|
static_cast<uint32_t>(inner_length));
|
|
}
|
|
|
|
static int GetCachedSize(const Key& key, const Value& value) {
|
|
// Tags for key and value will both be one byte (field numbers 1 and 2).
|
|
return 2 + KeyTypeHandler::GetCachedSize(key) +
|
|
ValueTypeHandler::GetCachedSize(value);
|
|
}
|
|
};
|
|
|
|
// MapEntryImpl is used to implement parsing and serialization of map entries.
|
|
// It uses Curious Recursive Template Pattern (CRTP) to provide the type of
|
|
// the eventual code to the template code.
|
|
template <typename Derived, typename Base, typename Key, typename Value,
|
|
WireFormatLite::FieldType kKeyFieldType,
|
|
WireFormatLite::FieldType kValueFieldType>
|
|
class MapEntryImpl : public Base {
|
|
public:
|
|
typedef MapEntryFuncs<Key, Value, kKeyFieldType, kValueFieldType> Funcs;
|
|
|
|
protected:
|
|
// Provide utilities to parse/serialize key/value. Provide utilities to
|
|
// manipulate internal stored type.
|
|
typedef MapTypeHandler<kKeyFieldType, Key> KeyTypeHandler;
|
|
typedef MapTypeHandler<kValueFieldType, Value> ValueTypeHandler;
|
|
|
|
// Define internal memory layout. Strings and messages are stored as
|
|
// pointers, while other types are stored as values.
|
|
typedef typename KeyTypeHandler::TypeOnMemory KeyOnMemory;
|
|
typedef typename ValueTypeHandler::TypeOnMemory ValueOnMemory;
|
|
|
|
// Enum type cannot be used for MapTypeHandler::Read. Define a type
|
|
// which will replace Enum with int.
|
|
typedef typename KeyTypeHandler::MapEntryAccessorType KeyMapEntryAccessorType;
|
|
typedef
|
|
typename ValueTypeHandler::MapEntryAccessorType ValueMapEntryAccessorType;
|
|
|
|
// Constants for field number.
|
|
static const int kKeyFieldNumber = 1;
|
|
static const int kValueFieldNumber = 2;
|
|
|
|
// Constants for field tag.
|
|
static const uint8_t kKeyTag =
|
|
GOOGLE_PROTOBUF_WIRE_FORMAT_MAKE_TAG(kKeyFieldNumber, KeyTypeHandler::kWireType);
|
|
static const uint8_t kValueTag = GOOGLE_PROTOBUF_WIRE_FORMAT_MAKE_TAG(
|
|
kValueFieldNumber, ValueTypeHandler::kWireType);
|
|
static const size_t kTagSize = 1;
|
|
|
|
public:
|
|
// Work-around for a compiler bug (see repeated_field.h).
|
|
typedef void MapEntryHasMergeTypeTrait;
|
|
typedef Derived EntryType;
|
|
typedef Key EntryKeyType;
|
|
typedef Value EntryValueType;
|
|
static const WireFormatLite::FieldType kEntryKeyFieldType = kKeyFieldType;
|
|
static const WireFormatLite::FieldType kEntryValueFieldType = kValueFieldType;
|
|
|
|
constexpr MapEntryImpl()
|
|
: key_(KeyTypeHandler::Constinit()),
|
|
value_(ValueTypeHandler::Constinit()),
|
|
_has_bits_{} {}
|
|
|
|
explicit MapEntryImpl(Arena* arena)
|
|
: Base(arena),
|
|
key_(KeyTypeHandler::Constinit()),
|
|
value_(ValueTypeHandler::Constinit()),
|
|
_has_bits_{} {}
|
|
|
|
~MapEntryImpl() {
|
|
if (Base::GetArenaForAllocation() != nullptr) return;
|
|
KeyTypeHandler::DeleteNoArena(key_);
|
|
ValueTypeHandler::DeleteNoArena(value_);
|
|
}
|
|
|
|
// accessors ======================================================
|
|
|
|
virtual inline const KeyMapEntryAccessorType& key() const {
|
|
return KeyTypeHandler::GetExternalReference(key_);
|
|
}
|
|
virtual inline const ValueMapEntryAccessorType& value() const {
|
|
return ValueTypeHandler::DefaultIfNotInitialized(value_);
|
|
}
|
|
inline KeyMapEntryAccessorType* mutable_key() {
|
|
set_has_key();
|
|
return KeyTypeHandler::EnsureMutable(&key_, Base::GetArenaForAllocation());
|
|
}
|
|
inline ValueMapEntryAccessorType* mutable_value() {
|
|
set_has_value();
|
|
return ValueTypeHandler::EnsureMutable(&value_,
|
|
Base::GetArenaForAllocation());
|
|
}
|
|
|
|
// implements MessageLite =========================================
|
|
|
|
// MapEntryImpl is for implementation only and this function isn't called
|
|
// anywhere. Just provide a fake implementation here for MessageLite.
|
|
std::string GetTypeName() const override { return ""; }
|
|
|
|
void CheckTypeAndMergeFrom(const MessageLite& other) override {
|
|
MergeFromInternal(*::google::protobuf::internal::DownCast<const Derived*>(&other));
|
|
}
|
|
|
|
const char* _InternalParse(const char* ptr, ParseContext* ctx) final {
|
|
while (!ctx->Done(&ptr)) {
|
|
uint32_t tag;
|
|
ptr = ReadTag(ptr, &tag);
|
|
GOOGLE_PROTOBUF_PARSER_ASSERT(ptr);
|
|
if (tag == kKeyTag) {
|
|
set_has_key();
|
|
KeyMapEntryAccessorType* key = mutable_key();
|
|
ptr = KeyTypeHandler::Read(ptr, ctx, key);
|
|
if (!Derived::ValidateKey(key)) return nullptr;
|
|
} else if (tag == kValueTag) {
|
|
set_has_value();
|
|
ValueMapEntryAccessorType* value = mutable_value();
|
|
ptr = ValueTypeHandler::Read(ptr, ctx, value);
|
|
if (!Derived::ValidateValue(value)) return nullptr;
|
|
} else {
|
|
if (tag == 0 || WireFormatLite::GetTagWireType(tag) ==
|
|
WireFormatLite::WIRETYPE_END_GROUP) {
|
|
ctx->SetLastTag(tag);
|
|
return ptr;
|
|
}
|
|
ptr = UnknownFieldParse(tag, static_cast<std::string*>(nullptr), ptr,
|
|
ctx);
|
|
}
|
|
GOOGLE_PROTOBUF_PARSER_ASSERT(ptr);
|
|
}
|
|
return ptr;
|
|
}
|
|
|
|
size_t ByteSizeLong() const override {
|
|
size_t size = 0;
|
|
size += kTagSize + static_cast<size_t>(KeyTypeHandler::ByteSize(key()));
|
|
size += kTagSize + static_cast<size_t>(ValueTypeHandler::ByteSize(value()));
|
|
return size;
|
|
}
|
|
|
|
::uint8_t* _InternalSerialize(
|
|
::uint8_t* ptr, io::EpsCopyOutputStream* stream) const override {
|
|
ptr = KeyTypeHandler::Write(kKeyFieldNumber, key(), ptr, stream);
|
|
return ValueTypeHandler::Write(kValueFieldNumber, value(), ptr, stream);
|
|
}
|
|
|
|
// Don't override SerializeWithCachedSizesToArray. Use MessageLite's.
|
|
|
|
int GetCachedSize() const override {
|
|
int size = 0;
|
|
size += has_key() ? static_cast<int>(kTagSize) +
|
|
KeyTypeHandler::GetCachedSize(key())
|
|
: 0;
|
|
size += has_value() ? static_cast<int>(kTagSize) +
|
|
ValueTypeHandler::GetCachedSize(value())
|
|
: 0;
|
|
return size;
|
|
}
|
|
|
|
bool IsInitialized() const override {
|
|
return ValueTypeHandler::IsInitialized(value_);
|
|
}
|
|
|
|
Base* New(Arena* arena) const override {
|
|
Derived* entry = Arena::CreateMessage<Derived>(arena);
|
|
return entry;
|
|
}
|
|
|
|
protected:
|
|
// We can't declare this function directly here as it would hide the other
|
|
// overload (const Message&).
|
|
void MergeFromInternal(const MapEntryImpl& from) {
|
|
if (from._has_bits_[0]) {
|
|
if (from.has_key()) {
|
|
KeyTypeHandler::EnsureMutable(&key_, Base::GetArenaForAllocation());
|
|
KeyTypeHandler::Merge(from.key(), &key_, Base::GetArenaForAllocation());
|
|
set_has_key();
|
|
}
|
|
if (from.has_value()) {
|
|
ValueTypeHandler::EnsureMutable(&value_, Base::GetArenaForAllocation());
|
|
ValueTypeHandler::Merge(from.value(), &value_,
|
|
Base::GetArenaForAllocation());
|
|
set_has_value();
|
|
}
|
|
}
|
|
}
|
|
|
|
public:
|
|
void Clear() override {
|
|
KeyTypeHandler::Clear(&key_, Base::GetArenaForAllocation());
|
|
ValueTypeHandler::Clear(&value_, Base::GetArenaForAllocation());
|
|
clear_has_key();
|
|
clear_has_value();
|
|
}
|
|
|
|
// Parsing using MergePartialFromCodedStream, above, is not as
|
|
// efficient as it could be. This helper class provides a speedier way.
|
|
template <typename MapField, typename Map>
|
|
class Parser {
|
|
public:
|
|
explicit Parser(MapField* mf) : mf_(mf), map_(mf->MutableMap()) {}
|
|
~Parser() {
|
|
if (entry_ != nullptr && entry_->GetArenaForAllocation() == nullptr)
|
|
delete entry_;
|
|
}
|
|
|
|
// This does what the typical MergePartialFromCodedStream() is expected to
|
|
// do, with the additional side-effect that if successful (i.e., if true is
|
|
// going to be its return value) it inserts the key-value pair into map_.
|
|
bool MergePartialFromCodedStream(io::CodedInputStream* input) {
|
|
// Look for the expected thing: a key and then a value. If it fails,
|
|
// invoke the enclosing class's MergePartialFromCodedStream, or return
|
|
// false if that would be pointless.
|
|
if (input->ExpectTag(kKeyTag)) {
|
|
if (!KeyTypeHandler::Read(input, &key_)) {
|
|
return false;
|
|
}
|
|
// Peek at the next byte to see if it is kValueTag. If not, bail out.
|
|
const void* data;
|
|
int size;
|
|
input->GetDirectBufferPointerInline(&data, &size);
|
|
// We could use memcmp here, but we don't bother. The tag is one byte.
|
|
static_assert(kTagSize == 1, "tag size must be 1");
|
|
if (size > 0 && *reinterpret_cast<const char*>(data) == kValueTag) {
|
|
typename Map::size_type map_size = map_->size();
|
|
value_ptr_ = &(*map_)[key_];
|
|
if (PROTOBUF_PREDICT_TRUE(map_size != map_->size())) {
|
|
// We created a new key-value pair. Fill in the value.
|
|
typedef
|
|
typename MapIf<ValueTypeHandler::kIsEnum, int*, Value*>::type T;
|
|
input->Skip(kTagSize); // Skip kValueTag.
|
|
if (!ValueTypeHandler::Read(input,
|
|
reinterpret_cast<T>(value_ptr_))) {
|
|
map_->erase(key_); // Failure! Undo insertion.
|
|
return false;
|
|
}
|
|
if (input->ExpectAtEnd()) return true;
|
|
return ReadBeyondKeyValuePair(input);
|
|
}
|
|
}
|
|
} else {
|
|
key_ = Key();
|
|
}
|
|
|
|
NewEntry();
|
|
*entry_->mutable_key() = key_;
|
|
const bool result = entry_->MergePartialFromCodedStream(input);
|
|
if (result) UseKeyAndValueFromEntry();
|
|
return result;
|
|
}
|
|
|
|
const char* _InternalParse(const char* ptr, ParseContext* ctx) {
|
|
if (PROTOBUF_PREDICT_TRUE(!ctx->Done(&ptr) && *ptr == kKeyTag)) {
|
|
ptr = KeyTypeHandler::Read(ptr + 1, ctx, &key_);
|
|
if (PROTOBUF_PREDICT_FALSE(!ptr || !Derived::ValidateKey(&key_))) {
|
|
return nullptr;
|
|
}
|
|
if (PROTOBUF_PREDICT_TRUE(!ctx->Done(&ptr) && *ptr == kValueTag)) {
|
|
typename Map::size_type map_size = map_->size();
|
|
value_ptr_ = &(*map_)[key_];
|
|
if (PROTOBUF_PREDICT_TRUE(map_size != map_->size())) {
|
|
using T =
|
|
typename MapIf<ValueTypeHandler::kIsEnum, int*, Value*>::type;
|
|
ptr = ValueTypeHandler::Read(ptr + 1, ctx,
|
|
reinterpret_cast<T>(value_ptr_));
|
|
if (PROTOBUF_PREDICT_FALSE(!ptr ||
|
|
!Derived::ValidateValue(value_ptr_))) {
|
|
map_->erase(key_); // Failure! Undo insertion.
|
|
return nullptr;
|
|
}
|
|
if (PROTOBUF_PREDICT_TRUE(ctx->Done(&ptr))) return ptr;
|
|
if (!ptr) return nullptr;
|
|
NewEntry();
|
|
ValueMover::Move(value_ptr_, entry_->mutable_value());
|
|
map_->erase(key_);
|
|
goto move_key;
|
|
}
|
|
} else {
|
|
if (!ptr) return nullptr;
|
|
}
|
|
NewEntry();
|
|
move_key:
|
|
KeyMover::Move(&key_, entry_->mutable_key());
|
|
} else {
|
|
if (!ptr) return nullptr;
|
|
NewEntry();
|
|
}
|
|
ptr = entry_->_InternalParse(ptr, ctx);
|
|
if (ptr) UseKeyAndValueFromEntry();
|
|
return ptr;
|
|
}
|
|
|
|
template <typename UnknownType>
|
|
const char* ParseWithEnumValidation(const char* ptr, ParseContext* ctx,
|
|
bool (*is_valid)(int),
|
|
uint32_t field_num,
|
|
InternalMetadata* metadata) {
|
|
auto entry = NewEntry();
|
|
ptr = entry->_InternalParse(ptr, ctx);
|
|
if (!ptr) return nullptr;
|
|
if (is_valid(entry->value())) {
|
|
UseKeyAndValueFromEntry();
|
|
} else {
|
|
WriteLengthDelimited(field_num, entry->SerializeAsString(),
|
|
metadata->mutable_unknown_fields<UnknownType>());
|
|
}
|
|
return ptr;
|
|
}
|
|
|
|
MapEntryImpl* NewEntry() { return entry_ = mf_->NewEntry(); }
|
|
|
|
const Key& key() const { return key_; }
|
|
const Value& value() const { return *value_ptr_; }
|
|
|
|
const Key& entry_key() const { return entry_->key(); }
|
|
const Value& entry_value() const { return entry_->value(); }
|
|
|
|
private:
|
|
void UseKeyAndValueFromEntry() {
|
|
// Update key_ in case we need it later (because key() is called).
|
|
// This is potentially inefficient, especially if the key is
|
|
// expensive to copy (e.g., a long string), but this is a cold
|
|
// path, so it's not a big deal.
|
|
key_ = entry_->key();
|
|
value_ptr_ = &(*map_)[key_];
|
|
ValueMover::Move(entry_->mutable_value(), value_ptr_);
|
|
}
|
|
|
|
// After reading a key and value successfully, and inserting that data
|
|
// into map_, we are not at the end of the input. This is unusual, but
|
|
// allowed by the spec.
|
|
bool ReadBeyondKeyValuePair(io::CodedInputStream* input) PROTOBUF_COLD {
|
|
NewEntry();
|
|
ValueMover::Move(value_ptr_, entry_->mutable_value());
|
|
map_->erase(key_);
|
|
KeyMover::Move(&key_, entry_->mutable_key());
|
|
const bool result = entry_->MergePartialFromCodedStream(input);
|
|
if (result) UseKeyAndValueFromEntry();
|
|
return result;
|
|
}
|
|
|
|
typedef MoveHelper<KeyTypeHandler::kIsEnum, KeyTypeHandler::kIsMessage,
|
|
KeyTypeHandler::kWireType ==
|
|
WireFormatLite::WIRETYPE_LENGTH_DELIMITED,
|
|
Key>
|
|
KeyMover;
|
|
typedef MoveHelper<ValueTypeHandler::kIsEnum, ValueTypeHandler::kIsMessage,
|
|
ValueTypeHandler::kWireType ==
|
|
WireFormatLite::WIRETYPE_LENGTH_DELIMITED,
|
|
Value>
|
|
ValueMover;
|
|
|
|
MapField* const mf_;
|
|
Map* const map_;
|
|
Key key_;
|
|
Value* value_ptr_;
|
|
MapEntryImpl* entry_ = nullptr;
|
|
};
|
|
|
|
protected:
|
|
void set_has_key() { _has_bits_[0] |= 0x00000001u; }
|
|
bool has_key() const { return (_has_bits_[0] & 0x00000001u) != 0; }
|
|
void clear_has_key() { _has_bits_[0] &= ~0x00000001u; }
|
|
void set_has_value() { _has_bits_[0] |= 0x00000002u; }
|
|
bool has_value() const { return (_has_bits_[0] & 0x00000002u) != 0; }
|
|
void clear_has_value() { _has_bits_[0] &= ~0x00000002u; }
|
|
|
|
public:
|
|
inline Arena* GetArena() const { return Base::GetArena(); }
|
|
|
|
public: // Needed for constructing tables
|
|
KeyOnMemory key_;
|
|
ValueOnMemory value_;
|
|
uint32_t _has_bits_[1];
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::Arena;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
template <typename C, typename K, typename V, WireFormatLite::FieldType,
|
|
WireFormatLite::FieldType>
|
|
friend class internal::MapEntry;
|
|
template <typename C, typename K, typename V, WireFormatLite::FieldType,
|
|
WireFormatLite::FieldType>
|
|
friend class internal::MapFieldLite;
|
|
|
|
GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(MapEntryImpl);
|
|
};
|
|
|
|
template <typename T, typename Key, typename Value,
|
|
WireFormatLite::FieldType kKeyFieldType,
|
|
WireFormatLite::FieldType kValueFieldType>
|
|
class MapEntryLite : public MapEntryImpl<T, MessageLite, Key, Value,
|
|
kKeyFieldType, kValueFieldType> {
|
|
public:
|
|
typedef MapEntryImpl<T, MessageLite, Key, Value, kKeyFieldType,
|
|
kValueFieldType>
|
|
SuperType;
|
|
constexpr MapEntryLite() {}
|
|
explicit MapEntryLite(Arena* arena) : SuperType(arena) {}
|
|
~MapEntryLite() {
|
|
MessageLite::_internal_metadata_.template Delete<std::string>();
|
|
}
|
|
void MergeFrom(const MapEntryLite& other) { MergeFromInternal(other); }
|
|
|
|
private:
|
|
GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(MapEntryLite);
|
|
};
|
|
// The completely unprincipled and unwieldy use of template parameters in
|
|
// the map code necessitates wrappers to make the code a little bit more
|
|
// manageable.
|
|
template <typename Derived>
|
|
struct DeconstructMapEntry;
|
|
|
|
template <typename T, typename K, typename V, WireFormatLite::FieldType key,
|
|
WireFormatLite::FieldType value>
|
|
struct DeconstructMapEntry<MapEntryLite<T, K, V, key, value> > {
|
|
typedef K Key;
|
|
typedef V Value;
|
|
static const WireFormatLite::FieldType kKeyFieldType = key;
|
|
static const WireFormatLite::FieldType kValueFieldType = value;
|
|
};
|
|
|
|
// Helpers for deterministic serialization =============================
|
|
|
|
// This struct can be used with any generic sorting algorithm. If the Key
|
|
// type is relatively small and easy to copy then copying Keys into an
|
|
// array of SortItems can be beneficial. Then all the data the sorting
|
|
// algorithm needs to touch is in that one array.
|
|
template <typename Key, typename PtrToKeyValuePair>
|
|
struct SortItem {
|
|
SortItem() {}
|
|
explicit SortItem(PtrToKeyValuePair p) : first(p->first), second(p) {}
|
|
|
|
Key first;
|
|
PtrToKeyValuePair second;
|
|
};
|
|
|
|
template <typename T>
|
|
struct CompareByFirstField {
|
|
bool operator()(const T& a, const T& b) const { return a.first < b.first; }
|
|
};
|
|
|
|
template <typename T>
|
|
struct CompareByDerefFirst {
|
|
bool operator()(const T& a, const T& b) const { return a->first < b->first; }
|
|
};
|
|
|
|
// Helper for table driven serialization
|
|
|
|
template <WireFormatLite::FieldType FieldType>
|
|
struct FromHelper {
|
|
template <typename T>
|
|
static const T& From(const T& x) {
|
|
return x;
|
|
}
|
|
};
|
|
|
|
template <>
|
|
struct FromHelper<WireFormatLite::TYPE_STRING> {
|
|
static ArenaStringPtr From(const std::string& x) {
|
|
ArenaStringPtr res;
|
|
TaggedPtr<std::string> ptr;
|
|
ptr.Set(const_cast<std::string*>(&x));
|
|
res.UnsafeSetTaggedPointer(ptr);
|
|
return res;
|
|
}
|
|
};
|
|
template <>
|
|
struct FromHelper<WireFormatLite::TYPE_BYTES> {
|
|
static ArenaStringPtr From(const std::string& x) {
|
|
ArenaStringPtr res;
|
|
TaggedPtr<std::string> ptr;
|
|
ptr.Set(const_cast<std::string*>(&x));
|
|
res.UnsafeSetTaggedPointer(ptr);
|
|
return res;
|
|
}
|
|
};
|
|
template <>
|
|
struct FromHelper<WireFormatLite::TYPE_MESSAGE> {
|
|
template <typename T>
|
|
static T* From(const T& x) {
|
|
return const_cast<T*>(&x);
|
|
}
|
|
};
|
|
|
|
template <typename MapEntryType>
|
|
struct MapEntryHelper;
|
|
|
|
template <typename T, typename Key, typename Value,
|
|
WireFormatLite::FieldType kKeyFieldType,
|
|
WireFormatLite::FieldType kValueFieldType>
|
|
struct MapEntryHelper<
|
|
MapEntryLite<T, Key, Value, kKeyFieldType, kValueFieldType> > {
|
|
// Provide utilities to parse/serialize key/value. Provide utilities to
|
|
// manipulate internal stored type.
|
|
typedef MapTypeHandler<kKeyFieldType, Key> KeyTypeHandler;
|
|
typedef MapTypeHandler<kValueFieldType, Value> ValueTypeHandler;
|
|
|
|
// Define internal memory layout. Strings and messages are stored as
|
|
// pointers, while other types are stored as values.
|
|
typedef typename KeyTypeHandler::TypeOnMemory KeyOnMemory;
|
|
typedef typename ValueTypeHandler::TypeOnMemory ValueOnMemory;
|
|
|
|
explicit MapEntryHelper(const MapPair<Key, Value>& map_pair)
|
|
: _has_bits_(3),
|
|
_cached_size_(2 + KeyTypeHandler::GetCachedSize(map_pair.first) +
|
|
ValueTypeHandler::GetCachedSize(map_pair.second)),
|
|
key_(FromHelper<kKeyFieldType>::From(map_pair.first)),
|
|
value_(FromHelper<kValueFieldType>::From(map_pair.second)) {}
|
|
|
|
// Purposely not following the style guide naming. These are the names
|
|
// the proto compiler would generate given the map entry descriptor.
|
|
// The proto compiler generates the offsets in this struct as if this was
|
|
// a regular message. This way the table driven code barely notices it's
|
|
// dealing with a map field.
|
|
uint32_t _has_bits_; // NOLINT
|
|
uint32_t _cached_size_; // NOLINT
|
|
KeyOnMemory key_; // NOLINT
|
|
ValueOnMemory value_; // NOLINT
|
|
};
|
|
|
|
} // namespace internal
|
|
} // namespace protobuf
|
|
} // namespace google
|
|
|
|
#include <thirdparty/protobuf/port_undef.inc>
|
|
|
|
#endif // GOOGLE_PROTOBUF_MAP_ENTRY_LITE_H__
|