mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2025-12-30 16:10:56 -06:00
1afe717563
Properly exit based on the path taken to shutdown master. Tested that shutting down through sigint or sigterm returns -1 Tested that a segfault exits the program properly Need to test that players who are trying to connect while master is shutting down are not able to spawn more child worlds.
396 lines
10 KiB
C++
396 lines
10 KiB
C++
#include "InstanceManager.h"
|
|
#include <string>
|
|
#include "Game.h"
|
|
#include "dServer.h"
|
|
#include "dLogger.h"
|
|
#include "dConfig.h"
|
|
#include "CDClientDatabase.h"
|
|
#include "CDClientManager.h"
|
|
#include "CDZoneTableTable.h"
|
|
#include "dMessageIdentifiers.h"
|
|
#include "MasterPackets.h"
|
|
#include "PacketUtils.h"
|
|
#include "BinaryPathFinder.h"
|
|
|
|
InstanceManager::InstanceManager(dLogger* logger, const std::string& externalIP) {
|
|
mLogger = logger;
|
|
mExternalIP = externalIP;
|
|
m_LastPort = std::atoi(Game::config->GetValue("world_port_start").c_str());
|
|
m_LastInstanceID = LWOINSTANCEID_INVALID;
|
|
}
|
|
|
|
InstanceManager::~InstanceManager() {
|
|
for (Instance* i : m_Instances) {
|
|
delete i;
|
|
i = nullptr;
|
|
}
|
|
}
|
|
|
|
Instance* InstanceManager::GetInstance(LWOMAPID mapID, bool isFriendTransfer, LWOCLONEID cloneID) {
|
|
mLogger->Log("InstanceManager", "Searching for an instance for mapID %i/%i", mapID, cloneID);
|
|
Instance* instance = FindInstance(mapID, isFriendTransfer, cloneID);
|
|
if (instance) return instance;
|
|
|
|
//TODO: Update this so that the IP is read from a configuration file instead
|
|
|
|
int softCap = 8;
|
|
int maxPlayers = 12;
|
|
|
|
if (mapID == 0) {
|
|
softCap = 999;
|
|
maxPlayers = softCap;
|
|
} else {
|
|
softCap = GetSoftCap(mapID);
|
|
maxPlayers = GetHardCap(mapID);
|
|
}
|
|
|
|
uint32_t port = GetFreePort();
|
|
instance = new Instance(mExternalIP, port, mapID, ++m_LastInstanceID, cloneID, softCap, maxPlayers);
|
|
|
|
//Start the actual process:
|
|
#ifdef _WIN32
|
|
std::string cmd = "start " + (BinaryPathFinder::GetBinaryDir() / "WorldServer.exe").string() + " -zone ";
|
|
#else
|
|
std::string cmd;
|
|
if (std::atoi(Game::config->GetValue("use_sudo_world").c_str())) {
|
|
cmd = "sudo " + (BinaryPathFinder::GetBinaryDir() / "WorldServer").string() + " -zone ";
|
|
} else {
|
|
cmd = (BinaryPathFinder::GetBinaryDir() / "WorldServer").string() + " -zone ";
|
|
}
|
|
#endif
|
|
|
|
cmd.append(std::to_string(mapID));
|
|
cmd.append(" -port ");
|
|
cmd.append(std::to_string(port));
|
|
cmd.append(" -instance ");
|
|
cmd.append(std::to_string(m_LastInstanceID));
|
|
cmd.append(" -maxclients ");
|
|
cmd.append(std::to_string(maxPlayers));
|
|
|
|
cmd.append(" -clone ");
|
|
cmd.append(std::to_string(cloneID));
|
|
|
|
#ifndef _WIN32
|
|
cmd.append("&"); //Sends our next process to the background on Linux
|
|
#endif
|
|
|
|
system(cmd.c_str());
|
|
|
|
m_Instances.push_back(instance);
|
|
|
|
if (instance) {
|
|
mLogger->Log("InstanceManager", "Created new instance: %i/%i/%i with min/max %i/%i", mapID, m_LastInstanceID, cloneID, softCap, maxPlayers);
|
|
return instance;
|
|
} else mLogger->Log("InstanceManager", "Failed to create a new instance!");
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
bool InstanceManager::IsPortInUse(uint32_t port) {
|
|
for (Instance* i : m_Instances) {
|
|
if (i && i->GetPort() == port) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
uint32_t InstanceManager::GetFreePort() {
|
|
uint32_t port = m_LastPort;
|
|
std::vector<uint32_t> usedPorts;
|
|
for (Instance* i : m_Instances) {
|
|
usedPorts.push_back(i->GetPort());
|
|
}
|
|
|
|
std::sort(usedPorts.begin(), usedPorts.end());
|
|
|
|
int portIdx = 0;
|
|
while (portIdx < usedPorts.size() && port == usedPorts[portIdx]) {
|
|
//increment by 3 since each instance uses 3 ports (instance, world-server, world-chat)
|
|
port += 3;
|
|
portIdx++;
|
|
}
|
|
|
|
return port;
|
|
}
|
|
|
|
void InstanceManager::AddPlayer(SystemAddress systemAddr, LWOMAPID mapID, LWOINSTANCEID instanceID) {
|
|
Instance* inst = FindInstance(mapID, instanceID);
|
|
if (inst) {
|
|
Player player;
|
|
player.addr = systemAddr;
|
|
player.id = 0; //TODO: Update this to include the LWOOBJID of the player's character.
|
|
inst->AddPlayer(player);
|
|
}
|
|
}
|
|
|
|
void InstanceManager::RemovePlayer(SystemAddress systemAddr, LWOMAPID mapID, LWOINSTANCEID instanceID) {
|
|
Instance* inst = FindInstance(mapID, instanceID);
|
|
if (inst) {
|
|
Player player;
|
|
player.addr = systemAddr;
|
|
player.id = 0; //TODO: Update this to include the LWOOBJID of the player's character.
|
|
inst->RemovePlayer(player);
|
|
}
|
|
}
|
|
|
|
std::vector<Instance*> InstanceManager::GetInstances() const {
|
|
return m_Instances;
|
|
}
|
|
|
|
void InstanceManager::AddInstance(Instance* instance) {
|
|
if (instance == nullptr) return;
|
|
|
|
m_Instances.push_back(instance);
|
|
}
|
|
|
|
void InstanceManager::RemoveInstance(Instance* instance) {
|
|
for (uint32_t i = 0; i < m_Instances.size(); ++i) {
|
|
if (m_Instances[i] == instance) {
|
|
instance->SetShutdownComplete(true);
|
|
|
|
if (!Game::shouldShutdown) RedirectPendingRequests(instance);
|
|
|
|
delete m_Instances[i];
|
|
|
|
m_Instances.erase(m_Instances.begin() + i);
|
|
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void InstanceManager::ReadyInstance(Instance* instance) {
|
|
instance->SetIsReady(true);
|
|
|
|
auto& pending = instance->GetPendingRequests();
|
|
|
|
for (const auto& request : pending) {
|
|
const auto& zoneId = instance->GetZoneID();
|
|
|
|
Game::logger->Log("InstanceManager", "Responding to pending request %llu -> %i (%i)", request, zoneId.GetMapID(), zoneId.GetCloneID());
|
|
|
|
MasterPackets::SendZoneTransferResponse(
|
|
Game::server,
|
|
request.sysAddr,
|
|
request.id,
|
|
request.mythranShift,
|
|
zoneId.GetMapID(),
|
|
zoneId.GetInstanceID(),
|
|
zoneId.GetCloneID(),
|
|
instance->GetIP(),
|
|
instance->GetPort()
|
|
);
|
|
}
|
|
|
|
pending.clear();
|
|
}
|
|
|
|
void InstanceManager::RequestAffirmation(Instance* instance, const PendingInstanceRequest& request) {
|
|
instance->GetPendingAffirmations().push_back(request);
|
|
|
|
CBITSTREAM;
|
|
|
|
PacketUtils::WriteHeader(bitStream, MASTER, MSG_MASTER_AFFIRM_TRANSFER_REQUEST);
|
|
|
|
bitStream.Write(request.id);
|
|
|
|
Game::server->Send(&bitStream, instance->GetSysAddr(), false);
|
|
|
|
Game::logger->Log("MasterServer", "Sent affirmation request %llu to %i/%i", request.id,
|
|
static_cast<int>(instance->GetZoneID().GetMapID()),
|
|
static_cast<int>(instance->GetZoneID().GetCloneID())
|
|
);
|
|
}
|
|
|
|
void InstanceManager::AffirmTransfer(Instance* instance, const uint64_t transferID) {
|
|
auto& pending = instance->GetPendingAffirmations();
|
|
|
|
for (auto i = 0u; i < pending.size(); ++i) {
|
|
const auto& request = pending[i];
|
|
|
|
if (request.id != transferID) continue;
|
|
|
|
const auto& zoneId = instance->GetZoneID();
|
|
|
|
MasterPackets::SendZoneTransferResponse(
|
|
Game::server,
|
|
request.sysAddr,
|
|
request.id,
|
|
request.mythranShift,
|
|
zoneId.GetMapID(),
|
|
zoneId.GetInstanceID(),
|
|
zoneId.GetCloneID(),
|
|
instance->GetIP(),
|
|
instance->GetPort()
|
|
);
|
|
|
|
pending.erase(pending.begin() + i);
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
void InstanceManager::RedirectPendingRequests(Instance* instance) {
|
|
const auto& zoneId = instance->GetZoneID();
|
|
|
|
for (const auto& request : instance->GetPendingAffirmations()) {
|
|
auto* in = Game::im->GetInstance(zoneId.GetMapID(), false, zoneId.GetCloneID());
|
|
|
|
if (!in->GetIsReady()) // Instance not ready, make a pending request
|
|
{
|
|
in->GetPendingRequests().push_back(request);
|
|
|
|
continue;
|
|
}
|
|
|
|
Game::im->RequestAffirmation(in, request);
|
|
}
|
|
}
|
|
|
|
Instance* InstanceManager::GetInstanceBySysAddr(SystemAddress& sysAddr) {
|
|
for (uint32_t i = 0; i < m_Instances.size(); ++i) {
|
|
if (m_Instances[i] && m_Instances[i]->GetSysAddr() == sysAddr) {
|
|
return m_Instances[i];
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
bool InstanceManager::IsInstanceFull(Instance* instance, bool isFriendTransfer) {
|
|
if (!isFriendTransfer && instance->GetSoftCap() > instance->GetCurrentClientCount())
|
|
return false;
|
|
else if (isFriendTransfer && instance->GetHardCap() > instance->GetCurrentClientCount())
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
Instance* InstanceManager::FindInstance(LWOMAPID mapID, bool isFriendTransfer, LWOCLONEID cloneId) {
|
|
for (Instance* i : m_Instances) {
|
|
if (i && i->GetMapID() == mapID && i->GetCloneID() == cloneId && !IsInstanceFull(i, isFriendTransfer) && !i->GetIsPrivate() && !i->GetShutdownComplete() && !i->GetIsShuttingDown()) {
|
|
return i;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
Instance* InstanceManager::FindInstance(LWOMAPID mapID, LWOINSTANCEID instanceID) {
|
|
for (Instance* i : m_Instances) {
|
|
if (i && i->GetMapID() == mapID && i->GetInstanceID() == instanceID && !i->GetIsPrivate() && !i->GetShutdownComplete() && !i->GetIsShuttingDown()) {
|
|
return i;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
Instance* InstanceManager::CreatePrivateInstance(LWOMAPID mapID, LWOCLONEID cloneID, const std::string& password) {
|
|
auto* instance = FindPrivateInstance(password);
|
|
|
|
if (instance != nullptr) {
|
|
return instance;
|
|
}
|
|
|
|
int maxPlayers = 999;
|
|
|
|
uint32_t port = GetFreePort();
|
|
instance = new Instance(mExternalIP, port, mapID, ++m_LastInstanceID, cloneID, maxPlayers, maxPlayers, true, password);
|
|
|
|
//Start the actual process:
|
|
std::string cmd = "start " + (BinaryPathFinder::GetBinaryDir() / "WorldServer").string() + " -zone ";
|
|
|
|
#ifndef _WIN32
|
|
cmd = (BinaryPathFinder::GetBinaryDir() / "WorldServer").string() + " -zone ";
|
|
#endif
|
|
|
|
cmd.append(std::to_string(mapID));
|
|
cmd.append(" -port ");
|
|
cmd.append(std::to_string(port));
|
|
cmd.append(" -instance ");
|
|
cmd.append(std::to_string(m_LastInstanceID));
|
|
cmd.append(" -maxclients ");
|
|
cmd.append(std::to_string(maxPlayers));
|
|
|
|
cmd.append(" -clone ");
|
|
cmd.append(std::to_string(cloneID));
|
|
|
|
#ifndef WIN32
|
|
cmd.append("&"); //Sends our next process to the background on Linux
|
|
#endif
|
|
|
|
system(cmd.c_str());
|
|
|
|
m_Instances.push_back(instance);
|
|
|
|
if (instance) return instance;
|
|
else mLogger->Log("InstanceManager", "Failed to create a new instance!");
|
|
|
|
return instance;
|
|
}
|
|
|
|
Instance* InstanceManager::FindPrivateInstance(const std::string& password) {
|
|
for (auto* instance : m_Instances) {
|
|
if (!instance) continue;
|
|
|
|
if (!instance->GetIsPrivate()) {
|
|
continue;
|
|
}
|
|
|
|
mLogger->Log("InstanceManager", "Password: %s == %s => %d", password.c_str(), instance->GetPassword().c_str(), password == instance->GetPassword());
|
|
|
|
if (instance->GetPassword() == password) {
|
|
return instance;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
int InstanceManager::GetSoftCap(LWOMAPID mapID) {
|
|
CDZoneTableTable* zoneTable = CDClientManager::Instance()->GetTable<CDZoneTableTable>("ZoneTable");
|
|
if (zoneTable) {
|
|
const CDZoneTable* zone = zoneTable->Query(mapID);
|
|
|
|
if (zone != nullptr) {
|
|
return zone->population_soft_cap;
|
|
}
|
|
}
|
|
|
|
return 8;
|
|
}
|
|
|
|
int InstanceManager::GetHardCap(LWOMAPID mapID) {
|
|
CDZoneTableTable* zoneTable = CDClientManager::Instance()->GetTable<CDZoneTableTable>("ZoneTable");
|
|
if (zoneTable) {
|
|
const CDZoneTable* zone = zoneTable->Query(mapID);
|
|
|
|
if (zone != nullptr) {
|
|
return zone->population_hard_cap;
|
|
}
|
|
}
|
|
|
|
return 12;
|
|
}
|
|
|
|
void Instance::SetShutdownComplete(const bool value) {
|
|
m_Shutdown = value;
|
|
}
|
|
|
|
bool Instance::GetShutdownComplete() const {
|
|
return m_Shutdown;
|
|
}
|
|
|
|
void Instance::Shutdown() {
|
|
CBITSTREAM;
|
|
|
|
PacketUtils::WriteHeader(bitStream, MASTER, MSG_MASTER_SHUTDOWN);
|
|
|
|
Game::server->Send(&bitStream, this->m_SysAddr, false);
|
|
|
|
Game::logger->Log("Instance", "Triggered world shutdown for zone/clone/instance %i/%i/%i", GetMapID(), GetCloneID(), GetInstanceID());
|
|
}
|