ExecutionDriver.cpp revision 97182985d530dbef488696c95a39c14fe56c995b
14a10645c70199c8d8567fbc46312158c419720abChris Lattner//===- ExecutionDriver.cpp - Allow execution of LLVM program --------------===// 27c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell// 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. 77c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell// 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> 22d0fde30ce850b78371fd1386338350591f9ff494Brian Gaekeusing namespace llvm; 23d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke 244a10645c70199c8d8567fbc46312158c419720abChris Lattnernamespace { 254a10645c70199c8d8567fbc46312158c419720abChris Lattner // OutputType - Allow the user to specify the way code should be run, to test 264a10645c70199c8d8567fbc46312158c419720abChris Lattner // for miscompilation. 274a10645c70199c8d8567fbc46312158c419720abChris Lattner // 284a10645c70199c8d8567fbc46312158c419720abChris Lattner enum OutputType { 29b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke AutoPick, RunLLI, RunJIT, RunLLC, RunCBE 304a10645c70199c8d8567fbc46312158c419720abChris Lattner }; 314148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman 324a10645c70199c8d8567fbc46312158c419720abChris Lattner cl::opt<OutputType> 334a10645c70199c8d8567fbc46312158c419720abChris Lattner InterpreterSel(cl::desc("Specify how LLVM code should be executed:"), 34b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke cl::values(clEnumValN(AutoPick, "auto", "Use best guess"), 35b687d82b029ba448b978e2c9ed424ee5d1f9f093Misha Brukman clEnumValN(RunLLI, "run-int", 36b687d82b029ba448b978e2c9ed424ee5d1f9f093Misha Brukman "Execute with the interpreter"), 375073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman clEnumValN(RunJIT, "run-jit", "Execute with JIT"), 385073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman clEnumValN(RunLLC, "run-llc", "Compile with LLC"), 395073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman clEnumValN(RunCBE, "run-cbe", "Compile with CBE"), 404d143ee01988e1b52e106ffccbb313937ca5e886Chris Lattner clEnumValEnd), 41b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke cl::init(AutoPick)); 423c053a0a98e19f5ac8905011ad4c8238b08d6aeaChris Lattner 43c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke cl::opt<bool> 44c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke CheckProgramExitCode("check-exit-code", 45eed80e23751ecc50c1fa5604f67be4b826d5b417Misha Brukman cl::desc("Assume nonzero exit code is failure (default on)"), 46c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke cl::init(true)); 47c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke 483c053a0a98e19f5ac8905011ad4c8238b08d6aeaChris Lattner cl::opt<std::string> 493c053a0a98e19f5ac8905011ad4c8238b08d6aeaChris Lattner InputFile("input", cl::init("/dev/null"), 503c053a0a98e19f5ac8905011ad4c8238b08d6aeaChris Lattner cl::desc("Filename to pipe in as stdin (default: /dev/null)")); 517dac658792425c10274594782d6fcf10208a16f0Chris Lattner 527dac658792425c10274594782d6fcf10208a16f0Chris Lattner cl::list<std::string> 537dac658792425c10274594782d6fcf10208a16f0Chris Lattner AdditionalSOs("additional-so", 547dac658792425c10274594782d6fcf10208a16f0Chris Lattner cl::desc("Additional shared objects to load " 557dac658792425c10274594782d6fcf10208a16f0Chris Lattner "into executing programs")); 567d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner 577d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner cl::opt<unsigned> 587d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner TimeoutValue("timeout", cl::init(300), cl::value_desc("seconds"), 597d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner cl::desc("Number of seconds program is allowed to run before it " 607d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner "is killed (default is 300s), 0 disables timeout")); 614a10645c70199c8d8567fbc46312158c419720abChris Lattner} 624a10645c70199c8d8567fbc46312158c419720abChris Lattner 63d0fde30ce850b78371fd1386338350591f9ff494Brian Gaekenamespace llvm { 64fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner // Anything specified after the --args option are taken as arguments to the 65fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner // program being debugged. 66fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner cl::list<std::string> 67fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner InputArgv("args", cl::Positional, cl::desc("<program arguments>..."), 6860083e2fc0f1165b7511757449f1ee3852b7229cChris Lattner cl::ZeroOrMore, cl::PositionalEatsArgs); 69636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke 70636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke cl::list<std::string> 71636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke ToolArgv("tool-args", cl::Positional, cl::desc("<tool arguments>..."), 7260083e2fc0f1165b7511757449f1ee3852b7229cChris Lattner cl::ZeroOrMore, cl::PositionalEatsArgs); 73fa76183e8e28985dfd17b1d6291c939dab4cbe1dChris Lattner} 749d679cbc6cb5c7dc8cca87a1e1548c480fb056b8Misha Brukman 754a10645c70199c8d8567fbc46312158c419720abChris Lattner//===----------------------------------------------------------------------===// 764a10645c70199c8d8567fbc46312158c419720abChris Lattner// BugDriver method implementation 774a10645c70199c8d8567fbc46312158c419720abChris Lattner// 784a10645c70199c8d8567fbc46312158c419720abChris Lattner 794a10645c70199c8d8567fbc46312158c419720abChris Lattner/// initializeExecutionEnvironment - This method is used to set up the 804a10645c70199c8d8567fbc46312158c419720abChris Lattner/// environment for executing LLVM programs. 814a10645c70199c8d8567fbc46312158c419720abChris Lattner/// 824a10645c70199c8d8567fbc46312158c419720abChris Lattnerbool BugDriver::initializeExecutionEnvironment() { 834a10645c70199c8d8567fbc46312158c419720abChris Lattner std::cout << "Initializing execution environment: "; 844a10645c70199c8d8567fbc46312158c419720abChris Lattner 854148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman // Create an instance of the AbstractInterpreter interface as specified on 864148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman // the command line 877bb11547e497d7b8fc87f61c1089eee808e3a1eeChris Lattner cbe = 0; 884a10645c70199c8d8567fbc46312158c419720abChris Lattner std::string Message; 89636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke 90cc876a7421f6dbcca98446058d5f0637092c6e1aChris Lattner switch (InterpreterSel) { 91b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke case AutoPick: 92b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = RunCBE; 93636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = cbe = AbstractInterpreter::createCBE(getToolName(), Message, 94636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 95b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke if (!Interpreter) { 96b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = RunJIT; 97636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createJIT(getToolName(), Message, 98636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 99b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke } 100b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke if (!Interpreter) { 101b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = RunLLC; 102636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createLLC(getToolName(), Message, 103636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 104b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke } 105b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke if (!Interpreter) { 106b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = RunLLI; 107636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createLLI(getToolName(), Message, 108636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 109b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke } 110b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke if (!Interpreter) { 111b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke InterpreterSel = AutoPick; 112b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke Message = "Sorry, I can't automatically select an interpreter!\n"; 113b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke } 114b5ee509be2ce78badc11fd649dda90ec78394d1aBrian Gaeke break; 115769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner case RunLLI: 116636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createLLI(getToolName(), Message, 117636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 118769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner break; 119769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner case RunLLC: 120636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createLLC(getToolName(), Message, 121636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 122769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner break; 123769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner case RunJIT: 124636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = AbstractInterpreter::createJIT(getToolName(), Message, 125636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 126769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner break; 127769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner case RunCBE: 128636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke Interpreter = cbe = AbstractInterpreter::createCBE(getToolName(), Message, 129636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke &ToolArgv); 130769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner break; 131cc876a7421f6dbcca98446058d5f0637092c6e1aChris Lattner default: 1324148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman Message = "Sorry, this back-end is not supported by bugpoint right now!\n"; 133cc876a7421f6dbcca98446058d5f0637092c6e1aChris Lattner break; 1344a10645c70199c8d8567fbc46312158c419720abChris Lattner } 1354148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman std::cerr << Message; 1364a10645c70199c8d8567fbc46312158c419720abChris Lattner 137a259c9be2acc9528ec7feb3cfd51dcde36d87bb3Misha Brukman // Initialize auxiliary tools for debugging 1387bb11547e497d7b8fc87f61c1089eee808e3a1eeChris Lattner if (!cbe) { 139636df3d7ec51a8c6fd759a98853ff709ae54d64eBrian Gaeke cbe = AbstractInterpreter::createCBE(getToolName(), Message, &ToolArgv); 1407bb11547e497d7b8fc87f61c1089eee808e3a1eeChris Lattner if (!cbe) { std::cout << Message << "\nExiting.\n"; exit(1); } 1417bb11547e497d7b8fc87f61c1089eee808e3a1eeChris Lattner } 142769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner gcc = GCC::create(getToolName(), Message); 143a259c9be2acc9528ec7feb3cfd51dcde36d87bb3Misha Brukman if (!gcc) { std::cout << Message << "\nExiting.\n"; exit(1); } 144a259c9be2acc9528ec7feb3cfd51dcde36d87bb3Misha Brukman 1454a10645c70199c8d8567fbc46312158c419720abChris Lattner // If there was an error creating the selected interpreter, quit with error. 1464a10645c70199c8d8567fbc46312158c419720abChris Lattner return Interpreter == 0; 1474a10645c70199c8d8567fbc46312158c419720abChris Lattner} 1484a10645c70199c8d8567fbc46312158c419720abChris Lattner 149ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner/// compileProgram - Try to compile the specified module, throwing an exception 150ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner/// if an error occurs, or returning normally if not. This is used for code 151ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner/// generation crash testing. 152ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner/// 153ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattnervoid BugDriver::compileProgram(Module *M) { 154ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner // Emit the program to a bytecode file... 15597182985d530dbef488696c95a39c14fe56c995bReid Spencer sys::Path BytecodeFile ("bugpoint-test-program.bc"); 15697182985d530dbef488696c95a39c14fe56c995bReid Spencer BytecodeFile.makeUnique(); 15797182985d530dbef488696c95a39c14fe56c995bReid Spencer if (writeProgramToFile(BytecodeFile.toString(), M)) { 158ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner std::cerr << ToolName << ": Error emitting bytecode to file '" 159ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner << BytecodeFile << "'!\n"; 160ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner exit(1); 161ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner } 162ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner 163ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner // Remove the temporary bytecode file when we are done. 16497182985d530dbef488696c95a39c14fe56c995bReid Spencer FileRemover BytecodeFileRemover(BytecodeFile.toString()); 165ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner 166ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner // Actually compile the program! 16797182985d530dbef488696c95a39c14fe56c995bReid Spencer Interpreter->compileProgram(BytecodeFile.toString()); 168ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner} 169ea9212ca964ff6587227016f86a44160e586a4c8Chris Lattner 1704a10645c70199c8d8567fbc46312158c419720abChris Lattner 1714a10645c70199c8d8567fbc46312158c419720abChris Lattner/// executeProgram - This method runs "Program", capturing the output of the 1724a10645c70199c8d8567fbc46312158c419720abChris Lattner/// program to a file, returning the filename of the file. A recommended 1734a10645c70199c8d8567fbc46312158c419720abChris Lattner/// filename may be optionally specified. 1744a10645c70199c8d8567fbc46312158c419720abChris Lattner/// 1754a10645c70199c8d8567fbc46312158c419720abChris Lattnerstd::string BugDriver::executeProgram(std::string OutputFile, 1765073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman std::string BytecodeFile, 177769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner const std::string &SharedObj, 178c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke AbstractInterpreter *AI, 179c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke bool *ProgramExitedNonzero) { 180769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner if (AI == 0) AI = Interpreter; 181769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner assert(AI && "Interpreter should have been created already!"); 1824a10645c70199c8d8567fbc46312158c419720abChris Lattner bool CreatedBytecode = false; 1834a10645c70199c8d8567fbc46312158c419720abChris Lattner if (BytecodeFile.empty()) { 1844a10645c70199c8d8567fbc46312158c419720abChris Lattner // Emit the program to a bytecode file... 18597182985d530dbef488696c95a39c14fe56c995bReid Spencer sys::Path uniqueFilename("bugpoint-test-program.bc"); 18697182985d530dbef488696c95a39c14fe56c995bReid Spencer uniqueFilename.makeUnique(); 18797182985d530dbef488696c95a39c14fe56c995bReid Spencer BytecodeFile = uniqueFilename.toString(); 1884a10645c70199c8d8567fbc46312158c419720abChris Lattner 1894a10645c70199c8d8567fbc46312158c419720abChris Lattner if (writeProgramToFile(BytecodeFile, Program)) { 1904a10645c70199c8d8567fbc46312158c419720abChris Lattner std::cerr << ToolName << ": Error emitting bytecode to file '" 1915073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman << BytecodeFile << "'!\n"; 1924a10645c70199c8d8567fbc46312158c419720abChris Lattner exit(1); 1934a10645c70199c8d8567fbc46312158c419720abChris Lattner } 1944a10645c70199c8d8567fbc46312158c419720abChris Lattner CreatedBytecode = true; 1954a10645c70199c8d8567fbc46312158c419720abChris Lattner } 1964a10645c70199c8d8567fbc46312158c419720abChris Lattner 1979709272675feaca030cce14d0f5eb7f342b3fc1dChris Lattner // Remove the temporary bytecode file when we are done. 1989709272675feaca030cce14d0f5eb7f342b3fc1dChris Lattner FileRemover BytecodeFileRemover(BytecodeFile, CreatedBytecode); 1999709272675feaca030cce14d0f5eb7f342b3fc1dChris Lattner 2004a10645c70199c8d8567fbc46312158c419720abChris Lattner if (OutputFile.empty()) OutputFile = "bugpoint-execution-output"; 2015073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2024a10645c70199c8d8567fbc46312158c419720abChris Lattner // Check to see if this is a valid output filename... 20397182985d530dbef488696c95a39c14fe56c995bReid Spencer sys::Path uniqueFile(OutputFile); 20497182985d530dbef488696c95a39c14fe56c995bReid Spencer uniqueFile.makeUnique(); 20597182985d530dbef488696c95a39c14fe56c995bReid Spencer OutputFile = uniqueFile.toString(); 2064a10645c70199c8d8567fbc46312158c419720abChris Lattner 207769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner // Figure out which shared objects to run, if any. 2087dac658792425c10274594782d6fcf10208a16f0Chris Lattner std::vector<std::string> SharedObjs(AdditionalSOs); 209769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner if (!SharedObj.empty()) 210769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner SharedObjs.push_back(SharedObj); 211769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner 2124a10645c70199c8d8567fbc46312158c419720abChris Lattner // Actually execute the program! 213769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner int RetVal = AI->ExecuteProgram(BytecodeFile, InputArgv, InputFile, 2147d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner OutputFile, SharedObjs, TimeoutValue); 2157d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner 2167d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner if (RetVal == -1) { 2177d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner std::cerr << "<timeout>"; 2187d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner static bool FirstTimeout = true; 2197d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner if (FirstTimeout) { 2207d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner std::cout << "\n" 2217d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner "*** Program execution timed out! This mechanism is designed to handle\n" 2227d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner " programs stuck in infinite loops gracefully. The -timeout option\n" 2237d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner " can be used to change the timeout threshold or disable it completely\n" 2247d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner " (with -timeout=0). This message is only displayed once.\n"; 2257d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner FirstTimeout = false; 2267d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner } 2277d91e49ff7bcc0fd10a54d45a6185bb05adf3d20Chris Lattner } 228769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner 229c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke if (ProgramExitedNonzero != 0) 230c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke *ProgramExitedNonzero = (RetVal != 0); 2314a10645c70199c8d8567fbc46312158c419720abChris Lattner 2324a10645c70199c8d8567fbc46312158c419720abChris Lattner // Return the filename we captured the output to. 2334a10645c70199c8d8567fbc46312158c419720abChris Lattner return OutputFile; 2344a10645c70199c8d8567fbc46312158c419720abChris Lattner} 2354a10645c70199c8d8567fbc46312158c419720abChris Lattner 236c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke/// executeProgramWithCBE - Used to create reference output with the C 237c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke/// backend, if reference output is not provided. 238c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke/// 239c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaekestd::string BugDriver::executeProgramWithCBE(std::string OutputFile) { 240c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke bool ProgramExitedNonzero; 241c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke std::string outFN = executeProgram(OutputFile, "", "", 242c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke (AbstractInterpreter*)cbe, 243c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke &ProgramExitedNonzero); 244c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke if (ProgramExitedNonzero) { 245c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke std::cerr 246c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke << "Warning: While generating reference output, program exited with\n" 247c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke << "non-zero exit code. This will NOT be treated as a failure.\n"; 248c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke CheckProgramExitCode = false; 249c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke } 250c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke return outFN; 251c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke} 2525073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 253a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattnerstd::string BugDriver::compileSharedObject(const std::string &BytecodeFile) { 2545073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman assert(Interpreter && "Interpreter should have been created already!"); 255a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattner std::string OutputCFile; 2565073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2575073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman // Using CBE 2585073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman cbe->OutputC(BytecodeFile, OutputCFile); 2595073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2605073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman#if 0 /* This is an alternative, as yet unimplemented */ 2615073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman // Using LLC 262a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattner std::string Message; 2634148556a9c767fdf3047ae8e004a759356b893ddMisha Brukman LLC *llc = createLLCtool(Message); 2645073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman if (llc->OutputAsm(BytecodeFile, OutputFile)) { 2655073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman std::cerr << "Could not generate asm code with `llc', exiting.\n"; 2665073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman exit(1); 2675073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman } 2685073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman#endif 2695073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 270a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattner std::string SharedObjectFile; 271769f1fe6728ffb5627ae0cedc392576d6e701a5aChris Lattner if (gcc->MakeSharedObject(OutputCFile, GCC::CFile, SharedObjectFile)) 272a0f5b15e1eb8642d92b3141a6b88a5729ea979dcChris Lattner exit(1); 2735073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2745073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman // Remove the intermediate C file 2755073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman removeFile(OutputCFile); 2765073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2776ebe44d22f6dd1ab9f7aa1f3cfd02be52145d535Chris Lattner return "./" + SharedObjectFile; 2785073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman} 2795073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2805073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman 2814a10645c70199c8d8567fbc46312158c419720abChris Lattner/// diffProgram - This method executes the specified module and diffs the output 2824a10645c70199c8d8567fbc46312158c419720abChris Lattner/// against the file specified by ReferenceOutputFile. If the output is 2834a10645c70199c8d8567fbc46312158c419720abChris Lattner/// different, true is returned. 2844a10645c70199c8d8567fbc46312158c419720abChris Lattner/// 2855073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukmanbool BugDriver::diffProgram(const std::string &BytecodeFile, 2865073336cd4da5df4ae13a167582d1dc90f32e4e0Misha Brukman const std::string &SharedObject, 287640f22e66d90439857a97a83896ee68c4f7128c9Chris Lattner bool RemoveBytecode) { 288c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke bool ProgramExitedNonzero; 289c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke 2904a10645c70199c8d8567fbc46312158c419720abChris Lattner // Execute the program, generating an output file... 291c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke std::string Output = executeProgram("", BytecodeFile, SharedObject, 0, 292c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke &ProgramExitedNonzero); 293c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke 294c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke // If we're checking the program exit code, assume anything nonzero is bad. 29558d84ced5f7367335e43ea54e5daf32a4fdaefb6Chris Lattner if (CheckProgramExitCode && ProgramExitedNonzero) { 29658d84ced5f7367335e43ea54e5daf32a4fdaefb6Chris Lattner removeFile(Output); 29758d84ced5f7367335e43ea54e5daf32a4fdaefb6Chris Lattner if (RemoveBytecode) removeFile(BytecodeFile); 298c5cad211d6ec50fe90a0a716dee701c6c4721385Brian Gaeke return true; 29958d84ced5f7367335e43ea54e5daf32a4fdaefb6Chris Lattner } 3004a10645c70199c8d8567fbc46312158c419720abChris Lattner 30165f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner std::string Error; 3024a10645c70199c8d8567fbc46312158c419720abChris Lattner bool FilesDifferent = false; 30365f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner if (DiffFiles(ReferenceOutputFile, Output, &Error)) { 30465f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner if (!Error.empty()) { 305eed80e23751ecc50c1fa5604f67be4b826d5b417Misha Brukman std::cerr << "While diffing output: " << Error << '\n'; 30665f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner exit(1); 30765f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner } 30865f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner FilesDifferent = true; 30965f62790d6e3e7f7ceb9cd12a7a51a66d95a3b03Chris Lattner } 3101a28a2b76d16913fe38f47032d31620a55dcce2eChris Lattner 3111a28a2b76d16913fe38f47032d31620a55dcce2eChris Lattner // Remove the generated output. 3121a28a2b76d16913fe38f47032d31620a55dcce2eChris Lattner removeFile(Output); 3134a10645c70199c8d8567fbc46312158c419720abChris Lattner 3141a28a2b76d16913fe38f47032d31620a55dcce2eChris Lattner // Remove the bytecode file if we are supposed to. 315640f22e66d90439857a97a83896ee68c4f7128c9Chris Lattner if (RemoveBytecode) removeFile(BytecodeFile); 3164a10645c70199c8d8567fbc46312158c419720abChris Lattner return FilesDifferent; 3174a10645c70199c8d8567fbc46312158c419720abChris Lattner} 31891eabc13d3a456cc4b387d3d7fdb041d976732c7Misha Brukman 31991eabc13d3a456cc4b387d3d7fdb041d976732c7Misha Brukmanbool BugDriver::isExecutingJIT() { 32091eabc13d3a456cc4b387d3d7fdb041d976732c7Misha Brukman return InterpreterSel == RunJIT; 32191eabc13d3a456cc4b387d3d7fdb041d976732c7Misha Brukman} 322d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke 323