mirror of
https://github.com/RPCSX/llvm.git
synced 2024-11-25 20:59:51 +00:00
Apply timeouts and memory limits in more places. In particular, when
bugpoint does "Running the code generator to test for a crash" this gets you a crash if llc goes into an infinite loop or uses up vast amounts of memory. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@104485 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
parent
414c0c43d3
commit
4139630212
@ -312,7 +312,7 @@ void BugDriver::compileProgram(Module *M, std::string *Error) {
|
||||
FileRemover BitcodeFileRemover(BitcodeFile, !SaveTemps);
|
||||
|
||||
// Actually compile the program!
|
||||
Interpreter->compileProgram(BitcodeFile.str(), Error);
|
||||
Interpreter->compileProgram(BitcodeFile.str(), Error, Timeout, MemoryLimit);
|
||||
}
|
||||
|
||||
|
||||
|
@ -133,7 +133,9 @@ static int RunProgramRemotelyWithTimeout(const sys::Path &RemoteClientPath,
|
||||
return ReturnCode;
|
||||
}
|
||||
|
||||
static std::string ProcessFailure(sys::Path ProgPath, const char** Args) {
|
||||
static std::string ProcessFailure(sys::Path ProgPath, const char** Args,
|
||||
unsigned Timeout = 0,
|
||||
unsigned MemoryLimit = 0) {
|
||||
std::ostringstream OS;
|
||||
OS << "\nError running tool:\n ";
|
||||
for (const char **Arg = Args; *Arg; ++Arg)
|
||||
@ -148,7 +150,8 @@ static std::string ProcessFailure(sys::Path ProgPath, const char** Args) {
|
||||
exit(1);
|
||||
}
|
||||
RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
|
||||
ErrorFilename); // FIXME: check return code ?
|
||||
ErrorFilename, Timeout, MemoryLimit);
|
||||
// FIXME: check return code ?
|
||||
|
||||
// Print out the error messages generated by GCC if possible...
|
||||
std::ifstream ErrorFile(ErrorFilename.c_str());
|
||||
@ -353,7 +356,8 @@ AbstractInterpreter *AbstractInterpreter::createCustom(
|
||||
// LLC Implementation of AbstractIntepreter interface
|
||||
//
|
||||
GCC::FileType LLC::OutputCode(const std::string &Bitcode,
|
||||
sys::Path &OutputAsmFile, std::string &Error) {
|
||||
sys::Path &OutputAsmFile, std::string &Error,
|
||||
unsigned Timeout, unsigned MemoryLimit) {
|
||||
const char *Suffix = (UseIntegratedAssembler ? ".llc.o" : ".llc.s");
|
||||
sys::Path uniqueFile(Bitcode + Suffix);
|
||||
std::string ErrMsg;
|
||||
@ -386,14 +390,17 @@ GCC::FileType LLC::OutputCode(const std::string &Bitcode,
|
||||
errs() << "\n";
|
||||
);
|
||||
if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
|
||||
sys::Path(), sys::Path(), sys::Path()))
|
||||
Error = ProcessFailure(sys::Path(LLCPath), &LLCArgs[0]);
|
||||
sys::Path(), sys::Path(), sys::Path(),
|
||||
Timeout, MemoryLimit))
|
||||
Error = ProcessFailure(sys::Path(LLCPath), &LLCArgs[0],
|
||||
Timeout, MemoryLimit);
|
||||
return UseIntegratedAssembler ? GCC::ObjectFile : GCC::AsmFile;
|
||||
}
|
||||
|
||||
void LLC::compileProgram(const std::string &Bitcode, std::string *Error) {
|
||||
void LLC::compileProgram(const std::string &Bitcode, std::string *Error,
|
||||
unsigned Timeout, unsigned MemoryLimit) {
|
||||
sys::Path OutputAsmFile;
|
||||
OutputCode(Bitcode, OutputAsmFile, *Error);
|
||||
OutputCode(Bitcode, OutputAsmFile, *Error, Timeout, MemoryLimit);
|
||||
OutputAsmFile.eraseFromDisk();
|
||||
}
|
||||
|
||||
@ -408,7 +415,8 @@ int LLC::ExecuteProgram(const std::string &Bitcode,
|
||||
unsigned MemoryLimit) {
|
||||
|
||||
sys::Path OutputAsmFile;
|
||||
GCC::FileType FileKind = OutputCode(Bitcode, OutputAsmFile, *Error);
|
||||
GCC::FileType FileKind = OutputCode(Bitcode, OutputAsmFile, *Error, Timeout,
|
||||
MemoryLimit);
|
||||
FileRemover OutFileRemover(OutputAsmFile, !SaveTemps);
|
||||
|
||||
std::vector<std::string> GCCArgs(ArgsForGCC);
|
||||
@ -528,7 +536,8 @@ AbstractInterpreter *AbstractInterpreter::createJIT(const char *Argv0,
|
||||
}
|
||||
|
||||
GCC::FileType CBE::OutputCode(const std::string &Bitcode,
|
||||
sys::Path &OutputCFile, std::string &Error) {
|
||||
sys::Path &OutputCFile, std::string &Error,
|
||||
unsigned Timeout, unsigned MemoryLimit) {
|
||||
sys::Path uniqueFile(Bitcode+".cbe.c");
|
||||
std::string ErrMsg;
|
||||
if (uniqueFile.makeUnique(true, &ErrMsg)) {
|
||||
@ -556,14 +565,15 @@ GCC::FileType CBE::OutputCode(const std::string &Bitcode,
|
||||
errs() << "\n";
|
||||
);
|
||||
if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
|
||||
sys::Path()))
|
||||
Error = ProcessFailure(LLCPath, &LLCArgs[0]);
|
||||
sys::Path(), Timeout, MemoryLimit))
|
||||
Error = ProcessFailure(LLCPath, &LLCArgs[0], Timeout, MemoryLimit);
|
||||
return GCC::CFile;
|
||||
}
|
||||
|
||||
void CBE::compileProgram(const std::string &Bitcode, std::string *Error) {
|
||||
void CBE::compileProgram(const std::string &Bitcode, std::string *Error,
|
||||
unsigned Timeout, unsigned MemoryLimit) {
|
||||
sys::Path OutputCFile;
|
||||
OutputCode(Bitcode, OutputCFile, *Error);
|
||||
OutputCode(Bitcode, OutputCFile, *Error, Timeout, MemoryLimit);
|
||||
OutputCFile.eraseFromDisk();
|
||||
}
|
||||
|
||||
@ -577,7 +587,7 @@ int CBE::ExecuteProgram(const std::string &Bitcode,
|
||||
unsigned Timeout,
|
||||
unsigned MemoryLimit) {
|
||||
sys::Path OutputCFile;
|
||||
OutputCode(Bitcode, OutputCFile, *Error);
|
||||
OutputCode(Bitcode, OutputCFile, *Error, Timeout, MemoryLimit);
|
||||
|
||||
FileRemover CFileRemove(OutputCFile, !SaveTemps);
|
||||
|
||||
|
@ -112,14 +112,17 @@ public:
|
||||
/// compileProgram - Compile the specified program from bitcode to executable
|
||||
/// code. This does not produce any output, it is only used when debugging
|
||||
/// the code generator. It returns false if the code generator fails.
|
||||
virtual void compileProgram(const std::string &Bitcode, std::string *Error) {}
|
||||
virtual void compileProgram(const std::string &Bitcode, std::string *Error,
|
||||
unsigned Timeout = 0, unsigned MemoryLimit = 0) {}
|
||||
|
||||
/// OutputCode - Compile the specified program from bitcode to code
|
||||
/// understood by the GCC driver (either C or asm). If the code generator
|
||||
/// fails, it sets Error, otherwise, this function returns the type of code
|
||||
/// emitted.
|
||||
virtual GCC::FileType OutputCode(const std::string &Bitcode,
|
||||
sys::Path &OutFile, std::string &Error) {
|
||||
sys::Path &OutFile, std::string &Error,
|
||||
unsigned Timeout = 0,
|
||||
unsigned MemoryLimit = 0) {
|
||||
Error = "OutputCode not supported by this AbstractInterpreter!";
|
||||
return GCC::AsmFile;
|
||||
}
|
||||
@ -161,7 +164,8 @@ public:
|
||||
/// compileProgram - Compile the specified program from bitcode to executable
|
||||
/// code. This does not produce any output, it is only used when debugging
|
||||
/// the code generator. Returns false if the code generator fails.
|
||||
virtual void compileProgram(const std::string &Bitcode, std::string *Error);
|
||||
virtual void compileProgram(const std::string &Bitcode, std::string *Error,
|
||||
unsigned Timeout = 0, unsigned MemoryLimit = 0);
|
||||
|
||||
virtual int ExecuteProgram(const std::string &Bitcode,
|
||||
const std::vector<std::string> &Args,
|
||||
@ -180,7 +184,9 @@ public:
|
||||
/// fails, it sets Error, otherwise, this function returns the type of code
|
||||
/// emitted.
|
||||
virtual GCC::FileType OutputCode(const std::string &Bitcode,
|
||||
sys::Path &OutFile, std::string &Error);
|
||||
sys::Path &OutFile, std::string &Error,
|
||||
unsigned Timeout = 0,
|
||||
unsigned MemoryLimit = 0);
|
||||
};
|
||||
|
||||
|
||||
@ -206,7 +212,8 @@ public:
|
||||
/// compileProgram - Compile the specified program from bitcode to executable
|
||||
/// code. This does not produce any output, it is only used when debugging
|
||||
/// the code generator. Returns false if the code generator fails.
|
||||
virtual void compileProgram(const std::string &Bitcode, std::string *Error);
|
||||
virtual void compileProgram(const std::string &Bitcode, std::string *Error,
|
||||
unsigned Timeout = 0, unsigned MemoryLimit = 0);
|
||||
|
||||
virtual int ExecuteProgram(const std::string &Bitcode,
|
||||
const std::vector<std::string> &Args,
|
||||
@ -225,7 +232,9 @@ public:
|
||||
/// fails, it sets Error, otherwise, this function returns the type of code
|
||||
/// emitted.
|
||||
virtual GCC::FileType OutputCode(const std::string &Bitcode,
|
||||
sys::Path &OutFile, std::string &Error);
|
||||
sys::Path &OutFile, std::string &Error,
|
||||
unsigned Timeout = 0,
|
||||
unsigned MemoryLimit = 0);
|
||||
};
|
||||
|
||||
} // End llvm namespace
|
||||
|
Loading…
Reference in New Issue
Block a user