1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138 | // Copyright (C) 2021-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 <dhcpsrv/lease_mgr_factory.h>
#include <hooks/hooks.h>
#include <process/daemon.h>
#include <pgsql_cb_impl.h><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <pgsql_cb_dhcp4.h><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <pgsql_cb_dhcp6.h><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <pgsql_cb_log.h><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <pgsql_hb_log.h><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <pgsql_host_data_source.h><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <pgsql_lb_log.h><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <pgsql_lease_mgr.h><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <sstream><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
#include <string><--- Include file: not found. Please note: Cppcheck does not need standard library headers to get proper results.
using namespace isc::asiolink;
using namespace isc::cb;
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 PostgreSQL CB factories with CB Managers
PgSqlConfigBackendDHCPv4::registerBackendType();
PgSqlConfigBackendDHCPv6::registerBackendType();
// Register PostgreSQL HB factories with Host Managers
HostDataSourceFactory::registerFactory("postgresql",
PgSqlHostDataSource::factory,
true,
PgSqlHostDataSource::getDBVersion);
// Register PostgreSQL LB factories with Lease Managers
LeaseMgrFactory::registerFactory("postgresql",
PgSqlLeaseMgr::factory,
true,
PgSqlLeaseMgr::getDBVersion);
LOG_INFO(pgsql_lb_logger, PGSQL_INIT_OK);
return (0);
}
/// @brief This function is called by the DHCPv4 server when it is configured.
///
/// The only purpose of this callout is to retrieve io_service_ reference.
///
/// @param handle callout handle passed to the callout.
/// @return 0 on success, 1 otherwise.
int dhcp4_srv_configured(CalloutHandle& /* handle */) {
PgSqlConfigBackendImpl::setIOService(IOServicePtr(new IOService()));
IOServiceMgr::instance().registerIOService(PgSqlConfigBackendImpl::getIOService());
return (0);
}
/// @brief This function is called by the DHCPv6 server when it is configured.
///
/// The only purpose of this callout is to retrieve io_service_ reference.
///
/// @param handle callout handle passed to the callout.
/// @return 0 on success, 1 otherwise.
int dhcp6_srv_configured(CalloutHandle& /* handle */) {
PgSqlConfigBackendImpl::setIOService(IOServicePtr(new IOService()));
IOServiceMgr::instance().registerIOService(PgSqlConfigBackendImpl::getIOService());
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 PostgreSQL backends
PgSqlConfigBackendDHCPv4::unregisterBackendType();
PgSqlConfigBackendDHCPv6::unregisterBackendType();
IOServicePtr io_service = PgSqlConfigBackendImpl::getIOService();
if (io_service) {
IOServiceMgr::instance().unregisterIOService(io_service);
io_service->stopAndPoll();
PgSqlConfigBackendImpl::setIOService(IOServicePtr());
}
// Unregister the factories and remove PostgreSQL backends
HostDataSourceFactory::deregisterFactory("postgresql", true);
// Unregister the factories and remove PostgreSQL backends
LeaseMgrFactory::deregisterFactory("postgresql", true);
LOG_INFO(pgsql_lb_logger, PGSQL_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"
|