xcat-core/xCAT/postscripts/xcataixscript

239 lines
7.0 KiB
Perl

#!/usr/bin/env perl -w
# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
#####################################################
#
# xCAT script resource for NIM (AIX nodes)
#
#####################################################
# if AIX - make sure we include perl 5.8.2 in INC path.
# Needed to find perl dependencies shipped in deps tarball.
if ($^O =~ /^aix/i) {
use lib "/usr/opt/perl5/lib/5.8.2/aix-thread-multi";
use lib "/usr/opt/perl5/lib/5.8.2";
use lib "/usr/opt/perl5/lib/site_perl/5.8.2/aix-thread-multi";
use lib "/usr/opt/perl5/lib/site_perl/5.8.2";
}
use File::Path;
use IO::Socket;
my $useSocketSSL=eval { require IO::Socket::SSL; };
use Getopt::Long;
use XML::Simple;
# 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;
# get hostname
$::shorthost = `hostname -s`;
chomp $::shorthost;
my $servnode;
# 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 xcataixscript: Could not find /etc/niminfo file.\n";
}
$servnode =~ s/^\s*//;
chomp $servnode;
# create the xcatpost dir
my $cmd = "mkdir -m 755 -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";
}
# need to find out what dir to mount - so need to get INSTALLDIR attribute
# from the node-specific script
# 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";
}
my $installdir;
if (-f $scriptname) {
my $cmd = "cat $scriptname | grep 'INSTALLDIR='";
&runcmd($cmd);
my $Sline = $::outref;
my $junk;
($junk, $installdir) = split(/=/, $Sline);
chomp $installdir;
$installdir =~ s/^\s*//;
$installdir =~ s/^(\'|\")(.*)(\"|\')$/$2/; # remove any surrounding quotes
}
if (!$installdir) {
$installdir="/install";
}
# get the contents of the $installdir/postscripts dir on the server
# - mount dir from server and copy files
my $mcmd = "mkdir -p /xcatmnt; mount -o nolock $servnode:$installdir/postscripts /xcatmnt";
if (&runcmd($mcmd) != 0) {
print "$::sdate xcataixpost: Could not mount $installdir/postscripts from $servnode.\n";
print $::LOG_FILE "$::sdate xcataixpost: Could not mount $installdir/postscripts from $servnode.\n";
}
my $cpcmd;
if ((@ARGV==0) || ($ARGV[0] != 2)) {
$cpcmd = "cp -r /xcatmnt/* /xcatpost >/dev/null 2>&1";
} 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";
}
# 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";
}
my $ucmd = "umount /xcatmnt; rmdir /xcatmnt";
if (&runcmd($ucmd) != 0) {
print "$::sdate xcataixpost: Could not unmount $installdir.\n";
print $::LOG_FILE "$::sdate xcataixpost: Could not unmount $installdir/postscripts.\n";
}
# add xcatd to /etc/inittab???
# see if it is already there
my $lsicmd = "/usr/sbin/lsitab xcat > /dev/null 2>&1";
if (&runcmd($lsicmd) != 0) {
# ok - create the entry
my $mkitab_cmd = 'mkitab "xcat:2:wait:/xcatpost/xcataixpost > /dev/console 2>&1"';
if (&runcmd($mkitab_cmd) != 0) {
print "$::sdate xcataixpost: Could not add xcataixpost to /etc/inittab.\n";
print $::LOG_FILE "$::sdate xcataixpost: Could not add xcataixpost to /etc/inittab.\n";
}
}
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 = "3001";
# open a socket to request credentials
my $remote = IO::Socket::SSL->new(
PeerAddr => $servnode,
PeerPort => $port,
Proto => 'tcp',
);
unless ($remote) {
print "$::sdate xcataixpost: Cannot connect to host \'$servnode\'\n";
print $::LOG_FILE "$::sdate xcataixpost: Cannot connect to host \'$servnode\'\n";
return 1;
}
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;
}
# setup the perl library path for xcat-dep packages
# this is necessary for AIX 5.3V/61H and beyond
# but does not hurt old AIX versions
print POSTSCRIPT "PERL5LIB='/usr/opt/perl5/lib/5.8.2:/usr/opt/perl5/lib/5.8.2/aix-thread-multi:/usr/opt/perl5/lib/site_perl/5.8.2:/usr/opt/perl5/lib/site_perl/5.8.2/aix-thread-multi'\n";
print POSTSCRIPT "export PERL5LIB\n";
# request must be in XML format
print $remote "<xcatrequest>\n";
print $remote " <command>getpostscript</command>\n";
print $remote "</xcatrequest>\n";
# get reponse in XML format
my $response='';
my $rsp;
while (<$remote>) {
$response .= $_;
if ($response =~ m/<\/xcatresponse>/) {
$rsp = eval { XMLin($response,SuppressEmpty=>undef,ForceArray=>1) };
if ($rsp->{serverdone}) {
last;
}
foreach my $line (@{$rsp->{data}}) {
$line =~ s/^\s+//; # strip any leading spaces
print POSTSCRIPT "$line";
}
$response='';
}
}
close(POSTSCRIPT);
close $remote;
return 0;
}
#####################################################
#
# run the command
#
#####################################################
sub runcmd
{
my ($cmd) = @_;
my $rc=0;
$cmd .= ' 2>&1' ;
$::outref = `$cmd`;
if ($?)
{
$rc = $? >> 8;
if ($rc > 0)
{
print "$::sdate xcataixpost: $::outref\n";
print $::LOG_FILE "$::sdate xcataixpost: $::outref\n";
}
}
return $rc;
}