#!/bin/bash

#==============================================================#

#   Description: bench test script                             #

#   Author: Teddysun <i@teddysun.com>                          #

#   modify: wwng <wwng@2333.me>                                #

#   Thanks: LookBack <admin@dwhd.org>                          #

#   Visit:  https://teddysun.com                               #

#==============================================================#

get_opsy() {

    [ -f /etc/os-release ] && awk -F'[= "]' '/PRETTY_NAME/{print $3,$4,$5}' /etc/os-release && return

    [ -f /etc/lsb-release ] && awk -F'[="]+' '/DESCRIPTION/{print $2}' /etc/lsb-release && return

    [ -f /etc/redhat-release ] && awk '{print ($1,$3~/^[0-9]/?$3:$4)}' /etc/redhat-release && return

}

next() {

    printf "%-70s\n" "-" | sed 's/\s/-/g'

}

speed_test() {

    speedtest=$(wget -4O /dev/null -T300 $1 2>&1 | awk '/\/dev\/null/ {speed=$3 $4} END {gsub(/\(|\)/,"",speed); print speed}')

    ipaddress=$(ping -c1 -n `awk -F'/' '{print $3}' <<< $1` | awk -F'[()]' '{print $2;exit}')

    nodeName=$2

    if   [ "${#nodeName}" -lt "8" ]; then

        echo -e "\e[33m$2\e[0m\t\t\t\t\e[32m$ipaddress\e[0m\t\t\e[31m$speedtest\e[0m"

    elif [ "${#nodeName}" -lt "13" ]; then

        echo -e "\e[33m$2\e[0m\t\t\t\e[32m$ipaddress\e[0m\t\t\e[31m$speedtest\e[0m"

    elif [ "${#nodeName}" -lt "24" ]; then

        echo -e "\e[33m$2\e[0m\t\t\e[32m$ipaddress\e[0m\t\t\e[31m$speedtest\e[0m"

    elif [ "${#nodeName}" -ge "24" ]; then

        echo -e "\e[33m$2\e[0m\t\e[32m$ipaddress\e[0m\t\t\e[31m$speedtest\e[0m"

    fi

}

speed() {

	speed_test 'http://cachefly.cachefly.net/100mb.test' 'CacheFly'

	speed_test 'https://storage.googleapis.com/google-code-archive-downloads/v2/code.google.com/jquery-speedtest/100MB.txt' 'Google GCE'

	speed_test 'http://speed.0738.cc/ldinfo_vod.rar' 'Loudi Telecom, Hunan, CN'

	speed_test 'http://101.95.50.18/test.img' 'Shanghai Telecom, Shanghai, CN'

	speed_test 'http://61.157.118.78/download/300.data' 'Suining Telecom, Sichuan, CN'

	#speed_test 'http://speed.myzone.cn/pc_elive_1.1.rar' 'China Telecom, Henan, CN'

	speed_test 'http://speedtest.newark.linode.com/100MB-newark.bin' 'Linode, Newark, NJ'

	speed_test 'http://speedtest.atlanta.linode.com/100MB-atlanta.bin' 'Linode, Atlanta, GA'

	speed_test 'http://speedtest.dallas.linode.com/100MB-dallas.bin' 'Linode, Dallas, TX'

	speed_test 'http://speedtest.fremont.linode.com/100MB-fremont.bin' 'Linode, Fremont, CA'

	speed_test 'http://speedtest.frankfurt.linode.com/100MB-frankfurt.bin' 'Linode, Frankfurt, DE'

	speed_test 'http://speedtest.london.linode.com/100MB-london.bin' 'Linode, London, UK'

	speed_test 'http://speedtest.singapore.linode.com/100MB-singapore.bin' 'Linode, Singapore, SG'

	speed_test 'http://speedtest.tokyo.linode.com/100MB-tokyo.bin' 'Linode, Tokyo, JP'

	speed_test 'http://speedtest.AMS01.softlayer.com/downloads/test100.zip' 'Softlayer, Amsterdam, NL'

	speed_test 'http://speedtest.CHE01.softlayer.com/downloads/test100.zip' 'Softlayer, Chennai, IN'

	#speed_test 'http://speedtest.DAL01.softlayer.com/downloads/test100.zip' 'Softlayer, Dallas, TX'

	#speed_test 'http://speedtest.FRA02.softlayer.com/downloads/test100.zip' 'Softlayer, Frankfurt, DE'

	speed_test 'http://speedtest.HKG02.softlayer.com/downloads/test100.zip' 'Softlayer, Hong Kong, CN'

	speed_test 'http://speedtest.HOU02.softlayer.com/downloads/test100.zip' 'Softlayer, Houston, TX'

	#speed_test 'http://speedtest.LON02.softlayer.com/downloads/test100.zip' 'Softlayer, London, UK'

	speed_test 'http://speedtest.MEL01.softlayer.com/downloads/test100.zip' 'Softlayer, Melbourne, AU'

	speed_test 'http://speedtest.MEX01.softlayer.com/downloads/test100.zip' 'Softlayer, Queretaro, MX'

	speed_test 'http://speedtest.MIL01.softlayer.com/downloads/test100.zip' 'Softlayer, Milan, IT'

	speed_test 'http://speedtest.MON01.softlayer.com/downloads/test100.zip' 'Softlayer, Montreal, CA'

	speed_test 'http://speedtest.PAR01.softlayer.com/downloads/test100.zip' 'Softlayer, Paris, FR'

	speed_test 'http://speedtest.SJC01.softlayer.com/downloads/test100.zip' 'Softlayer, San Jose, CA'

	speed_test 'http://speedtest.SAO01.softlayer.com/downloads/test100.zip' 'Softlayer, Sao Paulo, BR'

	speed_test 'http://speedtest.SEA01.softlayer.com/downloads/test100.zip' 'Softlayer, Seattle, WA'

	#speed_test 'http://speedtest.SNG01.softlayer.com/downloads/test100.zip' 'Softlayer, Singapore, SG'

	speed_test 'http://speedtest.SYD01.softlayer.com/downloads/test100.zip' 'Softlayer, Sydney, AU'

	#speed_test 'http://speedtest.TOK02.softlayer.com/downloads/test100.zip' 'Softlayer, Tokyo, JP'

	speed_test 'http://speedtest.TOR01.softlayer.com/downloads/test100.zip' 'Softlayer, Toronto, CA'

	speed_test 'http://speedtest.WDC01.softlayer.com/downloads/test100.zip' 'Softlayer, Washington, WA'

	speed_test 'http://speedtest-nyc1.digitalocean.com/100mb.test' 'Digitalocean, NewYork, NY'

	speed_test 'http://speedtest-ams2.digitalocean.com/100mb.test' 'Digitalocean, Amsterdam, NL'

	speed_test 'http://speedtest-sfo1.digitalocean.com/100mb.test' 'Digitalocean, San Francisco, CA'

	#speed_test 'http://speedtest-sgp1.digitalocean.com/100mb.test' 'Digitalocean, Singapore, SG'

	#speed_test 'http://speedtest-LON1.digitalocean.com/100mb.test' 'Digitalocean, London, UK'

	#speed_test 'http://speedtest-FRA1.digitalocean.com/100mb.test' 'Digitalocean, Frankfurt, DE'

	#speed_test 'http://speedtest-TOR1.digitalocean.com/100mb.test' 'Digitalocean, Toronto, CA'

	speed_test 'http://speedtest-blr1.digitalocean.com/100mb.test' 'Digitalocean, Bengaluru, IN'

	speed_test 'http://proof.ovh.net/files/100Mio.dat' 'OVH, Roubaix, FR'

	#speed_test 'http://mirror.nl.leaseweb.net/speedtest/100mb.bin' 'LeaseWeb, Amsterdam, NL'

	speed_test 'http://mirror.us.leaseweb.net/speedtest/100mb.bin' 'LeaseWeb, Manassas, VA'

	speed_test 'http://199.231.208.6/100MB.test' 'Enzu, Chicago, IL'

	#speed_test 'http://192.80.186.135/100MB.test' 'Enzu, Dallas, TX'

	speed_test 'http://192.157.214.6/100MB.test' 'Enzu, Los Angeles, CA'

	#speed_test 'http://172.246.125.7/100MB.test' 'Enzu, Miami, FL'

}

io_test() {

    (LANG=en_US dd if=/dev/zero of=test_$$ bs=64k count=16k conv=fdatasync && rm -f test_$$ ) 2>&1 | awk -F, '{io=$NF} END { print io}' | sed 's/^[ \t]*//;s/[ \t]*$//'

}

cname=$( awk -F: '/model name/ {name=$2} END {print name}' /proc/cpuinfo | sed 's/^[ \t]*//;s/[ \t]*$//' )

cores=$( awk -F: '/model name/ {core++} END {print core}' /proc/cpuinfo )

freq=$( awk -F: '/cpu MHz/ {freq=$2} END {print freq}' /proc/cpuinfo | sed 's/^[ \t]*//;s/[ \t]*$//' )

tram=$( free -m | awk '/Mem/ {print $2}' )

swap=$( free -m | awk '/Swap/ {print $2}' )

up=$( awk '{a=$1/86400;b=($1%86400)/3600;c=($1%3600)/60;d=$1%60} {printf("%ddays, %d:%d:%d\n",a,b,c,d)}' /proc/uptime )

opsy=$( get_opsy )

arch=$( uname -m )

lbit=$( getconf LONG_BIT )

host=$( hostname )

kern=$( uname -r )

clear

next

echo "CPU model            : $cname"

echo "Number of cores      : $cores"

echo "CPU frequency        : $freq MHz"

echo "Total amount of ram  : $tram MB"

echo "Total amount of swap : $swap MB"

echo "System uptime        : $up"

echo "OS                   : $opsy"

echo "Arch                 : $arch ($lbit Bit)"

echo "Kernel               : $kern"

next

if  [ -e '/usr/bin/wget' ]; then

    echo -e "Node Name\t\t\tIPv4 address\t\tDownload Speed"

    speed

else

    echo "Error: wget command not found. You must be install wget command at first."

    exit 1

fi

io1=$( io_test )

io2=$( io_test )

io3=$( io_test )

ioraw1=$( echo $io1 | awk 'NR==1 {print $1}' )

[ "`echo $io1 | awk 'NR==1 {print $2}'`" == "GB/s" ] && ioraw1=$( awk 'BEGIN{print '$ioraw1' * 1024}' )

ioraw2=$( echo $io2 | awk 'NR==1 {print $1}' )

[ "`echo $io2 | awk 'NR==1 {print $2}'`" == "GB/s" ] && ioraw2=$( awk 'BEGIN{print '$ioraw2' * 1024}' )

ioraw3=$( echo $io3 | awk 'NR==1 {print $1}' )

[ "`echo $io3 | awk 'NR==1 {print $2}'`" == "GB/s" ] && ioraw3=$( awk 'BEGIN{print '$ioraw3' * 1024}' )

ioall=$( awk 'BEGIN{print '$ioraw1' + '$ioraw2' + '$ioraw3'}' )

ioavg=$( awk 'BEGIN{print '$ioall'/3}' )

next

echo "I/O speed(1st run) : $io1"

echo "I/O speed(2nd run) : $io2"

echo "I/O speed(3rd run) : $io3"

echo "Average I/O: $ioavg MB/s"

echo ""