Linux 并发echo服务器之I/O复用之select

select

简介：

将待监听套接字加入集合，监测，若有数据到来，进行echo，若有新连接请求到来，accept并将对应套接字加入集合

函数：

int select(int , fd_set* rd, fd_set* wr , fd_set* except , const struct timeval * )

参数详解:
第一个：int，套接字集中值最大的加1（上限）
第二-四个：待监测集，若有事件发生，返回就绪描述符个数
第五个：timeval* ，监测时间

流程

1. 创建套接字->绑定地址->监听：s
2. 定义一个数组，存储套接字；
   定义一个int maxs，存储套接字值最大
   定义一个f’d_set rd（本程序进检测可读集）
3. while
   [1] 清空rd，添加监听套接字s和已连接客户端的套接字到rd中
   [2] 调用select
   [3] 若监听套接字可读，即有新连接请求，加入到rd中
   [4] 若rd集可读，有客户端发送消息，接收并回发
      #若出错或断开连接，从rd中删除对应套接字
4. close(s)

Code

#include <iostream>

#include <unistd.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/select.h>
#include <sys/time.h>

#include <netinet/in.h>
#include <arpa/inet.h>

#include <strings.h>
using std::cout;

const unsigned short port=8888;

int Socket(){
    int s=socket(AF_INET,SOCK_STREAM,0);
    
    struct sockaddr_in addr;
    bzero(&addr,sizeof(addr));
    addr.sin_family=AF_INET;
    addr.sin_port=htons(port);
    addr.sin_addr.s_addr=htonl(INADDR_ANY);

    bind(s,(struct sockaddr*)&addr,sizeof(addr));
    listen(s,5);
    return s;
}

/*
signal(SIGINT,sig_process);
    signal(SIGPIPE,sig_process);
void sig_process(int signo){
    printf("Catch a exit signal :%d",signo);
}
*/

void Chat(int c,int fds[],fd_set* rd){
	    ssize_t sz=0;
	    char buf[1024]={0};
        sz=recv(c,buf,sizeof(buf),0 );
        if(sz<=0){
            FD_CLR(c,rd);
            fds[c]=0;
        }
        cout<<buf<<'\n';
        sz=send(c,buf,sz,0);
        if(sz<=0){
            FD_CLR(c,rd);
            fds[c]=0;
        }

}
void Connect(int s){
    struct sockaddr_in addr;
    socklen_t len;
    
    int fds_sz=1024;
    int fds[fds_sz];
    for(int i=0;i<fds_sz;++i) fds[i]=0;
    
    fd_set rd;
    int maxfd=s;
    while(1){
        FD_ZERO(&rd);
        FD_SET(s,&rd);
        for(int i=0;i<fds_sz;++i){
            if(fds[i]==1)
                FD_SET(i,&rd);
        }       

        int err=select(maxfd+1,&rd,NULL,NULL,NULL);
        
        if( FD_ISSET(s,&rd) ) {
            int c=accept(s,(struct sockaddr*)&addr,&len);
            if(c<0) continue;
            cout<<"-----Connect:"<<c<<"\n";
            
            FD_SET(c,&rd);
            FD_CLR(s,&rd);
            if(maxfd<c) maxfd=c;
            fds[c]=1;
        }


        for(int i=0;i<fds_sz;++i){
            if(  FD_ISSET(i,&rd)  ){
                Chat(i,fds,&rd);
            }
        }
    }
}

int main()
{
	int s=Socket();
    Connect(s);
    close(s);
    return 0;
}