ac179d0f70
without keeping track of the avahi service resolvers and destroying them when the parent object goes away, callbacks might be called after after the dhe destruction and passing an invalid userdata pointer.
219 lines
9.0 KiB
C++
219 lines
9.0 KiB
C++
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
|
|
* *
|
|
* Copyright (C) 2016 Simon Stürz <simon.stuerz@guh.io> *
|
|
* *
|
|
* This file is part of guh. *
|
|
* *
|
|
* This library is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU Lesser General Public *
|
|
* License as published by the Free Software Foundation; either *
|
|
* version 2.1 of the License, or (at your option) any later version. *
|
|
* *
|
|
* This library is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU *
|
|
* Lesser General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Lesser General Public *
|
|
* License along with this library; If not, see *
|
|
* <http://www.gnu.org/licenses/>. *
|
|
* *
|
|
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
|
|
|
|
#include "qtavahiservicebrowser_p.h"
|
|
#include "qtavahiservicebrowser.h"
|
|
#include "avahiserviceentry.h"
|
|
#include "loggingcategories.h"
|
|
|
|
#include <avahi-common/strlst.h>
|
|
#include <avahi-common/error.h>
|
|
|
|
QtAvahiServiceBrowserPrivate::QtAvahiServiceBrowserPrivate(QtAvahiClient *client) :
|
|
client(client),
|
|
serviceTypeBrowser(NULL)
|
|
{
|
|
|
|
}
|
|
|
|
QtAvahiServiceBrowserPrivate::~QtAvahiServiceBrowserPrivate()
|
|
{
|
|
foreach (AvahiServiceResolver *resolver, m_serviceResolvers) {
|
|
avahi_service_resolver_free(resolver);
|
|
}
|
|
}
|
|
|
|
void QtAvahiServiceBrowserPrivate::callbackServiceTypeBrowser(AvahiServiceTypeBrowser *browser, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *type, const char *domain, AvahiLookupResultFlags flags, void *userdata)
|
|
{
|
|
Q_UNUSED(browser)
|
|
Q_UNUSED(interface)
|
|
Q_UNUSED(protocol)
|
|
Q_UNUSED(domain)
|
|
Q_UNUSED(flags)
|
|
|
|
QtAvahiServiceBrowser *serviceBrowser = static_cast<QtAvahiServiceBrowser *>(userdata);
|
|
if (!serviceBrowser)
|
|
return;
|
|
|
|
switch (event) {
|
|
case AVAHI_BROWSER_NEW:
|
|
if (!serviceBrowser->m_serviceTypes.contains(type)) {
|
|
serviceBrowser->m_serviceTypes.append(type);
|
|
qCDebug(dcAvahi()) << "Create service type browser for" << type;
|
|
serviceBrowser->createServiceBrowser(type);
|
|
}
|
|
break;
|
|
case AVAHI_BROWSER_REMOVE:
|
|
// Note: the browser for this serviceType will be deleted once all
|
|
// services from this type are removed
|
|
break;
|
|
case AVAHI_BROWSER_CACHE_EXHAUSTED:
|
|
break;
|
|
case AVAHI_BROWSER_ALL_FOR_NOW:
|
|
break;
|
|
case AVAHI_BROWSER_FAILURE:
|
|
qCWarning(dcAvahi()) << "Service type browser error:" << QString(avahi_strerror(avahi_client_errno(serviceBrowser->d_ptr->client->client)));
|
|
break;
|
|
}
|
|
}
|
|
|
|
void QtAvahiServiceBrowserPrivate::callbackServiceBrowser(AvahiServiceBrowser *browser, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AvahiLookupResultFlags flags, void *userdata)
|
|
{
|
|
Q_UNUSED(browser);
|
|
Q_UNUSED(flags);
|
|
|
|
QtAvahiServiceBrowser *serviceBrowser = static_cast<QtAvahiServiceBrowser *>(userdata);
|
|
if (!serviceBrowser)
|
|
return;
|
|
|
|
switch (event) {
|
|
case AVAHI_BROWSER_NEW: {
|
|
// Start resolving new service
|
|
AvahiServiceResolver *resolver = avahi_service_resolver_new(serviceBrowser->d_ptr->client->client,
|
|
interface,
|
|
protocol,
|
|
name,
|
|
type,
|
|
domain,
|
|
AVAHI_PROTO_UNSPEC,
|
|
(AvahiLookupFlags) 0,
|
|
QtAvahiServiceBrowserPrivate::callbackServiceResolver,
|
|
serviceBrowser);
|
|
if (resolver) {
|
|
serviceBrowser->d_ptr->m_serviceResolvers.append(resolver);
|
|
} else {
|
|
qCWarning(dcAvahi()) << "Failed to resolve service" << QString(name) << ":" << avahi_strerror(avahi_client_errno(serviceBrowser->d_ptr->client->client));
|
|
}
|
|
break;
|
|
}
|
|
case AVAHI_BROWSER_REMOVE: {
|
|
// Remove the service
|
|
foreach (const AvahiServiceEntry &entry, serviceBrowser->m_serviceEntries) {
|
|
// Check not only the name, but also the protocol
|
|
if (entry.name() == name && entry.protocol() == QtAvahiServiceBrowserPrivate::convertProtocol(protocol)) {
|
|
serviceBrowser->m_serviceEntries.removeAll(entry);
|
|
emit serviceBrowser->serviceEntryRemoved(entry);
|
|
}
|
|
}
|
|
|
|
// Check if this was the last service for this serviceType
|
|
foreach (const AvahiServiceEntry &entry, serviceBrowser->m_serviceEntries) {
|
|
if (entry.serviceType() == type)
|
|
return;
|
|
}
|
|
|
|
// This was the last service for this serviceType, lets delete the corresponding browser
|
|
AvahiServiceBrowser *browser = serviceBrowser->d_ptr->serviceBrowserTable.take(type);
|
|
if (browser)
|
|
avahi_service_browser_free(browser);
|
|
|
|
serviceBrowser->m_serviceTypes.removeAll(type);
|
|
break;
|
|
}
|
|
case AVAHI_BROWSER_ALL_FOR_NOW:
|
|
break;
|
|
case AVAHI_BROWSER_CACHE_EXHAUSTED:
|
|
break;
|
|
case AVAHI_BROWSER_FAILURE:
|
|
qCWarning(dcAvahi()) << "Service browser error:" << QString(avahi_strerror(avahi_client_errno(serviceBrowser->d_ptr->client->client)));
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
void QtAvahiServiceBrowserPrivate::callbackServiceResolver(AvahiServiceResolver *resolver, AvahiIfIndex interface, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const char *type, const char *domain, const char *host_name, const AvahiAddress *address, uint16_t port, AvahiStringList *txt, AvahiLookupResultFlags flags, void *userdata)
|
|
{
|
|
Q_UNUSED(interface);
|
|
Q_UNUSED(type);
|
|
Q_UNUSED(txt);
|
|
|
|
QPointer<QtAvahiServiceBrowser> serviceBrowser = static_cast<QtAvahiServiceBrowser *>(userdata);
|
|
if (serviceBrowser.isNull())
|
|
return;
|
|
|
|
switch (event) {
|
|
case AVAHI_RESOLVER_FAILURE:
|
|
break;
|
|
case AVAHI_RESOLVER_FOUND: {
|
|
char a[AVAHI_ADDRESS_STR_MAX];
|
|
avahi_address_snprint(a, sizeof(a), address);
|
|
|
|
// convert protocol
|
|
QAbstractSocket::NetworkLayerProtocol networkProtocol = QtAvahiServiceBrowserPrivate::convertProtocol(protocol);
|
|
QStringList txtList = QtAvahiServiceBrowserPrivate::convertTxtList(txt);
|
|
|
|
// create the new resolved service entry
|
|
AvahiServiceEntry entry = AvahiServiceEntry(name,
|
|
type,
|
|
QHostAddress(QString(a)),
|
|
QString(domain),
|
|
QString(host_name),
|
|
(quint16)port,
|
|
networkProtocol,
|
|
txtList,
|
|
flags);
|
|
|
|
serviceBrowser->m_serviceEntries.append(entry);
|
|
emit serviceBrowser->serviceEntryAdded(entry);
|
|
}
|
|
}
|
|
serviceBrowser->d_ptr->m_serviceResolvers.removeAll(resolver);
|
|
avahi_service_resolver_free(resolver);
|
|
|
|
}
|
|
|
|
QStringList QtAvahiServiceBrowserPrivate::convertTxtList(AvahiStringList *txt)
|
|
{
|
|
if (!txt)
|
|
return QStringList();
|
|
|
|
QStringList txtList;
|
|
txtList.append(QString(reinterpret_cast<char *>(txt->text)));
|
|
|
|
while (txt->next) {
|
|
AvahiStringList *next = txt->next;
|
|
txtList.append(QString(reinterpret_cast<char *>(next->text)));
|
|
txt = next;
|
|
}
|
|
|
|
return txtList;
|
|
}
|
|
|
|
QAbstractSocket::NetworkLayerProtocol QtAvahiServiceBrowserPrivate::convertProtocol(const AvahiProtocol &protocol)
|
|
{
|
|
QAbstractSocket::NetworkLayerProtocol networkProtocol = QAbstractSocket::UnknownNetworkLayerProtocol;
|
|
|
|
switch (protocol) {
|
|
case AVAHI_PROTO_INET:
|
|
networkProtocol = QAbstractSocket::IPv4Protocol;
|
|
break;
|
|
case AVAHI_PROTO_INET6:
|
|
networkProtocol = QAbstractSocket::IPv6Protocol;
|
|
break;
|
|
case AVAHI_PROTO_UNSPEC:
|
|
networkProtocol = QAbstractSocket::UnknownNetworkLayerProtocol;
|
|
break;
|
|
}
|
|
return networkProtocol;
|
|
}
|
|
|