系统提供select函数来实现多路复用输入/输出模型。select系统调用是用来让我们的程序监视 多个文件句柄的状态变化的。程序会停在select这里等待,直到被监视的文件句柄有一个或 多个发生了状态改变。
关于文件句柄,其实就是一个整数,我们最熟悉的句柄是0、1、2三 个,0是标准输入,1是标准输出,2是标准错误输出。0、1、2是整数表示的,对应的FILE * 结构的表⽰示就是stdin、stdout、stderr。
参数nfds是需要监视的最大的文件描述符值+1;
rdset,wrset,exset分别对应于需要检测的可读文件描述符的集合,可写文件描述符的集 合及异 常文件描述符的集合。
struct timeval结构用于描述一段时间长度,如果在这个时间内,需要监视的描述符没有事件 发生则函数返回,返回值为0。
下面的宏提供了处理这三种描述词组的方式:
FD_CLR(inr fd,fd_set* set);用来清除描述词组set中相关fd 的位
FD_ISSET(int fd,fd_set*set);用来测试描述词组set中相关fd 的位是否为真
FD_SET(int fd,fd_set*set);用来设置描述词组set中相关fd的位 >
FD_ZERO(fd_set*set);用来清除描述词组set的全部位
参数timeout为结构timeval,用来设置select()的等待时间,其结构定义如下:
struct timeval {
long tv_sec; /* seconds */
long tv_usec; /* microseconds */
};
如果参数timeout设为:
(1)NULL:则表示select()没有timeout,select将一直被阻塞,直到某个⽂文件描述符上发生了 事件。
(2)0:仅检测描述符集合的状态,然后立即返回,并不等待外部事件的发生。
(3)特定的时间值:如果在指定的时间段⾥里没有事件发⽣生,select将超时返回。
函数返回值: 执行成功则返回文件描述词状态已改变的个数
如果返回0代表在描述词状态改变前已超过timeout时间,没有返回;
当有错误发⽣生时则返回-1,错误原因存于errno,此时参数readfds,writefds,exceptfds和 timeout的值变成不可预测。
错误值可能为:
EBADF 文件描述词为无效的或该文件已关闭
EINTR 此调用被信号所中断
EINVAL 参数n 为负值。
ENOMEM 核心内存不足
理解select模型:
理解select模型的关键在于理解fd_set,为说明方便,取fd_set长度为1字节,fd_set中的每一bit 可以对应一个文件描述符fd。则1字节长的fd_set最大可以对应8个fd。
(1)执行 fd_set set; FD_ZERO(&set);则set用位表示是0000,0000。
(2)若fd=5,执行FD_SET(fd,&set);后set变为0001,0000(第5位置为1)
(3)若再加入fd=2,fd=1,则set变为0001,0011
(4)执行select(6,&set,0,0,0)阻塞等待
(5)若fd=1,fd=2上都发⽣生可读事件,则select返回,此时set变为0000,0011。注意:没有事件发生的fd=5被清空。select模型的特点:
(1)可监控的文件描述符个数取决与sizeof(fd_set)的值。
(2)将fd加入select监控集的同时,还要再使用一个数据结构array保存放到select监控集中的fd,
一是用于在select 返回后,array作为源数据和fd_set进行FD_ISSET判断。
二是select 返回后会把以前加入的但并无事件发生的fd清空,则每次开始 select前都要重新从array取得fd 逐一加入(FD_ZERO最先),扫描array的同时取得fd最大值maxfd,用于select的第一个 参 数。
(3)可见select模型必须在select前循环array(加fd,取maxfd),select返回后循环array (FD_ISSET判断是否有时间发生)。
select缺点:
(1)每次调用select,都需要把fd集合从用户态拷贝到内核态,这个开销在fd很多时会很大
(2)同时每次调用select都需要在内核遍历传递进来的所有fd,这个开销在fd很多时也很大
(3)select支持的文件描述符数量太小了,默认是1024
select使用:
使用select编写简单的网络服务器:
server端:
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <unistd.h>
#include <sys/select.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <string.h>
static void usage(const char* proc)
{
printf("Usage:%s [local_ip] [local_port]\n", proc);
}
int startup(const char* _ip, int _port)
{
int sock = socket(AF_INET, SOCK_STREAM, 0);
if(sock < 0)
{
perror("socket\n");
exit(2);
}
int opt = 1;
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
struct sockaddr_in local;
local.sin_family = AF_INET;
local.sin_port = htons(_port);
local.sin_addr.s_addr = inet_addr(_ip);
if(bind(sock, (struct sockaddr*)&local, sizeof(local)) < 0)
{
perror("bind");
exit(3);
}
if(listen(sock, 10) < 0)
{
perror("listen");
exit(4);
}
return sock;
}
int fds_array[sizeof(fd_set)*8];
int main(int argc, char* argv[])
{
if(argc != 3)
{
usage(argv[1]);
return 1;
}
int listen_sock = startup(argv[1], atoi(argv[2]));
int num = sizeof(fds_array)/sizeof(fds_array[0]);
int i = 0;
for(; i<num; ++i)
{
fds_array[i] = -1;
}
fds_array[1] = listen_sock;
while(1)
{
int maxfd = -1;
fd_set rfds;
fd_set wfds;
FD_ZERO(&rfds);
FD_ZERO(&wfds);
for(i=0; i<num; i++)
{
if(fds_array[i] == -1)
{
continue;
}
FD_SET(fds_array[i], &rfds);
FD_SET(fds_array[i], &wfds);
if(maxfd < fds_array[i])
{
maxfd = fds_array[i];
}//更新maxfd
}
struct timeval timeout = {5, 0};
switch( select(maxfd+1,&rfds, &wfds, NULL, NULL) )
{
case -1:
perror("select");
break;
case 0:
printf("timeout...\n");
break;
default:
{
for(i=0; i<num; i++)
{
if(fds_array[i] == -1)
{
continue;
}
if(i == 1 && FD_ISSET(fds_array[i], &rfds))
//监听到就绪的消息
{
struct sockaddr_in client;
int len = sizeof(client);
int new_sock = accept(listen_sock,(struct sockaddr*)&client, &len );
if(new_sock < 0)
{
perror("accept");
continue;
}
printf("get a new client...[%s : %d]\n", inet_ntoa(client.sin_addr),\
ntohs(client.sin_port));
int j = 1;
for(; j<num; j++)
{
if(fds_array[j] == -1)
{
break;
}
}
if(j == num)
{
printf("fds_array is full!!\n");
close(new_sock);
}
else
{
fds_array[j] = new_sock;
}
}//fi
else if(i != 1 && (FD_ISSET(fds_array[i], &rfds) \
|| FD_ISSET(fds_array[i], &wfds)))
//普通事件就绪
{
char buf[1024];
if(FD_ISSET(fds_array[i], &rfds))//读事件就绪
{
ssize_t s = read(fds_array[i], buf, sizeof(buf)-1);
if(s > 0)
{
buf[s] = 0;
printf("client# %s\n", buf);
if(FD_ISSET(fds_array[i],&wfds))
{
printf("please Enter# \n");
fflush(stdout);
ssize_t s = read(0, buf, sizeof(buf)-1);
if(s > 0)
{
buf[s-1] = 0;
ssize_t s = write(fds_array[i], buf, strlen(buf));
if(s > 0)
{
buf[s] = 0;
printf("server# %s\n", buf);
continue;
}
else if(s == 0)
{
close(fds_array[i]);
fds_array[i] = -1;
continue;
}
else
{
perror("write");
close(fds_array[i]);
fds_array[i] = -1;
continue;
}
}
else if(s == 0)
{
printf("client is quit...\n");
close(fds_array[i]);
fds_array[i] = -1;
continue;
}
else
{
perror("read");
close(fds_array[i]);
fds_array[i] = -1;
continue;
}
}
}
else if(s == 0)
{
printf("client is quit...\n");
close(fds_array[i]);
fds_array[i] = -1;
continue;
}
else
{
perror("read");
close(fds_array[i]);
fds_array[i] = -1;
continue;
}
}
}//fi else
}//for
}//default
}//switch
}//while
return 0;
}
client端:使用了一个简单的输出重定向(dup2)
可以参考上篇博客:
http://blog.csdn.net/qq_36221862/article/details/73839290
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <errno.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>
static void usage(const char* proc) //使用说明
{
printf("Usage:%s [local_ip] [local_port]\n", proc);
}
//tcp_client server_ip server_port
int main(int argc, char* argv[])
{
if(argc != 3)
{
usage(argv[0]);
return 1;
}
int sock = socket(AF_INET, SOCK_STREAM , 0);
if(sock < 0)
{
perror("socket\n");
return 2;
}
struct sockaddr_in server;
server.sin_family = AF_INET;
server.sin_port = htons(atoi(argv[2]));
server.sin_addr.s_addr = inet_addr(argv[1]);
if(connect(sock, (struct sockaddr*)&server, sizeof(server)) < 0)
{
perror("connect\n");
return 3;
}
umask(0);
int fd = open("./log", O_WRONLY|O_CREAT, 0644);
if(fd < 0)
{
perror("open");
return 4;
}
close(1);
int ret = dup2(fd, 1);
printf("connect success\n");
char buf[1024];
while(1)
{
printf("please Enter# \n");
fflush(stdout);
ssize_t s = read(0, buf, sizeof(buf)-1);
if(s > 0)
{
buf[s-1] = 0;
write(sock, buf, strlen(buf));
s = read(sock, buf, sizeof(buf)-1);
if(s > 0)
{
buf[s] = 0;
printf("server echo# %s\n ", buf);
}
}
}
close(fd);
close(sock);
return 0;
}
运行结果: