perfdhcp/main.cc

Go to the documentation of this file.

// Copyright (C) 2012-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/.
 
#include <config.h>
 
#include <exceptions/exceptions.h>
#include <log/logger_support.h>
#include <perfdhcp/avalanche_scen.h>
#include <perfdhcp/basic_scen.h>
#include <perfdhcp/command_options.h>
#include <util/filesystem.h>
 
#include <iostream>
#include <stdint.h>
 
using namespace isc::perfdhcp;
 
int
main(int argc, char* argv[]) {
    isc::util::file::setUmask();
 
    int ret_code = 0;
    std::string diags;
    bool parser_error = true;
    try {
        CommandOptions command_options;
        diags = command_options.getDiags();
        // If parser returns true it means that user specified
        // 'h' or 'v' command line option. Program shows the
        // help or version message and exits here.
        // The third argument indicates that the command line
        // should be printed when it gets parsed. This is useful
        // in particular when the command line needs to be
        // extracted from the log file.
        if (command_options.parse(argc, argv, true)) {
            return (ret_code);
        }
        isc::log::initLogger("perfdhcp");
        parser_error = false;
        auto scenario = command_options.getScenario();
        PerfSocket socket(command_options);
        if (scenario == Scenario::BASIC) {
            BasicScen scen(command_options, socket);
            ret_code = scen.run();
        } else if (scenario == Scenario::AVALANCHE) {
            AvalancheScen scen(command_options, socket);
            ret_code = scen.run();
        }
    } catch (const std::exception& e) {
        ret_code = 1;
        if (!parser_error) {
            std::cerr << std::endl << "ERROR: running perfdhcp: "
                      << e.what() << std::endl;
        } else {
            CommandOptions::usage();
            std::cerr << std::endl << "ERROR: parsing command line options: "
                      << e.what() << std::endl;
        }
        if (diags.find('e') != std::string::npos) {
            std::cerr << "Fatal error" << std::endl;
        }
    } catch (...) {
        ret_code = 1;
        if (!parser_error) {
            std::cerr << std::endl << "ERROR: running perfdhcp"
                      << std::endl;
        } else {
            CommandOptions::usage();
            std::cerr << std::endl << "ERROR: parsing command line options"
                      << std::endl;
        }
        if (diags.find('e') != std::string::npos) {
            std::cerr << "Fatal error" << std::endl;
        }
    }
    return (ret_code);
}