X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fhelperfuncs.cpp;h=913bc0861b16a2110c1addfcdef5661d3db85665;hb=dd8679b086c568f37daa1d746a972236938206ca;hp=b5d8f5630260200df40ab5695de3c6a7f010471a;hpb=bab14f0dd2345c9d7dcbc47c918563709e1ac094;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/helperfuncs.cpp b/src/helperfuncs.cpp index b5d8f5630..913bc0861 100644 --- a/src/helperfuncs.cpp +++ b/src/helperfuncs.cpp @@ -1 +1,558 @@ -/* +------------------------------------+ * | Inspire Internet Relay Chat Daemon | * +------------------------------------+ * * InspIRCd: (C) 2002-2007 InspIRCd Development Team * See: http://www.inspircd.org/wiki/index.php/Credits * * This program is free but copyrighted software; see * the file COPYING for details. * * --------------------------------------------------- */ #include "inspircd.h" #include #include "configreader.h" #include "users.h" #include "modules.h" #include "wildcard.h" #include "mode.h" #include "xline.h" #include "exitcodes.h" static char TIMESTR[26]; static time_t LAST = 0; /** 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, ...) { /* Do this check again here so that we save pointless vsnprintf calls */ if ((level < Config->LogLevel) && !Config->forcedebug) return; va_list argsPtr; char textbuffer[65536]; va_start(argsPtr, text); vsnprintf(textbuffer, 65536, text, argsPtr); va_end(argsPtr); this->Log(level, std::string(textbuffer)); } void InspIRCd::Log(int level, const std::string &text) { if (!this->Config) return; /* If we were given -debug we output all messages, regardless of configured loglevel */ if ((level < Config->LogLevel) && !Config->forcedebug) return; 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) { std::string out = std::string(TIMESTR) + " " + text.c_str() + "\n"; this->Logger->WriteLogLine(out); } if (Config->nofork) { printf("%s %s\n", TIMESTR, text.c_str()); } } std::string InspIRCd::GetServerDescription(const char* servername) { std::string description; FOREACH_MOD_I(this,I_OnGetServerDescription,OnGetServerDescription(servername,description)); if (!description.empty()) { return description; } else { // not a remote server that can be found, it must be me. return Config->ServerDesc; } } /* 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, ...) { char textbuffer[MAXBUF]; va_list argsPtr; va_start(argsPtr, text); vsnprintf(textbuffer, MAXBUF, text, argsPtr); va_end(argsPtr); this->WriteOpers(std::string(textbuffer)); } void InspIRCd::WriteOpers(const std::string &text) { for (std::vector::iterator i = this->all_opers.begin(); i != this->all_opers.end(); i++) { userrec* a = *i; if (IS_LOCAL(a) && a->modes[UM_SERVERNOTICE]) { // send server notices to all with +s a->WriteServ("NOTICE %s :%s",a->nick,text.c_str()); } } } void InspIRCd::ServerNoticeAll(char* text, ...) { if (!text) return; char textbuffer[MAXBUF]; char formatbuffer[MAXBUF]; va_list argsPtr; va_start (argsPtr, text); vsnprintf(textbuffer, MAXBUF, text, argsPtr); va_end(argsPtr); snprintf(formatbuffer,MAXBUF,"NOTICE $%s :%s",Config->ServerName,textbuffer); for (std::vector::const_iterator i = local_users.begin(); i != local_users.end(); i++) { userrec* t = *i; t->WriteServ(std::string(formatbuffer)); } } void InspIRCd::ServerPrivmsgAll(char* text, ...) { if (!text) return; 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++) { userrec* t = *i; t->WriteServ(std::string(formatbuffer)); } } void InspIRCd::WriteMode(const char* modes, int flags, const char* text, ...) { char textbuffer[MAXBUF]; int modelen; va_list argsPtr; if (!text || !modes || !flags) { this->Log(DEFAULT,"*** BUG *** WriteMode was given an invalid parameter"); return; } va_start(argsPtr, text); vsnprintf(textbuffer, MAXBUF, text, argsPtr); va_end(argsPtr); modelen = strlen(modes); if (flags == WM_AND) { for (std::vector::const_iterator i = local_users.begin(); i != local_users.end(); i++) { userrec* t = *i; bool send_to_user = true; for (int n = 0; n < modelen; n++) { if (!t->modes[modes[n]-65]) { send_to_user = false; break; } } if (send_to_user) t->WriteServ("NOTICE %s :%s",t->nick,textbuffer); } } else if (flags == WM_OR) { for (std::vector::const_iterator i = local_users.begin(); i != local_users.end(); i++) { userrec* t = *i; bool send_to_user = false; for (int n = 0; n < modelen; n++) { if (t->modes[modes[n]-65]) { send_to_user = true; break; } } 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_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 = 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 = chanlist->find(chan); if (iter == chanlist->end()) /* Couldn't find it */ return NULL; return iter->second; } chanrec* InspIRCd::FindChan(const std::string &chan) { chan_hash::iterator iter = chanlist->find(chan); 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 std::string &s) { for (std::vector::const_iterator i = this->local_users.begin(); i != this->local_users.end(); i++) { if ((*i)->registered == REG_ALL) { (*i)->WriteServ("NOTICE %s :%s",(*i)->nick,s.c_str()); } else { /* 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::UserCount() { 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::RegisteredUserCount() { return clientlist->size() - this->UnregisteredUserCount(); } int InspIRCd::ModeCount(const char mode) { ModeHandler* mh = this->Modes->FindMode(mode, MODETYPE_USER); if (mh) return mh->GetCount(); else return 0; } int InspIRCd::InvisibleUserCount() { return ModeCount('i'); } int InspIRCd::OperCount() { return this->all_opers.size(); } int InspIRCd::UnregisteredUserCount() { return this->unregistered_count; } long InspIRCd::ChannelCount() { return chanlist->size(); } long InspIRCd::LocalUserCount() { /* Doesnt count unregistered clients */ return (local_users.size() - this->UnregisteredUserCount()); } bool InspIRCd::IsChannel(const char *chname) { 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; } c = (char *)chname + 1; while (*c) { switch (*c) { case ' ': case ',': case 7: return false; } c++; } /* too long a name - note funky pointer arithmetic here. */ if ((c - chname) > CHANMAX) { return false; } return true; } bool InspIRCd::IsNick(const char* n) { if (!n || !*n) return false; int p = 0; for (char* i = (char*)n; *i; i++, p++) { if ((*i >= 'A') && (*i <= '}')) { /* "A"-"}" can occur anywhere in a nickname */ continue; } if ((((*i >= '0') && (*i <= '9')) || (*i == '-')) && (i > n)) { /* "0"-"9", "-" can occur anywhere BUT the first char of a nickname */ continue; } /* invalid character! abort */ return false; } /* too long? or not -- pointer arithmetic rocks */ return (p < NICKMAX - 1); } bool InspIRCd::IsIdent(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; } void InspIRCd::OpenLog(char** argv, int argc) { Config->MyDir = Config->GetFullProgDir(); if (!*this->LogFileName) { if (Config->logpath.empty()) { #ifndef DARWIN Config->logpath = Config->MyDir + "/ircd.log"; #else Config->logpath = "/var/log/ircd.log"; #endif } 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: %s\n\n", Config->logpath.c_str(), strerror(errno)); Exit(EXIT_STATUS_LOG); } this->Logger = new FileLogger(this, Config->log_file); } void InspIRCd::CheckRoot() { #ifndef DARWIN if (geteuid() == 0) { printf("WARNING!!! You are running an irc server as ROOT!!! DO NOT DO THIS!!!\n\n"); this->Log(DEFAULT,"Cant start as root"); #else if (geteuid() != 16) { printf("WARNING!!! You are not running inspircd as the ircdaemon user!!! YOU CAN NOT DO THIS!!!\n\n"); this->Log(DEFAULT,"Must start as user ircdaemon"); #endif Exit(EXIT_STATUS_ROOT); } } void InspIRCd::CheckDie() { if (*Config->DieValue) { printf("WARNING: %s\n\n",Config->DieValue); this->Log(DEFAULT,"Died because of tag: %s",Config->DieValue); Exit(EXIT_STATUS_DIETAG); } } /* We must load the modules AFTER initializing the socket engine, now */ void InspIRCd::LoadAllModules() { char configToken[MAXBUF]; Config->module_names.clear(); this->ModCount = -1; for (int count = 0; count < Config->ConfValueEnum(Config->config_data, "module"); count++) { 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 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_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); } void InspIRCd::SendWhoisLine(userrec* user, userrec* dest, int numeric, const std::string &text) { std::string copy_text = text; 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(userrec* user, userrec* 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)); } \ No newline at end of file +/* + * InspIRCd -- Internet Relay Chat Daemon + * + * Copyright (C) 2019 Matt Schatz + * Copyright (C) 2018 linuxdaemon + * Copyright (C) 2013 Daniel Vassdal + * Copyright (C) 2013 Adam + * Copyright (C) 2012-2015 Attila Molnar + * Copyright (C) 2012-2014, 2017-2018 Sadie Powell + * Copyright (C) 2012, 2018 Robby + * Copyright (C) 2012 ChrisTX + * Copyright (C) 2009-2010 Daniel De Graaf + * Copyright (C) 2007, 2010 Craig Edwards + * Copyright (C) 2007 Dennis Friis + * Copyright (C) 2006-2008 Robin Burchell + * + * 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. + * + * 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 . + */ + + +#ifdef _WIN32 +#define _CRT_RAND_S +#include +#endif + +#include "inspircd.h" +#include "xline.h" +#include "exitcodes.h" +#include + +/* Find a user record by nickname and return a pointer to it */ +User* InspIRCd::FindNick(const std::string &nick) +{ + if (!nick.empty() && isdigit(*nick.begin())) + return FindUUID(nick); + return FindNickOnly(nick); +} + +User* InspIRCd::FindNickOnly(const std::string &nick) +{ + user_hash::iterator iter = this->Users->clientlist.find(nick); + + if (iter == this->Users->clientlist.end()) + return NULL; + + return iter->second; +} + +User *InspIRCd::FindUUID(const std::string &uid) +{ + user_hash::iterator finduuid = this->Users->uuidlist.find(uid); + + if (finduuid == this->Users->uuidlist.end()) + return NULL; + + return finduuid->second; +} +/* find a channel record by channel name and return a pointer to it */ + +Channel* InspIRCd::FindChan(const std::string &chan) +{ + chan_hash::iterator iter = chanlist.find(chan); + + if (iter == chanlist.end()) + /* Couldn't find it */ + return NULL; + + return iter->second; +} + +bool InspIRCd::IsValidMask(const std::string &mask) +{ + const char* dest = mask.c_str(); + int exclamation = 0; + int atsign = 0; + + for (const char* i = dest; *i; i++) + { + /* out of range character, bad mask */ + if (*i < 32 || *i > 126) + { + return false; + } + + switch (*i) + { + case '!': + exclamation++; + break; + case '@': + atsign++; + break; + } + } + + /* valid masks only have 1 ! and @ */ + if (exclamation != 1 || atsign != 1) + return false; + + if (mask.length() > ServerInstance->Config->Limits.GetMaxMask()) + return false; + + return true; +} + +void InspIRCd::StripColor(std::string &sentence) +{ + /* refactor this completely due to SQUIT bug since the old code would strip last char and replace with \0 --peavey */ + int seq = 0; + + for (std::string::iterator i = sentence.begin(); i != sentence.end();) + { + 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::ProcessColors(file_cache& input) +{ + /* + * 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 + { + std::string character; + std::string replace; + special_chars(const std::string& c, const std::string& r) + : character(c) + , replace(r) + { + } + } special[] = { + special_chars("\\b", "\x02"), // Bold + special_chars("\\c", "\x03"), // Color + special_chars("\\i", "\x1D"), // Italic + special_chars("\\m", "\x11"), // Monospace + special_chars("\\r", "\x16"), // Reverse + special_chars("\\s", "\x1E"), // Strikethrough + special_chars("\\u", "\x1F"), // Underline + special_chars("\\x", "\x0F"), // Reset + special_chars("", "") + }; + + for(file_cache::iterator it = input.begin(), it_end = input.end(); it != it_end; it++) + { + std::string ret = *it; + for(int i = 0; special[i].character.empty() == false; ++i) + { + std::string::size_type pos = ret.find(special[i].character); + if(pos == std::string::npos) // Couldn't find the character, skip this line + continue; + + if((pos > 0) && (ret[pos-1] == '\\') && (ret[pos] == '\\')) + continue; // Skip double slashes. + + // Replace all our characters in the array + while(pos != std::string::npos) + { + 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()); + } + } + + // Replace double slashes with a single slash before we return + std::string::size_type pos = ret.find("\\\\"); + while(pos != std::string::npos) + { + ret = ret.substr(0, pos) + "\\" + ret.substr(pos + 2); + pos = ret.find("\\\\", pos + 1); + } + *it = ret; + } +} + +/* true for valid channel name, false else */ +bool InspIRCd::DefaultIsChannel(const std::string& chname) +{ + if (chname.empty() || chname.length() > ServerInstance->Config->Limits.ChanMax) + return false; + + if (chname[0] != '#') + return false; + + for (std::string::const_iterator i = chname.begin()+1; i != chname.end(); ++i) + { + switch (*i) + { + case ' ': + case ',': + case 7: + return false; + } + } + + return true; +} + +/* true for valid nickname, false else */ +bool InspIRCd::DefaultIsNick(const std::string& n) +{ + if (n.empty() || n.length() > ServerInstance->Config->Limits.NickMax) + return false; + + for (std::string::const_iterator i = n.begin(); i != n.end(); ++i) + { + if ((*i >= 'A') && (*i <= '}')) + { + /* "A"-"}" can occur anywhere in a nickname */ + continue; + } + + if ((((*i >= '0') && (*i <= '9')) || (*i == '-')) && (i != n.begin())) + { + /* "0"-"9", "-" can occur anywhere BUT the first char of a nickname */ + continue; + } + + /* invalid character! abort */ + return false; + } + + return true; +} + +/* return true for good ident, false else */ +bool InspIRCd::DefaultIsIdent(const std::string& n) +{ + if (n.empty()) + return false; + + for (std::string::const_iterator i = n.begin(); i != n.end(); ++i) + { + if ((*i >= 'A') && (*i <= '}')) + { + continue; + } + + if (((*i >= '0') && (*i <= '9')) || (*i == '-') || (*i == '.')) + { + continue; + } + + return false; + } + + return true; +} + +bool InspIRCd::IsHost(const std::string& host) +{ + // Hostnames must be non-empty and shorter than the maximum hostname length. + if (host.empty() || host.length() > ServerInstance->Config->Limits.MaxHost) + return false; + + unsigned int numdashes = 0; + unsigned int numdots = 0; + bool seendot = false; + const std::string::const_iterator hostend = host.end() - 1; + for (std::string::const_iterator iter = host.begin(); iter != host.end(); ++iter) + { + unsigned char chr = static_cast(*iter); + + // If the current character is a label separator. + if (chr == '.') + { + numdots++; + + // Consecutive separators are not allowed and dashes can not exist at the start or end + // of labels and separators must only exist between labels. + if (seendot || numdashes || iter == host.begin() || iter == hostend) + return false; + + seendot = true; + continue; + } + + // If this point is reached then the character is not a dot. + seendot = false; + + // If the current character is a dash. + if (chr == '-') + { + // Consecutive separators are not allowed and dashes can not exist at the start or end + // of labels and separators must only exist between labels. + if (seendot || numdashes >= 2 || iter == host.begin() || iter == hostend) + return false; + + numdashes += 1; + continue; + } + + // If this point is reached then the character is not a dash. + numdashes = 0; + + // Alphanumeric characters are allowed at any position. + if ((chr >= '0' && chr <= '9') || (chr >= 'A' && chr <= 'Z') || (chr >= 'a' && chr <= 'z')) + continue; + + return false; + } + + // Whilst simple hostnames (e.g. localhost) are valid we do not allow the server to use + // them to prevent issues with clients that differentiate between short client and server + // prefixes by checking whether the nickname contains a dot. + return numdots; +} + +bool InspIRCd::IsSID(const std::string &str) +{ + /* 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]))); +} + +/** A lookup table of values for multiplier characters used by + * InspIRCd::Duration(). In this lookup table, the indexes for + * the ascii values 'm' and 'M' have the value '60', the indexes + * for the ascii values 'D' and 'd' have a value of '86400', etc. + */ +static const unsigned int duration_multi[] = +{ + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 86400, 0, 0, 0, 3600, 0, 0, 0, 0, 60, 0, 0, + 0, 0, 0, 1, 0, 0, 0, 604800, 0, 31557600, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 86400, 0, 0, 0, 3600, 0, 0, 0, 0, 60, 0, 0, + 0, 0, 0, 1, 0, 0, 0, 604800, 0, 31557600, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, +}; + +bool InspIRCd::Duration(const std::string& str, unsigned long& duration) +{ + unsigned long total = 0; + unsigned long subtotal = 0; + + /* Iterate each item in the string, looking for number or multiplier */ + for (std::string::const_iterator i = str.begin(); i != str.end(); ++i) + { + /* Found a number, queue it onto the current number */ + if ((*i >= '0') && (*i <= '9')) + { + subtotal = (subtotal * 10) + (*i - '0'); + } + 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. + */ + unsigned int multiplier = duration_multi[static_cast(*i)]; + if (multiplier == 0) + return false; + + total += subtotal * multiplier; + + /* Next subtotal please */ + subtotal = 0; + } + } + /* Any trailing values built up are treated as raw seconds */ + duration = total + subtotal; + return true; +} + +unsigned long InspIRCd::Duration(const std::string& str) +{ + unsigned long out = 0; + InspIRCd::Duration(str, out); + return out; +} + +bool InspIRCd::IsValidDuration(const std::string& duration) +{ + for (std::string::const_iterator i = duration.begin(); i != duration.end(); ++i) + { + unsigned char c = *i; + if (((c >= '0') && (c <= '9'))) + continue; + + if (!duration_multi[c]) + return false; + } + return true; +} + +std::string InspIRCd::DurationString(time_t duration) +{ + if (duration == 0) + return "0s"; + + time_t years = duration / 31449600; + time_t weeks = (duration / 604800) % 52; + time_t days = (duration / 86400) % 7; + time_t hours = (duration / 3600) % 24; + time_t minutes = (duration / 60) % 60; + time_t seconds = duration % 60; + + std::string ret; + + if (years) + ret = ConvToStr(years) + "y"; + if (weeks) + ret += ConvToStr(weeks) + "w"; + if (days) + ret += ConvToStr(days) + "d"; + if (hours) + ret += ConvToStr(hours) + "h"; + if (minutes) + ret += ConvToStr(minutes) + "m"; + if (seconds) + ret += ConvToStr(seconds) + "s"; + + return ret; +} + +std::string InspIRCd::Format(va_list& vaList, const char* formatString) +{ + static std::vector formatBuffer(1024); + + while (true) + { + va_list dst; + va_copy(dst, vaList); + + int vsnret = vsnprintf(&formatBuffer[0], formatBuffer.size(), formatString, dst); + va_end(dst); + + if (vsnret > 0 && static_cast(vsnret) < formatBuffer.size()) + { + break; + } + + formatBuffer.resize(formatBuffer.size() * 2); + } + + return std::string(&formatBuffer[0]); +} + +std::string InspIRCd::Format(const char* formatString, ...) +{ + std::string ret; + VAFORMAT(ret, formatString, formatString); + return ret; +} + +std::string InspIRCd::TimeString(time_t curtime, const char* format, bool utc) +{ +#ifdef _WIN32 + if (curtime < 0) + curtime = 0; +#endif + + struct tm* timeinfo = utc ? gmtime(&curtime) : localtime(&curtime); + if (!timeinfo) + { + 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 %Y %H:%M:%S"; + + char buffer[512]; + if (!strftime(buffer, sizeof(buffer), format, timeinfo)) + buffer[0] = '\0'; + + return buffer; +} + +std::string InspIRCd::GenRandomStr(unsigned int length, bool printable) +{ + char* buf = new char[length]; + GenRandom(buf, length); + std::string rv; + rv.resize(length); + for(size_t i = 0; i < length; i++) + rv[i] = printable ? 0x3F + (buf[i] & 0x3F) : buf[i]; + delete[] buf; + return rv; +} + +// 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) +{ + unsigned long rv; + GenRandom((char*)&rv, sizeof(rv)); + return rv % max; +} + +// This is overridden by a higher-quality algorithm when SSL support is loaded +void InspIRCd::DefaultGenRandom(char* output, size_t max) +{ +#if defined HAS_ARC4RANDOM_BUF + arc4random_buf(output, max); +#else + for (unsigned int i = 0; i < max; ++i) +# ifdef _WIN32 + { + unsigned int uTemp; + if(rand_s(&uTemp) != 0) + output[i] = rand(); + else + output[i] = uTemp; + } +# else + output[i] = random(); +# endif +#endif +}