src/socketengines/socketengine_ports.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
   6  * See: http://www.inspircd.org/wiki/index.php/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "exitcodes.h"
  16 #include <port.h>
  17 #include "socketengines/socketengine_ports.h"
  18
  19 PortsEngine::PortsEngine(InspIRCd* Instance) : SocketEngine(Instance)
  20 {
  21         EngineHandle = port_create();
  22
  23         if (EngineHandle == -1)
  24         {
  25                 ServerInstance->Log(SPARSE,"ERROR: Could not initialize socket engine: %s", strerror(errno));
  26                 ServerInstance->Log(SPARSE,"ERROR: This is a fatal error, exiting now.");
  27                 printf("ERROR: Could not initialize socket engine: %s\n", strerror(errno));
  28                 printf("ERROR: This is a fatal error, exiting now.\n");
  29                 ServerInstance->Exit(EXIT_STATUS_SOCKETENGINE);
  30         }
  31         CurrentSetSize = 0;
  32 }
  33
  34 PortsEngine::~PortsEngine()
  35 {
  36         this->Close(EngineHandle);
  37 }
  38
  39 bool PortsEngine::AddFd(EventHandler* eh)
  40 {
  41         int fd = eh->GetFd();
  42         if ((fd < 0) || (fd > MAX_DESCRIPTORS))
  43                 return false;
  44
  45         if (GetRemainingFds() <= 1)
  46                 return false;
  47
  48         if (ref[fd])
  49                 return false;
  50
  51         ref[fd] = eh;
  52         port_associate(EngineHandle, PORT_SOURCE_FD, fd, eh->Readable() ? POLLRDNORM : POLLWRNORM, eh);
  53
  54         ServerInstance->Log(DEBUG,"New file descriptor: %d", fd);
  55         CurrentSetSize++;
  56         return true;
  57 }
  58
  59 void PortsEngine::WantWrite(EventHandler* eh)
  60 {
  61         port_associate(EngineHandle, PORT_SOURCE_FD, eh->GetFd(), POLLWRNORM, eh);
  62 }
  63
  64 bool PortsEngine::DelFd(EventHandler* eh, bool force)
  65 {
  66         int fd = eh->GetFd();
  67         if ((fd < 0) || (fd > MAX_DESCRIPTORS))
  68                 return false;
  69
  70         port_dissociate(EngineHandle, PORT_SOURCE_FD, fd);
  71
  72         CurrentSetSize--;
  73         ref[fd] = NULL;
  74
  75         ServerInstance->Log(DEBUG,"Remove file descriptor: %d", fd);
  76         return true;
  77 }
  78
  79 int PortsEngine::GetMaxFds()
  80 {
  81         return MAX_DESCRIPTORS;
  82 }
  83
  84 int PortsEngine::GetRemainingFds()
  85 {
  86         return MAX_DESCRIPTORS - CurrentSetSize;
  87 }
  88
  89 int PortsEngine::DispatchEvents()
  90 {
  91         struct timespec poll_time;
  92
  93         poll_time.tv_sec = 1;
  94         poll_time.tv_nsec = 0;
  95
  96         unsigned int nget = 1; // used to denote a retrieve request.
  97         int i = port_getn(EngineHandle, this->events, MAX_DESCRIPTORS, &nget, &poll_time);
  98
  99         // first handle an error condition
 100         if (i == -1)
 101                 return i;
 102
 103         for (i = 0; i < nget; i++)
 104         {
 105                 switch (this->events[i].portev_source)
 106                 {
 107                         case PORT_SOURCE_FD:
 108                         {
 109                                 int fd = this->events[i].portev_object;
 110                                 if (ref[fd])
 111                                 {
 112                                         // reinsert port for next time around
 113                                         port_associate(EngineHandle, PORT_SOURCE_FD, fd, POLLRDNORM, ref[fd]);
 114                                         ref[fd]->HandleEvent((this->events[i].portev_events & POLLRDNORM) ? EVENT_READ : EVENT_WRITE);
 115                                 }
 116                         }
 117                         default:
 118                         break;
 119                 }
 120         }
 121
 122         return i;
 123 }
 124
 125 std::string PortsEngine::GetName()
 126 {
 127         return "ports";
 128 }
 129