| //===- Writer.cpp ---------------------------------------------------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "Writer.h" |
| #include "Config.h" |
| #include "InputFiles.h" |
| #include "InputSection.h" |
| #include "MergedOutputSection.h" |
| #include "OutputSection.h" |
| #include "OutputSegment.h" |
| #include "SymbolTable.h" |
| #include "Symbols.h" |
| #include "SyntheticSections.h" |
| #include "Target.h" |
| #include "UnwindInfoSection.h" |
| |
| #include "lld/Common/ErrorHandler.h" |
| #include "lld/Common/Memory.h" |
| #include "llvm/BinaryFormat/MachO.h" |
| #include "llvm/Config/llvm-config.h" |
| #include "llvm/Support/LEB128.h" |
| #include "llvm/Support/MD5.h" |
| #include "llvm/Support/MathExtras.h" |
| #include "llvm/Support/Path.h" |
| |
| #include <algorithm> |
| |
| using namespace llvm; |
| using namespace llvm::MachO; |
| using namespace lld; |
| using namespace lld::macho; |
| |
| namespace { |
| class LCUuid; |
| |
| class Writer { |
| public: |
| Writer() : buffer(errorHandler().outputBuffer) {} |
| |
| void scanRelocations(); |
| void createOutputSections(); |
| void createLoadCommands(); |
| void assignAddresses(OutputSegment *); |
| |
| void openFile(); |
| void writeSections(); |
| void writeUuid(); |
| |
| void run(); |
| |
| std::unique_ptr<FileOutputBuffer> &buffer; |
| uint64_t addr = 0; |
| uint64_t fileOff = 0; |
| MachHeaderSection *header = nullptr; |
| StringTableSection *stringTableSection = nullptr; |
| SymtabSection *symtabSection = nullptr; |
| IndirectSymtabSection *indirectSymtabSection = nullptr; |
| UnwindInfoSection *unwindInfoSection = nullptr; |
| LCUuid *uuidCommand = nullptr; |
| }; |
| |
| // LC_DYLD_INFO_ONLY stores the offsets of symbol import/export information. |
| class LCDyldInfo : public LoadCommand { |
| public: |
| LCDyldInfo(RebaseSection *rebaseSection, BindingSection *bindingSection, |
| WeakBindingSection *weakBindingSection, |
| LazyBindingSection *lazyBindingSection, |
| ExportSection *exportSection) |
| : rebaseSection(rebaseSection), bindingSection(bindingSection), |
| weakBindingSection(weakBindingSection), |
| lazyBindingSection(lazyBindingSection), exportSection(exportSection) {} |
| |
| uint32_t getSize() const override { return sizeof(dyld_info_command); } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<dyld_info_command *>(buf); |
| c->cmd = LC_DYLD_INFO_ONLY; |
| c->cmdsize = getSize(); |
| if (rebaseSection->isNeeded()) { |
| c->rebase_off = rebaseSection->fileOff; |
| c->rebase_size = rebaseSection->getFileSize(); |
| } |
| if (bindingSection->isNeeded()) { |
| c->bind_off = bindingSection->fileOff; |
| c->bind_size = bindingSection->getFileSize(); |
| } |
| if (weakBindingSection->isNeeded()) { |
| c->weak_bind_off = weakBindingSection->fileOff; |
| c->weak_bind_size = weakBindingSection->getFileSize(); |
| } |
| if (lazyBindingSection->isNeeded()) { |
| c->lazy_bind_off = lazyBindingSection->fileOff; |
| c->lazy_bind_size = lazyBindingSection->getFileSize(); |
| } |
| if (exportSection->isNeeded()) { |
| c->export_off = exportSection->fileOff; |
| c->export_size = exportSection->getFileSize(); |
| } |
| } |
| |
| RebaseSection *rebaseSection; |
| BindingSection *bindingSection; |
| WeakBindingSection *weakBindingSection; |
| LazyBindingSection *lazyBindingSection; |
| ExportSection *exportSection; |
| }; |
| |
| class LCDysymtab : public LoadCommand { |
| public: |
| LCDysymtab(SymtabSection *symtabSection, |
| IndirectSymtabSection *indirectSymtabSection) |
| : symtabSection(symtabSection), |
| indirectSymtabSection(indirectSymtabSection) {} |
| |
| uint32_t getSize() const override { return sizeof(dysymtab_command); } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<dysymtab_command *>(buf); |
| c->cmd = LC_DYSYMTAB; |
| c->cmdsize = getSize(); |
| |
| c->ilocalsym = 0; |
| c->iextdefsym = c->nlocalsym = symtabSection->getNumLocalSymbols(); |
| c->nextdefsym = symtabSection->getNumExternalSymbols(); |
| c->iundefsym = c->iextdefsym + c->nextdefsym; |
| c->nundefsym = symtabSection->getNumUndefinedSymbols(); |
| |
| c->indirectsymoff = indirectSymtabSection->fileOff; |
| c->nindirectsyms = indirectSymtabSection->getNumSymbols(); |
| } |
| |
| SymtabSection *symtabSection; |
| IndirectSymtabSection *indirectSymtabSection; |
| }; |
| |
| class LCSegment : public LoadCommand { |
| public: |
| LCSegment(StringRef name, OutputSegment *seg) : name(name), seg(seg) {} |
| |
| uint32_t getSize() const override { |
| return sizeof(segment_command_64) + |
| seg->numNonHiddenSections() * sizeof(section_64); |
| } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<segment_command_64 *>(buf); |
| buf += sizeof(segment_command_64); |
| |
| c->cmd = LC_SEGMENT_64; |
| c->cmdsize = getSize(); |
| memcpy(c->segname, name.data(), name.size()); |
| c->fileoff = seg->fileOff; |
| c->maxprot = seg->maxProt; |
| c->initprot = seg->initProt; |
| |
| if (seg->getSections().empty()) |
| return; |
| |
| c->vmaddr = seg->firstSection()->addr; |
| c->vmsize = |
| seg->lastSection()->addr + seg->lastSection()->getSize() - c->vmaddr; |
| c->nsects = seg->numNonHiddenSections(); |
| |
| for (OutputSection *osec : seg->getSections()) { |
| if (!isZeroFill(osec->flags)) { |
| assert(osec->fileOff >= seg->fileOff); |
| c->filesize = std::max( |
| c->filesize, osec->fileOff + osec->getFileSize() - seg->fileOff); |
| } |
| |
| if (osec->isHidden()) |
| continue; |
| |
| auto *sectHdr = reinterpret_cast<section_64 *>(buf); |
| buf += sizeof(section_64); |
| |
| memcpy(sectHdr->sectname, osec->name.data(), osec->name.size()); |
| memcpy(sectHdr->segname, name.data(), name.size()); |
| |
| sectHdr->addr = osec->addr; |
| sectHdr->offset = osec->fileOff; |
| sectHdr->align = Log2_32(osec->align); |
| sectHdr->flags = osec->flags; |
| sectHdr->size = osec->getSize(); |
| sectHdr->reserved1 = osec->reserved1; |
| sectHdr->reserved2 = osec->reserved2; |
| } |
| } |
| |
| private: |
| StringRef name; |
| OutputSegment *seg; |
| }; |
| |
| class LCMain : public LoadCommand { |
| uint32_t getSize() const override { return sizeof(entry_point_command); } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<entry_point_command *>(buf); |
| c->cmd = LC_MAIN; |
| c->cmdsize = getSize(); |
| |
| if (config->entry->isInStubs()) |
| c->entryoff = |
| in.stubs->fileOff + config->entry->stubsIndex * target->stubSize; |
| else |
| c->entryoff = config->entry->getFileOffset(); |
| |
| c->stacksize = 0; |
| } |
| }; |
| |
| class LCSymtab : public LoadCommand { |
| public: |
| LCSymtab(SymtabSection *symtabSection, StringTableSection *stringTableSection) |
| : symtabSection(symtabSection), stringTableSection(stringTableSection) {} |
| |
| uint32_t getSize() const override { return sizeof(symtab_command); } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<symtab_command *>(buf); |
| c->cmd = LC_SYMTAB; |
| c->cmdsize = getSize(); |
| c->symoff = symtabSection->fileOff; |
| c->nsyms = symtabSection->getNumSymbols(); |
| c->stroff = stringTableSection->fileOff; |
| c->strsize = stringTableSection->getFileSize(); |
| } |
| |
| SymtabSection *symtabSection = nullptr; |
| StringTableSection *stringTableSection = nullptr; |
| }; |
| |
| // There are several dylib load commands that share the same structure: |
| // * LC_LOAD_DYLIB |
| // * LC_ID_DYLIB |
| // * LC_REEXPORT_DYLIB |
| class LCDylib : public LoadCommand { |
| public: |
| LCDylib(LoadCommandType type, StringRef path) : type(type), path(path) { |
| instanceCount++; |
| } |
| |
| uint32_t getSize() const override { |
| return alignTo(sizeof(dylib_command) + path.size() + 1, 8); |
| } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<dylib_command *>(buf); |
| buf += sizeof(dylib_command); |
| |
| c->cmd = type; |
| c->cmdsize = getSize(); |
| c->dylib.name = sizeof(dylib_command); |
| |
| memcpy(buf, path.data(), path.size()); |
| buf[path.size()] = '\0'; |
| } |
| |
| static uint32_t getInstanceCount() { return instanceCount; } |
| |
| private: |
| LoadCommandType type; |
| StringRef path; |
| static uint32_t instanceCount; |
| }; |
| |
| uint32_t LCDylib::instanceCount = 0; |
| |
| class LCLoadDylinker : public LoadCommand { |
| public: |
| uint32_t getSize() const override { |
| return alignTo(sizeof(dylinker_command) + path.size() + 1, 8); |
| } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<dylinker_command *>(buf); |
| buf += sizeof(dylinker_command); |
| |
| c->cmd = LC_LOAD_DYLINKER; |
| c->cmdsize = getSize(); |
| c->name = sizeof(dylinker_command); |
| |
| memcpy(buf, path.data(), path.size()); |
| buf[path.size()] = '\0'; |
| } |
| |
| private: |
| // Recent versions of Darwin won't run any binary that has dyld at a |
| // different location. |
| const StringRef path = "/usr/lib/dyld"; |
| }; |
| |
| class LCRPath : public LoadCommand { |
| public: |
| LCRPath(StringRef path) : path(path) {} |
| |
| uint32_t getSize() const override { |
| return alignTo(sizeof(rpath_command) + path.size() + 1, WordSize); |
| } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<rpath_command *>(buf); |
| buf += sizeof(rpath_command); |
| |
| c->cmd = LC_RPATH; |
| c->cmdsize = getSize(); |
| c->path = sizeof(rpath_command); |
| |
| memcpy(buf, path.data(), path.size()); |
| buf[path.size()] = '\0'; |
| } |
| |
| private: |
| StringRef path; |
| }; |
| |
| class LCBuildVersion : public LoadCommand { |
| public: |
| LCBuildVersion(const PlatformInfo &platform) : platform(platform) {} |
| |
| const int ntools = 1; |
| |
| uint32_t getSize() const override { |
| return sizeof(build_version_command) + ntools * sizeof(build_tool_version); |
| } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<build_version_command *>(buf); |
| c->cmd = LC_BUILD_VERSION; |
| c->cmdsize = getSize(); |
| c->platform = static_cast<uint32_t>(platform.kind); |
| c->minos = ((platform.minimum.getMajor() << 020) | |
| (platform.minimum.getMinor().getValueOr(0) << 010) | |
| platform.minimum.getSubminor().getValueOr(0)); |
| c->sdk = ((platform.sdk.getMajor() << 020) | |
| (platform.sdk.getMinor().getValueOr(0) << 010) | |
| platform.sdk.getSubminor().getValueOr(0)); |
| c->ntools = ntools; |
| auto *t = reinterpret_cast<build_tool_version *>(&c[1]); |
| t->tool = TOOL_LD; |
| t->version = (LLVM_VERSION_MAJOR << 020) | (LLVM_VERSION_MINOR << 010) | |
| LLVM_VERSION_PATCH; |
| } |
| |
| const PlatformInfo &platform; |
| }; |
| |
| // Stores a unique identifier for the output file based on an MD5 hash of its |
| // contents. In order to hash the contents, we must first write them, but |
| // LC_UUID itself must be part of the written contents in order for all the |
| // offsets to be calculated correctly. We resolve this circular paradox by |
| // first writing an LC_UUID with an all-zero UUID, then updating the UUID with |
| // its real value later. |
| class LCUuid : public LoadCommand { |
| public: |
| uint32_t getSize() const override { return sizeof(uuid_command); } |
| |
| void writeTo(uint8_t *buf) const override { |
| auto *c = reinterpret_cast<uuid_command *>(buf); |
| c->cmd = LC_UUID; |
| c->cmdsize = getSize(); |
| uuidBuf = c->uuid; |
| } |
| |
| void writeUuid(const std::array<uint8_t, 16> &uuid) const { |
| memcpy(uuidBuf, uuid.data(), uuid.size()); |
| } |
| |
| mutable uint8_t *uuidBuf; |
| }; |
| |
| } // namespace |
| |
| void Writer::scanRelocations() { |
| for (InputSection *isec : inputSections) { |
| // We do not wish to add rebase opcodes for __LD,__compact_unwind, because |
| // it doesn't actually end up in the final binary. TODO: filtering it out |
| // before Writer runs might be cleaner... |
| if (isec->segname == segment_names::ld) |
| continue; |
| |
| for (Reloc &r : isec->relocs) { |
| if (auto *s = r.referent.dyn_cast<lld::macho::Symbol *>()) { |
| if (isa<Undefined>(s)) |
| error("undefined symbol " + toString(*s) + ", referenced from " + |
| toString(isec->file)); |
| else |
| target->prepareSymbolRelocation(s, isec, r); |
| } else { |
| assert(r.referent.is<InputSection *>()); |
| if (!r.pcrel) |
| in.rebase->addEntry(isec, r.offset); |
| } |
| } |
| } |
| } |
| |
| void Writer::createLoadCommands() { |
| in.header->addLoadCommand(make<LCDyldInfo>( |
| in.rebase, in.binding, in.weakBinding, in.lazyBinding, in.exports)); |
| in.header->addLoadCommand(make<LCSymtab>(symtabSection, stringTableSection)); |
| in.header->addLoadCommand( |
| make<LCDysymtab>(symtabSection, indirectSymtabSection)); |
| for (StringRef path : config->runtimePaths) |
| in.header->addLoadCommand(make<LCRPath>(path)); |
| |
| switch (config->outputType) { |
| case MH_EXECUTE: |
| in.header->addLoadCommand(make<LCMain>()); |
| in.header->addLoadCommand(make<LCLoadDylinker>()); |
| break; |
| case MH_DYLIB: |
| in.header->addLoadCommand(make<LCDylib>(LC_ID_DYLIB, config->installName)); |
| break; |
| case MH_BUNDLE: |
| break; |
| default: |
| llvm_unreachable("unhandled output file type"); |
| } |
| |
| in.header->addLoadCommand(make<LCBuildVersion>(config->platform)); |
| |
| uuidCommand = make<LCUuid>(); |
| in.header->addLoadCommand(uuidCommand); |
| |
| uint8_t segIndex = 0; |
| for (OutputSegment *seg : outputSegments) { |
| in.header->addLoadCommand(make<LCSegment>(seg->name, seg)); |
| seg->index = segIndex++; |
| } |
| |
| uint64_t dylibOrdinal = 1; |
| for (InputFile *file : inputFiles) { |
| if (auto *dylibFile = dyn_cast<DylibFile>(file)) { |
| // TODO: dylibs that are only referenced by weak refs should also be |
| // loaded via LC_LOAD_WEAK_DYLIB. |
| LoadCommandType lcType = |
| dylibFile->forceWeakImport ? LC_LOAD_WEAK_DYLIB : LC_LOAD_DYLIB; |
| in.header->addLoadCommand(make<LCDylib>(lcType, dylibFile->dylibName)); |
| dylibFile->ordinal = dylibOrdinal++; |
| |
| if (dylibFile->reexport) |
| in.header->addLoadCommand( |
| make<LCDylib>(LC_REEXPORT_DYLIB, dylibFile->dylibName)); |
| } |
| } |
| |
| const uint32_t MACOS_MAXPATHLEN = 1024; |
| config->headerPad = std::max( |
| config->headerPad, (config->headerPadMaxInstallNames |
| ? LCDylib::getInstanceCount() * MACOS_MAXPATHLEN |
| : 0)); |
| } |
| |
| static size_t getSymbolPriority(const SymbolPriorityEntry &entry, |
| const InputFile &file) { |
| return std::max(entry.objectFiles.lookup(sys::path::filename(file.getName())), |
| entry.anyObjectFile); |
| } |
| |
| // Each section gets assigned the priority of the highest-priority symbol it |
| // contains. |
| static DenseMap<const InputSection *, size_t> buildInputSectionPriorities() { |
| DenseMap<const InputSection *, size_t> sectionPriorities; |
| |
| if (config->priorities.empty()) |
| return sectionPriorities; |
| |
| auto addSym = [&](Defined &sym) { |
| auto it = config->priorities.find(sym.getName()); |
| if (it == config->priorities.end()) |
| return; |
| |
| SymbolPriorityEntry &entry = it->second; |
| size_t &priority = sectionPriorities[sym.isec]; |
| priority = std::max(priority, getSymbolPriority(entry, *sym.isec->file)); |
| }; |
| |
| // TODO: Make sure this handles weak symbols correctly. |
| for (InputFile *file : inputFiles) |
| if (isa<ObjFile>(file) || isa<ArchiveFile>(file)) |
| for (lld::macho::Symbol *sym : file->symbols) |
| if (auto *d = dyn_cast<Defined>(sym)) |
| addSym(*d); |
| |
| return sectionPriorities; |
| } |
| |
| static int segmentOrder(OutputSegment *seg) { |
| return StringSwitch<int>(seg->name) |
| .Case(segment_names::pageZero, -2) |
| .Case(segment_names::text, -1) |
| // Make sure __LINKEDIT is the last segment (i.e. all its hidden |
| // sections must be ordered after other sections). |
| .Case(segment_names::linkEdit, std::numeric_limits<int>::max()) |
| .Default(0); |
| } |
| |
| static int sectionOrder(OutputSection *osec) { |
| StringRef segname = osec->parent->name; |
| // Sections are uniquely identified by their segment + section name. |
| if (segname == segment_names::text) { |
| return StringSwitch<int>(osec->name) |
| .Case(section_names::header, -1) |
| .Case(section_names::unwindInfo, std::numeric_limits<int>::max() - 1) |
| .Case(section_names::ehFrame, std::numeric_limits<int>::max()) |
| .Default(0); |
| } else if (segname == segment_names::linkEdit) { |
| return StringSwitch<int>(osec->name) |
| .Case(section_names::rebase, -8) |
| .Case(section_names::binding, -7) |
| .Case(section_names::weakBinding, -6) |
| .Case(section_names::lazyBinding, -5) |
| .Case(section_names::export_, -4) |
| .Case(section_names::symbolTable, -3) |
| .Case(section_names::indirectSymbolTable, -2) |
| .Case(section_names::stringTable, -1) |
| .Default(0); |
| } |
| // ZeroFill sections must always be the at the end of their segments, |
| // otherwise subsequent sections may get overwritten with zeroes at runtime. |
| if (isZeroFill(osec->flags)) |
| return std::numeric_limits<int>::max(); |
| return 0; |
| } |
| |
| template <typename T, typename F> |
| static std::function<bool(T, T)> compareByOrder(F ord) { |
| return [=](T a, T b) { return ord(a) < ord(b); }; |
| } |
| |
| // Sorting only can happen once all outputs have been collected. Here we sort |
| // segments, output sections within each segment, and input sections within each |
| // output segment. |
| static void sortSegmentsAndSections() { |
| llvm::stable_sort(outputSegments, |
| compareByOrder<OutputSegment *>(segmentOrder)); |
| |
| DenseMap<const InputSection *, size_t> isecPriorities = |
| buildInputSectionPriorities(); |
| |
| uint32_t sectionIndex = 0; |
| for (OutputSegment *seg : outputSegments) { |
| seg->sortOutputSections(compareByOrder<OutputSection *>(sectionOrder)); |
| for (auto *osec : seg->getSections()) { |
| // Now that the output sections are sorted, assign the final |
| // output section indices. |
| if (!osec->isHidden()) |
| osec->index = ++sectionIndex; |
| |
| if (!isecPriorities.empty()) { |
| if (auto *merged = dyn_cast<MergedOutputSection>(osec)) { |
| llvm::stable_sort(merged->inputs, |
| [&](InputSection *a, InputSection *b) { |
| return isecPriorities[a] > isecPriorities[b]; |
| }); |
| } |
| } |
| } |
| } |
| } |
| |
| void Writer::createOutputSections() { |
| // First, create hidden sections |
| stringTableSection = make<StringTableSection>(); |
| unwindInfoSection = make<UnwindInfoSection>(); // TODO(gkm): only when no -r |
| symtabSection = make<SymtabSection>(*stringTableSection); |
| indirectSymtabSection = make<IndirectSymtabSection>(); |
| |
| switch (config->outputType) { |
| case MH_EXECUTE: |
| make<PageZeroSection>(); |
| break; |
| case MH_DYLIB: |
| case MH_BUNDLE: |
| break; |
| default: |
| llvm_unreachable("unhandled output file type"); |
| } |
| |
| // Then merge input sections into output sections. |
| MapVector<std::pair<StringRef, StringRef>, MergedOutputSection *> |
| mergedOutputSections; |
| for (InputSection *isec : inputSections) { |
| // Instead of emitting DWARF sections, we emit STABS symbols to the object |
| // files that contain them. |
| if (isDebugSection(isec->flags) && isec->segname == segment_names::dwarf) |
| continue; |
| MergedOutputSection *&osec = |
| mergedOutputSections[{isec->segname, isec->name}]; |
| if (osec == nullptr) |
| osec = make<MergedOutputSection>(isec->name); |
| osec->mergeInput(isec); |
| } |
| |
| for (const auto &it : mergedOutputSections) { |
| StringRef segname = it.first.first; |
| MergedOutputSection *osec = it.second; |
| if (unwindInfoSection && segname == segment_names::ld) { |
| assert(osec->name == section_names::compactUnwind); |
| unwindInfoSection->setCompactUnwindSection(osec); |
| } else { |
| getOrCreateOutputSegment(segname)->addOutputSection(osec); |
| } |
| } |
| |
| for (SyntheticSection *ssec : syntheticSections) { |
| auto it = mergedOutputSections.find({ssec->segname, ssec->name}); |
| if (it == mergedOutputSections.end()) { |
| if (ssec->isNeeded()) |
| getOrCreateOutputSegment(ssec->segname)->addOutputSection(ssec); |
| } else { |
| error("section from " + toString(it->second->firstSection()->file) + |
| " conflicts with synthetic section " + ssec->segname + "," + |
| ssec->name); |
| } |
| } |
| } |
| |
| void Writer::assignAddresses(OutputSegment *seg) { |
| addr = alignTo(addr, PageSize); |
| fileOff = alignTo(fileOff, PageSize); |
| seg->fileOff = fileOff; |
| |
| for (auto *osec : seg->getSections()) { |
| if (!osec->isNeeded()) |
| continue; |
| addr = alignTo(addr, osec->align); |
| fileOff = alignTo(fileOff, osec->align); |
| osec->addr = addr; |
| osec->fileOff = isZeroFill(osec->flags) ? 0 : fileOff; |
| osec->finalize(); |
| |
| addr += osec->getSize(); |
| fileOff += osec->getFileSize(); |
| } |
| } |
| |
| void Writer::openFile() { |
| Expected<std::unique_ptr<FileOutputBuffer>> bufferOrErr = |
| FileOutputBuffer::create(config->outputFile, fileOff, |
| FileOutputBuffer::F_executable); |
| |
| if (!bufferOrErr) |
| error("failed to open " + config->outputFile + ": " + |
| llvm::toString(bufferOrErr.takeError())); |
| else |
| buffer = std::move(*bufferOrErr); |
| } |
| |
| void Writer::writeSections() { |
| uint8_t *buf = buffer->getBufferStart(); |
| for (OutputSegment *seg : outputSegments) |
| for (OutputSection *osec : seg->getSections()) |
| osec->writeTo(buf + osec->fileOff); |
| } |
| |
| void Writer::writeUuid() { |
| MD5 hash; |
| const auto *bufStart = reinterpret_cast<char *>(buffer->getBufferStart()); |
| const auto *bufEnd = reinterpret_cast<char *>(buffer->getBufferEnd()); |
| hash.update(StringRef(bufStart, bufEnd - bufStart)); |
| MD5::MD5Result result; |
| hash.final(result); |
| // Conform to UUID version 4 & 5 as specified in RFC 4122: |
| // 1. Set the version field to indicate that this is an MD5-based UUID. |
| result.Bytes[6] = (result.Bytes[6] & 0xf) | 0x30; |
| // 2. Set the two MSBs of uuid_t::clock_seq_hi_and_reserved to zero and one. |
| result.Bytes[8] = (result.Bytes[8] & 0x3f) | 0x80; |
| uuidCommand->writeUuid(result.Bytes); |
| } |
| |
| void Writer::run() { |
| // dyld requires __LINKEDIT segment to always exist (even if empty). |
| OutputSegment *linkEditSegment = |
| getOrCreateOutputSegment(segment_names::linkEdit); |
| |
| prepareBranchTarget(config->entry); |
| scanRelocations(); |
| if (in.stubHelper->isNeeded()) |
| in.stubHelper->setup(); |
| |
| for (const macho::Symbol *sym : symtab->getSymbols()) |
| if (const auto *defined = dyn_cast<Defined>(sym)) |
| if (defined->overridesWeakDef) |
| in.weakBinding->addNonWeakDefinition(defined); |
| |
| // Sort and assign sections to their respective segments. No more sections nor |
| // segments may be created after these methods run. |
| createOutputSections(); |
| sortSegmentsAndSections(); |
| |
| createLoadCommands(); |
| |
| // Ensure that segments (and the sections they contain) are allocated |
| // addresses in ascending order, which dyld requires. |
| // |
| // Note that at this point, __LINKEDIT sections are empty, but we need to |
| // determine addresses of other segments/sections before generating its |
| // contents. |
| for (OutputSegment *seg : outputSegments) |
| if (seg != linkEditSegment) |
| assignAddresses(seg); |
| |
| // Fill __LINKEDIT contents. |
| in.rebase->finalizeContents(); |
| in.binding->finalizeContents(); |
| in.weakBinding->finalizeContents(); |
| in.lazyBinding->finalizeContents(); |
| in.exports->finalizeContents(); |
| symtabSection->finalizeContents(); |
| indirectSymtabSection->finalizeContents(); |
| |
| // Now that __LINKEDIT is filled out, do a proper calculation of its |
| // addresses and offsets. |
| assignAddresses(linkEditSegment); |
| |
| openFile(); |
| if (errorCount()) |
| return; |
| |
| writeSections(); |
| writeUuid(); |
| |
| if (auto e = buffer->commit()) |
| error("failed to write to the output file: " + toString(std::move(e))); |
| } |
| |
| void macho::writeResult() { Writer().run(); } |
| |
| void macho::createSyntheticSections() { |
| in.header = make<MachHeaderSection>(); |
| in.rebase = make<RebaseSection>(); |
| in.binding = make<BindingSection>(); |
| in.weakBinding = make<WeakBindingSection>(); |
| in.lazyBinding = make<LazyBindingSection>(); |
| in.exports = make<ExportSection>(); |
| in.got = make<GotSection>(); |
| in.tlvPointers = make<TlvPointerSection>(); |
| in.lazyPointers = make<LazyPointerSection>(); |
| in.stubs = make<StubsSection>(); |
| in.stubHelper = make<StubHelperSection>(); |
| in.imageLoaderCache = make<ImageLoaderCacheSection>(); |
| } |