Merge pull request #249 from emmagamma/macos_network_bandwidth

add network bandwidth support for macOS
This commit is contained in:
Ethan Edwards 2024-01-30 18:12:45 -05:00 committed by GitHub
commit a56a5f6622
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -26,29 +26,49 @@ interface_get() {
name="$(ip -o route get 192.168.0.0 | awk '{print $5}')" name="$(ip -o route get 192.168.0.0 | awk '{print $5}')"
fi fi
;; ;;
Darwin)
if type route >/dev/null; then
name="$(route -n get 192.168.0.0 2>/dev/null | awk '/interface: / {print $2}')"
fi
;;
esac esac
fi fi
echo "$name" echo "$name"
} }
# interface_bytes give interface name and signal tx/rx return Bytes # interface_bytes give an interface name and return both tx/rx Bytes, separated by whitespace (upload first)
interface_bytes() { interface_bytes() {
cat "/sys/class/net/$1/statistics/$2_bytes" case "$(uname -s)" in
Linux)
upload=$(cat "/sys/class/net/$1/statistics/tx_bytes")
download=$(cat "/sys/class/net/$1/statistics/rx_bytes")
echo "$upload $download"
;;
Darwin)
# column 7 is Ibytes (in bytes, rx, download) and column 10 is Obytes (out bytes, tx, upload)
netstat -nbI "$1" | tail -n1 | awk '{print $10 " " $7}'
;;
esac
} }
# get_bandwidth return the number of bytes exchanged for tx and rx # get_bandwidth return the number of bytes exchanged for tx and rx
get_bandwidth() { get_bandwidth() {
upload="$(interface_bytes "$1" "tx")" local upload=0
download="$(interface_bytes "$1" "rx")" local download=0
#wait the interval for Wait for interval to calculate the difference IFS=' ' read -r upload download <<< "$(interface_bytes "$1")"
# wait for interval to calculate the difference
sleep "$INTERVAL" sleep "$INTERVAL"
upload="$(bc <<<"$(interface_bytes "$1" "tx") - $upload")" IFS=' ' read -r new_upload new_download <<< "$(interface_bytes "$1")"
download="$(bc <<<"$(interface_bytes "$1" "rx") - $download")"
#set to 0 by default useful for non-existent interface upload=$(( $new_upload - $upload ))
download=$(( $new_download - $download ))
# set to 0 by default
echo "${upload:-0} ${download:-0}" echo "${upload:-0} ${download:-0}"
} }
@ -65,7 +85,7 @@ bandwidth_to_unit() {
local result="0.00" local result="0.00"
if (($1 != 0)); then if (($1 != 0)); then
result="$(bc <<<"scale=2; $1 / $size")" result="$(awk -v a="$1" -v b="$size" 'BEGIN { printf "%.2f\n", a / b }' </dev/null)"
fi fi
echo "$result ${SIZE[$size]}" echo "$result ${SIZE[$size]}"
@ -83,12 +103,6 @@ main() {
interval_update=0 interval_update=0
fi fi
if ! command -v bc &> /dev/null
then
echo "command bc could not be found!"
exit 1
fi
while true; do while true; do
if ((counter == 0)); then if ((counter == 0)); then
counter=60 counter=60