UICommon: Add NetPlayIndex helper
This commit is contained in:
spycrab
parent
3dbf44417a
commit
827e40d78a
|
|
@ -5,6 +5,7 @@ add_library(uicommon
|
|||
DiscordPresence.cpp
|
||||
GameFile.cpp
|
||||
GameFileCache.cpp
|
||||
NetPlayIndex.cpp
|
||||
ResourcePack/Manager.cpp
|
||||
ResourcePack/Manifest.cpp
|
||||
ResourcePack/ResourcePack.cpp
|
||||
|
|
|
|||
|
|
@ -0,0 +1,321 @@
|
|||
// Copyright 2019 Dolphin Emulator Project
|
||||
// Licensed under GPLv2+
|
||||
// Refer to the license.txt file included.
|
||||
|
||||
#include "UICommon/NetPlayIndex.h"
|
||||
|
||||
#include <picojson/picojson.h>
|
||||
|
||||
#include "Common/HttpRequest.h"
|
||||
#include "Common/Thread.h"
|
||||
#include "Common/Version.h"
|
||||
|
||||
#include "Core/Config/NetplaySettings.h"
|
||||
|
||||
NetPlayIndex::NetPlayIndex()
|
||||
{
|
||||
}
|
||||
|
||||
NetPlayIndex::~NetPlayIndex()
|
||||
{
|
||||
if (!m_secret.empty())
|
||||
Remove();
|
||||
}
|
||||
|
||||
static std::optional<picojson::value> ParseResponse(std::vector<u8> response)
|
||||
{
|
||||
std::string response_string(reinterpret_cast<char*>(response.data()), response.size());
|
||||
|
||||
picojson::value json;
|
||||
|
||||
auto error = picojson::parse(json, response_string);
|
||||
|
||||
if (!error.empty())
|
||||
return {};
|
||||
|
||||
return json;
|
||||
}
|
||||
|
||||
std::optional<std::vector<NetPlaySession>>
|
||||
NetPlayIndex::List(const std::map<std::string, std::string>& filters)
|
||||
{
|
||||
Common::HttpRequest request;
|
||||
|
||||
std::string list_url = Config::Get(Config::NETPLAY_INDEX_URL) + "/v0/list";
|
||||
|
||||
if (!filters.empty())
|
||||
{
|
||||
list_url += "?";
|
||||
for (const auto& filter : filters)
|
||||
{
|
||||
list_url += filter.first + "=" + request.EscapeComponent(filter.second) + "&";
|
||||
}
|
||||
list_url = list_url.substr(0, list_url.size() - 1);
|
||||
}
|
||||
|
||||
auto response = request.Get(list_url);
|
||||
|
||||
if (!response)
|
||||
{
|
||||
m_last_error = "NO_RESPONSE";
|
||||
return {};
|
||||
}
|
||||
|
||||
auto json = ParseResponse(response.value());
|
||||
|
||||
if (!json)
|
||||
{
|
||||
m_last_error = "BAD_JSON";
|
||||
return {};
|
||||
}
|
||||
|
||||
const auto& status = json->get("status");
|
||||
|
||||
if (status.to_str() != "OK")
|
||||
{
|
||||
m_last_error = status.to_str();
|
||||
return {};
|
||||
}
|
||||
|
||||
const auto& entries = json->get("sessions");
|
||||
|
||||
std::vector<NetPlaySession> sessions;
|
||||
|
||||
for (const auto& entry : entries.get<picojson::array>())
|
||||
{
|
||||
NetPlaySession session;
|
||||
|
||||
const auto& name = entry.get("name");
|
||||
const auto& region = entry.get("region");
|
||||
const auto& method = entry.get("method");
|
||||
const auto& game_id = entry.get("game");
|
||||
const auto& server_id = entry.get("server_id");
|
||||
const auto& has_password = entry.get("password");
|
||||
const auto& player_count = entry.get("player_count");
|
||||
const auto& port = entry.get("port");
|
||||
const auto& in_game = entry.get("in_game");
|
||||
|
||||
if (!name.is<std::string>() || !region.is<std::string>() || !method.is<std::string>() ||
|
||||
!server_id.is<std::string>() || !game_id.is<std::string>() || !has_password.is<bool>() ||
|
||||
!player_count.is<double>() || !port.is<double>() || !in_game.is<bool>())
|
||||
continue;
|
||||
|
||||
session.name = name.to_str();
|
||||
session.region = region.to_str();
|
||||
session.game_id = game_id.to_str();
|
||||
session.server_id = server_id.to_str();
|
||||
session.method = method.to_str();
|
||||
session.has_password = has_password.get<bool>();
|
||||
session.player_count = static_cast<int>(player_count.get<double>());
|
||||
session.port = static_cast<int>(port.get<double>());
|
||||
session.in_game = in_game.get<bool>();
|
||||
|
||||
sessions.push_back(session);
|
||||
}
|
||||
|
||||
return sessions;
|
||||
}
|
||||
|
||||
void NetPlayIndex::NotificationLoop()
|
||||
{
|
||||
while (m_running.IsSet())
|
||||
{
|
||||
Common::HttpRequest request;
|
||||
auto response = request.Get(
|
||||
Config::Get(Config::NETPLAY_INDEX_URL) + "/v0/session/active?secret=" + m_secret +
|
||||
"&player_count=" + std::to_string(m_player_count) +
|
||||
"&game=" + request.EscapeComponent(m_game) + "&in_game=" + std::to_string(m_in_game));
|
||||
|
||||
if (!response)
|
||||
continue;
|
||||
|
||||
auto json = ParseResponse(response.value());
|
||||
|
||||
if (!json)
|
||||
{
|
||||
m_last_error = "BAD_JSON";
|
||||
m_running.Set(false);
|
||||
return;
|
||||
}
|
||||
|
||||
std::string status = json->get("status").to_str();
|
||||
|
||||
if (status != "OK")
|
||||
{
|
||||
m_last_error = status;
|
||||
m_running.Set(false);
|
||||
return;
|
||||
}
|
||||
|
||||
Common::SleepCurrentThread(1000 * 5);
|
||||
}
|
||||
}
|
||||
|
||||
bool NetPlayIndex::Add(NetPlaySession session)
|
||||
{
|
||||
m_running.Set(true);
|
||||
|
||||
Common::HttpRequest request;
|
||||
auto response = request.Get(
|
||||
Config::Get(Config::NETPLAY_INDEX_URL) + "/v0/session/add?name=" +
|
||||
request.EscapeComponent(session.name) + "®ion=" + request.EscapeComponent(session.region) +
|
||||
"&game=" + request.EscapeComponent(session.game_id) +
|
||||
"&password=" + std::to_string(session.has_password) + "&method=" + session.method +
|
||||
"&server_id=" + session.server_id + "&in_game=" + std::to_string(session.in_game) +
|
||||
"&port=" + std::to_string(session.port) +
|
||||
"&player_count=" + std::to_string(session.player_count) + "&version=" + Common::scm_desc_str);
|
||||
|
||||
if (!response.has_value())
|
||||
{
|
||||
m_last_error = "NO_RESPONSE";
|
||||
return false;
|
||||
}
|
||||
|
||||
auto json = ParseResponse(response.value());
|
||||
|
||||
if (!json)
|
||||
{
|
||||
m_last_error = "BAD_JSON";
|
||||
return false;
|
||||
}
|
||||
|
||||
std::string status = json->get("status").to_str();
|
||||
|
||||
if (status != "OK")
|
||||
{
|
||||
m_last_error = status;
|
||||
return false;
|
||||
}
|
||||
|
||||
m_secret = json->get("secret").to_str();
|
||||
m_in_game = session.in_game;
|
||||
m_player_count = session.player_count;
|
||||
m_game = session.game_id;
|
||||
|
||||
m_session_thread = std::thread([this] { NotificationLoop(); });
|
||||
|
||||
m_session_thread.detach();
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void NetPlayIndex::SetInGame(bool in_game)
|
||||
{
|
||||
m_in_game = in_game;
|
||||
}
|
||||
|
||||
void NetPlayIndex::SetPlayerCount(int player_count)
|
||||
{
|
||||
m_player_count = player_count;
|
||||
}
|
||||
|
||||
void NetPlayIndex::SetGame(const std::string& game)
|
||||
{
|
||||
m_game = game;
|
||||
}
|
||||
|
||||
void NetPlayIndex::Remove()
|
||||
{
|
||||
if (m_secret.empty())
|
||||
return;
|
||||
|
||||
m_running.Set(false);
|
||||
|
||||
if (m_session_thread.joinable())
|
||||
m_session_thread.join();
|
||||
|
||||
// We don't really care whether this fails or not
|
||||
Common::HttpRequest request;
|
||||
request.Get(Config::Get(Config::NETPLAY_INDEX_URL) + "/v0/session/remove?secret=" + m_secret);
|
||||
|
||||
m_secret = "";
|
||||
}
|
||||
|
||||
std::vector<std::string> NetPlayIndex::GetRegions()
|
||||
{
|
||||
static std::vector<std::string> regions{"AF", "CN", "EA", "EU", "NA", "OC"};
|
||||
|
||||
return regions;
|
||||
}
|
||||
|
||||
// This encryption system uses simple XOR operations and a checksum
|
||||
// It isn't very secure but is preferable to adding another dependency on mbedtls
|
||||
// The encrypted data is encoded as nibbles with the character 'A' as the base offset
|
||||
|
||||
bool NetPlaySession::EncryptID(const std::string& password)
|
||||
{
|
||||
if (password.empty())
|
||||
return false;
|
||||
|
||||
u8 i = 0;
|
||||
|
||||
std::string to_encrypt = server_id;
|
||||
|
||||
// Calculate and append checksum to ID
|
||||
u8 sum = 0;
|
||||
|
||||
for (char c : to_encrypt)
|
||||
sum += c;
|
||||
|
||||
to_encrypt += sum;
|
||||
|
||||
std::string encrypted_id;
|
||||
|
||||
for (const char& byte : to_encrypt)
|
||||
{
|
||||
char c = byte ^ password[i % password.size()];
|
||||
c += i;
|
||||
encrypted_id += 'A' + ((c & 0xF0) >> 4);
|
||||
encrypted_id += 'A' + (c & 0x0F);
|
||||
++i;
|
||||
}
|
||||
|
||||
server_id = encrypted_id;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
std::optional<std::string> NetPlaySession::DecryptID(const std::string& password) const
|
||||
{
|
||||
if (password.empty())
|
||||
return {};
|
||||
|
||||
// If the length of an encrypted session id is not divisble by two, it's invalid
|
||||
if (server_id.empty() || server_id.size() % 2 != 0)
|
||||
return {};
|
||||
|
||||
std::string decoded;
|
||||
|
||||
for (size_t i = 0; i < server_id.size(); i += 2)
|
||||
{
|
||||
char c = (server_id[i] - 'A') << 4 | (server_id[i + 1] - 'A');
|
||||
decoded.push_back(c);
|
||||
}
|
||||
|
||||
u8 i = 0;
|
||||
for (auto& c : decoded)
|
||||
{
|
||||
c -= i;
|
||||
c ^= password[i % password.size()];
|
||||
++i;
|
||||
}
|
||||
|
||||
// Verify checksum
|
||||
u8 expected_sum = decoded[decoded.size() - 1];
|
||||
|
||||
decoded = decoded.substr(0, decoded.size() - 1);
|
||||
|
||||
u8 sum = 0;
|
||||
for (char c : decoded)
|
||||
sum += c;
|
||||
|
||||
if (sum != expected_sum)
|
||||
return {};
|
||||
|
||||
return decoded;
|
||||
}
|
||||
|
||||
const std::string& NetPlayIndex::GetLastError() const
|
||||
{
|
||||
return m_last_error;
|
||||
}
|
||||
|
|
@ -0,0 +1,65 @@
|
|||
// Copyright 2019 Dolphin Emulator Project
|
||||
// Licensed under GPLv2+
|
||||
// Refer to the license.txt file included.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <map>
|
||||
#include <optional>
|
||||
#include <string>
|
||||
#include <thread>
|
||||
#include <vector>
|
||||
|
||||
#include "Common/Flag.h"
|
||||
|
||||
struct NetPlaySession
|
||||
{
|
||||
std::string name;
|
||||
std::string region;
|
||||
std::string method;
|
||||
std::string server_id;
|
||||
std::string game_id;
|
||||
|
||||
int player_count;
|
||||
int port;
|
||||
|
||||
bool has_password;
|
||||
bool in_game;
|
||||
|
||||
bool EncryptID(const std::string& password);
|
||||
std::optional<std::string> DecryptID(const std::string& password) const;
|
||||
};
|
||||
|
||||
class NetPlayIndex
|
||||
{
|
||||
public:
|
||||
explicit NetPlayIndex();
|
||||
~NetPlayIndex();
|
||||
|
||||
std::optional<std::vector<NetPlaySession>>
|
||||
List(const std::map<std::string, std::string>& filters = {});
|
||||
|
||||
static std::vector<std::string> GetRegions();
|
||||
|
||||
bool Add(NetPlaySession session);
|
||||
void Remove();
|
||||
|
||||
void SetPlayerCount(int player_count);
|
||||
void SetInGame(bool in_game);
|
||||
void SetGame(const std::string& game);
|
||||
|
||||
const std::string& GetLastError() const;
|
||||
|
||||
private:
|
||||
void NotificationLoop();
|
||||
|
||||
Common::Flag m_running;
|
||||
|
||||
std::string m_secret;
|
||||
std::string m_game;
|
||||
int m_player_count = 0;
|
||||
bool m_in_game = false;
|
||||
|
||||
std::string m_last_error;
|
||||
std::thread m_session_thread;
|
||||
};
|
||||
|
|
@ -53,6 +53,7 @@
|
|||
<ClCompile Include="AutoUpdate.cpp" />
|
||||
<ClCompile Include="CommandLineParse.cpp" />
|
||||
<ClCompile Include="DiscordPresence.cpp" />
|
||||
<ClCompile Include="NetPlayIndex.cpp" />
|
||||
<ClCompile Include="ResourcePack\Manager.cpp" />
|
||||
<ClCompile Include="ResourcePack\Manifest.cpp" />
|
||||
<ClCompile Include="ResourcePack\ResourcePack.cpp" />
|
||||
|
|
@ -69,6 +70,7 @@
|
|||
<ClInclude Include="AutoUpdate.h" />
|
||||
<ClInclude Include="CommandLineParse.h" />
|
||||
<ClInclude Include="DiscordPresence.h" />
|
||||
<ClInclude Include="NetPlayIndex.h" />
|
||||
<ClInclude Include="ResourcePack\Manager.h" />
|
||||
<ClInclude Include="ResourcePack\Manifest.h" />
|
||||
<ClInclude Include="ResourcePack\ResourcePack.h" />
|
||||
|
|
@ -86,4 +88,4 @@
|
|||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
|
||||
<ImportGroup Label="ExtensionTargets">
|
||||
</ImportGroup>
|
||||
</Project>
|
||||
</Project>
|
||||
|
|
|
|||
Loading…
Reference in New Issue