llvm-mc.cpp revision b950585cc5a0d665e9accfe5ce490cd269756f2e
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" 29cbc23f75cd8cd6889fd02f65b63f6c02512460bdChris Lattner#include "llvm/ADT/OwningPtr.h" 30f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/CommandLine.h" 31d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman#include "llvm/Support/FileUtilities.h" 32c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar#include "llvm/Support/FormattedStream.h" 33f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/ManagedStatic.h" 34f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/MemoryBuffer.h" 35f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner#include "llvm/Support/PrettyStackTrace.h" 36b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner#include "llvm/Support/SourceMgr.h" 37e4f1a9b8a272ff7452759019ee7774e9dbdf1568Dan Gohman#include "llvm/Support/ToolOutputFile.h" 381f6efa3996dd1929fbc129203ce5009b620e6969Michael J. Spencer#include "llvm/Support/Host.h" 391f6efa3996dd1929fbc129203ce5009b620e6969Michael J. Spencer#include "llvm/Support/Signals.h" 403e74d6fdd248e20a280f1dff3da9a6c689c2c4c3Evan Cheng#include "llvm/Support/TargetRegistry.h" 413e74d6fdd248e20a280f1dff3da9a6c689c2c4c3Evan Cheng#include "llvm/Support/TargetSelect.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 111b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloystatic cl::list<std::string> 112b950585cc5a0d665e9accfe5ce490cd269756f2eJames MolloyMAttrs("mattr", 113b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy cl::CommaSeparated, 114b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy cl::desc("Target specific attributes (-mattr=help for details)"), 115b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy cl::value_desc("a1,+a2,-a3,...")); 116b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy 117439661395fd2a2a832dba01c65bc88718528313cEvan Chengstatic cl::opt<Reloc::Model> 118439661395fd2a2a832dba01c65bc88718528313cEvan ChengRelocModel("relocation-model", 119439661395fd2a2a832dba01c65bc88718528313cEvan Cheng cl::desc("Choose relocation model"), 120439661395fd2a2a832dba01c65bc88718528313cEvan Cheng cl::init(Reloc::Default), 121439661395fd2a2a832dba01c65bc88718528313cEvan Cheng cl::values( 122439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValN(Reloc::Default, "default", 123439661395fd2a2a832dba01c65bc88718528313cEvan Cheng "Target default relocation model"), 124439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValN(Reloc::Static, "static", 125439661395fd2a2a832dba01c65bc88718528313cEvan Cheng "Non-relocatable code"), 126439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValN(Reloc::PIC_, "pic", 127439661395fd2a2a832dba01c65bc88718528313cEvan Cheng "Fully relocatable, position independent code"), 128439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValN(Reloc::DynamicNoPIC, "dynamic-no-pic", 129439661395fd2a2a832dba01c65bc88718528313cEvan Cheng "Relocatable external references, non-relocatable code"), 130439661395fd2a2a832dba01c65bc88718528313cEvan Cheng clEnumValEnd)); 131439661395fd2a2a832dba01c65bc88718528313cEvan Cheng 13234ad6db8b958fdc0d38e122edf753b5326e69b03Evan Chengstatic cl::opt<llvm::CodeModel::Model> 13334ad6db8b958fdc0d38e122edf753b5326e69b03Evan ChengCMModel("code-model", 13434ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng cl::desc("Choose code model"), 13534ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng cl::init(CodeModel::Default), 13634ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng cl::values(clEnumValN(CodeModel::Default, "default", 13734ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Target default code model"), 13834ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValN(CodeModel::Small, "small", 13934ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Small code model"), 14034ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValN(CodeModel::Kernel, "kernel", 14134ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Kernel code model"), 14234ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValN(CodeModel::Medium, "medium", 14334ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Medium code model"), 14434ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValN(CodeModel::Large, "large", 14534ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng "Large code model"), 14634ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng clEnumValEnd)); 14734ad6db8b958fdc0d38e122edf753b5326e69b03Evan Cheng 1485e6a7a248b980dd2a98a3613c72b75ba9d01d162Daniel Dunbarstatic cl::opt<bool> 149916a94b870042772568fca7995cf45aef7a6e333Bill WendlingNoInitialTextSection("n", cl::desc("Don't assume assembly file starts " 150916a94b870042772568fca7995cf45aef7a6e333Bill Wendling "in the text section")); 1515e6a7a248b980dd2a98a3613c72b75ba9d01d162Daniel Dunbar 152c6cf43d25853efb4a6765954eda52a45998a47f2Daniel Dunbarstatic cl::opt<bool> 153916a94b870042772568fca7995cf45aef7a6e333Bill WendlingSaveTempLabels("L", cl::desc("Don't discard temporary labels")); 154c6cf43d25853efb4a6765954eda52a45998a47f2Daniel Dunbar 155b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerenum ActionType { 15627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner AC_AsLex, 157ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan AC_Assemble, 158668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan AC_Disassemble, 159668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan AC_EDisassemble 160b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner}; 161f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 162b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerstatic cl::opt<ActionType> 163b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris LattnerAction(cl::desc("Action to perform:"), 16427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner cl::init(AC_Assemble), 16527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner cl::values(clEnumValN(AC_AsLex, "as-lex", 16627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner "Lex tokens from a .s file"), 16727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner clEnumValN(AC_Assemble, "assemble", 168b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner "Assemble a .s file (default)"), 169ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan clEnumValN(AC_Disassemble, "disassemble", 170ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan "Disassemble strings of hex bytes"), 171668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan clEnumValN(AC_EDisassemble, "edis", 172668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan "Enhanced disassembly of strings of hex bytes"), 173b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner clEnumValEnd)); 174f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 1759823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderbystatic const Target *GetTarget(const char *ProgName) { 176181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar // Figure out the target triple. 177181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar if (TripleName.empty()) 178181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar TripleName = sys::getHostTriple(); 179d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng Triple TheTriple(Triple::normalize(TripleName)); 180d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng 181d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng const Target *TheTarget = 0; 182181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar if (!ArchName.empty()) { 183d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng for (TargetRegistry::iterator it = TargetRegistry::begin(), 184d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng ie = TargetRegistry::end(); it != ie; ++it) { 185d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng if (ArchName == it->getName()) { 186d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng TheTarget = &*it; 187d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng break; 188d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng } 189d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng } 190d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng 191d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng if (!TheTarget) { 192d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng errs() << ProgName << ": error: invalid target '" << ArchName << "'.\n"; 193d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng return 0; 194d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng } 195d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng 196d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng // Adjust the triple to match (if known), otherwise stick with the 197d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng // module/host triple. 198d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng Triple::ArchType Type = Triple::getArchTypeForLLVMName(ArchName); 199d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng if (Type != Triple::UnknownArch) 200d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng TheTriple.setArch(Type); 201d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng } else { 202d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng // Get the target specific parser. 203d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng std::string Error; 204d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng TheTarget = TargetRegistry::lookupTarget(TheTriple.getTriple(), Error); 205d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng if (TheTarget == 0) { 206d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng errs() << ProgName << ": error: unable to get target for '" 207d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng << TheTriple.getTriple() 208d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng << "', see --version and --triple.\n"; 209d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng return 0; 210d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng } 211181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar } 212181ab6a854162da908310e23714b43b0091c9094Daniel Dunbar 213d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng TripleName = TheTriple.getTriple(); 214d6dcf39ca824f6df42de92328d08ad5d4b3d6bd2Evan Cheng return TheTarget; 2159823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby} 2169823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 217d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohmanstatic tool_output_file *GetOutputStream() { 218d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (OutputFilename == "") 219d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman OutputFilename = "-"; 220d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 221d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman std::string Err; 222d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman tool_output_file *Out = new tool_output_file(OutputFilename.c_str(), Err, 223d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman raw_fd_ostream::F_Binary); 224d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (!Err.empty()) { 225d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman errs() << Err << '\n'; 226d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman delete Out; 227d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return 0; 228d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman } 229d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman 230d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman return Out; 231d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman} 232d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 23327aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattnerstatic int AsLexInput(const char *ProgName) { 2343ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer OwningPtr<MemoryBuffer> BufferPtr; 2353ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputFilename, BufferPtr)) { 236333fb04506233255f10d8095c9e2de5e5f0fdc6fMichael J. Spencer errs() << ProgName << ": " << ec.message() << '\n'; 237f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner return 1; 238f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner } 2393ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer MemoryBuffer *Buffer = BufferPtr.take(); 240b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 241b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SourceMgr SrcMgr; 242f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 243b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Tell SrcMgr about this buffer, which is what TGParser will pick up. 244b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SrcMgr.AddNewSourceBuffer(Buffer, SMLoc()); 245f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 246b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Record the location of the include directories so that the lexer can find 247b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // it later. 248b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner SrcMgr.setIncludeDirs(IncludeDirs); 249a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner 2509823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby const Target *TheTarget = GetTarget(ProgName); 2519823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (!TheTarget) 2529823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby return 1; 2539823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 2541abf2cb59b8d63415780a03329307c0997b2670cEvan Cheng llvm::OwningPtr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(TripleName)); 2559823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby assert(MAI && "Unable to create target asm info!"); 2569823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby 257fd0b0288e2ee5ccf3f1d47090542710c67a77cf7Sean Callanan AsmLexer Lexer(*MAI); 258346cc61cd20c41ef15ed824d28c81f40f1b28590Daniel Dunbar Lexer.setBuffer(SrcMgr.getMemoryBuffer(0)); 259346cc61cd20c41ef15ed824d28c81f40f1b28590Daniel Dunbar 260d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman OwningPtr<tool_output_file> Out(GetOutputStream()); 261d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (!Out) 262d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return 1; 263d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 26427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner bool Error = false; 265a3c924f83aefd4c883ed17a200876d258e0ad1e2Daniel Dunbar while (Lexer.Lex().isNot(AsmToken::Eof)) { 26654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar AsmToken Tok = Lexer.getTok(); 26754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar 26854b636930b9d782be934c3881b296f971203ced3Daniel Dunbar switch (Tok.getKind()) { 26927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner default: 270bf2013ee22e7684ed37e4b78fca6937f38247ae7Sean Callanan SrcMgr.PrintMessage(Lexer.getLoc(), "unknown token", "warning"); 27127aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner Error = true; 27227aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner break; 2733f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Error: 27427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner Error = true; // error already printed. 2754651bca31bdad27184fa0d36640bf5ef1d83cf5cChris Lattner break; 2763f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Identifier: 27754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "identifier: " << Lexer.getTok().getString(); 278a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner break; 2793f87233d700eb4316cfaad59477834d2f5a2503bDaniel Dunbar case AsmToken::Integer: 28054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "int: " << Lexer.getTok().getString(); 281a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner break; 28254f0a625b0eb9afeece652a8462755010d237c78Daniel Dunbar case AsmToken::Real: 28354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "real: " << Lexer.getTok().getString(); 28454f0a625b0eb9afeece652a8462755010d237c78Daniel Dunbar break; 285f969868271f25df126047d0844f45b5e3b844b6dDaniel Dunbar case AsmToken::Register: 28654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "register: " << Lexer.getTok().getRegVal(); 287f969868271f25df126047d0844f45b5e3b844b6dDaniel Dunbar break; 288f969868271f25df126047d0844f45b5e3b844b6dDaniel Dunbar case AsmToken::String: 28954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "string: " << Lexer.getTok().getString(); 290f969868271f25df126047d0844f45b5e3b844b6dDaniel Dunbar break; 291165e8344d833147268810bbec18276dafe0122b9Daniel Dunbar 29254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Amp: Out->os() << "Amp"; break; 29354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::AmpAmp: Out->os() << "AmpAmp"; break; 29454b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::At: Out->os() << "At"; break; 29554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Caret: Out->os() << "Caret"; break; 29654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Colon: Out->os() << "Colon"; break; 29754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Comma: Out->os() << "Comma"; break; 29854b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Dollar: Out->os() << "Dollar"; break; 29954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Dot: Out->os() << "Dot"; break; 30054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::EndOfStatement: Out->os() << "EndOfStatement"; break; 30154b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Eof: Out->os() << "Eof"; break; 30254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Equal: Out->os() << "Equal"; break; 30354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::EqualEqual: Out->os() << "EqualEqual"; break; 30454b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Exclaim: Out->os() << "Exclaim"; break; 30554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::ExclaimEqual: Out->os() << "ExclaimEqual"; break; 30654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Greater: Out->os() << "Greater"; break; 30754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::GreaterEqual: Out->os() << "GreaterEqual"; break; 30854b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::GreaterGreater: Out->os() << "GreaterGreater"; break; 30954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Hash: Out->os() << "Hash"; break; 31054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LBrac: Out->os() << "LBrac"; break; 31154b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LCurly: Out->os() << "LCurly"; break; 31254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LParen: Out->os() << "LParen"; break; 31354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Less: Out->os() << "Less"; break; 31454b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LessEqual: Out->os() << "LessEqual"; break; 31554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LessGreater: Out->os() << "LessGreater"; break; 31654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::LessLess: Out->os() << "LessLess"; break; 31754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Minus: Out->os() << "Minus"; break; 31854b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Percent: Out->os() << "Percent"; break; 31954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Pipe: Out->os() << "Pipe"; break; 32054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::PipePipe: Out->os() << "PipePipe"; break; 32154b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Plus: Out->os() << "Plus"; break; 32254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::RBrac: Out->os() << "RBrac"; break; 32354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::RCurly: Out->os() << "RCurly"; break; 32454b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::RParen: Out->os() << "RParen"; break; 32554b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Slash: Out->os() << "Slash"; break; 32654b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Star: Out->os() << "Star"; break; 32754b636930b9d782be934c3881b296f971203ced3Daniel Dunbar case AsmToken::Tilde: Out->os() << "Tilde"; break; 328a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner } 32954b636930b9d782be934c3881b296f971203ced3Daniel Dunbar 33054b636930b9d782be934c3881b296f971203ced3Daniel Dunbar // Print the token string. 33154b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << " (\""; 33254b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os().write_escaped(Tok.getString()); 33354b636930b9d782be934c3881b296f971203ced3Daniel Dunbar Out->os() << "\")\n"; 334a59e8779964992457ada1af6a5f48068523cfd42Chris Lattner } 335bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 336d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman // Keep output if no errors. 337d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (Error == 0) Out->keep(); 338f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 339d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return Error; 340bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar} 341bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 342a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbarstatic int AssembleInput(const char *ProgName) { 343c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar const Target *TheTarget = GetTarget(ProgName); 344c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (!TheTarget) 345c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar return 1; 346c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 3473ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer OwningPtr<MemoryBuffer> BufferPtr; 3483ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputFilename, BufferPtr)) { 349333fb04506233255f10d8095c9e2de5e5f0fdc6fMichael J. Spencer errs() << ProgName << ": " << ec.message() << '\n'; 35027aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return 1; 35127aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner } 3523ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer MemoryBuffer *Buffer = BufferPtr.take(); 353f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 35427aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SourceMgr SrcMgr; 355f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 35612a8a447a4ca5b939f255ae92e7fef490344593aDaniel Dunbar // Tell SrcMgr about this buffer, which is what the parser will pick up. 35727aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SrcMgr.AddNewSourceBuffer(Buffer, SMLoc()); 358f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 35927aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner // Record the location of the include directories so that the lexer can find 36027aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner // it later. 36127aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner SrcMgr.setIncludeDirs(IncludeDirs); 362f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 363f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 3641abf2cb59b8d63415780a03329307c0997b2670cEvan Cheng llvm::OwningPtr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(TripleName)); 365c18409aed80ba1c6c5998befd3c3c8edc865c423Chris Lattner assert(MAI && "Unable to create target asm info!"); 366f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 3670e6a052331f674dd70e28af41f654a7874405eabEvan Cheng llvm::OwningPtr<MCRegisterInfo> MRI(TheTarget->createMCRegInfo(TripleName)); 3680e6a052331f674dd70e28af41f654a7874405eabEvan Cheng assert(MRI && "Unable to create target register info!"); 3690e6a052331f674dd70e28af41f654a7874405eabEvan Cheng 370e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng // FIXME: This is not pretty. MCContext has a ptr to MCObjectFileInfo and 371e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng // MCObjectFileInfo needs a MCContext reference in order to initialize itself. 372e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng OwningPtr<MCObjectFileInfo> MOFI(new MCObjectFileInfo()); 373203576aa0cb9d8bf2d2e4d910ebab4b7a63262aeEvan Cheng MCContext Ctx(*MAI, *MRI, MOFI.get()); 374203576aa0cb9d8bf2d2e4d910ebab4b7a63262aeEvan Cheng MOFI->InitMCObjectFileInfo(TripleName, RelocModel, CMModel, Ctx); 375e76a33b9567d78a5744dc52fcec3a6056d6fb576Evan Cheng 376c6cf43d25853efb4a6765954eda52a45998a47f2Daniel Dunbar if (SaveTempLabels) 377c6cf43d25853efb4a6765954eda52a45998a47f2Daniel Dunbar Ctx.setAllowTemporaryLabels(false); 37889b9372605db2ce3b0085c84089e389f7bc1fbddRafael Espindola 379b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy // Package up features to be passed to target/subtarget 380b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy std::string FeaturesStr; 381b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy if (MAttrs.size()) { 382b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy SubtargetFeatures Features; 383b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy for (unsigned i = 0; i != MAttrs.size(); ++i) 384b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy Features.AddFeature(MAttrs[i]); 385b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy FeaturesStr = Features.getString(); 386b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy } 387b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy 388d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman OwningPtr<tool_output_file> Out(GetOutputStream()); 389d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (!Out) 390d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return 1; 391d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 392d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman formatted_raw_ostream FOS(Out->os()); 393fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar OwningPtr<MCStreamer> Str; 394f3ce009fcb0a04a245c437d86657e9bd61ecc18fChris Lattner 39559ee62d2418df8db499eca1ae17f5900dc2dcbbaEvan Cheng OwningPtr<MCInstrInfo> MCII(TheTarget->createMCInstrInfo()); 396ffc0e73046f737d75e0a62b3a83ef19bcef111e3Evan Cheng OwningPtr<MCSubtargetInfo> 397ffc0e73046f737d75e0a62b3a83ef19bcef111e3Evan Cheng STI(TheTarget->createMCSubtargetInfo(TripleName, MCPU, FeaturesStr)); 398ffc0e73046f737d75e0a62b3a83ef19bcef111e3Evan Cheng 39996aa78c8c5ef1a5f268539c9edc86569b436d573Rafael Espindola // FIXME: There is a bit of code duplication with addPassesToEmitFile. 4009823ca971d5cb475401e59fde244caf5087c74a1Kevin Enderby if (FileType == OFT_AssemblyFile) { 4014c42a6de9f5456cc1b28f2d37db589f580f2adc7Chris Lattner MCInstPrinter *IP = 402b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy TheTarget->createMCInstPrinter(OutputAsmVariant, *MAI, *STI); 4031abcd06856df324eac98d4bf5ba673fb77ae6a11Benjamin Kramer MCCodeEmitter *CE = 0; 40478c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng MCAsmBackend *MAB = 0; 405745dacc91d7ee9531bfba76b21beb5d4eef93a7dDaniel Dunbar if (ShowEncoding) { 40628c85a81a17dd719a254dc00cbeb484774893197Evan Cheng CE = TheTarget->createMCCodeEmitter(*MCII, *STI, Ctx); 40778c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng MAB = TheTarget->createMCAsmBackend(TripleName); 408745dacc91d7ee9531bfba76b21beb5d4eef93a7dDaniel Dunbar } 40989b9372605db2ce3b0085c84089e389f7bc1fbddRafael Espindola Str.reset(TheTarget->createAsmStreamer(Ctx, FOS, /*asmverbose*/true, 410f1a5c7ec04002769f1638e64f7439589f0f926e6Rafael Espindola /*useLoc*/ true, 41178c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng /*useCFI*/ true, IP, CE, MAB, 412e266ce6c6eaf52ebe2b18d85b5e23788cf2f6ef4Bill Wendling ShowInst)); 4132d9f5d1d7f7aac13396778ba0d18cf1af8d0d02fDaniel Dunbar } else if (FileType == OFT_Null) { 4142d9f5d1d7f7aac13396778ba0d18cf1af8d0d02fDaniel Dunbar Str.reset(createNullStreamer(Ctx)); 415fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar } else { 416fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar assert(FileType == OFT_ObjectFile && "Invalid file type!"); 41728c85a81a17dd719a254dc00cbeb484774893197Evan Cheng MCCodeEmitter *CE = TheTarget->createMCCodeEmitter(*MCII, *STI, Ctx); 41878c10eeaa57d1c6c4b7781d3c0bcb0cfbbc43b5cEvan Cheng MCAsmBackend *MAB = TheTarget->createMCAsmBackend(TripleName); 41928c85a81a17dd719a254dc00cbeb484774893197Evan Cheng Str.reset(TheTarget->createMCObjectStreamer(TripleName, Ctx, *MAB, 42028c85a81a17dd719a254dc00cbeb484774893197Evan Cheng FOS, CE, RelaxAll, 42128c85a81a17dd719a254dc00cbeb484774893197Evan Cheng NoExecStack)); 422fb4a6b397665df011348ade24a8e38d2219f095aDaniel Dunbar } 423a0d1426af0bd05e1ae69481cdb75d2913e7e1ac1Daniel Dunbar 424fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar if (EnableLogging) { 425fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar Str.reset(createLoggingStreamer(Str.take(), errs())); 426fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar } 427fdb5a8617996a20370756962e34c2e80176d6e87Daniel Dunbar 4281b84cce77f8bccc905b4800927ce9016f76c1c40Jim Grosbach OwningPtr<MCAsmParser> Parser(createMCAsmParser(SrcMgr, Ctx, 4291b84cce77f8bccc905b4800927ce9016f76c1c40Jim Grosbach *Str.get(), *MAI)); 43094b9550a32d189704a8eae55505edf62662c0534Evan Cheng OwningPtr<MCTargetAsmParser> TAP(TheTarget->createMCAsmParser(*STI, *Parser)); 431c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar if (!TAP) { 432f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach errs() << ProgName 433c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar << ": error: this target does not support assembly parsing.\n"; 434a2edbabcb86f213eca6daeda5d801f8c7b1e44b2Daniel Dunbar return 1; 435c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar } 436c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar 4373c14ca47fc057c5999cd41ce2782e0ebaa2bcf7dDaniel Dunbar Parser->setShowParsedOperands(ShowInstOperands); 4389fbb37e10d3b2814877bb8d2893d863fe1da660bDaniel Dunbar Parser->setTargetParser(*TAP.get()); 439bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 4409fbb37e10d3b2814877bb8d2893d863fe1da660bDaniel Dunbar int Res = Parser->Run(NoInitialTextSection); 441bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar 442d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman // Keep output if no errors. 443d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (Res == 0) Out->keep(); 4444ca906560a3d583b25f3bc366fbcac60f0a34434Daniel Dunbar 445bfc0f34e34c4278b49d70b7a20e1b87cbce0a710Daniel Dunbar return Res; 446ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan} 447ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 448668b15467be158a5f0e0c4a5e1ec232da021892cSean Callananstatic int DisassembleInput(const char *ProgName, bool Enhanced) { 4494d4358e8b083b38e2d7e9de3ab102e5665ffb8e2Daniel Dunbar const Target *TheTarget = GetTarget(ProgName); 4504d4358e8b083b38e2d7e9de3ab102e5665ffb8e2Daniel Dunbar if (!TheTarget) 451ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan return 0; 452ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan 4533ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer OwningPtr<MemoryBuffer> Buffer; 4543ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer if (error_code ec = MemoryBuffer::getFileOrSTDIN(InputFilename, Buffer)) { 455333fb04506233255f10d8095c9e2de5e5f0fdc6fMichael J. Spencer errs() << ProgName << ": " << ec.message() << '\n'; 456ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan return 1; 457ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan } 458f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 459d4c454317a38d65957edebe62bfc69fc8d9885e8Dan Gohman OwningPtr<tool_output_file> Out(GetOutputStream()); 460d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (!Out) 461d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return 1; 462d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 463d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman int Res; 464a5c177e70a42f48e4885075c4c48aad0816a2817Bill Wendling if (Enhanced) { 4653ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer Res = 4663ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer Disassembler::disassembleEnhanced(TripleName, *Buffer.take(), Out->os()); 467a5c177e70a42f48e4885075c4c48aad0816a2817Bill Wendling } else { 468b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy // Package up features to be passed to target/subtarget 469b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy std::string FeaturesStr; 470b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy if (MAttrs.size()) { 471b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy SubtargetFeatures Features; 472b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy for (unsigned i = 0; i != MAttrs.size(); ++i) 473b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy Features.AddFeature(MAttrs[i]); 474b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy FeaturesStr = Features.getString(); 475b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy } 476b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy 477b950585cc5a0d665e9accfe5ce490cd269756f2eJames Molloy Res = Disassembler::disassemble(*TheTarget, TripleName, MCPU, FeaturesStr, 4783ff9563c3e391954b2e224afcf8b2b0fcc3888aaMichael J. Spencer *Buffer.take(), Out->os()); 479a5c177e70a42f48e4885075c4c48aad0816a2817Bill Wendling } 480d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 481d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman // Keep output if no errors. 482d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman if (Res == 0) Out->keep(); 483d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman 484d5826a33a5a7c298a8934541d11cda042028be3bDan Gohman return Res; 485ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan} 48627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner 487b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 488b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattnerint main(int argc, char **argv) { 489b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner // Print a stack trace if we signal out. 490b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner sys::PrintStackTraceOnErrorSignal(); 491b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner PrettyStackTraceProgram X(argc, argv); 492b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. 493b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar 494c22e0b2443afdedb6d9b225b938ad404d63cdbe6Daniel Dunbar // Initialize targets and assembly printers/parsers. 495b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar llvm::InitializeAllTargetInfos(); 496e78085a3c03de648a481e9751c3094c517bd7123Evan Cheng llvm::InitializeAllTargetMCs(); 497b4b53e5c13167925d6315a6f57c7b863e4e2b704Daniel Dunbar llvm::InitializeAllAsmParsers(); 498ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan llvm::InitializeAllDisassemblers(); 499f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 500b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner cl::ParseCommandLineOptions(argc, argv, "llvm machine code playground\n"); 50175ebbceeeda1e7a78efe1848bb90e034f0c6ba61Duncan Sands TripleName = Triple::normalize(TripleName); 502b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner 503b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner switch (Action) { 504b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner default: 50527aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner case AC_AsLex: 50627aa7d259b416a9d1bf837ed2c3c11463367b11cChris Lattner return AsLexInput(argv[0]); 507b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner case AC_Assemble: 508b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner return AssembleInput(argv[0]); 509ba847da571354e13f1caa3699ee06b2d57df9fe9Sean Callanan case AC_Disassemble: 510668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan return DisassembleInput(argv[0], false); 511668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan case AC_EDisassemble: 512668b15467be158a5f0e0c4a5e1ec232da021892cSean Callanan return DisassembleInput(argv[0], true); 513b23677e076bea7c3bf4763ba14d8ee96faf8a74bChris Lattner } 514f5bf3cf7e2a0ff1ca884a83a8b56b5a57f8a5c80Jim Grosbach 515f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner return 0; 516f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner} 517f9f065e45500823cdeb25bde2154d871ab6e9125Chris Lattner 518