[Sheepdog] [PATCH, RFC] Introduce block driver claim and release hooks

[Chris Webb]

During live migration, block devices with exclusive locking behaviour such as
Sheepdog [http://www.osrg.net/sheepdog/] are problematic, as both source and
destination need to have the device open simultaneously. However, the lock is
only really required while the vm is running not while it is stopped, and at
most one vm can be running at each point of migration. This patch introduces
bdrv_claim and bdrv_release hooks which can be used to claim and release the
lock 'just in time', allowing Sheepdog-backed guests to migrate.

This functionality could also be more generally useful. For example, it would
be possible to take fcntl() locks on qcow2 files, preventing corruption from
two qemu processes concurrently modifying qcow2 metadata. Doing this in
bdrv_open() is not possible as it would prevent live migration of guests
backed by qcow2 files on a shared filesystem.

Signed-off-by: Chris Webb <chris at arachsys.com>
---
 block.c     |  24 ++++++++++++++++++++++++
 block.h     |   2 ++
 block_int.h |   2 ++
 monitor.c   |   4 +++-
 vl.c        |   3 ++-
 5 files changed, 33 insertions(+), 2 deletions(-)

diff --git a/block.c b/block.c
--- a/block.c
+++ b/block.c
@@ -475,6 +475,30 @@
     return 0;
 }
 
+int bdrv_claim_all(void)
+{
+    BlockDriverState *bs;
+
+    for (bs = bdrv_first; bs != NULL; bs = bs->next) {
+        if (bs && bs->drv && bs->drv->bdrv_claim)
+            if (bs->drv->bdrv_claim(bs) < 0) {
+                bdrv_release_all();
+                return -1;
+            }
+    }
+    return 0;
+}
+
+void bdrv_release_all(void)
+{
+    BlockDriverState *bs;
+
+    for (bs = bdrv_first; bs != NULL; bs = bs->next) {
+        if (bs && bs->drv && bs->drv->bdrv_release)
+            bs->drv->bdrv_release(bs);
+    }
+}
+
 void bdrv_close(BlockDriverState *bs)
 {
     if (bs->drv) {
diff --git a/block.h b/block.h
--- a/block.h
+++ b/block.h
@@ -58,6 +58,8 @@
 int bdrv_open(BlockDriverState *bs, const char *filename, int flags);
 int bdrv_open2(BlockDriverState *bs, const char *filename, int flags,
                BlockDriver *drv);
+int bdrv_claim_all(void);
+void bdrv_release_all(void);
 void bdrv_close(BlockDriverState *bs);
 int bdrv_check(BlockDriverState *bs);
 int bdrv_read(BlockDriverState *bs, int64_t sector_num,
diff --git a/block_int.h b/block_int.h
--- a/block_int.h
+++ b/block_int.h
@@ -51,6 +51,8 @@
     int (*bdrv_probe)(const uint8_t *buf, int buf_size, const char *filename);
     int (*bdrv_probe_device)(const char *filename);
     int (*bdrv_open)(BlockDriverState *bs, const char *filename, int flags);
+    int (*bdrv_claim)(BlockDriverState *bs);
+    void (*bdrv_release)(BlockDriverState *bs);
     int (*bdrv_read)(BlockDriverState *bs, int64_t sector_num,
                      uint8_t *buf, int nb_sectors);
     int (*bdrv_write)(BlockDriverState *bs, int64_t sector_num,
diff --git a/monitor.c b/monitor.c
--- a/monitor.c
+++ b/monitor.c
@@ -456,7 +456,9 @@
 
 static void do_quit(Monitor *mon, const QDict *qdict)
 {
-	bdrv_close_all();
+    if (vm_running)
+        bdrv_release_all();
+    bdrv_close_all();
     exit(0);
 }
 
diff --git a/vl.c b/vl.c
--- a/vl.c
+++ b/vl.c
@@ -3219,7 +3219,7 @@
 
 void vm_start(void)
 {
-    if (!vm_running) {
+    if (!vm_running && bdrv_claim_all() >= 0) {
         cpu_enable_ticks();
         vm_running = 1;
         vm_state_notify(1, 0);
@@ -3293,6 +3293,7 @@
         vm_running = 0;
         pause_all_vcpus();
         vm_state_notify(0, reason);
+        bdrv_release_all();
     }
 }