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 26d38ea87aSPeter Maydell #include "qemu/osdep.h" 272f0c9fe6SPaolo Bonzini #include "qemu-common.h" 28737e150eSPaolo Bonzini #include "block/block.h" 29c87621eaSJohn Snow #include "block/blockjob_int.h" 30737e150eSPaolo Bonzini #include "block/block_int.h" 31c9de4050SJohn Snow #include "block/trace.h" 32373340b2SMax Reitz #include "sysemu/block-backend.h" 33e688df6bSMarkus Armbruster #include "qapi/error.h" 349af23989SMarkus Armbruster #include "qapi/qapi-events-block-core.h" 35cc7a8ea7SMarkus Armbruster #include "qapi/qmp/qerror.h" 3610817bf0SDaniel P. Berrange #include "qemu/coroutine.h" 371de7afc9SPaolo Bonzini #include "qemu/timer.h" 382f0c9fe6SPaolo Bonzini 39fc24908eSPaolo Bonzini /* Right now, this mutex is only needed to synchronize accesses to job->busy 40fc24908eSPaolo Bonzini * and job->sleep_timer, such as concurrent calls to block_job_do_yield and 41fc24908eSPaolo Bonzini * block_job_enter. */ 42fc24908eSPaolo Bonzini static QemuMutex block_job_mutex; 43fc24908eSPaolo Bonzini 44fc24908eSPaolo Bonzini static void block_job_lock(void) 45fc24908eSPaolo Bonzini { 46fc24908eSPaolo Bonzini qemu_mutex_lock(&block_job_mutex); 47fc24908eSPaolo Bonzini } 48fc24908eSPaolo Bonzini 49fc24908eSPaolo Bonzini static void block_job_unlock(void) 50fc24908eSPaolo Bonzini { 51fc24908eSPaolo Bonzini qemu_mutex_unlock(&block_job_mutex); 52fc24908eSPaolo Bonzini } 53fc24908eSPaolo Bonzini 54fc24908eSPaolo Bonzini static void __attribute__((__constructor__)) block_job_init(void) 55fc24908eSPaolo Bonzini { 56fc24908eSPaolo Bonzini qemu_mutex_init(&block_job_mutex); 57fc24908eSPaolo Bonzini } 58fc24908eSPaolo Bonzini 598254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job); 608254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg); 615f241594SJohn Snow static int block_job_event_pending(BlockJob *job); 62aa9ef2e6SJohn Snow static void block_job_enter_cond(BlockJob *job, bool(*fn)(BlockJob *job)); 638254b6d9SJohn Snow 64c55a832fSFam Zheng /* Transactional group of block jobs */ 65c55a832fSFam Zheng struct BlockJobTxn { 66c55a832fSFam Zheng 67c55a832fSFam Zheng /* Is this txn being cancelled? */ 68c55a832fSFam Zheng bool aborting; 69c55a832fSFam Zheng 70c55a832fSFam Zheng /* List of jobs */ 71c55a832fSFam Zheng QLIST_HEAD(, BlockJob) jobs; 72c55a832fSFam Zheng 73c55a832fSFam Zheng /* Reference count */ 74c55a832fSFam Zheng int refcnt; 75c55a832fSFam Zheng }; 76c55a832fSFam Zheng 7788691b37SPaolo Bonzini /* 7888691b37SPaolo Bonzini * The block job API is composed of two categories of functions. 7988691b37SPaolo Bonzini * 8088691b37SPaolo Bonzini * The first includes functions used by the monitor. The monitor is 8188691b37SPaolo Bonzini * peculiar in that it accesses the block job list with block_job_get, and 8288691b37SPaolo Bonzini * therefore needs consistency across block_job_get and the actual operation 8388691b37SPaolo Bonzini * (e.g. block_job_set_speed). The consistency is achieved with 8488691b37SPaolo Bonzini * aio_context_acquire/release. These functions are declared in blockjob.h. 8588691b37SPaolo Bonzini * 8688691b37SPaolo Bonzini * The second includes functions used by the block job drivers and sometimes 8788691b37SPaolo Bonzini * by the core block layer. These do not care about locking, because the 8888691b37SPaolo Bonzini * whole coroutine runs under the AioContext lock, and are declared in 8988691b37SPaolo Bonzini * blockjob_int.h. 9088691b37SPaolo Bonzini */ 9188691b37SPaolo Bonzini 92e7c1d78bSKevin Wolf static bool is_block_job(Job *job) 93a7112795SAlberto Garcia { 94e7c1d78bSKevin Wolf return job_type(job) == JOB_TYPE_BACKUP || 95e7c1d78bSKevin Wolf job_type(job) == JOB_TYPE_COMMIT || 96e7c1d78bSKevin Wolf job_type(job) == JOB_TYPE_MIRROR || 97e7c1d78bSKevin Wolf job_type(job) == JOB_TYPE_STREAM; 98a7112795SAlberto Garcia } 99e7c1d78bSKevin Wolf 100e7c1d78bSKevin Wolf BlockJob *block_job_next(BlockJob *bjob) 101e7c1d78bSKevin Wolf { 102e7c1d78bSKevin Wolf Job *job = bjob ? &bjob->job : NULL; 103e7c1d78bSKevin Wolf 104e7c1d78bSKevin Wolf do { 105e7c1d78bSKevin Wolf job = job_next(job); 106e7c1d78bSKevin Wolf } while (job && !is_block_job(job)); 107e7c1d78bSKevin Wolf 108e7c1d78bSKevin Wolf return job ? container_of(job, BlockJob, job) : NULL; 109a7112795SAlberto Garcia } 110a7112795SAlberto Garcia 111ffb1f10cSAlberto Garcia BlockJob *block_job_get(const char *id) 112ffb1f10cSAlberto Garcia { 113e7c1d78bSKevin Wolf Job *job = job_get(id); 114ffb1f10cSAlberto Garcia 115e7c1d78bSKevin Wolf if (job && is_block_job(job)) { 116e7c1d78bSKevin Wolf return container_of(job, BlockJob, job); 117e7c1d78bSKevin Wolf } else { 118ffb1f10cSAlberto Garcia return NULL; 119ffb1f10cSAlberto Garcia } 120e7c1d78bSKevin Wolf } 121ffb1f10cSAlberto Garcia 122c8ab5c2dSPaolo Bonzini BlockJobTxn *block_job_txn_new(void) 123c8ab5c2dSPaolo Bonzini { 124c8ab5c2dSPaolo Bonzini BlockJobTxn *txn = g_new0(BlockJobTxn, 1); 125c8ab5c2dSPaolo Bonzini QLIST_INIT(&txn->jobs); 126c8ab5c2dSPaolo Bonzini txn->refcnt = 1; 127c8ab5c2dSPaolo Bonzini return txn; 128c8ab5c2dSPaolo Bonzini } 129c8ab5c2dSPaolo Bonzini 130c8ab5c2dSPaolo Bonzini static void block_job_txn_ref(BlockJobTxn *txn) 131c8ab5c2dSPaolo Bonzini { 132c8ab5c2dSPaolo Bonzini txn->refcnt++; 133c8ab5c2dSPaolo Bonzini } 134c8ab5c2dSPaolo Bonzini 135c8ab5c2dSPaolo Bonzini void block_job_txn_unref(BlockJobTxn *txn) 136c8ab5c2dSPaolo Bonzini { 137c8ab5c2dSPaolo Bonzini if (txn && --txn->refcnt == 0) { 138c8ab5c2dSPaolo Bonzini g_free(txn); 139c8ab5c2dSPaolo Bonzini } 140c8ab5c2dSPaolo Bonzini } 141c8ab5c2dSPaolo Bonzini 142c8ab5c2dSPaolo Bonzini void block_job_txn_add_job(BlockJobTxn *txn, BlockJob *job) 143c8ab5c2dSPaolo Bonzini { 144c8ab5c2dSPaolo Bonzini if (!txn) { 145c8ab5c2dSPaolo Bonzini return; 146c8ab5c2dSPaolo Bonzini } 147c8ab5c2dSPaolo Bonzini 148c8ab5c2dSPaolo Bonzini assert(!job->txn); 149c8ab5c2dSPaolo Bonzini job->txn = txn; 150c8ab5c2dSPaolo Bonzini 151c8ab5c2dSPaolo Bonzini QLIST_INSERT_HEAD(&txn->jobs, job, txn_list); 152c8ab5c2dSPaolo Bonzini block_job_txn_ref(txn); 153c8ab5c2dSPaolo Bonzini } 154c8ab5c2dSPaolo Bonzini 155a865cebbSMarc-André Lureau static void block_job_txn_del_job(BlockJob *job) 156a865cebbSMarc-André Lureau { 157a865cebbSMarc-André Lureau if (job->txn) { 158a865cebbSMarc-André Lureau QLIST_REMOVE(job, txn_list); 159a865cebbSMarc-André Lureau block_job_txn_unref(job->txn); 160a865cebbSMarc-André Lureau job->txn = NULL; 161a865cebbSMarc-André Lureau } 162a865cebbSMarc-André Lureau } 163a865cebbSMarc-André Lureau 1644c7e813cSStefan Hajnoczi /* Assumes the block_job_mutex is held */ 1654c7e813cSStefan Hajnoczi static bool block_job_timer_pending(BlockJob *job) 1664c7e813cSStefan Hajnoczi { 1674c7e813cSStefan Hajnoczi return timer_pending(&job->sleep_timer); 1684c7e813cSStefan Hajnoczi } 1694c7e813cSStefan Hajnoczi 1704c7e813cSStefan Hajnoczi /* Assumes the block_job_mutex is held */ 1714c7e813cSStefan Hajnoczi static bool block_job_timer_not_pending(BlockJob *job) 1724c7e813cSStefan Hajnoczi { 1734c7e813cSStefan Hajnoczi return !block_job_timer_pending(job); 1744c7e813cSStefan Hajnoczi } 1754c7e813cSStefan Hajnoczi 176f321dcb5SPaolo Bonzini static void block_job_pause(BlockJob *job) 177f321dcb5SPaolo Bonzini { 178f321dcb5SPaolo Bonzini job->pause_count++; 179f321dcb5SPaolo Bonzini } 180f321dcb5SPaolo Bonzini 181f321dcb5SPaolo Bonzini static void block_job_resume(BlockJob *job) 182f321dcb5SPaolo Bonzini { 183f321dcb5SPaolo Bonzini assert(job->pause_count > 0); 184f321dcb5SPaolo Bonzini job->pause_count--; 185f321dcb5SPaolo Bonzini if (job->pause_count) { 186f321dcb5SPaolo Bonzini return; 187f321dcb5SPaolo Bonzini } 1884c7e813cSStefan Hajnoczi 1894c7e813cSStefan Hajnoczi /* kick only if no timer is pending */ 1904c7e813cSStefan Hajnoczi block_job_enter_cond(job, block_job_timer_not_pending); 191f321dcb5SPaolo Bonzini } 192f321dcb5SPaolo Bonzini 19305b0d8e3SPaolo Bonzini static void block_job_attached_aio_context(AioContext *new_context, 19405b0d8e3SPaolo Bonzini void *opaque); 19505b0d8e3SPaolo Bonzini static void block_job_detach_aio_context(void *opaque); 19605b0d8e3SPaolo Bonzini 197*80fa2c75SKevin Wolf void block_job_free(Job *job) 19805b0d8e3SPaolo Bonzini { 199*80fa2c75SKevin Wolf BlockJob *bjob = container_of(job, BlockJob, job); 200*80fa2c75SKevin Wolf BlockDriverState *bs = blk_bs(bjob->blk); 201*80fa2c75SKevin Wolf 202*80fa2c75SKevin Wolf assert(!bjob->txn); 203*80fa2c75SKevin Wolf 20405b0d8e3SPaolo Bonzini bs->job = NULL; 205*80fa2c75SKevin Wolf block_job_remove_all_bdrv(bjob); 206*80fa2c75SKevin Wolf blk_remove_aio_context_notifier(bjob->blk, 20705b0d8e3SPaolo Bonzini block_job_attached_aio_context, 208*80fa2c75SKevin Wolf block_job_detach_aio_context, bjob); 209*80fa2c75SKevin Wolf blk_unref(bjob->blk); 210*80fa2c75SKevin Wolf error_free(bjob->blocker); 211*80fa2c75SKevin Wolf assert(!timer_pending(&bjob->sleep_timer)); 21205b0d8e3SPaolo Bonzini } 21305b0d8e3SPaolo Bonzini 214463e0be1SStefan Hajnoczi static void block_job_attached_aio_context(AioContext *new_context, 215463e0be1SStefan Hajnoczi void *opaque) 216463e0be1SStefan Hajnoczi { 217463e0be1SStefan Hajnoczi BlockJob *job = opaque; 218463e0be1SStefan Hajnoczi 219463e0be1SStefan Hajnoczi if (job->driver->attached_aio_context) { 220463e0be1SStefan Hajnoczi job->driver->attached_aio_context(job, new_context); 221463e0be1SStefan Hajnoczi } 222463e0be1SStefan Hajnoczi 223463e0be1SStefan Hajnoczi block_job_resume(job); 224463e0be1SStefan Hajnoczi } 225463e0be1SStefan Hajnoczi 226bae8196dSPaolo Bonzini static void block_job_drain(BlockJob *job) 227bae8196dSPaolo Bonzini { 228bae8196dSPaolo Bonzini /* If job is !job->busy this kicks it into the next pause point. */ 229bae8196dSPaolo Bonzini block_job_enter(job); 230bae8196dSPaolo Bonzini 231bae8196dSPaolo Bonzini blk_drain(job->blk); 232bae8196dSPaolo Bonzini if (job->driver->drain) { 233bae8196dSPaolo Bonzini job->driver->drain(job); 234bae8196dSPaolo Bonzini } 235bae8196dSPaolo Bonzini } 236bae8196dSPaolo Bonzini 237463e0be1SStefan Hajnoczi static void block_job_detach_aio_context(void *opaque) 238463e0be1SStefan Hajnoczi { 239463e0be1SStefan Hajnoczi BlockJob *job = opaque; 240463e0be1SStefan Hajnoczi 241463e0be1SStefan Hajnoczi /* In case the job terminates during aio_poll()... */ 242*80fa2c75SKevin Wolf job_ref(&job->job); 243463e0be1SStefan Hajnoczi 244463e0be1SStefan Hajnoczi block_job_pause(job); 245463e0be1SStefan Hajnoczi 246463e0be1SStefan Hajnoczi while (!job->paused && !job->completed) { 247bae8196dSPaolo Bonzini block_job_drain(job); 248463e0be1SStefan Hajnoczi } 249463e0be1SStefan Hajnoczi 250*80fa2c75SKevin Wolf job_unref(&job->job); 251463e0be1SStefan Hajnoczi } 252463e0be1SStefan Hajnoczi 253f321dcb5SPaolo Bonzini static char *child_job_get_parent_desc(BdrvChild *c) 254f321dcb5SPaolo Bonzini { 255f321dcb5SPaolo Bonzini BlockJob *job = c->opaque; 256252291eaSKevin Wolf return g_strdup_printf("%s job '%s'", job_type_str(&job->job), job->job.id); 257f321dcb5SPaolo Bonzini } 258f321dcb5SPaolo Bonzini 259ad90febaSKevin Wolf static void child_job_drained_begin(BdrvChild *c) 260f321dcb5SPaolo Bonzini { 261ad90febaSKevin Wolf BlockJob *job = c->opaque; 262f321dcb5SPaolo Bonzini block_job_pause(job); 263f321dcb5SPaolo Bonzini } 264f321dcb5SPaolo Bonzini 265ad90febaSKevin Wolf static void child_job_drained_end(BdrvChild *c) 266f321dcb5SPaolo Bonzini { 267ad90febaSKevin Wolf BlockJob *job = c->opaque; 268f321dcb5SPaolo Bonzini block_job_resume(job); 269f321dcb5SPaolo Bonzini } 270f321dcb5SPaolo Bonzini 271ad90febaSKevin Wolf static const BdrvChildRole child_job = { 272ad90febaSKevin Wolf .get_parent_desc = child_job_get_parent_desc, 273ad90febaSKevin Wolf .drained_begin = child_job_drained_begin, 274ad90febaSKevin Wolf .drained_end = child_job_drained_end, 275ad90febaSKevin Wolf .stay_at_node = true, 276f321dcb5SPaolo Bonzini }; 277f321dcb5SPaolo Bonzini 278bbc02b90SKevin Wolf void block_job_remove_all_bdrv(BlockJob *job) 279bbc02b90SKevin Wolf { 280bbc02b90SKevin Wolf GSList *l; 281bbc02b90SKevin Wolf for (l = job->nodes; l; l = l->next) { 282bbc02b90SKevin Wolf BdrvChild *c = l->data; 283bbc02b90SKevin Wolf bdrv_op_unblock_all(c->bs, job->blocker); 284bbc02b90SKevin Wolf bdrv_root_unref_child(c); 285bbc02b90SKevin Wolf } 286bbc02b90SKevin Wolf g_slist_free(job->nodes); 287bbc02b90SKevin Wolf job->nodes = NULL; 288bbc02b90SKevin Wolf } 289bbc02b90SKevin Wolf 29076d554e2SKevin Wolf int block_job_add_bdrv(BlockJob *job, const char *name, BlockDriverState *bs, 29176d554e2SKevin Wolf uint64_t perm, uint64_t shared_perm, Error **errp) 29223d402d4SAlberto Garcia { 29376d554e2SKevin Wolf BdrvChild *c; 29476d554e2SKevin Wolf 29576d554e2SKevin Wolf c = bdrv_root_attach_child(bs, name, &child_job, perm, shared_perm, 29676d554e2SKevin Wolf job, errp); 29776d554e2SKevin Wolf if (c == NULL) { 29876d554e2SKevin Wolf return -EPERM; 29976d554e2SKevin Wolf } 30076d554e2SKevin Wolf 30176d554e2SKevin Wolf job->nodes = g_slist_prepend(job->nodes, c); 30223d402d4SAlberto Garcia bdrv_ref(bs); 30323d402d4SAlberto Garcia bdrv_op_block_all(bs, job->blocker); 30476d554e2SKevin Wolf 30576d554e2SKevin Wolf return 0; 30623d402d4SAlberto Garcia } 30723d402d4SAlberto Garcia 308559b935fSJohn Snow bool block_job_is_internal(BlockJob *job) 309559b935fSJohn Snow { 31033e9e9bdSKevin Wolf return (job->job.id == NULL); 311559b935fSJohn Snow } 312559b935fSJohn Snow 3135ccac6f1SJohn Snow static bool block_job_started(BlockJob *job) 3145ccac6f1SJohn Snow { 3155ccac6f1SJohn Snow return job->co; 3165ccac6f1SJohn Snow } 3175ccac6f1SJohn Snow 318bd21935bSKevin Wolf const BlockJobDriver *block_job_driver(BlockJob *job) 319bd21935bSKevin Wolf { 320bd21935bSKevin Wolf return job->driver; 321bd21935bSKevin Wolf } 322bd21935bSKevin Wolf 323e3796a24SJohn Snow /** 324e3796a24SJohn Snow * All jobs must allow a pause point before entering their job proper. This 325e3796a24SJohn Snow * ensures that jobs can be paused prior to being started, then resumed later. 326e3796a24SJohn Snow */ 327e3796a24SJohn Snow static void coroutine_fn block_job_co_entry(void *opaque) 328e3796a24SJohn Snow { 329e3796a24SJohn Snow BlockJob *job = opaque; 330e3796a24SJohn Snow 331e3796a24SJohn Snow assert(job && job->driver && job->driver->start); 332e3796a24SJohn Snow block_job_pause_point(job); 333e3796a24SJohn Snow job->driver->start(job); 334e3796a24SJohn Snow } 335e3796a24SJohn Snow 336fc24908eSPaolo Bonzini static void block_job_sleep_timer_cb(void *opaque) 337fc24908eSPaolo Bonzini { 338fc24908eSPaolo Bonzini BlockJob *job = opaque; 339fc24908eSPaolo Bonzini 340fc24908eSPaolo Bonzini block_job_enter(job); 341fc24908eSPaolo Bonzini } 342fc24908eSPaolo Bonzini 3435ccac6f1SJohn Snow void block_job_start(BlockJob *job) 3445ccac6f1SJohn Snow { 3455ccac6f1SJohn Snow assert(job && !block_job_started(job) && job->paused && 346e3796a24SJohn Snow job->driver && job->driver->start); 347e3796a24SJohn Snow job->co = qemu_coroutine_create(block_job_co_entry, job); 348e3796a24SJohn Snow job->pause_count--; 3495ccac6f1SJohn Snow job->busy = true; 350e3796a24SJohn Snow job->paused = false; 351a50c2ab8SKevin Wolf job_state_transition(&job->job, JOB_STATUS_RUNNING); 352aef4278cSFam Zheng bdrv_coroutine_enter(blk_bs(job->blk), job->co); 3535ccac6f1SJohn Snow } 3545ccac6f1SJohn Snow 3553925cd3bSJohn Snow static void block_job_decommission(BlockJob *job) 3563925cd3bSJohn Snow { 3573925cd3bSJohn Snow assert(job); 3583925cd3bSJohn Snow job->completed = true; 3593925cd3bSJohn Snow job->busy = false; 3603925cd3bSJohn Snow job->paused = false; 3613925cd3bSJohn Snow job->deferred_to_main_loop = true; 362a865cebbSMarc-André Lureau block_job_txn_del_job(job); 363a50c2ab8SKevin Wolf job_state_transition(&job->job, JOB_STATUS_NULL); 364*80fa2c75SKevin Wolf job_unref(&job->job); 3653925cd3bSJohn Snow } 3663925cd3bSJohn Snow 36775f71059SJohn Snow static void block_job_do_dismiss(BlockJob *job) 36875f71059SJohn Snow { 36975f71059SJohn Snow block_job_decommission(job); 37075f71059SJohn Snow } 37175f71059SJohn Snow 372e0cf0364SJohn Snow static void block_job_conclude(BlockJob *job) 373e0cf0364SJohn Snow { 374a50c2ab8SKevin Wolf job_state_transition(&job->job, JOB_STATUS_CONCLUDED); 37575f71059SJohn Snow if (job->auto_dismiss || !block_job_started(job)) { 37675f71059SJohn Snow block_job_do_dismiss(job); 37775f71059SJohn Snow } 378e0cf0364SJohn Snow } 379e0cf0364SJohn Snow 38035d6b368SJohn Snow static void block_job_update_rc(BlockJob *job) 38135d6b368SJohn Snow { 38235d6b368SJohn Snow if (!job->ret && block_job_is_cancelled(job)) { 38335d6b368SJohn Snow job->ret = -ECANCELED; 38435d6b368SJohn Snow } 38535d6b368SJohn Snow if (job->ret) { 386a50c2ab8SKevin Wolf job_state_transition(&job->job, JOB_STATUS_ABORTING); 38735d6b368SJohn Snow } 38835d6b368SJohn Snow } 38935d6b368SJohn Snow 3902da4617aSJohn Snow static int block_job_prepare(BlockJob *job) 3912da4617aSJohn Snow { 3922da4617aSJohn Snow if (job->ret == 0 && job->driver->prepare) { 3932da4617aSJohn Snow job->ret = job->driver->prepare(job); 3942da4617aSJohn Snow } 3952da4617aSJohn Snow return job->ret; 3962da4617aSJohn Snow } 3972da4617aSJohn Snow 39843628d93SJohn Snow static void block_job_commit(BlockJob *job) 39943628d93SJohn Snow { 40043628d93SJohn Snow assert(!job->ret); 40143628d93SJohn Snow if (job->driver->commit) { 40243628d93SJohn Snow job->driver->commit(job); 40343628d93SJohn Snow } 40443628d93SJohn Snow } 40543628d93SJohn Snow 40643628d93SJohn Snow static void block_job_abort(BlockJob *job) 40743628d93SJohn Snow { 40843628d93SJohn Snow assert(job->ret); 40943628d93SJohn Snow if (job->driver->abort) { 41043628d93SJohn Snow job->driver->abort(job); 41143628d93SJohn Snow } 41243628d93SJohn Snow } 41343628d93SJohn Snow 41443628d93SJohn Snow static void block_job_clean(BlockJob *job) 41543628d93SJohn Snow { 41643628d93SJohn Snow if (job->driver->clean) { 41743628d93SJohn Snow job->driver->clean(job); 41843628d93SJohn Snow } 41943628d93SJohn Snow } 42043628d93SJohn Snow 42111b61fbcSJohn Snow static int block_job_finalize_single(BlockJob *job) 422c55a832fSFam Zheng { 4234fb588e9SPaolo Bonzini assert(job->completed); 4244fb588e9SPaolo Bonzini 42535d6b368SJohn Snow /* Ensure abort is called for late-transactional failures */ 42635d6b368SJohn Snow block_job_update_rc(job); 42710a3fbb0SJohn Snow 428c55a832fSFam Zheng if (!job->ret) { 42943628d93SJohn Snow block_job_commit(job); 430c55a832fSFam Zheng } else { 43143628d93SJohn Snow block_job_abort(job); 432c55a832fSFam Zheng } 43343628d93SJohn Snow block_job_clean(job); 4348254b6d9SJohn Snow 4358254b6d9SJohn Snow if (job->cb) { 436c55a832fSFam Zheng job->cb(job->opaque, job->ret); 4378254b6d9SJohn Snow } 4385ccac6f1SJohn Snow 4395ccac6f1SJohn Snow /* Emit events only if we actually started */ 4405ccac6f1SJohn Snow if (block_job_started(job)) { 4418254b6d9SJohn Snow if (block_job_is_cancelled(job)) { 4428254b6d9SJohn Snow block_job_event_cancelled(job); 4438254b6d9SJohn Snow } else { 4448254b6d9SJohn Snow const char *msg = NULL; 4458254b6d9SJohn Snow if (job->ret < 0) { 4468254b6d9SJohn Snow msg = strerror(-job->ret); 4478254b6d9SJohn Snow } 4488254b6d9SJohn Snow block_job_event_completed(job, msg); 4498254b6d9SJohn Snow } 4505ccac6f1SJohn Snow } 4518254b6d9SJohn Snow 452a865cebbSMarc-André Lureau block_job_txn_del_job(job); 453e0cf0364SJohn Snow block_job_conclude(job); 4542da4617aSJohn Snow return 0; 455c55a832fSFam Zheng } 456c55a832fSFam Zheng 457b76e4458SLiang Li static void block_job_cancel_async(BlockJob *job, bool force) 4584c241cf5SPaolo Bonzini { 4594c241cf5SPaolo Bonzini if (job->iostatus != BLOCK_DEVICE_IO_STATUS_OK) { 4604c241cf5SPaolo Bonzini block_job_iostatus_reset(job); 4614c241cf5SPaolo Bonzini } 4624c241cf5SPaolo Bonzini if (job->user_paused) { 4634c241cf5SPaolo Bonzini /* Do not call block_job_enter here, the caller will handle it. */ 4644c241cf5SPaolo Bonzini job->user_paused = false; 4654c241cf5SPaolo Bonzini job->pause_count--; 4664c241cf5SPaolo Bonzini } 4674c241cf5SPaolo Bonzini job->cancelled = true; 468b76e4458SLiang Li /* To prevent 'force == false' overriding a previous 'force == true' */ 469b76e4458SLiang Li job->force |= force; 4704c241cf5SPaolo Bonzini } 4714c241cf5SPaolo Bonzini 4725f241594SJohn Snow static int block_job_txn_apply(BlockJobTxn *txn, int fn(BlockJob *), bool lock) 473efe4d4b7SJohn Snow { 474efe4d4b7SJohn Snow AioContext *ctx; 475efe4d4b7SJohn Snow BlockJob *job, *next; 4762da4617aSJohn Snow int rc = 0; 477efe4d4b7SJohn Snow 478efe4d4b7SJohn Snow QLIST_FOREACH_SAFE(job, &txn->jobs, txn_list, next) { 4795f241594SJohn Snow if (lock) { 480efe4d4b7SJohn Snow ctx = blk_get_aio_context(job->blk); 481efe4d4b7SJohn Snow aio_context_acquire(ctx); 4825f241594SJohn Snow } 4832da4617aSJohn Snow rc = fn(job); 4845f241594SJohn Snow if (lock) { 485efe4d4b7SJohn Snow aio_context_release(ctx); 4865f241594SJohn Snow } 4872da4617aSJohn Snow if (rc) { 4882da4617aSJohn Snow break; 489efe4d4b7SJohn Snow } 490efe4d4b7SJohn Snow } 4912da4617aSJohn Snow return rc; 4922da4617aSJohn Snow } 493efe4d4b7SJohn Snow 494c8ab5c2dSPaolo Bonzini static int block_job_finish_sync(BlockJob *job, 495c8ab5c2dSPaolo Bonzini void (*finish)(BlockJob *, Error **errp), 496c8ab5c2dSPaolo Bonzini Error **errp) 497c8ab5c2dSPaolo Bonzini { 498c8ab5c2dSPaolo Bonzini Error *local_err = NULL; 499c8ab5c2dSPaolo Bonzini int ret; 500c8ab5c2dSPaolo Bonzini 501c8ab5c2dSPaolo Bonzini assert(blk_bs(job->blk)->job == job); 502c8ab5c2dSPaolo Bonzini 503*80fa2c75SKevin Wolf job_ref(&job->job); 504c8ab5c2dSPaolo Bonzini 5054fb588e9SPaolo Bonzini if (finish) { 506c8ab5c2dSPaolo Bonzini finish(job, &local_err); 5074fb588e9SPaolo Bonzini } 508c8ab5c2dSPaolo Bonzini if (local_err) { 509c8ab5c2dSPaolo Bonzini error_propagate(errp, local_err); 510*80fa2c75SKevin Wolf job_unref(&job->job); 511c8ab5c2dSPaolo Bonzini return -EBUSY; 512c8ab5c2dSPaolo Bonzini } 513c8ab5c2dSPaolo Bonzini /* block_job_drain calls block_job_enter, and it should be enough to 514c8ab5c2dSPaolo Bonzini * induce progress until the job completes or moves to the main thread. 515c8ab5c2dSPaolo Bonzini */ 516c8ab5c2dSPaolo Bonzini while (!job->deferred_to_main_loop && !job->completed) { 517c8ab5c2dSPaolo Bonzini block_job_drain(job); 518c8ab5c2dSPaolo Bonzini } 519c8ab5c2dSPaolo Bonzini while (!job->completed) { 520c8ab5c2dSPaolo Bonzini aio_poll(qemu_get_aio_context(), true); 521c8ab5c2dSPaolo Bonzini } 522c8ab5c2dSPaolo Bonzini ret = (job->cancelled && job->ret == 0) ? -ECANCELED : job->ret; 523*80fa2c75SKevin Wolf job_unref(&job->job); 524c8ab5c2dSPaolo Bonzini return ret; 525c8ab5c2dSPaolo Bonzini } 526c8ab5c2dSPaolo Bonzini 527c55a832fSFam Zheng static void block_job_completed_txn_abort(BlockJob *job) 528c55a832fSFam Zheng { 529c55a832fSFam Zheng AioContext *ctx; 530c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 5314fb588e9SPaolo Bonzini BlockJob *other_job; 532c55a832fSFam Zheng 533c55a832fSFam Zheng if (txn->aborting) { 534c55a832fSFam Zheng /* 535c55a832fSFam Zheng * We are cancelled by another job, which will handle everything. 536c55a832fSFam Zheng */ 537c55a832fSFam Zheng return; 538c55a832fSFam Zheng } 539c55a832fSFam Zheng txn->aborting = true; 5404fb588e9SPaolo Bonzini block_job_txn_ref(txn); 5414fb588e9SPaolo Bonzini 542c55a832fSFam Zheng /* We are the first failed job. Cancel other jobs. */ 543c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 544b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 545c55a832fSFam Zheng aio_context_acquire(ctx); 546c55a832fSFam Zheng } 5474fb588e9SPaolo Bonzini 5484fb588e9SPaolo Bonzini /* Other jobs are effectively cancelled by us, set the status for 549c55a832fSFam Zheng * them; this job, however, may or may not be cancelled, depending 550c55a832fSFam Zheng * on the caller, so leave it. */ 5514fb588e9SPaolo Bonzini QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 552c55a832fSFam Zheng if (other_job != job) { 553b76e4458SLiang Li block_job_cancel_async(other_job, false); 554c55a832fSFam Zheng } 555c55a832fSFam Zheng } 5564fb588e9SPaolo Bonzini while (!QLIST_EMPTY(&txn->jobs)) { 5574fb588e9SPaolo Bonzini other_job = QLIST_FIRST(&txn->jobs); 558b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 5594fb588e9SPaolo Bonzini if (!other_job->completed) { 5604fb588e9SPaolo Bonzini assert(other_job->cancelled); 5614fb588e9SPaolo Bonzini block_job_finish_sync(other_job, NULL, NULL); 5624fb588e9SPaolo Bonzini } 56311b61fbcSJohn Snow block_job_finalize_single(other_job); 564c55a832fSFam Zheng aio_context_release(ctx); 565c55a832fSFam Zheng } 5664fb588e9SPaolo Bonzini 5674fb588e9SPaolo Bonzini block_job_txn_unref(txn); 568c55a832fSFam Zheng } 569c55a832fSFam Zheng 57011b61fbcSJohn Snow static int block_job_needs_finalize(BlockJob *job) 57111b61fbcSJohn Snow { 57211b61fbcSJohn Snow return !job->auto_finalize; 57311b61fbcSJohn Snow } 57411b61fbcSJohn Snow 57511b61fbcSJohn Snow static void block_job_do_finalize(BlockJob *job) 57611b61fbcSJohn Snow { 57711b61fbcSJohn Snow int rc; 57811b61fbcSJohn Snow assert(job && job->txn); 57911b61fbcSJohn Snow 58011b61fbcSJohn Snow /* prepare the transaction to complete */ 58111b61fbcSJohn Snow rc = block_job_txn_apply(job->txn, block_job_prepare, true); 58211b61fbcSJohn Snow if (rc) { 58311b61fbcSJohn Snow block_job_completed_txn_abort(job); 58411b61fbcSJohn Snow } else { 58511b61fbcSJohn Snow block_job_txn_apply(job->txn, block_job_finalize_single, true); 58611b61fbcSJohn Snow } 58711b61fbcSJohn Snow } 58811b61fbcSJohn Snow 589c55a832fSFam Zheng static void block_job_completed_txn_success(BlockJob *job) 590c55a832fSFam Zheng { 591c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 592efe4d4b7SJohn Snow BlockJob *other_job; 5932da4617aSJohn Snow 594a50c2ab8SKevin Wolf job_state_transition(&job->job, JOB_STATUS_WAITING); 595e8af5686SJohn Snow 596c55a832fSFam Zheng /* 597c55a832fSFam Zheng * Successful completion, see if there are other running jobs in this 598c55a832fSFam Zheng * txn. 599c55a832fSFam Zheng */ 600c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 601c55a832fSFam Zheng if (!other_job->completed) { 602c55a832fSFam Zheng return; 603c55a832fSFam Zheng } 604efe4d4b7SJohn Snow assert(other_job->ret == 0); 605c55a832fSFam Zheng } 6062da4617aSJohn Snow 6075f241594SJohn Snow block_job_txn_apply(txn, block_job_event_pending, false); 60811b61fbcSJohn Snow 60911b61fbcSJohn Snow /* If no jobs need manual finalization, automatically do so */ 61011b61fbcSJohn Snow if (block_job_txn_apply(txn, block_job_needs_finalize, false) == 0) { 61111b61fbcSJohn Snow block_job_do_finalize(job); 61211b61fbcSJohn Snow } 613c55a832fSFam Zheng } 614c55a832fSFam Zheng 6152f0c9fe6SPaolo Bonzini void block_job_set_speed(BlockJob *job, int64_t speed, Error **errp) 6162f0c9fe6SPaolo Bonzini { 617aa9ef2e6SJohn Snow int64_t old_speed = job->speed; 6182f0c9fe6SPaolo Bonzini 619a50c2ab8SKevin Wolf if (job_apply_verb(&job->job, JOB_VERB_SET_SPEED, errp)) { 6200ec4dfb8SJohn Snow return; 6210ec4dfb8SJohn Snow } 62218bb6928SKevin Wolf if (speed < 0) { 62318bb6928SKevin Wolf error_setg(errp, QERR_INVALID_PARAMETER, "speed"); 6242f0c9fe6SPaolo Bonzini return; 6252f0c9fe6SPaolo Bonzini } 6262f0c9fe6SPaolo Bonzini 62718bb6928SKevin Wolf ratelimit_set_speed(&job->limit, speed, BLOCK_JOB_SLICE_TIME); 62818bb6928SKevin Wolf 6292f0c9fe6SPaolo Bonzini job->speed = speed; 630d4fce188SJohn Snow if (speed && speed <= old_speed) { 631aa9ef2e6SJohn Snow return; 632aa9ef2e6SJohn Snow } 633aa9ef2e6SJohn Snow 634aa9ef2e6SJohn Snow /* kick only if a timer is pending */ 635aa9ef2e6SJohn Snow block_job_enter_cond(job, block_job_timer_pending); 6362f0c9fe6SPaolo Bonzini } 6372f0c9fe6SPaolo Bonzini 638dee81d51SKevin Wolf int64_t block_job_ratelimit_get_delay(BlockJob *job, uint64_t n) 639dee81d51SKevin Wolf { 640dee81d51SKevin Wolf if (!job->speed) { 641dee81d51SKevin Wolf return 0; 642dee81d51SKevin Wolf } 643dee81d51SKevin Wolf 644dee81d51SKevin Wolf return ratelimit_calculate_delay(&job->limit, n); 645dee81d51SKevin Wolf } 646dee81d51SKevin Wolf 647aeae883bSPaolo Bonzini void block_job_complete(BlockJob *job, Error **errp) 648aeae883bSPaolo Bonzini { 649559b935fSJohn Snow /* Should not be reachable via external interface for internal jobs */ 65033e9e9bdSKevin Wolf assert(job->job.id); 651a50c2ab8SKevin Wolf if (job_apply_verb(&job->job, JOB_VERB_COMPLETE, errp)) { 6520ec4dfb8SJohn Snow return; 6530ec4dfb8SJohn Snow } 6540ec4dfb8SJohn Snow if (job->pause_count || job->cancelled || !job->driver->complete) { 6559df229c3SAlberto Garcia error_setg(errp, "The active block job '%s' cannot be completed", 65633e9e9bdSKevin Wolf job->job.id); 657aeae883bSPaolo Bonzini return; 658aeae883bSPaolo Bonzini } 659aeae883bSPaolo Bonzini 6603fc4b10aSFam Zheng job->driver->complete(job, errp); 661aeae883bSPaolo Bonzini } 662aeae883bSPaolo Bonzini 66311b61fbcSJohn Snow void block_job_finalize(BlockJob *job, Error **errp) 66411b61fbcSJohn Snow { 66533e9e9bdSKevin Wolf assert(job && job->job.id); 666a50c2ab8SKevin Wolf if (job_apply_verb(&job->job, JOB_VERB_FINALIZE, errp)) { 66711b61fbcSJohn Snow return; 66811b61fbcSJohn Snow } 66911b61fbcSJohn Snow block_job_do_finalize(job); 67011b61fbcSJohn Snow } 67111b61fbcSJohn Snow 67275f71059SJohn Snow void block_job_dismiss(BlockJob **jobptr, Error **errp) 67375f71059SJohn Snow { 67475f71059SJohn Snow BlockJob *job = *jobptr; 67575f71059SJohn Snow /* similarly to _complete, this is QMP-interface only. */ 67633e9e9bdSKevin Wolf assert(job->job.id); 677a50c2ab8SKevin Wolf if (job_apply_verb(&job->job, JOB_VERB_DISMISS, errp)) { 67875f71059SJohn Snow return; 67975f71059SJohn Snow } 68075f71059SJohn Snow 68175f71059SJohn Snow block_job_do_dismiss(job); 68275f71059SJohn Snow *jobptr = NULL; 68375f71059SJohn Snow } 68475f71059SJohn Snow 6850ec4dfb8SJohn Snow void block_job_user_pause(BlockJob *job, Error **errp) 6860df4ba58SJohn Snow { 687a50c2ab8SKevin Wolf if (job_apply_verb(&job->job, JOB_VERB_PAUSE, errp)) { 6880ec4dfb8SJohn Snow return; 6890ec4dfb8SJohn Snow } 6900ec4dfb8SJohn Snow if (job->user_paused) { 6910ec4dfb8SJohn Snow error_setg(errp, "Job is already paused"); 6920ec4dfb8SJohn Snow return; 6930ec4dfb8SJohn Snow } 6940df4ba58SJohn Snow job->user_paused = true; 6950df4ba58SJohn Snow block_job_pause(job); 6960df4ba58SJohn Snow } 6970df4ba58SJohn Snow 6980df4ba58SJohn Snow bool block_job_user_paused(BlockJob *job) 6990df4ba58SJohn Snow { 7006573d9c6SPaolo Bonzini return job->user_paused; 7010df4ba58SJohn Snow } 7020df4ba58SJohn Snow 7030ec4dfb8SJohn Snow void block_job_user_resume(BlockJob *job, Error **errp) 7040df4ba58SJohn Snow { 7050ec4dfb8SJohn Snow assert(job); 7060ec4dfb8SJohn Snow if (!job->user_paused || job->pause_count <= 0) { 7070ec4dfb8SJohn Snow error_setg(errp, "Can't resume a job that was not paused"); 7080ec4dfb8SJohn Snow return; 7090ec4dfb8SJohn Snow } 710a50c2ab8SKevin Wolf if (job_apply_verb(&job->job, JOB_VERB_RESUME, errp)) { 7110ec4dfb8SJohn Snow return; 7120ec4dfb8SJohn Snow } 7132caf63a9SPaolo Bonzini block_job_iostatus_reset(job); 7144c241cf5SPaolo Bonzini job->user_paused = false; 7150df4ba58SJohn Snow block_job_resume(job); 7160df4ba58SJohn Snow } 7170df4ba58SJohn Snow 718b76e4458SLiang Li void block_job_cancel(BlockJob *job, bool force) 7198acc72a4SPaolo Bonzini { 720a50c2ab8SKevin Wolf if (job->job.status == JOB_STATUS_CONCLUDED) { 72175f71059SJohn Snow block_job_do_dismiss(job); 72211b61fbcSJohn Snow return; 72311b61fbcSJohn Snow } 724b76e4458SLiang Li block_job_cancel_async(job, force); 72511b61fbcSJohn Snow if (!block_job_started(job)) { 7265ccac6f1SJohn Snow block_job_completed(job, -ECANCELED); 72711b61fbcSJohn Snow } else if (job->deferred_to_main_loop) { 72811b61fbcSJohn Snow block_job_completed_txn_abort(job); 72911b61fbcSJohn Snow } else { 73011b61fbcSJohn Snow block_job_enter(job); 7315ccac6f1SJohn Snow } 7328acc72a4SPaolo Bonzini } 7338acc72a4SPaolo Bonzini 734b76e4458SLiang Li void block_job_user_cancel(BlockJob *job, bool force, Error **errp) 7350ec4dfb8SJohn Snow { 736a50c2ab8SKevin Wolf if (job_apply_verb(&job->job, JOB_VERB_CANCEL, errp)) { 7370ec4dfb8SJohn Snow return; 7380ec4dfb8SJohn Snow } 739b76e4458SLiang Li block_job_cancel(job, force); 7400ec4dfb8SJohn Snow } 7410ec4dfb8SJohn Snow 742345f9e1bSMax Reitz /* A wrapper around block_job_cancel() taking an Error ** parameter so it may be 743345f9e1bSMax Reitz * used with block_job_finish_sync() without the need for (rather nasty) 744345f9e1bSMax Reitz * function pointer casts there. */ 745345f9e1bSMax Reitz static void block_job_cancel_err(BlockJob *job, Error **errp) 746345f9e1bSMax Reitz { 747b76e4458SLiang Li block_job_cancel(job, false); 748345f9e1bSMax Reitz } 749345f9e1bSMax Reitz 750345f9e1bSMax Reitz int block_job_cancel_sync(BlockJob *job) 751345f9e1bSMax Reitz { 752345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_cancel_err, NULL); 753345f9e1bSMax Reitz } 754345f9e1bSMax Reitz 755a1a2af07SKevin Wolf void block_job_cancel_sync_all(void) 756a1a2af07SKevin Wolf { 757a1a2af07SKevin Wolf BlockJob *job; 758a1a2af07SKevin Wolf AioContext *aio_context; 759a1a2af07SKevin Wolf 760e7c1d78bSKevin Wolf while ((job = block_job_next(NULL))) { 761b6d2e599SKevin Wolf aio_context = blk_get_aio_context(job->blk); 762a1a2af07SKevin Wolf aio_context_acquire(aio_context); 763a1a2af07SKevin Wolf block_job_cancel_sync(job); 764a1a2af07SKevin Wolf aio_context_release(aio_context); 765a1a2af07SKevin Wolf } 766a1a2af07SKevin Wolf } 767a1a2af07SKevin Wolf 768345f9e1bSMax Reitz int block_job_complete_sync(BlockJob *job, Error **errp) 769345f9e1bSMax Reitz { 770345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_complete, errp); 771345f9e1bSMax Reitz } 772345f9e1bSMax Reitz 77305df8a6aSKevin Wolf void block_job_progress_update(BlockJob *job, uint64_t done) 77405df8a6aSKevin Wolf { 77505df8a6aSKevin Wolf job->offset += done; 77605df8a6aSKevin Wolf } 77705df8a6aSKevin Wolf 77805df8a6aSKevin Wolf void block_job_progress_set_remaining(BlockJob *job, uint64_t remaining) 77905df8a6aSKevin Wolf { 78005df8a6aSKevin Wolf job->len = job->offset + remaining; 78105df8a6aSKevin Wolf } 78205df8a6aSKevin Wolf 783559b935fSJohn Snow BlockJobInfo *block_job_query(BlockJob *job, Error **errp) 78430e628b7SPaolo Bonzini { 785559b935fSJohn Snow BlockJobInfo *info; 786559b935fSJohn Snow 787559b935fSJohn Snow if (block_job_is_internal(job)) { 788559b935fSJohn Snow error_setg(errp, "Cannot query QEMU internal jobs"); 789559b935fSJohn Snow return NULL; 790559b935fSJohn Snow } 791559b935fSJohn Snow info = g_new0(BlockJobInfo, 1); 792252291eaSKevin Wolf info->type = g_strdup(job_type_str(&job->job)); 79333e9e9bdSKevin Wolf info->device = g_strdup(job->job.id); 79430e628b7SPaolo Bonzini info->len = job->len; 795fc24908eSPaolo Bonzini info->busy = atomic_read(&job->busy); 796751ebd76SFam Zheng info->paused = job->pause_count > 0; 79730e628b7SPaolo Bonzini info->offset = job->offset; 79830e628b7SPaolo Bonzini info->speed = job->speed; 79932c81a4aSPaolo Bonzini info->io_status = job->iostatus; 800ef6dbf1eSMax Reitz info->ready = job->ready; 801a50c2ab8SKevin Wolf info->status = job->job.status; 802b40dacdcSJohn Snow info->auto_finalize = job->auto_finalize; 803b40dacdcSJohn Snow info->auto_dismiss = job->auto_dismiss; 804ab9ba614SJohn Snow info->has_error = job->ret != 0; 805ab9ba614SJohn Snow info->error = job->ret ? g_strdup(strerror(-job->ret)) : NULL; 80630e628b7SPaolo Bonzini return info; 80730e628b7SPaolo Bonzini } 80832c81a4aSPaolo Bonzini 80932c81a4aSPaolo Bonzini static void block_job_iostatus_set_err(BlockJob *job, int error) 81032c81a4aSPaolo Bonzini { 81132c81a4aSPaolo Bonzini if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 81232c81a4aSPaolo Bonzini job->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE : 81332c81a4aSPaolo Bonzini BLOCK_DEVICE_IO_STATUS_FAILED; 81432c81a4aSPaolo Bonzini } 81532c81a4aSPaolo Bonzini } 81632c81a4aSPaolo Bonzini 8178254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job) 818a66a2a36SPaolo Bonzini { 819559b935fSJohn Snow if (block_job_is_internal(job)) { 820559b935fSJohn Snow return; 821559b935fSJohn Snow } 822559b935fSJohn Snow 823252291eaSKevin Wolf qapi_event_send_block_job_cancelled(job_type(&job->job), 82433e9e9bdSKevin Wolf job->job.id, 825a66a2a36SPaolo Bonzini job->len, 826a66a2a36SPaolo Bonzini job->offset, 827bcada37bSWenchao Xia job->speed, 828bcada37bSWenchao Xia &error_abort); 829a66a2a36SPaolo Bonzini } 830a66a2a36SPaolo Bonzini 8318254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg) 832a66a2a36SPaolo Bonzini { 833559b935fSJohn Snow if (block_job_is_internal(job)) { 834559b935fSJohn Snow return; 835559b935fSJohn Snow } 836559b935fSJohn Snow 837252291eaSKevin Wolf qapi_event_send_block_job_completed(job_type(&job->job), 83833e9e9bdSKevin Wolf job->job.id, 839bcada37bSWenchao Xia job->len, 840bcada37bSWenchao Xia job->offset, 841bcada37bSWenchao Xia job->speed, 842bcada37bSWenchao Xia !!msg, 843bcada37bSWenchao Xia msg, 844bcada37bSWenchao Xia &error_abort); 845bcada37bSWenchao Xia } 846bcada37bSWenchao Xia 8475f241594SJohn Snow static int block_job_event_pending(BlockJob *job) 8485f241594SJohn Snow { 849a50c2ab8SKevin Wolf job_state_transition(&job->job, JOB_STATUS_PENDING); 8505f241594SJohn Snow if (!job->auto_finalize && !block_job_is_internal(job)) { 851252291eaSKevin Wolf qapi_event_send_block_job_pending(job_type(&job->job), 85233e9e9bdSKevin Wolf job->job.id, 8535f241594SJohn Snow &error_abort); 8545f241594SJohn Snow } 8555f241594SJohn Snow return 0; 8565f241594SJohn Snow } 8575f241594SJohn Snow 85888691b37SPaolo Bonzini /* 85988691b37SPaolo Bonzini * API for block job drivers and the block layer. These functions are 86088691b37SPaolo Bonzini * declared in blockjob_int.h. 86188691b37SPaolo Bonzini */ 86288691b37SPaolo Bonzini 86388691b37SPaolo Bonzini void *block_job_create(const char *job_id, const BlockJobDriver *driver, 86475859b94SJohn Snow BlockJobTxn *txn, BlockDriverState *bs, uint64_t perm, 86588691b37SPaolo Bonzini uint64_t shared_perm, int64_t speed, int flags, 86688691b37SPaolo Bonzini BlockCompletionFunc *cb, void *opaque, Error **errp) 86788691b37SPaolo Bonzini { 86888691b37SPaolo Bonzini BlockBackend *blk; 86988691b37SPaolo Bonzini BlockJob *job; 87088691b37SPaolo Bonzini int ret; 87188691b37SPaolo Bonzini 87288691b37SPaolo Bonzini if (bs->job) { 87388691b37SPaolo Bonzini error_setg(errp, QERR_DEVICE_IN_USE, bdrv_get_device_name(bs)); 87488691b37SPaolo Bonzini return NULL; 87588691b37SPaolo Bonzini } 87688691b37SPaolo Bonzini 87788691b37SPaolo Bonzini if (job_id == NULL && !(flags & BLOCK_JOB_INTERNAL)) { 87888691b37SPaolo Bonzini job_id = bdrv_get_device_name(bs); 87988691b37SPaolo Bonzini if (!*job_id) { 88088691b37SPaolo Bonzini error_setg(errp, "An explicit job ID is required for this node"); 88188691b37SPaolo Bonzini return NULL; 88288691b37SPaolo Bonzini } 88388691b37SPaolo Bonzini } 88488691b37SPaolo Bonzini 88588691b37SPaolo Bonzini if (job_id) { 88688691b37SPaolo Bonzini if (flags & BLOCK_JOB_INTERNAL) { 88788691b37SPaolo Bonzini error_setg(errp, "Cannot specify job ID for internal block job"); 88888691b37SPaolo Bonzini return NULL; 88988691b37SPaolo Bonzini } 89088691b37SPaolo Bonzini } 89188691b37SPaolo Bonzini 89288691b37SPaolo Bonzini blk = blk_new(perm, shared_perm); 89388691b37SPaolo Bonzini ret = blk_insert_bs(blk, bs, errp); 89488691b37SPaolo Bonzini if (ret < 0) { 89588691b37SPaolo Bonzini blk_unref(blk); 89688691b37SPaolo Bonzini return NULL; 89788691b37SPaolo Bonzini } 89888691b37SPaolo Bonzini 89933e9e9bdSKevin Wolf job = job_create(job_id, &driver->job_driver, errp); 90033e9e9bdSKevin Wolf if (job == NULL) { 90133e9e9bdSKevin Wolf blk_unref(blk); 90233e9e9bdSKevin Wolf return NULL; 90333e9e9bdSKevin Wolf } 90433e9e9bdSKevin Wolf 905e7c1d78bSKevin Wolf assert(is_block_job(&job->job)); 906*80fa2c75SKevin Wolf assert(job->job.driver->free == &block_job_free); 907e7c1d78bSKevin Wolf 90888691b37SPaolo Bonzini job->driver = driver; 90988691b37SPaolo Bonzini job->blk = blk; 91088691b37SPaolo Bonzini job->cb = cb; 91188691b37SPaolo Bonzini job->opaque = opaque; 91288691b37SPaolo Bonzini job->busy = false; 91388691b37SPaolo Bonzini job->paused = true; 91488691b37SPaolo Bonzini job->pause_count = 1; 9155f241594SJohn Snow job->auto_finalize = !(flags & BLOCK_JOB_MANUAL_FINALIZE); 91675f71059SJohn Snow job->auto_dismiss = !(flags & BLOCK_JOB_MANUAL_DISMISS); 917fc24908eSPaolo Bonzini aio_timer_init(qemu_get_aio_context(), &job->sleep_timer, 918fc24908eSPaolo Bonzini QEMU_CLOCK_REALTIME, SCALE_NS, 919fc24908eSPaolo Bonzini block_job_sleep_timer_cb, job); 92088691b37SPaolo Bonzini 92188691b37SPaolo Bonzini error_setg(&job->blocker, "block device is in use by block job: %s", 922252291eaSKevin Wolf job_type_str(&job->job)); 92388691b37SPaolo Bonzini block_job_add_bdrv(job, "main node", bs, 0, BLK_PERM_ALL, &error_abort); 92488691b37SPaolo Bonzini bs->job = job; 92588691b37SPaolo Bonzini 92688691b37SPaolo Bonzini bdrv_op_unblock(bs, BLOCK_OP_TYPE_DATAPLANE, job->blocker); 92788691b37SPaolo Bonzini 92888691b37SPaolo Bonzini blk_add_aio_context_notifier(blk, block_job_attached_aio_context, 92988691b37SPaolo Bonzini block_job_detach_aio_context, job); 93088691b37SPaolo Bonzini 93188691b37SPaolo Bonzini /* Only set speed when necessary to avoid NotSupported error */ 93288691b37SPaolo Bonzini if (speed != 0) { 93388691b37SPaolo Bonzini Error *local_err = NULL; 93488691b37SPaolo Bonzini 93588691b37SPaolo Bonzini block_job_set_speed(job, speed, &local_err); 93688691b37SPaolo Bonzini if (local_err) { 9373925cd3bSJohn Snow block_job_early_fail(job); 93888691b37SPaolo Bonzini error_propagate(errp, local_err); 93988691b37SPaolo Bonzini return NULL; 94088691b37SPaolo Bonzini } 94188691b37SPaolo Bonzini } 94275859b94SJohn Snow 94375859b94SJohn Snow /* Single jobs are modeled as single-job transactions for sake of 94475859b94SJohn Snow * consolidating the job management logic */ 94575859b94SJohn Snow if (!txn) { 94675859b94SJohn Snow txn = block_job_txn_new(); 94775859b94SJohn Snow block_job_txn_add_job(txn, job); 94875859b94SJohn Snow block_job_txn_unref(txn); 94975859b94SJohn Snow } else { 95075859b94SJohn Snow block_job_txn_add_job(txn, job); 95175859b94SJohn Snow } 95275859b94SJohn Snow 95388691b37SPaolo Bonzini return job; 95488691b37SPaolo Bonzini } 95588691b37SPaolo Bonzini 95688691b37SPaolo Bonzini void block_job_early_fail(BlockJob *job) 95788691b37SPaolo Bonzini { 958a50c2ab8SKevin Wolf assert(job->job.status == JOB_STATUS_CREATED); 9593925cd3bSJohn Snow block_job_decommission(job); 96088691b37SPaolo Bonzini } 96188691b37SPaolo Bonzini 96288691b37SPaolo Bonzini void block_job_completed(BlockJob *job, int ret) 96388691b37SPaolo Bonzini { 96475859b94SJohn Snow assert(job && job->txn && !job->completed); 96588691b37SPaolo Bonzini assert(blk_bs(job->blk)->job == job); 96688691b37SPaolo Bonzini job->completed = true; 96788691b37SPaolo Bonzini job->ret = ret; 96835d6b368SJohn Snow block_job_update_rc(job); 96935d6b368SJohn Snow trace_block_job_completed(job, ret, job->ret); 97035d6b368SJohn Snow if (job->ret) { 97188691b37SPaolo Bonzini block_job_completed_txn_abort(job); 97288691b37SPaolo Bonzini } else { 97388691b37SPaolo Bonzini block_job_completed_txn_success(job); 97488691b37SPaolo Bonzini } 97588691b37SPaolo Bonzini } 97688691b37SPaolo Bonzini 97788691b37SPaolo Bonzini static bool block_job_should_pause(BlockJob *job) 97888691b37SPaolo Bonzini { 97988691b37SPaolo Bonzini return job->pause_count > 0; 98088691b37SPaolo Bonzini } 98188691b37SPaolo Bonzini 982fc24908eSPaolo Bonzini /* Yield, and schedule a timer to reenter the coroutine after @ns nanoseconds. 983fc24908eSPaolo Bonzini * Reentering the job coroutine with block_job_enter() before the timer has 984fc24908eSPaolo Bonzini * expired is allowed and cancels the timer. 985fc24908eSPaolo Bonzini * 986fc24908eSPaolo Bonzini * If @ns is (uint64_t) -1, no timer is scheduled and block_job_enter() must be 987fc24908eSPaolo Bonzini * called explicitly. */ 988fc24908eSPaolo Bonzini static void block_job_do_yield(BlockJob *job, uint64_t ns) 989356f59b8SPaolo Bonzini { 990fc24908eSPaolo Bonzini block_job_lock(); 991fc24908eSPaolo Bonzini if (ns != -1) { 992fc24908eSPaolo Bonzini timer_mod(&job->sleep_timer, ns); 993fc24908eSPaolo Bonzini } 994356f59b8SPaolo Bonzini job->busy = false; 995fc24908eSPaolo Bonzini block_job_unlock(); 996356f59b8SPaolo Bonzini qemu_coroutine_yield(); 997356f59b8SPaolo Bonzini 998356f59b8SPaolo Bonzini /* Set by block_job_enter before re-entering the coroutine. */ 999356f59b8SPaolo Bonzini assert(job->busy); 1000356f59b8SPaolo Bonzini } 1001356f59b8SPaolo Bonzini 100288691b37SPaolo Bonzini void coroutine_fn block_job_pause_point(BlockJob *job) 100388691b37SPaolo Bonzini { 100488691b37SPaolo Bonzini assert(job && block_job_started(job)); 100588691b37SPaolo Bonzini 100688691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 100788691b37SPaolo Bonzini return; 100888691b37SPaolo Bonzini } 100988691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 101088691b37SPaolo Bonzini return; 101188691b37SPaolo Bonzini } 101288691b37SPaolo Bonzini 101388691b37SPaolo Bonzini if (job->driver->pause) { 101488691b37SPaolo Bonzini job->driver->pause(job); 101588691b37SPaolo Bonzini } 101688691b37SPaolo Bonzini 101788691b37SPaolo Bonzini if (block_job_should_pause(job) && !block_job_is_cancelled(job)) { 1018a50c2ab8SKevin Wolf JobStatus status = job->job.status; 1019a50c2ab8SKevin Wolf job_state_transition(&job->job, status == JOB_STATUS_READY 1020a50c2ab8SKevin Wolf ? JOB_STATUS_STANDBY 1021a50c2ab8SKevin Wolf : JOB_STATUS_PAUSED); 102288691b37SPaolo Bonzini job->paused = true; 1023fc24908eSPaolo Bonzini block_job_do_yield(job, -1); 102488691b37SPaolo Bonzini job->paused = false; 1025a50c2ab8SKevin Wolf job_state_transition(&job->job, status); 102688691b37SPaolo Bonzini } 102788691b37SPaolo Bonzini 102888691b37SPaolo Bonzini if (job->driver->resume) { 102988691b37SPaolo Bonzini job->driver->resume(job); 103088691b37SPaolo Bonzini } 103188691b37SPaolo Bonzini } 103288691b37SPaolo Bonzini 1033aa9ef2e6SJohn Snow /* 1034aa9ef2e6SJohn Snow * Conditionally enter a block_job pending a call to fn() while 1035aa9ef2e6SJohn Snow * under the block_job_lock critical section. 1036aa9ef2e6SJohn Snow */ 1037aa9ef2e6SJohn Snow static void block_job_enter_cond(BlockJob *job, bool(*fn)(BlockJob *job)) 103888691b37SPaolo Bonzini { 1039eb05e011SPaolo Bonzini if (!block_job_started(job)) { 1040eb05e011SPaolo Bonzini return; 1041eb05e011SPaolo Bonzini } 1042eb05e011SPaolo Bonzini if (job->deferred_to_main_loop) { 1043eb05e011SPaolo Bonzini return; 1044eb05e011SPaolo Bonzini } 1045eb05e011SPaolo Bonzini 1046fc24908eSPaolo Bonzini block_job_lock(); 1047356f59b8SPaolo Bonzini if (job->busy) { 1048fc24908eSPaolo Bonzini block_job_unlock(); 1049356f59b8SPaolo Bonzini return; 105088691b37SPaolo Bonzini } 1051356f59b8SPaolo Bonzini 1052aa9ef2e6SJohn Snow if (fn && !fn(job)) { 1053aa9ef2e6SJohn Snow block_job_unlock(); 1054aa9ef2e6SJohn Snow return; 1055aa9ef2e6SJohn Snow } 1056aa9ef2e6SJohn Snow 1057fc24908eSPaolo Bonzini assert(!job->deferred_to_main_loop); 1058fc24908eSPaolo Bonzini timer_del(&job->sleep_timer); 1059356f59b8SPaolo Bonzini job->busy = true; 1060fc24908eSPaolo Bonzini block_job_unlock(); 1061356f59b8SPaolo Bonzini aio_co_wake(job->co); 106288691b37SPaolo Bonzini } 106388691b37SPaolo Bonzini 1064aa9ef2e6SJohn Snow void block_job_enter(BlockJob *job) 1065aa9ef2e6SJohn Snow { 1066aa9ef2e6SJohn Snow block_job_enter_cond(job, NULL); 1067aa9ef2e6SJohn Snow } 1068aa9ef2e6SJohn Snow 106988691b37SPaolo Bonzini bool block_job_is_cancelled(BlockJob *job) 107088691b37SPaolo Bonzini { 107188691b37SPaolo Bonzini return job->cancelled; 107288691b37SPaolo Bonzini } 107388691b37SPaolo Bonzini 10745bf1d5a7SPaolo Bonzini void block_job_sleep_ns(BlockJob *job, int64_t ns) 107588691b37SPaolo Bonzini { 107688691b37SPaolo Bonzini assert(job->busy); 107788691b37SPaolo Bonzini 107888691b37SPaolo Bonzini /* Check cancellation *before* setting busy = false, too! */ 107988691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 108088691b37SPaolo Bonzini return; 108188691b37SPaolo Bonzini } 108288691b37SPaolo Bonzini 108388691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 1084fc24908eSPaolo Bonzini block_job_do_yield(job, qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + ns); 108588691b37SPaolo Bonzini } 108688691b37SPaolo Bonzini 108788691b37SPaolo Bonzini block_job_pause_point(job); 108888691b37SPaolo Bonzini } 108988691b37SPaolo Bonzini 109088691b37SPaolo Bonzini void block_job_yield(BlockJob *job) 109188691b37SPaolo Bonzini { 109288691b37SPaolo Bonzini assert(job->busy); 109388691b37SPaolo Bonzini 109488691b37SPaolo Bonzini /* Check cancellation *before* setting busy = false, too! */ 109588691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 109688691b37SPaolo Bonzini return; 109788691b37SPaolo Bonzini } 109888691b37SPaolo Bonzini 109988691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 1100fc24908eSPaolo Bonzini block_job_do_yield(job, -1); 110188691b37SPaolo Bonzini } 110288691b37SPaolo Bonzini 110388691b37SPaolo Bonzini block_job_pause_point(job); 110488691b37SPaolo Bonzini } 110588691b37SPaolo Bonzini 11062caf63a9SPaolo Bonzini void block_job_iostatus_reset(BlockJob *job) 11072caf63a9SPaolo Bonzini { 11084c241cf5SPaolo Bonzini if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 11094c241cf5SPaolo Bonzini return; 11104c241cf5SPaolo Bonzini } 11114c241cf5SPaolo Bonzini assert(job->user_paused && job->pause_count > 0); 11122caf63a9SPaolo Bonzini job->iostatus = BLOCK_DEVICE_IO_STATUS_OK; 11132caf63a9SPaolo Bonzini } 11142caf63a9SPaolo Bonzini 1115bcada37bSWenchao Xia void block_job_event_ready(BlockJob *job) 1116bcada37bSWenchao Xia { 1117a50c2ab8SKevin Wolf job_state_transition(&job->job, JOB_STATUS_READY); 1118ef6dbf1eSMax Reitz job->ready = true; 1119ef6dbf1eSMax Reitz 1120559b935fSJohn Snow if (block_job_is_internal(job)) { 1121559b935fSJohn Snow return; 1122559b935fSJohn Snow } 1123559b935fSJohn Snow 1124252291eaSKevin Wolf qapi_event_send_block_job_ready(job_type(&job->job), 112533e9e9bdSKevin Wolf job->job.id, 1126518848a2SMarkus Armbruster job->len, 1127518848a2SMarkus Armbruster job->offset, 1128518848a2SMarkus Armbruster job->speed, &error_abort); 1129a66a2a36SPaolo Bonzini } 1130a66a2a36SPaolo Bonzini 113181e254dcSKevin Wolf BlockErrorAction block_job_error_action(BlockJob *job, BlockdevOnError on_err, 113232c81a4aSPaolo Bonzini int is_read, int error) 113332c81a4aSPaolo Bonzini { 113432c81a4aSPaolo Bonzini BlockErrorAction action; 113532c81a4aSPaolo Bonzini 113632c81a4aSPaolo Bonzini switch (on_err) { 113732c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_ENOSPC: 11388c398252SKevin Wolf case BLOCKDEV_ON_ERROR_AUTO: 1139a589569fSWenchao Xia action = (error == ENOSPC) ? 1140a589569fSWenchao Xia BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT; 114132c81a4aSPaolo Bonzini break; 114232c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_STOP: 1143a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_STOP; 114432c81a4aSPaolo Bonzini break; 114532c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_REPORT: 1146a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_REPORT; 114732c81a4aSPaolo Bonzini break; 114832c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_IGNORE: 1149a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_IGNORE; 115032c81a4aSPaolo Bonzini break; 115132c81a4aSPaolo Bonzini default: 115232c81a4aSPaolo Bonzini abort(); 115332c81a4aSPaolo Bonzini } 1154559b935fSJohn Snow if (!block_job_is_internal(job)) { 115533e9e9bdSKevin Wolf qapi_event_send_block_job_error(job->job.id, 11565a2d2cbdSWenchao Xia is_read ? IO_OPERATION_TYPE_READ : 11575a2d2cbdSWenchao Xia IO_OPERATION_TYPE_WRITE, 11585a2d2cbdSWenchao Xia action, &error_abort); 1159559b935fSJohn Snow } 1160a589569fSWenchao Xia if (action == BLOCK_ERROR_ACTION_STOP) { 11610ec4dfb8SJohn Snow block_job_pause(job); 1162751ebd76SFam Zheng /* make the pause user visible, which will be resumed from QMP. */ 11630ec4dfb8SJohn Snow job->user_paused = true; 116432c81a4aSPaolo Bonzini block_job_iostatus_set_err(job, error); 116532c81a4aSPaolo Bonzini } 116632c81a4aSPaolo Bonzini return action; 116732c81a4aSPaolo Bonzini } 1168dec7d421SStefan Hajnoczi 1169dec7d421SStefan Hajnoczi typedef struct { 1170dec7d421SStefan Hajnoczi BlockJob *job; 1171dec7d421SStefan Hajnoczi AioContext *aio_context; 1172dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn; 1173dec7d421SStefan Hajnoczi void *opaque; 1174dec7d421SStefan Hajnoczi } BlockJobDeferToMainLoopData; 1175dec7d421SStefan Hajnoczi 1176dec7d421SStefan Hajnoczi static void block_job_defer_to_main_loop_bh(void *opaque) 1177dec7d421SStefan Hajnoczi { 1178dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = opaque; 1179dec7d421SStefan Hajnoczi AioContext *aio_context; 1180dec7d421SStefan Hajnoczi 1181dec7d421SStefan Hajnoczi /* Prevent race with block_job_defer_to_main_loop() */ 1182dec7d421SStefan Hajnoczi aio_context_acquire(data->aio_context); 1183dec7d421SStefan Hajnoczi 1184dec7d421SStefan Hajnoczi /* Fetch BDS AioContext again, in case it has changed */ 1185b6d2e599SKevin Wolf aio_context = blk_get_aio_context(data->job->blk); 1186d79df2a2SPaolo Bonzini if (aio_context != data->aio_context) { 1187dec7d421SStefan Hajnoczi aio_context_acquire(aio_context); 1188d79df2a2SPaolo Bonzini } 1189dec7d421SStefan Hajnoczi 1190dec7d421SStefan Hajnoczi data->fn(data->job, data->opaque); 1191dec7d421SStefan Hajnoczi 1192d79df2a2SPaolo Bonzini if (aio_context != data->aio_context) { 1193dec7d421SStefan Hajnoczi aio_context_release(aio_context); 1194d79df2a2SPaolo Bonzini } 1195dec7d421SStefan Hajnoczi 1196dec7d421SStefan Hajnoczi aio_context_release(data->aio_context); 1197dec7d421SStefan Hajnoczi 1198dec7d421SStefan Hajnoczi g_free(data); 1199dec7d421SStefan Hajnoczi } 1200dec7d421SStefan Hajnoczi 1201dec7d421SStefan Hajnoczi void block_job_defer_to_main_loop(BlockJob *job, 1202dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn, 1203dec7d421SStefan Hajnoczi void *opaque) 1204dec7d421SStefan Hajnoczi { 1205dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = g_malloc(sizeof(*data)); 1206dec7d421SStefan Hajnoczi data->job = job; 1207b6d2e599SKevin Wolf data->aio_context = blk_get_aio_context(job->blk); 1208dec7d421SStefan Hajnoczi data->fn = fn; 1209dec7d421SStefan Hajnoczi data->opaque = opaque; 1210794f0141SFam Zheng job->deferred_to_main_loop = true; 1211dec7d421SStefan Hajnoczi 1212fffb6e12SPaolo Bonzini aio_bh_schedule_oneshot(qemu_get_aio_context(), 1213fffb6e12SPaolo Bonzini block_job_defer_to_main_loop_bh, data); 1214dec7d421SStefan Hajnoczi } 1215