2022-4-16 splice实现回射服务器《Linux高性能服务器笔记》

服务端

#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<sys/types.h>
#include<sys/stat.h>
#include<fcntl.h>
#include<sys/sendfile.h>
//ssize_t splice(int fd_in, loff_t *off_in, int fd_out,
//                      loff_t *off_out, size_t len, unsigned int flags);
int main(void){
    
    
    //1.连接到客户端
    struct sockaddr_in address;
    bzero(&address,sizeof(address));
    address.sin_family = AF_INET;
    address.sin_addr.s_addr = INADDR_ANY;
    address.sin_port = htons(9999);
    int sock = socket(AF_INET,SOCK_STREAM,0);
    if(sock == -1){
    
    
        perror("socket");
        return -1;
    }
    //2.绑定地址
    int ret = bind(sock,(struct sockaddr*)&address,sizeof(address));
    if(ret == -1){
    
    
        perror("bind");
        return -1;
    }
    //3.监听
    ret = listen(sock,5);
    if(ret == -1){
    
    
        perror("listen");
        return -1;
    }
    struct sockaddr_in client;
    socklen_t client_addrlength = sizeof(client);
    int connfd = accept(sock,(struct sockaddr*)&client,&client_addrlength);
    //检测是否连接到客户端
    if(connfd == -1){
    
    
        perror("connect");
        return -1;
    }else {
    
    
        char clientIP[16];
        inet_ntop(AF_INET,&client.sin_addr.s_addr,clientIP,sizeof(clientIP));
        unsigned short port = ntohs(client.sin_port);
        printf("client's ip is %s,port is %d.\n",clientIP,port);

        int pipefd[2];
        ret = pipe(pipefd);
        ret = splice(connfd,NULL,pipefd[1],NULL,32768,SPLICE_F_MORE|SPLICE_F_MOVE);
        if(ret == -1){
    
    
            perror("splice1");
            return -1;
        }
        ret = splice(pipefd[0],NULL,connfd,NULL,32768,SPLICE_F_MORE|SPLICE_F_MOVE);
        if(ret == -1){
    
    
            perror("splice2");
            return -1;
        }
        close(connfd);
    }
    close(sock);
    return 0;
}

客户端

//tcp 通信客户端
#include<stdio.h>
#include<arpa/inet.h>
#include<stdlib.h>
#include<unistd.h>
#include<string.h>

int main(void){
    
    
    //1.创建套接字
    int fd = socket(AF_INET,SOCK_STREAM,0);
    if(fd == -1){
    
    
        perror("socket");
        exit(0);
    }

    //2.连接服务器端
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    inet_pton(AF_INET,"192.168.131.136",&serveraddr.sin_addr.s_addr);
    serveraddr.sin_port = htons(9999);
    int ret = connect(fd,(struct sockaddr*)&serveraddr,sizeof(serveraddr));
    if(ret == -1){
    
    
        perror("connect");
        exit(0);
    }

    //3.通信
    char *data = "hello , i am client.";
    write(fd,data,strlen(data));

    char recvBuf[1024] = {
    
    0}; 
    while(read(fd,recvBuf,sizeof(recvBuf)) > 0){
    
    
        printf("recv data :%s\n",recvBuf);
    }
    // int len2 = read(fd,recvBuf,sizeof(recvBuf));
    // if (len2 == -1)
    // {
    
    
    //     perror("read");
    //     exit(-1);
    // }
    // else if(len2 > 0){
    
    
    //     printf("recv client data : %s\n",recvBuf);

    // }else if(len2 == 0){
    
    
    //     //表示服务端断开连接
    //     printf("server closed.\n");
    // }
    close(fd);

    return 0;
}

服务端返回

客户端返回