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) throw(IOException);

    static Process spawn(const std::string& cmd,
        const std::vector<std::string>& args,
        int niceval=0) throw(IOException);

    static Process spawn(const std::string& cmd) throw(IOException);

    void kill(int signal) throw(IOException);

    int wait(bool hang, int *status) throw(IOException);

    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)
        throw(IOException);

    static void removePidFile();

    static void addEffectiveCapability(int cap) throw(Exception);

    static void clearEffectiveCapability(int cap) throw(Exception);

    static bool getEffectiveCapability(int cap) throw(Exception);

    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