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         proxy-ca-key.pem proxy-ca.pem proxy-invalid.pem proxy-key.pem
  33 }
  34 stop_servers() {
  35     if test -n "$pid_server"; then
  36         kill $pid_server || echo 'failed to kill gnutls-serv'
  37     fi
  38     if test -n "$pid_tlsproxy"; then
  39         kill $pid_tlsproxy || echo 'failed to kill tlsproxy'
  40     fi
  41 }
  42 abort() {
  43     echo "abort: $*"
  44     stop_servers
  45     exit 1
  46 }
  47 tlsproxy_setup() {
  48     "$srcdir/../src/tlsproxy-setup" >/dev/null 2>&1
  49 }
  50 tlsproxy_add() {
  51     "$srcdir/../src/tlsproxy-add" "$1" "$srcdir/$2" >/dev/null 2>&1
  52 }
  53 tlsproxy() {
  54     ../src/tlsproxy -d2 "$@" >tlsproxy-log 2>&1 &
  55     pid_tlsproxy=$!
  56 }
  57 server() {
  58     gnutls-serv --http --port 4712 "$@" >/dev/null 2>&1 &
  59     pid_server=$!
  60 }
  61 client() {
  62     ./client ./proxy-ca.pem "$@" > tmp 2>&1
  63 }
  64
  65 wait_for_ports() {
  66     while :; do
  67         sleep 1
  68
  69         # Check if each given port can be reached.
  70         ready=1
  71         for port in "$@"; do
  72             printf 'invalid request\r\n\r\n' \
  73                 | nc localhost "$port" >/dev/null 2>&1 \
  74                 || ready=
  75         done
  76
  77         if test -n "$ready"; then
  78             break
  79         fi
  80
  81         echo "waiting for ports $* ..."
  82     done
  83 }
  84
  85 test_proxy_failure() {
  86     grep 'proxy failure' tmp >/dev/null \
  87         || abort 'test_proxy_failure'
  88     grep 'response: HTTP/1.0 503 Forwarding failure' tmp >/dev/null \
  89         || abort 'test_proxy_failure 2'
  90 }
  91 test_proxy_successful() {
  92     grep 'response: HTTP/1.0 200 Connection established' tmp >/dev/null \
  93         || abort 'test_proxy_successful'
  94 }
  95 test_invalid_certificate() {
  96     grep 'certificate invalid' tmp >/dev/null \
  97         || abort 'test_invalid_certificate'
  98 }
  99 test_no_invalid_certificate() {
 100     grep 'certificate invalid' tmp >/dev/null \
 101         && abort 'test_no_invalid_certificate' || true
 102 }