1 //===- SectionSymbolSet.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 <mcld/LD/SectionSymbolSet.h> 10 #include <mcld/LD/LDSection.h> 11 #include <mcld/LD/RelocData.h> 12 #include <mcld/LD/SectionData.h> 13 #include <mcld/LD/EhFrame.h> 14 #include <mcld/LD/ResolveInfo.h> 15 #include <mcld/LD/LDSymbol.h> 16 #include <mcld/LD/NamePool.h> 17 #include <mcld/Fragment/FragmentRef.h> 18 #include <mcld/LD/LDFileFormat.h> 19 20 21 using namespace mcld; 22 23 //===----------------------------------------------------------------------===// 24 // SectionSymbolSet 25 //===----------------------------------------------------------------------===// 26 27 SectionSymbolSet::SectionSymbolSet() 28 { 29 m_pSectionSymbolMap = new SectHashTableType(16); 30 } 31 32 SectionSymbolSet::~SectionSymbolSet() 33 { 34 delete m_pSectionSymbolMap; 35 } 36 37 bool SectionSymbolSet::add(LDSection& pOutSect, NamePool& pNamePool) 38 { 39 // create the resolveInfo for this section symbol 40 llvm::StringRef sym_name = llvm::StringRef(pOutSect.name()); 41 ResolveInfo* sym_info = pNamePool.createSymbol(sym_name, 42 false, 43 ResolveInfo::Section, 44 ResolveInfo::Define, 45 ResolveInfo::Local, 46 0x0, // size 47 ResolveInfo::Default); 48 49 // create the output section symbol and set its fragRef to the first fragment 50 // of the section 51 LDSymbol* sym = LDSymbol::Create(*sym_info); 52 sym_info->setSymPtr(sym); 53 54 // insert the symbol to the Section to Symbol hash map 55 bool exist = false; 56 SectHashTableType::entry_type* entry = 57 m_pSectionSymbolMap->insert(&pOutSect, exist); 58 assert(!exist); 59 entry->setValue(sym); 60 61 return true; 62 } 63 64 bool SectionSymbolSet::finalize(LDSection& pOutSect, 65 SymbolTable& pSymTab) 66 { 67 if (0x0 == pOutSect.size()) 68 return true; 69 70 LDSymbol* sym = get(pOutSect); 71 assert(NULL != sym); 72 FragmentRef* frag_ref = NULL; 73 switch (pOutSect.kind()) { 74 case LDFileFormat::Relocation: 75 // Relocation section should not have section symbol. 76 return true; 77 78 case LDFileFormat::EhFrame: 79 frag_ref = FragmentRef::Create( 80 pOutSect.getEhFrame()->getSectionData().front(), 0x0); 81 break; 82 83 default: 84 frag_ref = FragmentRef::Create(pOutSect.getSectionData()->front(), 0x0); 85 break; 86 } 87 sym->setFragmentRef(frag_ref); 88 // push symbol into output symbol table 89 pSymTab.add(*sym); 90 91 return true; 92 } 93 94 LDSymbol* SectionSymbolSet::get(const LDSection& pOutSect) 95 { 96 SectHashTableType::iterator entry = m_pSectionSymbolMap->find(&pOutSect); 97 return entry.getEntry()->value(); 98 } 99 100 const LDSymbol* SectionSymbolSet::get(const LDSection& pOutSect) const 101 { 102 SectHashTableType::iterator entry = m_pSectionSymbolMap->find(&pOutSect); 103 return entry.getEntry()->value(); 104 } 105 106
