1 /* +------------------------------------+
2 * | Inspire Internet Relay Chat Daemon |
3 * +------------------------------------+
5 * InspIRCd: (C) 2002-2010 InspIRCd Development Team
6 * See: http://wiki.inspircd.org/Credits
8 * This program is free but copyrighted software; see
9 * the file COPYING for details.
11 * ---------------------------------------------------
18 #include "exitcodes.h"
19 #include "socketengine.h"
20 #include <sys/epoll.h>
24 /** A specialisation of the SocketEngine class, designed to use linux 2.6 epoll().
26 class EPollEngine : public SocketEngine
29 /** These are used by epoll() to hold socket events
31 struct epoll_event* events;
34 /** Create a new EPollEngine
37 /** Delete an EPollEngine
39 virtual ~EPollEngine();
40 virtual bool AddFd(EventHandler* eh, int event_mask);
41 virtual void OnSetEvent(EventHandler* eh, int old_mask, int new_mask);
42 virtual bool DelFd(EventHandler* eh, bool force = false);
43 virtual int DispatchEvents();
44 virtual std::string GetName();
47 EPollEngine::EPollEngine()
49 int max = ulimit(4, 0);
52 MAX_DESCRIPTORS = max;
56 ServerInstance->Logs->Log("SOCKET", DEFAULT, "ERROR: Can't determine maximum number of open sockets!");
57 printf("ERROR: Can't determine maximum number of open sockets!\n");
58 ServerInstance->Exit(EXIT_STATUS_SOCKETENGINE);
61 // This is not a maximum, just a hint at the eventual number of sockets that may be polled.
62 EngineHandle = epoll_create(GetMaxFds() / 4);
64 if (EngineHandle == -1)
66 ServerInstance->Logs->Log("SOCKET",DEFAULT, "ERROR: Could not initialize socket engine: %s", strerror(errno));
67 ServerInstance->Logs->Log("SOCKET",DEFAULT, "ERROR: Your kernel probably does not have the proper features. This is a fatal error, exiting now.");
68 printf("ERROR: Could not initialize epoll socket engine: %s\n", strerror(errno));
69 printf("ERROR: Your kernel probably does not have the proper features. This is a fatal error, exiting now.\n");
70 ServerInstance->Exit(EXIT_STATUS_SOCKETENGINE);
73 ref = new EventHandler* [GetMaxFds()];
74 events = new struct epoll_event[GetMaxFds()];
76 memset(ref, 0, GetMaxFds() * sizeof(EventHandler*));
79 EPollEngine::~EPollEngine()
81 this->Close(EngineHandle);
86 static int mask_to_epoll(int event_mask)
89 if (event_mask & (FD_WANT_POLL_READ | FD_WANT_POLL_WRITE | FD_WANT_SINGLE_WRITE))
91 // we need to use standard polling on this FD
92 if (event_mask & (FD_WANT_POLL_READ | FD_WANT_FAST_READ))
94 if (event_mask & (FD_WANT_POLL_WRITE | FD_WANT_FAST_WRITE | FD_WANT_SINGLE_WRITE))
99 // we can use edge-triggered polling on this FD
101 if (event_mask & (FD_WANT_FAST_READ | FD_WANT_EDGE_READ))
103 if (event_mask & (FD_WANT_FAST_WRITE | FD_WANT_EDGE_WRITE))
109 bool EPollEngine::AddFd(EventHandler* eh, int event_mask)
111 int fd = eh->GetFd();
112 if ((fd < 0) || (fd > GetMaxFds() - 1))
114 ServerInstance->Logs->Log("SOCKET",DEBUG,"AddFd out of range: (fd: %d, max: %d)", fd, GetMaxFds());
120 ServerInstance->Logs->Log("SOCKET",DEBUG,"Attempt to add duplicate fd: %d", fd);
124 struct epoll_event ev;
125 memset(&ev,0,sizeof(ev));
126 ev.events = mask_to_epoll(event_mask);
128 int i = epoll_ctl(EngineHandle, EPOLL_CTL_ADD, fd, &ev);
131 ServerInstance->Logs->Log("SOCKET",DEBUG,"Error adding fd: %d to socketengine: %s", fd, strerror(errno));
135 ServerInstance->Logs->Log("SOCKET",DEBUG,"New file descriptor: %d", fd);
138 SocketEngine::SetEventMask(eh, event_mask);
143 void EPollEngine::OnSetEvent(EventHandler* eh, int old_mask, int new_mask)
145 int old_events = mask_to_epoll(old_mask);
146 int new_events = mask_to_epoll(new_mask);
147 if (old_events != new_events)
149 // ok, we actually have something to tell the kernel about
150 struct epoll_event ev;
151 memset(&ev,0,sizeof(ev));
152 ev.events = new_events;
153 ev.data.fd = eh->GetFd();
154 epoll_ctl(EngineHandle, EPOLL_CTL_MOD, eh->GetFd(), &ev);
158 bool EPollEngine::DelFd(EventHandler* eh, bool force)
160 int fd = eh->GetFd();
161 if ((fd < 0) || (fd > GetMaxFds() - 1))
163 ServerInstance->Logs->Log("SOCKET",DEBUG,"DelFd out of range: (fd: %d, max: %d)", fd, GetMaxFds());
167 struct epoll_event ev;
168 memset(&ev,0,sizeof(ev));
170 int i = epoll_ctl(EngineHandle, EPOLL_CTL_DEL, fd, &ev);
174 ServerInstance->Logs->Log("SOCKET",DEBUG,"Cant remove socket: %s", strerror(errno));
180 ServerInstance->Logs->Log("SOCKET",DEBUG,"Remove file descriptor: %d", fd);
185 int EPollEngine::DispatchEvents()
187 socklen_t codesize = sizeof(int);
189 int i = epoll_wait(EngineHandle, events, GetMaxFds() - 1, 1000);
190 ServerInstance->UpdateTime();
194 for (int j = 0; j < i; j++)
196 EventHandler* eh = ref[events[j].data.fd];
199 if (events[j].events & EPOLLHUP)
202 eh->HandleEvent(EVENT_ERROR, 0);
205 if (events[j].events & EPOLLERR)
208 /* Get error number */
209 if (getsockopt(events[j].data.fd, SOL_SOCKET, SO_ERROR, &errcode, &codesize) < 0)
211 eh->HandleEvent(EVENT_ERROR, errcode);
214 int mask = eh->GetEventMask();
215 if (events[j].events & EPOLLIN)
216 mask &= ~FD_READ_WILL_BLOCK;
217 if (events[j].events & EPOLLOUT)
219 mask &= ~FD_WRITE_WILL_BLOCK;
220 if (mask & FD_WANT_SINGLE_WRITE)
222 int nm = mask & ~FD_WANT_SINGLE_WRITE;
223 OnSetEvent(eh, mask, nm);
227 SetEventMask(eh, mask);
228 if (events[j].events & EPOLLIN)
231 eh->HandleEvent(EVENT_READ);
233 if (events[j].events & EPOLLOUT)
236 eh->HandleEvent(EVENT_WRITE);
243 std::string EPollEngine::GetName()
248 SocketEngine* CreateSocketEngine()
250 return new EPollEngine;