3rdparty/yaml-cpp/src/binary.cpp

   1 #include "yaml-cpp/binary.h"
   2
   3 namespace YAML {
   4 static const char encoding[] =
   5     "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
   6
   7 std::string EncodeBase64(const unsigned char *data, std::size_t size) {
   8   const char PAD = '=';
   9
  10   std::string ret;
  11   ret.resize(4 * size / 3 + 3);
  12   char *out = &ret[0];
  13
  14   std::size_t chunks = size / 3;
  15   std::size_t remainder = size % 3;
  16
  17   for (std::size_t i = 0; i < chunks; i++, data += 3) {
  18     *out++ = encoding[data[0] >> 2];
  19     *out++ = encoding[((data[0] & 0x3) << 4) | (data[1] >> 4)];
  20     *out++ = encoding[((data[1] & 0xf) << 2) | (data[2] >> 6)];
  21     *out++ = encoding[data[2] & 0x3f];
  22   }
  23
  24   switch (remainder) {
  25     case 0:
  26       break;
  27     case 1:
  28       *out++ = encoding[data[0] >> 2];
  29       *out++ = encoding[((data[0] & 0x3) << 4)];
  30       *out++ = PAD;
  31       *out++ = PAD;
  32       break;
  33     case 2:
  34       *out++ = encoding[data[0] >> 2];
  35       *out++ = encoding[((data[0] & 0x3) << 4) | (data[1] >> 4)];
  36       *out++ = encoding[((data[1] & 0xf) << 2)];
  37       *out++ = PAD;
  38       break;
  39   }
  40
  41   ret.resize(out - &ret[0]);
  42   return ret;
  43 }
  44
  45 static const unsigned char decoding[] = {
  46     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  47     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  48     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 62,  255,
  49     255, 255, 63,  52,  53,  54,  55,  56,  57,  58,  59,  60,  61,  255, 255,
  50     255, 0,   255, 255, 255, 0,   1,   2,   3,   4,   5,   6,   7,   8,   9,
  51     10,  11,  12,  13,  14,  15,  16,  17,  18,  19,  20,  21,  22,  23,  24,
  52     25,  255, 255, 255, 255, 255, 255, 26,  27,  28,  29,  30,  31,  32,  33,
  53     34,  35,  36,  37,  38,  39,  40,  41,  42,  43,  44,  45,  46,  47,  48,
  54     49,  50,  51,  255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  55     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  56     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  57     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  58     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  59     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  60     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  61     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  62     255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255,
  63     255,
  64 };
  65
  66 std::vector<unsigned char> DecodeBase64(const std::string &input) {
  67   typedef std::vector<unsigned char> ret_type;
  68   if (input.empty())
  69     return ret_type();
  70
  71   ret_type ret(3 * input.size() / 4 + 1);
  72   unsigned char *out = &ret[0];
  73
  74   unsigned value = 0;
  75   for (std::size_t i = 0; i < input.size(); i++) {
  76     unsigned char d = decoding[static_cast<unsigned>(input[i])];
  77     if (d == 255)
  78       return ret_type();
  79
  80     value = (value << 6) | d;
  81     if (i % 4 == 3) {
  82       *out++ = value >> 16;
  83       if (i > 0 && input[i - 1] != '=')
  84         *out++ = value >> 8;
  85       if (input[i] != '=')
  86         *out++ = value;
  87     }
  88   }
  89
  90   ret.resize(out - &ret[0]);
  91   return ret;
  92 }
  93 }