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