48 "unsupported load command encountered");
56 SectionPred RemovePred = [](
const std::unique_ptr<Section> &) {
61 RemovePred = [&Config, RemovePred](
const std::unique_ptr<Section> &Sec) {
68 RemovePred = [RemovePred](
const std::unique_ptr<Section> &Sec) {
69 if (Sec->Segname ==
"__DWARF")
72 return RemovePred(Sec);
78 RemovePred = [&Config](
const std::unique_ptr<Section> &Sec) {
90 (*ISE.
Symbol)->Referenced =
true;
99 Sym.
Name = std::string(
I->getValue());
103 &Obj](
const std::unique_ptr<SymbolEntry> &
N) {
125template <
typename LCType>
128 "unsupported load command encountered");
133 LC.
Payload.assign(NewCmdsize -
sizeof(LCType), 0);
140 RPathLC.
cmd = MachO::LC_RPATH;
145 std::copy(Path.begin(), Path.end(), LC.
Payload.begin());
156 if (LC.MachOLoadCommand.load_command_data.cmd == MachO::LC_RPATH) {
179 "no LC_RPATH load command with path: %s",
180 RPath.str().c_str());
197 "no LC_RPATH load command with path: " + Old);
201 "' would create a duplicate load command");
207 case MachO::LC_ID_DYLIB:
209 updateLoadCommandPayloadString<MachO::dylib_command>(
213 case MachO::LC_RPATH: {
216 if (!NewRPath.
empty())
217 updateLoadCommandPayloadString<MachO::rpath_command>(LC, NewRPath);
223 case MachO::LC_LOAD_DYLIB:
224 case MachO::LC_LOAD_WEAK_DYLIB:
228 if (!NewInstallName.
empty())
229 updateLoadCommandPayloadString<MachO::dylib_command>(LC,
240 "' would create a duplicate load command");
249 "' would create a duplicate load command");
281 for (
const std::unique_ptr<Section> &Sec : LC.
Sections) {
282 if (Sec->CanonicalName == SecName) {
287 std::unique_ptr<FileOutputBuffer> Buf = std::move(*BufferOrErr);
288 llvm::copy(Sec->Content, Buf->getBufferStart());
290 if (
Error E = Buf->commit())
297 SecName.
str().c_str());
303 Section Sec(TargetSegName, Pair.second);
311 if (SegName && SegName == TargetSegName) {
313 for (
const std::unique_ptr<Section> &S : LC.
Sections)
314 Addr = std::max(
Addr, S->Addr + S->Size);
315 LC.
Sections.push_back(std::make_unique<Section>(Sec));
325 NewSegment.
Sections.push_back(std::make_unique<Section>(Sec));
332 std::tie(SegName, SecName) = SecName.
split(
",");
335 return LC.getSegmentName() == SegName;
337 if (FoundSeg == O.LoadCommands.end())
339 "could not find segment with name '%s'",
340 SegName.
str().c_str());
342 [SecName](
const std::unique_ptr<Section> &Sec) {
343 return Sec->Sectname == SecName;
345 if (FoundSec == FoundSeg->Sections.end())
347 "could not find section with name '%s'",
348 SecName.
str().c_str());
350 assert(FoundSec->get()->CanonicalName == (SegName +
"," + SecName).str());
357 if (!SecToUpdateOrErr)
359 Section &Sec = *SecToUpdateOrErr;
363 errc::invalid_argument,
364 "new section cannot be larger than previous section");
374 if (
Name.count(
',') != 1)
376 "invalid section name '%s' (should be formatted "
377 "as '<segment name>,<section name>')",
380 std::pair<StringRef, StringRef> Pair =
Name.split(
',');
381 if (Pair.first.size() > 16)
383 "too long segment name: '%s'",
384 Pair.first.str().c_str());
385 if (Pair.second.size() > 16)
387 "too long section name: '%s'",
388 Pair.second.str().c_str());
414 for (std::unique_ptr<Section> &Sec : LC.
Sections)
415 Sec->Relocations.clear();
446 if (O->get()->Header.FileType == MachO::HeaderFileType::MH_PRELOAD)
448 "%s: MH_PRELOAD files are not supported",
457 switch (In.getArch()) {
458 case Triple::ArchType::arm:
459 case Triple::ArchType::aarch64:
460 case Triple::ArchType::aarch64_32:
467 MachOWriter Writer(**O, In.is64Bit(), In.isLittleEndian(),
471 return Writer.
write();
479 for (
const auto &O : In.objects()) {
484 if (!NewArchiveMembersOrErr)
485 return NewArchiveMembersOrErr.
takeError();
486 auto Kind = (*ArOrErr)->kind();
491 (*ArOrErr)->hasSymbolTable(),
Kind,
493 (*ArOrErr)->isThin());
494 if (!OutputBufferOrErr)
501 std::move(*OutputBufferOrErr));
503 O.getCPUType(), O.getCPUSubType(),
504 O.getArchFlagName(), O.getAlign());
517 std::errc::invalid_argument,
518 "slice for '%s' of the universal Mach-O binary "
519 "'%s' is not a Mach-O object or an archive",
520 O.getArchFlagName().c_str(),
523 std::string ArchFlagName = O.getArchFlagName();
533 **ObjOrErr, MemStream))
536 auto MB = std::make_unique<SmallVectorMemoryBuffer>(
537 std::move(Buffer), ArchFlagName,
false);
541 Binaries.
emplace_back(std::move(*BinaryOrErr), std::move(MB));
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
This file defines the DenseSet and SmallDenseSet classes.
std::function< bool(const SectionBase &Sec)> SectionPred
static cl::opt< int > PageSize("imp-null-check-page-size", cl::desc("The page size of the target in bytes"), cl::init(4096), cl::Hidden)
static Error processLoadCommands(const MachOConfig &MachOConfig, Object &Obj)
static Expected< Section & > findSection(StringRef SecName, Object &O)
static Error isValidMachOCannonicalName(StringRef Name)
static void updateLoadCommandPayloadString(LoadCommand &LC, StringRef S)
static LoadCommand buildRPathLoadCommand(StringRef Path)
static bool isLoadCommandWithPayloadString(const LoadCommand &LC)
static void markSymbols(const CommonConfig &, Object &Obj)
static Error handleArgs(const CommonConfig &Config, const MachOConfig &MachOConfig, Object &Obj)
static Error removeSections(const CommonConfig &Config, Object &Obj)
std::function< bool(const LoadCommand &LC)> LoadCommandPred
static Error addSection(const NewSectionInfo &NewSection, Object &Obj)
static Error dumpSectionToFile(StringRef SecName, StringRef Filename, Object &Obj)
static StringRef getPayloadString(const LoadCommand &LC)
static void updateAndRemoveSymbols(const CommonConfig &Config, const MachOConfig &MachOConfig, Object &Obj)
static Error updateSection(const NewSectionInfo &NewSection, Object &O)
assert(ImpDefSCC.getReg()==AMDGPU::SCC &&ImpDefSCC.isDef())
Implements a dense probed hash-table based set.
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.
static Expected< std::unique_ptr< FileOutputBuffer > > create(StringRef FilePath, size_t Size, unsigned Flags=0)
Factory method to create an OutputBuffer object which manages a read/write buffer of the specified si...
reference emplace_back(ArgTypes &&... Args)
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
StringRef - Represent a constant reference to a string, i.e.
std::pair< StringRef, StringRef > split(char Separator) const
Split into two substrings around the first occurrence of a separator character.
std::string str() const
str - Get the contents as an std::string.
constexpr bool empty() const
empty - Check if the string is empty.
constexpr size_t size() const
size - Get the string size.
StringRef rtrim(char Char) const
Return string with consecutive Char characters starting from the right removed.
StringRef save(const char *S)
std::pair< iterator, bool > insert(const ValueT &V)
bool contains(const_arg_type_t< ValueT > V) const
Check if the set contains the given element.
bool erase(const ValueT &V)
size_type count(const_arg_type_t< ValueT > V) const
Return 1 if the specified key is in the set, 0 otherwise.
bool matches(StringRef S) const
virtual Expected< std::unique_ptr< Object > > create() const =0
This class implements an extremely fast bulk output stream that can only output to a stream.
A raw_ostream that writes to an SmallVector or SmallString.
Error executeObjcopyOnBinary(const CommonConfig &Config, const MachOConfig &MachOConfig, object::MachOObjectFile &In, raw_ostream &Out)
Apply the transformations described by Config and MachOConfig to In and writes the result into Out.
Error executeObjcopyOnMachOUniversalBinary(const MultiFormatConfig &Config, const object::MachOUniversalBinary &In, raw_ostream &Out)
Apply the transformations described by Config and MachOConfig to In and writes the result into Out.
Expected< std::vector< NewArchiveMember > > createNewArchiveMembers(const MultiFormatConfig &Config, const Archive &Ar)
Applies the transformations described by Config to each member in archive Ar.
Expected< std::unique_ptr< Binary > > createBinary(MemoryBufferRef Source, LLVMContext *Context=nullptr, bool InitContent=true)
Create a Binary from Source, autodetecting the file type.
Error writeUniversalBinaryToStream(ArrayRef< Slice > Slices, raw_ostream &Out)
StringRef filename(StringRef path, Style style=Style::native)
Get filename.
This is an optimization pass for GlobalISel generic memory operations.
Error createFileError(const Twine &F, Error E)
Concatenate a source file path and/or name with an Error.
Expected< std::unique_ptr< MemoryBuffer > > writeArchiveToBuffer(ArrayRef< NewArchiveMember > NewMembers, bool WriteSymtab, object::Archive::Kind Kind, bool Deterministic, bool Thin)
Error createStringError(std::error_code EC, char const *Fmt, const Ts &... Vals)
Create formatted StringError object.
uint64_t alignTo(uint64_t Size, Align A)
Returns a multiple of A needed to store Size bytes.
OutputIt copy(R &&Range, OutputIt Out)
auto find_if(R &&Range, UnaryPredicate P)
Provide wrappers to std::find_if which take ranges instead of having to pass begin/end explicitly.
void consumeError(Error Err)
Consume a Error without doing anything.
std::vector< StringRef > DumpSection
std::vector< NewSectionInfo > UpdateSection
bool DeterministicArchives
std::vector< NewSectionInfo > AddSection
StringMap< StringRef > SymbolsToRename
std::vector< StringRef > RPathToPrepend
DenseMap< StringRef, StringRef > InstallNamesToUpdate
std::optional< StringRef > SharedLibId
DenseSet< StringRef > EmptySegmentsToRemove
DenseSet< StringRef > RPathsToRemove
DenseMap< StringRef, StringRef > RPathsToUpdate
std::vector< StringRef > RPathToAdd
std::shared_ptr< MemoryBuffer > SectionData
std::optional< SymbolEntry * > Symbol
The Symbol referenced by this entry.
std::vector< IndirectSymbolEntry > Symbols
MachO::macho_load_command MachOLoadCommand
std::optional< StringRef > getSegmentName() const
std::vector< std::unique_ptr< Section > > Sections
std::optional< uint64_t > getSegmentVMAddr() const
std::vector< uint8_t > Payload
void updateLoadCommandIndexes()
StringSaver NewSectionsContents
Error removeLoadCommands(function_ref< bool(const LoadCommand &)> ToRemove)
Error removeSections(function_ref< bool(const std::unique_ptr< Section > &)> ToRemove)
std::vector< LoadCommand > LoadCommands
std::optional< uint32_t > SwiftVersion
IndirectSymbolTable IndirectSymTable
LoadCommand & addSegment(StringRef SegName, uint64_t SegVMSize)
Creates a new segment load command in the object and returns a reference to the newly created load co...
void removeSymbols(function_ref< bool(const std::unique_ptr< SymbolEntry > &)> ToRemove)