X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fhelperfuncs.cpp;h=8bb81481ef6807d3e22ada43f290c9db434388a4;hb=97b5e92c3d1195c4e3b3dc8dea3183c06399414e;hp=44958d44f471c4b15d04b30bb34f53b3a3105373;hpb=d17465716790010b6e3221f9ce49272110276ccf;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/helperfuncs.cpp b/src/helperfuncs.cpp index 44958d44f..8bb81481e 100644 --- a/src/helperfuncs.cpp +++ b/src/helperfuncs.cpp @@ -1,503 +1,432 @@ -/* +------------------------------------+ - * | Inspire Internet Relay Chat Daemon | - * +------------------------------------+ +/* + * InspIRCd -- Internet Relay Chat Daemon + * + * Copyright (C) 2009-2010 Daniel De Graaf + * Copyright (C) 2006-2008 Robin Burchell + * Copyright (C) 2005-2008 Craig Edwards + * Copyright (C) 2008 Thomas Stagner + * Copyright (C) 2006-2007 Oliver Lupton + * Copyright (C) 2007 Dennis Friis * - * InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev. - * E-mail: - * - * + * This file is part of InspIRCd. InspIRCd is free software: you can + * redistribute it and/or modify it under the terms of the GNU General Public + * License as published by the Free Software Foundation, version 2. * - * Written by Craig Edwards, Craig McLure, and others. - * This program is free but copyrighted software; see - * the file COPYING for details. + * This program is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more + * details. * - * --------------------------------------------------- + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . */ -#include -#include "inspircd_config.h" -#include "configreader.h" -#include -#include -#include -#include -#include -#include -#include -#ifdef HAS_EXECINFO -#include + +#ifdef _WIN32 +#define _CRT_RAND_S +#include #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" -static char TIMESTR[26]; -static time_t LAST = 0; +#include "inspircd.h" +#include "xline.h" +#include "exitcodes.h" +#include -/** Log() - * Write a line of text `text' to the logfile (and stdout, if in nofork) if the level `level' - * is greater than the configured loglevel. - */ -void InspIRCd::Log(int level, const char* text, ...) +/* Find a user record by nickname and return a pointer to it */ +User* InspIRCd::FindNick(const std::string &nick) { - va_list argsPtr; - char textbuffer[MAXBUF]; - - va_start(argsPtr, text); - vsnprintf(textbuffer, MAXBUF, text, argsPtr); - va_end(argsPtr); - - this->Log(level, std::string(textbuffer)); + if (!nick.empty() && isdigit(*nick.begin())) + return FindUUID(nick); + return FindNickOnly(nick); } -void InspIRCd::Log(int level, const std::string &text) +User* InspIRCd::FindNickOnly(const std::string &nick) { - if (!this->Config) - return; - - /* If we were given -debug we output all messages, regardless of configured loglevel */ - if ((level < Config->LogLevel) && !Config->forcedebug) - return; + user_hash::iterator iter = this->Users->clientlist.find(nick); - if (Time() != LAST) - { - time_t local = Time(); - struct tm *timeinfo = localtime(&local); - - strlcpy(TIMESTR,asctime(timeinfo),26); - TIMESTR[24] = ':'; - LAST = Time(); - } - - if (Config->log_file && Config->writelog) - { - fprintf(Config->log_file,"%s %s\n",TIMESTR,text.c_str()); - fflush(Config->log_file); - } + if (iter == this->Users->clientlist.end()) + return NULL; - if (Config->nofork) - { - printf("%s %s\n", TIMESTR, text.c_str()); - } + return iter->second; } -std::string InspIRCd::GetServerDescription(const char* servername) +User *InspIRCd::FindUUID(const std::string &uid) { - std::string description = ""; + user_hash::iterator finduuid = this->Users->uuidlist.find(uid); - FOREACH_MOD_I(this,I_OnGetServerDescription,OnGetServerDescription(servername,description)); + if (finduuid == this->Users->uuidlist.end()) + return NULL; - if (description != "") - { - return description; - } - else - { - // not a remote server that can be found, it must be me. - return Config->ServerDesc; - } + return finduuid->second; } +/* find a channel record by channel name and return a pointer to it */ -/* XXX - We don't use WriteMode for this because WriteMode is very slow and - * this isnt. Basically WriteMode has to iterate ALL the users 'n' times for - * the number of modes provided, e.g. if you send WriteMode 'og' to write to - * opers with globops, and you have 2000 users, thats 4000 iterations. WriteOpers - * uses the oper list, which means if you have 2000 users but only 5 opers, - * it iterates 5 times. - */ -void InspIRCd::WriteOpers(const char* text, ...) +Channel* InspIRCd::FindChan(const std::string &chan) { - char textbuffer[MAXBUF]; - va_list argsPtr; + chan_hash::iterator iter = chanlist.find(chan); - va_start(argsPtr, text); - vsnprintf(textbuffer, MAXBUF, text, argsPtr); - va_end(argsPtr); + if (iter == chanlist.end()) + /* Couldn't find it */ + return NULL; - this->WriteOpers(std::string(textbuffer)); + return iter->second; } -void InspIRCd::WriteOpers(const std::string &text) +bool InspIRCd::IsValidMask(const std::string &mask) { - for (std::vector::iterator i = this->all_opers.begin(); i != this->all_opers.end(); i++) + const char* dest = mask.c_str(); + int exclamation = 0; + int atsign = 0; + + for (const char* i = dest; *i; i++) { - userrec* a = *i; - if (IS_LOCAL(a) && a->modes[UM_SERVERNOTICE]) + /* out of range character, bad mask */ + if (*i < 32 || *i > 126) { - // send server notices to all with +s - a->WriteServ("NOTICE %s :%s",a->nick,text.c_str()); + return false; } - } -} -void InspIRCd::ServerNoticeAll(char* text, ...) -{ - if (!text) - return; + switch (*i) + { + case '!': + exclamation++; + break; + case '@': + atsign++; + break; + } + } - char textbuffer[MAXBUF]; - char formatbuffer[MAXBUF]; - va_list argsPtr; - va_start (argsPtr, text); - vsnprintf(textbuffer, MAXBUF, text, argsPtr); - va_end(argsPtr); + /* valid masks only have 1 ! and @ */ + if (exclamation != 1 || atsign != 1) + return false; - snprintf(formatbuffer,MAXBUF,"NOTICE $%s :%s",Config->ServerName,textbuffer); + if (mask.length() > 250) + return false; - for (std::vector::const_iterator i = local_users.begin(); i != local_users.end(); i++) - { - userrec* t = *i; - t->WriteServ(std::string(formatbuffer)); - } + return true; } -void InspIRCd::ServerPrivmsgAll(char* text, ...) +void InspIRCd::StripColor(std::string &sentence) { - if (!text) - return; + /* refactor this completely due to SQUIT bug since the old code would strip last char and replace with \0 --peavey */ + int seq = 0; - char textbuffer[MAXBUF]; - char formatbuffer[MAXBUF]; - va_list argsPtr; - va_start (argsPtr, text); - vsnprintf(textbuffer, MAXBUF, text, argsPtr); - va_end(argsPtr); - - snprintf(formatbuffer,MAXBUF,"PRIVMSG $%s :%s",Config->ServerName,textbuffer); - - for (std::vector::const_iterator i = local_users.begin(); i != local_users.end(); i++) + for (std::string::iterator i = sentence.begin(); i != sentence.end();) { - userrec* t = *i; - t->WriteServ(std::string(formatbuffer)); + if (*i == 3) + seq = 1; + else if (seq && (( ((*i >= '0') && (*i <= '9')) || (*i == ',') ) )) + { + seq++; + if ( (seq <= 4) && (*i == ',') ) + seq = 1; + else if (seq > 3) + seq = 0; + } + else + seq = 0; + + // Strip all control codes too except \001 for CTCP + if (seq || ((*i >= 0) && (*i < 32) && (*i != 1))) + i = sentence.erase(i); + else + ++i; } } -void InspIRCd::WriteMode(const char* modes, int flags, const char* text, ...) +void InspIRCd::ProcessColors(file_cache& input) { - char textbuffer[MAXBUF]; - int modelen; - va_list argsPtr; - - if ((!text) || (!modes) || (!flags)) + /* + * Replace all color codes from the special[] array to actual + * color code chars using C++ style escape sequences. You + * can append other chars to replace if you like -- Justasic + */ + static struct special_chars { - ilog(this,DEFAULT,"*** BUG *** WriteMode was given an invalid parameter"); - return; + std::string character; + std::string replace; + special_chars(const std::string &c, const std::string &r) : character(c), replace(r) { } } - va_start(argsPtr, text); - vsnprintf(textbuffer, MAXBUF, text, argsPtr); - va_end(argsPtr); - modelen = strlen(modes); - - for (std::vector::const_iterator i = local_users.begin(); i != local_users.end(); i++) + special[] = { + special_chars("\\002", "\002"), // Bold + special_chars("\\037", "\037"), // underline + special_chars("\\003", "\003"), // Color + special_chars("\\017", "\017"), // Stop colors + special_chars("\\u", "\037"), // Alias for underline + special_chars("\\b", "\002"), // Alias for Bold + special_chars("\\x", "\017"), // Alias for stop + special_chars("\\c", "\003"), // Alias for color + special_chars("", "") + }; + + for(file_cache::iterator it = input.begin(), it_end = input.end(); it != it_end; it++) { - userrec* t = (userrec*)(*i); - bool send_to_user = false; - - if (flags == WM_AND) + std::string ret = *it; + for(int i = 0; special[i].character.empty() == false; ++i) { - send_to_user = true; + std::string::size_type pos = ret.find(special[i].character); + if(pos == std::string::npos) // Couldn't find the character, skip this line + continue; - for (int n = 0; n < modelen; n++) - { - if (!t->modes[modes[n]-65]) - { - send_to_user = false; - break; - } - } - } - else if (flags == WM_OR) - { - send_to_user = false; + if((pos > 0) && (ret[pos-1] == '\\') && (ret[pos] == '\\')) + continue; // Skip double slashes. - for (int n = 0; n < modelen; n++) + // Replace all our characters in the array + while(pos != std::string::npos) { - if (t->modes[modes[n]-65]) - { - send_to_user = true; - break; - } + ret = ret.substr(0, pos) + special[i].replace + ret.substr(pos + special[i].character.size()); + pos = ret.find(special[i].character, pos + special[i].replace.size()); } } - if (send_to_user) + // Replace double slashes with a single slash before we return + std::string::size_type pos = ret.find("\\\\"); + while(pos != std::string::npos) { - t->WriteServ("NOTICE %s :%s",t->nick,textbuffer); + ret = ret.substr(0, pos) + "\\" + ret.substr(pos + 2); + pos = ret.find("\\\\", pos + 1); } + *it = ret; } } -/* 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 (iter == clientlist.end()) - /* Couldn't find it */ - return NULL; - - return iter->second; -} - -userrec* InspIRCd::FindNick(const char* nick) -{ - user_hash::iterator iter; - - if (!nick) - return NULL; - - 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 */ - -chanrec* InspIRCd::FindChan(const char* chan) -{ - chan_hash::iterator iter; - - if (!chan) - return NULL; - - iter = chanlist.find(chan); - - if (iter == chanlist.end()) - /* Couldn't find it */ - return NULL; - - return iter->second; -} - -chanrec* InspIRCd::FindChan(const std::string &chan) +/* true for valid channel name, false else */ +bool IsChannelHandler::Call(const std::string& chname) { - chan_hash::iterator iter = chanlist.find(chan); - - if (iter == chanlist.end()) - /* Couldn't find it */ - return NULL; - - return iter->second; -} + if (chname.empty() || chname.length() > ServerInstance->Config->Limits.ChanMax) + return false; + if (chname[0] != '#') + return false; -/* - * sends out an error notice to all connected clients (not to be used - * lightly!) - */ -void InspIRCd::SendError(const char *s) -{ - for (std::vector::const_iterator i = this->local_users.begin(); i != this->local_users.end(); i++) + for (std::string::const_iterator i = chname.begin()+1; i != chname.end(); ++i) { - userrec* t = (userrec*)(*i); - if (t->registered == REG_ALL) - { - t->WriteServ("NOTICE %s :%s",t->nick,s); - } - else + switch (*i) { - // fix - unregistered connections receive ERROR, not NOTICE - t->Write("ERROR :%s",s); + case ' ': + case ',': + case 7: + return false; } } -} -// this function counts all users connected, wether they are registered or NOT. -int InspIRCd::usercnt() -{ - return clientlist.size(); + return true; } -// 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() +/* true for valid nickname, false else */ +bool IsNickHandler::Call(const std::string& n) { - int c = 0; + if (n.empty() || n.length() > ServerInstance->Config->Limits.NickMax) + return false; - for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++) + for (std::string::const_iterator i = n.begin(); i != n.end(); ++i) { - c += (i->second->registered == REG_ALL); - } - - return c; -} + if ((*i >= 'A') && (*i <= '}')) + { + /* "A"-"}" can occur anywhere in a nickname */ + continue; + } -int InspIRCd::usercount_invisible() -{ - int c = 0; + if ((((*i >= '0') && (*i <= '9')) || (*i == '-')) && (i != n.begin())) + { + /* "0"-"9", "-" can occur anywhere BUT the first char of a nickname */ + continue; + } - for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++) - { - c += ((i->second->registered == REG_ALL) && (i->second->modes[UM_INVISIBLE])); + /* invalid character! abort */ + return false; } - return c; + return true; } -int InspIRCd::usercount_opers() +/* return true for good ident, false else */ +bool IsIdentHandler::Call(const std::string& n) { - int c = 0; + if (n.empty()) + return false; - for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++) + for (std::string::const_iterator i = n.begin(); i != n.end(); ++i) { - if (*(i->second->oper)) - c++; - } - return c; -} + if ((*i >= 'A') && (*i <= '}')) + { + continue; + } -int InspIRCd::usercount_unknown() -{ - int c = 0; + if (((*i >= '0') && (*i <= '9')) || (*i == '-') || (*i == '.')) + { + continue; + } - for (std::vector::const_iterator i = local_users.begin(); i != local_users.end(); i++) - { - userrec* t = (userrec*)(*i); - if (t->registered != REG_ALL) - c++; + return false; } - return c; + return true; } -long InspIRCd::chancount() +bool InspIRCd::IsSID(const std::string &str) { - return chanlist.size(); + /* Returns true if the string given is exactly 3 characters long, + * starts with a digit, and the other two characters are A-Z or digits + */ + return ((str.length() == 3) && isdigit(str[0]) && + ((str[1] >= 'A' && str[1] <= 'Z') || isdigit(str[1])) && + ((str[2] >= 'A' && str[2] <= 'Z') || isdigit(str[2]))); } -long InspIRCd::local_count() +void InspIRCd::CheckRoot() { - int c = 0; - - for (std::vector::const_iterator i = local_users.begin(); i != local_users.end(); i++) +#ifndef _WIN32 + if (geteuid() == 0) { - userrec* t = (userrec*)(*i); - if (t->registered == REG_ALL) - c++; + std::cout << "ERROR: You are running an irc server as root! DO NOT DO THIS!" << std::endl << std::endl; + this->Logs->Log("STARTUP", LOG_DEFAULT, "Can't start as root"); + Exit(EXIT_STATUS_ROOT); } - - return c; +#endif } -bool InspIRCd::IsChannel(const char *chname) +/** Refactored by Brain, Jun 2009. Much faster with some clever O(1) array + * lookups and pointer maths. + */ +unsigned long InspIRCd::Duration(const std::string &str) { - char *c; - - /* check for no name - don't check for !*chname, as if it is empty, it won't be '#'! */ - if (!chname || *chname != '#') - { - return false; - } + unsigned char multiplier = 0; + long total = 0; + long times = 1; + long subtotal = 0; - c = (char *)chname + 1; - while (*c) + /* Iterate each item in the string, looking for number or multiplier */ + for (std::string::const_reverse_iterator i = str.rbegin(); i != str.rend(); ++i) { - switch (*c) + /* Found a number, queue it onto the current number */ + if ((*i >= '0') && (*i <= '9')) { - case ' ': - case ',': - case 7: - return false; + subtotal = subtotal + ((*i - '0') * times); + times = times * 10; + } + else + { + /* Found something thats not a number, find out how much + * it multiplies the built up number by, multiply the total + * and reset the built up number. + */ + if (subtotal) + total += subtotal * duration_multi[multiplier]; + + /* Next subtotal please */ + subtotal = 0; + multiplier = *i; + times = 1; } - - c++; } - - /* too long a name - note funky pointer arithmetic here. */ - if ((c - chname) > CHANMAX) + if (multiplier) { - return false; + total += subtotal * duration_multi[multiplier]; + subtotal = 0; } - - return true; + /* Any trailing values built up are treated as raw seconds */ + return total + subtotal; } -void InspIRCd::OpenLog(char** argv, int argc) +const char* InspIRCd::Format(va_list &vaList, const char* formatString) { - if (!*this->LogFileName) - { - if (Config->logpath == "") - { - Config->logpath = ServerConfig::GetFullProgDir(argv,argc) + "/ircd.log"; - } - } - else + static std::vector formatBuffer(1024); + + while (true) { - Config->log_file = fopen(this->LogFileName,"a+"); + va_list dst; + va_copy(dst, vaList); - if (!Config->log_file) + int vsnret = vsnprintf(&formatBuffer[0], formatBuffer.size(), formatString, dst); + va_end(dst); + + if (vsnret > 0 && static_cast(vsnret) < formatBuffer.size()) { - printf("ERROR: Could not write to logfile %s, bailing!\n\n",Config->logpath.c_str()); - Exit(ERROR); + break; } - return; + + formatBuffer.resize(formatBuffer.size() * 2); } - Config->log_file = fopen(Config->logpath.c_str(),"a+"); + return &formatBuffer[0]; +} - if (!Config->log_file) - { - printf("ERROR: Could not write to logfile %s, bailing!\n\n",Config->logpath.c_str()); - Exit(ERROR); - } +const char* InspIRCd::Format(const char* formatString, ...) +{ + const char* ret; + VAFORMAT(ret, formatString, formatString); + return ret; } -void InspIRCd::CheckRoot() +std::string InspIRCd::TimeString(time_t curtime, const char* format, bool utc) { - if (geteuid() == 0) +#ifdef _WIN32 + if (curtime < 0) + curtime = 0; +#endif + + struct tm* timeinfo = utc ? gmtime(&curtime) : localtime(&curtime); + if (!timeinfo) { - 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); + curtime = 0; + timeinfo = localtime(&curtime); } + + // If the calculated year exceeds four digits or is less than the year 1000, + // the behavior of asctime() is undefined + if (timeinfo->tm_year + 1900 > 9999) + timeinfo->tm_year = 9999 - 1900; + else if (timeinfo->tm_year + 1900 < 1000) + timeinfo->tm_year = 0; + + // This is the default format used by asctime without the terminating new line. + if (!format) + format = "%a %b %d %H:%M:%S %Y"; + + char buffer[512]; + if (!strftime(buffer, sizeof(buffer), format, timeinfo)) + buffer[0] = '\0'; + + return buffer; } -void InspIRCd::CheckDie() +std::string InspIRCd::GenRandomStr(int length, bool printable) { - if (*Config->DieValue) - { - printf("WARNING: %s\n\n",Config->DieValue); - ilog(this,DEFAULT,"Died because of tag: %s",Config->DieValue); - Exit(ERROR); - } + char* buf = new char[length]; + GenRandom(buf, length); + std::string rv; + rv.resize(length); + for(int i=0; i < length; i++) + rv[i] = printable ? 0x3F + (buf[i] & 0x3F) : buf[i]; + delete[] buf; + return rv; } -/* We must load the modules AFTER initializing the socket engine, now */ -void InspIRCd::LoadAllModules() +// NOTE: this has a slight bias for lower values if max is not a power of 2. +// Don't use it if that matters. +unsigned long InspIRCd::GenRandomInt(unsigned long max) { - char configToken[MAXBUF]; - Config->module_names.clear(); - this->ModCount = -1; + unsigned long rv; + GenRandom((char*)&rv, sizeof(rv)); + return rv % max; +} - for (int count = 0; count < Config->ConfValueEnum(Config->config_data, "module"); count++) +// This is overridden by a higher-quality algorithm when SSL support is loaded +void GenRandomHandler::Call(char *output, size_t max) +{ + for(unsigned int i=0; i < max; i++) +#ifdef _WIN32 { - 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); - } + unsigned int uTemp; + if(rand_s(&uTemp) != 0) + output[i] = rand(); + else + output[i] = uTemp; } - 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); +#else + output[i] = random(); +#endif } -