Branch data Line data Source code
# 1 : : // Copyright (c) 2011-2021 The Bitcoin Core developers
# 2 : : // Distributed under the MIT software license, see the accompanying
# 3 : : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
# 4 : :
# 5 : : #include <util/strencodings.h>
# 6 : :
# 7 : : #include <boost/test/unit_test.hpp>
# 8 : : #include <string>
# 9 : :
# 10 : : using namespace std::literals;
# 11 : :
# 12 : : BOOST_AUTO_TEST_SUITE(base64_tests)
# 13 : :
# 14 : : BOOST_AUTO_TEST_CASE(base64_testvectors)
# 15 : 2 : {
# 16 : 2 : static const std::string vstrIn[] = {"","f","fo","foo","foob","fooba","foobar"};
# 17 : 2 : static const std::string vstrOut[] = {"","Zg==","Zm8=","Zm9v","Zm9vYg==","Zm9vYmE=","Zm9vYmFy"};
# 18 [ + + ]: 16 : for (unsigned int i=0; i<std::size(vstrIn); i++)
# 19 : 14 : {
# 20 : 14 : std::string strEnc = EncodeBase64(vstrIn[i]);
# 21 : 14 : BOOST_CHECK_EQUAL(strEnc, vstrOut[i]);
# 22 : 14 : std::string strDec = DecodeBase64(strEnc);
# 23 : 14 : BOOST_CHECK_EQUAL(strDec, vstrIn[i]);
# 24 : 14 : }
# 25 : :
# 26 : 2 : {
# 27 : 2 : const std::vector<uint8_t> in_u{0xff, 0x01, 0xff};
# 28 : 2 : const std::vector<std::byte> in_b{std::byte{0xff}, std::byte{0x01}, std::byte{0xff}};
# 29 : 2 : const std::string in_s{"\xff\x01\xff"};
# 30 : 2 : const std::string out_exp{"/wH/"};
# 31 : 2 : BOOST_CHECK_EQUAL(EncodeBase64(in_u), out_exp);
# 32 : 2 : BOOST_CHECK_EQUAL(EncodeBase64(in_b), out_exp);
# 33 : 2 : BOOST_CHECK_EQUAL(EncodeBase64(in_s), out_exp);
# 34 : 2 : }
# 35 : :
# 36 : : // Decoding strings with embedded NUL characters should fail
# 37 : 2 : bool failure;
# 38 : 2 : (void)DecodeBase64("invalid\0"s, &failure);
# 39 : 2 : BOOST_CHECK(failure);
# 40 : 2 : (void)DecodeBase64("nQB/pZw="s, &failure);
# 41 : 2 : BOOST_CHECK(!failure);
# 42 : 2 : (void)DecodeBase64("nQB/pZw=\0invalid"s, &failure);
# 43 : 2 : BOOST_CHECK(failure);
# 44 : 2 : (void)DecodeBase64("nQB/pZw=invalid\0"s, &failure);
# 45 : 2 : BOOST_CHECK(failure);
# 46 : 2 : }
# 47 : :
# 48 : : BOOST_AUTO_TEST_SUITE_END()
|