一个传送文件的小程序（linux）

client.c 用于发送文件，可以发送n个文件

/*
 * client.c
 *
 *  Created on: Apr 28, 2012
 *      Author: yyp
 */
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <dirent.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <time.h>

typedef struct com_meg{
    int  operation;
    time_t mod_time;
    char file_route[1024];
    int  state;
    int  type;
    size_t file_size;
} msg_exch;


int sendfile(char *path, int sockfd, size_t filesize);

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


    int sockfd;
    struct sockaddr_in server_addr;
    int socklen;
    int i;
    struct stat file_stat;
    msg_exch msg;

    if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
          perror("socket foundation");
           exit(EXIT_FAILURE);
      }
    /*
     *命名套接字，
     */
      server_addr.sin_family = AF_INET;
      server_addr.sin_port = htons(8080);
      server_addr.sin_addr.s_addr = inet_addr("10.2.1.163");
      socklen = sizeof(struct sockaddr_in);
      /*
       * 把套接字连接到服务器
           */
      if(connect(sockfd, (struct sockaddr *)&server_addr, socklen) == -1){
          perror("connect failure");
          exit(EXIT_FAILURE);
       }
       else{
        printf("connect sucess\n");
       }

       for(i = 1; i < argc; i++){

            memset(&msg, 0, sizeof(msg));
            /*initialize the exc_msg*/
            int res;
            res = lstat(argv[i], &file_stat);
            if(res == -1){
                perror("file stat");
                exit(EXIT_FAILURE);
            }
            strcpy(msg.file_route, argv[i]);
            msg.type = -1;
            msg.file_size = file_stat.st_size;
            printf("the file route is %s\n",msg.file_route);

            /*send exc_msg*/
            if(send(sockfd, &msg, sizeof(msg),0) < 0 ){
                perror("send pack");
                exit(EXIT_FAILURE);
            }

            printf("send exch sucess\n");
            if(recv(sockfd, &msg, sizeof(msg), 0) < 0){
                perror("recv pack");
                exit(EXIT_FAILURE);
            }

            printf("after send the type is %d\n", msg.type);

            printf("recv exch  sucess\n");
            if(msg.state != 0){
                fprintf(stderr,"the server have problem");
                exit(1);
            }

            if(sendfile(argv[i], sockfd, msg.file_size) != 0){
                fprintf(stderr,"%s",argv[i]);
                exit(EXIT_FAILURE);
            }
            else{
                printf("file %s done\n", argv[i]);
            }

            printf("\n one  done");
            printf("\n\n");

        }


        return 0;
}

int sendfile(char *path, int sockfd, size_t filesize){
    int nread;
    char buf[2048];
    int fd;
    int res;

    fd = open(path, O_RDONLY);
    if(fd == -1){
        perror("open file");
        exit(EXIT_FAILURE);
    }

    /*if filesize is too small,then send and return*/

    /*file is larger than 2048 then*/
    while((nread = read(fd, buf, 2048)) > 0){

        res = send(sockfd, buf, nread, 0);
        if(res == -1){
            perror("sendfile");
            exit(EXIT_FAILURE);
        }
    }

    printf("file: %s is done.\n", path);

    return 0;
}

server.c 用于接收文件

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <fcntl.h>

typedef struct com_meg{
    int operation;
    time_t mod_time;
    char file_route[1024];
    int state;
    int type;
    size_t file_size;
} msg_exch;


int getfile(int clientfd, msg_exch msg);

int main(int argc, char *argv[]){
        int sockfd, clientfd;
        int socklen, clientlen;
        int nrecv;
        int res;

        msg_exch msg;
        struct sockaddr_in server_addr, clietn_addr;

        sockfd = socket(AF_INET, SOCK_STREAM, 0);
        if(sockfd == -1){
            perror("socket");
            exit(EXIT_FAILURE);
        }

        server_addr.sin_family = AF_INET;
        server_addr.sin_port = htons(8080);
        server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
        socklen = sizeof(server_addr);

        res = bind(sockfd, (struct sockaddr *)&server_addr, socklen);
        if(res == -1){
            perror("bind");
            exit(EXIT_FAILURE);
        }

        listen(sockfd, 1);
        clientfd = accept(sockfd, (struct sockaddr *)&clietn_addr,(socklen_t *)&clientlen);
        printf("accept one: %d\n", clientfd);
        if(clientfd == -1){
            perror("bind");
            exit(EXIT_FAILURE);
        }

        while(1){
            /*initialize the exch again to against the wrong operation*/
            if(memset(&msg, 0, sizeof(msg)) == NULL){
                printf("memset error\n");
            }

            int size = 0;
            while(size < sizeof(msg)){
                nrecv = recv(clientfd, (char *)&msg + size, sizeof(msg)-size, 0);
                size = nrecv + size;
                if(nrecv <= 0){
                    fprintf(stderr,"cannot recv exchinfo or all files are done \n");
                    exit(EXIT_FAILURE);
                }
            }

            printf("file: %s\n", msg.file_route);
            if(msg.type == -1){
                msg.state = 0;
                send(clientfd, &msg, sizeof(msg), 0);

            }
            else{
                printf("all files send over\n");
                send(clientfd, &msg, sizeof(msg), 0);
                break;
            }

            if(getfile(clientfd, msg) == 0){
                printf("getfile %s\n",msg.file_route);
            }
            printf("\n");
        }

        return 0;
}


int getfile(int clientfd, msg_exch msg){

    int nread;
    int nwrite;
    char buf[2048];
     int fd;
    if((fd= open(msg.file_route, O_RDWR|O_CREAT|O_TRUNC, 0666)) == -1){
                    perror("create");
                    exit(EXIT_FAILURE);
            }


    /*file is larger recv and return 0*/
    size_t size = msg.file_size;
    size_t send_size;
    while(size  > 0 ){
        if(size <  2048){
            send_size = size;
        }
        else{
            send_size = 2048;
        }
        nread = recv(clientfd, buf, send_size, 0);
        size = size - nread;
        nwrite = write(fd, buf, nread);
        if(nwrite <= 0){
            fprintf(stderr, "write the the file\n");
            exit(EXIT_FAILURE);
        }
    }
    //sleep(1);
    close(fd);
    return 0;
}