Process.h

Go to the documentation of this file.

// -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 4; tab-width: 4; -*-
// vim: set shiftwidth=4 softtabstop=4 expandtab:
/*
 ********************************************************************
 ** NIDAS: NCAR In-situ Data Acquistion Software
 **
 ** 2006, Copyright University Corporation for Atmospheric Research
 **
 ** This program is free software; you can redistribute it and/or modify
 ** it under the terms of the GNU General Public License as published by
 ** the Free Software Foundation; either version 2 of the License, or
 ** (at your option) any later version.
 **
 ** This program is distributed in the hope that it will be useful,
 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 ** GNU General Public License for more details.
 **
 ** The LICENSE.txt file accompanying this software contains
 ** a copy of the GNU General Public License. If it is not found,
 ** write to the Free Software Foundation, Inc.,
 ** 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 **
 ********************************************************************
*/
 
#ifndef NIDAS_UTIL_PROCESS_H
#define NIDAS_UTIL_PROCESS_H
 
#include "ThreadSupport.h"
#include "IOException.h"
#include "auto_ptr.h"
 
#include <string>
#include <vector>
#include <map>
#include <memory>
#include <signal.h>
#include <sys/types.h>
#include <sys/wait.h>
 
#include <nidas/Config.h>
 
#ifdef HAVE_SYS_CAPABILITY_H 
#include <sys/capability.h>
#endif
 
#include <ext/stdio_filebuf.h>
 
namespace nidas { namespace util {
 
class Process {
public:
 
    Process(pid_t pid);
 
    Process();
 
    Process(const Process& x);
 
    Process& operator=(const Process& x);
 
    ~Process();
 
    static Process spawn(const std::string& cmd,
                         const std::vector<std::string>& args,
                         const std::map<std::string, std::string>& env,
                         int niceval=0);
 
    static Process spawn(const std::string& cmd,
                         const std::vector<std::string>& args,
                         int niceval=0);
 
    static Process spawn(const std::string& cmd);
 
    void kill(int signal);
 
    int wait(bool hang, int *status);
 
    int getInFd() const { return _infd; }
 
    std::ostream& inStream() { return *_instream_ap.get(); }
 
    void closeIn();
 
    int getOutFd() const { return _outfd; }
 
    std::istream& outStream() { return *_outstream_ap.get(); }
 
    void closeOut();
 
    int getErrFd() const { return _errfd; }
 
    std::istream& errStream() { return *_errstream_ap.get(); }
 
    void closeErr();
 
    pid_t getPid() const { return _pid; }
 
    static pid_t checkPidFile(const std::string& pidFile);
 
    static void removePidFile();
 
    static void addEffectiveCapability(int cap);
 
    static void clearEffectiveCapability(int cap);
 
    static bool getEffectiveCapability(int cap);
 
    static std::string expandEnvVars(std::string input);
    
    static bool getEnvVar(const std::string& name,std::string& value);
 
    static void setEnvVar(const std::string& name,const std::string& value);
 
    static void clearEnv();
 
    static unsigned long getVMemSize();
 
    static unsigned long getMaxRSS();
 
private:
 
    static std::string _pidFile;
 
    pid_t _pid;
 
    mutable int _infd;
 
    void setInFd(int val);
 
    mutable auto_ptr<__gnu_cxx::stdio_filebuf<char> > _inbuf_ap;
 
    mutable auto_ptr<std::ostream> _instream_ap;
 
    mutable int _outfd;
 
    void setOutFd(int val);
 
    mutable auto_ptr<__gnu_cxx::stdio_filebuf<char> > _outbuf_ap;
 
    mutable auto_ptr<std::istream> _outstream_ap;
 
    mutable int _errfd;
 
    void setErrFd(int val);
 
    mutable auto_ptr<__gnu_cxx::stdio_filebuf<char> > _errbuf_ap;
 
    mutable auto_ptr<std::istream> _errstream_ap;
 
    static std::map<std::string,char*> _environment;
 
    static Mutex _envLock;
};
 
}}  // namespace nidas namespace util
 
#endif