This patch adds a new flag "-coff-imports" to llvm-readobj.
[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 } // namespace opts
149
150 static int ReturnValue = EXIT_SUCCESS;
151
152 namespace llvm {
153
154 bool error(std::error_code EC) {
155   if (!EC)
156     return false;
157
158   ReturnValue = EXIT_FAILURE;
159   outs() << "\nError reading file: " << EC.message() << ".\n";
160   outs().flush();
161   return true;
162 }
163
164 bool relocAddressLess(RelocationRef a, RelocationRef b) {
165   uint64_t a_addr, b_addr;
166   if (error(a.getOffset(a_addr))) return false;
167   if (error(b.getOffset(b_addr))) return false;
168   return a_addr < b_addr;
169 }
170
171 } // namespace llvm
172
173 static void reportError(StringRef Input, std::error_code EC) {
174   if (Input == "-")
175     Input = "<stdin>";
176
177   errs() << Input << ": " << EC.message() << "\n";
178   errs().flush();
179   ReturnValue = EXIT_FAILURE;
180 }
181
182 static void reportError(StringRef Input, StringRef Message) {
183   if (Input == "-")
184     Input = "<stdin>";
185
186   errs() << Input << ": " << Message << "\n";
187   ReturnValue = EXIT_FAILURE;
188 }
189
190 static bool isMipsArch(unsigned Arch) {
191   switch (Arch) {
192   case llvm::Triple::mips:
193   case llvm::Triple::mipsel:
194   case llvm::Triple::mips64:
195   case llvm::Triple::mips64el:
196     return true;
197   default:
198     return false;
199   }
200 }
201
202 /// @brief Creates an format-specific object file dumper.
203 static std::error_code createDumper(const ObjectFile *Obj, StreamWriter &Writer,
204                                     std::unique_ptr<ObjDumper> &Result) {
205   if (!Obj)
206     return readobj_error::unsupported_file_format;
207
208   if (Obj->isCOFF())
209     return createCOFFDumper(Obj, Writer, Result);
210   if (Obj->isELF())
211     return createELFDumper(Obj, Writer, Result);
212   if (Obj->isMachO())
213     return createMachODumper(Obj, Writer, Result);
214
215   return readobj_error::unsupported_obj_file_format;
216 }
217
218 static StringRef getLoadName(const ObjectFile *Obj) {
219   if (auto *ELF = dyn_cast<ELF32LEObjectFile>(Obj))
220     return ELF->getLoadName();
221   if (auto *ELF = dyn_cast<ELF64LEObjectFile>(Obj))
222     return ELF->getLoadName();
223   if (auto *ELF = dyn_cast<ELF32BEObjectFile>(Obj))
224     return ELF->getLoadName();
225   if (auto *ELF = dyn_cast<ELF64BEObjectFile>(Obj))
226     return ELF->getLoadName();
227   llvm_unreachable("Not ELF");
228 }
229
230 /// @brief Dumps the specified object file.
231 static void dumpObject(const ObjectFile *Obj) {
232   StreamWriter Writer(outs());
233   std::unique_ptr<ObjDumper> Dumper;
234   if (std::error_code EC = createDumper(Obj, Writer, Dumper)) {
235     reportError(Obj->getFileName(), EC);
236     return;
237   }
238
239   outs() << '\n';
240   outs() << "File: " << Obj->getFileName() << "\n";
241   outs() << "Format: " << Obj->getFileFormatName() << "\n";
242   outs() << "Arch: "
243          << Triple::getArchTypeName((llvm::Triple::ArchType)Obj->getArch())
244          << "\n";
245   outs() << "AddressSize: " << (8*Obj->getBytesInAddress()) << "bit\n";
246   if (Obj->isELF())
247     outs() << "LoadName: " << getLoadName(Obj) << "\n";
248
249   if (opts::FileHeaders)
250     Dumper->printFileHeaders();
251   if (opts::Sections)
252     Dumper->printSections();
253   if (opts::Relocations)
254     Dumper->printRelocations();
255   if (opts::Symbols)
256     Dumper->printSymbols();
257   if (opts::DynamicSymbols)
258     Dumper->printDynamicSymbols();
259   if (opts::UnwindInfo)
260     Dumper->printUnwindInfo();
261   if (opts::DynamicTable)
262     Dumper->printDynamicTable();
263   if (opts::NeededLibraries)
264     Dumper->printNeededLibraries();
265   if (opts::ProgramHeaders)
266     Dumper->printProgramHeaders();
267   if (Obj->getArch() == llvm::Triple::arm && Obj->isELF())
268     if (opts::ARMAttributes)
269       Dumper->printAttributes();
270   if (isMipsArch(Obj->getArch()) && Obj->isELF())
271     if (opts::MipsPLTGOT)
272       Dumper->printMipsPLTGOT();
273   if (opts::COFFImports)
274     Dumper->printCOFFImports();
275 }
276
277
278 /// @brief Dumps each object file in \a Arc;
279 static void dumpArchive(const Archive *Arc) {
280   for (Archive::child_iterator ArcI = Arc->child_begin(),
281                                ArcE = Arc->child_end();
282                                ArcI != ArcE; ++ArcI) {
283     ErrorOr<std::unique_ptr<Binary>> ChildOrErr = ArcI->getAsBinary();
284     if (std::error_code EC = ChildOrErr.getError()) {
285       // Ignore non-object files.
286       if (EC != object_error::invalid_file_type)
287         reportError(Arc->getFileName(), EC.message());
288       continue;
289     }
290
291     if (ObjectFile *Obj = dyn_cast<ObjectFile>(&*ChildOrErr.get()))
292       dumpObject(Obj);
293     else
294       reportError(Arc->getFileName(), readobj_error::unrecognized_file_format);
295   }
296 }
297
298
299 /// @brief Opens \a File and dumps it.
300 static void dumpInput(StringRef File) {
301   // If file isn't stdin, check that it exists.
302   if (File != "-" && !sys::fs::exists(File)) {
303     reportError(File, readobj_error::file_not_found);
304     return;
305   }
306
307   // Attempt to open the binary.
308   ErrorOr<OwningBinary<Binary>> BinaryOrErr = createBinary(File);
309   if (std::error_code EC = BinaryOrErr.getError()) {
310     reportError(File, EC);
311     return;
312   }
313   Binary &Binary = *BinaryOrErr.get().getBinary();
314
315   if (Archive *Arc = dyn_cast<Archive>(&Binary))
316     dumpArchive(Arc);
317   else if (ObjectFile *Obj = dyn_cast<ObjectFile>(&Binary))
318     dumpObject(Obj);
319   else
320     reportError(File, readobj_error::unrecognized_file_format);
321 }
322
323
324 int main(int argc, const char *argv[]) {
325   sys::PrintStackTraceOnErrorSignal();
326   PrettyStackTraceProgram X(argc, argv);
327   llvm_shutdown_obj Y;
328
329   // Initialize targets.
330   llvm::InitializeAllTargetInfos();
331
332   // Register the target printer for --version.
333   cl::AddExtraVersionPrinter(TargetRegistry::printRegisteredTargetsForVersion);
334
335   cl::ParseCommandLineOptions(argc, argv, "LLVM Object Reader\n");
336
337   // Default to stdin if no filename is specified.
338   if (opts::InputFilenames.size() == 0)
339     opts::InputFilenames.push_back("-");
340
341   std::for_each(opts::InputFilenames.begin(), opts::InputFilenames.end(),
342                 dumpInput);
343
344   return ReturnValue;
345 }