arif/xcat-core
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
82a893d82b
xcat-core/xCAT/postscripts/xcataixpost
nott 14fc9d745e Move chomp inside conditional 
git-svn-id: https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@3158 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd
2009-04-09 17:49:05 +00:00

313 lines
8.8 KiB
Perl
Executable File
Raw Blame History

#!/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);
if ($tz) {
chomp $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);
$ENV{PATH}="/xcatpost:$ENV{PATH}";
&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);
if ($rpd) {
chomp $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;
}
Reference in New Issue View Git Blame Copy Permalink