套接字的多种可选项

  我们在创建套接字的时候,是可以对套接字进行一些属性设置的,以满足在开发中遇到的不同的需求。本文介绍一些与套接字属性相关的函数的使用。

getsockopt

  通过此函数我们可以获取套接字可选项。

#include <sys/socket.h>
int getsockopt(int sock, int level, int optname, void *optval, socklen_t *optlen);
  • sock:套接字文件描述符
  • level:要查看的可选项的协议层

  可选项如下表所示:

协议层选项名
SOL_SOCKETSO_SNDBUF
套接字相关通用可选项SO_RCVBUF
SO_REUSEADDR
SO_KEEPALIVE
SO_BROADCAST
SO_DONTROUTE
SO_OOBINLINE
SO_ERROR
SO_TYPE
IPPROTO_IPIP_TOS
IP层套接字相关属性IP_TTL
IP_MULTICAST_TTL
IP_MULTICAST_LOOP
IP_MULTICAST_IF
IPPROTO_TCPTCP_KEEPALIVE
TCP层套接字相关属性TCP_NODELAY
TCP_MAXSEG
  • optname:要查看的可选项名,即上表的右边
  • optva:保存查看结果的缓冲地址值
  • optlenoptval保存的可选项信息的字节数

实例

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>

void error_handing(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

int main(int argc, char *argv[])
{
    int tcp_sock, udp_sock;
    int sock_type;
    socklen_t optlen;
    int state;

    optlen = sizeof(sock_type);
    tcp_sock = socket(AF_INET, SOCK_STREAM, 0);
    udp_sock = socket(AF_INET, SOCK_DGRAM, 0);
    printf("SOCK_STREAM: %d \n", SOCK_STREAM);
    printf("SOCK_DGRAM: %d \n", SOCK_DGRAM);

    // 获取TCP套接字选项值
    state = getsockopt(tcp_sock, SOL_SOCKET, SO_TYPE, (void*)&sock_type, &optlen);
    if(state) error_handing("getsockopt error");
    printf("tcp_socket type: %d \n", sock_type);

    // 获取UDP套接字选项值
    state = getsockopt(udp_sock, SOL_SOCKET, SO_TYPE, (void*)&sock_type, &optlen);
    if(state) error_handing("getsockopt error");
    printf("udp_socket type: %d \n", sock_type);

    return 0;
}

  输出结果如下:
请添加图片描述
  可以看到,直接输出SOCK_STREAM的属性值为1,SOCK_DGRAM的属性值为2,然后我们分别将这两个属性设置到套接字中,通过getsockopt获取套接字的属性,与这两个属性的值是一致的。

SOCK_SNDBUF、SOCK_RCVBUF

  SOCK_SNDBUF是关于输入缓冲区大小的可选项,SOCK_RCVBUF是关于输出缓冲区大小的可选项。

实例

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>

void error_handing(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

int main(int argc, char *argv[])
{
    int sock;
    int snd_buf, rcv_buf;
    int state;
    socklen_t len;
    
    sock = socket(AF_INET, SOCK_STREAM, 0);

    // 获取套接字输出缓冲区大小
    len = sizeof(snd_buf);
    state = getsockopt(sock, SOL_SOCKET, SO_SNDBUF, (void*)&snd_buf, &len);
    if(state) error_handing("getsockopt error");

    // 获取套接字输入缓冲区大小
    len = sizeof(rcv_buf);
    state = getsockopt(sock, SOL_SOCKET, SO_RCVBUF, (void*)&rcv_buf, &len);
    if(state) error_handing("getsockopt error");

    printf("输入缓冲区大小: %d \n", rcv_buf);
    printf("输出缓冲区大小: %d \n", snd_buf);
    
    snd_buf = 1024 * 3;
    rcv_buf = 1024 * 3;

    // 设置输入缓冲区大小
    state = setsockopt(sock, SOL_SOCKET, SO_RCVBUF, (void *)&rcv_buf, sizeof(rcv_buf));
    if(state) error_handing("setsockopt error");

    // 设置输出缓冲区大小
    state = setsockopt(sock, SOL_SOCKET, SO_SNDBUF, (void *)&snd_buf, sizeof(snd_buf));
    if(state) error_handing("setsockopt error");

    // 获取套接字输出缓冲区大小
    len = sizeof(snd_buf);
    state = getsockopt(sock, SOL_SOCKET, SO_SNDBUF, (void*)&snd_buf, &len);
    if(state) error_handing("getsockopt error");

    // 获取套接字输入缓冲区大小
    len = sizeof(snd_buf);
    state = getsockopt(sock, SOL_SOCKET, SO_RCVBUF, (void*)&rcv_buf, &len);
    if(state) error_handing("getsockopt error");

    printf("输入缓冲区大小: %d \n", rcv_buf);
    printf("输出缓冲区大小: %d \n", snd_buf);

    return 0;
}

  输出结果:
请添加图片描述
  可以看出在我们设置输入输出缓冲区大小之前的大小,然后我们通过setsockopt设置了输入输出缓冲区的大小,明显发生了改变。但是与我们设置的3M字节大小的缓冲区的要求明显不符,这是因为我们只是将我们要设置的输入输出缓冲区大小告诉系统,但系统不会完全按照我们的要求来进行设置。总是,从这个实例我们可以通过getsockoptsetsockoptSO_SNDBUFSO_RCVBUF来获取和设置套接字输入输出缓冲区的大小。

SO_REUSEADDR

  一般来说,服务器与客户端会经过三次握手建立连接,断开的时候也会经过四次挥手,服务器端请求断开会向客户端发送FIN消息,客户端回复ACK,然后再发送一个FIN,服务器再回复一个ACK,然后就会进入一个Time-wait状态,当我们立即重启服务器的时候,在调用bind函数的时候就会发生错误,需要等待一段时间才可以正常重启服务器。

  对于这种情况,我们通过设置SO_REUSEADDR可选项,就可以立即正常调用bind函数,重启服务器。SO_REUSEADDR的默认值为0(假),我们将其设置为1(真)就可以达成这个目的。

实例

  • reuseaddr_serv.c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/socket.h>

void error_handing(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

int main(int argc, char *argv[])
{
    int serv_sock,clnt_sock;
    char message[30];
    int option, str_len;
    socklen_t optlen, clnt_addr_sz;
    struct sockaddr_in serv_addr;
    struct sockaddr_in clnt_addr;
    
    serv_sock = socket(AF_INET, SOCK_STREAM, 0);
    if(serv_sock == -1) error_handing("socket error");

    /*
    optlen = sizeof(option);
    option = 1;
    setsockopt(serv_sock, SOL_SOCKET, SO_REUSEADDR, (void *)&option, optlen);
    */

    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_addr.sin_port = htons(atoi(argv[1]));

    if(bind(serv_sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1) error_handing("bind() error");
    if(listen(serv_sock, 5) == -1) error_handing("listen() error");

    clnt_addr_sz = sizeof(clnt_addr);
    clnt_sock = accept(serv_sock, (struct sockaddr*) &clnt_addr, &clnt_addr_sz);
    if(clnt_sock == -1) error_handing("accept() error");

    while((str_len = read(clnt_sock, message, sizeof(message))) != 0)
    {
        write(clnt_sock, message, str_len);
        write(1, message, str_len);
    }
    close(clnt_sock);
    close(serv_sock);
    return 0;
}
  • reuseaddr_clnt.c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <string.h>
#include <sys/socket.h>

void error_handing(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

int main(int argc, char* argv[]){
    int sock;
    struct sockaddr_in serv_addr;
    char message[] = "hello world!";
    int str_len;

    if(argc != 3){
        printf("Usage: %s <IP> <port>\n", argv[0]);
        exit(1);
    }
	// 创建面向连接的套接字
    sock = socket(PF_INET, SOCK_STREAM, 0);
    if(sock == -1) error_handing("socket() error");
    
	// 初始化网络通信地址
    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
    serv_addr.sin_port = htons(atoi(argv[2]));

    if(connect(sock,(struct sockaddr*)&serv_addr, sizeof(serv_addr)) == -1) {
		error_handing("connect() error");
	}
    str_len = read(sock, message, sizeof(message) - 1);
    if(str_len == -1) error_handing("connect() error");

    printf("Message from server: %s \n", message);
    close(sock);
    return 0;
}

  我们将服务器与客户端进行连接,然后ctrl+ C强制中断服务器,并重启,就会在调用bind函数时发生错误:
请添加图片描述
  然后我们将服务器设置SO_REUSEADDR部分的注释标记去掉重新运行:
请添加图片描述
  当强制中断后立即重新运行服务器程序,我们发现不再会发生错误。

Nagle算法与TCP_NODELAY

  Nagle算法是为了防止数据包过多从而发生网络过载设计的,应用于TCP层。在不使用Nagle算法,TCP发送数据包的时候,只要向发送缓冲区中写入数据,发送缓冲区就会将消息发送出去,比如发送Hello,就会往外发送五个数据包,'H'、'e'、'l'、'l'、'o',而使用Nagle算法以后,TCP只有在收到上次消息的时确认报文才会发送下一条消息,在等待的这段时间,会持续的将要发送的消息写入发送缓冲区。首先写入H,然后将其发送出去(因为H之前没有其他消息发送出去,没有要等待的ACK),进入等待ACK的状态,在这个状态过程中,持续的将wllo写入发送缓冲区,在收到ACK报文后,将ello发送出去,这样就将之前的五个报文,变成了'H'、'ello'两个报文,大大缩减了报文数量,降低了网络负载。
  Nagle算法在TCP是默认开启的,如果要禁用的话,在Ubuntu中需要调用一下netinet/tcp/h,具体如下:

#include <netinet/tcp.h>
int opt_val = 1;
setsockopt(serv_sock, IPPROTO_TCP, TCP_NODELAY, (void *)&opt_val, sizeof(opt_val));
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值