blob: 1742213c3bb0429b6e3ff25595ce2073e7a40949 [file] [log] [blame]
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +00001//===- ELFObjcopy.cpp -----------------------------------------------------===//
2//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +00006//
7//===----------------------------------------------------------------------===//
8
9#include "ELFObjcopy.h"
10#include "Buffer.h"
11#include "CopyConfig.h"
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +000012#include "Object.h"
Jake Ehrlich8ad77792018-12-03 19:49:23 +000013#include "llvm-objcopy.h"
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +000014
15#include "llvm/ADT/BitmaskEnum.h"
16#include "llvm/ADT/Optional.h"
17#include "llvm/ADT/STLExtras.h"
18#include "llvm/ADT/SmallVector.h"
19#include "llvm/ADT/StringRef.h"
20#include "llvm/ADT/Twine.h"
21#include "llvm/BinaryFormat/ELF.h"
22#include "llvm/MC/MCTargetOptions.h"
23#include "llvm/Object/Binary.h"
24#include "llvm/Object/ELFObjectFile.h"
25#include "llvm/Object/ELFTypes.h"
26#include "llvm/Object/Error.h"
27#include "llvm/Option/Option.h"
28#include "llvm/Support/Casting.h"
29#include "llvm/Support/Compression.h"
Jake Ehrlich8ad77792018-12-03 19:49:23 +000030#include "llvm/Support/Errc.h"
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +000031#include "llvm/Support/Error.h"
32#include "llvm/Support/ErrorHandling.h"
33#include "llvm/Support/ErrorOr.h"
34#include "llvm/Support/Memory.h"
35#include "llvm/Support/Path.h"
36#include "llvm/Support/raw_ostream.h"
37#include <algorithm>
38#include <cassert>
39#include <cstdlib>
40#include <functional>
41#include <iterator>
42#include <memory>
43#include <string>
44#include <system_error>
45#include <utility>
46
47namespace llvm {
48namespace objcopy {
49namespace elf {
50
51using namespace object;
52using namespace ELF;
53using SectionPred = std::function<bool(const SectionBase &Sec)>;
54
55static bool isDebugSection(const SectionBase &Sec) {
56 return StringRef(Sec.Name).startswith(".debug") ||
57 StringRef(Sec.Name).startswith(".zdebug") || Sec.Name == ".gdb_index";
58}
59
60static bool isDWOSection(const SectionBase &Sec) {
61 return StringRef(Sec.Name).endswith(".dwo");
62}
63
64static bool onlyKeepDWOPred(const Object &Obj, const SectionBase &Sec) {
65 // We can't remove the section header string table.
66 if (&Sec == Obj.SectionNames)
67 return false;
68 // Short of keeping the string table we want to keep everything that is a DWO
69 // section and remove everything else.
70 return !isDWOSection(Sec);
71}
72
Jordan Rupprechtc8927412019-01-29 15:05:38 +000073static uint64_t setSectionFlagsPreserveMask(uint64_t OldFlags,
74 uint64_t NewFlags) {
75 // Preserve some flags which should not be dropped when setting flags.
76 // Also, preserve anything OS/processor dependant.
77 const uint64_t PreserveMask = ELF::SHF_COMPRESSED | ELF::SHF_EXCLUDE |
78 ELF::SHF_GROUP | ELF::SHF_LINK_ORDER |
79 ELF::SHF_MASKOS | ELF::SHF_MASKPROC |
80 ELF::SHF_TLS | ELF::SHF_INFO_LINK;
81 return (OldFlags & PreserveMask) | (NewFlags & ~PreserveMask);
82}
83
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +000084static ElfType getOutputElfType(const Binary &Bin) {
85 // Infer output ELF type from the input ELF object
86 if (isa<ELFObjectFile<ELF32LE>>(Bin))
87 return ELFT_ELF32LE;
88 if (isa<ELFObjectFile<ELF64LE>>(Bin))
89 return ELFT_ELF64LE;
90 if (isa<ELFObjectFile<ELF32BE>>(Bin))
91 return ELFT_ELF32BE;
92 if (isa<ELFObjectFile<ELF64BE>>(Bin))
93 return ELFT_ELF64BE;
94 llvm_unreachable("Invalid ELFType");
95}
96
97static ElfType getOutputElfType(const MachineInfo &MI) {
98 // Infer output ELF type from the binary arch specified
99 if (MI.Is64Bit)
100 return MI.IsLittleEndian ? ELFT_ELF64LE : ELFT_ELF64BE;
101 else
102 return MI.IsLittleEndian ? ELFT_ELF32LE : ELFT_ELF32BE;
103}
104
105static std::unique_ptr<Writer> createWriter(const CopyConfig &Config,
106 Object &Obj, Buffer &Buf,
107 ElfType OutputElfType) {
108 if (Config.OutputFormat == "binary") {
109 return llvm::make_unique<BinaryWriter>(Obj, Buf);
110 }
111 // Depending on the initial ELFT and OutputFormat we need a different Writer.
112 switch (OutputElfType) {
113 case ELFT_ELF32LE:
114 return llvm::make_unique<ELFWriter<ELF32LE>>(Obj, Buf,
115 !Config.StripSections);
116 case ELFT_ELF64LE:
117 return llvm::make_unique<ELFWriter<ELF64LE>>(Obj, Buf,
118 !Config.StripSections);
119 case ELFT_ELF32BE:
120 return llvm::make_unique<ELFWriter<ELF32BE>>(Obj, Buf,
121 !Config.StripSections);
122 case ELFT_ELF64BE:
123 return llvm::make_unique<ELFWriter<ELF64BE>>(Obj, Buf,
124 !Config.StripSections);
125 }
126 llvm_unreachable("Invalid output format");
127}
128
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000129template <class ELFT>
130static Expected<ArrayRef<uint8_t>>
131findBuildID(const object::ELFFile<ELFT> &In) {
132 for (const auto &Phdr : unwrapOrError(In.program_headers())) {
133 if (Phdr.p_type != PT_NOTE)
134 continue;
135 Error Err = Error::success();
David Blaikieba005aa2018-12-11 00:09:06 +0000136 for (const auto &Note : In.notes(Phdr, Err))
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000137 if (Note.getType() == NT_GNU_BUILD_ID && Note.getName() == ELF_NOTE_GNU)
138 return Note.getDesc();
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000139 if (Err)
140 return std::move(Err);
141 }
142 return createStringError(llvm::errc::invalid_argument,
143 "Could not find build ID.");
144}
145
146static Expected<ArrayRef<uint8_t>>
147findBuildID(const object::ELFObjectFileBase &In) {
148 if (auto *O = dyn_cast<ELFObjectFile<ELF32LE>>(&In))
149 return findBuildID(*O->getELFFile());
150 else if (auto *O = dyn_cast<ELFObjectFile<ELF64LE>>(&In))
151 return findBuildID(*O->getELFFile());
152 else if (auto *O = dyn_cast<ELFObjectFile<ELF32BE>>(&In))
153 return findBuildID(*O->getELFFile());
154 else if (auto *O = dyn_cast<ELFObjectFile<ELF64BE>>(&In))
155 return findBuildID(*O->getELFFile());
156
157 llvm_unreachable("Bad file format");
158}
159
Jordan Rupprechtfc832e92019-01-30 18:13:30 +0000160static Error linkToBuildIdDir(const CopyConfig &Config, StringRef ToLink,
161 StringRef Suffix,
162 ArrayRef<uint8_t> BuildIdBytes) {
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000163 SmallString<128> Path = Config.BuildIdLinkDir;
164 sys::path::append(Path, llvm::toHex(BuildIdBytes[0], /*LowerCase*/ true));
165 if (auto EC = sys::fs::create_directories(Path))
Jordan Rupprechtfc832e92019-01-30 18:13:30 +0000166 return createFileError(
167 Path.str(),
168 createStringError(EC, "cannot create build ID link directory"));
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000169
170 sys::path::append(Path,
171 llvm::toHex(BuildIdBytes.slice(1), /*LowerCase*/ true));
172 Path += Suffix;
173 if (auto EC = sys::fs::create_hard_link(ToLink, Path)) {
174 // Hard linking failed, try to remove the file first if it exists.
175 if (sys::fs::exists(Path))
176 sys::fs::remove(Path);
177 EC = sys::fs::create_hard_link(ToLink, Path);
178 if (EC)
Jordan Rupprechtfc832e92019-01-30 18:13:30 +0000179 return createStringError(EC, "cannot link %s to %s", ToLink.data(),
180 Path.data());
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000181 }
Jordan Rupprechtfc832e92019-01-30 18:13:30 +0000182 return Error::success();
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000183}
184
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000185static Error splitDWOToFile(const CopyConfig &Config, const Reader &Reader,
186 StringRef File, ElfType OutputElfType) {
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000187 auto DWOFile = Reader.create();
Jordan Rupprecht971d47622019-02-01 15:20:36 +0000188 auto OnlyKeepDWOPred = [&DWOFile](const SectionBase &Sec) {
189 return onlyKeepDWOPred(*DWOFile, Sec);
190 };
191 if (Error E = DWOFile->removeSections(OnlyKeepDWOPred))
192 return E;
Jordan Rupprecht70038e02019-01-07 16:59:12 +0000193 if (Config.OutputArch)
194 DWOFile->Machine = Config.OutputArch.getValue().EMachine;
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000195 FileBuffer FB(File);
196 auto Writer = createWriter(Config, *DWOFile, FB, OutputElfType);
Jordan Rupprecht881cae72019-01-22 23:49:16 +0000197 if (Error E = Writer->finalize())
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000198 return E;
199 return Writer->write();
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000200}
201
202static Error dumpSectionToFile(StringRef SecName, StringRef Filename,
203 Object &Obj) {
204 for (auto &Sec : Obj.sections()) {
205 if (Sec.Name == SecName) {
Jordan Rupprecht16a0de22018-12-20 00:57:06 +0000206 if (Sec.OriginalData.empty())
Martin Storsjo8010c6be2019-01-22 10:57:59 +0000207 return createStringError(
208 object_error::parse_failed,
209 "Can't dump section \"%s\": it has no contents",
210 SecName.str().c_str());
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000211 Expected<std::unique_ptr<FileOutputBuffer>> BufferOrErr =
212 FileOutputBuffer::create(Filename, Sec.OriginalData.size());
213 if (!BufferOrErr)
214 return BufferOrErr.takeError();
215 std::unique_ptr<FileOutputBuffer> Buf = std::move(*BufferOrErr);
216 std::copy(Sec.OriginalData.begin(), Sec.OriginalData.end(),
217 Buf->getBufferStart());
218 if (Error E = Buf->commit())
219 return E;
220 return Error::success();
221 }
222 }
Martin Storsjo8010c6be2019-01-22 10:57:59 +0000223 return createStringError(object_error::parse_failed, "Section not found");
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000224}
225
226static bool isCompressed(const SectionBase &Section) {
227 const char *Magic = "ZLIB";
228 return StringRef(Section.Name).startswith(".zdebug") ||
229 (Section.OriginalData.size() > strlen(Magic) &&
230 !strncmp(reinterpret_cast<const char *>(Section.OriginalData.data()),
231 Magic, strlen(Magic))) ||
232 (Section.Flags & ELF::SHF_COMPRESSED);
233}
234
235static bool isCompressable(const SectionBase &Section) {
236 return !isCompressed(Section) && isDebugSection(Section) &&
237 Section.Name != ".gdb_index";
238}
239
240static void replaceDebugSections(
241 const CopyConfig &Config, Object &Obj, SectionPred &RemovePred,
242 function_ref<bool(const SectionBase &)> shouldReplace,
243 function_ref<SectionBase *(const SectionBase *)> addSection) {
244 SmallVector<SectionBase *, 13> ToReplace;
245 SmallVector<RelocationSection *, 13> RelocationSections;
246 for (auto &Sec : Obj.sections()) {
247 if (RelocationSection *R = dyn_cast<RelocationSection>(&Sec)) {
248 if (shouldReplace(*R->getSection()))
249 RelocationSections.push_back(R);
250 continue;
251 }
252
253 if (shouldReplace(Sec))
254 ToReplace.push_back(&Sec);
255 }
256
257 for (SectionBase *S : ToReplace) {
258 SectionBase *NewSection = addSection(S);
259
260 for (RelocationSection *RS : RelocationSections) {
261 if (RS->getSection() == S)
262 RS->setSection(NewSection);
263 }
264 }
265
266 RemovePred = [shouldReplace, RemovePred](const SectionBase &Sec) {
267 return shouldReplace(Sec) || RemovePred(Sec);
268 };
269}
270
Eugene Leviant2db10622019-02-13 07:34:54 +0000271static bool isUnneededSymbol(const Symbol &Sym) {
272 return !Sym.Referenced &&
273 (Sym.Binding == STB_LOCAL || Sym.getShndx() == SHN_UNDEF) &&
274 Sym.Type != STT_FILE && Sym.Type != STT_SECTION;
275}
276
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000277// This function handles the high level operations of GNU objcopy including
278// handling command line options. It's important to outline certain properties
279// we expect to hold of the command line operations. Any operation that "keeps"
280// should keep regardless of a remove. Additionally any removal should respect
281// any previous removals. Lastly whether or not something is removed shouldn't
282// depend a) on the order the options occur in or b) on some opaque priority
283// system. The only priority is that keeps/copies overrule removes.
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000284static Error handleArgs(const CopyConfig &Config, Object &Obj,
285 const Reader &Reader, ElfType OutputElfType) {
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000286
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000287 if (!Config.SplitDWO.empty())
288 if (Error E =
289 splitDWOToFile(Config, Reader, Config.SplitDWO, OutputElfType))
290 return E;
291
Jordan Rupprecht70038e02019-01-07 16:59:12 +0000292 if (Config.OutputArch)
293 Obj.Machine = Config.OutputArch.getValue().EMachine;
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000294
295 // TODO: update or remove symbols only if there is an option that affects
296 // them.
297 if (Obj.SymbolTable) {
298 Obj.SymbolTable->updateSymbols([&](Symbol &Sym) {
Jordan Rupprechtbd7735f2019-01-31 16:45:16 +0000299 // Common and undefined symbols don't make sense as local symbols, and can
300 // even cause crashes if we localize those, so skip them.
301 if (!Sym.isCommon() && Sym.getShndx() != SHN_UNDEF &&
Jordan Rupprechtb47475c2018-11-01 17:26:36 +0000302 ((Config.LocalizeHidden &&
303 (Sym.Visibility == STV_HIDDEN || Sym.Visibility == STV_INTERNAL)) ||
Fangrui Songe4ee0662018-11-29 17:32:51 +0000304 is_contained(Config.SymbolsToLocalize, Sym.Name)))
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000305 Sym.Binding = STB_LOCAL;
306
307 // Note: these two globalize flags have very similar names but different
308 // meanings:
309 //
310 // --globalize-symbol: promote a symbol to global
311 // --keep-global-symbol: all symbols except for these should be made local
312 //
313 // If --globalize-symbol is specified for a given symbol, it will be
314 // global in the output file even if it is not included via
315 // --keep-global-symbol. Because of that, make sure to check
316 // --globalize-symbol second.
317 if (!Config.SymbolsToKeepGlobal.empty() &&
Jordan Rupprecht634820d2018-10-30 16:23:38 +0000318 !is_contained(Config.SymbolsToKeepGlobal, Sym.Name) &&
319 Sym.getShndx() != SHN_UNDEF)
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000320 Sym.Binding = STB_LOCAL;
321
Fangrui Songe4ee0662018-11-29 17:32:51 +0000322 if (is_contained(Config.SymbolsToGlobalize, Sym.Name) &&
Jordan Rupprecht634820d2018-10-30 16:23:38 +0000323 Sym.getShndx() != SHN_UNDEF)
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000324 Sym.Binding = STB_GLOBAL;
325
Fangrui Songe4ee0662018-11-29 17:32:51 +0000326 if (is_contained(Config.SymbolsToWeaken, Sym.Name) &&
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000327 Sym.Binding == STB_GLOBAL)
328 Sym.Binding = STB_WEAK;
329
330 if (Config.Weaken && Sym.Binding == STB_GLOBAL &&
331 Sym.getShndx() != SHN_UNDEF)
332 Sym.Binding = STB_WEAK;
333
334 const auto I = Config.SymbolsToRename.find(Sym.Name);
335 if (I != Config.SymbolsToRename.end())
336 Sym.Name = I->getValue();
337
338 if (!Config.SymbolsPrefix.empty() && Sym.Type != STT_SECTION)
339 Sym.Name = (Config.SymbolsPrefix + Sym.Name).str();
340 });
341
342 // The purpose of this loop is to mark symbols referenced by sections
343 // (like GroupSection or RelocationSection). This way, we know which
344 // symbols are still 'needed' and which are not.
Eugene Leviant2db10622019-02-13 07:34:54 +0000345 if (Config.StripUnneeded || !Config.UnneededSymbolsToRemove.empty()) {
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000346 for (auto &Section : Obj.sections())
347 Section.markSymbols();
348 }
349
Jordan Rupprecht971d47622019-02-01 15:20:36 +0000350 auto RemoveSymbolsPred = [&](const Symbol &Sym) {
Fangrui Songe4ee0662018-11-29 17:32:51 +0000351 if (is_contained(Config.SymbolsToKeep, Sym.Name) ||
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000352 (Config.KeepFileSymbols && Sym.Type == STT_FILE))
353 return false;
354
Jordan Rupprechtd0f7bcf2019-01-30 14:58:13 +0000355 if ((Config.DiscardMode == DiscardType::All ||
356 (Config.DiscardMode == DiscardType::Locals &&
357 StringRef(Sym.Name).startswith(".L"))) &&
358 Sym.Binding == STB_LOCAL && Sym.getShndx() != SHN_UNDEF &&
359 Sym.Type != STT_FILE && Sym.Type != STT_SECTION)
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000360 return true;
361
362 if (Config.StripAll || Config.StripAllGNU)
363 return true;
364
Fangrui Songe4ee0662018-11-29 17:32:51 +0000365 if (is_contained(Config.SymbolsToRemove, Sym.Name))
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000366 return true;
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000367
Eugene Leviant2db10622019-02-13 07:34:54 +0000368 if ((Config.StripUnneeded ||
369 is_contained(Config.UnneededSymbolsToRemove, Sym.Name)) &&
370 isUnneededSymbol(Sym))
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000371 return true;
372
373 return false;
Jordan Rupprecht971d47622019-02-01 15:20:36 +0000374 };
375 if (Error E = Obj.removeSymbols(RemoveSymbolsPred))
376 return E;
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000377 }
378
379 SectionPred RemovePred = [](const SectionBase &) { return false; };
380
381 // Removes:
382 if (!Config.ToRemove.empty()) {
383 RemovePred = [&Config](const SectionBase &Sec) {
384 return is_contained(Config.ToRemove, Sec.Name);
385 };
386 }
387
388 if (Config.StripDWO || !Config.SplitDWO.empty())
389 RemovePred = [RemovePred](const SectionBase &Sec) {
390 return isDWOSection(Sec) || RemovePred(Sec);
391 };
392
393 if (Config.ExtractDWO)
394 RemovePred = [RemovePred, &Obj](const SectionBase &Sec) {
395 return onlyKeepDWOPred(Obj, Sec) || RemovePred(Sec);
396 };
397
398 if (Config.StripAllGNU)
399 RemovePred = [RemovePred, &Obj](const SectionBase &Sec) {
400 if (RemovePred(Sec))
401 return true;
402 if ((Sec.Flags & SHF_ALLOC) != 0)
403 return false;
404 if (&Sec == Obj.SectionNames)
405 return false;
406 switch (Sec.Type) {
407 case SHT_SYMTAB:
408 case SHT_REL:
409 case SHT_RELA:
410 case SHT_STRTAB:
411 return true;
412 }
413 return isDebugSection(Sec);
414 };
415
416 if (Config.StripSections) {
417 RemovePred = [RemovePred](const SectionBase &Sec) {
418 return RemovePred(Sec) || (Sec.Flags & SHF_ALLOC) == 0;
419 };
420 }
421
422 if (Config.StripDebug) {
423 RemovePred = [RemovePred](const SectionBase &Sec) {
424 return RemovePred(Sec) || isDebugSection(Sec);
425 };
426 }
427
428 if (Config.StripNonAlloc)
429 RemovePred = [RemovePred, &Obj](const SectionBase &Sec) {
430 if (RemovePred(Sec))
431 return true;
432 if (&Sec == Obj.SectionNames)
433 return false;
434 return (Sec.Flags & SHF_ALLOC) == 0;
435 };
436
437 if (Config.StripAll)
438 RemovePred = [RemovePred, &Obj](const SectionBase &Sec) {
439 if (RemovePred(Sec))
440 return true;
441 if (&Sec == Obj.SectionNames)
442 return false;
443 if (StringRef(Sec.Name).startswith(".gnu.warning"))
444 return false;
445 return (Sec.Flags & SHF_ALLOC) == 0;
446 };
447
448 // Explicit copies:
Jake Ehrlich85985ed2018-12-06 02:03:53 +0000449 if (!Config.OnlySection.empty()) {
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000450 RemovePred = [&Config, RemovePred, &Obj](const SectionBase &Sec) {
451 // Explicitly keep these sections regardless of previous removes.
Jake Ehrlich85985ed2018-12-06 02:03:53 +0000452 if (is_contained(Config.OnlySection, Sec.Name))
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000453 return false;
454
455 // Allow all implicit removes.
456 if (RemovePred(Sec))
457 return true;
458
459 // Keep special sections.
460 if (Obj.SectionNames == &Sec)
461 return false;
462 if (Obj.SymbolTable == &Sec ||
463 (Obj.SymbolTable && Obj.SymbolTable->getStrTab() == &Sec))
464 return false;
465
466 // Remove everything else.
467 return true;
468 };
469 }
470
Jordan Rupprechtc5bae782018-11-13 19:32:27 +0000471 if (!Config.KeepSection.empty()) {
Fangrui Songe9f34b02018-11-12 23:46:22 +0000472 RemovePred = [&Config, RemovePred](const SectionBase &Sec) {
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000473 // Explicitly keep these sections regardless of previous removes.
Jordan Rupprechtc5bae782018-11-13 19:32:27 +0000474 if (is_contained(Config.KeepSection, Sec.Name))
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000475 return false;
476 // Otherwise defer to RemovePred.
477 return RemovePred(Sec);
478 };
479 }
480
481 // This has to be the last predicate assignment.
482 // If the option --keep-symbol has been specified
483 // and at least one of those symbols is present
484 // (equivalently, the updated symbol table is not empty)
485 // the symbol table and the string table should not be removed.
486 if ((!Config.SymbolsToKeep.empty() || Config.KeepFileSymbols) &&
487 Obj.SymbolTable && !Obj.SymbolTable->empty()) {
488 RemovePred = [&Obj, RemovePred](const SectionBase &Sec) {
489 if (&Sec == Obj.SymbolTable || &Sec == Obj.SymbolTable->getStrTab())
490 return false;
491 return RemovePred(Sec);
492 };
493 }
494
495 if (Config.CompressionType != DebugCompressionType::None)
496 replaceDebugSections(Config, Obj, RemovePred, isCompressable,
497 [&Config, &Obj](const SectionBase *S) {
498 return &Obj.addSection<CompressedSection>(
499 *S, Config.CompressionType);
500 });
501 else if (Config.DecompressDebugSections)
502 replaceDebugSections(
503 Config, Obj, RemovePred,
504 [](const SectionBase &S) { return isa<CompressedSection>(&S); },
505 [&Obj](const SectionBase *S) {
506 auto CS = cast<CompressedSection>(S);
507 return &Obj.addSection<DecompressedSection>(*CS);
508 });
509
Jordan Rupprecht971d47622019-02-01 15:20:36 +0000510 if (Error E = Obj.removeSections(RemovePred))
511 return E;
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000512
513 if (!Config.SectionsToRename.empty()) {
514 for (auto &Sec : Obj.sections()) {
515 const auto Iter = Config.SectionsToRename.find(Sec.Name);
516 if (Iter != Config.SectionsToRename.end()) {
517 const SectionRename &SR = Iter->second;
518 Sec.Name = SR.NewName;
Jordan Rupprechtc8927412019-01-29 15:05:38 +0000519 if (SR.NewFlags.hasValue())
520 Sec.Flags =
521 setSectionFlagsPreserveMask(Sec.Flags, SR.NewFlags.getValue());
522 }
523 }
524 }
525
526 if (!Config.SetSectionFlags.empty()) {
527 for (auto &Sec : Obj.sections()) {
528 const auto Iter = Config.SetSectionFlags.find(Sec.Name);
529 if (Iter != Config.SetSectionFlags.end()) {
530 const SectionFlagsUpdate &SFU = Iter->second;
531 Sec.Flags = setSectionFlagsPreserveMask(Sec.Flags, SFU.NewFlags);
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000532 }
533 }
534 }
535
536 if (!Config.AddSection.empty()) {
537 for (const auto &Flag : Config.AddSection) {
Jordan Rupprecht17dd4a22019-01-15 16:57:23 +0000538 std::pair<StringRef, StringRef> SecPair = Flag.split("=");
539 StringRef SecName = SecPair.first;
540 StringRef File = SecPair.second;
541 ErrorOr<std::unique_ptr<MemoryBuffer>> BufOrErr =
542 MemoryBuffer::getFile(File);
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000543 if (!BufOrErr)
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000544 return createFileError(File, errorCodeToError(BufOrErr.getError()));
Jordan Rupprecht17dd4a22019-01-15 16:57:23 +0000545 std::unique_ptr<MemoryBuffer> Buf = std::move(*BufOrErr);
546 ArrayRef<uint8_t> Data(
547 reinterpret_cast<const uint8_t *>(Buf->getBufferStart()),
548 Buf->getBufferSize());
549 OwnedDataSection &NewSection =
550 Obj.addSection<OwnedDataSection>(SecName, Data);
551 if (SecName.startswith(".note") && SecName != ".note.GNU-stack")
552 NewSection.Type = SHT_NOTE;
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000553 }
554 }
555
556 if (!Config.DumpSection.empty()) {
557 for (const auto &Flag : Config.DumpSection) {
558 std::pair<StringRef, StringRef> SecPair = Flag.split("=");
559 StringRef SecName = SecPair.first;
560 StringRef File = SecPair.second;
561 if (Error E = dumpSectionToFile(SecName, File, Obj))
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000562 return createFileError(Config.InputFilename, std::move(E));
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000563 }
564 }
565
566 if (!Config.AddGnuDebugLink.empty())
567 Obj.addSection<GnuDebugLinkSection>(Config.AddGnuDebugLink);
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000568
Eugene Leviant51c1f642019-02-25 14:12:41 +0000569 for (const NewSymbolInfo &SI : Config.SymbolsToAdd) {
570 SectionBase *Sec = Obj.findSection(SI.SectionName);
571 uint64_t Value = Sec ? Sec->Addr + SI.Value : SI.Value;
572 Obj.SymbolTable->addSymbol(SI.SymbolName, SI.Bind, SI.Type, Sec, Value,
573 SI.Visibility,
574 Sec ? SYMBOL_SIMPLE_INDEX : SHN_ABS, 0);
575 }
576
Eugene Leviant53350d02019-02-26 09:24:22 +0000577 if (Config.EntryExpr)
578 Obj.Entry = Config.EntryExpr(Obj.Entry);
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000579 return Error::success();
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000580}
581
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000582Error executeObjcopyOnRawBinary(const CopyConfig &Config, MemoryBuffer &In,
583 Buffer &Out) {
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000584 BinaryReader Reader(Config.BinaryArch, &In);
585 std::unique_ptr<Object> Obj = Reader.create();
586
Jordan Rupprecht70038e02019-01-07 16:59:12 +0000587 // Prefer OutputArch (-O<format>) if set, otherwise fallback to BinaryArch
588 // (-B<arch>).
589 const ElfType OutputElfType = getOutputElfType(
590 Config.OutputArch ? Config.OutputArch.getValue() : Config.BinaryArch);
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000591 if (Error E = handleArgs(Config, *Obj, Reader, OutputElfType))
592 return E;
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000593 std::unique_ptr<Writer> Writer =
594 createWriter(Config, *Obj, Out, OutputElfType);
Jordan Rupprecht881cae72019-01-22 23:49:16 +0000595 if (Error E = Writer->finalize())
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000596 return E;
597 return Writer->write();
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000598}
599
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000600Error executeObjcopyOnBinary(const CopyConfig &Config,
601 object::ELFObjectFileBase &In, Buffer &Out) {
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000602 ELFReader Reader(&In);
603 std::unique_ptr<Object> Obj = Reader.create();
Jordan Rupprecht70038e02019-01-07 16:59:12 +0000604 // Prefer OutputArch (-O<format>) if set, otherwise infer it from the input.
605 const ElfType OutputElfType =
606 Config.OutputArch ? getOutputElfType(Config.OutputArch.getValue())
607 : getOutputElfType(In);
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000608 ArrayRef<uint8_t> BuildIdBytes;
609
610 if (!Config.BuildIdLinkDir.empty()) {
611 BuildIdBytes = unwrapOrError(findBuildID(In));
612 if (BuildIdBytes.size() < 2)
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000613 return createFileError(
614 Config.InputFilename,
615 createStringError(object_error::parse_failed,
616 "build ID is smaller than two bytes."));
Jake Ehrlich8ad77792018-12-03 19:49:23 +0000617 }
618
Jordan Rupprechtfc832e92019-01-30 18:13:30 +0000619 if (!Config.BuildIdLinkDir.empty() && Config.BuildIdLinkInput)
620 if (Error E =
621 linkToBuildIdDir(Config, Config.InputFilename,
622 Config.BuildIdLinkInput.getValue(), BuildIdBytes))
623 return E;
624
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000625 if (Error E = handleArgs(Config, *Obj, Reader, OutputElfType))
626 return E;
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000627 std::unique_ptr<Writer> Writer =
628 createWriter(Config, *Obj, Out, OutputElfType);
Jordan Rupprecht881cae72019-01-22 23:49:16 +0000629 if (Error E = Writer->finalize())
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000630 return E;
Jordan Rupprecht881cae72019-01-22 23:49:16 +0000631 if (Error E = Writer->write())
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000632 return E;
Jordan Rupprechtfc832e92019-01-30 18:13:30 +0000633 if (!Config.BuildIdLinkDir.empty() && Config.BuildIdLinkOutput)
634 if (Error E =
635 linkToBuildIdDir(Config, Config.OutputFilename,
636 Config.BuildIdLinkOutput.getValue(), BuildIdBytes))
637 return E;
638
Jordan Rupprecht307deab2019-01-30 14:36:53 +0000639 return Error::success();
Alexander Shaposhnikovf4e75a52018-10-29 21:22:58 +0000640}
641
642} // end namespace elf
643} // end namespace objcopy
644} // end namespace llvm