656 lines
17 KiB
C++
656 lines
17 KiB
C++
|
// NeLNS - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
|
||
|
|
// Copyright (C) 2010 Winch Gate Property Limited
|
||
|
|
//
|
||
|
|
// This program is free software: you can redistribute it and/or modify
|
||
|
|
// it under the terms of the GNU Affero General Public License as
|
||
|
|
// published by the Free Software Foundation, either version 3 of the
|
||
|
|
// License, or (at your option) any later version.
|
||
|
|
//
|
||
|
|
// This program is distributed in the hope that it will be useful,
|
||
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
// GNU Affero General Public License for more details.
|
||
|
|
//
|
||
|
|
// You should have received a copy of the GNU Affero General Public License
|
||
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
#include "nel/misc/types_nl.h"
|
||
|
|
|
||
|
|
#include <cstdio>
|
||
|
|
#include <ctype.h>
|
||
|
|
#include <cmath>
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <map>
|
||
|
|
|
||
|
|
#include "nel/misc/debug.h"
|
||
|
|
#include "nel/misc/config_file.h"
|
||
|
|
#include "nel/misc/displayer.h"
|
||
|
|
#include "nel/misc/log.h"
|
||
|
|
|
||
|
|
#include "nel/net/service.h"
|
||
|
|
#include "nel/net/login_cookie.h"
|
||
|
|
|
||
|
|
#include "login_service.h"
|
||
|
|
#include "mysql_helper.h"
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Namespaces
|
||
|
|
//
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
using namespace NLMISC;
|
||
|
|
using namespace NLNET;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Variables
|
||
|
|
//
|
||
|
|
|
||
|
|
static uint RecordNbPlayers = 0;
|
||
|
|
uint NbPlayers = 0;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Functions
|
||
|
|
//
|
||
|
|
|
||
|
|
void refuseShard (TServiceId sid, const char *format, ...)
|
||
|
|
{
|
||
|
|
string reason;
|
||
|
|
NLMISC_CONVERT_VARGS (reason, format, NLMISC::MaxCStringSize);
|
||
|
|
nlwarning(reason.c_str ());
|
||
|
|
CMessage msgout("FAILED");
|
||
|
|
msgout.serial (reason);
|
||
|
|
CUnifiedNetwork::getInstance ()->send (sid, msgout);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void cbWSConnection (const std::string &serviceName, TServiceId sid, void *arg)
|
||
|
|
{
|
||
|
|
TSockId from;
|
||
|
|
CCallbackNetBase *cnb = CUnifiedNetwork::getInstance ()->getNetBase (sid, from);
|
||
|
|
const CInetAddress &ia = cnb->hostAddress (from);
|
||
|
|
|
||
|
|
nldebug("new potential shard: %s", ia.asString ().c_str ());
|
||
|
|
|
||
|
|
// if we accept external shard, don't need to check if address is valid
|
||
|
|
if(IService::getInstance ()->ConfigFile.getVar("AcceptExternalShards").asInt () == 1)
|
||
|
|
return;
|
||
|
|
|
||
|
|
string reason;
|
||
|
|
CMysqlResult result;
|
||
|
|
MYSQL_ROW row;
|
||
|
|
sint32 nbrow;
|
||
|
|
|
||
|
|
string query = "select * from shard where WSAddr='"+ia.ipAddress()+"'";
|
||
|
|
reason = sqlQuery(query, nbrow, row, result);
|
||
|
|
if (!reason.empty())
|
||
|
|
{
|
||
|
|
refuseShard (sid, "mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (nbrow == 0)
|
||
|
|
{
|
||
|
|
// if we are here, it s that the shard have not a valid wsaddr in the database
|
||
|
|
// we can't accept unknown shard
|
||
|
|
refuseShard (sid, "Bad shard identification, the shard (WSAddr %s) is not in the database and can't be added", ia.ipAddress ().c_str ());
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void cbWSDisconnection (const std::string &serviceName, TServiceId sid, void *arg)
|
||
|
|
{
|
||
|
|
TSockId from;
|
||
|
|
CCallbackNetBase *cnb = CUnifiedNetwork::getInstance ()->getNetBase (sid, from);
|
||
|
|
const CInetAddress &ia = cnb->hostAddress (from);
|
||
|
|
|
||
|
|
nldebug("shard disconnection: %s", ia.asString ().c_str ());
|
||
|
|
|
||
|
|
for (uint32 i = 0; i < Shards.size (); i++)
|
||
|
|
{
|
||
|
|
if (Shards[i].SId == sid)
|
||
|
|
{
|
||
|
|
// shard disconnected
|
||
|
|
nlinfo("ShardId %d with IP '%s' is offline!", Shards[i].ShardId, ia.asString ().c_str());
|
||
|
|
nlinfo("*** ShardId %3d NbPlayers %3d -> %3d", Shards[i].ShardId, Shards[i].NbPlayers, 0);
|
||
|
|
|
||
|
|
string query = "update shard set Online=0, NbPlayers=NbPlayers-"+toString(Shards[i].NbPlayers)+" where ShardId="+toString(Shards[i].ShardId);
|
||
|
|
sint ret = mysql_query (DatabaseConnection, query.c_str ());
|
||
|
|
if (ret != 0)
|
||
|
|
{
|
||
|
|
nlwarning ("mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
}
|
||
|
|
|
||
|
|
NbPlayers -= Shards[i].NbPlayers;
|
||
|
|
|
||
|
|
Shards[i].NbPlayers = 0;
|
||
|
|
|
||
|
|
// put users connected on this shard offline
|
||
|
|
|
||
|
|
query = "update user set State='Offline', ShardId=-1 where ShardId="+toString(Shards[i].ShardId);
|
||
|
|
ret = mysql_query (DatabaseConnection, query.c_str ());
|
||
|
|
if (ret != 0)
|
||
|
|
{
|
||
|
|
nlwarning ("mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
}
|
||
|
|
|
||
|
|
Shards.erase (Shards.begin () + i);
|
||
|
|
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
nlwarning("Shard %s goes offline but wasn't online!", ia.asString ().c_str ());
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Shard accepted the new user, so warn the user that he could connect to the shard now */
|
||
|
|
/*void cbClientShardAcceptedTheUser (CMessage &msgin, TSockId from, CCallbackNetBase &netbase)
|
||
|
|
{
|
||
|
|
uint8 res;
|
||
|
|
string IP;
|
||
|
|
uint32 Key, Id;
|
||
|
|
msgin.serial (IP);
|
||
|
|
msgin.serial (Key);
|
||
|
|
msgin.serial (Id);
|
||
|
|
msgin.serial (res);
|
||
|
|
|
||
|
|
CMessage msgout (netbase.getSIDA (), "ACC");
|
||
|
|
msgout.serial (res);
|
||
|
|
|
||
|
|
if(res)
|
||
|
|
{
|
||
|
|
// the shard accept the user
|
||
|
|
msgout.serial (IP);
|
||
|
|
msgout.serial (Key);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// the shard don't want him!
|
||
|
|
string reason;
|
||
|
|
msgin.serial (reason);
|
||
|
|
msgout.serial (reason);
|
||
|
|
}
|
||
|
|
|
||
|
|
// find the user
|
||
|
|
for (vector<CUser>::iterator it = Users.begin (); it != Users.end (); it++)
|
||
|
|
{
|
||
|
|
if ((*it).Authorized && (*it).Key == Key)
|
||
|
|
{
|
||
|
|
// send the answer to the user
|
||
|
|
netbase.send (msgout, (*it).SockId);
|
||
|
|
|
||
|
|
(*it).Authorized = false;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
|
||
|
|
|
||
|
|
/*
|
||
|
|
void cbShardComesIn (CMessage &msgin, TSockId from, CCallbackNetBase &netbase)
|
||
|
|
{
|
||
|
|
const CInetAddress &ia = netbase.hostAddress (from);
|
||
|
|
|
||
|
|
// at this time, it could be a new shard or a ne client.
|
||
|
|
|
||
|
|
nldebug("new potential shard: %s", ia.asString ().c_str ());
|
||
|
|
|
||
|
|
// first, check if it an authorized shard
|
||
|
|
for (sint32 i = 0; i < (sint32) Shards.size (); i++)
|
||
|
|
{
|
||
|
|
if (Shards[i].Address.ipAddress () == ia.ipAddress ())
|
||
|
|
{
|
||
|
|
if (Shards[i].Online)
|
||
|
|
{
|
||
|
|
nlwarning("Shard with ip '%s' is already online! Disconnect the new one", ia.asString().c_str ());
|
||
|
|
netbase.disconnect (from);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// new shard connected
|
||
|
|
Shards[i].Address = ia;
|
||
|
|
Shards[i].Online = true;
|
||
|
|
Shards[i].SockId = from;
|
||
|
|
nlinfo("Shard with ip '%s' is online!", Shards[i].Address.asString().c_str ());
|
||
|
|
}
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#if ACCEPT_EXTERNAL_SHARD
|
||
|
|
// New externam shard connected, add it in the file
|
||
|
|
Shards.push_back (CShard(ia));
|
||
|
|
sint32 pos = Shards.size()-1;
|
||
|
|
Shards[pos].Online = true;
|
||
|
|
Shards[pos].SockId = from;
|
||
|
|
nlinfo("External shard with ip '%s' is online!", Shards[pos].Address.asString().c_str ());
|
||
|
|
writeConfigFile ();
|
||
|
|
#else
|
||
|
|
nlwarning("It's not a authorized shard, disconnect it");
|
||
|
|
netbase.close (from);
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
|
||
|
|
//
|
||
|
|
static void cbWSIdentification (CMessage &msgin, const std::string &serviceName, TServiceId sid)
|
||
|
|
{
|
||
|
|
TSockId from;
|
||
|
|
CCallbackNetBase *cnb = CUnifiedNetwork::getInstance ()->getNetBase (sid, from);
|
||
|
|
const CInetAddress &ia = cnb->hostAddress (from);
|
||
|
|
|
||
|
|
sint32 shardId;
|
||
|
|
msgin.serial(shardId);
|
||
|
|
string application;
|
||
|
|
try {
|
||
|
|
msgin.serial(application);
|
||
|
|
} catch (Exception &) { }
|
||
|
|
nldebug("shard identification, It says to be ShardId %d, let's check that!", shardId);
|
||
|
|
|
||
|
|
string reason;
|
||
|
|
|
||
|
|
CMysqlResult result;
|
||
|
|
MYSQL_ROW row;
|
||
|
|
sint32 nbrow;
|
||
|
|
string query = "select * from shard where ShardId="+toString(shardId);
|
||
|
|
reason = sqlQuery(query, nbrow, row, result);
|
||
|
|
if (!reason.empty())
|
||
|
|
{
|
||
|
|
refuseShard (sid, "mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (nbrow == 0)
|
||
|
|
{
|
||
|
|
if(IService::getInstance ()->ConfigFile.getVar("AcceptExternalShards").asInt () == 1)
|
||
|
|
{
|
||
|
|
// we accept new shard, add it
|
||
|
|
query = "insert into shard (ShardId, WsAddr, Online, Name, ClientApplication) values ("+toString(shardId)+", '"+ia.ipAddress ()+"', 1, '"+ia.ipAddress ()+"', '"+application+"')";
|
||
|
|
reason = sqlQuery(query, nbrow, row, result);
|
||
|
|
if (!reason.empty())
|
||
|
|
{
|
||
|
|
refuseShard (sid, "mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
nlinfo("The ShardId %d with ip '%s' was inserted in the database and is online!", shardId, ia.ipAddress ().c_str ());
|
||
|
|
Shards.push_back (CShard (shardId, sid));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// can't accept new shard
|
||
|
|
refuseShard (sid, "Bad shard identification, The shard %d is not in the database and can't be added", shardId);
|
||
|
|
}
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
else if (nbrow == 1)
|
||
|
|
{
|
||
|
|
// check that the ip is ok
|
||
|
|
CInetAddress iadb;
|
||
|
|
iadb.setNameAndPort (row[1]);
|
||
|
|
nlinfo ("check %s with %s (%s)", ia.ipAddress ().c_str(), iadb.ipAddress().c_str(), row[1]);
|
||
|
|
if (ia.ipAddress () != iadb.ipAddress())
|
||
|
|
{
|
||
|
|
// good shard id but from a bad computer address
|
||
|
|
refuseShard (sid, "Bad shard identification, ShardId %d should come from '%s' and come from '%s'", shardId, row[1], ia.ipAddress ().c_str ());
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
sint32 s = findShard (shardId);
|
||
|
|
if (s != -1)
|
||
|
|
{
|
||
|
|
// the shard is already online, disconnect the old one and set the new one
|
||
|
|
refuseShard (Shards[s].SId, "A new shard connects with the same IP/ShardId, you was replaced");
|
||
|
|
|
||
|
|
Shards[s].SId = sid;
|
||
|
|
Shards[s].ShardId = shardId;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
string query = "update shard set Online=1 where ShardId="+toString(shardId);
|
||
|
|
sint ret = mysql_query (DatabaseConnection, query.c_str ());
|
||
|
|
if (ret != 0)
|
||
|
|
{
|
||
|
|
refuseShard (sid, "mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
Shards.push_back (CShard (shardId, sid));
|
||
|
|
}
|
||
|
|
|
||
|
|
// ok, the shard is identified correctly
|
||
|
|
nlinfo("ShardId %d with ip '%s' is online!", shardId, ia.ipAddress ().c_str ());
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
refuseShard (sid, "mysql problem, There's more than 1 shard with the shardId %d in the database", shardId);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
nlstop;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void cbWSClientConnected (CMessage &msgin, const std::string &serviceName, TServiceId sid)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// S16: Receive "CC" message from WS
|
||
|
|
//
|
||
|
|
|
||
|
|
// a WS tells me that a player is connected or disconnected
|
||
|
|
// find the user
|
||
|
|
uint32 Id;
|
||
|
|
uint8 con;
|
||
|
|
msgin.serial (Id);
|
||
|
|
msgin.serial (con); // con=1 means a client is connected on the shard, 0 means a client disconnected
|
||
|
|
|
||
|
|
if(con)
|
||
|
|
nlinfo ("Received a validation that a client is connected on the frontend");
|
||
|
|
else
|
||
|
|
nlinfo ("Received a validation that a client is disconnected on the frontend");
|
||
|
|
|
||
|
|
string reason;
|
||
|
|
CMysqlResult result;
|
||
|
|
MYSQL_ROW row;
|
||
|
|
sint32 nbrow;
|
||
|
|
|
||
|
|
string query = "select * from user where UId="+toString(Id);
|
||
|
|
reason = sqlQuery(query, nbrow, row, result);
|
||
|
|
if(!reason.empty()) return;
|
||
|
|
|
||
|
|
if(nbrow == 0)
|
||
|
|
{
|
||
|
|
nlwarning ("Id %d doesn't exist", Id);
|
||
|
|
Output->displayNL ("###: %3d UId doesn't exist", Id);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
else if (nbrow > 1)
|
||
|
|
{
|
||
|
|
nlerror ("Id %d have more than one entry!!!", Id);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// row[4] = State
|
||
|
|
if (con == 1 && string(row[4]) != string("Waiting"))
|
||
|
|
{
|
||
|
|
nlwarning("Id %d is not waiting", Id);
|
||
|
|
Output->displayNL("###: %3d User isn't waiting, his state is '%s'", Id, row[4]);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
else if (con == 0 && string(row[4]) != string ("Online"))
|
||
|
|
{
|
||
|
|
nlwarning ("Id %d wasn't connected on a shard", Id);
|
||
|
|
Output->displayNL ("###: %3d User wasn't connected on a shard, his state is '%s'", Id, row[4]);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
sint ShardPos = findShardWithSId (sid);
|
||
|
|
|
||
|
|
if (con == 1)
|
||
|
|
{
|
||
|
|
// new client on the shard
|
||
|
|
|
||
|
|
|
||
|
|
string query = "update user set State='Online', ShardId="+toString(Shards[ShardPos].ShardId)+" where UId="+toString(Id);
|
||
|
|
string rea = sqlQuery(query);
|
||
|
|
if(!rea.empty()) return;
|
||
|
|
|
||
|
|
if (ShardPos != -1)
|
||
|
|
{
|
||
|
|
nlinfo("*** ShardId %3d NbPlayers %3d -> %3d", Shards[ShardPos].ShardId, Shards[ShardPos].NbPlayers, Shards[ShardPos].NbPlayers+1);
|
||
|
|
Shards[ShardPos].NbPlayers++;
|
||
|
|
|
||
|
|
string query = "update shard set NbPlayers=NbPlayers+1 where ShardId="+toString(Shards[ShardPos].ShardId);
|
||
|
|
string rea = sqlQuery(query);
|
||
|
|
if(!rea.empty()) return;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
nlwarning ("user connected shard isn't in the shard list");
|
||
|
|
|
||
|
|
nldebug ("Id %d is connected on the shard", Id);
|
||
|
|
Output->displayNL ("###: %3d User connected to the shard (%d)", Id, Shards[ShardPos].ShardId);
|
||
|
|
|
||
|
|
NbPlayers++;
|
||
|
|
if (NbPlayers > RecordNbPlayers)
|
||
|
|
{
|
||
|
|
RecordNbPlayers = NbPlayers;
|
||
|
|
beep (2000, 1, 100, 0);
|
||
|
|
nlwarning("New player number record!!! %d players online on all shards", RecordNbPlayers);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// client removed from the shard (true is for potential other client with the same id that wait for a connection)
|
||
|
|
// disconnectClient (Users[pos], true, false);
|
||
|
|
|
||
|
|
string query = "update user set State='Offline', ShardId=-1 where UId="+toString(Id);
|
||
|
|
string rea = sqlQuery(query);
|
||
|
|
if(!rea.empty()) return;
|
||
|
|
|
||
|
|
if (ShardPos != -1)
|
||
|
|
{
|
||
|
|
nlinfo("*** ShardId %3d NbPlayers %3d -> %3d", Shards[ShardPos].ShardId, Shards[ShardPos].NbPlayers, Shards[ShardPos].NbPlayers-1);
|
||
|
|
Shards[ShardPos].NbPlayers--;
|
||
|
|
|
||
|
|
string query = "update shard set NbPlayers=NbPlayers-1 where ShardId="+toString(Shards[ShardPos].ShardId);
|
||
|
|
string rea = sqlQuery(query);
|
||
|
|
if(!rea.empty()) return;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
nlwarning ("user disconnected shard isn't in the shard list");
|
||
|
|
|
||
|
|
nldebug ("Id %d is disconnected from the shard", Id);
|
||
|
|
Output->displayNL ("###: %3d User disconnected from the shard (%d)", Id, Shards[ShardPos].ShardId);
|
||
|
|
|
||
|
|
NbPlayers--;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static void cbWSReportFSState(CMessage &msgin, const std::string &serviceName, TServiceId sid)
|
||
|
|
{
|
||
|
|
sint shardPos = findShardWithSId (sid);
|
||
|
|
|
||
|
|
if (shardPos == -1)
|
||
|
|
{
|
||
|
|
nlwarning ("unknown WS %d reported state of a fs", sid.get());
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
CShard& shard = Shards[shardPos];
|
||
|
|
|
||
|
|
TServiceId FSSId;
|
||
|
|
bool alive;
|
||
|
|
bool patching;
|
||
|
|
std::string patchURI;
|
||
|
|
|
||
|
|
msgin.serial(FSSId);
|
||
|
|
msgin.serial(alive);
|
||
|
|
msgin.serial(patching);
|
||
|
|
msgin.serial(patchURI);
|
||
|
|
|
||
|
|
if (!alive)
|
||
|
|
{
|
||
|
|
nlinfo("Shard %d frontend %d reported as offline", shard.ShardId, FSSId.get());
|
||
|
|
std::vector<CFrontEnd>::iterator itfs;
|
||
|
|
for (itfs=shard.FrontEnds.begin(); itfs!=shard.FrontEnds.end(); ++itfs)
|
||
|
|
{
|
||
|
|
if ((*itfs).SId == FSSId)
|
||
|
|
{
|
||
|
|
shard.FrontEnds.erase(itfs);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
CFrontEnd* updateFS = NULL;
|
||
|
|
std::vector<CFrontEnd>::iterator itfs;
|
||
|
|
for (itfs=shard.FrontEnds.begin(); itfs!=shard.FrontEnds.end(); ++itfs)
|
||
|
|
{
|
||
|
|
if ((*itfs).SId == FSSId)
|
||
|
|
{
|
||
|
|
// update FS state
|
||
|
|
CFrontEnd& fs = (*itfs);
|
||
|
|
fs.Patching = patching;
|
||
|
|
fs.PatchURI = patchURI;
|
||
|
|
|
||
|
|
updateFS = &fs;
|
||
|
|
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (itfs == shard.FrontEnds.end())
|
||
|
|
{
|
||
|
|
nlinfo("Shard %d frontend %d reported as online", shard.ShardId, FSSId.get());
|
||
|
|
// unknown fs, create new entry
|
||
|
|
shard.FrontEnds.push_back(CFrontEnd(FSSId, patching, patchURI));
|
||
|
|
|
||
|
|
updateFS = &(shard.FrontEnds.back());
|
||
|
|
}
|
||
|
|
|
||
|
|
if (updateFS != NULL)
|
||
|
|
{
|
||
|
|
nlinfo("Shard %d frontend %d status updated: patching=%s patchURI=%s", shard.ShardId, FSSId.get(), (updateFS->Patching ? "yes" : "no"), updateFS->PatchURI.c_str());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// update DynPatchURLS in database
|
||
|
|
std::string dynPatchURL;
|
||
|
|
uint i;
|
||
|
|
for (i=0; i<shard.FrontEnds.size(); ++i)
|
||
|
|
{
|
||
|
|
if (shard.FrontEnds[i].Patching)
|
||
|
|
{
|
||
|
|
if (!dynPatchURL.empty())
|
||
|
|
dynPatchURL += ' ';
|
||
|
|
dynPatchURL += shard.FrontEnds[i].PatchURI;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
string query = "UPDATE shard SET DynPatchURL='"+dynPatchURL+"' WHERE ShardId='"+toString(shard.ShardId)+"'";
|
||
|
|
sint ret = mysql_query (DatabaseConnection, query.c_str ());
|
||
|
|
if (ret != 0)
|
||
|
|
{
|
||
|
|
nlwarning ("mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void cbWSReportNoPatch(CMessage &msgin, const std::string &serviceName, TServiceId sid)
|
||
|
|
{
|
||
|
|
sint shardPos = findShardWithSId (sid);
|
||
|
|
|
||
|
|
if (shardPos == -1)
|
||
|
|
{
|
||
|
|
nlwarning ("unknown WS %d reported state of a fs", sid.get());
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
CShard& shard = Shards[shardPos];
|
||
|
|
|
||
|
|
uint i;
|
||
|
|
for (i=0; i<shard.FrontEnds.size(); ++i)
|
||
|
|
{
|
||
|
|
shard.FrontEnds[i].Patching = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
string query = "UPDATE shard SET DynPatchURL='' WHERE ShardId='"+toString(shard.ShardId)+"'";
|
||
|
|
sint ret = mysql_query (DatabaseConnection, query.c_str ());
|
||
|
|
if (ret != 0)
|
||
|
|
{
|
||
|
|
nlwarning ("mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void cbWSSetShardOpen(CMessage &msgin, const std::string &serviceName, TServiceId sid)
|
||
|
|
{
|
||
|
|
sint shardPos = findShardWithSId (sid);
|
||
|
|
|
||
|
|
if (shardPos == -1)
|
||
|
|
{
|
||
|
|
nlwarning ("unknown WS %d reported shard open state", sid.get());
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
uint8 shardOpenState;
|
||
|
|
msgin.serial(shardOpenState);
|
||
|
|
|
||
|
|
CShard& shard = Shards[shardPos];
|
||
|
|
|
||
|
|
string query = toString("UPDATE shard SET Online='%d' WHERE ShardId='%d'", shardOpenState+1, shard.ShardId);
|
||
|
|
sint ret = mysql_query (DatabaseConnection, query.c_str ());
|
||
|
|
if (ret != 0)
|
||
|
|
{
|
||
|
|
nlwarning ("mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static const TUnifiedCallbackItem WSCallbackArray[] =
|
||
|
|
{
|
||
|
|
{ "CC", cbWSClientConnected },
|
||
|
|
{ "WS_IDENT", cbWSIdentification },
|
||
|
|
|
||
|
|
{ "REPORT_FS_STATE", cbWSReportFSState },
|
||
|
|
{ "REPORT_NO_PATCH", cbWSReportNoPatch },
|
||
|
|
{ "SET_SHARD_OPEN", cbWSSetShardOpen },
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Functions
|
||
|
|
//
|
||
|
|
|
||
|
|
void connectionWSInit ()
|
||
|
|
{
|
||
|
|
CUnifiedNetwork::getInstance ()->addCallbackArray (WSCallbackArray, sizeof(WSCallbackArray)/sizeof(WSCallbackArray[0]));
|
||
|
|
|
||
|
|
CUnifiedNetwork::getInstance ()->setServiceUpCallback ("WS", cbWSConnection);
|
||
|
|
CUnifiedNetwork::getInstance ()->setServiceDownCallback ("WS", cbWSDisconnection);
|
||
|
|
}
|
||
|
|
|
||
|
|
void connectionWSUpdate ()
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
void connectionWSRelease ()
|
||
|
|
{
|
||
|
|
nlinfo ("I'm going down, clean the database");
|
||
|
|
|
||
|
|
|
||
|
|
while (!Shards.empty())
|
||
|
|
{
|
||
|
|
cbWSDisconnection ("", Shards[0].SId, NULL);
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
// we remove all shards online from my list
|
||
|
|
for (uint32 i = 0; i < Shards.size (); i++)
|
||
|
|
{
|
||
|
|
TSockId from;
|
||
|
|
CCallbackNetBase *cnb = CUnifiedNetwork::getInstance ()->getNetBase (Shards[i].SId, from);
|
||
|
|
const CInetAddress &ia = cnb->hostAddress (from);
|
||
|
|
|
||
|
|
// shard disconnected
|
||
|
|
nlinfo("Set ShardId %d with IP '%s' offline in the database and set %d players to offline", Shards[i].ShardId, ia.asString ().c_str());
|
||
|
|
|
||
|
|
string query = "update shard set Online=Online-1, NbPlayers=NbPlayers-"+toString(Shards[i].NbPlayers)+" where ShardId="+toString(Shards[i].ShardId);
|
||
|
|
sint ret = mysql_query (DatabaseConnection, query.c_str ());
|
||
|
|
if (ret != 0)
|
||
|
|
{
|
||
|
|
nlwarning ("mysql_query (%s) failed: %s", query.c_str (), mysql_error(DatabaseConnection));
|
||
|
|
}
|
||
|
|
|
||
|
|
// put users connected on this shard offline
|
||
|
|
}
|
||
|
|
Shards.clear ();
|
||
|
|
*/
|
||
|
|
|
||
|
|
}
|