| 1 | //===-- llvm/Support/ManagedStatic.h - Static Global wrapper ----*- C++ -*-===// | 
| 2 | // | 
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | 
| 4 | // See https://llvm.org/LICENSE.txt for license information. | 
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | 
| 6 | // | 
| 7 | //===----------------------------------------------------------------------===// | 
| 8 | // | 
| 9 | // This file defines the ManagedStatic class and the llvm_shutdown() function. | 
| 10 | // | 
| 11 | //===----------------------------------------------------------------------===// | 
| 12 |  | 
| 13 | #ifndef LLVM_SUPPORT_MANAGEDSTATIC_H | 
| 14 | #define LLVM_SUPPORT_MANAGEDSTATIC_H | 
| 15 |  | 
| 16 | #include <atomic> | 
| 17 | #include <cstddef> | 
| 18 |  | 
| 19 | namespace llvm { | 
| 20 |  | 
| 21 | /// object_creator - Helper method for ManagedStatic. | 
| 22 | template <class C> struct object_creator { | 
| 23 |   static void *call() { return new C(); } | 
| 24 | }; | 
| 25 |  | 
| 26 | /// object_deleter - Helper method for ManagedStatic. | 
| 27 | /// | 
| 28 | template <typename T> struct object_deleter { | 
| 29 |   static void call(void *Ptr) { delete (T *)Ptr; } | 
| 30 | }; | 
| 31 | template <typename T, size_t N> struct object_deleter<T[N]> { | 
| 32 |   static void call(void *Ptr) { delete[](T *)Ptr; } | 
| 33 | }; | 
| 34 |  | 
| 35 | // ManagedStatic must be initialized to zero, and it must *not* have a dynamic | 
| 36 | // initializer because managed statics are often created while running other | 
| 37 | // dynamic initializers. In standard C++11, the best way to accomplish this is | 
| 38 | // with a constexpr default constructor. However, different versions of the | 
| 39 | // Visual C++ compiler have had bugs where, even though the constructor may be | 
| 40 | // constexpr, a dynamic initializer may be emitted depending on optimization | 
| 41 | // settings. For the affected versions of MSVC, use the old linker | 
| 42 | // initialization pattern of not providing a constructor and leaving the fields | 
| 43 | // uninitialized. See http://llvm.org/PR41367 for details. | 
| 44 | #if !defined(_MSC_VER) || (_MSC_VER >= 1925) || defined(__clang__) | 
| 45 | #define LLVM_USE_CONSTEXPR_CTOR | 
| 46 | #endif | 
| 47 |  | 
| 48 | /// ManagedStaticBase - Common base class for ManagedStatic instances. | 
| 49 | class ManagedStaticBase { | 
| 50 | protected: | 
| 51 | #ifdef LLVM_USE_CONSTEXPR_CTOR | 
| 52 |   mutable std::atomic<void *> Ptr{}; | 
| 53 |   mutable void (*DeleterFn)(void *) = nullptr; | 
| 54 |   mutable const ManagedStaticBase *Next = nullptr; | 
| 55 | #else | 
| 56 |   // This should only be used as a static variable, which guarantees that this | 
| 57 |   // will be zero initialized. | 
| 58 |   mutable std::atomic<void *> Ptr; | 
| 59 |   mutable void (*DeleterFn)(void *); | 
| 60 |   mutable const ManagedStaticBase *Next; | 
| 61 | #endif | 
| 62 |  | 
| 63 |   void RegisterManagedStatic(void *(*creator)(), void (*deleter)(void*)) const; | 
| 64 |  | 
| 65 | public: | 
| 66 | #ifdef LLVM_USE_CONSTEXPR_CTOR | 
| 67 |   constexpr ManagedStaticBase() = default; | 
| 68 | #endif | 
| 69 |  | 
| 70 |   /// isConstructed - Return true if this object has not been created yet. | 
| 71 |   bool isConstructed() const { return Ptr != nullptr; } | 
| 72 |  | 
| 73 |   void destroy() const; | 
| 74 | }; | 
| 75 |  | 
| 76 | /// ManagedStatic - This transparently changes the behavior of global statics to | 
| 77 | /// be lazily constructed on demand (good for reducing startup times of dynamic | 
| 78 | /// libraries that link in LLVM components) and for making destruction be | 
| 79 | /// explicit through the llvm_shutdown() function call. | 
| 80 | /// | 
| 81 | template <class C, class Creator = object_creator<C>, | 
| 82 |           class Deleter = object_deleter<C>> | 
| 83 | class ManagedStatic : public ManagedStaticBase { | 
| 84 | public: | 
| 85 |   // Accessors. | 
| 86 |   C &operator*() { | 
| 87 |     void *Tmp = Ptr.load(std::memory_order_acquire); | 
| 88 |     if (!Tmp) | 
| 89 |       RegisterManagedStatic(creator: Creator::call, deleter: Deleter::call); | 
| 90 |  | 
| 91 |     return *static_cast<C *>(Ptr.load(std::memory_order_relaxed)); | 
| 92 |   } | 
| 93 |  | 
| 94 |   C *operator->() { return &**this; } | 
| 95 |  | 
| 96 |   const C &operator*() const { | 
| 97 |     void *Tmp = Ptr.load(std::memory_order_acquire); | 
| 98 |     if (!Tmp) | 
| 99 |       RegisterManagedStatic(creator: Creator::call, deleter: Deleter::call); | 
| 100 |  | 
| 101 |     return *static_cast<C *>(Ptr.load(std::memory_order_relaxed)); | 
| 102 |   } | 
| 103 |  | 
| 104 |   const C *operator->() const { return &**this; } | 
| 105 |  | 
| 106 |   // Extract the instance, leaving the ManagedStatic uninitialized. The | 
| 107 |   // user is then responsible for the lifetime of the returned instance. | 
| 108 |   C *claim() { | 
| 109 |     return static_cast<C *>(Ptr.exchange(nullptr)); | 
| 110 |   } | 
| 111 | }; | 
| 112 |  | 
| 113 | /// llvm_shutdown - Deallocate and destroy all ManagedStatic variables. | 
| 114 | void llvm_shutdown(); | 
| 115 |  | 
| 116 | /// llvm_shutdown_obj - This is a simple helper class that calls | 
| 117 | /// llvm_shutdown() when it is destroyed. | 
| 118 | struct llvm_shutdown_obj { | 
| 119 |   llvm_shutdown_obj() = default; | 
| 120 |   ~llvm_shutdown_obj() { llvm_shutdown(); } | 
| 121 | }; | 
| 122 |  | 
| 123 | } // end namespace llvm | 
| 124 |  | 
| 125 | #endif // LLVM_SUPPORT_MANAGEDSTATIC_H | 
| 126 |  |