GCOV - Varnish Cache


/*-
 * Copyright (c) 2010-2011 Varnish Software AS
 * All rights reserved.
 *
 * Author: Sanjoy Das <sanjoy@playingwithpointers.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.
 *
 * XXX: It might make sense to use just a single global list of all files
 * XXX: and use the call-private pointer to point to the file instance on
 * XXX: that list.
 * XXX: Duplicates in the global list can be avoided by examining the
 * XXX: dev+inode fields of the stat structure.
 * XXX: Individual files would need to be refcounted, so they can be
 * XXX: deleted when no VCL's reference them.
 *
 * XXX: We should periodically stat(2) the filename and check if the
 * XXX: underlying file has been updated.
 */

#include "config.h"

#include <stdlib.h>
#include <string.h>

#include "cache/cache.h"

#include "vfil.h"

#include "vcc_std_if.h"

struct frfile {
        unsigned                        magic;
#define CACHED_FILE_MAGIC 0xa8e9d87a
        char                            *file_name;
        void                            *contents;
        struct vrt_blob                 blob[1];
        int                             refcount;
        VTAILQ_ENTRY(frfile)            list;
};

static VTAILQ_HEAD(, frfile)    frlist = VTAILQ_HEAD_INITIALIZER(frlist);
static pthread_mutex_t          frmtx = PTHREAD_MUTEX_INITIALIZER;

static void
fini_frfile(VRT_CTX, void *ptr)
{
        struct frfile *frf;

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        CAST_OBJ_NOTNULL(frf, ptr, CACHED_FILE_MAGIC);

        PTOK(pthread_mutex_lock(&frmtx));
        if (--frf->refcount > 0)
                frf = NULL;
        else
                VTAILQ_REMOVE(&frlist, frf, list);
        PTOK(pthread_mutex_unlock(&frmtx));
        if (frf != NULL) {
                free(frf->contents);
                free(frf->file_name);
                FREE_OBJ(frf);
        }
}

static const struct vmod_priv_methods frfile_methods[1] = {{
        .magic = VMOD_PRIV_METHODS_MAGIC,
        .type = "vmod_std_fileread",
        .fini = fini_frfile
}};

static struct frfile *
find_frfile(struct vmod_priv *priv, VCL_STRING file_name)
{
        struct frfile *frf = NULL;
        char *s;
        ssize_t sz;

        AN(priv);

        if (file_name == NULL)
                return (NULL);

        if (priv->priv != NULL) {
                CAST_OBJ_NOTNULL(frf, priv->priv, CACHED_FILE_MAGIC);
                if (!strcmp(file_name, frf->file_name))
                        return (frf);
        }

        PTOK(pthread_mutex_lock(&frmtx));
        if (frf != NULL)
                frf->refcount--;
        VTAILQ_FOREACH(frf, &frlist, list) {
                if (!strcmp(file_name, frf->file_name)) {
                        frf->refcount++;
                        break;
                }
        }
        PTOK(pthread_mutex_unlock(&frmtx));
        if (frf != NULL) {
                priv->methods = frfile_methods;
                priv->priv = frf;
                return (frf);
        }

        s = VFIL_readfile(NULL, file_name, &sz);
        if (s != NULL) {
                assert(sz > 0);
                ALLOC_OBJ(frf, CACHED_FILE_MAGIC);
                AN(frf);
                REPLACE(frf->file_name, file_name);
                frf->refcount = 1;
                frf->contents = s;
                frf->blob->blob = s;
                frf->blob->len = (size_t)sz;
                priv->methods = frfile_methods;
                priv->priv = frf;
                PTOK(pthread_mutex_lock(&frmtx));
                VTAILQ_INSERT_HEAD(&frlist, frf, list);
                PTOK(pthread_mutex_unlock(&frmtx));
        }
        return (frf);
}

VCL_STRING v_matchproto_(td_std_fileread)
vmod_fileread(VRT_CTX, struct vmod_priv *priv, VCL_STRING file_name)
{
        struct frfile *frf;

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        AN(priv);

        frf = find_frfile(priv, file_name);
        if (frf == NULL)
                return (NULL);
        return (frf->contents);
}

VCL_BLOB v_matchproto_(td_std_blobread)
vmod_blobread(VRT_CTX, struct vmod_priv *priv, VCL_STRING file_name)
{
        struct frfile *frf;

        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
        AN(priv);

        frf = find_frfile(priv, file_name);
        if (frf == NULL)
                return (NULL);
        return (frf->blob);
}

		varnish-cache/vmod/vmod_std_fileread.c
0		/*-
1		* Copyright (c) 2010-2011 Varnish Software AS
2		* All rights reserved.
3		*
4		* Author: Sanjoy Das <sanjoy@playingwithpointers.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		* XXX: It might make sense to use just a single global list of all files
30		* XXX: and use the call-private pointer to point to the file instance on
31		* XXX: that list.
32		* XXX: Duplicates in the global list can be avoided by examining the
33		* XXX: dev+inode fields of the stat structure.
34		* XXX: Individual files would need to be refcounted, so they can be
35		* XXX: deleted when no VCL's reference them.
36		*
37		* XXX: We should periodically stat(2) the filename and check if the
38		* XXX: underlying file has been updated.
39		*/
40
41		#include "config.h"
42
43		#include <stdlib.h>
44		#include <string.h>
45
46		#include "cache/cache.h"
47
48		#include "vfil.h"
49
50		#include "vcc_std_if.h"
51
52		struct frfile {
53		unsigned magic;
54		#define CACHED_FILE_MAGIC 0xa8e9d87a
55		char *file_name;
56		void *contents;
57		struct vrt_blob blob[1];
58		int refcount;
59		VTAILQ_ENTRY(frfile) list;
60		};
61
62		static VTAILQ_HEAD(, frfile) frlist = VTAILQ_HEAD_INITIALIZER(frlist);
63		static pthread_mutex_t frmtx = PTHREAD_MUTEX_INITIALIZER;
64
65		static void
66	32	fini_frfile(VRT_CTX, void *ptr)
67		{
68		struct frfile *frf;
69
70	32	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
71	32	CAST_OBJ_NOTNULL(frf, ptr, CACHED_FILE_MAGIC);
72
73	32	PTOK(pthread_mutex_lock(&frmtx));
74	32	if (--frf->refcount > 0)
75	20	frf = NULL;
76		else
77	12	VTAILQ_REMOVE(&frlist, frf, list);
78	32	PTOK(pthread_mutex_unlock(&frmtx));
79	32	if (frf != NULL) {
80	12	free(frf->contents);
81	12	free(frf->file_name);
82	12	FREE_OBJ(frf);
83	12	}
84	32	}
85
86		static const struct vmod_priv_methods frfile_methods[1] = {{
87		.magic = VMOD_PRIV_METHODS_MAGIC,
88		.type = "vmod_std_fileread",
89		.fini = fini_frfile
90		}};
91
92		static struct frfile *
93	140	find_frfile(struct vmod_priv *priv, VCL_STRING file_name)
94		{
95	140	struct frfile *frf = NULL;
96		char *s;
97		ssize_t sz;
98
99	140	AN(priv);
100
101	140	if (file_name == NULL)
102	0	return (NULL);
103
104	140	if (priv->priv != NULL) {
105	80	CAST_OBJ_NOTNULL(frf, priv->priv, CACHED_FILE_MAGIC);
106	80	if (!strcmp(file_name, frf->file_name))
107	16	return (frf);
108	64	}
109
110	124	PTOK(pthread_mutex_lock(&frmtx));
111	124	if (frf != NULL)
112	64	frf->refcount--;
113	260	VTAILQ_FOREACH(frf, &frlist, list) {
114	216	if (!strcmp(file_name, frf->file_name)) {
115	80	frf->refcount++;
116	80	break;
117		}
118	136	}
119	124	PTOK(pthread_mutex_unlock(&frmtx));
120	124	if (frf != NULL) {
121	80	priv->methods = frfile_methods;
122	80	priv->priv = frf;
123	80	return (frf);
124		}
125
126	44	s = VFIL_readfile(NULL, file_name, &sz);
127	44	if (s != NULL) {
128	28	assert(sz > 0);
129	28	ALLOC_OBJ(frf, CACHED_FILE_MAGIC);
130	28	AN(frf);
131	28	REPLACE(frf->file_name, file_name);
132	28	frf->refcount = 1;
133	28	frf->contents = s;
134	28	frf->blob->blob = s;
135	28	frf->blob->len = (size_t)sz;
136	28	priv->methods = frfile_methods;
137	28	priv->priv = frf;
138	28	PTOK(pthread_mutex_lock(&frmtx));
139	28	VTAILQ_INSERT_HEAD(&frlist, frf, list);
140	28	PTOK(pthread_mutex_unlock(&frmtx));
141	28	}
142	44	return (frf);
143	140	}
144
145		VCL_STRING v_matchproto_(td_std_fileread)
146	104	vmod_fileread(VRT_CTX, struct vmod_priv *priv, VCL_STRING file_name)
147		{
148		struct frfile *frf;
149
150	104	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
151	104	AN(priv);
152
153	104	frf = find_frfile(priv, file_name);
154	104	if (frf == NULL)
155	16	return (NULL);
156	88	return (frf->contents);
157	104	}
158
159		VCL_BLOB v_matchproto_(td_std_blobread)
160	36	vmod_blobread(VRT_CTX, struct vmod_priv *priv, VCL_STRING file_name)
161		{
162		struct frfile *frf;
163
164	36	CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
165	36	AN(priv);
166
167	36	frf = find_frfile(priv, file_name);
168	36	if (frf == NULL)
169	0	return (NULL);
170	36	return (frf->blob);
171	36	}