AbstractHTTPScanner.h

#pragma once

#include <set>

#include <Poco/Net/HTTPRequest.h>
#include <Poco/Net/IPAddress.h>
#include <Poco/Net/NetworkInterface.h>
#include <Poco/Net/SocketAddress.h>
#include <Poco/Timespan.h>

#include "loop/StopControl.h"
#include "net/HTTPEntireResponse.h"
#include "net/IPAddressRange.h"
#include "util/Loggable.h"

namespace BeeeOn {

class AbstractHTTPScanner : protected Loggable {
public:
        AbstractHTTPScanner();
        AbstractHTTPScanner(const std::string& path, uint16_t port, const Poco::Net::IPAddress& minNetMask);
        ~AbstractHTTPScanner();

        std::vector<Poco::Net::SocketAddress> scan(const uint32_t maxResponseLength);

        void cancel();

        void setPath(const std::string& path);
        void setPort(Poco::UInt16 port);
        void setMinNetMask(const Poco::Net::IPAddress& minNetMask);
        void setPingTimeout(const Poco::Timespan& pingTimeout);
        void setHTTPTimeout(const Poco::Timespan& httpTimeout);
        void setBlackList(const std::set<std::string>& set);

        std::string path();
        Poco::UInt16 port();

protected:
        virtual void prepareRequest(Poco::Net::HTTPRequest& request) = 0;
        virtual bool isValidResponse(const std::string& response) = 0;

        void probeInterface(
                StopControl::Run &run,
                const Poco::Net::NetworkInterface& interface,
                std::vector<Poco::Net::SocketAddress>& devices,
                const Poco::Int64 maxResponseLength);

        void probeAddressRange(
                StopControl::Run &run,  
                const IPAddressRange& range,
                std::vector<Poco::Net::SocketAddress>& devices,
                const Poco::Int64 maxResponseLength);

        HTTPEntireResponse sendRequest(const Poco::Net::SocketAddress& socketAddress,
                const Poco::Int64 maxResponseLength);

        bool addressIncompatible(const Poco::Net::NetworkInterface::AddressTuple& addressTuple) const;

        std::vector<Poco::Net::NetworkInterface> listNetworkInterfaces();

private:
        std::string m_path;
        uint16_t m_port;
        Poco::Net::IPAddress m_minNetMask;
        Poco::Timespan m_pingTimeout;
        Poco::Timespan m_httpTimeout;
        std::set<std::string> m_blackList;
        StopControl m_stopControl;
};

}