12f0c9fe6SPaolo Bonzini /* 22f0c9fe6SPaolo Bonzini * QEMU System Emulator block driver 32f0c9fe6SPaolo Bonzini * 42f0c9fe6SPaolo Bonzini * Copyright (c) 2011 IBM Corp. 52f0c9fe6SPaolo Bonzini * Copyright (c) 2012 Red Hat, Inc. 62f0c9fe6SPaolo Bonzini * 72f0c9fe6SPaolo Bonzini * Permission is hereby granted, free of charge, to any person obtaining a copy 82f0c9fe6SPaolo Bonzini * of this software and associated documentation files (the "Software"), to deal 92f0c9fe6SPaolo Bonzini * in the Software without restriction, including without limitation the rights 102f0c9fe6SPaolo Bonzini * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 112f0c9fe6SPaolo Bonzini * copies of the Software, and to permit persons to whom the Software is 122f0c9fe6SPaolo Bonzini * furnished to do so, subject to the following conditions: 132f0c9fe6SPaolo Bonzini * 142f0c9fe6SPaolo Bonzini * The above copyright notice and this permission notice shall be included in 152f0c9fe6SPaolo Bonzini * all copies or substantial portions of the Software. 162f0c9fe6SPaolo Bonzini * 172f0c9fe6SPaolo Bonzini * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 182f0c9fe6SPaolo Bonzini * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 192f0c9fe6SPaolo Bonzini * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 202f0c9fe6SPaolo Bonzini * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 212f0c9fe6SPaolo Bonzini * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 222f0c9fe6SPaolo Bonzini * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 232f0c9fe6SPaolo Bonzini * THE SOFTWARE. 242f0c9fe6SPaolo Bonzini */ 252f0c9fe6SPaolo Bonzini 26d38ea87aSPeter Maydell #include "qemu/osdep.h" 272f0c9fe6SPaolo Bonzini #include "qemu-common.h" 28737e150eSPaolo Bonzini #include "block/block.h" 29c87621eaSJohn Snow #include "block/blockjob_int.h" 30737e150eSPaolo Bonzini #include "block/block_int.h" 31373340b2SMax Reitz #include "sysemu/block-backend.h" 32cc7a8ea7SMarkus Armbruster #include "qapi/qmp/qerror.h" 337b1b5d19SPaolo Bonzini #include "qapi/qmp/qjson.h" 3410817bf0SDaniel P. Berrange #include "qemu/coroutine.h" 357f0317cfSAlberto Garcia #include "qemu/id.h" 362f0c9fe6SPaolo Bonzini #include "qmp-commands.h" 371de7afc9SPaolo Bonzini #include "qemu/timer.h" 385a2d2cbdSWenchao Xia #include "qapi-event.h" 392f0c9fe6SPaolo Bonzini 408254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job); 418254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg); 428254b6d9SJohn Snow 43c55a832fSFam Zheng /* Transactional group of block jobs */ 44c55a832fSFam Zheng struct BlockJobTxn { 45c55a832fSFam Zheng 46c55a832fSFam Zheng /* Is this txn being cancelled? */ 47c55a832fSFam Zheng bool aborting; 48c55a832fSFam Zheng 49c55a832fSFam Zheng /* List of jobs */ 50c55a832fSFam Zheng QLIST_HEAD(, BlockJob) jobs; 51c55a832fSFam Zheng 52c55a832fSFam Zheng /* Reference count */ 53c55a832fSFam Zheng int refcnt; 54c55a832fSFam Zheng }; 55c55a832fSFam Zheng 56a7112795SAlberto Garcia static QLIST_HEAD(, BlockJob) block_jobs = QLIST_HEAD_INITIALIZER(block_jobs); 57a7112795SAlberto Garcia 5876d554e2SKevin Wolf static char *child_job_get_parent_desc(BdrvChild *c) 5976d554e2SKevin Wolf { 6076d554e2SKevin Wolf BlockJob *job = c->opaque; 6176d554e2SKevin Wolf return g_strdup_printf("%s job '%s'", 6276d554e2SKevin Wolf BlockJobType_lookup[job->driver->job_type], 6376d554e2SKevin Wolf job->id); 6476d554e2SKevin Wolf } 6576d554e2SKevin Wolf 6676d554e2SKevin Wolf static const BdrvChildRole child_job = { 6776d554e2SKevin Wolf .get_parent_desc = child_job_get_parent_desc, 6876d554e2SKevin Wolf .stay_at_node = true, 6976d554e2SKevin Wolf }; 7076d554e2SKevin Wolf 71a7112795SAlberto Garcia BlockJob *block_job_next(BlockJob *job) 72a7112795SAlberto Garcia { 73a7112795SAlberto Garcia if (!job) { 74a7112795SAlberto Garcia return QLIST_FIRST(&block_jobs); 75a7112795SAlberto Garcia } 76a7112795SAlberto Garcia return QLIST_NEXT(job, job_list); 77a7112795SAlberto Garcia } 78a7112795SAlberto Garcia 79ffb1f10cSAlberto Garcia BlockJob *block_job_get(const char *id) 80ffb1f10cSAlberto Garcia { 81ffb1f10cSAlberto Garcia BlockJob *job; 82ffb1f10cSAlberto Garcia 83ffb1f10cSAlberto Garcia QLIST_FOREACH(job, &block_jobs, job_list) { 84559b935fSJohn Snow if (job->id && !strcmp(id, job->id)) { 85ffb1f10cSAlberto Garcia return job; 86ffb1f10cSAlberto Garcia } 87ffb1f10cSAlberto Garcia } 88ffb1f10cSAlberto Garcia 89ffb1f10cSAlberto Garcia return NULL; 90ffb1f10cSAlberto Garcia } 91ffb1f10cSAlberto Garcia 92463e0be1SStefan Hajnoczi static void block_job_attached_aio_context(AioContext *new_context, 93463e0be1SStefan Hajnoczi void *opaque) 94463e0be1SStefan Hajnoczi { 95463e0be1SStefan Hajnoczi BlockJob *job = opaque; 96463e0be1SStefan Hajnoczi 97463e0be1SStefan Hajnoczi if (job->driver->attached_aio_context) { 98463e0be1SStefan Hajnoczi job->driver->attached_aio_context(job, new_context); 99463e0be1SStefan Hajnoczi } 100463e0be1SStefan Hajnoczi 101463e0be1SStefan Hajnoczi block_job_resume(job); 102463e0be1SStefan Hajnoczi } 103463e0be1SStefan Hajnoczi 104bae8196dSPaolo Bonzini static void block_job_drain(BlockJob *job) 105bae8196dSPaolo Bonzini { 106bae8196dSPaolo Bonzini /* If job is !job->busy this kicks it into the next pause point. */ 107bae8196dSPaolo Bonzini block_job_enter(job); 108bae8196dSPaolo Bonzini 109bae8196dSPaolo Bonzini blk_drain(job->blk); 110bae8196dSPaolo Bonzini if (job->driver->drain) { 111bae8196dSPaolo Bonzini job->driver->drain(job); 112bae8196dSPaolo Bonzini } 113bae8196dSPaolo Bonzini } 114bae8196dSPaolo Bonzini 115463e0be1SStefan Hajnoczi static void block_job_detach_aio_context(void *opaque) 116463e0be1SStefan Hajnoczi { 117463e0be1SStefan Hajnoczi BlockJob *job = opaque; 118463e0be1SStefan Hajnoczi 119463e0be1SStefan Hajnoczi /* In case the job terminates during aio_poll()... */ 120463e0be1SStefan Hajnoczi block_job_ref(job); 121463e0be1SStefan Hajnoczi 122463e0be1SStefan Hajnoczi block_job_pause(job); 123463e0be1SStefan Hajnoczi 124463e0be1SStefan Hajnoczi while (!job->paused && !job->completed) { 125bae8196dSPaolo Bonzini block_job_drain(job); 126463e0be1SStefan Hajnoczi } 127463e0be1SStefan Hajnoczi 128463e0be1SStefan Hajnoczi block_job_unref(job); 129463e0be1SStefan Hajnoczi } 130463e0be1SStefan Hajnoczi 131*bbc02b90SKevin Wolf void block_job_remove_all_bdrv(BlockJob *job) 132*bbc02b90SKevin Wolf { 133*bbc02b90SKevin Wolf GSList *l; 134*bbc02b90SKevin Wolf for (l = job->nodes; l; l = l->next) { 135*bbc02b90SKevin Wolf BdrvChild *c = l->data; 136*bbc02b90SKevin Wolf bdrv_op_unblock_all(c->bs, job->blocker); 137*bbc02b90SKevin Wolf bdrv_root_unref_child(c); 138*bbc02b90SKevin Wolf } 139*bbc02b90SKevin Wolf g_slist_free(job->nodes); 140*bbc02b90SKevin Wolf job->nodes = NULL; 141*bbc02b90SKevin Wolf } 142*bbc02b90SKevin Wolf 14376d554e2SKevin Wolf int block_job_add_bdrv(BlockJob *job, const char *name, BlockDriverState *bs, 14476d554e2SKevin Wolf uint64_t perm, uint64_t shared_perm, Error **errp) 14523d402d4SAlberto Garcia { 14676d554e2SKevin Wolf BdrvChild *c; 14776d554e2SKevin Wolf 14876d554e2SKevin Wolf c = bdrv_root_attach_child(bs, name, &child_job, perm, shared_perm, 14976d554e2SKevin Wolf job, errp); 15076d554e2SKevin Wolf if (c == NULL) { 15176d554e2SKevin Wolf return -EPERM; 15276d554e2SKevin Wolf } 15376d554e2SKevin Wolf 15476d554e2SKevin Wolf job->nodes = g_slist_prepend(job->nodes, c); 15523d402d4SAlberto Garcia bdrv_ref(bs); 15623d402d4SAlberto Garcia bdrv_op_block_all(bs, job->blocker); 15776d554e2SKevin Wolf 15876d554e2SKevin Wolf return 0; 15923d402d4SAlberto Garcia } 16023d402d4SAlberto Garcia 1617f0317cfSAlberto Garcia void *block_job_create(const char *job_id, const BlockJobDriver *driver, 162c6cc12bfSKevin Wolf BlockDriverState *bs, uint64_t perm, 163c6cc12bfSKevin Wolf uint64_t shared_perm, int64_t speed, int flags, 1647f0317cfSAlberto Garcia BlockCompletionFunc *cb, void *opaque, Error **errp) 1652f0c9fe6SPaolo Bonzini { 166b6d2e599SKevin Wolf BlockBackend *blk; 1672f0c9fe6SPaolo Bonzini BlockJob *job; 168d7086422SKevin Wolf int ret; 1692f0c9fe6SPaolo Bonzini 170628ff683SFam Zheng if (bs->job) { 171c6bd8c70SMarkus Armbruster error_setg(errp, QERR_DEVICE_IN_USE, bdrv_get_device_name(bs)); 1722f0c9fe6SPaolo Bonzini return NULL; 1732f0c9fe6SPaolo Bonzini } 174b6d2e599SKevin Wolf 175f81e0b45SJohn Snow if (job_id == NULL && !(flags & BLOCK_JOB_INTERNAL)) { 1767f0317cfSAlberto Garcia job_id = bdrv_get_device_name(bs); 1771562047cSKevin Wolf if (!*job_id) { 1781562047cSKevin Wolf error_setg(errp, "An explicit job ID is required for this node"); 1791562047cSKevin Wolf return NULL; 1801562047cSKevin Wolf } 1817f0317cfSAlberto Garcia } 1827f0317cfSAlberto Garcia 183f81e0b45SJohn Snow if (job_id) { 184f81e0b45SJohn Snow if (flags & BLOCK_JOB_INTERNAL) { 185f81e0b45SJohn Snow error_setg(errp, "Cannot specify job ID for internal block job"); 186f81e0b45SJohn Snow return NULL; 187f81e0b45SJohn Snow } 188f81e0b45SJohn Snow 1897f0317cfSAlberto Garcia if (!id_wellformed(job_id)) { 1907f0317cfSAlberto Garcia error_setg(errp, "Invalid job ID '%s'", job_id); 1917f0317cfSAlberto Garcia return NULL; 1927f0317cfSAlberto Garcia } 1937f0317cfSAlberto Garcia 1947f0317cfSAlberto Garcia if (block_job_get(job_id)) { 1957f0317cfSAlberto Garcia error_setg(errp, "Job ID '%s' already in use", job_id); 1967f0317cfSAlberto Garcia return NULL; 1977f0317cfSAlberto Garcia } 198f81e0b45SJohn Snow } 1997f0317cfSAlberto Garcia 200c6cc12bfSKevin Wolf blk = blk_new(perm, shared_perm); 201d7086422SKevin Wolf ret = blk_insert_bs(blk, bs, errp); 202d7086422SKevin Wolf if (ret < 0) { 203d7086422SKevin Wolf blk_unref(blk); 204d7086422SKevin Wolf return NULL; 205d7086422SKevin Wolf } 206b6d2e599SKevin Wolf 2073fc4b10aSFam Zheng job = g_malloc0(driver->instance_size); 2083718d8abSFam Zheng error_setg(&job->blocker, "block device is in use by block job: %s", 2093718d8abSFam Zheng BlockJobType_lookup[driver->job_type]); 21076d554e2SKevin Wolf block_job_add_bdrv(job, "main node", bs, 0, BLK_PERM_ALL, &error_abort); 211b112a65cSStefan Hajnoczi bdrv_op_unblock(bs, BLOCK_OP_TYPE_DATAPLANE, job->blocker); 2123718d8abSFam Zheng 2133fc4b10aSFam Zheng job->driver = driver; 2147f0317cfSAlberto Garcia job->id = g_strdup(job_id); 215b6d2e599SKevin Wolf job->blk = blk; 2162f0c9fe6SPaolo Bonzini job->cb = cb; 2172f0c9fe6SPaolo Bonzini job->opaque = opaque; 2185ccac6f1SJohn Snow job->busy = false; 2195ccac6f1SJohn Snow job->paused = true; 2205ccac6f1SJohn Snow job->pause_count = 1; 22118930ba3SFam Zheng job->refcnt = 1; 2222f0c9fe6SPaolo Bonzini bs->job = job; 2232f0c9fe6SPaolo Bonzini 224a7112795SAlberto Garcia QLIST_INSERT_HEAD(&block_jobs, job, job_list); 225a7112795SAlberto Garcia 226463e0be1SStefan Hajnoczi blk_add_aio_context_notifier(blk, block_job_attached_aio_context, 227463e0be1SStefan Hajnoczi block_job_detach_aio_context, job); 228463e0be1SStefan Hajnoczi 2292f0c9fe6SPaolo Bonzini /* Only set speed when necessary to avoid NotSupported error */ 2302f0c9fe6SPaolo Bonzini if (speed != 0) { 2312f0c9fe6SPaolo Bonzini Error *local_err = NULL; 2322f0c9fe6SPaolo Bonzini 2332f0c9fe6SPaolo Bonzini block_job_set_speed(job, speed, &local_err); 23484d18f06SMarkus Armbruster if (local_err) { 23518930ba3SFam Zheng block_job_unref(job); 2362f0c9fe6SPaolo Bonzini error_propagate(errp, local_err); 2372f0c9fe6SPaolo Bonzini return NULL; 2382f0c9fe6SPaolo Bonzini } 2392f0c9fe6SPaolo Bonzini } 2402f0c9fe6SPaolo Bonzini return job; 2412f0c9fe6SPaolo Bonzini } 2422f0c9fe6SPaolo Bonzini 243559b935fSJohn Snow bool block_job_is_internal(BlockJob *job) 244559b935fSJohn Snow { 245559b935fSJohn Snow return (job->id == NULL); 246559b935fSJohn Snow } 247559b935fSJohn Snow 2485ccac6f1SJohn Snow static bool block_job_started(BlockJob *job) 2495ccac6f1SJohn Snow { 2505ccac6f1SJohn Snow return job->co; 2515ccac6f1SJohn Snow } 2525ccac6f1SJohn Snow 2535ccac6f1SJohn Snow void block_job_start(BlockJob *job) 2545ccac6f1SJohn Snow { 2555ccac6f1SJohn Snow assert(job && !block_job_started(job) && job->paused && 2565ccac6f1SJohn Snow !job->busy && job->driver->start); 2575ccac6f1SJohn Snow job->co = qemu_coroutine_create(job->driver->start, job); 2585ccac6f1SJohn Snow if (--job->pause_count == 0) { 2595ccac6f1SJohn Snow job->paused = false; 2605ccac6f1SJohn Snow job->busy = true; 2615ccac6f1SJohn Snow qemu_coroutine_enter(job->co); 2625ccac6f1SJohn Snow } 2635ccac6f1SJohn Snow } 2645ccac6f1SJohn Snow 26518930ba3SFam Zheng void block_job_ref(BlockJob *job) 26697031164STing Wang { 26718930ba3SFam Zheng ++job->refcnt; 26818930ba3SFam Zheng } 26997031164STing Wang 27018930ba3SFam Zheng void block_job_unref(BlockJob *job) 27118930ba3SFam Zheng { 27218930ba3SFam Zheng if (--job->refcnt == 0) { 273b6d2e599SKevin Wolf BlockDriverState *bs = blk_bs(job->blk); 274b6d2e599SKevin Wolf bs->job = NULL; 275*bbc02b90SKevin Wolf block_job_remove_all_bdrv(job); 276463e0be1SStefan Hajnoczi blk_remove_aio_context_notifier(job->blk, 277463e0be1SStefan Hajnoczi block_job_attached_aio_context, 278463e0be1SStefan Hajnoczi block_job_detach_aio_context, job); 279b6d2e599SKevin Wolf blk_unref(job->blk); 28097031164STing Wang error_free(job->blocker); 2818ccb9569SKevin Wolf g_free(job->id); 282a7112795SAlberto Garcia QLIST_REMOVE(job, job_list); 28397031164STing Wang g_free(job); 28497031164STing Wang } 28518930ba3SFam Zheng } 28697031164STing Wang 287c55a832fSFam Zheng static void block_job_completed_single(BlockJob *job) 288c55a832fSFam Zheng { 289c55a832fSFam Zheng if (!job->ret) { 290c55a832fSFam Zheng if (job->driver->commit) { 291c55a832fSFam Zheng job->driver->commit(job); 292c55a832fSFam Zheng } 293c55a832fSFam Zheng } else { 294c55a832fSFam Zheng if (job->driver->abort) { 295c55a832fSFam Zheng job->driver->abort(job); 296c55a832fSFam Zheng } 297c55a832fSFam Zheng } 298e8a40bf7SJohn Snow if (job->driver->clean) { 299e8a40bf7SJohn Snow job->driver->clean(job); 300e8a40bf7SJohn Snow } 3018254b6d9SJohn Snow 3028254b6d9SJohn Snow if (job->cb) { 303c55a832fSFam Zheng job->cb(job->opaque, job->ret); 3048254b6d9SJohn Snow } 3055ccac6f1SJohn Snow 3065ccac6f1SJohn Snow /* Emit events only if we actually started */ 3075ccac6f1SJohn Snow if (block_job_started(job)) { 3088254b6d9SJohn Snow if (block_job_is_cancelled(job)) { 3098254b6d9SJohn Snow block_job_event_cancelled(job); 3108254b6d9SJohn Snow } else { 3118254b6d9SJohn Snow const char *msg = NULL; 3128254b6d9SJohn Snow if (job->ret < 0) { 3138254b6d9SJohn Snow msg = strerror(-job->ret); 3148254b6d9SJohn Snow } 3158254b6d9SJohn Snow block_job_event_completed(job, msg); 3168254b6d9SJohn Snow } 3175ccac6f1SJohn Snow } 3188254b6d9SJohn Snow 319c55a832fSFam Zheng if (job->txn) { 3201e93b9fbSVladimir Sementsov-Ogievskiy QLIST_REMOVE(job, txn_list); 321c55a832fSFam Zheng block_job_txn_unref(job->txn); 322c55a832fSFam Zheng } 323c55a832fSFam Zheng block_job_unref(job); 324c55a832fSFam Zheng } 325c55a832fSFam Zheng 326c55a832fSFam Zheng static void block_job_completed_txn_abort(BlockJob *job) 327c55a832fSFam Zheng { 328c55a832fSFam Zheng AioContext *ctx; 329c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 330c55a832fSFam Zheng BlockJob *other_job, *next; 331c55a832fSFam Zheng 332c55a832fSFam Zheng if (txn->aborting) { 333c55a832fSFam Zheng /* 334c55a832fSFam Zheng * We are cancelled by another job, which will handle everything. 335c55a832fSFam Zheng */ 336c55a832fSFam Zheng return; 337c55a832fSFam Zheng } 338c55a832fSFam Zheng txn->aborting = true; 339c55a832fSFam Zheng /* We are the first failed job. Cancel other jobs. */ 340c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 341b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 342c55a832fSFam Zheng aio_context_acquire(ctx); 343c55a832fSFam Zheng } 344c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 345c55a832fSFam Zheng if (other_job == job || other_job->completed) { 346c55a832fSFam Zheng /* Other jobs are "effectively" cancelled by us, set the status for 347c55a832fSFam Zheng * them; this job, however, may or may not be cancelled, depending 348c55a832fSFam Zheng * on the caller, so leave it. */ 349c55a832fSFam Zheng if (other_job != job) { 350c55a832fSFam Zheng other_job->cancelled = true; 351c55a832fSFam Zheng } 352c55a832fSFam Zheng continue; 353c55a832fSFam Zheng } 354c55a832fSFam Zheng block_job_cancel_sync(other_job); 355c55a832fSFam Zheng assert(other_job->completed); 356c55a832fSFam Zheng } 357c55a832fSFam Zheng QLIST_FOREACH_SAFE(other_job, &txn->jobs, txn_list, next) { 358b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 359c55a832fSFam Zheng block_job_completed_single(other_job); 360c55a832fSFam Zheng aio_context_release(ctx); 361c55a832fSFam Zheng } 362c55a832fSFam Zheng } 363c55a832fSFam Zheng 364c55a832fSFam Zheng static void block_job_completed_txn_success(BlockJob *job) 365c55a832fSFam Zheng { 366c55a832fSFam Zheng AioContext *ctx; 367c55a832fSFam Zheng BlockJobTxn *txn = job->txn; 368c55a832fSFam Zheng BlockJob *other_job, *next; 369c55a832fSFam Zheng /* 370c55a832fSFam Zheng * Successful completion, see if there are other running jobs in this 371c55a832fSFam Zheng * txn. 372c55a832fSFam Zheng */ 373c55a832fSFam Zheng QLIST_FOREACH(other_job, &txn->jobs, txn_list) { 374c55a832fSFam Zheng if (!other_job->completed) { 375c55a832fSFam Zheng return; 376c55a832fSFam Zheng } 377c55a832fSFam Zheng } 378c55a832fSFam Zheng /* We are the last completed job, commit the transaction. */ 379c55a832fSFam Zheng QLIST_FOREACH_SAFE(other_job, &txn->jobs, txn_list, next) { 380b6d2e599SKevin Wolf ctx = blk_get_aio_context(other_job->blk); 381c55a832fSFam Zheng aio_context_acquire(ctx); 382c55a832fSFam Zheng assert(other_job->ret == 0); 383c55a832fSFam Zheng block_job_completed_single(other_job); 384c55a832fSFam Zheng aio_context_release(ctx); 385c55a832fSFam Zheng } 386c55a832fSFam Zheng } 387c55a832fSFam Zheng 38865f46322SPaolo Bonzini void block_job_completed(BlockJob *job, int ret) 3892f0c9fe6SPaolo Bonzini { 390b6d2e599SKevin Wolf assert(blk_bs(job->blk)->job == job); 391a689dbf2SFam Zheng assert(!job->completed); 392a689dbf2SFam Zheng job->completed = true; 393a689dbf2SFam Zheng job->ret = ret; 394c55a832fSFam Zheng if (!job->txn) { 395c55a832fSFam Zheng block_job_completed_single(job); 396c55a832fSFam Zheng } else if (ret < 0 || block_job_is_cancelled(job)) { 397c55a832fSFam Zheng block_job_completed_txn_abort(job); 398c55a832fSFam Zheng } else { 399c55a832fSFam Zheng block_job_completed_txn_success(job); 400c55a832fSFam Zheng } 4012f0c9fe6SPaolo Bonzini } 4022f0c9fe6SPaolo Bonzini 4032f0c9fe6SPaolo Bonzini void block_job_set_speed(BlockJob *job, int64_t speed, Error **errp) 4042f0c9fe6SPaolo Bonzini { 4052f0c9fe6SPaolo Bonzini Error *local_err = NULL; 4062f0c9fe6SPaolo Bonzini 4073fc4b10aSFam Zheng if (!job->driver->set_speed) { 408c6bd8c70SMarkus Armbruster error_setg(errp, QERR_UNSUPPORTED); 4092f0c9fe6SPaolo Bonzini return; 4102f0c9fe6SPaolo Bonzini } 4113fc4b10aSFam Zheng job->driver->set_speed(job, speed, &local_err); 41284d18f06SMarkus Armbruster if (local_err) { 4132f0c9fe6SPaolo Bonzini error_propagate(errp, local_err); 4142f0c9fe6SPaolo Bonzini return; 4152f0c9fe6SPaolo Bonzini } 4162f0c9fe6SPaolo Bonzini 4172f0c9fe6SPaolo Bonzini job->speed = speed; 4182f0c9fe6SPaolo Bonzini } 4192f0c9fe6SPaolo Bonzini 420aeae883bSPaolo Bonzini void block_job_complete(BlockJob *job, Error **errp) 421aeae883bSPaolo Bonzini { 422559b935fSJohn Snow /* Should not be reachable via external interface for internal jobs */ 423559b935fSJohn Snow assert(job->id); 4245ccac6f1SJohn Snow if (job->pause_count || job->cancelled || 4255ccac6f1SJohn Snow !block_job_started(job) || !job->driver->complete) { 4269df229c3SAlberto Garcia error_setg(errp, "The active block job '%s' cannot be completed", 4279df229c3SAlberto Garcia job->id); 428aeae883bSPaolo Bonzini return; 429aeae883bSPaolo Bonzini } 430aeae883bSPaolo Bonzini 4313fc4b10aSFam Zheng job->driver->complete(job, errp); 432aeae883bSPaolo Bonzini } 433aeae883bSPaolo Bonzini 4348acc72a4SPaolo Bonzini void block_job_pause(BlockJob *job) 4352f0c9fe6SPaolo Bonzini { 436751ebd76SFam Zheng job->pause_count++; 4378acc72a4SPaolo Bonzini } 4388acc72a4SPaolo Bonzini 4390df4ba58SJohn Snow void block_job_user_pause(BlockJob *job) 4400df4ba58SJohn Snow { 4410df4ba58SJohn Snow job->user_paused = true; 4420df4ba58SJohn Snow block_job_pause(job); 4430df4ba58SJohn Snow } 4440df4ba58SJohn Snow 445a7f3b7ffSStefan Hajnoczi static bool block_job_should_pause(BlockJob *job) 4468acc72a4SPaolo Bonzini { 447751ebd76SFam Zheng return job->pause_count > 0; 4488acc72a4SPaolo Bonzini } 4498acc72a4SPaolo Bonzini 4500df4ba58SJohn Snow bool block_job_user_paused(BlockJob *job) 4510df4ba58SJohn Snow { 4520df4ba58SJohn Snow return job ? job->user_paused : 0; 4530df4ba58SJohn Snow } 4540df4ba58SJohn Snow 455fc9c0a9cSStefan Hajnoczi void coroutine_fn block_job_pause_point(BlockJob *job) 456fc9c0a9cSStefan Hajnoczi { 4575ccac6f1SJohn Snow assert(job && block_job_started(job)); 4585ccac6f1SJohn Snow 459fc9c0a9cSStefan Hajnoczi if (!block_job_should_pause(job)) { 460fc9c0a9cSStefan Hajnoczi return; 461fc9c0a9cSStefan Hajnoczi } 462fc9c0a9cSStefan Hajnoczi if (block_job_is_cancelled(job)) { 463fc9c0a9cSStefan Hajnoczi return; 464fc9c0a9cSStefan Hajnoczi } 465fc9c0a9cSStefan Hajnoczi 466fc9c0a9cSStefan Hajnoczi if (job->driver->pause) { 467fc9c0a9cSStefan Hajnoczi job->driver->pause(job); 468fc9c0a9cSStefan Hajnoczi } 469fc9c0a9cSStefan Hajnoczi 470fc9c0a9cSStefan Hajnoczi if (block_job_should_pause(job) && !block_job_is_cancelled(job)) { 471fc9c0a9cSStefan Hajnoczi job->paused = true; 472fc9c0a9cSStefan Hajnoczi job->busy = false; 473fc9c0a9cSStefan Hajnoczi qemu_coroutine_yield(); /* wait for block_job_resume() */ 474fc9c0a9cSStefan Hajnoczi job->busy = true; 475fc9c0a9cSStefan Hajnoczi job->paused = false; 476fc9c0a9cSStefan Hajnoczi } 477fc9c0a9cSStefan Hajnoczi 478fc9c0a9cSStefan Hajnoczi if (job->driver->resume) { 479fc9c0a9cSStefan Hajnoczi job->driver->resume(job); 480fc9c0a9cSStefan Hajnoczi } 481fc9c0a9cSStefan Hajnoczi } 482fc9c0a9cSStefan Hajnoczi 4838acc72a4SPaolo Bonzini void block_job_resume(BlockJob *job) 4848acc72a4SPaolo Bonzini { 485751ebd76SFam Zheng assert(job->pause_count > 0); 486751ebd76SFam Zheng job->pause_count--; 487751ebd76SFam Zheng if (job->pause_count) { 488751ebd76SFam Zheng return; 489751ebd76SFam Zheng } 490751ebd76SFam Zheng block_job_enter(job); 491751ebd76SFam Zheng } 492751ebd76SFam Zheng 4930df4ba58SJohn Snow void block_job_user_resume(BlockJob *job) 4940df4ba58SJohn Snow { 4950df4ba58SJohn Snow if (job && job->user_paused && job->pause_count > 0) { 4960df4ba58SJohn Snow job->user_paused = false; 4970df4ba58SJohn Snow block_job_resume(job); 4980df4ba58SJohn Snow } 4990df4ba58SJohn Snow } 5000df4ba58SJohn Snow 501751ebd76SFam Zheng void block_job_enter(BlockJob *job) 502751ebd76SFam Zheng { 5032f0c9fe6SPaolo Bonzini if (job->co && !job->busy) { 5040b8b8753SPaolo Bonzini qemu_coroutine_enter(job->co); 5052f0c9fe6SPaolo Bonzini } 5062f0c9fe6SPaolo Bonzini } 5072f0c9fe6SPaolo Bonzini 5088acc72a4SPaolo Bonzini void block_job_cancel(BlockJob *job) 5098acc72a4SPaolo Bonzini { 5105ccac6f1SJohn Snow if (block_job_started(job)) { 5118acc72a4SPaolo Bonzini job->cancelled = true; 51217bd51f9SStefan Hajnoczi block_job_iostatus_reset(job); 513751ebd76SFam Zheng block_job_enter(job); 5145ccac6f1SJohn Snow } else { 5155ccac6f1SJohn Snow block_job_completed(job, -ECANCELED); 5165ccac6f1SJohn Snow } 5178acc72a4SPaolo Bonzini } 5188acc72a4SPaolo Bonzini 5192f0c9fe6SPaolo Bonzini bool block_job_is_cancelled(BlockJob *job) 5202f0c9fe6SPaolo Bonzini { 5212f0c9fe6SPaolo Bonzini return job->cancelled; 5222f0c9fe6SPaolo Bonzini } 5232f0c9fe6SPaolo Bonzini 52432c81a4aSPaolo Bonzini void block_job_iostatus_reset(BlockJob *job) 52532c81a4aSPaolo Bonzini { 52632c81a4aSPaolo Bonzini job->iostatus = BLOCK_DEVICE_IO_STATUS_OK; 5273fc4b10aSFam Zheng if (job->driver->iostatus_reset) { 5283fc4b10aSFam Zheng job->driver->iostatus_reset(job); 5293bd293c3SPaolo Bonzini } 53032c81a4aSPaolo Bonzini } 53132c81a4aSPaolo Bonzini 532345f9e1bSMax Reitz static int block_job_finish_sync(BlockJob *job, 533345f9e1bSMax Reitz void (*finish)(BlockJob *, Error **errp), 534345f9e1bSMax Reitz Error **errp) 5352f0c9fe6SPaolo Bonzini { 536345f9e1bSMax Reitz Error *local_err = NULL; 53794db6d2dSFam Zheng int ret; 5382f0c9fe6SPaolo Bonzini 539b6d2e599SKevin Wolf assert(blk_bs(job->blk)->job == job); 5402f0c9fe6SPaolo Bonzini 54194db6d2dSFam Zheng block_job_ref(job); 542bae8196dSPaolo Bonzini 543345f9e1bSMax Reitz finish(job, &local_err); 544345f9e1bSMax Reitz if (local_err) { 545345f9e1bSMax Reitz error_propagate(errp, local_err); 54694db6d2dSFam Zheng block_job_unref(job); 547345f9e1bSMax Reitz return -EBUSY; 548345f9e1bSMax Reitz } 549bae8196dSPaolo Bonzini /* block_job_drain calls block_job_enter, and it should be enough to 550bae8196dSPaolo Bonzini * induce progress until the job completes or moves to the main thread. 551bae8196dSPaolo Bonzini */ 552bae8196dSPaolo Bonzini while (!job->deferred_to_main_loop && !job->completed) { 553bae8196dSPaolo Bonzini block_job_drain(job); 554bae8196dSPaolo Bonzini } 55594db6d2dSFam Zheng while (!job->completed) { 556bae8196dSPaolo Bonzini aio_poll(qemu_get_aio_context(), true); 5572f0c9fe6SPaolo Bonzini } 55894db6d2dSFam Zheng ret = (job->cancelled && job->ret == 0) ? -ECANCELED : job->ret; 55994db6d2dSFam Zheng block_job_unref(job); 56094db6d2dSFam Zheng return ret; 5612f0c9fe6SPaolo Bonzini } 5622f0c9fe6SPaolo Bonzini 563345f9e1bSMax Reitz /* A wrapper around block_job_cancel() taking an Error ** parameter so it may be 564345f9e1bSMax Reitz * used with block_job_finish_sync() without the need for (rather nasty) 565345f9e1bSMax Reitz * function pointer casts there. */ 566345f9e1bSMax Reitz static void block_job_cancel_err(BlockJob *job, Error **errp) 567345f9e1bSMax Reitz { 568345f9e1bSMax Reitz block_job_cancel(job); 569345f9e1bSMax Reitz } 570345f9e1bSMax Reitz 571345f9e1bSMax Reitz int block_job_cancel_sync(BlockJob *job) 572345f9e1bSMax Reitz { 573345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_cancel_err, NULL); 574345f9e1bSMax Reitz } 575345f9e1bSMax Reitz 576a1a2af07SKevin Wolf void block_job_cancel_sync_all(void) 577a1a2af07SKevin Wolf { 578a1a2af07SKevin Wolf BlockJob *job; 579a1a2af07SKevin Wolf AioContext *aio_context; 580a1a2af07SKevin Wolf 581a1a2af07SKevin Wolf while ((job = QLIST_FIRST(&block_jobs))) { 582b6d2e599SKevin Wolf aio_context = blk_get_aio_context(job->blk); 583a1a2af07SKevin Wolf aio_context_acquire(aio_context); 584a1a2af07SKevin Wolf block_job_cancel_sync(job); 585a1a2af07SKevin Wolf aio_context_release(aio_context); 586a1a2af07SKevin Wolf } 587a1a2af07SKevin Wolf } 588a1a2af07SKevin Wolf 589345f9e1bSMax Reitz int block_job_complete_sync(BlockJob *job, Error **errp) 590345f9e1bSMax Reitz { 591345f9e1bSMax Reitz return block_job_finish_sync(job, &block_job_complete, errp); 592345f9e1bSMax Reitz } 593345f9e1bSMax Reitz 5947483d1e5SAlex Bligh void block_job_sleep_ns(BlockJob *job, QEMUClockType type, int64_t ns) 5952f0c9fe6SPaolo Bonzini { 5968acc72a4SPaolo Bonzini assert(job->busy); 5978acc72a4SPaolo Bonzini 5982f0c9fe6SPaolo Bonzini /* Check cancellation *before* setting busy = false, too! */ 5998acc72a4SPaolo Bonzini if (block_job_is_cancelled(job)) { 6008acc72a4SPaolo Bonzini return; 6012f0c9fe6SPaolo Bonzini } 6028acc72a4SPaolo Bonzini 6038acc72a4SPaolo Bonzini job->busy = false; 604a7f3b7ffSStefan Hajnoczi if (!block_job_should_pause(job)) { 6050824afdaSAlberto Garcia co_aio_sleep_ns(blk_get_aio_context(job->blk), type, ns); 6060824afdaSAlberto Garcia } 6078acc72a4SPaolo Bonzini job->busy = true; 608fc9c0a9cSStefan Hajnoczi 609fc9c0a9cSStefan Hajnoczi block_job_pause_point(job); 6102f0c9fe6SPaolo Bonzini } 61130e628b7SPaolo Bonzini 612dc71ce45SFam Zheng void block_job_yield(BlockJob *job) 613dc71ce45SFam Zheng { 614dc71ce45SFam Zheng assert(job->busy); 615dc71ce45SFam Zheng 616dc71ce45SFam Zheng /* Check cancellation *before* setting busy = false, too! */ 617dc71ce45SFam Zheng if (block_job_is_cancelled(job)) { 618dc71ce45SFam Zheng return; 619dc71ce45SFam Zheng } 620dc71ce45SFam Zheng 621dc71ce45SFam Zheng job->busy = false; 622fc9c0a9cSStefan Hajnoczi if (!block_job_should_pause(job)) { 623dc71ce45SFam Zheng qemu_coroutine_yield(); 624fc9c0a9cSStefan Hajnoczi } 625dc71ce45SFam Zheng job->busy = true; 626fc9c0a9cSStefan Hajnoczi 627fc9c0a9cSStefan Hajnoczi block_job_pause_point(job); 628dc71ce45SFam Zheng } 629dc71ce45SFam Zheng 630559b935fSJohn Snow BlockJobInfo *block_job_query(BlockJob *job, Error **errp) 63130e628b7SPaolo Bonzini { 632559b935fSJohn Snow BlockJobInfo *info; 633559b935fSJohn Snow 634559b935fSJohn Snow if (block_job_is_internal(job)) { 635559b935fSJohn Snow error_setg(errp, "Cannot query QEMU internal jobs"); 636559b935fSJohn Snow return NULL; 637559b935fSJohn Snow } 638559b935fSJohn Snow info = g_new0(BlockJobInfo, 1); 63979e14bf7SFam Zheng info->type = g_strdup(BlockJobType_lookup[job->driver->job_type]); 6408ccb9569SKevin Wolf info->device = g_strdup(job->id); 64130e628b7SPaolo Bonzini info->len = job->len; 6428d65883fSPaolo Bonzini info->busy = job->busy; 643751ebd76SFam Zheng info->paused = job->pause_count > 0; 64430e628b7SPaolo Bonzini info->offset = job->offset; 64530e628b7SPaolo Bonzini info->speed = job->speed; 64632c81a4aSPaolo Bonzini info->io_status = job->iostatus; 647ef6dbf1eSMax Reitz info->ready = job->ready; 64830e628b7SPaolo Bonzini return info; 64930e628b7SPaolo Bonzini } 65032c81a4aSPaolo Bonzini 65132c81a4aSPaolo Bonzini static void block_job_iostatus_set_err(BlockJob *job, int error) 65232c81a4aSPaolo Bonzini { 65332c81a4aSPaolo Bonzini if (job->iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 65432c81a4aSPaolo Bonzini job->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE : 65532c81a4aSPaolo Bonzini BLOCK_DEVICE_IO_STATUS_FAILED; 65632c81a4aSPaolo Bonzini } 65732c81a4aSPaolo Bonzini } 65832c81a4aSPaolo Bonzini 6598254b6d9SJohn Snow static void block_job_event_cancelled(BlockJob *job) 660a66a2a36SPaolo Bonzini { 661559b935fSJohn Snow if (block_job_is_internal(job)) { 662559b935fSJohn Snow return; 663559b935fSJohn Snow } 664559b935fSJohn Snow 665bcada37bSWenchao Xia qapi_event_send_block_job_cancelled(job->driver->job_type, 6668ccb9569SKevin Wolf job->id, 667a66a2a36SPaolo Bonzini job->len, 668a66a2a36SPaolo Bonzini job->offset, 669bcada37bSWenchao Xia job->speed, 670bcada37bSWenchao Xia &error_abort); 671a66a2a36SPaolo Bonzini } 672a66a2a36SPaolo Bonzini 6738254b6d9SJohn Snow static void block_job_event_completed(BlockJob *job, const char *msg) 674a66a2a36SPaolo Bonzini { 675559b935fSJohn Snow if (block_job_is_internal(job)) { 676559b935fSJohn Snow return; 677559b935fSJohn Snow } 678559b935fSJohn Snow 679bcada37bSWenchao Xia qapi_event_send_block_job_completed(job->driver->job_type, 6808ccb9569SKevin Wolf job->id, 681bcada37bSWenchao Xia job->len, 682bcada37bSWenchao Xia job->offset, 683bcada37bSWenchao Xia job->speed, 684bcada37bSWenchao Xia !!msg, 685bcada37bSWenchao Xia msg, 686bcada37bSWenchao Xia &error_abort); 687bcada37bSWenchao Xia } 688bcada37bSWenchao Xia 689bcada37bSWenchao Xia void block_job_event_ready(BlockJob *job) 690bcada37bSWenchao Xia { 691ef6dbf1eSMax Reitz job->ready = true; 692ef6dbf1eSMax Reitz 693559b935fSJohn Snow if (block_job_is_internal(job)) { 694559b935fSJohn Snow return; 695559b935fSJohn Snow } 696559b935fSJohn Snow 697518848a2SMarkus Armbruster qapi_event_send_block_job_ready(job->driver->job_type, 6988ccb9569SKevin Wolf job->id, 699518848a2SMarkus Armbruster job->len, 700518848a2SMarkus Armbruster job->offset, 701518848a2SMarkus Armbruster job->speed, &error_abort); 702a66a2a36SPaolo Bonzini } 703a66a2a36SPaolo Bonzini 70481e254dcSKevin Wolf BlockErrorAction block_job_error_action(BlockJob *job, BlockdevOnError on_err, 70532c81a4aSPaolo Bonzini int is_read, int error) 70632c81a4aSPaolo Bonzini { 70732c81a4aSPaolo Bonzini BlockErrorAction action; 70832c81a4aSPaolo Bonzini 70932c81a4aSPaolo Bonzini switch (on_err) { 71032c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_ENOSPC: 7118c398252SKevin Wolf case BLOCKDEV_ON_ERROR_AUTO: 712a589569fSWenchao Xia action = (error == ENOSPC) ? 713a589569fSWenchao Xia BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT; 71432c81a4aSPaolo Bonzini break; 71532c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_STOP: 716a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_STOP; 71732c81a4aSPaolo Bonzini break; 71832c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_REPORT: 719a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_REPORT; 72032c81a4aSPaolo Bonzini break; 72132c81a4aSPaolo Bonzini case BLOCKDEV_ON_ERROR_IGNORE: 722a589569fSWenchao Xia action = BLOCK_ERROR_ACTION_IGNORE; 72332c81a4aSPaolo Bonzini break; 72432c81a4aSPaolo Bonzini default: 72532c81a4aSPaolo Bonzini abort(); 72632c81a4aSPaolo Bonzini } 727559b935fSJohn Snow if (!block_job_is_internal(job)) { 7288ccb9569SKevin Wolf qapi_event_send_block_job_error(job->id, 7295a2d2cbdSWenchao Xia is_read ? IO_OPERATION_TYPE_READ : 7305a2d2cbdSWenchao Xia IO_OPERATION_TYPE_WRITE, 7315a2d2cbdSWenchao Xia action, &error_abort); 732559b935fSJohn Snow } 733a589569fSWenchao Xia if (action == BLOCK_ERROR_ACTION_STOP) { 734751ebd76SFam Zheng /* make the pause user visible, which will be resumed from QMP. */ 7350df4ba58SJohn Snow block_job_user_pause(job); 73632c81a4aSPaolo Bonzini block_job_iostatus_set_err(job, error); 73732c81a4aSPaolo Bonzini } 73832c81a4aSPaolo Bonzini return action; 73932c81a4aSPaolo Bonzini } 740dec7d421SStefan Hajnoczi 741dec7d421SStefan Hajnoczi typedef struct { 742dec7d421SStefan Hajnoczi BlockJob *job; 743dec7d421SStefan Hajnoczi AioContext *aio_context; 744dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn; 745dec7d421SStefan Hajnoczi void *opaque; 746dec7d421SStefan Hajnoczi } BlockJobDeferToMainLoopData; 747dec7d421SStefan Hajnoczi 748dec7d421SStefan Hajnoczi static void block_job_defer_to_main_loop_bh(void *opaque) 749dec7d421SStefan Hajnoczi { 750dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = opaque; 751dec7d421SStefan Hajnoczi AioContext *aio_context; 752dec7d421SStefan Hajnoczi 753dec7d421SStefan Hajnoczi /* Prevent race with block_job_defer_to_main_loop() */ 754dec7d421SStefan Hajnoczi aio_context_acquire(data->aio_context); 755dec7d421SStefan Hajnoczi 756dec7d421SStefan Hajnoczi /* Fetch BDS AioContext again, in case it has changed */ 757b6d2e599SKevin Wolf aio_context = blk_get_aio_context(data->job->blk); 758dec7d421SStefan Hajnoczi aio_context_acquire(aio_context); 759dec7d421SStefan Hajnoczi 760794f0141SFam Zheng data->job->deferred_to_main_loop = false; 761dec7d421SStefan Hajnoczi data->fn(data->job, data->opaque); 762dec7d421SStefan Hajnoczi 763dec7d421SStefan Hajnoczi aio_context_release(aio_context); 764dec7d421SStefan Hajnoczi 765dec7d421SStefan Hajnoczi aio_context_release(data->aio_context); 766dec7d421SStefan Hajnoczi 767dec7d421SStefan Hajnoczi g_free(data); 768dec7d421SStefan Hajnoczi } 769dec7d421SStefan Hajnoczi 770dec7d421SStefan Hajnoczi void block_job_defer_to_main_loop(BlockJob *job, 771dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopFn *fn, 772dec7d421SStefan Hajnoczi void *opaque) 773dec7d421SStefan Hajnoczi { 774dec7d421SStefan Hajnoczi BlockJobDeferToMainLoopData *data = g_malloc(sizeof(*data)); 775dec7d421SStefan Hajnoczi data->job = job; 776b6d2e599SKevin Wolf data->aio_context = blk_get_aio_context(job->blk); 777dec7d421SStefan Hajnoczi data->fn = fn; 778dec7d421SStefan Hajnoczi data->opaque = opaque; 779794f0141SFam Zheng job->deferred_to_main_loop = true; 780dec7d421SStefan Hajnoczi 781fffb6e12SPaolo Bonzini aio_bh_schedule_oneshot(qemu_get_aio_context(), 782fffb6e12SPaolo Bonzini block_job_defer_to_main_loop_bh, data); 783dec7d421SStefan Hajnoczi } 784c55a832fSFam Zheng 785c55a832fSFam Zheng BlockJobTxn *block_job_txn_new(void) 786c55a832fSFam Zheng { 787c55a832fSFam Zheng BlockJobTxn *txn = g_new0(BlockJobTxn, 1); 788c55a832fSFam Zheng QLIST_INIT(&txn->jobs); 789c55a832fSFam Zheng txn->refcnt = 1; 790c55a832fSFam Zheng return txn; 791c55a832fSFam Zheng } 792c55a832fSFam Zheng 793c55a832fSFam Zheng static void block_job_txn_ref(BlockJobTxn *txn) 794c55a832fSFam Zheng { 795c55a832fSFam Zheng txn->refcnt++; 796c55a832fSFam Zheng } 797c55a832fSFam Zheng 798c55a832fSFam Zheng void block_job_txn_unref(BlockJobTxn *txn) 799c55a832fSFam Zheng { 800c55a832fSFam Zheng if (txn && --txn->refcnt == 0) { 801c55a832fSFam Zheng g_free(txn); 802c55a832fSFam Zheng } 803c55a832fSFam Zheng } 804c55a832fSFam Zheng 805c55a832fSFam Zheng void block_job_txn_add_job(BlockJobTxn *txn, BlockJob *job) 806c55a832fSFam Zheng { 807c55a832fSFam Zheng if (!txn) { 808c55a832fSFam Zheng return; 809c55a832fSFam Zheng } 810c55a832fSFam Zheng 811c55a832fSFam Zheng assert(!job->txn); 812c55a832fSFam Zheng job->txn = txn; 813c55a832fSFam Zheng 814c55a832fSFam Zheng QLIST_INSERT_HEAD(&txn->jobs, job, txn_list); 815c55a832fSFam Zheng block_job_txn_ref(txn); 816c55a832fSFam Zheng } 817