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 }