-sub add_trace {
- my ($text) = @_;
-
- if ($debug) {
- print $text;
- } else {
- $trace .= $text;
- }
-}
-
-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) = @_;
-
- add_trace("ENTER chain $chain\n");
-
- if ($chain eq 'PVEFW-Drop') {
- add_trace("LEAVE chain $chain\n");
- return 'DROP';
- }
- if ($chain eq 'PVEFW-reject') {
- add_trace("LEAVE chain $chain\n");
- return 'REJECT';
- }
-
- if ($chain eq 'PVEFW-tcpflags') {
- add_trace("LEAVE chain $chain\n");
- return undef;
- }
-
- my $rules = $ruleset->{$chain} ||
- die "no such chain '$chain'";
-
- foreach my $rule (@$rules) {
- my ($goto, $action) = rule_match($chain, $rule, $pkg);
- if (!defined($action)) {
- add_trace("SKIP: $rule\n");
- next;
- }
- add_trace("MATCH: $rule\n");
-
- if ($action eq 'ACCEPT' || $action eq 'DROP' || $action eq 'REJECT') {
- add_trace("TERMINATE chain $chain: $action\n");
- return $action;
- } elsif ($action eq 'RETURN') {
- add_trace("RETURN FROM chain $chain\n");
- last;
- } else {
- if ($goto) {
- add_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;
- }
- add_trace("CONTINUE chain $chain\n");
- }
- }
- }
-
- add_trace("LEAVE chain $chain\n");
- if ($chain =~ m/^PVEFW-(INPUT|OUTPUT|FORWARD)$/) {
- return 'ACCEPT'; # default policy
- }
-
- return undef;
-}
-
-sub copy_packet {
- my ($pkg) = @_;
-
- my $res = {};
-
- while (my ($k,$v) = each %$pkg) {
- $res->{$k} = $v;
- }
-
- return $res;
-}
-
-# Try to simulate packet traversal inside kernel. This invokes iptable
-# checks several times.
-sub route_packet {
- my ($ruleset, $ipset_ruleset, $pkg, $from_info, $target, $start_state) = @_;
-
- my $route_state = $start_state;
-
- my $physdev_in;
-
- while ($route_state ne $target->{iface}) {
-
- my $chain;
- my $next_route_state;
- my $next_physdev_in;
-
- $pkg->{iface_in} = $pkg->{iface_out} = undef;
- $pkg->{physdev_in} = $pkg->{physdev_out} = undef;
-
- if ($route_state eq 'host') {
-
- if ($target->{type} eq 'ct') {
- $pkg->{iface_in} = 'lo';
- $pkg->{iface_out} = 'venet0';
- $chain = 'PVEFW-OUTPUT';
- $next_route_state = 'venet-in';
- } elsif ($target->{type} eq 'vm') {
- $pkg->{iface_in} = 'lo';
- $pkg->{iface_out} = $target->{bridge} || die 'internal error';
- $chain = 'PVEFW-OUTPUT';
- $next_route_state = 'fwbr-in';
- } else {
- die "implement me";
- }
-
- } elsif ($route_state eq 'venet-out') {
-
- if ($target->{type} eq 'host') {
-
- $chain = 'PVEFW-INPUT';
- $pkg->{iface_in} = 'venet0';
- $pkg->{iface_out} = 'lo';
- $next_route_state = 'host';
-
- } elsif ($target->{type} eq 'vm') {
-
- $chain = 'PVEFW-FORWARD';
- $pkg->{iface_in} = 'venet0';
- $pkg->{iface_out} = $target->{bridge} || die 'internal error';
- $next_route_state = 'fwbr-in';
-
- } elsif ($target->{type} eq 'ct') {
-
- $chain = 'PVEFW-FORWARD';
- $pkg->{iface_in} = 'venet0';
- $pkg->{iface_out} = 'venet0';
- $next_route_state = 'venet-in';
-
- } else {
- die "implement me";
- }
-
- } elsif ($route_state eq 'fwbr-out') {
-
- $chain = 'PVEFW-FORWARD';
- $next_route_state = $from_info->{bridge} || die 'internal error';
- $next_physdev_in = $from_info->{fwpr} || die 'internal error';
- $pkg->{iface_in} = $from_info->{fwbr} || die 'internal error';
- $pkg->{iface_out} = $from_info->{fwbr} || die 'internal error';
- $pkg->{physdev_in} = $from_info->{tapdev} || die 'internal error';
- $pkg->{physdev_out} = $from_info->{fwln} || die 'internal error';
-
- } elsif ($route_state eq 'fwbr-in') {
-
- $chain = 'PVEFW-FORWARD';
- $next_route_state = $target->{tapdev};
- $pkg->{iface_in} = $target->{fwbr} || die 'internal error';
- $pkg->{iface_out} = $target->{fwbr} || die 'internal error';
- $pkg->{physdev_in} = $target->{fwln} || die 'internal error';
- $pkg->{physdev_out} = $target->{tapdev} || die 'internal error';
-
- } elsif ($route_state =~ m/^vmbr\d+$/) {
-
- die "missing physdev_in - internal error?" if !$physdev_in;
-
- if ($target->{type} eq 'host') {
-
- $chain = 'PVEFW-INPUT';
- $pkg->{iface_in} = $route_state;
- $pkg->{iface_out} = 'lo';
- $next_route_state = 'host';
-
- } elsif ($target->{type} eq 'ct') {
-
- $chain = 'PVEFW-FORWARD';
- $pkg->{iface_in} = $route_state;
- $pkg->{iface_out} = 'venet0';
- $next_route_state = 'venet-in';
-
- } elsif ($target->{type} eq 'vm') {
-
- $chain = 'PVEFW-FORWARD';
- if ($route_state eq $target->{bridge}) {
- $pkg->{iface_in} = $route_state;
- $pkg->{iface_out} = $route_state;
- $pkg->{physdev_in} = $physdev_in;
- $pkg->{physdev_out} = $target->{fwpr} || die 'internal error';
- } else {
- $pkg->{iface_in} = $route_state;
- $pkg->{iface_out} = $route_state;
- $pkg->{physdev_in} = $physdev_in;
- # do not set physdev_out (same behavior as kernel)
- }
- $next_route_state = 'fwbr-in';
-
- } else {
- die "implement me";
- }
-
- } else {
- die "implement me $route_state";
- }
-
- die "internal error" if !defined($next_route_state);
-
- if ($chain) {
- add_trace("IPT check at $route_state (chain $chain)\n");
- add_trace(Dumper($pkg));
- my $res = ruleset_simulate_chain($ruleset, $chain, $pkg);
- return $res if $res ne 'ACCEPT';
- }
-
- $route_state = $next_route_state;
-
- $physdev_in = $next_physdev_in;
- }
-
- return 'ACCEPT';
-}
-
-sub extract_ct_info {
- my ($vmdata, $vmid) = @_;
-
- my $info = { type => 'ct', vmid => $vmid };
-
- my $conf = $vmdata->{openvz}->{$vmid} || die "no such CT '$vmid'";
- if ($conf->{ip_address}) {
- $info->{ip_address} = $conf->{ip_address}->{value};
- } else {
- die "implement me";
- }
- return $info;