splice
函数用于在两个文件描述符之间移动数据,也是零拷贝。
splice
定义如下:
#include <fcntl.h>
ssize_t splice(int fd, loff_f* off_in, int fd_out, loff_t* off_out, size_t len, unsigned int flags);
如果fd_in是一个管道文件描述符,则参数必须设置为NULL。如果不是,则表述从输入数据流何处开始读数据。此时,若off_in被设置为NULL,则表示从输入流的当前偏移量开始读。
fd_in和fd_out必须至少有一个是管道文件描述符。
下面我们使用splice
函数来实现一个零拷贝的回射服务器,它将客户端发送的数据原样返回给客户端:
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <assert.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <libgen.h>
#include <fcntl.h>
#define BACKLOG 5
int main(int argc, char* argv[]) {
if (argc <= 2) {
printf("usage: %s ip_address port_number", basename(argv[0]));
return 1;
}
const char* ip = argv[1];
int port = atoi(argv[2]);
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(ret != -1);
ret = listen(sock, BACKLOG);
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) {
printf("errno is %d\n", errno);
} else {
int pipefd[2];
assert(ret != -1);
ret = pipe(pipefd); // 创建管道
// 将connfd上流入的客户数据定向到管道中
ret = splice(connfd, NULL, pipefd[1], NULL, 32768, SPLICE_F_MORE | SPLICE_F_MOVE);
assert( ret != -1);
// 将管道的输出定向到connfd客户连接的文件描述符
ret = splice(pipefd[0], NULL, connfd, NULL, 32768, SPLICE_F_MORE | SPLICE_F_MOVE);
assert(ret != -1);
close(connfd);
}
close(sock);
return 0;
}
我们通过splice
函数将客户端的内容读入到pipefd[1]
中,然后再使用splice
函数从pipefd[0]
中读出该内容到客户端,从而实现了简单高效的回射服务。
整个过程从未执行recv
或send
操作,因此也未涉及用户空间和内和空间之间的数据拷贝。
【注】:
pipe
函数定义:
#include<unistd.h>
pipe(int fd[2]);
它由输出型参数fd返回两个文件描述符,fd[0]为读而打开,fd[1]为写而打开,fd[1]的输出是fd[0]的输入,当管道创建成功后pipe函数返回0,如果创建失败则返回-1