summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorbrain <brain@e03df62e-2008-0410-955e-edbf42e46eb7>2005-12-12 11:50:45 +0000
committerbrain <brain@e03df62e-2008-0410-955e-edbf42e46eb7>2005-12-12 11:50:45 +0000
commitcef3e32fae0132d51b2fcca16ef05907b174e2a3 (patch)
tree6c56f7d3401068ee3e16486bcbd54a9151bc4b05
parent4f2e601fe71febf62db0f98a418a25598e24377b (diff)
EXPERIMENTAL new socket engine code
git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@2326 e03df62e-2008-0410-955e-edbf42e46eb7
-rw-r--r--include/socketengine.h7
-rw-r--r--src/commands.cpp29
-rw-r--r--src/inspircd.cpp713
-rw-r--r--src/modules.cpp31
-rw-r--r--src/socketengine.cpp20
5 files changed, 309 insertions, 491 deletions
diff --git a/include/socketengine.h b/include/socketengine.h
index 41cd459d8..3a52283a5 100644
--- a/include/socketengine.h
+++ b/include/socketengine.h
@@ -32,6 +32,13 @@
#include <sys/time.h>
#endif
+const char X_LISTEN = 0;
+const char X_ESTAB_CLIENT = 1;
+const char X_ESTAB_MODULE = 2;
+const char X_ESTAB_DNS = 3;
+
+const char X_READBIT = 0x80;
+
class SocketEngine {
std::vector<int> fds;
diff --git a/src/commands.cpp b/src/commands.cpp
index a53d9ee00..0797c3f19 100644
--- a/src/commands.cpp
+++ b/src/commands.cpp
@@ -71,14 +71,9 @@ using namespace std;
#include "dnsqueue.h"
#include "helperfuncs.h"
#include "hashcomp.h"
+#include "socketengine.h"
-#ifdef USE_KQUEUE
-extern int kq;
-#endif
-
-#ifdef USE_EPOLL
-int ep;
-#endif
+extern SocketEngine* SE;
extern int MODCOUNT;
extern std::vector<Module*> modules;
@@ -947,25 +942,7 @@ void handle_quit(char **parameters, int pcnt, userrec *user)
/* push the socket on a stack of sockets due to be closed at the next opportunity */
if (user->fd > -1)
{
-#ifdef USE_KQUEUE
- struct kevent ke;
- EV_SET(&ke, user->fd, EVFILT_READ, EV_DELETE, 0, 0, NULL);
- int i = kevent(kq, &ke, 1, 0, 0, NULL);
- if (i == -1)
- {
- log(DEBUG,"kqueue: Failed to remove user from queue!");
- }
-#endif
-#ifdef USE_EPOLL
- struct epoll_event ev;
- ev.events = EPOLLIN | EPOLLET;
- ev.data.fd = user->fd;
- int i = epoll_ctl(ep, EPOLL_CTL_DEL, user->fd, &ev);
- if (i < 0)
- {
- log(DEBUG,"epoll: List deletion failure!");
- }
-#endif
+ SE->DelFd(user->fd);
user->CloseSocket();
}
diff --git a/src/inspircd.cpp b/src/inspircd.cpp
index 87da59b57..5d10c088f 100644
--- a/src/inspircd.cpp
+++ b/src/inspircd.cpp
@@ -87,6 +87,7 @@ char list[MAXBUF];
char PrefixQuit[MAXBUF];
char DieValue[MAXBUF];
char DNSServer[MAXBUF];
+char data[65536];
int debugging = 0;
int WHOWAS_STALE = 48; // default WHOWAS Entries last 2 days before they go 'stale'
int WHOWAS_MAX = 100; // default 100 people maximum in the WHOWAS list
@@ -118,13 +119,7 @@ bool unlimitcore = false;
time_t TIME = time(NULL), OLDTIME = time(NULL);
-#ifdef USE_KQUEUE
-int kq, lkq, skq;
-#endif
-
-#ifdef USE_EPOLL
-int ep, lep, sep;
-#endif
+SocketEngine* SE = NULL;
bool has_been_netsplit = false;
extern std::vector<std::string> include_stack;
@@ -1118,7 +1113,7 @@ void kill_link(userrec *user,const char* r)
if (user->fd > -1)
{
FOREACH_MOD OnRawSocketClose(user->fd);
- engine_delete_fd;
+ SE->DelFd(user->fd);
user->CloseSocket();
}
@@ -1171,7 +1166,7 @@ void kill_link_silent(userrec *user,const char* r)
if (user->fd > -1)
{
FOREACH_MOD OnRawSocketClose(user->fd);
- engine_delete_fd;
+ SE->DelFd(user->fd);
user->CloseSocket();
}
@@ -1507,7 +1502,7 @@ void AddClient(int socket, char* host, int port, bool iscached, char* ip)
}
}
fd_ref_table[socket] = clientlist[tempnick];
- engine_add_fd;
+ SE->AddFd(socket,true,X_ESTAB_CLIENT);
// initialize their dns lookup thread
//if (pthread_create(&clientlist[tempnick]->dnsthread, NULL, dns_task, (void *)clientlist[tempnick]) != 0)
@@ -2280,45 +2275,262 @@ bool LoadModule(const char* filename)
}
-int InspIRCd(char** argv, int argc)
+void ProcessUser(userrec* cu)
{
- struct sockaddr_in client,server;
- char addrs[MAXBUF][255];
- int incomingSockfd, result = TRUE;
- socklen_t length;
- int count = 0;
-#ifdef USE_SELECT
- int selectResult = 0, selectResult2 = 0;
- fd_set selectFds;
-#endif
- char configToken[MAXBUF], Addr[MAXBUF], Type[MAXBUF];
- timeval tv;
+ int result = EAGAIN;
+ log(DEBUG,"Processing user with fd %d",cu->fd);
+ int MOD_RESULT = 0;
+ int result2 = 0;
+ FOREACH_RESULT(OnRawSocketRead(cu->fd,data,65535,result2));
+ if (!MOD_RESULT)
+ {
+ result = cu->ReadData(data, 65535);
+ }
+ else
+ {
+ log(DEBUG,"Data result returned by module: %d",MOD_RESULT);
+ result = result2;
+ }
+ log(DEBUG,"Read result: %d",result);
+ if (result)
+ {
+ statsRecv += result;
+ // perform a check on the raw buffer as an array (not a string!) to remove
+ // characters 0 and 7 which are illegal in the RFC - replace them with spaces.
+ // hopefully this should stop even more people whining about "Unknown command: *"
+ for (int checker = 0; checker < result; checker++)
+ {
+ if ((data[checker] == 0) || (data[checker] == 7))
+ data[checker] = ' ';
+ }
+ if (result > 0)
+ data[result] = '\0';
+ userrec* current = cu;
+ int currfd = current->fd;
+ int floodlines = 0;
+ // add the data to the users buffer
+ if (result > 0)
+ {
+ if (!current->AddBuffer(data))
+ {
+ // AddBuffer returned false, theres too much data in the user's buffer and theyre up to no good.
+ if (current->registered == 7)
+ {
+ kill_link(current,"RecvQ exceeded");
+ }
+ else
+ {
+ WriteOpers("*** Excess flood from %s",current->ip);
+ log(DEFAULT,"Excess flood from: %s",current->ip);
+ add_zline(120,ServerName,"Flood from unregistered connection",current->ip);
+ apply_lines();
+ }
+ return;
+ }
+ if (current->recvq.length() > (unsigned)NetBufferSize)
+ {
+ if (current->registered == 7)
+ {
+ kill_link(current,"RecvQ exceeded");
+ }
+ else
+ {
+ WriteOpers("*** Excess flood from %s",current->ip);
+ log(DEFAULT,"Excess flood from: %s",current->ip);
+ add_zline(120,ServerName,"Flood from unregistered connection",current->ip);
+ apply_lines();
+ }
+ return;
+ }
+ // while there are complete lines to process...
+ while (current->BufferIsReady())
+ {
+ floodlines++;
+ if (TIME > current->reset_due)
+ {
+ current->reset_due = TIME + current->threshold;
+ current->lines_in = 0;
+ }
+ current->lines_in++;
+ if (current->lines_in > current->flood)
+ {
+ log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
+ WriteOpers("*** Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
+ kill_link(current,"Excess flood");
+ return;
+ }
+ if ((floodlines > current->flood) && (current->flood != 0))
+ {
+ if (current->registered == 7)
+ {
+ log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
+ WriteOpers("*** Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
+ kill_link(current,"Excess flood");
+ }
+ else
+ {
+ add_zline(120,ServerName,"Flood from unregistered connection",current->ip);
+ apply_lines();
+ }
+ return;
+ }
+ char sanitized[MAXBUF];
+ // use GetBuffer to copy single lines into the sanitized string
+ std::string single_line = current->GetBuffer();
+ current->bytes_in += single_line.length();
+ current->cmds_in++;
+ if (single_line.length()>512)
+ {
+ log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
+ WriteOpers("*** Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
+ kill_link(current,"Excess flood");
+ return;
+ }
+ strlcpy(sanitized,single_line.c_str(),MAXBUF);
+ if (*sanitized)
+ {
+ userrec* old_comp = fd_ref_table[currfd];
+ // we're gonna re-scan to check if the nick is gone, after every
+ // command - if it has, we're gonna bail
+ process_buffer(sanitized,current);
+ // look for the user's record in case it's changed... if theyve quit,
+ // we cant do anything more with their buffer, so bail.
+ // there used to be an ugly, slow loop here. Now we have a reference
+ // table, life is much easier (and FASTER)
+ userrec* new_comp = fd_ref_table[currfd];
+ if ((currfd < 0) || (!fd_ref_table[currfd]) || (old_comp != new_comp))
+ return;
+ }
+ }
+ return;
+ }
+ if ((result == -1) && (errno != EAGAIN) && (errno != EINTR))
+ {
+ log(DEBUG,"killing: %s",cu->nick);
+ kill_link(cu,strerror(errno));
+ return;
+ }
+ }
+ // result EAGAIN means nothing read
+ else if (result == EAGAIN)
+ {
+ }
+ else if (result == 0)
+ {
+ log(DEBUG,"InspIRCd: Exited: %s",cu->nick);
+ kill_link(cu,"Client exited");
+ log(DEBUG,"Bailing from client exit");
+ return;
+ }
+}
+void DoBackgroundUserStuff()
+{
+ for (user_hash::iterator count2 = clientlist.begin(); count2 != clientlist.end(); count2++)
+ {
+ userrec* curr = NULL;
+ if (count2->second)
+ curr = count2->second;
+ if ((long)curr == -1)
+ return;
+
+ if ((curr) && (curr->fd != 0))
+ {
+ int currfd = curr->fd;
+ // we don't check the state of remote users.
+ if ((currfd != -1) && (currfd != FD_MAGIC_NUMBER))
+ {
+ curr->FlushWriteBuf();
+ if (curr->GetWriteError() != "")
+ {
+ log(DEBUG,"InspIRCd: write error: %s",curr->GetWriteError().c_str());
+ kill_link(curr,curr->GetWriteError().c_str());
+ return;
+ }
+ // registration timeout -- didnt send USER/NICK/HOST in the time specified in
+ // their connection class.
+ if (((unsigned)TIME > (unsigned)curr->timeout) && (curr->registered != 7))
+ {
+ log(DEBUG,"InspIRCd: registration timeout: %s",curr->nick);
+ kill_link(curr,"Registration timeout");
+ return;
+ }
+ if ((TIME > curr->signon) && (curr->registered == 3) && (AllModulesReportReady(curr)))
+ {
+ log(DEBUG,"signon exceed, registered=3, and modules ready, OK: %d %d",TIME,curr->signon);
+ curr->dns_done = true;
+ statsDnsBad++;
+ FullConnectUser(curr);
+ if (fd_ref_table[currfd] != curr) // something changed, bail pronto
+ return;
+ }
+ if ((curr->dns_done) && (curr->registered == 3) && (AllModulesReportReady(curr)))
+ {
+ log(DEBUG,"dns done, registered=3, and modules ready, OK");
+ FullConnectUser(curr);
+ if (fd_ref_table[currfd] != curr) // something changed, bail pronto
+ return;
+ }
+ if ((TIME > curr->nping) && (isnick(curr->nick)) && (curr->registered == 7))
+ {
+ if ((!curr->lastping) && (curr->registered == 7))
+ {
+ log(DEBUG,"InspIRCd: ping timeout: %s",curr->nick);
+ kill_link(curr,"Ping timeout");
+ return;
+ }
+ Write(curr->fd,"PING :%s",ServerName);
+ log(DEBUG,"InspIRCd: pinging: %s",curr->nick);
+ curr->lastping = 0;
+ curr->nping = TIME+curr->pingmax; // was hard coded to 120
+ }
+ }
+ }
+ }
+}
+
+void OpenLog(char** argv, int argc)
+{
std::string logpath = GetFullProgDir(argv,argc) + "/ircd.log";
- log_file = fopen(logpath.c_str(),"a+");
- if (!log_file)
- {
- printf("ERROR: Could not write to logfile %s, bailing!\n\n",logpath.c_str());
- Exit(ERROR);
+ log_file = fopen(logpath.c_str(),"a+");
+ if (!log_file)
+ {
+ printf("ERROR: Could not write to logfile %s, bailing!\n\n",logpath.c_str());
+ Exit(ERROR);
}
-
#ifdef IS_CYGWIN
- printf("Logging to ircd.log...\n");
+ printf("Logging to ircd.log...\n");
#else
- printf("Logging to %s...\n",logpath.c_str());
+ printf("Logging to %s...\n",logpath.c_str());
#endif
+}
- log(DEFAULT,"$Id$");
+void CheckRoot()
+{
if (geteuid() == 0)
{
printf("WARNING!!! You are running an irc server as ROOT!!! DO NOT DO THIS!!!\n\n");
- Exit(ERROR);
log(DEFAULT,"InspIRCd: startup: not starting with UID 0!");
+ Exit(ERROR);
}
+}
+
+int InspIRCd(char** argv, int argc)
+{
+ struct sockaddr_in client,server;
+ char addrs[MAXBUF][255];
+ int incomingSockfd;
+ socklen_t length;
+ int count = 0;
+ int clientportcount = 0;
+ char configToken[MAXBUF], Addr[MAXBUF], Type[MAXBUF];
+
+ OpenLog(argv, argc);
+ CheckRoot();
SetupCommandTable();
- log(DEBUG,"InspIRCd: startup: default command table set up");
-
ReadConfig(true,NULL);
+ AddServerName(ServerName);
+
if (DieValue[0])
{
printf("WARNING: %s\n\n",DieValue);
@@ -2327,10 +2539,6 @@ int InspIRCd(char** argv, int argc)
}
log(DEBUG,"InspIRCd: startup: read config");
- AddServerName(ServerName);
-
- int clientportcount = 0;
-
for (count = 0; count < ConfValueEnum("bind",&config_f); count++)
{
ConfValue("bind","port",count,configToken,&config_f);
@@ -2349,10 +2557,9 @@ int InspIRCd(char** argv, int argc)
log(DEBUG,"InspIRCd: startup: read binding %s:%s [%s] from config",Addr,configToken, Type);
}
portCount = clientportcount;
-
+
log(DEBUG,"InspIRCd: startup: read %lu total client ports",(unsigned long)portCount);
- log(DEBUG,"InspIRCd: startup: InspIRCd is now starting!");
-
+
printf("\n");
/* BugFix By Craig! :p */
@@ -2376,12 +2583,7 @@ int InspIRCd(char** argv, int argc)
ConfValue("pid","file",0,PID,&config_f);
// write once here, to try it out and make sure its ok
WritePID(PID);
-
- /* setup select call */
-#ifdef USE_SELECT
- FD_ZERO(&selectFds);
-#endif
- log(DEBUG,"InspIRCd: startup: zero selects");
+
log(VERBOSE,"InspIRCd: startup: portCount = %lu", (unsigned long)portCount);
for (count = 0; count < portCount; count++)
@@ -2428,37 +2630,25 @@ int InspIRCd(char** argv, int argc)
}
}
- engine_init;
- engine_server_fill;
+ SE = new SocketEngine();
- WritePID(PID);
+ /* Add the listening sockets used for client inbound connections
+ * to the socket engine
+ */
+ for (count = 0; count < portCount; count++)
+ SE->AddFd(openSockfd[count],true,X_LISTEN);
- length = sizeof (client);
- engine_structs;
- timeval tvs;
- tvs.tv_usec = 10000L;
- tvs.tv_sec = 0;
- tv.tv_sec = 0;
- tv.tv_usec = 10000L;
- char data[65536];
- timeval tval;
- tval.tv_usec = 10000L;
- tval.tv_sec = 0;
- int total_in_this_set = 0;
- int i = 0, v = 0, j = 0, cycle_iter = 0;
+ std::vector<int> activefds;
+
+ WritePID(PID);
bool expire_run = false;
- cycle_iter = i = j = 0; // stop the compiler whining - these may be marked unused in certain socket engine types
-
/* main loop, this never returns */
for (;;)
{
#ifdef _POSIX_PRIORITY_SCHEDULING
sched_yield();
#endif
-#ifdef USE_SELECT
- FD_ZERO(&sfd);
-#endif
// we only read time() once per iteration rather than tons of times!
OLDTIME = TIME;
TIME = time(NULL);
@@ -2495,380 +2685,51 @@ int InspIRCd(char** argv, int argc)
if ((TIME % 5) == 1)
expire_run = false;
- // fix by brain - this must be below any manipulation of the hashmap by modules
- user_hash::iterator count2 = clientlist.begin();
+ DoBackgroundUserStuff();
+ SE->Wait(activefds);
- while (count2 != clientlist.end())
+ for (unsigned int activefd = 0; activefd < activefds.size(); activefd++)
{
-#ifdef USE_SELECT
- FD_ZERO(&sfd);
-#endif
-
- total_in_this_set = 0;
-
- user_hash::iterator xcount = count2;
- user_hash::iterator endingiter = count2;
-
- if (count2 == clientlist.end()) break;
-
- userrec* curr = NULL;
-
- if (count2->second)
- curr = count2->second;
-
- if ((long)curr == -1)
- goto label;
-
- if ((curr) && (curr->fd != 0))
+ userrec* cu = fd_ref_table[activefds[activefd]];
+ if (cu)
{
-#ifdef _POSIX_PRIORITY_SCHEDULING
- sched_yield();
-#endif
- // assemble up to 64 sockets into an fd_set
- // to implement a pooling mechanism.
- //
- // This should be up to 64x faster than the
- // old implementation.
-#ifdef USE_SELECT
- while (total_in_this_set < 1024)
- {
- if (count2 != clientlist.end())
- {
- curr = count2->second;
- if ((long)curr == -1)
- goto label;
- int currfd = curr->fd;
- // we don't check the state of remote users.
- if ((currfd != -1) && (currfd != FD_MAGIC_NUMBER))
- {
- curr->FlushWriteBuf();
- if (curr->GetWriteError() != "")
- {
- log(DEBUG,"InspIRCd: write error: %s",curr->GetWriteError().c_str());
- kill_link(curr,curr->GetWriteError().c_str());
- goto label;
- }
-
- FD_SET (curr->fd, &sfd);
-
- // registration timeout -- didnt send USER/NICK/HOST in the time specified in
- // their connection class.
- if (((unsigned)TIME > (unsigned)curr->timeout) && (curr->registered != 7))
- {
- log(DEBUG,"InspIRCd: registration timeout: %s",curr->nick);
- kill_link(curr,"Registration timeout");
- goto label;
- }
- if ((TIME > curr->signon) && (curr->registered == 3) && (AllModulesReportReady(curr)))
- {
- log(DEBUG,"signon exceed, registered=3, and modules ready, OK");
- curr->dns_done = true;
- statsDnsBad++;
- FullConnectUser(curr);
- if (fd_ref_table[currfd] != curr) // something changed, bail pronto
- goto label;
- }
- if ((curr->dns_done) && (curr->registered == 3) && (AllModulesReportReady(curr))) // both NICK and USER... and DNS
- {
- log(DEBUG,"dns done, registered=3, and modules ready, OK");
- FullConnectUser(curr);
- if (fd_ref_table[currfd] != curr) // something changed, bail pronto
- goto label;
- }
- if ((TIME > curr->nping) && (isnick(curr->nick)) && (curr->registered == 7))
- {
- if ((!curr->lastping) && (curr->registered == 7))
- {
- log(DEBUG,"InspIRCd: ping timeout: %s",curr->nick);
- kill_link(curr,"Ping timeout");
- goto label;
- }
- Write(curr->fd,"PING :%s",ServerName);
- log(DEBUG,"InspIRCd: pinging: %s",curr->nick);
- curr->lastping = 0;
- curr->nping = TIME+curr->pingmax; // was hard coded to 120
- }
- }
- count2++;
- total_in_this_set++;
- }
- else break;
- }
- endingiter = count2;
- count2 = xcount; // roll back to where we were
-#else
- // KQUEUE and EPOLL: We don't go through a loop to fill the fd_set so instead we must manually do this loop every now and again.
- // TODO: We dont need to do all this EVERY loop iteration, tone down the visits to this if we're using kqueue.
- cycle_iter++;
- if (cycle_iter > 20) while (count2 != clientlist.end())
+ /* It's a user */
+ ProcessUser(cu);
+ }
+ else
+ {
+ /* It maybe a listener */
+ for (count = 0; count < boundPortCount; count++)
{
- cycle_iter = 0;
- if (count2 != clientlist.end())
+ if (activefds[activefd] == openSockfd[count])
{
- curr = count2->second;
- if ((long)curr == -1)
- goto label;
- int currfd = curr->fd;
- // we don't check the state of remote users.
- if ((currfd != -1) && (currfd != FD_MAGIC_NUMBER))
+ char target[MAXBUF], resolved[MAXBUF];
+ length = sizeof (client);
+ incomingSockfd = accept (openSockfd[count], (struct sockaddr *) &client, &length);
+ log(DEBUG,"Accepted socket %d",incomingSockfd);
+ strlcpy (target, (char *) inet_ntoa (client.sin_addr), MAXBUF);
+ strlcpy (resolved, target, MAXBUF);
+ if (incomingSockfd >= 0)
{
- curr->FlushWriteBuf();
- if (curr->GetWriteError() != "")
- {
- log(DEBUG,"InspIRCd: write error: %s",curr->GetWriteError().c_str());
- kill_link(curr,curr->GetWriteError().c_str());
- goto label;
- }
-
- // registration timeout -- didnt send USER/NICK/HOST in the time specified in
- // their connection class.
- if (((unsigned)TIME > (unsigned)curr->timeout) && (curr->registered != 7))
- {
- log(DEBUG,"InspIRCd: registration timeout: %s",curr->nick);
- kill_link(curr,"Registration timeout");
- goto label;
-
- }
- if ((TIME > curr->signon) && (curr->registered == 3) && (AllModulesReportReady(curr)))
- {
- log(DEBUG,"signon exceed, registered=3, and modules ready, OK: %d %d",TIME,curr->signon);
- curr->dns_done = true;
- statsDnsBad++;
- FullConnectUser(curr);
- if (fd_ref_table[currfd] != curr) // something changed, bail pronto
- goto label;
- }
- if ((curr->dns_done) && (curr->registered == 3) && (AllModulesReportReady(curr)))
- {
- log(DEBUG,"dns done, registered=3, and modules ready, OK");
- FullConnectUser(curr);
- if (fd_ref_table[currfd] != curr) // something changed, bail pronto
- goto label;
- }
- if ((TIME > curr->nping) && (isnick(curr->nick)) && (curr->registered == 7))
- {
- if ((!curr->lastping) && (curr->registered == 7))
- {
- log(DEBUG,"InspIRCd: ping timeout: %s",curr->nick);
- kill_link(curr,"Ping timeout");
- goto label;
- }
- Write(curr->fd,"PING :%s",ServerName);
- log(DEBUG,"InspIRCd: pinging: %s",curr->nick);
- curr->lastping = 0;
- curr->nping = TIME+curr->pingmax; // was hard coded to 120
- }
- }
- }
- else break;
- count2++;
- }
- // increment the counter right to the end of the list, as kqueue processes everything in one go
-#endif
-
- v = 0;
- engine_fill;
-
-#ifdef _POSIX_PRIORITY_SCHEDULING
- sched_yield();
-#endif
- result = EAGAIN;
- if (engine_check)
- {
- log(DEBUG,"Data waiting on socket %d",cu->fd);
- int MOD_RESULT = 0;
- int result2 = 0;
- FOREACH_RESULT(OnRawSocketRead(cu->fd,data,65535,result2));
- if (!MOD_RESULT)
- {
- result = cu->ReadData(data, 65535);
+ FOREACH_MOD OnRawSocketAccept(incomingSockfd, resolved, ports[count]);
+ statsAccept++;
+ AddClient(incomingSockfd, resolved, ports[count], false, inet_ntoa (client.sin_addr));
+ log(DEBUG,"Adding client on port %lu fd=%lu",(unsigned long)ports[count],(unsigned long)incomingSockfd);
}
else
{
- log(DEBUG,"Data result returned by module: %d",MOD_RESULT);
- result = result2;
+ WriteOpers("*** WARNING: accept() failed on port %lu (%s)",(unsigned long)ports[count],target);
+ log(DEBUG,"accept failed: %lu",(unsigned long)ports[count]);
+ statsRefused++;
}
- log(DEBUG,"Read result: %d",result);
- if (result)
- {
- statsRecv += result;
- // perform a check on the raw buffer as an array (not a string!) to remove
- // characters 0 and 7 which are illegal in the RFC - replace them with spaces.
- // hopefully this should stop even more people whining about "Unknown command: *"
- for (int checker = 0; checker < result; checker++)
- {
- if ((data[checker] == 0) || (data[checker] == 7))
- data[checker] = ' ';
- }
- if (result > 0)
- data[result] = '\0';
- userrec* current = cu;
- int currfd = current->fd;
- int floodlines = 0;
- // add the data to the users buffer
- if (result > 0)
- if (!current->AddBuffer(data))
- {
- // AddBuffer returned false, theres too much data in the user's buffer and theyre up to no good.
- if (current->registered == 7)
- {
- kill_link(current,"RecvQ exceeded");
- }
- else
- {
- WriteOpers("*** Excess flood from %s",current->ip);
- log(DEFAULT,"Excess flood from: %s",current->ip);
- add_zline(120,ServerName,"Flood from unregistered connection",current->ip);
- apply_lines();
- }
- goto label;
- }
- if (current->recvq.length() > (unsigned)NetBufferSize)
- {
- if (current->registered == 7)
- {
- kill_link(current,"RecvQ exceeded");
- }
- else
- {
- WriteOpers("*** Excess flood from %s",current->ip);
- log(DEFAULT,"Excess flood from: %s",current->ip);
- add_zline(120,ServerName,"Flood from unregistered connection",current->ip);
- apply_lines();
- }
- goto label;
- }
- // while there are complete lines to process...
- while (current->BufferIsReady())
- {
- floodlines++;
- if (TIME > current->reset_due)
- {
- current->reset_due = TIME + current->threshold;
- current->lines_in = 0;
- }
- current->lines_in++;
- if (current->lines_in > current->flood)
- {
- log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
- WriteOpers("*** Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
- kill_link(current,"Excess flood");
- goto label;
- }
- if ((floodlines > current->flood) && (current->flood != 0))
- {
- if (current->registered == 7)
- {
- log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
- WriteOpers("*** Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
- kill_link(current,"Excess flood");
- }
- else
- {
- add_zline(120,ServerName,"Flood from unregistered connection",current->ip);
- apply_lines();
- }
- goto label;
- }
- char sanitized[MAXBUF];
- // use GetBuffer to copy single lines into the sanitized string
- std::string single_line = current->GetBuffer();
- current->bytes_in += single_line.length();
- current->cmds_in++;
- if (single_line.length()>512)
- {
- log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
- WriteOpers("*** Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
- kill_link(current,"Excess flood");
- goto label;
- }
- strlcpy(sanitized,single_line.c_str(),MAXBUF);
- if (*sanitized)
- {
- userrec* old_comp = fd_ref_table[currfd];
- // we're gonna re-scan to check if the nick is gone, after every
- // command - if it has, we're gonna bail
- process_buffer(sanitized,current);
- // look for the user's record in case it's changed... if theyve quit,
- // we cant do anything more with their buffer, so bail.
- // there used to be an ugly, slow loop here. Now we have a reference
- // table, life is much easier (and FASTER)
- userrec* new_comp = fd_ref_table[currfd];
- if ((currfd < 0) || (!fd_ref_table[currfd]) || (old_comp != new_comp))
- goto label;
-
- }
- }
- goto label;
- }
-
- if ((result == -1) && (errno != EAGAIN) && (errno != EINTR))
- {
- log(DEBUG,"killing: %s",cu->nick);
- kill_link(cu,strerror(errno));
- goto label;
- }
- }
- // result EAGAIN means nothing read
- if (result == EAGAIN)
- {
- }
- else
- if (result == 0)
- {
- engine_cleanup;
- }
- else if (result > 0)
- {
}
}
}
- for (int q = 0; q < total_in_this_set; q++)
- {
- count2++;
- }
}
-
-#ifdef _POSIX_PRIORITY_SCHEDULING
- sched_yield();
-#endif
-
- engine_scanset;
- char target[MAXBUF], resolved[MAXBUF];
- length = sizeof (client);
- incomingSockfd = accept (openSockfd[count], (struct sockaddr *) &client, &length);
- log(DEBUG,"Accepted socket %d",incomingSockfd);
-
- strlcpy (target, (char *) inet_ntoa (client.sin_addr), MAXBUF);
- strlcpy (resolved, target, MAXBUF);
-
- if (incomingSockfd < 0)
- {
- WriteOpers("*** WARNING: Accept failed on port %lu (%s)",(unsigned long)ports[count],target);
- log(DEBUG,"InspIRCd: accept failed: %lu",(unsigned long)ports[count]);
- statsRefused++;
- }
- else
- {
- FOREACH_MOD OnRawSocketAccept(incomingSockfd, resolved, ports[count]);
- statsAccept++;
- AddClient(incomingSockfd, resolved, ports[count], false, inet_ntoa (client.sin_addr));
- log(DEBUG,"InspIRCd: adding client on port %lu fd=%lu",(unsigned long)ports[count],(unsigned long)incomingSockfd);
- }
- }
- }
}
- label:
- if (0) {};
-#ifdef _POSIX_PRIORITY_SCHEDULING
- sched_yield();
- sched_yield();
-#endif
-}
-/* not reached */
-close (incomingSockfd);
-return 0;
+ /* This is never reached -- we hope! */
+ return 0;
}
diff --git a/src/modules.cpp b/src/modules.cpp
index 3c930cea3..6afc36b47 100644
--- a/src/modules.cpp
+++ b/src/modules.cpp
@@ -58,15 +58,9 @@ using namespace std;
#include "helperfuncs.h"
#include "hashcomp.h"
#include "socket.h"
+#include "socketengine.h"
-#ifdef USE_KQUEUE
-extern int kq;
-#endif
-
-#ifdef USE_EPOLL
-int ep;
-#endif
-
+extern SocketEngine* SE;
extern int MODCOUNT;
extern std::vector<Module*> modules;
extern std::vector<ircd_module*> factory;
@@ -678,26 +672,7 @@ bool Server::UserToPseudo(userrec* user,std::string message)
user->fd = FD_MAGIC_NUMBER;
user->ClearBuffer();
Write(old_fd,"ERROR :Closing link (%s@%s) [%s]",user->ident,user->host,message.c_str());
-#ifdef USE_KQUEUE
- struct kevent ke;
- EV_SET(&ke, old_fd, EVFILT_READ, EV_DELETE, 0, 0, NULL);
- int i = kevent(kq, &ke, 1, 0, 0, NULL);
- if (i == -1)
- {
- log(DEBUG,"kqueue: Failed to remove user from queue!");
- }
-#endif
-#ifdef USE_EPOLL
- struct epoll_event ev;
- ev.events = EPOLLIN | EPOLLET;
- ev.data.fd = old_fd;
- int i = epoll_ctl(ep, EPOLL_CTL_DEL, old_fd, &ev);
- if (i < 0)
- {
- log(DEBUG,"epoll: List deletion failure!");
- }
-#endif
-
+ SE->DelFd(old_fd);
shutdown(old_fd,2);
close(old_fd);
return true;
diff --git a/src/socketengine.cpp b/src/socketengine.cpp
index 7501d19db..0c816943e 100644
--- a/src/socketengine.cpp
+++ b/src/socketengine.cpp
@@ -16,15 +16,9 @@
char ref[65535];
-const char X_LISTEN = 0;
-const char X_ESTAB_CLIENT = 1;
-const char X_ESTAB_MODULE = 2;
-const char X_ESTAB_DNS = 3;
-
-const char X_READBIT = 0x80;
-
SocketEngine::SocketEngine()
{
+ log(DEBUG,"SocketEngine::SocketEngine()");
#ifdef USE_EPOLL
EngineHandle = epoll_create(65535);
#endif
@@ -35,6 +29,7 @@ SocketEngine::SocketEngine()
SocketEngine::~SocketEngine()
{
+ log(DEBUG,"SocketEngine::~SocketEngine()");
#ifdef USE_EPOLL
close(EngineHandle);
#endif
@@ -50,7 +45,7 @@ bool SocketEngine::AddFd(int fd, bool readable, char type)
ref[fd] |= X_READBIT;
#ifdef USE_EPOLL
struct epoll_event ev;
- log(DEBUG,"epoll: Adduser to events, ep=%d socket=%d",EngineHandle,fd);
+ log(DEBUG,"epoll: Add socket to events, ep=%d socket=%d",EngineHandle,fd);
readable ? ev.events = EPOLLIN | EPOLLET : ev.events = EPOLLOUT | EPOLLET;
ev.data.fd = fd;
int i = epoll_ctl(EngineHandle, EPOLL_CTL_ADD, fd, &ev);
@@ -62,7 +57,7 @@ bool SocketEngine::AddFd(int fd, bool readable, char type)
#endif
#ifdef USE_KQUEUE
struct kevent ke;
- log(DEBUG,"kqueue: Add user to events, kq=%d socket=%d",EngineHandle,fd);
+ log(DEBUG,"kqueue: Add socket to events, kq=%d socket=%d",EngineHandle,fd);
EV_SET(&ke, fd, readable ? EVFILT_READ : EVFILT_WRITE, EV_ADD, 0, 0, NULL);
int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
if (i == -1)
@@ -76,6 +71,7 @@ return true;
bool SocketEngine::DelFd(int fd)
{
+ log(DEBUG,"SocketEngine::DelFd(%d)",fd);
bool found = false;
for (std::vector<int>::iterator i = fds.begin(); i != fds.end(); i++)
{
@@ -83,6 +79,7 @@ bool SocketEngine::DelFd(int fd)
{
fds.erase(i);
found = true;
+ break;
}
}
#ifdef USE_KQUEUE
@@ -91,7 +88,7 @@ bool SocketEngine::DelFd(int fd)
int i = kevent(EngineHandle, &ke, 1, 0, 0, NULL);
if (i == -1)
{
- log(DEBUG,"kqueue: Failed to remove user from queue!");
+ log(DEBUG,"kqueue: Failed to remove socket from queue!");
return false;
}
#endif
@@ -114,7 +111,8 @@ bool SocketEngine::Wait(std::vector<int> &fdlist)
{
fdlist.clear();
#ifdef USE_SELECT
- int highest_fd = 0;
+ FD_ZERO(&wfdset);
+ FD_ZERO(&rfdset);
timeval tval;
int sresult;
for (unsigned int a = 0; a < fds.size(); a++)