tcp client epoll 多socket 复用

最新推荐文章于 2024-09-19 21:05:19 发布

ChangeMe_Q

最新推荐文章于 2024-09-19 21:05:19 发布

阅读量1.1k

点赞数

分类专栏： c

本文链接：https://blog.csdn.net/weixin_39227749/article/details/80883677

版权

c 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

tcp client epoll 多socket 复用

#include <stdio.h>    
#include <sys/types.h>    
#include <sys/socket.h>    
#include <netinet/in.h>    
#include <arpa/inet.h>
#include <sys/epoll.h> 
#include <errno.h>     
#include <string.h>  
#include <stdlib.h> 
#include <fcntl.h> 

#define EPOLLEVENTS 20
#define MAXSIZE 1024

static void add_event(int epollfd,int fd,int state)
{
    struct epoll_event ev;
    ev.events = state;
    ev.data.fd = fd;
    epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&ev);
}

static void delete_event(int epollfd,int fd,int state)
{
    struct epoll_event ev;
    ev.events = state;
    ev.data.fd = fd;
    epoll_ctl(epollfd,EPOLL_CTL_DEL,fd,&ev);
}

static void handle_events(int epollfd,struct epoll_event *events,int num)
{
    int fd;
    int i;
    static int j = 0;
    static int k = 0;
    char ip[32] = {0};
    for (i = 0;i < num;i++)
    {
        k++;
        fd = events[i].data.fd;
        if (events[i].events & EPOLLIN){
            printf("do_read : %d\n", k);
        }
            //do_read(epollfd,fd,sockfd,buf);
        else if (events[i].events & EPOLLOUT){
            printf("do_write : %d\n", k);
            //do_write(epollfd,fd,sockfd,buf);
        }else{
            printf("other op : %d\n", k);
        }

        struct sockaddr_in sa;
        int len = sizeof(sa);
        if(!getpeername(fd, (struct sockaddr *)&sa, &len))
        {
            j++;
            printf( "==ip%s %d\n", inet_ntoa(sa.sin_addr),j);
        }

        delete_event(epollfd,fd,EPOLLIN | EPOLLOUT | EPOLLET);
        close(fd);
    }
}

int init_connect(int epollfd, char *ip, int port)
{
    int satate;   
    int client_sockfd;     
    int len;     
    struct sockaddr_in remote_addr; // 服务器端网络地址结构体     
    memset(&remote_addr,0,sizeof(remote_addr)); // 数据初始化--清零     
    remote_addr.sin_family=AF_INET; // 设置为IP通信     
    remote_addr.sin_addr.s_addr=inet_addr(ip);// 服务器IP地址     
    remote_addr.sin_port=htons(port); // 服务器端口号     
    // 创建客户端套接字--IPv4协议，面向连接通信，TCP协议   
    if((client_sockfd=socket(PF_INET,SOCK_STREAM,0))<0)     
    {     
        perror("client socket creation failed");     
        return -1; 
    } 
    //设置非阻塞socket
    int flags;
    flags = fcntl(client_sockfd, F_GETFL, NULL);
    fcntl(client_sockfd, F_SETFL, flags | O_NONBLOCK);

    // 将套接字绑定到服务器的网络地址上   
    connect(client_sockfd,(struct sockaddr *)&remote_addr,sizeof(struct sockaddr));
    if (errno != EINPROGRESS)
    {
        close(client_sockfd);
        perror("connect..2");
        return -1;
    }

    add_event(epollfd,client_sockfd,EPOLLIN | EPOLLOUT | EPOLLET);

    return 0;
}

int main(int argc, char *argv[])     
{  

    int epollfd = 0;
    struct epoll_event events[EPOLLEVENTS];
    epollfd = epoll_create(MAXSIZE);

    char ip[32] = {0};
    int i;
    for(i = 0; i < 254; ++i){
        memset(ip, 0 ,32);
        sprintf(ip,"192.168.6.%d",i);
        init_connect(epollfd, ip, 22);
    }

    int ret;
    while(1){
        //创建epoll
        ret = epoll_wait(epollfd,events,EPOLLEVENTS,-1);
        handle_events(epollfd,events,ret);
    }

    return 0;  
}

epoll 使用说明

在linux的网络编程中，很长的时间都在使用select来做事件触发。在linux新的内核中，有了一种替换它的机制，就是epoll。
相比于select，epoll最大的好处在于它不会随着监听fd数目的增长而降低效率。因为在内核中的select实现中，它是采用轮询来处理的，轮询的fd数目越多，自然耗时越多。并且，在linux/posix_types.h头文件有这样的声明：
#define __FD_SETSIZE 1024
表示select最多同时监听1024个fd，当然，可以通过修改头文件再重编译内核来扩大这个数目，但这似乎并不治本。

epoll的接口非常简单，一共就三个函数：
1. int epoll_create(int size);
创建一个epoll的句柄，size用来告诉内核这个监听的数目一共有多大。这个参数不同于select()中的第一个参数，给出最大监听的fd+1的值。需要注意的是，当创建好epoll句柄后，它就是会占用一个fd值，在linux下如果查看/proc/进程id/fd/，是能够看到这个fd的，所以在使用完epoll后，必须调用close()关闭，否则可能导致fd被耗尽。

int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);
epoll的事件注册函数，它不同与select()是在监听事件时告诉内核要监听什么类型的事件，而是在这里先注册要监听的事件类型。第一个参数是epoll_create()的返回值，第二个参数表示动作，用三个宏来表示：
EPOLL_CTL_ADD：注册新的fd到epfd中；
EPOLL_CTL_MOD：修改已经注册的fd的监听事件；
EPOLL_CTL_DEL：从epfd中删除一个fd；
第三个参数是需要监听的fd，第四个参数是告诉内核需要监听什么事，struct epoll_event结构如下：

typedef union epoll_data {
void *ptr;
int fd;
__uint32_t u32;
__uint64_t u64;
} epoll_data_t;

struct epoll_event {
__uint32_t events; /* Epoll events */
epoll_data_t data; /* User data variable */
};

events可以是以下几个宏的集合：
EPOLLIN ：表示对应的文件描述符可以读（包括对端SOCKET正常关闭）；
EPOLLOUT：表示对应的文件描述符可以写；
EPOLLPRI：表示对应的文件描述符有紧急的数据可读（这里应该表示有带外数据到来）；
EPOLLERR：表示对应的文件描述符发生错误；
EPOLLHUP：表示对应的文件描述符被挂断；
EPOLLET：将EPOLL设为边缘触发(Edge Triggered)模式，这是相对于水平触发(Level Triggered)来说的。
EPOLLONESHOT：只监听一次事件，当监听完这次事件之后，如果还需要继续监听这个socket的话，需要再次把这个socket加入到EPOLL队列里

int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);
等待事件的产生，类似于select()调用。参数events用来从内核得到事件的集合，maxevents告之内核这个events有多大，这个 maxevents的值不能大于创建epoll_create()时的size，参数timeout是超时时间（毫秒，0会立即返回，-1将不确定，也有说法说是永久阻塞）。该函数返回需要处理的事件数目，如返回0表示已超时。

4、关于ET、LT两种工作模式：
可以得出这样的结论:
ET模式仅当状态发生变化的时候才获得通知,这里所谓的状态的变化并不包括缓冲区中还有未处理的数据,也就是说,如果要采用ET模式,需要一直read/write直到出错为止,很多人反映为什么采用ET模式只接收了一部分数据就再也得不到通知了,大多因为这样;而LT模式是只要有数据没有处理就会一直通知下去的.

那么究竟如何来使用epoll呢？其实非常简单。
通过在包含一个头文件#include