]> git.proxmox.com Git - mirror_qemu.git/blob - util/event_notifier-posix.c
event_notifier: add event_notifier_get_wfd()
[mirror_qemu.git] / util / event_notifier-posix.c
1 /*
2 * event notifier support
3 *
4 * Copyright Red Hat, Inc. 2010
5 *
6 * Authors:
7 * Michael S. Tsirkin <mst@redhat.com>
8 *
9 * This work is licensed under the terms of the GNU GPL, version 2 or later.
10 * See the COPYING file in the top-level directory.
11 */
12
13 #include "qemu/osdep.h"
14 #include "qemu-common.h"
15 #include "qemu/cutils.h"
16 #include "qemu/event_notifier.h"
17 #include "qemu/main-loop.h"
18
19 #ifdef CONFIG_EVENTFD
20 #include <sys/eventfd.h>
21 #endif
22
23 #ifdef CONFIG_EVENTFD
24 /*
25 * Initialize @e with existing file descriptor @fd.
26 * @fd must be a genuine eventfd object, emulation with pipe won't do.
27 */
28 void event_notifier_init_fd(EventNotifier *e, int fd)
29 {
30 e->rfd = fd;
31 e->wfd = fd;
32 e->initialized = true;
33 }
34 #endif
35
36 int event_notifier_init(EventNotifier *e, int active)
37 {
38 int fds[2];
39 int ret;
40
41 #ifdef CONFIG_EVENTFD
42 ret = eventfd(0, EFD_NONBLOCK | EFD_CLOEXEC);
43 #else
44 ret = -1;
45 errno = ENOSYS;
46 #endif
47 if (ret >= 0) {
48 e->rfd = e->wfd = ret;
49 } else {
50 if (errno != ENOSYS) {
51 return -errno;
52 }
53 if (qemu_pipe(fds) < 0) {
54 return -errno;
55 }
56 ret = fcntl_setfl(fds[0], O_NONBLOCK);
57 if (ret < 0) {
58 ret = -errno;
59 goto fail;
60 }
61 ret = fcntl_setfl(fds[1], O_NONBLOCK);
62 if (ret < 0) {
63 ret = -errno;
64 goto fail;
65 }
66 e->rfd = fds[0];
67 e->wfd = fds[1];
68 }
69 e->initialized = true;
70 if (active) {
71 event_notifier_set(e);
72 }
73 return 0;
74
75 fail:
76 close(fds[0]);
77 close(fds[1]);
78 return ret;
79 }
80
81 void event_notifier_cleanup(EventNotifier *e)
82 {
83 if (!e->initialized) {
84 return;
85 }
86
87 if (e->rfd != e->wfd) {
88 close(e->rfd);
89 }
90
91 e->rfd = -1;
92 close(e->wfd);
93 e->wfd = -1;
94 e->initialized = false;
95 }
96
97 int event_notifier_get_fd(const EventNotifier *e)
98 {
99 return e->rfd;
100 }
101
102 int event_notifier_get_wfd(const EventNotifier *e)
103 {
104 return e->wfd;
105 }
106
107 int event_notifier_set(EventNotifier *e)
108 {
109 static const uint64_t value = 1;
110 ssize_t ret;
111
112 if (!e->initialized) {
113 return -1;
114 }
115
116 do {
117 ret = write(e->wfd, &value, sizeof(value));
118 } while (ret < 0 && errno == EINTR);
119
120 /* EAGAIN is fine, a read must be pending. */
121 if (ret < 0 && errno != EAGAIN) {
122 return -errno;
123 }
124 return 0;
125 }
126
127 int event_notifier_test_and_clear(EventNotifier *e)
128 {
129 int value;
130 ssize_t len;
131 char buffer[512];
132
133 if (!e->initialized) {
134 return 0;
135 }
136
137 /* Drain the notify pipe. For eventfd, only 8 bytes will be read. */
138 value = 0;
139 do {
140 len = read(e->rfd, buffer, sizeof(buffer));
141 value |= (len > 0);
142 } while ((len == -1 && errno == EINTR) || len == sizeof(buffer));
143
144 return value;
145 }