kdeconnect-kde/core/backends/lan/mdns_wrapper.h
Albert Vaca Cintora fd25318a20 Add MDNS e2e test
2023-08-28 18:39:30 +02:00

102 lines
2.6 KiB
C++

/**
* SPDX-FileCopyrightText: 2023 Albert Vaca Cintora <albertvaka@gmail.com>
*
* SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
*/
#ifndef KDECONNECT_MDNS_WRAPPER_H
#define KDECONNECT_MDNS_WRAPPER_H
#include <QHostAddress>
#include <QMap>
#include <QSocketNotifier>
#include <QString>
#include <QVector>
#include "kdeconnectcore_export.h"
/**
* A Qt wrapper for the mdns.h header-only library
* from https://github.com/mjansson/mdns
*/
namespace MdnsWrapper
{
class KDECONNECTCORE_EXPORT Discoverer : public QObject
{
Q_OBJECT
public:
struct MdnsService {
QString name; // The instance-name part in "<instance-name>._<service-type>._tcp.local."
uint16_t port;
QHostAddress address; // An IPv4 address (IPv6 addresses are ignored)
QMap<QString, QString> txtRecords;
};
// serviceType must be of the form "_<service-type>._<tcp/udp>.local"
void startDiscovering(const QString &serviceType);
void stopDiscovering();
void sendQuery(const QString &serviceType);
Q_SIGNALS:
void serviceFound(const MdnsWrapper::Discoverer::MdnsService &service);
private:
int listenForQueryResponses();
void stopListeningForQueryResponses();
QVector<QSocketNotifier *> responseSocketNotifiers;
};
class KDECONNECTCORE_EXPORT Announcer : public QObject
{
Q_OBJECT
public:
struct AnnouncedInfo {
QByteArray serviceType; // ie: "_<service-type>._tcp.local."
QByteArray serviceInstance; // ie: "<instance-name>._<service-type>._tcp.local."
QByteArray hostname; // ie: "<hostname>.local."
QVector<QHostAddress> addressesV4;
QVector<QHostAddress> addressesV6;
uint16_t port;
QHash<QByteArray, QByteArray> txtRecords;
};
// serviceType must be of the form "_<service-type>._<tcp/udp>.local"
Announcer(const QString &instanceName, const QString &serviceType, uint16_t port);
void putTxtRecord(const QString &key, const QString &value)
{
self.txtRecords[key.toLatin1()] = value.toLatin1();
}
void startAnnouncing();
void stopAnnouncing();
void sendMulticastAnnounce(bool isGoodbye);
public Q_SLOTS:
// notify that network interfaces changed since the creation of this Announcer
void onNetworkChange()
{
detectHostAddresses();
}
private:
int listenForQueries();
void stopListeningForQueries();
void detectHostAddresses();
AnnouncedInfo self;
QSocketNotifier *socketNotifier = nullptr;
QSocketNotifier *socketNotifierV6 = nullptr;
};
} // namespace MdnsWrapper
#endif