vyatta-cfg-system: update new ntp updating routine to work with dhcp

Update the new NTP updating routines so they operate outside of
configuration mode (dhcp triggered updates).  This also requires
changes to the vyatta-dhcp3 package (see Bug #500)

Bug #94 http://bugzilla.vyos.net/show_bug.cgi?id=94

1 files changed, 24 insertions, 6 deletions

diff --git a/scripts/system/vyatta_update_ntp.pl b/scripts/system/vyatta_update_ntp.pl
index 2f9b29f1..f32a9407 100755
--- a/scripts/system/vyatta_update_ntp.pl
+++ b/scripts/system/vyatta_update_ntp.pl
@@ -22,8 +22,12 @@ use strict;
 use lib "/opt/vyatta/share/perl5";
 use Vyatta::Config;
 use NetAddr::IP;
+use Getopt::Long;
 
-die "$0 expects no arguments\n" if (@ARGV);
+my $dhclient_script = 0;
+
+GetOptions("dhclient-script=i" => \$dhclient_script,
+);
 
 sub ntp_format {
     my ($cidr_or_host) = @_;
@@ -72,22 +76,36 @@ foreach my $line (@ntp) {
    }
 }
 
-if ($cfg->exists("server")) {
+my @servers;
+my @clients;
+
+if ($dhclient_script == 1) {
+    @servers = $cfg->listOrigNodes("server");
+    @clients = $cfg->returnOrigValues("client address");
+} else {
+    @servers = $cfg->listNodes("server");
+    @clients = $cfg->returnValues("client address");
+}
+
+if (scalar(@servers) > 0) {
     print $output "# Servers\n\n";
-    foreach my $server ($cfg->listNodes("server")) {
+    foreach my $server (@servers) {
         my $server_addr = ntp_format($server);
         print $output "server $server_addr iburst";
         for my $property (qw(dynamic noselect preempt prefer)) {
-	    print $output " $property" if ($cfg->exists("server $server $property"));
+            if ($dhclient_script == 1) {
+                print $output " $property" if ($cfg->existsOrig("server $server $property"));
+            } else {
+                print $output " $property" if ($cfg->exists("server $server $property"));
+            }
         }
         print $output "\nrestrict $server_addr nomodify notrap nopeer noquery\n";
     }
     print $output "\n";
 }
 
-if ($cfg->exists("client")) {
+if (scalar(@clients) > 0) {
     print $output "# Clients\n\n";
-    my @clients = $cfg->returnValues("client address");
     foreach my $client (@clients) {
         my $address = ntp_format($client);
         print $output "restrict $address nomodify notrap nopeer\n";