llvm-readobj: add support to dump COFF export tables
[oota-llvm.git] / tools / llvm-readobj / llvm-readobj.cpp
1 //===- llvm-readobj.cpp - Dump contents of an Object File -----------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This is a tool similar to readelf, except it works on multiple object file
11 // formats. The main purpose of this tool is to provide detailed output suitable
12 // for FileCheck.
13 //
14 // Flags should be similar to readelf where supported, but the output format
15 // does not need to be identical. The point is to not make users learn yet
16 // another set of flags.
17 //
18 // Output should be specialized for each format where appropriate.
19 //
20 //===----------------------------------------------------------------------===//
21
22 #include "llvm-readobj.h"
23 #include "Error.h"
24 #include "ObjDumper.h"
25 #include "StreamWriter.h"
26 #include "llvm/Object/Archive.h"
27 #include "llvm/Object/ELFObjectFile.h"
28 #include "llvm/Object/ObjectFile.h"
29 #include "llvm/Support/Casting.h"
30 #include "llvm/Support/CommandLine.h"
31 #include "llvm/Support/DataTypes.h"
32 #include "llvm/Support/Debug.h"
33 #include "llvm/Support/FileSystem.h"
34 #include "llvm/Support/ManagedStatic.h"
35 #include "llvm/Support/PrettyStackTrace.h"
36 #include "llvm/Support/Signals.h"
37 #include "llvm/Support/TargetRegistry.h"
38 #include "llvm/Support/TargetSelect.h"
39 #include <string>
40 #include <system_error>
41
42
43 using namespace llvm;
44 using namespace llvm::object;
45
46 namespace opts {
47   cl::list<std::string> InputFilenames(cl::Positional,
48     cl::desc("<input object files>"),
49     cl::ZeroOrMore);
50
51   // -file-headers, -h
52   cl::opt<bool> FileHeaders("file-headers",
53     cl::desc("Display file headers "));
54   cl::alias FileHeadersShort("h",
55     cl::desc("Alias for --file-headers"),
56     cl::aliasopt(FileHeaders));
57
58   // -sections, -s
59   cl::opt<bool> Sections("sections",
60     cl::desc("Display all sections."));
61   cl::alias SectionsShort("s",
62     cl::desc("Alias for --sections"),
63     cl::aliasopt(Sections));
64
65   // -section-relocations, -sr
66   cl::opt<bool> SectionRelocations("section-relocations",
67     cl::desc("Display relocations for each section shown."));
68   cl::alias SectionRelocationsShort("sr",
69     cl::desc("Alias for --section-relocations"),
70     cl::aliasopt(SectionRelocations));
71
72   // -section-symbols, -st
73   cl::opt<bool> SectionSymbols("section-symbols",
74     cl::desc("Display symbols for each section shown."));
75   cl::alias SectionSymbolsShort("st",
76     cl::desc("Alias for --section-symbols"),
77     cl::aliasopt(SectionSymbols));
78
79   // -section-data, -sd
80   cl::opt<bool> SectionData("section-data",
81     cl::desc("Display section data for each section shown."));
82   cl::alias SectionDataShort("sd",
83     cl::desc("Alias for --section-data"),
84     cl::aliasopt(SectionData));
85
86   // -relocations, -r
87   cl::opt<bool> Relocations("relocations",
88     cl::desc("Display the relocation entries in the file"));
89   cl::alias RelocationsShort("r",
90     cl::desc("Alias for --relocations"),
91     cl::aliasopt(Relocations));
92
93   // -symbols, -t
94   cl::opt<bool> Symbols("symbols",
95     cl::desc("Display the symbol table"));
96   cl::alias SymbolsShort("t",
97     cl::desc("Alias for --symbols"),
98     cl::aliasopt(Symbols));
99
100   // -dyn-symbols, -dt
101   cl::opt<bool> DynamicSymbols("dyn-symbols",
102     cl::desc("Display the dynamic symbol table"));
103   cl::alias DynamicSymbolsShort("dt",
104     cl::desc("Alias for --dyn-symbols"),
105     cl::aliasopt(DynamicSymbols));
106
107   // -unwind, -u
108   cl::opt<bool> UnwindInfo("unwind",
109     cl::desc("Display unwind information"));
110   cl::alias UnwindInfoShort("u",
111     cl::desc("Alias for --unwind"),
112     cl::aliasopt(UnwindInfo));
113
114   // -dynamic-table
115   cl::opt<bool> DynamicTable("dynamic-table",
116     cl::desc("Display the ELF .dynamic section table"));
117
118   // -needed-libs
119   cl::opt<bool> NeededLibraries("needed-libs",
120     cl::desc("Display the needed libraries"));
121
122   // -program-headers
123   cl::opt<bool> ProgramHeaders("program-headers",
124     cl::desc("Display ELF program headers"));
125
126   // -expand-relocs
127   cl::opt<bool> ExpandRelocs("expand-relocs",
128     cl::desc("Expand each shown relocation to multiple lines"));
129
130   // -codeview-linetables
131   cl::opt<bool> CodeViewLineTables("codeview-linetables",
132     cl::desc("Display CodeView line table information"));
133
134   // -arm-attributes, -a
135   cl::opt<bool> ARMAttributes("arm-attributes",
136                               cl::desc("Display the ARM attributes section"));
137   cl::alias ARMAttributesShort("-a", cl::desc("Alias for --arm-attributes"),
138                                cl::aliasopt(ARMAttributes));
139
140   // -mips-plt-got
141   cl::opt<bool>
142   MipsPLTGOT("mips-plt-got",
143              cl::desc("Display the MIPS GOT and PLT GOT sections"));
144
145   // -coff-imports
146   cl::opt<bool>
147   COFFImports("coff-imports", cl::desc("Display the PE/COFF import table"));
148
149   // -coff-exports
150   cl::opt<bool>
151   COFFExports("coff-exports", cl::desc("Display the PE/COFF export table"));
152
153   // -coff-directives
154   cl::opt<bool>
155   COFFDirectives("coff-directives",
156                  cl::desc("Display the PE/COFF .drectve section"));
157
158   // -coff-basereloc
159   cl::opt<bool>
160   COFFBaseRelocs("coff-basereloc",
161                  cl::desc("Display the PE/COFF .reloc section"));
162 } // namespace opts
163
164 static int ReturnValue = EXIT_SUCCESS;
165
166 namespace llvm {
167
168 bool error(std::error_code EC) {
169   if (!EC)
170     return false;
171
172   ReturnValue = EXIT_FAILURE;
173   outs() << "\nError reading file: " << EC.message() << ".\n";
174   outs().flush();
175   return true;
176 }
177
178 bool relocAddressLess(RelocationRef a, RelocationRef b) {
179   uint64_t a_addr, b_addr;
180   if (error(a.getOffset(a_addr))) exit(ReturnValue);
181   if (error(b.getOffset(b_addr))) exit(ReturnValue);
182   return a_addr < b_addr;
183 }
184
185 } // namespace llvm
186
187 static void reportError(StringRef Input, std::error_code EC) {
188   if (Input == "-")
189     Input = "<stdin>";
190
191   errs() << Input << ": " << EC.message() << "\n";
192   errs().flush();
193   ReturnValue = EXIT_FAILURE;
194 }
195
196 static void reportError(StringRef Input, StringRef Message) {
197   if (Input == "-")
198     Input = "<stdin>";
199
200   errs() << Input << ": " << Message << "\n";
201   ReturnValue = EXIT_FAILURE;
202 }
203
204 static bool isMipsArch(unsigned Arch) {
205   switch (Arch) {
206   case llvm::Triple::mips:
207   case llvm::Triple::mipsel:
208   case llvm::Triple::mips64:
209   case llvm::Triple::mips64el:
210     return true;
211   default:
212     return false;
213   }
214 }
215
216 /// @brief Creates an format-specific object file dumper.
217 static std::error_code createDumper(const ObjectFile *Obj, StreamWriter &Writer,
218                                     std::unique_ptr<ObjDumper> &Result) {
219   if (!Obj)
220     return readobj_error::unsupported_file_format;
221
222   if (Obj->isCOFF())
223     return createCOFFDumper(Obj, Writer, Result);
224   if (Obj->isELF())
225     return createELFDumper(Obj, Writer, Result);
226   if (Obj->isMachO())
227     return createMachODumper(Obj, Writer, Result);
228
229   return readobj_error::unsupported_obj_file_format;
230 }
231
232 static StringRef getLoadName(const ObjectFile *Obj) {
233   if (auto *ELF = dyn_cast<ELF32LEObjectFile>(Obj))
234     return ELF->getLoadName();
235   if (auto *ELF = dyn_cast<ELF64LEObjectFile>(Obj))
236     return ELF->getLoadName();
237   if (auto *ELF = dyn_cast<ELF32BEObjectFile>(Obj))
238     return ELF->getLoadName();
239   if (auto *ELF = dyn_cast<ELF64BEObjectFile>(Obj))
240     return ELF->getLoadName();
241   llvm_unreachable("Not ELF");
242 }
243
244 /// @brief Dumps the specified object file.
245 static void dumpObject(const ObjectFile *Obj) {
246   StreamWriter Writer(outs());
247   std::unique_ptr<ObjDumper> Dumper;
248   if (std::error_code EC = createDumper(Obj, Writer, Dumper)) {
249     reportError(Obj->getFileName(), EC);
250     return;
251   }
252
253   outs() << '\n';
254   outs() << "File: " << Obj->getFileName() << "\n";
255   outs() << "Format: " << Obj->getFileFormatName() << "\n";
256   outs() << "Arch: "
257          << Triple::getArchTypeName((llvm::Triple::ArchType)Obj->getArch())
258          << "\n";
259   outs() << "AddressSize: " << (8*Obj->getBytesInAddress()) << "bit\n";
260   if (Obj->isELF())
261     outs() << "LoadName: " << getLoadName(Obj) << "\n";
262
263   if (opts::FileHeaders)
264     Dumper->printFileHeaders();
265   if (opts::Sections)
266     Dumper->printSections();
267   if (opts::Relocations)
268     Dumper->printRelocations();
269   if (opts::Symbols)
270     Dumper->printSymbols();
271   if (opts::DynamicSymbols)
272     Dumper->printDynamicSymbols();
273   if (opts::UnwindInfo)
274     Dumper->printUnwindInfo();
275   if (opts::DynamicTable)
276     Dumper->printDynamicTable();
277   if (opts::NeededLibraries)
278     Dumper->printNeededLibraries();
279   if (opts::ProgramHeaders)
280     Dumper->printProgramHeaders();
281   if (Obj->getArch() == llvm::Triple::arm && Obj->isELF())
282     if (opts::ARMAttributes)
283       Dumper->printAttributes();
284   if (isMipsArch(Obj->getArch()) && Obj->isELF())
285     if (opts::MipsPLTGOT)
286       Dumper->printMipsPLTGOT();
287   if (opts::COFFImports)
288     Dumper->printCOFFImports();
289   if (opts::COFFExports)
290     Dumper->printCOFFExports();
291   if (opts::COFFDirectives)
292     Dumper->printCOFFDirectives();
293   if (opts::COFFBaseRelocs)
294     Dumper->printCOFFBaseReloc();
295 }
296
297
298 /// @brief Dumps each object file in \a Arc;
299 static void dumpArchive(const Archive *Arc) {
300   for (Archive::child_iterator ArcI = Arc->child_begin(),
301                                ArcE = Arc->child_end();
302                                ArcI != ArcE; ++ArcI) {
303     ErrorOr<std::unique_ptr<Binary>> ChildOrErr = ArcI->getAsBinary();
304     if (std::error_code EC = ChildOrErr.getError()) {
305       // Ignore non-object files.
306       if (EC != object_error::invalid_file_type)
307         reportError(Arc->getFileName(), EC.message());
308       continue;
309     }
310
311     if (ObjectFile *Obj = dyn_cast<ObjectFile>(&*ChildOrErr.get()))
312       dumpObject(Obj);
313     else
314       reportError(Arc->getFileName(), readobj_error::unrecognized_file_format);
315   }
316 }
317
318
319 /// @brief Opens \a File and dumps it.
320 static void dumpInput(StringRef File) {
321   // If file isn't stdin, check that it exists.
322   if (File != "-" && !sys::fs::exists(File)) {
323     reportError(File, readobj_error::file_not_found);
324     return;
325   }
326
327   // Attempt to open the binary.
328   ErrorOr<OwningBinary<Binary>> BinaryOrErr = createBinary(File);
329   if (std::error_code EC = BinaryOrErr.getError()) {
330     reportError(File, EC);
331     return;
332   }
333   Binary &Binary = *BinaryOrErr.get().getBinary();
334
335   if (Archive *Arc = dyn_cast<Archive>(&Binary))
336     dumpArchive(Arc);
337   else if (ObjectFile *Obj = dyn_cast<ObjectFile>(&Binary))
338     dumpObject(Obj);
339   else
340     reportError(File, readobj_error::unrecognized_file_format);
341 }
342
343
344 int main(int argc, const char *argv[]) {
345   sys::PrintStackTraceOnErrorSignal();
346   PrettyStackTraceProgram X(argc, argv);
347   llvm_shutdown_obj Y;
348
349   // Initialize targets.
350   llvm::InitializeAllTargetInfos();
351
352   // Register the target printer for --version.
353   cl::AddExtraVersionPrinter(TargetRegistry::printRegisteredTargetsForVersion);
354
355   cl::ParseCommandLineOptions(argc, argv, "LLVM Object Reader\n");
356
357   // Default to stdin if no filename is specified.
358   if (opts::InputFilenames.size() == 0)
359     opts::InputFilenames.push_back("-");
360
361   std::for_each(opts::InputFilenames.begin(), opts::InputFilenames.end(),
362                 dumpInput);
363
364   return ReturnValue;
365 }