config/qos/makeqosscripts.pl

   1 #!/usr/bin/perl
   2 ###############################################################################
   3 #                                                                             #
   4 # IPFire.org - A linux based firewall                                         #
   5 # Copyright (C) 2007  Michael Tremer & Christian Schmidt                      #
   6 #                                                                             #
   7 # This program is free software: you can redistribute it and/or modify        #
   8 # it under the terms of the GNU General Public License as published by        #
   9 # the Free Software Foundation, either version 3 of the License, or           #
  10 # (at your option) any later version.                                         #
  11 #                                                                             #
  12 # This program is distributed in the hope that it will be useful,             #
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of              #
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
  15 # GNU General Public License for more details.                                #
  16 #                                                                             #
  17 # You should have received a copy of the GNU General Public License           #
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
  19 #                                                                             #
  20 ###############################################################################
  21
  22 use strict;
  23 # enable only the following on debugging purpose
  24 # use warnings;
  25
  26 require '/var/ipfire/general-functions.pl';
  27 require "${General::swroot}/lang.pl";
  28 require "${General::swroot}/header.pl";
  29
  30 my %qossettings = ();
  31 my %checked = ();
  32 my %netsettings = ();
  33 my $message = "";
  34 my $errormessage = "";
  35 my $c = "";
  36 my $direntry = "";
  37 my $classentry = "";
  38 my $subclassentry = "";
  39 my $l7ruleentry = "";
  40 my $portruleentry = "";
  41 my $tosruleentry = "";
  42 my @tmp = ();
  43 my @classes = ();
  44 my @subclasses = ();
  45 my @l7rules = ();
  46 my @portrules = ();
  47 my @tosrules = ();
  48 my @tmpline = ();
  49 my @classline = ();
  50 my @subclassline = ();
  51 my @tosruleline = ();
  52 my @l7ruleline = ();
  53 my @portruleline = ();
  54 my @proto = ();
  55 my %selected= () ;
  56 my $classfile = "/var/ipfire/qos/classes";
  57 my $subclassfile = "/var/ipfire/qos/subclasses";
  58 my $level7file = "/var/ipfire/qos/level7config";
  59 my $portfile = "/var/ipfire/qos/portconfig";
  60 my $tosfile = "/var/ipfire/qos/tosconfig";
  61
  62 &General::readhash("${General::swroot}/ethernet/settings", \%netsettings);
  63
  64 $qossettings{'ENABLED'} = 'off';
  65 $qossettings{'EDIT'} = 'no';
  66 $qossettings{'OUT_SPD'} = '';
  67 $qossettings{'INC_SPD'} = '';
  68 $qossettings{'DEF_OUT_SPD'} = '';
  69 $qossettings{'DEF_INC_SPD'} = '';
  70 $qossettings{'DEFCLASS_INC'} = '';
  71 $qossettings{'DEFCLASS_OUT'} = '';
  72 $qossettings{'ACK'} = '';
  73 $qossettings{'MTU'} = '1492';
  74 $qossettings{'RED_DEV'} = `cat /var/ipfire/red/iface`;
  75 $qossettings{'IMQ_DEV'} = 'imq0';
  76 $qossettings{'TOS'} = '';
  77 $qossettings{'VALID'} = 'yes';
  78 $qossettings{'IMQ_MODE'} = 'PREROUTING';
  79
  80 &General::readhash("${General::swroot}/qos/settings", \%qossettings);
  81
  82 open( FILE, "< $classfile" ) or die "Unable to read $classfile";
  83 @classes = <FILE>;
  84 close FILE;
  85 open( FILE, "< $subclassfile" ) or die "Unable to read $subclassfile";
  86 @subclasses = <FILE>;
  87 close FILE;
  88 open( FILE, "< $level7file" ) or die "Unable to read $level7file";
  89 @l7rules = <FILE>;
  90 close FILE;
  91 open( FILE, "< $portfile" ) or die "Unable to read $portfile";
  92 @portrules = <FILE>;
  93 close FILE;
  94 open( FILE, "< $tosfile" ) or die "Unable to read $tosfile";
  95 @tosrules = <FILE>;
  96 close FILE;
  97
  98 ############################################################################################################################
  99 ############################################################################################################################
 100
 101 print <<END
 102 #/bin/bash
 103 #################################################
 104 # This is an autocreated QoS-Script for         #
 105 # IPFIRE                                        #
 106 # Copyright by the IPFire Team (GPLv2)          #
 107 # www.ipfire.org                                #
 108 #################################################
 109
 110 ### SYSTEMVARIABLES:
 111 # RED INTERFACE:        $qossettings{'RED_DEV'}
 112 # IMQ DEVICE:           $qossettings{'IMQ_DEV'}
 113
 114 eval $(/usr/local/bin/readhash /var/ipfire/main/settings)
 115 if [ "$RRDLOG" == "" ]; then
 116         RRDLOG=/var/log/rrd
 117 fi
 118
 119 case "\$1" in
 120
 121   status)
 122         case "\$2" in
 123           qdisc)
 124                 echo "[qdisc]"
 125                 tc -s qdisc show dev $qossettings{'RED_DEV'}
 126                 tc -s qdisc show dev $qossettings{'IMQ_DEV'}
 127                 exit 0
 128           ;;
 129           class)
 130                 echo "[class]"
 131                 tc -s class show dev $qossettings{'RED_DEV'}
 132                 tc -s class show dev $qossettings{'IMQ_DEV'}
 133                 exit 0
 134           ;;
 135           filter)
 136                 echo "[filter]"
 137                 tc -s filter show dev $qossettings{'RED_DEV'}
 138                 tc -s filter show dev $qossettings{'IMQ_DEV'}
 139                 exit 0
 140           ;;
 141           iptables)
 142                 echo "[iptables]"
 143                 iptables -t mangle -n -L QOS-OUT -v -x 2> /dev/null
 144                 iptables -t mangle -n -L QOS-INC -v -x 2> /dev/null
 145                 iptables -t mangle -n -L QOS-TOS -v -x 2> /dev/null
 146                 exit 0
 147           ;;
 148         esac
 149         \$0 \$1 qdisc
 150         \$0 \$1 class
 151         \$0 \$1 filter
 152         \$0 \$1 iptables
 153         exit 0
 154   ;;
 155   start)
 156         ###
 157         ### $qossettings{'RED_DEV'}
 158         ###
 159
 160         ### INIT KERNEL
 161         modprobe sch_htb
 162
 163         ### SET QUEUE LENGTH & MTU - has just to be tested!!! IMPORTANT
 164         ip link set dev $qossettings{'RED_DEV'} qlen $qossettings{'QLENGTH'}
 165         #ip link set dev $qossettings{'RED_DEV'} mtu $qossettings{'MTU'}
 166
 167         ### ADD HTB QDISC FOR $qossettings{'RED_DEV'}
 168         tc qdisc add dev $qossettings{'RED_DEV'} root handle 1: htb default $qossettings{'DEFCLASS_OUT'}
 169
 170         ### MAIN RATE LIMIT
 171         tc class add dev $qossettings{'RED_DEV'} parent 1: classid 1:1 htb rate $qossettings{'OUT_SPD'}kbit
 172
 173         ### CLASSES FOR $qossettings{'RED_DEV'}
 174 END
 175 ;
 176 foreach $classentry (sort @classes)
 177 {
 178         @classline = split( /\;/, $classentry );
 179         if ($qossettings{'RED_DEV'} eq $classline[0]) {
 180                 $qossettings{'DEVICE'} = $classline[0];
 181                 $qossettings{'CLASS'} = $classline[1];
 182                 $qossettings{'PRIO'} = $classline[2];
 183                 $qossettings{'RATE'} = $classline[3];
 184                 $qossettings{'CEIL'} = $classline[4];
 185                 $qossettings{'BURST'} = $classline[5];
 186                 $qossettings{'CBURST'} = $classline[6];
 187                 print "\ttc class add dev $qossettings{'DEVICE'} parent 1:1 classid 1:$qossettings{'CLASS'} htb rate $qossettings{'RATE'}kbit ceil $qossettings{'CEIL'}kbit prio $qossettings{'PRIO'} ";
 188                 if (($qossettings{'BURST'} ne '') && ($qossettings{'BURST'} ne 0)) {
 189                         print "burst $qossettings{'BURST'}k ";
 190                 }
 191                 if (($qossettings{'CBURST'} ne '') && ($qossettings{'CBURST'} ne 0)) {
 192                         print "cburst $qossettings{'CBURST'}k";
 193                 }
 194                 print "\n";
 195         }
 196 }
 197 foreach $subclassentry (sort @subclasses) {
 198         @subclassline = split( /\;/, $subclassentry );
 199         if ($qossettings{'RED_DEV'} eq $subclassline[0]) {
 200                 $qossettings{'DEVICE'} = $subclassline[0];
 201                 $qossettings{'CLASS'} = $subclassline[1];
 202                 $qossettings{'SCLASS'} = $subclassline[2];
 203                 $qossettings{'SPRIO'} = $subclassline[3];
 204                 $qossettings{'SRATE'} = $subclassline[4];
 205                 $qossettings{'SCEIL'} = $subclassline[5];
 206                 $qossettings{'SBURST'} = $subclassline[6];
 207                 $qossettings{'SCBURST'} = $subclassline[7];
 208                 print "\ttc class add dev $qossettings{'DEVICE'} parent 1:$qossettings{'CLASS'} classid 1:$qossettings{'SCLASS'} htb rate $qossettings{'SRATE'}kbit ceil $qossettings{'SCEIL'}kbit prio $qossettings{'SPRIO'} ";
 209                 if ($qossettings{'SBURST'} > 0) {
 210                         print "burst $qossettings{'SBURST'}k ";
 211                 }
 212                 if (($qossettings{'SCBURST'} ne '') && ($qossettings{'SCBURST'} ne 0)) {
 213                         print "cburst $qossettings{'CBURST'}k";
 214                 }
 215                 print "\n";
 216         }
 217 }
 218
 219 print "\n\t### ATTACH QDISC TO LEAF CLASSES\n";
 220 foreach $classentry (sort @classes)
 221 {
 222         @classline = split( /\;/, $classentry );
 223         if ($qossettings{'RED_DEV'} eq $classline[0]) {
 224                 $qossettings{'DEVICE'} = $classline[0];
 225                 $qossettings{'CLASS'} = $classline[1];
 226                 print "\ttc qdisc add dev $qossettings{'DEVICE'} parent 1:$qossettings{'CLASS'} handle $qossettings{'CLASS'}: sfq perturb $qossettings{'SFQ_PERTUB'}\n";
 227         }
 228 }
 229 foreach $subclassentry (sort @subclasses) {
 230         @subclassline = split( /\;/, $subclassentry );
 231         if ($qossettings{'RED_DEV'} eq $subclassline[0]) {
 232                 $qossettings{'DEVICE'} = $subclassline[0];
 233                 $qossettings{'SCLASS'} = $subclassline[2];
 234                 print "\ttc qdisc add dev $qossettings{'DEVICE'} parent 1:$qossettings{'SCLASS'} handle $qossettings{'SCLASS'}: sfq perturb $qossettings{'SFQ_PERTUB'}\n";
 235         }
 236 }
 237 print "\n\t### FILTER TRAFFIC INTO CLASSES\n";
 238 foreach $classentry (sort @classes)
 239 {
 240         @classline = split( /\;/, $classentry );
 241         if ($qossettings{'RED_DEV'} eq $classline[0]) {
 242                 $qossettings{'DEVICE'} = $classline[0];
 243                 $qossettings{'CLASS'} = $classline[1];
 244                 print "\ttc filter add dev $qossettings{'DEVICE'} parent 1:0 prio 0 protocol ip handle $qossettings{'CLASS'} fw flowid 1:$qossettings{'CLASS'}\n";
 245         }
 246 }
 247 foreach $subclassentry (sort @subclasses) {
 248         @subclassline = split( /\;/, $subclassentry );
 249         if ($qossettings{'RED_DEV'} eq $subclassline[0]) {
 250                 $qossettings{'DEVICE'} = $subclassline[0];
 251                 $qossettings{'CLASS'} = $subclassline[1];
 252                 $qossettings{'SCLASS'} = $subclassline[2];
 253                 print "\ttc filter add dev $qossettings{'DEVICE'} parent 1:0 prio 0 protocol ip handle $qossettings{'SCLASS'} fw flowid 1:$qossettings{'SCLASS'}\n";
 254         }
 255 }
 256 print <<END
 257
 258         ### ADD QOS-OUT CHAIN TO THE MANGLE TABLE IN IPTABLES
 259         iptables -t mangle -N QOS-OUT
 260         iptables -t mangle -N QOS-TOS
 261         iptables -t mangle -I POSTROUTING -o $qossettings{'RED_DEV'} -j QOS-OUT
 262         iptables -t mangle -A POSTROUTING -o $qossettings{'RED_DEV'} -j QOS-TOS
 263
 264         ### MARK ACKs
 265         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags SYN,RST SYN -j TOS --set-tos 4
 266         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags SYN,RST SYN -j MARK --set-mark $qossettings{'ACK'}
 267         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags SYN,RST SYN -j RETURN
 268
 269         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p icmp -m length --length 40:100 -j MARK --set-mark $qossettings{'ACK'}
 270         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p icmp -m length --length 40:100 -j RETURN
 271
 272         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --syn -m length --length 40:68 -j TOS --set-tos 4
 273         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --syn -m length --length 40:68 -j MARK --set-mark $qossettings{'ACK'}
 274         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --syn -m length --length 40:68 -j RETURN
 275
 276         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL SYN,ACK -m length --length 40:68 -j TOS --set-tos 4
 277         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL SYN,ACK -m length --length 40:68 -j MARK --set-mark $qossettings{'ACK'}
 278         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL SYN,ACK -m length --length 40:68 -j RETURN
 279
 280         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK -m length --length 40:100 -j TOS --set-tos 4
 281         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK -m length --length 40:100 -j MARK --set-mark $qossettings{'ACK'}
 282         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK -m length --length 40:100 -j RETURN
 283
 284         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL RST -j TOS --set-tos 4
 285         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL RST -j MARK --set-mark $qossettings{'ACK'}
 286         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL RST -j RETURN
 287
 288         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK,RST -j TOS --set-tos 4
 289         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK,RST -j MARK --set-mark $qossettings{'ACK'}
 290         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK,RST -j RETURN
 291
 292         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK,FIN -j TOS --set-tos 4
 293         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK,FIN -j MARK --set-mark $qossettings{'ACK'}
 294         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -p tcp --tcp-flags ALL ACK,FIN -j RETURN
 295
 296         ### SET TOS
 297 END
 298 ;
 299         foreach $tosruleentry (sort @tosrules)
 300         {
 301                 @tosruleline = split( /\;/, $tosruleentry );
 302                 $qossettings{'CLASS'} = $tosruleline[0];
 303                 $qossettings{'TOS'} = abs $tosruleline[2] * 2;
 304                 if ( $tosruleline[1] eq $qossettings{'RED_DEV'} )
 305                 {
 306                         print "\tiptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -m tos --tos $qossettings{'TOS'} -j MARK --set-mark $qossettings{'CLASS'}\n";
 307                         print "\tiptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -m tos --tos $qossettings{'TOS'} -j RETURN\n";
 308                 }
 309         }
 310
 311 print "\n\t### SET PORT-RULES\n";
 312         foreach $portruleentry (sort @portrules)
 313         {
 314                 @portruleline = split( /\;/, $portruleentry );
 315                 if ( $portruleline[1] eq $qossettings{'RED_DEV'} )
 316                 {
 317                         $qossettings{'CLASS'} = $portruleline[0];
 318                         $qossettings{'DEVICE'} = $portruleline[1];
 319                         $qossettings{'PPROT'} = $portruleline[2];
 320                         $qossettings{'QIP'} = $portruleline[3];
 321                         $qossettings{'QPORT'} = $portruleline[4];
 322                         $qossettings{'DIP'} = $portruleline[5];
 323                         $qossettings{'DPORT'} = $portruleline[6];
 324                         print "\tiptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} ";
 325                         if ($qossettings{'QIP'} ne ''){
 326                                 print "-s $qossettings{'QIP'} ";
 327                         }
 328                         if ($qossettings{'DIP'} ne ''){
 329                                 print "-d $qossettings{'DIP'} ";
 330                         }
 331                         print "-p $qossettings{'PPROT'} ";
 332 #                       if (($qossettings{'QPORT'} ne '') || ($qossettings{'DPORT'} ne '')){
 333 #                               print "-m multiport ";
 334 #                       }
 335                         if ($qossettings{'QPORT'} ne ''){
 336                                 print "--sport $qossettings{'QPORT'} ";
 337                         }
 338                         if ($qossettings{'DPORT'} ne ''){
 339                                 print "--dport $qossettings{'DPORT'} ";
 340                         }
 341                         print "-j MARK --set-mark $qossettings{'CLASS'}\n";
 342                         print "\tiptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} ";
 343                         if ($qossettings{'QIP'} ne ''){
 344                                 print "-s $qossettings{'QIP'} ";
 345                         }
 346                         if ($qossettings{'DIP'} ne ''){
 347                                 print "-d $qossettings{'DIP'} ";
 348                         }
 349                         print "-p $qossettings{'PPROT'} ";
 350 #                       if (($qossettings{'QPORT'} ne '') || ($qossettings{'DPORT'} ne '')){
 351 #                               print "-m multiport ";
 352 #                       }
 353                         if ($qossettings{'QPORT'} ne ''){
 354                                 print "--sport $qossettings{'QPORT'} ";
 355                         }
 356                         if ($qossettings{'DPORT'} ne ''){
 357                                 print "--dport $qossettings{'DPORT'} ";
 358                         }
 359                         print "-j RETURN\n\n";
 360                 }
 361         }
 362
 363 print <<END
 364
 365         ### SET LEVEL7-RULES
 366 END
 367 ;
 368         foreach $l7ruleentry (sort @l7rules)
 369         {
 370                 @l7ruleline = split( /\;/, $l7ruleentry );
 371                 if ( $l7ruleline[1] eq $qossettings{'RED_DEV'} )
 372                 {
 373                         $qossettings{'CLASS'} = $l7ruleline[0];
 374                         $qossettings{'DEVICE'} = $l7ruleline[1];
 375                         $qossettings{'L7PROT'} = $l7ruleline[2];
 376                         $qossettings{'QIP'} = $l7ruleline[3];
 377                         $qossettings{'DIP'} = $l7ruleline[4];
 378                         print "\tiptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} ";
 379                         if ($qossettings{'QIP'} ne ''){
 380                                 print "-s $qossettings{'QIP'} ";
 381                         }
 382                         if ($qossettings{'DIP'} ne ''){
 383                                 print "-d $qossettings{'DIP'} ";
 384                         }
 385                         print "-m layer7 --l7dir /etc/l7-protocols/protocols --l7proto $qossettings{'L7PROT'} -j MARK --set-mark $qossettings{'CLASS'}\n";
 386                         print "\tiptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} ";
 387                         if ($qossettings{'QIP'} ne ''){
 388                                 print "-s $qossettings{'QIP'} ";
 389                         }
 390                         if ($qossettings{'DIP'} ne ''){
 391                                 print "-d $qossettings{'DIP'} ";
 392                         }
 393                         print "-m layer7 --l7dir /etc/l7-protocols/protocols --l7proto $qossettings{'L7PROT'} -j RETURN\n";
 394                 }
 395         }
 396
 397 print <<END
 398
 399         ### REDUNDANT: SET ALL NONMARKED PACKETS TO DEFAULT CLASS
 400         iptables -t mangle -A QOS-OUT -o $qossettings{'RED_DEV'} -m mark --mark 0 -j MARK --set-mark $qossettings{'DEFCLASS_OUT'}
 401
 402         ###
 403         ### $qossettings{'IMQ_DEV'}
 404         ###
 405
 406         ### BRING UP $qossettings{'IMQ_DEV'}
 407         if [ `lsmod | grep -q ipt_IMQ` ]; then
 408                 insmod ipt_IMQ
 409                 sleep 2
 410         fi
 411         modprobe imq numdevs=1
 412         ip link set $qossettings{'IMQ_DEV'} up
 413
 414         ### SET QUEUE LENGTH & MTU - has just to be tested!!! IMPORTANT
 415         ip link set dev $qossettings{'IMQ_DEV'} qlen $qossettings{'QLENGTH'}
 416         # ip link set dev $qossettings{'IMQ_DEV'} mtu $qossettings{'MTU'}
 417
 418         ### ADD HTB QDISC FOR $qossettings{'IMQ_DEV'}
 419         tc qdisc add dev $qossettings{'IMQ_DEV'} root handle 2: htb default $qossettings{'DEFCLASS_INC'}
 420
 421         ### MAIN RATE LIMIT
 422         tc class add dev $qossettings{'IMQ_DEV'} parent 2: classid 2:1 htb rate $qossettings{'INC_SPD'}kbit
 423
 424         ### CLASSES FOR $qossettings{'IMQ_DEV'}
 425 END
 426 ;
 427 foreach $classentry (sort @classes)
 428 {
 429         @classline = split( /\;/, $classentry );
 430         if ($qossettings{'IMQ_DEV'} eq $classline[0]) {
 431                 $qossettings{'DEVICE'} = $classline[0];
 432                 $qossettings{'CLASS'} = $classline[1];
 433                 $qossettings{'PRIO'} = $classline[2];
 434                 $qossettings{'RATE'} = $classline[3];
 435                 $qossettings{'CEIL'} = $classline[4];
 436                 $qossettings{'BURST'} = $classline[5];
 437                 $qossettings{'CBURST'} = $classline[6];
 438                 print "\ttc class add dev $qossettings{'DEVICE'} parent 2:1 classid 2:$qossettings{'CLASS'} htb rate $qossettings{'RATE'}kbit ceil $qossettings{'CEIL'}kbit prio $qossettings{'PRIO'} ";
 439                 if (($qossettings{'BURST'} ne '') && ($qossettings{'BURST'} ne 0)) {
 440                         print "burst $qossettings{'BURST'}k ";
 441                 }
 442                 if (($qossettings{'CBURST'} ne '') && ($qossettings{'CBURST'} ne 0)) {
 443                         print "cburst $qossettings{'CBURST'}k";
 444                 }
 445                 print "\n";
 446         }
 447 }
 448 foreach $subclassentry (sort @subclasses) {
 449         @subclassline = split( /\;/, $subclassentry );
 450         if ($qossettings{'IMQ_DEV'} eq $subclassline[0]) {
 451                 $qossettings{'DEVICE'} = $subclassline[0];
 452                 $qossettings{'CLASS'} = $subclassline[1];
 453                 $qossettings{'SCLASS'} = $subclassline[2];
 454                 $qossettings{'SPRIO'} = $subclassline[3];
 455                 $qossettings{'SRATE'} = $subclassline[4];
 456                 $qossettings{'SCEIL'} = $subclassline[5];
 457                 $qossettings{'SBURST'} = $subclassline[6];
 458                 $qossettings{'SCBURST'} = $subclassline[7];
 459                 print "\ttc class add dev $qossettings{'DEVICE'} parent 2:$qossettings{'CLASS'} classid 2:$qossettings{'SCLASS'} htb rate $qossettings{'SRATE'}kbit ceil $qossettings{'SCEIL'}kbit prio $qossettings{'SPRIO'} ";
 460                 if ($qossettings{'SBURST'} > 0) {
 461                         print "burst $qossettings{'SBURST'}k ";
 462                 }
 463                 if (($qossettings{'SCBURST'} ne '') && ($qossettings{'SCBURST'} ne 0)) {
 464                         print "cburst $qossettings{'CBURST'}k";
 465                 }
 466                 print "\n";
 467         }
 468 }
 469
 470 print "\n\t### ATTACH QDISC TO LEAF CLASSES\n";
 471 foreach $classentry (sort @classes)
 472 {
 473         @classline = split( /\;/, $classentry );
 474         if ($qossettings{'IMQ_DEV'} eq $classline[0]) {
 475                 $qossettings{'DEVICE'} = $classline[0];
 476                 $qossettings{'CLASS'} = $classline[1];
 477                 print "\ttc qdisc add dev $qossettings{'DEVICE'} parent 2:$qossettings{'CLASS'} handle $qossettings{'CLASS'}: sfq perturb $qossettings{'SFQ_PERTUB'}\n";
 478         }
 479 }
 480 foreach $subclassentry (sort @subclasses) {
 481         @subclassline = split( /\;/, $subclassentry );
 482         if ($qossettings{'IMQ_DEV'} eq $subclassline[0]) {
 483                 $qossettings{'DEVICE'} = $subclassline[0];
 484                 $qossettings{'SCLASS'} = $subclassline[2];
 485                 print "\ttc qdisc add dev $qossettings{'DEVICE'} parent 2:$qossettings{'SCLASS'} handle $qossettings{'SCLASS'}: sfq perturb $qossettings{'SFQ_PERTUB'}\n";
 486         }
 487 }
 488 print "\n\t### FILTER TRAFFIC INTO CLASSES\n";
 489 foreach $classentry (sort @classes)
 490 {
 491         @classline = split( /\;/, $classentry );
 492         if ($qossettings{'IMQ_DEV'} eq $classline[0]) {
 493                 $qossettings{'DEVICE'} = $classline[0];
 494                 $qossettings{'CLASS'} = $classline[1];
 495                 print "\ttc filter add dev $qossettings{'DEVICE'} parent 2:0 prio 0 protocol ip handle $qossettings{'CLASS'} fw flowid 2:$qossettings{'CLASS'}\n";
 496         }
 497 }
 498 foreach $subclassentry (sort @subclasses) {
 499         @subclassline = split( /\;/, $subclassentry );
 500         if ($qossettings{'IMQ_DEV'} eq $subclassline[0]) {
 501                 $qossettings{'DEVICE'} = $subclassline[0];
 502                 $qossettings{'CLASS'} = $subclassline[1];
 503                 $qossettings{'SCLASS'} = $subclassline[2];
 504                 print "\ttc filter add dev $qossettings{'DEVICE'} parent 2:0 prio 0 protocol ip handle $qossettings{'SCLASS'} fw flowid 2:$qossettings{'SCLASS'}\n";
 505         }
 506 }
 507
 508 if ( $qossettings{'IMQ_MODE'} eq 'POSTROUTING' )
 509 {
 510 print <<END
 511
 512         ### ADD QOS-INC CHAIN TO THE MANGLE TABLE IN IPTABLES
 513         iptables -t mangle -N QOS-INC
 514         iptables -t mangle -A POSTROUTING -m mark ! --mark 0 -o ! $qossettings{'RED_DEV'} -j IMQ --todev 0
 515         iptables -t mangle -I FORWARD -i $qossettings{'RED_DEV'} -j QOS-INC
 516         iptables -t mangle -A FORWARD -i $qossettings{'RED_DEV'} -j QOS-TOS
 517
 518         ### SET TOS
 519 END
 520 ;
 521 }
 522 else
 523 {
 524 print <<END
 525
 526         ### ADD QOS-INC CHAIN TO THE MANGLE TABLE IN IPTABLES
 527         iptables -t mangle -N QOS-INC
 528         iptables -t mangle -A PREROUTING -i $qossettings{'RED_DEV'} -j IMQ --todev 0
 529         iptables -t mangle -I PREROUTING -i $qossettings{'RED_DEV'} -j QOS-INC
 530         iptables -t mangle -A PREROUTING -i $qossettings{'RED_DEV'} -j QOS-TOS
 531
 532         ### SET TOS
 533 END
 534 ;
 535 }
 536
 537         foreach $tosruleentry (sort @tosrules)
 538         {
 539                 @tosruleline = split( /\;/, $tosruleentry );
 540                 $qossettings{'CLASS'} = $tosruleline[0];
 541                 $qossettings{'TOS'} = abs $tosruleline[2] * 2;
 542                 if ( $tosruleline[1] eq $qossettings{'IMQ_DEV'} )
 543                 {
 544                         print "\tiptables -t mangle -A QOS-INC -i $qossettings{'RED_DEV'} -m tos --tos $qossettings{'TOS'} -j MARK --set-mark $qossettings{'CLASS'}\n";
 545                         print "\tiptables -t mangle -A QOS-INC -i $qossettings{'RED_DEV'} -m tos --tos $qossettings{'TOS'} -j RETURN\n";
 546                 }
 547
 548         }
 549
 550 print "\n\t### SET PORT-RULES\n";
 551         foreach $portruleentry (sort @portrules)
 552         {
 553                 @portruleline = split( /\;/, $portruleentry );
 554                 if ( $portruleline[1] eq $qossettings{'IMQ_DEV'} )
 555                 {
 556                         $qossettings{'CLASS'} = $portruleline[0];
 557                         $qossettings{'DEVICE'} = $portruleline[1];
 558                         $qossettings{'PPROT'} = $portruleline[2];
 559                         $qossettings{'QIP'} = $portruleline[3];
 560                         $qossettings{'QPORT'} = $portruleline[4];
 561                         $qossettings{'DIP'} = $portruleline[5];
 562                         $qossettings{'DPORT'} = $portruleline[6];
 563                         print "\tiptables -t mangle -A QOS-INC -i $qossettings{'RED_DEV'} ";
 564                         if ($qossettings{'QIP'} ne ''){
 565                                 print "-s $qossettings{'QIP'} ";
 566                         }
 567                         if ($qossettings{'DIP'} ne ''){
 568                                 print "-d $qossettings{'DIP'} ";
 569                         }
 570                         print "-p $qossettings{'PPROT'} ";
 571 #                       if (($qossettings{'QPORT'} ne '') || ($qossettings{'DPORT'} ne '')){
 572 #                               print "-m multiport ";
 573 #                       }
 574                         if ($qossettings{'QPORT'} ne ''){
 575                                 print "--sport $qossettings{'QPORT'} ";
 576                         }
 577                         if ($qossettings{'DPORT'} ne ''){
 578                                 print "--dport $qossettings{'DPORT'} ";
 579                         }
 580                         print "-j MARK --set-mark $qossettings{'CLASS'}\n";
 581                         print "\tiptables -t mangle -A QOS-INC -i $qossettings{'RED_DEV'} ";
 582                         if ($qossettings{'QIP'} ne ''){
 583                                 print "-s $qossettings{'QIP'} ";
 584                         }
 585                         if ($qossettings{'DIP'} ne ''){
 586                                 print "-d $qossettings{'DIP'} ";
 587                         }
 588                         print "-p $qossettings{'PPROT'} ";
 589 #                       if (($qossettings{'QPORT'} ne '') || ($qossettings{'DPORT'} ne '')){
 590 #                               print "-m multiport ";
 591 #                       }
 592                         if ($qossettings{'QPORT'} ne ''){
 593                                 print "--sport $qossettings{'QPORT'} ";
 594                         }
 595                         if ($qossettings{'DPORT'} ne ''){
 596                                 print "--dport $qossettings{'DPORT'} ";
 597                         }
 598                         print "-j RETURN\n\n";
 599                 }
 600         }
 601
 602 print <<END
 603
 604         ### SET LEVEL7-RULES
 605 END
 606 ;
 607         foreach $l7ruleentry (sort @l7rules)
 608         {
 609                 @l7ruleline = split( /\;/, $l7ruleentry );
 610                 if ( $l7ruleline[1] eq $qossettings{'IMQ_DEV'} )
 611                 {
 612                         $qossettings{'CLASS'} = $l7ruleline[0];
 613                         $qossettings{'DEVICE'} = $l7ruleline[1];
 614                         $qossettings{'L7PROT'} = $l7ruleline[2];
 615                         $qossettings{'QIP'} = $l7ruleline[3];
 616                         $qossettings{'DIP'} = $l7ruleline[4];
 617                         print "\tiptables -t mangle -A QOS-INC -i $qossettings{'RED_DEV'} ";
 618                         if ($qossettings{'QIP'} ne ''){
 619                                 print "-s $qossettings{'QIP'} ";
 620                         }
 621                         if ($qossettings{'DIP'} ne ''){
 622                                 print "-d $qossettings{'DIP'} ";
 623                         }
 624                         print "-m layer7 --l7dir /etc/l7-protocols/protocols --l7proto $qossettings{'L7PROT'} -j MARK --set-mark $qossettings{'CLASS'}\n";
 625                         print "\tiptables -t mangle -A QOS-INC -i $qossettings{'RED_DEV'} ";
 626                         if ($qossettings{'QIP'} ne ''){
 627                                 print "-s $qossettings{'QIP'} ";
 628                         }
 629                         if ($qossettings{'DIP'} ne ''){
 630                                 print "-d $qossettings{'DIP'} ";
 631                         }
 632                         print "-m layer7 --l7dir /etc/l7-protocols/protocols --l7proto $qossettings{'L7PROT'} -j RETURN\n";
 633                 }
 634         }
 635
 636 print <<END
 637         ### REDUNDANT: SET ALL NONMARKED PACKETS TO DEFAULT CLASS
 638         iptables -t mangle -A QOS-INC -i $qossettings{'RED_DEV'} -m mark --mark 0 -j MARK --set-mark $qossettings{'DEFCLASS_INC'}
 639
 640         ### SETTING TOS BITS
 641 END
 642 ;
 643         foreach $classentry (sort @classes)
 644         {
 645                 @classline = split( /\;/, $classentry );
 646                 $qossettings{'CLASS'} = $classline[1];
 647                 $qossettings{'TOS'} = abs $classline[7] * 2;
 648                 if ($qossettings{'TOS'} ne "0") {
 649                         print "\tiptables -t mangle -A QOS-TOS -m mark --mark $qossettings{'CLASS'} -j TOS --set-tos $qossettings{'TOS'}\n";
 650                         print "\tiptables -t mangle -A QOS-TOS -m mark --mark $qossettings{'CLASS'} -j RETURN\n";
 651                 }
 652         }
 653         foreach $subclassentry (sort @subclasses)
 654         {
 655                 @subclassline = split( /\;/, $subclassentry );
 656                 $qossettings{'SUBCLASS'} = $subclassline[1];
 657                 $qossettings{'TOS'} = $subclassline[8];
 658                 $qossettings{'TOS'} = abs $qossettings{'TOS'} * 2;
 659                 if ($qossettings{'TOS'} ne "0") {
 660                         print "\tiptables -t mangle -A QOS-TOS -m mark --mark $qossettings{'SUBCLASS'} -j TOS --set-tos $qossettings{'TOS'}\n";
 661                         print "\tiptables -t mangle -A QOS-TOS -m mark --mark $qossettings{'SUBCLASS'} -j RETURN\n";
 662                 }
 663         }
 664
 665 print <<END
 666
 667         ## STARTING COLLECTOR
 668         ( sleep 10 && /usr/local/bin/qosd $qossettings{'RED_DEV'} >/dev/null 2>&1) &
 669         ( sleep 10 && /usr/local/bin/qosd $qossettings{'IMQ_DEV'} >/dev/null 2>&1) &
 670
 671         for i in \$(ls \$RRDLOG/class_*.rrd); do
 672                 rrdtool update \$i \$(date +%s):
 673         done
 674
 675         echo "Quality of Service was successfully started!"
 676         exit 0
 677   ;;
 678   clear|stop)
 679         ### RESET EVERYTHING TO A KNOWN STATE
 680         killall qosd >/dev/null 2>&1
 681         (sleep 3 && killall -9 qosd &>/dev/null) &
 682         # DELETE QDISCS
 683         tc qdisc del dev $qossettings{'RED_DEV'} root >/dev/null 2>&1
 684         tc qdisc del dev $qossettings{'IMQ_DEV'} root >/dev/null 2>&1
 685         # STOP IMQ-DEVICE
 686         ip link set $qossettings{'IMQ_DEV'} down >/dev/null 2>&1
 687         iptables -t mangle --delete POSTROUTING -m mark ! --mark 0 -o ! $qossettings{'RED_DEV'} -j IMQ --todev 0 >/dev/null 2>&1
 688         iptables -t mangle --delete PREROUTING -i $qossettings{'RED_DEV'} -j IMQ --todev 0  >/dev/null 2>&1
 689         # rmmod imq # this crash on 2.6.25.xx
 690         # REMOVE & FLUSH CHAINS
 691         iptables -t mangle --delete POSTROUTING -o $qossettings{'RED_DEV'} -j QOS-OUT >/dev/null 2>&1
 692         iptables -t mangle --delete POSTROUTING -o $qossettings{'RED_DEV'} -j QOS-TOS >/dev/null 2>&1
 693         iptables -t mangle --flush  QOS-OUT >/dev/null 2>&1
 694         iptables -t mangle --delete-chain QOS-OUT >/dev/null 2>&1
 695         iptables -t mangle --delete FORWARD -i $qossettings{'RED_DEV'} -j QOS-INC >/dev/null 2>&1
 696         iptables -t mangle --delete FORWARD -i $qossettings{'RED_DEV'} -j QOS-TOS >/dev/null 2>&1
 697         iptables -t mangle --delete PREROUTING -i $qossettings{'RED_DEV'} -j QOS-INC >/dev/null 2>&1
 698         iptables -t mangle --delete PREROUTING -i $qossettings{'RED_DEV'} -j QOS-TOS >/dev/null 2>&1
 699         iptables -t mangle --flush  QOS-INC >/dev/null 2>&1
 700         iptables -t mangle --delete-chain QOS-INC >/dev/null 2>&1
 701         iptables -t mangle --flush  QOS-TOS >/dev/null 2>&1
 702         iptables -t mangle --delete-chain QOS-TOS >/dev/null 2>&1
 703         rmmod sch_htb >/dev/null 2>&1
 704
 705         for i in \$(ls \$RRDLOG/class_*.rrd); do
 706                 rrdtool update \$i \$(date +%s):
 707         done
 708
 709         echo "Quality of Service was successfully cleared!"
 710   ;;
 711   gen|generate)
 712         echo -n "Generateing the QoS-Scripts..."
 713         /usr/bin/perl /var/ipfire/qos/bin/makeqosscripts.pl > /var/ipfire/qos/bin/qos.sh
 714         echo ".Done!"
 715         exit 0
 716         ;;
 717   restart)
 718         ### FIRST CLEAR EVERYTHING
 719         \$0 clear
 720
 721         ### THEN START
 722         \$0 start
 723         ;;
 724 esac
 725 ### EOF
 726 END
 727 ;
 728
 729 ############################################################################################################################
 730 ############################################################################################################################