mirror of
https://dev.iopsys.eu/bbf/bbfdm.git
synced 2025-12-10 07:44:39 +01:00
94 lines
4.6 KiB
Bash
Executable file
94 lines
4.6 KiB
Bash
Executable file
#!/bin/sh
|
|
# Copyright (C) 2022 iopsys Software Solutions AB
|
|
# Author: IMEN Bhiri <imen.bhiri@pivasoftware.com>
|
|
# Author: AMIN Ben Ramdhane <amin.benramdhane@pivasoftware.com>
|
|
|
|
. /usr/share/bbfdm/bbf_api
|
|
|
|
CAPTURE_FILE="/tmp/download_dump"
|
|
|
|
download_launch() {
|
|
url=$1
|
|
device=$2
|
|
|
|
[ "$url" = "" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=Error_InitConnectionFailed; $UCI_COMMIT_BBF_DMMAP; return; }
|
|
interface=$(uci_get_bbf_dmmap dmmap_diagnostics.download.interface)
|
|
protocol=$(uci_get_bbf_dmmap dmmap_diagnostics.download.ProtocolVersion Any)
|
|
if [ "$protocol" = "IPv4" ]; then proto="--ipv4"; elif [ "$protocol" = "IPv6" ]; then proto="--ipv6"; else proto=""; fi
|
|
ip_addr_used=$(get_ip_addr_used $protocol $interface)
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.download.IPAddressUsed="$ip_addr_used"
|
|
# Disable acceleration on Broadcom devices to capture all packets with tcpdump
|
|
[ -e /usr/sbin/fcctl ] && { fcctl disable >/dev/null 2>&1; fcctl flush >/dev/null 2>&1; }
|
|
|
|
FREE_SIZE=`free | grep Mem | awk '{print $4}'`
|
|
FREE_SIZE_BYTE=$((${FREE_SIZE}/1000))
|
|
|
|
tcpdump -i "$device" -C ${FREE_SIZE_BYTE} tcp -w ${CAPTURE_FILE} > /dev/null 2>&1 &
|
|
PID=$!
|
|
sleep 1
|
|
if [ ${url:0:7} = http:// -o ${url:0:6} = ftp:// ]; then
|
|
|
|
tx_bytes_before=$(ubus call network.device status "{'name':'$device'}" | jsonfilter -e @.statistics.tx_bytes)
|
|
rx_bytes_before=$(ubus call network.device status "{'name':'$device'}" | jsonfilter -e @.statistics.rx_bytes)
|
|
time1=$(date +%s)
|
|
curl $proto --fail --silent -o /dev/null "$url"
|
|
error_code="$?"
|
|
time2=$(date +%s)
|
|
tx_bytes_after=$(ubus call network.device status "{'name':'$device'}" | jsonfilter -e @.statistics.tx_bytes)
|
|
rx_bytes_after=$(ubus call network.device status "{'name':'$device'}" | jsonfilter -e @.statistics.rx_bytes)
|
|
[ "$error_code" = "6" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=Error_CannotResolveHostName; $UCI_COMMIT_BBF_DMMAP; kill $PID 2> /dev/null; return; }
|
|
[ "$error_code" = "7" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=Error_InitConnectionFailed; $UCI_COMMIT_BBF_DMMAP; kill $PID 2> /dev/null; return; }
|
|
[ "$error_code" = "22" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=Error_NoResponse; $UCI_COMMIT_BBF_DMMAP; kill $PID 2> /dev/null; return; }
|
|
[ "$error_code" = "27" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=Error_IncorrectSize; $UCI_COMMIT_BBF_DMMAP; kill $PID 2> /dev/null; return; }
|
|
[ "$error_code" = "28" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=Error_Timeout; $UCI_COMMIT_BBF_DMMAP; kill $PID 2> /dev/null; return; }
|
|
[ "$error_code" != "0" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=Error_Other; $UCI_COMMIT_BBF_DMMAP; kill $PID 2> /dev/null; return; }
|
|
fi
|
|
tx_bytes=$((tx_bytes_after-tx_bytes_before))
|
|
rx_bytes=$((rx_bytes_after-rx_bytes_before))
|
|
periodtime=$(($((time2-time1))*1000000))
|
|
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=Complete
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.download.TotalBytesReceived=$rx_bytes
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.download.TotalBytesSent=$tx_bytes
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.download.PeriodOfFullLoading=$periodtime
|
|
perconnection=$($UCI_GET_BBF_DMMAP dmmap_diagnostics.download.EnablePerConnection)
|
|
if [ "$perconnection" = "true" ] || [ "$perconnection" = "1" ]; then
|
|
$UCI_ADD_BBF_DMMAP dmmap_diagnostics DownloadPerConnection
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.@DownloadPerConnection[0].TotalBytesReceived=$rx_bytes
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.@DownloadPerConnection[0].TotalBytesSent=$tx_bytes
|
|
else
|
|
$UCI_DELETE_BBF_DMMAP dmmap_diagnostics.@DownloadPerConnection[0]
|
|
fi
|
|
$UCI_COMMIT_BBF_DMMAP
|
|
sleep 1
|
|
kill $PID >/dev/null 2>&1
|
|
# Enable acceleration on Broadcom devices after killing the tcpdump pid
|
|
[ -e /usr/sbin/fcctl ] && { fcctl enable >/dev/null 2>&1; fcctl flush >/dev/null 2>&1; }
|
|
}
|
|
|
|
download_stop_diagnostic() {
|
|
$UCI_DELETE_BBF_DMMAP dmmap_diagnostics.@DownloadPerConnection[0]
|
|
pids=$(pgrep -f download_launch.*run)
|
|
if [ -n "$pids" ]; then
|
|
kill -9 "$pids" >/dev/null 2>&1
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=None
|
|
fi
|
|
pids=$(pgrep -f download_launch.*run)
|
|
if [ -n "$pids" ]; then
|
|
kids=$(grep -l "PPid.*$pids" /proc/*/task/*/status | grep -o "[0-9]*")
|
|
for kid in $kids; do
|
|
kill -9 "$kid" >/dev/null 2>&1
|
|
done
|
|
kill -9 "$pids" >/dev/null 2>&1
|
|
$UCI_SET_BBF_DMMAP dmmap_diagnostics.download.DiagnosticState=None
|
|
fi
|
|
$UCI_COMMIT_BBF_DMMAP
|
|
}
|
|
|
|
if [ "$1" = "run" ] ; then
|
|
download_launch "$2" "$3"
|
|
elif [ "$1" = "stop" ]; then
|
|
download_stop_diagnostic
|
|
else
|
|
return
|
|
fi
|