1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27#include "qemu/osdep.h"
28#include "qemu/atomic.h"
29#include "sysemu/kvm.h"
30#include "sysemu/balloon.h"
31#include "trace-root.h"
32#include "qapi/error.h"
33#include "qapi/qapi-commands-misc.h"
34#include "qapi/qmp/qerror.h"
35
36static QEMUBalloonEvent *balloon_event_fn;
37static QEMUBalloonStatus *balloon_stat_fn;
38static void *balloon_opaque;
39static int balloon_inhibit_count;
40
41bool qemu_balloon_is_inhibited(void)
42{
43 return atomic_read(&balloon_inhibit_count) > 0;
44}
45
46void qemu_balloon_inhibit(bool state)
47{
48 if (state) {
49 atomic_inc(&balloon_inhibit_count);
50 } else {
51 atomic_dec(&balloon_inhibit_count);
52 }
53
54 assert(atomic_read(&balloon_inhibit_count) >= 0);
55}
56
57static bool have_balloon(Error **errp)
58{
59 if (kvm_enabled() && !kvm_has_sync_mmu()) {
60 error_set(errp, ERROR_CLASS_KVM_MISSING_CAP,
61 "Using KVM without synchronous MMU, balloon unavailable");
62 return false;
63 }
64 if (!balloon_event_fn) {
65 error_set(errp, ERROR_CLASS_DEVICE_NOT_ACTIVE,
66 "No balloon device has been activated");
67 return false;
68 }
69 return true;
70}
71
72int qemu_add_balloon_handler(QEMUBalloonEvent *event_func,
73 QEMUBalloonStatus *stat_func, void *opaque)
74{
75 if (balloon_event_fn || balloon_stat_fn || balloon_opaque) {
76
77
78
79 return -1;
80 }
81 balloon_event_fn = event_func;
82 balloon_stat_fn = stat_func;
83 balloon_opaque = opaque;
84 return 0;
85}
86
87void qemu_remove_balloon_handler(void *opaque)
88{
89 if (balloon_opaque != opaque) {
90 return;
91 }
92 balloon_event_fn = NULL;
93 balloon_stat_fn = NULL;
94 balloon_opaque = NULL;
95}
96
97BalloonInfo *qmp_query_balloon(Error **errp)
98{
99 BalloonInfo *info;
100
101 if (!have_balloon(errp)) {
102 return NULL;
103 }
104
105 info = g_malloc0(sizeof(*info));
106 balloon_stat_fn(balloon_opaque, info);
107 return info;
108}
109
110void qmp_balloon(int64_t target, Error **errp)
111{
112 if (!have_balloon(errp)) {
113 return;
114 }
115
116 if (target <= 0) {
117 error_setg(errp, QERR_INVALID_PARAMETER_VALUE, "target", "a size");
118 return;
119 }
120
121 trace_balloon_event(balloon_opaque, target);
122 balloon_event_fn(balloon_opaque, target);
123}
124