X-Git-Url: https://git.proxmox.com/?p=pve-firewall.git;a=blobdiff_plain;f=test%2Ffwtester.pl;h=457f352100b09395dbe05813cd8cd91aca2c82f2;hp=97e9cfa55485d0bd2b4ef054c0cc5d8d7d445efc;hb=HEAD;hpb=f1bafd371437d8eb39613324259a64b2841c4bc4 diff --git a/test/fwtester.pl b/test/fwtester.pl index 97e9cfa..457f352 100755 --- a/test/fwtester.pl +++ b/test/fwtester.pl @@ -1,304 +1,101 @@ #!/usr/bin/perl use lib '../src'; + use strict; use warnings; -use Data::Dumper; -use PVE::Firewall; - -my $mark; -my $trace; - -sub rule_match { - my ($chain, $rule, $pkg) = @_; - - $rule =~ s/^-A $chain // || die "got strange rule: $rule"; - - if ($rule =~ s/^-m conntrack\s*//) { - return undef; # simply ignore - } - - if ($rule =~ s/^-m addrtype\s*//) { - return undef; # simply ignore - } - - if ($rule =~ s/^-i (\S+)\s*//) { - die "missing iface_in" if !$pkg->{iface_in}; - return undef if $pkg->{iface_in} ne $1; # no match - } - if ($rule =~ s/^-o (\S+)\s*//) { - die "missing iface_out" if !$pkg->{iface_out}; - return undef if $pkg->{iface_out} ne $1; # no match - } - if ($rule =~ s/^-p (tcp|udp)\s*//) { - die "missing proto" if !$pkg->{proto}; - return undef if $pkg->{proto} ne $1; # no match - } - - if ($rule =~ s/^--dport (\d+):(\d+)\s*//) { - die "missing dport" if !$pkg->{dport}; - return undef if ($pkg->{dport} < $1) || ($pkg->{dport} > $2); # no match - } - - if ($rule =~ s/^--dport (\d+)\s*//) { - die "missing dport" if !$pkg->{dport}; - return undef if $pkg->{dport} != $1; # no match - } - - if ($rule =~ s/^-s (\S+)\s*//) { - die "missing source" if !$pkg->{source}; - return undef if $pkg->{source} ne $1; # no match - } - - if ($rule =~ s/^-d (\S+)\s*//) { - die "missing destination" if !$pkg->{dest}; - return undef if $pkg->{dest} ne $1; # no match - } - - if ($rule =~ s/^-m mac ! --mac-source (\S+)\s*//) { - die "missing source mac" if !$pkg->{mac_source}; - return undef if $pkg->{mac_source} eq $1; # no match - } - - if ($rule =~ s/^-m physdev --physdev-is-bridged --physdev-in (\S+)\s*//) { - my $devre = $1; - $devre =~ s/\+/\.\*/; - return undef if !$pkg->{physdev_in}; - return undef if $pkg->{physdev_in} !~ m/^${devre}$/; - } - - if ($rule =~ s/^-m physdev --physdev-is-bridged --physdev-out (\S+)\s*//) { - my $devre = $1; - $devre =~ s/\+/\.\*/; - return undef if !$pkg->{physdev_out}; - return undef if $pkg->{physdev_out} !~ m/^${devre}$/; - } - - if ($rule =~ s/^-j MARK --set-mark (\d+)\s*$//) { - $mark = $1; - return undef; - } - - if ($rule =~ s/^-j (\S+)\s*$//) { - return (0, $1); - } - - if ($rule =~ s/^-g (\S+)\s*$//) { - return (1, $1); - } - - die "unable to parse rule: $rule"; -} - -sub ruleset_simulate_chain { - my ($ruleset, $chain, $pkg) = @_; - - $trace .= "ENTER chain $chain\n"; - - if ($chain eq 'PVEFW-Drop') { - $trace .= "LEAVE chain $chain\n"; - return 'DROP'; - } - if ($chain eq 'PVEFW-reject') { - $trace .= "LEAVE chain $chain\n"; - return 'REJECT'; - } - - if ($chain eq 'PVEFW-tcpflags') { - $trace .= "LEAVE chain $chain\n"; - return undef; - } - - my $rules = $ruleset->{$chain} || - die "no such chain '$chain'"; +use Data::Dumper; +use File::Basename; +use Getopt::Long; +use Net::IP; - foreach my $rule (@$rules) { - my ($goto, $action) = rule_match($chain, $rule, $pkg); - if (!defined($action)) { - $trace .= "SKIP: $rule\n"; - next; - } - $trace .= "MATCH: $rule\n"; - - if ($action eq 'ACCEPT' || $action eq 'DROP' || $action eq 'REJECT') { - $trace .= "TERMINATE chain $chain: $action\n"; - return $action; - } elsif ($action eq 'RETURN') { - $trace .= "RETURN FROM chain $chain\n"; - last; - } else { - if ($goto) { - $trace .= "LEAVE chain $chain - goto $action\n"; - return ruleset_simulate_chain($ruleset, $action, $pkg) - #$chain = $action; - #$rules = $ruleset->{$chain} || die "no such chain '$chain'"; - } else { - if ($action = ruleset_simulate_chain($ruleset, $action, $pkg)) { - return $action; - } - $trace .= "CONTINUE chain $chain\n"; - } - } - } +use PVE::Corosync; +use PVE::FirewallSimulator; +use PVE::INotify; - $trace .= "LEAVE chain $chain\n"; - if ($chain =~ m/^PVEFW-(INPUT|OUTPUT|FORWARD)$/) { - return 'ACCEPT'; # default policy - } +my $debug = 0; - return undef; +sub print_usage_and_exit { + die "usage: $0 [--debug] [testfile [testid]]\n"; } -sub copy_packet { - my ($pkg) = @_; - - my $res = {}; - - while (my ($k,$v) = each %$pkg) { - $res->{$k} = $v; - } - - return $res; +if (!GetOptions ('debug' => \$debug)) { + print_usage_and_exit(); } +# load dummy corosync config to have fw create according rules +my $corosync_conf_fn = "corosync.conf"; +my $raw = PVE::Tools::file_get_contents($corosync_conf_fn); +my $local_hostname = PVE::INotify::nodename(); +(my $raw_replaced = $raw) =~ s/proxself$/$local_hostname\n/gm; +my $corosync_conf = PVE::Corosync::parse_conf($corosync_conf_fn, $raw_replaced); -sub simulate_firewall { - my ($ruleset, $ipset_ruleset, $vmdata, $test) = @_; - - my $from = delete $test->{from} || die "missing 'from' field"; - my $to = delete $test->{to} || die "missing 'to' field"; - my $action = delete $test->{action} || die "missing 'action'"; - - die "from/to needs to be different" if $from eq $to; +PVE::FirewallSimulator::debug($debug); - my $pkg = { - iface_in => 'lo', - iface_out => 'lo', - proto => 'tcp', - sport => '1234', - dport => '4321', - source => '10.11.12.13', - dest => '10.11.12.14', - }; +my $testfilename = shift; +my $testid = shift; - while (my ($k,$v) = each %$test) { - $pkg->{$k} = $v; - } - - my $pre_test; +sub run_tests { + my ($vmdata, $testdir, $testfile, $testid) = @_; - if ($from eq 'host') { - $pre_test = ['PVEFW-OUTPUT', $pkg]; - } elsif ($from =~ m/^ct(\d+)$/) { - my $vmid = $1; - my $conf = $vmdata->{openvz}->{$vmid} || die "no such CT '$vmid'"; - if ($conf->{ip_address}) { - $pkg->{source} = $conf->{ip_address}->{value}; - $pkg->{iface_in} = 'venet0'; - } else { - die "implement me"; - } - $pre_test = ['PVEFW-FORWARD', $pkg]; - } elsif ($from =~ m/^vm(\d+)$/) { - my $vmid = $1; - my $conf = $vmdata->{qemu}->{$vmid} || die "no such VM '$vmid'"; - my $net = PVE::QemuServer::parse_net($conf->{net0}); - my $macaddr = $net->{macaddr} || die "unable to get mac address"; - $pkg->{iface_in} = $net->{bridge} || die "unable to get bridge"; - $pkg->{mac_source} = $macaddr; - my $brpkg = copy_packet($pkg); - $brpkg->{physdev_in} = "tap${vmid}i0"; - $brpkg->{physdev_out} = "link${vmid}i0"; - $brpkg->{iface_in} = $brpkg->{iface_out} = "fwbr${vmid}i0"; - $pre_test = ['PVEFW-FORWARD', $brpkg]; - } else { - die "implement me"; - } + $testfile = 'tests' if !$testfile; - my $post_test; - if ($to eq 'host') { - $post_test = ['PVEFW-INPUT', $pkg]; - } elsif ($to =~ m/^ct(\d+)$/) { - my $vmid = $1; - my $conf = $vmdata->{openvz}->{$vmid} || die "no such CT '$vmid'"; - if ($conf->{ip_address}) { - $pkg->{dest} = $conf->{ip_address}->{value}; - $pkg->{iface_out} = 'venet0'; - } else { - die "implement me"; - } - $post_test = ['PVEFW-FORWARD', $pkg]; - } elsif ($to =~ m/^vm(\d+)$/) { - my $vmid = $1; - my $conf = $vmdata->{qemu}->{$vmid} || die "no such VM '$vmid'"; - my $net = PVE::QemuServer::parse_net($conf->{net0}); - $pkg->{iface_out} = $net->{bridge} || die "unable to get bridge"; - my $brpkg = copy_packet($pkg); - $brpkg->{physdev_out} = "tap${vmid}i0"; - $brpkg->{physdev_in} = "link${vmid}i0"; - $brpkg->{iface_in} = $brpkg->{iface_out} = "fwbr${vmid}i0"; - $post_test = ['PVEFW-FORWARD', $brpkg]; - } else { - die "implement me"; - } - my $res = 'UNKNOWN'; - if ($pre_test) { - my ($chain, $testpkg) = @$pre_test; - $trace .= "PRE TEST $chain: " . Dumper($testpkg); - $res = ruleset_simulate_chain($ruleset, $chain, $testpkg); - if ($res ne 'ACCEPT') { - die "test failed ($res != $action)\n" if $action ne $res; - return undef; # sucess - } - } - - if ($post_test) { - my ($chain, $testpkg) = @$post_test; - $trace .= "POST TEST $chain: " . Dumper($testpkg); - $res = ruleset_simulate_chain($ruleset, $chain, $testpkg); - if ($res ne 'ACCEPT') { - die "test failed ($res != $action)\n" if $action ne $res; - return undef; # sucess - } - } - - die "test failed ($res != $action)\n" if $action ne $res; # fixme: remove - - return undef; -} + $vmdata->{testdir} = $testdir; -sub run_tests { - my ($vmdata, $testdir) = @_; + my $host_ip = '172.16.1.2'; - $vmdata->{testdir} = $testdir; + PVE::Firewall::local_network('172.16.1.0/24'); my ($ruleset, $ipset_ruleset) = - PVE::Firewall::compile(undef, undef, $vmdata); + PVE::Firewall::compile(undef, undef, $vmdata, $corosync_conf); - my $testfile = "$testdir/tests"; - my $fh = IO::File->new($testfile) || - die "unable to open '$testfile' - $!\n"; + my $filename = "$testdir/$testfile"; + my $fh = IO::File->new($filename) || + die "unable to open '$filename' - $!\n"; + my $testcount = 0; while (defined(my $line = <$fh>)) { next if $line =~ m/^\s*$/; next if $line =~ m/^#.*$/; if ($line =~ m/^\{.*\}\s*$/) { my $test = eval $line; die $@ if $@; - $trace = ''; - eval { simulate_firewall($ruleset, $ipset_ruleset, $vmdata, $test); }; + next if defined($testid) && (!defined($test->{id}) || ($testid ne $test->{id})); + PVE::FirewallSimulator::reset_trace(); + print Dumper($ruleset->{filter}) if $debug; + $testcount++; + eval { + my @test_zones = qw(host outside nfvm vm100 ct200); + if (!defined($test->{from}) && !defined($test->{to})) { + die "missing zone speification (from, to)\n"; + } elsif (!defined($test->{to})) { + foreach my $zone (@test_zones) { + next if $zone eq $test->{from}; + $test->{to} = $zone; + PVE::FirewallSimulator::add_trace("Set Zone: to => '$zone'\n"); + PVE::FirewallSimulator::simulate_firewall($ruleset->{filter}, $ipset_ruleset, + $host_ip, $vmdata, $test); + } + } elsif (!defined($test->{from})) { + foreach my $zone (@test_zones) { + next if $zone eq $test->{to}; + $test->{from} = $zone; + PVE::FirewallSimulator::add_trace("Set Zone: from => '$zone'\n"); + PVE::FirewallSimulator::simulate_firewall($ruleset->{filter}, $ipset_ruleset, + $host_ip, $vmdata, $test); + } + } else { + PVE::FirewallSimulator::simulate_firewall($ruleset->{filter}, $ipset_ruleset, + $host_ip, $vmdata, $test); + } + }; if (my $err = $@) { - - print Dumper($ruleset); - - print "$trace\n"; - - print "$testfile line $.: $line"; - + print Dumper($ruleset->{filter}) if !$debug; + print PVE::FirewallSimulator::get_trace() . "\n" if !$debug; + print "$filename line $.: $line"; print "test failed: $err\n"; - exit(-1); } } else { @@ -306,7 +103,9 @@ sub run_tests { } } - print "PASS: $testfile\n"; + die "no tests found\n" if $testcount <= 0; + + print "PASS: $filename\n"; return undef; } @@ -314,19 +113,48 @@ sub run_tests { my $vmdata = { qemu => { 100 => { - net0 => "e1000=0E:0B:38:B8:B3:21,bridge=vmbr0,firewall=0", + net0 => "e1000=0E:0B:38:B8:B3:21,bridge=vmbr0,firewall=1", + net1 => "e1000=0E:0B:38:B9:B4:21,bridge=vmbr1,firewall=1", + net2 => "e1000=0E:0B:38:BA:B4:21,bridge=vmbr2,firewall=1", + }, + 101 => { + net0 => "e1000=0E:0B:38:B8:B3:22,bridge=vmbr0,firewall=1", + }, + # on bridge vmbr1 + 110 => { + net0 => "e1000=0E:0B:38:B8:B4:21,bridge=vmbr1,firewall=1", }, }, - openvz => { + lxc => { 200 => { - ip_address => { value => '10.0.200.1' }, + net0 => "name=eth0,hwaddr=0E:18:24:41:2C:43,bridge=vmbr0,firewall=1,ip=10.0.200.1/24", + }, + 201 => { + net0 => "name=eth0,hwaddr=0E:18:24:41:2C:44,bridge=vmbr0,firewall=1,ip=10.0.200.2/24", }, }, }; -foreach my $dir () { - next if ! -d $dir; - run_tests($vmdata, $dir); +if ($testfilename) { + my $testfile; + my $dir; + + if (-d $testfilename) { + $dir = $testfilename; + } elsif (-f $testfilename) { + $dir = dirname($testfilename); + $testfile = basename($testfilename); + } else { + die "no such file/dir '$testfilename'\n"; + } + + run_tests($vmdata, $dir, $testfile, $testid); + +} else { + foreach my $dir () { + next if ! -d $dir; + run_tests($vmdata, $dir); + } } print "OK - all tests passed\n";