diff options
author | brain <brain@e03df62e-2008-0410-955e-edbf42e46eb7> | 2005-12-12 11:50:45 +0000 |
---|---|---|
committer | brain <brain@e03df62e-2008-0410-955e-edbf42e46eb7> | 2005-12-12 11:50:45 +0000 |
commit | cef3e32fae0132d51b2fcca16ef05907b174e2a3 (patch) | |
tree | 6c56f7d3401068ee3e16486bcbd54a9151bc4b05 | |
parent | 4f2e601fe71febf62db0f98a418a25598e24377b (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.h | 7 | ||||
-rw-r--r-- | src/commands.cpp | 29 | ||||
-rw-r--r-- | src/inspircd.cpp | 713 | ||||
-rw-r--r-- | src/modules.cpp | 31 | ||||
-rw-r--r-- | src/socketengine.cpp | 20 |
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++) |