linux poll 使用简单例子

本文介绍如何使用poll()函数实现服务器端监听多个客户端连接,代码简洁且效率高,适用于需要同时处理多个并发连接的场景。通过示例代码演示了创建socket、绑定地址、监听及接受客户端连接的过程。

         可以监听多个客户端,代码比select简洁。直接上服务端代码,测试可以用telnet。 

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <assert.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <poll.h>

#define MAX_FD_NUM 20
#define MAXLEN 1024

int main(int argc,char* argv[])
{
    printf("server start up\n");

    if(argc <= 2)
    {
        printf("usage:%s ip port\n",basename(argv[0]));
        return 1;
    }

    //IP地址
    const char* ip = argv[1];
    //端口号
    int port = atoi(argv[2]);
    //内核监听队列的最大长度(完全连接的socket)
    //int backlog = atoi(argv[3]);

    //创建socket	(TIP/IP协议族,流式socket)
    int server_sockfd = socket(PF_INET,SOCK_STREAM,0);

    //TCP/IP协议族的socket地址结构体
    struct sockaddr_in server_addr;
    bzero(&server_addr,sizeof(server_addr));
    server_addr.sin_family = AF_INET;			//TCP/IPv4的地址族
    inet_pton(AF_INET,ip,&server_addr.sin_addr);		//将IP地址字符串转换为二进制的整数并赋给addr.sin_server_addr
    server_addr.sin_port = htons(port);			//端口,host to net,将主机字节序(小端)转换为网络字节序(大端)

    //将文件描述符sock和socket地址关联,仅服务端需要,客户端自动绑定地址
    //注意需要强制转换为 struct sockaddr*
    int ret = bind(server_sockfd,(struct sockaddr*)&server_addr,sizeof(server_addr));
    assert(ret != -1);

    //监听
    ret = listen(server_sockfd,MAX_FD_NUM-1);
    assert(ret != -1);

    //等待客户端做些连接等相关工作
    sleep(3);

    //客户端地址信息
    struct sockaddr_in client_addr;
    socklen_t client_addr_len = sizeof(struct sockaddr_in);

    //poll fds
    struct pollfd pollfdArry[MAX_FD_NUM];
    for(int i=0;i<MAX_FD_NUM;++i)
    {
        pollfdArry[i].fd = -1;
    }

    //insert the server socket fd
    pollfdArry[0].fd = server_sockfd;
    pollfdArry[0].events = POLLIN;

    int cur_fd_num = 1;
    char buf[MAXLEN]={0};

    //
    while (1)
    {
        int nready = poll(pollfdArry,cur_fd_num,-1);

        //server socket fd
        if(pollfdArry[0].revents & POLLIN)
        {
            if(cur_fd_num > MAX_FD_NUM)
            {
                printf("socket num to much\n");
            }
            else
            {
                //接受连接,并将被接受的远端sock地址信息保存在第二个参数中
                //只是从监听队列中取出连接,即使客户端已经断开网络连接也会accept成功
                int client_sockfd = accept(server_sockfd,(struct sockaddr*)&client_addr,&client_addr_len);

                if(client_sockfd < 0)
                {
                   perror("accept");
                }
                else
                {
                    //inet_ntoa(struct addr_in) 将IP地址转换为字符串并返回
                    printf("accept client_addr %s\n",inet_ntoa(client_addr.sin_addr));

                    for(int i=0;i<MAX_FD_NUM;++i)
                    {
                        if(pollfdArry[i].fd == -1)
                        {
                            pollfdArry[i].fd = client_sockfd;
                            pollfdArry[i].events = POLLIN;
                            cur_fd_num++;
                            break;
                        }

                    }
                }

            }

            if(--nready <= 0)
            {
                continue;
            }
        }

        for(int i=0;i<MAX_FD_NUM;++i)
        {
            if(pollfdArry[i].fd < 0)
            {
                continue;
            }

            if(pollfdArry[i].revents & (POLLIN |POLLERR))
            {

                int n = recv(pollfdArry[i].fd,buf,MAXLEN,0);

                if(n < 0)
                {
                    if(ECONNRESET == errno)
                    {
                        close(pollfdArry[i].fd);
                        pollfdArry[i].fd = -1;
                        cur_fd_num--;
                    }
                    else
                    {
                        perror("recv");
                    }
                }
                else if(n == 0)
                {
                    close(pollfdArry[i].fd);
                    pollfdArry[i].fd = -1;
                    cur_fd_num--;
                }
                else
                {
                    printf("recived:%s\n",buf);
                }

            }

            if(--nready)
            {
                break;
            }
        }
    }


    //关闭连接,实际只是socket的引用-1,必须引用为0才会真正关闭
    for(int i=0;i<MAX_FD_NUM;++i)
    {
        if(pollfdArry[i].fd != -1)
        {
            close(pollfdArry[i].fd);
        }
    }

    return 0;
}

 

<think>我们正在回答用户关于“Linuxepoll简单使用示例代码”的请求。根据引用,我们知道epollLinux下高效的I/O多路复用机制。引用中提到了epoll的基本使用方法,包括创建epoll实例、添加事件等。我们还可以参考引用中的线程池示例,但用户主要需要简单的epoll示例,所以我们以基础使用为主。我们将提供一个简单的epoll使用示例,包括:1.创建epoll实例2.添加需要监视的文件描述符(如socket)和事件3.进入循环等待事件发生4.处理事件(如可读、可写等)注意:本示例以TCP服务器为例,监听套接字,接受连接,并使用epoll管理多个连接。代码结构如下:-创建socket,绑定,监听-创建epoll实例-将监听socket添加到epoll-进入事件循环-当事件发生时,如果是监听socket则接受新连接并加入epoll,如果是客户端连接则处理数据由于用户要求简单示例,我们只实现基本框架,不涉及复杂的错误处理和资源释放。</think> 以下是 Linux `epoll` 的简单使用示例代码及分步解释: ### 基础使用步骤 1. **创建 epoll 实例** ```c int epoll_fd = epoll_create1(0); if (epoll_fd == -1) { perror("epoll_create1"); exit(EXIT_FAILURE); } ``` 2. **添加监听事件**(以 TCP 服务器为例) ```c struct epoll_event ev; ev.events = EPOLLIN; // 监听可读事件 ev.data.fd = server_socket; // 服务器套接字 if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, server_socket, &ev) == -1) { perror("epoll_ctl"); exit(EXIT_FAILURE); } ``` 3. **事件循环处理** ```c #define MAX_EVENTS 10 struct epoll_event events[MAX_EVENTS]; while (1) { int nfds = epoll_wait(epoll_fd, events, MAX_EVENTS, -1); // 阻塞等待 if (nfds == -1) { perror("epoll_wait"); break; } for (int i = 0; i < nfds; i++) { if (events[i].data.fd == server_socket) { // 处理新连接 accept_connection(server_socket, epoll_fd); } else { // 处理客户端数据 handle_client(events[i].data.fd); } } } ``` ### 完整示例代码 ```c #include <sys/epoll.h> #include <sys/socket.h> #include <netinet/in.h> #include <unistd.h> #include <fcntl.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #define PORT 8080 #define MAX_EVENTS 10 void setnonblocking(int sockfd) { int flags = fcntl(sockfd, F_GETFL, 0); fcntl(sockfd, F_SETFL, flags | O_NONBLOCK); } void handle_client(int client_fd) { char buffer[1024]; ssize_t bytes_read = read(client_fd, buffer, sizeof(buffer)-1); if (bytes_read > 0) { buffer[bytes_read] = '\0'; printf("Received: %s\n", buffer); write(client_fd, "HTTP/1.1 200 OK\r\n\r\nHello", 25); } close(client_fd); // 简化处理:响应后立即关闭 } int main() { int server_fd = socket(AF_INET, SOCK_STREAM, 0); struct sockaddr_in addr = { .sin_family = AF_INET, .sin_addr.s_addr = INADDR_ANY, .sin_port = htons(PORT) }; bind(server_fd, (struct sockaddr*)&addr, sizeof(addr)); listen(server_fd, SOMAXCONN); setnonblocking(server_fd); // 非阻塞模式 int epoll_fd = epoll_create1(0); struct epoll_event ev; ev.events = EPOLLIN | EPOLLET; // 边缘触发模式 ev.data.fd = server_fd; epoll_ctl(epoll_fd, EPOLL_CTL_ADD, server_fd, &ev); while (1) { struct epoll_event events[MAX_EVENTS]; int nfds = epoll_wait(epoll_fd, events, MAX_EVENTS, -1); for (int i = 0; i < nfds; i++) { if (events[i].data.fd == server_fd) { // 接受所有新连接 while (1) { struct sockaddr_in client_addr; socklen_t addr_len = sizeof(client_addr); int client_fd = accept(server_fd, (struct sockaddr*)&client_addr, &addr_len); if (client_fd < 0) break; setnonblocking(client_fd); ev.events = EPOLLIN | EPOLLET; ev.data.fd = client_fd; epoll_ctl(epoll_fd, EPOLL_CTL_ADD, client_fd, &ev); } } else { handle_client(events[i].data.fd); } } } close(epoll_fd); close(server_fd); return 0; } ``` ### 关键说明 1. **边缘触发(EPOLLET)** - 事件仅通知一次,需一次性处理完所有数据 - 需搭配非阻塞套接字,避免 `read()` 阻塞 2. **水平触发(默认)** - 删除 `EPOLLET` 标志即可 - 只要缓冲区有数据会持续通知 3. **性能优势** `epoll` 避免了像 `select/poll` 的 O(n) 遍历,时间复杂度为 O(1) 级别,特别适合高并发场景[^2]。
评论 3
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值