e68fe9d2db
git-svn-id: https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@1561 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd
136 lines
3.0 KiB
Perl
136 lines
3.0 KiB
Perl
package xCAT_plugin::nodestat;
|
|
|
|
use Socket;
|
|
use IO::Handle;
|
|
my $stat;
|
|
|
|
sub handled_commands {
|
|
return {
|
|
nodestat => 'nodestat',
|
|
};
|
|
}
|
|
|
|
sub pinghost {
|
|
my $node = shift;
|
|
my $rc = system("ping -q -n -c 1 -w 1 $node > /dev/null");
|
|
if ($rc == 0) {
|
|
return 1;
|
|
} else {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
sub nodesockopen {
|
|
my $node = shift;
|
|
my $port = shift;
|
|
my $socket;
|
|
my $addr = gethostbyname($node);
|
|
my $sin = sockaddr_in($port,$addr);
|
|
my $proto = getprotobyname('tcp');
|
|
socket($socket,PF_INET,SOCK_STREAM,$proto) || return 0;
|
|
connect($socket,$sin) || return 0;
|
|
return 1;
|
|
}
|
|
|
|
sub installer_query {
|
|
my $node = shift;
|
|
my $destport = 3001;
|
|
my $socket;
|
|
my $text = "";
|
|
my $proto = getprotobyname('tcp');
|
|
socket($socket,PF_INET,SOCK_STREAM,$proto) || return 0;
|
|
my $addr = gethostbyname($node);
|
|
my $sin = sockaddr_in($destport,$addr);
|
|
connect($socket,$sin) || return 0;
|
|
print $socket "stat \n";
|
|
$socket->flush;
|
|
while (<$socket>) {
|
|
$text.=$_;
|
|
}
|
|
$text =~ s/\n.*//;
|
|
return $text;
|
|
close($socket);
|
|
}
|
|
|
|
|
|
sub getstat {
|
|
my $response = shift;
|
|
$stat = $response->{node}->[0]->{data}->[0];
|
|
}
|
|
|
|
#-------------------------------------------------------
|
|
|
|
=head3 preprocess_request
|
|
|
|
Check and setup for hierarchy
|
|
|
|
=cut
|
|
|
|
#-------------------------------------------------------
|
|
sub preprocess_request
|
|
{
|
|
my $req = shift;
|
|
my $cb = shift;
|
|
my %sn;
|
|
if ($req->{_xcatdest}) { return [$req]; } #exit if preprocessed
|
|
my $nodes = $req->{node};
|
|
my $service = "xcat";
|
|
|
|
# find service nodes for requested nodes
|
|
# build an individual request for each service node
|
|
$sn = xCAT::Utils->get_ServiceNode($nodes, $service, "MN");
|
|
|
|
# build each request for each service node
|
|
my @requests;
|
|
foreach my $snkey (keys %$sn)
|
|
{
|
|
my $reqcopy = {%$req};
|
|
$reqcopy->{node} = $sn->{$snkey};
|
|
$reqcopy->{'_xcatdest'} = $snkey;
|
|
push @requests, $reqcopy;
|
|
|
|
}
|
|
return \@requests;
|
|
}
|
|
|
|
sub process_request {
|
|
my $request = shift;
|
|
my $callback = shift;
|
|
my $doreq = shift;
|
|
my @nodes = @{$request->{node}};
|
|
my $node;
|
|
foreach $node (@nodes) {
|
|
my %rsp;
|
|
my $text="";
|
|
$rsp{name}=[$node];
|
|
unless (pinghost($node)) {
|
|
$rsp{data} = [ 'noping' ];
|
|
$callback->({node=>[\%rsp]});
|
|
next;
|
|
}
|
|
if (nodesockopen($node,15002)) {
|
|
$rsp{data} = [ 'pbs' ];
|
|
$callback->({node=>[\%rsp]});
|
|
next;
|
|
} elsif (nodesockopen($node,22)) {
|
|
$rsp{data} = [ 'sshd' ];
|
|
$callback->({node=>[\%rsp]});
|
|
next;
|
|
} elsif ($text = installer_query($node)) {
|
|
$rsp{data} = [ $text ];
|
|
$callback->({node=>[\%rsp]});
|
|
next;
|
|
} else {
|
|
$doreq->({command=>['nodeset'],
|
|
node=>[$node],
|
|
arg=>['stat']},
|
|
\&getstat);
|
|
$rsp{data} = [ 'ping '.$stat ];
|
|
$callback->({node=>[\%rsp]});
|
|
next;
|
|
}
|
|
}
|
|
}
|
|
|
|
1;
|