ARK-OS/arkos/kernel/tools/lib/perf/tests/test-cpumap.c

// SPDX-License-Identifier: GPL-2.0
#include <stdarg.h>
#include <stdio.h>
#include <perf/cpumap.h>
#include <internal/tests.h>
#include "tests.h"

static int libperf_print(enum libperf_print_level level,
			 const char *fmt, va_list ap)
{
	return vfprintf(stderr, fmt, ap);
}

int test_cpumap(int argc, char **argv)
{
	struct perf_cpu_map *cpus;
	struct perf_cpu cpu;
	int idx;

	__T_START;

	libperf_init(libperf_print);

	cpus = perf_cpu_map__new_any_cpu();
	if (!cpus)
		return -1;

	perf_cpu_map__get(cpus);
	perf_cpu_map__put(cpus);
	perf_cpu_map__put(cpus);

	cpus = perf_cpu_map__new_online_cpus();
	if (!cpus)
		return -1;

	perf_cpu_map__for_each_cpu(cpu, idx, cpus)
		__T("wrong cpu number", cpu.cpu != -1);

	perf_cpu_map__put(cpus);

	__T_END;
	return tests_failed == 0 ? 0 : -1;
}