khanat-code-old/code/nelns/login_system/nel_launcher_qt/connection.cpp
2013-02-18 13:22:03 +01:00

268 lines
6.5 KiB
C++

#include "connection.h"
#include <iostream>
#include <nel/misc/ucstring.h>
#include <nel/misc/md5.h>
#include <nel/net/sock.h>
#include <nel/net/login_client.h>
#include "nel_launcher_dlg.h"
bool CNelLauncherConnection::connect()
{
std::string server = ConfigFile.getVar("StartupHost").asString(0);
if(m_Sock.connected())
return true;
try
{
// add the default port if no port in the cfg
if(server.find(':') == std::string::npos)
server+=":80";
m_Sock.connect(NLNET::CInetAddress(server));
if(!m_Sock.connected())
{
nlwarning("Can't connect to web server '%s'", server.c_str());
goto end;
}
}
catch(NLMISC::Exception &e)
{
nlwarning("Can't connect to web server '%s': %s", server.c_str(), e.what());
goto end;
}
return true;
end:
if(m_Sock.connected())
m_Sock.close ();
return false;
}
bool CNelLauncherConnection::send(const std::string &url)
{
nlassert(m_Sock.connected());
std::string buffer = "GET " + url + "\r\n";
uint32 size = buffer.size();
if(!url.empty())
{
if(m_Sock.send((uint8 *)buffer.c_str(), size, false) != NLNET::CSock::Ok)
{
nlwarning ("Can't send data to the server");
return false;
}
}
return true;
}
bool CNelLauncherConnection::receive(std::string &res)
{
nlassert(m_Sock.connected());
uint32 size;
res = "";
uint8 buf[1024];
//if(VerboseLog) nlinfo("Receiving");
while (true)
{
size = 1023;
if(m_Sock.receive((uint8*)buf, size, false) == NLNET::CSock::Ok)
{
//if(VerboseLog) nlinfo("Received OK %d bytes", size);
buf[1023] = '\0';
res += (char*)buf;
//nlinfo("block received '%s'", buf);
}
else
{
//if(VerboseLog) nlinfo("Received CLOSE %d bytes", size);
buf[size] = '\0';
res += (char*)buf;
//nlwarning ("server connection closed");
break;
}
}
// trim off whitespace.
res = NLMISC::trim(res);
nlinfo("all received '%s'", res.c_str());
return true;
}
std::string CNelLauncherConnection::checkLogin(const std::string &login, const std::string &password, const std::string &clientApp)
{
m_Shards.clear();
m_Login = m_Password = m_ClientApp = "";
if(ConfigFile.exists("UseDirectClient") && ConfigFile.getVar("UseDirectClient").asBool())
{
ucstring pwd = ucstring(password);
NLMISC::CHashKeyMD5 hk = NLMISC::getMD5((uint8*)pwd.c_str(), pwd.size());
std::string cpwd = hk.toString();
nlinfo("The crypted password is %s", cpwd.c_str());
std::string result = NLNET::CLoginClient::authenticate(ConfigFile.getVar("StartupHost").asString(), login, cpwd, clientApp);
if (!result.empty()) return result;
for(uint i = 0; i < NLNET::CLoginClient::ShardList.size(); ++i)
{
nldebug("Shard '%u' '%s' '%u'", NLNET::CLoginClient::ShardList[i].Id, NLNET::CLoginClient::ShardList[i].Name.toString().c_str(), NLNET::CLoginClient::ShardList[i].NbPlayers);
m_Shards.push_back(CShard("1", true,
NLNET::CLoginClient::ShardList[i].Id, NLNET::CLoginClient::ShardList[i].Name.toString(), NLNET::CLoginClient::ShardList[i].NbPlayers,
"1", "1"));
}
m_Login = login;
m_Password = password;
m_ClientApp = clientApp;
return "";
}
if(!connect())
return "Can't connect (error code 1)";
//if(VerboseLog) nlinfo("Connected");
if(!send(ConfigFile.getVar("StartupPage").asString()+"?login="+login+"&password="+password+"&clientApplication="+clientApp))
return "Can't send (error code 2)";
//if(VerboseLog) nlinfo("Sent request login check");
std::string res;
if(!receive(res))
return "Can't receive (error code 3)";
//if(VerboseLog) nlinfo("Received request login check");
if(res.empty())
return "Empty answer from server (error code 4)";
if(res[0] == '0')
{
// server returns an error
nlwarning("server error: %s", res.substr(2).c_str());
return res.substr(2);
}
else if(res[0] == '1')
{
// server returns ok, we have the list of shard
uint nbs = atoi(res.substr(2).c_str());
std::vector<std::string> lines;
NLMISC::explode(res, std::string("\n"), lines, true);
// if(VerboseLog)
// {
//nlinfo ("Exploded, with nl, %d res", lines.size());
/* for (uint i = 0; i < lines.size(); i++)
{
nlinfo (" > '%s'", lines[i].c_str());
}*/
// }
if(lines.size() != nbs+1)
{
nlwarning("bad shard lines number %d != %d", lines.size(), nbs+1);
nlwarning("'%s'", res.c_str());
return "bad lines numbers (error code 5)";
}
for(uint i = 1; i < lines.size(); i++)
{
std::vector<std::string> res;
NLMISC::explode(lines[i], std::string("|"), res);
// if(VerboseLog)
// {
// nlinfo ("Exploded with '%s', %d res", "|", res.size());
/* for (uint i = 0; i < res.size(); i++)
{
nlinfo (" > '%s'", res[i].c_str());
}*/
// }
if(res.size() != 7)
{
nlwarning("bad | numbers %d != %d", res.size(), 7);
nlwarning("'%s'", lines[i].c_str());
return "bad pipe numbers (error code 6)";
}
m_Shards.push_back(CShard(res[0], atoi(res[1].c_str())>0, atoi(res[2].c_str()), res[3], atoi(res[4].c_str()), res[5], res[6]));
}
}
else
{
// server returns ???
nlwarning("%s", res.c_str());
return res;
}
m_Login = login;
m_Password = password;
m_ClientApp = clientApp;
return "";
}
std::string CNelLauncherConnection::selectShard(uint32 shardId, std::string &cookie, std::string &addr)
{
cookie = addr = "";
if(ConfigFile.exists("UseDirectClient") && ConfigFile.getVar("UseDirectClient").asBool())
return NLNET::CLoginClient::wantToConnectToShard(shardId, addr, cookie);
if(!connect()) return "Can't connect (error code 7)";
if(m_Login.empty()) return "Empty Login (error code 8)";
if(m_Password.empty()) return "Empty Password (error code 9)";
if(m_ClientApp.empty()) return "Empty Client Application (error code 10)";
if(!send(ConfigFile.getVar("StartupPage").asString()+"?cmd=login&shardid="+NLMISC::toString(shardId)+"&login="+m_Login+"&password="+m_Password+"&clientApplication="+m_ClientApp))
return "Can't send (error code 11)";
std::string res;
if(!receive(res))
return "Can't receive (error code 12)";
if(res.empty())
return "Empty result (error code 13)";
if(res[0] == '0')
{
// server returns an error
nlwarning("server error: %s", res.substr(2).c_str());
return res.substr(2);
}
else if(res[0] == '1')
{
// server returns ok, we have the access
std::vector<std::string> line;
NLMISC::explode(res, std::string(" "), line, true);
if(line.size() != 2)
{
nlwarning("bad launch lines number %d != %d", line.size(), 2);
return "bad launch line number (error code 14)";
}
cookie = line[0].substr(2);
addr = line[1];
}
else
{
// server returns ???
nlwarning("%s", res.c_str());
return res;
}
return "";
}