Modify the snapshot load path to find and load snapshots contained in backing
disks, useful when the current disk is a differencing disk.

Add the source of a snapshot when listing snapshots.

This should only break backwards compatibility for scenarios depending on not
being able to load snapshots from backing disks (which doesn't seem like a
problem), and for code which parses the snapshot list output (if any).

Signed-off-by: Rob Earhart <earh...@google.com>
---
 block.c                |   10 +++++
 block.h                |    2 +
 block/qcow2-snapshot.c |  105 ++++++++++++++++++++++++++++++++++++------------
 qemu-img.c             |   25 +++++++-----
 savevm.c               |   92 +++++++++++++++++++++++++++++-------------
 5 files changed, 170 insertions(+), 64 deletions(-)

diff --git a/block.c b/block.c
index 31d1ba4..3588700 100644
--- a/block.c
+++ b/block.c
@@ -1481,6 +1481,16 @@ void bdrv_get_backing_filename(BlockDriverState *bs,
     }
 }

+const char *bdrv_get_filename(BlockDriverState *bs)
+{
+    return bs->filename;
+}
+
+BlockDriverState* bdrv_get_backing_bdrv(BlockDriverState *bs)
+{
+    return bs->backing_hd;
+}
+
 int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
                           const uint8_t *buf, int nb_sectors)
 {
diff --git a/block.h b/block.h
index edf5704..1d04322 100644
--- a/block.h
+++ b/block.h
@@ -179,6 +179,8 @@ int bdrv_get_info(BlockDriverState *bs,
BlockDriverInfo *bdi);
 const char *bdrv_get_encrypted_filename(BlockDriverState *bs);
 void bdrv_get_backing_filename(BlockDriverState *bs,
                                char *filename, int filename_size);
+const char *bdrv_get_filename(BlockDriverState *bs);
+BlockDriverState* bdrv_get_backing_bdrv(BlockDriverState *bs);
 int bdrv_snapshot_create(BlockDriverState *bs,
                          QEMUSnapshotInfo *sn_info);
 int bdrv_snapshot_goto(BlockDriverState *bs,
diff --git a/block/qcow2-snapshot.c b/block/qcow2-snapshot.c
index 8ddaea2..ca06fcf 100644
--- a/block/qcow2-snapshot.c
+++ b/block/qcow2-snapshot.c
@@ -191,16 +191,23 @@ static int qcow_write_snapshots(BlockDriverState *bs)
 static void find_new_snapshot_id(BlockDriverState *bs,
                                  char *id_str, int id_str_size)
 {
-    BDRVQcowState *s = bs->opaque;
-    QCowSnapshot *sn;
-    int i, id, id_max = 0;
+    int id_max = 0;
+
+    assert(bs);
+
+    do {
+        BDRVQcowState *s = bs->opaque;
+        QCowSnapshot *sn;
+        int i, id;
+
+        for(i = 0; i < s->nb_snapshots; i++) {
+            sn = s->snapshots + i;
+            id = strtoul(sn->id_str, NULL, 10);
+            if (id > id_max)
+                id_max = id;
+        }
+    } while ((bs = bdrv_get_backing_bdrv(bs)));

-    for(i = 0; i < s->nb_snapshots; i++) {
-        sn = s->snapshots + i;
-        id = strtoul(sn->id_str, NULL, 10);
-        if (id > id_max)
-            id_max = id;
-    }
     snprintf(id_str, id_str_size, "%d", id_max + 1);
 }

@@ -316,41 +323,88 @@ int qcow2_snapshot_goto(BlockDriverState *bs,
const char *snapshot_id)
 {
     BDRVQcowState *s = bs->opaque;
     QCowSnapshot *sn;
-    int i, snapshot_index, l1_size2;
+    int i, snapshot_index, l1_size2, ret;

     snapshot_index = find_snapshot_by_id_or_name(bs, snapshot_id);
-    if (snapshot_index < 0)
-        return -ENOENT;
+
+    if (snapshot_index < 0) {
+        /* if there is a backing file, use it. In this case, the
+         * current image needs to be reset to be consistent
+         */
+        if (bs->backing_hd) {
+            ret = bdrv_snapshot_goto(bs->backing_hd, snapshot_id);
+            if (ret < 0) {
+                return ret;
+            }
+            /* reset the sectors of the current image (Note: there can
+             * still be snapshots, so the refcounts must be kept
+             * consistent)
+             */
+            if (qcow2_update_snapshot_refcount(bs, s->l1_table_offset,
+                                               s->l1_size, -1) < 0) {
+                return -EIO;
+            }
+            l1_size2 = s->l1_size * sizeof(uint64_t);
+            memset(s->l1_table, 0, l1_size2);
+            if (bdrv_pwrite(s->hd, s->l1_table_offset,
+                            s->l1_table, l1_size2) != l1_size2) {
+                return -EIO;
+            }
+            return 0;
+        } else {
+            return -ENOENT;
+        }
+    }
+
+    assert(0 <= snapshot_index);
+
     sn = &s->snapshots[snapshot_index];

-    if (qcow2_update_snapshot_refcount(bs, s->l1_table_offset,
s->l1_size, -1) < 0)
-        goto fail;
+    /* if the file is read-only, all can be done by updating the
+     * in-memory L1 table. Otherwise, we'll need to do some writes.
+     */

-    if (qcow2_grow_l1_table(bs, sn->l1_size) < 0)
-        goto fail;
+    if (!bs->read_only) {
+        if (qcow2_update_snapshot_refcount(bs, s->l1_table_offset,
+                                           s->l1_size, -1)
+            < 0) {
+            return -EIO;
+        }
+    }
+
+    if (qcow2_grow_l1_table(bs, sn->l1_size) < 0) {
+        return -EIO;
+    }

     s->l1_size = sn->l1_size;
     l1_size2 = s->l1_size * sizeof(uint64_t);
     /* copy the snapshot l1 table to the current l1 table */
     if (bdrv_pread(s->hd, sn->l1_table_offset,
-                   s->l1_table, l1_size2) != l1_size2)
-        goto fail;
-    if (bdrv_pwrite(s->hd, s->l1_table_offset,
-                    s->l1_table, l1_size2) != l1_size2)
-        goto fail;
+                   s->l1_table, l1_size2) != l1_size2) {
+        return -EIO;
+    }
+    if (!bs->read_only) {
+        if (bdrv_pwrite(s->hd, s->l1_table_offset,
+                        s->l1_table, l1_size2) != l1_size2) {
+            return -EIO;
+        }
+    }
     for(i = 0;i < s->l1_size; i++) {
         be64_to_cpus(&s->l1_table[i]);
     }

-    if (qcow2_update_snapshot_refcount(bs, s->l1_table_offset,
s->l1_size, 1) < 0)
-        goto fail;
+    if (!bs->read_only) {
+        if (qcow2_update_snapshot_refcount(bs, s->l1_table_offset,
+                                           s->l1_size, 1)
+            < 0) {
+            return -EIO;
+        }
+    }

 #ifdef DEBUG_ALLOC
     qcow2_check_refcounts(bs);
 #endif
     return 0;
- fail:
-    return -EIO;
 }

 int qcow2_snapshot_delete(BlockDriverState *bs, const char *snapshot_id)
@@ -416,4 +470,3 @@ int qcow2_snapshot_list(BlockDriverState *bs,
QEMUSnapshotInfo **psn_tab)
     *psn_tab = sn_tab;
     return s->nb_snapshots;
 }
-
diff --git a/qemu-img.c b/qemu-img.c
index 0c9f2d4..345a768 100644
--- a/qemu-img.c
+++ b/qemu-img.c
@@ -873,16 +873,21 @@ static void dump_snapshots(BlockDriverState *bs)
     int nb_sns, i;
     char buf[256];

-    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
-    if (nb_sns <= 0)
-        return;
-    printf("Snapshot list:\n");
-    printf("%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
-    for(i = 0; i < nb_sns; i++) {
-        sn = &sn_tab[i];
-        printf("%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
-    }
-    qemu_free(sn_tab);
+    assert(bs);
+
+    do {
+        nb_sns = bdrv_snapshot_list(bs, &sn_tab);
+        if (nb_sns <= 0)
+            continue;
+        printf("Snapshot list (from %s):\n",
+               bdrv_get_filename(bs));
+        printf("%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
+        for(i = 0; i < nb_sns; i++) {
+            sn = &sn_tab[i];
+            printf("%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
+        }
+        qemu_free(sn_tab);
+    } while ((bs = bdrv_get_backing_bdrv(bs)));
 }

 static int img_info(int argc, char **argv)
diff --git a/savevm.c b/savevm.c
index 2fd3de6..8a81c42 100644
--- a/savevm.c
+++ b/savevm.c
@@ -1616,25 +1616,53 @@ static BlockDriverState *get_bs_snapshots(void)
     return bs;
 }

+/* Find a snapshot with the given name.
+ * sn_bs is an out parameter: if NULL, only the current BlockDriverState (bs)
+ * is checked; otherwise, *sb_bs is set to point to the backing block driver
+ * state where the found snapshot resides.
+ */
 static int bdrv_snapshot_find(BlockDriverState *bs, QEMUSnapshotInfo *sn_info,
-                              const char *name)
+                              const char *name, BlockDriverState **sn_bs)
 {
     QEMUSnapshotInfo *sn_tab, *sn;
     int nb_sns, i, ret;

+    assert(bs);
+
     ret = -ENOENT;
-    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
-    if (nb_sns < 0)
-        return ret;
-    for(i = 0; i < nb_sns; i++) {
-        sn = &sn_tab[i];
-        if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
-            *sn_info = *sn;
-            ret = 0;
+    do {
+        nb_sns = bdrv_snapshot_list(bs, &sn_tab);
+        if (nb_sns < 0) {
+            if (nb_sns == -ENOTSUP) {
+                /* If this device does not support snapshots,
+                 * its backing file still might; continue on
+                 * to the backing file.
+                 */
+                continue;
+            } else {
+                return ret;
+            }
+        }
+        for (i = 0; i < nb_sns; i++) {
+            sn = &sn_tab[i];
+            if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
+                *sn_info = *sn;
+                if (sn_bs != 0) {
+                    *sn_bs = bs;
+                }
+                ret = 0;
+                break;
+            }
+        }
+        qemu_free(sn_tab);
+        if (ret == 0) {
             break;
         }
-    }
-    qemu_free(sn_tab);
+        if (!sn_bs) {
+            break;
+        }
+    } while ((bs = bdrv_get_backing_bdrv(bs)));
+
     return ret;
 }

@@ -1651,7 +1679,7 @@ static int del_existing_snapshots(Monitor *mon,
const char *name)
     QTAILQ_FOREACH(dinfo, &drives, next) {
         bs = dinfo->bdrv;
         if (bdrv_can_snapshot(bs) &&
-            bdrv_snapshot_find(bs, snapshot, name) >= 0)
+            bdrv_snapshot_find(bs, snapshot, name, NULL) >= 0)
         {
             ret = bdrv_snapshot_delete(bs, name);
             if (ret < 0) {
@@ -1696,7 +1724,7 @@ void do_savevm(Monitor *mon, const QDict *qdict)

     memset(sn, 0, sizeof(*sn));
     if (name) {
-        ret = bdrv_snapshot_find(bs, old_sn, name);
+        ret = bdrv_snapshot_find(bs, old_sn, name, NULL);
         if (ret >= 0) {
             pstrcpy(sn->name, sizeof(sn->name), old_sn->name);
             pstrcpy(sn->id_str, sizeof(sn->id_str), old_sn->id_str);
@@ -1759,7 +1787,7 @@ void do_savevm(Monitor *mon, const QDict *qdict)
 int load_vmstate(Monitor *mon, const char *name)
 {
     DriveInfo *dinfo;
-    BlockDriverState *bs, *bs1;
+    BlockDriverState *bs, *bs1, *sn_bs;
     QEMUSnapshotInfo sn;
     QEMUFile *f;
     int ret;
@@ -1804,12 +1832,12 @@ int load_vmstate(Monitor *mon, const char *name)
     }

     /* Don't even try to load empty VM states */
-    ret = bdrv_snapshot_find(bs, &sn, name);
+    ret = bdrv_snapshot_find(bs, &sn, name, &sn_bs);
     if ((ret >= 0) && (sn.vm_state_size == 0))
         return -EINVAL;

     /* restore the VM state */
-    f = qemu_fopen_bdrv(bs, 0);
+    f = qemu_fopen_bdrv(sn_bs, 0);
     if (!f) {
         monitor_printf(mon, "Could not open VM state file\n");
         return -EINVAL;
@@ -1876,17 +1904,25 @@ void do_info_snapshots(Monitor *mon)
     }
     monitor_printf(mon, "\n");

-    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
-    if (nb_sns < 0) {
-        monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
-        return;
-    }
-    monitor_printf(mon, "Snapshot list (from %s):\n",
-                   bdrv_get_device_name(bs));
-    monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
-    for(i = 0; i < nb_sns; i++) {
-        sn = &sn_tab[i];
-        monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
+    for (; bs; bs = bdrv_get_backing_bdrv(bs)) {
+        nb_sns = bdrv_snapshot_list(bs, &sn_tab);
+        if (nb_sns < 0) {
+            monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
+            continue;
+        }
+        const char* device_name = bdrv_get_device_name(bs);
+        if (device_name && device_name[0]) {
+          monitor_printf(mon, "Snapshot list (device %s, from %s):\n",
+                         device_name, bdrv_get_filename(bs));
+        } else {
+          monitor_printf(mon, "Snapshot list (from %s):\n",
+                         bdrv_get_filename(bs));
+        }
+        monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf,
sizeof(buf), NULL));
+        for(i = 0; i < nb_sns; i++) {
+            sn = &sn_tab[i];
+            monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf,
sizeof(buf), sn));
+        }
+        qemu_free(sn_tab);
     }
-    qemu_free(sn_tab);
 }
-- 
1.6.6.2


Reply via email to