linux中fd_set的内部实现

一、在网络编程中，经常用到selec系统调用来判断套接字上是否存在数据可读，或者能否向一个套接字写入数据。其原型为：

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

其中，fd_set是一个socket集合，常用如下宏来对fd_set进行操作：

1 2 3 4

FD_CLR( s, *set)     //从set中删除句柄s; FD_ISSET( s, *set)   //检查句柄s是否存在与set中; FD_SET( s, *set )    //把句柄s添加到set中; FD_ZERO( *set )      //把set队列初始为空.

需要说明一点，在内核中，socket对应struct socket结构，但在返回给用户空间之前，内核做了一个关联：调用get_unused_fd_flags从当前进程中获取一个可用的文件描述符fd ，将struct socket结构关联到该fd，并返回fd给用户空间。所以在用户空间中，socket为文件描述符。另外，进程可以打开的文件数是有限制的，为1024，故socket的取值小于1024。

二、fd_set是如何实现的呢?

试想由你实现这样一个集合，可以往里添加任意0~1024之间的数(FD_SET操作），也可以将加入到集合中的数移除——移除一个（FD_CLR操作）或全部（FD_ZERO），你会如何实现？

一种比较好的思路是使用位图bitmap，往集合了添加n时只需将第n个bit位置1，移除n时只需将第n个比特置为0，移除所有数据时，只需将所有bit置为0，可以通过memset操作来实现。fd_set的实现就是采用位图bitmap（关于位图可以参考《编程珠玑》第一章）。

其定义如下：

1 2 3 4 5 6 7 8 9

#define __NFDBITS (8 * sizeof(unsigned long))                //每个ulong型可以表示多少个bit, #define __FD_SETSIZE 1024                                          //socket最大取值为1024 #define __FDSET_LONGS (__FD_SETSIZE/__NFDBITS)     //bitmap一共有1024个bit，共需要多少个ulong   typedef  struct  {      unsigned  long  fds_bits [__FDSET_LONGS];                  //用ulong数组来表示bitmap } __kernel_fd_set;   typedef  __kernel_fd_set   fd_set;

对应的操作如下：

1 2 3 4 5 6 7

//每个ulong为32位，可以表示32个bit。 //fd  >> 5 即 fd / 32，找到对应的ulong下标i；fd & 31 即fd % 32，找到在ulong[i]内部的位置   #define __FD_SET(fd, fdsetp)   (((fd_set *)(fdsetp))->fds_bits[(fd) >> 5] |= (1<<((fd) & 31)))             //设置对应的bit #define __FD_CLR(fd, fdsetp)   (((fd_set *)(fdsetp))->fds_bits[(fd) >> 5] &= ~(1<<((fd) & 31)))            //清除对应的bit #define __FD_ISSET(fd, fdsetp)   ((((fd_set *)(fdsetp))->fds_bits[(fd) >> 5] & (1<<((fd) & 31))) != 0)     //判断对应的bit是否为1 #define __FD_ZERO(fdsetp)   (memset (fdsetp, 0, sizeof (*(fd_set *)(fdsetp))))                             //memset bitmap

转自： http://www.cnblogs.com/scope-beyound/p/3628217.html