Fix mistakenly using Clang instead of GCC on older FreeBSD versions.

[user/henk/code/inspircd.git] / src / inspstring.cpp

diff --git a/src/inspstring.cpp b/src/inspstring.cpp
index 4f4cbd8ad226ad4200ec1d427122cf9ffe5b9af3..534b7ce008558a32ac466eaa786ea1f6c1194560 100644 (file)
--- a/src/inspstring.cpp
+++ b/src/inspstring.cpp
@@ -1,17 +1,23 @@
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ *   Copyright (C) 2010 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ *   Copyright (C) 2005-2007 Craig Edwards <craigedwards@brainbox.cc>
  *
- *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
- * See: http://wiki.inspircd.org/Credits
+ * This file is part of InspIRCd.  InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
  *
- * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+ * details.
  *
- * ---------------------------------------------------
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-/* $Core */
 
 #include "inspircd.h"
 
@@ -152,3 +158,75 @@ std::string BinToHex(const std::string& data)
        }
        return rv;
 }
+
+static const char b64table[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+
+std::string BinToBase64(const std::string& data_str, const char* table, char pad)
+{
+       if (!table)
+               table = b64table;
+
+       uint32_t buffer;
+       uint8_t* data = (uint8_t*)data_str.data();
+       std::string rv;
+       size_t i = 0;
+       while (i + 2 < data_str.length())
+       {
+               buffer = (data[i] << 16 | data[i+1] << 8 | data[i+2]);
+               rv.push_back(table[0x3F & (buffer >> 18)]);
+               rv.push_back(table[0x3F & (buffer >> 12)]);
+               rv.push_back(table[0x3F & (buffer >>  6)]);
+               rv.push_back(table[0x3F & (buffer >>  0)]);
+               i += 3;
+       }
+       if (data_str.length() == i)
+       {
+               // no extra characters
+       }
+       else if (data_str.length() == i + 1)
+       {
+               buffer = data[i] << 16;
+               rv.push_back(table[0x3F & (buffer >> 18)]);
+               rv.push_back(table[0x3F & (buffer >> 12)]);
+               if (pad)
+               {
+                       rv.push_back(pad);
+                       rv.push_back(pad);
+               }
+       }
+       else if (data_str.length() == i + 2)
+       {
+               buffer = (data[i] << 16 | data[i+1] << 8);
+               rv.push_back(table[0x3F & (buffer >> 18)]);
+               rv.push_back(table[0x3F & (buffer >> 12)]);
+               rv.push_back(table[0x3F & (buffer >>  6)]);
+               if (pad)
+                       rv.push_back(pad);
+       }
+       return rv;
+}
+
+std::string Base64ToBin(const std::string& data_str, const char* table)
+{
+       if (!table)
+               table = b64table;
+
+       int bitcount = 0;
+       uint32_t buffer = 0;
+       const char* data = data_str.c_str();
+       std::string rv;
+       while (true)
+       {
+               const char* find = strchr(table, *data++);
+               if (!find || find >= table + 64)
+                       break;
+               buffer = (buffer << 6) | (find - table);
+               bitcount += 6;
+               if (bitcount >= 8)
+               {
+                       bitcount -= 8;
+                       rv.push_back((buffer >> bitcount) & 0xFF);
+               }
+       }
+       return rv;
+}