MDNS.hxx

Go to the documentation of this file.

 
#ifndef _OS_MDNS_HXX_
#define _OS_MDNS_HXX_
 
#if defined (__linux__)
#include <netdb.h>
#include <stdio.h>
#include <semaphore.h>
#include <avahi-client/client.h>
#include <avahi-client/lookup.h>
#include <avahi-client/publish.h>
#include <avahi-common/error.h>
#include <avahi-common/malloc.h>
#include <avahi-common/simple-watch.h>
 
#include "os/OS.hxx"
 
#elif defined(__FreeRTOS__)
#include <netdb.h>
#endif
 
struct addrinfo;
 
#include "utils/macros.h"
 
 
class MDNS
#if defined (__linux__)
    : public OSThread
#endif
{
public:
    MDNS()
#if defined (__linux__)
        : 
          OSThread()
        , sem_()
        , group_(nullptr)
        , simplePoll_(nullptr)
        , client_(nullptr)
#endif
    {
#if defined (__linux__)
        start("mDNS Server", 0, 2048);
        printf("mDNS client start\n");        
        sem_.wait();
#endif
    }
 
    ~MDNS()
    {
#if defined(__linux__)
        shutdown();
#endif
    }
 
#if defined(__linux__)
    void shutdown();
#endif
 
    void publish(const char *name, const char *service, uint16_t port);
 
    void unpublish(const char *name, const char *service);
    
    void commit()
    {
#if defined (__linux__)
        avahi_entry_group_commit(group_);
#endif
    }
 
    static int lookup(const char *service, struct addrinfo *hints,
                      struct addrinfo **addr);
 
    static void scan(const char *service);
 
private:
#if defined (__linux__)
    struct LookupUserdata
    {
        LookupUserdata(const struct addrinfo *h)
            : count(0)
            , done(0)
            , c(nullptr)
            , sp(nullptr)
            , hints(h)
            , addr(nullptr)
        {
        }
 
        unsigned count : 31;
        unsigned done  :  1;
        AvahiClient *c;
        AvahiSimplePoll *sp;
        const struct addrinfo *hints;
        struct addrinfo *addr;
    };
 
    static void resolve_callback(AvahiServiceResolver *r,
                                 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);
 
    static void browse_callback(AvahiServiceBrowser *b, AvahiIfIndex interface,
                                AvahiProtocol protocol, AvahiBrowserEvent event,
                                const char *name, const char *type,
                                const char *domain,
                                AvahiLookupResultFlags flags, void* userdata);
 
    static void entry_group_callback(AvahiEntryGroup *g,
                                     AvahiEntryGroupState state, void *userdata);
 
    static void client_callback(AvahiClient *c, AvahiClientState state,
                                void * userdata);
 
    void *entry() override;
 
    OSSem sem_;
 
    AvahiEntryGroup *group_;
 
    AvahiSimplePoll *simplePoll_;
 
    AvahiClient *client_;
#endif
 
    DISALLOW_COPY_AND_ASSIGN(MDNS);
};
 
#endif /* _OS_MDNS_HXX_ */