Logo Search packages:      
Sourcecode: hedgewars version File versions  Download package

newnetclient.cpp

/*
 * Hedgewars, a free turn based strategy game
 * Copyright (c) 2006-2008 Ulyanov Igor <iulyanov@gmail.com>
 * Copyright (c) 2009 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 "gameuiconfig.h"
#include "game.h"
#include "gamecfgwidget.h"
#include "teamselect.h"
#include "misc.h"

char delimeter='\n';

HWNewNet::HWNewNet(GameUIConfig * config, GameCFGWidget* pGameCFGWidget, TeamSelWidget* pTeamSelWidget) :
  config(config),
  m_pGameCFGWidget(pGameCFGWidget),
  m_pTeamSelWidget(pTeamSelWidget),
  isChief(false),
  m_game_connected(false),
  loginStep(0),
  netClientState(0)
{
// 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)));

// config stuff
      connect(this, SIGNAL(paramChanged(const QString &, const QStringList &)), pGameCFGWidget, SLOT(setParam(const QString &, const QStringList &)));
      connect(pGameCFGWidget, SIGNAL(paramChanged(const QString &, const QStringList &)), this, SLOT(onParamChanged(const QString &, const QStringList &)));
      connect(this, SIGNAL(configAsked()), pGameCFGWidget, SLOT(fullNetConfig()));
}

HWNewNet::~HWNewNet()
{
      if (m_game_connected)
      {
            RawSendNet(QString("QUIT%1%2").arg(delimeter).arg("User quit"));
            emit Disconnected();
      }
      NetSocket.flush();
}

void HWNewNet::Connect(const QString & hostName, quint16 port, const QString & nick)
{
      mynick = nick;
      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 != 2)
      {
            qWarning("Illegal try to create room!");
            return;
      }
      
      RawSendNet(QString("CREATE_ROOM%1%2").arg(delimeter).arg(room));
      m_pGameCFGWidget->setEnabled(true);
      m_pTeamSelWidget->setInteractivity(true);
      isChief = true;
}

void HWNewNet::JoinRoom(const QString & room)
{
      if(netClientState != 2)
      {
            qWarning("Illegal try to join room!");
            return;
      }
      
      loginStep++;

      RawSendNet(QString("JOIN_ROOM%1%2").arg(delimeter).arg(room));
      m_pGameCFGWidget->setEnabled(false);
      m_pTeamSelWidget->setInteractivity(false);
      isChief = false;
}

void HWNewNet::AddTeam(const HWTeam & team)
{
      QString cmd = QString("ADD_TEAM") + delimeter +
           team.TeamName + delimeter +
           team.teamColor.name() + delimeter +
           team.Grave + delimeter +
           team.Fort + delimeter +
           team.Voicepack + delimeter +
           QString::number(team.difficulty);

      for(int i = 0; i < 8; ++i)
      {
            cmd.append(delimeter);
            cmd.append(team.HHName[i]);
            cmd.append(delimeter);
            cmd.append(team.HHHat[i]);
      }
      RawSendNet(cmd);
}

void HWNewNet::RemoveTeam(const HWTeam & team)
{
      RawSendNet(QString("REMOVE_TEAM") + delimeter + team.TeamName);
}

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().trimmed());

            if (s.size() == 0) {
                  ParseCmd(cmdbuf);
                  cmdbuf.clear();
            } else
                  cmdbuf << s;
      }
}

void HWNewNet::OnConnect()
{
}

void HWNewNet::OnDisconnect()
{
      if(m_game_connected) emit Disconnected();
      m_game_connected = false;
}

void HWNewNet::displayError(QAbstractSocket::SocketError socketError)
{
      emit Disconnected();
      
      switch (socketError) {
            case QAbstractSocket::RemoteHostClosedError:
                  break;
            case QAbstractSocket::HostNotFoundError:
                  emit showMessage(tr("The host was not found. Please check the host name and port settings."));
                  break;
            case QAbstractSocket::ConnectionRefusedError:
                  emit showMessage(tr("Connection refused"));
                  break;
            default:
                  emit showMessage(NetSocket.errorString());
            }
}

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 showMessage("Error: " + lst[1]);
            else
                  emit showMessage("Unknown error");
            return;
      }

      if (lst[0] == "WARNING") {
            if (lst.size() == 2)
                  emit showMessage("Warning: " + lst[1]);
            else
                  emit showMessage("Unknown warning");
            return;
      }

      if (lst[0] == "CONNECTED") {
            RawSendNet(QString("NICK%1%2").arg(delimeter).arg(mynick));
            RawSendNet(QString("PROTO%1%2").arg(delimeter).arg(*cProtoVer));
            netClientState = 1;
            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 == 2)
                  emit chatStringLobby(formatChatMsg(lst[1], lst[2]));
            else
                  emit chatStringFromNet(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 == 2)
                  emit chatStringLobby(tmp.join("\n"));
            else
                  emit chatStringFromNet(tmp.join("\n"));
            return;
      }

      if (lst[0] == "READY") {
            if(lst.size() < 2)
            {
                  qWarning("Net: Malformed READY message");
                  return;
            }
            for(int i = 1; i < lst.size(); ++i)
            {
                  if (lst[i] == mynick)
                        emit setMyReadyStatus(true);
                  emit setReadyStatus(lst[i], true);
            }
            return;
      }
      
      if (lst[0] == "NOT_READY") {
            if(lst.size() < 2)
            {
                  qWarning("Net: Malformed NOT_READY message");
                  return;
            }
            for(int i = 1; i < lst.size(); ++i)
            {
                  if (lst[i] == mynick)
                        emit setMyReadyStatus(false);
                  emit setReadyStatus(lst[i], false);
            }
            return;
      }

      if (lst[0] == "ADD_TEAM") {
            if(lst.size() != 23)
            {
                  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;
            }
            m_pTeamSelWidget->removeNetTeam(HWTeam(lst[1]));
            return;
      }

      if(lst[0] == "ROOMABANDONED") {
            netClientState = 2;
            emit showMessage(HWNewNet::tr("Room destroyed"));
            emit LeftRoom();
            return;
      }

      if(lst[0] == "KICKED") {
            netClientState = 2;
            emit showMessage(HWNewNet::tr("You got kicked"));
            emit LeftRoom();
            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 = 3;
                        emit EnteredGame();
                        emit roomMaster(isChief);
                        if (isChief)
                              emit configAsked();
                  }
                  emit nickAdded(lst[i]);
                  emit chatStringFromNet(QString(tr("*** %1 joined")).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 = 2;
                        RawSendNet(QString("LIST"));
                        emit Connected();
                  }

                  emit nickAddedLobby(lst[i]);
                  emit chatStringLobby(QString(tr("*** %1 joined")).arg(lst[i]));
            }
            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(QString(tr("*** %1 left")).arg(lst[1]));
            else
                  emit chatStringFromNet(QString(tr("*** %1 left (%2)")).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(QString(tr("*** %1 left")).arg(lst[1]));
            else
                  emit chatStringLobby(QString(tr("*** %1 left (%2)")).arg(lst[1], lst[2]));
            return;
      }

      if (lst[0] == "RUN_GAME") {
            netClientState = 5;
            RunGame();
            return;
      }

      if (lst[0] == "ASKPASSWORD") {
            QString password = QInputDialog::getText(0, tr("Password"), tr("Enter your password:"), QLineEdit::Password);

            QString hash = QCryptographicHash::hash(password.toLatin1(), QCryptographicHash::Md5).toHex();

            RawSendNet(QString("PASSWORD%1%2").arg(delimeter).arg(hash));
            return;
      }

      if (lst[0] == "TEAM_ACCEPTED") {
            if (lst.size() != 2)
            {
                  qWarning("Net: Bad TEAM_ACCEPTED message");
                  return;
            }
            m_pTeamSelWidget->changeTeamStatus(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.numHedgehogs = 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.teamColor = 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;
            }
            emit showMessage(HWNewNet::tr("Quit reason: ") + lst[1]);
            return;
      }


      if (lst[0] == "ADMIN_ACCESS") {
            emit adminAccess(true);
            return;
      }

      qWarning() << "Net: Unknown message:" << lst;
}

void HWNewNet::RunGame()
{
      emit AskForRunGame();
}

void HWNewNet::onHedgehogsNumChanged(const HWTeam& team)
{
      if (isChief)
      RawSendNet(QString("HH_NUM%1%2%1%3")
                  .arg(delimeter)
                  .arg(team.TeamName)
                  .arg(team.numHedgehogs));
}

void HWNewNet::onTeamColorChanged(const HWTeam& team)
{
      if (isChief)
      RawSendNet(QString("TEAM_COLOR%1%2%1%3")
                  .arg(delimeter)
                  .arg(team.TeamName)
                  .arg(team.teamColor.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(formatChatMsg(mynick, str)));
      }
}

void HWNewNet::chatLineToLobby(const QString& str)
{
      if(str != "") {
            RawSendNet(QString("CHAT") + delimeter + str);
            emit(chatStringFromMeLobby(formatChatMsg(mynick, str)));
      }
}

void HWNewNet::askRoomsList()
{
      if(netClientState != 2)
      {
            qWarning("Illegal try to get rooms list!");
            return;
      }
      RawSendNet(QString("LIST"));
}

bool HWNewNet::isRoomChief()
{
      return isChief;
}

void HWNewNet::gameFinished()
{
      netClientState = 3;
      RawSendNet(QString("ROUNDFINISHED"));
}

QString HWNewNet::formatChatMsg(const QString & nick, const QString & msg)
{
      if(msg.left(4) == "/me ")
            return QString("* %1 %2").arg(nick).arg(msg.mid(4));
      else
            return QString("%1: %2").arg(nick).arg(msg);
}

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::startGame()
{
      RawSendNet(QString("START_GAME"));
}

void HWNewNet::toggleRestrictJoins()
{
      RawSendNet(QString("TOGGLE_RESTRICT_JOINS"));
}

void HWNewNet::toggleRestrictTeamAdds()
{
      RawSendNet(QString("TOGGLE_RESTRICT_TEAMS"));
}

void HWNewNet::partRoom()
{
      netClientState = 2;
      RawSendNet(QString("PART"));
}

bool HWNewNet::isInRoom()
{
      return netClientState > 2;
}

void HWNewNet::newServerMessage(const QString & msg)
{
      RawSendNet(QString("SET_SERVER_MESSAGE%1%2").arg(delimeter).arg(msg));
}

Generated by  Doxygen 1.6.0   Back to index