// Copyright (c) eBPF for Windows contributors
// SPDX-License-Identifier: MIT

#define CATCH_CONFIG_MAIN

#include "bpf/libbpf.h"
#include "catch_wrapper.hpp"
#include "ebpf_api.h"

#include <iostream>

#define SAMPLE_PATH ""
#define CILIUM_XDP_SECTIONS_SNAT 10
#define CILIUM_XDP_SECTIONS_DSR 12

void
verify_program(_In_z_ const char* file, uint32_t expected_section_count)
{
    struct bpf_object_open_opts opts = {0};
    bpf_program* program = nullptr;
    uint32_t section_count = 0;
    struct bpf_object* object = bpf_object__open_file(file, &opts);
    REQUIRE(object != nullptr);

    while (true) {
        program = bpf_object__next_program(object, program);
        if (program == nullptr) {
            break;
        }
        section_count++;
        const char* section_name = bpf_program__section_name(program);
        REQUIRE(section_name != nullptr);

        const char* program_name = bpf_program__name(program);
        REQUIRE(program_name != nullptr);

#ifndef SKIP_VERIFICATION
        uint32_t result;
        ebpf_api_verifier_stats_t stats;
        const char* log_buffer = nullptr;
        const char* report = nullptr;
        REQUIRE(
            (result = ebpf_api_elf_verify_program_from_file(
                 file,
                 section_name,
                 program_name,
                 &EBPF_PROGRAM_TYPE_XDP,
                 EBPF_VERIFICATION_VERBOSITY_NORMAL,
                 &report,
                 &log_buffer,
                 &stats),
             ebpf_free_string(log_buffer),
             log_buffer = nullptr,
             result == 0));
        REQUIRE(report != nullptr);
        ebpf_free_string(report);
#endif
    }

    REQUIRE(section_count == expected_section_count);
}

TEST_CASE("verify_snat_program", "[cilium][xdp]")
{
    verify_program(SAMPLE_PATH "bpf_xdp_snat.o", CILIUM_XDP_SECTIONS_SNAT);
}
TEST_CASE("verify_dsr_program", "[cilium][xdp]")
{
    verify_program(SAMPLE_PATH "bpf_xdp_dsr.o", CILIUM_XDP_SECTIONS_DSR);
}