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 40fc24908eSPaolo Bonzini /* Right now, this mutex is only needed to synchronize accesses to job->busy 41fc24908eSPaolo Bonzini * and job->sleep_timer, such as concurrent calls to block_job_do_yield and 42fc24908eSPaolo Bonzini * block_job_enter. */ 43fc24908eSPaolo Bonzini static QemuMutex block_job_mutex; 44fc24908eSPaolo Bonzini 45fc24908eSPaolo Bonzini static void block_job_lock(void) 46fc24908eSPaolo Bonzini { 47fc24908eSPaolo Bonzini qemu_mutex_lock(&block_job_mutex); 48fc24908eSPaolo Bonzini } 49fc24908eSPaolo Bonzini 50fc24908eSPaolo Bonzini static void block_job_unlock(void) 51fc24908eSPaolo Bonzini { 52fc24908eSPaolo Bonzini qemu_mutex_unlock(&block_job_mutex); 53fc24908eSPaolo Bonzini } 54fc24908eSPaolo Bonzini 55fc24908eSPaolo Bonzini static void __attribute__((__constructor__)) block_job_init(void) 56fc24908eSPaolo Bonzini { 57fc24908eSPaolo Bonzini qemu_mutex_init(&block_job_mutex); 58fc24908eSPaolo Bonzini } 59fc24908eSPaolo Bonzini 608254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job); 618254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg); 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 77a7112795SAlberto Garcia static QLIST_HEAD(, BlockJob) block_jobs = QLIST_HEAD_INITIALIZER(block_jobs); 78a7112795SAlberto Garcia 7988691b37SPaolo Bonzini /* 8088691b37SPaolo Bonzini * The block job API is composed of two categories of functions. 8188691b37SPaolo Bonzini * 8288691b37SPaolo Bonzini * The first includes functions used by the monitor. The monitor is 8388691b37SPaolo Bonzini * peculiar in that it accesses the block job list with block_job_get, and 8488691b37SPaolo Bonzini * therefore needs consistency across block_job_get and the actual operation 8588691b37SPaolo Bonzini * (e.g. block_job_set_speed). The consistency is achieved with 8688691b37SPaolo Bonzini * aio_context_acquire/release. These functions are declared in blockjob.h. 8788691b37SPaolo Bonzini * 8888691b37SPaolo Bonzini * The second includes functions used by the block job drivers and sometimes 8988691b37SPaolo Bonzini * by the core block layer. These do not care about locking, because the 9088691b37SPaolo Bonzini * whole coroutine runs under the AioContext lock, and are declared in 9188691b37SPaolo Bonzini * blockjob_int.h. 9288691b37SPaolo Bonzini */ 9388691b37SPaolo Bonzini 94a7112795SAlberto Garcia BlockJob *block_job_next(BlockJob *job) 95a7112795SAlberto Garcia { 96a7112795SAlberto Garcia if (!job) { 97a7112795SAlberto Garcia return QLIST_FIRST(&block_jobs); 98a7112795SAlberto Garcia } 99a7112795SAlberto Garcia return QLIST_NEXT(job, job_list); 100a7112795SAlberto Garcia } 101a7112795SAlberto Garcia 102ffb1f10cSAlberto Garcia BlockJob *block_job_get(const char *id) 103ffb1f10cSAlberto Garcia { 104ffb1f10cSAlberto Garcia BlockJob *job; 105ffb1f10cSAlberto Garcia 106ffb1f10cSAlberto Garcia QLIST_FOREACH(job, &block_jobs, job_list) { 107559b935fSJohn Snow if (job->id && !strcmp(id, job->id)) { 108ffb1f10cSAlberto Garcia return job; 109ffb1f10cSAlberto Garcia } 110ffb1f10cSAlberto Garcia } 111ffb1f10cSAlberto Garcia 112ffb1f10cSAlberto Garcia return NULL; 113ffb1f10cSAlberto Garcia } 114ffb1f10cSAlberto Garcia 115c8ab5c2dSPaolo Bonzini BlockJobTxn *block_job_txn_new(void) 116c8ab5c2dSPaolo Bonzini { 117c8ab5c2dSPaolo Bonzini BlockJobTxn *txn = g_new0(BlockJobTxn, 1); 118c8ab5c2dSPaolo Bonzini QLIST_INIT(&txn->jobs); 119c8ab5c2dSPaolo Bonzini txn->refcnt = 1; 120c8ab5c2dSPaolo Bonzini return txn; 121c8ab5c2dSPaolo Bonzini } 122c8ab5c2dSPaolo Bonzini 123c8ab5c2dSPaolo Bonzini static void block_job_txn_ref(BlockJobTxn *txn) 124c8ab5c2dSPaolo Bonzini { 125c8ab5c2dSPaolo Bonzini txn->refcnt++; 126c8ab5c2dSPaolo Bonzini } 127c8ab5c2dSPaolo Bonzini 128c8ab5c2dSPaolo Bonzini void block_job_txn_unref(BlockJobTxn *txn) 129c8ab5c2dSPaolo Bonzini { 130c8ab5c2dSPaolo Bonzini if (txn && --txn->refcnt == 0) { 131c8ab5c2dSPaolo Bonzini g_free(txn); 132c8ab5c2dSPaolo Bonzini } 133c8ab5c2dSPaolo Bonzini } 134c8ab5c2dSPaolo Bonzini 135c8ab5c2dSPaolo Bonzini void block_job_txn_add_job(BlockJobTxn *txn, BlockJob *job) 136c8ab5c2dSPaolo Bonzini { 137c8ab5c2dSPaolo Bonzini if (!txn) { 138c8ab5c2dSPaolo Bonzini return; 139c8ab5c2dSPaolo Bonzini } 140c8ab5c2dSPaolo Bonzini 141c8ab5c2dSPaolo Bonzini assert(!job->txn); 142c8ab5c2dSPaolo Bonzini job->txn = txn; 143c8ab5c2dSPaolo Bonzini 144c8ab5c2dSPaolo Bonzini QLIST_INSERT_HEAD(&txn->jobs, job, txn_list); 145c8ab5c2dSPaolo Bonzini block_job_txn_ref(txn); 146c8ab5c2dSPaolo Bonzini } 147c8ab5c2dSPaolo Bonzini 148f321dcb5SPaolo Bonzini static void block_job_pause(BlockJob *job) 149f321dcb5SPaolo Bonzini { 150f321dcb5SPaolo Bonzini job->pause_count++; 151f321dcb5SPaolo Bonzini } 152f321dcb5SPaolo Bonzini 153f321dcb5SPaolo Bonzini static void block_job_resume(BlockJob *job) 154f321dcb5SPaolo Bonzini { 155f321dcb5SPaolo Bonzini assert(job->pause_count > 0); 156f321dcb5SPaolo Bonzini job->pause_count--; 157f321dcb5SPaolo Bonzini if (job->pause_count) { 158f321dcb5SPaolo Bonzini return; 159f321dcb5SPaolo Bonzini } 160f321dcb5SPaolo Bonzini block_job_enter(job); 161f321dcb5SPaolo Bonzini } 162f321dcb5SPaolo Bonzini 1634172a003Ssochin.jiang void block_job_ref(BlockJob *job) 16405b0d8e3SPaolo Bonzini { 16505b0d8e3SPaolo Bonzini ++job->refcnt; 16605b0d8e3SPaolo Bonzini } 16705b0d8e3SPaolo Bonzini 16805b0d8e3SPaolo Bonzini static void block_job_attached_aio_context(AioContext *new_context, 16905b0d8e3SPaolo Bonzini void *opaque); 17005b0d8e3SPaolo Bonzini static void block_job_detach_aio_context(void *opaque); 17105b0d8e3SPaolo Bonzini 1724172a003Ssochin.jiang void block_job_unref(BlockJob *job) 17305b0d8e3SPaolo Bonzini { 17405b0d8e3SPaolo Bonzini if (--job->refcnt == 0) { 17505b0d8e3SPaolo Bonzini BlockDriverState *bs = blk_bs(job->blk); 1760a3e155fSAlberto Garcia QLIST_REMOVE(job, job_list); 17705b0d8e3SPaolo Bonzini bs->job = NULL; 17805b0d8e3SPaolo Bonzini block_job_remove_all_bdrv(job); 17905b0d8e3SPaolo Bonzini blk_remove_aio_context_notifier(job->blk, 18005b0d8e3SPaolo Bonzini block_job_attached_aio_context, 18105b0d8e3SPaolo Bonzini block_job_detach_aio_context, job); 18205b0d8e3SPaolo Bonzini blk_unref(job->blk); 18305b0d8e3SPaolo Bonzini error_free(job->blocker); 18405b0d8e3SPaolo Bonzini g_free(job->id); 185fc24908eSPaolo Bonzini assert(!timer_pending(&job->sleep_timer)); 18605b0d8e3SPaolo Bonzini g_free(job); 18705b0d8e3SPaolo Bonzini } 18805b0d8e3SPaolo Bonzini } 18905b0d8e3SPaolo Bonzini 190463e0be1SStefan Hajnoczi static void block_job_attached_aio_context(AioContext *new_context, 191463e0be1SStefan Hajnoczi void *opaque) 192463e0be1SStefan Hajnoczi { 193463e0be1SStefan Hajnoczi BlockJob *job = opaque; 194463e0be1SStefan Hajnoczi 195463e0be1SStefan Hajnoczi if (job->driver->attached_aio_context) { 196463e0be1SStefan Hajnoczi job->driver->attached_aio_context(job, new_context); 197463e0be1SStefan Hajnoczi } 198463e0be1SStefan Hajnoczi 199463e0be1SStefan Hajnoczi block_job_resume(job); 200463e0be1SStefan Hajnoczi } 201463e0be1SStefan Hajnoczi 202bae8196dSPaolo Bonzini static void block_job_drain(BlockJob *job) 203bae8196dSPaolo Bonzini { 204bae8196dSPaolo Bonzini /* If job is !job->busy this kicks it into the next pause point. */ 205bae8196dSPaolo Bonzini block_job_enter(job); 206bae8196dSPaolo Bonzini 207bae8196dSPaolo Bonzini blk_drain(job->blk); 208bae8196dSPaolo Bonzini if (job->driver->drain) { 209bae8196dSPaolo Bonzini job->driver->drain(job); 210bae8196dSPaolo Bonzini } 211bae8196dSPaolo Bonzini } 212bae8196dSPaolo Bonzini 213463e0be1SStefan Hajnoczi static void block_job_detach_aio_context(void *opaque) 214463e0be1SStefan Hajnoczi { 215463e0be1SStefan Hajnoczi BlockJob *job = opaque; 216463e0be1SStefan Hajnoczi 217463e0be1SStefan Hajnoczi /* In case the job terminates during aio_poll()... */ 218463e0be1SStefan Hajnoczi block_job_ref(job); 219463e0be1SStefan Hajnoczi 220463e0be1SStefan Hajnoczi block_job_pause(job); 221463e0be1SStefan Hajnoczi 222463e0be1SStefan Hajnoczi while (!job->paused && !job->completed) { 223bae8196dSPaolo Bonzini block_job_drain(job); 224463e0be1SStefan Hajnoczi } 225463e0be1SStefan Hajnoczi 226463e0be1SStefan Hajnoczi block_job_unref(job); 227463e0be1SStefan Hajnoczi } 228463e0be1SStefan Hajnoczi 229f321dcb5SPaolo Bonzini static char *child_job_get_parent_desc(BdrvChild *c) 230f321dcb5SPaolo Bonzini { 231f321dcb5SPaolo Bonzini BlockJob *job = c->opaque; 232f321dcb5SPaolo Bonzini return g_strdup_printf("%s job '%s'", 233977c736fSMarkus Armbruster BlockJobType_str(job->driver->job_type), 234f321dcb5SPaolo Bonzini job->id); 235f321dcb5SPaolo Bonzini } 236f321dcb5SPaolo Bonzini 237*ad90febaSKevin Wolf static void child_job_drained_begin(BdrvChild *c) 238f321dcb5SPaolo Bonzini { 239*ad90febaSKevin Wolf BlockJob *job = c->opaque; 240f321dcb5SPaolo Bonzini block_job_pause(job); 241f321dcb5SPaolo Bonzini } 242f321dcb5SPaolo Bonzini 243*ad90febaSKevin Wolf static void child_job_drained_end(BdrvChild *c) 244f321dcb5SPaolo Bonzini { 245*ad90febaSKevin Wolf BlockJob *job = c->opaque; 246f321dcb5SPaolo Bonzini block_job_resume(job); 247f321dcb5SPaolo Bonzini } 248f321dcb5SPaolo Bonzini 249*ad90febaSKevin Wolf static const BdrvChildRole child_job = { 250*ad90febaSKevin Wolf .get_parent_desc = child_job_get_parent_desc, 251*ad90febaSKevin Wolf .drained_begin = child_job_drained_begin, 252*ad90febaSKevin Wolf .drained_end = child_job_drained_end, 253*ad90febaSKevin Wolf .stay_at_node = true, 254f321dcb5SPaolo Bonzini }; 255f321dcb5SPaolo Bonzini 256bbc02b90SKevin Wolf void block_job_remove_all_bdrv(BlockJob *job) 257bbc02b90SKevin Wolf { 258bbc02b90SKevin Wolf GSList *l; 259bbc02b90SKevin Wolf for (l = job->nodes; l; l = l->next) { 260bbc02b90SKevin Wolf BdrvChild *c = l->data; 261bbc02b90SKevin Wolf bdrv_op_unblock_all(c->bs, job->blocker); 262bbc02b90SKevin Wolf bdrv_root_unref_child(c); 263bbc02b90SKevin Wolf } 264bbc02b90SKevin Wolf g_slist_free(job->nodes); 265bbc02b90SKevin Wolf job->nodes = NULL; 266bbc02b90SKevin Wolf } 267bbc02b90SKevin Wolf 26876d554e2SKevin Wolf int block_job_add_bdrv(BlockJob *job, const char *name, BlockDriverState *bs, 26976d554e2SKevin Wolf uint64_t perm, uint64_t shared_perm, Error **errp) 27023d402d4SAlberto Garcia { 27176d554e2SKevin Wolf BdrvChild *c; 27276d554e2SKevin Wolf 27376d554e2SKevin Wolf c = bdrv_root_attach_child(bs, name, &child_job, perm, shared_perm, 27476d554e2SKevin Wolf job, errp); 27576d554e2SKevin Wolf if (c == NULL) { 27676d554e2SKevin Wolf return -EPERM; 27776d554e2SKevin Wolf } 27876d554e2SKevin Wolf 27976d554e2SKevin Wolf job->nodes = g_slist_prepend(job->nodes, c); 28023d402d4SAlberto Garcia bdrv_ref(bs); 28123d402d4SAlberto Garcia bdrv_op_block_all(bs, job->blocker); 28276d554e2SKevin Wolf 28376d554e2SKevin Wolf return 0; 28423d402d4SAlberto Garcia } 28523d402d4SAlberto Garcia 286559b935fSJohn Snow bool block_job_is_internal(BlockJob *job) 287559b935fSJohn Snow { 288559b935fSJohn Snow return (job->id == NULL); 289559b935fSJohn Snow } 290559b935fSJohn Snow 2915ccac6f1SJohn Snow static bool block_job_started(BlockJob *job) 2925ccac6f1SJohn Snow { 2935ccac6f1SJohn Snow return job->co; 2945ccac6f1SJohn Snow } 2955ccac6f1SJohn Snow 296e3796a24SJohn Snow /** 297e3796a24SJohn Snow * All jobs must allow a pause point before entering their job proper. This 298e3796a24SJohn Snow * ensures that jobs can be paused prior to being started, then resumed later. 299e3796a24SJohn Snow */ 300e3796a24SJohn Snow static void coroutine_fn block_job_co_entry(void *opaque) 301e3796a24SJohn Snow { 302e3796a24SJohn Snow BlockJob *job = opaque; 303e3796a24SJohn Snow 304e3796a24SJohn Snow assert(job && job->driver && job->driver->start); 305e3796a24SJohn Snow block_job_pause_point(job); 306e3796a24SJohn Snow job->driver->start(job); 307e3796a24SJohn Snow } 308e3796a24SJohn Snow 309fc24908eSPaolo Bonzini static void block_job_sleep_timer_cb(void *opaque) 310fc24908eSPaolo Bonzini { 311fc24908eSPaolo Bonzini BlockJob *job = opaque; 312fc24908eSPaolo Bonzini 313fc24908eSPaolo Bonzini block_job_enter(job); 314fc24908eSPaolo Bonzini } 315fc24908eSPaolo Bonzini 3165ccac6f1SJohn Snow void block_job_start(BlockJob *job) 3175ccac6f1SJohn Snow { 3185ccac6f1SJohn Snow assert(job && !block_job_started(job) && job->paused && 319e3796a24SJohn Snow job->driver && job->driver->start); 320e3796a24SJohn Snow job->co = qemu_coroutine_create(block_job_co_entry, job); 321e3796a24SJohn Snow job->pause_count--; 3225ccac6f1SJohn Snow job->busy = true; 323e3796a24SJohn Snow job->paused = false; 324aef4278cSFam Zheng bdrv_coroutine_enter(blk_bs(job->blk), job->co); 3255ccac6f1SJohn Snow } 3265ccac6f1SJohn Snow 327c55a832fSFam Zheng static void block_job_completed_single(BlockJob *job) 328c55a832fSFam Zheng { 3294fb588e9SPaolo Bonzini assert(job->completed); 3304fb588e9SPaolo Bonzini 331c55a832fSFam Zheng if (!job->ret) { 332c55a832fSFam Zheng if (job->driver->commit) { 333c55a832fSFam Zheng job->driver->commit(job); 334c55a832fSFam Zheng } 335c55a832fSFam Zheng } else { 336c55a832fSFam Zheng if (job->driver->abort) { 337c55a832fSFam Zheng job->driver->abort(job); 338c55a832fSFam Zheng } 339c55a832fSFam Zheng } 340e8a40bf7SJohn Snow if (job->driver->clean) { 341e8a40bf7SJohn Snow job->driver->clean(job); 342e8a40bf7SJohn Snow } 3438254b6d9SJohn Snow 3448254b6d9SJohn Snow if (job->cb) { 345c55a832fSFam Zheng job->cb(job->opaque, job->ret); 3468254b6d9SJohn Snow } 3475ccac6f1SJohn Snow 3485ccac6f1SJohn Snow /* Emit events only if we actually started */ 3495ccac6f1SJohn Snow if (block_job_started(job)) { 3508254b6d9SJohn Snow if (block_job_is_cancelled(job)) { 3518254b6d9SJohn Snow block_job_event_cancelled(job); 3528254b6d9SJohn Snow } else { 3538254b6d9SJohn Snow const char *msg = NULL; 3548254b6d9SJohn Snow if (job->ret < 0) { 3558254b6d9SJohn Snow msg = strerror(-job->ret); 3568254b6d9SJohn Snow } 3578254b6d9SJohn Snow block_job_event_completed(job, msg); 3588254b6d9SJohn Snow } 3595ccac6f1SJohn Snow } 3608254b6d9SJohn Snow 361c55a832fSFam Zheng if (job->txn) { 3621e93b9fbSVladimir Sementsov-Ogievskiy QLIST_REMOVE(job, txn_list); 363c55a832fSFam Zheng block_job_txn_unref(job->txn); 364c55a832fSFam Zheng } 365c55a832fSFam Zheng block_job_unref(job); 366c55a832fSFam Zheng } 367c55a832fSFam Zheng 3684c241cf5SPaolo Bonzini static void block_job_cancel_async(BlockJob *job) 3694c241cf5SPaolo Bonzini { 3704c241cf5SPaolo Bonzini if (job->iostatus != BLOCK_DEVICE_IO_STATUS_OK) { 3714c241cf5SPaolo Bonzini block_job_iostatus_reset(job); 3724c241cf5SPaolo Bonzini } 3734c241cf5SPaolo Bonzini if (job->user_paused) { 3744c241cf5SPaolo Bonzini /* Do not call block_job_enter here, the caller will handle it. */ 3754c241cf5SPaolo Bonzini job->user_paused = false; 3764c241cf5SPaolo Bonzini job->pause_count--; 3774c241cf5SPaolo Bonzini } 3784c241cf5SPaolo Bonzini job->cancelled = true; 3794c241cf5SPaolo Bonzini } 3804c241cf5SPaolo Bonzini 381c8ab5c2dSPaolo Bonzini static int block_job_finish_sync(BlockJob *job, 382c8ab5c2dSPaolo Bonzini void (*finish)(BlockJob *, Error **errp), 383c8ab5c2dSPaolo Bonzini Error **errp) 384c8ab5c2dSPaolo Bonzini { 385c8ab5c2dSPaolo Bonzini Error *local_err = NULL; 386c8ab5c2dSPaolo Bonzini int ret; 387c8ab5c2dSPaolo Bonzini 388c8ab5c2dSPaolo Bonzini assert(blk_bs(job->blk)->job == job); 389c8ab5c2dSPaolo Bonzini 390c8ab5c2dSPaolo Bonzini block_job_ref(job); 391c8ab5c2dSPaolo Bonzini 3924fb588e9SPaolo Bonzini if (finish) { 393c8ab5c2dSPaolo Bonzini finish(job, &local_err); 3944fb588e9SPaolo Bonzini } 395c8ab5c2dSPaolo Bonzini if (local_err) { 396c8ab5c2dSPaolo Bonzini error_propagate(errp, local_err); 397c8ab5c2dSPaolo Bonzini block_job_unref(job); 398c8ab5c2dSPaolo Bonzini return -EBUSY; 399c8ab5c2dSPaolo Bonzini } 400c8ab5c2dSPaolo Bonzini /* block_job_drain calls block_job_enter, and it should be enough to 401c8ab5c2dSPaolo Bonzini * induce progress until the job completes or moves to the main thread. 402c8ab5c2dSPaolo Bonzini */ 403c8ab5c2dSPaolo Bonzini while (!job->deferred_to_main_loop && !job->completed) { 404c8ab5c2dSPaolo Bonzini block_job_drain(job); 405c8ab5c2dSPaolo Bonzini } 406c8ab5c2dSPaolo Bonzini while (!job->completed) { 407c8ab5c2dSPaolo Bonzini aio_poll(qemu_get_aio_context(), true); 408c8ab5c2dSPaolo Bonzini } 409c8ab5c2dSPaolo Bonzini ret = (job->cancelled && job->ret == 0) ? -ECANCELED : job->ret; 410c8ab5c2dSPaolo Bonzini block_job_unref(job); 411c8ab5c2dSPaolo Bonzini return ret; 412c8ab5c2dSPaolo Bonzini } 413c8ab5c2dSPaolo Bonzini 414c55a832fSFam Zheng static void block_job_completed_txn_abort(BlockJob *job) 415c55a832fSFam Zheng { 416c55a832fSFam Zheng AioContext *ctx; 417c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 4184fb588e9SPaolo Bonzini BlockJob *other_job; 419c55a832fSFam Zheng 420c55a832fSFam Zheng if (txn->aborting) { 421c55a832fSFam Zheng /* 422c55a832fSFam Zheng * We are cancelled by another job, which will handle everything. 423c55a832fSFam Zheng */ 424c55a832fSFam Zheng return; 425c55a832fSFam Zheng } 426c55a832fSFam Zheng txn->aborting = true; 4274fb588e9SPaolo Bonzini block_job_txn_ref(txn); 4284fb588e9SPaolo Bonzini 429c55a832fSFam Zheng /* We are the first failed job. Cancel other jobs. */ 430c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 431b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 432c55a832fSFam Zheng aio_context_acquire(ctx); 433c55a832fSFam Zheng } 4344fb588e9SPaolo Bonzini 4354fb588e9SPaolo Bonzini /* Other jobs are effectively cancelled by us, set the status for 436c55a832fSFam Zheng * them; this job, however, may or may not be cancelled, depending 437c55a832fSFam Zheng * on the caller, so leave it. */ 4384fb588e9SPaolo Bonzini QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 439c55a832fSFam Zheng if (other_job != job) { 4404c241cf5SPaolo Bonzini block_job_cancel_async(other_job); 441c55a832fSFam Zheng } 442c55a832fSFam Zheng } 4434fb588e9SPaolo Bonzini while (!QLIST_EMPTY(&txn->jobs)) { 4444fb588e9SPaolo Bonzini other_job = QLIST_FIRST(&txn->jobs); 445b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 4464fb588e9SPaolo Bonzini if (!other_job->completed) { 4474fb588e9SPaolo Bonzini assert(other_job->cancelled); 4484fb588e9SPaolo Bonzini block_job_finish_sync(other_job, NULL, NULL); 4494fb588e9SPaolo Bonzini } 450c55a832fSFam Zheng block_job_completed_single(other_job); 451c55a832fSFam Zheng aio_context_release(ctx); 452c55a832fSFam Zheng } 4534fb588e9SPaolo Bonzini 4544fb588e9SPaolo Bonzini block_job_txn_unref(txn); 455c55a832fSFam Zheng } 456c55a832fSFam Zheng 457c55a832fSFam Zheng static void block_job_completed_txn_success(BlockJob *job) 458c55a832fSFam Zheng { 459c55a832fSFam Zheng AioContext *ctx; 460c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 461c55a832fSFam Zheng BlockJob *other_job, *next; 462c55a832fSFam Zheng /* 463c55a832fSFam Zheng * Successful completion, see if there are other running jobs in this 464c55a832fSFam Zheng * txn. 465c55a832fSFam Zheng */ 466c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 467c55a832fSFam Zheng if (!other_job->completed) { 468c55a832fSFam Zheng return; 469c55a832fSFam Zheng } 470c55a832fSFam Zheng } 471c55a832fSFam Zheng /* We are the last completed job, commit the transaction. */ 472c55a832fSFam Zheng QLIST_FOREACH_SAFE(other_job, &txn->jobs, txn_list, next) { 473b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 474c55a832fSFam Zheng aio_context_acquire(ctx); 475c55a832fSFam Zheng assert(other_job->ret == 0); 476c55a832fSFam Zheng block_job_completed_single(other_job); 477c55a832fSFam Zheng aio_context_release(ctx); 478c55a832fSFam Zheng } 479c55a832fSFam Zheng } 480c55a832fSFam Zheng 481aa9ef2e6SJohn Snow /* Assumes the block_job_mutex is held */ 482aa9ef2e6SJohn Snow static bool block_job_timer_pending(BlockJob *job) 483aa9ef2e6SJohn Snow { 484aa9ef2e6SJohn Snow return timer_pending(&job->sleep_timer); 485aa9ef2e6SJohn Snow } 486aa9ef2e6SJohn Snow 4872f0c9fe6SPaolo Bonzini void block_job_set_speed(BlockJob *job, int64_t speed, Error **errp) 4882f0c9fe6SPaolo Bonzini { 4892f0c9fe6SPaolo Bonzini Error *local_err = NULL; 490aa9ef2e6SJohn Snow int64_t old_speed = job->speed; 4912f0c9fe6SPaolo Bonzini 4923fc4b10aSFam Zheng if (!job->driver->set_speed) { 493c6bd8c70SMarkus Armbruster error_setg(errp, QERR_UNSUPPORTED); 4942f0c9fe6SPaolo Bonzini return; 4952f0c9fe6SPaolo Bonzini } 4963fc4b10aSFam Zheng job->driver->set_speed(job, speed, &local_err); 49784d18f06SMarkus Armbruster if (local_err) { 4982f0c9fe6SPaolo Bonzini error_propagate(errp, local_err); 4992f0c9fe6SPaolo Bonzini return; 5002f0c9fe6SPaolo Bonzini } 5012f0c9fe6SPaolo Bonzini 5022f0c9fe6SPaolo Bonzini job->speed = speed; 503aa9ef2e6SJohn Snow if (speed <= old_speed) { 504aa9ef2e6SJohn Snow return; 505aa9ef2e6SJohn Snow } 506aa9ef2e6SJohn Snow 507aa9ef2e6SJohn Snow /* kick only if a timer is pending */ 508aa9ef2e6SJohn Snow block_job_enter_cond(job, block_job_timer_pending); 5092f0c9fe6SPaolo Bonzini } 5102f0c9fe6SPaolo Bonzini 511aeae883bSPaolo Bonzini void block_job_complete(BlockJob *job, Error **errp) 512aeae883bSPaolo Bonzini { 513559b935fSJohn Snow /* Should not be reachable via external interface for internal jobs */ 514559b935fSJohn Snow assert(job->id); 5155ccac6f1SJohn Snow if (job->pause_count || job->cancelled || 5165ccac6f1SJohn Snow !block_job_started(job) || !job->driver->complete) { 5179df229c3SAlberto Garcia error_setg(errp, "The active block job '%s' cannot be completed", 5189df229c3SAlberto Garcia job->id); 519aeae883bSPaolo Bonzini return; 520aeae883bSPaolo Bonzini } 521aeae883bSPaolo Bonzini 5223fc4b10aSFam Zheng job->driver->complete(job, errp); 523aeae883bSPaolo Bonzini } 524aeae883bSPaolo Bonzini 5250df4ba58SJohn Snow void block_job_user_pause(BlockJob *job) 5260df4ba58SJohn Snow { 5270df4ba58SJohn Snow job->user_paused = true; 5280df4ba58SJohn Snow block_job_pause(job); 5290df4ba58SJohn Snow } 5300df4ba58SJohn Snow 5310df4ba58SJohn Snow bool block_job_user_paused(BlockJob *job) 5320df4ba58SJohn Snow { 5336573d9c6SPaolo Bonzini return job->user_paused; 5340df4ba58SJohn Snow } 5350df4ba58SJohn Snow 5360df4ba58SJohn Snow void block_job_user_resume(BlockJob *job) 5370df4ba58SJohn Snow { 5380df4ba58SJohn Snow if (job && job->user_paused && job->pause_count > 0) { 5392caf63a9SPaolo Bonzini block_job_iostatus_reset(job); 5404c241cf5SPaolo Bonzini job->user_paused = false; 5410df4ba58SJohn Snow block_job_resume(job); 5420df4ba58SJohn Snow } 5430df4ba58SJohn Snow } 5440df4ba58SJohn Snow 5458acc72a4SPaolo Bonzini void block_job_cancel(BlockJob *job) 5468acc72a4SPaolo Bonzini { 5475ccac6f1SJohn Snow if (block_job_started(job)) { 5484c241cf5SPaolo Bonzini block_job_cancel_async(job); 549751ebd76SFam Zheng block_job_enter(job); 5505ccac6f1SJohn Snow } else { 5515ccac6f1SJohn Snow block_job_completed(job, -ECANCELED); 5525ccac6f1SJohn Snow } 5538acc72a4SPaolo Bonzini } 5548acc72a4SPaolo Bonzini 555345f9e1bSMax Reitz /* A wrapper around block_job_cancel() taking an Error ** parameter so it may be 556345f9e1bSMax Reitz * used with block_job_finish_sync() without the need for (rather nasty) 557345f9e1bSMax Reitz * function pointer casts there. */ 558345f9e1bSMax Reitz static void block_job_cancel_err(BlockJob *job, Error **errp) 559345f9e1bSMax Reitz { 560345f9e1bSMax Reitz block_job_cancel(job); 561345f9e1bSMax Reitz } 562345f9e1bSMax Reitz 563345f9e1bSMax Reitz int block_job_cancel_sync(BlockJob *job) 564345f9e1bSMax Reitz { 565345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_cancel_err, NULL); 566345f9e1bSMax Reitz } 567345f9e1bSMax Reitz 568a1a2af07SKevin Wolf void block_job_cancel_sync_all(void) 569a1a2af07SKevin Wolf { 570a1a2af07SKevin Wolf BlockJob *job; 571a1a2af07SKevin Wolf AioContext *aio_context; 572a1a2af07SKevin Wolf 573a1a2af07SKevin Wolf while ((job = QLIST_FIRST(&block_jobs))) { 574b6d2e599SKevin Wolf aio_context = blk_get_aio_context(job->blk); 575a1a2af07SKevin Wolf aio_context_acquire(aio_context); 576a1a2af07SKevin Wolf block_job_cancel_sync(job); 577a1a2af07SKevin Wolf aio_context_release(aio_context); 578a1a2af07SKevin Wolf } 579a1a2af07SKevin Wolf } 580a1a2af07SKevin Wolf 581345f9e1bSMax Reitz int block_job_complete_sync(BlockJob *job, Error **errp) 582345f9e1bSMax Reitz { 583345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_complete, errp); 584345f9e1bSMax Reitz } 585345f9e1bSMax Reitz 586559b935fSJohn Snow BlockJobInfo *block_job_query(BlockJob *job, Error **errp) 58730e628b7SPaolo Bonzini { 588559b935fSJohn Snow BlockJobInfo *info; 589559b935fSJohn Snow 590559b935fSJohn Snow if (block_job_is_internal(job)) { 591559b935fSJohn Snow error_setg(errp, "Cannot query QEMU internal jobs"); 592559b935fSJohn Snow return NULL; 593559b935fSJohn Snow } 594559b935fSJohn Snow info = g_new0(BlockJobInfo, 1); 595977c736fSMarkus Armbruster info->type = g_strdup(BlockJobType_str(job->driver->job_type)); 5968ccb9569SKevin Wolf info->device = g_strdup(job->id); 59730e628b7SPaolo Bonzini info->len = job->len; 598fc24908eSPaolo Bonzini info->busy = atomic_read(&job->busy); 599751ebd76SFam Zheng info->paused = job->pause_count > 0; 60030e628b7SPaolo Bonzini info->offset = job->offset; 60130e628b7SPaolo Bonzini info->speed = job->speed; 60232c81a4aSPaolo Bonzini info->io_status = job->iostatus; 603ef6dbf1eSMax Reitz info->ready = job->ready; 60430e628b7SPaolo Bonzini return info; 60530e628b7SPaolo Bonzini } 60632c81a4aSPaolo Bonzini 60732c81a4aSPaolo Bonzini static void block_job_iostatus_set_err(BlockJob *job, int error) 60832c81a4aSPaolo Bonzini { 60932c81a4aSPaolo Bonzini if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 61032c81a4aSPaolo Bonzini job->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE : 61132c81a4aSPaolo Bonzini BLOCK_DEVICE_IO_STATUS_FAILED; 61232c81a4aSPaolo Bonzini } 61332c81a4aSPaolo Bonzini } 61432c81a4aSPaolo Bonzini 6158254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job) 616a66a2a36SPaolo Bonzini { 617559b935fSJohn Snow if (block_job_is_internal(job)) { 618559b935fSJohn Snow return; 619559b935fSJohn Snow } 620559b935fSJohn Snow 621bcada37bSWenchao Xia qapi_event_send_block_job_cancelled(job->driver->job_type, 6228ccb9569SKevin Wolf job->id, 623a66a2a36SPaolo Bonzini job->len, 624a66a2a36SPaolo Bonzini job->offset, 625bcada37bSWenchao Xia job->speed, 626bcada37bSWenchao Xia &error_abort); 627a66a2a36SPaolo Bonzini } 628a66a2a36SPaolo Bonzini 6298254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg) 630a66a2a36SPaolo Bonzini { 631559b935fSJohn Snow if (block_job_is_internal(job)) { 632559b935fSJohn Snow return; 633559b935fSJohn Snow } 634559b935fSJohn Snow 635bcada37bSWenchao Xia qapi_event_send_block_job_completed(job->driver->job_type, 6368ccb9569SKevin Wolf job->id, 637bcada37bSWenchao Xia job->len, 638bcada37bSWenchao Xia job->offset, 639bcada37bSWenchao Xia job->speed, 640bcada37bSWenchao Xia !!msg, 641bcada37bSWenchao Xia msg, 642bcada37bSWenchao Xia &error_abort); 643bcada37bSWenchao Xia } 644bcada37bSWenchao Xia 64588691b37SPaolo Bonzini /* 64688691b37SPaolo Bonzini * API for block job drivers and the block layer. These functions are 64788691b37SPaolo Bonzini * declared in blockjob_int.h. 64888691b37SPaolo Bonzini */ 64988691b37SPaolo Bonzini 65088691b37SPaolo Bonzini void *block_job_create(const char *job_id, const BlockJobDriver *driver, 65188691b37SPaolo Bonzini BlockDriverState *bs, uint64_t perm, 65288691b37SPaolo Bonzini uint64_t shared_perm, int64_t speed, int flags, 65388691b37SPaolo Bonzini BlockCompletionFunc *cb, void *opaque, Error **errp) 65488691b37SPaolo Bonzini { 65588691b37SPaolo Bonzini BlockBackend *blk; 65688691b37SPaolo Bonzini BlockJob *job; 65788691b37SPaolo Bonzini int ret; 65888691b37SPaolo Bonzini 65988691b37SPaolo Bonzini if (bs->job) { 66088691b37SPaolo Bonzini error_setg(errp, QERR_DEVICE_IN_USE, bdrv_get_device_name(bs)); 66188691b37SPaolo Bonzini return NULL; 66288691b37SPaolo Bonzini } 66388691b37SPaolo Bonzini 66488691b37SPaolo Bonzini if (job_id == NULL && !(flags & BLOCK_JOB_INTERNAL)) { 66588691b37SPaolo Bonzini job_id = bdrv_get_device_name(bs); 66688691b37SPaolo Bonzini if (!*job_id) { 66788691b37SPaolo Bonzini error_setg(errp, "An explicit job ID is required for this node"); 66888691b37SPaolo Bonzini return NULL; 66988691b37SPaolo Bonzini } 67088691b37SPaolo Bonzini } 67188691b37SPaolo Bonzini 67288691b37SPaolo Bonzini if (job_id) { 67388691b37SPaolo Bonzini if (flags & BLOCK_JOB_INTERNAL) { 67488691b37SPaolo Bonzini error_setg(errp, "Cannot specify job ID for internal block job"); 67588691b37SPaolo Bonzini return NULL; 67688691b37SPaolo Bonzini } 67788691b37SPaolo Bonzini 67888691b37SPaolo Bonzini if (!id_wellformed(job_id)) { 67988691b37SPaolo Bonzini error_setg(errp, "Invalid job ID '%s'", job_id); 68088691b37SPaolo Bonzini return NULL; 68188691b37SPaolo Bonzini } 68288691b37SPaolo Bonzini 68388691b37SPaolo Bonzini if (block_job_get(job_id)) { 68488691b37SPaolo Bonzini error_setg(errp, "Job ID '%s' already in use", job_id); 68588691b37SPaolo Bonzini return NULL; 68688691b37SPaolo Bonzini } 68788691b37SPaolo Bonzini } 68888691b37SPaolo Bonzini 68988691b37SPaolo Bonzini blk = blk_new(perm, shared_perm); 69088691b37SPaolo Bonzini ret = blk_insert_bs(blk, bs, errp); 69188691b37SPaolo Bonzini if (ret < 0) { 69288691b37SPaolo Bonzini blk_unref(blk); 69388691b37SPaolo Bonzini return NULL; 69488691b37SPaolo Bonzini } 69588691b37SPaolo Bonzini 69688691b37SPaolo Bonzini job = g_malloc0(driver->instance_size); 69788691b37SPaolo Bonzini job->driver = driver; 69888691b37SPaolo Bonzini job->id = g_strdup(job_id); 69988691b37SPaolo Bonzini job->blk = blk; 70088691b37SPaolo Bonzini job->cb = cb; 70188691b37SPaolo Bonzini job->opaque = opaque; 70288691b37SPaolo Bonzini job->busy = false; 70388691b37SPaolo Bonzini job->paused = true; 70488691b37SPaolo Bonzini job->pause_count = 1; 70588691b37SPaolo Bonzini job->refcnt = 1; 706fc24908eSPaolo Bonzini aio_timer_init(qemu_get_aio_context(), &job->sleep_timer, 707fc24908eSPaolo Bonzini QEMU_CLOCK_REALTIME, SCALE_NS, 708fc24908eSPaolo Bonzini block_job_sleep_timer_cb, job); 70988691b37SPaolo Bonzini 71088691b37SPaolo Bonzini error_setg(&job->blocker, "block device is in use by block job: %s", 711977c736fSMarkus Armbruster BlockJobType_str(driver->job_type)); 71288691b37SPaolo Bonzini block_job_add_bdrv(job, "main node", bs, 0, BLK_PERM_ALL, &error_abort); 71388691b37SPaolo Bonzini bs->job = job; 71488691b37SPaolo Bonzini 71588691b37SPaolo Bonzini bdrv_op_unblock(bs, BLOCK_OP_TYPE_DATAPLANE, job->blocker); 71688691b37SPaolo Bonzini 71788691b37SPaolo Bonzini QLIST_INSERT_HEAD(&block_jobs, job, job_list); 71888691b37SPaolo Bonzini 71988691b37SPaolo Bonzini blk_add_aio_context_notifier(blk, block_job_attached_aio_context, 72088691b37SPaolo Bonzini block_job_detach_aio_context, job); 72188691b37SPaolo Bonzini 72288691b37SPaolo Bonzini /* Only set speed when necessary to avoid NotSupported error */ 72388691b37SPaolo Bonzini if (speed != 0) { 72488691b37SPaolo Bonzini Error *local_err = NULL; 72588691b37SPaolo Bonzini 72688691b37SPaolo Bonzini block_job_set_speed(job, speed, &local_err); 72788691b37SPaolo Bonzini if (local_err) { 72888691b37SPaolo Bonzini block_job_unref(job); 72988691b37SPaolo Bonzini error_propagate(errp, local_err); 73088691b37SPaolo Bonzini return NULL; 73188691b37SPaolo Bonzini } 73288691b37SPaolo Bonzini } 73388691b37SPaolo Bonzini return job; 73488691b37SPaolo Bonzini } 73588691b37SPaolo Bonzini 736f321dcb5SPaolo Bonzini void block_job_pause_all(void) 737f321dcb5SPaolo Bonzini { 738f321dcb5SPaolo Bonzini BlockJob *job = NULL; 739f321dcb5SPaolo Bonzini while ((job = block_job_next(job))) { 740f321dcb5SPaolo Bonzini AioContext *aio_context = blk_get_aio_context(job->blk); 741f321dcb5SPaolo Bonzini 742f321dcb5SPaolo Bonzini aio_context_acquire(aio_context); 7433d5d319eSAlberto Garcia block_job_ref(job); 744f321dcb5SPaolo Bonzini block_job_pause(job); 745f321dcb5SPaolo Bonzini aio_context_release(aio_context); 746f321dcb5SPaolo Bonzini } 747f321dcb5SPaolo Bonzini } 748f321dcb5SPaolo Bonzini 74988691b37SPaolo Bonzini void block_job_early_fail(BlockJob *job) 75088691b37SPaolo Bonzini { 75188691b37SPaolo Bonzini block_job_unref(job); 75288691b37SPaolo Bonzini } 75388691b37SPaolo Bonzini 75488691b37SPaolo Bonzini void block_job_completed(BlockJob *job, int ret) 75588691b37SPaolo Bonzini { 75688691b37SPaolo Bonzini assert(blk_bs(job->blk)->job == job); 75788691b37SPaolo Bonzini assert(!job->completed); 75888691b37SPaolo Bonzini job->completed = true; 75988691b37SPaolo Bonzini job->ret = ret; 76088691b37SPaolo Bonzini if (!job->txn) { 76188691b37SPaolo Bonzini block_job_completed_single(job); 76288691b37SPaolo Bonzini } else if (ret < 0 || block_job_is_cancelled(job)) { 76388691b37SPaolo Bonzini block_job_completed_txn_abort(job); 76488691b37SPaolo Bonzini } else { 76588691b37SPaolo Bonzini block_job_completed_txn_success(job); 76688691b37SPaolo Bonzini } 76788691b37SPaolo Bonzini } 76888691b37SPaolo Bonzini 76988691b37SPaolo Bonzini static bool block_job_should_pause(BlockJob *job) 77088691b37SPaolo Bonzini { 77188691b37SPaolo Bonzini return job->pause_count > 0; 77288691b37SPaolo Bonzini } 77388691b37SPaolo Bonzini 774fc24908eSPaolo Bonzini /* Yield, and schedule a timer to reenter the coroutine after @ns nanoseconds. 775fc24908eSPaolo Bonzini * Reentering the job coroutine with block_job_enter() before the timer has 776fc24908eSPaolo Bonzini * expired is allowed and cancels the timer. 777fc24908eSPaolo Bonzini * 778fc24908eSPaolo Bonzini * If @ns is (uint64_t) -1, no timer is scheduled and block_job_enter() must be 779fc24908eSPaolo Bonzini * called explicitly. */ 780fc24908eSPaolo Bonzini static void block_job_do_yield(BlockJob *job, uint64_t ns) 781356f59b8SPaolo Bonzini { 782fc24908eSPaolo Bonzini block_job_lock(); 783fc24908eSPaolo Bonzini if (ns != -1) { 784fc24908eSPaolo Bonzini timer_mod(&job->sleep_timer, ns); 785fc24908eSPaolo Bonzini } 786356f59b8SPaolo Bonzini job->busy = false; 787fc24908eSPaolo Bonzini block_job_unlock(); 788356f59b8SPaolo Bonzini qemu_coroutine_yield(); 789356f59b8SPaolo Bonzini 790356f59b8SPaolo Bonzini /* Set by block_job_enter before re-entering the coroutine. */ 791356f59b8SPaolo Bonzini assert(job->busy); 792356f59b8SPaolo Bonzini } 793356f59b8SPaolo Bonzini 79488691b37SPaolo Bonzini void coroutine_fn block_job_pause_point(BlockJob *job) 79588691b37SPaolo Bonzini { 79688691b37SPaolo Bonzini assert(job && block_job_started(job)); 79788691b37SPaolo Bonzini 79888691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 79988691b37SPaolo Bonzini return; 80088691b37SPaolo Bonzini } 80188691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 80288691b37SPaolo Bonzini return; 80388691b37SPaolo Bonzini } 80488691b37SPaolo Bonzini 80588691b37SPaolo Bonzini if (job->driver->pause) { 80688691b37SPaolo Bonzini job->driver->pause(job); 80788691b37SPaolo Bonzini } 80888691b37SPaolo Bonzini 80988691b37SPaolo Bonzini if (block_job_should_pause(job) && !block_job_is_cancelled(job)) { 81088691b37SPaolo Bonzini job->paused = true; 811fc24908eSPaolo Bonzini block_job_do_yield(job, -1); 81288691b37SPaolo Bonzini job->paused = false; 81388691b37SPaolo Bonzini } 81488691b37SPaolo Bonzini 81588691b37SPaolo Bonzini if (job->driver->resume) { 81688691b37SPaolo Bonzini job->driver->resume(job); 81788691b37SPaolo Bonzini } 81888691b37SPaolo Bonzini } 81988691b37SPaolo Bonzini 820f321dcb5SPaolo Bonzini void block_job_resume_all(void) 821f321dcb5SPaolo Bonzini { 8223d5d319eSAlberto Garcia BlockJob *job, *next; 8233d5d319eSAlberto Garcia 8243d5d319eSAlberto Garcia QLIST_FOREACH_SAFE(job, &block_jobs, job_list, next) { 825f321dcb5SPaolo Bonzini AioContext *aio_context = blk_get_aio_context(job->blk); 826f321dcb5SPaolo Bonzini 827f321dcb5SPaolo Bonzini aio_context_acquire(aio_context); 828f321dcb5SPaolo Bonzini block_job_resume(job); 8293d5d319eSAlberto Garcia block_job_unref(job); 830f321dcb5SPaolo Bonzini aio_context_release(aio_context); 831f321dcb5SPaolo Bonzini } 832f321dcb5SPaolo Bonzini } 833f321dcb5SPaolo Bonzini 834aa9ef2e6SJohn Snow /* 835aa9ef2e6SJohn Snow * Conditionally enter a block_job pending a call to fn() while 836aa9ef2e6SJohn Snow * under the block_job_lock critical section. 837aa9ef2e6SJohn Snow */ 838aa9ef2e6SJohn Snow static void block_job_enter_cond(BlockJob *job, bool(*fn)(BlockJob *job)) 83988691b37SPaolo Bonzini { 840eb05e011SPaolo Bonzini if (!block_job_started(job)) { 841eb05e011SPaolo Bonzini return; 842eb05e011SPaolo Bonzini } 843eb05e011SPaolo Bonzini if (job->deferred_to_main_loop) { 844eb05e011SPaolo Bonzini return; 845eb05e011SPaolo Bonzini } 846eb05e011SPaolo Bonzini 847fc24908eSPaolo Bonzini block_job_lock(); 848356f59b8SPaolo Bonzini if (job->busy) { 849fc24908eSPaolo Bonzini block_job_unlock(); 850356f59b8SPaolo Bonzini return; 85188691b37SPaolo Bonzini } 852356f59b8SPaolo Bonzini 853aa9ef2e6SJohn Snow if (fn && !fn(job)) { 854aa9ef2e6SJohn Snow block_job_unlock(); 855aa9ef2e6SJohn Snow return; 856aa9ef2e6SJohn Snow } 857aa9ef2e6SJohn Snow 858fc24908eSPaolo Bonzini assert(!job->deferred_to_main_loop); 859fc24908eSPaolo Bonzini timer_del(&job->sleep_timer); 860356f59b8SPaolo Bonzini job->busy = true; 861fc24908eSPaolo Bonzini block_job_unlock(); 862356f59b8SPaolo Bonzini aio_co_wake(job->co); 86388691b37SPaolo Bonzini } 86488691b37SPaolo Bonzini 865aa9ef2e6SJohn Snow void block_job_enter(BlockJob *job) 866aa9ef2e6SJohn Snow { 867aa9ef2e6SJohn Snow block_job_enter_cond(job, NULL); 868aa9ef2e6SJohn Snow } 869aa9ef2e6SJohn Snow 87088691b37SPaolo Bonzini bool block_job_is_cancelled(BlockJob *job) 87188691b37SPaolo Bonzini { 87288691b37SPaolo Bonzini return job->cancelled; 87388691b37SPaolo Bonzini } 87488691b37SPaolo Bonzini 8755bf1d5a7SPaolo Bonzini void block_job_sleep_ns(BlockJob *job, int64_t ns) 87688691b37SPaolo Bonzini { 87788691b37SPaolo Bonzini assert(job->busy); 87888691b37SPaolo Bonzini 87988691b37SPaolo Bonzini /* Check cancellation *before* setting busy = false, too! */ 88088691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 88188691b37SPaolo Bonzini return; 88288691b37SPaolo Bonzini } 88388691b37SPaolo Bonzini 88488691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 885fc24908eSPaolo Bonzini block_job_do_yield(job, qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + ns); 88688691b37SPaolo Bonzini } 88788691b37SPaolo Bonzini 88888691b37SPaolo Bonzini block_job_pause_point(job); 88988691b37SPaolo Bonzini } 89088691b37SPaolo Bonzini 89188691b37SPaolo Bonzini void block_job_yield(BlockJob *job) 89288691b37SPaolo Bonzini { 89388691b37SPaolo Bonzini assert(job->busy); 89488691b37SPaolo Bonzini 89588691b37SPaolo Bonzini /* Check cancellation *before* setting busy = false, too! */ 89688691b37SPaolo Bonzini if (block_job_is_cancelled(job)) { 89788691b37SPaolo Bonzini return; 89888691b37SPaolo Bonzini } 89988691b37SPaolo Bonzini 90088691b37SPaolo Bonzini if (!block_job_should_pause(job)) { 901fc24908eSPaolo Bonzini block_job_do_yield(job, -1); 90288691b37SPaolo Bonzini } 90388691b37SPaolo Bonzini 90488691b37SPaolo Bonzini block_job_pause_point(job); 90588691b37SPaolo Bonzini } 90688691b37SPaolo Bonzini 9072caf63a9SPaolo Bonzini void block_job_iostatus_reset(BlockJob *job) 9082caf63a9SPaolo Bonzini { 9094c241cf5SPaolo Bonzini if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 9104c241cf5SPaolo Bonzini return; 9114c241cf5SPaolo Bonzini } 9124c241cf5SPaolo Bonzini assert(job->user_paused && job->pause_count > 0); 9132caf63a9SPaolo Bonzini job->iostatus = BLOCK_DEVICE_IO_STATUS_OK; 9142caf63a9SPaolo Bonzini } 9152caf63a9SPaolo Bonzini 916bcada37bSWenchao Xia void block_job_event_ready(BlockJob *job) 917bcada37bSWenchao Xia { 918ef6dbf1eSMax Reitz job->ready = true; 919ef6dbf1eSMax Reitz 920559b935fSJohn Snow if (block_job_is_internal(job)) { 921559b935fSJohn Snow return; 922559b935fSJohn Snow } 923559b935fSJohn Snow 924518848a2SMarkus Armbruster qapi_event_send_block_job_ready(job->driver->job_type, 9258ccb9569SKevin Wolf job->id, 926518848a2SMarkus Armbruster job->len, 927518848a2SMarkus Armbruster job->offset, 928518848a2SMarkus Armbruster job->speed, &error_abort); 929a66a2a36SPaolo Bonzini } 930a66a2a36SPaolo Bonzini 93181e254dcSKevin Wolf BlockErrorAction block_job_error_action(BlockJob *job, BlockdevOnError on_err, 93232c81a4aSPaolo Bonzini int is_read, int error) 93332c81a4aSPaolo Bonzini { 93432c81a4aSPaolo Bonzini BlockErrorAction action; 93532c81a4aSPaolo Bonzini 93632c81a4aSPaolo Bonzini switch (on_err) { 93732c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_ENOSPC: 9388c398252SKevin Wolf case BLOCKDEV_ON_ERROR_AUTO: 939a589569fSWenchao Xia action = (error == ENOSPC) ? 940a589569fSWenchao Xia BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT; 94132c81a4aSPaolo Bonzini break; 94232c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_STOP: 943a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_STOP; 94432c81a4aSPaolo Bonzini break; 94532c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_REPORT: 946a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_REPORT; 94732c81a4aSPaolo Bonzini break; 94832c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_IGNORE: 949a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_IGNORE; 95032c81a4aSPaolo Bonzini break; 95132c81a4aSPaolo Bonzini default: 95232c81a4aSPaolo Bonzini abort(); 95332c81a4aSPaolo Bonzini } 954559b935fSJohn Snow if (!block_job_is_internal(job)) { 9558ccb9569SKevin Wolf qapi_event_send_block_job_error(job->id, 9565a2d2cbdSWenchao Xia is_read ? IO_OPERATION_TYPE_READ : 9575a2d2cbdSWenchao Xia IO_OPERATION_TYPE_WRITE, 9585a2d2cbdSWenchao Xia action, &error_abort); 959559b935fSJohn Snow } 960a589569fSWenchao Xia if (action == BLOCK_ERROR_ACTION_STOP) { 961751ebd76SFam Zheng /* make the pause user visible, which will be resumed from QMP. */ 9620df4ba58SJohn Snow block_job_user_pause(job); 96332c81a4aSPaolo Bonzini block_job_iostatus_set_err(job, error); 96432c81a4aSPaolo Bonzini } 96532c81a4aSPaolo Bonzini return action; 96632c81a4aSPaolo Bonzini } 967dec7d421SStefan Hajnoczi 968dec7d421SStefan Hajnoczi typedef struct { 969dec7d421SStefan Hajnoczi BlockJob *job; 970dec7d421SStefan Hajnoczi AioContext *aio_context; 971dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn; 972dec7d421SStefan Hajnoczi void *opaque; 973dec7d421SStefan Hajnoczi } BlockJobDeferToMainLoopData; 974dec7d421SStefan Hajnoczi 975dec7d421SStefan Hajnoczi static void block_job_defer_to_main_loop_bh(void *opaque) 976dec7d421SStefan Hajnoczi { 977dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = opaque; 978dec7d421SStefan Hajnoczi AioContext *aio_context; 979dec7d421SStefan Hajnoczi 980dec7d421SStefan Hajnoczi /* Prevent race with block_job_defer_to_main_loop() */ 981dec7d421SStefan Hajnoczi aio_context_acquire(data->aio_context); 982dec7d421SStefan Hajnoczi 983dec7d421SStefan Hajnoczi /* Fetch BDS AioContext again, in case it has changed */ 984b6d2e599SKevin Wolf aio_context = blk_get_aio_context(data->job->blk); 985d79df2a2SPaolo Bonzini if (aio_context != data->aio_context) { 986dec7d421SStefan Hajnoczi aio_context_acquire(aio_context); 987d79df2a2SPaolo Bonzini } 988dec7d421SStefan Hajnoczi 989dec7d421SStefan Hajnoczi data->fn(data->job, data->opaque); 990dec7d421SStefan Hajnoczi 991d79df2a2SPaolo Bonzini if (aio_context != data->aio_context) { 992dec7d421SStefan Hajnoczi aio_context_release(aio_context); 993d79df2a2SPaolo Bonzini } 994dec7d421SStefan Hajnoczi 995dec7d421SStefan Hajnoczi aio_context_release(data->aio_context); 996dec7d421SStefan Hajnoczi 997dec7d421SStefan Hajnoczi g_free(data); 998dec7d421SStefan Hajnoczi } 999dec7d421SStefan Hajnoczi 1000dec7d421SStefan Hajnoczi void block_job_defer_to_main_loop(BlockJob *job, 1001dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn, 1002dec7d421SStefan Hajnoczi void *opaque) 1003dec7d421SStefan Hajnoczi { 1004dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = g_malloc(sizeof(*data)); 1005dec7d421SStefan Hajnoczi data->job = job; 1006b6d2e599SKevin Wolf data->aio_context = blk_get_aio_context(job->blk); 1007dec7d421SStefan Hajnoczi data->fn = fn; 1008dec7d421SStefan Hajnoczi data->opaque = opaque; 1009794f0141SFam Zheng job->deferred_to_main_loop = true; 1010dec7d421SStefan Hajnoczi 1011fffb6e12SPaolo Bonzini aio_bh_schedule_oneshot(qemu_get_aio_context(), 1012fffb6e12SPaolo Bonzini block_job_defer_to_main_loop_bh, data); 1013dec7d421SStefan Hajnoczi } 1014