void* wk_fileServ(void* sockp)
{
FileServer_Private client;
pthread_mutex_lock(&s_clientMutex);
client.m_sock = reinterpret_cast<DS::SocketHandle>(sockp);
client.m_readerId = 0;
s_clients.push_back(&client);
pthread_mutex_unlock(&s_clientMutex);
try {
file_init(client);
for ( ;; ) {
DS::RecvValue<uint32_t>(client.m_sock); // Message size
uint32_t msgId = DS::RecvValue<uint32_t>(client.m_sock);
switch (msgId) {
case e_CliToFile_PingRequest:
cb_ping(client);
break;
case e_CliToFile_BuildIdRequest:
cb_buildId(client);
break;
case e_CliToFile_ManifestRequest:
cb_manifest(client);
break;
case e_CliToFile_ManifestEntryAck:
cb_manifestAck(client);
break;
case e_CliToFile_DownloadRequest:
cb_downloadStart(client);
break;
case e_CliToFile_DownloadChunkAck:
cb_downloadNext(client);
break;
default:
/* Invalid message */
fprintf(stderr, "[File] Got invalid message ID %d from %s\n",
msgId, DS::SockIpAddress(client.m_sock).c_str());
DS::CloseSock(client.m_sock);
throw DS::SockHup();
}
}
} catch (DS::AssertException ex) {
fprintf(stderr, "[File] Assertion failed at %s:%ld: %s\n",
ex.m_file, ex.m_line, ex.m_cond);
} catch (DS::SockHup) {
// Socket closed...
}
pthread_mutex_lock(&s_clientMutex);
auto client_iter = s_clients.begin();
while (client_iter != s_clients.end()) {
if (*client_iter == &client)
client_iter = s_clients.erase(client_iter);
else
++client_iter;
}
pthread_mutex_unlock(&s_clientMutex);
DS::FreeSock(client.m_sock);
return 0;
}
void cb_sockRead(AuthServer_Private& client)
{
uint16_t msgId = DS::CryptRecvValue<uint16_t>(client.m_sock, client.m_crypt);
switch (msgId) {
case e_CliToAuth_PingRequest:
cb_ping(client);
break;
case e_CliToAuth_ClientRegisterRequest:
cb_register(client);
break;
case e_CliToAuth_AcctLoginRequest:
cb_login(client);
break;
case e_CliToAuth_AcctSetPlayerRequest:
cb_setPlayer(client);
break;
case e_CliToAuth_PlayerCreateRequest:
cb_playerCreate(client);
break;
case e_CliToAuth_PlayerDeleteRequest:
cb_playerDelete(client);
break;
case e_CliToAuth_VaultNodeCreate:
cb_nodeCreate(client);
break;
case e_CliToAuth_VaultNodeFetch:
cb_nodeFetch(client);
break;
case e_CliToAuth_VaultNodeSave:
cb_nodeUpdate(client);
break;
case e_CliToAuth_VaultNodeAdd:
cb_nodeRef(client);
break;
case e_CliToAuth_VaultNodeRemove:
cb_nodeUnref(client);
break;
case e_CliToAuth_VaultFetchNodeRefs:
cb_nodeTree(client);
break;
case e_CliToAuth_VaultInitAgeRequest:
cb_ageCreate(client);
break;
case e_CliToAuth_VaultNodeFind:
cb_nodeFind(client);
break;
case e_CliToAuth_VaultSendNode:
cb_nodeSend(client);
break;
case e_CliToAuth_AgeRequest:
cb_ageRequest(client, false);
break;
case e_CliToAuth_AgeRequestEx:
cb_ageRequest(client, true);
break;
case e_CliToAuth_FileListRequest:
cb_fileList(client);
break;
case e_CliToAuth_FileDownloadRequest:
cb_downloadStart(client);
break;
case e_CliToAuth_FileDownloadChunkAck:
cb_downloadNext(client);
break;
case e_CliToAuth_LogPythonTraceback:
printf("[Auth] Got client python traceback:\n%s\n",
DS::CryptRecvString(client.m_sock, client.m_crypt).c_str());
break;
case e_CliToAuth_LogStackDump:
printf("[Auth] Got client stackdump:\n%s\n",
DS::CryptRecvString(client.m_sock, client.m_crypt).c_str());
break;
case e_CliToAuth_LogClientDebuggerConnect:
// Nobody cares
break;
case e_CliToAuth_ScoreCreate:
cb_scoreCreate(client);
break;
case e_CliToAuth_ScoreGetScores:
cb_scoreGetScores(client);
break;
case e_CliToAuth_ScoreAddPoints:
cb_scoreAddPoints(client);
break;
case e_CliToAuth_ScoreTransferPoints:
cb_scoreTransferPoints(client);
break;
case e_CliToAuth_ScoreSetPoints:
cb_scoreSetPoints(client);
break;
case e_CliToAuth_ScoreGetHighScores:
cb_scoreGetHighScores(client);
break;
case e_CliToAuth_GetPublicAgeList:
cb_getPublicAges(client);
break;
case e_CliToAuth_SetAgePublic:
cb_setAgePublic(client);
break;
case e_CliToAuth_ClientSetCCRLevel:
case e_CliToAuth_AcctSetRolesRequest:
case e_CliToAuth_AcctSetBillingTypeRequest:
case e_CliToAuth_AcctActivateRequest:
case e_CliToAuth_AcctCreateFromKeyRequest:
case e_CliToAuth_VaultNodeDelete:
case e_CliToAuth_UpgradeVisitorRequest:
case e_CliToAuth_SetPlayerBanStatusRequest:
case e_CliToAuth_KickPlayer:
fprintf(stderr, "[Auth] Got unsupported client message %d from %s\n",
msgId, DS::SockIpAddress(client.m_sock).c_str());
DS::CloseSock(client.m_sock);
throw DS::SockHup();
default:
/* Invalid message */
fprintf(stderr, "[Auth] Got invalid message ID %d from %s\n",
msgId, DS::SockIpAddress(client.m_sock).c_str());
DS::CloseSock(client.m_sock);
throw DS::SockHup();
}
}
void wk_fileServ(DS::SocketHandle sockp)
{
FileServer_Private client;
s_clientMutex.lock();
client.m_sock = sockp;
client.m_readerId = 0;
s_clients.push_back(&client);
s_clientMutex.unlock();
try {
file_init(client);
for ( ;; ) {
DS::RecvValue<uint32_t>(client.m_sock); // Message size
uint32_t msgId = DS::RecvValue<uint32_t>(client.m_sock);
switch (msgId) {
case e_CliToFile_PingRequest:
cb_ping(client);
break;
case e_CliToFile_BuildIdRequest:
cb_buildId(client);
break;
case e_CliToFile_ManifestRequest:
cb_manifest(client);
break;
case e_CliToFile_ManifestEntryAck:
cb_manifestAck(client);
break;
case e_CliToFile_DownloadRequest:
cb_downloadStart(client);
break;
case e_CliToFile_DownloadChunkAck:
cb_downloadNext(client);
break;
default:
/* Invalid message */
fprintf(stderr, "[File] Got invalid message ID %d from %s\n",
msgId, DS::SockIpAddress(client.m_sock).c_str());
DS::CloseSock(client.m_sock);
throw DS::SockHup();
}
}
} catch (const DS::AssertException& ex) {
fprintf(stderr, "[File] Assertion failed at %s:%ld: %s\n",
ex.m_file, ex.m_line, ex.m_cond);
} catch (const DS::PacketSizeOutOfBounds& ex) {
fprintf(stderr, "[File] Client packet size too large: Requested %u bytes\n",
ex.requestedSize());
} catch (const DS::SockHup&) {
// Socket closed...
}
s_clientMutex.lock();
auto client_iter = s_clients.begin();
while (client_iter != s_clients.end()) {
if (*client_iter == &client)
client_iter = s_clients.erase(client_iter);
else
++client_iter;
}
s_clientMutex.unlock();
DS::FreeSock(client.m_sock);
}
