-#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/LinkLocal/DNSSD/DNSSDServiceID.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 DNSSDServiceID&);

- virtual void stopResolvingService(const DNSSDServiceID&);

-

- 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<DNSSDServiceID, DNSServiceRef> ServiceSDRefMap;

- ServiceSDRefMap resolveSDRefs;

- typedef std::vector<DNSServiceRef> HostnameSDRefs;

- HostnameSDRefs hostnameResolveSDRefs;

- };

-}