#!/bin/bash
#==============================================================#
#   Description: bench test script                             #
#   Author: Teddysun <i@teddysun.com>                          #
#   Thanks: LookBack <admin@dwhd.org>                          #
#   Visit:  https://teddysun.com                               #
#==============================================================#

cname=$( awk -F: '/model name/ {name=$2} END {print name}' /proc/cpuinfo )
if  [ "$cname" =  "" ]; then
    cname=$( awk -F: '/Processor/ {name=$2} END {print name}' /proc/cpuinfo )
fi
cores=$( awk -F: '/model name/ {core++} END {print core}' /proc/cpuinfo )
freq=$( awk -F: '/cpu MHz/ {freq=$2} END {print freq}' /proc/cpuinfo )
tram=$( free -m | awk 'NR==2 {print $2}' )
swap=$( free -m | awk 'NR==4 {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 )

next() {
    printf "%-60s\n" "-" | sed 's/\s/-/g'
}

speed_test() {
    speedtest=$(wget -4O /dev/null $1 2>&1 | awk '/\/dev\/null/ {speed=$3 $4} END {gsub(/\(|\)/,"",speed); print speed}')
    ipaddress=$(ping -c1 `awk -F'/' '{print $3}' <<< $1` | awk -F'[()]' '{print $2;exit}')
    echo -e "\e[34m[$3]\e[0m\e[33m$2\e[0m$4\e[32m$ipaddress\e[0m\t\t\e[31m$speedtest\e[0m"
}

speed() {
    echo -e "Node Name\t\t\t\tIP Address\t\tSpeed"
    speed_test 'http://cachefly.cachefly.net/100mb.test' '   CacheFly' 'Global' '\t\t\t'
    echo -e "\e[37m[America]\e[0m"
    speed_test 'http://speedtest.fremont.linode.com/100MB-fremont.bin' '  Linode, Fremont' 'US West' '\t\t'
    speed_test 'http://lax-ca-us-ping.vultr.com/vultr.com.100MB.bin' '  Vultr, Los Angeles' 'US West' '\t\t'
    speed_test 'http://speedtest.newark.linode.com/100MB-newark.bin' '  Linode, Newark' 'US East' '\t\t'
    speed_test 'http://speedtest.wdc01.softlayer.com/downloads/test100.zip' '  Softlayer, Washington' 'US East' '\t'
    speed_test 'http://68.169.39.189/speedtest.256mb' '   UK2, Montreal' 'Canada' '\t'
    speed_test 'http://proof.ovh.ca/files/100Mb.dat' '   OVH, Montreal' 'Canada' '\t'
    echo -e "\e[37m[Europe]\e[0m"
    speed_test 'http://proof.ovh.net/files/100Mb.dat' '   OVH, Roubaix' 'France' '\t\t'
    speed_test 'http://ping.online.net/100Mo.dat' '   Online.net, Paris' 'France' '\t'
    speed_test 'http://speedtest.frankfurt.linode.com/100MB-frankfurt.bin' '  Linode, Frankfurt' 'Germany' '\t'
    speed_test 'http://speedtest.fra02.softlayer.com/downloads/test100.zip' '  Softlayer, Frankfurt' 'Germany' '\t'
    echo -e "\e[37m[Asia]\e[0m"
    speed_test 'http://test.cc163.net/down/test2.rar' '    China Unicom, Jilin' 'China' '\t'
    speed_test 'http://58.32.224.18/jre-7u67-windows-i586.exe' '    Chinanet, Shanghai' 'China' '\t'
    speed_test 'http://speedtest.hkg02.softlayer.com/downloads/test100.zip' ' Softlayer, HongKong' 'HongKong' '\t'
    speed_test 'http://43.249.37.9/hMINOJJhxH/100mb.bin' ' LeaseWeb, HongKong' 'HongKong' '\t'
    speed_test 'http://speedtest.tokyo.linode.com/100MB-tokyo.bin' '    Linode, Tokyo' 'Japan' '\t\t'
    speed_test 'http://hnd-jp-ping.vultr.com/vultr.com.100MB.bin' '    Vultr, Tokyo' 'Japan' '\t\t'
    speed_test 'http://speedtest.sng01.softlayer.com/downloads/test100.zip' 'Softlayer, Singapore' 'Singapore' '\t'
    speed_test 'http://speedtest.oneasiahost.com/100mb.bin' 'OneAsiaHost, Singapore' 'Singapore' '\t'

}

clear
if [ "$cname" =  "" ] ; then cname='UnKnown'; fi
if [ "$cores" =  "" ] ; then cores='UnKnown'; fi
if [ "$freq" =  "" ] ; then freq='UnKnown'; fi
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"
next

if  [ -e '/usr/bin/wget' ]; then
    speed && next
else
    echo "Error: wget command not found. You must be install wget command at first."
    exit 1
fi


io=$((dd if=/dev/zero of=test_$$ bs=64k count=16k conv=fdatasync && /bin/rm -f test_$$) 2>&1 | awk 'END{print}')
echo "I/O speed : $io"
echo ""