1 /*
2  * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <err.h>
18 #include <errno.h>
19 #include <stdio.h>
20 #include <string.h>
21 
22 #include "iri.h"
23 
24 static int
25 resolve(const char *base, const char *ref, const char *expected)
26 {
27 	static struct iri	i;
28 	char			buf[512];
29 
30 	if (iri_parse(base, ref, &i) == -1) {
31 		fprintf(stderr, "FAIL resolve(\"%s\", \"%s\") %s\n", base, ref,
32 		    strerror(errno));
33 		return (1);
34 	}
35 
36 	if (iri_unparse(&i, buf, sizeof(buf)) == -1) {
37 		fprintf(stderr, "FAIL resolve(\"%s\", \"%s\") %s\n", base, ref,
38 		    strerror(errno));
39 		return (1);
40 	}
41 
42 	if (strcmp(buf, expected) != 0) {
43 		fprintf(stderr, "FAIL resolve(\"%s\", \"%s\")\n", base, ref);
44 		fprintf(stderr, "got:\t%s\n", buf);
45 		fprintf(stderr, "want:\t%s\n", expected);
46 		return (1);
47 	}
48 
49 	fprintf(stderr, "OK resolve(\"%s\", \"%s\") -> %s\n", base, ref,
50 	    expected);
51 	return (0);
52 }
53 
54 static int
55 setquery(const char *iri, const char *query, const char *expected)
56 {
57 	static struct iri	i;
58 	char			buf[512];
59 
60 	if (iri_parse(NULL, iri, &i) == -1) {
61 		fprintf(stderr, "FAIL can't parse <%s>: %s\n", iri,
62 		    strerror(errno));
63 		return (1);
64 	}
65 
66 	if (iri_setquery(&i, query) == -1) {
67 		fprintf(stderr, "FAIL setting query \"%s\": %s\n", query,
68 		    strerror(errno));
69 		return (1);
70 	}
71 
72 	if (iri_unparse(&i, buf, sizeof(buf)) == -1) {
73 		fprintf(stderr, "FAIL unparsing <%s> with query %s\n",
74 		    iri, query);
75 		return (1);
76 	}
77 
78 	if (strcmp(buf, expected) != 0) {
79 		fprintf(stderr, "FAIL setquery(\"%s\", \"%s\")\n", iri, query);
80 		fprintf(stderr, "got:\t%s\n", buf);
81 		fprintf(stderr, "want:\t%s\n", expected);
82 		return (1);
83 	}
84 
85 	fprintf(stderr, "OK setquery(\"%s\", \"%s\") -> %s\n", iri, query,
86 	    expected);
87 	return (0);
88 }
89 
90 int
91 main(void)
92 {
93 	const char	*base;
94 	int		 ret = 0;
95 
96 	/* RFC 3986 tests */
97 
98 	base = "http://a/b/c/d;p?q";
99 
100 	ret |= resolve(base, "g:h", "g:h");
101 	ret |= resolve(base, "g", "http://a/b/c/g");
102 	ret |= resolve(base, "./g", "http://a/b/c/g");
103 	ret |= resolve(base, "g/", "http://a/b/c/g/");
104 	ret |= resolve(base, "/g", "http://a/g");
105 
106 	/*
107 	 * the RFC says "http://g" but we always normalize an
108 	 * empty path to "/" if there is an authority.
109 	 */
110 	ret |= resolve(base, "//g", "http://g/");
111 
112 	ret |= resolve(base, "?y", "http://a/b/c/d;p?y");
113 	ret |= resolve(base, "g?y", "http://a/b/c/g?y");
114 	ret |= resolve(base, "#s", "http://a/b/c/d;p?q#s");
115 	ret |= resolve(base, "g#s", "http://a/b/c/g#s");
116 	ret |= resolve(base, "g?y#s", "http://a/b/c/g?y#s");
117 	ret |= resolve(base, ";x", "http://a/b/c/;x");
118 	ret |= resolve(base, "g;x", "http://a/b/c/g;x");
119 	ret |= resolve(base, "g;x?y#s", "http://a/b/c/g;x?y#s");
120 	ret |= resolve(base, "", "http://a/b/c/d;p?q");
121 	ret |= resolve(base, ".", "http://a/b/c/");
122 	ret |= resolve(base, "./", "http://a/b/c/");
123 	ret |= resolve(base, "..", "http://a/b/");
124 	ret |= resolve(base, "../", "http://a/b/");
125 	ret |= resolve(base, "../g", "http://a/b/g");
126 	ret |= resolve(base, "../..", "http://a/");
127 	ret |= resolve(base, "../../", "http://a/");
128 	ret |= resolve(base, "../../g", "http://a/g");
129 
130 	ret |= resolve(base, "../../../g", "http://a/g");
131 	ret |= resolve(base, "../../../../g", "http://a/g");
132 	ret |= resolve(base, "/./g", "http://a/g");
133 	ret |= resolve(base, "/../g", "http://a/g");
134 	ret |= resolve(base, "g.", "http://a/b/c/g.");
135 	ret |= resolve(base, ".g", "http://a/b/c/.g");
136 	ret |= resolve(base, "g..", "http://a/b/c/g..");
137 	ret |= resolve(base, "..g", "http://a/b/c/..g");
138 
139 	ret |= resolve(base, "./../g", "http://a/b/g");
140 	ret |= resolve(base, "./g/.", "http://a/b/c/g/");
141 	ret |= resolve(base, "g/./h", "http://a/b/c/g/h");
142 	ret |= resolve(base, "g/../h", "http://a/b/c/h");
143 	ret |= resolve(base, "g;x=1/./y", "http://a/b/c/g;x=1/y");
144 	ret |= resolve(base, "g;x=1/../y", "http://a/b/c/y");
145 
146 	ret |= resolve(base, "g?y/./x", "http://a/b/c/g?y/./x");
147 	ret |= resolve(base, "g?y/../x", "http://a/b/c/g?y/../x");
148 	ret |= resolve(base, "g#s/./x", "http://a/b/c/g#s/./x");
149 	ret |= resolve(base, "g#s/../x", "http://a/b/c/g#s/../x");
150 
151 	ret |= resolve(base, "http:g", "http:g");
152 
153 	/* extra tests */
154 
155 	base = "gemini://a/b/c";
156 
157 	ret |= setquery(base, "hw", "gemini://a/b/c?hw");
158 	ret |= setquery(base, "h w", "gemini://a/b/c?h%20w");
159 	ret |= setquery(base, "100%", "gemini://a/b/c?100%25");
160 	ret |= setquery(base, "%20", "gemini://a/b/c?%2520");
161 
162 	return (ret);
163 }