#!/bin/sh
# Copyright (C) 2022 iopsys Software Solutions AB
# Author: MOHAMED Kallel <mohamed.kallel@pivasoftware.com>
# Author: AMIN Ben Ramdhane <amin.benramdhane@pivasoftware.com>

. /usr/share/bbfdm/bbf_api

ipping_error()
{
	cnt=$(uci_get_bbf_dmmap dmmap_diagnostics.ipping.NumberOfRepetitions 3)
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.SuccessCount=0
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.FailureCount="$cnt"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.AverageResponseTime=0
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.MinimumResponseTime=9999
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.MaximumResponseTime=0
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.AverageResponseTimeDetailed=0
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.MinimumResponseTimeDetailed=999999999
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.MaximumResponseTimeDetailed=0
	$UCI_COMMIT_BBF_DMMAP
}

ipping_launch() {
	host=$(uci_get_bbf_dmmap dmmap_diagnostics.ipping.Host)
	cnt=$(uci_get_bbf_dmmap dmmap_diagnostics.ipping.NumberOfRepetitions 3)
	dsize=$(uci_get_bbf_dmmap dmmap_diagnostics.ipping.DataBlockSize 64)
	timeout=$(uci_get_bbf_dmmap dmmap_diagnostics.ipping.Timeout 1000)
	interface=$(uci_get_bbf_dmmap dmmap_diagnostics.ipping.interface)
	protocol=$(uci_get_bbf_dmmap dmmap_diagnostics.ipping.ProtocolVersion Any)
	[ -n "$interface" ] && device=$(ifstatus "$interface" | jsonfilter -e @.device) && device="-I $device" || device=""
	if [ "$protocol" = "IPv4" ]; then proto="-4"; elif [ "$protocol" = "IPv6" ]; then proto="-6"; else proto=""; fi
	ip_addr_used=$(get_ip_addr_used $protocol $interface)
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.IPAddressUsed="$ip_addr_used"
	[ "$host" = "" ] && return
	timeout=$((timeout/1000))
	[ "$timeout" = "0" ] && timeout="1"
	micros=1000
	success_count=0
	avg_time_sum=0
	avg_time_sum_det=0
	min=9999999
	min_det=999999999
	max=0
	max_det=0
	i=0

	while [ $i -lt "$cnt" ]; do
		i=$((i+1))
		res=$(ping -q $proto -c 1 -s $dsize -W $timeout $device $host 2>&1)
		ba=$(echo "$res" | grep "bad address")
		[ -n "$ba" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.DiagnosticState=Error_CannotResolveHostName; ipping_error; return; }
		ba=$(echo "$res" | grep "unknown host")
		[ -n "$ba" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.DiagnosticState=Error_CannotResolveHostName; ipping_error; return; }
		stc=$(echo "$res" | grep "received")
		[ -z "$stc" ] && { $UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.DiagnosticState=Error_Other; ipping_error; return; }
		times=$(echo "$res" | grep "min/avg/max")
		[ -z "$times" ] && continue
		sc1=$(echo "$stc" | awk '{print $4}')
		sc1=${sc1:-0}
		success_count=$((success_count+sc1))
		times=$(echo "$times" | awk -F'=' '{ print $2 }')
		min_time=$(echo "$times" | awk -F'[=/ ]' '{ print $2 }')
		avg_time=$(echo "$times" | awk -F'[=/ ]' '{ print $3 }')
		max_time=$(echo "$times" | awk -F'[=/ ]' '{ print $4 }')
		min_time=${min_time:-0}
		avg_time=${avg_time:-0}
		max_time=${max_time:-0}
		min_time_det=$(echo "$min_time" "$micros" | awk '{printf "%3.0f\n",$1*$2}')
		avg_time_det=$(echo "$avg_time" "$micros" | awk '{printf "%3.0f\n",$1*$2}')
		max_time_det=$(echo "$max_time" "$micros" | awk '{printf "%3.0f\n",$1*$2}')
		min_time=$(awk "BEGIN{print $min_time * 1000}")
		avg_time=$(awk "BEGIN{print $avg_time * 1000}")
		max_time=$(awk "BEGIN{print $max_time * 1000}")
		[ "$min_time" -lt $min ] && min=$min_time
		[ "$max_time" -gt $max ] && max=$max_time
		avg_time_sum=$((avg_time_sum+avg_time))
		[ "$min_time_det" -lt $min_det ] && min_det=$min_time_det
		[ "$max_time_det" -gt $max_det ] && max_det=$max_time_det
		avg_time_sum_det=$((avg_time_sum_det+avg_time_det))
	done
	failure_count=$((cnt-success_count))
	[ $success_count -gt 0 ] && avg_time=$((avg_time_sum/success_count)) || avg_time=0
	[ $success_count -gt 0 ] && avg_time_det=$((avg_time_sum_det/success_count)) || avg_time_det=0
	avg_time=$(awk "BEGIN{print int($avg_time / 1000)}")
	min_time=$(awk "BEGIN{print int($min / 1000)}")
	max_time=$(awk "BEGIN{print int($max / 1000)}")
	min_time_det=$min_det
	max_time_det=$max_det
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.DiagnosticState="Complete"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.SuccessCount="$success_count"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.FailureCount="$failure_count"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.AverageResponseTime="$avg_time"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.MinimumResponseTime="$min_time"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.MaximumResponseTime="$max_time"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.AverageResponseTimeDetailed="$avg_time_det"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.MinimumResponseTimeDetailed="$min_time_det"
	$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.MaximumResponseTimeDetailed="$max_time_det"
	$UCI_COMMIT_BBF_DMMAP
}

ipping_stop_diagnostic() {
	pids=$(pgrep -f ipping_launch)
	if [ -n "$pids" ]; then
		kill -9 "$pids" >/dev/null 2>&1
		$UCI_SET_BBF_DMMAP dmmap_diagnostics.ipping.DiagnosticState=None
		$UCI_COMMIT_BBF_DMMAP
	fi
}

if [ "$1" = "run" ]; then
	ipping_launch
elif [ "$1" = "stop" ]; then
	ipping_stop_diagnostic
else
	return
fi