c++ - recv 在接收到 transmitFile() 函数后没有停止

Question

我正在为类使用 Winsocket 编写一个小型服务器/客户端文件传输，它基本上可以工作，除了在我收到文件并将其写入我的硬盘后无法在套接字上收到任何更多消息。

传输代码看起来像这样:

    long size = GetFileSize(hFile, NULL);
TransmitFile(_socket, hFile, size, 0, NULL, NULL, TF_DISCONNECT);
ok = ::recv(_socket, cantwortfilename, 100, 0); // getting a confirmation (1)
cantwortfilename[ok] = '\0';
cout << cantwortfilename << endl;
char test[] = "ok!";
::send(_socket, test, strlen(test), 0); // to test if server receives again (2)

我尝试用 0 而不是 size 得到了相同的结果。

现在服务器端接收:

    HANDLE hFile = CreateFile(filepathlong, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

while (1)
{
    ReadFile((HANDLE)_socket, &buffer, MAX_PATH, &dwbr, &AsyncInfo);
    GetOverlappedResult((HANDLE)_socket, &AsyncInfo, &dwbr, TRUE);
    if (dwbr == 0) break;
    WriteFile(hFile, buffer, dwbr, &dwbr, NULL);
}
char test[] = "alles ok!";
::send(_socket, test, strlen(test), 0); // sending the confirmation (1)

CloseHandle(hFile);

 i = ::recv(_socket, test, 10, 0); // receiving the test (2)
test[i] = '\0';
cout << "empfangen:" << test << endl;

据我所知，文件的传输工作正常(试过 rar jpg .h)和

::send(_socket, test, strlen(test), 0); // sending the confirmation (1)

出去也很好。

但是之后的接收没有给我任何信息？还是空的？我猜是空的，因为 recv 也不会阻止程序。但是当我发出它时，“i”将为 0。

为了检查 while(1) 循环中的内容是否出现了某种错误，我尝试了另一种方法来接收文件。

第二次尝试:

    int r;
ofstream file(filepath, std::ios::out | std::ios::binary | std::ios::trunc);
char *memblock;
int size = 7766; // was the size of the file i was testing with
memblock = new char[size];
memset(memblock, 0, size);
if (file.is_open()){
    //while memory blocks are still being received
    while ((r = recv(_socket, memblock, 256, 0)) != 0)
    {
        //if there's a socket error, abort
        if (r == SOCKET_ERROR)
        {
            cout << "error" << endl;
        }
        //write client's file blocks to file on server
        file.write(memblock, r);
    }
    delete[] memblock;
    //finished sending memory blocks; file is completely transferred.
    file.close();
}

之后，再次发送一个具有相同结果的 recv。该文件有效，但再次收到时我发现了一些空的东西。

所以谁能告诉我为什么以及如何解决这个问题？尽可能少的改变？

谢谢，马丁

编辑:现在对我有用的代码:

    char csize[256];
rc = recv(_socket, csize, 256, 0);
csize[rc] = '\0';
int size = atoi(csize);
int bytes_read = 0, len = 0;

int r;
ofstream file(filepath, std::ios::out | std::ios::binary | std::ios::trunc);
char *memblock;
memblock = new char[size];
memset(memblock, 0, size);

if (file.is_open()){
    while (bytes_read < size){
        len = recv(_socket, memblock + bytes_read, size - bytes_read, 0);
        bytes_read += len;
    }
    file.write(memblock, size);
    delete[] memblock;
    file.close();
}

Answer 1

您的发射器正在使用 TF_DSCONNECT 标志调用 TransmitFile()。这将在文件发送完成后关闭套接字。一旦 TransmitFile() 退出，发送器就不能再发送任何数据(我不确定 TransmitFile() 是否只进行半双工关闭以关闭发送方向但让接收方向保持打开状态，或者如果它在发送和接收方向上都进行了全双工关闭 - 我会谨慎行事并假设后者)。

根据您所展示的内容，您需要使用 shutdown() 和 closesocket() 而不是 TF_DISCONNECT 以便您拥有更多控制权关于实际断开连接的时间和方式。

此外，TransmitFile() 不发送文件大小，只发送文件数据，因此您的接收方无法知道文件数据何时真正完成接收，除非它等待超时(这不是可靠的解决方案)。您需要在发送文件数据之前将发送器更改为发送文件大小，然后将接收器更改为先读取文件大小然后循环直到接收到指定的字节数。