]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/helperfuncs.cpp
Remove the last uses of localised fake clients. This removes a lot of allocations...
[user/henk/code/inspircd.git] / src / helperfuncs.cpp
index dab21744aafaf1780f0751cd03b9584903156dd9..608ed2f1bf953186ac20fa0b1ff8a250adeff138 100644 (file)
  * ---------------------------------------------------
  */
 
+#include "inspircd.h"
 #include <stdarg.h>
-#include "configreader.h"
-#include "users.h"
-#include "modules.h"
 #include "wildcard.h"
-#include "mode.h"
 #include "xline.h"
-#include "inspircd.h"
+#include "exitcodes.h"
 
 static char TIMESTR[26];
 static time_t LAST = 0;
@@ -29,6 +26,10 @@ static time_t LAST = 0;
  */
 void InspIRCd::Log(int level, const char* text, ...)
 {
+       /* sanity check, just in case */
+       if (!this->Config || !this->Logger)
+               return;
+
        /* Do this check again here so that we save pointless vsnprintf calls */
        if ((level < Config->LogLevel) && !Config->forcedebug)
                return;
@@ -45,7 +46,8 @@ void InspIRCd::Log(int level, const char* text, ...)
 
 void InspIRCd::Log(int level, const std::string &text)
 {
-       if (!this->Config)
+       /* sanity check, just in case */
+       if (!this->Config || !this->Logger)
                return;
 
        /* If we were given -debug we output all messages, regardless of configured loglevel */
@@ -80,7 +82,7 @@ std::string InspIRCd::GetServerDescription(const char* servername)
 
        FOREACH_MOD_I(this,I_OnGetServerDescription,OnGetServerDescription(servername,description));
 
-       if (description != "")
+       if (!description.empty())
        {
                return description;
        }
@@ -115,7 +117,7 @@ void InspIRCd::WriteOpers(const std::string &text)
        for (std::vector<userrec*>::iterator i = this->all_opers.begin(); i != this->all_opers.end(); i++)
        {
                userrec* a = *i;
-               if (IS_LOCAL(a) && a->modes[UM_SERVERNOTICE])
+               if (IS_LOCAL(a) && a->IsModeSet('s'))
                {
                        // send server notices to all with +s
                        a->WriteServ("NOTICE %s :%s",a->nick,text.c_str());
@@ -191,18 +193,19 @@ void InspIRCd::WriteMode(const char* modes, int flags, const char* text, ...)
 
                        for (int n = 0; n < modelen; n++)
                        {
-                               if (!t->modes[modes[n]-65])
+                               if (!t->IsModeSet(modes[n]))
                                {
                                        send_to_user = false;
                                        break;
                                }
                        }
                        if (send_to_user)
-                               t->WriteServ("NOTICE %s :%s",t->nick,textbuffer);
+                       {
+                               t->WriteServ("NOTICE %s :%s", t->nick, textbuffer);
+                       }
                }
        }
-       else
-       if (flags == WM_OR)
+       else if (flags == WM_OR)
        {
                for (std::vector<userrec*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
                {
@@ -211,25 +214,30 @@ void InspIRCd::WriteMode(const char* modes, int flags, const char* text, ...)
 
                        for (int n = 0; n < modelen; n++)
                        {
-                               if (t->modes[modes[n]-65])
+                               if (t->IsModeSet(modes[n]))
                                {
                                        send_to_user = true;
                                        break;
                                }
                        }
+
                        if (send_to_user)
-                               t->WriteServ("NOTICE %s :%s",t->nick,textbuffer);
+                       {
+                               t->WriteServ("NOTICE %s :%s", t->nick, textbuffer);
+                       }
                }
        }
 }
 
 /* Find a user record by nickname and return a pointer to it */
-
 userrec* InspIRCd::FindNick(const std::string &nick)
 {
-       user_hash::iterator iter = clientlist.find(nick);
+       if (!nick.empty() && isdigit(*nick.begin()))
+               return FindUUID(nick);
 
-       if (iter == clientlist.end())
+       user_hash::iterator iter = clientlist->find(nick);
+
+       if (iter == clientlist->end())
                /* Couldn't find it */
                return NULL;
 
@@ -238,21 +246,58 @@ userrec* InspIRCd::FindNick(const std::string &nick)
 
 userrec* InspIRCd::FindNick(const char* nick)
 {
-       user_hash::iterator iter = clientlist.find(nick);
+       if (isdigit(*nick))
+               return FindUUID(nick);
+
+       user_hash::iterator iter = clientlist->find(nick);
        
-       if (iter == clientlist.end())
+       if (iter == clientlist->end())
                return NULL;
 
        return iter->second;
 }
 
-/* find a channel record by channel name and return a pointer to it */
+userrec* InspIRCd::FindNickOnly(const std::string &nick)
+{
+       user_hash::iterator iter = clientlist->find(nick);
+
+       if (iter == clientlist->end())
+               return NULL;
 
+       return iter->second;
+}
+
+userrec* InspIRCd::FindNickOnly(const char* nick)
+{
+       user_hash::iterator iter = clientlist->find(nick);
+
+       if (iter == clientlist->end())
+               return NULL;
+
+       return iter->second;
+}
+
+userrec *InspIRCd::FindUUID(const std::string &uid)
+{
+       return FindUUID(uid.c_str());
+}
+
+userrec *InspIRCd::FindUUID(const char *uid)
+{
+       user_hash::iterator finduuid = uuidlist->find(uid);
+
+       if (finduuid == uuidlist->end())
+               return NULL;
+
+       return finduuid->second;
+}
+
+/* find a channel record by channel name and return a pointer to it */
 chanrec* InspIRCd::FindChan(const char* chan)
 {
-       chan_hash::iterator iter = chanlist.find(chan);
+       chan_hash::iterator iter = chanlist->find(chan);
 
-       if (iter == chanlist.end())
+       if (iter == chanlist->end())
                /* Couldn't find it */
                return NULL;
 
@@ -261,105 +306,92 @@ chanrec* InspIRCd::FindChan(const char* chan)
 
 chanrec* InspIRCd::FindChan(const std::string &chan)
 {
-       chan_hash::iterator iter = chanlist.find(chan);
+       chan_hash::iterator iter = chanlist->find(chan);
 
-       if (iter == chanlist.end())
+       if (iter == chanlist->end())
                /* Couldn't find it */
                return NULL;
 
        return iter->second;
 }
 
-/*
- * sends out an error notice to all connected clients (not to be used
- * lightly!)
- */
-void InspIRCd::SendError(const char *s)
+/* Send an error notice to all users, registered or not */
+void InspIRCd::SendError(const std::string &s)
 {
        for (std::vector<userrec*>::const_iterator i = this->local_users.begin(); i != this->local_users.end(); i++)
        {
-               userrec* t = (userrec*)(*i);
-               if (t->registered == REG_ALL)
+               if ((*i)->registered == REG_ALL)
                {
-                       t->WriteServ("NOTICE %s :%s",t->nick,s);
+                       (*i)->WriteServ("NOTICE %s :%s",(*i)->nick,s.c_str());
                }
                else
                {
-                       // fix - unregistered connections receive ERROR, not NOTICE
-                       t->Write("ERROR :%s",s);
+                       /* Unregistered connections receive ERROR, not a NOTICE */
+                       (*i)->Write("ERROR :" + s);
                }
+               /* This might generate a whole load of EAGAIN, but we dont really
+                * care about this, as if we call SendError something catastrophic
+                * has occured anyway, and we wont receive the events for these.
+                */
+               (*i)->FlushWriteBuf();
        }
 }
 
-// this function counts all users connected, wether they are registered or NOT.
+/* this function counts all users connected, wether they are registered or NOT. */
 int InspIRCd::UserCount()
 {
-       return clientlist.size();
+       return clientlist->size();
 }
 
-// this counts only registered users, so that the percentages in /MAP don't mess up when users are sitting in an unregistered state
+/* this counts only registered users, so that the percentages in /MAP don't mess up when users are sitting in an unregistered state */
 int InspIRCd::RegisteredUserCount()
 {
-       return clientlist.size() - this->UnregisteredUserCount();
+       return clientlist->size() - this->UnregisteredUserCount();
 }
 
-int InspIRCd::InvisibleUserCount()
+/* return how many users have a given mode e.g. 'a' */
+int InspIRCd::ModeCount(const char mode)
 {
-       int c = 0;
+       ModeHandler* mh = this->Modes->FindMode(mode, MODETYPE_USER);
 
-       for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++)
-       {
-               c += ((i->second->registered == REG_ALL) && (i->second->modes[UM_INVISIBLE]));
-       }
+       if (mh)
+               return mh->GetCount();
+       else
+               return 0;
+}
 
-       return c;
+/* wrapper for readability */
+int InspIRCd::InvisibleUserCount()
+{
+       return ModeCount('i');
 }
 
+/* return how many users are opered */
 int InspIRCd::OperCount()
 {
-       int c = 0;
-
-       for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++)
-       {
-               if (*(i->second->oper))
-                       c++;
-       }
-       return c;
+       return this->all_opers.size();
 }
 
+/* return how many users are unregistered */
 int InspIRCd::UnregisteredUserCount()
 {
-       int c = 0;
-
-       for (std::vector<userrec*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
-       {
-               userrec* t = (userrec*)(*i);
-               if (t->registered != REG_ALL)
-                       c++;
-       }
-
-       return c;
+       return this->unregistered_count;
 }
 
+/* return channel count */
 long InspIRCd::ChannelCount()
 {
-       return chanlist.size();
+       return chanlist->size();
 }
 
+/* return how many local registered users there are */
 long InspIRCd::LocalUserCount()
 {
-       int c = 0;
-
-       for (std::vector<userrec*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
-       {
-               userrec* t = (userrec*)(*i);
-               if (t->registered == REG_ALL)
-                       c++;
-       }
-
-       return c;
+       /* Doesnt count unregistered clients */
+       return (local_users.size() - this->UnregisteredUserCount());
 }
 
+/* true for valid channel name, false else */
 bool InspIRCd::IsChannel(const char *chname)
 {
        char *c;
@@ -393,7 +425,8 @@ bool InspIRCd::IsChannel(const char *chname)
        return true;
 }
 
-bool InspIRCd::IsNick(const char* n)
+/* true for valid nickname, false else */
+bool IsNickHandler::Call(const char* n)
 {
        if (!n || !*n)
                return false;
@@ -421,13 +454,40 @@ bool InspIRCd::IsNick(const char* n)
        return (p < NICKMAX - 1);
 }
 
-void InspIRCd::OpenLog(char** argv, int argc)
+/* return true for good ident, false else */
+bool IsIdentHandler::Call(const char* n)
+{
+       if (!n || !*n)
+               return false;
+
+       for (char* i = (char*)n; *i; i++)
+       {
+               if ((*i >= 'A') && (*i <= '}'))
+               {
+                       continue;
+               }
+
+               if (((*i >= '0') && (*i <= '9')) || (*i == '-') || (*i == '.'))
+               {
+                       continue;
+               }
+
+               return false;
+       }
+
+       return true;
+}
+
+/* open the proper logfile */
+bool InspIRCd::OpenLog(char** argv, int argc)
 {
+       Config->MyDir = Config->GetFullProgDir();
+
        if (!*this->LogFileName)
        {
-               if (Config->logpath == "")
+               if (Config->logpath.empty())
                {
-                       Config->logpath = ServerConfig::GetFullProgDir(argv,argc) + "/ircd.log";
+                       Config->logpath = Config->MyDir + "/ircd.log";
                }
 
                Config->log_file = fopen(Config->logpath.c_str(),"a+");
@@ -439,11 +499,12 @@ void InspIRCd::OpenLog(char** argv, int argc)
 
        if (!Config->log_file)
        {
-               printf("ERROR: Could not write to logfile %s: %s\n\n", Config->logpath.c_str(), strerror(errno));
-               Exit(ERROR);
+               this->Logger = NULL;
+               return false;
        }
 
        this->Logger = new FileLogger(this, Config->log_file);
+       return true;
 }
 
 void InspIRCd::CheckRoot()
@@ -452,7 +513,7 @@ void InspIRCd::CheckRoot()
        {
                printf("WARNING!!! You are running an irc server as ROOT!!! DO NOT DO THIS!!!\n\n");
                this->Log(DEFAULT,"Cant start as root");
-               Exit(ERROR);
+               Exit(EXIT_STATUS_ROOT);
        }
 }
 
@@ -462,7 +523,7 @@ void InspIRCd::CheckDie()
        {
                printf("WARNING: %s\n\n",Config->DieValue);
                this->Log(DEFAULT,"Died because of <die> tag: %s",Config->DieValue);
-               Exit(ERROR);
+               Exit(EXIT_STATUS_DIETAG);
        }
 }
 
@@ -475,17 +536,17 @@ void InspIRCd::LoadAllModules()
 
        for (int count = 0; count < Config->ConfValueEnum(Config->config_data, "module"); count++)
        {
-               Config->ConfValue(Config->config_data, "module","name",count,configToken,MAXBUF);
-               printf("[\033[1;32m*\033[0m] Loading module:\t\033[1;32m%s\033[0m\n",configToken);
+               Config->ConfValue(Config->config_data, "module", "name", count, configToken, MAXBUF);
+               printf_c("[\033[1;32m*\033[0m] Loading module:\t\033[1;32m%s\033[0m\n",configToken);
                
                if (!this->LoadModule(configToken))             
                {
-                       this->Log(DEFAULT,"There was an error loading a module: %s", this->ModuleError());
-                       printf("\nThere was an error loading a module: %s\n\n",this->ModuleError());
-                       Exit(ERROR);
+                       this->Log(DEFAULT,"There was an error loading the module '%s': %s", configToken, this->ModuleError());
+                       printf_c("\n[\033[1;31m*\033[0m] There was an error loading the module '%s': %s\n\n", configToken, this->ModuleError());
+                       Exit(EXIT_STATUS_MODULE);
                }
        }
-       printf("\nA total of \033[1;32m%d\033[0m module%s been loaded.\n", this->ModCount+1, this->ModCount+1 == 1 ? " has" : "s have");
+       printf_c("\nA total of \033[1;32m%d\033[0m module%s been loaded.\n", this->ModCount+1, this->ModCount+1 == 1 ? " has" : "s have");
        this->Log(DEFAULT,"Total loaded modules: %d", this->ModCount+1);
 }