1 //===- NamePool.cpp -------------------------------------------------------===//
2 //
3 // The MCLinker Project
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 #include <llvm/Support/raw_ostream.h>
10 #include <mcld/LD/NamePool.h>
11 #include <mcld/LD/StaticResolver.h>
12
13 using namespace mcld;
14
15 //===----------------------------------------------------------------------===//
16 // NamePool
17 //===----------------------------------------------------------------------===//
NamePool(NamePool::size_type pSize)18 NamePool::NamePool(NamePool::size_type pSize)
19 : m_pResolver(new StaticResolver()), m_Table(pSize) {
20 }
21
~NamePool()22 NamePool::~NamePool()
23 {
24 delete m_pResolver;
25
26 FreeInfoSet::iterator info, iEnd = m_FreeInfoSet.end();
27 for (info = m_FreeInfoSet.begin(); info != iEnd; ++info) {
28 ResolveInfo::Destroy(*info);
29 }
30 }
31
32 /// createSymbol - create a symbol
createSymbol(const llvm::StringRef & pName,bool pIsDyn,ResolveInfo::Type pType,ResolveInfo::Desc pDesc,ResolveInfo::Binding pBinding,ResolveInfo::SizeType pSize,ResolveInfo::Visibility pVisibility)33 ResolveInfo* NamePool::createSymbol(const llvm::StringRef& pName,
34 bool pIsDyn,
35 ResolveInfo::Type pType,
36 ResolveInfo::Desc pDesc,
37 ResolveInfo::Binding pBinding,
38 ResolveInfo::SizeType pSize,
39 ResolveInfo::Visibility pVisibility)
40 {
41 ResolveInfo** result = m_FreeInfoSet.allocate();
42 (*result) = ResolveInfo::Create(pName);
43 (*result)->setIsSymbol(true);
44 (*result)->setSource(pIsDyn);
45 (*result)->setType(pType);
46 (*result)->setDesc(pDesc);
47 (*result)->setBinding(pBinding);
48 (*result)->setVisibility(pVisibility);
49 (*result)->setSize(pSize);
50 return *result;
51 }
52
53 /// insertSymbol - insert a symbol and resolve it immediately
54 /// @return the pointer of resolved ResolveInfo
55 /// @return is the symbol existent?
insertSymbol(const llvm::StringRef & pName,bool pIsDyn,ResolveInfo::Type pType,ResolveInfo::Desc pDesc,ResolveInfo::Binding pBinding,ResolveInfo::SizeType pSize,LDSymbol::ValueType pValue,ResolveInfo::Visibility pVisibility,ResolveInfo * pOldInfo,Resolver::Result & pResult)56 void NamePool::insertSymbol(const llvm::StringRef& pName,
57 bool pIsDyn,
58 ResolveInfo::Type pType,
59 ResolveInfo::Desc pDesc,
60 ResolveInfo::Binding pBinding,
61 ResolveInfo::SizeType pSize,
62 LDSymbol::ValueType pValue,
63 ResolveInfo::Visibility pVisibility,
64 ResolveInfo* pOldInfo,
65 Resolver::Result& pResult)
66 {
67 // We should check if there is any symbol with the same name existed.
68 // If it already exists, we should use resolver to decide which symbol
69 // should be reserved. Otherwise, we insert the symbol and set up its
70 // attributes.
71 bool exist = false;
72 ResolveInfo* old_symbol = m_Table.insert(pName, exist);
73 ResolveInfo* new_symbol = NULL;
74 if (exist && old_symbol->isSymbol()) {
75 new_symbol = m_Table.getEntryFactory().produce(pName);
76 }
77 else {
78 exist = false;
79 new_symbol = old_symbol;
80 }
81
82 new_symbol->setIsSymbol(true);
83 new_symbol->setSource(pIsDyn);
84 new_symbol->setType(pType);
85 new_symbol->setDesc(pDesc);
86 new_symbol->setBinding(pBinding);
87 new_symbol->setVisibility(pVisibility);
88 new_symbol->setSize(pSize);
89
90 if (!exist) {
91 // old_symbol is neither existed nor a symbol.
92 pResult.info = new_symbol;
93 pResult.existent = false;
94 pResult.overriden = true;
95 return;
96 }
97 else if (NULL != pOldInfo) {
98 // existent, remember its attribute
99 pOldInfo->override(*old_symbol);
100 }
101
102 // exist and is a symbol
103 // symbol resolution
104 bool override = false;
105 unsigned int action = Resolver::LastAction;
106 if (m_pResolver->resolve(*old_symbol, *new_symbol, override, pValue)) {
107 pResult.info = old_symbol;
108 pResult.existent = true;
109 pResult.overriden = override;
110 }
111 else {
112 m_pResolver->resolveAgain(*this, action, *old_symbol, *new_symbol, pResult);
113 }
114
115 m_Table.getEntryFactory().destroy(new_symbol);
116 return;
117 }
118
insertString(const llvm::StringRef & pString)119 llvm::StringRef NamePool::insertString(const llvm::StringRef& pString)
120 {
121 bool exist = false;
122 ResolveInfo* resolve_info = m_Table.insert(pString, exist);
123 return llvm::StringRef(resolve_info->name(), resolve_info->nameSize());
124 }
125
reserve(NamePool::size_type pSize)126 void NamePool::reserve(NamePool::size_type pSize)
127 {
128 m_Table.rehash(pSize);
129 }
130
capacity() const131 NamePool::size_type NamePool::capacity() const
132 {
133 return (m_Table.numOfBuckets() - m_Table.numOfEntries());
134 }
135
136 /// findInfo - find the resolved ResolveInfo
findInfo(const llvm::StringRef & pName)137 ResolveInfo* NamePool::findInfo(const llvm::StringRef& pName)
138 {
139 Table::iterator iter = m_Table.find(pName);
140 return iter.getEntry();
141 }
142
143 /// findInfo - find the resolved ResolveInfo
findInfo(const llvm::StringRef & pName) const144 const ResolveInfo* NamePool::findInfo(const llvm::StringRef& pName) const
145 {
146 Table::const_iterator iter = m_Table.find(pName);
147 return iter.getEntry();
148 }
149
150 /// findSymbol - find the resolved output LDSymbol
findSymbol(const llvm::StringRef & pName)151 LDSymbol* NamePool::findSymbol(const llvm::StringRef& pName)
152 {
153 ResolveInfo* info = findInfo(pName);
154 if (NULL == info)
155 return NULL;
156 return info->outSymbol();
157 }
158
159 /// findSymbol - find the resolved output LDSymbol
findSymbol(const llvm::StringRef & pName) const160 const LDSymbol* NamePool::findSymbol(const llvm::StringRef& pName) const
161 {
162 const ResolveInfo* info = findInfo(pName);
163 if (NULL == info)
164 return NULL;
165 return info->outSymbol();
166 }
167
168