X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fuserprocess.cpp;h=aaaf137105223923b1d536755162b7fa03f9fd0f;hb=1b7c615062a7b203c7fc3ce4c56e16eb671f7c15;hp=fd1976a222b254f109abc8a24412a93f14b83cb2;hpb=d0f0cbea0727e3a3f670a308b173559061a0ca1d;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/userprocess.cpp b/src/userprocess.cpp index fd1976a22..aaaf13710 100644 --- a/src/userprocess.cpp +++ b/src/userprocess.cpp @@ -2,7 +2,7 @@ * | Inspire Internet Relay Chat Daemon | * +------------------------------------+ * - * Inspire is copyright (C) 2002-2005 ChatSpike-Dev. + * InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev. * E-mail: * * @@ -18,372 +18,309 @@ using namespace std; -#include "inspircd_config.h" -#include "inspircd.h" -#include "inspircd_io.h" -#include "inspircd_util.h" -#include -#include -#include -#include -#include -#include -#include -#ifdef GCC3 -#include -#else -#include -#endif -#include -#include -#include -#include -#include -#ifdef THREADED_DNS -#include -#endif +#include "configreader.h" #include "users.h" -#include "ctables.h" -#include "globals.h" #include "modules.h" -#include "dynamic.h" #include "wildcard.h" -#include "message.h" -#include "mode.h" -#include "commands.h" #include "xline.h" -#include "inspstring.h" -#include "dnsqueue.h" -#include "helperfuncs.h" -#include "hashcomp.h" #include "socketengine.h" -#include "typedefs.h" +#include "userprocess.h" +#include "inspircd.h" +#include "command_parse.h" +#include "cull_list.h" -extern int MODCOUNT; -extern struct sockaddr_in client,server; -extern socklen_t length; -extern std::vector modules; -extern std::vector factory; -extern time_t TIME; -extern time_t OLDTIME; +void InspIRCd::ProcessUser(userrec* cu) +{ + int result = EAGAIN; -extern InspIRCd* ServerInstance; -extern SocketEngine* SE; -extern serverstats* stats; -extern ServerConfig *Config; + if (cu->GetFd() == FD_MAGIC_NUMBER) + return; -extern userrec* fd_ref_table[65536]; -char data[65536]; + if (this->Config->GetIOHook(cu->GetPort())) + { + int result2 = 0; + int MOD_RESULT = 0; -extern user_hash clientlist; -extern chan_hash chanlist; + try + { + MOD_RESULT = this->Config->GetIOHook(cu->GetPort())->OnRawSocketRead(cu->GetFd(),ReadBuffer,sizeof(ReadBuffer),result2); + this->Log(DEBUG,"Data result returned by module: %d",MOD_RESULT); + } + catch (ModuleException& modexcept) + { + this->Log(DEBUG,"Module exception caught: %s",modexcept.GetReason()); + } -void ProcessUser(userrec* cu) -{ - 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) - { - stats->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,Config->ServerName,"Flood from unregistered connection",current->ip); - apply_lines(APPLY_ZLINES); - } - return; - } - if (current->recvq.length() > (unsigned)Config->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,Config->ServerName,"Flood from unregistered connection",current->ip); - apply_lines(APPLY_ZLINES); - } - 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,Config->ServerName,"Flood from unregistered connection",current->ip); - apply_lines(APPLY_ZLINES); - } - 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; - } - else - { - /* The user is still here, flush their buffer */ - current->FlushWriteBuf(); - } - } - } - 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; - } -} + if (MOD_RESULT < 0) + { + result = -EAGAIN; + } + else + { + result = result2; + } + } + else + { + result = cu->ReadData(ReadBuffer, sizeof(ReadBuffer)); + } + + this->Log(DEBUG,"Read result: %d",result); + + if ((result) && (result != -EAGAIN)) + { + userrec *current; + int currfd; + int floodlines = 0; + + this->stats->statsRecv += result; + /* + * perform a check on the raw buffer as an array (not a string!) to remove + * character 0 which is illegal in the RFC - replace them with spaces. + * XXX - no garauntee there's not \0's in the middle of the data, + * and no reason for it to be terminated either. -- Om + */ + + for (int checker = 0; checker < result; checker++) + { + if (ReadBuffer[checker] == 0) + ReadBuffer[checker] = ' '; + } + + if (result > 0) + ReadBuffer[result] = '\0'; + + current = cu; + currfd = current->GetFd(); + + // add the data to the users buffer + if (result > 0) + { + if (!current->AddBuffer(ReadBuffer)) + { + // AddBuffer returned false, theres too much data in the user's buffer and theyre up to no good. + if (current->registered == REG_ALL) + { + // Make sure they arn't flooding long lines. + if (TIME > current->reset_due) + { + current->reset_due = TIME + current->threshold; + current->lines_in = 0; + } + + current->lines_in++; + + if (current->lines_in > current->flood) + { + this->Log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host); + this->SNO->WriteToSnoMask('f',"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host); + userrec::QuitUser(this, current,"Excess flood"); + return; + } + else + { + current->WriteServ("NOTICE %s :Your previous line was too long and was not delivered (Over 512chars) Please shorten it.", current->nick); + current->recvq = ""; + } + } + else + { + this->WriteOpers("*** Excess flood from %s",current->GetIPString()); + this->SNO->WriteToSnoMask('f',"Excess flood from: %s",current->GetIPString()); + XLines->add_zline(120,this->Config->ServerName,"Flood from unregistered connection",current->GetIPString()); + XLines->apply_lines(APPLY_ZLINES); + } + + return; + } + + if (current->recvq.length() > (unsigned)this->Config->NetBufferSize) + { + if (current->registered == REG_ALL) + { + userrec::QuitUser(this, current,"RecvQ exceeded"); + } + else + { + this->WriteOpers("*** Excess flood from %s",current->GetIPString()); + this->SNO->WriteToSnoMask('f',"Excess flood from: %s",current->GetIPString()); + XLines->add_zline(120,this->Config->ServerName,"Flood from unregistered connection",current->GetIPString()); + XLines->apply_lines(APPLY_ZLINES); + } + + return; + } + + // while there are complete lines to process... + while (current->BufferIsReady()) + { + if (TIME > current->reset_due) + { + current->reset_due = TIME + current->threshold; + current->lines_in = 0; + } + + if (++current->lines_in > current->flood) + { + this->Log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host); + this->SNO->WriteToSnoMask('f',"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host); + userrec::QuitUser(this, current,"Excess flood"); + return; + } + if ((++floodlines > current->flood) && (current->flood != 0)) + { + if (current->registered == REG_ALL) + { + this->Log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host); + SNO->WriteToSnoMask('f',"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host); + userrec::QuitUser(this, current,"Excess flood"); + } + else + { + XLines->add_zline(120,this->Config->ServerName,"Flood from unregistered connection",current->GetIPString()); + XLines->apply_lines(APPLY_ZLINES); + } + + return; + } + + // 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) + single_line.resize(512); + + EventHandler* old_comp = this->SE->GetRef(currfd); + + this->Parser->ProcessBuffer(single_line,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) + */ + EventHandler* new_comp = this->SE->GetRef(currfd); + if (old_comp != new_comp) + { + return; + } + else + { + /* The user is still here, flush their buffer */ + current->FlushWriteBuf(); + } + } + + return; + } + + if ((result == -1) && (errno != EAGAIN) && (errno != EINTR)) + { + this->Log(DEBUG,"killing: %s",cu->nick); + userrec::QuitUser(this,cu,strerror(errno)); + return; + } + } + + // result EAGAIN means nothing read + else if ((result == EAGAIN) || (result == -EAGAIN)) + { + /* do nothing */ + } + else if (result == 0) + { + this->Log(DEBUG,"InspIRCd: Exited: %s",cu->nick); + userrec::QuitUser(this,cu,"Client exited"); + this->Log(DEBUG,"Bailing from client exit"); + return; + } +} /** * This function is called once a second from the mainloop. * It is intended to do background checking on all the user structs, e.g. - * stuff like ping checks, registration timeouts, etc. - * The function returns false when it is finished, and true if - * it needs to be run again (e.g. it has processed one of a batch of - * QUIT messages, but couldnt continue iterating because the iterator - * became invalid). This function is also responsible for checking - * if InspSocket derived classes are timed out. + * stuff like ping checks, registration timeouts, etc. This function is + * also responsible for checking if InspSocket derived classes are timed out. */ -bool DoBackgroundUserStuff(time_t TIME) +void InspIRCd::DoBackgroundUserStuff(time_t TIME) { - unsigned int numsockets = ServerInstance->module_sockets.size(); - for (std::vector::iterator a = ServerInstance->module_sockets.begin(); a < ServerInstance->module_sockets.end(); a++) - { - InspSocket* s = (InspSocket*)*a; - if (s->Timeout(TIME)) - { - log(DEBUG,"Socket poll returned false, close and bail"); - SE->DelFd(s->GetFd()); - s->Close(); - ServerInstance->module_sockets.erase(a); - delete s; - break; - } - if (ServerInstance->module_sockets.size() != numsockets) break; - } - /* TODO: We need a seperate hash containing only local users for this - */ - for (user_hash::iterator count2 = clientlist.begin(); count2 != clientlist.end(); count2++) - { - /* Sanity checks for corrupted iterators (yes, really) */ - userrec* curr = NULL; - if (count2->second) - curr = count2->second; - if ((long)curr == -1) - return false; - - if ((curr) && (curr->fd != 0)) /* XXX - why are we checking fd != 0? --w00t */ - { - int currfd = curr->fd; - // we don't check the state of remote users. - if (IS_LOCAL(curr)) - { - curr->FlushWriteBuf(); - if (curr->GetWriteError() != "") - { - log(DEBUG,"InspIRCd: write error: %s",curr->GetWriteError().c_str()); - kill_link(curr,curr->GetWriteError().c_str()); - return true; - } - // 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 true; - } - 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; - stats->statsDnsBad++; - FullConnectUser(curr); - if (fd_ref_table[currfd] != curr) // something changed, bail pronto - return true; - } - 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 true; - } - 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 true; - } - Write(curr->fd,"PING :%s",Config->ServerName); - log(DEBUG,"InspIRCd: pinging: %s",curr->nick); - curr->lastping = 0; - curr->nping = TIME+curr->pingmax; // was hard coded to 120 - } - } - } - } - return false; -} + CullList GlobalGoners(this); -void OpenLog(char** argv, int argc) -{ - std::string logpath = GetFullProgDir(argv,argc) + "/ircd.log"; - Config->log_file = fopen(logpath.c_str(),"a+"); - if (!Config->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"); -#else - printf("Logging to %s...\n",logpath.c_str()); -#endif -} + /* XXX: IT IS NOT SAFE TO USE AN ITERATOR HERE. DON'T EVEN THINK ABOUT IT. */ + for (unsigned long count2 = 0; count2 != this->local_users.size(); count2++) + { + if (count2 >= this->local_users.size()) + break; + userrec* curr = this->local_users[count2]; -void CheckRoot() -{ - if (geteuid() == 0) - { - printf("WARNING!!! You are running an irc server as ROOT!!! DO NOT DO THIS!!!\n\n"); - log(DEFAULT,"InspIRCd: startup: not starting with UID 0!"); - Exit(ERROR); - } -} + if (curr) + { + /* + * registration timeout -- didnt send USER/NICK/HOST + * in the time specified in their connection class. + */ + if (((unsigned)TIME > (unsigned)curr->timeout) && (curr->registered != REG_ALL)) + { + this->Log(DEBUG,"InspIRCd: registration timeout: %s",curr->nick); + //ZapThisDns(curr->fd); + GlobalGoners.AddItem(curr,"Registration timeout"); + continue; + } + /* + * user has signed on with USER/NICK/PASS, and dns has completed, all the modules + * say this user is ok to proceed, fully connect them. + */ + if ((TIME > curr->signon) && (curr->registered == REG_NICKUSER) && (AllModulesReportReady(curr))) + { + curr->dns_done = true; + //ZapThisDns(curr->fd); + this->stats->statsDnsBad++; + curr->FullConnect(&GlobalGoners); + continue; + } + if ((curr->dns_done) && (curr->registered == REG_NICKUSER) && (AllModulesReportReady(curr))) + { + this->Log(DEBUG,"dns done, registered=3, and modules ready, OK"); + curr->FullConnect(&GlobalGoners); + //ZapThisDns(curr->fd); + continue; + } + // It's time to PING this user. Send them a ping. + if ((TIME > curr->nping) && (curr->registered == REG_ALL)) + { + // This user didn't answer the last ping, remove them + if (!curr->lastping) + { + GlobalGoners.AddItem(curr,"Ping timeout"); + curr->lastping = 1; + curr->nping = TIME+curr->pingmax; + continue; + } + curr->Write("PING :%s",this->Config->ServerName); + curr->lastping = 0; + curr->nping = TIME+curr->pingmax; + } + /* + * We can flush the write buffer as the last thing we do, because if they + * match any of the above conditions its no use flushing their buffer anyway. + */ + + curr->FlushWriteBuf(); + if (*curr->GetWriteError()) + { + GlobalGoners.AddItem(curr,curr->GetWriteError()); + continue; + } + } -void CheckDie() -{ - if (*Config->DieValue) - { - printf("WARNING: %s\n\n",Config->DieValue); - log(DEFAULT,"Ut-Oh, somebody didn't read their config file: '%s'",Config->DieValue); - exit(0); - } -} + } -void LoadAllModules() -{ - /* We must load the modules AFTER initializing the socket engine, now */ - MODCOUNT = -1; - char configToken[MAXBUF]; - for (int count = 0; count < Config->ConfValueEnum("module",&Config->config_f); count++) - { - Config->ConfValue("module","name",count,configToken,&Config->config_f); - printf("Loading module... \033[1;32m%s\033[0m\n",configToken); - if (!LoadModule(configToken)) - { - log(DEFAULT,"Exiting due to a module loader error."); - printf("\nThere was an error loading a module: %s\n\nYou might want to do './inspircd start' instead of 'bin/inspircd'\n\n",ModuleError()); - Exit(0); - } - } - log(DEFAULT,"Total loaded modules: %lu",(unsigned long)MODCOUNT+1); -} + /* Remove all the queued users who are due to be quit, free memory used. */ + GlobalGoners.Apply(); +}