Commit 7e7ef1bf authored by Masami Hiramatsu (Google)'s avatar Masami Hiramatsu (Google) Committed by Steven Rostedt (Google)

lib/test_fprobe: Add a test case for nr_maxactive

Add a test case for nr_maxactive. If the number of active
functions is more than nr_maxactive, it must be skipped.

Link: https://lkml.kernel.org/r/167526698856.433354.4430007340787176666.stgit@mhiramat.roam.corp.google.com

Cc: Florent Revest <revest@chromium.org>
Cc: Mark Rutland <mark.rutland@arm.com>
Cc: Will Deacon <will@kernel.org>
Signed-off-by: default avatarMasami Hiramatsu (Google) <mhiramat@kernel.org>
Signed-off-by: default avatarSteven Rostedt (Google) <rostedt@goodmis.org>
parent 59a7a298
...@@ -17,8 +17,10 @@ static u32 rand1, entry_val, exit_val; ...@@ -17,8 +17,10 @@ static u32 rand1, entry_val, exit_val;
/* Use indirect calls to avoid inlining the target functions */ /* Use indirect calls to avoid inlining the target functions */
static u32 (*target)(u32 value); static u32 (*target)(u32 value);
static u32 (*target2)(u32 value); static u32 (*target2)(u32 value);
static u32 (*target_nest)(u32 value, u32 (*nest)(u32));
static unsigned long target_ip; static unsigned long target_ip;
static unsigned long target2_ip; static unsigned long target2_ip;
static unsigned long target_nest_ip;
static noinline u32 fprobe_selftest_target(u32 value) static noinline u32 fprobe_selftest_target(u32 value)
{ {
...@@ -30,6 +32,11 @@ static noinline u32 fprobe_selftest_target2(u32 value) ...@@ -30,6 +32,11 @@ static noinline u32 fprobe_selftest_target2(u32 value)
return (value / div_factor) + 1; return (value / div_factor) + 1;
} }
static noinline u32 fprobe_selftest_nest_target(u32 value, u32 (*nest)(u32))
{
return nest(value + 2);
}
static notrace void fp_entry_handler(struct fprobe *fp, unsigned long ip, static notrace void fp_entry_handler(struct fprobe *fp, unsigned long ip,
struct pt_regs *regs, void *data) struct pt_regs *regs, void *data)
{ {
...@@ -67,6 +74,19 @@ static notrace void fp_exit_handler(struct fprobe *fp, unsigned long ip, ...@@ -67,6 +74,19 @@ static notrace void fp_exit_handler(struct fprobe *fp, unsigned long ip,
KUNIT_EXPECT_NULL(current_test, data); KUNIT_EXPECT_NULL(current_test, data);
} }
static notrace void nest_entry_handler(struct fprobe *fp, unsigned long ip,
struct pt_regs *regs, void *data)
{
KUNIT_EXPECT_FALSE(current_test, preemptible());
}
static notrace void nest_exit_handler(struct fprobe *fp, unsigned long ip,
struct pt_regs *regs, void *data)
{
KUNIT_EXPECT_FALSE(current_test, preemptible());
KUNIT_EXPECT_EQ(current_test, ip, target_nest_ip);
}
/* Test entry only (no rethook) */ /* Test entry only (no rethook) */
static void test_fprobe_entry(struct kunit *test) static void test_fprobe_entry(struct kunit *test)
{ {
...@@ -163,6 +183,25 @@ static void test_fprobe_data(struct kunit *test) ...@@ -163,6 +183,25 @@ static void test_fprobe_data(struct kunit *test)
KUNIT_EXPECT_EQ(test, 0, unregister_fprobe(&fp)); KUNIT_EXPECT_EQ(test, 0, unregister_fprobe(&fp));
} }
/* Test nr_maxactive */
static void test_fprobe_nest(struct kunit *test)
{
static const char *syms[] = {"fprobe_selftest_target", "fprobe_selftest_nest_target"};
struct fprobe fp = {
.entry_handler = nest_entry_handler,
.exit_handler = nest_exit_handler,
.nr_maxactive = 1,
};
current_test = test;
KUNIT_EXPECT_EQ(test, 0, register_fprobe_syms(&fp, syms, 2));
target_nest(rand1, target);
KUNIT_EXPECT_EQ(test, 1, fp.nmissed);
KUNIT_EXPECT_EQ(test, 0, unregister_fprobe(&fp));
}
static unsigned long get_ftrace_location(void *func) static unsigned long get_ftrace_location(void *func)
{ {
unsigned long size, addr = (unsigned long)func; unsigned long size, addr = (unsigned long)func;
...@@ -178,8 +217,10 @@ static int fprobe_test_init(struct kunit *test) ...@@ -178,8 +217,10 @@ static int fprobe_test_init(struct kunit *test)
rand1 = get_random_u32_above(div_factor); rand1 = get_random_u32_above(div_factor);
target = fprobe_selftest_target; target = fprobe_selftest_target;
target2 = fprobe_selftest_target2; target2 = fprobe_selftest_target2;
target_nest = fprobe_selftest_nest_target;
target_ip = get_ftrace_location(target); target_ip = get_ftrace_location(target);
target2_ip = get_ftrace_location(target2); target2_ip = get_ftrace_location(target2);
target_nest_ip = get_ftrace_location(target_nest);
return 0; return 0;
} }
...@@ -189,6 +230,7 @@ static struct kunit_case fprobe_testcases[] = { ...@@ -189,6 +230,7 @@ static struct kunit_case fprobe_testcases[] = {
KUNIT_CASE(test_fprobe), KUNIT_CASE(test_fprobe),
KUNIT_CASE(test_fprobe_syms), KUNIT_CASE(test_fprobe_syms),
KUNIT_CASE(test_fprobe_data), KUNIT_CASE(test_fprobe_data),
KUNIT_CASE(test_fprobe_nest),
{} {}
}; };
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment