]>
Commit | Line | Data |
---|---|---|
b02db2d9 DB |
1 | /* |
2 | * QEMU I/O task | |
3 | * | |
4 | * Copyright (c) 2015 Red Hat, Inc. | |
5 | * | |
6 | * This library is free software; you can redistribute it and/or | |
7 | * modify it under the terms of the GNU Lesser General Public | |
8 | * License as published by the Free Software Foundation; either | |
c8198bd5 | 9 | * version 2.1 of the License, or (at your option) any later version. |
b02db2d9 DB |
10 | * |
11 | * This library is distributed in the hope that it will be useful, | |
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | * Lesser General Public License for more details. | |
15 | * | |
16 | * You should have received a copy of the GNU Lesser General Public | |
17 | * License along with this library; if not, see <http://www.gnu.org/licenses/>. | |
18 | * | |
19 | */ | |
20 | ||
cae9fc56 | 21 | #include "qemu/osdep.h" |
b02db2d9 | 22 | #include "io/task.h" |
da34e65c | 23 | #include "qapi/error.h" |
b02db2d9 | 24 | #include "qemu/thread.h" |
78f8d497 | 25 | #include "qom/object.h" |
b02db2d9 DB |
26 | #include "trace.h" |
27 | ||
52d6cfec DB |
28 | struct QIOTaskThreadData { |
29 | QIOTaskWorker worker; | |
30 | gpointer opaque; | |
31 | GDestroyNotify destroy; | |
32 | GMainContext *context; | |
dbb44504 | 33 | GSource *completion; |
52d6cfec DB |
34 | }; |
35 | ||
36 | ||
b02db2d9 DB |
37 | struct QIOTask { |
38 | Object *source; | |
39 | QIOTaskFunc func; | |
40 | gpointer opaque; | |
41 | GDestroyNotify destroy; | |
1a447e4f | 42 | Error *err; |
52dd99e8 DB |
43 | gpointer result; |
44 | GDestroyNotify destroyResult; | |
dbb44504 DB |
45 | QemuMutex thread_lock; |
46 | QemuCond thread_cond; | |
52d6cfec | 47 | struct QIOTaskThreadData *thread; |
b02db2d9 DB |
48 | }; |
49 | ||
50 | ||
51 | QIOTask *qio_task_new(Object *source, | |
52 | QIOTaskFunc func, | |
53 | gpointer opaque, | |
54 | GDestroyNotify destroy) | |
55 | { | |
56 | QIOTask *task; | |
57 | ||
58 | task = g_new0(QIOTask, 1); | |
59 | ||
60 | task->source = source; | |
61 | object_ref(source); | |
62 | task->func = func; | |
63 | task->opaque = opaque; | |
64 | task->destroy = destroy; | |
dbb44504 DB |
65 | qemu_mutex_init(&task->thread_lock); |
66 | qemu_cond_init(&task->thread_cond); | |
b02db2d9 DB |
67 | |
68 | trace_qio_task_new(task, source, func, opaque); | |
69 | ||
70 | return task; | |
71 | } | |
72 | ||
73 | static void qio_task_free(QIOTask *task) | |
74 | { | |
dbb44504 | 75 | qemu_mutex_lock(&task->thread_lock); |
52d6cfec DB |
76 | if (task->thread) { |
77 | if (task->thread->destroy) { | |
78 | task->thread->destroy(task->thread->opaque); | |
79 | } | |
80 | ||
81 | if (task->thread->context) { | |
82 | g_main_context_unref(task->thread->context); | |
83 | } | |
84 | ||
85 | g_free(task->thread); | |
86 | } | |
87 | ||
b02db2d9 DB |
88 | if (task->destroy) { |
89 | task->destroy(task->opaque); | |
90 | } | |
52dd99e8 DB |
91 | if (task->destroyResult) { |
92 | task->destroyResult(task->result); | |
93 | } | |
1a447e4f DB |
94 | if (task->err) { |
95 | error_free(task->err); | |
96 | } | |
b02db2d9 DB |
97 | object_unref(task->source); |
98 | ||
dbb44504 DB |
99 | qemu_mutex_unlock(&task->thread_lock); |
100 | qemu_mutex_destroy(&task->thread_lock); | |
101 | qemu_cond_destroy(&task->thread_cond); | |
102 | ||
b02db2d9 DB |
103 | g_free(task); |
104 | } | |
105 | ||
106 | ||
7c28768f | 107 | static gboolean qio_task_thread_result(gpointer opaque) |
b02db2d9 | 108 | { |
52d6cfec | 109 | QIOTask *task = opaque; |
b02db2d9 | 110 | |
52d6cfec DB |
111 | trace_qio_task_thread_result(task); |
112 | qio_task_complete(task); | |
b02db2d9 DB |
113 | |
114 | return FALSE; | |
115 | } | |
116 | ||
117 | ||
118 | static gpointer qio_task_thread_worker(gpointer opaque) | |
119 | { | |
52d6cfec | 120 | QIOTask *task = opaque; |
b02db2d9 | 121 | |
52d6cfec DB |
122 | trace_qio_task_thread_run(task); |
123 | ||
124 | task->thread->worker(task, task->thread->opaque); | |
b02db2d9 DB |
125 | |
126 | /* We're running in the background thread, and must only | |
127 | * ever report the task results in the main event loop | |
128 | * thread. So we schedule an idle callback to report | |
129 | * the worker results | |
130 | */ | |
52d6cfec | 131 | trace_qio_task_thread_exit(task); |
a17536c5 | 132 | |
dbb44504 DB |
133 | qemu_mutex_lock(&task->thread_lock); |
134 | ||
135 | task->thread->completion = g_idle_source_new(); | |
136 | g_source_set_callback(task->thread->completion, | |
137 | qio_task_thread_result, task, NULL); | |
138 | g_source_attach(task->thread->completion, | |
139 | task->thread->context); | |
b65cb867 | 140 | g_source_unref(task->thread->completion); |
dbb44504 DB |
141 | trace_qio_task_thread_source_attach(task, task->thread->completion); |
142 | ||
143 | qemu_cond_signal(&task->thread_cond); | |
144 | qemu_mutex_unlock(&task->thread_lock); | |
a17536c5 | 145 | |
b02db2d9 DB |
146 | return NULL; |
147 | } | |
148 | ||
149 | ||
150 | void qio_task_run_in_thread(QIOTask *task, | |
151 | QIOTaskWorker worker, | |
152 | gpointer opaque, | |
a17536c5 PX |
153 | GDestroyNotify destroy, |
154 | GMainContext *context) | |
b02db2d9 DB |
155 | { |
156 | struct QIOTaskThreadData *data = g_new0(struct QIOTaskThreadData, 1); | |
157 | QemuThread thread; | |
158 | ||
a17536c5 PX |
159 | if (context) { |
160 | g_main_context_ref(context); | |
161 | } | |
162 | ||
b02db2d9 DB |
163 | data->worker = worker; |
164 | data->opaque = opaque; | |
165 | data->destroy = destroy; | |
a17536c5 | 166 | data->context = context; |
b02db2d9 | 167 | |
52d6cfec DB |
168 | task->thread = data; |
169 | ||
b02db2d9 DB |
170 | trace_qio_task_thread_start(task, worker, opaque); |
171 | qemu_thread_create(&thread, | |
172 | "io-task-worker", | |
173 | qio_task_thread_worker, | |
52d6cfec | 174 | task, |
b02db2d9 DB |
175 | QEMU_THREAD_DETACHED); |
176 | } | |
177 | ||
178 | ||
dbb44504 DB |
179 | void qio_task_wait_thread(QIOTask *task) |
180 | { | |
181 | qemu_mutex_lock(&task->thread_lock); | |
182 | g_assert(task->thread != NULL); | |
183 | while (task->thread->completion == NULL) { | |
184 | qemu_cond_wait(&task->thread_cond, &task->thread_lock); | |
185 | } | |
186 | ||
187 | trace_qio_task_thread_source_cancel(task, task->thread->completion); | |
188 | g_source_destroy(task->thread->completion); | |
189 | qemu_mutex_unlock(&task->thread_lock); | |
190 | ||
191 | qio_task_thread_result(task); | |
192 | } | |
193 | ||
194 | ||
b02db2d9 DB |
195 | void qio_task_complete(QIOTask *task) |
196 | { | |
60e705c5 | 197 | task->func(task, task->opaque); |
b02db2d9 DB |
198 | trace_qio_task_complete(task); |
199 | qio_task_free(task); | |
200 | } | |
201 | ||
b02db2d9 | 202 | |
1a447e4f DB |
203 | void qio_task_set_error(QIOTask *task, |
204 | Error *err) | |
205 | { | |
206 | error_propagate(&task->err, err); | |
207 | } | |
208 | ||
209 | ||
210 | bool qio_task_propagate_error(QIOTask *task, | |
211 | Error **errp) | |
212 | { | |
213 | if (task->err) { | |
214 | error_propagate(errp, task->err); | |
80fb34ed | 215 | task->err = NULL; |
1a447e4f DB |
216 | return true; |
217 | } | |
218 | ||
219 | return false; | |
220 | } | |
221 | ||
222 | ||
52dd99e8 DB |
223 | void qio_task_set_result_pointer(QIOTask *task, |
224 | gpointer result, | |
225 | GDestroyNotify destroy) | |
226 | { | |
227 | task->result = result; | |
228 | task->destroyResult = destroy; | |
229 | } | |
230 | ||
231 | ||
232 | gpointer qio_task_get_result_pointer(QIOTask *task) | |
233 | { | |
234 | return task->result; | |
235 | } | |
236 | ||
237 | ||
b02db2d9 DB |
238 | Object *qio_task_get_source(QIOTask *task) |
239 | { | |
b02db2d9 DB |
240 | return task->source; |
241 | } |