Snake/SnakeServer/ClientProtocol/server.cpp

#include "server.h"
#include "quasarapp.h"
#include <QTcpSocket>
#include <factorynetobjects.h>

namespace ClientProtocol {

bool Server::parsePackage(const Package &pkg, QTcpSocket* sender) {
    if (!pkg.isValid()) {
        QuasarAppUtils::Params::verboseLog("incomming package is not valid!");
        changeKarma(qHash(sender->peerAddress()), CRITICAL_ERROOR);
        return false;
    }

    if (!sender->isValid()) {
        QuasarAppUtils::Params::verboseLog("incomming package is not valid!");
        changeKarma(qHash(sender->peerAddress()), LOGICK_ERROOR);
        return false;
    }

    switch (pkg.hdr.command) {
    case NetworkClasses::Ping: {

        if (pkg.hdr.type != Request) {
            return false;
        }

        Package pcg;

        QVariantMap map;
        if (!(FactoryNetObjects::build(NetworkClasses::Ping, map) && pcg.create(map, Responke))) {
            return false;
        };

        if (!sendPackage(pcg, sender)) {
            QuasarAppUtils::Params::verboseLog("!responce not sendet!");
        }
        break;
    }

    default: {
        QVariantMap data;
        pkg.parse(data);
        emit incomingReques(data, qHash(sender->peerAddress()));
    }
    }

    return true;
}

bool Server::sendPackage(Package &pkg, QTcpSocket * target) {
    if (!pkg.isValid()) {
        return false;
    }

    if (!target || !target->isValid()) {
        qCritical() << "destination server not valid!";
        return false;
    }

    if (!target->waitForConnected()) {
        qCritical() << "no connected to server! " << target->errorString();
        return false;
    }

    auto bytes = pkg.toBytes();
    bool sendet = bytes.size() == target->write(bytes);

    return sendet;
}

void Server::ban(quint32 target) {
    _connections[target].karma = BANED_KARMA;
}

void Server::unBan(quint32 target) {
    _connections[target].karma = RESTORE_KARMA;
}

void Server::registerSocket(QTcpSocket *socket) {
    auto address = qHash(socket->peerAddress());

    if (!_connections[address].sct) {

        _connections[address].karma = DEFAULT_KARMA;
        _connections[address].sct = socket;

        connect(socket, &QTcpSocket::readyRead, this,  &Server::avelableBytes);
    }
}

bool Server::changeKarma(quint32 addresss, int diff) {
    auto objKarma = _connections.value(addresss).karma;

    if (objKarma >= NOT_VALID_CARMA) {
        return false;
    }

    if (objKarma <= BANED_KARMA) {
        return false;
    }

    _connections[addresss].karma += diff;
    return true;
}

bool Server::isBaned(const QTcpSocket * adr) const {
    return _connections.value(qHash(adr->peerAddress())).karma < 1;
}

void Server::saveKarma() const {

}

bool Server::loadKarma() {
    return false;
}

void Server::avelableBytes() {
    auto client = dynamic_cast<QTcpSocket*>(sender());

    if (!client) {
        return;
    }

    auto array = client->readAll();

    if (_downloadPackage.hdr.isValid()) {
        _downloadPackage.data.append(array);

    } else {
        memcpy(&_downloadPackage.hdr,
               array.data(), sizeof(Header));
        _downloadPackage.data.append(array.mid(sizeof(Header)));
    }

    if (_downloadPackage.isValid()) {
        parsePackage(_downloadPackage, client);
        _downloadPackage.reset();
        return;
    }
}

void Server::handleIncommingConnection() {
    while (hasPendingConnections()) {
        auto socket = nextPendingConnection();

        if (isBaned(socket)) {
            socket->abort();
            break;
        }

        registerSocket(socket);
    }
}

Server::Server(QObject *ptr) :
    QTcpServer (ptr) {

    connect(this, &Server::newConnection, this, &Server::handleIncommingConnection);
}

Server::~Server() {

}

bool Server::run(const QString &ip, unsigned short port) {
    if (!listen(QHostAddress(ip), port) ) {
        QuasarAppUtils::Params::verboseLog("listing fail " + this->errorString());
        return false;
    }

    return true;
}

void Server::stop(bool reset) {
    close();

    if (reset) {
        _connections.clear();
    }
}

void Server::badRequest(quint32 address) {
    auto client = _connections.value(address);

    if (!changeKarma(address, REQUEST_ERROR)) {
        return;
    }

    QVariantMap map;

    Package pcg;
    if (!(FactoryNetObjects::build(NetworkClasses::BadRequest, map) && pcg.create(map, Request))) {
        return;
    };

    if (!sendPackage(pcg, client.sct)) {
        return;
    }
}

QString Server::getWorkState() const {
    if (isListening()) {
        if (hasPendingConnections())
            return "Work";
        else {
            return "overload";
        }
    }

    return "Not running";
}

QString Server::connectionState() const {
    return QString("%0 / %1").arg(_connections.size()).arg(maxPendingConnections());
}

QStringList Server::baned() const {
    QStringList list = {};
    for (auto i : _connections) {
        if (i.karma <= 0) {
            list.push_back(i.sct->peerAddress().toString());
        }
    }

    return list;
}

Connectioninfo::Connectioninfo(QTcpSocket *tcp, int kar) {
    sct = tcp;
    karma = kar;
}

Connectioninfo::~Connectioninfo() {
    if (sct) {
        sct->deleteLater();
    }
}

}