llvm-mc.cpp revision 28c85a81a17dd719a254dc00cbeb484774893197
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 159fbb37e10d3b2814877bb8d2893d863fe1da660bDaniel Dunbar#include "llvm/MC/MCParser/AsmLexer.h" 16c6ef277a0b8f43af22d86aea9d5053749cacfbbbChris Lattner#include "llvm/MC/MCParser/MCAsmLexer.h" 1778c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng#include "llvm/MC/MCAsmBackend.h" 18cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner#include "llvm/MC/MCContext.h" 194a0abd80f18f9c2a10bf5b14cd6731d51972a426Daniel Dunbar#include "llvm/MC/MCCodeEmitter.h" 2090edac0e8b35f766599362b6301863229f0ddcdbChris Lattner#include "llvm/MC/MCInstPrinter.h" 2159ee62d2418df8db499eca1ae17f5900dc2dcbbaEvan Cheng#include "llvm/MC/MCInstrInfo.h" 22e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng#include "llvm/MC/MCObjectFileInfo.h" 23e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng#include "llvm/MC/MCRegisterInfo.h" 24f9bdeddb96043559c61f176f8077e3b91a0c544fChris Lattner#include "llvm/MC/MCSectionMachO.h" 25cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner#include "llvm/MC/MCStreamer.h" 26ffc0e73046f737d75e0a62b3a83ef19bcef111e3Evan Cheng#include "llvm/MC/MCSubtargetInfo.h" 2794b9550a32d189704a8eae55505edf62662c0534Evan Cheng#include "llvm/MC/MCTargetAsmParser.h" 28ab8be96fd30ca9396e6b84fdddf1ac6208984cadEvan Cheng#include "llvm/MC/SubtargetFeature.h" 29519466c32fe08bdf9afabfabd60a42f6c827ea6bChris Lattner#include "llvm/Target/TargetRegistry.h" 30519466c32fe08bdf9afabfabd60a42f6c827ea6bChris Lattner#include "llvm/Target/TargetSelect.h" 31cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner#include "llvm/ADT/OwningPtr.h" 32f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/CommandLine.h" 33d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman#include "llvm/Support/FileUtilities.h" 34c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar#include "llvm/Support/FormattedStream.h" 35f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/ManagedStatic.h" 36f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/MemoryBuffer.h" 37f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/PrettyStackTrace.h" 38b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner#include "llvm/Support/SourceMgr.h" 39e4f1a9b8a272ff7452759019ee7774e9dbdf1568Dan Gohman#include "llvm/Support/ToolOutputFile.h" 401f6efa3996dd1929fbc129203ce5009b620e6969Michael J. Spencer#include "llvm/Support/Host.h" 411f6efa3996dd1929fbc129203ce5009b620e6969Michael J. Spencer#include "llvm/Support/Signals.h" 42333fb04506233255f10d8095c9e2de5e5f0fdc6fMichael J. Spencer#include "llvm/Support/system_error.h" 43a3dcfb130044f306632a5fab43854eda4095a09cChris Lattner#include "Disassembler.h" 44f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerusing namespace llvm; 45f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 46f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerstatic cl::opt<std::string> 47f9f065e45500823cdeb25bde2154d871ab6e9125Chris LattnerInputFilename(cl::Positional, cl::desc("<input file>"), cl::init("-")); 48f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 49f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattnerstatic cl::opt<std::string> 50f9f065e45500823cdeb25bde2154d871ab6e9125Chris LattnerOutputFilename("o", cl::desc("Output filename"), 51f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner cl::value_desc("filename")); 52f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 53f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbarstatic cl::opt<bool> 54f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel DunbarShowEncoding("show-encoding", cl::desc("Show instruction encodings")); 55f2f6b0c0e9088da25a0367c88263ef8e7637a82cDaniel Dunbar 569dee8e3009408fd08c656558397a8ac8604139baDaniel Dunbarstatic cl::opt<bool> 579dee8e3009408fd08c656558397a8ac8604139baDaniel DunbarShowInst("show-inst", cl::desc("Show internal instruction representation")); 589dee8e3009408fd08c656558397a8ac8604139baDaniel Dunbar 593c14ca47fc057c5999cd41ce2782e0ebaa2bcf7dDaniel Dunbarstatic cl::opt<bool> 603c14ca47fc057c5999cd41ce2782e0ebaa2bcf7dDaniel DunbarShowInstOperands("show-inst-operands", 613c14ca47fc057c5999cd41ce2782e0ebaa2bcf7dDaniel Dunbar cl::desc("Show instructions operands as parsed")); 623c14ca47fc057c5999cd41ce2782e0ebaa2bcf7dDaniel Dunbar 63e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattnerstatic cl::opt<unsigned> 64e895c6151589c1b7f6ac9ca992b76106fa197a37Chris LattnerOutputAsmVariant("output-asm-variant", 65e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattner cl::desc("Syntax variant to use for output printing")); 66e895c6151589c1b7f6ac9ca992b76106fa197a37Chris Lattner 67ac2884a717daf3ad2aa8425320795d661e8a980bDaniel Dunbarstatic cl::opt<bool> 68ac2884a717daf3ad2aa8425320795d661e8a980bDaniel DunbarRelaxAll("mc-relax-all", cl::desc("Relax all fixups")); 69ac2884a717daf3ad2aa8425320795d661e8a980bDaniel Dunbar 70fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbarstatic cl::opt<bool> 7196aa78c8c5ef1a5f268539c9edc86569b436d573Rafael EspindolaNoExecStack("mc-no-exec-stack", cl::desc("File doesn't need an exec stack")); 7296aa78c8c5ef1a5f268539c9edc86569b436d573Rafael Espindola 7396aa78c8c5ef1a5f268539c9edc86569b436d573Rafael Espindolastatic cl::opt<bool> 74fdb5a8617996a20370756962e34c2e80176d6e87Daniel DunbarEnableLogging("enable-api-logging", cl::desc("Enable MC API logging")); 75fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar 76fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbarenum OutputFileType { 772d9f5d1d7f7aac13396778ba0d18cf1af8d0d02fDaniel Dunbar OFT_Null, 78fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OFT_AssemblyFile, 79fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OFT_ObjectFile 80fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar}; 81fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbarstatic cl::opt<OutputFileType> 82fb4a6b397665df011348ade24a8e38d2219f095aDaniel DunbarFileType("filetype", cl::init(OFT_AssemblyFile), 83fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar cl::desc("Choose an output file type:"), 84fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar cl::values( 85fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValN(OFT_AssemblyFile, "asm", 86fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar "Emit an assembly ('.s') file"), 872d9f5d1d7f7aac13396778ba0d18cf1af8d0d02fDaniel Dunbar clEnumValN(OFT_Null, "null", 882d9f5d1d7f7aac13396778ba0d18cf1af8d0d02fDaniel Dunbar "Don't emit anything (for timing purposes)"), 89fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValN(OFT_ObjectFile, "obj", 90fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar "Emit a native object ('.o') file"), 91fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar clEnumValEnd)); 92fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar 93b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerstatic cl::list<std::string> 94b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris LattnerIncludeDirs("I", cl::desc("Directory of include files"), 95b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner cl::value_desc("directory"), cl::Prefix); 96f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 97b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbarstatic cl::opt<std::string> 98181ab6a854162da908310e23714b43b0091c9094Daniel DunbarArchName("arch", cl::desc("Target arch to assemble for, " 99916a94b870042772568fca7995cf45aef7a6e333Bill Wendling "see -version for available targets")); 100181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar 101181ab6a854162da908310e23714b43b0091c9094Daniel Dunbarstatic cl::opt<std::string> 102ed1f1687b790b444c29c9cdc5ea646be0783c1cdNick LewyckyTripleName("triple", cl::desc("Target triple to assemble for, " 103181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar "see -version for available targets")); 104b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 105d2f14256f06725840f35adcc2557b48fc9ade694Jim Grosbachstatic cl::opt<std::string> 106d2f14256f06725840f35adcc2557b48fc9ade694Jim GrosbachMCPU("mcpu", 107d2f14256f06725840f35adcc2557b48fc9ade694Jim Grosbach cl::desc("Target a specific cpu type (-mcpu=help for details)"), 108d2f14256f06725840f35adcc2557b48fc9ade694Jim Grosbach cl::value_desc("cpu-name"), 109d2f14256f06725840f35adcc2557b48fc9ade694Jim Grosbach cl::init("")); 110d2f14256f06725840f35adcc2557b48fc9ade694Jim Grosbach 111439661395fd2a2a832dba01c65bc88718528313cEvan Chengstatic cl::opt<Reloc::Model> 112439661395fd2a2a832dba01c65bc88718528313cEvan ChengRelocModel("relocation-model", 113439661395fd2a2a832dba01c65bc88718528313cEvan Cheng cl::desc("Choose relocation model"), 114439661395fd2a2a832dba01c65bc88718528313cEvan Cheng cl::init(Reloc::Default), 115439661395fd2a2a832dba01c65bc88718528313cEvan Cheng cl::values( 116439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValN(Reloc::Default, "default", 117439661395fd2a2a832dba01c65bc88718528313cEvan Cheng "Target default relocation model"), 118439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValN(Reloc::Static, "static", 119439661395fd2a2a832dba01c65bc88718528313cEvan Cheng "Non-relocatable code"), 120439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValN(Reloc::PIC_, "pic", 121439661395fd2a2a832dba01c65bc88718528313cEvan Cheng "Fully relocatable, position independent code"), 122439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValN(Reloc::DynamicNoPIC, "dynamic-no-pic", 123439661395fd2a2a832dba01c65bc88718528313cEvan Cheng "Relocatable external references, non-relocatable code"), 124439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValEnd)); 125439661395fd2a2a832dba01c65bc88718528313cEvan Cheng 12634ad6db8b958fdc0d38e122edf753b5326e69b03Evan Chengstatic cl::opt<llvm::CodeModel::Model> 12734ad6db8b958fdc0d38e122edf753b5326e69b03Evan ChengCMModel("code-model", 12834ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng cl::desc("Choose code model"), 12934ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng cl::init(CodeModel::Default), 13034ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng cl::values(clEnumValN(CodeModel::Default, "default", 13134ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Target default code model"), 13234ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValN(CodeModel::Small, "small", 13334ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Small code model"), 13434ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValN(CodeModel::Kernel, "kernel", 13534ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Kernel code model"), 13634ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValN(CodeModel::Medium, "medium", 13734ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Medium code model"), 13834ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValN(CodeModel::Large, "large", 13934ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Large code model"), 14034ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValEnd)); 14134ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng 1425e6a7a248b980dd2a98a3613c72b75ba9d01d162Daniel Dunbarstatic cl::opt<bool> 143916a94b870042772568fca7995cf45aef7a6e333Bill WendlingNoInitialTextSection("n", cl::desc("Don't assume assembly file starts " 144916a94b870042772568fca7995cf45aef7a6e333Bill Wendling "in the text section")); 1455e6a7a248b980dd2a98a3613c72b75ba9d01d162Daniel Dunbar 146c6cf43d25853efb4a6765954eda52a45998a47f2Daniel Dunbarstatic cl::opt<bool> 147916a94b870042772568fca7995cf45aef7a6e333Bill WendlingSaveTempLabels("L", cl::desc("Don't discard temporary labels")); 148c6cf43d25853efb4a6765954eda52a45998a47f2Daniel Dunbar 149b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerenum ActionType { 15027aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner AC_AsLex, 151ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan AC_Assemble, 152668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan AC_Disassemble, 153668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan AC_EDisassemble 154b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner}; 155f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 156b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerstatic cl::opt<ActionType> 157b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris LattnerAction(cl::desc("Action to perform:"), 15827aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner cl::init(AC_Assemble), 15927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner cl::values(clEnumValN(AC_AsLex, "as-lex", 16027aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner "Lex tokens from a .s file"), 16127aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner clEnumValN(AC_Assemble, "assemble", 162b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner "Assemble a .s file (default)"), 163ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan clEnumValN(AC_Disassemble, "disassemble", 164ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan "Disassemble strings of hex bytes"), 165668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan clEnumValN(AC_EDisassemble, "edis", 166668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan "Enhanced disassembly of strings of hex bytes"), 167b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner clEnumValEnd)); 168f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 1699823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderbystatic const Target *GetTarget(const char *ProgName) { 170181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar // Figure out the target triple. 171181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar if (TripleName.empty()) 172181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar TripleName = sys::getHostTriple(); 173181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar if (!ArchName.empty()) { 174181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar llvm::Triple TT(TripleName); 175181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar TT.setArchName(ArchName); 176181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar TripleName = TT.str(); 177181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar } 178181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar 1799823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby // Get the target specific parser. 1809823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby std::string Error; 1819823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const Target *TheTarget = TargetRegistry::lookupTarget(TripleName, Error); 1829823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (TheTarget) 1839823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return TheTarget; 1849823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 1859823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby errs() << ProgName << ": error: unable to get target for '" << TripleName 1869823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby << "', see --version and --triple.\n"; 1879823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return 0; 1889823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby} 1899823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 190d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohmanstatic tool_output_file *GetOutputStream() { 191d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (OutputFilename == "") 192d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman OutputFilename = "-"; 193d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 194d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman std::string Err; 195d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman tool_output_file *Out = new tool_output_file(OutputFilename.c_str(), Err, 196d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman raw_fd_ostream::F_Binary); 197d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (!Err.empty()) { 198d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman errs() << Err << '\n'; 199d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman delete Out; 200d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return 0; 201d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman } 202d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman 203d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman return Out; 204d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman} 205d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 20627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattnerstatic int AsLexInput(const char *ProgName) { 2073ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer OwningPtr<MemoryBuffer> BufferPtr; 2083ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputFilename, BufferPtr)) { 209333fb04506233255f10d8095c9e2de5e5f0fdc6fMichael J. Spencer errs() << ProgName << ": " << ec.message() << '\n'; 210f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner return 1; 211f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner } 2123ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer MemoryBuffer *Buffer = BufferPtr.take(); 213b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 214b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SourceMgr SrcMgr; 215f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 216b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Tell SrcMgr about this buffer, which is what TGParser will pick up. 217b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SrcMgr.AddNewSourceBuffer(Buffer, SMLoc()); 218f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 219b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Record the location of the include directories so that the lexer can find 220b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // it later. 221b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SrcMgr.setIncludeDirs(IncludeDirs); 222a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner 2239823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const Target *TheTarget = GetTarget(ProgName); 2249823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (!TheTarget) 2259823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return 1; 2269823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 2271abf2cb59b8d63415780a03329307c0997b2670cEvan Cheng llvm::OwningPtr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(TripleName)); 2289823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby assert(MAI && "Unable to create target asm info!"); 2299823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 230fd0b0288e2ee5ccf3f1d47090542710c67a77cf7Sean Callanan AsmLexer Lexer(*MAI); 231346cc61cd20c41ef15ed824d28c81f40f1b28590Daniel Dunbar Lexer.setBuffer(SrcMgr.getMemoryBuffer(0)); 232346cc61cd20c41ef15ed824d28c81f40f1b28590Daniel Dunbar 233d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman OwningPtr<tool_output_file> Out(GetOutputStream()); 234d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (!Out) 235d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return 1; 236d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 23727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner bool Error = false; 238a3c924f83aefd4c883ed17a200876d258e0ad1e2Daniel Dunbar while (Lexer.Lex().isNot(AsmToken::Eof)) { 23954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar AsmToken Tok = Lexer.getTok(); 24054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar 24154b636930b9d782be934c3881b296f971203ced3Daniel Dunbar switch (Tok.getKind()) { 24227aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner default: 243bf2013ee22e7684ed37e4b78fca6937f38247ae7Sean Callanan SrcMgr.PrintMessage(Lexer.getLoc(), "unknown token", "warning"); 24427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner Error = true; 24527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner break; 2463f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Error: 24727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner Error = true; // error already printed. 2484651bca31bdad27184fa0d36640bf5ef1d83cf5cChris Lattner break; 2493f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Identifier: 25054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "identifier: " << Lexer.getTok().getString(); 251a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner break; 2523f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Integer: 25354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "int: " << Lexer.getTok().getString(); 254a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner break; 25554f0a625b0eb9afeece652a8462755010d237c78Daniel Dunbar case AsmToken::Real: 25654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "real: " << Lexer.getTok().getString(); 25754f0a625b0eb9afeece652a8462755010d237c78Daniel Dunbar break; 258f969868271f25df126047d0844f45b5e3b844b6dDaniel Dunbar case AsmToken::Register: 25954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "register: " << Lexer.getTok().getRegVal(); 260f969868271f25df126047d0844f45b5e3b844b6dDaniel Dunbar break; 261f969868271f25df126047d0844f45b5e3b844b6dDaniel Dunbar case AsmToken::String: 26254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "string: " << Lexer.getTok().getString(); 263f969868271f25df126047d0844f45b5e3b844b6dDaniel Dunbar break; 264165e8344d833147268810bbec18276dafe0122b9Daniel Dunbar 26554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Amp: Out->os() << "Amp"; break; 26654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::AmpAmp: Out->os() << "AmpAmp"; break; 26754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::At: Out->os() << "At"; break; 26854b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Caret: Out->os() << "Caret"; break; 26954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Colon: Out->os() << "Colon"; break; 27054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Comma: Out->os() << "Comma"; break; 27154b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Dollar: Out->os() << "Dollar"; break; 27254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Dot: Out->os() << "Dot"; break; 27354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::EndOfStatement: Out->os() << "EndOfStatement"; break; 27454b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Eof: Out->os() << "Eof"; break; 27554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Equal: Out->os() << "Equal"; break; 27654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::EqualEqual: Out->os() << "EqualEqual"; break; 27754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Exclaim: Out->os() << "Exclaim"; break; 27854b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::ExclaimEqual: Out->os() << "ExclaimEqual"; break; 27954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Greater: Out->os() << "Greater"; break; 28054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::GreaterEqual: Out->os() << "GreaterEqual"; break; 28154b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::GreaterGreater: Out->os() << "GreaterGreater"; break; 28254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Hash: Out->os() << "Hash"; break; 28354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LBrac: Out->os() << "LBrac"; break; 28454b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LCurly: Out->os() << "LCurly"; break; 28554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LParen: Out->os() << "LParen"; break; 28654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Less: Out->os() << "Less"; break; 28754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LessEqual: Out->os() << "LessEqual"; break; 28854b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LessGreater: Out->os() << "LessGreater"; break; 28954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LessLess: Out->os() << "LessLess"; break; 29054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Minus: Out->os() << "Minus"; break; 29154b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Percent: Out->os() << "Percent"; break; 29254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Pipe: Out->os() << "Pipe"; break; 29354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::PipePipe: Out->os() << "PipePipe"; break; 29454b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Plus: Out->os() << "Plus"; break; 29554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::RBrac: Out->os() << "RBrac"; break; 29654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::RCurly: Out->os() << "RCurly"; break; 29754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::RParen: Out->os() << "RParen"; break; 29854b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Slash: Out->os() << "Slash"; break; 29954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Star: Out->os() << "Star"; break; 30054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Tilde: Out->os() << "Tilde"; break; 301a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner } 30254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar 30354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar // Print the token string. 30454b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << " (\""; 30554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os().write_escaped(Tok.getString()); 30654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "\")\n"; 307a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner } 308bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 309d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman // Keep output if no errors. 310d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (Error == 0) Out->keep(); 311f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 312d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return Error; 313bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar} 314bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 315a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbarstatic int AssembleInput(const char *ProgName) { 316c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar const Target *TheTarget = GetTarget(ProgName); 317c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (!TheTarget) 318c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar return 1; 319c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 3203ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer OwningPtr<MemoryBuffer> BufferPtr; 3213ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputFilename, BufferPtr)) { 322333fb04506233255f10d8095c9e2de5e5f0fdc6fMichael J. Spencer errs() << ProgName << ": " << ec.message() << '\n'; 32327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return 1; 32427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner } 3253ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer MemoryBuffer *Buffer = BufferPtr.take(); 326f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 32727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SourceMgr SrcMgr; 328f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 32912a8a447a4ca5b939f255ae92e7fef490344593aDaniel Dunbar // Tell SrcMgr about this buffer, which is what the parser will pick up. 33027aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SrcMgr.AddNewSourceBuffer(Buffer, SMLoc()); 331f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 33227aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner // Record the location of the include directories so that the lexer can find 33327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner // it later. 33427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SrcMgr.setIncludeDirs(IncludeDirs); 335f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 336f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 3371abf2cb59b8d63415780a03329307c0997b2670cEvan Cheng llvm::OwningPtr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(TripleName)); 338c18409aed80ba1c6c5998befd3c3c8edc865c423Chris Lattner assert(MAI && "Unable to create target asm info!"); 339f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 3400e6a052331f674dd70e28af41f654a7874405eabEvan Cheng llvm::OwningPtr<MCRegisterInfo> MRI(TheTarget->createMCRegInfo(TripleName)); 3410e6a052331f674dd70e28af41f654a7874405eabEvan Cheng assert(MRI && "Unable to create target register info!"); 3420e6a052331f674dd70e28af41f654a7874405eabEvan Cheng 343d2f14256f06725840f35adcc2557b48fc9ade694Jim Grosbach // Package up features to be passed to target/subtarget 344d2f14256f06725840f35adcc2557b48fc9ade694Jim Grosbach std::string FeaturesStr; 345d2f14256f06725840f35adcc2557b48fc9ade694Jim Grosbach 346e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng // FIXME: This is not pretty. MCContext has a ptr to MCObjectFileInfo and 347e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng // MCObjectFileInfo needs a MCContext reference in order to initialize itself. 348e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng OwningPtr<MCObjectFileInfo> MOFI(new MCObjectFileInfo()); 349203576aa0cb9d8bf2d2e4d910ebab4b7a63262aeEvan Cheng MCContext Ctx(*MAI, *MRI, MOFI.get()); 350203576aa0cb9d8bf2d2e4d910ebab4b7a63262aeEvan Cheng MOFI->InitMCObjectFileInfo(TripleName, RelocModel, CMModel, Ctx); 351e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng 352c6cf43d25853efb4a6765954eda52a45998a47f2Daniel Dunbar if (SaveTempLabels) 353c6cf43d25853efb4a6765954eda52a45998a47f2Daniel Dunbar Ctx.setAllowTemporaryLabels(false); 35489b9372605db2ce3b0085c84089e389f7bc1fbddRafael Espindola 355d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman OwningPtr<tool_output_file> Out(GetOutputStream()); 356d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (!Out) 357d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return 1; 358d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 359d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman formatted_raw_ostream FOS(Out->os()); 360fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OwningPtr<MCStreamer> Str; 361f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner 36259ee62d2418df8db499eca1ae17f5900dc2dcbbaEvan Cheng OwningPtr<MCInstrInfo> MCII(TheTarget->createMCInstrInfo()); 363ffc0e73046f737d75e0a62b3a83ef19bcef111e3Evan Cheng OwningPtr<MCSubtargetInfo> 364ffc0e73046f737d75e0a62b3a83ef19bcef111e3Evan Cheng STI(TheTarget->createMCSubtargetInfo(TripleName, MCPU, FeaturesStr)); 365ffc0e73046f737d75e0a62b3a83ef19bcef111e3Evan Cheng 36696aa78c8c5ef1a5f268539c9edc86569b436d573Rafael Espindola // FIXME: There is a bit of code duplication with addPassesToEmitFile. 3679823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (FileType == OFT_AssemblyFile) { 3684c42a6de9f5456cc1b28f2d37db589f580f2adc7Chris Lattner MCInstPrinter *IP = 369b262799d49891b036daa00eddf51947487346c98Evan Cheng TheTarget->createMCInstPrinter(OutputAsmVariant, *MAI); 3701abcd06856df324eac98d4bf5ba673fb77ae6a11Benjamin Kramer MCCodeEmitter *CE = 0; 37178c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng MCAsmBackend *MAB = 0; 372745dacc91d7ee9531bfba76b21beb5d4eef93a7dDaniel Dunbar if (ShowEncoding) { 37328c85a81a17dd719a254dc00cbeb484774893197Evan Cheng CE = TheTarget->createMCCodeEmitter(*MCII, *STI, Ctx); 37478c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng MAB = TheTarget->createMCAsmBackend(TripleName); 375745dacc91d7ee9531bfba76b21beb5d4eef93a7dDaniel Dunbar } 37689b9372605db2ce3b0085c84089e389f7bc1fbddRafael Espindola Str.reset(TheTarget->createAsmStreamer(Ctx, FOS, /*asmverbose*/true, 377f1a5c7ec04002769f1638e64f7439589f0f926e6Rafael Espindola /*useLoc*/ true, 37878c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng /*useCFI*/ true, IP, CE, MAB, 379e266ce6c6eaf52ebe2b18d85b5e23788cf2f6ef4Bill Wendling ShowInst)); 3802d9f5d1d7f7aac13396778ba0d18cf1af8d0d02fDaniel Dunbar } else if (FileType == OFT_Null) { 3812d9f5d1d7f7aac13396778ba0d18cf1af8d0d02fDaniel Dunbar Str.reset(createNullStreamer(Ctx)); 382fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar } else { 383fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar assert(FileType == OFT_ObjectFile && "Invalid file type!"); 38428c85a81a17dd719a254dc00cbeb484774893197Evan Cheng MCCodeEmitter *CE = TheTarget->createMCCodeEmitter(*MCII, *STI, Ctx); 38578c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng MCAsmBackend *MAB = TheTarget->createMCAsmBackend(TripleName); 38628c85a81a17dd719a254dc00cbeb484774893197Evan Cheng Str.reset(TheTarget->createMCObjectStreamer(TripleName, Ctx, *MAB, 38728c85a81a17dd719a254dc00cbeb484774893197Evan Cheng FOS, CE, RelaxAll, 38828c85a81a17dd719a254dc00cbeb484774893197Evan Cheng NoExecStack)); 389fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar } 390a0d1426af0bd05e1ae69481cdb75d2913e7e1ac1Daniel Dunbar 391fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar if (EnableLogging) { 392fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar Str.reset(createLoggingStreamer(Str.take(), errs())); 393fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar } 394fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar 3959fbb37e10d3b2814877bb8d2893d863fe1da660bDaniel Dunbar OwningPtr<MCAsmParser> Parser(createMCAsmParser(*TheTarget, SrcMgr, Ctx, 3969fbb37e10d3b2814877bb8d2893d863fe1da660bDaniel Dunbar *Str.get(), *MAI)); 39794b9550a32d189704a8eae55505edf62662c0534Evan Cheng OwningPtr<MCTargetAsmParser> TAP(TheTarget->createMCAsmParser(*STI, *Parser)); 398c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (!TAP) { 399f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach errs() << ProgName 400c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar << ": error: this target does not support assembly parsing.\n"; 401a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar return 1; 402c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar } 403c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 4043c14ca47fc057c5999cd41ce2782e0ebaa2bcf7dDaniel Dunbar Parser->setShowParsedOperands(ShowInstOperands); 4059fbb37e10d3b2814877bb8d2893d863fe1da660bDaniel Dunbar Parser->setTargetParser(*TAP.get()); 406bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 4079fbb37e10d3b2814877bb8d2893d863fe1da660bDaniel Dunbar int Res = Parser->Run(NoInitialTextSection); 408bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 409d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman // Keep output if no errors. 410d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (Res == 0) Out->keep(); 4114ca906560a3d583b25f3bc366fbcac60f0a34434Daniel Dunbar 412bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar return Res; 413ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan} 414ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 415668b15467be158a5f0e0c4a5e1ec232da021892cSean Callananstatic int DisassembleInput(const char *ProgName, bool Enhanced) { 4164d4358e8b083b38e2d7e9de3ab102e5665ffb8e2Daniel Dunbar const Target *TheTarget = GetTarget(ProgName); 4174d4358e8b083b38e2d7e9de3ab102e5665ffb8e2Daniel Dunbar if (!TheTarget) 418ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan return 0; 419ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 4203ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer OwningPtr<MemoryBuffer> Buffer; 4213ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputFilename, Buffer)) { 422333fb04506233255f10d8095c9e2de5e5f0fdc6fMichael J. Spencer errs() << ProgName << ": " << ec.message() << '\n'; 423ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan return 1; 424ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan } 425f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 426d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman OwningPtr<tool_output_file> Out(GetOutputStream()); 427d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (!Out) 428d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return 1; 429d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 430d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman int Res; 431a5c177e70a42f48e4885075c4c48aad0816a2817Bill Wendling if (Enhanced) { 4323ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer Res = 4333ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer Disassembler::disassembleEnhanced(TripleName, *Buffer.take(), Out->os()); 434a5c177e70a42f48e4885075c4c48aad0816a2817Bill Wendling } else { 435b262799d49891b036daa00eddf51947487346c98Evan Cheng Res = Disassembler::disassemble(*TheTarget, TripleName, 4363ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer *Buffer.take(), Out->os()); 437a5c177e70a42f48e4885075c4c48aad0816a2817Bill Wendling } 438d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 439d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman // Keep output if no errors. 440d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (Res == 0) Out->keep(); 441d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 442d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return Res; 443ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan} 44427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 445b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 446b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerint main(int argc, char **argv) { 447b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Print a stack trace if we signal out. 448b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner sys::PrintStackTraceOnErrorSignal(); 449b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner PrettyStackTraceProgram X(argc, argv); 450b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. 451b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 452c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar // Initialize targets and assembly printers/parsers. 453b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar llvm::InitializeAllTargetInfos(); 454e78085a3c03de648a481e9751c3094c517bd7123Evan Cheng llvm::InitializeAllTargetMCs(); 455b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar llvm::InitializeAllAsmParsers(); 456ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan llvm::InitializeAllDisassemblers(); 457f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 458b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner cl::ParseCommandLineOptions(argc, argv, "llvm machine code playground\n"); 45975ebbceeeda1e7a78efe1848bb90e034f0c6ba61Duncan Sands TripleName = Triple::normalize(TripleName); 460b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 461b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner switch (Action) { 462b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner default: 46327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner case AC_AsLex: 46427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return AsLexInput(argv[0]); 465b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner case AC_Assemble: 466b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner return AssembleInput(argv[0]); 467ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan case AC_Disassemble: 468668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan return DisassembleInput(argv[0], false); 469668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan case AC_EDisassemble: 470668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan return DisassembleInput(argv[0], true); 471b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner } 472f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 473f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner return 0; 474f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner} 475f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 476