Linux select 实现并发服务器和客户端

转载：http://blog.csdn.net/szkbsgy/article/details/10558881

[cpp] view plaincopy  
 <span style="font-size:18px;">服务端：  
 #include <stdio.h>  
 #include <stdlib.h>  
 #include <string.h>  
 #include <sys/time.h>  
 #include <sys/types.h>  
 #include <sys/socket.h>  
 #include <netinet/in.h>  
 #include <sys/select.h>  
 #include <errno.h>  
   
   
 #define SERVER_IP "127.0.0.1"  
 #define SERVER_PORT 8000  
   
   
 #define MAX_RECV_LEN 1024  
 #define MAX_CLIENT_NUM 30  
 #define BACK_LOG 20  
   
   
 static int running = 1;  
   
   
 int main(int argc, char *argv[])  
 {  
     int sock_fd = -1;  
     int ret = -1;  
     struct sockaddr_in serv_addr;  
     struct sockaddr_in cli_addr;  
     socklen_t serv_addr_len = 0;  
     socklen_t cli_addr_len = 0;  
     int client_fd[MAX_CLIENT_NUM];  
     char recv_buf[MAX_RECV_LEN];  
     int new_conn_fd = -1;  
     int i = 0;  
     int max_fd = -1;  
     int num = -1;  
     struct timeval timeout;  
   
   
     fd_set read_set;  
     fd_set write_set;  
     fd_set select_read_set;  
   
   
     FD_ZERO(&read_set);  
     FD_ZERO(&write_set);  
     FD_ZERO(&select_read_set);  
   
     for (i = 0; i < MAX_CLIENT_NUM; i++)  
     {  
         client_fd[i] = -1;  
     }   
   
     memset(&serv_addr, 0, sizeof(serv_addr));  
     memset(&cli_addr, 0, sizeof(cli_addr));  
   
   
     sock_fd = socket(AF_INET, SOCK_STREAM, 0);  
     if (sock_fd < 0)  
     {  
         perror("Fail to socket");  
         exit(1);  
     }  
   
   
     serv_addr.sin_family = AF_INET;  
     serv_addr.sin_port = htons(SERVER_PORT);  
     serv_addr.sin_addr.s_addr = inet_addr(SERVER_IP);  
   
   
     unsigned int value = 1;  
     if (setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR,  
                 (void *)&value, sizeof(value)) < 0)  
     {  
         perror("Fail to setsockopt");  
         exit(1);  
     }  
   
   
     serv_addr_len = sizeof(serv_addr);  
     if (bind(sock_fd, (struct sockaddr*)&serv_addr, serv_addr_len) < 0)  
     {  
         perror("Fail to bind");  
         exit(1);  
     }  
     if (listen(sock_fd, BACK_LOG) < 0)  
     {  
         perror("Fail to listen");  
         exit(1);  
     }  
   
     char buf[1024];  
     max_fd = sock_fd;  
     int len;  
     FD_SET(sock_fd, &read_set);  
     while (running)  
     {  
         timeout.tv_sec = 5;  
         timeout.tv_usec = 0;  
   
   
         max_fd = sock_fd;  
         for (i = 0; i < MAX_CLIENT_NUM; i++)  
         {  
             if (max_fd < client_fd[i])  
             {  
                 max_fd = client_fd[i];  
             }  
         }  
   
   
         select_read_set = read_set;  
         ret = select(max_fd + 1, &select_read_set, NULL, NULL, &timeout);  
         if (ret == 0)  
         {  
             printf("timeout\n");  
         }  
         else if (ret < 0)  
         {  
             printf("error occur\n");  
         }  
         else  
         {  
             if (FD_ISSET(sock_fd, &select_read_set))  
             {  
                 printf("new client comes\n");  
                 len = sizeof(cli_addr);  
                 new_conn_fd = accept(sock_fd, (struct sockaddr*)&cli_addr, &len);  
                 if (new_conn_fd < 0)  
                 {  
                     perror("Fail to accept");  
                     exit(1);  
                 }  
                 else  
                 {  
                     for (i = 0; i < MAX_CLIENT_NUM; i++)  
                     {  
                         if (client_fd[i] == -1)  
                         {  
                             client_fd[i] = new_conn_fd;  
                             FD_SET(new_conn_fd, &read_set);  
                             break;  
                         }  
                         if (max_fd < new_conn_fd)  
                         {  
                             max_fd = new_conn_fd;  
                         }  
                     }  
                 }  
             }  
             else  
             {  
                 for (i = 0; i < MAX_CLIENT_NUM; i++)  
                 {  
                     if (-1 == client_fd[i]) {  
                         continue;  
                     }  
                     memset(recv_buf, 0, MAX_RECV_LEN);  
                     if (FD_ISSET(client_fd[i], &select_read_set))  
                     {  
                         num = read(client_fd[i], recv_buf, MAX_RECV_LEN);  
                         if (num < 0)  
                         {  
                             printf("Client(%d) left\n", client_fd[i]);  
                             FD_CLR(client_fd[i], &read_set);  
                             close(client_fd[i]);  
                             client_fd[i] = -1;  
                         }  
                         else if (num > 0)  
                         {  
                             recv_buf[num] = '\0';  
                             printf("Recieve client(%d) data\n", client_fd[i]);  
                             printf("Data: %s\n\n", recv_buf);  
                         } if (num == 0)  
                         {  
                             printf("Client(%d) exit\n", client_fd[i]);  
                             FD_CLR(client_fd[i], &read_set);  
                             close(client_fd[i]);  
                             client_fd[i] = -1;  
                         }  
                     }  
                 }  
             }  
         }  
     }  
     return 0;  
 }  
   
 客户端：  
 #include <stdio.h>  
 #include <stdlib.h>  
 #include <string.h>  
 #include <unistd.h>  
 #include <sys/time.h>  
 #include <sys/types.h>  
 #include <sys/socket.h>  
 #include <netinet/in.h>  
 #include <sys/select.h>  
   
   
 #define SERVER_IP "127.0.0.1"  
 #define SERVER_PORT 8000  
   
   
 #define MAX_RECV_LEN 1024  
   
   
 static int running = 1;  
   
   
 int main(int argc, char *argv[])  
 {  
     int sock_fd = -1;  
     int ret = -1;  
     struct sockaddr_in serv_addr;  
     struct sockaddr_in cli_addr;  
     socklen_t serv_addr_len = 0;  
   
     memset(&serv_addr, 0, sizeof(serv_addr));  
   
   
     sock_fd = socket(AF_INET, SOCK_STREAM, 0);  
     if (sock_fd < 0)  
     {  
         perror("Fail to socket");  
         exit(1);  
     }  
   
   
     serv_addr.sin_family = AF_INET;  
     serv_addr.sin_port = htons(SERVER_PORT);  
     serv_addr.sin_addr.s_addr = inet_addr(SERVER_IP);  
   
   
     serv_addr_len = sizeof(serv_addr);  
     if (connect(sock_fd, (struct sockaddr *)&serv_addr, serv_addr_len) < 0)  
     {  
         perror("Fail to connect");  
         exit(1);  
     }  
     char buf[1024];  
     int num = 0;  
     while (running)  
     {  
         num = read(STDIN_FILENO, buf, MAX_RECV_LEN);  
         if (num > 0)  
         {  
             buf[num] = '\0';  
             printf("buf: %s\n", buf);  
             num = write(sock_fd, buf, num);  
             if (num < 0)  
             {  
                 printf("write failed\n");  
                 exit(1);  
             }  
   
   
             if (strncmp(buf, "exit", strlen("exit")) == 0)  
             {  
                 printf("Client exit\n");  
                 close(sock_fd);  
                 return 0;  
             }   
         }  
     }  
     return 0;  
 } </span>  