BLC 사용 예제 - BASE64 인코딩/디코딩

#include "blc/console/blc_conio.h"
#include "blc/bit/blc_bit.h"
#include "blc/bimap/blc_bidirectional_map.h"
#include <iostream>
#include <string>
#include <vector>
#include <cstdint>
// BASE64 변환 테이블 타입
using BASE64TableT = blc::bimap::BidirectionalMap<
uint8_t, uint8_t, blc::bimap::ContainerUnorderedMap
>;
// BASE64 변환 테이블
const BASE64TableT kBASE64Table = {
{ 0, 'A' },
{ 1, 'B' },
{ 2, 'C' },
{ 3, 'D' },
{ 4, 'E' },
{ 5, 'F' },
{ 6, 'G' },
{ 7, 'H' },
{ 8, 'I' },
{ 9, 'J' },
{ 10, 'K' },
{ 11, 'L' },
{ 12, 'M' },
{ 13, 'N' },
{ 14, 'O' },
{ 15, 'P' },
{ 16, 'Q' },
{ 17, 'R' },
{ 18, 'S' },
{ 19, 'T' },
{ 20, 'U' },
{ 21, 'V' },
{ 22, 'W' },
{ 23, 'X' },
{ 24, 'Y' },
{ 25, 'Z' },
{ 26, 'a' },
{ 27, 'b' },
{ 28, 'c' },
{ 29, 'd' },
{ 30, 'e' },
{ 31, 'f' },
{ 32, 'g' },
{ 33, 'h' },
{ 34, 'i' },
{ 35, 'j' },
{ 36, 'k' },
{ 37, 'l' },
{ 38, 'm' },
{ 39, 'n' },
{ 40, 'o' },
{ 41, 'p' },
{ 42, 'q' },
{ 43, 'r' },
{ 44, 's' },
{ 45, 't' },
{ 46, 'u' },
{ 47, 'v' },
{ 48, 'w' },
{ 49, 'x' },
{ 50, 'y' },
{ 51, 'z' },
{ 52, '0' },
{ 53, '1' },
{ 54, '2' },
{ 55, '3' },
{ 56, '4' },
{ 57, '5' },
{ 58, '6' },
{ 59, '7' },
{ 60, '8' },
{ 61, '9' },
{ 62, '+' },
{ 63, '/' },
};
// 포인터 data가 가리키는 위치부터
// size 바이트 길이의 데이터를BASE64로 인코딩
std::string BASE64Encode(const void* data, size_t size)
{
using namespace std;
using namespace blc::bit;
string result;
ConstBitPtr src(data);
uint8_t buffer = 0;
const BitPtr dst(&buffer);
result.reserve(size * 4 / 3 + 1); // 메모리 재할당 최소화
for (; size >= 1; src += 24) { // data를 24비트씩 나눠서 처리
{ // 남은 길이가 1 바이트 이상일 때의 공통 처리 블록
BitMove(dst, src + 2, 6);
result += *kBASE64Table.FirstToSecond(buffer);
BitMove(dst + 4, src, 2);
}
if (size >= 2) {
{ // 남은 길이가 2 바이트 이상일 때의 공통 처리 블록
BitMove(dst, src + 12, 4);
result += *kBASE64Table.FirstToSecond(buffer);
BitMove(dst + 2, src + 8, 4);
}
if (size >= 3) { // 남은 길이가 3 바이트 이상인 경우 처리
BitMove(dst, src + 22, 2);
result += *kBASE64Table.FirstToSecond(buffer);
BitMove(dst, src + 16, 6);
result += *kBASE64Table.FirstToSecond(buffer);
size -= 3;
}
else { // 남은 길이가 2 바이트인 경우 처리
BitSet(dst, false, 2);
result += *kBASE64Table.FirstToSecond(buffer);
result += '='; // 패딩 삽입
size -= 2;
}
}
else { // 남은 길이가 1 바이트인 경우 처리
BitSet(dst, false, 4);
result += *kBASE64Table.FirstToSecond(buffer);
result += "=="; // 패딩 삽입
--size;
}
}
return result;
}
// BASE64로 인코딩된 BASE64 문자열을 디코딩해서
// 그 결과를 벡터에 담아 반환
std::vector<uint8_t> BASE64Decode(const std::string& BASE64)
{
using namespace std;
using namespace blc::bit;
size_t size = BASE64.size();
if ((size & 3) || !size) return {}; // BASE64로 인코딩된 데이터의 길이(패딩 포함)는
// 4의 배수인지, BASE64가 빈 문자열이 아닌지 점검
vector<uint8_t> result(size * 3 / 4, 0); // 필요한 메모리 확보
BitPtr dst(&result[0]);
for (size_t i = 0; i < size; dst += 24) { // 24비트씩 나눠서 처리
// 남은 길이(패딩 제외)가 2 바이트 이상일 때의 공통 처리------
ConstBitPtr src(kBASE64Table.SecondToFirst(BASE64[i++]));
if (!src) return {};
BitMove(dst + 2, src, 6);
src.set_const_base_ptr(kBASE64Table.SecondToFirst(BASE64[i++]));
if (!src) return {};
BitMove(dst, src + 4, 2);
//-------------------------------------------------
if (BASE64[i] != '=') {
// 남은 길이(패딩 제외)가 3 바이트 이상일 때의 공통 처리------
BitMove(dst + 12, src, 4);
src.set_const_base_ptr(kBASE64Table.SecondToFirst(BASE64[i++]));
if (!src) return {};
BitMove(dst + 8, src + 2, 4);
//-------------------------------------------------
if (BASE64[i] != '=') { // 남은 길이(패딩 제외)가
// 4 바이트 이상인 경우 처리
BitMove(dst + 22, src, 2);
src.set_const_base_ptr(kBASE64Table.SecondToFirst(BASE64[i++]));
if (!src) return {};
BitMove(dst + 16, src, 6);
}
else { // 남은 길이(패딩 제외)가
// 3 바이트인 경우 처리
++i;
result.resize(result.size() - 1);
}
}
else { // 남은 길이(패딩 제외)가
// 2 바이트인 경우 처리
i += 2;
result.resize(result.size() - 2);
}
}
return result;
}
int main()
{
using namespace blc::console;
using namespace blc::ustr;
using namespace std;
SetCharacterWidthFunc(mk_wcwidth_cjk); // 한중일 문자 및 특수문자
// 사용시 문자폭 계산함수 설정
ConsoleWindow::SetTitle(L"BASE64 인코딩/디코딩 예제"s);
auto str = u8""s;
ConsoleWindow::Space space(0, 0, 60, 1); // (0, 0)좌표에서 폭 60, 높이 1의 공간
cout << u8"입력한 문자열을 BASE64로 인코딩/디코딩 합니다."s << endl << endl;
space.set_y(ConsoleWindow::WhereY()); // space의 y좌표 갱신
ConsoleWindow::ModifyStr(str, space); // 문자열 입력 용도로 사용
cout << endl << endl;
auto base64 = BASE64Encode(str.c_str(), str.size()); // BASE64로 인코딩
cout << u8">>BASE64 인코딩 결과:"s << endl;
cout << base64 << endl << endl;
auto buffer = BASE64Decode(base64); // BASE64를 디코딩
cout << u8">>BASE64 디코딩 결과:"s << endl;
cout << string(buffer.cbegin(), buffer.cend()) << endl << endl;
return 0;
}

예제 코드 접기