#!/bin/sh                                                                                                                                                                                                         
                                                                                                                                                                                                                  
. "$MUNIN_LIBDIR/plugins/plugin.sh"                                                                                                                                                                               
                                                                                                                                                                                                                  
: <<=cut                                                                                                                                                                                                          
                                                                                                                                                                                                                  
=head1 NAME                                                                                                                                                                                                       
                                                                                                                                                                                                                  
multibandwidth - Plugin to monitor the bandwidth between localhost and serveral hosts.                                                                                                                            
                                                                                                                                                                                                                  
=head1 APPLICABLE SYSTEMS                                                                                                                                                                                         
                                                                                                                                                                                                                  
All systems with “bash”, and “munin”                                                                                                                                                                              
                                                                                                                                                                                                                  
=head1 REQUIREMENTS                                                                                                                                                                                               
                                                                                                                                                                                                                  
bing installed.                                                                                                                                                                                                   
                                                                                                                                                                                                                  
You can install bing by using (Ubuntu/Debian): apt-get install bing                                                                                                                                               
                                                                                                                                                                                                                  
=head1 CONFIGURATION                                                                                                                                                                                              
                                                                                                                                                                                                                  
The following is the default configuration                                                                                                                                                                        
                                                                                                                                                                                                                  
[multibandwidth]                                                                                                                                                                                                  
user root                                                                                                                                                                                                         
env.hosts example.org example2.org example3.org                                                                                                                                                                   
env.samples 15                                                                                                                                                                                                    
env.small_packet_size 44                                                                                                                                                                                          
env.big_packet_size 108                                                                                                                                                                                           
env.max_mbps 15728640                                                                                                                                                                                             
                                                                                                                                                                                                                  
- env.hosts explanation: hostname or IP of the hosts to calculate the bandwidth.                                                                                                                                  
                                                                                                                                                                                                                  
- env.samples explanation: Reset stats after sending samples ECHO_REQUEST packets.                                                                                                                                

- env.small_packet_size explanation: Specifies the number of data bytes to be sent in the small
        packets. The default and minimum value is 44.

- env.big_packet_size explanation: Specifies the number of data bytes to be sent in the big
        packets. The default is 108. The size should be chosen so that big packet roundtrip times
        are long enough to be accurately measured.

- env.max_mbps explanation: bing have some random spikes. This variable is used to indicate
        the maximum value of mbps that can be recorded (in bps).

=head1 MAGIC MARKERS

#%# capabilities=autoconf

=head1 VERSION

1.1.17

=head1 AUTHOR

Jose Manuel Febrer Cortés <https://www.linkedin.com/in/jfebrer/>
Marco Bertola’s help <https://www.linkedin.com/in/bertolamarco/>

=head1 LICENSE

GPLv2

=cut



case $1 in
   config)
                echo graph_title MultiBandwidth
                echo 'graph_vlabel bps'
                echo 'graph_args --base 1024 -l 0'
                echo 'graph_scale yes'
                echo 'graph_category network'
                echo 'graph_info This graph shows the bandwidth between localhost and serveral hosts'
                for host in $hosts; do
                        fieldname="host_$(clean_fieldname "$host")"
                        echo "$fieldname.label $host"
                        echo "$fieldname.draw LINE2"
                        echo "$fieldname.info Bandwidth statistics for $host"
                done
        exit 0;;
    autoconf)
                if command -v bing >/dev/null 2>&1; then
                        echo 'yes'
                        exit 0;
                else
                        echo 'no (bing not installed)'
                        exit 0;
                fi

esac


#Calculating the bandwidth
for host in $hosts; do
        fieldname="host_$(clean_fieldname "$host")"
        printf "$fieldname.value ";

        SPEED=$(timeout 6 bing localhost "$host" -n -c 1 -e "$samples" -s "$small_packet_size" -S "$big_packet_size" 2>/dev/null \
                |grep "estimated link" -A 2 \
                | grep bps \
                | awk '{print $2}' \
                | cut -d "b" -f1)

        if (echo "$SPEED" | grep -q "M"); then
                VALUE=`echo "$SPEED" | sed 's/.$//'`
                RATE=`echo "$VALUE * 1048576" | bc -l`

                if [ $(echo "$RATE" >= "$max_mbps" | bc >/dev/null && echo "no" || echo "yes") = "yes" ]; then
                        echo "$max_mbps"
                else
                        echo "$RATE"
                fi
        elif (echo "$SPEED" | grep -q "K"); then
                VALUE=`echo "$SPEED" | sed 's/.$//'`
                echo "$VALUE * 1024" | bc -l
        elif (echo "$SPEED" | grep -q "G"); then
                VALUE=`echo "$SPEED" | sed 's/.$//'`
                echo "$VALUE * 1073742000" | bc -l
        else
                echo "Error: no data (timeout)" >&2
        fi
done