#!/bin/bash
#
# Added --fork compatibility to support multiple control sockets
#
# Original by ahu @ pdns-dev team
# Modified by Daniel Selans <dan.s at hostdime.com>
# 09.02.07
#

SOCKETDIR=/var/run # Leave off the trailing slash
CHKFORK=`ps -C pdns_recursor --no-headers | wc -l`

# Quick sanity check
if [ CHKFORK = 0 ]
then
         echo "pdns_recursor is not running!"
         exit
fi

TSTAMP=$(date +%s)

VARIABLES="questions tcp-questions cache-entries concurrent-queries\
            nxdomain-answers noerror-answers\
            servfail-answers tcp-outqueries\
            outgoing-timeouts nsspeeds-entries negcache-entries all-outqueries throttled-out\
            cache-hits cache-misses answers0-1 answers1-10  answers10-100 answers100-1000 answers-slow\
            qa-latency throttle-entries sys-msec user-msec  unauthorized-udp unauthorized-tcp client-parse-errors\
            server-parse-errors"

UVARIABLES=$(echo $VARIABLES | tr '[a-z]' '[A-Z]' | tr - _ )



if [ $CHKFORK = 2 ]
then
         # Running with --fork
         SOCKETS=`ls -t -1 $SOCKETDIR/pdns_recursor.controlsocket.* | head --lines 2`
         FORK="yes"
         COUNT=0

         for i in $SOCKETS
         do
                 PID=`echo $i | cut -d . -f3`
                 rec_control --socket-dir=$SOCKETDIR --socket-pid=$PID GET $VARIABLES |
                 (
                 for a in $UVARIABLES
                 do
                         read $a
                 done
                 rrdtool update pdns_recursor.$COUNT.rrd \
                         -t \
                         $(for a in $VARIABLES
                         do
                                 echo -n $a:
                         done | sed 's/:$//' ) \
                 $TSTAMP$(
                         for a in $UVARIABLES
                         do
                                 echo -n :${!a}
                         done)
                 )
                 COUNT=$(($COUNT + 1))
         done

else
         rec_control --socket-dir=$SOCKETDIR  GET $VARIABLES |
         (
         for a in $UVARIABLES
         do
                 read $a
         done
         rrdtool update pdns_recursor.rrd \
                 -t \
                 $(for a in $VARIABLES
                 do
                         echo -n $a:
                 done | sed 's/:$//' ) \
         $TSTAMP$(
                 for a in $UVARIABLES
                 do
                         echo -n :${!a}
                 done)
         )
fi