gcontini
2020-03-15 ef7df8048d1067d3148287f27af9047743f1e2ae
src/library/base/base64.cpp
New file
@@ -0,0 +1,153 @@
#include <stdio.h>
#include <stdlib.h>
#include <algorithm>
#include "base64.h"
namespace license {
using namespace std;
const static char* b64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
// maps A=>0,B=>1..
const static unsigned char unb64[] = {
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 10
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 20
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 30
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 40
   0,  0,  0,  62, 0,  0,  0,  63, 52, 53,  // 50
   54, 55, 56, 57, 58, 59, 60, 61, 0,  0,  // 60
   0,  0,  0,  0,  0,  0,  1,  2,  3,  4,  // 70
   5,  6,  7,  8,  9,  10, 11, 12, 13, 14,  // 80
   15, 16, 17, 18, 19, 20, 21, 22, 23, 24,  // 90
   25, 0,  0,  0,  0,  0,  0,  26, 27, 28,  // 100
   29, 30, 31, 32, 33, 34, 35, 36, 37, 38,  // 110
   39, 40, 41, 42, 43, 44, 45, 46, 47, 48,  // 120
   49, 50, 51, 0,  0,  0,  0,  0,  0,  0,  // 130
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 140
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 150
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 160
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 170
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 180
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 190
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 200
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 210
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 220
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 230
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 240
   0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  // 250
   0,  0,  0,  0,  0,  0,
};  // This array has 255 elements
// review api
void add_CR_if_needed(string& encodeBuffer, int lineLenght) {
   if (lineLenght > 0 && ((encodeBuffer.size() + 1) % lineLenght) == 0) {
      encodeBuffer += '\n';
   }
}
string base64(const void* binaryData, size_t len, int lineLenght) {
   const unsigned char* bin = (const unsigned char*)binaryData;
   int rc = 0;  // result counter
   int byteNo;  // I need this after the loop
   int modulusLen = len % 3;
   int pad = ((modulusLen & 1) << 1) + ((modulusLen & 2) >> 1);  // 2 gives 1 and 1 gives 2, but 0 gives 0.
   const size_t flen = 4 * (len + pad) / 3;
   size_t totalLength = flen;
   if (lineLenght > 0) {
      totalLength += ((int)flen / lineLenght) + 3;
   }
   string encodeBuffer;
   encodeBuffer.reserve(totalLength);
   for (byteNo = 0; byteNo <= len - 3; byteNo += 3) {
      unsigned char BYTE0 = bin[byteNo];
      unsigned char BYTE1 = bin[byteNo + 1];
      unsigned char BYTE2 = bin[byteNo + 2];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[BYTE0 >> 2];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[((0x3 & BYTE0) << 4) + (BYTE1 >> 4)];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[((0x0f & BYTE1) << 2) + (BYTE2 >> 6)];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[0x3f & BYTE2];
   }
   if (pad == 2) {
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[bin[byteNo] >> 2];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[(0x3 & bin[byteNo]) << 4];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += '=';
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += '=';
   } else if (pad == 1) {
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[bin[byteNo] >> 2];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[((0x3 & bin[byteNo]) << 4) + (bin[byteNo + 1] >> 4)];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += b64[(0x0f & bin[byteNo + 1]) << 2];
      add_CR_if_needed(encodeBuffer, lineLenght);
      encodeBuffer += '=';
   }
   if (lineLenght && encodeBuffer[encodeBuffer.length() - 1] != '\n') {
      encodeBuffer += '\n';
   }
   return encodeBuffer;
}
std::vector<uint8_t> unbase64(const std::string& base64_data) {
   string tmp_str(base64_data);
   tmp_str.erase(std::remove(tmp_str.begin(), tmp_str.end(), '\n'), tmp_str.end());
   const unsigned char* safeAsciiPtr = (const unsigned char*)tmp_str.c_str();
   std::vector<uint8_t> bin;
   int cb = 0;
   int charNo;
   int pad = 0;
   size_t len = tmp_str.size();
   if (len < 2) {  // 2 accesses below would be OOB.
      // catch empty string, return NULL as result.
      puts("ERROR: You passed an invalid base64 string (too short). You get NULL back.");
      return bin;
   }
   if (safeAsciiPtr[len - 1] == '=') ++pad;
   if (safeAsciiPtr[len - 2] == '=') ++pad;
   size_t flen = 3 * len / 4 - pad;
   bin.reserve(flen);
   for (charNo = 0; charNo <= len - 4 - pad; charNo += 4) {
      int A = unb64[safeAsciiPtr[charNo]];
      int B = unb64[safeAsciiPtr[charNo + 1]];
      int C = unb64[safeAsciiPtr[charNo + 2]];
      int D = unb64[safeAsciiPtr[charNo + 3]];
      bin.push_back((A << 2) | (B >> 4));
      bin.push_back((B << 4) | (C >> 2));
      bin.push_back((C << 6) | (D));
   }
   if (pad == 1) {
      int A = unb64[safeAsciiPtr[charNo]];
      int B = unb64[safeAsciiPtr[charNo + 1]];
      int C = unb64[safeAsciiPtr[charNo + 2]];
      bin.push_back((A << 2) | (B >> 4));
      bin.push_back((B << 4) | (C >> 2));
   } else if (pad == 2) {
      int A = unb64[safeAsciiPtr[charNo]];
      int B = unb64[safeAsciiPtr[charNo + 1]];
      bin.push_back((A << 2) | (B >> 4));
   }
   return bin;
}
}  // namespace license