Add base64 encode and decode.
Chunting Gu
6 years ago
parent
ee56fe9d1b
commit
18db152419
@ -0,0 +1,29 @@
|
||||
#include "gtest/gtest.h"
|
||||
|
||||
#include "webcc/base64.h"
|
||||
|
||||
TEST(Base64, Encode) {
|
||||
std::string encoded;
|
||||
|
||||
encoded = webcc::Base64Encode("ban");
|
||||
EXPECT_EQ("YmFu", encoded);
|
||||
|
||||
encoded = webcc::Base64Encode("bana");
|
||||
EXPECT_EQ("YmFuYQ==", encoded);
|
||||
|
||||
encoded = webcc::Base64Encode("banan");
|
||||
EXPECT_EQ("YmFuYW4=", encoded);
|
||||
}
|
||||
|
||||
TEST(Base64, Decode) {
|
||||
std::string decoded;
|
||||
|
||||
decoded = webcc::Base64Decode("YmFu");
|
||||
EXPECT_EQ("ban", decoded);
|
||||
|
||||
decoded = webcc::Base64Decode("YmFuYQ==");
|
||||
EXPECT_EQ("bana", decoded);
|
||||
|
||||
decoded = webcc::Base64Decode("YmFuYW4=");
|
||||
EXPECT_EQ("banan", decoded);
|
||||
}
|
||||
@ -0,0 +1,183 @@
|
||||
#include "webcc/base64.h"
|
||||
|
||||
// Modified from Boost.Beast (boost\beast\core\detail\base64.hpp)
|
||||
// See the comments below for the original authors and copyrights.
|
||||
|
||||
//
|
||||
// Copyright (c) 2016-2017 Vinnie Falco (vinnie dot falco at gmail dot com)
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
||||
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// Official repository: https://github.com/boostorg/beast
|
||||
//
|
||||
|
||||
/*
|
||||
Portions from http://www.adp-gmbh.ch/cpp/common/base64.html
|
||||
Copyright notice:
|
||||
|
||||
base64.cpp and base64.h
|
||||
|
||||
Copyright (C) 2004-2008 Rene Nyffenegger
|
||||
|
||||
This source code is provided 'as-is', without any express or implied
|
||||
warranty. In no event will the author be held liable for any damages
|
||||
arising from the use of this software.
|
||||
|
||||
Permission is granted to anyone to use this software for any purpose,
|
||||
including commercial applications, and to alter it and redistribute it
|
||||
freely, subject to the following restrictions:
|
||||
|
||||
1. The origin of this source code must not be misrepresented; you must not
|
||||
claim that you wrote the original source code. If you use this source code
|
||||
in a product, an acknowledgment in the product documentation would be
|
||||
appreciated but is not required.
|
||||
|
||||
2. Altered source versions must be plainly marked as such, and must not be
|
||||
misrepresented as being the original source code.
|
||||
|
||||
3. This notice may not be removed or altered from any source distribution.
|
||||
|
||||
Rene Nyffenegger rene.nyffenegger@adp-gmbh.ch
|
||||
*/
|
||||
|
||||
namespace webcc {
|
||||
|
||||
namespace base64 {
|
||||
|
||||
// The Base 64 Alphabet.
|
||||
static const char kAlphabet[] =
|
||||
"ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||||
"abcdefghijklmnopqrstuvwxyz"
|
||||
"0123456789+/";
|
||||
|
||||
// kInverse['z'] -> 51 which is the index of 'z' in kAlphabet.
|
||||
static const char kInverse[] = {
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 0-15
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 16-31
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, -1, -1, 63, // 32-47
|
||||
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1, // 48-63
|
||||
-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, // 64-79
|
||||
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1, // 80-95
|
||||
-1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, // 96-111
|
||||
41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -1, -1, -1, -1, -1, // 112-127
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 128-143
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 144-159
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 160-175
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 176-191
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 192-207
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 208-223
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, // 224-239
|
||||
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1 // 240-255
|
||||
};
|
||||
|
||||
// Return max chars needed to encode a base64 string.
|
||||
inline std::size_t constexpr EncodedSize(std::size_t n) {
|
||||
return 4 * ((n + 2) / 3);
|
||||
}
|
||||
|
||||
// Return max bytes needed to decode a base64 string.
|
||||
inline std::size_t constexpr DecodedSize(std::size_t n) {
|
||||
// return 3 * n / 4;
|
||||
return n / 4 * 3; // requires n&3==0, smaller
|
||||
}
|
||||
|
||||
std::size_t Encode(const void* src, std::size_t len, void* dst) {
|
||||
char* out = static_cast<char*>(dst);
|
||||
const char* in = static_cast<const char*>(src);
|
||||
|
||||
for (auto n = len / 3; n--;) {
|
||||
*out++ = kAlphabet[ (in[0] & 0xfc) >> 2];
|
||||
*out++ = kAlphabet[((in[0] & 0x03) << 4) + ((in[1] & 0xf0) >> 4)];
|
||||
*out++ = kAlphabet[((in[1] & 0x0f) << 2) + ((in[2] & 0xc0) >> 6)];
|
||||
*out++ = kAlphabet[ in[2] & 0x3f];
|
||||
in += 3;
|
||||
}
|
||||
|
||||
switch (len % 3) {
|
||||
case 2:
|
||||
*out++ = kAlphabet[ (in[0] & 0xfc) >> 2];
|
||||
*out++ = kAlphabet[((in[0] & 0x03) << 4) + ((in[1] & 0xf0) >> 4)];
|
||||
*out++ = kAlphabet[ (in[1] & 0x0f) << 2];
|
||||
*out++ = '=';
|
||||
break;
|
||||
|
||||
case 1:
|
||||
*out++ = kAlphabet[ (in[0] & 0xfc) >> 2];
|
||||
*out++ = kAlphabet[((in[0] & 0x03) << 4)];
|
||||
*out++ = '=';
|
||||
*out++ = '=';
|
||||
break;
|
||||
|
||||
case 0:
|
||||
break;
|
||||
}
|
||||
|
||||
return out - static_cast<char*>(dst);
|
||||
}
|
||||
|
||||
typedef std::pair<std::size_t, std::size_t> SizePair;
|
||||
|
||||
SizePair Decode(const char* src, std::size_t len, void* dst) {
|
||||
auto in = reinterpret_cast<const unsigned char*>(src);
|
||||
unsigned char* out = static_cast<unsigned char*>(dst);
|
||||
|
||||
char c4[4];
|
||||
unsigned char c3[3];
|
||||
int i = 0;
|
||||
|
||||
while (len-- && *in != '=') {
|
||||
char v = kInverse[*in];
|
||||
if(v == -1) { break; }
|
||||
++in;
|
||||
c4[i] = v;
|
||||
|
||||
if(++i == 4) {
|
||||
c3[0] = (c4[0] << 2) + ((c4[1] & 0x30) >> 4);
|
||||
c3[1] = ((c4[1] & 0xf) << 4) + ((c4[2] & 0x3c) >> 2);
|
||||
c3[2] = ((c4[2] & 0x3) << 6) + c4[3];
|
||||
|
||||
for (i = 0; i < 3; i++) {
|
||||
*out++ = c3[i];
|
||||
}
|
||||
|
||||
i = 0;
|
||||
}
|
||||
}
|
||||
|
||||
if (i > 0) {
|
||||
c3[0] = ( c4[0] << 2) + ((c4[1] & 0x30) >> 4);
|
||||
c3[1] = ((c4[1] & 0xf) << 4) + ((c4[2] & 0x3c) >> 2);
|
||||
c3[2] = ((c4[2] & 0x3) << 6) + c4[3];
|
||||
|
||||
for (int j = 0; j < i - 1; j++) {
|
||||
*out++ = c3[j];
|
||||
}
|
||||
}
|
||||
|
||||
return {out - static_cast<unsigned char*>(dst),
|
||||
in - reinterpret_cast<const unsigned char*>(src)};
|
||||
}
|
||||
|
||||
} // namespace base64
|
||||
|
||||
std::string Base64Encode(const std::uint8_t* data, std::size_t len) {
|
||||
std::string dst;
|
||||
dst.resize(base64::EncodedSize(len));
|
||||
dst.resize(base64::Encode(data, len, &dst[0]));
|
||||
return dst;
|
||||
}
|
||||
|
||||
std::string Base64Encode(const std::string& s) {
|
||||
return Base64Encode(reinterpret_cast<const std::uint8_t*>(s.data()), s.size());
|
||||
}
|
||||
|
||||
std::string Base64Decode(const std::string& data) {
|
||||
std::string dst;
|
||||
dst.resize(base64::DecodedSize(data.size()));
|
||||
auto result = base64::Decode(data.data(), data.size(), &dst[0]);
|
||||
dst.resize(result.first);
|
||||
return dst;
|
||||
}
|
||||
|
||||
} // namespace webcc
|
||||
@ -0,0 +1,17 @@
|
||||
#ifndef WEBCC_BASE64_H_
|
||||
#define WEBCC_BASE64_H_
|
||||
|
||||
#include <cctype>
|
||||
#include <string>
|
||||
|
||||
namespace webcc {
|
||||
|
||||
std::string Base64Encode(const std::uint8_t* data, std::size_t len);
|
||||
|
||||
std::string Base64Encode(const std::string& input);
|
||||
|
||||
std::string Base64Decode(const std::string& data);
|
||||
|
||||
} // namespace webcc
|
||||
|
||||
#endif // WEBCC_BASE64_H_
|
||||
Loading…
Reference in New Issue