// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2021 Red Hat Inc, Daniel Bristot de Oliveira <bristot@kernel.org>
 */

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

#include "osnoise.h"
#include "timerlat.h"

/*
 * rtla_usage - print rtla usage
 */
static void rtla_usage(int err)
{
 int i;

 static const char *msg[] = {
  "",
  "rtla version " VERSION,
  "",
  "  usage: rtla COMMAND ...",
  "",
  "  commands:",
  "     osnoise  - gives information about the operating system noise (osnoise)",
  "     hwnoise  - gives information about hardware-related noise",
  "     timerlat - measures the timer irq and thread latency",
  "",
  NULL,
 };

 for (i = 0; msg[i]; i++)
  fprintf(stderr, "%s\n", msg[i]);
 exit(err);
}

/*
 * run_command - try to run a rtla tool command
 *
 * It returns 0 if it fails. The tool's main will generally not
 * return as they should call exit().
 */
int run_command(int argc, char **argv, int start_position)
{
 if (strcmp(argv[start_position], "osnoise") == 0) {
  osnoise_main(argc-start_position, &argv[start_position]);
  goto ran;
 } else if (strcmp(argv[start_position], "hwnoise") == 0) {
  hwnoise_main(argc-start_position, &argv[start_position]);
  goto ran;
 } else if (strcmp(argv[start_position], "timerlat") == 0) {
  timerlat_main(argc-start_position, &argv[start_position]);
  goto ran;
 }

 return 0;
ran:
 return 1;
}

int main(int argc, char *argv[])
{
 int retval;

 /* is it an alias? */
 retval = run_command(argc, argv, 0);
 if (retval)
  exit(0);

 if (argc < 2)
  goto usage;

 if (strcmp(argv[1], "-h") == 0) {
  rtla_usage(0);
 } else if (strcmp(argv[1], "--help") == 0) {
  rtla_usage(0);
 }

 retval = run_command(argc, argv, 1);
 if (retval)
  exit(0);

usage:
 rtla_usage(1);
 exit(1);
}