src/socketengine.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
   6  *                       E-mail:
   7  *                <brain@chatspike.net>
   8  *                <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *            the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 #include "inspircd_config.h"
  18 #include "globals.h"
  19 #include "inspircd.h"
  20 #ifdef USE_EPOLL
  21 #include <sys/epoll.h>
  22 #endif
  23 #ifdef USE_KQUEUE
  24 #include <sys/types.h>
  25 #include <sys/event.h>
  26 #include <sys/time.h>
  27 #endif
  28 #include <vector>
  29 #include <string>
  30 #include "socketengine.h"
  31 #include "helperfuncs.h"
  32
  33 char ref[MAX_DESCRIPTORS];
  34
  35 SocketEngine::SocketEngine()
  36 {
  37         log(DEBUG,"SocketEngine::SocketEngine()");
  38 #ifdef USE_EPOLL
  39         EngineHandle = epoll_create(MAX_DESCRIPTORS);
  40 #endif
  41 #ifdef USE_KQUEUE
  42         EngineHandle = kqueue();
  43 #endif
  44 #ifdef USE_SELECT
  45         EngineHandle = 0;
  46 #endif
  47         if (EngineHandle == -1)
  48         {
  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.");
  53                 Exit(0);
  54         }
  55         CurrentSetSize = 0;
  56 }
  57
  58 SocketEngine::~SocketEngine()
  59 {
  60         log(DEBUG,"SocketEngine::~SocketEngine()");
  61 #ifdef USE_EPOLL
  62         close(EngineHandle);
  63 #endif
  64 #ifdef USE_KQUEUE
  65         close(EngineHandle);
  66 #endif
  67 }
  68
  69 char SocketEngine::GetType(int fd)
  70 {
  71         if ((fd < 0) || (fd > MAX_DESCRIPTORS))
  72                 return X_EMPTY_SLOT;
  73         /* Mask off the top bit used for 'read/write' state */
  74         return (ref[fd] & ~0x80);
  75 }
  76
  77 bool SocketEngine::AddFd(int fd, bool readable, char type)
  78 {
  79         if ((fd < 0) || (fd > MAX_DESCRIPTORS))
  80         {
  81                 log(DEFAULT,"ERROR: FD of %d added above max of %d",fd,MAX_DESCRIPTORS);
  82                 return false;
  83         }
  84         if (GetRemainingFds() <= 1)
  85         {
  86                 log(DEFAULT,"ERROR: System out of file descriptors!");
  87                 return false;
  88         }
  89 #ifdef USE_SELECT
  90         fds[fd] = fd;
  91 #endif
  92         ref[fd] = type;
  93         if (readable)
  94         {
  95                 log(DEBUG,"Set readbit");
  96                 ref[fd] |= X_READBIT;
  97         }
  98         log(DEBUG,"Add socket %d",fd);
  99 #ifdef USE_EPOLL
 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;
 104         ev.data.fd = fd;
 105         int i = epoll_ctl(EngineHandle, EPOLL_CTL_ADD, fd, &ev);
 106         if (i < 0)
 107         {
 108                 log(DEBUG,"epoll: List insertion failure!");
 109                 return false;
 110         }
 111 #endif
 112 #ifdef USE_KQUEUE
 113         struct kevent ke;
 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);
 117         if (i == -1)
 118         {
 119                 log(DEBUG,"kqueue: List insertion failure!");
 120                 return false;
 121         }
 122 #endif
 123         CurrentSetSize++;
 124         return true;
 125 }
 126
 127 bool SocketEngine::HasFd(int fd)
 128 {
 129         if ((fd < 0) || (fd > MAX_DESCRIPTORS))
 130                 return false;
 131         return (ref[fd] != 0);
 132 }
 133
 134 bool SocketEngine::DelFd(int fd)
 135 {
 136         log(DEBUG,"SocketEngine::DelFd(%d)",fd);
 137
 138         if ((fd < 0) || (fd > MAX_DESCRIPTORS))
 139                 return false;
 140
 141 #ifdef USE_SELECT
 142         std::map<int,int>::iterator t = fds.find(fd);
 143         if (t != fds.end())
 144         {
 145                 fds.erase(t);
 146                 log(DEBUG,"Deleted fd %d",fd);
 147         }
 148 #endif
 149 #ifdef USE_KQUEUE
 150         struct kevent ke;
 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);
 153         if (i == -1)
 154         {
 155                 log(DEBUG,"kqueue: Failed to remove socket from queue!");
 156                 return false;
 157         }
 158 #endif
 159 #ifdef USE_EPOLL
 160         struct epoll_event ev;
 161         memset(&ev,0,sizeof(struct epoll_event));
 162         ref[fd] && X_READBIT ? ev.events = EPOLLIN : ev.events = EPOLLOUT;
 163         ev.data.fd = fd;
 164         int i = epoll_ctl(EngineHandle, EPOLL_CTL_DEL, fd, &ev);
 165         if (i < 0)
 166         {
 167                 log(DEBUG,"epoll: List deletion failure!");
 168                 return false;
 169         }
 170 #endif
 171         CurrentSetSize--;
 172         ref[fd] = 0;
 173         return true;
 174 }
 175
 176 int SocketEngine::GetMaxFds()
 177 {
 178 #ifdef USE_SELECT
 179         return FD_SETSIZE;
 180 #endif
 181 #ifdef USE_KQUEUE
 182         return MAX_DESCRIPTORS;
 183 #endif
 184 #ifdef USE_EPOLL
 185         return MAX_DESCRIPTORS;
 186 #endif
 187 }
 188
 189 int SocketEngine::GetRemainingFds()
 190 {
 191 #ifdef USE_SELECT
 192         return FD_SETSIZE - CurrentSetSize;
 193 #endif
 194 #ifdef USE_KQUEUE
 195         return MAX_DESCRIPTORS - CurrentSetSize;
 196 #endif
 197 #ifdef USE_EPOLL
 198         return MAX_DESCRIPTORS - CurrentSetSize;
 199 #endif
 200 }
 201
 202 int SocketEngine::Wait(int* fdlist)
 203 {
 204         int result = 0;
 205 #ifdef USE_SELECT
 206         FD_ZERO(&wfdset);
 207         FD_ZERO(&rfdset);
 208         timeval tval;
 209         int sresult;
 210         for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
 211         {
 212                 if (ref[a->second] & X_READBIT)
 213                 {
 214                         FD_SET (a->second, &rfdset);
 215                 }
 216                 else
 217                 {
 218                         FD_SET (a->second, &wfdset);
 219                 }
 220
 221         }
 222         tval.tv_sec = 0;
 223         tval.tv_usec = 50L;
 224         sresult = select(FD_SETSIZE, &rfdset, &wfdset, NULL, &tval);
 225         if (sresult > 0)
 226         {
 227                 for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
 228                 {
 229                         if ((FD_ISSET (a->second, &rfdset)) || (FD_ISSET (a->second, &wfdset)))
 230                                 fdlist[result++] = a->second;
 231                 }
 232         }
 233 #endif
 234 #ifdef USE_KQUEUE
 235         ts.tv_nsec = 5000L;
 236         ts.tv_sec = 0;
 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;
 240 #endif
 241 #ifdef USE_EPOLL
 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;
 245 #endif
 246         return result;
 247 }
 248
 249 std::string SocketEngine::GetName()
 250 {
 251 #ifdef USE_SELECT
 252         return "select";
 253 #endif
 254 #ifdef USE_KQUEUE
 255         return "kqueue";
 256 #endif
 257 #ifdef USE_EPOLL
 258         return "epoll";
 259 #endif
 260         return "misconfigured";
 261 }