lease_query_impl.h

Go to the documentation of this file.

// Copyright (C) 2020-2025 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
#ifndef LEASE_QUERY_IMPL_H
#define LEASE_QUERY_IMPL_H
 
#include <exceptions/exceptions.h>
#include <asiolink/io_address.h>
#include <asiolink/io_service.h>
#include <dhcp/pkt.h>
#include <cc/data.h>
#include <cc/simple_parser.h>
 
#include <boost/functional/hash.hpp>
#include <unordered_set>
 
namespace isc {
namespace lease_query {

class QueryTerminated : public isc::Exception {
public:
    QueryTerminated(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) {
    }
};
 
#define CHECK_TERMINATED \
    if (LeaseQueryImpl::terminated_) { \
        isc_throw(isc::lease_query::QueryTerminated, "terminated"); \
    }

class AddressList {
public:
    AddressList(uint16_t family)
    : family_(family) { };

    void insert(const isc::asiolink::IOAddress& address);

    bool contains(const isc::asiolink::IOAddress& address) const;

    size_t size() const {
        return (addresses_.size());
    }

    uint16_t getFamily() {
        return (family_);
    }
 
private:
    uint16_t family_;

    std::unordered_set<asiolink::IOAddress, boost::hash<asiolink::IOAddress> > addresses_;
};

class LeaseQueryImpl : public boost::noncopyable {
public:
    LeaseQueryImpl(uint16_t family, const isc::data::ConstElementPtr config);

    virtual ~LeaseQueryImpl();

    bool isRequester(const isc::asiolink::IOAddress& address) const;

    size_t getNumRequesters() const {
        return (address_list_.size());
    }

    virtual void processQuery(isc::dhcp::PktPtr base_query) const = 0;

    static const isc::data::SimpleKeywords LEASE_QUERY_KEYWORDS;

    uint16_t getFamily() {
        return (address_list_.getFamily());
    }

    isc::asiolink::IOServicePtr getIOService() {
        return (io_service_);
    }

    void setIOService(isc::asiolink::IOServicePtr io_service) {
        io_service_ = io_service;
    }

    static bool terminated_;

    static size_t PageSize;
 
private:

    isc::asiolink::IOServicePtr io_service_;

    AddressList address_list_;
};

typedef boost::shared_ptr<LeaseQueryImpl> LeaseQueryImplPtr;
 
} // end of namespace isc::lease_query
} // end of namespace isc
 
#endif // LEASE_QUERY_IMPL_H