X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=dropcount.sh;h=2620b5dd6f836ec25d6a7ad4b0e959e462e367c9;hb=0714b6c732880fb8ceabe8102afb305ba5ecf95b;hp=3cfc4ab922c4bba79a3074104aa34a7560ffaf38;hpb=262b3b5dc1aaa6f46546f653a7fd8dcb3792fa5a;p=flowspec-xdp

diff --git a/dropcount.sh b/dropcount.sh
index 3cfc4ab..2620b5d 100755
--- a/dropcount.sh
+++ b/dropcount.sh
@@ -1,29 +1,45 @@
-#!/bin/sh
-KEY=""
+#!/bin/bash
+function PRINTCNT() {
+if [ "$KEY" != "" ]; then
+	if [ "$KEY" = "0" ]; then
+		echo -e "$CNT:\tInvalid packet length"
+	elif [ "$KEY" = "1" ]; then
+		echo -e "$CNT:\tInvalid VLAN tag"
+	elif [ "$KEY" = "2" ]; then
+		echo -e "$CNT:\tInvalid/rejected IHL IPv4 field"
+	elif [ "$KEY" = "3" ]; then
+		echo -e "$CNT:\tRejected IPv6 fragments"
+	else
+		echo -en "$CNT:\t"
+		cat "$(dirname ${BASH_SOURCE[0]})/installed-rules.txt" | head -n $(( $KEY - 3 )) | tail -n1
+	fi
+fi
 CNT=0
-bpftool map dump id "$(bpftool map show | grep drop_cnt_map | awk '{ print $1 }' | tr -d ':')" | while read LINE; do
-	case "$LINE" in
-		"key:") ;;
-		"value"*)
-			CNT=$(( $CNT + $(echo "$LINE" | awk '{ print "0x" $11 $10 $9 $8 $7 $6 $5 $4 }') ))
-			;;
-		"Found "*) ;;
-		*)
-			if [ "$KEY" != "" ]; then
-				if [ "$KEY" = "0x00000000" ]; then
-					echo "Invalid packet length: $CNT"
-				elif [ "$KEY" = "0x00000001" ]; then
-					echo "Invalid VLAN tag: $CNT"
-				elif [ "$KEY" = "0x00000002" ]; then
-					echo "Invalid/rejected IHL IPv4 field: $CNT"
-				elif [ "$KEY" = "0x00000003" ]; then
-					echo "Rejected IPv6 fragments: $CNT"
-				else
-					echo "$KEY: $CNT"
+}
+MAP_CONTENTS="$(bpftool map show | grep drop_cnt_map | awk '{ print $1 }' | tr -d ':' | while read IF; do
+	bpftool map dump id "$IF"
+done)"
+echo "$MAP_CONTENTS" | {
+	declare -a COUNTS
+	KEY=""
+	while read LINE; do
+		case "$LINE" in
+			"key:") ;;
+			"value"*)
+				COUNTS["$KEY"]=$(( ${COUNTS["$KEY"]} + $(echo "$LINE" | awk '{ print "0x" $11 $10 $9 $8 $7 $6 $5 $4 }') ))
+				;;
+			"Found "*) ;;
+			*)
+				KEY=$((16#$(echo "$LINE" | awk '{ print $4 $3 $2 $1 }')))
+				if [ "$COUNTS["$KEY"]" = "" ]; then
+					COUNTS["$KEY"]=0
 				fi
-			fi
-			KEY=$(echo "$LINE" | awk '{ print "0x" $4 $3 $2 $1 }')
-			CNT=0
-			;;
-	esac
-done
+				;;
+		esac
+	done
+	for C in "${!COUNTS[@]}"; do
+		KEY=$C
+		CNT="${COUNTS["$KEY"]}"
+		PRINTCNT
+	done
+}