脚本应用之三：实时监控网卡流量

作用：实时监控网卡接收和发送的速率，以及数据总量，并设置报警数值。

测试版本：centos 系列

实现：通过ifconfig指令获取数据，计算后循环同一位置打印信息，一秒一次； tput bel指令报警

不足：一次只能监控一块网卡；

由于7.x 于5.x和6.x，ifconfig获取信息有差异，需要根据系统版本手动指定相关参数， 7.x 请将第一个位置参数设置为 “7”，其他版本为“6”（默认，可不指定）

演示：

脚本会自动列出发现的网卡，选择欲监控的网卡时，需要指定网卡名称，而不是序号！

HR和HT是在脚本中设置的报警数值，数据包速率超过任何一个相应的数值后都会有声音报警，单位kBps

脚本6s清屏一次，纠正用户误按键造成的显示混乱：

代码专区：

#/bin/bash

#  LY
#  ------------------
#  Copyright 2016.4.13, LingYi ([email protected]) QQ:1519952564
#  "Monitor network traffic"

#系统版本选择
#CentOS 5.x 6.x -> ‘func6x‘
#CentOS 7.x -> ‘func7x‘
func=${1:-func6x}
[[ $1 == ‘6‘ ]] && func=func6x
[[ $1 == ‘7‘ ]] && func=func7x

#报警的相关参数设置
#kBps
Speed_alarm_Rx=${2:-10000}
Speed_alarm_Tx=${3:-10000}
Alarm_times=2
#second
Alarm_speed=0.5

[[ $func == ‘func6x‘ || $func == ‘func7x‘ ]] || { echo -e "variable func wrong !!\n$0 [6|7]" && exit; }

#工具安装检查
for tool in net-tools bc; do
	if ! rpm -q --quiet $tool; then
		echo -n "Install $tool, wait ..."
		yum install -y $tool >/dev/null 2>&1
		[[ $? -eq 0 ]] && echo OK || { echo Fail && exit 1; }
	fi
done

#打印网卡列表，接收用户输入
[[ $# -ne 0 ]] && ifconfig $1 >/dev/null 2>&1 && setnic=1 && Nic=$1
while [[ $setnic -ne 1 ]]; do
	ifconfig | grep ‘^[a-zA-Z0-9]‘|awk ‘{print "[",NR,"]",$1}‘
	read -p "Input the Nic: " Nic
	[[ -z $Nic ]] && continue
	ifconfig $Nic >/dev/null 2>&1 && break
done

#针对不同系统版本采用不用函数获取流量相关数值
func6x(){
	Rx_Sum=$( ifconfig $1 | grep -i tx | grep -i Rx | awk ‘{print $3,$4}‘ | tr -d ‘[()]‘ )
        Tx_Sum=$( ifconfig $1 | grep -i tx | grep -i Rx | awk ‘{print $7,$8}‘ | tr -d ‘[()]‘ )
        Rx_Before=$( ifconfig $1 | grep -i tx | grep -i Rx | awk ‘{print $2}‘ | tr -c -d ‘0-9\n‘ )
        Tx_Before=$( ifconfig $1 | grep -i tx | grep -i Rx | awk ‘{print $6}‘ | tr -c -d ‘0-9\n‘ )
        sleep 1
        Rx_After=$( ifconfig $1 | grep -i tx | grep -i Rx | awk ‘{print $2}‘ | tr -c -d ‘0-9\n‘ )
        Tx_After=$( ifconfig $1 | grep -i tx | grep -i Rx | awk ‘{print $6}‘ | tr -c -d ‘0-9\n‘ )
        Rx=$( echo "scale=3; ( ${Rx_After} - ${Rx_Before} ) / 1024" | bc )
        Tx=$( echo "scale=3; ( ${Tx_After} - ${Tx_Before} ) / 1024" | bc )
}

func7x(){
        Rx_Sum=$( ifconfig eth0 | grep -i ‘rx packets‘ | awk ‘{print $6,$7}‘|tr -d ‘[()]‘ )
        Tx_Sum=$( ifconfig eth0 | grep -i ‘tx packets‘ | awk ‘{print $6,$7}‘|tr -d ‘[()]‘ )
        Rx_Before=$( ifconfig eth0 | grep -i ‘rx packets‘ | awk ‘{print $5}‘ )
        Tx_Before=$( ifconfig eth0 | grep -i ‘tx packets‘ | awk ‘{print $5}‘ )
        sleep 1
        Rx_After=$( ifconfig eth0 | grep -i ‘rx packets‘ | awk ‘{print $5}‘ )
        Tx_After=$( ifconfig eth0 | grep -i ‘tx packets‘ | awk ‘{print $5}‘ )
        Rx=$( echo "scale=3; ( ${Rx_After} - ${Rx_Before} ) / 1024" | bc )
        Tx=$( echo "scale=3; ( ${Tx_After} - ${Tx_Before} ) / 1024" | bc )
}

#报警函数
Alarm(){
    [[ $(awk -v n1=$1 -v n2=${Speed_alarm_Rx} ‘BEGIN{print(n1>=n2)?"0":"1"}‘) -ne 0 ]] && {
    [[ $(awk -v n1=$2 -v n2=${Speed_alarm_Tx} ‘BEGIN{print(n1>=n2)?"0":"1"}‘) -ne 0 ]] && return; }
    for((i=1; i<=Alarm_times; i++)); do tput bel; sleep $Alarm_speed; done &
}

#信号捕捉，处理“ctrl+c”操作
tput cup 0 0; tput ed; tput civis; trap ‘clear; tput cnorm; exit 0‘ 2
set $Nic; i=1
while true; do
	tput cup 0 0
	$func $1
	#判断接收和发送速率是否超过阈值
	[[ $(echo ${Rx} | cut -b 1) == "." ]] && Rx=0.$(echo ${Rx} | awk -F "." ‘{print $2}‘)
	[[ $(echo ${Tx} | cut -b 1) == "." ]] && Tx=0.$(echo ${Tx} | awk -F "." ‘{print $2}‘)
	#6秒清屏一次
	[[ $( expr ${i} % 7 ) -eq 6 ]] && tput ed
	Alarm $Rx $Tx
	printf " [ ${1} ]\tHR: ${Speed_alarm_Rx}  HT:${Speed_alarm_Tx}\n"
	printf "%50s\n" "" | tr ‘ ‘ ‘-‘  
	printf " %-5s%-11s%-4s\n %-5s%-11s%-4s\n" Rx: ${Rx} kB/s Tx: ${Tx} kB/s
	echo -e " Rx:  ${Rx_Sum}\n Tx:  ${Tx_Sum}\n\t\t\t$( date "+%H:%M:%S %Y.%m.%d" )"
	printf "%50s\n" ""|tr ‘ ‘ ‘-‘
	let i++
done