Discovery Server for match making setup

This commit is contained in:
HikikoMarmy
2025-04-14 03:54:05 +01:00
parent b9fa6b6a23
commit ed2a53d76a
2 changed files with 31 additions and 308 deletions

View File

@@ -10,6 +10,8 @@
DiscoveryServer::DiscoveryServer()
{
m_running = false;
m_socket = INVALID_SOCKET;
m_recvBuffer.resize( 1024 );
}
@@ -32,7 +34,7 @@ void DiscoveryServer::Start( std::string ip, int32_t port )
service.sin_family = AF_INET;
service.sin_port = htons( port );
if (ip == "0.0.0.0")
if( ip == "0.0.0.0" )
{
service.sin_addr.s_addr = ADDR_ANY;
}
@@ -66,60 +68,35 @@ void DiscoveryServer::Stop()
void DiscoveryServer::Run()
{
sockaddr_in clientAddr;
int clientAddrLen = sizeof(clientAddr);
int clientAddrLen = sizeof( clientAddr );
while( m_running )
{
std::this_thread::sleep_for(std::chrono::milliseconds(1));
std::this_thread::sleep_for( std::chrono::milliseconds( 1 ) );
auto bytesReceived = recvfrom(m_socket, (char*)m_recvBuffer.data(), 1024, 0, (struct sockaddr*)&clientAddr, &clientAddrLen);
auto bytesReceived = recvfrom( m_socket, ( char * )m_recvBuffer.data(), 1024, 0, ( struct sockaddr * )&clientAddr, &clientAddrLen );
if (bytesReceived == SOCKET_ERROR || bytesReceived < 4)
if( bytesReceived == SOCKET_ERROR || bytesReceived < 4 )
{
continue;
}
ProcessPacket(&clientAddr, std::make_shared< ByteStream >(m_recvBuffer.data(), bytesReceived));
ProcessPacket( &clientAddr, std::make_shared< ByteStream >( m_recvBuffer.data(), bytesReceived ) );
}
}
void DiscoveryServer::ProcessPacket(sockaddr_in* clientAddr, sptr_byte_stream stream)
void DiscoveryServer::ProcessPacket( sockaddr_in *clientAddr, sptr_byte_stream stream )
{
Log::Packet( stream->data, stream->data.size(), false );
auto head = stream->read_u32();
if (0x20 == head)
{
ProcessDiscoveryInitial(clientAddr, stream);
}
auto user = RealmUserManager::Get().GetUserByAddress(clientAddr);
if (nullptr == user)
if( 0x20 != stream->read_u32() )
{
return;
}
stream->set_position(0);
if (user->is_host)
{
ProcessHost(user, stream);
}
else
{
ProcessJoin(user, stream);
}
}
void DiscoveryServer::ProcessDiscoveryInitial( sockaddr_in *clientAddr, sptr_byte_stream stream )
{
auto encryptedBytes = stream->read_bytes( 0x20 );
auto decryptedBytes = RealmCrypt::decryptSymmetric(encryptedBytes);
auto decryptedBytes = RealmCrypt::decryptSymmetric( encryptedBytes );
std::wstring sessionId(16, L'\0');
std::memcpy(sessionId.data(), decryptedBytes.data(), 32);
std::wstring sessionId( 16, L'\0' );
std::memcpy( sessionId.data(), decryptedBytes.data(), 0x20 );
if( sessionId.empty() || sessionId.size() != 16 )
{
@@ -127,271 +104,40 @@ void DiscoveryServer::ProcessDiscoveryInitial( sockaddr_in *clientAddr, sptr_byt
return;
}
// Get the users remote IP and Port for discovery.
auto remoteIp = inet_ntoa( clientAddr->sin_addr );
auto remotePort = ntohs( clientAddr->sin_port );
auto user = RealmUserManager::Get().GetUser(sessionId);
if (user == nullptr)
Log::Debug( "Discovery Handshake from %s:%d", remoteIp, remotePort );
// Find the user associated with the session ID
auto user = RealmUserManager::Get().GetUser( sessionId );
if( user == nullptr )
{
Log::Error("User not found! [%S]", sessionId.c_str());
Log::Error( "User not found! [%S]", sessionId.c_str() );
return;
}
if (remoteIp != user->tcp->remote_ip)
if( remoteIp != user->sock->remote_ip )
{
Log::Error( "Discovery Handshake from invalid IP!" );
return;
}
user->udp->fd = m_socket;
user->udp->remote_addr = *clientAddr;
user->udp->remote_ip = remoteIp;
user->udp->remote_port = remotePort;
// Initialize our discovery information
user->m_discoveryAddr = remoteIp;
user->m_discoveryPort = remotePort;
user->discovery_addr = remoteIp;
user->discovery_port = remotePort;
//Log::Debug( "New Discovery Connection : [%S] %s:%d", sessionId.c_str(), remoteIp, remotePort );
}
void DiscoveryServer::ProcessHost(sptr_user user, sptr_byte_stream stream)
{
static int cmd = 0;
switch (user->discovery_state)
if( user->m_isHost )
{
case DiscoveryState::Initial_Connect:
{
if (false == GameSessionManager::Get().SetGameOpen(user))
{
Log::Error("Failed to set game open! [%S]", user->session_id.c_str());
return;
}
user->discovery_state = DiscoveryState::Host_Waiting;
user->udp->send(stream);
} break;
case DiscoveryState::Host_Waiting:
{
Send_Tmp(&user->udp->remote_addr);
} break;
GameSessionManager::Get().RequestOpen( user );
}
}
void DiscoveryServer::ProcessJoin(sptr_user user, sptr_byte_stream stream)
{
switch (user->discovery_state)
else
{
case DiscoveryState::Initial_Connect:
{
if (false == GameSessionManager::Get().JoinGame(user))
{
Log::Error("Failed to join game session! [%S]", user->session_id.c_str());
return;
}
GameSessionManager::Get().RequestJoin( user );
user->discovery_state = DiscoveryState::Negotiating;
} break;
case DiscoveryState::Negotiating:
{
// Initialize Realm Server Client
Send_07(user);
user->discovery_state = DiscoveryState::Joining;
} break;
case DiscoveryState::Joining:
{
if (false == Process_0C(user, stream))
{
Log::Error("Failed to parse user character data...");
return;
}
user->discovery_state = DiscoveryState::Joined;
} break;
case DiscoveryState::Joined:
{
Send_08(user);
} break;
// Theory: It may be possible to get the joining clients local address
// by tricking it with an 07 packet, then sending NotifyClientDiscovered.
// This would allow local network users to play together.
}
}
void DiscoveryServer::PrepareJoinGame(sptr_user user)
{
if (false == GameSessionManager::Get().JoinGame(user))
{
Log::Error("Failed to join game session! [%S]", user->session_id.c_str());
//return;
}
// Realm Server Client Initialize
Send_07( user );
user->discovery_state = DiscoveryState::Initial_Connect;
}
void DiscoveryServer::PrepareHostGame(sptr_user user)
{
if (false == GameSessionManager::Get().SetGameOpen(user))
{
return;
}
user->discovery_state = DiscoveryState::Host_Waiting;
}
void DiscoveryServer::Send_07( sptr_user user )
{
ByteStream reply;
reply.write_u32(0x07);
user->udp->send(reply);
}
void DiscoveryServer::Send_08( sptr_user user )
{
auto session = GameSessionManager::Get().FindGame(user->game_id);
if (nullptr == session)
{
Log::Error("Game session not found! [%d]", user->game_id);
return;
}
ByteStream reply;
{
reply.write_u32(0x08);
reply.write_sz_utf8("", 24);
reply.write_sz_utf8(std::to_string(session->m_userList.size()));
reply.write_sz_utf8("4");
for (auto& member : session->m_userList)
{
reply.write_sz_utf8(member->player_name, 24);
reply.write_sz_utf8(member->player_class, 24);
reply.write_u32(member->player_level);
}
}
for( auto &member : session->m_userList )
{
if (member->is_host)
continue;
if( !member->is_ready )
continue;
member->udp->send(reply);
}
}
void DiscoveryServer::Send_09( sptr_user user )
{
auto session = GameSessionManager::Get().FindGame( user->game_id );
if( nullptr == session)
{
Log::Error( "Game session not found! [%d]", user->game_id);
return;
}
ByteStream reply;
{
reply.write_u32(0x09);
reply.write_u32(session->m_userList.size());
for( auto &member : session->m_userList )
{
reply.write_sz_utf8(member->udp->remote_ip, 24);
reply.write_u32(member->udp->remote_port);
}
}
for( auto &member : session->m_userList )
{
member->udp->send(reply);
}
}
bool DiscoveryServer::Process_0C( sptr_user user, sptr_byte_stream stream)
{
if (0x0C != stream->read_u32())
{
return false;
}
//Log::Packet(stream->data, stream->data.size(), false);
if (nullptr == user)
{
return false;
}
user->local_addr = stream->read_utf8(24);
user->discovery_addr = stream->read_utf8(24);
user->discovery_port = stream->read_u32();
user->player_name = stream->read_utf8(24);
user->player_class = stream->read_utf8(24);
user->player_level = stream->read_u32();
Log::Debug("Join User Info : %s %s %d", user->player_name.c_str(), user->player_class.c_str(), user->player_level);
user->is_ready = true;
Send_08(user);
Send_0C(user);
return true;
}
void DiscoveryServer::Send_0C( sptr_user user )
{
auto session = GameSessionManager::Get().FindGame(user->game_id);
if (nullptr == session)
{
Log::Error("Game session not found! [%d]", user->game_id);
return;
}
auto host = session->GetHost();
ByteStream reply;
{
reply.write_u32(0x0C);
reply.write_sz_utf8(user->local_addr, 24);
reply.write_sz_utf8(user->discovery_addr, 24);
reply.write_u32(user->discovery_port);
reply.write_sz_utf8(host->player_name, 24);
reply.write_sz_utf8(host->player_class, 24);
reply.write_u32(host->player_level);
}
host->udp->send(reply);
}
void DiscoveryServer::Send_0D( sptr_user user )
{
// Send Reply to clientAddr
std::vector< uint8_t > reply = { 0x0D, 0x00, 0x00, 0x00 };
}
void DiscoveryServer::Send_Tmp(sockaddr_in* clientAddr)
{
ByteStream reply;
{
reply.write_u32(0x08);
reply.write_sz_utf8("193.149.164.146", 24);
reply.write_u32(10101);
}
sendto(m_socket, (char*)reply.data.data(), reply.data.size(), 0, (struct sockaddr*)clientAddr, sizeof(sockaddr_in));
}

View File

@@ -1,9 +1,5 @@
#pragma once
#include <memory>
#include <array>
#include <unordered_map>
class DiscoveryServer
{
private:
@@ -38,25 +34,6 @@ public:
private:
void ProcessPacket(sockaddr_in* clientAddr, sptr_byte_stream stream);
void ProcessDiscoveryInitial( sockaddr_in *clientAddr, sptr_byte_stream stream );
void ProcessHost(sptr_user user, sptr_byte_stream stream);
// Joinee Handlers
void ProcessJoin(sptr_user user, sptr_byte_stream stream);
void PrepareJoinGame( sptr_user user );
void PrepareHostGame( sptr_user user );
void Send_07( sptr_user user ); // Realm Initialize
void Send_08( sptr_user user ); // Member Info Update
void Send_09( sptr_user user ); // Member Address Info / Start Game
bool Process_0C( sptr_user user, sptr_byte_stream stream ); // Member Character Information
void Send_0C(sptr_user user); // Joining Member Character Info
void Send_0D(sptr_user user);
void Send_Tmp(sockaddr_in* clientAddr);
std::atomic< bool > m_running;
std::thread m_thread;