forked from OSchip/llvm-project
87 lines
2.4 KiB
C++
87 lines
2.4 KiB
C++
//===------ utils/obj2yaml.cpp - obj2yaml conversion tool -------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "obj2yaml.h"
|
|
#include "llvm/ADT/OwningPtr.h"
|
|
#include "llvm/Object/Archive.h"
|
|
#include "llvm/Object/COFF.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/ManagedStatic.h"
|
|
#include "llvm/Support/PrettyStackTrace.h"
|
|
#include "llvm/Support/Signals.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace objyaml { // generic yaml-writing specific routines
|
|
|
|
unsigned char printable(unsigned char Ch) {
|
|
return Ch >= ' ' && Ch <= '~' ? Ch : '.';
|
|
}
|
|
|
|
raw_ostream &writeHexStream(raw_ostream &Out, const ArrayRef<uint8_t> arr) {
|
|
const char *hex = "0123456789ABCDEF";
|
|
Out << " !hex \"";
|
|
|
|
typedef ArrayRef<uint8_t>::const_iterator iter_t;
|
|
const iter_t end = arr.end();
|
|
for (iter_t iter = arr.begin(); iter != end; ++iter)
|
|
Out << hex[(*iter >> 4) & 0x0F] << hex[(*iter & 0x0F)];
|
|
|
|
Out << "\" # |";
|
|
for (iter_t iter = arr.begin(); iter != end; ++iter)
|
|
Out << printable(*iter);
|
|
Out << "|\n";
|
|
|
|
return Out;
|
|
}
|
|
|
|
raw_ostream &writeHexNumber(raw_ostream &Out, unsigned long long N) {
|
|
if (N >= 10)
|
|
Out << "0x";
|
|
Out.write_hex(N);
|
|
return Out;
|
|
}
|
|
|
|
} // end namespace yaml
|
|
|
|
namespace {
|
|
enum ObjectFileType {
|
|
coff
|
|
};
|
|
}
|
|
|
|
cl::opt<ObjectFileType> InputFormat(
|
|
cl::desc("Choose input format"),
|
|
cl::values(clEnumVal(coff, "process COFF object files"), clEnumValEnd));
|
|
|
|
cl::opt<std::string> InputFilename(cl::Positional, cl::desc("<input file>"),
|
|
cl::init("-"));
|
|
|
|
int main(int argc, char *argv[]) {
|
|
cl::ParseCommandLineOptions(argc, argv);
|
|
sys::PrintStackTraceOnErrorSignal();
|
|
PrettyStackTraceProgram X(argc, argv);
|
|
llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
|
|
|
|
// Process the input file
|
|
OwningPtr<MemoryBuffer> buf;
|
|
|
|
// TODO: If this is an archive, then burst it and dump each entry
|
|
if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputFilename, buf)) {
|
|
errs() << "Error: '" << ec.message() << "' opening file '" << InputFilename
|
|
<< "'\n";
|
|
} else {
|
|
ec = coff2yaml(outs(), buf.take());
|
|
if (ec)
|
|
errs() << "Error: " << ec.message() << " dumping COFF file\n";
|
|
}
|
|
|
|
return 0;
|
|
}
|