Сетевой обмен через QTcpSocket's

#ifndef TCPClient_H
#define TCPClient_H
 
#include "include.h"
 
class MyClient : public QDialog
{
    Q_OBJECT
public:
    static const quint8 MESSAGE = 1;
    static const quint8 UNSECURED_CONNECTION = 2;
    static const quint8 SECURITY_CHECK = 3;
    MyClient(QWidget *parent);
    bool isConnectionSecured()
    {
        return ConnectionSecurityStatus;
    }
private:
    QTcpSocket *_socket;    // Сокет
    QTcpServer *_server;
    quint16 _blockSize;     // Текущий размер блока данных
    bool ConnectionSecurityStatus;  // TRUE - ключи для шифрования созданы и проверены
                                    // FALSE - ключи не были сгенерированы или проверка
                                    // шифрования заданой строки выявила ошибку
    bool isConnectionExist = 0; // По задумке чат может быть только 1 на 1, т.о. если
                                // isConnectionExist = 1, то при запросе на соединение
                                // отправится сообщение о занятости юзверя
    uint nPort = 9100;
    // Отправляем сообщение клиенту отсюда
    void SendToClient (QTcpSocket *socket, const QString *str);
private slots:
// Слоты "клиента":
    void SocketConnected();
    void SocketDisconnected();
    // Cигнал readyRead вызывается, когда сокет получает пакет
    // (который может быть лишь частью отправленых данных) байтов
    void SocketReadyRead();
    void SocketDisplayError(int socketError);
// Слоты "сервера":
    // Создаем соединения тут
    void ServerNewConnection();
    // Показ ошибки при соединении
    void ShowServerError(int serverError);
public slots:
    // Слот Connect получает сигнал (с данными) от Dialog для создания подключения
    void Connect(QHostAddress IP, quint16 Port);
    // Сюда приходит зашифрованное сообщение для отправки
    void SendMessageOut (const QString *message);
signals:
    // Сигнал для MainWindow c зашифрованным сообщением
    void NewMessage(const QString *message);
    // Сигнал установки соединения (закрывает Dialog'овое окно)
    void ConnectionValid();
};
 
#endif // TCPClient_H

#include "tcpClient.h"
 
MyClient::MyClient(QWidget *parent):
    QDialog(parent)
{
    // Создаем сокет
    _socket = new QTcpSocket(this);
    // Создаем сервер
    _server = new QTcpServer(this);
 
    // Проверка сокета
/*    if(!_socket->isValid())
    {
        QMessageBox::critical(0, "Socket Error",
                              "Unable to create a socket: "
                              + _socket->errorString());
        _socket->abort();
        return;
    }*/
    // Проверка сервера
    if(!_server->listen(QHostAddress::Any, nPort))
    {
        if(!_server->listen(QHostAddress::Any, nPort+10))
        {
            QMessageBox::critical(0, "Server Error",
                              "Unable to start the server: "
                              + _server->errorString());
            _server->close();
            return;
        }
    }
    // Подключаем сигналы "сервера"
    connect(_server, SIGNAL(newConnection()), this, SLOT(ServerNewConnection()));
//    connect(_server, SIGNAL(acceptError(QAbstractSocket::SocketError)),
//                  this, SLOT(ShowServerError(int)));
    // Подключаем сигналы "клиента"
    connect(_socket, SIGNAL(readyRead()), this, SLOT(SocketReadyRead()));
    connect(_socket, SIGNAL(connected()), this, SLOT(SocketConnected()));
    connect(_socket, SIGNAL(disconnected()), this, SLOT(SocketDisconnected()));
//    connect(_socket, SIGNAL(error(QAbstractSocket::SocketError)),this,
//                        SLOT(SocketDisplayError(int)));
}
 
void MyClient::Connect(QHostAddress IP, quint16 Port)
{
    _socket->connectToHost(IP, Port);
    if(_socket->isValid())
    {
        isConnectionExist = 1;
        emit ConnectionValid();
    }
    else
    {
        QMessageBox::critical(0, "Socket Error",
                              "Unable to create a socket: "
                              + _socket->errorString());
        _socket->abort();
        return;
    }
}
 
void MyClient::ServerNewConnection()
{
    QTcpSocket *ClientSocket = _server->nextPendingConnection();
 
    // Если одно соединение уже существует, то вежливо извиняемся и обрываем соединение
    if(isConnectionExist)
    {
        QString Apologizing_Message = "This user is alredy chating with someone(";
        SendToClient(ClientSocket, &Apologizing_Message);
        ClientSocket->disconnectFromHost();
        return;
    }
    else
    {
        QString Message = "Connection established!";
        SendToClient(ClientSocket, &Message);
    }
    // Теперь _socket соединен с запрашивававшимм соединение
    _socket = ClientSocket;
    // Запоминаем, что соединение установлено
    isConnectionExist = 1;
}
 
void MyClient::SendToClient(QTcpSocket *socket, const QString *str)
{
    // Если вдруг мы считаем, что соединения нет
//    if(!isConnectionExist)
//        return;
 
    // Если считаем, что соединение не безопасно
    if(!ConnectionSecurityStatus)
    {
        // todo: Сигнал на новый обмен кодами шифрования
    }
 
    QByteArray *message;
    message = new QByteArray;
    QDataStream out(message, QIODevice::WriteOnly);
 
    out.setVersion(QDataStream::Qt_5_4);
 
    // Оставляем два пустых байта под размер сообщения, третий байт - код, показывающий
    // как сообщение будет обработано и в конец записываем само сообщение
    out << quint16(0) << quint8(MESSAGE) << str;
 
    // Возвращаемся на начало
    out.device()->seek(0);
 
    // И дописываем в оставленные пустыми два байта размер сообщение
    // (размер только самой строки)
    out << quint16(message->size() - sizeof(quint16));
 
    // Наконец отправляем сообщение
    socket->write(message->toStdString().c_str());
    emit socket->readyRead();
    delete message;
}
 
void MyClient::SocketConnected()
{
// После подключения следует отправить запрос на авторизацию
    QByteArray block;
    QDataStream out(&block, QIODevice::WriteOnly);
 
    out.setVersion(QDataStream::Qt_5_4);
 
//резервируем 2 байта для размера блока. Класс MyClient используется в реализации сервера, но тут используем статические члены этого класса - константы команд
//третий байт - команда
    out << (quint16)0 << (quint8)MyClient::UNSECURED_CONNECTION;
 
//возваращаемся в начало
    out.device()->seek(0);
//вписываем размер блока на зарезервированное место
    out << (quint16)(block.size() - sizeof(quint16));
    if(_socket->write(block) == -1)
    {
        QMessageBox::critical(this, "Error",
                              "Error while writing in socket");
    }
}
 
void MyClient::SocketDisconnected()
{
    isConnectionExist = 0;
    _socket->disconnectFromHost();
    _server->close();
}
 
void MyClient::SocketDisplayError(int socketError)
{
    QMessageBox::critical(0, "Socket Error!", tr("Code: %1").arg(socketError));
}
 
void MyClient::ShowServerError(int serverError)
{
    QMessageBox::critical(0, "Server Error!", tr("Code: %1").arg(serverError));
}
 
void MyClient::SendMessageOut (const QString *message)
{
   emit SendToClient(_socket, message);
}
 
void MyClient::SocketReadyRead()
{
//тут обрабатываются данные от сервера
    QDataStream in(_socket);
//если считываем новый блок первые 2 байта это его размер
 
    if (_blockSize == 0) {
//если пришло меньше 2 байт ждем пока будет 2 байта
        if (_socket->bytesAvailable() < (int)sizeof(quint16))
            return;
//считываем размер (2 байта)
        in >> _blockSize;
    }
//ждем пока блок придет полностью
    if (_socket->bytesAvailable() < _blockSize)
        return;
    else //можно принимать новый блок
        _blockSize = 0;
//3 байт - команда серверу
    quint8 command;
    in >> command;
 
    switch (command)
    {
//общее сообщение от сервера
        case MyClient::MESSAGE:
        {
//Скидываем сообщение на расшифровку
            QString message;
            in >> message;
            emit NewMessage(&message);
        }
        case MyClient::UNSECURED_CONNECTION:
        {
 
            // todo: Обработку команды сервера о незащищенном канале (т.е. обмен ключами)
        }
    }
}

void MyClient::SocketReadyRead()
{
//тут обрабатываются данные от сервера
    QTcpSocket *Socket = qobject_cast<QTcpSocket*>(sender());
                                  
    QDataStream in(Socket);
//если считываем новый блок первые 2 байта это его размер
 
    if (_blockSize == 0) {
//если пришло меньше 2 байт ждем пока будет 2 байта
        if (Socket->bytesAvailable() < (int)sizeof(quint16))
            return;
//считываем размер (2 байта)
        in >> _blockSize;
    }
//ждем пока блок придет полностью
    if (Socket->bytesAvailable() < _blockSize)
        return;
    else //можно принимать новый блок
        _blockSize = 0;
//3 байт - команда серверу
    quint8 command;
    in >> command;
 
    switch (command)
    {
        default:
        case MyClient::UNSECURED_CONNECTION:
        {
 
            // todo: Обработку команды сервера о незащищенном канале (т.е. обмен ключами)
        }
        //общее сообщение от сервера
        case MyClient::MESSAGE:
        {
            //Скидываем сообщение на расшифровку
            QString message;
            in >> message;  // message получается всегда равным "", т.е. пустым
            emit NewMessage(&message);
            break;
        }
    }
}

void MyClient::SendToClient(QTcpSocket *socket, const QString *str)
{
    QByteArray *message;
    message = new QByteArray;
    QDataStream out(message, QIODevice::WriteOnly);
 
    out.setVersion(QDataStream::Qt_5_4);
    // Оставляем два пустых байта под размер сообщения, третий байт - код, показывающий
    // как сообщение будет обработано и в конец записываем само сообщение
    out << quint16(0) << quint8(MESSAGE) << str;
 
    // Возвращаемся на начало
    out.device()->seek(0);
 
    // И дописываем в оставленные пустыми два байта размер сообщение
    // (размер только самой строки)
    out << quint16(message->size() - sizeof(quint16));
 
    // Наконец отправляем сообщение
    socket->write(*message);
    delete message;
}