/*
 * Boot order test cases.
 *
 * Copyright (c) 2013 Red Hat Inc.
 *
 * Authors:
 *  Markus Armbruster <armbru@redhat.com>,
 *
 * This work is licensed under the terms of the GNU GPL, version 2 or later.
 * See the COPYING file in the top-level directory.
 */

#include <string.h>
#include <glib.h>
#include "libqos/fw_cfg.h"
#include "libqtest.h"

static void test_pc_cmos_byte(int reg, int expected)
{
    int actual;

    outb(0x70, reg);
    actual = inb(0x71);
    g_assert_cmphex(actual, ==, expected);
}

static void test_pc_cmos(uint8_t boot1, uint8_t boot2)
{
    test_pc_cmos_byte(0x38, boot1);
    test_pc_cmos_byte(0x3d, boot2);
}

static void test_pc_with_args(const char *test_args,
                              uint8_t boot1, uint8_t boot2,
                              uint8_t reboot1, uint8_t reboot2)
{
    char *args = g_strdup_printf("-nodefaults -display none %s", test_args);

    qtest_start(args);
    test_pc_cmos(boot1, boot2);
    qmp("{ 'execute': 'system_reset' }");
    /*
     * system_reset only requests reset.  We get a RESET event after
     * the actual reset completes.  Need to wait for that.
     */
    qmp("");                    /* HACK: wait for event */
    test_pc_cmos(reboot1, reboot2);
    qtest_quit(global_qtest);
    g_free(args);
}

static void test_pc_boot_order(void)
{
    test_pc_with_args("", 0x30, 0x12, 0x30, 0x12);
    test_pc_with_args("-no-fd-bootchk", 0x31, 0x12, 0x31, 0x12);
    test_pc_with_args("-boot c", 0, 0x02, 0, 0x02);
    test_pc_with_args("-boot nda", 0x10, 0x34, 0x10, 0x34);
    test_pc_with_args("-boot order=", 0, 0, 0, 0);
    test_pc_with_args("-boot order= -boot order=c", 0, 0x02, 0, 0x02);
    test_pc_with_args("-boot once=a", 0, 0x01, 0x30, 0x12);
    test_pc_with_args("-boot once=a -no-fd-bootchk", 0x01, 0x01, 0x31, 0x12);
    test_pc_with_args("-boot once=a,order=c", 0, 0x01, 0, 0x02);
    test_pc_with_args("-boot once=d -boot order=nda", 0, 0x03, 0x10, 0x34);
    test_pc_with_args("-boot once=a -boot once=b -boot once=c",
                      0, 0x02, 0x30, 0x12);
}

#define G3BEIGE_CFG_ADDR 0xf0000510
#define MAC99_CFG_ADDR   0xf0000510

#define NO_QEMU_PROTOS
#include "hw/nvram/fw_cfg.h"
#undef NO_QEMU_PROTOS

static void test_powermac_with_args(bool newworld, const char *extra_args,
                                    uint16_t expected_boot,
                                    uint16_t expected_reboot)
{
    char *args = g_strdup_printf("-nodefaults -display none -machine %s %s",
                                 newworld ? "mac99" : "g3beige", extra_args);
    QFWCFG *fw_cfg = mm_fw_cfg_init(newworld ? MAC99_CFG_ADDR
                                    : G3BEIGE_CFG_ADDR);
    uint16_t actual;

    qtest_start(args);
    actual = qfw_cfg_get_u16(fw_cfg, FW_CFG_BOOT_DEVICE);
    g_assert_cmphex(actual, ==, expected_boot);
    qmp("{ 'execute': 'system_reset' }");
    actual = qfw_cfg_get_u16(fw_cfg, FW_CFG_BOOT_DEVICE);
    g_assert_cmphex(actual, ==, expected_reboot);
    qtest_quit(global_qtest);
    g_free(args);
}

static void test_powermac_boot_order(void)
{
    int i;

    for (i = 0; i < 2; i++) {
        bool newworld = (i == 1);

        test_powermac_with_args(newworld, "", 'c', 'c');
        test_powermac_with_args(newworld, "-boot c", 'c', 'c');
        test_powermac_with_args(newworld, "-boot d", 'd', 'd');
        test_powermac_with_args(newworld, "-boot once=d,order=c", 'd', 'c');
    }
}

int main(int argc, char *argv[])
{
    const char *arch = qtest_get_arch();

    g_test_init(&argc, &argv, NULL);

    if (strcmp(arch, "i386") == 0 || strcmp(arch, "x86_64") == 0) {
        qtest_add_func("boot-order/pc", test_pc_boot_order);
    } else if (strcmp(arch, "ppc") == 0 || strcmp(arch, "ppc64") == 0) {
        qtest_add_func("boot-order/powermac", test_powermac_boot_order);
    }

    return g_test_run();
}