重定向代码

最新推荐文章于 2024-01-02 14:56:05 发布

weixin_34260991

最新推荐文章于 2024-01-02 14:56:05 发布

阅读量110

点赞数

文章标签： python

原文链接：https://my.oschina.net/jjyuangu/blog/616769

版权

2019独角兽企业重金招聘Python工程师标准>>>

#include <fcntl.h>
#include <stdio.h>
#include <string>
#include <io.h>

class StdCapture
{
public:
 StdCapture() : m_capturing(false), m_init(false), m_oldStdOut(0), m_oldStdErr(0)
 {
  m_pipe[READ] = 0;
  m_pipe[WRITE] = 0;
  if (_pipe(m_pipe, 65536, O_BINARY) == -1)
   return;
  m_oldStdOut = _dup(_fileno(stdout));
  m_oldStdErr = _dup(_fileno(stderr));
  if (m_oldStdOut == -1 || m_oldStdErr == -1)
   return;
  m_init = true;
 }
 ~StdCapture()
 {
  if (m_capturing)
  {
   EndCapture();
  }
  if (m_oldStdOut > 0)
   _close(m_oldStdOut);
  if (m_oldStdErr > 0)
   _close(m_oldStdErr);
  if (m_pipe[READ] > 0)
   _close(m_pipe[READ]);
  if (m_pipe[WRITE] > 0)
   _close(m_pipe[WRITE]);
 }

 void BeginCapture()
 {
  if (!m_init)
   return;
  if (m_capturing)
   EndCapture();
  fflush(stdout);
  fflush(stderr);
  _dup2(m_pipe[WRITE], _fileno(stdout));
  _dup2(m_pipe[WRITE], _fileno(stderr));
  m_capturing = true;
 }
 bool EndCapture()
 {
  if (!m_init)
   return false;
  if (!m_capturing)
   return false;
  fflush(stdout);
  fflush(stderr);
  _dup2(m_oldStdOut, _fileno(stdout));
  _dup2(m_oldStdErr, _fileno(stderr));
  m_captured.clear();
  std::string buf;
  const int bufSize = 1024;
  buf.resize(bufSize);
  int bytesRead = 0;
  if (!_eof(m_pipe[READ]))
  {
   bytesRead = _read(m_pipe[READ], &(*buf.begin()), bufSize);
  }
  while (bytesRead == bufSize)
  {
   m_captured += buf;
   bytesRead = 0;
   if (!_eof(m_pipe[READ]))
   {
    bytesRead = _read(m_pipe[READ], &(*buf.begin()), bufSize);
   }
  }
  if (bytesRead > 0)
  {
   buf.resize(bytesRead);
   m_captured += buf;
  }
  return true;
 }
 std::string GetCapture() const
 {
  std::string::size_type idx = m_captured.find_last_not_of("\r\n");
  if (idx == std::string::npos)
  {
   return m_captured;
  }
  else
  {
   return m_captured.substr(0, idx + 1);
  }
 }
private:
 enum PIPES { READ, WRITE };
 int m_pipe[2];
 int m_oldStdOut;
 int m_oldStdErr;
 bool m_capturing;
 bool m_init;
 std::string m_captured;
};

转载于:https://my.oschina.net/jjyuangu/blog/616769