]>
git.proxmox.com Git - pve-firewall.git/blob - test/fwtester.pl
a99a7906ded4d1865bfd9b579dc4e016baae7332
17 sub print_usage_and_exit
{
18 die "usage: $0 [--debug] [testfile [testid]]\n";
21 if (!GetOptions
('debug' => \
$debug)) {
22 print_usage_and_exit
();
25 my $testfilename = shift;
39 my ($devre, $dev) = @_;
41 $devre =~ s/\+$/\.\*/;
42 return ($dev =~ m/^${devre}$/) ?
1 : 0;
46 my ($ipsetname, $ipset, $ipaddr) = @_;
48 my $ip = Net
::IP-
>new($ipaddr);
50 foreach my $entry (@$ipset) {
51 next if $entry =~ m/^create/; # simply ignore
52 if ($entry =~ m/add \S+ (\S+)$/) {
53 my $test = Net
::IP-
>new($1);
54 if ($test->overlaps($ip)) {
55 add_trace
("IPSET $ipsetname match $ipaddr\n");
67 my ($ipset_ruleset, $chain, $rule, $pkg) = @_;
69 $rule =~ s/^-A $chain // || die "got strange rule: $rule";
71 while (length($rule)) {
73 if ($rule =~ s/^-m conntrack --ctstate (\S+)\s*//) {
76 return undef if $cstate eq 'INVALID'; # no match
77 return undef if $cstate eq 'RELATED,ESTABLISHED'; # no match
79 next if $cstate =~ m/NEW/;
81 die "please implement cstate test '$cstate'";
84 if ($rule =~ s/^-m addrtype --src-type (\S+)\s*//) {
86 die "missing srctype" if !$pkg->{srctype
};
87 return undef if $atype ne $pkg->{srctype
};
90 if ($rule =~ s/^-m addrtype --dst-type (\S+)\s*//) {
92 die "missing dsttype" if !$pkg->{dsttype
};
93 return undef if $atype ne $pkg->{dsttype
};
96 if ($rule =~ s/^-i (\S+)\s*//) {
98 die "missing iface_in" if !$pkg->{iface_in
};
99 return undef if !nf_dev_match
($devre, $pkg->{iface_in
});
103 if ($rule =~ s/^-o (\S+)\s*//) {
105 die "missing iface_out" if !$pkg->{iface_out
};
106 return undef if !nf_dev_match
($devre, $pkg->{iface_out
});
110 if ($rule =~ s/^-p (tcp|udp)\s*//) {
111 die "missing proto" if !$pkg->{proto
};
112 return undef if $pkg->{proto
} ne $1; # no match
116 if ($rule =~ s/^--dport (\d+):(\d+)\s*//) {
117 die "missing dport" if !$pkg->{dport
};
118 return undef if ($pkg->{dport
} < $1) || ($pkg->{dport
} > $2); # no match
122 if ($rule =~ s/^--dport (\d+)\s*//) {
123 die "missing dport" if !$pkg->{dport
};
124 return undef if $pkg->{dport
} != $1; # no match
128 if ($rule =~ s/^-s (\S+)\s*//) {
129 die "missing source" if !$pkg->{source
};
130 my $ip = Net
::IP-
>new($1);
131 return undef if !$ip->overlaps(Net
::IP-
>new($pkg->{source
})); # no match
135 if ($rule =~ s/^-d (\S+)\s*//) {
136 die "missing destination" if !$pkg->{dest
};
137 my $ip = Net
::IP-
>new($1);
138 return undef if !$ip->overlaps(Net
::IP-
>new($pkg->{dest
})); # no match
142 if ($rule =~ s/^-m set --match-set (\S+) src\s*//) {
143 die "missing source" if !$pkg->{source
};
144 my $ipset = $ipset_ruleset->{$1};
145 die "no such ip set '$1'" if !$ipset;
146 return undef if !ipset_match
($1, $ipset, $pkg->{source
});
150 if ($rule =~ s/^-m set --match-set (\S+) dst\s*//) {
151 die "missing destination" if !$pkg->{dest
};
152 my $ipset = $ipset_ruleset->{$1};
153 die "no such ip set '$1'" if !$ipset;
154 return undef if !ipset_match
($1, $ipset, $pkg->{dest
});
158 if ($rule =~ s/^-m mac ! --mac-source (\S+)\s*//) {
159 die "missing source mac" if !$pkg->{mac_source
};
160 return undef if $pkg->{mac_source
} eq $1; # no match
164 if ($rule =~ s/^-m physdev --physdev-is-bridged --physdev-in (\S+)\s*//) {
166 return undef if !$pkg->{physdev_in
};
167 return undef if !nf_dev_match
($devre, $pkg->{physdev_in
});
171 if ($rule =~ s/^-m physdev --physdev-is-bridged --physdev-out (\S+)\s*//) {
173 return undef if !$pkg->{physdev_out
};
174 return undef if !nf_dev_match
($devre, $pkg->{physdev_out
});
178 if ($rule =~ s/^-m mark --mark (\d+)\s*//) {
179 return undef if !defined($mark) || $mark != $1;
185 if ($rule =~ s/^-j MARK --set-mark (\d+)\s*$//) {
190 if ($rule =~ s/^-j (\S+)\s*$//) {
194 if ($rule =~ s/^-g (\S+)\s*$//) {
198 if ($rule =~ s/^-j NFLOG --nflog-prefix \"[^\"]+\"$//) {
205 die "unable to parse rule: $rule";
208 sub ruleset_simulate_chain
{
209 my ($ruleset, $ipset_ruleset, $chain, $pkg) = @_;
211 add_trace
("ENTER chain $chain\n");
215 if ($chain eq 'PVEFW-Drop') {
216 add_trace
("LEAVE chain $chain\n");
217 return ('DROP', $counter);
219 if ($chain eq 'PVEFW-reject') {
220 add_trace
("LEAVE chain $chain\n");
221 return ('REJECT', $counter);
224 if ($chain eq 'PVEFW-tcpflags') {
225 add_trace
("LEAVE chain $chain\n");
226 return (undef, $counter);
229 my $rules = $ruleset->{$chain} ||
230 die "no such chain '$chain'";
232 foreach my $rule (@$rules) {
234 my ($goto, $action) = rule_match
($ipset_ruleset, $chain, $rule, $pkg);
235 if (!defined($action)) {
236 add_trace
("SKIP: $rule\n");
239 add_trace
("MATCH: $rule\n");
241 if ($action eq 'ACCEPT' || $action eq 'DROP' || $action eq 'REJECT') {
242 add_trace
("TERMINATE chain $chain: $action\n");
243 return ($action, $counter);
244 } elsif ($action eq 'RETURN') {
245 add_trace
("RETURN FROM chain $chain\n");
249 add_trace
("LEAVE chain $chain - goto $action\n");
250 return ruleset_simulate_chain
($ruleset, $ipset_ruleset, $action, $pkg)
252 #$rules = $ruleset->{$chain} || die "no such chain '$chain'";
254 my ($act, $ctr) = ruleset_simulate_chain
($ruleset, $ipset_ruleset, $action, $pkg);
256 return ($act, $counter) if $act;
257 add_trace
("CONTINUE chain $chain\n");
262 add_trace
("LEAVE chain $chain\n");
263 if ($chain =~ m/^PVEFW-(INPUT|OUTPUT|FORWARD)$/) {
264 return ('ACCEPT', $counter); # default policy
267 return (undef, $counter);
275 while (my ($k,$v) = each %$pkg) {
282 # Try to simulate packet traversal inside kernel. This invokes iptable
283 # checks several times.
285 my ($ruleset, $ipset_ruleset, $pkg, $from_info, $target, $start_state) = @_;
287 my $route_state = $start_state;
291 my $ipt_invocation_counter = 0;
292 my $rule_check_counter = 0;
294 while ($route_state ne $target->{iface
}) {
297 my $next_route_state;
300 $pkg->{iface_in
} = $pkg->{iface_out
} = undef;
301 $pkg->{physdev_in
} = $pkg->{physdev_out
} = undef;
303 if ($route_state eq 'from-bport') {
304 $next_route_state = $from_info->{bridge
} || die 'internal error';
305 $next_physdev_in = $from_info->{iface
} || die 'internal error';
306 } elsif ($route_state eq 'host') {
308 if ($target->{type
} eq 'bport') {
309 $pkg->{iface_in
} = 'lo';
310 $pkg->{iface_out
} = $target->{bridge
} || die 'internal error';
311 $chain = 'PVEFW-OUTPUT';
312 $next_route_state = $target->{iface
} || die 'internal error';
313 } elsif ($target->{type
} eq 'ct') {
314 $pkg->{iface_in
} = 'lo';
315 $pkg->{iface_out
} = 'venet0';
316 $chain = 'PVEFW-OUTPUT';
317 $next_route_state = 'venet-in';
318 } elsif ($target->{type
} eq 'vm') {
319 $pkg->{iface_in
} = 'lo';
320 $pkg->{iface_out
} = $target->{bridge
} || die 'internal error';
321 $chain = 'PVEFW-OUTPUT';
322 $next_route_state = 'fwbr-in';
327 } elsif ($route_state eq 'venet-out') {
329 if ($target->{type
} eq 'host') {
331 $chain = 'PVEFW-INPUT';
332 $pkg->{iface_in
} = 'venet0';
333 $pkg->{iface_out
} = 'lo';
334 $next_route_state = 'host';
336 } elsif ($target->{type
} eq 'bport') {
338 $chain = 'PVEFW-FORWARD';
339 $pkg->{iface_in
} = 'venet0';
340 $pkg->{iface_out
} = $target->{bridge
} || die 'internal error';
341 $next_route_state = $target->{iface
} || die 'internal error';
343 } elsif ($target->{type
} eq 'vm') {
345 $chain = 'PVEFW-FORWARD';
346 $pkg->{iface_in
} = 'venet0';
347 $pkg->{iface_out
} = $target->{bridge
} || die 'internal error';
348 $next_route_state = 'fwbr-in';
350 } elsif ($target->{type
} eq 'ct') {
352 $chain = 'PVEFW-FORWARD';
353 $pkg->{iface_in
} = 'venet0';
354 $pkg->{iface_out
} = 'venet0';
355 $next_route_state = 'venet-in';
361 } elsif ($route_state eq 'fwbr-out') {
363 $chain = 'PVEFW-FORWARD';
364 $next_route_state = $from_info->{bridge
} || die 'internal error';
365 $next_physdev_in = $from_info->{fwpr
} || die 'internal error';
366 $pkg->{iface_in
} = $from_info->{fwbr
} || die 'internal error';
367 $pkg->{iface_out
} = $from_info->{fwbr
} || die 'internal error';
368 $pkg->{physdev_in
} = $from_info->{tapdev
} || die 'internal error';
369 $pkg->{physdev_out
} = $from_info->{fwln
} || die 'internal error';
371 } elsif ($route_state eq 'fwbr-in') {
373 $chain = 'PVEFW-FORWARD';
374 $next_route_state = $target->{tapdev
};
375 $pkg->{iface_in
} = $target->{fwbr
} || die 'internal error';
376 $pkg->{iface_out
} = $target->{fwbr
} || die 'internal error';
377 $pkg->{physdev_in
} = $target->{fwln
} || die 'internal error';
378 $pkg->{physdev_out
} = $target->{tapdev
} || die 'internal error';
380 } elsif ($route_state =~ m/^vmbr\d+$/) {
382 die "missing physdev_in - internal error?" if !$physdev_in;
383 $pkg->{physdev_in
} = $physdev_in;
385 if ($target->{type
} eq 'host') {
387 $chain = 'PVEFW-INPUT';
388 $pkg->{iface_in
} = $route_state;
389 $pkg->{iface_out
} = 'lo';
390 $next_route_state = 'host';
392 } elsif ($target->{type
} eq 'bport') {
394 $chain = 'PVEFW-FORWARD';
395 $pkg->{iface_in
} = $route_state;
396 $pkg->{iface_out
} = $target->{bridge
} || die 'internal error';
397 # conditionally set physdev_out (same behavior as kernel)
398 if ($route_state eq $target->{bridge
}) {
399 $pkg->{physdev_out
} = $target->{iface
} || die 'internal error';
401 $next_route_state = $target->{iface
};
403 } elsif ($target->{type
} eq 'ct') {
405 $chain = 'PVEFW-FORWARD';
406 $pkg->{iface_in
} = $route_state;
407 $pkg->{iface_out
} = 'venet0';
408 $next_route_state = 'venet-in';
410 } elsif ($target->{type
} eq 'vm') {
412 $chain = 'PVEFW-FORWARD';
413 $pkg->{iface_in
} = $route_state;
414 $pkg->{iface_out
} = $target->{bridge
};
415 # conditionally set physdev_out (same behavior as kernel)
416 if ($route_state eq $target->{bridge
}) {
417 $pkg->{physdev_out
} = $target->{fwpr
} || die 'internal error';
419 $next_route_state = 'fwbr-in';
426 die "implement me $route_state";
429 die "internal error" if !defined($next_route_state);
432 add_trace
("IPT check at $route_state (chain $chain)\n");
433 add_trace
(Dumper
($pkg));
434 $ipt_invocation_counter++;
435 my ($res, $ctr) = ruleset_simulate_chain
($ruleset, $ipset_ruleset, $chain, $pkg);
436 $rule_check_counter += $ctr;
437 return ($res, $ipt_invocation_counter, $rule_check_counter) if $res ne 'ACCEPT';
440 $route_state = $next_route_state;
442 $physdev_in = $next_physdev_in;
445 return ('ACCEPT', $ipt_invocation_counter, $rule_check_counter);
448 sub extract_ct_info
{
449 my ($vmdata, $vmid) = @_;
451 my $info = { type
=> 'ct', vmid
=> $vmid };
453 my $conf = $vmdata->{openvz
}->{$vmid} || die "no such CT '$vmid'";
454 if ($conf->{ip_address
}) {
455 $info->{ip_address
} = $conf->{ip_address
}->{value
};
462 sub extract_vm_info
{
463 my ($vmdata, $vmid) = @_;
465 my $info = { type
=> 'vm', vmid
=> $vmid };
467 my $conf = $vmdata->{qemu
}->{$vmid} || die "no such VM '$vmid'";
468 my $net = PVE
::QemuServer
::parse_net
($conf->{net0
});
469 $info->{macaddr
} = $net->{macaddr
} || die "unable to get mac address";
470 $info->{bridge
} = $net->{bridge
} || die "unable to get bridge";
471 $info->{fwbr
} = "fwbr${vmid}i0";
472 $info->{tapdev
} = "tap${vmid}i0";
473 $info->{fwln
} = "fwln${vmid}i0";
474 $info->{fwpr
} = "fwpr${vmid}p0";
479 sub simulate_firewall
{
480 my ($ruleset, $ipset_ruleset, $vmdata, $test) = @_;
482 my $from = $test->{from
} || die "missing 'from' field";
483 my $to = $test->{to
} || die "missing 'to' field";
484 my $action = $test->{action
} || die "missing 'action'";
486 my $testid = $test->{id
};
488 die "from/to needs to be different" if $from eq $to;
496 srctype
=> 'UNICAST',
497 dsttype
=> 'UNICAST',
500 while (my ($k,$v) = each %$test) {
501 next if $k eq 'from';
503 next if $k eq 'action';
505 die "unknown attribute '$k'\n" if !exists($pkg->{$k});
513 my $host_ip = '172.16.1.2';
515 if ($from eq 'host') {
516 $from_info->{type
} = 'host';
517 $start_state = 'host';
518 $pkg->{source
} = $host_ip if !defined($pkg->{source
});
519 } elsif ($from =~ m
|^(vmbr\d
+)/(\S
+)$|) {
520 $from_info->{type
} = 'bport';
521 $from_info->{bridge
} = $1;
522 $from_info->{iface
} = $2;
523 $start_state = 'from-bport';
524 } elsif ($from eq 'outside') {
525 $from_info->{type
} = 'bport';
526 $from_info->{bridge
} = 'vmbr0';
527 $from_info->{iface
} = 'eth0';
528 $start_state = 'from-bport';
529 } elsif ($from eq 'nfvm') {
530 $from_info->{type
} = 'bport';
531 $from_info->{bridge
} = 'vmbr0';
532 $from_info->{iface
} = 'tapXYZ';
533 $start_state = 'from-bport';
534 } elsif ($from =~ m/^ct(\d+)$/) {
536 $from_info = extract_ct_info
($vmdata, $vmid);
537 if ($from_info->{ip_address
}) {
538 $pkg->{source
} = $from_info->{ip_address
} if !defined($pkg->{source
});
539 $start_state = 'venet-out';
543 } elsif ($from =~ m/^vm(\d+)$/) {
545 $from_info = extract_vm_info
($vmdata, $vmid);
546 $start_state = 'fwbr-out';
547 $pkg->{mac_source
} = $from_info->{macaddr
};
549 die "unable to parse \"from => '$from'\"\n";
555 $target->{type
} = 'host';
556 $target->{iface
} = 'host';
557 $pkg->{dest
} = $host_ip if !defined($pkg->{dest
});
558 } elsif ($to =~ m
|^(vmbr\d
+)/(\S
+)$|) {
559 $target->{type
} = 'bport';
560 $target->{bridge
} = $1;
561 $target->{iface
} = $2;
562 } elsif ($to eq 'outside') {
563 $target->{type
} = 'bport';
564 $target->{bridge
} = 'vmbr0';
565 $target->{iface
} = 'eth0';
566 } elsif ($to eq 'nfvm') {
567 $target->{type
} = 'bport';
568 $target->{bridge
} = 'vmbr0';
569 $target->{iface
} = 'tapXYZ';
570 } elsif ($to =~ m/^ct(\d+)$/) {
572 $target = extract_ct_info
($vmdata, $vmid);
573 $target->{iface
} = 'venet-in';
575 if ($target->{ip_address
}) {
576 $pkg->{dest
} = $target->{ip_address
};
580 } elsif ($to =~ m/^vm(\d+)$/) {
582 $target = extract_vm_info
($vmdata, $vmid);
583 $target->{iface
} = $target->{tapdev
};
585 die "unable to parse \"to => '$to'\"\n";
588 $pkg->{source
} = '100.100.1.2' if !defined($pkg->{source
});
589 $pkg->{dest
} = '100.200.3.4' if !defined($pkg->{dest
});
591 my ($res, $ic, $rc) = route_packet
($ruleset, $ipset_ruleset, $pkg,
592 $from_info, $target, $start_state);
594 add_trace
("IPT statistics: invocation = $ic, checks = $rc\n");
596 die "test failed ($res != $action)\n" if $action ne $res;
602 my ($vmdata, $testdir, $testfile, $testid) = @_;
604 $testfile = 'tests' if !$testfile;
606 $vmdata->{testdir
} = $testdir;
608 PVE
::Firewall
::cluster_network
('172.16.1.0/24');
610 my ($ruleset, $ipset_ruleset) =
611 PVE
::Firewall
::compile
(undef, undef, $vmdata);
613 my $filename = "$testdir/$testfile";
614 my $fh = IO
::File-
>new($filename) ||
615 die "unable to open '$filename' - $!\n";
618 while (defined(my $line = <$fh>)) {
619 next if $line =~ m/^\s*$/;
620 next if $line =~ m/^#.*$/;
621 if ($line =~ m/^\{.*\}\s*$/) {
622 my $test = eval $line;
624 next if defined($testid) && (!defined($test->{id
}) || ($testid ne $test->{id
}));
626 print Dumper
($ruleset) if $debug;
629 my @test_zones = qw(host outside nfvm vm100 ct200);
630 if (!defined($test->{from
}) && !defined($test->{to
})) {
631 die "missing zone speification (from, to)\n";
632 } elsif (!defined($test->{to
})) {
633 foreach my $zone (@test_zones) {
634 next if $zone eq $test->{from
};
636 add_trace
("Set Zone: to => '$zone'\n");
637 simulate_firewall
($ruleset, $ipset_ruleset, $vmdata, $test);
639 } elsif (!defined($test->{from
})) {
640 foreach my $zone (@test_zones) {
641 next if $zone eq $test->{to
};
642 $test->{from
} = $zone;
643 add_trace
("Set Zone: from => '$zone'\n");
644 simulate_firewall
($ruleset, $ipset_ruleset, $vmdata, $test);
647 simulate_firewall
($ruleset, $ipset_ruleset, $vmdata, $test);
652 print Dumper
($ruleset) if !$debug;
654 print "$trace\n" if !$debug;
656 print "$filename line $.: $line";
658 print "test failed: $err\n";
667 die "no tests found\n" if $testcount <= 0;
669 print "PASS: $filename\n";
677 net0
=> "e1000=0E:0B:38:B8:B3:21,bridge=vmbr0,firewall=1",
680 net0
=> "e1000=0E:0B:38:B8:B3:22,bridge=vmbr0,firewall=1",
684 net0
=> "e1000=0E:0B:38:B8:B4:21,bridge=vmbr1,firewall=1",
689 ip_address
=> { value
=> '10.0.200.1' },
692 ip_address
=> { value
=> '10.0.200.2' },
701 if (-d
$testfilename) {
702 $dir = $testfilename;
703 } elsif (-f
$testfilename) {
704 $dir = dirname
($testfilename);
705 $testfile = basename
($testfilename);
707 die "no such file/dir '$testfilename'\n";
710 run_tests
($vmdata, $dir, $testfile, $testid);
713 foreach my $dir (<test-
*>) {
715 run_tests
($vmdata, $dir);
719 print "OK - all tests passed\n";