src/socketengine.cpp

   1 #include "inspircd_config.h"
   2 #include "globals.h"
   3 #include "inspircd.h"
   4 #ifdef USE_EPOLL
   5 #include <sys/epoll.h>
   6 #define EP_DELAY 5
   7 #endif
   8 #ifdef USE_KQUEUE
   9 #include <sys/types.h>
  10 #include <sys/event.h>
  11 #include <sys/time.h>
  12 #endif
  13 #include <vector>
  14 #include <string>
  15 #include "socketengine.h"
  16
  17 char ref[65535];
  18
  19 SocketEngine::SocketEngine()
  20 {
  21         log(DEBUG,"SocketEngine::SocketEngine()");
  22 #ifdef USE_EPOLL
  23         EngineHandle = epoll_create(65535);
  24 #endif
  25 #ifdef USE_KQUEUE
  26         EngineHandle = kqueue();
  27 #endif
  28 }
  29
  30 SocketEngine::~SocketEngine()
  31 {
  32         log(DEBUG,"SocketEngine::~SocketEngine()");
  33 #ifdef USE_EPOLL
  34         close(EngineHandle);
  35 #endif
  36 #ifdef USE_KQUEUE
  37         close(EngineHandle);
  38 #endif
  39 }
  40
  41 bool SocketEngine::AddFd(int fd, bool readable, char type)
  42 {
  43         this->fds.push_back(fd);
  44         if (readable)
  45                 ref[fd] |= X_READBIT;
  46 #ifdef USE_EPOLL
  47         struct epoll_event ev;
  48         log(DEBUG,"epoll: Add socket to events, ep=%d socket=%d",EngineHandle,fd);
  49         readable ? ev.events = EPOLLIN | EPOLLET : ev.events = EPOLLOUT | EPOLLET;
  50         ev.data.fd = fd;
  51         int i = epoll_ctl(EngineHandle, EPOLL_CTL_ADD, fd, &ev);
  52         if (i < 0)
  53         {
  54                 log(DEBUG,"epoll: List insertion failure!");
  55                 return false;
  56         }
  57 #endif
  58 #ifdef USE_KQUEUE
  59         struct kevent ke;
  60         log(DEBUG,"kqueue: Add socket to events, kq=%d socket=%d",EngineHandle,fd);
  61         EV_SET(&ke, fd, readable ? EVFILT_READ : EVFILT_WRITE, EV_ADD, 0, 0, NULL);
  62         int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
  63         if (i == -1)
  64         {
  65                 log(DEBUG,"kqueue: List insertion failure!");
  66                 return false;
  67         }
  68 #endif
  69 return true;
  70 }
  71
  72 bool SocketEngine::DelFd(int fd)
  73 {
  74         log(DEBUG,"SocketEngine::DelFd(%d)",fd);
  75         bool found = false;
  76         for (std::vector<int>::iterator i = fds.begin(); i != fds.end(); i++)
  77         {
  78                 if (*i == fd)
  79                 {
  80                         fds.erase(i);
  81                         found = true;
  82                         break;
  83                 }
  84         }
  85 #ifdef USE_KQUEUE
  86         struct kevent ke;
  87         EV_SET(&ke, fd, ref[fd] && X_READBIT ? EVFILT_READ : EVFILT_WRITE, EV_DELETE, 0, 0, NULL);
  88         int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
  89         if (i == -1)
  90         {
  91                 log(DEBUG,"kqueue: Failed to remove socket from queue!");
  92                 return false;
  93         }
  94 #endif
  95 #ifdef USE_EPOLL
  96         struct epoll_event ev;
  97         ref[fd] && X_READBIT ? ev.events = EPOLLIN | EPOLLET : ev.events = EPOLLOUT | EPOLLET;
  98         ev.data.fd = fd;
  99         int i = epoll_ctl(EngineHandle, EPOLL_CTL_DEL, fd, &ev);
 100         if (i < 0)
 101         {
 102                 log(DEBUG,"epoll: List deletion failure!");
 103                 return false;
 104         }
 105 #endif
 106         ref[fd] = 0;
 107         return found;
 108 }
 109
 110 bool SocketEngine::Wait(std::vector<int> &fdlist)
 111 {
 112         fdlist.clear();
 113 #ifdef USE_SELECT
 114         FD_ZERO(&wfdset);
 115         FD_ZERO(&rfdset);
 116         timeval tval;
 117         int sresult;
 118         for (unsigned int a = 0; a < fds.size(); a++)
 119         {
 120                 if (ref[fds[a]] && X_READBIT)
 121                 {
 122                         FD_SET (fds[a], &rfdset);
 123                 }
 124                 else
 125                 {
 126                         FD_SET (fds[a], &wfdset);
 127                 }
 128
 129         }
 130         tval.tv_sec = 0;
 131         tval.tv_usec = 1000L;
 132         sresult = select(FD_SETSIZE, &rfdset, &wfdset, NULL, &tval);
 133         if (sresult > 0)
 134         {
 135                 for (unsigned int a = 0; a < fds.size(); a++)
 136                 {
 137                         if ((FD_ISSET (fds[a], &rfdset)) || (FD_ISSET (fds[a], &wfdset)))
 138                                 fdlist.push_back(fds[a]);
 139                 }
 140         }
 141 #endif
 142 #ifdef USE_KQUEUE
 143         ts.tv_nsec = 1000L;
 144         ts.tv_sec = 0;
 145         int i = kevent(EngineHandle, NULL, 0, &ke_list[0], 65535, &ts);
 146         for (int j = 0; j < i; j++)
 147                 fdlist.push_back(ke_list[j].ident);
 148 #endif
 149 #ifdef USE_EPOLL
 150         int i = epoll_wait(EngineHandle, events, 65535, 1);
 151         for (int j = 0; j < i; j++)
 152                 fdlist.push_back(events[j].data.fd);
 153 #endif
 154         return true;
 155 }
 156
 157 std::string SocketEngine::GetName()
 158 {
 159 #ifdef USE_SELECT
 160         return "select";
 161 #endif
 162 #ifdef USE_KQUEUE
 163         return "kqueue";
 164 #endif
 165 #ifdef USE_EPOLL
 166         return "epoll";
 167 #endif
 168         return "misconfigured";
 169 }