ShellExec.cpp

Go to the documentation of this file.

#include <sstream>
#include <string.h>
#include <stdio.h>
#include <sys/wait.h>
#include <Poco/TemporaryFile.h>

#include "ShellExec.hpp"

//-----------------------------------------------------------------------------
ShellExec::ShellExec(const std::string& cmd_)
:inherited(cmd_), exitStatus(0)
{
}
//-----------------------------------------------------------------------------
bool ShellExec::exec(std::ostream& os, std::ostream& es)
{
  const std::string tmpFile = Poco::TemporaryFile::tempName("");

  _IsError = false;
  char buff[SZ]={0};
  FILE* f = NULL;
  FILE* err = freopen (tmpFile.c_str(),"w", stderr);

  f=popen(cmd.c_str(), "r");
  if(f!=NULL)
  {
    while(fgets(buff, sizeof(buff), f))
    {
      os << buff;
    }
    int ret = pclose(f);
    exitStatus = ret;

    if (ret != 0)
    {
      _IsError = true;
      es << strerror(errno) << std::endl;
    }
  }
  else
  {
    _IsError = true;
    es << strerror(errno) << std::endl;
  }

  if (err!=NULL)
  {
    if (fclose(err)!=0)
      es << strerror(errno) << std::endl;

    err = fopen(tmpFile.c_str(), "r");
    if (err!=NULL)
    {
      memset(buff, 0, sizeof(buff));
      while(fgets(buff, sizeof(buff), err))
      {
        es << buff;
      }

      if (fclose(err)!=0)
        es << strerror(errno) << std::endl;

      if (remove(tmpFile.c_str())!=0)
        es << strerror(errno) << std::endl;
    }
  }
  return !_IsError;
}
//-----------------------------------------------------------------------------
std::ostream& operator << (std::ostream& os, const ShellExec& shellExec)
{
  const_cast<ShellExec&>(shellExec).exec(os, os);
  return os;
}
//-----------------------------------------------------------------------------
std::string& operator << (std::string& str, const ShellExec& shellExec)
{
  std::stringstream outMsg;
  outMsg << shellExec;
  if (!shellExec.isError())
    str = outMsg.str();
  return str;
}
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------