1. #include <stdio.h>
    2. #include <sys/types.h>
    3. #include <sys/socket.h>
    4. #include <netdb.h>
    5. #include <stdlib.h>
    6. #include <unistd.h>
    7. #include <arpa/inet.h>
    8. #include <netinet/in.h>
    9. #include <errno.h>
    10. #include <string.h>
    11. #include <ctype.h>
    12. #include <sys/select.h>
    14. int main()
    15. {
    16.     // 创建监听套接字
    17.     int lfd = socket(AF_INET, SOCK_STREAM, 0);
    18.     if (lfd == -1)
    19.     {
    20.         perror("socket error");
    21.         exit(1);
    22.     }
    24.     // 设置端口复用
    25.     int opt = 1;
    26.     setsockopt(lfd, SOL_SOCKET, SO_REUSEADDR, (const void *)&opt, sizeof(opt));
    28.     //IP地址与端口号信息
    29.     struct sockaddr_in serv_addr;
    30.     memset(&serv_addr, 0, sizeof(serv_addr));
    31.     serv_addr.sin_family = AF_INET;
    32.     serv_addr.sin_port = htons(9317);
    33.     serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    35.     // 绑定、监听
    36.     int ret = bind(lfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));
    37.     if (ret == -1)
    38.     {
    39.         perror("bind error");
    40.         exit(1);
    41.     }
    43.     ret = listen(lfd, 64);
    44.     if (ret == -1)
    45.     {
    46.         perror("listen error");
    47.         exit(1);
    48.     }
    50.     // 用来接收客户端ip信息
    51.     struct sockaddr_in client_addr;
    52.     socklen_t cli_addr_len = sizeof(struct sockaddr_in);
    54.     // select用的两个事件数组
    55.     fd_set reads, temps;
    56.     // 最大的事件fd
    57.     int max_fd;
    58.     // 初始化reads事件表
    59.     FD_ZERO(&reads);
    60.     // 设置内核监视监听套接字
    61.     FD_SET(lfd, &reads);
    62.     max_fd = lfd;
    64.     while (1)
    65.     {
    66.         // 复制reads表传参，保留原reads表不变，因为select会修改传进去的表
    67.         temps = reads;
    69.         int ret = select(max_fd + 1, &temps, NULL, NULL, NULL);
    70.         if (ret < 0)
    71.         {
    72.             fprintf(stderr, "select error: %s\n", strerror(errno));
    73.             exit(1);
    74.         }
    76.         // 监听套接字有事件
    77.         if (FD_ISSET(lfd, &temps))
    78.         {
    79.             int cfd;
    80.             if ((cfd = accept(lfd, (struct sockaddr *)&client_addr, &cli_addr_len)) < 0)
    81.             {
    82.                 fprintf(stderr, "accept error: %s\n", strerror(errno));
    83.                 exit(1);
    84.             }
    85.             // 新套接字（客户套接字）加入reads表
    86.             FD_SET(cfd, &reads);
    87.             max_fd = max_fd > cfd ? max_fd : cfd;
    88.         }
    89.         int k;
    90.         for (k = lfd + 1; k <= max_fd; k++)
    91.         {
    92.             // 客户端有事件，处理
    93.             if (FD_ISSET(k, &temps))
    94.             {
    95.                 char buff[1024];
    96.                 int rec_len, i;
    97.                 if ((rec_len = recv(k, buff, sizeof(buff), 0)) < 0)
    98.                 {
    99.                     fprintf(stderr, "read error: %s\n", strerror(errno));
    100.                     exit(1);
    101.                 }
    102.                 else if (rec_len == 0)
    103.                 {
    104.                     // 客户端关闭连接
    105.                     close(k);
    106.                     FD_CLR(k, &reads);
    107.                 }
    108.                 else
    109.                 {
    110.                     for (i = 0; i < rec_len; ++i)
    111.                     {
    112.                         buff[i] = toupper(buff[i]);
    113.                     }
    114.                     send(k, buff, rec_len, 0);
    115.                 }
    116.             }
    117.         }
    118.     }
    119.     close(lfd);
    120.     return 0;
    121. }