GCOV - Varnish Cache


/*-
 * Copyright (c) 2006 Verdens Gang AS
 * Copyright (c) 2006-2011 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"

#if defined(HAVE_KQUEUE)

#include "cache/cache_varnishd.h"

#include <sys/event.h>

#include <stdlib.h>

#include "waiter/waiter.h"
#include "waiter/waiter_priv.h"
#include "vtim.h"

#define NKEV    256

struct vwk {
        unsigned                magic;
#define VWK_MAGIC               0x1cc2acc2
        int                     kq;
        struct waiter           *waiter;
        pthread_t               thread;
        double                  next;
        int                     pipe[2];
        unsigned                nwaited;
        int                     die;
        struct lock             mtx;
};

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

static void *
vwk_thread(void *priv)
{
        struct vwk *vwk;
        struct kevent ke[NKEV], *kp;
        int j, n;
        double now, then;
        struct timespec ts;
        struct waited *wp;
        struct waiter *w;
        char c;

        CAST_OBJ_NOTNULL(vwk, priv, VWK_MAGIC);
        w = vwk->waiter;
        CHECK_OBJ_NOTNULL(w, WAITER_MAGIC);
        THR_SetName("cache-kqueue");
        THR_Init();

        now = VTIM_real();
        while (1) {
                while (1) {
                        Lck_Lock(&vwk->mtx);
                        /*
                         * XXX: We could avoid many syscalls here if we were
                         * XXX: allowed to just close the fd's on timeout.
                         */
                        then = Wait_HeapDue(w, &wp);
                        if (wp == NULL) {
                                vwk->next = now + 100;
                                break;
                        } else if (then > now) {
                                vwk->next = then;
                                break;
                        }
                        CHECK_OBJ_NOTNULL(wp, WAITED_MAGIC);
                        EV_SET(ke, wp->fd, EVFILT_READ, EV_DELETE, 0, 0, NULL);
                        AZ(kevent(vwk->kq, ke, 1, NULL, 0, NULL));
                        AN(Wait_HeapDelete(w, wp));
                        Lck_Unlock(&vwk->mtx);
                        Wait_Call(w, wp, WAITER_TIMEOUT, now);
                }
                then = vwk->next - now;
                ts = VTIM_timespec(then);
                Lck_Unlock(&vwk->mtx);
                n = kevent(vwk->kq, NULL, 0, ke, NKEV, &ts);
                assert(n >= 0);
                assert(n <= NKEV);
                now = VTIM_real();
                for (kp = ke, j = 0; j < n; j++, kp++) {
                        assert(kp->filter == EVFILT_READ);
                        if ((uintptr_t)ke[j].udata == (uintptr_t)vwk) {
                                assert(read(vwk->pipe[0], &c, 1) == 1);
                                continue;
                        }
                        CAST_OBJ_NOTNULL(wp, (void*)ke[j].udata, WAITED_MAGIC);
                        Lck_Lock(&vwk->mtx);
                        AN(Wait_HeapDelete(w, wp));
                        Lck_Unlock(&vwk->mtx);
                        vwk->nwaited--;
                        if (kp->flags & EV_EOF &&
                            recv(wp->fd, &c, 1, MSG_PEEK) == 0)
                                Wait_Call(w, wp, WAITER_REMCLOSE, now);
                        else
                                Wait_Call(w, wp, WAITER_ACTION, now);
                }
                if (vwk->nwaited == 0 && vwk->die)
                        break;
        }
        closefd(&vwk->pipe[0]);
        closefd(&vwk->pipe[1]);
        closefd(&vwk->kq);
        return (NULL);
}

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

static int v_matchproto_(waiter_enter_f)
vwk_enter(void *priv, struct waited *wp)
{
        struct vwk *vwk;
        struct kevent ke;

        CAST_OBJ_NOTNULL(vwk, priv, VWK_MAGIC);
        EV_SET(&ke, wp->fd, EVFILT_READ, EV_ADD|EV_ONESHOT, 0, 0, wp);
        Lck_Lock(&vwk->mtx);
        vwk->nwaited++;
        Wait_HeapInsert(vwk->waiter, wp);
        AZ(kevent(vwk->kq, &ke, 1, NULL, 0, NULL));

        /* If the kqueue isn't due before our timeout, poke it via the pipe */
        if (Wait_When(wp) < vwk->next)
                assert(write(vwk->pipe[1], "X", 1) == 1);

        Lck_Unlock(&vwk->mtx);
        return (0);
}

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

static void v_matchproto_(waiter_init_f)
vwk_init(struct waiter *w)
{
        struct vwk *vwk;
        struct kevent ke;

        CHECK_OBJ_NOTNULL(w, WAITER_MAGIC);
        vwk = w->priv;
        INIT_OBJ(vwk, VWK_MAGIC);
        vwk->waiter = w;

        vwk->kq = kqueue();
        assert(vwk->kq >= 0);
        Lck_New(&vwk->mtx, lck_waiter);
        AZ(pipe(vwk->pipe));
        EV_SET(&ke, vwk->pipe[0], EVFILT_READ, EV_ADD, 0, 0, vwk);
        AZ(kevent(vwk->kq, &ke, 1, NULL, 0, NULL));

        PTOK(pthread_create(&vwk->thread, NULL, vwk_thread, vwk));
}

/*--------------------------------------------------------------------
 * It is the callers responsibility to trigger all fd's waited on to
 * fail somehow.
 */

static void v_matchproto_(waiter_fini_f)
vwk_fini(struct waiter *w)
{
        struct vwk *vwk;
        void *vp;

        CAST_OBJ_NOTNULL(vwk, w->priv, VWK_MAGIC);
        Lck_Lock(&vwk->mtx);
        vwk->die = 1;
        assert(write(vwk->pipe[1], "Y", 1) == 1);
        Lck_Unlock(&vwk->mtx);
        PTOK(pthread_join(vwk->thread, &vp));
        Lck_Delete(&vwk->mtx);
}

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

#include "waiter/mgt_waiter.h"

const struct waiter_impl waiter_kqueue = {
        .name =         "kqueue",
        .init =         vwk_init,
        .fini =         vwk_fini,
        .enter =        vwk_enter,
        .size =         sizeof(struct vwk),
};

#endif /* defined(HAVE_KQUEUE) */

		varnish-cache/bin/varnishd/waiter/cache_waiter_kqueue.c
0		/*-
1		* Copyright (c) 2006 Verdens Gang AS
2		* Copyright (c) 2006-2011 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
32		#include "config.h"
33
34		#if defined(HAVE_KQUEUE)
35
36		#include "cache/cache_varnishd.h"
37
38		#include <sys/event.h>
39
40		#include <stdlib.h>
41
42		#include "waiter/waiter.h"
43		#include "waiter/waiter_priv.h"
44		#include "vtim.h"
45
46		#define NKEV 256
47
48		struct vwk {
49		unsigned magic;
50		#define VWK_MAGIC 0x1cc2acc2
51		int kq;
52		struct waiter *waiter;
53		pthread_t thread;
54		double next;
55		int pipe[2];
56		unsigned nwaited;
57		int die;
58		struct lock mtx;
59		};
60
61		/--------------------------------------------------------------------/
62
63		static void *
64	40	vwk_thread(void *priv)
65		{
66		struct vwk *vwk;
67		struct kevent ke[NKEV], *kp;
68		int j, n;
69		double now, then;
70		struct timespec ts;
71		struct waited *wp;
72		struct waiter *w;
73		char c;
74
75	40	CAST_OBJ_NOTNULL(vwk, priv, VWK_MAGIC);
76	40	w = vwk->waiter;
77	40	CHECK_OBJ_NOTNULL(w, WAITER_MAGIC);
78	40	THR_SetName("cache-kqueue");
79	40	THR_Init();
80
81	40	now = VTIM_real();
82	188365	while (1) {
83	188885	while (1) {
84	188885	Lck_Lock(&vwk->mtx);
85		/*
86		* XXX: We could avoid many syscalls here if we were
87		* XXX: allowed to just close the fd's on timeout.
88		*/
89	188885	then = Wait_HeapDue(w, &wp);
90	188885	if (wp == NULL) {
91	148731	vwk->next = now + 100;
92	148731	break;
93	40154	} else if (then > now) {
94	39634	vwk->next = then;
95	39634	break;
96		}
97	520	CHECK_OBJ_NOTNULL(wp, WAITED_MAGIC);
98	520	EV_SET(ke, wp->fd, EVFILT_READ, EV_DELETE, 0, 0, NULL);
99	520	AZ(kevent(vwk->kq, ke, 1, NULL, 0, NULL));
100	520	AN(Wait_HeapDelete(w, wp));
101	520	Lck_Unlock(&vwk->mtx);
102	520	Wait_Call(w, wp, WAITER_TIMEOUT, now);
103		}
104	188365	then = vwk->next - now;
105	188365	ts = VTIM_timespec(then);
106	188365	Lck_Unlock(&vwk->mtx);
107	188365	n = kevent(vwk->kq, NULL, 0, ke, NKEV, &ts);
108	188365	assert(n >= 0);
109	188365	assert(n <= NKEV);
110	188365	now = VTIM_real();
111	332650	for (kp = ke, j = 0; j < n; j++, kp++) {
112	144285	assert(kp->filter == EVFILT_READ);
113	144285	if ((uintptr_t)ke[j].udata == (uintptr_t)vwk) {
114	72105	assert(read(vwk->pipe[0], &c, 1) == 1);
115	72105	continue;
116		}
117	72180	CAST_OBJ_NOTNULL(wp, (void*)ke[j].udata, WAITED_MAGIC);
118	72180	Lck_Lock(&vwk->mtx);
119	72180	AN(Wait_HeapDelete(w, wp));
120	72180	Lck_Unlock(&vwk->mtx);
121	72180	vwk->nwaited--;
122	72180	if (kp->flags & EV_EOF &&
123	46332	recv(wp->fd, &c, 1, MSG_PEEK) == 0)
124	38920	Wait_Call(w, wp, WAITER_REMCLOSE, now);
125		else
126	33260	Wait_Call(w, wp, WAITER_ACTION, now);
127	72180	}
128	188365	if (vwk->nwaited == 0 && vwk->die)
129	40	break;
130		}
131	40	closefd(&vwk->pipe[0]);
132	40	closefd(&vwk->pipe[1]);
133	40	closefd(&vwk->kq);
134	40	return (NULL);
135		}
136
137		/--------------------------------------------------------------------/
138
139		static int v_matchproto_(waiter_enter_f)
140	73580	vwk_enter(void priv, struct waited wp)
141		{
142		struct vwk *vwk;
143		struct kevent ke;
144
145	73580	CAST_OBJ_NOTNULL(vwk, priv, VWK_MAGIC);
146	73580	EV_SET(&ke, wp->fd, EVFILT_READ, EV_ADD\|EV_ONESHOT, 0, 0, wp);
147	73580	Lck_Lock(&vwk->mtx);
148	73580	vwk->nwaited++;
149	73580	Wait_HeapInsert(vwk->waiter, wp);
150	73580	AZ(kevent(vwk->kq, &ke, 1, NULL, 0, NULL));
151
152		/* If the kqueue isn't due before our timeout, poke it via the pipe */
153	73580	if (Wait_When(wp) < vwk->next)
154	72068	assert(write(vwk->pipe[1], "X", 1) == 1);
155
156	73580	Lck_Unlock(&vwk->mtx);
157	73580	return (0);
158		}
159
160		/--------------------------------------------------------------------/
161
162		static void v_matchproto_(waiter_init_f)
163	74777	vwk_init(struct waiter *w)
164		{
165		struct vwk *vwk;
166		struct kevent ke;
167
168	74777	CHECK_OBJ_NOTNULL(w, WAITER_MAGIC);
169	74777	vwk = w->priv;
170	74777	INIT_OBJ(vwk, VWK_MAGIC);
171	74777	vwk->waiter = w;
172
173	74777	vwk->kq = kqueue();
174	74777	assert(vwk->kq >= 0);
175	74777	Lck_New(&vwk->mtx, lck_waiter);
176	74777	AZ(pipe(vwk->pipe));
177	74777	EV_SET(&ke, vwk->pipe[0], EVFILT_READ, EV_ADD, 0, 0, vwk);
178	74777	AZ(kevent(vwk->kq, &ke, 1, NULL, 0, NULL));
179
180	74777	PTOK(pthread_create(&vwk->thread, NULL, vwk_thread, vwk));
181	74777	}
182
183		/*--------------------------------------------------------------------
184		* It is the callers responsibility to trigger all fd's waited on to
185		* fail somehow.
186		*/
187
188		static void v_matchproto_(waiter_fini_f)
189	40	vwk_fini(struct waiter *w)
190		{
191		struct vwk *vwk;
192		void *vp;
193
194	40	CAST_OBJ_NOTNULL(vwk, w->priv, VWK_MAGIC);
195	40	Lck_Lock(&vwk->mtx);
196	40	vwk->die = 1;
197	40	assert(write(vwk->pipe[1], "Y", 1) == 1);
198	40	Lck_Unlock(&vwk->mtx);
199	40	PTOK(pthread_join(vwk->thread, &vp));
200	40	Lck_Delete(&vwk->mtx);
201	40	}
202
203		/--------------------------------------------------------------------/
204
205		#include "waiter/mgt_waiter.h"
206
207		const struct waiter_impl waiter_kqueue = {
208		.name = "kqueue",
209		.init = vwk_init,
210		.fini = vwk_fini,
211		.enter = vwk_enter,
212		.size = sizeof(struct vwk),
213		};
214
215		#endif /* defined(HAVE_KQUEUE) */