lease_cmds_callouts.cc

Go to the documentation of this file.

// Copyright (C) 2017-2026 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/.
 
// Functions accessed by the hooks framework use C linkage to avoid the name
// mangling that accompanies use of the C++ compiler as well as to avoid
// issues related to namespaces.
 
#include <config.h>
 
#include <cc/command_interpreter.h>
#include <dhcpsrv/cfgmgr.h>
#include <hooks/hooks.h>
#include <lease_cmds/binding_variables.h>
#include <lease_cmds/lease_cmds.h>
#include <process/daemon.h>
 
#include <lease_cmds_log.h>
#include <sflq_cmds.h>
 
using namespace isc::config;
using namespace isc::data;
using namespace isc::dhcp;
using namespace isc::hooks;
using namespace isc::process;
using namespace isc::lease_cmds;
 
namespace isc {
namespace lease_cmds {

BindingVariableMgrPtr binding_var_mgr;
 
} // end of namespace lease_cmds
} // end of namespace isc
 
 
extern "C" {

int lease4_add(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.leaseAddHandler(handle));
}

int lease6_add(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.leaseAddHandler(handle));
}

int lease6_bulk_apply(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.lease6BulkApplyHandler(handle));
}

int lease4_get(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.leaseGetHandler(handle));
}

int lease6_get(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.leaseGetHandler(handle));
}

int lease4_get_all(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetAllHandler(handle));
}

int lease6_get_all(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetAllHandler(handle));
}

int lease4_get_page(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetPageHandler(handle));
}

int lease6_get_page(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetPageHandler(handle));
}

int lease4_get_by_hw_address(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetByHwAddressHandler(handle));
}

int lease6_get_by_hw_address(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetByHwAddressHandler(handle));
}

int lease4_get_by_client_id(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetByClientIdHandler(handle));
}

int lease6_get_by_duid(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetByDuidHandler(handle));
}

int lease4_get_by_state(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetByStateHandler(handle));
}

int lease6_get_by_state(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetByStateHandler(handle));
}

int lease4_get_by_hostname(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetByHostnameHandler(handle));
}

int lease6_get_by_hostname(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return (lease_cmds.leaseGetByHostnameHandler(handle));
}

int lease4_del(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.lease4DelHandler(handle));
}

int lease6_del(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.lease6DelHandler(handle));
}

int lease4_update(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.lease4UpdateHandler(handle));
}

int lease6_update(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.lease6UpdateHandler(handle));
}

int lease4_wipe(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.lease4WipeHandler(handle));
}

int lease6_wipe(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.lease6WipeHandler(handle));
}

int lease4_resend_ddns(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.lease4ResendDdnsHandler(handle));
}

int lease6_resend_ddns(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.lease6ResendDdnsHandler(handle));
}

int lease4_write(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.leaseWriteHandler(handle));
}

int lease6_write(CalloutHandle& handle) {
    LeaseCmds lease_cmds;
    return(lease_cmds.leaseWriteHandler(handle));
}

int sflq_pool4_rebuild(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool4RebuildHandler(handle));
}

int sflq_pool4_get_all(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool4GetAllHandler(handle));
}

int sflq_pool4_get_by_subnet(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool4GetBySubnetHandler(handle));
}

int sflq_pool4_get_by_range(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool4GetByRangeHandler(handle));
}

int sflq_pool4_del(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool4DelHandler(handle));
}

int sflq_pool6_rebuild(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool6RebuildHandler(handle));
}

int sflq_pool6_get_all(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool6GetAllHandler(handle));
}

int sflq_pool6_get_by_subnet(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool6GetBySubnetHandler(handle));
}

int sflq_pool6_get_by_range(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool6GetByRangeHandler(handle));
}

int sflq_pool6_del(CalloutHandle& handle) {
    SflqCmds sflq_cmds;
    return(sflq_cmds.sflqPool6DelHandler(handle));
}

int load(LibraryHandle& handle) {
    // Make the hook library not loadable by d2.
    uint16_t family = CfgMgr::instance().getFamily();
    const std::string& proc_name = Daemon::getProcName();
    if (family == AF_INET) {
        if (proc_name != "kea-dhcp4") {
            isc_throw(isc::Unexpected, "Bad process name: " << proc_name
                      << ", expected kea-dhcp4");
        }
    } else {
        if (proc_name != "kea-dhcp6") {
            isc_throw(isc::Unexpected, "Bad process name: " << proc_name
                      << ", expected kea-dhcp6");
        }
    }
 
    handle.registerCommandCallout("lease4-add", lease4_add);
    handle.registerCommandCallout("lease6-add", lease6_add);
    handle.registerCommandCallout("lease6-bulk-apply", lease6_bulk_apply);
    handle.registerCommandCallout("lease4-get", lease4_get);
    handle.registerCommandCallout("lease6-get", lease6_get);
    handle.registerCommandCallout("lease4-get-all", lease4_get_all);
    handle.registerCommandCallout("lease6-get-all", lease6_get_all);
    handle.registerCommandCallout("lease4-get-page", lease4_get_page);
    handle.registerCommandCallout("lease6-get-page", lease6_get_page);
    handle.registerCommandCallout("lease4-get-by-hw-address",
                                  lease4_get_by_hw_address);
    handle.registerCommandCallout("lease6-get-by-hw-address",
                                  lease6_get_by_hw_address);
    handle.registerCommandCallout("lease4-get-by-client-id",
                                  lease4_get_by_client_id);
    handle.registerCommandCallout("lease6-get-by-duid", lease6_get_by_duid);
    handle.registerCommandCallout("lease4-get-by-state", lease4_get_by_state);
    handle.registerCommandCallout("lease6-get-by-state", lease6_get_by_state);
    handle.registerCommandCallout("lease4-get-by-hostname",
                                  lease4_get_by_hostname);
    handle.registerCommandCallout("lease6-get-by-hostname",
                                  lease6_get_by_hostname);
    handle.registerCommandCallout("lease4-del", lease4_del);
    handle.registerCommandCallout("lease6-del", lease6_del);
    handle.registerCommandCallout("lease4-update", lease4_update);
    handle.registerCommandCallout("lease6-update", lease6_update);
    handle.registerCommandCallout("lease4-wipe", lease4_wipe);
    handle.registerCommandCallout("lease6-wipe", lease6_wipe);
    handle.registerCommandCallout("lease4-resend-ddns", lease4_resend_ddns);
    handle.registerCommandCallout("lease6-resend-ddns", lease6_resend_ddns);
    handle.registerCommandCallout("lease4-write", lease4_write);
    handle.registerCommandCallout("lease6-write", lease6_write);
    handle.registerCommandCallout("sflq-pool4-rebuild", sflq_pool4_rebuild);
    handle.registerCommandCallout("sflq-pool4-get-all", sflq_pool4_get_all);
    handle.registerCommandCallout("sflq-pool4-get-by-subnet", sflq_pool4_get_by_subnet);
    handle.registerCommandCallout("sflq-pool4-get-by-range", sflq_pool4_get_by_range);
    handle.registerCommandCallout("sflq-pool4-del", sflq_pool4_del);
    handle.registerCommandCallout("sflq-pool6-rebuild", sflq_pool6_rebuild);
    handle.registerCommandCallout("sflq-pool6-get-all", sflq_pool6_get_all);
    handle.registerCommandCallout("sflq-pool6-get-by-subnet", sflq_pool6_get_by_subnet);
    handle.registerCommandCallout("sflq-pool6-get-by-range", sflq_pool6_get_by_range);
    handle.registerCommandCallout("sflq-pool6-del", sflq_pool6_del);
 
    // Instantiate the binding-variables manager singleton.
    binding_var_mgr.reset(new BindingVariableMgr(family));
 
    try {
        // Configure binding variable manager using the hook library's parameters.
        ConstElementPtr json = handle.getParameters();
        if (json) {
            binding_var_mgr->configure(json);
        }
    } catch (const std::exception& ex) {
        LOG_ERROR(lease_cmds_logger, LEASE_CMDS_LOAD_ERROR)
            .arg(ex.what());
        return (1);
    }
 
    LOG_INFO(lease_cmds_logger, LEASE_CMDS_INIT_OK);
    return (0);
}

int unload() {
    LOG_INFO(lease_cmds_logger, LEASE_CMDS_DEINIT_OK);
    return (0);
}

int multi_threading_compatible() {
    return (1);
}

int lease4_offer(CalloutHandle& handle) {
    CalloutHandle::CalloutNextStep status = handle.getStatus();
    if (status == CalloutHandle::NEXT_STEP_DROP ||
        status == CalloutHandle::NEXT_STEP_SKIP) {
        return (0);
    }
 
    try {
        LeaseCmds lease_cmds;
        lease_cmds.lease4Offer(handle, binding_var_mgr);
    } catch (const std::exception& ex) {
        LOG_ERROR(lease_cmds_logger, LEASE_CMDS_LEASE4_OFFER_FAILED)
            .arg(ex.what());
        return (1);
    }
 
    return (0);
}

int leases4_committed(CalloutHandle& handle) {
    CalloutHandle::CalloutNextStep status = handle.getStatus();
    if (status == CalloutHandle::NEXT_STEP_DROP ||
        status == CalloutHandle::NEXT_STEP_SKIP) {
        return (0);
    }
 
    try {
        LeaseCmds lease_cmds;
        lease_cmds.leases4Committed(handle, binding_var_mgr);
    } catch (const std::exception& ex) {
        LOG_ERROR(lease_cmds_logger, LEASE_CMDS_LEASES4_COMMITTED_FAILED)
            .arg(ex.what());
        return (1);
    }
 
    return (0);
}

int leases6_committed(CalloutHandle& handle) {
    CalloutHandle::CalloutNextStep status = handle.getStatus();
    if (status == CalloutHandle::NEXT_STEP_DROP ||
        status == CalloutHandle::NEXT_STEP_SKIP) {
        return (0);
    }
 
    try {
        LeaseCmds lease_cmds;
        lease_cmds.leases6Committed(handle, binding_var_mgr);
    } catch (const std::exception& ex) {
        LOG_ERROR(lease_cmds_logger, LEASE_CMDS_LEASES6_COMMITTED_FAILED)
            .arg(ex.what());
        return (1);
    }
 
    return (0);
}
 
} // end extern "C"