I/O复用模型之select学习

时间:2021-10-04 21:59:27

linux下的I/O模型可以分为5种:

1、阻塞式I/O模型

2、非阻塞式I/O模型

3、I/O复用模型

4、信号驱动I/O模型

5、异步I/O模型

简单解释:

阻塞和非阻塞:就是说需要做一件事的时候能不能立即得到返回应答,如果不能立即得到返回应答而需要等待,那就是阻塞了,否则就是非阻塞。

同步和异步:就是说每次只能昨晚一件事才能做另外一件事,那就是同步,反之,如果你能够同时做几件事,并且不一定非要一件事做完了再去做另外一件事,这就是异步。

下面具体说一下上面提到的5种I/O模型:

阻塞I/O就是指那种recv,read之类的,一直需要等,等到有了拷贝数据才返回。

非阻塞I/O就是不用等,立即返回,只要设置描述符为非阻塞就可以了,但是要进程自己检查是否可读。

I/O复用其实也是阻塞的,只是可以用来等很多描述符。

信号驱动采用信号机制等待。

异步I/O就不用等了,当他告知你的时候,已经可以返回了,数据都拷贝好了。

其中前4种I/O模型都是同步模型,只有最后一种才是异步模型。

I/O复用模型详细介绍:

1、select系统调用:

select系统调用是用来让我们的程序监视多个文件描述符的状态变化,程序会停在select这里等待,直到被监视的文件描述符有某一个或多个发生了状态变化。

select()机制中提供了fd_set的数据结构,实际上是long类型的数组,每一个数组元素都能与一打开的文件句柄建立联系,建立联系的工作由程序员完成,当调用select()时,由内核根据I/O状态更改fd_set的内容,由此来通知执行了select()的进程那哪些socket或者文件可读可写。

select函数原型:

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

函数参数说明:

1、nfds:select监视的文件句柄数,视进程中的打开文件数而定,一般设为要监视各文件中最大的文件描述符加1。

2、readfds:这个文件描述符集合监视文件集中的任何文件是否有数据可读,当select函数返回时,readfds将清除其中不可读的文件描述符,只留下可读的文件描述符。

3、writefds:这个文件描述符集合监视文件集中的任何文件是否有数据可写,当select函数返回时,writefds将清除其中不可写的文件描述符,只留下可写的文件描述符。

4、exceptfds:这个文件集将监视文件集中的任何文件是否发生了错误,当select函数返回的时候,exceptfds将清除其中的其他文件描述符,只留下带有标记的OOB数据的文件描述符。

5、timeout:本次select()的超时结束时间,它可以使select处于三种状态。

(1)若将NULL以形参传入,即不传时间结构体,就是讲select置于阻塞状态,一定等到监视描述文件符集合中的某个文件描述符发生变化为止。

(2)若将时间设置为0秒0毫秒,就变成了一个纯粹的非阻塞函数,不过文件描述符是否发生变化,都立刻返回继续执行,文件无变化返回0,有变化返回一个正值。

(3)timeout的值大于0,这就是等待的超时时间,即select在timeout时间内阻塞,超过时间之内有事件到来就返回,否则在超时之后不管怎样一定返回,返回值同上所述。

其中timeval结构体如下:

struct timeval { long tv_sec; long tv_usec; } ;其中tv_sec是秒,tv_usec是微秒。

函数返回值:

正值:表示监听的文件集中有文件描述符符合要求。

零值:表示select监视超时。

负值:表示发生了错误,错误值有errno指定。

宏操作:

FD_ZERO(fd_set * set):用来清除描述词组set的全部位。

FD_SET(int fd, fd_set * set):用来设置描述词组set中的相关fd的位。

FD_ISSET(int fd, fd_set *set):检测集合中的某一个描述符是否准备就绪了。

FD_CLR(int fd, fd_set *set):用来清除描述词组set中相关fd的位。

需要注意的事项:

(1)对于可写性的检查,最好放在需要写数据的时候进行检查,如果和可读性放在同一个地方进行检查, 那么select很可能每次都会因为可写性检查成功而返回。

(2)select()调用会清空传递给它的集合参数中的内容,也就是会清空readfds, writefds, exceptfds这三个指针参数所指定的描述符集合。因此,在每次调用select()之前,必须重新初始化并把需要监视的描述符填写到相应的描述符集合中。select()调用也会清空timeout指针所指向的struct timeval结构,所以在每次调用之前select()之前也要重新填充timeout指针所指向的struct timeval结构。

一个简单的select调用例子:

服务器端:

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h> #define MYPORT 1234 //连接时使用的端口
#define MAXCLINE 7 //连接队列中的个数
#define BUF_SIZE 200 int fd[MAXCLINE]; //连接的fd int conn_amount; //当前的连接数 void showclient()
{
printf("client amount: %d\n", conn_amount);
for (int i = 0; i < MAXCLINE; ++i) {
printf("[%d]:%d ", i, fd[i]);
}
puts("");
} int main()
{
int sock_fd, new_fd; //监听套接字, 连接套接字
struct sockaddr_in server_addr; //服务器的地址信息
struct sockaddr_in client_addr; //客户端的地址信息 socklen_t sin_size; int yes = 1;
char buf[BUF_SIZE]; //建立sock_fd套接字
if ((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
perror("can not create socket");
exit(1);
} //设置与sock_fd套接字关联的选项
//SOL_SOCKET表示普通的套接字, SO_REUSEADDR表示允许重用本地地址和端口
if (setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1) {
perror("setsockopt error");
exit(1);
} server_addr.sin_family = AF_INET; //主机字节序
server_addr.sin_port = htons(MYPORT);
server_addr.sin_addr.s_addr = INADDR_ANY; //通配IP memset(server_addr.sin_zero, '\0', sizeof(server_addr.sin_zero)); if (bind(sock_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
perror("bind error");
exit(1);
} if (listen(sock_fd, MAXCLINE) == -1) {
perror("listen error");
exit(1);
} printf("listen port %d\n", MYPORT); fd_set fdsr; //定义文件描述符集 int maxsock = sock_fd + 1;
struct timeval tv; conn_amount = 0;
sin_size = sizeof(client_addr); while (true) { //初始化文件描述符集合
FD_ZERO(&fdsr); //清除描述符集
FD_SET(sock_fd, &fdsr); //把sock_fd加入描述符集 //设定超时时间
tv.tv_sec = 40;
tv.tv_usec = 0; //添加活动连接
for (int i = 0; i < MAXCLINE; ++i) {
if (fd[i] != 0) {
FD_SET(fd[i], &fdsr);
}
} //如果文件描述符中有连接请求,会做出相应的处理,实现I/O复用,多用户的连接通讯
int ret = select(maxsock, &fdsr, NULL, NULL, &tv); //没有找到有效的连接,失败
if (ret < 0) {
perror("select error");
break;
} else if (ret == 0) { //指定的时间到了,超时
puts("timeout");
continue;
} //循环判断有效的连接是否有数据到达
int cnt = 0;
for (int i = 0; i < conn_amount; ++i) {
if (FD_ISSET(fd[i], &fdsr)) {
ret = recv(fd[i], buf, sizeof(buf), 0);
//客户端连接关闭,清除文件描述符集中的相应的位
if (ret <= 0) {
printf("client [%d] close\n", i);
close(fd[i]);
FD_CLR(fd[i], &fdsr);
fd[i] = 0;
++cnt;
} else {
//有相应的数据发送过来
if (ret < BUF_SIZE) {
memset(&buf[ret], '\0', 1);
}
printf("client [%d] send:%s\n", i, buf);
}
}
} conn_amount -= cnt; if (FD_ISSET(sock_fd, &fdsr)) {
new_fd = accept(sock_fd, (struct sockaddr *)&client_addr, &sin_size); //每次返回的是一个新文件描述符
if (new_fd <= 0) {
perror("accept error");
continue;
} //添加新的fd到数组中,判断有效的连接数是否小于最大的连接数,
//如果小于的话,就把新的连接套接字加入集合
if (conn_amount < MAXCLINE) {
for (int i = 0; i < MAXCLINE; ++i) {
if (fd[i ] == 0) {
fd[i] = new_fd;
break;
}
} ++conn_amount; //连接数加1
printf("new connection client [%d]%s:%d\n", conn_amount, inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
if (new_fd > maxsock - 1) {
maxsock = new_fd + 1;
}
} else {
puts("max connections arrive, exit...");
send(new_fd, "bye", 4, 0);
close(new_fd);
continue;
}
} showclient(); } for (int i = 0; i < MAXCLINE; ++i) {
if (fd[i] != 0) {
fd[i] = 0;
close(fd[i]);
}
} return 0;
}

客户端:

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <unistd.h>
#include <stdio.h>
#include <netdb.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h> #define SERVPORT 1234 int main()
{
int sock;
struct sockaddr_in sa;
char buffer[1024]; sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sock == -1) {
perror("can not create socket");
exit(1);
} memset(&sa, 0, sizeof(struct sockaddr_in)); sa.sin_family = AF_INET;
sa.sin_port = htons(SERVPORT);
sa.sin_addr.s_addr = htonl(0x7f000001); if (connect(sock, (struct sockaddr *)&sa, sizeof(struct sockaddr)) == -1) {
perror("connect error");
exit(1);
}
freopen("2_select_client", "r", stdin);
while (gets(buffer)) {
int sendbytes = write(sock, buffer, 100);
if (sendbytes == -1) {
perror("send error");
exit(1);
}
}
fclose(stdin);
close(sock);
return 0;
}