Socket網絡編程--聊天程序(5)

　　上一小節我們講了使用select來避免使用多進程的資源浪費問題。上次只是實現了從多個客戶端發送數據給服務器端，接下來就要實現從服務器端發送數據給各個客戶端。

 

　　使用select多路轉換處理聊天程序2

　　client.c 使用上一節用的那個，在那個基礎上修改下面幾句

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

    close(sockfd);
    

    return 0;
}

　　server.c 我們在上一小節的基礎上加上一個處理服務器stdin讀取數據，然后遍歷所有fd_A里面還連接着的客戶端，然后一個一個的發送數據。

    while(1)
    {
        FD_ZERO(&servfd);//清空所有server的fd
        FD_ZERO(&recvfd);//清空所有client的fd
        FD_SET(sockfd,&servfd);
        //timeout.tv_sec=30;//可以減少判斷的次數
        switch(select(max_servfd+1,&servfd,NULL,NULL,&timeout))
        {
　　　　　　　　　　...
        }
        //FD_COPY(recvfd,servfd);
        for(i=0;i<MAX_CON_NO;i++)//最大隊列進行判斷，優化的話，可以使用鏈表
        {
            if(fd_A[i]!=0)
            {
                FD_SET(fd_A[i],&recvfd);
            }
        }

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

        /*用於檢測stdin是否有數據*/
        FD_ZERO(&servfd);
        FD_SET(STDIN_FILENO,&servfd);

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

　　廢話不多說，直接上運行時的截圖

　　程序運行的順序是，運行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(1)
    {
        FD_ZERO(&servfd);//清空所有server的fd
        FD_ZERO(&recvfd);//清空所有client的fd
        FD_SET(sockfd,&servfd);
        //timeout.tv_sec=30;//可以減少判斷的次數
        switch(select(max_servfd+1,&servfd,NULL,NULL,&timeout))
        {
            ...
        }
　　　　　　　...
        switch(select(max_recvfd+1,&recvfd,NULL,NULL,&timeout))
        {
            case -1:
                //select error
                break;
            case 0:
                //timeout
                break;
            default:
                for(i=0;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,0))==-1 || recvSize==0)
                        {
                            //perror("fail to receive datas");
                            //表示該client是關閉的
                            printf("fd %d close\n",fd_A[i]);
                            FD_CLR(fd_A[i],&recvfd);
                            fd_A[i]=0;
                        }
                        else//客戶端發送數據過來，然后這里進行轉發
                        {
                            /*send datas to client*/
                            for(j=0;j<MAX_CON_NO;j++)
                            {
                                if(fd_A[j]!=0&&i!=j)
                                {
                                    printf("數據發往%d,",fd_A[j]);
                                    if((sendSize=send(fd_A[j],recvBuf,strlen(recvBuf),0))!=strlen(recvBuf))
                                    {
                                        perror("fail");
                                        exit(1);
                                    }
                                    else
                                    {
                                        printf("Success\n");
                                    }
                                }
                            }
                            //可以判斷recvBuf是否為bye來判斷是否可以close
                            memset(recvBuf,0,MAX_DATA_SIZE);
                        }
                    }
                }
                break;
        }
    }
    return 0;
}

　　直接上運行時圖片

　　各個程序運行的循序是先運行server，然后運行client1，client2，然后cli1發送數據data1，此時cli2接收到server發來的轉發數據，然后... ...

　　好了，到現在為止已經完成了聊天室功能了，還可以隨時進入隨時出來。好像還不錯的樣子。下一節就介紹引入用戶名登陸的字符界面(由於我不會圖形化界面，所以就不花時間去學那個了，至於字符界面下有個curses.h據說可以弄得很好看，我這里就不弄了。有興趣的可以去弄一下)，還有一些人性化的中文提示，然后整理一下代碼。由於基本沒有什么技術含量，所以如果到時候篇幅不夠就再加個數據庫設計，考慮使用數據庫來保存用戶(為了方便就使用mysql吧)。

 

　　小小劇透一下，如果接下來有時間將會實現以下功能 用戶功能，指令功能，私聊功能，vip功能，文件發送功能，多服務器問題

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