* start QEMU with 2 unmapped cpus,
 * while in preconfig state
    * add 2 numa nodes
    * assign cpus to them
 * exit preconfig and in running state check that cpus
   are mapped correctly.
Signed-off-by: Igor Mammedov <imammedo@redhat.com>
Message-Id: <1526556607-268163-1-git-send-email-imammedo@redhat.com>
Signed-off-by: Eduardo Habkost <ehabkost@redhat.com>
		
	
			
		
			
				
	
	
		
			355 lines
		
	
	
		
			9.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			355 lines
		
	
	
		
			9.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * NUMA configuration test cases
 | 
						|
 *
 | 
						|
 * Copyright (c) 2017 Red Hat Inc.
 | 
						|
 * Authors:
 | 
						|
 *  Igor Mammedov <imammedo@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 "qemu/osdep.h"
 | 
						|
#include "libqtest.h"
 | 
						|
#include "qapi/qmp/qdict.h"
 | 
						|
#include "qapi/qmp/qlist.h"
 | 
						|
 | 
						|
static char *make_cli(const char *generic_cli, const char *test_cli)
 | 
						|
{
 | 
						|
    return g_strdup_printf("%s %s", generic_cli ? generic_cli : "", test_cli);
 | 
						|
}
 | 
						|
 | 
						|
static void test_mon_explicit(const void *data)
 | 
						|
{
 | 
						|
    char *s;
 | 
						|
    char *cli;
 | 
						|
 | 
						|
    cli = make_cli(data, "-smp 8 "
 | 
						|
                   "-numa node,nodeid=0,cpus=0-3 "
 | 
						|
                   "-numa node,nodeid=1,cpus=4-7 ");
 | 
						|
    qtest_start(cli);
 | 
						|
 | 
						|
    s = hmp("info numa");
 | 
						|
    g_assert(strstr(s, "node 0 cpus: 0 1 2 3"));
 | 
						|
    g_assert(strstr(s, "node 1 cpus: 4 5 6 7"));
 | 
						|
    g_free(s);
 | 
						|
 | 
						|
    qtest_end();
 | 
						|
    g_free(cli);
 | 
						|
}
 | 
						|
 | 
						|
static void test_mon_default(const void *data)
 | 
						|
{
 | 
						|
    char *s;
 | 
						|
    char *cli;
 | 
						|
 | 
						|
    cli = make_cli(data, "-smp 8 -numa node -numa node");
 | 
						|
    qtest_start(cli);
 | 
						|
 | 
						|
    s = hmp("info numa");
 | 
						|
    g_assert(strstr(s, "node 0 cpus: 0 2 4 6"));
 | 
						|
    g_assert(strstr(s, "node 1 cpus: 1 3 5 7"));
 | 
						|
    g_free(s);
 | 
						|
 | 
						|
    qtest_end();
 | 
						|
    g_free(cli);
 | 
						|
}
 | 
						|
 | 
						|
static void test_mon_partial(const void *data)
 | 
						|
{
 | 
						|
    char *s;
 | 
						|
    char *cli;
 | 
						|
 | 
						|
    cli = make_cli(data, "-smp 8 "
 | 
						|
                   "-numa node,nodeid=0,cpus=0-1 "
 | 
						|
                   "-numa node,nodeid=1,cpus=4-5 ");
 | 
						|
    qtest_start(cli);
 | 
						|
 | 
						|
    s = hmp("info numa");
 | 
						|
    g_assert(strstr(s, "node 0 cpus: 0 1 2 3 6 7"));
 | 
						|
    g_assert(strstr(s, "node 1 cpus: 4 5"));
 | 
						|
    g_free(s);
 | 
						|
 | 
						|
    qtest_end();
 | 
						|
    g_free(cli);
 | 
						|
}
 | 
						|
 | 
						|
static QList *get_cpus(QDict **resp)
 | 
						|
{
 | 
						|
    *resp = qmp("{ 'execute': 'query-cpus' }");
 | 
						|
    g_assert(*resp);
 | 
						|
    g_assert(qdict_haskey(*resp, "return"));
 | 
						|
    return qdict_get_qlist(*resp, "return");
 | 
						|
}
 | 
						|
 | 
						|
static void test_query_cpus(const void *data)
 | 
						|
{
 | 
						|
    char *cli;
 | 
						|
    QDict *resp;
 | 
						|
    QList *cpus;
 | 
						|
    QObject *e;
 | 
						|
 | 
						|
    cli = make_cli(data, "-smp 8 -numa node,cpus=0-3 -numa node,cpus=4-7");
 | 
						|
    qtest_start(cli);
 | 
						|
    cpus = get_cpus(&resp);
 | 
						|
    g_assert(cpus);
 | 
						|
 | 
						|
    while ((e = qlist_pop(cpus))) {
 | 
						|
        QDict *cpu, *props;
 | 
						|
        int64_t cpu_idx, node;
 | 
						|
 | 
						|
        cpu = qobject_to(QDict, e);
 | 
						|
        g_assert(qdict_haskey(cpu, "CPU"));
 | 
						|
        g_assert(qdict_haskey(cpu, "props"));
 | 
						|
 | 
						|
        cpu_idx = qdict_get_int(cpu, "CPU");
 | 
						|
        props = qdict_get_qdict(cpu, "props");
 | 
						|
        g_assert(qdict_haskey(props, "node-id"));
 | 
						|
        node = qdict_get_int(props, "node-id");
 | 
						|
        if (cpu_idx >= 0 && cpu_idx < 4) {
 | 
						|
            g_assert_cmpint(node, ==, 0);
 | 
						|
        } else {
 | 
						|
            g_assert_cmpint(node, ==, 1);
 | 
						|
        }
 | 
						|
        qobject_unref(e);
 | 
						|
    }
 | 
						|
 | 
						|
    qobject_unref(resp);
 | 
						|
    qtest_end();
 | 
						|
    g_free(cli);
 | 
						|
}
 | 
						|
 | 
						|
static void pc_numa_cpu(const void *data)
 | 
						|
{
 | 
						|
    char *cli;
 | 
						|
    QDict *resp;
 | 
						|
    QList *cpus;
 | 
						|
    QObject *e;
 | 
						|
 | 
						|
    cli = make_cli(data, "-cpu pentium -smp 8,sockets=2,cores=2,threads=2 "
 | 
						|
        "-numa node,nodeid=0 -numa node,nodeid=1 "
 | 
						|
        "-numa cpu,node-id=1,socket-id=0 "
 | 
						|
        "-numa cpu,node-id=0,socket-id=1,core-id=0 "
 | 
						|
        "-numa cpu,node-id=0,socket-id=1,core-id=1,thread-id=0 "
 | 
						|
        "-numa cpu,node-id=1,socket-id=1,core-id=1,thread-id=1");
 | 
						|
    qtest_start(cli);
 | 
						|
    cpus = get_cpus(&resp);
 | 
						|
    g_assert(cpus);
 | 
						|
 | 
						|
    while ((e = qlist_pop(cpus))) {
 | 
						|
        QDict *cpu, *props;
 | 
						|
        int64_t socket, core, thread, node;
 | 
						|
 | 
						|
        cpu = qobject_to(QDict, e);
 | 
						|
        g_assert(qdict_haskey(cpu, "props"));
 | 
						|
        props = qdict_get_qdict(cpu, "props");
 | 
						|
 | 
						|
        g_assert(qdict_haskey(props, "node-id"));
 | 
						|
        node = qdict_get_int(props, "node-id");
 | 
						|
        g_assert(qdict_haskey(props, "socket-id"));
 | 
						|
        socket = qdict_get_int(props, "socket-id");
 | 
						|
        g_assert(qdict_haskey(props, "core-id"));
 | 
						|
        core = qdict_get_int(props, "core-id");
 | 
						|
        g_assert(qdict_haskey(props, "thread-id"));
 | 
						|
        thread = qdict_get_int(props, "thread-id");
 | 
						|
 | 
						|
        if (socket == 0) {
 | 
						|
            g_assert_cmpint(node, ==, 1);
 | 
						|
        } else if (socket == 1 && core == 0) {
 | 
						|
            g_assert_cmpint(node, ==, 0);
 | 
						|
        } else if (socket == 1 && core == 1 && thread == 0) {
 | 
						|
            g_assert_cmpint(node, ==, 0);
 | 
						|
        } else if (socket == 1 && core == 1 && thread == 1) {
 | 
						|
            g_assert_cmpint(node, ==, 1);
 | 
						|
        } else {
 | 
						|
            g_assert(false);
 | 
						|
        }
 | 
						|
        qobject_unref(e);
 | 
						|
    }
 | 
						|
 | 
						|
    qobject_unref(resp);
 | 
						|
    qtest_end();
 | 
						|
    g_free(cli);
 | 
						|
}
 | 
						|
 | 
						|
static void spapr_numa_cpu(const void *data)
 | 
						|
{
 | 
						|
    char *cli;
 | 
						|
    QDict *resp;
 | 
						|
    QList *cpus;
 | 
						|
    QObject *e;
 | 
						|
 | 
						|
    cli = make_cli(data, "-smp 4,cores=4 "
 | 
						|
        "-numa node,nodeid=0 -numa node,nodeid=1 "
 | 
						|
        "-numa cpu,node-id=0,core-id=0 "
 | 
						|
        "-numa cpu,node-id=0,core-id=1 "
 | 
						|
        "-numa cpu,node-id=0,core-id=2 "
 | 
						|
        "-numa cpu,node-id=1,core-id=3");
 | 
						|
    qtest_start(cli);
 | 
						|
    cpus = get_cpus(&resp);
 | 
						|
    g_assert(cpus);
 | 
						|
 | 
						|
    while ((e = qlist_pop(cpus))) {
 | 
						|
        QDict *cpu, *props;
 | 
						|
        int64_t core, node;
 | 
						|
 | 
						|
        cpu = qobject_to(QDict, e);
 | 
						|
        g_assert(qdict_haskey(cpu, "props"));
 | 
						|
        props = qdict_get_qdict(cpu, "props");
 | 
						|
 | 
						|
        g_assert(qdict_haskey(props, "node-id"));
 | 
						|
        node = qdict_get_int(props, "node-id");
 | 
						|
        g_assert(qdict_haskey(props, "core-id"));
 | 
						|
        core = qdict_get_int(props, "core-id");
 | 
						|
 | 
						|
        if (core >= 0 && core < 3) {
 | 
						|
            g_assert_cmpint(node, ==, 0);
 | 
						|
        } else if (core == 3) {
 | 
						|
            g_assert_cmpint(node, ==, 1);
 | 
						|
        } else {
 | 
						|
            g_assert(false);
 | 
						|
        }
 | 
						|
        qobject_unref(e);
 | 
						|
    }
 | 
						|
 | 
						|
    qobject_unref(resp);
 | 
						|
    qtest_end();
 | 
						|
    g_free(cli);
 | 
						|
}
 | 
						|
 | 
						|
static void aarch64_numa_cpu(const void *data)
 | 
						|
{
 | 
						|
    char *cli;
 | 
						|
    QDict *resp;
 | 
						|
    QList *cpus;
 | 
						|
    QObject *e;
 | 
						|
 | 
						|
    cli = make_cli(data, "-smp 2 "
 | 
						|
        "-numa node,nodeid=0 -numa node,nodeid=1 "
 | 
						|
        "-numa cpu,node-id=1,thread-id=0 "
 | 
						|
        "-numa cpu,node-id=0,thread-id=1");
 | 
						|
    qtest_start(cli);
 | 
						|
    cpus = get_cpus(&resp);
 | 
						|
    g_assert(cpus);
 | 
						|
 | 
						|
    while ((e = qlist_pop(cpus))) {
 | 
						|
        QDict *cpu, *props;
 | 
						|
        int64_t thread, node;
 | 
						|
 | 
						|
        cpu = qobject_to(QDict, e);
 | 
						|
        g_assert(qdict_haskey(cpu, "props"));
 | 
						|
        props = qdict_get_qdict(cpu, "props");
 | 
						|
 | 
						|
        g_assert(qdict_haskey(props, "node-id"));
 | 
						|
        node = qdict_get_int(props, "node-id");
 | 
						|
        g_assert(qdict_haskey(props, "thread-id"));
 | 
						|
        thread = qdict_get_int(props, "thread-id");
 | 
						|
 | 
						|
        if (thread == 0) {
 | 
						|
            g_assert_cmpint(node, ==, 1);
 | 
						|
        } else if (thread == 1) {
 | 
						|
            g_assert_cmpint(node, ==, 0);
 | 
						|
        } else {
 | 
						|
            g_assert(false);
 | 
						|
        }
 | 
						|
        qobject_unref(e);
 | 
						|
    }
 | 
						|
 | 
						|
    qobject_unref(resp);
 | 
						|
    qtest_end();
 | 
						|
    g_free(cli);
 | 
						|
}
 | 
						|
 | 
						|
static void pc_dynamic_cpu_cfg(const void *data)
 | 
						|
{
 | 
						|
    QObject *e;
 | 
						|
    QDict *resp;
 | 
						|
    QList *cpus;
 | 
						|
    QTestState *qs;
 | 
						|
 | 
						|
    qs = qtest_startf("%s %s", data ? (char *)data : "",
 | 
						|
                              "-nodefaults --preconfig -smp 2");
 | 
						|
 | 
						|
    /* create 2 numa nodes */
 | 
						|
    g_assert(!qmp_rsp_is_err(qtest_qmp(qs, "{ 'execute': 'set-numa-node',"
 | 
						|
        " 'arguments': { 'type': 'node', 'nodeid': 0 } }")));
 | 
						|
    g_assert(!qmp_rsp_is_err(qtest_qmp(qs, "{ 'execute': 'set-numa-node',"
 | 
						|
        " 'arguments': { 'type': 'node', 'nodeid': 1 } }")));
 | 
						|
 | 
						|
    /* map 2 cpus in non default reverse order
 | 
						|
     * i.e socket1->node0, socket0->node1
 | 
						|
     */
 | 
						|
    g_assert(!qmp_rsp_is_err(qtest_qmp(qs, "{ 'execute': 'set-numa-node',"
 | 
						|
        " 'arguments': { 'type': 'cpu', 'node-id': 0, 'socket-id': 1 } }")));
 | 
						|
    g_assert(!qmp_rsp_is_err(qtest_qmp(qs, "{ 'execute': 'set-numa-node',"
 | 
						|
        " 'arguments': { 'type': 'cpu', 'node-id': 1, 'socket-id': 0 } }")));
 | 
						|
 | 
						|
    /* let machine initialization to complete and run */
 | 
						|
    g_assert(!qmp_rsp_is_err(qtest_qmp(qs, "{ 'execute': 'exit-preconfig' }")));
 | 
						|
    qtest_qmp_eventwait(qs, "RESUME");
 | 
						|
 | 
						|
    /* check that CPUs are mapped as expected */
 | 
						|
    resp = qtest_qmp(qs, "{ 'execute': 'query-hotpluggable-cpus'}");
 | 
						|
    g_assert(qdict_haskey(resp, "return"));
 | 
						|
    cpus = qdict_get_qlist(resp, "return");
 | 
						|
    g_assert(cpus);
 | 
						|
    while ((e = qlist_pop(cpus))) {
 | 
						|
        const QDict *cpu, *props;
 | 
						|
        int64_t socket, node;
 | 
						|
 | 
						|
        cpu = qobject_to(QDict, e);
 | 
						|
        g_assert(qdict_haskey(cpu, "props"));
 | 
						|
        props = qdict_get_qdict(cpu, "props");
 | 
						|
 | 
						|
        g_assert(qdict_haskey(props, "node-id"));
 | 
						|
        node = qdict_get_int(props, "node-id");
 | 
						|
        g_assert(qdict_haskey(props, "socket-id"));
 | 
						|
        socket = qdict_get_int(props, "socket-id");
 | 
						|
 | 
						|
        if (socket == 0) {
 | 
						|
            g_assert_cmpint(node, ==, 1);
 | 
						|
        } else if (socket == 1) {
 | 
						|
            g_assert_cmpint(node, ==, 0);
 | 
						|
        } else {
 | 
						|
            g_assert(false);
 | 
						|
        }
 | 
						|
        qobject_unref(e);
 | 
						|
    }
 | 
						|
    qobject_unref(resp);
 | 
						|
 | 
						|
    qtest_quit(qs);
 | 
						|
}
 | 
						|
 | 
						|
int main(int argc, char **argv)
 | 
						|
{
 | 
						|
    const char *args = NULL;
 | 
						|
    const char *arch = qtest_get_arch();
 | 
						|
 | 
						|
    if (strcmp(arch, "aarch64") == 0) {
 | 
						|
        args = "-machine virt";
 | 
						|
    }
 | 
						|
 | 
						|
    g_test_init(&argc, &argv, NULL);
 | 
						|
 | 
						|
    qtest_add_data_func("/numa/mon/default", args, test_mon_default);
 | 
						|
    qtest_add_data_func("/numa/mon/cpus/explicit", args, test_mon_explicit);
 | 
						|
    qtest_add_data_func("/numa/mon/cpus/partial", args, test_mon_partial);
 | 
						|
    qtest_add_data_func("/numa/qmp/cpus/query-cpus", args, test_query_cpus);
 | 
						|
 | 
						|
    if (!strcmp(arch, "i386") || !strcmp(arch, "x86_64")) {
 | 
						|
        qtest_add_data_func("/numa/pc/cpu/explicit", args, pc_numa_cpu);
 | 
						|
        qtest_add_data_func("/numa/pc/dynamic/cpu", args, pc_dynamic_cpu_cfg);
 | 
						|
    }
 | 
						|
 | 
						|
    if (!strcmp(arch, "ppc64")) {
 | 
						|
        qtest_add_data_func("/numa/spapr/cpu/explicit", args, spapr_numa_cpu);
 | 
						|
    }
 | 
						|
 | 
						|
    if (!strcmp(arch, "aarch64")) {
 | 
						|
        qtest_add_data_func("/numa/aarch64/cpu/explicit", args,
 | 
						|
                            aarch64_numa_cpu);
 | 
						|
    }
 | 
						|
 | 
						|
    return g_test_run();
 | 
						|
}
 |