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++, so we don't hash the \0 */
  91                 t = 5 * t + lowermap[(unsigned char)*x];
  92         return size_t(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                 if ((last_pushed) && (*n == ':'))
 221                 {
 222                         /* If we find a token thats not the first and starts with :,
 223                          * this is the last token on the line
 224                          */
 225                         std::string::iterator curr = ++n;
 226                         n = tokens.end();
 227                         return std::string(curr, tokens.end());
 228                 }
 229
 230                 last_pushed = false;
 231
 232                 if ((*n == ' ') || (n+1 == tokens.end()))
 233                 {
 234                         /* If we find a space, or end of string, this is the end of a token.
 235                          */
 236                         last_starting_position = n+1;
 237                         last_pushed = true;
 238
 239                         std::string strip(lsp, n+1 == tokens.end() ? n+1  : n++);
 240                         while ((strip.length()) && (strip.find_last_of(' ') == strip.length() - 1))
 241                                 strip.erase(strip.end() - 1);
 242
 243                         return strip;
 244                 }
 245
 246                 n++;
 247         }
 248         return "";
 249 }
 250
 251 irc::sepstream::sepstream(const std::string &source, char seperator) : tokens(source), sep(seperator)
 252 {
 253         last_starting_position = tokens.begin();
 254         n = tokens.begin();
 255 }
 256
 257 const std::string irc::sepstream::GetToken()
 258 {
 259         std::string::iterator lsp = last_starting_position;
 260
 261         while (n != tokens.end())
 262         {
 263                 if ((*n == sep) || (n+1 == tokens.end()))
 264                 {
 265                         last_starting_position = n+1;
 266                         std::string strip = std::string(lsp, n+1 == tokens.end() ? n+1  : n++);
 267
 268                         while ((strip.length()) && (strip.find_last_of(sep) == strip.length() - 1))
 269                                 strip.erase(strip.end() - 1);
 270
 271                         return strip;
 272                 }
 273
 274                 n++;
 275         }
 276
 277         return "";
 278 }
 279
 280 irc::sepstream::~sepstream()
 281 {
 282 }
 283
 284 std::string irc::hex(const unsigned char *raw, size_t rawsz)
 285 {
 286         if (!rawsz)
 287                 return "";
 288
 289         char buf[rawsz*2+1];
 290         size_t i;
 291
 292         for (i = 0; i < rawsz; i++)
 293         {
 294                 sprintf (&(buf[i*2]), "%02x", raw[i]);
 295         }
 296         buf[i*2] = 0;
 297
 298         return buf;
 299 }
 300
 301 const char* irc::Spacify(char* n)
 302 {
 303         static char x[MAXBUF];
 304         strlcpy(x,n,MAXBUF);
 305         for (char* y = x; *y; y++)
 306                 if (*y == '_')
 307                         *y = ' ';
 308         return x;
 309 }
 310
 311
 312 irc::modestacker::modestacker(bool add) : adding(add)
 313 {
 314         sequence.clear();
 315         sequence.push_back("");
 316 }
 317
 318 void irc::modestacker::Push(char modeletter, const std::string &parameter)
 319 {
 320         *(sequence.begin()) += modeletter;
 321         sequence.push_back(parameter);
 322 }
 323
 324 void irc::modestacker::Push(char modeletter)
 325 {
 326         this->Push(modeletter,"");
 327 }
 328
 329 void irc::modestacker::PushPlus()
 330 {
 331         this->Push('+',"");
 332 }
 333
 334 void irc::modestacker::PushMinus()
 335 {
 336         this->Push('-',"");
 337 }
 338
 339 int irc::modestacker::GetStackedLine(std::deque<std::string> &result)
 340 {
 341         int n = 0;
 342         result.clear();
 343         result.push_back(adding ? "+" : "-");
 344
 345         while (!sequence[0].empty() && (sequence.size() > 1) && (result.size() < MAXMODES+1))
 346         {
 347                 result[0] += *(sequence[0].begin());
 348                 if (!sequence[1].empty())
 349                         result.push_back(sequence[1]);
 350                 sequence[0].erase(sequence[0].begin());
 351                 sequence.erase(sequence.begin() + 1);
 352                 n++;
 353         }
 354
 355         return n;
 356 }
 357
 358 irc::stringjoiner::stringjoiner(const std::string &seperator, const std::vector<std::string> &sequence, int begin, int end)
 359 {
 360         for (int v = begin; v < end; v++)
 361                 joined.append(sequence[v]).append(seperator);
 362         joined.append(sequence[end]);
 363 }
 364
 365 irc::stringjoiner::stringjoiner(const std::string &seperator, const std::deque<std::string> &sequence, int begin, int end)
 366 {
 367         for (int v = begin; v < end; v++)
 368                 joined.append(sequence[v]).append(seperator);
 369         joined.append(sequence[end]);
 370 }
 371
 372 irc::stringjoiner::stringjoiner(const std::string &seperator, const char** sequence, int begin, int end)
 373 {
 374         for (int v = begin; v < end; v++)
 375                 joined.append(sequence[v]).append(seperator);
 376         joined.append(sequence[end]);
 377 }
 378
 379 std::string& irc::stringjoiner::GetJoined()
 380 {
 381         return joined;
 382 }
 383