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"
32 char ref[MAX_DESCRIPTORS];
34 SocketEngine::SocketEngine()
36 log(DEBUG,"SocketEngine::SocketEngine()");
38 EngineHandle = epoll_create(MAX_DESCRIPTORS);
41 EngineHandle = kqueue();
46 if (EngineHandle == -1)
48 log(SPARSE,"ERROR: Could not initialize socket engine. Your kernel probably does not have the proper features.");
49 log(SPARSE,"ERROR: this is a fatal error, exiting now.");
50 printf("ERROR: Could not initialize socket engine. Your kernel probably does not have the proper features.");
51 printf("ERROR: this is a fatal error, exiting now.");
57 SocketEngine::~SocketEngine()
59 log(DEBUG,"SocketEngine::~SocketEngine()");
68 char SocketEngine::GetType(int fd)
70 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
72 /* Mask off the top bit used for 'read/write' state */
73 return (ref[fd] & ~0x80);
76 bool SocketEngine::AddFd(int fd, bool readable, char type)
78 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
80 log(DEFAULT,"ERROR: FD of %d added above max of %d",fd,MAX_DESCRIPTORS);
83 if (GetRemainingFds() <= 1)
85 log(DEFAULT,"ERROR: System out of file descriptors!");
94 log(DEBUG,"Set readbit");
97 log(DEBUG,"Add socket %d",fd);
99 struct epoll_event ev;
100 memset(&ev,0,sizeof(struct epoll_event));
101 log(DEBUG,"epoll: Add socket to events, ep=%d socket=%d",EngineHandle,fd);
102 readable ? ev.events = EPOLLIN : ev.events = EPOLLOUT;
104 int i = epoll_ctl(EngineHandle, EPOLL_CTL_ADD, fd, &ev);
107 log(DEBUG,"epoll: List insertion failure!");
113 log(DEBUG,"kqueue: Add socket to events, kq=%d socket=%d",EngineHandle,fd);
114 EV_SET(&ke, fd, readable ? EVFILT_READ : EVFILT_WRITE, EV_ADD, 0, 0, NULL);
115 int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
118 log(DEBUG,"kqueue: List insertion failure!");
126 bool SocketEngine::HasFd(int fd)
128 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
130 return (ref[fd] != 0);
133 bool SocketEngine::DelFd(int fd)
135 log(DEBUG,"SocketEngine::DelFd(%d)",fd);
137 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
141 std::map<int,int>::iterator t = fds.find(fd);
145 log(DEBUG,"Deleted fd %d",fd);
150 EV_SET(&ke, fd, ref[fd] & X_READBIT ? EVFILT_READ : EVFILT_WRITE, EV_DELETE, 0, 0, NULL);
151 int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
154 log(DEBUG,"kqueue: Failed to remove socket from queue!");
159 struct epoll_event ev;
160 memset(&ev,0,sizeof(struct epoll_event));
161 ref[fd] && X_READBIT ? ev.events = EPOLLIN : ev.events = EPOLLOUT;
163 int i = epoll_ctl(EngineHandle, EPOLL_CTL_DEL, fd, &ev);
166 log(DEBUG,"epoll: List deletion failure!");
175 int SocketEngine::GetMaxFds()
181 return MAX_DESCRIPTORS;
184 return MAX_DESCRIPTORS;
188 int SocketEngine::GetRemainingFds()
191 return FD_SETSIZE - CurrentSetSize;
194 return MAX_DESCRIPTORS - CurrentSetSize;
197 return MAX_DESCRIPTORS - CurrentSetSize;
201 int SocketEngine::Wait(int* fdlist)
209 for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
211 if (ref[a->second] & X_READBIT)
213 FD_SET (a->second, &rfdset);
217 FD_SET (a->second, &wfdset);
223 sresult = select(FD_SETSIZE, &rfdset, &wfdset, NULL, &tval);
226 for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
228 if ((FD_ISSET (a->second, &rfdset)) || (FD_ISSET (a->second, &wfdset)))
229 fdlist[result++] = a->second;
236 int i = kevent(EngineHandle, NULL, 0, &ke_list[0], MAX_DESCRIPTORS, &ts);
237 for (int j = 0; j < i; j++)
238 fdlist[result++] = ke_list[j].ident;
241 int i = epoll_wait(EngineHandle, events, MAX_DESCRIPTORS, 50);
242 for (int j = 0; j < i; j++)
243 fdlist[result++] = events[j].data.fd;
248 std::string SocketEngine::GetName()
259 return "misconfigured";