1 /* +------------------------------------+
2 * | Inspire Internet Relay Chat Daemon |
3 * +------------------------------------+
5 * InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
7 * <brain@chatspike.net>
8 * <Craig@chatspike.net>
10 * Written by Craig Edwards, Craig McLure, and others.
11 * This program is free but copyrighted software; see
12 * the file COPYING for details.
14 * ---------------------------------------------------
17 #include "inspircd_config.h"
21 #include <sys/epoll.h>
24 #include <sys/types.h>
25 #include <sys/event.h>
30 #include "socketengine.h"
31 #include "helperfuncs.h"
33 char ref[MAX_DESCRIPTORS];
35 SocketEngine::SocketEngine()
37 log(DEBUG,"SocketEngine::SocketEngine()");
39 EngineHandle = epoll_create(MAX_DESCRIPTORS);
42 EngineHandle = kqueue();
47 if (EngineHandle == -1)
49 log(SPARSE,"ERROR: Could not initialize socket engine. Your kernel probably does not have the proper features.");
50 log(SPARSE,"ERROR: this is a fatal error, exiting now.");
51 printf("ERROR: Could not initialize socket engine. Your kernel probably does not have the proper features.");
52 printf("ERROR: this is a fatal error, exiting now.");
58 SocketEngine::~SocketEngine()
60 log(DEBUG,"SocketEngine::~SocketEngine()");
69 char SocketEngine::GetType(int fd)
71 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
73 /* Mask off the top bit used for 'read/write' state */
74 return (ref[fd] & ~0x80);
77 bool SocketEngine::AddFd(int fd, bool readable, char type)
79 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
81 log(DEFAULT,"ERROR: FD of %d added above max of %d",fd,MAX_DESCRIPTORS);
84 if (GetRemainingFds() <= 1)
86 log(DEFAULT,"ERROR: System out of file descriptors!");
95 log(DEBUG,"Set readbit");
98 log(DEBUG,"Add socket %d",fd);
100 struct epoll_event ev;
101 memset(&ev,0,sizeof(struct epoll_event));
102 log(DEBUG,"epoll: Add socket to events, ep=%d socket=%d",EngineHandle,fd);
103 readable ? ev.events = EPOLLIN : ev.events = EPOLLOUT;
105 int i = epoll_ctl(EngineHandle, EPOLL_CTL_ADD, fd, &ev);
108 log(DEBUG,"epoll: List insertion failure!");
114 log(DEBUG,"kqueue: Add socket to events, kq=%d socket=%d",EngineHandle,fd);
115 EV_SET(&ke, fd, readable ? EVFILT_READ : EVFILT_WRITE, EV_ADD, 0, 0, NULL);
116 int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
119 log(DEBUG,"kqueue: List insertion failure!");
127 bool SocketEngine::HasFd(int fd)
129 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
131 return (ref[fd] != 0);
134 bool SocketEngine::DelFd(int fd)
136 log(DEBUG,"SocketEngine::DelFd(%d)",fd);
138 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
142 std::map<int,int>::iterator t = fds.find(fd);
146 log(DEBUG,"Deleted fd %d",fd);
151 EV_SET(&ke, fd, ref[fd] & X_READBIT ? EVFILT_READ : EVFILT_WRITE, EV_DELETE, 0, 0, NULL);
152 int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
155 log(DEBUG,"kqueue: Failed to remove socket from queue!");
160 struct epoll_event ev;
161 memset(&ev,0,sizeof(struct epoll_event));
162 ref[fd] && X_READBIT ? ev.events = EPOLLIN : ev.events = EPOLLOUT;
164 int i = epoll_ctl(EngineHandle, EPOLL_CTL_DEL, fd, &ev);
167 log(DEBUG,"epoll: List deletion failure!");
176 int SocketEngine::GetMaxFds()
182 return MAX_DESCRIPTORS;
185 return MAX_DESCRIPTORS;
189 int SocketEngine::GetRemainingFds()
192 return FD_SETSIZE - CurrentSetSize;
195 return MAX_DESCRIPTORS - CurrentSetSize;
198 return MAX_DESCRIPTORS - CurrentSetSize;
202 int SocketEngine::Wait(int* fdlist)
210 for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
212 if (ref[a->second] & X_READBIT)
214 FD_SET (a->second, &rfdset);
218 FD_SET (a->second, &wfdset);
224 sresult = select(FD_SETSIZE, &rfdset, &wfdset, NULL, &tval);
227 for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
229 if ((FD_ISSET (a->second, &rfdset)) || (FD_ISSET (a->second, &wfdset)))
230 fdlist[result++] = a->second;
237 int i = kevent(EngineHandle, NULL, 0, &ke_list[0], MAX_DESCRIPTORS, &ts);
238 for (int j = 0; j < i; j++)
239 fdlist[result++] = ke_list[j].ident;
242 int i = epoll_wait(EngineHandle, events, MAX_DESCRIPTORS, 50);
243 for (int j = 0; j < i; j++)
244 fdlist[result++] = events[j].data.fd;
249 std::string SocketEngine::GetName()
260 return "misconfigured";