// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) 2025 Red Hat, Inc.*/
#include <test_progs.h>
#include "string_kfuncs_success.skel.h"
#include "string_kfuncs_failure1.skel.h"
#include "string_kfuncs_failure2.skel.h"
#include <sys/mman.h>

static const char * const test_cases[] = {
        "strcmp",
        "strcasecmp",
        "strncasecmp",
        "strchr",
        "strchrnul",
        "strnchr",
        "strrchr",
        "strlen",
        "strnlen",
        "strspn_str",
        "strspn_accept",
        "strcspn_str",
        "strcspn_reject",
        "strstr",
        "strcasestr",
        "strnstr",
        "strncasestr",
};

void run_too_long_tests(void)
{
        struct string_kfuncs_failure2 *skel;
        struct bpf_program *prog;
        char test_name[256];
        int err, i;

        skel = string_kfuncs_failure2__open_and_load();
        if (!ASSERT_OK_PTR(skel, "string_kfuncs_failure2__open_and_load"))
                return;

        memset(skel->bss->long_str, 'a', sizeof(skel->bss->long_str));

        for (i = 0; i < ARRAY_SIZE(test_cases); i++) {
                sprintf(test_name, "test_%s_too_long", test_cases[i]);
                if (!test__start_subtest(test_name))
                        continue;

                prog = bpf_object__find_program_by_name(skel->obj, test_name);
                if (!ASSERT_OK_PTR(prog, "bpf_object__find_program_by_name"))
                        goto cleanup;

                LIBBPF_OPTS(bpf_test_run_opts, topts);
                err = bpf_prog_test_run_opts(bpf_program__fd(prog), &topts);
                if (!ASSERT_OK(err, "bpf_prog_test_run"))
                        goto cleanup;

                ASSERT_EQ(topts.retval, -E2BIG, "reading too long string fails with -E2BIG");
        }

cleanup:
        string_kfuncs_failure2__destroy(skel);
}

void test_string_kfuncs(void)
{
        RUN_TESTS(string_kfuncs_success);
        RUN_TESTS(string_kfuncs_failure1);

        run_too_long_tests();
}