mirror of https://github.com/OpenRCT2/OpenRCT2.git
Add NetworkServerAdvertiser class
This commit is contained in:
parent
6266e77fd5
commit
a346d8a048
|
@ -119,6 +119,7 @@
|
||||||
<ClCompile Include="src\network\NetworkKey.cpp" />
|
<ClCompile Include="src\network\NetworkKey.cpp" />
|
||||||
<ClCompile Include="src\network\NetworkPacket.cpp" />
|
<ClCompile Include="src\network\NetworkPacket.cpp" />
|
||||||
<ClCompile Include="src\network\NetworkPlayer.cpp" />
|
<ClCompile Include="src\network\NetworkPlayer.cpp" />
|
||||||
|
<ClCompile Include="src\network\NetworkServerAdvertiser.cpp" />
|
||||||
<ClCompile Include="src\network\NetworkUser.cpp" />
|
<ClCompile Include="src\network\NetworkUser.cpp" />
|
||||||
<ClCompile Include="src\network\TcpSocket.cpp" />
|
<ClCompile Include="src\network\TcpSocket.cpp" />
|
||||||
<ClCompile Include="src\network\twitch.cpp" />
|
<ClCompile Include="src\network\twitch.cpp" />
|
||||||
|
@ -438,6 +439,7 @@
|
||||||
<ClInclude Include="src\network\NetworkGroup.h" />
|
<ClInclude Include="src\network\NetworkGroup.h" />
|
||||||
<ClInclude Include="src\network\NetworkPacket.h" />
|
<ClInclude Include="src\network\NetworkPacket.h" />
|
||||||
<ClInclude Include="src\network\NetworkPlayer.h" />
|
<ClInclude Include="src\network\NetworkPlayer.h" />
|
||||||
|
<ClInclude Include="src\network\NetworkServerAdvertiser.h" />
|
||||||
<ClInclude Include="src\network\NetworkTypes.h" />
|
<ClInclude Include="src\network\NetworkTypes.h" />
|
||||||
<ClInclude Include="src\network\NetworkUser.h" />
|
<ClInclude Include="src\network\NetworkUser.h" />
|
||||||
<ClInclude Include="src\network\TcpSocket.h" />
|
<ClInclude Include="src\network\TcpSocket.h" />
|
||||||
|
|
|
@ -0,0 +1,258 @@
|
||||||
|
#pragma region Copyright (c) 2014-2016 OpenRCT2 Developers
|
||||||
|
/*****************************************************************************
|
||||||
|
* OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
|
||||||
|
*
|
||||||
|
* OpenRCT2 is the work of many authors, a full list can be found in contributors.md
|
||||||
|
* For more information, visit https://github.com/OpenRCT2/OpenRCT2
|
||||||
|
*
|
||||||
|
* OpenRCT2 is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* A full copy of the GNU General Public License can be found in licence.txt
|
||||||
|
*****************************************************************************/
|
||||||
|
#pragma endregion
|
||||||
|
|
||||||
|
#ifndef DISABLE_NETWORK
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include "../core/Console.hpp"
|
||||||
|
#include "../core/String.hpp"
|
||||||
|
#include "../core/Util.hpp"
|
||||||
|
#include "network.h"
|
||||||
|
#include "NetworkServerAdvertiser.h"
|
||||||
|
|
||||||
|
extern "C"
|
||||||
|
{
|
||||||
|
#include "../addresses.h"
|
||||||
|
#include "../config.h"
|
||||||
|
#include "../localisation/date.h"
|
||||||
|
#include "../management/finance.h"
|
||||||
|
#include "../peep/peep.h"
|
||||||
|
#include "../world/map.h"
|
||||||
|
#include "../world/park.h"
|
||||||
|
#include "http.h"
|
||||||
|
}
|
||||||
|
|
||||||
|
enum MASTER_SERVER_STATUS
|
||||||
|
{
|
||||||
|
MASTER_SERVER_STATUS_OK = 200,
|
||||||
|
MASTER_SERVER_STATUS_INVALID_TOKEN = 401,
|
||||||
|
MASTER_SERVER_STATUS_SERVER_NOT_FOUND = 404,
|
||||||
|
MASTER_SERVER_STATUS_INTERNAL_ERROR = 500
|
||||||
|
};
|
||||||
|
|
||||||
|
constexpr int MASTER_SERVER_REGISTER_TIME = 120 * 1000; // 2 minutes
|
||||||
|
constexpr int MASTER_SERVER_HEARTBEAT_TIME = 60 * 1000; // 1 minute
|
||||||
|
|
||||||
|
class NetworkServerAdvertiser : public INetworkServerAdvertiser
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
uint16 _port;
|
||||||
|
|
||||||
|
ADVERTISE_STATUS _status = ADVERTISE_STATUS_UNREGISTERED;
|
||||||
|
uint32 _lastAdvertiseTime = 0;
|
||||||
|
uint32 _lastHeartbeatTime = 0;
|
||||||
|
|
||||||
|
// Our unique token for this server
|
||||||
|
std::string _token;
|
||||||
|
|
||||||
|
// Key received from the master server
|
||||||
|
std::string _key;
|
||||||
|
|
||||||
|
public:
|
||||||
|
NetworkServerAdvertiser(uint16 port)
|
||||||
|
{
|
||||||
|
_port = port;
|
||||||
|
_key = GenerateAdvertiseKey();
|
||||||
|
}
|
||||||
|
|
||||||
|
ADVERTISE_STATUS GetStatus() override
|
||||||
|
{
|
||||||
|
return _status;
|
||||||
|
}
|
||||||
|
|
||||||
|
void Update() override
|
||||||
|
{
|
||||||
|
switch (_status) {
|
||||||
|
case ADVERTISE_STATUS_UNREGISTERED:
|
||||||
|
if (_lastAdvertiseTime == 0 || SDL_TICKS_PASSED(SDL_GetTicks(), _lastAdvertiseTime + MASTER_SERVER_REGISTER_TIME))
|
||||||
|
{
|
||||||
|
SendRegistration();
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case ADVERTISE_STATUS_REGISTERED:
|
||||||
|
if (SDL_TICKS_PASSED(SDL_GetTicks(), _lastHeartbeatTime + MASTER_SERVER_HEARTBEAT_TIME))
|
||||||
|
{
|
||||||
|
SendHeartbeat();
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
// exhaust enum values to satisfy clang
|
||||||
|
case ADVERTISE_STATUS_DISABLED:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
void SendRegistration()
|
||||||
|
{
|
||||||
|
_lastAdvertiseTime = SDL_GetTicks();
|
||||||
|
|
||||||
|
// Send the registration request
|
||||||
|
http_json_request request;
|
||||||
|
request.tag = this;
|
||||||
|
request.url = GetMasterServerUrl();
|
||||||
|
request.method = HTTP_METHOD_POST;
|
||||||
|
|
||||||
|
json_t *body = json_object();
|
||||||
|
json_object_set_new(body, "key", json_string(_key.c_str()));
|
||||||
|
json_object_set_new(body, "port", json_integer(_port));
|
||||||
|
request.body = body;
|
||||||
|
|
||||||
|
http_request_json_async(&request, [](http_json_response * response) -> void
|
||||||
|
{
|
||||||
|
if (response == nullptr)
|
||||||
|
{
|
||||||
|
Console::WriteLine("Unable to connect to master server");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
auto advertiser = (NetworkServerAdvertiser *)response->tag;
|
||||||
|
advertiser->OnRegistrationResponse(response->root);
|
||||||
|
http_request_json_dispose(response);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
json_decref(body);
|
||||||
|
}
|
||||||
|
|
||||||
|
void SendHeartbeat()
|
||||||
|
{
|
||||||
|
http_json_request request;
|
||||||
|
request.tag = this;
|
||||||
|
request.url = GetMasterServerUrl();
|
||||||
|
request.method = HTTP_METHOD_PUT;
|
||||||
|
|
||||||
|
json_t * jsonBody = GetHeartbeatJson();
|
||||||
|
request.body = jsonBody;
|
||||||
|
|
||||||
|
_lastHeartbeatTime = SDL_GetTicks();
|
||||||
|
http_request_json_async(&request, [](http_json_response *response) -> void
|
||||||
|
{
|
||||||
|
if (response == nullptr)
|
||||||
|
{
|
||||||
|
log_warning("Unable to connect to master server");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
auto advertiser = (NetworkServerAdvertiser *)response->tag;
|
||||||
|
advertiser->OnHeartbeatResponse(response->root);
|
||||||
|
http_request_json_dispose(response);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
json_decref(jsonBody);
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnRegistrationResponse(json_t * jsonRoot)
|
||||||
|
{
|
||||||
|
json_t *jsonStatus = json_object_get(jsonRoot, "status");
|
||||||
|
if (json_is_integer(jsonStatus))
|
||||||
|
{
|
||||||
|
int status = (int)json_integer_value(jsonStatus);
|
||||||
|
if (status == MASTER_SERVER_STATUS_OK)
|
||||||
|
{
|
||||||
|
json_t * jsonToken = json_object_get(jsonRoot, "token");
|
||||||
|
if (json_is_string(jsonToken))
|
||||||
|
{
|
||||||
|
_token = std::string(json_string_value(jsonToken));
|
||||||
|
_status = ADVERTISE_STATUS_REGISTERED;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
const char * message = "Invalid response from server";
|
||||||
|
json_t * jsonMessage = json_object_get(jsonRoot, "message");
|
||||||
|
if (json_is_string(jsonMessage))
|
||||||
|
{
|
||||||
|
message = json_string_value(jsonMessage);
|
||||||
|
}
|
||||||
|
Console::Error::WriteLine("Unable to advertise: %s", message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnHeartbeatResponse(json_t * jsonRoot)
|
||||||
|
{
|
||||||
|
json_t *jsonStatus = json_object_get(jsonRoot, "status");
|
||||||
|
if (json_is_integer(jsonStatus))
|
||||||
|
{
|
||||||
|
int status = (int)json_integer_value(jsonStatus);
|
||||||
|
if (status == MASTER_SERVER_STATUS_OK)
|
||||||
|
{
|
||||||
|
// Master server has successfully updated our server status
|
||||||
|
}
|
||||||
|
else if (status == MASTER_SERVER_STATUS_INVALID_TOKEN)
|
||||||
|
{
|
||||||
|
_status = ADVERTISE_STATUS_UNREGISTERED;
|
||||||
|
Console::WriteLine("Master server heartbeat failed: Invalid Token");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
json_t * GetHeartbeatJson()
|
||||||
|
{
|
||||||
|
uint32 numPlayers = network_get_num_players();
|
||||||
|
|
||||||
|
json_t * root = json_object();
|
||||||
|
json_object_set_new(root, "token", json_string(_token.c_str()));
|
||||||
|
json_object_set_new(root, "players", json_integer(numPlayers));
|
||||||
|
|
||||||
|
json_t * gameInfo = json_object();
|
||||||
|
json_object_set_new(gameInfo, "mapSize", json_integer(gMapSize - 2));
|
||||||
|
json_object_set_new(gameInfo, "day", json_integer(gDateMonthTicks));
|
||||||
|
json_object_set_new(gameInfo, "month", json_integer(gDateMonthsElapsed));
|
||||||
|
json_object_set_new(gameInfo, "guests", json_integer(gNumGuestsInPark));
|
||||||
|
json_object_set_new(gameInfo, "parkValue", json_integer(gParkValue));
|
||||||
|
if (!(gParkFlags & PARK_FLAGS_NO_MONEY))
|
||||||
|
{
|
||||||
|
money32 cash = DECRYPT_MONEY(gCashEncrypted);
|
||||||
|
json_object_set_new(gameInfo, "cash", json_integer(cash));
|
||||||
|
}
|
||||||
|
json_object_set_new(root, "gameInfo", gameInfo);
|
||||||
|
|
||||||
|
return root;
|
||||||
|
}
|
||||||
|
|
||||||
|
static std::string GenerateAdvertiseKey()
|
||||||
|
{
|
||||||
|
// Generate a string of 16 random hex characters (64-integer key as a hex formatted string)
|
||||||
|
static const char hexChars[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f' };
|
||||||
|
char key[17];
|
||||||
|
for (int i = 0; i < 16; i++)
|
||||||
|
{
|
||||||
|
int hexCharIndex = rand() % Util::CountOf(hexChars);
|
||||||
|
key[i] = hexChars[hexCharIndex];
|
||||||
|
}
|
||||||
|
key[Util::CountOf(key) - 1] = 0;
|
||||||
|
return key;
|
||||||
|
}
|
||||||
|
|
||||||
|
static const char * GetMasterServerUrl()
|
||||||
|
{
|
||||||
|
const char * result = OPENRCT2_MASTER_SERVER_URL;
|
||||||
|
if (!String::IsNullOrEmpty(gConfigNetwork.master_server_url))
|
||||||
|
{
|
||||||
|
result = gConfigNetwork.master_server_url;
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
INetworkServerAdvertiser * CreateServerAdvertiser(uint16 port)
|
||||||
|
{
|
||||||
|
return new NetworkServerAdvertiser(port);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // DISABLE_NETWORK
|
|
@ -0,0 +1,36 @@
|
||||||
|
#pragma region Copyright (c) 2014-2016 OpenRCT2 Developers
|
||||||
|
/*****************************************************************************
|
||||||
|
* OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
|
||||||
|
*
|
||||||
|
* OpenRCT2 is the work of many authors, a full list can be found in contributors.md
|
||||||
|
* For more information, visit https://github.com/OpenRCT2/OpenRCT2
|
||||||
|
*
|
||||||
|
* OpenRCT2 is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* A full copy of the GNU General Public License can be found in licence.txt
|
||||||
|
*****************************************************************************/
|
||||||
|
#pragma endregion
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "../common.h"
|
||||||
|
|
||||||
|
enum ADVERTISE_STATUS
|
||||||
|
{
|
||||||
|
ADVERTISE_STATUS_DISABLED,
|
||||||
|
ADVERTISE_STATUS_UNREGISTERED,
|
||||||
|
ADVERTISE_STATUS_REGISTERED,
|
||||||
|
};
|
||||||
|
|
||||||
|
interface INetworkServerAdvertiser
|
||||||
|
{
|
||||||
|
virtual ~INetworkServerAdvertiser() { }
|
||||||
|
|
||||||
|
virtual ADVERTISE_STATUS GetStatus() abstract;
|
||||||
|
virtual void Update() abstract;
|
||||||
|
};
|
||||||
|
|
||||||
|
INetworkServerAdvertiser * CreateServerAdvertiser(uint16 port);
|
Loading…
Reference in New Issue