GCOV - Varnish Cache


/*-
 * Copyright 2018 UPLEX - Nils Goroll Systemoptimierung
 * All rights reserved.
 *
 * Author: Geoffrey Simmons <geoffrey.simmons@uplex.de>
 *
 * 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 <sys/socket.h>
#include <sys/un.h>

#include <unistd.h>
#include <string.h>
#include <poll.h>
#include <stdio.h>

#include "vdef.h"
#include "vas.h"
#include "vus.h"
#include "vtcp.h"

static int
sun_init(struct sockaddr_un *uds, const char *path, const char **err)
{
        AN(uds);
        AN(path);
        assert(VUS_is(path));

        if (err)
                *err = NULL;

        if (strlen(path) + 1 > sizeof(uds->sun_path)) {
                errno = ENAMETOOLONG;
                if (err)
                        *err = "Path too long for a Unix domain socket";
                return (-1);
        }
        if (! strcmp(path, "@")) {
                errno = EINVAL;
                if (err)
                        *err = "The empty abstract socket name is not"
                            " supported";
                return (-1);
        }
        memset(uds->sun_path, 0, sizeof(uds->sun_path));
        if (*path == '@')
                bprintf(uds->sun_path, "%c%s", 0, path + 1);
        else
                bprintf(uds->sun_path, "%s", path);
        uds->sun_family = PF_UNIX;
        return (0);
}

int
VUS_resolver(const char *path, vus_resolved_f *func, void *priv,
             const char **err)
{
        struct sockaddr_un uds;
        int ret;

        AN(err);

        ret = sun_init(&uds, path, err);
        if (ret)
                return (ret);

        assert(uds.sun_path[1] != '\0');

        if (func != NULL)
                ret = func(priv, &uds);
        return (ret);
}

int
VUS_bind(const struct sockaddr_un *uds, const char **errp)
{
        int sd, e;
        socklen_t sl;

        sl = VUS_socklen(uds);

        if (errp != NULL)
                *errp = NULL;

        sd = socket(PF_UNIX, SOCK_STREAM, 0);
        if (sd < 0) {
                if (errp != NULL)
                        *errp = "socket(2)";
                return (-1);
        }

        if (unlink(uds->sun_path) != 0 && errno != ENOENT) {
                if (errp != NULL)
                        *errp = "unlink(2)";
                e = errno;
                closefd(&sd);
                errno = e;
                return (-1);
        }

        if (bind(sd, (const void*)uds, sl) != 0) {
                if (errp != NULL)
                        *errp = "bind(2)";
                e = errno;
                closefd(&sd);
                errno = e;
                return (-1);
        }
        return (sd);
}

int
VUS_connect(const char *path, int msec)
{
        int s, i;
        struct pollfd fds[1];
        struct sockaddr_un uds;
        socklen_t sl;

        if (path == NULL)
                return (-1);
        i = sun_init(&uds, path, NULL);
        if (i)
                return (i);

        assert(uds.sun_path[1] != '\0');

        sl = VUS_socklen(&uds);

        AN(sl);

        s = socket(PF_UNIX, SOCK_STREAM, 0);
        if (s < 0)
                return (s);

        /* Set the socket non-blocking */
        if (msec != 0)
                VTCP_nonblocking(s);

        i = connect(s, (const void*)&uds, sl);
        if (i == 0)
                return (s);
        if (errno != EINPROGRESS) {
                closefd(&s);
                return (-1);
        }

        if (msec < 0) {
                /*
                 * Caller is responsible for waiting and
                 * calling VTCP_connected
                 */
                return (s);
        }

        assert(msec > 0);
        /* Exercise our patience, polling for write */
        fds[0].fd = s;
        fds[0].events = POLLWRNORM;
        fds[0].revents = 0;
        i = poll(fds, 1, msec);

        if (i == 0) {
                /* Timeout, close and give up */
                closefd(&s);
                errno = ETIMEDOUT;
                return (-1);
        }

        return (VTCP_connected(s));
}

socklen_t
VUS_socklen(const struct sockaddr_un *uds)
{
        socklen_t sl;
        const char *p;
        if (*uds->sun_path)
                sl = sizeof(*uds);
        else {
                p = strchr(uds->sun_path + 1, '\0');
                assert(p != NULL);
                sl = p - (const char*)uds;
        }
        assert(sl <= sizeof(*uds));
        return sl;
}

		varnish-cache/lib/libvarnish/vus.c
0		/*-
1		* Copyright 2018 UPLEX - Nils Goroll Systemoptimierung
2		* All rights reserved.
3		*
4		* Author: Geoffrey Simmons <geoffrey.simmons@uplex.de>
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 <sys/socket.h>
31		#include <sys/un.h>
32
33		#include <unistd.h>
34		#include <string.h>
35		#include <poll.h>
36		#include <stdio.h>
37
38		#include "vdef.h"
39		#include "vas.h"
40		#include "vus.h"
41		#include "vtcp.h"
42
43		static int
44	560	sun_init(struct sockaddr_un uds, const char path, const char **err)
45		{
46	560	AN(uds);
47	560	AN(path);
48	560	assert(VUS_is(path));
49
50	560	if (err)
51	442	*err = NULL;
52
53	560	if (strlen(path) + 1 > sizeof(uds->sun_path)) {
54	2	errno = ENAMETOOLONG;
55	2	if (err)
56	2	*err = "Path too long for a Unix domain socket";
57	2	return (-1);
58		}
59	558	if (! strcmp(path, "@")) {
60	0	errno = EINVAL;
61	0	if (err)
62	0	*err = "The empty abstract socket name is not"
63		" supported";
64	0	return (-1);
65		}
66	558	memset(uds->sun_path, 0, sizeof(uds->sun_path));
67	558	if (*path == '@')
68	1	bprintf(uds->sun_path, "%c%s", 0, path + 1);
69		else
70	557	bprintf(uds->sun_path, "%s", path);
71	558	uds->sun_family = PF_UNIX;
72	558	return (0);
73	560	}
74
75		int
76	443	VUS_resolver(const char path, vus_resolved_f func, void *priv,
77		const char **err)
78		{
79		struct sockaddr_un uds;
80		int ret;
81
82	443	AN(err);
83
84	443	ret = sun_init(&uds, path, err);
85	443	if (ret)
86	2	return (ret);
87
88	441	assert(uds.sun_path[1] != '\0');
89
90	441	if (func != NULL)
91	440	ret = func(priv, &uds);
92	441	return (ret);
93	441	}
94
95		int
96	123	VUS_bind(const struct sockaddr_un uds, const char *errp)
97		{
98		int sd, e;
99		socklen_t sl;
100
101	123	sl = VUS_socklen(uds);
102
103	123	if (errp != NULL)
104	59	*errp = NULL;
105
106	123	sd = socket(PF_UNIX, SOCK_STREAM, 0);
107	123	if (sd < 0) {
108	0	if (errp != NULL)
109	0	*errp = "socket(2)";
110	0	return (-1);
111		}
112
113	123	if (unlink(uds->sun_path) != 0 && errno != ENOENT) {
114	0	if (errp != NULL)
115	0	*errp = "unlink(2)";
116	0	e = errno;
117	0	closefd(&sd);
118	0	errno = e;
119	0	return (-1);
120		}
121
122	123	if (bind(sd, (const void*)uds, sl) != 0) {
123	1	if (errp != NULL)
124	0	*errp = "bind(2)";
125	1	e = errno;
126	1	closefd(&sd);
127	1	errno = e;
128	1	return (-1);
129		}
130	122	return (sd);
131	123	}
132
133		int
134	118	VUS_connect(const char *path, int msec)
135		{
136		int s, i;
137		struct pollfd fds[1];
138		struct sockaddr_un uds;
139		socklen_t sl;
140
141	118	if (path == NULL)
142	0	return (-1);
143	118	i = sun_init(&uds, path, NULL);
144	118	if (i)
145	0	return (i);
146
147	118	assert(uds.sun_path[1] != '\0');
148
149	118	sl = VUS_socklen(&uds);
150
151	118	AN(sl);
152
153	118	s = socket(PF_UNIX, SOCK_STREAM, 0);
154	118	if (s < 0)
155	0	return (s);
156
157		/* Set the socket non-blocking */
158	118	if (msec != 0)
159	118	VTCP_nonblocking(s);
160
161	118	i = connect(s, (const void*)&uds, sl);
162	118	if (i == 0)
163	117	return (s);
164	1	if (errno != EINPROGRESS) {
165	1	closefd(&s);
166	1	return (-1);
167		}
168
169	0	if (msec < 0) {
170		/*
171		* Caller is responsible for waiting and
172		* calling VTCP_connected
173		*/
174	0	return (s);
175		}
176
177	0	assert(msec > 0);
178		/* Exercise our patience, polling for write */
179	0	fds[0].fd = s;
180	0	fds[0].events = POLLWRNORM;
181	0	fds[0].revents = 0;
182	0	i = poll(fds, 1, msec);
183
184	0	if (i == 0) {
185		/* Timeout, close and give up */
186	0	closefd(&s);
187	0	errno = ETIMEDOUT;
188	0	return (-1);
189		}
190
191	0	return (VTCP_connected(s));
192	118	}
193
194		socklen_t
195	493	VUS_socklen(const struct sockaddr_un *uds)
196		{
197		socklen_t sl;
198		const char *p;
199	493	if (*uds->sun_path)
200	492	sl = sizeof(*uds);
201		else {
202	1	p = strchr(uds->sun_path + 1, '\0');
203	1	assert(p != NULL);
204	1	sl = p - (const char*)uds;
205		}
206	493	assert(sl <= sizeof(*uds));
207	493	return sl;
208		}