hyperion.ng/libsrc/leddevice/dev_net/ProviderUdp.cpp

// STL includes
#include <cstring>
#include <cstdio>
#include <iostream>
#include <exception>
// Linux includes
#include <fcntl.h>

#include <QStringList>
#include <QUdpSocket>
#include <QHostInfo>

// 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 = 1;
}

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<int>(config_port);
				Debug( _log, "UDP socket will write to %s:%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)
{
	qint64 retVal = _udpSocket->writeDatagram((const char *)data,size,_address,_port);

	WarningIf((retVal<0), _log, "&s", QSTRING_CSTR(QString
								("(%1:%2) Write Error: (%3) %4").arg(_address.toString()).arg(_port).arg(_udpSocket->error()).arg(_udpSocket->errorString())));

	return retVal;
}