C++中TCP socket传输文件

87 篇文章 17 订阅
44 篇文章 1 订阅

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

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

在客户端发送接收阶段:

    //1.发送文件头
    char path[260] = {0};
    cout<<"请输入文件路径"<<endl;
    cin>>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<<"是否要接受文件"<<endl;
    cin>>str;
    send(sockWaiter ,str,sizeof(str),0);
    char szPath[MAX_SIZE] = {0};
    cout<<"请输入要存储的路径"<<endl;
    cin>>szPath;
    char szPathName[MAX_SIZE] = {0};
    sprintf(szPathName,"%s%s",szPath,fh.str);//拼接路径和文件名
    cout<<szPathName<<endl;
    fstream fs;
    fs.open(szPathName,fstream::out|fstream::binary|fstream::trunc);//以空文件的形式打开
    int FileSize = fh.size;
    int len;
    char content[ONE_PAGE] = {0};
    while(FileSize)
    {
        len = recv(sockWaiter,content,ONE_PAGE,0);
        if(len > 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<netinet/in.h>   
    #include<sys/types.h>   
    #include<sys/socket.h>   
    #include<stdio.h>   
    #include<stdlib.h>   
    #include<string.h>   
      
    #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<netinet/in.h>                         // for sockaddr_in  
    #include<sys/types.h>                          // for socket  
    #include<sys/socket.h>                         // for socket  
    #include<stdio.h>                              // for printf  
    #include<stdlib.h>                             // for exit  
    #include<string.h>                             // 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等难度低,简单易用,在符合使用场景时是首选。

 

  • 1
    点赞
  • 30
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,下面是一个简单的封装TcpSocket类的示例代码: ```c #include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <errno.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #define MAX_BUFFER_SIZE 1024 class TcpSocket { private: int sockfd; struct sockaddr_in server_addr; public: TcpSocket() { sockfd = socket(AF_INET, SOCK_STREAM, 0); if (sockfd == -1) { perror("socket"); exit(EXIT_FAILURE); } memset(&server_addr, 0, sizeof(struct sockaddr_in)); server_addr.sin_family = AF_INET; } ~TcpSocket() { close(sockfd); } bool connect(const char* server_ip, uint16_t server_port) { server_addr.sin_addr.s_addr = inet_addr(server_ip); server_addr.sin_port = htons(server_port); int ret = ::connect(sockfd, (struct sockaddr*)&server_addr, sizeof(struct sockaddr_in)); if (ret == -1) { perror("connect"); return false; } return true; } bool send(const char* data, size_t len) { size_t total_sent = 0; while (total_sent < len) { ssize_t sent = ::send(sockfd, data + total_sent, len - total_sent, 0); if (sent == -1) { perror("send"); return false; } total_sent += sent; } return true; } bool recv(char* buffer, size_t len, size_t& recv_len) { memset(buffer, 0, len); ssize_t recv = ::recv(sockfd, buffer, len, 0); if (recv == -1) { perror("recv"); return false; } recv_len = recv; return true; } }; ``` 这个类封装了一个基本的TcpSocket连接,包括创建socket、连接服务器、发送数据和接收数据等操作。其,connect函数用于连接服务器,send函数用于发送数据,recv函数用于接收数据。注意,在send和recv函数,由于网络传输可能会出现分包的情况,因此需要循环发送和接收数据,直到数据完整传输或接收完毕为止。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值