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
SectionSymbolSet()27 SectionSymbolSet::SectionSymbolSet()
28 {
29 m_pSectionSymbolMap = new SectHashTableType(16);
30 }
31
~SectionSymbolSet()32 SectionSymbolSet::~SectionSymbolSet()
33 {
34 delete m_pSectionSymbolMap;
35 }
36
add(LDSection & pOutSect,NamePool & pNamePool)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
finalize(LDSection & pOutSect,SymbolTable & pSymTab)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
get(const LDSection & pOutSect)94 LDSymbol* SectionSymbolSet::get(const LDSection& pOutSect)
95 {
96 SectHashTableType::iterator entry = m_pSectionSymbolMap->find(&pOutSect);
97 return entry.getEntry()->value();
98 }
99
get(const LDSection & pOutSect) const100 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