1 /*
2  * Block tests for iothreads
3  *
4  * Copyright (c) 2018 Kevin Wolf <kwolf@redhat.com>
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to deal
8  * in the Software without restriction, including without limitation the rights
9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10  * copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in
14  * all copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22  * THE SOFTWARE.
23  */
24 
25 #include "qemu/osdep.h"
26 #include "block/block.h"
27 #include "block/block_int-global-state.h"
28 #include "block/blockjob_int.h"
29 #include "sysemu/block-backend.h"
30 #include "qapi/error.h"
31 #include "qapi/qmp/qdict.h"
32 #include "qemu/main-loop.h"
33 #include "iothread.h"
34 
35 static int coroutine_fn bdrv_test_co_preadv(BlockDriverState *bs,
36                                             int64_t offset, int64_t bytes,
37                                             QEMUIOVector *qiov,
38                                             BdrvRequestFlags flags)
39 {
40     return 0;
41 }
42 
43 static int coroutine_fn bdrv_test_co_pwritev(BlockDriverState *bs,
44                                              int64_t offset, int64_t bytes,
45                                              QEMUIOVector *qiov,
46                                              BdrvRequestFlags flags)
47 {
48     return 0;
49 }
50 
51 static int coroutine_fn bdrv_test_co_pdiscard(BlockDriverState *bs,
52                                               int64_t offset, int64_t bytes)
53 {
54     return 0;
55 }
56 
57 static int coroutine_fn
58 bdrv_test_co_truncate(BlockDriverState *bs, int64_t offset, bool exact,
59                       PreallocMode prealloc, BdrvRequestFlags flags,
60                       Error **errp)
61 {
62     return 0;
63 }
64 
65 static int coroutine_fn bdrv_test_co_block_status(BlockDriverState *bs,
66                                                   bool want_zero,
67                                                   int64_t offset, int64_t count,
68                                                   int64_t *pnum, int64_t *map,
69                                                   BlockDriverState **file)
70 {
71     *pnum = count;
72     return 0;
73 }
74 
75 static BlockDriver bdrv_test = {
76     .format_name            = "test",
77     .instance_size          = 1,
78 
79     .bdrv_co_preadv         = bdrv_test_co_preadv,
80     .bdrv_co_pwritev        = bdrv_test_co_pwritev,
81     .bdrv_co_pdiscard       = bdrv_test_co_pdiscard,
82     .bdrv_co_truncate       = bdrv_test_co_truncate,
83     .bdrv_co_block_status   = bdrv_test_co_block_status,
84 };
85 
86 static void test_sync_op_pread(BdrvChild *c)
87 {
88     uint8_t buf[512];
89     int ret;
90 
91     /* Success */
92     ret = bdrv_pread(c, 0, sizeof(buf), buf, 0);
93     g_assert_cmpint(ret, ==, 0);
94 
95     /* Early error: Negative offset */
96     ret = bdrv_pread(c, -2, sizeof(buf), buf, 0);
97     g_assert_cmpint(ret, ==, -EIO);
98 }
99 
100 static void test_sync_op_pwrite(BdrvChild *c)
101 {
102     uint8_t buf[512] = { 0 };
103     int ret;
104 
105     /* Success */
106     ret = bdrv_pwrite(c, 0, sizeof(buf), buf, 0);
107     g_assert_cmpint(ret, ==, 0);
108 
109     /* Early error: Negative offset */
110     ret = bdrv_pwrite(c, -2, sizeof(buf), buf, 0);
111     g_assert_cmpint(ret, ==, -EIO);
112 }
113 
114 static void test_sync_op_blk_pread(BlockBackend *blk)
115 {
116     uint8_t buf[512];
117     int ret;
118 
119     /* Success */
120     ret = blk_pread(blk, 0, sizeof(buf), buf, 0);
121     g_assert_cmpint(ret, ==, 0);
122 
123     /* Early error: Negative offset */
124     ret = blk_pread(blk, -2, sizeof(buf), buf, 0);
125     g_assert_cmpint(ret, ==, -EIO);
126 }
127 
128 static void test_sync_op_blk_pwrite(BlockBackend *blk)
129 {
130     uint8_t buf[512] = { 0 };
131     int ret;
132 
133     /* Success */
134     ret = blk_pwrite(blk, 0, sizeof(buf), buf, 0);
135     g_assert_cmpint(ret, ==, 0);
136 
137     /* Early error: Negative offset */
138     ret = blk_pwrite(blk, -2, sizeof(buf), buf, 0);
139     g_assert_cmpint(ret, ==, -EIO);
140 }
141 
142 static void test_sync_op_blk_preadv(BlockBackend *blk)
143 {
144     uint8_t buf[512];
145     QEMUIOVector qiov = QEMU_IOVEC_INIT_BUF(qiov, buf, sizeof(buf));
146     int ret;
147 
148     /* Success */
149     ret = blk_preadv(blk, 0, sizeof(buf), &qiov, 0);
150     g_assert_cmpint(ret, ==, 0);
151 
152     /* Early error: Negative offset */
153     ret = blk_preadv(blk, -2, sizeof(buf), &qiov, 0);
154     g_assert_cmpint(ret, ==, -EIO);
155 }
156 
157 static void test_sync_op_blk_pwritev(BlockBackend *blk)
158 {
159     uint8_t buf[512] = { 0 };
160     QEMUIOVector qiov = QEMU_IOVEC_INIT_BUF(qiov, buf, sizeof(buf));
161     int ret;
162 
163     /* Success */
164     ret = blk_pwritev(blk, 0, sizeof(buf), &qiov, 0);
165     g_assert_cmpint(ret, ==, 0);
166 
167     /* Early error: Negative offset */
168     ret = blk_pwritev(blk, -2, sizeof(buf), &qiov, 0);
169     g_assert_cmpint(ret, ==, -EIO);
170 }
171 
172 static void test_sync_op_blk_preadv_part(BlockBackend *blk)
173 {
174     uint8_t buf[512];
175     QEMUIOVector qiov = QEMU_IOVEC_INIT_BUF(qiov, buf, sizeof(buf));
176     int ret;
177 
178     /* Success */
179     ret = blk_preadv_part(blk, 0, sizeof(buf), &qiov, 0, 0);
180     g_assert_cmpint(ret, ==, 0);
181 
182     /* Early error: Negative offset */
183     ret = blk_preadv_part(blk, -2, sizeof(buf), &qiov, 0, 0);
184     g_assert_cmpint(ret, ==, -EIO);
185 }
186 
187 static void test_sync_op_blk_pwritev_part(BlockBackend *blk)
188 {
189     uint8_t buf[512] = { 0 };
190     QEMUIOVector qiov = QEMU_IOVEC_INIT_BUF(qiov, buf, sizeof(buf));
191     int ret;
192 
193     /* Success */
194     ret = blk_pwritev_part(blk, 0, sizeof(buf), &qiov, 0, 0);
195     g_assert_cmpint(ret, ==, 0);
196 
197     /* Early error: Negative offset */
198     ret = blk_pwritev_part(blk, -2, sizeof(buf), &qiov, 0, 0);
199     g_assert_cmpint(ret, ==, -EIO);
200 }
201 
202 static void test_sync_op_blk_pwrite_compressed(BlockBackend *blk)
203 {
204     uint8_t buf[512] = { 0 };
205     int ret;
206 
207     /* Late error: Not supported */
208     ret = blk_pwrite_compressed(blk, 0, sizeof(buf), buf);
209     g_assert_cmpint(ret, ==, -ENOTSUP);
210 
211     /* Early error: Negative offset */
212     ret = blk_pwrite_compressed(blk, -2, sizeof(buf), buf);
213     g_assert_cmpint(ret, ==, -EIO);
214 }
215 
216 static void test_sync_op_blk_pwrite_zeroes(BlockBackend *blk)
217 {
218     int ret;
219 
220     /* Success */
221     ret = blk_pwrite_zeroes(blk, 0, 512, 0);
222     g_assert_cmpint(ret, ==, 0);
223 
224     /* Early error: Negative offset */
225     ret = blk_pwrite_zeroes(blk, -2, 512, 0);
226     g_assert_cmpint(ret, ==, -EIO);
227 }
228 
229 static void test_sync_op_load_vmstate(BdrvChild *c)
230 {
231     uint8_t buf[512];
232     int ret;
233 
234     /* Error: Driver does not support snapshots */
235     ret = bdrv_load_vmstate(c->bs, buf, 0, sizeof(buf));
236     g_assert_cmpint(ret, ==, -ENOTSUP);
237 }
238 
239 static void test_sync_op_save_vmstate(BdrvChild *c)
240 {
241     uint8_t buf[512] = { 0 };
242     int ret;
243 
244     /* Error: Driver does not support snapshots */
245     ret = bdrv_save_vmstate(c->bs, buf, 0, sizeof(buf));
246     g_assert_cmpint(ret, ==, -ENOTSUP);
247 }
248 
249 static void test_sync_op_pdiscard(BdrvChild *c)
250 {
251     int ret;
252 
253     /* Normal success path */
254     c->bs->open_flags |= BDRV_O_UNMAP;
255     ret = bdrv_pdiscard(c, 0, 512);
256     g_assert_cmpint(ret, ==, 0);
257 
258     /* Early success: UNMAP not supported */
259     c->bs->open_flags &= ~BDRV_O_UNMAP;
260     ret = bdrv_pdiscard(c, 0, 512);
261     g_assert_cmpint(ret, ==, 0);
262 
263     /* Early error: Negative offset */
264     ret = bdrv_pdiscard(c, -2, 512);
265     g_assert_cmpint(ret, ==, -EIO);
266 }
267 
268 static void test_sync_op_blk_pdiscard(BlockBackend *blk)
269 {
270     int ret;
271 
272     /* Early success: UNMAP not supported */
273     ret = blk_pdiscard(blk, 0, 512);
274     g_assert_cmpint(ret, ==, 0);
275 
276     /* Early error: Negative offset */
277     ret = blk_pdiscard(blk, -2, 512);
278     g_assert_cmpint(ret, ==, -EIO);
279 }
280 
281 static void test_sync_op_truncate(BdrvChild *c)
282 {
283     int ret;
284 
285     /* Normal success path */
286     ret = bdrv_truncate(c, 65536, false, PREALLOC_MODE_OFF, 0, NULL);
287     g_assert_cmpint(ret, ==, 0);
288 
289     /* Early error: Negative offset */
290     ret = bdrv_truncate(c, -2, false, PREALLOC_MODE_OFF, 0, NULL);
291     g_assert_cmpint(ret, ==, -EINVAL);
292 
293     /* Error: Read-only image */
294     c->bs->open_flags &= ~BDRV_O_RDWR;
295 
296     ret = bdrv_truncate(c, 65536, false, PREALLOC_MODE_OFF, 0, NULL);
297     g_assert_cmpint(ret, ==, -EACCES);
298 
299     c->bs->open_flags |= BDRV_O_RDWR;
300 }
301 
302 static void test_sync_op_blk_truncate(BlockBackend *blk)
303 {
304     int ret;
305 
306     /* Normal success path */
307     ret = blk_truncate(blk, 65536, false, PREALLOC_MODE_OFF, 0, NULL);
308     g_assert_cmpint(ret, ==, 0);
309 
310     /* Early error: Negative offset */
311     ret = blk_truncate(blk, -2, false, PREALLOC_MODE_OFF, 0, NULL);
312     g_assert_cmpint(ret, ==, -EINVAL);
313 }
314 
315 /* Disable TSA to make bdrv_test.bdrv_co_block_status writable */
316 static void TSA_NO_TSA test_sync_op_block_status(BdrvChild *c)
317 {
318     int ret;
319     int64_t n;
320 
321     /* Normal success path */
322     ret = bdrv_is_allocated(c->bs, 0, 65536, &n);
323     g_assert_cmpint(ret, ==, 0);
324 
325     /* Early success: No driver support */
326     bdrv_test.bdrv_co_block_status = NULL;
327     ret = bdrv_is_allocated(c->bs, 0, 65536, &n);
328     g_assert_cmpint(ret, ==, 1);
329 
330     /* Early success: bytes = 0 */
331     ret = bdrv_is_allocated(c->bs, 0, 0, &n);
332     g_assert_cmpint(ret, ==, 0);
333 
334     /* Early success: Offset > image size*/
335     ret = bdrv_is_allocated(c->bs, 0x1000000, 0x1000000, &n);
336     g_assert_cmpint(ret, ==, 0);
337 }
338 
339 static void test_sync_op_flush(BdrvChild *c)
340 {
341     int ret;
342 
343     /* Normal success path */
344     ret = bdrv_flush(c->bs);
345     g_assert_cmpint(ret, ==, 0);
346 
347     /* Early success: Read-only image */
348     c->bs->open_flags &= ~BDRV_O_RDWR;
349 
350     ret = bdrv_flush(c->bs);
351     g_assert_cmpint(ret, ==, 0);
352 
353     c->bs->open_flags |= BDRV_O_RDWR;
354 }
355 
356 static void test_sync_op_blk_flush(BlockBackend *blk)
357 {
358     BlockDriverState *bs = blk_bs(blk);
359     int ret;
360 
361     /* Normal success path */
362     ret = blk_flush(blk);
363     g_assert_cmpint(ret, ==, 0);
364 
365     /* Early success: Read-only image */
366     bs->open_flags &= ~BDRV_O_RDWR;
367 
368     ret = blk_flush(blk);
369     g_assert_cmpint(ret, ==, 0);
370 
371     bs->open_flags |= BDRV_O_RDWR;
372 }
373 
374 static void test_sync_op_check(BdrvChild *c)
375 {
376     BdrvCheckResult result;
377     int ret;
378 
379     /* Error: Driver does not implement check */
380     ret = bdrv_check(c->bs, &result, 0);
381     g_assert_cmpint(ret, ==, -ENOTSUP);
382 }
383 
384 static void test_sync_op_activate(BdrvChild *c)
385 {
386     /* Early success: Image is not inactive */
387     bdrv_activate(c->bs, NULL);
388 }
389 
390 
391 typedef struct SyncOpTest {
392     const char *name;
393     void (*fn)(BdrvChild *c);
394     void (*blkfn)(BlockBackend *blk);
395 } SyncOpTest;
396 
397 const SyncOpTest sync_op_tests[] = {
398     {
399         .name   = "/sync-op/pread",
400         .fn     = test_sync_op_pread,
401         .blkfn  = test_sync_op_blk_pread,
402     }, {
403         .name   = "/sync-op/pwrite",
404         .fn     = test_sync_op_pwrite,
405         .blkfn  = test_sync_op_blk_pwrite,
406     }, {
407         .name   = "/sync-op/preadv",
408         .fn     = NULL,
409         .blkfn  = test_sync_op_blk_preadv,
410     }, {
411         .name   = "/sync-op/pwritev",
412         .fn     = NULL,
413         .blkfn  = test_sync_op_blk_pwritev,
414     }, {
415         .name   = "/sync-op/preadv_part",
416         .fn     = NULL,
417         .blkfn  = test_sync_op_blk_preadv_part,
418     }, {
419         .name   = "/sync-op/pwritev_part",
420         .fn     = NULL,
421         .blkfn  = test_sync_op_blk_pwritev_part,
422     }, {
423         .name   = "/sync-op/pwrite_compressed",
424         .fn     = NULL,
425         .blkfn  = test_sync_op_blk_pwrite_compressed,
426     }, {
427         .name   = "/sync-op/pwrite_zeroes",
428         .fn     = NULL,
429         .blkfn  = test_sync_op_blk_pwrite_zeroes,
430     }, {
431         .name   = "/sync-op/load_vmstate",
432         .fn     = test_sync_op_load_vmstate,
433     }, {
434         .name   = "/sync-op/save_vmstate",
435         .fn     = test_sync_op_save_vmstate,
436     }, {
437         .name   = "/sync-op/pdiscard",
438         .fn     = test_sync_op_pdiscard,
439         .blkfn  = test_sync_op_blk_pdiscard,
440     }, {
441         .name   = "/sync-op/truncate",
442         .fn     = test_sync_op_truncate,
443         .blkfn  = test_sync_op_blk_truncate,
444     }, {
445         .name   = "/sync-op/block_status",
446         .fn     = test_sync_op_block_status,
447     }, {
448         .name   = "/sync-op/flush",
449         .fn     = test_sync_op_flush,
450         .blkfn  = test_sync_op_blk_flush,
451     }, {
452         .name   = "/sync-op/check",
453         .fn     = test_sync_op_check,
454     }, {
455         .name   = "/sync-op/activate",
456         .fn     = test_sync_op_activate,
457     },
458 };
459 
460 /* Test synchronous operations that run in a different iothread, so we have to
461  * poll for the coroutine there to return. */
462 static void test_sync_op(const void *opaque)
463 {
464     const SyncOpTest *t = opaque;
465     IOThread *iothread = iothread_new();
466     AioContext *ctx = iothread_get_aio_context(iothread);
467     BlockBackend *blk;
468     BlockDriverState *bs;
469     BdrvChild *c;
470 
471     blk = blk_new(qemu_get_aio_context(), BLK_PERM_ALL, BLK_PERM_ALL);
472     bs = bdrv_new_open_driver(&bdrv_test, "base", BDRV_O_RDWR, &error_abort);
473     bs->total_sectors = 65536 / BDRV_SECTOR_SIZE;
474     blk_insert_bs(blk, bs, &error_abort);
475     c = QLIST_FIRST(&bs->parents);
476 
477     blk_set_aio_context(blk, ctx, &error_abort);
478     aio_context_acquire(ctx);
479     if (t->fn) {
480         t->fn(c);
481     }
482     if (t->blkfn) {
483         t->blkfn(blk);
484     }
485     blk_set_aio_context(blk, qemu_get_aio_context(), &error_abort);
486     aio_context_release(ctx);
487 
488     bdrv_unref(bs);
489     blk_unref(blk);
490 }
491 
492 typedef struct TestBlockJob {
493     BlockJob common;
494     bool should_complete;
495     int n;
496 } TestBlockJob;
497 
498 static int test_job_prepare(Job *job)
499 {
500     g_assert(qemu_get_current_aio_context() == qemu_get_aio_context());
501     return 0;
502 }
503 
504 static int coroutine_fn test_job_run(Job *job, Error **errp)
505 {
506     TestBlockJob *s = container_of(job, TestBlockJob, common.job);
507 
508     job_transition_to_ready(&s->common.job);
509     while (!s->should_complete) {
510         s->n++;
511         g_assert(qemu_get_current_aio_context() == job->aio_context);
512 
513         /* Avoid job_sleep_ns() because it marks the job as !busy. We want to
514          * emulate some actual activity (probably some I/O) here so that the
515          * drain involved in AioContext switches has to wait for this activity
516          * to stop. */
517         qemu_co_sleep_ns(QEMU_CLOCK_REALTIME, 1000000);
518 
519         job_pause_point(&s->common.job);
520     }
521 
522     g_assert(qemu_get_current_aio_context() == job->aio_context);
523     return 0;
524 }
525 
526 static void test_job_complete(Job *job, Error **errp)
527 {
528     TestBlockJob *s = container_of(job, TestBlockJob, common.job);
529     s->should_complete = true;
530 }
531 
532 BlockJobDriver test_job_driver = {
533     .job_driver = {
534         .instance_size  = sizeof(TestBlockJob),
535         .free           = block_job_free,
536         .user_resume    = block_job_user_resume,
537         .run            = test_job_run,
538         .complete       = test_job_complete,
539         .prepare        = test_job_prepare,
540     },
541 };
542 
543 static void test_attach_blockjob(void)
544 {
545     IOThread *iothread = iothread_new();
546     AioContext *ctx = iothread_get_aio_context(iothread);
547     BlockBackend *blk;
548     BlockDriverState *bs;
549     TestBlockJob *tjob;
550 
551     blk = blk_new(qemu_get_aio_context(), BLK_PERM_ALL, BLK_PERM_ALL);
552     bs = bdrv_new_open_driver(&bdrv_test, "base", BDRV_O_RDWR, &error_abort);
553     blk_insert_bs(blk, bs, &error_abort);
554 
555     tjob = block_job_create("job0", &test_job_driver, NULL, bs,
556                             0, BLK_PERM_ALL,
557                             0, 0, NULL, NULL, &error_abort);
558     job_start(&tjob->common.job);
559 
560     while (tjob->n == 0) {
561         aio_poll(qemu_get_aio_context(), false);
562     }
563 
564     blk_set_aio_context(blk, ctx, &error_abort);
565 
566     tjob->n = 0;
567     while (tjob->n == 0) {
568         aio_poll(qemu_get_aio_context(), false);
569     }
570 
571     aio_context_acquire(ctx);
572     blk_set_aio_context(blk, qemu_get_aio_context(), &error_abort);
573     aio_context_release(ctx);
574 
575     tjob->n = 0;
576     while (tjob->n == 0) {
577         aio_poll(qemu_get_aio_context(), false);
578     }
579 
580     blk_set_aio_context(blk, ctx, &error_abort);
581 
582     tjob->n = 0;
583     while (tjob->n == 0) {
584         aio_poll(qemu_get_aio_context(), false);
585     }
586 
587     WITH_JOB_LOCK_GUARD() {
588         job_complete_sync_locked(&tjob->common.job, &error_abort);
589     }
590     aio_context_acquire(ctx);
591     blk_set_aio_context(blk, qemu_get_aio_context(), &error_abort);
592     aio_context_release(ctx);
593 
594     bdrv_unref(bs);
595     blk_unref(blk);
596 }
597 
598 /*
599  * Test that changing the AioContext for one node in a tree (here through blk)
600  * changes all other nodes as well:
601  *
602  *  blk
603  *   |
604  *   |  bs_verify [blkverify]
605  *   |   /               \
606  *   |  /                 \
607  *  bs_a [bdrv_test]    bs_b [bdrv_test]
608  *
609  */
610 static void test_propagate_basic(void)
611 {
612     IOThread *iothread = iothread_new();
613     AioContext *ctx = iothread_get_aio_context(iothread);
614     AioContext *main_ctx;
615     BlockBackend *blk;
616     BlockDriverState *bs_a, *bs_b, *bs_verify;
617     QDict *options;
618 
619     /*
620      * Create bs_a and its BlockBackend.  We cannot take the RESIZE
621      * permission because blkverify will not share it on the test
622      * image.
623      */
624     blk = blk_new(qemu_get_aio_context(), BLK_PERM_ALL & ~BLK_PERM_RESIZE,
625                   BLK_PERM_ALL);
626     bs_a = bdrv_new_open_driver(&bdrv_test, "bs_a", BDRV_O_RDWR, &error_abort);
627     blk_insert_bs(blk, bs_a, &error_abort);
628 
629     /* Create bs_b */
630     bs_b = bdrv_new_open_driver(&bdrv_test, "bs_b", BDRV_O_RDWR, &error_abort);
631 
632     /* Create blkverify filter that references both bs_a and bs_b */
633     options = qdict_new();
634     qdict_put_str(options, "driver", "blkverify");
635     qdict_put_str(options, "test", "bs_a");
636     qdict_put_str(options, "raw", "bs_b");
637 
638     bs_verify = bdrv_open(NULL, NULL, options, BDRV_O_RDWR, &error_abort);
639 
640     /* Switch the AioContext */
641     blk_set_aio_context(blk, ctx, &error_abort);
642     g_assert(blk_get_aio_context(blk) == ctx);
643     g_assert(bdrv_get_aio_context(bs_a) == ctx);
644     g_assert(bdrv_get_aio_context(bs_verify) == ctx);
645     g_assert(bdrv_get_aio_context(bs_b) == ctx);
646 
647     /* Switch the AioContext back */
648     main_ctx = qemu_get_aio_context();
649     aio_context_acquire(ctx);
650     blk_set_aio_context(blk, main_ctx, &error_abort);
651     aio_context_release(ctx);
652     g_assert(blk_get_aio_context(blk) == main_ctx);
653     g_assert(bdrv_get_aio_context(bs_a) == main_ctx);
654     g_assert(bdrv_get_aio_context(bs_verify) == main_ctx);
655     g_assert(bdrv_get_aio_context(bs_b) == main_ctx);
656 
657     bdrv_unref(bs_verify);
658     bdrv_unref(bs_b);
659     bdrv_unref(bs_a);
660     blk_unref(blk);
661 }
662 
663 /*
664  * Test that diamonds in the graph don't lead to endless recursion:
665  *
666  *              blk
667  *               |
668  *      bs_verify [blkverify]
669  *       /              \
670  *      /                \
671  *   bs_b [raw]         bs_c[raw]
672  *      \                /
673  *       \              /
674  *       bs_a [bdrv_test]
675  */
676 static void test_propagate_diamond(void)
677 {
678     IOThread *iothread = iothread_new();
679     AioContext *ctx = iothread_get_aio_context(iothread);
680     AioContext *main_ctx;
681     BlockBackend *blk;
682     BlockDriverState *bs_a, *bs_b, *bs_c, *bs_verify;
683     QDict *options;
684 
685     /* Create bs_a */
686     bs_a = bdrv_new_open_driver(&bdrv_test, "bs_a", BDRV_O_RDWR, &error_abort);
687 
688     /* Create bs_b and bc_c */
689     options = qdict_new();
690     qdict_put_str(options, "driver", "raw");
691     qdict_put_str(options, "file", "bs_a");
692     qdict_put_str(options, "node-name", "bs_b");
693     bs_b = bdrv_open(NULL, NULL, options, BDRV_O_RDWR, &error_abort);
694 
695     options = qdict_new();
696     qdict_put_str(options, "driver", "raw");
697     qdict_put_str(options, "file", "bs_a");
698     qdict_put_str(options, "node-name", "bs_c");
699     bs_c = bdrv_open(NULL, NULL, options, BDRV_O_RDWR, &error_abort);
700 
701     /* Create blkverify filter that references both bs_b and bs_c */
702     options = qdict_new();
703     qdict_put_str(options, "driver", "blkverify");
704     qdict_put_str(options, "test", "bs_b");
705     qdict_put_str(options, "raw", "bs_c");
706 
707     bs_verify = bdrv_open(NULL, NULL, options, BDRV_O_RDWR, &error_abort);
708     /*
709      * Do not take the RESIZE permission: This would require the same
710      * from bs_c and thus from bs_a; however, blkverify will not share
711      * it on bs_b, and thus it will not be available for bs_a.
712      */
713     blk = blk_new(qemu_get_aio_context(), BLK_PERM_ALL & ~BLK_PERM_RESIZE,
714                   BLK_PERM_ALL);
715     blk_insert_bs(blk, bs_verify, &error_abort);
716 
717     /* Switch the AioContext */
718     blk_set_aio_context(blk, ctx, &error_abort);
719     g_assert(blk_get_aio_context(blk) == ctx);
720     g_assert(bdrv_get_aio_context(bs_verify) == ctx);
721     g_assert(bdrv_get_aio_context(bs_a) == ctx);
722     g_assert(bdrv_get_aio_context(bs_b) == ctx);
723     g_assert(bdrv_get_aio_context(bs_c) == ctx);
724 
725     /* Switch the AioContext back */
726     main_ctx = qemu_get_aio_context();
727     aio_context_acquire(ctx);
728     blk_set_aio_context(blk, main_ctx, &error_abort);
729     aio_context_release(ctx);
730     g_assert(blk_get_aio_context(blk) == main_ctx);
731     g_assert(bdrv_get_aio_context(bs_verify) == main_ctx);
732     g_assert(bdrv_get_aio_context(bs_a) == main_ctx);
733     g_assert(bdrv_get_aio_context(bs_b) == main_ctx);
734     g_assert(bdrv_get_aio_context(bs_c) == main_ctx);
735 
736     blk_unref(blk);
737     bdrv_unref(bs_verify);
738     bdrv_unref(bs_c);
739     bdrv_unref(bs_b);
740     bdrv_unref(bs_a);
741 }
742 
743 static void test_propagate_mirror(void)
744 {
745     IOThread *iothread = iothread_new();
746     AioContext *ctx = iothread_get_aio_context(iothread);
747     AioContext *main_ctx = qemu_get_aio_context();
748     BlockDriverState *src, *target, *filter;
749     BlockBackend *blk;
750     Job *job;
751     Error *local_err = NULL;
752 
753     /* Create src and target*/
754     src = bdrv_new_open_driver(&bdrv_test, "src", BDRV_O_RDWR, &error_abort);
755     target = bdrv_new_open_driver(&bdrv_test, "target", BDRV_O_RDWR,
756                                   &error_abort);
757 
758     /* Start a mirror job */
759     mirror_start("job0", src, target, NULL, JOB_DEFAULT, 0, 0, 0,
760                  MIRROR_SYNC_MODE_NONE, MIRROR_OPEN_BACKING_CHAIN, false,
761                  BLOCKDEV_ON_ERROR_REPORT, BLOCKDEV_ON_ERROR_REPORT,
762                  false, "filter_node", MIRROR_COPY_MODE_BACKGROUND,
763                  &error_abort);
764     WITH_JOB_LOCK_GUARD() {
765         job = job_get_locked("job0");
766     }
767     filter = bdrv_find_node("filter_node");
768 
769     /* Change the AioContext of src */
770     bdrv_try_change_aio_context(src, ctx, NULL, &error_abort);
771     g_assert(bdrv_get_aio_context(src) == ctx);
772     g_assert(bdrv_get_aio_context(target) == ctx);
773     g_assert(bdrv_get_aio_context(filter) == ctx);
774     g_assert(job->aio_context == ctx);
775 
776     /* Change the AioContext of target */
777     aio_context_acquire(ctx);
778     bdrv_try_change_aio_context(target, main_ctx, NULL, &error_abort);
779     aio_context_release(ctx);
780     g_assert(bdrv_get_aio_context(src) == main_ctx);
781     g_assert(bdrv_get_aio_context(target) == main_ctx);
782     g_assert(bdrv_get_aio_context(filter) == main_ctx);
783 
784     /* With a BlockBackend on src, changing target must fail */
785     blk = blk_new(qemu_get_aio_context(), 0, BLK_PERM_ALL);
786     blk_insert_bs(blk, src, &error_abort);
787 
788     bdrv_try_change_aio_context(target, ctx, NULL, &local_err);
789     error_free_or_abort(&local_err);
790 
791     g_assert(blk_get_aio_context(blk) == main_ctx);
792     g_assert(bdrv_get_aio_context(src) == main_ctx);
793     g_assert(bdrv_get_aio_context(target) == main_ctx);
794     g_assert(bdrv_get_aio_context(filter) == main_ctx);
795 
796     /* ...unless we explicitly allow it */
797     aio_context_acquire(ctx);
798     blk_set_allow_aio_context_change(blk, true);
799     bdrv_try_change_aio_context(target, ctx, NULL, &error_abort);
800     aio_context_release(ctx);
801 
802     g_assert(blk_get_aio_context(blk) == ctx);
803     g_assert(bdrv_get_aio_context(src) == ctx);
804     g_assert(bdrv_get_aio_context(target) == ctx);
805     g_assert(bdrv_get_aio_context(filter) == ctx);
806 
807     job_cancel_sync_all();
808 
809     aio_context_acquire(ctx);
810     blk_set_aio_context(blk, main_ctx, &error_abort);
811     bdrv_try_change_aio_context(target, main_ctx, NULL, &error_abort);
812     aio_context_release(ctx);
813 
814     blk_unref(blk);
815     bdrv_unref(src);
816     bdrv_unref(target);
817 }
818 
819 static void test_attach_second_node(void)
820 {
821     IOThread *iothread = iothread_new();
822     AioContext *ctx = iothread_get_aio_context(iothread);
823     AioContext *main_ctx = qemu_get_aio_context();
824     BlockBackend *blk;
825     BlockDriverState *bs, *filter;
826     QDict *options;
827 
828     blk = blk_new(ctx, BLK_PERM_ALL, BLK_PERM_ALL);
829     bs = bdrv_new_open_driver(&bdrv_test, "base", BDRV_O_RDWR, &error_abort);
830     blk_insert_bs(blk, bs, &error_abort);
831 
832     options = qdict_new();
833     qdict_put_str(options, "driver", "raw");
834     qdict_put_str(options, "file", "base");
835 
836     aio_context_acquire(ctx);
837     filter = bdrv_open(NULL, NULL, options, BDRV_O_RDWR, &error_abort);
838     aio_context_release(ctx);
839 
840     g_assert(blk_get_aio_context(blk) == ctx);
841     g_assert(bdrv_get_aio_context(bs) == ctx);
842     g_assert(bdrv_get_aio_context(filter) == ctx);
843 
844     aio_context_acquire(ctx);
845     blk_set_aio_context(blk, main_ctx, &error_abort);
846     aio_context_release(ctx);
847     g_assert(blk_get_aio_context(blk) == main_ctx);
848     g_assert(bdrv_get_aio_context(bs) == main_ctx);
849     g_assert(bdrv_get_aio_context(filter) == main_ctx);
850 
851     bdrv_unref(filter);
852     bdrv_unref(bs);
853     blk_unref(blk);
854 }
855 
856 static void test_attach_preserve_blk_ctx(void)
857 {
858     IOThread *iothread = iothread_new();
859     AioContext *ctx = iothread_get_aio_context(iothread);
860     BlockBackend *blk;
861     BlockDriverState *bs;
862 
863     blk = blk_new(ctx, BLK_PERM_ALL, BLK_PERM_ALL);
864     bs = bdrv_new_open_driver(&bdrv_test, "base", BDRV_O_RDWR, &error_abort);
865     bs->total_sectors = 65536 / BDRV_SECTOR_SIZE;
866 
867     /* Add node to BlockBackend that has an iothread context assigned */
868     blk_insert_bs(blk, bs, &error_abort);
869     g_assert(blk_get_aio_context(blk) == ctx);
870     g_assert(bdrv_get_aio_context(bs) == ctx);
871 
872     /* Remove the node again */
873     aio_context_acquire(ctx);
874     blk_remove_bs(blk);
875     aio_context_release(ctx);
876     g_assert(blk_get_aio_context(blk) == ctx);
877     g_assert(bdrv_get_aio_context(bs) == qemu_get_aio_context());
878 
879     /* Re-attach the node */
880     blk_insert_bs(blk, bs, &error_abort);
881     g_assert(blk_get_aio_context(blk) == ctx);
882     g_assert(bdrv_get_aio_context(bs) == ctx);
883 
884     aio_context_acquire(ctx);
885     blk_set_aio_context(blk, qemu_get_aio_context(), &error_abort);
886     aio_context_release(ctx);
887     bdrv_unref(bs);
888     blk_unref(blk);
889 }
890 
891 int main(int argc, char **argv)
892 {
893     int i;
894 
895     bdrv_init();
896     qemu_init_main_loop(&error_abort);
897 
898     g_test_init(&argc, &argv, NULL);
899 
900     for (i = 0; i < ARRAY_SIZE(sync_op_tests); i++) {
901         const SyncOpTest *t = &sync_op_tests[i];
902         g_test_add_data_func(t->name, t, test_sync_op);
903     }
904 
905     g_test_add_func("/attach/blockjob", test_attach_blockjob);
906     g_test_add_func("/attach/second_node", test_attach_second_node);
907     g_test_add_func("/attach/preserve_blk_ctx", test_attach_preserve_blk_ctx);
908     g_test_add_func("/propagate/basic", test_propagate_basic);
909     g_test_add_func("/propagate/diamond", test_propagate_diamond);
910     g_test_add_func("/propagate/mirror", test_propagate_mirror);
911 
912     return g_test_run();
913 }
914