Commit 84239a24 authored by Yonghong Song's avatar Yonghong Song Committed by Andrii Nakryiko

selftests/bpf: Replace CHECK with ASSERT_* in ns_current_pid_tgid test

Replace CHECK in selftest ns_current_pid_tgid with recommended ASSERT_* style.
I also shortened subtest name as the prefix of subtest name is covered
by the test name already.

This patch does fix a testing issue. Currently even if bss->user_{pid,tgid}
is not correct, the test still passed since the clone func returns 0.
I fixed it to return a non-zero value if bss->user_{pid,tgid} is incorrect.
Signed-off-by: default avatarYonghong Song <yonghong.song@linux.dev>
Signed-off-by: default avatarAndrii Nakryiko <andrii@kernel.org>
Acked-by: default avatarJiri Olsa <jolsa@kernel.org>
Link: https://lore.kernel.org/bpf/20240315184859.2975543-1-yonghong.song@linux.dev
parent eb166e52
...@@ -20,19 +20,19 @@ static int test_current_pid_tgid(void *args) ...@@ -20,19 +20,19 @@ static int test_current_pid_tgid(void *args)
{ {
struct test_ns_current_pid_tgid__bss *bss; struct test_ns_current_pid_tgid__bss *bss;
struct test_ns_current_pid_tgid *skel; struct test_ns_current_pid_tgid *skel;
int err = -1, duration = 0; int ret = -1, err;
pid_t tgid, pid; pid_t tgid, pid;
struct stat st; struct stat st;
skel = test_ns_current_pid_tgid__open_and_load(); skel = test_ns_current_pid_tgid__open_and_load();
if (CHECK(!skel, "skel_open_load", "failed to load skeleton\n")) if (!ASSERT_OK_PTR(skel, "test_ns_current_pid_tgid__open_and_load"))
goto cleanup; goto out;
pid = syscall(SYS_gettid); pid = syscall(SYS_gettid);
tgid = getpid(); tgid = getpid();
err = stat("/proc/self/ns/pid", &st); err = stat("/proc/self/ns/pid", &st);
if (CHECK(err, "stat", "failed /proc/self/ns/pid: %d\n", err)) if (!ASSERT_OK(err, "stat /proc/self/ns/pid"))
goto cleanup; goto cleanup;
bss = skel->bss; bss = skel->bss;
...@@ -42,24 +42,26 @@ static int test_current_pid_tgid(void *args) ...@@ -42,24 +42,26 @@ static int test_current_pid_tgid(void *args)
bss->user_tgid = 0; bss->user_tgid = 0;
err = test_ns_current_pid_tgid__attach(skel); err = test_ns_current_pid_tgid__attach(skel);
if (CHECK(err, "skel_attach", "skeleton attach failed: %d\n", err)) if (!ASSERT_OK(err, "test_ns_current_pid_tgid__attach"))
goto cleanup; goto cleanup;
/* trigger tracepoint */ /* trigger tracepoint */
usleep(1); usleep(1);
ASSERT_EQ(bss->user_pid, pid, "pid"); if (!ASSERT_EQ(bss->user_pid, pid, "pid"))
ASSERT_EQ(bss->user_tgid, tgid, "tgid"); goto cleanup;
err = 0; if (!ASSERT_EQ(bss->user_tgid, tgid, "tgid"))
goto cleanup;
ret = 0;
cleanup: cleanup:
test_ns_current_pid_tgid__destroy(skel); test_ns_current_pid_tgid__destroy(skel);
out:
return err; return ret;
} }
static void test_ns_current_pid_tgid_new_ns(void) static void test_ns_current_pid_tgid_new_ns(void)
{ {
int wstatus, duration = 0; int wstatus;
pid_t cpid; pid_t cpid;
/* Create a process in a new namespace, this process /* Create a process in a new namespace, this process
...@@ -68,21 +70,21 @@ static void test_ns_current_pid_tgid_new_ns(void) ...@@ -68,21 +70,21 @@ static void test_ns_current_pid_tgid_new_ns(void)
cpid = clone(test_current_pid_tgid, child_stack + STACK_SIZE, cpid = clone(test_current_pid_tgid, child_stack + STACK_SIZE,
CLONE_NEWPID | SIGCHLD, NULL); CLONE_NEWPID | SIGCHLD, NULL);
if (CHECK(cpid == -1, "clone", "%s\n", strerror(errno))) if (!ASSERT_NEQ(cpid, -1, "clone"))
return; return;
if (CHECK(waitpid(cpid, &wstatus, 0) == -1, "waitpid", "%s\n", strerror(errno))) if (!ASSERT_NEQ(waitpid(cpid, &wstatus, 0), -1, "waitpid"))
return; return;
if (CHECK(WEXITSTATUS(wstatus) != 0, "newns_pidtgid", "failed")) if (!ASSERT_OK(WEXITSTATUS(wstatus), "newns_pidtgid"))
return; return;
} }
/* TODO: use a different tracepoint */ /* TODO: use a different tracepoint */
void serial_test_ns_current_pid_tgid(void) void serial_test_ns_current_pid_tgid(void)
{ {
if (test__start_subtest("ns_current_pid_tgid_root_ns")) if (test__start_subtest("root_ns_tp"))
test_current_pid_tgid(NULL); test_current_pid_tgid(NULL);
if (test__start_subtest("ns_current_pid_tgid_new_ns")) if (test__start_subtest("new_ns_tp"))
test_ns_current_pid_tgid_new_ns(); test_ns_current_pid_tgid_new_ns();
} }
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