#!/bin/bash

if [ `id -u` -ne 0 ] ; then echo "Please run as root" ; exit 1 ; fi

tput setaf 5 ;
echo "============================================="
echo "Cloud Assessment Data Capture - Pugmarks Inc."
echo "============================================="
tput sgr0;

command -v curl >/dev/null 2>&1 || {
	tput setaf 1 ;
	echo  "This script requires curl. Please install curl and run this script again. Aborting.";
	tput sgr0;
	exit 1;
}

if [ $# -ne 4 ]; then
	echo ; echo "Usage: \n$0 <email> <secret> <sample-time> <test-period>\n"
	echo "Where:"
	echo -e "email \t\t  User email"
	echo -e "secret \t\t  Server secret"
	echo -e "sample-time \t  Sampling time interval in minutes"
	echo -e "test-period \t  Duration to test in hours"; echo
	exit 1
fi

email=$1
secret=$2
sample_t=$3
period=$4

no_runs=`expr $period \* 60 / $sample_t`
sleep_t=`expr $sample_t \* 60`
tmpfile=`mktemp`
samplefile=`mktemp`
#echo $tmpfile

#if [ ! -w "/dev/mem" ]; then echo "Could not write to temp file" ; exit 1 ; fi

echo "Preparing environment...."
ARCH=$(uname -m | sed 's/x86_//;s/i[3-6]86/32/')
if [ -f /usr/bin/lsb_release ]; then
	OS=$(lsb_release -d|sed 's/Description:\t//')
	apt-get -y install dmidecode >/dev/null 2>&1
elif [ -f /etc/redhat-release ]; then
	OS=$(cat /etc/redhat-release)
	yum -y install dmidecode >/dev/null 2>&1
else
	OS="$(uname -s) $(uname -r)"
fi
echo "Done."

sys_uuid=`dmidecode -s system-uuid`
out_file=$sys_uuid-`date +%F@%R:%S`

if [ -z "$sys_uuid" ]; then
	tput setaf 1 ;
	echo "Could not read the UUID for this server. Data was not submitted."
	tput sgr0;
	exit 1
fi

echo "{\"sysinfo\":" >> $tmpfile
echo "Capturing one-time system info data...."
echo "{" >> $tmpfile
echo "\"info\":\"Cloud Assessment Data Capture - Pugmarks Inc.\"," >> $tmpfile

echo "\"time\":\"`date`\"," >> $tmpfile
echo "\"hostname\":\"`hostname`\"," >> $tmpfile
echo "\"manufacturer\":\"`dmidecode -s system-manufacturer`\"," >> $tmpfile
echo "\"product\":\"`dmidecode -s system-product-name`\"," >> $tmpfile
echo "\"serial\":\"`dmidecode -s system-serial-number`\"," >> $tmpfile
echo "\"uuid\":\"$sys_uuid\"," >> $tmpfile
echo "\"user\":\"`whoami`\"," >> $tmpfile
echo "\"os\":\"$OS\"," >> $tmpfile
echo "\"os_arch\":\"$ARCH\"," >> $tmpfile
echo "\"processor\":\"`cat /proc/cpuinfo|grep name|head -1|cut -f2 -d:`\"," >> $tmpfile
echo "\"processor_count\":\"`dmidecode -s processor-version|wc -l`\"," >> $tmpfile
echo "\"core-thread_count\":\"`cat /proc/cpuinfo|grep processor|wc -l`\"," >> $tmpfile
echo "\"total_memory\":\"`cat /proc/meminfo|grep MemTotal|awk '{print $2 $3}'`\"," >> $tmpfile
echo "\"free_memory\":\"`cat /proc/meminfo|grep MemFree|awk '{print $2 $3}'`\"," >> $tmpfile
echo "\"total_swap\":\"`cat /proc/meminfo|grep SwapTotal|awk '{print $2 $3}'`\"," >> $tmpfile
echo "\"free_swap\":\"`cat /proc/meminfo|grep SwapFree|awk '{print $2 $3}'`\"," >> $tmpfile
echo "\"disk\":\"`lsblk -d -o NAME,TYPE|grep disk|wc -l`\"," >> $tmpfile
echo "\"disk_size\":\"`df -h --total --block-size=K|grep total|awk '{print $2}'`\"," >> $tmpfile
echo "\"disk_used\":\"`df -h --total --block-size=K|grep total|awk '{print $3}'`\"," >> $tmpfile

echo "\"processors\":[" >> $tmpfile
count=0
cat /proc/cpuinfo|grep name |{
	while read t; do
	count=$((count+1))
	done 

	for i in $(seq $count)
	do
	p=$(printf "%sp" "$i")
	echo "{" >> $tmpfile ;
	cat /proc/cpuinfo|grep name | sed -n $p |
	while read a b c cpu; do
		echo "\"caption\":\"$cpu\"," >> $tmpfile
	done
	cat /proc/cpuinfo|grep family | sed -n $p |
	while read a b c family; do
		echo "\"family\":\"$family\"," >> $tmpfile
	done
	cat /proc/cpuinfo|grep cores | sed -n $p |
	while read a b c cores; do
		echo "\"cores\":\"$cores\"," >> $tmpfile
	done
	lscpu|grep Architecture|
	while read a arch; do
		echo "\"version\":\"$arch\"" >> $tmpfile
	done
	if [ $i == $count ]; then
		echo "}" >> $tmpfile
	else
		echo "}," >> $tmpfile
	fi
	done
}
echo "]," >> $tmpfile

echo "\"disks\":[" >> $tmpfile
count=`df -h --block-size=MB -T|grep -v blocks |grep -v tmpfs|wc -l`
i=0
df -h --block-size=MB -T|grep -v blocks |grep -v tmpfs |
while read name ptype total used free usage mount; do
	echo "{" >> $tmpfile ;
	echo "\"name\":\"$name\"," >> $tmpfile
	echo "\"freesize\":\"$free\"," >> $tmpfile
	echo "\"totalsize\":\"$total\"," >> $tmpfile
	echo "\"usedsize\":\"$used\"" >> $tmpfile
	i=`expr $i + 1`
	if [ $i == $count ]; then
		echo "}" >> $tmpfile
	else
		echo "}," >> $tmpfile
	fi
done
echo "]," >> $tmpfile

echo "\"ips\":[" >> $tmpfile
count=`ip a|grep 'inet '|grep -v 127.0 | wc -l`
i=0
ip a|grep 'inet '|grep -v 127.0 |
while read ntype ip scope link; do
	i=`expr $i + 1`
	if [ $i == $count ]; then
		echo "\"$ip\"" >> $tmpfile
	else
		echo "\"$ip\"," >> $tmpfile
	fi
done
echo "]," >> $tmpfile

echo "\"processes\":[" >> $tmpfile
count=`ps aux | sort -nr -k 3 |grep -v TTY | head -10 | wc -l`
i=0
ps aux | sort -nr -k 3 |grep -v TTY | head -10  |
while read user pid cpu mem vsz rss tty stat start t cmnd; do
	echo "{" >> $tmpfile ;
	echo "\"user\":\"$user\"," >> $tmpfile
	echo "\"pid\":\"$pid\"," >> $tmpfile
	echo "\"cpu\":\"$cpu\"," >> $tmpfile
	echo "\"mem\":\"$mem\"," >> $tmpfile
	echo "\"vsz\":\"$vsz\"," >> $tmpfile
	echo "\"rss\":\"$rss\"," >> $tmpfile
	echo "\"tty\":\"$tty\"," >> $tmpfile
	echo "\"stat\":\"$stat\"," >> $tmpfile
	echo "\"start\":\"$start\"," >> $tmpfile
	echo "\"t\":\"$t\"," >> $tmpfile
	cmnd=`echo $cmnd|tr '"' ':'|tr ',' ':'`
	echo "\"name\":\"$cmnd\"" >> $tmpfile
	i=`expr $i + 1`
	if [ $i == $count ]; then
		echo "}" >> $tmpfile
	else
		echo "}," >> $tmpfile
	fi
done
echo "]," >> $tmpfile

echo "\"services\":[" >> $tmpfile
if grep '^NAME="Ubuntu"$' /etc/os-release > /dev/null 2>&1; then
	service --status-all 2>&1 | grep '\[ + ]' | awk '{print $4}' |
	while read name options; do
	echo "{" >> $tmpfile ;
	echo "\"name\":\"$name\"," >> $tmpfile
		echo "\"status\":\"running\"" >> $tmpfile
	echo "}," >> $tmpfile
	done
else
	chkconfig --list | grep '3:on' |
	while read name options; do
	echo "{" >> $tmpfile ;
	echo "\"name\":\"$name\"," >> $tmpfile
		echo "\"status\":\"running\"" >> $tmpfile
	echo "}," >> $tmpfile
	done
fi

echo "]," >> $tmpfile
echo "\"ports\":[" >> $tmpfile ;
netstat -plan | grep LISTEN | grep -v ^unix |
while read protocol mss win loc fore status bnd; do
	echo "{" >> $tmpfile ;
	echo "\"protocol\":\"$protocol\"," >> $tmpfile
	echo "\"local_address\":\"$loc\"," >> $tmpfile
	echo "\"foreign_address\":\"$fore\"," >> $tmpfile
	echo "\"status\":\"$status\"" >> $tmpfile
	echo "}," >> $tmpfile
done
echo "]" >> $tmpfile

echo "}" >> $tmpfile

echo "Done."


tput setaf 6 ;
echo "Script will continue to run for the next $period hours taking samples each $sample_t minutes"
tput sgr0;

echo ",\"performance\":" >> $tmpfile

echo "[" >> $tmpfile
for i in `seq 1 $no_runs`; do

	tmpsample=`mktemp`
	if grep '^NAME="Ubuntu"$' /etc/os-release > /dev/null 2>&1; then
	top  -n 1 -b |head -4 > $tmpsample; df --total -m|tail -1 >> $tmpsample
	idle=`grep ^%Cpu $tmpsample|awk '{print $8}'`
	else
	top  -n 1 -b |head -4 > $tmpsample; df --total -m|tail -1 >> $tmpsample
	idle=`grep ^Cpu $tmpsample|cut -f4 -d%|awk '{print $2}'`
	fi
	if [ `echo $idle|cut -f2 -d.` -ge 5 ]; then export cpu=`echo $idle|cut -f1 -d.`; cpu=`expr 100 - $cpu -1`; else export cpu=`echo $idle|cut -f1 -d.`; cpu=`expr 100 - $cpu`; fi
	load_avg=`grep ^top $tmpsample|cut -f5 -d,|awk '{print $1}'`
	io_wait=`grep ^Cpu $tmpsample|cut -f5 -d%|awk '{print $2}'`
	memused=`grep ^Mem $tmpsample|awk '{print $4}'`
	membuffered=`grep ^Mem $tmpsample|awk '{print $8}'`
	memtotal=`grep ^Mem $tmpsample|awk '{print $2}'`
	diskused=`grep ^total $tmpsample|awk '{print $3}'`
	disktotal=`grep ^total $tmpsample|awk '{print $2}'`

	echo "{" >> $tmpfile
	echo "\"time\":\"`date`\"," >> $tmpfile
	echo "\"cpu\":\"$cpu\"," >> $tmpfile
	echo "\"load_avg\":\"$load_avg\"," >> $tmpfile
	echo "\"io_wait\":\"$io_wait\"," >> $tmpfile
	echo "\"memused\":\"$memused\"," >> $tmpfile
	echo "\"membuffered\":\"$membuffered\"," >> $tmpfile
	echo "\"memtotal\":\"$memtotal\"," >> $tmpfile
	echo "\"diskused\":\"$diskused\"," >> $tmpfile
	echo "\"disktotal\":\"$disktotal\"," >> $tmpfile
	echo "}," >> $tmpfile

	tput setaf 2 ;
	echo "Result:"
	echo "CPU \t\t$cpu"
	echo "Load Avarage \t$load_avg"
	echo "Mem Total \t$memtotal"
	echo "Mem Used \t$memused"
	echo "Mem Buffered \t$membuffered"
	echo "Disk Total \t$disktotal"
	echo "Disk Used \t$diskused"
	tput sgr0;

	echo "Sleeping $sleep_t seconds..."

	rm $tmpsample

	sleep $sleep_t

done

echo "]" >> $tmpfile
echo "}" >> $tmpfile
echo "Done."

tput setaf 6 ;
echo -e "\n\nSubmitting data to server..."

json=$(cat $tmpfile)
curl -X POST -H "Content-Type: application/json" \-d @$tmpfile \--url "http://207.191.234.131/api/submit?email=$email&secret=$secret"

tput setaf 2 ;
echo -e "\nData submission complete."
tput sgr0;

tput setaf 5 ;
echo -e "\n\n============================================="
echo "Submitted data saved locally at : $tmpfile"
echo -e "=============================================\n\n"
tput sgr0;