llvm-mc.cpp revision e895c6151589c1b7f6ac9ca992b76106fa197a37
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" 35f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerusing namespace llvm; 36f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 37f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerstatic cl::opt<std::string> 38f9f065e45500823cdeb25bde2154d871ab6e9125Chris LattnerInputFilename(cl::Positional, cl::desc("<input file>"), cl::init("-")); 39f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 40f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerstatic cl::opt<std::string> 41f9f065e45500823cdeb25bde2154d871ab6e9125Chris LattnerOutputFilename("o", cl::desc("Output filename"), 42f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner cl::value_desc("filename")); 43f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 44f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbarstatic cl::opt<bool> 45f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel DunbarShowEncoding("show-encoding", cl::desc("Show instruction encodings")); 46f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbar 47e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattnerstatic cl::opt<unsigned> 48e895c6151589c1b7f6ac9ca992b76106fa197a37Chris LattnerOutputAsmVariant("output-asm-variant", 49e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattner cl::desc("Syntax variant to use for output printing")); 50e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattner 51fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbarenum OutputFileType { 52fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OFT_AssemblyFile, 53fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OFT_ObjectFile 54fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar}; 55fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbarstatic cl::opt<OutputFileType> 56fb4a6b397665df011348ade24a8e38d2219f095aDaniel DunbarFileType("filetype", cl::init(OFT_AssemblyFile), 57fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar cl::desc("Choose an output file type:"), 58fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar cl::values( 59fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValN(OFT_AssemblyFile, "asm", 60fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar "Emit an assembly ('.s') file"), 61fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValN(OFT_ObjectFile, "obj", 62fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar "Emit a native object ('.o') file"), 63fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValEnd)); 64fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar 65bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbarstatic cl::opt<bool> 66baa26395ccf17fc988bb9cf62d6659ca8415ece9Dan GohmanForce("f", cl::desc("Enable binary output on terminals")); 67bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 68b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerstatic cl::list<std::string> 69b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris LattnerIncludeDirs("I", cl::desc("Directory of include files"), 70b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner cl::value_desc("directory"), cl::Prefix); 71f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 72b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbarstatic cl::opt<std::string> 738977d087c693fd581db82bcff134d12da0f48bd3Daniel DunbarTripleName("triple", cl::desc("Target triple to assemble for," 74b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar "see -version for available targets"), 75867aadfedfb6d9b64fd5da1ca67850a992605a3bDaniel Dunbar cl::init(LLVM_HOSTTRIPLE)); 76b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 77b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerenum ActionType { 7827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner AC_AsLex, 79b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner AC_Assemble 80b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner}; 81f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 82b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerstatic cl::opt<ActionType> 83b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris LattnerAction(cl::desc("Action to perform:"), 8427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner cl::init(AC_Assemble), 8527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner cl::values(clEnumValN(AC_AsLex, "as-lex", 8627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner "Lex tokens from a .s file"), 8727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner clEnumValN(AC_Assemble, "assemble", 88b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner "Assemble a .s file (default)"), 89b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner clEnumValEnd)); 90f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 919823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderbystatic const Target *GetTarget(const char *ProgName) { 929823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby // Get the target specific parser. 939823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby std::string Error; 949823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const Target *TheTarget = TargetRegistry::lookupTarget(TripleName, Error); 959823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (TheTarget) 969823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return TheTarget; 979823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 989823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby errs() << ProgName << ": error: unable to get target for '" << TripleName 999823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby << "', see --version and --triple.\n"; 1009823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return 0; 1019823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby} 1029823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 10327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattnerstatic int AsLexInput(const char *ProgName) { 104b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner std::string ErrorMessage; 105b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner MemoryBuffer *Buffer = MemoryBuffer::getFileOrSTDIN(InputFilename, 106b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner &ErrorMessage); 107f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner if (Buffer == 0) { 108b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner errs() << ProgName << ": "; 109f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner if (ErrorMessage.size()) 110f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner errs() << ErrorMessage << "\n"; 111f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner else 112f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner errs() << "input file didn't read correctly.\n"; 113f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner return 1; 114f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner } 115b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 116b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SourceMgr SrcMgr; 117b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 118b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Tell SrcMgr about this buffer, which is what TGParser will pick up. 119b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SrcMgr.AddNewSourceBuffer(Buffer, SMLoc()); 120b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 121b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Record the location of the include directories so that the lexer can find 122b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // it later. 123b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SrcMgr.setIncludeDirs(IncludeDirs); 124a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner 1259823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const Target *TheTarget = GetTarget(ProgName); 1269823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (!TheTarget) 1279823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return 1; 1289823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 1299823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const MCAsmInfo *MAI = TheTarget->createAsmInfo(TripleName); 1309823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby assert(MAI && "Unable to create target asm info!"); 1319823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 1329823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby AsmLexer Lexer(SrcMgr, *MAI); 133f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 13427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner bool Error = false; 13527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 136a3c924f83aefd4c883ed17a200876d258e0ad1e2Daniel Dunbar while (Lexer.Lex().isNot(AsmToken::Eof)) { 137a3c924f83aefd4c883ed17a200876d258e0ad1e2Daniel Dunbar switch (Lexer.getKind()) { 13827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner default: 1393fb7683bec8c8edb24e80c95f3b0668c6ecc0ae6Daniel Dunbar Lexer.PrintMessage(Lexer.getLoc(), "unknown token", "warning"); 14027aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner Error = true; 14127aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner break; 1423f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Error: 14327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner Error = true; // error already printed. 1444651bca31bdad27184fa0d36640bf5ef1d83cf5cChris Lattner break; 1453f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Identifier: 146419adedaa1638fbe4e078c997f81e94327ebff5aDaniel Dunbar outs() << "identifier: " << Lexer.getTok().getString() << '\n'; 147a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner break; 1483f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::String: 149419adedaa1638fbe4e078c997f81e94327ebff5aDaniel Dunbar outs() << "string: " << Lexer.getTok().getString() << '\n'; 15010a907d70fb54c40eecabb889e81c79b44092221Chris Lattner break; 1513f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Integer: 152419adedaa1638fbe4e078c997f81e94327ebff5aDaniel Dunbar outs() << "int: " << Lexer.getTok().getString() << '\n'; 153a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner break; 154165e8344d833147268810bbec18276dafe0122b9Daniel Dunbar 1553f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Amp: outs() << "Amp\n"; break; 1563f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::AmpAmp: outs() << "AmpAmp\n"; break; 1573f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Caret: outs() << "Caret\n"; break; 1583f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Colon: outs() << "Colon\n"; break; 1593f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Comma: outs() << "Comma\n"; break; 1603f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Dollar: outs() << "Dollar\n"; break; 1613f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::EndOfStatement: outs() << "EndOfStatement\n"; break; 1623f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Eof: outs() << "Eof\n"; break; 1633f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Equal: outs() << "Equal\n"; break; 1643f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::EqualEqual: outs() << "EqualEqual\n"; break; 1653f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Exclaim: outs() << "Exclaim\n"; break; 1663f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::ExclaimEqual: outs() << "ExclaimEqual\n"; break; 1673f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Greater: outs() << "Greater\n"; break; 1683f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::GreaterEqual: outs() << "GreaterEqual\n"; break; 1693f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::GreaterGreater: outs() << "GreaterGreater\n"; break; 1703f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::LParen: outs() << "LParen\n"; break; 1713f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Less: outs() << "Less\n"; break; 1723f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::LessEqual: outs() << "LessEqual\n"; break; 1733f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::LessGreater: outs() << "LessGreater\n"; break; 1743f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::LessLess: outs() << "LessLess\n"; break; 1753f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Minus: outs() << "Minus\n"; break; 1763f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Percent: outs() << "Percent\n"; break; 1773f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Pipe: outs() << "Pipe\n"; break; 1783f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::PipePipe: outs() << "PipePipe\n"; break; 1793f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Plus: outs() << "Plus\n"; break; 1803f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::RParen: outs() << "RParen\n"; break; 1813f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Slash: outs() << "Slash\n"; break; 1823f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Star: outs() << "Star\n"; break; 1833f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Tilde: outs() << "Tilde\n"; break; 184a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner } 185a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner } 186f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 18727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return Error; 188b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner} 189b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 190c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbarstatic formatted_raw_ostream *GetOutputStream() { 19117e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner if (OutputFilename == "") 19217e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner OutputFilename = "-"; 193bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 194bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar // Make sure that the Out file gets unlinked from the disk if we get a 19517e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner // SIGINT. 19617e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner if (OutputFilename != "-") 19717e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner sys::RemoveFileOnSignal(sys::Path(OutputFilename)); 198bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 199bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar std::string Err; 20017e9edc4a7bbeadf756494cf39fcacc9eff72202Chris Lattner raw_fd_ostream *Out = new raw_fd_ostream(OutputFilename.c_str(), Err, 201baa26395ccf17fc988bb9cf62d6659ca8415ece9Dan Gohman raw_fd_ostream::F_Binary); 202bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar if (!Err.empty()) { 203bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar errs() << Err << '\n'; 204bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar delete Out; 205bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar return 0; 206bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar } 207bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 208c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar return new formatted_raw_ostream(*Out, formatted_raw_ostream::DELETE_STREAM); 209bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar} 210bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 211a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbarstatic int AssembleInput(const char *ProgName) { 212c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar const Target *TheTarget = GetTarget(ProgName); 213c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (!TheTarget) 214c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar return 1; 215c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 216a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar std::string Error; 217a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar MemoryBuffer *Buffer = MemoryBuffer::getFileOrSTDIN(InputFilename, &Error); 21827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner if (Buffer == 0) { 21927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner errs() << ProgName << ": "; 220a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar if (Error.size()) 221a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar errs() << Error << "\n"; 22227aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner else 22327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner errs() << "input file didn't read correctly.\n"; 22427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return 1; 22527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner } 22627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 22727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SourceMgr SrcMgr; 22827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 22912a8a447a4ca5b939f255ae92e7fef490344593aDaniel Dunbar // Tell SrcMgr about this buffer, which is what the parser will pick up. 23027aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SrcMgr.AddNewSourceBuffer(Buffer, SMLoc()); 23127aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 23227aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner // Record the location of the include directories so that the lexer can find 23327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner // it later. 23427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SrcMgr.setIncludeDirs(IncludeDirs); 23527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 236cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner MCContext Ctx; 237c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar formatted_raw_ostream *Out = GetOutputStream(); 238bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar if (!Out) 239bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar return 1; 240c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 241c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 242c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar // FIXME: We shouldn't need to do this (and link in codegen). 243c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar OwningPtr<TargetMachine> TM(TheTarget->createTargetMachine(TripleName, "")); 244c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 245f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner if (!TM) { 246f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner errs() << ProgName << ": error: could not create target for triple '" 247f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner << TripleName << "'.\n"; 248f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner return 1; 249c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar } 250c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 25190edac0e8b35f766599362b6301863229f0ddcdbChris Lattner OwningPtr<MCInstPrinter> IP; 2524a0abd80f18f9c2a10bf5b14cd6731d51972a426Daniel Dunbar OwningPtr<MCCodeEmitter> CE; 253fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OwningPtr<MCStreamer> Str; 254f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner 2559823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const MCAsmInfo *MAI = TheTarget->createAsmInfo(TripleName); 2569823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby assert(MAI && "Unable to create target asm info!"); 257fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar 2589823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (FileType == OFT_AssemblyFile) { 259e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattner IP.reset(TheTarget->createMCInstPrinter(OutputAsmVariant, *MAI, *Out)); 260f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbar if (ShowEncoding) 261f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbar CE.reset(TheTarget->createCodeEmitter(*TM)); 26290edac0e8b35f766599362b6301863229f0ddcdbChris Lattner Str.reset(createAsmStreamer(Ctx, *Out, *MAI, IP.get(), CE.get())); 263fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar } else { 264fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar assert(FileType == OFT_ObjectFile && "Invalid file type!"); 2654fac74950a1ff08b995b366bfb84369c1507faefDaniel Dunbar CE.reset(TheTarget->createCodeEmitter(*TM)); 2664fac74950a1ff08b995b366bfb84369c1507faefDaniel Dunbar Str.reset(createMachOStreamer(Ctx, *Out, CE.get())); 267fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar } 268a0d1426af0bd05e1ae69481cdb75d2913e7e1ac1Daniel Dunbar 2699823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby AsmParser Parser(SrcMgr, Ctx, *Str.get(), *MAI); 270c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar OwningPtr<TargetAsmParser> TAP(TheTarget->createAsmParser(Parser)); 271c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (!TAP) { 272c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar errs() << ProgName 273c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar << ": error: this target does not support assembly parsing.\n"; 274a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar return 1; 275c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar } 276c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 27716cdcb38b2a5c5cdc216f9abafabd20e1ef7a254Daniel Dunbar Parser.setTargetParser(*TAP.get()); 278bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 279bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar int Res = Parser.Run(); 280c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (Out != &fouts()) 281bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar delete Out; 282bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 283bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar return Res; 28427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner} 28527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 286b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 287b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerint main(int argc, char **argv) { 288b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Print a stack trace if we signal out. 289b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner sys::PrintStackTraceOnErrorSignal(); 290b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner PrettyStackTraceProgram X(argc, argv); 291b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. 292b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 293c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar // Initialize targets and assembly printers/parsers. 294b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar llvm::InitializeAllTargetInfos(); 295c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar // FIXME: We shouldn't need to initialize the Target(Machine)s. 296c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar llvm::InitializeAllTargets(); 297c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar llvm::InitializeAllAsmPrinters(); 298b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar llvm::InitializeAllAsmParsers(); 299b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 300b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner cl::ParseCommandLineOptions(argc, argv, "llvm machine code playground\n"); 301b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 302b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner switch (Action) { 303b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner default: 30427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner case AC_AsLex: 30527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return AsLexInput(argv[0]); 306b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner case AC_Assemble: 307b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner return AssembleInput(argv[0]); 308b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner } 309b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 310f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner return 0; 311f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner} 312f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 313