mirror of https://github.com/nymea/nymea.git
358 lines
13 KiB
C++
358 lines
13 KiB
C++
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
|
|
* *
|
|
* Copyright (C) 2015 Simon Stuerz <simon.stuerz@guh.guru> *
|
|
* Copyright (C) 2014 Michael Zanetti <michael_zanetti@gmx.net> *
|
|
* *
|
|
* This file is part of guh. *
|
|
* *
|
|
* Guh is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation, version 2 of the License. *
|
|
* *
|
|
* Guh is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with guh. If not, see <http://www.gnu.org/licenses/>. *
|
|
* *
|
|
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
|
|
|
|
#include "tcpserver.h"
|
|
#include "loggingcategories.h"
|
|
#include "guhsettings.h"
|
|
#include "guhcore.h"
|
|
#include "jsonrpcserver.h"
|
|
|
|
#include <QDebug>
|
|
#include <QJsonDocument>
|
|
|
|
namespace guhserver {
|
|
|
|
TcpServer::TcpServer(QObject *parent) :
|
|
TransportInterface(parent)
|
|
{
|
|
// Timer for scanning network interfaces ever 5 seconds
|
|
// Note: QNetworkConfigurationManager does not work on embedded platforms
|
|
m_timer = new QTimer(this);
|
|
m_timer->setSingleShot(false);
|
|
m_timer->setInterval(5000);
|
|
connect (m_timer, &QTimer::timeout, this, &TcpServer::onTimeout);
|
|
|
|
// load JSON-RPC server settings
|
|
GuhSettings settings(GuhSettings::SettingsRoleGlobal);
|
|
qCDebug(dcTcpServer) << "Loading Tcp server settings from:" << settings.fileName();
|
|
settings.beginGroup("JSONRPC");
|
|
|
|
// load port
|
|
m_port = settings.value("port", 1234).toUInt();
|
|
|
|
// load interfaces
|
|
QStringList interfaceList = settings.value("interfaces", QStringList("all")).toStringList();
|
|
if (interfaceList.contains("all")) {
|
|
m_networkInterfaces = QNetworkInterface::allInterfaces();
|
|
} else {
|
|
foreach (const QNetworkInterface &interface, QNetworkInterface::allInterfaces()) {
|
|
if (interfaceList.contains(interface.name())) {
|
|
m_networkInterfaces.append(interface);
|
|
}
|
|
}
|
|
}
|
|
|
|
// load IP versions (IPv4, IPv6 or any)
|
|
m_ipVersions = settings.value("ip", QStringList("any")).toStringList();
|
|
settings.endGroup();
|
|
}
|
|
|
|
void TcpServer::sendData(const QList<QUuid> &clients, const QVariantMap &data)
|
|
{
|
|
foreach (const QUuid &client, clients) {
|
|
sendData(client, data);
|
|
}
|
|
}
|
|
|
|
void TcpServer::reloadNetworkInterfaces()
|
|
{
|
|
GuhSettings settings(GuhSettings::SettingsRoleGlobal);
|
|
settings.beginGroup("JSONRPC");
|
|
|
|
// reload network interfaces
|
|
m_networkInterfaces.clear();
|
|
|
|
QStringList interfaceList = settings.value("interfaces", QStringList("all")).toStringList();
|
|
if (interfaceList.contains("all")) {
|
|
m_networkInterfaces = QNetworkInterface::allInterfaces();
|
|
} else {
|
|
foreach (const QNetworkInterface &interface, QNetworkInterface::allInterfaces()) {
|
|
if (interfaceList.contains(interface.name())) {
|
|
m_networkInterfaces.append(interface);
|
|
}
|
|
}
|
|
}
|
|
settings.endGroup();
|
|
}
|
|
|
|
|
|
void TcpServer::validateMessage(const QUuid &clientId, const QByteArray &data)
|
|
{
|
|
QJsonParseError error;
|
|
QJsonDocument jsonDoc = QJsonDocument::fromJson(data, &error);
|
|
|
|
if(error.error != QJsonParseError::NoError) {
|
|
qCWarning(dcJsonRpc) << "Failed to parse JSON data" << data << ":" << error.errorString();
|
|
sendErrorResponse(clientId, -1, QString("Failed to parse JSON data: %1").arg(error.errorString()));
|
|
return;
|
|
}
|
|
|
|
QVariantMap message = jsonDoc.toVariant().toMap();
|
|
|
|
bool success;
|
|
int commandId = message.value("id").toInt(&success);
|
|
if (!success) {
|
|
qCWarning(dcJsonRpc) << "Error parsing command. Missing \"id\":" << message;
|
|
sendErrorResponse(clientId, commandId, "Error parsing command. Missing 'id'");
|
|
return;
|
|
}
|
|
|
|
QStringList commandList = message.value("method").toString().split('.');
|
|
if (commandList.count() != 2) {
|
|
qCWarning(dcJsonRpc) << "Error parsing method.\nGot:" << message.value("method").toString() << "\nExpected: \"Namespace.method\"";
|
|
sendErrorResponse(clientId, commandId, QString("Error parsing method. Got: '%1'', Expected: 'Namespace.method'").arg(message.value("method").toString()));
|
|
return;
|
|
}
|
|
|
|
QString targetNamespace = commandList.first();
|
|
QString method = commandList.last();
|
|
|
|
JsonHandler *handler = GuhCore::instance()->jsonRPCServer()->handlers().value(targetNamespace);
|
|
if (!handler) {
|
|
sendErrorResponse(clientId, commandId, "No such namespace");
|
|
return;
|
|
}
|
|
if (!handler->hasMethod(method)) {
|
|
sendErrorResponse(clientId, commandId, "No such method");
|
|
return;
|
|
}
|
|
|
|
emit dataAvailable(clientId, targetNamespace, method, message);
|
|
}
|
|
|
|
void TcpServer::sendResponse(const QUuid &clientId, int commandId, const QVariantMap ¶ms)
|
|
{
|
|
QVariantMap response;
|
|
response.insert("id", commandId);
|
|
response.insert("status", "success");
|
|
response.insert("params", params);
|
|
|
|
sendData(clientId, response);
|
|
}
|
|
|
|
void TcpServer::sendErrorResponse(const QUuid &clientId, int commandId, const QString &error)
|
|
{
|
|
QVariantMap errorResponse;
|
|
errorResponse.insert("id", commandId);
|
|
errorResponse.insert("status", "error");
|
|
errorResponse.insert("error", error);
|
|
|
|
sendData(clientId, errorResponse);
|
|
}
|
|
|
|
|
|
void TcpServer::sendData(const QUuid &clientId, const QVariantMap &data)
|
|
{
|
|
QTcpSocket *client = m_clientList.value(clientId);
|
|
if (client) {
|
|
client->write(QJsonDocument::fromVariant(data).toJson());
|
|
}
|
|
}
|
|
|
|
void TcpServer::newClientConnected()
|
|
{
|
|
// got a new client connected
|
|
QTcpServer *server = qobject_cast<QTcpServer*>(sender());
|
|
QTcpSocket *newConnection = server->nextPendingConnection();
|
|
qCDebug(dcConnection) << "Tcp server: new client connected:" << newConnection->peerAddress().toString();
|
|
|
|
QUuid clientId = QUuid::createUuid();
|
|
|
|
// append the new client to the client list
|
|
m_clientList.insert(clientId, newConnection);
|
|
|
|
connect(newConnection, SIGNAL(readyRead()),this,SLOT(readPackage()));
|
|
connect(newConnection,SIGNAL(disconnected()),this,SLOT(onClientDisconnected()));
|
|
|
|
emit clientConnected(clientId);
|
|
}
|
|
|
|
|
|
void TcpServer::readPackage()
|
|
{
|
|
QTcpSocket *client = qobject_cast<QTcpSocket*>(sender());
|
|
qCDebug(dcTcpServer) << "data comming from" << client->peerAddress().toString();
|
|
QByteArray message;
|
|
while (client->canReadLine()) {
|
|
QByteArray dataLine = client->readLine();
|
|
qCDebug(dcTcpServer) << "line in:" << dataLine;
|
|
message.append(dataLine);
|
|
if (dataLine.endsWith('\n')) {
|
|
validateMessage(m_clientList.key(client), message);
|
|
message.clear();
|
|
}
|
|
}
|
|
}
|
|
|
|
void TcpServer::onClientDisconnected()
|
|
{
|
|
QTcpSocket *client = qobject_cast<QTcpSocket*>(sender());
|
|
qCDebug(dcConnection) << "Tcp server: client disconnected:" << client->peerAddress().toString();
|
|
QUuid clientId = m_clientList.key(client);
|
|
m_clientList.take(clientId)->deleteLater();
|
|
}
|
|
|
|
void TcpServer::onError(const QAbstractSocket::SocketError &error)
|
|
{
|
|
QTcpServer *server = qobject_cast<QTcpServer *>(sender());
|
|
QUuid uuid = m_serverList.key(server);
|
|
qCWarning(dcTcpServer) << "Tcp server" << server->serverAddress().toString() << "error:" << error << server->errorString();
|
|
qCWarning(dcTcpServer) << "shutting down Tcp server" << server->serverAddress().toString();
|
|
|
|
server->close();
|
|
m_serverList.take(uuid)->deleteLater();
|
|
}
|
|
|
|
void TcpServer::onTimeout()
|
|
{
|
|
bool ipV4 = m_ipVersions.contains("IPv4");
|
|
bool ipV6 = m_ipVersions.contains("IPv6");
|
|
if (m_ipVersions.contains("any")) {
|
|
ipV4 = ipV6 = true;
|
|
}
|
|
|
|
reloadNetworkInterfaces();
|
|
|
|
// get all available host addresses
|
|
QList<QHostAddress> allAddresses;
|
|
foreach (const QNetworkInterface &interface, m_networkInterfaces) {
|
|
QList<QNetworkAddressEntry> addresseEntries = interface.addressEntries();
|
|
|
|
foreach (QNetworkAddressEntry entry, addresseEntries) {
|
|
if (ipV4 && entry.ip().protocol() == QAbstractSocket::IPv4Protocol) {
|
|
allAddresses.append(entry.ip());
|
|
}
|
|
if (ipV6 && entry.ip().protocol() == QAbstractSocket::IPv6Protocol) {
|
|
allAddresses.append(entry.ip());
|
|
}
|
|
}
|
|
}
|
|
|
|
// check if a new server should be created
|
|
QList<QHostAddress> serversToCreate;
|
|
QList<QTcpServer *> serversToDelete;
|
|
|
|
foreach (const QHostAddress &address, allAddresses) {
|
|
// check if there is a server for this host address
|
|
bool found = false;
|
|
foreach (QTcpServer *s, m_serverList) {
|
|
if (s->serverAddress() == address) {
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
if (!found)
|
|
serversToCreate.append(address);
|
|
}
|
|
|
|
// delete every server which has no longer an interface
|
|
foreach (QTcpServer *s, m_serverList) {
|
|
if (!allAddresses.contains(s->serverAddress())) {
|
|
qCDebug(dcConnection) << "Closing Tcp server on" << s->serverAddress().toString() << s->serverPort();
|
|
QUuid uuid = m_serverList.key(s);
|
|
s->close();
|
|
m_serverList.take(uuid)->deleteLater();
|
|
}
|
|
|
|
}
|
|
|
|
foreach(const QHostAddress &address, serversToCreate){
|
|
QTcpServer *server = new QTcpServer(this);
|
|
if(server->listen(address, m_port)) {
|
|
qCDebug(dcConnection) << "Started TCP server on" << address.toString() << m_port;
|
|
connect(server, SIGNAL(newConnection()), SLOT(newClientConnected()));
|
|
m_serverList.insert(QUuid::createUuid(), server);
|
|
} else {
|
|
qCWarning(dcConnection) << "Tcp server error: can not listen on" << address.toString() << m_port;
|
|
delete server;
|
|
}
|
|
}
|
|
|
|
if (!serversToCreate.isEmpty() || !serversToDelete.isEmpty()) {
|
|
qCDebug(dcConnection) << "Current server list:";
|
|
foreach (QTcpServer *s, m_serverList) {
|
|
qCDebug(dcConnection) << " - Tcp server on" << s->serverAddress().toString() << s->serverPort();
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
bool TcpServer::startServer()
|
|
{
|
|
bool ipV4 = m_ipVersions.contains("IPv4");
|
|
bool ipV6 = m_ipVersions.contains("IPv6");
|
|
if (m_ipVersions.contains("any")) {
|
|
ipV4 = ipV6 = true;
|
|
}
|
|
|
|
foreach (const QNetworkInterface &interface, m_networkInterfaces) {
|
|
QList<QNetworkAddressEntry> addresseEntries = interface.addressEntries();
|
|
QList<QHostAddress> addresses;
|
|
|
|
foreach (QNetworkAddressEntry entry, addresseEntries) {
|
|
if (ipV4 && entry.ip().protocol() == QAbstractSocket::IPv4Protocol) {
|
|
addresses.append(entry.ip());
|
|
}
|
|
if (ipV6 && entry.ip().protocol() == QAbstractSocket::IPv6Protocol) {
|
|
addresses.append(entry.ip());
|
|
}
|
|
}
|
|
|
|
// create a tcp server for each address found for the corresponding settings
|
|
foreach(const QHostAddress &address, addresses){
|
|
QTcpServer *server = new QTcpServer(this);
|
|
if(server->listen(address, m_port)) {
|
|
qCDebug(dcConnection) << "Started Tcp server on" << server->serverAddress().toString() << server->serverPort();
|
|
connect(server, SIGNAL(newConnection()), SLOT(newClientConnected()));
|
|
m_serverList.insert(QUuid::createUuid(), server);
|
|
} else {
|
|
qCWarning(dcConnection) << "Tcp server error: can not listen on" << interface.name() << address.toString() << m_port;
|
|
delete server;
|
|
}
|
|
}
|
|
}
|
|
|
|
m_timer->start();
|
|
|
|
if (m_serverList.empty())
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool TcpServer::stopServer()
|
|
{
|
|
// Listen on all Networkinterfaces
|
|
foreach (QTcpServer *s, m_serverList) {
|
|
qCDebug(dcConnection) << "Closing Tcp server on" << s->serverAddress().toString() << s->serverPort();
|
|
QUuid uuid = m_serverList.key(s);
|
|
s->close();
|
|
m_serverList.take(uuid)->deleteLater();
|
|
}
|
|
|
|
if (!m_serverList.empty())
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
}
|