无法通过套接字C发送文件

时间:2015-08-29 11:23:35

标签: c windows sockets

只有当我使用自己的计算机时它才有效,但是当我使用另一台计算机时,收到的文件很乱,而且大小总是另一个。

服务器:

while (1) {
        ZeroMemory(r_buf, MAX_BUFF_RECV - 1);
        bytesReceived = recv(sock, r_buf, FILE_DOWNLOAD_SIZE, 0);

        if (bytesReceived < 0) {
            // ERROR
            puts("[!] recv failed");
            return;
        }
        else if (bytesReceived < FILE_DOWNLOAD_SIZE) {
            // LAST CHUNCK
            if (WriteFile(out, r_buf, bytesReceived, &n, NULL) == FALSE) {
                printf("WriteFile() error %d\n", GetLastError());
            }
            break;
        }
        else {
            if (WriteFile(out, r_buf, bytesReceived, &n, NULL) == FALSE) {
                printf("WriteFile() error %d\n", GetLastError());
            }
        }
    }

客户端:

while (1)
    {
        ZeroMemory(buff, FILE_DOWNLOAD_SIZE);
        nread = fread(buff, 1, FILE_DOWNLOAD_SIZE, f);

        if (nread != FILE_DOWNLOAD_SIZE) {
            send(s, buff, nread, 0);
            break;
        }
        else {
            send(s, buff, nread, 0);
        }
    }
我已经好几个小时了,请告诉我什么是错的

1 个答案:

答案 0 :(得分:1)

在面向连接的传输中,如TCP,send()recv()不是1对1。我们无法保证您在一个数据包中传输send()的内容,或recv()将一次性收到您send()的所有内容。

无法保证send()甚至会接受您提供的所有内容。它会将它可以复制到套接字的出站缓冲区中,并返回复制的实际字节数。因此它可以接受比请求更少的字节。

同样,recv()将复制套接字入站缓冲区中当前可用的任何数据,至少 1个字节但超过指定的字节数,进入指定的缓冲区并返回实际复制的字节数。因此它可以接收比请求更少的字节。

您还需要知道实际传输了多少文件数据,以便知道何时停止读取数据。由于recv()可以返回比请求的更少的字节,因此不能仅依靠缓冲区大小来指示EOF。

您的代码中没有考虑任何这些因素。

尝试更像这样的东西:

服务器:

bool readRaw(SOCKET sock, void *buf, int bufSize)
{
    char *pbuf = (char*) buf;
    while (bufSize > 0)
    {
        int bytesReceived = recv(sock, pbuf, bufSize, 0);
        if (bytesReceived < 0)
        {
            printf("[!] recv() error %d\n", WSAGetLastError());
            return false;
        }
        else if (bytesReceived == 0)
        {
            puts("[!] client disconnected");
            return false;
        }
        else
        {
            pBuf += bytesReceived;
            bufSize -= bytesReceived;
        }
    }
    return true;
}

...

unsigned __int64 fileSize = 0;
if (readRaw(sock, &fileSize, sizeof(fileSize)))
{
    while (fileSize > 0)
    {
        int bufSize = min(fileSize, MAX_BUFF_RECV);

        if (!readRaw(sock, r_buf, bufSize))
            break;

        if (!WriteFile(out, r_buf, bufSize, &n, NULL))
        {
            printf("[!] WriteFile() error %d\n", GetLastError());
            break;
        }

        fileSize -= bufSize;
    }
}

客户端:

bool sendRaw(SOCKET sock, void *buf, int bufSize)
{
    char *pbuf = (char*) buf;
    while (bufSize > 0)
    {
        int bytesSent = send(sock, pbuf, bufSize, 0);
        if (bytesSent < 0)
        {
            printf("[!] send() error %d\n", WSAGetLastError());
            return false;
        }
        else
        {
            pBuf += bytesSent;
            bufSize -= bytesSent;
        }
    }

    return true;
}

...

fseek(f, 0, SEEK_END);
long int pos = ftell(f);
fseek(f, 0, SEEK_SET);

if (pos == -1)
    printf("[!] ftell() error\n");
else
{
    unsigned __int64 fileSize = pos;
    if (sendRaw(s, &fileSize, sizeof(fileSize)))
    {
        while (fileSize > 0)
        {
            int bufSize = min(FILE_DOWNLOAD_SIZE, fileSize);

            nread = fread(buff, 1, bufSize, f);
            if (nread == 0)
            {
                printf("[!] fread() error\n");
                break;
            }

            if (!sendRaw(s, buff, nread))
                break;

            uifileSize -= nread;
        }
    }
}