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   ~TrieBuilder();
26   void setImageBase(uint64_t addr) { imageBase = addr; }
27   void addSymbol(const Symbol &sym) { exported.push_back(&sym); }
28   // Returns the size in bytes of the serialized trie.
29   size_t build();
30   void writeTo(uint8_t *buf) const;
31 
32 private:
33   TrieNode *makeNode();
34   void sortAndBuild(llvm::MutableArrayRef<const Symbol *> vec, TrieNode *node,
35                     size_t lastPos, size_t pos);
36 
37   uint64_t imageBase = 0;
38   std::vector<const Symbol *> exported;
39   std::vector<TrieNode *> nodes;
40 };
41 
42 using TrieEntryCallback =
43     llvm::function_ref<void(const llvm::Twine & /*name*/, uint64_t /*flags*/)>;
44 
45 void parseTrie(const uint8_t *buf, size_t size, const TrieEntryCallback &);
46 
47 } // namespace macho
48 } // namespace lld
49 
50 #endif
51