43-使用 select 改进客户端

原创

mb63083a7dd962a 2022-08-26 11:58:28 ©著作权

文章标签 select 网络编程 IO多路复用服务器客户端 文章分类 运维

©著作权归作者所有：来自51CTO博客作者mb63083a7dd962a的原创作品，请联系作者获取转载授权，否则将追究法律责任

1. 程序路径

代码托管在 gitos 上，请使用下面的命令获取：

git clone

如果你已经 clone 过这个代码了，请使用 git pull 更新一下。本节程序所使用的程序路径是 unp/program/echo/multiplexing_select_client。

2. 回忆 select

2.1 函数原型

/* According to POSIX.1-2001 */
#include <sys/select.h>

/* According to earlier standards */
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>

int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout);

void FD_CLR(int fd, fd_set *set);
int  FD_ISSET(int fd, fd_set *set);
void FD_SET(int fd, fd_set *set);
void FD_ZERO(fd_set *set);

上面这些函数相信你不陌生，后面我们写程序会用到。

2.2 selece 的参数与返回值

（1）参数

nfds 表示三个集合中最大描述符的值 + 1
后面三个 fd_set 集合分别表示监听哪种类型的事件，分别表示读事件，写事件和异常事件集合
最后一个参数是超时参数，可以为 NULL，表示永远等待

（2）返回值

小于 0，失败
等于 0，超时时间到
大于 0，发生了 IO 事件的个数

3. 改进客户端

在旧版本（processzombie 以其之前我们写过的程序）中，我们使用的客户端基本上都是这样：

void doClient(int sockfd) {
  // 客户端阻塞在标准输入上，一旦客户端收到了服务器的 FIN，也无能为力
  while(fgets(buf, stdin)) {
    write(sockfd, buf);
    read(sockfd, buf);
    puts(buf);
  }
}

缺点是它无法感知服务器进程发来的 FIN 段，现在改进如下（伪代码）：

void doClient(int sockfd) {
  fds.add(STDIN_FILENO);
  fds.add(sockfd);
  maxfd = max(STDIN_FILENO, sockfd);

  while(1) {
    rfds = fds;
    // 只监听了标准输入和套接字 sockfd
    nready = select(maxfd + 1, &rfds, NULL, NULL, NULL);
    if (STDIN_FILENO in rfds) {
      if (fgets(buf, stdin) != NULL) {
        write(sockfd, buf);
      }
      else {
        break;
      }
    }

    if (sockfd in rfds) {
      if (read(sockfd, buf) == 0) {
        puts("peer closed");
        break;
      }
      puts(buf);
    }
  }
}