// SPDX-License-Identifier: GPL-2.0
#include "tests.h"
#include "debug.h"
#include "evlist.h"
#include "cgroup.h"
#include "rblist.h"
#include "metricgroup.h"
#include "parse-events.h"
#include "pmu-events/pmu-events.h"
#include "pfm.h"
#include <subcmd/parse-options.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

static int test_expand_events(struct evlist *evlist)
{
 int i, ret = TEST_FAIL;
 int nr_events;
 bool was_group_event;
 int nr_members;  /* for the first evsel only */
 const char cgrp_str[] = "A,B,C";
 const char *cgrp_name[] = { "A", "B", "C" };
 int nr_cgrps = ARRAY_SIZE(cgrp_name);
 char **ev_name;
 struct evsel *evsel;

 TEST_ASSERT_VAL("evlist is empty", !evlist__empty(evlist));

 nr_events = evlist->core.nr_entries;
 ev_name = calloc(nr_events, sizeof(*ev_name));
 if (ev_name == NULL) {
  pr_debug("memory allocation failure\n");
  return TEST_FAIL;
 }
 i = 0;
 evlist__for_each_entry(evlist, evsel) {
  ev_name[i] = strdup(evsel->name);
  if (ev_name[i] == NULL) {
   pr_debug("memory allocation failure\n");
   goto out;
  }
  i++;
 }
 /* remember grouping info */
 was_group_event = evsel__is_group_event(evlist__first(evlist));
 nr_members = evlist__first(evlist)->core.nr_members;

 ret = evlist__expand_cgroup(evlist, cgrp_str, false);
 if (ret < 0) {
  pr_debug("failed to expand events for cgroups\n");
  goto out;
 }

 ret = TEST_FAIL;
 if (evlist->core.nr_entries != nr_events * nr_cgrps) {
  pr_debug("event count doesn't match\n");
  goto out;
 }

 i = 0;
 evlist__for_each_entry(evlist, evsel) {
  if (!evsel__name_is(evsel, ev_name[i % nr_events])) {
   pr_debug("event name doesn't match:\n");
   pr_debug("  evsel[%d]: %s\n  expected: %s\n",
     i, evsel->name, ev_name[i % nr_events]);
   goto out;
  }
  if (strcmp(evsel->cgrp->name, cgrp_name[i / nr_events])) {
   pr_debug("cgroup name doesn't match:\n");
   pr_debug("  evsel[%d]: %s\n  expected: %s\n",
     i, evsel->cgrp->name, cgrp_name[i / nr_events]);
   goto out;
  }

  if ((i % nr_events) == 0) {
   if (evsel__is_group_event(evsel) != was_group_event) {
    pr_debug("event group doesn't match: got %s, expect %s\n",
      evsel__is_group_event(evsel) ? "true" : "false",
      was_group_event ? "true" : "false");
    goto out;
   }
   if (evsel->core.nr_members != nr_members) {
    pr_debug("event group member doesn't match: %d vs %d\n",
      evsel->core.nr_members, nr_members);
    goto out;
   }
  }
  i++;
 }
 ret = TEST_OK;

out: for (i = 0; i < nr_events; i++)
  free(ev_name[i]);
 free(ev_name);
 return ret;
}

static int expand_default_events(void)
{
 int ret;
 struct evlist *evlist = evlist__new_default();

 TEST_ASSERT_VAL("failed to get evlist", evlist);

 ret = test_expand_events(evlist);
 evlist__delete(evlist);
 return ret;
}

static int expand_group_events(void)
{
 int ret;
 struct evlist *evlist;
 struct parse_events_error err;
 const char event_str[] = "{cycles,instructions}";

 symbol_conf.event_group = true;

 evlist = evlist__new();
 TEST_ASSERT_VAL("failed to get evlist", evlist);

 parse_events_error__init(&err);
 ret = parse_events(evlist, event_str, &err);
 if (ret < 0) {
  pr_debug("failed to parse event '%s', err %d\n", event_str, ret);
  parse_events_error__print(&err, event_str);
  goto out;
 }

 ret = test_expand_events(evlist);
out:
 parse_events_error__exit(&err);
 evlist__delete(evlist);
 return ret;
}

static int expand_libpfm_events(void)
{
 int ret;
 struct evlist *evlist;
 const char event_str[] = "CYCLES";
 struct option opt = {
  .value = &evlist,
 };

 symbol_conf.event_group = true;

 evlist = evlist__new();
 TEST_ASSERT_VAL("failed to get evlist", evlist);

 ret = parse_libpfm_events_option(&opt, event_str, 0);
 if (ret < 0) {
  pr_debug("failed to parse libpfm event '%s', err %d\n",
    event_str, ret);
  goto out;
 }
 if (evlist__empty(evlist)) {
  pr_debug("libpfm was not enabled\n");
  goto out;
 }

 ret = test_expand_events(evlist);
out:
 evlist__delete(evlist);
 return ret;
}

static int expand_metric_events(void)
{
 int ret;
 struct evlist *evlist;
 const char metric_str[] = "CPI";
 const struct pmu_metrics_table *pme_test;

 evlist = evlist__new();
 TEST_ASSERT_VAL("failed to get evlist", evlist);

 pme_test = find_core_metrics_table("testarch", "testcpu");
 ret = metricgroup__parse_groups_test(evlist, pme_test, metric_str);
 if (ret < 0) {
  pr_debug("failed to parse '%s' metric\n", metric_str);
  goto out;
 }

 ret = test_expand_events(evlist);

out:
 evlist__delete(evlist);
 return ret;
}

static int test__expand_cgroup_events(struct test_suite *test __maybe_unused,
          int subtest __maybe_unused)
{
 int ret;

 ret = expand_default_events();
 TEST_ASSERT_EQUAL("failed to expand default events", ret, 0);

 ret = expand_group_events();
 TEST_ASSERT_EQUAL("failed to expand event group", ret, 0);

 ret = expand_libpfm_events();
 TEST_ASSERT_EQUAL("failed to expand event group", ret, 0);

 ret = expand_metric_events();
 TEST_ASSERT_EQUAL("failed to expand metric events", ret, 0);

 return ret;
}

DEFINE_SUITE("Event expansion for cgroups", expand_cgroup_events);