深入研究socket编程（6）——高级I/O复用（select、poll 、 epoll）

问题聚焦：

    前篇提到了I/O处理单元的四种I/O模型。

    本篇详细介绍实现这些I/O模型所用到的相关技术。

    核心思想：I/O复用

---

使用情景：

• 客户端程序要同时处理多个socket。
• 客户端程序要同时处理用户输入和网络连接。
• TCP服务器要同时处理监听socket和连接socket，这是使用最多的场合。
• 服务器要同时处理TCP请求和UDP请求。
• 服务器要同时监听多个端口或者处理多种服务。

主要技术：

• select
• poll
• epoll

---

select系统调用

作用：

    在一段指定时间内，监听用户感兴趣的文件描述符的可读、可写和异常等事件。

    

select API

原型：

[cpp]  view plain copy print ?

1. #include <sys/select.h>  
2. int select ( int nfds, fd_set* readfds, fde_set* writefds, fd_set* exceptfds, struct timeval* timeout );  

函数说明：

• nfds： 指定被监听的文件描述符的总数，通常为所有文件描述符中的最大值+1
• readfds、writefds 、exceptfds： 可读、可写和异常等事件对应的文件描述符集合。
• fd_set结构：仅包含一个整型数组，该数组的每个元素的每一位标记了一个文件描述符。fd_set能容纳的文件描述符数量由FD_SETSIZE指定，这就限制了select能同时处理的文件描述符的总量。

       select中的fd_set集合容量的限制为FD_SETSIZE，一般为1024 。修改它，需要重新编译内核。

fd_set相关的位操作：

[cpp]  view plain copy print ?

1. #include <sys/select.h>  
2. FD_ZERO( fd_set *fdset );  
3. FD_SET( int fd, fd_set *fdset );  
4. FD_CLR( int fd, fd_set *fdset );  
5. int FD_ISSET( int fd, fd_set *fdset );  

• timeout：设置select的超时时间。这是timeval结构指针，用来告诉内核select等待多久。不过我们不能完全信任select调用返回后的timeout值，比如调用失败时，timeout值是不确定的。timeval结构体如下：

[cpp]  view plain copy print ?

1. struct timeval  
2. {  
3.      long tv_sec;       \\秒数  
4.      long tv_usec;     \\微秒  
5. }  

       如果给timeout变量的tv_sec成员和tv_usec成员都传递0，则select将立即返回。如果给timeout传递NULL，则select将一直阻塞，直到某个文件描述符就绪。

返回状态：

• select成功时返回就绪（可读、可写和异常）文件描述符的总数。
• 如果在超时时间内没有任何文件描述符就绪，select将返回0。
• select失败时返回-1并设置errno。
• 如果select 等待期间，程序接收到信号，则select立即返回-1，并设置errno为EINTR。

文件描述符就绪条件

可读：

• socket内核接收缓冲区中的字节数大于或等于其低水位标记SO_RCVLOWAT。此时我们可以无阻塞地对该socket，并且读操作返回的字节数大于0。
• socket通信的对方关闭连接，此时读操作返回0。
• 监听socket上有新的连接请求。
• socekt上有未处理的错误，此时我们可以使用getsockopt来读取和清除该错误。

可写：

• socket内核发送缓冲区中的可用字节数大于或等于其低水位标记SO_SNDLOWAT。此时我们可以无阻塞地写该socket，并且写操作返回的字节数大于0。
• socket的写操作被关闭。对写操作被关闭的socket执行写操作将出发一个SIGPIPE信号。
• socket使用非阻塞connect连接成功或者失败之后。
• socket上有未处理的错误，此时我们可以使用getsockopt来读取和清除该错误。

异常：

• socket上接收到带外数据。

处理带外数据

socket上接收到普通数据和带外数据都将使select返回，但socket处于不同的就绪状态：前者处于可读状态，后者处于异常状态。

[cpp]  view plain copy print ?

1. #include <sys/types.h>  
2. #include <sys/socket.h>  
3. #include <netinet/in.h>  
4. #include <arpa/inet.h>  
5. #include <assert.h>  
6. #include <stdio.h>  
7. #include <unistd.h>  
8. #include <errno.h>  
9. #include <string.h>  
10. #include <fcntl.h>  
11. #include <stdlib.h>  
12.   
13. int main( int argc, char* argv[] )  
14. {  
15.         if( argc <= 2 )  
16.         {  
17.             printf( "usage: %s ip_address port_number\n", basename( argv[0] ) );  
18.             return 1;      
19.         }  
20.         const char* ip = argv[1];  
21.         int port = atoi( argv[2] );  
22.         printf( "ip is %s and port is %d\n", ip, port );  
23.           
24.         int ret = 0;  
25.         struct sockaddr_in address;  
26.         bzero( &address, sizeof( address ) );  
27.         address.sin_family = AF_INET;  
28.         inet_pton( AF_INET, ip, &address.sin_addr );  
29.         address.sin_port = htons( port );  
30.   
31.         int listenfd = socket( PF_INET, SOCK_STREAM, 0 );  
32.         assert( listenfd >= 0 );  
33.   
34.         ret = bind( listenfd, ( struct sockaddr* )&address, sizeof( address ) );  
35.         assert( ret != -1 );  
36.   
37.         ret = listen( listenfd, 5 );  
38.         assert( ret != -1 );  
39.   
40.         struct sockaddr_in client_address;  
41.         socklen_t client_addrlength = sizeof( client_address );  
42.         int connfd = accept( listenfd, ( struct sockaddr* )&client_address, &client_addrlength );  
43.         if ( connfd < 0 )  
44.         {  
45.              printf( "errno is: %d\n", errno );  
46.              close( listenfd );  
47.         }  
48.   
49.         char remote_addr[INET_ADDRSTRLEN];  
50.         printf( "connected with ip: %s an