Freesco, NND, CDN, EOS

#!/bin/bash
# rchtb 0.22 Zciech
#
VER=0.22-05.09.09

# Idea i podstawowe zasady:
# rc.htb 0.3.3-stable, (C)PriSM 2oo3, GNU GPL
# <http://www.rc.htb.prv.pl/>

# umiescic w /etc/rchtb/rchtb
# Wszystkie predkosci przesylu podane w kilobitach na sekunde
#
# wymaga pliku /etc/htb.conf o zawartosci
# -----------------------------8><------------------------------
# #IP             rano        wieczor          noc
# #         07.00-16.59      17.00-22.59      23.00-06.59
# #         down   upl      down    upl      down    upl
# #
# 192.168.0.2      128   120      128   120      180   160
# 192.168.0.3      128   120      128   120      180   160
# 192.168.0.4      180   180      180   180      180   180
# 192.168.0.4      180   180      180   180      180   180
# # i tak dalej dla wszystkich hostow, podajmy max download i upload w danej porze dnia
# -----------------------------><8------------------------------

# do pliku crona dopisac:
# -----------------------------8><------------------------------
# # Restart HTB
# 2 7,15,22 * * * /etc/rchtb/rchtb restart >>/dev/null
# -----------------------------><8------------------------------

# --------------------------------------------
# Prorytert klas htb
# klasa         LAN(download)   WAN(upload)
# ACK               1
# ICMP         1      2
# Gry         2      3
# Userzy      3      4
# Goscie                                5
# p2p         4      6
# ruch lokalny,SQUD   5
# --------------------------------------------
# Priorytety filtrow
# filtr            LAN(download)   WAN(upload)
# ruch lokalny          1
# ACK                                   1
# ICMP                  2               2
# SUID         2
# Gry                   3               3
# p2p                   4               4
# Userzy      5      5
# Goscie                      9

       
#  Konfiguracja
htb_config=/etc/htb.conf
WANINT=eth0   # iterfejs do internetu
LANINT=eth1   # interfejs do Lanu
WANDOWN=1024   # Download w kbit/s
WANUP=256   # Upload w kbit/s
P2P_ON=1   # Wlaczenie klasy p2p - 1
P2PDOWN=0   # max download p2p w kbit/s
P2PUP=0      # max upload p2p w kbit/s
#LANSPD=4094   # LAN zwykly w [kbit/s]
LANSPD=102400   # LAN dla ftp w [kbit/s]
ICMPSPD=25   # Predkosc dla klasy ICMP (rate)
SERVSPD=15   # Predkosc dla gosci (rate)
PRIOPORT=22   # Port priorytetowy
SQUID_PORT=3128   # Port squida jesli masz, jesli nie zostaw pusty =""
ipp2p=0 #ipp2p on=1 off=0
layer7=1 #layer7 on=1 off=0
#programy markowane przez layera7
programy="fasttrack bittorrent edonkey applejuice directconnect gnutella openft mute"

# Koniec konfiguracji


SFQ="sfq perturb 10"   # SFQ
STARTMRK=2      # Startowy MARK
#nalezy tu wpisac prawdziwe sciezki do tc i iptraf
h=/usr/sbin/rchtb_tc
i=/usr/sbin/iptables

stop()
{
 $i -F -t mangle
 $i -X -t mangle
 $h qdisc del root dev $WANINT 2> /dev/null
 $h qdisc del root dev $LANINT 2> /dev/null
   
}

start()
{
 stop # dla bezpieczeństwa
 godz=`date +%H%M`   
 if [ $godz -ge  "0700" -a $godz -lt  "1459" ];then
    pora=1 # dzien
 elif [ $godz -ge  "1500" -a $godz -lt  "2159" ];then
    pora=2 # wieczor
 elif [ $godz -ge  "2200" -a $godz -lt  "2359" ];then
    pora=3 # noc
 elif [ $godz -ge  "0000" -a $godz -lt  "0659" ];then
    pora=3 # noc
 fi
 #P2P-ustawienia zaleznie od pory dnia
if [ $pora = 1 ];then
    #dzień d=75kB u=14kB
    P2PDOWN=600
    P2PUP=112
elif [ $pora = 2 ];then
    #wieczor d=1kB u=0,5kB :(
    P2PDOWN=8
    P2PUP=4
elif [ $pora = 3 ];then
    #noc d=85kB u=16kB
    P2PDOWN=680
    P2PUP=128
fi
 # obliczenia
 WANDOWN=$[$WANDOWN*98/100]
 WANUP=$[$WANUP*98/100]
 LANSPD=$[$LANSPD*95/100]
 LOCAL=$[$LANSPD-$WANDOWN]
 ACKSPD=$[$WANDOWN/20]   
 CNT=1
 for ii in `cat $htb_config|grep [0123456789] | cut -d "#" -f1 |awk '{print $1}'`;do
    CNT=$[$CNT+1]
 done 
 USERDOWN=$[$WANDOWN/$CNT]
 USERUP=$[$WANUP-$ACKSPD-$ICMPSPD-$SERVSPD]
 USERUP=$[$USERUP/$CNT]
 if [ $USERDOWN \< 1 ];then
    USERDOWN=1
 fi
 if [ $USERUP \< 1 ];then
     USERUP=1
 fi
 # INTERFEJS LANINT (ruch przychodzący)
 $h qdisc add dev $LANINT root handle 1:0 htb default 3 r2q 1
 
 # Główna klasa dla LANINT
 $h class add dev $LANINT parent 1:0 classid 1:1 htb rate ${LANSPD}kbit ceil ${LANSPD}kbit
 
 # Podział na pasmo dla łącza internetowego i resztę
 $h class add dev $LANINT parent 1:1 classid 1:2 htb rate ${WANDOWN}kbit ceil ${WANDOWN}kbit
 $h class add dev $LANINT parent 1:1 classid 1:3 htb rate ${LOCAL}kbit ceil ${LOCAL}kbit prio 5
 $h qdisc add dev $LANINT parent 1:3 $SFQ

 # pasmo priorytetowe dla ICMP, TOS 0x10 (min. delay) oraz wybranego portu 22
 $h class add dev $LANINT parent 1:2 classid 1:4 htb rate ${USERDOWN}kbit ceil ${WANDOWN}kbit prio 1 quantum 1500
 $h qdisc add dev $LANINT parent 1:4 sfq perturb 10
 $h filter add dev $LANINT protocol ip prio 2 parent 1:0 u32 match ip sport 22 0xffff flowid 1:4
 # $h filter add dev $LANINT protocol ip prio 2 parent 1:0 u32 match ip tos 0x10 0xff flowid 1:4
 $h filter add dev $LANINT protocol ip prio 2 parent 1:0 u32 match ip protocol 1 0xff flowid 1:4

 # transfer serwer->LAN
 $h filter add dev $LANINT protocol ip prio 1 parent 1:0 handle 1 fw flowid 1:3
 
 # squid transparent proxy
 if [ $SQUID_PORT ]; then
  $h filter add dev $LANINT protocol ip prio 2 parent 1:0 u32 match ip sport $SQUID_PORT 0xffff match ip tos 0x00 0xff flowid 1:3
  $i -t mangle -A OUTPUT -o $LANINT -p tcp --sport ! $SQUID_PORT -j MARK --set-mark 1
  $i -t mangle -A OUTPUT -o $LANINT -p udp -j MARK --set-mark 1
 else
  $i -t mangle -A OUTPUT -o $LANINT -j MARK --set-mark 1     
 fi
   
 # INTERFEJS  (ruch wychodzący)
 $h qdisc add dev $WANINT root handle 2:0 htb default 11 r2q 1
     
 # główna klasa
 $h class add dev $WANINT parent 2:0 classid 2:1 htb rate ${WANUP}kbit ceil ${WANUP}kbit
       
 # klasa priorytetowa ACK
 $h class add dev $WANINT parent 2:1 classid 2:2 htb rate ${ACKSPD}kbit ceil ${WANUP}kbit prio 1 quantum 1500
 $h qdisc add dev $WANINT parent 2:2 sfq perturb 10
 $h filter add dev $WANINT parent 2:0 protocol ip prio 1 u32 match ip protocol 6 0xff \
 match u8 0x05 0x0f at 0 match u16 0x0000 0xffc0 at 1 match u8 0x10 0xff at 33 flowid 2:2 # ACK

 # klasa priorytetowa TOS 0x10 oraz ICMP i DNS
 $h class add dev $WANINT parent 2:1 classid 2:3 htb rate ${ICMPSPD}kbit ceil ${WANUP}kbit prio 2 quantum 1500
 $h qdisc add dev $WANINT parent 2:3 $SFQ
 # $h filter add dev $WANINT parent 2:0 protocol ip prio 1 u32 match ip tos 0x10 0xff flowid 2:3 # TOS
 $h filter add dev $WANINT parent 2:0 protocol ip prio 2 u32 match ip dport 22 0xffff flowid 2:3 # port 22
 $h filter add dev $WANINT parent 2:0 protocol ip prio 2 u32 match ip sport 22 0xffff flowid 2:3 # port 22
 $h filter add dev $WANINT parent 2:0 protocol ip prio 2 u32 match ip dport 53 0xffff flowid 2:3 # port 53
 $h filter add dev $WANINT parent 2:0 protocol ip prio 2 u32 match ip protocol 1 0xff flowid 2:3 # ICMP
 
 # ruch dla gosci
 $h class add dev $WANINT parent 2:1 classid 2:4 htb rate ${SERVSPD}kbit ceil ${WANUP}kbit prio 5 quantum 1500
 $h qdisc add dev $WANINT parent 2:4 $SFQ
 $h filter add dev $WANINT parent 2:0 protocol ip prio 6 handle 1 fw flowid 2:4
 
 $i -t mangle -A OUTPUT -o $WANINT -j MARK --set-mark 1     
 # Pasmo dla p2p
  $h class add dev $LANINT parent 1:2 classid 1:5 htb rate 4kbit ceil ${P2PDOWN}kbit prio 4 quantum 1500
  $h qdisc add dev $LANINT parent 1:5 $SFQ
  $h class add dev $WANINT parent 2:1 classid 2:5 htb rate 4kbit ceil ${P2PUP}kbit prio 6 quantum 1500
  $h qdisc add dev $WANINT parent 2:5 $SFQ
 if [ $P2P_ON = 1 ];then

 if [ "$ipp2p" == "1" ]; then
 $i -t mangle -A PREROUTING -j CONNMARK --restore-mark
 $i -t mangle -A PREROUTING -m mark ! --mark 0 -j ACCEPT   
 $i -t mangle -A PREROUTING -m ipp2p --ipp2p -j MARK --set-mark 2
 $i -t mangle -A PREROUTING -m mark --mark 2 -j CONNMARK --save-mark
 $i -t mangle -A POSTROUTING -m mark --mark 2 -j RETURN
  fi
 
  if [ "$layer7" == "1" ]; then
   for p in $programy; do
    $i -t mangle -A PREROUTING -m layer7 --l7proto $p -j MARK --set-mark 2
     done
      fi
     
 
  $h filter add dev $LANINT parent 1:0 protocol ip prio 4 handle 2 fw flowid 1:5
  $h filter add dev $WANINT parent 2:0 protocol ip prio 4 handle 2 fw flowid 2:5
 fi

 # Userzy
 CNT=1
 MARK=$[$STARTMRK+2]
 grep "^" $htb_config | grep -v "^#"|grep [0123456789] | while read IP USERDOWN[1] USERUP[1] USERDOWN[2] USERUP[2] USERDOWN[3] USERUP[3] REM; do
    USERDOWN0=$USERDOWN
    USERUP0=$USERUP
    if [ $USERDOWN -gt ${USERDOWN[$pora]} ]; then
   USERDOWN0=${USERDOWN[$pora]}
    fi   
    if [ $USERUP -gt ${USERUP[$pora]} ]; then
        USERUP0=${USERUP[$pora]}
    fi
    if [ $WANDOWN -lt ${USERDOWN[$pora]} ]; then
        USERDOWN[$pora]=$WANDOWN
    fi
    if [ $WANUP -lt ${USERUP[$pora]} ]; then
        USERUP[$pora]=$WANUP
    fi
    $h class add dev $LANINT parent 1:2 classid 1:$[10+$CNT] htb rate ${USERDOWN0}kbit ceil ${USERDOWN[$pora]}kbit prio 3 quantum 1500
    $h qdisc add dev $LANINT parent 1:$[10+$CNT] $SFQ
    $h filter add dev $LANINT protocol ip prio 5 parent 1:0 u32 match ip dst $IP flowid 1:$[10+$CNT]
    $i -t mangle -A PREROUTING -s $IP -j MARK --set-mark $MARK
    $i -t mangle -A PREROUTING -j MARK --set-mark $MARK
    $h class add dev $WANINT parent 2:1 classid 2:$[10+$CNT] htb rate ${USERUP0}kbit ceil ${USERUP[$pora]}kbit prio 4 quantum 1500
    $h qdisc add dev $WANINT parent 2:$[10+$CNT] $SFQ
    $h filter add dev $WANINT parent 2:0 protocol ip prio 5 handle $MARK fw flowid 2:$[10+$CNT]
    MARK=$[$MARK+1]
    CNT=$[$CNT+1]
 done


# -----------------------------8><------------------------------
 # Pasmo dla gier, zaplotkuj lub wykasuj jesli nie uzywasz
 #$h class add dev $LANINT parent 1:2 classid 1:6 htb rate 40kbit ceil 200kbit prio 2 quantum 1500
 #$h qdisc add dev $LANINT parent 1:6 $SFQ
 #$h class add dev $WANINT parent 2:1 classid 2:6 htb rate 40kbit ceil 200kbit prio 3 quantum 1500
 #$h qdisc add dev $WANINT parent 2:6 $SFQ   
 #$i -t mangle -A PREROUTING -p tcp -m multiport --port 6111,6112 -j MARK --set-mark 3
 #$h filter add dev $LANINT parent 1:0 protocol ip prio 3 handle 3 fw flowid 1:6
 #$h filter add dev $WANINT parent 2:0 protocol ip prio 3 handle 3 fw flowid 2:6
# -----------------------------><8-------------------------------
   
 # reszta   
 $h filter add dev $WANINT parent 2:0 protocol ip prio 9 u32 match ip dst 0/0 flowid 2:4
}

echo "htb $VER zciech"
case "$1" in
 'start')
    echo -n "Uruchamianie kolejkowania..."
    start
    echo " wykonano."
    ;;
 'stop')
    echo -n "Zatrzymywanie kolejkowania..."
    stop
    echo " wykonano."   
    ;;
 'restart')
    echo -n "Restartowanie kolejkowania: stop"
    stop
    echo -n " start"
    start
    echo " wykonano."
    ;;
 'status')
    echo "Klasy na interfejsie $LANINT"
    echo "================================"
    tc class show dev $LANINT | grep root
    tc class show dev $LANINT | grep -v root | sort | nl
    echo "Klasy na interfejsie $WANINT"
    echo "==================================="
    tc class show dev $WANINT | grep root
    tc class show dev $WANINT | grep -v root | sort | nl
    ;;
 *)
    echo -e "\nUżycie: rc.htb start|stop|restart|status"
    ;;
esac