1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
2 // vim: ts=8 sw=2 smarttab
4 * Ceph - scalable distributed file system
6 * Copyright (C) 2014 UnitedStack <haomai@unitedstack.com>
8 * Author: Haomai Wang <haomaiwang@gmail.com>
10 * This is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License version 2.1, as published by the Free Software
13 * Foundation. See file COPYING.
17 #include "common/errno.h"
18 #include "EventEpoll.h"
20 #define dout_subsys ceph_subsys_ms
23 #define dout_prefix *_dout << "EpollDriver."
25 int EpollDriver::init(EventCenter *c, int nevent)
27 events = (struct epoll_event*)malloc(sizeof(struct epoll_event)*nevent);
29 lderr(cct) << __func__ << " unable to malloc memory. " << dendl;
32 memset(events, 0, sizeof(struct epoll_event)*nevent);
34 epfd = epoll_create(1024); /* 1024 is just an hint for the kernel */
36 lderr(cct) << __func__ << " unable to do epoll_create: "
37 << cpp_strerror(errno) << dendl;
46 int EpollDriver::add_event(int fd, int cur_mask, int add_mask)
48 ldout(cct, 20) << __func__ << " add event fd=" << fd << " cur_mask=" << cur_mask
49 << " add_mask=" << add_mask << " to " << epfd << dendl;
50 struct epoll_event ee;
51 /* If the fd was already monitored for some event, we need a MOD
52 * operation. Otherwise we need an ADD operation. */
54 op = cur_mask == EVENT_NONE ? EPOLL_CTL_ADD: EPOLL_CTL_MOD;
57 add_mask |= cur_mask; /* Merge old events */
58 if (add_mask & EVENT_READABLE)
60 if (add_mask & EVENT_WRITABLE)
61 ee.events |= EPOLLOUT;
62 ee.data.u64 = 0; /* avoid valgrind warning */
64 if (epoll_ctl(epfd, op, fd, &ee) == -1) {
65 lderr(cct) << __func__ << " epoll_ctl: add fd=" << fd << " failed. "
66 << cpp_strerror(errno) << dendl;
73 int EpollDriver::del_event(int fd, int cur_mask, int delmask)
75 ldout(cct, 20) << __func__ << " del event fd=" << fd << " cur_mask=" << cur_mask
76 << " delmask=" << delmask << " to " << epfd << dendl;
77 struct epoll_event ee;
78 int mask = cur_mask & (~delmask);
82 if (mask & EVENT_READABLE) ee.events |= EPOLLIN;
83 if (mask & EVENT_WRITABLE) ee.events |= EPOLLOUT;
84 ee.data.u64 = 0; /* avoid valgrind warning */
86 if (mask != EVENT_NONE) {
87 if ((r = epoll_ctl(epfd, EPOLL_CTL_MOD, fd, &ee)) < 0) {
88 lderr(cct) << __func__ << " epoll_ctl: modify fd=" << fd << " mask=" << mask
89 << " failed." << cpp_strerror(errno) << dendl;
93 /* Note, Kernel < 2.6.9 requires a non null event pointer even for
95 if ((r = epoll_ctl(epfd, EPOLL_CTL_DEL, fd, &ee)) < 0) {
96 lderr(cct) << __func__ << " epoll_ctl: delete fd=" << fd
97 << " failed." << cpp_strerror(errno) << dendl;
104 int EpollDriver::resize_events(int newsize)
109 int EpollDriver::event_wait(vector<FiredFileEvent> &fired_events, struct timeval *tvp)
111 int retval, numevents = 0;
113 retval = epoll_wait(epfd, events, size,
114 tvp ? (tvp->tv_sec*1000 + tvp->tv_usec/1000) : -1);
119 fired_events.resize(numevents);
120 for (j = 0; j < numevents; j++) {
122 struct epoll_event *e = events + j;
124 if (e->events & EPOLLIN) mask |= EVENT_READABLE;
125 if (e->events & EPOLLOUT) mask |= EVENT_WRITABLE;
126 if (e->events & EPOLLERR) mask |= EVENT_READABLE|EVENT_WRITABLE;
127 if (e->events & EPOLLHUP) mask |= EVENT_READABLE|EVENT_WRITABLE;
128 fired_events[j].fd = e->data.fd;
129 fired_events[j].mask = mask;