Linux IO复用之select

最新推荐文章于 2025-04-22 15:41:37 发布

原创最新推荐文章于 2025-04-22 15:41:37 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

网络编程专栏收录该内容

23 篇文章

订阅专栏

本文介绍了一个使用C语言实现的简易TCP服务器程序，该程序利用select进行多客户端连接的监听和处理。通过创建socket并绑定地址，服务器可以接受多个客户端连接，并能够读取客户端发送的数据，同时将接收到的数据打印出来并回传给客户端。

select：

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/epoll.h>
#include <sys/socket.h>
#include <netinet/in.h>

int main(int argc, char*argv[])
{
        int sockfd = socket(AF_INET, SOCK_STREAM, 0);

        struct sockaddr_in addrSrv;
        bzero(&addrSrv, sizeof(struct sockaddr_in));
        addrSrv.sin_addr.s_addr = htonl(INADDR_ANY);
        addrSrv.sin_family = AF_INET;
        addrSrv.sin_port = htons(6000);
        bind(sockfd, (struct sockaddr*)&addrSrv, sizeof(struct sockaddr));

        listen(sockfd, 3);

        struct sockaddr_in addrConn;
        socklen_t len = sizeof(struct sockaddr_in);

        fd_set allset, rset;
        FD_ZERO(&allset);
        FD_SET(sockfd, &allset);

        int maxfd = sockfd;

        int client[FD_SETSIZE];
        int i;
        for (i=0;i<FD_SETSIZE;i++)
                client[i] = -1;

        int maxi = -1;

        int sockConn;
        while (1)
        {
                rset = allset;
                int count = select(maxfd+1, &rset, NULL, NULL, NULL);
                if (FD_ISSET(sockfd, &rset))
                {
                        sockConn = accept(sockfd, (struct sockaddr*)&addrConn, &len);
                        for (i=0;i<FD_SETSIZE; i++)
                                if (client[i]<0)
                                {
                                        client[i] = sockConn;
                                        break;
                                }

                        if (maxi < i)
                                maxi = i;

                        FD_SET(sockConn, &allset);

                        if (maxfd < sockConn)
                                maxfd = sockConn;
                }

                for (i=0;i<=maxi;i++)
                {
                        if ((sockConn = client[i]) < -1)
                                continue;

                        if (FD_ISSET(sockConn, &rset))
                        {
                                char buf[1024] = {0};
                                int n = read(sockConn, buf, 1024);
                                if (n == 0)
                                {
                                        close(sockConn);
                                        client[i] = -1;
                                        FD_CLR(sockConn, &allset);
                                }
                                else
                                {
                                        printf("server recv:%s\n", buf);
                                        write(sockConn, "server data", strlen("server data"));
                                }
                        }
                }
        }

        close(sockfd);

        return 0;
}