#!/bin/bash
#
# Contributed by: radim.roska@gmail.com
#
# 4 interfaces, 16 cores
# 
# for setting CPU affinity either use https://github.com/jelaas/eth-affinity/tree/22f0707e980a4d74f34ab9775bc07fadd0dda9d7
# or script supplied by intel with its ixgbe driver...

grep -q pktgen /proc/modules | modprobe pktgen

#ctrl+c stops generation but still shows result
trap bashtrap INT

USAGE="$0 -c <#cpus> -e <ethX> -s <#bytes> -d <ns> -n <#> -r <pps>"

CPUS=16
DEVICE=eth3
MIN=0
MAX=0
N=100000000
SIZE=1512
myDELAY=0
PROFILE=0

PPS_SUM=0

OPROFILE_REPORT=/root/setups/opreport_`date +"%H:%M:%S-%d-%m"`.log

while getopts "c:e:d:s:n:hp" opts ; do
    case $opts in
	p) PROFILE=1;;
	c) CPUS=$OPTARG;;
        e) DEVICE=eth$OPTARG;;
        s) SIZE=$OPTARG;;
	r) PPKS=$OPTARG;;
	d) myDELAY=$OPTARG;;
	n) N=$OPTARG;;
        h) echo -e $USAGE;exit 1;;
        ?) echo -e $USAGE;
        exit 1;;
    esac
done

#================================================================================================

function print_config() {
	echo "--------------------------------------------------------"
	echo `date`
	echo "N=$N, CPUS=$CPUS, DEV=$DEVICE, pkt_size=$SIZE"
	echo "--------------------------------------------------------"
}

function pgset() {
	local result
	echo $1 > $PGDEV
}

function bashtrap() {

	echo "CTRL+C Detected !...stop generation"
	pgset "stop"

}

function print_result() {

	echo "--------------"
	echo "Results for $1"
	grep -h pps /proc/net/pktgen/${1}*

	SUM=0
	for a in `grep -h pps /proc/net/pktgen/${1}* | cut -d"p" -f1 `;do 
		SUM=$((SUM+a))
	done

	echo "pps = " $SUM
	PPS_SUM=$((PPS_SUM + SUM))
	SUM=0
	for a in `grep -h pkts-sofar /proc/net/pktgen/${1}* | awk '{print $2}'`;do 
		SUM=$((SUM+a))
	done
	echo "packets sent = " $SUM

	echo `cat /proc/net//pktgen/eth* | grep Mb  | awk '{s+=$2} END {print s}'` Mbps
	
}

#================================================================================================

# Config Start Here

PKTS=$((N / CPUS))
RATEP=$((PPS / CPUS))
CLONE_SKB=" clone_skb 1000"
PKT_SIZE=" pkt_size $SIZE "
COUNT=" count $PKTS"

# delay 0 means maximum speed.
if [[ -z $PPKS ]]; then
        DELAY="delay $myDELAY"
else
        DELAY="ratep $PPKS"
fi

print_config

#specify interfaces you want pktgen to run
ETHs=("eth2" "eth3" "eth4" "eth5")

#do mapping  - manually !number of ETHs must correspond 
#NIC1 is next to CPU1 and NIC2 to CPU2 - trying to take that into account
CPU_ETH0=(4 5 12 13)
CPU_ETH1=(6 7 14 15)
CPU_ETH2=(0 1 8 9)
CPU_ETH3=(2 3 10 11)

for pgdev in /proc/net/pktgen/kpktgend_*
do
	PGDEV=$pgdev
	# echo "Removing all devices"
	pgset "rem_device_all"
done

#ETH0
for cpu_core in ${CPU_ETH0[@]}
do
	PGDEV=/proc/net/pktgen/kpktgend_$cpu_core
#	echo "Adding ${ETHs[0]} `seq 0 $MAX_QUEUE`"
	pgset "add_device ${ETHs[0]}@$cpu_core "
done
#ETH1
for cpu_core in ${CPU_ETH1[@]}
do
	PGDEV=/proc/net/pktgen/kpktgend_$cpu_core
#	echo "Adding ${ETHs[1]} `seq 0 $MAX_QUEUE`"
	pgset "add_device ${ETHs[1]}@$cpu_core "
done
#ETH2
for cpu_core in ${CPU_ETH2[@]}
do
	PGDEV=/proc/net/pktgen/kpktgend_$cpu_core
#	echo "Adding ${ETHs[1]} `seq 0 $MAX_QUEUE`"
	pgset "add_device ${ETHs[2]}@$cpu_core "
done
#ETH3
for cpu_core in ${CPU_ETH3[@]}
do
	PGDEV=/proc/net/pktgen/kpktgend_$cpu_core
#	echo "Adding ${ETHs[1]} `seq 0 $MAX_QUEUE`"
	pgset "add_device ${ETHs[3]}@$cpu_core "
done

for iface in ${ETHs[@]}
do
	#max number of queues our systems is 16 - thats just easiest way how to make it work :)
	for queue in {0..15}
	do
		PGDEV=/proc/net/pktgen/$iface@$queue

		[[ ! -f $PGDEV ]] && continue

		echo " Configuring $PGDEV"
		pgset "$COUNT"
		pgset "flag QUEUE_MAP_CPU"
		pgset "$CLONE_SKB"
		pgset "$PKT_SIZE"
		pgset "$DELAY"

	#	pgset "ratep $RATEP"
	#	pgset "dst 10.11.11.11"
	#	pgset "dst_mac $MAC"

	#----------------------------------
	#single flow
	#----------------------------------
		pgset "dst 10.0.0.3" 
		#pgset "dst 10.0.0.5" 


	#----------------------------------
	#multiple src flow - IP
	#----------------------------------
		pgset "flag IPSRC_RND"
		pgset "src_min 10.0.0.1"            #Set the minimum (or only) source IP.
		pgset "src_max 10.0.0.254"          #Set the maximum source IP.


	#----------------------------------
	#multiple dest flow - IP
	#----------------------------------
		#pgset " flag IPDST_RND"
		# pgset "dst_min 10.0.5.1"            #Same as dst
		# pgset "dst_max 10.0.5.254"          #Set the maximum destination IP.
		


	#----------------------------------
	#dest ports
	#----------------------------------
		pgset "flag UDPDST_RND"
		#pgset "config 1"
		PORT=$((10000 + processor + RANDOM%10000))
		pgset "udp_dst_min $PORT"
		pgset "udp_dst_max $PORT"


		#my eth3 mac
		#pgset "src_mac 00:25:90:0e:59:39" 
		#test
		pgset "dst_mac  00:00:00:00:00:01"
		#pgset "dst_mac  00:25:90:0e:59:57"
	done
done

# Time to run
PGDEV=/proc/net/pktgen/pgctrl
echo "Running . . . ctrl^C to stop "
pgset "start"
echo "Done"

#ETH0
print_result ${ETHs[0]}
#ETH1
print_result ${ETHs[1]}
#ETH2
print_result ${ETHs[2]}
#ETH3
print_result ${ETHs[3]}

echo sum = $PPS_SUM