src/hashcomp.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
   6  *                     E-mail:
   7  *              <brain@chatspike.net>
   8  *              <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *          the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 using namespace std;
  18
  19 #include "inspircd.h"
  20 #include "hashcomp.h"
  21 #include <ext/hash_map>
  22 #define nspace __gnu_cxx
  23
  24 /******************************************************
  25  *
  26  * The hash functions of InspIRCd are the centrepoint
  27  * of the entire system. If these functions are
  28  * inefficient or wasteful, the whole program suffers
  29  * as a result. A lot of C programmers in the ircd
  30  * scene spend a lot of time debating (arguing) about
  31  * the best way to write hash functions to hash irc
  32  * nicknames, channels etc.
  33  * We are lucky as C++ developers as hash_map does
  34  * a lot of this for us. It does intellegent memory
  35  * requests, bucketing, search functions, insertion
  36  * and deletion etc. All we have to do is write some
  37  * overloaded comparison and hash value operators which
  38  * cause it to act in an irc-like way. The features we
  39  * add to the standard hash_map are:
  40  *
  41  * Case insensitivity: The hash_map will be case
  42  * insensitive.
  43  *
  44  * Scandanavian Comparisons: The characters [, ], \ will
  45  * be considered the lowercase of {, } and |.
  46  *
  47  * This file also contains hashing methods for hashing
  48  * insp_inaddr structs, we use this if we want to cache IP
  49  * addresses.
  50  *
  51  ******************************************************/
  52
  53 using namespace std;
  54 using namespace irc::sockets;
  55
  56 /* convert a string to lowercase. Note following special circumstances
  57  * taken from RFC 1459. Many "official" server branches still hold to this
  58  * rule so i will too;
  59  *
  60  *  Because of IRC's scandanavian origin, the characters {}| are
  61  *  considered to be the lower case equivalents of the characters []\,
  62  *  respectively. This is a critical issue when determining the
  63  *  equivalence of two nicknames.
  64  */
  65 void nspace::strlower(char *n)
  66 {
  67         if (n)
  68         {
  69                 for (char* t = n; *t; t++)
  70                         *t = lowermap[(unsigned char)*t];
  71         }
  72 }
  73
  74 size_t nspace::hash<insp_inaddr>::operator()(const insp_inaddr &a) const
  75 {
  76         size_t q;
  77         memcpy(&q,&a,sizeof(size_t));
  78         return q;
  79 }
  80
  81 size_t nspace::hash<string>::operator()(const string &s) const
  82 {
  83         /* XXX: NO DATA COPIES! :)
  84          * The hash function here is practically
  85          * a copy of the one in STL's hash_fun.h,
  86          * only with *x replaced with lowermap[*x].
  87          * This avoids a copy to use hash<const char*>
  88          */
  89         register size_t t = 0;
  90         for (std::string::const_iterator x = s.begin(); x != s.end(); ++x) /* ++x not x++, as its faster */
  91                 t = 5 * t + lowermap[(unsigned char)*x];
  92         return t;
  93 }
  94
  95 bool irc::StrHashComp::operator()(const std::string& s1, const std::string& s2) const
  96 {
  97         unsigned char* n1 = (unsigned char*)s1.c_str();
  98         unsigned char* n2 = (unsigned char*)s2.c_str();
  99         for (; *n1 && *n2; n1++, n2++)
 100                 if (lowermap[*n1] != lowermap[*n2])
 101                         return false;
 102         return (lowermap[*n1] == lowermap[*n2]);
 103 }
 104
 105 bool irc::InAddr_HashComp::operator()(const insp_inaddr &s1, const insp_inaddr &s2) const
 106 {
 107 #ifdef IPV6
 108         for (int n = 0; n < 16; n++)
 109                 if (s2.s6_addr[n] != s1.s6_addr[n])
 110                         return false;
 111         return true;
 112 #else
 113         return (s1.s_addr == s1.s_addr);
 114 #endif
 115 }
 116
 117 /******************************************************
 118  *
 119  * This is the implementation of our special irc::string
 120  * class which is a case-insensitive equivalent to
 121  * std::string which is not only case-insensitive but
 122  * can also do scandanavian comparisons, e.g. { = [, etc.
 123  *
 124  * This class depends on the const array 'lowermap'.
 125  *
 126  ******************************************************/
 127
 128 bool irc::irc_char_traits::eq(char c1st, char c2nd)
 129 {
 130         return lowermap[(unsigned char)c1st] == lowermap[(unsigned char)c2nd];
 131 }
 132
 133 bool irc::irc_char_traits::ne(char c1st, char c2nd)
 134 {
 135         return lowermap[(unsigned char)c1st] != lowermap[(unsigned char)c2nd];
 136 }
 137
 138 bool irc::irc_char_traits::lt(char c1st, char c2nd)
 139 {
 140         return lowermap[(unsigned char)c1st] < lowermap[(unsigned char)c2nd];
 141 }
 142
 143 int irc::irc_char_traits::compare(const char* str1, const char* str2, size_t n)
 144 {
 145         for(unsigned int i = 0; i < n; i++)
 146         {
 147                 if(lowermap[(unsigned char)*str1] > lowermap[(unsigned char)*str2])
 148                         return 1;
 149
 150                 if(lowermap[(unsigned char)*str1] < lowermap[(unsigned char)*str2])
 151                         return -1;
 152
 153                 if(*str1 == 0 || *str2 == 0)
 154                         return 0;
 155
 156                 str1++;
 157                 str2++;
 158         }
 159         return 0;
 160 }
 161
 162 std::string operator+ (std::string& leftval, irc::string& rightval)
 163 {
 164         return leftval + std::string(rightval.c_str());
 165 }
 166
 167 irc::string operator+ (irc::string& leftval, std::string& rightval)
 168 {
 169         return leftval + irc::string(rightval.c_str());
 170 }
 171
 172 bool operator== (std::string& leftval, irc::string& rightval)
 173 {
 174         return (leftval == std::string(rightval.c_str()));
 175 }
 176
 177 bool operator== (irc::string& leftval, std::string& rightval)
 178 {
 179         return (rightval == std::string(leftval.c_str()));
 180 }
 181
 182 const char* irc::irc_char_traits::find(const char* s1, int  n, char c)
 183 {
 184         while(n-- > 0 && lowermap[(unsigned char)*s1] != lowermap[(unsigned char)c])
 185                 s1++;
 186         return s1;
 187 }
 188
 189 /* See hashcomp.h if you care about these... */
 190 std::ostream& operator<<(std::ostream &os, const irc::string &str)
 191 {
 192         return os << str.c_str();
 193 }
 194
 195 std::istream& operator>>(std::istream &is, irc::string &str)
 196 {
 197         std::string tmp;
 198         is >> tmp;
 199         str = tmp.c_str();
 200         return is;
 201 }
 202
 203 irc::tokenstream::tokenstream(const std::string &source) : tokens(source), last_pushed(false)
 204 {
 205         /* Record starting position and current position */
 206         last_starting_position = tokens.begin();
 207         n = tokens.begin();
 208 }
 209
 210 irc::tokenstream::~tokenstream()
 211 {
 212 }
 213
 214 const std::string irc::tokenstream::GetToken()
 215 {
 216         std::string::iterator lsp = last_starting_position;
 217
 218         while (n != tokens.end())
 219         {
 220                 /** Skip multi space, converting "  " into " "
 221                  */
 222                 while ((n+1 != tokens.end()) && (*n == ' ') && (*(n+1) == ' '))
 223                         n++;
 224
 225                 if ((last_pushed) && (*n == ':'))
 226                 {
 227                         /* If we find a token thats not the first and starts with :,
 228                          * this is the last token on the line
 229                          */
 230                         std::string::iterator curr = ++n;
 231                         n = tokens.end();
 232                         return std::string(curr, tokens.end());
 233                 }
 234
 235                 last_pushed = false;
 236
 237                 if ((*n == ' ') || (n+1 == tokens.end()))
 238                 {
 239                         /* If we find a space, or end of string, this is the end of a token.
 240                          */
 241                         last_starting_position = n+1;
 242                         last_pushed = true;
 243
 244                         std::string strip(lsp, n+1 == tokens.end() ? n+1  : n++);
 245                         while ((strip.length()) && (strip.find_last_of(' ') == strip.length() - 1))
 246                                 strip.erase(strip.end() - 1);
 247
 248                         return strip;
 249                 }
 250
 251                 n++;
 252         }
 253         return "";
 254 }
 255
 256 irc::sepstream::sepstream(const std::string &source, char seperator) : tokens(source), sep(seperator)
 257 {
 258         last_starting_position = tokens.begin();
 259         n = tokens.begin();
 260 }
 261
 262 const std::string irc::sepstream::GetToken()
 263 {
 264         std::string::iterator lsp = last_starting_position;
 265
 266         while (n != tokens.end())
 267         {
 268                 if ((*n == sep) || (n+1 == tokens.end()))
 269                 {
 270                         last_starting_position = n+1;
 271                         std::string strip = std::string(lsp, n+1 == tokens.end() ? n+1  : n++);
 272
 273                         while ((strip.length()) && (strip.find_last_of(sep) == strip.length() - 1))
 274                                 strip.erase(strip.end() - 1);
 275
 276                         return strip;
 277                 }
 278
 279                 n++;
 280         }
 281
 282         return "";
 283 }
 284
 285 irc::sepstream::~sepstream()
 286 {
 287 }
 288
 289 std::string irc::hex(const unsigned char *raw, size_t rawsz)
 290 {
 291         if (!rawsz)
 292                 return "";
 293
 294         char buf[rawsz*2+1];
 295         size_t i;
 296
 297         for (i = 0; i < rawsz; i++)
 298         {
 299                 sprintf (&(buf[i*2]), "%02x", raw[i]);
 300         }
 301         buf[i*2] = 0;
 302
 303         return buf;
 304 }
 305
 306 const char* irc::Spacify(char* n)
 307 {
 308         static char x[MAXBUF];
 309         strlcpy(x,n,MAXBUF);
 310         for (char* y = x; *y; y++)
 311                 if (*y == '_')
 312                         *y = ' ';
 313         return x;
 314 }
 315
 316
 317 irc::modestacker::modestacker(bool add) : adding(add)
 318 {
 319         sequence.clear();
 320         sequence.push_back("");
 321 }
 322
 323 void irc::modestacker::Push(char modeletter, const std::string &parameter)
 324 {
 325         *(sequence.begin()) += modeletter;
 326         sequence.push_back(parameter);
 327 }
 328
 329 void irc::modestacker::Push(char modeletter)
 330 {
 331         this->Push(modeletter,"");
 332 }
 333
 334 void irc::modestacker::PushPlus()
 335 {
 336         this->Push('+',"");
 337 }
 338
 339 void irc::modestacker::PushMinus()
 340 {
 341         this->Push('-',"");
 342 }
 343
 344 int irc::modestacker::GetStackedLine(std::deque<std::string> &result)
 345 {
 346         int n = 0;
 347         result.clear();
 348         result.push_back(adding ? "+" : "-");
 349
 350         while (!sequence[0].empty() && (sequence.size() > 1) && (result.size() < MAXMODES+1))
 351         {
 352                 result[0] += *(sequence[0].begin());
 353                 if (!sequence[1].empty())
 354                         result.push_back(sequence[1]);
 355                 sequence[0].erase(sequence[0].begin());
 356                 sequence.erase(sequence.begin() + 1);
 357                 n++;
 358         }
 359
 360         return n;
 361 }
 362
 363 irc::stringjoiner::stringjoiner(const std::string &seperator, const std::vector<std::string> &sequence, int begin, int end)
 364 {
 365         for (int v = begin; v < end; v++)
 366                 joined.append(sequence[v]).append(seperator);
 367         joined.append(sequence[end]);
 368 }
 369
 370 irc::stringjoiner::stringjoiner(const std::string &seperator, const std::deque<std::string> &sequence, int begin, int end)
 371 {
 372         for (int v = begin; v < end; v++)
 373                 joined.append(sequence[v]).append(seperator);
 374         joined.append(sequence[end]);
 375 }
 376
 377 irc::stringjoiner::stringjoiner(const std::string &seperator, const char** sequence, int begin, int end)
 378 {
 379         for (int v = begin; v < end; v++)
 380                 joined.append(sequence[v]).append(seperator);
 381         joined.append(sequence[end]);
 382 }
 383
 384 std::string& irc::stringjoiner::GetJoined()
 385 {
 386         return joined;
 387 }
 388
 389 irc::portparser::portparser(const std::string &source, bool allow_overlapped) : in_range(0), range_begin(0), range_end(0), overlapped(allow_overlapped)
 390 {
 391         sep = new irc::commasepstream(source);
 392         overlap_set.clear();
 393 }
 394
 395 irc::portparser::~portparser()
 396 {
 397         delete sep;
 398 }
 399
 400 bool irc::portparser::Overlaps(long val)
 401 {
 402         if (!overlapped)
 403                 return false;
 404
 405         if (overlap_set.find(val) == overlap_set.end())
 406         {
 407                 overlap_set[val] = true;
 408                 return false;
 409         }
 410         else
 411                 return true;
 412 }
 413
 414 long irc::portparser::GetToken()
 415 {
 416         if (in_range > 0)
 417         {
 418                 in_range++;
 419                 if (in_range <= range_end)
 420                 {
 421                         if (!Overlaps(in_range))
 422                         {
 423                                 return in_range;
 424                         }
 425                         else
 426                         {
 427                                 while (((Overlaps(in_range)) && (in_range <= range_end)))
 428                                         in_range++;
 429
 430                                 if (in_range <= range_end)
 431                                         return in_range;
 432                         }
 433                 }
 434                 else
 435                         in_range = 0;
 436         }
 437
 438         std::string x = sep->GetToken();
 439
 440         if (x == "")
 441                 return 0;
 442
 443         while (Overlaps(atoi(x.c_str())))
 444         {
 445                 x = sep->GetToken();
 446
 447                 if (x == "")
 448                         return 0;
 449         }
 450
 451         std::string::size_type dash = x.rfind('-');
 452         if (dash != std::string::npos)
 453         {
 454                 std::string sbegin = x.substr(0, dash);
 455                 std::string send = x.substr(dash+1, x.length());
 456                 range_begin = atoi(sbegin.c_str());
 457                 range_end = atoi(send.c_str());
 458
 459                 if ((range_begin > 0) && (range_end > 0) && (range_begin < 65536) && (range_end < 65536) && (range_begin < range_end))
 460                 {
 461                         in_range = range_begin;
 462                         return in_range;
 463                 }
 464                 else
 465                 {
 466                         /* Assume its just the one port */
 467                         return atoi(sbegin.c_str());
 468                 }
 469         }
 470         else
 471         {
 472                 return atoi(x.c_str());
 473         }
 474 }
 475