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" 32e688df6bSMarkus Armbruster #include "qapi/error.h" 33*9af23989SMarkus Armbruster #include "qapi/qapi-events-block-core.h" 34cc7a8ea7SMarkus Armbruster #include "qapi/qmp/qerror.h" 3510817bf0SDaniel P. Berrange #include "qemu/coroutine.h" 367f0317cfSAlberto Garcia #include "qemu/id.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); 61aa9ef2e6SJohn Snow static void block_job_enter_cond(BlockJob *job, bool(*fn)(BlockJob *job)); 628254b6d9SJohn Snow 63c55a832fSFam Zheng /* Transactional group of block jobs */ 64c55a832fSFam Zheng struct BlockJobTxn { 65c55a832fSFam Zheng 66c55a832fSFam Zheng /* Is this txn being cancelled? */ 67c55a832fSFam Zheng bool aborting; 68c55a832fSFam Zheng 69c55a832fSFam Zheng /* List of jobs */ 70c55a832fSFam Zheng QLIST_HEAD(, BlockJob) jobs; 71c55a832fSFam Zheng 72c55a832fSFam Zheng /* Reference count */ 73c55a832fSFam Zheng int refcnt; 74c55a832fSFam Zheng }; 75c55a832fSFam Zheng 76a7112795SAlberto Garcia static QLIST_HEAD(, BlockJob) block_jobs = QLIST_HEAD_INITIALIZER(block_jobs); 77a7112795SAlberto Garcia 7888691b37SPaolo Bonzini /* 7988691b37SPaolo Bonzini * The block job API is composed of two categories of functions. 8088691b37SPaolo Bonzini * 8188691b37SPaolo Bonzini * The first includes functions used by the monitor. The monitor is 8288691b37SPaolo Bonzini * peculiar in that it accesses the block job list with block_job_get, and 8388691b37SPaolo Bonzini * therefore needs consistency across block_job_get and the actual operation 8488691b37SPaolo Bonzini * (e.g. block_job_set_speed). The consistency is achieved with 8588691b37SPaolo Bonzini * aio_context_acquire/release. These functions are declared in blockjob.h. 8688691b37SPaolo Bonzini * 8788691b37SPaolo Bonzini * The second includes functions used by the block job drivers and sometimes 8888691b37SPaolo Bonzini * by the core block layer. These do not care about locking, because the 8988691b37SPaolo Bonzini * whole coroutine runs under the AioContext lock, and are declared in 9088691b37SPaolo Bonzini * blockjob_int.h. 9188691b37SPaolo Bonzini */ 9288691b37SPaolo Bonzini 93a7112795SAlberto Garcia BlockJob *block_job_next(BlockJob *job) 94a7112795SAlberto Garcia { 95a7112795SAlberto Garcia if (!job) { 96a7112795SAlberto Garcia return QLIST_FIRST(&block_jobs); 97a7112795SAlberto Garcia } 98a7112795SAlberto Garcia return QLIST_NEXT(job, job_list); 99a7112795SAlberto Garcia } 100a7112795SAlberto Garcia 101ffb1f10cSAlberto Garcia BlockJob *block_job_get(const char *id) 102ffb1f10cSAlberto Garcia { 103ffb1f10cSAlberto Garcia BlockJob *job; 104ffb1f10cSAlberto Garcia 105ffb1f10cSAlberto Garcia QLIST_FOREACH(job, &block_jobs, job_list) { 106559b935fSJohn Snow if (job->id && !strcmp(id, job->id)) { 107ffb1f10cSAlberto Garcia return job; 108ffb1f10cSAlberto Garcia } 109ffb1f10cSAlberto Garcia } 110ffb1f10cSAlberto Garcia 111ffb1f10cSAlberto Garcia return NULL; 112ffb1f10cSAlberto Garcia } 113ffb1f10cSAlberto Garcia 114c8ab5c2dSPaolo Bonzini BlockJobTxn *block_job_txn_new(void) 115c8ab5c2dSPaolo Bonzini { 116c8ab5c2dSPaolo Bonzini BlockJobTxn *txn = g_new0(BlockJobTxn, 1); 117c8ab5c2dSPaolo Bonzini QLIST_INIT(&txn->jobs); 118c8ab5c2dSPaolo Bonzini txn->refcnt = 1; 119c8ab5c2dSPaolo Bonzini return txn; 120c8ab5c2dSPaolo Bonzini } 121c8ab5c2dSPaolo Bonzini 122c8ab5c2dSPaolo Bonzini static void block_job_txn_ref(BlockJobTxn *txn) 123c8ab5c2dSPaolo Bonzini { 124c8ab5c2dSPaolo Bonzini txn->refcnt++; 125c8ab5c2dSPaolo Bonzini } 126c8ab5c2dSPaolo Bonzini 127c8ab5c2dSPaolo Bonzini void block_job_txn_unref(BlockJobTxn *txn) 128c8ab5c2dSPaolo Bonzini { 129c8ab5c2dSPaolo Bonzini if (txn && --txn->refcnt == 0) { 130c8ab5c2dSPaolo Bonzini g_free(txn); 131c8ab5c2dSPaolo Bonzini } 132c8ab5c2dSPaolo Bonzini } 133c8ab5c2dSPaolo Bonzini 134c8ab5c2dSPaolo Bonzini void block_job_txn_add_job(BlockJobTxn *txn, BlockJob *job) 135c8ab5c2dSPaolo Bonzini { 136c8ab5c2dSPaolo Bonzini if (!txn) { 137c8ab5c2dSPaolo Bonzini return; 138c8ab5c2dSPaolo Bonzini } 139c8ab5c2dSPaolo Bonzini 140c8ab5c2dSPaolo Bonzini assert(!job->txn); 141c8ab5c2dSPaolo Bonzini job->txn = txn; 142c8ab5c2dSPaolo Bonzini 143c8ab5c2dSPaolo Bonzini QLIST_INSERT_HEAD(&txn->jobs, job, txn_list); 144c8ab5c2dSPaolo Bonzini block_job_txn_ref(txn); 145c8ab5c2dSPaolo Bonzini } 146c8ab5c2dSPaolo Bonzini 147f321dcb5SPaolo Bonzini static void block_job_pause(BlockJob *job) 148f321dcb5SPaolo Bonzini { 149f321dcb5SPaolo Bonzini job->pause_count++; 150f321dcb5SPaolo Bonzini } 151f321dcb5SPaolo Bonzini 152f321dcb5SPaolo Bonzini static void block_job_resume(BlockJob *job) 153f321dcb5SPaolo Bonzini { 154f321dcb5SPaolo Bonzini assert(job->pause_count > 0); 155f321dcb5SPaolo Bonzini job->pause_count--; 156f321dcb5SPaolo Bonzini if (job->pause_count) { 157f321dcb5SPaolo Bonzini return; 158f321dcb5SPaolo Bonzini } 159f321dcb5SPaolo Bonzini block_job_enter(job); 160f321dcb5SPaolo Bonzini } 161f321dcb5SPaolo Bonzini 1624172a003Ssochin.jiang void block_job_ref(BlockJob *job) 16305b0d8e3SPaolo Bonzini { 16405b0d8e3SPaolo Bonzini ++job->refcnt; 16505b0d8e3SPaolo Bonzini } 16605b0d8e3SPaolo Bonzini 16705b0d8e3SPaolo Bonzini static void block_job_attached_aio_context(AioContext *new_context, 16805b0d8e3SPaolo Bonzini void *opaque); 16905b0d8e3SPaolo Bonzini static void block_job_detach_aio_context(void *opaque); 17005b0d8e3SPaolo Bonzini 1714172a003Ssochin.jiang void block_job_unref(BlockJob *job) 17205b0d8e3SPaolo Bonzini { 17305b0d8e3SPaolo Bonzini if (--job->refcnt == 0) { 17405b0d8e3SPaolo Bonzini BlockDriverState *bs = blk_bs(job->blk); 1750a3e155fSAlberto Garcia QLIST_REMOVE(job, job_list); 17605b0d8e3SPaolo Bonzini bs->job = NULL; 17705b0d8e3SPaolo Bonzini block_job_remove_all_bdrv(job); 17805b0d8e3SPaolo Bonzini blk_remove_aio_context_notifier(job->blk, 17905b0d8e3SPaolo Bonzini block_job_attached_aio_context, 18005b0d8e3SPaolo Bonzini block_job_detach_aio_context, job); 18105b0d8e3SPaolo Bonzini blk_unref(job->blk); 18205b0d8e3SPaolo Bonzini error_free(job->blocker); 18305b0d8e3SPaolo Bonzini g_free(job->id); 184fc24908eSPaolo Bonzini assert(!timer_pending(&job->sleep_timer)); 18505b0d8e3SPaolo Bonzini g_free(job); 18605b0d8e3SPaolo Bonzini } 18705b0d8e3SPaolo Bonzini } 18805b0d8e3SPaolo Bonzini 189463e0be1SStefan Hajnoczi static void block_job_attached_aio_context(AioContext *new_context, 190463e0be1SStefan Hajnoczi void *opaque) 191463e0be1SStefan Hajnoczi { 192463e0be1SStefan Hajnoczi BlockJob *job = opaque; 193463e0be1SStefan Hajnoczi 194463e0be1SStefan Hajnoczi if (job->driver->attached_aio_context) { 195463e0be1SStefan Hajnoczi job->driver->attached_aio_context(job, new_context); 196463e0be1SStefan Hajnoczi } 197463e0be1SStefan Hajnoczi 198463e0be1SStefan Hajnoczi block_job_resume(job); 199463e0be1SStefan Hajnoczi } 200463e0be1SStefan Hajnoczi 201bae8196dSPaolo Bonzini static void block_job_drain(BlockJob *job) 202bae8196dSPaolo Bonzini { 203bae8196dSPaolo Bonzini /* If job is !job->busy this kicks it into the next pause point. */ 204bae8196dSPaolo Bonzini block_job_enter(job); 205bae8196dSPaolo Bonzini 206bae8196dSPaolo Bonzini blk_drain(job->blk); 207bae8196dSPaolo Bonzini if (job->driver->drain) { 208bae8196dSPaolo Bonzini job->driver->drain(job); 209bae8196dSPaolo Bonzini } 210bae8196dSPaolo Bonzini } 211bae8196dSPaolo Bonzini 212463e0be1SStefan Hajnoczi static void block_job_detach_aio_context(void *opaque) 213463e0be1SStefan Hajnoczi { 214463e0be1SStefan Hajnoczi BlockJob *job = opaque; 215463e0be1SStefan Hajnoczi 216463e0be1SStefan Hajnoczi /* In case the job terminates during aio_poll()... */ 217463e0be1SStefan Hajnoczi block_job_ref(job); 218463e0be1SStefan Hajnoczi 219463e0be1SStefan Hajnoczi block_job_pause(job); 220463e0be1SStefan Hajnoczi 221463e0be1SStefan Hajnoczi while (!job->paused && !job->completed) { 222bae8196dSPaolo Bonzini block_job_drain(job); 223463e0be1SStefan Hajnoczi } 224463e0be1SStefan Hajnoczi 225463e0be1SStefan Hajnoczi block_job_unref(job); 226463e0be1SStefan Hajnoczi } 227463e0be1SStefan Hajnoczi 228f321dcb5SPaolo Bonzini static char *child_job_get_parent_desc(BdrvChild *c) 229f321dcb5SPaolo Bonzini { 230f321dcb5SPaolo Bonzini BlockJob *job = c->opaque; 231f321dcb5SPaolo Bonzini return g_strdup_printf("%s job '%s'", 232977c736fSMarkus Armbruster BlockJobType_str(job->driver->job_type), 233f321dcb5SPaolo Bonzini job->id); 234f321dcb5SPaolo Bonzini } 235f321dcb5SPaolo Bonzini 236ad90febaSKevin Wolf static void child_job_drained_begin(BdrvChild *c) 237f321dcb5SPaolo Bonzini { 238ad90febaSKevin Wolf BlockJob *job = c->opaque; 239f321dcb5SPaolo Bonzini block_job_pause(job); 240f321dcb5SPaolo Bonzini } 241f321dcb5SPaolo Bonzini 242ad90febaSKevin Wolf static void child_job_drained_end(BdrvChild *c) 243f321dcb5SPaolo Bonzini { 244ad90febaSKevin Wolf BlockJob *job = c->opaque; 245f321dcb5SPaolo Bonzini block_job_resume(job); 246f321dcb5SPaolo Bonzini } 247f321dcb5SPaolo Bonzini 248ad90febaSKevin Wolf static const BdrvChildRole child_job = { 249ad90febaSKevin Wolf .get_parent_desc = child_job_get_parent_desc, 250ad90febaSKevin Wolf .drained_begin = child_job_drained_begin, 251ad90febaSKevin Wolf .drained_end = child_job_drained_end, 252ad90febaSKevin Wolf .stay_at_node = true, 253f321dcb5SPaolo Bonzini }; 254f321dcb5SPaolo Bonzini 255bbc02b90SKevin Wolf void block_job_remove_all_bdrv(BlockJob *job) 256bbc02b90SKevin Wolf { 257bbc02b90SKevin Wolf GSList *l; 258bbc02b90SKevin Wolf for (l = job->nodes; l; l = l->next) { 259bbc02b90SKevin Wolf BdrvChild *c = l->data; 260bbc02b90SKevin Wolf bdrv_op_unblock_all(c->bs, job->blocker); 261bbc02b90SKevin Wolf bdrv_root_unref_child(c); 262bbc02b90SKevin Wolf } 263bbc02b90SKevin Wolf g_slist_free(job->nodes); 264bbc02b90SKevin Wolf job->nodes = NULL; 265bbc02b90SKevin Wolf } 266bbc02b90SKevin Wolf 26776d554e2SKevin Wolf int block_job_add_bdrv(BlockJob *job, const char *name, BlockDriverState *bs, 26876d554e2SKevin Wolf uint64_t perm, uint64_t shared_perm, Error **errp) 26923d402d4SAlberto Garcia { 27076d554e2SKevin Wolf BdrvChild *c; 27176d554e2SKevin Wolf 27276d554e2SKevin Wolf c = bdrv_root_attach_child(bs, name, &child_job, perm, shared_perm, 27376d554e2SKevin Wolf job, errp); 27476d554e2SKevin Wolf if (c == NULL) { 27576d554e2SKevin Wolf return -EPERM; 27676d554e2SKevin Wolf } 27776d554e2SKevin Wolf 27876d554e2SKevin Wolf job->nodes = g_slist_prepend(job->nodes, c); 27923d402d4SAlberto Garcia bdrv_ref(bs); 28023d402d4SAlberto Garcia bdrv_op_block_all(bs, job->blocker); 28176d554e2SKevin Wolf 28276d554e2SKevin Wolf return 0; 28323d402d4SAlberto Garcia } 28423d402d4SAlberto Garcia 285559b935fSJohn Snow bool block_job_is_internal(BlockJob *job) 286559b935fSJohn Snow { 287559b935fSJohn Snow return (job->id == NULL); 288559b935fSJohn Snow } 289559b935fSJohn Snow 2905ccac6f1SJohn Snow static bool block_job_started(BlockJob *job) 2915ccac6f1SJohn Snow { 2925ccac6f1SJohn Snow return job->co; 2935ccac6f1SJohn Snow } 2945ccac6f1SJohn Snow 295e3796a24SJohn Snow /** 296e3796a24SJohn Snow * All jobs must allow a pause point before entering their job proper. This 297e3796a24SJohn Snow * ensures that jobs can be paused prior to being started, then resumed later. 298e3796a24SJohn Snow */ 299e3796a24SJohn Snow static void coroutine_fn block_job_co_entry(void *opaque) 300e3796a24SJohn Snow { 301e3796a24SJohn Snow BlockJob *job = opaque; 302e3796a24SJohn Snow 303e3796a24SJohn Snow assert(job && job->driver && job->driver->start); 304e3796a24SJohn Snow block_job_pause_point(job); 305e3796a24SJohn Snow job->driver->start(job); 306e3796a24SJohn Snow } 307e3796a24SJohn Snow 308fc24908eSPaolo Bonzini static void block_job_sleep_timer_cb(void *opaque) 309fc24908eSPaolo Bonzini { 310fc24908eSPaolo Bonzini BlockJob *job = opaque; 311fc24908eSPaolo Bonzini 312fc24908eSPaolo Bonzini block_job_enter(job); 313fc24908eSPaolo Bonzini } 314fc24908eSPaolo Bonzini 3155ccac6f1SJohn Snow void block_job_start(BlockJob *job) 3165ccac6f1SJohn Snow { 3175ccac6f1SJohn Snow assert(job && !block_job_started(job) && job->paused && 318e3796a24SJohn Snow job->driver && job->driver->start); 319e3796a24SJohn Snow job->co = qemu_coroutine_create(block_job_co_entry, job); 320e3796a24SJohn Snow job->pause_count--; 3215ccac6f1SJohn Snow job->busy = true; 322e3796a24SJohn Snow job->paused = false; 323aef4278cSFam Zheng bdrv_coroutine_enter(blk_bs(job->blk), job->co); 3245ccac6f1SJohn Snow } 3255ccac6f1SJohn Snow 326c55a832fSFam Zheng static void block_job_completed_single(BlockJob *job) 327c55a832fSFam Zheng { 3284fb588e9SPaolo Bonzini assert(job->completed); 3294fb588e9SPaolo Bonzini 330c55a832fSFam Zheng if (!job->ret) { 331c55a832fSFam Zheng if (job->driver->commit) { 332c55a832fSFam Zheng job->driver->commit(job); 333c55a832fSFam Zheng } 334c55a832fSFam Zheng } else { 335c55a832fSFam Zheng if (job->driver->abort) { 336c55a832fSFam Zheng job->driver->abort(job); 337c55a832fSFam Zheng } 338c55a832fSFam Zheng } 339e8a40bf7SJohn Snow if (job->driver->clean) { 340e8a40bf7SJohn Snow job->driver->clean(job); 341e8a40bf7SJohn Snow } 3428254b6d9SJohn Snow 3438254b6d9SJohn Snow if (job->cb) { 344c55a832fSFam Zheng job->cb(job->opaque, job->ret); 3458254b6d9SJohn Snow } 3465ccac6f1SJohn Snow 3475ccac6f1SJohn Snow /* Emit events only if we actually started */ 3485ccac6f1SJohn Snow if (block_job_started(job)) { 3498254b6d9SJohn Snow if (block_job_is_cancelled(job)) { 3508254b6d9SJohn Snow block_job_event_cancelled(job); 3518254b6d9SJohn Snow } else { 3528254b6d9SJohn Snow const char *msg = NULL; 3538254b6d9SJohn Snow if (job->ret < 0) { 3548254b6d9SJohn Snow msg = strerror(-job->ret); 3558254b6d9SJohn Snow } 3568254b6d9SJohn Snow block_job_event_completed(job, msg); 3578254b6d9SJohn Snow } 3585ccac6f1SJohn Snow } 3598254b6d9SJohn Snow 360c55a832fSFam Zheng if (job->txn) { 3611e93b9fbSVladimir Sementsov-Ogievskiy QLIST_REMOVE(job, txn_list); 362c55a832fSFam Zheng block_job_txn_unref(job->txn); 363c55a832fSFam Zheng } 364c55a832fSFam Zheng block_job_unref(job); 365c55a832fSFam Zheng } 366c55a832fSFam Zheng 3674c241cf5SPaolo Bonzini static void block_job_cancel_async(BlockJob *job) 3684c241cf5SPaolo Bonzini { 3694c241cf5SPaolo Bonzini if (job->iostatus != BLOCK_DEVICE_IO_STATUS_OK) { 3704c241cf5SPaolo Bonzini block_job_iostatus_reset(job); 3714c241cf5SPaolo Bonzini } 3724c241cf5SPaolo Bonzini if (job->user_paused) { 3734c241cf5SPaolo Bonzini /* Do not call block_job_enter here, the caller will handle it. */ 3744c241cf5SPaolo Bonzini job->user_paused = false; 3754c241cf5SPaolo Bonzini job->pause_count--; 3764c241cf5SPaolo Bonzini } 3774c241cf5SPaolo Bonzini job->cancelled = true; 3784c241cf5SPaolo Bonzini } 3794c241cf5SPaolo Bonzini 380c8ab5c2dSPaolo Bonzini static int block_job_finish_sync(BlockJob *job, 381c8ab5c2dSPaolo Bonzini void (*finish)(BlockJob *, Error **errp), 382c8ab5c2dSPaolo Bonzini Error **errp) 383c8ab5c2dSPaolo Bonzini { 384c8ab5c2dSPaolo Bonzini Error *local_err = NULL; 385c8ab5c2dSPaolo Bonzini int ret; 386c8ab5c2dSPaolo Bonzini 387c8ab5c2dSPaolo Bonzini assert(blk_bs(job->blk)->job == job); 388c8ab5c2dSPaolo Bonzini 389c8ab5c2dSPaolo Bonzini block_job_ref(job); 390c8ab5c2dSPaolo Bonzini 3914fb588e9SPaolo Bonzini if (finish) { 392c8ab5c2dSPaolo Bonzini finish(job, &local_err); 3934fb588e9SPaolo Bonzini } 394c8ab5c2dSPaolo Bonzini if (local_err) { 395c8ab5c2dSPaolo Bonzini error_propagate(errp, local_err); 396c8ab5c2dSPaolo Bonzini block_job_unref(job); 397c8ab5c2dSPaolo Bonzini return -EBUSY; 398c8ab5c2dSPaolo Bonzini } 399c8ab5c2dSPaolo Bonzini /* block_job_drain calls block_job_enter, and it should be enough to 400c8ab5c2dSPaolo Bonzini * induce progress until the job completes or moves to the main thread. 401c8ab5c2dSPaolo Bonzini */ 402c8ab5c2dSPaolo Bonzini while (!job->deferred_to_main_loop && !job->completed) { 403c8ab5c2dSPaolo Bonzini block_job_drain(job); 404c8ab5c2dSPaolo Bonzini } 405c8ab5c2dSPaolo Bonzini while (!job->completed) { 406c8ab5c2dSPaolo Bonzini aio_poll(qemu_get_aio_context(), true); 407c8ab5c2dSPaolo Bonzini } 408c8ab5c2dSPaolo Bonzini ret = (job->cancelled && job->ret == 0) ? -ECANCELED : job->ret; 409c8ab5c2dSPaolo Bonzini block_job_unref(job); 410c8ab5c2dSPaolo Bonzini return ret; 411c8ab5c2dSPaolo Bonzini } 412c8ab5c2dSPaolo Bonzini 413c55a832fSFam Zheng static void block_job_completed_txn_abort(BlockJob *job) 414c55a832fSFam Zheng { 415c55a832fSFam Zheng AioContext *ctx; 416c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 4174fb588e9SPaolo Bonzini BlockJob *other_job; 418c55a832fSFam Zheng 419c55a832fSFam Zheng if (txn->aborting) { 420c55a832fSFam Zheng /* 421c55a832fSFam Zheng * We are cancelled by another job, which will handle everything. 422c55a832fSFam Zheng */ 423c55a832fSFam Zheng return; 424c55a832fSFam Zheng } 425c55a832fSFam Zheng txn->aborting = true; 4264fb588e9SPaolo Bonzini block_job_txn_ref(txn); 4274fb588e9SPaolo Bonzini 428c55a832fSFam Zheng /* We are the first failed job. Cancel other jobs. */ 429c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 430b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 431c55a832fSFam Zheng aio_context_acquire(ctx); 432c55a832fSFam Zheng } 4334fb588e9SPaolo Bonzini 4344fb588e9SPaolo Bonzini /* Other jobs are effectively cancelled by us, set the status for 435c55a832fSFam Zheng * them; this job, however, may or may not be cancelled, depending 436c55a832fSFam Zheng * on the caller, so leave it. */ 4374fb588e9SPaolo Bonzini QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 438c55a832fSFam Zheng if (other_job != job) { 4394c241cf5SPaolo Bonzini block_job_cancel_async(other_job); 440c55a832fSFam Zheng } 441c55a832fSFam Zheng } 4424fb588e9SPaolo Bonzini while (!QLIST_EMPTY(&txn->jobs)) { 4434fb588e9SPaolo Bonzini other_job = QLIST_FIRST(&txn->jobs); 444b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 4454fb588e9SPaolo Bonzini if (!other_job->completed) { 4464fb588e9SPaolo Bonzini assert(other_job->cancelled); 4474fb588e9SPaolo Bonzini block_job_finish_sync(other_job, NULL, NULL); 4484fb588e9SPaolo Bonzini } 449c55a832fSFam Zheng block_job_completed_single(other_job); 450c55a832fSFam Zheng aio_context_release(ctx); 451c55a832fSFam Zheng } 4524fb588e9SPaolo Bonzini 4534fb588e9SPaolo Bonzini block_job_txn_unref(txn); 454c55a832fSFam Zheng } 455c55a832fSFam Zheng 456c55a832fSFam Zheng static void block_job_completed_txn_success(BlockJob *job) 457c55a832fSFam Zheng { 458c55a832fSFam Zheng AioContext *ctx; 459c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 460c55a832fSFam Zheng BlockJob *other_job, *next; 461c55a832fSFam Zheng /* 462c55a832fSFam Zheng * Successful completion, see if there are other running jobs in this 463c55a832fSFam Zheng * txn. 464c55a832fSFam Zheng */ 465c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 466c55a832fSFam Zheng if (!other_job->completed) { 467c55a832fSFam Zheng return; 468c55a832fSFam Zheng } 469c55a832fSFam Zheng } 470c55a832fSFam Zheng /* We are the last completed job, commit the transaction. */ 471c55a832fSFam Zheng QLIST_FOREACH_SAFE(other_job, &txn->jobs, txn_list, next) { 472b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 473c55a832fSFam Zheng aio_context_acquire(ctx); 474c55a832fSFam Zheng assert(other_job->ret == 0); 475c55a832fSFam Zheng block_job_completed_single(other_job); 476c55a832fSFam Zheng aio_context_release(ctx); 477c55a832fSFam Zheng } 478c55a832fSFam Zheng } 479c55a832fSFam Zheng 480aa9ef2e6SJohn Snow /* Assumes the block_job_mutex is held */ 481aa9ef2e6SJohn Snow static bool block_job_timer_pending(BlockJob *job) 482aa9ef2e6SJohn Snow { 483aa9ef2e6SJohn Snow return timer_pending(&job->sleep_timer); 484aa9ef2e6SJohn Snow } 485aa9ef2e6SJohn Snow 4862f0c9fe6SPaolo Bonzini void block_job_set_speed(BlockJob *job, int64_t speed, Error **errp) 4872f0c9fe6SPaolo Bonzini { 4882f0c9fe6SPaolo Bonzini Error *local_err = NULL; 489aa9ef2e6SJohn Snow int64_t old_speed = job->speed; 4902f0c9fe6SPaolo Bonzini 4913fc4b10aSFam Zheng if (!job->driver->set_speed) { 492c6bd8c70SMarkus Armbruster error_setg(errp, QERR_UNSUPPORTED); 4932f0c9fe6SPaolo Bonzini return; 4942f0c9fe6SPaolo Bonzini } 4953fc4b10aSFam Zheng job->driver->set_speed(job, speed, &local_err); 49684d18f06SMarkus Armbruster if (local_err) { 4972f0c9fe6SPaolo Bonzini error_propagate(errp, local_err); 4982f0c9fe6SPaolo Bonzini return; 4992f0c9fe6SPaolo Bonzini } 5002f0c9fe6SPaolo Bonzini 5012f0c9fe6SPaolo Bonzini job->speed = speed; 502aa9ef2e6SJohn Snow if (speed <= old_speed) { 503aa9ef2e6SJohn Snow return; 504aa9ef2e6SJohn Snow } 505aa9ef2e6SJohn Snow 506aa9ef2e6SJohn Snow /* kick only if a timer is pending */ 507aa9ef2e6SJohn Snow block_job_enter_cond(job, block_job_timer_pending); 5082f0c9fe6SPaolo Bonzini } 5092f0c9fe6SPaolo Bonzini 510aeae883bSPaolo Bonzini void block_job_complete(BlockJob *job, Error **errp) 511aeae883bSPaolo Bonzini { 512559b935fSJohn Snow /* Should not be reachable via external interface for internal jobs */ 513559b935fSJohn Snow assert(job->id); 5145ccac6f1SJohn Snow if (job->pause_count || job->cancelled || 5155ccac6f1SJohn Snow !block_job_started(job) || !job->driver->complete) { 5169df229c3SAlberto Garcia error_setg(errp, "The active block job '%s' cannot be completed", 5179df229c3SAlberto Garcia job->id); 518aeae883bSPaolo Bonzini return; 519aeae883bSPaolo Bonzini } 520aeae883bSPaolo Bonzini 5213fc4b10aSFam Zheng job->driver->complete(job, errp); 522aeae883bSPaolo Bonzini } 523aeae883bSPaolo Bonzini 5240df4ba58SJohn Snow void block_job_user_pause(BlockJob *job) 5250df4ba58SJohn Snow { 5260df4ba58SJohn Snow job->user_paused = true; 5270df4ba58SJohn Snow block_job_pause(job); 5280df4ba58SJohn Snow } 5290df4ba58SJohn Snow 5300df4ba58SJohn Snow bool block_job_user_paused(BlockJob *job) 5310df4ba58SJohn Snow { 5326573d9c6SPaolo Bonzini return job->user_paused; 5330df4ba58SJohn Snow } 5340df4ba58SJohn Snow 5350df4ba58SJohn Snow void block_job_user_resume(BlockJob *job) 5360df4ba58SJohn Snow { 5370df4ba58SJohn Snow if (job && job->user_paused && job->pause_count > 0) { 5382caf63a9SPaolo Bonzini block_job_iostatus_reset(job); 5394c241cf5SPaolo Bonzini job->user_paused = false; 5400df4ba58SJohn Snow block_job_resume(job); 5410df4ba58SJohn Snow } 5420df4ba58SJohn Snow } 5430df4ba58SJohn Snow 5448acc72a4SPaolo Bonzini void block_job_cancel(BlockJob *job) 5458acc72a4SPaolo Bonzini { 5465ccac6f1SJohn Snow if (block_job_started(job)) { 5474c241cf5SPaolo Bonzini block_job_cancel_async(job); 548751ebd76SFam Zheng block_job_enter(job); 5495ccac6f1SJohn Snow } else { 5505ccac6f1SJohn Snow block_job_completed(job, -ECANCELED); 5515ccac6f1SJohn Snow } 5528acc72a4SPaolo Bonzini } 5538acc72a4SPaolo Bonzini 554345f9e1bSMax Reitz /* A wrapper around block_job_cancel() taking an Error ** parameter so it may be 555345f9e1bSMax Reitz * used with block_job_finish_sync() without the need for (rather nasty) 556345f9e1bSMax Reitz * function pointer casts there. */ 557345f9e1bSMax Reitz static void block_job_cancel_err(BlockJob *job, Error **errp) 558345f9e1bSMax Reitz { 559345f9e1bSMax Reitz block_job_cancel(job); 560345f9e1bSMax Reitz } 561345f9e1bSMax Reitz 562345f9e1bSMax Reitz int block_job_cancel_sync(BlockJob *job) 563345f9e1bSMax Reitz { 564345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_cancel_err, NULL); 565345f9e1bSMax Reitz } 566345f9e1bSMax Reitz 567a1a2af07SKevin Wolf void block_job_cancel_sync_all(void) 568a1a2af07SKevin Wolf { 569a1a2af07SKevin Wolf BlockJob *job; 570a1a2af07SKevin Wolf AioContext *aio_context; 571a1a2af07SKevin Wolf 572a1a2af07SKevin Wolf while ((job = QLIST_FIRST(&block_jobs))) { 573b6d2e599SKevin Wolf aio_context = blk_get_aio_context(job->blk); 574a1a2af07SKevin Wolf aio_context_acquire(aio_context); 575a1a2af07SKevin Wolf block_job_cancel_sync(job); 576a1a2af07SKevin Wolf aio_context_release(aio_context); 577a1a2af07SKevin Wolf } 578a1a2af07SKevin Wolf } 579a1a2af07SKevin Wolf 580345f9e1bSMax Reitz int block_job_complete_sync(BlockJob *job, Error **errp) 581345f9e1bSMax Reitz { 582345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_complete, errp); 583345f9e1bSMax Reitz } 584345f9e1bSMax Reitz 585559b935fSJohn Snow BlockJobInfo *block_job_query(BlockJob *job, Error **errp) 58630e628b7SPaolo Bonzini { 587559b935fSJohn Snow BlockJobInfo *info; 588559b935fSJohn Snow 589559b935fSJohn Snow if (block_job_is_internal(job)) { 590559b935fSJohn Snow error_setg(errp, "Cannot query QEMU internal jobs"); 591559b935fSJohn Snow return NULL; 592559b935fSJohn Snow } 593559b935fSJohn Snow info = g_new0(BlockJobInfo, 1); 594977c736fSMarkus Armbruster info->type = g_strdup(BlockJobType_str(job->driver->job_type)); 5958ccb9569SKevin Wolf info->device = g_strdup(job->id); 59630e628b7SPaolo Bonzini info->len = job->len; 597fc24908eSPaolo Bonzini info->busy = atomic_read(&job->busy); 598751ebd76SFam Zheng info->paused = job->pause_count > 0; 59930e628b7SPaolo Bonzini info->offset = job->offset; 60030e628b7SPaolo Bonzini info->speed = job->speed; 60132c81a4aSPaolo Bonzini info->io_status = job->iostatus; 602ef6dbf1eSMax Reitz info->ready = job->ready; 60330e628b7SPaolo Bonzini return info; 60430e628b7SPaolo Bonzini } 60532c81a4aSPaolo Bonzini 60632c81a4aSPaolo Bonzini static void block_job_iostatus_set_err(BlockJob *job, int error) 60732c81a4aSPaolo Bonzini { 60832c81a4aSPaolo Bonzini if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 60932c81a4aSPaolo Bonzini job->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE : 61032c81a4aSPaolo Bonzini BLOCK_DEVICE_IO_STATUS_FAILED; 61132c81a4aSPaolo Bonzini } 61232c81a4aSPaolo Bonzini } 61332c81a4aSPaolo Bonzini 6148254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job) 615a66a2a36SPaolo Bonzini { 616559b935fSJohn Snow if (block_job_is_internal(job)) { 617559b935fSJohn Snow return; 618559b935fSJohn Snow } 619559b935fSJohn Snow 620bcada37bSWenchao Xia qapi_event_send_block_job_cancelled(job->driver->job_type, 6218ccb9569SKevin Wolf job->id, 622a66a2a36SPaolo Bonzini job->len, 623a66a2a36SPaolo Bonzini job->offset, 624bcada37bSWenchao Xia job->speed, 625bcada37bSWenchao Xia &error_abort); 626a66a2a36SPaolo Bonzini } 627a66a2a36SPaolo Bonzini 6288254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg) 629a66a2a36SPaolo Bonzini { 630559b935fSJohn Snow if (block_job_is_internal(job)) { 631559b935fSJohn Snow return; 632559b935fSJohn Snow } 633559b935fSJohn Snow 634bcada37bSWenchao Xia qapi_event_send_block_job_completed(job->driver->job_type, 6358ccb9569SKevin Wolf job->id, 636bcada37bSWenchao Xia job->len, 637bcada37bSWenchao Xia job->offset, 638bcada37bSWenchao Xia job->speed, 639bcada37bSWenchao Xia !!msg, 640bcada37bSWenchao Xia msg, 641bcada37bSWenchao Xia &error_abort); 642bcada37bSWenchao Xia } 643bcada37bSWenchao Xia 64488691b37SPaolo Bonzini /* 64588691b37SPaolo Bonzini * API for block job drivers and the block layer. These functions are 64688691b37SPaolo Bonzini * declared in blockjob_int.h. 64788691b37SPaolo Bonzini */ 64888691b37SPaolo Bonzini 64988691b37SPaolo Bonzini void *block_job_create(const char *job_id, const BlockJobDriver *driver, 65088691b37SPaolo Bonzini BlockDriverState *bs, uint64_t perm, 65188691b37SPaolo Bonzini uint64_t shared_perm, int64_t speed, int flags, 65288691b37SPaolo Bonzini BlockCompletionFunc *cb, void *opaque, Error **errp) 65388691b37SPaolo Bonzini { 65488691b37SPaolo Bonzini BlockBackend *blk; 65588691b37SPaolo Bonzini BlockJob *job; 65688691b37SPaolo Bonzini int ret; 65788691b37SPaolo Bonzini 65888691b37SPaolo Bonzini if (bs->job) { 65988691b37SPaolo Bonzini error_setg(errp, QERR_DEVICE_IN_USE, bdrv_get_device_name(bs)); 66088691b37SPaolo Bonzini return NULL; 66188691b37SPaolo Bonzini } 66288691b37SPaolo Bonzini 66388691b37SPaolo Bonzini if (job_id == NULL && !(flags & BLOCK_JOB_INTERNAL)) { 66488691b37SPaolo Bonzini job_id = bdrv_get_device_name(bs); 66588691b37SPaolo Bonzini if (!*job_id) { 66688691b37SPaolo Bonzini error_setg(errp, "An explicit job ID is required for this node"); 66788691b37SPaolo Bonzini return NULL; 66888691b37SPaolo Bonzini } 66988691b37SPaolo Bonzini } 67088691b37SPaolo Bonzini 67188691b37SPaolo Bonzini if (job_id) { 67288691b37SPaolo Bonzini if (flags & BLOCK_JOB_INTERNAL) { 67388691b37SPaolo Bonzini error_setg(errp, "Cannot specify job ID for internal block job"); 67488691b37SPaolo Bonzini return NULL; 67588691b37SPaolo Bonzini } 67688691b37SPaolo Bonzini 67788691b37SPaolo Bonzini if (!id_wellformed(job_id)) { 67888691b37SPaolo Bonzini error_setg(errp, "Invalid job ID '%s'", job_id); 67988691b37SPaolo Bonzini return NULL; 68088691b37SPaolo Bonzini } 68188691b37SPaolo Bonzini 68288691b37SPaolo Bonzini if (block_job_get(job_id)) { 68388691b37SPaolo Bonzini error_setg(errp, "Job ID '%s' already in use", job_id); 68488691b37SPaolo Bonzini return NULL; 68588691b37SPaolo Bonzini } 68688691b37SPaolo Bonzini } 68788691b37SPaolo Bonzini 68888691b37SPaolo Bonzini blk = blk_new(perm, shared_perm); 68988691b37SPaolo Bonzini ret = blk_insert_bs(blk, bs, errp); 69088691b37SPaolo Bonzini if (ret < 0) { 69188691b37SPaolo Bonzini blk_unref(blk); 69288691b37SPaolo Bonzini return NULL; 69388691b37SPaolo Bonzini } 69488691b37SPaolo Bonzini 69588691b37SPaolo Bonzini job = g_malloc0(driver->instance_size); 69688691b37SPaolo Bonzini job->driver = driver; 69788691b37SPaolo Bonzini job->id = g_strdup(job_id); 69888691b37SPaolo Bonzini job->blk = blk; 69988691b37SPaolo Bonzini job->cb = cb; 70088691b37SPaolo Bonzini job->opaque = opaque; 70188691b37SPaolo Bonzini job->busy = false; 70288691b37SPaolo Bonzini job->paused = true; 70388691b37SPaolo Bonzini job->pause_count = 1; 70488691b37SPaolo Bonzini job->refcnt = 1; 705fc24908eSPaolo Bonzini aio_timer_init(qemu_get_aio_context(), &job->sleep_timer, 706fc24908eSPaolo Bonzini QEMU_CLOCK_REALTIME, SCALE_NS, 707fc24908eSPaolo Bonzini block_job_sleep_timer_cb, job); 70888691b37SPaolo Bonzini 70988691b37SPaolo Bonzini error_setg(&job->blocker, "block device is in use by block job: %s", 710977c736fSMarkus Armbruster BlockJobType_str(driver->job_type)); 71188691b37SPaolo Bonzini block_job_add_bdrv(job, "main node", bs, 0, BLK_PERM_ALL, &error_abort); 71288691b37SPaolo Bonzini bs->job = job; 71388691b37SPaolo Bonzini 71488691b37SPaolo Bonzini bdrv_op_unblock(bs, BLOCK_OP_TYPE_DATAPLANE, job->blocker); 71588691b37SPaolo Bonzini 71688691b37SPaolo Bonzini QLIST_INSERT_HEAD(&block_jobs, job, job_list); 71788691b37SPaolo Bonzini 71888691b37SPaolo Bonzini blk_add_aio_context_notifier(blk, block_job_attached_aio_context, 71988691b37SPaolo Bonzini block_job_detach_aio_context, job); 72088691b37SPaolo Bonzini 72188691b37SPaolo Bonzini /* Only set speed when necessary to avoid NotSupported error */ 72288691b37SPaolo Bonzini if (speed != 0) { 72388691b37SPaolo Bonzini Error *local_err = NULL; 72488691b37SPaolo Bonzini 72588691b37SPaolo Bonzini block_job_set_speed(job, speed, &local_err); 72688691b37SPaolo Bonzini if (local_err) { 72788691b37SPaolo Bonzini block_job_unref(job); 72888691b37SPaolo Bonzini error_propagate(errp, local_err); 72988691b37SPaolo Bonzini return NULL; 73088691b37SPaolo Bonzini } 73188691b37SPaolo Bonzini } 73288691b37SPaolo Bonzini return job; 73388691b37SPaolo Bonzini } 73488691b37SPaolo Bonzini 735f321dcb5SPaolo Bonzini void block_job_pause_all(void) 736f321dcb5SPaolo Bonzini { 737f321dcb5SPaolo Bonzini BlockJob *job = NULL; 738f321dcb5SPaolo Bonzini while ((job = block_job_next(job))) { 739f321dcb5SPaolo Bonzini AioContext *aio_context = blk_get_aio_context(job->blk); 740f321dcb5SPaolo Bonzini 741f321dcb5SPaolo Bonzini aio_context_acquire(aio_context); 7423d5d319eSAlberto Garcia block_job_ref(job); 743f321dcb5SPaolo Bonzini block_job_pause(job); 744f321dcb5SPaolo Bonzini aio_context_release(aio_context); 745f321dcb5SPaolo Bonzini } 746f321dcb5SPaolo Bonzini } 747f321dcb5SPaolo Bonzini 74888691b37SPaolo Bonzini void block_job_early_fail(BlockJob *job) 74988691b37SPaolo Bonzini { 75088691b37SPaolo Bonzini block_job_unref(job); 75188691b37SPaolo Bonzini } 75288691b37SPaolo Bonzini 75388691b37SPaolo Bonzini void block_job_completed(BlockJob *job, int ret) 75488691b37SPaolo Bonzini { 75588691b37SPaolo Bonzini assert(blk_bs(job->blk)->job == job); 75688691b37SPaolo Bonzini assert(!job->completed); 75788691b37SPaolo Bonzini job->completed = true; 75888691b37SPaolo Bonzini job->ret = ret; 75988691b37SPaolo Bonzini if (!job->txn) { 76088691b37SPaolo Bonzini block_job_completed_single(job); 76188691b37SPaolo Bonzini } else if (ret < 0 || block_job_is_cancelled(job)) { 76288691b37SPaolo Bonzini block_job_completed_txn_abort(job); 76388691b37SPaolo Bonzini } else { 76488691b37SPaolo Bonzini block_job_completed_txn_success(job); 76588691b37SPaolo Bonzini } 76688691b37SPaolo Bonzini } 76788691b37SPaolo Bonzini 76888691b37SPaolo Bonzini static bool block_job_should_pause(BlockJob *job) 76988691b37SPaolo Bonzini { 77088691b37SPaolo Bonzini return job->pause_count > 0; 77188691b37SPaolo Bonzini } 77288691b37SPaolo Bonzini 773fc24908eSPaolo Bonzini /* Yield, and schedule a timer to reenter the coroutine after @ns nanoseconds. 774fc24908eSPaolo Bonzini * Reentering the job coroutine with block_job_enter() before the timer has 775fc24908eSPaolo Bonzini * expired is allowed and cancels the timer. 776fc24908eSPaolo Bonzini * 777fc24908eSPaolo Bonzini * If @ns is (uint64_t) -1, no timer is scheduled and block_job_enter() must be 778fc24908eSPaolo Bonzini * called explicitly. */ 779fc24908eSPaolo Bonzini static void block_job_do_yield(BlockJob *job, uint64_t ns) 780356f59b8SPaolo Bonzini { 781fc24908eSPaolo Bonzini block_job_lock(); 782fc24908eSPaolo Bonzini if (ns != -1) { 783fc24908eSPaolo Bonzini timer_mod(&job->sleep_timer, ns); 784fc24908eSPaolo Bonzini } 785356f59b8SPaolo Bonzini job->busy = false; 786fc24908eSPaolo Bonzini block_job_unlock(); 787356f59b8SPaolo Bonzini qemu_coroutine_yield(); 788356f59b8SPaolo Bonzini 789356f59b8SPaolo Bonzini /* Set by block_job_enter before re-entering the coroutine. */ 790356f59b8SPaolo Bonzini assert(job->busy); 791356f59b8SPaolo Bonzini } 792356f59b8SPaolo Bonzini 79388691b37SPaolo Bonzini void coroutine_fn block_job_pause_point(BlockJob *job) 79488691b37SPaolo Bonzini { 79588691b37SPaolo Bonzini assert(job && block_job_started(job)); 79688691b37SPaolo Bonzini 79788691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 79888691b37SPaolo Bonzini return; 79988691b37SPaolo Bonzini } 80088691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 80188691b37SPaolo Bonzini return; 80288691b37SPaolo Bonzini } 80388691b37SPaolo Bonzini 80488691b37SPaolo Bonzini if (job->driver->pause) { 80588691b37SPaolo Bonzini job->driver->pause(job); 80688691b37SPaolo Bonzini } 80788691b37SPaolo Bonzini 80888691b37SPaolo Bonzini if (block_job_should_pause(job) && !block_job_is_cancelled(job)) { 80988691b37SPaolo Bonzini job->paused = true; 810fc24908eSPaolo Bonzini block_job_do_yield(job, -1); 81188691b37SPaolo Bonzini job->paused = false; 81288691b37SPaolo Bonzini } 81388691b37SPaolo Bonzini 81488691b37SPaolo Bonzini if (job->driver->resume) { 81588691b37SPaolo Bonzini job->driver->resume(job); 81688691b37SPaolo Bonzini } 81788691b37SPaolo Bonzini } 81888691b37SPaolo Bonzini 819f321dcb5SPaolo Bonzini void block_job_resume_all(void) 820f321dcb5SPaolo Bonzini { 8213d5d319eSAlberto Garcia BlockJob *job, *next; 8223d5d319eSAlberto Garcia 8233d5d319eSAlberto Garcia QLIST_FOREACH_SAFE(job, &block_jobs, job_list, next) { 824f321dcb5SPaolo Bonzini AioContext *aio_context = blk_get_aio_context(job->blk); 825f321dcb5SPaolo Bonzini 826f321dcb5SPaolo Bonzini aio_context_acquire(aio_context); 827f321dcb5SPaolo Bonzini block_job_resume(job); 8283d5d319eSAlberto Garcia block_job_unref(job); 829f321dcb5SPaolo Bonzini aio_context_release(aio_context); 830f321dcb5SPaolo Bonzini } 831f321dcb5SPaolo Bonzini } 832f321dcb5SPaolo Bonzini 833aa9ef2e6SJohn Snow /* 834aa9ef2e6SJohn Snow * Conditionally enter a block_job pending a call to fn() while 835aa9ef2e6SJohn Snow * under the block_job_lock critical section. 836aa9ef2e6SJohn Snow */ 837aa9ef2e6SJohn Snow static void block_job_enter_cond(BlockJob *job, bool(*fn)(BlockJob *job)) 83888691b37SPaolo Bonzini { 839eb05e011SPaolo Bonzini if (!block_job_started(job)) { 840eb05e011SPaolo Bonzini return; 841eb05e011SPaolo Bonzini } 842eb05e011SPaolo Bonzini if (job->deferred_to_main_loop) { 843eb05e011SPaolo Bonzini return; 844eb05e011SPaolo Bonzini } 845eb05e011SPaolo Bonzini 846fc24908eSPaolo Bonzini block_job_lock(); 847356f59b8SPaolo Bonzini if (job->busy) { 848fc24908eSPaolo Bonzini block_job_unlock(); 849356f59b8SPaolo Bonzini return; 85088691b37SPaolo Bonzini } 851356f59b8SPaolo Bonzini 852aa9ef2e6SJohn Snow if (fn && !fn(job)) { 853aa9ef2e6SJohn Snow block_job_unlock(); 854aa9ef2e6SJohn Snow return; 855aa9ef2e6SJohn Snow } 856aa9ef2e6SJohn Snow 857fc24908eSPaolo Bonzini assert(!job->deferred_to_main_loop); 858fc24908eSPaolo Bonzini timer_del(&job->sleep_timer); 859356f59b8SPaolo Bonzini job->busy = true; 860fc24908eSPaolo Bonzini block_job_unlock(); 861356f59b8SPaolo Bonzini aio_co_wake(job->co); 86288691b37SPaolo Bonzini } 86388691b37SPaolo Bonzini 864aa9ef2e6SJohn Snow void block_job_enter(BlockJob *job) 865aa9ef2e6SJohn Snow { 866aa9ef2e6SJohn Snow block_job_enter_cond(job, NULL); 867aa9ef2e6SJohn Snow } 868aa9ef2e6SJohn Snow 86988691b37SPaolo Bonzini bool block_job_is_cancelled(BlockJob *job) 87088691b37SPaolo Bonzini { 87188691b37SPaolo Bonzini return job->cancelled; 87288691b37SPaolo Bonzini } 87388691b37SPaolo Bonzini 8745bf1d5a7SPaolo Bonzini void block_job_sleep_ns(BlockJob *job, int64_t ns) 87588691b37SPaolo Bonzini { 87688691b37SPaolo Bonzini assert(job->busy); 87788691b37SPaolo Bonzini 87888691b37SPaolo Bonzini /* Check cancellation *before* setting busy = false, too! */ 87988691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 88088691b37SPaolo Bonzini return; 88188691b37SPaolo Bonzini } 88288691b37SPaolo Bonzini 88388691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 884fc24908eSPaolo Bonzini block_job_do_yield(job, qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + ns); 88588691b37SPaolo Bonzini } 88688691b37SPaolo Bonzini 88788691b37SPaolo Bonzini block_job_pause_point(job); 88888691b37SPaolo Bonzini } 88988691b37SPaolo Bonzini 89088691b37SPaolo Bonzini void block_job_yield(BlockJob *job) 89188691b37SPaolo Bonzini { 89288691b37SPaolo Bonzini assert(job->busy); 89388691b37SPaolo Bonzini 89488691b37SPaolo Bonzini /* Check cancellation *before* setting busy = false, too! */ 89588691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 89688691b37SPaolo Bonzini return; 89788691b37SPaolo Bonzini } 89888691b37SPaolo Bonzini 89988691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 900fc24908eSPaolo Bonzini block_job_do_yield(job, -1); 90188691b37SPaolo Bonzini } 90288691b37SPaolo Bonzini 90388691b37SPaolo Bonzini block_job_pause_point(job); 90488691b37SPaolo Bonzini } 90588691b37SPaolo Bonzini 9062caf63a9SPaolo Bonzini void block_job_iostatus_reset(BlockJob *job) 9072caf63a9SPaolo Bonzini { 9084c241cf5SPaolo Bonzini if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 9094c241cf5SPaolo Bonzini return; 9104c241cf5SPaolo Bonzini } 9114c241cf5SPaolo Bonzini assert(job->user_paused && job->pause_count > 0); 9122caf63a9SPaolo Bonzini job->iostatus = BLOCK_DEVICE_IO_STATUS_OK; 9132caf63a9SPaolo Bonzini } 9142caf63a9SPaolo Bonzini 915bcada37bSWenchao Xia void block_job_event_ready(BlockJob *job) 916bcada37bSWenchao Xia { 917ef6dbf1eSMax Reitz job->ready = true; 918ef6dbf1eSMax Reitz 919559b935fSJohn Snow if (block_job_is_internal(job)) { 920559b935fSJohn Snow return; 921559b935fSJohn Snow } 922559b935fSJohn Snow 923518848a2SMarkus Armbruster qapi_event_send_block_job_ready(job->driver->job_type, 9248ccb9569SKevin Wolf job->id, 925518848a2SMarkus Armbruster job->len, 926518848a2SMarkus Armbruster job->offset, 927518848a2SMarkus Armbruster job->speed, &error_abort); 928a66a2a36SPaolo Bonzini } 929a66a2a36SPaolo Bonzini 93081e254dcSKevin Wolf BlockErrorAction block_job_error_action(BlockJob *job, BlockdevOnError on_err, 93132c81a4aSPaolo Bonzini int is_read, int error) 93232c81a4aSPaolo Bonzini { 93332c81a4aSPaolo Bonzini BlockErrorAction action; 93432c81a4aSPaolo Bonzini 93532c81a4aSPaolo Bonzini switch (on_err) { 93632c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_ENOSPC: 9378c398252SKevin Wolf case BLOCKDEV_ON_ERROR_AUTO: 938a589569fSWenchao Xia action = (error == ENOSPC) ? 939a589569fSWenchao Xia BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT; 94032c81a4aSPaolo Bonzini break; 94132c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_STOP: 942a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_STOP; 94332c81a4aSPaolo Bonzini break; 94432c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_REPORT: 945a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_REPORT; 94632c81a4aSPaolo Bonzini break; 94732c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_IGNORE: 948a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_IGNORE; 94932c81a4aSPaolo Bonzini break; 95032c81a4aSPaolo Bonzini default: 95132c81a4aSPaolo Bonzini abort(); 95232c81a4aSPaolo Bonzini } 953559b935fSJohn Snow if (!block_job_is_internal(job)) { 9548ccb9569SKevin Wolf qapi_event_send_block_job_error(job->id, 9555a2d2cbdSWenchao Xia is_read ? IO_OPERATION_TYPE_READ : 9565a2d2cbdSWenchao Xia IO_OPERATION_TYPE_WRITE, 9575a2d2cbdSWenchao Xia action, &error_abort); 958559b935fSJohn Snow } 959a589569fSWenchao Xia if (action == BLOCK_ERROR_ACTION_STOP) { 960751ebd76SFam Zheng /* make the pause user visible, which will be resumed from QMP. */ 9610df4ba58SJohn Snow block_job_user_pause(job); 96232c81a4aSPaolo Bonzini block_job_iostatus_set_err(job, error); 96332c81a4aSPaolo Bonzini } 96432c81a4aSPaolo Bonzini return action; 96532c81a4aSPaolo Bonzini } 966dec7d421SStefan Hajnoczi 967dec7d421SStefan Hajnoczi typedef struct { 968dec7d421SStefan Hajnoczi BlockJob *job; 969dec7d421SStefan Hajnoczi AioContext *aio_context; 970dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn; 971dec7d421SStefan Hajnoczi void *opaque; 972dec7d421SStefan Hajnoczi } BlockJobDeferToMainLoopData; 973dec7d421SStefan Hajnoczi 974dec7d421SStefan Hajnoczi static void block_job_defer_to_main_loop_bh(void *opaque) 975dec7d421SStefan Hajnoczi { 976dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = opaque; 977dec7d421SStefan Hajnoczi AioContext *aio_context; 978dec7d421SStefan Hajnoczi 979dec7d421SStefan Hajnoczi /* Prevent race with block_job_defer_to_main_loop() */ 980dec7d421SStefan Hajnoczi aio_context_acquire(data->aio_context); 981dec7d421SStefan Hajnoczi 982dec7d421SStefan Hajnoczi /* Fetch BDS AioContext again, in case it has changed */ 983b6d2e599SKevin Wolf aio_context = blk_get_aio_context(data->job->blk); 984d79df2a2SPaolo Bonzini if (aio_context != data->aio_context) { 985dec7d421SStefan Hajnoczi aio_context_acquire(aio_context); 986d79df2a2SPaolo Bonzini } 987dec7d421SStefan Hajnoczi 988dec7d421SStefan Hajnoczi data->fn(data->job, data->opaque); 989dec7d421SStefan Hajnoczi 990d79df2a2SPaolo Bonzini if (aio_context != data->aio_context) { 991dec7d421SStefan Hajnoczi aio_context_release(aio_context); 992d79df2a2SPaolo Bonzini } 993dec7d421SStefan Hajnoczi 994dec7d421SStefan Hajnoczi aio_context_release(data->aio_context); 995dec7d421SStefan Hajnoczi 996dec7d421SStefan Hajnoczi g_free(data); 997dec7d421SStefan Hajnoczi } 998dec7d421SStefan Hajnoczi 999dec7d421SStefan Hajnoczi void block_job_defer_to_main_loop(BlockJob *job, 1000dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn, 1001dec7d421SStefan Hajnoczi void *opaque) 1002dec7d421SStefan Hajnoczi { 1003dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = g_malloc(sizeof(*data)); 1004dec7d421SStefan Hajnoczi data->job = job; 1005b6d2e599SKevin Wolf data->aio_context = blk_get_aio_context(job->blk); 1006dec7d421SStefan Hajnoczi data->fn = fn; 1007dec7d421SStefan Hajnoczi data->opaque = opaque; 1008794f0141SFam Zheng job->deferred_to_main_loop = true; 1009dec7d421SStefan Hajnoczi 1010fffb6e12SPaolo Bonzini aio_bh_schedule_oneshot(qemu_get_aio_context(), 1011fffb6e12SPaolo Bonzini block_job_defer_to_main_loop_bh, data); 1012dec7d421SStefan Hajnoczi } 1013