-/* +------------------------------------+
- * | Inspire Internet Relay Chat Daemon |
- * +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
*
- * InspIRCd: (C) 2002-2008 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
+ * Copyright (C) 2014 Adam <Adam@anope.org>
+ * Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ * Copyright (C) 2007-2008 Craig Edwards <craigedwards@brainbox.cc>
*
- * This program is free but copyrighted software; see
- * the file COPYING for details.
+ * This file is part of InspIRCd. InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
*
- * ---------------------------------------------------
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
+
#include "inspircd.h"
-#ifndef WINDOWS
+
+#ifndef _WIN32
#include <sys/select.h>
-#endif // WINDOWS
-#include "socketengines/socketengine_select.h"
+#endif // _WIN32
+/** A specialisation of the SocketEngine class, designed to use traditional select().
+ */
+namespace
+{
+ fd_set ReadSet, WriteSet, ErrSet;
+ int MaxFD = 0;
+}
-SelectEngine::SelectEngine(InspIRCd* Instance) : SocketEngine(Instance)
+void SocketEngine::Init()
{
- EngineHandle = 0;
- CurrentSetSize = 0;
+ MAX_DESCRIPTORS = FD_SETSIZE;
- writeable = new bool [GetMaxFds()];
- memset(writeable, 0, sizeof(writeable));
- ref = new EventHandler* [GetMaxFds()];
- memset(ref, 0, GetMaxFds() * sizeof(EventHandler*));
+ FD_ZERO(&ReadSet);
+ FD_ZERO(&WriteSet);
+ FD_ZERO(&ErrSet);
}
-SelectEngine::~SelectEngine()
+void SocketEngine::Deinit()
{
- delete[] ref;
}
-bool SelectEngine::AddFd(EventHandler* eh)
+void SocketEngine::RecoverFromFork()
+{
+}
+
+bool SocketEngine::AddFd(EventHandler* eh, int event_mask)
{
int fd = eh->GetFd();
if ((fd < 0) || (fd > GetMaxFds() - 1))
return false;
- if (GetRemainingFds() <= 1)
+ if (!SocketEngine::AddFdRef(eh))
return false;
- if (ref[fd])
- return false;
-
- fds[fd] = fd;
- ref[fd] = eh;
- CurrentSetSize++;
+ eh->SetEventMask(event_mask);
+ OnSetEvent(eh, 0, event_mask);
+ FD_SET(fd, &ErrSet);
+ if (fd > MaxFD)
+ MaxFD = fd;
- ServerInstance->Logs->Log("SOCKET",DEBUG,"New file descriptor: %d", fd);
+ ServerInstance->Logs->Log("SOCKET", 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)
+void SocketEngine::DelFd(EventHandler* eh)
{
int fd = eh->GetFd();
if ((fd < 0) || (fd > GetMaxFds() - 1))
- return false;
+ return;
- std::map<int,int>::iterator t = fds.find(fd);
- if (t != fds.end())
- fds.erase(t);
+ SocketEngine::DelFdRef(eh);
- CurrentSetSize--;
- ref[fd] = NULL;
+ FD_CLR(fd, &ReadSet);
+ FD_CLR(fd, &WriteSet);
+ FD_CLR(fd, &ErrSet);
+ if (fd == MaxFD)
+ --MaxFD;
- ServerInstance->Logs->Log("SOCKET",DEBUG,"Remove file descriptor: %d", fd);
- return true;
+ ServerInstance->Logs->Log("SOCKET", LOG_DEBUG, "Remove file descriptor: %d", fd);
}
-int SelectEngine::GetMaxFds()
+void SocketEngine::OnSetEvent(EventHandler* eh, int old_mask, int new_mask)
{
- return FD_SETSIZE;
-}
+ int fd = eh->GetFd();
+ int diff = old_mask ^ new_mask;
-int SelectEngine::GetRemainingFds()
-{
- return GetMaxFds() - CurrentSetSize;
+ if (diff & (FD_WANT_POLL_READ | FD_WANT_FAST_READ))
+ {
+ if (new_mask & (FD_WANT_POLL_READ | FD_WANT_FAST_READ))
+ FD_SET(fd, &ReadSet);
+ else
+ FD_CLR(fd, &ReadSet);
+ }
+ if (diff & (FD_WANT_POLL_WRITE | FD_WANT_FAST_WRITE | FD_WANT_SINGLE_WRITE))
+ {
+ if (new_mask & (FD_WANT_POLL_WRITE | FD_WANT_FAST_WRITE | FD_WANT_SINGLE_WRITE))
+ FD_SET(fd, &WriteSet);
+ else
+ FD_CLR(fd, &WriteSet);
+ }
}
-int SelectEngine::DispatchEvents()
+int SocketEngine::DispatchEvents()
{
- int result = 0;
timeval tval;
- int sresult = 0;
- EventHandler* ev[GetMaxFds()];
- socklen_t codesize;
- int errcode;
+ tval.tv_sec = 1;
+ tval.tv_usec = 0;
+
+ fd_set rfdset = ReadSet, wfdset = WriteSet, errfdset = ErrSet;
- FD_ZERO(&wfdset);
- FD_ZERO(&rfdset);
- FD_ZERO(&errfdset);
+ int sresult = select(MaxFD + 1, &rfdset, &wfdset, &errfdset, &tval);
+ ServerInstance->UpdateTime();
- for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
+ for (int i = 0, j = sresult; i <= MaxFD && j > 0; i++)
{
- 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);
+ int has_read = FD_ISSET(i, &rfdset), has_write = FD_ISSET(i, &wfdset), has_error = FD_ISSET(i, &errfdset);
- FD_SET (a->second, &errfdset);
- }
+ if (!(has_read || has_write || has_error))
+ continue;
- tval.tv_sec = 1;
- tval.tv_usec = 0;
+ --j;
- sresult = select(FD_SETSIZE, &rfdset, &wfdset, &errfdset, &tval);
+ EventHandler* ev = GetRef(i);
+ if (!ev)
+ continue;
- if (sresult > 0)
- {
- for (std::map<int,int>::iterator a = fds.begin(); a != fds.end(); a++)
+ if (has_error)
{
- if ((FD_ISSET (a->second, &rfdset)) || (FD_ISSET (a->second, &wfdset)) || FD_ISSET (a->second, &errfdset))
- {
- ev[result++] = ref[a->second];
- }
- }
- }
+ stats.ErrorEvents++;
- /** 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))
- {
- ErrorEvents++;
-
- if (getsockopt(ev[i]->GetFd(), SOL_SOCKET, SO_ERROR, (char*)&errcode, &codesize) < 0)
- errcode = errno;
+ socklen_t codesize = sizeof(int);
+ int errcode = 0;
+ if (getsockopt(i, SOL_SOCKET, SO_ERROR, (char*)&errcode, &codesize) < 0)
+ errcode = errno;
- ev[i]->HandleEvent(EVENT_ERROR, errcode);
+ ev->OnEventHandlerError(errcode);
+ continue;
+ }
+ if (has_read)
+ {
+ stats.ReadEvents++;
+ ev->SetEventMask(ev->GetEventMask() & ~FD_READ_WILL_BLOCK);
+ ev->OnEventHandlerRead();
+ if (ev != GetRef(i))
continue;
- }
- else
- {
- if (writeable[ev[i]->GetFd()])
- {
- WriteEvents++;
- writeable[ev[i]->GetFd()] = false;
- ev[i]->HandleEvent(EVENT_WRITE);
- }
- else
- {
- if (ev[i]->Readable())
- {
- ReadEvents++;
- ev[i]->HandleEvent(EVENT_READ);
- }
- else
- {
- WriteEvents++;
- ev[i]->HandleEvent(EVENT_WRITE);
- }
- }
- }
}
- }
- return result;
-}
+ if (has_write)
+ {
+ stats.WriteEvents++;
+ int newmask = (ev->GetEventMask() & ~(FD_WRITE_WILL_BLOCK | FD_WANT_SINGLE_WRITE));
+ SocketEngine::OnSetEvent(ev, ev->GetEventMask(), newmask);
+ ev->SetEventMask(newmask);
+ ev->OnEventHandlerWrite();
+ }
+ }
-std::string SelectEngine::GetName()
-{
- return "select";
+ return sresult;
}