qdev: remove baked in notion of aliases (v2)
[qemu.git] / test-coroutine.c
1 /*
2 * Coroutine tests
3 *
4 * Copyright IBM, Corp. 2011
5 *
6 * Authors:
7 * Stefan Hajnoczi <stefanha@linux.vnet.ibm.com>
8 *
9 * This work is licensed under the terms of the GNU LGPL, version 2 or later.
10 * See the COPYING.LIB file in the top-level directory.
11 *
12 */
13
14 #include <glib.h>
15 #include "qemu-coroutine.h"
16
17 /*
18 * Check that qemu_in_coroutine() works
19 */
20
21 static void coroutine_fn verify_in_coroutine(void *opaque)
22 {
23 g_assert(qemu_in_coroutine());
24 }
25
26 static void test_in_coroutine(void)
27 {
28 Coroutine *coroutine;
29
30 g_assert(!qemu_in_coroutine());
31
32 coroutine = qemu_coroutine_create(verify_in_coroutine);
33 qemu_coroutine_enter(coroutine, NULL);
34 }
35
36 /*
37 * Check that qemu_coroutine_self() works
38 */
39
40 static void coroutine_fn verify_self(void *opaque)
41 {
42 g_assert(qemu_coroutine_self() == opaque);
43 }
44
45 static void test_self(void)
46 {
47 Coroutine *coroutine;
48
49 coroutine = qemu_coroutine_create(verify_self);
50 qemu_coroutine_enter(coroutine, coroutine);
51 }
52
53 /*
54 * Check that coroutines may nest multiple levels
55 */
56
57 typedef struct {
58 unsigned int n_enter; /* num coroutines entered */
59 unsigned int n_return; /* num coroutines returned */
60 unsigned int max; /* maximum level of nesting */
61 } NestData;
62
63 static void coroutine_fn nest(void *opaque)
64 {
65 NestData *nd = opaque;
66
67 nd->n_enter++;
68
69 if (nd->n_enter < nd->max) {
70 Coroutine *child;
71
72 child = qemu_coroutine_create(nest);
73 qemu_coroutine_enter(child, nd);
74 }
75
76 nd->n_return++;
77 }
78
79 static void test_nesting(void)
80 {
81 Coroutine *root;
82 NestData nd = {
83 .n_enter = 0,
84 .n_return = 0,
85 .max = 128,
86 };
87
88 root = qemu_coroutine_create(nest);
89 qemu_coroutine_enter(root, &nd);
90
91 /* Must enter and return from max nesting level */
92 g_assert_cmpint(nd.n_enter, ==, nd.max);
93 g_assert_cmpint(nd.n_return, ==, nd.max);
94 }
95
96 /*
97 * Check that yield/enter transfer control correctly
98 */
99
100 static void coroutine_fn yield_5_times(void *opaque)
101 {
102 bool *done = opaque;
103 int i;
104
105 for (i = 0; i < 5; i++) {
106 qemu_coroutine_yield();
107 }
108 *done = true;
109 }
110
111 static void test_yield(void)
112 {
113 Coroutine *coroutine;
114 bool done = false;
115 int i = -1; /* one extra time to return from coroutine */
116
117 coroutine = qemu_coroutine_create(yield_5_times);
118 while (!done) {
119 qemu_coroutine_enter(coroutine, &done);
120 i++;
121 }
122 g_assert_cmpint(i, ==, 5); /* coroutine must yield 5 times */
123 }
124
125 /*
126 * Check that creation, enter, and return work
127 */
128
129 static void coroutine_fn set_and_exit(void *opaque)
130 {
131 bool *done = opaque;
132
133 *done = true;
134 }
135
136 static void test_lifecycle(void)
137 {
138 Coroutine *coroutine;
139 bool done = false;
140
141 /* Create, enter, and return from coroutine */
142 coroutine = qemu_coroutine_create(set_and_exit);
143 qemu_coroutine_enter(coroutine, &done);
144 g_assert(done); /* expect done to be true (first time) */
145
146 /* Repeat to check that no state affects this test */
147 done = false;
148 coroutine = qemu_coroutine_create(set_and_exit);
149 qemu_coroutine_enter(coroutine, &done);
150 g_assert(done); /* expect done to be true (second time) */
151 }
152
153 /*
154 * Lifecycle benchmark
155 */
156
157 static void coroutine_fn empty_coroutine(void *opaque)
158 {
159 /* Do nothing */
160 }
161
162 static void perf_lifecycle(void)
163 {
164 Coroutine *coroutine;
165 unsigned int i, max;
166 double duration;
167
168 max = 1000000;
169
170 g_test_timer_start();
171 for (i = 0; i < max; i++) {
172 coroutine = qemu_coroutine_create(empty_coroutine);
173 qemu_coroutine_enter(coroutine, NULL);
174 }
175 duration = g_test_timer_elapsed();
176
177 g_test_message("Lifecycle %u iterations: %f s\n", max, duration);
178 }
179
180 int main(int argc, char **argv)
181 {
182 g_test_init(&argc, &argv, NULL);
183 g_test_add_func("/basic/lifecycle", test_lifecycle);
184 g_test_add_func("/basic/yield", test_yield);
185 g_test_add_func("/basic/nesting", test_nesting);
186 g_test_add_func("/basic/self", test_self);
187 g_test_add_func("/basic/in_coroutine", test_in_coroutine);
188 if (g_test_perf()) {
189 g_test_add_func("/perf/lifecycle", perf_lifecycle);
190 }
191 return g_test_run();
192 }