1affc150dc44fab1911775a49636d0ce85333b634Zonr Chang//===- NamePool.cpp -------------------------------------------------------===// 25460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 35460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// The MCLinker Project 45460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 55460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// This file is distributed under the University of Illinois Open Source 65460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// License. See LICENSE.TXT for details. 75460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 85460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//===----------------------------------------------------------------------===// 95460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <llvm/Support/raw_ostream.h> 10affc150dc44fab1911775a49636d0ce85333b634Zonr Chang#include <mcld/LD/NamePool.h> 1122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao#include <mcld/LD/StaticResolver.h> 125460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 135460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaousing namespace mcld; 145460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1522add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao//===----------------------------------------------------------------------===// 16affc150dc44fab1911775a49636d0ce85333b634Zonr Chang// NamePool 1722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao//===----------------------------------------------------------------------===// 1822add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei LiaoNamePool::NamePool(NamePool::size_type pSize) 1922add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao : m_pResolver(new StaticResolver()), m_Table(pSize) { 205460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 215460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 22affc150dc44fab1911775a49636d0ce85333b634Zonr ChangNamePool::~NamePool() 235460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 2422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao delete m_pResolver; 256f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines 266f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines FreeInfoSet::iterator info, iEnd = m_FreeInfoSet.end(); 276f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines for (info = m_FreeInfoSet.begin(); info != iEnd; ++info) { 286f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines ResolveInfo::Destroy(*info); 296f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines } 305460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 315460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 325460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/// createSymbol - create a symbol 33affc150dc44fab1911775a49636d0ce85333b634Zonr ChangResolveInfo* NamePool::createSymbol(const llvm::StringRef& pName, 3422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao bool pIsDyn, 3522add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao ResolveInfo::Type pType, 3622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao ResolveInfo::Desc pDesc, 3722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao ResolveInfo::Binding pBinding, 3822add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao ResolveInfo::SizeType pSize, 3922add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao ResolveInfo::Visibility pVisibility) 405460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 416f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines ResolveInfo** result = m_FreeInfoSet.allocate(); 426f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines (*result) = ResolveInfo::Create(pName); 436f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines (*result)->setIsSymbol(true); 446f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines (*result)->setSource(pIsDyn); 456f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines (*result)->setType(pType); 466f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines (*result)->setDesc(pDesc); 476f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines (*result)->setBinding(pBinding); 486f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines (*result)->setVisibility(pVisibility); 496f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines (*result)->setSize(pSize); 506f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines return *result; 515460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 525460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 535460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/// insertSymbol - insert a symbol and resolve it immediately 545460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/// @return the pointer of resolved ResolveInfo 555460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/// @return is the symbol existent? 56affc150dc44fab1911775a49636d0ce85333b634Zonr Changvoid NamePool::insertSymbol(const llvm::StringRef& pName, 575460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao bool pIsDyn, 585460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo::Type pType, 595460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo::Desc pDesc, 605460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo::Binding pBinding, 615460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo::SizeType pSize, 62f33f6de54db174aa679a4b6d1e040d37e95541c0Stephen Hines LDSymbol::ValueType pValue, 635460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo::Visibility pVisibility, 645460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo* pOldInfo, 655460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao Resolver::Result& pResult) 665460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 675460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // We should check if there is any symbol with the same name existed. 685460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // If it already exists, we should use resolver to decide which symbol 695460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // should be reserved. Otherwise, we insert the symbol and set up its 705460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // attributes. 715460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao bool exist = false; 725460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo* old_symbol = m_Table.insert(pName, exist); 735460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo* new_symbol = NULL; 745460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (exist && old_symbol->isSymbol()) { 755460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol = m_Table.getEntryFactory().produce(pName); 765460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 775460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao else { 785460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao exist = false; 795460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol = old_symbol; 805460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 815460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 825460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol->setIsSymbol(true); 835460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol->setSource(pIsDyn); 845460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol->setType(pType); 855460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol->setDesc(pDesc); 865460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol->setBinding(pBinding); 875460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol->setVisibility(pVisibility); 885460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao new_symbol->setSize(pSize); 895460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 905460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (!exist) { 916f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines // old_symbol is neither existed nor a symbol. 925460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pResult.info = new_symbol; 935460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pResult.existent = false; 945460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pResult.overriden = true; 955460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return; 965460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 975460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao else if (NULL != pOldInfo) { 985460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // existent, remember its attribute 995460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pOldInfo->override(*old_symbol); 1005460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 1015460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 102d0fbbb227051be16931a1aa9b4a7722ac039c698Shih-wei Liao // exist and is a symbol 1035460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // symbol resolution 1045460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao bool override = false; 1055460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao unsigned int action = Resolver::LastAction; 106f33f6de54db174aa679a4b6d1e040d37e95541c0Stephen Hines if (m_pResolver->resolve(*old_symbol, *new_symbol, override, pValue)) { 107affc150dc44fab1911775a49636d0ce85333b634Zonr Chang pResult.info = old_symbol; 108affc150dc44fab1911775a49636d0ce85333b634Zonr Chang pResult.existent = true; 109affc150dc44fab1911775a49636d0ce85333b634Zonr Chang pResult.overriden = override; 1105460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 1116f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines else { 112affc150dc44fab1911775a49636d0ce85333b634Zonr Chang m_pResolver->resolveAgain(*this, action, *old_symbol, *new_symbol, pResult); 1136f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines } 1146f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines 1156f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines m_Table.getEntryFactory().destroy(new_symbol); 1165460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return; 1175460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1185460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 119affc150dc44fab1911775a49636d0ce85333b634Zonr Changllvm::StringRef NamePool::insertString(const llvm::StringRef& pString) 1205460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1215460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao bool exist = false; 1225460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo* resolve_info = m_Table.insert(pString, exist); 1235460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return llvm::StringRef(resolve_info->name(), resolve_info->nameSize()); 1245460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1255460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 126affc150dc44fab1911775a49636d0ce85333b634Zonr Changvoid NamePool::reserve(NamePool::size_type pSize) 1275460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1285460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_Table.rehash(pSize); 1295460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1305460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 131affc150dc44fab1911775a49636d0ce85333b634Zonr ChangNamePool::size_type NamePool::capacity() const 1325460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1335460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return (m_Table.numOfBuckets() - m_Table.numOfEntries()); 1345460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1355460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1365460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/// findInfo - find the resolved ResolveInfo 137affc150dc44fab1911775a49636d0ce85333b634Zonr ChangResolveInfo* NamePool::findInfo(const llvm::StringRef& pName) 1385460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1395460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao Table::iterator iter = m_Table.find(pName); 1405460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return iter.getEntry(); 1415460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1425460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1435460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/// findInfo - find the resolved ResolveInfo 144affc150dc44fab1911775a49636d0ce85333b634Zonr Changconst ResolveInfo* NamePool::findInfo(const llvm::StringRef& pName) const 1455460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1465460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao Table::const_iterator iter = m_Table.find(pName); 1475460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return iter.getEntry(); 1485460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1495460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1505460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/// findSymbol - find the resolved output LDSymbol 151affc150dc44fab1911775a49636d0ce85333b634Zonr ChangLDSymbol* NamePool::findSymbol(const llvm::StringRef& pName) 1525460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1535460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ResolveInfo* info = findInfo(pName); 1545460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (NULL == info) 1555460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return NULL; 1565460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return info->outSymbol(); 1575460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1585460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1595460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao/// findSymbol - find the resolved output LDSymbol 160affc150dc44fab1911775a49636d0ce85333b634Zonr Changconst LDSymbol* NamePool::findSymbol(const llvm::StringRef& pName) const 1615460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1625460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao const ResolveInfo* info = findInfo(pName); 1635460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (NULL == info) 1645460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return NULL; 1655460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return info->outSymbol(); 1665460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1675460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 168