1 #!/bin/sh
2 
3 set -e
4 
5 ggflags=
6 
7 # usage: config <global config> <stuff for localhost>
8 # generates a configuration file reg.conf
9 config() {
10 	cat <<EOF > reg.conf
11 ipv6 off
12 port 10965
13 $1
14 server "localhost" {
15 	cert "$PWD/cert.pem"
16 	key  "$PWD/key.pem"
17 	root "$PWD/testdata"
18 	$2
19 }
20 EOF
21 }
22 
23 checkconf() {
24 	./../gmid -n -c reg.conf
25 }
26 
27 # usage: get <path>
28 # return the body of the request on stdout
29 get() {
30 	./../gg -b $ggflags "gemini://localhost:10965/$1"
31 }
32 
33 # usage: head <path>
34 # return the meta response line on stdout
35 head() {
36 	./../gg -h $ggflags "gemini://localhost:10965/$1"
37 }
38 
39 # usage: raw <path>
40 # return both header and body
41 raw() {
42 	./../gg $ggflags "gemini://localhost:10965/$1"
43 }
44 
45 run() {
46 	# filter out logs for GET requests
47 	(./../gmid -f -c reg.conf 2>&1 | grep -v GET) >&2 &
48 	pid=$!
49 	# give gmid time to bind the port, otherwise we end up
50 	# executing gg when gmid isn't ready yet.
51 	sleep 1
52 }
53 
54 # usage: check [exit-message]
55 # check if gmid is still running
56 check() {
57 	if ! ps $pid >/dev/null; then
58 		echo ${1:-"gmid crashed?"}
59 		exit 1
60 	fi
61 }
62 
63 restart() {
64 	pkill -SIGHUP gmid
65 	sleep 1
66 }
67 
68 # quit gmid
69 quit() {
70 	pkill gmid || true
71 	wait || true
72 }
73 
74 count() {
75 	wc -l | xargs
76 }
77 
78 # usage: eq a b errmsg
79 # if a and b aren't equal strings, exit with errmsg
80 eq() {
81 	if ! [ "$1" = "$2" ]; then
82 		echo "$3: \"$1\" not equal \"$2\""
83 		exit 1
84 	fi
85 }
86 
87 onexit() {
88 	rm -f bigfile bigfile.sha
89 	quit
90 }
91 
92 # tests
93 
94 trap 'onexit' INT TERM EXIT
95 
96 endl=`printf "\r\n"`
97 lf=`echo`
98 
99 config "" ""
100 checkconf
101 run
102 
103 eq "$(head /)"		"20 text/gemini"	"Unexpected head for /"
104 eq "$(get /)"		"# hello world$ln"	"Unexpected body for /"
105 echo OK GET /
106 
107 eq "$(head /foo)"	"51 not found"		"Unexpected head /foo"
108 eq "$(get /foo)"	""			"Unexpected body /foo"
109 echo OK GET /foo
110 
111 # should redirect if asked for a directory but without the trailing /
112 eq "$(head /dir)"	"30 /dir/"		"Unexpected redirect for /dir"
113 eq "$(get  /dir)"	""			"Unexpected body for redirect"
114 echo OK GET /dir
115 
116 # 51 for a directory without index.gmi
117 eq "$(head /dir/)"	"51 not found"		"Unexpected head for /"
118 eq "$(get  /dir/)"	""			"Unexpected body for error"
119 echo OK GET /dir/
120 
121 eq "$(head /dir/foo.gmi)" "20 text/gemini"	"Unexpected head for /dir/foo.gmi"
122 eq "$(get  /dir/foo.gmi)" "# hello world$ln"	"Unexpected body for /dir/foo.gmi"
123 echo OK GET /dir/foo.gmi
124 
125 # try a big file
126 eq "$(head /bigfile)"	"20 application/octet-stream" "Unexpected head for /bigfile"
127 get /bigfile > bigfile
128 ./sha bigfile bigfile.sha
129 eq "$(cat bigfile.sha)"	"$(cat testdata/bigfile.sha)" "Unexpected sha for /bigfile"
130 echo OK GET /bigfile
131 
132 # shouldn't be executing cgi scripts
133 eq "$(head /hello)"	"20 application/octet-stream" "Unexpected head for /hello"
134 echo OK GET /hello
135 
136 check "should be running"
137 
138 # try with custom mime
139 config 'mime "text/x-funny-text" "gmi"' 'default type "application/x-trash"'
140 checkconf
141 restart
142 
143 eq "$(head /)"		"20 text/x-funny-text"		"Unexpected head for /"
144 echo OK GET / with custom mime
145 
146 eq "$(head /hello)"	"20 application/x-trash"	"Unexpected head for /hello"
147 echo OK GET /hello with custom mime
148 
149 check "should be running"
150 
151 # try with custom lang
152 config '' 'lang "it"'
153 checkconf
154 restart
155 
156 eq "$(head /)"		"20 text/gemini; lang=it"	"Unexpected head for /"
157 echo OK GET / with custom lang
158 
159 check "should be running"
160 
161 # make sure we can use different lang in different location rules
162 config '' 'lang "it" location "/en/*" { lang "en" } location "/de/*" { lang "de" }'
163 checkconf
164 echo OK parse multiple locations correctly
165 restart
166 
167 # try with CGI scripts
168 config '' 'cgi "*"'
169 checkconf
170 restart
171 
172 eq "$(head /hello)"	"20 text/gemini"		"Unexpected head for /hello"
173 eq "$(get  /hello)"	"# hello world$ln"		"Unexpected body for /hello"
174 echo OK GET /hello with cgi
175 
176 eq "$(head /slow)"	"20 text/gemini"		"Unexpected head for /slow"
177 eq "$(get  /slow)"	"# hello world$ln"		"Unexpected body for /slow"
178 echo OK GET /slow with cgi
179 
180 eq "$(head /err)"	"42 CGI error"	"Unexpected head for /err"
181 eq "$(get  /err)"	""		"Unexpected body for /err"
182 echo OK GET /err with cgi
183 
184 eq "$(raw /invalid | wc -c | xargs)" 2048		"Unexpected body for /invalid"
185 echo OK GET /invalid with cgi
186 
187 # try a big file
188 eq "$(head /serve-bigfile)"	"20 application/octet-stream" "Unexpected head for /serve-bigfile"
189 get /bigfile > bigfile
190 ./sha bigfile bigfile.sha
191 eq "$(cat bigfile.sha)"	"$(cat testdata/bigfile.sha)" "Unexpected sha for /serve-bigfile"
192 echo OK GET /serve-bigfile with cgi
193 
194 # ensure we split the query correctly
195 eq "$(get /env | awk /^-/ | count)"		1	"Unexpected number of arguments"
196 eq "$(get /env?foo | awk /^-/ | count)"		2	"Unexpected number of arguments"
197 eq "$(get /env?foo+bar | awk /^-/ | count)"	3	"Unexpected number of arguments"
198 eq "$(get /env?foo+bar=5 | awk /^-/ | count)"	1	"Unexpected number of arguments"
199 eq "$(get /env?foo+bar%3d5 | awk /^-/ | count)"	3	"Unexpected number of arguments"
200 
201 check "should be running"
202 
203 config '' 'index "foo.gmi"'
204 checkconf
205 restart
206 
207 eq "$(head /dir/)"	"20 text/gemini"	"Unexpected head for /"
208 eq "$(get  /dir/)"	"# hello world$ln"	"Unexpected body for error"
209 echo OK GET /dir/ with custom index
210 
211 check "should be running"
212 
213 config '' 'location "/dir/*" { default type "text/plain" index "hello" }'
214 checkconf
215 restart
216 
217 eq "$(head /dir/hello)"	"20 text/plain"			"Unexpected head for /"
218 echo OK GET /dir/hello with location and default type
219 
220 eq "$(head /dir/)"	"20 text/plain"			"Unexpected head for /dir/"
221 eq "$(get  /dir/|tail -1)" 'echo "# hello world"'	"Unexpected body for /dir/"
222 echo OK GET /dir/ with location and custom index
223 
224 check "should be running"
225 
226 config '' 'location "/dir/*" { auto index on }'
227 checkconf
228 restart
229 
230 eq "$(head /)"		"20 text/gemini"		"Unexpected head for /"
231 eq "$(get  /)"		"# hello world$ln"		"Unexpected body for /"
232 echo OK GET / with auto index
233 
234 eq "$(head /dir)"		"30 /dir/"		"Unexpected head for /dir"
235 eq "$(head /dir/)"		"20 text/gemini"	"Unexpected head for /dir/"
236 eq "$(get /dir/|wc -l|xargs)"	"5"			"Unexpected body for /dir/"
237 echo OK GET /dir/ with auto index on
238 
239 check "should be running"
240 
241 # test block return and strip
242 
243 config '' 'location "*" { block }'
244 checkconf
245 restart
246 
247 eq "$(head /)"		"40 temporary failure"		"Unexpected head for /"
248 eq "$(get /)"		""				"Unexpected body for /"
249 echo OK GET / with block
250 
251 eq "$(head /nonexists)"	"40 temporary failure"		"Unexpected head for /nonexists"
252 eq "$(get /nonexists)"	""				"Unexpected body for /nonexists"
253 echo OK GET /nonexists with block
254 
255 check "should be running"
256 
257 config '' '
258 location "/dir" {
259 	strip 1
260 	block return 40 "%% %p %q %P %N test"
261 }
262 location "*" {
263 	strip 99
264 	block return 40 "%% %p %q %P %N test"
265 }'
266 checkconf
267 restart
268 
269 eq "$(head /dir/foo.gmi)"	"40 % /foo.gmi  10965 localhost test"
270 echo OK GET /dir/foo.gmi with strip and block
271 
272 eq "$(head /bigfile)"		"40 %   10965 localhost test"
273 echo OK GET /bigfile with strip and block
274 
275 check "should be running"
276 
277 # test the entrypoint
278 
279 config '' 'entrypoint "/env"'
280 checkconf
281 restart
282 
283 eq "$(head /foo/bar)"	"20 text/plain; lang=en"	"Unknown head for /foo/bar"
284 eq "$(get /foo/bar|grep PATH_INFO)" "PATH_INFO=/foo/bar" "Unexpected PATH_INFO"
285 echo OK GET /foo/bar with entrypoint
286 
287 # test with require ca
288 
289 config '' 'require client ca "'$PWD'/testca.pem"'
290 checkconf
291 restart
292 
293 eq "$(head /)"		"60 client certificate required" "Unexpected head for /"
294 echo OK GET / without client certificate
295 
296 ggflags="-C valid.crt -K valid.key"
297 eq "$(head /)"		"20 text/gemini"		"Unexpected head for /"
298 echo OK GET / with valid client certificate
299 
300 ggflags="-C invalid.cert.pem -K invalid.key.pem"
301 eq "$(head /)"		"61 certificate not authorised"	"Unexpected head for /"
302 echo OK GET / with invalid client certificate
303 
304 ggflags=''
305 
306 quit