| // Protocol Buffers - Google's data interchange format | |
| // Copyright 2008 Google Inc. 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 | |
| // Author: kenton@google.com (Kenton Varda) | |
| // Based on original Protocol Buffers design by | |
| // Sanjay Ghemawat, Jeff Dean, and others. | |
| // | |
| // Defines Message, the abstract interface implemented by non-lite | |
| // protocol message objects. | |
| // | |
| // This is only intended to be extended by protoc created gencode or types | |
| // defined in the Protobuf runtime. It is not intended or supported for | |
| // application code to extend this class, and any protected methods may be | |
| // removed without being it being considered a breaking change as long as the | |
| // corresponding gencode does not use it. | |
| // | |
| // Example usage: | |
| // | |
| // Say you have a message defined as: | |
| // | |
| // message Foo { | |
| // optional string text = 1; | |
| // repeated int32 numbers = 2; | |
| // } | |
| // | |
| // Then, if you used the protocol compiler to generate a class from the above | |
| // definition, you could use it like so: | |
| // | |
| // std::string data; // Will store a serialized version of the message. | |
| // | |
| // { | |
| // // Create a message and serialize it. | |
| // Foo foo; | |
| // foo.set_text("Hello World!"); | |
| // foo.add_numbers(1); | |
| // foo.add_numbers(5); | |
| // foo.add_numbers(42); | |
| // | |
| // foo.SerializeToString(&data); | |
| // } | |
| // | |
| // { | |
| // // Parse the serialized message and check that it contains the | |
| // // correct data. | |
| // Foo foo; | |
| // foo.ParseFromString(data); | |
| // | |
| // assert(foo.text() == "Hello World!"); | |
| // assert(foo.numbers_size() == 3); | |
| // assert(foo.numbers(0) == 1); | |
| // assert(foo.numbers(1) == 5); | |
| // assert(foo.numbers(2) == 42); | |
| // } | |
| // | |
| // { | |
| // // Same as the last block, but do it dynamically via the Message | |
| // // reflection interface. | |
| // Message* foo = new Foo; | |
| // const Descriptor* descriptor = foo->GetDescriptor(); | |
| // | |
| // // Get the descriptors for the fields we're interested in and verify | |
| // // their types. | |
| // const FieldDescriptor* text_field = descriptor->FindFieldByName("text"); | |
| // assert(text_field != nullptr); | |
| // assert(text_field->type() == FieldDescriptor::TYPE_STRING); | |
| // assert(!text_field->is_required()); | |
| // assert(!text_field->is_repeated()); | |
| // const FieldDescriptor* numbers_field = descriptor-> | |
| // FindFieldByName("numbers"); | |
| // assert(numbers_field != nullptr); | |
| // assert(numbers_field->type() == FieldDescriptor::TYPE_INT32); | |
| // assert(numbers_field->is_repeated()); | |
| // | |
| // // Parse the message. | |
| // foo->ParseFromString(data); | |
| // | |
| // // Use the reflection interface to examine the contents. | |
| // const Reflection* reflection = foo->GetReflection(); | |
| // assert(reflection->GetString(*foo, text_field) == "Hello World!"); | |
| // assert(reflection->FieldSize(*foo, numbers_field) == 3); | |
| // assert(reflection->GetRepeatedInt32(*foo, numbers_field, 0) == 1); | |
| // assert(reflection->GetRepeatedInt32(*foo, numbers_field, 1) == 5); | |
| // assert(reflection->GetRepeatedInt32(*foo, numbers_field, 2) == 42); | |
| // | |
| // delete foo; | |
| // } | |
| #ifndef GOOGLE_PROTOBUF_MESSAGE_H__ | |
| #define GOOGLE_PROTOBUF_MESSAGE_H__ | |
| #include <cstddef> | |
| #include <cstdint> | |
| #include <memory> | |
| #include <optional> | |
| #include <string> | |
| #include <type_traits> | |
| #include <utility> | |
| #include <vector> | |
| #include "absl/base/attributes.h" | |
| #include "absl/base/call_once.h" | |
| #include "absl/base/macros.h" | |
| #include "absl/base/optimization.h" | |
| #include "absl/log/absl_check.h" | |
| #include "absl/memory/memory.h" | |
| #include "absl/strings/cord.h" | |
| #include "absl/strings/string_view.h" | |
| #include "absl/types/optional.h" | |
| #include "google/protobuf/arena.h" | |
| #include "google/protobuf/descriptor.h" | |
| #include "google/protobuf/generated_message_reflection.h" | |
| #include "google/protobuf/generated_message_tctable_decl.h" | |
| #include "google/protobuf/generated_message_util.h" | |
| #include "google/protobuf/map.h" // TODO: cleanup | |
| #include "google/protobuf/message_lite.h" | |
| #include "google/protobuf/port.h" | |
| #include "google/protobuf/reflection.h" | |
| // Must be included last. | |
| #include "google/protobuf/port_def.inc" | |
| #ifdef SWIG | |
| #error "You cannot SWIG proto headers" | |
| #endif | |
| namespace google { | |
| namespace protobuf { | |
| // Defined in this file. | |
| class Message; | |
| class Reflection; | |
| class MessageFactory; | |
| // Defined in other files. | |
| class AssignDescriptorsHelper; | |
| class ConstMapIterator; | |
| class DynamicMessageFactory; | |
| class GeneratedMessageReflectionTestHelper; | |
| class MapKey; | |
| class MapValueConstRef; | |
| class MapValueRef; | |
| class MapIterator; | |
| class MapReflectionTester; | |
| class TextFormat; | |
| namespace internal { | |
| struct FuzzPeer; | |
| struct DescriptorTable; | |
| struct DescriptorMethodsFriend; | |
| template <bool is_oneof> | |
| struct DynamicFieldInfoHelper; | |
| class HasBitsTestPeer; | |
| template <typename MessageT> | |
| struct MapDynamicFieldInfo; | |
| class MapFieldBase; | |
| class MessageUtil; | |
| class ReflectionVisit; | |
| class SwapFieldHelper; | |
| class CachedSize; | |
| struct TailCallTableInfo; | |
| template <typename MessageT, typename FieldT> | |
| struct RepeatedEntityDynamicFieldInfoBase; | |
| template <typename MessageT, typename FieldT> | |
| struct RepeatedPtrEntityDynamicFieldInfoBase; | |
| class LazyFieldForUnion; | |
| namespace field_layout { | |
| enum TransformValidation : uint16_t; | |
| } // namespace field_layout | |
| } // namespace internal | |
| class UnknownFieldSet; // unknown_field_set.h | |
| namespace io { | |
| class EpsCopyOutputStream; // coded_stream.h | |
| class ZeroCopyInputStream; // zero_copy_stream.h | |
| class ZeroCopyOutputStream; // zero_copy_stream.h | |
| class CodedInputStream; // coded_stream.h | |
| class CodedOutputStream; // coded_stream.h | |
| } // namespace io | |
| namespace python { | |
| class RepeatedScalarContainerFriend; // repeated_scalar_container.cc | |
| class MapReflectionFriend; // scalar_map_container.h | |
| class MessageReflectionFriend; | |
| } // namespace python | |
| namespace expr { | |
| class CelMapReflectionFriend; // field_backed_map_impl.cc | |
| class SudoMapReflectionFriend; | |
| } // namespace expr | |
| namespace internal { | |
| class MapFieldPrinterHelper; // text_format.cc | |
| PROTOBUF_EXPORT std::string StringifyMessage( | |
| const Message& message); // text_format.cc | |
| } // namespace internal | |
| PROTOBUF_EXPORT std::string ShortFormat( | |
| const Message& message); // text_format.cc | |
| PROTOBUF_EXPORT std::string Utf8Format( | |
| const Message& message); // text_format.cc | |
| namespace util { | |
| class MessageDifferencer; | |
| } // namespace util | |
| namespace json_internal { | |
| struct UnparseProto2Descriptor; | |
| } // namespace json_internal | |
| namespace internal { | |
| class ReflectionAccessor; // message.cc | |
| class ReflectionOps; // reflection_ops.h | |
| class MapKeySorter; // wire_format.cc | |
| class WireFormat; // wire_format.h | |
| class MapFieldReflectionTest; // map_test.cc | |
| } // namespace internal | |
| template <typename T> | |
| class RepeatedField; // repeated_field.h | |
| template <typename T> | |
| class RepeatedPtrField; // repeated_field.h | |
| // A container to hold message metadata. | |
| struct Metadata { | |
| const Descriptor* descriptor; | |
| const Reflection* reflection; | |
| }; | |
| namespace internal { | |
| template <class To> | |
| inline To* GetPointerAtOffset(void* message, uint32_t offset) { | |
| return reinterpret_cast<To*>(reinterpret_cast<char*>(message) + offset); | |
| } | |
| template <class To> | |
| const To* GetConstPointerAtOffset(const void* message, uint32_t offset) { | |
| return reinterpret_cast<const To*>(reinterpret_cast<const char*>(message) + | |
| offset); | |
| } | |
| template <class To> | |
| const To& GetConstRefAtOffset(const Message& message, uint32_t offset) { | |
| return *GetConstPointerAtOffset<To>(&message, offset); | |
| } | |
| bool CreateUnknownEnumValues(const FieldDescriptor* field); | |
| // Returns true if "message" is a descendant of "root". | |
| PROTOBUF_EXPORT bool IsDescendant(const Message& root, const Message& message); | |
| inline void MaybePoisonAfterClear(Message* root); | |
| } // namespace internal | |
| // Abstract interface for protocol messages. | |
| // | |
| // See also MessageLite, which contains most every-day operations. Message | |
| // adds descriptors and reflection on top of that. | |
| // | |
| // The methods of this class that are virtual but not pure-virtual have | |
| // default implementations based on reflection. Message classes which are | |
| // optimized for speed will want to override these with faster implementations, | |
| // but classes optimized for code size may be happy with keeping them. See | |
| // the optimize_for option in descriptor.proto. | |
| // | |
| // Users must not derive from this class. Only the protocol compiler and | |
| // the internal library are allowed to create subclasses. | |
| class PROTOBUF_EXPORT Message : public MessageLite { | |
| public: | |
| Message(const Message&) = delete; | |
| Message& operator=(const Message&) = delete; | |
| // Basic Operations ------------------------------------------------ | |
| // Construct a new instance of the same type. Ownership is passed to the | |
| // caller. (This is also defined in MessageLite, but is defined again here | |
| // for return-type covariance.) | |
| [[nodiscard]] Message* New() const { return New(/*arena=*/nullptr); } | |
| // Construct a new instance on the arena. Ownership is passed to the caller | |
| // if arena is a nullptr. | |
| [[nodiscard]] Message* New(Arena* arena) const { | |
| return static_cast<Message*>(MessageLite::New(arena)); | |
| } | |
| // Make this message into a copy of the given message. The given message | |
| // must have the same descriptor, but need not necessarily be the same class. | |
| // By default this is just implemented as "Clear(); MergeFrom(from);". | |
| void CopyFrom(const Message& from); | |
| // Merge the fields from the given message into this message. Singular | |
| // fields will be overwritten, if specified in from, except for embedded | |
| // messages which will be merged. Repeated fields will be concatenated. | |
| // The given message must be of the same type as this message (i.e. the | |
| // exact same class). | |
| void MergeFrom(const Message& from); | |
| // Verifies that IsInitialized() returns true. ABSL_CHECK-fails otherwise, | |
| // with a nice error message. | |
| void CheckInitialized() const; | |
| // Slowly build a list of all required fields that are not set. | |
| // This is much, much slower than IsInitialized() as it is implemented | |
| // purely via reflection. Generally, you should not call this unless you | |
| // have already determined that an error exists by calling IsInitialized(). | |
| void FindInitializationErrors(std::vector<std::string>* errors) const; | |
| // Like FindInitializationErrors, but joins all the strings, delimited by | |
| // commas, and returns them. | |
| [[nodiscard]] std::string InitializationErrorString() const; | |
| // Clears all unknown fields from this message and all embedded messages. | |
| // Normally, if unknown tag numbers are encountered when parsing a message, | |
| // the tag and value are stored in the message's UnknownFieldSet and | |
| // then written back out when the message is serialized. This allows servers | |
| // which simply route messages to other servers to pass through messages | |
| // that have new field definitions which they don't yet know about. However, | |
| // this behavior can have security implications. To avoid it, call this | |
| // method after parsing. | |
| // | |
| // See Reflection::GetUnknownFields() for more on unknown fields. | |
| void DiscardUnknownFields(); | |
| // Computes (an estimate of) the total number of bytes currently used for | |
| // storing the message in memory. | |
| // | |
| // SpaceUsedLong() is noticeably slower than ByteSizeLong(), as it is | |
| // implemented using reflection (rather than the generated code implementation | |
| // for ByteSizeLong()). Like ByteSizeLong(), its CPU time is linear in the | |
| // number of fields defined for the proto. | |
| // | |
| // Note: The precise value of this method should never be depended on, and can | |
| // change substantially due to internal details. In debug builds, this will | |
| // include a random fuzz factor to prevent these dependencies. | |
| [[nodiscard]] size_t SpaceUsedLong() const; | |
| [[deprecated("Please use SpaceUsedLong() instead")]] int SpaceUsed() const { | |
| return internal::ToIntSize(SpaceUsedLong()); | |
| } | |
| // Debugging & Testing---------------------------------------------- | |
| // Generates a human-readable form of this message for debugging purposes. | |
| // Note that the format and content of a debug string is not guaranteed, may | |
| // change without notice, and should not be depended on. Code that does | |
| // anything except display a string to assist in debugging should use | |
| // TextFormat instead. | |
| [[nodiscard]] std::string DebugString() const; | |
| // Like DebugString(), but with less whitespace. | |
| [[nodiscard]] std::string ShortDebugString() const; | |
| // Like DebugString(), but do not escape UTF-8 byte sequences. | |
| [[nodiscard]] std::string Utf8DebugString() const; | |
| // Convenience function useful in GDB. Prints DebugString() to stdout. | |
| void PrintDebugString() const; | |
| // Implementation of the `AbslStringify` interface. This adds something | |
| // similar to either `ShortDebugString()` or `DebugString()` to the sink. | |
| // Do not rely on exact format. | |
| template <typename Sink> | |
| friend void AbslStringify(Sink& sink, const google::protobuf::Message& message) { | |
| sink.Append(internal::StringifyMessage(message)); | |
| } | |
| // Reflection-based methods ---------------------------------------- | |
| // These methods are pure-virtual in MessageLite, but Message provides | |
| // reflection-based default implementations. | |
| #if !defined(PROTOBUF_CUSTOM_VTABLE) | |
| void Clear() override; | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD size_t ByteSizeLong() const override; | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD uint8_t* _InternalSerialize( | |
| uint8_t* target, io::EpsCopyOutputStream* stream) const override; | |
| #endif // !PROTOBUF_CUSTOM_VTABLE | |
| // Introspection --------------------------------------------------- | |
| // Get a non-owning pointer to a Descriptor for this message's type. This | |
| // describes what fields the message contains, the types of those fields, etc. | |
| // This object remains property of the Message. | |
| [[nodiscard]] const Descriptor* GetDescriptor() const { | |
| return GetMetadata().descriptor; | |
| } | |
| // Get a non-owning pointer to the Reflection interface for this Message, | |
| // which can be used to read and modify the fields of the Message dynamically | |
| // (in other words, without knowing the message type at compile time). This | |
| // object remains property of the Message. | |
| [[nodiscard]] const Reflection* GetReflection() const { | |
| return GetMetadata().reflection; | |
| } | |
| // Abseil flag support for Messages. | |
| // | |
| // Flag syntax is `:format,options...:value` where: | |
| // - `format` is one of `text`, `serialized`. | |
| // - `options` is a possibly empty list of options. Each format has its | |
| // supported options. | |
| // - `value` is the payload in the specified format. | |
| // | |
| // The valid options are: | |
| // | |
| // * For `text`: | |
| // - `base64`: indicates that `value` is encoded as base64. | |
| // - `ignore_unknown`: when specified, unknown field/extensions are | |
| // dropped. Otherwise, they cause a parse failure. | |
| // | |
| // * For `serialized`: | |
| // - `base64`: indicates that `value` is encoded as base64. It is | |
| // recommended to use `serialized` with `base64` given that passing | |
| // binary data in shells is difficult and error prone. | |
| friend bool AbslParseFlag(absl::string_view text, Message* msg, | |
| std::string* error) { | |
| return msg->AbslParseFlagImpl(text, *error); | |
| } | |
| friend std::string AbslUnparseFlag(const Message& msg) { | |
| return msg.AbslUnparseFlagImpl(); | |
| } | |
| protected: | |
| #if !defined(PROTOBUF_CUSTOM_VTABLE) | |
| constexpr Message() {} | |
| #endif // PROTOBUF_CUSTOM_VTABLE | |
| #if defined(PROTOBUF_PROTECTED_MESSAGE_BASE_DESTRUCTOR) | |
| // Explicitly define the destructor as protected so it can't be called | |
| // directly. | |
| ~Message() = default; | |
| #endif // PROTOBUF_PROTECTED_MESSAGE_BASE_DESTRUCTOR | |
| using MessageLite::MessageLite; | |
| // Get a struct containing the metadata for the Message, which is used in turn | |
| // to implement GetDescriptor() and GetReflection() above. | |
| Metadata GetMetadata() const; | |
| static Metadata GetMetadataImpl(const internal::ClassData& data); | |
| // For CODE_SIZE types | |
| static bool IsInitializedImpl(const MessageLite&); | |
| bool AbslParseFlagImpl(absl::string_view text, std::string& error); | |
| std::string AbslUnparseFlagImpl() const; | |
| size_t ComputeUnknownFieldsSize( | |
| size_t total_size, const internal::CachedSize* cached_size) const; | |
| size_t MaybeComputeUnknownFieldsSize( | |
| size_t total_size, const internal::CachedSize* cached_size) const; | |
| // Reflection based version for reflection based types. | |
| static void MergeImpl(MessageLite& to, const MessageLite& from); | |
| static void ClearImpl(MessageLite& msg); | |
| static size_t ByteSizeLongImpl(const MessageLite& msg); | |
| static uint8_t* _InternalSerializeImpl(const MessageLite& msg, | |
| uint8_t* target, | |
| io::EpsCopyOutputStream* stream); | |
| }; | |
| namespace internal { | |
| // Creates and returns an allocation for a split message. | |
| void CreateSplitMessageGeneric(Arena* arena, void** split, size_t size); | |
| // Forward-declare interfaces used to implement RepeatedFieldRef. | |
| // These are protobuf internals that users shouldn't care about. | |
| class RepeatedFieldAccessor; | |
| extern PROTOBUF_EXPORT const DescriptorMethods kDescriptorMethods; | |
| } // namespace internal | |
| // This interface contains methods that can be used to dynamically access | |
| // and modify the fields of a protocol message. Their semantics are | |
| // similar to the accessors the protocol compiler generates. | |
| // | |
| // To get the Reflection for a given Message, call Message::GetReflection(). | |
| // | |
| // This interface is separate from Message only for efficiency reasons; | |
| // the vast majority of implementations of Message will share the same | |
| // implementation of Reflection (GeneratedMessageReflection, | |
| // defined in generated_message.h), and all Messages of a particular class | |
| // should share the same Reflection object (though you should not rely on | |
| // the latter fact). | |
| // | |
| // There are several ways that these methods can be used incorrectly. For | |
| // example, any of the following conditions will lead to undefined | |
| // results (probably assertion failures): | |
| // - The FieldDescriptor is not a field of this message type. | |
| // - The method called is not appropriate for the field's type. For | |
| // each field type in FieldDescriptor::TYPE_*, there is only one | |
| // Get*() method, one Set*() method, and one Add*() method that is | |
| // valid for that type. It should be obvious which (except maybe | |
| // for TYPE_BYTES, which are represented using strings in C++). | |
| // - A Get*() or Set*() method for singular fields is called on a repeated | |
| // field. | |
| // - GetRepeated*(), SetRepeated*(), or Add*() is called on a non-repeated | |
| // field. | |
| // - The Message object passed to any method is not of the right type for | |
| // this Reflection object (i.e. message.GetReflection() != reflection). | |
| // | |
| // You might wonder why there is not any abstract representation for a field | |
| // of arbitrary type. E.g., why isn't there just a "GetField()" method that | |
| // returns "const Field&", where "Field" is some class with accessors like | |
| // "GetInt32Value()". The problem is that someone would have to deal with | |
| // allocating these Field objects. For generated message classes, having to | |
| // allocate space for an additional object to wrap every field would at least | |
| // double the message's memory footprint, probably worse. Allocating the | |
| // objects on-demand, on the other hand, would be expensive and prone to | |
| // memory leaks. So, instead we ended up with this flat interface. | |
| class PROTOBUF_EXPORT Reflection final { | |
| public: | |
| Reflection(const Reflection&) = delete; | |
| Reflection& operator=(const Reflection&) = delete; | |
| ~Reflection(); | |
| // Get the UnknownFieldSet for the message. This contains fields which | |
| // were seen when the Message was parsed but were not recognized according | |
| // to the Message's definition. | |
| [[nodiscard]] const UnknownFieldSet& GetUnknownFields( | |
| const Message& message) const; | |
| // Get a mutable pointer to the UnknownFieldSet for the message. This | |
| // contains fields which were seen when the Message was parsed but were not | |
| // recognized according to the Message's definition. | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD UnknownFieldSet* MutableUnknownFields( | |
| Message* message) const; | |
| // Estimate the amount of memory used by the message object. | |
| [[nodiscard]] size_t SpaceUsedLong(const Message& message) const; | |
| [[nodiscard]] [[deprecated("Please use SpaceUsedLong() instead")]] int | |
| SpaceUsed(const Message& message) const { | |
| return internal::ToIntSize(SpaceUsedLong(message)); | |
| } | |
| // Returns true if the given message is a default message instance. | |
| [[nodiscard]] bool IsDefaultInstance(const Message& message) const { | |
| ABSL_DCHECK_EQ(message.GetReflection(), this); | |
| return schema_.IsDefaultInstance(message); | |
| } | |
| // Check if the given non-repeated field is set. | |
| [[nodiscard]] bool HasField(const Message& message, | |
| const FieldDescriptor* field) const; | |
| // Get the number of elements of a repeated field. | |
| [[nodiscard]] int FieldSize(const Message& message, | |
| const FieldDescriptor* field) const; | |
| // Clear the value of a field, so that HasField() returns false or | |
| // FieldSize() returns zero. | |
| void ClearField(Message* message, const FieldDescriptor* field) const; | |
| // Check if the oneof is set. Returns true if any field in oneof | |
| // is set, false otherwise. | |
| [[nodiscard]] bool HasOneof(const Message& message, | |
| const OneofDescriptor* oneof_descriptor) const; | |
| void ClearOneof(Message* message, | |
| const OneofDescriptor* oneof_descriptor) const; | |
| // Returns the field descriptor if the oneof is set. nullptr otherwise. | |
| [[nodiscard]] const FieldDescriptor* GetOneofFieldDescriptor( | |
| const Message& message, const OneofDescriptor* oneof_descriptor) const; | |
| // Removes the last element of a repeated field. | |
| // We don't provide a way to remove any element other than the last | |
| // because it invites inefficient use, such as O(n^2) filtering loops | |
| // that should have been O(n). If you want to remove an element other | |
| // than the last, the best way to do it is to re-arrange the elements | |
| // (using Swap()) so that the one you want removed is at the end, then | |
| // call RemoveLast(). | |
| void RemoveLast(Message* message, const FieldDescriptor* field) const; | |
| // Removes the last element of a repeated message field, and returns the | |
| // pointer to the caller. Caller takes ownership of the returned pointer. | |
| [[nodiscard]] Message* ReleaseLast(Message* message, | |
| const FieldDescriptor* field) const; | |
| // Similar to ReleaseLast() without internal safety and ownershp checks. This | |
| // method should only be used when the objects are on the same arena or paired | |
| // with a call to `UnsafeArenaAddAllocatedMessage`. | |
| Message* UnsafeArenaReleaseLast(Message* message, | |
| const FieldDescriptor* field) const; | |
| // Swap the complete contents of two messages. | |
| void Swap(Message* message1, Message* message2) const; | |
| // Swap fields listed in fields vector of two messages. | |
| void SwapFields(Message* message1, Message* message2, | |
| const std::vector<const FieldDescriptor*>& fields) const; | |
| // Swap two elements of a repeated field. | |
| void SwapElements(Message* message, const FieldDescriptor* field, int index1, | |
| int index2) const; | |
| // Swap without internal safety and ownership checks. This method should only | |
| // be used when the objects are on the same arena. | |
| void UnsafeArenaSwap(Message* lhs, Message* rhs) const; | |
| // SwapFields without internal safety and ownership checks. This method should | |
| // only be used when the objects are on the same arena. | |
| void UnsafeArenaSwapFields( | |
| Message* lhs, Message* rhs, | |
| const std::vector<const FieldDescriptor*>& fields) const; | |
| // List all fields of the message which are currently set, except for unknown | |
| // fields, but including extension known to the parser (i.e. compiled in). | |
| // Singular fields will only be listed if HasField(field) would return true | |
| // and repeated fields will only be listed if FieldSize(field) would return | |
| // non-zero. Fields (both normal fields and extension fields) will be listed | |
| // ordered by field number. | |
| // Use Reflection::GetUnknownFields() or message.unknown_fields() to also get | |
| // access to fields/extensions unknown to the parser. | |
| void ListFields(const Message& message, | |
| std::vector<const FieldDescriptor*>* output) const; | |
| // Returns true if ListFields would have given no results, and there are | |
| // no unknown fields. | |
| [[nodiscard]] bool IsEmpty(const Message& message) const; | |
| // Returns true if ListFields would have given no results. | |
| [[nodiscard]] bool IsEmptyIgnoringUnknownFields(const Message& message) const; | |
| // Singular field getters ------------------------------------------ | |
| // These get the value of a non-repeated field. They return the default | |
| // value for fields that aren't set. | |
| [[nodiscard]] int32_t GetInt32(const Message& message, | |
| const FieldDescriptor* field) const; | |
| [[nodiscard]] int64_t GetInt64(const Message& message, | |
| const FieldDescriptor* field) const; | |
| [[nodiscard]] uint32_t GetUInt32(const Message& message, | |
| const FieldDescriptor* field) const; | |
| [[nodiscard]] uint64_t GetUInt64(const Message& message, | |
| const FieldDescriptor* field) const; | |
| [[nodiscard]] float GetFloat(const Message& message, | |
| const FieldDescriptor* field) const; | |
| [[nodiscard]] double GetDouble(const Message& message, | |
| const FieldDescriptor* field) const; | |
| [[nodiscard]] bool GetBool(const Message& message, | |
| const FieldDescriptor* field) const; | |
| [[nodiscard]] std::string GetString(const Message& message, | |
| const FieldDescriptor* field) const; | |
| [[nodiscard]] [[deprecated( | |
| "Please use GetEnumValue() instead. GetEnum() will be " | |
| "removed in Q1 2027.")]] const EnumValueDescriptor* | |
| GetEnum(const Message& message, const FieldDescriptor* field) const; | |
| // GetEnumValue() returns an enum field's value as an integer rather than | |
| // an EnumValueDescriptor*. If the integer value does not correspond to a | |
| // known value descriptor, a new value descriptor is created. (Such a value | |
| // will only be present when the new unknown-enum-value semantics are enabled | |
| // for a message.) | |
| [[nodiscard]] int GetEnumValue(const Message& message, | |
| const FieldDescriptor* field) const; | |
| // See MutableMessage() for the meaning of the "factory" parameter. | |
| [[nodiscard]] const Message& GetMessage( | |
| const Message& message, const FieldDescriptor* field, | |
| MessageFactory* factory = nullptr) const; | |
| // Get a string value without copying, if possible. | |
| // | |
| // GetString() necessarily returns a copy of the string. This can be | |
| // inefficient when the std::string is already stored in a std::string object | |
| // in the underlying message. GetStringReference() will return a reference to | |
| // the underlying std::string in this case. Otherwise, it will copy the | |
| // string into *scratch and return that. | |
| // | |
| // Note: It is perfectly reasonable and useful to write code like: | |
| // str = reflection->GetStringReference(message, field, &str); | |
| // This line would ensure that only one copy of the string is made | |
| // regardless of the field's underlying representation. When initializing | |
| // a newly-constructed string, though, it's just as fast and more | |
| // readable to use code like: | |
| // std::string str = reflection->GetString(message, field); | |
| [[nodiscard]] const std::string& GetStringReference( | |
| const Message& message, const FieldDescriptor* field, | |
| std::string* scratch) const; | |
| // Returns a Cord containing the value of the string field. If the | |
| // underlying field is stored as a cord (e.g. it has the [ctype=CORD] | |
| // option), this involves no copies (just reference counting). If the | |
| // underlying representation is not a Cord, a copy will have to be made. | |
| [[nodiscard]] absl::Cord GetCord(const Message& message, | |
| const FieldDescriptor* field) const; | |
| // Enables GetStringView() and GetRepeatedStringView() APIs to return | |
| // absl::string_view even though the underlying implementation doesn't have | |
| // contiguous bytes; e.g. absl::Cord. | |
| class ScratchSpace { | |
| public: | |
| ScratchSpace() = default; | |
| ScratchSpace(const ScratchSpace&) = delete; | |
| ScratchSpace& operator=(const ScratchSpace&) = delete; | |
| private: | |
| friend class Reflection; | |
| absl::string_view CopyFromCord(const absl::Cord& cord) { | |
| if (auto flat = cord.TryFlat()) { | |
| return *flat; | |
| } | |
| if (!buffer_) { | |
| buffer_ = std::make_unique<std::string>(); | |
| } | |
| absl::CopyCordToString(cord, buffer_.get()); | |
| return *buffer_; | |
| } | |
| std::unique_ptr<std::string> buffer_; | |
| }; | |
| // Returns a view into the contents of a string field. "scratch" is used to | |
| // flatten bytes if it is non-contiguous. The lifetime of absl::string_view is | |
| // either tied to "message" (contiguous) or "scratch" (otherwise). | |
| [[nodiscard]] absl::string_view GetStringView( | |
| const Message& message, const FieldDescriptor* field, | |
| ScratchSpace& scratch ABSL_ATTRIBUTE_LIFETIME_BOUND) const; | |
| // Singular field mutators ----------------------------------------- | |
| // These mutate the value of a non-repeated field. | |
| void SetInt32(Message* message, const FieldDescriptor* field, | |
| int32_t value) const; | |
| void SetInt64(Message* message, const FieldDescriptor* field, | |
| int64_t value) const; | |
| void SetUInt32(Message* message, const FieldDescriptor* field, | |
| uint32_t value) const; | |
| void SetUInt64(Message* message, const FieldDescriptor* field, | |
| uint64_t value) const; | |
| void SetFloat(Message* message, const FieldDescriptor* field, | |
| float value) const; | |
| void SetDouble(Message* message, const FieldDescriptor* field, | |
| double value) const; | |
| void SetBool(Message* message, const FieldDescriptor* field, | |
| bool value) const; | |
| void SetString(Message* message, const FieldDescriptor* field, | |
| std::string value) const; | |
| // Set a string field to a Cord value. If the underlying field is | |
| // represented using a Cord already, this involves no copies (just | |
| // reference counting). Otherwise, a copy must be made. | |
| void SetString(Message* message, const FieldDescriptor* field, | |
| const absl::Cord& value) const; | |
| void SetEnum(Message* message, const FieldDescriptor* field, | |
| const EnumValueDescriptor* value) const; | |
| // Set an enum field's value with an integer rather than EnumValueDescriptor. | |
| // For proto3 this is just setting the enum field to the value specified, for | |
| // proto2 it's more complicated. If value is a known enum value the field is | |
| // set as usual. If the value is unknown then it is added to the unknown field | |
| // set. Note this matches the behavior of parsing unknown enum values. | |
| // If multiple calls with unknown values happen than they are all added to the | |
| // unknown field set in order of the calls. | |
| void SetEnumValue(Message* message, const FieldDescriptor* field, | |
| int value) const; | |
| // Get a mutable pointer to a field with a message type. If a MessageFactory | |
| // is provided, it will be used to construct instances of the sub-message; | |
| // otherwise, the default factory is used. If the field is an extension that | |
| // does not live in the same pool as the containing message's descriptor (e.g. | |
| // it lives in an overlay pool), then a MessageFactory must be provided. | |
| // If you have no idea what that meant, then you probably don't need to worry | |
| // about it (don't provide a MessageFactory). WARNING: If the | |
| // FieldDescriptor is for a compiled-in extension, then | |
| // factory->GetPrototype(field->message_type()) MUST return an instance of | |
| // the compiled-in class for this type, NOT DynamicMessage. | |
| Message* MutableMessage(Message* message, const FieldDescriptor* field, | |
| MessageFactory* factory = nullptr) const; | |
| // Replaces the message specified by 'field' with the already-allocated object | |
| // sub_message, passing ownership to the message. If the field contained a | |
| // message, that message is deleted. If sub_message is nullptr, the field is | |
| // cleared. | |
| void SetAllocatedMessage(Message* message, Message* sub_message, | |
| const FieldDescriptor* field) const; | |
| // Similar to `SetAllocatedMessage`, but omits all internal safety and | |
| // ownership checks. This method should only be used when the objects are on | |
| // the same arena or paired with a call to `UnsafeArenaReleaseMessage`. | |
| void UnsafeArenaSetAllocatedMessage(Message* message, Message* sub_message, | |
| const FieldDescriptor* field) const; | |
| // Releases the message specified by 'field' and returns the pointer, | |
| // ReleaseMessage() will return the message the message object if it exists. | |
| // Otherwise, it may or may not return nullptr. In any case, if the return | |
| // value is non-null, the caller takes ownership of the pointer. | |
| // If the field existed (HasField() is true), then the returned pointer will | |
| // be the same as the pointer returned by MutableMessage(). | |
| // This function has the same effect as ClearField(). | |
| [[nodiscard]] Message* ReleaseMessage( | |
| Message* message, const FieldDescriptor* field, | |
| MessageFactory* factory = nullptr) const; | |
| // Similar to `ReleaseMessage`, but omits all internal safety and ownership | |
| // checks. This method should only be used when the objects are on the same | |
| // arena or paired with a call to `UnsafeArenaSetAllocatedMessage`. | |
| Message* UnsafeArenaReleaseMessage(Message* message, | |
| const FieldDescriptor* field, | |
| MessageFactory* factory = nullptr) const; | |
| // Repeated field getters ------------------------------------------ | |
| // These get the value of one element of a repeated field. | |
| [[nodiscard]] int32_t GetRepeatedInt32(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] int64_t GetRepeatedInt64(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] uint32_t GetRepeatedUInt32(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] uint64_t GetRepeatedUInt64(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] float GetRepeatedFloat(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] double GetRepeatedDouble(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] bool GetRepeatedBool(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] std::string GetRepeatedString(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] [[deprecated( | |
| "Please use GetRepeatedEnumValue() instead. GetRepeatedEnum() will be " | |
| "removed in Q1 2027.")]] const EnumValueDescriptor* | |
| GetRepeatedEnum(const Message& message, const FieldDescriptor* field, | |
| int index) const; | |
| // GetRepeatedEnumValue() returns an enum field's value as an integer rather | |
| // than an EnumValueDescriptor*. If the integer value does not correspond to a | |
| // known value descriptor, a new value descriptor is created. (Such a value | |
| // will only be present when the new unknown-enum-value semantics are enabled | |
| // for a message.) | |
| [[nodiscard]] int GetRepeatedEnumValue(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| [[nodiscard]] const Message& GetRepeatedMessage(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| // See GetStringReference(), above. | |
| [[nodiscard]] const std::string& GetRepeatedStringReference( | |
| const Message& message, const FieldDescriptor* field, int index, | |
| std::string* scratch) const; | |
| // See GetStringView(), above. | |
| [[nodiscard]] absl::string_view GetRepeatedStringView( | |
| const Message& message, const FieldDescriptor* field, int index, | |
| ScratchSpace& scratch ABSL_ATTRIBUTE_LIFETIME_BOUND) const; | |
| // Repeated field mutators ----------------------------------------- | |
| // These mutate the value of one element of a repeated field. | |
| void SetRepeatedInt32(Message* message, const FieldDescriptor* field, | |
| int index, int32_t value) const; | |
| void SetRepeatedInt64(Message* message, const FieldDescriptor* field, | |
| int index, int64_t value) const; | |
| void SetRepeatedUInt32(Message* message, const FieldDescriptor* field, | |
| int index, uint32_t value) const; | |
| void SetRepeatedUInt64(Message* message, const FieldDescriptor* field, | |
| int index, uint64_t value) const; | |
| void SetRepeatedFloat(Message* message, const FieldDescriptor* field, | |
| int index, float value) const; | |
| void SetRepeatedDouble(Message* message, const FieldDescriptor* field, | |
| int index, double value) const; | |
| void SetRepeatedBool(Message* message, const FieldDescriptor* field, | |
| int index, bool value) const; | |
| void SetRepeatedString(Message* message, const FieldDescriptor* field, | |
| int index, std::string value) const; | |
| void SetRepeatedEnum(Message* message, const FieldDescriptor* field, | |
| int index, const EnumValueDescriptor* value) const; | |
| // Set an enum field's value with an integer rather than EnumValueDescriptor. | |
| // For proto3 this is just setting the enum field to the value specified, for | |
| // proto2 it's more complicated. If value is a known enum value the field is | |
| // set as usual. If the value is unknown then it is added to the unknown field | |
| // set. Note this matches the behavior of parsing unknown enum values. | |
| // If multiple calls with unknown values happen than they are all added to the | |
| // unknown field set in order of the calls. | |
| void SetRepeatedEnumValue(Message* message, const FieldDescriptor* field, | |
| int index, int value) const; | |
| // Get a mutable pointer to an element of a repeated field with a message | |
| // type. | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD Message* MutableRepeatedMessage( | |
| Message* message, const FieldDescriptor* field, int index) const; | |
| // Repeated field adders ------------------------------------------- | |
| // These add an element to a repeated field. | |
| void AddInt32(Message* message, const FieldDescriptor* field, | |
| int32_t value) const; | |
| void AddInt64(Message* message, const FieldDescriptor* field, | |
| int64_t value) const; | |
| void AddUInt32(Message* message, const FieldDescriptor* field, | |
| uint32_t value) const; | |
| void AddUInt64(Message* message, const FieldDescriptor* field, | |
| uint64_t value) const; | |
| void AddFloat(Message* message, const FieldDescriptor* field, | |
| float value) const; | |
| void AddDouble(Message* message, const FieldDescriptor* field, | |
| double value) const; | |
| void AddBool(Message* message, const FieldDescriptor* field, | |
| bool value) const; | |
| void AddString(Message* message, const FieldDescriptor* field, | |
| std::string value) const; | |
| void AddEnum(Message* message, const FieldDescriptor* field, | |
| const EnumValueDescriptor* value) const; | |
| // Add an integer value to a repeated enum field rather than | |
| // EnumValueDescriptor. For proto3 this is just setting the enum field to the | |
| // value specified, for proto2 it's more complicated. If value is a known enum | |
| // value the field is set as usual. If the value is unknown then it is added | |
| // to the unknown field set. Note this matches the behavior of parsing unknown | |
| // enum values. If multiple calls with unknown values happen than they are all | |
| // added to the unknown field set in order of the calls. | |
| void AddEnumValue(Message* message, const FieldDescriptor* field, | |
| int value) const; | |
| // See MutableMessage() for comments on the "factory" parameter. | |
| Message* AddMessage(Message* message, const FieldDescriptor* field, | |
| MessageFactory* factory = nullptr) const; | |
| // Appends an already-allocated object 'new_entry' to the repeated field | |
| // specified by 'field' passing ownership to the message. | |
| void AddAllocatedMessage(Message* message, const FieldDescriptor* field, | |
| Message* new_entry) const; | |
| // Similar to AddAllocatedMessage() without internal safety and ownership | |
| // checks. This method should only be used when the objects are on the same | |
| // arena or paired with a call to `UnsafeArenaReleaseLast`. | |
| void UnsafeArenaAddAllocatedMessage(Message* message, | |
| const FieldDescriptor* field, | |
| Message* new_entry) const; | |
| // Get a RepeatedFieldRef object that can be used to read the underlying | |
| // repeated field. The type parameter T must be set according to the | |
| // field's cpp type. The following table shows the mapping from cpp type | |
| // to acceptable T. | |
| // | |
| // field->cpp_type() T | |
| // CPPTYPE_INT32 int32_t | |
| // CPPTYPE_UINT32 uint32_t | |
| // CPPTYPE_INT64 int64_t | |
| // CPPTYPE_UINT64 uint64_t | |
| // CPPTYPE_DOUBLE double | |
| // CPPTYPE_FLOAT float | |
| // CPPTYPE_BOOL bool | |
| // CPPTYPE_ENUM generated enum type or int32_t | |
| // CPPTYPE_STRING std::string | |
| // CPPTYPE_MESSAGE generated message type or google::protobuf::Message | |
| // | |
| // A RepeatedFieldRef object can be copied and the resulted object will point | |
| // to the same repeated field in the same message. The object can be used as | |
| // long as the message is not destroyed. | |
| // | |
| // Note that to use this method users need to include the header file | |
| // "reflection.h" (which defines the RepeatedFieldRef class templates). | |
| template <typename T> | |
| [[nodiscard]] RepeatedFieldRef<T> GetRepeatedFieldRef( | |
| const Message& message, const FieldDescriptor* field) const; | |
| // Like GetRepeatedFieldRef() but return an object that can also be used | |
| // manipulate the underlying repeated field. | |
| template <typename T> | |
| MutableRepeatedFieldRef<T> GetMutableRepeatedFieldRef( | |
| Message* message, const FieldDescriptor* field) const; | |
| // DEPRECATED. Please use Get(Mutable)RepeatedFieldRef() for repeated field | |
| // access. The following repeated field accessors will be removed in the | |
| // future. | |
| // | |
| // Repeated field accessors ------------------------------------------------- | |
| // The methods above, e.g. GetRepeatedInt32(msg, fd, index), provide singular | |
| // access to the data in a RepeatedField. The methods below provide aggregate | |
| // access by exposing the RepeatedField object itself with the Message. | |
| // Applying these templates to inappropriate types will lead to an undefined | |
| // reference at link time (e.g. GetRepeatedField<***double>), or possibly a | |
| // template matching error at compile time (e.g. GetRepeatedPtrField<File>). | |
| // | |
| // Usage example: my_doubs = refl->GetRepeatedField<double>(msg, fd); | |
| // DEPRECATED. Please use GetRepeatedFieldRef(). | |
| // | |
| // for T = Cord and all protobuf scalar types except enums. | |
| template <typename T> | |
| [[nodiscard]] [[deprecated( | |
| "Please use GetRepeatedFieldRef() instead")]] const RepeatedField<T>& | |
| GetRepeatedField(const Message& msg, const FieldDescriptor* d) const { | |
| return GetRepeatedFieldInternal<T>(msg, d, | |
| GetRepeatedFieldIntent::kExposeDirectly); | |
| } | |
| // DEPRECATED. Please use GetMutableRepeatedFieldRef(). | |
| // | |
| // for T = Cord and all protobuf scalar types except enums. | |
| template <typename T> | |
| [[deprecated( | |
| "Please use GetMutableRepeatedFieldRef() instead")]] RepeatedField<T>* | |
| MutableRepeatedField(Message* msg, const FieldDescriptor* d) const { | |
| return MutableRepeatedFieldInternal<T>( | |
| msg, d, GetRepeatedFieldIntent::kExposeDirectly); | |
| } | |
| // DEPRECATED. Please use GetRepeatedFieldRef(). | |
| // | |
| // for T = std::string, google::protobuf::internal::StringPieceField | |
| // google::protobuf::Message & descendants. | |
| template <typename T> | |
| [[nodiscard]] [[deprecated( | |
| "Please use GetRepeatedFieldRef() instead")]] const RepeatedPtrField<T>& | |
| GetRepeatedPtrField(const Message& msg, const FieldDescriptor* d) const { | |
| return GetRepeatedPtrFieldInternal<T>( | |
| msg, d, GetRepeatedFieldIntent::kExposeDirectly); | |
| } | |
| // DEPRECATED. Please use GetMutableRepeatedFieldRef(). | |
| // | |
| // for T = std::string, google::protobuf::internal::StringPieceField | |
| // google::protobuf::Message & descendants. | |
| template <typename T> | |
| [[deprecated( | |
| "Please use GetMutableRepeatedFieldRef() instead")]] RepeatedPtrField<T>* | |
| MutableRepeatedPtrField(Message* msg, const FieldDescriptor* d) const { | |
| return MutableRepeatedPtrFieldInternal<T>( | |
| msg, d, GetRepeatedFieldIntent::kExposeDirectly); | |
| } | |
| // Extensions ---------------------------------------------------------------- | |
| // Try to find an extension of this message type by fully-qualified field | |
| // name. Returns nullptr if no extension is known for this name or number. | |
| [[nodiscard]] const FieldDescriptor* FindKnownExtensionByName( | |
| absl::string_view name) const; | |
| // Try to find an extension of this message type by field number. | |
| // Returns nullptr if no extension is known for this name or number. | |
| [[nodiscard]] const FieldDescriptor* FindKnownExtensionByNumber( | |
| int number) const; | |
| // Returns the MessageFactory associated with this message. This can be | |
| // useful for determining if a message is a generated message or not, for | |
| // example: | |
| // if (message->GetReflection()->GetMessageFactory() == | |
| // google::protobuf::MessageFactory::generated_factory()) { | |
| // // This is a generated message. | |
| // } | |
| // It can also be used to create more messages of this type, though | |
| // Message::New() is an easier way to accomplish this. | |
| [[nodiscard]] MessageFactory* GetMessageFactory() const; | |
| private: | |
| enum class GetRepeatedFieldIntent { | |
| // The caller intents to return a reference/pointer to the raw repeated | |
| // field directly to application code. | |
| kExposeDirectly, | |
| // The caller will either wrap the raw repeated field in a proxy before | |
| // returning it to application code, or will not return it to application | |
| // code. | |
| kHiddenOrInternal, | |
| }; | |
| const internal::ReflectionSchema& Schema() const { return schema_; } | |
| bool IsRepeatedOrMapFieldEmpty(const Message& message, | |
| const FieldDescriptor* field) const; | |
| template <typename T> | |
| const RepeatedField<T>& GetRepeatedFieldInternal( | |
| const Message& message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const; | |
| template <typename T> | |
| RepeatedField<T>* MutableRepeatedFieldInternal( | |
| Message* message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const; | |
| template <typename T> | |
| const RepeatedPtrField<T>& GetRepeatedPtrFieldInternal( | |
| const Message& message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const; | |
| template <typename T> | |
| RepeatedPtrField<T>* MutableRepeatedPtrFieldInternal( | |
| Message* message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const; | |
| // REQUIRES: If the field is Cord, then `scratch != nullptr`. | |
| absl::string_view GetStringViewImpl(const Message& message, | |
| const FieldDescriptor* field, | |
| ScratchSpace* scratch) const; | |
| absl::string_view GetRepeatedStringViewImpl(const Message& message, | |
| const FieldDescriptor* field, | |
| int index, | |
| ScratchSpace* scratch) const; | |
| // Obtain a pointer to a Repeated Field Structure and do some type checking: | |
| // on field->cpp_type(), | |
| // on field->field_option().ctype() (if ctype >= 0) | |
| // of field->message_type() (if message_type != nullptr). | |
| // We use 2 routine rather than 4 (const vs mutable) x (scalar vs pointer). | |
| void* MutableRawRepeatedField(Message* message, const FieldDescriptor* field, | |
| FieldDescriptor::CppType cpptype, int ctype, | |
| const Descriptor* desc, | |
| GetRepeatedFieldIntent intent) const; | |
| const void* GetRawRepeatedField(const Message& message, | |
| const FieldDescriptor* field, | |
| FieldDescriptor::CppType cpptype, int ctype, | |
| const Descriptor* desc, | |
| GetRepeatedFieldIntent intent) const; | |
| // The following methods are used to implement (Mutable)RepeatedFieldRef. | |
| // A Ref object will store a raw pointer to the repeated field data (obtained | |
| // from RepeatedFieldData()) and a pointer to a Accessor (obtained from | |
| // RepeatedFieldAccessor) which will be used to access the raw data. | |
| // Returns a raw pointer to the repeated field | |
| // | |
| // "cpp_type" and "message_type" are deduced from the type parameter T passed | |
| // to Get(Mutable)RepeatedFieldRef. If T is a generated message type, | |
| // "message_type" should be set to its descriptor. Otherwise "message_type" | |
| // should be set to nullptr. Implementations of this method should check | |
| // whether "cpp_type"/"message_type" is consistent with the actual type of the | |
| // field. | |
| const void* RepeatedFieldData(const Message& message, | |
| const FieldDescriptor* field, | |
| FieldDescriptor::CppType cpp_type, | |
| const Descriptor* message_type) const; | |
| void* RepeatedFieldData(Message* message, const FieldDescriptor* field, | |
| FieldDescriptor::CppType cpp_type, | |
| const Descriptor* message_type) const; | |
| // The returned pointer should point to a singleton instance which implements | |
| // the RepeatedFieldAccessor interface. | |
| const internal::RepeatedFieldAccessor* RepeatedFieldAccessor( | |
| const FieldDescriptor* field) const; | |
| // Returns true if the message field is backed by a LazyField. | |
| // | |
| // A message field may be backed by a LazyField without the user annotation | |
| // ([lazy = true]). While the user-annotated LazyField is lazily verified on | |
| // first touch (i.e. failure on access rather than parsing if the LazyField is | |
| // not initialized), the inferred LazyField is eagerly verified to avoid lazy | |
| // parsing error at the cost of lower efficiency. When reflecting a message | |
| // field, use this API instead of checking field->options().lazy(). | |
| bool IsLazyField(const FieldDescriptor* field) const { | |
| return IsLazilyVerifiedLazyField(field) || | |
| IsEagerlyVerifiedLazyField(field); | |
| } | |
| // Returns true if the field is lazy extension. It is meant to allow python | |
| // reparse lazy field until b/157559327 is fixed. | |
| bool IsLazyExtension(const Message& message, | |
| const FieldDescriptor* field) const; | |
| bool IsLazilyVerifiedLazyField(const FieldDescriptor* field) const; | |
| bool IsEagerlyVerifiedLazyField(const FieldDescriptor* field) const; | |
| internal::field_layout::TransformValidation GetLazyStyle( | |
| const FieldDescriptor* field) const; | |
| bool IsSplit(const FieldDescriptor* field) const { | |
| return schema_.IsSplit(field); | |
| } | |
| // Walks the message tree from "root" and poisons (under ASAN) the memory to | |
| // force subsequent accesses to fail. Always calls Clear beforehand to clear | |
| // strings, etc. | |
| void MaybePoisonAfterClear(Message& root) const; | |
| friend class FastReflectionBase; | |
| friend class FastReflectionMessageMutator; | |
| friend class internal::HasBitsTestPeer; | |
| template <typename MessageT> | |
| friend struct internal::MapDynamicFieldInfo; | |
| friend class internal::ReflectionVisit; | |
| friend internal::DescriptorMethodsFriend; | |
| friend json_internal::UnparseProto2Descriptor; | |
| friend bool internal::IsDescendant(const Message& root, | |
| const Message& message); | |
| friend void internal::MaybePoisonAfterClear(Message* root); | |
| // The table-driven parser table. | |
| // This table is generated on demand for Message types that did not override | |
| // _InternalParse. It uses the reflection information to do so. | |
| mutable absl::once_flag tcparse_table_once_; | |
| const Descriptor* const descriptor_; | |
| MessageFactory* const message_factory_; | |
| const DescriptorPool* const descriptor_pool_; | |
| const internal::ReflectionSchema schema_; | |
| using TcParseTableBase = internal::TcParseTableBase; | |
| mutable const TcParseTableBase* tcparse_table_ = nullptr; | |
| const TcParseTableBase* GetTcParseTable() const { | |
| absl::call_once(tcparse_table_once_, | |
| [&] { tcparse_table_ = CreateTcParseTable(); }); | |
| return tcparse_table_; | |
| } | |
| const TcParseTableBase* CreateTcParseTable() const; | |
| void PopulateTcParseFastEntries(const internal::TailCallTableInfo& table_info, | |
| TcParseTableBase* tc_table) const; | |
| void PopulateTcParseEntries(internal::TailCallTableInfo& table_info, | |
| TcParseTableBase::FieldEntry* entries) const; | |
| void PopulateTcParseFieldAux(const internal::TailCallTableInfo& table_info, | |
| TcParseTableBase::FieldAux* field_aux) const; | |
| template <bool kForIsEmpty, typename MaybeFieldDescriptorVec> | |
| int32_t IsEmptyOrCollectSetFields(const Message& message, | |
| const Descriptor& descriptor, | |
| MaybeFieldDescriptorVec output) const; | |
| bool IsEmptyIgnoringUnknownFieldsImpl(const Message& message) const; | |
| template <typename T, typename Enable> | |
| friend class RepeatedFieldRef; | |
| template <typename T, typename Enable> | |
| friend class MutableRepeatedFieldRef; | |
| template <typename MessageT, typename FieldT> | |
| friend struct internal::RepeatedEntityDynamicFieldInfoBase; | |
| template <typename MessageT, typename FieldT> | |
| friend struct internal::RepeatedPtrEntityDynamicFieldInfoBase; | |
| friend class Message; | |
| friend class MessageLayoutInspector; | |
| friend class AssignDescriptorsHelper; | |
| friend class DynamicMessageFactory; | |
| friend class GeneratedMessageReflectionTestHelper; | |
| friend class python::MapReflectionFriend; | |
| friend class python::MessageReflectionFriend; | |
| friend class python::RepeatedScalarContainerFriend; | |
| friend class util::MessageDifferencer; | |
| #define GOOGLE_PROTOBUF_HAS_CEL_MAP_REFLECTION_FRIEND | |
| friend class expr::CelMapReflectionFriend; | |
| friend class internal::MapFieldReflectionTest; | |
| friend class internal::MapKeySorter; | |
| friend class internal::MessageUtil; | |
| friend class internal::WireFormat; | |
| friend class internal::ReflectionOps; | |
| friend class internal::SwapFieldHelper; | |
| template <bool is_oneof> | |
| friend struct internal::DynamicFieldInfoHelper; | |
| friend struct internal::FuzzPeer; | |
| // Needed for implementing text format for map. | |
| friend class internal::MapFieldPrinterHelper; | |
| Reflection(const Descriptor* descriptor, | |
| const internal::ReflectionSchema& schema, | |
| const DescriptorPool* pool, MessageFactory* factory); | |
| // Special version for specialized implementations of string. We can't | |
| // call MutableRawRepeatedField directly here because we don't have access to | |
| // FieldOptions::* which are defined in descriptor.pb.h. Including that | |
| // file here is not possible because it would cause a circular include cycle. | |
| const void* GetRawRepeatedString(const Message& message, | |
| const FieldDescriptor* field, bool is_string, | |
| GetRepeatedFieldIntent intent) const; | |
| void* MutableRawRepeatedString(Message* message, const FieldDescriptor* field, | |
| bool is_string, | |
| GetRepeatedFieldIntent intent) const; | |
| friend class MapReflectionTester; | |
| // Returns true if key is in map. Returns false if key is not in map field. | |
| bool ContainsMapKey(const Message& message, const FieldDescriptor* field, | |
| const MapKey& key) const; | |
| // If key is in map field: Saves the value pointer to val and returns | |
| // false. If key in not in map field: Insert the key into map, saves | |
| // value pointer to val and returns true. Users are able to modify the | |
| // map value by MapValueRef. | |
| bool InsertOrLookupMapValue(Message* message, const FieldDescriptor* field, | |
| const MapKey& key, MapValueRef* val) const; | |
| // If key is in map field: Saves the value pointer to val and returns true. | |
| // Returns false if key is not in map field. Users are NOT able to modify | |
| // the value by MapValueConstRef. | |
| bool LookupMapValue(const Message& message, const FieldDescriptor* field, | |
| const MapKey& key, MapValueConstRef* val) const; | |
| bool LookupMapValue(const Message&, const FieldDescriptor*, const MapKey&, | |
| MapValueRef*) const = delete; | |
| // Delete and returns true if key is in the map field. Returns false | |
| // otherwise. | |
| bool DeleteMapValue(Message* message, const FieldDescriptor* field, | |
| const MapKey& key) const; | |
| // Returns a MapIterator referring to the first element in the map field. | |
| // If the map field is empty, this function returns the same as | |
| // reflection::MapEnd. Mutation to the field may invalidate the iterator. | |
| MapIterator MapBegin(Message* message, const FieldDescriptor* field) const; | |
| // Returns a MapIterator referring to the theoretical element that would | |
| // follow the last element in the map field. It does not point to any | |
| // real element. Mutation to the field may invalidate the iterator. | |
| MapIterator MapEnd(Message* message, const FieldDescriptor* field) const; | |
| // Returns a ConstMapIterator referring to the first element in the map field. | |
| // If the map field is empty, this function returns the same as | |
| // reflection::ConstMapEnd. Mutation to the field may invalidate the iterator. | |
| ConstMapIterator ConstMapBegin(const Message* message, | |
| const FieldDescriptor* field) const; | |
| // Returns a ConstMapIterator referring to the theoretical element that would | |
| // follow the last element in the map field. It does not point to any | |
| // real element. Mutation to the field may invalidate the iterator. | |
| ConstMapIterator ConstMapEnd(const Message* message, | |
| const FieldDescriptor* field) const; | |
| // Get the number of <key, value> pair of a map field. The result may be | |
| // different from FieldSize which can have duplicate keys. | |
| int MapSize(const Message& message, const FieldDescriptor* field) const; | |
| // Help method for MapIterator. | |
| template <bool> | |
| friend class MapIteratorBase; | |
| friend class WireFormatForMapFieldTest; | |
| internal::MapFieldBase* MutableMapData(Message* message, | |
| const FieldDescriptor* field) const; | |
| const internal::MapFieldBase* GetMapData(const Message& message, | |
| const FieldDescriptor* field) const; | |
| // Check that the type passed for the unsafe cast matches what we expect from | |
| // the field. | |
| // This makes it easier to catch bugs in callers. | |
| template <typename T> | |
| void VerifyFieldType(const FieldDescriptor* field) const; | |
| template <typename Type> | |
| const Type& GetRaw(const Message& message, | |
| const FieldDescriptor* field) const; | |
| void* MutableRawSplitImpl(Message* message, | |
| const FieldDescriptor* field) const; | |
| template <typename Type> | |
| Type* MutableRaw(Message* message, const FieldDescriptor* field) const; | |
| template <typename Type> | |
| const Type& DefaultRaw(const FieldDescriptor* field) const; | |
| const internal::ClassData* GetMessageClassData( | |
| const FieldDescriptor* field) const; | |
| const uint32_t* GetHasBits(const Message& message) const; | |
| inline uint32_t* MutableHasBits(Message* message) const; | |
| uint32_t GetOneofCase(const Message& message, | |
| const OneofDescriptor* oneof_descriptor) const; | |
| inline uint32_t* MutableOneofCase( | |
| Message* message, const OneofDescriptor* oneof_descriptor) const; | |
| bool HasExtensionSet(const Message& /* message */) const { | |
| return schema_.HasExtensionSet(); | |
| } | |
| const internal::ExtensionSet& GetExtensionSet(const Message& message) const; | |
| internal::ExtensionSet* MutableExtensionSet(Message* message) const; | |
| const internal::InternalMetadata& GetInternalMetadata( | |
| const Message& message) const { | |
| return message._internal_metadata_; | |
| } | |
| internal::InternalMetadata* MutableInternalMetadata(Message* message) const { | |
| return &message->_internal_metadata_; | |
| } | |
| bool IsInlined(const FieldDescriptor* field) const { | |
| return schema_.IsFieldInlined(field); | |
| } | |
| bool IsMicroString(const FieldDescriptor* field) const { | |
| return schema_.IsFieldMicroString(field); | |
| } | |
| // For implicit-presence (including repeated) fields, returns true if the | |
| // field is populated, i.e., nonzero/nonempty. False otherwise. | |
| bool IsImplicitPresenceFieldNonEmpty(const Message& message, | |
| const FieldDescriptor* field) const; | |
| // Returns whether the field is present if there are usable hasbits in the | |
| // field schema. (Note that in some cases hasbits are merely a hint to | |
| // indicate "possible presence", and another empty-check is required). | |
| bool IsFieldPresentGivenHasbits(const Message& message, | |
| const FieldDescriptor* field, | |
| const uint32_t* hasbits, | |
| uint32_t hasbit_index) const; | |
| // Returns true if the field is considered to be present. | |
| // Requires the input to be non-extension, non-oneof, non-weak. | |
| // For explicit presence fields, a field is present iff the hasbit is set. | |
| // For implicit presence fields, a field is present iff it is | |
| // nonzero/nonempty. The hasbit may be checked as a shortcut in the empty | |
| // case. | |
| bool HasFieldWithHasbits(const Message& message, | |
| const FieldDescriptor* field) const; | |
| void SetHasBit(Message* message, const FieldDescriptor* field) const; | |
| void ClearHasBit(Message* message, const FieldDescriptor* field) const; | |
| // Naively swaps the hasbit without checking for field existence. | |
| // For explicit presence fields, the hasbit is swapped normally. | |
| // For implicit presence fields, the hasbit is swapped without checking for | |
| // field emptiness. That is, the destination message may have hasbit set even | |
| // if the field is empty. This should still result in correct behaviour due to | |
| // HasbitMode being set to kHintHasbits for implicit presence fields. | |
| inline void NaiveSwapHasBit(Message* message1, Message* message2, | |
| const FieldDescriptor* field) const; | |
| // Returns the `_split_` pointer. Requires: IsSplit() == true. | |
| inline const void* GetSplitField(const Message* message) const; | |
| // Returns the address of the `_split_` pointer. Requires: IsSplit() == true. | |
| inline void** MutableSplitField(Message* message) const; | |
| // Allocate the split instance if needed. | |
| void PrepareSplitMessageForWrite(Message* message) const; | |
| // Shallow-swap fields listed in fields vector of two messages. It is the | |
| // caller's responsibility to make sure shallow swap is safe. | |
| void UnsafeShallowSwapFields( | |
| Message* message1, Message* message2, | |
| const std::vector<const FieldDescriptor*>& fields) const; | |
| // This function only swaps the field. Should swap corresponding has_bit | |
| // before or after using this function. | |
| void SwapField(Message* message1, Message* message2, | |
| const FieldDescriptor* field) const; | |
| // Unsafe but shallow version of SwapField. | |
| void UnsafeShallowSwapField(Message* message1, Message* message2, | |
| const FieldDescriptor* field) const; | |
| template <bool unsafe_shallow_swap> | |
| void SwapFieldsImpl(Message* message1, Message* message2, | |
| const std::vector<const FieldDescriptor*>& fields) const; | |
| template <bool unsafe_shallow_swap, typename FromType, typename ToType> | |
| void InternalMoveOneofField(const FieldDescriptor* field, FromType* from, | |
| ToType* to) const; | |
| template <bool unsafe_shallow_swap> | |
| void SwapOneofField(Message* lhs, Message* rhs, | |
| const OneofDescriptor* oneof_descriptor) const; | |
| void InternalSwap(Message* lhs, Message* rhs) const; | |
| inline bool HasOneofField(const Message& message, | |
| const FieldDescriptor* field) const; | |
| inline void SetOneofCase(Message* message, | |
| const FieldDescriptor* field) const; | |
| void ClearOneofField(Message* message, const FieldDescriptor* field) const; | |
| template <typename Type> | |
| inline const Type& GetField(const Message& message, | |
| const FieldDescriptor* field) const; | |
| template <typename Type> | |
| inline void SetField(Message* message, const FieldDescriptor* field, | |
| const Type& value) const; | |
| template <typename Type> | |
| inline Type* MutableField(Message* message, | |
| const FieldDescriptor* field) const; | |
| template <typename Type> | |
| inline const Type& GetRepeatedField(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| template <typename Type> | |
| inline const Type& GetRepeatedPtrField(const Message& message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| template <typename Type> | |
| inline void SetRepeatedField(Message* message, const FieldDescriptor* field, | |
| int index, Type value) const; | |
| template <typename Type> | |
| inline Type* MutableRepeatedField(Message* message, | |
| const FieldDescriptor* field, | |
| int index) const; | |
| template <typename Type> | |
| inline void AddField(Message* message, const FieldDescriptor* field, | |
| const Type& value) const; | |
| template <typename Type> | |
| inline Type* AddField(Message* message, const FieldDescriptor* field) const; | |
| int GetExtensionNumberOrDie(const Descriptor* type) const; | |
| // Internal versions of EnumValue API perform no checking. Called after checks | |
| // by public methods. | |
| void SetEnumValueInternal(Message* message, const FieldDescriptor* field, | |
| int value) const; | |
| void SetRepeatedEnumValueInternal(Message* message, | |
| const FieldDescriptor* field, int index, | |
| int value) const; | |
| void AddEnumValueInternal(Message* message, const FieldDescriptor* field, | |
| int value) const; | |
| friend inline const char* ParseLenDelim(int field_number, | |
| const FieldDescriptor* field, | |
| Message* msg, | |
| const Reflection* reflection, | |
| const char* ptr, | |
| internal::ParseContext* ctx); | |
| friend inline const char* ParsePackedField(const FieldDescriptor* field, | |
| Message* msg, | |
| const Reflection* reflection, | |
| const char* ptr, | |
| internal::ParseContext* ctx); | |
| void SetStringView(Message* message, const FieldDescriptor* field, | |
| absl::string_view value) const; | |
| void SetRepeatedStringView(Message* message, const FieldDescriptor* field, | |
| int index, absl::string_view value) const; | |
| void AddStringView(Message* message, const FieldDescriptor* field, | |
| absl::string_view value) const; | |
| // Supports const absl::Cord&, std::string&& and absl::string_view. | |
| template <typename String> | |
| void SetStringImpl(Message* message, const FieldDescriptor* field, | |
| String&& value) const; | |
| // Supports const absl::Cord&, std::string&& and absl::string_view. | |
| template <typename String> | |
| void SetRepeatedStringImpl(Message* message, const FieldDescriptor* field, | |
| int index, String&& value) const; | |
| // Supports const absl::Cord&, std::string&& and absl::string_view. | |
| template <typename String> | |
| void AddStringImpl(Message* message, const FieldDescriptor* field, | |
| String&& value) const; | |
| }; | |
| extern template void Reflection::SwapFieldsImpl<true>( | |
| Message* message1, Message* message2, | |
| const std::vector<const FieldDescriptor*>& fields) const; | |
| extern template void Reflection::SwapFieldsImpl<false>( | |
| Message* message1, Message* message2, | |
| const std::vector<const FieldDescriptor*>& fields) const; | |
| // Abstract interface for a factory for message objects. | |
| // | |
| // The thread safety for this class is implementation dependent, see comments | |
| // around GetPrototype for details | |
| class PROTOBUF_EXPORT MessageFactory { | |
| public: | |
| MessageFactory() = default; | |
| MessageFactory(const MessageFactory&) = delete; | |
| MessageFactory& operator=(const MessageFactory&) = delete; | |
| virtual ~MessageFactory(); | |
| // Given a Descriptor, gets or constructs the default (prototype) Message | |
| // of that type. You can then call that message's New() method to construct | |
| // a mutable message of that type. | |
| // | |
| // Calling this method twice with the same Descriptor returns the same | |
| // object. The returned object remains property of the factory. Also, any | |
| // objects created by calling the prototype's New() method share some data | |
| // with the prototype, so these must be destroyed before the MessageFactory | |
| // is destroyed. | |
| // | |
| // The given descriptor must outlive the returned message, and hence must | |
| // outlive the MessageFactory. | |
| // | |
| // Some implementations do not support all types. GetPrototype() will | |
| // return nullptr if the descriptor passed in is not supported. | |
| // | |
| // This method may or may not be thread-safe depending on the implementation. | |
| // Each implementation should document its own degree thread-safety. | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD virtual const Message* GetPrototype( | |
| const Descriptor* type) = 0; | |
| // Gets a MessageFactory which supports all generated, compiled-in messages. | |
| // In other words, for any compiled-in type FooMessage, the following is true: | |
| // MessageFactory::generated_factory()->GetPrototype( | |
| // FooMessage::descriptor()) == FooMessage::default_instance() | |
| // This factory supports all types which are found in | |
| // DescriptorPool::generated_pool(). If given a descriptor from any other | |
| // pool, GetPrototype() will return nullptr. (You can also check if a | |
| // descriptor is for a generated message by checking if | |
| // descriptor->file()->pool() == DescriptorPool::generated_pool().) | |
| // | |
| // This factory is 100% thread-safe; calling GetPrototype() does not modify | |
| // any shared data. | |
| // | |
| // This factory is a singleton. The caller must not delete the object. | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD static MessageFactory* | |
| generated_factory(); | |
| // For internal use only: Registers a .proto file at static initialization | |
| // time, to be placed in generated_factory. The first time GetPrototype() | |
| // is called with a descriptor from this file, |register_messages| will be | |
| // called, with the file name as the parameter. It must call | |
| // InternalRegisterGeneratedMessage() (below) to register each message type | |
| // in the file. This strange mechanism is necessary because descriptors are | |
| // built lazily, so we can't register types by their descriptor until we | |
| // know that the descriptor exists. |filename| must be a permanent string. | |
| static void InternalRegisterGeneratedFile( | |
| const google::protobuf::internal::DescriptorTable* table); | |
| // For internal use only: Registers a message type. Called only by the | |
| // functions which are registered with InternalRegisterGeneratedFile(), | |
| // above. | |
| static void InternalRegisterGeneratedMessage(const Descriptor* descriptor, | |
| const Message* prototype); | |
| private: | |
| friend class DynamicMessageFactory; | |
| static const Message* TryGetGeneratedPrototype(const Descriptor* type); | |
| }; | |
| #define DECLARE_GET_REPEATED_FIELD(TYPE) \ | |
| template <> \ | |
| PROTOBUF_EXPORT const RepeatedField<TYPE>& \ | |
| Reflection::GetRepeatedFieldInternal<TYPE>( \ | |
| const Message& message, const FieldDescriptor* field, \ | |
| GetRepeatedFieldIntent intent) const; \ | |
| \ | |
| template <> \ | |
| PROTOBUF_EXPORT RepeatedField<TYPE>* \ | |
| Reflection::MutableRepeatedFieldInternal<TYPE>( \ | |
| Message * message, const FieldDescriptor* field, \ | |
| GetRepeatedFieldIntent intent) const; | |
| DECLARE_GET_REPEATED_FIELD(int32_t) | |
| DECLARE_GET_REPEATED_FIELD(int64_t) | |
| DECLARE_GET_REPEATED_FIELD(uint32_t) | |
| DECLARE_GET_REPEATED_FIELD(uint64_t) | |
| DECLARE_GET_REPEATED_FIELD(float) | |
| DECLARE_GET_REPEATED_FIELD(double) | |
| DECLARE_GET_REPEATED_FIELD(bool) | |
| #undef DECLARE_GET_REPEATED_FIELD | |
| // Call this function to ensure that this message's reflection is linked into | |
| // the binary: | |
| // | |
| // google::protobuf::LinkMessageReflection<pkg::FooMessage>(); | |
| // | |
| // This will ensure that the following lookup will succeed: | |
| // | |
| // DescriptorPool::generated_pool()->FindMessageTypeByName("pkg.FooMessage"); | |
| // | |
| // As a side-effect, it will also guarantee that anything else from the same | |
| // .proto file will also be available for lookup in the generated pool. | |
| // | |
| // This function does not actually register the message, so it does not need | |
| // to be called before the lookup. However it does need to occur in a function | |
| // that cannot be stripped from the binary (ie. it must be reachable from main). | |
| // | |
| // Best practice is to call this function as close as possible to where the | |
| // reflection is actually needed. This function is very cheap to call, so you | |
| // should not need to worry about its runtime overhead except in the tightest | |
| // of loops (on x86-64 it compiles into two "mov" instructions). | |
| template <typename T> | |
| void LinkMessageReflection() { | |
| internal::StrongReferenceToType<T>(); | |
| } | |
| // Specializations to handle smart pointers to `Message`. Without these, | |
| // `google::protobuf::DynamicCastMessage(std::shared_ptr<Message>)` is ambiguous | |
| // (`const MessageLite` vs `MessageLite`). | |
| template <typename T> | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD std::shared_ptr<T> DynamicCastMessage( | |
| std::shared_ptr<Message> ptr) { | |
| if (auto* res = DynamicCastMessage<T>(ptr.get())) { | |
| // Use aliasing constructor to keep the same control block. | |
| return std::shared_ptr<T>(std::move(ptr), res); | |
| } else { | |
| return nullptr; | |
| } | |
| } | |
| template <typename T> | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD std::shared_ptr<const T> DynamicCastMessage( | |
| std::shared_ptr<const Message> ptr) { | |
| if (auto* res = DynamicCastMessage<T>(ptr.get())) { | |
| // Use aliasing constructor to keep the same control block. | |
| return std::shared_ptr<const T>(std::move(ptr), res); | |
| } else { | |
| return nullptr; | |
| } | |
| } | |
| // Overloads for `std::shared_ptr` to substitute `down_pointer_cast` | |
| template <typename T> | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD std::shared_ptr<T> DownCastMessage( | |
| std::shared_ptr<Message> ptr) { | |
| auto* res = DownCastMessage<T>(ptr.get()); | |
| // Use aliasing constructor to keep the same control block. | |
| return std::shared_ptr<T>(std::move(ptr), res); | |
| } | |
| template <typename T> | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD std::shared_ptr<const T> DownCastMessage( | |
| std::shared_ptr<const Message> ptr) { | |
| auto* res = DownCastMessage<T>(ptr.get()); | |
| // Use aliasing constructor to keep the same control block. | |
| return std::shared_ptr<const T>(std::move(ptr), res); | |
| } | |
| // ============================================================================= | |
| // Implementation details for {Get,Mutable}RawRepeatedPtrField. We provide | |
| // specializations for <std::string>, <StringPieceField> and <Message> and | |
| // handle everything else with the default template which will match any type | |
| // having a method with signature "static const google::protobuf::Descriptor* | |
| // descriptor()". Such a type presumably is a descendant of google::protobuf::Message. | |
| template <> | |
| inline const RepeatedPtrField<std::string>& | |
| Reflection::GetRepeatedPtrFieldInternal<std::string>( | |
| const Message& message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const { | |
| return *static_cast<const RepeatedPtrField<std::string>*>( | |
| GetRawRepeatedString(message, field, /*is_string=*/true, intent)); | |
| } | |
| template <> | |
| inline RepeatedPtrField<std::string>* | |
| Reflection::MutableRepeatedPtrFieldInternal<std::string>( | |
| Message* message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const { | |
| if (!field->is_extension()) { | |
| SetHasBit(message, field); | |
| } | |
| return static_cast<RepeatedPtrField<std::string>*>( | |
| MutableRawRepeatedString(message, field, /*is_string=*/true, intent)); | |
| } | |
| // ----- | |
| template <> | |
| inline const RepeatedPtrField<Message>& Reflection::GetRepeatedPtrFieldInternal( | |
| const Message& message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const { | |
| return *static_cast<const RepeatedPtrField<Message>*>( | |
| GetRawRepeatedField(message, field, FieldDescriptor::CPPTYPE_MESSAGE, -1, | |
| /*desc=*/nullptr, intent)); | |
| } | |
| template <> | |
| inline RepeatedPtrField<Message>* Reflection::MutableRepeatedPtrFieldInternal( | |
| Message* message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const { | |
| if (!field->is_extension()) { | |
| SetHasBit(message, field); | |
| } | |
| return static_cast<RepeatedPtrField<Message>*>( | |
| MutableRawRepeatedField(message, field, FieldDescriptor::CPPTYPE_MESSAGE, | |
| -1, /*desc=*/nullptr, intent)); | |
| } | |
| template <typename PB> | |
| inline const RepeatedPtrField<PB>& Reflection::GetRepeatedPtrFieldInternal( | |
| const Message& message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const { | |
| return *static_cast<const RepeatedPtrField<PB>*>( | |
| GetRawRepeatedField(message, field, FieldDescriptor::CPPTYPE_MESSAGE, -1, | |
| PB::default_instance().GetDescriptor(), intent)); | |
| } | |
| template <typename PB> | |
| inline RepeatedPtrField<PB>* Reflection::MutableRepeatedPtrFieldInternal( | |
| Message* message, const FieldDescriptor* field, | |
| GetRepeatedFieldIntent intent) const { | |
| if (!field->is_extension()) { | |
| SetHasBit(message, field); | |
| } | |
| return static_cast<RepeatedPtrField<PB>*>(MutableRawRepeatedField( | |
| message, field, FieldDescriptor::CPPTYPE_MESSAGE, -1, | |
| PB::default_instance().GetDescriptor(), intent)); | |
| } | |
| template <typename Type> | |
| const Type& Reflection::DefaultRaw(const FieldDescriptor* field) const { | |
| return *reinterpret_cast<const Type*>(schema_.GetFieldDefault(field)); | |
| } | |
| bool Reflection::HasOneofField(const Message& message, | |
| const FieldDescriptor* field) const { | |
| return (GetOneofCase(message, field->containing_oneof()) == | |
| static_cast<uint32_t>(field->number())); | |
| } | |
| const void* Reflection::GetSplitField(const Message* message) const { | |
| ABSL_DCHECK(schema_.IsSplit()); | |
| return *internal::GetConstPointerAtOffset<void*>(message, | |
| schema_.SplitOffset()); | |
| } | |
| void** Reflection::MutableSplitField(Message* message) const { | |
| ABSL_DCHECK(schema_.IsSplit()); | |
| return internal::GetPointerAtOffset<void*>(message, schema_.SplitOffset()); | |
| } | |
| namespace internal { | |
| // In some cases, (Get|Mutable)Raw may be called with a type that is different | |
| // from the final type; e.g. char. As a defensive coding to this unfortunate | |
| // practices, we should only assume extra indirection (or a lack thereof) for | |
| // the well known, complex types. | |
| template <typename T> | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD bool SplitFieldHasExtraIndirectionStatic( | |
| const FieldDescriptor* field) { | |
| if (std::is_base_of<RepeatedFieldBase, T>() || | |
| std::is_base_of<RepeatedPtrFieldBase, T>()) { | |
| ABSL_DCHECK(SplitFieldHasExtraIndirection(field)); | |
| return true; | |
| } else if (std::is_base_of<MessageLite, T>()) { | |
| ABSL_DCHECK(!SplitFieldHasExtraIndirection(field)); | |
| return false; | |
| } | |
| return SplitFieldHasExtraIndirection(field); | |
| } | |
| inline void MaybePoisonAfterClear(Message* root) { | |
| if (root == nullptr) return; | |
| if constexpr (HasMemoryPoisoning()) { | |
| const Reflection* reflection = root->GetReflection(); | |
| reflection->MaybePoisonAfterClear(*root); | |
| } else { | |
| root->Clear(); | |
| } | |
| } | |
| template <typename T> | |
| inline constexpr std::false_type IsRepeatedT{}; | |
| template <typename T> | |
| inline constexpr std::true_type IsRepeatedT<RepeatedField<T>>{}; | |
| template <typename T> | |
| inline constexpr std::true_type IsRepeatedT<RepeatedPtrField<T>>{}; | |
| template <> | |
| inline constexpr std::true_type IsRepeatedT<internal::RepeatedPtrFieldBase>{}; | |
| template <> | |
| inline constexpr std::true_type IsRepeatedT<internal::MapFieldBase>{}; | |
| template <typename T> | |
| PROTOBUF_FUTURE_ADD_EARLY_NODISCARD constexpr FieldDescriptor::CppType | |
| GetCppType() { | |
| if constexpr (IsRepeatedT<T>) { | |
| return GetCppType<typename T::value_type>(); | |
| } else { | |
| if (std::is_same_v<T, int32_t>) return FieldDescriptor::CPPTYPE_INT32; | |
| if (std::is_same_v<T, int64_t>) return FieldDescriptor::CPPTYPE_INT64; | |
| if (std::is_same_v<T, uint32_t>) return FieldDescriptor::CPPTYPE_UINT32; | |
| if (std::is_same_v<T, uint64_t>) return FieldDescriptor::CPPTYPE_UINT64; | |
| if (std::is_same_v<T, float>) return FieldDescriptor::CPPTYPE_FLOAT; | |
| if (std::is_same_v<T, double>) return FieldDescriptor::CPPTYPE_DOUBLE; | |
| if (std::is_same_v<T, bool>) return FieldDescriptor::CPPTYPE_BOOL; | |
| using CV = std::remove_cv_t<T>; | |
| using PCV = std::remove_cv_t<std::remove_pointer_t<T>>; | |
| // strings | |
| if (std::is_same_v<PCV, internal::ArenaStringPtr> || | |
| std::is_same_v<PCV, std::string> || | |
| std::is_same_v<PCV, internal::MicroString> || | |
| std::is_same_v<PCV, absl::Cord>) { | |
| return FieldDescriptor::CPPTYPE_STRING; | |
| } | |
| // messages | |
| if (std::is_same_v<PCV, Message> || // | |
| std::is_same_v<PCV, MessageLite> || // | |
| std::is_same_v<CV, internal::LazyField> || | |
| std::is_same_v<CV, internal::LazyFieldForUnion>) { | |
| return FieldDescriptor::CPPTYPE_MESSAGE; | |
| } | |
| } | |
| // Return an invalid type to make the caller fail with a nice error message in | |
| // case we missed something. | |
| return FieldDescriptor::CppType{}; | |
| } | |
| } // namespace internal | |
| template <typename T> | |
| void Reflection::VerifyFieldType(const FieldDescriptor* field) const { | |
| if constexpr (!internal::PerformDebugChecks()) { | |
| return; | |
| } | |
| if constexpr (std::is_const_v<T>) { | |
| return VerifyFieldType<std::remove_const_t<T>>(field); | |
| } | |
| // `char` and `void` are used in places where we don't know the type yet. | |
| if constexpr (std::is_same_v<T, char> || std::is_same_v<T, void>) { | |
| return; | |
| } | |
| const auto error = [&] { | |
| return absl::StrFormat("Invalid cast of %s to type %s.", field->full_name(), | |
| internal::RttiTypeName<T>().value_or("unknown")); | |
| }; | |
| ABSL_DCHECK_EQ(field->is_repeated(), internal::IsRepeatedT<T>) << error(); | |
| if constexpr (std::is_same_v<T, internal::MapFieldBase>) { | |
| ABSL_DCHECK(field->is_map()) << error(); | |
| } else if constexpr (std::is_same_v<T, internal::RepeatedPtrFieldBase>) { | |
| // It has to be string or message. | |
| ABSL_DCHECK(field->cpp_type() == field->CPPTYPE_STRING || | |
| field->cpp_type() == field->CPPTYPE_MESSAGE) | |
| << error(); | |
| } else { | |
| auto cpp_type = field->cpp_type(); | |
| // Collapse ENUM to INT32 because they are the same through reflection. | |
| if (cpp_type == field->CPPTYPE_ENUM) cpp_type = field->CPPTYPE_INT32; | |
| ABSL_DCHECK_EQ(+cpp_type, +internal::GetCppType<T>()) << error(); | |
| // Check subfield types for message. | |
| if constexpr (internal::GetCppType<T>() == | |
| FieldDescriptor::CPPTYPE_MESSAGE) { | |
| // Singular/oneof messages are by pointer, except Lazy. | |
| // Lazy uses LazyField by-value for normal fields and LazyFieldForUnion | |
| // by-value for oneof fields. | |
| if (!field->is_repeated() && !IsLazyField(field)) { | |
| ABSL_DCHECK(std::is_pointer_v<T>) << error(); | |
| } | |
| } | |
| // Check subfield types for string. | |
| if constexpr (internal::GetCppType<T>() == | |
| FieldDescriptor::CPPTYPE_STRING) { | |
| switch (field->cpp_string_type()) { | |
| case FieldDescriptor::CppStringType::kView: | |
| case FieldDescriptor::CppStringType::kString: | |
| if (IsMicroString(field)) { | |
| ABSL_DCHECK((std::is_same_v<T, internal::MicroString>)) << error(); | |
| } else { | |
| ABSL_DCHECK((std::is_same_v<T, internal::ArenaStringPtr> || | |
| std::is_same_v<T, RepeatedPtrField<std::string>>)) | |
| << error(); | |
| } | |
| break; | |
| case FieldDescriptor::CppStringType::kCord: | |
| if (field->real_containing_oneof() != nullptr) { | |
| ABSL_DCHECK((std::is_same_v<T, absl::Cord*>)) << error(); | |
| } else { | |
| ABSL_DCHECK((std::is_same_v<T, absl::Cord> || | |
| std::is_same_v<T, RepeatedField<absl::Cord>>)) | |
| << error(); | |
| } | |
| break; | |
| } | |
| } | |
| } | |
| } | |
| template <typename Type> | |
| const Type& Reflection::GetRaw(const Message& message, | |
| const FieldDescriptor* field) const { | |
| VerifyFieldType<Type>(field); | |
| const uint32_t field_offset = schema_.GetFieldOffset(field); | |
| if (ABSL_PREDICT_FALSE(schema_.IsSplit(field))) { | |
| ABSL_DCHECK(!schema_.InRealOneof(field)) | |
| << "Field = " << field->full_name(); | |
| const void* split = GetSplitField(&message); | |
| if (internal::SplitFieldHasExtraIndirectionStatic<Type>(field)) { | |
| return **internal::GetConstPointerAtOffset<Type*>(split, field_offset); | |
| } | |
| return *internal::GetConstPointerAtOffset<Type>(split, field_offset); | |
| } | |
| return internal::GetConstRefAtOffset<Type>(message, field_offset); | |
| } | |
| template <typename T> | |
| RepeatedFieldRef<T> Reflection::GetRepeatedFieldRef( | |
| const Message& message, const FieldDescriptor* field) const { | |
| ABSL_DCHECK_EQ(message.GetReflection(), this); | |
| return RepeatedFieldRef<T>(message, field); | |
| } | |
| template <typename T> | |
| MutableRepeatedFieldRef<T> Reflection::GetMutableRepeatedFieldRef( | |
| Message* message, const FieldDescriptor* field) const { | |
| ABSL_DCHECK_EQ(message->GetReflection(), this); | |
| if (!field->is_extension()) { | |
| SetHasBit(message, field); | |
| } | |
| return MutableRepeatedFieldRef<T>(message, field); | |
| } | |
| template <typename Type> | |
| Type* Reflection::MutableRaw(Message* message, | |
| const FieldDescriptor* field) const { | |
| VerifyFieldType<Type>(field); | |
| if (ABSL_PREDICT_FALSE(schema_.IsSplit(field))) { | |
| return reinterpret_cast<Type*>(MutableRawSplitImpl(message, field)); | |
| } | |
| const uint32_t field_offset = schema_.GetFieldOffset(field); | |
| return internal::GetPointerAtOffset<Type>(message, field_offset); | |
| } | |
| } // namespace protobuf | |
| } // namespace google | |
| #include "google/protobuf/port_undef.inc" | |
| #endif // GOOGLE_PROTOBUF_MESSAGE_H__ |