]> ruderich.org/simon Gitweb - tlsproxy/tlsproxy.git/blob - tests/common.sh
37c691beaf71bbb43480940b98dc58c9e17e1875
[tlsproxy/tlsproxy.git] / tests / common.sh
1 # Functions used by all tests.
2 #
3 # Copyright (C) 2011-2013  Simon Ruderich
4 #
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
17
18
19 set -e
20
21
22 cleanup() {
23     rm -f \
24         tmp tlsproxy-log \
25         certificate-localhost-proxy.pem certificate-localhost-server.pem \
26         proxy-ca-key.pem proxy-ca.pem proxy-invalid.pem proxy-key.pem
27 }
28 stop_servers() {
29     kill $pid_server || echo 'failed to kill gnutls-serv'
30     kill $pid_tlsproxy || echo 'failed to kill tlsproxy'
31 }
32 abort() {
33     echo "abort: $*"
34     stop_servers
35     exit 1
36 }
37 tlsproxy_setup() {
38     "$srcdir/../src/tlsproxy-setup" >/dev/null 2>&1
39 }
40 tlsproxy_add() {
41     "$srcdir/../src/tlsproxy-add" "$1" "$srcdir/$2" >/dev/null 2>&1
42 }
43 tlsproxy() {
44     ../src/tlsproxy -d2 "$@" >tlsproxy-log 2>&1 &
45     pid_tlsproxy=$!
46 }
47 server() {
48     gnutls-serv --http --port 4712 "$@" >/dev/null 2>&1 &
49     pid_server=$!
50 }
51 client() {
52     ./client ./proxy-ca.pem "$@" > tmp 2>&1
53 }
54
55 wait_for_ports() {
56     while :; do
57         sleep 1
58
59         # Check if each given port can be reached.
60         ready=1
61         for port in "$@"; do
62             printf 'invalid request\r\n\r\n' \
63                 | nc localhost "$port" >/dev/null 2>&1 \
64                 || ready=
65         done
66
67         if test -n "$ready"; then
68             break
69         fi
70
71         echo "waiting for ports $* ..."
72     done
73 }
74
75 test_proxy_failure() {
76     grep 'proxy failure' tmp >/dev/null \
77         || abort 'test_proxy_failure'
78     grep 'response: HTTP/1.0 503 Forwarding failure' tmp >/dev/null \
79         || abort 'test_proxy_failure 2'
80 }
81 test_proxy_successful() {
82     grep 'response: HTTP/1.0 200 Connection established' tmp >/dev/null \
83         || abort 'test_proxy_successful'
84 }
85 test_invalid_certificate() {
86     grep 'certificate invalid' tmp >/dev/null \
87         || abort 'test_invalid_certificate'
88 }
89 test_no_invalid_certificate() {
90     grep 'certificate invalid' tmp >/dev/null \
91         && abort 'test_no_invalid_certificate' || true
92 }