linux/tools/perf/tests/perf-hooks.c

// SPDX-License-Identifier: GPL-2.0
#include <signal.h>
#include <stdlib.h>

#include "tests.h"
#include "debug.h"
#include "perf-hooks.h"

static void sigsegv_handler(int sig __maybe_unused)
{
	pr_debug("SIGSEGV is observed as expected, try to recover.\n");
	perf_hooks__recover();
	signal(SIGSEGV, SIG_DFL);
	raise(SIGSEGV);
	exit(-1);
}


static void the_hook(void *_hook_flags)
{
	int *hook_flags = _hook_flags;

	*hook_flags = 1234;

	/* Generate a segfault, test perf_hooks__recover */
	raise(SIGSEGV);
}

static int test__perf_hooks(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
{
	int hook_flags = 0;

	signal(SIGSEGV, sigsegv_handler);
	perf_hooks__set_hook("test", the_hook, &hook_flags);
	perf_hooks__invoke_test();

	/* hook is triggered? */
	if (hook_flags != 1234) {
		pr_debug("Setting failed: %d (%p)\n", hook_flags, &hook_flags);
		return TEST_FAIL;
	}

	/* the buggy hook is removed? */
	if (perf_hooks__get_hook("test"))
		return TEST_FAIL;
	return TEST_OK;
}

DEFINE_SUITE("perf hooks", perf_hooks);