enable peer-groups

1 files changed, 82 insertions, 1 deletions

diff --git a/scripts/bgp/vyatta-bgp.pl b/scripts/bgp/vyatta-bgp.pl
index d20ad8fa..ee4c9a49 100755
--- a/scripts/bgp/vyatta-bgp.pl
+++ b/scripts/bgp/vyatta-bgp.pl
@@ -1,12 +1,29 @@
 #!/usr/bin/perl
 use lib "/opt/vyatta/share/perl5/";
+use VyattaConfig;
 use VyattaMisc;
 use Getopt::Long;
 
 GetOptions("check-peer-name=s"      => \$peername,
+	   "check-as"		    => \$checkas,
+	   "check-peer-groups"	    => \$checkpeergroups,
+           "peergroup=s"    	    => \$pg,
+           "as=s"		    => \$as,
+           "neighbor=s"		    => \$neighbor,
 );
 
-if (defined $peername) { check_peer_name($peername); }
+if    (defined $peername) 	 	{ check_peer_name($peername); }
+elsif (defined $checkpeergroups &&
+       defined $pg &&
+       defined $as) 			{ check_for_peer_groups($pg, $as); }
+elsif (defined $neighbor && 
+       defined $as && 
+       defined $checkas && 
+       defined $pg) 		 	{ check_as($pg, $neighbor, $as); }
+elsif (defined $neighbor && 
+       defined $as && 
+       defined $checkas)  	 	{ check_as(-1, $neighbor, $as); }
+
 
 exit 0;
 
@@ -20,3 +37,67 @@ sub check_peer_name() {
   }
   exit 0;
 }
+
+sub check_for_peer_groups() {
+  my $config = new VyattaConfig;
+  my $pg = shift;
+  my $as = shift;
+  my $node;
+  my @peers, @neighbors;
+
+  $config->setLevel("protocols bgp $as neighbor");
+  my @neighbors = $config->listNodes();
+  
+  foreach $node (@neighbors) { 
+    my $peergroup = $config->returnValue("$node peer-group");
+    if ($peergroup eq $pg) { push @peers, $node; }
+  }
+
+  if (@peers) { 
+    foreach $node (@peers) {
+      print "neighbor $node uses peer-group $pg\n";
+    }
+    
+    print "please delete these peers before removing the peer-group\n";
+    exit 1;
+  }
+
+  exit 0;
+}
+
+sub check_as() {
+  my $pg = shift;
+  my $neighbor = shift;
+  my $as = shift;
+  my $config = new VyattaConfig;
+  my $pgtest = $neighbor;
+
+  $pgtest =~ s/\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}//;
+  if ($pgtest ne "") { return; }
+
+  $config->setLevel("protocols bgp $as neighbor $neighbor");
+  $remoteas = $config->returnValue("remote-as");
+
+  if (! defined $remoteas) {
+    if ($pg > 0) { 
+      $peergroup = 1; 
+      $peergroupas = 1;
+    }
+    else { 
+      $peergroup = $config->returnValue("peer-group"); 
+      $peergroupas = $config->returnValue(" .. $peergroup remote-as");
+    }
+
+    if (! defined $peergroup) {
+      print "You must define a remote-as or peer-group for neighbor $neighbor before commiting\n";
+      exit 1;
+    }
+
+    if (! defined $peergroupas) {
+      print "You must define a remote-as in neighbor $neighbor or peer-group $peergroup before commiting\n";
+      exit 1;
+    }
+  }
+
+  exit 0;
+}