#!/usr/bin/perl -w
#
# Module: vyatta-wanloadbalance.pl
# 
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License version 2 as published
# by the Free Software Foundation.
# 
# **** End License ****
#
use lib "/opt/vyatta/share/perl5/";
use VyattaConfig;
use VyattaMisc;

use warnings;
use strict;
use POSIX;
use File::Copy;

sub write_health {
#open conf
    my $config = new VyattaConfig;

    my $valid = "false";

    $config->setLevel("load-balancing wan interface-health");
    my @eths = $config->listNodes();
    
    print FILE_LCK "health {\n";

    foreach my $ethNode (@eths) {
	
	print FILE_LCK "\tinterface " . $ethNode . " {\n";
	
	my $option = $config->returnValue("$ethNode failure-count");
	if (defined $option) {
	    print FILE_LCK "\t\tfailure-ct  " . $option . "\n";
	}
	
	$option = $config->returnValue("$ethNode ping");
	if (defined $option) {
	    print FILE_LCK "\t\ttarget  " . $option . "\n";
	    $valid = "true";
	}
	
	$option = $config->returnValue("$ethNode resp-time");
	if (defined $option) {
	    print FILE_LCK "\t\tping-resp  " . $option*1000 . "\n";
	}
	
	$option = $config->returnValue("$ethNode success-count");
	if (defined $option) {
	    print FILE_LCK "\t\tsuccess-ct  " . $option . "\n";
	}
	print FILE_LCK "\t}\n";
    }
    print FILE_LCK "}\n\n";

    if ($valid eq "false") {
	print "valid wan load-balance configuration requires an interface with a ping target\n";
    }
    return $valid;
}

sub write_rules {
    my $config = new VyattaConfig;

    my $valid = "false";

    $config->setLevel('load-balancing wan rule');
    my @rules = $config->listNodes();
    
    #destination
    foreach my $rule (@rules) {
	print FILE_LCK "rule " . $rule . " {\n";
	
	$config->setLevel('load-balancing wan rule');

	my $protocol = $config->returnValue("$rule protocol");
	if (defined $protocol) {
	    print FILE_LCK "\tprotocol " . $protocol . "\n"
	}
	else {
	    $protocol = "";
	}

	#destination
	print FILE_LCK "\tdestination {\n";
	my $daddr = $config->returnValue("$rule destination address");
	if (defined $daddr) {
	    print FILE_LCK "\t\taddress " . $daddr . "\n";
	}

	my $dnet = $config->returnValue("$rule destination network");
	if (defined $dnet && !defined $daddr) {
	    print FILE_LCK "\t\tnetwork " . $dnet . "\n";
	}
	elsif (defined $dnet && defined $daddr) {
	    print "Please specify either destination address or source network\n";
	    exit 2;
	}

	my $option = $config->returnValue("$rule destination port");
	if (defined $option) {
	    if ($protocol ne "tcp" && $protocol ne "udp") {
		print "Please specify protocol tcp or udp when configuring ports\n";
		exit 2;
	    }
	    print FILE_LCK "\t\tport " . $option . "\n";
	}

	print FILE_LCK "\t}\n";

	#source
	$config->setLevel('load-balancing wan rule');


	print FILE_LCK "\tsource {\n";
	my $saddr = $config->returnValue("$rule source address");
	if (defined $saddr) {
	    print FILE_LCK "\t\taddress " . $saddr . "\n";
	}

	my $snet = $config->returnValue("$rule source network");
	if (defined $snet && !defined $saddr) {
	    print FILE_LCK "\t\tnetwork " . $snet . "\n";
	}
	elsif (defined $snet && defined $saddr) {
	    print "Please specify either source address or source network\n";
	    exit 2;
	}

	$option = $config->returnValue("$rule source port");
	if (defined $option) {
	    if ($protocol ne "tcp" && $protocol ne "udp") {
		print "Please specify protocol tcp or udp when configuring ports\n";
		exit 2;
	    }
	    print FILE_LCK "\t\tport " . $option . "\n";
	}
	print FILE_LCK "\t}\n";

	#interface
	$config->setLevel("load-balancing wan rule $rule interface");
	my @eths = $config->listNodes();
	
	foreach my $ethNode (@eths) {

	    $valid = "true";

	    print FILE_LCK "\tinterface " . $ethNode . " {\n";
	    
	    $option = $config->returnValue("$ethNode weight");
	    if (defined $option) {
		print FILE_LCK "\t\tweight  " . $option . "\n";
	    }
	    print FILE_LCK "\t}\n";
	}
	print FILE_LCK "}\n";
    }

    if ($valid eq "false") {
	print "At least one interface must be configured for a rule to be valid\n";
    }
    return $valid;
}



####main
my $conf_file = '/var/load-balance/wlb.conf';
my $conf_lck_file = '/var/load-balance/wlb.conf.lck';

#open file
open(FILE, "<$conf_file") or die "Can't open wlb config file"; 
open(FILE_LCK, "+>$conf_lck_file") or die "Can't open wlb lock file";

my $success = write_health();
if ($success eq "false") {
    exit 1;
}

$success = write_rules();
if ($success eq "false") {
    exit 1;
}

close FILE;
close FILE_LCK;

copy ($conf_lck_file,$conf_file);
unlink($conf_lck_file);


#finally kick the process
system "/opt/vyatta/sbin/vyatta-wanloadbalance.init restart $conf_file 2>/dev/null";

exit 0;