event-loop.c 10.6 KB
Newer Older
1 2 3
/*
 * Copyright © 2008 Kristian Høgsberg
 *
4 5 6 7 8 9 10
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
11
 *
12 13 14 15 16 17 18 19 20 21 22 23
 * The above copyright notice and this permission notice (including the
 * next paragraph) shall be included in all copies or substantial
 * portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
24 25
 */

Kristian Høgsberg's avatar
Kristian Høgsberg committed
26 27 28
#include <stddef.h>
#include <stdio.h>
#include <errno.h>
29
#include <signal.h>
Kristian Høgsberg's avatar
Kristian Høgsberg committed
30
#include <stdlib.h>
Yong Bakos's avatar
Yong Bakos committed
31
#include <stdint.h>
Kristian Høgsberg's avatar
Kristian Høgsberg committed
32
#include <string.h>
33
#include <fcntl.h>
Kristian Høgsberg's avatar
Kristian Høgsberg committed
34 35 36
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/epoll.h>
37
#include <sys/signalfd.h>
38
#include <sys/timerfd.h>
Kristian Høgsberg's avatar
Kristian Høgsberg committed
39
#include <unistd.h>
40
#include "wayland-util.h"
41
#include "wayland-private.h"
42
#include "wayland-server-core.h"
Pekka Paalanen's avatar
Pekka Paalanen committed
43
#include "wayland-os.h"
Kristian Høgsberg's avatar
Kristian Høgsberg committed
44 45 46

struct wl_event_loop {
	int epoll_fd;
47
	struct wl_list check_list;
48
	struct wl_list idle_list;
49
	struct wl_list destroy_list;
50 51

	struct wl_signal destroy_signal;
52 53 54
};

struct wl_event_source_interface {
55 56
	int (*dispatch)(struct wl_event_source *source,
			struct epoll_event *ep);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
57 58 59
};

struct wl_event_source {
60 61
	struct wl_event_source_interface *interface;
	struct wl_event_loop *loop;
62 63
	struct wl_list link;
	void *data;
64
	int fd;
65 66 67 68
};

struct wl_event_source_fd {
	struct wl_event_source base;
69
	wl_event_loop_fd_func_t func;
70
	int fd;
Kristian Høgsberg's avatar
Kristian Høgsberg committed
71 72
};

73
static int
74 75 76 77 78 79 80 81 82 83
wl_event_source_fd_dispatch(struct wl_event_source *source,
			    struct epoll_event *ep)
{
	struct wl_event_source_fd *fd_source = (struct wl_event_source_fd *) source;
	uint32_t mask;

	mask = 0;
	if (ep->events & EPOLLIN)
		mask |= WL_EVENT_READABLE;
	if (ep->events & EPOLLOUT)
84
		mask |= WL_EVENT_WRITABLE;
85 86 87 88
	if (ep->events & EPOLLHUP)
		mask |= WL_EVENT_HANGUP;
	if (ep->events & EPOLLERR)
		mask |= WL_EVENT_ERROR;
89

90
	return fd_source->func(fd_source->fd, mask, source->data);
91 92 93 94 95 96
}

struct wl_event_source_interface fd_source_interface = {
	wl_event_source_fd_dispatch,
};

97 98 99
static struct wl_event_source *
add_source(struct wl_event_loop *loop,
	   struct wl_event_source *source, uint32_t mask, void *data)
Kristian Høgsberg's avatar
Kristian Høgsberg committed
100 101 102
{
	struct epoll_event ep;

103 104
	if (source->fd < 0) {
		free(source);
105
		return NULL;
106
	}
107

108 109 110
	source->loop = loop;
	source->data = data;
	wl_list_init(&source->link);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
111

112
	memset(&ep, 0, sizeof ep);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
113 114
	if (mask & WL_EVENT_READABLE)
		ep.events |= EPOLLIN;
115
	if (mask & WL_EVENT_WRITABLE)
Kristian Høgsberg's avatar
Kristian Høgsberg committed
116 117 118
		ep.events |= EPOLLOUT;
	ep.data.ptr = source;

119 120
	if (epoll_ctl(loop->epoll_fd, EPOLL_CTL_ADD, source->fd, &ep) < 0) {
		close(source->fd);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
121 122 123 124
		free(source);
		return NULL;
	}

125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140
	return source;
}

WL_EXPORT struct wl_event_source *
wl_event_loop_add_fd(struct wl_event_loop *loop,
		     int fd, uint32_t mask,
		     wl_event_loop_fd_func_t func,
		     void *data)
{
	struct wl_event_source_fd *source;

	source = malloc(sizeof *source);
	if (source == NULL)
		return NULL;

	source->base.interface = &fd_source_interface;
141
	source->base.fd = wl_os_dupfd_cloexec(fd, 0);
142 143 144 145
	source->func = func;
	source->fd = fd;

	return add_source(loop, &source->base, mask, data);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
146 147
}

148
WL_EXPORT int
149
wl_event_source_fd_update(struct wl_event_source *source, uint32_t mask)
150
{
151
	struct wl_event_loop *loop = source->loop;
152 153
	struct epoll_event ep;

154
	memset(&ep, 0, sizeof ep);
155 156
	if (mask & WL_EVENT_READABLE)
		ep.events |= EPOLLIN;
157
	if (mask & WL_EVENT_WRITABLE)
158 159 160
		ep.events |= EPOLLOUT;
	ep.data.ptr = source;

161
	return epoll_ctl(loop->epoll_fd, EPOLL_CTL_MOD, source->fd, &ep);
162 163 164 165 166 167 168
}

struct wl_event_source_timer {
	struct wl_event_source base;
	wl_event_loop_timer_func_t func;
};

169
static int
170 171 172 173 174 175
wl_event_source_timer_dispatch(struct wl_event_source *source,
			       struct epoll_event *ep)
{
	struct wl_event_source_timer *timer_source =
		(struct wl_event_source_timer *) source;
	uint64_t expires;
176
	int len;
177

178
	len = read(source->fd, &expires, sizeof expires);
179
	if (!(len == -1 && errno == EAGAIN) && len != sizeof expires)
180
		/* Is there anything we can do here?  Will this ever happen? */
Kristian Høgsberg's avatar
Kristian Høgsberg committed
181
		wl_log("timerfd read error: %m\n");
182

183
	return timer_source->func(timer_source->base.data);
184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201
}

struct wl_event_source_interface timer_source_interface = {
	wl_event_source_timer_dispatch,
};

WL_EXPORT struct wl_event_source *
wl_event_loop_add_timer(struct wl_event_loop *loop,
			wl_event_loop_timer_func_t func,
			void *data)
{
	struct wl_event_source_timer *source;

	source = malloc(sizeof *source);
	if (source == NULL)
		return NULL;

	source->base.interface = &timer_source_interface;
202
	source->base.fd = timerfd_create(CLOCK_MONOTONIC,
203
					 TFD_CLOEXEC | TFD_NONBLOCK);
204 205
	source->func = func;

206
	return add_source(loop, &source->base, WL_EVENT_READABLE, data);
207 208 209 210 211 212 213 214 215
}

WL_EXPORT int
wl_event_source_timer_update(struct wl_event_source *source, int ms_delay)
{
	struct itimerspec its;

	its.it_interval.tv_sec = 0;
	its.it_interval.tv_nsec = 0;
216 217
	its.it_value.tv_sec = ms_delay / 1000;
	its.it_value.tv_nsec = (ms_delay % 1000) * 1000 * 1000;
Kristian Høgsberg's avatar
Kristian Høgsberg committed
218
	if (timerfd_settime(source->fd, 0, &its, NULL) < 0)
219 220 221 222 223
		return -1;

	return 0;
}

224 225 226 227 228 229
struct wl_event_source_signal {
	struct wl_event_source base;
	int signal_number;
	wl_event_loop_signal_func_t func;
};

230
static int
231
wl_event_source_signal_dispatch(struct wl_event_source *source,
232
				struct epoll_event *ep)
233 234 235 236
{
	struct wl_event_source_signal *signal_source =
		(struct wl_event_source_signal *) source;
	struct signalfd_siginfo signal_info;
237
	int len;
238

239
	len = read(source->fd, &signal_info, sizeof signal_info);
240
	if (!(len == -1 && errno == EAGAIN) && len != sizeof signal_info)
241
		/* Is there anything we can do here?  Will this ever happen? */
Kristian Høgsberg's avatar
Kristian Høgsberg committed
242
		wl_log("signalfd read error: %m\n");
243

244 245
	return signal_source->func(signal_source->signal_number,
				   signal_source->base.data);
246 247 248 249 250 251 252 253
}

struct wl_event_source_interface signal_source_interface = {
	wl_event_source_signal_dispatch,
};

WL_EXPORT struct wl_event_source *
wl_event_loop_add_signal(struct wl_event_loop *loop,
254 255 256
			 int signal_number,
			 wl_event_loop_signal_func_t func,
			 void *data)
257 258 259 260 261 262 263 264 265
{
	struct wl_event_source_signal *source;
	sigset_t mask;

	source = malloc(sizeof *source);
	if (source == NULL)
		return NULL;

	source->base.interface = &signal_source_interface;
266
	source->signal_number = signal_number;
267 268 269

	sigemptyset(&mask);
	sigaddset(&mask, signal_number);
270
	source->base.fd = signalfd(-1, &mask, SFD_CLOEXEC | SFD_NONBLOCK);
271 272 273 274
	sigprocmask(SIG_BLOCK, &mask, NULL);

	source->func = func;

275
	return add_source(loop, &source->base, WL_EVENT_READABLE, data);
276 277
}

278 279 280 281 282 283
struct wl_event_source_idle {
	struct wl_event_source base;
	wl_event_loop_idle_func_t func;
};

struct wl_event_source_interface idle_source_interface = {
284
	NULL,
285 286 287 288 289 290 291 292 293 294 295 296 297 298 299
};

WL_EXPORT struct wl_event_source *
wl_event_loop_add_idle(struct wl_event_loop *loop,
		       wl_event_loop_idle_func_t func,
		       void *data)
{
	struct wl_event_source_idle *source;

	source = malloc(sizeof *source);
	if (source == NULL)
		return NULL;

	source->base.interface = &idle_source_interface;
	source->base.loop = loop;
300
	source->base.fd = -1;
301 302

	source->func = func;
303
	source->base.data = data;
304 305

	wl_list_insert(loop->idle_list.prev, &source->base.link);
306 307 308 309

	return &source->base;
}

310
WL_EXPORT void
311
wl_event_source_check(struct wl_event_source *source)
312
{
313
	wl_list_insert(source->loop->check_list.prev, &source->link);
314 315
}

316 317 318
WL_EXPORT int
wl_event_source_remove(struct wl_event_source *source)
{
319 320
	struct wl_event_loop *loop = source->loop;

321 322 323 324
	/* We need to explicitly remove the fd, since closing the fd
	 * isn't enough in case we've dup'ed the fd. */
	if (source->fd >= 0) {
		epoll_ctl(loop->epoll_fd, EPOLL_CTL_DEL, source->fd, NULL);
325
		close(source->fd);
326 327
		source->fd = -1;
	}
328

329
	wl_list_remove(&source->link);
330
	wl_list_insert(&loop->destroy_list, &source->link);
331 332

	return 0;
333 334
}

335 336 337 338 339 340 341 342 343 344 345
static void
wl_event_loop_process_destroy_list(struct wl_event_loop *loop)
{
	struct wl_event_source *source, *next;

	wl_list_for_each_safe(source, next, &loop->destroy_list, link)
		free(source);

	wl_list_init(&loop->destroy_list);
}

346
WL_EXPORT struct wl_event_loop *
Kristian Høgsberg's avatar
Kristian Høgsberg committed
347 348 349 350 351 352 353 354
wl_event_loop_create(void)
{
	struct wl_event_loop *loop;

	loop = malloc(sizeof *loop);
	if (loop == NULL)
		return NULL;

Pekka Paalanen's avatar
Pekka Paalanen committed
355
	loop->epoll_fd = wl_os_epoll_create_cloexec();
Kristian Høgsberg's avatar
Kristian Høgsberg committed
356 357 358 359
	if (loop->epoll_fd < 0) {
		free(loop);
		return NULL;
	}
360
	wl_list_init(&loop->check_list);
361
	wl_list_init(&loop->idle_list);
362
	wl_list_init(&loop->destroy_list);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
363

364 365
	wl_signal_init(&loop->destroy_signal);

Kristian Høgsberg's avatar
Kristian Høgsberg committed
366 367 368
	return loop;
}

369
WL_EXPORT void
Kristian Høgsberg's avatar
Kristian Høgsberg committed
370 371
wl_event_loop_destroy(struct wl_event_loop *loop)
{
372 373
	wl_signal_emit(&loop->destroy_signal, loop);

374
	wl_event_loop_process_destroy_list(loop);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
375 376 377 378
	close(loop->epoll_fd);
	free(loop);
}

379 380 381 382 383 384 385 386 387 388 389 390 391 392 393
static int
post_dispatch_check(struct wl_event_loop *loop)
{
	struct epoll_event ep;
	struct wl_event_source *source, *next;
	int n;

	ep.events = 0;
	n = 0;
	wl_list_for_each_safe(source, next, &loop->check_list, link)
		n += source->interface->dispatch(source, &ep);

	return n;
}

394 395
WL_EXPORT void
wl_event_loop_dispatch_idle(struct wl_event_loop *loop)
396
{
397
	struct wl_event_source_idle *source;
398

399 400 401
	while (!wl_list_empty(&loop->idle_list)) {
		source = container_of(loop->idle_list.next,
				      struct wl_event_source_idle, base.link);
402 403 404 405 406
		source->func(source->base.data);
		wl_event_source_remove(&source->base);
	}
}

407
WL_EXPORT int
408
wl_event_loop_dispatch(struct wl_event_loop *loop, int timeout)
Kristian Høgsberg's avatar
Kristian Høgsberg committed
409 410
{
	struct epoll_event ep[32];
411 412
	struct wl_event_source *source;
	int i, count, n;
413

414
	wl_event_loop_dispatch_idle(loop);
415

416
	count = epoll_wait(loop->epoll_fd, ep, ARRAY_LENGTH(ep), timeout);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
417 418
	if (count < 0)
		return -1;
David Herrmann's avatar
David Herrmann committed
419

Kristian Høgsberg's avatar
Kristian Høgsberg committed
420 421
	for (i = 0; i < count; i++) {
		source = ep[i].data.ptr;
422
		if (source->fd != -1)
David Herrmann's avatar
David Herrmann committed
423
			source->interface->dispatch(source, &ep[i]);
Kristian Høgsberg's avatar
Kristian Høgsberg committed
424 425
	}

426 427
	wl_event_loop_process_destroy_list(loop);

428 429
	wl_event_loop_dispatch_idle(loop);

430
	do {
431
		n = post_dispatch_check(loop);
432
	} while (n > 0);
433

Kristian Høgsberg's avatar
Kristian Høgsberg committed
434 435
	return 0;
}
436 437 438 439 440 441

WL_EXPORT int
wl_event_loop_get_fd(struct wl_event_loop *loop)
{
	return loop->epoll_fd;
}
442 443 444 445 446 447 448 449 450 451 452 453

WL_EXPORT void
wl_event_loop_add_destroy_listener(struct wl_event_loop *loop,
				   struct wl_listener *listener)
{
	wl_signal_add(&loop->destroy_signal, listener);
}

WL_EXPORT struct wl_listener *
wl_event_loop_get_destroy_listener(struct wl_event_loop *loop,
				   wl_notify_func_t notify)
{
454
	return wl_signal_get(&loop->destroy_signal, notify);
455 456
}