#include <stdio.h>

 #include <stdlib.h>

 #include <errno.h>

 #include <string.h>

 #include <netdb.h>

 #include <sys/types.h>

 #include <sys/socket.h>

 #include <sys/time.h>

 #include <sys/un.h>

 #include <sys/ioctl.h>

 #include <sys/wait.h>

 #include <sys/select.h>

 #include <netinet/in.h>

 #include <arpa/inet.h>

 #define SERVER_PORT 12138

 #define BACKLOG 20

 #define MAX_CON_NO 10

 #define MAX_DATA_SIZE 4096

 int MAX(int a,int b)

 {

     if(a>b) return a;

     return b;

 }

 int main(int argc,char *argv[])

 {

     struct sockaddr_in serverSockaddr,clientSockaddr;

     char sendBuf[MAX_DATA_SIZE],recvBuf[MAX_DATA_SIZE];

     int sendSize,recvSize;

     int sockfd,clientfd;

     fd_set servfd,recvfd;//用于select处理用的

     int fd_A[BACKLOG+];//保存客户端的socket描述符

     int conn_amount;//用于计算客户端的个数

     int max_servfd,max_recvfd;

     int on=;

     socklen_t sinSize=;

     char username[];

     int pid;

     int i;

     struct timeval timeout;

     if(argc != )

     {

         printf("usage: ./server [username]\n");

         exit();

     }

     strcpy(username,argv[]);

     printf("username:%s\n",username);

     /*establish a socket*/

     if((sockfd = socket(AF_INET,SOCK_STREAM,))==-)

     {

         perror("fail to establish a socket");

         exit();

     }

     printf("Success to establish a socket...\n");

     /*init sockaddr_in*/

     serverSockaddr.sin_family=AF_INET;

     serverSockaddr.sin_port=htons(SERVER_PORT);

     serverSockaddr.sin_addr.s_addr=htonl(INADDR_ANY);

     bzero(&(serverSockaddr.sin_zero),);

     /*

      * SOL_SOCKET.SO_REUSEADDR 允许重用本地地址

      * */

     setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on));

     /*bind socket*/

     if(bind(sockfd,(struct sockaddr *)&serverSockaddr,sizeof(struct sockaddr))==-)

     {

         perror("fail to bind");

         exit();

     }

     printf("Success to bind the socket...\n");

     /*listen on the socket*/

     if(listen(sockfd,BACKLOG)==-)

     {

         perror("fail to listen");

         exit();

     }

     timeout.tv_sec=;//1秒遍历一遍

     timeout.tv_usec=;

     sinSize=sizeof(clientSockaddr);//注意要写上，否则获取不了IP和端口

     FD_ZERO(&servfd);//清空所有server的fd

     FD_ZERO(&recvfd);//清空所有client的fd

     FD_SET(sockfd,&servfd);

     conn_amount=;

     max_servfd=sockfd;//记录最大的server端描述符

     max_recvfd=;//记录最大的client端的socket描述符

     while()

     {

         FD_ZERO(&servfd);//清空所有server的fd

         FD_ZERO(&recvfd);//清空所有client的fd

         FD_SET(sockfd,&servfd);

         //timeout.tv_sec=30;//可以减少判断的次数

         switch(select(max_servfd+,&servfd,NULL,NULL,&timeout))//为什么要+1，是因为第一个参数是所有描述符中最大的描述符fd号加一，原因的话在APUE中有讲，因为内部是一个数组，第一个参数是要生成一个这样大小的数组

         {

             case -:

                 perror("select error");

                 break;

             case :

                 //在timeout时间内，如果没有一个描述符有数据，那么就会返回0

                 break;

             default:

                 //返回准备就绪的描述符数目

                 if(FD_ISSET(sockfd,&servfd))//sockfd 有数据表示可以进行accept

                 {

                     /*accept a client's request*/

                     if((clientfd=accept(sockfd,(struct sockaddr  *)&clientSockaddr, &sinSize))==-)

                     {

                         perror("fail to accept");

                         exit();

                     }

                     printf("Success to accpet a connection request...\n");

                     printf(">>>>>> %s:%d join in!\n",inet_ntoa(clientSockaddr.sin_addr),ntohs(clientSockaddr.sin_port));

                     //每加入一个客户端都向fd_A写入

                     fd_A[conn_amount++]=clientfd;

                     max_recvfd=MAX(max_recvfd,clientfd);

                 }

                 break;

         }

         //FD_COPY(recvfd,servfd);

         for(i=;i<MAX_CON_NO;i++)//最大队列进行判断，优化的话，可以使用链表

         {

             if(fd_A[i]!=)

             {

                 FD_SET(fd_A[i],&recvfd);//对所有还连着服务器的客户端都放到fd_set中用于下面select的判断

             }

         }

         switch(select(max_recvfd+,&recvfd,NULL,NULL,&timeout))

         {

             case -:

                 //select error

                 break;

             case :

                 //timeout

                 break;

             default:

                 for(i=;i<conn_amount;i++)

                 {

                     if(FD_ISSET(fd_A[i],&recvfd))

                     {

                         /*receive datas from client*/

                         if((recvSize=recv(fd_A[i],recvBuf,MAX_DATA_SIZE,))==-)

                         {

                             //perror("fail to receive datas");

                             //表示该client是关闭的

                             printf("close\n");

                             FD_CLR(fd_A[i],&recvfd);

                             fd_A[i]=;//表示该描述符已经关闭

                         }

                         else

                         {

                             printf("Client:%s\n",recvBuf);

                             //可以判断recvBuf是否为bye来判断是否可以close

                             memset(recvBuf,,MAX_DATA_SIZE);

                         }

                     }

                 }

                 break;

         }

     }

     return ;

 }