#include "qemu/osdep.h"
#include "qobject/qdict.h"
#include "qobject/qlist.h"
#include "qobject/qnum.h"
#include "qobject/qbool.h"
#include "libqtest-single.h"

static char *get_cpu0_qom_path(void)
{
    QDict *resp;
    QList *ret;
    QDict *cpu0;
    char *path;

    resp = qmp("{'execute': 'query-cpus-fast', 'arguments': {}}");
    g_assert(qdict_haskey(resp, "return"));
    ret = qdict_get_qlist(resp, "return");

    cpu0 = qobject_to(QDict, qlist_peek(ret));
    path = g_strdup(qdict_get_str(cpu0, "qom-path"));
    qobject_unref(resp);
    return path;
}

static QObject *qom_get(const char *path, const char *prop)
{
    QDict *resp = qmp("{ 'execute': 'qom-get',"
                      "  'arguments': { 'path': %s,"
                      "                 'property': %s } }",
                      path, prop);
    QObject *ret = qdict_get(resp, "return");
    qobject_ref(ret);
    qobject_unref(resp);
    return ret;
}

static bool qom_get_bool(const char *path, const char *prop)
{
    QBool *value = qobject_to(QBool, qom_get(path, prop));
    bool b = qbool_get_bool(value);

    qobject_unref(value);
    return b;
}

typedef struct CpuidTestArgs {
    /* test name */
    const char *name;
    /* CPU type */
    const char *cpu;
    /* CPU features (may be NULL) */
    const char *cpufeat;
    /* machine type (may be NULL to use default machine) */
    const char *machine;
    /* CPU property to read */
    const char *property;
    /* expected value of the property */
    int64_t expected_value;
} CpuidTestArgs;

static void test_cpuid_prop(const void *data)
{
    const CpuidTestArgs *args = data;
    char *cmdline;
    char *save;
    char *path;
    QNum *value;
    int64_t val;

    cmdline = g_strdup_printf("-cpu %s", args->cpu);

    if (args->cpufeat) {
        save = cmdline;
        cmdline = g_strdup_printf("%s,%s", cmdline, args->cpufeat);
        g_free(save);
    }
    if (args->machine) {
        save = cmdline;
        cmdline = g_strdup_printf("-machine %s %s", args->machine, cmdline);
        g_free(save);
    }

    qtest_start(cmdline);
    path = get_cpu0_qom_path();
    value = qobject_to(QNum, qom_get(path, args->property));
    g_assert(qnum_get_try_int(value, &val));
    g_assert_cmpint(val, ==, args->expected_value);
    qtest_end();

    qobject_unref(value);
    g_free(path);
    g_free(cmdline);
}

/* Parameters to a add_feature_test() test case */
typedef struct FeatureTestArgs {
    /* Test name */
    const char *name;
    /* CPU type */
    const char *cpu;
    /* CPU features, may be NULL */
    const char *cpufeat;
    /*
     * cpuid-input-eax and cpuid-input-ecx values to look for,
     * in "feature-words" and "filtered-features" properties.
     */
    uint32_t in_eax, in_ecx;
    /* The register name to look for, in the X86CPUFeatureWordInfo array */
    const char *reg;
    /* The bit to check in X86CPUFeatureWordInfo.features */
    int bitnr;
    /* The expected value for the bit in (X86CPUFeatureWordInfo.features) */
    bool expected_value;
} FeatureTestArgs;

/* Get the value for a feature word in a X86CPUFeatureWordInfo list */
static uint32_t get_feature_word(QList *features, uint32_t eax, uint32_t ecx,
                                 const char *reg)
{
    const QListEntry *e;

    for (e = qlist_first(features); e; e = qlist_next(e)) {
        QDict *w = qobject_to(QDict, qlist_entry_obj(e));
        const char *rreg = qdict_get_str(w, "cpuid-register");
        uint32_t reax = qdict_get_int(w, "cpuid-input-eax");
        bool has_ecx = qdict_haskey(w, "cpuid-input-ecx");
        uint32_t recx = 0;
        int64_t val;

        if (has_ecx) {
            recx = qdict_get_int(w, "cpuid-input-ecx");
        }
        if (eax == reax && (!has_ecx || ecx == recx) && !strcmp(rreg, reg)) {
            g_assert(qnum_get_try_int(qobject_to(QNum,
                                                 qdict_get(w, "features")),
                                      &val));
            return val;
        }
    }
    return 0;
}

static void test_feature_flag(const void *data)
{
    const FeatureTestArgs *args = data;
    char *path;
    char *cmdline;
    QList *present, *filtered;
    uint32_t value;

    if (args->cpufeat) {
        cmdline = g_strdup_printf("-cpu %s,%s", args->cpu, args->cpufeat);
    } else {
        cmdline = g_strdup_printf("-cpu %s", args->cpu);
    }

    qtest_start(cmdline);
    path = get_cpu0_qom_path();
    present = qobject_to(QList, qom_get(path, "feature-words"));
    filtered = qobject_to(QList, qom_get(path, "filtered-features"));
    value = get_feature_word(present, args->in_eax, args->in_ecx, args->reg);
    value |= get_feature_word(filtered, args->in_eax, args->in_ecx, args->reg);
    qtest_end();

    g_assert(!!(value & (1U << args->bitnr)) == args->expected_value);

    qobject_unref(present);
    qobject_unref(filtered);
    g_free(path);
    g_free(cmdline);
}

static void test_plus_minus_subprocess(void)
{
    char *path;

    if (!qtest_has_cpu_model("pentium")) {
        return;
    }

    /* Rules:
     * 1)"-foo" overrides "+foo"
     * 2) "[+-]foo" overrides "foo=..."
     * 3) Old feature names with underscores (e.g. "sse4_2")
     *    should keep working
     *
     * Note: rules 1 and 2 are planned to be removed soon, and
     * should generate a warning.
     */
    qtest_start("-cpu pentium,-fpu,+fpu,-mce,mce=on,+cx8,cx8=off,+sse4_1,sse4_2=on");
    path = get_cpu0_qom_path();

    g_assert_false(qom_get_bool(path, "fpu"));
    g_assert_false(qom_get_bool(path, "mce"));
    g_assert_true(qom_get_bool(path, "cx8"));

    /* Test both the original and the alias feature names: */
    g_assert_true(qom_get_bool(path, "sse4-1"));
    g_assert_true(qom_get_bool(path, "sse4.1"));

    g_assert_true(qom_get_bool(path, "sse4-2"));
    g_assert_true(qom_get_bool(path, "sse4.2"));

    qtest_end();
    g_free(path);
}

static void test_plus_minus(void)
{
    if (!qtest_has_cpu_model("pentium")) {
        return;
    }

    g_test_trap_subprocess("/x86/cpuid/parsing-plus-minus/subprocess", 0, 0);
    g_test_trap_assert_passed();
    g_test_trap_assert_stderr("*Ambiguous CPU model string. "
                              "Don't mix both \"-mce\" and \"mce=on\"*");
    g_test_trap_assert_stderr("*Ambiguous CPU model string. "
                              "Don't mix both \"+cx8\" and \"cx8=off\"*");
    g_test_trap_assert_stdout("");
}

static const CpuidTestArgs cpuid_tests[] = {
    /* Original level values for CPU models: */
    {
        "x86/cpuid/phenom/level",
        "phenom", NULL, NULL, "level", 5,
    },
    {
        "x86/cpuid/Conroe/level",
        "Conroe", NULL, NULL, "level", 10,
    },
    {
        "x86/cpuid/SandyBridge/level",
        "SandyBridge", NULL, NULL, "level", 0xd,
    },
    {
        "x86/cpuid/486/xlevel",
        "486", NULL, NULL, "xlevel", 0,
    },
    {
        "x86/cpuid/core2duo/xlevel",
        "core2duo", NULL, NULL, "xlevel", 0x80000008,
    },
    {
        "x86/cpuid/phenom/xlevel",
        "phenom", NULL, NULL, "xlevel", 0x8000001A,
    },
    {
        "x86/cpuid/athlon/xlevel",
        "athlon", NULL, NULL, "xlevel", 0x80000008,
    },
    /* If level is not large enough, it should increase automatically: */
    /* CPUID[6].EAX: */
    {
        "x86/cpuid/auto-level/486/arat",
        "486", "arat=on", NULL, "level", 6,
    },
    /* CPUID[EAX=7,ECX=0].EBX: */
    {
        "x86/cpuid/auto-level/phenom/fsgsbase",
        "phenom", "fsgsbase=on", NULL, "level", 7,
    },
    /* CPUID[EAX=7,ECX=0].ECX: */
    {
        "x86/cpuid/auto-level/phenom/avx512vbmi",
        "phenom", "avx512vbmi=on", NULL, "level", 7,
    },
    /* CPUID[EAX=0xd,ECX=1].EAX: */
    {
        "x86/cpuid/auto-level/phenom/xsaveopt",
        "phenom", "xsaveopt=on", NULL, "level", 0xd,
    },
    /* CPUID[8000_0001].EDX: */
    {
        "x86/cpuid/auto-xlevel/486/3dnow",
        "486", "3dnow=on", NULL, "xlevel", 0x80000001,
    },
    /* CPUID[8000_0001].ECX: */
    {
        "x86/cpuid/auto-xlevel/486/sse4a",
        "486", "sse4a=on", NULL, "xlevel", 0x80000001,
    },
    /* CPUID[8000_0007].EDX: */
    {
        "x86/cpuid/auto-xlevel/486/invtsc",
        "486", "invtsc=on", NULL, "xlevel", 0x80000007,
    },
    /* CPUID[8000_000A].EDX: */
    {
        "x86/cpuid/auto-xlevel/486/npt",
        "486", "svm=on,npt=on", NULL, "xlevel", 0x8000000A,
    },
    /* CPUID[C000_0001].EDX: */
    {
        "x86/cpuid/auto-xlevel2/phenom/xstore",
        "phenom", "xstore=on", NULL, "xlevel2", 0xC0000001,
    },
    /* SVM needs CPUID[0x8000000A] */
    {
        "x86/cpuid/auto-xlevel/athlon/svm",
        "athlon", "svm=on", NULL, "xlevel", 0x8000000A,
    },
    /* If level is already large enough, it shouldn't change: */
    {
        "x86/cpuid/auto-level/SandyBridge/multiple",
        "SandyBridge", "arat=on,fsgsbase=on,avx512vbmi=on",
        NULL, "level", 0xd,
    },
    /* If level is explicitly set, it shouldn't change: */
    {
        "x86/cpuid/auto-level/486/fixed/0xF",
        "486",
        "level=0xF,arat=on,fsgsbase=on,avx512vbmi=on,xsaveopt=on",
        NULL, "level", 0xF,
    },
    {
        "x86/cpuid/auto-level/486/fixed/2",
        "486",
        "level=2,arat=on,fsgsbase=on,avx512vbmi=on,xsaveopt=on",
        NULL, "level", 2,
    },
    {
        "x86/cpuid/auto-level/486/fixed/0",
        "486",
        "level=0,arat=on,fsgsbase=on,avx512vbmi=on,xsaveopt=on",
        NULL, "level", 0,
    },
    /* if xlevel is already large enough, it shouldn't change: */
    {
        "x86/cpuid/auto-xlevel/phenom/3dnow",
        "phenom", "3dnow=on,sse4a=on,invtsc=on,npt=on,svm=on",
        NULL, "xlevel", 0x8000001A,
    },
    /* If xlevel is explicitly set, it shouldn't change: */
    {
        "x86/cpuid/auto-xlevel/486/fixed/80000002",
        "486",
        "xlevel=0x80000002,3dnow=on,sse4a=on,invtsc=on,npt=on,svm=on",
        NULL, "xlevel", 0x80000002,
    },
    {
        "x86/cpuid/auto-xlevel/486/fixed/8000001A",
        "486",
        "xlevel=0x8000001A,3dnow=on,sse4a=on,invtsc=on,npt=on,svm=on",
        NULL, "xlevel", 0x8000001A,
    },
    {
        "x86/cpuid/auto-xlevel/phenom/fixed/0",
        "486",
        "xlevel=0,3dnow=on,sse4a=on,invtsc=on,npt=on,svm=on",
        NULL, "xlevel", 0,
    },
    /* if xlevel2 is already large enough, it shouldn't change: */
    {
        "x86/cpuid/auto-xlevel2/486/fixed",
        "486", "xlevel2=0xC0000002,xstore=on",
        NULL, "xlevel2", 0xC0000002,
    },
};

/*
 * Test cases to ensure that a given feature flag is set in
 * either "feature-words" or "filtered-features", when running QEMU
 * using cmdline
 */
static const FeatureTestArgs feature_tests[] = {
    /* Test feature parsing */
    {
        "x86/cpuid/features/plus",
        "486", "+arat",
        6, 0, "EAX", 2, true,
    },
    {
        "x86/cpuid/features/minus",
        "pentium", "-mmx",
        1, 0, "EDX", 23, false,
    },
    {
        "x86/cpuid/features/on",
        "486", "arat=on",
        6, 0, "EAX", 2, true,
    },
    {
        "x86/cpuid/features/off",
        "pentium", "mmx=off",
        1, 0, "EDX", 23, false,
    },

    {
        "x86/cpuid/features/max-plus-invtsc",
        "max" , "+invtsc",
        0x80000007, 0, "EDX", 8, true,
    },
    {
        "x86/cpuid/features/max-invtsc-on",
        "max", "invtsc=on",
        0x80000007, 0, "EDX", 8, true,
    },
    {
        "x86/cpuid/features/max-minus-mmx",
        "max", "-mmx",
        1, 0, "EDX", 23, false,
    },
    {
        "x86/cpuid/features/max-invtsc-on,mmx=off",
        "max", "mmx=off",
        1, 0, "EDX", 23, false,
    },
};

int main(int argc, char **argv)
{
    g_test_init(&argc, &argv, NULL);

    g_test_add_func("/x86/cpuid/parsing-plus-minus/subprocess",
                    test_plus_minus_subprocess);
    g_test_add_func("/x86/cpuid/parsing-plus-minus", test_plus_minus);

    for (int i = 0; i < ARRAY_SIZE(cpuid_tests); i++) {
        if (!qtest_has_cpu_model(cpuid_tests[i].cpu)) {
            continue;
        }
        qtest_add_data_func(cpuid_tests[i].name,
                            &cpuid_tests[i], test_cpuid_prop);
    }


    for (int i = 0; i < ARRAY_SIZE(feature_tests); i++) {
        if (!qtest_has_cpu_model(feature_tests[i].cpu)) {
            continue;
        }
        qtest_add_data_func(feature_tests[i].name,
                            &feature_tests[i], test_feature_flag);
    }

    return g_test_run();
}
