]>
Commit | Line | Data |
---|---|---|
e1878eb5 VSO |
1 | /* |
2 | * Helper functionality for distributing a fixed total amount of | |
3 | * an abstract resource among multiple coroutines. | |
4 | * | |
5 | * Copyright (c) 2022 Virtuozzo International GmbH | |
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 "qemu/coroutine.h" | |
28 | #include "block/aio.h" | |
29 | ||
30 | typedef struct QemuCoTimeoutState { | |
31 | CoroutineEntry *entry; | |
32 | void *opaque; | |
33 | QemuCoSleep sleep_state; | |
34 | bool marker; | |
35 | CleanupFunc *clean; | |
36 | } QemuCoTimeoutState; | |
37 | ||
38 | static void coroutine_fn qemu_co_timeout_entry(void *opaque) | |
39 | { | |
40 | QemuCoTimeoutState *s = opaque; | |
41 | ||
42 | s->entry(s->opaque); | |
43 | ||
44 | if (s->marker) { | |
45 | assert(!s->sleep_state.to_wake); | |
46 | /* .marker set by qemu_co_timeout, it have been failed */ | |
47 | if (s->clean) { | |
48 | s->clean(s->opaque); | |
49 | } | |
50 | g_free(s); | |
51 | } else { | |
52 | s->marker = true; | |
53 | qemu_co_sleep_wake(&s->sleep_state); | |
54 | } | |
55 | } | |
56 | ||
57 | int coroutine_fn qemu_co_timeout(CoroutineEntry *entry, void *opaque, | |
58 | uint64_t timeout_ns, CleanupFunc clean) | |
59 | { | |
60 | QemuCoTimeoutState *s; | |
61 | Coroutine *co; | |
62 | ||
63 | if (timeout_ns == 0) { | |
64 | entry(opaque); | |
65 | return 0; | |
66 | } | |
67 | ||
68 | s = g_new(QemuCoTimeoutState, 1); | |
69 | *s = (QemuCoTimeoutState) { | |
70 | .entry = entry, | |
71 | .opaque = opaque, | |
72 | .clean = clean | |
73 | }; | |
74 | ||
75 | co = qemu_coroutine_create(qemu_co_timeout_entry, s); | |
76 | ||
77 | aio_co_enter(qemu_get_current_aio_context(), co); | |
78 | qemu_co_sleep_ns_wakeable(&s->sleep_state, QEMU_CLOCK_REALTIME, timeout_ns); | |
79 | ||
80 | if (s->marker) { | |
81 | /* .marker set by qemu_co_timeout_entry, success */ | |
82 | g_free(s); | |
83 | return 0; | |
84 | } | |
85 | ||
86 | /* Don't free s, as we can't cancel qemu_co_timeout_entry execution */ | |
87 | s->marker = true; | |
88 | return -ETIMEDOUT; | |
89 | } |