GCOV - Varnish Cache


/*-
 * Copyright (c) 2015 Varnish Software AS
 * All rights reserved.
 *
 * Author: Dridi Boukelmoune <dridi@varnish-software.com>
 *
 * 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 <netdb.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <unistd.h>

#include "cache/cache.h"

#include "vsa.h"
#include "vss.h"
#include "vus.h"
#include "vcc_debug_if.h"

struct xyzzy_debug_dyn {
        unsigned                magic;
#define VMOD_DEBUG_DYN_MAGIC    0x9b77ccbd
        pthread_mutex_t         mtx;
        char                    *vcl_name;
        VCL_BACKEND             dir;
};

struct xyzzy_debug_dyn_uds {
        unsigned                magic;
#define VMOD_DEBUG_UDS_MAGIC    0x6c7370e6
        pthread_mutex_t         mtx;
        char                    *vcl_name;
        VCL_BACKEND             dir;
};

static void
dyn_dir_init(VRT_CTX, struct xyzzy_debug_dyn *dyn,
    VCL_STRING addr, VCL_STRING port, VCL_PROBE probe, VCL_BACKEND via)
{
        const struct suckaddr *sa;
        VCL_BACKEND dir, dir2;
        struct vrt_endpoint vep;
        struct vrt_backend vrt;

        CHECK_OBJ_NOTNULL(dyn, VMOD_DEBUG_DYN_MAGIC);
        XXXAN(addr);
        XXXAN(port);
        CHECK_OBJ_ORNULL(via, DIRECTOR_MAGIC);

        INIT_OBJ(&vep, VRT_ENDPOINT_MAGIC);
        VRT_BACKEND_INIT(&vrt);
        vrt.endpoint = &vep;
        vrt.vcl_name = dyn->vcl_name;
        vrt.hosthdr = addr;
        vrt.authority = addr;
        vrt.probe = probe;

        sa = VSS_ResolveOne(NULL, addr, port, AF_UNSPEC, SOCK_STREAM, 0);
        AN(sa);
        if (VSA_Get_Proto(sa) == AF_INET)
                vep.ipv4 = sa;
        else if (VSA_Get_Proto(sa) == AF_INET6)
                vep.ipv6 = sa;
        else
                WRONG("Wrong proto family");

        dir = VRT_new_backend(ctx, &vrt, via);
        AN(dir);

        /*
         * NB: A real dynamic backend should not replace the previous
         * instance if the new one is identical.  We do it here because
         * the d* tests requires a replacement.
         */
        PTOK(pthread_mutex_lock(&dyn->mtx));
        dir2 = dyn->dir;
        dyn->dir = dir;
        PTOK(pthread_mutex_unlock(&dyn->mtx));

        if (dir2 != NULL)
                VRT_delete_backend(ctx, &dir2);

        VSA_free(&sa);
}

VCL_VOID
xyzzy_dyn__init(VRT_CTX, struct xyzzy_debug_dyn **dynp,
    const char *vcl_name, VCL_STRING addr, VCL_STRING port, VCL_PROBE probe,
    VCL_BACKEND via)
{
        struct xyzzy_debug_dyn *dyn;

        ASSERT_CLI();
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        AN(dynp);
        AZ(*dynp);
        AN(vcl_name);

        if (*addr == '\0' || *port == '\0') {
                AZ(VRT_handled(ctx));
                VRT_fail(ctx, "Missing dynamic backend address or port");
                return;
        }

        ALLOC_OBJ(dyn, VMOD_DEBUG_DYN_MAGIC);
        AN(dyn);
        REPLACE(dyn->vcl_name, vcl_name);

        PTOK(pthread_mutex_init(&dyn->mtx, NULL));

        dyn_dir_init(ctx, dyn, addr, port, probe, via);
        XXXAN(dyn->dir);
        *dynp = dyn;
}

VCL_VOID v_matchproto_(td_xyzzy_debug_dyn__fini)
xyzzy_dyn__fini(struct xyzzy_debug_dyn **dynp)
{
        struct xyzzy_debug_dyn *dyn;

        TAKE_OBJ_NOTNULL(dyn, dynp, VMOD_DEBUG_DYN_MAGIC);
        /* at this point all backends will be deleted by the vcl */
        free(dyn->vcl_name);
        PTOK(pthread_mutex_destroy(&dyn->mtx));
        FREE_OBJ(dyn);
}

VCL_BACKEND v_matchproto_()
xyzzy_dyn_backend(VRT_CTX, struct xyzzy_debug_dyn *dyn)
{
        VCL_BACKEND retval;
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(dyn, VMOD_DEBUG_DYN_MAGIC);
        PTOK(pthread_mutex_lock(&dyn->mtx));
        retval = dyn->dir;
        PTOK(pthread_mutex_unlock(&dyn->mtx));
        AN(retval);
        return (retval);
}

VCL_VOID
xyzzy_dyn_refresh(VRT_CTX, struct xyzzy_debug_dyn *dyn,
    VCL_STRING addr, VCL_STRING port, VCL_PROBE probe, VCL_BACKEND via)
{
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(dyn, VMOD_DEBUG_DYN_MAGIC);
        dyn_dir_init(ctx, dyn, addr, port, probe, via);
}

static int
dyn_uds_init(VRT_CTX, struct xyzzy_debug_dyn_uds *uds, VCL_STRING path)
{
        VCL_BACKEND dir, dir2;
        struct vrt_endpoint vep;
        struct vrt_backend vrt;
        struct stat st;

        if (path == NULL) {
                VRT_fail(ctx, "path is NULL");
                return (-1);
        }
        if (! VUS_is(path)) {
                VRT_fail(ctx, "path must be an absolute path: %s", path);
                return (-1);
        }

        /* XXX: now that we accept that sockets may come after vcl.load
         * and change during the life of a VCL, do we still need to check
         * this? It looks like both if blocks can be retired.
         */
        errno = 0;
        if (stat(path, &st) != 0) {
                VRT_fail(ctx, "Cannot stat path %s: %s", path, strerror(errno));
                return (-1);
        }
        if (!S_ISSOCK(st.st_mode)) {
                VRT_fail(ctx, "%s is not a socket", path);
                return (-1);
        }

        INIT_OBJ(&vep, VRT_ENDPOINT_MAGIC);
        VRT_BACKEND_INIT(&vrt);
        vrt.endpoint = &vep;
        vep.uds_path = path;
        vrt.vcl_name = uds->vcl_name;
        vrt.hosthdr = "localhost";
        vep.ipv4 = NULL;
        vep.ipv6 = NULL;

        // we support via: uds -> ip, but not via: ip -> uds
        if ((dir = VRT_new_backend(ctx, &vrt, NULL)) == NULL)
                return (-1);

        PTOK(pthread_mutex_lock(&uds->mtx));
        dir2 = uds->dir;
        uds->dir = dir;
        PTOK(pthread_mutex_unlock(&uds->mtx));

        if (dir2 != NULL)
                VRT_delete_backend(ctx, &dir2);
        return (0);
}

VCL_VOID v_matchproto_(td_xyzzy_debug_dyn_uds__init)
xyzzy_dyn_uds__init(VRT_CTX, struct xyzzy_debug_dyn_uds **udsp,
    const char *vcl_name, VCL_STRING path)
{
        struct xyzzy_debug_dyn_uds *uds;

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        AN(udsp);
        AZ(*udsp);
        AN(vcl_name);

        ALLOC_OBJ(uds, VMOD_DEBUG_UDS_MAGIC);
        AN(uds);
        REPLACE(uds->vcl_name, vcl_name);
        PTOK(pthread_mutex_init(&uds->mtx, NULL));

        if (dyn_uds_init(ctx, uds, path) != 0) {
                free(uds->vcl_name);
                PTOK(pthread_mutex_destroy(&uds->mtx));
                FREE_OBJ(uds);
                return;
        }

        *udsp = uds;
}

VCL_VOID v_matchproto_(td_debug_dyn_uds__fini)
xyzzy_dyn_uds__fini(struct xyzzy_debug_dyn_uds **udsp)
{
        struct xyzzy_debug_dyn_uds *uds;

        TAKE_OBJ_NOTNULL(uds, udsp, VMOD_DEBUG_UDS_MAGIC);
        free(uds->vcl_name);
        PTOK(pthread_mutex_destroy(&uds->mtx));
        FREE_OBJ(uds);
}

VCL_BACKEND v_matchproto_(td_debug_dyn_uds_backend)
xyzzy_dyn_uds_backend(VRT_CTX, struct xyzzy_debug_dyn_uds *uds)
{
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(uds, VMOD_DEBUG_UDS_MAGIC);
        AN(uds->dir);
        return (uds->dir);
}

VCL_VOID v_matchproto_(td_debug_dyn_uds_refresh)
xyzzy_dyn_uds_refresh(VRT_CTX, struct xyzzy_debug_dyn_uds *uds, VCL_STRING path)
{
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(uds, VMOD_DEBUG_UDS_MAGIC);
        (void) dyn_uds_init(ctx, uds, path);
}

		varnish-cache/vmod/vmod_debug_dyn.c
0		/*-
1		* Copyright (c) 2015 Varnish Software AS
2		* All rights reserved.
3		*
4		* Author: Dridi Boukelmoune <dridi@varnish-software.com>
5		*
6		* SPDX-License-Identifier: BSD-2-Clause
7		*
8		* Redistribution and use in source and binary forms, with or without
9		* modification, are permitted provided that the following conditions
10		* are met:
11		* 1. Redistributions of source code must retain the above copyright
12		* notice, this list of conditions and the following disclaimer.
13		* 2. Redistributions in binary form must reproduce the above copyright
14		* notice, this list of conditions and the following disclaimer in the
15		* documentation and/or other materials provided with the distribution.
16		*
17		* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18		* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19		* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20		* ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
21		* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22		* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23		* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24		* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25		* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26		* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27		* SUCH DAMAGE.
28		*/
29
30		#include "config.h"
31
32		#include <netdb.h>
33		#include <stdlib.h>
34		#include <string.h>
35		#include <sys/socket.h>
36		#include <sys/stat.h>
37		#include <unistd.h>
38
39		#include "cache/cache.h"
40
41		#include "vsa.h"
42		#include "vss.h"
43		#include "vus.h"
44		#include "vcc_debug_if.h"
45
46		struct xyzzy_debug_dyn {
47		unsigned magic;
48		#define VMOD_DEBUG_DYN_MAGIC 0x9b77ccbd
49		pthread_mutex_t mtx;
50		char *vcl_name;
51		VCL_BACKEND dir;
52		};
53
54		struct xyzzy_debug_dyn_uds {
55		unsigned magic;
56		#define VMOD_DEBUG_UDS_MAGIC 0x6c7370e6
57		pthread_mutex_t mtx;
58		char *vcl_name;
59		VCL_BACKEND dir;
60		};
61
62		static void
63	18	dyn_dir_init(VRT_CTX, struct xyzzy_debug_dyn *dyn,
64		VCL_STRING addr, VCL_STRING port, VCL_PROBE probe, VCL_BACKEND via)
65		{
66		const struct suckaddr *sa;
67		VCL_BACKEND dir, dir2;
68		struct vrt_endpoint vep;
69		struct vrt_backend vrt;
70
71	18	CHECK_OBJ_NOTNULL(dyn, VMOD_DEBUG_DYN_MAGIC);
72	18	XXXAN(addr);
73	18	XXXAN(port);
74	18	CHECK_OBJ_ORNULL(via, DIRECTOR_MAGIC);
75
76	18	INIT_OBJ(&vep, VRT_ENDPOINT_MAGIC);
77	18	VRT_BACKEND_INIT(&vrt);
78	18	vrt.endpoint = &vep;
79	18	vrt.vcl_name = dyn->vcl_name;
80	18	vrt.hosthdr = addr;
81	18	vrt.authority = addr;
82	18	vrt.probe = probe;
83
84	18	sa = VSS_ResolveOne(NULL, addr, port, AF_UNSPEC, SOCK_STREAM, 0);
85	18	AN(sa);
86	18	if (VSA_Get_Proto(sa) == AF_INET)
87	18	vep.ipv4 = sa;
88	0	else if (VSA_Get_Proto(sa) == AF_INET6)
89	0	vep.ipv6 = sa;
90		else
91	0	WRONG("Wrong proto family");
92
93	18	dir = VRT_new_backend(ctx, &vrt, via);
94	18	AN(dir);
95
96		/*
97		* NB: A real dynamic backend should not replace the previous
98		* instance if the new one is identical. We do it here because
99		* the d* tests requires a replacement.
100		*/
101	18	PTOK(pthread_mutex_lock(&dyn->mtx));
102	18	dir2 = dyn->dir;
103	18	dyn->dir = dir;
104	18	PTOK(pthread_mutex_unlock(&dyn->mtx));
105
106	18	if (dir2 != NULL)
107	9	VRT_delete_backend(ctx, &dir2);
108
109	18	VSA_free(&sa);
110	18	}
111
112		VCL_VOID
113	10	xyzzy_dyn__init(VRT_CTX, struct xyzzy_debug_dyn **dynp,
114		const char *vcl_name, VCL_STRING addr, VCL_STRING port, VCL_PROBE probe,
115		VCL_BACKEND via)
116		{
117		struct xyzzy_debug_dyn *dyn;
118
119	10	ASSERT_CLI();
120	10	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
121	10	AN(dynp);
122	10	AZ(*dynp);
123	10	AN(vcl_name);
124
125	10	if (addr == '\0' \|\| port == '\0') {
126	1	AZ(VRT_handled(ctx));
127	1	VRT_fail(ctx, "Missing dynamic backend address or port");
128	1	return;
129		}
130
131	9	ALLOC_OBJ(dyn, VMOD_DEBUG_DYN_MAGIC);
132	9	AN(dyn);
133	9	REPLACE(dyn->vcl_name, vcl_name);
134
135	9	PTOK(pthread_mutex_init(&dyn->mtx, NULL));
136
137	9	dyn_dir_init(ctx, dyn, addr, port, probe, via);
138	9	XXXAN(dyn->dir);
139	9	*dynp = dyn;
140	10	}
141
142		VCL_VOID v_matchproto_(td_xyzzy_debug_dyn__fini)
143	1	xyzzy_dyn__fini(struct xyzzy_debug_dyn **dynp)
144		{
145		struct xyzzy_debug_dyn *dyn;
146
147	1	TAKE_OBJ_NOTNULL(dyn, dynp, VMOD_DEBUG_DYN_MAGIC);
148		/* at this point all backends will be deleted by the vcl */
149	1	free(dyn->vcl_name);
150	1	PTOK(pthread_mutex_destroy(&dyn->mtx));
151	1	FREE_OBJ(dyn);
152	1	}
153
154		VCL_BACKEND v_matchproto_()
155	16	xyzzy_dyn_backend(VRT_CTX, struct xyzzy_debug_dyn *dyn)
156		{
157		VCL_BACKEND retval;
158	16	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
159	16	CHECK_OBJ_NOTNULL(dyn, VMOD_DEBUG_DYN_MAGIC);
160	16	PTOK(pthread_mutex_lock(&dyn->mtx));
161	16	retval = dyn->dir;
162	16	PTOK(pthread_mutex_unlock(&dyn->mtx));
163	16	AN(retval);
164	16	return (retval);
165		}
166
167		VCL_VOID
168	9	xyzzy_dyn_refresh(VRT_CTX, struct xyzzy_debug_dyn *dyn,
169		VCL_STRING addr, VCL_STRING port, VCL_PROBE probe, VCL_BACKEND via)
170		{
171	9	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
172	9	CHECK_OBJ_NOTNULL(dyn, VMOD_DEBUG_DYN_MAGIC);
173	9	dyn_dir_init(ctx, dyn, addr, port, probe, via);
174	9	}
175
176		static int
177	17	dyn_uds_init(VRT_CTX, struct xyzzy_debug_dyn_uds *uds, VCL_STRING path)
178		{
179		VCL_BACKEND dir, dir2;
180		struct vrt_endpoint vep;
181		struct vrt_backend vrt;
182		struct stat st;
183
184	17	if (path == NULL) {
185	0	VRT_fail(ctx, "path is NULL");
186	0	return (-1);
187		}
188	17	if (! VUS_is(path)) {
189	2	VRT_fail(ctx, "path must be an absolute path: %s", path);
190	2	return (-1);
191		}
192
193		/* XXX: now that we accept that sockets may come after vcl.load
194		* and change during the life of a VCL, do we still need to check
195		* this? It looks like both if blocks can be retired.
196		*/
197	15	errno = 0;
198	15	if (stat(path, &st) != 0) {
199	1	VRT_fail(ctx, "Cannot stat path %s: %s", path, strerror(errno));
200	1	return (-1);
201		}
202	14	if (!S_ISSOCK(st.st_mode)) {
203	1	VRT_fail(ctx, "%s is not a socket", path);
204	1	return (-1);
205		}
206
207	13	INIT_OBJ(&vep, VRT_ENDPOINT_MAGIC);
208	13	VRT_BACKEND_INIT(&vrt);
209	13	vrt.endpoint = &vep;
210	13	vep.uds_path = path;
211	13	vrt.vcl_name = uds->vcl_name;
212	13	vrt.hosthdr = "localhost";
213	13	vep.ipv4 = NULL;
214	13	vep.ipv6 = NULL;
215
216		// we support via: uds -> ip, but not via: ip -> uds
217	13	if ((dir = VRT_new_backend(ctx, &vrt, NULL)) == NULL)
218	0	return (-1);
219
220	13	PTOK(pthread_mutex_lock(&uds->mtx));
221	13	dir2 = uds->dir;
222	13	uds->dir = dir;
223	13	PTOK(pthread_mutex_unlock(&uds->mtx));
224
225	13	if (dir2 != NULL)
226	5	VRT_delete_backend(ctx, &dir2);
227	13	return (0);
228	17	}
229
230		VCL_VOID v_matchproto_(td_xyzzy_debug_dyn_uds__init)
231	12	xyzzy_dyn_uds__init(VRT_CTX, struct xyzzy_debug_dyn_uds **udsp,
232		const char *vcl_name, VCL_STRING path)
233		{
234		struct xyzzy_debug_dyn_uds *uds;
235
236	12	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
237	12	AN(udsp);
238	12	AZ(*udsp);
239	12	AN(vcl_name);
240
241	12	ALLOC_OBJ(uds, VMOD_DEBUG_UDS_MAGIC);
242	12	AN(uds);
243	12	REPLACE(uds->vcl_name, vcl_name);
244	12	PTOK(pthread_mutex_init(&uds->mtx, NULL));
245
246	12	if (dyn_uds_init(ctx, uds, path) != 0) {
247	4	free(uds->vcl_name);
248	4	PTOK(pthread_mutex_destroy(&uds->mtx));
249	4	FREE_OBJ(uds);
250	4	return;
251		}
252
253	8	*udsp = uds;
254	12	}
255
256		VCL_VOID v_matchproto_(td_debug_dyn_uds__fini)
257	1	xyzzy_dyn_uds__fini(struct xyzzy_debug_dyn_uds **udsp)
258		{
259		struct xyzzy_debug_dyn_uds *uds;
260
261	1	TAKE_OBJ_NOTNULL(uds, udsp, VMOD_DEBUG_UDS_MAGIC);
262	1	free(uds->vcl_name);
263	1	PTOK(pthread_mutex_destroy(&uds->mtx));
264	1	FREE_OBJ(uds);
265	1	}
266
267		VCL_BACKEND v_matchproto_(td_debug_dyn_uds_backend)
268	13	xyzzy_dyn_uds_backend(VRT_CTX, struct xyzzy_debug_dyn_uds *uds)
269		{
270	13	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
271	13	CHECK_OBJ_NOTNULL(uds, VMOD_DEBUG_UDS_MAGIC);
272	13	AN(uds->dir);
273	13	return (uds->dir);
274		}
275
276		VCL_VOID v_matchproto_(td_debug_dyn_uds_refresh)
277	5	xyzzy_dyn_uds_refresh(VRT_CTX, struct xyzzy_debug_dyn_uds *uds, VCL_STRING path)
278		{
279	5	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
280	5	CHECK_OBJ_NOTNULL(uds, VMOD_DEBUG_UDS_MAGIC);
281	5	(void) dyn_uds_init(ctx, uds, path);
282	5	}