大家好,又见面了,我是你们的朋友全栈君。
sendfile函數linux內核新加的函數,可以使得網絡傳輸文件時用户层无需分配缓冲区给将要传输的文件,从而能够节约内存,并直接调用系统调用
#include <sys/sendfile.h>
ssize_t sendfile(int out_fd,int in_fd,off_t offset,size_t count);
实例:
#include <stdio.h>
#include <sys/socket.h>
#include <sys/sendfile.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <sys/stat.h>
#include <stdlib.h>
#define LISTENQ 100
int main(int argc,char** argv) {
if (argc != 3) {
printf("please add <port> <sendfile-name>\n");
return -1;
}
int sockfd;
if ((sockfd = socket(AF_INET,SOCK_STREAM,0)) < 0) {
printf("socket error: %s\n",strerror(errno));
return -1;
}
struct sockaddr_in server;
bzero(&server,sizeof(server));
server.sin_family = AF_INET;
server.sin_addr.s_addr = htonl(INADDR_ANY);
server.sin_port = htons(atoi(argv[1]));
if (bind(sockfd,(struct sockaddr*)&server,sizeof(server)) < 0) {
printf("bind error: %s\n",strerror(errno));
return -1;
}
if (listen(sockfd,LISTENQ) < 0) {
printf("listen error: %s\n",strerror(errno));
return -1;
}
int connfd;
for (; ;) {
if ((connfd = accept(sockfd,NULL,NULL)) < 0) {
if (errno == EINTR) {
continue;
}
printf("accept error: %s\n",strerror(errno));
return -1;
}
struct stat file_stat;
bool work = true;
if (stat(argv[2],&file_stat) < 0) {
printf("stat error: %s\n",strerror(errno));
work = false;
}
if (S_ISREG(file_stat.st_mode)) {
int fd;
if ((fd = open(argv[2],O_RDONLY)) < 0) {
printf("open error: %s\n",strerror(errno));
return -1;
}
if (sendfile(connfd,fd,0,file_stat.st_size) != file_stat.st_size) {
work = false;
printf("sendfile error: %s\n",strerror(errno));
return -1;
}
work = false;
close(connfd);
close(fd);
continue;
}
if (!work) {
char buf[100] = "Invalid request\n";
if (write(connfd,buf,strlen(buf)) != strlen(buf)) {
close(connfd);
printf("write error: %s\n",strerror(errno));
continue;
}
close(connfd);
}
}
return 0;
}
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/135647.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...