Socket网络编程--聊天程序(5)

　　上一小节我们讲了使用select来避免使用多进程的资源浪费问题。上次只是实现了从多个客户端发送数据给服务器端，接下来就要实现从服务器端发送数据给各个客户端。

　　使用select多路转换处理聊天程序2

　　client.c 使用上一节用的那个，在那个基础上修改下面几句

     //send-recv 一些返回指没有判断，具体可以看server.c
     if((pid=fork())<)
     {
         perror("fork error\n");
     }
     else if(pid==)/*child*/
     {
         while()
         {
             fgets(sendBuf,MAX_BUF,stdin);
             if(send(sockfd,sendBuf,strlen(sendBuf),)==-)
             {
                 perror("fail to receive datas.");
             }
             memset(sendBuf,,sizeof(sendBuf));
         }
     }
     else
     {
         while()
         {
             if((recvSize=recv(sockfd,recvBuf,MAX_BUF,)==-))
             {
                 printf("Server maybe shutdown!");
                 break;
             }
             printf("Server:%s\n",recvBuf);
             memset(recvBuf,,sizeof(recvBuf));
         }
         kill(pid,SIGKILL);
     }

     close(sockfd);

     return ;
 }

　　server.c 我们在上一小节的基础上加上一个处理服务器stdin读取数据，然后遍历所有fd_A里面还连接着的客户端，然后一个一个的发送数据。

     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))
         {
　　　　　　　　　　...
         }
         //FD_COPY(recvfd,servfd);
         for(i=;i<MAX_CON_NO;i++)//最大队列进行判断，优化的话，可以使用链表
         {
             if(fd_A[i]!=)
             {
                 FD_SET(fd_A[i],&recvfd);
             }
         }

         switch(select(max_recvfd+,&recvfd,NULL,NULL,&timeout))
         {
　　　　　　　　　　...         }

         /*用于检测stdin是否有数据*/
         FD_ZERO(&servfd);
         FD_SET(STDIN_FILENO,&servfd);

         switch(select(STDIN_FILENO+,&servfd,NULL,NULL,&timeout))
         {
             case -:
                 break;
             case :
                 break;
             default:
                 /*send datas to client*/
                 if(FD_ISSET(STDIN_FILENO,&servfd))
                 {
                     fgets(sendBuf,MAX_DATA_SIZE,stdin);
                     for(i=;i<MAX_CON_NO;i++)
                     {
                         if(fd_A[i]!=)
                         {
                             printf("数据发往%d,",fd_A[i]);
                             if((sendSize=send(fd_A[i],sendBuf,strlen(sendBuf),))!=strlen(sendBuf))
                             {
                                 perror("fail");
                                 exit();
                             }
                             else
                             {
                                 printf("Success\n");
                             }
                         }
                     }
                     memset(sendBuf,,MAX_DATA_SIZE);
                     fflush(stdin);
                 }
                 break;
         }
     }
     return ;
 }

　　废话不多说，直接上运行时的截图

　　程序运行的顺序是，运行server，然后运行client1，server发送data1，client2连接server，server发送data2，client3连接server，server发送data3，此时client1退出连接，server获取fd 4 close，发送data4的时候没有发送给fd4。然后client4连接上去。分配为fd7。(虽然fd4已经关闭了，按说可以给client4分配fd4，不过太麻烦我这里就没有实现。)

　　实现聊天室功能

　　接下来就是实现聊天室功能。上面server.c中有三个select，第一个是处理随时可能来的客户端连接，第二个select是处理随时可能从客户器端来的数据，第三个select是处理随时可能从控制台输入数据，并send到各个客户端。要实现聊天室功能，就对第二个和第三个select进行合并。具体不多说，直接上代码

　　client.c 基本不变

　　server.c

 #include <stdio.h>
　　　...
 int main(int argc,char *argv[])
 {
　　　　...
     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))
         {
             ...
         }
　　　　　　　...
         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,))==- || recvSize==)
                         {
                             //perror("fail to receive datas");
                             //表示该client是关闭的
                             printf("fd %d close\n",fd_A[i]);
                             FD_CLR(fd_A[i],&recvfd);
                             fd_A[i]=;
                         }
                         else//客户端发送数据过来，然后这里进行转发
                         {
                             /*send datas to client*/
                             for(j=;j<MAX_CON_NO;j++)
                             {
                                 if(fd_A[j]!=&&i!=j)
                                 {
                                     printf("数据发往%d,",fd_A[j]);
                                     if((sendSize=send(fd_A[j],recvBuf,strlen(recvBuf),))!=strlen(recvBuf))
                                     {
                                         perror("fail");
                                         exit();
                                     }
                                     else
                                     {
                                         printf("Success\n");
                                     }
                                 }
                             }
                             //可以判断recvBuf是否为bye来判断是否可以close
                             memset(recvBuf,,MAX_DATA_SIZE);
                         }
                     }
                 }
                 break;
         }
     }
     return ;
 }

　　直接上运行时图片

　　各个程序运行的循序是先运行server，然后运行client1，client2，然后cli1发送数据data1，此时cli2接收到server发来的转发数据，然后... ...

　　好了，到现在为止已经完成了聊天室功能了，还可以随时进入随时出来。好像还不错的样子。下一节就介绍引入用户名登陆的字符界面(由于我不会图形化界面，所以就不花时间去学那个了，至于字符界面下有个curses.h据说可以弄得很好看，我这里就不弄了。有兴趣的可以去弄一下)，还有一些人性化的中文提示，然后整理一下代码。由于基本没有什么技术含量，所以如果到时候篇幅不够就再加个数据库设计，考虑使用数据库来保存用户(为了方便就使用mysql吧)。

　　小小剧透一下，如果接下来有时间将会实现以下功能 用户功能，指令功能，私聊功能，vip功能，文件发送功能，多服务器问题

　　本文地址: http://www.cnblogs.com/wunaozai/p/3871563.html