/*
* Hedgewars, a free turn based strategy game
* Copyright (c) 2006-2008 Igor Ulyanov <iulyanov@gmail.com>
* Copyright (c) 2008-2011 Andrey Korotaev <unC0Rr@gmail.com>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; version 2 of the License
*
* 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
*/
#include <QDebug>
#include <QInputDialog>
#include <QCryptographicHash>
#include "hwconsts.h"
#include "newnetclient.h"
#include "proto.h"
#include "game.h"
char delimeter='\n';
HWNewNet::HWNewNet() :
isChief(false),
m_game_connected(false),
loginStep(0),
netClientState(Disconnected)
{
// socket stuff
connect(&NetSocket, SIGNAL(readyRead()), this, SLOT(ClientRead()));
connect(&NetSocket, SIGNAL(connected()), this, SLOT(OnConnect()));
connect(&NetSocket, SIGNAL(disconnected()), this, SLOT(OnDisconnect()));
connect(&NetSocket, SIGNAL(error(QAbstractSocket::SocketError)), this,
SLOT(displayError(QAbstractSocket::SocketError)));
}
HWNewNet::~HWNewNet()
{
if (m_game_connected)
{
RawSendNet(QString("QUIT%1%2").arg(delimeter).arg("User quit"));
emit disconnected(tr("User quit"));
}
NetSocket.flush();
}
void HWNewNet::Connect(const QString & hostName, quint16 port, const QString & nick)
{
netClientState = Connecting;
mynick = nick;
myhost = hostName + QString(":%1").arg(port);
NetSocket.connectToHost(hostName, port);
}
void HWNewNet::Disconnect()
{
if (m_game_connected)
RawSendNet(QString("QUIT%1%2").arg(delimeter).arg("User quit"));
m_game_connected = false;
NetSocket.disconnectFromHost();
}
void HWNewNet::CreateRoom(const QString & room)
{
if(netClientState != InLobby)
{
qWarning("Illegal try to create room!");
return;
}
myroom = room;
RawSendNet(QString("CREATE_ROOM%1%2").arg(delimeter).arg(room));
isChief = true;
}
void HWNewNet::JoinRoom(const QString & room)
{
if(netClientState != InLobby)
{
qWarning("Illegal try to join room!");
return;
}
myroom = room;
RawSendNet(QString("JOIN_ROOM%1%2").arg(delimeter).arg(room));
isChief = false;
}
void HWNewNet::AddTeam(const HWTeam & team)
{
QString cmd = QString("ADD_TEAM") + delimeter +
team.name() + delimeter +
team.color().name() + delimeter +
team.grave() + delimeter +
team.fort() + delimeter +
team.voicepack() + delimeter +
team.flag() + delimeter +
QString::number(team.difficulty());
for(int i = 0; i < HEDGEHOGS_PER_TEAM; ++i)
{
cmd.append(delimeter);
cmd.append(team.hedgehog(i).Name);
cmd.append(delimeter);
cmd.append(team.hedgehog(i).Hat);
}
RawSendNet(cmd);
}
void HWNewNet::RemoveTeam(const HWTeam & team)
{
RawSendNet(QString("REMOVE_TEAM") + delimeter + team.name());
}
void HWNewNet::NewNick(const QString & nick)
{
RawSendNet(QString("NICK%1%2").arg(delimeter).arg(nick));
}
void HWNewNet::ToggleReady()
{
RawSendNet(QString("TOGGLE_READY"));
}
void HWNewNet::SendNet(const QByteArray & buf)
{
QString msg = QString(buf.toBase64());
RawSendNet(QString("EM%1%2").arg(delimeter).arg(msg));
}
void HWNewNet::RawSendNet(const QString & str)
{
RawSendNet(str.toUtf8());
}
void HWNewNet::RawSendNet(const QByteArray & buf)
{
qDebug() << "Client: " << QString(buf).split("\n");
NetSocket.write(buf);
NetSocket.write("\n\n", 2);
}
void HWNewNet::ClientRead()
{
while (NetSocket.canReadLine()) {
QString s = QString::fromUtf8(NetSocket.readLine());
if (s.endsWith('\n')) s.chop(1);
if (s.size() == 0) {
ParseCmd(cmdbuf);
cmdbuf.clear();
} else
cmdbuf << s;
}
}
void HWNewNet::OnConnect()
{
netClientState = Connected;
}
void HWNewNet::OnDisconnect()
{
netClientState = Disconnected;
if(m_game_connected) emit disconnected("");
m_game_connected = false;
}
void HWNewNet::displayError(QAbstractSocket::SocketError socketError)
{
m_game_connected = false;
switch (socketError) {
case QAbstractSocket::RemoteHostClosedError:
break;
case QAbstractSocket::HostNotFoundError:
emit disconnected(tr("The host was not found. Please check the host name and port settings."));
break;
case QAbstractSocket::ConnectionRefusedError:
emit disconnected(tr("Connection refused"));
break;
default:
emit disconnected(NetSocket.errorString());
}
}
void HWNewNet::SendPasswordHash(const QString & hash)
{
RawSendNet(QString("PASSWORD%1%2").arg(delimeter).arg(hash));
}
void HWNewNet::ParseCmd(const QStringList & lst)
{
qDebug() << "Server: " << lst;
if(!lst.size())
{
qWarning("Net client: Bad message");
return;
}
if (lst[0] == "NICK")
{
mynick = lst[1];
return ;
}
if (lst[0] == "PROTO")
return ;
if (lst[0] == "ERROR") {
if (lst.size() == 2)
emit Error(lst[1]);
else
emit Error("Unknown error");
return;
}
if (lst[0] == "WARNING") {
if (lst.size() == 2)
emit Warning(lst[1]);
else
emit Warning("Unknown warning");
return;
}
if (lst[0] == "CONNECTED") {
if(lst.size() < 3 || lst[2].toInt() < cMinServerVersion)
{
// TODO: Warn user, disconnect
qWarning() << "Server too old";
}
RawSendNet(QString("NICK%1%2").arg(delimeter).arg(mynick));
RawSendNet(QString("PROTO%1%2").arg(delimeter).arg(*cProtoVer));
netClientState = Connected;
m_game_connected = true;
emit adminAccess(false);
return;
}
if (lst[0] == "PING") {
if (lst.size() > 1)
RawSendNet(QString("PONG%1%2").arg(delimeter).arg(lst[1]));
else
RawSendNet(QString("PONG"));
return;
}
if (lst[0] == "ROOMS") {
QStringList tmp = lst;
tmp.removeFirst();
emit roomsList(tmp);
return;
}
if (lst[0] == "SERVER_MESSAGE") {
if(lst.size() < 2)
{
qWarning("Net: Empty SERVERMESSAGE message");
return;
}
emit serverMessage(lst[1]);
return;
}
if (lst[0] == "CHAT") {
if(lst.size() < 3)
{
qWarning("Net: Empty CHAT message");
return;
}
if (netClientState == InLobby)
emit chatStringLobby(lst[1], HWProto::formatChatMsgForFrontend(lst[2]));
else
emit chatStringFromNet(HWProto::formatChatMsg(lst[1], lst[2]));
return;
}
if (lst[0] == "INFO") {
if(lst.size() < 5)
{
qWarning("Net: Malformed INFO message");
return;
}
QStringList tmp = lst;
tmp.removeFirst();
if (netClientState == InLobby)
emit chatStringLobby(tmp.join("\n").prepend('\x01'));
else
emit chatStringFromNet(tmp.join("\n").prepend('\x01'));
return;
}
if (lst[0] == "SERVER_VARS") {
QStringList tmp = lst;
tmp.removeFirst();
while (tmp.size() >= 2)
{
if(tmp[0] == "MOTD_NEW") emit serverMessageNew(tmp[1]);
else if(tmp[0] == "MOTD_OLD") emit serverMessageOld(tmp[1]);
else if(tmp[0] == "LATEST_PROTO") emit latestProtocolVar(tmp[1].toInt());
tmp.removeFirst();
tmp.removeFirst();
}
return;
}
if (lst[0] == "CLIENT_FLAGS")
{
if(lst.size() < 3 || lst[1].size() < 2)
{
qWarning("Net: Malformed CLIENT_FLAGS message");
return;
}
QString flags = lst[1];
bool setFlag = flags[0] == '+';
while(flags.size() > 1)
{
flags.remove(0, 1);
char c = flags[0].toAscii();
switch(c)
{
case 'r':
{
for(int i = 2; i < lst.size(); ++i)
{
if (lst[i] == mynick)
emit setMyReadyStatus(setFlag);
emit setReadyStatus(lst[i], setFlag);
}
}
}
}
return;
}
if (lst[0] == "ADD_TEAM") {
if(lst.size() != 24)
{
qWarning("Net: Bad ADDTEAM message");
return;
}
QStringList tmp = lst;
tmp.removeFirst();
emit AddNetTeam(tmp);
return;
}
if (lst[0] == "REMOVE_TEAM") {
if(lst.size() != 2)
{
qWarning("Net: Bad REMOVETEAM message");
return;
}
emit RemoveNetTeam(HWTeam(lst[1]));
return;
}
if(lst[0] == "ROOMABANDONED") {
netClientState = InLobby;
emit LeftRoom(tr("Room destroyed"));
return;
}
if(lst[0] == "KICKED") {
netClientState = InLobby;
emit LeftRoom(tr("You got kicked"));
return;
}
if(lst[0] == "JOINED") {
if(lst.size() < 2)
{
qWarning("Net: Bad JOINED message");
return;
}
for(int i = 1; i < lst.size(); ++i)
{
if (lst[i] == mynick)
{
netClientState = InRoom;
emit EnteredGame();
emit roomMaster(isChief);
if (isChief)
emit configAsked();
}
emit nickAdded(lst[i], isChief && (lst[i] != mynick));
emit chatStringFromNet(tr("%1 *** %2 has joined the room").arg('\x03').arg(lst[i]));
}
return;
}
if(lst[0] == "LOBBY:JOINED") {
if(lst.size() < 2)
{
qWarning("Net: Bad JOINED message");
return;
}
for(int i = 1; i < lst.size(); ++i)
{
if (lst[i] == mynick)
{
netClientState = InLobby;
RawSendNet(QString("LIST"));
emit connected();
}
emit nickAddedLobby(lst[i], false);
emit chatStringLobby(lst[i], tr("%1 *** %2 has joined").arg('\x03').arg("|nick|"));
}
return;
}
if(lst[0] == "LEFT") {
if(lst.size() < 2)
{
qWarning("Net: Bad LEFT message");
return;
}
emit nickRemoved(lst[1]);
if (lst.size() < 3)
emit chatStringFromNet(tr("%1 *** %2 has left").arg('\x03').arg(lst[1]));
else
emit chatStringFromNet(tr("%1 *** %2 has left (%3)").arg('\x03').arg(lst[1], lst[2]));
return;
}
if(lst[0] == "ROOM") {
if(lst.size() < 2)
{
qWarning("Net: Bad ROOM message");
return;
}
RawSendNet(QString("LIST"));
return;
}
if(lst[0] == "LOBBY:LEFT") {
if(lst.size() < 2)
{
qWarning("Net: Bad LOBBY:LEFT message");
return;
}
emit nickRemovedLobby(lst[1]);
if (lst.size() < 3)
emit chatStringLobby(tr("%1 *** %2 has left").arg('\x03').arg(lst[1]));
else
emit chatStringLobby(tr("%1 *** %2 has left (%3)").arg('\x03').arg(lst[1], lst[2]));
return;
}
if (lst[0] == "RUN_GAME") {
netClientState = InGame;
emit AskForRunGame();
return;
}
if (lst[0] == "ASKPASSWORD") {
emit AskForPassword(mynick);
return;
}
if (lst[0] == "NOTICE") {
if(lst.size() < 2)
{
qWarning("Net: Bad NOTICE message");
return;
}
bool ok;
int n = lst[1].toInt(&ok);
if(!ok)
{
qWarning("Net: Bad NOTICE message");
return;
}
handleNotice(n);
return;
}
if (lst[0] == "TEAM_ACCEPTED") {
if (lst.size() != 2)
{
qWarning("Net: Bad TEAM_ACCEPTED message");
return;
}
emit TeamAccepted(lst[1]);
return;
}
if (lst[0] == "CFG") {
if(lst.size() < 3)
{
qWarning("Net: Bad CFG message");
return;
}
QStringList tmp = lst;
tmp.removeFirst();
tmp.removeFirst();
if (lst[1] == "SCHEME")
emit netSchemeConfig(tmp);
else
emit paramChanged(lst[1], tmp);
return;
}
if (lst[0] == "HH_NUM") {
if (lst.size() != 3)
{
qWarning("Net: Bad TEAM_ACCEPTED message");
return;
}
HWTeam tmptm(lst[1]);
tmptm.setNumHedgehogs(lst[2].toUInt());
emit hhnumChanged(tmptm);
return;
}
if (lst[0] == "TEAM_COLOR") {
if (lst.size() != 3)
{
qWarning("Net: Bad TEAM_COLOR message");
return;
}
HWTeam tmptm(lst[1]);
tmptm.setColor(QColor(lst[2]));
emit teamColorChanged(tmptm);
return;
}
if (lst[0] == "EM") {
if(lst.size() < 2)
{
qWarning("Net: Bad EM message");
return;
}
for(int i = 1; i < lst.size(); ++i)
{
QByteArray em = QByteArray::fromBase64(lst[i].toAscii());
emit FromNet(em);
}
return;
}
if (lst[0] == "BYE") {
if (lst.size() < 2)
{
qWarning("Net: Bad BYE message");
return;
}
if (lst[1] == "Authentication failed")
{
emit AuthFailed();
}
m_game_connected = false;
Disconnect();
emit disconnected(lst[1]);
return;
}
if (lst[0] == "ADMIN_ACCESS") {
emit adminAccess(true);
return;
}
if (lst[0] == "ROOM_CONTROL_ACCESS") {
if (lst.size() < 2)
{
qWarning("Net: Bad ROOM_CONTROL_ACCESS message");
return;
}
isChief = (lst[1] != "0");
emit roomMaster(isChief);
return;
}
qWarning() << "Net: Unknown message:" << lst;
}
void HWNewNet::onHedgehogsNumChanged(const HWTeam& team)
{
if (isChief)
RawSendNet(QString("HH_NUM%1%2%1%3")
.arg(delimeter)
.arg(team.name())
.arg(team.numHedgehogs()));
}
void HWNewNet::onTeamColorChanged(const HWTeam& team)
{
if (isChief)
RawSendNet(QString("TEAM_COLOR%1%2%1%3")
.arg(delimeter)
.arg(team.name())
.arg(team.color().name()));
}
void HWNewNet::onParamChanged(const QString & param, const QStringList & value)
{
if (isChief)
RawSendNet(
QString("CFG%1%2%1%3")
.arg(delimeter)
.arg(param)
.arg(value.join(QString(delimeter)))
);
}
void HWNewNet::chatLineToNet(const QString& str)
{
if(str != "") {
RawSendNet(QString("CHAT") + delimeter + str);
emit(chatStringFromMe(HWProto::formatChatMsg(mynick, str)));
}
}
void HWNewNet::chatLineToLobby(const QString& str)
{
if(str != "") {
RawSendNet(QString("CHAT") + delimeter + str);
emit chatStringLobby(mynick, HWProto::formatChatMsgForFrontend(str));
}
}
void HWNewNet::SendTeamMessage(const QString& str)
{
RawSendNet(QString("TEAMCHAT") + delimeter + str);
}
void HWNewNet::askRoomsList()
{
if(netClientState != InLobby)
{
qWarning("Illegal try to get rooms list!");
return;
}
RawSendNet(QString("LIST"));
}
HWNewNet::ClientState HWNewNet::clientState()
{
return netClientState;
}
QString HWNewNet::getNick()
{
return mynick;
}
QString HWNewNet::getRoom()
{
return myroom;
}
QString HWNewNet::getHost()
{
return myhost;
}
bool HWNewNet::isRoomChief()
{
return isChief;
}
void HWNewNet::gameFinished(bool correctly)
{
if (netClientState == InGame) netClientState = InRoom;
RawSendNet(QString("ROUNDFINISHED%1%2").arg(delimeter).arg(correctly ? "1" : "0"));
}
void HWNewNet::banPlayer(const QString & nick)
{
RawSendNet(QString("BAN%1%2").arg(delimeter).arg(nick));
}
void HWNewNet::kickPlayer(const QString & nick)
{
RawSendNet(QString("KICK%1%2").arg(delimeter).arg(nick));
}
void HWNewNet::infoPlayer(const QString & nick)
{
RawSendNet(QString("INFO%1%2").arg(delimeter).arg(nick));
}
void HWNewNet::followPlayer(const QString & nick)
{
if (!isInRoom()) {
RawSendNet(QString("FOLLOW%1%2").arg(delimeter).arg(nick));
isChief = false;
}
}
void HWNewNet::startGame()
{
RawSendNet(QString("START_GAME"));
}
void HWNewNet::updateRoomName(const QString & name)
{
RawSendNet(QString("ROOM_NAME%1%2").arg(delimeter).arg(name));
}
void HWNewNet::toggleRestrictJoins()
{
RawSendNet(QString("TOGGLE_RESTRICT_JOINS"));
}
void HWNewNet::toggleRestrictTeamAdds()
{
RawSendNet(QString("TOGGLE_RESTRICT_TEAMS"));
}
void HWNewNet::clearAccountsCache()
{
RawSendNet(QString("CLEAR_ACCOUNTS_CACHE"));
}
void HWNewNet::partRoom()
{
netClientState = InLobby;
RawSendNet(QString("PART"));
}
bool HWNewNet::isInRoom()
{
return netClientState >= InRoom;
}
void HWNewNet::setServerMessageNew(const QString & msg)
{
RawSendNet(QString("SET_SERVER_VAR%1MOTD_NEW%1%2").arg(delimeter).arg(msg));
}
void HWNewNet::setServerMessageOld(const QString & msg)
{
RawSendNet(QString("SET_SERVER_VAR%1MOTD_OLD%1%2").arg(delimeter).arg(msg));
}
void HWNewNet::setLatestProtocolVar(int proto)
{
RawSendNet(QString("SET_SERVER_VAR%1LATEST_PROTO%1%2").arg(delimeter).arg(proto));
}
void HWNewNet::askServerVars()
{
RawSendNet(QString("GET_SERVER_VAR"));
}
void HWNewNet::handleNotice(int n)
{
switch(n)
{
case 0:
{
emit NickTaken(mynick);
break;
}
}
}