]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/helperfuncs.cpp
In the grand tradition of huge fucking commits:
[user/henk/code/inspircd.git] / src / helperfuncs.cpp
index 44958d44f471c4b15d04b30bb34f53b3a3105373..a90dd01050063f5d83605c5cf326f65bfd29e968 100644 (file)
@@ -2,46 +2,20 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                    E-mail:
- *             <brain@chatspike.net>
- *             <Craig@chatspike.net>
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
  *
- * Written by Craig Edwards, Craig McLure, and others.
  * This program is free but copyrighted software; see
- *         the file COPYING for details.
+ *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
+#include "inspircd.h"
 #include <stdarg.h>
-#include "inspircd_config.h"
-#include "configreader.h"
-#include <unistd.h>
-#include <fcntl.h>
-#include <sys/errno.h>
-#include <signal.h>
-#include <time.h>
-#include <string>
-#include <sstream>
-#ifdef HAS_EXECINFO
-#include <execinfo.h>
-#endif
-#include "connection.h"
-#include "users.h"
-#include "ctables.h"
-#include "globals.h"
-#include "modules.h"
-#include "dynamic.h"
 #include "wildcard.h"
-#include "mode.h"
 #include "xline.h"
-#include "commands.h"
-#include "inspstring.h"
-#include "helperfuncs.h"
-#include "hashcomp.h"
-#include "typedefs.h"
-#include "inspircd.h"
+#include "exitcodes.h"
 
 static char TIMESTR[26];
 static time_t LAST = 0;
@@ -52,11 +26,19 @@ 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;
+
        va_list argsPtr;
-       char textbuffer[MAXBUF];
+       char textbuffer[65536];
 
        va_start(argsPtr, text);
-       vsnprintf(textbuffer, MAXBUF, text, argsPtr);
+       vsnprintf(textbuffer, 65536, text, argsPtr);
        va_end(argsPtr);
 
        this->Log(level, std::string(textbuffer));
@@ -64,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 */
@@ -83,8 +66,8 @@ void InspIRCd::Log(int level, const std::string &text)
 
        if (Config->log_file && Config->writelog)
        {
-               fprintf(Config->log_file,"%s %s\n",TIMESTR,text.c_str());
-               fflush(Config->log_file);
+               std::string out = std::string(TIMESTR) + " " + text.c_str() + "\n";
+               this->Logger->WriteLogLine(out);
        }
 
        if (Config->nofork)
@@ -95,11 +78,11 @@ void InspIRCd::Log(int level, const std::string &text)
 
 std::string InspIRCd::GetServerDescription(const char* servername)
 {
-       std::string description = "";
+       std::string description;
 
        FOREACH_MOD_I(this,I_OnGetServerDescription,OnGetServerDescription(servername,description));
 
-       if (description != "")
+       if (!description.empty())
        {
                return description;
        }
@@ -131,10 +114,10 @@ void InspIRCd::WriteOpers(const char* text, ...)
 
 void InspIRCd::WriteOpers(const std::string &text)
 {
-       for (std::vector<userrec*>::iterator i = this->all_opers.begin(); i != this->all_opers.end(); i++)
+       for (std::list<User*>::iterator i = this->all_opers.begin(); i != this->all_opers.end(); i++)
        {
-               userrec* a = *i;
-               if (IS_LOCAL(a) && a->modes[UM_SERVERNOTICE])
+               User* a = *i;
+               if (IS_LOCAL(a) && a->IsModeSet('s'))
                {
                        // send server notices to all with +s
                        a->WriteServ("NOTICE %s :%s",a->nick,text.c_str());
@@ -156,9 +139,9 @@ void InspIRCd::ServerNoticeAll(char* text, ...)
 
        snprintf(formatbuffer,MAXBUF,"NOTICE $%s :%s",Config->ServerName,textbuffer);
 
-       for (std::vector<userrec*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
+       for (std::vector<User*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
        {
-               userrec* t = *i;
+               User* t = *i;
                t->WriteServ(std::string(formatbuffer));
        }
 }
@@ -177,9 +160,9 @@ void InspIRCd::ServerPrivmsgAll(char* text, ...)
 
        snprintf(formatbuffer,MAXBUF,"PRIVMSG $%s :%s",Config->ServerName,textbuffer);
 
-       for (std::vector<userrec*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
+       for (std::vector<User*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
        {
-               userrec* t = *i;
+               User* t = *i;
                t->WriteServ(std::string(formatbuffer));
        }
 }
@@ -190,9 +173,9 @@ void InspIRCd::WriteMode(const char* modes, int flags, const char* text, ...)
        int modelen;
        va_list argsPtr;
 
-       if ((!text) || (!modes) || (!flags))
+       if (!text || !modes || !flags)
        {
-               ilog(this,DEFAULT,"*** BUG *** WriteMode was given an invalid parameter");
+               this->Log(DEFAULT,"*** BUG *** WriteMode was given an invalid parameter");
                return;
        }
 
@@ -201,200 +184,214 @@ void InspIRCd::WriteMode(const char* modes, int flags, const char* text, ...)
        va_end(argsPtr);
        modelen = strlen(modes);
 
-       for (std::vector<userrec*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
+       if (flags == WM_AND)
        {
-               userrec* t = (userrec*)(*i);
-               bool send_to_user = false;
-
-               if (flags == WM_AND)
+               for (std::vector<User*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
                {
-                       send_to_user = true;
+                       User* t = *i;
+                       bool send_to_user = true;
 
                        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);
+                       }
                }
-               else if (flags == WM_OR)
+       }
+       else if (flags == WM_OR)
+       {
+               for (std::vector<User*>::const_iterator i = local_users.begin(); i != local_users.end(); i++)
                {
-                       send_to_user = false;
+                       User* t = *i;
+                       bool send_to_user = false;
 
                        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);
+                       if (send_to_user)
+                       {
+                               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* 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;
 
        return iter->second;
 }
 
-userrec* InspIRCd::FindNick(const char* nick)
+User* InspIRCd::FindNick(const char* nick)
 {
-       user_hash::iterator iter;
+       if (isdigit(*nick))
+               return FindUUID(nick);
 
-       if (!nick)
+       user_hash::iterator iter = clientlist->find(nick);
+       
+       if (iter == clientlist->end())
                return NULL;
 
-       iter = clientlist.find(nick);
-       
-       if (iter == clientlist.end())
+       return iter->second;
+}
+
+User* InspIRCd::FindNickOnly(const std::string &nick)
+{
+       user_hash::iterator iter = clientlist->find(nick);
+
+       if (iter == clientlist->end())
                return NULL;
 
        return iter->second;
 }
 
-/* find a channel record by channel name and return a pointer to it */
+User* InspIRCd::FindNickOnly(const char* nick)
+{
+       user_hash::iterator iter = clientlist->find(nick);
 
-chanrec* InspIRCd::FindChan(const char* chan)
+       if (iter == clientlist->end())
+               return NULL;
+
+       return iter->second;
+}
+
+User *InspIRCd::FindUUID(const std::string &uid)
+{
+       return FindUUID(uid.c_str());
+}
+
+User *InspIRCd::FindUUID(const char *uid)
 {
-       chan_hash::iterator iter;
+       user_hash::iterator finduuid = uuidlist->find(uid);
 
-       if (!chan)
+       if (finduuid == uuidlist->end())
                return NULL;
 
-       iter = chanlist.find(chan);
+       return finduuid->second;
+}
 
-       if (iter == chanlist.end())
+/* find a channel record by channel name and return a pointer to it */
+Channel* InspIRCd::FindChan(const char* chan)
+{
+       chan_hash::iterator iter = chanlist->find(chan);
+
+       if (iter == chanlist->end())
                /* Couldn't find it */
                return NULL;
 
        return iter->second;
 }
 
-chanrec* InspIRCd::FindChan(const std::string &chan)
+Channel* 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++)
+       for (std::vector<User*>::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.
-int InspIRCd::usercnt()
+/* 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
-int InspIRCd::registered_usercount()
+/* 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()
 {
-       int c = 0;
-
-       for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++)
-       {
-               c += (i->second->registered == REG_ALL);
-       }
-
-       return c;
+       return clientlist->size() - this->UnregisteredUserCount();
 }
 
-int InspIRCd::usercount_invisible()
+/* return how many users have a given mode e.g. 'a' */
+int InspIRCd::ModeCount(const char mode)
 {
-       int c = 0;
-
-       for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++)
-       {
-               c += ((i->second->registered == REG_ALL) && (i->second->modes[UM_INVISIBLE]));
-       }
+       ModeHandler* mh = this->Modes->FindMode(mode, MODETYPE_USER);
 
-       return c;
+       if (mh)
+               return mh->GetCount();
+       else
+               return 0;
 }
 
-int InspIRCd::usercount_opers()
+/* wrapper for readability */
+int InspIRCd::InvisibleUserCount()
 {
-       int c = 0;
-
-       for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++)
-       {
-               if (*(i->second->oper))
-                       c++;
-       }
-       return c;
+       return ModeCount('i');
 }
 
-int InspIRCd::usercount_unknown()
+/* return how many users are opered */
+int InspIRCd::OperCount()
 {
-       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->all_opers.size();
 }
 
-long InspIRCd::chancount()
+/* return how many users are unregistered */
+int InspIRCd::UnregisteredUserCount()
 {
-       return chanlist.size();
+       return this->unregistered_count;
 }
 
-long InspIRCd::local_count()
+/* return channel count */
+long InspIRCd::ChannelCount()
 {
-       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 chanlist->size();
+}
 
-       return c;
+/* return how many local registered users there are */
+long InspIRCd::LocalUserCount()
+{
+       /* 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;
@@ -428,34 +425,86 @@ bool InspIRCd::IsChannel(const char *chname)
        return true;
 }
 
-void InspIRCd::OpenLog(char** argv, int argc)
+/* true for valid nickname, false else */
+bool IsNickHandler::Call(const char* n)
 {
-       if (!*this->LogFileName)
+       if (!n || !*n)
+               return false;
+       int p = 0;
+       for (char* i = (char*)n; *i; i++, p++)
        {
-               if (Config->logpath == "")
+               if ((*i >= 'A') && (*i <= '}'))
+               {
+                       /* "A"-"}" can occur anywhere in a nickname */
+                       continue;
+               }
+
+               if ((((*i >= '0') && (*i <= '9')) || (*i == '-')) && (i > n))
                {
-                       Config->logpath = ServerConfig::GetFullProgDir(argv,argc) + "/ircd.log";
+                       /* "0"-"9", "-" can occur anywhere BUT the first char of a nickname */
+                       continue;
                }
+
+               /* invalid character! abort */
+               return false;
        }
-       else
+
+       /* too long? or not -- pointer arithmetic rocks */
+       return (p < NICKMAX - 1);
+}
+
+/* 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++)
        {
-               Config->log_file = fopen(this->LogFileName,"a+");
+               if ((*i >= 'A') && (*i <= '}'))
+               {
+                       continue;
+               }
 
-               if (!Config->log_file)
+               if (((*i >= '0') && (*i <= '9')) || (*i == '-') || (*i == '.'))
                {
-                       printf("ERROR: Could not write to logfile %s, bailing!\n\n",Config->logpath.c_str());
-                       Exit(ERROR);
+                       continue;
                }
-               return;
+
+               return false;
        }
 
-       Config->log_file = fopen(Config->logpath.c_str(),"a+");
+       return true;
+}
+
+/* open the proper logfile */
+bool InspIRCd::OpenLog(char** argv, int argc)
+{
+       Config->MyDir = Config->GetFullProgDir();
+
+       if (!*this->LogFileName)
+       {
+               if (Config->logpath.empty())
+               {
+                       Config->logpath = Config->MyDir + "/ircd.log";
+               }
+
+               Config->log_file = fopen(Config->logpath.c_str(),"a+");
+       }
+       else
+       {
+               Config->log_file = fopen(this->LogFileName,"a+");
+       }
 
        if (!Config->log_file)
        {
-               printf("ERROR: Could not write to logfile %s, bailing!\n\n",Config->logpath.c_str());
-               Exit(ERROR);
+               this->Logger = NULL;
+               return false;
        }
+
+       this->Logger = new FileLogger(this, Config->log_file);
+       return true;
 }
 
 void InspIRCd::CheckRoot()
@@ -463,8 +512,8 @@ void InspIRCd::CheckRoot()
        if (geteuid() == 0)
        {
                printf("WARNING!!! You are running an irc server as ROOT!!! DO NOT DO THIS!!!\n\n");
-               ilog(this,DEFAULT,"Cant start as root");
-               Exit(ERROR);
+               this->Log(DEFAULT,"Cant start as root");
+               Exit(EXIT_STATUS_ROOT);
        }
 }
 
@@ -473,31 +522,29 @@ void InspIRCd::CheckDie()
        if (*Config->DieValue)
        {
                printf("WARNING: %s\n\n",Config->DieValue);
-               ilog(this,DEFAULT,"Died because of <die> tag: %s",Config->DieValue);
-               Exit(ERROR);
+               this->Log(DEFAULT,"Died because of <die> tag: %s",Config->DieValue);
+               Exit(EXIT_STATUS_DIETAG);
        }
 }
 
-/* We must load the modules AFTER initializing the socket engine, now */
-void InspIRCd::LoadAllModules()
+void InspIRCd::SendWhoisLine(User* user, User* dest, int numeric, const std::string &text)
 {
-       char configToken[MAXBUF];
-       Config->module_names.clear();
-       this->ModCount = -1;
+       std::string copy_text = text;
 
-       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);
-               
-               if (!this->LoadModule(configToken))             
-               {
-                       ilog(this,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);
-               }
-       }
-       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");
-       ilog(this,DEFAULT,"Total loaded modules: %d", this->ModCount+1);
+       int MOD_RESULT = 0;
+       FOREACH_RESULT_I(this, I_OnWhoisLine, OnWhoisLine(user, dest, numeric, copy_text));
+
+       if (!MOD_RESULT)
+               user->WriteServ("%d %s", numeric, copy_text.c_str());
 }
 
+void InspIRCd::SendWhoisLine(User* user, User* dest, int numeric, const char* format, ...)
+{
+       char textbuffer[MAXBUF];
+       va_list argsPtr;
+       va_start (argsPtr, format);
+       vsnprintf(textbuffer, MAXBUF, format, argsPtr);
+       va_end(argsPtr);
+
+       this->SendWhoisLine(user, dest, numeric, std::string(textbuffer));
+}