// system includes #include // project includes #include #include #include "protoserver/ProtoConnection.h" #include "ProtoClientConnection.h" ProtoServer::ProtoServer(uint16_t port) : QObject() , _hyperion(Hyperion::getInstance()) , _server() , _openConnections() , _log(Logger::getInstance("PROTOSERVER")) , _forwarder_enabled(true) { 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("PROTOSERVER 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("PROTOSERVER ERROR: Could not bind to port"); } // Set trigger for incoming connections connect(&_server, SIGNAL(newConnection()), this, SLOT(newConnection())); connect( _hyperion, SIGNAL(componentStateChanged(hyperion::Components,bool)), this, SLOT(componentStateChanged(hyperion::Components,bool))); } 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) { Debug(_log, "New connection"); ProtoClientConnection * connection = new ProtoClientConnection(socket); _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*))); // register forward signal for video mode connect(this, SIGNAL(videoMode(VideoMode)), connection, SLOT(setVideoMode(VideoMode))); } } 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 & image, int duration_ms) { if ( _forwarder_enabled ) { for (int i = 0; i < _proxy_connections.size(); ++i) _proxy_connections.at(i)->setImage(image, priority, duration_ms); } } void ProtoServer::componentStateChanged(const hyperion::Components component, bool enable) { if (component == hyperion::COMP_FORWARDER) { if (_forwarder_enabled != enable) { _forwarder_enabled = enable; Info(_log, "forwarder change state to %s", (_forwarder_enabled ? "enabled" : "disabled") ); } _hyperion->getComponentRegister().componentStateChanged(component, _forwarder_enabled); } } void ProtoServer::closedConnection(ProtoClientConnection *connection) { Debug(_log, "Connection closed"); _openConnections.remove(connection); // schedule to delete the connection object connection->deleteLater(); }