summaryrefslogtreecommitdiff
path: root/scripts/VyattaClusterConfig.pm
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/VyattaClusterConfig.pm')
-rw-r--r--scripts/VyattaClusterConfig.pm312
1 files changed, 312 insertions, 0 deletions
diff --git a/scripts/VyattaClusterConfig.pm b/scripts/VyattaClusterConfig.pm
new file mode 100644
index 0000000..affb3e8
--- /dev/null
+++ b/scripts/VyattaClusterConfig.pm
@@ -0,0 +1,312 @@
+package VyattaClusterConfig;
+
+use strict;
+use lib "/opt/vyatta/share/perl5/";
+use VyattaConfig;
+
+my $DEFAULT_INITDEAD = 120;
+my $DEFAULT_LOG_FACILITY = 'daemon';
+my $SERVICE_DIR = "/etc/init.d";
+
+my %fields = (
+ _interface => undef,
+ _pre_shared => undef,
+ _keepalive_itvl => undef,
+ _dead_itvl => undef,
+ _groups => {},
+ _is_empty => 1,
+);
+
+sub new {
+ my $that = shift;
+ my $class = ref ($that) || $that;
+ my $self = {
+ %fields,
+ };
+
+ bless $self, $class;
+ return $self;
+}
+
+sub setup {
+ my ( $self, $level ) = @_;
+ my $config = new VyattaConfig;
+
+ $config->setLevel("$level");
+ my @nodes = $config->listNodes();
+ if (scalar(@nodes) <= 0) {
+ $self->{_is_empty} = 1;
+ return 0;
+ } else {
+ $self->{_is_empty} = 0;
+ }
+
+ my @tmp = $config->returnValues("interface");
+ $self->{_interface} = [ @tmp ];
+ $self->{_pre_shared} = $config->returnValue("pre-shared-secret");
+ $self->{_keepalive_itvl} = $config->returnValue("keepalive-interval");
+ $self->{_dead_itvl} = $config->returnValue("dead-interval");
+
+ $config->setLevel("$level group");
+ my @groups = $config->listNodes();
+ my $group;
+ for $group (@groups) {
+ my $hashref = {};
+ $config->setLevel("$level group $group");
+ $hashref->{_primary} = $config->returnValue("primary");
+ @tmp = $config->returnValues("secondary");
+ $hashref->{_secondary} = [ @tmp ];
+ $hashref->{_auto_failback} = $config->returnValue("auto-failback");
+ @tmp = $config->returnValues("monitor");
+ $hashref->{_monitor} = [ @tmp ];
+ @tmp = $config->returnValues("service");
+ $hashref->{_service} = [ @tmp ];
+ $self->{_groups}->{$group} = $hashref;
+ }
+
+ return 0;
+}
+
+sub setupOrig {
+ my ( $self, $level ) = @_;
+ my $config = new VyattaConfig;
+
+ $config->setLevel("$level");
+ my @nodes = $config->listOrigNodes();
+ if (scalar(@nodes) <= 0) {
+ $self->{_is_empty} = 1;
+ return 0;
+ } else {
+ $self->{_is_empty} = 0;
+ }
+
+ my @tmp = $config->returnOrigValues("interface");
+ $self->{_interface} = [ @tmp ];
+ $self->{_pre_shared} = $config->returnOrigValue("pre-shared-secret");
+ $self->{_keepalive_itvl} = $config->returnOrigValue("keepalive-interval");
+ $self->{_dead_itvl} = $config->returnOrigValue("dead-interval");
+
+ $config->setLevel("$level group");
+ my @groups = $config->listOrigNodes();
+ my $group;
+ for $group (@groups) {
+ my $hashref = {};
+ $config->setLevel("$level group $group");
+ $hashref->{_primary} = $config->returnOrigValue("primary");
+ @tmp = $config->returnOrigValues("secondary");
+ $hashref->{_secondary} = [ @tmp ];
+ $hashref->{_auto_failback} = $config->returnOrigValue("auto-failback");
+ @tmp = $config->returnOrigValues("monitor");
+ $hashref->{_monitor} = [ @tmp ];
+ @tmp = $config->returnOrigValues("service");
+ $hashref->{_service} = [ @tmp ];
+ $self->{_groups}->{$group} = $hashref;
+ }
+
+ return 0;
+}
+
+sub primaryNode {
+ my ($self) = @_;
+ my @groups = keys %{$self->{_groups}};
+ my $hashref = $self->{_groups}->{$groups[0]};
+ return $hashref->{_primary};
+}
+
+sub secondaryNode {
+ my ($self) = @_;
+ my @groups = keys %{$self->{_groups}};
+ my $hashref = $self->{_groups}->{$groups[0]};
+ return ${$hashref->{_secondary}}[0];
+}
+
+sub monitorNodes {
+ my ($self) = @_;
+ my @groups = keys %{$self->{_groups}};
+ my $hashref = $self->{_groups}->{$groups[0]};
+ return @{$hashref->{_monitor}};
+}
+
+sub serviceStr {
+ my ($self) = @_;
+ my @groups = keys %{$self->{_groups}};
+ my $hashref = $self->{_groups}->{$groups[0]};
+ return (join " ", @{$hashref->{_service}});
+}
+
+sub isEmpty {
+ my ($self) = @_;
+ return $self->{_is_empty};
+}
+
+sub authkeys {
+ my ($self) = @_;
+ my $key = $self->{_pre_shared};
+ return (undef, "pre-shared secret not defined") if (!defined($key));
+ my $str =<<EOS;
+auth 1
+1 sha1 $key
+EOS
+ return ($str, undef);
+}
+
+sub check_interfaces {
+ my @interfaces = @_;
+ foreach (@interfaces) {
+ system("ip addr show $_ >& /dev/null");
+ if ($? >> 8) {
+ return "interface $_ does not exist";
+ }
+ }
+ return undef;
+}
+
+sub ha_cf {
+ my ($self) = @_;
+ my @groups = keys %{$self->{_groups}};
+ return (undef, "no resource group defined") if ($#groups < 0);
+ return (undef, "using multiple resource groups is not supported yet")
+ if ($#groups > 0);
+
+ my $ierr = check_interfaces(@{$self->{_interface}});
+ if (defined($ierr)) {
+ return (undef, $ierr);
+ }
+ my $interfaces = join " ", @{$self->{_interface}};
+ my $kitvl = $self->{_keepalive_itvl};
+ my $ditvl = $self->{_dead_itvl};
+
+ my $hashref = $self->{_groups}->{$groups[0]};
+ my $primary = $hashref->{_primary};
+ my @secondaries = @{$hashref->{_secondary}};
+ my $pings = join " ", @{$hashref->{_monitor}};
+ my $auto_failback = ($hashref->{_auto_failback} eq "true") ?
+ "on" : "off";
+ my $my_name = `uname -n`;
+ chomp $my_name;
+
+ return (undef, "heartbeat interface(s) not defined") if ($interfaces eq "");
+ return (undef, "keepalive interval not defined") if (!defined($kitvl));
+ return (undef, "dead interval not defined") if (!defined($ditvl));
+ return (undef, "cluster primary system not defined")
+ if (!defined($primary));
+ return (undef, "cluster secondary node(s) not defined")
+ if ($#secondaries < 0);
+ return (undef, "using multiple secondary nodes is not supported yet")
+ if ($#secondaries > 0);
+ return (undef,
+ "dead interval must be more than twice the keepalive interval")
+ if ($ditvl <= (2 * $kitvl));
+ return (undef,
+ "dead interval must be smaller than $DEFAULT_INITDEAD seconds")
+ if ($ditvl >= $DEFAULT_INITDEAD);
+ return (undef,
+ "the current node '$my_name' is not defined in the configuration")
+ if (($my_name ne $primary) && ($my_name ne $secondaries[0]));
+
+ my $monitor_str = "";
+ if ($pings ne "") {
+ $monitor_str = "\nping $pings\n"
+ . "respawn hacluster /usr/lib/heartbeat/ipfail";
+ }
+
+ my $wtime = int($kitvl * 2);
+ if ($wtime > $ditvl) {
+ $wtime = $ditvl;
+ }
+
+ my $str =<<EOS;
+keepalive $kitvl
+deadtime $ditvl
+warntime $wtime
+initdead $DEFAULT_INITDEAD
+logfacility $DEFAULT_LOG_FACILITY
+bcast $interfaces
+auto_failback $auto_failback
+node $primary $secondaries[0]$monitor_str
+EOS
+
+ return ($str, undef);
+}
+
+sub isValidIPv4 {
+ my $str = shift;
+ return 0 if (!($str =~ m/^(\d+)\.(\d+)\.(\d+)\.(\d+)$/));
+ my ($a1, $a2, $a3, $a4) = ($1, $2, $3, $4);
+ return 1 if (($a1 >= 0 && $a1 < 256) && ($a2 >= 0 && $a2 < 256)
+ && ($a3 >= 0 && $a3 < 256) && ($a4 >= 0 && $a4 < 256));
+ return 0;
+}
+
+my @service_list = ();
+sub isValidService {
+ my $service = shift;
+ if (scalar(@service_list) == 0) {
+ opendir(SDIR, "$SERVICE_DIR")
+ or (print STDERR "Error: can't open $SERVICE_DIR" && return 0);
+ @service_list = grep !/^\./, readdir SDIR;
+ }
+ return 1 if (grep /^$service$/, @service_list);
+ return 0;
+}
+
+sub haresources {
+ my ($self) = @_;
+ my @groups = keys %{$self->{_groups}};
+ return (undef, "no resource group defined") if ($#groups < 0);
+ return (undef, "using multiple resource groups is not supported yet")
+ if ($#groups > 0);
+
+ my $hashref = $self->{_groups}->{$groups[0]};
+ my $primary = $hashref->{_primary};
+
+ my @init_services = ();
+ my @ip_addresses = ();
+ foreach (@{$hashref->{_service}}) {
+ if (!isValidIPv4($_)) {
+ if (isValidService($_)) {
+ push @init_services, $_;
+ } else {
+ return (undef, "\"$_\" is not a valid IP address or service name");
+ }
+ } else {
+ push @ip_addresses, $_;
+ }
+ }
+ # this forces all ip addresses to be before all services, which may not
+ # be the desirable behavior in all cases.
+ my $ip_str = join " ", @ip_addresses;
+ my $serv_str = join " ", @init_services;
+ my $services = join " ", ($ip_str, $serv_str);
+ return (undef, "cluster primary system not defined") if (!defined($primary));
+ return (undef, "cluster service(s) not defined") if ($services eq "");
+
+ my $str =<<EOS;
+$primary $services
+EOS
+
+ return ($str, undef, @init_services);
+}
+
+sub print_str {
+ my ($self) = @_;
+ my $str = "cluster";
+ $str .= "\n interface " . (join ",", @{$self->{_interface}});
+ $str .= "\n pre-shared-secret $self->{_pre_shared}";
+ $str .= "\n keepalive-interval $self->{_keepalive_itvl}";
+ $str .= "\n dead-interval $self->{_dead_itvl}";
+ my $group;
+ foreach $group (keys %{$self->{_groups}}) {
+ $str .= "\n group $group";
+ my $hashref = $self->{_groups}->{$group};
+ $str .= "\n primary $hashref->{_primary}";
+ $str .= "\n secondary " . (join ",", @{$hashref->{_secondary}});
+ $str .= "\n monitor " . (join ",", @{$hashref->{_monitor}});
+ $str .= "\n service " . (join ",", @{$hashref->{_service}});
+ }
+
+ return $str;
+}
+
+1;
+