ping-tcp

#!/usr/bin/env bash

#  +--------------------------------------------------------------------------------------+
#  | Title        : ping-tcp                                                              |
#  |                                                                                      |
#  | Description  : Check if host is reachable for a specific TCP Port                    |
#  |                                                                                      |
#  | Author       : Sven Wick <sven.wick@gmx.de>                                          |
#  |                                                                                      |
#  | Based On     : https://github.com/vaporup/ssh-tools/blob/master/ssh-ping             |
#  |                https://stackoverflow.com/a/35338529                                  |
#  |                                                                                      |
#  +--------------------------------------------------------------------------------------+

# trap CTRL-C and call print_statistics()
trap print_statistics SIGINT

#
# Some colors for better output
#

     RED='\033[0;91m'
   GREEN='\033[0;92m'
  YELLOW='\033[0;93m'
    BLUE='\033[0;94m'
 MAGENTA='\033[0;95m'
    CYAN='\033[0;96m'
   WHITE='\033[0;97m'
    BOLD='\033[1m'
   RESET='\033[0m'

#
# Defaults
#

ping_count=0           # How many requests to do
ping_interval=1        # Seconds to wait between sending each request
connect_timeout=16     # Seconds to wait for a response
seq=1                  # Request Counter
requests_transmitted=0 # Count how often we sent a request
requests_received=0    # Count how often we got an answer
requests_lost=0        # Count how often we lost an answer
quiet="no"             # Do not suppress output

nomen_est_omen=$(basename "$0")

    case ${nomen_est_omen} in
        ping-ftp )
            port=21
            ;;
        ping-ssh )
            port=22
            echo -e "\n    ${YELLOW}Checkout ssh-ping from ssh-tools (much better)${RESET}"
            ;;
        ping-smtp )
            port=25
            ;;
        ping-http )
            port=80
            ;;
        ping-pop3 )
            port=110
            ;;
        ping-imap2 )
            port=143
            ;;
        ping-ldap )
            port=389
            ;;
        ping-https )
            port=443
            ;;
        ping-cifs|ping-smb )
            port=445
            ;;
        ping-ldaps )
            port=636
            ;;
        ping-rsync )
            port=873
            ;;
        ping-imaps )
            port=993
            ;;
        ping-pop3s )
            port=995
            ;;
        ping-mssql )
            port=1433
            ;;
        ping-mysql )
            port=3306
            ;;
        ping-rdp )
            port=3389
            ;;
    esac

#
# Usage/Help message
#

function usage() {

cat << EOF

    Usage: ${0##*/} [OPTIONS] host

    OPTIONS:

        -c count       Stop after sending <count> request packets
        -h             Show this message
        -i interval    Wait <interval> seconds between sending each request.
                       The default is 1 second.
        -D             Print timestamp (unix time + microseconds as in gettimeofday) before each line
        -H             Print timestamp (human readable) before each line
        -W timeout     Time to wait for a response, in seconds
        -p port        Port to connect to on the remote host.
        -q             Quiet output.
                       Nothing is displayed except the summary lines at startup time and when finished
        -n             No colors.
                       (e.g. for black on white terminals)

    ENVIRONMENT_VARIABLES:

        TCP_PING_NO_COLORS    if set, no colors are shown (like -n)

        Example:      TCP_PING_NO_COLORS=true ${0##*/} -c 1 hostname

    EXIT_CODES:

        0             No requests lost
        1             More than 1 request lost
        2             All requests lost

        Example:      ${0##*/} -q -c 1 my_website >/dev/null || ...



EOF

}

if [[ -z $1 || $1 == "--help" ]]; then
    usage
    exit 1
fi

function command_exists() {

    command -v "${1}" >/dev/null 2>&1

}

function get_timestamp() {

    if [[ "${print_timestamp_human_readable}" == "yes" ]]; then
      date
    else
      if [[ "${OSTYPE}" == "linux-gnu" ]]; then
        date +%s.%6N
      else
        command_exists perl && perl -MTime::HiRes=time -e 'printf "%.6f", time'
      fi
    fi

}

function get_request_timestamp() {

    if [[ "${OSTYPE}" == "linux-gnu" ]]; then
        date +%s%3N
    else
        command_exists perl && perl -MTime::HiRes=time -e 'printf "%i", time * 1000'
    fi

}

function print_statistics() {

    [[ ${requests_transmitted} -eq 0 ]] && exit

    requests_loss=$(( 100 * requests_lost / requests_transmitted ))

    echo ""
    echo -e "${WHITE}---${RESET} ${YELLOW}${host}${RESET} ${WHITE}ping statistics${RESET} ${WHITE}---${RESET}"

    statistics_ok="${GREEN}${requests_transmitted}${RESET} ${WHITE}requests transmitted${RESET}, "
    statistics_ok+="${GREEN}${requests_received}${RESET} ${WHITE}requests received${RESET}, "
    statistics_ok+="${GREEN}${requests_loss}%${RESET} ${WHITE}request loss${RESET}"

    statistics_warn="${YELLOW}${requests_transmitted}${RESET} ${WHITE}requests transmitted${RESET}, "
    statistics_warn+="${YELLOW}${requests_received}${RESET} ${WHITE}requests received${RESET}, "
    statistics_warn+="${YELLOW}${requests_loss}%${RESET} ${WHITE}request loss${RESET}"

    statistics_crit="${RED}${requests_transmitted}${RESET} ${WHITE}requests transmitted${RESET}, "
    statistics_crit+="${RED}${requests_received}${RESET} ${WHITE}requests received${RESET}, "
    statistics_crit+="${RED}${requests_loss}%${RESET} ${WHITE}request loss${RESET}"

    [[ ${requests_loss} -eq 100 ]] &&  echo -e "${statistics_crit}" && exit 2
    [[ ${requests_loss} -gt   1 ]] &&  echo -e "${statistics_warn}" && exit 1
    [[ ${requests_loss} -eq   0 ]] &&  echo -e "${statistics_ok}"   && exit

}

#
# Command line Options
#

# shellcheck disable=SC2249
while getopts ":c:hi:DHp:W:qn" opt; do
    case ${opt} in
        c )
            [[ ${OPTARG} =~ ^[0-9]+$ ]] && ping_count=${OPTARG}
        ;;
        h )
            usage
            exit 1
        ;;
        i )
            ping_interval=${OPTARG}
        ;;
        D )
            print_timestamp="yes"
        ;;
        H )
            print_timestamp="yes"
            print_timestamp_human_readable="yes"
        ;;
        p )
            [[ ${OPTARG} =~ ^[0-9]+$ ]] && port=${OPTARG}
        ;;
        W )
            [[ ${OPTARG} =~ ^[0-9]+$ ]] && connect_timeout=${OPTARG}
        ;;
        q )
            quiet="yes"
        ;;
        n )
            colors="no"
        ;;
        \? )
            echo "Invalid option: ${OPTARG}" 1>&2
            usage
            exit 1
        ;;
    esac
done

shift $((OPTIND - 1))

host=${1}

#
# Colors are counter productive
# on black on white terminals
#

if [[ -n "${TCP_PING_NO_COLORS}" ]]; then

    colors="no"

fi

if [[ ${colors} == no ]]; then

    unset -v RED GREEN YELLOW BLUE MAGENTA CYAN WHITE BOLD

fi

[[ -z "${host}" ]] && { echo -e "\n  ${RED}Error: No target host given${RESET}" ; usage; exit 1; }
[[ -z "${port}" ]] && { echo -e "\n  ${RED}Error: No TCP port given given${RESET}" ; usage; exit 1; }

#
# Output header
#

echo -e "${BOLD}PING_TCP_${port}${RESET} ${YELLOW}${host}${RESET}"

if [[ ! "${OSTYPE}" == "linux-gnu" ]]; then
  command_exists perl || echo -e "${YELLOW}WARNING:${RESET} No perl found, time measure probably fails (${WHITE}time${RESET}=${RED}0${RESET} ms)" >&2
fi

while true; do

    #
    # ping only $count times or forever if $count = 0
    #

    [[ ${ping_count} -gt 0 ]] && [[ ${seq} -gt ${ping_count} ]] && break

    #
    # used for -D and or -H option
    #

    timestamp=$( get_timestamp )

    #
    # Doing the actual request and measure its execution time
    #

    start_request=$( get_request_timestamp )

    status=$( timeout --foreground $connect_timeout bash -c "</dev/tcp/${host}/${port}" > /dev/null 2>&1 && echo reply || echo denied)

    end_request=$( get_request_timestamp )
    time_request=$((end_request-start_request))

    if [[ ${status} == reply ]]; then
        requests_received=$(( requests_received + 1 ))
        [[ ${quiet} == no ]] && [[ ${print_timestamp} == yes ]] && echo -e -n "${WHITE}[${RESET}${MAGENTA}${timestamp}${RESET}${WHITE}]${RESET} "
        [[ ${quiet} == no ]] && echo -e "${GREEN}Reply${RESET} ${WHITE}from${RESET} ${YELLOW}${host}${RESET}${WHITE}:${RED}${port}${RESET} seq${RESET}=${RED}${seq}${RESET} ${WHITE}time${RESET}=${RED}${time_request}${RESET} ms"
    else
        requests_lost=$(( requests_lost + 1 ))
    fi

    requests_transmitted=${seq}
    seq=$(( seq + 1 ))

    #
    # Don't sleep if we do just 1 request
    #

    [[ ${ping_count} -eq 1 ]] || sleep "${ping_interval}"

done

print_statistics