23using namespace object;
26Error COFFReader::readExecutableHeaders(Object &Obj)
const {
28 Obj.Is64 = COFFObj.
is64();
34 if (DH->AddressOfNewExeHeader >
sizeof(*DH))
35 Obj.DosStub = ArrayRef<uint8_t>(
reinterpret_cast<const uint8_t *
>(&DH[1]),
36 DH->AddressOfNewExeHeader -
sizeof(*DH));
44 Obj.BaseOfData = PE32->BaseOfData;
47 for (
size_t I = 0;
I < Obj.PeHeader.NumberOfRvaAndSize;
I++) {
51 Obj.DataDirectories.emplace_back(*Dir);
56Error COFFReader::readSections(Object &Obj)
const {
57 std::vector<Section> Sections;
60 Expected<const coff_section *> SecOrErr = COFFObj.
getSection(
I);
62 return SecOrErr.takeError();
63 const coff_section *Sec = *SecOrErr;
68 ArrayRef<uint8_t> Contents;
71 S.setContentsRef(Contents);
77 return NameOrErr.takeError();
79 Obj.addSections(Sections);
83Error COFFReader::readSymbols(Object &Obj,
bool IsBigObj)
const {
84 std::vector<Symbol> Symbols;
86 ArrayRef<Section> Sections = Obj.getSections();
88 Expected<COFFSymbolRef> SymOrErr = COFFObj.
getSymbol(
I);
90 return SymOrErr.takeError();
91 COFFSymbolRef SymRef = *SymOrErr;
93 Symbols.push_back(
Symbol());
98 *
reinterpret_cast<const coff_symbol32 *
>(SymRef.getRawPtr()));
101 *
reinterpret_cast<const coff_symbol16 *
>(SymRef.getRawPtr()));
104 return NameOrErr.takeError();
105 Sym.Name = *NameOrErr;
109 assert(AuxData.size() == SymSize * SymRef.getNumberOfAuxSymbols());
116 if (SymRef.isFileRecord())
117 Sym.AuxFile = StringRef(
reinterpret_cast<const char *
>(AuxData.data()),
121 for (
size_t I = 0;
I < SymRef.getNumberOfAuxSymbols();
I++)
122 Sym.AuxData.push_back(AuxData.slice(
I * SymSize,
sizeof(AuxSymbol)));
125 if (SymRef.getSectionNumber() <=
127 Sym.TargetSectionId = SymRef.getSectionNumber();
128 else if (
static_cast<uint32_t>(SymRef.getSectionNumber() - 1) <
130 Sym.TargetSectionId = Sections[SymRef.getSectionNumber() - 1].UniqueId;
133 "section number out of range");
136 const coff_aux_section_definition *SD = SymRef.getSectionDefinition();
137 const coff_aux_weak_external *WE = SymRef.getWeakExternal();
139 int32_t
Index = SD->getNumber(IsBigObj);
142 "unexpected associative section index");
143 Sym.AssociativeComdatTargetSectionId = Sections[
Index - 1].UniqueId;
148 Sym.WeakTargetSymbolId = WE->TagIndex;
150 I += 1 + SymRef.getNumberOfAuxSymbols();
152 Obj.addSymbols(Symbols);
156Error COFFReader::setSymbolTargets(Object &Obj)
const {
157 std::vector<const Symbol *> RawSymbolTable;
158 for (
const Symbol &
Sym : Obj.getSymbols()) {
159 RawSymbolTable.push_back(&
Sym);
160 for (
size_t I = 0;
I <
Sym.Sym.NumberOfAuxSymbols;
I++)
161 RawSymbolTable.push_back(
nullptr);
163 for (Symbol &
Sym : Obj.getMutableSymbols()) {
166 if (
Sym.WeakTargetSymbolId) {
167 if (*
Sym.WeakTargetSymbolId >= RawSymbolTable.size())
169 "weak external reference out of range");
171 if (Target ==
nullptr)
173 "invalid SymbolTableIndex");
174 Sym.WeakTargetSymbolId =
Target->UniqueId;
177 for (Section &Sec : Obj.getMutableSections()) {
178 for (Relocation &R : Sec.Relocs) {
179 if (
R.Reloc.SymbolTableIndex >= RawSymbolTable.size())
181 "SymbolTableIndex out of range");
182 const Symbol *
Sym = RawSymbolTable[
R.Reloc.SymbolTableIndex];
185 "invalid SymbolTableIndex");
186 R.Target =
Sym->UniqueId;
187 R.TargetName =
Sym->Name;
194 auto Obj = std::make_unique<Object>();
196 bool IsBigObj =
false;
198 Obj->CoffFileHeader = *CFH;
203 "no COFF file header returned");
206 Obj->CoffFileHeader.Machine = CBFH->
Machine;
211 if (
Error E = readExecutableHeaders(*Obj))
213 if (
Error E = readSections(*Obj))
215 if (
Error E = readSymbols(*Obj, IsBigObj))
217 if (
Error E = setSymbolTargets(*Obj))
220 return std::move(Obj);
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
Lightweight error class with error context and mandatory checking.
static ErrorSuccess success()
Create a success value.
Tagged union holding either a T or a Error.
Expected< std::unique_ptr< Object > > create() const
const dos_header * getDOSHeader() const
Expected< StringRef > getSectionName(DataRefImpl Sec) const override
const pe32_header * getPE32Header() const
Expected< COFFSymbolRef > getSymbol(uint32_t index) const
Expected< StringRef > getSymbolName(DataRefImpl Symb) const override
uint32_t getNumberOfSymbols() const
const coff_file_header * getCOFFHeader() const
const coff_bigobj_file_header * getCOFFBigObjHeader() const
uint32_t getNumberOfSections() const
Expected< const coff_section * > getSection(int32_t index) const
Expected< ArrayRef< uint8_t > > getSectionContents(DataRefImpl Sec) const override
const data_directory * getDataDirectory(uint32_t index) const
ArrayRef< uint8_t > getSymbolAuxData(COFFSymbolRef Symbol) const
ArrayRef< coff_relocation > getRelocations(const coff_section *Sec) const
const pe32plus_header * getPE32PlusHeader() const
@ IMAGE_SCN_LNK_NRELOC_OVFL
@ IMAGE_COMDAT_SELECT_ASSOCIATIVE
void copySymbol(Symbol1Ty &Dest, const Symbol2Ty &Src)
void copyPeHeader(PeHeader1Ty &Dest, const PeHeader2Ty &Src)
coff_symbol< support::ulittle32_t > coff_symbol32
coff_symbol< support::ulittle16_t > coff_symbol16
This is an optimization pass for GlobalISel generic memory operations.
void append_range(Container &C, Range &&R)
Wrapper function to append range R to container C.
Error createStringError(std::error_code EC, char const *Fmt, const Ts &... Vals)
Create formatted StringError object.
LLVM_ABI Error errorCodeToError(std::error_code EC)
Helper for converting an std::error_code to a Error.