]> git.proxmox.com Git - pve-firewall.git/blame - PVE/Firewall.pm
use accept mark for security groups
[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
9aab3127 18my $macros;
3a616aa0 19
a332200b
DM
20# todo: implement some kind of MACROS, like shorewall /usr/share/shorewall/macro.*
21sub get_firewall_macros {
9aab3127
DM
22
23 return $macros if $macros;
24
a332200b
DM
25 #foreach my $path (</usr/share/shorewall/macro.*>) {
26 # if ($path =~ m|/macro\.(\S+)$|) {
27 # $macros->{$1} = 1;
28 # }
29 #}
30
31 $macros = {}; # fixme: implemet me
32
9aab3127
DM
33 return $macros;
34}
35
fcba0beb
DM
36my $etc_services;
37
38sub get_etc_services {
39
40 return $etc_services if $etc_services;
41
42 my $filename = "/etc/services";
43
44 my $fh = IO::File->new($filename, O_RDONLY);
45 if (!$fh) {
46 warn "unable to read '$filename' - $!\n";
47 return {};
48 }
49
50 my $services = {};
51
52 while (my $line = <$fh>) {
53 chomp ($line);
54 next if $line =~m/^#/;
55 next if ($line =~m/^\s*$/);
56
57 if ($line =~ m!^(\S+)\s+(\S+)/(tcp|udp).*$!) {
58 $services->{byid}->{$2}->{name} = $1;
59 $services->{byid}->{$2}->{$3} = 1;
60 $services->{byname}->{$1} = $services->{byid}->{$2};
61 }
62 }
63
64 close($fh);
65
66 $etc_services = $services;
67
3a616aa0 68
fcba0beb
DM
69 return $etc_services;
70}
71
72my $etc_protocols;
73
74sub get_etc_protocols {
75 return $etc_protocols if $etc_protocols;
76
77 my $filename = "/etc/protocols";
78
79 my $fh = IO::File->new($filename, O_RDONLY);
80 if (!$fh) {
81 warn "unable to read '$filename' - $!\n";
82 return {};
83 }
84
85 my $protocols = {};
86
87 while (my $line = <$fh>) {
88 chomp ($line);
89 next if $line =~m/^#/;
90 next if ($line =~m/^\s*$/);
91
92 if ($line =~ m!^(\S+)\s+(\d+)\s+.*$!) {
93 $protocols->{byid}->{$2}->{name} = $1;
94 $protocols->{byname}->{$1} = $protocols->{byid}->{$2};
95 }
96 }
97
98 close($fh);
99
100 $etc_protocols = $protocols;
101
102 return $etc_protocols;
103}
104
ecbea048
DM
105sub parse_address_list {
106 my ($str) = @_;
107
d6de1dc2 108 my $nbaor = 0;
ecbea048
DM
109 foreach my $aor (split(/,/, $str)) {
110 if (!Net::IP->new($aor)) {
111 my $err = Net::IP::Error();
112 die "invalid IP address: $err\n";
d6de1dc2
AD
113 }else{
114 $nbaor++;
ecbea048
DM
115 }
116 }
d6de1dc2 117 return $nbaor;
ecbea048 118}
dddd9413 119
fcba0beb
DM
120sub parse_port_name_number_or_range {
121 my ($str) = @_;
122
123 my $services = PVE::Firewall::get_etc_services();
4cdbb3b7 124 my $nbports = 0;
fcba0beb 125 foreach my $item (split(/,/, $str)) {
4cdbb3b7 126 my $portlist = "";
fcba0beb 127 foreach my $pon (split(':', $item, 2)) {
4cdbb3b7
AD
128 if ($pon =~ m/^\d+$/){
129 die "invalid port '$pon'\n" if $pon < 0 && $pon > 65536;
130 }else{
131 die "invalid port $services->{byname}->{$pon}\n" if !$services->{byname}->{$pon};
132 }
133 $nbports++;
fcba0beb
DM
134 }
135 }
136
4cdbb3b7 137 return ($nbports);
fcba0beb
DM
138}
139
780bcc0f
DM
140my $bridge_firewall_enabled = 0;
141
142sub enable_bridge_firewall {
143
144 return if $bridge_firewall_enabled; # only once
145
146 system("echo 1 > /proc/sys/net/bridge/bridge-nf-call-iptables");
147 system("echo 1 > /proc/sys/net/bridge/bridge-nf-call-ip6tables");
148
149 $bridge_firewall_enabled = 1;
150}
151
8cebfa6f 152my $rule_format = "%-15s %-30s %-30s %-15s %-15s %-15s\n";
dddd9413 153
3a616aa0
AD
154sub iptables {
155 my ($cmd) = @_;
156
157 run_command("/sbin/iptables $cmd", outfunc => sub {}, errfunc => sub {});
158}
159
b16e818e
DM
160sub iptables_restore_cmdlist {
161 my ($cmdlist) = @_;
3a616aa0 162
3fa83edf 163 run_command("/sbin/iptables-restore -n", input => $cmdlist);
3a616aa0
AD
164}
165
de2a57cd
DM
166sub iptables_get_chains {
167
168 my $res = {};
169
170 # check what chains we want to track
171 my $is_pvefw_chain = sub {
172 my $name = shift;
173
dec84fcd
DM
174 return 1 if $name =~ m/^PVEFW-\S+$/;
175
de2a57cd 176 return 1 if $name =~ m/^tap\d+i\d+-(:?IN|OUT)$/;
b16e818e 177 return 1 if $name =~ m/^vmbr\d+-(:?IN|OUT)$/;
f50656c2 178 return 1 if $name =~ m/^GROUP-(:?[^\s\-]+)-(:?IN|OUT)$/;
de2a57cd
DM
179
180 return undef;
181 };
182
183 my $table = '';
184
de2a57cd
DM
185 my $parser = sub {
186 my $line = shift;
187
188 return if $line =~ m/^#/;
189 return if $line =~ m/^\s*$/;
190
191 if ($line =~ m/^\*(\S+)$/) {
192 $table = $1;
193 return;
194 }
195
196 return if $table ne 'filter';
197
198 if ($line =~ m/^:(\S+)\s/) {
199 my $chain = $1;
200 return if !&$is_pvefw_chain($chain);
3fa83edf 201 $res->{$chain} = "unknown";
09d5f68e 202 } elsif ($line =~ m/^-A\s+(\S+)\s.*--comment\s+\"PVESIG:(\S+)\"/) {
3fa83edf 203 my ($chain, $sig) = ($1, $2);
de2a57cd 204 return if !&$is_pvefw_chain($chain);
3fa83edf 205 $res->{$chain} = $sig;
de2a57cd
DM
206 } else {
207 # simply ignore the rest
208 return;
209 }
210 };
211
212 run_command("/sbin/iptables-save", outfunc => $parser);
213
de2a57cd
DM
214 return $res;
215}
216
3a616aa0
AD
217sub iptables_chain_exist {
218 my ($chain) = @_;
219
220 eval{
221 iptables("-n --list $chain");
222 };
223 return undef if $@;
224
225 return 1;
226}
227
228sub iptables_rule_exist {
229 my ($rule) = @_;
230
231 eval{
232 iptables("-C $rule");
233 };
234 return undef if $@;
235
236 return 1;
237}
238
3fa83edf 239sub ruleset_generate_rule {
4e6112f9 240 my ($ruleset, $chain, $rule, $goto) = @_;
3a616aa0 241
3fa83edf 242 my $cmd = '';
3a616aa0 243
d6de1dc2 244 $cmd .= " -m iprange --src-range" if $rule->{nbsource} && $rule->{nbsource} > 1;
3a616aa0 245 $cmd .= " -s $rule->{source}" if $rule->{source};
d6de1dc2 246 $cmd .= " -m iprange --dst-range" if $rule->{nbdest} && $rule->{nbdest} > 1;
3a616aa0
AD
247 $cmd .= " -d $rule->{dest}" if $rule->{destination};
248 $cmd .= " -p $rule->{proto}" if $rule->{proto};
4cdbb3b7 249 $cmd .= " --match multiport" if $rule->{nbdport} && $rule->{nbdport} > 1;
3a616aa0 250 $cmd .= " --dport $rule->{dport}" if $rule->{dport};
4cdbb3b7 251 $cmd .= " --match multiport" if $rule->{nbsport} && $rule->{nbsport} > 1;
3a616aa0 252 $cmd .= " --sport $rule->{sport}" if $rule->{sport};
4e6112f9
DM
253
254 if (my $action = $rule->{action}) {
255 $goto = 1 if !defined($goto) && $action eq 'PVEFW-SET-ACCEPT-MARK';
256 $cmd .= $goto ? " -g $action" : " -j $action";
257 };
3a616aa0 258
3fa83edf
DM
259 ruleset_addrule($ruleset, $chain, $cmd) if $cmd;
260}
261
262sub ruleset_create_chain {
263 my ($ruleset, $chain) = @_;
3a616aa0 264
d050c724
DM
265 die "Invalid chain name '$chain' (28 char max)\n" if length($chain) > 28;
266
3fa83edf
DM
267 die "chain '$chain' already exists\n" if $ruleset->{$chain};
268
269 $ruleset->{$chain} = [];
3a616aa0
AD
270}
271
3fa83edf
DM
272sub ruleset_chain_exist {
273 my ($ruleset, $chain) = @_;
3a616aa0 274
3fa83edf
DM
275 return $ruleset->{$chain} ? 1 : undef;
276}
3a616aa0 277
3fa83edf
DM
278sub ruleset_addrule {
279 my ($ruleset, $chain, $rule) = @_;
3a616aa0 280
3fa83edf 281 die "no such chain '$chain'\n" if !$ruleset->{$chain};
3a616aa0 282
3fa83edf
DM
283 push @{$ruleset->{$chain}}, "-A $chain $rule";
284}
3a616aa0 285
3fa83edf
DM
286sub ruleset_insertrule {
287 my ($ruleset, $chain, $rule) = @_;
3a616aa0 288
3fa83edf 289 die "no such chain '$chain'\n" if !$ruleset->{$chain};
3a616aa0 290
3fa83edf
DM
291 unshift @{$ruleset->{$chain}}, "-A $chain $rule";
292}
3a616aa0 293
3fa83edf
DM
294sub generate_bridge_chains {
295 my ($ruleset, $bridge) = @_;
3a616aa0 296
dec84fcd
DM
297 if (!ruleset_chain_exist($ruleset, "PVEFW-BRIDGE-IN")){
298 ruleset_create_chain($ruleset, "PVEFW-BRIDGE-IN");
d4d0fd1d
AD
299 }
300
dec84fcd
DM
301 if (!ruleset_chain_exist($ruleset, "PVEFW-BRIDGE-OUT")){
302 ruleset_create_chain($ruleset, "PVEFW-BRIDGE-OUT");
d4d0fd1d 303 }
3fa83edf 304
dec84fcd
DM
305 if (!ruleset_chain_exist($ruleset, "PVEFW-FORWARD")){
306 ruleset_create_chain($ruleset, "PVEFW-FORWARD");
3fa83edf 307
dec84fcd
DM
308 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-m state --state RELATED,ESTABLISHED -j ACCEPT");
309 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-m physdev --physdev-is-in --physdev-is-bridged -j PVEFW-BRIDGE-OUT");
310 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-m physdev --physdev-is-out --physdev-is-bridged -j PVEFW-BRIDGE-IN");
3a616aa0
AD
311 }
312
3fa83edf
DM
313 if (!ruleset_chain_exist($ruleset, "$bridge-IN")) {
314 ruleset_create_chain($ruleset, "$bridge-IN");
dec84fcd
DM
315 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-i $bridge -j DROP"); # disable interbridge routing
316 ruleset_addrule($ruleset, "PVEFW-BRIDGE-IN", "-j $bridge-IN");
3fa83edf
DM
317 ruleset_addrule($ruleset, "$bridge-IN", "-j ACCEPT");
318 }
3a616aa0 319
3fa83edf
DM
320 if (!ruleset_chain_exist($ruleset, "$bridge-OUT")) {
321 ruleset_create_chain($ruleset, "$bridge-OUT");
dec84fcd
DM
322 ruleset_addrule($ruleset, "PVEFW-FORWARD", "-o $bridge -j DROP"); # disable interbridge routing
323 ruleset_addrule($ruleset, "PVEFW-BRIDGE-OUT", "-j $bridge-OUT");
3fa83edf
DM
324 }
325}
3a616aa0
AD
326
327sub generate_tap_rules_direction {
c29f55c9 328 my ($ruleset, $iface, $netid, $macaddr, $rules, $bridge, $direction) = @_;
3a616aa0
AD
329
330 my $tapchain = "$iface-$direction";
331
3fa83edf 332 ruleset_create_chain($ruleset, $tapchain);
3a616aa0 333
3fa83edf
DM
334 ruleset_addrule($ruleset, $tapchain, "-m state --state INVALID -j DROP");
335 ruleset_addrule($ruleset, $tapchain, "-m state --state RELATED,ESTABLISHED -j ACCEPT");
3a616aa0 336
c29f55c9
DM
337 if ($direction eq 'OUT' && defined($macaddr)) {
338 ruleset_addrule($ruleset, $tapchain, "-m mac ! --mac-source $macaddr -j DROP");
339 }
340
3fa83edf 341 if ($rules) {
3a616aa0
AD
342 foreach my $rule (@$rules) {
343 next if $rule->{iface} && $rule->{iface} ne $netid;
344 if($rule->{action} =~ m/^(GROUP-(\S+))$/){
3fa83edf
DM
345 $rule->{action} .= "-$direction";
346 # generate empty group rule if don't exist
347 if(!ruleset_chain_exist($ruleset, $rule->{action})){
348 generate_group_rules($ruleset, $2);
349 }
4e6112f9
DM
350 ruleset_generate_rule($ruleset, $tapchain, $rule);
351 ruleset_addrule($ruleset, $tapchain, "-m mark --mark 1 -g $bridge-IN");
352 } else {
353 # we go to vmbr-IN if accept in out rules
354 $rule->{action} = "$bridge-IN" if $rule->{action} eq 'ACCEPT' && $direction eq 'OUT';
355 ruleset_generate_rule($ruleset, $tapchain, $rule);
3a616aa0 356 }
4e6112f9 357 }
3a616aa0
AD
358 }
359
3fa83edf
DM
360 ruleset_addrule($ruleset, $tapchain, "-j LOG --log-prefix \"$tapchain-dropped: \" --log-level 4");
361 ruleset_addrule($ruleset, $tapchain, "-j DROP");
3a616aa0 362
3fa83edf
DM
363 # plug the tap chain to bridge chain
364 my $physdevdirection = $direction eq 'IN' ? "out" : "in";
365 my $rule = "-m physdev --physdev-$physdevdirection $iface --physdev-is-bridged -j $tapchain";
366 ruleset_insertrule($ruleset, "$bridge-$direction", $rule);
3a616aa0 367
3fa83edf
DM
368 if ($direction eq 'OUT'){
369 # add tap->host rules
370 my $rule = "-m physdev --physdev-$physdevdirection $iface -j $tapchain";
dec84fcd 371 ruleset_addrule($ruleset, "PVEFW-INPUT", $rule);
3a616aa0
AD
372 }
373}
374
0bd5f137 375sub enablehostfw {
3fa83edf 376 my ($ruleset) = @_;
0bd5f137
AD
377
378 my $filename = "/etc/pve/local/host.fw";
379 my $fh = IO::File->new($filename, O_RDONLY);
380 return if !$fh;
381
382 my $rules = parse_fw_rules($filename, $fh);
0bd5f137 383
3fa83edf 384 # host inbound firewall
dec84fcd
DM
385 my $chain = "PVEFW-HOST-IN";
386 ruleset_create_chain($ruleset, $chain);
0bd5f137 387
dec84fcd
DM
388 ruleset_addrule($ruleset, $chain, "-m state --state INVALID -j DROP");
389 ruleset_addrule($ruleset, $chain, "-m state --state RELATED,ESTABLISHED -j ACCEPT");
390 ruleset_addrule($ruleset, $chain, "-i lo -j ACCEPT");
391 ruleset_addrule($ruleset, $chain, "-m addrtype --dst-type MULTICAST -j ACCEPT");
392 ruleset_addrule($ruleset, $chain, "-p udp -m state --state NEW -m multiport --dports 5404,5405 -j ACCEPT");
393 ruleset_addrule($ruleset, $chain, "-p udp -m udp --dport 9000 -j ACCEPT"); #corosync
0bd5f137 394
3fa83edf
DM
395 if ($rules->{in}) {
396 foreach my $rule (@{$rules->{in}}) {
397 # we use RETURN because we need to check also tap rules
398 $rule->{action} = 'RETURN' if $rule->{action} eq 'ACCEPT';
dec84fcd 399 ruleset_generate_rule($ruleset, $chain, $rule);
3fa83edf 400 }
0bd5f137
AD
401 }
402
dec84fcd
DM
403 ruleset_addrule($ruleset, $chain, "-j LOG --log-prefix \"kvmhost-IN dropped: \" --log-level 4");
404 ruleset_addrule($ruleset, $chain, "-j DROP");
0bd5f137 405
3fa83edf 406 # host outbound firewall
aadd745e 407 $chain = "PVEFW-HOST-OUT";
dec84fcd
DM
408 ruleset_create_chain($ruleset, $chain);
409
410 ruleset_addrule($ruleset, $chain, "-m state --state INVALID -j DROP");
411 ruleset_addrule($ruleset, $chain, "-m state --state RELATED,ESTABLISHED -j ACCEPT");
412 ruleset_addrule($ruleset, $chain, "-o lo -j ACCEPT");
413 ruleset_addrule($ruleset, $chain, "-m addrtype --dst-type MULTICAST -j ACCEPT");
414 ruleset_addrule($ruleset, $chain, "-p udp -m state --state NEW -m multiport --dports 5404,5405 -j ACCEPT");
415 ruleset_addrule($ruleset, $chain, "-p udp -m udp --dport 9000 -j ACCEPT"); #corosync
0bd5f137 416
3fa83edf
DM
417 if ($rules->{out}) {
418 foreach my $rule (@{$rules->{out}}) {
419 # we use RETURN because we need to check also tap rules
420 $rule->{action} = 'RETURN' if $rule->{action} eq 'ACCEPT';
dec84fcd 421 ruleset_generate_rule($ruleset, $chain, $rule);
3fa83edf 422 }
0bd5f137
AD
423 }
424
dec84fcd
DM
425 ruleset_addrule($ruleset, $chain, "-j LOG --log-prefix \"kvmhost-OUT dropped: \" --log-level 4");
426 ruleset_addrule($ruleset, $chain, "-j DROP");
9d31b418 427
dec84fcd
DM
428 ruleset_addrule($ruleset, "PVEFW-OUTPUT", "-j PVEFW-HOST-OUT");
429 ruleset_addrule($ruleset, "PVEFW-INPUT", "-j PVEFW-HOST-IN");
9d31b418
AD
430}
431
432sub generate_group_rules {
3fa83edf 433 my ($ruleset, $group) = @_;
9d31b418
AD
434
435 my $filename = "/etc/pve/firewall/groups.fw";
436 my $fh = IO::File->new($filename, O_RDONLY);
437 return if !$fh;
438
439 my $rules = parse_fw_rules($filename, $fh, $group);
9d31b418 440
3fa83edf 441 my $chain = "GROUP-${group}-IN";
9d31b418 442
3fa83edf 443 ruleset_create_chain($ruleset, $chain);
9d31b418 444
3fa83edf
DM
445 if ($rules->{in}) {
446 foreach my $rule (@{$rules->{in}}) {
447 ruleset_generate_rule($ruleset, $chain, $rule);
9d31b418
AD
448 }
449 }
450
3fa83edf 451 $chain = "GROUP-${group}-OUT";
9d31b418 452
3fa83edf 453 ruleset_create_chain($ruleset, $chain);
4e6112f9 454 ruleset_addrule($ruleset, $chain, "-j MARK --set-mark 0"); # clear mark
9d31b418 455
3fa83edf
DM
456 if ($rules->{out}) {
457 foreach my $rule (@{$rules->{out}}) {
4e6112f9
DM
458 # we go the PVEFW-SET-ACCEPT-MARK Instead of ACCEPT) because we need to
459 # check also other tap rules (and group rules can be set on any bridge,
460 # so we can't go to VMBRXX-IN)
461 $rule->{action} = 'PVEFW-SET-ACCEPT-MARK' if $rule->{action} eq 'ACCEPT';
6a6c7af7 462 ruleset_generate_rule($ruleset, $chain, $rule);
9d31b418
AD
463 }
464 }
9d31b418
AD
465}
466
5e1267a5 467sub parse_fw_rules {
9d31b418 468 my ($filename, $fh, $group) = @_;
5e1267a5
DM
469
470 my $section;
9d31b418
AD
471 my $securitygroup;
472 my $securitygroupexist;
5e1267a5
DM
473
474 my $res = { in => [], out => [] };
475
a332200b 476 my $macros = get_firewall_macros();
fcba0beb
DM
477 my $protocols = get_etc_protocols();
478
5e1267a5
DM
479 while (defined(my $line = <$fh>)) {
480 next if $line =~ m/^#/;
481 next if $line =~ m/^\s*$/;
482
9d31b418 483 if ($line =~ m/^\[(in|out)(:(\S+))?\]\s*$/i) {
5e1267a5 484 $section = lc($1);
9d31b418
AD
485 $securitygroup = lc($3) if $3;
486 $securitygroupexist = 1 if $securitygroup && $securitygroup eq $group;
5e1267a5
DM
487 next;
488 }
489 next if !$section;
9d31b418 490 next if $group && $securitygroup ne $group;
5e1267a5
DM
491
492 my ($action, $iface, $source, $dest, $proto, $dport, $sport) =
493 split(/\s+/, $line);
494
495 if (!$action) {
496 warn "skip incomplete line\n";
497 next;
498 }
499
500 my $service;
3a616aa0 501 if ($action =~ m/^(ACCEPT|DROP|REJECT|GROUP-(\S+))$/) {
5e1267a5
DM
502 # OK
503 } elsif ($action =~ m/^(\S+)\((ACCEPT|DROP|REJECT)\)$/) {
504 ($service, $action) = ($1, $2);
505 if (!$macros->{$service}) {
506 warn "unknown service '$service'\n";
507 next;
508 }
509 } else {
510 warn "unknown action '$action'\n";
511 next;
512 }
513
514 $iface = undef if $iface && $iface eq '-';
515 if ($iface && $iface !~ m/^(net0|net1|net2|net3|net4|net5)$/) {
516 warn "unknown interface '$iface'\n";
517 next;
518 }
519
520 $proto = undef if $proto && $proto eq '-';
fcba0beb
DM
521 if ($proto && !(defined($protocols->{byname}->{$proto}) ||
522 defined($protocols->{byid}->{$proto}))) {
5e1267a5
DM
523 warn "unknown protokol '$proto'\n";
524 next;
525 }
526
527 $source = undef if $source && $source eq '-';
5e1267a5 528 $dest = undef if $dest && $dest eq '-';
5e1267a5
DM
529
530 $dport = undef if $dport && $dport eq '-';
531 $sport = undef if $sport && $sport eq '-';
4cdbb3b7
AD
532 my $nbdport = undef;
533 my $nbsport = undef;
d6de1dc2
AD
534 my $nbsource = undef;
535 my $nbdest = undef;
5e1267a5 536
ecbea048 537 eval {
d6de1dc2
AD
538 $nbsource = parse_address_list($source) if $source;
539 $nbdest = parse_address_list($dest) if $dest;
4cdbb3b7
AD
540 $nbdport = parse_port_name_number_or_range($dport) if $dport;
541 $nbsport = parse_port_name_number_or_range($sport) if $sport;
ecbea048
DM
542 };
543 if (my $err = $@) {
544 warn $err;
545 next;
546
547 }
548
549
5e1267a5
DM
550 my $rule = {
551 action => $action,
552 service => $service,
553 iface => $iface,
554 source => $source,
555 dest => $dest,
d6de1dc2
AD
556 nbsource => $nbsource,
557 nbdest => $nbdest,
5e1267a5
DM
558 proto => $proto,
559 dport => $dport,
560 sport => $sport,
4cdbb3b7
AD
561 nbdport => $nbdport,
562 nbsport => $nbsport,
563
5e1267a5
DM
564 };
565
566 push @{$res->{$section}}, $rule;
567 }
568
9d31b418 569 die "security group $group don't exist" if $group && !$securitygroupexist;
5e1267a5
DM
570 return $res;
571}
572
06320eb0
DM
573sub run_locked {
574 my ($code, @param) = @_;
575
576 my $timeout = 10;
577
578 my $res = lock_file($pve_fw_lock_filename, $timeout, $code, @param);
579
580 die $@ if $@;
581
582 return $res;
583}
584
5e1267a5
DM
585sub read_local_vm_config {
586
587 my $openvz = {};
588
589 my $qemu = {};
590
591 my $list = PVE::QemuServer::config_list();
592
593 foreach my $vmid (keys %$list) {
5e1267a5
DM
594 my $cfspath = PVE::QemuServer::cfs_config_path($vmid);
595 if (my $conf = PVE::Cluster::cfs_read_file($cfspath)) {
596 $qemu->{$vmid} = $conf;
597 }
598 }
599
600 my $vmdata = { openvz => $openvz, qemu => $qemu };
601
602 return $vmdata;
603};
604
605sub read_vm_firewall_rules {
606 my ($vmdata) = @_;
607 my $rules = {};
608 foreach my $vmid (keys %{$vmdata->{qemu}}, keys %{$vmdata->{openvz}}) {
609 my $filename = "/etc/pve/firewall/$vmid.fw";
610 my $fh = IO::File->new($filename, O_RDONLY);
611 next if !$fh;
612
613 $rules->{$vmid} = parse_fw_rules($filename, $fh);
614 }
615
616 return $rules;
617}
618
619sub compile {
5e1267a5
DM
620 my $vmdata = read_local_vm_config();
621 my $rules = read_vm_firewall_rules($vmdata);
622
3fa83edf
DM
623 #print Dumper($rules);
624
625 my $ruleset = {};
626
627 # setup host firewall rules
dec84fcd
DM
628 ruleset_create_chain($ruleset, "PVEFW-INPUT");
629 ruleset_create_chain($ruleset, "PVEFW-OUTPUT");
3fa83edf 630
4e6112f9
DM
631 ruleset_create_chain($ruleset, "PVEFW-SET-ACCEPT-MARK");
632 ruleset_addrule($ruleset, "PVEFW-SET-ACCEPT-MARK", "-j MARK --set-mark 1");
633
3fa83edf
DM
634 enablehostfw($ruleset);
635
636 # generate firewall rules for QEMU VMs
637 foreach my $vmid (keys %{$vmdata->{qemu}}) {
638 my $conf = $vmdata->{qemu}->{$vmid};
639 next if !$rules->{$vmid};
640
641 foreach my $netid (keys %$conf) {
642 next if $netid !~ m/^net(\d+)$/;
643 my $net = PVE::QemuServer::parse_net($conf->{$netid});
644 next if !$net;
645 my $iface = "tap${vmid}i$1";
5e1267a5 646
3fa83edf
DM
647 my $bridge = $net->{bridge};
648 next if !$bridge; # fixme: ?
649
650 $bridge .= "v$net->{tag}" if $net->{tag};
651
652 generate_bridge_chains($ruleset, $bridge);
653
c29f55c9
DM
654 my $macaddr = $net->{macaddr};
655 generate_tap_rules_direction($ruleset, $iface, $netid, $macaddr, $rules->{$vmid}->{in}, $bridge, 'IN');
656 generate_tap_rules_direction($ruleset, $iface, $netid, $macaddr, $rules->{$vmid}->{out}, $bridge, 'OUT');
3fa83edf
DM
657 }
658 }
3fa83edf
DM
659 return $ruleset;
660}
661
662sub get_ruleset_status {
663 my ($ruleset, $verbose) = @_;
664
665 my $active_chains = iptables_get_chains();
666
667 my $statushash = {};
668
669 foreach my $chain (sort keys %$ruleset) {
09d5f68e 670 my $digest = Digest::SHA->new('sha1');
3fa83edf
DM
671 foreach my $cmd (@{$ruleset->{$chain}}) {
672 $digest->add("$cmd\n");
673 }
674 my $sig = $digest->b64digest;
675 $statushash->{$chain}->{sig} = $sig;
676
677 my $oldsig = $active_chains->{$chain};
678 if (!defined($oldsig)) {
679 $statushash->{$chain}->{action} = 'create';
680 } else {
681 if ($oldsig eq $sig) {
682 $statushash->{$chain}->{action} = 'exists';
683 } else {
684 $statushash->{$chain}->{action} = 'update';
685 }
686 }
687 print "$statushash->{$chain}->{action} $chain ($sig)\n" if $verbose;
688 foreach my $cmd (@{$ruleset->{$chain}}) {
689 print "\t$cmd\n" if $verbose;
690 }
691 }
692
693 foreach my $chain (sort keys %$active_chains) {
694 if (!defined($ruleset->{$chain})) {
695 my $sig = $active_chains->{$chain};
696 $statushash->{$chain}->{action} = 'delete';
697 $statushash->{$chain}->{sig} = $sig;
698 print "delete $chain ($sig)\n" if $verbose;
699 }
700 }
701
702 return $statushash;
703}
704
705sub print_ruleset {
706 my ($ruleset) = @_;
707
708 get_ruleset_status($ruleset, 1);
709}
710
711sub print_sig_rule {
712 my ($chain, $sig) = @_;
713
09d5f68e
DM
714 # We just use this to store a SHA1 checksum used to detect changes
715 return "-A $chain -m comment --comment \"PVESIG:$sig\"\n";
b6360c3f
DM
716}
717
a84f4d96
DM
718sub apply_ruleset {
719 my ($ruleset, $verbose) = @_;
3fa83edf 720
a84f4d96 721 enable_bridge_firewall();
3fa83edf
DM
722
723 my $cmdlist = "*filter\n"; # we pass this to iptables-restore;
724
725 my $statushash = get_ruleset_status($ruleset, $verbose);
726
727 # create missing chains first
728 foreach my $chain (sort keys %$ruleset) {
729 my $stat = $statushash->{$chain};
730 die "internal error" if !$stat;
731 next if $stat->{action} ne 'create';
886aba9c 732
3fa83edf
DM
733 $cmdlist .= ":$chain - [0:0]\n";
734 }
735
dec84fcd 736 my $rule = "INPUT -j PVEFW-INPUT";
3fa83edf
DM
737 if (!PVE::Firewall::iptables_rule_exist($rule)) {
738 $cmdlist .= "-A $rule\n";
739 }
dec84fcd 740 $rule = "OUTPUT -j PVEFW-OUTPUT";
3fa83edf
DM
741 if (!PVE::Firewall::iptables_rule_exist($rule)) {
742 $cmdlist .= "-A $rule\n";
743 }
744
dec84fcd 745 $rule = "FORWARD -j PVEFW-FORWARD";
3fa83edf
DM
746 if (!PVE::Firewall::iptables_rule_exist($rule)) {
747 $cmdlist .= "-A $rule\n";
748 }
749
750 foreach my $chain (sort keys %$ruleset) {
751 my $stat = $statushash->{$chain};
752 die "internal error" if !$stat;
753
754 if ($stat->{action} eq 'update' || $stat->{action} eq 'create') {
755 $cmdlist .= "-F $chain\n";
756 foreach my $cmd (@{$ruleset->{$chain}}) {
757 $cmdlist .= "$cmd\n";
758 }
759 $cmdlist .= print_sig_rule($chain, $stat->{sig});
760 } elsif ($stat->{action} eq 'delete') {
f5d28682 761 die "internal error"; # this should not happen
3fa83edf
DM
762 } elsif ($stat->{action} eq 'exists') {
763 # do nothing
764 } else {
765 die "internal error - unknown status '$stat->{action}'";
766 }
767 }
768
f5d28682
DM
769 foreach my $chain (keys %$statushash) {
770 next if $statushash->{$chain}->{action} ne 'delete';
771 $cmdlist .= "-F $chain\n";
772 }
773 foreach my $chain (keys %$statushash) {
774 next if $statushash->{$chain}->{action} ne 'delete';
775 $cmdlist .= "-X $chain\n";
776 }
777
3fa83edf
DM
778 $cmdlist .= "COMMIT\n";
779
780 print $cmdlist if $verbose;
781
782 iptables_restore_cmdlist($cmdlist);
783
784 # test: re-read status and check if everything is up to date
785 $statushash = get_ruleset_status($ruleset);
786
787 my $errors;
788 foreach my $chain (sort keys %$ruleset) {
789 my $stat = $statushash->{$chain};
790 if ($stat->{action} ne 'exists') {
791 warn "unable to update chain '$chain'\n";
792 $errors = 1;
793 }
794 }
b6360c3f 795
3fa83edf 796 die "unable to apply firewall changes\n" if $errors;
b6360c3f
DM
797}
798
b6360c3f 7991;