#!/usr/bin/env bash
trap _exit INT QUIT TERM
_red() { printf '\033[0;31;31m%b\033[0m' "$1" }
_green() { printf '\033[0;31;32m%b\033[0m' "$1" }
_yellow() { printf '\033[0;31;33m%b\033[0m' "$1" }
_blue() { printf '\033[0;31;36m%b\033[0m' "$1" }
_exists() { local cmd="$1" if eval type type > /dev/null 2>&1; then eval type "$cmd" > /dev/null 2>&1 elif command > /dev/null 2>&1; then command -v "$cmd" > /dev/null 2>&1 else which "$cmd" > /dev/null 2>&1 fi local rt=$? return ${rt} }
_exit() { _red "\nThe script has been terminated.\n" rm -fr speedtest.tgz speedtest-cli benchtest_* exit 1 }
get_opsy() { [ -f /etc/redhat-release ] && awk '{print $0}' /etc/redhat-release && return [ -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 }
next() { printf "%-70s\n" "-" | sed 's/\s/-/g' }
speed_test() { local nodeName="$2" [ -z "$1" ] && ./speedtest-cli/speedtest --progress=no --accept-license --accept-gdpr > ./speedtest-cli/speedtest.log 2>&1 || \ ./speedtest-cli/speedtest --progress=no --server-id=$1 --accept-license --accept-gdpr > ./speedtest-cli/speedtest.log 2>&1 if [ $? -eq 0 ]; then local dl_speed=$(awk '/Download/{print $3" "$4}' ./speedtest-cli/speedtest.log) local up_speed=$(awk '/Upload/{print $3" "$4}' ./speedtest-cli/speedtest.log) local latency=$(awk '/Latency/{print $2" "$3}' ./speedtest-cli/speedtest.log) if [[ -n "${dl_speed}" && -n "${up_speed}" && -n "${latency}" ]]; then printf "\033[0;33m%-18s\033[0;32m%-18s\033[0;31m%-20s\033[0;36m%-12s\033[0m\n" " ${nodeName}" "${up_speed}" "${dl_speed}" "${latency}" fi fi }
speed() { speed_test '' 'Speedtest.net' speed_test '21541' 'Los Angeles, US' speed_test '43860' 'Dallas, US' speed_test '40879' 'Montreal, CA' speed_test '24215' 'Paris, FR' speed_test '28922' 'Amsterdam, NL' speed_test '24447' 'Shanghai, CN' speed_test '26352' 'Nanjing, CN' speed_test '27594' 'Guangzhou, CN' speed_test '32155' 'Hongkong, CN' speed_test '6527' 'Seoul, KR' speed_test '7311' 'Singapore, SG' speed_test '21569' 'Tokyo, JP' }
io_test() { (LANG=C dd if=/dev/zero of=benchtest_$$ bs=512k count=$1 conv=fdatasync && rm -f benchtest_$$ ) 2>&1 | awk -F, '{io=$NF} END { print io}' | sed 's/^[ \t]*//;s/[ \t]*$//' }
calc_size() { local raw=$1 local total_size=0 local num=1 local unit="KB" if ! [[ ${raw} =~ ^[0-9]+$ ]] ; then echo "" return fi if [ "${raw}" -ge 1073741824 ]; then num=1073741824 unit="TB" elif [ "${raw}" -ge 1048576 ]; then num=1048576 unit="GB" elif [ "${raw}" -ge 1024 ]; then num=1024 unit="MB" elif [ "${raw}" -eq 0 ]; then echo "${total_size}" return fi total_size=$( awk 'BEGIN{printf "%.1f", '$raw' / '$num'}' ) echo "${total_size} ${unit}" }
check_virt(){ _exists "dmesg" && virtualx="$(dmesg 2>/dev/null)" if _exists "dmidecode"; then sys_manu="$(dmidecode -s system-manufacturer 2>/dev/null)" sys_product="$(dmidecode -s system-product-name 2>/dev/null)" sys_ver="$(dmidecode -s system-version 2>/dev/null)" else sys_manu="" sys_product="" sys_ver="" fi if grep -qa docker /proc/1/cgroup; then virt="Docker" elif grep -qa lxc /proc/1/cgroup; then virt="LXC" elif grep -qa container=lxc /proc/1/environ; then virt="LXC" elif [[ -f /proc/user_beancounters ]]; then virt="OpenVZ" elif [[ "${virtualx}" == *kvm-clock* ]]; then virt="KVM" elif [[ "${sys_product}" == *KVM* ]]; then virt="KVM" elif [[ "${cname}" == *KVM* ]]; then virt="KVM" elif [[ "${cname}" == *QEMU* ]]; then virt="KVM" elif [[ "${virtualx}" == *"VMware Virtual Platform"* ]]; then virt="VMware" elif [[ "${sys_product}" == *"VMware Virtual Platform"* ]]; then virt="VMware" elif [[ "${virtualx}" == *"Parallels Software International"* ]]; then virt="Parallels" elif [[ "${virtualx}" == *VirtualBox* ]]; then virt="VirtualBox" elif [[ -e /proc/xen ]]; then if grep -q "control_d" "/proc/xen/capabilities" 2>/dev/null; then virt="Xen-Dom0" else virt="Xen-DomU" fi elif [ -f "/sys/hypervisor/type" ] && grep -q "xen" "/sys/hypervisor/type"; then virt="Xen" elif [[ "${sys_manu}" == *"Microsoft Corporation"* ]]; then if [[ "${sys_product}" == *"Virtual Machine"* ]]; then if [[ "${sys_ver}" == *"7.0"* || "${sys_ver}" == *"Hyper-V" ]]; then virt="Hyper-V" else virt="Microsoft Virtual Machine" fi fi else virt="Dedicated" fi }
ipv4_info() { local org="$(wget -q -T10 -O- ipinfo.io/org)" local city="$(wget -q -T10 -O- ipinfo.io/city)" local country="$(wget -q -T10 -O- ipinfo.io/country)" local region="$(wget -q -T10 -O- ipinfo.io/region)" if [[ -n "$org" ]]; then echo " Organization : $(_blue "$org")" fi if [[ -n "$city" && -n "country" ]]; then echo " Location : $(_blue "$city / $country")" fi if [[ -n "$region" ]]; then echo " Region : $(_yellow "$region")" fi if [[ -z "$org" ]]; then echo " Region : $(_red "No ISP detected")" fi }
install_speedtest() { if [ ! -e "./speedtest-cli/speedtest" ]; then sys_bit="" local sysarch="$(uname -m)" if [ "${sysarch}" = "unknown" ] || [ "${sysarch}" = "" ]; then local sysarch="$(arch)" fi if [ "${sysarch}" = "x86_64" ]; then sys_bit="x86_64" fi if [ "${sysarch}" = "i386" ] || [ "${sysarch}" = "i686" ]; then sys_bit="i386" fi if [ "${sysarch}" = "armv8" ] || [ "${sysarch}" = "armv8l" ] || [ "${sysarch}" = "aarch64" ] || [ "${sysarch}" = "arm64" ]; then sys_bit="aarch64" fi if [ "${sysarch}" = "armv7" ] || [ "${sysarch}" = "armv7l" ]; then sys_bit="armhf" fi if [ "${sysarch}" = "armv6" ]; then sys_bit="armel" fi [ -z "${sys_bit}" ] && _red "Error: Unsupported system architecture (${sysarch}).\n" && exit 1 url1="https://install.speedtest.net/app/cli/ookla-speedtest-1.1.1-linux-${sys_bit}.tgz" url2="https://dl.lamp.sh/files/ookla-speedtest-1.1.1-linux-${sys_bit}.tgz" wget --no-check-certificate -q -T10 -O speedtest.tgz ${url1} if [ $? -ne 0 ]; then wget --no-check-certificate -q -T10 -O speedtest.tgz ${url2} [ $? -ne 0 ] && _red "Error: Failed to download speedtest-cli.\n" && exit 1 fi mkdir -p speedtest-cli && tar zxf speedtest.tgz -C ./speedtest-cli && chmod +x ./speedtest-cli/speedtest rm -f speedtest.tgz fi printf "%-18s%-18s%-20s%-12s\n" " Node Name" "Upload Speed" "Download Speed" "Latency" }
print_intro() { echo "-------------------- A Bench.sh Script By Teddysun -------------------" echo " Version : $(_green v2022-06-01)" echo " Usage : $(_red "wget -qO- bench.sh | bash")" }
get_system_info() { cname=$( awk -F: '/model name/ {name=$2} END {print name}' /proc/cpuinfo | sed 's/^[ \t]*//;s/[ \t]*$//' ) cores=$( awk -F: '/processor/ {core++} END {print core}' /proc/cpuinfo ) freq=$( awk -F'[ :]' '/cpu MHz/ {print $4;exit}' /proc/cpuinfo ) ccache=$( awk -F: '/cache size/ {cache=$2} END {print cache}' /proc/cpuinfo | sed 's/^[ \t]*//;s/[ \t]*$//' ) cpu_aes=$( grep -i 'aes' /proc/cpuinfo ) cpu_virt=$( grep -Ei 'vmx|svm' /proc/cpuinfo ) tram=$( LANG=C; free | awk '/Mem/ {print $2}' ) tram=$( calc_size $tram ) uram=$( LANG=C; free | awk '/Mem/ {print $3}' ) uram=$( calc_size $uram ) swap=$( LANG=C; free | awk '/Swap/ {print $2}' ) swap=$( calc_size $swap ) uswap=$( LANG=C; free | awk '/Swap/ {print $3}' ) uswap=$( calc_size $uswap ) up=$( awk '{a=$1/86400;b=($1%86400)/3600;c=($1%3600)/60} {printf("%d days, %d hour %d min\n",a,b,c)}' /proc/uptime ) if _exists "w"; then load=$( LANG=C; w | head -1 | awk -F'load average:' '{print $2}' | sed 's/^[ \t]*//;s/[ \t]*$//' ) elif _exists "uptime"; then load=$( LANG=C; uptime | head -1 | awk -F'load average:' '{print $2}' | sed 's/^[ \t]*//;s/[ \t]*$//' ) fi opsy=$( get_opsy ) arch=$( uname -m ) if _exists "getconf"; then lbit=$( getconf LONG_BIT ) else echo ${arch} | grep -q "64" && lbit="64" || lbit="32" fi kern=$( uname -r ) disk_total_size=$( LANG=C; df -t simfs -t ext2 -t ext3 -t ext4 -t btrfs -t xfs -t vfat -t ntfs -t swap --total 2>/dev/null | grep total | awk '{ print $2 }' ) disk_total_size=$( calc_size $disk_total_size ) disk_used_size=$( LANG=C; df -t simfs -t ext2 -t ext3 -t ext4 -t btrfs -t xfs -t vfat -t ntfs -t swap --total 2>/dev/null | grep total | awk '{ print $3 }' ) disk_used_size=$( calc_size $disk_used_size ) tcpctrl=$( sysctl net.ipv4.tcp_congestion_control | awk -F ' ' '{print $3}' ) }
print_system_info() { if [ -n "$cname" ]; then echo " CPU Model : $(_blue "$cname")" else echo " CPU Model : $(_blue "CPU model not detected")" fi if [ -n "$freq" ]; then echo " CPU Cores : $(_blue "$cores @ $freq MHz")" else echo " CPU Cores : $(_blue "$cores")" fi if [ -n "$ccache" ]; then echo " CPU Cache : $(_blue "$ccache")" fi if [ -n "$cpu_aes" ]; then echo " AES-NI : $(_green "Enabled")" else echo " AES-NI : $(_red "Disabled")" fi if [ -n "$cpu_virt" ]; then echo " VM-x/AMD-V : $(_green "Enabled")" else echo " VM-x/AMD-V : $(_red "Disabled")" fi echo " Total Disk : $(_yellow "$disk_total_size") $(_blue "($disk_used_size Used)")" echo " Total Mem : $(_yellow "$tram") $(_blue "($uram Used)")" if [ "$swap" != "0" ]; then echo " Total Swap : $(_blue "$swap ($uswap Used)")" fi echo " System uptime : $(_blue "$up")" echo " Load average : $(_blue "$load")" echo " OS : $(_blue "$opsy")" echo " Arch : $(_blue "$arch ($lbit Bit)")" echo " Kernel : $(_blue "$kern")" echo " TCP CC : $(_yellow "$tcpctrl")" echo " Virtualization : $(_blue "$virt")" }
print_io_test() { freespace=$( df -m . | awk 'NR==2 {print $4}' ) if [ -z "${freespace}" ]; then freespace=$( df -m . | awk 'NR==3 {print $3}' ) fi if [ ${freespace} -gt 1024 ]; then writemb=2048 io1=$( io_test ${writemb} ) echo " I/O Speed(1st run) : $(_yellow "$io1")" io2=$( io_test ${writemb} ) echo " I/O Speed(2nd run) : $(_yellow "$io2")" io3=$( io_test ${writemb} ) echo " I/O Speed(3rd run) : $(_yellow "$io3")" 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{printf "%.1f", '$ioall' / 3}' ) echo " I/O Speed(average) : $(_yellow "$ioavg MB/s")" else echo " $(_red "Not enough space for I/O Speed test!")" fi }
print_end_time() { end_time=$(date +%s) time=$(( ${end_time} - ${start_time} )) if [ ${time} -gt 60 ]; then min=$(expr $time / 60) sec=$(expr $time % 60) echo " Finished in : ${min} min ${sec} sec" else echo " Finished in : ${time} sec" fi date_time=$(date '+%Y-%m-%d %H:%M:%S %Z') echo " Timestamp : $date_time" }
! _exists "wget" && _red "Error: wget command not found.\n" && exit 1 ! _exists "free" && _red "Error: free command not found.\n" && exit 1 start_time=$(date +%s) get_system_info check_virt clear print_intro next print_system_info ipv4_info next print_io_test next install_speedtest && speed && rm -fr speedtest-cli next print_end_time next
|