-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathEpoll.cpp
80 lines (69 loc) · 2.02 KB
/
Epoll.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
//
// Created by 裴沛东 on 2021/12/27.
//
#include <sys/epoll.h>
#include <vector>
#include <unistd.h>
#include <cstring>
#include <memory>
#include <cstdio>
#include "util.h"
#include "Epoll.h"
Epoll::Epoll():epollfd(-1),events(nullptr){
epollfd = epoll_create(1);
errif(epollfd == -1,"epoll_create failed");
events = new epoll_event[MAX_EVENTS];
bzero(events, sizeof(epoll_event) * MAX_EVENTS);
ActiveEvents.reserve(MAX_EVENTS);
}
void Epoll::addFd(int fd, uint32_t option) {
struct epoll_event ev;
bzero(&ev, sizeof(ev));
ev.data.fd = fd;
ev.events = option;
errif(epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &ev) == -1, "epoll add event error");
}
Epoll::~Epoll() {
if(epollfd != -1) {
close(epollfd);
epollfd = -1;
}
delete[] events;
}
bool Epoll::poll(int timeout) {
int nfds = epoll_wait(epollfd, events, MAX_EVENTS, timeout);
errif(nfds == -1, "epoll wait error");
for(int i = 0; i < nfds; ++i){
Channel *channel = (Channel*)events[i].data.ptr;
ActiveEvents.emplace_back(channel);
}
return true;
}
std::vector<Channel*> & Epoll::getActiveEvents() {
return ActiveEvents;
}
void Epoll::updateChannel(Channel *channel) {
int fd = channel->getFd();
struct epoll_event event;
bzero(&event,sizeof(event));
event.data.ptr = channel;
event.events = channel->getEvents();
//如果不在树上
if(!channel->getInpoll()) {
errif(epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&event) == -1,"epoll add event failed");
channel->setInpoll();
} else {
errif(epoll_ctl(epollfd,EPOLL_CTL_MOD,fd,&event) == -1,"epoll modify event failed");
}
}
void Epoll::deleteChannel(Channel *channel) {
struct epoll_event event;
bzero(&event,sizeof(event));
event.data.ptr = channel;
event.events = channel->getEvents();
if(channel->getInpoll()) {
errif(epoll_ctl(epollfd,EPOLL_CTL_DEL,channel->getFd(),&event) == -1,"epoll delete event failed");
}
}
void Epoll::modifyChannel(Channel *channel) {
}