我们现在用poll替代select重写tcp回射服务器程序。在使用select早先那个版本中,我们必须分配一个client数组以及一个名为rset的描述符集。改用poll后,我们只需分配一个pollfd结构的数组来维护客户信息,而不必分配另外一个数组。
服务端程序如下所示:
分配pollfd结构数组
我们声明在pollfd结构数组中存在OPEN_MAX个元素。确定一个进程任何时刻能够打开的最大描述符数目并不容易。
方法之一是以参数_SC_OPEN_MAX调用POSIX的sysconf函数,然后动态分配一个合适大小的数组。然而sysconf的可能返回之一是“indeterminate”(不确定),意味着我们任然不得不猜测一个值。这里我们就用POSIX得OPEN_MAX常值。
我们把client数组得第一项用于监听套接字,并把其余各项的描述符成员置为-1。我们还给第一项设置POLLRDNORM事件,这样当有新的连接准备好被接受时poll将通知我们。maxi变量含有client数组当前正在使用的最大下标值。
调用poll,检查新的连接
我们调用poll以等待新的连接或者现有连接上有数据可读。当一个新的连接被接受后,我们在client数组中查找第一个描述符成员为负的可用项。注意,我们从下标1开始搜索,因为client【0】固定用于监听套接字。找到一个可用项之后,我们把新连接的描述符保存到其中,并设置POLLRDNORM事件。
检查某个现有连接上的数据
我们检查的两个返回事件是POLLRDNORM和POLLERR。其中我们并没有在event成员中设置第二个事件,因为他在条件成立时总是返回。我们检查POLLERR的原因在于,有些实现在一个连接上接收到RST时返回的是POLLERR事件,而其他实现返回的只是POLLRNNORM事件。不论哪种情形,我们都调用read,当有错误发生时,read将返回这个错误。当有一个现有连接由他的客户终止时,我们就把他的fd成员设置为-1。