summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/dedicated_room/yuzu_room.cpp14
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";
76static constexpr char token_delimiter{':'}; 76static constexpr char token_delimiter{':'};
77 77
78static void PadToken(std::string& token) { 78static 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}