avahi-remoteservice.cpp

/* This file is part of the KDE project
 *
 * Copyright (C) 2004, 2005 Jakub Stachowski <[email protected]>
 * Copyright (C) 2018 Harald Sitter <[email protected]>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public License
 * along with this library; see the file COPYING.LIB.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include "avahi-remoteservice_p.h"
#include <netinet/in.h>
#include <QEventLoop>
#include <QCoreApplication>
#include <QDebug>
#include "remoteservice.h"
#include "avahi_server_interface.h"
#include "avahi_serviceresolver_interface.h"
namespace KDNSSD
{

RemoteService::RemoteService(const QString &name, const QString &type, const QString &domain)
    : ServiceBase(new RemoteServicePrivate(this, name, type, domain))
{
}

RemoteService::~RemoteService()
{
}

bool RemoteService::resolve()
{
    K_D;
    resolveAsync();
    while (d->m_running && !d->m_resolved) {
        QCoreApplication::processEvents(QEventLoop::ExcludeUserInputEvents);
    }
    return d->m_resolved;
}

void RemoteService::resolveAsync()
{
    K_D;
    if (d->m_running) {
        return;
    }
    d->m_resolved = false;
    registerTypes();

    // Do not race!
    // https://github.com/lathiat/avahi/issues/9
    // Avahi's DBus API is incredibly racey with signals getting fired
    // immediately after a request was made even though we may not yet be
    // listening. In lieu of a proper upstream fix for this we'll unfortunately
    // have to resort to this hack:
    // We register to all signals regardless of path and then filter them once
    // we know what "our" path is. This is much more fragile than a proper
    // QDBusInterface assisted signal connection but unfortunately the only way
    // we can reliably prevent signals getting lost in the race.
    // This uses a fancy trick whereby using QDBusMessage as last argument will
    // give us the correct signal argument types as well as the underlying
    // message so that we may check the message path.
    QDBusConnection::systemBus()
            .connect("org.freedesktop.Avahi",
                     "",
                     "org.freedesktop.Avahi.ServiceResolver",
                     "Found",
                     d,
                     SLOT(gotGlobalFound(int,int,QString,QString,QString,QString,
                                         int,QString,ushort,QList<QByteArray>,
                                         uint,QDBusMessage)));
    QDBusConnection::systemBus()
            .connect("org.freedesktop.Avahi",
                     "",
                     "org.freedesktop.Avahi.ServiceResolver",
                     "Failure",
                     d,
                     SLOT(gotGlobalError(QDBusMessage)));
    d->m_dbusObjectPath.clear();

    //qDebug() << this << ":Starting resolve of : " << d->m_serviceName << " " << d->m_type << " " << d->m_domain << "\n";
    org::freedesktop::Avahi::Server s(QStringLiteral("org.freedesktop.Avahi"), QStringLiteral("/"), QDBusConnection::systemBus());
    //FIXME: don't use LOOKUP_NO_ADDRESS if NSS unavailable
    QDBusReply<QDBusObjectPath> rep = s.ServiceResolverNew(-1, -1, d->m_serviceName, d->m_type,
                                      domainToDNS(d->m_domain), -1, 8 /*AVAHI_LOOKUP_NO_ADDRESS*/);
    if (!rep.isValid()) {
        emit resolved(false);
        return;
    }

    d->m_dbusObjectPath = rep.value().path();

    // This is held because we need to explicitly Free it!
    d->m_resolver = new org::freedesktop::Avahi::ServiceResolver(
                s.service(),
                d->m_dbusObjectPath,
                s.connection());
    d->m_running = true;
}

bool RemoteService::isResolved() const
{
    K_D;
    return d->m_resolved;
}

void RemoteServicePrivate::gotError()
{
    m_resolved = false;
    stop();

    emit m_parent->resolved(false);
}

void RemoteServicePrivate::gotGlobalFound(int interface,
                                          int protocol,
                                          const QString &name,
                                          const QString &type,
                                          const QString &domain,
                                          const QString &host,
                                          int aprotocol,
                                          const QString &address,
                                          ushort port,
                                          const QList<QByteArray> &txt,
                                          uint flags,
                                          QDBusMessage msg)
{
    if (!isOurMsg(msg)) {
        return;
    }
    gotFound(interface, protocol, name, type, domain, host, aprotocol, address,
             port, txt, flags);
}

void RemoteServicePrivate::gotGlobalError(QDBusMessage msg)
{
    if (!isOurMsg(msg)) {
        return;
    }
    gotError();
}

void RemoteServicePrivate::gotFound(int, int, const QString &name, const QString &, const QString &domain, const QString &host, int, const QString &, ushort port, const QList<QByteArray> &txt, uint)
{
    m_serviceName = name;
    m_hostName = host;
    m_port = port;
    m_domain = DNSToDomain(domain);
    for (const QByteArray &x : txt) {
        int pos = x.indexOf("=");
        if (pos == -1) {
            m_textData[x] = QByteArray();
        } else {
            m_textData[x.mid(0, pos)] = x.mid(pos + 1, x.size() - pos);
        }
    }
    m_resolved = true;
    emit m_parent->resolved(true);
}

void RemoteServicePrivate::stop()
{
    if (m_resolver) {
        m_resolver->Free();
    }
    delete m_resolver;
    m_resolver = nullptr;
    m_running = false;
}

void RemoteService::virtual_hook(int, void *)
{
    // BASE::virtual_hook(int, void*);
}

}

#include "moc_remoteservice.cpp"
#include "moc_avahi-remoteservice_p.cpp"