// Code for handling calls to "post" that are resume related.
//
// Copyright (C) 2008  Kevin O'Connor <kevin@koconnor.net>
//
// This file may be distributed under the terms of the GNU GPLv3 license.

#include "util.h" // dprintf
#include "ioport.h" // outb
#include "pic.h" // eoi_pic2
#include "biosvar.h" // struct bios_data_area_s
#include "bregs.h" // struct bregs
#include "acpi.h" // find_resume_vector

// Reset DMA controller
void
init_dma()
{
    // first reset the DMA controllers
    outb(0, PORT_DMA1_MASTER_CLEAR);
    outb(0, PORT_DMA2_MASTER_CLEAR);

    // then initialize the DMA controllers
    outb(0xc0, PORT_DMA2_MODE_REG);
    outb(0x00, PORT_DMA2_MASK_REG);
}

// Handler for post calls that look like a resume.
void VISIBLE16
handle_resume(u8 status)
{
    init_dma();

    debug_serial_setup();
    dprintf(1, "In resume (status=%d)\n", status);

    struct bios_data_area_s *bda = MAKE_FARPTR(SEG_BDA, 0);
    switch (status) {
    case 0xfe:
        // S3 resume request.  Jump to 32bit mode to handle the resume.
        asm volatile(
            "movw %%ax, %%ss\n"
            "movl %0, %%esp\n"
            "pushl $_code32_s3_resume\n"
            "jmp transition32\n"
            : : "i"(BUILD_S3RESUME_STACK_ADDR), "a"(0)
            );
        break;

    case 0x00:
    case 0x09:
    case 0x0d ... 0xfd:
    case 0xff:
        // Normal post - now that status has been cleared a reset will
        // run regular boot code..
        reset_vector();
        break;

    case 0x05:
        // flush keyboard (issue EOI) and jump via 40h:0067h
        eoi_pic2();
        // NO BREAK
    case 0x0a:
        // resume execution by jump via 40h:0067h
        asm volatile(
            "movw %%ax, %%ds\n"
            "ljmpw *%0\n"
            : : "m"(bda->jump_ip), "a"(SEG_BDA)
            );
        break;

    case 0x0b:
        // resume execution via IRET via 40h:0067h
        asm volatile(
            "movw %%ax, %%ds\n"
            "movw %0, %%sp\n"
            "movw %1, %%ss\n"
            "iretw\n"
            : : "m"(bda->jump_ip), "m"(bda->jump_cs), "a"(SEG_BDA)
            );
        break;

    case 0x0c:
        // resume execution via RETF via 40h:0067h
        asm volatile(
            "movw %%ax, %%ds\n"
            "movw %0, %%sp\n"
            "movw %1, %%ss\n"
            "lretw\n"
            : : "m"(bda->jump_ip), "m"(bda->jump_cs), "a"(SEG_BDA)
            );
        break;
    }

    BX_PANIC("Unimplemented shutdown status: %02x\n", status);
}

void VISIBLE32
s3_resume()
{
    dprintf(1, "In 32bit resume\n");

    smm_init();

    make_bios_readonly();

    u32 s3_resume_vector = find_resume_vector();

    // Invoke the resume vector.
    struct bregs br;
    memset(&br, 0, sizeof(br));
    if (s3_resume_vector) {
        dprintf(1, "Jump to resume vector (%x)\n", s3_resume_vector);
        br.ip = FARPTR_TO_OFFSET(s3_resume_vector);
        br.cs = FARPTR_TO_SEG(s3_resume_vector);
    } else {
        dprintf(1, "No resume vector set!\n");
        // Jump to the post vector to restart with a normal boot.
        br.ip = (u32)reset_vector - BUILD_BIOS_ADDR;
        br.cs = SEG_BIOS;
    }
    call16big(&br);
}

// Ughh - some older gcc compilers have a bug which causes VISIBLE32
// functions to not be exported as global variables.
asm(".global s3_resume");