163 lines
3.3 KiB
Perl
163 lines
3.3 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 xCAT::Table;
|
||
|
|
|
||
|
|
use xCAT::Utils;
|
||
|
|
|
||
|
|
use xCAT::MsgUtils;
|
||
|
|
|
||
|
|
1;
|
||
|
|
|
||
|
|
#-------------------------------------------------------
|
||
|
|
|
||
|
|
=head3 handled_commands
|
||
|
|
|
||
|
|
Return list of commands handled by this plugin
|
||
|
|
|
||
|
|
=cut
|
||
|
|
|
||
|
|
#-------------------------------------------------------
|
||
|
|
|
||
|
|
sub handled_commands
|
||
|
|
{
|
||
|
|
return {
|
||
|
|
xdsh => "xdsh",
|
||
|
|
xdcp => "xdsh"
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
#-------------------------------------------------------
|
||
|
|
|
||
|
|
=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 %rsp;
|
||
|
|
$::DSH = "/opt/csm/bin/dsh";
|
||
|
|
$::DCP = "/opt/csm/bin/dcp";
|
||
|
|
|
||
|
|
# check that dsh is installed
|
||
|
|
if (!-e $::DSH)
|
||
|
|
{
|
||
|
|
$rsp->{data}->[0] =
|
||
|
|
"dsh is not installed. Cannot process the command\n";
|
||
|
|
xCAT::MsgUtils->message("E", $rsp, $callback);
|
||
|
|
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
|
||
|
|
if (($command eq "xdsh") || ($command eq "xdcp"))
|
||
|
|
{
|
||
|
|
return
|
||
|
|
xdsh($nodes, $args, $callback, $command,
|
||
|
|
$request->{noderange}->[0]);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{ # error
|
||
|
|
$rsp->{data}->[0] =
|
||
|
|
"Unknown command $command. Cannot process the command\n";
|
||
|
|
xCAT::MsgUtils->message("E", $rsp, $callback);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#-------------------------------------------------------
|
||
|
|
|
||
|
|
=head3 xdsh
|
||
|
|
|
||
|
|
Builds and runs the dsh or dcp command
|
||
|
|
|
||
|
|
|
||
|
|
=cut
|
||
|
|
|
||
|
|
#-------------------------------------------------------
|
||
|
|
sub xdsh
|
||
|
|
{
|
||
|
|
my $nodes = shift;
|
||
|
|
my $args = shift;
|
||
|
|
my $callback = shift;
|
||
|
|
my $command = shift;
|
||
|
|
my $noderange = shift;
|
||
|
|
|
||
|
|
#
|
||
|
|
# set XCAT Context
|
||
|
|
#
|
||
|
|
$ENV{DSH_CONTEXT} = "XCAT";
|
||
|
|
|
||
|
|
#
|
||
|
|
# if nodes, Put nodes in a file so we do
|
||
|
|
# not risk hitting a command line
|
||
|
|
# limit
|
||
|
|
my $node_file;
|
||
|
|
if ($nodes)
|
||
|
|
{
|
||
|
|
$node_file = xCAT::Utils->make_node_list_file($nodes);
|
||
|
|
$ENV{'DSH_LIST'} = $node_file; # export the file for dsh
|
||
|
|
}
|
||
|
|
|
||
|
|
#
|
||
|
|
# call dsh or dcp
|
||
|
|
#
|
||
|
|
|
||
|
|
my $dsh_dcp_command = "";
|
||
|
|
my %rsp;
|
||
|
|
if ($command eq "xdsh")
|
||
|
|
{
|
||
|
|
$dsh_dcp_command = $::DSH;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
$dsh_dcp_command = $::DCP;
|
||
|
|
}
|
||
|
|
$dsh_dcp_command .= " ";
|
||
|
|
|
||
|
|
foreach my $arg (@$args)
|
||
|
|
{ # add arguments
|
||
|
|
$dsh_dcp_command .= $arg; # last argument must be command to run
|
||
|
|
$dsh_dcp_command .= " ";
|
||
|
|
}
|
||
|
|
$dsh_dcp_command .= "2>&1";
|
||
|
|
my @local_results = `$dsh_dcp_command`; # run the dsh command
|
||
|
|
my $rc = $? >> 8;
|
||
|
|
my $i = 0;
|
||
|
|
chop @local_results;
|
||
|
|
foreach my $line (@local_results)
|
||
|
|
{
|
||
|
|
$rsp->{data}->[$i] = $line;
|
||
|
|
$i++;
|
||
|
|
}
|
||
|
|
|
||
|
|
#$rsp->{data}->[$i] = "Return Code = $rc\n";
|
||
|
|
xCAT::Utils->close_delete_file($::NODE_LIST_FILE, $node_file);
|
||
|
|
xCAT::MsgUtils->message("I", $rsp, $callback);
|
||
|
|
|
||
|
|
#xCAT::MsgUtils->message("I", $rsp);
|
||
|
|
#$callback->($rsp);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|