学步园

1. server.c  

(a)为了将传过来的文件保存为不同的文件内容，故用本机时间的十六进制作为文件名

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>

#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>

#include<sys/stat.h>
#include<fcntl.h>
#include<time.h>

#include<errno.h>
#define oops(msg) { perror(msg); exit(errno); }

int main(){
        int sd = socket(PF_INET, SOCK_STREAM, 0);
        if(sd == -1) oops("socket");

        struct sockaddr_in address;
        bzero((void *)&address, sizeof(address));
        address.sin_family = AF_INET;
        address.sin_addr.s_addr = inet_addr("192.168.227.128");
        address.sin_port =  8888;

        int bind_return =  bind(sd, (struct sockaddr *)&address, sizeof(address));
        if(bind_return == -1) oops("bind");

        int listen_return = listen(sd, 1);
        if(listen_return == -1) oops("listen");

        while(1){
                int client_fd = accept(sd, NULL, NULL);
                if(client_fd == -1) oops("accept");

                time_t current_time = time(NULL);
                char fileName[50];
                sprintf(fileName, "/root/study/socket/%x", current_time);
                int fd = open(fileName, O_WRONLY|O_CREAT, S_IRUSR|S_IWUSR);
                if(fd == -1) oops("open");

                char buff[1024];
                ssize_t length;
                while((length = recv(client_fd, buff, sizeof(buff), 0)) > 0){
                        write(fd, buff, length);
                }

                close(fd);
        }
        return EXIT_SUCCESS;
}

2. client.c

(1) 通过sendfile进行内核空间向socket传文件，也就是所为的zero copy。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>

#include<sys/types.h>
#include<sys/socket.h>
#include<sys/sendfile.h>
#include<netinet/in.h>
#include<arpa/inet.h>

#include<sys/stat.h>
#include<fcntl.h>

#include<errno.h>

#define oops(msg) { perror(msg); exit(errno); }

int main(int argc, char *argv[]){
        if(argc != 2)oops("which file not specified?");

        int sd = socket(PF_INET, SOCK_STREAM, 0);
        if(sd == -1) oops("socket");

        struct sockaddr_in address;
        bzero(&address, sizeof(address));
        address.sin_family = AF_INET;
        address.sin_addr.s_addr = inet_addr("192.168.227.128");
        address.sin_port =  8888;

        int connect_ret = connect(sd, (struct sockaddr*)&address, sizeof(address));
        if(connect_ret == -1) oops("connect");

        int fd = open(argv[1], O_RDONLY);
        off_t offset = 0;
        struct stat fileStatus;
        int fstat_ret = fstat(fd, &fileStatus);
        if(fstat_ret == -1)oops("fstat");

        ssize_t length = sendfile(sd, fd, &offset, fileStatus.st_size);
        if(length == -1)
                {oops("sendfile");}
        else
                fprintf(stdout, "%ld characters have been send to server.\n", length);

        close(fd);

        close(sd);
        return EXIT_SUCCESS;
}