1 5df699d1 2022-08-29 op ran_no=0
  2 5df699d1 2022-08-29 op failed_no=0
  3 176179b2 2021-10-04 op failed=
  4 176179b2 2021-10-04 op 
  5 7a4ae106 2021-12-29 op gg="./../gg"
  6 176179b2 2021-10-04 op gmid="./../gmid"
  7 176179b2 2021-10-04 op current_test=
  8 176179b2 2021-10-04 op 
  9 176179b2 2021-10-04 op run_test() {
 10 176179b2 2021-10-04 op 	ggflags=
 11 176179b2 2021-10-04 op 	port=10965
 12 7b7def0a 2022-01-03 op 	config_common="prefork 1
 13 176179b2 2021-10-04 op ipv6 off
 14 176179b2 2021-10-04 op port $port
 15 176179b2 2021-10-04 op "
 16 176179b2 2021-10-04 op 	hdr=
 17 176179b2 2021-10-04 op 	body=
 18 ae0d3cdb 2021-12-09 op 	dont_check_server_alive=no
 19 176179b2 2021-10-04 op 
 20 5df699d1 2022-08-29 op 	ran_no=$((ran_no + 1))
 21 5df699d1 2022-08-29 op 
 22 176179b2 2021-10-04 op 	current_test=$1
 23 176179b2 2021-10-04 op 	rm -f reg.conf
 24 176179b2 2021-10-04 op 
 25 176179b2 2021-10-04 op 	if ! $1; then
 26 176179b2 2021-10-04 op 		echo "$1 failed"
 27 176179b2 2021-10-04 op 		failed="$failed $1"
 28 5df699d1 2022-08-29 op 		failed_no=$((failed_no + 1))
 29 176179b2 2021-10-04 op 	else
 30 176179b2 2021-10-04 op 		echo "$1 passed"
 31 176179b2 2021-10-04 op 	fi
 32 176179b2 2021-10-04 op 
 33 ae0d3cdb 2021-12-09 op 	if [ "$dont_check_server_alive" != 'no' ]; then
 34 176179b2 2021-10-04 op 		return
 35 176179b2 2021-10-04 op 	fi
 36 176179b2 2021-10-04 op 
 37 176179b2 2021-10-04 op 	if ! check; then
 38 176179b2 2021-10-04 op 		echo "gmid crashed?"
 39 176179b2 2021-10-04 op 		failed="$failed $1"
 40 5df699d1 2022-08-29 op 		failed_no=$((failed_no + 1))
 41 176179b2 2021-10-04 op 	fi
 42 176179b2 2021-10-04 op }
 43 176179b2 2021-10-04 op 
 44 4a2a525d 2021-10-04 op tests_done() {
 45 5df699d1 2022-08-29 op 	ok=$((ran_no - failed_no))
 46 5df699d1 2022-08-29 op 	echo
 47 5df699d1 2022-08-29 op 	echo "tests: $ran_no / passed: $ok / failed: $failed_no"
 48 4a2a525d 2021-10-04 op 	if [ "$failed" != "" ]; then
 49 4a2a525d 2021-10-04 op 		echo
 50 4a2a525d 2021-10-04 op 		echo "failed tests:$failed"
 51 4a2a525d 2021-10-04 op 		exit 1
 52 4a2a525d 2021-10-04 op 	fi
 53 4a2a525d 2021-10-04 op 	exit 0
 54 4a2a525d 2021-10-04 op }
 55 4a2a525d 2021-10-04 op 
 56 176179b2 2021-10-04 op # usage: gen_config <global config> <server config>
 57 176179b2 2021-10-04 op # generates a configuration file reg.conf
 58 176179b2 2021-10-04 op gen_config() {
 59 176179b2 2021-10-04 op 	cat <<EOF > reg.conf
 60 176179b2 2021-10-04 op $config_common
 61 176179b2 2021-10-04 op $1
 62 176179b2 2021-10-04 op server "localhost" {
 63 176179b2 2021-10-04 op 	cert "$PWD/cert.pem"
 64 176179b2 2021-10-04 op 	key  "$PWD/key.pem"
 65 176179b2 2021-10-04 op 	root "$PWD/testdata"
 66 176179b2 2021-10-04 op 	$2
 67 176179b2 2021-10-04 op }
 68 176179b2 2021-10-04 op EOF
 69 176179b2 2021-10-04 op 	if ! checkconf; then
 70 176179b2 2021-10-04 op 		echo "failed to parse the config" >&2
 71 176179b2 2021-10-04 op 		return 1
 72 176179b2 2021-10-04 op 	fi
 73 176179b2 2021-10-04 op }
 74 176179b2 2021-10-04 op 
 75 92a9f41d 2022-01-03 op set_proxy() {
 76 92a9f41d 2022-01-03 op 	cat <<EOF >>reg.conf
 77 92a9f41d 2022-01-03 op server "localhost.local" {
 78 92a9f41d 2022-01-03 op 	cert "$PWD/cert.pem"
 79 92a9f41d 2022-01-03 op 	key "$PWD/key.pem"
 80 92a9f41d 2022-01-03 op 	proxy {
 81 92a9f41d 2022-01-03 op 		relay-to "localhost:$port"
 82 92a9f41d 2022-01-03 op 		$1
 83 92a9f41d 2022-01-03 op 	}
 84 92a9f41d 2022-01-03 op }
 85 92a9f41d 2022-01-03 op EOF
 86 92a9f41d 2022-01-03 op 
 87 92a9f41d 2022-01-03 op 	if ! checkconf; then
 88 92a9f41d 2022-01-03 op 		echo "failed to parse the config" >&2
 89 92a9f41d 2022-01-03 op 		return 1
 90 92a9f41d 2022-01-03 op 	fi
 91 92a9f41d 2022-01-03 op }
 92 92a9f41d 2022-01-03 op 
 93 176179b2 2021-10-04 op checkconf() {
 94 dd208aa2 2022-01-03 op 	if ! $gmid -n -c reg.conf >/dev/null 2>&1; then
 95 dd208aa2 2022-01-03 op 		$gmid -n -c reg.conf
 96 dd208aa2 2022-01-03 op 	fi
 97 176179b2 2021-10-04 op }
 98 176179b2 2021-10-04 op 
 99 176179b2 2021-10-04 op # usage: setup_simple_test <global config> <server config>
100 176179b2 2021-10-04 op # generates a configuration file with `gen_config', validates it and
101 176179b2 2021-10-04 op # launches the daemon
102 176179b2 2021-10-04 op setup_simple_test() {
103 176179b2 2021-10-04 op 	gen_config "$1" "$2"
104 176179b2 2021-10-04 op 	run
105 176179b2 2021-10-04 op }
106 176179b2 2021-10-04 op 
107 176179b2 2021-10-04 op # usage: get <path>
108 176179b2 2021-10-04 op # return the body of the request on stdout
109 176179b2 2021-10-04 op get() {
110 7a4ae106 2021-12-29 op 	$gg -T10 $ggflags "gemini://localhost:10965/$1" || true
111 176179b2 2021-10-04 op }
112 176179b2 2021-10-04 op 
113 176179b2 2021-10-04 op # usage: head <path>
114 176179b2 2021-10-04 op # return the meta response line on stdout
115 176179b2 2021-10-04 op head() {
116 7a4ae106 2021-12-29 op 	$gg -T10 -d header $ggflags "gemini://localhost:10965/$1" || true
117 176179b2 2021-10-04 op }
118 176179b2 2021-10-04 op 
119 176179b2 2021-10-04 op # usage: fetch <path>
120 176179b2 2021-10-04 op # fetches the header and the body.  They're returned in $hdr and
121 176179b2 2021-10-04 op # $body.
122 176179b2 2021-10-04 op fetch() {
123 176179b2 2021-10-04 op 	if ! hdr="$(head $1)" || ! body="$(get $1)"; then
124 176179b2 2021-10-04 op 		return 1
125 176179b2 2021-10-04 op 	fi
126 176179b2 2021-10-04 op }
127 176179b2 2021-10-04 op 
128 176179b2 2021-10-04 op # usage: fetch_hdr <path>
129 176179b2 2021-10-04 op # fetches the header into $hdr
130 176179b2 2021-10-04 op fetch_hdr() {
131 176179b2 2021-10-04 op 	hdr="$(head $1)"
132 176179b2 2021-10-04 op 	body=""
133 176179b2 2021-10-04 op }
134 176179b2 2021-10-04 op 
135 176179b2 2021-10-04 op # usage: check_reply header body
136 176179b2 2021-10-04 op # checks that $hdr and $body are equal to the given strings
137 176179b2 2021-10-04 op check_reply() {
138 176179b2 2021-10-04 op 	if [ "$hdr" != "$1" ]; then
139 176179b2 2021-10-04 op 		echo "Header mismatch" >&2
140 176179b2 2021-10-04 op 		echo "wants : $1"      >&2
141 176179b2 2021-10-04 op 		echo "got   : $hdr"    >&2
142 176179b2 2021-10-04 op 		return 1
143 176179b2 2021-10-04 op 	fi
144 176179b2 2021-10-04 op 
145 176179b2 2021-10-04 op 	if [ "$body" != "$2" ]; then
146 176179b2 2021-10-04 op 		echo "Body mismatch" >&2
147 ce09569f 2021-12-29 op 		echo "wants : $2"    >&2
148 176179b2 2021-10-04 op 		echo "got   : $body" >&2
149 176179b2 2021-10-04 op 		return 1
150 176179b2 2021-10-04 op 	fi
151 176179b2 2021-10-04 op }
152 176179b2 2021-10-04 op 
153 176179b2 2021-10-04 op run() {
154 176179b2 2021-10-04 op 	if check; then
155 176179b2 2021-10-04 op 		kill -HUP "$(cat gmid.pid)"
156 176179b2 2021-10-04 op 		sleep 1
157 176179b2 2021-10-04 op 		return
158 176179b2 2021-10-04 op 	fi
159 176179b2 2021-10-04 op 
160 176179b2 2021-10-04 op 	$gmid -P gmid.pid -c reg.conf
161 176179b2 2021-10-04 op 
162 176179b2 2021-10-04 op 	# give gmid time to bind the port, otherwise we end up
163 176179b2 2021-10-04 op 	# executing gg when gmid isn't ready yet.
164 176179b2 2021-10-04 op 	sleep 1
165 176179b2 2021-10-04 op }
166 176179b2 2021-10-04 op 
167 176179b2 2021-10-04 op check() {
168 176179b2 2021-10-04 op 	if [ ! -f gmid.pid ]; then
169 176179b2 2021-10-04 op 		return 1
170 176179b2 2021-10-04 op 	fi
171 176179b2 2021-10-04 op 
172 176179b2 2021-10-04 op 	pid="$(cat gmid.pid || true)"
173 5266f500 2022-01-30 op 	if [ "$pid" = "" ]; then
174 176179b2 2021-10-04 op 		return 1
175 176179b2 2021-10-04 op 	fi
176 176179b2 2021-10-04 op 
177 176179b2 2021-10-04 op 	# remember: we're running under ``set -e''
178 176179b2 2021-10-04 op 	if ps $pid >/dev/null; then
179 176179b2 2021-10-04 op 		return 0
180 176179b2 2021-10-04 op 	fi
181 176179b2 2021-10-04 op 
182 176179b2 2021-10-04 op 	return 1
183 176179b2 2021-10-04 op }
184 176179b2 2021-10-04 op 
185 176179b2 2021-10-04 op count() {
186 176179b2 2021-10-04 op 	wc -l | xargs
187 176179b2 2021-10-04 op }
188 176179b2 2021-10-04 op 
189 176179b2 2021-10-04 op quit() {
190 176179b2 2021-10-04 op 	pid="$(cat gmid.pid || true)"
191 176179b2 2021-10-04 op 	if [ "$pid" != "" ]; then
192 176179b2 2021-10-04 op 		kill $pid || true
193 176179b2 2021-10-04 op 		wait || true
194 176179b2 2021-10-04 op 	fi
195 176179b2 2021-10-04 op }
196 176179b2 2021-10-04 op 
197 176179b2 2021-10-04 op onexit() {
198 176179b2 2021-10-04 op 	rm -f bigfile bigfile.sha
199 176179b2 2021-10-04 op 	quit
200 176179b2 2021-10-04 op }