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 * ---------------------------------------------------
18 #include <sys/select.h>
19 #include "socketengine_select.h"
22 SelectEngine::SelectEngine(InspIRCd* Instance) : SocketEngine(Instance)
24 ServerInstance->Log(DEBUG,"SelectEngine::SelectEngine()");
27 memset(writeable, 0, sizeof(writeable));
30 SelectEngine::~SelectEngine()
32 ServerInstance->Log(DEBUG,"SelectEngine::~SelectEngine()");
35 bool SelectEngine::AddFd(EventHandler* eh)
38 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
40 ServerInstance->Log(DEBUG,"ERROR: FD of %d added above max of %d",fd,MAX_DESCRIPTORS);
43 if (GetRemainingFds() <= 1)
45 ServerInstance->Log(DEBUG,"ERROR: System out of file descriptors!");
55 ServerInstance->Log(DEBUG,"Add socket %d",fd);
61 void SelectEngine::WantWrite(EventHandler* eh)
63 writeable[eh->GetFd()] = true;
66 bool SelectEngine::DelFd(EventHandler* eh)
70 ServerInstance->Log(DEBUG,"SelectEngine::DelFd(%d)",fd);
72 if ((fd < 0) || (fd > MAX_DESCRIPTORS))
75 std::map<int,int>::iterator t = fds.find(fd);
79 ServerInstance->Log(DEBUG,"Deleted fd %d",fd);
87 int SelectEngine::GetMaxFds()
92 int SelectEngine::GetRemainingFds()
94 return FD_SETSIZE - CurrentSetSize;
97 int SelectEngine::DispatchEvents()
102 EventHandler* ev[MAX_DESCRIPTORS];
110 for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
112 if (ref[a->second]->Readable())
113 FD_SET (a->second, &rfdset);
115 FD_SET (a->second, &wfdset);
116 if (writeable[a->second])
117 FD_SET (a->second, &wfdset);
119 FD_SET (a->second, &errfdset);
123 sresult = select(FD_SETSIZE, &rfdset, &wfdset, &errfdset, &tval);
126 for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
128 if ((FD_ISSET (a->second, &rfdset)) || (FD_ISSET (a->second, &wfdset)) || FD_ISSET (a->second, &errfdset))
130 ev[result++] = ref[a->second];
135 /** An event handler may remove its own descriptor from the list, therefore it is not
136 * safe to directly iterate over the list and dispatch events there with STL iterators.
137 * Thats a shame because it makes this code slower and more resource intensive, but maybe
138 * the user should stop using select(), as select() smells anyway.
140 for (int i = 0; i < result; i++)
144 if (FD_ISSET (ev[i]->GetFd(), &errfdset))
148 if (getsockopt(ev[i]->GetFd(), SOL_SOCKET, SO_ERROR, &errcode, &codesize) < 0)
150 ev[i]->HandleEvent(EVENT_ERROR, errcode);
154 ServerInstance->Log(DEBUG,"Handle %s event on fd %d",writeable[ev[i]->GetFd()] || !ev[i]->Readable() ? "write" : "read", ev[i]->GetFd());
155 if (writeable[ev[i]->GetFd()])
158 ev[i]->HandleEvent(EVENT_WRITE);
159 writeable[ev[i]->GetFd()] = false;
165 ev[i]->HandleEvent(ev[i]->Readable() ? EVENT_READ : EVENT_WRITE);
173 std::string SelectEngine::GetName()