// STL includes #include #include #include #include // Linux includes #include #include #include #include // Local Hyperion includes #include "ProviderUdp.h" const ushort MAX_PORT = 65535; ProviderUdp::ProviderUdp(const QJsonObject& deviceConfig) : LedDevice(deviceConfig) , _udpSocket(nullptr) , _port(1) , _defaultHost("127.0.0.1") { _latchTime_ms = 0; } ProviderUdp::~ProviderUdp() { delete _udpSocket; } bool ProviderUdp::init(const QJsonObject& deviceConfig) { bool isInitOK = false; // Initialise sub-class if (LedDevice::init(deviceConfig)) { QString host = deviceConfig["host"].toString(_defaultHost); if (_address.setAddress(host)) { Debug(_log, "Successfully parsed %s as an IP-address.", QSTRING_CSTR(_address.toString())); } else { QHostInfo hostInfo = QHostInfo::fromName(host); if (hostInfo.error() == QHostInfo::NoError) { _address = hostInfo.addresses().first(); Debug(_log, "Successfully resolved IP-address (%s) for hostname (%s).", QSTRING_CSTR(_address.toString()), QSTRING_CSTR(host)); } else { QString errortext = QString("Failed resolving IP-address for [%1], (%2) %3").arg(host).arg(hostInfo.error()).arg(hostInfo.errorString()); this->setInError(errortext); isInitOK = false; } } if (!_isDeviceInError) { int config_port = deviceConfig["port"].toInt(_port); if (config_port <= 0 || config_port > MAX_PORT) { QString errortext = QString("Invalid target port [%1]!").arg(config_port); this->setInError(errortext); isInitOK = false; } else { _port = static_cast(config_port); Debug(_log, "UDP socket will write to %s port: %u", QSTRING_CSTR(_address.toString()), _port); _udpSocket = new QUdpSocket(this); isInitOK = true; } } } return isInitOK; } int ProviderUdp::open() { int retval = -1; _isDeviceReady = false; // Try to bind the UDP-Socket if (_udpSocket != nullptr) { if (_udpSocket->state() != QAbstractSocket::BoundState) { QHostAddress localAddress = QHostAddress::Any; quint16 localPort = 0; if (!_udpSocket->bind(localAddress, localPort)) { QString warntext = QString("Could not bind local address: %1, (%2) %3").arg(localAddress.toString()).arg(_udpSocket->error()).arg(_udpSocket->errorString()); Warning(_log, "%s", QSTRING_CSTR(warntext)); } } // Everything is OK, device is ready _isDeviceReady = true; retval = 0; } else { this->setInError(" Open error. UDP Socket not initialised!"); } return retval; } int ProviderUdp::close() { int retval = 0; _isDeviceReady = false; if (_udpSocket != nullptr) { // Test, if device requires closing if (_udpSocket->isOpen()) { Debug(_log, "Close UDP-device: %s", QSTRING_CSTR(this->getActiveDeviceType())); _udpSocket->close(); // Everything is OK -> device is closed } } return retval; } int ProviderUdp::writeBytes(const unsigned size, const uint8_t* data) { int rc = 0; qint64 bytesWritten = _udpSocket->writeDatagram(reinterpret_cast(data), size, _address, _port); if (bytesWritten == -1 || bytesWritten != size) { Warning(_log, "%s", QSTRING_CSTR(QString("(%1:%2) Write Error: (%3) %4").arg(_address.toString()).arg(_port).arg(_udpSocket->error()).arg(_udpSocket->errorString()))); rc = -1; } return rc; } int ProviderUdp::writeBytes(const QByteArray& bytes) { int rc = 0; qint64 bytesWritten = _udpSocket->writeDatagram(bytes, _address, _port); if (bytesWritten == -1 || bytesWritten != bytes.size()) { Warning(_log, "%s", QSTRING_CSTR(QString("(%1:%2) Write Error: (%3) %4").arg(_address.toString()).arg(_port).arg(_udpSocket->error()).arg(_udpSocket->errorString()))); rc = -1; } return rc; }