I/O复用 select函数简单示例-CSDN博客

本文链接：https://blog.csdn.net/sono_io/article/details/132364535

本文详细介绍了使用C语言实现的一个基础服务端程序，涉及socket编程、bind、listen、accept和select等关键操作，展示了如何处理新连接和接收客户端消息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

服务端

#include <stdio.h>
#include <unistd.h>
#include <sys/time.h>
#include <sys/select.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>

#define BUFSIZE 30

void handingError(char *msg)
{
    fputs(msg, stderr);
    fputc('\n', stderr);
    exit(2);
}

int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        handingError("argc error");
    }
    char buff[BUFSIZE];
    int servSocket, clntSocket;
    struct sockaddr_in servAddr, clntAddr;
    servSocket = socket(PF_INET, SOCK_STREAM, 0);
    memset(&servAddr, 0, sizeof(servAddr));
    servAddr.sin_family = PF_INET;
    servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servAddr.sin_port = htons(atoi(argv[1]));
    if (bind(servSocket, (struct sockaddr *)&servAddr, sizeof(servAddr)) == -1)
    {
        handingError("bind error");
    }
    if (listen(servSocket, 5) == -1)
    {
        handingError("listen error");
    }
    int i;
    fd_set reads, cpy_reads;
    int fd_max, fd_num;
    FD_ZERO(&reads);
    FD_SET(servSocket, &reads);
    fd_max = servSocket;
    struct timeval timeOut;
    int str_len;
    while (1)
    {
        cpy_reads = reads;
        timeOut.tv_sec = 5;
        timeOut.tv_usec = 0;
        if ((fd_num = select(fd_max + 1, &cpy_reads, 0, 0, &timeOut)) == -1)
        {
            break;
        }
        if (fd_num == 0)
        {
            printf("error, fd_num\n");

            continue;
        }
        for (i = 0; i < fd_max + 1; i++)
        {
            if (FD_ISSET(i, &cpy_reads))
            {
                if (i == servSocket) // new connect
                {
                    socklen_t sLen = sizeof(clntAddr);
                    clntSocket = accept(servSocket, (struct sockaddr *)&clntAddr, &sLen);
                    FD_SET(clntSocket, &reads);
                    if (fd_max < clntSocket)
                    {
                        fd_max = clntSocket;
                    }
                    printf("new connect: %d\n", clntSocket);
                }
                else // read msg
                {
                    str_len = read(i, buff, BUFSIZE);
                    if (str_len == 0)
                    {
                        FD_CLR(i, &reads);
                        close(i);
                        printf("close client: %d\n", i);
                    }
                    else
                    {
                        write(i, buff, str_len);
                    }
                }
            }
        }
    }
    close(servSocket);
    return 0;
}