#!/bin/bash
# For use with intel/asynch_mode_nginx, which is licensed under the
# BSD 3-Clause License.
# See https://github.com/intel/asynch_mode_nginx/blob/master/LICENSE

######################################
############# USER INPUT #############
######################################
ip_address="$1"
_time=10
clients=200
port=443
cipher=AES128-SHA
######################################
############# USER INPUT #############
######################################

helpAndError () {
        echo "This script is used to run the ConnectionsPerSecond(CPS) testing HTTPS."
        echo "To use this script: ./connection_test.sh <ip_address>"
        echo "To do a dry-run, use the emulation flag:"
        echo "./connection_test.sh <ip_address> --emulation"
        exit 0
}

# Check for h flag or no command line args
if [[ -z $ip_address || $@ == *-h* || $@ == *--help* ]]; then
        helpAndError
fi

# Check for emulation flag
emulation=0
if [[ $@ == **emulation** ]]
then
        emulation=1

fi

# cmd1 is the first part of the commandline and cmd2 is the second part
# The total commandline will be cmd1 + $ip_address:$port + cmd2
cmd1="openssl s_time -connect"
cmd2="-new -cipher $cipher -time $_time"

# Print out variables to check
printf " IP Addresses:                  $ip_address\n"
printf " Time:                          $_time\n"
printf " Clients:                       $clients\n"
printf " Port:                          $port\n"
printf " Cipher:                        $cipher\n"

# Remove previous .test files
rm -rf ./.test_*

# Get starttime
starttime=$(date +%s)

# Kick off the tests after checking for emulation
if [[ $emulation -eq 1 ]]
then
        for (( i = 0; i < ${clients}; i++ )); do
                printf "$cmd1 $ip_address:$(($port)) $cmd2 > .test_$(($port))_$i &\n"
        done
        exit 0
else
        for (( i = 0; i < ${clients}; i++ )); do
        $cmd1 $ip_address:$(($port)) $cmd2 > .test_$(($port))_$i &
        done
fi

waitstarttime=$(date +%s)

# wait until all processes complete
# The clients are child processes of this script, so wait returns once every
# one of them has exited.
wait

sumTotal=$(cat ./.test_$(($port))* | awk '(/^[0-9]* connections in [0-9]* real/){ total += $1/$4 } END {print total}')
printf "Connections per second:      $sumTotal CPS\n"
printf "Finished in %d seconds (%d seconds waiting for procs to start)\n" $(($(date +%s) - $starttime)) $(($waitstarttime - $starttime))
rm -rf ./.test_*