qemu-iotests: Improve and make use of QMPTestCase.wait_until_completed()

[qemu.git] / qmp.c

diff --git a/qmp.c b/qmp.c
index f556a04d1981f7424d01c48c39168c5f9573a11e..87a28f797d5969af1805f60ec883049c23f86d43 100644 (file)
--- a/qmp.c
+++ b/qmp.c
@@ -114,8 +114,11 @@ void qmp_cpu(int64_t index, Error **errp)
 
 void qmp_cpu_add(int64_t id, Error **errp)
 {
-    if (current_machine->hot_add_cpu) {
-        current_machine->hot_add_cpu(id, errp);
+    MachineClass *mc;
+
+    mc = MACHINE_GET_CLASS(current_machine);
+    if (mc->qemu_machine->hot_add_cpu) {
+        mc->qemu_machine->hot_add_cpu(id, errp);
     } else {
         error_setg(errp, "Not supported");
     }