root / hw / virtio-balloon.c @ d33ea50a
History | View | Annotate | Download (8 kB)
1 | bd322087 | aliguori | /*
|
---|---|---|---|
2 | bd322087 | aliguori | * Virtio Block Device
|
3 | bd322087 | aliguori | *
|
4 | bd322087 | aliguori | * Copyright IBM, Corp. 2008
|
5 | bd322087 | aliguori | *
|
6 | bd322087 | aliguori | * Authors:
|
7 | bd322087 | aliguori | * Anthony Liguori <aliguori@us.ibm.com>
|
8 | bd322087 | aliguori | *
|
9 | bd322087 | aliguori | * This work is licensed under the terms of the GNU GPL, version 2. See
|
10 | bd322087 | aliguori | * the COPYING file in the top-level directory.
|
11 | bd322087 | aliguori | *
|
12 | bd322087 | aliguori | */
|
13 | bd322087 | aliguori | |
14 | fa6111f2 | Amit Shah | #include "iov.h" |
15 | bd322087 | aliguori | #include "qemu-common.h" |
16 | bd322087 | aliguori | #include "virtio.h" |
17 | bd322087 | aliguori | #include "pc.h" |
18 | bd322087 | aliguori | #include "sysemu.h" |
19 | bd322087 | aliguori | #include "cpu.h" |
20 | 625a5bef | Adam Litke | #include "monitor.h" |
21 | bd322087 | aliguori | #include "balloon.h" |
22 | bd322087 | aliguori | #include "virtio-balloon.h" |
23 | bd322087 | aliguori | #include "kvm.h" |
24 | 625a5bef | Adam Litke | #include "qlist.h" |
25 | 625a5bef | Adam Litke | #include "qint.h" |
26 | 625a5bef | Adam Litke | #include "qstring.h" |
27 | bd322087 | aliguori | |
28 | bd322087 | aliguori | #if defined(__linux__)
|
29 | bd322087 | aliguori | #include <sys/mman.h> |
30 | bd322087 | aliguori | #endif
|
31 | bd322087 | aliguori | |
32 | 07b0403d | Eduardo Habkost | /* Disable guest-provided stats by now (https://bugzilla.redhat.com/show_bug.cgi?id=623903) */
|
33 | 07b0403d | Eduardo Habkost | #define ENABLE_GUEST_STATS 0 |
34 | 07b0403d | Eduardo Habkost | |
35 | 07b0403d | Eduardo Habkost | |
36 | bd322087 | aliguori | typedef struct VirtIOBalloon |
37 | bd322087 | aliguori | { |
38 | bd322087 | aliguori | VirtIODevice vdev; |
39 | 625a5bef | Adam Litke | VirtQueue *ivq, *dvq, *svq; |
40 | bd322087 | aliguori | uint32_t num_pages; |
41 | bd322087 | aliguori | uint32_t actual; |
42 | 625a5bef | Adam Litke | uint64_t stats[VIRTIO_BALLOON_S_NR]; |
43 | 625a5bef | Adam Litke | VirtQueueElement stats_vq_elem; |
44 | 625a5bef | Adam Litke | size_t stats_vq_offset; |
45 | 625a5bef | Adam Litke | MonitorCompletion *stats_callback; |
46 | 625a5bef | Adam Litke | void *stats_opaque_callback_data;
|
47 | bd322087 | aliguori | } VirtIOBalloon; |
48 | bd322087 | aliguori | |
49 | bd322087 | aliguori | static VirtIOBalloon *to_virtio_balloon(VirtIODevice *vdev)
|
50 | bd322087 | aliguori | { |
51 | bd322087 | aliguori | return (VirtIOBalloon *)vdev;
|
52 | bd322087 | aliguori | } |
53 | bd322087 | aliguori | |
54 | bd322087 | aliguori | static void balloon_page(void *addr, int deflate) |
55 | bd322087 | aliguori | { |
56 | bd322087 | aliguori | #if defined(__linux__)
|
57 | bd322087 | aliguori | if (!kvm_enabled() || kvm_has_sync_mmu())
|
58 | e78815a5 | Andreas Färber | qemu_madvise(addr, TARGET_PAGE_SIZE, |
59 | e78815a5 | Andreas Färber | deflate ? QEMU_MADV_WILLNEED : QEMU_MADV_DONTNEED); |
60 | bd322087 | aliguori | #endif
|
61 | bd322087 | aliguori | } |
62 | bd322087 | aliguori | |
63 | 625a5bef | Adam Litke | /*
|
64 | 625a5bef | Adam Litke | * reset_stats - Mark all items in the stats array as unset
|
65 | 625a5bef | Adam Litke | *
|
66 | 625a5bef | Adam Litke | * This function needs to be called at device intialization and before
|
67 | 625a5bef | Adam Litke | * before updating to a set of newly-generated stats. This will ensure that no
|
68 | 625a5bef | Adam Litke | * stale values stick around in case the guest reports a subset of the supported
|
69 | 625a5bef | Adam Litke | * statistics.
|
70 | 625a5bef | Adam Litke | */
|
71 | 625a5bef | Adam Litke | static inline void reset_stats(VirtIOBalloon *dev) |
72 | 625a5bef | Adam Litke | { |
73 | 625a5bef | Adam Litke | int i;
|
74 | 625a5bef | Adam Litke | for (i = 0; i < VIRTIO_BALLOON_S_NR; dev->stats[i++] = -1); |
75 | 625a5bef | Adam Litke | } |
76 | 625a5bef | Adam Litke | |
77 | 625a5bef | Adam Litke | static void stat_put(QDict *dict, const char *label, uint64_t val) |
78 | 625a5bef | Adam Litke | { |
79 | 625a5bef | Adam Litke | if (val != -1) |
80 | 625a5bef | Adam Litke | qdict_put(dict, label, qint_from_int(val)); |
81 | 625a5bef | Adam Litke | } |
82 | 625a5bef | Adam Litke | |
83 | 625a5bef | Adam Litke | static QObject *get_stats_qobject(VirtIOBalloon *dev)
|
84 | 625a5bef | Adam Litke | { |
85 | 625a5bef | Adam Litke | QDict *dict = qdict_new(); |
86 | bd12ff9d | Adam Litke | uint64_t actual = ram_size - ((uint64_t) dev->actual << |
87 | bd12ff9d | Adam Litke | VIRTIO_BALLOON_PFN_SHIFT); |
88 | 625a5bef | Adam Litke | |
89 | 625a5bef | Adam Litke | stat_put(dict, "actual", actual);
|
90 | 07b0403d | Eduardo Habkost | #if ENABLE_GUEST_STATS
|
91 | 625a5bef | Adam Litke | stat_put(dict, "mem_swapped_in", dev->stats[VIRTIO_BALLOON_S_SWAP_IN]);
|
92 | 625a5bef | Adam Litke | stat_put(dict, "mem_swapped_out", dev->stats[VIRTIO_BALLOON_S_SWAP_OUT]);
|
93 | 625a5bef | Adam Litke | stat_put(dict, "major_page_faults", dev->stats[VIRTIO_BALLOON_S_MAJFLT]);
|
94 | 625a5bef | Adam Litke | stat_put(dict, "minor_page_faults", dev->stats[VIRTIO_BALLOON_S_MINFLT]);
|
95 | 625a5bef | Adam Litke | stat_put(dict, "free_mem", dev->stats[VIRTIO_BALLOON_S_MEMFREE]);
|
96 | 625a5bef | Adam Litke | stat_put(dict, "total_mem", dev->stats[VIRTIO_BALLOON_S_MEMTOT]);
|
97 | 07b0403d | Eduardo Habkost | #endif
|
98 | 625a5bef | Adam Litke | |
99 | 625a5bef | Adam Litke | return QOBJECT(dict);
|
100 | 625a5bef | Adam Litke | } |
101 | 625a5bef | Adam Litke | |
102 | bd322087 | aliguori | static void virtio_balloon_handle_output(VirtIODevice *vdev, VirtQueue *vq) |
103 | bd322087 | aliguori | { |
104 | bd322087 | aliguori | VirtIOBalloon *s = to_virtio_balloon(vdev); |
105 | bd322087 | aliguori | VirtQueueElement elem; |
106 | bd322087 | aliguori | |
107 | bd322087 | aliguori | while (virtqueue_pop(vq, &elem)) {
|
108 | bd322087 | aliguori | size_t offset = 0;
|
109 | bd322087 | aliguori | uint32_t pfn; |
110 | bd322087 | aliguori | |
111 | fa6111f2 | Amit Shah | while (iov_to_buf(elem.out_sg, elem.out_num, &pfn, offset, 4) == 4) { |
112 | c227f099 | Anthony Liguori | ram_addr_t pa; |
113 | c227f099 | Anthony Liguori | ram_addr_t addr; |
114 | bd322087 | aliguori | |
115 | c227f099 | Anthony Liguori | pa = (ram_addr_t)ldl_p(&pfn) << VIRTIO_BALLOON_PFN_SHIFT; |
116 | bd322087 | aliguori | offset += 4;
|
117 | bd322087 | aliguori | |
118 | bd322087 | aliguori | addr = cpu_get_physical_page_desc(pa); |
119 | bd322087 | aliguori | if ((addr & ~TARGET_PAGE_MASK) != IO_MEM_RAM)
|
120 | bd322087 | aliguori | continue;
|
121 | bd322087 | aliguori | |
122 | 5c130f65 | pbrook | /* Using qemu_get_ram_ptr is bending the rules a bit, but
|
123 | 5c130f65 | pbrook | should be OK because we only want a single page. */
|
124 | 5c130f65 | pbrook | balloon_page(qemu_get_ram_ptr(addr), !!(vq == s->dvq)); |
125 | bd322087 | aliguori | } |
126 | bd322087 | aliguori | |
127 | bd322087 | aliguori | virtqueue_push(vq, &elem, offset); |
128 | bd322087 | aliguori | virtio_notify(vdev, vq); |
129 | bd322087 | aliguori | } |
130 | bd322087 | aliguori | } |
131 | bd322087 | aliguori | |
132 | 625a5bef | Adam Litke | static void complete_stats_request(VirtIOBalloon *vb) |
133 | 625a5bef | Adam Litke | { |
134 | 625a5bef | Adam Litke | QObject *stats; |
135 | 625a5bef | Adam Litke | |
136 | 625a5bef | Adam Litke | if (!vb->stats_opaque_callback_data)
|
137 | 625a5bef | Adam Litke | return;
|
138 | 625a5bef | Adam Litke | |
139 | 625a5bef | Adam Litke | stats = get_stats_qobject(vb); |
140 | 625a5bef | Adam Litke | vb->stats_callback(vb->stats_opaque_callback_data, stats); |
141 | 625a5bef | Adam Litke | qobject_decref(stats); |
142 | 625a5bef | Adam Litke | vb->stats_opaque_callback_data = NULL;
|
143 | 625a5bef | Adam Litke | vb->stats_callback = NULL;
|
144 | 625a5bef | Adam Litke | } |
145 | 625a5bef | Adam Litke | |
146 | 625a5bef | Adam Litke | static void virtio_balloon_receive_stats(VirtIODevice *vdev, VirtQueue *vq) |
147 | 625a5bef | Adam Litke | { |
148 | 625a5bef | Adam Litke | VirtIOBalloon *s = DO_UPCAST(VirtIOBalloon, vdev, vdev); |
149 | 625a5bef | Adam Litke | VirtQueueElement *elem = &s->stats_vq_elem; |
150 | 625a5bef | Adam Litke | VirtIOBalloonStat stat; |
151 | 625a5bef | Adam Litke | size_t offset = 0;
|
152 | 625a5bef | Adam Litke | |
153 | 625a5bef | Adam Litke | if (!virtqueue_pop(vq, elem)) {
|
154 | 625a5bef | Adam Litke | return;
|
155 | 625a5bef | Adam Litke | } |
156 | 625a5bef | Adam Litke | |
157 | 625a5bef | Adam Litke | /* Initialize the stats to get rid of any stale values. This is only
|
158 | 625a5bef | Adam Litke | * needed to handle the case where a guest supports fewer stats than it
|
159 | 625a5bef | Adam Litke | * used to (ie. it has booted into an old kernel).
|
160 | 625a5bef | Adam Litke | */
|
161 | 625a5bef | Adam Litke | reset_stats(s); |
162 | 625a5bef | Adam Litke | |
163 | fa6111f2 | Amit Shah | while (iov_to_buf(elem->out_sg, elem->out_num, &stat, offset, sizeof(stat)) |
164 | fa6111f2 | Amit Shah | == sizeof(stat)) {
|
165 | 625a5bef | Adam Litke | uint16_t tag = tswap16(stat.tag); |
166 | 625a5bef | Adam Litke | uint64_t val = tswap64(stat.val); |
167 | 625a5bef | Adam Litke | |
168 | 625a5bef | Adam Litke | offset += sizeof(stat);
|
169 | 625a5bef | Adam Litke | if (tag < VIRTIO_BALLOON_S_NR)
|
170 | 625a5bef | Adam Litke | s->stats[tag] = val; |
171 | 625a5bef | Adam Litke | } |
172 | 625a5bef | Adam Litke | s->stats_vq_offset = offset; |
173 | 625a5bef | Adam Litke | |
174 | 625a5bef | Adam Litke | complete_stats_request(s); |
175 | 625a5bef | Adam Litke | } |
176 | 625a5bef | Adam Litke | |
177 | bd322087 | aliguori | static void virtio_balloon_get_config(VirtIODevice *vdev, uint8_t *config_data) |
178 | bd322087 | aliguori | { |
179 | bd322087 | aliguori | VirtIOBalloon *dev = to_virtio_balloon(vdev); |
180 | bd322087 | aliguori | struct virtio_balloon_config config;
|
181 | bd322087 | aliguori | |
182 | bd322087 | aliguori | config.num_pages = cpu_to_le32(dev->num_pages); |
183 | bd322087 | aliguori | config.actual = cpu_to_le32(dev->actual); |
184 | bd322087 | aliguori | |
185 | bd322087 | aliguori | memcpy(config_data, &config, 8);
|
186 | bd322087 | aliguori | } |
187 | bd322087 | aliguori | |
188 | bd322087 | aliguori | static void virtio_balloon_set_config(VirtIODevice *vdev, |
189 | bd322087 | aliguori | const uint8_t *config_data)
|
190 | bd322087 | aliguori | { |
191 | bd322087 | aliguori | VirtIOBalloon *dev = to_virtio_balloon(vdev); |
192 | bd322087 | aliguori | struct virtio_balloon_config config;
|
193 | bd322087 | aliguori | memcpy(&config, config_data, 8);
|
194 | bd322087 | aliguori | dev->actual = config.actual; |
195 | bd322087 | aliguori | } |
196 | bd322087 | aliguori | |
197 | 8172539d | Michael S. Tsirkin | static uint32_t virtio_balloon_get_features(VirtIODevice *vdev, uint32_t f)
|
198 | bd322087 | aliguori | { |
199 | 625a5bef | Adam Litke | f |= (1 << VIRTIO_BALLOON_F_STATS_VQ);
|
200 | 8172539d | Michael S. Tsirkin | return f;
|
201 | bd322087 | aliguori | } |
202 | bd322087 | aliguori | |
203 | 625a5bef | Adam Litke | static void virtio_balloon_to_target(void *opaque, ram_addr_t target, |
204 | 625a5bef | Adam Litke | MonitorCompletion cb, void *cb_data)
|
205 | bd322087 | aliguori | { |
206 | bd322087 | aliguori | VirtIOBalloon *dev = opaque; |
207 | bd322087 | aliguori | |
208 | bd322087 | aliguori | if (target > ram_size)
|
209 | bd322087 | aliguori | target = ram_size; |
210 | bd322087 | aliguori | |
211 | bd322087 | aliguori | if (target) {
|
212 | bd322087 | aliguori | dev->num_pages = (ram_size - target) >> VIRTIO_BALLOON_PFN_SHIFT; |
213 | bd322087 | aliguori | virtio_notify_config(&dev->vdev); |
214 | 625a5bef | Adam Litke | } else {
|
215 | 625a5bef | Adam Litke | /* For now, only allow one request at a time. This restriction can be
|
216 | 625a5bef | Adam Litke | * removed later by queueing callback and data pairs.
|
217 | 625a5bef | Adam Litke | */
|
218 | 625a5bef | Adam Litke | if (dev->stats_callback != NULL) { |
219 | 625a5bef | Adam Litke | return;
|
220 | 625a5bef | Adam Litke | } |
221 | 625a5bef | Adam Litke | dev->stats_callback = cb; |
222 | 625a5bef | Adam Litke | dev->stats_opaque_callback_data = cb_data; |
223 | 07b0403d | Eduardo Habkost | if (ENABLE_GUEST_STATS && (dev->vdev.guest_features & (1 << VIRTIO_BALLOON_F_STATS_VQ))) { |
224 | 625a5bef | Adam Litke | virtqueue_push(dev->svq, &dev->stats_vq_elem, dev->stats_vq_offset); |
225 | 625a5bef | Adam Litke | virtio_notify(&dev->vdev, dev->svq); |
226 | 625a5bef | Adam Litke | } else {
|
227 | 625a5bef | Adam Litke | /* Stats are not supported. Clear out any stale values that might
|
228 | 625a5bef | Adam Litke | * have been set by a more featureful guest kernel.
|
229 | 625a5bef | Adam Litke | */
|
230 | 625a5bef | Adam Litke | reset_stats(dev); |
231 | 625a5bef | Adam Litke | complete_stats_request(dev); |
232 | 625a5bef | Adam Litke | } |
233 | bd322087 | aliguori | } |
234 | bd322087 | aliguori | } |
235 | bd322087 | aliguori | |
236 | bd322087 | aliguori | static void virtio_balloon_save(QEMUFile *f, void *opaque) |
237 | bd322087 | aliguori | { |
238 | bd322087 | aliguori | VirtIOBalloon *s = opaque; |
239 | bd322087 | aliguori | |
240 | bd322087 | aliguori | virtio_save(&s->vdev, f); |
241 | bd322087 | aliguori | |
242 | bd322087 | aliguori | qemu_put_be32(f, s->num_pages); |
243 | bd322087 | aliguori | qemu_put_be32(f, s->actual); |
244 | bd322087 | aliguori | } |
245 | bd322087 | aliguori | |
246 | bd322087 | aliguori | static int virtio_balloon_load(QEMUFile *f, void *opaque, int version_id) |
247 | bd322087 | aliguori | { |
248 | bd322087 | aliguori | VirtIOBalloon *s = opaque; |
249 | bd322087 | aliguori | |
250 | bd322087 | aliguori | if (version_id != 1) |
251 | bd322087 | aliguori | return -EINVAL;
|
252 | bd322087 | aliguori | |
253 | bd322087 | aliguori | virtio_load(&s->vdev, f); |
254 | bd322087 | aliguori | |
255 | bd322087 | aliguori | s->num_pages = qemu_get_be32(f); |
256 | bd322087 | aliguori | s->actual = qemu_get_be32(f); |
257 | bd322087 | aliguori | return 0; |
258 | bd322087 | aliguori | } |
259 | bd322087 | aliguori | |
260 | 53c25cea | Paul Brook | VirtIODevice *virtio_balloon_init(DeviceState *dev) |
261 | bd322087 | aliguori | { |
262 | bd322087 | aliguori | VirtIOBalloon *s; |
263 | bd322087 | aliguori | |
264 | 53c25cea | Paul Brook | s = (VirtIOBalloon *)virtio_common_init("virtio-balloon",
|
265 | 53c25cea | Paul Brook | VIRTIO_ID_BALLOON, |
266 | 53c25cea | Paul Brook | 8, sizeof(VirtIOBalloon)); |
267 | bd322087 | aliguori | |
268 | bd322087 | aliguori | s->vdev.get_config = virtio_balloon_get_config; |
269 | bd322087 | aliguori | s->vdev.set_config = virtio_balloon_set_config; |
270 | bd322087 | aliguori | s->vdev.get_features = virtio_balloon_get_features; |
271 | bd322087 | aliguori | |
272 | bd322087 | aliguori | s->ivq = virtio_add_queue(&s->vdev, 128, virtio_balloon_handle_output);
|
273 | bd322087 | aliguori | s->dvq = virtio_add_queue(&s->vdev, 128, virtio_balloon_handle_output);
|
274 | 625a5bef | Adam Litke | s->svq = virtio_add_queue(&s->vdev, 128, virtio_balloon_receive_stats);
|
275 | bd322087 | aliguori | |
276 | 625a5bef | Adam Litke | reset_stats(s); |
277 | bd322087 | aliguori | qemu_add_balloon_handler(virtio_balloon_to_target, s); |
278 | bd322087 | aliguori | |
279 | 0be71e32 | Alex Williamson | register_savevm(dev, "virtio-balloon", -1, 1, |
280 | 0be71e32 | Alex Williamson | virtio_balloon_save, virtio_balloon_load, s); |
281 | bd322087 | aliguori | |
282 | 53c25cea | Paul Brook | return &s->vdev;
|
283 | bd322087 | aliguori | } |