#!/usr/bin/env perl -w
# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
#####################################################
#
#   xCAT post script for AIX nodes
#
#               This script is run from the /etc/inittab file on both
#               diskless and diskfull AIX nodes
#
#               It is also run by the updatenode cmd
#
#####################################################

use File::Path;
use IO::Socket;
use Getopt::Long;

sleep int(rand(10));

# since we don't have syslog set up yet we'll
# just save msgs in a local log file
$logdir = "/var/log/xcat";

if (!-d $logdir) {
	mkpath($logdir);
}

$::sdate = `/bin/date`;
chomp $::sdate;
my $logfile = $logdir . "/xcat.log";

# this log should not contain much so it might be ok to let it grow?
# at least we'll have the errors preserved
open(LOGFILE,">>",$logfile);
$::LOG_FILE = \*LOGFILE;

#
# Process the command line...
#
#  the -c means redo the copy of scripts from the server
if (!GetOptions('copy|c' => \$::opt_c,))
{    # Gather options
        exit 1;
}

# get hostname
$::shorthost = `hostname -s`;
chomp $::shorthost;

my $servnode;
if (-f "/etc/xcatinfo") {
	# if this file exists assume it has the correct server name
	my $cmd = "cat /etc/xcatinfo | grep 'XCATSERVER'";
	&runcmd($cmd);
	my $SNline = $::outref;
	my $junk;
	($junk, $servnode) =  split(/=/, $SNline);
} else {
	# get the name of my service node/NIM master from the /etc/niminfo file
	if (-f "/etc/niminfo") {
    	$cmd = "cat /etc/niminfo | grep 'NIM_MASTER_HOSTNAME'";
    	&runcmd($cmd);
    	my $SNline = $::outref;
		my $junk;
    	($junk, $servnode) =  split(/=/, $SNline);

		# save the servnode from niminfo in xcatinfo
		my $xcatinfo="/etc/xcatinfo";
		open(XCATINFO,">",$xcatinfo);
		print XCATINFO "XCATSERVER=$servnode\n";
		close(XCATINFO);
	} else {
		print "$::sdate  xcataixpost: Could not find /etc/niminfo file.\n";
    	print $::LOG_FILE "$::sdate  xcataixpost: Could not find /etc/niminfo file.\n";
	}
}
$servnode =~ s/^\s*//;
chomp $servnode;

# if the /xcatpost dir has not been populated or the -c option is
#   specified then do the mount and copy of /install/postscripts
if ( !(-f "/xcatpost/xcataixpost") || $::opt_c ) {
	# create the xcatpost dir
	my $cmd = "mkdir -p /xcatpost";
	if (&runcmd($cmd) != 0) {
		print "$::sdate xcataixpost: Could not make the /xcatpost directory.\n";
		print $::LOG_FILE "$::sdate xcataixpost: Could not make the /xcatpost directory.\n";
	}

	# get the contents of the /install/postscripts dir on the server
	#   - mount dir from server and copy files
	my $mcmd = "mkdir -p /xcatmnt; mount $servnode:/install/postscripts /xcatmnt";
	if (&runcmd($mcmd) != 0) {
    	print "$::sdate xcataixpost: Could not mount /install/postscripts from $servnode.\n";
    	print $::LOG_FILE "$::sdate xcataixpost: Could not mount /install/postscripts from $servnode.\n";
	}

	if ((@ARGV==0) || ($ARGV[0] != 2)) {
    	$cpcmd = "cp -r /xcatmnt/* /xcatpost 2>/dev/null";
	} else {
    	# when argv[1]=2, there is only one postscript file,
    	#   user wants only download it to save time
    	$cpcmd= "cp /xcatmnt/$ARGV[1] /xcatpost >/dev/null 2>&1";
	}

	if (&runcmd($cpcmd) != 0) {
	#    print "$::sdate xcataixpost: Could not copy postscripts to /xcatpost.\n";
	#	print $::LOG_FILE "$::sdate xcataixpost: Could not copy postscripts to /xcatpost.\n";
	}

	my $ucmd = "umount /xcatmnt; rmdir /xcatmnt";
	if (&runcmd($ucmd) != 0) {
    	print "$::sdate xcataixpost: Could not unmount /install.\n";
    	print $::LOG_FILE "$::sdate xcataixpost: Could not unmount /install/postscripts.\n";
	}

} # end recopy postscripts


# request the xCAT postscript for this particular node
$scriptname = "/xcatpost/myxcatpost_" . $::shorthost;
if (&getmypost != 0) {
	print "$::sdate xcataixpost: Could not get the xCAT post script for this node.\n";
	print $::LOG_FILE "$::sdate xcataixpost: Could not get the xCAT post script for this node.\n";
}

# make sure all are executable
my $chcmd = "chmod +x /xcatpost/*";
if (&runcmd($chcmd) != 0) {
	print "$::sdate xcataixpost: Could not change /xcatpost file permissions.\n";
	print $::LOG_FILE "$::sdate xcataixpost: Could not change /xcatpost file permissions.\n";
}

if (-f $scriptname) {
	# set the timezone
    my $cmd = "cat $scriptname | grep 'TIMEZONE='";
    &runcmd($cmd);
    my $TZline = $::outref;
    my ($junk, $tz) =  split(/=/, $TZline);
	chomp $tz;
    if ($tz) {
        my $tzcmd = qq~/usr/bin/chtz $tz >/dev/null 2>&1~;
        if (&runcmd($tzcmd) != 0) {
            $msg = "$::sdate xcataixpost: Could not set timezone.\n";
            `logger -t xcat $msg`;
        } else {
			&runcmd("export TZ=$tz");
		}
    }
}

# check & run the postscript
my $nodesetstat="standalone";
if (-f $scriptname)
{
    if (@ARGV>1) {
      my $scripts=$ARGV[1];
      my $POSTS=join('\n', split(',', $scripts));
      #print "scripts=$scripts\n";
      #remove all the postscripts
      my $TMP=`sed "/postscripts-start-here/,/postscripts-end-here/ d" $scriptname`;
      `echo "$TMP" > $scriptname`;
      #add requested postscripts in
      `echo "$POSTS" | tr "," "\n" >> $scriptname`;
   }

   $nodesetstat=`grep "NODESETSTATE=" $scriptname|awk -F \= '{print \$2}'`;
   chomp($nodesetstat);

    &runcmd("cd /xcatpost;$scriptname");
} else {
	print "$::sdate  xcataixpost: Could not find post script for $::shorthost.\n";
	print $::LOG_FILE "$::sdate  xcataixpost: Could not find post script for $::shorthost.\n";
}

if (@ARGV<1) {
  my $update=1;
  if (-f $scriptname) {
      my $CNS=`grep NODESTATUS= $scriptname`;
      if ((!$?) && ($CNS =~ /\'0\'|\'N\'|\'n\'/)) { $update=0;}  #only update is nonodestatus is not set in the site table
  }
  if ($update) {
      if (&updateflag($nodesetstat) != 0) {
	  print "$::sdate xcataixpost: Failed to update the xCAT server.\n";
	  print $::LOG_FILE "$::sdate xcataixpost: Failed to update the xCAT server..\n";
      }
  }
} else {
  #called by updatenode, infrom user it is done
  print "returned";
}

if (-f $scriptname) {

	# Set the local password - if provided
	my $cmd = "cat $scriptname | grep 'ROOTPW='";
    &runcmd($cmd);
    my $PWline = $::outref;
    my ($junk, $rpd) =  split(/=/, $PWline);
	chomp $rpd;
	if ($rpd) {
		my $pwcmd = qq~echo "root:$rpd" | /bin/chpasswd -c >/dev/null 2>&1~;
		if (&runcmd($pwcmd) != 0) {
			$msg = "$::sdate xcataixpost: Could not set password.\n";
			`logger -t xcat $msg`;
		}
	}
}

close($::LOG_FILE);

exit 0;

#####################################################
#
#   getmypost
#               Get the xCAT post script info for this node
#					and write it to a file 
#
#####################################################
sub  getmypost {

	my $port = "3002";
	
	my $remote = IO::Socket::INET->new( Proto => "tcp", PeerAddr  => $servnode, PeerPort  => $port, );

	unless ($remote) { 
		print "$::sdate  xcataixpost: Cannot connect to host \'$servnode\'\n";
		print $::LOG_FILE "$::sdate  xcataixpost: Cannot connect to host \'$servnode\'\n";
		return 1;
	}

	$remote->autoflush(1);
	print $remote "getpostscript\n";

	if (!open(POSTSCRIPT, ">$scriptname") ) {
		print "$::sdate  xcataixpost: Could not open $scriptname.\n";
		print $::LOG_FILE "$::sdate  xcataixpost: Could not open $scriptname.\n";
		close $remote;
		return 1;
	}

	my $line;
	while (defined ($line = <$remote>)) {
		chomp $line;
		if ( ($line eq "ready") || ($line eq "done")) {
			next;
		}
		print POSTSCRIPT "$line\n";
	}
	close(POSTSCRIPT);
	close $remote;
	return 0;
}

############################################################
#
#   updateflag
#     Tells xCAT on the server that the post scripts is done.      
#
############################################################
sub  updateflag {
    my $nodesetstat=shift;
    my $state="booted";
    if ($nodesetstat eq "standalone") { $state="installed booting"; }

 
    my $port = "3002";
    my $remote = IO::Socket::INET->new( Proto => "tcp", PeerAddr  => $servnode, PeerPort  => $port, );
    unless ($remote) { 
        print "$::sdate  xcataixpost: Cannot connect to host \'$servnode\'\n";
        print $::LOG_FILE "$::sdate  xcataixpost: Cannot connect to host \'$servnode\'\n";
	return 1;
    }
    $remote->autoflush(1);

    my $line;
    while (defined ($line = <$remote>)) {
        chomp $line;
        if ($line eq "ready") {
           print $remote "installstatus $state\n";
        } elsif ($line eq "done") {
	    last;
        }
    }
    close $remote;
    return 0;
}

#####################################################
#
# run the command
#
#####################################################
sub runcmd
{
    my ($cmd) = @_;
	my $rc=0;
    $cmd .= ' 2>&1' ;
    $::outref = [];
    $::outref = `$cmd`;
    if ($?)
    {
        $rc = $? >> 8;
        if ($rc > 0)
        {
	#		print "$::sdate  xcataixpost: $::outref\n";
	#		print $::LOG_FILE "$::sdate  xcataixpost: $::outref\n";
        }
    }
 return $rc;
}