llvm-mc.cpp revision a3dcfb130044f306632a5fab43854eda4095a09c
1bb4688a9cf385ccada90ebeb04b3ba5827bf213dChris Lattner//===-- llvm-mc.cpp - Machine Code Hacking Driver -------------------------===// 2f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner// 3f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner// The LLVM Compiler Infrastructure 4f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner// 5f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner// This file is distributed under the University of Illinois Open Source 6f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner// License. See LICENSE.TXT for details. 7f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner// 8f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner//===----------------------------------------------------------------------===// 9f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner// 10bb4688a9cf385ccada90ebeb04b3ba5827bf213dChris Lattner// This utility is a simple driver that allows command line hacking on machine 11bb4688a9cf385ccada90ebeb04b3ba5827bf213dChris Lattner// code. 12f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner// 13f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner//===----------------------------------------------------------------------===// 14f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 1590edac0e8b35f766599362b6301863229f0ddcdbChris Lattner#include "llvm/MC/MCAsmLexer.h" 16cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner#include "llvm/MC/MCContext.h" 174a0abd80f18f9c2a10bf5b14cd6731d51972a426Daniel Dunbar#include "llvm/MC/MCCodeEmitter.h" 1890edac0e8b35f766599362b6301863229f0ddcdbChris Lattner#include "llvm/MC/MCInstPrinter.h" 19f9bdeddb96043559c61f176f8077e3b91a0c544fChris Lattner#include "llvm/MC/MCSectionMachO.h" 20cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner#include "llvm/MC/MCStreamer.h" 21cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner#include "llvm/ADT/OwningPtr.h" 22f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/CommandLine.h" 23c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar#include "llvm/Support/FormattedStream.h" 24f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/ManagedStatic.h" 25f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/MemoryBuffer.h" 26f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/PrettyStackTrace.h" 27b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner#include "llvm/Support/SourceMgr.h" 28f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/raw_ostream.h" 29f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/System/Signals.h" 30a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar#include "llvm/Target/TargetAsmParser.h" 31b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar#include "llvm/Target/TargetRegistry.h" 3290edac0e8b35f766599362b6301863229f0ddcdbChris Lattner#include "llvm/Target/TargetMachine.h" // FIXME. 33b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar#include "llvm/Target/TargetSelect.h" 3427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner#include "AsmParser.h" 35a3dcfb130044f306632a5fab43854eda4095a09cChris Lattner#include "Disassembler.h" 36f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerusing namespace llvm; 37f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 38f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerstatic cl::opt<std::string> 39f9f065e45500823cdeb25bde2154d871ab6e9125Chris LattnerInputFilename(cl::Positional, cl::desc("<input file>"), cl::init("-")); 40f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 41f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerstatic cl::opt<std::string> 42f9f065e45500823cdeb25bde2154d871ab6e9125Chris LattnerOutputFilename("o", cl::desc("Output filename"), 43f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner cl::value_desc("filename")); 44f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 45f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbarstatic cl::opt<bool> 46f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel DunbarShowEncoding("show-encoding", cl::desc("Show instruction encodings")); 47f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbar 48e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattnerstatic cl::opt<unsigned> 49e895c6151589c1b7f6ac9ca992b76106fa197a37Chris LattnerOutputAsmVariant("output-asm-variant", 50e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattner cl::desc("Syntax variant to use for output printing")); 51e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattner 52fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbarenum OutputFileType { 53fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OFT_AssemblyFile, 54fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OFT_ObjectFile 55fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar}; 56fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbarstatic cl::opt<OutputFileType> 57fb4a6b397665df011348ade24a8e38d2219f095aDaniel DunbarFileType("filetype", cl::init(OFT_AssemblyFile), 58fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar cl::desc("Choose an output file type:"), 59fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar cl::values( 60fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValN(OFT_AssemblyFile, "asm", 61fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar "Emit an assembly ('.s') file"), 62fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValN(OFT_ObjectFile, "obj", 63fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar "Emit a native object ('.o') file"), 64fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValEnd)); 65fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar 66bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbarstatic cl::opt<bool> 67baa26395ccf17fc988bb9cf62d6659ca8415ece9Dan GohmanForce("f", cl::desc("Enable binary output on terminals")); 68bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 69b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerstatic cl::list<std::string> 70b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris LattnerIncludeDirs("I", cl::desc("Directory of include files"), 71b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner cl::value_desc("directory"), cl::Prefix); 72f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 73b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbarstatic cl::opt<std::string> 74ed1f1687b790b444c29c9cdc5ea646be0783c1cdNick LewyckyTripleName("triple", cl::desc("Target triple to assemble for, " 75ed1f1687b790b444c29c9cdc5ea646be0783c1cdNick Lewycky "see -version for available targets"), 76f4e748bc3f7090468f7b7e5b6e2ebe323c1e16fdNick Lewycky cl::init(LLVM_HOSTTRIPLE)); 77b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 78b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerenum ActionType { 7927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner AC_AsLex, 80ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan AC_Assemble, 81ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan AC_Disassemble 82b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner}; 83f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 84b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerstatic cl::opt<ActionType> 85b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris LattnerAction(cl::desc("Action to perform:"), 8627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner cl::init(AC_Assemble), 8727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner cl::values(clEnumValN(AC_AsLex, "as-lex", 8827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner "Lex tokens from a .s file"), 8927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner clEnumValN(AC_Assemble, "assemble", 90b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner "Assemble a .s file (default)"), 91ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan clEnumValN(AC_Disassemble, "disassemble", 92ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan "Disassemble strings of hex bytes"), 93b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner clEnumValEnd)); 94f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 959823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderbystatic const Target *GetTarget(const char *ProgName) { 969823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby // Get the target specific parser. 979823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby std::string Error; 989823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const Target *TheTarget = TargetRegistry::lookupTarget(TripleName, Error); 999823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (TheTarget) 1009823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return TheTarget; 1019823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 1029823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby errs() << ProgName << ": error: unable to get target for '" << TripleName 1039823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby << "', see --version and --triple.\n"; 1049823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return 0; 1059823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby} 1069823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 10727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattnerstatic int AsLexInput(const char *ProgName) { 108b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner std::string ErrorMessage; 109b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner MemoryBuffer *Buffer = MemoryBuffer::getFileOrSTDIN(InputFilename, 110b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner &ErrorMessage); 111f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner if (Buffer == 0) { 112b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner errs() << ProgName << ": "; 113f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner if (ErrorMessage.size()) 114f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner errs() << ErrorMessage << "\n"; 115f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner else 116f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner errs() << "input file didn't read correctly.\n"; 117f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner return 1; 118f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner } 119b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 120b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SourceMgr SrcMgr; 121b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 122b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Tell SrcMgr about this buffer, which is what TGParser will pick up. 123b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SrcMgr.AddNewSourceBuffer(Buffer, SMLoc()); 124b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 125b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Record the location of the include directories so that the lexer can find 126b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // it later. 127b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SrcMgr.setIncludeDirs(IncludeDirs); 128a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner 1299823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const Target *TheTarget = GetTarget(ProgName); 1309823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (!TheTarget) 1319823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return 1; 1329823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 1339823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const MCAsmInfo *MAI = TheTarget->createAsmInfo(TripleName); 1349823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby assert(MAI && "Unable to create target asm info!"); 1359823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 1369823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby AsmLexer Lexer(SrcMgr, *MAI); 137f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 13827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner bool Error = false; 13927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 140a3c924f83aefd4c883ed17a200876d258e0ad1e2Daniel Dunbar while (Lexer.Lex().isNot(AsmToken::Eof)) { 141a3c924f83aefd4c883ed17a200876d258e0ad1e2Daniel Dunbar switch (Lexer.getKind()) { 14227aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner default: 1433fb7683bec8c8edb24e80c95f3b0668c6ecc0ae6Daniel Dunbar Lexer.PrintMessage(Lexer.getLoc(), "unknown token", "warning"); 14427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner Error = true; 14527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner break; 1463f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Error: 14727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner Error = true; // error already printed. 1484651bca31bdad27184fa0d36640bf5ef1d83cf5cChris Lattner break; 1493f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Identifier: 150419adedaa1638fbe4e078c997f81e94327ebff5aDaniel Dunbar outs() << "identifier: " << Lexer.getTok().getString() << '\n'; 151a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner break; 1523f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::String: 153419adedaa1638fbe4e078c997f81e94327ebff5aDaniel Dunbar outs() << "string: " << Lexer.getTok().getString() << '\n'; 15410a907d70fb54c40eecabb889e81c79b44092221Chris Lattner break; 1553f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Integer: 156419adedaa1638fbe4e078c997f81e94327ebff5aDaniel Dunbar outs() << "int: " << Lexer.getTok().getString() << '\n'; 157a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner break; 158165e8344d833147268810bbec18276dafe0122b9Daniel Dunbar 1593f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Amp: outs() << "Amp\n"; break; 1603f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::AmpAmp: outs() << "AmpAmp\n"; break; 1613f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Caret: outs() << "Caret\n"; break; 1623f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Colon: outs() << "Colon\n"; break; 1633f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Comma: outs() << "Comma\n"; break; 1643f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Dollar: outs() << "Dollar\n"; break; 1653f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::EndOfStatement: outs() << "EndOfStatement\n"; break; 1663f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Eof: outs() << "Eof\n"; break; 1673f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Equal: outs() << "Equal\n"; break; 1683f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::EqualEqual: outs() << "EqualEqual\n"; break; 1693f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Exclaim: outs() << "Exclaim\n"; break; 1703f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::ExclaimEqual: outs() << "ExclaimEqual\n"; break; 1713f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Greater: outs() << "Greater\n"; break; 1723f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::GreaterEqual: outs() << "GreaterEqual\n"; break; 1733f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::GreaterGreater: outs() << "GreaterGreater\n"; break; 1743f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::LParen: outs() << "LParen\n"; break; 1753f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Less: outs() << "Less\n"; break; 1763f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::LessEqual: outs() << "LessEqual\n"; break; 1773f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::LessGreater: outs() << "LessGreater\n"; break; 1783f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::LessLess: outs() << "LessLess\n"; break; 1793f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Minus: outs() << "Minus\n"; break; 1803f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Percent: outs() << "Percent\n"; break; 1813f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Pipe: outs() << "Pipe\n"; break; 1823f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::PipePipe: outs() << "PipePipe\n"; break; 1833f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Plus: outs() << "Plus\n"; break; 1843f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::RParen: outs() << "RParen\n"; break; 1853f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Slash: outs() << "Slash\n"; break; 1863f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Star: outs() << "Star\n"; break; 1873f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Tilde: outs() << "Tilde\n"; break; 188a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner } 189a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner } 190f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 19127aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return Error; 192b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner} 193b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 194c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbarstatic formatted_raw_ostream *GetOutputStream() { 19517e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner if (OutputFilename == "") 19617e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner OutputFilename = "-"; 197bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 198bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar // Make sure that the Out file gets unlinked from the disk if we get a 19917e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner // SIGINT. 20017e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner if (OutputFilename != "-") 20117e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner sys::RemoveFileOnSignal(sys::Path(OutputFilename)); 202bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 203bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar std::string Err; 20417e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner raw_fd_ostream *Out = new raw_fd_ostream(OutputFilename.c_str(), Err, 205baa26395ccf17fc988bb9cf62d6659ca8415ece9Dan Gohman raw_fd_ostream::F_Binary); 206bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar if (!Err.empty()) { 207bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar errs() << Err << '\n'; 208bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar delete Out; 209bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar return 0; 210bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar } 211bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 212c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar return new formatted_raw_ostream(*Out, formatted_raw_ostream::DELETE_STREAM); 213bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar} 214bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 215a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbarstatic int AssembleInput(const char *ProgName) { 216c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar const Target *TheTarget = GetTarget(ProgName); 217c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (!TheTarget) 218c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar return 1; 219c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 220a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar std::string Error; 221a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar MemoryBuffer *Buffer = MemoryBuffer::getFileOrSTDIN(InputFilename, &Error); 22227aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner if (Buffer == 0) { 22327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner errs() << ProgName << ": "; 224a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar if (Error.size()) 225a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar errs() << Error << "\n"; 22627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner else 22727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner errs() << "input file didn't read correctly.\n"; 22827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return 1; 22927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner } 23027aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 23127aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SourceMgr SrcMgr; 23227aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 23312a8a447a4ca5b939f255ae92e7fef490344593aDaniel Dunbar // Tell SrcMgr about this buffer, which is what the parser will pick up. 23427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SrcMgr.AddNewSourceBuffer(Buffer, SMLoc()); 23527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 23627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner // Record the location of the include directories so that the lexer can find 23727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner // it later. 23827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SrcMgr.setIncludeDirs(IncludeDirs); 23927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 240cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner MCContext Ctx; 241c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar formatted_raw_ostream *Out = GetOutputStream(); 242bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar if (!Out) 243bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar return 1; 244c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 245c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 246c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar // FIXME: We shouldn't need to do this (and link in codegen). 247c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar OwningPtr<TargetMachine> TM(TheTarget->createTargetMachine(TripleName, "")); 248c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 249f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner if (!TM) { 250f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner errs() << ProgName << ": error: could not create target for triple '" 251f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner << TripleName << "'.\n"; 252f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner return 1; 253c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar } 254c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 25590edac0e8b35f766599362b6301863229f0ddcdbChris Lattner OwningPtr<MCInstPrinter> IP; 2564a0abd80f18f9c2a10bf5b14cd6731d51972a426Daniel Dunbar OwningPtr<MCCodeEmitter> CE; 257fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OwningPtr<MCStreamer> Str; 258f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner 2599823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const MCAsmInfo *MAI = TheTarget->createAsmInfo(TripleName); 2609823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby assert(MAI && "Unable to create target asm info!"); 261fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar 2629823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (FileType == OFT_AssemblyFile) { 263e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattner IP.reset(TheTarget->createMCInstPrinter(OutputAsmVariant, *MAI, *Out)); 264f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbar if (ShowEncoding) 265f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbar CE.reset(TheTarget->createCodeEmitter(*TM)); 26690edac0e8b35f766599362b6301863229f0ddcdbChris Lattner Str.reset(createAsmStreamer(Ctx, *Out, *MAI, IP.get(), CE.get())); 267fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar } else { 268fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar assert(FileType == OFT_ObjectFile && "Invalid file type!"); 2694fac74950a1ff08b995b366bfb84369c1507faefDaniel Dunbar CE.reset(TheTarget->createCodeEmitter(*TM)); 2704fac74950a1ff08b995b366bfb84369c1507faefDaniel Dunbar Str.reset(createMachOStreamer(Ctx, *Out, CE.get())); 271fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar } 272a0d1426af0bd05e1ae69481cdb75d2913e7e1ac1Daniel Dunbar 2739823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby AsmParser Parser(SrcMgr, Ctx, *Str.get(), *MAI); 274c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar OwningPtr<TargetAsmParser> TAP(TheTarget->createAsmParser(Parser)); 275c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (!TAP) { 276c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar errs() << ProgName 277c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar << ": error: this target does not support assembly parsing.\n"; 278a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar return 1; 279c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar } 280c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 28116cdcb38b2a5c5cdc216f9abafabd20e1ef7a254Daniel Dunbar Parser.setTargetParser(*TAP.get()); 282bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 283bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar int Res = Parser.Run(); 284c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (Out != &fouts()) 285bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar delete Out; 286bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 287bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar return Res; 288ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan} 289ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 290ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callananstatic int DisassembleInput(const char *ProgName) { 291ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan std::string Error; 292ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan const Target *TheTarget = TargetRegistry::lookupTarget(TripleName, Error); 293ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan if (TheTarget == 0) { 294ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan errs() << ProgName << ": error: unable to get target for '" << TripleName 295ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan << "', see --version and --triple.\n"; 296ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan return 0; 297ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan } 298ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 299ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan std::string ErrorMessage; 300ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 301ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan MemoryBuffer *Buffer = MemoryBuffer::getFileOrSTDIN(InputFilename, 302ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan &ErrorMessage); 303ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 304ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan if (Buffer == 0) { 305ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan errs() << ProgName << ": "; 306ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan if (ErrorMessage.size()) 307ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan errs() << ErrorMessage << "\n"; 308ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan else 309ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan errs() << "input file didn't read correctly.\n"; 310ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan return 1; 311ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan } 312ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 313a3dcfb130044f306632a5fab43854eda4095a09cChris Lattner return Disassembler::disassemble(*TheTarget, TripleName, *Buffer); 314ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan} 31527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 316b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 317b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerint main(int argc, char **argv) { 318b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Print a stack trace if we signal out. 319b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner sys::PrintStackTraceOnErrorSignal(); 320b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner PrettyStackTraceProgram X(argc, argv); 321b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. 322b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 323c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar // Initialize targets and assembly printers/parsers. 324b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar llvm::InitializeAllTargetInfos(); 325c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar // FIXME: We shouldn't need to initialize the Target(Machine)s. 326c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar llvm::InitializeAllTargets(); 327c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar llvm::InitializeAllAsmPrinters(); 328b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar llvm::InitializeAllAsmParsers(); 329ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan llvm::InitializeAllDisassemblers(); 330b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 331b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner cl::ParseCommandLineOptions(argc, argv, "llvm machine code playground\n"); 332b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 333b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner switch (Action) { 334b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner default: 33527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner case AC_AsLex: 33627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return AsLexInput(argv[0]); 337b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner case AC_Assemble: 338b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner return AssembleInput(argv[0]); 339ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan case AC_Disassemble: 340ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan return DisassembleInput(argv[0]); 341b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner } 342b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 343f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner return 0; 344f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner} 345f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 346