#include <linux/tables.h>
#include <asm/x86_init_fn.h>
#include <linux/ranges.h>
#include <linux/kprobes.h>

DEFINE_SECTION_RANGE(kprobes, SECTION_TEXT);

void __kprobes test_kprobe_0001(void)
{
	pr_info("test_kprobe\n");
}

void test_kprobe_0002(void)
{
	pr_info("test_kprobe\n");
}

void test_kprobe_addr(const char *test, unsigned long addr, bool should_match)
{
	if (SECTION_ADDR_IN_RANGE(kprobes, addr))
		if (should_match)
			pr_info("== OK: %s within range!\n", test);
		else
			pr_info("== FAIL: %s should not be in range...\n",
				test);
	else
		if (should_match)
			pr_info("== FAIL: %s should be in range...\n", test);
		else
			pr_info("== OK: %s not in range as expected!\n", test);
}

void early_init_kprobes(void)
{
	unsigned long addr;

	addr = (unsigned long) &test_kprobe_0001;

	test_kprobe_addr("test_kprobe_0001", addr, true);

	addr = (unsigned long) &test_kprobe_0002;

	test_kprobe_addr("test_kprobe_0002", addr, false);
}

X86_INIT_EARLY_ALL(kprobes, NULL, NULL, early_init_kprobes, NULL, NULL);