#!/bin/bash
# Récupération des volumes de données passées par les interfaces
# dépendances : snmp, jq

# Tableau associatif principal pour stocker uniquement les interfaces à monitorer de l'équipement :
# interfaces[nom] = index incrémental applicatif
# ex : interfaces["Gi1/0/1"] = 0
# Cet index sert à identifier l'interface choisie dans les tableaux suivants.
# Il s'agît du tableau des équipements/interfaces demandés du fichier json. Ils ou elles peuvent ne pas être trouvé(e)s dans la réalité.
#
# Un tableau indexé classique par attribut monitoré est ensuite créé :
# Id dans les OID de l'interface : ifIds[index] = ifId  
# nom de l'interface réellement monitorée ifMonitor[index] = nom (tableau inverse de interfaces, contenant seulement les monitorées) 
# description de l'interface : ifDescriptions[index] = ifDescr
# vitesse de l'interface  : ifSpeeds[index] = ifSpeed
# compteur d'octets in : ifBytesIN[index] = ifBytesIN
# compteur d'octets out : ifBytesOUT[index] = ifBytesOUT

# path de snmpwalk sur le système
snmpwalkPath=/usr/bin/snmpwalk
# path de snmpget sur le système
snmpgetPath=/usr/bin/snmpget
# fichier de log du script
ficLog=bandwith.log
# fichier des resultats des mesures
ficMes=bandwith.json

# délai d'attente avant la prochaine mesure en secondes
delay=1
# nombre de mesures à conserver
nbMesures=5


# OID de base des noms d'interface
oidIfnames=1.3.6.1.2.1.31.1.1.1.1
# OID de base des descriptions d'interface
oidIfDescriptions=1.3.6.1.2.1.2.2.1.2
# OID de base des vitesses d'interface
oidIfSpeeds=1.3.6.1.2.1.2.2.1.5
# OID de base des compteurs octets IN
oidIfBytesIN=1.3.6.1.2.1.2.2.1.10
# OID de base des compteurs octets OUT
oidIfBytesOUT=1.3.6.1.2.1.2.2.1.16


# Test arguments entrés
if [[ $# -eq 0 ]]; then
    echo 'Erreur de paramètre : le fichier de configuration de bandwith doit être passé'
    echo 'Usage : bandwith.sh fichier_bandwith.conf'
    exit 1
fi;
# Source de la configuration et tests fichiers
if [ ! -e $1 ]; then
    echo 'Erreur : fichier '$1' non trouvé.'
    exit 1
fi;
source $1;


# Index des interfaces de l'équipement : tableau associatif, exemple : interfaces["Gi1/0/6"] = 3
declare -A interfaces
for i in ${!ifnames[@]}; do
    interfaces[${ifnames[$i]}]=$i
done

# Déclaration des tableaux
ifMonitor=()
ifIds=()
ifDescriptions=()
ifSpeeds=()
ifBytesIN=()
ifBytesOUT=()


# Recherche des noms d'interface de l'équipement.
# Paramètres de snmpwalk
walkParams=' -v '$version' -c '$community' -Onq '$ip' '
# Récupération des interfaces de l'équipement avec leur nom
walkIfname=$($snmpwalkPath $walkParams $oidIfnames)


# Recherche dans tous ces noms d'interfaces, ceux demandés par le monitoring.
# Quand un nom est trouvé, on stocke son Id  dans le tableau ifIds
# https://www.tldp.org/LDP/abs/html/string-manipulation.html
n=1                     # compteur
currentId=              # valeur courante Id d'interface
substringToRemove=.$oidIfnames.  # chaîne à retirer du résultat '.'+oidDeBase+'.'
for w in $walkIfname; do
    # $walkInfame contient une suite de paires OID, nom. On ne retiendra donc que l'OID correspondant au nom, soit l'élément précédent
    # on ne traite donc qu'un élément sur deux en calculant le modulo 2 du compteur incrémental.  
    modulo=$(($n%2))
    if [ $modulo == 0 ]
    then
        # le résultat courant du walk fait-il partie des clés du tableau des interfaces à monitorer ?
        # si oui, on le stocke son Id dans le tableau idIds
        for i in ${!interfaces[@]}; do
            nom=\"$i\"
            if [ $nom = $w ]; then
                # récupération de l'index applicatif dans le tableau global des interfaces monitorées
                index=${interfaces[$i]}

                # pour comprendre la valeur renvoyée ici par le walk : echo $oidIdCurrent 
                
                # retrait de la chaine OID de base entourée de '.' pour ne conserver que l'Id final de l'interface
                ifId=${currentId#$substringToRemove}   

                # stockage de l'Id de l'interface à l'index $index du tableau des Id
                ifIds[$index]=$ifId

                # stockage du nom de l'interface monitorée
                ifMonitor[$index]=$i 

                # log de l'entrée monitorée
                echo $i monitorée >> $ficLog
        
                # plus besoin de continuer la boucle si on a trouvé l'interface
                break
            fi
        done
    else
        currentId=$w
    fi

    n=$((n+1))
done

#######################################
#                                     #
# Boucle de monitoring                #
#                                     #
#######################################

# nombre de lignes à conserver dans le fichier de mesure = nb interfaces * nbMesures
nbLignes=$((${#ifMonitor[@]} * $nbMesures))

# Paramètres de snmpget
getParams=' -v '$version' -c '$community' -Ovq '$ip' '

while :
do
    # boucle sur les interfaces réellement monitorées
    for index in ${!ifMonitor[@]}; do
        # récupération de l'Id de l'interface
        ifId=${ifIds[$index]}
        # récupération du nom
        nom=${ifMonitor[$index]}
        # collecte de la description
        resDescription=$($snmpgetPath $getParams $oidIfDescriptions.$ifId)
        # collecte de la vitesse
        resSpeed=$($snmpgetPath $getParams $oidIfSpeeds.$ifId)
        # collecte des octets IN
        resIN=$($snmpgetPath $getParams $oidIfBytesIN.$ifId)
        # collecte des octets OUT
        resOUT=$($snmpgetPath $getParams $oidIfBytesOUT.$ifId)

        echo $nom $resDescription $resSpeed $resIN $resOUT $(date +%s)>> $ficMes
        tail -n $nbLignes $ficMes > tmp
        cat tmp > $ficMes

    done
    sleep $delay
done