select简单实现C/S模型----多路IO转接之高并发

多路IO转接服务器之select简介

多路IO转接服务器也叫做多任务IO服务器。该类服务器实现的主旨思想是，不再由应用程序自己监视客户端连接，取而代之由内核替应用程序监视文件。

1. select能监听的文件描述符个数受限于FD_SETSIZE,一般为1024，单纯改变进程打开的文件描述符个数并不能改变select监听文件个数
2. 解决1024以下客户端时使用select是很合适的，但如果链接客户端过多，select采用的是轮询模型，会大大降低服务器响应效率，不应在select上投入更多精力

#include <sys/select.h>
/* According to earlier standards */
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
int select(int nfds, fd_set *readfds, fd_set *writefds,
			fd_set *exceptfds, struct timeval *timeout);

	nfds: 		监控的文件描述符集里最大文件描述符加1，因为此参数会告诉内核检测前多少个文件描述符的状态
	readfds：	监控有读数据到达文件描述符集合，传入传出参数
	writefds：	监控写数据到达文件描述符集合，传入传出参数
	exceptfds：	监控异常发生达文件描述符集合,如带外数据到达异常，传入传出参数
	timeout：	定时阻塞监控时间，3种情况
				1.NULL，永远等下去
				2.设置timeval，等待固定时间
				3.设置timeval里时间均为0，检查描述字后立即返回，轮询
	struct timeval {
		long tv_sec; /* seconds */
		long tv_usec; /* microseconds */微秒
	};
	void FD_CLR(int fd, fd_set *set); 	//把文件描述符集合里fd清0
	int FD_ISSET(int fd, fd_set *set); 	//测试文件描述符集合里fd是否置1
	void FD_SET(int fd, fd_set *set); 	//把文件描述符集合里fd位置1
	void FD_ZERO(fd_set *set); 			//把文件描述符集合里所有位清0

select实现C/S通信模型

服务器端

#include "wrap.h"
#include "sys/types.h"
#include "sys/time.h"
#include "ctype.h"

#define SERV_PORT 9527

int main()
{
    int listenfd, connfd;   //设置监听socket，传输sokcet                                                                                                                                                                                                                                                                                    
    char buf[128];  

    //socket地址结构
    struct sockaddr_in serv_addr, clie_addr;
    //初始化
    bzero(&serv_addr, sizeof(serv_addr));
    bzero(&clie_addr, sizeof(clie_addr));
    socklen_t clie_addr_len = sizeof(clie_addr);

    listenfd = Socket(AF_INET, SOCK_STREAM, 0); 

    //重复利用端口号
    int opt = 1;
    setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(SERV_PORT);
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);

    //绑定监听
    Bind(listenfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr));

    //设置监听上限
    Listen(listenfd, 128);

    //设置select的监听读集合 和 总的监听集合
    fd_set rset, rAllset;
    FD_ZERO(&rAllset);
    FD_SET(listenfd, &rAllset);
    //记录监听文件的最大描述符
    int maxfd = listenfd;
    while(1)
    {   
        //重置监听读集合
        rset = rAllset;
        int ret = select(maxfd + 1, &rset, NULL, NULL, NULL);
        if(-1 == ret)
        {   
            perr_exit("select failed..\n");
        }   
        if(0 == ret)
            continue;
        if(0 < ret)
        {   
            //监听到请求连接
            if(FD_ISSET(listenfd, &rset))
            {   
                // 非阻塞方式来accept，因为有请求连接可以马上处理
                connfd = Accept(listenfd, (struct sockaddr*)&clie_addr, &clie_addr_len);    
                //最大文件描述符变更
                if(maxfd < connfd)
                    maxfd = connfd;
                //把新的文件描述符加入到监听集合
                FD_SET(connfd, &rAllset);

                if(0 == --ret)
                    continue;
            }   

            //循环找出监听到要读的socketfd
            for(int i = listenfd + 1; i <= maxfd; i++)
            {   
                if(FD_ISSET(i, &rset))
                {   
                    //进行读写
                    int len = Read(i, buf, sizeof(buf));    
                    //当终止客户端或者客户端输入回车结束传输
                    if(len == 0 || len == 1)
                    {   
                        FD_CLR(i, &rAllset);
                        close(i);
                    }
                    else if(len >= 2)
                    {
                        for(int j = 0; j < len; j++)
                        {
                            buf[j] = toupper(buf[j]);
                        }
                        Write(i, buf, len);
                        Write(STDOUT_FILENO, buf, len);

                    }
                    //提前终止轮询, 当数量减为0说明没有其他socket在通信。
                    if(0 == --ret)
                        break;
                }
            }
        }


    }
    close(listenfd);


    return 0;
}

客户端

#include <stdio.h>                                                                                                                                                                                                                                                                                                                          
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <errno.h>

int main()
{
    int sockFd = socket(AF_INET, SOCK_STREAM, 0); 
    if(-1 == sockFd)
    {   
        perror("socket error\n");
        exit(0);
    }   
    struct sockaddr_in serv_addr;
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(9527);
    inet_pton(AF_INET, "127.0.0.1", &serv_addr.sin_addr.s_addr);
    int cfd = connect(sockFd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));
    if(-1 == cfd)
    {   
        perror("connect error\n");
        exit(0);
    }   
    int loop = 5;
    char buf[1024];
    while(loop--)
    {   
        int ret = read(STDIN_FILENO, buf, sizeof(buf));
        write(sockFd, buf, ret);        //写数据到服务端
        ret = read(sockFd, buf, sizeof(buf));   //从服务端读取数据
        write(STDOUT_FILENO, buf, ret);         //数据打印
        //sleep(1);
    }   
    close(sockFd);
    return 0;
}