linux下文件传送程序代码两例（分别基于tcp和udp）

基于TCP的文件传输程序：

 

 服务器端代码：

#include     // for sockaddr_in
#include     // for socket
#include     // for socket
#include         // for printf
#include         // for exit
#include         // for bzero
/*
#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)
{
    //设置一个socket地址结构server_addr,代表服务器internet地址, 端口
    struct sockaddr_in server_addr;
    bzero(&server_addr,sizeof(server_addr)); //把一段内存区的内容全部设置为0
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htons(INADDR_ANY);
    server_addr.sin_port = htons(HELLO_WORLD_SERVER_PORT);

    //创建用于internet的流协议(TCP)socket,用server_socket代表服务器socket
    int server_socket = socket(PF_INET,SOCK_STREAM,0);
    if( server_socket < 0)
    {
        printf("Create Socket Failed!");
        exit(1);
    }
   
    //把socket和socket地址结构联系起来
    if( bind(server_socket,(struct sockaddr*)&server_addr,sizeof(server_addr)))
    {
        printf("Server Bind Port : %d Failed!", HELLO_WORLD_SERVER_PORT);
        exit(1);
    }
   
    //server_socket用于监听
    if ( listen(server_socket, LENGTH_OF_LISTEN_QUEUE) )
    {
        printf("Server Listen Failed!");
        exit(1);
    }
    while (1) //服务器端要一直运行
    {
        //定义客户端的socket地址结构client_addr
        struct sockaddr_in client_addr;
        socklen_t length = sizeof(client_addr);

        //接受一个到server_socket代表的socket的一个连接
        //如果没有连接请求,就等待到有连接请求--这是accept函数的特性
        //accept函数返回一个新的socket,这个socket(new_server_socket)用于同连接到的客户的通信
        //new_server_socket代表了服务器和客户端之间的一个通信通道
        //accept函数把连接到的客户端信息填写到客户端的socket地址结构client_addr中
        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, BUFFER_SIZE);
        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, FILE_NAME_MAX_SIZE+1);
        strncpy(file_name, buffer, strlen(buffer)>FILE_NAME_MAX_SIZE?FILE_NAME_MAX_SIZE:strlen(buffer));
//        int fp = open(file_name, O_RDONLY);
//        if( fp < 0 )
        FILE * fp = fopen(file_name,"a");
        if(NULL == fp )
        {
            printf("File:/t%s Can Not creat file!/n", file_name);
        }
        else
        {
            bzero(buffer, BUFFER_SIZE);
            int file_block_length = 0;
//            while( (file_block_length = read(fp,buffer,BUFFER_SIZE))>0)
  int length = 0;
      while( length = recv(new_server_socket,buffer,BUFFER_SIZE,0))
      {
          if(length < 0)
          {
               printf("Recieve Data From Server %s Failed!/n", argv[1]);
               break;
          }
//          int write_length = write(fp, buffer,length);
          int write_length = fwrite(buffer,sizeof(char),length,fp);
          if (write_length          {
               printf("File:/t%s Write Failed/n", file_name);
               break;
          }
          bzero(buffer,BUFFER_SIZE);   
      }
           
//            close(fp);
            fclose(fp);
      printf("Recieve File:/t %s From Server Finished/n",file_name);
        }
        //关闭与客户端的连接
        close(new_server_socket);
    }
    //关闭监听用的socket
    close(server_socket);
    return 0;
}

客户端代码：

#include     // for sockaddr_in
#include     // for socket
#include     // for socket
#include         // for printf
#include         // for exit
#include         // for bzero
/*
#include
#include
#include
#include
*/

#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)); //把一段内存区的内容全部设置为0
    client_addr.sin_family = AF_INET;    //internet协议族
    client_addr.sin_addr.s_addr = htons(INADDR_ANY);//INADDR_ANY表示自动获取本机地址
    client_addr.sin_port = htons(0);    //0表示让系统自动分配一个空闲端口
    //创建用于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;
    if(inet_aton(argv[1],&server_addr.sin_addr) == 0) //服务器的IP地址来自程序的参数
    {
        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, FILE_NAME_MAX_SIZE+1);
    printf("Please Input File Name On Server:/t");
    scanf("%s", file_name);
   
    char buffer[BUFFER_SIZE];
    bzero(buffer,BUFFER_SIZE);
    strncpy(buffer, file_name, strlen(file_name)>BUFFER_SIZE?BUFFER_SIZE:strlen(file_name));
    //向服务器发送buffer中的数据
    send(client_socket,buffer,BUFFER_SIZE,0);

//    int fp = open(file_name, O_WRONLY|O_CREAT);
//    if( fp < 0 )
    FILE * fp = fopen(file_name,"r");
    if(NULL == fp )
    {
        printf("nFile:/t%s Not Found/n", file_name);
        exit(1);
    }
   
    //从服务器接收数据到buffer中
    bzero(buffer,BUFFER_SIZE);
    int file_block_length;
    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(client_socket,buffer,file_block_length,0)<0)
                {
                    printf("Send File:/t%s Failed/n", file_name);
                    break;
                }
                bzero(buffer, BUFFER_SIZE);
            }

 

    printf("File:/t%s Transfer Finished/n",file_name);   
    close(fp);
    //关闭socket
    close(client_socket);
    return 0;
}

 

 

 

测试过程：

 

服务器端：   [root@localhost c++]# ./file_serve
                  Recieve File:    cxt From Server Finished

 

客户端：      [root@localhost file_transmit]# ./file_clien 192.168.0.130
                  Please Input File Name On Server:       cxt
                  file_block_length = 49
                  File:   cxt Transfer Finished

运行结果：客户端将cxt文件传送到服务器端。

 

基于UDP的文件传输程序：

server.c：

#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;

#define SERV_PORT 2500
#define MAX_SIZE 1024*40

void recvUDP(char name[20],int sockfd)
{
        int ret,fd;
        mode_t fdmode = (S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH);
        char mesg[MAX_SIZE];
        fd_set rdset;
        struct timeval tv;
        int rlen,wlen;
        int i;

        fd = open(name,O_RDWR|O_CREAT|O_APPEND,fdmode);
        if(fd == -1)
        {
                printf("open file %s error:%n",name,strerror(errno));
                exit(-1);
        }

        for(i=0;;i++)
        {
                tv.tv_sec = 1;
                tv.tv_usec = 0;

                FD_ZERO(&rdset);
                FD_SET(sockfd,&rdset);
               
                ret = select(sockfd+1,&rdset,NULL,NULL,&tv);
                if(ret == -1)
                {
                        printf("select error %s/n",strerror(errno));
                        exit(-1);
                }
                else if(ret==0)
                {
                        printf("select timeout,continue circle/n");
                        continue;
                }
               
                if(FD_ISSET(sockfd,&rdset))
                {
                        memset(mesg,0,MAX_SIZE);
                        rlen = read(sockfd,mesg,MAX_SIZE);
                        if(rlen <=0 )
                        {
                                printf("read error %s/n",strerror(errno));
                                exit(-1);
                        }
                       
                        if(!strcmp(mesg,"end"))
                        {
                                printf("recv end./n");
                                break;
                        }

                        wlen = write(fd,mesg,rlen);
                        if(wlen != rlen )
                        {
                                printf("write error %s/n",strerror(errno));
                                exit(-1);
                        }       
                }
               
                printf("The %d times write/n",i);
        }

        close(fd);
}

int main(int argc, char *argv[])
{
        int sockfd;
        int r;
        struct sockaddr_in servaddr;

        sockfd = socket(AF_INET,SOCK_DGRAM,0); /*create a socket*/

        /*init servaddr*/
        bzero(&servaddr,sizeof(servaddr));
        servaddr.sin_family = AF_INET;
        servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
        servaddr.sin_port = htons(SERV_PORT);

        /*bind address and port to socket*/
        if(bind(sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr)) == -1)
        {
                perror("bind error");
                exit(-1);
        }

        r = fcntl(sockfd, F_GETFL, 0);
        fcntl(sockfd, F_SETFL, r & ~O_NONBLOCK);
       
        recvUDP(argv[1],sockfd);

        return 0;

}

client.c:

#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;
#include;

#define MAX_SIZE 1024*40
#define SERV_PORT 2500

void connectUDP(char name[20],int sockfd,struct sockaddr *pservaddr,socklen_t servlen)
{
                char buf[MAX_SIZE];
                fd_set wrset;
                struct timeval tv;
                int rlen,wlen;
                int fd;
                int ret;
                int i;

                if(connect(sockfd,(struct sockaddr *)pservaddr,servlen) == -1)
                {
                        perror("connet error");
                        exit(1);
                }
                else
                        printf("connect server ok!/n");

                fd = open(name,O_RDONLY);
                if(fd==-1)
                {
                        printf("fopen error %s/n",strerror(errno));
                        exit(-1);
                }
               
                i=0;
               
                while(1)
                {       
                        tv.tv_sec = 1;
                        tv.tv_usec = 0;
                       
                        FD_ZERO(&wrset);
                        FD_SET(sockfd,&wrset);
                       
                        ret = select(sockfd+1,NULL,&wrset,NULL,&tv);
                        if(ret == -1)
                        {
                                printf("select error %s/n",strerror(errno));
                                exit(-1);
                        }
                        else if(ret==0)
                        {
                                printf("select timeout,continue circle/n");
                                continue;
                        }
                       
                        memset(buf,0,MAX_SIZE);
                        if(FD_ISSET(sockfd,&wrset))
                        {                                       
                                rlen = read(fd,buf,MAX_SIZE);
                                if(rlen <0)
                                {
                                        printf("fread data error %s/n",strerror(errno));
                                        exit(-1);
                                }
                                else if(rlen==0)
                                {
                                        wlen = write(sockfd,"end",3);
                                        if(wlen !=3)
                                        {
                                                printf("write end error/n",strerror(errno));
                                                exit(-1);
                                        }

                                        printf("all complete/n");

                                        close(fd);
                                        close(sockfd);
                                        exit(0);
                                }

                                wlen = write(sockfd,buf,rlen);
                                if(wlen != rlen)
                                {
                                        printf("write data to sockfd error:%s/n",strerror(errno));
                                        exit(-1);
                                }
                               
                                i++;
                               
                                usleep(500);
                                printf("The %d times read/n",i);
                        }
                }
        }

int main(int argc ,char *argv[])
{       
        char *fh;
        struct sysinfo s_info;
        float time1,time2;
        int error1,error2;
        int sockfd;
        struct stat fsize;
        struct sockaddr_in servaddr;
        error1= sysinfo(&s_info);
        time1 = s_info.uptime;
        int r;

        if(argc != 3)
        {
                printf("useage:udpclient;/n");
                exit(1);
        }
        bzero(&servaddr,sizeof(servaddr));
        servaddr.sin_family= AF_INET;
        servaddr.sin_port = htons(SERV_PORT);
       
        if(inet_pton(AF_INET,argv[1],&servaddr.sin_addr) <= 0)
        {
                printf("[%s]is not a valid IPaddress/n",argv[1]);
                exit(1);
        }
        fh = argv[2];

        sockfd =socket(AF_INET,SOCK_DGRAM,0);
       
        r = fcntl(sockfd, F_GETFL, 0);
        fcntl(sockfd, F_SETFL, r & ~O_NONBLOCK);
       
        connectUDP(argv[2],sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr));       
        fprintf(stderr,"ServerIP:/t%s/n",argv[1]);       
        if(stat(argv[2],&fsize) == -1)
                perror("failed to get fiel statusi/n");
        else       
                fprintf(stderr,"file name:/t%s/nfile size:/t%dK/n",argv[2],fsize.st_size/1024);
        error2=sysinfo(&s_info);
        time2 = s_info.uptime;
        printf("tranfice file time =%fs/n",(time2-time1));
}