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