LXR qemu/tests/qtest/cpu-plug-test.c

   1/*
   2 * QTest testcase for CPU plugging
   3 *
   4 * Copyright (c) 2015 SUSE Linux GmbH
   5 *
   6 * This work is licensed under the terms of the GNU GPL, version 2 or later.
   7 * See the COPYING file in the top-level directory.
   8 */
   9
  10#include "qemu/osdep.h"
  11
  12#include "qemu-common.h"
  13#include "libqtest-single.h"
  14#include "qapi/qmp/qdict.h"
  15#include "qapi/qmp/qlist.h"
  16
  17struct PlugTestData {
  18    char *machine;
  19    const char *cpu_model;
  20    char *device_model;
  21    unsigned sockets;
  22    unsigned cores;
  23    unsigned threads;
  24    unsigned maxcpus;
  25};
  26typedef struct PlugTestData PlugTestData;
  27
  28static void test_plug_with_device_add(gconstpointer data)
  29{
  30    const PlugTestData *td = data;
  31    char *args;
  32    QTestState *qts;
  33    QDict *resp;
  34    QList *cpus;
  35    QObject *e;
  36    int hotplugged = 0;
  37
  38    args = g_strdup_printf("-machine %s -cpu %s "
  39                           "-smp 1,sockets=%u,cores=%u,threads=%u,maxcpus=%u",
  40                           td->machine, td->cpu_model,
  41                           td->sockets, td->cores, td->threads, td->maxcpus);
  42    qts = qtest_init(args);
  43
  44    resp = qtest_qmp(qts, "{ 'execute': 'query-hotpluggable-cpus'}");
  45    g_assert(qdict_haskey(resp, "return"));
  46    cpus = qdict_get_qlist(resp, "return");
  47    g_assert(cpus);
  48
  49    while ((e = qlist_pop(cpus))) {
  50        const QDict *cpu, *props;
  51
  52        cpu = qobject_to(QDict, e);
  53        if (qdict_haskey(cpu, "qom-path")) {
  54            qobject_unref(e);
  55            continue;
  56        }
  57
  58        g_assert(qdict_haskey(cpu, "props"));
  59        props = qdict_get_qdict(cpu, "props");
  60
  61        qtest_qmp_device_add_qdict(qts, td->device_model, props);
  62        hotplugged++;
  63        qobject_unref(e);
  64    }
  65
  66    /* make sure that there were hotplugged CPUs */
  67    g_assert(hotplugged);
  68    qobject_unref(resp);
  69    qtest_quit(qts);
  70    g_free(args);
  71}
  72
  73static void test_data_free(gpointer data)
  74{
  75    PlugTestData *pc = data;
  76
  77    g_free(pc->machine);
  78    g_free(pc->device_model);
  79    g_free(pc);
  80}
  81
  82static void add_pc_test_case(const char *mname)
  83{
  84    char *path;
  85    PlugTestData *data;
  86
  87    if (!g_str_has_prefix(mname, "pc-")) {
  88        return;
  89    }
  90    data = g_new(PlugTestData, 1);
  91    data->machine = g_strdup(mname);
  92    data->cpu_model = "Haswell"; /* 1.3+ theoretically */
  93    data->device_model = g_strdup_printf("%s-%s-cpu", data->cpu_model,
  94                                         qtest_get_arch());
  95    data->sockets = 1;
  96    data->cores = 3;
  97    data->threads = 2;
  98    data->maxcpus = data->sockets * data->cores * data->threads;
  99
 100    path = g_strdup_printf("cpu-plug/%s/device-add/%ux%ux%u&maxcpus=%u",
 101                           mname, data->sockets, data->cores,
 102                           data->threads, data->maxcpus);
 103    qtest_add_data_func_full(path, data, test_plug_with_device_add,
 104                             test_data_free);
 105    g_free(path);
 106}
 107
 108static void add_pseries_test_case(const char *mname)
 109{
 110    char *path;
 111    PlugTestData *data;
 112
 113    if (!g_str_has_prefix(mname, "pseries-") ||
 114        (g_str_has_prefix(mname, "pseries-2.") && atoi(&mname[10]) < 7)) {
 115        return;
 116    }
 117    data = g_new(PlugTestData, 1);
 118    data->machine = g_strdup(mname);
 119    data->cpu_model = "power8_v2.0";
 120    data->device_model = g_strdup("power8_v2.0-spapr-cpu-core");
 121    data->sockets = 2;
 122    data->cores = 3;
 123    data->threads = 1;
 124    data->maxcpus = data->sockets * data->cores * data->threads;
 125
 126    path = g_strdup_printf("cpu-plug/%s/device-add/%ux%ux%u&maxcpus=%u",
 127                           mname, data->sockets, data->cores,
 128                           data->threads, data->maxcpus);
 129    qtest_add_data_func_full(path, data, test_plug_with_device_add,
 130                             test_data_free);
 131    g_free(path);
 132}
 133
 134static void add_s390x_test_case(const char *mname)
 135{
 136    char *path;
 137    PlugTestData *data;
 138
 139    if (!g_str_has_prefix(mname, "s390-ccw-virtio-")) {
 140        return;
 141    }
 142
 143    data = g_new(PlugTestData, 1);
 144    data->machine = g_strdup(mname);
 145    data->cpu_model = "qemu";
 146    data->device_model = g_strdup("qemu-s390x-cpu");
 147    data->sockets = 1;
 148    data->cores = 3;
 149    data->threads = 1;
 150    data->maxcpus = data->sockets * data->cores * data->threads;
 151
 152    path = g_strdup_printf("cpu-plug/%s/device-add/%ux%ux%u&maxcpus=%u",
 153                           mname, data->sockets, data->cores,
 154                           data->threads, data->maxcpus);
 155    qtest_add_data_func_full(path, data, test_plug_with_device_add,
 156                             test_data_free);
 157    g_free(path);
 158}
 159
 160int main(int argc, char **argv)
 161{
 162    const char *arch = qtest_get_arch();
 163
 164    g_test_init(&argc, &argv, NULL);
 165
 166    if (strcmp(arch, "i386") == 0 || strcmp(arch, "x86_64") == 0) {
 167        qtest_cb_for_every_machine(add_pc_test_case, g_test_quick());
 168    } else if (g_str_equal(arch, "ppc64")) {
 169        qtest_cb_for_every_machine(add_pseries_test_case, g_test_quick());
 170    } else if (g_str_equal(arch, "s390x")) {
 171        qtest_cb_for_every_machine(add_s390x_test_case, g_test_quick());
 172    }
 173
 174    return g_test_run();
 175}
 176