+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+Permalink
+
+
+
+
+
+
+
+
+
+
+
+
+ 
+ 
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ executable file
+
+ 162 lines (151 sloc)
+
+ 3.57 KB
+
+
+
+
+
+
+
+| + | #!/usr/bin/env bash | +
| + | # Use this script to test if a given TCP host/port are available | +
| + | + | +
| + | cmdname=$(basename $0) | +
| + | + | +
| + | echoerr() { if [[ $QUIET -ne 1 ]]; then echo "$@" 1>&2; fi } | +
| + | + | +
| + | usage() | +
| + | { | +
| + | cat << USAGE >&2 | +
| + | Usage: | +
| + | $cmdname host:port [-s] [-t timeout] [-- command args] | +
| + | -h HOST | --host=HOST Host or IP under test | +
| + | -p PORT | --port=PORT TCP port under test | +
| + | Alternatively, you specify the host and port as host:port | +
| + | -s | --strict Only execute subcommand if the test succeeds | +
| + | -q | --quiet Don't output any status messages | +
| + | -t TIMEOUT | --timeout=TIMEOUT | +
| + | Timeout in seconds, zero for no timeout | +
| + | -- COMMAND ARGS Execute command with args after the test finishes | +
| + | USAGE | +
| + | exit 1 | +
| + | } | +
| + | + |
| + | wait_for() | +
| + | { | +
| + | if [[ $TIMEOUT -gt 0 ]]; then | +
| + | echoerr "$cmdname: waiting $TIMEOUT seconds for $HOST:$PORT" | +
| + | else | +
| + | echoerr "$cmdname: waiting for $HOST:$PORT without a timeout" | +
| + | fi | +
| + | start_ts=$(date +%s) | +
| + | while : | +
| + | do | +
| + | (echo > /dev/tcp/$HOST/$PORT) >/dev/null 2>&1 | +
| + | result=$? | +
| + | if [[ $result -eq 0 ]]; then | +
| + | end_ts=$(date +%s) | +
| + | echoerr "$cmdname: $HOST:$PORT is available after $((end_ts - start_ts)) seconds" | +
| + | break | +
| + | fi | +
| + | sleep 1 | +
| + | done | +
| + | return $result | +
| + | } | +
| + | + |
| + | wait_for_wrapper() | +
| + | { | +
| + | # In order to support SIGINT during timeout: http://unix.stackexchange.com/a/57692 | +
| + | if [[ $QUIET -eq 1 ]]; then | +
| + | timeout $TIMEOUT $0 --quiet --child --host=$HOST --port=$PORT --timeout=$TIMEOUT & | +
| + | else | +
| + | timeout $TIMEOUT $0 --child --host=$HOST --port=$PORT --timeout=$TIMEOUT & | +
| + | fi | +
| + | PID=$! | +
| + | trap "kill -INT -$PID" INT | +
| + | wait $PID | +
| + | RESULT=$? | +
| + | if [[ $RESULT -ne 0 ]]; then | +
| + | echoerr "$cmdname: timeout occurred after waiting $TIMEOUT seconds for $HOST:$PORT" | +
| + | fi | +
| + | return $RESULT | +
| + | } | +
| + | + |
| + | # process arguments | +
| + | while [[ $# -gt 0 ]] | +
| + | do | +
| + | case "$1" in | +
| + | *:* ) | +
| + | hostport=(${1//:/ }) | +
| + | HOST=${hostport[0]} | +
| + | PORT=${hostport[1]} | +
| + | shift 1 | +
| + | ;; | +
| + | --child) | +
| + | CHILD=1 | +
| + | shift 1 | +
| + | ;; | +
| + | -q | --quiet) | +
| + | QUIET=1 | +
| + | shift 1 | +
| + | ;; | +
| + | -s | --strict) | +
| + | STRICT=1 | +
| + | shift 1 | +
| + | ;; | +
| + | -h) | +
| + | HOST="$2" | +
| + | if [[ $HOST == "" ]]; then break; fi | +
| + | shift 2 | +
| + | ;; | +
| + | --host=*) | +
| + | HOST="${1#*=}" | +
| + | shift 1 | +
| + | ;; | +
| + | -p) | +
| + | PORT="$2" | +
| + | if [[ $PORT == "" ]]; then break; fi | +
| + | shift 2 | +
| + | ;; | +
| + | --port=*) | +
| + | PORT="${1#*=}" | +
| + | shift 1 | +
| + | ;; | +
| + | -t) | +
| + | TIMEOUT="$2" | +
| + | if [[ $TIMEOUT == "" ]]; then break; fi | +
| + | shift 2 | +
| + | ;; | +
| + | --timeout=*) | +
| + | TIMEOUT="${1#*=}" | +
| + | shift 1 | +
| + | ;; | +
| + | --) | +
| + | shift | +
| + | CLI="$@" | +
| + | break | +
| + | ;; | +
| + | --help) | +
| + | usage | +
| + | ;; | +
| + | *) | +
| + | echoerr "Unknown argument: $1" | +
| + | usage | +
| + | ;; | +
| + | esac | +
| + | done | +
| + | + |
| + | if [[ "$HOST" == "" || "$PORT" == "" ]]; then | +
| + | echoerr "Error: you need to provide a host and port to test." | +
| + | usage | +
| + | fi | +
| + | + |
| + | TIMEOUT=${TIMEOUT:-15} | +
| + | STRICT=${STRICT:-0} | +
| + | CHILD=${CHILD:-0} | +
| + | QUIET=${QUIET:-0} | +
| + | + |
| + | if [[ $CHILD -gt 0 ]]; then | +
| + | wait_for | +
| + | RESULT=$? | +
| + | exit $RESULT | +
| + | else | +
| + | if [[ $TIMEOUT -gt 0 ]]; then | +
| + | wait_for_wrapper | +
| + | RESULT=$? | +
| + | else | +
| + | wait_for | +
| + | RESULT=$? | +
| + | fi | +
| + | fi | +
| + | + |
| + | if [[ $CLI != "" ]]; then | +
| + | if [[ $RESULT -ne 0 && $STRICT -eq 1 ]]; then | +
| + | echoerr "$cmdname: strict mode, refusing to execute subprocess" | +
| + | exit $RESULT | +
| + | fi | +
| + | exec $CLI | +
| + | else | +
| + | exit $RESULT | +
| + | fi | +