Linux下的單進程多用戶TCP服務器,採用select方法實現。


  1. /************************************************* 
  2. * File name   : server.c 
  3. * Description : 單進程併發服務器 
  4. * Author      : [email protected] 
  5. * Version     : V1.0 
  6. * Date        :  
  7. * Compiler    : arm-linux-gcc-4.4.3 
  8. * Target      : mini2440(Linux-2.6.32) 
  9. * History     :  
  10. *   <author>  <time>   <version >   <desc> 
  11. *************************************************/  
  12. #include <stdio.h>   
  13. #include <string.h>   
  14. #include <unistd.h>   
  15. #include <sys/types.h>   
  16. #include <sys/socket.h>   
  17. #include <netinet/in.h>   
  18. #include <arpa/inet.h>   
  19. #include <sys/time.h>   
  20. #include <stdlib.h>   
  21.   
  22. #define PORT 1234               //服務器端口   
  23. #define BACKLOG 5               //listen隊列中等待的連接數   
  24. #define MAXDATASIZE 1024        //緩衝區大小   
  25.   
  26. typedef struct _CLIENT  
  27. {  
  28.     int fd;                     //客戶端socket描述符   
  29.     char name[20];              //客戶端名稱   
  30.     struct sockaddr_in addr;    //客戶端地址信息結構體   
  31.     char data[MAXDATASIZE];     //客戶端私有數據指針   
  32. } CLIENT;  
  33.   
  34. void process_client(CLIENT * client, char *recvbuf, int len);   //客戶請求處理函數   
  35.   
  36. /************************************************* 
  37. * Function    : main() 
  38. * Description :  
  39. * Calls       : process_client() 
  40. * Called By   :  
  41. * Input       :  
  42. * Output      :  
  43. * Return      :  
  44. *************************************************/  
  45. void main(int argc ,char **argv)  
  46. {  
  47.     int i, maxi, maxfd, sockfd;  
  48.     int nready;  
  49.     ssize_t n;  
  50.     fd_set rset, allset;        //select所需的文件描述符集合   
  51.     int listenfd, connectfd;    //socket文件描述符   
  52.     struct sockaddr_in server;  //服務器地址信息結構體   
  53.   
  54.     CLIENT client[FD_SETSIZE];  //FD_SETSIZE爲select函數支持的最大描述符個數   
  55.     char recvbuf[MAXDATASIZE];  //緩衝區   
  56.     int sin_size;               //地址信息結構體大小   
  57.   
  58.     if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)  
  59.     {                           //調用socket創建用於監聽客戶端的socket   
  60.         perror("Creating socket failed.");  
  61.         exit(1);  
  62.     }  
  63.   
  64.     int opt = SO_REUSEADDR;  
  65.     setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));  //設置socket屬性   
  66.   
  67.     bzero(&server, sizeof(server));  
  68.     server.sin_family = AF_INET;  
  69.     server.sin_port = htons(PORT);  
  70.     server.sin_addr.s_addr = htonl(INADDR_ANY);  
  71.   
  72.     if (bind(listenfd, (struct sockaddr *)&server, sizeof(struct sockaddr)) == -1)  
  73.     {                           //調用bind綁定地址   
  74.         perror("Bind error.");  
  75.         exit(1);  
  76.     }  
  77.   
  78.     if (listen(listenfd, BACKLOG) == -1)  
  79.     {                           //調用listen開始監聽   
  80.         perror("listen() error\n");  
  81.         exit(1);  
  82.     }  
  83.   
  84.     //初始化select   
  85.     maxfd = listenfd;  
  86.     maxi = -1;  
  87.     for (i = 0; i < FD_SETSIZE; i++)  
  88.     {  
  89.         client[i].fd = -1;  
  90.     }  
  91.     FD_ZERO(&allset);           //清空   
  92.     FD_SET(listenfd, &allset);  //將監聽socket加入select檢測的描述符集合   
  93.   
  94.     while (1)  
  95.     {  
  96.         struct sockaddr_in addr;  
  97.         rset = allset;  
  98.         nready = select(maxfd + 1, &rset, NULL, NULL, NULL);    //調用select   
  99.         printf("Select() break and the return num is %d. \n", nready);  
  100.   
  101.         if (FD_ISSET(listenfd, &rset))  
  102.         {                       //檢測是否有新客戶端請求   
  103.             printf("Accept a connection.\n");  
  104.             //調用accept,返回服務器與客戶端連接的socket描述符   
  105.             sin_size = sizeof(struct sockaddr_in);  
  106.             if ((connectfd =  
  107.                  accept(listenfd, (struct sockaddr *)&addr, (socklen_t *) & sin_size)) == -1)  
  108.             {  
  109.                 perror("Accept() error\n");  
  110.                 continue;  
  111.             }  
  112.   
  113.             //將新客戶端的加入數組   
  114.             for (i = 0; i < FD_SETSIZE; i++)  
  115.             {  
  116.                 if (client[i].fd < 0)  
  117.                 {  
  118.                     char buffer[20];  
  119.                     client[i].fd = connectfd;   //保存客戶端描述符   
  120.                     memset(buffer, '0'sizeof(buffer));  
  121.                     sprintf(buffer, "Client[%.2d]", i);  
  122.                     memcpy(client[i].name, buffer, strlen(buffer));  
  123.                     client[i].addr = addr;  
  124.                     memset(buffer, '0'sizeof(buffer));  
  125.                     sprintf(buffer, "Only For Test!");  
  126.                     memcpy(client[i].data, buffer, strlen(buffer));  
  127.                     printf("You got a connection from %s:%d.\n", inet_ntoa(client[i].addr.sin_addr),ntohs(client[i].addr.sin_port));  
  128.                     printf("Add a new connection:%s\n",client[i].name);  
  129.                     break;  
  130.                 }  
  131.             }  
  132.               
  133.             if (i == FD_SETSIZE)  
  134.                 printf("Too many clients\n");  
  135.             FD_SET(connectfd, &allset); //將新socket連接放入select監聽集合   
  136.             if (connectfd > maxfd)  
  137.                 maxfd = connectfd;  //確認maxfd是最大描述符   
  138.             if (i > maxi)       //數組最大元素值   
  139.                 maxi = i;  
  140.             if (--nready <= 0)  
  141.                 continue;       //如果沒有新客戶端連接,繼續循環   
  142.         }  
  143.   
  144.         for (i = 0; i <= maxi; i++)  
  145.         {  
  146.             if ((sockfd = client[i].fd) < 0)    //如果客戶端描述符小於0,則沒有客戶端連接,檢測下一個   
  147.                 continue;  
  148.             // 有客戶連接,檢測是否有數據   
  149.             if (FD_ISSET(sockfd, &rset))  
  150.             {  
  151.                 printf("Receive from connect fd[%d].\n", i);  
  152.                 if ((n = recv(sockfd, recvbuf, MAXDATASIZE, 0)) == 0)  
  153.                 {               //從客戶端socket讀數據,等於0表示網絡中斷   
  154.                     close(sockfd);  //關閉socket連接   
  155.                     printf("%s closed. User's data: %s\n", client[i].name, client[i].data);  
  156.                     FD_CLR(sockfd, &allset);    //從監聽集合中刪除此socket連接   
  157.                     client[i].fd = -1;  //數組元素設初始值,表示沒客戶端連接   
  158.                 }  
  159.                 else  
  160.                     process_client(&client[i], recvbuf, n); //接收到客戶數據,開始處理   
  161.                 if (--nready <= 0)  
  162.                     break;      //如果沒有新客戶端有數據,跳出for循環回到while循環   
  163.             }  
  164.         }  
  165.     }  
  166.     close(listenfd);            //關閉服務器監聽socket        
  167. }  
  168.   
  169. /************************************************* 
  170. * Function    : process_client() 
  171. * Description : 處理客戶端連接函數 
  172. * Calls       :  
  173. * Called By   : main() 
  174. * Input       :  
  175. * Output      :  
  176. * Return      :  
  177. *************************************************/  
  178. void process_client(CLIENT * client, char *recvbuf, int len)  
  179. {  
  180.     char sendbuf[MAXDATASIZE];  
  181.     int i;  
  182.   
  183.     printf("Received client( %s ) message: %s\n", client->name, recvbuf);  
  184.      
  185.     for (i = 0; i < len - 1; i++)  
  186.     {  
  187.         sendbuf[i] = recvbuf[len - i - 2];  
  188.     }  
  189.       
  190.     sendbuf[len - 1] = '\0';  
  191.   
  192.     send(client->fd, sendbuf, strlen(sendbuf), 0);  
  193. }  

測試結果:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章