]> git.proxmox.com Git - pve-qemu.git/blame - debian/patches/pve/0026-PVE-Backup-add-vma-backup-format-code.patch
bump version to 6.2.0-11
[pve-qemu.git] / debian / patches / pve / 0026-PVE-Backup-add-vma-backup-format-code.patch
CommitLineData
23102ed6 1From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
6402d961 2From: Dietmar Maurer <dietmar@proxmox.com>
83faa3fe
TL
3Date: Mon, 6 Apr 2020 12:16:57 +0200
4Subject: [PATCH] PVE-Backup: add vma backup format code
95259824 5
ddbf7a87 6Signed-off-by: Thomas Lamprecht <t.lamprecht@proxmox.com>
2bf61c3e
FE
7[FE: create: register all streams before entering coroutines]
8Signed-off-by: Fabian Ebner <f.ebner@proxmox.com>
95259824 9---
817b7667
SR
10 block/meson.build | 2 +
11 meson.build | 5 +
2653a5f0 12 vma-reader.c | 860 ++++++++++++++++++++++++++++++++++++++++++++++
817b7667 13 vma-writer.c | 790 ++++++++++++++++++++++++++++++++++++++++++
2653a5f0 14 vma.c | 849 +++++++++++++++++++++++++++++++++++++++++++++
817b7667 15 vma.h | 150 ++++++++
2653a5f0 16 6 files changed, 2656 insertions(+)
95259824
WB
17 create mode 100644 vma-reader.c
18 create mode 100644 vma-writer.c
19 create mode 100644 vma.c
20 create mode 100644 vma.h
21
817b7667 22diff --git a/block/meson.build b/block/meson.build
4567474e 23index c9d1fdca7d..72081a9974 100644
817b7667
SR
24--- a/block/meson.build
25+++ b/block/meson.build
f376b2b9 26@@ -44,6 +44,8 @@ block_ss.add(files(
817b7667 27 'zeroinit.c',
8dca018b 28 ), zstd, zlib, gnutls)
6838f038 29
817b7667
SR
30+block_ss.add(files('../vma-writer.c'), libuuid)
31+
32 softmmu_ss.add(when: 'CONFIG_TCG', if_true: files('blkreplay.c'))
95259824 33
817b7667
SR
34 block_ss.add(when: 'CONFIG_QCOW1', if_true: files('qcow.c'))
35diff --git a/meson.build b/meson.build
4567474e 36index 96de1a6ef9..54c23b9567 100644
817b7667
SR
37--- a/meson.build
38+++ b/meson.build
4567474e 39@@ -1202,6 +1202,8 @@ keyutils = dependency('libkeyutils', required: false,
95259824 40
817b7667 41 has_gettid = cc.has_function('gettid')
95259824 42
817b7667
SR
43+libuuid = cc.find_library('uuid', required: true)
44+
4567474e
FE
45 # libselinux
46 selinux = dependency('libselinux',
47 required: get_option('selinux'),
48@@ -3070,6 +3072,9 @@ if have_tools
49 dependencies: [blockdev, qemuutil, gnutls, selinux],
50 install: true)
95259824 51
c5e8e7c9 52+ vma = executable('vma', files('vma.c', 'vma-reader.c') + genh,
817b7667
SR
53+ dependencies: [authz, block, crypto, io, qom], install: true)
54+
55 subdir('storage-daemon')
56 subdir('contrib/rdmacm-mux')
57 subdir('contrib/elf2dmp')
95259824
WB
58diff --git a/vma-reader.c b/vma-reader.c
59new file mode 100644
2653a5f0 60index 0000000000..4f4ee2b47b
95259824
WB
61--- /dev/null
62+++ b/vma-reader.c
2653a5f0 63@@ -0,0 +1,860 @@
95259824
WB
64+/*
65+ * VMA: Virtual Machine Archive
66+ *
67+ * Copyright (C) 2012 Proxmox Server Solutions
68+ *
69+ * Authors:
70+ * Dietmar Maurer (dietmar@proxmox.com)
71+ *
72+ * This work is licensed under the terms of the GNU GPL, version 2 or later.
73+ * See the COPYING file in the top-level directory.
74+ *
75+ */
76+
77+#include "qemu/osdep.h"
78+#include <glib.h>
79+#include <uuid/uuid.h>
80+
81+#include "qemu-common.h"
82+#include "qemu/timer.h"
83+#include "qemu/ratelimit.h"
84+#include "vma.h"
85+#include "block/block.h"
86+#include "sysemu/block-backend.h"
87+
88+static unsigned char zero_vma_block[VMA_BLOCK_SIZE];
89+
90+typedef struct VmaRestoreState {
67af0fa4 91+ BlockBackend *target;
95259824
WB
92+ bool write_zeroes;
93+ unsigned long *bitmap;
94+ int bitmap_size;
95+} VmaRestoreState;
96+
97+struct VmaReader {
98+ int fd;
99+ GChecksum *md5csum;
100+ GHashTable *blob_hash;
101+ unsigned char *head_data;
102+ VmaDeviceInfo devinfo[256];
103+ VmaRestoreState rstate[256];
104+ GList *cdata_list;
105+ guint8 vmstate_stream;
106+ uint32_t vmstate_clusters;
107+ /* to show restore percentage if run with -v */
108+ time_t start_time;
109+ int64_t cluster_count;
110+ int64_t clusters_read;
67af0fa4
WB
111+ int64_t zero_cluster_data;
112+ int64_t partial_zero_cluster_data;
95259824
WB
113+ int clusters_read_per;
114+};
115+
116+static guint
117+g_int32_hash(gconstpointer v)
118+{
119+ return *(const uint32_t *)v;
120+}
121+
122+static gboolean
123+g_int32_equal(gconstpointer v1, gconstpointer v2)
124+{
125+ return *((const uint32_t *)v1) == *((const uint32_t *)v2);
126+}
127+
128+static int vma_reader_get_bitmap(VmaRestoreState *rstate, int64_t cluster_num)
129+{
130+ assert(rstate);
131+ assert(rstate->bitmap);
132+
133+ unsigned long val, idx, bit;
134+
135+ idx = cluster_num / BITS_PER_LONG;
136+
137+ assert(rstate->bitmap_size > idx);
138+
139+ bit = cluster_num % BITS_PER_LONG;
140+ val = rstate->bitmap[idx];
141+
142+ return !!(val & (1UL << bit));
143+}
144+
145+static void vma_reader_set_bitmap(VmaRestoreState *rstate, int64_t cluster_num,
146+ int dirty)
147+{
148+ assert(rstate);
149+ assert(rstate->bitmap);
150+
151+ unsigned long val, idx, bit;
152+
153+ idx = cluster_num / BITS_PER_LONG;
154+
155+ assert(rstate->bitmap_size > idx);
156+
157+ bit = cluster_num % BITS_PER_LONG;
158+ val = rstate->bitmap[idx];
159+ if (dirty) {
160+ if (!(val & (1UL << bit))) {
161+ val |= 1UL << bit;
162+ }
163+ } else {
164+ if (val & (1UL << bit)) {
165+ val &= ~(1UL << bit);
166+ }
167+ }
168+ rstate->bitmap[idx] = val;
169+}
170+
171+typedef struct VmaBlob {
172+ uint32_t start;
173+ uint32_t len;
174+ void *data;
175+} VmaBlob;
176+
177+static const VmaBlob *get_header_blob(VmaReader *vmar, uint32_t pos)
178+{
179+ assert(vmar);
180+ assert(vmar->blob_hash);
181+
182+ return g_hash_table_lookup(vmar->blob_hash, &pos);
183+}
184+
185+static const char *get_header_str(VmaReader *vmar, uint32_t pos)
186+{
187+ const VmaBlob *blob = get_header_blob(vmar, pos);
188+ if (!blob) {
189+ return NULL;
190+ }
191+ const char *res = (char *)blob->data;
192+ if (res[blob->len-1] != '\0') {
193+ return NULL;
194+ }
195+ return res;
196+}
197+
198+static ssize_t
199+safe_read(int fd, unsigned char *buf, size_t count)
200+{
201+ ssize_t n;
202+
203+ do {
204+ n = read(fd, buf, count);
205+ } while (n < 0 && errno == EINTR);
206+
207+ return n;
208+}
209+
210+static ssize_t
211+full_read(int fd, unsigned char *buf, size_t len)
212+{
213+ ssize_t n;
214+ size_t total;
215+
216+ total = 0;
217+
218+ while (len > 0) {
219+ n = safe_read(fd, buf, len);
220+
221+ if (n == 0) {
222+ return total;
223+ }
224+
225+ if (n <= 0) {
226+ break;
227+ }
228+
229+ buf += n;
230+ total += n;
231+ len -= n;
232+ }
233+
234+ if (len) {
235+ return -1;
236+ }
237+
238+ return total;
239+}
240+
241+void vma_reader_destroy(VmaReader *vmar)
242+{
243+ assert(vmar);
244+
245+ if (vmar->fd >= 0) {
246+ close(vmar->fd);
247+ }
248+
249+ if (vmar->cdata_list) {
250+ g_list_free(vmar->cdata_list);
251+ }
252+
253+ int i;
254+ for (i = 1; i < 256; i++) {
255+ if (vmar->rstate[i].bitmap) {
256+ g_free(vmar->rstate[i].bitmap);
257+ }
2653a5f0
TL
258+ if (vmar->rstate[i].target) {
259+ blk_unref(vmar->rstate[i].target);
260+ }
95259824
WB
261+ }
262+
263+ if (vmar->md5csum) {
264+ g_checksum_free(vmar->md5csum);
265+ }
266+
267+ if (vmar->blob_hash) {
268+ g_hash_table_destroy(vmar->blob_hash);
269+ }
270+
271+ if (vmar->head_data) {
272+ g_free(vmar->head_data);
273+ }
274+
275+ g_free(vmar);
276+
277+};
278+
279+static int vma_reader_read_head(VmaReader *vmar, Error **errp)
280+{
281+ assert(vmar);
282+ assert(errp);
283+ assert(*errp == NULL);
284+
285+ unsigned char md5sum[16];
286+ int i;
287+ int ret = 0;
288+
289+ vmar->head_data = g_malloc(sizeof(VmaHeader));
290+
291+ if (full_read(vmar->fd, vmar->head_data, sizeof(VmaHeader)) !=
292+ sizeof(VmaHeader)) {
293+ error_setg(errp, "can't read vma header - %s",
294+ errno ? g_strerror(errno) : "got EOF");
295+ return -1;
296+ }
297+
298+ VmaHeader *h = (VmaHeader *)vmar->head_data;
299+
300+ if (h->magic != VMA_MAGIC) {
301+ error_setg(errp, "not a vma file - wrong magic number");
302+ return -1;
303+ }
304+
305+ uint32_t header_size = GUINT32_FROM_BE(h->header_size);
306+ int need = header_size - sizeof(VmaHeader);
307+ if (need <= 0) {
308+ error_setg(errp, "wrong vma header size %d", header_size);
309+ return -1;
310+ }
311+
312+ vmar->head_data = g_realloc(vmar->head_data, header_size);
313+ h = (VmaHeader *)vmar->head_data;
314+
315+ if (full_read(vmar->fd, vmar->head_data + sizeof(VmaHeader), need) !=
316+ need) {
317+ error_setg(errp, "can't read vma header data - %s",
318+ errno ? g_strerror(errno) : "got EOF");
319+ return -1;
320+ }
321+
322+ memcpy(md5sum, h->md5sum, 16);
323+ memset(h->md5sum, 0, 16);
324+
325+ g_checksum_reset(vmar->md5csum);
326+ g_checksum_update(vmar->md5csum, vmar->head_data, header_size);
327+ gsize csize = 16;
328+ g_checksum_get_digest(vmar->md5csum, (guint8 *)(h->md5sum), &csize);
329+
330+ if (memcmp(md5sum, h->md5sum, 16) != 0) {
331+ error_setg(errp, "wrong vma header chechsum");
332+ return -1;
333+ }
334+
335+ /* we can modify header data after checksum verify */
336+ h->header_size = header_size;
337+
338+ h->version = GUINT32_FROM_BE(h->version);
339+ if (h->version != 1) {
340+ error_setg(errp, "wrong vma version %d", h->version);
341+ return -1;
342+ }
343+
344+ h->ctime = GUINT64_FROM_BE(h->ctime);
345+ h->blob_buffer_offset = GUINT32_FROM_BE(h->blob_buffer_offset);
346+ h->blob_buffer_size = GUINT32_FROM_BE(h->blob_buffer_size);
347+
348+ uint32_t bstart = h->blob_buffer_offset + 1;
349+ uint32_t bend = h->blob_buffer_offset + h->blob_buffer_size;
350+
351+ if (bstart <= sizeof(VmaHeader)) {
352+ error_setg(errp, "wrong vma blob buffer offset %d",
353+ h->blob_buffer_offset);
354+ return -1;
355+ }
356+
357+ if (bend > header_size) {
358+ error_setg(errp, "wrong vma blob buffer size %d/%d",
359+ h->blob_buffer_offset, h->blob_buffer_size);
360+ return -1;
361+ }
362+
363+ while ((bstart + 2) <= bend) {
364+ uint32_t size = vmar->head_data[bstart] +
365+ (vmar->head_data[bstart+1] << 8);
366+ if ((bstart + size + 2) <= bend) {
367+ VmaBlob *blob = g_new0(VmaBlob, 1);
368+ blob->start = bstart - h->blob_buffer_offset;
369+ blob->len = size;
370+ blob->data = vmar->head_data + bstart + 2;
371+ g_hash_table_insert(vmar->blob_hash, &blob->start, blob);
372+ }
373+ bstart += size + 2;
374+ }
375+
376+
377+ int count = 0;
378+ for (i = 1; i < 256; i++) {
379+ VmaDeviceInfoHeader *dih = &h->dev_info[i];
380+ uint32_t devname_ptr = GUINT32_FROM_BE(dih->devname_ptr);
381+ uint64_t size = GUINT64_FROM_BE(dih->size);
382+ const char *devname = get_header_str(vmar, devname_ptr);
383+
384+ if (size && devname) {
385+ count++;
386+ vmar->devinfo[i].size = size;
387+ vmar->devinfo[i].devname = devname;
388+
389+ if (strcmp(devname, "vmstate") == 0) {
390+ vmar->vmstate_stream = i;
391+ }
392+ }
393+ }
394+
95259824
WB
395+ for (i = 0; i < VMA_MAX_CONFIGS; i++) {
396+ uint32_t name_ptr = GUINT32_FROM_BE(h->config_names[i]);
397+ uint32_t data_ptr = GUINT32_FROM_BE(h->config_data[i]);
398+
399+ if (!(name_ptr && data_ptr)) {
400+ continue;
401+ }
402+ const char *name = get_header_str(vmar, name_ptr);
403+ const VmaBlob *blob = get_header_blob(vmar, data_ptr);
404+
405+ if (!(name && blob)) {
406+ error_setg(errp, "vma contains invalid data pointers");
407+ return -1;
408+ }
409+
410+ VmaConfigData *cdata = g_new0(VmaConfigData, 1);
411+ cdata->name = name;
412+ cdata->data = blob->data;
413+ cdata->len = blob->len;
414+
415+ vmar->cdata_list = g_list_append(vmar->cdata_list, cdata);
416+ }
417+
418+ return ret;
419+};
420+
421+VmaReader *vma_reader_create(const char *filename, Error **errp)
422+{
423+ assert(filename);
424+ assert(errp);
425+
426+ VmaReader *vmar = g_new0(VmaReader, 1);
427+
428+ if (strcmp(filename, "-") == 0) {
429+ vmar->fd = dup(0);
430+ } else {
431+ vmar->fd = open(filename, O_RDONLY);
432+ }
433+
434+ if (vmar->fd < 0) {
435+ error_setg(errp, "can't open file %s - %s\n", filename,
436+ g_strerror(errno));
437+ goto err;
438+ }
439+
440+ vmar->md5csum = g_checksum_new(G_CHECKSUM_MD5);
441+ if (!vmar->md5csum) {
442+ error_setg(errp, "can't allocate cmsum\n");
443+ goto err;
444+ }
445+
446+ vmar->blob_hash = g_hash_table_new_full(g_int32_hash, g_int32_equal,
447+ NULL, g_free);
448+
449+ if (vma_reader_read_head(vmar, errp) < 0) {
450+ goto err;
451+ }
452+
453+ return vmar;
454+
455+err:
456+ if (vmar) {
457+ vma_reader_destroy(vmar);
458+ }
459+
460+ return NULL;
461+}
462+
463+VmaHeader *vma_reader_get_header(VmaReader *vmar)
464+{
465+ assert(vmar);
466+ assert(vmar->head_data);
467+
468+ return (VmaHeader *)(vmar->head_data);
469+}
470+
471+GList *vma_reader_get_config_data(VmaReader *vmar)
472+{
473+ assert(vmar);
474+ assert(vmar->head_data);
475+
476+ return vmar->cdata_list;
477+}
478+
479+VmaDeviceInfo *vma_reader_get_device_info(VmaReader *vmar, guint8 dev_id)
480+{
481+ assert(vmar);
482+ assert(dev_id);
483+
484+ if (vmar->devinfo[dev_id].size && vmar->devinfo[dev_id].devname) {
485+ return &vmar->devinfo[dev_id];
486+ }
487+
488+ return NULL;
489+}
490+
67af0fa4
WB
491+static void allocate_rstate(VmaReader *vmar, guint8 dev_id,
492+ BlockBackend *target, bool write_zeroes)
493+{
494+ assert(vmar);
495+ assert(dev_id);
496+
497+ vmar->rstate[dev_id].target = target;
498+ vmar->rstate[dev_id].write_zeroes = write_zeroes;
499+
500+ int64_t size = vmar->devinfo[dev_id].size;
501+
502+ int64_t bitmap_size = (size/BDRV_SECTOR_SIZE) +
503+ (VMA_CLUSTER_SIZE/BDRV_SECTOR_SIZE) * BITS_PER_LONG - 1;
504+ bitmap_size /= (VMA_CLUSTER_SIZE/BDRV_SECTOR_SIZE) * BITS_PER_LONG;
505+
506+ vmar->rstate[dev_id].bitmap_size = bitmap_size;
507+ vmar->rstate[dev_id].bitmap = g_new0(unsigned long, bitmap_size);
508+
509+ vmar->cluster_count += size/VMA_CLUSTER_SIZE;
510+}
511+
512+int vma_reader_register_bs(VmaReader *vmar, guint8 dev_id, BlockBackend *target,
95259824
WB
513+ bool write_zeroes, Error **errp)
514+{
515+ assert(vmar);
67af0fa4 516+ assert(target != NULL);
95259824 517+ assert(dev_id);
67af0fa4 518+ assert(vmar->rstate[dev_id].target == NULL);
95259824 519+
67af0fa4 520+ int64_t size = blk_getlength(target);
95259824
WB
521+ int64_t size_diff = size - vmar->devinfo[dev_id].size;
522+
523+ /* storage types can have different size restrictions, so it
524+ * is not always possible to create an image with exact size.
525+ * So we tolerate a size difference up to 4MB.
526+ */
527+ if ((size_diff < 0) || (size_diff > 4*1024*1024)) {
528+ error_setg(errp, "vma_reader_register_bs for stream %s failed - "
529+ "unexpected size %zd != %zd", vmar->devinfo[dev_id].devname,
530+ size, vmar->devinfo[dev_id].size);
531+ return -1;
532+ }
533+
67af0fa4 534+ allocate_rstate(vmar, dev_id, target, write_zeroes);
95259824
WB
535+
536+ return 0;
537+}
538+
539+static ssize_t safe_write(int fd, void *buf, size_t count)
540+{
541+ ssize_t n;
542+
543+ do {
544+ n = write(fd, buf, count);
545+ } while (n < 0 && errno == EINTR);
546+
547+ return n;
548+}
549+
550+static size_t full_write(int fd, void *buf, size_t len)
551+{
552+ ssize_t n;
553+ size_t total;
554+
555+ total = 0;
556+
557+ while (len > 0) {
558+ n = safe_write(fd, buf, len);
559+ if (n < 0) {
560+ return n;
561+ }
562+ buf += n;
563+ total += n;
564+ len -= n;
565+ }
566+
567+ if (len) {
568+ /* incomplete write ? */
569+ return -1;
570+ }
571+
572+ return total;
573+}
574+
575+static int restore_write_data(VmaReader *vmar, guint8 dev_id,
67af0fa4 576+ BlockBackend *target, int vmstate_fd,
95259824
WB
577+ unsigned char *buf, int64_t sector_num,
578+ int nb_sectors, Error **errp)
579+{
580+ assert(vmar);
581+
582+ if (dev_id == vmar->vmstate_stream) {
583+ if (vmstate_fd >= 0) {
584+ int len = nb_sectors * BDRV_SECTOR_SIZE;
585+ int res = full_write(vmstate_fd, buf, len);
586+ if (res < 0) {
587+ error_setg(errp, "write vmstate failed %d", res);
588+ return -1;
589+ }
590+ }
591+ } else {
67af0fa4 592+ int res = blk_pwrite(target, sector_num * BDRV_SECTOR_SIZE, buf, nb_sectors * BDRV_SECTOR_SIZE, 0);
95259824 593+ if (res < 0) {
67af0fa4
WB
594+ error_setg(errp, "blk_pwrite to %s failed (%d)",
595+ bdrv_get_device_name(blk_bs(target)), res);
95259824
WB
596+ return -1;
597+ }
598+ }
599+ return 0;
600+}
67af0fa4 601+
95259824
WB
602+static int restore_extent(VmaReader *vmar, unsigned char *buf,
603+ int extent_size, int vmstate_fd,
67af0fa4 604+ bool verbose, bool verify, Error **errp)
95259824
WB
605+{
606+ assert(vmar);
607+ assert(buf);
608+
609+ VmaExtentHeader *ehead = (VmaExtentHeader *)buf;
610+ int start = VMA_EXTENT_HEADER_SIZE;
611+ int i;
612+
613+ for (i = 0; i < VMA_BLOCKS_PER_EXTENT; i++) {
614+ uint64_t block_info = GUINT64_FROM_BE(ehead->blockinfo[i]);
615+ uint64_t cluster_num = block_info & 0xffffffff;
616+ uint8_t dev_id = (block_info >> 32) & 0xff;
617+ uint16_t mask = block_info >> (32+16);
618+ int64_t max_sector;
619+
620+ if (!dev_id) {
621+ continue;
622+ }
623+
624+ VmaRestoreState *rstate = &vmar->rstate[dev_id];
67af0fa4 625+ BlockBackend *target = NULL;
95259824
WB
626+
627+ if (dev_id != vmar->vmstate_stream) {
67af0fa4
WB
628+ target = rstate->target;
629+ if (!verify && !target) {
95259824
WB
630+ error_setg(errp, "got wrong dev id %d", dev_id);
631+ return -1;
632+ }
633+
634+ if (vma_reader_get_bitmap(rstate, cluster_num)) {
635+ error_setg(errp, "found duplicated cluster %zd for stream %s",
636+ cluster_num, vmar->devinfo[dev_id].devname);
637+ return -1;
638+ }
639+ vma_reader_set_bitmap(rstate, cluster_num, 1);
640+
641+ max_sector = vmar->devinfo[dev_id].size/BDRV_SECTOR_SIZE;
642+ } else {
643+ max_sector = G_MAXINT64;
644+ if (cluster_num != vmar->vmstate_clusters) {
645+ error_setg(errp, "found out of order vmstate data");
646+ return -1;
647+ }
648+ vmar->vmstate_clusters++;
649+ }
650+
651+ vmar->clusters_read++;
652+
653+ if (verbose) {
654+ time_t duration = time(NULL) - vmar->start_time;
655+ int percent = (vmar->clusters_read*100)/vmar->cluster_count;
656+ if (percent != vmar->clusters_read_per) {
657+ printf("progress %d%% (read %zd bytes, duration %zd sec)\n",
658+ percent, vmar->clusters_read*VMA_CLUSTER_SIZE,
659+ duration);
660+ fflush(stdout);
661+ vmar->clusters_read_per = percent;
662+ }
663+ }
664+
665+ /* try to write whole clusters to speedup restore */
666+ if (mask == 0xffff) {
667+ if ((start + VMA_CLUSTER_SIZE) > extent_size) {
668+ error_setg(errp, "short vma extent - too many blocks");
669+ return -1;
670+ }
671+ int64_t sector_num = (cluster_num * VMA_CLUSTER_SIZE) /
672+ BDRV_SECTOR_SIZE;
673+ int64_t end_sector = sector_num +
674+ VMA_CLUSTER_SIZE/BDRV_SECTOR_SIZE;
675+
676+ if (end_sector > max_sector) {
677+ end_sector = max_sector;
678+ }
679+
680+ if (end_sector <= sector_num) {
9b05d1d4 681+ error_setg(errp, "got wrong block address - write beyond end");
95259824
WB
682+ return -1;
683+ }
684+
67af0fa4
WB
685+ if (!verify) {
686+ int nb_sectors = end_sector - sector_num;
687+ if (restore_write_data(vmar, dev_id, target, vmstate_fd,
688+ buf + start, sector_num, nb_sectors,
689+ errp) < 0) {
690+ return -1;
691+ }
95259824
WB
692+ }
693+
694+ start += VMA_CLUSTER_SIZE;
695+ } else {
696+ int j;
697+ int bit = 1;
698+
699+ for (j = 0; j < 16; j++) {
700+ int64_t sector_num = (cluster_num*VMA_CLUSTER_SIZE +
701+ j*VMA_BLOCK_SIZE)/BDRV_SECTOR_SIZE;
702+
703+ int64_t end_sector = sector_num +
704+ VMA_BLOCK_SIZE/BDRV_SECTOR_SIZE;
705+ if (end_sector > max_sector) {
706+ end_sector = max_sector;
707+ }
708+
709+ if (mask & bit) {
710+ if ((start + VMA_BLOCK_SIZE) > extent_size) {
711+ error_setg(errp, "short vma extent - too many blocks");
712+ return -1;
713+ }
714+
715+ if (end_sector <= sector_num) {
716+ error_setg(errp, "got wrong block address - "
9b05d1d4 717+ "write beyond end");
95259824
WB
718+ return -1;
719+ }
720+
67af0fa4
WB
721+ if (!verify) {
722+ int nb_sectors = end_sector - sector_num;
723+ if (restore_write_data(vmar, dev_id, target, vmstate_fd,
724+ buf + start, sector_num,
725+ nb_sectors, errp) < 0) {
726+ return -1;
727+ }
95259824
WB
728+ }
729+
730+ start += VMA_BLOCK_SIZE;
731+
732+ } else {
733+
67af0fa4
WB
734+
735+ if (end_sector > sector_num) {
95259824
WB
736+ /* Todo: use bdrv_co_write_zeroes (but that need to
737+ * be run inside coroutine?)
738+ */
739+ int nb_sectors = end_sector - sector_num;
67af0fa4
WB
740+ int zero_size = BDRV_SECTOR_SIZE*nb_sectors;
741+ vmar->zero_cluster_data += zero_size;
742+ if (mask != 0) {
743+ vmar->partial_zero_cluster_data += zero_size;
744+ }
745+
746+ if (rstate->write_zeroes && !verify) {
747+ if (restore_write_data(vmar, dev_id, target, vmstate_fd,
748+ zero_vma_block, sector_num,
749+ nb_sectors, errp) < 0) {
750+ return -1;
751+ }
95259824
WB
752+ }
753+ }
754+ }
755+
756+ bit = bit << 1;
757+ }
758+ }
759+ }
760+
761+ if (start != extent_size) {
762+ error_setg(errp, "vma extent error - missing blocks");
763+ return -1;
764+ }
765+
766+ return 0;
767+}
768+
67af0fa4
WB
769+static int vma_reader_restore_full(VmaReader *vmar, int vmstate_fd,
770+ bool verbose, bool verify,
771+ Error **errp)
95259824
WB
772+{
773+ assert(vmar);
774+ assert(vmar->head_data);
775+
776+ int ret = 0;
777+ unsigned char buf[VMA_MAX_EXTENT_SIZE];
778+ int buf_pos = 0;
779+ unsigned char md5sum[16];
780+ VmaHeader *h = (VmaHeader *)vmar->head_data;
781+
782+ vmar->start_time = time(NULL);
783+
784+ while (1) {
785+ int bytes = full_read(vmar->fd, buf + buf_pos, sizeof(buf) - buf_pos);
786+ if (bytes < 0) {
787+ error_setg(errp, "read failed - %s", g_strerror(errno));
788+ return -1;
789+ }
790+
791+ buf_pos += bytes;
792+
793+ if (!buf_pos) {
794+ break; /* EOF */
795+ }
796+
797+ if (buf_pos < VMA_EXTENT_HEADER_SIZE) {
798+ error_setg(errp, "read short extent (%d bytes)", buf_pos);
799+ return -1;
800+ }
801+
802+ VmaExtentHeader *ehead = (VmaExtentHeader *)buf;
803+
804+ /* extract md5sum */
805+ memcpy(md5sum, ehead->md5sum, sizeof(ehead->md5sum));
806+ memset(ehead->md5sum, 0, sizeof(ehead->md5sum));
807+
808+ g_checksum_reset(vmar->md5csum);
809+ g_checksum_update(vmar->md5csum, buf, VMA_EXTENT_HEADER_SIZE);
810+ gsize csize = 16;
811+ g_checksum_get_digest(vmar->md5csum, ehead->md5sum, &csize);
812+
813+ if (memcmp(md5sum, ehead->md5sum, 16) != 0) {
814+ error_setg(errp, "wrong vma extent header chechsum");
815+ return -1;
816+ }
817+
818+ if (memcmp(h->uuid, ehead->uuid, sizeof(ehead->uuid)) != 0) {
819+ error_setg(errp, "wrong vma extent uuid");
820+ return -1;
821+ }
822+
823+ if (ehead->magic != VMA_EXTENT_MAGIC || ehead->reserved1 != 0) {
824+ error_setg(errp, "wrong vma extent header magic");
825+ return -1;
826+ }
827+
828+ int block_count = GUINT16_FROM_BE(ehead->block_count);
829+ int extent_size = VMA_EXTENT_HEADER_SIZE + block_count*VMA_BLOCK_SIZE;
830+
831+ if (buf_pos < extent_size) {
832+ error_setg(errp, "short vma extent (%d < %d)", buf_pos,
833+ extent_size);
834+ return -1;
835+ }
836+
837+ if (restore_extent(vmar, buf, extent_size, vmstate_fd, verbose,
67af0fa4 838+ verify, errp) < 0) {
95259824
WB
839+ return -1;
840+ }
841+
842+ if (buf_pos > extent_size) {
843+ memmove(buf, buf + extent_size, buf_pos - extent_size);
844+ buf_pos = buf_pos - extent_size;
845+ } else {
846+ buf_pos = 0;
847+ }
848+ }
849+
850+ bdrv_drain_all();
851+
852+ int i;
853+ for (i = 1; i < 256; i++) {
854+ VmaRestoreState *rstate = &vmar->rstate[i];
67af0fa4 855+ if (!rstate->target) {
95259824
WB
856+ continue;
857+ }
858+
67af0fa4
WB
859+ if (blk_flush(rstate->target) < 0) {
860+ error_setg(errp, "vma blk_flush %s failed",
95259824
WB
861+ vmar->devinfo[i].devname);
862+ return -1;
863+ }
864+
865+ if (vmar->devinfo[i].size &&
866+ (strcmp(vmar->devinfo[i].devname, "vmstate") != 0)) {
867+ assert(rstate->bitmap);
868+
869+ int64_t cluster_num, end;
870+
871+ end = (vmar->devinfo[i].size + VMA_CLUSTER_SIZE - 1) /
872+ VMA_CLUSTER_SIZE;
873+
874+ for (cluster_num = 0; cluster_num < end; cluster_num++) {
875+ if (!vma_reader_get_bitmap(rstate, cluster_num)) {
876+ error_setg(errp, "detected missing cluster %zd "
877+ "for stream %s", cluster_num,
878+ vmar->devinfo[i].devname);
879+ return -1;
880+ }
881+ }
882+ }
883+ }
884+
67af0fa4
WB
885+ if (verbose) {
886+ if (vmar->clusters_read) {
887+ printf("total bytes read %zd, sparse bytes %zd (%.3g%%)\n",
888+ vmar->clusters_read*VMA_CLUSTER_SIZE,
889+ vmar->zero_cluster_data,
890+ (double)(100.0*vmar->zero_cluster_data)/
891+ (vmar->clusters_read*VMA_CLUSTER_SIZE));
892+
893+ int64_t datasize = vmar->clusters_read*VMA_CLUSTER_SIZE-vmar->zero_cluster_data;
894+ if (datasize) { // this does not make sense for empty files
895+ printf("space reduction due to 4K zero blocks %.3g%%\n",
896+ (double)(100.0*vmar->partial_zero_cluster_data) / datasize);
897+ }
898+ } else {
899+ printf("vma archive contains no image data\n");
900+ }
901+ }
95259824
WB
902+ return ret;
903+}
904+
67af0fa4
WB
905+int vma_reader_restore(VmaReader *vmar, int vmstate_fd, bool verbose,
906+ Error **errp)
907+{
908+ return vma_reader_restore_full(vmar, vmstate_fd, verbose, false, errp);
909+}
910+
911+int vma_reader_verify(VmaReader *vmar, bool verbose, Error **errp)
912+{
913+ guint8 dev_id;
914+
915+ for (dev_id = 1; dev_id < 255; dev_id++) {
916+ if (vma_reader_get_device_info(vmar, dev_id)) {
917+ allocate_rstate(vmar, dev_id, NULL, false);
918+ }
919+ }
920+
921+ return vma_reader_restore_full(vmar, -1, verbose, true, errp);
922+}
923+
95259824
WB
924diff --git a/vma-writer.c b/vma-writer.c
925new file mode 100644
817b7667 926index 0000000000..11d8321ffd
95259824
WB
927--- /dev/null
928+++ b/vma-writer.c
d7f4e01a 929@@ -0,0 +1,790 @@
95259824
WB
930+/*
931+ * VMA: Virtual Machine Archive
932+ *
933+ * Copyright (C) 2012 Proxmox Server Solutions
934+ *
935+ * Authors:
936+ * Dietmar Maurer (dietmar@proxmox.com)
937+ *
938+ * This work is licensed under the terms of the GNU GPL, version 2 or later.
939+ * See the COPYING file in the top-level directory.
940+ *
941+ */
942+
943+#include "qemu/osdep.h"
944+#include <glib.h>
945+#include <uuid/uuid.h>
946+
947+#include "vma.h"
948+#include "block/block.h"
949+#include "monitor/monitor.h"
950+#include "qemu/main-loop.h"
951+#include "qemu/coroutine.h"
952+#include "qemu/cutils.h"
953+
954+#define DEBUG_VMA 0
955+
956+#define DPRINTF(fmt, ...)\
957+ do { if (DEBUG_VMA) { printf("vma: " fmt, ## __VA_ARGS__); } } while (0)
958+
959+#define WRITE_BUFFERS 5
67af0fa4
WB
960+#define HEADER_CLUSTERS 8
961+#define HEADERBUF_SIZE (VMA_CLUSTER_SIZE*HEADER_CLUSTERS)
95259824
WB
962+
963+struct VmaWriter {
964+ int fd;
965+ FILE *cmd;
966+ int status;
967+ char errmsg[8192];
968+ uuid_t uuid;
969+ bool header_written;
970+ bool closed;
971+
972+ /* we always write extents */
67af0fa4 973+ unsigned char *outbuf;
95259824
WB
974+ int outbuf_pos; /* in bytes */
975+ int outbuf_count; /* in VMA_BLOCKS */
976+ uint64_t outbuf_block_info[VMA_BLOCKS_PER_EXTENT];
977+
67af0fa4 978+ unsigned char *headerbuf;
95259824
WB
979+
980+ GChecksum *md5csum;
95259824
WB
981+ CoMutex flush_lock;
982+ Coroutine *co_writer;
983+
984+ /* drive informations */
985+ VmaStreamInfo stream_info[256];
986+ guint stream_count;
987+
988+ guint8 vmstate_stream;
989+ uint32_t vmstate_clusters;
990+
991+ /* header blob table */
992+ char *header_blob_table;
993+ uint32_t header_blob_table_size;
994+ uint32_t header_blob_table_pos;
995+
996+ /* store for config blobs */
997+ uint32_t config_names[VMA_MAX_CONFIGS]; /* offset into blob_buffer table */
998+ uint32_t config_data[VMA_MAX_CONFIGS]; /* offset into blob_buffer table */
999+ uint32_t config_count;
1000+};
1001+
1002+void vma_writer_set_error(VmaWriter *vmaw, const char *fmt, ...)
1003+{
1004+ va_list ap;
1005+
1006+ if (vmaw->status < 0) {
1007+ return;
1008+ }
1009+
1010+ vmaw->status = -1;
1011+
1012+ va_start(ap, fmt);
1013+ g_vsnprintf(vmaw->errmsg, sizeof(vmaw->errmsg), fmt, ap);
1014+ va_end(ap);
1015+
1016+ DPRINTF("vma_writer_set_error: %s\n", vmaw->errmsg);
1017+}
1018+
1019+static uint32_t allocate_header_blob(VmaWriter *vmaw, const char *data,
1020+ size_t len)
1021+{
1022+ if (len > 65535) {
1023+ return 0;
1024+ }
1025+
1026+ if (!vmaw->header_blob_table ||
1027+ (vmaw->header_blob_table_size <
1028+ (vmaw->header_blob_table_pos + len + 2))) {
1029+ int newsize = vmaw->header_blob_table_size + ((len + 2 + 511)/512)*512;
1030+
1031+ vmaw->header_blob_table = g_realloc(vmaw->header_blob_table, newsize);
1032+ memset(vmaw->header_blob_table + vmaw->header_blob_table_size,
1033+ 0, newsize - vmaw->header_blob_table_size);
1034+ vmaw->header_blob_table_size = newsize;
1035+ }
1036+
1037+ uint32_t cpos = vmaw->header_blob_table_pos;
1038+ vmaw->header_blob_table[cpos] = len & 255;
1039+ vmaw->header_blob_table[cpos+1] = (len >> 8) & 255;
1040+ memcpy(vmaw->header_blob_table + cpos + 2, data, len);
1041+ vmaw->header_blob_table_pos += len + 2;
1042+ return cpos;
1043+}
1044+
1045+static uint32_t allocate_header_string(VmaWriter *vmaw, const char *str)
1046+{
1047+ assert(vmaw);
1048+
1049+ size_t len = strlen(str) + 1;
1050+
1051+ return allocate_header_blob(vmaw, str, len);
1052+}
1053+
1054+int vma_writer_add_config(VmaWriter *vmaw, const char *name, gpointer data,
1055+ gsize len)
1056+{
1057+ assert(vmaw);
1058+ assert(!vmaw->header_written);
1059+ assert(vmaw->config_count < VMA_MAX_CONFIGS);
1060+ assert(name);
1061+ assert(data);
95259824
WB
1062+
1063+ gchar *basename = g_path_get_basename(name);
1064+ uint32_t name_ptr = allocate_header_string(vmaw, basename);
1065+ g_free(basename);
1066+
1067+ if (!name_ptr) {
1068+ return -1;
1069+ }
1070+
1071+ uint32_t data_ptr = allocate_header_blob(vmaw, data, len);
1072+ if (!data_ptr) {
1073+ return -1;
1074+ }
1075+
1076+ vmaw->config_names[vmaw->config_count] = name_ptr;
1077+ vmaw->config_data[vmaw->config_count] = data_ptr;
1078+
1079+ vmaw->config_count++;
1080+
1081+ return 0;
1082+}
1083+
1084+int vma_writer_register_stream(VmaWriter *vmaw, const char *devname,
1085+ size_t size)
1086+{
1087+ assert(vmaw);
1088+ assert(devname);
1089+ assert(!vmaw->status);
1090+
1091+ if (vmaw->header_written) {
1092+ vma_writer_set_error(vmaw, "vma_writer_register_stream: header "
1093+ "already written");
1094+ return -1;
1095+ }
1096+
1097+ guint n = vmaw->stream_count + 1;
1098+
1099+ /* we can have dev_ids form 1 to 255 (0 reserved)
1100+ * 255(-1) reseverd for safety
1101+ */
1102+ if (n > 254) {
1103+ vma_writer_set_error(vmaw, "vma_writer_register_stream: "
1104+ "too many drives");
1105+ return -1;
1106+ }
1107+
1108+ if (size <= 0) {
1109+ vma_writer_set_error(vmaw, "vma_writer_register_stream: "
1110+ "got strange size %zd", size);
1111+ return -1;
1112+ }
1113+
1114+ DPRINTF("vma_writer_register_stream %s %zu %d\n", devname, size, n);
1115+
1116+ vmaw->stream_info[n].devname = g_strdup(devname);
1117+ vmaw->stream_info[n].size = size;
1118+
1119+ vmaw->stream_info[n].cluster_count = (size + VMA_CLUSTER_SIZE - 1) /
1120+ VMA_CLUSTER_SIZE;
1121+
1122+ vmaw->stream_count = n;
1123+
1124+ if (strcmp(devname, "vmstate") == 0) {
1125+ vmaw->vmstate_stream = n;
1126+ }
1127+
1128+ return n;
1129+}
1130+
6402d961 1131+static void coroutine_fn yield_until_fd_writable(int fd)
95259824 1132+{
6402d961
TL
1133+ assert(qemu_in_coroutine());
1134+ AioContext *ctx = qemu_get_current_aio_context();
1135+ aio_set_fd_handler(ctx, fd, false, NULL, (IOHandler *)qemu_coroutine_enter,
1136+ NULL, qemu_coroutine_self());
1137+ qemu_coroutine_yield();
1138+ aio_set_fd_handler(ctx, fd, false, NULL, NULL, NULL, NULL);
95259824
WB
1139+}
1140+
1141+static ssize_t coroutine_fn
67af0fa4 1142+vma_queue_write(VmaWriter *vmaw, const void *buf, size_t bytes)
95259824 1143+{
67af0fa4 1144+ DPRINTF("vma_queue_write enter %zd\n", bytes);
95259824 1145+
67af0fa4
WB
1146+ assert(vmaw);
1147+ assert(buf);
1148+ assert(bytes <= VMA_MAX_EXTENT_SIZE);
95259824 1149+
67af0fa4
WB
1150+ size_t done = 0;
1151+ ssize_t ret;
95259824
WB
1152+
1153+ assert(vmaw->co_writer == NULL);
1154+
1155+ vmaw->co_writer = qemu_coroutine_self();
1156+
95259824 1157+ while (done < bytes) {
67af0fa4
WB
1158+ if (vmaw->status < 0) {
1159+ DPRINTF("vma_queue_write detected canceled backup\n");
1160+ done = -1;
1161+ break;
1162+ }
6402d961 1163+ yield_until_fd_writable(vmaw->fd);
95259824
WB
1164+ ret = write(vmaw->fd, buf + done, bytes - done);
1165+ if (ret > 0) {
1166+ done += ret;
67af0fa4 1167+ DPRINTF("vma_queue_write written %zd %zd\n", done, ret);
95259824
WB
1168+ } else if (ret < 0) {
1169+ if (errno == EAGAIN || errno == EWOULDBLOCK) {
67af0fa4
WB
1170+ /* try again */
1171+ } else {
1172+ vma_writer_set_error(vmaw, "vma_queue_write: write error - %s",
95259824
WB
1173+ g_strerror(errno));
1174+ done = -1; /* always return failure for partial writes */
1175+ break;
1176+ }
1177+ } else if (ret == 0) {
1178+ /* should not happen - simply try again */
1179+ }
1180+ }
1181+
95259824
WB
1182+ vmaw->co_writer = NULL;
1183+
67af0fa4 1184+ return (done == bytes) ? bytes : -1;
95259824
WB
1185+}
1186+
1187+VmaWriter *vma_writer_create(const char *filename, uuid_t uuid, Error **errp)
1188+{
1189+ const char *p;
1190+
1191+ assert(sizeof(VmaHeader) == (4096 + 8192));
1192+ assert(G_STRUCT_OFFSET(VmaHeader, config_names) == 2044);
1193+ assert(G_STRUCT_OFFSET(VmaHeader, config_data) == 3068);
1194+ assert(G_STRUCT_OFFSET(VmaHeader, dev_info) == 4096);
1195+ assert(sizeof(VmaExtentHeader) == 512);
1196+
1197+ VmaWriter *vmaw = g_new0(VmaWriter, 1);
1198+ vmaw->fd = -1;
1199+
1200+ vmaw->md5csum = g_checksum_new(G_CHECKSUM_MD5);
1201+ if (!vmaw->md5csum) {
1202+ error_setg(errp, "can't allocate cmsum\n");
1203+ goto err;
1204+ }
1205+
1206+ if (strstart(filename, "exec:", &p)) {
1207+ vmaw->cmd = popen(p, "w");
1208+ if (vmaw->cmd == NULL) {
1209+ error_setg(errp, "can't popen command '%s' - %s\n", p,
1210+ g_strerror(errno));
1211+ goto err;
1212+ }
1213+ vmaw->fd = fileno(vmaw->cmd);
1214+
67af0fa4 1215+ /* try to use O_NONBLOCK */
95259824 1216+ fcntl(vmaw->fd, F_SETFL, fcntl(vmaw->fd, F_GETFL)|O_NONBLOCK);
95259824
WB
1217+
1218+ } else {
1219+ struct stat st;
1220+ int oflags;
1221+ const char *tmp_id_str;
1222+
1223+ if ((stat(filename, &st) == 0) && S_ISFIFO(st.st_mode)) {
67af0fa4 1224+ oflags = O_NONBLOCK|O_WRONLY;
817b7667 1225+ vmaw->fd = qemu_open(filename, oflags, errp);
95259824 1226+ } else if (strstart(filename, "/dev/fdset/", &tmp_id_str)) {
67af0fa4 1227+ oflags = O_NONBLOCK|O_WRONLY;
817b7667 1228+ vmaw->fd = qemu_open(filename, oflags, errp);
95259824 1229+ } else if (strstart(filename, "/dev/fdname/", &tmp_id_str)) {
817b7667 1230+ vmaw->fd = monitor_get_fd(monitor_cur(), tmp_id_str, errp);
95259824
WB
1231+ if (vmaw->fd < 0) {
1232+ goto err;
1233+ }
67af0fa4 1234+ /* try to use O_NONBLOCK */
95259824 1235+ fcntl(vmaw->fd, F_SETFL, fcntl(vmaw->fd, F_GETFL)|O_NONBLOCK);
95259824 1236+ } else {
817b7667
SR
1237+ oflags = O_NONBLOCK|O_DIRECT|O_WRONLY|O_EXCL;
1238+ vmaw->fd = qemu_create(filename, oflags, 0644, errp);
95259824
WB
1239+ }
1240+
1241+ if (vmaw->fd < 0) {
1242+ error_setg(errp, "can't open file %s - %s\n", filename,
1243+ g_strerror(errno));
1244+ goto err;
1245+ }
1246+ }
1247+
1248+ /* we use O_DIRECT, so we need to align IO buffers */
67af0fa4
WB
1249+
1250+ vmaw->outbuf = qemu_memalign(512, VMA_MAX_EXTENT_SIZE);
1251+ vmaw->headerbuf = qemu_memalign(512, HEADERBUF_SIZE);
95259824
WB
1252+
1253+ vmaw->outbuf_count = 0;
1254+ vmaw->outbuf_pos = VMA_EXTENT_HEADER_SIZE;
1255+
1256+ vmaw->header_blob_table_pos = 1; /* start at pos 1 */
1257+
95259824 1258+ qemu_co_mutex_init(&vmaw->flush_lock);
95259824
WB
1259+
1260+ uuid_copy(vmaw->uuid, uuid);
1261+
1262+ return vmaw;
1263+
1264+err:
1265+ if (vmaw) {
1266+ if (vmaw->cmd) {
1267+ pclose(vmaw->cmd);
1268+ } else if (vmaw->fd >= 0) {
1269+ close(vmaw->fd);
1270+ }
1271+
1272+ if (vmaw->md5csum) {
1273+ g_checksum_free(vmaw->md5csum);
1274+ }
1275+
1276+ g_free(vmaw);
1277+ }
1278+
1279+ return NULL;
1280+}
1281+
1282+static int coroutine_fn vma_write_header(VmaWriter *vmaw)
1283+{
1284+ assert(vmaw);
67af0fa4 1285+ unsigned char *buf = vmaw->headerbuf;
95259824
WB
1286+ VmaHeader *head = (VmaHeader *)buf;
1287+
1288+ int i;
1289+
1290+ DPRINTF("VMA WRITE HEADER\n");
1291+
1292+ if (vmaw->status < 0) {
1293+ return vmaw->status;
1294+ }
1295+
67af0fa4 1296+ memset(buf, 0, HEADERBUF_SIZE);
95259824
WB
1297+
1298+ head->magic = VMA_MAGIC;
1299+ head->version = GUINT32_TO_BE(1); /* v1 */
1300+ memcpy(head->uuid, vmaw->uuid, 16);
1301+
1302+ time_t ctime = time(NULL);
1303+ head->ctime = GUINT64_TO_BE(ctime);
1304+
95259824
WB
1305+ for (i = 0; i < VMA_MAX_CONFIGS; i++) {
1306+ head->config_names[i] = GUINT32_TO_BE(vmaw->config_names[i]);
1307+ head->config_data[i] = GUINT32_TO_BE(vmaw->config_data[i]);
1308+ }
1309+
1310+ /* 32 bytes per device (12 used currently) = 8192 bytes max */
1311+ for (i = 1; i <= 254; i++) {
1312+ VmaStreamInfo *si = &vmaw->stream_info[i];
1313+ if (si->size) {
1314+ assert(si->devname);
1315+ uint32_t devname_ptr = allocate_header_string(vmaw, si->devname);
1316+ if (!devname_ptr) {
1317+ return -1;
1318+ }
1319+ head->dev_info[i].devname_ptr = GUINT32_TO_BE(devname_ptr);
1320+ head->dev_info[i].size = GUINT64_TO_BE(si->size);
1321+ }
1322+ }
1323+
1324+ uint32_t header_size = sizeof(VmaHeader) + vmaw->header_blob_table_size;
1325+ head->header_size = GUINT32_TO_BE(header_size);
1326+
67af0fa4 1327+ if (header_size > HEADERBUF_SIZE) {
95259824
WB
1328+ return -1; /* just to be sure */
1329+ }
1330+
1331+ uint32_t blob_buffer_offset = sizeof(VmaHeader);
1332+ memcpy(buf + blob_buffer_offset, vmaw->header_blob_table,
1333+ vmaw->header_blob_table_size);
1334+ head->blob_buffer_offset = GUINT32_TO_BE(blob_buffer_offset);
1335+ head->blob_buffer_size = GUINT32_TO_BE(vmaw->header_blob_table_pos);
1336+
1337+ g_checksum_reset(vmaw->md5csum);
1338+ g_checksum_update(vmaw->md5csum, (const guchar *)buf, header_size);
1339+ gsize csize = 16;
1340+ g_checksum_get_digest(vmaw->md5csum, (guint8 *)(head->md5sum), &csize);
1341+
1342+ return vma_queue_write(vmaw, buf, header_size);
1343+}
1344+
1345+static int coroutine_fn vma_writer_flush(VmaWriter *vmaw)
1346+{
1347+ assert(vmaw);
1348+
1349+ int ret;
1350+ int i;
1351+
1352+ if (vmaw->status < 0) {
1353+ return vmaw->status;
1354+ }
1355+
1356+ if (!vmaw->header_written) {
1357+ vmaw->header_written = true;
1358+ ret = vma_write_header(vmaw);
1359+ if (ret < 0) {
1360+ vma_writer_set_error(vmaw, "vma_writer_flush: write header failed");
1361+ return ret;
1362+ }
1363+ }
1364+
1365+ DPRINTF("VMA WRITE FLUSH %d %d\n", vmaw->outbuf_count, vmaw->outbuf_pos);
1366+
1367+
1368+ VmaExtentHeader *ehead = (VmaExtentHeader *)vmaw->outbuf;
1369+
1370+ ehead->magic = VMA_EXTENT_MAGIC;
1371+ ehead->reserved1 = 0;
1372+
1373+ for (i = 0; i < VMA_BLOCKS_PER_EXTENT; i++) {
1374+ ehead->blockinfo[i] = GUINT64_TO_BE(vmaw->outbuf_block_info[i]);
1375+ }
1376+
1377+ guint16 block_count = (vmaw->outbuf_pos - VMA_EXTENT_HEADER_SIZE) /
1378+ VMA_BLOCK_SIZE;
1379+
1380+ ehead->block_count = GUINT16_TO_BE(block_count);
1381+
1382+ memcpy(ehead->uuid, vmaw->uuid, sizeof(ehead->uuid));
1383+ memset(ehead->md5sum, 0, sizeof(ehead->md5sum));
1384+
1385+ g_checksum_reset(vmaw->md5csum);
1386+ g_checksum_update(vmaw->md5csum, vmaw->outbuf, VMA_EXTENT_HEADER_SIZE);
1387+ gsize csize = 16;
1388+ g_checksum_get_digest(vmaw->md5csum, ehead->md5sum, &csize);
1389+
1390+ int bytes = vmaw->outbuf_pos;
1391+ ret = vma_queue_write(vmaw, vmaw->outbuf, bytes);
1392+ if (ret != bytes) {
1393+ vma_writer_set_error(vmaw, "vma_writer_flush: failed write");
1394+ }
1395+
1396+ vmaw->outbuf_count = 0;
1397+ vmaw->outbuf_pos = VMA_EXTENT_HEADER_SIZE;
1398+
1399+ for (i = 0; i < VMA_BLOCKS_PER_EXTENT; i++) {
1400+ vmaw->outbuf_block_info[i] = 0;
1401+ }
1402+
1403+ return vmaw->status;
1404+}
1405+
1406+static int vma_count_open_streams(VmaWriter *vmaw)
1407+{
1408+ g_assert(vmaw != NULL);
1409+
1410+ int i;
1411+ int open_drives = 0;
1412+ for (i = 0; i <= 255; i++) {
1413+ if (vmaw->stream_info[i].size && !vmaw->stream_info[i].finished) {
1414+ open_drives++;
1415+ }
1416+ }
1417+
1418+ return open_drives;
1419+}
1420+
67af0fa4
WB
1421+
1422+/**
1423+ * You need to call this if the vma archive does not contain
1424+ * any data stream.
1425+ */
1426+int coroutine_fn
1427+vma_writer_flush_output(VmaWriter *vmaw)
1428+{
1429+ qemu_co_mutex_lock(&vmaw->flush_lock);
1430+ int ret = vma_writer_flush(vmaw);
1431+ qemu_co_mutex_unlock(&vmaw->flush_lock);
1432+ if (ret < 0) {
1433+ vma_writer_set_error(vmaw, "vma_writer_flush_header failed");
1434+ }
1435+ return ret;
1436+}
1437+
95259824
WB
1438+/**
1439+ * all jobs should call this when there is no more data
1440+ * Returns: number of remaining stream (0 ==> finished)
1441+ */
1442+int coroutine_fn
1443+vma_writer_close_stream(VmaWriter *vmaw, uint8_t dev_id)
1444+{
1445+ g_assert(vmaw != NULL);
1446+
1447+ DPRINTF("vma_writer_set_status %d\n", dev_id);
1448+ if (!vmaw->stream_info[dev_id].size) {
1449+ vma_writer_set_error(vmaw, "vma_writer_close_stream: "
1450+ "no such stream %d", dev_id);
1451+ return -1;
1452+ }
1453+ if (vmaw->stream_info[dev_id].finished) {
1454+ vma_writer_set_error(vmaw, "vma_writer_close_stream: "
1455+ "stream already closed %d", dev_id);
1456+ return -1;
1457+ }
1458+
1459+ vmaw->stream_info[dev_id].finished = true;
1460+
1461+ int open_drives = vma_count_open_streams(vmaw);
1462+
1463+ if (open_drives <= 0) {
1464+ DPRINTF("vma_writer_set_status all drives completed\n");
67af0fa4 1465+ vma_writer_flush_output(vmaw);
95259824
WB
1466+ }
1467+
1468+ return open_drives;
1469+}
1470+
1471+int vma_writer_get_status(VmaWriter *vmaw, VmaStatus *status)
1472+{
1473+ int i;
1474+
1475+ g_assert(vmaw != NULL);
1476+
1477+ if (status) {
1478+ status->status = vmaw->status;
1479+ g_strlcpy(status->errmsg, vmaw->errmsg, sizeof(status->errmsg));
1480+ for (i = 0; i <= 255; i++) {
1481+ status->stream_info[i] = vmaw->stream_info[i];
1482+ }
1483+
1484+ uuid_unparse_lower(vmaw->uuid, status->uuid_str);
1485+ }
1486+
1487+ status->closed = vmaw->closed;
1488+
1489+ return vmaw->status;
1490+}
1491+
1492+static int vma_writer_get_buffer(VmaWriter *vmaw)
1493+{
1494+ int ret = 0;
1495+
1496+ qemu_co_mutex_lock(&vmaw->flush_lock);
1497+
1498+ /* wait until buffer is available */
1499+ while (vmaw->outbuf_count >= (VMA_BLOCKS_PER_EXTENT - 1)) {
1500+ ret = vma_writer_flush(vmaw);
1501+ if (ret < 0) {
1502+ vma_writer_set_error(vmaw, "vma_writer_get_buffer: flush failed");
1503+ break;
1504+ }
1505+ }
1506+
1507+ qemu_co_mutex_unlock(&vmaw->flush_lock);
1508+
1509+ return ret;
1510+}
1511+
1512+
1513+int64_t coroutine_fn
1514+vma_writer_write(VmaWriter *vmaw, uint8_t dev_id, int64_t cluster_num,
6838f038 1515+ const unsigned char *buf, size_t *zero_bytes)
95259824
WB
1516+{
1517+ g_assert(vmaw != NULL);
1518+ g_assert(zero_bytes != NULL);
1519+
1520+ *zero_bytes = 0;
1521+
1522+ if (vmaw->status < 0) {
1523+ return vmaw->status;
1524+ }
1525+
1526+ if (!dev_id || !vmaw->stream_info[dev_id].size) {
1527+ vma_writer_set_error(vmaw, "vma_writer_write: "
1528+ "no such stream %d", dev_id);
1529+ return -1;
1530+ }
1531+
1532+ if (vmaw->stream_info[dev_id].finished) {
1533+ vma_writer_set_error(vmaw, "vma_writer_write: "
1534+ "stream already closed %d", dev_id);
1535+ return -1;
1536+ }
1537+
1538+
1539+ if (cluster_num >= (((uint64_t)1)<<32)) {
1540+ vma_writer_set_error(vmaw, "vma_writer_write: "
1541+ "cluster number out of range");
1542+ return -1;
1543+ }
1544+
1545+ if (dev_id == vmaw->vmstate_stream) {
1546+ if (cluster_num != vmaw->vmstate_clusters) {
1547+ vma_writer_set_error(vmaw, "vma_writer_write: "
1548+ "non sequential vmstate write");
1549+ }
1550+ vmaw->vmstate_clusters++;
1551+ } else if (cluster_num >= vmaw->stream_info[dev_id].cluster_count) {
1552+ vma_writer_set_error(vmaw, "vma_writer_write: cluster number too big");
1553+ return -1;
1554+ }
1555+
1556+ /* wait until buffer is available */
1557+ if (vma_writer_get_buffer(vmaw) < 0) {
1558+ vma_writer_set_error(vmaw, "vma_writer_write: "
1559+ "vma_writer_get_buffer failed");
1560+ return -1;
1561+ }
1562+
1563+ DPRINTF("VMA WRITE %d %zd\n", dev_id, cluster_num);
1564+
d7f4e01a 1565+ uint64_t dev_size = vmaw->stream_info[dev_id].size;
95259824
WB
1566+ uint16_t mask = 0;
1567+
1568+ if (buf) {
1569+ int i;
1570+ int bit = 1;
d7f4e01a 1571+ uint64_t byte_offset = cluster_num * VMA_CLUSTER_SIZE;
95259824 1572+ for (i = 0; i < 16; i++) {
6838f038 1573+ const unsigned char *vmablock = buf + (i*VMA_BLOCK_SIZE);
d7f4e01a
TL
1574+
1575+ // Note: If the source is not 64k-aligned, we might reach 4k blocks
1576+ // after the end of the device. Always mark these as zero in the
1577+ // mask, so the restore handles them correctly.
1578+ if (byte_offset < dev_size &&
1579+ !buffer_is_zero(vmablock, VMA_BLOCK_SIZE))
1580+ {
95259824
WB
1581+ mask |= bit;
1582+ memcpy(vmaw->outbuf + vmaw->outbuf_pos, vmablock,
1583+ VMA_BLOCK_SIZE);
d7f4e01a
TL
1584+
1585+ // prevent memory leakage on unaligned last block
1586+ if (byte_offset + VMA_BLOCK_SIZE > dev_size) {
1587+ uint64_t real_data_in_block = dev_size - byte_offset;
1588+ memset(vmaw->outbuf + vmaw->outbuf_pos + real_data_in_block,
1589+ 0, VMA_BLOCK_SIZE - real_data_in_block);
1590+ }
1591+
95259824
WB
1592+ vmaw->outbuf_pos += VMA_BLOCK_SIZE;
1593+ } else {
1594+ DPRINTF("VMA WRITE %zd ZERO BLOCK %d\n", cluster_num, i);
1595+ vmaw->stream_info[dev_id].zero_bytes += VMA_BLOCK_SIZE;
1596+ *zero_bytes += VMA_BLOCK_SIZE;
1597+ }
1598+
d7f4e01a 1599+ byte_offset += VMA_BLOCK_SIZE;
95259824
WB
1600+ bit = bit << 1;
1601+ }
1602+ } else {
1603+ DPRINTF("VMA WRITE %zd ZERO CLUSTER\n", cluster_num);
1604+ vmaw->stream_info[dev_id].zero_bytes += VMA_CLUSTER_SIZE;
1605+ *zero_bytes += VMA_CLUSTER_SIZE;
1606+ }
1607+
1608+ uint64_t block_info = ((uint64_t)mask) << (32+16);
1609+ block_info |= ((uint64_t)dev_id) << 32;
1610+ block_info |= (cluster_num & 0xffffffff);
1611+ vmaw->outbuf_block_info[vmaw->outbuf_count] = block_info;
1612+
1613+ DPRINTF("VMA WRITE MASK %zd %zx\n", cluster_num, block_info);
1614+
1615+ vmaw->outbuf_count++;
1616+
1617+ /** NOTE: We allways write whole clusters, but we correctly set
1618+ * transferred bytes. So transferred == size when when everything
1619+ * went OK.
1620+ */
1621+ size_t transferred = VMA_CLUSTER_SIZE;
1622+
1623+ if (dev_id != vmaw->vmstate_stream) {
1624+ uint64_t last = (cluster_num + 1) * VMA_CLUSTER_SIZE;
d7f4e01a
TL
1625+ if (last > dev_size) {
1626+ uint64_t diff = last - dev_size;
95259824
WB
1627+ if (diff >= VMA_CLUSTER_SIZE) {
1628+ vma_writer_set_error(vmaw, "vma_writer_write: "
1629+ "read after last cluster");
1630+ return -1;
1631+ }
1632+ transferred -= diff;
1633+ }
1634+ }
1635+
1636+ vmaw->stream_info[dev_id].transferred += transferred;
1637+
1638+ return transferred;
1639+}
1640+
67af0fa4
WB
1641+void vma_writer_error_propagate(VmaWriter *vmaw, Error **errp)
1642+{
1643+ if (vmaw->status < 0 && *errp == NULL) {
1644+ error_setg(errp, "%s", vmaw->errmsg);
1645+ }
1646+}
1647+
95259824
WB
1648+int vma_writer_close(VmaWriter *vmaw, Error **errp)
1649+{
1650+ g_assert(vmaw != NULL);
1651+
1652+ int i;
1653+
6402d961 1654+ qemu_co_mutex_lock(&vmaw->flush_lock); // wait for pending writes
95259824 1655+
67af0fa4
WB
1656+ assert(vmaw->co_writer == NULL);
1657+
95259824
WB
1658+ if (vmaw->cmd) {
1659+ if (pclose(vmaw->cmd) < 0) {
1660+ vma_writer_set_error(vmaw, "vma_writer_close: "
1661+ "pclose failed - %s", g_strerror(errno));
1662+ }
1663+ } else {
1664+ if (close(vmaw->fd) < 0) {
1665+ vma_writer_set_error(vmaw, "vma_writer_close: "
1666+ "close failed - %s", g_strerror(errno));
1667+ }
1668+ }
1669+
1670+ for (i = 0; i <= 255; i++) {
1671+ VmaStreamInfo *si = &vmaw->stream_info[i];
1672+ if (si->size) {
1673+ if (!si->finished) {
1674+ vma_writer_set_error(vmaw, "vma_writer_close: "
1675+ "detected open stream '%s'", si->devname);
1676+ } else if ((si->transferred != si->size) &&
1677+ (i != vmaw->vmstate_stream)) {
1678+ vma_writer_set_error(vmaw, "vma_writer_close: "
1679+ "incomplete stream '%s' (%zd != %zd)",
1680+ si->devname, si->transferred, si->size);
1681+ }
1682+ }
1683+ }
1684+
1685+ for (i = 0; i <= 255; i++) {
1686+ vmaw->stream_info[i].finished = 1; /* mark as closed */
1687+ }
1688+
1689+ vmaw->closed = 1;
1690+
1691+ if (vmaw->status < 0 && *errp == NULL) {
1692+ error_setg(errp, "%s", vmaw->errmsg);
1693+ }
1694+
6402d961
TL
1695+ qemu_co_mutex_unlock(&vmaw->flush_lock);
1696+
95259824
WB
1697+ return vmaw->status;
1698+}
1699+
1700+void vma_writer_destroy(VmaWriter *vmaw)
1701+{
1702+ assert(vmaw);
1703+
1704+ int i;
1705+
1706+ for (i = 0; i <= 255; i++) {
1707+ if (vmaw->stream_info[i].devname) {
1708+ g_free(vmaw->stream_info[i].devname);
1709+ }
1710+ }
1711+
1712+ if (vmaw->md5csum) {
1713+ g_checksum_free(vmaw->md5csum);
1714+ }
1715+
d7f4e01a
TL
1716+ qemu_vfree(vmaw->headerbuf);
1717+ qemu_vfree(vmaw->outbuf);
95259824
WB
1718+ g_free(vmaw);
1719+}
1720diff --git a/vma.c b/vma.c
1721new file mode 100644
2653a5f0 1722index 0000000000..89440733b1
95259824
WB
1723--- /dev/null
1724+++ b/vma.c
2653a5f0 1725@@ -0,0 +1,849 @@
95259824
WB
1726+/*
1727+ * VMA: Virtual Machine Archive
1728+ *
1729+ * Copyright (C) 2012-2013 Proxmox Server Solutions
1730+ *
1731+ * Authors:
1732+ * Dietmar Maurer (dietmar@proxmox.com)
1733+ *
1734+ * This work is licensed under the terms of the GNU GPL, version 2 or later.
1735+ * See the COPYING file in the top-level directory.
1736+ *
1737+ */
1738+
1739+#include "qemu/osdep.h"
1740+#include <glib.h>
1741+
1742+#include "vma.h"
1743+#include "qemu-common.h"
6402d961 1744+#include "qemu/module.h"
95259824
WB
1745+#include "qemu/error-report.h"
1746+#include "qemu/main-loop.h"
6402d961
TL
1747+#include "qemu/cutils.h"
1748+#include "qapi/qmp/qdict.h"
67af0fa4 1749+#include "sysemu/block-backend.h"
95259824
WB
1750+
1751+static void help(void)
1752+{
1753+ const char *help_msg =
1754+ "usage: vma command [command options]\n"
1755+ "\n"
1756+ "vma list <filename>\n"
67af0fa4
WB
1757+ "vma config <filename> [-c config]\n"
1758+ "vma create <filename> [-c config] pathname ...\n"
95259824 1759+ "vma extract <filename> [-r <fifo>] <targetdir>\n"
67af0fa4 1760+ "vma verify <filename> [-v]\n"
95259824
WB
1761+ ;
1762+
1763+ printf("%s", help_msg);
1764+ exit(1);
1765+}
1766+
1767+static const char *extract_devname(const char *path, char **devname, int index)
1768+{
1769+ assert(path);
1770+
1771+ const char *sep = strchr(path, '=');
1772+
1773+ if (sep) {
1774+ *devname = g_strndup(path, sep - path);
1775+ path = sep + 1;
1776+ } else {
1777+ if (index >= 0) {
1778+ *devname = g_strdup_printf("disk%d", index);
1779+ } else {
1780+ *devname = NULL;
1781+ }
1782+ }
1783+
1784+ return path;
1785+}
1786+
1787+static void print_content(VmaReader *vmar)
1788+{
1789+ assert(vmar);
1790+
1791+ VmaHeader *head = vma_reader_get_header(vmar);
1792+
1793+ GList *l = vma_reader_get_config_data(vmar);
1794+ while (l && l->data) {
1795+ VmaConfigData *cdata = (VmaConfigData *)l->data;
1796+ l = g_list_next(l);
1797+ printf("CFG: size: %d name: %s\n", cdata->len, cdata->name);
1798+ }
1799+
1800+ int i;
1801+ VmaDeviceInfo *di;
1802+ for (i = 1; i < 255; i++) {
1803+ di = vma_reader_get_device_info(vmar, i);
1804+ if (di) {
1805+ if (strcmp(di->devname, "vmstate") == 0) {
1806+ printf("VMSTATE: dev_id=%d memory: %zd\n", i, di->size);
1807+ } else {
1808+ printf("DEV: dev_id=%d size: %zd devname: %s\n",
1809+ i, di->size, di->devname);
1810+ }
1811+ }
1812+ }
1813+ /* ctime is the last entry we print */
1814+ printf("CTIME: %s", ctime(&head->ctime));
1815+ fflush(stdout);
1816+}
1817+
1818+static int list_content(int argc, char **argv)
1819+{
1820+ int c, ret = 0;
1821+ const char *filename;
1822+
1823+ for (;;) {
1824+ c = getopt(argc, argv, "h");
1825+ if (c == -1) {
1826+ break;
1827+ }
1828+ switch (c) {
1829+ case '?':
1830+ case 'h':
1831+ help();
1832+ break;
1833+ default:
1834+ g_assert_not_reached();
1835+ }
1836+ }
1837+
1838+ /* Get the filename */
1839+ if ((optind + 1) != argc) {
1840+ help();
1841+ }
1842+ filename = argv[optind++];
1843+
1844+ Error *errp = NULL;
1845+ VmaReader *vmar = vma_reader_create(filename, &errp);
1846+
1847+ if (!vmar) {
1848+ g_error("%s", error_get_pretty(errp));
1849+ }
1850+
1851+ print_content(vmar);
1852+
1853+ vma_reader_destroy(vmar);
1854+
1855+ return ret;
1856+}
1857+
1858+typedef struct RestoreMap {
1859+ char *devname;
1860+ char *path;
67af0fa4 1861+ char *format;
6402d961
TL
1862+ uint64_t throttling_bps;
1863+ char *throttling_group;
1864+ char *cache;
95259824
WB
1865+ bool write_zero;
1866+} RestoreMap;
1867+
6402d961
TL
1868+static bool try_parse_option(char **line, const char *optname, char **out, const char *inbuf) {
1869+ size_t optlen = strlen(optname);
1870+ if (strncmp(*line, optname, optlen) != 0 || (*line)[optlen] != '=') {
1871+ return false;
1872+ }
1873+ if (*out) {
1874+ g_error("read map failed - duplicate value for option '%s'", optname);
1875+ }
1876+ char *value = (*line) + optlen + 1; /* including a '=' */
1877+ char *colon = strchr(value, ':');
1878+ if (!colon) {
1879+ g_error("read map failed - option '%s' not terminated ('%s')",
1880+ optname, inbuf);
1881+ }
1882+ *line = colon+1;
1883+ *out = g_strndup(value, colon - value);
1884+ return true;
1885+}
1886+
1887+static uint64_t verify_u64(const char *text) {
1888+ uint64_t value;
1889+ const char *endptr = NULL;
1890+ if (qemu_strtou64(text, &endptr, 0, &value) != 0 || !endptr || *endptr) {
1891+ g_error("read map failed - not a number: %s", text);
1892+ }
1893+ return value;
1894+}
1895+
95259824
WB
1896+static int extract_content(int argc, char **argv)
1897+{
1898+ int c, ret = 0;
1899+ int verbose = 0;
1900+ const char *filename;
1901+ const char *dirname;
1902+ const char *readmap = NULL;
1903+
1904+ for (;;) {
1905+ c = getopt(argc, argv, "hvr:");
1906+ if (c == -1) {
1907+ break;
1908+ }
1909+ switch (c) {
1910+ case '?':
1911+ case 'h':
1912+ help();
1913+ break;
1914+ case 'r':
1915+ readmap = optarg;
1916+ break;
1917+ case 'v':
1918+ verbose = 1;
1919+ break;
1920+ default:
1921+ help();
1922+ }
1923+ }
1924+
1925+ /* Get the filename */
1926+ if ((optind + 2) != argc) {
1927+ help();
1928+ }
1929+ filename = argv[optind++];
1930+ dirname = argv[optind++];
1931+
1932+ Error *errp = NULL;
1933+ VmaReader *vmar = vma_reader_create(filename, &errp);
1934+
1935+ if (!vmar) {
1936+ g_error("%s", error_get_pretty(errp));
1937+ }
1938+
1939+ if (mkdir(dirname, 0777) < 0) {
1940+ g_error("unable to create target directory %s - %s",
1941+ dirname, g_strerror(errno));
1942+ }
1943+
1944+ GList *l = vma_reader_get_config_data(vmar);
1945+ while (l && l->data) {
1946+ VmaConfigData *cdata = (VmaConfigData *)l->data;
1947+ l = g_list_next(l);
1948+ char *cfgfn = g_strdup_printf("%s/%s", dirname, cdata->name);
1949+ GError *err = NULL;
1950+ if (!g_file_set_contents(cfgfn, (gchar *)cdata->data, cdata->len,
1951+ &err)) {
1952+ g_error("unable to write file: %s", err->message);
1953+ }
1954+ }
1955+
1956+ GHashTable *devmap = g_hash_table_new(g_str_hash, g_str_equal);
1957+
1958+ if (readmap) {
1959+ print_content(vmar);
1960+
1961+ FILE *map = fopen(readmap, "r");
1962+ if (!map) {
1963+ g_error("unable to open fifo %s - %s", readmap, g_strerror(errno));
1964+ }
1965+
1966+ while (1) {
1967+ char inbuf[8192];
1968+ char *line = fgets(inbuf, sizeof(inbuf), map);
6402d961
TL
1969+ char *format = NULL;
1970+ char *bps = NULL;
1971+ char *group = NULL;
1972+ char *cache = NULL;
95259824
WB
1973+ if (!line || line[0] == '\0' || !strcmp(line, "done\n")) {
1974+ break;
1975+ }
1976+ int len = strlen(line);
1977+ if (line[len - 1] == '\n') {
1978+ line[len - 1] = '\0';
1979+ if (len == 1) {
1980+ break;
1981+ }
1982+ }
1983+
6402d961
TL
1984+ while (1) {
1985+ if (!try_parse_option(&line, "format", &format, inbuf) &&
1986+ !try_parse_option(&line, "throttling.bps", &bps, inbuf) &&
1987+ !try_parse_option(&line, "throttling.group", &group, inbuf) &&
1988+ !try_parse_option(&line, "cache", &cache, inbuf))
1989+ {
1990+ break;
67af0fa4 1991+ }
6402d961
TL
1992+ }
1993+
1994+ uint64_t bps_value = 0;
1995+ if (bps) {
1996+ bps_value = verify_u64(bps);
1997+ g_free(bps);
67af0fa4
WB
1998+ }
1999+
95259824
WB
2000+ const char *path;
2001+ bool write_zero;
2002+ if (line[0] == '0' && line[1] == ':') {
67af0fa4 2003+ path = line + 2;
95259824
WB
2004+ write_zero = false;
2005+ } else if (line[0] == '1' && line[1] == ':') {
67af0fa4 2006+ path = line + 2;
95259824
WB
2007+ write_zero = true;
2008+ } else {
2009+ g_error("read map failed - parse error ('%s')", inbuf);
2010+ }
2011+
2012+ char *devname = NULL;
2013+ path = extract_devname(path, &devname, -1);
2014+ if (!devname) {
2015+ g_error("read map failed - no dev name specified ('%s')",
2016+ inbuf);
2017+ }
2018+
2019+ RestoreMap *map = g_new0(RestoreMap, 1);
2020+ map->devname = g_strdup(devname);
2021+ map->path = g_strdup(path);
67af0fa4 2022+ map->format = format;
6402d961
TL
2023+ map->throttling_bps = bps_value;
2024+ map->throttling_group = group;
2025+ map->cache = cache;
95259824
WB
2026+ map->write_zero = write_zero;
2027+
2028+ g_hash_table_insert(devmap, map->devname, map);
2029+
2030+ };
2031+ }
2032+
2033+ int i;
2034+ int vmstate_fd = -1;
2035+ guint8 vmstate_stream = 0;
2036+
2037+ for (i = 1; i < 255; i++) {
2038+ VmaDeviceInfo *di = vma_reader_get_device_info(vmar, i);
2039+ if (di && (strcmp(di->devname, "vmstate") == 0)) {
2040+ vmstate_stream = i;
2041+ char *statefn = g_strdup_printf("%s/vmstate.bin", dirname);
2042+ vmstate_fd = open(statefn, O_WRONLY|O_CREAT|O_EXCL, 0644);
2043+ if (vmstate_fd < 0) {
2044+ g_error("create vmstate file '%s' failed - %s", statefn,
2045+ g_strerror(errno));
2046+ }
2047+ g_free(statefn);
2048+ } else if (di) {
2049+ char *devfn = NULL;
67af0fa4 2050+ const char *format = NULL;
6402d961
TL
2051+ uint64_t throttling_bps = 0;
2052+ const char *throttling_group = NULL;
2053+ const char *cache = NULL;
2054+ int flags = BDRV_O_RDWR;
95259824
WB
2055+ bool write_zero = true;
2056+
2653a5f0
TL
2057+ BlockBackend *blk = NULL;
2058+
95259824
WB
2059+ if (readmap) {
2060+ RestoreMap *map;
2061+ map = (RestoreMap *)g_hash_table_lookup(devmap, di->devname);
2062+ if (map == NULL) {
2063+ g_error("no device name mapping for %s", di->devname);
2064+ }
2065+ devfn = map->path;
67af0fa4 2066+ format = map->format;
6402d961
TL
2067+ throttling_bps = map->throttling_bps;
2068+ throttling_group = map->throttling_group;
2069+ cache = map->cache;
95259824
WB
2070+ write_zero = map->write_zero;
2071+ } else {
2072+ devfn = g_strdup_printf("%s/tmp-disk-%s.raw",
2073+ dirname, di->devname);
2074+ printf("DEVINFO %s %zd\n", devfn, di->size);
2075+
2076+ bdrv_img_create(devfn, "raw", NULL, NULL, NULL, di->size,
6838f038 2077+ flags, true, &errp);
95259824
WB
2078+ if (errp) {
2079+ g_error("can't create file %s: %s", devfn,
2080+ error_get_pretty(errp));
2081+ }
2082+
2083+ /* Note: we created an empty file above, so there is no
2084+ * need to write zeroes (so we generate a sparse file)
2085+ */
2086+ write_zero = false;
2087+ }
2088+
67af0fa4
WB
2089+ size_t devlen = strlen(devfn);
2090+ QDict *options = NULL;
6402d961 2091+ bool writethrough;
67af0fa4
WB
2092+ if (format) {
2093+ /* explicit format from commandline */
2094+ options = qdict_new();
6402d961 2095+ qdict_put_str(options, "driver", format);
67af0fa4
WB
2096+ } else if ((devlen > 4 && strcmp(devfn+devlen-4, ".raw") == 0) ||
2097+ strncmp(devfn, "/dev/", 5) == 0)
2098+ {
2099+ /* This part is now deprecated for PVE as well (just as qemu
2100+ * deprecated not specifying an explicit raw format, too.
2101+ */
2102+ /* explicit raw format */
2103+ options = qdict_new();
6402d961 2104+ qdict_put_str(options, "driver", "raw");
67af0fa4 2105+ }
6402d961
TL
2106+ if (cache && bdrv_parse_cache_mode(cache, &flags, &writethrough)) {
2107+ g_error("invalid cache option: %s\n", cache);
2108+ }
67af0fa4
WB
2109+
2110+ if (errp || !(blk = blk_new_open(devfn, NULL, options, flags, &errp))) {
95259824
WB
2111+ g_error("can't open file %s - %s", devfn,
2112+ error_get_pretty(errp));
2113+ }
67af0fa4 2114+
6402d961
TL
2115+ if (cache) {
2116+ blk_set_enable_write_cache(blk, !writethrough);
2117+ }
2118+
2119+ if (throttling_group) {
2120+ blk_io_limits_enable(blk, throttling_group);
2121+ }
2122+
2123+ if (throttling_bps) {
2124+ if (!throttling_group) {
2125+ blk_io_limits_enable(blk, devfn);
2126+ }
2127+
2128+ ThrottleConfig cfg;
2129+ throttle_config_init(&cfg);
2130+ cfg.buckets[THROTTLE_BPS_WRITE].avg = throttling_bps;
2131+ Error *err = NULL;
2132+ if (!throttle_is_valid(&cfg, &err)) {
2133+ error_report_err(err);
2134+ g_error("failed to apply throttling");
2135+ }
2136+ blk_set_io_limits(blk, &cfg);
2137+ }
2138+
67af0fa4 2139+ if (vma_reader_register_bs(vmar, i, blk, write_zero, &errp) < 0) {
95259824
WB
2140+ g_error("%s", error_get_pretty(errp));
2141+ }
2142+
2143+ if (!readmap) {
2144+ g_free(devfn);
2145+ }
2146+ }
2147+ }
2148+
2149+ if (vma_reader_restore(vmar, vmstate_fd, verbose, &errp) < 0) {
2150+ g_error("restore failed - %s", error_get_pretty(errp));
2151+ }
2152+
2153+ if (!readmap) {
2154+ for (i = 1; i < 255; i++) {
2155+ VmaDeviceInfo *di = vma_reader_get_device_info(vmar, i);
2156+ if (di && (i != vmstate_stream)) {
2157+ char *tmpfn = g_strdup_printf("%s/tmp-disk-%s.raw",
2158+ dirname, di->devname);
2159+ char *fn = g_strdup_printf("%s/disk-%s.raw",
2160+ dirname, di->devname);
2161+ if (rename(tmpfn, fn) != 0) {
2162+ g_error("rename %s to %s failed - %s",
2163+ tmpfn, fn, g_strerror(errno));
2164+ }
2165+ }
2166+ }
2167+ }
2168+
2169+ vma_reader_destroy(vmar);
2170+
67af0fa4
WB
2171+ bdrv_close_all();
2172+
2173+ return ret;
2174+}
2175+
2176+static int verify_content(int argc, char **argv)
2177+{
2178+ int c, ret = 0;
2179+ int verbose = 0;
2180+ const char *filename;
2181+
2182+ for (;;) {
2183+ c = getopt(argc, argv, "hv");
2184+ if (c == -1) {
2185+ break;
2186+ }
2187+ switch (c) {
2188+ case '?':
2189+ case 'h':
2190+ help();
2191+ break;
2192+ case 'v':
2193+ verbose = 1;
2194+ break;
2195+ default:
2196+ help();
2197+ }
2198+ }
2199+
2200+ /* Get the filename */
2201+ if ((optind + 1) != argc) {
2202+ help();
2203+ }
2204+ filename = argv[optind++];
2205+
2206+ Error *errp = NULL;
2207+ VmaReader *vmar = vma_reader_create(filename, &errp);
2208+
2209+ if (!vmar) {
2210+ g_error("%s", error_get_pretty(errp));
2211+ }
2212+
2213+ if (verbose) {
2214+ print_content(vmar);
2215+ }
2216+
2217+ if (vma_reader_verify(vmar, verbose, &errp) < 0) {
2218+ g_error("verify failed - %s", error_get_pretty(errp));
2219+ }
2220+
2221+ vma_reader_destroy(vmar);
2222+
95259824
WB
2223+ bdrv_close_all();
2224+
2225+ return ret;
2226+}
2227+
2228+typedef struct BackupJob {
67af0fa4 2229+ BlockBackend *target;
95259824
WB
2230+ int64_t len;
2231+ VmaWriter *vmaw;
2232+ uint8_t dev_id;
2233+} BackupJob;
2234+
2235+#define BACKUP_SECTORS_PER_CLUSTER (VMA_CLUSTER_SIZE / BDRV_SECTOR_SIZE)
2236+
67af0fa4
WB
2237+static void coroutine_fn backup_run_empty(void *opaque)
2238+{
2239+ VmaWriter *vmaw = (VmaWriter *)opaque;
2240+
2241+ vma_writer_flush_output(vmaw);
2242+
2243+ Error *err = NULL;
2244+ if (vma_writer_close(vmaw, &err) != 0) {
2245+ g_warning("vma_writer_close failed %s", error_get_pretty(err));
2246+ }
2247+}
2248+
95259824
WB
2249+static void coroutine_fn backup_run(void *opaque)
2250+{
2251+ BackupJob *job = (BackupJob *)opaque;
2252+ struct iovec iov;
2253+ QEMUIOVector qiov;
2254+
2255+ int64_t start, end;
2256+ int ret = 0;
2257+
67af0fa4 2258+ unsigned char *buf = blk_blockalign(job->target, VMA_CLUSTER_SIZE);
95259824
WB
2259+
2260+ start = 0;
2261+ end = DIV_ROUND_UP(job->len / BDRV_SECTOR_SIZE,
2262+ BACKUP_SECTORS_PER_CLUSTER);
2263+
2264+ for (; start < end; start++) {
2265+ iov.iov_base = buf;
2266+ iov.iov_len = VMA_CLUSTER_SIZE;
2267+ qemu_iovec_init_external(&qiov, &iov, 1);
2268+
67af0fa4
WB
2269+ ret = blk_co_preadv(job->target, start * VMA_CLUSTER_SIZE,
2270+ VMA_CLUSTER_SIZE, &qiov, 0);
95259824
WB
2271+ if (ret < 0) {
2272+ vma_writer_set_error(job->vmaw, "read error", -1);
2273+ goto out;
2274+ }
2275+
2276+ size_t zb = 0;
2277+ if (vma_writer_write(job->vmaw, job->dev_id, start, buf, &zb) < 0) {
2278+ vma_writer_set_error(job->vmaw, "backup_dump_cb vma_writer_write failed", -1);
2279+ goto out;
2280+ }
2281+ }
2282+
2283+
2284+out:
2285+ if (vma_writer_close_stream(job->vmaw, job->dev_id) <= 0) {
2286+ Error *err = NULL;
2287+ if (vma_writer_close(job->vmaw, &err) != 0) {
2288+ g_warning("vma_writer_close failed %s", error_get_pretty(err));
2289+ }
2290+ }
d7f4e01a 2291+ qemu_vfree(buf);
95259824
WB
2292+}
2293+
2294+static int create_archive(int argc, char **argv)
2295+{
2296+ int i, c;
2297+ int verbose = 0;
2298+ const char *archivename;
2bf61c3e 2299+ GList *backup_coroutines = NULL;
95259824
WB
2300+ GList *config_files = NULL;
2301+
2302+ for (;;) {
2303+ c = getopt(argc, argv, "hvc:");
2304+ if (c == -1) {
2305+ break;
2306+ }
2307+ switch (c) {
2308+ case '?':
2309+ case 'h':
2310+ help();
2311+ break;
2312+ case 'c':
2313+ config_files = g_list_append(config_files, optarg);
2314+ break;
2315+ case 'v':
2316+ verbose = 1;
2317+ break;
2318+ default:
2319+ g_assert_not_reached();
2320+ }
2321+ }
2322+
2323+
67af0fa4
WB
2324+ /* make sure we an archive name */
2325+ if ((optind + 1) > argc) {
95259824
WB
2326+ help();
2327+ }
2328+
2329+ archivename = argv[optind++];
2330+
2331+ uuid_t uuid;
2332+ uuid_generate(uuid);
2333+
2334+ Error *local_err = NULL;
2335+ VmaWriter *vmaw = vma_writer_create(archivename, uuid, &local_err);
2336+
2337+ if (vmaw == NULL) {
2338+ g_error("%s", error_get_pretty(local_err));
2339+ }
2340+
2341+ GList *l = config_files;
2342+ while (l && l->data) {
2343+ char *name = l->data;
2344+ char *cdata = NULL;
2345+ gsize clen = 0;
2346+ GError *err = NULL;
2347+ if (!g_file_get_contents(name, &cdata, &clen, &err)) {
2348+ unlink(archivename);
2349+ g_error("Unable to read file: %s", err->message);
2350+ }
2351+
2352+ if (vma_writer_add_config(vmaw, name, cdata, clen) != 0) {
2353+ unlink(archivename);
2354+ g_error("Unable to append config data %s (len = %zd)",
2355+ name, clen);
2356+ }
2357+ l = g_list_next(l);
2358+ }
2359+
67af0fa4 2360+ int devcount = 0;
95259824
WB
2361+ while (optind < argc) {
2362+ const char *path = argv[optind++];
2363+ char *devname = NULL;
67af0fa4 2364+ path = extract_devname(path, &devname, devcount++);
95259824
WB
2365+
2366+ Error *errp = NULL;
67af0fa4 2367+ BlockBackend *target;
95259824 2368+
67af0fa4
WB
2369+ target = blk_new_open(path, NULL, NULL, 0, &errp);
2370+ if (!target) {
95259824
WB
2371+ unlink(archivename);
2372+ g_error("bdrv_open '%s' failed - %s", path, error_get_pretty(errp));
2373+ }
67af0fa4 2374+ int64_t size = blk_getlength(target);
95259824
WB
2375+ int dev_id = vma_writer_register_stream(vmaw, devname, size);
2376+ if (dev_id <= 0) {
2377+ unlink(archivename);
83faa3fe 2378+ g_error("vma_writer_register_stream '%s' failed", devname);
95259824
WB
2379+ }
2380+
2381+ BackupJob *job = g_new0(BackupJob, 1);
2382+ job->len = size;
67af0fa4 2383+ job->target = target;
95259824
WB
2384+ job->vmaw = vmaw;
2385+ job->dev_id = dev_id;
2386+
2387+ Coroutine *co = qemu_coroutine_create(backup_run, job);
2bf61c3e
FE
2388+ // Don't enter coroutine yet, because it might write the header before
2389+ // all streams can be registered.
2390+ backup_coroutines = g_list_append(backup_coroutines, co);
95259824
WB
2391+ }
2392+
83faa3fe 2393+ VmaStatus vmastat;
95259824
WB
2394+ int percent = 0;
2395+ int last_percent = -1;
2396+
67af0fa4 2397+ if (devcount) {
2bf61c3e
FE
2398+ GList *entry = backup_coroutines;
2399+ while (entry && entry->data) {
2400+ Coroutine *co = entry->data;
2401+ qemu_coroutine_enter(co);
2402+ entry = g_list_next(entry);
2403+ }
2404+
67af0fa4
WB
2405+ while (1) {
2406+ main_loop_wait(false);
2407+ vma_writer_get_status(vmaw, &vmastat);
95259824 2408+
67af0fa4 2409+ if (verbose) {
95259824 2410+
67af0fa4
WB
2411+ uint64_t total = 0;
2412+ uint64_t transferred = 0;
2413+ uint64_t zero_bytes = 0;
95259824 2414+
67af0fa4
WB
2415+ int i;
2416+ for (i = 0; i < 256; i++) {
2417+ if (vmastat.stream_info[i].size) {
2418+ total += vmastat.stream_info[i].size;
2419+ transferred += vmastat.stream_info[i].transferred;
2420+ zero_bytes += vmastat.stream_info[i].zero_bytes;
2421+ }
95259824 2422+ }
67af0fa4
WB
2423+ percent = (transferred*100)/total;
2424+ if (percent != last_percent) {
2425+ fprintf(stderr, "progress %d%% %zd/%zd %zd\n", percent,
2426+ transferred, total, zero_bytes);
2427+ fflush(stderr);
95259824 2428+
67af0fa4
WB
2429+ last_percent = percent;
2430+ }
95259824 2431+ }
95259824 2432+
67af0fa4
WB
2433+ if (vmastat.closed) {
2434+ break;
2435+ }
95259824
WB
2436+ }
2437+ } else {
2438+ Coroutine *co = qemu_coroutine_create(backup_run_empty, vmaw);
2439+ qemu_coroutine_enter(co);
2440+ while (1) {
2441+ main_loop_wait(false);
2442+ vma_writer_get_status(vmaw, &vmastat);
2443+ if (vmastat.closed) {
2444+ break;
2445+ }
2446+ }
2447+ }
2448+
2449+ bdrv_drain_all();
2450+
2451+ vma_writer_get_status(vmaw, &vmastat);
2452+
2453+ if (verbose) {
2454+ for (i = 0; i < 256; i++) {
2455+ VmaStreamInfo *si = &vmastat.stream_info[i];
2456+ if (si->size) {
2457+ fprintf(stderr, "image %s: size=%zd zeros=%zd saved=%zd\n",
2458+ si->devname, si->size, si->zero_bytes,
2459+ si->size - si->zero_bytes);
2460+ }
2461+ }
2462+ }
2463+
2464+ if (vmastat.status < 0) {
2465+ unlink(archivename);
2466+ g_error("creating vma archive failed");
2467+ }
2468+
2bf61c3e
FE
2469+ g_list_free(backup_coroutines);
2470+ g_list_free(config_files);
d7f4e01a 2471+ vma_writer_destroy(vmaw);
95259824
WB
2472+ return 0;
2473+}
2474+
67af0fa4
WB
2475+static int dump_config(int argc, char **argv)
2476+{
2477+ int c, ret = 0;
2478+ const char *filename;
2479+ const char *config_name = "qemu-server.conf";
2480+
2481+ for (;;) {
2482+ c = getopt(argc, argv, "hc:");
2483+ if (c == -1) {
2484+ break;
2485+ }
2486+ switch (c) {
2487+ case '?':
2488+ case 'h':
2489+ help();
2490+ break;
2491+ case 'c':
2492+ config_name = optarg;
2493+ break;
2494+ default:
2495+ help();
2496+ }
2497+ }
2498+
2499+ /* Get the filename */
2500+ if ((optind + 1) != argc) {
2501+ help();
2502+ }
2503+ filename = argv[optind++];
2504+
2505+ Error *errp = NULL;
2506+ VmaReader *vmar = vma_reader_create(filename, &errp);
2507+
2508+ if (!vmar) {
2509+ g_error("%s", error_get_pretty(errp));
2510+ }
2511+
2512+ int found = 0;
2513+ GList *l = vma_reader_get_config_data(vmar);
2514+ while (l && l->data) {
2515+ VmaConfigData *cdata = (VmaConfigData *)l->data;
2516+ l = g_list_next(l);
2517+ if (strcmp(cdata->name, config_name) == 0) {
2518+ found = 1;
2519+ fwrite(cdata->data, cdata->len, 1, stdout);
2520+ break;
2521+ }
2522+ }
2523+
2524+ vma_reader_destroy(vmar);
2525+
2526+ bdrv_close_all();
2527+
2528+ if (!found) {
2529+ fprintf(stderr, "unable to find configuration data '%s'\n", config_name);
2530+ return -1;
2531+ }
2532+
2533+ return ret;
2534+}
2535+
95259824
WB
2536+int main(int argc, char **argv)
2537+{
2538+ const char *cmdname;
2539+ Error *main_loop_err = NULL;
2540+
6402d961
TL
2541+ error_init(argv[0]);
2542+ module_call_init(MODULE_INIT_TRACE);
2543+ qemu_init_exec_dir(argv[0]);
95259824
WB
2544+
2545+ if (qemu_init_main_loop(&main_loop_err)) {
2546+ g_error("%s", error_get_pretty(main_loop_err));
2547+ }
2548+
2549+ bdrv_init();
6402d961 2550+ module_call_init(MODULE_INIT_QOM);
95259824
WB
2551+
2552+ if (argc < 2) {
2553+ help();
2554+ }
2555+
2556+ cmdname = argv[1];
2557+ argc--; argv++;
2558+
2559+
2560+ if (!strcmp(cmdname, "list")) {
2561+ return list_content(argc, argv);
2562+ } else if (!strcmp(cmdname, "create")) {
2563+ return create_archive(argc, argv);
2564+ } else if (!strcmp(cmdname, "extract")) {
2565+ return extract_content(argc, argv);
67af0fa4
WB
2566+ } else if (!strcmp(cmdname, "verify")) {
2567+ return verify_content(argc, argv);
2568+ } else if (!strcmp(cmdname, "config")) {
2569+ return dump_config(argc, argv);
95259824
WB
2570+ }
2571+
2572+ help();
2573+ return 0;
2574+}
2575diff --git a/vma.h b/vma.h
2576new file mode 100644
6838f038 2577index 0000000000..c895c97f6d
95259824
WB
2578--- /dev/null
2579+++ b/vma.h
6838f038 2580@@ -0,0 +1,150 @@
95259824
WB
2581+/*
2582+ * VMA: Virtual Machine Archive
2583+ *
2584+ * Copyright (C) Proxmox Server Solutions
2585+ *
2586+ * Authors:
2587+ * Dietmar Maurer (dietmar@proxmox.com)
2588+ *
2589+ * This work is licensed under the terms of the GNU GPL, version 2 or later.
2590+ * See the COPYING file in the top-level directory.
2591+ *
2592+ */
2593+
2594+#ifndef BACKUP_VMA_H
2595+#define BACKUP_VMA_H
2596+
2597+#include <uuid/uuid.h>
2598+#include "qapi/error.h"
2599+#include "block/block.h"
2600+
2601+#define VMA_BLOCK_BITS 12
2602+#define VMA_BLOCK_SIZE (1<<VMA_BLOCK_BITS)
2603+#define VMA_CLUSTER_BITS (VMA_BLOCK_BITS+4)
2604+#define VMA_CLUSTER_SIZE (1<<VMA_CLUSTER_BITS)
2605+
2606+#if VMA_CLUSTER_SIZE != 65536
2607+#error unexpected cluster size
2608+#endif
2609+
2610+#define VMA_EXTENT_HEADER_SIZE 512
2611+#define VMA_BLOCKS_PER_EXTENT 59
2612+#define VMA_MAX_CONFIGS 256
2613+
2614+#define VMA_MAX_EXTENT_SIZE \
2615+ (VMA_EXTENT_HEADER_SIZE+VMA_CLUSTER_SIZE*VMA_BLOCKS_PER_EXTENT)
2616+#if VMA_MAX_EXTENT_SIZE != 3867136
2617+#error unexpected VMA_EXTENT_SIZE
2618+#endif
2619+
2620+/* File Format Definitions */
2621+
2622+#define VMA_MAGIC (GUINT32_TO_BE(('V'<<24)|('M'<<16)|('A'<<8)|0x00))
2623+#define VMA_EXTENT_MAGIC (GUINT32_TO_BE(('V'<<24)|('M'<<16)|('A'<<8)|'E'))
2624+
2625+typedef struct VmaDeviceInfoHeader {
2626+ uint32_t devname_ptr; /* offset into blob_buffer table */
2627+ uint32_t reserved0;
2628+ uint64_t size; /* device size in bytes */
2629+ uint64_t reserved1;
2630+ uint64_t reserved2;
2631+} VmaDeviceInfoHeader;
2632+
2633+typedef struct VmaHeader {
2634+ uint32_t magic;
2635+ uint32_t version;
2636+ unsigned char uuid[16];
2637+ int64_t ctime;
2638+ unsigned char md5sum[16];
2639+
2640+ uint32_t blob_buffer_offset;
2641+ uint32_t blob_buffer_size;
2642+ uint32_t header_size;
2643+
2644+ unsigned char reserved[1984];
2645+
2646+ uint32_t config_names[VMA_MAX_CONFIGS]; /* offset into blob_buffer table */
2647+ uint32_t config_data[VMA_MAX_CONFIGS]; /* offset into blob_buffer table */
2648+
2649+ uint32_t reserved1;
2650+
2651+ VmaDeviceInfoHeader dev_info[256];
2652+} VmaHeader;
2653+
2654+typedef struct VmaExtentHeader {
2655+ uint32_t magic;
2656+ uint16_t reserved1;
2657+ uint16_t block_count;
2658+ unsigned char uuid[16];
2659+ unsigned char md5sum[16];
2660+ uint64_t blockinfo[VMA_BLOCKS_PER_EXTENT];
2661+} VmaExtentHeader;
2662+
2663+/* functions/definitions to read/write vma files */
2664+
2665+typedef struct VmaReader VmaReader;
2666+
2667+typedef struct VmaWriter VmaWriter;
2668+
2669+typedef struct VmaConfigData {
2670+ const char *name;
2671+ const void *data;
2672+ uint32_t len;
2673+} VmaConfigData;
2674+
2675+typedef struct VmaStreamInfo {
2676+ uint64_t size;
2677+ uint64_t cluster_count;
2678+ uint64_t transferred;
2679+ uint64_t zero_bytes;
2680+ int finished;
2681+ char *devname;
2682+} VmaStreamInfo;
2683+
2684+typedef struct VmaStatus {
2685+ int status;
2686+ bool closed;
2687+ char errmsg[8192];
2688+ char uuid_str[37];
2689+ VmaStreamInfo stream_info[256];
2690+} VmaStatus;
2691+
2692+typedef struct VmaDeviceInfo {
2693+ uint64_t size; /* device size in bytes */
2694+ const char *devname;
2695+} VmaDeviceInfo;
2696+
2697+VmaWriter *vma_writer_create(const char *filename, uuid_t uuid, Error **errp);
2698+int vma_writer_close(VmaWriter *vmaw, Error **errp);
67af0fa4 2699+void vma_writer_error_propagate(VmaWriter *vmaw, Error **errp);
95259824
WB
2700+void vma_writer_destroy(VmaWriter *vmaw);
2701+int vma_writer_add_config(VmaWriter *vmaw, const char *name, gpointer data,
2702+ size_t len);
2703+int vma_writer_register_stream(VmaWriter *vmaw, const char *devname,
2704+ size_t size);
2705+
2706+int64_t coroutine_fn vma_writer_write(VmaWriter *vmaw, uint8_t dev_id,
6838f038
WB
2707+ int64_t cluster_num,
2708+ const unsigned char *buf,
95259824
WB
2709+ size_t *zero_bytes);
2710+
2711+int coroutine_fn vma_writer_close_stream(VmaWriter *vmaw, uint8_t dev_id);
67af0fa4 2712+int coroutine_fn vma_writer_flush_output(VmaWriter *vmaw);
95259824
WB
2713+
2714+int vma_writer_get_status(VmaWriter *vmaw, VmaStatus *status);
2715+void vma_writer_set_error(VmaWriter *vmaw, const char *fmt, ...);
2716+
2717+
2718+VmaReader *vma_reader_create(const char *filename, Error **errp);
2719+void vma_reader_destroy(VmaReader *vmar);
2720+VmaHeader *vma_reader_get_header(VmaReader *vmar);
2721+GList *vma_reader_get_config_data(VmaReader *vmar);
2722+VmaDeviceInfo *vma_reader_get_device_info(VmaReader *vmar, guint8 dev_id);
2723+int vma_reader_register_bs(VmaReader *vmar, guint8 dev_id,
67af0fa4 2724+ BlockBackend *target, bool write_zeroes,
95259824
WB
2725+ Error **errp);
2726+int vma_reader_restore(VmaReader *vmar, int vmstate_fd, bool verbose,
2727+ Error **errp);
67af0fa4 2728+int vma_reader_verify(VmaReader *vmar, bool verbose, Error **errp);
95259824
WB
2729+
2730+#endif /* BACKUP_VMA_H */