From fc2271f3f5cf25a739f625ba7dfc6d618d26db7b Mon Sep 17 00:00:00 2001 From: Viktor Kutuzov Date: Sat, 18 Jul 2009 18:39:24 +0000 Subject: Require a remote command to exit with the exit status of the test program or with 255 if an error occurred. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@76323 91177308-0d34-0410-b5e6-96231b3b80d8 --- tools/bugpoint/ToolRunner.cpp | 70 +++++++++++++++++++++++++++++++++------ utils/bugpoint/RemoteRunSafely.sh | 21 +++++------- 2 files changed, 68 insertions(+), 23 deletions(-) diff --git a/tools/bugpoint/ToolRunner.cpp b/tools/bugpoint/ToolRunner.cpp index a5e1e8b8a4..e6db42a08b 100644 --- a/tools/bugpoint/ToolRunner.cpp +++ b/tools/bugpoint/ToolRunner.cpp @@ -46,8 +46,8 @@ namespace { ToolExecutionError::~ToolExecutionError() throw() { } -/// RunProgramWithTimeout - This function provides an alternate interface to the -/// sys::Program::ExecuteAndWait interface. +/// RunProgramWithTimeout - This function provides an alternate interface +/// to the sys::Program::ExecuteAndWait interface. /// @see sys:Program::ExecuteAndWait static int RunProgramWithTimeout(const sys::Path &ProgramPath, const char **Args, @@ -60,20 +60,74 @@ static int RunProgramWithTimeout(const sys::Path &ProgramPath, redirects[0] = &StdInFile; redirects[1] = &StdOutFile; redirects[2] = &StdErrFile; - - if (0) { + +#if 0 // For debug purposes + { errs() << "RUN:"; for (unsigned i = 0; Args[i]; ++i) errs() << " " << Args[i]; errs() << "\n"; } +#endif return sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects, NumSeconds, MemoryLimit); } +/// RunProgramRemotelyWithTimeout - This function runs the given program +/// remotely using the given remote client and the sys::Program::ExecuteAndWait. +/// Returns the remote program exit code or reports a remote client error if it +/// fails. Remote client is required to return 255 if it failed or program exit +/// code otherwise. +/// @see sys:Program::ExecuteAndWait +static int RunProgramRemotelyWithTimeout(const sys::Path &RemoteClientPath, + const char **Args, + const sys::Path &StdInFile, + const sys::Path &StdOutFile, + const sys::Path &StdErrFile, + unsigned NumSeconds = 0, + unsigned MemoryLimit = 0) { + const sys::Path* redirects[3]; + redirects[0] = &StdInFile; + redirects[1] = &StdOutFile; + redirects[2] = &StdErrFile; + +#if 0 // For debug purposes + { + errs() << "RUN:"; + for (unsigned i = 0; Args[i]; ++i) + errs() << " " << Args[i]; + errs() << "\n"; + } +#endif + + // Run the program remotely with the remote client + int ReturnCode = sys::Program::ExecuteAndWait(RemoteClientPath, Args, + 0, redirects, NumSeconds, MemoryLimit); + + // Has the remote client fail? + if (255 == ReturnCode) { + std::ostringstream OS; + OS << "\nError running remote client:\n "; + for (const char **Arg = Args; *Arg; ++Arg) + OS << " " << *Arg; + OS << "\n"; + + // The error message is in the output file, let's print it out from there. + std::ifstream ErrorFile(StdOutFile.c_str()); + if (ErrorFile) { + std::copy(std::istreambuf_iterator(ErrorFile), + std::istreambuf_iterator(), + std::ostreambuf_iterator(OS)); + ErrorFile.close(); + } + throw ToolExecutionError(OS.str()); + } + + return ReturnCode; +} static void ProcessFailure(sys::Path ProgPath, const char** Args) { std::ostringstream OS; @@ -680,16 +734,10 @@ int GCC::ExecuteProgram(const std::string &ProgramFile, Timeout, MemoryLimit); } else { outs() << ""; outs().flush(); - int RemoteClientStatus = RunProgramWithTimeout(sys::Path(RemoteClientPath), + return RunProgramRemotelyWithTimeout(sys::Path(RemoteClientPath), &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile), Timeout, MemoryLimit); - if (RemoteClientStatus != 0) { - errs() << "Remote Client failed with an error: " << - RemoteClientStatus << ".\n"; - } } - - return 0; } int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType, diff --git a/utils/bugpoint/RemoteRunSafely.sh b/utils/bugpoint/RemoteRunSafely.sh index 543a22554e..f390e339ea 100644 --- a/utils/bugpoint/RemoteRunSafely.sh +++ b/utils/bugpoint/RemoteRunSafely.sh @@ -2,11 +2,9 @@ # # Program: RemoteRunSafely.sh # -# Synopsis: This script simply runs another program remotely using rsh. -# It always returns the another program exit code. -# -# (?) only exit statuses that indicates that the program could not be executed -# normally is considered to indicate a test failure. +# Synopsis: This script simply runs another program remotely using ssh. +# It always returns the another program exit code or exit with +# code 255 which indicates that the program could not be executed. # # Syntax: # @@ -25,8 +23,8 @@ printUsageAndExit() { echo "Usage:" echo "./RemoteRunSafely.sh [-l ] [-p ] " \ - "[cd ] " - exit 1 + " " + exit 255 } moreArgsExpected() @@ -88,7 +86,7 @@ fi local_program=$WORKING_DIR"/"$PROGRAM if [ ! -x "$local_program" ]; then echo "File "$local_program" does not exist or is not an executable.." - exit 2 + exit 255 fi connection=$RUSER'@'$RHOST @@ -98,11 +96,10 @@ remote="./"$PROGRAM $RCLIENT $connection $RPORT \ 'rm -f '$remote' ; ' \ 'cat > '$remote' ; chmod +x '$remote' ; '$remote' '$*' ; ' \ - 'echo exit $? ; ' \ - 'rm -f '$remote + 'err=$? ; rm -f '$remote' ; exit $err' ) +err=$? -#DEBUG: err=$? #DEBUG: echo script exit $err -#DEBUG: exit $err +exit $err -- cgit v1.2.3-18-g5258