macho_dylib_file.cpp [plain text]
#include <stdint.h>
#include <math.h>
#include <unistd.h>
#include <sys/param.h>
#include <sys/mman.h>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
#include "Architectures.hpp"
#include "Bitcode.hpp"
#include "MachOFileAbstraction.hpp"
#include "MachOTrie.hpp"
#include "generic_dylib_file.hpp"
#include "macho_dylib_file.h"
#include "../code-sign-blobs/superblob.h"
namespace mach_o {
namespace dylib {
template <typename A>
class File final : public generic::dylib::File<A>
{
using Base = generic::dylib::File<A>;
public:
static bool validFile(const uint8_t* fileContent, bool executableOrDylib, bool subTypeMustMatch=false);
File(const uint8_t* fileContent, uint64_t fileLength, const char* path,
time_t mTime, ld::File::Ordinal ordinal, bool linkingFlatNamespace,
bool linkingMainExecutable, bool hoistImplicitPublicDylibs,
const ld::VersionSet& platforms, bool allowWeakImports,
bool allowSimToMacOSX, bool addVers, bool buildingForSimulator,
bool logAllFiles, const char* installPath,
bool indirectDylib, bool ignoreMismatchPlatform, bool usingBitcode);
virtual ~File() noexcept {}
virtual const ld::VersionSet& platforms() const { return this->_platforms; }
private:
using P = typename A::P;
using E = typename A::P::E;
using pint_t = typename A::P::uint_t;
void addDyldFastStub();
void buildExportHashTableFromExportInfo(const macho_dyld_info_command<P>* dyldInfo,
const uint8_t* fileContent);
void buildExportHashTableFromSymbolTable(const macho_dysymtab_command<P>* dynamicInfo,
const macho_nlist<P>* symbolTable, const char* strings,
const uint8_t* fileContent);
void addSymbol(const char* name, bool weakDef = false, bool tlv = false, pint_t address = 0);
static const char* objCInfoSegmentName();
static const char* objCInfoSectionName();
static bool useSimulatorVariant();
uint64_t _fileLength;
uint32_t _linkeditStartOffset;
};
template <> const char* File<x86_64>::objCInfoSegmentName() { return "__DATA"; }
template <> const char* File<arm>::objCInfoSegmentName() { return "__DATA"; }
template <typename A> const char* File<A>::objCInfoSegmentName() { return "__OBJC"; }
template <> const char* File<x86_64>::objCInfoSectionName() { return "__objc_imageinfo"; }
template <> const char* File<arm>::objCInfoSectionName() { return "__objc_imageinfo"; }
template <typename A> const char* File<A>::objCInfoSectionName() { return "__image_info"; }
template <> bool File<x86>::useSimulatorVariant() { return true; }
template <> bool File<x86_64>::useSimulatorVariant() { return true; }
template <typename A> bool File<A>::useSimulatorVariant() { return false; }
template <typename A>
File<A>::File(const uint8_t* fileContent, uint64_t fileLength, const char* path, time_t mTime,
ld::File::Ordinal ord, bool linkingFlatNamespace, bool linkingMainExecutable,
bool hoistImplicitPublicDylibs, const ld::VersionSet& platforms, bool allowWeakImports,
bool allowSimToMacOSX, bool addVers, bool buildingForSimulator, bool logAllFiles,
const char* targetInstallPath, bool indirectDylib, bool ignoreMismatchPlatform, bool usingBitcode)
: Base(strdup(path), mTime, ord, platforms, allowWeakImports, linkingFlatNamespace,
hoistImplicitPublicDylibs, allowSimToMacOSX, addVers), _fileLength(fileLength), _linkeditStartOffset(0)
{
const macho_header<P>* header = (const macho_header<P>*)fileContent;
const uint32_t cmd_count = header->ncmds();
const macho_load_command<P>* const cmds = (macho_load_command<P>*)((char*)header + sizeof(macho_header<P>));
const macho_load_command<P>* const cmdsEnd = (macho_load_command<P>*)((char*)header + sizeof(macho_header<P>) + header->sizeofcmds());
if ( logAllFiles )
printf("%s\n", path);
if ( (header->filetype() == MH_DYLIB_STUB) && (cmd_count == 0) ) {
munmap((caddr_t)fileContent, fileLength);
return;
}
this->_noRexports = (header->flags() & MH_NO_REEXPORTED_DYLIBS)
|| (header->filetype() == MH_BUNDLE)
|| (header->filetype() == MH_EXECUTE); this->_hasWeakExports = (header->flags() & MH_WEAK_DEFINES);
this->_deadStrippable = (header->flags() & MH_DEAD_STRIPPABLE_DYLIB);
this->_appExtensionSafe = (header->flags() & MH_APP_EXTENSION_SAFE);
const macho_dysymtab_command<P>* dynamicInfo = nullptr;
const macho_dyld_info_command<P>* dyldInfo = nullptr;
const macho_nlist<P>* symbolTable = nullptr;
const macho_symtab_command<P>* symtab = nullptr;
const char* strings = nullptr;
bool compressedLinkEdit = false;
uint32_t dependentLibCount = 0;
ld::VersionSet lcPlatforms;
const macho_load_command<P>* cmd = cmds;
for (uint32_t i = 0; i < cmd_count; ++i) {
macho_dylib_command<P>* dylibID;
uint32_t cmdLength = cmd->cmdsize();
switch (cmd->cmd()) {
case LC_SYMTAB:
symtab = (macho_symtab_command<P>*)cmd;
symbolTable = (const macho_nlist<P>*)((char*)header + symtab->symoff());
strings = (char*)header + symtab->stroff();
if ( (symtab->stroff() + symtab->strsize()) > fileLength )
throwf("mach-o string pool extends beyond end of file in %s", path);
break;
case LC_DYSYMTAB:
dynamicInfo = (macho_dysymtab_command<P>*)cmd;
break;
case LC_DYLD_INFO:
case LC_DYLD_INFO_ONLY:
dyldInfo = (macho_dyld_info_command<P>*)cmd;
compressedLinkEdit = true;
break;
case LC_ID_DYLIB:
dylibID = (macho_dylib_command<P>*)cmd;
if ( dylibID->name_offset() > cmdLength )
throwf("malformed mach-o: LC_ID_DYLIB load command has offset (%u) outside its size (%u)", dylibID->name_offset(), cmdLength);
if ( (dylibID->name_offset() + strlen(dylibID->name()) + 1) > cmdLength )
throwf("malformed mach-o: LC_ID_DYLIB load command string extends beyond end of load command");
this->_dylibInstallPath = strdup(dylibID->name());
this->_dylibTimeStamp = dylibID->timestamp();
this->_dylibCurrentVersion = dylibID->current_version();
this->_dylibCompatibilityVersion = dylibID->compatibility_version();
this->_hasPublicInstallName = this->isPublicLocation(this->_dylibInstallPath);
break;
case LC_LOAD_DYLIB:
case LC_LOAD_WEAK_DYLIB:
++dependentLibCount;
break;
case LC_REEXPORT_DYLIB:
this->_explictReExportFound = true;
++dependentLibCount;
break;
case LC_SUB_FRAMEWORK:
this->_parentUmbrella = strdup(((macho_sub_framework_command<P>*)cmd)->umbrella());
break;
case LC_SUB_CLIENT:
this->_allowableClients.push_back(strdup(((macho_sub_client_command<P>*)cmd)->client()));
this->_hasPublicInstallName = false;
break;
case LC_RPATH:
this->_rpaths.push_back(strdup(((macho_rpath_command<P>*)cmd)->path()));
break;
case LC_VERSION_MIN_MACOSX:
case LC_VERSION_MIN_IPHONEOS:
case LC_VERSION_MIN_WATCHOS:
case LC_VERSION_MIN_TVOS:
lcPlatforms.add({Options::platformForLoadCommand(cmd->cmd(), useSimulatorVariant()), ((macho_version_min_command<P>*)cmd)->version()});
break;
case LC_BUILD_VERSION:
{
const macho_build_version_command<P>* buildVersCmd = (macho_build_version_command<P>*)cmd;
lcPlatforms.add({(ld::Platform)buildVersCmd->platform(), buildVersCmd->minos()});
}
break;
case LC_CODE_SIGNATURE:
break;
case macho_segment_command<P>::CMD:
if ( strncmp(((macho_segment_command<P>*)cmd)->segname(), objCInfoSegmentName(), 6) == 0 ) {
const macho_segment_command<P>* segment = (macho_segment_command<P>*)cmd;
const macho_section<P>* const sectionsStart = (macho_section<P>*)((char*)segment + sizeof(macho_segment_command<P>));
const macho_section<P>* const sectionsEnd = §ionsStart[segment->nsects()];
for (const macho_section<P>* sect=sectionsStart; sect < sectionsEnd; ++sect) {
if ( strncmp(sect->sectname(), objCInfoSectionName(), strlen(objCInfoSectionName())) == 0 ) {
const uint32_t* contents = (uint32_t*)(&fileContent[sect->offset()]);
if ( (sect->size() >= 8) && (contents[0] == 0) ) {
uint32_t flags = E::get32(contents[1]);
this->_swiftVersion = ((flags >> 8) & 0xFF);
}
else if ( sect->size() > 0 ) {
warning("can't parse %s/%s section in %s", objCInfoSegmentName(), objCInfoSectionName(), path);
}
}
}
}
else if ( strcmp(((macho_segment_command<P>*)cmd)->segname(), "__LLVM") == 0 ) {
const macho_section<P>* const sect = (macho_section<P>*)((char*)cmd + sizeof(macho_segment_command<P>));
if ( strncmp(sect->sectname(), "__bundle", 8) == 0 )
this->_bitcode = std::unique_ptr<ld::Bitcode>(new ld::Bitcode(NULL, sect->size()));
}
else if ( strcmp(((macho_segment_command<P>*)cmd)->segname(), "__LINKEDIT") == 0 ) {
_linkeditStartOffset = ((macho_segment_command<P>*)cmd)->fileoff();
}
}
cmd = (const macho_load_command<P>*)(((char*)cmd)+cmdLength);
if ( cmd > cmdsEnd )
throwf("malformed dylb, load command #%d is outside size of load commands in %s", i, path);
}
if (lcPlatforms.empty() &&
(std::is_same<A, arm>::value || std::is_same<A, arm64>::value))
lcPlatforms.add({ld::kPlatform_iOS, 0});
platforms.forEach(^(ld::Platform platform, uint32_t version, bool &stop) {
if (!lcPlatforms.contains(platform) ) {
this->_wrongOS = true;
if ( this->_addVersionLoadCommand && !indirectDylib && !ignoreMismatchPlatform ) {
if (buildingForSimulator && !this->_allowSimToMacOSXLinking) {
if ( usingBitcode )
throwf("building for %s simulator, but linking against dylib built for %s,",
platforms.to_str().c_str(), lcPlatforms.to_str().c_str());
else
warning("URGENT: building for %s simulator, but linking against dylib (%s) built for %s. "
"Note: This will be an error in the future.",
platforms.to_str().c_str(), path, lcPlatforms.to_str().c_str());
}
} else {
if ( usingBitcode )
throwf("building for %s, but linking against dylib built for %s,",
platforms.to_str().c_str(), lcPlatforms.to_str().c_str());
else if ( (getenv("RC_XBS") != NULL) && (getenv("RC_BUILDIT") == NULL) ) warning("URGENT: building for %s, but linking against dylib (%s) built for %s. "
"Note: This will be an error in the future.",
platforms.to_str().c_str(), path, lcPlatforms.to_str().c_str());
}
}
});
bool processDependentLibraries = true;
if ( compressedLinkEdit && this->_noRexports && !linkingFlatNamespace)
processDependentLibraries = false;
if ( processDependentLibraries ) {
this->_dependentDylibs.reserve(dependentLibCount);
cmd = cmds;
unsigned int reExportDylibCount = 0;
for (uint32_t i = 0; i < cmd_count; ++i) {
uint32_t cmdLength = cmd->cmdsize();
const macho_dylib_command<P>* dylibCmd = (macho_dylib_command<P>*)cmd;
switch (cmd->cmd()) {
case LC_LOAD_DYLIB:
case LC_LOAD_WEAK_DYLIB:
if ( compressedLinkEdit && !linkingFlatNamespace )
break;
case LC_REEXPORT_DYLIB:
++reExportDylibCount;
if ( dylibCmd->name_offset() > cmdLength )
throwf("malformed mach-o: LC_*_DYLIB load command has offset (%u) outside its size (%u)", dylibCmd->name_offset(), cmdLength);
if ( (dylibCmd->name_offset() + strlen(dylibCmd->name()) + 1) > cmdLength )
throwf("malformed mach-o: LC_*_DYLIB load command string extends beyond end of load command");
const char *path = strdup(dylibCmd->name());
bool reExport = (cmd->cmd() == LC_REEXPORT_DYLIB);
if ( (targetInstallPath == nullptr) || (strcmp(targetInstallPath, path) != 0) )
this->_dependentDylibs.emplace_back(path, reExport);
break;
}
cmd = (const macho_load_command<P>*)(((char*)cmd)+cmdLength);
}
if ( compressedLinkEdit && !linkingFlatNamespace ) {
if ( reExportDylibCount == 0 )
throwf("malformed dylib has MH_NO_REEXPORTED_DYLIBS flag but no LC_REEXPORT_DYLIB load commands: %s", path);
}
cmd = cmds;
for (uint32_t i = 0; i < cmd_count; ++i) {
const char* frameworkLeafName;
const char* dylibBaseName;
switch (cmd->cmd()) {
case LC_SUB_UMBRELLA:
frameworkLeafName = ((macho_sub_umbrella_command<P>*)cmd)->sub_umbrella();
for (auto &dep : this->_dependentDylibs) {
const char* dylibName = dep.path;
const char* lastSlash = strrchr(dylibName, '/');
if ( (lastSlash != nullptr) && (strcmp(&lastSlash[1], frameworkLeafName) == 0) )
dep.reExport = true;
}
break;
case LC_SUB_LIBRARY:
dylibBaseName = ((macho_sub_library_command<P>*)cmd)->sub_library();
for (auto &dep : this->_dependentDylibs) {
const char* dylibName = dep.path;
const char* lastSlash = strrchr(dylibName, '/');
const char* leafStart = &lastSlash[1];
if ( lastSlash == nullptr )
leafStart = dylibName;
const char* firstDot = strchr(leafStart, '.');
int len = strlen(leafStart);
if ( firstDot != nullptr )
len = firstDot - leafStart;
if ( strncmp(leafStart, dylibBaseName, len) == 0 )
dep.reExport = true;
}
break;
}
cmd = (const macho_load_command<P>*)(((char*)cmd)+cmd->cmdsize());
}
}
if ( this->_dylibInstallPath != NULL ) {
const char* lastSlash = strrchr(this->_dylibInstallPath, '/');
if ( lastSlash != NULL ) {
const char* leafName = lastSlash+1;
char frname[strlen(leafName)+32];
strcpy(frname, leafName);
strcat(frname, ".framework/");
if ( strstr(this->_dylibInstallPath, frname) != NULL )
this->_frameworkName = leafName;
}
}
if ( (this->_dylibInstallPath == nullptr) && ((header->filetype() == MH_DYLIB) || (header->filetype() == MH_DYLIB_STUB)) )
throwf("dylib %s missing LC_ID_DYLIB load command", path);
if ( dyldInfo == nullptr ) {
if ( symbolTable == nullptr )
throw "binary missing LC_SYMTAB load command";
if ( dynamicInfo == nullptr )
throw "binary missing LC_DYSYMTAB load command";
}
if ( symtab != nullptr ) {
if ( symtab->symoff() < _linkeditStartOffset )
throwf("malformed mach-o, symbol table not in __LINKEDIT");
if ( symtab->stroff() < _linkeditStartOffset )
throwf("malformed mach-o, symbol table strings not in __LINKEDIT");
}
if ( linkingFlatNamespace && linkingMainExecutable && ((header->flags() & MH_TWOLEVEL) == 0) ) {
std::vector<const char*> importNames;
importNames.reserve(dynamicInfo->nundefsym());
const macho_nlist<P>* start = &symbolTable[dynamicInfo->iundefsym()];
const macho_nlist<P>* end = &start[dynamicInfo->nundefsym()];
for (const macho_nlist<P>* sym=start; sym < end; ++sym) {
importNames.push_back(&strings[sym->n_strx()]);
}
this->_importAtom = new generic::dylib::ImportAtom<A>(*this, importNames);
}
if ( dyldInfo != nullptr )
buildExportHashTableFromExportInfo(dyldInfo, fileContent);
else
buildExportHashTableFromSymbolTable(dynamicInfo, symbolTable, strings, fileContent);
munmap((caddr_t)fileContent, fileLength);
}
template <typename A>
void File<A>::buildExportHashTableFromSymbolTable(const macho_dysymtab_command<P>* dynamicInfo,
const macho_nlist<P>* symbolTable,
const char* strings, const uint8_t* fileContent)
{
if ( dynamicInfo->tocoff() == 0 ) {
if ( this->_s_logHashtable )
fprintf(stderr, "ld: building hashtable of %u toc entries for %s\n", dynamicInfo->nextdefsym(), this->path());
const macho_nlist<P>* start = &symbolTable[dynamicInfo->iextdefsym()];
const macho_nlist<P>* end = &start[dynamicInfo->nextdefsym()];
this->_atoms.reserve(dynamicInfo->nextdefsym()); for (const macho_nlist<P>* sym=start; sym < end; ++sym) {
this->addSymbol(&strings[sym->n_strx()], (sym->n_desc() & N_WEAK_DEF) != 0, false, sym->n_value());
}
}
else {
int32_t count = dynamicInfo->ntoc();
this->_atoms.reserve(count); if ( this->_s_logHashtable )
fprintf(stderr, "ld: building hashtable of %u entries for %s\n", count, this->path());
const auto* toc = reinterpret_cast<const dylib_table_of_contents*>(fileContent + dynamicInfo->tocoff());
for (int32_t i = 0; i < count; ++i) {
const uint32_t index = E::get32(toc[i].symbol_index);
const macho_nlist<P>* sym = &symbolTable[index];
this->addSymbol(&strings[sym->n_strx()], (sym->n_desc() & N_WEAK_DEF) != 0, false, sym->n_value());
}
}
if ( (this->_dylibInstallPath != nullptr) && (strcmp(this->_dylibInstallPath, "/usr/lib/libSystem.B.dylib") == 0) )
addDyldFastStub();
}
template <typename A>
void File<A>::buildExportHashTableFromExportInfo(const macho_dyld_info_command<P>* dyldInfo,
const uint8_t* fileContent)
{
if ( this->_s_logHashtable )
fprintf(stderr, "ld: building hashtable from export info in %s\n", this->path());
if ( dyldInfo->export_size() > 0 ) {
const uint8_t* start = fileContent + dyldInfo->export_off();
const uint8_t* end = &start[dyldInfo->export_size()];
if ( (dyldInfo->export_off() + dyldInfo->export_size()) > _fileLength )
throwf("malformed mach-o dylib, exports trie extends beyond end of file, ");
std::vector<mach_o::trie::Entry> list;
parseTrie(start, end, list);
for (const auto &entry : list)
this->addSymbol(entry.name,
entry.flags & EXPORT_SYMBOL_FLAGS_WEAK_DEFINITION,
(entry.flags & EXPORT_SYMBOL_FLAGS_KIND_MASK) == EXPORT_SYMBOL_FLAGS_KIND_THREAD_LOCAL,
entry.address);
}
}
template <typename A>
void File<A>::addSymbol(const char* name, bool weakDef, bool tlv, pint_t address)
{
__block uint32_t linkMinOSVersion = 0;
this->platforms().forEach(^(ld::Platform platform, uint32_t version, bool &stop) {
if (linkMinOSVersion == 0)
linkMinOSVersion = version;
if (platform == ld::kPlatform_macOS)
linkMinOSVersion = version;
});
if ( strncmp(name, "$ld$", 4) == 0 ) {
const char* symAction = &name[4];
const char* symCond = strchr(symAction, '$');
if ( symCond != nullptr ) {
char curOSVers[16];
sprintf(curOSVers, "$os%d.%d$", (linkMinOSVersion >> 16), ((linkMinOSVersion >> 8) & 0xFF));
if ( strncmp(symCond, curOSVers, strlen(curOSVers)) == 0 ) {
const char* symName = strchr(&symCond[1], '$');
if ( symName != nullptr ) {
++symName;
if ( strncmp(symAction, "hide$", 5) == 0 ) {
if ( this->_s_logHashtable )
fprintf(stderr, " adding %s to ignore set for %s\n", symName, this->path());
this->_ignoreExports.insert(strdup(symName));
return;
}
else if ( strncmp(symAction, "add$", 4) == 0 ) {
this->addSymbol(symName, weakDef);
return;
}
else if ( strncmp(symAction, "weak$", 5) == 0 ) {
if ( !this->_allowWeakImports )
this->_ignoreExports.insert(strdup(symName));
}
else if ( strncmp(symAction, "install_name$", 13) == 0 ) {
this->_dylibInstallPath = strdup(symName);
this->_installPathOverride = true;
if ( strcmp(this->_dylibInstallPath, "/System/Library/Frameworks/ApplicationServices.framework/Versions/A/ApplicationServices") == 0 )
this->_dylibCompatibilityVersion = Options::parseVersionNumber32("1.0");
return;
}
else if ( strncmp(symAction, "compatibility_version$", 22) == 0 ) {
this->_dylibCompatibilityVersion = Options::parseVersionNumber32(symName);
return;
}
else {
warning("bad symbol action: %s in dylib %s", name, this->path());
}
}
}
}
else {
warning("bad symbol condition: %s in dylib %s", name, this->path());
}
}
if ( this->_ignoreExports.count(name) == 0 ) {
typename Base::AtomAndWeak bucket = { nullptr, weakDef, tlv, address };
if ( this->_s_logHashtable )
fprintf(stderr, " adding %s to hash table for %s\n", name, this->path());
this->_atoms[strdup(name)] = bucket;
}
}
template <>
void File<x86_64>::addDyldFastStub()
{
addSymbol("dyld_stub_binder");
}
template <>
void File<x86>::addDyldFastStub()
{
addSymbol("dyld_stub_binder");
}
template <typename A>
void File<A>::addDyldFastStub()
{
}
template <typename A>
class Parser
{
public:
using P = typename A::P;
static bool validFile(const uint8_t* fileContent, bool executableOrDyliborBundle, bool subTypeMustMatch=false, uint32_t subType=0);
static const char* fileKind(const uint8_t* fileContent);
static ld::dylib::File* parse(const uint8_t* fileContent, uint64_t fileLength, const char* path,
time_t mTime, ld::File::Ordinal ordinal, const Options& opts,
bool indirectDylib)
{
return new File<A>(fileContent, fileLength, path, mTime, ordinal, opts.flatNamespace(),
opts.linkingMainExecutable(), opts.implicitlyLinkIndirectPublicDylibs(),
opts.platforms(), opts.allowWeakImports(),
opts.allowSimulatorToLinkWithMacOSX(), opts.addVersionLoadCommand(),
opts.targetIOSSimulator(), opts.logAllFiles(), opts.installPath(),
indirectDylib, opts.outputKind() == Options::kPreload, opts.bundleBitcode());
}
};
template <>
bool Parser<x86>::validFile(const uint8_t* fileContent, bool executableOrDyliborBundle, bool subTypeMustMatch, uint32_t subType)
{
const auto* header = reinterpret_cast<const macho_header<P>*>(fileContent);
if ( header->magic() != MH_MAGIC )
return false;
if ( header->cputype() != CPU_TYPE_I386 )
return false;
switch ( header->filetype() ) {
case MH_DYLIB:
case MH_DYLIB_STUB:
return true;
case MH_BUNDLE:
if ( executableOrDyliborBundle )
return true;
else
throw "can't link with bundle (MH_BUNDLE) only dylibs (MH_DYLIB)";
case MH_EXECUTE:
if ( executableOrDyliborBundle )
return true;
else
throw "can't link with a main executable";
default:
return false;
}
}
template <>
bool Parser<x86_64>::validFile(const uint8_t* fileContent, bool executableOrDyliborBundle, bool subTypeMustMatch, uint32_t subType)
{
const auto* header = reinterpret_cast<const macho_header<P>*>(fileContent);
if ( header->magic() != MH_MAGIC_64 )
return false;
if ( header->cputype() != CPU_TYPE_X86_64 )
return false;
switch ( header->filetype() ) {
case MH_DYLIB:
case MH_DYLIB_STUB:
return true;
case MH_BUNDLE:
if ( executableOrDyliborBundle )
return true;
else
throw "can't link with bundle (MH_BUNDLE) only dylibs (MH_DYLIB)";
case MH_EXECUTE:
if ( executableOrDyliborBundle )
return true;
else
throw "can't link with a main executable";
default:
return false;
}
}
template <>
bool Parser<arm>::validFile(const uint8_t* fileContent, bool executableOrDyliborBundle, bool subTypeMustMatch, uint32_t subType)
{
const auto* header = reinterpret_cast<const macho_header<P>*>(fileContent);
if ( header->magic() != MH_MAGIC )
return false;
if ( header->cputype() != CPU_TYPE_ARM )
return false;
if ( subTypeMustMatch && (header->cpusubtype() != subType) )
return false;
switch ( header->filetype() ) {
case MH_DYLIB:
case MH_DYLIB_STUB:
return true;
case MH_BUNDLE:
if ( executableOrDyliborBundle )
return true;
else
throw "can't link with bundle (MH_BUNDLE) only dylibs (MH_DYLIB)";
case MH_EXECUTE:
if ( executableOrDyliborBundle )
return true;
else
throw "can't link with a main executable";
default:
return false;
}
}
template <>
bool Parser<arm64>::validFile(const uint8_t* fileContent, bool executableOrDyliborBundle, bool subTypeMustMatch, uint32_t subType)
{
const auto* header = reinterpret_cast<const macho_header<P>*>(fileContent);
if ( header->magic() != MH_MAGIC_64 )
return false;
if ( header->cputype() != CPU_TYPE_ARM64 )
return false;
switch ( header->filetype() ) {
case MH_DYLIB:
case MH_DYLIB_STUB:
return true;
case MH_BUNDLE:
if ( executableOrDyliborBundle )
return true;
else
throw "can't link with bundle (MH_BUNDLE) only dylibs (MH_DYLIB)";
case MH_EXECUTE:
if ( executableOrDyliborBundle )
return true;
else
throw "can't link with a main executable";
default:
return false;
}
}
bool isDylibFile(const uint8_t* fileContent, cpu_type_t* result, cpu_subtype_t* subResult)
{
if ( Parser<x86_64>::validFile(fileContent, false) ) {
*result = CPU_TYPE_X86_64;
const auto* header = reinterpret_cast<const macho_header<Pointer64<LittleEndian>>*>(fileContent);
*subResult = header->cpusubtype();
return true;
}
if ( Parser<x86>::validFile(fileContent, false) ) {
*result = CPU_TYPE_I386;
*subResult = CPU_SUBTYPE_X86_ALL;
return true;
}
if ( Parser<arm>::validFile(fileContent, false) ) {
*result = CPU_TYPE_ARM;
const auto* header = reinterpret_cast<const macho_header<Pointer32<LittleEndian>>*>(fileContent);
*subResult = header->cpusubtype();
return true;
}
if ( Parser<arm64>::validFile(fileContent, false) ) {
*result = CPU_TYPE_ARM64;
const auto* header = reinterpret_cast<const macho_header<Pointer32<LittleEndian>>*>(fileContent);
*subResult = header->cpusubtype();
return true;
}
return false;
}
template <>
const char* Parser<x86>::fileKind(const uint8_t* fileContent)
{
const auto* header = reinterpret_cast<const macho_header<P>*>(fileContent);
if ( header->magic() != MH_MAGIC )
return nullptr;
if ( header->cputype() != CPU_TYPE_I386 )
return nullptr;
return "i386";
}
template <>
const char* Parser<x86_64>::fileKind(const uint8_t* fileContent)
{
const auto* header = reinterpret_cast<const macho_header<P>*>(fileContent);
if ( header->magic() != MH_MAGIC_64 )
return nullptr;
if ( header->cputype() != CPU_TYPE_X86_64 )
return nullptr;
return "x86_64";
}
template <>
const char* Parser<arm>::fileKind(const uint8_t* fileContent)
{
const auto* header = reinterpret_cast<const macho_header<P>*>(fileContent);
if ( header->magic() != MH_MAGIC )
return nullptr;
if ( header->cputype() != CPU_TYPE_ARM )
return nullptr;
for (const auto* t = archInfoArray; t->archName != nullptr; ++t) {
if ( (t->cpuType == CPU_TYPE_ARM) && ((cpu_subtype_t)header->cpusubtype() == t->cpuSubType) ) {
return t->archName;
}
}
return "arm???";
}
#if SUPPORT_ARCH_arm64
template <>
const char* Parser<arm64>::fileKind(const uint8_t* fileContent)
{
const auto* header = reinterpret_cast<const macho_header<P>*>(fileContent);
if ( header->magic() != MH_MAGIC_64 )
return nullptr;
if ( header->cputype() != CPU_TYPE_ARM64 )
return nullptr;
return "arm64";
}
#endif
const char* archName(const uint8_t* fileContent)
{
if ( Parser<x86_64>::validFile(fileContent, true) ) {
return Parser<x86_64>::fileKind(fileContent);
}
if ( Parser<x86>::validFile(fileContent, true) ) {
return Parser<x86>::fileKind(fileContent);
}
if ( Parser<arm>::validFile(fileContent, true) ) {
return Parser<arm>::fileKind(fileContent);
}
#if SUPPORT_ARCH_arm64
if ( Parser<arm64>::validFile(fileContent, true) ) {
return Parser<arm64>::fileKind(fileContent);
}
#endif
return nullptr;
}
static ld::dylib::File* parseAsArchitecture(const uint8_t* fileContent, uint64_t fileLength, const char* path,
time_t modTime, const Options& opts, ld::File::Ordinal ordinal,
bool bundleLoader, bool indirectDylib,
cpu_type_t architecture, cpu_subtype_t subArchitecture)
{
bool subTypeMustMatch = opts.enforceDylibSubtypesMatch();
switch ( architecture) {
#if SUPPORT_ARCH_x86_64
case CPU_TYPE_X86_64:
if ( Parser<x86_64>::validFile(fileContent, bundleLoader, subTypeMustMatch, subArchitecture) )
return Parser<x86_64>::parse(fileContent, fileLength, path, modTime, ordinal, opts, indirectDylib);
break;
#endif
#if SUPPORT_ARCH_i386
case CPU_TYPE_I386:
if ( Parser<x86>::validFile(fileContent, bundleLoader, subTypeMustMatch, subArchitecture) )
return Parser<x86>::parse(fileContent, fileLength, path, modTime, ordinal, opts, indirectDylib);
break;
#endif
#if SUPPORT_ARCH_arm_any
case CPU_TYPE_ARM:
if ( Parser<arm>::validFile(fileContent, bundleLoader, subTypeMustMatch, subArchitecture) )
return Parser<arm>::parse(fileContent, fileLength, path, modTime, ordinal, opts, indirectDylib);
break;
#endif
#if SUPPORT_ARCH_arm64
case CPU_TYPE_ARM64:
if ( Parser<arm64>::validFile(fileContent, bundleLoader, subTypeMustMatch, subArchitecture) )
return Parser<arm64>::parse(fileContent, fileLength, path, modTime, ordinal, opts, indirectDylib);
break;
#endif
}
return nullptr;
}
ld::dylib::File* parse(const uint8_t* fileContent, uint64_t fileLength, const char* path,
time_t modtime, const Options& opts, ld::File::Ordinal ordinal,
bool bundleLoader, bool indirectDylib)
{
if (!archName(fileContent))
return nullptr;
auto file = parseAsArchitecture(fileContent, fileLength, path, modtime, opts, ordinal, bundleLoader, indirectDylib, opts.architecture(), opts.subArchitecture());
if (!file && opts.fallbackArchitecture()) {
warning("architecture %s not present in dylib file %s, attempting fallback", opts.architectureName(), path);
file = parseAsArchitecture(fileContent, fileLength, path, modtime, opts, ordinal, bundleLoader, indirectDylib, opts.fallbackArchitecture(), opts.fallbackSubArchitecture());
}
return file;
}
}; };