// MPTable generation (on emulators)
//
// Copyright (C) 2008,2009  Kevin O'Connor <kevin@koconnor.net>
// Copyright (C) 2006 Fabrice Bellard
//
// This file may be distributed under the terms of the GNU LGPLv3 license.

#include "util.h" // dprintf
#include "config.h" // CONFIG_*
#include "mptable.h" // MPTABLE_SIGNATURE

#if CONFIG_KVM
int irq0override = 1;
#else
int irq0override = 0;
#endif

void
mptable_init(void)
{
    if (! CONFIG_MPTABLE)
        return;

    dprintf(3, "init MPTable\n");

    int smp_cpus = CountCPUs;
    if (smp_cpus <= 1)
        // Building an mptable on uniprocessor machines confuses some OSes.
        return;

    int length = (sizeof(struct mptable_config_s)
                  + sizeof(struct mpt_cpu) * smp_cpus
                  + sizeof(struct mpt_bus)
                  + sizeof(struct mpt_ioapic)
                  + sizeof(struct mpt_intsrc) * 16);
    struct mptable_config_s *config = malloc_fseg(length);
    struct mptable_floating_s *floating = malloc_fseg(sizeof(*floating));
    if (!config || !floating) {
        dprintf(1, "No room for MPTABLE!\n");
        return;
    }

    /* floating pointer structure */
    memset(floating, 0, sizeof(*floating));
    floating->signature = MPTABLE_SIGNATURE;
    floating->physaddr = (u32)config;
    floating->length = 1;
    floating->spec_rev = 4;
    floating->checksum -= checksum(floating, sizeof(*floating));

    // Config structure.
    memset(config, 0, sizeof(*config));
    config->signature = MPCONFIG_SIGNATURE;
    config->spec = 4;
    memcpy(config->oemid, CONFIG_CPUNAME8, sizeof(config->oemid));
    memcpy(config->productid, "0.1         ", sizeof(config->productid));
    config->entrycount = smp_cpus + 2 + 16;
    config->lapic = BUILD_APIC_ADDR;

    // CPU definitions.
    u32 cpuid_signature, ebx, ecx, cpuid_features;
    cpuid(1, &cpuid_signature, &ebx, &ecx, &cpuid_features);
    struct mpt_cpu *cpus = (void*)&config[1];
    int i;
    for (i = 0; i < smp_cpus; i++) {
        struct mpt_cpu *cpu = &cpus[i];
        memset(cpu, 0, sizeof(*cpu));
        cpu->type = MPT_TYPE_CPU;
        cpu->apicid = i;
        cpu->apicver = 0x11;
        /* cpu flags: enabled, bootstrap cpu */
        cpu->cpuflag = (i == 0 ? 3 : 1);
        if (cpuid_signature) {
            cpu->cpusignature = cpuid_signature;
            cpu->featureflag = cpuid_features;
        } else {
            cpu->cpusignature = 0x600;
            cpu->featureflag = 0x201;
        }
    }

    /* isa bus */
    struct mpt_bus *bus = (void*)&cpus[smp_cpus];
    memset(bus, 0, sizeof(*bus));
    bus->type = MPT_TYPE_BUS;
    memcpy(bus->bustype, "ISA   ", sizeof(bus->bustype));

    /* ioapic */
    u8 ioapic_id = smp_cpus;
    struct mpt_ioapic *ioapic = (void*)&bus[1];
    memset(ioapic, 0, sizeof(*ioapic));
    ioapic->type = MPT_TYPE_IOAPIC;
    ioapic->apicid = ioapic_id;
    ioapic->apicver = 0x11;
    ioapic->flags = 1; // enable
    ioapic->apicaddr = BUILD_IOAPIC_ADDR;

    /* irqs */
    struct mpt_intsrc *intsrc = (void*)&ioapic[1];
    for (i = 0; i < 16; i++) {
        memset(intsrc, 0, sizeof(*intsrc));
        intsrc->type = MPT_TYPE_INTSRC;
        intsrc->srcbusirq = i;
        intsrc->dstapic = ioapic_id;
        intsrc->dstirq = i;
        if (irq0override) {
            /* Destination 2 is covered by irq0->inti2 override (i ==
               0). Source IRQ 2 is unused */
            if (i == 0)
                intsrc->dstirq = 2;
            else if (i == 2)
                intsrc--;
        }
        intsrc++;
    }

    // Set checksum.
    config->length = (void*)intsrc - (void*)config;
    config->checksum -= checksum(config, config->length);

    dprintf(1, "MP table addr=%p MPC table addr=%p size=%d\n",
            floating, config, config->length);
}
