|
|
|
#include <cerrno>#include <cstdlib>#include <exception>#include <iostream>#include <string>#include <sys/wait.h>#include <unistd.h>#include <cstring>#include "boost/version.hpp"#include "boost/filesystem/convenience.hpp"#include "FWCore/Utilities/interface/EDMException.h"#include "FWCore/Utilities/interface/Exception.h"#include "FWCore/Utilities/interface/RegexMatch.h"#include "FWCore/Utilities/interface/TestHelper.h"Go to the source code of this file.
Functions | |
| int | do_work (int argc, char *argv[], char **env) |
| int | ptomaine (int argc, char *argv[], char **env) |
| int | run_script (std::string const &shell, std::string const &script) |
| int do_work | ( | int | argc, |
| char * | argv[], | ||
| char ** | env | ||
| ) |
Definition at line 57 of file TestHelper.cc.
References dir2webdir::argc, dtNoiseDBValidation_cfg::cerr, gather_cfg::cout, alignCSCRings::e, i, scaleCards::path, run_script(), checkforupdate::scriptname, and edm::untaintString().
Referenced by ptomaine().
{
bf::path currentPath(bf::initial_path().string());
if(argc < 4) {
std::cout << "Usage: " << argv[0] << " shell subdir script1 script2 ... scriptN\n\n"
<< "where shell is the path+shell (e.g., /bin/bash) intended to run the scripts\n"
<< "and subdir is the subsystem/package/subdir in which the scripts are found\n"
<< "(e.g., FWCore/Utilities/test)\n"
<< std::endl;
std::cout << "Current directory is: " << currentPath.string() << '\n';
std::cout << "Current environment:\n";
std::cout << "---------------------\n";
for(int i = 0; env[i] != 0; ++i) std::cout << env[i] << '\n';
std::cout << "---------------------\n";
std::cout << "Executable name: " << argv[0] << '\n';
return -1;
}
char const* goodDirectory = "[A-Za-z0-9/_.-]+";
for(int i = 0; i < argc; ++i) {
std::cout << "argument " << i << ": " << argv[i] << '\n';
}
std::string shell(argv[1]);
std::cerr << "shell is: " << shell << '\n';
std::cout << "Current directory is: " << currentPath.string() << '\n';
// It is unclear about which of these environment variables should
// be used.
char const* topdir = getenv("SCRAMRT_LOCALRT");
if(!topdir) topdir = getenv("LOCALRT");
try {
if(!edm::untaintString(topdir, goodDirectory)) {
std::cerr << "Invalid top directory '" << topdir << "'" << std::endl;;
return -1;
}
}
catch(std::runtime_error const& e) {
std::cerr << "Invalid top directory '" << topdir << "'" << std::endl;;
std::cerr << "e.what" << std::endl;;
return -1;
}
char const* arch = getenv("SCRAM_ARCH");
if(!arch) {
// Try to synthesize SCRAM_ARCH value.
bf::path exepath(argv[0]);
#if (BOOST_VERSION / 100000) >= 1 && ((BOOST_VERSION / 100) % 1000) >= 47
std::string maybe_arch = exepath.parent_path().filename().string();
#else
std::string maybe_arch = exepath.branch_path().leaf();
#endif
if(setenv("SCRAM_ARCH", maybe_arch.c_str(), 1) != 0) {
std::cerr << "SCRAM_ARCH not set and attempt to set it failed\n";
return -1;
}
arch = getenv("SCRAM_ARCH");
}
int rc = 0;
if(!topdir) {
std::cerr << "Neither SCRAMRT_LOCALRT nor LOCALRT is defined" << std::endl;;
return -1;
}
try {
if(!edm::untaintString(argv[2], goodDirectory)) {
std::cerr << "Invalid test directory '" << argv[2] << "'" << std::endl;;
return -1;
}
}
catch(std::runtime_error const& e) {
std::cerr << "Invalid test directory '" << argv[2] << "'" << std::endl;;
std::cerr << "e.what" << std::endl;;
return -1;
}
std::string testdir(topdir); testdir += "/src/"; testdir += argv[2];
std::string tmpdir(topdir); tmpdir += "/tmp/"; tmpdir += arch;
std::string testbin(topdir); testbin += "/test/"; testbin += arch;
std::cout << "topdir is: " << topdir << '\n';
std::cout << "testdir is: " << testdir << '\n';
std::cout << "tmpdir is: " << tmpdir << '\n';
std::cout << "testbin is: " << testbin << '\n';
if(setenv("LOCAL_TEST_DIR", testdir.c_str(),1) != 0) {
std::cerr << "Could not set LOCAL_TEST_DIR to " << testdir << std::endl;;
return -1;
}
if(setenv("LOCAL_TMP_DIR", tmpdir.c_str(),1) != 0) {
std::cerr << "Could not set LOCAL_TMP_DIR to " << tmpdir << std::endl;;
return -1;
}
if(setenv("LOCAL_TOP_DIR", topdir,1) != 0) {
std::cerr << "Could not set LOCAL_TOP_DIR to " << topdir << std::endl;;
return -1;
}
if(setenv("LOCAL_TEST_BIN", testbin.c_str(),1) != 0) {
std::cerr << "Could not set LOCAL_TEST_BIN to " << testbin << std::endl;;
return -1;
}
testdir += "/";
for(int i = 3; i < argc && rc == 0; ++i) {
std::string scriptname(testdir);
scriptname += argv[i];
std::cout << "Running script: " << scriptname << std::endl;
rc = run_script(shell, scriptname);
}
std::cout << "status = " << rc << std::endl;
return rc == 0 ? 0 : -1;
}
| int ptomaine | ( | int | argc, |
| char * | argv[], | ||
| char ** | env | ||
| ) |
Definition at line 177 of file TestHelper.cc.
References dtNoiseDBValidation_cfg::cerr, do_work(), exception, and x.
{
int rc = 1;
try {
rc = do_work(argc, argv, env);
}
catch(edm::Exception& x) {
std::cerr << "Caught an edm::Exception in "
<< argv[0] << '\n'
<< x;
}
catch(cms::Exception& x) {
std::cerr << "Caught a cms::Exception in "
<< argv[0] << '\n'
<< x;
}
catch(std::exception& x) {
std::cerr << "Caught a std::exception in "
<< argv[0] << '\n'
<< x.what();
}
catch(...) {
std::cerr << "Caught an unknown exception in "
<< argv[0];
}
return rc;
}
| int run_script | ( | std::string const & | shell, |
| std::string const & | script | ||
| ) |
Definition at line 23 of file TestHelper.cc.
References dtNoiseDBValidation_cfg::cerr, compareJSON::const, evf::utils::pid, and ntuplemaker::status.
Referenced by do_work().
{
pid_t pid = 0;
int status = 0;
if((pid = fork()) < 0) {
std::cerr << "fork failed, to run " << script << std::endl;;
return -1;
}
if(pid == 0) { // child
execlp(shell.c_str(), "sh", "-c", script.c_str(), static_cast<char const*>(0));
std::cerr << "child failed becuase '" << strerror(errno) << "'\n";
_exit(127); // signal parent and children processes
} else { // parent
while(waitpid(pid, &status, 0) < 0) {
if(errno != EINTR) {
std::cerr << "child process failed " << strerror(errno) << "\n";
status = -1;
break;
} else {
if(WIFSIGNALED(status)) {
std::cerr << "child existed because of a signal " << WTERMSIG(status) << "\n";
}
}
}
if(WIFSIGNALED(status)) {
std::cerr << "child existed because of a signal " << WTERMSIG(status) << "\n";
}
if(WIFEXITED(status)) {
}
}
return status;
}
1.7.3