summaryrefslogtreecommitdiff
path: root/url.c
blob: 26e737f0c0dc15cc889ed9fb48f5088bed1652b2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*
 * Copyright 2023, Vincent Douillet <vincent@vdouillet.fr>
 *
 * 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.
 *
 * 3. Neither the name of the copyright holder nor the names of its contributors
 * may be used to endorse or promote products derived from this software without
 * specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS 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 THE COPYRIGHT HOLDER 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/types.h>

#include <assert.h>
#include <limits.h>
#include <stdarg.h>
#include <stdint.h>
#include <stdlib.h>
#include <kcgi.h>
#include <string.h>

#include "config.h"
#include "str.h"
#include "url.h"

char   *
url_encode(const char *str)
{
	char   *result;

	if ((result = khttp_urlencode(str)) == NULL)
		return NULL;

	str_replace(result, '+', ' ');
	return result;
}

size_t
url_build(char *dst, size_t dst_sz,...)
{
	va_list path_list;
	char 	p [PATH_MAX];
	char   *encoded, *path, *slash;
	size_t 	c_len, p_len, w_len;

	dst[0] = '\0';
	w_len = 0;
	va_start(path_list, dst_sz);
	while ((path = va_arg(path_list, char *)) != NULL) {
		/* no more space in dst buffer */
		if (w_len >= dst_sz) {
			w_len = 0;
			goto end;
		}

		if (path[0] == '\0')
			continue;

		if (path[0] == '.' && path[1] == '\0')
			continue;

		while (path[0] != '\0') {
			/* no more space in dst buffer */
			if (w_len >= dst_sz) {
				w_len = 0;
				goto end;
			}

			/* skip heading slashes */
			slash = strchr(path, '/');
			if (slash == path) {
				path++;
				continue;
			}

			/* copy string to encode into buffer
			 * if no slash found, copy everything, otherwise stop
			 * before the slash */
			c_len = slash == NULL ? strlen(path) :
			    (size_t) (slash - path);
			if (c_len >= PATH_MAX) {
				w_len = 0;
				goto end;
			}
			p_len = strlcpy(p, path, c_len + 1);
			if (p_len < c_len) {
				w_len = 0;
				goto end;
			}

			/* encode string */
			if ((encoded = url_encode(p)) == NULL) {
				w_len = 0;
				goto end;
			}

			/* append it to the path */
			strlcat(dst, "/", dst_sz);
			w_len = strlcat(dst, encoded, dst_sz);
			free(encoded);
			path += c_len;
		}
	}

end:
	va_end(path_list);
	return w_len;
}