1 /* $OpenBSD: accept.c,v 1.7 2016/05/23 19:11:42 renato Exp $ */
2
3 /*
4 * Copyright (c) 2012 Claudio Jeker <claudio@openbsd.org>
5 *
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 */
18
19 #include <sys/types.h>
20 #include <stdlib.h>
21
22 #include "ldpd.h"
23 #include "ldpe.h"
24 #include "log.h"
25
26 struct accept_ev {
27 LIST_ENTRY(accept_ev) entry;
28 struct event ev;
29 void (*accept_cb)(int, short, void *);
30 void *arg;
31 int fd;
32 };
33
34 struct {
35 LIST_HEAD(, accept_ev) queue;
36 struct event evt;
37 } accept_queue;
38
39 static void accept_arm(void);
40 static void accept_unarm(void);
41 static void accept_cb(int, short, void *);
42 static void accept_timeout(int, short, void *);
43
44 void
accept_init(void)45 accept_init(void)
46 {
47 LIST_INIT(&accept_queue.queue);
48 evtimer_set(&accept_queue.evt, accept_timeout, NULL);
49 }
50
51 int
accept_add(int fd,void (* cb)(int,short,void *),void * arg)52 accept_add(int fd, void (*cb)(int, short, void *), void *arg)
53 {
54 struct accept_ev *av;
55
56 if ((av = calloc(1, sizeof(*av))) == NULL)
57 return (-1);
58 av->fd = fd;
59 av->accept_cb = cb;
60 av->arg = arg;
61 LIST_INSERT_HEAD(&accept_queue.queue, av, entry);
62
63 event_set(&av->ev, av->fd, EV_READ, accept_cb, av);
64 event_add(&av->ev, NULL);
65
66 log_debug("%s: accepting on fd %d", __func__, fd);
67
68 return (0);
69 }
70
71 void
accept_del(int fd)72 accept_del(int fd)
73 {
74 struct accept_ev *av;
75
76 LIST_FOREACH(av, &accept_queue.queue, entry)
77 if (av->fd == fd) {
78 log_debug("%s: %d removed from queue", __func__, fd);
79 event_del(&av->ev);
80 LIST_REMOVE(av, entry);
81 free(av);
82 return;
83 }
84 }
85
86 void
accept_pause(void)87 accept_pause(void)
88 {
89 struct timeval evtpause = { 1, 0 };
90
91 log_debug(__func__);
92 accept_unarm();
93 evtimer_add(&accept_queue.evt, &evtpause);
94 }
95
96 void
accept_unpause(void)97 accept_unpause(void)
98 {
99 if (evtimer_pending(&accept_queue.evt, NULL)) {
100 log_debug(__func__);
101 evtimer_del(&accept_queue.evt);
102 accept_arm();
103 }
104 }
105
106 static void
accept_arm(void)107 accept_arm(void)
108 {
109 struct accept_ev *av;
110 LIST_FOREACH(av, &accept_queue.queue, entry)
111 event_add(&av->ev, NULL);
112 }
113
114 static void
accept_unarm(void)115 accept_unarm(void)
116 {
117 struct accept_ev *av;
118 LIST_FOREACH(av, &accept_queue.queue, entry)
119 event_del(&av->ev);
120 }
121
122 static void
accept_cb(int fd,short event,void * arg)123 accept_cb(int fd, short event, void *arg)
124 {
125 struct accept_ev *av = arg;
126 event_add(&av->ev, NULL);
127 av->accept_cb(fd, event, av->arg);
128 }
129
130 static void
accept_timeout(int fd,short event,void * bula)131 accept_timeout(int fd, short event, void *bula)
132 {
133 log_debug(__func__);
134 accept_arm();
135 }
136