1、基本概念
IO多路复用是指内核一旦发现进程指定的一个或者多个IO条件准备读取,它就通知该进程。IO多路复用适用如下场合:
与多进程和多线程技术相比,I/O多路复用技术的最大优势是系统开销小,系统不必创建进程/线程,也不必维护这些进程/线程,从而大大减小了系统的开销。
2、select函数
该函数准许进程指示内核等待多个事件中的任何一个发送,并只在有一个或多个事件发生或经历一段指定的时间后才唤醒。函数原型如下:
#include
#include
int select(int maxfdp1,fd_set *readset,fd_set *writeset,fd_set *exceptset,const struct timeval *timeout)
返回值:就绪描述符的数目,超时返回0,出错返回-1
函数参数介绍如下:
void FD_ZERO(fd_set *fdset); //清空集合
void FD_SET(int fd, fd_set *fdset); //将一个给定的文件描述符加入集合之中
void FD_CLR(int fd, fd_set *fdset); //将一个给定的文件描述符从集合中删除
int FD_ISSET(int fd, fd_set *fdset); // 检查集合中指定的文件描述符是否可以读写
这个参数有三种可能:
(1)永远等待下去:仅在有一个描述字准备好I/O时才返回。为此,把该参数设置为空指针NULL。
(2)等待一段固定时间:在有一个描述字准备好I/O时返回,但是不超过由该参数所指向的timeval结构中指定的秒数和微秒数。
(3)根本不等待:检查描述字后立即返回,这称为轮询。为此,该参数必须指向一个timeval结构,而且其中的定时器值必须为0。
下面是我写的一个select的demo,以便以后用到或者大家需要
/*********************************************************
- Copyright (C): 2016
- File name : server.c
- Author : - Zhaoxinan -
- Date : 2016年09月29日 星期四 15时09分06秒
- Description : 服务器selcetIO多路复用
* *******************************************************/
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
//端口号8000
#define PORT 8000
//最大连接数80
#define MAX_LINE 80
int main()
{
int i;
int ret;
int sockfd;
int new_fd;
int work_fd;
int sin_size;
int opt = 1;
int maxi;
int maxfd;
fd_set readset;
fd_set allset;
int client[FD_SETSIZE]; //客户端连接的套接字描述符数组
char buffer[MAX_LINE];
socklen_t addr_len; //地址结构长度
struct sockaddr_in server_addr;
struct sockaddr_in client_addr;
//创建一个TCP套接口
if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
{
fprintf(stderr, "Socket error!\n", strerror(errno));
exit(1);
}
//填充服务器端信息
bzero(&server_addr, sizeof(struct sockaddr_in));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
server_addr.sin_port = htons(PORT);
//让服务器退出后不用等待30s
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
//绑定IP和端口号
if (bind(sockfd, (struct sockaddr*)(&server_addr), sizeof(struct sockaddr)) == -1)
{
fprintf(stderr, "Bind error:%s\n\a", strerror(errno));
exit(1);
}
printf("bind\n");
//设置最大监听数
if (listen(sockfd, 30) == -1)
{
fprintf(stderr, "Listen error:%s\n\a", strerror(errno));
exit(1);
}
printf("listening!\n");
//对最大描述符进行初始化
maxfd = sockfd;
maxi = -1;
for (i = 0; i < FD_SETSIZE; i++)
{
client[i] = -1;
}
//清空描述符集
FD_ZERO(&allset);
//将sockfd加入描述符集
FD_SET(sockfd, &allset);
while (1)
{
readset = allset;
//得到当前可以读的文件描述符数
ret = select(maxfd + 1, &readset, NULL, NULL, NULL);
if (ret == -1)
{
printf("select error!\n");
exit(1);
}
//检测sockfd在集合中的状态是否变化
if (FD_ISSET(sockfd, &readset))
{
//进入if说明右客户端连接需要给它创建一个client_fd
addr_len = sizeof(server_addr);
if ((new_fd = accept(sockfd, (struct sockaddr*)(&client_addr), &addr_len)) == -1)
{
fprintf(stderr, "Accept error:%s\n\a", strerror(errno));
exit(1);
}
printf("accept\n");
printf("新客户端fd = %d上线\n", new_fd);
//找一个空闲的位置将newfd存储起来
//这个数组的作用就是统计目前的描述符集是否溢出
for (i = 0; i < FD_SETSIZE; i++)
{
if (client[i] <= 0)
{
client[i] = new_fd;
break;
}
}
//太多的客户端超过服务器设置的描述符集最大值,服务器拒绝链接
if (i == FD_SETSIZE)
{
printf("too many clients\n");
exit(1);
}
//将新的描述符加入描述符集
FD_SET(new_fd, &allset);
if (new_fd > maxfd)
{
maxfd = new_fd;
}
if (i > maxi)
{
maxi = i;
}
}
//对每一个连接描述符做处理
for (i = 0; i < FD_SETSIZE; i++)
{
if ((work_fd = client[i]) < 0)
{
continue;
}
//检测该fd是否变化,如果变化则进if
if (FD_ISSET(work_fd, &readset))
{
printf("reading\n");
ret = read(work_fd, buffer, MAX_LINE);
if (ret == 0)
{
//断开链接
printf("the other side has been close\n");
fflush(stdout);
close(work_fd);
//从描述符集中删除该描述符
FD_CLR(work_fd, &allset);
client[i] = -1;
}
else
{
printf("当前客户fd = %d发送信息\n", work_fd);
printf("%s\n", buffer);
printf("__________________\n");
printf("readok!\n");
}
}
}
}
close(sockfd);
return 0;
}
/*********************************************************
- Copyright (C): 2016
- File name : client.c
- Author : - Zhaoxinan -
- Date : 2016年10月07日 星期五 22时03分58秒
- Description : IO多路复用客户端
* *******************************************************/
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define PORT 8000
int main(int argc, char **argv)
{
int ret;
int sockfd;
int n_write;
char buffer[80];
struct sockaddr_in server_addr;
if (argc != 2)
{
exit(1);
}
if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
{
exit(1);
}
bzero(&server_addr, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(PORT);
server_addr.sin_addr.s_addr = inet_addr(argv[1]);
if (connect(sockfd, (struct sockaddr *)(&server_addr), sizeof(struct sockaddr)) == -1)
{
exit(1);
}
while (1)
{
memset(buffer, 0, sizeof(buffer));
scanf("%s", buffer);
n_write = write(sockfd, buffer, sizeof(buffer));
if (n_write == -1)
{
printf("write error!\n");
}
printf("_________________\n");
}
return 0;
}