Linux高性能服务器编程5.11.2修改缓冲区大小

最新推荐文章于 2024-09-15 11:40:06 发布

SJLongX

最新推荐文章于 2024-09-15 11:40:06 发布

阅读量408

点赞数 10

文章标签： c++

本文链接：https://blog.csdn.net/qq_40912479/article/details/135758003

版权

文章描述了在C语言中，通过使用`socket`、`setsockopt`等系统调用，客户端和服务器端如何设置SO_SNDBUF和SO_RCVBUF来控制数据包发送和接收的缓冲区大小。

摘要由CSDN通过智能技术生成

书中说大小会自动扩大，并且有最小值，我还以为都是2的倍数，实际上并不是

在这里插入图片描述

这个不知道是不是改了系统参数了。

客户端代码如下：

#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>
#define BUFFER_SIZE 1024

int main(int argc, char* argv[]){
    if(argc <= 3){
        printf("ip and port and size!!!\n");
        return 0;
    }

    char* ip = argv[1];
    int port = atoi(argv[2]);

    int sockfd = socket(AF_INET, SOCK_STREAM, 0);

    struct sockaddr_in serveraddr;
    bzero(&serveraddr, sizeof(serveraddr));
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(port);
    inet_pton(AF_INET, ip, &serveraddr.sin_addr);

    int sendbuf = atoi(argv[3]);
    int len = sizeof(sendbuf);
    setsockopt(sockfd, SOL_SOCKET, SO_SNDBUF, &sendbuf, len);
    getsockopt(sockfd, SOL_SOCKET, SO_SNDBUF, &sendbuf, (socklen_t*)&len);
    printf("sendbuf size : %d \n", sendbuf);

    if((connect(sockfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr))) < 0){
        printf("connect failure \n");
    }else{
        char buf[BUFFER_SIZE];
        memset(buf, 'a', BUFFER_SIZE);
        send(sockfd, buf, sizeof(buf), 0);
        printf("send_finish \n");
    }
    close(sockfd);
    return 0;
}

服务器端代码如下：

#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>
#define BUFFER_SIZE 1024
int main(int argc, char* argv[]){
    if(argc <= 3){
        printf("ip and port and size !!!\n");
        return 0;
    }

    char* ip = argv[1];
    int port = atoi(argv[2]);

    int sockfd = socket(AF_INET, SOCK_STREAM, 0);

    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    inet_pton(AF_INET, ip, &addr.sin_addr);

    int ret = bind(sockfd, (struct sockaddr*)&addr, sizeof(addr));
    assert(ret != -1);

    ret = listen(sockfd, 5);
    assert(ret != -1);

    int recvbuf = atoi(argv[3]);
    int len = sizeof(recvbuf);
    setsockopt(sockfd, SOL_SOCKET, SO_RCVBUF, &recvbuf, len);
    getsockopt(sockfd, SOL_SOCKET, SO_RCVBUF, &recvbuf, (socklen_t*)&len);
    printf("recvbuf size : %d \n", recvbuf);

    struct sockaddr_in clientaddr;
    socklen_t clientaddrlen;
    int connfd = accept(sockfd, (struct sockaddr*)&clientaddr, &clientaddrlen);
    if(connfd < 0){
        printf("connect failure !!!");
    }else{
        char buf[BUFFER_SIZE];
        memset(buf, '\0', BUFFER_SIZE);
        while((ret = recv(connfd, buf, BUFFER_SIZE - 1, 0)) > 0){}
        close(connfd);
    }
    close(sockfd);
    return 0;
}