Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //=== Registry.h - Linker-supported plugin registries -----------*- 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 | // Defines a registry template for discovering pluggable modules. |
||
| 10 | // |
||
| 11 | //===----------------------------------------------------------------------===// |
||
| 12 | |||
| 13 | #ifndef LLVM_SUPPORT_REGISTRY_H |
||
| 14 | #define LLVM_SUPPORT_REGISTRY_H |
||
| 15 | |||
| 16 | #include "llvm/ADT/STLExtras.h" |
||
| 17 | #include "llvm/ADT/StringRef.h" |
||
| 18 | #include "llvm/ADT/iterator_range.h" |
||
| 19 | #include "llvm/Support/Compiler.h" |
||
| 20 | #include "llvm/Support/DynamicLibrary.h" |
||
| 21 | #include <memory> |
||
| 22 | |||
| 23 | namespace llvm { |
||
| 24 | /// A simple registry entry which provides only a name, description, and |
||
| 25 | /// no-argument constructor. |
||
| 26 | template <typename T> |
||
| 27 | class SimpleRegistryEntry { |
||
| 28 | StringRef Name, Desc; |
||
| 29 | std::unique_ptr<T> (*Ctor)(); |
||
| 30 | |||
| 31 | public: |
||
| 32 | SimpleRegistryEntry(StringRef N, StringRef D, std::unique_ptr<T> (*C)()) |
||
| 33 | : Name(N), Desc(D), Ctor(C) {} |
||
| 34 | |||
| 35 | StringRef getName() const { return Name; } |
||
| 36 | StringRef getDesc() const { return Desc; } |
||
| 37 | std::unique_ptr<T> instantiate() const { return Ctor(); } |
||
| 38 | }; |
||
| 39 | |||
| 40 | /// A global registry used in conjunction with static constructors to make |
||
| 41 | /// pluggable components (like targets or garbage collectors) "just work" when |
||
| 42 | /// linked with an executable. |
||
| 43 | template <typename T> |
||
| 44 | class Registry { |
||
| 45 | public: |
||
| 46 | typedef T type; |
||
| 47 | typedef SimpleRegistryEntry<T> entry; |
||
| 48 | |||
| 49 | class node; |
||
| 50 | class iterator; |
||
| 51 | |||
| 52 | private: |
||
| 53 | Registry() = delete; |
||
| 54 | |||
| 55 | friend class node; |
||
| 56 | static node *Head, *Tail; |
||
| 57 | |||
| 58 | public: |
||
| 59 | /// Node in linked list of entries. |
||
| 60 | /// |
||
| 61 | class node { |
||
| 62 | friend class iterator; |
||
| 63 | friend Registry<T>; |
||
| 64 | |||
| 65 | node *Next; |
||
| 66 | const entry& Val; |
||
| 67 | |||
| 68 | public: |
||
| 69 | node(const entry &V) : Next(nullptr), Val(V) {} |
||
| 70 | }; |
||
| 71 | |||
| 72 | /// Add a node to the Registry: this is the interface between the plugin and |
||
| 73 | /// the executable. |
||
| 74 | /// |
||
| 75 | /// This function is exported by the executable and called by the plugin to |
||
| 76 | /// add a node to the executable's registry. Therefore it's not defined here |
||
| 77 | /// to avoid it being instantiated in the plugin and is instead defined in |
||
| 78 | /// the executable (see LLVM_INSTANTIATE_REGISTRY below). |
||
| 79 | static void add_node(node *N); |
||
| 80 | |||
| 81 | /// Iterators for registry entries. |
||
| 82 | /// |
||
| 83 | class iterator |
||
| 84 | : public llvm::iterator_facade_base<iterator, std::forward_iterator_tag, |
||
| 85 | const entry> { |
||
| 86 | const node *Cur; |
||
| 87 | |||
| 88 | public: |
||
| 89 | explicit iterator(const node *N) : Cur(N) {} |
||
| 90 | |||
| 91 | bool operator==(const iterator &That) const { return Cur == That.Cur; } |
||
| 92 | iterator &operator++() { Cur = Cur->Next; return *this; } |
||
| 93 | const entry &operator*() const { return Cur->Val; } |
||
| 94 | }; |
||
| 95 | |||
| 96 | // begin is not defined here in order to avoid usage of an undefined static |
||
| 97 | // data member, instead it's instantiated by LLVM_INSTANTIATE_REGISTRY. |
||
| 98 | static iterator begin(); |
||
| 99 | static iterator end() { return iterator(nullptr); } |
||
| 100 | |||
| 101 | static iterator_range<iterator> entries() { |
||
| 102 | return make_range(begin(), end()); |
||
| 103 | } |
||
| 104 | |||
| 105 | /// A static registration template. Use like such: |
||
| 106 | /// |
||
| 107 | /// Registry<Collector>::Add<FancyGC> |
||
| 108 | /// X("fancy-gc", "Newfangled garbage collector."); |
||
| 109 | /// |
||
| 110 | /// Use of this template requires that: |
||
| 111 | /// |
||
| 112 | /// 1. The registered subclass has a default constructor. |
||
| 113 | template <typename V> |
||
| 114 | class Add { |
||
| 115 | entry Entry; |
||
| 116 | node Node; |
||
| 117 | |||
| 118 | static std::unique_ptr<T> CtorFn() { return std::make_unique<V>(); } |
||
| 119 | |||
| 120 | public: |
||
| 121 | Add(StringRef Name, StringRef Desc) |
||
| 122 | : Entry(Name, Desc, CtorFn), Node(Entry) { |
||
| 123 | add_node(&Node); |
||
| 124 | } |
||
| 125 | }; |
||
| 126 | }; |
||
| 127 | } // end namespace llvm |
||
| 128 | |||
| 129 | /// Instantiate a registry class. |
||
| 130 | /// |
||
| 131 | /// This provides template definitions of add_node, begin, and the Head and Tail |
||
| 132 | /// pointers, then explicitly instantiates them. We could explicitly specialize |
||
| 133 | /// them, instead of the two-step process of define then instantiate, but |
||
| 134 | /// strictly speaking that's not allowed by the C++ standard (we would need to |
||
| 135 | /// have explicit specialization declarations in all translation units where the |
||
| 136 | /// specialization is used) so we don't. |
||
| 137 | #define LLVM_INSTANTIATE_REGISTRY(REGISTRY_CLASS) \ |
||
| 138 | namespace llvm { \ |
||
| 139 | template<typename T> typename Registry<T>::node *Registry<T>::Head = nullptr;\ |
||
| 140 | template<typename T> typename Registry<T>::node *Registry<T>::Tail = nullptr;\ |
||
| 141 | template<typename T> \ |
||
| 142 | void Registry<T>::add_node(typename Registry<T>::node *N) { \ |
||
| 143 | if (Tail) \ |
||
| 144 | Tail->Next = N; \ |
||
| 145 | else \ |
||
| 146 | Head = N; \ |
||
| 147 | Tail = N; \ |
||
| 148 | } \ |
||
| 149 | template<typename T> typename Registry<T>::iterator Registry<T>::begin() { \ |
||
| 150 | return iterator(Head); \ |
||
| 151 | } \ |
||
| 152 | template REGISTRY_CLASS::node *Registry<REGISTRY_CLASS::type>::Head; \ |
||
| 153 | template REGISTRY_CLASS::node *Registry<REGISTRY_CLASS::type>::Tail; \ |
||
| 154 | template \ |
||
| 155 | void Registry<REGISTRY_CLASS::type>::add_node(REGISTRY_CLASS::node*); \ |
||
| 156 | template REGISTRY_CLASS::iterator Registry<REGISTRY_CLASS::type>::begin(); \ |
||
| 157 | } |
||
| 158 | |||
| 159 | #endif // LLVM_SUPPORT_REGISTRY_H |