config/qos/makeqosscripts.pl

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