Home | History | Annotate | Download | only in protobuf
      1 // Protocol Buffers - Google's data interchange format
      2 // Copyright 2008 Google Inc.  All rights reserved.
      3 // https://developers.google.com/protocol-buffers/
      4 //
      5 // Redistribution and use in source and binary forms, with or without
      6 // modification, are permitted provided that the following conditions are
      7 // met:
      8 //
      9 //     * Redistributions of source code must retain the above copyright
     10 // notice, this list of conditions and the following disclaimer.
     11 //     * Redistributions in binary form must reproduce the above
     12 // copyright notice, this list of conditions and the following disclaimer
     13 // in the documentation and/or other materials provided with the
     14 // distribution.
     15 //     * Neither the name of Google Inc. nor the names of its
     16 // contributors may be used to endorse or promote products derived from
     17 // this software without specific prior written permission.
     18 //
     19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     30 
     31 #ifndef GOOGLE_PROTOBUF_ARENA_H__
     32 #define GOOGLE_PROTOBUF_ARENA_H__
     33 
     34 #include <limits>
     35 #ifdef max
     36 #undef max  // Visual Studio defines this macro
     37 #endif
     38 #if __cplusplus >= 201103L
     39 #include <google/protobuf/stubs/type_traits.h>
     40 #endif
     41 #if defined(_MSC_VER) && !_HAS_EXCEPTIONS
     42 // Work around bugs in MSVC <typeinfo> header when _HAS_EXCEPTIONS=0.
     43 #include <exception>
     44 #include <typeinfo>
     45 namespace std {
     46 using type_info = ::type_info;
     47 }
     48 #else
     49 #include <typeinfo>
     50 #endif
     51 
     52 #include <google/protobuf/stubs/atomic_sequence_num.h>
     53 #include <google/protobuf/stubs/atomicops.h>
     54 #include <google/protobuf/stubs/common.h>
     55 #include <google/protobuf/stubs/logging.h>
     56 #include <google/protobuf/stubs/mutex.h>
     57 #include <google/protobuf/stubs/type_traits.h>
     58 
     59 
     60 namespace google {
     61 namespace protobuf {
     62 
     63 class Arena;       // defined below
     64 class Message;     // message.h
     65 
     66 namespace internal {
     67 class ArenaString; // arenastring.h
     68 class LazyField;   // lazy_field.h
     69 
     70 template<typename Type>
     71 class GenericTypeHandler; // repeated_field.h
     72 
     73 // Templated cleanup methods.
     74 template<typename T> void arena_destruct_object(void* object) {
     75   reinterpret_cast<T*>(object)->~T();
     76 }
     77 template<typename T> void arena_delete_object(void* object) {
     78   delete reinterpret_cast<T*>(object);
     79 }
     80 inline void arena_free(void* object, size_t /* size */) {
     81   free(object);
     82 }
     83 
     84 }  // namespace internal
     85 
     86 // ArenaOptions provides optional additional parameters to arena construction
     87 // that control its block-allocation behavior.
     88 struct ArenaOptions {
     89   // This defines the size of the first block requested from the system malloc.
     90   // Subsequent block sizes will increase in a geometric series up to a maximum.
     91   size_t start_block_size;
     92 
     93   // This defines the maximum block size requested from system malloc (unless an
     94   // individual arena allocation request occurs with a size larger than this
     95   // maximum). Requested block sizes increase up to this value, then remain
     96   // here.
     97   size_t max_block_size;
     98 
     99   // An initial block of memory for the arena to use, or NULL for none. If
    100   // provided, the block must live at least as long as the arena itself. The
    101   // creator of the Arena retains ownership of the block after the Arena is
    102   // destroyed.
    103   char* initial_block;
    104 
    105   // The size of the initial block, if provided.
    106   size_t initial_block_size;
    107 
    108   // A function pointer to an alloc method that returns memory blocks of size
    109   // requested. By default, it contains a ptr to the malloc function.
    110   //
    111   // NOTE: block_alloc and dealloc functions are expected to behave like
    112   // malloc and free, including Asan poisoning.
    113   void* (*block_alloc)(size_t);
    114   // A function pointer to a dealloc method that takes ownership of the blocks
    115   // from the arena. By default, it contains a ptr to a wrapper function that
    116   // calls free.
    117   void (*block_dealloc)(void*, size_t);
    118 
    119   // Hooks for adding external functionality such as user-specific metrics
    120   // collection, specific debugging abilities, etc.
    121   // Init hook may return a pointer to a cookie to be stored in the arena.
    122   // reset and destruction hooks will then be called with the same cookie
    123   // pointer. This allows us to save an external object per arena instance and
    124   // use it on the other hooks (Note: It is just as legal for init to return
    125   // NULL and not use the cookie feature).
    126   // on_arena_reset and on_arena_destruction also receive the space used in
    127   // the arena just before the reset.
    128   void* (*on_arena_init)(Arena* arena);
    129   void (*on_arena_reset)(Arena* arena, void* cookie, uint64 space_used);
    130   void (*on_arena_destruction)(Arena* arena, void* cookie, uint64 space_used);
    131 
    132   // type_info is promised to be static - its lifetime extends to
    133   // match program's lifetime (It is given by typeid operator).
    134   // Note: typeid(void) will be passed as allocated_type every time we
    135   // intentionally want to avoid monitoring an allocation. (i.e. internal
    136   // allocations for managing the arena)
    137   void (*on_arena_allocation)(const std::type_info* allocated_type,
    138       uint64 alloc_size, void* cookie);
    139 
    140   ArenaOptions()
    141       : start_block_size(kDefaultStartBlockSize),
    142         max_block_size(kDefaultMaxBlockSize),
    143         initial_block(NULL),
    144         initial_block_size(0),
    145         block_alloc(&malloc),
    146         block_dealloc(&internal::arena_free),
    147         on_arena_init(NULL),
    148         on_arena_reset(NULL),
    149         on_arena_destruction(NULL),
    150         on_arena_allocation(NULL) {}
    151 
    152  private:
    153   // Constants define default starting block size and max block size for
    154   // arena allocator behavior -- see descriptions above.
    155   static const size_t kDefaultStartBlockSize = 256;
    156   static const size_t kDefaultMaxBlockSize   = 8192;
    157 };
    158 
    159 // Support for non-RTTI environments. (The metrics hooks API uses type
    160 // information.)
    161 #ifndef GOOGLE_PROTOBUF_NO_RTTI
    162 #define RTTI_TYPE_ID(type) (&typeid(type))
    163 #else
    164 #define RTTI_TYPE_ID(type) (NULL)
    165 #endif
    166 
    167 // Arena allocator. Arena allocation replaces ordinary (heap-based) allocation
    168 // with new/delete, and improves performance by aggregating allocations into
    169 // larger blocks and freeing allocations all at once. Protocol messages are
    170 // allocated on an arena by using Arena::CreateMessage<T>(Arena*), below, and
    171 // are automatically freed when the arena is destroyed.
    172 //
    173 // This is a thread-safe implementation: multiple threads may allocate from the
    174 // arena concurrently. Destruction is not thread-safe and the destructing
    175 // thread must synchronize with users of the arena first.
    176 //
    177 // An arena provides two allocation interfaces: CreateMessage<T>, which works
    178 // for arena-enabled proto2 message types as well as other types that satisfy
    179 // the appropriate protocol (described below), and Create<T>, which works for
    180 // any arbitrary type T. CreateMessage<T> is better when the type T supports it,
    181 // because this interface (i) passes the arena pointer to the created object so
    182 // that its sub-objects and internal allocations can use the arena too, and (ii)
    183 // elides the object's destructor call when possible. Create<T> does not place
    184 // any special requirements on the type T, and will invoke the object's
    185 // destructor when the arena is destroyed.
    186 //
    187 // The arena message allocation protocol, required by CreateMessage<T>, is as
    188 // follows:
    189 //
    190 // - The type T must have (at least) two constructors: a constructor with no
    191 //   arguments, called when a T is allocated on the heap; and a constructor with
    192 //   a google::protobuf::Arena* argument, called when a T is allocated on an arena. If the
    193 //   second constructor is called with a NULL arena pointer, it must be
    194 //   equivalent to invoking the first (no-argument) constructor.
    195 //
    196 // - The type T must have a particular type trait: a nested type
    197 //   |InternalArenaConstructable_|. This is usually a typedef to |void|. If no
    198 //   such type trait exists, then the instantiation CreateMessage<T> will fail
    199 //   to compile.
    200 //
    201 // - The type T *may* have the type trait |DestructorSkippable_|. If this type
    202 //   trait is present in the type, then its destructor will not be called if and
    203 //   only if it was passed a non-NULL arena pointer. If this type trait is not
    204 //   present on the type, then its destructor is always called when the
    205 //   containing arena is destroyed.
    206 //
    207 // - One- and two-user-argument forms of CreateMessage<T>() also exist that
    208 //   forward these constructor arguments to T's constructor: for example,
    209 //   CreateMessage<T>(Arena*, arg1, arg2) forwards to a constructor T(Arena*,
    210 //   arg1, arg2).
    211 //
    212 // This protocol is implemented by all arena-enabled proto2 message classes as
    213 // well as RepeatedPtrField.
    214 
    215 #if __cplusplus >= 201103L
    216 class LIBPROTOBUF_EXPORT Arena final {
    217 #else
    218 class LIBPROTOBUF_EXPORT Arena {
    219 #endif
    220  public:
    221   // Arena constructor taking custom options. See ArenaOptions below for
    222   // descriptions of the options available.
    223   explicit Arena(const ArenaOptions& options) : options_(options) {
    224     Init();
    225   }
    226 
    227   // Default constructor with sensible default options, tuned for average
    228   // use-cases.
    229   Arena() {
    230     Init();
    231   }
    232 
    233   // Destructor deletes all owned heap allocated objects, and destructs objects
    234   // that have non-trivial destructors, except for proto2 message objects whose
    235   // destructors can be skipped. Also, frees all blocks except the initial block
    236   // if it was passed in.
    237   ~Arena();
    238 
    239   // API to create proto2 message objects on the arena. If the arena passed in
    240   // is NULL, then a heap allocated object is returned. Type T must be a message
    241   // defined in a .proto file with cc_enable_arenas set to true, otherwise a
    242   // compilation error will occur.
    243   //
    244   // RepeatedField and RepeatedPtrField may also be instantiated directly on an
    245   // arena with this method.
    246   //
    247   // This function also accepts any type T that satisfies the arena message
    248   // allocation protocol, documented above.
    249   template <typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    250   static T* CreateMessage(::google::protobuf::Arena* arena) {
    251     if (arena == NULL) {
    252       return new T;
    253     } else {
    254       return arena->CreateMessageInternal<T>(static_cast<T*>(0));
    255     }
    256   }
    257 
    258   // One-argument form of CreateMessage. This is useful for constructing objects
    259   // that implement the arena message construction protocol described above but
    260   // take additional constructor arguments.
    261   template <typename T, typename Arg> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    262   static T* CreateMessage(::google::protobuf::Arena* arena, const Arg& arg) {
    263     if (arena == NULL) {
    264       return new T(NULL, arg);
    265     } else {
    266       return arena->CreateMessageInternal<T>(static_cast<T*>(0),
    267                                              arg);
    268     }
    269   }
    270 
    271   // Two-argument form of CreateMessage. This is useful for constructing objects
    272   // that implement the arena message construction protocol described above but
    273   // take additional constructor arguments.
    274   template <typename T, typename Arg1, typename Arg2> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    275   static T* CreateMessage(::google::protobuf::Arena* arena,
    276                           const Arg1& arg1,
    277                           const Arg2& arg2) {
    278     if (arena == NULL) {
    279       return new T(NULL, arg1, arg2);
    280     } else {
    281       return arena->CreateMessageInternal<T>(static_cast<T*>(0),
    282                                              arg1, arg2);
    283     }
    284   }
    285 
    286   // API to create any objects on the arena. Note that only the object will
    287   // be created on the arena; the underlying ptrs (in case of a proto2 message)
    288   // will be still heap allocated. Proto messages should usually be allocated
    289   // with CreateMessage<T>() instead.
    290   //
    291   // Note that even if T satisfies the arena message construction protocol
    292   // (InternalArenaConstructable_ trait and optional DestructorSkippable_
    293   // trait), as described above, this function does not follow the protocol;
    294   // instead, it treats T as a black-box type, just as if it did not have these
    295   // traits. Specifically, T's constructor arguments will always be only those
    296   // passed to Create<T>() -- no additional arena pointer is implicitly added.
    297   // Furthermore, the destructor will always be called at arena destruction time
    298   // (unless the destructor is trivial). Hence, from T's point of view, it is as
    299   // if the object were allocated on the heap (except that the underlying memory
    300   // is obtained from the arena).
    301   template <typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    302   static T* Create(::google::protobuf::Arena* arena) {
    303     if (arena == NULL) {
    304       return new T();
    305     } else {
    306       return arena->CreateInternal<T>(google::protobuf::internal::has_trivial_destructor<T>::value);
    307     }
    308   }
    309 
    310   // Version of the above with one constructor argument for the created object.
    311   template <typename T, typename Arg> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    312   static T* Create(::google::protobuf::Arena* arena, const Arg& arg) {
    313     if (arena == NULL) {
    314       return new T(arg);
    315     } else {
    316       return arena->CreateInternal<T>(google::protobuf::internal::has_trivial_destructor<T>::value,
    317                                       arg);
    318     }
    319   }
    320 
    321   // Version of the above with two constructor arguments for the created object.
    322   template <typename T, typename Arg1, typename Arg2> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    323   static T* Create(::google::protobuf::Arena* arena, const Arg1& arg1, const Arg2& arg2) {
    324     if (arena == NULL) {
    325       return new T(arg1, arg2);
    326     } else {
    327       return arena->CreateInternal<T>(google::protobuf::internal::has_trivial_destructor<T>::value,
    328                                       arg1, arg2);
    329     }
    330   }
    331 
    332   // Version of the above with three constructor arguments for the created
    333   // object.
    334   template <typename T, typename Arg1, typename Arg2, typename Arg3>
    335   GOOGLE_ATTRIBUTE_ALWAYS_INLINE static T* Create(::google::protobuf::Arena* arena,
    336                                            const Arg1& arg1, const Arg2& arg2,
    337                                            const Arg3& arg3) {
    338     if (arena == NULL) {
    339       return new T(arg1, arg2, arg3);
    340     } else {
    341       return arena->CreateInternal<T>(google::protobuf::internal::has_trivial_destructor<T>::value,
    342                                       arg1, arg2, arg3);
    343     }
    344   }
    345 
    346   // Version of the above with four constructor arguments for the created
    347   // object.
    348   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    349             typename Arg4>
    350   GOOGLE_ATTRIBUTE_ALWAYS_INLINE static T* Create(::google::protobuf::Arena* arena,
    351                                            const Arg1& arg1, const Arg2& arg2,
    352                                            const Arg3& arg3, const Arg4& arg4) {
    353     if (arena == NULL) {
    354       return new T(arg1, arg2, arg3, arg4);
    355     } else {
    356       return arena->CreateInternal<T>(google::protobuf::internal::has_trivial_destructor<T>::value,
    357                                       arg1, arg2, arg3, arg4);
    358     }
    359   }
    360 
    361   // Version of the above with five constructor arguments for the created
    362   // object.
    363   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    364             typename Arg4, typename Arg5>
    365   GOOGLE_ATTRIBUTE_ALWAYS_INLINE static T* Create(::google::protobuf::Arena* arena,
    366                                            const Arg1& arg1, const Arg2& arg2,
    367                                            const Arg3& arg3, const Arg4& arg4,
    368                                            const Arg5& arg5) {
    369     if (arena == NULL) {
    370       return new T(arg1, arg2, arg3, arg4, arg5);
    371     } else {
    372       return arena->CreateInternal<T>(google::protobuf::internal::has_trivial_destructor<T>::value,
    373                                       arg1, arg2, arg3, arg4, arg5);
    374     }
    375   }
    376 
    377   // Version of the above with six constructor arguments for the created
    378   // object.
    379   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    380             typename Arg4, typename Arg5, typename Arg6>
    381   GOOGLE_ATTRIBUTE_ALWAYS_INLINE static T* Create(::google::protobuf::Arena* arena,
    382                                            const Arg1& arg1, const Arg2& arg2,
    383                                            const Arg3& arg3, const Arg4& arg4,
    384                                            const Arg5& arg5, const Arg6& arg6) {
    385     if (arena == NULL) {
    386       return new T(arg1, arg2, arg3, arg4, arg5, arg6);
    387     } else {
    388       return arena->CreateInternal<T>(google::protobuf::internal::has_trivial_destructor<T>::value,
    389                                       arg1, arg2, arg3, arg4, arg5, arg6);
    390     }
    391   }
    392 
    393   // Version of the above with seven constructor arguments for the created
    394   // object.
    395   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    396             typename Arg4, typename Arg5, typename Arg6, typename Arg7>
    397   GOOGLE_ATTRIBUTE_ALWAYS_INLINE static T* Create(::google::protobuf::Arena* arena,
    398                                            const Arg1& arg1, const Arg2& arg2,
    399                                            const Arg3& arg3, const Arg4& arg4,
    400                                            const Arg5& arg5, const Arg6& arg6,
    401                                            const Arg7& arg7) {
    402     if (arena == NULL) {
    403       return new T(arg1, arg2, arg3, arg4, arg5, arg6, arg7);
    404     } else {
    405       return arena->CreateInternal<T>(google::protobuf::internal::has_trivial_destructor<T>::value,
    406                                       arg1, arg2, arg3, arg4, arg5, arg6, arg7);
    407     }
    408   }
    409 
    410   // Version of the above with eight constructor arguments for the created
    411   // object.
    412   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    413             typename Arg4, typename Arg5, typename Arg6, typename Arg7,
    414             typename Arg8>
    415   GOOGLE_ATTRIBUTE_ALWAYS_INLINE static T* Create(::google::protobuf::Arena* arena,
    416                                            const Arg1& arg1, const Arg2& arg2,
    417                                            const Arg3& arg3, const Arg4& arg4,
    418                                            const Arg5& arg5, const Arg6& arg6,
    419                                            const Arg7& arg7, const Arg8& arg8) {
    420     if (arena == NULL) {
    421       return new T(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8);
    422     } else {
    423       return arena->CreateInternal<T>(
    424           google::protobuf::internal::has_trivial_destructor<T>::value,
    425           arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8);
    426     }
    427   }
    428 
    429   // Create an array of object type T on the arena *without* invoking the
    430   // constructor of T. If `arena` is null, then the return value should be freed
    431   // with `delete[] x;` (or `::operator delete[](x);`).
    432   // To ensure safe uses, this function checks at compile time
    433   // (when compiled as C++11) that T is trivially default-constructible and
    434   // trivially destructible.
    435   template <typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    436   static T* CreateArray(::google::protobuf::Arena* arena, size_t num_elements) {
    437     GOOGLE_CHECK_LE(num_elements,
    438              std::numeric_limits<size_t>::max() / sizeof(T))
    439         << "Requested size is too large to fit into size_t.";
    440     if (arena == NULL) {
    441       return static_cast<T*>(::operator new[](num_elements * sizeof(T)));
    442     } else {
    443       return arena->CreateInternalRawArray<T>(num_elements);
    444     }
    445   }
    446 
    447   // Returns the total space used by the arena, which is the sums of the sizes
    448   // of the underlying blocks. The total space used may not include the new
    449   // blocks that are allocated by this arena from other threads concurrently
    450   // with the call to this method.
    451   GOOGLE_ATTRIBUTE_NOINLINE uint64 SpaceAllocated() const;
    452   // As above, but does not include any free space in underlying blocks.
    453   GOOGLE_ATTRIBUTE_NOINLINE uint64 SpaceUsed() const;
    454 
    455   // Combines SpaceAllocated and SpaceUsed. Returns a pair of
    456   // <space_allocated, space_used>.
    457   GOOGLE_ATTRIBUTE_NOINLINE pair<uint64, uint64> SpaceAllocatedAndUsed() const;
    458 
    459   // Frees all storage allocated by this arena after calling destructors
    460   // registered with OwnDestructor() and freeing objects registered with Own().
    461   // Any objects allocated on this arena are unusable after this call. It also
    462   // returns the total space used by the arena which is the sums of the sizes
    463   // of the allocated blocks. This method is not thread-safe.
    464   GOOGLE_ATTRIBUTE_NOINLINE uint64 Reset();
    465 
    466   // Adds |object| to a list of heap-allocated objects to be freed with |delete|
    467   // when the arena is destroyed or reset.
    468   template <typename T> GOOGLE_ATTRIBUTE_NOINLINE
    469   void Own(T* object) {
    470     OwnInternal(object, google::protobuf::internal::is_convertible<T*, ::google::protobuf::Message*>());
    471   }
    472 
    473   // Adds |object| to a list of objects whose destructors will be manually
    474   // called when the arena is destroyed or reset. This differs from Own() in
    475   // that it does not free the underlying memory with |delete|; hence, it is
    476   // normally only used for objects that are placement-newed into
    477   // arena-allocated memory.
    478   template <typename T> GOOGLE_ATTRIBUTE_NOINLINE
    479   void OwnDestructor(T* object) {
    480     if (object != NULL) {
    481       AddListNode(object, &internal::arena_destruct_object<T>);
    482     }
    483   }
    484 
    485   // Adds a custom member function on an object to the list of destructors that
    486   // will be manually called when the arena is destroyed or reset. This differs
    487   // from OwnDestructor() in that any member function may be specified, not only
    488   // the class destructor.
    489   GOOGLE_ATTRIBUTE_NOINLINE void OwnCustomDestructor(void* object,
    490                                               void (*destruct)(void*)) {
    491     AddListNode(object, destruct);
    492   }
    493 
    494   // Retrieves the arena associated with |value| if |value| is an arena-capable
    495   // message, or NULL otherwise. This differs from value->GetArena() in that the
    496   // latter is a virtual call, while this method is a templated call that
    497   // resolves at compile-time.
    498   template<typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    499   static ::google::protobuf::Arena* GetArena(const T* value) {
    500     return GetArenaInternal(value, static_cast<T*>(0));
    501   }
    502 
    503  private:
    504   struct InternalIsArenaConstructableHelper {
    505     template<typename U>
    506     static char ArenaConstructable(
    507         const typename U::InternalArenaConstructable_*);
    508     template<typename U>
    509     static double ArenaConstructable(...);
    510   };
    511 
    512  public:
    513   // Helper typetrait that indicates support for arenas in a type T at compile
    514   // time. This is public only to allow construction of higher-level templated
    515   // utilities. is_arena_constructable<T>::value is true if the message type T
    516   // has arena support enabled, and false otherwise.
    517   //
    518   // This is inside Arena because only Arena has the friend relationships
    519   // necessary to see the underlying generated code traits.
    520   template <typename T>
    521   struct is_arena_constructable
    522       : public google::protobuf::internal::integral_constant<
    523             bool, sizeof(InternalIsArenaConstructableHelper::ArenaConstructable<
    524                          const T>(static_cast<const T*>(0))) == sizeof(char)> {
    525   };
    526 
    527  private:
    528   // Blocks are variable length malloc-ed objects.  The following structure
    529   // describes the common header for all blocks.
    530   struct Block {
    531     void* owner;   // &ThreadCache of thread that owns this block, or
    532                    // &this->owner if not yet owned by a thread.
    533     Block* next;   // Next block in arena (may have different owner)
    534     // ((char*) &block) + pos is next available byte. It is always
    535     // aligned at a multiple of 8 bytes.
    536     size_t pos;
    537     size_t size;  // total size of the block.
    538     GOOGLE_ATTRIBUTE_ALWAYS_INLINE size_t avail() const { return size - pos; }
    539     // data follows
    540   };
    541 
    542   template<typename Type> friend class ::google::protobuf::internal::GenericTypeHandler;
    543   friend class MockArena;              // For unit-testing.
    544   friend class internal::ArenaString;  // For AllocateAligned.
    545   friend class internal::LazyField;    // For CreateMaybeMessage.
    546 
    547   struct ThreadCache {
    548     // The ThreadCache is considered valid as long as this matches the
    549     // lifecycle_id of the arena being used.
    550     int64 last_lifecycle_id_seen;
    551     Block* last_block_used_;
    552   };
    553 
    554   static const size_t kHeaderSize = sizeof(Block);
    555   static google::protobuf::internal::SequenceNumber lifecycle_id_generator_;
    556 #if defined(GOOGLE_PROTOBUF_NO_THREADLOCAL)
    557   // Android ndk does not support GOOGLE_THREAD_LOCAL keyword so we use a custom thread
    558   // local storage class we implemented.
    559   // iOS also does not support the GOOGLE_THREAD_LOCAL keyword.
    560   static ThreadCache& thread_cache();
    561 #elif defined(PROTOBUF_USE_DLLS)
    562   // Thread local variables cannot be exposed through DLL interface but we can
    563   // wrap them in static functions.
    564   static ThreadCache& thread_cache();
    565 #else
    566   static GOOGLE_THREAD_LOCAL ThreadCache thread_cache_;
    567   static ThreadCache& thread_cache() { return thread_cache_; }
    568 #endif
    569 
    570   // SFINAE for skipping addition to delete list for a message type when created
    571   // with CreateMessage. This is mainly to skip proto2/proto1 message objects
    572   // with cc_enable_arenas=true from being part of the delete list. Also, note,
    573   // compiler will optimize out the branch in CreateInternal<T>.
    574   template<typename T>
    575   static inline bool SkipDeleteList(typename T::DestructorSkippable_*) {
    576     return true;
    577   }
    578 
    579   // For message objects that don't have the DestructorSkippable_ trait, we
    580   // always add to the delete list.
    581   template<typename T>
    582   static inline bool SkipDeleteList(...) {
    583     return google::protobuf::internal::has_trivial_destructor<T>::value;
    584   }
    585 
    586  private:
    587   struct InternalIsDestructorSkippableHelper {
    588     template<typename U>
    589     static char DestructorSkippable(
    590         const typename U::DestructorSkippable_*);
    591     template<typename U>
    592     static double DestructorSkippable(...);
    593   };
    594 
    595  public:
    596   // Helper typetrait that indicates whether the desctructor of type T should be
    597   // called when arena is destroyed at compile time. This is only to allow
    598   // construction of higher-level templated utilities.
    599   // is_destructor_skippable<T>::value is true if the destructor of the message
    600   // type T should not be called when arena is destroyed or false otherwise.
    601   // This is inside Arena because only Arena has the friend relationships
    602   // necessary to see the underlying generated code traits.
    603   template<typename T>
    604   struct is_destructor_skippable
    605       : public google::protobuf::internal::integral_constant<
    606             bool,
    607             sizeof(InternalIsDestructorSkippableHelper::DestructorSkippable<
    608                    const T>(static_cast<const T*>(0))) == sizeof(char) ||
    609                 google::protobuf::internal::has_trivial_destructor<T>::value> {};
    610 
    611   // CreateMessage<T> requires that T supports arenas, but this private method
    612   // works whether or not T supports arenas. These are not exposed to user code
    613   // as it can cause confusing API usages, and end up having double free in
    614   // user code. These are used only internally from LazyField and Repeated
    615   // fields, since they are designed to work in all mode combinations.
    616   template<typename Msg> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    617   static Msg* CreateMaybeMessage(
    618       Arena* arena, typename Msg::InternalArenaConstructable_*) {
    619     return CreateMessage<Msg>(arena);
    620   }
    621 
    622   template<typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    623   static T* CreateMaybeMessage(Arena* arena, ...) {
    624     return Create<T>(arena);
    625   }
    626 
    627   // Just allocate the required size for the given type assuming the
    628   // type has a trivial constructor.
    629   template<typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    630   T* CreateInternalRawArray(size_t num_elements) {
    631     GOOGLE_CHECK_LE(num_elements,
    632              std::numeric_limits<size_t>::max() / sizeof(T))
    633         << "Requested size is too large to fit into size_t.";
    634     return static_cast<T*>(
    635         AllocateAligned(RTTI_TYPE_ID(T), sizeof(T) * num_elements));
    636   }
    637 
    638   template <typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    639   T* CreateInternal(bool skip_explicit_ownership) {
    640     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T))) T();
    641     if (!skip_explicit_ownership) {
    642       AddListNode(t, &internal::arena_destruct_object<T>);
    643     }
    644     return t;
    645   }
    646 
    647   template <typename T, typename Arg> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    648   T* CreateInternal(bool skip_explicit_ownership, const Arg& arg) {
    649     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T))) T(arg);
    650     if (!skip_explicit_ownership) {
    651       AddListNode(t, &internal::arena_destruct_object<T>);
    652     }
    653     return t;
    654   }
    655 
    656   template <typename T, typename Arg1, typename Arg2> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    657   T* CreateInternal(
    658       bool skip_explicit_ownership, const Arg1& arg1, const Arg2& arg2) {
    659     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T))) T(arg1, arg2);
    660     if (!skip_explicit_ownership) {
    661       AddListNode(t, &internal::arena_destruct_object<T>);
    662     }
    663     return t;
    664   }
    665 
    666   template <typename T, typename Arg1, typename Arg2, typename Arg3>
    667   GOOGLE_ATTRIBUTE_ALWAYS_INLINE T* CreateInternal(bool skip_explicit_ownership,
    668                                             const Arg1& arg1,
    669                                             const Arg2& arg2,
    670                                             const Arg3& arg3) {
    671     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T)))
    672         T(arg1, arg2, arg3);
    673     if (!skip_explicit_ownership) {
    674       AddListNode(t, &internal::arena_destruct_object<T>);
    675     }
    676     return t;
    677   }
    678 
    679   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    680             typename Arg4>
    681   GOOGLE_ATTRIBUTE_ALWAYS_INLINE T* CreateInternal(bool skip_explicit_ownership,
    682                                             const Arg1& arg1,
    683                                             const Arg2& arg2,
    684                                             const Arg3& arg3,
    685                                             const Arg4& arg4) {
    686     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T)))
    687         T(arg1, arg2, arg3, arg4);
    688     if (!skip_explicit_ownership) {
    689       AddListNode(t, &internal::arena_destruct_object<T>);
    690     }
    691     return t;
    692   }
    693 
    694   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    695             typename Arg4, typename Arg5>
    696   GOOGLE_ATTRIBUTE_ALWAYS_INLINE T* CreateInternal(bool skip_explicit_ownership,
    697                                             const Arg1& arg1,
    698                                             const Arg2& arg2,
    699                                             const Arg3& arg3,
    700                                             const Arg4& arg4,
    701                                             const Arg5& arg5) {
    702     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T)))
    703         T(arg1, arg2, arg3, arg4, arg5);
    704     if (!skip_explicit_ownership) {
    705       AddListNode(t, &internal::arena_destruct_object<T>);
    706     }
    707     return t;
    708   }
    709 
    710   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    711             typename Arg4, typename Arg5, typename Arg6>
    712   GOOGLE_ATTRIBUTE_ALWAYS_INLINE T* CreateInternal(bool skip_explicit_ownership,
    713                                             const Arg1& arg1,
    714                                             const Arg2& arg2,
    715                                             const Arg3& arg3,
    716                                             const Arg4& arg4,
    717                                             const Arg5& arg5,
    718                                             const Arg6& arg6) {
    719     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T)))
    720         T(arg1, arg2, arg3, arg4, arg5, arg6);
    721     if (!skip_explicit_ownership) {
    722       AddListNode(t, &internal::arena_destruct_object<T>);
    723     }
    724     return t;
    725   }
    726 
    727   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    728             typename Arg4, typename Arg5, typename Arg6, typename Arg7>
    729   GOOGLE_ATTRIBUTE_ALWAYS_INLINE T* CreateInternal(bool skip_explicit_ownership,
    730                                             const Arg1& arg1,
    731                                             const Arg2& arg2,
    732                                             const Arg3& arg3,
    733                                             const Arg4& arg4,
    734                                             const Arg5& arg5,
    735                                             const Arg6& arg6,
    736                                             const Arg7& arg7) {
    737     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T)))
    738         T(arg1, arg2, arg3, arg4, arg5, arg6, arg7);
    739     if (!skip_explicit_ownership) {
    740       AddListNode(t, &internal::arena_destruct_object<T>);
    741     }
    742     return t;
    743   }
    744 
    745   template <typename T, typename Arg1, typename Arg2, typename Arg3,
    746             typename Arg4, typename Arg5, typename Arg6, typename Arg7,
    747             typename Arg8>
    748   GOOGLE_ATTRIBUTE_ALWAYS_INLINE T* CreateInternal(bool skip_explicit_ownership,
    749                                             const Arg1& arg1,
    750                                             const Arg2& arg2,
    751                                             const Arg3& arg3,
    752                                             const Arg4& arg4,
    753                                             const Arg5& arg5,
    754                                             const Arg6& arg6,
    755                                             const Arg7& arg7,
    756                                             const Arg8& arg8) {
    757     T* t = new (AllocateAligned(RTTI_TYPE_ID(T), sizeof(T)))
    758         T(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8);
    759     if (!skip_explicit_ownership) {
    760       AddListNode(t, &internal::arena_destruct_object<T>);
    761     }
    762     return t;
    763   }
    764 
    765   template <typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    766   T* CreateMessageInternal(typename T::InternalArenaConstructable_*) {
    767     return CreateInternal<T, Arena*>(SkipDeleteList<T>(static_cast<T*>(0)),
    768                                      this);
    769   }
    770 
    771   template <typename T, typename Arg> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    772   T* CreateMessageInternal(typename T::InternalArenaConstructable_*,
    773                            const Arg& arg) {
    774     return CreateInternal<T, Arena*>(SkipDeleteList<T>(static_cast<T*>(0)),
    775                                      this, arg);
    776   }
    777 
    778   template <typename T, typename Arg1, typename Arg2> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    779   T* CreateMessageInternal(typename T::InternalArenaConstructable_*,
    780                            const Arg1& arg1, const Arg2& arg2) {
    781     return CreateInternal<T, Arena*>(SkipDeleteList<T>(static_cast<T*>(0)),
    782                                      this, arg1, arg2);
    783   }
    784 
    785   // CreateInArenaStorage is used to implement map field. Without it,
    786   // google::protobuf::Map need to call generated message's protected arena constructor,
    787   // which needs to declare google::protobuf::Map as friend of generated message.
    788   template <typename T>
    789   static void CreateInArenaStorage(T* ptr, Arena* arena) {
    790     CreateInArenaStorageInternal(ptr, arena,
    791                                  typename is_arena_constructable<T>::type());
    792     RegisterDestructorInternal(ptr, arena,
    793                                typename is_destructor_skippable<T>::type());
    794   }
    795 
    796   template <typename T>
    797   static void CreateInArenaStorageInternal(
    798       T* ptr, Arena* arena, google::protobuf::internal::true_type) {
    799     new (ptr) T(arena);
    800   }
    801   template <typename T>
    802   static void CreateInArenaStorageInternal(
    803       T* ptr, Arena* arena, google::protobuf::internal::false_type) {
    804     new (ptr) T;
    805   }
    806 
    807   template <typename T>
    808   static void RegisterDestructorInternal(
    809       T* ptr, Arena* arena, google::protobuf::internal::true_type) {}
    810   template <typename T>
    811   static void RegisterDestructorInternal(
    812       T* ptr, Arena* arena, google::protobuf::internal::false_type) {
    813     arena->OwnDestructor(ptr);
    814   }
    815 
    816   // These implement Own(), which registers an object for deletion (destructor
    817   // call and operator delete()). The second parameter has type 'true_type' if T
    818   // is a subtype of ::google::protobuf::Message and 'false_type' otherwise. Collapsing
    819   // all template instantiations to one for generic Message reduces code size,
    820   // using the virtual destructor instead.
    821   template<typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    822   void OwnInternal(T* object, google::protobuf::internal::true_type) {
    823     if (object != NULL) {
    824       AddListNode(object, &internal::arena_delete_object< ::google::protobuf::Message >);
    825     }
    826   }
    827   template<typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    828   void OwnInternal(T* object, google::protobuf::internal::false_type) {
    829     if (object != NULL) {
    830       AddListNode(object, &internal::arena_delete_object<T>);
    831     }
    832   }
    833 
    834   // Implementation for GetArena(). Only message objects with
    835   // InternalArenaConstructable_ tags can be associated with an arena, and such
    836   // objects must implement a GetArenaNoVirtual() method.
    837   template<typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    838   static ::google::protobuf::Arena* GetArenaInternal(
    839       const T* value, typename T::InternalArenaConstructable_*) {
    840     return value->GetArenaNoVirtual();
    841   }
    842 
    843   template<typename T> GOOGLE_ATTRIBUTE_ALWAYS_INLINE
    844   static ::google::protobuf::Arena* GetArenaInternal(const T* value, ...) {
    845     return NULL;
    846   }
    847 
    848   // Allocate and also optionally call on_arena_allocation callback with the
    849   // allocated type info when the hooks are in place in ArenaOptions and
    850   // the cookie is not null.
    851   void* AllocateAligned(const std::type_info* allocated, size_t n);
    852 
    853   // Allocate an internal allocation, avoiding optional typed monitoring.
    854   GOOGLE_ATTRIBUTE_ALWAYS_INLINE void* AllocateAligned(size_t n) {
    855     return AllocateAligned(NULL, n);
    856   }
    857 
    858   void Init();
    859 
    860   // Free all blocks and return the total space used which is the sums of sizes
    861   // of the all the allocated blocks.
    862   uint64 FreeBlocks();
    863 
    864   // Add object pointer and cleanup function pointer to the list.
    865   // TODO(rohananil, cfallin): We could pass in a sub-arena into this method
    866   // to avoid polluting blocks of this arena with list nodes. This would help in
    867   // mixed mode (where many protobufs have cc_enable_arenas=false), and is an
    868   // alternative to a chunked linked-list, but with extra overhead of *next.
    869   void AddListNode(void* elem, void (*cleanup)(void*));
    870   // Delete or Destruct all objects owned by the arena.
    871   void CleanupList();
    872   uint64 ResetInternal();
    873 
    874   inline void SetThreadCacheBlock(Block* block) {
    875     thread_cache().last_block_used_ = block;
    876     thread_cache().last_lifecycle_id_seen = lifecycle_id_;
    877   }
    878 
    879   int64 lifecycle_id_;  // Unique for each arena. Changes on Reset().
    880 
    881   google::protobuf::internal::AtomicWord blocks_;  // Head of linked list of all allocated blocks
    882   google::protobuf::internal::AtomicWord hint_;    // Fast thread-local block access
    883 
    884   // Node contains the ptr of the object to be cleaned up and the associated
    885   // cleanup function ptr.
    886   struct Node {
    887     void* elem;              // Pointer to the object to be cleaned up.
    888     void (*cleanup)(void*);  // Function pointer to the destructor or deleter.
    889     Node* next;              // Next node in the list.
    890   };
    891 
    892   google::protobuf::internal::AtomicWord cleanup_list_;  // Head of a linked list of nodes containing object
    893                              // ptrs and cleanup methods.
    894 
    895   bool owns_first_block_;    // Indicates that arena owns the first block
    896   Mutex blocks_lock_;
    897 
    898   void AddBlock(Block* b);
    899   // Access must be synchronized, either by blocks_lock_ or by being called from
    900   // Init()/Reset().
    901   void AddBlockInternal(Block* b);
    902   void* SlowAlloc(size_t n);
    903   Block* FindBlock(void* me);
    904   Block* NewBlock(void* me, Block* my_last_block, size_t n,
    905                   size_t start_block_size, size_t max_block_size);
    906   static void* AllocFromBlock(Block* b, size_t n);
    907   template <typename Key, typename T>
    908   friend class Map;
    909 
    910   // The arena may save a cookie it receives from the external on_init hook
    911   // and then use it when calling the on_reset and on_destruction hooks.
    912   void* hooks_cookie_;
    913 
    914   ArenaOptions options_;
    915 
    916   GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(Arena);
    917 };
    918 
    919 // Defined above for supporting environments without RTTI.
    920 #undef RTTI_TYPE_ID
    921 
    922 }  // namespace protobuf
    923 
    924 }  // namespace google
    925 #endif  // GOOGLE_PROTOBUF_ARENA_H__
    926