ExecutionDriver.cpp revision 51ab5c8862466bbddcd5c4369779c472978ed309
14a10645c70199c8d8567fbc46312158c419720abChris Lattner//===- ExecutionDriver.cpp - Allow execution of LLVM program --------------===// 23da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman// 37c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell// The LLVM Compiler Infrastructure 47c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell// 57c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell// This file was developed by the LLVM research group and is distributed under 67c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell// the University of Illinois Open Source License. See LICENSE.TXT for details. 73da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman// 87c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell//===----------------------------------------------------------------------===// 94a10645c70199c8d8567fbc46312158c419720abChris Lattner// 104a10645c70199c8d8567fbc46312158c419720abChris Lattner// This file contains code used to execute the program utilizing one of the 114a10645c70199c8d8567fbc46312158c419720abChris Lattner// various ways of running LLVM bytecode. 124a10645c70199c8d8567fbc46312158c419720abChris Lattner// 134a10645c70199c8d8567fbc46312158c419720abChris Lattner//===----------------------------------------------------------------------===// 144a10645c70199c8d8567fbc46312158c419720abChris Lattner 154a10645c70199c8d8567fbc46312158c419720abChris Lattner#include "BugDriver.h" 16b687d82b029ba448b978e2c9ed424ee5d1f9f093Misha Brukman#include "llvm/Support/ToolRunner.h" 17551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/Support/CommandLine.h" 18551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/Support/Debug.h" 19551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/Support/FileUtilities.h" 20551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/Support/SystemUtils.h" 214a10645c70199c8d8567fbc46312158c419720abChris Lattner#include <fstream> 22e31a9ccb9c7568e6f185f667b53c274c0be9e603Chris Lattner#include <iostream> 2351ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer 24d0fde30ce850b78371fd1386338350591f9ff494Brian Gaekeusing namespace llvm; 25d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke 264a10645c70199c8d8567fbc46312158c419720abChris Lattnernamespace { 274a10645c70199c8d8567fbc46312158c419720abChris Lattner // OutputType - Allow the user to specify the way code should be run, to test 284a10645c70199c8d8567fbc46312158c419720abChris Lattner // for miscompilation. 294a10645c70199c8d8567fbc46312158c419720abChris Lattner // 304a10645c70199c8d8567fbc46312158c419720abChris Lattner enum OutputType { 31b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke AutoPick, RunLLI, RunJIT, RunLLC, RunCBE 324a10645c70199c8d8567fbc46312158c419720abChris Lattner }; 334148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman 34a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner cl::opt<double> 35a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner AbsTolerance("abs-tolerance", cl::desc("Absolute error tolerated"), 36a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner cl::init(0.0)); 37a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner cl::opt<double> 38a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner RelTolerance("rel-tolerance", cl::desc("Relative error tolerated"), 39a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner cl::init(0.0)); 40a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner 414a10645c70199c8d8567fbc46312158c419720abChris Lattner cl::opt<OutputType> 424a10645c70199c8d8567fbc46312158c419720abChris Lattner InterpreterSel(cl::desc("Specify how LLVM code should be executed:"), 43b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke cl::values(clEnumValN(AutoPick, "auto", "Use best guess"), 44b687d82b029ba448b978e2c9ed424ee5d1f9f093Misha Brukman clEnumValN(RunLLI, "run-int", 45b687d82b029ba448b978e2c9ed424ee5d1f9f093Misha Brukman "Execute with the interpreter"), 465073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman clEnumValN(RunJIT, "run-jit", "Execute with JIT"), 475073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman clEnumValN(RunLLC, "run-llc", "Compile with LLC"), 485073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman clEnumValN(RunCBE, "run-cbe", "Compile with CBE"), 494d143ee01988e1b52e106ffccbb313937ca5e886Chris Lattner clEnumValEnd), 50b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke cl::init(AutoPick)); 513c053a0a98e19f5ac8905011ad4c8238b08d6aeaChris Lattner 52c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke cl::opt<bool> 53c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke CheckProgramExitCode("check-exit-code", 54eed80e23751ecc50c1fa5604f67be4b826d5b417Misha Brukman cl::desc("Assume nonzero exit code is failure (default on)"), 55c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke cl::init(true)); 56c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke 573c053a0a98e19f5ac8905011ad4c8238b08d6aeaChris Lattner cl::opt<std::string> 583c053a0a98e19f5ac8905011ad4c8238b08d6aeaChris Lattner InputFile("input", cl::init("/dev/null"), 593c053a0a98e19f5ac8905011ad4c8238b08d6aeaChris Lattner cl::desc("Filename to pipe in as stdin (default: /dev/null)")); 607dac658792425c10274594782d6fcf10208a16f0Chris Lattner 617dac658792425c10274594782d6fcf10208a16f0Chris Lattner cl::list<std::string> 627dac658792425c10274594782d6fcf10208a16f0Chris Lattner AdditionalSOs("additional-so", 637dac658792425c10274594782d6fcf10208a16f0Chris Lattner cl::desc("Additional shared objects to load " 647dac658792425c10274594782d6fcf10208a16f0Chris Lattner "into executing programs")); 657d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner 667d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner cl::opt<unsigned> 677d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner TimeoutValue("timeout", cl::init(300), cl::value_desc("seconds"), 687d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner cl::desc("Number of seconds program is allowed to run before it " 697d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner "is killed (default is 300s), 0 disables timeout")); 7051ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer 7151ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer cl::list<std::string> 7251ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer AdditionalLinkerArgs("Xlinker", 7351ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer cl::desc("Additional arguments to pass to the linker")); 744a10645c70199c8d8567fbc46312158c419720abChris Lattner} 754a10645c70199c8d8567fbc46312158c419720abChris Lattner 76d0fde30ce850b78371fd1386338350591f9ff494Brian Gaekenamespace llvm { 77fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner // Anything specified after the --args option are taken as arguments to the 78fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner // program being debugged. 79fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner cl::list<std::string> 80fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner InputArgv("args", cl::Positional, cl::desc("<program arguments>..."), 8160083e2fc0f1165b7511757449f1ee3852b7229cChris Lattner cl::ZeroOrMore, cl::PositionalEatsArgs); 82636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke 83636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke cl::list<std::string> 84636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke ToolArgv("tool-args", cl::Positional, cl::desc("<tool arguments>..."), 8560083e2fc0f1165b7511757449f1ee3852b7229cChris Lattner cl::ZeroOrMore, cl::PositionalEatsArgs); 86fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner} 879d679cbc6cb5c7dc8cca87a1e1548c480fb056b8Misha Brukman 884a10645c70199c8d8567fbc46312158c419720abChris Lattner//===----------------------------------------------------------------------===// 894a10645c70199c8d8567fbc46312158c419720abChris Lattner// BugDriver method implementation 904a10645c70199c8d8567fbc46312158c419720abChris Lattner// 914a10645c70199c8d8567fbc46312158c419720abChris Lattner 924a10645c70199c8d8567fbc46312158c419720abChris Lattner/// initializeExecutionEnvironment - This method is used to set up the 934a10645c70199c8d8567fbc46312158c419720abChris Lattner/// environment for executing LLVM programs. 944a10645c70199c8d8567fbc46312158c419720abChris Lattner/// 954a10645c70199c8d8567fbc46312158c419720abChris Lattnerbool BugDriver::initializeExecutionEnvironment() { 964a10645c70199c8d8567fbc46312158c419720abChris Lattner std::cout << "Initializing execution environment: "; 974a10645c70199c8d8567fbc46312158c419720abChris Lattner 984148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman // Create an instance of the AbstractInterpreter interface as specified on 994148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman // the command line 1007bb11547e497d7b8fc87f61c1089eee808e3a1eeChris Lattner cbe = 0; 1014a10645c70199c8d8567fbc46312158c419720abChris Lattner std::string Message; 102636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke 103cc876a7421f6dbcca98446058d5f0637092c6e1aChris Lattner switch (InterpreterSel) { 104b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke case AutoPick: 105b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = RunCBE; 106636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = cbe = AbstractInterpreter::createCBE(getToolName(), Message, 107636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 108b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke if (!Interpreter) { 109b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = RunJIT; 110636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createJIT(getToolName(), Message, 111636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 112b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke } 113b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke if (!Interpreter) { 114b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = RunLLC; 115636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createLLC(getToolName(), Message, 116636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 117b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke } 118b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke if (!Interpreter) { 119b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = RunLLI; 120636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createLLI(getToolName(), Message, 121636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 122b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke } 123b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke if (!Interpreter) { 124b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = AutoPick; 125b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke Message = "Sorry, I can't automatically select an interpreter!\n"; 126b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke } 127b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke break; 128769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner case RunLLI: 129636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createLLI(getToolName(), Message, 130636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 131769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner break; 132769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner case RunLLC: 133636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createLLC(getToolName(), Message, 134636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 135769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner break; 136769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner case RunJIT: 137636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createJIT(getToolName(), Message, 138636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 139769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner break; 140769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner case RunCBE: 141636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = cbe = AbstractInterpreter::createCBE(getToolName(), Message, 142636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 143769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner break; 144cc876a7421f6dbcca98446058d5f0637092c6e1aChris Lattner default: 1454148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman Message = "Sorry, this back-end is not supported by bugpoint right now!\n"; 146cc876a7421f6dbcca98446058d5f0637092c6e1aChris Lattner break; 1474a10645c70199c8d8567fbc46312158c419720abChris Lattner } 1484148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman std::cerr << Message; 1494a10645c70199c8d8567fbc46312158c419720abChris Lattner 150a259c9be2acc9528ec7feb3cfd51dcde36d87bb3Misha Brukman // Initialize auxiliary tools for debugging 1517bb11547e497d7b8fc87f61c1089eee808e3a1eeChris Lattner if (!cbe) { 152636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke cbe = AbstractInterpreter::createCBE(getToolName(), Message, &ToolArgv); 1537bb11547e497d7b8fc87f61c1089eee808e3a1eeChris Lattner if (!cbe) { std::cout << Message << "\nExiting.\n"; exit(1); } 1547bb11547e497d7b8fc87f61c1089eee808e3a1eeChris Lattner } 155769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner gcc = GCC::create(getToolName(), Message); 156a259c9be2acc9528ec7feb3cfd51dcde36d87bb3Misha Brukman if (!gcc) { std::cout << Message << "\nExiting.\n"; exit(1); } 157a259c9be2acc9528ec7feb3cfd51dcde36d87bb3Misha Brukman 1584a10645c70199c8d8567fbc46312158c419720abChris Lattner // If there was an error creating the selected interpreter, quit with error. 1594a10645c70199c8d8567fbc46312158c419720abChris Lattner return Interpreter == 0; 1604a10645c70199c8d8567fbc46312158c419720abChris Lattner} 1614a10645c70199c8d8567fbc46312158c419720abChris Lattner 162ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner/// compileProgram - Try to compile the specified module, throwing an exception 163ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner/// if an error occurs, or returning normally if not. This is used for code 164ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner/// generation crash testing. 165ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner/// 166ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattnervoid BugDriver::compileProgram(Module *M) { 167ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner // Emit the program to a bytecode file... 16897182985d530dbef488696c95a39c14fe56c995bReid Spencer sys::Path BytecodeFile ("bugpoint-test-program.bc"); 16997182985d530dbef488696c95a39c14fe56c995bReid Spencer BytecodeFile.makeUnique(); 17097182985d530dbef488696c95a39c14fe56c995bReid Spencer if (writeProgramToFile(BytecodeFile.toString(), M)) { 171ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner std::cerr << ToolName << ": Error emitting bytecode to file '" 172ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner << BytecodeFile << "'!\n"; 173ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner exit(1); 174ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner } 175ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner 176ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner // Remove the temporary bytecode file when we are done. 1775f76760c880e6d61c229d2058c5699b033caeae1Reid Spencer FileRemover BytecodeFileRemover(BytecodeFile); 178ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner 179ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner // Actually compile the program! 18097182985d530dbef488696c95a39c14fe56c995bReid Spencer Interpreter->compileProgram(BytecodeFile.toString()); 181ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner} 182ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner 1834a10645c70199c8d8567fbc46312158c419720abChris Lattner 1844a10645c70199c8d8567fbc46312158c419720abChris Lattner/// executeProgram - This method runs "Program", capturing the output of the 1854a10645c70199c8d8567fbc46312158c419720abChris Lattner/// program to a file, returning the filename of the file. A recommended 1864a10645c70199c8d8567fbc46312158c419720abChris Lattner/// filename may be optionally specified. 1874a10645c70199c8d8567fbc46312158c419720abChris Lattner/// 1884a10645c70199c8d8567fbc46312158c419720abChris Lattnerstd::string BugDriver::executeProgram(std::string OutputFile, 1895073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman std::string BytecodeFile, 190769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner const std::string &SharedObj, 191c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke AbstractInterpreter *AI, 192c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke bool *ProgramExitedNonzero) { 193769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner if (AI == 0) AI = Interpreter; 194769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner assert(AI && "Interpreter should have been created already!"); 1954a10645c70199c8d8567fbc46312158c419720abChris Lattner bool CreatedBytecode = false; 1964a10645c70199c8d8567fbc46312158c419720abChris Lattner if (BytecodeFile.empty()) { 1974a10645c70199c8d8567fbc46312158c419720abChris Lattner // Emit the program to a bytecode file... 19897182985d530dbef488696c95a39c14fe56c995bReid Spencer sys::Path uniqueFilename("bugpoint-test-program.bc"); 19997182985d530dbef488696c95a39c14fe56c995bReid Spencer uniqueFilename.makeUnique(); 20097182985d530dbef488696c95a39c14fe56c995bReid Spencer BytecodeFile = uniqueFilename.toString(); 2014a10645c70199c8d8567fbc46312158c419720abChris Lattner 2024a10645c70199c8d8567fbc46312158c419720abChris Lattner if (writeProgramToFile(BytecodeFile, Program)) { 2034a10645c70199c8d8567fbc46312158c419720abChris Lattner std::cerr << ToolName << ": Error emitting bytecode to file '" 2045073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman << BytecodeFile << "'!\n"; 2054a10645c70199c8d8567fbc46312158c419720abChris Lattner exit(1); 2064a10645c70199c8d8567fbc46312158c419720abChris Lattner } 2074a10645c70199c8d8567fbc46312158c419720abChris Lattner CreatedBytecode = true; 2084a10645c70199c8d8567fbc46312158c419720abChris Lattner } 2094a10645c70199c8d8567fbc46312158c419720abChris Lattner 2109709272675feaca030cce14d0f5eb7f342b3fc1dChris Lattner // Remove the temporary bytecode file when we are done. 21106c375bd1d402ddcf2ee8e34005a74a0addbd7ccBrian Gaeke sys::Path BytecodePath (BytecodeFile); 21206c375bd1d402ddcf2ee8e34005a74a0addbd7ccBrian Gaeke FileRemover BytecodeFileRemover(BytecodePath, CreatedBytecode); 2139709272675feaca030cce14d0f5eb7f342b3fc1dChris Lattner 2144a10645c70199c8d8567fbc46312158c419720abChris Lattner if (OutputFile.empty()) OutputFile = "bugpoint-execution-output"; 2155073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2164a10645c70199c8d8567fbc46312158c419720abChris Lattner // Check to see if this is a valid output filename... 21797182985d530dbef488696c95a39c14fe56c995bReid Spencer sys::Path uniqueFile(OutputFile); 21897182985d530dbef488696c95a39c14fe56c995bReid Spencer uniqueFile.makeUnique(); 21997182985d530dbef488696c95a39c14fe56c995bReid Spencer OutputFile = uniqueFile.toString(); 2204a10645c70199c8d8567fbc46312158c419720abChris Lattner 221769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner // Figure out which shared objects to run, if any. 2227dac658792425c10274594782d6fcf10208a16f0Chris Lattner std::vector<std::string> SharedObjs(AdditionalSOs); 223769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner if (!SharedObj.empty()) 224769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner SharedObjs.push_back(SharedObj); 225769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner 22651ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer 22751ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer // If this is an LLC or CBE run, then the GCC compiler might get run to 22851ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer // compile the program. If so, we should pass the user's -Xlinker options 22951ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer // as the GCCArgs. 23051ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer int RetVal = 0; 23151ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer if (InterpreterSel == RunLLC || InterpreterSel == RunCBE) 23251ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer RetVal = AI->ExecuteProgram(BytecodeFile, InputArgv, InputFile, 23351ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer OutputFile, AdditionalLinkerArgs, SharedObjs, 23451ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer TimeoutValue); 23551ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer else 23651ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer RetVal = AI->ExecuteProgram(BytecodeFile, InputArgv, InputFile, 23751ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer OutputFile, std::vector<std::string>(), 23851ab5c8862466bbddcd5c4369779c472978ed309Reid Spencer SharedObjs, TimeoutValue); 2397d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner 2407d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner if (RetVal == -1) { 2417d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner std::cerr << "<timeout>"; 2427d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner static bool FirstTimeout = true; 2437d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner if (FirstTimeout) { 2447d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner std::cout << "\n" 2457d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner "*** Program execution timed out! This mechanism is designed to handle\n" 2467d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner " programs stuck in infinite loops gracefully. The -timeout option\n" 2477d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner " can be used to change the timeout threshold or disable it completely\n" 2487d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner " (with -timeout=0). This message is only displayed once.\n"; 2497d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner FirstTimeout = false; 2507d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner } 2517d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner } 252769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner 253c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke if (ProgramExitedNonzero != 0) 254c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke *ProgramExitedNonzero = (RetVal != 0); 2554a10645c70199c8d8567fbc46312158c419720abChris Lattner 2564a10645c70199c8d8567fbc46312158c419720abChris Lattner // Return the filename we captured the output to. 2574a10645c70199c8d8567fbc46312158c419720abChris Lattner return OutputFile; 2584a10645c70199c8d8567fbc46312158c419720abChris Lattner} 2594a10645c70199c8d8567fbc46312158c419720abChris Lattner 260c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke/// executeProgramWithCBE - Used to create reference output with the C 261c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke/// backend, if reference output is not provided. 262c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke/// 263c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaekestd::string BugDriver::executeProgramWithCBE(std::string OutputFile) { 264c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke bool ProgramExitedNonzero; 265c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke std::string outFN = executeProgram(OutputFile, "", "", 266c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke (AbstractInterpreter*)cbe, 267c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke &ProgramExitedNonzero); 268c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke if (ProgramExitedNonzero) { 269c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke std::cerr 270c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke << "Warning: While generating reference output, program exited with\n" 271c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke << "non-zero exit code. This will NOT be treated as a failure.\n"; 272c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke CheckProgramExitCode = false; 273c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke } 274c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke return outFN; 275c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke} 2765073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 277a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattnerstd::string BugDriver::compileSharedObject(const std::string &BytecodeFile) { 2785073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman assert(Interpreter && "Interpreter should have been created already!"); 2795f76760c880e6d61c229d2058c5699b033caeae1Reid Spencer sys::Path OutputCFile; 2805073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2815073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman // Using CBE 2825073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman cbe->OutputC(BytecodeFile, OutputCFile); 2835073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2845073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman#if 0 /* This is an alternative, as yet unimplemented */ 2855073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman // Using LLC 286a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattner std::string Message; 2874148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman LLC *llc = createLLCtool(Message); 2885073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman if (llc->OutputAsm(BytecodeFile, OutputFile)) { 2895073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman std::cerr << "Could not generate asm code with `llc', exiting.\n"; 2905073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman exit(1); 2915073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman } 2925073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman#endif 2935073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 294a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattner std::string SharedObjectFile; 2953da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman if (gcc->MakeSharedObject(OutputCFile.toString(), GCC::CFile, 2965f76760c880e6d61c229d2058c5699b033caeae1Reid Spencer SharedObjectFile)) 297a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattner exit(1); 2985073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2995073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman // Remove the intermediate C file 300a229c5cce75209047db32c6039aa0b0fd481f049Reid Spencer OutputCFile.eraseFromDisk(); 3015073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 3026ebe44d22f6dd1ab9f7aa1f3cfd02be52145d535Chris Lattner return "./" + SharedObjectFile; 3035073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman} 3045073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 3055073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 3064a10645c70199c8d8567fbc46312158c419720abChris Lattner/// diffProgram - This method executes the specified module and diffs the output 3074a10645c70199c8d8567fbc46312158c419720abChris Lattner/// against the file specified by ReferenceOutputFile. If the output is 3084a10645c70199c8d8567fbc46312158c419720abChris Lattner/// different, true is returned. 3094a10645c70199c8d8567fbc46312158c419720abChris Lattner/// 3105073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukmanbool BugDriver::diffProgram(const std::string &BytecodeFile, 3115073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman const std::string &SharedObject, 312640f22e66d90439857a97a83896ee68c4f7128c9Chris Lattner bool RemoveBytecode) { 313c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke bool ProgramExitedNonzero; 314c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke 3154a10645c70199c8d8567fbc46312158c419720abChris Lattner // Execute the program, generating an output file... 3165f76760c880e6d61c229d2058c5699b033caeae1Reid Spencer sys::Path Output (executeProgram("", BytecodeFile, SharedObject, 0, 3175f76760c880e6d61c229d2058c5699b033caeae1Reid Spencer &ProgramExitedNonzero)); 318c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke 319c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke // If we're checking the program exit code, assume anything nonzero is bad. 32058d84ced5f7367335e43ea54e5daf32a4fdaefb6Chris Lattner if (CheckProgramExitCode && ProgramExitedNonzero) { 321a229c5cce75209047db32c6039aa0b0fd481f049Reid Spencer Output.eraseFromDisk(); 3223da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman if (RemoveBytecode) 323a229c5cce75209047db32c6039aa0b0fd481f049Reid Spencer sys::Path(BytecodeFile).eraseFromDisk(); 324c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke return true; 32558d84ced5f7367335e43ea54e5daf32a4fdaefb6Chris Lattner } 3264a10645c70199c8d8567fbc46312158c419720abChris Lattner 32765f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner std::string Error; 3284a10645c70199c8d8567fbc46312158c419720abChris Lattner bool FilesDifferent = false; 329a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner if (int Diff = DiffFilesWithTolerance(sys::Path(ReferenceOutputFile), 330a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner sys::Path(Output.toString()), 331a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner AbsTolerance, RelTolerance, &Error)) { 332a328c51bb99666ee0c045a57ea6d6ce2b0198f9bChris Lattner if (Diff == 2) { 333eed80e23751ecc50c1fa5604f67be4b826d5b417Misha Brukman std::cerr << "While diffing output: " << Error << '\n'; 33465f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner exit(1); 33565f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner } 33665f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner FilesDifferent = true; 33765f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner } 3383da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman 3391a28a2b76d16913fe38f47032d31620a55dcce2eChris Lattner // Remove the generated output. 340a229c5cce75209047db32c6039aa0b0fd481f049Reid Spencer Output.eraseFromDisk(); 3414a10645c70199c8d8567fbc46312158c419720abChris Lattner 3421a28a2b76d16913fe38f47032d31620a55dcce2eChris Lattner // Remove the bytecode file if we are supposed to. 34300b16889ab461b7ecef1c91ade101186b7f1fce2Jeff Cohen if (RemoveBytecode) 344a229c5cce75209047db32c6039aa0b0fd481f049Reid Spencer sys::Path(BytecodeFile).eraseFromDisk(); 3454a10645c70199c8d8567fbc46312158c419720abChris Lattner return FilesDifferent; 3464a10645c70199c8d8567fbc46312158c419720abChris Lattner} 34791eabc13d3a456cc4b387d3d7fdb041d976732c7Misha Brukman 34891eabc13d3a456cc4b387d3d7fdb041d976732c7Misha Brukmanbool BugDriver::isExecutingJIT() { 34991eabc13d3a456cc4b387d3d7fdb041d976732c7Misha Brukman return InterpreterSel == RunJIT; 35091eabc13d3a456cc4b387d3d7fdb041d976732c7Misha Brukman} 351d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke 352