// SPDX-License-Identifier: GPL-2.0
#include <linux/compiler.h>
#include "event.h"
#include "tests.h"
#include "stat.h"
#include "counts.h"
#include "debug.h"
#include "util/synthetic-events.h"

static bool has_term(struct perf_record_stat_config *config,
       u64 tag, u64 val)
{
 unsigned i;

 for (i = 0; i < config->nr; i++) {
  if ((config->data[i].tag == tag) &&
      (config->data[i].val == val))
   return true;
 }

 return false;
}

static int process_stat_config_event(const struct perf_tool *tool __maybe_unused,
         union perf_event *event,
         struct perf_sample *sample __maybe_unused,
         struct machine *machine __maybe_unused)
{
 struct perf_record_stat_config *config = &event->stat_config;
 struct perf_stat_config test_stat_config = {};

#define HAS(term, val) \
 has_term(config, PERF_STAT_CONFIG_TERM__##term, val)

 TEST_ASSERT_VAL("wrong nr",        config->nr == PERF_STAT_CONFIG_TERM__MAX);
 TEST_ASSERT_VAL("wrong aggr_mode", HAS(AGGR_MODE, AGGR_CORE));
 TEST_ASSERT_VAL("wrong scale",     HAS(SCALE, 1));
 TEST_ASSERT_VAL("wrong interval",  HAS(INTERVAL, 1));

#undef HAS

 perf_event__read_stat_config(&test_stat_config, config);

 TEST_ASSERT_VAL("wrong aggr_mode", test_stat_config.aggr_mode == AGGR_CORE);
 TEST_ASSERT_VAL("wrong scale",     test_stat_config.scale == 1);
 TEST_ASSERT_VAL("wrong interval",  test_stat_config.interval == 1);
 return 0;
}

static int test__synthesize_stat_config(struct test_suite *test __maybe_unused,
     int subtest __maybe_unused)
{
 struct perf_stat_config test_stat_config = {
  .aggr_mode = AGGR_CORE,
  .scale  = 1,
  .interval = 1,
 };

 TEST_ASSERT_VAL("failed to synthesize stat_config",
  !perf_event__synthesize_stat_config(NULL, &test_stat_config,
          process_stat_config_event,
          NULL));

 return 0;
}

static int process_stat_event(const struct perf_tool *tool __maybe_unused,
         union perf_event *event,
         struct perf_sample *sample __maybe_unused,
         struct machine *machine __maybe_unused)
{
 struct perf_record_stat *st = &event->stat;

 TEST_ASSERT_VAL("wrong cpu",    st->cpu    == 1);
 TEST_ASSERT_VAL("wrong thread", st->thread == 2);
 TEST_ASSERT_VAL("wrong id",     st->id     == 3);
 TEST_ASSERT_VAL("wrong val",    st->val    == 100);
 TEST_ASSERT_VAL("wrong run",    st->ena    == 200);
 TEST_ASSERT_VAL("wrong ena",    st->run    == 300);
 return 0;
}

static int test__synthesize_stat(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
{
 struct perf_counts_values count;

 count.val = 100;
 count.ena = 200;
 count.run = 300;

 TEST_ASSERT_VAL("failed to synthesize stat_config",
   !perf_event__synthesize_stat(NULL, (struct perf_cpu){.cpu = 1}, 2, 3,
           &count, process_stat_event, NULL));

 return 0;
}

static int process_stat_round_event(const struct perf_tool *tool __maybe_unused,
        union perf_event *event,
        struct perf_sample *sample __maybe_unused,
        struct machine *machine __maybe_unused)
{
 struct perf_record_stat_round *stat_round = &event->stat_round;

 TEST_ASSERT_VAL("wrong time", stat_round->time == 0xdeadbeef);
 TEST_ASSERT_VAL("wrong type", stat_round->type == PERF_STAT_ROUND_TYPE__INTERVAL);
 return 0;
}

static int test__synthesize_stat_round(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
{
 TEST_ASSERT_VAL("failed to synthesize stat_config",
  !perf_event__synthesize_stat_round(NULL, 0xdeadbeef, PERF_STAT_ROUND_TYPE__INTERVAL,
         process_stat_round_event, NULL));

 return 0;
}

DEFINE_SUITE("Synthesize stat config", synthesize_stat_config);
DEFINE_SUITE("Synthesize stat", synthesize_stat);
DEFINE_SUITE("Synthesize stat round", synthesize_stat_round);