X-Git-Url: https://git.proxmox.com/?a=blobdiff_plain;f=balloon.c;h=f104b42961b08dc5c4407178ea20f5cc382fcc51;hb=2e36bc1b884a6fc6bc626b13db4960773ad976f7;hp=f2ef50cf77cb0d59e6902b3f77869eb5f5cfd3df;hpb=d38ea87ac54af64ef611de434d07c12dc0399216;p=mirror_qemu.git

diff --git a/balloon.c b/balloon.c
index f2ef50cf77..f104b42961 100644
--- a/balloon.c
+++ b/balloon.c
@@ -25,28 +25,33 @@
  */
 
 #include "qemu/osdep.h"
-#include "qemu-common.h"
-#include "exec/cpu-common.h"
+#include "qemu/atomic.h"
 #include "sysemu/kvm.h"
 #include "sysemu/balloon.h"
-#include "trace.h"
-#include "qmp-commands.h"
+#include "trace-root.h"
+#include "qapi/error.h"
+#include "qapi/qapi-commands-misc.h"
 #include "qapi/qmp/qerror.h"
-#include "qapi/qmp/qjson.h"
 
 static QEMUBalloonEvent *balloon_event_fn;
 static QEMUBalloonStatus *balloon_stat_fn;
 static void *balloon_opaque;
-static bool balloon_inhibited;
+static int balloon_inhibit_count;
 
 bool qemu_balloon_is_inhibited(void)
 {
-    return balloon_inhibited;
+    return atomic_read(&balloon_inhibit_count) > 0;
 }
 
 void qemu_balloon_inhibit(bool state)
 {
-    balloon_inhibited = state;
+    if (state) {
+        atomic_inc(&balloon_inhibit_count);
+    } else {
+        atomic_dec(&balloon_inhibit_count);
+    }
+
+    assert(atomic_read(&balloon_inhibit_count) >= 0);
 }
 
 static bool have_balloon(Error **errp)