C++中TCP socket传输文件

在两个文件中都定义文件头和用到的宏：

#define MAX_SIZE 10
#define ONE_PAGE 4096
struct FileHead
{
    char str[260];
    int size;
};

在客户端发送接收阶段：

    //1.发送文件头
    char path[260] = {0};
    cout<<"请输入文件路径"<>path;
    //截取文件名
    char* ptemp = path;
    while(*ptemp++ != '');
    while(*(--ptemp) != '\');
    ptemp++;

    fstream fs;
    fs.open(path,fstream::in|fstream::binary);
    fs.seekg(0,fstream::end);//以最后的位置为基准不偏移
    int nlen = fs.tellg();//取得文件大小
    fs.seekg(0,fstream::beg);

    FileHead fh;
    fh.size = nlen;
    memcpy(fh.str,ptemp,MAX_PATH);
    nlen = send(sockConnect ,(char*)&fh,sizeof(fh),0);
    //2,.如果接受到的内容为是
    char  szResult[MAX_SIZE] = {0};
    recv(sockConnect,szResult,sizeof(szResult),0);
    char szBuf[ONE_PAGE] ={0};
    if(0 == strcmp(szResult,"是"))
    {
        //读文件
        while(!fs.eof())
        {
            fs.read(szBuf,ONE_PAGE);
            int len = fs.gcount();
            //if(len == 0 ) break;
            send(sockConnect,szBuf,len,0);
        }
    }
    
    //3.关闭文件流
    fs.close();

在服务器端，接收和发送阶段

    char str[1024] = {0};
    int nlen;
    nlen = recv(sockWaiter,(char*)&fh,sizeof(fh),0);
    cout<<"是否要接受文件"<>str;
    send(sockWaiter ,str,sizeof(str),0);
    char szPath[MAX_SIZE] = {0};
    cout<<"请输入要存储的路径"<>szPath;
    char szPathName[MAX_SIZE] = {0};
    sprintf(szPathName,"%s%s",szPath,fh.str);//拼接路径和文件名
    cout< 0)
        {
            fs.write(content,len);
            FileSize -= len;
        }
    }
    fs.close();

/***************************************************

C/C++大文件/数据网络传输方法总结

在C/C++网络编程中难免会遇到须要传输大数据、大文件的状况，而因为socket自己缓冲区的限制，大概一次只能发送4K左右的数据，因此在传输大数据时客户端就须要进行分包，在目的地从新组包。而实际上已有一些消息/通信中间件对此进行了封装，提供了直接发送大数据/文件的接口；除此以外，利用共享目录，ftp，ssh等系统命令来实现大文件/数据也不失为一种好的方法。程序员

1.基础的基于socket进行传输

基础的基于socket进行传输关键在于控制，须要本身行分包和组包。apache

原理很简单那，咱们就直接看一下代码吧。
代码引用自： http://blog.csdn.net/ljd_1986413/article/details/7940938编程

  
    // file_server.c -- socket文件传输服务器端示例代码   
    // /  
    #include   
    #include   
    #include   
    #include   
    #include   
    #include   
      
    #define HELLO_WORLD_SERVER_PORT    6666  
    #define LENGTH_OF_LISTEN_QUEUE     20  
    #define BUFFER_SIZE                1024  
    #define FILE_NAME_MAX_SIZE         512  
      
    int main(int argc, char **argv)  
    {  
        // set socket's address information   
        // 设置一个socket地址结构server_addr,表明服务器internet的地址和端口  
        struct sockaddr_in   server_addr;  
        bzero(&server_addr, sizeof(server_addr));  
        server_addr.sin_family = AF_INET;  
        server_addr.sin_addr.s_addr = htons(INADDR_ANY);  
        server_addr.sin_port = htons(HELLO_WORLD_SERVER_PORT);  
      
        // create a stream socket   
        // 建立用于internet的流协议(TCP)socket，用server_socket表明服务器向客户端提供服务的接口  
        int server_socket = socket(PF_INET, SOCK_STREAM, 0);  
        if (server_socket < 0)  
        {  
            printf("Create Socket Failed!\n");  
            exit(1);  
        }  
      
        // 把socket和socket地址结构绑定   
        if (bind(server_socket, (struct sockaddr*)&server_addr, sizeof(server_addr)))  
        {  
            printf("Server Bind Port: %d Failed!\n", HELLO_WORLD_SERVER_PORT);  
            exit(1);  
        }  
      
        // server_socket用于监听   
        if (listen(server_socket, LENGTH_OF_LISTEN_QUEUE))  
        {  
            printf("Server Listen Failed!\n");  
            exit(1);  
        }  
      
        // 服务器端一直运行用以持续为客户端提供服务   
        while(1)  
        {  
            // 定义客户端的socket地址结构client_addr，当收到来自客户端的请求后，调用accept  
            // 接受此请求，同时将client端的地址和端口等信息写入client_addr中  
            struct sockaddr_in client_addr;  
            socklen_t          length = sizeof(client_addr);  
      
            // 接受一个从client端到达server端的链接请求,将客户端的信息保存在client_addr中  
            // 若是没有链接请求，则一直等待直到有链接请求为止，这是accept函数的特性，能够  
            // 用select()来实现超时检测   
            // accpet返回一个新的socket,这个socket用来与这次链接到server的client进行通讯  
            // 这里的new_server_socket表明了这个通讯通道  
            int new_server_socket = accept(server_socket, (struct sockaddr*)&client_addr, &length);  
            if (new_server_socket < 0)  
            {  
                printf("Server Accept Failed!\n");  
                break;  
            }  
      
            char buffer[BUFFER_SIZE];  
            bzero(buffer, sizeof(buffer));  
            length = recv(new_server_socket, buffer, BUFFER_SIZE, 0);  
            if (length < 0)  
            {  
                printf("Server Recieve Data Failed!\n");  
                break;  
            }  
      
            char file_name[FILE_NAME_MAX_SIZE + 1];  
            bzero(file_name, sizeof(file_name));  
            strncpy(file_name, buffer,  
                    strlen(buffer) > FILE_NAME_MAX_SIZE ? FILE_NAME_MAX_SIZE : strlen(buffer));  
      
            FILE *fp = fopen(file_name, "r");  
            if (fp == NULL)  
            {  
                printf("File:\t%s Not Found!\n", file_name);  
            }  
            else  
            {  
                bzero(buffer, BUFFER_SIZE);  
                int file_block_length = 0;  
                while( (file_block_length = fread(buffer, sizeof(char), BUFFER_SIZE, fp)) > 0)  
                {  
                    printf("file_block_length = %d\n", file_block_length);  
      
                    // 发送buffer中的字符串到new_server_socket,实际上就是发送给客户端  
                    if (send(new_server_socket, buffer, file_block_length, 0) < 0)  
                    {  
                        printf("Send File:\t%s Failed!\n", file_name);  
                        break;  
                    }  
      
                    bzero(buffer, sizeof(buffer));  
                }  
                fclose(fp);  
                printf("File:\t%s Transfer Finished!\n", file_name);  
            }  
      
            close(new_server_socket);  
        }  
      
        close(server_socket);  
      
        return 0;  
    }

//  
    // file_client.c  socket传输文件的client端示例程序   
    // ///  
    #include                         // for sockaddr_in  
    #include                          // for socket  
    #include                         // for socket  
    #include                              // for printf  
    #include                             // for exit  
    #include                             // for bzero  
      
    #define HELLO_WORLD_SERVER_PORT       6666  
    #define BUFFER_SIZE                   1024  
    #define FILE_NAME_MAX_SIZE            512  
      
    int main(int argc, char **argv)  
    {  
        if (argc != 2)  
        {  
            printf("Usage: ./%s ServerIPAddress\n", argv[0]);  
            exit(1);  
        }  
      
        // 设置一个socket地址结构client_addr, 表明客户机的internet地址和端口  
        struct sockaddr_in client_addr;  
        bzero(&client_addr, sizeof(client_addr));  
        client_addr.sin_family = AF_INET; // internet协议族  
        client_addr.sin_addr.s_addr = htons(INADDR_ANY); // INADDR_ANY表示自动获取本机地址  
        client_addr.sin_port = htons(0); // auto allocated, 让系统自动分配一个空闲端口  
      
        // 建立用于internet的流协议(TCP)类型socket，用client_socket表明客户端socket  
        int client_socket = socket(AF_INET, SOCK_STREAM, 0);  
        if (client_socket < 0)  
        {  
            printf("Create Socket Failed!\n");  
            exit(1);  
        }  
      
        // 把客户端的socket和客户端的socket地址结构绑定   
        if (bind(client_socket, (struct sockaddr*)&client_addr, sizeof(client_addr)))  
        {  
            printf("Client Bind Port Failed!\n");  
            exit(1);  
        }  
      
        // 设置一个socket地址结构server_addr,表明服务器的internet地址和端口  
        struct sockaddr_in  server_addr;  
        bzero(&server_addr, sizeof(server_addr));  
        server_addr.sin_family = AF_INET;  
      
        // 服务器的IP地址来自程序的参数   
        if (inet_aton(argv[1], &server_addr.sin_addr) == 0)  
        {  
            printf("Server IP Address Error!\n");  
            exit(1);  
        }  
      
        server_addr.sin_port = htons(HELLO_WORLD_SERVER_PORT);  
        socklen_t server_addr_length = sizeof(server_addr);  
      
        // 向服务器发起链接请求，链接成功后client_socket表明客户端和服务器端的一个socket链接  
        if (connect(client_socket, (struct sockaddr*)&server_addr, server_addr_length) < 0)  
        {  
            printf("Can Not Connect To %s!\n", argv[1]);  
            exit(1);  
        }  
      
        char file_name[FILE_NAME_MAX_SIZE + 1];  
        bzero(file_name, sizeof(file_name));  
        printf("Please Input File Name On Server.\t");  
        scanf("%s", file_name);  
      
        char buffer[BUFFER_SIZE];  
        bzero(buffer, sizeof(buffer));  
        strncpy(buffer, file_name, strlen(file_name) > BUFFER_SIZE ? BUFFER_SIZE : strlen(file_name));  
        // 向服务器发送buffer中的数据，此时buffer中存放的是客户端须要接收的文件的名字  
        send(client_socket, buffer, BUFFER_SIZE, 0);  
      
        FILE *fp = fopen(file_name, "w");  
        if (fp == NULL)  
        {  
            printf("File:\t%s Can Not Open To Write!\n", file_name);  
            exit(1);  
        }  
      
        // 从服务器端接收数据到buffer中   
        bzero(buffer, sizeof(buffer));  
        int length = 0;  
        while(length = recv(client_socket, buffer, BUFFER_SIZE, 0))  
        {  
            if (length < 0)  
            {  
                printf("Recieve Data From Server %s Failed!\n", argv[1]);  
                break;  
            }  
      
            int write_length = fwrite(buffer, sizeof(char), length, fp);  
            if (write_length < length)  
            {  
                printf("File:\t%s Write Failed!\n", file_name);  
                break;  
            }  
            bzero(buffer, BUFFER_SIZE);  
        }  
      
        printf("Recieve File:\t %s From Server[%s] Finished!\n", file_name, argv[1]);  
      
        // 传输完毕，关闭socket   
        fclose(fp);  
        close(client_socket);  
        return 0;  
      
    }

2.使用现有的通信中间件

2.1 ActiveMQ 传送文件接口

为了解决传输大文件的问题，ActiveMQ在jms规范以外引入了jms streams的概念。PTP模式下，连到同一个destination的两端，能够经过broker中转来传输大文件。服务器

发送端使用connection.createOutputStream打开一个输出流，往流里写文件。网络

OutputStream out =connection.createOutputStream(destination);ssh

接收端则简单的使用connection.createInputStream拿到一个输入流，从中读取文件数据便可。socket

InputStream in = connection.createInputStream(destination)

详见：http://activemq.apache.org/jms-streams.htmltcp

2.2 ZeroMQ 接口

ZeroMQ没有直接提供传送文件的接口。但ZeroMQ中send（void * data, size_t len）接口已经作好了封装，能够send任意大小的数据。代码以下：函数

zmq::context_t ctx(1);
    zmq::socket_t sock(ctx, ZMQ_REQ);

	sock.connect("tcp://192.168.20.111:20310");

	sock.send(pData,len);//数据大小没有限制，能够直接发送任意大小的数据

	
	char reply[100];
	sock.recv (reply,100);

	sock.disconnect(addr);

接收端代码以下：

m_context = new zmq::context_t(1);
    m_socket = new zmq::socket_t (*m_context, ZMQ_REP);
	m_socket->bind ("tcp://*:20310");

	zmq::message_t request;

        // Wait for next request from client
        m_socket->recv (&request) // request能够接受发送来的任意大小的数据
        
        m_socket->send("ok",2);

 

是否是很简单呢？
 

3.基于共享文件、ftp、scp等

这就不细说了。要么就是写共享目录，要么就是调用系统命令。

4.总结

1)直接基于socket编程难度较高，因此不推荐。 2）使用现有的库方便，但须要学习。通常推荐。 3）共享文件、ftp、scp等难度低，简单易用，在符合使用场景时是首选。