mirror of
https://gitlab.isc.org/isc-projects/kea
synced 2025-09-05 00:15:17 +00:00
83 lines
2.5 KiB
C++
83 lines
2.5 KiB
C++
// Copyright (C) 2024 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 <asiolink/io_service_mgr.h>
|
|
#include <dhcpsrv/base_host_data_source.h>
|
|
#include <dhcpsrv/cfgmgr.h>
|
|
#include <hooks/hooks.h>
|
|
#include <process/daemon.h>
|
|
#include <mysql_hb_log.h>
|
|
#include <mysql_host_data_source.h>
|
|
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
using namespace isc::asiolink;
|
|
using namespace isc::dhcp;
|
|
using namespace isc::hooks;
|
|
using namespace isc::log;
|
|
using namespace isc::process;
|
|
using namespace std;
|
|
|
|
extern "C" {
|
|
|
|
/// @brief This function is called when the library is loaded.
|
|
///
|
|
/// @param handle library handle
|
|
/// @return 0 when initialization is successful, 1 otherwise
|
|
|
|
int load(LibraryHandle& /* handle */) {
|
|
// Make the hook library not loadable by d2 or ca.
|
|
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");
|
|
}
|
|
}
|
|
|
|
// Register MySQL HB factories with Host Managers
|
|
isc::dhcp::HostDataSourceFactory::registerFactory("mysql", MySqlHostDataSource::factory, true);
|
|
|
|
LOG_INFO(mysql_hb_logger, MYSQL_HB_INIT_OK);
|
|
return (0);
|
|
}
|
|
|
|
/// @brief This function is called when the library is unloaded.
|
|
///
|
|
/// @return 0 if deregistration was successful, 1 otherwise
|
|
int unload() {
|
|
// Unregister the factories and remove MySQL backends
|
|
isc::dhcp::HostDataSourceFactory::deregisterFactory("mysql", true);
|
|
|
|
LOG_INFO(mysql_hb_logger, MYSQL_HB_DEINIT_OK);
|
|
return (0);
|
|
}
|
|
|
|
/// @brief This function is called to retrieve the multi-threading compatibility.
|
|
///
|
|
/// @note: the compatibility is based on the assumption this hook library
|
|
/// is always called from the main thread.
|
|
///
|
|
/// @return 1 which means compatible with multi-threading.
|
|
int multi_threading_compatible() {
|
|
return (1);
|
|
}
|
|
|
|
} // end extern "C"
|