diff options
| -rw-r--r-- | src/dedicated_room/yuzu_room.cpp | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/src/dedicated_room/yuzu_room.cpp b/src/dedicated_room/yuzu_room.cpp index 8d8ac1ed7..359891883 100644 --- a/src/dedicated_room/yuzu_room.cpp +++ b/src/dedicated_room/yuzu_room.cpp | |||
| @@ -76,8 +76,18 @@ static constexpr char BanListMagic[] = "YuzuRoom-BanList-1"; | |||
| 76 | static constexpr char token_delimiter{':'}; | 76 | static constexpr char token_delimiter{':'}; |
| 77 | 77 | ||
| 78 | static void PadToken(std::string& token) { | 78 | static void PadToken(std::string& token) { |
| 79 | const auto remainder = token.size() % 3; | 79 | std::size_t outlen = 0; |
| 80 | for (size_t i = 0; i < (3 - remainder); i++) { | 80 | |
| 81 | std::array<unsigned char, 512> output{}; | ||
| 82 | std::array<unsigned char, 2048> roundtrip{}; | ||
| 83 | for (size_t i = 0; i < 3; i++) { | ||
| 84 | mbedtls_base64_decode(output.data(), output.size(), &outlen, | ||
| 85 | reinterpret_cast<const unsigned char*>(token.c_str()), | ||
| 86 | token.length()); | ||
| 87 | mbedtls_base64_encode(roundtrip.data(), roundtrip.size(), &outlen, output.data(), outlen); | ||
| 88 | if (memcmp(roundtrip.data(), token.data(), token.size()) == 0) { | ||
| 89 | break; | ||
| 90 | } | ||
| 81 | token.push_back('='); | 91 | token.push_back('='); |
| 82 | } | 92 | } |
| 83 | } | 93 | } |