src/inspstring.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 /* $Core */
  15
  16 #include "inspircd.h"
  17
  18 /*
  19  * Copyright (c) 1998 Todd C. Miller <Todd.Miller@courtesan.com>
  20  * All rights reserved.
  21  *
  22  * Redistribution and use in source and binary forms, with or without
  23  * modification, are permitted provided that the following conditions
  24  * are met:
  25  * 1. Redistributions of source code must retain the above copyright
  26  *    notice, this list of conditions and the following disclaimer.
  27  * 2. Redistributions in binary form must reproduce the above copyright
  28  *    notice, this list of conditions and the following disclaimer in the
  29  *    documentation and/or other materials provided with the distribution.
  30  * 3. The name of the author may not be used to endorse or promote products
  31  *    derived from this software without specific prior written permission.
  32  *
  33  * THIS SOFTWARE IS PROVIDED `AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
  34  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
  35  * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
  36  * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  37  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  38  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  39  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  40  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  41  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  42  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  43  */
  44
  45 #ifndef HAS_STRLCPY
  46 CoreExport size_t strlcat(char *dst, const char *src, size_t siz)
  47 {
  48         char *d = dst;
  49         const char *s = src;
  50         size_t n = siz, dlen;
  51
  52         while (n-- != 0 && *d != '\0')
  53                 d++;
  54
  55         dlen = d - dst;
  56         n = siz - dlen;
  57
  58         if (n == 0)
  59                 return(dlen + strlen(s));
  60
  61         while (*s != '\0')
  62         {
  63                 if (n != 1)
  64                 {
  65                         *d++ = *s;
  66                         n--;
  67                 }
  68
  69                 s++;
  70         }
  71
  72         *d = '\0';
  73         return(dlen + (s - src)); /* count does not include NUL */
  74 }
  75
  76 CoreExport size_t strlcpy(char *dst, const char *src, size_t siz)
  77 {
  78         char *d = dst;
  79         const char *s = src;
  80         size_t n = siz;
  81
  82         /* Copy as many bytes as will fit */
  83         if (n != 0 && --n != 0)
  84         {
  85                 do
  86                 {
  87                         if ((*d++ = *s++) == 0)
  88                                 break;
  89                 } while (--n != 0);
  90         }
  91
  92         /* Not enough room in dst, add NUL and traverse rest of src */
  93         if (n == 0)
  94         {
  95                 if (siz != 0)
  96                         *d = '\0'; /* NUL-terminate dst */
  97                 while (*s++);
  98         }
  99
 100         return(s - src - 1); /* count does not include NUL */
 101 }
 102 #endif
 103
 104 CoreExport int charlcat(char* x,char y,int z)
 105 {
 106         char* x__n = x;
 107         int v = 0;
 108
 109         while(*x__n++)
 110                 v++;
 111
 112         if (v < z - 1)
 113         {
 114                 *--x__n = y;
 115                 *++x__n = 0;
 116         }
 117
 118         return v;
 119 }
 120
 121 CoreExport bool charremove(char* mp, char remove)
 122 {
 123         char* mptr = mp;
 124         bool shift_down = false;
 125
 126         while (*mptr)
 127         {
 128                 if (*mptr == remove)
 129                 shift_down = true;
 130
 131                 if (shift_down)
 132                         *mptr = *(mptr+1);
 133
 134                 mptr++;
 135         }
 136
 137         return shift_down;
 138 }
 139
 140 static const char hextable[] = "0123456789abcdef";
 141
 142 std::string BinToHex(const std::string& data)
 143 {
 144         int l = data.length();
 145         std::string rv;
 146         rv.reserve(l * 2);
 147         for(int i=0; i < l; i++)
 148         {
 149                 unsigned char c = data[i];
 150                 rv.append(1, hextable[c >> 4]);
 151                 rv.append(1, hextable[c & 0xF]);
 152         }
 153         return rv;
 154 }
 155
 156 static const char b64table[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
 157
 158 std::string BinToBase64(const std::string& data_str, const char* table, char pad)
 159 {
 160         if (!table)
 161                 table = b64table;
 162
 163         uint32_t buffer;
 164         uint8_t* data = (uint8_t*)data_str.data();
 165         std::string rv;
 166         size_t i = 0;
 167         while (i + 2 < data_str.length())
 168         {
 169                 buffer = (data[i] << 16 | data[i+1] << 8 | data[i+2]);
 170                 rv.push_back(table[0x3F & (buffer >> 18)]);
 171                 rv.push_back(table[0x3F & (buffer >> 12)]);
 172                 rv.push_back(table[0x3F & (buffer >>  6)]);
 173                 rv.push_back(table[0x3F & (buffer >>  0)]);
 174                 i += 3;
 175         }
 176         if (data_str.length() == i)
 177         {
 178                 // no extra characters
 179         }
 180         else if (data_str.length() == i + 1)
 181         {
 182                 buffer = data[i] << 16;
 183                 rv.push_back(table[0x3F & (buffer >> 18)]);
 184                 rv.push_back(table[0x3F & (buffer >> 12)]);
 185                 if (pad)
 186                 {
 187                         rv.push_back(pad);
 188                         rv.push_back(pad);
 189                 }
 190         }
 191         else if (data_str.length() == i + 2)
 192         {
 193                 buffer = (data[i] << 16 | data[i] << 8);
 194                 rv.push_back(table[0x3F & (buffer >> 18)]);
 195                 rv.push_back(table[0x3F & (buffer >> 12)]);
 196                 rv.push_back(table[0x3F & (buffer >>  6)]);
 197                 if (pad)
 198                         rv.push_back(pad);
 199         }
 200         return rv;
 201 }
 202
 203 std::string Base64ToBin(const std::string& data_str, const char* table)
 204 {
 205         if (!table)
 206                 table = b64table;
 207
 208         bool ok = true;
 209         int bitcount = 0;
 210         uint32_t buffer;
 211         const char* data = data_str.c_str();
 212         std::string rv;
 213         while (ok)
 214         {
 215                 const char* find = strchr(table, *data);
 216                 ok = (find && find < table + 64);
 217                 buffer = (buffer << 6) | (ok ? find - table : 0);
 218                 bitcount += 6;
 219                 if (bitcount >= 8)
 220                 {
 221                         bitcount -= 8;
 222                         rv.push_back((buffer >> bitcount) & 0xFF);
 223                 }
 224         }
 225         return rv;
 226 }