load_unload.cc

Go to the documentation of this file.

// Copyright (C) 2013-2015 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 <hooks/hooks.h>
#include <user_chk_log.h>
#include <user_registry.h>
#include <user_file.h>
 
#include <iostream>
#include <fstream>
#include <errno.h>
 
using namespace isc::hooks;
using namespace user_chk;
 
UserRegistryPtr user_registry;
 
std::fstream user_chk_output;
 
const char* registry_fname = "/tmp/user_chk_registry.txt";
 
const char* user_chk_output_fname = "/tmp/user_chk_outcome.txt";
 
const char* query_user_id_label = "query_user_id";
 
const char* registered_user_label = "registered_user";
 
const char* default_user4_id_str = "00000000";
 
const char *default_user6_id_str = "00000000";
 
// 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.
extern "C" {
 
int load(LibraryHandle&) {
    // non-zero indicates an error.
    int ret_val = 0;
 
    try {
        // Instantiate the registry.
        user_registry.reset(new UserRegistry());
 
        // Create the data source.
        UserDataSourcePtr user_file(new UserFile(registry_fname));
 
        // Set the registry's data source
        user_registry->setSource(user_file);
 
        // Do an initial load of the registry.
        user_registry->refresh();
 
        // zero out the errno to be safe
        errno = 0;
 
        // Open up the output file for user_chk results.
        user_chk_output.open(user_chk_output_fname,
                     std::fstream::out | std::fstream::app);
 
        if (!user_chk_output) {
            // Grab the system error message.
            const char* errmsg = strerror(errno);
            isc_throw(isc::Unexpected, "Cannot open output file: "
                                       << user_chk_output_fname
                                       << " reason: " << errmsg);
        }
    }
    catch (const std::exception& ex) {
        // Log the error and return failure.
        LOG_ERROR(user_chk_logger, USER_CHK_HOOK_LOAD_ERROR)
            .arg(ex.what());
        ret_val = 1;
    }
 
    return (ret_val);
}
 
int unload() {
    try {
        user_registry.reset();
        if (user_chk_output.is_open()) {
            user_chk_output.close();
        }
    } catch (const std::exception& ex) {
        // On the off chance something goes awry, catch it and log it.
        // @todo Not sure if we should return a non-zero result or not.
        LOG_ERROR(user_chk_logger, USER_CHK_HOOK_UNLOAD_ERROR)
            .arg(ex.what());
    }
 
    return (0);
}
 
}