]> git.proxmox.com Git - qemu.git/blob - coroutine-gthread.c
fw_cfg: Splash image loader can overrun a stack variable, fix
[qemu.git] / coroutine-gthread.c
1 /*
2 * GThread coroutine initialization code
3 *
4 * Copyright (C) 2006 Anthony Liguori <anthony@codemonkey.ws>
5 * Copyright (C) 2011 Aneesh Kumar K.V <aneesh.kumar@linux.vnet.ibm.com>
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.0 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, see <http://www.gnu.org/licenses/>.
19 */
20
21 #include <glib.h>
22 #include "qemu-common.h"
23 #include "block/coroutine_int.h"
24
25 typedef struct {
26 Coroutine base;
27 GThread *thread;
28 bool runnable;
29 bool free_on_thread_exit;
30 CoroutineAction action;
31 } CoroutineGThread;
32
33 static GStaticMutex coroutine_lock = G_STATIC_MUTEX_INIT;
34
35 /* GLib 2.31 and beyond deprecated various parts of the thread API,
36 * but the new interfaces are not available in older GLib versions
37 * so we have to cope with both.
38 */
39 #if GLIB_CHECK_VERSION(2, 31, 0)
40 /* Default zero-initialisation is sufficient for 2.31+ GCond */
41 static GCond the_coroutine_cond;
42 static GCond *coroutine_cond = &the_coroutine_cond;
43 static inline void init_coroutine_cond(void)
44 {
45 }
46
47 /* Awkwardly, the GPrivate API doesn't provide a way to update the
48 * GDestroyNotify handler for the coroutine key dynamically. So instead
49 * we track whether or not the CoroutineGThread should be freed on
50 * thread exit / coroutine key update using the free_on_thread_exit
51 * field.
52 */
53 static void coroutine_destroy_notify(gpointer data)
54 {
55 CoroutineGThread *co = data;
56 if (co && co->free_on_thread_exit) {
57 g_free(co);
58 }
59 }
60
61 static GPrivate coroutine_key = G_PRIVATE_INIT(coroutine_destroy_notify);
62
63 static inline CoroutineGThread *get_coroutine_key(void)
64 {
65 return g_private_get(&coroutine_key);
66 }
67
68 static inline void set_coroutine_key(CoroutineGThread *co,
69 bool free_on_thread_exit)
70 {
71 /* Unlike g_static_private_set() this does not call the GDestroyNotify
72 * if the previous value of the key was NULL. Fortunately we only need
73 * the GDestroyNotify in the non-NULL key case.
74 */
75 co->free_on_thread_exit = free_on_thread_exit;
76 g_private_replace(&coroutine_key, co);
77 }
78
79 static inline GThread *create_thread(GThreadFunc func, gpointer data)
80 {
81 return g_thread_new("coroutine", func, data);
82 }
83
84 #else
85
86 /* Handle older GLib versions */
87 static GCond *coroutine_cond;
88 static inline void init_coroutine_cond(void)
89 {
90 coroutine_cond = g_cond_new();
91 }
92
93 static GStaticPrivate coroutine_key = G_STATIC_PRIVATE_INIT;
94
95 static inline CoroutineGThread *get_coroutine_key(void)
96 {
97 return g_static_private_get(&coroutine_key);
98 }
99
100 static inline void set_coroutine_key(CoroutineGThread *co,
101 bool free_on_thread_exit)
102 {
103 g_static_private_set(&coroutine_key, co,
104 free_on_thread_exit ? (GDestroyNotify)g_free : NULL);
105 }
106
107 static inline GThread *create_thread(GThreadFunc func, gpointer data)
108 {
109 return g_thread_create_full(func, data, 0, TRUE, TRUE,
110 G_THREAD_PRIORITY_NORMAL, NULL);
111 }
112
113 #endif
114
115
116 static void __attribute__((constructor)) coroutine_init(void)
117 {
118 if (!g_thread_supported()) {
119 #if !GLIB_CHECK_VERSION(2, 31, 0)
120 g_thread_init(NULL);
121 #else
122 fprintf(stderr, "glib threading failed to initialize.\n");
123 exit(1);
124 #endif
125 }
126
127 init_coroutine_cond();
128 }
129
130 static void coroutine_wait_runnable_locked(CoroutineGThread *co)
131 {
132 while (!co->runnable) {
133 g_cond_wait(coroutine_cond, g_static_mutex_get_mutex(&coroutine_lock));
134 }
135 }
136
137 static void coroutine_wait_runnable(CoroutineGThread *co)
138 {
139 g_static_mutex_lock(&coroutine_lock);
140 coroutine_wait_runnable_locked(co);
141 g_static_mutex_unlock(&coroutine_lock);
142 }
143
144 static gpointer coroutine_thread(gpointer opaque)
145 {
146 CoroutineGThread *co = opaque;
147
148 set_coroutine_key(co, false);
149 coroutine_wait_runnable(co);
150 co->base.entry(co->base.entry_arg);
151 qemu_coroutine_switch(&co->base, co->base.caller, COROUTINE_TERMINATE);
152 return NULL;
153 }
154
155 Coroutine *qemu_coroutine_new(void)
156 {
157 CoroutineGThread *co;
158
159 co = g_malloc0(sizeof(*co));
160 co->thread = create_thread(coroutine_thread, co);
161 if (!co->thread) {
162 g_free(co);
163 return NULL;
164 }
165 return &co->base;
166 }
167
168 void qemu_coroutine_delete(Coroutine *co_)
169 {
170 CoroutineGThread *co = DO_UPCAST(CoroutineGThread, base, co_);
171
172 g_thread_join(co->thread);
173 g_free(co);
174 }
175
176 CoroutineAction qemu_coroutine_switch(Coroutine *from_,
177 Coroutine *to_,
178 CoroutineAction action)
179 {
180 CoroutineGThread *from = DO_UPCAST(CoroutineGThread, base, from_);
181 CoroutineGThread *to = DO_UPCAST(CoroutineGThread, base, to_);
182
183 g_static_mutex_lock(&coroutine_lock);
184 from->runnable = false;
185 from->action = action;
186 to->runnable = true;
187 to->action = action;
188 g_cond_broadcast(coroutine_cond);
189
190 if (action != COROUTINE_TERMINATE) {
191 coroutine_wait_runnable_locked(from);
192 }
193 g_static_mutex_unlock(&coroutine_lock);
194 return from->action;
195 }
196
197 Coroutine *qemu_coroutine_self(void)
198 {
199 CoroutineGThread *co = get_coroutine_key();
200 if (!co) {
201 co = g_malloc0(sizeof(*co));
202 co->runnable = true;
203 set_coroutine_key(co, true);
204 }
205
206 return &co->base;
207 }
208
209 bool qemu_in_coroutine(void)
210 {
211 CoroutineGThread *co = get_coroutine_key();
212
213 return co && co->base.caller;
214 }