sendfile(零拷贝技术)写的一个简单的服务器程序

痛定思痛。 2022-08-06 09:21 145阅读 0赞

服务器给客户端发送文件时，总是先把数据从内核空间读到用户空间，再发送给客户端，即先建立一个缓冲区，用read函数或者fread函数把文件内容读入缓冲区，再进行发送。而从内核空间拷贝到用户空间属于系统调用，是会消耗系统资源的，所以出现了零拷贝技术，即直接从服务器端的内核空间发送到客户端，少了一个拷贝的过程，从而降低系统资源消耗。

下面这段程序，是一个简单的服务器端代码，客户机连上服务器，就会收到服务器发送的文件内容，可以用telnet测试。

#include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include <unistd.h>
    #include <sys/socket.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include <assert.h>
    #include <sys/types.h>
    #include <sys/stat.h>
    #include <fcntl.h>
    #include <sys/sendfile.h>
    #include <errno.h>
    
    
    int main(int argc, char *argv[])
    {
        if (argc <= 3)
        {
            printf("usage:%s ip, port, filename\n", basename(argv[0]));
            return 1;
        }
        else
        {
            //argc == 3
        }
    
    
        const char* ip = argv[1];
        int port = atoi(argv[2]);
        const char* file_name = argv[3];
    
    
        int filefd = open(file_name, O_RDONLY);
        assert(filefd > 0);
        struct stat stat_buf;
        fstat(filefd, &stat_buf);
        
        struct sockaddr_in address;
        bzero(&address, sizeof(address));
        address.sin_family = AF_INET;
        inet_pton(AF_INET, ip, &address.sin_addr);
        address.sin_port = htons(port);
    
    
        int sock = socket(PF_INET, SOCK_STREAM, 0);
        assert(sock >= 0);
    
    
        int ret = bind(sock, (struct sockaddr*)&address, sizeof(address));
        assert(-1 != ret);
    
    
        ret = listen(sock, 5);
        assert(-1 != ret);
    
    
        struct sockaddr_in client;
        socklen_t client_addrlength = sizeof(client);
        int connfd = accept(sock, (struct sockaddr*)&client, &client_addrlength);
    
    
        if (connfd < 0)
        {
            printf("errno is: %d\n", errno);
        }
        else
        {
            sendfile(connfd, filefd, NULL, stat_buf.st_size);
            close(connfd);
        }
    
    
        close(sock);
        return 0;
    }