xref: /freebsd/contrib/llvm-project/llvm/tools/llvm-readobj/llvm-readobj.cpp (revision f425b8be7e7a38e47677d7bd68819dccbdcc3134) 
1 //===- llvm-readobj.cpp - Dump contents of an Object File -----------------===//
2 //
3 // 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
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This is a tool similar to readelf, except it works on multiple object file
10 // formats. The main purpose of this tool is to provide detailed output suitable
11 // for FileCheck.
12 //
13 // Flags should be similar to readelf where supported, but the output format
14 // does not need to be identical. The point is to not make users learn yet
15 // another set of flags.
16 //
17 // Output should be specialized for each format where appropriate.
18 //
19 //===----------------------------------------------------------------------===//
20 
21 #include "llvm-readobj.h"
22 #include "Error.h"
23 #include "ObjDumper.h"
24 #include "WindowsResourceDumper.h"
25 #include "llvm/DebugInfo/CodeView/GlobalTypeTableBuilder.h"
26 #include "llvm/DebugInfo/CodeView/MergingTypeTableBuilder.h"
27 #include "llvm/Object/Archive.h"
28 #include "llvm/Object/COFFImportFile.h"
29 #include "llvm/Object/MachOUniversal.h"
30 #include "llvm/Object/ObjectFile.h"
31 #include "llvm/Object/WindowsResource.h"
32 #include "llvm/Support/Casting.h"
33 #include "llvm/Support/CommandLine.h"
34 #include "llvm/Support/DataTypes.h"
35 #include "llvm/Support/Debug.h"
36 #include "llvm/Support/FileSystem.h"
37 #include "llvm/Support/FormatVariadic.h"
38 #include "llvm/Support/InitLLVM.h"
39 #include "llvm/Support/Path.h"
40 #include "llvm/Support/ScopedPrinter.h"
41 #include "llvm/Support/TargetRegistry.h"
42 #include "llvm/Support/WithColor.h"
43 
44 using namespace llvm;
45 using namespace llvm::object;
46 
47 namespace opts {
48   cl::list<std::string> InputFilenames(cl::Positional,
49     cl::desc("<input object files>"),
50     cl::ZeroOrMore);
51 
52   // --all, -a
53   cl::opt<bool>
54       All("all",
55           cl::desc("Equivalent to setting: --file-headers, --program-headers, "
56                    "--section-headers, --symbols, --relocations, "
57                    "--dynamic-table, --notes, --version-info, --unwind, "
58                    "--section-groups and --elf-hash-histogram."));
59   cl::alias AllShort("a", cl::desc("Alias for --all"), cl::aliasopt(All));
60 
61   // --headers -e
62   cl::opt<bool>
63       Headers("headers",
64           cl::desc("Equivalent to setting: --file-headers, --program-headers, "
65                    "--section-headers"));
66   cl::alias HeadersShort("e", cl::desc("Alias for --headers"),
67      cl::aliasopt(Headers));
68 
69   // --wide, -W
70   cl::opt<bool>
71       WideOutput("wide", cl::desc("Ignored for compatibility with GNU readelf"),
72                  cl::Hidden);
73   cl::alias WideOutputShort("W",
74     cl::desc("Alias for --wide"),
75     cl::aliasopt(WideOutput));
76 
77   // --file-headers, --file-header, -h
78   cl::opt<bool> FileHeaders("file-headers",
79     cl::desc("Display file headers "));
80   cl::alias FileHeadersShort("h", cl::desc("Alias for --file-headers"),
81                              cl::aliasopt(FileHeaders), cl::NotHidden);
82   cl::alias FileHeadersSingular("file-header",
83                                 cl::desc("Alias for --file-headers"),
84                                 cl::aliasopt(FileHeaders));
85 
86   // --section-headers, --sections, -S
87   // Also -s in llvm-readobj mode.
88   cl::opt<bool> SectionHeaders("section-headers",
89                                cl::desc("Display all section headers."));
90   cl::alias SectionsShortUpper("S", cl::desc("Alias for --section-headers"),
91                                cl::aliasopt(SectionHeaders), cl::NotHidden);
92   cl::alias SectionHeadersAlias("sections",
93                                 cl::desc("Alias for --section-headers"),
94                                 cl::aliasopt(SectionHeaders), cl::NotHidden);
95 
96   // --section-relocations
97   // Also --sr in llvm-readobj mode.
98   cl::opt<bool> SectionRelocations("section-relocations",
99     cl::desc("Display relocations for each section shown."));
100 
101   // --section-symbols
102   // Also --st in llvm-readobj mode.
103   cl::opt<bool> SectionSymbols("section-symbols",
104     cl::desc("Display symbols for each section shown."));
105 
106   // --section-data
107   // Also --sd in llvm-readobj mode.
108   cl::opt<bool> SectionData("section-data",
109     cl::desc("Display section data for each section shown."));
110 
111   // --section-mapping
112   cl::opt<cl::boolOrDefault>
113       SectionMapping("section-mapping",
114                      cl::desc("Display the section to segment mapping."));
115 
116   // --relocations, --relocs, -r
117   cl::opt<bool> Relocations("relocations",
118     cl::desc("Display the relocation entries in the file"));
119   cl::alias RelocationsShort("r", cl::desc("Alias for --relocations"),
120                              cl::aliasopt(Relocations), cl::NotHidden);
121   cl::alias RelocationsGNU("relocs", cl::desc("Alias for --relocations"),
122                            cl::aliasopt(Relocations));
123 
124   // --notes, -n
125   cl::opt<bool> Notes("notes", cl::desc("Display the ELF notes in the file"));
126   cl::alias NotesShort("n", cl::desc("Alias for --notes"), cl::aliasopt(Notes));
127 
128   // --dyn-relocations
129   cl::opt<bool> DynRelocs("dyn-relocations",
130     cl::desc("Display the dynamic relocation entries in the file"));
131 
132   // --symbols
133   // Also -s in llvm-readelf mode, or -t in llvm-readobj mode.
134   cl::opt<bool>
135       Symbols("symbols",
136               cl::desc("Display the symbol table. Also display the dynamic "
137                        "symbol table when using GNU output style for ELF"));
138   cl::alias SymbolsGNU("syms", cl::desc("Alias for --symbols"),
139                        cl::aliasopt(Symbols));
140 
141   // --dyn-symbols, --dyn-syms
142   // Also --dt in llvm-readobj mode.
143   cl::opt<bool> DynamicSymbols("dyn-symbols",
144     cl::desc("Display the dynamic symbol table"));
145   cl::alias DynSymsGNU("dyn-syms", cl::desc("Alias for --dyn-symbols"),
146                        cl::aliasopt(DynamicSymbols));
147 
148   // --hash-symbols
149   cl::opt<bool> HashSymbols(
150       "hash-symbols",
151       cl::desc("Display the dynamic symbols derived from the hash section"));
152 
153   // --unwind, -u
154   cl::opt<bool> UnwindInfo("unwind",
155     cl::desc("Display unwind information"));
156   cl::alias UnwindInfoShort("u",
157     cl::desc("Alias for --unwind"),
158     cl::aliasopt(UnwindInfo));
159 
160   // --dynamic-table, --dynamic, -d
161   cl::opt<bool> DynamicTable("dynamic-table",
162     cl::desc("Display the ELF .dynamic section table"));
163   cl::alias DynamicTableShort("d", cl::desc("Alias for --dynamic-table"),
164                               cl::aliasopt(DynamicTable), cl::NotHidden);
165   cl::alias DynamicTableAlias("dynamic", cl::desc("Alias for --dynamic-table"),
166                               cl::aliasopt(DynamicTable));
167 
168   // --needed-libs
169   cl::opt<bool> NeededLibraries("needed-libs",
170     cl::desc("Display the needed libraries"));
171 
172   // --program-headers, --segments, -l
173   cl::opt<bool> ProgramHeaders("program-headers",
174     cl::desc("Display ELF program headers"));
175   cl::alias ProgramHeadersShort("l", cl::desc("Alias for --program-headers"),
176                                 cl::aliasopt(ProgramHeaders), cl::NotHidden);
177   cl::alias SegmentsAlias("segments", cl::desc("Alias for --program-headers"),
178                           cl::aliasopt(ProgramHeaders));
179 
180   // --string-dump, -p
181   cl::list<std::string> StringDump("string-dump", cl::desc("<number|name>"),
182                                    cl::ZeroOrMore);
183   cl::alias StringDumpShort("p", cl::desc("Alias for --string-dump"),
184                             cl::aliasopt(StringDump), cl::Prefix);
185 
186   // --hex-dump, -x
187   cl::list<std::string> HexDump("hex-dump", cl::desc("<number|name>"),
188                                 cl::ZeroOrMore);
189   cl::alias HexDumpShort("x", cl::desc("Alias for --hex-dump"),
190                          cl::aliasopt(HexDump), cl::Prefix);
191 
192   // --demangle, -C
193   cl::opt<bool> Demangle("demangle",
194                          cl::desc("Demangle symbol names in output"));
195   cl::alias DemangleShort("C", cl::desc("Alias for --demangle"),
196                           cl::aliasopt(Demangle), cl::NotHidden);
197 
198   // --hash-table
199   cl::opt<bool> HashTable("hash-table",
200     cl::desc("Display ELF hash table"));
201 
202   // --gnu-hash-table
203   cl::opt<bool> GnuHashTable("gnu-hash-table",
204     cl::desc("Display ELF .gnu.hash section"));
205 
206   // --expand-relocs
207   cl::opt<bool> ExpandRelocs("expand-relocs",
208     cl::desc("Expand each shown relocation to multiple lines"));
209 
210   // --raw-relr
211   cl::opt<bool> RawRelr("raw-relr",
212     cl::desc("Do not decode relocations in SHT_RELR section, display raw contents"));
213 
214   // --codeview
215   cl::opt<bool> CodeView("codeview",
216                          cl::desc("Display CodeView debug information"));
217 
218   // --codeview-merged-types
219   cl::opt<bool>
220       CodeViewMergedTypes("codeview-merged-types",
221                           cl::desc("Display the merged CodeView type stream"));
222 
223   // --codeview-ghash
224   cl::opt<bool> CodeViewEnableGHash(
225       "codeview-ghash",
226       cl::desc(
227           "Enable global hashing for CodeView type stream de-duplication"));
228 
229   // --codeview-subsection-bytes
230   cl::opt<bool> CodeViewSubsectionBytes(
231       "codeview-subsection-bytes",
232       cl::desc("Dump raw contents of codeview debug sections and records"));
233 
234   // --arm-attributes
235   cl::opt<bool> ARMAttributes("arm-attributes",
236                               cl::desc("Display the ARM attributes section"));
237 
238   // --mips-plt-got
239   cl::opt<bool>
240   MipsPLTGOT("mips-plt-got",
241              cl::desc("Display the MIPS GOT and PLT GOT sections"));
242 
243   // --mips-abi-flags
244   cl::opt<bool> MipsABIFlags("mips-abi-flags",
245                              cl::desc("Display the MIPS.abiflags section"));
246 
247   // --mips-reginfo
248   cl::opt<bool> MipsReginfo("mips-reginfo",
249                             cl::desc("Display the MIPS .reginfo section"));
250 
251   // --mips-options
252   cl::opt<bool> MipsOptions("mips-options",
253                             cl::desc("Display the MIPS .MIPS.options section"));
254 
255   // --coff-imports
256   cl::opt<bool>
257   COFFImports("coff-imports", cl::desc("Display the PE/COFF import table"));
258 
259   // --coff-exports
260   cl::opt<bool>
261   COFFExports("coff-exports", cl::desc("Display the PE/COFF export table"));
262 
263   // --coff-directives
264   cl::opt<bool>
265   COFFDirectives("coff-directives",
266                  cl::desc("Display the PE/COFF .drectve section"));
267 
268   // --coff-basereloc
269   cl::opt<bool>
270   COFFBaseRelocs("coff-basereloc",
271                  cl::desc("Display the PE/COFF .reloc section"));
272 
273   // --coff-debug-directory
274   cl::opt<bool>
275   COFFDebugDirectory("coff-debug-directory",
276                      cl::desc("Display the PE/COFF debug directory"));
277 
278   // --coff-resources
279   cl::opt<bool> COFFResources("coff-resources",
280                               cl::desc("Display the PE/COFF .rsrc section"));
281 
282   // --coff-load-config
283   cl::opt<bool>
284   COFFLoadConfig("coff-load-config",
285                  cl::desc("Display the PE/COFF load config"));
286 
287   // --elf-linker-options
288   cl::opt<bool>
289   ELFLinkerOptions("elf-linker-options",
290                    cl::desc("Display the ELF .linker-options section"));
291 
292   // --macho-data-in-code
293   cl::opt<bool>
294   MachODataInCode("macho-data-in-code",
295                   cl::desc("Display MachO Data in Code command"));
296 
297   // --macho-indirect-symbols
298   cl::opt<bool>
299   MachOIndirectSymbols("macho-indirect-symbols",
300                   cl::desc("Display MachO indirect symbols"));
301 
302   // --macho-linker-options
303   cl::opt<bool>
304   MachOLinkerOptions("macho-linker-options",
305                   cl::desc("Display MachO linker options"));
306 
307   // --macho-segment
308   cl::opt<bool>
309   MachOSegment("macho-segment",
310                   cl::desc("Display MachO Segment command"));
311 
312   // --macho-version-min
313   cl::opt<bool>
314   MachOVersionMin("macho-version-min",
315                   cl::desc("Display MachO version min command"));
316 
317   // --macho-dysymtab
318   cl::opt<bool>
319   MachODysymtab("macho-dysymtab",
320                   cl::desc("Display MachO Dysymtab command"));
321 
322   // --stackmap
323   cl::opt<bool>
324   PrintStackMap("stackmap",
325                 cl::desc("Display contents of stackmap section"));
326 
327   // --version-info, -V
328   cl::opt<bool>
329       VersionInfo("version-info",
330                   cl::desc("Display ELF version sections (if present)"));
331   cl::alias VersionInfoShort("V", cl::desc("Alias for -version-info"),
332                              cl::aliasopt(VersionInfo));
333 
334   // --elf-section-groups, --section-groups, -g
335   cl::opt<bool> SectionGroups("elf-section-groups",
336                               cl::desc("Display ELF section group contents"));
337   cl::alias SectionGroupsAlias("section-groups",
338                                cl::desc("Alias for -elf-sections-groups"),
339                                cl::aliasopt(SectionGroups));
340   cl::alias SectionGroupsShort("g", cl::desc("Alias for -elf-sections-groups"),
341                                cl::aliasopt(SectionGroups));
342 
343   // --elf-hash-histogram, --histogram, -I
344   cl::opt<bool> HashHistogram(
345       "elf-hash-histogram",
346       cl::desc("Display bucket list histogram for hash sections"));
347   cl::alias HashHistogramShort("I", cl::desc("Alias for -elf-hash-histogram"),
348                                cl::aliasopt(HashHistogram));
349   cl::alias HistogramAlias("histogram",
350                            cl::desc("Alias for --elf-hash-histogram"),
351                            cl::aliasopt(HashHistogram));
352 
353   // --elf-cg-profile
354   cl::opt<bool> CGProfile("elf-cg-profile", cl::desc("Display callgraph profile section"));
355 
356   // -addrsig
357   cl::opt<bool> Addrsig("addrsig",
358                         cl::desc("Display address-significance table"));
359 
360   // -elf-output-style
361   cl::opt<OutputStyleTy>
362       Output("elf-output-style", cl::desc("Specify ELF dump style"),
363              cl::values(clEnumVal(LLVM, "LLVM default style"),
364                         clEnumVal(GNU, "GNU readelf style")),
365              cl::init(LLVM));
366 
367   cl::extrahelp
368       HelpResponse("\nPass @FILE as argument to read options from FILE.\n");
369 } // namespace opts
370 
371 namespace llvm {
372 
373 LLVM_ATTRIBUTE_NORETURN void reportError(Twine Msg) {
374   fouts().flush();
375   errs() << "\n";
376   WithColor::error(errs()) << Msg << "\n";
377   exit(1);
378 }
379 
380 void reportError(StringRef Input, Error Err) {
381   if (Input == "-")
382     Input = "<stdin>";
383   error(createFileError(Input, std::move(Err)));
384 }
385 
386 void reportWarning(Twine Msg) {
387   fouts().flush();
388   errs() << "\n";
389   WithColor::warning(errs()) << Msg << "\n";
390 }
391 
392 void warn(Error Err) {
393   handleAllErrors(std::move(Err), [&](const ErrorInfoBase &EI) {
394     reportWarning(EI.message());
395   });
396 }
397 
398 void error(Error EC) {
399   if (!EC)
400     return;
401   handleAllErrors(std::move(EC),
402                   [&](const ErrorInfoBase &EI) { reportError(EI.message()); });
403 }
404 
405 void error(std::error_code EC) {
406   if (!EC)
407     return;
408   reportError(EC.message());
409 }
410 
411 } // namespace llvm
412 
413 static void reportError(StringRef Input, std::error_code EC) {
414   reportError(Input, errorCodeToError(EC));
415 }
416 
417 static bool isMipsArch(unsigned Arch) {
418   switch (Arch) {
419   case llvm::Triple::mips:
420   case llvm::Triple::mipsel:
421   case llvm::Triple::mips64:
422   case llvm::Triple::mips64el:
423     return true;
424   default:
425     return false;
426   }
427 }
428 namespace {
429 struct ReadObjTypeTableBuilder {
430   ReadObjTypeTableBuilder()
431       : Allocator(), IDTable(Allocator), TypeTable(Allocator),
432         GlobalIDTable(Allocator), GlobalTypeTable(Allocator) {}
433 
434   llvm::BumpPtrAllocator Allocator;
435   llvm::codeview::MergingTypeTableBuilder IDTable;
436   llvm::codeview::MergingTypeTableBuilder TypeTable;
437   llvm::codeview::GlobalTypeTableBuilder GlobalIDTable;
438   llvm::codeview::GlobalTypeTableBuilder GlobalTypeTable;
439   std::vector<OwningBinary<Binary>> Binaries;
440 };
441 } // namespace
442 static ReadObjTypeTableBuilder CVTypes;
443 
444 /// Creates an format-specific object file dumper.
445 static std::error_code createDumper(const ObjectFile *Obj,
446                                     ScopedPrinter &Writer,
447                                     std::unique_ptr<ObjDumper> &Result) {
448   if (!Obj)
449     return readobj_error::unsupported_file_format;
450 
451   if (Obj->isCOFF())
452     return createCOFFDumper(Obj, Writer, Result);
453   if (Obj->isELF())
454     return createELFDumper(Obj, Writer, Result);
455   if (Obj->isMachO())
456     return createMachODumper(Obj, Writer, Result);
457   if (Obj->isWasm())
458     return createWasmDumper(Obj, Writer, Result);
459   if (Obj->isXCOFF())
460     return createXCOFFDumper(Obj, Writer, Result);
461 
462   return readobj_error::unsupported_obj_file_format;
463 }
464 
465 /// Dumps the specified object file.
466 static void dumpObject(const ObjectFile *Obj, ScopedPrinter &Writer,
467                        const Archive *A = nullptr) {
468   std::string FileStr =
469           A ? Twine(A->getFileName() + "(" + Obj->getFileName() + ")").str()
470             : Obj->getFileName().str();
471 
472   std::unique_ptr<ObjDumper> Dumper;
473   if (std::error_code EC = createDumper(Obj, Writer, Dumper))
474     reportError(FileStr, EC);
475 
476   Writer.startLine() << "\n";
477   if (opts::Output == opts::LLVM) {
478     Writer.printString("File", FileStr);
479     Writer.printString("Format", Obj->getFileFormatName());
480     Writer.printString("Arch", Triple::getArchTypeName(
481                                    (llvm::Triple::ArchType)Obj->getArch()));
482     Writer.printString("AddressSize",
483                        formatv("{0}bit", 8 * Obj->getBytesInAddress()));
484     Dumper->printLoadName();
485   } else if (opts::Output == opts::GNU && A) {
486     Writer.printString("File", FileStr);
487   }
488 
489   if (opts::FileHeaders)
490     Dumper->printFileHeaders();
491   if (opts::SectionHeaders)
492     Dumper->printSectionHeaders();
493   if (opts::Relocations)
494     Dumper->printRelocations();
495   if (opts::DynRelocs)
496     Dumper->printDynamicRelocations();
497   if (opts::Symbols || opts::DynamicSymbols)
498     Dumper->printSymbols(opts::Symbols, opts::DynamicSymbols);
499   if (opts::HashSymbols)
500     Dumper->printHashSymbols();
501   if (opts::UnwindInfo)
502     Dumper->printUnwindInfo();
503   if (opts::DynamicTable)
504     Dumper->printDynamicTable();
505   if (opts::NeededLibraries)
506     Dumper->printNeededLibraries();
507   if (opts::ProgramHeaders || opts::SectionMapping == cl::BOU_TRUE)
508     Dumper->printProgramHeaders(opts::ProgramHeaders, opts::SectionMapping);
509   if (!opts::StringDump.empty())
510     Dumper->printSectionsAsString(Obj, opts::StringDump);
511   if (!opts::HexDump.empty())
512     Dumper->printSectionsAsHex(Obj, opts::HexDump);
513   if (opts::HashTable)
514     Dumper->printHashTable();
515   if (opts::GnuHashTable)
516     Dumper->printGnuHashTable();
517   if (opts::VersionInfo)
518     Dumper->printVersionInfo();
519   if (Obj->isELF()) {
520     if (opts::ELFLinkerOptions)
521       Dumper->printELFLinkerOptions();
522     if (Obj->getArch() == llvm::Triple::arm)
523       if (opts::ARMAttributes)
524         Dumper->printAttributes();
525     if (isMipsArch(Obj->getArch())) {
526       if (opts::MipsPLTGOT)
527         Dumper->printMipsPLTGOT();
528       if (opts::MipsABIFlags)
529         Dumper->printMipsABIFlags();
530       if (opts::MipsReginfo)
531         Dumper->printMipsReginfo();
532       if (opts::MipsOptions)
533         Dumper->printMipsOptions();
534     }
535     if (opts::SectionGroups)
536       Dumper->printGroupSections();
537     if (opts::HashHistogram)
538       Dumper->printHashHistogram();
539     if (opts::CGProfile)
540       Dumper->printCGProfile();
541     if (opts::Addrsig)
542       Dumper->printAddrsig();
543     if (opts::Notes)
544       Dumper->printNotes();
545   }
546   if (Obj->isCOFF()) {
547     if (opts::COFFImports)
548       Dumper->printCOFFImports();
549     if (opts::COFFExports)
550       Dumper->printCOFFExports();
551     if (opts::COFFDirectives)
552       Dumper->printCOFFDirectives();
553     if (opts::COFFBaseRelocs)
554       Dumper->printCOFFBaseReloc();
555     if (opts::COFFDebugDirectory)
556       Dumper->printCOFFDebugDirectory();
557     if (opts::COFFResources)
558       Dumper->printCOFFResources();
559     if (opts::COFFLoadConfig)
560       Dumper->printCOFFLoadConfig();
561     if (opts::Addrsig)
562       Dumper->printAddrsig();
563     if (opts::CodeView)
564       Dumper->printCodeViewDebugInfo();
565     if (opts::CodeViewMergedTypes)
566       Dumper->mergeCodeViewTypes(CVTypes.IDTable, CVTypes.TypeTable,
567                                  CVTypes.GlobalIDTable, CVTypes.GlobalTypeTable,
568                                  opts::CodeViewEnableGHash);
569   }
570   if (Obj->isMachO()) {
571     if (opts::MachODataInCode)
572       Dumper->printMachODataInCode();
573     if (opts::MachOIndirectSymbols)
574       Dumper->printMachOIndirectSymbols();
575     if (opts::MachOLinkerOptions)
576       Dumper->printMachOLinkerOptions();
577     if (opts::MachOSegment)
578       Dumper->printMachOSegment();
579     if (opts::MachOVersionMin)
580       Dumper->printMachOVersionMin();
581     if (opts::MachODysymtab)
582       Dumper->printMachODysymtab();
583   }
584   if (opts::PrintStackMap)
585     Dumper->printStackMap();
586 }
587 
588 /// Dumps each object file in \a Arc;
589 static void dumpArchive(const Archive *Arc, ScopedPrinter &Writer) {
590   Error Err = Error::success();
591   for (auto &Child : Arc->children(Err)) {
592     Expected<std::unique_ptr<Binary>> ChildOrErr = Child.getAsBinary();
593     if (!ChildOrErr) {
594       if (auto E = isNotObjectErrorInvalidFileType(ChildOrErr.takeError())) {
595         reportError(Arc->getFileName(), std::move(E));
596       }
597       continue;
598     }
599     if (ObjectFile *Obj = dyn_cast<ObjectFile>(&*ChildOrErr.get()))
600       dumpObject(Obj, Writer, Arc);
601     else if (COFFImportFile *Imp = dyn_cast<COFFImportFile>(&*ChildOrErr.get()))
602       dumpCOFFImportFile(Imp, Writer);
603     else
604       reportError(Arc->getFileName(), readobj_error::unrecognized_file_format);
605   }
606   if (Err)
607     reportError(Arc->getFileName(), std::move(Err));
608 }
609 
610 /// Dumps each object file in \a MachO Universal Binary;
611 static void dumpMachOUniversalBinary(const MachOUniversalBinary *UBinary,
612                                      ScopedPrinter &Writer) {
613   for (const MachOUniversalBinary::ObjectForArch &Obj : UBinary->objects()) {
614     Expected<std::unique_ptr<MachOObjectFile>> ObjOrErr = Obj.getAsObjectFile();
615     if (ObjOrErr)
616       dumpObject(&*ObjOrErr.get(), Writer);
617     else if (auto E = isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) {
618       reportError(UBinary->getFileName(), ObjOrErr.takeError());
619     }
620     else if (Expected<std::unique_ptr<Archive>> AOrErr = Obj.getAsArchive())
621       dumpArchive(&*AOrErr.get(), Writer);
622   }
623 }
624 
625 /// Dumps \a WinRes, Windows Resource (.res) file;
626 static void dumpWindowsResourceFile(WindowsResource *WinRes,
627                                     ScopedPrinter &Printer) {
628   WindowsRes::Dumper Dumper(WinRes, Printer);
629   if (auto Err = Dumper.printData())
630     reportError(WinRes->getFileName(), std::move(Err));
631 }
632 
633 
634 /// Opens \a File and dumps it.
635 static void dumpInput(StringRef File, ScopedPrinter &Writer) {
636   // Attempt to open the binary.
637   Expected<OwningBinary<Binary>> BinaryOrErr = createBinary(File);
638   if (!BinaryOrErr)
639     reportError(File, BinaryOrErr.takeError());
640   Binary &Binary = *BinaryOrErr.get().getBinary();
641 
642   if (Archive *Arc = dyn_cast<Archive>(&Binary))
643     dumpArchive(Arc, Writer);
644   else if (MachOUniversalBinary *UBinary =
645                dyn_cast<MachOUniversalBinary>(&Binary))
646     dumpMachOUniversalBinary(UBinary, Writer);
647   else if (ObjectFile *Obj = dyn_cast<ObjectFile>(&Binary))
648     dumpObject(Obj, Writer);
649   else if (COFFImportFile *Import = dyn_cast<COFFImportFile>(&Binary))
650     dumpCOFFImportFile(Import, Writer);
651   else if (WindowsResource *WinRes = dyn_cast<WindowsResource>(&Binary))
652     dumpWindowsResourceFile(WinRes, Writer);
653   else
654     reportError(File, readobj_error::unrecognized_file_format);
655 
656   CVTypes.Binaries.push_back(std::move(*BinaryOrErr));
657 }
658 
659 /// Registers aliases that should only be allowed by readobj.
660 static void registerReadobjAliases() {
661   // -s has meant --sections for a very long time in llvm-readobj despite
662   // meaning --symbols in readelf.
663   static cl::alias SectionsShort("s", cl::desc("Alias for --section-headers"),
664                                  cl::aliasopt(opts::SectionHeaders),
665                                  cl::NotHidden);
666 
667   // Only register -t in llvm-readobj, as readelf reserves it for
668   // --section-details (not implemented yet).
669   static cl::alias SymbolsShort("t", cl::desc("Alias for --symbols"),
670                                 cl::aliasopt(opts::Symbols), cl::NotHidden);
671 
672   // The following two-letter aliases are only provided for readobj, as readelf
673   // allows single-letter args to be grouped together.
674   static cl::alias SectionRelocationsShort(
675       "sr", cl::desc("Alias for --section-relocations"),
676       cl::aliasopt(opts::SectionRelocations));
677   static cl::alias SectionDataShort("sd", cl::desc("Alias for --section-data"),
678                                     cl::aliasopt(opts::SectionData));
679   static cl::alias SectionSymbolsShort("st",
680                                        cl::desc("Alias for --section-symbols"),
681                                        cl::aliasopt(opts::SectionSymbols));
682   static cl::alias DynamicSymbolsShort("dt",
683                                        cl::desc("Alias for --dyn-symbols"),
684                                        cl::aliasopt(opts::DynamicSymbols));
685 }
686 
687 /// Registers aliases that should only be allowed by readelf.
688 static void registerReadelfAliases() {
689   // -s is here because for readobj it means --sections.
690   static cl::alias SymbolsShort("s", cl::desc("Alias for --symbols"),
691                                 cl::aliasopt(opts::Symbols), cl::NotHidden,
692                                 cl::Grouping);
693 
694   // Allow all single letter flags to be grouped together.
695   for (auto &OptEntry : cl::getRegisteredOptions()) {
696     StringRef ArgName = OptEntry.getKey();
697     cl::Option *Option = OptEntry.getValue();
698     if (ArgName.size() == 1)
699       apply(Option, cl::Grouping);
700   }
701 }
702 
703 int main(int argc, const char *argv[]) {
704   InitLLVM X(argc, argv);
705 
706   // Register the target printer for --version.
707   cl::AddExtraVersionPrinter(TargetRegistry::printRegisteredTargetsForVersion);
708 
709   if (sys::path::stem(argv[0]).contains("readelf")) {
710     opts::Output = opts::GNU;
711     registerReadelfAliases();
712   } else {
713     registerReadobjAliases();
714   }
715 
716   cl::ParseCommandLineOptions(argc, argv, "LLVM Object Reader\n");
717 
718   if (opts::All) {
719     opts::FileHeaders = true;
720     opts::ProgramHeaders = true;
721     opts::SectionHeaders = true;
722     opts::Symbols = true;
723     opts::Relocations = true;
724     opts::DynamicTable = true;
725     opts::Notes = true;
726     opts::VersionInfo = true;
727     opts::UnwindInfo = true;
728     opts::SectionGroups = true;
729     opts::HashHistogram = true;
730   }
731 
732   if (opts::Headers) {
733     opts::FileHeaders = true;
734     opts::ProgramHeaders = true;
735     opts::SectionHeaders = true;
736   }
737 
738   // Default to stdin if no filename is specified.
739   if (opts::InputFilenames.empty())
740     opts::InputFilenames.push_back("-");
741 
742   ScopedPrinter Writer(fouts());
743   for (const std::string &I : opts::InputFilenames)
744     dumpInput(I, Writer);
745 
746   if (opts::CodeViewMergedTypes) {
747     if (opts::CodeViewEnableGHash)
748       dumpCodeViewMergedTypes(Writer, CVTypes.GlobalIDTable.records(),
749                               CVTypes.GlobalTypeTable.records());
750     else
751       dumpCodeViewMergedTypes(Writer, CVTypes.IDTable.records(),
752                               CVTypes.TypeTable.records());
753   }
754 
755   return 0;
756 }
757