/**
 * \file
 * \brief PCI
 */

/*
 * Copyright (c) 2007, 2008, 2009, 2011, 2016 ETH Zurich.
 * All rights reserved.
 *
 * This file is distributed under the terms in the attached LICENSE file.
 * If you do not find this file, copies can be found by writing to:
 * ETH Zurich D-INFK, Universitaetsstrasse 6, CH-8092 Zurich. Attn: Systems Group.
 */

#include <barrelfish/barrelfish.h>
#include <barrelfish/capabilities.h>
#include <barrelfish/nameservice_client.h>
#include <stdio.h>
#include <stdlib.h>
#include <mm/mm.h>
#include <if/monitor_blocking_defs.h>

#include <octopus/octopus.h>
#include <skb/skb.h>
#include <hw_records.h>

#include "acpi_debug.h"
#include "acpi_shared.h"
#include "acpi_allocators.h"
#include "acpidump.h"



uintptr_t my_hw_id;


static errval_t setup_skb_info(void)
{
    skb_execute("[pci_queries].");
    errval_t err = skb_read_error_code();
    if (err_is_fail(err)) {
        ACPI_DEBUG("\npcimain.c: Could not load pci_queries.pl.\n"
               "SKB returned: %s\nSKB error: %s\n",
                skb_get_output(), skb_get_error_output());
        return err;
    }

    skb_add_fact("mem_region_type(%d,ram).", RegionType_Empty);
    skb_add_fact("mem_region_type(%d,roottask).", RegionType_RootTask);
    skb_add_fact("mem_region_type(%d,phyaddr).", RegionType_PhyAddr);
    skb_add_fact("mem_region_type(%d,multiboot_module).", RegionType_Module);
    skb_add_fact("mem_region_type(%d,platform_data).", RegionType_PlatformData);


    return acpi_arch_skb_set_info();
}

extern int ApDumpAllTables (void);
static void wait_for_iommu(void)
{   
    errval_t err;
    char**names = NULL;
    size_t len = 0;
    
    err = oct_get_names(&names, &len, HW_PCI_IOMMU_RECORD_REGEX);
    if (err_is_fail(err)) {
        if (err == OCT_ERR_NO_RECORD) {
            debug_printf("No Iommus available, continue withouth waiting\n");
            return;
        }
        goto out;
    }

    if (len > 0) {

        char* key;
        char* record;
        uint64_t type, flags, segment, address, idx;
        err = oct_get(&record, names[0]);
        if (err_is_fail(err)) {
            goto out;
        }

        err = oct_read(record, "%s { " HW_PCI_IOMMU_RECORD_FIELDS_READ " }",
                       &key, &idx, &type, &flags, &segment, &address);
        if (err_is_fail(err)) {
            goto out;
        }
        
        if (type == HW_PCI_IOMMU_DMAR_FAIL) {
            debug_printf("Reading DMAR failed, not waiting for iommus \n");
            goto out;
        }

        debug_printf("Waiting for all iommus to start up (num_iommu=%zu) \n", len);
        err = oct_barrier_enter("barrier.iommu", &record ,2);
        if (err_is_fail(err)) {
            goto out;    
        }
        if (record) {
            free(record);
        }
    }

out:
    oct_free_names(names, len);
}

int main(int argc, char *argv[])
{
    errval_t err;

    // Parse CMD Arguments
    bool got_apic_id = false;
    bool do_video_init = false;
    bool ignore_irq_override = false;
    bool dump_acpi_tables = false;

    for (int i = 1; i < argc; i++) {
        if(sscanf(argv[i], "apicid=%" PRIuPTR, &my_hw_id) == 1) {
            got_apic_id = true;
        } else if (strcmp(argv[i], "video_init") == 0) {
            do_video_init = true;
        } else if (strncmp(argv[i], "dump_tables", strlen("dump_tables")) == 0) {
            dump_acpi_tables = true;
        } else if (strncmp(argv[i], "ignore_irq_override", strlen("ignore_irq_override")) == 0) {
            ignore_irq_override = true;
        }

    }

    if(got_apic_id == false) {
        fprintf(stderr, "Usage: %s APIC_ID\n", argv[0]);
        fprintf(stderr, "Wrong monitor version?\n");
        return EXIT_FAILURE;
    }

    err = oct_init();
    if (err_is_fail(err)) {
        USER_PANIC_ERR(err, "Initialize dist");
    }

    //connect to the SKB
    ACPI_DEBUG("acpi: connecting to the SKB...\n");
    err = skb_client_connect();
    if (err_is_fail(err)) {
        USER_PANIC_ERR(err, "Connecting to SKB failed.");
    }

    ACPI_DEBUG("acpi: connecting to the SKB...\n");
    err = setup_skb_info();
    if (err_is_fail(err)) {
        USER_PANIC_ERR(err, "Populating SKB failed.");
    }

    err = acpi_allocators_init();
    if (err_is_fail(err)) {
        USER_PANIC_ERR(err, "Init memory allocator");
    }

    err = acpi_arch_copy_bios_mem();
    if (err_is_fail(err)) {
        USER_PANIC_ERR(err, "Copy BIOS Memory");
    }


    if (dump_acpi_tables) {
        debug_printf("DUMPING ACPI TABLES.\n");
        debug_printf("======================================\n");
        AcpiOsInitialize ();
        Gbl_OutputFile = stdout;
        ApDumpAllTables ();
        debug_printf("======================================\n");
    }



    err = acpi_arch_load_irq_routing_new();
    if (err_is_fail(err)) {
        USER_PANIC_ERR(err, "load irq routing new.");
    }

    int r = init_acpi();
    assert(r == 0);

    buttons_init();

    if (do_video_init) {
        acpi_arch_video_init();
    }

    start_service();
 
    wait_for_iommu(); 
   
    // synchronize ACPI/KALUGA/PCI
    char* record = NULL;
    debug_printf("barrier.pci.bridges\n");
    err = oct_barrier_enter("barrier.pci.bridges", &record, 3);
    if (err_is_fail(err)) {
        USER_PANIC_ERR(err, "Could not wait for PCI Barrier 'barrier.pci.bridges'\n");
    }

    err = acpi_interrupts_arch_setup();
    if (err_is_fail(err)) {
        USER_PANIC_ERR(err, "setup skb irq controllers");
    }

    if(ignore_irq_override){
        err = skb_execute("retractall(interrupt_override(_,_,_,_)).");
        if(err_is_fail(err)){
            DEBUG_SKB_ERR(err, "couldnt remove interrupt overrides");
        }
    }

    ACPI_DEBUG("####################### Entering message handler loop \n");
    messages_handler_loop();
}