1//===------ utils/obj2yaml.cpp - obj2yaml conversion tool -------*- C++ -*-===//
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#include "Error.h"
11#include "obj2yaml.h"
12#include "llvm/Object/Archive.h"
13#include "llvm/Object/COFF.h"
14#include "llvm/Support/CommandLine.h"
15#include "llvm/Support/ManagedStatic.h"
16#include "llvm/Support/PrettyStackTrace.h"
17#include "llvm/Support/Signals.h"
18
19using namespace llvm;
20using namespace llvm::object;
21
22static std::error_code dumpObject(const ObjectFile &Obj) {
23  if (Obj.isCOFF())
24    return coff2yaml(outs(), cast<COFFObjectFile>(Obj));
25  if (Obj.isELF())
26    return elf2yaml(outs(), Obj);
27
28  return obj2yaml_error::unsupported_obj_file_format;
29}
30
31static std::error_code dumpInput(StringRef File) {
32  if (File != "-" && !sys::fs::exists(File))
33    return obj2yaml_error::file_not_found;
34
35  ErrorOr<Binary *> BinaryOrErr = createBinary(File);
36  if (std::error_code EC = BinaryOrErr.getError())
37    return EC;
38
39  std::unique_ptr<Binary> Binary(BinaryOrErr.get());
40  // TODO: If this is an archive, then burst it and dump each entry
41  if (ObjectFile *Obj = dyn_cast<ObjectFile>(Binary.get()))
42    return dumpObject(*Obj);
43
44  return obj2yaml_error::unrecognized_file_format;
45}
46
47cl::opt<std::string> InputFilename(cl::Positional, cl::desc("<input file>"),
48                                   cl::init("-"));
49
50int main(int argc, char *argv[]) {
51  cl::ParseCommandLineOptions(argc, argv);
52  sys::PrintStackTraceOnErrorSignal();
53  PrettyStackTraceProgram X(argc, argv);
54  llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
55
56  if (std::error_code EC = dumpInput(InputFilename)) {
57    errs() << "Error: '" << EC.message() << "'\n";
58    return 1;
59  }
60
61  return 0;
62}
63