From 4fb6191acb1fdff8170a26ba9acd835c9eaf8218 Mon Sep 17 00:00:00 2001
From: Alexandre Derumier <aderumier@odiso.com>
Date: Tue, 13 Sep 2016 01:57:56 +0200
Subject: [PATCH 42/48] qmp_snapshot_drive: add aiocontext

Signed-off-by: Alexandre Derumier <aderumier@odiso.com>
---
 savevm-async.c | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/savevm-async.c b/savevm-async.c
index 2f4766c..5913a90 100644
--- a/savevm-async.c
+++ b/savevm-async.c
@@ -345,6 +345,7 @@ void qmp_snapshot_drive(const char *device, const char *name, Error **errp)
     BlockBackend *blk;
     BlockDriverState *bs;
     QEMUSnapshotInfo sn1, *sn = &sn1;
+    AioContext *aio_context;
     int ret;
 #ifdef _WIN32
     struct _timeb tb;
@@ -371,20 +372,23 @@ void qmp_snapshot_drive(const char *device, const char *name, Error **errp)
         return;
     }
 
+    aio_context = bdrv_get_aio_context(bs);
+    aio_context_acquire(aio_context);
+
     if (bdrv_is_read_only(bs)) {
         error_setg(errp, "Node '%s' is read only", device);
-        return;
+        goto out;
     }
 
     if (!bdrv_can_snapshot(bs)) {
         error_setg(errp, QERR_UNSUPPORTED);
-        return;
+        goto out;
     }
 
     if (bdrv_snapshot_find(bs, sn, name) >= 0) {
         error_set(errp, ERROR_CLASS_GENERIC_ERROR,
                   "snapshot '%s' already exists", name);
-        return;
+        goto out;
     }
 
     sn = &sn1;
@@ -409,8 +413,11 @@ void qmp_snapshot_drive(const char *device, const char *name, Error **errp)
     if (ret < 0) {
         error_set(errp, ERROR_CLASS_GENERIC_ERROR,
                   "Error while creating snapshot on '%s'\n", device);
-        return;
+        goto out;
     }
+
+out:
+    aio_context_release(aio_context);
 }
 
 void qmp_delete_drive_snapshot(const char *device, const char *name,
-- 
2.1.4

