-#include "Swiften/LinkLocal/AppleDNSSDService.h"

-

-#include <algorithm>

-#include <unistd.h>

-#include <iostream>

-#include <sys/socket.h>

-#include <netinet/in.h>

-#include <fcntl.h>

-

-#include "Swiften/EventLoop/MainEventLoop.h"

-#include "Swiften/LinkLocal/LinkLocalServiceInfo.h"

-#include "Swiften/Network/HostAddress.h"

-

-namespace Swift {

-

-AppleDNSSDService::AppleDNSSDService() : thread(0), stopRequested(false), haveError(false), browseSDRef(0), registerSDRef(0) {

- int fds[2];

- int result = pipe(fds);

- assert(result == 0);

- interruptSelectReadSocket = fds[0];

- fcntl(interruptSelectReadSocket, F_SETFL, fcntl(interruptSelectReadSocket, F_GETFL)|O_NONBLOCK);

- interruptSelectWriteSocket = fds[1];

-}

-

-AppleDNSSDService::~AppleDNSSDService() {

- stop();

-}

-

-void AppleDNSSDService::start() {

- stop();

- thread = new boost::thread(boost::bind(&AppleDNSSDService::doStart, shared_from_this()));

-}

-

-void AppleDNSSDService::stop() {

- if (thread) {

- stopRequested = true;

- interruptSelect();

- thread->join();

- delete thread;

- stopRequested = false;

- }

-}

-

-void AppleDNSSDService::registerService(const String& name, int port, const LinkLocalServiceInfo& info) {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

-

- assert(!registerSDRef);

- ByteArray txtRecord = info.toTXTRecord();

- DNSServiceErrorType result = DNSServiceRegister(&registerSDRef, 0, 0, name.getUTF8Data(), "_presence._tcp", NULL, NULL, port, txtRecord.getSize(), txtRecord.getData(), &AppleDNSSDService::handleServiceRegisteredGlobal, this);

- if (result != kDNSServiceErr_NoError) {

- std::cerr << "Error creating service registration" << std::endl;

- haveError = true;

- }

-

- interruptSelect();

-}

-

-void AppleDNSSDService::updateService(const LinkLocalServiceInfo& info) {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

- assert(registerSDRef);

- ByteArray txtRecord = info.toTXTRecord();

- DNSServiceUpdateRecord(registerSDRef, NULL, NULL, txtRecord.getSize(), txtRecord.getData(), 0);

-}

-

-void AppleDNSSDService::unregisterService() {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

-

- assert(registerSDRef);

- DNSServiceRefDeallocate(registerSDRef); // Interrupts select()

- registerSDRef = NULL;

-}

-

-void AppleDNSSDService::startResolvingService(const Service& service) {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

-

- DNSServiceRef resolveSDRef;

- DNSServiceErrorType result = DNSServiceResolve(&resolveSDRef, 0, service.networkInterface, service.name.getUTF8Data(), service.type.getUTF8Data(), service.domain.getUTF8Data(), &AppleDNSSDService::handleServiceResolvedGlobal, this);

- if (result != kDNSServiceErr_NoError) {

- std::cerr << "Error creating service resolve query" << std::endl;

- haveError = true;

- }

- else {

- bool isNew = resolveSDRefs.insert(std::make_pair(service, resolveSDRef)).second;

- assert(isNew);

- }

-

- interruptSelect();

-}

-

-void AppleDNSSDService::stopResolvingService(const Service& service) {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

-

- ServiceSDRefMap::iterator i = resolveSDRefs.find(service);

- assert(i != resolveSDRefs.end());

- DNSServiceRefDeallocate(i->second); // Interrupts select()

- resolveSDRefs.erase(i);

-}

-

-void AppleDNSSDService::resolveHostname(const String& hostname, int interfaceIndex) {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

-

- DNSServiceRef hostnameResolveSDRef;

- DNSServiceErrorType result = DNSServiceGetAddrInfo(&hostnameResolveSDRef, 0, interfaceIndex, kDNSServiceProtocol_IPv4, hostname.getUTF8Data(), &AppleDNSSDService::handleHostnameResolvedGlobal, this);

- if (result != kDNSServiceErr_NoError) {

- std::cerr << "Error creating hostname resolve query" << std::endl;

- haveError = true;

- }

- else {

- hostnameResolveSDRefs.push_back(hostnameResolveSDRef);

- }

-

- interruptSelect();

-}

-

-void AppleDNSSDService::doStart() {

- haveError = false;

- onStarted();

-

- // Listen for new services

- {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

- assert(!browseSDRef);

- DNSServiceErrorType result = DNSServiceBrowse(&browseSDRef, 0, 0, "_presence._tcp", 0, &AppleDNSSDService::handleServiceDiscoveredGlobal , this);

- if (result != kDNSServiceErr_NoError) {

- std::cerr << "Error creating browse query" << std::endl;

- haveError = true;

- }

- }

-

- // Run the main loop

- while (!haveError && !stopRequested) {

- fd_set fdSet;

- FD_ZERO(&fdSet);

- int maxSocket = interruptSelectReadSocket;

- FD_SET(interruptSelectReadSocket, &fdSet);

-

- {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

-

- // Browsing

- int browseSocket = DNSServiceRefSockFD(browseSDRef);

- maxSocket = std::max(maxSocket, browseSocket);

- FD_SET(browseSocket, &fdSet);

-

- // Registration

- if (registerSDRef) {

- int registerSocket = DNSServiceRefSockFD(registerSDRef);

- maxSocket = std::max(maxSocket, registerSocket);

- FD_SET(registerSocket, &fdSet);

- }

-

- // Service resolving

- for (ServiceSDRefMap::const_iterator i = resolveSDRefs.begin(); i != resolveSDRefs.end(); ++i) {

- int resolveSocket = DNSServiceRefSockFD(i->second);

- maxSocket = std::max(maxSocket, resolveSocket);

- FD_SET(resolveSocket, &fdSet);

- }

-

- // Hostname resolving

- for (HostnameSDRefs::const_iterator i = hostnameResolveSDRefs.begin(); i != hostnameResolveSDRefs.end(); ++i) {

- int hostnameResolveSocket = DNSServiceRefSockFD(*i);

- maxSocket = std::max(maxSocket, hostnameResolveSocket);

- FD_SET(hostnameResolveSocket, &fdSet);

- }

- }

-

- int selectResult = select(maxSocket+1, &fdSet, NULL, NULL, 0);

-

- // Flush the interruptSelectReadSocket

- if (FD_ISSET(interruptSelectReadSocket, &fdSet)) {

- char dummy;

- while (read(interruptSelectReadSocket, &dummy, 1) > 0) {}

- }

-

- {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

- if (selectResult <= 0) {

- continue;

- }

- if (FD_ISSET(DNSServiceRefSockFD(browseSDRef), &fdSet)) {

- DNSServiceProcessResult(browseSDRef);

- }

- if (registerSDRef && FD_ISSET(DNSServiceRefSockFD(registerSDRef), &fdSet)) {

- DNSServiceProcessResult(registerSDRef);

- }

- for (ServiceSDRefMap::const_iterator i = resolveSDRefs.begin(); i != resolveSDRefs.end(); ++i) {

- if (FD_ISSET(DNSServiceRefSockFD(i->second), &fdSet)) {

- DNSServiceProcessResult(i->second);

- }

- }

- for (HostnameSDRefs::const_iterator i = hostnameResolveSDRefs.begin(); i != hostnameResolveSDRefs.end(); ++i) {

- if (FD_ISSET(DNSServiceRefSockFD(*i), &fdSet)) {

- DNSServiceProcessResult(*i);

- hostnameResolveSDRefs.erase(std::remove(hostnameResolveSDRefs.begin(), hostnameResolveSDRefs.end(), *i), hostnameResolveSDRefs.end());

- DNSServiceRefDeallocate(*i);

- break; // Stop the loop, because we removed an element

- }

- }

- }

- }

-

- {

- boost::lock_guard<boost::mutex> lock(sdRefsMutex);

-

- for (ServiceSDRefMap::const_iterator i = resolveSDRefs.begin(); i != resolveSDRefs.end(); ++i) {

- DNSServiceRefDeallocate(i->second);

- }

- resolveSDRefs.clear();

-

- for (HostnameSDRefs::const_iterator i = hostnameResolveSDRefs.begin(); i != hostnameResolveSDRefs.end(); ++i) {

- DNSServiceRefDeallocate(*i);

- }

- hostnameResolveSDRefs.clear();

-

- if (registerSDRef) {

- DNSServiceRefDeallocate(registerSDRef);

- registerSDRef = NULL;

- }

-

- DNSServiceRefDeallocate(browseSDRef);

- browseSDRef = NULL;

- }

-

- MainEventLoop::postEvent(boost::bind(boost::ref(onStopped), haveError), shared_from_this());

-}

-

-void AppleDNSSDService::interruptSelect() {

- char c = 0;

- write(interruptSelectWriteSocket, &c, 1);

-}

-

-void AppleDNSSDService::handleServiceDiscoveredGlobal(DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *serviceName, const char *regtype, const char *replyDomain, void *context ) {

- static_cast<AppleDNSSDService*>(context)->handleServiceDiscovered(sdRef, flags, interfaceIndex, errorCode, serviceName, regtype, replyDomain);

-}

-

-void AppleDNSSDService::handleServiceDiscovered(DNSServiceRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *serviceName, const char *regtype, const char *replyDomain) {

- if (errorCode != kDNSServiceErr_NoError) {

- return;

- }

- else {

- Service service(serviceName, regtype, replyDomain, interfaceIndex);

- if (flags & kDNSServiceFlagsAdd) {

- MainEventLoop::postEvent(boost::bind(boost::ref(onServiceAdded), service), shared_from_this());

- }

- else {

- MainEventLoop::postEvent(boost::bind(boost::ref(onServiceRemoved), service), shared_from_this());

- }

- }

-}

-

-void AppleDNSSDService::handleServiceRegisteredGlobal(DNSServiceRef sdRef, DNSServiceFlags flags, DNSServiceErrorType errorCode, const char *name, const char *regtype, const char *domain, void *context) {

- static_cast<AppleDNSSDService*>(context)->handleServiceRegistered(sdRef, flags, errorCode, name, regtype, domain);

-}

-

-void AppleDNSSDService::handleServiceRegistered(DNSServiceRef, DNSServiceFlags, DNSServiceErrorType errorCode, const char *name, const char *regtype, const char *domain) {

- if (errorCode != kDNSServiceErr_NoError) {

- std::cerr << "Error registering service" << std::endl;

- haveError = true;

- }

- else {

- MainEventLoop::postEvent(boost::bind(boost::ref(onServiceRegistered), Service(name, regtype, domain, 0)), shared_from_this());

- }

-}

-

-void AppleDNSSDService::handleServiceResolvedGlobal(DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *fullname, const char *hosttarget, uint16_t port, uint16_t txtLen, const unsigned char *txtRecord, void *context ) {

- static_cast<AppleDNSSDService*>(context)->handleServiceResolved(sdRef, flags, interfaceIndex, errorCode, fullname, hosttarget, port, txtLen, txtRecord);

-}

-

-void AppleDNSSDService::handleServiceResolved(DNSServiceRef sdRef, DNSServiceFlags, uint32_t, DNSServiceErrorType errorCode, const char *, const char *hosttarget, uint16_t port, uint16_t txtLen, const unsigned char *txtRecord) {

- if (errorCode != kDNSServiceErr_NoError) {

- // TODO

- std::cerr << "Resolve error " << hosttarget << std::endl;

- return;

- }

- for (ServiceSDRefMap::const_iterator i = resolveSDRefs.begin(); i != resolveSDRefs.end(); ++i) {

- if (i->second == sdRef) {

- MainEventLoop::postEvent(boost::bind(boost::ref(onServiceResolved), i->first, ResolveResult(hosttarget, port, LinkLocalServiceInfo::createFromTXTRecord(ByteArray(reinterpret_cast<const char*>(txtRecord), txtLen)))), shared_from_this());

- return;

- }

- }

- assert(false);

-}

-

-void AppleDNSSDService::handleHostnameResolvedGlobal(DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *hostname, const struct sockaddr *address, uint32_t ttl, void *context) {

- static_cast<AppleDNSSDService*>(context)->handleHostnameResolved(sdRef, flags, interfaceIndex, errorCode, hostname, address, ttl);

-}

-

-void AppleDNSSDService::handleHostnameResolved(DNSServiceRef, DNSServiceFlags, uint32_t, DNSServiceErrorType errorCode, const char *hostname, const struct sockaddr *rawAddress, uint32_t) {

- if (errorCode) {

- std::cerr << "Error resolving hostname" << std::endl;

- MainEventLoop::postEvent(boost::bind(boost::ref(onHostnameResolved), hostname, boost::optional<HostAddress>()), shared_from_this());

- }

- else {

- assert(rawAddress->sa_family == AF_INET);

- const sockaddr_in* sa = reinterpret_cast<const sockaddr_in*>(rawAddress);

- uint32_t address = ntohl(sa->sin_addr.s_addr);

- MainEventLoop::postEvent(boost::bind(boost::ref(onHostnameResolved), String(hostname), HostAddress(reinterpret_cast<unsigned char*>(&address), 4)), shared_from_this());

- }

-}

-

-}

-#pragma once

-

-#include <boost/enable_shared_from_this.hpp>

-#include <boost/thread.hpp>

-#include <boost/thread/mutex.hpp>

-#include <dns_sd.h>

-

-#include "Swiften/LinkLocal/DNSSDService.h"

-#include "Swiften/EventLoop/EventOwner.h"

-

-namespace Swift {

- class AppleDNSSDService : public DNSSDService, public EventOwner, public boost::enable_shared_from_this<AppleDNSSDService> {

- public:

- AppleDNSSDService();

- ~AppleDNSSDService();

-

- virtual void start();

- virtual void stop();

-

- virtual void registerService(const String& name, int port, const LinkLocalServiceInfo&);

- virtual void updateService(const LinkLocalServiceInfo&);

- virtual void unregisterService();

-

- virtual void startResolvingService(const Service&);

- virtual void stopResolvingService(const Service&);

-

- virtual void resolveHostname(const String& hostname, int interfaceIndex = 0);

-

- private:

- void doStart();

- void interruptSelect();

-

- static void handleServiceDiscoveredGlobal(DNSServiceRef, DNSServiceFlags, uint32_t, DNSServiceErrorType, const char *, const char *, const char *, void*);

- void handleServiceDiscovered(DNSServiceRef, DNSServiceFlags, uint32_t, DNSServiceErrorType, const char *, const char *, const char *);

- static void handleServiceRegisteredGlobal(DNSServiceRef, DNSServiceFlags, DNSServiceErrorType, const char *, const char *, const char *, void *);

- void handleServiceRegistered(DNSServiceRef, DNSServiceFlags, DNSServiceErrorType, const char *, const char *, const char *);

- static void handleServiceResolvedGlobal(DNSServiceRef, DNSServiceFlags, uint32_t, DNSServiceErrorType, const char *, const char *, uint16_t, uint16_t, const unsigned char *, void *);

- void handleServiceResolved(DNSServiceRef, DNSServiceFlags, uint32_t, DNSServiceErrorType, const char *, const char *, uint16_t, uint16_t, const unsigned char *);

- static void handleHostnameResolvedGlobal(DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *hostname, const struct sockaddr *address, uint32_t ttl, void *context);

- void handleHostnameResolved(DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *hostname, const struct sockaddr *address, uint32_t ttl);

-

- private:

- boost::thread* thread;

- bool stopRequested;

- bool haveError;

- int interruptSelectReadSocket;

- int interruptSelectWriteSocket;

- boost::mutex sdRefsMutex;

- DNSServiceRef browseSDRef;

- DNSServiceRef registerSDRef;

- typedef std::map<Service, DNSServiceRef> ServiceSDRefMap;

- ServiceSDRefMap resolveSDRefs;

- typedef std::vector<DNSServiceRef> HostnameSDRefs;

- HostnameSDRefs hostnameResolveSDRefs;

- };

-}

-#include "Swiften/LinkLocal/AvahiDNSSDService.h"

-

-#include <boost/bind.hpp>

-

-#include "Swiften/EventLoop/MainEventLoop.h"

-#include "Swiften/LinkLocal/LinkLocalServiceInfo.h"

-#include "Swiften/Network/HostAddress.h"

-

-namespace Swift {

-

-AvahiDNSSDService::AvahiDNSSDService() : client(NULL), threadedPoll(0), serviceBrowser(0) {

-}

-

-AvahiDNSSDService::~AvahiDNSSDService() {

-}

-

-void AvahiDNSSDService::start() {

- threadedPoll = avahi_threaded_poll_new();

-

- int error;

- client = avahi_client_new(

- avahi_threaded_poll_get(threadedPoll),

- static_cast<AvahiClientFlags>(0), NULL, this, &error); // TODO

- if (!client) {

- // TODO

- std::cerr << "Avahi Error: " << avahi_strerror(error) << std::endl;

- return;

- }

-

- serviceBrowser = avahi_service_browser_new(

- client, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC,

- "_presence._tcp", NULL, static_cast<AvahiLookupFlags>(0),

- handleServiceDiscoveredGlobal, this);

- if (!serviceBrowser) {

- // TODO

- std::cerr << "Avahi Error: " << avahi_strerror(error) << std::endl;

- return;

- }

-

- avahi_threaded_poll_start(threadedPoll);

-}

-

-void AvahiDNSSDService::stop() {

- avahi_threaded_poll_stop(threadedPoll);

- avahi_service_browser_free(serviceBrowser);

- avahi_client_free(client);

- avahi_threaded_poll_free(threadedPoll);

-}

-

-void AvahiDNSSDService::registerService(const String& name, int port, const LinkLocalServiceInfo& info) {

- avahi_threaded_poll_lock(threadedPoll);

- avahi_threaded_poll_unlock(threadedPoll);

-}

-

-void AvahiDNSSDService::updateService(const LinkLocalServiceInfo& info) {

- avahi_threaded_poll_lock(threadedPoll);

- avahi_threaded_poll_unlock(threadedPoll);

-}

-

-void AvahiDNSSDService::unregisterService() {

- avahi_threaded_poll_lock(threadedPoll);

- avahi_threaded_poll_unlock(threadedPoll);

-}

-

-void AvahiDNSSDService::startResolvingService(const Service& service) {

- avahi_threaded_poll_lock(threadedPoll);

-

- AvahiServiceResolver* resolver = avahi_service_resolver_new(

- client,

- service.networkInterface,

- AVAHI_PROTO_INET,

- service.name.getUTF8Data(),

- service.type.getUTF8Data(),

- service.domain.getUTF8Data(),

- AVAHI_PROTO_UNSPEC,

- static_cast<AvahiLookupFlags>(0),

- &AvahiDNSSDService::handleServiceResolvedGlobal,

- this);

- assert(serviceResolvers.find(service) == serviceResolvers.end());

- serviceResolvers[service] = resolver;

-

- avahi_threaded_poll_unlock(threadedPoll);

-}

-

-void AvahiDNSSDService::stopResolvingService(const Service& service) {

- avahi_threaded_poll_lock(threadedPoll);

-

- ServiceResolverMap::iterator i = serviceResolvers.find(service);

- assert(i != serviceResolvers.end());

- avahi_service_resolver_free(i->second);

- serviceResolvers.erase(i);

-

- avahi_threaded_poll_unlock(threadedPoll);

-}

-

-// TODO: Take interfaceIndex into account

-void AvahiDNSSDService::resolveHostname(const String& hostname, int) {

- HostnameAddressMap::const_iterator i = hostnameAddresses.find(hostname);

- if (i == hostnameAddresses.end()) {

- MainEventLoop::postEvent(boost::bind(boost::ref(onHostnameResolved), hostname, boost::optional<HostAddress>()), shared_from_this());

- }

- else {

- MainEventLoop::postEvent(boost::bind(boost::ref(onHostnameResolved), hostname, boost::optional<HostAddress>(i->second)), shared_from_this());

- }

-}

-

-void AvahiDNSSDService::handleServiceDiscovered(AvahiServiceBrowser *, AvahiIfIndex interfaceIndex, AvahiProtocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AvahiLookupResultFlags) {

- switch(event) {

- case AVAHI_BROWSER_FAILURE:

- std::cerr << "Browse error" << std::endl;

- // TODO

- return;

- case AVAHI_BROWSER_NEW: {

- std::cerr << "Service added: " << name << " " << type << " " << domain << std::endl;

- Service service(name, type, domain, interfaceIndex);

- MainEventLoop::postEvent(boost::bind(boost::ref(onServiceAdded), service), shared_from_this());

- }

- break;

- case AVAHI_BROWSER_REMOVE: {

- std::cerr << "Service removed: " << name << " " << type << " " << domain << std::endl;

- Service service(name, type, domain, interfaceIndex);

- MainEventLoop::postEvent(boost::bind(boost::ref(onServiceRemoved), service), shared_from_this());

- }

- break;

- case AVAHI_BROWSER_ALL_FOR_NOW:

- case AVAHI_BROWSER_CACHE_EXHAUSTED:

- break;

- }

-}

-

-void AvahiDNSSDService::handleServiceResolved(AvahiServiceResolver *, AvahiIfIndex interfaceIndex, AvahiProtocol, AvahiResolverEvent event, const char *name, const char *type, const char *domain, const char *hostname, const AvahiAddress *address, uint16_t port, AvahiStringList *txt, AvahiLookupResultFlags) {

- switch(event) {

- case AVAHI_RESOLVER_FAILURE:

- //TODO

- std::cerr << "Resolve error" << std::endl;

- break;

- case AVAHI_RESOLVER_FOUND:

- ByteArray data;

- for(AvahiStringList* i = txt; i; i = avahi_string_list_get_next(i)) {

- char size = i->size;

- data += ByteArray(&size, 1);

- data += ByteArray(reinterpret_cast<char*>(avahi_string_list_get_text(i)), avahi_string_list_get_size(i));

- }

-

- assert(address->proto == AVAHI_PROTO_INET);

- HostAddress hostAddress(reinterpret_cast<const unsigned char*>(&address->data.ipv4.address), 4);

- hostnameAddresses[String(hostname)] = hostAddress;

- MainEventLoop::postEvent(boost::bind(boost::ref(onServiceResolved),

- Service(name, type, domain, interfaceIndex),

- ResolveResult(hostname, port,

- LinkLocalServiceInfo::createFromTXTRecord(data))),

- shared_from_this());

- break;

- }

-}

-

-}

-#pragma once

-

-#include <map>

-#include <boost/enable_shared_from_this.hpp>

-#include <avahi-client/client.h>

-#include <avahi-client/lookup.h>

-#include <avahi-common/thread-watch.h>

-#include <avahi-common/watch.h>

-#include <avahi-common/malloc.h>

-#include <avahi-common/error.h>

-

-

-#include "Swiften/EventLoop/EventOwner.h"

-#include "Swiften/LinkLocal/DNSSDService.h"

-

-namespace Swift {

- class AvahiDNSSDService : public DNSSDService, public EventOwner, public boost::enable_shared_from_this<AvahiDNSSDService> {

- public:

- AvahiDNSSDService();

- ~AvahiDNSSDService();

-

- virtual void start();

- virtual void stop();

-

- virtual void registerService(const String& name, int port, const LinkLocalServiceInfo&);

- virtual void updateService(const LinkLocalServiceInfo&);

- virtual void unregisterService();

-

- virtual void startResolvingService(const Service&);

- virtual void stopResolvingService(const Service&);

-

- virtual void resolveHostname(const String& hostname, int interfaceIndex = 0);

-

- private:

- static void handleServiceDiscoveredGlobal(AvahiServiceBrowser *b, AvahiIfIndex networkInterface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AvahiLookupResultFlags flags, void* userdata) {

- static_cast<AvahiDNSSDService*>(userdata)->handleServiceDiscovered(b, networkInterface, protocol, event, name, type, domain, flags);

- }

- void handleServiceDiscovered(AvahiServiceBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AVAHI_GCC_UNUSED AvahiLookupResultFlags flags);

-

- static void handleServiceResolvedGlobal(AvahiServiceResolver *r, AvahiIfIndex interfaceIndex, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const char *type, const char *domain, const char *hostname, const AvahiAddress *address, uint16_t port, AvahiStringList *txt, AvahiLookupResultFlags flags, void* userdata) {

- static_cast<AvahiDNSSDService*>(userdata)->handleServiceResolved(r, interfaceIndex, protocol, event, name, type, domain, hostname, address, port, txt, flags);

- }

- void handleServiceResolved(AvahiServiceResolver *r, AvahiIfIndex interface, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const char *type, const char *domain, const char *hostname, const AvahiAddress *address, uint16_t port, AvahiStringList *txt, AvahiLookupResultFlags flags);

-

-

- private:

- AvahiClient* client;

- AvahiThreadedPoll* threadedPoll;

- AvahiServiceBrowser* serviceBrowser;

- typedef std::map<Service, AvahiServiceResolver*> ServiceResolverMap;

- ServiceResolverMap serviceResolvers;

- typedef std::map<String, HostAddress> HostnameAddressMap;

- HostnameAddressMap hostnameAddresses;

- };

-}

-#include "Swiften/LinkLocal/DNSSDService.h"

-

-namespace Swift {

-

-DNSSDService::~DNSSDService() {

-}

-

-}

-#pragma once

-

-#include <boost/signal.hpp>

-#include <map>

-

-#include "Swiften/Base/String.h"

-#include "Swiften/LinkLocal/LinkLocalServiceInfo.h"

-

-namespace Swift {

- class LinkLocalServiceInfo;

- class HostAddress;

-

- class DNSSDService {

- public:

- struct Service {

- Service(const String& name, const String& type, const String& domain, int networkInterface) : name(name), type(type), domain(domain), networkInterface(networkInterface) {}

- bool operator==(const Service& o) const {

- return name == o.name && type == o.type && domain == o.domain && (networkInterface != 0 && o.networkInterface != 0 ? networkInterface == o.networkInterface : true);

- }

- bool operator<(const Service& o) const {

- if (o.name == name) {

- if (o.type == type) {

- if (o.domain == domain) {

- return networkInterface < o.networkInterface;

- }

- else {

- return domain < o.domain;

- }

- }

- else {

- return type < o.type;

- }

- }

- else {

- return o.name < name;

- }

- }

-

- String name;

- String type;

- String domain;

- int networkInterface;

- };

-

- struct ResolveResult {

- ResolveResult(const String& host, int port, const LinkLocalServiceInfo& info) : host(host), port(port), info(info) {}

- String host;

- int port;

- LinkLocalServiceInfo info;

- };

-

- virtual ~DNSSDService();

-

- virtual void start() = 0;

- virtual void stop() = 0;

-

- virtual void registerService(const String& name, int port, const LinkLocalServiceInfo&) = 0;

- virtual void updateService(const LinkLocalServiceInfo&) = 0;

- virtual void unregisterService() = 0;

-

- virtual void startResolvingService(const Service&) = 0;

- virtual void stopResolvingService(const Service&) = 0;

-

- virtual void resolveHostname(const String& hostname, int interfaceIndex = 0) = 0;

-

- boost::signal<void ()> onStarted;

- boost::signal<void (bool)> onStopped;

- boost::signal<void (const Service&)> onServiceAdded;

- boost::signal<void (const Service&)> onServiceRemoved;

- boost::signal<void (const Service&)> onServiceRegistered;

- boost::signal<void (const Service&, const ResolveResult&)> onServiceResolved;

- boost::signal<void (const String&, const boost::optional<HostAddress>&)> onHostnameResolved;

- };

-}

-#include "Swiften/LinkLocal/DNSSDService.h"

- boost::shared_ptr<DNSSDService> resolver,

- remoteJID_(remoteJID),

- hostname_(hostname),

- port_(port),

- resolver_(resolver),

- connection_(connection),

- resolving_(false) {

- resolving_ = true;

- resolver_->onHostnameResolved.connect(boost::bind(&LinkLocalConnector::handleHostnameResolved, boost::dynamic_pointer_cast<LinkLocalConnector>(shared_from_this()), _1, _2));

- resolver_->resolveHostname(hostname_);

-void LinkLocalConnector::handleHostnameResolved(const String& hostname, const boost::optional<HostAddress>& address) {

- if (resolving_) {

- if (hostname == hostname_) {

- resolving_ = false;

- if (address) {

- connection_->onConnectFinished.connect(boost::bind(boost::ref(onConnectFinished), _1));

- connection_->connect(HostAddressPort(*address, port_));

- }

- else {

- onConnectFinished(false);

- }

- }

- queuedElements_.push_back(element);

- class DNSSDService;

- boost::shared_ptr<DNSSDService> resolver,

- return remoteJID_;

- return queuedElements_;

- return connection_;

- void handleHostnameResolved(const String& hostname, const boost::optional<HostAddress>& address);

- JID remoteJID_;

- String hostname_;

- int port_;

- boost::shared_ptr<DNSSDService> resolver_;

- boost::shared_ptr<Connection> connection_;

- bool resolving_;

- std::vector<boost::shared_ptr<Element> > queuedElements_;

-#include <boost/bind.hpp>

-#include <iostream>

-

-#include "Swiften/LinkLocal/LinkLocalRoster.h"

-#include "Swiften/Network/HostAddress.h"

-

-namespace Swift {

-

-LinkLocalRoster::LinkLocalRoster(boost::shared_ptr<DNSSDService> service) : dnsSDService(service) {

- dnsSDService->onStopped.connect(boost::bind(&LinkLocalRoster::handleStopped, this, _1));

- dnsSDService->onServiceRegistered.connect(boost::bind(&LinkLocalRoster::handleServiceRegistered, this, _1));

- dnsSDService->onServiceAdded.connect(boost::bind(&LinkLocalRoster::handleServiceAdded, this, _1));

- dnsSDService->onServiceRemoved.connect(boost::bind(&LinkLocalRoster::handleServiceRemoved, this, _1));

- dnsSDService->onServiceResolved.connect(boost::bind(&LinkLocalRoster::handleServiceResolved, this, _1, _2));

-}

-

-boost::shared_ptr<RosterPayload> LinkLocalRoster::getRoster() const {

- boost::shared_ptr<RosterPayload> roster(new RosterPayload());

- for(ServiceMap::const_iterator i = services.begin(); i != services.end(); ++i) {

- roster->addItem(getRosterItem(i->first, i->second));

- }

- return roster;

-}

-

-std::vector<boost::shared_ptr<Presence> > LinkLocalRoster::getAllPresence() const {

- std::vector<boost::shared_ptr<Presence> > result;

- for(ServiceMap::const_iterator i = services.begin(); i != services.end(); ++i) {

- result.push_back(getPresence(i->first, i->second));

- }

- return result;

-}

-

-RosterItemPayload LinkLocalRoster::getRosterItem(const DNSSDService::Service& service, const DNSSDService::ResolveResult& resolveResult) const {

- return RosterItemPayload(getJIDForService(service), getRosterName(service, resolveResult), RosterItemPayload::Both);

-}

-

-String LinkLocalRoster::getRosterName(const DNSSDService::Service& service, const DNSSDService::ResolveResult& resolveResult) const {

- if (!resolveResult.info.getNick().isEmpty()) {

- return resolveResult.info.getNick();

- }

- else if (!resolveResult.info.getFirstName().isEmpty()) {

- String result = resolveResult.info.getFirstName();

- if (!resolveResult.info.getLastName().isEmpty()) {

- result += " " + resolveResult.info.getLastName();

- }

- return result;

- }

- else if (!resolveResult.info.getLastName().isEmpty()) {

- return resolveResult.info.getLastName();

- }

- return service.name;

-}

-

-JID LinkLocalRoster::getJIDForService(const DNSSDService::Service& service) const {

- return JID(service.name);

-}

-

-boost::shared_ptr<Presence> LinkLocalRoster::getPresence(const DNSSDService::Service& service, const DNSSDService::ResolveResult& resolveResult) const {

- boost::shared_ptr<Presence> presence(new Presence());

- presence->setFrom(getJIDForService(service));

- switch (resolveResult.info.getStatus()) {

- case LinkLocalServiceInfo::Available:

- presence->setShow(StatusShow::Online);

- break;

- case LinkLocalServiceInfo::Away:

- presence->setShow(StatusShow::Away);

- break;

- case LinkLocalServiceInfo::DND:

- presence->setShow(StatusShow::DND);

- break;

- }

- presence->setStatus(resolveResult.info.getMessage());

- return presence;

-}

-

-void LinkLocalRoster::handleServiceAdded(const DNSSDService::Service& service) {

- if (selfService && *selfService == service) {

- return;

- }

- dnsSDService->startResolvingService(service);

-}

-

-void LinkLocalRoster::handleServiceRemoved(const DNSSDService::Service& service) {

- if (selfService && *selfService == service) {

- return;

- }

- dnsSDService->stopResolvingService(service);

- services.erase(service);

- boost::shared_ptr<RosterPayload> roster(new RosterPayload());

- roster->addItem(RosterItemPayload(getJIDForService(service), "", RosterItemPayload::Remove));

- onRosterChanged(roster);

-}

-

-void LinkLocalRoster::handleServiceResolved(const DNSSDService::Service& service, const DNSSDService::ResolveResult& result) {

- std::pair<ServiceMap::iterator, bool> r = services.insert(std::make_pair(service, result));

- if (r.second) {

- boost::shared_ptr<RosterPayload> roster(new RosterPayload());

- roster->addItem(getRosterItem(service, result));

- onRosterChanged(roster);

- }

- else {

- r.first->second = result;

- }

- onPresenceChanged(getPresence(service, result));

-}

-

-void LinkLocalRoster::handleServiceRegistered(const DNSSDService::Service& service) {

- selfService = service;

-}

-

-void LinkLocalRoster::handleStopped(bool error) {

- std::cout << "DNSSDService stopped: " << error << std::endl;

-}

-

-bool LinkLocalRoster::hasItem(const JID& j) const {

- for(ServiceMap::const_iterator i = services.begin(); i != services.end(); ++i) {

- if (getJIDForService(i->first) == j) {

- return true;

- }

- }

- return false;

-}

-

-String LinkLocalRoster::getHostname(const JID& j) const {

- for(ServiceMap::const_iterator i = services.begin(); i != services.end(); ++i) {

- if (getJIDForService(i->first) == j) {

- return i->second.host;

- }

- }

- return "";

-}

-

-int LinkLocalRoster::getPort(const JID& j) const {

- for(ServiceMap::const_iterator i = services.begin(); i != services.end(); ++i) {

- if (getJIDForService(i->first) == j) {

- return i->second.port;

- }

- }

- return 0;

-}

-

-}

-#pragma once

-

-#include <boost/shared_ptr.hpp>

-#include <boost/optional.hpp>

-#include <set>

-

-#include "Swiften/Base/String.h"

-#include "Swiften/JID/JID.h"

-#include "Swiften/LinkLocal/DNSSDService.h"

-#include "Swiften/Elements/RosterPayload.h"

-#include "Swiften/Elements/Presence.h"

-

-namespace Swift {

- class HostAddress;

-

- class LinkLocalRoster {

- public:

- LinkLocalRoster(boost::shared_ptr<DNSSDService> service);

-

- boost::shared_ptr<RosterPayload> getRoster() const;

- std::vector<boost::shared_ptr<Presence> > getAllPresence() const;

-

- boost::signal<void (boost::shared_ptr<RosterPayload>)> onRosterChanged;

- boost::signal<void (boost::shared_ptr<Presence>)> onPresenceChanged;

-

- bool hasItem(const JID&) const;

- String getHostname(const JID&) const;

- int getPort(const JID&) const;

-

- private:

- RosterItemPayload getRosterItem(const DNSSDService::Service& service, const DNSSDService::ResolveResult& info) const;

- String getRosterName(const DNSSDService::Service& service, const DNSSDService::ResolveResult& info) const;

- JID getJIDForService(const DNSSDService::Service& service) const;

- boost::shared_ptr<Presence> getPresence(const DNSSDService::Service& service, const DNSSDService::ResolveResult& info) const;

-

- void handleStopped(bool);

- void handleServiceRegistered(const DNSSDService::Service& service);

- void handleServiceAdded(const DNSSDService::Service&);

- void handleServiceRemoved(const DNSSDService::Service&);

- void handleServiceResolved(const DNSSDService::Service& service, const DNSSDService::ResolveResult& result);

-

- private:

- boost::shared_ptr<DNSSDService> dnsSDService;

- boost::optional<DNSSDService::Service> selfService;

- typedef std::map<DNSSDService::Service, DNSSDService::ResolveResult> ServiceMap;

- ServiceMap services;

- };

-}

- Swiften/LinkLocal/DNSSDService.cpp \

- Swiften/LinkLocal/LinkLocalRoster.cpp \

-ifeq ($(MACOSX),1)

-SWIFTEN_SOURCES += \

- Swiften/LinkLocal/AppleDNSSDService.cpp

-endif

-ifeq ($(HAVE_AVAHI),yes)

-SWIFTEN_SOURCES += \

- Swiften/LinkLocal/AvahiDNSSDService.cpp

-endif

-

- LinkLocalServiceInfoTest() {}

-