43-使用 select 改进客户端

1. 程序路径

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

git clone https://git.oschina.net/ivan_allen/unp.git

如果你已经 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);     }   } }

详细代码请参考 unp/program/echo/multiplexing_select_client/echo.cc.

4. 运行结果

  • 在 flower 机器上启动服务器
flower $ ./echo -s -h flower
  • 在 sun 机器上启动客户端
sun $ ./echo -h flower

随意输入一些字符,服务器正常回射。接下来,将服务器子进程强制 kill 掉,出现图 1 的画面:


这里写图片描述
图1 服务器进程杀死后,客户端立即反应过来


这里写图片描述
图2 四次挥手,优雅断开

5. 总结

  • 掌握在网络编程中使用 select

练习 1:将本程序中的 select 改为 poll.
练习 2:使用 select 改进服务器(将多进程改为单进程)
思考:这个程序有很严重的 bug,想想在哪里?

说明:本文转自blog.csdn.net,用于学习交流分享,仅代表原文作者观点。如有疑问,请联系我们删除~