// SPDX-License-Identifier: GPL-2.0-or-later
/*  cpufreq-bench CPUFreq microbenchmark
 *
 *  Copyright (C) 2008 Christian Kornacker <ckornacker@suse.de>
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <getopt.h>
#include <errno.h>

#include "config.h"
#include "system.h"
#include "benchmark.h"

static struct option long_options[] = {
        {"output",      1,      0,      'o'},
        {"sleep",       1,      0,      's'},
        {"load",        1,      0,      'l'},
        {"verbose",     0,      0,      'v'},
        {"cpu",         1,      0,      'c'},
        {"governor",    1,      0,      'g'},
        {"prio",        1,      0,      'p'},
        {"file",        1,      0,      'f'},
        {"cycles",      1,      0,      'n'},
        {"rounds",      1,      0,      'r'},
        {"load-step",   1,      0,      'x'},
        {"sleep-step",  1,      0,      'y'},
        {"help",        0,      0,      'h'},
        {0, 0, 0, 0}
};

/*******************************************************************
 usage
*******************************************************************/

void usage()
{
        printf("usage: ./bench\n");
        printf("Options:\n");
        printf(" -l, --load=<long int>\t\tinitial load time in us\n");
        printf(" -s, --sleep=<long int>\t\tinitial sleep time in us\n");
        printf(" -x, --load-step=<long int>\ttime to be added to load time, in us\n");
        printf(" -y, --sleep-step=<long int>\ttime to be added to sleep time, in us\n");
        printf(" -c, --cpu=<cpu #>\t\t\tCPU Nr. to use, starting at 0\n");
        printf(" -p, --prio=<priority>\t\t\tscheduler priority, HIGH, LOW or DEFAULT\n");
        printf(" -g, --governor=<governor>\t\tcpufreq governor to test\n");
        printf(" -n, --cycles=<int>\t\t\tload/sleep cycles\n");
        printf(" -r, --rounds<int>\t\t\tload/sleep rounds\n");
        printf(" -f, --file=<configfile>\t\tconfig file to use\n");
        printf(" -o, --output=<dir>\t\t\toutput path. Filename will be OUTPUTPATH/benchmark_TIMESTAMP.log\n");
        printf(" -v, --verbose\t\t\t\tverbose output on/off\n");
        printf(" -h, --help\t\t\t\tPrint this help screen\n");
        exit(1);
}

/*******************************************************************
 main
*******************************************************************/

int main(int argc, char **argv)
{
        int c;
        int option_index = 0;
        struct config *config = NULL;

        config = prepare_default_config();

        if (config == NULL)
                return EXIT_FAILURE;

        while (1) {
                c = getopt_long (argc, argv, "hg:o:s:l:vc:p:f:n:r:x:y:",
                                long_options, &option_index);
                if (c == -1)
                        break;

                switch (c) {
                case 'o':
                        if (config->output != NULL)
                                fclose(config->output);

                        config->output = prepare_output(optarg);

                        if (config->output == NULL)
                                return EXIT_FAILURE;

                        dprintf("user output path -> %s\n", optarg);
                        break;
                case 's':
                        sscanf(optarg, "%li", &config->sleep);
                        dprintf("user sleep time -> %s\n", optarg);
                        break;
                case 'l':
                        sscanf(optarg, "%li", &config->load);
                        dprintf("user load time -> %s\n", optarg);
                        break;
                case 'c':
                        sscanf(optarg, "%u", &config->cpu);
                        dprintf("user cpu -> %s\n", optarg);
                        break;
                case 'g':
                        strncpy(config->governor, optarg, 14);
                        dprintf("user governor -> %s\n", optarg);
                        break;
                case 'p':
                        if (string_to_prio(optarg) != SCHED_ERR) {
                                config->prio = string_to_prio(optarg);
                                dprintf("user prio -> %s\n", optarg);
                        } else {
                                if (config != NULL) {
                                        if (config->output != NULL)
                                                fclose(config->output);
                                        free(config);
                                }
                                usage();
                        }
                        break;
                case 'n':
                        sscanf(optarg, "%u", &config->cycles);
                        dprintf("user cycles -> %s\n", optarg);
                        break;
                case 'r':
                        sscanf(optarg, "%u", &config->rounds);
                        dprintf("user rounds -> %s\n", optarg);
                        break;
                case 'x':
                        sscanf(optarg, "%li", &config->load_step);
                        dprintf("user load_step -> %s\n", optarg);
                        break;
                case 'y':
                        sscanf(optarg, "%li", &config->sleep_step);
                        dprintf("user sleep_step -> %s\n", optarg);
                        break;
                case 'f':
                        if (prepare_config(optarg, config))
                                return EXIT_FAILURE;
                        break;
                case 'v':
                        config->verbose = 1;
                        dprintf("verbose output enabled\n");
                        break;
                case 'h':
                case '?':
                default:
                        if (config != NULL) {
                                if (config->output != NULL)
                                        fclose(config->output);
                                free(config);
                        }
                        usage();
                }
        }

        if (config->verbose) {
                printf("starting benchmark with parameters:\n");
                printf("config:\n\t"
                       "sleep=%li\n\t"
                       "load=%li\n\t"
                       "sleep_step=%li\n\t"
                       "load_step=%li\n\t"
                       "cpu=%u\n\t"
                       "cycles=%u\n\t"
                       "rounds=%u\n\t"
                       "governor=%s\n\n",
                       config->sleep,
                       config->load,
                       config->sleep_step,
                       config->load_step,
                       config->cpu,
                       config->cycles,
                       config->rounds,
                       config->governor);
        }

        prepare_user(config);
        prepare_system(config);
        start_benchmark(config);

        if (config->output != stdout)
                fclose(config->output);

        free(config);

        return EXIT_SUCCESS;
}