多路复用——select

        在之前我们说过TCP的模型，用户可以通过连接服务器从而通过服务器去加载服务端对应功能的文件来达到通信，但是我们知道，一个用户加载文件需要一个进程，在Linux中，一个进程就要占用4G的虚拟内存，如果在并发量高的情况下，很明显服务器是不堪重负的。下面来讲解决方案即多路复用技术。

        多路复用，举个例子就好比把多条狭窄的乡间小路修成一条双向16车道的快速路，即在这条大路上去进行网络通信。多路复用技术的核心就在于在单个网络信道上传输多个网络信号，能能大幅提高通信效率，并且能够很大程度上降低服务器的负担。

下面我们用TCP作为通信模型来看多路复用的第一种 I/O 机制，即select：

服务端如下：

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

//tcp server 只有server端能用多路复用
int main(int argc, char *argv[])
{
    //创建socket
    int svr_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (svr_fd < 0)
     {
        perror("socket");
        return -1;
    }
    //准备通信地址
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(8888);
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    socklen_t addrlen=sizeof(addr);
    //绑定地址
    if (bind(svr_fd, (struct sockaddr*)&addr, addrlen) < 0)
    {
        perror("bind");
        return -1;
    }
    //监听连接
    if (listen(svr_fd, 10))
    {
        perror("listen");
        return -1;
    }
    //定义一个文件描述符集合并清空
 	fd_set reads;
	FD_ZERO(&reads);
	//把需要等待的socket描述符添加到集合中
	FD_SET(svr_fd,&reads);
	//定义超时时间
	struct timeval timeout={5,500};
	//记录最大的socket描述符
	int max_fd=svr_fd;//当前socket描述符就这一个，所以也是最大值

	char buf[4096];
	size_t buf_size=sizeof(buf);

	while(1)
	{
		//若有多个新的连接时，集合会发生变化，而把之前的没发生变化的文件描述符给覆盖了
		fd_set reads_copy=reads;//备份集合
		int ret=select(max_fd+1,&reads_copy,NULL,NULL,&timeout);
		if(ret>0)
		{
			//一开始，先测试网络等待的socket描述符
			if(FD_ISSET(svr_fd,&reads_copy))
			{
				//调用accpet连接客户端
				int cli_fd=accept(svr_fd,(struct sockaddr*)&addr,&addrlen);
				if(cli_fd<0)
				{
					perror("accept");
				}
				else
				{
					//把客户端的socket描述符添加到监控集合中
					FD_SET(cli_fd,&reads);
					if(cli_fd>max_fd)
					{
						max_fd=cli_fd;
					}
				}
			}
			else
			{
				//测试其他socket描述符是否发生变化
				for(int fd=3;fd<=max_fd;fd++)
				{
					if(FD_ISSET(fd,&reads_copy)&&fd!=svr_fd)
					{
						int ret=recv(fd,buf,buf_size,0);
						if(ret<=0)
						{
							FD_CLR(fd,&reads);
							printf("客户端%d退出\n",fd);
							continue;
						}
						printf("recv:%s bits:%d\n",buf,ret);
						strcat(buf,":return");
						ret=send(fd,buf,strlen(buf)+1,0);
						if(ret<=0 || 0==strcmp("quit",buf))
						{
							FD_CLR(fd,&reads);
							printf("客户端%d退出\n",fd);
							continue;
						}
					}
				}
			}
		}
	}

	return 0;
}

客户端如下：

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


typedef struct sockaddr *SP;

int main(int argc,const char* argv[])
{
	//创建socket
	int cli_fd=socket(AF_INET,SOCK_STREAM,0);
	if(cli_fd<0)
	{
		perror("socket");
		return -1;
	}
	//准备通信地址
	struct sockaddr_in addr={};
	addr.sin_family=AF_INET;
	addr.sin_port=htons(8888);
	addr.sin_addr.s_addr=inet_addr("127.0.0.1");
	socklen_t addrlen=sizeof(addr);
	//连接服务器
	if(connect(cli_fd,(SP)&addr,addrlen))
	{
		perror("connect");
		return -1;
	}

	char buf[4096];
	size_t buf_size=sizeof(buf);
	while(1)
	{
		//发送请求
		printf(">>>>>");
		scanf("%s",buf);
		int ret=send(cli_fd,buf,strlen(buf)+1,0);
		//ret=write(cli_fd,buf,strlen(buf)+1);
		if(ret<=0)
		{
			printf("服务器正在升级，请稍后重试\n");
			break;
		}
		if(0==strcmp("quit",buf))
		{
			printf("通信结束\n");
			break;
		}
		//接收请求
		//int ret=read(cli_fd,buf,buf_size);
		ret=recv(cli_fd,buf,buf_size,0);
		if(ret<=0)
		{
			printf("服务器正在维护，请稍候重试\n");
			break;
		}
		printf("read:%s bits:%d\n",buf,ret);
	}
		
	return 0;
}

over