Index: tools/llvm-objcopy/Object.h =================================================================== --- tools/llvm-objcopy/Object.h +++ tools/llvm-objcopy/Object.h @@ -126,6 +126,20 @@ void writeSection(FileOutputBuffer &Out) const override; }; +class RawDataSection : public SectionBase { +private: + std::vector Data; + +public: + RawDataSection(StringRef SecName, ArrayRef Data) + : Data(std::begin(Data), std::end(Data)) { + Name = SecName; + Type = ELF::SHT_PROGBITS; + Size = Data.size(); + } + void writeSection(FileOutputBuffer &Out) const override; +}; + // There are two types of string tables that can exist, dynamic and not dynamic. // In the dynamic case the string table is allocated. Changing a dynamic string // table would mean altering virtual addresses and thus the memory image. So @@ -372,6 +386,7 @@ const SymbolTableSection *getSymTab() const { return SymbolTable; } const SectionBase *getSectionHeaderStrTab() const { return SectionNames; } void removeSections(std::function ToRemove); + void addSection(StringRef SecName, ArrayRef Data); virtual size_t totalSize() const = 0; virtual void finalize() = 0; virtual void write(FileOutputBuffer &Out) const = 0; Index: tools/llvm-objcopy/Object.cpp =================================================================== --- tools/llvm-objcopy/Object.cpp +++ tools/llvm-objcopy/Object.cpp @@ -81,6 +81,11 @@ std::copy(std::begin(Contents), std::end(Contents), Buf); } +void RawDataSection::writeSection(FileOutputBuffer &Out) const { + uint8_t *Buf = Out.getBufferStart() + Offset; + std::copy(std::begin(Data), std::end(Data), Buf); +} + void StringTableSection::addString(StringRef Name) { StrTabBuilder.add(Name); Size = StrTabBuilder.getSize(); @@ -678,6 +683,13 @@ Sections.erase(Iter, std::end(Sections)); } +template +void Object::addSection(StringRef SecName, ArrayRef Data) { + auto Sec = llvm::make_unique(SecName, Data); + Sec->OriginalOffset = ~0ULL; + Sections.push_back(std::move(Sec)); +} + template void ELFObject::sortSections() { // Put all sections in offset order. Maintain the ordering as closely as // possible while meeting that demand however. Index: tools/llvm-objcopy/llvm-objcopy.cpp =================================================================== --- tools/llvm-objcopy/llvm-objcopy.cpp +++ tools/llvm-objcopy/llvm-objcopy.cpp @@ -112,6 +112,9 @@ cl::desc("Equivalent to extract-dwo on the input file to " ", then strip-dwo on the input file"), cl::value_desc("dwo-file")); +static cl::list AddSection("add-section", + cl::desc("Make a section named
with the contents of ."), + cl::value_desc("section=file")); using SectionPred = std::function; @@ -178,6 +181,21 @@ if (!SplitDWO.empty()) SplitDWOToFile(ObjFile, SplitDWO.getValue()); + if (!AddSection.empty()) { + for (const auto &Flag : AddSection) { + auto SecPair = StringRef(Flag).split("="); + auto SecName = SecPair.first; + auto File = SecPair.second; + auto BufOrErr = MemoryBuffer::getFile(File); + if (!BufOrErr) + reportError(File, BufOrErr.getError()); + auto Buf = std::move(*BufOrErr); + auto BufPtr = reinterpret_cast(Buf->getBufferStart()); + auto BufSize = Buf->getBufferSize(); + Obj->addSection(SecName, ArrayRef(BufPtr, BufSize)); + } + } + SectionPred RemovePred = [](const SectionBase &) { return false; }; // Removes: