1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21#include "qemu/osdep.h"
22#include <glib.h>
23
24#include "io/task.h"
25#include "qapi/error.h"
26
27#define TYPE_DUMMY "qemu:dummy"
28
29typedef struct DummyObject DummyObject;
30typedef struct DummyObjectClass DummyObjectClass;
31
32struct DummyObject {
33 Object parent;
34};
35
36struct DummyObjectClass {
37 ObjectClass parent;
38};
39
40static const TypeInfo dummy_info = {
41 .parent = TYPE_OBJECT,
42 .name = TYPE_DUMMY,
43 .instance_size = sizeof(DummyObject),
44 .class_size = sizeof(DummyObjectClass),
45};
46
47struct TestTaskData {
48 Object *source;
49 Error *err;
50 bool freed;
51};
52
53
54static void task_callback(Object *source,
55 Error *err,
56 gpointer opaque)
57{
58 struct TestTaskData *data = opaque;
59
60 data->source = source;
61 data->err = err;
62}
63
64
65static void test_task_complete(void)
66{
67 QIOTask *task;
68 Object *obj = object_new(TYPE_DUMMY);
69 Object *src;
70 struct TestTaskData data = { NULL, NULL, false };
71
72 task = qio_task_new(obj, task_callback, &data, NULL);
73 src = qio_task_get_source(task);
74
75 qio_task_complete(task);
76
77 g_assert(obj == src);
78
79 object_unref(obj);
80 object_unref(src);
81
82 g_assert(data.source == obj);
83 g_assert(data.err == NULL);
84 g_assert(data.freed == false);
85}
86
87
88static void task_data_free(gpointer opaque)
89{
90 struct TestTaskData *data = opaque;
91
92 data->freed = true;
93}
94
95
96static void test_task_data_free(void)
97{
98 QIOTask *task;
99 Object *obj = object_new(TYPE_DUMMY);
100 struct TestTaskData data = { NULL, NULL, false };
101
102 task = qio_task_new(obj, task_callback, &data, task_data_free);
103
104 qio_task_complete(task);
105
106 object_unref(obj);
107
108 g_assert(data.source == obj);
109 g_assert(data.err == NULL);
110 g_assert(data.freed == true);
111}
112
113
114static void test_task_error(void)
115{
116 QIOTask *task;
117 Object *obj = object_new(TYPE_DUMMY);
118 struct TestTaskData data = { NULL, NULL, false };
119 Error *err = NULL;
120
121 task = qio_task_new(obj, task_callback, &data, NULL);
122
123 error_setg(&err, "Some error");
124
125 qio_task_abort(task, err);
126
127 error_free(err);
128 object_unref(obj);
129
130 g_assert(data.source == obj);
131 g_assert(data.err == err);
132 g_assert(data.freed == false);
133
134}
135
136
137struct TestThreadWorkerData {
138 Object *source;
139 Error *err;
140 bool fail;
141 GThread *worker;
142 GThread *complete;
143 GMainLoop *loop;
144};
145
146static int test_task_thread_worker(QIOTask *task,
147 Error **errp,
148 gpointer opaque)
149{
150 struct TestThreadWorkerData *data = opaque;
151
152 data->worker = g_thread_self();
153
154 if (data->fail) {
155 error_setg(errp, "Testing fail");
156 return -1;
157 }
158
159 return 0;
160}
161
162
163static void test_task_thread_callback(Object *source,
164 Error *err,
165 gpointer opaque)
166{
167 struct TestThreadWorkerData *data = opaque;
168
169 data->source = source;
170 data->err = err;
171
172 data->complete = g_thread_self();
173
174 g_main_loop_quit(data->loop);
175}
176
177
178static void test_task_thread_complete(void)
179{
180 QIOTask *task;
181 Object *obj = object_new(TYPE_DUMMY);
182 struct TestThreadWorkerData data = { 0 };
183 GThread *self;
184
185 data.loop = g_main_loop_new(g_main_context_default(),
186 TRUE);
187
188 task = qio_task_new(obj,
189 test_task_thread_callback,
190 &data,
191 NULL);
192
193 qio_task_run_in_thread(task,
194 test_task_thread_worker,
195 &data,
196 NULL);
197
198 g_main_loop_run(data.loop);
199
200 g_main_loop_unref(data.loop);
201 object_unref(obj);
202
203 g_assert(data.source == obj);
204 g_assert(data.err == NULL);
205
206 self = g_thread_self();
207
208
209
210 g_assert(data.worker != self);
211
212
213
214 g_assert(data.complete == self);
215}
216
217
218static void test_task_thread_error(void)
219{
220 QIOTask *task;
221 Object *obj = object_new(TYPE_DUMMY);
222 struct TestThreadWorkerData data = { 0 };
223 GThread *self;
224
225 data.loop = g_main_loop_new(g_main_context_default(),
226 TRUE);
227 data.fail = true;
228
229 task = qio_task_new(obj,
230 test_task_thread_callback,
231 &data,
232 NULL);
233
234 qio_task_run_in_thread(task,
235 test_task_thread_worker,
236 &data,
237 NULL);
238
239 g_main_loop_run(data.loop);
240
241 g_main_loop_unref(data.loop);
242 object_unref(obj);
243
244 g_assert(data.source == obj);
245 g_assert(data.err != NULL);
246
247 self = g_thread_self();
248
249
250
251 g_assert(data.worker != self);
252
253
254
255 g_assert(data.complete == self);
256}
257
258
259int main(int argc, char **argv)
260{
261 g_test_init(&argc, &argv, NULL);
262 module_call_init(MODULE_INIT_QOM);
263 type_register_static(&dummy_info);
264 g_test_add_func("/crypto/task/complete", test_task_complete);
265 g_test_add_func("/crypto/task/datafree", test_task_data_free);
266 g_test_add_func("/crypto/task/error", test_task_error);
267 g_test_add_func("/crypto/task/thread_complete", test_task_thread_complete);
268 g_test_add_func("/crypto/task/thread_error", test_task_thread_error);
269 return g_test_run();
270}
271