Edit: /opt/dedrads/check_bandwidth (4546B)
#!/bin/bash
# Shared RADS Tools, written by Erik Soroka (eriks@imhadmin.net, ext 834)
CHECKUSER=$1
SESSIONID=$2
if [ -z $1 ]; then
echo
echo "Usage: check_bandwidth [username]"
echo
exit 0
fi
if [ -z $2 ]; then
SESSIONID=`date +%s`
DETAILED=YES
fi
if [ "$2" == "--plaintext" ]; then
RADSCOLORS=off
fi
MONTH=`date +%m | cut -d0 -f2`
YEAR=`date +%Y`
echo
#lets make sure we have bandwidth data for this user
if [ -f /var/cpanel/bandwidth/$CHECKUSER ]; then
# COLLECT HTTP BANDWIDTH DATA
cat /var/cpanel/bandwidth/$CHECKUSER |grep http |grep "$MONTH...$YEAR" > /tmp/.$SESSIONID
echo -e "\033[40;37;2m HTTPD \033[0m" > /tmp/.$SESSIONID.http
for bw in `cat /tmp/.$SESSIONID`; do
HTTPDATE=`echo $bw | cut -d'-' -f1`
HTTPBYTES=`echo $bw | awk -F "=" '{print $2}'`
HTTPMB=`echo $HTTPBYTES/1024/1024 | bc`
if [ "$RADSCOLORS" == "off" ]; then
echo "HTTPDATE : $HTTPMB MB"
else
echo -e "\033[29;30;1m$HTTPDATE \033[31;34m:\033[0m\033[29;33m $HTTPMB MB \033[0m"
fi
done >> /tmp/.$SESSIONID.http
# COLLECT IMAP BANDWIDTH DATA
cat /var/cpanel/bandwidth/$CHECKUSER |grep imap |grep "$MONTH...$YEAR" > /tmp/.$SESSIONID
echo -e "\033[40;37;2m IMAP \033[0m" > /tmp/.$SESSIONID.imap
for bw in `cat /tmp/.$SESSIONID`; do
IMAPDATE=`echo $bw | cut -d'-' -f1`
IMAPBYTES=`echo $bw | awk -F "=" '{print $2}'`
IMAPMB=`echo $IMAPBYTES/1024/1024 | bc`
if [ "$RADSCOLORS" == "off" ]; then
echo "$IMAPDATE : $IMAPMB MB"
else
echo -e "\033[29;30;1m$IMAPDATE \033[31;34m:\033[0m\033[29;33m $IMAPMB MB \033[0m"
fi
done >> /tmp/.$SESSIONID.imap
# COLLECT POP3 BANDWIDTH DATA
cat /var/cpanel/bandwidth/$CHECKUSER |grep pop3 |grep "$MONTH...$YEAR" > /tmp/.$SESSIONID
echo -e "\033[40;37;2m POP3 \033[0m" > /tmp/.$SESSIONID.pop3
for bw in `cat /tmp/.$SESSIONID`; do
POP3DATE=`echo $bw | cut -d'-' -f1`
POP3BYTES=`echo $bw | awk -F "=" '{print $2}'`
POP3MB=`echo $POP3BYTES/1024/1024 | bc`
if [ "$RADSCOLORS" == "off" ]; then
echo "$POP3DATE : $POP3MB MB"
else
echo -e "\033[29;30;1m$POP3DATE \033[31;34m:\033[0m\033[29;33m $POP3MB MB \033[0m"
fi
done >> /tmp/.$SESSIONID.pop3
# COLLECT FTP BANDWIDTH DATA
cat /var/cpanel/bandwidth/$CHECKUSER |grep ftp |grep "$MONTH...$YEAR" > /tmp/.$SESSIONID
echo -e "\033[40;37;2m FTP \033[0m" > /tmp/.$SESSIONID.ftp
for bw in `cat /tmp/.$SESSIONID`; do
FTPDATE=`echo $bw | cut -d'-' -f1`
FTPBYTES=`echo $bw | awk -F "=" '{print $2}'`
FTPMB=`echo $FTPBYTES/1024/1024 | bc`
if [ "$RADSCOLORS" == "off" ]; then
echo "$FTPDATE : $FTPMB MB"
else
echo -e "\033[29;30;1m$FTPDATE \033[31;34m:\033[0m\033[29;33m $FTPMB MB \033[0m"
fi
done >> /tmp/.$SESSIONID.ftp
# we're just giving the summary since this was invoked from the check script
if [ "$RADSCOLORS" == "off" ]; then
echo " Displaying summary of bandwidth transfer for this user this month: "
else
echo -e " \033[40;37;2m Displaying summary of bandwidth transfer for this user this month: \033[0m"
fi
HTTPTOTALMB=`cat /tmp/.$SESSIONID.http | awk '{print $3}' | awk '{total = total + $1}END{print total}'`
IMAPTOTALMB=`cat /tmp/.$SESSIONID.imap | awk '{print $3}' | awk '{total = total + $1}END{print total}'`
FTPTOTALMB=`cat /tmp/.$SESSIONID.ftp | awk '{print $3}' | awk '{total = total + $1}END{print total}'`
POP3TOTALMB=`cat /tmp/.$SESSIONID.pop3 | awk '{print $3}' | awk '{total = total + $1}END{print total}'`
if [ "$RADSCOLORS" == "off" ]; then
echo " HTTP: $HTTPTOTALMB MB IMAP: $IMAPTOTALMB MB POP3: $POP3TOTALMB MB FTP: $FTPTOTALMB MB"
else
echo -e " HTTP: \033[33;32m $HTTPTOTALMB MB\033[0m IMAP: \033[33;32m $IMAPTOTALMB MB\033[0m POP3: \033[33;32m$POP3TOTALMB MB \033[0m FTP: \033[33;32m$FTPTOTALMB MB \033[0m"
fi
echo
#no session ID was received earlier, we were invoked directly, lets display add'l history:
if [ "$DETAILED" == "YES" ]; then
echo
echo -e "\033[40;37;2m Displaying details of user's bandwidth transfer for various services over the past month \033[0m"
echo
paste /tmp/.$SESSIONID.http /tmp/.$SESSIONID.imap /tmp/.$SESSIONID.pop3 /tmp/.$SESSIONID.ftp
echo
echo -ne "\033[0m"
echo
fi
rm -f /tmp/.$SESSIONID
rm -f /tmp/.$SESSIONID.http
rm -f /tmp/.$SESSIONID.imap
rm -f /tmp/.$SESSIONID.pop3
rm -f /tmp/.$SESSIONID.ftp
rm -f /tmp/.$SESSIONID.smtp
else
#we do not have bandwidth data for this user
echo -e " ERROR: Could not locate any bandwidth data for $CHECKUSER in /var/cpanel/bandwidth/ \033[0m "
echo
exit 0
fi