124 lines
3.2 KiB
C++
124 lines
3.2 KiB
C++
/*
|
|
* Deskflow -- mouse and keyboard sharing utility
|
|
* SPDX-FileCopyrightText: (C) 2021 Symless Ltd.
|
|
* SPDX-License-Identifier: GPL-2.0-only WITH LicenseRef-OpenSSL-Exception
|
|
*/
|
|
|
|
#include "ServerConnection.h"
|
|
|
|
#include "Messages.h"
|
|
#include "ServerMessage.h"
|
|
#include "common/Settings.h"
|
|
|
|
#include <QMessageBox>
|
|
#include <QPushButton>
|
|
|
|
namespace deskflow::gui {
|
|
|
|
//
|
|
// ServerConnection::Deps
|
|
//
|
|
|
|
messages::NewClientPromptResult ServerConnection::Deps::showNewClientPrompt(
|
|
QWidget *parent, const QString &clientName, bool serverRequiresPeerAuth
|
|
) const
|
|
{
|
|
return messages::showNewClientPrompt(parent, clientName, serverRequiresPeerAuth);
|
|
}
|
|
|
|
//
|
|
// ServerConnection
|
|
//
|
|
|
|
ServerConnection::ServerConnection(QWidget *parent, IServerConfig &serverConfig, std::shared_ptr<Deps> deps)
|
|
: m_pParent(parent),
|
|
m_serverConfig(serverConfig),
|
|
m_pDeps(deps)
|
|
{
|
|
}
|
|
|
|
void ServerConnection::handleLogLine(const QString &logLine)
|
|
{
|
|
ServerMessage message(logLine);
|
|
const auto &clientName = message.getClientName();
|
|
|
|
if (message.isDisconnectedMessage()) {
|
|
m_connectedClients.remove(clientName);
|
|
Q_EMIT clientsChanged(connectedClients());
|
|
return;
|
|
}
|
|
|
|
if (message.isConnectedMessage()) {
|
|
m_connectedClients.insert(clientName);
|
|
Q_EMIT clientsChanged(connectedClients());
|
|
return;
|
|
}
|
|
|
|
if (!message.isNewClientMessage()) {
|
|
return;
|
|
}
|
|
|
|
if (m_messageShowing) {
|
|
qDebug("new client message already shown, skipping for now");
|
|
return;
|
|
}
|
|
|
|
if (m_serverConfigDialogVisible) {
|
|
qDebug("server config dialog visible, skipping new client prompt");
|
|
return;
|
|
}
|
|
|
|
if (Settings::value(Settings::Server::ExternalConfig).toBool()) {
|
|
qDebug("external config enabled, skipping new client prompt");
|
|
return;
|
|
}
|
|
|
|
if (m_connectedClients.contains(clientName)) {
|
|
qDebug("already got request, skipping new client prompt for: %s", qPrintable(clientName));
|
|
return;
|
|
}
|
|
|
|
handleNewClient(clientName);
|
|
}
|
|
|
|
void ServerConnection::handleNewClient(const QString &clientName)
|
|
{
|
|
using enum messages::NewClientPromptResult;
|
|
|
|
if (m_serverConfig.isFull()) {
|
|
qDebug("server config full, skipping new client prompt for: %s", qPrintable(clientName));
|
|
return;
|
|
}
|
|
|
|
if (m_serverConfig.screenExists(clientName)) {
|
|
qDebug("client already added, skipping new client prompt for: %s", qPrintable(clientName));
|
|
return;
|
|
}
|
|
|
|
Q_EMIT messageShowing();
|
|
|
|
m_messageShowing = true;
|
|
const bool tlsEnabled = Settings::value(Settings::Security::TlsEnabled).toBool();
|
|
const bool requireCerts = Settings::value(Settings::Security::CheckPeers).toBool();
|
|
const auto result = m_pDeps->showNewClientPrompt(m_pParent, clientName, tlsEnabled && requireCerts);
|
|
m_messageShowing = false;
|
|
|
|
if (result == Add) {
|
|
qDebug("accepted dialog, adding client: %s", qPrintable(clientName));
|
|
Q_EMIT configureClient(clientName);
|
|
} else if (result == Ignore) {
|
|
qDebug("declined dialog, ignoring client: %s", qPrintable(clientName));
|
|
} else {
|
|
qFatal("unexpected add client result");
|
|
}
|
|
|
|
m_connectedClients.insert(clientName);
|
|
Q_EMIT clientsChanged(connectedClients());
|
|
}
|
|
|
|
QStringList ServerConnection::connectedClients() const
|
|
{
|
|
return QStringList(m_connectedClients.begin(), m_connectedClients.end());
|
|
}
|
|
|
|
} // namespace deskflow::gui
|