#include <kernaux/multiboot2.h>

#include <assert.h>

#include "multiboot2_example1.h"
#include "multiboot2_example2.h"

static const struct {
    struct KernAux_Multiboot2 multiboot2;
    struct KernAux_Multiboot2_Tag_None tag_none;
} multiboot2_without_boot_cmd_line = {
    .multiboot2 = {
        .total_size = sizeof(multiboot2_without_boot_cmd_line),
        .reserved1 = 0,
    },
    .tag_none = {
        .base = {
            .type = KERNAUX_MULTIBOOT2_TAGTYPE_NONE,
            .size = sizeof(multiboot2_without_boot_cmd_line.tag_none),
        },
    },
};

static const struct {
    struct KernAux_Multiboot2 multiboot2;

    struct {
        struct KernAux_Multiboot2_Tag_BootCmdLine tag;
        char cmdline[14];
    } tag_boot_cmd_line;

    unsigned char _align1[2];

    struct KernAux_Multiboot2_Tag_None tag_none;
} multiboot2_with_some_boot_cmd_line = {
    .multiboot2 = {
        .total_size = sizeof(multiboot2_with_some_boot_cmd_line),
        .reserved1 = 0,
    },
    .tag_boot_cmd_line = {
        .tag = {
            .base = {
                .type = KERNAUX_MULTIBOOT2_TAGTYPE_BOOT_CMD_LINE,
                .size = sizeof(
                    multiboot2_with_some_boot_cmd_line.tag_boot_cmd_line
                ),
            },
        },
        .cmdline = "Hello, World!\0",
    },
    .tag_none = {
        .base = {
            .type = KERNAUX_MULTIBOOT2_TAGTYPE_NONE,
            .size = sizeof(multiboot2_with_some_boot_cmd_line.tag_none),
        },
    },
};

static const struct {
    struct KernAux_Multiboot2 multiboot2;

    struct {
        struct KernAux_Multiboot2_Tag_BootCmdLine tag;
        char cmdline[14];
    } tag_boot_cmd_line1;

    unsigned char _align1[2];

    struct {
        struct KernAux_Multiboot2_Tag_BootCmdLine tag;
        char cmdline[13];
    } tag_boot_cmd_line2;

    unsigned char _align2[3];

    struct KernAux_Multiboot2_Tag_None tag_none;
} multiboot2_with_two_boot_cmd_lines = {
    .multiboot2 = {
        .total_size = sizeof(multiboot2_with_two_boot_cmd_lines),
        .reserved1 = 0,
    },
    .tag_boot_cmd_line1 = {
        .tag = {
            .base = {
                .type = KERNAUX_MULTIBOOT2_TAGTYPE_BOOT_CMD_LINE,
                .size = sizeof(
                    multiboot2_with_two_boot_cmd_lines.tag_boot_cmd_line1
                ),
            },
        },
        .cmdline = "Hello, World!\0",
    },
    .tag_boot_cmd_line2 = {
        .tag = {
            .base = {
                .type = KERNAUX_MULTIBOOT2_TAGTYPE_BOOT_CMD_LINE,
                .size = sizeof(
                    multiboot2_with_two_boot_cmd_lines.tag_boot_cmd_line2
                ),
            },
        },
        .cmdline = "Hello again!\0",
    },
    .tag_none = {
        .base = {
            .type = KERNAUX_MULTIBOOT2_TAGTYPE_NONE,
            .size = sizeof(multiboot2_with_two_boot_cmd_lines.tag_none),
        },
    },
};

int main()
{
    assert(KernAux_Multiboot2_is_valid(
        (struct KernAux_Multiboot2*)multiboot2_example1
    ));

    assert(KernAux_Multiboot2_is_valid(
        &multiboot2_example2.multiboot2
    ));

    assert(KernAux_Multiboot2_is_valid(
        &multiboot2_without_boot_cmd_line.multiboot2
    ));

    assert(KernAux_Multiboot2_is_valid(
        &multiboot2_with_some_boot_cmd_line.multiboot2
    ));

    assert(KernAux_Multiboot2_is_valid(
        &multiboot2_with_two_boot_cmd_lines.multiboot2
    ));

    assert(
        KernAux_Multiboot2_boot_cmd_line(
            &multiboot2_without_boot_cmd_line.multiboot2
        ) == 0
    );

    // KernAux_Multiboot2_first_tag_with_type

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_32BIT_SYSTEM_TABLE_PTR
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_64BIT_SYSTEM_TABLE_PTR
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_SMBIOS_TABLES
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_ACPI_NEW_RSDP
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_NETWORKING_INFO
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_MEMORY_MAP
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_BOOT_SERVICES_NOT_TERMINATED
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_32BIT_IMAGE_HANDLE_PTR
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            (struct KernAux_Multiboot2*)multiboot2_example1,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_64BIT_IMAGE_HANDLE_PTR
        ) == 0
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_NONE
        ) == &multiboot2_example2.tag_none.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_BOOT_CMD_LINE
        ) == &multiboot2_example2.tag_boot_cmd_line.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_BOOT_LOADER_NAME
        ) == &multiboot2_example2.tag_boot_loader_name.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_MODULE
        ) == &multiboot2_example2.tag_module1.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_BASIC_MEMORY_INFO
        ) == &multiboot2_example2.tag_basic_memory_info.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_FRAMEBUFFER_INFO
        ) == &multiboot2_example2.tag_framebuffer_info.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_ELF_SYMBOLS
        ) == &multiboot2_example2.tag_elf_symbols.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_APM_TABLE
        ) == &multiboot2_example2.tag_apm_table.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_32BIT_SYSTEM_TABLE_PTR
        ) == &multiboot2_example2.tag_efi_32bit_system_table_ptr.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_64BIT_SYSTEM_TABLE_PTR
        ) == &multiboot2_example2.tag_efi_64bit_system_table_ptr.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_SMBIOS_TABLES
        ) == &multiboot2_example2.tag_smbios_tables.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_ACPI_OLD_RSDP
        ) == &multiboot2_example2.tag_acpi_old_rsdp.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_ACPI_NEW_RSDP
        ) == &multiboot2_example2.tag_acpi_new_rsdp.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_NETWORKING_INFO
        ) == &multiboot2_example2.tag_networking_info.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_MEMORY_MAP
        ) == &multiboot2_example2.tag_efi_memory_map.tag.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_BOOT_SERVICES_NOT_TERMINATED
        ) == &multiboot2_example2.tag_efi_boot_services_not_terminated.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_32BIT_IMAGE_HANDLE_PTR
        ) == &multiboot2_example2.tag_efi_32bit_image_handle_ptr.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_EFI_64BIT_IMAGE_HANDLE_PTR
        ) == &multiboot2_example2.tag_efi_64bit_image_handle_ptr.base
    );

    assert(
        KernAux_Multiboot2_first_tag_with_type(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_IMAGE_LOAD_BASE_PHYS_ADDR
        ) == &multiboot2_example2.tag_image_load_base_phys_addr.base
    );

    // KernAux_Multiboot2_tag_with_type_after

    assert(
        KernAux_Multiboot2_tag_with_type_after(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_MODULE,
            KernAux_Multiboot2_first_tag_with_type(
                &multiboot2_example2.multiboot2,
                KERNAUX_MULTIBOOT2_TAGTYPE_MODULE
            ) - 1
        ) == (struct KernAux_Multiboot2_TagBase*)
            &multiboot2_example2.tag_module1
    );

    assert(
        KernAux_Multiboot2_tag_with_type_after(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_MODULE,
            KernAux_Multiboot2_first_tag_with_type(
                &multiboot2_example2.multiboot2,
                KERNAUX_MULTIBOOT2_TAGTYPE_MODULE
            )
        ) == (struct KernAux_Multiboot2_TagBase*)
            &multiboot2_example2.tag_module2
    );

    assert(
        KernAux_Multiboot2_tag_with_type_after(
            &multiboot2_example2.multiboot2,
            KERNAUX_MULTIBOOT2_TAGTYPE_MODULE,
            KernAux_Multiboot2_tag_with_type_after(
                &multiboot2_example2.multiboot2,
                KERNAUX_MULTIBOOT2_TAGTYPE_MODULE,
                KernAux_Multiboot2_first_tag_with_type(
                    &multiboot2_example2.multiboot2,
                    KERNAUX_MULTIBOOT2_TAGTYPE_MODULE
                )
            )
        ) == 0
    );

    // KernAux_Multiboot2_boot_cmd_line

    assert(
        KernAux_Multiboot2_boot_cmd_line(
            &multiboot2_with_some_boot_cmd_line.multiboot2
        ) == multiboot2_with_some_boot_cmd_line.tag_boot_cmd_line.cmdline
    );

    assert(
        KernAux_Multiboot2_boot_cmd_line(
            &multiboot2_with_two_boot_cmd_lines.multiboot2
        ) == multiboot2_with_two_boot_cmd_lines.tag_boot_cmd_line1.cmdline
    );

    return 0;
}