Teeworlds.cpp

/*            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
 *                    Version 2, December 2004
 *
 * Copyright (C) 2004 Sam Hocevar
 *  14 rue de Plaisance, 75014 Paris, France
 * Everyone is permitted to copy and distribute verbatim or modified
 * copies of this license document, and changing it is allowed as long
 * as the name is changed.
 *
 *            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
 *   TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
 *
 *  0. You just DO WHAT THE FUCK YOU WANT TO.
 */

#include "Teeworlds.h"
#include <QHostInfo>
#include <QtDebug>

TeeworldsServer::TeeworldsServer(const QString &param)
{
    QRegExp reg("^([^ ]+) ([0-9]+)$");
    if(reg.indexIn(param) != -1)
    {
        bool ok;
        int port = reg.cap(2).toInt(&ok, 10);
        if(ok && port >= 1 && port <= 65535)
        {
            setup(reg.cap(1).toLatin1(), port);
            return ;
        }
    }

    throw ServerError(tr("TeeworldsServer: invalid configuration"));
}

TeeworldsServer::TeeworldsServer(const char *host, int port)
{
    setup(host, port);
}

void TeeworldsServer::setup(const char *host, int port)
{
    m_sHost = host; m_iPort = port; m_iNumPlayers = 0; m_iMaxPlayers = 0;

    m_pUdpSocket = new QUdpSocket(this);
    {
        int port = 5000;
        while(!m_pUdpSocket->bind(QHostAddress::Any, port))
        {
            delete m_pUdpSocket; m_pUdpSocket = new QUdpSocket(this); // FIXME
            port++;
            if(port == 5040)
                throw ServerError(tr("TeeworldsServer: "
                    "can't listen: ") + m_pUdpSocket->errorString());
        }
        qDebug() << tr("TeeworldsServer: listening on port %1").arg(port);
    }
    connect(m_pUdpSocket, SIGNAL(readyRead()), this, SLOT(receiveData()));

    m_pTimer = new QTimer(this);
    m_pTimer->setSingleShot(false);
    connect(m_pTimer, SIGNAL(timeout()), this, SLOT(query()));
    m_pTimer->start(30000);

    query();
}

unsigned int TeeworldsServer::numPlayers() const
{
    return m_iNumPlayers;
}

unsigned int TeeworldsServer::maxPlayers() const
{
    return m_iMaxPlayers;
}

QString TeeworldsServer::map() const
{
    return m_sMap;
}

QString TeeworldsServer::mode() const
{
    return m_sMode;
}

void TeeworldsServer::query()
{
    QHostInfo ns = QHostInfo::fromName(m_sHost);
    if(!ns.addresses().isEmpty())
    {
        if(m_pUdpSocket->writeDatagram(
            "\xff\xff\xff\xff\xff\xff\xff\xff\xff\xffgief", 14,
            ns.addresses().first(), m_iPort) == -1)
        {
            emit errorEncountered(m_pUdpSocket->errorString());
        }
    }
    else
        emit errorEncountered(ns.errorString());
}

void TeeworldsServer::refresh()
{
    query();
    m_pTimer->start();
}

void TeeworldsServer::receiveData()
{
    while(m_pUdpSocket->hasPendingDatagrams())
    {
        QByteArray datagram;
        datagram.resize(m_pUdpSocket->pendingDatagramSize());
        QHostAddress sender;
        quint16 senderPort;

        m_pUdpSocket->readDatagram(datagram.data(), datagram.size(),
            &sender, &senderPort);

        QRegExp regexp("info([0-9]+\\.[0-9]+\\.[0-9]+)" // server version
            "\\x0000(.*)" // server name
            "\\x0000(.*)" // map name
            "\\x0000(.*)" // game type
            "\\x0000(-?[0-9]*)"   // flags
            "\\x0000(-?[0-9]*)" // progression
            "\\x0000([0-9]*)"   // player count
            "\\x0000([0-9]*)"   // max players
            "\\x0000");
        // We use fromAscii because the implicit cast QByteArray -> QString
        // will stop the string at the first null byte
        if(regexp.indexIn(
            QString::fromAscii(datagram.constData(), datagram.size())) == -1)
        {
            emit errorEncountered(tr("Invalid answer"));
        }

        bool ok, changed = false;
        unsigned int old_players = m_iNumPlayers;
        changed = confirm_assign(&m_iNumPlayers,
            (unsigned)regexp.cap(7).toInt(&ok, 10)) || changed;
        changed = confirm_assign(&m_iMaxPlayers,
            (unsigned)regexp.cap(8).toInt(&ok, 10)) || changed;
        changed = confirm_assign(&m_sMap, regexp.cap(3)) || changed;
        changed = confirm_assign(&m_sMode, regexp.cap(4)) || changed;
        if(changed)
            emit infosChanged(m_iNumPlayers, m_iMaxPlayers, m_sMap, m_sMode,
                old_players == 0 && m_iNumPlayers > 0);
    }
}