# Copyright (C) 2013  Miroslav Lichvar <mlichvar@redhat.com>

#

# 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

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

export LC_ALL=C

export PATH=$(pwd):$PATH

if [ ! -e clknetsim ]; then

	git clone https://github.com/mlichvar/clknetsim.git || exit 1

fi

if [ ! -x clknetsim/clknetsim -o ! -e clknetsim/clknetsim.so ]; then

	pushd clknetsim

	make || exit 1

	popd

fi

export CLKNETSIM_PATH=clknetsim

. $CLKNETSIM_PATH/clknetsim.bash

default_limit=500

default_time_offset=1e-1

default_freq_offset=1e-4

default_base_delay=1e-7

default_jitter=1e-7

default_wander=1e-9

default_time_rms_limit=1e-6

default_freq_rms_limit=1e-6

default_time_max_limit=2e-6

default_freq_max_limit=2e-6

default_min_sync_time=10

default_max_sync_time=40

default_max_ptp4l_delay_limit=1e-4

default_nodes=3

default_subnets=""

default_master_node=1

default_transparent_nodes=""

default_free_running_nodes=""

default_pmc_node=0

default_nsm_node=0

default_master_start=0.0

default_slave_start=9.0

default_pmc_start=50.0

default_nsm_start=50.0

default_master_step=""

default_slave_step=""

default_update_interval=0

default_log_packets=0

default_master_conf=""

default_slave_conf=""

default_transparent_conf=""

default_pmc_conf=""

default_nsm_conf=""

default_extra_ptp4l_options=""

for defopt in $(declare | grep '^default_'); do

	defoptname=${defopt%%=*}

	optname=${defoptname#default_}

	eval "[ -z \"\${$optname:+a}\" ] && $optname=\"\$$defoptname\""

done

test_start() {

	rm -f tmp/*

	echo "Testing $@:"

}

test_pass() {

	echo "PASS"

	exit 0

}

test_fail() {

	echo "FAIL"

	exit 1

}

test_ok() {

	pad_line

	echo -e "\tOK"

	return 0

}

test_bad() {

	pad_line

	echo -e "\tBAD"

	return 1

}

test_error() {

	pad_line

	echo -e "\tERROR"

	return 1

}

msg_length=0

pad_line() {

	local line_length=56

	[ $msg_length -lt $line_length ] && \

		printf "%$[$line_length - $msg_length]s" ""

	msg_length=0

}

test_message() {

	local level=$1 eol=$2

	shift 2

	local msg="$*"

	while [ $level -gt 0 ]; do

		echo -n "  "

		level=$[$level - 1]

		msg_length=$[$msg_length + 2]

	done

	echo -n "$msg"

	msg_length=$[$msg_length + ${#msg}]

	if [ $eol -ne 0 ]; then

		echo

		msg_length=0

	fi

}

check_sync() {

	local i sync_time max_time_error max_freq_error ret=0

	local rms_time_error rms_freq_error

	test_message 2 1 "checking clock sync time, max/rms time/freq error:"

	for i in $(seq 1 $nodes); do

		[ $i -eq $master_node ] && continue

		[ $i -eq $pmc_node ] && continue

		[ $i -eq $nsm_node ] && continue

		[[ " $free_running_nodes " =~ [^0-9]$i[^0-9] ]] && continue

		sync_time=$(find_sync tmp/log.offset tmp/log.freq $i \

			$time_max_limit $freq_max_limit 1.0)

		max_time_error=$(get_stat 'Maximum absolute offset' $i)

		max_freq_error=$(get_stat 'Maximum absolute frequency' $i)

		rms_time_error=$(get_stat 'RMS offset' $i)

		rms_freq_error=$(get_stat 'RMS frequency' $i)

		test_message 3 0 "node $i: $sync_time $(printf '%.2e %.2e %.2e %.2e' \

			$max_time_error $max_freq_error $rms_time_error $rms_freq_error)"

		check_stat $sync_time $min_sync_time $max_sync_time && \

			check_stat $max_time_error 0.0 $time_max_limit && \

			check_stat $max_freq_error 0.0 $freq_max_limit && \

			check_stat $rms_time_error 0.0 $time_rms_limit && \

			check_stat $rms_freq_error 0.0 $freq_rms_limit && \

			test_ok || test_bad

		[ $? -eq 0 ] || ret=1

	done

	return $ret

}

check_ptp4l_delay() {

	local i max_delay ret=0

	test_message 2 1 "checking max abs path delay measured by ptp4l:"

	for i in $(seq 1 $nodes); do

		[ $i -eq $master_node ] && continue

		[ $i -eq $pmc_node ] && continue

		[ $i -eq $nsm_node ] && continue

		max_delay=$(grep -oE 'path delay *[-0-9]+ +[-0-9]+$' tmp/log.$i | awk '

			BEGIN {

				max = 0.0

			} { 

				abs = $4 < 0 ? -$4 : $4

				if (max < abs)

					max = abs

			} END {

				print max / 1e9

			}')

		test_message 3 0 "node $i: $(printf '%.2e' $max_delay)"

		check_stat $max_delay 0.0 $max_ptp4l_delay_limit && test_ok || test_bad

		[ $? -eq 0 ] || ret=1

	done

	return $ret

}

check_pmc_output() {

	local pattern=$1

	test_message 2 0 "checking pmc output:"

	[[ "$(cat tmp/log.$pmc_node)" =~ $pattern ]] && test_ok || test_bad

}

check_nsm_output() {

	local pattern=$1

	test_message 2 0 "checking nsm output:"

	[[ "$(cat tmp/log.$nsm_node)" =~ $pattern ]] && test_ok || test_bad

}

print_nondefaults() {

	local defopt defoptname optname

	test_message 2 1 "non-default settings:"

	declare | grep '^default_*' | while read defopt; do

		defoptname=${defopt%%=*}

		optname=${defoptname#default_}

		eval "[ \"\$$optname\" = \"\$$defoptname\" ]" || \

			test_message 3 1 $(eval "echo $optname=\$$optname")

	done

}

get_wander_expr() {

	local scaled_wander

	scaled_wander=$(awk "BEGIN {print $wander / \

		sqrt($update_interval < 0 ? 2^-($update_interval) : 1)}")

	echo "(+ $freq_offset (sum (* $scaled_wander (normal))))"

}

run_simulation() {

	test_message 2 0 "running simulation:"

	start_server $nodes \

		-n $[$(echo "$subnets" | tr -cd '|' | wc -c) + 1] \

		-o tmp/log.offset -f tmp/log.freq \

		$([ $log_packets -ne 0 ] && echo -p tmp/log.packets) \

		-R $(awk "BEGIN {print $update_interval < 0 ? 2^-($update_interval) : 1}") \

		-r $(awk "BEGIN {print $max_sync_time * 2^$update_interval}") \

		-l $(awk "BEGIN {print $limit * 2^$update_interval}") && test_ok || test_error

}

run_ptp4l() {

	local i s subnet ifaces test_subnets conf start

	test_message 1 1 "ptp4l test with $nodes nodes:"

	print_nondefaults

	[ -z "$subnets" ] && test_subnets=$(seq -s ' ' 1 $nodes) || test_subnets=$subnets

	generate_config4 "$master_node $pmc_node $nsm_node" "$test_subnets" $time_offset \

		"$(get_wander_expr)" \

		"(+ $base_delay (* $jitter (exponential)))"

	for i in $(seq 1 $nodes); do

		ifaces="" subnet=0

		for s in $test_subnets; do

			[ $s = "|" ] && subnet=$[$subnet + 1] && continue

			[ $s -eq $i ] && ifaces="$ifaces -i eth$subnet"

		done

		test_message 2 0 "starting node $i:"

		if [ $i -eq $pmc_node ]; then

			echo "node${i}_start = $pmc_start" >> tmp/conf

			start_client $i pmc "$pmc_conf" "" "-b 0 $ifaces" && \

				test_ok || test_error

		elif [ $i -eq $nsm_node ]; then

			echo "node${i}_start = $nsm_start" >> tmp/conf

			start_client $i nsm "$nsm_conf" "" "$ifaces" && \

				test_ok || test_error

		else

			if [[ " $transparent_nodes " =~ [^0-9]$i[^0-9] ]]; then

				start=$slave_start

				conf=$transparent_conf

				step=$slave_step

			elif [ $i -eq $master_node ]; then

				start=$master_start

				conf=$master_conf

				step=$master_step

			else

				start=$slave_start

				conf=$slave_conf

				step=$slave_step

			fi

			[ -z "$step" ] || echo "node${i}_step = $step" >> tmp/conf

			echo "node${i}_start = $start" >> tmp/conf

			start_client $i ptp4l "$conf" "" "$ifaces $extra_ptp4l_options" && \

				test_ok || test_error

		fi

		[ $? -ne 0 ] && return 1

	done

	run_simulation

}

run_phc2sys() {

	local i

	test_message 1 1 "phc2sys test with $nodes nodes:"

	print_nondefaults

	for i in $(seq 1 $nodes); do

		cat >> tmp/conf <<-EOF

			node${i}_freq = $(get_wander_expr)

			node${i}_refclock = (* $jitter (normal))

			node${i}_offset = $time_offset

			node${i}_start = $slave_start

		EOF

		test_message 2 0 "starting node $i:"

		start_client $i phc2sys "$slave_conf" && test_ok || test_error

		[ $? -ne 0 ] && return 1

	done

	run_simulation

}