linux/tools/testing/selftests/bpf/prog_tests/test_struct_ops_maybe_null.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();
}