summaryrefslogtreecommitdiff
path: root/src/socketengines/socketengine_select.cpp
diff options
context:
space:
mode:
authorbrain <brain@e03df62e-2008-0410-955e-edbf42e46eb7>2007-10-23 22:30:25 +0000
committerbrain <brain@e03df62e-2008-0410-955e-edbf42e46eb7>2007-10-23 22:30:25 +0000
commitcbb8fda4a7e763738e9675ec9f96286c87879176 (patch)
tree0ed6862e3b9f5f35ea86ea57e8a799f038523c28 /src/socketengines/socketengine_select.cpp
parent71e400674fc9abc16b29004eff6fe18dd1c75f4e (diff)
Move socketengines into their own dir. This was all w00t's idea, but i told him no because i didnt think it would work. Now ive done it myself :P ner ner ne ner ner :)
git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@8332 e03df62e-2008-0410-955e-edbf42e46eb7
Diffstat (limited to 'src/socketengines/socketengine_select.cpp')
-rw-r--r--src/socketengines/socketengine_select.cpp165
1 files changed, 165 insertions, 0 deletions
diff --git a/src/socketengines/socketengine_select.cpp b/src/socketengines/socketengine_select.cpp
new file mode 100644
index 000000000..1fe138f14
--- /dev/null
+++ b/src/socketengines/socketengine_select.cpp
@@ -0,0 +1,165 @@
+/* +------------------------------------+
+ * | Inspire Internet Relay Chat Daemon |
+ * +------------------------------------+
+ *
+ * InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
+ * This program is free but copyrighted software; see
+ * the file COPYING for details.
+ *
+ * ---------------------------------------------------
+ */
+
+#include "inspircd.h"
+#include <sys/select.h>
+#include "socketengines/socketengine_select.h"
+
+
+SelectEngine::SelectEngine(InspIRCd* Instance) : SocketEngine(Instance)
+{
+ EngineHandle = 0;
+ CurrentSetSize = 0;
+ memset(writeable, 0, sizeof(writeable));
+}
+
+SelectEngine::~SelectEngine()
+{
+}
+
+bool SelectEngine::AddFd(EventHandler* eh)
+{
+ int fd = eh->GetFd();
+ if ((fd < 0) || (fd > MAX_DESCRIPTORS))
+ return false;
+
+ if (GetRemainingFds() <= 1)
+ return false;
+
+ if (ref[fd])
+ return false;
+
+ fds[fd] = fd;
+ ref[fd] = eh;
+ CurrentSetSize++;
+
+ ServerInstance->Log(DEBUG,"New file descriptor: %d", fd);
+ return true;
+}
+
+void SelectEngine::WantWrite(EventHandler* eh)
+{
+ writeable[eh->GetFd()] = true;
+}
+
+bool SelectEngine::DelFd(EventHandler* eh, bool force)
+{
+ int fd = eh->GetFd();
+
+ if ((fd < 0) || (fd > MAX_DESCRIPTORS))
+ return false;
+
+ std::map<int,int>::iterator t = fds.find(fd);
+ if (t != fds.end())
+ fds.erase(t);
+
+ CurrentSetSize--;
+ ref[fd] = NULL;
+ fds[fd] = 0;
+
+ ServerInstance->Log(DEBUG,"Remove file descriptor: %d", fd);
+ return true;
+}
+
+int SelectEngine::GetMaxFds()
+{
+ return FD_SETSIZE;
+}
+
+int SelectEngine::GetRemainingFds()
+{
+ return FD_SETSIZE - CurrentSetSize;
+}
+
+int SelectEngine::DispatchEvents()
+{
+ int result = 0;
+ timeval tval;
+ int sresult = 0;
+ EventHandler* ev[MAX_DESCRIPTORS];
+ socklen_t codesize;
+ int errcode;
+
+ FD_ZERO(&wfdset);
+ FD_ZERO(&rfdset);
+ FD_ZERO(&errfdset);
+
+ for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
+ {
+ if (ref[a->second]->Readable())
+ FD_SET (a->second, &rfdset);
+ else
+ FD_SET (a->second, &wfdset);
+ if (writeable[a->second])
+ FD_SET (a->second, &wfdset);
+
+ FD_SET (a->second, &errfdset);
+ }
+ tval.tv_sec = 1;
+ tval.tv_usec = 0;
+ sresult = select(FD_SETSIZE, &rfdset, &wfdset, &errfdset, &tval);
+ if (sresult > 0)
+ {
+ for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
+ {
+ if ((FD_ISSET (a->second, &rfdset)) || (FD_ISSET (a->second, &wfdset)) || FD_ISSET (a->second, &errfdset))
+ {
+ ev[result++] = ref[a->second];
+ }
+ }
+ }
+
+ /** An event handler may remove its own descriptor from the list, therefore it is not
+ * safe to directly iterate over the list and dispatch events there with STL iterators.
+ * Thats a shame because it makes this code slower and more resource intensive, but maybe
+ * the user should stop using select(), as select() smells anyway.
+ */
+ for (int i = 0; i < result; i++)
+ {
+ if (ev[i])
+ {
+ if (FD_ISSET (ev[i]->GetFd(), &errfdset))
+ {
+ if (ev[i])
+ {
+ if (getsockopt(ev[i]->GetFd(), SOL_SOCKET, SO_ERROR, &errcode, &codesize) < 0)
+ errcode = errno;
+
+ ev[i]->HandleEvent(EVENT_ERROR, errcode);
+ }
+ continue;
+ }
+ if (ev[i])
+ {
+ if (writeable[ev[i]->GetFd()])
+ {
+ writeable[ev[i]->GetFd()] = false;
+ if (ev[i])
+ ev[i]->HandleEvent(EVENT_WRITE);
+ }
+ else
+ {
+ if (ev[i])
+ ev[i]->HandleEvent(ev[i]->Readable() ? EVENT_READ : EVENT_WRITE);
+ }
+ }
+ }
+ }
+
+ return result;
+}
+
+std::string SelectEngine::GetName()
+{
+ return "select";
+}