1. /* 采用链表来保存【连接socket】 */
    2. #include "wrap.h"
    3. #include <stdlib.h>
    4. #include <stdio.h>
    5. #include <unistd.h>
    6. #include <errno.h>
    7. #include <sys/socket.h>    
    8. #include <arpa/inet.h>
    9. #include <ctype.h>
    10. #include <strings.h>
    11. #define SRV_PORT 2222
    12. #define LISTEN_BACKLOG 128
    13. #define CLIENT_MAXSIZE 1024
    14. struct cfd_node{
    15.     uint connfd;
    16.     struct cfd_node* next;
    17. };
    18. struct cfd_node* head=NULL;
    20. int main(int argc,char* argv[]){
    21.     uint lfd,cfd,max_fd;
    22.     struct cfd_node* ptr,*preptr;
    23.     char buf[BUFSIZ];
    24.     struct sockaddr_in srv_addr,client_addr;
    25.     socklen_t client_addr_len,srv_addr_len;
    26.     char client_ip_addr[20];
    27.     lfd=Socket(AF_INET,SOCK_STREAM,0);
    28.     max_fd=lfd;
    29.     srv_addr_len=sizeof(srv_addr);
    30.     init_sockaddr_in((struct sockaddr*)&srv_addr,srv_addr_len,AF_INET,SRV_PORT,INADDR_ANY);
    32.     Bind(lfd,(struct sockaddr*)&srv_addr,srv_addr_len);
    33.     int opt=1;
    34.     setsockopt(lfd,SOL_SOCKET,SO_REUSEPORT,&opt,sizeof(opt)); /* 多路复用 */
    35.     Listen(lfd,LISTEN_BACKLOG);
    37.     fd_set rset,allset;  /* 读集合，备份集合 */
    38.     FD_ZERO(&allset);       
    39.     FD_SET(lfd,&allset);
    40.     while(1){
    41.         rset=allset;
    42.         int sel_ret=Select(max_fd+1,&rset,NULL,NULL,NULL);
    43.         printf("Listen success! current fd num:%d\n",sel_ret);
    44.         if(sel_ret>0){
    45.             if(FD_ISSET(lfd,&rset)){
    46.                 client_addr_len=sizeof(client_addr);
    47.                 cfd=Accept(lfd,(struct sockaddr*)&client_addr,&client_addr_len);
    48.                 FD_SET(cfd,&allset);
    49.                 /* 头插法 */
    50.                 struct cfd_node* node=malloc(sizeof(struct cfd_node));
    51.                 node->connfd=cfd;
    52.                 node->next=head;
    53.                 head=node;
    55.                 printf("new client coming! ip: %s port %d \n",
    56.         inet_ntop(AF_INET,&client_addr.sin_addr.s_addr,client_ip_addr,sizeof(client_ip_addr))
    57.         ,ntohs(client_addr.sin_port));
    58.                 max_fd=(cfd>max_fd?cfd:max_fd);
    59.                 if(sel_ret==1) continue;
    60.             }
    61. #ifdef SIMPLE
    62.             /* 数据处理-轮询 */
    64.             for(int connfd=lfd+1;connfd<max_fd+1;connfd++){
    65.                 printf("work!\t");
    66.                 if(FD_ISSET(connfd,&rset)){
    67.                     int n=Read(connfd,buf,sizeof(buf));
    68.                     if( n == 0){
    69.                         printf("connect close!\n");
    70.                         Close(connfd);
    71.                         FD_CLR(connfd,&allset);
    72.                     }else if(n>0){
    73.                         printf("%d bytes data coming!\n",n);
    74.                         Write(STDOUT_FILENO,buf,n);
    75.                         for(int j=0;j<n;j++){
    76.                             buf[j]=toupper(buf[j]);
    77.                         }
    78.                         Write(connfd,buf,n);
    79.                     }
    80.                 }
    81.             }
    82. #else
    83.             /* 链表轮询 */
    84.             ptr=head;
    85.             preptr=NULL;
    86.             while(ptr!=NULL){
    87.                 int connfd=ptr->connfd;
    88.                 if(FD_ISSET(connfd,&rset)){
    89.                     int n=Read(connfd,buf,sizeof(buf));
    90.                     if( n == 0){
    91.                         printf("connect close!\n");
    92.                         Close(connfd);
    93.                         FD_CLR(connfd,&allset);
    94.                         /* 链表中删除失效socket */
    95.                         if(preptr==NULL){//头结点
    96.                             head=ptr->next;
    97.                         }
    98.                         else if(ptr->next!=NULL){
    99.                             preptr->next=ptr->next;
    100.                         }
    101.                         free(ptr);
    102.                     }else if(n>0){
    103.                         Write(STDOUT_FILENO,buf,n);
    104.                         for(int j=0;j<n;j++){
    105.                             buf[j]=toupper(buf[j]);
    106.                         }
    107.                         Write(connfd,buf,n);
    108.                     }
    109.                 }
    110.                 preptr=ptr;
    111.                 ptr=ptr->next;
    112.             }
    113. #endif
    114.         }
    115.     }
    116.     return 0;
    117. }