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

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