linux-stable/tools/testing/selftests/bpf/prog_tests/test_struct_ops_maybe_null.c

47 lines
1.2 KiB
C

// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */
#include <test_progs.h>
#include "struct_ops_maybe_null.skel.h"
#include "struct_ops_maybe_null_fail.skel.h"
/* Test that the verifier accepts a program that access a nullable pointer
* with a proper check.
*/
static void maybe_null(void)
{
struct struct_ops_maybe_null *skel;
skel = struct_ops_maybe_null__open_and_load();
if (!ASSERT_OK_PTR(skel, "struct_ops_module_open_and_load"))
return;
struct_ops_maybe_null__destroy(skel);
}
/* Test that the verifier rejects a program that access a nullable pointer
* without a check beforehand.
*/
static void maybe_null_fail(void)
{
struct struct_ops_maybe_null_fail *skel;
skel = struct_ops_maybe_null_fail__open_and_load();
if (ASSERT_ERR_PTR(skel, "struct_ops_module_fail__open_and_load"))
return;
struct_ops_maybe_null_fail__destroy(skel);
}
void test_struct_ops_maybe_null(void)
{
/* The verifier verifies the programs at load time, so testing both
* programs in the same compile-unit is complicated. We run them in
* separate objects to simplify the testing.
*/
if (test__start_subtest("maybe_null"))
maybe_null();
if (test__start_subtest("maybe_null_fail"))
maybe_null_fail();
}