lld/MachO/ExportTrie.h

   1 //===- ExportTrie.h ---------------------------------------------*- 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 #ifndef LLD_MACHO_EXPORT_TRIE_H
  10 #define LLD_MACHO_EXPORT_TRIE_H
  11
  12 #include "llvm/ADT/ArrayRef.h"
  13 #include "llvm/ADT/STLExtras.h"
  14
  15 #include <vector>
  16
  17 namespace lld {
  18 namespace macho {
  19
  20 struct TrieNode;
  21 class Symbol;
  22
  23 class TrieBuilder {
  24 public:
  25   void setImageBase(uint64_t addr) { imageBase = addr; }
  26   void addSymbol(const Symbol &sym) { exported.push_back(&sym); }
  27   // Returns the size in bytes of the serialized trie.
  28   size_t build();
  29   void writeTo(uint8_t *buf) const;
  30
  31 private:
  32   TrieNode *makeNode();
  33   void sortAndBuild(llvm::MutableArrayRef<const Symbol *> vec, TrieNode *node,
  34                     size_t lastPos, size_t pos);
  35
  36   uint64_t imageBase = 0;
  37   std::vector<const Symbol *> exported;
  38   std::vector<TrieNode *> nodes;
  39 };
  40
  41 using TrieEntryCallback =
  42     llvm::function_ref<void(const llvm::Twine & /*name*/, uint64_t /*flags*/)>;
  43
  44 void parseTrie(const uint8_t *buf, size_t size, const TrieEntryCallback &);
  45
  46 } // namespace macho
  47 } // namespace lld
  48
  49 #endif