xref: /openbmc/qemu/job.c (revision e4ec5ad4)
1 /*
2  * Background jobs (long-running operations)
3  *
4  * Copyright (c) 2011 IBM Corp.
5  * Copyright (c) 2012, 2018 Red Hat, Inc.
6  *
7  * Permission is hereby granted, free of charge, to any person obtaining a copy
8  * of this software and associated documentation files (the "Software"), to deal
9  * in the Software without restriction, including without limitation the rights
10  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11  * copies of the Software, and to permit persons to whom the Software is
12  * furnished to do so, subject to the following conditions:
13  *
14  * The above copyright notice and this permission notice shall be included in
15  * all copies or substantial portions of the Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
20  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23  * THE SOFTWARE.
24  */
25 
26 #include "qemu/osdep.h"
27 #include "qapi/error.h"
28 #include "qemu/job.h"
29 #include "qemu/id.h"
30 #include "qemu/main-loop.h"
31 #include "block/aio-wait.h"
32 #include "trace-root.h"
33 #include "qapi/qapi-events-job.h"
34 
35 static QLIST_HEAD(, Job) jobs = QLIST_HEAD_INITIALIZER(jobs);
36 
37 /* Job State Transition Table */
38 bool JobSTT[JOB_STATUS__MAX][JOB_STATUS__MAX] = {
39                                     /* U, C, R, P, Y, S, W, D, X, E, N */
40     /* U: */ [JOB_STATUS_UNDEFINED] = {0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0},
41     /* C: */ [JOB_STATUS_CREATED]   = {0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 1},
42     /* R: */ [JOB_STATUS_RUNNING]   = {0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 0},
43     /* P: */ [JOB_STATUS_PAUSED]    = {0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0},
44     /* Y: */ [JOB_STATUS_READY]     = {0, 0, 0, 0, 0, 1, 1, 0, 1, 0, 0},
45     /* S: */ [JOB_STATUS_STANDBY]   = {0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0},
46     /* W: */ [JOB_STATUS_WAITING]   = {0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0},
47     /* D: */ [JOB_STATUS_PENDING]   = {0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0},
48     /* X: */ [JOB_STATUS_ABORTING]  = {0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0},
49     /* E: */ [JOB_STATUS_CONCLUDED] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1},
50     /* N: */ [JOB_STATUS_NULL]      = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
51 };
52 
53 bool JobVerbTable[JOB_VERB__MAX][JOB_STATUS__MAX] = {
54                                     /* U, C, R, P, Y, S, W, D, X, E, N */
55     [JOB_VERB_CANCEL]               = {0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0},
56     [JOB_VERB_PAUSE]                = {0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0},
57     [JOB_VERB_RESUME]               = {0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0},
58     [JOB_VERB_SET_SPEED]            = {0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0},
59     [JOB_VERB_COMPLETE]             = {0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0},
60     [JOB_VERB_FINALIZE]             = {0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0},
61     [JOB_VERB_DISMISS]              = {0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0},
62 };
63 
64 /* Transactional group of jobs */
65 struct JobTxn {
66 
67     /* Is this txn being cancelled? */
68     bool aborting;
69 
70     /* List of jobs */
71     QLIST_HEAD(, Job) jobs;
72 
73     /* Reference count */
74     int refcnt;
75 };
76 
77 /* Right now, this mutex is only needed to synchronize accesses to job->busy
78  * and job->sleep_timer, such as concurrent calls to job_do_yield and
79  * job_enter. */
80 static QemuMutex job_mutex;
81 
82 static void job_lock(void)
83 {
84     qemu_mutex_lock(&job_mutex);
85 }
86 
87 static void job_unlock(void)
88 {
89     qemu_mutex_unlock(&job_mutex);
90 }
91 
92 static void __attribute__((__constructor__)) job_init(void)
93 {
94     qemu_mutex_init(&job_mutex);
95 }
96 
97 JobTxn *job_txn_new(void)
98 {
99     JobTxn *txn = g_new0(JobTxn, 1);
100     QLIST_INIT(&txn->jobs);
101     txn->refcnt = 1;
102     return txn;
103 }
104 
105 static void job_txn_ref(JobTxn *txn)
106 {
107     txn->refcnt++;
108 }
109 
110 void job_txn_unref(JobTxn *txn)
111 {
112     if (txn && --txn->refcnt == 0) {
113         g_free(txn);
114     }
115 }
116 
117 void job_txn_add_job(JobTxn *txn, Job *job)
118 {
119     if (!txn) {
120         return;
121     }
122 
123     assert(!job->txn);
124     job->txn = txn;
125 
126     QLIST_INSERT_HEAD(&txn->jobs, job, txn_list);
127     job_txn_ref(txn);
128 }
129 
130 static void job_txn_del_job(Job *job)
131 {
132     if (job->txn) {
133         QLIST_REMOVE(job, txn_list);
134         job_txn_unref(job->txn);
135         job->txn = NULL;
136     }
137 }
138 
139 static int job_txn_apply(JobTxn *txn, int fn(Job *))
140 {
141     Job *job, *next;
142     int rc = 0;
143 
144     QLIST_FOREACH_SAFE(job, &txn->jobs, txn_list, next) {
145         rc = fn(job);
146         if (rc) {
147             break;
148         }
149     }
150     return rc;
151 }
152 
153 bool job_is_internal(Job *job)
154 {
155     return (job->id == NULL);
156 }
157 
158 static void job_state_transition(Job *job, JobStatus s1)
159 {
160     JobStatus s0 = job->status;
161     assert(s1 >= 0 && s1 < JOB_STATUS__MAX);
162     trace_job_state_transition(job, job->ret,
163                                JobSTT[s0][s1] ? "allowed" : "disallowed",
164                                JobStatus_str(s0), JobStatus_str(s1));
165     assert(JobSTT[s0][s1]);
166     job->status = s1;
167 
168     if (!job_is_internal(job) && s1 != s0) {
169         qapi_event_send_job_status_change(job->id, job->status);
170     }
171 }
172 
173 int job_apply_verb(Job *job, JobVerb verb, Error **errp)
174 {
175     JobStatus s0 = job->status;
176     assert(verb >= 0 && verb < JOB_VERB__MAX);
177     trace_job_apply_verb(job, JobStatus_str(s0), JobVerb_str(verb),
178                          JobVerbTable[verb][s0] ? "allowed" : "prohibited");
179     if (JobVerbTable[verb][s0]) {
180         return 0;
181     }
182     error_setg(errp, "Job '%s' in state '%s' cannot accept command verb '%s'",
183                job->id, JobStatus_str(s0), JobVerb_str(verb));
184     return -EPERM;
185 }
186 
187 JobType job_type(const Job *job)
188 {
189     return job->driver->job_type;
190 }
191 
192 const char *job_type_str(const Job *job)
193 {
194     return JobType_str(job_type(job));
195 }
196 
197 bool job_is_cancelled(Job *job)
198 {
199     return job->cancelled;
200 }
201 
202 bool job_is_ready(Job *job)
203 {
204     switch (job->status) {
205     case JOB_STATUS_UNDEFINED:
206     case JOB_STATUS_CREATED:
207     case JOB_STATUS_RUNNING:
208     case JOB_STATUS_PAUSED:
209     case JOB_STATUS_WAITING:
210     case JOB_STATUS_PENDING:
211     case JOB_STATUS_ABORTING:
212     case JOB_STATUS_CONCLUDED:
213     case JOB_STATUS_NULL:
214         return false;
215     case JOB_STATUS_READY:
216     case JOB_STATUS_STANDBY:
217         return true;
218     default:
219         g_assert_not_reached();
220     }
221     return false;
222 }
223 
224 bool job_is_completed(Job *job)
225 {
226     switch (job->status) {
227     case JOB_STATUS_UNDEFINED:
228     case JOB_STATUS_CREATED:
229     case JOB_STATUS_RUNNING:
230     case JOB_STATUS_PAUSED:
231     case JOB_STATUS_READY:
232     case JOB_STATUS_STANDBY:
233         return false;
234     case JOB_STATUS_WAITING:
235     case JOB_STATUS_PENDING:
236     case JOB_STATUS_ABORTING:
237     case JOB_STATUS_CONCLUDED:
238     case JOB_STATUS_NULL:
239         return true;
240     default:
241         g_assert_not_reached();
242     }
243     return false;
244 }
245 
246 static bool job_started(Job *job)
247 {
248     return job->co;
249 }
250 
251 static bool job_should_pause(Job *job)
252 {
253     return job->pause_count > 0;
254 }
255 
256 Job *job_next(Job *job)
257 {
258     if (!job) {
259         return QLIST_FIRST(&jobs);
260     }
261     return QLIST_NEXT(job, job_list);
262 }
263 
264 Job *job_get(const char *id)
265 {
266     Job *job;
267 
268     QLIST_FOREACH(job, &jobs, job_list) {
269         if (job->id && !strcmp(id, job->id)) {
270             return job;
271         }
272     }
273 
274     return NULL;
275 }
276 
277 static void job_sleep_timer_cb(void *opaque)
278 {
279     Job *job = opaque;
280 
281     job_enter(job);
282 }
283 
284 void *job_create(const char *job_id, const JobDriver *driver, JobTxn *txn,
285                  AioContext *ctx, int flags, BlockCompletionFunc *cb,
286                  void *opaque, Error **errp)
287 {
288     Job *job;
289 
290     if (job_id) {
291         if (flags & JOB_INTERNAL) {
292             error_setg(errp, "Cannot specify job ID for internal job");
293             return NULL;
294         }
295         if (!id_wellformed(job_id)) {
296             error_setg(errp, "Invalid job ID '%s'", job_id);
297             return NULL;
298         }
299         if (job_get(job_id)) {
300             error_setg(errp, "Job ID '%s' already in use", job_id);
301             return NULL;
302         }
303     } else if (!(flags & JOB_INTERNAL)) {
304         error_setg(errp, "An explicit job ID is required");
305         return NULL;
306     }
307 
308     job = g_malloc0(driver->instance_size);
309     job->driver        = driver;
310     job->id            = g_strdup(job_id);
311     job->refcnt        = 1;
312     job->aio_context   = ctx;
313     job->busy          = false;
314     job->paused        = true;
315     job->pause_count   = 1;
316     job->auto_finalize = !(flags & JOB_MANUAL_FINALIZE);
317     job->auto_dismiss  = !(flags & JOB_MANUAL_DISMISS);
318     job->cb            = cb;
319     job->opaque        = opaque;
320 
321     notifier_list_init(&job->on_finalize_cancelled);
322     notifier_list_init(&job->on_finalize_completed);
323     notifier_list_init(&job->on_pending);
324     notifier_list_init(&job->on_ready);
325 
326     job_state_transition(job, JOB_STATUS_CREATED);
327     aio_timer_init(qemu_get_aio_context(), &job->sleep_timer,
328                    QEMU_CLOCK_REALTIME, SCALE_NS,
329                    job_sleep_timer_cb, job);
330 
331     QLIST_INSERT_HEAD(&jobs, job, job_list);
332 
333     /* Single jobs are modeled as single-job transactions for sake of
334      * consolidating the job management logic */
335     if (!txn) {
336         txn = job_txn_new();
337         job_txn_add_job(txn, job);
338         job_txn_unref(txn);
339     } else {
340         job_txn_add_job(txn, job);
341     }
342 
343     return job;
344 }
345 
346 void job_ref(Job *job)
347 {
348     ++job->refcnt;
349 }
350 
351 void job_unref(Job *job)
352 {
353     if (--job->refcnt == 0) {
354         assert(job->status == JOB_STATUS_NULL);
355         assert(!timer_pending(&job->sleep_timer));
356         assert(!job->txn);
357 
358         if (job->driver->free) {
359             job->driver->free(job);
360         }
361 
362         QLIST_REMOVE(job, job_list);
363 
364         error_free(job->err);
365         g_free(job->id);
366         g_free(job);
367     }
368 }
369 
370 void job_progress_update(Job *job, uint64_t done)
371 {
372     job->progress_current += done;
373 }
374 
375 void job_progress_set_remaining(Job *job, uint64_t remaining)
376 {
377     job->progress_total = job->progress_current + remaining;
378 }
379 
380 void job_progress_increase_remaining(Job *job, uint64_t delta)
381 {
382     job->progress_total += delta;
383 }
384 
385 void job_event_cancelled(Job *job)
386 {
387     notifier_list_notify(&job->on_finalize_cancelled, job);
388 }
389 
390 void job_event_completed(Job *job)
391 {
392     notifier_list_notify(&job->on_finalize_completed, job);
393 }
394 
395 static void job_event_pending(Job *job)
396 {
397     notifier_list_notify(&job->on_pending, job);
398 }
399 
400 static void job_event_ready(Job *job)
401 {
402     notifier_list_notify(&job->on_ready, job);
403 }
404 
405 static void job_event_idle(Job *job)
406 {
407     notifier_list_notify(&job->on_idle, job);
408 }
409 
410 void job_enter_cond(Job *job, bool(*fn)(Job *job))
411 {
412     if (!job_started(job)) {
413         return;
414     }
415     if (job->deferred_to_main_loop) {
416         return;
417     }
418 
419     job_lock();
420     if (job->busy) {
421         job_unlock();
422         return;
423     }
424 
425     if (fn && !fn(job)) {
426         job_unlock();
427         return;
428     }
429 
430     assert(!job->deferred_to_main_loop);
431     timer_del(&job->sleep_timer);
432     job->busy = true;
433     job_unlock();
434     aio_co_enter(job->aio_context, job->co);
435 }
436 
437 void job_enter(Job *job)
438 {
439     job_enter_cond(job, NULL);
440 }
441 
442 /* Yield, and schedule a timer to reenter the coroutine after @ns nanoseconds.
443  * Reentering the job coroutine with job_enter() before the timer has expired
444  * is allowed and cancels the timer.
445  *
446  * If @ns is (uint64_t) -1, no timer is scheduled and job_enter() must be
447  * called explicitly. */
448 static void coroutine_fn job_do_yield(Job *job, uint64_t ns)
449 {
450     job_lock();
451     if (ns != -1) {
452         timer_mod(&job->sleep_timer, ns);
453     }
454     job->busy = false;
455     job_event_idle(job);
456     job_unlock();
457     qemu_coroutine_yield();
458 
459     /* Set by job_enter_cond() before re-entering the coroutine.  */
460     assert(job->busy);
461 }
462 
463 void coroutine_fn job_pause_point(Job *job)
464 {
465     assert(job && job_started(job));
466 
467     if (!job_should_pause(job)) {
468         return;
469     }
470     if (job_is_cancelled(job)) {
471         return;
472     }
473 
474     if (job->driver->pause) {
475         job->driver->pause(job);
476     }
477 
478     if (job_should_pause(job) && !job_is_cancelled(job)) {
479         JobStatus status = job->status;
480         job_state_transition(job, status == JOB_STATUS_READY
481                                   ? JOB_STATUS_STANDBY
482                                   : JOB_STATUS_PAUSED);
483         job->paused = true;
484         job_do_yield(job, -1);
485         job->paused = false;
486         job_state_transition(job, status);
487     }
488 
489     if (job->driver->resume) {
490         job->driver->resume(job);
491     }
492 }
493 
494 void job_yield(Job *job)
495 {
496     assert(job->busy);
497 
498     /* Check cancellation *before* setting busy = false, too!  */
499     if (job_is_cancelled(job)) {
500         return;
501     }
502 
503     if (!job_should_pause(job)) {
504         job_do_yield(job, -1);
505     }
506 
507     job_pause_point(job);
508 }
509 
510 void coroutine_fn job_sleep_ns(Job *job, int64_t ns)
511 {
512     assert(job->busy);
513 
514     /* Check cancellation *before* setting busy = false, too!  */
515     if (job_is_cancelled(job)) {
516         return;
517     }
518 
519     if (!job_should_pause(job)) {
520         job_do_yield(job, qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + ns);
521     }
522 
523     job_pause_point(job);
524 }
525 
526 /* Assumes the block_job_mutex is held */
527 static bool job_timer_not_pending(Job *job)
528 {
529     return !timer_pending(&job->sleep_timer);
530 }
531 
532 void job_pause(Job *job)
533 {
534     job->pause_count++;
535 }
536 
537 void job_resume(Job *job)
538 {
539     assert(job->pause_count > 0);
540     job->pause_count--;
541     if (job->pause_count) {
542         return;
543     }
544 
545     /* kick only if no timer is pending */
546     job_enter_cond(job, job_timer_not_pending);
547 }
548 
549 void job_user_pause(Job *job, Error **errp)
550 {
551     if (job_apply_verb(job, JOB_VERB_PAUSE, errp)) {
552         return;
553     }
554     if (job->user_paused) {
555         error_setg(errp, "Job is already paused");
556         return;
557     }
558     job->user_paused = true;
559     job_pause(job);
560 }
561 
562 bool job_user_paused(Job *job)
563 {
564     return job->user_paused;
565 }
566 
567 void job_user_resume(Job *job, Error **errp)
568 {
569     assert(job);
570     if (!job->user_paused || job->pause_count <= 0) {
571         error_setg(errp, "Can't resume a job that was not paused");
572         return;
573     }
574     if (job_apply_verb(job, JOB_VERB_RESUME, errp)) {
575         return;
576     }
577     if (job->driver->user_resume) {
578         job->driver->user_resume(job);
579     }
580     job->user_paused = false;
581     job_resume(job);
582 }
583 
584 static void job_do_dismiss(Job *job)
585 {
586     assert(job);
587     job->busy = false;
588     job->paused = false;
589     job->deferred_to_main_loop = true;
590 
591     job_txn_del_job(job);
592 
593     job_state_transition(job, JOB_STATUS_NULL);
594     job_unref(job);
595 }
596 
597 void job_dismiss(Job **jobptr, Error **errp)
598 {
599     Job *job = *jobptr;
600     /* similarly to _complete, this is QMP-interface only. */
601     assert(job->id);
602     if (job_apply_verb(job, JOB_VERB_DISMISS, errp)) {
603         return;
604     }
605 
606     job_do_dismiss(job);
607     *jobptr = NULL;
608 }
609 
610 void job_early_fail(Job *job)
611 {
612     assert(job->status == JOB_STATUS_CREATED);
613     job_do_dismiss(job);
614 }
615 
616 static void job_conclude(Job *job)
617 {
618     job_state_transition(job, JOB_STATUS_CONCLUDED);
619     if (job->auto_dismiss || !job_started(job)) {
620         job_do_dismiss(job);
621     }
622 }
623 
624 static void job_update_rc(Job *job)
625 {
626     if (!job->ret && job_is_cancelled(job)) {
627         job->ret = -ECANCELED;
628     }
629     if (job->ret) {
630         if (!job->err) {
631             error_setg(&job->err, "%s", strerror(-job->ret));
632         }
633         job_state_transition(job, JOB_STATUS_ABORTING);
634     }
635 }
636 
637 static void job_commit(Job *job)
638 {
639     assert(!job->ret);
640     if (job->driver->commit) {
641         job->driver->commit(job);
642     }
643 }
644 
645 static void job_abort(Job *job)
646 {
647     assert(job->ret);
648     if (job->driver->abort) {
649         job->driver->abort(job);
650     }
651 }
652 
653 static void job_clean(Job *job)
654 {
655     if (job->driver->clean) {
656         job->driver->clean(job);
657     }
658 }
659 
660 static int job_finalize_single(Job *job)
661 {
662     assert(job_is_completed(job));
663 
664     /* Ensure abort is called for late-transactional failures */
665     job_update_rc(job);
666 
667     if (!job->ret) {
668         job_commit(job);
669     } else {
670         job_abort(job);
671     }
672     job_clean(job);
673 
674     if (job->cb) {
675         job->cb(job->opaque, job->ret);
676     }
677 
678     /* Emit events only if we actually started */
679     if (job_started(job)) {
680         if (job_is_cancelled(job)) {
681             job_event_cancelled(job);
682         } else {
683             job_event_completed(job);
684         }
685     }
686 
687     job_txn_del_job(job);
688     job_conclude(job);
689     return 0;
690 }
691 
692 static void job_cancel_async(Job *job, bool force)
693 {
694     if (job->user_paused) {
695         /* Do not call job_enter here, the caller will handle it.  */
696         if (job->driver->user_resume) {
697             job->driver->user_resume(job);
698         }
699         job->user_paused = false;
700         assert(job->pause_count > 0);
701         job->pause_count--;
702     }
703     job->cancelled = true;
704     /* To prevent 'force == false' overriding a previous 'force == true' */
705     job->force_cancel |= force;
706 }
707 
708 static void job_completed_txn_abort(Job *job)
709 {
710     AioContext *outer_ctx = job->aio_context;
711     AioContext *ctx;
712     JobTxn *txn = job->txn;
713     Job *other_job;
714 
715     if (txn->aborting) {
716         /*
717          * We are cancelled by another job, which will handle everything.
718          */
719         return;
720     }
721     txn->aborting = true;
722     job_txn_ref(txn);
723 
724     /* We can only hold the single job's AioContext lock while calling
725      * job_finalize_single() because the finalization callbacks can involve
726      * calls of AIO_WAIT_WHILE(), which could deadlock otherwise. */
727     aio_context_release(outer_ctx);
728 
729     /* Other jobs are effectively cancelled by us, set the status for
730      * them; this job, however, may or may not be cancelled, depending
731      * on the caller, so leave it. */
732     QLIST_FOREACH(other_job, &txn->jobs, txn_list) {
733         if (other_job != job) {
734             ctx = other_job->aio_context;
735             aio_context_acquire(ctx);
736             job_cancel_async(other_job, false);
737             aio_context_release(ctx);
738         }
739     }
740     while (!QLIST_EMPTY(&txn->jobs)) {
741         other_job = QLIST_FIRST(&txn->jobs);
742         ctx = other_job->aio_context;
743         aio_context_acquire(ctx);
744         if (!job_is_completed(other_job)) {
745             assert(job_is_cancelled(other_job));
746             job_finish_sync(other_job, NULL, NULL);
747         }
748         job_finalize_single(other_job);
749         aio_context_release(ctx);
750     }
751 
752     aio_context_acquire(outer_ctx);
753 
754     job_txn_unref(txn);
755 }
756 
757 static int job_prepare(Job *job)
758 {
759     if (job->ret == 0 && job->driver->prepare) {
760         job->ret = job->driver->prepare(job);
761         job_update_rc(job);
762     }
763     return job->ret;
764 }
765 
766 static int job_needs_finalize(Job *job)
767 {
768     return !job->auto_finalize;
769 }
770 
771 static void job_do_finalize(Job *job)
772 {
773     int rc;
774     assert(job && job->txn);
775 
776     /* prepare the transaction to complete */
777     rc = job_txn_apply(job->txn, job_prepare);
778     if (rc) {
779         job_completed_txn_abort(job);
780     } else {
781         job_txn_apply(job->txn, job_finalize_single);
782     }
783 }
784 
785 void job_finalize(Job *job, Error **errp)
786 {
787     assert(job && job->id);
788     if (job_apply_verb(job, JOB_VERB_FINALIZE, errp)) {
789         return;
790     }
791     job_do_finalize(job);
792 }
793 
794 static int job_transition_to_pending(Job *job)
795 {
796     job_state_transition(job, JOB_STATUS_PENDING);
797     if (!job->auto_finalize) {
798         job_event_pending(job);
799     }
800     return 0;
801 }
802 
803 void job_transition_to_ready(Job *job)
804 {
805     job_state_transition(job, JOB_STATUS_READY);
806     job_event_ready(job);
807 }
808 
809 static void job_completed_txn_success(Job *job)
810 {
811     JobTxn *txn = job->txn;
812     Job *other_job;
813 
814     job_state_transition(job, JOB_STATUS_WAITING);
815 
816     /*
817      * Successful completion, see if there are other running jobs in this
818      * txn.
819      */
820     QLIST_FOREACH(other_job, &txn->jobs, txn_list) {
821         if (!job_is_completed(other_job)) {
822             return;
823         }
824         assert(other_job->ret == 0);
825     }
826 
827     job_txn_apply(txn, job_transition_to_pending);
828 
829     /* If no jobs need manual finalization, automatically do so */
830     if (job_txn_apply(txn, job_needs_finalize) == 0) {
831         job_do_finalize(job);
832     }
833 }
834 
835 static void job_completed(Job *job)
836 {
837     assert(job && job->txn && !job_is_completed(job));
838 
839     job_update_rc(job);
840     trace_job_completed(job, job->ret);
841     if (job->ret) {
842         job_completed_txn_abort(job);
843     } else {
844         job_completed_txn_success(job);
845     }
846 }
847 
848 /** Useful only as a type shim for aio_bh_schedule_oneshot. */
849 static void job_exit(void *opaque)
850 {
851     Job *job = (Job *)opaque;
852     AioContext *ctx = job->aio_context;
853 
854     aio_context_acquire(ctx);
855 
856     /* This is a lie, we're not quiescent, but still doing the completion
857      * callbacks. However, completion callbacks tend to involve operations that
858      * drain block nodes, and if .drained_poll still returned true, we would
859      * deadlock. */
860     job->busy = false;
861     job_event_idle(job);
862 
863     job_completed(job);
864 
865     aio_context_release(ctx);
866 }
867 
868 /**
869  * All jobs must allow a pause point before entering their job proper. This
870  * ensures that jobs can be paused prior to being started, then resumed later.
871  */
872 static void coroutine_fn job_co_entry(void *opaque)
873 {
874     Job *job = opaque;
875 
876     assert(job && job->driver && job->driver->run);
877     job_pause_point(job);
878     job->ret = job->driver->run(job, &job->err);
879     job->deferred_to_main_loop = true;
880     job->busy = true;
881     aio_bh_schedule_oneshot(qemu_get_aio_context(), job_exit, job);
882 }
883 
884 void job_start(Job *job)
885 {
886     assert(job && !job_started(job) && job->paused &&
887            job->driver && job->driver->run);
888     job->co = qemu_coroutine_create(job_co_entry, job);
889     job->pause_count--;
890     job->busy = true;
891     job->paused = false;
892     job_state_transition(job, JOB_STATUS_RUNNING);
893     aio_co_enter(job->aio_context, job->co);
894 }
895 
896 void job_cancel(Job *job, bool force)
897 {
898     if (job->status == JOB_STATUS_CONCLUDED) {
899         job_do_dismiss(job);
900         return;
901     }
902     job_cancel_async(job, force);
903     if (!job_started(job)) {
904         job_completed(job);
905     } else if (job->deferred_to_main_loop) {
906         job_completed_txn_abort(job);
907     } else {
908         job_enter(job);
909     }
910 }
911 
912 void job_user_cancel(Job *job, bool force, Error **errp)
913 {
914     if (job_apply_verb(job, JOB_VERB_CANCEL, errp)) {
915         return;
916     }
917     job_cancel(job, force);
918 }
919 
920 /* A wrapper around job_cancel() taking an Error ** parameter so it may be
921  * used with job_finish_sync() without the need for (rather nasty) function
922  * pointer casts there. */
923 static void job_cancel_err(Job *job, Error **errp)
924 {
925     job_cancel(job, false);
926 }
927 
928 int job_cancel_sync(Job *job)
929 {
930     return job_finish_sync(job, &job_cancel_err, NULL);
931 }
932 
933 void job_cancel_sync_all(void)
934 {
935     Job *job;
936     AioContext *aio_context;
937 
938     while ((job = job_next(NULL))) {
939         aio_context = job->aio_context;
940         aio_context_acquire(aio_context);
941         job_cancel_sync(job);
942         aio_context_release(aio_context);
943     }
944 }
945 
946 int job_complete_sync(Job *job, Error **errp)
947 {
948     return job_finish_sync(job, job_complete, errp);
949 }
950 
951 void job_complete(Job *job, Error **errp)
952 {
953     /* Should not be reachable via external interface for internal jobs */
954     assert(job->id);
955     if (job_apply_verb(job, JOB_VERB_COMPLETE, errp)) {
956         return;
957     }
958     if (job->pause_count || job_is_cancelled(job) || !job->driver->complete) {
959         error_setg(errp, "The active block job '%s' cannot be completed",
960                    job->id);
961         return;
962     }
963 
964     job->driver->complete(job, errp);
965 }
966 
967 int job_finish_sync(Job *job, void (*finish)(Job *, Error **errp), Error **errp)
968 {
969     Error *local_err = NULL;
970     int ret;
971 
972     job_ref(job);
973 
974     if (finish) {
975         finish(job, &local_err);
976     }
977     if (local_err) {
978         error_propagate(errp, local_err);
979         job_unref(job);
980         return -EBUSY;
981     }
982 
983     AIO_WAIT_WHILE(job->aio_context,
984                    (job_enter(job), !job_is_completed(job)));
985 
986     ret = (job_is_cancelled(job) && job->ret == 0) ? -ECANCELED : job->ret;
987     job_unref(job);
988     return ret;
989 }
990