optimize bridge chains
[pve-firewall.git] / PVE / Firewall.pm
CommitLineData
b6360c3f
DM
1package PVE::Firewall;
2
3use warnings;
4use strict;
5use Data::Dumper;
09d5f68e 6use Digest::SHA;
f789653a 7use PVE::Tools;
b6360c3f 8use PVE::QemuServer;
5e1267a5
DM
9use File::Path;
10use IO::File;
ecbea048 11use Net::IP;
06320eb0 12use PVE::Tools qw(run_command lock_file);
ecbea048 13
5e1267a5 14use Data::Dumper;
b6360c3f 15
06320eb0
DM
16my $pve_fw_lock_filename = "/var/lock/pvefw.lck";
17
961b4928 18# todo: define more MACROS
857f62c8 19# imported/converted from: /usr/share/shorewall/macro.*
961b4928 20my $pve_fw_macros = {
857f62c8
DM
21 'Amanda' => [
22 { action => 'PARAM', proto => 'udp', dport => '10080' },
23 { action => 'PARAM', proto => 'tcp', dport => '10080' },
24 ],
25 'Auth' => [
26 { action => 'PARAM', proto => 'tcp', dport => '113' },
27 ],
28 'BGP' => [
29 { action => 'PARAM', proto => 'tcp', dport => '179' },
30 ],
31 'BitTorrent' => [
961b4928 32 { action => 'PARAM', proto => 'tcp', dport => '6881:6889' },
857f62c8
DM
33 { action => 'PARAM', proto => 'udp', dport => '6881' },
34 ],
35 'BitTorrent32' => [
36 { action => 'PARAM', proto => 'tcp', dport => '6881:6999' },
37 { action => 'PARAM', proto => 'udp', dport => '6881' },
38 ],
39 'CVS' => [
40 { action => 'PARAM', proto => 'tcp', dport => '2401' },
41 ],
42 'Citrix' => [
43 { action => 'PARAM', proto => 'tcp', dport => '1494' },
44 { action => 'PARAM', proto => 'udp', dport => '1604' },
45 { action => 'PARAM', proto => 'tcp', dport => '2598' },
46 ],
47 'DAAP' => [
48 { action => 'PARAM', proto => 'tcp', dport => '3689' },
49 { action => 'PARAM', proto => 'udp', dport => '3689' },
50 ],
51 'DCC' => [
52 { action => 'PARAM', proto => 'tcp', dport => '6277' },
53 ],
54 'DHCPfwd' => [
55 { action => 'PARAM', proto => 'udp', dport => '67:68', sport => '67:68' },
56 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '67:68', sport => '67:68' },
57 ],
58 'DNS' => [
59 { action => 'PARAM', proto => 'udp', dport => '53' },
60 { action => 'PARAM', proto => 'tcp', dport => '53' },
61 ],
62 'Distcc' => [
63 { action => 'PARAM', proto => 'tcp', dport => '3632' },
64 ],
65 'Edonkey' => [
66 { action => 'PARAM', proto => 'tcp', dport => '4662' },
67 { action => 'PARAM', proto => 'udp', dport => '4665' },
961b4928 68 ],
857f62c8
DM
69 'FTP' => [
70 { action => 'PARAM', proto => 'tcp', dport => '21' },
71 ],
72 'Finger' => [
73 { action => 'PARAM', proto => 'tcp', dport => '79' },
74 ],
75 'GNUnet' => [
76 { action => 'PARAM', proto => 'tcp', dport => '2086' },
77 { action => 'PARAM', proto => 'udp', dport => '2086' },
78 { action => 'PARAM', proto => 'tcp', dport => '1080' },
79 { action => 'PARAM', proto => 'udp', dport => '1080' },
80 ],
81 'GRE' => [
82 { action => 'PARAM', proto => '47' },
83 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => '47' },
84 ],
85 'Git' => [
86 { action => 'PARAM', proto => 'tcp', dport => '9418' },
87 ],
88 'Gnutella' => [
89 { action => 'PARAM', proto => 'tcp', dport => '6346' },
90 { action => 'PARAM', proto => 'udp', dport => '6346' },
91 ],
92 'HKP' => [
93 { action => 'PARAM', proto => 'tcp', dport => '11371' },
94 ],
95 'HTTP' => [
961b4928
DM
96 { action => 'PARAM', proto => 'tcp', dport => '80' },
97 ],
857f62c8
DM
98 'HTTPS' => [
99 { action => 'PARAM', proto => 'tcp', dport => '443' },
100 ],
101 'ICPV2' => [
102 { action => 'PARAM', proto => 'udp', dport => '3130' },
103 ],
104 'ICQ' => [
105 { action => 'PARAM', proto => 'tcp', dport => '5190' },
106 ],
107 'IMAP' => [
108 { action => 'PARAM', proto => 'tcp', dport => '143' },
109 ],
110 'IMAPS' => [
111 { action => 'PARAM', proto => 'tcp', dport => '993' },
112 ],
113 'IPIP' => [
114 { action => 'PARAM', proto => '94' },
115 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => '94' },
116 ],
117 'IPP' => [
118 { action => 'PARAM', proto => 'tcp', dport => '631' },
119 ],
120 'IPPbrd' => [
121 { action => 'PARAM', proto => 'udp', dport => '631' },
122 ],
123 'IPPserver' => [
124 { action => 'PARAM', source => 'SOURCE', dest => 'DEST', proto => 'tcp', dport => '631' },
125 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '631' },
126 ],
127 'IPsec' => [
128 { action => 'PARAM', proto => 'udp', dport => '500', sport => '500' },
129 { action => 'PARAM', proto => '50' },
130 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '500', sport => '500' },
131 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => '50' },
132 ],
133 'IPsecah' => [
134 { action => 'PARAM', proto => 'udp', dport => '500', sport => '500' },
135 { action => 'PARAM', proto => '51' },
136 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '500', sport => '500' },
137 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => '51' },
138 ],
139 'IPsecnat' => [
140 { action => 'PARAM', proto => 'udp', dport => '500' },
141 { action => 'PARAM', proto => 'udp', dport => '4500' },
142 { action => 'PARAM', proto => '50' },
143 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '500' },
144 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '4500' },
145 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => '50' },
146 ],
147 'IRC' => [
148 { action => 'PARAM', proto => 'tcp', dport => '6667' },
149 ],
150 'JabberPlain' => [
151 { action => 'PARAM', proto => 'tcp', dport => '5222' },
152 ],
153 'JabberSecure' => [
154 { action => 'PARAM', proto => 'tcp', dport => '5223' },
155 ],
156 'Jabberd' => [
157 { action => 'PARAM', proto => 'tcp', dport => '5269' },
158 ],
159 'Jetdirect' => [
160 { action => 'PARAM', proto => 'tcp', dport => '9100' },
161 ],
162 'L2TP' => [
163 { action => 'PARAM', proto => 'udp', dport => '1701' },
164 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '1701' },
165 ],
166 'LDAP' => [
167 { action => 'PARAM', proto => 'tcp', dport => '389' },
168 ],
169 'LDAPS' => [
170 { action => 'PARAM', proto => 'tcp', dport => '636' },
171 ],
172 'MSNP' => [
173 { action => 'PARAM', proto => 'tcp', dport => '1863' },
174 ],
175 'MSSQL' => [
176 { action => 'PARAM', proto => 'tcp', dport => '1433' },
177 ],
178 'Mail' => [
179 { action => 'PARAM', proto => 'tcp', dport => '25' },
180 { action => 'PARAM', proto => 'tcp', dport => '465' },
181 { action => 'PARAM', proto => 'tcp', dport => '587' },
182 ],
183 'Munin' => [
184 { action => 'PARAM', proto => 'tcp', dport => '4949' },
185 ],
186 'MySQL' => [
187 { action => 'PARAM', proto => 'tcp', dport => '3306' },
188 ],
189 'NNTP' => [
190 { action => 'PARAM', proto => 'tcp', dport => '119' },
191 ],
192 'NNTPS' => [
193 { action => 'PARAM', proto => 'tcp', dport => '563' },
194 ],
195 'NTP' => [
196 { action => 'PARAM', proto => 'udp', dport => '123' },
197 ],
198 'NTPbi' => [
199 { action => 'PARAM', proto => 'udp', dport => '123' },
200 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '123' },
201 ],
202 'NTPbrd' => [
203 { action => 'PARAM', proto => 'udp', dport => '123' },
204 { action => 'PARAM', proto => 'udp', dport => '1024:65535', sport => '123' },
205 ],
206 'OSPF' => [
207 { action => 'PARAM', proto => '89' },
208 ],
209 'OpenVPN' => [
210 { action => 'PARAM', proto => 'udp', dport => '1194' },
211 ],
212 'PCA' => [
213 { action => 'PARAM', proto => 'udp', dport => '5632' },
214 { action => 'PARAM', proto => 'tcp', dport => '5631' },
215 ],
216 'POP3' => [
217 { action => 'PARAM', proto => 'tcp', dport => '110' },
218 ],
219 'POP3S' => [
220 { action => 'PARAM', proto => 'tcp', dport => '995' },
221 ],
222 'PPtP' => [
223 { action => 'PARAM', proto => '47' },
224 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => '47' },
225 { action => 'PARAM', proto => 'tcp', dport => '1723' },
226 ],
227 'Ping' => [
228 { action => 'PARAM', proto => 'icmp', dport => '8' },
229 ],
230 'PostgreSQL' => [
231 { action => 'PARAM', proto => 'tcp', dport => '5432' },
232 ],
233 'Printer' => [
234 { action => 'PARAM', proto => 'tcp', dport => '515' },
235 ],
236 'RDP' => [
237 { action => 'PARAM', proto => 'tcp', dport => '3389' },
238 ],
239 'RIPbi' => [
240 { action => 'PARAM', proto => 'udp', dport => '520' },
241 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '520' },
242 ],
243 'RNDC' => [
244 { action => 'PARAM', proto => 'tcp', dport => '953' },
245 ],
246 'Razor' => [
247 { action => 'ACCEPT', proto => 'tcp', dport => '2703' },
248 ],
249 'Rdate' => [
250 { action => 'PARAM', proto => 'tcp', dport => '37' },
251 ],
252 'Rsync' => [
253 { action => 'PARAM', proto => 'tcp', dport => '873' },
254 ],
255 'SANE' => [
256 { action => 'PARAM', proto => 'tcp', dport => '6566' },
257 ],
258 'SMB' => [
259 { action => 'PARAM', proto => 'udp', dport => '135,445' },
260 { action => 'PARAM', proto => 'udp', dport => '137:139' },
261 { action => 'PARAM', proto => 'udp', dport => '1024:65535', sport => '137' },
262 { action => 'PARAM', proto => 'tcp', dport => '135,139,445' },
263 ],
264 'SMBBI' => [
265 { action => 'PARAM', proto => 'udp', dport => '135,445' },
266 { action => 'PARAM', proto => 'udp', dport => '137:139' },
267 { action => 'PARAM', proto => 'udp', dport => '1024:65535', sport => '137' },
268 { action => 'PARAM', proto => 'tcp', dport => '135,139,445' },
269 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '135,445' },
270 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '137:139' },
271 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'udp', dport => '1024:65535', sport => '137' },
272 { action => 'PARAM', source => 'DEST', dest => 'SOURCE', proto => 'tcp', dport => '135,139,445' },
273 ],
274 'SMBswat' => [
275 { action => 'PARAM', proto => 'tcp', dport => '901' },
276 ],
277 'SMTP' => [
278 { action => 'PARAM', proto => 'tcp', dport => '25' },
279 ],
280 'SMTPS' => [
281 { action => 'PARAM', proto => 'tcp', dport => '465' },
282 ],
283 'SNMP' => [
284 { action => 'PARAM', proto => 'udp', dport => '161:162' },
285 { action => 'PARAM', proto => 'tcp', dport => '161' },
286 ],
287 'SPAMD' => [
288 { action => 'PARAM', proto => 'tcp', dport => '783' },
289 ],
290 'SSH' => [
291 { action => 'PARAM', proto => 'tcp', dport => '22' },
292 ],
293 'SVN' => [
294 { action => 'PARAM', proto => 'tcp', dport => '3690' },
295 ],
296 'SixXS' => [
297 { action => 'PARAM', proto => 'tcp', dport => '3874' },
298 { action => 'PARAM', proto => 'udp', dport => '3740' },
299 { action => 'PARAM', proto => '41' },
300 { action => 'PARAM', proto => 'udp', dport => '5072,8374' },
301 ],
302 'Squid' => [
303 { action => 'PARAM', proto => 'tcp', dport => '3128' },
304 ],
305 'Submission' => [
306 { action => 'PARAM', proto => 'tcp', dport => '587' },
307 ],
308 'Syslog' => [
309 { action => 'PARAM', proto => 'udp', dport => '514' },
310 { action => 'PARAM', proto => 'tcp', dport => '514' },
311 ],
312 'TFTP' => [
313 { action => 'PARAM', proto => 'udp', dport => '69' },
314 ],
315 'Telnet' => [
316 { action => 'PARAM', proto => 'tcp', dport => '23' },
317 ],
318 'Telnets' => [
319 { action => 'PARAM', proto => 'tcp', dport => '992' },
320 ],
321 'Time' => [
322 { action => 'PARAM', proto => 'tcp', dport => '37' },
323 ],
324 'Trcrt' => [
325 { action => 'PARAM', proto => 'udp', dport => '33434:33524' },
326 { action => 'PARAM', proto => 'icmp', dport => '8' },
327 ],
328 'VNC' => [
329 { action => 'PARAM', proto => 'tcp', dport => '5900:5909' },
330 ],
331 'VNCL' => [
332 { action => 'PARAM', proto => 'tcp', dport => '5500' },
333 ],
334 'Web' => [
335 { action => 'PARAM', proto => 'tcp', dport => '80' },
961b4928
DM
336 { action => 'PARAM', proto => 'tcp', dport => '443' },
337 ],
857f62c8
DM
338 'Webcache' => [
339 { action => 'PARAM', proto => 'tcp', dport => '8080' },
340 ],
341 'Webmin' => [
342 { action => 'PARAM', proto => 'tcp', dport => '10000' },
343 ],
344 'Whois' => [
345 { action => 'PARAM', proto => 'tcp', dport => '43' },
346 ],
961b4928
DM
347};
348
349my $pve_fw_parsed_macros;
350my $pve_fw_preferred_macro_names = {};
3a616aa0 351
a332200b 352sub get_firewall_macros {
9aab3127 353
961b4928
DM
354 return $pve_fw_parsed_macros if $pve_fw_parsed_macros;
355
356 $pve_fw_parsed_macros = {};
9aab3127 357
961b4928
DM
358 foreach my $k (keys %$pve_fw_macros) {
359 my $name = lc($k);
a332200b 360
961b4928
DM
361 my $macro = $pve_fw_macros->{$k};
362 $pve_fw_preferred_macro_names->{$name} = $k;
363 $pve_fw_parsed_macros->{$name} = $macro;
364 }
a332200b 365
961b4928 366 return $pve_fw_parsed_macros;
9aab3127
DM
367}
368
fcba0beb
DM
369my $etc_services;
370
371sub get_etc_services {
372
373 return $etc_services if $etc_services;
374
375 my $filename = "/etc/services";
376
377 my $fh = IO::File->new($filename, O_RDONLY);
378 if (!$fh) {
379 warn "unable to read '$filename' - $!\n";
380 return {};
381 }
382
383 my $services = {};
384
385 while (my $line = <$fh>) {
386 chomp ($line);
387 next if $line =~m/^#/;
388 next if ($line =~m/^\s*$/);
389
390 if ($line =~ m!^(\S+)\s+(\S+)/(tcp|udp).*$!) {
391 $services->{byid}->{$2}->{name} = $1;
35b66e9d 392 $services->{byid}->{$2}->{port} = $2;
fcba0beb
DM
393 $services->{byid}->{$2}->{$3} = 1;
394 $services->{byname}->{$1} = $services->{byid}->{$2};
395 }
396 }
397
398 close($fh);
399
400 $etc_services = $services;
401
3a616aa0 402
fcba0beb
DM
403 return $etc_services;
404}
405
406my $etc_protocols;
407
408sub get_etc_protocols {
409 return $etc_protocols if $etc_protocols;
410
411 my $filename = "/etc/protocols";
412
413 my $fh = IO::File->new($filename, O_RDONLY);
414 if (!$fh) {
415 warn "unable to read '$filename' - $!\n";
416 return {};
417 }
418
419 my $protocols = {};
420
421 while (my $line = <$fh>) {
422 chomp ($line);
423 next if $line =~m/^#/;
424 next if ($line =~m/^\s*$/);
425
426 if ($line =~ m!^(\S+)\s+(\d+)\s+.*$!) {
427 $protocols->{byid}->{$2}->{name} = $1;
428 $protocols->{byname}->{$1} = $protocols->{byid}->{$2};
429 }
430 }
431
432 close($fh);
433
434 $etc_protocols = $protocols;
435
436 return $etc_protocols;
437}
438
ecbea048
DM
439sub parse_address_list {
440 my ($str) = @_;
441
d6de1dc2 442 my $nbaor = 0;
ecbea048
DM
443 foreach my $aor (split(/,/, $str)) {
444 if (!Net::IP->new($aor)) {
445 my $err = Net::IP::Error();
446 die "invalid IP address: $err\n";
d6de1dc2
AD
447 }else{
448 $nbaor++;
ecbea048
DM
449 }
450 }
d6de1dc2 451 return $nbaor;
ecbea048 452}
dddd9413 453
fcba0beb
DM
454sub parse_port_name_number_or_range {
455 my ($str) = @_;
456
457 my $services = PVE::Firewall::get_etc_services();
4cdbb3b7 458 my $nbports = 0;
fcba0beb 459 foreach my $item (split(/,/, $str)) {
4cdbb3b7 460 my $portlist = "";
35b66e9d 461 my $oldpon = undef;
fcba0beb 462 foreach my $pon (split(':', $item, 2)) {
35b66e9d 463 $pon = $services->{byname}->{$pon}->{port} if $services->{byname}->{$pon}->{port};
4cdbb3b7 464 if ($pon =~ m/^\d+$/){
857f62c8 465 die "invalid port '$pon'\n" if $pon < 0 && $pon > 65535;
35b66e9d
AD
466 die "port '$pon' must be bigger than port '$oldpon' \n" if $oldpon && ($pon < $oldpon);
467 $oldpon = $pon;
4cdbb3b7
AD
468 }else{
469 die "invalid port $services->{byname}->{$pon}\n" if !$services->{byname}->{$pon};
470 }
471 $nbports++;
fcba0beb
DM
472 }
473 }
474
4cdbb3b7 475 return ($nbports);
fcba0beb
DM
476}
477
780bcc0f
DM
478my $bridge_firewall_enabled = 0;
479
480sub enable_bridge_firewall {
481
482 return if $bridge_firewall_enabled; # only once
483
484 system("echo 1 > /proc/sys/net/bridge/bridge-nf-call-iptables");
485 system("echo 1 > /proc/sys/net/bridge/bridge-nf-call-ip6tables");
486
487 $bridge_firewall_enabled = 1;
488}
489
8cebfa6f 490my $rule_format = "%-15s %-30s %-30s %-15s %-15s %-15s\n";
dddd9413 491
3a616aa0
AD
492sub iptables {
493 my ($cmd) = @_;
494
495 run_command("/sbin/iptables $cmd", outfunc => sub {}, errfunc => sub {});
496}
497
b16e818e
DM
498sub iptables_restore_cmdlist {
499 my ($cmdlist) = @_;
3a616aa0 500
3fa83edf 501 run_command("/sbin/iptables-restore -n", input => $cmdlist);
3a616aa0
AD
502}
503
de2a57cd
DM
504sub iptables_get_chains {
505
506 my $res = {};
507
508 # check what chains we want to track
509 my $is_pvefw_chain = sub {
510 my $name = shift;
511
dec84fcd
DM
512 return 1 if $name =~ m/^PVEFW-\S+$/;
513
de2a57cd 514 return 1 if $name =~ m/^tap\d+i\d+-(:?IN|OUT)$/;
b16e818e 515 return 1 if $name =~ m/^vmbr\d+-(:?IN|OUT)$/;
f50656c2 516 return 1 if $name =~ m/^GROUP-(:?[^\s\-]+)-(:?IN|OUT)$/;
de2a57cd
DM
517
518 return undef;
519 };
520
521 my $table = '';
522
de2a57cd
DM
523 my $parser = sub {
524 my $line = shift;
525
526 return if $line =~ m/^#/;
527 return if $line =~ m/^\s*$/;
528
529 if ($line =~ m/^\*(\S+)$/) {
530 $table = $1;
531 return;
532 }
533
534 return if $table ne 'filter';
535
536 if ($line =~ m/^:(\S+)\s/) {
537 my $chain = $1;
538 return if !&$is_pvefw_chain($chain);
3fa83edf 539 $res->{$chain} = "unknown";
09d5f68e 540 } elsif ($line =~ m/^-A\s+(\S+)\s.*--comment\s+\"PVESIG:(\S+)\"/) {
3fa83edf 541 my ($chain, $sig) = ($1, $2);
de2a57cd 542 return if !&$is_pvefw_chain($chain);
3fa83edf 543 $res->{$chain} = $sig;
de2a57cd
DM
544 } else {
545 # simply ignore the rest
546 return;
547 }
548 };
549
550 run_command("/sbin/iptables-save", outfunc => $parser);
551
de2a57cd
DM
552 return $res;
553}
554
3a616aa0
AD
555sub iptables_chain_exist {
556 my ($chain) = @_;
557
558 eval{
559 iptables("-n --list $chain");
560 };
561 return undef if $@;
562
563 return 1;
564}
565
566sub iptables_rule_exist {
567 my ($rule) = @_;
568
569 eval{
570 iptables("-C $rule");
571 };
572 return undef if $@;
573
574 return 1;
575}
576
3fa83edf 577sub ruleset_generate_rule {
4e6112f9 578 my ($ruleset, $chain, $rule, $goto) = @_;
3a616aa0 579
3fa83edf 580 my $cmd = '';
3a616aa0 581
d6de1dc2 582 $cmd .= " -m iprange --src-range" if $rule->{nbsource} && $rule->{nbsource} > 1;
3a616aa0 583 $cmd .= " -s $rule->{source}" if $rule->{source};
d6de1dc2 584 $cmd .= " -m iprange --dst-range" if $rule->{nbdest} && $rule->{nbdest} > 1;
8c32a215 585 $cmd .= " -d $rule->{dest}" if $rule->{dest};
3a616aa0 586 $cmd .= " -p $rule->{proto}" if $rule->{proto};
4cdbb3b7 587 $cmd .= " --match multiport" if $rule->{nbdport} && $rule->{nbdport} > 1;
3a616aa0 588 $cmd .= " --dport $rule->{dport}" if $rule->{dport};
4cdbb3b7 589 $cmd .= " --match multiport" if $rule->{nbsport} && $rule->{nbsport} > 1;
3a616aa0 590 $cmd .= " --sport $rule->{sport}" if $rule->{sport};
4e6112f9
DM
591
592 if (my $action = $rule->{action}) {
593 $goto = 1 if !defined($goto) && $action eq 'PVEFW-SET-ACCEPT-MARK';
594 $cmd .= $goto ? " -g $action" : " -j $action";
595 };
3a616aa0 596
3fa83edf
DM
597 ruleset_addrule($ruleset, $chain, $cmd) if $cmd;
598}
599
600sub ruleset_create_chain {
601 my ($ruleset, $chain) = @_;
3a616aa0 602
d050c724
DM
603 die "Invalid chain name '$chain' (28 char max)\n" if length($chain) > 28;
604
3fa83edf
DM
605 die "chain '$chain' already exists\n" if $ruleset->{$chain};
606
607 $ruleset->{$chain} = [];
3a616aa0
AD
608}
609
3fa83edf
DM
610sub ruleset_chain_exist {
611 my ($ruleset, $chain) = @_;
3a616aa0 612
3fa83edf
DM
613 return $ruleset->{$chain} ? 1 : undef;
614}
3a616aa0 615
3fa83edf
DM
616sub ruleset_addrule {
617 my ($ruleset, $chain, $rule) = @_;
3a616aa0 618
3fa83edf 619 die "no such chain '$chain'\n" if !$ruleset->{$chain};
3a616aa0 620
3fa83edf
DM
621 push @{$ruleset->{$chain}}, "-A $chain $rule";
622}
3a616aa0 623
3fa83edf
DM
624sub ruleset_insertrule {
625 my ($ruleset, $chain, $rule) = @_;
3a616aa0 626
3fa83edf 627 die "no such chain '$chain'\n" if !$ruleset->{$chain};
3a616aa0 628
3fa83edf
DM
629 unshift @{$ruleset->{$chain}}, "-A $chain $rule";
630}
3a616aa0 631
3fa83edf
DM
632sub generate_bridge_chains {
633 my ($ruleset, $bridge) = @_;
3a616aa0 634
dec84fcd
DM
635 if (!ruleset_chain_exist($ruleset, "PVEFW-FORWARD")){
636 ruleset_create_chain($ruleset, "PVEFW-FORWARD");
7fab95bc 637 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-m conntrack --ctstate RELATED,ESTABLISHED -j ACCEPT");
3a616aa0
AD
638 }
639
9fcad984
AD
640 if (!ruleset_chain_exist($ruleset, "$bridge")) {
641 ruleset_create_chain($ruleset, "$bridge");
642 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-o $bridge -m physdev --physdev-is-bridged -j $bridge");
643 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-i $bridge -m physdev --physdev-is-bridged -j $bridge");
644 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-o $bridge -j DROP"); # disable interbridge routing
645 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-i $bridge -j DROP"); # disable interbridge routing
3fa83edf 646 }
3a616aa0 647
3fa83edf
DM
648 if (!ruleset_chain_exist($ruleset, "$bridge-OUT")) {
649 ruleset_create_chain($ruleset, "$bridge-OUT");
9fcad984
AD
650 ruleset_addrule($ruleset, "$bridge", "-m physdev --physdev-is-bridged --physdev-is-in -j $bridge-OUT");
651 }
652
653 if (!ruleset_chain_exist($ruleset, "$bridge-IN")) {
654 ruleset_create_chain($ruleset, "$bridge-IN");
655 ruleset_addrule($ruleset, "$bridge", "-m physdev --physdev-is-bridged --physdev-is-out -j $bridge-IN");
3fa83edf
DM
656 }
657}
3a616aa0
AD
658
659sub generate_tap_rules_direction {
51bae274 660 my ($ruleset, $group_rules, $iface, $netid, $macaddr, $rules, $bridge, $direction) = @_;
3a616aa0
AD
661
662 my $tapchain = "$iface-$direction";
663
3fa83edf 664 ruleset_create_chain($ruleset, $tapchain);
3a616aa0 665
7fab95bc
DM
666 ruleset_addrule($ruleset, $tapchain, "-m conntrack --ctstate INVALID -j DROP");
667 ruleset_addrule($ruleset, $tapchain, "-m conntrack --ctstate RELATED,ESTABLISHED -j ACCEPT");
3a616aa0 668
c29f55c9
DM
669 if ($direction eq 'OUT' && defined($macaddr)) {
670 ruleset_addrule($ruleset, $tapchain, "-m mac ! --mac-source $macaddr -j DROP");
671 }
672
3fa83edf 673 if ($rules) {
3a616aa0
AD
674 foreach my $rule (@$rules) {
675 next if $rule->{iface} && $rule->{iface} ne $netid;
bfbfa8b6 676 # we go to $bridge-IN if accept in out rules
3a616aa0 677 if($rule->{action} =~ m/^(GROUP-(\S+))$/){
3fa83edf
DM
678 $rule->{action} .= "-$direction";
679 # generate empty group rule if don't exist
680 if(!ruleset_chain_exist($ruleset, $rule->{action})){
51bae274 681 generate_group_rules($ruleset, $group_rules, $2);
3fa83edf 682 }
4e6112f9 683 ruleset_generate_rule($ruleset, $tapchain, $rule);
31ff3ef6
DM
684 ruleset_addrule($ruleset, $tapchain, "-m mark --mark 1 -g $bridge-IN")
685 if $direction eq 'OUT';
4e6112f9 686 } else {
4e6112f9
DM
687 $rule->{action} = "$bridge-IN" if $rule->{action} eq 'ACCEPT' && $direction eq 'OUT';
688 ruleset_generate_rule($ruleset, $tapchain, $rule);
3a616aa0 689 }
4e6112f9 690 }
3a616aa0
AD
691 }
692
3fa83edf
DM
693 ruleset_addrule($ruleset, $tapchain, "-j LOG --log-prefix \"$tapchain-dropped: \" --log-level 4");
694 ruleset_addrule($ruleset, $tapchain, "-j DROP");
3a616aa0 695
3fa83edf
DM
696 # plug the tap chain to bridge chain
697 my $physdevdirection = $direction eq 'IN' ? "out" : "in";
698 my $rule = "-m physdev --physdev-$physdevdirection $iface --physdev-is-bridged -j $tapchain";
699 ruleset_insertrule($ruleset, "$bridge-$direction", $rule);
3a616aa0 700
3fa83edf
DM
701 if ($direction eq 'OUT'){
702 # add tap->host rules
703 my $rule = "-m physdev --physdev-$physdevdirection $iface -j $tapchain";
dec84fcd 704 ruleset_addrule($ruleset, "PVEFW-INPUT", $rule);
3a616aa0
AD
705 }
706}
707
0bd5f137 708sub enablehostfw {
51bae274 709 my ($ruleset, $rules, $group_rules) = @_;
0bd5f137 710
51bae274 711 # fixme: allow security groups
0bd5f137 712
3fa83edf 713 # host inbound firewall
dec84fcd
DM
714 my $chain = "PVEFW-HOST-IN";
715 ruleset_create_chain($ruleset, $chain);
0bd5f137 716
7fab95bc
DM
717 ruleset_addrule($ruleset, $chain, "-m conntrack --ctstate INVALID -j DROP");
718 ruleset_addrule($ruleset, $chain, "-m conntrack --ctstate RELATED,ESTABLISHED -j ACCEPT");
dec84fcd
DM
719 ruleset_addrule($ruleset, $chain, "-i lo -j ACCEPT");
720 ruleset_addrule($ruleset, $chain, "-m addrtype --dst-type MULTICAST -j ACCEPT");
7fab95bc 721 ruleset_addrule($ruleset, $chain, "-p udp -m conntrack --ctstate NEW -m multiport --dports 5404,5405 -j ACCEPT");
dec84fcd 722 ruleset_addrule($ruleset, $chain, "-p udp -m udp --dport 9000 -j ACCEPT"); #corosync
0bd5f137 723
3fa83edf
DM
724 if ($rules->{in}) {
725 foreach my $rule (@{$rules->{in}}) {
726 # we use RETURN because we need to check also tap rules
727 $rule->{action} = 'RETURN' if $rule->{action} eq 'ACCEPT';
dec84fcd 728 ruleset_generate_rule($ruleset, $chain, $rule);
3fa83edf 729 }
0bd5f137
AD
730 }
731
dec84fcd
DM
732 ruleset_addrule($ruleset, $chain, "-j LOG --log-prefix \"kvmhost-IN dropped: \" --log-level 4");
733 ruleset_addrule($ruleset, $chain, "-j DROP");
0bd5f137 734
3fa83edf 735 # host outbound firewall
aadd745e 736 $chain = "PVEFW-HOST-OUT";
dec84fcd
DM
737 ruleset_create_chain($ruleset, $chain);
738
7fab95bc
DM
739 ruleset_addrule($ruleset, $chain, "-m conntrack --ctstate INVALID -j DROP");
740 ruleset_addrule($ruleset, $chain, "-m conntrack --ctstate RELATED,ESTABLISHED -j ACCEPT");
dec84fcd
DM
741 ruleset_addrule($ruleset, $chain, "-o lo -j ACCEPT");
742 ruleset_addrule($ruleset, $chain, "-m addrtype --dst-type MULTICAST -j ACCEPT");
7fab95bc 743 ruleset_addrule($ruleset, $chain, "-p udp -m conntrack --ctstate NEW -m multiport --dports 5404,5405 -j ACCEPT");
dec84fcd 744 ruleset_addrule($ruleset, $chain, "-p udp -m udp --dport 9000 -j ACCEPT"); #corosync
0bd5f137 745
3fa83edf
DM
746 if ($rules->{out}) {
747 foreach my $rule (@{$rules->{out}}) {
748 # we use RETURN because we need to check also tap rules
749 $rule->{action} = 'RETURN' if $rule->{action} eq 'ACCEPT';
dec84fcd 750 ruleset_generate_rule($ruleset, $chain, $rule);
3fa83edf 751 }
0bd5f137
AD
752 }
753
dec84fcd
DM
754 ruleset_addrule($ruleset, $chain, "-j LOG --log-prefix \"kvmhost-OUT dropped: \" --log-level 4");
755 ruleset_addrule($ruleset, $chain, "-j DROP");
9d31b418 756
dec84fcd
DM
757 ruleset_addrule($ruleset, "PVEFW-OUTPUT", "-j PVEFW-HOST-OUT");
758 ruleset_addrule($ruleset, "PVEFW-INPUT", "-j PVEFW-HOST-IN");
9d31b418
AD
759}
760
761sub generate_group_rules {
51bae274 762 my ($ruleset, $group_rules, $group) = @_;
9d31b418 763
51bae274 764 my $rules = $group_rules->{$group};
9d31b418 765
51bae274
DM
766 die "no such security group '$group'\n" if !$rules;
767
3fa83edf 768 my $chain = "GROUP-${group}-IN";
9d31b418 769
3fa83edf 770 ruleset_create_chain($ruleset, $chain);
9d31b418 771
3fa83edf
DM
772 if ($rules->{in}) {
773 foreach my $rule (@{$rules->{in}}) {
774 ruleset_generate_rule($ruleset, $chain, $rule);
9d31b418
AD
775 }
776 }
777
3fa83edf 778 $chain = "GROUP-${group}-OUT";
9d31b418 779
3fa83edf 780 ruleset_create_chain($ruleset, $chain);
4e6112f9 781 ruleset_addrule($ruleset, $chain, "-j MARK --set-mark 0"); # clear mark
9d31b418 782
3fa83edf
DM
783 if ($rules->{out}) {
784 foreach my $rule (@{$rules->{out}}) {
4e6112f9
DM
785 # we go the PVEFW-SET-ACCEPT-MARK Instead of ACCEPT) because we need to
786 # check also other tap rules (and group rules can be set on any bridge,
787 # so we can't go to VMBRXX-IN)
788 $rule->{action} = 'PVEFW-SET-ACCEPT-MARK' if $rule->{action} eq 'ACCEPT';
6a6c7af7 789 ruleset_generate_rule($ruleset, $chain, $rule);
9d31b418
AD
790 }
791 }
9d31b418
AD
792}
793
51bae274
DM
794my $MAX_NETS = 32;
795my $valid_netdev_names = {};
796for (my $i = 0; $i < $MAX_NETS; $i++) {
797 $valid_netdev_names->{"net$i"} = 1;
798}
5e1267a5 799
51bae274
DM
800sub parse_fw_rule {
801 my ($line, $need_iface, $allow_groups) = @_;
5e1267a5 802
a332200b 803 my $macros = get_firewall_macros();
fcba0beb 804 my $protocols = get_etc_protocols();
51bae274
DM
805
806 my ($action, $iface, $source, $dest, $proto, $dport, $sport);
807
808 $line =~ s/#.*$//;
809
810 my @data = split(/\s+/, $line);
811 my $expected_elements = $need_iface ? 7 : 6;
812
813 die "wrong number of rule elements\n" if scalar(@data) > $expected_elements;
814
815 if ($need_iface) {
816 ($action, $iface, $source, $dest, $proto, $dport, $sport) = @data
817 } else {
818 ($action, $source, $dest, $proto, $dport, $sport) = @data;
819 }
820
821 die "incomplete rule\n" if !$action;
822
961b4928
DM
823 my $macro;
824 my $macro_name;
825
51bae274
DM
826 if ($action =~ m/^(ACCEPT|DROP|REJECT)$/) {
827 # OK
828 } elsif ($allow_groups && $action =~ m/^GROUP-(:?\S+)$/) {
829 # OK
830 } elsif ($action =~ m/^(\S+)\((ACCEPT|DROP|REJECT)\)$/) {
961b4928
DM
831 ($macro_name, $action) = ($1, $2);
832 my $lc_macro_name = lc($macro_name);
833 my $preferred_name = $pve_fw_preferred_macro_names->{$lc_macro_name};
834 $macro_name = $preferred_name if $preferred_name;
835 $macro = $macros->{$lc_macro_name};
836 die "unknown macro '$macro_name'\n" if !$macro;
51bae274
DM
837 } else {
838 die "unknown action '$action'\n";
839 }
840
841 if ($need_iface) {
842 $iface = undef if $iface && $iface eq '-';
843 die "unknown interface '$iface'\n"
844 if defined($iface) && !$valid_netdev_names->{$iface};
845 }
846
847 $proto = undef if $proto && $proto eq '-';
848 die "unknown protokol '$proto'\n" if $proto &&
849 !(defined($protocols->{byname}->{$proto}) ||
850 defined($protocols->{byid}->{$proto}));
851
852 $source = undef if $source && $source eq '-';
853 $dest = undef if $dest && $dest eq '-';
854
855 $dport = undef if $dport && $dport eq '-';
856 $sport = undef if $sport && $sport eq '-';
857
51bae274
DM
858 my $nbsource = undef;
859 my $nbdest = undef;
860
861 $nbsource = parse_address_list($source) if $source;
862 $nbdest = parse_address_list($dest) if $dest;
961b4928
DM
863
864 my $rules = [];
51bae274 865
961b4928 866 my $param = {
51bae274 867 action => $action,
51bae274
DM
868 iface => $iface,
869 source => $source,
870 dest => $dest,
871 nbsource => $nbsource,
872 nbdest => $nbdest,
873 proto => $proto,
874 dport => $dport,
875 sport => $sport,
51bae274 876 };
961b4928
DM
877
878 if ($macro) {
879 foreach my $templ (@$macro) {
880 my $rule = {};
881 foreach my $k (keys %$templ) {
882 my $v = $templ->{$k};
857f62c8
DM
883 if ($v eq 'PARAM') {
884 $v = $param->{$k};
885 } elsif ($v eq 'DEST') {
886 $v = $param->{dest};
887 } elsif ($v eq 'SOURCE') {
888 $v = $param->{source};
889 }
890
961b4928
DM
891 die "missing parameter '$k' in macro '$macro_name'\n" if !defined($v);
892 $rule->{$k} = $v;
893 }
894 push @$rules, $rule;
895 }
896 } else {
897 push @$rules, $param;
898 }
899
900 foreach my $rule (@$rules) {
901 $rule->{nbdport} = parse_port_name_number_or_range($rule->{dport})
902 if defined($rule->{dport});
903 $rule->{nbsport} = parse_port_name_number_or_range($rule->{sport})
904 if defined($rule->{sport});
905 }
906
907 return $rules;
51bae274
DM
908}
909
85c6eaed
DM
910sub parse_fw_option {
911 my ($line) = @_;
912
913 my ($opt, $value);
914
915 if ($line =~ m/^enable:\s*(0|1)\s*$/i) {
916 $opt = 'enable';
917 $value = int($1);
918 } elsif ($line =~ m/^(policy-(in|out)):\s*(ACCEPT|DROP|REJECT)\s*$/i) {
919 $opt = lc($1);
920 $value = uc($3);
921 } else {
922 chomp $line;
923 die "can't parse option '$line'\n"
924 }
925
926 return ($opt, $value);
927}
928
51bae274
DM
929sub parse_vm_fw_rules {
930 my ($filename, $fh) = @_;
931
85c6eaed 932 my $res = { in => [], out => [], options => {}};
51bae274
DM
933
934 my $section;
935
5e1267a5
DM
936 while (defined(my $line = <$fh>)) {
937 next if $line =~ m/^#/;
938 next if $line =~ m/^\s*$/;
939
961b4928
DM
940 my $linenr = $fh->input_line_number();
941 my $prefix = "$filename (line $linenr)";
942
85c6eaed 943 if ($line =~ m/^\[(\S+)\]\s*$/i) {
5e1267a5 944 $section = lc($1);
85c6eaed 945 warn "$prefix: ignore unknown section '$section'\n" if !$res->{$section};
5e1267a5
DM
946 next;
947 }
51bae274 948 if (!$section) {
961b4928 949 warn "$prefix: skip line - no section";
5e1267a5
DM
950 next;
951 }
952
85c6eaed
DM
953 next if !$res->{$section}; # skip undefined section
954
955 if ($section eq 'options') {
956 eval {
957 my ($opt, $value) = parse_fw_option($line);
958 $res->{options}->{$opt} = $value;
959 };
960 warn "$prefix: $@" if $@;
961 next;
962 }
963
961b4928
DM
964 my $rules;
965 eval { $rules = parse_fw_rule($line, 1, 1); };
51bae274 966 if (my $err = $@) {
961b4928 967 warn "$prefix: $err";
5e1267a5
DM
968 next;
969 }
970
961b4928 971 push @{$res->{$section}}, @$rules;
51bae274
DM
972 }
973
974 return $res;
975}
976
977sub parse_host_fw_rules {
978 my ($filename, $fh) = @_;
979
980 my $res = { in => [], out => [] };
981
982 my $section;
983
984 while (defined(my $line = <$fh>)) {
985 next if $line =~ m/^#/;
986 next if $line =~ m/^\s*$/;
987
961b4928
DM
988 my $linenr = $fh->input_line_number();
989 my $prefix = "$filename (line $linenr)";
990
51bae274
DM
991 if ($line =~ m/^\[(in|out)\]\s*$/i) {
992 $section = lc($1);
5e1267a5
DM
993 next;
994 }
51bae274 995 if (!$section) {
961b4928 996 warn "$prefix: skip line - no section";
5e1267a5
DM
997 next;
998 }
999
961b4928
DM
1000 my $rules;
1001 eval { $rules = parse_fw_rule($line, 1, 1); };
ecbea048 1002 if (my $err = $@) {
961b4928 1003 warn "$prefix: $err";
ecbea048 1004 next;
ecbea048
DM
1005 }
1006
961b4928 1007 push @{$res->{$section}}, @$rules;
51bae274 1008 }
ecbea048 1009
51bae274
DM
1010 return $res;
1011}
4cdbb3b7 1012
51bae274
DM
1013sub parse_group_fw_rules {
1014 my ($filename, $fh) = @_;
5e1267a5 1015
51bae274
DM
1016 my $section;
1017 my $group;
1018
1019 my $res = { in => [], out => [] };
1020
1021 while (defined(my $line = <$fh>)) {
1022 next if $line =~ m/^#/;
1023 next if $line =~ m/^\s*$/;
1024
961b4928
DM
1025 my $linenr = $fh->input_line_number();
1026 my $prefix = "$filename (line $linenr)";
1027
51bae274
DM
1028 if ($line =~ m/^\[(in|out):(\S+)\]\s*$/i) {
1029 $section = lc($1);
1030 $group = lc($2);
1031 next;
1032 }
1033 if (!$section || !$group) {
961b4928 1034 warn "$prefix: skip line - no section";
51bae274
DM
1035 next;
1036 }
1037
961b4928
DM
1038 my $rules;
1039 eval { $rules = parse_fw_rule($line, 0, 0); };
51bae274 1040 if (my $err = $@) {
961b4928 1041 warn "$prefix: $err";
51bae274
DM
1042 next;
1043 }
1044
961b4928 1045 push @{$res->{$group}->{$section}}, @$rules;
5e1267a5
DM
1046 }
1047
1048 return $res;
1049}
1050
06320eb0
DM
1051sub run_locked {
1052 my ($code, @param) = @_;
1053
1054 my $timeout = 10;
1055
1056 my $res = lock_file($pve_fw_lock_filename, $timeout, $code, @param);
1057
1058 die $@ if $@;
1059
1060 return $res;
1061}
1062
5e1267a5
DM
1063sub read_local_vm_config {
1064
1065 my $openvz = {};
1066
1067 my $qemu = {};
1068
1069 my $list = PVE::QemuServer::config_list();
1070
1071 foreach my $vmid (keys %$list) {
5e1267a5
DM
1072 my $cfspath = PVE::QemuServer::cfs_config_path($vmid);
1073 if (my $conf = PVE::Cluster::cfs_read_file($cfspath)) {
1074 $qemu->{$vmid} = $conf;
1075 }
1076 }
1077
1078 my $vmdata = { openvz => $openvz, qemu => $qemu };
1079
1080 return $vmdata;
1081};
1082
1083sub read_vm_firewall_rules {
1084 my ($vmdata) = @_;
1085 my $rules = {};
1086 foreach my $vmid (keys %{$vmdata->{qemu}}, keys %{$vmdata->{openvz}}) {
1087 my $filename = "/etc/pve/firewall/$vmid.fw";
1088 my $fh = IO::File->new($filename, O_RDONLY);
1089 next if !$fh;
1090
51bae274 1091 $rules->{$vmid} = parse_vm_fw_rules($filename, $fh);
5e1267a5
DM
1092 }
1093
1094 return $rules;
1095}
1096
1097sub compile {
5e1267a5
DM
1098 my $vmdata = read_local_vm_config();
1099 my $rules = read_vm_firewall_rules($vmdata);
51bae274
DM
1100
1101 my $group_rules = {};
1102 my $filename = "/etc/pve/firewall/groups.fw";
1103 if (my $fh = IO::File->new($filename, O_RDONLY)) {
1104 $group_rules = parse_group_fw_rules($filename, $fh);
1105 }
5e1267a5 1106
3fa83edf
DM
1107 #print Dumper($rules);
1108
1109 my $ruleset = {};
1110
dec84fcd
DM
1111 ruleset_create_chain($ruleset, "PVEFW-INPUT");
1112 ruleset_create_chain($ruleset, "PVEFW-OUTPUT");
fadb13dd 1113 ruleset_create_chain($ruleset, "PVEFW-FORWARD");
3fa83edf 1114
4e6112f9
DM
1115 ruleset_create_chain($ruleset, "PVEFW-SET-ACCEPT-MARK");
1116 ruleset_addrule($ruleset, "PVEFW-SET-ACCEPT-MARK", "-j MARK --set-mark 1");
1117
fadb13dd 1118 my $enable_hostfw = 0;
51bae274
DM
1119 $filename = "/etc/pve/local/host.fw";
1120 if (my $fh = IO::File->new($filename, O_RDONLY)) {
1121 my $host_rules = parse_host_fw_rules($filename, $fh);
fadb13dd
DM
1122
1123 $enable_hostfw = 1;
1124
51bae274
DM
1125 enablehostfw($ruleset, $host_rules, $group_rules);
1126 }
3fa83edf
DM
1127
1128 # generate firewall rules for QEMU VMs
1129 foreach my $vmid (keys %{$vmdata->{qemu}}) {
1130 my $conf = $vmdata->{qemu}->{$vmid};
fadb13dd 1131
3fa83edf 1132 next if !$rules->{$vmid};
fadb13dd
DM
1133 my $options = $rules->{$vmid}->{options};
1134 next if defined($options->{enable}) && ($options->{enable} == 0);
3fa83edf
DM
1135
1136 foreach my $netid (keys %$conf) {
1137 next if $netid !~ m/^net(\d+)$/;
1138 my $net = PVE::QemuServer::parse_net($conf->{$netid});
1139 next if !$net;
1140 my $iface = "tap${vmid}i$1";
5e1267a5 1141
3fa83edf
DM
1142 my $bridge = $net->{bridge};
1143 next if !$bridge; # fixme: ?
1144
1145 $bridge .= "v$net->{tag}" if $net->{tag};
1146
fadb13dd 1147
3fa83edf
DM
1148 generate_bridge_chains($ruleset, $bridge);
1149
c29f55c9 1150 my $macaddr = $net->{macaddr};
51bae274
DM
1151 generate_tap_rules_direction($ruleset, $group_rules, $iface, $netid, $macaddr, $rules->{$vmid}->{in}, $bridge, 'IN');
1152 generate_tap_rules_direction($ruleset, $group_rules, $iface, $netid, $macaddr, $rules->{$vmid}->{out}, $bridge, 'OUT');
3fa83edf
DM
1153 }
1154 }
c0413e35 1155
fadb13dd
DM
1156 if ($enable_hostfw) {
1157 # allow traffic from lo (ourself)
1158 ruleset_addrule($ruleset, "PVEFW-INPUT", "-i lo -j ACCEPT");
1159 }
c0413e35 1160
3fa83edf
DM
1161 return $ruleset;
1162}
1163
1164sub get_ruleset_status {
1165 my ($ruleset, $verbose) = @_;
1166
1167 my $active_chains = iptables_get_chains();
1168
1169 my $statushash = {};
1170
1171 foreach my $chain (sort keys %$ruleset) {
09d5f68e 1172 my $digest = Digest::SHA->new('sha1');
3fa83edf
DM
1173 foreach my $cmd (@{$ruleset->{$chain}}) {
1174 $digest->add("$cmd\n");
1175 }
1176 my $sig = $digest->b64digest;
1177 $statushash->{$chain}->{sig} = $sig;
1178
1179 my $oldsig = $active_chains->{$chain};
1180 if (!defined($oldsig)) {
1181 $statushash->{$chain}->{action} = 'create';
1182 } else {
1183 if ($oldsig eq $sig) {
1184 $statushash->{$chain}->{action} = 'exists';
1185 } else {
1186 $statushash->{$chain}->{action} = 'update';
1187 }
1188 }
1189 print "$statushash->{$chain}->{action} $chain ($sig)\n" if $verbose;
1190 foreach my $cmd (@{$ruleset->{$chain}}) {
1191 print "\t$cmd\n" if $verbose;
1192 }
1193 }
1194
1195 foreach my $chain (sort keys %$active_chains) {
1196 if (!defined($ruleset->{$chain})) {
1197 my $sig = $active_chains->{$chain};
1198 $statushash->{$chain}->{action} = 'delete';
1199 $statushash->{$chain}->{sig} = $sig;
1200 print "delete $chain ($sig)\n" if $verbose;
1201 }
1202 }
1203
1204 return $statushash;
1205}
1206
1207sub print_ruleset {
1208 my ($ruleset) = @_;
1209
1210 get_ruleset_status($ruleset, 1);
1211}
1212
1213sub print_sig_rule {
1214 my ($chain, $sig) = @_;
1215
09d5f68e
DM
1216 # We just use this to store a SHA1 checksum used to detect changes
1217 return "-A $chain -m comment --comment \"PVESIG:$sig\"\n";
b6360c3f
DM
1218}
1219
a84f4d96
DM
1220sub apply_ruleset {
1221 my ($ruleset, $verbose) = @_;
3fa83edf 1222
a84f4d96 1223 enable_bridge_firewall();
3fa83edf
DM
1224
1225 my $cmdlist = "*filter\n"; # we pass this to iptables-restore;
1226
1227 my $statushash = get_ruleset_status($ruleset, $verbose);
1228
1229 # create missing chains first
1230 foreach my $chain (sort keys %$ruleset) {
1231 my $stat = $statushash->{$chain};
1232 die "internal error" if !$stat;
1233 next if $stat->{action} ne 'create';
886aba9c 1234
3fa83edf
DM
1235 $cmdlist .= ":$chain - [0:0]\n";
1236 }
1237
dec84fcd 1238 my $rule = "INPUT -j PVEFW-INPUT";
3fa83edf
DM
1239 if (!PVE::Firewall::iptables_rule_exist($rule)) {
1240 $cmdlist .= "-A $rule\n";
1241 }
dec84fcd 1242 $rule = "OUTPUT -j PVEFW-OUTPUT";
3fa83edf
DM
1243 if (!PVE::Firewall::iptables_rule_exist($rule)) {
1244 $cmdlist .= "-A $rule\n";
1245 }
1246
dec84fcd 1247 $rule = "FORWARD -j PVEFW-FORWARD";
3fa83edf
DM
1248 if (!PVE::Firewall::iptables_rule_exist($rule)) {
1249 $cmdlist .= "-A $rule\n";
1250 }
1251
1252 foreach my $chain (sort keys %$ruleset) {
1253 my $stat = $statushash->{$chain};
1254 die "internal error" if !$stat;
1255
1256 if ($stat->{action} eq 'update' || $stat->{action} eq 'create') {
1257 $cmdlist .= "-F $chain\n";
1258 foreach my $cmd (@{$ruleset->{$chain}}) {
1259 $cmdlist .= "$cmd\n";
1260 }
1261 $cmdlist .= print_sig_rule($chain, $stat->{sig});
1262 } elsif ($stat->{action} eq 'delete') {
f5d28682 1263 die "internal error"; # this should not happen
3fa83edf
DM
1264 } elsif ($stat->{action} eq 'exists') {
1265 # do nothing
1266 } else {
1267 die "internal error - unknown status '$stat->{action}'";
1268 }
1269 }
1270
f5d28682
DM
1271 foreach my $chain (keys %$statushash) {
1272 next if $statushash->{$chain}->{action} ne 'delete';
1273 $cmdlist .= "-F $chain\n";
1274 }
1275 foreach my $chain (keys %$statushash) {
1276 next if $statushash->{$chain}->{action} ne 'delete';
fadb13dd
DM
1277 next if $chain eq 'PVEFW-INPUT';
1278 next if $chain eq 'PVEFW-OUTPUT';
1279 next if $chain eq 'PVEFW-FORWARD';
f5d28682
DM
1280 $cmdlist .= "-X $chain\n";
1281 }
1282
3fa83edf
DM
1283 $cmdlist .= "COMMIT\n";
1284
1285 print $cmdlist if $verbose;
1286
1287 iptables_restore_cmdlist($cmdlist);
1288
1289 # test: re-read status and check if everything is up to date
1290 $statushash = get_ruleset_status($ruleset);
1291
1292 my $errors;
1293 foreach my $chain (sort keys %$ruleset) {
1294 my $stat = $statushash->{$chain};
1295 if ($stat->{action} ne 'exists') {
1296 warn "unable to update chain '$chain'\n";
1297 $errors = 1;
1298 }
1299 }
b6360c3f 1300
3fa83edf 1301 die "unable to apply firewall changes\n" if $errors;
b6360c3f
DM
1302}
1303
b6360c3f 13041;