MFC控制台指令的重定向输出到文件方法

前一篇文章讲过如何隐藏控制台窗口，并把结果返回。但是控制台有个输出重定向输出到文件功能，这个在createprocess中是不支持的。因此那个函数改进如下

TCHAR strLog[1024]={0};

CString strCmp =_T("test.exe   abc>abc.h");

RunCMD(strLog,strCmp.GetBuffer(),TRUE);//这个是用法

BOOL RunCMD(TCHAR *pLog,TCHAR *pCmd,BOOL bReconstruct)//最后这个参数说明是否重定向到文件

{

    SECURITY_ATTRIBUTES sa;

    BOOL ret;

    sa.nLength=256;

    sa.lpSecurityDescriptor=0;

    sa.bInheritHandle=TRUE;

 

    HANDLE hReadPipe1,hWritePipe1,hReadPipe2,hWritePipe2;

    HANDLE hOpenFile;

    CString strReconstruct;

    ////////////

    _SECURITY_ATTRIBUTES  sec;

 

    sec.bInheritHandle=TRUE;//这个东西不添加就不会写入文件   解决运行完毕但是文件为空的问题

    sec.lpSecurityDescriptor=NULL;//这个东西不添加就会返回998

    /////////////

    ret=CreatePipe(&hReadPipe2,&hWritePipe2,&sa,0);

    if(bReconstruct)

    {

        strReconstruct = pCmd;

        strReconstruct = strReconstruct.Right(strReconstruct.GetLength()-strReconstruct.ReverseFind('>')-1);

        hOpenFile = (HANDLE)CreateFile(strReconstruct.GetBuffer(), GENERIC_WRITE|GENERIC_READ, FILE_SHARE_READ,&sec, CREATE_NEW,NULL, NULL);

        if (hOpenFile == INVALID_HANDLE_VALUE)

        {

             int iRes = GetLastError();

             hOpenFile = NULL;

             return FALSE;

        }

 

    }

    else

    {

        ret=CreatePipe(&hReadPipe1,&hWritePipe1,&sa,0);//创建两个匿名管道

 

    }

 

    STARTUPINFO si;

    PROCESS_INFORMATION pi;

    ZeroMemory( &si,sizeof(si));

    si.cb = sizeof(si);

    si.dwFlags=STARTF_USESHOWWINDOW|STARTF_USESTDHANDLES;

    si.wShowWindow=SW_HIDE;

    si.hStdInput=hReadPipe2;

 

    if(bReconstruct)

    {

        si.hStdOutput=si.hStdError=hOpenFile;

    }

    else

    {

        si.hStdOutput=si.hStdError=hWritePipe1;

    }

    ZeroMemory( &pi,sizeof(pi));

    unsigned long lBytesRead;

    char Buff[1024]={0};

    CString strLog;

    CreateProcess(NULL,pCmd,NULL,NULL,TRUE,0,NULL,NULL,&si,&pi);

    while(1&&!bReconstruct)

    {

        ret=PeekNamedPipe(hReadPipe1,Buff,1024,&lBytesRead,0,0);//管道是否有数据可读

        if(lBytesRead&&pLog)

        {

             ret=ReadFile(hReadPipe1,Buff,lBytesRead,&lBytesRead,0);//读取管道里的数据

            

             strLog.Format(_T("%S"),Buff);

             StrCpy(pLog,strLog.GetBuffer());

             if(!ret)

                 break;

            

 

        }

        else

             break;

 

 

    }

 

    if(bReconstruct)

    {

        CloseHandle(hOpenFile);

    }

    CloseHandle( pi.hProcess );

    CloseHandle( pi.hThread );

    return TRUE;

}