MDNS/src/include/Network/Mdns/Server.h

Go to the documentation of this file.

/****
 * Sming Framework Project - Open Source framework for high efficiency native ESP8266 development.
 * Created 2015 by Skurydin Alexey
 * http://github.com/SmingHub/Sming
 * All files of the Sming Core are provided under the LGPL v3 license.
 *
 * Server.h
 *
 * Code originally based on MrDunk's mDNS code distributed under the MIT license.
 *
 * MIT license: https://github.com/mrdunk/esp8266_mdns/blob/master/LICENCE.txt
 *
 * However, code has been essentially rewritten so Sming LGPL v3 license applies.
 *
 */

#pragma once

#include <Delegate.h>
#include <Network/UdpConnection.h>
#include "Request.h"
#include "Handler.h"

namespace mDNS
{
class Server : protected UdpConnection
{
public:
    using MessageDelegate = Delegate<bool(Message& message)>;

    using PacketDelegate = Delegate<bool(IpAddress remoteIP, uint16_t remotePort, const uint8_t* data, size_t length)>;

    ~Server();

    bool begin();

    void end();

    bool restart()
    {
        end();
        return begin();
    }

    void addHandler(Handler& handler)
    {
        handlers.add(&handler);
    }

    void removeHandler(Handler& handler)
    {
        handlers.remove(&handler);
    }

    void onSend(MessageDelegate callback)
    {
        sendCallback = callback;
    }

    void onPacket(PacketDelegate callback)
    {
        packetCallback = callback;
    }

    bool search(const String& hostname, ResourceType type = ResourceType::PTR);

    bool send(Message& message);

protected:
    void onReceive(pbuf* buf, IpAddress remoteIP, uint16_t remotePort) override;

private:
    /*
     * Need a separate UDP connection for sending requests
     */
    class UdpOut : public UdpConnection
    {
    protected:
        void onReceive(pbuf* buf, IpAddress remoteIP, uint16_t remotePort) override;
    };

    Handler::List handlers;
    MessageDelegate sendCallback;
    PacketDelegate packetCallback;
    UdpOut out;
    bool active{false};
};

extern Server server;

} // namespace mDNS