1da668aa1SThomas Huth /*
2da668aa1SThomas Huth * QEMU I/O task tests
3da668aa1SThomas Huth *
4da668aa1SThomas Huth * Copyright (c) 2015 Red Hat, Inc.
5da668aa1SThomas Huth *
6da668aa1SThomas Huth * This library is free software; you can redistribute it and/or
7da668aa1SThomas Huth * modify it under the terms of the GNU Lesser General Public
8da668aa1SThomas Huth * License as published by the Free Software Foundation; either
9da668aa1SThomas Huth * version 2.1 of the License, or (at your option) any later version.
10da668aa1SThomas Huth *
11da668aa1SThomas Huth * This library is distributed in the hope that it will be useful,
12da668aa1SThomas Huth * but WITHOUT ANY WARRANTY; without even the implied warranty of
13da668aa1SThomas Huth * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14da668aa1SThomas Huth * Lesser General Public License for more details.
15da668aa1SThomas Huth *
16da668aa1SThomas Huth * You should have received a copy of the GNU Lesser General Public
17da668aa1SThomas Huth * License along with this library; if not, see <http://www.gnu.org/licenses/>.
18da668aa1SThomas Huth *
19da668aa1SThomas Huth */
20da668aa1SThomas Huth
21da668aa1SThomas Huth #include "qemu/osdep.h"
22da668aa1SThomas Huth
23da668aa1SThomas Huth #include "qom/object.h"
24da668aa1SThomas Huth #include "io/task.h"
25da668aa1SThomas Huth #include "qapi/error.h"
26da668aa1SThomas Huth #include "qemu/module.h"
27da668aa1SThomas Huth
28*05f2320dSThomas Huth #define TYPE_DUMMY "qemu-dummy"
29da668aa1SThomas Huth
30da668aa1SThomas Huth typedef struct DummyObject DummyObject;
31da668aa1SThomas Huth typedef struct DummyObjectClass DummyObjectClass;
32da668aa1SThomas Huth
33da668aa1SThomas Huth struct DummyObject {
34da668aa1SThomas Huth Object parent;
35da668aa1SThomas Huth };
36da668aa1SThomas Huth
37da668aa1SThomas Huth struct DummyObjectClass {
38da668aa1SThomas Huth ObjectClass parent;
39da668aa1SThomas Huth };
40da668aa1SThomas Huth
41da668aa1SThomas Huth static const TypeInfo dummy_info = {
42da668aa1SThomas Huth .parent = TYPE_OBJECT,
43da668aa1SThomas Huth .name = TYPE_DUMMY,
44da668aa1SThomas Huth .instance_size = sizeof(DummyObject),
45da668aa1SThomas Huth .class_size = sizeof(DummyObjectClass),
46da668aa1SThomas Huth };
47da668aa1SThomas Huth
48da668aa1SThomas Huth struct TestTaskData {
49da668aa1SThomas Huth Object *source;
50da668aa1SThomas Huth Error *err;
51da668aa1SThomas Huth bool freed;
52da668aa1SThomas Huth };
53da668aa1SThomas Huth
54da668aa1SThomas Huth
task_callback(QIOTask * task,gpointer opaque)55da668aa1SThomas Huth static void task_callback(QIOTask *task,
56da668aa1SThomas Huth gpointer opaque)
57da668aa1SThomas Huth {
58da668aa1SThomas Huth struct TestTaskData *data = opaque;
59da668aa1SThomas Huth
60da668aa1SThomas Huth data->source = qio_task_get_source(task);
61da668aa1SThomas Huth qio_task_propagate_error(task, &data->err);
62da668aa1SThomas Huth }
63da668aa1SThomas Huth
64da668aa1SThomas Huth
test_task_complete(void)65da668aa1SThomas Huth static void test_task_complete(void)
66da668aa1SThomas Huth {
67da668aa1SThomas Huth QIOTask *task;
68da668aa1SThomas Huth Object *obj = object_new(TYPE_DUMMY);
69da668aa1SThomas Huth Object *src;
70da668aa1SThomas Huth struct TestTaskData data = { NULL, NULL, false };
71da668aa1SThomas Huth
72da668aa1SThomas Huth task = qio_task_new(obj, task_callback, &data, NULL);
73da668aa1SThomas Huth src = qio_task_get_source(task);
74da668aa1SThomas Huth
75da668aa1SThomas Huth qio_task_complete(task);
76da668aa1SThomas Huth
77da668aa1SThomas Huth g_assert(obj == src);
78da668aa1SThomas Huth
79da668aa1SThomas Huth object_unref(obj);
80da668aa1SThomas Huth
81da668aa1SThomas Huth g_assert(data.source == obj);
82da668aa1SThomas Huth g_assert(data.err == NULL);
83da668aa1SThomas Huth g_assert(data.freed == false);
84da668aa1SThomas Huth }
85da668aa1SThomas Huth
86da668aa1SThomas Huth
task_data_free(gpointer opaque)87da668aa1SThomas Huth static void task_data_free(gpointer opaque)
88da668aa1SThomas Huth {
89da668aa1SThomas Huth struct TestTaskData *data = opaque;
90da668aa1SThomas Huth
91da668aa1SThomas Huth data->freed = true;
92da668aa1SThomas Huth }
93da668aa1SThomas Huth
94da668aa1SThomas Huth
test_task_data_free(void)95da668aa1SThomas Huth static void test_task_data_free(void)
96da668aa1SThomas Huth {
97da668aa1SThomas Huth QIOTask *task;
98da668aa1SThomas Huth Object *obj = object_new(TYPE_DUMMY);
99da668aa1SThomas Huth struct TestTaskData data = { NULL, NULL, false };
100da668aa1SThomas Huth
101da668aa1SThomas Huth task = qio_task_new(obj, task_callback, &data, task_data_free);
102da668aa1SThomas Huth
103da668aa1SThomas Huth qio_task_complete(task);
104da668aa1SThomas Huth
105da668aa1SThomas Huth object_unref(obj);
106da668aa1SThomas Huth
107da668aa1SThomas Huth g_assert(data.source == obj);
108da668aa1SThomas Huth g_assert(data.err == NULL);
109da668aa1SThomas Huth g_assert(data.freed == true);
110da668aa1SThomas Huth }
111da668aa1SThomas Huth
112da668aa1SThomas Huth
test_task_failure(void)113da668aa1SThomas Huth static void test_task_failure(void)
114da668aa1SThomas Huth {
115da668aa1SThomas Huth QIOTask *task;
116da668aa1SThomas Huth Object *obj = object_new(TYPE_DUMMY);
117da668aa1SThomas Huth struct TestTaskData data = { NULL, NULL, false };
118da668aa1SThomas Huth Error *err = NULL;
119da668aa1SThomas Huth
120da668aa1SThomas Huth task = qio_task_new(obj, task_callback, &data, NULL);
121da668aa1SThomas Huth
122da668aa1SThomas Huth error_setg(&err, "Some error");
123da668aa1SThomas Huth
124da668aa1SThomas Huth qio_task_set_error(task, err);
125da668aa1SThomas Huth qio_task_complete(task);
126da668aa1SThomas Huth
127da668aa1SThomas Huth object_unref(obj);
128da668aa1SThomas Huth
129da668aa1SThomas Huth g_assert(data.source == obj);
130da668aa1SThomas Huth g_assert(data.err == err);
131da668aa1SThomas Huth g_assert(data.freed == false);
132da668aa1SThomas Huth error_free(data.err);
133da668aa1SThomas Huth }
134da668aa1SThomas Huth
135da668aa1SThomas Huth
136da668aa1SThomas Huth struct TestThreadWorkerData {
137da668aa1SThomas Huth Object *source;
138da668aa1SThomas Huth Error *err;
139da668aa1SThomas Huth bool fail;
140da668aa1SThomas Huth GThread *worker;
141da668aa1SThomas Huth GThread *complete;
142da668aa1SThomas Huth GMainLoop *loop;
143da668aa1SThomas Huth };
144da668aa1SThomas Huth
test_task_thread_worker(QIOTask * task,gpointer opaque)145da668aa1SThomas Huth static void test_task_thread_worker(QIOTask *task,
146da668aa1SThomas Huth gpointer opaque)
147da668aa1SThomas Huth {
148da668aa1SThomas Huth struct TestThreadWorkerData *data = opaque;
149da668aa1SThomas Huth
150da668aa1SThomas Huth data->worker = g_thread_self();
151da668aa1SThomas Huth
152da668aa1SThomas Huth if (data->fail) {
153da668aa1SThomas Huth Error *err = NULL;
154da668aa1SThomas Huth error_setg(&err, "Testing fail");
155da668aa1SThomas Huth qio_task_set_error(task, err);
156da668aa1SThomas Huth }
157da668aa1SThomas Huth }
158da668aa1SThomas Huth
159da668aa1SThomas Huth
test_task_thread_callback(QIOTask * task,gpointer opaque)160da668aa1SThomas Huth static void test_task_thread_callback(QIOTask *task,
161da668aa1SThomas Huth gpointer opaque)
162da668aa1SThomas Huth {
163da668aa1SThomas Huth struct TestThreadWorkerData *data = opaque;
164da668aa1SThomas Huth
165da668aa1SThomas Huth data->source = qio_task_get_source(task);
166da668aa1SThomas Huth qio_task_propagate_error(task, &data->err);
167da668aa1SThomas Huth
168da668aa1SThomas Huth data->complete = g_thread_self();
169da668aa1SThomas Huth
170da668aa1SThomas Huth g_main_loop_quit(data->loop);
171da668aa1SThomas Huth }
172da668aa1SThomas Huth
173da668aa1SThomas Huth
test_task_thread_complete(void)174da668aa1SThomas Huth static void test_task_thread_complete(void)
175da668aa1SThomas Huth {
176da668aa1SThomas Huth QIOTask *task;
177da668aa1SThomas Huth Object *obj = object_new(TYPE_DUMMY);
178da668aa1SThomas Huth struct TestThreadWorkerData data = { 0 };
179da668aa1SThomas Huth GThread *self;
180da668aa1SThomas Huth
181da668aa1SThomas Huth data.loop = g_main_loop_new(g_main_context_default(),
182da668aa1SThomas Huth TRUE);
183da668aa1SThomas Huth
184da668aa1SThomas Huth task = qio_task_new(obj,
185da668aa1SThomas Huth test_task_thread_callback,
186da668aa1SThomas Huth &data,
187da668aa1SThomas Huth NULL);
188da668aa1SThomas Huth
189da668aa1SThomas Huth qio_task_run_in_thread(task,
190da668aa1SThomas Huth test_task_thread_worker,
191da668aa1SThomas Huth &data,
192da668aa1SThomas Huth NULL,
193da668aa1SThomas Huth NULL);
194da668aa1SThomas Huth
195da668aa1SThomas Huth g_main_loop_run(data.loop);
196da668aa1SThomas Huth
197da668aa1SThomas Huth g_main_loop_unref(data.loop);
198da668aa1SThomas Huth object_unref(obj);
199da668aa1SThomas Huth
200da668aa1SThomas Huth g_assert(data.source == obj);
201da668aa1SThomas Huth g_assert(data.err == NULL);
202da668aa1SThomas Huth
203da668aa1SThomas Huth self = g_thread_self();
204da668aa1SThomas Huth
205da668aa1SThomas Huth /* Make sure the test_task_thread_worker actually got
206da668aa1SThomas Huth * run in a different thread */
207da668aa1SThomas Huth g_assert(data.worker != self);
208da668aa1SThomas Huth
209da668aa1SThomas Huth /* And that the test_task_thread_callback got rnu in
210da668aa1SThomas Huth * the main loop thread (ie this one) */
211da668aa1SThomas Huth g_assert(data.complete == self);
212da668aa1SThomas Huth }
213da668aa1SThomas Huth
214da668aa1SThomas Huth
test_task_thread_failure(void)215da668aa1SThomas Huth static void test_task_thread_failure(void)
216da668aa1SThomas Huth {
217da668aa1SThomas Huth QIOTask *task;
218da668aa1SThomas Huth Object *obj = object_new(TYPE_DUMMY);
219da668aa1SThomas Huth struct TestThreadWorkerData data = { 0 };
220da668aa1SThomas Huth GThread *self;
221da668aa1SThomas Huth
222da668aa1SThomas Huth data.loop = g_main_loop_new(g_main_context_default(),
223da668aa1SThomas Huth TRUE);
224da668aa1SThomas Huth data.fail = true;
225da668aa1SThomas Huth
226da668aa1SThomas Huth task = qio_task_new(obj,
227da668aa1SThomas Huth test_task_thread_callback,
228da668aa1SThomas Huth &data,
229da668aa1SThomas Huth NULL);
230da668aa1SThomas Huth
231da668aa1SThomas Huth qio_task_run_in_thread(task,
232da668aa1SThomas Huth test_task_thread_worker,
233da668aa1SThomas Huth &data,
234da668aa1SThomas Huth NULL,
235da668aa1SThomas Huth NULL);
236da668aa1SThomas Huth
237da668aa1SThomas Huth g_main_loop_run(data.loop);
238da668aa1SThomas Huth
239da668aa1SThomas Huth g_main_loop_unref(data.loop);
240da668aa1SThomas Huth object_unref(obj);
241da668aa1SThomas Huth
242da668aa1SThomas Huth g_assert(data.source == obj);
243da668aa1SThomas Huth error_free_or_abort(&data.err);
244da668aa1SThomas Huth
245da668aa1SThomas Huth self = g_thread_self();
246da668aa1SThomas Huth
247da668aa1SThomas Huth /* Make sure the test_task_thread_worker actually got
248da668aa1SThomas Huth * run in a different thread */
249da668aa1SThomas Huth g_assert(data.worker != self);
250da668aa1SThomas Huth
251da668aa1SThomas Huth /* And that the test_task_thread_callback got rnu in
252da668aa1SThomas Huth * the main loop thread (ie this one) */
253da668aa1SThomas Huth g_assert(data.complete == self);
254da668aa1SThomas Huth }
255da668aa1SThomas Huth
256da668aa1SThomas Huth
main(int argc,char ** argv)257da668aa1SThomas Huth int main(int argc, char **argv)
258da668aa1SThomas Huth {
259da668aa1SThomas Huth g_test_init(&argc, &argv, NULL);
260da668aa1SThomas Huth module_call_init(MODULE_INIT_QOM);
261da668aa1SThomas Huth type_register_static(&dummy_info);
262da668aa1SThomas Huth g_test_add_func("/crypto/task/complete", test_task_complete);
263da668aa1SThomas Huth g_test_add_func("/crypto/task/datafree", test_task_data_free);
264da668aa1SThomas Huth g_test_add_func("/crypto/task/failure", test_task_failure);
265da668aa1SThomas Huth g_test_add_func("/crypto/task/thread_complete", test_task_thread_complete);
266da668aa1SThomas Huth g_test_add_func("/crypto/task/thread_failure", test_task_thread_failure);
267da668aa1SThomas Huth return g_test_run();
268da668aa1SThomas Huth }
269