b2subprocess.H

//------------------------------------------------------------------------
// b2subprocess.H --
//
// written by doreille
//
// Copyright (c) 2009 SMR Engineering & Development SA
//               2502 Bienne, Switzerland
//
// All Rights Reserved.  Proprietary source code.  The contents of
// this file may not be disclosed to third parties, copied or
// duplicated in any form, in whole or in part, without the prior
// written permission of SMR.
//------------------------------------------------------------------------
 
#ifndef B2SUBPROCESS_H_
#define B2SUBPROCESS_H_
 
#include <sys/wait.h>
 
#include <sstream>
 
#include "utils/b2exception.H"
 
namespace b2000 {
 
class Subprocess : public std::ostringstream {
public:
    int call(
          const std::string& out = "cout", const std::string& err = "cerr",
          bool raise_exception_on_error = false) {
        std::string s = str();
        if (out != "cout") {
            s += " >";
            s += out;
        }
        if (err != "cerr") {
            if (err == "cout") {
                s += " 2>&1";
            } else {
                s += " 2>";
                s += err;
            }
        }
 
        returncode = ::system(s.c_str());
        if (WIFSIGNALED(returncode) || (WEXITSTATUS(returncode) && raise_exception_on_error)) {
            Exception() << get_error() << THROW;
        }
        return WEXITSTATUS(returncode);
    }
 
    std::string get_error() const {
        std::ostringstream res;
        if (WIFSIGNALED(returncode)) {
            res << "Command " << str() << " killed by signal " << WTERMSIG(returncode);
        } else if (WEXITSTATUS(returncode) > 0) {
            res << "Command " << str() << " terminates with nonzero exit status "
                << WEXITSTATUS(returncode);
        }
        return res.str();
    }
 
private:
    ::pid_t returncode;
};
 
}  // namespace b2000
 
#endif /* B2SUBPROCESS_H_ */