// 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();
}