1

我正在开发一个类似程序的远程shell。为了在远程机器上运行命令，我创建了一个 cmd.exe 进程，并将其 stdin 和 stdout 重定向到一对管道。我使用这些管道发送命令并获得结果。我使用 ReadFile 函数从 cmd.exe 返回命令的输出，但我不知道必须从管道读取的确切数据量。因此，如果数据尚未准备好，则 readfile 将进入阻塞模式并等待数据。

我正在使用 MSVC++ 2010，我的操作系统是 Win7。这是我的代码的一部分作为示例：

void CpipesDlg::OnBnClickedBtnredirstd()
{
    char Cmd[]="dir *.*\r\n";
    char Buff[129];
    CString str;
    HANDLE hStdIn_Read, hStdIn_Write;
    HANDLE hStdOut_Read, hStdOut_Write;
    SECURITY_ATTRIBUTES sAttr;

    STARTUPINFOA StartInf;
    PROCESS_INFORMATION procInf;

    DWORD dwBytesToWrite,dwBytesReadFrom;

    sAttr.nLength = sizeof(sAttr);
    sAttr.bInheritHandle = TRUE;
    sAttr.lpSecurityDescriptor = NULL;

    CreatePipe(&hStdIn_Read,&hStdIn_Write,&sAttr,0);
    CreatePipe(&hStdOut_Read,&hStdOut_Write,&sAttr,0);

    //SetHandleInformation(hStdIn_Read, HANDLE_FLAG_INHERIT, 0);
    //SetHandleInformation(hStdIn_Write, HANDLE_FLAG_INHERIT, 0);

    memset(&StartInf,0, sizeof(StartInf));
    memset(&procInf,0,sizeof(procInf));

    StartInf.cb = sizeof(StartInf);
    StartInf.dwFlags = STARTF_USESTDHANDLES;
    StartInf.hStdError = hStdOut_Write;
    StartInf.hStdOutput = hStdOut_Write;
    StartInf.hStdInput = hStdIn_Read;

    WriteFile(hStdIn_Write,Cmd,sizeof(Cmd),&dwBytesToWrite,NULL);

    if(!CreateProcessA(NULL,"cmd.exe",NULL,NULL,TRUE,NORMAL_PRIORITY_CLASS | CREATE_NO_WINDOW ,NULL,NULL,&StartInf,&procInf))
    {
        MessageBoxA(m_hWnd, "Can't Create Process","Error",MB_OK | MB_ICONERROR);
    }
        WriteFile(hStdIn_Write,Cmd,sizeof(Cmd),&dwBytesToWrite,NULL);

    BOOL bSUCCESS =TRUE;
    Sleep(100);
    while(bSUCCESS)
    {

        BOOL bResult  = ReadFile(hStdOut_Read,Buff,70,&dwBytesReadFrom,NULL);
        if(!bResult)
        {
            break;
        }
        Buff[dwBytesReadFrom]=0;
        str+= Buff;
        bSUCCESS = dwBytesReadFrom!=0;
    }
    m_Disp = str;
    UpdateData(FALSE);
    CloseHandle(hStdIn_Read);
    CloseHandle(hStdIn_Write);
    CloseHandle(hStdOut_Read);
    CloseHandle(hStdOut_Write);
}

在上面的代码中，在调试模式下，第一次调用 ReadFile 函数返回真实数据，但最后一次调用它会阻塞，因为没有足够的数据可以读取。

这是我的问题：如何避免阻塞模式或获取要读取的确切字节数？

问候！

4

1 回答 1

1

您可能想要使用该PeekNamedPipe功能，它允许您查询管道以查看在任何给定时间有多少可用数据。如果lpBuffer是，NULL则没有数据被读取。

于 2013-11-02T10:16:55.123 回答