Blame SOURCES/kvm-block-ignore_bds_parents-parameter-for-drain-functio.patch

ae23c9
From b4a41557a3fd97307e750c84916f9d2237f08242 Mon Sep 17 00:00:00 2001
ae23c9
From: Kevin Wolf <kwolf@redhat.com>
ae23c9
Date: Wed, 10 Oct 2018 20:21:47 +0100
ae23c9
Subject: [PATCH 21/49] block: ignore_bds_parents parameter for drain functions
ae23c9
MIME-Version: 1.0
ae23c9
Content-Type: text/plain; charset=UTF-8
ae23c9
Content-Transfer-Encoding: 8bit
ae23c9
ae23c9
RH-Author: Kevin Wolf <kwolf@redhat.com>
ae23c9
Message-id: <20181010202213.7372-9-kwolf@redhat.com>
ae23c9
Patchwork-id: 82598
ae23c9
O-Subject: [RHEL-8 qemu-kvm PATCH 18/44] block: ignore_bds_parents parameter for drain functions
ae23c9
Bugzilla: 1637976
ae23c9
RH-Acked-by: Max Reitz <mreitz@redhat.com>
ae23c9
RH-Acked-by: John Snow <jsnow@redhat.com>
ae23c9
RH-Acked-by: Thomas Huth <thuth@redhat.com>
ae23c9
ae23c9
In the future, bdrv_drained_all_begin/end() will drain all invidiual
ae23c9
nodes separately rather than whole subtrees. This means that we don't
ae23c9
want to propagate the drain to all parents any more: If the parent is a
ae23c9
BDS, it will already be drained separately. Recursing to all parents is
ae23c9
unnecessary work and would make it an O(n²) operation.
ae23c9
ae23c9
Prepare the drain function for the changed drain_all by adding an
ae23c9
ignore_bds_parents parameter to the internal implementation that
ae23c9
prevents the propagation of the drain to BDS parents. We still (have to)
ae23c9
propagate it to non-BDS parents like BlockBackends or Jobs because those
ae23c9
are not drained separately.
ae23c9
ae23c9
Signed-off-by: Kevin Wolf <kwolf@redhat.com>
ae23c9
(cherry picked from commit 6cd5c9d7b2df93ef54144f170d4c908934a4767f)
ae23c9
Signed-off-by: Kevin Wolf <kwolf@redhat.com>
ae23c9
Signed-off-by: Danilo C. L. de Paula <ddepaula@redhat.com>
ae23c9
---
ae23c9
 block.c                   | 11 +++---
ae23c9
 block/io.c                | 88 ++++++++++++++++++++++++++++-------------------
ae23c9
 block/vvfat.c             |  1 +
ae23c9
 include/block/block.h     | 16 ++++++---
ae23c9
 include/block/block_int.h |  6 ++++
ae23c9
 5 files changed, 78 insertions(+), 44 deletions(-)
ae23c9
ae23c9
diff --git a/block.c b/block.c
ae23c9
index 5a50de6..519be6e 100644
ae23c9
--- a/block.c
ae23c9
+++ b/block.c
ae23c9
@@ -818,13 +818,13 @@ static char *bdrv_child_get_parent_desc(BdrvChild *c)
ae23c9
 static void bdrv_child_cb_drained_begin(BdrvChild *child)
ae23c9
 {
ae23c9
     BlockDriverState *bs = child->opaque;
ae23c9
-    bdrv_do_drained_begin_quiesce(bs, NULL);
ae23c9
+    bdrv_do_drained_begin_quiesce(bs, NULL, false);
ae23c9
 }
ae23c9
 
ae23c9
 static bool bdrv_child_cb_drained_poll(BdrvChild *child)
ae23c9
 {
ae23c9
     BlockDriverState *bs = child->opaque;
ae23c9
-    return bdrv_drain_poll(bs, false, NULL);
ae23c9
+    return bdrv_drain_poll(bs, false, NULL, false);
ae23c9
 }
ae23c9
 
ae23c9
 static void bdrv_child_cb_drained_end(BdrvChild *child)
ae23c9
@@ -908,6 +908,7 @@ static void bdrv_inherited_options(int *child_flags, QDict *child_options,
ae23c9
 }
ae23c9
 
ae23c9
 const BdrvChildRole child_file = {
ae23c9
+    .parent_is_bds   = true,
ae23c9
     .get_parent_desc = bdrv_child_get_parent_desc,
ae23c9
     .inherit_options = bdrv_inherited_options,
ae23c9
     .drained_begin   = bdrv_child_cb_drained_begin,
ae23c9
@@ -933,6 +934,7 @@ static void bdrv_inherited_fmt_options(int *child_flags, QDict *child_options,
ae23c9
 }
ae23c9
 
ae23c9
 const BdrvChildRole child_format = {
ae23c9
+    .parent_is_bds   = true,
ae23c9
     .get_parent_desc = bdrv_child_get_parent_desc,
ae23c9
     .inherit_options = bdrv_inherited_fmt_options,
ae23c9
     .drained_begin   = bdrv_child_cb_drained_begin,
ae23c9
@@ -1051,6 +1053,7 @@ static int bdrv_backing_update_filename(BdrvChild *c, BlockDriverState *base,
ae23c9
 }
ae23c9
 
ae23c9
 const BdrvChildRole child_backing = {
ae23c9
+    .parent_is_bds   = true,
ae23c9
     .get_parent_desc = bdrv_child_get_parent_desc,
ae23c9
     .attach          = bdrv_backing_attach,
ae23c9
     .detach          = bdrv_backing_detach,
ae23c9
@@ -4901,7 +4904,7 @@ void bdrv_set_aio_context(BlockDriverState *bs, AioContext *new_context)
ae23c9
     AioContext *ctx = bdrv_get_aio_context(bs);
ae23c9
 
ae23c9
     aio_disable_external(ctx);
ae23c9
-    bdrv_parent_drained_begin(bs, NULL);
ae23c9
+    bdrv_parent_drained_begin(bs, NULL, false);
ae23c9
     bdrv_drain(bs); /* ensure there are no in-flight requests */
ae23c9
 
ae23c9
     while (aio_poll(ctx, false)) {
ae23c9
@@ -4915,7 +4918,7 @@ void bdrv_set_aio_context(BlockDriverState *bs, AioContext *new_context)
ae23c9
      */
ae23c9
     aio_context_acquire(new_context);
ae23c9
     bdrv_attach_aio_context(bs, new_context);
ae23c9
-    bdrv_parent_drained_end(bs, NULL);
ae23c9
+    bdrv_parent_drained_end(bs, NULL, false);
ae23c9
     aio_enable_external(ctx);
ae23c9
     aio_context_release(new_context);
ae23c9
 }
ae23c9
diff --git a/block/io.c b/block/io.c
ae23c9
index b58c91b..0021fefd 100644
ae23c9
--- a/block/io.c
ae23c9
+++ b/block/io.c
ae23c9
@@ -41,12 +41,13 @@
ae23c9
 static int coroutine_fn bdrv_co_do_pwrite_zeroes(BlockDriverState *bs,
ae23c9
     int64_t offset, int bytes, BdrvRequestFlags flags);
ae23c9
 
ae23c9
-void bdrv_parent_drained_begin(BlockDriverState *bs, BdrvChild *ignore)
ae23c9
+void bdrv_parent_drained_begin(BlockDriverState *bs, BdrvChild *ignore,
ae23c9
+                               bool ignore_bds_parents)
ae23c9
 {
ae23c9
     BdrvChild *c, *next;
ae23c9
 
ae23c9
     QLIST_FOREACH_SAFE(c, &bs->parents, next_parent, next) {
ae23c9
-        if (c == ignore) {
ae23c9
+        if (c == ignore || (ignore_bds_parents && c->role->parent_is_bds)) {
ae23c9
             continue;
ae23c9
         }
ae23c9
         if (c->role->drained_begin) {
ae23c9
@@ -55,12 +56,13 @@ void bdrv_parent_drained_begin(BlockDriverState *bs, BdrvChild *ignore)
ae23c9
     }
ae23c9
 }
ae23c9
 
ae23c9
-void bdrv_parent_drained_end(BlockDriverState *bs, BdrvChild *ignore)
ae23c9
+void bdrv_parent_drained_end(BlockDriverState *bs, BdrvChild *ignore,
ae23c9
+                             bool ignore_bds_parents)
ae23c9
 {
ae23c9
     BdrvChild *c, *next;
ae23c9
 
ae23c9
     QLIST_FOREACH_SAFE(c, &bs->parents, next_parent, next) {
ae23c9
-        if (c == ignore) {
ae23c9
+        if (c == ignore || (ignore_bds_parents && c->role->parent_is_bds)) {
ae23c9
             continue;
ae23c9
         }
ae23c9
         if (c->role->drained_end) {
ae23c9
@@ -69,13 +71,14 @@ void bdrv_parent_drained_end(BlockDriverState *bs, BdrvChild *ignore)
ae23c9
     }
ae23c9
 }
ae23c9
 
ae23c9
-static bool bdrv_parent_drained_poll(BlockDriverState *bs, BdrvChild *ignore)
ae23c9
+static bool bdrv_parent_drained_poll(BlockDriverState *bs, BdrvChild *ignore,
ae23c9
+                                     bool ignore_bds_parents)
ae23c9
 {
ae23c9
     BdrvChild *c, *next;
ae23c9
     bool busy = false;
ae23c9
 
ae23c9
     QLIST_FOREACH_SAFE(c, &bs->parents, next_parent, next) {
ae23c9
-        if (c == ignore) {
ae23c9
+        if (c == ignore || (ignore_bds_parents && c->role->parent_is_bds)) {
ae23c9
             continue;
ae23c9
         }
ae23c9
         if (c->role->drained_poll) {
ae23c9
@@ -166,6 +169,7 @@ typedef struct {
ae23c9
     bool recursive;
ae23c9
     bool poll;
ae23c9
     BdrvChild *parent;
ae23c9
+    bool ignore_bds_parents;
ae23c9
 } BdrvCoDrainData;
ae23c9
 
ae23c9
 static void coroutine_fn bdrv_drain_invoke_entry(void *opaque)
ae23c9
@@ -219,11 +223,11 @@ static void bdrv_drain_invoke(BlockDriverState *bs, bool begin)
ae23c9
 
ae23c9
 /* Returns true if BDRV_POLL_WHILE() should go into a blocking aio_poll() */
ae23c9
 bool bdrv_drain_poll(BlockDriverState *bs, bool recursive,
ae23c9
-                     BdrvChild *ignore_parent)
ae23c9
+                     BdrvChild *ignore_parent, bool ignore_bds_parents)
ae23c9
 {
ae23c9
     BdrvChild *child, *next;
ae23c9
 
ae23c9
-    if (bdrv_parent_drained_poll(bs, ignore_parent)) {
ae23c9
+    if (bdrv_parent_drained_poll(bs, ignore_parent, ignore_bds_parents)) {
ae23c9
         return true;
ae23c9
     }
ae23c9
 
ae23c9
@@ -232,8 +236,9 @@ bool bdrv_drain_poll(BlockDriverState *bs, bool recursive,
ae23c9
     }
ae23c9
 
ae23c9
     if (recursive) {
ae23c9
+        assert(!ignore_bds_parents);
ae23c9
         QLIST_FOREACH_SAFE(child, &bs->children, next, next) {
ae23c9
-            if (bdrv_drain_poll(child->bs, recursive, child)) {
ae23c9
+            if (bdrv_drain_poll(child->bs, recursive, child, false)) {
ae23c9
                 return true;
ae23c9
             }
ae23c9
         }
ae23c9
@@ -249,13 +254,14 @@ static bool bdrv_drain_poll_top_level(BlockDriverState *bs, bool recursive,
ae23c9
      * have executed. */
ae23c9
     while (aio_poll(bs->aio_context, false));
ae23c9
 
ae23c9
-    return bdrv_drain_poll(bs, recursive, ignore_parent);
ae23c9
+    return bdrv_drain_poll(bs, recursive, ignore_parent, false);
ae23c9
 }
ae23c9
 
ae23c9
 static void bdrv_do_drained_begin(BlockDriverState *bs, bool recursive,
ae23c9
-                                  BdrvChild *parent, bool poll);
ae23c9
+                                  BdrvChild *parent, bool ignore_bds_parents,
ae23c9
+                                  bool poll);
ae23c9
 static void bdrv_do_drained_end(BlockDriverState *bs, bool recursive,
ae23c9
-                                BdrvChild *parent);
ae23c9
+                                BdrvChild *parent, bool ignore_bds_parents);
ae23c9
 
ae23c9
 static void bdrv_co_drain_bh_cb(void *opaque)
ae23c9
 {
ae23c9
@@ -266,9 +272,11 @@ static void bdrv_co_drain_bh_cb(void *opaque)
ae23c9
     if (bs) {
ae23c9
         bdrv_dec_in_flight(bs);
ae23c9
         if (data->begin) {
ae23c9
-            bdrv_do_drained_begin(bs, data->recursive, data->parent, data->poll);
ae23c9
+            bdrv_do_drained_begin(bs, data->recursive, data->parent,
ae23c9
+                                  data->ignore_bds_parents, data->poll);
ae23c9
         } else {
ae23c9
-            bdrv_do_drained_end(bs, data->recursive, data->parent);
ae23c9
+            bdrv_do_drained_end(bs, data->recursive, data->parent,
ae23c9
+                                data->ignore_bds_parents);
ae23c9
         }
ae23c9
     } else {
ae23c9
         assert(data->begin);
ae23c9
@@ -281,7 +289,9 @@ static void bdrv_co_drain_bh_cb(void *opaque)
ae23c9
 
ae23c9
 static void coroutine_fn bdrv_co_yield_to_drain(BlockDriverState *bs,
ae23c9
                                                 bool begin, bool recursive,
ae23c9
-                                                BdrvChild *parent, bool poll)
ae23c9
+                                                BdrvChild *parent,
ae23c9
+                                                bool ignore_bds_parents,
ae23c9
+                                                bool poll)
ae23c9
 {
ae23c9
     BdrvCoDrainData data;
ae23c9
 
ae23c9
@@ -296,6 +306,7 @@ static void coroutine_fn bdrv_co_yield_to_drain(BlockDriverState *bs,
ae23c9
         .begin = begin,
ae23c9
         .recursive = recursive,
ae23c9
         .parent = parent,
ae23c9
+        .ignore_bds_parents = ignore_bds_parents,
ae23c9
         .poll = poll,
ae23c9
     };
ae23c9
     if (bs) {
ae23c9
@@ -311,7 +322,7 @@ static void coroutine_fn bdrv_co_yield_to_drain(BlockDriverState *bs,
ae23c9
 }
ae23c9
 
ae23c9
 void bdrv_do_drained_begin_quiesce(BlockDriverState *bs,
ae23c9
-                                   BdrvChild *parent)
ae23c9
+                                   BdrvChild *parent, bool ignore_bds_parents)
ae23c9
 {
ae23c9
     assert(!qemu_in_coroutine());
ae23c9
 
ae23c9
@@ -320,26 +331,30 @@ void bdrv_do_drained_begin_quiesce(BlockDriverState *bs,
ae23c9
         aio_disable_external(bdrv_get_aio_context(bs));
ae23c9
     }
ae23c9
 
ae23c9
-    bdrv_parent_drained_begin(bs, parent);
ae23c9
+    bdrv_parent_drained_begin(bs, parent, ignore_bds_parents);
ae23c9
     bdrv_drain_invoke(bs, true);
ae23c9
 }
ae23c9
 
ae23c9
 static void bdrv_do_drained_begin(BlockDriverState *bs, bool recursive,
ae23c9
-                                  BdrvChild *parent, bool poll)
ae23c9
+                                  BdrvChild *parent, bool ignore_bds_parents,
ae23c9
+                                  bool poll)
ae23c9
 {
ae23c9
     BdrvChild *child, *next;
ae23c9
 
ae23c9
     if (qemu_in_coroutine()) {
ae23c9
-        bdrv_co_yield_to_drain(bs, true, recursive, parent, poll);
ae23c9
+        bdrv_co_yield_to_drain(bs, true, recursive, parent, ignore_bds_parents,
ae23c9
+                               poll);
ae23c9
         return;
ae23c9
     }
ae23c9
 
ae23c9
-    bdrv_do_drained_begin_quiesce(bs, parent);
ae23c9
+    bdrv_do_drained_begin_quiesce(bs, parent, ignore_bds_parents);
ae23c9
 
ae23c9
     if (recursive) {
ae23c9
+        assert(!ignore_bds_parents);
ae23c9
         bs->recursive_quiesce_counter++;
ae23c9
         QLIST_FOREACH_SAFE(child, &bs->children, next, next) {
ae23c9
-            bdrv_do_drained_begin(child->bs, true, child, false);
ae23c9
+            bdrv_do_drained_begin(child->bs, true, child, ignore_bds_parents,
ae23c9
+                                  false);
ae23c9
         }
ae23c9
     }
ae23c9
 
ae23c9
@@ -353,28 +368,30 @@ static void bdrv_do_drained_begin(BlockDriverState *bs, bool recursive,
ae23c9
      * nodes.
ae23c9
      */
ae23c9
     if (poll) {
ae23c9
+        assert(!ignore_bds_parents);
ae23c9
         BDRV_POLL_WHILE(bs, bdrv_drain_poll_top_level(bs, recursive, parent));
ae23c9
     }
ae23c9
 }
ae23c9
 
ae23c9
 void bdrv_drained_begin(BlockDriverState *bs)
ae23c9
 {
ae23c9
-    bdrv_do_drained_begin(bs, false, NULL, true);
ae23c9
+    bdrv_do_drained_begin(bs, false, NULL, false, true);
ae23c9
 }
ae23c9
 
ae23c9
 void bdrv_subtree_drained_begin(BlockDriverState *bs)
ae23c9
 {
ae23c9
-    bdrv_do_drained_begin(bs, true, NULL, true);
ae23c9
+    bdrv_do_drained_begin(bs, true, NULL, false, true);
ae23c9
 }
ae23c9
 
ae23c9
-void bdrv_do_drained_end(BlockDriverState *bs, bool recursive,
ae23c9
-                         BdrvChild *parent)
ae23c9
+static void bdrv_do_drained_end(BlockDriverState *bs, bool recursive,
ae23c9
+                                BdrvChild *parent, bool ignore_bds_parents)
ae23c9
 {
ae23c9
     BdrvChild *child, *next;
ae23c9
     int old_quiesce_counter;
ae23c9
 
ae23c9
     if (qemu_in_coroutine()) {
ae23c9
-        bdrv_co_yield_to_drain(bs, false, recursive, parent, false);
ae23c9
+        bdrv_co_yield_to_drain(bs, false, recursive, parent, ignore_bds_parents,
ae23c9
+                               false);
ae23c9
         return;
ae23c9
     }
ae23c9
     assert(bs->quiesce_counter > 0);
ae23c9
@@ -382,27 +399,28 @@ void bdrv_do_drained_end(BlockDriverState *bs, bool recursive,
ae23c9
 
ae23c9
     /* Re-enable things in child-to-parent order */
ae23c9
     bdrv_drain_invoke(bs, false);
ae23c9
-    bdrv_parent_drained_end(bs, parent);
ae23c9
+    bdrv_parent_drained_end(bs, parent, ignore_bds_parents);
ae23c9
     if (old_quiesce_counter == 1) {
ae23c9
         aio_enable_external(bdrv_get_aio_context(bs));
ae23c9
     }
ae23c9
 
ae23c9
     if (recursive) {
ae23c9
+        assert(!ignore_bds_parents);
ae23c9
         bs->recursive_quiesce_counter--;
ae23c9
         QLIST_FOREACH_SAFE(child, &bs->children, next, next) {
ae23c9
-            bdrv_do_drained_end(child->bs, true, child);
ae23c9
+            bdrv_do_drained_end(child->bs, true, child, ignore_bds_parents);
ae23c9
         }
ae23c9
     }
ae23c9
 }
ae23c9
 
ae23c9
 void bdrv_drained_end(BlockDriverState *bs)
ae23c9
 {
ae23c9
-    bdrv_do_drained_end(bs, false, NULL);
ae23c9
+    bdrv_do_drained_end(bs, false, NULL, false);
ae23c9
 }
ae23c9
 
ae23c9
 void bdrv_subtree_drained_end(BlockDriverState *bs)
ae23c9
 {
ae23c9
-    bdrv_do_drained_end(bs, true, NULL);
ae23c9
+    bdrv_do_drained_end(bs, true, NULL, false);
ae23c9
 }
ae23c9
 
ae23c9
 void bdrv_apply_subtree_drain(BdrvChild *child, BlockDriverState *new_parent)
ae23c9
@@ -410,7 +428,7 @@ void bdrv_apply_subtree_drain(BdrvChild *child, BlockDriverState *new_parent)
ae23c9
     int i;
ae23c9
 
ae23c9
     for (i = 0; i < new_parent->recursive_quiesce_counter; i++) {
ae23c9
-        bdrv_do_drained_begin(child->bs, true, child, true);
ae23c9
+        bdrv_do_drained_begin(child->bs, true, child, false, true);
ae23c9
     }
ae23c9
 }
ae23c9
 
ae23c9
@@ -419,7 +437,7 @@ void bdrv_unapply_subtree_drain(BdrvChild *child, BlockDriverState *old_parent)
ae23c9
     int i;
ae23c9
 
ae23c9
     for (i = 0; i < old_parent->recursive_quiesce_counter; i++) {
ae23c9
-        bdrv_do_drained_end(child->bs, true, child);
ae23c9
+        bdrv_do_drained_end(child->bs, true, child, false);
ae23c9
     }
ae23c9
 }
ae23c9
 
ae23c9
@@ -471,7 +489,7 @@ void bdrv_drain_all_begin(void)
ae23c9
     BdrvNextIterator it;
ae23c9
 
ae23c9
     if (qemu_in_coroutine()) {
ae23c9
-        bdrv_co_yield_to_drain(NULL, true, false, NULL, true);
ae23c9
+        bdrv_co_yield_to_drain(NULL, true, false, NULL, false, true);
ae23c9
         return;
ae23c9
     }
ae23c9
 
ae23c9
@@ -485,7 +503,7 @@ void bdrv_drain_all_begin(void)
ae23c9
         AioContext *aio_context = bdrv_get_aio_context(bs);
ae23c9
 
ae23c9
         aio_context_acquire(aio_context);
ae23c9
-        bdrv_do_drained_begin(bs, true, NULL, true);
ae23c9
+        bdrv_do_drained_begin(bs, true, NULL, false, true);
ae23c9
         aio_context_release(aio_context);
ae23c9
     }
ae23c9
 
ae23c9
@@ -503,7 +521,7 @@ void bdrv_drain_all_end(void)
ae23c9
         AioContext *aio_context = bdrv_get_aio_context(bs);
ae23c9
 
ae23c9
         aio_context_acquire(aio_context);
ae23c9
-        bdrv_do_drained_end(bs, true, NULL);
ae23c9
+        bdrv_do_drained_end(bs, true, NULL, false);
ae23c9
         aio_context_release(aio_context);
ae23c9
     }
ae23c9
 }
ae23c9
diff --git a/block/vvfat.c b/block/vvfat.c
ae23c9
index 4595f33..c7d2ed2 100644
ae23c9
--- a/block/vvfat.c
ae23c9
+++ b/block/vvfat.c
ae23c9
@@ -3134,6 +3134,7 @@ static void vvfat_qcow_options(int *child_flags, QDict *child_options,
ae23c9
 }
ae23c9
 
ae23c9
 static const BdrvChildRole child_vvfat_qcow = {
ae23c9
+    .parent_is_bds      = true,
ae23c9
     .inherit_options    = vvfat_qcow_options,
ae23c9
 };
ae23c9
 
ae23c9
diff --git a/include/block/block.h b/include/block/block.h
ae23c9
index 43f29b5..6e91803 100644
ae23c9
--- a/include/block/block.h
ae23c9
+++ b/include/block/block.h
ae23c9
@@ -585,7 +585,8 @@ void bdrv_io_unplug(BlockDriverState *bs);
ae23c9
  * Begin a quiesced section of all users of @bs. This is part of
ae23c9
  * bdrv_drained_begin.
ae23c9
  */
ae23c9
-void bdrv_parent_drained_begin(BlockDriverState *bs, BdrvChild *ignore);
ae23c9
+void bdrv_parent_drained_begin(BlockDriverState *bs, BdrvChild *ignore,
ae23c9
+                               bool ignore_bds_parents);
ae23c9
 
ae23c9
 /**
ae23c9
  * bdrv_parent_drained_end:
ae23c9
@@ -593,18 +594,23 @@ void bdrv_parent_drained_begin(BlockDriverState *bs, BdrvChild *ignore);
ae23c9
  * End a quiesced section of all users of @bs. This is part of
ae23c9
  * bdrv_drained_end.
ae23c9
  */
ae23c9
-void bdrv_parent_drained_end(BlockDriverState *bs, BdrvChild *ignore);
ae23c9
+void bdrv_parent_drained_end(BlockDriverState *bs, BdrvChild *ignore,
ae23c9
+                             bool ignore_bds_parents);
ae23c9
 
ae23c9
 /**
ae23c9
  * bdrv_drain_poll:
ae23c9
  *
ae23c9
  * Poll for pending requests in @bs, its parents (except for @ignore_parent),
ae23c9
- * and if @recursive is true its children as well.
ae23c9
+ * and if @recursive is true its children as well (used for subtree drain).
ae23c9
+ *
ae23c9
+ * If @ignore_bds_parents is true, parents that are BlockDriverStates must
ae23c9
+ * ignore the drain request because they will be drained separately (used for
ae23c9
+ * drain_all).
ae23c9
  *
ae23c9
  * This is part of bdrv_drained_begin.
ae23c9
  */
ae23c9
 bool bdrv_drain_poll(BlockDriverState *bs, bool recursive,
ae23c9
-                     BdrvChild *ignore_parent);
ae23c9
+                     BdrvChild *ignore_parent, bool ignore_bds_parents);
ae23c9
 
ae23c9
 /**
ae23c9
  * bdrv_drained_begin:
ae23c9
@@ -625,7 +631,7 @@ void bdrv_drained_begin(BlockDriverState *bs);
ae23c9
  * running requests to complete.
ae23c9
  */
ae23c9
 void bdrv_do_drained_begin_quiesce(BlockDriverState *bs,
ae23c9
-                                   BdrvChild *parent);
ae23c9
+                                   BdrvChild *parent, bool ignore_bds_parents);
ae23c9
 
ae23c9
 /**
ae23c9
  * Like bdrv_drained_begin, but recursively begins a quiesced section for
ae23c9
diff --git a/include/block/block_int.h b/include/block/block_int.h
ae23c9
index beeacde..0ad8a76 100644
ae23c9
--- a/include/block/block_int.h
ae23c9
+++ b/include/block/block_int.h
ae23c9
@@ -582,6 +582,12 @@ struct BdrvChildRole {
ae23c9
      * points to. */
ae23c9
     bool stay_at_node;
ae23c9
 
ae23c9
+    /* If true, the parent is a BlockDriverState and bdrv_next_all_states()
ae23c9
+     * will return it. This information is used for drain_all, where every node
ae23c9
+     * will be drained separately, so the drain only needs to be propagated to
ae23c9
+     * non-BDS parents. */
ae23c9
+    bool parent_is_bds;
ae23c9
+
ae23c9
     void (*inherit_options)(int *child_flags, QDict *child_options,
ae23c9
                             int parent_flags, QDict *parent_options);
ae23c9
 
ae23c9
-- 
ae23c9
1.8.3.1
ae23c9