From ef7df8048d1067d3148287f27af9047743f1e2ae Mon Sep 17 00:00:00 2001 From: gcontini <1121667+gcontini@users.noreply.github.com> Date: 周日, 15 3月 2020 16:26:21 +0800 Subject: [PATCH] Merge branch 'feature/pc_identifiers' into develop issues #2 #3 #14 #49 --- src/library/base/base64.cpp | 153 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 153 insertions(+), 0 deletions(-) diff --git a/src/library/base/base64.cpp b/src/library/base/base64.cpp new file mode 100644 index 0000000..f508d4a --- /dev/null +++ b/src/library/base/base64.cpp @@ -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 -- Gitblit v1.9.1