本篇博客介绍如何用select函数实现并发服务器。
下面贴代码:
服务端代码
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/select.h>
#define SERV_PORT 8888
#define SERV_IP "127.0.0.1" //本地回环接口
#define LIST 20 //服务器最大接受连接
#define MAX_FD 10 //FD_SET支持描述符数量
int main(void)
{
int sockfd;
int err;
int i;
int connfd;
int fd_all[MAX_FD]; //保存所有描述符,用于select调用后,判断哪个可读
//下面两个备份原因是select调用后,会发生变化,再次调用select前,需要重新赋值
fd_set fd_read; //FD_SET数据备份
fd_set fd_select; //用于select
struct timeval timeout; //超时时间备份
struct timeval timeout_select; //用于select
struct sockaddr_in serv_addr; //服务器地址
struct sockaddr_in cli_addr; //客户端地址
socklen_t serv_len;
socklen_t cli_len;
//超时时间设置
timeout.tv_sec = 10;
timeout.tv_usec = 0;
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if(sockfd < 0)
{
perror("fail to socket");
exit(1);
}
memset(&serv_addr, 0, sizeof(serv_addr));
memset(&cli_addr, 0, sizeof(cli_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(SERV_PORT);
serv_addr.sin_addr.s_addr = inet_addr(SERV_IP);
serv_len = sizeof(serv_addr);
err = bind(sockfd, (struct sockaddr *)&serv_addr, serv_len);
if(err < 0)
{
perror("fail to bind");
exit(1);
}
err = listen(sockfd, LIST);
if(err < 0)
{
perror("fail to listen");
exit(1);
}
//初始化fd_all数组
memset(&fd_all, -1, sizeof(fd_all));
fd_all[0] = sockfd; //第一个为监听套接字
FD_ZERO(&fd_read);
FD_SET(sockfd, &fd_read); //将监听套接字加入fd_set
char buf[1024]; //读写缓冲区
int num;
int maxfd;
maxfd = fd_all[0]; //监听的最大套接字
while(1)
{
//每次都需要重新赋值
fd_select = fd_read;
timeout_select = timeout;
// err = select(maxfd+1, &fd_select, NULL, NULL, NULL);
err = select(maxfd+1, &fd_select, NULL, NULL, (struct timeval *)&timeout_select);
if(err < 0)
{
perror("fail to select");
exit(1);
}
if(err == 0)
printf("timeout\n");
//检测监听套接字是否可读
if(FD_ISSET(sockfd, &fd_select))
{
//可读,证明有新客户端连接服务器
cli_len = sizeof(cli_addr);
connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &cli_len);
if(connfd < 0)
{
perror("fail to accept");
exit(1);
}
//将新连接套接字加入fd_all及fd_read
for(i=0; i<MAX_FD; i++)
{
if(fd_all[i] != -1)
{
continue;
}
else
{
fd_all[i] = connfd;
printf("client fd_all[%d] join\n", i);
break;
}
}
FD_SET(connfd, &fd_read);
if(maxfd < connfd)
{
maxfd = connfd; //更新maxfd
}
}
//从1开始查看连接套接字是否可读,因为上面已经处理过0(sockfd)
for(i=1; i<MAX_FD; i++)
{
if(FD_ISSET(fd_all[i], &fd_select))
{
printf("fd_all[%d] is ok\n", i);
num = read(fd_all[i], buf, 1024);
if(num > 0)
{
if(strncmp("exit", buf, 4) == 0)
{
//客户端退出,关闭套接字,并从监听集合清除
printf("client:fd_all[%d] exit\n", i);
FD_CLR(fd_all[i], &fd_read);
close(fd_all[i]);
fd_all[i] = -1;
continue;
}
//收到 客户端数据并打印
buf[num] = '\0';
printf("receive buf from client fd_all[%d] is: %s\n", i, buf);
}
//回复客户端
num = write(fd_all[i], "ok", sizeof("ok"));
if(num < 0)
{
perror("fail to write ");
exit(1);
}
else
{
printf("send reply\n");
}
}
else
{
//printf("no data\n");
}
}
}
return 0;
}
select函数解析:
函数原型
int select(int maxfdp,fd_set *readfds,fd_set *writefds,fd_set *errorfds,struct timeval *timeout);
先解释数据结构 struct fd_set ,可以理解为一个集合,实际上是一个位图,每一个特定为来标志相应大小文件描述符,这个集合中存放的是文件描述符(file descriptor),即文件句柄(也就是位图上的每一位都能与一个打开的文件句柄(文件描述符)建立联系,这个工作由程序员来完成),这可以是我们所说的普通意义的文件,当然Unix下任何设备、管道、FIFO等都是文件形式,全部包括在内,所以毫无疑问一个socket就是一个文件,socket句柄就是一个文件描述符。fd_set集合可以通过一些宏由人为来操作,程序员通过操作4类宏,来完成最fd_set的操作:
(1)、FD_ZERO(fd_set *) 清空一个文件描述符集合;
(2)、FD_SET(int ,fd_set *)将一个文件描述符添加到一个指定的文件描述符集合中;
(3)、FD_CLR(int ,fd_set*) 将一个给定的文件描述符从集合中删除;
(4)、FD_ISSET(int ,fd_set* )检查集合中指定的文件描述符是否可以读写。
深入的理解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)int maxfdp是一个整数值,是指集合中所有文件描述符的范围,即所有文件描述符的最大值加1,不能错。
(2)fd_set *readfds是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的读变化的,即我们关心是否可以从这些文件中读取数据了,如果这个集合中有一个文件可读,select就会返回一个大于0的值,表示有文件可读;如果没有可读的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的读变化。
(3)fd_set *writefds是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的写变化的,即我们关心是否可以向这些文件中写入数据了,如果这个集合中有一个文件可写,select就会返回一个大于0的值,表示有文件可写,如果没有可写的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的写变化。
(4)fd_set *errorfds同上面两个参数的意图,用来监视文件错误异常文件。
(5)struct timeval* timeout是select的超时时间,这个参数至关重要,它可以使select处于三种状态,第一,若将NULL以形参传入,即不传入时间结构,就是将select置于阻塞状态,一定等到监视文件描述符集合中某个文件描述符发生变化为止;第二,若将时间值设为0秒0毫秒,就变成一个纯粹的非阻塞函数,不管文件描述符是否有变化,都立刻返回继续执行,文件无变化返回0,有变化返回一个正值;第三,timeout的值大于0,这就是等待的超时时间,即 select在timeout时间内阻塞,超时时间之内有事件到来就返回了,否则在超时后不管怎样一定返回,返回值同上述。
客户端代码
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define SERV_PORT 8888
#define SERV_IP "127.0.0.1"
int main()
{
int sockfd;
int err;
int connfd;
struct sockaddr_in serv_addr;
struct sockaddr_in cli_addr;
socklen_t serv_len;
socklen_t cli_len;
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if(sockfd < 0)
{
perror("fail to socket");
exit(1);
}
memset(&serv_addr, 0, sizeof(serv_addr));
memset(&cli_addr, 0, sizeof(cli_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(SERV_PORT);
serv_addr.sin_addr.s_addr = inet_addr(SERV_IP);
serv_len = sizeof(serv_addr);
//客户端不需要绑定,直接连接即可
err = connect(sockfd, (struct sockaddr *)&serv_addr, serv_len);
if(err < 0)
{
perror("fail to bind");
exit(1);
}
char buf[1024];
int num;
while(1)
{
sleep(2);
num = read(STDIN_FILENO, buf, 1024);
if(num > 0)
{
//exit代表退出
if(strncmp("exit", buf, 4) == 0)
{
write(sockfd, buf, num);
break;
}
}
write(sockfd, buf, num);
}
num = read(sockfd, buf, 1024);
if(num > 0)
{
buf[num] = '\0';
printf("server reply: %s\n", buf);
}
else
printf("error to read\n");
close(sockfd);
return 0;
}