GCOV - Varnish Cache


/*-
 * Copyright (c) 2006 Verdens Gang AS
 * Copyright (c) 2006-2009 Varnish Software AS
 * All rights reserved.
 *
 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include "config.h"

#include <time.h>
#include <poll.h>
#include <pthread.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "vdef.h"
#include "miniobj.h"
#include "vas.h"

#include "vbh.h"
#include "vev.h"
#include "vtim.h"

#undef DEBUG_EVENTS

/* INFTIM indicates an infinite timeout for poll(2) */
#ifndef INFTIM
#define INFTIM -1
#endif

struct vevsig {
        struct vev_root         *vevb;
        struct vev              *vev;
        struct sigaction        sigact;
        unsigned char           happened;
        siginfo_t               siginfo[1];
};

static struct vevsig            *vev_sigs;
static int                      vev_nsig;

struct vev_root {
        unsigned                magic;
#define VEV_BASE_MAGIC          0x477bcf3d
        unsigned                n_fd_events;
        struct pollfd           *pfd;
        struct vev              **pev;
        unsigned                npfd;
        unsigned                lpfd;
        struct vbh              *binheap;
        unsigned                psig;
        pthread_t               thread;
#ifdef DEBUG_EVENTS
        FILE                    *debug;
#endif
};

/*--------------------------------------------------------------------*/

#ifdef DEBUG_EVENTS
#define DBG(evb, ...) do {                              \
        if ((evb)->debug != NULL)                       \
                fprintf((evb)->debug, __VA_ARGS__);     \
        } while (0);
#else
#define DBG(evb, ...)   /* ... */
//#define DBG(evb, ...) fprintf(stderr, __VA_ARGS__);
#endif

/*--------------------------------------------------------------------*/

static void v_matchproto_(vbh_update_t)
vev_bh_update(void *priv, void *a, unsigned u)
{
        struct vev_root *evb;
        struct vev *e;

        CAST_OBJ_NOTNULL(evb, priv, VEV_BASE_MAGIC);
        CAST_OBJ_NOTNULL(e, a, VEV_MAGIC);
        assert(u < evb->lpfd);
        e->__binheap_idx = u;
        if (u != VBH_NOIDX) {
                evb->pev[u] = e;
                evb->pfd[u].fd = e->fd;
                evb->pfd[u].events =
                    e->fd_flags & (VEV__RD|VEV__WR|VEV__ERR|VEV__HUP);
        }
}

static int v_matchproto_(vbh_cmp_t)
vev_bh_cmp(void *priv, const void *a, const void *b)
{
        struct vev_root *evb;
        const struct vev *ea, *eb;

        CAST_OBJ_NOTNULL(evb, priv, VEV_BASE_MAGIC);
        CAST_OBJ_NOTNULL(ea, a, VEV_MAGIC);
        CAST_OBJ_NOTNULL(eb, b, VEV_MAGIC);
        return (ea->__when < eb->__when);
}

/*--------------------------------------------------------------------*/

static int
vev_get_pfd(struct vev_root *evb)
{
        unsigned u;

        if (evb->lpfd + 1 < evb->npfd)
                return (0);

        if (evb->npfd < 8)
                u = 8;
        else if (evb->npfd > 256)
                u = evb->npfd + 256;
        else
                u = evb->npfd * 2;
        evb->npfd = u;
        evb->pfd = realloc(evb->pfd, sizeof(*evb->pfd) * u);
        AN(evb->pfd);
        evb->pev = realloc(evb->pev, sizeof(*evb->pev) * u);
        AN(evb->pev);
        return (0);
}

/*--------------------------------------------------------------------*/

static int
vev_get_sig(int sig)
{
        struct vevsig *os;

        if (sig < vev_nsig)
                return (0);

        os = calloc((sig + 1L), sizeof *os);
        if (os == NULL)
                return (ENOMEM);

        if (vev_sigs != NULL) {
                memcpy(os, vev_sigs, vev_nsig * sizeof *os);
                free(vev_sigs);
        }
        vev_sigs = os;
        vev_nsig = sig + 1;

        return (0);
}

/*--------------------------------------------------------------------*/

static void
vev_sigaction(int sig, siginfo_t *siginfo, void *ctx)
{
        struct vevsig *es;

        (void)ctx;
        assert(sig < vev_nsig);
        assert(vev_sigs != NULL);
        es = &vev_sigs[sig];
        if (!es->happened) {
                es->vevb->psig++;
                memcpy(es->siginfo, siginfo, sizeof *es->siginfo);
                es->vev->siginfo = es->siginfo;
        }
        es->happened = 1;
}

/*--------------------------------------------------------------------*/

struct vev_root *
VEV_New(void)
{
        struct vev_root *evb;

        evb = calloc(1, sizeof *evb);
        if (evb == NULL)
                return (evb);
        evb->lpfd = VBH_NOIDX + 1;
        if (vev_get_pfd(evb)) {
                free(evb);
                return (NULL);
        }
        evb->magic = VEV_BASE_MAGIC;
        evb->binheap = VBH_new(evb, vev_bh_cmp, vev_bh_update);
        evb->thread = pthread_self();
#ifdef DEBUG_EVENTS
        evb->debug = fopen("/tmp/_.events", "w");
        AN(evb->debug);
        setbuf(evb->debug, NULL);
        DBG(evb, "\n\nStart debugging\n");
#endif
        return (evb);
}

/*--------------------------------------------------------------------*/

void
VEV_Destroy(struct vev_root **evbp)
{
        struct vev_root *evb;
        struct vev *e;

        TAKE_OBJ_NOTNULL(evb, evbp, VEV_BASE_MAGIC);
        assert(pthread_equal(evb->thread, pthread_self()));
        while ((e = VBH_root(evb->binheap)) != NULL) {
                VEV_Stop(evb, e);
                free(e);
        }
        VBH_destroy(&evb->binheap);
        free(evb->pfd);
        free(evb->pev);
        FREE_OBJ(evb);
}

/*--------------------------------------------------------------------*/

struct vev *
VEV_Alloc(void)
{
        struct vev *e;

        e = calloc(1, sizeof *e);
        if (e != NULL) {
                e->fd = -1;
        }
        return (e);
}

/*--------------------------------------------------------------------*/

int
VEV_Start(struct vev_root *evb, struct vev *e)
{
        struct vevsig *es;

        CHECK_OBJ_NOTNULL(evb, VEV_BASE_MAGIC);
        assert(e->magic != VEV_MAGIC);
        assert(e->callback != NULL);
        assert(e->sig >= 0);
        assert(e->timeout >= 0.0);
        assert(e->fd < 0 || e->fd_flags);
        assert(pthread_equal(evb->thread, pthread_self()));
        DBG(evb, "ev_add(%p) fd = %d\n", e, e->fd);

        if (vev_get_pfd(evb))
                return (ENOMEM);

        if (e->sig > 0) {
                if (vev_get_sig(e->sig))
                        return (ENOMEM);

                assert(e->fd < 0);
                es = &vev_sigs[e->sig];
                if (es->vev != NULL)
                        return (EBUSY);
                AZ(es->happened);
                es->vev = e;
                es->vevb = evb;
                es->sigact.sa_flags = e->sig_flags | SA_SIGINFO;
                es->sigact.sa_sigaction = vev_sigaction;
        } else {
                es = NULL;
        }

        e->magic = VEV_MAGIC;   /* before VBH_insert() */

        if (e->timeout != 0.0)
                e->__when += VTIM_mono() + e->timeout;
        else
                e->__when = 9e99;

        evb->lpfd++;
        VBH_insert(evb->binheap, e);
        assert(e->__binheap_idx != VBH_NOIDX);

        e->__vevb = evb;
        e->__privflags = 0;

        if (e->sig > 0) {
                assert(es != NULL);
                AZ(sigaction(e->sig, &es->sigact, NULL));
        }

        return (0);
}

/*--------------------------------------------------------------------*/

void
VEV_Stop(struct vev_root *evb, struct vev *e)
{
        struct vevsig *es;

        CHECK_OBJ_NOTNULL(evb, VEV_BASE_MAGIC);
        CHECK_OBJ_NOTNULL(e, VEV_MAGIC);
        DBG(evb, "ev_del(%p) fd = %d i=%u L=%d\n", e, e->fd, e->__binheap_idx, evb->lpfd);
        assert(evb == e->__vevb);
        assert(pthread_equal(evb->thread, pthread_self()));
        assert(evb->pev[e->__binheap_idx] == e);

        assert(e->__binheap_idx != VBH_NOIDX);
        e->fd = -1;
        VBH_delete(evb->binheap, e->__binheap_idx);
        assert(e->__binheap_idx == VBH_NOIDX);
        evb->lpfd--;

        if (e->fd_events) {
                assert(evb->n_fd_events > 0);
                evb->n_fd_events--;
                e->fd_events = 0;
        }

        if (e->sig > 0) {
                assert(e->sig < vev_nsig);
                es = &vev_sigs[e->sig];
                assert(es->vev == e);
                es->vev = NULL;
                es->vevb = NULL;
                es->sigact.sa_flags = 0;
                es->sigact.sa_handler = SIG_DFL;
                AZ(sigaction(e->sig, &es->sigact, NULL));
                es->happened = 0;
        }

        e->__vevb = NULL;
        FINI_OBJ(e);
}

/*--------------------------------------------------------------------*/

int
VEV_Loop(struct vev_root *evb)
{
        int i;

        CHECK_OBJ_NOTNULL(evb, VEV_BASE_MAGIC);
        assert(pthread_equal(evb->thread, pthread_self()));
        do
                i = VEV_Once(evb);
        while (i == 1);
        return (i);
}

/*--------------------------------------------------------------------*/

static int
vev_sched_timeout(struct vev_root *evb, struct vev *e, vtim_mono t)
{
        int i;


        i = e->callback(e, 0);
        if (i) {
                VEV_Stop(evb, e);
                free(e);
        } else {
                e->__when = t + e->timeout;
                VBH_delete(evb->binheap, e->__binheap_idx);
                VBH_insert(evb->binheap, e);
        }
        return (1);
}

static int
vev_sched_signal(struct vev_root *evb)
{
        int i, j, retval = 1;
        struct vevsig *es;
        struct vev *e;

        es = vev_sigs;
        for (j = 0; j < vev_nsig; j++, es++) {
                if (!es->happened || es->vevb != evb)
                        continue;
                evb->psig--;
                es->happened = 0;
                e = es->vev;
                assert(e != NULL);
                i = e->callback(e, VEV__SIG);
                if (e->siginfo != NULL) {
                        e->siginfo = NULL;
                        memset(es->siginfo, 0, sizeof *es->siginfo);
                }
                if (i) {
                        VEV_Stop(evb, e);
                        free(e);
                }
                if (i < 0)
                        retval = i;
        }
        return (retval);
}

int
VEV_Once(struct vev_root *evb)
{
        double t;
        struct vev *e;
        int i, k, tmo, retval = 1;
        unsigned u;
        int progress;

        CHECK_OBJ_NOTNULL(evb, VEV_BASE_MAGIC);
        assert(pthread_equal(evb->thread, pthread_self()));
        assert(evb->lpfd < evb->npfd);

        if (evb->psig)
                return (vev_sched_signal(evb));

        tmo = INFTIM;
        e = VBH_root(evb->binheap);
        if (e != NULL) {
                CHECK_OBJ(e, VEV_MAGIC);
                assert(e->__binheap_idx == VBH_NOIDX + 1);
                t = VTIM_mono();
                if (e->__when <= t)
                        return (vev_sched_timeout(evb, e, t));
                if (e->__when < 9e99)
                        tmo = (int)((e->__when - t) * 1e3);
                if (tmo == 0)
                        tmo = 1;
        }

        if (tmo == INFTIM && evb->lpfd == VBH_NOIDX + 1)
                return (0);

        i = poll(evb->pfd + 1, evb->lpfd - 1, tmo);
        if (i == -1 && errno == EINTR)
                return (vev_sched_signal(evb));
        if (i == -1)
                return (-1);

        if (i == 0) {
                assert(e != NULL);
                t = VTIM_mono();
                if (e->__when <= t)
                        return (vev_sched_timeout(evb, e, t));
        }

        AZ(evb->n_fd_events);
        for (u = 1; u < evb->lpfd; u++) {
                AZ(evb->pev[u]->fd_events);
                evb->pev[u]->fd_events = evb->pfd[u].revents;
                if (evb->pev[u]->fd_events)
                        evb->n_fd_events++;
        }
        assert(evb->n_fd_events == i);

        DBG(evb, "EVENTS %d\n", i);
        while (evb->n_fd_events > 0) {
                progress = 0;
                for (u = VBH_NOIDX + 1; u < evb->lpfd; u++) {
                        e = evb->pev[u];
                        if (e->fd_events == 0)
                                continue;
                        DBG(evb, "EVENT %p u=%u fd=%d ev=0x%x %d\n",
                            e, u, e->fd, e->fd_events, i);
                        k = e->callback(e, e->fd_events);
                        e->fd_events = 0;
                        assert(evb->n_fd_events > 0);
                        evb->n_fd_events--;
                        progress++;
                        if (k) {
                                VEV_Stop(evb, e);
                                free(e);
                        }
                        if (k < 0)
                                retval = k;
                }
                assert(progress > 0);
        }

        return (retval);
}

		varnish-cache/lib/libvarnish/vev.c
0		/*-
1		* Copyright (c) 2006 Verdens Gang AS
2		* Copyright (c) 2006-2009 Varnish Software AS
3		* All rights reserved.
4		*
5		* Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
6		*
7		* SPDX-License-Identifier: BSD-2-Clause
8		*
9		* Redistribution and use in source and binary forms, with or without
10		* modification, are permitted provided that the following conditions
11		* are met:
12		* 1. Redistributions of source code must retain the above copyright
13		* notice, this list of conditions and the following disclaimer.
14		* 2. Redistributions in binary form must reproduce the above copyright
15		* notice, this list of conditions and the following disclaimer in the
16		* documentation and/or other materials provided with the distribution.
17		*
18		* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19		* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20		* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21		* ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
22		* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23		* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24		* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25		* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26		* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27		* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28		* SUCH DAMAGE.
29		*/
30
31		#include "config.h"
32
33		#include <time.h>
34		#include <poll.h>
35		#include <pthread.h>
36		#include <signal.h>
37		#include <stdio.h>
38		#include <stdlib.h>
39		#include <string.h>
40
41		#include "vdef.h"
42		#include "miniobj.h"
43		#include "vas.h"
44
45		#include "vbh.h"
46		#include "vev.h"
47		#include "vtim.h"
48
49		#undef DEBUG_EVENTS
50
51		/* INFTIM indicates an infinite timeout for poll(2) */
52		#ifndef INFTIM
53		#define INFTIM -1
54		#endif
55
56		struct vevsig {
57		struct vev_root *vevb;
58		struct vev *vev;
59		struct sigaction sigact;
60		unsigned char happened;
61		siginfo_t siginfo[1];
62		};
63
64		static struct vevsig *vev_sigs;
65		static int vev_nsig;
66
67		struct vev_root {
68		unsigned magic;
69		#define VEV_BASE_MAGIC 0x477bcf3d
70		unsigned n_fd_events;
71		struct pollfd *pfd;
72		struct vev **pev;
73		unsigned npfd;
74		unsigned lpfd;
75		struct vbh *binheap;
76		unsigned psig;
77		pthread_t thread;
78		#ifdef DEBUG_EVENTS
79		FILE *debug;
80		#endif
81		};
82
83		/--------------------------------------------------------------------/
84
85		#ifdef DEBUG_EVENTS
86		#define DBG(evb, ...) do { \
87		if ((evb)->debug != NULL) \
88		fprintf((evb)->debug, __VA_ARGS__); \
89		} while (0);
90		#else
91		#define DBG(evb, ...) /* ... */
92		//#define DBG(evb, ...) fprintf(stderr, __VA_ARGS__);
93		#endif
94
95		/--------------------------------------------------------------------/
96
97		static void v_matchproto_(vbh_update_t)
98	373538	vev_bh_update(void priv, void a, unsigned u)
99		{
100		struct vev_root *evb;
101		struct vev *e;
102
103	373538	CAST_OBJ_NOTNULL(evb, priv, VEV_BASE_MAGIC);
104	373538	CAST_OBJ_NOTNULL(e, a, VEV_MAGIC);
105	373538	assert(u < evb->lpfd);
106	373538	e->__binheap_idx = u;
107	373538	if (u != VBH_NOIDX) {
108	315191	evb->pev[u] = e;
109	315191	evb->pfd[u].fd = e->fd;
110	315191	evb->pfd[u].events =
111	315191	e->fd_flags & (VEV__RD\|VEV__WR\|VEV__ERR\|VEV__HUP);
112	315191	}
113	373538	}
114
115		static int v_matchproto_(vbh_cmp_t)
116	214769	vev_bh_cmp(void priv, const void a, const void *b)
117		{
118		struct vev_root *evb;
119		const struct vev ea, eb;
120
121	214769	CAST_OBJ_NOTNULL(evb, priv, VEV_BASE_MAGIC);
122	214769	CAST_OBJ_NOTNULL(ea, a, VEV_MAGIC);
123	214769	CAST_OBJ_NOTNULL(eb, b, VEV_MAGIC);
124	214769	return (ea->__when < eb->__when);
125		}
126
127		/--------------------------------------------------------------------/
128
129		static int
130	51477	vev_get_pfd(struct vev_root *evb)
131		{
132		unsigned u;
133
134	51477	if (evb->lpfd + 1 < evb->npfd)
135	39486	return (0);
136
137	11991	if (evb->npfd < 8)
138	9159	u = 8;
139	2832	else if (evb->npfd > 256)
140	0	u = evb->npfd + 256;
141		else
142	2832	u = evb->npfd * 2;
143	11991	evb->npfd = u;
144	11991	evb->pfd = realloc(evb->pfd, sizeof(evb->pfd) u);
145	11991	AN(evb->pfd);
146	11991	evb->pev = realloc(evb->pev, sizeof(evb->pev) u);
147	11991	AN(evb->pev);
148	11991	return (0);
149	51477	}
150
151		/--------------------------------------------------------------------/
152
153		static int
154	5622	vev_get_sig(int sig)
155		{
156		struct vevsig *os;
157
158	5622	if (sig < vev_nsig)
159	2811	return (0);
160
161	2811	os = calloc((sig + 1L), sizeof *os);
162	2811	if (os == NULL)
163	0	return (ENOMEM);
164
165	2811	if (vev_sigs != NULL) {
166	0	memcpy(os, vev_sigs, vev_nsig * sizeof *os);
167	0	free(vev_sigs);
168	0	}
169	2811	vev_sigs = os;
170	2811	vev_nsig = sig + 1;
171
172	2811	return (0);
173	5622	}
174
175		/--------------------------------------------------------------------/
176
177		static void
178	15	vev_sigaction(int sig, siginfo_t siginfo, void ctx)
179		{
180		struct vevsig *es;
181
182	15	(void)ctx;
183	15	assert(sig < vev_nsig);
184	15	assert(vev_sigs != NULL);
185	15	es = &vev_sigs[sig];
186	15	if (!es->happened) {
187	15	es->vevb->psig++;
188	15	memcpy(es->siginfo, siginfo, sizeof *es->siginfo);
189	15	es->vev->siginfo = es->siginfo;
190	15	}
191	15	es->happened = 1;
192	15	}
193
194		/--------------------------------------------------------------------/
195
196		struct vev_root *
197	9159	VEV_New(void)
198		{
199		struct vev_root *evb;
200
201	9159	evb = calloc(1, sizeof *evb);
202	9159	if (evb == NULL)
203	0	return (evb);
204	9159	evb->lpfd = VBH_NOIDX + 1;
205	9159	if (vev_get_pfd(evb)) {
206	0	free(evb);
207	0	return (NULL);
208		}
209	9159	evb->magic = VEV_BASE_MAGIC;
210	9159	evb->binheap = VBH_new(evb, vev_bh_cmp, vev_bh_update);
211	9159	evb->thread = pthread_self();
212		#ifdef DEBUG_EVENTS
213		evb->debug = fopen("/tmp/_.events", "w");
214		AN(evb->debug);
215		setbuf(evb->debug, NULL);
216		DBG(evb, "\n\nStart debugging\n");
217		#endif
218	9159	return (evb);
219	9159	}
220
221		/--------------------------------------------------------------------/
222
223		void
224	3015	VEV_Destroy(struct vev_root **evbp)
225		{
226		struct vev_root *evb;
227		struct vev *e;
228
229	3015	TAKE_OBJ_NOTNULL(evb, evbp, VEV_BASE_MAGIC);
230	3015	assert(pthread_equal(evb->thread, pthread_self()));
231	22904	while ((e = VBH_root(evb->binheap)) != NULL) {
232	19889	VEV_Stop(evb, e);
233	19889	free(e);
234		}
235	3015	VBH_destroy(&evb->binheap);
236	3015	free(evb->pfd);
237	3015	free(evb->pev);
238	3015	FREE_OBJ(evb);
239	3015	}
240
241		/--------------------------------------------------------------------/
242
243		struct vev *
244	42318	VEV_Alloc(void)
245		{
246		struct vev *e;
247
248	42318	e = calloc(1, sizeof *e);
249	42318	if (e != NULL) {
250	42318	e->fd = -1;
251	42318	}
252	42318	return (e);
253		}
254
255		/--------------------------------------------------------------------/
256
257		int
258	42318	VEV_Start(struct vev_root evb, struct vev e)
259		{
260		struct vevsig *es;
261
262	42318	CHECK_OBJ_NOTNULL(evb, VEV_BASE_MAGIC);
263	42318	assert(e->magic != VEV_MAGIC);
264	42318	assert(e->callback != NULL);
265	42318	assert(e->sig >= 0);
266	42318	assert(e->timeout >= 0.0);
267	42318	assert(e->fd < 0 \|\| e->fd_flags);
268	42318	assert(pthread_equal(evb->thread, pthread_self()));
269		DBG(evb, "ev_add(%p) fd = %d\n", e, e->fd);
270
271	42318	if (vev_get_pfd(evb))
272	0	return (ENOMEM);
273
274	42318	if (e->sig > 0) {
275	5622	if (vev_get_sig(e->sig))
276	0	return (ENOMEM);
277
278	5622	assert(e->fd < 0);
279	5622	es = &vev_sigs[e->sig];
280	5622	if (es->vev != NULL)
281	0	return (EBUSY);
282	5622	AZ(es->happened);
283	5622	es->vev = e;
284	5622	es->vevb = evb;
285	5622	es->sigact.sa_flags = e->sig_flags \| SA_SIGINFO;
286	5622	es->sigact.sa_sigaction = vev_sigaction;
287	5622	} else {
288	36696	es = NULL;
289		}
290
291	42318	e->magic = VEV_MAGIC; /* before VBH_insert() */
292
293	42318	if (e->timeout != 0.0)
294	15273	e->__when += VTIM_mono() + e->timeout;
295		else
296	27045	e->__when = 9e99;
297
298	42318	evb->lpfd++;
299	42318	VBH_insert(evb->binheap, e);
300	42318	assert(e->__binheap_idx != VBH_NOIDX);
301
302	42318	e->__vevb = evb;
303	42318	e->__privflags = 0;
304
305	42318	if (e->sig > 0) {
306	5622	assert(es != NULL);
307	5622	AZ(sigaction(e->sig, &es->sigact, NULL));
308	5622	}
309
310	42318	return (0);
311	42318	}
312
313		/--------------------------------------------------------------------/
314
315		void
316	42087	VEV_Stop(struct vev_root evb, struct vev e)
317		{
318		struct vevsig *es;
319
320	42087	CHECK_OBJ_NOTNULL(evb, VEV_BASE_MAGIC);
321	42087	CHECK_OBJ_NOTNULL(e, VEV_MAGIC);
322		DBG(evb, "ev_del(%p) fd = %d i=%u L=%d\n", e, e->fd, e->__binheap_idx, evb->lpfd);
323	42087	assert(evb == e->__vevb);
324	42087	assert(pthread_equal(evb->thread, pthread_self()));
325	42087	assert(evb->pev[e->__binheap_idx] == e);
326
327	42087	assert(e->__binheap_idx != VBH_NOIDX);
328	42087	e->fd = -1;
329	42087	VBH_delete(evb->binheap, e->__binheap_idx);
330	42087	assert(e->__binheap_idx == VBH_NOIDX);
331	42087	evb->lpfd--;
332
333	42087	if (e->fd_events) {
334	0	assert(evb->n_fd_events > 0);
335	0	evb->n_fd_events--;
336	0	e->fd_events = 0;
337	0	}
338
339	42087	if (e->sig > 0) {
340	5622	assert(e->sig < vev_nsig);
341	5622	es = &vev_sigs[e->sig];
342	5622	assert(es->vev == e);
343	5622	es->vev = NULL;
344	5622	es->vevb = NULL;
345	5622	es->sigact.sa_flags = 0;
346	5622	es->sigact.sa_handler = SIG_DFL;
347	5622	AZ(sigaction(e->sig, &es->sigact, NULL));
348	5622	es->happened = 0;
349	5622	}
350
351	42087	e->__vevb = NULL;
352	42087	FINI_OBJ(e);
353	42087	}
354
355		/--------------------------------------------------------------------/
356
357		int
358	2811	VEV_Loop(struct vev_root *evb)
359		{
360		int i;
361
362	2811	CHECK_OBJ_NOTNULL(evb, VEV_BASE_MAGIC);
363	2811	assert(pthread_equal(evb->thread, pthread_self()));
364	2811	do
365	102961	i = VEV_Once(evb);
366	102961	while (i == 1);
367	2811	return (i);
368		}
369
370		/--------------------------------------------------------------------/
371
372		static int
373	16260	vev_sched_timeout(struct vev_root evb, struct vev e, vtim_mono t)
374		{
375		int i;
376
377
378	16260	i = e->callback(e, 0);
379	16260	if (i) {
380	0	VEV_Stop(evb, e);
381	0	free(e);
382	0	} else {
383	16260	e->__when = t + e->timeout;
384	16260	VBH_delete(evb->binheap, e->__binheap_idx);
385	16260	VBH_insert(evb->binheap, e);
386		}
387	16260	return (1);
388		}
389
390		static int
391	12	vev_sched_signal(struct vev_root *evb)
392		{
393	12	int i, j, retval = 1;
394		struct vevsig *es;
395		struct vev *e;
396
397	12	es = vev_sigs;
398	204	for (j = 0; j < vev_nsig; j++, es++) {
399	192	if (!es->happened \|\| es->vevb != evb)
400	180	continue;
401	12	evb->psig--;
402	12	es->happened = 0;
403	12	e = es->vev;
404	12	assert(e != NULL);
405	12	i = e->callback(e, VEV__SIG);
406	12	if (e->siginfo != NULL) {
407	12	e->siginfo = NULL;
408	12	memset(es->siginfo, 0, sizeof *es->siginfo);
409	12	}
410	12	if (i) {
411	12	VEV_Stop(evb, e);
412	12	free(e);
413	12	}
414	12	if (i < 0)
415	12	retval = i;
416	12	}
417	12	return (retval);
418		}
419
420		int
421	123075	VEV_Once(struct vev_root *evb)
422		{
423		double t;
424		struct vev *e;
425	123075	int i, k, tmo, retval = 1;
426		unsigned u;
427		int progress;
428
429	123075	CHECK_OBJ_NOTNULL(evb, VEV_BASE_MAGIC);
430	123075	assert(pthread_equal(evb->thread, pthread_self()));
431	123075	assert(evb->lpfd < evb->npfd);
432
433	123075	if (evb->psig)
434	0	return (vev_sched_signal(evb));
435
436	123075	tmo = INFTIM;
437	123075	e = VBH_root(evb->binheap);
438	123075	if (e != NULL) {
439	119967	CHECK_OBJ(e, VEV_MAGIC);
440	119967	assert(e->__binheap_idx == VBH_NOIDX + 1);
441	119967	t = VTIM_mono();
442	119967	if (e->__when <= t)
443	3972	return (vev_sched_timeout(evb, e, t));
444	115995	if (e->__when < 9e99)
445	102895	tmo = (int)((e->__when - t) * 1e3);
446	115995	if (tmo == 0)
447	628	tmo = 1;
448	115995	}
449
450	119103	if (tmo == INFTIM && evb->lpfd == VBH_NOIDX + 1)
451	3123	return (0);
452
453	115980	i = poll(evb->pfd + 1, evb->lpfd - 1, tmo);
454	115980	if (i == -1 && errno == EINTR)
455	12	return (vev_sched_signal(evb));
456	115968	if (i == -1)
457	0	return (-1);
458
459	115968	if (i == 0) {
460	12817	assert(e != NULL);
461	12817	t = VTIM_mono();
462	12817	if (e->__when <= t)
463	12288	return (vev_sched_timeout(evb, e, t));
464	529	}
465
466	103680	AZ(evb->n_fd_events);
467	998411	for (u = 1; u < evb->lpfd; u++) {
468	894731	AZ(evb->pev[u]->fd_events);
469	894731	evb->pev[u]->fd_events = evb->pfd[u].revents;
470	894731	if (evb->pev[u]->fd_events)
471	105953	evb->n_fd_events++;
472	894731	}
473	103680	assert(evb->n_fd_events == i);
474
475		DBG(evb, "EVENTS %d\n", i);
476	207084	while (evb->n_fd_events > 0) {
477	103404	progress = 0;
478	989733	for (u = VBH_NOIDX + 1; u < evb->lpfd; u++) {
479	886329	e = evb->pev[u];
480	886329	if (e->fd_events == 0)
481	780391	continue;
482		DBG(evb, "EVENT %p u=%u fd=%d ev=0x%x %d\n",
483		e, u, e->fd, e->fd_events, i);
484	105938	k = e->callback(e, e->fd_events);
485	105938	e->fd_events = 0;
486	105938	assert(evb->n_fd_events > 0);
487	105938	evb->n_fd_events--;
488	105938	progress++;
489	105938	if (k) {
490	12655	VEV_Stop(evb, e);
491	12655	free(e);
492	12655	}
493	105938	if (k < 0)
494	2799	retval = k;
495	105938	}
496	103404	assert(progress > 0);
497		}
498
499	103680	return (retval);