Форум программистов [Powered by Invision Power Board]

Код

//---------------------------------------------------------------------------


#pragma hdrstop

#include "ConsRedirect.h"


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

#include "LastError.h"

//#pragma package(smart_init)

#define   MY_PIPE_SIZE  1024
#define   OUT_BUFF_SIZE 512
#define   READ_FD 0
#define   WRITE_FD 1
#define   BEEP_CHAR 7

char szBuffer[OUT_BUFF_SIZE];

int fdStdErrPipe[2] = {-100,-100};
int fdStdOutPipe[2] = {-100, -100};

class Sincro
{
    CRITICAL_SECTION cs;
public:
    Sincro()
    {
        InitializeCriticalSection(&cs);
        EnterCriticalSection(&cs);
    }

    ~Sincro()
    {
        LeaveCriticalSection(&cs);
        DeleteCriticalSection(&cs);
    }
};

//---------------------------------------------------------------------------

const char * REDIRECTEDSTD::ConsoleDfnTitle()
{
    return  "FOR_HIDE";
}

int REDIRECTEDSTD::GetStdOutWriteEnd()
{
    return   fdStdOutPipe[WRITE_FD];
}
int REDIRECTEDSTD::GetStdErrWriteEnd()
{
     return fdStdErrPipe[WRITE_FD];
}

void REDIRECTEDSTD::RedirectStd()
{

    int fdStdOut;
    int fdStdErr;

    Sincro sinc;

    _pipe(fdStdOutPipe, MY_PIPE_SIZE, _O_BINARY );
    _pipe(fdStdErrPipe, MY_PIPE_SIZE, _O_BINARY );

    fdStdOut = _dup(_fileno(stdout));
    fdStdErr = _dup(_fileno(stderr));

#ifdef __BORLANDC__
    dup2(fdStdOutPipe[WRITE_FD], _fileno(stdout));
    dup2(fdStdErrPipe[WRITE_FD], _fileno(stderr));
#else
    _dup2(fdStdOutPipe[WRITE_FD], _fileno(stdout));
    _dup2(fdStdErrPipe[WRITE_FD], _fileno(stderr));
#endif
//    _close(fdStdOutPipe[WRITE_FD]);
//    _close(fdStdErrPipe[WRITE_FD]);

    _close(fdStdOut);
    _close(fdStdErr);
    SetConsoleTitle(ConsoleDfnTitle());
    HWND h = FindWindow(NULL, ConsoleDfnTitle());
    if(h)
    {
        ShowWindow(h, false);
    }
}

std::string  REDIRECTEDSTD::ReadStdErr()
{
    if(fdStdErrPipe[READ_FD] == -100) return "";
    Sincro sinc;
    szBuffer[0] = 0;
    int nOutRead = 0;

    fflush(stderr);

    if(!_eof(fdStdErrPipe[READ_FD]))
    {
        nOutRead = _read(fdStdErrPipe[READ_FD],
                            szBuffer, OUT_BUFF_SIZE);
        if(nOutRead)
        {
            szBuffer[nOutRead] = 0;
        }
    }

    return  std::string(szBuffer);
}

std::string  REDIRECTEDSTD::ReadStdOut()
{
    if(fdStdOutPipe[READ_FD] == -100) return "";
    Sincro sinc;
    szBuffer[0] = 0;
    int nOutRead = 0;

    fflush(stdout);

    if(!_eof(fdStdOutPipe[READ_FD]))
    {
        nOutRead = _read(fdStdOutPipe[READ_FD],
                            szBuffer, OUT_BUFF_SIZE);
        if(nOutRead)
        {
            szBuffer[nOutRead] = 0;
        }
    }
    return  std::string(szBuffer);
}

std::string REDIRECTEDSTD::ReadAll()
{
    std::string s  = ReadStdErr();
    std::string s1 = ReadStdOut();

    if( !s1.empty() ) s = s + "\n" + s1;
    return s;
}