Linux C select多路复用并发服务器详解

Linux C select多路复用并发服务器详解

先看代码，然后我们再来看下详解：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <strings.h>
#include <sys/types.h>          
#include <sys/socket.h>
#include <unistd.h>
#include <sys/time.h>
#include <sys/select.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define BUF_SIZE 128 //接收和发送数据长度

int main()
{
	//创建socket套接字
	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sockfd == -1)
	{
		perror("Socket");
		return -1;
	}

	//创建服务器端口
	struct sockaddr_in serveraddr;
	int serveraddr_size = sizeof(serveraddr);
	memset(&serveraddr, 0, serveraddr_size);
	
	//创建客户端端口
	struct sockaddr_in clientaddr;
	int clientaddr_size = sizeof(clientaddr);
	memset(&clientaddr, 0, clientaddr_size);
	
	//端口设置
	serveraddr.sin_family = AF_INET;
	serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
	serveraddr.sin_port = htons(8000);
	
	//允许端口重用
	int val = 1;
	setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
	
	//绑定端口
	int ret = bind(sockfd, (struct sockaddr*)&serveraddr, serveraddr_size);
	if(ret == -1)
	{
		perror("Bind");
		close(sockfd);
		return -1;
	}

	//监听客户端连接请求
	ret = listen(sockfd, 5);
	if(ret == -1)
	{
		perror("Listen");
		close(sockfd);
		return -1;
	}

	//创建消息队列
	fd_set readfds, temfds;

	//清空消息队列
	FD_ZERO(&readfds);
	//把套接字加入队列
	FD_SET(sockfd, &readfds);
	//临时队列复制
	temfds = readfds;
	//队列最大的描述符
	int maxfd = sockfd;

	int i;
	while(1)
	{
		//基准队列
		readfds = temfds;

		//阻塞等待
		ret = select(maxfd+1, &readfds, NULL, NULL, NULL);
		if(ret == -1)
		{
			perror("Selsect");
			exit(-1);
		}
		
		//循环查询是哪个描述符
		for(i=sockfd; i<maxfd+1; i++)
		{
			//判断
			if(FD_ISSET(i, &readfds))
			{
				//检测描述符是否是套接字描述符
				if(i==sockfd)
				{
					
					//建立连接
					int	connetfd = accept(sockfd, (struct sockaddr*)&clientaddr, &clientaddr_size);
					printf("connetfd:%d is link\n", connetfd);
					//将建立的连接加入队列
					FD_SET(connetfd, &temfds);
					//修改最大文件描述符
					if(maxfd < connetfd)
					{
						maxfd = connetfd;
					}
				}
				else
				{
					char readbuf[BUF_SIZE];
					memset(readbuf, 0, sizeof(readbuf));
					//接收数据
					int read_len = recv(i, readbuf, sizeof(readbuf), 0);
					if(read_len == -1)//错误
					{
						perror("Recv");
						exit(-1);
					}
					else if(read_len == 0)//连接断开
					{
						printf("Connetfd:%d is unlink\n", i);
						close(i);
						FD_CLR(i, &temfds);
					}
					puts(readbuf);
					//发送数据
					send(i, readbuf, strlen(readbuf), 0);
					printf("Message:%s\n", readbuf);
				}
			}
		}
	}
	return 0;
}

好了，我们看完了代码，我们再来看一下具体是怎么实现的：
代码的实现思想：建立出一个消息队列，新用户连接时加入队列中，当接收的任意文件描述符的消息时轮训查询是哪一个文件描述符，执行完后进入下一轮的等待接收。

selcet函数

函数原型：

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

函数功能：选择-同步I/O多路复用，监视多个文件描述符
函数参数：
++++ nfds：最大文件描述符+1；
++++ readfds：读数据文件描述符队列
++++ writefds：写数据文件描述符队列
++++ exceptfds：异常文件描述符队列
++++ timeout：阻塞超时
介绍完这个函数后不得不介绍这几个宏函数了：

void FD_CLR(int fd, fd_set *set);   //从队列中删除文件描述符fd
int  FD_ISSET(int fd, fd_set *set); //判断文件描述符fd是否返回真（有响应）
void FD_SET(int fd, fd_set *set);   //将文件描述符fd注册到队列中
void FD_ZERO(fd_set *set);			//清空队列

函数参数：
fd：文件描述符
set：队列表

好了总的来说我们只需要，不断的查询是否有请求，哪一个文件描述符发起的请求，我们就响应相应的请求就可以了。