mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
0f17e031e8
But this doesn't protect if not forwarded to localhost "127.0.0.1". Loop connections across different hosts arn't detected too. Former-commit-id: 464a80708ebd11c0f7c83dc87a1afe0f10e0e078
91 lines
2.5 KiB
C++
91 lines
2.5 KiB
C++
// system includes
|
|
#include <stdexcept>
|
|
|
|
// project includes
|
|
#include <hyperion/MessageForwarder.h>
|
|
#include <protoserver/ProtoServer.h>
|
|
#include "protoserver/ProtoConnection.h"
|
|
#include "ProtoClientConnection.h"
|
|
|
|
ProtoServer::ProtoServer(Hyperion *hyperion, uint16_t port) :
|
|
QObject(),
|
|
_hyperion(hyperion),
|
|
_server(),
|
|
_openConnections()
|
|
{
|
|
|
|
MessageForwarder * forwarder = hyperion->getForwarder();
|
|
QStringList slaves = forwarder->getProtoSlaves();
|
|
|
|
for (int i = 0; i < slaves.size(); ++i) {
|
|
if ( QString("127.0.0.1:%1").arg(port) == slaves.at(i) ) {
|
|
throw std::runtime_error("Loop between proto server and forwarder detected. Fix your config!");
|
|
}
|
|
|
|
ProtoConnection* p = new ProtoConnection(slaves.at(i).toLocal8Bit().constData());
|
|
p->setSkipReply(true);
|
|
_proxy_connections << p;
|
|
}
|
|
|
|
if (!_server.listen(QHostAddress::Any, port))
|
|
{
|
|
throw std::runtime_error("Proto server could not bind to port");
|
|
}
|
|
|
|
// Set trigger for incoming connections
|
|
connect(&_server, SIGNAL(newConnection()), this, SLOT(newConnection()));
|
|
}
|
|
|
|
ProtoServer::~ProtoServer()
|
|
{
|
|
foreach (ProtoClientConnection * connection, _openConnections) {
|
|
delete connection;
|
|
}
|
|
|
|
while (!_proxy_connections.isEmpty())
|
|
delete _proxy_connections.takeFirst();
|
|
}
|
|
|
|
uint16_t ProtoServer::getPort() const
|
|
{
|
|
return _server.serverPort();
|
|
}
|
|
|
|
void ProtoServer::newConnection()
|
|
{
|
|
QTcpSocket * socket = _server.nextPendingConnection();
|
|
|
|
if (socket != nullptr)
|
|
{
|
|
std::cout << "New proto connection" << std::endl;
|
|
ProtoClientConnection * connection = new ProtoClientConnection(socket, _hyperion);
|
|
_openConnections.insert(connection);
|
|
|
|
// register slot for cleaning up after the connection closed
|
|
connect(connection, SIGNAL(connectionClosed(ProtoClientConnection*)), this, SLOT(closedConnection(ProtoClientConnection*)));
|
|
connect(connection, SIGNAL(newMessage(const proto::HyperionRequest*)), this, SLOT(newMessage(const proto::HyperionRequest*)));
|
|
|
|
}
|
|
}
|
|
|
|
void ProtoServer::newMessage(const proto::HyperionRequest * message)
|
|
{
|
|
for (int i = 0; i < _proxy_connections.size(); ++i)
|
|
_proxy_connections.at(i)->sendMessage(*message);
|
|
}
|
|
|
|
void ProtoServer::sendImageToProtoSlaves(int priority, const Image<ColorRgb> & image, int duration_ms)
|
|
{
|
|
for (int i = 0; i < _proxy_connections.size(); ++i)
|
|
_proxy_connections.at(i)->setImage(image, priority, duration_ms);
|
|
}
|
|
|
|
void ProtoServer::closedConnection(ProtoClientConnection *connection)
|
|
{
|
|
std::cout << "Proto connection closed" << std::endl;
|
|
_openConnections.remove(connection);
|
|
|
|
// schedule to delete the connection object
|
|
connection->deleteLater();
|
|
}
|