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