2 * InspIRCd -- Internet Relay Chat Daemon
4 * Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
5 * Copyright (C) 2006-2008 Robin Burchell <robin+git@viroteck.net>
6 * Copyright (C) 2005-2008 Craig Edwards <craigedwards@brainbox.cc>
7 * Copyright (C) 2008 Thomas Stagner <aquanight@inspircd.org>
8 * Copyright (C) 2006-2007 Oliver Lupton <oliverlupton@gmail.com>
9 * Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
11 * This file is part of InspIRCd. InspIRCd is free software: you can
12 * redistribute it and/or modify it under the terms of the GNU General Public
13 * License as published by the Free Software Foundation, version 2.
15 * This program is distributed in the hope that it will be useful, but WITHOUT
16 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
17 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
20 * You should have received a copy of the GNU General Public License
21 * along with this program. If not, see <http://www.gnu.org/licenses/>.
32 #include "exitcodes.h"
35 /* Find a user record by nickname and return a pointer to it */
36 User* InspIRCd::FindNick(const std::string &nick)
38 if (!nick.empty() && isdigit(*nick.begin()))
39 return FindUUID(nick);
40 return FindNickOnly(nick);
43 User* InspIRCd::FindNickOnly(const std::string &nick)
45 user_hash::iterator iter = this->Users->clientlist.find(nick);
47 if (iter == this->Users->clientlist.end())
53 User *InspIRCd::FindUUID(const std::string &uid)
55 user_hash::iterator finduuid = this->Users->uuidlist.find(uid);
57 if (finduuid == this->Users->uuidlist.end())
60 return finduuid->second;
62 /* find a channel record by channel name and return a pointer to it */
64 Channel* InspIRCd::FindChan(const std::string &chan)
66 chan_hash::iterator iter = chanlist.find(chan);
68 if (iter == chanlist.end())
69 /* Couldn't find it */
75 bool InspIRCd::IsValidMask(const std::string &mask)
77 const char* dest = mask.c_str();
81 for (const char* i = dest; *i; i++)
83 /* out of range character, bad mask */
84 if (*i < 32 || *i > 126)
100 /* valid masks only have 1 ! and @ */
101 if (exclamation != 1 || atsign != 1)
104 if (mask.length() > 250)
110 void InspIRCd::StripColor(std::string &sentence)
112 /* refactor this completely due to SQUIT bug since the old code would strip last char and replace with \0 --peavey */
115 for (std::string::iterator i = sentence.begin(); i != sentence.end();)
119 else if (seq && (( ((*i >= '0') && (*i <= '9')) || (*i == ',') ) ))
122 if ( (seq <= 4) && (*i == ',') )
130 // Strip all control codes too except \001 for CTCP
131 if (seq || ((*i >= 0) && (*i < 32) && (*i != 1)))
132 i = sentence.erase(i);
138 void InspIRCd::ProcessColors(file_cache& input)
141 * Replace all color codes from the special[] array to actual
142 * color code chars using C++ style escape sequences. You
143 * can append other chars to replace if you like -- Justasic
145 static struct special_chars
147 std::string character;
149 special_chars(const std::string &c, const std::string &r) : character(c), replace(r) { }
153 special_chars("\\002", "\002"), // Bold
154 special_chars("\\037", "\037"), // underline
155 special_chars("\\003", "\003"), // Color
156 special_chars("\\017", "\017"), // Stop colors
157 special_chars("\\u", "\037"), // Alias for underline
158 special_chars("\\b", "\002"), // Alias for Bold
159 special_chars("\\x", "\017"), // Alias for stop
160 special_chars("\\c", "\003"), // Alias for color
161 special_chars("", "")
164 for(file_cache::iterator it = input.begin(), it_end = input.end(); it != it_end; it++)
166 std::string ret = *it;
167 for(int i = 0; special[i].character.empty() == false; ++i)
169 std::string::size_type pos = ret.find(special[i].character);
170 if(pos == std::string::npos) // Couldn't find the character, skip this line
173 if((pos > 0) && (ret[pos-1] == '\\') && (ret[pos] == '\\'))
174 continue; // Skip double slashes.
176 // Replace all our characters in the array
177 while(pos != std::string::npos)
179 ret = ret.substr(0, pos) + special[i].replace + ret.substr(pos + special[i].character.size());
180 pos = ret.find(special[i].character, pos + special[i].replace.size());
184 // Replace double slashes with a single slash before we return
185 std::string::size_type pos = ret.find("\\\\");
186 while(pos != std::string::npos)
188 ret = ret.substr(0, pos) + "\\" + ret.substr(pos + 2);
189 pos = ret.find("\\\\", pos + 1);
195 /* true for valid channel name, false else */
196 bool IsChannelHandler::Call(const std::string& chname)
198 if (chname.empty() || chname.length() > ServerInstance->Config->Limits.ChanMax)
201 if (chname[0] != '#')
204 for (std::string::const_iterator i = chname.begin()+1; i != chname.end(); ++i)
218 /* true for valid nickname, false else */
219 bool IsNickHandler::Call(const std::string& n)
221 if (n.empty() || n.length() > ServerInstance->Config->Limits.NickMax)
224 for (std::string::const_iterator i = n.begin(); i != n.end(); ++i)
226 if ((*i >= 'A') && (*i <= '}'))
228 /* "A"-"}" can occur anywhere in a nickname */
232 if ((((*i >= '0') && (*i <= '9')) || (*i == '-')) && (i != n.begin()))
234 /* "0"-"9", "-" can occur anywhere BUT the first char of a nickname */
238 /* invalid character! abort */
245 /* return true for good ident, false else */
246 bool IsIdentHandler::Call(const std::string& n)
251 for (std::string::const_iterator i = n.begin(); i != n.end(); ++i)
253 if ((*i >= 'A') && (*i <= '}'))
258 if (((*i >= '0') && (*i <= '9')) || (*i == '-') || (*i == '.'))
269 bool InspIRCd::IsSID(const std::string &str)
271 /* Returns true if the string given is exactly 3 characters long,
272 * starts with a digit, and the other two characters are A-Z or digits
274 return ((str.length() == 3) && isdigit(str[0]) &&
275 ((str[1] >= 'A' && str[1] <= 'Z') || isdigit(str[1])) &&
276 ((str[2] >= 'A' && str[2] <= 'Z') || isdigit(str[2])));
279 void InspIRCd::CheckRoot()
284 std::cout << "ERROR: You are running an irc server as root! DO NOT DO THIS!" << std::endl << std::endl;
285 this->Logs->Log("STARTUP", LOG_DEFAULT, "Can't start as root");
286 Exit(EXIT_STATUS_ROOT);
291 /** Refactored by Brain, Jun 2009. Much faster with some clever O(1) array
292 * lookups and pointer maths.
294 unsigned long InspIRCd::Duration(const std::string &str)
296 unsigned char multiplier = 0;
301 /* Iterate each item in the string, looking for number or multiplier */
302 for (std::string::const_reverse_iterator i = str.rbegin(); i != str.rend(); ++i)
304 /* Found a number, queue it onto the current number */
305 if ((*i >= '0') && (*i <= '9'))
307 subtotal = subtotal + ((*i - '0') * times);
312 /* Found something thats not a number, find out how much
313 * it multiplies the built up number by, multiply the total
314 * and reset the built up number.
317 total += subtotal * duration_multi[multiplier];
319 /* Next subtotal please */
327 total += subtotal * duration_multi[multiplier];
330 /* Any trailing values built up are treated as raw seconds */
331 return total + subtotal;
334 const char* InspIRCd::Format(va_list &vaList, const char* formatString)
336 static std::vector<char> formatBuffer(1024);
341 va_copy(dst, vaList);
343 int vsnret = vsnprintf(&formatBuffer[0], formatBuffer.size(), formatString, dst);
346 if (vsnret > 0 && static_cast<unsigned>(vsnret) < formatBuffer.size())
351 formatBuffer.resize(formatBuffer.size() * 2);
354 return &formatBuffer[0];
357 const char* InspIRCd::Format(const char* formatString, ...)
360 VAFORMAT(ret, formatString, formatString);
364 std::string InspIRCd::TimeString(time_t curtime, const char* format, bool utc)
371 struct tm* timeinfo = utc ? gmtime(&curtime) : localtime(&curtime);
375 timeinfo = localtime(&curtime);
378 // If the calculated year exceeds four digits or is less than the year 1000,
379 // the behavior of asctime() is undefined
380 if (timeinfo->tm_year + 1900 > 9999)
381 timeinfo->tm_year = 9999 - 1900;
382 else if (timeinfo->tm_year + 1900 < 1000)
383 timeinfo->tm_year = 0;
385 // This is the default format used by asctime without the terminating new line.
387 format = "%a %b %d %H:%M:%S %Y";
390 if (!strftime(buffer, sizeof(buffer), format, timeinfo))
396 std::string InspIRCd::GenRandomStr(int length, bool printable)
398 char* buf = new char[length];
399 GenRandom(buf, length);
402 for(int i=0; i < length; i++)
403 rv[i] = printable ? 0x3F + (buf[i] & 0x3F) : buf[i];
408 // NOTE: this has a slight bias for lower values if max is not a power of 2.
409 // Don't use it if that matters.
410 unsigned long InspIRCd::GenRandomInt(unsigned long max)
413 GenRandom((char*)&rv, sizeof(rv));
417 // This is overridden by a higher-quality algorithm when SSL support is loaded
418 void GenRandomHandler::Call(char *output, size_t max)
420 for(unsigned int i=0; i < max; i++)
424 if(rand_s(&uTemp) != 0)
430 output[i] = random();
434 ModResult OnCheckExemptionHandler::Call(User* user, Channel* chan, const std::string& restriction)
436 unsigned int mypfx = chan->GetPrefixValue(user);
440 irc::spacesepstream defaultstream(ServerInstance->Config->ConfValue("options")->getString("exemptchanops"));
442 while (defaultstream.GetToken(current))
444 std::string::size_type pos = current.find(':');
445 if (pos == std::string::npos)
447 if (!current.compare(0, pos, restriction))
448 minmode = current[pos+1];
451 PrefixMode* mh = ServerInstance->Modes->FindPrefixMode(minmode);
452 if (mh && mypfx >= mh->GetPrefixRank())
453 return MOD_RES_ALLOW;
454 if (mh || minmode == '*')
456 return MOD_RES_PASSTHRU;