| // Copyright 2023 The Dawn & Tint Authors |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are met: |
| // |
| // 1. Redistributions of source code must retain the above copyright notice, this |
| // list of conditions and the following disclaimer. |
| // |
| // 2. Redistributions in binary form must reproduce the above copyright notice, |
| // this list of conditions and the following disclaimer in the documentation |
| // and/or other materials provided with the distribution. |
| // |
| // 3. Neither the name of the copyright holder nor the names of its |
| // contributors may be used to endorse or promote products derived from |
| // this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE |
| // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR |
| // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
| // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
| // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| #include "src/tint/utils/text/base64.h" |
| |
| #include <optional> |
| #include <string> |
| #include <vector> |
| |
| #include "gtest/gtest.h" |
| #include "src/tint/utils/containers/transform.h" |
| #include "src/tint/utils/text/string.h" |
| |
| namespace tint::utils { |
| namespace { |
| |
| struct DecodeBase64Case { |
| char in; |
| std::optional<uint8_t> out; |
| }; |
| |
| using DecodeBase64Test = testing::TestWithParam<DecodeBase64Case>; |
| |
| TEST_P(DecodeBase64Test, Char) { |
| EXPECT_EQ(DecodeBase64(GetParam().in), GetParam().out); |
| } |
| |
| INSTANTIATE_TEST_SUITE_P(Valid, |
| DecodeBase64Test, |
| testing::Values(DecodeBase64Case{'A', 0}, |
| DecodeBase64Case{'B', 1}, |
| DecodeBase64Case{'C', 2}, |
| DecodeBase64Case{'D', 3}, |
| DecodeBase64Case{'E', 4}, |
| DecodeBase64Case{'F', 5}, |
| DecodeBase64Case{'G', 6}, |
| DecodeBase64Case{'H', 7}, |
| DecodeBase64Case{'I', 8}, |
| DecodeBase64Case{'J', 9}, |
| DecodeBase64Case{'K', 10}, |
| DecodeBase64Case{'L', 11}, |
| DecodeBase64Case{'M', 12}, |
| DecodeBase64Case{'N', 13}, |
| DecodeBase64Case{'O', 14}, |
| DecodeBase64Case{'P', 15}, |
| DecodeBase64Case{'Q', 16}, |
| DecodeBase64Case{'R', 17}, |
| DecodeBase64Case{'S', 18}, |
| DecodeBase64Case{'T', 19}, |
| DecodeBase64Case{'U', 20}, |
| DecodeBase64Case{'V', 21}, |
| DecodeBase64Case{'W', 22}, |
| DecodeBase64Case{'X', 23}, |
| DecodeBase64Case{'Y', 24}, |
| DecodeBase64Case{'Z', 25}, |
| DecodeBase64Case{'a', 26}, |
| DecodeBase64Case{'b', 27}, |
| DecodeBase64Case{'c', 28}, |
| DecodeBase64Case{'d', 29}, |
| DecodeBase64Case{'e', 30}, |
| DecodeBase64Case{'f', 31}, |
| DecodeBase64Case{'g', 32}, |
| DecodeBase64Case{'h', 33}, |
| DecodeBase64Case{'i', 34}, |
| DecodeBase64Case{'j', 35}, |
| DecodeBase64Case{'k', 36}, |
| DecodeBase64Case{'l', 37}, |
| DecodeBase64Case{'m', 38}, |
| DecodeBase64Case{'n', 39}, |
| DecodeBase64Case{'o', 40}, |
| DecodeBase64Case{'p', 41}, |
| DecodeBase64Case{'q', 42}, |
| DecodeBase64Case{'r', 43}, |
| DecodeBase64Case{'s', 44}, |
| DecodeBase64Case{'t', 45}, |
| DecodeBase64Case{'u', 46}, |
| DecodeBase64Case{'v', 47}, |
| DecodeBase64Case{'w', 48}, |
| DecodeBase64Case{'x', 49}, |
| DecodeBase64Case{'y', 50}, |
| DecodeBase64Case{'z', 51}, |
| DecodeBase64Case{'0', 52}, |
| DecodeBase64Case{'1', 53}, |
| DecodeBase64Case{'2', 54}, |
| DecodeBase64Case{'3', 55}, |
| DecodeBase64Case{'4', 56}, |
| DecodeBase64Case{'5', 57}, |
| DecodeBase64Case{'6', 58}, |
| DecodeBase64Case{'7', 59}, |
| DecodeBase64Case{'8', 60}, |
| DecodeBase64Case{'9', 61}, |
| DecodeBase64Case{'+', 62}, |
| DecodeBase64Case{'/', 63})); |
| |
| INSTANTIATE_TEST_SUITE_P(Invalid, |
| DecodeBase64Test, |
| testing::Values(DecodeBase64Case{'@', std::nullopt}, |
| DecodeBase64Case{'#', std::nullopt}, |
| DecodeBase64Case{'^', std::nullopt}, |
| DecodeBase64Case{'&', std::nullopt}, |
| DecodeBase64Case{'!', std::nullopt}, |
| DecodeBase64Case{'*', std::nullopt}, |
| DecodeBase64Case{'(', std::nullopt}, |
| DecodeBase64Case{')', std::nullopt}, |
| DecodeBase64Case{'-', std::nullopt}, |
| DecodeBase64Case{'.', std::nullopt}, |
| DecodeBase64Case{'\0', std::nullopt}, |
| DecodeBase64Case{'\n', std::nullopt})); |
| |
| INSTANTIATE_TEST_SUITE_P(Padding, |
| DecodeBase64Test, |
| testing::Values(DecodeBase64Case{'=', std::nullopt})); |
| |
| struct DecodeBase64FromCommentsCase { |
| std::string_view wgsl; |
| Vector<int, 0> expected; |
| }; |
| |
| using DecodeBase64FromCommentsTest = ::testing::TestWithParam<DecodeBase64FromCommentsCase>; |
| |
| TEST_P(DecodeBase64FromCommentsTest, None) { |
| auto got_bytes = DecodeBase64FromComments(GetParam().wgsl); |
| auto got = Transform(got_bytes, [](std::byte byte) { return static_cast<int>(byte); }); |
| EXPECT_EQ(got, GetParam().expected); |
| } |
| |
| INSTANTIATE_TEST_SUITE_P(, |
| DecodeBase64FromCommentsTest, |
| testing::ValuesIn(std::vector<DecodeBase64FromCommentsCase>{ |
| {"", Empty}, |
| {"//", Empty}, |
| {"abc", Empty}, |
| {"abc//", Empty}, |
| { |
| R"(a |
| b |
| c)", |
| Empty, |
| }, |
| {"// abc", {26, 27, 28}}, |
| {"a // bc", {27, 28}}, |
| {"ab // c", {28}}, |
| {"// a.b.c", {26, 27, 28}}, |
| { |
| R"(a |
| b |
| c)", |
| Empty, |
| }, |
| { |
| R"(a |
| // b |
| c)", |
| {27}, |
| }, |
| { |
| R"(// a |
| // b |
| // c)", |
| {26, 27, 28}, |
| }, |
| { |
| R"(/* a |
| b |
| c |
| */)", |
| {26, 27, 28}, |
| }, |
| { |
| R"(/* a |
| b |
| */ |
| c)", |
| {26, 27}, |
| }, |
| { |
| R"(a/* |
| b |
| */ |
| c)", |
| {27}, |
| }, |
| { |
| "x/*a*/b/*c*/y", |
| {26, 28}, |
| }, |
| { |
| "x/*a/*b*/c*/z", |
| {26, 27, 28}, |
| }, |
| })); |
| |
| struct EncodeBase64Case { |
| std::vector<uint8_t> in; |
| std::optional<std::string> expected; |
| }; |
| |
| using EncodeBase64Test = testing::TestWithParam<EncodeBase64Case>; |
| |
| TEST_P(EncodeBase64Test, Bytes) { |
| tint::Vector<std::byte, 0> bytes; |
| for (auto b : GetParam().in) { |
| bytes.Push(std::byte{b}); |
| } |
| auto got = EncodeBase64(bytes.AsSpan()); |
| if (GetParam().expected) { |
| ASSERT_EQ(got, Success); |
| EXPECT_EQ(got.Get(), *GetParam().expected); |
| } else { |
| EXPECT_NE(got, Success); |
| } |
| } |
| |
| INSTANTIATE_TEST_SUITE_P(, |
| EncodeBase64Test, |
| testing::Values(EncodeBase64Case{{}, std::string{""}}, |
| EncodeBase64Case{{0}, std::string{"A"}}, |
| EncodeBase64Case{{1}, std::string{"B"}}, |
| EncodeBase64Case{{1, 2, 3}, std::string{"BCD"}}, |
| EncodeBase64Case{{62, 63}, std::string{"+/"}}, |
| EncodeBase64Case{{64}, std::nullopt}, |
| EncodeBase64Case{{255}, std::nullopt})); |
| |
| struct EncodeBase64SingleCase { |
| uint8_t in; |
| char expected; |
| }; |
| |
| using EncodeBase64SingleTest = testing::TestWithParam<EncodeBase64SingleCase>; |
| |
| TEST_P(EncodeBase64SingleTest, Char) { |
| std::byte b{GetParam().in}; |
| auto got = EncodeBase64(std::span{&b, 1}); |
| ASSERT_EQ(got, Success); |
| EXPECT_EQ(got.Get(), std::string(1, GetParam().expected)); |
| } |
| |
| INSTANTIATE_TEST_SUITE_P(Valid, |
| EncodeBase64SingleTest, |
| testing::Values(EncodeBase64SingleCase{0, 'A'}, |
| EncodeBase64SingleCase{1, 'B'}, |
| EncodeBase64SingleCase{2, 'C'}, |
| EncodeBase64SingleCase{3, 'D'}, |
| EncodeBase64SingleCase{4, 'E'}, |
| EncodeBase64SingleCase{5, 'F'}, |
| EncodeBase64SingleCase{6, 'G'}, |
| EncodeBase64SingleCase{7, 'H'}, |
| EncodeBase64SingleCase{8, 'I'}, |
| EncodeBase64SingleCase{9, 'J'}, |
| EncodeBase64SingleCase{10, 'K'}, |
| EncodeBase64SingleCase{11, 'L'}, |
| EncodeBase64SingleCase{12, 'M'}, |
| EncodeBase64SingleCase{13, 'N'}, |
| EncodeBase64SingleCase{14, 'O'}, |
| EncodeBase64SingleCase{15, 'P'}, |
| EncodeBase64SingleCase{16, 'Q'}, |
| EncodeBase64SingleCase{17, 'R'}, |
| EncodeBase64SingleCase{18, 'S'}, |
| EncodeBase64SingleCase{19, 'T'}, |
| EncodeBase64SingleCase{20, 'U'}, |
| EncodeBase64SingleCase{21, 'V'}, |
| EncodeBase64SingleCase{22, 'W'}, |
| EncodeBase64SingleCase{23, 'X'}, |
| EncodeBase64SingleCase{24, 'Y'}, |
| EncodeBase64SingleCase{25, 'Z'}, |
| EncodeBase64SingleCase{26, 'a'}, |
| EncodeBase64SingleCase{27, 'b'}, |
| EncodeBase64SingleCase{28, 'c'}, |
| EncodeBase64SingleCase{29, 'd'}, |
| EncodeBase64SingleCase{30, 'e'}, |
| EncodeBase64SingleCase{31, 'f'}, |
| EncodeBase64SingleCase{32, 'g'}, |
| EncodeBase64SingleCase{33, 'h'}, |
| EncodeBase64SingleCase{34, 'i'}, |
| EncodeBase64SingleCase{35, 'j'}, |
| EncodeBase64SingleCase{36, 'k'}, |
| EncodeBase64SingleCase{37, 'l'}, |
| EncodeBase64SingleCase{38, 'm'}, |
| EncodeBase64SingleCase{39, 'n'}, |
| EncodeBase64SingleCase{40, 'o'}, |
| EncodeBase64SingleCase{41, 'p'}, |
| EncodeBase64SingleCase{42, 'q'}, |
| EncodeBase64SingleCase{43, 'r'}, |
| EncodeBase64SingleCase{44, 's'}, |
| EncodeBase64SingleCase{45, 't'}, |
| EncodeBase64SingleCase{46, 'u'}, |
| EncodeBase64SingleCase{47, 'v'}, |
| EncodeBase64SingleCase{48, 'w'}, |
| EncodeBase64SingleCase{49, 'x'}, |
| EncodeBase64SingleCase{50, 'y'}, |
| EncodeBase64SingleCase{51, 'z'}, |
| EncodeBase64SingleCase{52, '0'}, |
| EncodeBase64SingleCase{53, '1'}, |
| EncodeBase64SingleCase{54, '2'}, |
| EncodeBase64SingleCase{55, '3'}, |
| EncodeBase64SingleCase{56, '4'}, |
| EncodeBase64SingleCase{57, '5'}, |
| EncodeBase64SingleCase{58, '6'}, |
| EncodeBase64SingleCase{59, '7'}, |
| EncodeBase64SingleCase{60, '8'}, |
| EncodeBase64SingleCase{61, '9'}, |
| EncodeBase64SingleCase{62, '+'}, |
| EncodeBase64SingleCase{63, '/'})); |
| |
| TEST(Base64Test, RoundtripSingleComment) { |
| std::string input = "// AQID\n@compute fn main() {}"; |
| auto decoded = DecodeBase64FromComments(input); |
| auto reencoded = EncodeBase64(decoded.AsSpan()); |
| ASSERT_EQ(reencoded, Success); |
| std::string output = "// " + reencoded.Get() + "\n@compute fn main() {}"; |
| EXPECT_EQ(input, output); |
| } |
| |
| TEST(Base64Test, RoundtripMultiComment) { |
| std::string input = "// AQ\n// ID\n@compute fn main() {}"; |
| auto decoded_input = DecodeBase64FromComments(input); |
| |
| auto reencoded = EncodeBase64(decoded_input.AsSpan()); |
| ASSERT_EQ(reencoded, Success); |
| std::string output = "// " + reencoded.Get() + "\n@compute fn main() {}"; |
| |
| // Reconstructed string has a single comment, so it is different. |
| EXPECT_NE(input, output); |
| EXPECT_EQ(output, "// AQID\n@compute fn main() {}"); |
| |
| // But the binary produced by both should be the same. |
| auto decoded_output = DecodeBase64FromComments(output); |
| auto got_input = Transform(decoded_input, [](std::byte b) { return static_cast<int>(b); }); |
| auto got_output = Transform(decoded_output, [](std::byte b) { return static_cast<int>(b); }); |
| EXPECT_EQ(got_input, got_output); |
| } |
| |
| } // namespace |
| } // namespace tint::utils |
