#!/bin/bash

#

# shape         program for performing traffic shaping

#

# chkconfig: 2345 80 30

# description: shape is a script which sets the HTB traffic shaping \

#              parameters and uses simple text configuration file. \

#              htb stands for Hierarchical Token Bucket

#

# processname: shape

# config: /etc/sysconfig/htb.conf

#

# 17/08/11: first version xxxxxxxxxxxxxx{}fsit.cz



# Source function library.

. /etc/rc.d/init.d/functions



CONFIG="/etc/sysconfig/htb.conf"



if [ ! -f $CONFIG ] then

        echo "No config in $CONFIG"

        echo "Sample config is included at end of this script. See $0"

        exit 1

fi



start() {

        echo -n $"Starting $prog: "

        init

        #check13 || exit 1

        #RETVAL=$?

        #echo

        #[ $RETVAL = 0 ] && touch ${lockfile}

        #return $RETVAL

}



init() {

        echo

        echo "# setup parent 1:0 for upload"



        count=0

        for foo in $( seq $( cat $CONFIG | grep WAN_NAME | wc -l ) )

        do

                WAN_NAME=$( cat $CONFIG | grep WAN_NAME_${count} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                WAN_IFACE=$( cat $CONFIG | grep WAN_IFACE_${count} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                WAN_SPEED=$( cat $CONFIG | grep WAN_SPEED_${count} | sed -n 's/.*="\([^"]*\).*/\1/p' )



                echo "tc qdisc add dev $WAN_IFACE root handle 1:0 htb default 1"

                tc qdisc add dev $WAN_IFACE root handle 1:0 htb default 1



                echo "tc class add dev $WAN_IFACE parent 1:0 classid 1:1 htb rate $WAN_SPEED burst 512k"

                tc class add dev $WAN_IFACE parent 1:0 classid 1:1 htb rate $WAN_SPEED burst 512k



                count=$(( $count + 1 ))



                count2=0

                htb_class=2



                echo "# now setup all the local networks for upload"



                for foo2 in $( seq $( cat $CONFIG | grep LAN_NAME | wc -l ) )

                do

                        LAN_NAME=$( cat $CONFIG | grep LAN_NAME_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                        LAN_IFACE=$( cat $CONFIG | grep LAN_IFACE_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                        LAN_RATE=$( cat $CONFIG | grep LAN_RATE_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                        LAN_CEIL=$( cat $CONFIG | grep LAN_CEIL_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                        LAN_IP_ORIG=$( cat $CONFIG | grep LAN_IP_ORIG_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                        LAN_IP_MASQ=$( cat $CONFIG | grep LAN_IP_MASQ_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )



                        echo "# setup network/host $LAN_NAME ($LAN_IP_ORIG) for upload, rate is $LAN_RATE, ceil is $LAN_CEIL"



                        echo "   tc class add dev $WAN_IFACE parent 1:1 classid 1:${htb_class} htb rate $LAN_RATE ceil $LAN_C

EIL"

                        tc class add dev $WAN_IFACE parent 1:1 classid 1:${htb_class} htb rate $LAN_RATE ceil $LAN_CEIL



                        echo "   tc filter add dev $WAN_IFACE protocol ip parent 1:0 handle ::$(( ${count2} + 100 )) prio 1 u

32 match ip src $LAN_IP_MASQ flowid 1:${htb_class}"

                        tc filter add dev $WAN_IFACE protocol ip parent 1:0 handle ::$(( ${count2} + 100 )) prio 1 u32 match

ip src $LAN_IP_MASQ flowid 1:${htb_class}

                        echo



                        count2=$(( $count2 + 1 ))

                        htb_class=$(( $htb_class + 1 ))

                done

        done



        echo "# setup parent 1:0 for download"



        for foo in $( cat $CONFIG | grep LAN_IFACE | sed -n 's/.*="\([^"]*\).*/\1/p' | sort | uniq )

        do

                WAN_SPEED=$( cat $CONFIG | grep WAN_SPEED_0 | sed -n 's/.*="\([^"]*\).*/\1/p' )



                echo "tc qdisc add dev ${foo} root handle 1:0 htb default 1"

                tc qdisc add dev ${foo} root handle 1:0 htb default 1



                echo "tc class add dev ${foo} parent 1:0 classid 1:1 htb rate $WAN_SPEED ceil $WAN_SPEED"

                tc class add dev ${foo} parent 1:0 classid 1:1 htb rate $WAN_SPEED ceil $WAN_SPEED

        done

        echo





        htb_class=2

        count2=0

        for foo in  $( seq $( cat $CONFIG | grep LAN_NAME | wc -l ) )

        do

                LAN_NAME=$( cat $CONFIG | grep LAN_NAME_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                LAN_IFACE=$( cat $CONFIG | grep LAN_IFACE_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                LAN_RATE=$( cat $CONFIG | grep LAN_RATE_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                LAN_CEIL=$( cat $CONFIG | grep LAN_CEIL_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )



                echo "# setup network/host $LAN_NAME ($LAN_IP_ORIG) for download, rate is $LAN_RATE, ceil is $LAN_CEIL"



                echo "   tc class add dev $LAN_IFACE parent 1:1 classid 1:${htb_class} htb rate $LAN_RATE ceil $LAN_CEIL"

                tc class add dev $LAN_IFACE parent 1:1 classid 1:${htb_class} htb rate $LAN_RATE ceil $LAN_CEIL



                echo "   tc filter add dev $LAN_IFACE protocol ip parent 1:0 handle ::$(( ${count2} + 100 )) prio 1 u32 match

 ip dst $LAN_IP_ORIG flowid 1:${htb_class}"

                tc filter add dev $LAN_IFACE protocol ip parent 1:0 handle ::$(( ${count2} + 100 )) prio 1 u32 match ip dst $

LAN_IP_ORIG flowid 1:${htb_class}

                echo



                count2=$(( $count2 + 1 ))

                htb_class=$(( $htb_class + 1 ))

        done



        exit

}



stop() {

        delete_filters

        delete_qdiscs

}



delete_qdiscs() {

        # delete Qdisc WAN

        for foo in  $( cat $CONFIG | grep WAN_IFACE | sed -n 's/.*="\([^"]*\).*/\1/p' | sort | uniq )

        do

                echo "Stopping shaping on interface $foo"

                echo "tc qdisc del dev $foo root"

                tc qdisc del dev $foo root

        done





        # delete Qdisc LAN

        for foo in  $( cat $CONFIG | grep LAN_IFACE | sed -n 's/.*="\([^"]*\).*/\1/p' | sort | uniq )

        do

                echo "Stopping shaping on interface $foo"

                echo "tc qdisc del dev $foo root"

                tc qdisc del dev $foo root

        done

}



delete_filters() {

        # delete all the upload filters for network on WAN_IFACE(s)

        for foo in  $( cat $CONFIG | grep WAN_IFACE | sed -n 's/.*="\([^"]*\).*/\1/p' | sort | uniq )

        do



                count2=0

                for foo2 in $( seq $( cat $CONFIG | grep LAN_NAME | wc -l ) )

                do

                        echo "tc filter del dev $foo parent 1: protocol ip prio 1 handle 800::$(( ${count2} + 100 )) u32"   -

                        tc filter del dev $foo parent 1: protocol ip prio 1 handle 800::$(( ${count2} + 100 )) u32

                        count2=$(( $count2 + 1 ))

                done

        done



        # delete all the download filters for LAN interfaces

        count2=0

        for foo in  $( seq $( cat $CONFIG | grep LAN_NAME | wc -l ) )

        do

                LAN_IFACE=$( cat $CONFIG | grep LAN_IFACE_${count2} | sed -n 's/.*="\([^"]*\).*/\1/p' )

                echo "tc filter del dev $LAN_IFACE parent 1: protocol ip prio 1 handle 800::$(( ${count2} + 100 )) u32"

                tc filter del dev $LAN_IFACE parent 1: protocol ip prio 1 handle 800::$(( ${count2} + 100 )) u32

                count2=$(( $count2 + 1 ))

        done

}





status() {

        echo ": upload --------------------------------------------"

        for foo in  $( cat $CONFIG | grep WAN_IFACE | sed -n 's/.*="\([^"]*\).*/\1/p' | sort | uniq )

        do

                echo

                echo "Queing discipline for $foo"

                tc qdisc show dev $foo

                echo "Classes for device $foo"

                #tc -s class show dev $foo

                tc -s class show dev $foo

                echo "Filter for device $foo"

                tc filter show dev $foo

                echo

        done



        echo ": download ------------------------------------------"

        for foo in  $( cat $CONFIG | grep LAN_IFACE | sed -n 's/.*="\([^"]*\).*/\1/p' | sort | uniq )

        do

                echo

                echo "Queing discipline for $foo"

                tc qdisc show dev $foo

                echo "Classes for device $foo"

                #tc -s class show dev $foo

                tc -s class show dev $foo

                echo "Filter for device $foo"

                tc filter show dev $foo

                echo

        done

}



# See how we were called.

case "$1" in

  start)

        start

        ;;

  stop)

        stop

        ;;

  status)

        #status -p ${pidfile} $httpd

        #RETVAL=$?

        status

        ;;

  restart)

        stop

        start

        ;;

  *)

        echo $"Usage: $prog {start|stop|restart|status}

"

        exit 1

esac



exit $RETVAL



------------------ sample config file ----------------------

/etc/sysconfig/htb.conf

------------------------------------------------------------

WAN_NAME_0="Internet"

WAN_IFACE_0="eth0"

WAN_SPEED_0="8192Kbit"



LAN_NAME_0="wifi1"

LAN_IFACE_0="eth1"

LAN_IP_ORIG_0="192.168.1.0/24"

LAN_IP_MASQ_0="30.20.40.100/32"

LAN_RATE_0="1024Kbit"

LAN_CEIL_0="2048Kbit"



LAN_NAME_1="wifi2"

LAN_IFACE_1="eth2"

LAN_IP_ORIG_1="192.168.2.0/24"

LAN_IP_MASQ_1="30.20.40.101/32"

LAN_RATE_1="2048Kbit"

LAN_CEIL_1="4092Kbit"

------------------------ end -------------------------------