#!/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

#

WWWPREFIX=.

WSIZE=800

HSIZE=250

# only recent rrds offer slope-mode:

GRAPHOPTS=--slope-mode

function makeGraphs()

{

   if [ $MULTI = 1 ]

   then

         IMGFILEEND="$3.png"

         RRDFILEEND="$3.rrd"

   else

         IMGFILEEND="png"

         RRDFILEEND="rrd"

   fi

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/questions-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -t "Questions and answers per second" \

         -v "packets" \

         DEF:questions=pdns_recursor.$RRDFILEEND:questions:AVERAGE  \

         DEF:nxdomainanswers=pdns_recursor.$RRDFILEEND:nxdomain-answers:AVERAGE \

         DEF:noerroranswers=pdns_recursor.$RRDFILEEND:noerror-answers:AVERAGE \

         DEF:servfailanswers=pdns_recursor.$RRDFILEEND:servfail-answers:AVERAGE \

         LINE1:questions#0000ff:"questions/s"\

         AREA:noerroranswers#00ff00:"noerror answers/s"  \

         STACK:nxdomainanswers#ffa500:"nxdomain answers/s"\

         STACK:servfailanswers#ff0000:"servfail answers/s"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/tcp-questions-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -t "TCP questions and answers per second, unauthorized packets/s" \

         -v "packets" \

         DEF:tcpquestions=pdns_recursor.$RRDFILEEND:tcp-questions:AVERAGE  \

         DEF:unauthudp=pdns_recursor.$RRDFILEEND:unauthorized-udp:AVERAGE  \

         DEF:unauthtcp=pdns_recursor.$RRDFILEEND:unauthorized-tcp:AVERAGE  \

         LINE1:tcpquestions#0000ff:"tcp questions/s" \

         LINE1:unauthudp#ff0000:"udp unauth/s"  \

         LINE1:unauthtcp#00ff00:"tcp unauth/s"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/packet-errors-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -t "Parsing errors per second" \

         -v "packets" \

         DEF:clientparseerrors=pdns_recursor.$RRDFILEEND:client-parse-errors:AVERAGE  \

         DEF:serverparseerrors=pdns_recursor.$RRDFILEEND:server-parse-errors:AVERAGE  \

         LINE1:clientparseerrors#0000ff:"bad packets from clients" \

         LINE1:serverparseerrors#00ff00:"bad packets from servers"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/latencies-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -t "Questions answered within latency" \

         -v "questions" \

         DEF:questions=pdns_recursor.$RRDFILEEND:questions:AVERAGE  \

         DEF:answers01=pdns_recursor.$RRDFILEEND:answers0-1:AVERAGE \

         DEF:answers110=pdns_recursor.$RRDFILEEND:answers1-10:AVERAGE \

         DEF:answers10100=pdns_recursor.$RRDFILEEND:answers10-100:AVERAGE \

         DEF:answers1001000=pdns_recursor.$RRDFILEEND:answers100-1000:AVERAGE \

         DEF:answersslow=pdns_recursor.$RRDFILEEND:answers-slow:AVERAGE \

         LINE1:questions#0000ff:"questions/s" \

         AREA:answers01#00ff00:"<1 ms" \

         STACK:answers110#0000ff:"<10 ms" \

         STACK:answers10100#00ffff:"<100 ms" \

         STACK:answers1001000#ffff00:"<1000 ms" \

         STACK:answersslow#ff0000:">1000 ms"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/qoutq-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0 \

         -t "Questions/outqueries per second" \

         -v "packets" \

         DEF:questions=pdns_recursor.$RRDFILEEND:questions:AVERAGE  \

         DEF:alloutqueries=pdns_recursor.$RRDFILEEND:all-outqueries:AVERAGE \

         LINE1:questions#ff0000:"questions/s"\

         LINE1:alloutqueries#00ff00:"outqueries/s"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/qa-latency-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0 \

         -t "Questions/answer latency in milliseconds" \

         -v "msec" \

         DEF:qalatency=pdns_recursor.$RRDFILEEND:qa-latency:AVERAGE  \

         CDEF:mqalatency=qalatency,1000,/ \

         LINE1:mqalatency#ff0000:"questions/s"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/timeouts-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -t "Outqueries/timeouts per second" \

         -v "events" \

         DEF:alloutqueries=pdns_recursor.$RRDFILEEND:all-outqueries:AVERAGE  \

         DEF:outgoingtimeouts=pdns_recursor.$RRDFILEEND:outgoing-timeouts:AVERAGE \

         DEF:throttledout=pdns_recursor.$RRDFILEEND:throttled-out:AVERAGE \

         LINE1:alloutqueries#ff0000:"outqueries/s"\

         LINE1:outgoingtimeouts#00ff00:"outgoing timeouts/s"\

         LINE1:throttledout#0000ff:"throttled outqueries/s"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/caches-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -t "Cache sizes" \

         -v "entries" \

         DEF:cacheentries=pdns_recursor.$RRDFILEEND:cache-entries:AVERAGE  \

         DEF:negcacheentries=pdns_recursor.$RRDFILEEND:negcache-entries:AVERAGE  \

         DEF:nsspeedsentries=pdns_recursor.$RRDFILEEND:nsspeeds-entries:AVERAGE  \

         DEF:throttleentries=pdns_recursor.$RRDFILEEND:throttle-entries:AVERAGE  \

         LINE1:cacheentries#ff0000:"cache entries" \

         LINE1:negcacheentries#0000ff:"negative cache entries" \

         LINE1:nsspeedsentries#00ff00:"NS speeds entries" \

         LINE1:throttleentries#00ff00:"throttle map entries"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/caches2-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -t "Cache sizes" \

         -v "entries" \

         DEF:negcacheentries=pdns_recursor.$RRDFILEEND:negcache-entries:AVERAGE  \

         DEF:nsspeedsentries=pdns_recursor.$RRDFILEEND:nsspeeds-entries:AVERAGE  \

         DEF:throttleentries=pdns_recursor.$RRDFILEEND:throttle-entries:AVERAGE  \

         LINE1:negcacheentries#0000ff:"negative cache entries" \

         LINE1:nsspeedsentries#00ff00:"NS speeds entries" \

         LINE1:throttleentries#ffa000:"throttle map entries"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/load-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -v "MThreads" \

         -t "Concurrent queries" \

         DEF:concurrentqueries=pdns_recursor.$RRDFILEEND:concurrent-queries:AVERAGE  \

         LINE1:concurrentqueries#0000ff:"concurrent queries"

   rrdtool graph $GRAPHOPTS --start -$1 $WWWPREFIX/hitrate-$2.$IMGFILEEND -w $WSIZE -h $HSIZE -l 0\

         -v "percentage" \

         -t "cache hits, cpu load" \

         DEF:cachehits=pdns_recursor.$RRDFILEEND:cache-hits:AVERAGE  \

         DEF:cachemisses=pdns_recursor.$RRDFILEEND:cache-misses:AVERAGE  \

         DEF:usermsec=pdns_recursor.$RRDFILEEND:user-msec:AVERAGE \

         DEF:sysmsec=pdns_recursor.$RRDFILEEND:sys-msec:AVERAGE \

         DEF:musermsec=pdns_recursor.$RRDFILEEND:user-msec:MAX \

         DEF:msysmsec=pdns_recursor.$RRDFILEEND:sys-msec:MAX \

         CDEF:perc=cachehits,100,*,cachehits,cachemisses,+,/ \

         CDEF:userperc=usermsec,10,/ \

         CDEF:sysperc=sysmsec,10,/ \

         CDEF:totmperc=musermsec,msysmsec,+,10,/ \

         LINE1:perc#0000ff:"percentage cache hits"  \

         LINE1:totmperc#ffff00:"max cpu use" \

         AREA:userperc#ff0000:"user cpu percentage" \

         STACK:sysperc#00ff00:"system cpu percentage" \

         COMMENT:"\l" \

         COMMENT:"Cache hits " \

         GPRINT:perc:AVERAGE:"avg %-3.1lf%%\t" \

         GPRINT:perc:LAST:"last %-3.1lf%%\t" \

         GPRINT:perc:MAX:"max %-3.1lf%%" \

         COMMENT:"\l" \

         COMMENT:"System cpu " \

         GPRINT:sysperc:AVERAGE:"avg %-3.1lf%%\t" \

         GPRINT:sysperc:LAST:"last %-3.1lf%%\t" \

         GPRINT:sysperc:MAX:"max %-3.1lf%%\t" \

         COMMENT:"\l" \

         COMMENT:"User cpu   " \

         GPRINT:userperc:AVERAGE:"avg %-3.1lf%%\t" \

         GPRINT:userperc:LAST:"last %-3.1lf%%\t" \

         GPRINT:userperc:MAX:"max %-3.1lf%%"

}

CHKFORK=`ps -C pdns_recursor --no-headers | wc -l`

# Quick sanity check

if [ CHKFORK = 0 ]

then

         echo "pdns_recursor is not running!"

         exit

fi

if [ $CHKFORK = 2 ]

then

         COUNT=0

         MULTI=1

         while [ $COUNT != $CHKFORK ]

         do

                 makeGraphs 6h 6h $COUNT

                 makeGraphs 24h day $COUNT

                 #makeGraphs 7d week $COUNT

                 #makeGraphs 1m month $COUNT

                 #makeGraphs 1y year $COUNT

                 COUNT=$(($COUNT + 1))

         done

else

         MULTI=0

         makeGraphs 6h 6h

         makeGraphs 24h day

         #makeGraphs 7d week

         #makeGraphs 1m month

         #makeGraphs 1y year

fi