fix issues shellcheck and support vpn
This commit is contained in:
parent
557015eaa0
commit
65b9c8e23c
@ -2,6 +2,7 @@
|
|||||||
# Copyright (C) 2012 Stefan Breunig <stefan+measure-net-speed@mathphys.fsk.uni-heidelberg.de>
|
# Copyright (C) 2012 Stefan Breunig <stefan+measure-net-speed@mathphys.fsk.uni-heidelberg.de>
|
||||||
# Copyright (C) 2014 kaueraal
|
# Copyright (C) 2014 kaueraal
|
||||||
# Copyright (C) 2015 Thiago Perrotta <perrotta dot thiago at poli dot ufrj dot br>
|
# Copyright (C) 2015 Thiago Perrotta <perrotta dot thiago at poli dot ufrj dot br>
|
||||||
|
# Copyright (C) 2019 Jesús E. <heckyel@hyperbola.info>
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
# This program is free software: you can redistribute it and/or modify
|
||||||
# it under the terms of the GNU General Public License as published by
|
# it under the terms of the GNU General Public License as published by
|
||||||
@ -16,15 +17,42 @@
|
|||||||
# You should have received a copy of the GNU General Public License
|
# You should have received a copy of the GNU General Public License
|
||||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# Get custom IN and OUT labels if provided by command line arguments
|
||||||
|
while [[ $# -gt 1 ]]; do
|
||||||
|
key="$1"
|
||||||
|
case "$key" in
|
||||||
|
-i|--inlabel)
|
||||||
|
INLABEL="$2"
|
||||||
|
shift;;
|
||||||
|
-o|--outlabel)
|
||||||
|
OUTLABEL="$2"
|
||||||
|
shift;;
|
||||||
|
esac
|
||||||
|
shift
|
||||||
|
done
|
||||||
|
|
||||||
|
[[ -z "$INLABEL" ]] && INLABEL="IN "
|
||||||
|
[[ -z "$OUTLABEL" ]] && OUTLABEL="OUT "
|
||||||
|
|
||||||
# Use the provided interface, otherwise the device used for the default route.
|
# Use the provided interface, otherwise the device used for the default route.
|
||||||
if [[ -n $BLOCK_INSTANCE ]]; then
|
if [[ -z $INTERFACE ]] && [[ -n $BLOCK_INSTANCE ]]; then
|
||||||
INTERFACE=$BLOCK_INSTANCE
|
INTERFACE=$BLOCK_INSTANCE
|
||||||
else
|
elif [[ -z $INTERFACE ]]; then
|
||||||
INTERFACE=$(ip route | awk '/^default/ { print $5 ; exit }')
|
# Verify tun, vpn connection
|
||||||
|
if [[ $(ip route | awk '/^default/ { print $5 ; exit }') = "tun0" ]]; then
|
||||||
|
INTERFACE=$(ip route | awk -v i=5 -v j=3 'FNR == i {print $j}')
|
||||||
|
else
|
||||||
|
INTERFACE=$(ip route | awk '/^default/ { print $5 ; exit }')
|
||||||
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# Exit if there is no default route
|
||||||
|
[[ -z "$INTERFACE" ]] && exit
|
||||||
|
|
||||||
# Issue #36 compliant.
|
# Issue #36 compliant.
|
||||||
if ! [ -e "/sys/class/net/${INTERFACE}/operstate" ] || ! [ "`cat /sys/class/net/${INTERFACE}/operstate`" = "up" ]
|
if ! [ -e "/sys/class/net/${INTERFACE}/operstate" ] || \
|
||||||
|
{ ! [ "$TREAT_UNKNOWN_AS_UP" = "1" ] &&
|
||||||
|
! [ "$(cat "/sys/class/net/${INTERFACE}/operstate")" = "up" ]; }
|
||||||
then
|
then
|
||||||
echo "$INTERFACE down"
|
echo "$INTERFACE down"
|
||||||
echo "$INTERFACE down"
|
echo "$INTERFACE down"
|
||||||
@ -33,11 +61,11 @@ then
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
# path to store the old results in
|
# path to store the old results in
|
||||||
path="/dev/shm/$(basename $0)-${INTERFACE}"
|
path="/dev/shm/$(basename "$0")-${INTERFACE}"
|
||||||
|
|
||||||
# grabbing data for each adapter.
|
# grabbing data for each adapter.
|
||||||
read rx < "/sys/class/net/${INTERFACE}/statistics/rx_bytes"
|
read -r rx < "/sys/class/net/${INTERFACE}/statistics/rx_bytes"
|
||||||
read tx < "/sys/class/net/${INTERFACE}/statistics/tx_bytes"
|
read -r tx < "/sys/class/net/${INTERFACE}/statistics/tx_bytes"
|
||||||
|
|
||||||
# get time
|
# get time
|
||||||
time=$(date +%s)
|
time=$(date +%s)
|
||||||
@ -45,46 +73,46 @@ time=$(date +%s)
|
|||||||
# write current data if file does not exist. Do not exit, this will cause
|
# write current data if file does not exist. Do not exit, this will cause
|
||||||
# problems if this file is sourced instead of executed as another process.
|
# problems if this file is sourced instead of executed as another process.
|
||||||
if ! [[ -f "${path}" ]]; then
|
if ! [[ -f "${path}" ]]; then
|
||||||
echo "${time} ${rx} ${tx}" > "${path}"
|
echo "${time} ${rx} ${tx}" > "${path}"
|
||||||
chmod 0666 "${path}"
|
chmod 0666 "${path}"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# read previous state and update data storage
|
# read previous state and update data storage
|
||||||
read old < "${path}"
|
read -r old < "${path}"
|
||||||
echo "${time} ${rx} ${tx}" > "${path}"
|
echo "${time} ${rx} ${tx}" > "${path}"
|
||||||
|
|
||||||
# parse old data and calc time passed
|
# parse old data and calc time passed
|
||||||
old=(${old//;/ })
|
old=(${old//;/ })
|
||||||
time_diff=$(( $time - ${old[0]} ))
|
time_diff=$(( time - old[0] ))
|
||||||
|
|
||||||
# sanity check: has a positive amount of time passed
|
# sanity check: has a positive amount of time passed
|
||||||
[[ "${time_diff}" -gt 0 ]] || exit
|
[[ "${time_diff}" -gt 0 ]] || exit
|
||||||
|
|
||||||
# calc bytes transferred, and their rate in byte/s
|
# calc bytes transferred, and their rate in byte/s
|
||||||
rx_diff=$(( $rx - ${old[1]} ))
|
rx_diff=$(( rx - old[1] ))
|
||||||
tx_diff=$(( $tx - ${old[2]} ))
|
tx_diff=$(( tx - old[2] ))
|
||||||
rx_rate=$(( $rx_diff / $time_diff ))
|
rx_rate=$(( rx_diff / time_diff ))
|
||||||
tx_rate=$(( $tx_diff / $time_diff ))
|
tx_rate=$(( tx_diff / time_diff ))
|
||||||
|
|
||||||
# shift by 10 bytes to get KiB/s. If the value is larger than
|
# shift by 10 bytes to get KiB/s. If the value is larger than
|
||||||
# 1024^2 = 1048576, then display MiB/s instead
|
# 1024^2 = 1048576, then display MiB/s instead
|
||||||
|
|
||||||
# incoming
|
# incoming
|
||||||
echo -n "IN "
|
echo -n "$INLABEL"
|
||||||
rx_kib=$(( $rx_rate >> 10 ))
|
rx_kib=$(( rx_rate >> 10 ))
|
||||||
if [[ "$rx_rate" -gt 1048576 ]]; then
|
if hash bc 2>/dev/null && [[ "$rx_rate" -gt 1048576 ]]; then
|
||||||
printf '%sM' "`echo "scale=1; $rx_kib / 1024" | bc`"
|
printf '%sM' "$(echo "scale=1; $rx_kib / 1024" | bc)"
|
||||||
else
|
else
|
||||||
echo -n "${rx_kib}K"
|
echo -n "${rx_kib}K"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo -n " "
|
echo -n " "
|
||||||
|
|
||||||
# outgoing
|
# outgoing
|
||||||
echo -n "OUT "
|
echo -n "$OUTLABEL"
|
||||||
tx_kib=$(( $tx_rate >> 10 ))
|
tx_kib=$(( tx_rate >> 10 ))
|
||||||
if [[ "$tx_rate" -gt 1048576 ]]; then
|
if hash bc 2>/dev/null && [[ "$tx_rate" -gt 1048576 ]]; then
|
||||||
printf '%sM' "`echo "scale=1; $tx_kib / 1024" | bc`"
|
printf '%sM\n' "$(echo "scale=1; $tx_kib / 1024" | bc)"
|
||||||
else
|
else
|
||||||
echo -n "${tx_kib}K"
|
echo -n "${tx_kib}K"
|
||||||
fi
|
fi
|
||||||
|
Loading…
x
Reference in New Issue
Block a user