#!/bin/bash
USER_IFACE='eth1.2 eth1.4 eth1.5 eth1.6 eth1.7 eth1.8 eth1.9 eth1.14 eth1.16 eth1.19'
EXT_IFACE='eth0'
DATA='/usr/unlimit.dat'
UFLAG='/usr/unlimit.flg'
TCBIN='/sbin/tc'
IPT='/usr/local/sbin/iptables'

flush () {
    $IPT -t mangle -F
    for i in $USER_IFACE $EXT_IFACE; do
        ${TCBIN} qdisc del dev $i root >/dev/null 2>&1
    done
}
init () {
    flush
    for i in $USER_IFACE $EXT_IFACE; do
        ${TCBIN} qdisc add dev $i root handle 1: htb
    done
}
build () {
   init
   n=3
   cat $DATA | while read ip rate;do
   USER_IFACE=`/sbin/ip -o route get $ip | sed 's/.*dev \(\S*\).*/\1/'`
   ${TCBIN} class add dev $EXT_IFACE parent 1: classid 1:$n htb rate $rate ceil $rate
   ${TCBIN} filter add dev $EXT_IFACE protocol ip parent 1: prio 0 handle $n fw classid 1:$n
   ${TCBIN} qdisc add dev $EXT_IFACE parent 1:$n handle $n: sfq perturb 10
   $IPT -t mangle -A FORWARD -s $ip/32 -j MARK --set-mark $n
   let n++                            
   ${TCBIN} class add dev $USER_IFACE parent 1: classid 1:$n htb rate $rate ceil $rate
   ${TCBIN} filter add dev $USER_IFACE protocol ip parent 1: prio 0  handle $n fw classid 1:$n
   ${TCBIN} qdisc add dev $USER_IFACE parent 1:$n handle $n: sfq perturb 10
   $IPT -t mangle -A FORWARD -d $ip/32 -j MARK --set-mark $n
   let n++
   done
   if [ -f ${UFLAG} ]; then
     /bin/rm ${UFLAG}
   fi
}
show() {
    for i in $USER_IFACE $EXT_IFACE; do
        echo --== $i ==-- --== $i ==-- --== $i ==-- --== $i ==--
        ${TCBIN} qdisc show dev $i
        echo =Classes= =Classes= =Classes= =Classes=
        ${TCBIN} class show dev $i
        echo =Filters= =Filters= =Filters= =Filters=
        ${TCBIN} filter show dev $i
    done
}
case $1 in
        start)
                build
        ;;
        stop)
                flush
        ;;
        show)
                show
        ;;
        *)
          echo "Usage: {start|stop|show}"
          exit 1
esac