网络间通信socket传输任意格式任意大小的文件

 

              最近看到很多网友写的socket程序，试验了几个，发送"Hello,world"时都没有问题，但在传输较大文件，比如一个200M的文件时，都不能成功，即使成功了，server端也存在保存的问题， 有的是乱码，也有文件大小不对的情况，这两天工作不忙，我也草草写了一些代码，实现了可以传输任意格式任意大小的文件，几位同事帮我一起做了测试，传送几百M的文件速度还是很快的，而且也没有丢数据的问题。  代码原理很简单，所以也没有做注释，大家可以把代码copy下来，自己把玩一下。现在我跟同事之间传送datasheet，都在使用我的程序， 其实我也是个入行不深的人，代码并没有经过优化，很多printf() 是多余的，也有的地方写的很麻烦。希望大家给以批评，给以帮助。

              由于 网络接口MTU的限制(一般mtu为1500),大些的文件只能分多次发送,这样就有几个问题：分几次发送？一次发送多大？ 保存端的怎么保存？

我的办法是：通过定义一个shouldoplen，来说明一次操作需要操作的长度，如果要发送的文件较小（跟buf相比），shouldoplen 就是读取的文件大小，如果文件较大，需要多次发送，那么shouldoplen 就是buf  的长度，通过多次读取，发送， 直到发送出去的总长度oplencount  等于文件的大小，这时一个文件就算完整发送成功了。

编译时：  

    gcc  socket_server.c -lpthread  -o server

    gcc  socket_client.c -o client

使用时：

在一端打开server

       #./server

令一端使用client  

     #./client ./han/docunt/Linux_dd.pdf  Linux_dd.pdf

这样就把./han/docunt/Linux_dd.pdf 这就文件  发送server端，保存名为Linux_dd.pdf        

当然，前提是在sock_client.c  中的把目的IP改为你要连接的目的IP。

***********************这是data.h****************

#ifndef DATA_H
#define DATA_H

#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<string.h>
#include<fcntl.h>
#include<signal.h>
#include<pthread.h>
#include<sys/socket.h>
#include<errno.h>
#include<arpa/inet.h>
#include<netinet/in.h>


#include "data.h"


typedef  unsigned int uint;

struct data{


    char filename[30];              //要发送的文件名（可含路径）

    char filesavename[30];    //要保存的文件名（可含路径）
    
    uint filelen;                        //文件从字节数

    uint shouldoplen;             //一次要操作的字节数
    
    uint oplencount;                // 操作的字节总数

    char filebuf[1300];                //由于mtu为1500，在不改动mtu的情况下，1300没有问题。


    
};


struct data_return {
        
    int writelen;                           //server端保存文件时，一次写的长度
    
    int writelencount;                //写的总数
    
    int recvlen;                             //一次接受到的长度
};





#endif

***********************这是sock_sever 端********************************

#include "data.h"



int sockid;

pthread_t tid;


int fileopen;


__sighandler_t f_quit(void)
{

    printf("\nserver closed right now!!\n");
    close(sockid);
    close(fileopen);
    exit(0);


}


struct thread_para{

    int sockid;
    
    struct sockaddr_in addr;

    struct sockaddr_in fromaddr;

    struct data datarecv;
};


int main(int argc,char* argv[])
{
    

    signal(SIGINT,(__sighandler_t)f_quit);


    struct thread_para thread_do;

    memset(&thread_do,0,sizeof(thread_do));

    

    sockid = socket(AF_INET,SOCK_STREAM,0);
    
    if(sockid == -1)
        printf("socker create error!"),exit(-1);

    printf("sockid = %d\n",sockid);

    struct sockaddr_in addr;    
    
    addr.sin_family = AF_INET;

    addr.sin_port = htons(8888);

    addr.sin_addr.s_addr = INADDR_ANY;

    int r = bind(sockid,(struct sockaddr*)&addr,sizeof(addr));

    if(r<0)
        printf("bind error!,r=%s\n",strerror(errno)),exit(-1);
    
    printf("bind success!\n");



    if((r = listen(sockid,10))<0)
        printf("listen error!,r=%s\n",strerror(errno)),exit(-1);
    
    printf("listen successfully!\n");


    int fd,writelen,writelencount,recvlen,opnumber;
    
    struct sockaddr_in fromaddr;


    socklen_t len = sizeof(fromaddr);


    void* handler(void*);


    while(1){
        
        
        if( (fd = accept(sockid,(struct sockaddr*)&fromaddr,&len)) < 0 ){

            printf("accept error,errno = %s\n",strerror(errno)),exit(-1);
        }

    
        printf("accept successfuluy!,new sockid=%d\n",fd);
    
        memset(&thread_do,0,sizeof(thread_do));
    
        thread_do.sockid = fd;

        thread_do.fromaddr = fromaddr;
    
        pthread_create(&tid,NULL,handler,&thread_do);


    }

}

void* handler(void* p)
{

    struct thread_para * thread_do = p;


    int writelen,writelencount,recvlen,opnumber;

    writelen = 0;

    writelencount = 0;    
    
    recvlen = 0;

    opnumber = 1;

    printf(" there are the data from %s :\n",inet_ntoa(thread_do->fromaddr.sin_addr) );
    

    do{


        memset(&thread_do->datarecv,0,sizeof(thread_do->datarecv));
    
        recvlen = recv( thread_do->sockid,&thread_do->datarecv,sizeof(thread_do->datarecv)-1,0);        
    
        if(recvlen<0)
            
            printf("recv error!,errro=%s\n",strerror(errno)),exit(-1);

        else if(recvlen == 0 )
        
            printf(" client has closed!\n");
        
        else
    
            printf("the filename from client:\n\t%s\n",thread_do->datarecv.filename);
        
            printf("the file save name:\n\t%s\n",thread_do->datarecv.filesavename);
    
            printf("the filelen =%d,the recvlen=%d,the filebuf = %d\n",thread_do->datarecv.filelen,recvlen,sizeof(thread_do->datarecv.filebuf)-1 );

            printf("the shouldoplen = %d\n",thread_do->datarecv.shouldoplen);

            printf("the oplencount = %d\n",thread_do->datarecv.oplencount);
    

            umask(0);

            int fileopen = open((const char*)thread_do->datarecv.filesavename,O_RDWR | O_CREAT | O_APPEND, S_IRWXU | S_IRWXG | S_IROTH );    
    
            if(fileopen<0){

                printf("fileopen error!,%s\n",strerror(errno)),exit(-1);
        
            }

            printf("file %s create and open successfully!\n",thread_do->datarecv.filesavename);



            if( (  writelen =  write(fileopen,thread_do->datarecv.filebuf,thread_do->datarecv.shouldoplen)  ) <0 ){

                printf("write error!\n"),exit(-1);
            }
            
            writelencount  += writelen;

            printf(" opnumber = %d,writelen=%d,writelencount=%d\n \n",opnumber++,writelen,writelencount);

            struct data_return data_r;
    
            data_r.writelen = writelen;
    
            data_r.writelencount = writelencount;    
        
            data_r.recvlen = recvlen;

            write(thread_do->sockid, &data_r ,sizeof(data_r));
        
            close(fileopen);
    
            }while( writelencount != thread_do->datarecv.filelen );

        

}    

***********************************这是sock_client 端*******************************

#include "data.h"


int sockid;



__sighandler_t f_quit(void)
{

        printf("\n client closed right now!!\n");
        close(sockid);
 
        exit(0);

 
}


int main(int argc,char* argv[])
{

    signal(SIGINT,(__sighandler_t)f_quit);

    sockid = socket(AF_INET,SOCK_STREAM,0);
    
    if(sockid==-1){
    
        printf("socket create error!"),exit(-1);
    
    }
    
    struct sockaddr_in addr;

    memset(&addr,0,sizeof(addr));

    addr.sin_family = AF_INET;

    addr.sin_port = htons(8888);

    inet_aton("192.168.10.52",&addr.sin_addr);

    
    int r = connect(sockid,(struct sockaddr*)&addr,sizeof(addr));

    if(r<0)printf("connect error!\n"),exit(-1);

    printf("connect success!  \n");

    struct data datasent;
    
    memset(&datasent,0,sizeof(datasent));
    
    struct  data_return data_r;

    memset(&data_r,0,sizeof(data_r));

    strcpy(datasent.filename, argv[1]);
    
    strcpy(datasent.filesavename,argv[2]);

    int fd  = open(datasent.filename,O_RDWR);    

    if(fd<0){
    
        printf("open %s failed!\n",datasent.filename),exit(-1);
    }
    
    datasent.filelen = lseek(fd,0,SEEK_END);
    
    lseek(fd,0,SEEK_SET);

    printf("open file %s success!,the fd = %d,file length  = %d\n",datasent.filename,fd,datasent.filelen);

    datasent.oplencount = 0;

    int opnumber = 1;

    do{
    
        lseek(fd,datasent.oplencount,SEEK_SET);


        if( (datasent.filelen - datasent.oplencount ) >= ( sizeof(datasent.filebuf) - 1 ) ){                    //  比较： （ filelen - oplencount   ） 与  filebuf 的大小        
    
            datasent.shouldoplen = sizeof(datasent.filebuf) - 1 ;                                                             //通过比较，确定出shouoplen的值
        }

        else{
            datasent.shouldoplen = datasent.filelen -datasent.oplencount;    
        }

        memset(&datasent.filebuf,0,sizeof(datasent.filebuf));

        readdata(fd,&datasent,datasent.shouldoplen);

        datasent.oplencount += datasent.shouldoplen;

        sentdata(datasent);

        printf(" opnumber = %d,shouldoplen = %d ,datasent.oplencount = %d \n",opnumber++,datasent.shouldoplen,datasent.oplencount);

        read(sockid,&data_r,sizeof(data_r));
    
        printf("server info:recvlen = %d, writelen = %d, writelencount = %d \n\n",data_r.recvlen,data_r.writelen,data_r.writelencount);


        } while( datasent.filelen > datasent.oplencount );


        close(sockid);

        close(fd);                
                                
}

int readdata(int fd,struct data* datasent,int len)
{
    int readlen;

    if((readlen = read( fd,datasent->filebuf,len ) )<0){                                                            //通过len来确定一次要读多少
        
        printf("%m\n"),exit(-1);
    }

    printf("read %s successfully ! readlen=%d\n",datasent->filename,readlen);

}


int sentdata(struct data datasent)
{

    int sendlen;    

    if((sendlen = send( sockid,&datasent,sizeof(datasent)-1,0 ))<0){
    
        printf("write error ,writelen=%d, sizeof(datasent): %d, error: %s\n",sendlen, sizeof(datasent), strerror(errno));
    
        exit(-1);
    }

    printf("file %s send successfully!,sendlen=%d\n",datasent.filename,sendlen);


}