]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/hashcomp.cpp
'svn propset -R svn:eol-style CR *' Set to UNIX-style always. Binaries are auto skipp...
[user/henk/code/inspircd.git] / src / hashcomp.cpp
1 /*       +------------------------------------+\r *       | Inspire Internet Relay Chat Daemon |\r *       +------------------------------------+\r *\r *  InspIRCd: (C) 2002-2007 InspIRCd Development Team\r * See: http://www.inspircd.org/wiki/index.php/Credits\r *\r * This program is free but copyrighted software; see\r *            the file COPYING for details.\r *\r * ---------------------------------------------------\r */\r\r#include "inspircd.h"\r#include "hashcomp.h"\r#ifndef WIN32\r#include <ext/hash_map>\r#define nspace __gnu_cxx\r#else\r#include <hash_map>\r#define nspace stdext\rusing stdext::hash_map;\r#endif\r\r/******************************************************\r *\r * The hash functions of InspIRCd are the centrepoint\r * of the entire system. If these functions are\r * inefficient or wasteful, the whole program suffers\r * as a result. A lot of C programmers in the ircd\r * scene spend a lot of time debating (arguing) about\r * the best way to write hash functions to hash irc\r * nicknames, channels etc.\r * We are lucky as C++ developers as hash_map does\r * a lot of this for us. It does intellegent memory\r * requests, bucketing, search functions, insertion\r * and deletion etc. All we have to do is write some\r * overloaded comparison and hash value operators which\r * cause it to act in an irc-like way. The features we\r * add to the standard hash_map are:\r *\r * Case insensitivity: The hash_map will be case\r * insensitive.\r *\r * Scandanavian Comparisons: The characters [, ], \ will\r * be considered the lowercase of {, } and |.\r *\r ******************************************************/\r\rusing namespace irc::sockets;\r\r/* convert a string to lowercase. Note following special circumstances\r * taken from RFC 1459. Many "official" server branches still hold to this\r * rule so i will too;\r *\r *  Because of IRC's scandanavian origin, the characters {}| are\r *  considered to be the lower case equivalents of the characters []\,\r *  respectively. This is a critical issue when determining the\r *  equivalence of two nicknames.\r */\rvoid nspace::strlower(char *n)\r{\r        if (n)\r {\r              for (char* t = n; *t; t++)\r                     *t = lowermap[(unsigned char)*t];\r      }\r}\r\r#ifndef WIN32\rsize_t nspace::hash<string>::operator()(const string &s) const\r#else\rsize_t nspace::hash_compare<string, std::less<string> >::operator()(const string &s) const\r#endif\r{\r    /* XXX: NO DATA COPIES! :)\r      * The hash function here is practically\r        * a copy of the one in STL's hash_fun.h,\r       * only with *x replaced with lowermap[*x].\r     * This avoids a copy to use hash<const char*>\r  */\r    register size_t t = 0;\r for (std::string::const_iterator x = s.begin(); x != s.end(); ++x) /* ++x not x++, as its faster */\r            t = 5 * t + lowermap[(unsigned char)*x];\r       return t;\r}\r\r#ifndef WIN32\rsize_t nspace::hash<irc::string>::operator()(const irc::string &s) const\r#else\rsize_t nspace::hash_compare<irc::string, std::less<irc::string> >::operator()(const irc::string &s) const\r#endif\r{\r   register size_t t = 0;\r for (irc::string::const_iterator x = s.begin(); x != s.end(); ++x) /* ++x not x++, as its faster */\r            t = 5 * t + lowermap[(unsigned char)*x];\r       return t;\r}\r\rbool irc::StrHashComp::operator()(const std::string& s1, const std::string& s2) const\r{\r   unsigned char* n1 = (unsigned char*)s1.c_str();\r        unsigned char* n2 = (unsigned char*)s2.c_str();\r        for (; *n1 && *n2; n1++, n2++)\r         if (lowermap[*n1] != lowermap[*n2])\r                    return false;\r  return (lowermap[*n1] == lowermap[*n2]);\r}\r\r/******************************************************\r *\r * This is the implementation of our special irc::string\r * class which is a case-insensitive equivalent to\r * std::string which is not only case-insensitive but\r * can also do scandanavian comparisons, e.g. { = [, etc.\r *\r * This class depends on the const array 'lowermap'.\r *\r ******************************************************/\r\rbool irc::irc_char_traits::eq(char c1st, char c2nd)\r{\r  return lowermap[(unsigned char)c1st] == lowermap[(unsigned char)c2nd];\r}\r\rbool irc::irc_char_traits::ne(char c1st, char c2nd)\r{\r        return lowermap[(unsigned char)c1st] != lowermap[(unsigned char)c2nd];\r}\r\rbool irc::irc_char_traits::lt(char c1st, char c2nd)\r{\r        return lowermap[(unsigned char)c1st] < lowermap[(unsigned char)c2nd];\r}\r\rint irc::irc_char_traits::compare(const char* str1, const char* str2, size_t n)\r{\r     for(unsigned int i = 0; i < n; i++)\r    {\r              if(lowermap[(unsigned char)*str1] > lowermap[(unsigned char)*str2])\r                    return 1;\r\r             if(lowermap[(unsigned char)*str1] < lowermap[(unsigned char)*str2])\r                    return -1;\r\r            if(*str1 == 0 || *str2 == 0)\r                   return 0;\r\r             str1++;\r                str2++;\r        }\r      return 0;\r}\r\rconst char* irc::irc_char_traits::find(const char* s1, int  n, char c)\r{\r  while(n-- > 0 && lowermap[(unsigned char)*s1] != lowermap[(unsigned char)c])\r           s1++;\r  return s1;\r}\r\rirc::tokenstream::tokenstream(const std::string &source) : tokens(source), last_pushed(false)\r{\r  /* Record starting position and current position */\r    last_starting_position = tokens.begin();\r       n = tokens.begin();\r}\r\rirc::tokenstream::~tokenstream()\r{\r}\r\rbool irc::tokenstream::GetToken(std::string &token)\r{\r     std::string::iterator lsp = last_starting_position;\r\r   while (n != tokens.end())\r      {\r              /** Skip multi space, converting "  " into " "\r          */\r            while ((n+1 != tokens.end()) && (*n == ' ') && (*(n+1) == ' '))\r                        n++;\r\r          if ((last_pushed) && (*n == ':'))\r              {\r                      /* If we find a token thats not the first and starts with :,\r                    * this is the last token on the line\r                   */\r                    std::string::iterator curr = ++n;\r                      n = tokens.end();\r                      token = std::string(curr, tokens.end());\r                       return true;\r           }\r\r             last_pushed = false;\r\r          if ((*n == ' ') || (n+1 == tokens.end()))\r              {\r                      /* If we find a space, or end of string, this is the end of a token.\r                    */\r                    last_starting_position = n+1;\r                  last_pushed = true;\r\r                   std::string strip(lsp, n+1 == tokens.end() ? n+1  : n++);\r                      while ((strip.length()) && (strip.find_last_of(' ') == strip.length() - 1))\r                            strip.erase(strip.end() - 1);\r\r                 token = strip;\r                 return !token.empty();\r         }\r\r             n++;\r   }\r      token.clear();\r return false;\r}\r\rbool irc::tokenstream::GetToken(irc::string &token)\r{\r std::string stdstring;\r bool returnval = GetToken(stdstring);\r  token = assign(stdstring);\r     return returnval;\r}\r\rbool irc::tokenstream::GetToken(int &token)\r{\r     std::string tok;\r       bool returnval = GetToken(tok);\r        token = ConvToInt(tok);\r        return returnval;\r}\r\rbool irc::tokenstream::GetToken(long &token)\r{\r    std::string tok;\r       bool returnval = GetToken(tok);\r        token = ConvToInt(tok);\r        return returnval;\r}\r\rirc::sepstream::sepstream(const std::string &source, char seperator) : tokens(source), sep(seperator)\r{\r   last_starting_position = tokens.begin();\r       n = tokens.begin();\r}\r\rconst std::string irc::sepstream::GetToken()\r{\r  std::string::iterator lsp = last_starting_position;\r\r   while (n != tokens.end())\r      {\r              if ((*n == sep) || (n+1 == tokens.end()))\r              {\r                      last_starting_position = n+1;\r                  std::string strip = std::string(lsp, n+1 == tokens.end() ? n+1  : n++);\r\r                       while ((strip.length()) && (strip.find_last_of(sep) == strip.length() - 1))\r                            strip.erase(strip.end() - 1);\r\r                 return strip;\r          }\r\r             n++;\r   }\r\r     return "";\r}\r\rconst std::string irc::sepstream::GetRemaining()\r{\r       return std::string(n, tokens.end());\r}\r\rbool irc::sepstream::StreamEnd()\r{\r     return ((n + 1) == tokens.end());\r}\r\rirc::sepstream::~sepstream()\r{\r}\r\rstd::string irc::hex(const unsigned char *raw, size_t rawsz)\r{\r  if (!rawsz)\r            return "";\r\r    /* EWW! This used to be using sprintf, which is WAY inefficient. -Special */\r   \r       const char *hex = "0123456789abcdef";\r  static char hexbuf[MAXBUF];\r\r   size_t i, j;\r   for (i = 0, j = 0; j < rawsz; ++j)\r     {\r              hexbuf[i++] = hex[raw[j] / 16];\r                hexbuf[i++] = hex[raw[j] % 16];\r        }\r      hexbuf[i] = 0;\r\r        return hexbuf;\r}\r\rCoreExport const char* irc::Spacify(const char* n)\r{\r static char x[MAXBUF];\r strlcpy(x,n,MAXBUF);\r   for (char* y = x; *y; y++)\r             if (*y == '_')\r                 *y = ' ';\r      return x;\r}\r\r\rirc::modestacker::modestacker(bool add) : adding(add)\r{\r  sequence.clear();\r      sequence.push_back("");\r}\r\rvoid irc::modestacker::Push(char modeletter, const std::string &parameter)\r{\r        *(sequence.begin()) += modeletter;\r     sequence.push_back(parameter);\r}\r\rvoid irc::modestacker::Push(char modeletter)\r{\r       this->Push(modeletter,"");\r}\r\rvoid irc::modestacker::PushPlus()\r{\r      this->Push('+',"");\r}\r\rvoid irc::modestacker::PushMinus()\r{\r    this->Push('-',"");\r}\r\rint irc::modestacker::GetStackedLine(std::deque<std::string> &result, int max_line_size)\r{\r      if (sequence.empty())\r  {\r              result.clear();\r                return 0;\r      }\r\r     int n = 0;\r     int size = 1; /* Account for initial +/- char */\r       int nextsize = 0;\r      result.clear();\r        result.push_back(adding ? "+" : "-");\r\r if (sequence.size() > 1)\r               nextsize = sequence[1].length() + 2;\r\r  while (!sequence[0].empty() && (sequence.size() > 1) && (result.size() < MAXMODES) && ((size + nextsize) < max_line_size))\r     {\r              result[0] += *(sequence[0].begin());\r           if (!sequence[1].empty())\r              {\r                      result.push_back(sequence[1]);\r                 size += nextsize; /* Account for mode character and whitespace */\r              }\r              sequence[0].erase(sequence[0].begin());\r                sequence.erase(sequence.begin() + 1);\r\r         if (sequence.size() > 1)\r                       nextsize = sequence[1].length() + 2;\r\r          n++;\r   }\r\r     return n;\r}\r\rirc::stringjoiner::stringjoiner(const std::string &seperator, const std::vector<std::string> &sequence, int begin, int end)\r{\r     for (int v = begin; v < end; v++)\r              joined.append(sequence[v]).append(seperator);\r  joined.append(sequence[end]);\r}\r\rirc::stringjoiner::stringjoiner(const std::string &seperator, const std::deque<std::string> &sequence, int begin, int end)\r{\r  for (int v = begin; v < end; v++)\r              joined.append(sequence[v]).append(seperator);\r  joined.append(sequence[end]);\r}\r\rirc::stringjoiner::stringjoiner(const std::string &seperator, const char** sequence, int begin, int end)\r{\r    for (int v = begin; v < end; v++)\r              joined.append(sequence[v]).append(seperator);\r  joined.append(sequence[end]);\r}\r\rstd::string& irc::stringjoiner::GetJoined()\r{\r return joined;\r}\r\rirc::portparser::portparser(const std::string &source, bool allow_overlapped) : in_range(0), range_begin(0), range_end(0), overlapped(allow_overlapped)\r{\r    sep = new irc::commasepstream(source);\r overlap_set.clear();\r}\r\rirc::portparser::~portparser()\r{\r       delete sep;\r}\r\rbool irc::portparser::Overlaps(long val)\r{\r      if (!overlapped)\r               return false;\r\r if (overlap_set.find(val) == overlap_set.end())\r        {\r              overlap_set[val] = true;\r               return false;\r  }\r      else\r           return true;\r}\r\rlong irc::portparser::GetToken()\r{\r     if (in_range > 0)\r      {\r              in_range++;\r            if (in_range <= range_end)\r             {\r                      if (!Overlaps(in_range))\r                       {\r                              return in_range;\r                       }\r                      else\r                   {\r                              while (((Overlaps(in_range)) && (in_range <= range_end)))\r                                      in_range++;\r                            \r                               if (in_range <= range_end)\r                                     return in_range;\r                       }\r              }\r              else\r                   in_range = 0;\r  }\r\r     std::string x = sep->GetToken();\r\r      if (x.empty())\r         return 0;\r\r     while (Overlaps(atoi(x.c_str())))\r      {\r              x = sep->GetToken();\r\r          if (x.empty())\r                 return 0;\r      }\r\r     std::string::size_type dash = x.rfind('-');\r    if (dash != std::string::npos)\r {\r              std::string sbegin = x.substr(0, dash);\r                std::string send = x.substr(dash+1, x.length());\r               range_begin = atoi(sbegin.c_str());\r            range_end = atoi(send.c_str());\r\r               if ((range_begin > 0) && (range_end > 0) && (range_begin < 65536) && (range_end < 65536) && (range_begin < range_end))\r         {\r                      in_range = range_begin;\r                        return in_range;\r               }\r              else\r           {\r                      /* Assume its just the one port */\r                     return atoi(sbegin.c_str());\r           }\r      }\r      else\r   {\r              return atoi(x.c_str());\r        }\r}\r\rirc::dynamicbitmask::dynamicbitmask() : bits_size(4)\r{\r    /* We start with 4 bytes allocated which is room\r        * for 4 items. Something makes me doubt its worth\r      * allocating less than 4 bytes.\r        */\r    bits = new unsigned char[bits_size];\r   memset(bits, 0, bits_size);\r}\r\rirc::dynamicbitmask::~dynamicbitmask()\r{\r        /* Tidy up the entire used memory on delete */\r delete[] bits;\r}\r\rirc::bitfield irc::dynamicbitmask::Allocate()\r{\r      /* Yeah, this isnt too efficient, however a module or the core\r  * should only be allocating bitfields on load, the Toggle and\r  * Get methods are O(1) as these are called much more often.\r    */\r    unsigned char* freebits = this->GetFreeBits();\r for (unsigned char i = 0; i < bits_size; i++)\r  {\r              /* Yes, this is right. You'll notice we terminate the  loop when !current_pos,\r          * this is because we logic shift our bit off the end of unsigned char, and its\r                 * lost, making the loop counter 0 when we're done.\r             */\r            for (unsigned char current_pos = 1; current_pos; current_pos = current_pos << 1)\r               {\r                      if (!(freebits[i] & current_pos))\r                      {\r                              freebits[i] |= current_pos;\r                            return std::make_pair(i, current_pos);\r                 }\r              }\r      }\r      /* We dont have any free space left, increase by one */\r\r       if (bits_size == 255)\r          /* Oh dear, cant grow it any further */\r                throw std::bad_alloc();\r\r       unsigned char old_bits_size = bits_size;\r       bits_size++;\r   /* Allocate new bitfield space */\r      unsigned char* temp_bits = new unsigned char[bits_size];\r       unsigned char* temp_freebits = new unsigned char[bits_size];\r   /* Copy the old data in */\r     memcpy(temp_bits, bits, old_bits_size);\r        memcpy(temp_freebits, freebits, old_bits_size);\r        /* Delete the old data pointers */\r     delete[] bits;\r delete[] freebits;\r     /* Swap the pointers over so now the new \r       * pointers point to our member values\r  */\r    bits = temp_bits;\r      freebits = temp_freebits;\r      this->SetFreeBits(freebits);\r   /* Initialize the new byte on the end of\r        * the bitfields, pre-allocate the one bit\r      * for this allocation\r  */\r    bits[old_bits_size] = 0;\r       freebits[old_bits_size] = 1;\r   /* We already know where we just allocated\r      * the bitfield, so no loop needed\r      */\r    return std::make_pair(old_bits_size, 1);\r}\r\rbool irc::dynamicbitmask::Deallocate(irc::bitfield &pos)\r{\r /* We dont bother to shrink the bitfield\r        * on deallocation, the most we could do\r        * is save one byte (!) and this would cost\r     * us a loop (ugly O(n) stuff) so we just\r       * clear the bit and leave the memory\r   * claimed -- nobody will care about one\r        * byte.\r        */\r    if (pos.first < bits_size)\r     {\r              this->GetFreeBits()[pos.first] &= ~pos.second;\r         return true;\r   }\r      /* They gave a bitfield outside of the\r  * length of our array. BAD programmer.\r         */\r    return false;\r}\r\rvoid irc::dynamicbitmask::Toggle(irc::bitfield &pos, bool state)\r{\r    /* Range check the value */\r    if (pos.first < bits_size)\r     {\r              if (state)\r                     /* Set state, OR the state in */\r                       bits[pos.first] |= pos.second;\r         else\r                   /* Clear state, AND the !state out */\r                  bits[pos.first] &= ~pos.second;\r        }\r}\r\rbool irc::dynamicbitmask::Get(irc::bitfield &pos)\r{\r       /* Range check the value */\r    if (pos.first < bits_size)\r             return (bits[pos.first] & pos.second);\r else\r           /* We can't return false, otherwise we can't\r            * distinguish between failure and a cleared bit!\r               * Our only sensible choice is to throw (ew).\r           */\r            throw ModuleException("irc::dynamicbitmask::Get(): Invalid bitfield, out of range");\r}\r\runsigned char irc::dynamicbitmask::GetSize()\r{\r return bits_size;\r}\r\r