]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - include/hashcomp.h
Crappy warnings
[user/henk/code/inspircd.git] / include / hashcomp.h
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
6  * See: http://www.inspircd.org/wiki/index.php/Credits
7  *
8  * This program is free but copyrighted software; see
9  *            the file COPYING for details.
10  *
11  * ---------------------------------------------------
12  */
13
14 #ifndef _HASHCOMP_H_
15 #define _HASHCOMP_H_
16
17 #include "inspircd_config.h"
18 #include "socket.h"
19 #include "hash_map.h"
20
21 #define _CRT_SECURE_NO_DEPRECATE
22 #define _SCL_SECURE_NO_DEPRECATE
23
24 /*******************************************************
25  * This file contains classes and templates that deal
26  * with the comparison and hashing of 'irc strings'.
27  * An 'irc string' is a string which compares in a
28  * case insensitive manner, and as per RFC 1459 will
29  * treat [ identical to {, ] identical to }, and \
30  * as identical to |.
31  *
32  * Our hashing functions are designed  to accept
33  * std::string and compare/hash them as type irc::string
34  * by converting them internally. This makes them
35  * backwards compatible with other code which is not
36  * aware of irc::string.
37  *******************************************************/
38  
39 using namespace std;
40 using irc::sockets::insp_aton;
41 using irc::sockets::insp_ntoa;
42
43 #ifndef LOWERMAP
44 #define LOWERMAP
45 /** A mapping of uppercase to lowercase, including scandinavian
46  * 'oddities' as specified by RFC1459, e.g. { -> [, and | -> \
47  */
48 unsigned const char lowermap[256] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19,                             /* 0-19 */
49                                 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39,                         /* 20-39 */
50                                 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59,                         /* 40-59 */
51                                 60, 61, 62, 63, 64, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111,             /* 60-79 */
52                                 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 94, 95, 96, 97, 98, 99,           /* 80-99 */
53                                 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119,     /* 100-119 */
54                                 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139,     /* 120-139 */
55                                 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159,     /* 140-159 */
56                                 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179,     /* 160-179 */
57                                 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199,     /* 180-199 */
58                                 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219,     /* 200-219 */
59                                 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239,     /* 220-239 */
60                                 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255                          /* 240-255 */
61 };
62 #endif
63
64 /** The irc namespace contains a number of helper classes.
65  */
66 namespace irc
67 {
68
69         /** This class returns true if two strings match.
70          * Case sensitivity is ignored, and the RFC 'character set'
71          * is adhered to
72          */
73         struct StrHashComp
74         {
75                 /** The operator () does the actual comparison in hash_map
76                  */
77                 bool operator()(const std::string& s1, const std::string& s2) const;
78         };
79
80
81         /** irc::stringjoiner joins string lists into a string, using
82          * the given seperator string.
83          * This class can join a vector of std::string, a deque of
84          * std::string, or a const char** array, using overloaded
85          * constructors.
86          */
87         class CoreExport stringjoiner
88         {
89          private:
90                 /** Output string
91                  */
92                 std::string joined;
93          public:
94                 /** Join elements of a vector, between (and including) begin and end
95                  * @param seperator The string to seperate values with
96                  * @param sequence One or more items to seperate
97                  * @param begin The starting element in the sequence to be joined
98                  * @param end The ending element in the sequence to be joined
99                  */
100                 stringjoiner(const std::string &seperator, const std::vector<std::string> &sequence, int begin, int end);
101                 /** Join elements of a deque, between (and including) begin and end
102                  * @param seperator The string to seperate values with
103                  * @param sequence One or more items to seperate
104                  * @param begin The starting element in the sequence to be joined
105                  * @param end The ending element in the sequence to be joined
106                  */
107                 stringjoiner(const std::string &seperator, const std::deque<std::string> &sequence, int begin, int end);
108                 /** Join elements of an array of char arrays, between (and including) begin and end
109                  * @param seperator The string to seperate values with
110                  * @param sequence One or more items to seperate
111                  * @param begin The starting element in the sequence to be joined
112                  * @param end The ending element in the sequence to be joined
113                  */
114                 stringjoiner(const std::string &seperator, const char** sequence, int begin, int end);
115
116                 /** Get the joined sequence
117                  * @return A reference to the joined string
118                  */
119                 std::string& GetJoined();
120         };
121
122         /** irc::modestacker stacks mode sequences into a list.
123          * It can then reproduce this list, clamped to a maximum of MAXMODES
124          * values per line.
125          */
126         class CoreExport modestacker
127         {
128          private:
129                 /** The mode sequence and its parameters
130                  */
131                 std::deque<std::string> sequence;
132                 /** True if the mode sequence is initially adding
133                  * characters, false if it is initially removing
134                  * them
135                  */
136                 bool adding;
137          public:
138                 /** Construct a new modestacker.
139                  * @param add True if the stack is adding modes,
140                  * false if it is removing them
141                  */
142                 modestacker(bool add);
143                 /** Push a modeletter and its parameter onto the stack.
144                  * No checking is performed as to if this mode actually
145                  * requires a parameter. If you stack invalid mode
146                  * sequences, they will be tidied if and when they are
147                  * passed to a mode parser.
148                  * @param modeletter The mode letter to insert
149                  * @param parameter The parameter for the mode
150                  */
151                 void Push(char modeletter, const std::string &parameter);
152                 /** Push a modeletter without parameter onto the stack.
153                  * No checking is performed as to if this mode actually
154                  * requires a parameter. If you stack invalid mode
155                  * sequences, they will be tidied if and when they are
156                  * passed to a mode parser.
157                  * @param modeletter The mode letter to insert
158                  */
159                 void Push(char modeletter);
160                 /** Push a '+' symbol onto the stack.
161                  */
162                 void PushPlus();
163                 /** Push a '-' symbol onto the stack.
164                  */
165                 void PushMinus();
166                 /** Return zero or more elements which form the
167                  * mode line. This will be clamped to a max of
168                  * MAXMODES+1 items (MAXMODES mode parameters and
169                  * one mode sequence string), and max_line_size
170                  * characters. As specified below, this function
171                  * should be called in a loop until it returns zero,
172                  * indicating there are no more modes to return.
173                  * @param result The deque to populate. This will
174                  * be cleared before it is used.
175                  * @param max_line_size The maximum size of the line
176                  * to build, in characters, seperate to MAXMODES.
177                  * @return The number of elements in the deque.
178                  * The function should be called repeatedly until it
179                  * returns 0, in case there are multiple lines of
180                  * mode changes to be obtained.
181                  */
182                 int GetStackedLine(std::deque<std::string> &result, int max_line_size = 360);
183         };
184
185         /** irc::tokenstream reads a string formatted as per RFC1459 and RFC2812.
186          * It will split the string into 'tokens' each containing one parameter
187          * from the string.
188          * For instance, if it is instantiated with the string:
189          * "PRIVMSG #test :foo bar baz qux"
190          * then each successive call to tokenstream::GetToken() will return
191          * "PRIVMSG", "#test", "foo bar baz qux", "".
192          * Note that if the whole string starts with a colon this is not taken
193          * to mean the string is all one parameter, and the first item in the
194          * list will be ":item". This is to allow for parsing 'source' fields
195          * from data.
196          */
197         class CoreExport tokenstream
198         {
199          private:
200                 /** Original string
201                  */
202                 std::string tokens;
203                 /** Last position of a seperator token
204                  */
205                 std::string::iterator last_starting_position;
206                 /** Current string position
207                  */
208                 std::string::iterator n;
209                 /** True if the last value was an ending value
210                  */
211                 bool last_pushed;
212          public:
213                 /** Create a tokenstream and fill it with the provided data
214                  */
215                 tokenstream(const std::string &source);
216                 ~tokenstream();
217
218                 /** Fetch the next token from the stream
219                  * @return The next token is returned, or an empty string if none remain
220                  */
221                 bool GetToken(std::string &token);
222         };
223
224         /** irc::sepstream allows for splitting token seperated lists.
225          * Each successive call to sepstream::GetToken() returns
226          * the next token, until none remain, at which point the method returns
227          * an empty string.
228          */
229         class CoreExport sepstream : public classbase
230         {
231          private:
232                 /** Original string
233                  */
234                 std::string tokens;
235                 /** Last position of a seperator token
236                  */
237                 std::string::iterator last_starting_position;
238                 /** Current string position
239                  */
240                 std::string::iterator n;
241                 /** Seperator value
242                  */
243                 char sep;
244          public:
245                 /** Create a sepstream and fill it with the provided data
246                  */
247                 sepstream(const std::string &source, char seperator);
248                 virtual ~sepstream();
249
250                 /** Fetch the next token from the stream
251                  * @return The next token is returned, or an empty string if none remain
252                  */
253                 virtual const std::string GetToken();
254                 
255                 /** Fetch the entire remaining stream, without tokenizing
256                  * @return The remaining part of the stream
257                  */
258                 virtual const std::string GetRemaining();
259                 
260                 /** Returns true if the end of the stream has been reached
261                  * @return True if the end of the stream has been reached, otherwise false
262                  */
263                 virtual bool StreamEnd();
264         };
265
266         /** A derived form of sepstream, which seperates on commas
267          */
268         class CoreExport commasepstream : public sepstream
269         {
270          public:
271                 commasepstream(const std::string &source) : sepstream(source, ',')
272                 {
273                 }
274         };
275
276         /** A derived form of sepstream, which seperates on spaces
277          */
278         class CoreExport spacesepstream : public sepstream
279         {
280          public:
281                 spacesepstream(const std::string &source) : sepstream(source, ' ')
282                 {
283                 }
284         };
285
286         /** The portparser class seperates out a port range into integers.
287          * A port range may be specified in the input string in the form
288          * "6660,6661,6662-6669,7020". The end of the stream is indicated by
289          * a return value of 0 from portparser::GetToken(). If you attempt
290          * to specify an illegal range (e.g. one where start >= end, or
291          * start or end < 0) then GetToken() will return the first element
292          * of the pair of numbers.
293          */
294         class CoreExport portparser : public classbase
295         {
296          private:
297                 /** Used to split on commas
298                  */
299                 commasepstream* sep;
300                 /** Current position in a range of ports
301                  */
302                 long in_range;
303                 /** Starting port in a range of ports
304                  */
305                 long range_begin;
306                 /** Ending port in a range of ports
307                  */
308                 long range_end;
309                 /** Allow overlapped port ranges
310                  */
311                 bool overlapped;
312                 /** Used to determine overlapping of ports
313                  * without O(n) algorithm being used
314                  */
315                 std::map<long, bool> overlap_set;
316                 /** Returns true if val overlaps an existing range
317                  */
318                 bool Overlaps(long val);
319          public:
320                 /** Create a portparser and fill it with the provided data
321                  * @param source The source text to parse from
322                  * @param allow_overlapped Allow overlapped ranges
323                  */
324                 portparser(const std::string &source, bool allow_overlapped = true);
325                 /** Frees the internal commasepstream object
326                  */
327                 ~portparser();
328                 /** Fetch the next token from the stream
329                  * @return The next port number is returned, or 0 if none remain
330                  */
331                 long GetToken();
332         };
333
334         /** Used to hold a bitfield definition in dynamicbitmask.
335          * You must be allocated one of these by dynamicbitmask::Allocate(),
336          * you should not fill the values yourself!
337          */
338         typedef std::pair<size_t, unsigned char> bitfield;
339
340         /** The irc::dynamicbitmask class is used to maintain a bitmap of
341          * boolean values, which can grow to any reasonable size no matter
342          * how many bitfields are in it.
343          *
344          * It starts off at 32 bits in size, large enough to hold 32 boolean
345          * values, with a memory allocation of 8 bytes. If you allocate more
346          * than 32 bits, the class will grow the bitmap by 8 bytes at a time
347          * for each set of 8 bitfields you allocate with the Allocate()
348          * method.
349          *
350          * This method is designed so that multiple modules can be allocated
351          * bit values in a bitmap dynamically, rather than having to define
352          * costs in a fixed size unsigned integer and having the possibility
353          * of collisions of values in different third party modules.
354          *
355          * IMPORTANT NOTE:
356          *
357          * To use this class, you must derive from it.
358          * This is because each derived instance has its own freebits array
359          * which can determine what bitfields are allocated on a TYPE BY TYPE
360          * basis, e.g. an irc::dynamicbitmask type for userrecs, and one for
361          * chanrecs, etc. You should inheret it in a very simple way as follows.
362          * The base class will resize and maintain freebits as required, you are
363          * just required to make the pointer static and specific to this class
364          * type.
365          *
366          * \code
367          * class mydbitmask : public irc::dynamicbitmask
368          * {
369          *  private:
370          *
371          *      static unsigned char* freebits;
372          *
373          *  public:
374          *
375          *      mydbitmask() : irc::dynamicbitmask()
376          *      {
377          *          freebits = new unsigned char[this->bits_size];
378          *          memset(freebits, 0, this->bits_size);
379          *      }
380          *
381          *      ~mydbitmask()
382          *      {
383          *          delete[] freebits;
384          *      }
385          *
386          *      unsigned char* GetFreeBits()
387          *      {
388          *          return freebits;
389          *      }
390          *
391          *      void SetFreeBits(unsigned char* freebt)
392          *      {
393          *          freebits = freebt;
394          *      }
395          * };
396          * \endcode
397          */
398         class CoreExport dynamicbitmask : public classbase
399         {
400          private:
401                 /** Data bits. We start with four of these,
402                  * and we grow the bitfield as we allocate
403                  * more than 32 entries with Allocate().
404                  */
405                 unsigned char* bits;
406          protected:
407                 /** Current set size (size of freebits and bits).
408                  * Both freebits and bits will ALWAYS be the
409                  * same length.
410                  */
411                 unsigned char bits_size;
412          public:
413                 /** Allocate the initial memory for bits and
414                  * freebits and zero the memory.
415                  */
416                 dynamicbitmask();
417
418                 /** Free the memory used by bits and freebits
419                  */
420                 virtual ~dynamicbitmask();
421
422                 /** Allocate an irc::bitfield.
423                  * @return An irc::bitfield which can be used
424                  * with Get, Deallocate and Toggle methods.
425                  * @throw Can throw std::bad_alloc if there is
426                  * no ram left to grow the bitmask.
427                  */
428                 bitfield Allocate();
429
430                 /** Deallocate an irc::bitfield.
431                  * @param An irc::bitfield to deallocate.
432                  * @return True if the bitfield could be
433                  * deallocated, false if it could not.
434                  */
435                 bool Deallocate(bitfield &pos);
436
437                 /** Toggle the value of a bitfield.
438                  * @param pos A bitfield to allocate, previously
439                  * allocated by dyamicbitmask::Allocate().
440                  * @param state The state to set the field to.
441                  */
442                 void Toggle(bitfield &pos, bool state);
443
444                 /** Get the value of a bitfield.
445                  * @param pos A bitfield to retrieve, previously
446                  * allocated by dyamicbitmask::Allocate().
447                  * @return The value of the bitfield.
448                  * @throw Will throw ModuleException if the bitfield
449                  * you provide is outside of the range
450                  * 0 >= bitfield.first < size_bits.
451                  */
452                 bool Get(bitfield &pos);
453
454                 /** Return the size in bytes allocated to the bits
455                  * array.
456                  * Note that the actual allocation is twice this,
457                  * as there are an equal number of bytes allocated
458                  * for the freebits array.
459                  */
460                 unsigned char GetSize();
461
462                 virtual unsigned char* GetFreeBits() { return NULL; }
463
464                 virtual void SetFreeBits(unsigned char* freebits) { }
465         };
466
467         /** The irc_char_traits class is used for RFC-style comparison of strings.
468          * This class is used to implement irc::string, a case-insensitive, RFC-
469          * comparing string class.
470          */
471         struct irc_char_traits : std::char_traits<char> {
472
473                 /** Check if two chars match
474                  */
475                 static bool eq(char c1st, char c2nd);
476
477                 /** Check if two chars do NOT match
478                  */
479                 static bool ne(char c1st, char c2nd);
480
481                 /** Check if one char is less than another
482                  */
483                 static bool lt(char c1st, char c2nd);
484
485                 /** Compare two strings of size n
486                  */
487                 static CoreExport int compare(const char* str1, const char* str2, size_t n);
488
489                 /** Find a char within a string up to position n
490                  */
491                 static CoreExport const char* find(const char* s1, int  n, char c);
492         };
493
494         CoreExport std::string hex(const unsigned char *raw, size_t rawsz);
495
496         /** This typedef declares irc::string based upon irc_char_traits
497          */
498         typedef basic_string<char, irc_char_traits, allocator<char> > string;
499
500         CoreExport const char* Spacify(const char* n);
501 }
502
503 /* Define operators for using >> and << with irc::string to an ostream on an istream. */
504 /* This was endless fun. No. Really. */
505 /* It was also the first core change Ommeh made, if anyone cares */
506 inline std::ostream& operator<<(std::ostream &os, const irc::string &str) { return os << str.c_str(); }
507 inline std::istream& operator>>(std::istream &is, irc::string &str)
508 {
509         std::string tmp;
510         is >> tmp;
511         str = tmp.c_str();
512         return is;
513 }
514
515 /* Define operators for + and == with irc::string to std::string for easy assignment
516  * and comparison - Brain
517  */
518 inline std::string operator+ (std::string& leftval, irc::string& rightval)
519 {
520         return leftval + std::string(rightval.c_str());
521 }
522
523 inline irc::string operator+ (irc::string& leftval, std::string& rightval)
524 {
525         return leftval + irc::string(rightval.c_str());
526 }
527
528 inline bool operator== (const std::string& leftval, const irc::string& rightval)
529 {
530         return (leftval.c_str() == rightval);
531 }
532
533 inline bool operator== (const irc::string& leftval, const std::string& rightval)
534 {
535         return (leftval == rightval.c_str());
536 }
537
538 inline std::string assign(const irc::string &other) { return other.c_str(); }
539 inline irc::string assign(const std::string &other) { return other.c_str(); }
540 inline std::string& trim(std::string &str)
541 {
542         std::string::size_type start = str.find_first_not_of(" ");
543         std::string::size_type end = str.find_last_not_of(" ");
544         if (start == std::string::npos || end == std::string::npos)
545                 str = "";
546         else
547                 str = str.substr(start, end-start+1);
548
549         return str;
550 }
551
552 /* Hashing stuff is totally different on vc++'s hash_map implementation, so to save a buttload of #ifdefs we'll just
553    do it all at once - Burlex */
554
555 namespace nspace
556 {
557         /** Hashing function to hash irc::string
558          */
559 #ifdef WINDOWS
560         template<> class CoreExport hash_compare<irc::string, std::less<irc::string> >
561         {
562         public:
563                 enum { bucket_size = 4, min_buckets = 8 };                      // Got these numbers from the CRT source,
564                                                                                                                         // if anyone wants to change them feel free.
565                 bool operator()(const irc::string & s1, const irc::string & s2) const
566                 {
567                         if(s1.length() != s2.length()) return true;
568                         return (irc::irc_char_traits::compare(s1.c_str(), s2.c_str(), s1.length()) < 0);
569                 }
570                 
571                 size_t operator()(const irc::string & s) const;
572         };
573
574         template<> class CoreExport hash_compare<std::string, std::less<std::string> >
575         {
576         public:
577                 enum { bucket_size = 4, min_buckets = 8 };
578                 bool operator()(const std::string & s1, const std::string & s2) const
579                 {
580                         if(s1.length() != s2.length()) return true;
581                         return (irc::irc_char_traits::compare(s1.c_str(), s2.c_str(), s1.length()) < 0);
582                 }
583                 
584                 /** Hash a std::string using RFC1459 case sensitivity rules
585                 * @param s A string to hash
586                 * @return The hash value
587                 */
588                 size_t operator()(const std::string & s) const;
589         };
590 #else
591         template<> struct hash<irc::string>
592         {
593                 size_t operator()(const irc::string &s) const;
594         };
595
596         template<> struct hash<std::string>
597         {
598                 /** Hash a std::string using RFC1459 case sensitivity rules
599                 * @param s A string to hash
600                 * @return The hash value
601                 */
602                 size_t operator()(const string &s) const;
603         };
604 #endif
605
606         /** Convert a string to lower case respecting RFC1459
607         * @param n A string to lowercase
608         */
609         void strlower(char *n);
610 }
611
612 #endif