#!/bin/bash

. /etc/rc.conf
. /etc/rc.d/functions

mrtg_stop() {
        PID="`pidof -o %PPID -x mrtg.run mrtg`"
        stat_busy "Zatrzymuję MRTG"
        for i in `cat /etc/mrtg/ipki.txt`; do
                iptables -D mrtg_traffic -d $i -j RETURN 2>/dev/null
                iptables -D mrtg_traffic -s $i -j RETURN 2>/dev/null
        done
        iptables -D INPUT -j mrtg_traffic 2>/dev/null
        iptables -D FORWARD -j mrtg_traffic 2>/dev/null
        iptables -D OUTPUT -j mrtg_traffic 2>/dev/null
        iptables -F mrtg_traffic 2>/dev/null
        iptables -X mrtg_traffic 2>/dev/null
        [ ! -z "$PID" ]  && kill -9 $PID 2>/dev/null
        if [ $? -gt 0 ]; then
                stat_fail
        else
                rm_daemon mrtg
                stat_done
        fi
}

mrtg_start() {
        PID="`pidof -o %PPID -x mrtg.run mrtg`"
        stat_busy "Uruchamiam MRTG"
        if [ ! -z "$PID" ]; then
                stat_fail
        else
                add_daemon mrtg
                #Na wszelki wypadek...
                for i in `cat /etc/mrtg/ipki.txt`; do
                        iptables -D mrtg_traffic -d $i -j RETURN 2>/dev/null
                        iptables -D mrtg_traffic -s $i -j RETURN 2>/dev/null
                done
                iptables -D INPUT -j mrtg_traffic 2>/dev/null
                iptables -D FORWARD -j mrtg_traffic 2>/dev/null
                iptables -D OUTPUT -j mrtg_traffic 2>/dev/null
                iptables -F mrtg_traffic 2>/dev/null
                iptables -X mrtg_traffic 2>/dev/null
                #Start
                iptables -N mrtg_traffic 2>/dev/null
                for i in `cat /etc/mrtg/ipki.txt`; do
                        iptables -A mrtg_traffic -d $i -j RETURN 2>/dev/null
                        iptables -A mrtg_traffic -s $i -j RETURN 2>/dev/null
                done
                iptables -I INPUT -j mrtg_traffic 2>/dev/null
                iptables -I FORWARD -j mrtg_traffic 2>/dev/null
                iptables -I OUTPUT -j mrtg_traffic 2>/dev/null
                /etc/mrtg/mrtg.run &
                stat_done
        fi
}

case "$1" in
        start)
                mrtg_start
                ;;
        stop)
                mrtg_stop
                ;;
        restart)
                mrtg_stop
                sleep 3
                mrtg_start
                ;;
        *)
                echo "Użycie: $0 {start|stop|restart}"
                ;;
esac
exit 0

::1
127.0.0.1
192.168.1.1
192.168.1.2
192.168.1.3
192.168.1.4
192.168.1.5
...