host_mgr.h

Go to the documentation of this file.

// Copyright (C) 2014-2021 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 HOST_MGR_H
#define HOST_MGR_H

#include <asiolink/io_service.h>
#include <database/database_connection.h>
#include <dhcpsrv/base_host_data_source.h>
#include <dhcpsrv/cache_host_data_source.h>
#include <dhcpsrv/host.h>
#include <dhcpsrv/subnet_id.h>
#include <boost/noncopyable.hpp>
#include <boost/scoped_ptr.hpp>
#include <string>

namespace isc {
namespace dhcp {

class HostMgr : public boost::noncopyable, public BaseHostDataSource {
public:

    static void create();

    static void addBackend(const std::string& access);

    static bool delBackend(const std::string& db_type);

    static bool delBackend(const std::string& db_type,
                           const std::string& access,
                           bool if_unusable = false);

    static void delAllBackends();

    static bool checkCacheBackend(bool logging = false);

    static HostMgr& instance();

    virtual ConstHostCollection
    getAll(const Host::IdentifierType& identifier_type,
           const uint8_t* identifier_begin,
           const size_t identifier_len) const;

    virtual ConstHostCollection
    getAll4(const SubnetID& subnet_id) const;

    virtual ConstHostCollection
    getAll6(const SubnetID& subnet_id) const;

    virtual ConstHostCollection
    getAllbyHostname(const std::string& hostname) const;

    virtual ConstHostCollection
    getAllbyHostname4(const std::string& hostname, const SubnetID& subnet_id) const;

    virtual ConstHostCollection
    getAllbyHostname6(const std::string& hostname, const SubnetID& subnet_id) const;

    virtual ConstHostCollection
    getPage4(const SubnetID& subnet_id,
             size_t& source_index,
             uint64_t lower_host_id,
             const HostPageSize& page_size) const;

    virtual ConstHostCollection
    getPage6(const SubnetID& subnet_id,
             size_t& source_index,
             uint64_t lower_host_id,
             const HostPageSize& page_size) const;

    virtual ConstHostCollection
    getPage4(size_t& source_index,
             uint64_t lower_host_id,
             const HostPageSize& page_size) const;

    virtual ConstHostCollection
    getPage6(size_t& source_index,
             uint64_t lower_host_id,
             const HostPageSize& page_size) const;

    virtual ConstHostCollection
    getAll4(const asiolink::IOAddress& address) const;

    virtual ConstHostPtr
    get4Any(const SubnetID& subnet_id,
            const Host::IdentifierType& identifier_type,
            const uint8_t* identifier_begin,
            const size_t identifier_len) const;

    virtual ConstHostPtr
    get4(const SubnetID& subnet_id, const Host::IdentifierType& identifier_type,
         const uint8_t* identifier_begin, const size_t identifier_len) const;

    virtual ConstHostPtr
    get4(const SubnetID& subnet_id, const asiolink::IOAddress& address) const;

    virtual ConstHostCollection
    getAll4(const SubnetID& subnet_id,
            const asiolink::IOAddress& address) const;

    virtual ConstHostPtr
    get6Any(const SubnetID& subnet_id,
            const Host::IdentifierType& identifier_type,
            const uint8_t* identifier_begin,
            const size_t identifier_len) const;

    virtual ConstHostPtr
    get6(const SubnetID& subnet_id, const Host::IdentifierType& identifier_type,
         const uint8_t* identifier_begin, const size_t identifier_len) const;

    virtual ConstHostPtr
    get6(const asiolink::IOAddress& prefix, const uint8_t prefix_len) const;

    virtual ConstHostPtr
    get6(const SubnetID& subnet_id, const asiolink::IOAddress& addr) const;

    virtual ConstHostCollection
    getAll6(const SubnetID& subnet_id,
            const asiolink::IOAddress& address) const;

    virtual void add(const HostPtr& host);

    virtual bool del(const SubnetID& subnet_id, const asiolink::IOAddress& addr);

    virtual bool
    del4(const SubnetID& subnet_id, const Host::IdentifierType& identifier_type,
         const uint8_t* identifier_begin, const size_t identifier_len);

    virtual bool
    del6(const SubnetID& subnet_id, const Host::IdentifierType& identifier_type,
         const uint8_t* identifier_begin, const size_t identifier_len);

    virtual std::string getType() const {
        return (std::string("host_mgr"));
    }

    HostDataSourceList& getHostDataSourceList() {
        return (alternate_sources_);
    }

    HostDataSourcePtr getHostDataSource() const;

    bool getNegativeCaching() const {
        return (negative_caching_);
    }

    void setNegativeCaching(bool negative_caching) {
        negative_caching_ = negative_caching;
    }

    bool getDisableSingleQuery() const {
        return (disable_single_query_);
    }

    void setDisableSingleQuery(bool disable_single_query) {
        disable_single_query_ = disable_single_query;
    }

    virtual bool setIPReservationsUnique(const bool unique);

    bool getIPReservationsUnique() const {
        return (ip_reservations_unique_);
    }

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

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

protected:

    bool negative_caching_;

    bool disable_single_query_;

    virtual void cache(ConstHostPtr host) const;

    virtual void cacheNegative(const SubnetID& ipv4_subnet_id,
                               const SubnetID& ipv6_subnet_id,
                               const Host::IdentifierType& identifier_type,
                               const uint8_t* identifier_begin,
                               const size_t identifier_len) const;

private:

    bool ip_reservations_unique_;

    HostMgr() : negative_caching_(false), disable_single_query_(false),
                ip_reservations_unique_(true) { }

    HostDataSourceList alternate_sources_;

    CacheHostDataSourcePtr cache_ptr_;

    static boost::scoped_ptr<HostMgr>& getHostMgrPtr();

    static isc::asiolink::IOServicePtr io_service_;
};

}  // namespace dhcp
}  // namespace isc

#endif // HOST_MGR_H