d2_controller.cc

Go to the documentation of this file.

// Copyright (C) 2013-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/.
 
#include <config.h>
 
#include <config/command_mgr.h>
#include <config/http_command_mgr.h>
#include <config/unix_command_mgr.h>
#include <d2/d2_controller.h>
#include <d2/d2_process.h>
#include <d2/parser_context.h>
#include <process/cfgrpt/config_report.h>
#include <stats/stats_mgr.h>
 
#include <stdlib.h>
 
using namespace isc::config;
using namespace isc::process;
using namespace isc::stats;
namespace ph = std::placeholders;
 
namespace isc {
namespace d2 {
 
const char* D2Controller::d2_app_name_ = "DhcpDdns";
 
const char* D2Controller::d2_bin_name_ = "kea-dhcp-ddns";
 
DControllerBasePtr&
D2Controller::instance() {
    // If the instance hasn't been created yet, create it.  Note this method
    // must use the base class singleton instance methods.
    if (!getController()) {
        DControllerBasePtr controller_ptr(new D2Controller());
        setController(controller_ptr);
    }
 
    return (getController());
}
 
DProcessBase* D2Controller::createProcess() {
    // Instantiate and return an instance of the D2 application process. Note
    // that the process is passed the controller's io_service.
    return (new D2Process(getAppName().c_str(), getIOService()));
}
 
D2Controller::D2Controller()
    : DControllerBase(d2_app_name_, d2_bin_name_) {
}
 
void
D2Controller::registerCommands() {
    // These are the commands always supported by the D2 server.
    // Please keep the list in alphabetic order.
    CommandMgr::instance().registerCommand(BUILD_REPORT_COMMAND,
        std::bind(&D2Controller::buildReportHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(CONFIG_GET_COMMAND,
        std::bind(&D2Controller::configGetHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(CONFIG_HASH_GET_COMMAND,
        std::bind(&D2Controller::configHashGetHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(CONFIG_RELOAD_COMMAND,
        std::bind(&D2Controller::configReloadHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(CONFIG_SET_COMMAND,
        std::bind(&D2Controller::configSetHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(CONFIG_TEST_COMMAND,
        std::bind(&D2Controller::configTestHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(CONFIG_WRITE_COMMAND,
        std::bind(&D2Controller::configWriteHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(SHUT_DOWN_COMMAND,
        std::bind(&D2Controller::shutdownHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(STATUS_GET_COMMAND,
        std::bind(&DControllerBase::statusGetHandler, this, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand(VERSION_GET_COMMAND,
        std::bind(&D2Controller::versionGetHandler, this, ph::_1, ph::_2));
 
    // Register statistic related commands.
    CommandMgr::instance().registerCommand("statistic-get",
        std::bind(&StatsMgr::statisticGetHandler, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand("statistic-get-all",
        std::bind(&StatsMgr::statisticGetAllHandler, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand("statistic-reset",
        std::bind(&StatsMgr::statisticResetHandler, ph::_1, ph::_2));
 
    CommandMgr::instance().registerCommand("statistic-reset-all",
        std::bind(&StatsMgr::statisticResetAllHandler, ph::_1, ph::_2));
}
 
void
D2Controller::deregisterCommands() {
    try {
        // Close command sockets.
        UnixCommandMgr::instance().closeCommandSocket();
        HttpCommandMgr::instance().close();
 
        // Deregister any registered commands (please keep in alphabetic order)
        CommandMgr::instance().deregisterCommand(BUILD_REPORT_COMMAND);
        CommandMgr::instance().deregisterCommand(CONFIG_GET_COMMAND);
        CommandMgr::instance().deregisterCommand(CONFIG_HASH_GET_COMMAND);
        CommandMgr::instance().deregisterCommand(CONFIG_RELOAD_COMMAND);
        CommandMgr::instance().deregisterCommand(CONFIG_SET_COMMAND);
        CommandMgr::instance().deregisterCommand(CONFIG_TEST_COMMAND);
        CommandMgr::instance().deregisterCommand(CONFIG_WRITE_COMMAND);
        CommandMgr::instance().deregisterCommand(SHUT_DOWN_COMMAND);
        CommandMgr::instance().deregisterCommand("statistic-get");
        CommandMgr::instance().deregisterCommand("statistic-get-all");
        CommandMgr::instance().deregisterCommand("statistic-reset");
        CommandMgr::instance().deregisterCommand("statistic-reset-all");
        CommandMgr::instance().deregisterCommand(STATUS_GET_COMMAND);
        CommandMgr::instance().deregisterCommand(VERSION_GET_COMMAND);
 
    } catch (...) {
        // What to do? Simply ignore...
    }
}
 
isc::data::ConstElementPtr
D2Controller::parseFile(const std::string& file_name) {
    isc::data::ConstElementPtr elements;
 
    // Read contents of the file and parse it as JSON
    D2ParserContext parser;
    elements = parser.parseFile(file_name, D2ParserContext::PARSER_DHCPDDNS);
    if (!elements) {
        isc_throw(isc::BadValue, "no configuration found in file");
    }
 
    return (elements);
}
 
D2Controller::~D2Controller() {
}
 
} // end namespace isc::d2
} // end namespace isc