Forum Debian Users Gang

#!/bin/bash
#
# rc.htb 1.4, (C)Lenthir 2oo4, GNU GPL
# 2004-08-13 19:30
VER=1.4
MAX=3330
#######################
#Konfiguracja interfejsów:
out="eth0"                                       #wyjście na świat (modem)

lan="`/root/kolejkowanie/lan100`"                #adresy IP dla interfejsów lan, bez IP serwera
srv="83.17.x.x"                               #IP zewnętrzny serwera
srv_l="192.168.100.1"

#echo ${lan}

ile=`echo ${lan}|awk '{print NF}'`                 #ilości komputerów w podsieciach, bez serwera

TC=/sbin/tc                                        #ścieżka do tc
MODPROBE=/sbin/modprobe                            #ścieżka do modprobe
IP=/sbin/ip                                        #ścieżka do ip
IPTABLES=/sbin/iptables                            #ścieżka do iptables
firewall="/etc/init.d/lfirewall restart"            #jak zrestartować firewall

u_r2q=1                                            #r2q
u_quantum=2500 #default 1500                                    #quantum

#220 443 993 143 195 65188
porty_tcp="20 21 22 23 25 53 80 55541 3128 110 27015 27016 33333 27018"
porty_udp="53 80 55541 3128 53 27015 27016 27018 33333"
pping=1

#(1Mbit = 1024kbit = 1048576bity = 0,125MB/s = 128kB/s = 131072B/s):
wan_dwl=1000                                    #prędkość łacza na świat w kbit/s
wan_upl=190
ratio_d=1                                       #prędkość maksymalna jaką mogą osiągać użytkownicy (1 = 100%)
ratio_u=1

#Konfiguracja serwera
srv_dwl=20                                      #minimalna prędkość downloadu z serwera
srv_upl=20                                      #minimalna prędkość uploadu z serwera

#Konfiguracja LAN
lan_dwl[0]=100                                    #prędkość sieci LAN w Mbit/s (wartość rzeczywista!!)
lan_upl[0]=100
lan_dwl[1]=100
lan_upl[1]=100
#######################

if [[ $ile -ge $MAX ]]
        then
        echo "Niedopuszczalnie dużo komputerów!"
        exit 1
        fi

#######################

stop()
{
$TC qdisc del root dev $out 2> /dev/null
$TC qdisc del root dev imq 2> /dev/null
$IP link set imq down
}
start()
{ stop

# download
tmp=$(echo "($wan_dwl-$srv_dwl)%$ile" | bc)
srv_dwl=$(echo "$srv_dwl+$tmp" | bc)
min=$(echo "($wan_dwl-$srv_dwl)/$ile" | bc)
pri_min=$(echo "$min/2+$min%2" | bc)
sec_min=$(echo "$min/2" | bc)
max=$(echo "$wan_dwl*$ratio_d" | bc)

echo " Ilość komputerów: $ile"
echo "%===================================================================%"
echo " Ustawianie kolejki downloadu."
echo " Maksymalny download: $max kbit/s"
echo " Minimalny download serwera: $srv_dwl kbit/s"
echo " Minimalny download userów: $min kbit/s"
echo " Primary: $pri_min kbit/s    Secondary: $sec_min kbit/s"
echo " "

if [[ u_r2q=="" ]]; then
$TC qdisc add dev imq root handle 1:0 htb
else
$TC qdisc add dev imq root handle 1:0 htb r2q $u_r2q
fi

$TC class add dev imq parent 1:0 classid 1:1 htb rate ${wan_dwl}kbit ceil ${wan_dwl}kbit

# Serwer
$TC class add dev imq parent 1:1 classid 1:2 htb rate ${srv_dwl}kbit ceil ${max}kbit quantum $u_quantum
$TC qdisc add dev imq parent 1:2 sfq perturb 10
$TC filter add dev imq protocol ip parent 1:0 u32 match ip dst $srv flowid 1:2

j=3
for usr in $lan
        do
        $TC class add dev imq parent 1:1 classid 1:$j htb rate ${min}kbit ceil ${max}kbit quantum $u_quantum
        $TC class add dev imq parent 1:$j classid 1:$(($j+1)) htb rate ${pri_min}kbit ceil ${max}kbit quantum $u_quantum
        $TC class add dev imq parent 1:$j classid 1:$(($j+2)) htb rate ${sec_min}kbit ceil ${max}kbit quantum $u_quantum
        $TC qdisc add dev imq parent 1:$(($j+1)) sfq perturb 10
        $TC qdisc add dev imq parent 1:$(($j+2)) sfq perturb 10
        for prt in $porty_tcp
                 do
                 $TC filter add dev imq protocol ip parent 1:0 pref 2 u32 match ip dst $usr match ip protocol 6 0xff match ip sport $prt 0xffff flowid 1:$(($j+1))
                 done
        for prt in $porty_udp
                 do
                 $TC filter add dev imq protocol ip parent 1:0 pref 2 u32 match ip dst $usr match ip protocol 17 0xff match ip sport $prt 0xffff flowid 1:$(($j+1))
                 done
        if [[ $pping -eq 1 ]]; then
                 $TC filter add dev imq protocol ip parent 1:0 pref 2 u32 match ip dst $usr match ip protocol 1 0xff flowid 1:$(($j+1))
                 fi
        $TC filter add dev imq protocol ip parent 1:0 pref 3 u32 match ip dst $usr flowid 1:$((j+2))
        let "j=j+3"
        done
i=0
for uvs in $srv_l
        do
        $TC class add dev imq parent 1:0 classid 1:$j htb rate ${lan_dwl[$i]}Mbit ceil ${lan_dwl[$i]}Mbit quantum $u_quantum
        $TC qdisc add dev imq parent 1:$j sfq perturb 10
        $TC filter add dev imq protocol ip parent 1:0 pref 1 u32 match ip dst $uvs flowid 1:$j
        let "j=j+1"
        let "i=i+1"
        done

$IP link set imq up

# upload
tmp=$(echo "($wan_upl-$srv_upl)%$ile" | bc)
srv_upl=$(echo "$srv_upl+$tmp" | bc)
min=$(echo "($wan_upl-$srv_upl)/$ile" | bc)
pri_min=$(echo "$min/2+$min%2" | bc)
sec_min=$(echo "$min/2" | bc)
max=$(echo "$wan_upl*$ratio_u" | bc)

echo " Ustawianie kolejki uploadu."
echo " Maksymalny upload: $max kbit/s"
echo " Minimalny upload serwera: $srv_upl kbit/s"
echo " Minimalny upload userów: $min kbit/s"
echo " Primary: $pri_min kbit/s    Secondary: $sec_min kbit/s"
echo "%===================================================================%"

if [[ u_r2q=="" ]]; then
$TC qdisc add dev $out root handle 2:0 htb
else
$TC qdisc add dev $out root handle 2:0 htb r2q $u_r2q
fi

$TC class add dev $out parent 2:0 classid 2:1 htb rate ${wan_upl}kbit ceil ${wan_upl}kbit

$TC class add dev $out parent 2:1 classid 2:2 htb rate ${srv_upl}kbit ceil ${max}kbit quantum $u_quantum
$TC qdisc add dev $out parent 2:2 sfq perturb 10
$IPTABLES -t mangle -A POSTROUTING -s $srv -j MARK --set-mark 1
$TC filter add dev $out protocol ip parent 2:0 handle 1 fw flowid 2:2

j=3
for usr in $lan
        do
        $TC class add dev $out parent 2:1 classid 2:$j htb rate ${min}kbit ceil ${max}kbit quantum $u_quantum
        $TC class add dev $out parent 2:$j classid 2:$(($j+1)) htb rate ${pri_min}kbit ceil ${max}kbit quantum $u_quantum
        $TC class add dev $out parent 2:$j classid 2:$(($j+2)) htb rate ${sec_min}kbit ceil ${max}kbit quantum $u_quantum
        $TC qdisc add dev $out parent 2:$(($j+1)) sfq perturb 10
        $TC qdisc add dev $out parent 2:$(($j+2)) sfq perturb 10
        $IPTABLES -t mangle -A POSTROUTING -s $usr -j MARK --set-mark $(($j+2))
        for prt in $porty_tcp
                 do
                 $IPTABLES -t mangle -A POSTROUTING -p tcp -s $usr --dport $prt -j MARK --set-mark $(($j+1))
                 done
        for prt in $porty_udp
                 do
                 $IPTABLES -t mangle -A POSTROUTING -p udp -s $usr --dport $prt -j MARK --set-mark $(($j+1))
                 done
        if [[ $pping -eq 1 ]]; then
                 $IPTABLES -t mangle -A POSTROUTING -p icmp -s $usr -j MARK --set-mark $(($j+1))
                 fi
        $TC filter add dev $out protocol ip parent 2:0 pref 2 handle $(($j+1)) fw flowid 2:$(($j+1))
        $TC filter add dev $out protocol ip parent 2:0 pref 3 handle $(($j+2)) fw flowid 2:$(($j+2))
        let "j=j+3"
        done

i=0
for uvs in $srv_l
        do
        $TC class add dev $out parent 2:0 classid 2:$j htb rate ${lan_upl[$i]}Mbit ceil ${lan_upl[$i]}Mbit quantum $u_quantum
        $TC qdisc add dev $out parent 2:$j sfq perturb 10
        $IPTABLES -t mangle -A POSTROUTING -s $uvs -j MARK --set-mark $j
        $TC filter add dev $out protocol ip parent 2:0 pref 1 handle $j fw flowid 2:$j
        let "j=j+1"
        let "i=i+1"
        done
}
echo "rc.htb 1.4, (C)Lenthir 2oo4, GNU GPL"
echo "2004-08-13 19:30"
case "$1" in
    'start')
      echo "Uruchamianie kolejkowania..."
      start
      echo "Gotowe."
      exit 0
      ;;
    'stop')
      echo -n "Zatrzymywanie kolejkowania..."
      stop
      echo " wykonano."
      ;;
    'restart')
      echo "Restartowanie kolejkowania..."
      $firewall
      start
      echo "Gotowe."
      ;;
    'status')
      echo "Klasy na interfejsie imq!"
      echo "%====================================%"
      $TC class show dev imq | grep root
      $TC class show dev imq | grep -v root | sort | nl
      echo "Klasy na interfejsie ${out}!"
      echo "%====================================%"
      $TC class show dev $out | grep root
      $TC class show dev $out | grep -v root | sort | nl
      ;;
    'stat')
      if [[ "$2" == "" ]]; then
        echo "Script need second argument - adress IP"
        exit 1
      elif [[ "$2" == "server" ]]; then
        watch -d -n 1 "echo "Serwer:" && tc -s class show dev imq | grep -A 3 "htb 1:2 " && tc -s class show dev $out | grep -A 3 "htb 2:2 ""
      elif [[ "$2" == "lan" ]]; then
      j=3
        for usr in $lan
                do
                let "j=j+3"
                done
      i=0;trs="echo "Transfer: klient -> serwer -> klient ""
        for uvs in $srv_l
                do
                trs="$trs && echo "Lan[$i]:" && tc -s class show dev imq | grep -A 3 "htb 1:$j " && tc -s class show dev $out | grep -A 3 "htb 2:$j ""
                let "j=j+1"
                let "i=i+1"
                done
        watch -d -n 1 "$trs"
      else
      j=3
        for usr in $lan
                do
                if [[ "$2" == "$usr" ]]; then
                        watch -d -n 1 "echo "Download: " && tc -s class show dev imq | grep -A 3 "htb 1:$j " && tc -s class show dev imq | grep -A 3 "1:$(($j+1)) " && tc -s class show dev imq | grep -A 3 "1:$(($j+2)) " && echo && echo "Upload: " && tc -s class show dev $out | grep -A 3 "htb 2:$j " && tc -s class show dev $out | grep -A 3 "2:$(($j+1)) " && tc -s class show dev $out | grep -A 3 "2:$(($j+2)) ""
                        echo "Zakończono."
                        exit 0
                fi
                let "j=j+3"
                done
      fi
      ;;
    *)
      echo
      echo "Użycie: rc.htb start|stop|restart|status"
      echo "rc.htb stat <server|adress IP>"
      exit 1
      ;;
esac