xref: /qemu/job.c (revision 34dc97b9)
133e9e9bdSKevin Wolf /*
233e9e9bdSKevin Wolf  * Background jobs (long-running operations)
333e9e9bdSKevin Wolf  *
433e9e9bdSKevin Wolf  * Copyright (c) 2011 IBM Corp.
533e9e9bdSKevin Wolf  * Copyright (c) 2012, 2018 Red Hat, Inc.
633e9e9bdSKevin Wolf  *
733e9e9bdSKevin Wolf  * Permission is hereby granted, free of charge, to any person obtaining a copy
833e9e9bdSKevin Wolf  * of this software and associated documentation files (the "Software"), to deal
933e9e9bdSKevin Wolf  * in the Software without restriction, including without limitation the rights
1033e9e9bdSKevin Wolf  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
1133e9e9bdSKevin Wolf  * copies of the Software, and to permit persons to whom the Software is
1233e9e9bdSKevin Wolf  * furnished to do so, subject to the following conditions:
1333e9e9bdSKevin Wolf  *
1433e9e9bdSKevin Wolf  * The above copyright notice and this permission notice shall be included in
1533e9e9bdSKevin Wolf  * all copies or substantial portions of the Software.
1633e9e9bdSKevin Wolf  *
1733e9e9bdSKevin Wolf  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1833e9e9bdSKevin Wolf  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1933e9e9bdSKevin Wolf  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
2033e9e9bdSKevin Wolf  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
2133e9e9bdSKevin Wolf  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
2233e9e9bdSKevin Wolf  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
2333e9e9bdSKevin Wolf  * THE SOFTWARE.
2433e9e9bdSKevin Wolf  */
2533e9e9bdSKevin Wolf 
2633e9e9bdSKevin Wolf #include "qemu/osdep.h"
2733e9e9bdSKevin Wolf #include "qemu-common.h"
2833e9e9bdSKevin Wolf #include "qapi/error.h"
2933e9e9bdSKevin Wolf #include "qemu/job.h"
3033e9e9bdSKevin Wolf #include "qemu/id.h"
311908a559SKevin Wolf #include "qemu/main-loop.h"
32a50c2ab8SKevin Wolf #include "trace-root.h"
331dac83f1SKevin Wolf #include "qapi/qapi-events-job.h"
3433e9e9bdSKevin Wolf 
35e7c1d78bSKevin Wolf static QLIST_HEAD(, Job) jobs = QLIST_HEAD_INITIALIZER(jobs);
36e7c1d78bSKevin Wolf 
37a50c2ab8SKevin Wolf /* Job State Transition Table */
38a50c2ab8SKevin Wolf bool JobSTT[JOB_STATUS__MAX][JOB_STATUS__MAX] = {
39a50c2ab8SKevin Wolf                                     /* U, C, R, P, Y, S, W, D, X, E, N */
40a50c2ab8SKevin Wolf     /* U: */ [JOB_STATUS_UNDEFINED] = {0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0},
41a50c2ab8SKevin Wolf     /* C: */ [JOB_STATUS_CREATED]   = {0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 1},
42a50c2ab8SKevin Wolf     /* R: */ [JOB_STATUS_RUNNING]   = {0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 0},
43a50c2ab8SKevin Wolf     /* P: */ [JOB_STATUS_PAUSED]    = {0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0},
44a50c2ab8SKevin Wolf     /* Y: */ [JOB_STATUS_READY]     = {0, 0, 0, 0, 0, 1, 1, 0, 1, 0, 0},
45a50c2ab8SKevin Wolf     /* S: */ [JOB_STATUS_STANDBY]   = {0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0},
46a50c2ab8SKevin Wolf     /* W: */ [JOB_STATUS_WAITING]   = {0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0},
47a50c2ab8SKevin Wolf     /* D: */ [JOB_STATUS_PENDING]   = {0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0},
48a50c2ab8SKevin Wolf     /* X: */ [JOB_STATUS_ABORTING]  = {0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0},
49a50c2ab8SKevin Wolf     /* E: */ [JOB_STATUS_CONCLUDED] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1},
50a50c2ab8SKevin Wolf     /* N: */ [JOB_STATUS_NULL]      = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
51a50c2ab8SKevin Wolf };
52a50c2ab8SKevin Wolf 
53a50c2ab8SKevin Wolf bool JobVerbTable[JOB_VERB__MAX][JOB_STATUS__MAX] = {
54a50c2ab8SKevin Wolf                                     /* U, C, R, P, Y, S, W, D, X, E, N */
55a50c2ab8SKevin Wolf     [JOB_VERB_CANCEL]               = {0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0},
56a50c2ab8SKevin Wolf     [JOB_VERB_PAUSE]                = {0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0},
57a50c2ab8SKevin Wolf     [JOB_VERB_RESUME]               = {0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0},
58a50c2ab8SKevin Wolf     [JOB_VERB_SET_SPEED]            = {0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0},
59a50c2ab8SKevin Wolf     [JOB_VERB_COMPLETE]             = {0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0},
60a50c2ab8SKevin Wolf     [JOB_VERB_FINALIZE]             = {0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0},
61a50c2ab8SKevin Wolf     [JOB_VERB_DISMISS]              = {0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0},
62a50c2ab8SKevin Wolf };
63a50c2ab8SKevin Wolf 
647eaa8fb5SKevin Wolf /* Transactional group of jobs */
657eaa8fb5SKevin Wolf struct JobTxn {
667eaa8fb5SKevin Wolf 
677eaa8fb5SKevin Wolf     /* Is this txn being cancelled? */
687eaa8fb5SKevin Wolf     bool aborting;
697eaa8fb5SKevin Wolf 
707eaa8fb5SKevin Wolf     /* List of jobs */
717eaa8fb5SKevin Wolf     QLIST_HEAD(, Job) jobs;
727eaa8fb5SKevin Wolf 
737eaa8fb5SKevin Wolf     /* Reference count */
747eaa8fb5SKevin Wolf     int refcnt;
757eaa8fb5SKevin Wolf };
767eaa8fb5SKevin Wolf 
77da01ff7fSKevin Wolf /* Right now, this mutex is only needed to synchronize accesses to job->busy
78da01ff7fSKevin Wolf  * and job->sleep_timer, such as concurrent calls to job_do_yield and
79da01ff7fSKevin Wolf  * job_enter. */
80da01ff7fSKevin Wolf static QemuMutex job_mutex;
81da01ff7fSKevin Wolf 
82da01ff7fSKevin Wolf static void job_lock(void)
83da01ff7fSKevin Wolf {
84da01ff7fSKevin Wolf     qemu_mutex_lock(&job_mutex);
85da01ff7fSKevin Wolf }
86da01ff7fSKevin Wolf 
87da01ff7fSKevin Wolf static void job_unlock(void)
88da01ff7fSKevin Wolf {
89da01ff7fSKevin Wolf     qemu_mutex_unlock(&job_mutex);
90da01ff7fSKevin Wolf }
91da01ff7fSKevin Wolf 
92da01ff7fSKevin Wolf static void __attribute__((__constructor__)) job_init(void)
93da01ff7fSKevin Wolf {
94da01ff7fSKevin Wolf     qemu_mutex_init(&job_mutex);
95da01ff7fSKevin Wolf }
96da01ff7fSKevin Wolf 
977eaa8fb5SKevin Wolf JobTxn *job_txn_new(void)
987eaa8fb5SKevin Wolf {
997eaa8fb5SKevin Wolf     JobTxn *txn = g_new0(JobTxn, 1);
1007eaa8fb5SKevin Wolf     QLIST_INIT(&txn->jobs);
1017eaa8fb5SKevin Wolf     txn->refcnt = 1;
1027eaa8fb5SKevin Wolf     return txn;
1037eaa8fb5SKevin Wolf }
1047eaa8fb5SKevin Wolf 
1057eaa8fb5SKevin Wolf static void job_txn_ref(JobTxn *txn)
1067eaa8fb5SKevin Wolf {
1077eaa8fb5SKevin Wolf     txn->refcnt++;
1087eaa8fb5SKevin Wolf }
1097eaa8fb5SKevin Wolf 
1107eaa8fb5SKevin Wolf void job_txn_unref(JobTxn *txn)
1117eaa8fb5SKevin Wolf {
1127eaa8fb5SKevin Wolf     if (txn && --txn->refcnt == 0) {
1137eaa8fb5SKevin Wolf         g_free(txn);
1147eaa8fb5SKevin Wolf     }
1157eaa8fb5SKevin Wolf }
1167eaa8fb5SKevin Wolf 
1177eaa8fb5SKevin Wolf void job_txn_add_job(JobTxn *txn, Job *job)
1187eaa8fb5SKevin Wolf {
1197eaa8fb5SKevin Wolf     if (!txn) {
1207eaa8fb5SKevin Wolf         return;
1217eaa8fb5SKevin Wolf     }
1227eaa8fb5SKevin Wolf 
1237eaa8fb5SKevin Wolf     assert(!job->txn);
1247eaa8fb5SKevin Wolf     job->txn = txn;
1257eaa8fb5SKevin Wolf 
1267eaa8fb5SKevin Wolf     QLIST_INSERT_HEAD(&txn->jobs, job, txn_list);
1277eaa8fb5SKevin Wolf     job_txn_ref(txn);
1287eaa8fb5SKevin Wolf }
1297eaa8fb5SKevin Wolf 
1307eaa8fb5SKevin Wolf static void job_txn_del_job(Job *job)
1317eaa8fb5SKevin Wolf {
1327eaa8fb5SKevin Wolf     if (job->txn) {
1337eaa8fb5SKevin Wolf         QLIST_REMOVE(job, txn_list);
1347eaa8fb5SKevin Wolf         job_txn_unref(job->txn);
1357eaa8fb5SKevin Wolf         job->txn = NULL;
1367eaa8fb5SKevin Wolf     }
1377eaa8fb5SKevin Wolf }
1387eaa8fb5SKevin Wolf 
13949880165SFam Zheng static int job_txn_apply(JobTxn *txn, int fn(Job *))
1407eaa8fb5SKevin Wolf {
1417eaa8fb5SKevin Wolf     Job *job, *next;
1427eaa8fb5SKevin Wolf     int rc = 0;
1437eaa8fb5SKevin Wolf 
1447eaa8fb5SKevin Wolf     QLIST_FOREACH_SAFE(job, &txn->jobs, txn_list, next) {
1457eaa8fb5SKevin Wolf         rc = fn(job);
1467eaa8fb5SKevin Wolf         if (rc) {
1477eaa8fb5SKevin Wolf             break;
1487eaa8fb5SKevin Wolf         }
1497eaa8fb5SKevin Wolf     }
1507eaa8fb5SKevin Wolf     return rc;
1517eaa8fb5SKevin Wolf }
1527eaa8fb5SKevin Wolf 
153456273b0SKevin Wolf bool job_is_internal(Job *job)
1541dac83f1SKevin Wolf {
1551dac83f1SKevin Wolf     return (job->id == NULL);
1561dac83f1SKevin Wolf }
1571dac83f1SKevin Wolf 
1582e1795b5SKevin Wolf static void job_state_transition(Job *job, JobStatus s1)
159a50c2ab8SKevin Wolf {
160a50c2ab8SKevin Wolf     JobStatus s0 = job->status;
161a50c2ab8SKevin Wolf     assert(s1 >= 0 && s1 <= JOB_STATUS__MAX);
1624ad35181SKevin Wolf     trace_job_state_transition(job, job->ret,
163a50c2ab8SKevin Wolf                                JobSTT[s0][s1] ? "allowed" : "disallowed",
164a50c2ab8SKevin Wolf                                JobStatus_str(s0), JobStatus_str(s1));
165a50c2ab8SKevin Wolf     assert(JobSTT[s0][s1]);
166a50c2ab8SKevin Wolf     job->status = s1;
1671dac83f1SKevin Wolf 
1681dac83f1SKevin Wolf     if (!job_is_internal(job) && s1 != s0) {
1693ab72385SPeter Xu         qapi_event_send_job_status_change(job->id, job->status);
1701dac83f1SKevin Wolf     }
171a50c2ab8SKevin Wolf }
172a50c2ab8SKevin Wolf 
173a50c2ab8SKevin Wolf int job_apply_verb(Job *job, JobVerb verb, Error **errp)
174a50c2ab8SKevin Wolf {
175a50c2ab8SKevin Wolf     JobStatus s0 = job->status;
176a50c2ab8SKevin Wolf     assert(verb >= 0 && verb <= JOB_VERB__MAX);
177a50c2ab8SKevin Wolf     trace_job_apply_verb(job, JobStatus_str(s0), JobVerb_str(verb),
178a50c2ab8SKevin Wolf                          JobVerbTable[verb][s0] ? "allowed" : "prohibited");
179a50c2ab8SKevin Wolf     if (JobVerbTable[verb][s0]) {
180a50c2ab8SKevin Wolf         return 0;
181a50c2ab8SKevin Wolf     }
182a50c2ab8SKevin Wolf     error_setg(errp, "Job '%s' in state '%s' cannot accept command verb '%s'",
183a50c2ab8SKevin Wolf                job->id, JobStatus_str(s0), JobVerb_str(verb));
184a50c2ab8SKevin Wolf     return -EPERM;
185a50c2ab8SKevin Wolf }
186a50c2ab8SKevin Wolf 
187252291eaSKevin Wolf JobType job_type(const Job *job)
188252291eaSKevin Wolf {
189252291eaSKevin Wolf     return job->driver->job_type;
190252291eaSKevin Wolf }
191252291eaSKevin Wolf 
192252291eaSKevin Wolf const char *job_type_str(const Job *job)
193252291eaSKevin Wolf {
194252291eaSKevin Wolf     return JobType_str(job_type(job));
195252291eaSKevin Wolf }
196252291eaSKevin Wolf 
197daa7f2f9SKevin Wolf bool job_is_cancelled(Job *job)
198daa7f2f9SKevin Wolf {
199daa7f2f9SKevin Wolf     return job->cancelled;
200daa7f2f9SKevin Wolf }
201daa7f2f9SKevin Wolf 
202df956ae2SKevin Wolf bool job_is_ready(Job *job)
203df956ae2SKevin Wolf {
204df956ae2SKevin Wolf     switch (job->status) {
205df956ae2SKevin Wolf     case JOB_STATUS_UNDEFINED:
206df956ae2SKevin Wolf     case JOB_STATUS_CREATED:
207df956ae2SKevin Wolf     case JOB_STATUS_RUNNING:
208df956ae2SKevin Wolf     case JOB_STATUS_PAUSED:
209df956ae2SKevin Wolf     case JOB_STATUS_WAITING:
210df956ae2SKevin Wolf     case JOB_STATUS_PENDING:
211df956ae2SKevin Wolf     case JOB_STATUS_ABORTING:
212df956ae2SKevin Wolf     case JOB_STATUS_CONCLUDED:
213df956ae2SKevin Wolf     case JOB_STATUS_NULL:
214df956ae2SKevin Wolf         return false;
215df956ae2SKevin Wolf     case JOB_STATUS_READY:
216df956ae2SKevin Wolf     case JOB_STATUS_STANDBY:
217df956ae2SKevin Wolf         return true;
218df956ae2SKevin Wolf     default:
219df956ae2SKevin Wolf         g_assert_not_reached();
220df956ae2SKevin Wolf     }
221df956ae2SKevin Wolf     return false;
222df956ae2SKevin Wolf }
223df956ae2SKevin Wolf 
224dbe5e6c1SKevin Wolf bool job_is_completed(Job *job)
225dbe5e6c1SKevin Wolf {
226dbe5e6c1SKevin Wolf     switch (job->status) {
227dbe5e6c1SKevin Wolf     case JOB_STATUS_UNDEFINED:
228dbe5e6c1SKevin Wolf     case JOB_STATUS_CREATED:
229dbe5e6c1SKevin Wolf     case JOB_STATUS_RUNNING:
230dbe5e6c1SKevin Wolf     case JOB_STATUS_PAUSED:
231dbe5e6c1SKevin Wolf     case JOB_STATUS_READY:
232dbe5e6c1SKevin Wolf     case JOB_STATUS_STANDBY:
233dbe5e6c1SKevin Wolf         return false;
234dbe5e6c1SKevin Wolf     case JOB_STATUS_WAITING:
235dbe5e6c1SKevin Wolf     case JOB_STATUS_PENDING:
236dbe5e6c1SKevin Wolf     case JOB_STATUS_ABORTING:
237dbe5e6c1SKevin Wolf     case JOB_STATUS_CONCLUDED:
238dbe5e6c1SKevin Wolf     case JOB_STATUS_NULL:
239dbe5e6c1SKevin Wolf         return true;
240dbe5e6c1SKevin Wolf     default:
241dbe5e6c1SKevin Wolf         g_assert_not_reached();
242dbe5e6c1SKevin Wolf     }
243dbe5e6c1SKevin Wolf     return false;
244dbe5e6c1SKevin Wolf }
245dbe5e6c1SKevin Wolf 
2463d70ff53SKevin Wolf static bool job_started(Job *job)
247da01ff7fSKevin Wolf {
248da01ff7fSKevin Wolf     return job->co;
249da01ff7fSKevin Wolf }
250da01ff7fSKevin Wolf 
251198c49ccSKevin Wolf static bool job_should_pause(Job *job)
252da01ff7fSKevin Wolf {
253da01ff7fSKevin Wolf     return job->pause_count > 0;
254da01ff7fSKevin Wolf }
255da01ff7fSKevin Wolf 
256e7c1d78bSKevin Wolf Job *job_next(Job *job)
257e7c1d78bSKevin Wolf {
258e7c1d78bSKevin Wolf     if (!job) {
259e7c1d78bSKevin Wolf         return QLIST_FIRST(&jobs);
260e7c1d78bSKevin Wolf     }
261e7c1d78bSKevin Wolf     return QLIST_NEXT(job, job_list);
262e7c1d78bSKevin Wolf }
263e7c1d78bSKevin Wolf 
264e7c1d78bSKevin Wolf Job *job_get(const char *id)
265e7c1d78bSKevin Wolf {
266e7c1d78bSKevin Wolf     Job *job;
267e7c1d78bSKevin Wolf 
268e7c1d78bSKevin Wolf     QLIST_FOREACH(job, &jobs, job_list) {
269e7c1d78bSKevin Wolf         if (job->id && !strcmp(id, job->id)) {
270e7c1d78bSKevin Wolf             return job;
271e7c1d78bSKevin Wolf         }
272e7c1d78bSKevin Wolf     }
273e7c1d78bSKevin Wolf 
274e7c1d78bSKevin Wolf     return NULL;
275e7c1d78bSKevin Wolf }
276e7c1d78bSKevin Wolf 
2775d43e86eSKevin Wolf static void job_sleep_timer_cb(void *opaque)
2785d43e86eSKevin Wolf {
2795d43e86eSKevin Wolf     Job *job = opaque;
2805d43e86eSKevin Wolf 
2815d43e86eSKevin Wolf     job_enter(job);
2825d43e86eSKevin Wolf }
2835d43e86eSKevin Wolf 
2847eaa8fb5SKevin Wolf void *job_create(const char *job_id, const JobDriver *driver, JobTxn *txn,
2857eaa8fb5SKevin Wolf                  AioContext *ctx, int flags, BlockCompletionFunc *cb,
2867eaa8fb5SKevin Wolf                  void *opaque, Error **errp)
28733e9e9bdSKevin Wolf {
28833e9e9bdSKevin Wolf     Job *job;
28933e9e9bdSKevin Wolf 
29033e9e9bdSKevin Wolf     if (job_id) {
291bb02b65cSKevin Wolf         if (flags & JOB_INTERNAL) {
292bb02b65cSKevin Wolf             error_setg(errp, "Cannot specify job ID for internal job");
293bb02b65cSKevin Wolf             return NULL;
294bb02b65cSKevin Wolf         }
29533e9e9bdSKevin Wolf         if (!id_wellformed(job_id)) {
29633e9e9bdSKevin Wolf             error_setg(errp, "Invalid job ID '%s'", job_id);
29733e9e9bdSKevin Wolf             return NULL;
29833e9e9bdSKevin Wolf         }
299e7c1d78bSKevin Wolf         if (job_get(job_id)) {
300e7c1d78bSKevin Wolf             error_setg(errp, "Job ID '%s' already in use", job_id);
301e7c1d78bSKevin Wolf             return NULL;
302e7c1d78bSKevin Wolf         }
303bb02b65cSKevin Wolf     } else if (!(flags & JOB_INTERNAL)) {
304bb02b65cSKevin Wolf         error_setg(errp, "An explicit job ID is required");
305bb02b65cSKevin Wolf         return NULL;
30633e9e9bdSKevin Wolf     }
30733e9e9bdSKevin Wolf 
30833e9e9bdSKevin Wolf     job = g_malloc0(driver->instance_size);
30933e9e9bdSKevin Wolf     job->driver        = driver;
31033e9e9bdSKevin Wolf     job->id            = g_strdup(job_id);
31180fa2c75SKevin Wolf     job->refcnt        = 1;
31208be6fe2SKevin Wolf     job->aio_context   = ctx;
313da01ff7fSKevin Wolf     job->busy          = false;
314da01ff7fSKevin Wolf     job->paused        = true;
315da01ff7fSKevin Wolf     job->pause_count   = 1;
316bb02b65cSKevin Wolf     job->auto_finalize = !(flags & JOB_MANUAL_FINALIZE);
317bb02b65cSKevin Wolf     job->auto_dismiss  = !(flags & JOB_MANUAL_DISMISS);
3184ad35181SKevin Wolf     job->cb            = cb;
3194ad35181SKevin Wolf     job->opaque        = opaque;
32033e9e9bdSKevin Wolf 
321139a9f02SKevin Wolf     notifier_list_init(&job->on_finalize_cancelled);
322139a9f02SKevin Wolf     notifier_list_init(&job->on_finalize_completed);
323139a9f02SKevin Wolf     notifier_list_init(&job->on_pending);
3242e1795b5SKevin Wolf     notifier_list_init(&job->on_ready);
325139a9f02SKevin Wolf 
326a50c2ab8SKevin Wolf     job_state_transition(job, JOB_STATUS_CREATED);
3275d43e86eSKevin Wolf     aio_timer_init(qemu_get_aio_context(), &job->sleep_timer,
3285d43e86eSKevin Wolf                    QEMU_CLOCK_REALTIME, SCALE_NS,
3295d43e86eSKevin Wolf                    job_sleep_timer_cb, job);
330a50c2ab8SKevin Wolf 
331e7c1d78bSKevin Wolf     QLIST_INSERT_HEAD(&jobs, job, job_list);
332e7c1d78bSKevin Wolf 
3337eaa8fb5SKevin Wolf     /* Single jobs are modeled as single-job transactions for sake of
3347eaa8fb5SKevin Wolf      * consolidating the job management logic */
3357eaa8fb5SKevin Wolf     if (!txn) {
3367eaa8fb5SKevin Wolf         txn = job_txn_new();
3377eaa8fb5SKevin Wolf         job_txn_add_job(txn, job);
3387eaa8fb5SKevin Wolf         job_txn_unref(txn);
3397eaa8fb5SKevin Wolf     } else {
3407eaa8fb5SKevin Wolf         job_txn_add_job(txn, job);
3417eaa8fb5SKevin Wolf     }
3427eaa8fb5SKevin Wolf 
34333e9e9bdSKevin Wolf     return job;
34433e9e9bdSKevin Wolf }
345fd61a701SKevin Wolf 
34680fa2c75SKevin Wolf void job_ref(Job *job)
347fd61a701SKevin Wolf {
34880fa2c75SKevin Wolf     ++job->refcnt;
34980fa2c75SKevin Wolf }
35080fa2c75SKevin Wolf 
35180fa2c75SKevin Wolf void job_unref(Job *job)
35280fa2c75SKevin Wolf {
35380fa2c75SKevin Wolf     if (--job->refcnt == 0) {
35480fa2c75SKevin Wolf         assert(job->status == JOB_STATUS_NULL);
3555d43e86eSKevin Wolf         assert(!timer_pending(&job->sleep_timer));
3567eaa8fb5SKevin Wolf         assert(!job->txn);
35780fa2c75SKevin Wolf 
35880fa2c75SKevin Wolf         if (job->driver->free) {
35980fa2c75SKevin Wolf             job->driver->free(job);
36080fa2c75SKevin Wolf         }
36180fa2c75SKevin Wolf 
362e7c1d78bSKevin Wolf         QLIST_REMOVE(job, job_list);
363e7c1d78bSKevin Wolf 
3643d1f8b07SJohn Snow         error_free(job->err);
365fd61a701SKevin Wolf         g_free(job->id);
366fd61a701SKevin Wolf         g_free(job);
367fd61a701SKevin Wolf     }
36880fa2c75SKevin Wolf }
3691908a559SKevin Wolf 
37030a5c887SKevin Wolf void job_progress_update(Job *job, uint64_t done)
37130a5c887SKevin Wolf {
37230a5c887SKevin Wolf     job->progress_current += done;
37330a5c887SKevin Wolf }
37430a5c887SKevin Wolf 
37530a5c887SKevin Wolf void job_progress_set_remaining(Job *job, uint64_t remaining)
37630a5c887SKevin Wolf {
37730a5c887SKevin Wolf     job->progress_total = job->progress_current + remaining;
37830a5c887SKevin Wolf }
37930a5c887SKevin Wolf 
38062f13600SMax Reitz void job_progress_increase_remaining(Job *job, uint64_t delta)
38162f13600SMax Reitz {
38262f13600SMax Reitz     job->progress_total += delta;
38362f13600SMax Reitz }
38462f13600SMax Reitz 
385139a9f02SKevin Wolf void job_event_cancelled(Job *job)
386139a9f02SKevin Wolf {
387139a9f02SKevin Wolf     notifier_list_notify(&job->on_finalize_cancelled, job);
388139a9f02SKevin Wolf }
389139a9f02SKevin Wolf 
390139a9f02SKevin Wolf void job_event_completed(Job *job)
391139a9f02SKevin Wolf {
392139a9f02SKevin Wolf     notifier_list_notify(&job->on_finalize_completed, job);
393139a9f02SKevin Wolf }
394139a9f02SKevin Wolf 
3957eaa8fb5SKevin Wolf static void job_event_pending(Job *job)
396139a9f02SKevin Wolf {
397139a9f02SKevin Wolf     notifier_list_notify(&job->on_pending, job);
398139a9f02SKevin Wolf }
399139a9f02SKevin Wolf 
4002e1795b5SKevin Wolf static void job_event_ready(Job *job)
4012e1795b5SKevin Wolf {
4022e1795b5SKevin Wolf     notifier_list_notify(&job->on_ready, job);
4032e1795b5SKevin Wolf }
4042e1795b5SKevin Wolf 
405*34dc97b9SKevin Wolf static void job_event_idle(Job *job)
406*34dc97b9SKevin Wolf {
407*34dc97b9SKevin Wolf     notifier_list_notify(&job->on_idle, job);
408*34dc97b9SKevin Wolf }
409*34dc97b9SKevin Wolf 
410da01ff7fSKevin Wolf void job_enter_cond(Job *job, bool(*fn)(Job *job))
411da01ff7fSKevin Wolf {
412da01ff7fSKevin Wolf     if (!job_started(job)) {
413da01ff7fSKevin Wolf         return;
414da01ff7fSKevin Wolf     }
415da01ff7fSKevin Wolf     if (job->deferred_to_main_loop) {
416da01ff7fSKevin Wolf         return;
417da01ff7fSKevin Wolf     }
418da01ff7fSKevin Wolf 
419da01ff7fSKevin Wolf     job_lock();
420da01ff7fSKevin Wolf     if (job->busy) {
421da01ff7fSKevin Wolf         job_unlock();
422da01ff7fSKevin Wolf         return;
423da01ff7fSKevin Wolf     }
424da01ff7fSKevin Wolf 
425da01ff7fSKevin Wolf     if (fn && !fn(job)) {
426da01ff7fSKevin Wolf         job_unlock();
427da01ff7fSKevin Wolf         return;
428da01ff7fSKevin Wolf     }
429da01ff7fSKevin Wolf 
430da01ff7fSKevin Wolf     assert(!job->deferred_to_main_loop);
431da01ff7fSKevin Wolf     timer_del(&job->sleep_timer);
432da01ff7fSKevin Wolf     job->busy = true;
433da01ff7fSKevin Wolf     job_unlock();
434da01ff7fSKevin Wolf     aio_co_wake(job->co);
435da01ff7fSKevin Wolf }
436da01ff7fSKevin Wolf 
4375d43e86eSKevin Wolf void job_enter(Job *job)
4385d43e86eSKevin Wolf {
4395d43e86eSKevin Wolf     job_enter_cond(job, NULL);
4405d43e86eSKevin Wolf }
4415d43e86eSKevin Wolf 
442da01ff7fSKevin Wolf /* Yield, and schedule a timer to reenter the coroutine after @ns nanoseconds.
4433d70ff53SKevin Wolf  * Reentering the job coroutine with job_enter() before the timer has expired
4443d70ff53SKevin Wolf  * is allowed and cancels the timer.
445da01ff7fSKevin Wolf  *
4463d70ff53SKevin Wolf  * If @ns is (uint64_t) -1, no timer is scheduled and job_enter() must be
447da01ff7fSKevin Wolf  * called explicitly. */
448198c49ccSKevin Wolf static void coroutine_fn job_do_yield(Job *job, uint64_t ns)
449da01ff7fSKevin Wolf {
450da01ff7fSKevin Wolf     job_lock();
451da01ff7fSKevin Wolf     if (ns != -1) {
452da01ff7fSKevin Wolf         timer_mod(&job->sleep_timer, ns);
453da01ff7fSKevin Wolf     }
454da01ff7fSKevin Wolf     job->busy = false;
455*34dc97b9SKevin Wolf     job_event_idle(job);
456da01ff7fSKevin Wolf     job_unlock();
457da01ff7fSKevin Wolf     qemu_coroutine_yield();
458da01ff7fSKevin Wolf 
459da01ff7fSKevin Wolf     /* Set by job_enter_cond() before re-entering the coroutine.  */
460da01ff7fSKevin Wolf     assert(job->busy);
461da01ff7fSKevin Wolf }
462da01ff7fSKevin Wolf 
463da01ff7fSKevin Wolf void coroutine_fn job_pause_point(Job *job)
464da01ff7fSKevin Wolf {
465da01ff7fSKevin Wolf     assert(job && job_started(job));
466da01ff7fSKevin Wolf 
467da01ff7fSKevin Wolf     if (!job_should_pause(job)) {
468da01ff7fSKevin Wolf         return;
469da01ff7fSKevin Wolf     }
470da01ff7fSKevin Wolf     if (job_is_cancelled(job)) {
471da01ff7fSKevin Wolf         return;
472da01ff7fSKevin Wolf     }
473da01ff7fSKevin Wolf 
474da01ff7fSKevin Wolf     if (job->driver->pause) {
475da01ff7fSKevin Wolf         job->driver->pause(job);
476da01ff7fSKevin Wolf     }
477da01ff7fSKevin Wolf 
478da01ff7fSKevin Wolf     if (job_should_pause(job) && !job_is_cancelled(job)) {
479da01ff7fSKevin Wolf         JobStatus status = job->status;
480da01ff7fSKevin Wolf         job_state_transition(job, status == JOB_STATUS_READY
481da01ff7fSKevin Wolf                                   ? JOB_STATUS_STANDBY
482da01ff7fSKevin Wolf                                   : JOB_STATUS_PAUSED);
483da01ff7fSKevin Wolf         job->paused = true;
484da01ff7fSKevin Wolf         job_do_yield(job, -1);
485da01ff7fSKevin Wolf         job->paused = false;
486da01ff7fSKevin Wolf         job_state_transition(job, status);
487da01ff7fSKevin Wolf     }
488da01ff7fSKevin Wolf 
489da01ff7fSKevin Wolf     if (job->driver->resume) {
490da01ff7fSKevin Wolf         job->driver->resume(job);
491da01ff7fSKevin Wolf     }
492da01ff7fSKevin Wolf }
493da01ff7fSKevin Wolf 
494198c49ccSKevin Wolf void job_yield(Job *job)
495198c49ccSKevin Wolf {
496198c49ccSKevin Wolf     assert(job->busy);
497198c49ccSKevin Wolf 
498198c49ccSKevin Wolf     /* Check cancellation *before* setting busy = false, too!  */
499198c49ccSKevin Wolf     if (job_is_cancelled(job)) {
500198c49ccSKevin Wolf         return;
501198c49ccSKevin Wolf     }
502198c49ccSKevin Wolf 
503198c49ccSKevin Wolf     if (!job_should_pause(job)) {
504198c49ccSKevin Wolf         job_do_yield(job, -1);
505198c49ccSKevin Wolf     }
506198c49ccSKevin Wolf 
507198c49ccSKevin Wolf     job_pause_point(job);
508198c49ccSKevin Wolf }
509198c49ccSKevin Wolf 
5105d43e86eSKevin Wolf void coroutine_fn job_sleep_ns(Job *job, int64_t ns)
5115d43e86eSKevin Wolf {
5125d43e86eSKevin Wolf     assert(job->busy);
5135d43e86eSKevin Wolf 
5145d43e86eSKevin Wolf     /* Check cancellation *before* setting busy = false, too!  */
5155d43e86eSKevin Wolf     if (job_is_cancelled(job)) {
5165d43e86eSKevin Wolf         return;
5175d43e86eSKevin Wolf     }
5185d43e86eSKevin Wolf 
5195d43e86eSKevin Wolf     if (!job_should_pause(job)) {
5205d43e86eSKevin Wolf         job_do_yield(job, qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + ns);
5215d43e86eSKevin Wolf     }
5225d43e86eSKevin Wolf 
5235d43e86eSKevin Wolf     job_pause_point(job);
5245d43e86eSKevin Wolf }
5255d43e86eSKevin Wolf 
526b69f777dSKevin Wolf void job_drain(Job *job)
527b69f777dSKevin Wolf {
528b69f777dSKevin Wolf     /* If job is !busy this kicks it into the next pause point. */
529b69f777dSKevin Wolf     job_enter(job);
530b69f777dSKevin Wolf 
531b69f777dSKevin Wolf     if (job->driver->drain) {
532b69f777dSKevin Wolf         job->driver->drain(job);
533b69f777dSKevin Wolf     }
534b69f777dSKevin Wolf }
535b69f777dSKevin Wolf 
536b15de828SKevin Wolf /* Assumes the block_job_mutex is held */
537b15de828SKevin Wolf static bool job_timer_not_pending(Job *job)
538b15de828SKevin Wolf {
539b15de828SKevin Wolf     return !timer_pending(&job->sleep_timer);
540b15de828SKevin Wolf }
541b15de828SKevin Wolf 
542b15de828SKevin Wolf void job_pause(Job *job)
543b15de828SKevin Wolf {
544b15de828SKevin Wolf     job->pause_count++;
545b15de828SKevin Wolf }
546b15de828SKevin Wolf 
547b15de828SKevin Wolf void job_resume(Job *job)
548b15de828SKevin Wolf {
549b15de828SKevin Wolf     assert(job->pause_count > 0);
550b15de828SKevin Wolf     job->pause_count--;
551b15de828SKevin Wolf     if (job->pause_count) {
552b15de828SKevin Wolf         return;
553b15de828SKevin Wolf     }
554b15de828SKevin Wolf 
555b15de828SKevin Wolf     /* kick only if no timer is pending */
556b15de828SKevin Wolf     job_enter_cond(job, job_timer_not_pending);
557b15de828SKevin Wolf }
558b15de828SKevin Wolf 
559b15de828SKevin Wolf void job_user_pause(Job *job, Error **errp)
560b15de828SKevin Wolf {
561b15de828SKevin Wolf     if (job_apply_verb(job, JOB_VERB_PAUSE, errp)) {
562b15de828SKevin Wolf         return;
563b15de828SKevin Wolf     }
564b15de828SKevin Wolf     if (job->user_paused) {
565b15de828SKevin Wolf         error_setg(errp, "Job is already paused");
566b15de828SKevin Wolf         return;
567b15de828SKevin Wolf     }
568b15de828SKevin Wolf     job->user_paused = true;
569b15de828SKevin Wolf     job_pause(job);
570b15de828SKevin Wolf }
571b15de828SKevin Wolf 
572b15de828SKevin Wolf bool job_user_paused(Job *job)
573b15de828SKevin Wolf {
574b15de828SKevin Wolf     return job->user_paused;
575b15de828SKevin Wolf }
576b15de828SKevin Wolf 
577b15de828SKevin Wolf void job_user_resume(Job *job, Error **errp)
578b15de828SKevin Wolf {
579b15de828SKevin Wolf     assert(job);
580b15de828SKevin Wolf     if (!job->user_paused || job->pause_count <= 0) {
581b15de828SKevin Wolf         error_setg(errp, "Can't resume a job that was not paused");
582b15de828SKevin Wolf         return;
583b15de828SKevin Wolf     }
584b15de828SKevin Wolf     if (job_apply_verb(job, JOB_VERB_RESUME, errp)) {
585b15de828SKevin Wolf         return;
586b15de828SKevin Wolf     }
587b15de828SKevin Wolf     if (job->driver->user_resume) {
588b15de828SKevin Wolf         job->driver->user_resume(job);
589b15de828SKevin Wolf     }
590b15de828SKevin Wolf     job->user_paused = false;
591b15de828SKevin Wolf     job_resume(job);
592b15de828SKevin Wolf }
593b15de828SKevin Wolf 
5945f9a6a08SKevin Wolf static void job_do_dismiss(Job *job)
5954ad35181SKevin Wolf {
5964ad35181SKevin Wolf     assert(job);
5974ad35181SKevin Wolf     job->busy = false;
5984ad35181SKevin Wolf     job->paused = false;
5994ad35181SKevin Wolf     job->deferred_to_main_loop = true;
6004ad35181SKevin Wolf 
6017eaa8fb5SKevin Wolf     job_txn_del_job(job);
6024ad35181SKevin Wolf 
6034ad35181SKevin Wolf     job_state_transition(job, JOB_STATUS_NULL);
6044ad35181SKevin Wolf     job_unref(job);
6054ad35181SKevin Wolf }
6064ad35181SKevin Wolf 
6075f9a6a08SKevin Wolf void job_dismiss(Job **jobptr, Error **errp)
6085f9a6a08SKevin Wolf {
6095f9a6a08SKevin Wolf     Job *job = *jobptr;
6105f9a6a08SKevin Wolf     /* similarly to _complete, this is QMP-interface only. */
6115f9a6a08SKevin Wolf     assert(job->id);
6125f9a6a08SKevin Wolf     if (job_apply_verb(job, JOB_VERB_DISMISS, errp)) {
6135f9a6a08SKevin Wolf         return;
6145f9a6a08SKevin Wolf     }
6155f9a6a08SKevin Wolf 
6165f9a6a08SKevin Wolf     job_do_dismiss(job);
6175f9a6a08SKevin Wolf     *jobptr = NULL;
6185f9a6a08SKevin Wolf }
6195f9a6a08SKevin Wolf 
6204ad35181SKevin Wolf void job_early_fail(Job *job)
6214ad35181SKevin Wolf {
6224ad35181SKevin Wolf     assert(job->status == JOB_STATUS_CREATED);
6234ad35181SKevin Wolf     job_do_dismiss(job);
6244ad35181SKevin Wolf }
6254ad35181SKevin Wolf 
6264ad35181SKevin Wolf static void job_conclude(Job *job)
6274ad35181SKevin Wolf {
6284ad35181SKevin Wolf     job_state_transition(job, JOB_STATUS_CONCLUDED);
6294ad35181SKevin Wolf     if (job->auto_dismiss || !job_started(job)) {
6304ad35181SKevin Wolf         job_do_dismiss(job);
6314ad35181SKevin Wolf     }
6324ad35181SKevin Wolf }
6334ad35181SKevin Wolf 
6343d70ff53SKevin Wolf static void job_update_rc(Job *job)
6354ad35181SKevin Wolf {
6364ad35181SKevin Wolf     if (!job->ret && job_is_cancelled(job)) {
6374ad35181SKevin Wolf         job->ret = -ECANCELED;
6384ad35181SKevin Wolf     }
6394ad35181SKevin Wolf     if (job->ret) {
6403d1f8b07SJohn Snow         if (!job->err) {
6413d1f8b07SJohn Snow             error_setg(&job->err, "%s", strerror(-job->ret));
6421266c9b9SKevin Wolf         }
6434ad35181SKevin Wolf         job_state_transition(job, JOB_STATUS_ABORTING);
6444ad35181SKevin Wolf     }
6454ad35181SKevin Wolf }
6464ad35181SKevin Wolf 
6474ad35181SKevin Wolf static void job_commit(Job *job)
6484ad35181SKevin Wolf {
6494ad35181SKevin Wolf     assert(!job->ret);
6504ad35181SKevin Wolf     if (job->driver->commit) {
6514ad35181SKevin Wolf         job->driver->commit(job);
6524ad35181SKevin Wolf     }
6534ad35181SKevin Wolf }
6544ad35181SKevin Wolf 
6554ad35181SKevin Wolf static void job_abort(Job *job)
6564ad35181SKevin Wolf {
6574ad35181SKevin Wolf     assert(job->ret);
6584ad35181SKevin Wolf     if (job->driver->abort) {
6594ad35181SKevin Wolf         job->driver->abort(job);
6604ad35181SKevin Wolf     }
6614ad35181SKevin Wolf }
6624ad35181SKevin Wolf 
6634ad35181SKevin Wolf static void job_clean(Job *job)
6644ad35181SKevin Wolf {
6654ad35181SKevin Wolf     if (job->driver->clean) {
6664ad35181SKevin Wolf         job->driver->clean(job);
6674ad35181SKevin Wolf     }
6684ad35181SKevin Wolf }
6694ad35181SKevin Wolf 
6707eaa8fb5SKevin Wolf static int job_finalize_single(Job *job)
6714ad35181SKevin Wolf {
6724ad35181SKevin Wolf     assert(job_is_completed(job));
6734ad35181SKevin Wolf 
6744ad35181SKevin Wolf     /* Ensure abort is called for late-transactional failures */
6754ad35181SKevin Wolf     job_update_rc(job);
6764ad35181SKevin Wolf 
6774ad35181SKevin Wolf     if (!job->ret) {
6784ad35181SKevin Wolf         job_commit(job);
6794ad35181SKevin Wolf     } else {
6804ad35181SKevin Wolf         job_abort(job);
6814ad35181SKevin Wolf     }
6824ad35181SKevin Wolf     job_clean(job);
6834ad35181SKevin Wolf 
6844ad35181SKevin Wolf     if (job->cb) {
6854ad35181SKevin Wolf         job->cb(job->opaque, job->ret);
6864ad35181SKevin Wolf     }
6874ad35181SKevin Wolf 
6884ad35181SKevin Wolf     /* Emit events only if we actually started */
6894ad35181SKevin Wolf     if (job_started(job)) {
6904ad35181SKevin Wolf         if (job_is_cancelled(job)) {
6914ad35181SKevin Wolf             job_event_cancelled(job);
6924ad35181SKevin Wolf         } else {
6934ad35181SKevin Wolf             job_event_completed(job);
6944ad35181SKevin Wolf         }
6954ad35181SKevin Wolf     }
6964ad35181SKevin Wolf 
6977eaa8fb5SKevin Wolf     job_txn_del_job(job);
6984ad35181SKevin Wolf     job_conclude(job);
6994ad35181SKevin Wolf     return 0;
7004ad35181SKevin Wolf }
7014ad35181SKevin Wolf 
7023d70ff53SKevin Wolf static void job_cancel_async(Job *job, bool force)
7037eaa8fb5SKevin Wolf {
7047eaa8fb5SKevin Wolf     if (job->user_paused) {
7057eaa8fb5SKevin Wolf         /* Do not call job_enter here, the caller will handle it.  */
7067eaa8fb5SKevin Wolf         if (job->driver->user_resume) {
7077eaa8fb5SKevin Wolf             job->driver->user_resume(job);
7087eaa8fb5SKevin Wolf         }
709e321c059SJeff Cody         job->user_paused = false;
7107eaa8fb5SKevin Wolf         assert(job->pause_count > 0);
7117eaa8fb5SKevin Wolf         job->pause_count--;
7127eaa8fb5SKevin Wolf     }
7137eaa8fb5SKevin Wolf     job->cancelled = true;
7147eaa8fb5SKevin Wolf     /* To prevent 'force == false' overriding a previous 'force == true' */
7157eaa8fb5SKevin Wolf     job->force_cancel |= force;
7167eaa8fb5SKevin Wolf }
7177eaa8fb5SKevin Wolf 
7183d70ff53SKevin Wolf static void job_completed_txn_abort(Job *job)
7197eaa8fb5SKevin Wolf {
7207eaa8fb5SKevin Wolf     AioContext *ctx;
7217eaa8fb5SKevin Wolf     JobTxn *txn = job->txn;
7227eaa8fb5SKevin Wolf     Job *other_job;
7237eaa8fb5SKevin Wolf 
7247eaa8fb5SKevin Wolf     if (txn->aborting) {
7257eaa8fb5SKevin Wolf         /*
7267eaa8fb5SKevin Wolf          * We are cancelled by another job, which will handle everything.
7277eaa8fb5SKevin Wolf          */
7287eaa8fb5SKevin Wolf         return;
7297eaa8fb5SKevin Wolf     }
7307eaa8fb5SKevin Wolf     txn->aborting = true;
7317eaa8fb5SKevin Wolf     job_txn_ref(txn);
7327eaa8fb5SKevin Wolf 
7337eaa8fb5SKevin Wolf     /* We are the first failed job. Cancel other jobs. */
7347eaa8fb5SKevin Wolf     QLIST_FOREACH(other_job, &txn->jobs, txn_list) {
7357eaa8fb5SKevin Wolf         ctx = other_job->aio_context;
7367eaa8fb5SKevin Wolf         aio_context_acquire(ctx);
7377eaa8fb5SKevin Wolf     }
7387eaa8fb5SKevin Wolf 
7397eaa8fb5SKevin Wolf     /* Other jobs are effectively cancelled by us, set the status for
7407eaa8fb5SKevin Wolf      * them; this job, however, may or may not be cancelled, depending
7417eaa8fb5SKevin Wolf      * on the caller, so leave it. */
7427eaa8fb5SKevin Wolf     QLIST_FOREACH(other_job, &txn->jobs, txn_list) {
7437eaa8fb5SKevin Wolf         if (other_job != job) {
7447eaa8fb5SKevin Wolf             job_cancel_async(other_job, false);
7457eaa8fb5SKevin Wolf         }
7467eaa8fb5SKevin Wolf     }
7477eaa8fb5SKevin Wolf     while (!QLIST_EMPTY(&txn->jobs)) {
7487eaa8fb5SKevin Wolf         other_job = QLIST_FIRST(&txn->jobs);
7497eaa8fb5SKevin Wolf         ctx = other_job->aio_context;
7507eaa8fb5SKevin Wolf         if (!job_is_completed(other_job)) {
7517eaa8fb5SKevin Wolf             assert(job_is_cancelled(other_job));
7527eaa8fb5SKevin Wolf             job_finish_sync(other_job, NULL, NULL);
7537eaa8fb5SKevin Wolf         }
7547eaa8fb5SKevin Wolf         job_finalize_single(other_job);
7557eaa8fb5SKevin Wolf         aio_context_release(ctx);
7567eaa8fb5SKevin Wolf     }
7577eaa8fb5SKevin Wolf 
7587eaa8fb5SKevin Wolf     job_txn_unref(txn);
7597eaa8fb5SKevin Wolf }
7607eaa8fb5SKevin Wolf 
7617eaa8fb5SKevin Wolf static int job_prepare(Job *job)
7627eaa8fb5SKevin Wolf {
7637eaa8fb5SKevin Wolf     if (job->ret == 0 && job->driver->prepare) {
7647eaa8fb5SKevin Wolf         job->ret = job->driver->prepare(job);
7651266c9b9SKevin Wolf         job_update_rc(job);
7667eaa8fb5SKevin Wolf     }
7677eaa8fb5SKevin Wolf     return job->ret;
7687eaa8fb5SKevin Wolf }
7697eaa8fb5SKevin Wolf 
7707eaa8fb5SKevin Wolf static int job_needs_finalize(Job *job)
7717eaa8fb5SKevin Wolf {
7727eaa8fb5SKevin Wolf     return !job->auto_finalize;
7737eaa8fb5SKevin Wolf }
7747eaa8fb5SKevin Wolf 
7757eaa8fb5SKevin Wolf static void job_do_finalize(Job *job)
7767eaa8fb5SKevin Wolf {
7777eaa8fb5SKevin Wolf     int rc;
7787eaa8fb5SKevin Wolf     assert(job && job->txn);
7797eaa8fb5SKevin Wolf 
7807eaa8fb5SKevin Wolf     /* prepare the transaction to complete */
78149880165SFam Zheng     rc = job_txn_apply(job->txn, job_prepare);
7827eaa8fb5SKevin Wolf     if (rc) {
7837eaa8fb5SKevin Wolf         job_completed_txn_abort(job);
7847eaa8fb5SKevin Wolf     } else {
78549880165SFam Zheng         job_txn_apply(job->txn, job_finalize_single);
7867eaa8fb5SKevin Wolf     }
7877eaa8fb5SKevin Wolf }
7887eaa8fb5SKevin Wolf 
7897eaa8fb5SKevin Wolf void job_finalize(Job *job, Error **errp)
7907eaa8fb5SKevin Wolf {
7917eaa8fb5SKevin Wolf     assert(job && job->id);
7927eaa8fb5SKevin Wolf     if (job_apply_verb(job, JOB_VERB_FINALIZE, errp)) {
7937eaa8fb5SKevin Wolf         return;
7947eaa8fb5SKevin Wolf     }
7957eaa8fb5SKevin Wolf     job_do_finalize(job);
7967eaa8fb5SKevin Wolf }
7977eaa8fb5SKevin Wolf 
7987eaa8fb5SKevin Wolf static int job_transition_to_pending(Job *job)
7997eaa8fb5SKevin Wolf {
8007eaa8fb5SKevin Wolf     job_state_transition(job, JOB_STATUS_PENDING);
8017eaa8fb5SKevin Wolf     if (!job->auto_finalize) {
8027eaa8fb5SKevin Wolf         job_event_pending(job);
8037eaa8fb5SKevin Wolf     }
8047eaa8fb5SKevin Wolf     return 0;
8057eaa8fb5SKevin Wolf }
8067eaa8fb5SKevin Wolf 
8072e1795b5SKevin Wolf void job_transition_to_ready(Job *job)
8082e1795b5SKevin Wolf {
8092e1795b5SKevin Wolf     job_state_transition(job, JOB_STATUS_READY);
8102e1795b5SKevin Wolf     job_event_ready(job);
8112e1795b5SKevin Wolf }
8122e1795b5SKevin Wolf 
8133d70ff53SKevin Wolf static void job_completed_txn_success(Job *job)
8147eaa8fb5SKevin Wolf {
8157eaa8fb5SKevin Wolf     JobTxn *txn = job->txn;
8167eaa8fb5SKevin Wolf     Job *other_job;
8177eaa8fb5SKevin Wolf 
8187eaa8fb5SKevin Wolf     job_state_transition(job, JOB_STATUS_WAITING);
8197eaa8fb5SKevin Wolf 
8207eaa8fb5SKevin Wolf     /*
8217eaa8fb5SKevin Wolf      * Successful completion, see if there are other running jobs in this
8227eaa8fb5SKevin Wolf      * txn.
8237eaa8fb5SKevin Wolf      */
8247eaa8fb5SKevin Wolf     QLIST_FOREACH(other_job, &txn->jobs, txn_list) {
8257eaa8fb5SKevin Wolf         if (!job_is_completed(other_job)) {
8267eaa8fb5SKevin Wolf             return;
8277eaa8fb5SKevin Wolf         }
8287eaa8fb5SKevin Wolf         assert(other_job->ret == 0);
8297eaa8fb5SKevin Wolf     }
8307eaa8fb5SKevin Wolf 
83149880165SFam Zheng     job_txn_apply(txn, job_transition_to_pending);
8327eaa8fb5SKevin Wolf 
8337eaa8fb5SKevin Wolf     /* If no jobs need manual finalization, automatically do so */
83449880165SFam Zheng     if (job_txn_apply(txn, job_needs_finalize) == 0) {
8357eaa8fb5SKevin Wolf         job_do_finalize(job);
8367eaa8fb5SKevin Wolf     }
8377eaa8fb5SKevin Wolf }
8387eaa8fb5SKevin Wolf 
839404ff28dSJohn Snow static void job_completed(Job *job)
8403d70ff53SKevin Wolf {
8413d70ff53SKevin Wolf     assert(job && job->txn && !job_is_completed(job));
8421266c9b9SKevin Wolf 
8433d70ff53SKevin Wolf     job_update_rc(job);
844404ff28dSJohn Snow     trace_job_completed(job, job->ret);
8453d70ff53SKevin Wolf     if (job->ret) {
8463d70ff53SKevin Wolf         job_completed_txn_abort(job);
8473d70ff53SKevin Wolf     } else {
8483d70ff53SKevin Wolf         job_completed_txn_success(job);
8493d70ff53SKevin Wolf     }
8503d70ff53SKevin Wolf }
8513d70ff53SKevin Wolf 
852ccbfb331SJohn Snow /** Useful only as a type shim for aio_bh_schedule_oneshot. */
853ccbfb331SJohn Snow static void job_exit(void *opaque)
854ccbfb331SJohn Snow {
855ccbfb331SJohn Snow     Job *job = (Job *)opaque;
856d1756c78SKevin Wolf     AioContext *ctx = job->aio_context;
857d1756c78SKevin Wolf 
858d1756c78SKevin Wolf     aio_context_acquire(ctx);
859ccbfb331SJohn Snow     job_completed(job);
860d1756c78SKevin Wolf     aio_context_release(ctx);
861ccbfb331SJohn Snow }
862ccbfb331SJohn Snow 
863ccbfb331SJohn Snow /**
864ccbfb331SJohn Snow  * All jobs must allow a pause point before entering their job proper. This
865ccbfb331SJohn Snow  * ensures that jobs can be paused prior to being started, then resumed later.
866ccbfb331SJohn Snow  */
867ccbfb331SJohn Snow static void coroutine_fn job_co_entry(void *opaque)
868ccbfb331SJohn Snow {
869ccbfb331SJohn Snow     Job *job = opaque;
870ccbfb331SJohn Snow 
871ccbfb331SJohn Snow     assert(job && job->driver && job->driver->run);
872ccbfb331SJohn Snow     job_pause_point(job);
873ccbfb331SJohn Snow     job->ret = job->driver->run(job, &job->err);
874*34dc97b9SKevin Wolf     job_event_idle(job);
875ccbfb331SJohn Snow     job->deferred_to_main_loop = true;
876ccbfb331SJohn Snow     aio_bh_schedule_oneshot(qemu_get_aio_context(), job_exit, job);
877ccbfb331SJohn Snow }
878ccbfb331SJohn Snow 
879ccbfb331SJohn Snow void job_start(Job *job)
880ccbfb331SJohn Snow {
881ccbfb331SJohn Snow     assert(job && !job_started(job) && job->paused &&
882ccbfb331SJohn Snow            job->driver && job->driver->run);
883ccbfb331SJohn Snow     job->co = qemu_coroutine_create(job_co_entry, job);
884ccbfb331SJohn Snow     job->pause_count--;
885ccbfb331SJohn Snow     job->busy = true;
886ccbfb331SJohn Snow     job->paused = false;
887ccbfb331SJohn Snow     job_state_transition(job, JOB_STATUS_RUNNING);
888ccbfb331SJohn Snow     aio_co_enter(job->aio_context, job->co);
889ccbfb331SJohn Snow }
890ccbfb331SJohn Snow 
8913d70ff53SKevin Wolf void job_cancel(Job *job, bool force)
8923d70ff53SKevin Wolf {
8933d70ff53SKevin Wolf     if (job->status == JOB_STATUS_CONCLUDED) {
8943d70ff53SKevin Wolf         job_do_dismiss(job);
8953d70ff53SKevin Wolf         return;
8963d70ff53SKevin Wolf     }
8973d70ff53SKevin Wolf     job_cancel_async(job, force);
8983d70ff53SKevin Wolf     if (!job_started(job)) {
899404ff28dSJohn Snow         job_completed(job);
9003d70ff53SKevin Wolf     } else if (job->deferred_to_main_loop) {
9013d70ff53SKevin Wolf         job_completed_txn_abort(job);
9023d70ff53SKevin Wolf     } else {
9033d70ff53SKevin Wolf         job_enter(job);
9043d70ff53SKevin Wolf     }
9053d70ff53SKevin Wolf }
9063d70ff53SKevin Wolf 
9073d70ff53SKevin Wolf void job_user_cancel(Job *job, bool force, Error **errp)
9083d70ff53SKevin Wolf {
9093d70ff53SKevin Wolf     if (job_apply_verb(job, JOB_VERB_CANCEL, errp)) {
9103d70ff53SKevin Wolf         return;
9113d70ff53SKevin Wolf     }
9123d70ff53SKevin Wolf     job_cancel(job, force);
9133d70ff53SKevin Wolf }
9143d70ff53SKevin Wolf 
9153d70ff53SKevin Wolf /* A wrapper around job_cancel() taking an Error ** parameter so it may be
9163d70ff53SKevin Wolf  * used with job_finish_sync() without the need for (rather nasty) function
9173d70ff53SKevin Wolf  * pointer casts there. */
9183d70ff53SKevin Wolf static void job_cancel_err(Job *job, Error **errp)
9193d70ff53SKevin Wolf {
9203d70ff53SKevin Wolf     job_cancel(job, false);
9213d70ff53SKevin Wolf }
9223d70ff53SKevin Wolf 
9233d70ff53SKevin Wolf int job_cancel_sync(Job *job)
9243d70ff53SKevin Wolf {
9253d70ff53SKevin Wolf     return job_finish_sync(job, &job_cancel_err, NULL);
9263d70ff53SKevin Wolf }
9273d70ff53SKevin Wolf 
9283d70ff53SKevin Wolf void job_cancel_sync_all(void)
9293d70ff53SKevin Wolf {
9303d70ff53SKevin Wolf     Job *job;
9313d70ff53SKevin Wolf     AioContext *aio_context;
9323d70ff53SKevin Wolf 
9333d70ff53SKevin Wolf     while ((job = job_next(NULL))) {
9343d70ff53SKevin Wolf         aio_context = job->aio_context;
9353d70ff53SKevin Wolf         aio_context_acquire(aio_context);
9363d70ff53SKevin Wolf         job_cancel_sync(job);
9373d70ff53SKevin Wolf         aio_context_release(aio_context);
9383d70ff53SKevin Wolf     }
9393d70ff53SKevin Wolf }
9403d70ff53SKevin Wolf 
9413d70ff53SKevin Wolf int job_complete_sync(Job *job, Error **errp)
9423d70ff53SKevin Wolf {
9433d70ff53SKevin Wolf     return job_finish_sync(job, job_complete, errp);
9443d70ff53SKevin Wolf }
9453d70ff53SKevin Wolf 
9463453d972SKevin Wolf void job_complete(Job *job, Error **errp)
9473453d972SKevin Wolf {
9483453d972SKevin Wolf     /* Should not be reachable via external interface for internal jobs */
9493453d972SKevin Wolf     assert(job->id);
9503453d972SKevin Wolf     if (job_apply_verb(job, JOB_VERB_COMPLETE, errp)) {
9513453d972SKevin Wolf         return;
9523453d972SKevin Wolf     }
9533453d972SKevin Wolf     if (job->pause_count || job_is_cancelled(job) || !job->driver->complete) {
9543453d972SKevin Wolf         error_setg(errp, "The active block job '%s' cannot be completed",
9553453d972SKevin Wolf                    job->id);
9563453d972SKevin Wolf         return;
9573453d972SKevin Wolf     }
9583453d972SKevin Wolf 
9593453d972SKevin Wolf     job->driver->complete(job, errp);
9603453d972SKevin Wolf }
9613453d972SKevin Wolf 
9626a74c075SKevin Wolf int job_finish_sync(Job *job, void (*finish)(Job *, Error **errp), Error **errp)
9636a74c075SKevin Wolf {
9646a74c075SKevin Wolf     Error *local_err = NULL;
9656a74c075SKevin Wolf     int ret;
9666a74c075SKevin Wolf 
9676a74c075SKevin Wolf     job_ref(job);
9686a74c075SKevin Wolf 
9696a74c075SKevin Wolf     if (finish) {
9706a74c075SKevin Wolf         finish(job, &local_err);
9716a74c075SKevin Wolf     }
9726a74c075SKevin Wolf     if (local_err) {
9736a74c075SKevin Wolf         error_propagate(errp, local_err);
9746a74c075SKevin Wolf         job_unref(job);
9756a74c075SKevin Wolf         return -EBUSY;
9766a74c075SKevin Wolf     }
9776a74c075SKevin Wolf     /* job_drain calls job_enter, and it should be enough to induce progress
9786a74c075SKevin Wolf      * until the job completes or moves to the main thread. */
9796a74c075SKevin Wolf     while (!job->deferred_to_main_loop && !job_is_completed(job)) {
9806a74c075SKevin Wolf         job_drain(job);
9816a74c075SKevin Wolf     }
9826a74c075SKevin Wolf     while (!job_is_completed(job)) {
9836a74c075SKevin Wolf         aio_poll(qemu_get_aio_context(), true);
9846a74c075SKevin Wolf     }
9856a74c075SKevin Wolf     ret = (job_is_cancelled(job) && job->ret == 0) ? -ECANCELED : job->ret;
9866a74c075SKevin Wolf     job_unref(job);
9876a74c075SKevin Wolf     return ret;
9886a74c075SKevin Wolf }
989