| // 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 |