21void MachOReader::readHeader(
Object &O)
const {
31template <
typename SectionType>
33 StringRef SegName(Sec.segname, strnlen(Sec.segname,
sizeof(Sec.segname)));
34 StringRef SectName(Sec.sectname, strnlen(Sec.sectname,
sizeof(Sec.sectname)));
60template <
typename SectionType,
typename SegmentType>
64 std::vector<std::unique_ptr<Section>> Sections;
65 for (
auto Curr =
reinterpret_cast<const SectionType *
>(LoadCmd.
Ptr +
67 End =
reinterpret_cast<const SectionType *
>(LoadCmd.
Ptr +
71 memcpy((
void *)&Sec,
reinterpret_cast<const char *
>(Curr),
90 return Data.takeError();
113 "Incorrect number of relocations");
115 return std::move(Sections);
118Error MachOReader::readLoadCommands(
Object &O)
const {
121 static constexpr char TextSegmentName[] =
"__TEXT";
124 switch (LoadCmd.C.cmd) {
125 case MachO::LC_CODE_SIGNATURE:
126 O.CodeSignatureCommandIndex =
O.LoadCommands.size();
128 case MachO::LC_SEGMENT:
132 if (
StringRef(
reinterpret_cast<const char *
>(
135 O.TextSegmentCommandIndex =
O.LoadCommands.size();
137 if (
Expected<std::vector<std::unique_ptr<Section>>> Sections =
138 extractSections<MachO::section, MachO::segment_command>(
139 LoadCmd, MachOObj, NextSectionIndex))
142 return Sections.takeError();
144 case MachO::LC_SEGMENT_64:
148 if (
StringRef(
reinterpret_cast<const char *
>(
151 O.TextSegmentCommandIndex =
O.LoadCommands.size();
153 if (
Expected<std::vector<std::unique_ptr<Section>>> Sections =
154 extractSections<MachO::section_64, MachO::segment_command_64>(
155 LoadCmd, MachOObj, NextSectionIndex))
158 return Sections.takeError();
160 case MachO::LC_SYMTAB:
161 O.SymTabCommandIndex =
O.LoadCommands.size();
163 case MachO::LC_DYSYMTAB:
164 O.DySymTabCommandIndex =
O.LoadCommands.size();
166 case MachO::LC_DYLD_INFO:
167 case MachO::LC_DYLD_INFO_ONLY:
168 O.DyLdInfoCommandIndex =
O.LoadCommands.size();
170 case MachO::LC_DATA_IN_CODE:
171 O.DataInCodeCommandIndex =
O.LoadCommands.size();
173 case MachO::LC_LINKER_OPTIMIZATION_HINT:
174 O.LinkerOptimizationHintCommandIndex =
O.LoadCommands.size();
176 case MachO::LC_FUNCTION_STARTS:
177 O.FunctionStartsCommandIndex =
O.LoadCommands.size();
179 case MachO::LC_DYLIB_CODE_SIGN_DRS:
180 O.DylibCodeSignDRsIndex =
O.LoadCommands.size();
182 case MachO::LC_DYLD_EXPORTS_TRIE:
183 O.ExportsTrieCommandIndex =
O.LoadCommands.size();
185 case MachO::LC_DYLD_CHAINED_FIXUPS:
186 O.ChainedFixupsCommandIndex =
O.LoadCommands.size();
189#define HANDLE_LOAD_COMMAND(LCName, LCValue, LCStruct) \
190 case MachO::LCName: \
191 memcpy((void *)&(LC.MachOLoadCommand.LCStruct##_data), LoadCmd.Ptr, \
192 sizeof(MachO::LCStruct)); \
193 if (MachOObj.isLittleEndian() != sys::IsLittleEndianHost) \
194 MachO::swapStruct(LC.MachOLoadCommand.LCStruct##_data); \
195 if (LoadCmd.C.cmdsize > sizeof(MachO::LCStruct)) \
196 LC.Payload = ArrayRef<uint8_t>( \
197 reinterpret_cast<uint8_t *>(const_cast<char *>(LoadCmd.Ptr)) + \
198 sizeof(MachO::LCStruct), \
199 LoadCmd.C.cmdsize - sizeof(MachO::LCStruct)); \
202 switch (LoadCmd.C.cmd) {
210 reinterpret_cast<uint8_t *
>(
const_cast<char *
>(LoadCmd.Ptr)) +
214#include "llvm/BinaryFormat/MachO.def"
216 O.LoadCommands.push_back(std::move(LC));
221template <
typename nlist_t>
224 "n_strx exceeds the size of the string table");
234void MachOReader::readSymbolTable(
Object &O)
const {
236 for (
auto Symbol : MachOObj.
symbols()) {
240 Symbol.getRawDataRefImpl()))
242 Symbol.getRawDataRefImpl())));
244 O.SymTable.Symbols.push_back(std::make_unique<SymbolEntry>(SE));
248void MachOReader::setSymbolInRelocationInfo(
Object &O)
const {
249 std::vector<const Section *> Sections;
250 for (
auto &LC :
O.LoadCommands)
251 for (std::unique_ptr<Section> &Sec : LC.
Sections)
252 Sections.push_back(Sec.get());
255 for (std::unique_ptr<Section> &Sec : LC.
Sections)
256 for (
auto &Reloc : Sec->Relocations)
257 if (!Reloc.Scattered && !Reloc.IsAddend) {
261 Reloc.Symbol =
O.SymTable.getSymbolByIndex(SymbolNum);
265 assert(SymbolNum >= 1 && SymbolNum <= Sections.size() &&
266 "Invalid section index.");
267 Reloc.Sec = Sections[SymbolNum - 1];
272void MachOReader::readRebaseInfo(
Object &O)
const {
276void MachOReader::readBindInfo(
Object &O)
const {
280void MachOReader::readWeakBindInfo(
Object &O)
const {
284void MachOReader::readLazyBindInfo(
Object &O)
const {
288void MachOReader::readExportInfo(
Object &O)
const {
293 O.Exports.Trie = Trie;
296void MachOReader::readLinkData(
Object &O, std::optional<size_t> LCIndex,
301 O.LoadCommands[*LCIndex].MachOLoadCommand.linkedit_data_command_data;
306void MachOReader::readDataInCodeData(
Object &O)
const {
307 return readLinkData(O,
O.DataInCodeCommandIndex,
O.DataInCode);
310void MachOReader::readLinkerOptimizationHint(
Object &O)
const {
311 return readLinkData(O,
O.LinkerOptimizationHintCommandIndex,
312 O.LinkerOptimizationHint);
315void MachOReader::readFunctionStartsData(
Object &O)
const {
316 return readLinkData(O,
O.FunctionStartsCommandIndex,
O.FunctionStarts);
319void MachOReader::readDylibCodeSignDRs(
Object &O)
const {
320 return readLinkData(O,
O.DylibCodeSignDRsIndex,
O.DylibCodeSignDRs);
323void MachOReader::readExportsTrie(
Object &O)
const {
324 return readLinkData(O,
O.ExportsTrieCommandIndex,
O.ExportsTrie);
327void MachOReader::readChainedFixups(
Object &O)
const {
328 return readLinkData(O,
O.ChainedFixupsCommandIndex,
O.ChainedFixups);
331void MachOReader::readIndirectSymbolTable(
Object &O)
const {
337 if ((
Index & AbsOrLocalMask) != 0)
338 O.IndirectSymTable.Symbols.emplace_back(
Index, std::nullopt);
340 O.IndirectSymTable.Symbols.emplace_back(
345void MachOReader::readSwiftVersion(
Object &O)
const {
346 struct ObjCImageInfo {
352 for (
const std::unique_ptr<Section> &Sec : LC.
Sections)
353 if (Sec->Sectname ==
"__objc_imageinfo" &&
354 (Sec->Segname ==
"__DATA" || Sec->Segname ==
"__DATA_CONST" ||
355 Sec->Segname ==
"__DATA_DIRTY") &&
356 Sec->Content.size() >=
sizeof(ObjCImageInfo)) {
357 memcpy(&ImageInfo, Sec->Content.data(),
sizeof(ObjCImageInfo));
362 O.SwiftVersion = (ImageInfo.Flags >> 8) & 0xff;
368 auto Obj = std::make_unique<Object>();
370 if (
Error E = readLoadCommands(*Obj))
372 readSymbolTable(*Obj);
373 setSymbolInRelocationInfo(*Obj);
374 readRebaseInfo(*Obj);
376 readWeakBindInfo(*Obj);
377 readLazyBindInfo(*Obj);
378 readExportInfo(*Obj);
379 readDataInCodeData(*Obj);
380 readLinkerOptimizationHint(*Obj);
381 readFunctionStartsData(*Obj);
382 readDylibCodeSignDRs(*Obj);
383 readExportsTrie(*Obj);
384 readChainedFixups(*Obj);
385 readIndirectSymbolTable(*Obj);
386 readSwiftVersion(*Obj);
387 return std::move(Obj);
#define offsetof(TYPE, MEMBER)
static Section constructSectionCommon(const SectionType &Sec, uint32_t Index)
static Expected< std::vector< std::unique_ptr< Section > > > extractSections(const object::MachOObjectFile::LoadCommandInfo &LoadCmd, const object::MachOObjectFile &MachOObj, uint32_t &NextSectionIndex)
Section constructSection(const MachO::section &Sec, uint32_t Index)
SymbolEntry constructSymbolEntry(StringRef StrTable, const nlist_t &nlist)
assert(ImpDefSCC.getReg()==AMDGPU::SCC &&ImpDefSCC.isDef())
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory),...
bool empty() const
empty - Check if the array is empty.
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.
Error takeError()
Take ownership of the stored error.
StringRef - Represent a constant reference to a string, i.e.
std::string str() const
str - Get the contents as an std::string.
constexpr StringRef substr(size_t Start, size_t N=npos) const
Return a reference to the substring from [Start, Start + N).
constexpr size_t size() const
size - Get the string size.
constexpr const char * data() const
data - Get a pointer to the start of the string (which may not be null terminated).
The instances of the Type class are immutable: once they are created, they are never changed.
Expected< std::unique_ptr< Object > > create() const override
StringRef getData() const
bool isLittleEndian() const
Expected< SectionRef > getSection(unsigned SectionIndex) const
ArrayRef< uint8_t > getDyldInfoLazyBindOpcodes() const
MachO::dysymtab_command getDysymtabLoadCommand() const
relocation_iterator section_rel_begin(DataRefImpl Sec) const override
ArrayRef< uint8_t > getSectionContents(uint32_t Offset, uint64_t Size) const
relocation_iterator section_rel_end(DataRefImpl Sec) const override
ArrayRef< uint8_t > getDyldInfoExportsTrie() const
MachO::nlist_64 getSymbol64TableEntry(DataRefImpl DRI) const
unsigned getAnyRelocationType(const MachO::any_relocation_info &RE) const
ArrayRef< uint8_t > getDyldExportsTrie() const
bool getPlainRelocationExternal(const MachO::any_relocation_info &RE) const
const MachO::mach_header & getHeader() const
ArrayRef< uint8_t > getDyldInfoRebaseOpcodes() const
iterator_range< load_command_iterator > load_commands() const
bool is64Bit() const override
MachO::any_relocation_info getRelocation(DataRefImpl Rel) const
ArrayRef< uint8_t > getDyldInfoBindOpcodes() const
ArrayRef< uint8_t > getDyldInfoWeakBindOpcodes() const
bool isRelocationScattered(const MachO::any_relocation_info &RE) const
StringRef getStringTableData() const
uint32_t getIndirectSymbolTableEntry(const MachO::dysymtab_command &DLC, unsigned Index) const
symbol_iterator_range symbols() const
void swapStruct(fat_header &mh)
static const bool IsLittleEndianHost
void swapByteOrder(T &Value)
This is an optimization pass for GlobalISel generic memory operations.
MachO::macho_load_command MachOLoadCommand
std::vector< std::unique_ptr< Section > > Sections
std::vector< uint8_t > Payload
std::optional< uint32_t > OriginalOffset
std::vector< RelocationInfo > Relocations