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