WinSock Non-Blocking I/O

WinSock Non-Blocking I/O

本文关键字：Non-Blocking WinSock      更新时间：2023-10-16

我有一个WSASocket，当连接时调用CreateProcess和服务器cmd.exe，我想在进程hStdInput和套接字句柄之间实现一个管道来解析通过套接字发送的命令，除了当我运行像"ping 127.0.0.1"这样的命令并且必须等待输出时， 在我通过套接字发送更多数据之前，什么都没有显示，似乎我的 ReadFile 调用阻止了 hStdOut 处理程序发送任何内容。有什么办法可以解决这个问题吗？请不要被我的代码冒犯，我正在编写这个项目作为学习练习，任何帮助将不胜感激。

int syncShell(SOCKET *ConnectSocket) {
int iResult = 0;
printf("Spawning processn");
char Process[] = "C:\Windows\System32\cmd.exe";
STARTUPINFO sinfo;
PROCESS_INFORMATION pinfo;
memset(&sinfo, 0, sizeof(sinfo));
sinfo.cb = sizeof(sinfo);
sinfo.dwFlags = (STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW);
// create pipe for external commands
SECURITY_ATTRIBUTES  saAttr;
saAttr.nLength = sizeof(SECURITY_ATTRIBUTES); 
saAttr.bInheritHandle = TRUE; 
saAttr.lpSecurityDescriptor = NULL;
HANDLE hReadPipe = NULL, hWritePipe = NULL;
iResult = CreatePipe(&hReadPipe, &hWritePipe, &saAttr, DEFAULT_BUFLEN);
if (iResult == 0) {
printf("Pipe Error");
}
sinfo.hStdOutput = sinfo.hStdError = (HANDLE) *ConnectSocket;
sinfo.hStdInput = hReadPipe;
if (!CreateProcessA(NULL, Process, NULL, NULL, TRUE, 0, NULL, NULL, &sinfo, &pinfo)) {
printf("CreateProcess failed (%d).n", GetLastError());
}
// implement pipe logic
char buf[DEFAULT_BUFLEN];
DWORD len = 0;
WSABUF DataBuf;
while (1) {
// causing the block?
iResult = ReadFile((HANDLE) *ConnectSocket, buf, DEFAULT_BUFLEN, &len, NULL);
if (iResult == 0) {
printf("File Error or non-blocking");
}
else {
printf("%d: %.*sn", len, len, buf);
WriteFile(hWritePipe, buf, len, NULL, NULL);
}
Sleep(1000);
}
WaitForSingleObject(pinfo.hProcess, INFINITE); // waits till proc finishes
CloseHandle(pinfo.hProcess);
CloseHandle(pinfo.hThread);
printf("Process exitedn");
return 0;
}

首先，根据 [ReadFile] 文档：

对于异步读取操作，hFile 可以是任何句柄 通过CreateFile函数用FILE_FLAG_OVERLAPPED标志打开， 或socket或accept函数返回的套接字句柄。

默认情况下，socket创建带有WSA_FLAG_OVERLAPPED的套接字句柄。 如果传递重叠句柄并将ReadFile的最后一个参数设置为NULL，您将收到错误代码 87(ERROR_INVALID_PARAMETER(。 要使用的示例重叠：

OVERLAPPED oRead = { 0 };
oRead.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);  
iResult = ReadFile((HANDLE)ConnectSocket, buf, DEFAULT_BUFLEN, &len, &oRead);
if (!iResult && GetLastError() == ERROR_IO_PENDING)
{
WaitForSingleObject(oRead.hEvent, INFINITE);
}
buf[oRead.InternalHigh] = 0;  //set string terminator for printf
printf("%sn", buf);
WriteFile(hWritePipe1, buf, oRead.InternalHigh, NULL, NULL);

最好直接使用recv()：

iResult = recv(ConnectSocket, buf, DEFAULT_BUFLEN, 0);
buf[iResult] = 0;  //set string terminator for printf
printf("%sn", buf);
WriteFile(hWritePipe1, buf, iResult, NULL, NULL);

此外，重叠的套接字可用于将 IO 重定向到子进程 您可以创建 2 个管道来与子进程通信：

iResult = CreatePipe(&hReadPipe1, &hWritePipe1, &saAttr, DEFAULT_BUFLEN);
if (iResult == 0) {
printf("Pipe Error");
}
iResult = CreatePipe(&hReadPipe2, &hWritePipe2, &saAttr, DEFAULT_BUFLEN);
if (iResult == 0) {
printf("Pipe Error");
}

从子进程(cmd.exe(读取并发送到客户端。

或

只需使用WSASocket而不是socket，并且不要指定WSA_FLAG_OVERLAPPED。(推荐(

相关文章：