1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
// SPDX-License-Identifier: ISC
/* $OpenBSD$ */
/*
* Copyright (c) 2012 Claudio Jeker <claudio@openbsd.org>
*/
#include <zebra.h>
#include "ldpd.h"
#include "ldpe.h"
#include "log.h"
struct accept_ev {
LIST_ENTRY(accept_ev) entry;
struct thread *ev;
void (*accept_cb)(struct thread *);
void *arg;
int fd;
};
struct {
LIST_HEAD(, accept_ev) queue;
struct thread *evt;
} accept_queue;
static void accept_arm(void);
static void accept_unarm(void);
static void accept_cb(struct thread *);
static void accept_timeout(struct thread *);
void
accept_init(void)
{
LIST_INIT(&accept_queue.queue);
}
int accept_add(int fd, void (*cb)(struct thread *), void *arg)
{
struct accept_ev *av;
if ((av = calloc(1, sizeof(*av))) == NULL)
return (-1);
av->fd = fd;
av->accept_cb = cb;
av->arg = arg;
LIST_INSERT_HEAD(&accept_queue.queue, av, entry);
thread_add_read(master, accept_cb, av, av->fd, &av->ev);
log_debug("%s: accepting on fd %d", __func__, fd);
return (0);
}
void
accept_del(int fd)
{
struct accept_ev *av;
LIST_FOREACH(av, &accept_queue.queue, entry)
if (av->fd == fd) {
log_debug("%s: %d removed from queue", __func__, fd);
THREAD_OFF(av->ev);
LIST_REMOVE(av, entry);
free(av);
return;
}
}
void
accept_pause(void)
{
log_debug(__func__);
accept_unarm();
thread_add_timer(master, accept_timeout, NULL, 1, &accept_queue.evt);
}
void
accept_unpause(void)
{
if (accept_queue.evt != NULL) {
log_debug(__func__);
THREAD_OFF(accept_queue.evt);
accept_arm();
}
}
static void
accept_arm(void)
{
struct accept_ev *av;
LIST_FOREACH(av, &accept_queue.queue, entry) {
thread_add_read(master, accept_cb, av, av->fd, &av->ev);
}
}
static void
accept_unarm(void)
{
struct accept_ev *av;
LIST_FOREACH(av, &accept_queue.queue, entry)
THREAD_OFF(av->ev);
}
static void accept_cb(struct thread *thread)
{
struct accept_ev *av = THREAD_ARG(thread);
thread_add_read(master, accept_cb, av, av->fd, &av->ev);
av->accept_cb(thread);
}
static void accept_timeout(struct thread *thread)
{
accept_queue.evt = NULL;
log_debug(__func__);
accept_arm();
}
|
