毫无套路的【accept/listen 超时】解决指南

闲谈

2023-11-10 04:51:43

解决 accept/listen 超时难题的终极指南

问题背景

在网络编程中，accept/listen 超时是最常见的噩梦之一。它们会导致服务器无法处理新连接，从而严重影响系统性能。传统的解决方案往往不起作用，让人抓狂不已。

终极解决方案

经过广泛的测试，我们发现了一种简单有效的解决方法：使用 select() 函数 。select() 可以同时监视多个套接字，并仅在有数据可读、可写或出错时返回。

代码示例

以下是使用 select() 函数设置 accept/listen 超时时间的代码示例：

#include <sys/socket.h>
#include <sys/select.h>
#include <stdio.h>

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

  // 绑定端口
  struct sockaddr_in servaddr;
  memset(&servaddr, 0, sizeof(servaddr));
  servaddr.sin_family = AF_INET;
  servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
  servaddr.sin_port = htons(8080);

  if (bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) == -1) {
    perror("bind");
    return -1;
  }

  // 监听端口
  if (listen(listenfd, 10) == -1) {
    perror("listen");
    return -1;
  }

  // 初始化文件符集合
  fd_set readfds;
  FD_ZERO(&readfds);
  FD_SET(listenfd, &readfds);

  // 设置超时时间
  struct timeval timeout;
  timeout.tv_sec = 10;
  timeout.tv_usec = 0;

  // 使用 select() 监视套接字
  int ret = select(listenfd + 1, &readfds, NULL, NULL, &timeout);
  if (ret == -1) {
    perror("select");
    return -1;
  }

  // 检查是否有新连接
  if (FD_ISSET(listenfd, &readfds)) {
    struct sockaddr_in clientaddr;
    socklen_t len = sizeof(clientaddr);
    int connfd = accept(listenfd, (struct sockaddr *)&clientaddr, &len);
    if (connfd == -1) {
      perror("accept");
      return -1;
    }

    // 处理客户端连接
  }

  return 0;
}