1*5ffd83dbSDimitry Andric //===- ExportTrie.h ---------------------------------------------*- C++ -*-===// 2*5ffd83dbSDimitry Andric // 3*5ffd83dbSDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*5ffd83dbSDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*5ffd83dbSDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*5ffd83dbSDimitry Andric // 7*5ffd83dbSDimitry Andric //===----------------------------------------------------------------------===// 8*5ffd83dbSDimitry Andric 9*5ffd83dbSDimitry Andric #ifndef LLD_MACHO_EXPORT_TRIE_H 10*5ffd83dbSDimitry Andric #define LLD_MACHO_EXPORT_TRIE_H 11*5ffd83dbSDimitry Andric 12*5ffd83dbSDimitry Andric #include "llvm/ADT/ArrayRef.h" 13*5ffd83dbSDimitry Andric #include "llvm/ADT/STLExtras.h" 14*5ffd83dbSDimitry Andric 15*5ffd83dbSDimitry Andric #include <vector> 16*5ffd83dbSDimitry Andric 17*5ffd83dbSDimitry Andric namespace lld { 18*5ffd83dbSDimitry Andric namespace macho { 19*5ffd83dbSDimitry Andric 20*5ffd83dbSDimitry Andric struct TrieNode; 21*5ffd83dbSDimitry Andric class Symbol; 22*5ffd83dbSDimitry Andric 23*5ffd83dbSDimitry Andric class TrieBuilder { 24*5ffd83dbSDimitry Andric public: 25*5ffd83dbSDimitry Andric void addSymbol(const Symbol &sym) { exported.push_back(&sym); } 26*5ffd83dbSDimitry Andric // Returns the size in bytes of the serialized trie. 27*5ffd83dbSDimitry Andric size_t build(); 28*5ffd83dbSDimitry Andric void writeTo(uint8_t *buf) const; 29*5ffd83dbSDimitry Andric 30*5ffd83dbSDimitry Andric private: 31*5ffd83dbSDimitry Andric TrieNode *makeNode(); 32*5ffd83dbSDimitry Andric void sortAndBuild(llvm::MutableArrayRef<const Symbol *> vec, TrieNode *node, 33*5ffd83dbSDimitry Andric size_t lastPos, size_t pos); 34*5ffd83dbSDimitry Andric 35*5ffd83dbSDimitry Andric std::vector<const Symbol *> exported; 36*5ffd83dbSDimitry Andric std::vector<TrieNode *> nodes; 37*5ffd83dbSDimitry Andric }; 38*5ffd83dbSDimitry Andric 39*5ffd83dbSDimitry Andric using TrieEntryCallback = 40*5ffd83dbSDimitry Andric llvm::function_ref<void(const llvm::Twine & /*name*/, uint64_t /*flags*/)>; 41*5ffd83dbSDimitry Andric 42*5ffd83dbSDimitry Andric void parseTrie(const uint8_t *buf, size_t size, const TrieEntryCallback &); 43*5ffd83dbSDimitry Andric 44*5ffd83dbSDimitry Andric } // namespace macho 45*5ffd83dbSDimitry Andric } // namespace lld 46*5ffd83dbSDimitry Andric 47*5ffd83dbSDimitry Andric #endif 48