Socket (三) : select poll epoll

Socket demo



一、select

1.原理

  select是Linux系统提供的一种服务,无需使用循环不断询问或因等待IO而导致进程被阻塞
  select就是将我们想要监听的sockfd交给select,而由系统来管理select,如果sockfd有事件产生,则select则通知程序。
  select使用一个结构体来管理sockfd,其内有一个bitmap,向select添加sockfd也就是将bitmap对应值进行更改为1。同理,当有事件发生时,系统返回的bitmap中如果值为1则代表此sockfd有事件 发生。由于select返回的是一个bitmap而不是发生事件的集合,因此还需要遍历整个bitmap找出有事件发生的sockfd,导致其效率很低。
  select只有水平触发,监听的最大值为1024,也可更改

2.常用函数

void FD_CLR(int fd, fd_set *set);//清除某一个被监视的文件描述符。
int  FD_ISSET(int fd, fd_set *set);//测试一个文件描述符是否是集合中的一员
void FD_SET(int fd, fd_set *set);//添加一个文件描述符,将set中的某一位设置成1;
void FD_ZERO(fd_set *set);//清空集合中的文件描述符,将每一位都设置为0;
int select(int maxfd, fd_set* readset, fd_set* writeset, fd_set* exceptset, 
const struct timeval* timeout);返回发生事件的sockfd的个数

3.代码

#include "TcpServer.h"
using namespace MySocket;

const int CLIENTSIZE = 100;

int main(int argc, char *argv[])
{
    if (argc != 2)
        return -1;

    TcpServer server;
    server.InitServer(atoi(argv[1]));

    int maxfd = server.listenfd;
    int client[CLIENTSIZE];

    for (int i = 0; i < CLIENTSIZE; ++i)
    {
        client[i] = -1;
    }
    fd_set allset;//
    FD_ZERO(&allset);//将allset都清空
    FD_SET(server.listenfd, &allset);//将listenfd设置为监听

    while (1)
    {
        fd_set tmpset = allset;//这里复制的原因是:返回发生事件的bitmap将原来的bitmap覆盖了
        int ret = select(maxfd + 1, &tmpset, NULL, NULL, NULL);//返回发生事件的sockfd的个数

        if (ret < 0)
        {
            printf("select failed..\n");
            break;
        }
        else if (ret == 0)
        {
            printf("select timeout..\n");
            continue;
        }
        for (int i = 0; i <= maxfd; ++i)
        {
            // printf("%d %d\n", i, maxfd);
            if (!FD_ISSET(i, &tmpset))
                continue;
            if (i == server.listenfd)
            {
                printf("waiting for conneting...\n");
                struct sockaddr_in cliaddr;
                socklen_t clilen = sizeof(cliaddr);
                if (server.Accept() == false)
                {
                    printf("connect failed!\n");
                    return -1;
                }
                int confd = server.clientfd;
                if (confd > CLIENTSIZE)
                {
                    printf("too many connection...\n");
                    return -1;
                }
                printf("sock=%d connet success.\n\n", confd);
                client[confd] = confd;
                FD_SET(confd, &allset);
                if (confd > maxfd)
                {
                    maxfd = confd;
                }
            }
            else
            {
                server.clientfd = i;
                char buf[1000];
                printf("dealing client sockfd..\n");
                if (server.Read(buf, 0))
                {
                    printf("%s\n", buf);
                }
                else
                {
                    printf("socket=%d disconnect\n", i);
                    server.CloseClient();
                    FD_CLR(i, &allset);
                }
            }
        }
    }

    return 0;
}

二、poll

1.原理

  相比select,poll变化不大,原理也相同,poll使用数组来取代bitmap,因而也就没有大小限制。同样,poll也需要遍历整个数组来找出有事件发生的sockfd,其也只有水平触发。

2.常用函数

poll结构体
struct pollfd{
	int fd;			//文件描述符
	short events;	//等待的事件
	short revents;	//实际发生的事件
};
int poll(struct pollfd *fds, nfds_t nfds, int timeout);监视并等待多个文件描述符的属性变化

3.代码

#include "TcpServer.h"
#include "poll.h"
using namespace MySocket;

const int CLIENTSIZE = 100;
const int FDSIZE=1024;

int main(int argc, char *argv[])
{
    if (argc != 2)
        return -1;

    TcpServer server;
    server.InitServer(atoi(argv[1]));


    int client[CLIENTSIZE];
    for (int i = 0; i < CLIENTSIZE; ++i)
    {
        client[i] = -1;
    }

    //poll设置
    struct pollfd fds[FDSIZE];
    for(int i=0;i<FDSIZE;++i){
        fds[i].fd=-1;
    }
    fds[server.listenfd].fd=server.listenfd;
    fds[server.listenfd].events=POLLIN;
    int maxfd = server.listenfd;

    while (1)
    {
        int ret=poll(fds,maxfd+1,-1);
        if (ret < 0)
        {
            printf("poll failed..\n");
            break;
        }
        else if (ret == 0)
        {
            printf("poll timeout..\n");
            continue;
        }
        for (int i = 0; i <= maxfd; ++i)
        {
            // printf("%d %d\n", i, maxfd);
            if(fds[i].fd<0)continue;
            if((fds[i].revents&POLLIN)==0) continue;
            if (i == server.listenfd)
            {
                printf("waiting for conneting...\n");
                struct sockaddr_in cliaddr;
                socklen_t clilen = sizeof(cliaddr);
                if (server.Accept() == false)
                {
                    printf("connect failed!\n");
                    return -1;
                }
                int confd = server.clientfd;
                if (confd > CLIENTSIZE)
                {
                    printf("too many connection...\n");
                    return -1;
                }
                printf("sock=%d connet success.\n\n", confd);
                client[confd] = confd;
                fds[confd].fd=confd;
                fds[confd].events=POLLIN;
                fds[confd].revents=0;
                if (confd > maxfd)
                {
                    maxfd = confd;
                }
            }
            else
            {
                server.clientfd = i;
                char buf[1000];
                printf("deal client sockfd..\n");
                if (server.Read(buf, 0))
                {
                    printf("%s\n", buf);
                }
                else
                {
                    printf("socket=%d disconnect\n", i);
                    server.CloseClient();
                    fds[i].fd=-1;
                }
            }
        }
    }

    return 0;
}

三、epoll

1.原理

  select和poll效率都比较低,其原因就是每次都需要遍历才能找到发生事件的sockfd。epoll的一个巨大改进之处在于将发生事件的sockfd使用一个数据进行保存,提高了效率。epoll具有水平触发和边缘触发两种触发模式。

2.常用函数

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 */
};

int epoll_create(int size);//创建一个epoll的句柄。自从linux2.6.8之后,size参数是被忽略的。
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。第四个参数是告诉内核需要监听什么事
int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);//收集在epoll监控的事件中已经发送的事件。参数events是分配好的epoll_event结构体数组,epoll将会把发生的事件赋值到events数组中

3. 代码

#include "TcpServer.h"
#include "sys/epoll.h"
using namespace MySocket;

const int CLIENTSIZE = 100;
const int EVENTSIZE = 100;

int main(int argc, char *argv[])
{
    if (argc != 2)
        return -1;

    TcpServer server;
    server.InitServer(atoi(argv[1]));

    int client[CLIENTSIZE];

    for (int i = 0; i < CLIENTSIZE; ++i)
    {
        client[i] = -1;
    }

    int epollfd = epoll_create(1); // linux2.6.8版本后此值被忽略,只要大于0即可
    struct epoll_event ev;
    ev.data.fd = server.listenfd;
    ev.events = EPOLLIN;
    epoll_ctl(epollfd, EPOLL_CTL_ADD, server.listenfd, &ev);
    int maxfd = server.listenfd;

    while (1)
    {
        epoll_event events[EVENTSIZE];
        int ret = epoll_wait(epollfd, events, EVENTSIZE, -1);//-1:忽略超时

        if (ret < 0)
        {
            printf("epoll failed..\n");
            break;
        }
        else if (ret == 0)
        {
            printf("epoll timeout..\n");
            continue;
        }
        for (int i = 0; i < ret; ++i)
        {
            // printf("%d %d\n", i, maxfd);
            if ((events[i].data.fd == server.listenfd) && (events[i].events & EPOLLIN))
            {
                printf("waiting for conneting...\n");
                struct sockaddr_in cliaddr;
                socklen_t clilen = sizeof(cliaddr);
                if (server.Accept() == false)
                {
                    printf("connect failed!\n");
                    return -1;
                }
                int confd = server.clientfd;
                if (confd > CLIENTSIZE)
                {
                    printf("too many connection...\n");
                    return -1;
                }
                printf("sock=%d connet success.\n\n", confd);
                client[confd] = confd;

                memcpy(&ev, 0, sizeof(ev));
                ev.data.fd = confd;
                ev.events = EPOLLIN;
                epoll_ctl(epollfd, EPOLL_CTL_ADD, confd, &ev);
                if (confd > maxfd)
                {
                    maxfd = confd;
                }
            }
            else
            {
                server.clientfd = i;
                char buf[1000];
                printf("dealing client sockfd..\n");
                if (server.Read(buf, 0))
                {
                    printf("%s\n", buf);
                }
                else
                {
                    printf("socket=%d disconnect\n", i);
                    server.CloseClient();
                    memcpy(&ev, 0, sizeof(ev));
                    ev.data.fd = events[i].data.fd;
                    ev.events = EPOLLIN;
                    epoll_ctl(epollfd,EPOLL_CTL_DEL,events[i].data.fd,&ev);
                    close(events[i].data.fd);
                }
            }
        }
    }

    return 0;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值