blob: cb73da6d69b7683063cb2075751a28d8a5ba0528 [file] [log] [blame]
// Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <brillo/data_encoding.h>
#include <algorithm>
#include <numeric>
#include <gtest/gtest.h>
namespace brillo {
namespace data_encoding {
TEST(data_encoding, UrlEncoding) {
std::string test = "\"http://sample/path/0014.html \"";
std::string encoded = UrlEncode(test.c_str());
EXPECT_EQ("%22http%3A%2F%2Fsample%2Fpath%2F0014.html+%22", encoded);
EXPECT_EQ(test, UrlDecode(encoded.c_str()));
test = "\"http://sample/path/0014.html \"";
encoded = UrlEncode(test.c_str(), false);
EXPECT_EQ("%22http%3A%2F%2Fsample%2Fpath%2F0014.html%20%22", encoded);
EXPECT_EQ(test, UrlDecode(encoded.c_str()));
}
TEST(data_encoding, WebParamsEncoding) {
std::string encoded =
WebParamsEncode({{"q", "test"}, {"path", "/usr/bin"}, {"#", "%"}});
EXPECT_EQ("q=test&path=%2Fusr%2Fbin&%23=%25", encoded);
auto params = WebParamsDecode(encoded);
EXPECT_EQ(3, params.size());
EXPECT_EQ("q", params[0].first);
EXPECT_EQ("test", params[0].second);
EXPECT_EQ("path", params[1].first);
EXPECT_EQ("/usr/bin", params[1].second);
EXPECT_EQ("#", params[2].first);
EXPECT_EQ("%", params[2].second);
}
TEST(data_encoding, Base64Encode) {
const std::string text1 = "hello world";
const std::string encoded1 = "aGVsbG8gd29ybGQ=";
const std::string text2 =
"Lorem ipsum dolor sit amet, facilisis erat nec aliquam, scelerisque "
"molestie commodo. Viverra tincidunt integer erat ipsum, integer "
"molestie, arcu in, sit mauris ac a sed sit etiam.";
const std::string encoded2 =
"TG9yZW0gaXBzdW0gZG9sb3Igc2l0IGFtZXQsIGZhY2lsaXNpcyBlcmF0IG5lYyBhbGlxdWF"
"tLCBzY2VsZXJpc3F1ZSBtb2xlc3RpZSBjb21tb2RvLiBWaXZlcnJhIHRpbmNpZHVudCBpbn"
"RlZ2VyIGVyYXQgaXBzdW0sIGludGVnZXIgbW9sZXN0aWUsIGFyY3UgaW4sIHNpdCBtYXVya"
"XMgYWMgYSBzZWQgc2l0IGV0aWFtLg==";
brillo::Blob data3(256);
std::iota(data3.begin(), data3.end(), 0); // Fills the buffer with 0x00-0xFF.
const std::string encoded3 =
"AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUmJygpKissLS4vMDEyMzQ"
"1Njc4OTo7PD0+P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5fYGFiY2RlZmdoaW"
"prbG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6PkJGSk5SVlpeYmZqbnJ2en"
"6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/wMHCw8TFxsfIycrLzM3Oz9DR0tPU"
"1dbX2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v8PHy8/T19vf4+fr7/P3+/w==";
EXPECT_EQ(encoded1, Base64Encode(text1));
EXPECT_EQ(encoded2, Base64Encode(text2));
EXPECT_EQ(encoded3, Base64Encode(data3));
}
TEST(data_encoding, Base64EncodeWrapLines) {
const std::string text1 = "hello world";
const std::string encoded1 = "aGVsbG8gd29ybGQ=\n";
const std::string text2 =
"Lorem ipsum dolor sit amet, facilisis erat nec aliquam, scelerisque "
"molestie commodo. Viverra tincidunt integer erat ipsum, integer "
"molestie, arcu in, sit mauris ac a sed sit etiam.";
const std::string encoded2 =
"TG9yZW0gaXBzdW0gZG9sb3Igc2l0IGFtZXQsIGZhY2lsaXNpcyBlcmF0IG5lYyBh\n"
"bGlxdWFtLCBzY2VsZXJpc3F1ZSBtb2xlc3RpZSBjb21tb2RvLiBWaXZlcnJhIHRp\n"
"bmNpZHVudCBpbnRlZ2VyIGVyYXQgaXBzdW0sIGludGVnZXIgbW9sZXN0aWUsIGFy\n"
"Y3UgaW4sIHNpdCBtYXVyaXMgYWMgYSBzZWQgc2l0IGV0aWFtLg==\n";
brillo::Blob data3(256);
std::iota(data3.begin(), data3.end(), 0); // Fills the buffer with 0x00-0xFF.
const std::string encoded3 =
"AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUmJygpKissLS4v\n"
"MDEyMzQ1Njc4OTo7PD0+P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5f\n"
"YGFiY2RlZmdoaWprbG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6P\n"
"kJGSk5SVlpeYmZqbnJ2en6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/\n"
"wMHCw8TFxsfIycrLzM3Oz9DR0tPU1dbX2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v\n"
"8PHy8/T19vf4+fr7/P3+/w==\n";
EXPECT_EQ(encoded1, Base64EncodeWrapLines(text1));
EXPECT_EQ(encoded2, Base64EncodeWrapLines(text2));
EXPECT_EQ(encoded3, Base64EncodeWrapLines(data3));
}
TEST(data_encoding, Base64Decode) {
const std::string encoded1 = "dGVzdCBzdHJpbmc=";
const std::string encoded2 =
"TG9yZW0gaXBzdW0gZG9sb3Igc2l0IGFtZXQsIGZhY2lsaXNpcyBlcmF0IG5lYyBh\n"
"bGlxdWFtLCBzY2VsZXJpc3F1ZSBtb2xlc3RpZSBjb21tb2RvLiBWaXZlcnJhIHRp\r\n"
"bmNpZHVudCBpbnRlZ2VyIGVyYXQgaXBzdW0sIGludGVnZXIgbW9sZXN0aWUsIGFy\r"
"Y3UgaW4sIHNpdCBtYXVyaXMgYWMgYSBzZWQgc2l0IGV0aWFt\n"
"Lg==\n\n\n";
const std::string decoded2 =
"Lorem ipsum dolor sit amet, facilisis erat nec aliquam, scelerisque "
"molestie commodo. Viverra tincidunt integer erat ipsum, integer "
"molestie, arcu in, sit mauris ac a sed sit etiam.";
const std::string encoded3 =
"AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUmJygpKissLS4vMDEyMzQ"
"1Njc4OTo7PD0+P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5fYGFiY2RlZmdoaW"
"prbG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6PkJGSk5SVlpeYmZqbnJ2en"
"6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/wMHCw8TFxsfIycrLzM3Oz9DR0tPU"
"1dbX2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v8PHy8/T19vf4+fr7/P3+/w==";
brillo::Blob decoded3(256);
std::iota(decoded3.begin(), decoded3.end(), 0); // Fill with 0x00..0xFF.
std::string decoded;
EXPECT_TRUE(Base64Decode(encoded1, &decoded));
EXPECT_EQ("test string", decoded);
EXPECT_TRUE(Base64Decode(encoded2, &decoded));
EXPECT_EQ(decoded2, decoded);
brillo::Blob decoded_blob;
EXPECT_TRUE(Base64Decode(encoded3, &decoded_blob));
EXPECT_EQ(decoded3, decoded_blob);
EXPECT_FALSE(Base64Decode("A", &decoded_blob));
EXPECT_TRUE(decoded_blob.empty());
EXPECT_TRUE(Base64Decode("/w==", &decoded_blob));
EXPECT_EQ((brillo::Blob{0xFF}), decoded_blob);
EXPECT_TRUE(Base64Decode("//8=", &decoded_blob));
EXPECT_EQ((brillo::Blob{0xFF, 0xFF}), decoded_blob);
EXPECT_FALSE(Base64Decode("AAECAwQFB,cI", &decoded_blob));
EXPECT_TRUE(decoded_blob.empty());
}
} // namespace data_encoding
} // namespace brillo