]>
Commit | Line | Data |
---|---|---|
d3b12f5d PB |
1 | /* |
2 | * QEMU System Emulator | |
3 | * | |
4 | * Copyright (c) 2003-2008 Fabrice Bellard | |
5 | * | |
6 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
7 | * of this software and associated documentation files (the "Software"), to deal | |
8 | * in the Software without restriction, including without limitation the rights | |
9 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
10 | * copies of the Software, and to permit persons to whom the Software is | |
11 | * furnished to do so, subject to the following conditions: | |
12 | * | |
13 | * The above copyright notice and this permission notice shall be included in | |
14 | * all copies or substantial portions of the Software. | |
15 | * | |
16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
19 | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
20 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
21 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | |
22 | * THE SOFTWARE. | |
23 | */ | |
d3b12f5d | 24 | |
d38ea87a | 25 | #include "qemu/osdep.h" |
da34e65c | 26 | #include "qapi/error.h" |
f348b6d1 | 27 | #include "qemu/cutils.h" |
1de7afc9 | 28 | #include "qemu/timer.h" |
740b1759 | 29 | #include "sysemu/cpu-timers.h" |
d759c951 | 30 | #include "sysemu/replay.h" |
1de7afc9 | 31 | #include "qemu/main-loop.h" |
737e150e | 32 | #include "block/aio.h" |
71ad4713 | 33 | #include "block/thread-pool.h" |
8297be80 | 34 | #include "qemu/error-report.h" |
e9ed92bd | 35 | #include "qemu/queue.h" |
c905a368 | 36 | #include "qemu/compiler.h" |
70ac26b9 | 37 | #include "qom/object.h" |
e9ed92bd PB |
38 | |
39 | #ifndef _WIN32 | |
40 | #include <sys/wait.h> | |
41 | #endif | |
d3b12f5d PB |
42 | |
43 | #ifndef _WIN32 | |
44 | ||
d3b12f5d PB |
45 | /* If we have signalfd, we mask out the signals we want to handle and then |
46 | * use signalfd to listen for them. We rely on whatever the current signal | |
47 | * handler is to dispatch the signals when we receive them. | |
48 | */ | |
c905a368 DB |
49 | /* |
50 | * Disable CFI checks. | |
51 | * We are going to call a signal hander directly. Such handler may or may not | |
52 | * have been defined in our binary, so there's no guarantee that the pointer | |
53 | * used to set the handler is a cfi-valid pointer. Since the handlers are | |
54 | * stored in kernel memory, changing the handler to an attacker-defined | |
55 | * function requires being able to call a sigaction() syscall, | |
56 | * which is not as easy as overwriting a pointer in memory. | |
57 | */ | |
58 | QEMU_DISABLE_CFI | |
d3b12f5d PB |
59 | static void sigfd_handler(void *opaque) |
60 | { | |
61 | int fd = (intptr_t)opaque; | |
62 | struct qemu_signalfd_siginfo info; | |
63 | struct sigaction action; | |
64 | ssize_t len; | |
65 | ||
66 | while (1) { | |
67 | do { | |
68 | len = read(fd, &info, sizeof(info)); | |
69 | } while (len == -1 && errno == EINTR); | |
70 | ||
71 | if (len == -1 && errno == EAGAIN) { | |
72 | break; | |
73 | } | |
74 | ||
75 | if (len != sizeof(info)) { | |
372a87a1 TH |
76 | error_report("read from sigfd returned %zd: %s", len, |
77 | g_strerror(errno)); | |
d3b12f5d PB |
78 | return; |
79 | } | |
80 | ||
81 | sigaction(info.ssi_signo, NULL, &action); | |
82 | if ((action.sa_flags & SA_SIGINFO) && action.sa_sigaction) { | |
d98d4072 | 83 | sigaction_invoke(&action, &info); |
d3b12f5d PB |
84 | } else if (action.sa_handler) { |
85 | action.sa_handler(info.ssi_signo); | |
86 | } | |
87 | } | |
88 | } | |
89 | ||
78524330 | 90 | static int qemu_signal_init(Error **errp) |
d3b12f5d PB |
91 | { |
92 | int sigfd; | |
93 | sigset_t set; | |
94 | ||
95 | /* | |
96 | * SIG_IPI must be blocked in the main thread and must not be caught | |
97 | * by sigwait() in the signal thread. Otherwise, the cpu thread will | |
98 | * not catch it reliably. | |
99 | */ | |
100 | sigemptyset(&set); | |
101 | sigaddset(&set, SIG_IPI); | |
d3b12f5d PB |
102 | sigaddset(&set, SIGIO); |
103 | sigaddset(&set, SIGALRM); | |
104 | sigaddset(&set, SIGBUS); | |
3e9418e1 JK |
105 | /* SIGINT cannot be handled via signalfd, so that ^C can be used |
106 | * to interrupt QEMU when it is being run under gdb. SIGHUP and | |
107 | * SIGTERM are also handled asynchronously, even though it is not | |
108 | * strictly necessary, because they use the same handler as SIGINT. | |
109 | */ | |
d3b12f5d PB |
110 | pthread_sigmask(SIG_BLOCK, &set, NULL); |
111 | ||
4aa7534d | 112 | sigdelset(&set, SIG_IPI); |
d3b12f5d PB |
113 | sigfd = qemu_signalfd(&set); |
114 | if (sigfd == -1) { | |
78524330 | 115 | error_setg_errno(errp, errno, "failed to create signalfd"); |
d3b12f5d PB |
116 | return -errno; |
117 | } | |
118 | ||
4d14cb0c | 119 | g_unix_set_fd_nonblocking(sigfd, true, NULL); |
d3b12f5d | 120 | |
82e1cc4b | 121 | qemu_set_fd_handler(sigfd, sigfd_handler, NULL, (void *)(intptr_t)sigfd); |
d3b12f5d PB |
122 | |
123 | return 0; | |
124 | } | |
125 | ||
126 | #else /* _WIN32 */ | |
127 | ||
78524330 | 128 | static int qemu_signal_init(Error **errp) |
d3b12f5d | 129 | { |
d3b12f5d PB |
130 | return 0; |
131 | } | |
4c8d0d27 PB |
132 | #endif |
133 | ||
134 | static AioContext *qemu_aio_context; | |
edec47cf PB |
135 | static QEMUBH *qemu_notify_bh; |
136 | ||
137 | static void notify_event_cb(void *opaque) | |
138 | { | |
139 | /* No need to do anything; this bottom half is only used to | |
140 | * kick the kernel out of ppoll/poll/WaitForMultipleObjects. | |
141 | */ | |
142 | } | |
d3b12f5d | 143 | |
5f3aa1ff SH |
144 | AioContext *qemu_get_aio_context(void) |
145 | { | |
146 | return qemu_aio_context; | |
147 | } | |
148 | ||
d3b12f5d PB |
149 | void qemu_notify_event(void) |
150 | { | |
4c8d0d27 | 151 | if (!qemu_aio_context) { |
ee77dfb2 MR |
152 | return; |
153 | } | |
edec47cf | 154 | qemu_bh_schedule(qemu_notify_bh); |
d3b12f5d PB |
155 | } |
156 | ||
cbff4b34 SH |
157 | static GArray *gpollfds; |
158 | ||
2f78e491 | 159 | int qemu_init_main_loop(Error **errp) |
d3b12f5d PB |
160 | { |
161 | int ret; | |
82cbbdc6 | 162 | GSource *src; |
d3b12f5d | 163 | |
3f53bc61 | 164 | init_clocks(qemu_timer_notify_cb); |
172061a0 | 165 | |
78524330 | 166 | ret = qemu_signal_init(errp); |
d3b12f5d PB |
167 | if (ret) { |
168 | return ret; | |
169 | } | |
170 | ||
668f62ec | 171 | qemu_aio_context = aio_context_new(errp); |
2f78e491 | 172 | if (!qemu_aio_context) { |
2f78e491 CN |
173 | return -EMFILE; |
174 | } | |
5f50be9b | 175 | qemu_set_current_aio_context(qemu_aio_context); |
28ba61e7 | 176 | qemu_notify_bh = qemu_bh_new(notify_event_cb, NULL); |
cbff4b34 | 177 | gpollfds = g_array_new(FALSE, FALSE, sizeof(GPollFD)); |
82cbbdc6 | 178 | src = aio_get_g_source(qemu_aio_context); |
c3ff757d | 179 | g_source_set_name(src, "aio-context"); |
82cbbdc6 PB |
180 | g_source_attach(src, NULL); |
181 | g_source_unref(src); | |
f3926945 | 182 | src = iohandler_get_g_source(); |
c3ff757d | 183 | g_source_set_name(src, "io-handler"); |
f3926945 FZ |
184 | g_source_attach(src, NULL); |
185 | g_source_unref(src); | |
d3b12f5d PB |
186 | return 0; |
187 | } | |
188 | ||
70ac26b9 NSJ |
189 | static void main_loop_update_params(EventLoopBase *base, Error **errp) |
190 | { | |
71ad4713 NSJ |
191 | ERRP_GUARD(); |
192 | ||
70ac26b9 NSJ |
193 | if (!qemu_aio_context) { |
194 | error_setg(errp, "qemu aio context not ready"); | |
195 | return; | |
196 | } | |
197 | ||
198 | aio_context_set_aio_params(qemu_aio_context, base->aio_max_batch, errp); | |
71ad4713 NSJ |
199 | if (*errp) { |
200 | return; | |
201 | } | |
202 | ||
203 | aio_context_set_thread_pool_params(qemu_aio_context, base->thread_pool_min, | |
204 | base->thread_pool_max, errp); | |
70ac26b9 NSJ |
205 | } |
206 | ||
207 | MainLoop *mloop; | |
208 | ||
209 | static void main_loop_init(EventLoopBase *base, Error **errp) | |
210 | { | |
211 | MainLoop *m = MAIN_LOOP(base); | |
212 | ||
213 | if (mloop) { | |
214 | error_setg(errp, "only one main-loop instance allowed"); | |
215 | return; | |
216 | } | |
217 | ||
218 | main_loop_update_params(base, errp); | |
219 | ||
220 | mloop = m; | |
221 | return; | |
222 | } | |
223 | ||
224 | static bool main_loop_can_be_deleted(EventLoopBase *base) | |
225 | { | |
226 | return false; | |
227 | } | |
228 | ||
229 | static void main_loop_class_init(ObjectClass *oc, void *class_data) | |
230 | { | |
231 | EventLoopBaseClass *bc = EVENT_LOOP_BASE_CLASS(oc); | |
232 | ||
233 | bc->init = main_loop_init; | |
234 | bc->update_params = main_loop_update_params; | |
235 | bc->can_be_deleted = main_loop_can_be_deleted; | |
236 | } | |
237 | ||
238 | static const TypeInfo main_loop_info = { | |
239 | .name = TYPE_MAIN_LOOP, | |
240 | .parent = TYPE_EVENT_LOOP_BASE, | |
241 | .class_init = main_loop_class_init, | |
242 | .instance_size = sizeof(MainLoop), | |
243 | }; | |
244 | ||
245 | static void main_loop_register_types(void) | |
246 | { | |
247 | type_register_static(&main_loop_info); | |
248 | } | |
249 | ||
250 | type_init(main_loop_register_types) | |
251 | ||
d3b12f5d PB |
252 | static int max_priority; |
253 | ||
ea26ce76 | 254 | #ifndef _WIN32 |
48ce11ff SH |
255 | static int glib_pollfds_idx; |
256 | static int glib_n_poll_fds; | |
257 | ||
8c278762 TH |
258 | void qemu_fd_register(int fd) |
259 | { | |
260 | } | |
261 | ||
7b595f35 | 262 | static void glib_pollfds_fill(int64_t *cur_timeout) |
d3b12f5d PB |
263 | { |
264 | GMainContext *context = g_main_context_default(); | |
4dae83ae | 265 | int timeout = 0; |
7b595f35 | 266 | int64_t timeout_ns; |
48ce11ff | 267 | int n; |
d3b12f5d PB |
268 | |
269 | g_main_context_prepare(context, &max_priority); | |
270 | ||
48ce11ff SH |
271 | glib_pollfds_idx = gpollfds->len; |
272 | n = glib_n_poll_fds; | |
273 | do { | |
274 | GPollFD *pfds; | |
275 | glib_n_poll_fds = n; | |
276 | g_array_set_size(gpollfds, glib_pollfds_idx + glib_n_poll_fds); | |
277 | pfds = &g_array_index(gpollfds, GPollFD, glib_pollfds_idx); | |
278 | n = g_main_context_query(context, max_priority, &timeout, pfds, | |
279 | glib_n_poll_fds); | |
280 | } while (n != glib_n_poll_fds); | |
d3b12f5d | 281 | |
7b595f35 AB |
282 | if (timeout < 0) { |
283 | timeout_ns = -1; | |
284 | } else { | |
285 | timeout_ns = (int64_t)timeout * (int64_t)SCALE_MS; | |
d3b12f5d | 286 | } |
7b595f35 AB |
287 | |
288 | *cur_timeout = qemu_soonest_timeout(timeout_ns, *cur_timeout); | |
d3b12f5d PB |
289 | } |
290 | ||
48ce11ff | 291 | static void glib_pollfds_poll(void) |
d3b12f5d PB |
292 | { |
293 | GMainContext *context = g_main_context_default(); | |
48ce11ff | 294 | GPollFD *pfds = &g_array_index(gpollfds, GPollFD, glib_pollfds_idx); |
d3b12f5d | 295 | |
48ce11ff | 296 | if (g_main_context_check(context, max_priority, pfds, glib_n_poll_fds)) { |
d3b12f5d PB |
297 | g_main_context_dispatch(context); |
298 | } | |
299 | } | |
300 | ||
893986fe AL |
301 | #define MAX_MAIN_LOOP_SPIN (1000) |
302 | ||
7b595f35 | 303 | static int os_host_main_loop_wait(int64_t timeout) |
15455536 | 304 | { |
ecbddbb1 | 305 | GMainContext *context = g_main_context_default(); |
15455536 PB |
306 | int ret; |
307 | ||
ecbddbb1 RJ |
308 | g_main_context_acquire(context); |
309 | ||
48ce11ff | 310 | glib_pollfds_fill(&timeout); |
15455536 | 311 | |
d759c951 AB |
312 | qemu_mutex_unlock_iothread(); |
313 | replay_mutex_unlock(); | |
15455536 | 314 | |
7b595f35 | 315 | ret = qemu_poll_ns((GPollFD *)gpollfds->data, gpollfds->len, timeout); |
cbff4b34 | 316 | |
d759c951 AB |
317 | replay_mutex_lock(); |
318 | qemu_mutex_lock_iothread(); | |
15455536 | 319 | |
48ce11ff | 320 | glib_pollfds_poll(); |
ecbddbb1 RJ |
321 | |
322 | g_main_context_release(context); | |
323 | ||
15455536 PB |
324 | return ret; |
325 | } | |
326 | #else | |
d3b12f5d PB |
327 | /***********************************************************/ |
328 | /* Polling handling */ | |
329 | ||
330 | typedef struct PollingEntry { | |
331 | PollingFunc *func; | |
332 | void *opaque; | |
333 | struct PollingEntry *next; | |
334 | } PollingEntry; | |
335 | ||
336 | static PollingEntry *first_polling_entry; | |
337 | ||
338 | int qemu_add_polling_cb(PollingFunc *func, void *opaque) | |
339 | { | |
340 | PollingEntry **ppe, *pe; | |
b21e2380 | 341 | pe = g_new0(PollingEntry, 1); |
d3b12f5d PB |
342 | pe->func = func; |
343 | pe->opaque = opaque; | |
344 | for(ppe = &first_polling_entry; *ppe != NULL; ppe = &(*ppe)->next); | |
345 | *ppe = pe; | |
346 | return 0; | |
347 | } | |
348 | ||
349 | void qemu_del_polling_cb(PollingFunc *func, void *opaque) | |
350 | { | |
351 | PollingEntry **ppe, *pe; | |
352 | for(ppe = &first_polling_entry; *ppe != NULL; ppe = &(*ppe)->next) { | |
353 | pe = *ppe; | |
354 | if (pe->func == func && pe->opaque == opaque) { | |
355 | *ppe = pe->next; | |
356 | g_free(pe); | |
357 | break; | |
358 | } | |
359 | } | |
360 | } | |
361 | ||
362 | /***********************************************************/ | |
363 | /* Wait objects support */ | |
364 | typedef struct WaitObjects { | |
365 | int num; | |
06ac7d49 | 366 | int revents[MAXIMUM_WAIT_OBJECTS + 1]; |
d3b12f5d PB |
367 | HANDLE events[MAXIMUM_WAIT_OBJECTS + 1]; |
368 | WaitObjectFunc *func[MAXIMUM_WAIT_OBJECTS + 1]; | |
369 | void *opaque[MAXIMUM_WAIT_OBJECTS + 1]; | |
370 | } WaitObjects; | |
371 | ||
372 | static WaitObjects wait_objects = {0}; | |
373 | ||
374 | int qemu_add_wait_object(HANDLE handle, WaitObjectFunc *func, void *opaque) | |
375 | { | |
376 | WaitObjects *w = &wait_objects; | |
377 | if (w->num >= MAXIMUM_WAIT_OBJECTS) { | |
378 | return -1; | |
379 | } | |
380 | w->events[w->num] = handle; | |
381 | w->func[w->num] = func; | |
382 | w->opaque[w->num] = opaque; | |
06ac7d49 | 383 | w->revents[w->num] = 0; |
d3b12f5d PB |
384 | w->num++; |
385 | return 0; | |
386 | } | |
387 | ||
388 | void qemu_del_wait_object(HANDLE handle, WaitObjectFunc *func, void *opaque) | |
389 | { | |
390 | int i, found; | |
391 | WaitObjects *w = &wait_objects; | |
392 | ||
393 | found = 0; | |
394 | for (i = 0; i < w->num; i++) { | |
395 | if (w->events[i] == handle) { | |
396 | found = 1; | |
397 | } | |
398 | if (found) { | |
399 | w->events[i] = w->events[i + 1]; | |
400 | w->func[i] = w->func[i + 1]; | |
401 | w->opaque[i] = w->opaque[i + 1]; | |
06ac7d49 | 402 | w->revents[i] = w->revents[i + 1]; |
d3b12f5d PB |
403 | } |
404 | } | |
405 | if (found) { | |
406 | w->num--; | |
407 | } | |
408 | } | |
409 | ||
d3385eb4 PB |
410 | void qemu_fd_register(int fd) |
411 | { | |
4c8d0d27 PB |
412 | WSAEventSelect(fd, event_notifier_get_handle(&qemu_aio_context->notifier), |
413 | FD_READ | FD_ACCEPT | FD_CLOSE | | |
d3385eb4 PB |
414 | FD_CONNECT | FD_WRITE | FD_OOB); |
415 | } | |
416 | ||
cbff4b34 SH |
417 | static int pollfds_fill(GArray *pollfds, fd_set *rfds, fd_set *wfds, |
418 | fd_set *xfds) | |
419 | { | |
420 | int nfds = -1; | |
421 | int i; | |
422 | ||
423 | for (i = 0; i < pollfds->len; i++) { | |
424 | GPollFD *pfd = &g_array_index(pollfds, GPollFD, i); | |
425 | int fd = pfd->fd; | |
426 | int events = pfd->events; | |
8db165b3 | 427 | if (events & G_IO_IN) { |
cbff4b34 SH |
428 | FD_SET(fd, rfds); |
429 | nfds = MAX(nfds, fd); | |
430 | } | |
8db165b3 | 431 | if (events & G_IO_OUT) { |
cbff4b34 SH |
432 | FD_SET(fd, wfds); |
433 | nfds = MAX(nfds, fd); | |
434 | } | |
435 | if (events & G_IO_PRI) { | |
436 | FD_SET(fd, xfds); | |
437 | nfds = MAX(nfds, fd); | |
438 | } | |
439 | } | |
440 | return nfds; | |
441 | } | |
442 | ||
443 | static void pollfds_poll(GArray *pollfds, int nfds, fd_set *rfds, | |
444 | fd_set *wfds, fd_set *xfds) | |
445 | { | |
446 | int i; | |
447 | ||
448 | for (i = 0; i < pollfds->len; i++) { | |
449 | GPollFD *pfd = &g_array_index(pollfds, GPollFD, i); | |
450 | int fd = pfd->fd; | |
451 | int revents = 0; | |
452 | ||
453 | if (FD_ISSET(fd, rfds)) { | |
8db165b3 | 454 | revents |= G_IO_IN; |
cbff4b34 SH |
455 | } |
456 | if (FD_ISSET(fd, wfds)) { | |
8db165b3 | 457 | revents |= G_IO_OUT; |
cbff4b34 SH |
458 | } |
459 | if (FD_ISSET(fd, xfds)) { | |
460 | revents |= G_IO_PRI; | |
461 | } | |
462 | pfd->revents = revents & pfd->events; | |
463 | } | |
464 | } | |
465 | ||
7b595f35 | 466 | static int os_host_main_loop_wait(int64_t timeout) |
d3b12f5d | 467 | { |
ea26ce76 | 468 | GMainContext *context = g_main_context_default(); |
48ce11ff | 469 | GPollFD poll_fds[1024 * 2]; /* this is probably overkill */ |
134a03e0 | 470 | int select_ret = 0; |
48ce11ff | 471 | int g_poll_ret, ret, i, n_poll_fds; |
d3b12f5d | 472 | PollingEntry *pe; |
d3385eb4 | 473 | WaitObjects *w = &wait_objects; |
42fe1c24 | 474 | gint poll_timeout; |
7b595f35 | 475 | int64_t poll_timeout_ns; |
15455536 | 476 | static struct timeval tv0; |
9cbaacf9 SH |
477 | fd_set rfds, wfds, xfds; |
478 | int nfds; | |
d3b12f5d | 479 | |
ecbddbb1 RJ |
480 | g_main_context_acquire(context); |
481 | ||
d3b12f5d PB |
482 | /* XXX: need to suppress polling by better using win32 events */ |
483 | ret = 0; | |
484 | for (pe = first_polling_entry; pe != NULL; pe = pe->next) { | |
485 | ret |= pe->func(pe->opaque); | |
486 | } | |
d3385eb4 | 487 | if (ret != 0) { |
ecbddbb1 | 488 | g_main_context_release(context); |
d3385eb4 PB |
489 | return ret; |
490 | } | |
d3b12f5d | 491 | |
3cb8c205 SH |
492 | FD_ZERO(&rfds); |
493 | FD_ZERO(&wfds); | |
494 | FD_ZERO(&xfds); | |
495 | nfds = pollfds_fill(gpollfds, &rfds, &wfds, &xfds); | |
496 | if (nfds >= 0) { | |
497 | select_ret = select(nfds + 1, &rfds, &wfds, &xfds, &tv0); | |
498 | if (select_ret != 0) { | |
499 | timeout = 0; | |
500 | } | |
501 | if (select_ret > 0) { | |
502 | pollfds_poll(gpollfds, nfds, &rfds, &wfds, &xfds); | |
503 | } | |
504 | } | |
505 | ||
ea26ce76 | 506 | g_main_context_prepare(context, &max_priority); |
42fe1c24 | 507 | n_poll_fds = g_main_context_query(context, max_priority, &poll_timeout, |
ea26ce76 | 508 | poll_fds, ARRAY_SIZE(poll_fds)); |
6512e34b | 509 | g_assert(n_poll_fds + w->num <= ARRAY_SIZE(poll_fds)); |
ea26ce76 | 510 | |
06ac7d49 | 511 | for (i = 0; i < w->num; i++) { |
58b9630d | 512 | poll_fds[n_poll_fds + i].fd = (DWORD_PTR)w->events[i]; |
ea26ce76 | 513 | poll_fds[n_poll_fds + i].events = G_IO_IN; |
06ac7d49 PB |
514 | } |
515 | ||
7b595f35 AB |
516 | if (poll_timeout < 0) { |
517 | poll_timeout_ns = -1; | |
518 | } else { | |
519 | poll_timeout_ns = (int64_t)poll_timeout * (int64_t)SCALE_MS; | |
3239ad04 SW |
520 | } |
521 | ||
7b595f35 AB |
522 | poll_timeout_ns = qemu_soonest_timeout(poll_timeout_ns, timeout); |
523 | ||
d3385eb4 | 524 | qemu_mutex_unlock_iothread(); |
d759c951 AB |
525 | |
526 | replay_mutex_unlock(); | |
527 | ||
7b595f35 AB |
528 | g_poll_ret = qemu_poll_ns(poll_fds, n_poll_fds + w->num, poll_timeout_ns); |
529 | ||
d759c951 AB |
530 | replay_mutex_lock(); |
531 | ||
d3385eb4 | 532 | qemu_mutex_lock_iothread(); |
5e3bc735 | 533 | if (g_poll_ret > 0) { |
06ac7d49 | 534 | for (i = 0; i < w->num; i++) { |
ea26ce76 | 535 | w->revents[i] = poll_fds[n_poll_fds + i].revents; |
d3385eb4 | 536 | } |
06ac7d49 PB |
537 | for (i = 0; i < w->num; i++) { |
538 | if (w->revents[i] && w->func[i]) { | |
539 | w->func[i](w->opaque[i]); | |
d3b12f5d | 540 | } |
d3b12f5d PB |
541 | } |
542 | } | |
543 | ||
ea26ce76 PB |
544 | if (g_main_context_check(context, max_priority, poll_fds, n_poll_fds)) { |
545 | g_main_context_dispatch(context); | |
546 | } | |
547 | ||
ecbddbb1 RJ |
548 | g_main_context_release(context); |
549 | ||
5e3bc735 | 550 | return select_ret || g_poll_ret; |
d3b12f5d PB |
551 | } |
552 | #endif | |
553 | ||
1ab67b98 MAL |
554 | static NotifierList main_loop_poll_notifiers = |
555 | NOTIFIER_LIST_INITIALIZER(main_loop_poll_notifiers); | |
556 | ||
557 | void main_loop_poll_add_notifier(Notifier *notify) | |
558 | { | |
559 | notifier_list_add(&main_loop_poll_notifiers, notify); | |
560 | } | |
561 | ||
562 | void main_loop_poll_remove_notifier(Notifier *notify) | |
563 | { | |
564 | notifier_remove(notify); | |
565 | } | |
566 | ||
de5f852f | 567 | void main_loop_wait(int nonblocking) |
d3b12f5d | 568 | { |
1ab67b98 MAL |
569 | MainLoopPoll mlpoll = { |
570 | .state = MAIN_LOOP_POLL_FILL, | |
571 | .timeout = UINT32_MAX, | |
572 | .pollfds = gpollfds, | |
573 | }; | |
7c7db755 | 574 | int ret; |
7b595f35 | 575 | int64_t timeout_ns; |
d3b12f5d PB |
576 | |
577 | if (nonblocking) { | |
1ab67b98 | 578 | mlpoll.timeout = 0; |
d3b12f5d PB |
579 | } |
580 | ||
d3b12f5d | 581 | /* poll any events */ |
cbff4b34 | 582 | g_array_set_size(gpollfds, 0); /* reset for new iteration */ |
d3b12f5d | 583 | /* XXX: separate device handlers from system ones */ |
1ab67b98 | 584 | notifier_list_notify(&main_loop_poll_notifiers, &mlpoll); |
7b595f35 | 585 | |
1ab67b98 | 586 | if (mlpoll.timeout == UINT32_MAX) { |
7b595f35 AB |
587 | timeout_ns = -1; |
588 | } else { | |
1ab67b98 | 589 | timeout_ns = (uint64_t)mlpoll.timeout * (int64_t)(SCALE_MS); |
7b595f35 AB |
590 | } |
591 | ||
592 | timeout_ns = qemu_soonest_timeout(timeout_ns, | |
593 | timerlistgroup_deadline_ns( | |
594 | &main_loop_tlg)); | |
595 | ||
596 | ret = os_host_main_loop_wait(timeout_ns); | |
1ab67b98 MAL |
597 | mlpoll.state = ret < 0 ? MAIN_LOOP_POLL_ERR : MAIN_LOOP_POLL_OK; |
598 | notifier_list_notify(&main_loop_poll_notifiers, &mlpoll); | |
d3b12f5d | 599 | |
740b1759 CF |
600 | if (icount_enabled()) { |
601 | /* | |
602 | * CPU thread can infinitely wait for event after | |
603 | * missing the warp | |
604 | */ | |
8191d368 | 605 | icount_start_warp_timer(); |
740b1759 | 606 | } |
40daca54 | 607 | qemu_clock_run_all_timers(); |
d3b12f5d | 608 | } |
f627aab1 PB |
609 | |
610 | /* Functions to operate on the main QEMU AioContext. */ | |
611 | ||
0f08586c | 612 | QEMUBH *qemu_bh_new_full(QEMUBHFunc *cb, void *opaque, const char *name) |
f627aab1 | 613 | { |
0f08586c | 614 | return aio_bh_new_full(qemu_aio_context, cb, opaque, name); |
f627aab1 | 615 | } |
e9ed92bd PB |
616 | |
617 | /* | |
618 | * Functions to operate on the I/O handler AioContext. | |
619 | * This context runs on top of main loop. We can't reuse qemu_aio_context | |
620 | * because iohandlers mustn't be polled by aio_poll(qemu_aio_context). | |
621 | */ | |
622 | static AioContext *iohandler_ctx; | |
623 | ||
624 | static void iohandler_init(void) | |
625 | { | |
626 | if (!iohandler_ctx) { | |
627 | iohandler_ctx = aio_context_new(&error_abort); | |
628 | } | |
629 | } | |
630 | ||
631 | AioContext *iohandler_get_aio_context(void) | |
632 | { | |
633 | iohandler_init(); | |
634 | return iohandler_ctx; | |
635 | } | |
636 | ||
637 | GSource *iohandler_get_g_source(void) | |
638 | { | |
639 | iohandler_init(); | |
640 | return aio_get_g_source(iohandler_ctx); | |
641 | } | |
642 | ||
643 | void qemu_set_fd_handler(int fd, | |
644 | IOHandler *fd_read, | |
645 | IOHandler *fd_write, | |
646 | void *opaque) | |
647 | { | |
648 | iohandler_init(); | |
649 | aio_set_fd_handler(iohandler_ctx, fd, false, | |
826cc324 | 650 | fd_read, fd_write, NULL, NULL, opaque); |
e9ed92bd PB |
651 | } |
652 | ||
653 | void event_notifier_set_handler(EventNotifier *e, | |
654 | EventNotifierHandler *handler) | |
655 | { | |
656 | iohandler_init(); | |
657 | aio_set_event_notifier(iohandler_ctx, e, false, | |
826cc324 | 658 | handler, NULL, NULL); |
e9ed92bd | 659 | } |