66a4ef81e6
git-svn-id: https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@4439 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd
399 lines
12 KiB
Perl
399 lines
12 KiB
Perl
# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
|
|
#-------------------------------------------------------
|
|
|
|
=head1
|
|
xCAT plugin package to handle xdsh
|
|
|
|
Supported command:
|
|
xdsh-> dsh
|
|
xdcp-> dcp
|
|
|
|
=cut
|
|
|
|
#-------------------------------------------------------
|
|
package xCAT_plugin::xdsh;
|
|
use strict;
|
|
use Storable qw(dclone);
|
|
use File::Basename;
|
|
use File::Path;
|
|
require xCAT::Table;
|
|
|
|
require xCAT::Utils;
|
|
|
|
require xCAT::MsgUtils;
|
|
use Getopt::Long;
|
|
require xCAT::DSHCLI;
|
|
1;
|
|
|
|
#-------------------------------------------------------
|
|
|
|
=head3 handled_commands
|
|
|
|
Return list of commands handled by this plugin
|
|
|
|
=cut
|
|
|
|
#-------------------------------------------------------
|
|
|
|
sub handled_commands
|
|
{
|
|
return {
|
|
xdsh => "xdsh",
|
|
xdcp => "xdsh"
|
|
};
|
|
}
|
|
|
|
#-------------------------------------------------------
|
|
|
|
=head3 preprocess_request
|
|
|
|
Check and setup for hierarchy
|
|
|
|
=cut
|
|
|
|
#-------------------------------------------------------
|
|
sub preprocess_request
|
|
{
|
|
my $req = shift;
|
|
my $cb = shift;
|
|
my %sn;
|
|
my $sn;
|
|
my $command = $req->{command}->[0]; # xdsh vs xdcp
|
|
|
|
#if already preprocessed, go straight to request
|
|
if ($req->{_xcatpreprocessed}->[0] == 1) { return [$req]; }
|
|
my $nodes = $req->{node};
|
|
my $service = "xcat";
|
|
my @requests;
|
|
my $syncsn = 0;
|
|
my $syncsnfile;
|
|
my $dcppull = 0;
|
|
|
|
# read the environment variables for rsync setup
|
|
foreach my $envar (@{$req->{env}})
|
|
{
|
|
my ($var, $value) = split(/=/, $envar, 2);
|
|
if ($var eq "RSYNCSNONLY")
|
|
{ # syncing SN, will change noderange to list of SN
|
|
# we are only syncing the service node ( -s flag)
|
|
$syncsn = 1;
|
|
}
|
|
if ($var eq "DSH_RSYNC_FILE") # from -F flag
|
|
{ # if hierarchy,need to copy file to the SN
|
|
$syncsnfile = $value; # in the new /tmp/xcatrf.tmp
|
|
}
|
|
if ($var eq "DCP_PULL") # from -P flag
|
|
{
|
|
$dcppull = 1; # TBD handle pull hierarchy
|
|
}
|
|
}
|
|
|
|
# find service nodes for requested nodes
|
|
# build an individual request for each service node
|
|
# find out the names for the Management Node
|
|
my @MNnodeinfo = xCAT::Utils->determinehostname;
|
|
my $MNnodename = pop @MNnodeinfo; # hostname
|
|
my @MNnodeipaddr = @MNnodeinfo; # ipaddresses
|
|
my $mnname = $MNnodeipaddr[0];
|
|
my $tmpsyncsnfile = "/tmp/xcatrf.tmp";
|
|
my $SNpath;
|
|
|
|
my $synfiledir = "/var/xcat/syncfiles"; # default
|
|
if ($nodes)
|
|
{
|
|
$sn = xCAT::Utils->get_ServiceNode($nodes, $service, "MN");
|
|
my @snodes;
|
|
my @snoderange;
|
|
|
|
# check to see if service nodes and not just the MN
|
|
if ($sn)
|
|
{
|
|
foreach my $snkey (keys %$sn)
|
|
{
|
|
if (!grep(/$snkey/, @MNnodeipaddr))
|
|
{ # if not the MN
|
|
push @snodes, $snkey;
|
|
$snoderange[0] .= "$snkey,";
|
|
|
|
}
|
|
}
|
|
|
|
if (@snodes)
|
|
{
|
|
|
|
# get the directory on the servicenode to put the files in
|
|
my @syndir = xCAT::Utils->get_site_attribute("SNsyncfiledir");
|
|
if ($syndir[0])
|
|
{
|
|
$synfiledir = $syndir[0];
|
|
}
|
|
|
|
# if -F command and service nodes first need to rsync the SN
|
|
if ($syncsnfile)
|
|
{
|
|
|
|
#change noderange to the service nodes
|
|
my $addreq;
|
|
chop $snoderange[0];
|
|
$addreq->{'_xcatdest'} = $mnname;
|
|
$addreq->{node} = \@snodes;
|
|
$addreq->{noderange} = \@snoderange;
|
|
$addreq->{arg}->[0] = "-s";
|
|
$addreq->{arg}->[1] = "-F";
|
|
$addreq->{arg}->[2] = $syncsnfile;
|
|
$addreq->{command}->[0] = $req->{command}->[0];
|
|
$addreq->{cwd}->[0] = $req->{cwd}->[0];
|
|
push @requests, $addreq;
|
|
|
|
# need to add to the queue to copy rsync file( -F input)
|
|
# to the service node to the /tmp/xcatrf.tmp file
|
|
my $addreq;
|
|
$addreq->{'_xcatdest'} = $mnname;
|
|
$addreq->{node} = \@snodes;
|
|
$addreq->{noderange} = \@snoderange;
|
|
$addreq->{arg}->[0] = $syncsnfile;
|
|
$addreq->{arg}->[1] = $tmpsyncsnfile;
|
|
$addreq->{command}->[0] = $req->{command}->[0];
|
|
$addreq->{cwd}->[0] = $req->{cwd}->[0];
|
|
push @requests, $addreq;
|
|
}
|
|
else
|
|
{
|
|
|
|
# if other xdcp command
|
|
# mk the diretory on the SN to hold the files
|
|
# to be sent to the CN.
|
|
# build a command to update the service nodes
|
|
# change the destination to the tmp location on
|
|
# the service node, if not pull function
|
|
if (($command eq "xdcp") && ($dcppull == 0))
|
|
{
|
|
|
|
#make the needed directory on the service node
|
|
# create new directory for path on Service Node
|
|
my $frompath = $req->{arg}->[-2];
|
|
$SNpath = $synfiledir;
|
|
$SNpath .= $frompath;
|
|
my $SNdir;
|
|
$SNdir = dirname($SNpath); # get directory
|
|
my $addreq= dclone($req);
|
|
$addreq->{'_xcatdest'} = $mnname;
|
|
$addreq->{node} = \@snodes;
|
|
$addreq->{noderange} = \@snoderange;
|
|
$addreq->{arg}->[0] = "mkdir ";
|
|
$addreq->{arg}->[1] = "-p ";
|
|
$addreq->{arg}->[2] = $SNdir;
|
|
$addreq->{command}->[0] = "xdsh";
|
|
$addreq->{cwd}->[0] = $req->{cwd}->[0];
|
|
push @requests, $addreq;
|
|
|
|
# now sync file to the service node to the new
|
|
# tmp path
|
|
my $addreq = dclone($req);
|
|
$addreq->{'_xcatdest'} = $mnname;
|
|
chop $snoderange[0];
|
|
$addreq->{node} = \@snodes;
|
|
$addreq->{noderange} = \@snoderange;
|
|
$addreq->{arg}->[-1] = $SNdir;
|
|
push @requests, $addreq;
|
|
|
|
}
|
|
}
|
|
}
|
|
} # end if SN
|
|
|
|
# if not only syncing the service nodes ( -s flag)
|
|
# for each node build the
|
|
# the command, to sync from the service node
|
|
if ($syncsn == 0)
|
|
{ #syncing nodes ( no -s flag)
|
|
foreach my $snkey (keys %$sn)
|
|
{
|
|
|
|
if (!grep(/$snkey/, @MNnodeipaddr))
|
|
{ # entries run from the Service Node
|
|
|
|
# if the -F option to sync the nodes
|
|
# then for a Service Node
|
|
# change the command to use the -F /tmp/xcatrf.tmp
|
|
# because that is where the file was put on the SN
|
|
#
|
|
my $newSNreq = dclone($req);
|
|
if ($syncsnfile) # -F option
|
|
{
|
|
my $args = $newSNreq->{arg};
|
|
|
|
my $i = 0;
|
|
foreach my $argument (@$args)
|
|
{
|
|
|
|
# find the -F and change the name of the
|
|
# file in the next array entry to the tmp file
|
|
if ($argument eq "-F")
|
|
{
|
|
$i++;
|
|
$newSNreq->{arg}->[$i] = $tmpsyncsnfile;
|
|
last;
|
|
}
|
|
$i++;
|
|
}
|
|
}
|
|
else
|
|
{ # if other dcp command, change from directory
|
|
# to be the tmp directory on the service node
|
|
# if not pull (-P) funcion
|
|
if (($command eq "xdcp") && ($dcppull == 0))
|
|
{
|
|
$newSNreq->{arg}->[-2] = $SNpath;
|
|
}
|
|
}
|
|
$newSNreq->{node} = $sn->{$snkey};
|
|
$newSNreq->{'_xcatdest'} = $snkey;
|
|
$newSNreq->{_xcatpreprocessed}->[0] = 1;
|
|
push @requests, $newSNreq;
|
|
}
|
|
else
|
|
{ # just run normal dsh dcp
|
|
my $reqcopy = {%$req};
|
|
$reqcopy->{node} = $sn->{$snkey};
|
|
$reqcopy->{'_xcatdest'} = $snkey;
|
|
$reqcopy->{_xcatpreprocessed}->[0] = 1;
|
|
push @requests, $reqcopy;
|
|
|
|
}
|
|
} # end foreach
|
|
} # end syncing only service nodes
|
|
|
|
}
|
|
else
|
|
{ # running local on image
|
|
return [$req];
|
|
}
|
|
return \@requests;
|
|
}
|
|
|
|
#-------------------------------------------------------
|
|
|
|
=head3 process_request
|
|
|
|
Process the command
|
|
|
|
=cut
|
|
|
|
#-------------------------------------------------------
|
|
sub process_request
|
|
{
|
|
|
|
my $request = shift;
|
|
my $callback = shift;
|
|
my $nodes = $request->{node};
|
|
my $command = $request->{command}->[0];
|
|
my $args = $request->{arg};
|
|
my $envs = $request->{env};
|
|
my $rsp = {};
|
|
# get the Environment Variables and set them in the current environment
|
|
foreach my $envar (@{$request->{env}})
|
|
{
|
|
my ($var, $value) = split(/=/, $envar, 2);
|
|
$ENV{$var} = $value;
|
|
}
|
|
if ($command eq "xdsh")
|
|
{
|
|
xdsh($nodes, $args, $callback, $command, $request->{noderange}->[0]);
|
|
}
|
|
else
|
|
{
|
|
if ($command eq "xdcp")
|
|
{
|
|
xdcp($nodes, $args, $callback, $command,
|
|
$request->{noderange}->[0]);
|
|
}
|
|
else
|
|
{
|
|
my $rsp = {};
|
|
$rsp->{data}->[0] =
|
|
"Unknown command $command. Cannot process the command.";
|
|
xCAT::MsgUtils->message("E", $rsp, $callback, 1);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
#-------------------------------------------------------
|
|
|
|
=head3 xdsh
|
|
|
|
Parses Builds and runs the dsh
|
|
|
|
|
|
=cut
|
|
|
|
#-------------------------------------------------------
|
|
sub xdsh
|
|
{
|
|
my ($nodes, $args, $callback, $command, $noderange) = @_;
|
|
|
|
|
|
# parse dsh input
|
|
my @local_results =
|
|
xCAT::DSHCLI->parse_and_run_dsh($nodes, $args, $callback,
|
|
$command, $noderange);
|
|
#print $local_results[0];
|
|
#print "\n";
|
|
#`echo $local_results[0] >> /tmp/lissa/testxdsh`;
|
|
my $maxlines=10000;
|
|
my $arraylen=@local_results;
|
|
my $rsp = {};
|
|
my $i=0;
|
|
my $j;
|
|
while ($i < $arraylen) {
|
|
for ($j = 0 ; $j < $maxlines ; $j++) {
|
|
if ($i > $arraylen) {
|
|
last;
|
|
} else {
|
|
$rsp->{data}->[$j]= $local_results[$i]; # send max lines
|
|
}
|
|
$i++
|
|
}
|
|
xCAT::MsgUtils->message("D", $rsp, $callback);
|
|
}
|
|
return;
|
|
}
|
|
|
|
#-------------------------------------------------------
|
|
|
|
=head3 xdcp
|
|
|
|
Parses, Builds and runs the dcp command
|
|
|
|
|
|
=cut
|
|
|
|
#-------------------------------------------------------
|
|
sub xdcp
|
|
{
|
|
my ($nodes, $args, $callback, $command, $noderange) = @_;
|
|
|
|
#`touch /tmp/lissadebug`;
|
|
# parse dcp input
|
|
my @local_results =
|
|
xCAT::DSHCLI->parse_and_run_dcp($nodes, $args, $callback,
|
|
$command, $noderange);
|
|
my $rsp = {};
|
|
my $i = 0;
|
|
## process return data
|
|
foreach my $line (@local_results)
|
|
{
|
|
$rsp->{data}->[$i] = $line;
|
|
$i++;
|
|
}
|
|
|
|
xCAT::MsgUtils->message("D", $rsp, $callback);
|
|
if (-e "/tmp/xcatrf.tmp")
|
|
{ # used tmp file for -F option
|
|
#`rm /tmp/xcatrf.tmp`;
|
|
}
|
|
return;
|
|
}
|
|
|