xref: /qemu/blockjob.c (revision 970311646a701eecb103eb28093e8924d2fa6861)
12f0c9fe6SPaolo Bonzini /*
22f0c9fe6SPaolo Bonzini  * QEMU System Emulator block driver
32f0c9fe6SPaolo Bonzini  *
42f0c9fe6SPaolo Bonzini  * Copyright (c) 2011 IBM Corp.
52f0c9fe6SPaolo Bonzini  * Copyright (c) 2012 Red Hat, Inc.
62f0c9fe6SPaolo Bonzini  *
72f0c9fe6SPaolo Bonzini  * Permission is hereby granted, free of charge, to any person obtaining a copy
82f0c9fe6SPaolo Bonzini  * of this software and associated documentation files (the "Software"), to deal
92f0c9fe6SPaolo Bonzini  * in the Software without restriction, including without limitation the rights
102f0c9fe6SPaolo Bonzini  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
112f0c9fe6SPaolo Bonzini  * copies of the Software, and to permit persons to whom the Software is
122f0c9fe6SPaolo Bonzini  * furnished to do so, subject to the following conditions:
132f0c9fe6SPaolo Bonzini  *
142f0c9fe6SPaolo Bonzini  * The above copyright notice and this permission notice shall be included in
152f0c9fe6SPaolo Bonzini  * all copies or substantial portions of the Software.
162f0c9fe6SPaolo Bonzini  *
172f0c9fe6SPaolo Bonzini  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
182f0c9fe6SPaolo Bonzini  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
192f0c9fe6SPaolo Bonzini  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
202f0c9fe6SPaolo Bonzini  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
212f0c9fe6SPaolo Bonzini  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
222f0c9fe6SPaolo Bonzini  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
232f0c9fe6SPaolo Bonzini  * THE SOFTWARE.
242f0c9fe6SPaolo Bonzini  */
252f0c9fe6SPaolo Bonzini 
262f0c9fe6SPaolo Bonzini #include "config-host.h"
272f0c9fe6SPaolo Bonzini #include "qemu-common.h"
282f0c9fe6SPaolo Bonzini #include "trace.h"
29737e150eSPaolo Bonzini #include "block/block.h"
30737e150eSPaolo Bonzini #include "block/blockjob.h"
31737e150eSPaolo Bonzini #include "block/block_int.h"
32cc7a8ea7SMarkus Armbruster #include "qapi/qmp/qerror.h"
337b1b5d19SPaolo Bonzini #include "qapi/qmp/qjson.h"
34737e150eSPaolo Bonzini #include "block/coroutine.h"
352f0c9fe6SPaolo Bonzini #include "qmp-commands.h"
361de7afc9SPaolo Bonzini #include "qemu/timer.h"
375a2d2cbdSWenchao Xia #include "qapi-event.h"
382f0c9fe6SPaolo Bonzini 
393fc4b10aSFam Zheng void *block_job_create(const BlockJobDriver *driver, BlockDriverState *bs,
40097310b5SMarkus Armbruster                        int64_t speed, BlockCompletionFunc *cb,
412f0c9fe6SPaolo Bonzini                        void *opaque, Error **errp)
422f0c9fe6SPaolo Bonzini {
432f0c9fe6SPaolo Bonzini     BlockJob *job;
442f0c9fe6SPaolo Bonzini 
45628ff683SFam Zheng     if (bs->job) {
46c6bd8c70SMarkus Armbruster         error_setg(errp, QERR_DEVICE_IN_USE, bdrv_get_device_name(bs));
472f0c9fe6SPaolo Bonzini         return NULL;
482f0c9fe6SPaolo Bonzini     }
49fa510ebfSFam Zheng     bdrv_ref(bs);
503fc4b10aSFam Zheng     job = g_malloc0(driver->instance_size);
513718d8abSFam Zheng     error_setg(&job->blocker, "block device is in use by block job: %s",
523718d8abSFam Zheng                BlockJobType_lookup[driver->job_type]);
533718d8abSFam Zheng     bdrv_op_block_all(bs, job->blocker);
54b112a65cSStefan Hajnoczi     bdrv_op_unblock(bs, BLOCK_OP_TYPE_DATAPLANE, job->blocker);
553718d8abSFam Zheng 
563fc4b10aSFam Zheng     job->driver        = driver;
572f0c9fe6SPaolo Bonzini     job->bs            = bs;
582f0c9fe6SPaolo Bonzini     job->cb            = cb;
592f0c9fe6SPaolo Bonzini     job->opaque        = opaque;
602f0c9fe6SPaolo Bonzini     job->busy          = true;
612f0c9fe6SPaolo Bonzini     bs->job = job;
622f0c9fe6SPaolo Bonzini 
632f0c9fe6SPaolo Bonzini     /* Only set speed when necessary to avoid NotSupported error */
642f0c9fe6SPaolo Bonzini     if (speed != 0) {
652f0c9fe6SPaolo Bonzini         Error *local_err = NULL;
662f0c9fe6SPaolo Bonzini 
672f0c9fe6SPaolo Bonzini         block_job_set_speed(job, speed, &local_err);
6884d18f06SMarkus Armbruster         if (local_err) {
69*97031164STing Wang             block_job_release(bs);
702f0c9fe6SPaolo Bonzini             error_propagate(errp, local_err);
712f0c9fe6SPaolo Bonzini             return NULL;
722f0c9fe6SPaolo Bonzini         }
732f0c9fe6SPaolo Bonzini     }
742f0c9fe6SPaolo Bonzini     return job;
752f0c9fe6SPaolo Bonzini }
762f0c9fe6SPaolo Bonzini 
77*97031164STing Wang void block_job_release(BlockDriverState *bs)
78*97031164STing Wang {
79*97031164STing Wang     BlockJob *job = bs->job;
80*97031164STing Wang 
81*97031164STing Wang     bs->job = NULL;
82*97031164STing Wang     bdrv_op_unblock_all(bs, job->blocker);
83*97031164STing Wang     error_free(job->blocker);
84*97031164STing Wang     g_free(job);
85*97031164STing Wang }
86*97031164STing Wang 
8765f46322SPaolo Bonzini void block_job_completed(BlockJob *job, int ret)
882f0c9fe6SPaolo Bonzini {
892f0c9fe6SPaolo Bonzini     BlockDriverState *bs = job->bs;
902f0c9fe6SPaolo Bonzini 
912f0c9fe6SPaolo Bonzini     assert(bs->job == job);
922f0c9fe6SPaolo Bonzini     job->cb(job->opaque, ret);
93*97031164STing Wang     block_job_release(bs);
942f0c9fe6SPaolo Bonzini }
952f0c9fe6SPaolo Bonzini 
962f0c9fe6SPaolo Bonzini void block_job_set_speed(BlockJob *job, int64_t speed, Error **errp)
972f0c9fe6SPaolo Bonzini {
982f0c9fe6SPaolo Bonzini     Error *local_err = NULL;
992f0c9fe6SPaolo Bonzini 
1003fc4b10aSFam Zheng     if (!job->driver->set_speed) {
101c6bd8c70SMarkus Armbruster         error_setg(errp, QERR_UNSUPPORTED);
1022f0c9fe6SPaolo Bonzini         return;
1032f0c9fe6SPaolo Bonzini     }
1043fc4b10aSFam Zheng     job->driver->set_speed(job, speed, &local_err);
10584d18f06SMarkus Armbruster     if (local_err) {
1062f0c9fe6SPaolo Bonzini         error_propagate(errp, local_err);
1072f0c9fe6SPaolo Bonzini         return;
1082f0c9fe6SPaolo Bonzini     }
1092f0c9fe6SPaolo Bonzini 
1102f0c9fe6SPaolo Bonzini     job->speed = speed;
1112f0c9fe6SPaolo Bonzini }
1122f0c9fe6SPaolo Bonzini 
113aeae883bSPaolo Bonzini void block_job_complete(BlockJob *job, Error **errp)
114aeae883bSPaolo Bonzini {
115751ebd76SFam Zheng     if (job->pause_count || job->cancelled || !job->driver->complete) {
116c6bd8c70SMarkus Armbruster         error_setg(errp, QERR_BLOCK_JOB_NOT_READY,
117bfb197e0SMarkus Armbruster                    bdrv_get_device_name(job->bs));
118aeae883bSPaolo Bonzini         return;
119aeae883bSPaolo Bonzini     }
120aeae883bSPaolo Bonzini 
1213fc4b10aSFam Zheng     job->driver->complete(job, errp);
122aeae883bSPaolo Bonzini }
123aeae883bSPaolo Bonzini 
1248acc72a4SPaolo Bonzini void block_job_pause(BlockJob *job)
1252f0c9fe6SPaolo Bonzini {
126751ebd76SFam Zheng     job->pause_count++;
1278acc72a4SPaolo Bonzini }
1288acc72a4SPaolo Bonzini 
1298acc72a4SPaolo Bonzini bool block_job_is_paused(BlockJob *job)
1308acc72a4SPaolo Bonzini {
131751ebd76SFam Zheng     return job->pause_count > 0;
1328acc72a4SPaolo Bonzini }
1338acc72a4SPaolo Bonzini 
1348acc72a4SPaolo Bonzini void block_job_resume(BlockJob *job)
1358acc72a4SPaolo Bonzini {
136751ebd76SFam Zheng     assert(job->pause_count > 0);
137751ebd76SFam Zheng     job->pause_count--;
138751ebd76SFam Zheng     if (job->pause_count) {
139751ebd76SFam Zheng         return;
140751ebd76SFam Zheng     }
141751ebd76SFam Zheng     block_job_enter(job);
142751ebd76SFam Zheng }
143751ebd76SFam Zheng 
144751ebd76SFam Zheng void block_job_enter(BlockJob *job)
145751ebd76SFam Zheng {
14632c81a4aSPaolo Bonzini     block_job_iostatus_reset(job);
1472f0c9fe6SPaolo Bonzini     if (job->co && !job->busy) {
1482f0c9fe6SPaolo Bonzini         qemu_coroutine_enter(job->co, NULL);
1492f0c9fe6SPaolo Bonzini     }
1502f0c9fe6SPaolo Bonzini }
1512f0c9fe6SPaolo Bonzini 
1528acc72a4SPaolo Bonzini void block_job_cancel(BlockJob *job)
1538acc72a4SPaolo Bonzini {
1548acc72a4SPaolo Bonzini     job->cancelled = true;
155751ebd76SFam Zheng     block_job_enter(job);
1568acc72a4SPaolo Bonzini }
1578acc72a4SPaolo Bonzini 
1582f0c9fe6SPaolo Bonzini bool block_job_is_cancelled(BlockJob *job)
1592f0c9fe6SPaolo Bonzini {
1602f0c9fe6SPaolo Bonzini     return job->cancelled;
1612f0c9fe6SPaolo Bonzini }
1622f0c9fe6SPaolo Bonzini 
16332c81a4aSPaolo Bonzini void block_job_iostatus_reset(BlockJob *job)
16432c81a4aSPaolo Bonzini {
16532c81a4aSPaolo Bonzini     job->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
1663fc4b10aSFam Zheng     if (job->driver->iostatus_reset) {
1673fc4b10aSFam Zheng         job->driver->iostatus_reset(job);
1683bd293c3SPaolo Bonzini     }
16932c81a4aSPaolo Bonzini }
17032c81a4aSPaolo Bonzini 
171345f9e1bSMax Reitz struct BlockFinishData {
1722f0c9fe6SPaolo Bonzini     BlockJob *job;
173097310b5SMarkus Armbruster     BlockCompletionFunc *cb;
1742f0c9fe6SPaolo Bonzini     void *opaque;
1752f0c9fe6SPaolo Bonzini     bool cancelled;
1762f0c9fe6SPaolo Bonzini     int ret;
1772f0c9fe6SPaolo Bonzini };
1782f0c9fe6SPaolo Bonzini 
179345f9e1bSMax Reitz static void block_job_finish_cb(void *opaque, int ret)
1802f0c9fe6SPaolo Bonzini {
181345f9e1bSMax Reitz     struct BlockFinishData *data = opaque;
1822f0c9fe6SPaolo Bonzini 
1832f0c9fe6SPaolo Bonzini     data->cancelled = block_job_is_cancelled(data->job);
1842f0c9fe6SPaolo Bonzini     data->ret = ret;
1852f0c9fe6SPaolo Bonzini     data->cb(data->opaque, ret);
1862f0c9fe6SPaolo Bonzini }
1872f0c9fe6SPaolo Bonzini 
188345f9e1bSMax Reitz static int block_job_finish_sync(BlockJob *job,
189345f9e1bSMax Reitz                                  void (*finish)(BlockJob *, Error **errp),
190345f9e1bSMax Reitz                                  Error **errp)
1912f0c9fe6SPaolo Bonzini {
192345f9e1bSMax Reitz     struct BlockFinishData data;
1932f0c9fe6SPaolo Bonzini     BlockDriverState *bs = job->bs;
194345f9e1bSMax Reitz     Error *local_err = NULL;
1952f0c9fe6SPaolo Bonzini 
1962f0c9fe6SPaolo Bonzini     assert(bs->job == job);
1972f0c9fe6SPaolo Bonzini 
1982f0c9fe6SPaolo Bonzini     /* Set up our own callback to store the result and chain to
1992f0c9fe6SPaolo Bonzini      * the original callback.
2002f0c9fe6SPaolo Bonzini      */
2012f0c9fe6SPaolo Bonzini     data.job = job;
2022f0c9fe6SPaolo Bonzini     data.cb = job->cb;
2032f0c9fe6SPaolo Bonzini     data.opaque = job->opaque;
2042f0c9fe6SPaolo Bonzini     data.ret = -EINPROGRESS;
205345f9e1bSMax Reitz     job->cb = block_job_finish_cb;
2062f0c9fe6SPaolo Bonzini     job->opaque = &data;
207345f9e1bSMax Reitz     finish(job, &local_err);
208345f9e1bSMax Reitz     if (local_err) {
209345f9e1bSMax Reitz         error_propagate(errp, local_err);
210345f9e1bSMax Reitz         return -EBUSY;
211345f9e1bSMax Reitz     }
2122f0c9fe6SPaolo Bonzini     while (data.ret == -EINPROGRESS) {
213b47ec2c4SPaolo Bonzini         aio_poll(bdrv_get_aio_context(bs), true);
2142f0c9fe6SPaolo Bonzini     }
2152f0c9fe6SPaolo Bonzini     return (data.cancelled && data.ret == 0) ? -ECANCELED : data.ret;
2162f0c9fe6SPaolo Bonzini }
2172f0c9fe6SPaolo Bonzini 
218345f9e1bSMax Reitz /* A wrapper around block_job_cancel() taking an Error ** parameter so it may be
219345f9e1bSMax Reitz  * used with block_job_finish_sync() without the need for (rather nasty)
220345f9e1bSMax Reitz  * function pointer casts there. */
221345f9e1bSMax Reitz static void block_job_cancel_err(BlockJob *job, Error **errp)
222345f9e1bSMax Reitz {
223345f9e1bSMax Reitz     block_job_cancel(job);
224345f9e1bSMax Reitz }
225345f9e1bSMax Reitz 
226345f9e1bSMax Reitz int block_job_cancel_sync(BlockJob *job)
227345f9e1bSMax Reitz {
228345f9e1bSMax Reitz     return block_job_finish_sync(job, &block_job_cancel_err, NULL);
229345f9e1bSMax Reitz }
230345f9e1bSMax Reitz 
231345f9e1bSMax Reitz int block_job_complete_sync(BlockJob *job, Error **errp)
232345f9e1bSMax Reitz {
233345f9e1bSMax Reitz     return block_job_finish_sync(job, &block_job_complete, errp);
234345f9e1bSMax Reitz }
235345f9e1bSMax Reitz 
2367483d1e5SAlex Bligh void block_job_sleep_ns(BlockJob *job, QEMUClockType type, int64_t ns)
2372f0c9fe6SPaolo Bonzini {
2388acc72a4SPaolo Bonzini     assert(job->busy);
2398acc72a4SPaolo Bonzini 
2402f0c9fe6SPaolo Bonzini     /* Check cancellation *before* setting busy = false, too!  */
2418acc72a4SPaolo Bonzini     if (block_job_is_cancelled(job)) {
2428acc72a4SPaolo Bonzini         return;
2432f0c9fe6SPaolo Bonzini     }
2448acc72a4SPaolo Bonzini 
2458acc72a4SPaolo Bonzini     job->busy = false;
2468acc72a4SPaolo Bonzini     if (block_job_is_paused(job)) {
2478acc72a4SPaolo Bonzini         qemu_coroutine_yield();
2488acc72a4SPaolo Bonzini     } else {
2490b9caf9bSFam Zheng         co_aio_sleep_ns(bdrv_get_aio_context(job->bs), type, ns);
2508acc72a4SPaolo Bonzini     }
2518acc72a4SPaolo Bonzini     job->busy = true;
2522f0c9fe6SPaolo Bonzini }
25330e628b7SPaolo Bonzini 
254dc71ce45SFam Zheng void block_job_yield(BlockJob *job)
255dc71ce45SFam Zheng {
256dc71ce45SFam Zheng     assert(job->busy);
257dc71ce45SFam Zheng 
258dc71ce45SFam Zheng     /* Check cancellation *before* setting busy = false, too!  */
259dc71ce45SFam Zheng     if (block_job_is_cancelled(job)) {
260dc71ce45SFam Zheng         return;
261dc71ce45SFam Zheng     }
262dc71ce45SFam Zheng 
263dc71ce45SFam Zheng     job->busy = false;
264dc71ce45SFam Zheng     qemu_coroutine_yield();
265dc71ce45SFam Zheng     job->busy = true;
266dc71ce45SFam Zheng }
267dc71ce45SFam Zheng 
26830e628b7SPaolo Bonzini BlockJobInfo *block_job_query(BlockJob *job)
26930e628b7SPaolo Bonzini {
27030e628b7SPaolo Bonzini     BlockJobInfo *info = g_new0(BlockJobInfo, 1);
27179e14bf7SFam Zheng     info->type      = g_strdup(BlockJobType_lookup[job->driver->job_type]);
27230e628b7SPaolo Bonzini     info->device    = g_strdup(bdrv_get_device_name(job->bs));
27330e628b7SPaolo Bonzini     info->len       = job->len;
2748d65883fSPaolo Bonzini     info->busy      = job->busy;
275751ebd76SFam Zheng     info->paused    = job->pause_count > 0;
27630e628b7SPaolo Bonzini     info->offset    = job->offset;
27730e628b7SPaolo Bonzini     info->speed     = job->speed;
27832c81a4aSPaolo Bonzini     info->io_status = job->iostatus;
279ef6dbf1eSMax Reitz     info->ready     = job->ready;
28030e628b7SPaolo Bonzini     return info;
28130e628b7SPaolo Bonzini }
28232c81a4aSPaolo Bonzini 
28332c81a4aSPaolo Bonzini static void block_job_iostatus_set_err(BlockJob *job, int error)
28432c81a4aSPaolo Bonzini {
28532c81a4aSPaolo Bonzini     if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
28632c81a4aSPaolo Bonzini         job->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
28732c81a4aSPaolo Bonzini                                           BLOCK_DEVICE_IO_STATUS_FAILED;
28832c81a4aSPaolo Bonzini     }
28932c81a4aSPaolo Bonzini }
29032c81a4aSPaolo Bonzini 
291bcada37bSWenchao Xia void block_job_event_cancelled(BlockJob *job)
292a66a2a36SPaolo Bonzini {
293bcada37bSWenchao Xia     qapi_event_send_block_job_cancelled(job->driver->job_type,
294a66a2a36SPaolo Bonzini                                         bdrv_get_device_name(job->bs),
295a66a2a36SPaolo Bonzini                                         job->len,
296a66a2a36SPaolo Bonzini                                         job->offset,
297bcada37bSWenchao Xia                                         job->speed,
298bcada37bSWenchao Xia                                         &error_abort);
299a66a2a36SPaolo Bonzini }
300a66a2a36SPaolo Bonzini 
301bcada37bSWenchao Xia void block_job_event_completed(BlockJob *job, const char *msg)
302a66a2a36SPaolo Bonzini {
303bcada37bSWenchao Xia     qapi_event_send_block_job_completed(job->driver->job_type,
304bcada37bSWenchao Xia                                         bdrv_get_device_name(job->bs),
305bcada37bSWenchao Xia                                         job->len,
306bcada37bSWenchao Xia                                         job->offset,
307bcada37bSWenchao Xia                                         job->speed,
308bcada37bSWenchao Xia                                         !!msg,
309bcada37bSWenchao Xia                                         msg,
310bcada37bSWenchao Xia                                         &error_abort);
311bcada37bSWenchao Xia }
312bcada37bSWenchao Xia 
313bcada37bSWenchao Xia void block_job_event_ready(BlockJob *job)
314bcada37bSWenchao Xia {
315ef6dbf1eSMax Reitz     job->ready = true;
316ef6dbf1eSMax Reitz 
317518848a2SMarkus Armbruster     qapi_event_send_block_job_ready(job->driver->job_type,
318518848a2SMarkus Armbruster                                     bdrv_get_device_name(job->bs),
319518848a2SMarkus Armbruster                                     job->len,
320518848a2SMarkus Armbruster                                     job->offset,
321518848a2SMarkus Armbruster                                     job->speed, &error_abort);
322a66a2a36SPaolo Bonzini }
323a66a2a36SPaolo Bonzini 
32432c81a4aSPaolo Bonzini BlockErrorAction block_job_error_action(BlockJob *job, BlockDriverState *bs,
32532c81a4aSPaolo Bonzini                                         BlockdevOnError on_err,
32632c81a4aSPaolo Bonzini                                         int is_read, int error)
32732c81a4aSPaolo Bonzini {
32832c81a4aSPaolo Bonzini     BlockErrorAction action;
32932c81a4aSPaolo Bonzini 
33032c81a4aSPaolo Bonzini     switch (on_err) {
33132c81a4aSPaolo Bonzini     case BLOCKDEV_ON_ERROR_ENOSPC:
332a589569fSWenchao Xia         action = (error == ENOSPC) ?
333a589569fSWenchao Xia                  BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT;
33432c81a4aSPaolo Bonzini         break;
33532c81a4aSPaolo Bonzini     case BLOCKDEV_ON_ERROR_STOP:
336a589569fSWenchao Xia         action = BLOCK_ERROR_ACTION_STOP;
33732c81a4aSPaolo Bonzini         break;
33832c81a4aSPaolo Bonzini     case BLOCKDEV_ON_ERROR_REPORT:
339a589569fSWenchao Xia         action = BLOCK_ERROR_ACTION_REPORT;
34032c81a4aSPaolo Bonzini         break;
34132c81a4aSPaolo Bonzini     case BLOCKDEV_ON_ERROR_IGNORE:
342a589569fSWenchao Xia         action = BLOCK_ERROR_ACTION_IGNORE;
34332c81a4aSPaolo Bonzini         break;
34432c81a4aSPaolo Bonzini     default:
34532c81a4aSPaolo Bonzini         abort();
34632c81a4aSPaolo Bonzini     }
347823c6863SMarkus Armbruster     qapi_event_send_block_job_error(bdrv_get_device_name(job->bs),
3485a2d2cbdSWenchao Xia                                     is_read ? IO_OPERATION_TYPE_READ :
3495a2d2cbdSWenchao Xia                                     IO_OPERATION_TYPE_WRITE,
3505a2d2cbdSWenchao Xia                                     action, &error_abort);
351a589569fSWenchao Xia     if (action == BLOCK_ERROR_ACTION_STOP) {
352751ebd76SFam Zheng         /* make the pause user visible, which will be resumed from QMP. */
353751ebd76SFam Zheng         job->user_paused = true;
35432c81a4aSPaolo Bonzini         block_job_pause(job);
35532c81a4aSPaolo Bonzini         block_job_iostatus_set_err(job, error);
35632c81a4aSPaolo Bonzini         if (bs != job->bs) {
35732c81a4aSPaolo Bonzini             bdrv_iostatus_set_err(bs, error);
35832c81a4aSPaolo Bonzini         }
35932c81a4aSPaolo Bonzini     }
36032c81a4aSPaolo Bonzini     return action;
36132c81a4aSPaolo Bonzini }
362dec7d421SStefan Hajnoczi 
363dec7d421SStefan Hajnoczi typedef struct {
364dec7d421SStefan Hajnoczi     BlockJob *job;
365dec7d421SStefan Hajnoczi     QEMUBH *bh;
366dec7d421SStefan Hajnoczi     AioContext *aio_context;
367dec7d421SStefan Hajnoczi     BlockJobDeferToMainLoopFn *fn;
368dec7d421SStefan Hajnoczi     void *opaque;
369dec7d421SStefan Hajnoczi } BlockJobDeferToMainLoopData;
370dec7d421SStefan Hajnoczi 
371dec7d421SStefan Hajnoczi static void block_job_defer_to_main_loop_bh(void *opaque)
372dec7d421SStefan Hajnoczi {
373dec7d421SStefan Hajnoczi     BlockJobDeferToMainLoopData *data = opaque;
374dec7d421SStefan Hajnoczi     AioContext *aio_context;
375dec7d421SStefan Hajnoczi 
376dec7d421SStefan Hajnoczi     qemu_bh_delete(data->bh);
377dec7d421SStefan Hajnoczi 
378dec7d421SStefan Hajnoczi     /* Prevent race with block_job_defer_to_main_loop() */
379dec7d421SStefan Hajnoczi     aio_context_acquire(data->aio_context);
380dec7d421SStefan Hajnoczi 
381dec7d421SStefan Hajnoczi     /* Fetch BDS AioContext again, in case it has changed */
382dec7d421SStefan Hajnoczi     aio_context = bdrv_get_aio_context(data->job->bs);
383dec7d421SStefan Hajnoczi     aio_context_acquire(aio_context);
384dec7d421SStefan Hajnoczi 
385dec7d421SStefan Hajnoczi     data->fn(data->job, data->opaque);
386dec7d421SStefan Hajnoczi 
387dec7d421SStefan Hajnoczi     aio_context_release(aio_context);
388dec7d421SStefan Hajnoczi 
389dec7d421SStefan Hajnoczi     aio_context_release(data->aio_context);
390dec7d421SStefan Hajnoczi 
391dec7d421SStefan Hajnoczi     g_free(data);
392dec7d421SStefan Hajnoczi }
393dec7d421SStefan Hajnoczi 
394dec7d421SStefan Hajnoczi void block_job_defer_to_main_loop(BlockJob *job,
395dec7d421SStefan Hajnoczi                                   BlockJobDeferToMainLoopFn *fn,
396dec7d421SStefan Hajnoczi                                   void *opaque)
397dec7d421SStefan Hajnoczi {
398dec7d421SStefan Hajnoczi     BlockJobDeferToMainLoopData *data = g_malloc(sizeof(*data));
399dec7d421SStefan Hajnoczi     data->job = job;
400dec7d421SStefan Hajnoczi     data->bh = qemu_bh_new(block_job_defer_to_main_loop_bh, data);
401dec7d421SStefan Hajnoczi     data->aio_context = bdrv_get_aio_context(job->bs);
402dec7d421SStefan Hajnoczi     data->fn = fn;
403dec7d421SStefan Hajnoczi     data->opaque = opaque;
404dec7d421SStefan Hajnoczi 
405dec7d421SStefan Hajnoczi     qemu_bh_schedule(data->bh);
406dec7d421SStefan Hajnoczi }
407