amnezia-client/client/managementserver.cpp
pokamest c2a7d66cb4
Release 1.0 (#4)
* Crash fix in management server

* Openvpn scripts fixes
some refactoring

* deploy fix

* Scripts fix for macos

* OpenVpn runtime error codes handling

* MacOS deploy script fix

* easyrsa scripts for MacOS

* Refactoring
Ui improvements
Bug fixes

* new server page fix
2021-01-09 20:08:41 +03:00

113 lines
2.7 KiB
C++

#include <QDebug>
#include <QTcpServer>
#include <QTcpSocket>
#include "managementserver.h"
ManagementServer::ManagementServer(QObject *parent) : QObject(parent),
m_tcpServer(nullptr)
{
}
ManagementServer::~ManagementServer()
{
}
bool ManagementServer::isOpen() const
{
return (m_socket && m_socket->isOpen());
}
void ManagementServer::stop()
{
m_tcpServer->close();
}
void ManagementServer::onAcceptError(QAbstractSocket::SocketError socketError)
{
qDebug().noquote() << QString("Accept error: %1").arg(socketError);
}
qint64 ManagementServer::writeCommand(const QString& message)
{
if (!isOpen()) {
return 0;
}
const QString command = message + "\n";
return m_socket->write(command.toStdString().c_str());
}
void ManagementServer::onNewConnection()
{
qDebug() << "New incoming connection";
m_socket = QPointer<QTcpSocket>(m_tcpServer->nextPendingConnection());
m_tcpServer->close();
QObject::connect(m_socket.data(), SIGNAL(disconnected()), this, SLOT(onSocketDisconnected()));
QObject::connect(m_socket.data(), SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(onSocketError(QAbstractSocket::SocketError)));
QObject::connect(m_socket.data(), SIGNAL(readyRead()), this, SLOT(onReadyRead()));
}
void ManagementServer::onSocketError(QAbstractSocket::SocketError socketError)
{
Q_UNUSED(socketError);
qDebug().noquote() << QString("Mananement server error: %1").arg(m_socket->errorString());
}
void ManagementServer::onSocketDisconnected()
{
if (m_socket) m_socket->deleteLater();
}
QPointer<QTcpSocket> ManagementServer::socket() const
{
if (!isOpen()) {
return nullptr;
}
return m_socket;
}
void ManagementServer::onReadyRead()
{
emit readyRead();
}
bool ManagementServer::start(const QString& host, unsigned int port)
{
if (m_tcpServer) {
m_tcpServer->deleteLater();
}
m_tcpServer = QSharedPointer<QTcpServer>(new QTcpServer(this), [](QTcpServer *s){
if (s) s->deleteLater();
});
m_tcpServer->setMaxPendingConnections(1);
connect(m_tcpServer.data(), SIGNAL(acceptError(QAbstractSocket::SocketError)), this, SLOT(onAcceptError(QAbstractSocket::SocketError)));
connect(m_tcpServer.data(), SIGNAL(newConnection()), this, SLOT(onNewConnection()));
if (m_tcpServer->listen(QHostAddress(host), port)) {
emit serverStarted();
return true;
}
qDebug().noquote() << QString("Can't start TCP server, %1,%2")
.arg(m_tcpServer->serverError())
.arg(m_tcpServer->errorString());
return false;
}
QString ManagementServer::readLine()
{
if (!isOpen()) {
qDebug() << "Socket is not opened";
return QString();
}
return m_socket->readLine();
}