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