GCOV - Varnish Cache


/*-
 * Copyright (c) 2013-2015 Varnish Software AS
 * All rights reserved.
 *
 * Author: Poul-Henning Kamp <phk@FreeBSD.org>
 *
 * 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 <stdlib.h>
#include <string.h>

#include "cache/cache.h"

#include "vcc_directors_if.h"

#include "vmod_directors.h"
#include "vsb.h"
#include "vbm.h"

struct vmod_directors_fallback {
        unsigned                                magic;
#define VMOD_DIRECTORS_FALLBACK_MAGIC           0xad4e26ba
        struct vdir                             *vd;
        VCL_BOOL                                st;
        unsigned                                cur;
};

static VCL_BOOL v_matchproto_(vdi_healthy)
vmod_fallback_healthy(VRT_CTX, VCL_BACKEND dir, VCL_TIME *changed)
{
        struct vmod_directors_fallback *fb;

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
        CAST_OBJ_NOTNULL(fb, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
        return (vdir_any_healthy(ctx, fb->vd, changed));
}

static void v_matchproto_(vdi_list_f)
vmod_fallback_list(VRT_CTX, VCL_BACKEND dir, struct vsb *vsb, int pflag,
    int jflag)
{
        struct vmod_directors_fallback *fb;
        struct vdir *vd;
        VCL_BACKEND be;
        VCL_BOOL h;
        unsigned u, nh;

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
        CAST_OBJ_NOTNULL(fb, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
        CAST_OBJ_NOTNULL(vd, fb->vd, VDIR_MAGIC);

        if (pflag) {
                if (jflag) {
                        VSB_cat(vsb, "{\n");
                        VSB_indent(vsb, 2);
                        VSB_printf(vsb, "\"sticky\": %s,\n",
                            fb->st ? "true" : "false");
                        VSB_cat(vsb, "\"backends\": {\n");
                        VSB_indent(vsb, 2);
                } else {
                        VSB_cat(vsb, "\n\n\tBackend\tCurrent\tHealth\n");
                }
        }

        vdir_rdlock(vd);
        vdir_update_health(ctx, vd);
        for (u = 0; pflag && u < vd->n_backend; u++) {
                be = vd->backend[u];
                CHECK_OBJ_NOTNULL(be, DIRECTOR_MAGIC);

                h = vbit_test(vd->healthy, u);

                if (jflag) {
                        if (u)
                                VSB_cat(vsb, ",\n");
                        VSB_printf(vsb, "\"%s\": {\n", be->vcl_name);
                        VSB_indent(vsb, 2);

                        if (fb->cur == u)
                                VSB_cat(vsb, "\"current\": true,\n");
                        else
                                VSB_cat(vsb, "\"current\": false,\n");

                        if (h)
                                VSB_cat(vsb, "\"health\": \"healthy\"\n");
                        else
                                VSB_cat(vsb, "\"health\": \"sick\"\n");

                        VSB_indent(vsb, -2);
                        VSB_cat(vsb, "}");
                } else {
                        VSB_cat(vsb, "\t");
                        VSB_cat(vsb, be->vcl_name);
                        if (fb->cur == u)
                                VSB_cat(vsb, "\t*\t");
                        else
                                VSB_cat(vsb, "\t\t");
                        VSB_cat(vsb, h ? "healthy" : "sick");
                        VSB_cat(vsb, "\n");
                }
        }
        nh = vd->n_healthy;
        u = vd->n_backend;
        vdir_unlock(vd);

        if (jflag && (pflag)) {
                VSB_cat(vsb, "\n");
                VSB_indent(vsb, -2);
                VSB_cat(vsb, "}\n");
                VSB_indent(vsb, -2);
                VSB_cat(vsb, "},\n");
        }

        if (pflag)
                return;

        if (jflag)
                VSB_printf(vsb, "[%u, %u, \"%s\"]", nh, u,
                    nh ? "healthy" : "sick");
        else
                VSB_printf(vsb, "%u/%u\t%s", nh, u, nh ? "healthy" : "sick");
}

static VCL_BACKEND v_matchproto_(vdi_resolve_f)
vmod_fallback_resolve(VRT_CTX, VCL_BACKEND dir)
{
        struct vmod_directors_fallback *fb;
        unsigned u;
        VCL_BACKEND be = NULL;

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
        CAST_OBJ_NOTNULL(fb, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);

        vdir_wrlock(fb->vd);
        if (!fb->st)
                fb->cur = 0;
        for (u = 0; u < fb->vd->n_backend; u++) {
                be = fb->vd->backend[fb->cur];
                CHECK_OBJ_NOTNULL(be, DIRECTOR_MAGIC);
                if (VRT_Healthy(ctx, be, NULL))
                        break;
                if (++fb->cur == fb->vd->n_backend)
                        fb->cur = 0;
        }
        if (u == fb->vd->n_backend)
                be = NULL;
        vdir_unlock(fb->vd);
        return (be);
}

static void v_matchproto_(vdi_release_f)
vmod_fallback_release(VCL_BACKEND dir)
{
        struct vmod_directors_fallback *fallback;

        CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
        CAST_OBJ_NOTNULL(fallback, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
        vdir_release(fallback->vd);
}

static void v_matchproto_(vdi_destroy_f)
vmod_fallback_destroy(VCL_BACKEND dir)
{
        struct vmod_directors_fallback *fallback;

        CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
        CAST_OBJ_NOTNULL(fallback, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
        vdir_delete(&fallback->vd);
        FREE_OBJ(fallback);
}

static const struct vdi_methods vmod_fallback_methods[1] = {{
        .magic =                VDI_METHODS_MAGIC,
        .type =                 "fallback",
        .healthy =              vmod_fallback_healthy,
        .resolve =              vmod_fallback_resolve,
        .release =              vmod_fallback_release,
        .destroy =              vmod_fallback_destroy,
        .list =                 vmod_fallback_list
}};


VCL_VOID v_matchproto_()
vmod_fallback__init(VRT_CTX,
    struct vmod_directors_fallback **fbp, const char *vcl_name, VCL_BOOL sticky)
{
        struct vmod_directors_fallback *fb;

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        AN(fbp);
        AZ(*fbp);
        ALLOC_OBJ(fb, VMOD_DIRECTORS_FALLBACK_MAGIC);
        AN(fb);
        *fbp = fb;
        vdir_new(ctx, &fb->vd, vcl_name, vmod_fallback_methods, fb);
        fb->st = sticky;
}

VCL_VOID v_matchproto_()
vmod_fallback__fini(struct vmod_directors_fallback **fbp)
{
        struct vmod_directors_fallback *fb;

        TAKE_OBJ_NOTNULL(fb, fbp, VMOD_DIRECTORS_FALLBACK_MAGIC);
        VRT_DelDirector(&fb->vd->dir);
}

VCL_VOID v_matchproto_()
vmod_fallback_add_backend(VRT_CTX,
    struct vmod_directors_fallback *fb, VCL_BACKEND be)
{

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(fb, VMOD_DIRECTORS_FALLBACK_MAGIC);
        vdir_add_backend(ctx, fb->vd, be, 0.0);
}

VCL_VOID v_matchproto_()
vmod_fallback_remove_backend(VRT_CTX,
    struct vmod_directors_fallback *fb, VCL_BACKEND be)
{
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(fb, VMOD_DIRECTORS_FALLBACK_MAGIC);
        vdir_remove_backend(ctx, fb->vd, be, &fb->cur);
}

VCL_BACKEND v_matchproto_()
vmod_fallback_backend(VRT_CTX,
    struct vmod_directors_fallback *fb)
{
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CHECK_OBJ_NOTNULL(fb, VMOD_DIRECTORS_FALLBACK_MAGIC);
        return (fb->vd->dir);
}

		varnish-cache/vmod/vmod_directors_fall_back.c
0		/*-
1		* Copyright (c) 2013-2015 Varnish Software AS
2		* All rights reserved.
3		*
4		* Author: Poul-Henning Kamp <phk@FreeBSD.org>
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 <stdlib.h>
33		#include <string.h>
34
35		#include "cache/cache.h"
36
37		#include "vcc_directors_if.h"
38
39		#include "vmod_directors.h"
40		#include "vsb.h"
41		#include "vbm.h"
42
43		struct vmod_directors_fallback {
44		unsigned magic;
45		#define VMOD_DIRECTORS_FALLBACK_MAGIC 0xad4e26ba
46		struct vdir *vd;
47		VCL_BOOL st;
48		unsigned cur;
49		};
50
51		static VCL_BOOL v_matchproto_(vdi_healthy)
52	4	vmod_fallback_healthy(VRT_CTX, VCL_BACKEND dir, VCL_TIME *changed)
53		{
54		struct vmod_directors_fallback *fb;
55
56	4	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
57	4	CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
58	4	CAST_OBJ_NOTNULL(fb, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
59	4	return (vdir_any_healthy(ctx, fb->vd, changed));
60		}
61
62		static void v_matchproto_(vdi_list_f)
63	13	vmod_fallback_list(VRT_CTX, VCL_BACKEND dir, struct vsb *vsb, int pflag,
64		int jflag)
65		{
66		struct vmod_directors_fallback *fb;
67		struct vdir *vd;
68		VCL_BACKEND be;
69		VCL_BOOL h;
70		unsigned u, nh;
71
72	13	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
73	13	CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
74	13	CAST_OBJ_NOTNULL(fb, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
75	13	CAST_OBJ_NOTNULL(vd, fb->vd, VDIR_MAGIC);
76
77	13	if (pflag) {
78	2	if (jflag) {
79	1	VSB_cat(vsb, "{\n");
80	1	VSB_indent(vsb, 2);
81	2	VSB_printf(vsb, "\"sticky\": %s,\n",
82	1	fb->st ? "true" : "false");
83	1	VSB_cat(vsb, "\"backends\": {\n");
84	1	VSB_indent(vsb, 2);
85	1	} else {
86	1	VSB_cat(vsb, "\n\n\tBackend\tCurrent\tHealth\n");
87		}
88	2	}
89
90	13	vdir_rdlock(vd);
91	13	vdir_update_health(ctx, vd);
92	21	for (u = 0; pflag && u < vd->n_backend; u++) {
93	8	be = vd->backend[u];
94	8	CHECK_OBJ_NOTNULL(be, DIRECTOR_MAGIC);
95
96	8	h = vbit_test(vd->healthy, u);
97
98	8	if (jflag) {
99	4	if (u)
100	3	VSB_cat(vsb, ",\n");
101	4	VSB_printf(vsb, "\"%s\": {\n", be->vcl_name);
102	4	VSB_indent(vsb, 2);
103
104	4	if (fb->cur == u)
105	1	VSB_cat(vsb, "\"current\": true,\n");
106		else
107	3	VSB_cat(vsb, "\"current\": false,\n");
108
109	4	if (h)
110	3	VSB_cat(vsb, "\"health\": \"healthy\"\n");
111		else
112	1	VSB_cat(vsb, "\"health\": \"sick\"\n");
113
114	4	VSB_indent(vsb, -2);
115	4	VSB_cat(vsb, "}");
116	4	} else {
117	4	VSB_cat(vsb, "\t");
118	4	VSB_cat(vsb, be->vcl_name);
119	4	if (fb->cur == u)
120	1	VSB_cat(vsb, "\t*\t");
121		else
122	3	VSB_cat(vsb, "\t\t");
123	4	VSB_cat(vsb, h ? "healthy" : "sick");
124	4	VSB_cat(vsb, "\n");
125		}
126	8	}
127	13	nh = vd->n_healthy;
128	13	u = vd->n_backend;
129	13	vdir_unlock(vd);
130
131	13	if (jflag && (pflag)) {
132	1	VSB_cat(vsb, "\n");
133	1	VSB_indent(vsb, -2);
134	1	VSB_cat(vsb, "}\n");
135	1	VSB_indent(vsb, -2);
136	1	VSB_cat(vsb, "},\n");
137	1	}
138
139	13	if (pflag)
140	2	return;
141
142	11	if (jflag)
143	6	VSB_printf(vsb, "[%u, %u, \"%s\"]", nh, u,
144	3	nh ? "healthy" : "sick");
145		else
146	8	VSB_printf(vsb, "%u/%u\t%s", nh, u, nh ? "healthy" : "sick");
147	13	}
148
149		static VCL_BACKEND v_matchproto_(vdi_resolve_f)
150	27	vmod_fallback_resolve(VRT_CTX, VCL_BACKEND dir)
151		{
152		struct vmod_directors_fallback *fb;
153		unsigned u;
154	27	VCL_BACKEND be = NULL;
155
156	27	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
157	27	CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
158	27	CAST_OBJ_NOTNULL(fb, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
159
160	27	vdir_wrlock(fb->vd);
161	27	if (!fb->st)
162	18	fb->cur = 0;
163	45	for (u = 0; u < fb->vd->n_backend; u++) {
164	44	be = fb->vd->backend[fb->cur];
165	44	CHECK_OBJ_NOTNULL(be, DIRECTOR_MAGIC);
166	44	if (VRT_Healthy(ctx, be, NULL))
167	26	break;
168	18	if (++fb->cur == fb->vd->n_backend)
169	2	fb->cur = 0;
170	18	}
171	27	if (u == fb->vd->n_backend)
172	1	be = NULL;
173	27	vdir_unlock(fb->vd);
174	27	return (be);
175		}
176
177		static void v_matchproto_(vdi_release_f)
178	1	vmod_fallback_release(VCL_BACKEND dir)
179		{
180		struct vmod_directors_fallback *fallback;
181
182	1	CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
183	1	CAST_OBJ_NOTNULL(fallback, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
184	1	vdir_release(fallback->vd);
185	1	}
186
187		static void v_matchproto_(vdi_destroy_f)
188	1	vmod_fallback_destroy(VCL_BACKEND dir)
189		{
190		struct vmod_directors_fallback *fallback;
191
192	1	CHECK_OBJ_NOTNULL(dir, DIRECTOR_MAGIC);
193	1	CAST_OBJ_NOTNULL(fallback, dir->priv, VMOD_DIRECTORS_FALLBACK_MAGIC);
194	1	vdir_delete(&fallback->vd);
195	1	FREE_OBJ(fallback);
196	1	}
197
198		static const struct vdi_methods vmod_fallback_methods[1] = {{
199		.magic = VDI_METHODS_MAGIC,
200		.type = "fallback",
201		.healthy = vmod_fallback_healthy,
202		.resolve = vmod_fallback_resolve,
203		.release = vmod_fallback_release,
204		.destroy = vmod_fallback_destroy,
205		.list = vmod_fallback_list
206		}};
207
208
209		VCL_VOID v_matchproto_()
210	5	vmod_fallback__init(VRT_CTX,
211		struct vmod_directors_fallback *fbp, const char vcl_name, VCL_BOOL sticky)
212		{
213		struct vmod_directors_fallback *fb;
214
215	5	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
216	5	AN(fbp);
217	5	AZ(*fbp);
218	5	ALLOC_OBJ(fb, VMOD_DIRECTORS_FALLBACK_MAGIC);
219	5	AN(fb);
220	5	*fbp = fb;
221	5	vdir_new(ctx, &fb->vd, vcl_name, vmod_fallback_methods, fb);
222	5	fb->st = sticky;
223	5	}
224
225		VCL_VOID v_matchproto_()
226	1	vmod_fallback__fini(struct vmod_directors_fallback **fbp)
227		{
228		struct vmod_directors_fallback *fb;
229
230	1	TAKE_OBJ_NOTNULL(fb, fbp, VMOD_DIRECTORS_FALLBACK_MAGIC);
231	1	VRT_DelDirector(&fb->vd->dir);
232	1	}
233
234		VCL_VOID v_matchproto_()
235	16	vmod_fallback_add_backend(VRT_CTX,
236		struct vmod_directors_fallback *fb, VCL_BACKEND be)
237		{
238
239	16	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
240	16	CHECK_OBJ_NOTNULL(fb, VMOD_DIRECTORS_FALLBACK_MAGIC);
241	16	vdir_add_backend(ctx, fb->vd, be, 0.0);
242	16	}
243
244		VCL_VOID v_matchproto_()
245	4	vmod_fallback_remove_backend(VRT_CTX,
246		struct vmod_directors_fallback *fb, VCL_BACKEND be)
247		{
248	4	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
249	4	CHECK_OBJ_NOTNULL(fb, VMOD_DIRECTORS_FALLBACK_MAGIC);
250	4	vdir_remove_backend(ctx, fb->vd, be, &fb->cur);
251	4	}
252
253		VCL_BACKEND v_matchproto_()
254	27	vmod_fallback_backend(VRT_CTX,
255		struct vmod_directors_fallback *fb)
256		{
257	27	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
258	27	CHECK_OBJ_NOTNULL(fb, VMOD_DIRECTORS_FALLBACK_MAGIC);
259	27	return (fb->vd->dir);
260		}