Update README

[flowspec-xdp] / dropcount.sh

diff --git a/dropcount.sh b/dropcount.sh
index f67abf7251fb0c52c050272e26005923f949f4f7..2620b5dd6f836ec25d6a7ad4b0e959e462e367c9 100755 (executable)
--- a/dropcount.sh
+++ b/dropcount.sh
@@ -1,19 +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
-                               echo "$KEY: $CNT"
-                       fi
-                       KEY=$(echo "$LINE" | awk '{ print "0x" $4 $3 $2 $1 }')
-                       CNT=0
-                       ;;
-       esac
-done
+}
+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
+                               ;;
+               esac
+       done
+       for C in "${!COUNTS[@]}"; do
+               KEY=$C
+               CNT="${COUNTS["$KEY"]}"
+               PRINTCNT
+       done
+}