**
一、五种网络I/O模型
**
假设有这样一个需求:我们需要在一个程序里要查看按键是否要按下,同时他还要从串口里读取数据进行处理,也要处理网络 上来的数据。按照我们以前阻塞的模式来写的话这个程序的伪代码逻辑就是:
if( read(fd_keypad, buf, sizeof(buf)) > 0) //阻塞1
{
turn_buzzer(on);
}
if( read(fd_serialport, buf, sizeof(buf)) > 0 ) //阻塞2
{
process_serialport_data(buf);
}
if( read(fd_socket, buf, sizeof(buf)) > 0 ) //阻塞3
{
process_socket_data(buf);
}
在上面的伪代码中我们可以看到,如果按键这时没按下(即数据没有准备好)read()系统调用不会返回在那,即使现在串口或网 络socket有数据到来也没法处理。下面我们使用多路复用来解决这个问题
**
二、select多路复用
** select()函数允许进程指示内核等待多个事件(文件描述符)中的任何一个发生,并只在有一个或多个事件发生或经历一段指定时 间后才唤醒它,然后接下来判断究竟是哪个文件描述符发生了事件并进行相应的处理。
调用select函数头文件:
#include <sys/select.h>
#include <sys/time.h>
函数:
int select(int max_fd, fd_set *readset, fd_set *writeset, fd_set *exceptset, struct timeval *timeout);
- select函数的返回值是就绪描述符的数目,超时时返回0,出错返回-1; 2. 第一个参数max_fd指待测试的fd的总个数,它的值是待测试的最大文件描述符加1。Linux内核从0开始到max_fd-1扫描文件描述 符,如果有数据出现事件(读、写、异常)将会返回;假设需要监测的文件描述符是8,9,10,那么Linux内核实际也要监测0~7,此时真 正带测试的文件描述符是0~10总共11个,即max(8,9,10)+1,所以第一个参数是所有要监听的文件描述符中最大的+1。 3. 中间三个参数readset、writeset和exceptset指定要让内核测试读、写和异常条件的fd集合,如果不需要测试的可以设置为 NULL; 4. 最后一个参数是设置select的超时时间,如果设置为NULL则永不超时。
- 需要注意的是待测试的描述集总是从0, 1, 2, …开始的。 所以, 假如你要检测的描述符为8, 9, 10, 那么系统实际也要 监测0, 1, 2, 3, 4, 5, 6, 7, 此时真正待测试的描述符的个数为11个, 也就是max(8, 9, 10) + 1
- 在Linux内核有个参数__FD_SETSIZE定义了每个FD_SET的句柄个数中,这也意味着select所用到的FD_SET是有限的,也正是这个 原因select()默认只能同时处理1024个客户端的连接请求
timeval实例
truct timeval old_response_timeout;
struct timeval response_timeout;
/* Save original timeout */
modbus_get_response_timeout(ctx, &old_response_timeout);
/* Define a new and too short timeout! */
response_timeout.tv_sec = 0;
response_timeout.tv_usec = 0;
modbus_set_response_timeout(ctx, &response_timeout);
下面是使用select()多路复用实现的服务器端示例代码:
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <time.h>
#include <pthread.h>
#include <getopt.h>
#include <libgen.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#define ARRAY_SIZE(x) (sizeof(x)/sizeof(x[0]))
static inline void msleep(unsigned long ms);
static inline void print_usage(char *progname);
int socket_server_init(char *listen_ip, int listen_port);
int main(int argc, char **argv) {
int listenfd, connfd;
int serv_port = 0;
int daemon_run = 0; char *progname = NULL; int opt; fd_set rdset;
int rv;
int i, j;
int found;
int maxfd=0;
char buf[1024];
int fds_array[1024];
struct option long_options[] = { {"daemon", no_argument, NULL, 'b'}, {"port", required_argument, NULL, 'p'}, {"help", no_argument, NULL, 'h'}, {NULL, 0, NULL, 0} };
progname = basename(argv[0]);
/* Parser the command line parameters */
while ((opt = getopt_long(argc, argv, "bp:h", long_options, NULL)) != -1) {
switch (opt) {
case 'b': daemon_run=1;
break;
case 'p': serv_port = atoi(optarg);
break;
case 'h': /* Get help information */
print_usage(progname);
return EXIT_SUCCESS;
default:
break;
}
}
if( !serv_port )
{
print_usage(progname); return -1;
}
if( (listenfd=socket_server_init(NULL, serv_port)) < 0 ) {
printf("ERROR: %s server listen on port %d failure\n", argv[0],serv_port); return -2; } printf("%s server start to listen on port %d\n", argv[0],serv_port);
/* set program running on background */ if( daemon_run ) { daemon(0, 0); }
for(i=0; i<ARRAY_SIZE(fds_array) ; i++) { fds_array[i]=-1;
}
fds_array[0] = listenfd;
for ( ; ; )
{
FD_ZERO(&rdset);
for(i=0; i<ARRAY_SIZE(fds_array) ; i++)
{
if( fds_array[i] < 0 )
continue;
maxfd = fds_array[i]>maxfd ? fds_array[i] : maxfd;
FD_SET(fds_array[i], &rdset);
}
/* program will blocked here */
rv = select(maxfd+1, &rdset, NULL, NULL, NULL);
if(rv < 0)
{
printf("select failure: %s\n", strerror(errno));
break;
}
else if(rv == 0)
{
printf("select get timeout\n");
continue;
}
/* listen socket get event means new client start connect now */ if ( FD_ISSET(listenfd, &rdset) )
{
if( (connfd=accept(listenfd, (struct sockaddr *)NULL, NULL)) < 0)
{
printf("accept new client failure: %s\n", strerror(errno));
continue; }
found = 0;
for(i=0; i<ARRAY_SIZE(fds_array) ; i++) {
if( fds_array[i] < 0 )
{ printf("accept new client[%d] and add it into array\n", connfd );
fds_array[i] = connfd;
found = 1; break; }
}
if( !found )
{
printf("accept new client[%d] but full, so refuse it\n", connfd);
close(connfd);
}
} else /* data arrive from already connected client */
{
for(i=0; i<ARRAY_SIZE(fds_array); i++)
{
if( fds_array[i]<0 || !FD_ISSET(fds_array[i], &rdset)
) continue;
if( (rv=read(fds_array[i], buf, sizeof(buf))) <= 0) { printf("socket[%d] read failure or get disconncet.\n", fds_array[i]); close(fds_array[i]); fds_array[i] = -1; } else { printf("socket[%d] read get %d bytes data\n", fds_array[i], rv);
/* convert letter from lowercase to uppercase */ for(j=0; j<rv; j++) buf[j]=toupper(buf[j]);
if( write(fds_array[i], buf, rv) < 0 ) { printf("socket[%d] write failure: %s\n", fds_array[i], strerror(errno)); close(fds_array[i]); fds_array[i] = -1; } } } } }
CleanUp: close(listenfd); return 0; }
static inline void msleep(unsigned long ms) { struct timeval tv;
tv.tv_sec = ms/1000; tv.tv_usec = (ms%1000)*1000; select(0, NULL, NULL, NULL, &tv); }
static inline void print_usage(char *progname) { printf("Usage: %s [OPTION]...\n", progname); printf(" %s is a socket server program, which used to verify client and echo back string from it\n", progname); printf("\nMandatory arguments to long options are mandatory for short options too:\n"); printf(" -b[daemon ] set program running on background\n"); printf(" -p[port ] Socket server port address\n"); printf(" -h[help ] Display this help information\n");
printf("\nExample: %s -b -p 8900\n", progname); return ; }
int socket_server_init(char *listen_ip, int listen_port) { struct sockaddr_in servaddr; int rv = 0; int on = 1; int listenfd;
if ( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) { printf("Use socket() to create a TCP socket failure: %s\n", strerror(errno)); return -1; }
/* Set socket port reuseable, fix 'Address already in use' bug when socket server restart */ setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
memset(&servaddr, 0, sizeof(servaddr)); servaddr.sin_family = AF_INET; servaddr.sin_port = htons(listen_port);
if( !listen_ip ) /* Listen all the local IP address */ { servaddr.sin_addr.s_addr = htonl(INADDR_ANY); } else /* listen the specified IP address */ { if (inet_pton(AF_INET, listen_ip, &servaddr.sin_addr) <= 0) { printf("inet_pton() set listen IP address failure.\n"); rv = -2; goto CleanUp; } }
if(bind(listenfd, (struct sockaddr *) &servaddr, sizeof(servaddr)) < 0) { printf("Use bind() to bind the TCP socket failure: %s\n", strerror(errno)); rv = -3; goto CleanUp; }
if(listen(listenfd, 13) < 0) { printf("Use bind() to bind the TCP socket failure: %s\n", strerror(errno)); rv = -4; goto CleanUp; }
CleanUp:
if(rv<0)
close(listenfd);
else
rv = listenfd;
return rv; }
基于select的I/O复用模型的是单进程执行可以为多个客户端服务,这样可以减少创建线程或进程所需要的CPU时间片或内存 资源的开销;此外几乎所有的平台上都支持select(),其良好跨平台支持是它的另一个优点。当然它也有两个主要的缺点:
- 每次调用 select()都需要把fd集合从用户态拷贝到内核态,之后内核需要遍历所有传递进来的fd,这时如果客户端fd很多 时会导致系统开销很大; 2. 单个进程能够监视的文件描述符的数量存在最大限制,在Linux上一般为1024,可以通过setrlimit()、修改宏定义甚至重 新编译内核等方式来提升这一限制,但是这样也会造成效率的降低;