157 lines
3.1 KiB
C++
157 lines
3.1 KiB
C++
#ifndef PROVIDERUDPSSL_H
|
|
#define PROVIDERUDPSSL_H
|
|
|
|
#include <leddevice/LedDevice.h>
|
|
#include <utils/Logger.h>
|
|
|
|
// Qt includes
|
|
#include <QMutex>
|
|
#include <QMutexLocker>
|
|
#include <QHostInfo>
|
|
#include <QThread>
|
|
|
|
//----------- mbedtls
|
|
#if defined(USE_MBEDTLS3)
|
|
#include <mbedtls/build_info.h>
|
|
#else
|
|
#if !defined(MBEDTLS_CONFIG_FILE)
|
|
#include <mbedtls/config.h>
|
|
#else
|
|
#include MBEDTLS_CONFIG_FILE
|
|
#endif
|
|
#endif
|
|
|
|
#if defined(MBEDTLS_PLATFORM_C)
|
|
#include <mbedtls/platform.h>
|
|
#endif
|
|
|
|
#include <string.h>
|
|
#include <cstring>
|
|
#include <chrono>
|
|
|
|
#include <mbedtls/net_sockets.h>
|
|
#include <mbedtls/ssl_ciphersuites.h>
|
|
#include <mbedtls/entropy.h>
|
|
#include <mbedtls/timing.h>
|
|
#include <mbedtls/ctr_drbg.h>
|
|
#include <mbedtls/error.h>
|
|
#include <mbedtls/debug.h>
|
|
|
|
class ProviderUdpSSL : public LedDevice
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
///
|
|
/// @brief Constructs an UDP SSL LED-device
|
|
///
|
|
ProviderUdpSSL(const QJsonObject &deviceConfig);
|
|
|
|
///
|
|
/// @brief Destructor of the LED-device
|
|
///
|
|
~ProviderUdpSSL() override;
|
|
|
|
///
|
|
QString _hostName;
|
|
QHostAddress _address;
|
|
int _port;
|
|
|
|
protected:
|
|
|
|
///
|
|
/// @brief Initialise the UDP-SSL device's configuration and network address details
|
|
///
|
|
/// @param[in] deviceConfig the JSON device configuration
|
|
/// @return True, if success#endif // PROVIDERUDP_H
|
|
///
|
|
bool init(const QJsonObject &deviceConfig) override;
|
|
|
|
///
|
|
/// @brief Opens the output device.
|
|
///
|
|
/// @return Zero on success (i.e. device is ready), else negative
|
|
///
|
|
int open() override;
|
|
|
|
///
|
|
/// @brief Closes the output device.
|
|
///
|
|
/// @return Zero on success (i.e. device is closed), else negative
|
|
///
|
|
int close() override;
|
|
|
|
///
|
|
/// @brief Initialise device's network details
|
|
///
|
|
/// @return True, if success
|
|
///
|
|
bool initNetwork();
|
|
|
|
///
|
|
/// @brief Start astreaming connection
|
|
///
|
|
/// @return True, if success
|
|
///
|
|
bool startConnection();
|
|
|
|
///
|
|
/// @brief Stop the streaming connection
|
|
///
|
|
void stopConnection();
|
|
|
|
///
|
|
/// Writes the given bytes/bits to the UDP-device and sleeps the latch time to ensure that the
|
|
/// values are latched.
|
|
///
|
|
/// @param[in] size The length of the data
|
|
/// @param[in] data The data
|
|
///
|
|
void writeBytes(unsigned int size, const uint8_t* data, bool flush = false);
|
|
|
|
///
|
|
/// get ciphersuites list from mbedtls_ssl_list_ciphersuites
|
|
///
|
|
/// @return const int * array
|
|
///
|
|
virtual const int * getCiphersuites() const;
|
|
|
|
private:
|
|
|
|
bool initConnection();
|
|
|
|
bool seedingRNG();
|
|
bool setupStructure();
|
|
|
|
bool setupPSK();
|
|
bool startSSLHandshake();
|
|
|
|
QString errorMsg(int ret);
|
|
void closeSSLNotify();
|
|
void freeSSLConnection();
|
|
|
|
mbedtls_net_context client_fd;
|
|
mbedtls_entropy_context entropy;
|
|
mbedtls_ssl_context ssl;
|
|
mbedtls_ssl_config conf;
|
|
mbedtls_x509_crt cacert;
|
|
mbedtls_ctr_drbg_context ctr_drbg;
|
|
mbedtls_timing_delay_context timer;
|
|
|
|
QString _transport_type;
|
|
QString _custom;
|
|
int _ssl_port;
|
|
QString _server_name;
|
|
QString _psk;
|
|
QString _psk_identity;
|
|
|
|
int _handshake_attempts;
|
|
uint32_t _handshake_timeout_min;
|
|
uint32_t _handshake_timeout_max;
|
|
|
|
bool _streamReady;
|
|
bool _streamPaused;
|
|
};
|
|
|
|
#endif // PROVIDERUDPSSL_H
|