读取整个文件并通过套接字发送

Question

我制作了一个应该传输文件的服务器和客户端。 我试图让它读取整个文件并发送。 但现在看来，我遇到了问题。 当客户端连接时，服务器应自动发送文件。 但是文件是空的，不知道问题出在哪里 您可以看到我正在尝试发送 .txt 文件。但我想以后发送一个大文件，但不能大于1MB。）

编辑：

图片在这里：http://img819.imageshack.us/img819/8259/aadi.jpg

• 左侧：我尝试发送的文件。
• 右侧：我收到的文件

问题：我收到的文件已损坏，无法使用。

服务器：

#include <WinSock2.h>
#include <Windows.h>
#include <stdio.h>
#include <iostream>
using namespace std;

#pragma comment(lib, "Ws2_32.lib")
#define Port 6000

SOCKET Socket, Sub;
WSADATA Winsock;
sockaddr_in Addr;
sockaddr_in IncomingAddress;
int AddressLen = sizeof(IncomingAddress);

int main()
{
    WSAStartup(MAKEWORD(2, 2), &Winsock);    // Start Winsock

    if(LOBYTE(Winsock.wVersion) != 2 || HIBYTE(Winsock.wVersion) != 2)    // Check version
    {
        WSACleanup();
        return 0;
    }

    Socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    ZeroMemory(&Addr, sizeof(Addr));
    Addr.sin_family = AF_INET;
    Addr.sin_port = htons(Port);  
    bind(Socket, (sockaddr*)&Addr, sizeof(Addr));

    if(listen(Socket, 1) == SOCKET_ERROR)
    {
        printf("listening error\n");
    }
    else
    {
        printf("listening ok\n");
    }

    if(Sub = accept(Socket, (sockaddr*)&IncomingAddress, &AddressLen))
    {
        char *ClientIP = inet_ntoa(IncomingAddress.sin_addr);
        int ClientPort = ntohs(IncomingAddress.sin_port);
        printf("Client conncted!\n");
        printf("IP: %s:%d\n", ClientIP, ClientPort);

        printf("Sending file .. \n");

        FILE *File;
        char *Buffer;
        unsigned long Size;

        File = fopen("C:\\Prog.rar", "rb");
        if(!File)
        {
            printf("Error while readaing the file\n");
            getchar();
            return 0;
        }

        fseek(File, 0, SEEK_END);
        Size = ftell(File);
        fseek(File, 0, SEEK_SET);

        Buffer = new char[Size];

        fread(Buffer, Size, 1, File);
        char cSize[MAX_PATH];
        sprintf(cSize, "%i", Size);

        fclose(File);
        send(Sub, cSize, MAX_PATH, 0); // File size

        //int len = Size;
        //char *data = Buffer;

        int Offset = 0;
        while(Size > Offset)
        {
            int Amount = send(Sub, Buffer + Offset, Size - Offset, 0);

            if(Amount <= 0)
            {
                cout << "Error: " << WSAGetLastError() << endl;
                break;
            }
            else
            {
                Offset += Amount;
                printf("2\n");
            }
        }


        free(Buffer);
        closesocket(Sub);
        closesocket(Socket);
        WSACleanup();
    }

    getchar();
    return 0;
}

客户：

#include <WinSock2.h>
#include <Windows.h>
#include <stdio.h>
#include <iostream>
using namespace std;

#pragma comment(lib, "Ws2_32.lib")

SOCKET Socket;
WSADATA Winsock;
sockaddr_in Addr;
int Addrlen = sizeof(Addr);

int main()
{
    WSAStartup(MAKEWORD(2, 2), &Winsock);    // Start Winsock

    if(LOBYTE(Winsock.wVersion) != 2 || HIBYTE(Winsock.wVersion) != 2)    // Check version
    {
        WSACleanup();
        return 0;
    }

     Socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    ZeroMemory(&Addr, sizeof(Addr));    // clear the struct
    Addr.sin_family = AF_INET;    // set the address family
    Addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    Addr.sin_port = htons(6000);    // set the port

    if(connect(Socket, (sockaddr*)&Addr, sizeof(Addr)) < 0)
    {
        printf("Connection failed !\n");
        getchar();
        return 0;
    }

    printf("Connection successful !\n");

    printf("Receiving file .. \n");



    int Size;
    char *Filesize = new char[1024];

    if(recv(Socket, Filesize, 1024, 0)) // File size
    {
        Size = atoi((const char*)Filesize);
        printf("File size: %d\n", Size);
    }

    char *Buffer = new char[Size];

    //int len = Size;
    //char *data = Buffer;

    int Offset = 0;
    while(Size > Offset)
    {
        int Amount = recv(Socket, Buffer + Offset, Size - Offset, 0);

        if(Amount <= 0)
        {
            cout << "Error: " << WSAGetLastError() << endl;
            break;
        }
        else
        {
            Offset += Amount;
            printf("2\n");
        }
    }

    FILE *File;
    File = fopen("Prog.rar", "wb");
    fwrite(Buffer, 1, Size, File);
    fclose(File);

    getchar();
    closesocket(Socket);
    WSACleanup();
    return 0;
}

Answer 1

send

API 可能无法发送您请求发送的所有数据。所以，你必须注意返回值，并从上次发送结束的地方重试发送。举个例子：

offset = 0;
while (offset < bufsize) {
    r = send(socket, buf+offset, bufsize-offset);
    if (r <= 0) break;
    offset += r;
}

当您对文件传输执行类似操作时，您无法确定您的文件大小是否如此。

发送文件大小时，只需发送代表大小的字符串，而不是整个

MAX_PATH

。然后，接收器需要解析第一个字符串以确定大小，但在第一个字符串末尾之后读入的任何数据都需要被视为文件的一部分。但是，由于您正在尝试发送

MAX_PATH

，因此接收者应该收到相同的金额。您的客户端代码接收 1024 字节，但没有迹象表明这与

MAX_PATH

的大小相同。

recv

API 也可能返回比请求更少的字节。您使用循环来处理读取文件，但您可能需要一个循环来读取包含文件大小的整个消息。

在客户端接收循环中，您正在递增

data

指针。这使得稍后写出该文件无法使用。不过你已经有了

Buffer

，所以用它来写出你的文件。

fwrite(Buffer, 1, len, File);

如果在执行套接字 I/O 时遇到错误，可以使用

WSAGetLastError()

检索错误，或者可以使用

getsockopt()

选项在套接字上发出

SO_ERROR

。这些可能返回不同的值，但错误原因应该是相关的。

Answer 2

我遇到了同样的问题，在谷歌搜索后我发现

send()

API可以基于依赖于操作系统的低级TCP缓冲区发送最大数量的数据。因此，为了发送一个大文件，我们需要执行文件分块，即：以块的形式发送文件。

const int FILE_CHUNK_SIZE = 2000; 

// get file size
ifstream file("myFile.file", ios::binary);
file.seekg(0, ios::end);
unsigned int fileSize = file.tellg();
file.close();

// get the file
char* fileBuffer = new char[fileSize];
file.open("myFile.file", ios::binary);
file.seekg (0, ios::beg);
file.read (fileBuffer, fileSize);
file.close();

// send file in chunks
unsigned int bytesSent = 0;
int bytesToSend = 0;

while (bytesSent < fileSize)
{
    if(fileSize - bytesSent >= FILE_CHUNK_SIZE)
        bytesToSend = FILE_CHUNK_SIZE;
    else
        bytesToSend = fileSize - bytesSent;
    send(ConnectSocket, fileBuffer + bytesSent, bytesToSend, 0 );
    bytesSent += bytesToSend;
}
delete [] fileBuffer;

在接收端，我们需要调用

recv()

API，直到读取整个文件内容。

感谢 shacktar cplusplus.com