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 # Terminate all running servers when the script terminates.
  23 trap 'stop_servers' 0
  24 # Same when the user presses Ctrl-C.
  25 trap 'abort SIGINT' INT
  26
  27
  28 cleanup() {
  29     rm -f \
  30         tmp tlsproxy-log \
  31         certificate-localhost-proxy.pem certificate-localhost-server.pem
  32 }
  33 stop_servers() {
  34     if test -n "$pid_server"; then
  35         kill $pid_server || echo 'failed to kill gnutls-serv'
  36     fi
  37     if test -n "$pid_tlsproxy"; then
  38         kill $pid_tlsproxy || echo 'failed to kill tlsproxy'
  39     fi
  40 }
  41 abort() {
  42     echo "abort: $*"
  43     stop_servers
  44     exit 1
  45 }
  46 tlsproxy_setup() {
  47     # Don't waste time running `tlsproxy-setup` when all files are still
  48     # present.
  49     if test -f proxy-ca-key.pem &&
  50             test -f proxy-ca.pem &&
  51             test -f proxy-dh.pem &&
  52             test -f proxy-invalid.pem &&
  53             test -f proxy-key.pem
  54     then
  55         return
  56     fi
  57
  58     "$srcdir/../src/tlsproxy-setup" >/dev/null 2>&1
  59 }
  60 tlsproxy_add() {
  61     # `make distcheck` marks the source PEM-file as read-only which is copied
  62     # over when using `cp`. This causes a check failure when tlsproxy_add() is
  63     # called again with the same hostname. Instead use redirection which
  64     # doesn't use `cp`.
  65     "$srcdir/../src/tlsproxy-add" "$1" < "$srcdir/$2" >/dev/null 2>&1
  66 }
  67 tlsproxy() {
  68     #valgrind --leak-check=full --error-exitcode=1 --track-fds=yes \
  69     ../src/tlsproxy -d2 "$@" >tlsproxy-log 2>&1 &
  70     pid_tlsproxy=$!
  71 }
  72 server() {
  73     gnutls-serv --http --port 4712 "$@" >/dev/null 2>&1 &
  74     pid_server=$!
  75 }
  76 client() {
  77     ./client ./proxy-ca.pem "$@" > tmp 2>&1
  78 }
  79
  80 wait_for_ports() {
  81     for x in 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20; do
  82         sleep 1
  83
  84         # Check if each given port can be reached.
  85         ready=1
  86         for port in "$@"; do
  87             printf 'invalid request\r\n\r\n' \
  88                 | nc localhost "$port" >/dev/null 2>&1 \
  89                 || ready=
  90         done
  91
  92         if test -n "$ready"; then
  93             break
  94         fi
  95
  96         echo "waiting for ports $* ..."
  97     done
  98 }
  99
 100 test_proxy_failure() {
 101     grep 'proxy failure' tmp >/dev/null \
 102         || abort 'test_proxy_failure'
 103     grep 'response: HTTP/1.0 503 Forwarding failure' tmp >/dev/null \
 104         || abort 'test_proxy_failure 2'
 105 }
 106 test_proxy_authentication_failure() {
 107     grep 'proxy failure' tmp >/dev/null \
 108         || abort 'test_proxy_authentication_failure'
 109     grep 'response: HTTP/1.0 407 Proxy Authentication Required' tmp >/dev/null \
 110         || abort 'test_proxy_authentication_failure 2'
 111 }
 112 test_authentication_missing() {
 113     grep 'response: HTTP/1.0 407 Proxy Authentication Required' tmp >/dev/null \
 114         || abort 'test_authentication_missing'
 115 }
 116 test_proxy_successful() {
 117     grep 'response: HTTP/1.0 200 Connection established' tmp >/dev/null \
 118         || abort 'test_proxy_successful'
 119 }
 120 test_invalid_certificate() {
 121     grep 'certificate invalid' tmp >/dev/null \
 122         || abort 'test_invalid_certificate'
 123 }
 124 test_no_invalid_certificate() {
 125     grep 'certificate invalid' tmp >/dev/null \
 126         && abort 'test_no_invalid_certificate' || true
 127 }