1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
|
#pragma once
#include "core/convar/convar.h"
#include "server/serverpresence.h"
#include <winsock2.h>
#include <string>
#include <cstring>
#include <future>
#include <unordered_set>
extern ConVar* Cvar_ns_masterserver_hostname;
extern ConVar* Cvar_ns_curl_log_enable;
struct RemoteModInfo
{
public:
std::string Name;
std::string Version;
};
class RemoteServerInfo
{
public:
char id[33]; // 32 bytes + nullterminator
// server info
char name[64];
std::string description;
char map[32];
char playlist[16];
char region[32];
std::vector<RemoteModInfo> requiredMods;
int playerCount;
int maxPlayers;
// connection stuff
bool requiresPassword;
public:
RemoteServerInfo(
const char* newId,
const char* newName,
const char* newDescription,
const char* newMap,
const char* newPlaylist,
const char* newRegion,
int newPlayerCount,
int newMaxPlayers,
bool newRequiresPassword);
};
struct RemoteServerConnectionInfo
{
public:
char authToken[32];
in_addr ip;
unsigned short port;
};
struct MainMenuPromoData
{
public:
std::string newInfoTitle1;
std::string newInfoTitle2;
std::string newInfoTitle3;
std::string largeButtonTitle;
std::string largeButtonText;
std::string largeButtonUrl;
int largeButtonImageIndex;
std::string smallButton1Title;
std::string smallButton1Url;
int smallButton1ImageIndex;
std::string smallButton2Title;
std::string smallButton2Url;
int smallButton2ImageIndex;
};
class MasterServerManager
{
private:
bool m_bRequestingServerList = false;
bool m_bAuthenticatingWithGameServer = false;
public:
char m_sOwnServerId[33];
char m_sOwnServerAuthToken[33];
char m_sOwnClientAuthToken[33];
std::string m_sOwnModInfoJson;
bool m_bOriginAuthWithMasterServerDone = false;
bool m_bOriginAuthWithMasterServerInProgress = false;
bool m_bOriginAuthWithMasterServerSuccessful = false;
std::string m_sOriginAuthWithMasterServerErrorCode = "";
std::string m_sOriginAuthWithMasterServerErrorMessage = "";
bool m_bSavingPersistentData = false;
bool m_bScriptRequestingServerList = false;
bool m_bSuccessfullyConnected = true;
bool m_bNewgameAfterSelfAuth = false;
bool m_bScriptAuthenticatingWithGameServer = false;
bool m_bSuccessfullyAuthenticatedWithGameServer = false;
std::string m_sAuthFailureReason {};
bool m_bHasPendingConnectionInfo = false;
RemoteServerConnectionInfo m_pendingConnectionInfo;
std::vector<RemoteServerInfo> m_vRemoteServers;
bool m_bHasMainMenuPromoData = false;
MainMenuPromoData m_sMainMenuPromoData;
std::optional<RemoteServerInfo> m_currentServer;
std::string m_sCurrentServerPassword;
std::unordered_set<std::string> m_handledServerConnections;
public:
MasterServerManager();
void ClearServerList();
void RequestServerList();
void RequestMainMenuPromos();
void AuthenticateOriginWithMasterServer(const char* uid, const char* originToken);
void AuthenticateWithOwnServer(const char* uid, const char* playerToken);
void AuthenticateWithServer(const char* uid, const char* playerToken, RemoteServerInfo server, const char* password);
void WritePlayerPersistentData(const char* playerId, const char* pdata, size_t pdataSize);
void ProcessConnectionlessPacketSigreq1(std::string req);
};
extern MasterServerManager* g_pMasterServerManager;
extern ConVar* Cvar_ns_masterserver_hostname;
/** Result returned in the std::future of a MasterServerPresenceReporter::ReportPresence() call. */
enum class MasterServerReportPresenceResult
{
// Adding this server to the MS was successful.
Success,
// We failed to add this server to the MS and should retry.
Failed,
// We failed to add this server to the MS and shouldn't retry.
FailedNoRetry,
// We failed to even reach the MS.
FailedNoConnect,
// We failed to add the server because an existing server with the same ip:port exists.
FailedDuplicateServer,
};
class MasterServerPresenceReporter : public ServerPresenceReporter
{
public:
/** Full data returned in the std::future of a MasterServerPresenceReporter::ReportPresence() call. */
struct ReportPresenceResultData
{
MasterServerReportPresenceResult result;
std::optional<std::string> id;
std::optional<std::string> serverAuthToken;
};
const int MAX_REGISTRATION_ATTEMPTS = 5;
// Called to initialise the master server presence reporter's state.
void CreatePresence(const ServerPresence* pServerPresence) override;
// Run on an internal to either add the server to the MS or update it.
void ReportPresence(const ServerPresence* pServerPresence) override;
// Called when we need to remove the server from the master server.
void DestroyPresence(const ServerPresence* pServerPresence) override;
// Called every frame.
void RunFrame(double flCurrentTime, const ServerPresence* pServerPresence) override;
protected:
// Contains the async logic to add the server to the MS.
void InternalAddServer(const ServerPresence* pServerPresence);
// Contains the async logic to update the server on the MS.
void InternalUpdateServer(const ServerPresence* pServerPresence);
// The future used for InternalAddServer() calls.
std::future<ReportPresenceResultData> addServerFuture;
// The future used for InternalAddServer() calls.
std::future<ReportPresenceResultData> updateServerFuture;
int m_nNumRegistrationAttempts;
double m_fNextAddServerAttemptTime;
};
|
