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" 31373340b2SMax Reitz #include "sysemu/block-backend.h" 32cc7a8ea7SMarkus Armbruster #include "qapi/qmp/qerror.h" 337b1b5d19SPaolo Bonzini #include "qapi/qmp/qjson.h" 3410817bf0SDaniel P. Berrange #include "qemu/coroutine.h" 357f0317cfSAlberto Garcia #include "qemu/id.h" 362f0c9fe6SPaolo Bonzini #include "qmp-commands.h" 371de7afc9SPaolo Bonzini #include "qemu/timer.h" 385a2d2cbdSWenchao Xia #include "qapi-event.h" 392f0c9fe6SPaolo Bonzini 408254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job); 418254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg); 428254b6d9SJohn Snow 43c55a832fSFam Zheng /* Transactional group of block jobs */ 44c55a832fSFam Zheng struct BlockJobTxn { 45c55a832fSFam Zheng 46c55a832fSFam Zheng /* Is this txn being cancelled? */ 47c55a832fSFam Zheng bool aborting; 48c55a832fSFam Zheng 49c55a832fSFam Zheng /* List of jobs */ 50c55a832fSFam Zheng QLIST_HEAD(, BlockJob) jobs; 51c55a832fSFam Zheng 52c55a832fSFam Zheng /* Reference count */ 53c55a832fSFam Zheng int refcnt; 54c55a832fSFam Zheng }; 55c55a832fSFam Zheng 56a7112795SAlberto Garcia static QLIST_HEAD(, BlockJob) block_jobs = QLIST_HEAD_INITIALIZER(block_jobs); 57a7112795SAlberto Garcia 5876d554e2SKevin Wolf static char *child_job_get_parent_desc(BdrvChild *c) 5976d554e2SKevin Wolf { 6076d554e2SKevin Wolf BlockJob *job = c->opaque; 6176d554e2SKevin Wolf return g_strdup_printf("%s job '%s'", 6276d554e2SKevin Wolf BlockJobType_lookup[job->driver->job_type], 6376d554e2SKevin Wolf job->id); 6476d554e2SKevin Wolf } 6576d554e2SKevin Wolf 6676d554e2SKevin Wolf static const BdrvChildRole child_job = { 6776d554e2SKevin Wolf .get_parent_desc = child_job_get_parent_desc, 6876d554e2SKevin Wolf .stay_at_node = true, 6976d554e2SKevin Wolf }; 7076d554e2SKevin Wolf 71600ac6a0SJohn Snow static void block_job_drained_begin(void *opaque) 72600ac6a0SJohn Snow { 73600ac6a0SJohn Snow BlockJob *job = opaque; 74600ac6a0SJohn Snow block_job_pause(job); 75600ac6a0SJohn Snow } 76600ac6a0SJohn Snow 77600ac6a0SJohn Snow static void block_job_drained_end(void *opaque) 78600ac6a0SJohn Snow { 79600ac6a0SJohn Snow BlockJob *job = opaque; 80600ac6a0SJohn Snow block_job_resume(job); 81600ac6a0SJohn Snow } 82600ac6a0SJohn Snow 83600ac6a0SJohn Snow static const BlockDevOps block_job_dev_ops = { 84600ac6a0SJohn Snow .drained_begin = block_job_drained_begin, 85600ac6a0SJohn Snow .drained_end = block_job_drained_end, 86600ac6a0SJohn Snow }; 87600ac6a0SJohn Snow 88a7112795SAlberto Garcia BlockJob *block_job_next(BlockJob *job) 89a7112795SAlberto Garcia { 90a7112795SAlberto Garcia if (!job) { 91a7112795SAlberto Garcia return QLIST_FIRST(&block_jobs); 92a7112795SAlberto Garcia } 93a7112795SAlberto Garcia return QLIST_NEXT(job, job_list); 94a7112795SAlberto Garcia } 95a7112795SAlberto Garcia 96ffb1f10cSAlberto Garcia BlockJob *block_job_get(const char *id) 97ffb1f10cSAlberto Garcia { 98ffb1f10cSAlberto Garcia BlockJob *job; 99ffb1f10cSAlberto Garcia 100ffb1f10cSAlberto Garcia QLIST_FOREACH(job, &block_jobs, job_list) { 101559b935fSJohn Snow if (job->id && !strcmp(id, job->id)) { 102ffb1f10cSAlberto Garcia return job; 103ffb1f10cSAlberto Garcia } 104ffb1f10cSAlberto Garcia } 105ffb1f10cSAlberto Garcia 106ffb1f10cSAlberto Garcia return NULL; 107ffb1f10cSAlberto Garcia } 108ffb1f10cSAlberto Garcia 109*05b0d8e3SPaolo Bonzini static void block_job_ref(BlockJob *job) 110*05b0d8e3SPaolo Bonzini { 111*05b0d8e3SPaolo Bonzini ++job->refcnt; 112*05b0d8e3SPaolo Bonzini } 113*05b0d8e3SPaolo Bonzini 114*05b0d8e3SPaolo Bonzini static void block_job_attached_aio_context(AioContext *new_context, 115*05b0d8e3SPaolo Bonzini void *opaque); 116*05b0d8e3SPaolo Bonzini static void block_job_detach_aio_context(void *opaque); 117*05b0d8e3SPaolo Bonzini 118*05b0d8e3SPaolo Bonzini static void block_job_unref(BlockJob *job) 119*05b0d8e3SPaolo Bonzini { 120*05b0d8e3SPaolo Bonzini if (--job->refcnt == 0) { 121*05b0d8e3SPaolo Bonzini BlockDriverState *bs = blk_bs(job->blk); 122*05b0d8e3SPaolo Bonzini bs->job = NULL; 123*05b0d8e3SPaolo Bonzini block_job_remove_all_bdrv(job); 124*05b0d8e3SPaolo Bonzini blk_remove_aio_context_notifier(job->blk, 125*05b0d8e3SPaolo Bonzini block_job_attached_aio_context, 126*05b0d8e3SPaolo Bonzini block_job_detach_aio_context, job); 127*05b0d8e3SPaolo Bonzini blk_unref(job->blk); 128*05b0d8e3SPaolo Bonzini error_free(job->blocker); 129*05b0d8e3SPaolo Bonzini g_free(job->id); 130*05b0d8e3SPaolo Bonzini QLIST_REMOVE(job, job_list); 131*05b0d8e3SPaolo Bonzini g_free(job); 132*05b0d8e3SPaolo Bonzini } 133*05b0d8e3SPaolo Bonzini } 134*05b0d8e3SPaolo Bonzini 135463e0be1SStefan Hajnoczi static void block_job_attached_aio_context(AioContext *new_context, 136463e0be1SStefan Hajnoczi void *opaque) 137463e0be1SStefan Hajnoczi { 138463e0be1SStefan Hajnoczi BlockJob *job = opaque; 139463e0be1SStefan Hajnoczi 140463e0be1SStefan Hajnoczi if (job->driver->attached_aio_context) { 141463e0be1SStefan Hajnoczi job->driver->attached_aio_context(job, new_context); 142463e0be1SStefan Hajnoczi } 143463e0be1SStefan Hajnoczi 144463e0be1SStefan Hajnoczi block_job_resume(job); 145463e0be1SStefan Hajnoczi } 146463e0be1SStefan Hajnoczi 147bae8196dSPaolo Bonzini static void block_job_drain(BlockJob *job) 148bae8196dSPaolo Bonzini { 149bae8196dSPaolo Bonzini /* If job is !job->busy this kicks it into the next pause point. */ 150bae8196dSPaolo Bonzini block_job_enter(job); 151bae8196dSPaolo Bonzini 152bae8196dSPaolo Bonzini blk_drain(job->blk); 153bae8196dSPaolo Bonzini if (job->driver->drain) { 154bae8196dSPaolo Bonzini job->driver->drain(job); 155bae8196dSPaolo Bonzini } 156bae8196dSPaolo Bonzini } 157bae8196dSPaolo Bonzini 158463e0be1SStefan Hajnoczi static void block_job_detach_aio_context(void *opaque) 159463e0be1SStefan Hajnoczi { 160463e0be1SStefan Hajnoczi BlockJob *job = opaque; 161463e0be1SStefan Hajnoczi 162463e0be1SStefan Hajnoczi /* In case the job terminates during aio_poll()... */ 163463e0be1SStefan Hajnoczi block_job_ref(job); 164463e0be1SStefan Hajnoczi 165463e0be1SStefan Hajnoczi block_job_pause(job); 166463e0be1SStefan Hajnoczi 167463e0be1SStefan Hajnoczi while (!job->paused && !job->completed) { 168bae8196dSPaolo Bonzini block_job_drain(job); 169463e0be1SStefan Hajnoczi } 170463e0be1SStefan Hajnoczi 171463e0be1SStefan Hajnoczi block_job_unref(job); 172463e0be1SStefan Hajnoczi } 173463e0be1SStefan Hajnoczi 174bbc02b90SKevin Wolf void block_job_remove_all_bdrv(BlockJob *job) 175bbc02b90SKevin Wolf { 176bbc02b90SKevin Wolf GSList *l; 177bbc02b90SKevin Wolf for (l = job->nodes; l; l = l->next) { 178bbc02b90SKevin Wolf BdrvChild *c = l->data; 179bbc02b90SKevin Wolf bdrv_op_unblock_all(c->bs, job->blocker); 180bbc02b90SKevin Wolf bdrv_root_unref_child(c); 181bbc02b90SKevin Wolf } 182bbc02b90SKevin Wolf g_slist_free(job->nodes); 183bbc02b90SKevin Wolf job->nodes = NULL; 184bbc02b90SKevin Wolf } 185bbc02b90SKevin Wolf 18676d554e2SKevin Wolf int block_job_add_bdrv(BlockJob *job, const char *name, BlockDriverState *bs, 18776d554e2SKevin Wolf uint64_t perm, uint64_t shared_perm, Error **errp) 18823d402d4SAlberto Garcia { 18976d554e2SKevin Wolf BdrvChild *c; 19076d554e2SKevin Wolf 19176d554e2SKevin Wolf c = bdrv_root_attach_child(bs, name, &child_job, perm, shared_perm, 19276d554e2SKevin Wolf job, errp); 19376d554e2SKevin Wolf if (c == NULL) { 19476d554e2SKevin Wolf return -EPERM; 19576d554e2SKevin Wolf } 19676d554e2SKevin Wolf 19776d554e2SKevin Wolf job->nodes = g_slist_prepend(job->nodes, c); 19823d402d4SAlberto Garcia bdrv_ref(bs); 19923d402d4SAlberto Garcia bdrv_op_block_all(bs, job->blocker); 20076d554e2SKevin Wolf 20176d554e2SKevin Wolf return 0; 20223d402d4SAlberto Garcia } 20323d402d4SAlberto Garcia 2047f0317cfSAlberto Garcia void *block_job_create(const char *job_id, const BlockJobDriver *driver, 205c6cc12bfSKevin Wolf BlockDriverState *bs, uint64_t perm, 206c6cc12bfSKevin Wolf uint64_t shared_perm, int64_t speed, int flags, 2077f0317cfSAlberto Garcia BlockCompletionFunc *cb, void *opaque, Error **errp) 2082f0c9fe6SPaolo Bonzini { 209b6d2e599SKevin Wolf BlockBackend *blk; 2102f0c9fe6SPaolo Bonzini BlockJob *job; 211d7086422SKevin Wolf int ret; 2122f0c9fe6SPaolo Bonzini 213628ff683SFam Zheng if (bs->job) { 214c6bd8c70SMarkus Armbruster error_setg(errp, QERR_DEVICE_IN_USE, bdrv_get_device_name(bs)); 2152f0c9fe6SPaolo Bonzini return NULL; 2162f0c9fe6SPaolo Bonzini } 217b6d2e599SKevin Wolf 218f81e0b45SJohn Snow if (job_id == NULL && !(flags & BLOCK_JOB_INTERNAL)) { 2197f0317cfSAlberto Garcia job_id = bdrv_get_device_name(bs); 2201562047cSKevin Wolf if (!*job_id) { 2211562047cSKevin Wolf error_setg(errp, "An explicit job ID is required for this node"); 2221562047cSKevin Wolf return NULL; 2231562047cSKevin Wolf } 2247f0317cfSAlberto Garcia } 2257f0317cfSAlberto Garcia 226f81e0b45SJohn Snow if (job_id) { 227f81e0b45SJohn Snow if (flags & BLOCK_JOB_INTERNAL) { 228f81e0b45SJohn Snow error_setg(errp, "Cannot specify job ID for internal block job"); 229f81e0b45SJohn Snow return NULL; 230f81e0b45SJohn Snow } 231f81e0b45SJohn Snow 2327f0317cfSAlberto Garcia if (!id_wellformed(job_id)) { 2337f0317cfSAlberto Garcia error_setg(errp, "Invalid job ID '%s'", job_id); 2347f0317cfSAlberto Garcia return NULL; 2357f0317cfSAlberto Garcia } 2367f0317cfSAlberto Garcia 2377f0317cfSAlberto Garcia if (block_job_get(job_id)) { 2387f0317cfSAlberto Garcia error_setg(errp, "Job ID '%s' already in use", job_id); 2397f0317cfSAlberto Garcia return NULL; 2407f0317cfSAlberto Garcia } 241f81e0b45SJohn Snow } 2427f0317cfSAlberto Garcia 243c6cc12bfSKevin Wolf blk = blk_new(perm, shared_perm); 244d7086422SKevin Wolf ret = blk_insert_bs(blk, bs, errp); 245d7086422SKevin Wolf if (ret < 0) { 246d7086422SKevin Wolf blk_unref(blk); 247d7086422SKevin Wolf return NULL; 248d7086422SKevin Wolf } 249b6d2e599SKevin Wolf 2503fc4b10aSFam Zheng job = g_malloc0(driver->instance_size); 2513fc4b10aSFam Zheng job->driver = driver; 2527f0317cfSAlberto Garcia job->id = g_strdup(job_id); 253b6d2e599SKevin Wolf job->blk = blk; 2542f0c9fe6SPaolo Bonzini job->cb = cb; 2552f0c9fe6SPaolo Bonzini job->opaque = opaque; 2565ccac6f1SJohn Snow job->busy = false; 2575ccac6f1SJohn Snow job->paused = true; 2585ccac6f1SJohn Snow job->pause_count = 1; 25918930ba3SFam Zheng job->refcnt = 1; 260600ac6a0SJohn Snow 261600ac6a0SJohn Snow error_setg(&job->blocker, "block device is in use by block job: %s", 262600ac6a0SJohn Snow BlockJobType_lookup[driver->job_type]); 263600ac6a0SJohn Snow block_job_add_bdrv(job, "main node", bs, 0, BLK_PERM_ALL, &error_abort); 2642f0c9fe6SPaolo Bonzini bs->job = job; 2652f0c9fe6SPaolo Bonzini 266600ac6a0SJohn Snow blk_set_dev_ops(blk, &block_job_dev_ops, job); 267600ac6a0SJohn Snow bdrv_op_unblock(bs, BLOCK_OP_TYPE_DATAPLANE, job->blocker); 268600ac6a0SJohn Snow 269a7112795SAlberto Garcia QLIST_INSERT_HEAD(&block_jobs, job, job_list); 270a7112795SAlberto Garcia 271463e0be1SStefan Hajnoczi blk_add_aio_context_notifier(blk, block_job_attached_aio_context, 272463e0be1SStefan Hajnoczi block_job_detach_aio_context, job); 273463e0be1SStefan Hajnoczi 2742f0c9fe6SPaolo Bonzini /* Only set speed when necessary to avoid NotSupported error */ 2752f0c9fe6SPaolo Bonzini if (speed != 0) { 2762f0c9fe6SPaolo Bonzini Error *local_err = NULL; 2772f0c9fe6SPaolo Bonzini 2782f0c9fe6SPaolo Bonzini block_job_set_speed(job, speed, &local_err); 27984d18f06SMarkus Armbruster if (local_err) { 28018930ba3SFam Zheng block_job_unref(job); 2812f0c9fe6SPaolo Bonzini error_propagate(errp, local_err); 2822f0c9fe6SPaolo Bonzini return NULL; 2832f0c9fe6SPaolo Bonzini } 2842f0c9fe6SPaolo Bonzini } 2852f0c9fe6SPaolo Bonzini return job; 2862f0c9fe6SPaolo Bonzini } 2872f0c9fe6SPaolo Bonzini 288559b935fSJohn Snow bool block_job_is_internal(BlockJob *job) 289559b935fSJohn Snow { 290559b935fSJohn Snow return (job->id == NULL); 291559b935fSJohn Snow } 292559b935fSJohn Snow 2935ccac6f1SJohn Snow static bool block_job_started(BlockJob *job) 2945ccac6f1SJohn Snow { 2955ccac6f1SJohn Snow return job->co; 2965ccac6f1SJohn Snow } 2975ccac6f1SJohn Snow 298e3796a24SJohn Snow /** 299e3796a24SJohn Snow * All jobs must allow a pause point before entering their job proper. This 300e3796a24SJohn Snow * ensures that jobs can be paused prior to being started, then resumed later. 301e3796a24SJohn Snow */ 302e3796a24SJohn Snow static void coroutine_fn block_job_co_entry(void *opaque) 303e3796a24SJohn Snow { 304e3796a24SJohn Snow BlockJob *job = opaque; 305e3796a24SJohn Snow 306e3796a24SJohn Snow assert(job && job->driver && job->driver->start); 307e3796a24SJohn Snow block_job_pause_point(job); 308e3796a24SJohn Snow job->driver->start(job); 309e3796a24SJohn Snow } 310e3796a24SJohn Snow 3115ccac6f1SJohn Snow void block_job_start(BlockJob *job) 3125ccac6f1SJohn Snow { 3135ccac6f1SJohn Snow assert(job && !block_job_started(job) && job->paused && 314e3796a24SJohn Snow job->driver && job->driver->start); 315e3796a24SJohn Snow job->co = qemu_coroutine_create(block_job_co_entry, job); 316e3796a24SJohn Snow job->pause_count--; 3175ccac6f1SJohn Snow job->busy = true; 318e3796a24SJohn Snow job->paused = false; 319aef4278cSFam Zheng bdrv_coroutine_enter(blk_bs(job->blk), job->co); 3205ccac6f1SJohn Snow } 3215ccac6f1SJohn Snow 322*05b0d8e3SPaolo Bonzini void block_job_early_fail(BlockJob *job) 32397031164STing Wang { 324*05b0d8e3SPaolo Bonzini block_job_unref(job); 32518930ba3SFam Zheng } 32697031164STing Wang 327c55a832fSFam Zheng static void block_job_completed_single(BlockJob *job) 328c55a832fSFam Zheng { 329c55a832fSFam Zheng if (!job->ret) { 330c55a832fSFam Zheng if (job->driver->commit) { 331c55a832fSFam Zheng job->driver->commit(job); 332c55a832fSFam Zheng } 333c55a832fSFam Zheng } else { 334c55a832fSFam Zheng if (job->driver->abort) { 335c55a832fSFam Zheng job->driver->abort(job); 336c55a832fSFam Zheng } 337c55a832fSFam Zheng } 338e8a40bf7SJohn Snow if (job->driver->clean) { 339e8a40bf7SJohn Snow job->driver->clean(job); 340e8a40bf7SJohn Snow } 3418254b6d9SJohn Snow 3428254b6d9SJohn Snow if (job->cb) { 343c55a832fSFam Zheng job->cb(job->opaque, job->ret); 3448254b6d9SJohn Snow } 3455ccac6f1SJohn Snow 3465ccac6f1SJohn Snow /* Emit events only if we actually started */ 3475ccac6f1SJohn Snow if (block_job_started(job)) { 3488254b6d9SJohn Snow if (block_job_is_cancelled(job)) { 3498254b6d9SJohn Snow block_job_event_cancelled(job); 3508254b6d9SJohn Snow } else { 3518254b6d9SJohn Snow const char *msg = NULL; 3528254b6d9SJohn Snow if (job->ret < 0) { 3538254b6d9SJohn Snow msg = strerror(-job->ret); 3548254b6d9SJohn Snow } 3558254b6d9SJohn Snow block_job_event_completed(job, msg); 3568254b6d9SJohn Snow } 3575ccac6f1SJohn Snow } 3588254b6d9SJohn Snow 359c55a832fSFam Zheng if (job->txn) { 3601e93b9fbSVladimir Sementsov-Ogievskiy QLIST_REMOVE(job, txn_list); 361c55a832fSFam Zheng block_job_txn_unref(job->txn); 362c55a832fSFam Zheng } 363c55a832fSFam Zheng block_job_unref(job); 364c55a832fSFam Zheng } 365c55a832fSFam Zheng 366c55a832fSFam Zheng static void block_job_completed_txn_abort(BlockJob *job) 367c55a832fSFam Zheng { 368c55a832fSFam Zheng AioContext *ctx; 369c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 370c55a832fSFam Zheng BlockJob *other_job, *next; 371c55a832fSFam Zheng 372c55a832fSFam Zheng if (txn->aborting) { 373c55a832fSFam Zheng /* 374c55a832fSFam Zheng * We are cancelled by another job, which will handle everything. 375c55a832fSFam Zheng */ 376c55a832fSFam Zheng return; 377c55a832fSFam Zheng } 378c55a832fSFam Zheng txn->aborting = true; 379c55a832fSFam Zheng /* We are the first failed job. Cancel other jobs. */ 380c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 381b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 382c55a832fSFam Zheng aio_context_acquire(ctx); 383c55a832fSFam Zheng } 384c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 385c55a832fSFam Zheng if (other_job == job || other_job->completed) { 386c55a832fSFam Zheng /* Other jobs are "effectively" cancelled by us, set the status for 387c55a832fSFam Zheng * them; this job, however, may or may not be cancelled, depending 388c55a832fSFam Zheng * on the caller, so leave it. */ 389c55a832fSFam Zheng if (other_job != job) { 390c55a832fSFam Zheng other_job->cancelled = true; 391c55a832fSFam Zheng } 392c55a832fSFam Zheng continue; 393c55a832fSFam Zheng } 394c55a832fSFam Zheng block_job_cancel_sync(other_job); 395c55a832fSFam Zheng assert(other_job->completed); 396c55a832fSFam Zheng } 397c55a832fSFam Zheng QLIST_FOREACH_SAFE(other_job, &txn->jobs, txn_list, next) { 398b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 399c55a832fSFam Zheng block_job_completed_single(other_job); 400c55a832fSFam Zheng aio_context_release(ctx); 401c55a832fSFam Zheng } 402c55a832fSFam Zheng } 403c55a832fSFam Zheng 404c55a832fSFam Zheng static void block_job_completed_txn_success(BlockJob *job) 405c55a832fSFam Zheng { 406c55a832fSFam Zheng AioContext *ctx; 407c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 408c55a832fSFam Zheng BlockJob *other_job, *next; 409c55a832fSFam Zheng /* 410c55a832fSFam Zheng * Successful completion, see if there are other running jobs in this 411c55a832fSFam Zheng * txn. 412c55a832fSFam Zheng */ 413c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 414c55a832fSFam Zheng if (!other_job->completed) { 415c55a832fSFam Zheng return; 416c55a832fSFam Zheng } 417c55a832fSFam Zheng } 418c55a832fSFam Zheng /* We are the last completed job, commit the transaction. */ 419c55a832fSFam Zheng QLIST_FOREACH_SAFE(other_job, &txn->jobs, txn_list, next) { 420b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 421c55a832fSFam Zheng aio_context_acquire(ctx); 422c55a832fSFam Zheng assert(other_job->ret == 0); 423c55a832fSFam Zheng block_job_completed_single(other_job); 424c55a832fSFam Zheng aio_context_release(ctx); 425c55a832fSFam Zheng } 426c55a832fSFam Zheng } 427c55a832fSFam Zheng 42865f46322SPaolo Bonzini void block_job_completed(BlockJob *job, int ret) 4292f0c9fe6SPaolo Bonzini { 430b6d2e599SKevin Wolf assert(blk_bs(job->blk)->job == job); 431a689dbf2SFam Zheng assert(!job->completed); 432a689dbf2SFam Zheng job->completed = true; 433a689dbf2SFam Zheng job->ret = ret; 434c55a832fSFam Zheng if (!job->txn) { 435c55a832fSFam Zheng block_job_completed_single(job); 436c55a832fSFam Zheng } else if (ret < 0 || block_job_is_cancelled(job)) { 437c55a832fSFam Zheng block_job_completed_txn_abort(job); 438c55a832fSFam Zheng } else { 439c55a832fSFam Zheng block_job_completed_txn_success(job); 440c55a832fSFam Zheng } 4412f0c9fe6SPaolo Bonzini } 4422f0c9fe6SPaolo Bonzini 4432f0c9fe6SPaolo Bonzini void block_job_set_speed(BlockJob *job, int64_t speed, Error **errp) 4442f0c9fe6SPaolo Bonzini { 4452f0c9fe6SPaolo Bonzini Error *local_err = NULL; 4462f0c9fe6SPaolo Bonzini 4473fc4b10aSFam Zheng if (!job->driver->set_speed) { 448c6bd8c70SMarkus Armbruster error_setg(errp, QERR_UNSUPPORTED); 4492f0c9fe6SPaolo Bonzini return; 4502f0c9fe6SPaolo Bonzini } 4513fc4b10aSFam Zheng job->driver->set_speed(job, speed, &local_err); 45284d18f06SMarkus Armbruster if (local_err) { 4532f0c9fe6SPaolo Bonzini error_propagate(errp, local_err); 4542f0c9fe6SPaolo Bonzini return; 4552f0c9fe6SPaolo Bonzini } 4562f0c9fe6SPaolo Bonzini 4572f0c9fe6SPaolo Bonzini job->speed = speed; 4582f0c9fe6SPaolo Bonzini } 4592f0c9fe6SPaolo Bonzini 460aeae883bSPaolo Bonzini void block_job_complete(BlockJob *job, Error **errp) 461aeae883bSPaolo Bonzini { 462559b935fSJohn Snow /* Should not be reachable via external interface for internal jobs */ 463559b935fSJohn Snow assert(job->id); 4645ccac6f1SJohn Snow if (job->pause_count || job->cancelled || 4655ccac6f1SJohn Snow !block_job_started(job) || !job->driver->complete) { 4669df229c3SAlberto Garcia error_setg(errp, "The active block job '%s' cannot be completed", 4679df229c3SAlberto Garcia job->id); 468aeae883bSPaolo Bonzini return; 469aeae883bSPaolo Bonzini } 470aeae883bSPaolo Bonzini 4713fc4b10aSFam Zheng job->driver->complete(job, errp); 472aeae883bSPaolo Bonzini } 473aeae883bSPaolo Bonzini 4748acc72a4SPaolo Bonzini void block_job_pause(BlockJob *job) 4752f0c9fe6SPaolo Bonzini { 476751ebd76SFam Zheng job->pause_count++; 4778acc72a4SPaolo Bonzini } 4788acc72a4SPaolo Bonzini 4790df4ba58SJohn Snow void block_job_user_pause(BlockJob *job) 4800df4ba58SJohn Snow { 4810df4ba58SJohn Snow job->user_paused = true; 4820df4ba58SJohn Snow block_job_pause(job); 4830df4ba58SJohn Snow } 4840df4ba58SJohn Snow 485a7f3b7ffSStefan Hajnoczi static bool block_job_should_pause(BlockJob *job) 4868acc72a4SPaolo Bonzini { 487751ebd76SFam Zheng return job->pause_count > 0; 4888acc72a4SPaolo Bonzini } 4898acc72a4SPaolo Bonzini 4900df4ba58SJohn Snow bool block_job_user_paused(BlockJob *job) 4910df4ba58SJohn Snow { 4926573d9c6SPaolo Bonzini return job->user_paused; 4930df4ba58SJohn Snow } 4940df4ba58SJohn Snow 495fc9c0a9cSStefan Hajnoczi void coroutine_fn block_job_pause_point(BlockJob *job) 496fc9c0a9cSStefan Hajnoczi { 4975ccac6f1SJohn Snow assert(job && block_job_started(job)); 4985ccac6f1SJohn Snow 499fc9c0a9cSStefan Hajnoczi if (!block_job_should_pause(job)) { 500fc9c0a9cSStefan Hajnoczi return; 501fc9c0a9cSStefan Hajnoczi } 502fc9c0a9cSStefan Hajnoczi if (block_job_is_cancelled(job)) { 503fc9c0a9cSStefan Hajnoczi return; 504fc9c0a9cSStefan Hajnoczi } 505fc9c0a9cSStefan Hajnoczi 506fc9c0a9cSStefan Hajnoczi if (job->driver->pause) { 507fc9c0a9cSStefan Hajnoczi job->driver->pause(job); 508fc9c0a9cSStefan Hajnoczi } 509fc9c0a9cSStefan Hajnoczi 510fc9c0a9cSStefan Hajnoczi if (block_job_should_pause(job) && !block_job_is_cancelled(job)) { 511fc9c0a9cSStefan Hajnoczi job->paused = true; 512fc9c0a9cSStefan Hajnoczi job->busy = false; 513fc9c0a9cSStefan Hajnoczi qemu_coroutine_yield(); /* wait for block_job_resume() */ 514fc9c0a9cSStefan Hajnoczi job->busy = true; 515fc9c0a9cSStefan Hajnoczi job->paused = false; 516fc9c0a9cSStefan Hajnoczi } 517fc9c0a9cSStefan Hajnoczi 518fc9c0a9cSStefan Hajnoczi if (job->driver->resume) { 519fc9c0a9cSStefan Hajnoczi job->driver->resume(job); 520fc9c0a9cSStefan Hajnoczi } 521fc9c0a9cSStefan Hajnoczi } 522fc9c0a9cSStefan Hajnoczi 5238acc72a4SPaolo Bonzini void block_job_resume(BlockJob *job) 5248acc72a4SPaolo Bonzini { 525751ebd76SFam Zheng assert(job->pause_count > 0); 526751ebd76SFam Zheng job->pause_count--; 527751ebd76SFam Zheng if (job->pause_count) { 528751ebd76SFam Zheng return; 529751ebd76SFam Zheng } 530751ebd76SFam Zheng block_job_enter(job); 531751ebd76SFam Zheng } 532751ebd76SFam Zheng 5330df4ba58SJohn Snow void block_job_user_resume(BlockJob *job) 5340df4ba58SJohn Snow { 5350df4ba58SJohn Snow if (job && job->user_paused && job->pause_count > 0) { 5360df4ba58SJohn Snow job->user_paused = false; 5370df4ba58SJohn Snow block_job_resume(job); 5380df4ba58SJohn Snow } 5390df4ba58SJohn Snow } 5400df4ba58SJohn Snow 541751ebd76SFam Zheng void block_job_enter(BlockJob *job) 542751ebd76SFam Zheng { 5432f0c9fe6SPaolo Bonzini if (job->co && !job->busy) { 544aef4278cSFam Zheng bdrv_coroutine_enter(blk_bs(job->blk), job->co); 5452f0c9fe6SPaolo Bonzini } 5462f0c9fe6SPaolo Bonzini } 5472f0c9fe6SPaolo Bonzini 5488acc72a4SPaolo Bonzini void block_job_cancel(BlockJob *job) 5498acc72a4SPaolo Bonzini { 5505ccac6f1SJohn Snow if (block_job_started(job)) { 5518acc72a4SPaolo Bonzini job->cancelled = true; 55217bd51f9SStefan Hajnoczi block_job_iostatus_reset(job); 553751ebd76SFam Zheng block_job_enter(job); 5545ccac6f1SJohn Snow } else { 5555ccac6f1SJohn Snow block_job_completed(job, -ECANCELED); 5565ccac6f1SJohn Snow } 5578acc72a4SPaolo Bonzini } 5588acc72a4SPaolo Bonzini 5592f0c9fe6SPaolo Bonzini bool block_job_is_cancelled(BlockJob *job) 5602f0c9fe6SPaolo Bonzini { 5612f0c9fe6SPaolo Bonzini return job->cancelled; 5622f0c9fe6SPaolo Bonzini } 5632f0c9fe6SPaolo Bonzini 56432c81a4aSPaolo Bonzini void block_job_iostatus_reset(BlockJob *job) 56532c81a4aSPaolo Bonzini { 56632c81a4aSPaolo Bonzini job->iostatus = BLOCK_DEVICE_IO_STATUS_OK; 56732c81a4aSPaolo Bonzini } 56832c81a4aSPaolo Bonzini 569345f9e1bSMax Reitz static int block_job_finish_sync(BlockJob *job, 570345f9e1bSMax Reitz void (*finish)(BlockJob *, Error **errp), 571345f9e1bSMax Reitz Error **errp) 5722f0c9fe6SPaolo Bonzini { 573345f9e1bSMax Reitz Error *local_err = NULL; 57494db6d2dSFam Zheng int ret; 5752f0c9fe6SPaolo Bonzini 576b6d2e599SKevin Wolf assert(blk_bs(job->blk)->job == job); 5772f0c9fe6SPaolo Bonzini 57894db6d2dSFam Zheng block_job_ref(job); 579bae8196dSPaolo Bonzini 580345f9e1bSMax Reitz finish(job, &local_err); 581345f9e1bSMax Reitz if (local_err) { 582345f9e1bSMax Reitz error_propagate(errp, local_err); 58394db6d2dSFam Zheng block_job_unref(job); 584345f9e1bSMax Reitz return -EBUSY; 585345f9e1bSMax Reitz } 586bae8196dSPaolo Bonzini /* block_job_drain calls block_job_enter, and it should be enough to 587bae8196dSPaolo Bonzini * induce progress until the job completes or moves to the main thread. 588bae8196dSPaolo Bonzini */ 589bae8196dSPaolo Bonzini while (!job->deferred_to_main_loop && !job->completed) { 590bae8196dSPaolo Bonzini block_job_drain(job); 591bae8196dSPaolo Bonzini } 59294db6d2dSFam Zheng while (!job->completed) { 593bae8196dSPaolo Bonzini aio_poll(qemu_get_aio_context(), true); 5942f0c9fe6SPaolo Bonzini } 59594db6d2dSFam Zheng ret = (job->cancelled && job->ret == 0) ? -ECANCELED : job->ret; 59694db6d2dSFam Zheng block_job_unref(job); 59794db6d2dSFam Zheng return ret; 5982f0c9fe6SPaolo Bonzini } 5992f0c9fe6SPaolo Bonzini 600345f9e1bSMax Reitz /* A wrapper around block_job_cancel() taking an Error ** parameter so it may be 601345f9e1bSMax Reitz * used with block_job_finish_sync() without the need for (rather nasty) 602345f9e1bSMax Reitz * function pointer casts there. */ 603345f9e1bSMax Reitz static void block_job_cancel_err(BlockJob *job, Error **errp) 604345f9e1bSMax Reitz { 605345f9e1bSMax Reitz block_job_cancel(job); 606345f9e1bSMax Reitz } 607345f9e1bSMax Reitz 608345f9e1bSMax Reitz int block_job_cancel_sync(BlockJob *job) 609345f9e1bSMax Reitz { 610345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_cancel_err, NULL); 611345f9e1bSMax Reitz } 612345f9e1bSMax Reitz 613a1a2af07SKevin Wolf void block_job_cancel_sync_all(void) 614a1a2af07SKevin Wolf { 615a1a2af07SKevin Wolf BlockJob *job; 616a1a2af07SKevin Wolf AioContext *aio_context; 617a1a2af07SKevin Wolf 618a1a2af07SKevin Wolf while ((job = QLIST_FIRST(&block_jobs))) { 619b6d2e599SKevin Wolf aio_context = blk_get_aio_context(job->blk); 620a1a2af07SKevin Wolf aio_context_acquire(aio_context); 621a1a2af07SKevin Wolf block_job_cancel_sync(job); 622a1a2af07SKevin Wolf aio_context_release(aio_context); 623a1a2af07SKevin Wolf } 624a1a2af07SKevin Wolf } 625a1a2af07SKevin Wolf 626345f9e1bSMax Reitz int block_job_complete_sync(BlockJob *job, Error **errp) 627345f9e1bSMax Reitz { 628345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_complete, errp); 629345f9e1bSMax Reitz } 630345f9e1bSMax Reitz 6317483d1e5SAlex Bligh void block_job_sleep_ns(BlockJob *job, QEMUClockType type, int64_t ns) 6322f0c9fe6SPaolo Bonzini { 6338acc72a4SPaolo Bonzini assert(job->busy); 6348acc72a4SPaolo Bonzini 6352f0c9fe6SPaolo Bonzini /* Check cancellation *before* setting busy = false, too! */ 6368acc72a4SPaolo Bonzini if (block_job_is_cancelled(job)) { 6378acc72a4SPaolo Bonzini return; 6382f0c9fe6SPaolo Bonzini } 6398acc72a4SPaolo Bonzini 6408acc72a4SPaolo Bonzini job->busy = false; 641a7f3b7ffSStefan Hajnoczi if (!block_job_should_pause(job)) { 6420824afdaSAlberto Garcia co_aio_sleep_ns(blk_get_aio_context(job->blk), type, ns); 6430824afdaSAlberto Garcia } 6448acc72a4SPaolo Bonzini job->busy = true; 645fc9c0a9cSStefan Hajnoczi 646fc9c0a9cSStefan Hajnoczi block_job_pause_point(job); 6472f0c9fe6SPaolo Bonzini } 64830e628b7SPaolo Bonzini 649dc71ce45SFam Zheng void block_job_yield(BlockJob *job) 650dc71ce45SFam Zheng { 651dc71ce45SFam Zheng assert(job->busy); 652dc71ce45SFam Zheng 653dc71ce45SFam Zheng /* Check cancellation *before* setting busy = false, too! */ 654dc71ce45SFam Zheng if (block_job_is_cancelled(job)) { 655dc71ce45SFam Zheng return; 656dc71ce45SFam Zheng } 657dc71ce45SFam Zheng 658dc71ce45SFam Zheng job->busy = false; 659fc9c0a9cSStefan Hajnoczi if (!block_job_should_pause(job)) { 660dc71ce45SFam Zheng qemu_coroutine_yield(); 661fc9c0a9cSStefan Hajnoczi } 662dc71ce45SFam Zheng job->busy = true; 663fc9c0a9cSStefan Hajnoczi 664fc9c0a9cSStefan Hajnoczi block_job_pause_point(job); 665dc71ce45SFam Zheng } 666dc71ce45SFam Zheng 667559b935fSJohn Snow BlockJobInfo *block_job_query(BlockJob *job, Error **errp) 66830e628b7SPaolo Bonzini { 669559b935fSJohn Snow BlockJobInfo *info; 670559b935fSJohn Snow 671559b935fSJohn Snow if (block_job_is_internal(job)) { 672559b935fSJohn Snow error_setg(errp, "Cannot query QEMU internal jobs"); 673559b935fSJohn Snow return NULL; 674559b935fSJohn Snow } 675559b935fSJohn Snow info = g_new0(BlockJobInfo, 1); 67679e14bf7SFam Zheng info->type = g_strdup(BlockJobType_lookup[job->driver->job_type]); 6778ccb9569SKevin Wolf info->device = g_strdup(job->id); 67830e628b7SPaolo Bonzini info->len = job->len; 6798d65883fSPaolo Bonzini info->busy = job->busy; 680751ebd76SFam Zheng info->paused = job->pause_count > 0; 68130e628b7SPaolo Bonzini info->offset = job->offset; 68230e628b7SPaolo Bonzini info->speed = job->speed; 68332c81a4aSPaolo Bonzini info->io_status = job->iostatus; 684ef6dbf1eSMax Reitz info->ready = job->ready; 68530e628b7SPaolo Bonzini return info; 68630e628b7SPaolo Bonzini } 68732c81a4aSPaolo Bonzini 68832c81a4aSPaolo Bonzini static void block_job_iostatus_set_err(BlockJob *job, int error) 68932c81a4aSPaolo Bonzini { 69032c81a4aSPaolo Bonzini if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 69132c81a4aSPaolo Bonzini job->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE : 69232c81a4aSPaolo Bonzini BLOCK_DEVICE_IO_STATUS_FAILED; 69332c81a4aSPaolo Bonzini } 69432c81a4aSPaolo Bonzini } 69532c81a4aSPaolo Bonzini 6968254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job) 697a66a2a36SPaolo Bonzini { 698559b935fSJohn Snow if (block_job_is_internal(job)) { 699559b935fSJohn Snow return; 700559b935fSJohn Snow } 701559b935fSJohn Snow 702bcada37bSWenchao Xia qapi_event_send_block_job_cancelled(job->driver->job_type, 7038ccb9569SKevin Wolf job->id, 704a66a2a36SPaolo Bonzini job->len, 705a66a2a36SPaolo Bonzini job->offset, 706bcada37bSWenchao Xia job->speed, 707bcada37bSWenchao Xia &error_abort); 708a66a2a36SPaolo Bonzini } 709a66a2a36SPaolo Bonzini 7108254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg) 711a66a2a36SPaolo Bonzini { 712559b935fSJohn Snow if (block_job_is_internal(job)) { 713559b935fSJohn Snow return; 714559b935fSJohn Snow } 715559b935fSJohn Snow 716bcada37bSWenchao Xia qapi_event_send_block_job_completed(job->driver->job_type, 7178ccb9569SKevin Wolf job->id, 718bcada37bSWenchao Xia job->len, 719bcada37bSWenchao Xia job->offset, 720bcada37bSWenchao Xia job->speed, 721bcada37bSWenchao Xia !!msg, 722bcada37bSWenchao Xia msg, 723bcada37bSWenchao Xia &error_abort); 724bcada37bSWenchao Xia } 725bcada37bSWenchao Xia 726bcada37bSWenchao Xia void block_job_event_ready(BlockJob *job) 727bcada37bSWenchao Xia { 728ef6dbf1eSMax Reitz job->ready = true; 729ef6dbf1eSMax Reitz 730559b935fSJohn Snow if (block_job_is_internal(job)) { 731559b935fSJohn Snow return; 732559b935fSJohn Snow } 733559b935fSJohn Snow 734518848a2SMarkus Armbruster qapi_event_send_block_job_ready(job->driver->job_type, 7358ccb9569SKevin Wolf job->id, 736518848a2SMarkus Armbruster job->len, 737518848a2SMarkus Armbruster job->offset, 738518848a2SMarkus Armbruster job->speed, &error_abort); 739a66a2a36SPaolo Bonzini } 740a66a2a36SPaolo Bonzini 74181e254dcSKevin Wolf BlockErrorAction block_job_error_action(BlockJob *job, BlockdevOnError on_err, 74232c81a4aSPaolo Bonzini int is_read, int error) 74332c81a4aSPaolo Bonzini { 74432c81a4aSPaolo Bonzini BlockErrorAction action; 74532c81a4aSPaolo Bonzini 74632c81a4aSPaolo Bonzini switch (on_err) { 74732c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_ENOSPC: 7488c398252SKevin Wolf case BLOCKDEV_ON_ERROR_AUTO: 749a589569fSWenchao Xia action = (error == ENOSPC) ? 750a589569fSWenchao Xia BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT; 75132c81a4aSPaolo Bonzini break; 75232c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_STOP: 753a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_STOP; 75432c81a4aSPaolo Bonzini break; 75532c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_REPORT: 756a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_REPORT; 75732c81a4aSPaolo Bonzini break; 75832c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_IGNORE: 759a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_IGNORE; 76032c81a4aSPaolo Bonzini break; 76132c81a4aSPaolo Bonzini default: 76232c81a4aSPaolo Bonzini abort(); 76332c81a4aSPaolo Bonzini } 764559b935fSJohn Snow if (!block_job_is_internal(job)) { 7658ccb9569SKevin Wolf qapi_event_send_block_job_error(job->id, 7665a2d2cbdSWenchao Xia is_read ? IO_OPERATION_TYPE_READ : 7675a2d2cbdSWenchao Xia IO_OPERATION_TYPE_WRITE, 7685a2d2cbdSWenchao Xia action, &error_abort); 769559b935fSJohn Snow } 770a589569fSWenchao Xia if (action == BLOCK_ERROR_ACTION_STOP) { 771751ebd76SFam Zheng /* make the pause user visible, which will be resumed from QMP. */ 7720df4ba58SJohn Snow block_job_user_pause(job); 77332c81a4aSPaolo Bonzini block_job_iostatus_set_err(job, error); 77432c81a4aSPaolo Bonzini } 77532c81a4aSPaolo Bonzini return action; 77632c81a4aSPaolo Bonzini } 777dec7d421SStefan Hajnoczi 778dec7d421SStefan Hajnoczi typedef struct { 779dec7d421SStefan Hajnoczi BlockJob *job; 780dec7d421SStefan Hajnoczi AioContext *aio_context; 781dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn; 782dec7d421SStefan Hajnoczi void *opaque; 783dec7d421SStefan Hajnoczi } BlockJobDeferToMainLoopData; 784dec7d421SStefan Hajnoczi 785dec7d421SStefan Hajnoczi static void block_job_defer_to_main_loop_bh(void *opaque) 786dec7d421SStefan Hajnoczi { 787dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = opaque; 788dec7d421SStefan Hajnoczi AioContext *aio_context; 789dec7d421SStefan Hajnoczi 790dec7d421SStefan Hajnoczi /* Prevent race with block_job_defer_to_main_loop() */ 791dec7d421SStefan Hajnoczi aio_context_acquire(data->aio_context); 792dec7d421SStefan Hajnoczi 793dec7d421SStefan Hajnoczi /* Fetch BDS AioContext again, in case it has changed */ 794b6d2e599SKevin Wolf aio_context = blk_get_aio_context(data->job->blk); 795d79df2a2SPaolo Bonzini if (aio_context != data->aio_context) { 796dec7d421SStefan Hajnoczi aio_context_acquire(aio_context); 797d79df2a2SPaolo Bonzini } 798dec7d421SStefan Hajnoczi 799794f0141SFam Zheng data->job->deferred_to_main_loop = false; 800dec7d421SStefan Hajnoczi data->fn(data->job, data->opaque); 801dec7d421SStefan Hajnoczi 802d79df2a2SPaolo Bonzini if (aio_context != data->aio_context) { 803dec7d421SStefan Hajnoczi aio_context_release(aio_context); 804d79df2a2SPaolo Bonzini } 805dec7d421SStefan Hajnoczi 806dec7d421SStefan Hajnoczi aio_context_release(data->aio_context); 807dec7d421SStefan Hajnoczi 808dec7d421SStefan Hajnoczi g_free(data); 809dec7d421SStefan Hajnoczi } 810dec7d421SStefan Hajnoczi 811dec7d421SStefan Hajnoczi void block_job_defer_to_main_loop(BlockJob *job, 812dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn, 813dec7d421SStefan Hajnoczi void *opaque) 814dec7d421SStefan Hajnoczi { 815dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = g_malloc(sizeof(*data)); 816dec7d421SStefan Hajnoczi data->job = job; 817b6d2e599SKevin Wolf data->aio_context = blk_get_aio_context(job->blk); 818dec7d421SStefan Hajnoczi data->fn = fn; 819dec7d421SStefan Hajnoczi data->opaque = opaque; 820794f0141SFam Zheng job->deferred_to_main_loop = true; 821dec7d421SStefan Hajnoczi 822fffb6e12SPaolo Bonzini aio_bh_schedule_oneshot(qemu_get_aio_context(), 823fffb6e12SPaolo Bonzini block_job_defer_to_main_loop_bh, data); 824dec7d421SStefan Hajnoczi } 825c55a832fSFam Zheng 826c55a832fSFam Zheng BlockJobTxn *block_job_txn_new(void) 827c55a832fSFam Zheng { 828c55a832fSFam Zheng BlockJobTxn *txn = g_new0(BlockJobTxn, 1); 829c55a832fSFam Zheng QLIST_INIT(&txn->jobs); 830c55a832fSFam Zheng txn->refcnt = 1; 831c55a832fSFam Zheng return txn; 832c55a832fSFam Zheng } 833c55a832fSFam Zheng 834c55a832fSFam Zheng static void block_job_txn_ref(BlockJobTxn *txn) 835c55a832fSFam Zheng { 836c55a832fSFam Zheng txn->refcnt++; 837c55a832fSFam Zheng } 838c55a832fSFam Zheng 839c55a832fSFam Zheng void block_job_txn_unref(BlockJobTxn *txn) 840c55a832fSFam Zheng { 841c55a832fSFam Zheng if (txn && --txn->refcnt == 0) { 842c55a832fSFam Zheng g_free(txn); 843c55a832fSFam Zheng } 844c55a832fSFam Zheng } 845c55a832fSFam Zheng 846c55a832fSFam Zheng void block_job_txn_add_job(BlockJobTxn *txn, BlockJob *job) 847c55a832fSFam Zheng { 848c55a832fSFam Zheng if (!txn) { 849c55a832fSFam Zheng return; 850c55a832fSFam Zheng } 851c55a832fSFam Zheng 852c55a832fSFam Zheng assert(!job->txn); 853c55a832fSFam Zheng job->txn = txn; 854c55a832fSFam Zheng 855c55a832fSFam Zheng QLIST_INSERT_HEAD(&txn->jobs, job, txn_list); 856c55a832fSFam Zheng block_job_txn_ref(txn); 857c55a832fSFam Zheng } 858