Fix "show nat translation monitor" to work with both snat & dnat.

1 files changed, 17 insertions, 1 deletions

diff --git a/scripts/vyatta-nat-translations.pl b/scripts/vyatta-nat-translations.pl
index abd5f5c..d156df8 100755
--- a/scripts/vyatta-nat-translations.pl
+++ b/scripts/vyatta-nat-translations.pl
@@ -33,7 +33,7 @@ use strict;
 
 my $dump = 0;
 my ($xml_file, $verbose, $proto, $stats, $ipaddr, $pipe);
-my $mode='both';
+my $mode;
 my $verbose_format = "%-20s %-18s %-20s %-18s\n";
 my $format         = "%-20s %-20s %-4s  %-4s  %-8s";
 
@@ -61,6 +61,18 @@ sub print_xml {
     print Dumper($data); 
 }
 
+sub guess_snat_dnat {
+    my ($src, $dst) = @_;
+
+    if ($src->{original} eq $dst->{reply}) {
+	return "dnat";
+    }
+    if ($dst->{original} eq $src->{reply}) {
+	return "snat";
+    }
+    return "unkn";
+}
+
 sub nat_print_xml {
     my ($data, $mode) = @_;
 
@@ -116,6 +128,9 @@ sub nat_print_xml {
 	if (defined $verbose) {
 	    printf($verbose_format, $in_src, $in_dst, $out_src, $out_dst);
 	}
+	if (! defined $mode) {
+	    $mode = guess_snat_dnat(\%src, \%dst);
+	}
 	if (defined $mode) {
 	    my ($from, $to);
 	    if ($mode eq 'snat') {
@@ -222,6 +237,7 @@ if (defined $xml_file) {
     } else {
 	$proto = "";
     }
+    $mode = 'both' if ! defined $mode;
     if ($mode eq 'both' or $mode eq 'snat') {
 	my $ipopt = "";
 	if (defined $ipaddr) {