_ Git - flowspec-xdp/blob - dropcount.sh

   1 #!/bin/bash
   2 function PRINTCNT() {
   3 if [ "$1" != "" ]; then
   4         if [ "$1" = "0" ]; then
   5                 echo -e "$2\t$3\tInvalid packet length"
   6         elif [ "$1" = "1" ]; then
   7                 echo -e "$2\t$3\tInvalid VLAN tag"
   8         elif [ "$1" = "2" ]; then
   9                 echo -e "$2\t$3\tInvalid/rejected IHL IPv4 field"
  10         elif [ "$1" = "3" ]; then
  11                 echo -e "$2\t$3\tRejected IPv6 fragments"
  12         else
  13                 echo -en "$2\t$3\t"
  14                 cat "$(dirname ${BASH_SOURCE[0]})/installed-rules.txt" | head -n $(( $1 - 3 )) | tail -n1
  15         fi
  16 fi
  17 CNT=0
  18 }
  19 MAP_CONTENTS="$(bpftool map show | grep drop_cnt_map | awk '{ print $1 }' | tr -d ':' | while read IF; do
  20         bpftool map dump id "$IF" | grep "bytes\|packets\|key" | grep -v '\(bytes\|packets\)": 0\(,\)*$'
  21 done)"
  22 echo "$MAP_CONTENTS" | {
  23         declare -a BYTES
  24         declare -a PACKETS
  25         KEY=""
  26         while read LINE; do
  27                 case "$LINE" in
  28                         *"key"*)
  29                                 KEY=$(echo "$LINE" | awk '{ print $2 }' | tr -d ',')
  30                                 if [ "${BYTES["${KEY}"]}" = "" ]; then
  31                                         BYTES["${KEY}"]=0
  32                                         PACKETS["${KEY}"]=0
  33                                 fi
  34                                 ;;
  35                         *"bytes"*)
  36                                 BYTES["${KEY}"]=$(( ${BYTES["$KEY"]} + $(echo "$LINE" | awk '{ print $2 }' | tr -d ',') ))
  37                                 ;;
  38                         *"packets"*)
  39                                 PACKETS["$KEY"]=$(( ${PACKETS["$KEY"]} + $(echo "$LINE" | awk '{ print $2 }' | tr -d ',') ))
  40                                 ;;
  41                 esac
  42         done
  43         echo -e "pkts\tKBytes\tRule"
  44         for C in "${!BYTES[@]}"; do
  45                 PRINTCNT $C "${PACKETS["$C"]}" "$(( ${BYTES["$C"]} / 1000 ))"
  46         done
  47 }