1 176179b2 2021-10-04 op failed=
  2 176179b2 2021-10-04 op 
  3 176179b2 2021-10-04 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 176179b2 2021-10-04 op 	dont_check=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 176179b2 2021-10-04 op 	if [ "$dont_check" != '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 176179b2 2021-10-04 op # usage: gen_config <global config> <server config>
 39 176179b2 2021-10-04 op # generates a configuration file reg.conf
 40 176179b2 2021-10-04 op gen_config() {
 41 176179b2 2021-10-04 op 	cat <<EOF > reg.conf
 42 176179b2 2021-10-04 op $config_common
 43 176179b2 2021-10-04 op $1
 44 176179b2 2021-10-04 op server "localhost" {
 45 176179b2 2021-10-04 op 	cert "$PWD/cert.pem"
 46 176179b2 2021-10-04 op 	key  "$PWD/key.pem"
 47 176179b2 2021-10-04 op 	root "$PWD/testdata"
 48 176179b2 2021-10-04 op 	$2
 49 176179b2 2021-10-04 op }
 50 176179b2 2021-10-04 op EOF
 51 176179b2 2021-10-04 op 	if ! checkconf; then
 52 176179b2 2021-10-04 op 		echo "failed to parse the config" >&2
 53 176179b2 2021-10-04 op 		return 1
 54 176179b2 2021-10-04 op 	fi
 55 176179b2 2021-10-04 op }
 56 176179b2 2021-10-04 op 
 57 176179b2 2021-10-04 op checkconf() {
 58 176179b2 2021-10-04 op 	$gmid -n -c reg.conf >/dev/null
 59 176179b2 2021-10-04 op }
 60 176179b2 2021-10-04 op 
 61 176179b2 2021-10-04 op # usage: setup_simple_test <global config> <server config>
 62 176179b2 2021-10-04 op # generates a configuration file with `gen_config', validates it and
 63 176179b2 2021-10-04 op # launches the daemon
 64 176179b2 2021-10-04 op setup_simple_test() {
 65 176179b2 2021-10-04 op 	gen_config "$1" "$2"
 66 176179b2 2021-10-04 op 	run
 67 176179b2 2021-10-04 op }
 68 176179b2 2021-10-04 op 
 69 176179b2 2021-10-04 op # usage: get <path>
 70 176179b2 2021-10-04 op # return the body of the request on stdout
 71 176179b2 2021-10-04 op get() {
 72 176179b2 2021-10-04 op 	$gg -T30 -b $ggflags "gemini://localhost:10965/$1"
 73 176179b2 2021-10-04 op }
 74 176179b2 2021-10-04 op 
 75 176179b2 2021-10-04 op # usage: head <path>
 76 176179b2 2021-10-04 op # return the meta response line on stdout
 77 176179b2 2021-10-04 op head() {
 78 176179b2 2021-10-04 op 	$gg -T30 -h $ggflags "gemini://localhost:10965/$1"
 79 176179b2 2021-10-04 op }
 80 176179b2 2021-10-04 op 
 81 176179b2 2021-10-04 op # usage: raw <path>
 82 176179b2 2021-10-04 op # return both header and body
 83 176179b2 2021-10-04 op raw() {
 84 176179b2 2021-10-04 op 	$gg -T30 $ggflags "gemini://localhost:10965/$1"
 85 176179b2 2021-10-04 op }
 86 176179b2 2021-10-04 op 
 87 176179b2 2021-10-04 op # usage: fetch <path>
 88 176179b2 2021-10-04 op # fetches the header and the body.  They're returned in $hdr and
 89 176179b2 2021-10-04 op # $body.
 90 176179b2 2021-10-04 op fetch() {
 91 176179b2 2021-10-04 op 	if ! hdr="$(head $1)" || ! body="$(get $1)"; then
 92 176179b2 2021-10-04 op 		return 1
 93 176179b2 2021-10-04 op 	fi
 94 176179b2 2021-10-04 op }
 95 176179b2 2021-10-04 op 
 96 176179b2 2021-10-04 op # usage: fetch_hdr <path>
 97 176179b2 2021-10-04 op # fetches the header into $hdr
 98 176179b2 2021-10-04 op fetch_hdr() {
 99 176179b2 2021-10-04 op 	hdr="$(head $1)"
100 176179b2 2021-10-04 op 	body=""
101 176179b2 2021-10-04 op }
102 176179b2 2021-10-04 op 
103 176179b2 2021-10-04 op # usage: check_reply header body
104 176179b2 2021-10-04 op # checks that $hdr and $body are equal to the given strings
105 176179b2 2021-10-04 op check_reply() {
106 176179b2 2021-10-04 op 	if [ "$hdr" != "$1" ]; then
107 176179b2 2021-10-04 op 		echo "Header mismatch" >&2
108 176179b2 2021-10-04 op 		echo "wants : $1"      >&2
109 176179b2 2021-10-04 op 		echo "got   : $hdr"    >&2
110 176179b2 2021-10-04 op 		return 1
111 176179b2 2021-10-04 op 	fi
112 176179b2 2021-10-04 op 
113 176179b2 2021-10-04 op 	if [ "$body" != "$2" ]; then
114 176179b2 2021-10-04 op 		echo "Body mismatch" >&2
115 176179b2 2021-10-04 op 		echo "wants : $1"    >&2
116 176179b2 2021-10-04 op 		echo "got   : $body" >&2
117 176179b2 2021-10-04 op 		return 1
118 176179b2 2021-10-04 op 	fi
119 176179b2 2021-10-04 op }
120 176179b2 2021-10-04 op 
121 176179b2 2021-10-04 op run() {
122 176179b2 2021-10-04 op 	if check; then
123 176179b2 2021-10-04 op 		kill -HUP "$(cat gmid.pid)"
124 176179b2 2021-10-04 op 		sleep 1
125 176179b2 2021-10-04 op 		return
126 176179b2 2021-10-04 op 	fi
127 176179b2 2021-10-04 op 
128 176179b2 2021-10-04 op 	$gmid -P gmid.pid -c reg.conf
129 176179b2 2021-10-04 op 
130 176179b2 2021-10-04 op 	# give gmid time to bind the port, otherwise we end up
131 176179b2 2021-10-04 op 	# executing gg when gmid isn't ready yet.
132 176179b2 2021-10-04 op 	sleep 1
133 176179b2 2021-10-04 op }
134 176179b2 2021-10-04 op 
135 176179b2 2021-10-04 op check() {
136 176179b2 2021-10-04 op 	if [ ! -f gmid.pid ]; then
137 176179b2 2021-10-04 op 		return 1
138 176179b2 2021-10-04 op 	fi
139 176179b2 2021-10-04 op 
140 176179b2 2021-10-04 op 	pid="$(cat gmid.pid || true)"
141 176179b2 2021-10-04 op 	if [ "$pid" == "" ]; then
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 	# remember: we're running under ``set -e''
146 176179b2 2021-10-04 op 	if ps $pid >/dev/null; then
147 176179b2 2021-10-04 op 		return 0
148 176179b2 2021-10-04 op 	fi
149 176179b2 2021-10-04 op 
150 176179b2 2021-10-04 op 	return 1
151 176179b2 2021-10-04 op }
152 176179b2 2021-10-04 op 
153 176179b2 2021-10-04 op # usage: sha in out
154 176179b2 2021-10-04 op # writes the sha256 of `in' to `out'
155 176179b2 2021-10-04 op sha() {
156 176179b2 2021-10-04 op 	if which sha256 >/dev/null 2>&1; then
157 176179b2 2021-10-04 op 		sha256 < "$1" > "$2"
158 176179b2 2021-10-04 op 		return $?
159 176179b2 2021-10-04 op 	fi
160 176179b2 2021-10-04 op 
161 176179b2 2021-10-04 op 	if which sha256sum >/dev/null 2>&1; then
162 176179b2 2021-10-04 op 		sha256sum "$1" | awk '{print $1}' > "$2"
163 176179b2 2021-10-04 op 		return $?
164 176179b2 2021-10-04 op 	fi
165 176179b2 2021-10-04 op 
166 176179b2 2021-10-04 op 	echo "No sha binary found" >&2
167 176179b2 2021-10-04 op 	exit 1
168 176179b2 2021-10-04 op }
169 176179b2 2021-10-04 op 
170 176179b2 2021-10-04 op count() {
171 176179b2 2021-10-04 op 	wc -l | xargs
172 176179b2 2021-10-04 op }
173 176179b2 2021-10-04 op 
174 176179b2 2021-10-04 op quit() {
175 176179b2 2021-10-04 op 	pid="$(cat gmid.pid || true)"
176 176179b2 2021-10-04 op 	if [ "$pid" != "" ]; then
177 176179b2 2021-10-04 op 		kill $pid || true
178 176179b2 2021-10-04 op 		wait || true
179 176179b2 2021-10-04 op 	fi
180 176179b2 2021-10-04 op }
181 176179b2 2021-10-04 op 
182 176179b2 2021-10-04 op onexit() {
183 176179b2 2021-10-04 op 	rm -f bigfile bigfile.sha
184 176179b2 2021-10-04 op 	quit
185 176179b2 2021-10-04 op }