【IO复用-select】

IO复用-select

1.select API

select系统调用原型如下：

#include <sys/select.h>

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

参数说明：

1. nfds指定被监听的文件描述符总数。通常设置为select监听的所有文件描述符中最大值+1
2. readfds、writefds、exceptfds分别指向可读、写、异常事件对应的文件描述符集合。fd_set本质是一个整型数组，该数组的每一个bit位标记一个文件描述符。
3. timeout设置tv_sec和tv_usec都为0时select立即返回。设置为NULL，则一直阻塞直到有文件描述符就绪。
4. select成功时返回就绪的文件描述符（读、写、异常）总数；超时时间内没有任何文件描述符就绪，返回0；失败返回-1并设置errno为EINTR；

处理fd_set的几个宏：

#include <sys/select.h>

FD_ZERO(fd_set *fdset);					//清除fdset所有位
FD_SET(int fd, fd_set *fdset);			//设置fdset的位fd
FD_CLR(int fd, fd_set *fdset);			//清除fdset的位fd
int FD_ISSET(int fd, fd_set *fdset);	//测试fdset的位fd是否被设置

2.文件描述符就绪条件

下列情况下socket可读：

• socket内核接收缓存区字节数>=SO_RCVLOWAT。此时读操作返回字节数大于0；

• socket通信对方关闭连接。此时读操作返回0；

• 监听socket上有新的连接请求；

• socket上有未处理的错误。此时可以使用getsockopt来读取和清除该错误；

  下列情况下socket可写：

• socket内核接收缓存区可用字节数>=SO_SNDLOWAT。此时读操作返回字节数大于0；

• socket写操作被关闭。此时写将触发一个SIGPIPE；

• socket使用非阻塞connect连接成功或失败（超时）之后；

• socket上有未处理的错误。此时可以使用getsockopt来读取和清除该错误；
  下列情况socket能处理异常:

• socket上接收到带外数据；

3. select应用-处理带外数据实例

#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <arpa/inet.h>
#include <assert.h>
#include <errno.h>
#include <sys/select.h>

int main(int argc, char const *argv[])
{
    if (argc != 3)
    {
        printf("usage: %s <IP> <Port>\n", argv[0]);
        return 1;
    }
    const char *ip = argv[1];
    int port = atoi(argv[2]);
    /*设置sockaddr*/
    struct sockaddr_in addr;
    memset(&addr, 0, sizeof(addr));
    addr.sin_family = AF_INET;
    inet_pton(AF_INET, ip, &addr.sin_addr);
    addr.sin_port = htons(port);
	/*创建socket*/
    int sock = socket(PF_INET, SOCK_STREAM, 0);
    assert(sock >= 0);
	/*命名socket*/
    int result = bind(sock, (struct sockaddr*)&addr, sizeof(addr));
    assert(result != -1);
	/*进入监听状态*/
    result = listen(sock, 5);
    assert(result != -1);
	/*等待接收客户端连接*/
    struct sockaddr_in client_addr;
    socklen_t client_addrlen = sizeof(client_addr);
    int connfd = accept(sock, (struct sockaddr*)&client_addr, &client_addrlen);
    if (connfd < 0)
    {
        printf("accept failed, errno: %d\n", errno);
        close(sock);
    }
    /*初始化读和异常文件描述符集合*/
    char buf[1024];
    fd_set read_set;
    fd_set exception_set;
    FD_ZERO(&read_set);
    FD_ZERO(&exception_set);
    
    for (; ; )
    {
    	/*监听connfd可读和异常就绪*/
        FD_SET(connfd, &read_set);
        FD_SET(connfd, &exception_set);
        /*select阻塞等待读和异常位就绪*/
        result = select(connfd + 1, &read_set, NULL, &exception_set, NULL);
        if (result < 1)
        {
            printf("select failed!\n");
            break;
        }
		/*读集合的connfd被置位，普通数据*/
        if (FD_ISSET(connfd, &read_set))
        {
            result = recv(connfd, buf, sizeof(buf)-1, 0);
            if (result <= 0)
            {
                break;
            }
            printf("get %d bytes of normal data: %s", result, buf);
        }
        /*异常集合的connfd被置位，OOB数据*/
        if (FD_ISSET(connfd, &exception_set))
        {
            result = recv(connfd, buf, sizeof(buf)-1, MSG_OOB);
            if (result <= 0)
            {
                break;
            }
            printf("get %d bytes of OOB data: %s", result, buf);
        }
    }
    /*关闭文件*/
    close(connfd);
    close(sock);
    
    return 0;
}