linux网络编程八:高效的sendfile,实现零拷贝

来源:互联网 发布:百度推广账户换域名 编辑:程序博客网 时间:2024/06/10 01:16

最近在看《linux高性能服务器编程》,在此做个日记,以激励自己,同时分享于有需要的朋友。


1. sendfile函数

#include <sys/sendfile.h>ssize_t sendfile(int out_fd, int in_fd, off_t *offset, size_t count);

sendfile在两个文件描术符之间直接传递数据,完全在内核中操作,从而避免了内核缓冲区到用户缓冲区的拷贝,因此效率很高,称为零拷贝。

out_fd参数是待写入描述符。

in_fd参数是待读出的描述符。

offset参数指定从读入文件流的哪个位置开始读,如果为空,则从文件流的默认超始位置读入。

count参数指定传输的字节数。

调用成功时返回传输的字节数,失败则为-1,并设置errno


注: man手册指出,in_fd必须是一个支持类似mmap函数的文件描述符,即它必须指向真实的文件,不能是socket和管道;而out_fd必须是一个socket。可见,sendfile专为网络传输文件而生。



2. 代码:

//用sendfile函数传输文件#include <stdio.h>#include <stdlib.h>#include <unistd.h>#include <string.h>#include <assert.h>#include <errno.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#include <sys/types.h>#include <sys/stat.h>#include <sys/sendfile.h>#include <fcntl.h>int main(int argc, char **argv){if (argc <= 3) {fprintf(stderr, "Usage: %s ip port filename\n", basename(argv[0]));return 1;}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;address.sin_port = htons(port);inet_pton(AF_INET, ip, &address.sin_addr);int sock = socket(PF_INET, SOCK_STREAM, 0);assert(sock >= 0);int ret = bind(sock, (struct sockaddr*)&address, sizeof(address));assert(ret != -1);ret = listen(sock, 5);assert(ret != -1);struct sockaddr_in client;socklen_t client_addrlength = sizeof(client);int connfd = accept(sock, (struct sockaddr*)&client, &client_addrlength);if (connfd < 0) {fprintf(stderr, "errno is: %s\n", strerror(errno));}else {sendfile(connfd, filefd, NULL, stat_buf.st_size);close(connfd);}close(sock);return 0;}





0 0