arif/xcat-core
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
7d2335289b
xcat-core/xCAT-server/lib/xcat/plugins/route.pm
daniceexi f9c0de1b52 defect 3111514: fixed some minor syntax problems 
git-svn-id: https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@8480 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd
2010-12-20 10:03:58 +00:00

435 lines
9.7 KiB
Perl
Raw Blame History

# IBM(c) 2010 EPL license http://www.eclipse.org/legal/epl-v10.html
#-------------------------------------------------------
=head1
xCAT plugin package to handle xdsh
Supported command:
nodenetconn
ipforward (internal command)
=cut
#-------------------------------------------------------
package xCAT_plugin::route;
BEGIN
{
$::XCATROOT = $ENV{'XCATROOT'} ? $ENV{'XCATROOT'} : '/opt/xcat';
}
use lib "$::XCATROOT/lib/perl";
use strict;
use xCAT::Table;
use xCAT::Utils;
use xCAT::NetworkUtils;
use xCAT::MsgUtils;
use Getopt::Long;
use xCAT::NodeRange;
use Data::Dumper;
1;
#-------------------------------------------------------
=head3 handled_commands
Return list of commands handled by this plugin
=cut
#-------------------------------------------------------
sub handled_commands
{
return {
makeroutes => "route",
ipforward => "route"
};
}
#-------------------------------------------------------
=head3 preprocess_request
Preprocess the command
=cut
#-------------------------------------------------------
sub preprocess_request
{
my $request = shift;
my $callback = shift;
my $sub_req = shift;
my $command = $request->{command}->[0];
#if already preprocessed, go straight to request
if ((defined($request->{_xcatpreprocessed}))
&& ($request->{_xcatpreprocessed}->[0] == 1))
{
return [$request];
}
if ($command eq "ipforward") {
my $nodes=$request->{node};
my @sns=();
if ($nodes) {
@sns=@$nodes;
}
print "sns=@sns\n";
my @requests=();
foreach (@sns) {
my $reqcopy = {%$request};
$reqcopy->{'node'}=[];
$reqcopy->{'_xcatdest'}=$_;
$reqcopy->{_xcatpreprocessed}->[0] = 1;
push @requests, $reqcopy;
}
return \@requests;
} else {
my $reqcopy = {%$request};
$reqcopy->{_xcatpreprocessed}->[0] = 1;
return [$reqcopy];
}
}
#-------------------------------------------------------
=head3 process_request
Process the command
=cut
#-------------------------------------------------------
sub process_request
{
my $request = shift;
my $callback = shift;
my $sub_req = shift;
my $command = $request->{command}->[0];
if ($command eq "makeroutes") {
return process_makeroutes($request, $callback, $sub_req);
} elsif ($command eq "ipforward") {
return process_ipforward($request, $callback, $sub_req);
}
return;
}
sub process_makeroutes {
my $request = shift;
my $callback = shift;
my $sub_req = shift;
my $command = $request->{command}->[0];
my $args = $request->{arg};
# parse the options
@ARGV=();
if ($args) {
@ARGV=@{$args};
}
Getopt::Long::Configure("bundling");
Getopt::Long::Configure("no_pass_through");
my $netmask_in;
my $net_in;
my $gateway_in;
my $routelist_in;
my $delete=0;
if(!GetOptions(
'h|help' => \$::HELP,
'v|version' => \$::VERSION,
'r|routename=s' => \$routelist_in,
'n|net=s' => \$net_in,
'm|mask=s' => \$netmask_in,
'g|gateway=s' => \$gateway_in,
'd|delete' => \$delete,))
{
&usage($callback);
return 1;
}
# display the usage if -h or --help is specified
if ($::HELP) {
&usage($callback);
return 0;
}
if (@ARGV > 0) {
&usage($callback);
return 1;
}
# display the version statement if -v or --verison is specified
if ($::VERSION)
{
my $rsp={};
$rsp->{data}->[0]= xCAT::Utils->Version();
$callback->($rsp);
return 0;
}
#get all the routes from the routes table
my %all_routes=();
my $routestab=xCAT::Table->new("routes", -create =>1);
if ($routestab) {
my @tmp1=$routestab->getAllAttribs(('routename', 'net', 'mask', 'gateway', 'ifname'));
if (@tmp1 > 0) {
foreach(@tmp1) {
$all_routes{$_->{routename}} = $_;
$_->{process} = 0;
}
}
}
#print Dumper(%all_routes);
#get the routes to be handled
if ($routelist_in) {
if ($net_in || $gateway_in || $netmask_in) {
my $rsp={};
$rsp->{error}->[0]= "-r cannot be combined with -n,-g or -m flag.";
$callback->($rsp);
return 1;
}
#check if the route names are valid
foreach(split(',', $routelist_in)) {
if (! exists($all_routes{$_})) {
my $rsp={};
$rsp->{error}->[0]= "route $_ is not defined in the routes table.";
$callback->($rsp);
return 1;
} else {
$all_routes{$_}->{process}=1;
}
}
}
elsif ( $net_in || $gateway_in || $netmask_in) {
my $count=0;
foreach (keys %all_routes) {
my $process=1;
if (($net_in) && ($all_routes{$_}->{net} ne $net_in)) { $process=0; }
if (($netmask_in) && ($all_routes{$_}->{mask} ne $netmask_in)) { $process=0; }
if (($gateway_in) && ($all_routes{$_}->{gateway} ne $gateway_in)) { $process=0; }
$all_routes{$_}->{process}=$process;
if ($process == 1) { $count++; }
}
if ($count==0) {
my $rsp={};
$rsp->{data}->[0]= "No route in the routes table matches the input.";
$callback->($rsp);
return 1;
}
}
else {
#handle all routes
foreach (keys %all_routes) {
$all_routes{$_}->{process}= 1;
}
}
#print Dumper(%all_routes);
#now let handle the route creatation and deletion
my @sns=();
while (my ($routename, $route_hash) = each(%all_routes)) {
if ($route_hash->{process}) {
my ($gw_name, $gw_ip)=xCAT::NetworkUtils->gethostnameandip($route_hash->{gateway});
push(@sns, $gw_name);
if ($delete) {
delete_route($callback, $route_hash->{net}, $route_hash->{mask}, $gw_ip, $gw_name);
}
else {
set_route($callback, $route_hash->{net}, $route_hash->{mask}, $gw_ip, $gw_name);
}
}
}
#not all gateways are service nodes
my %sn_hash=();
my @allSN=xCAT::Utils->getAllSN();
my %allSN_hash=();
foreach(@allSN) {$allSN_hash{$_}=1;}
foreach my $sn (@sns) {
if (exists($allSN_hash{$sn})) {
$sn_hash{$sn}=1;
}
}
#update servicenode.ipforward
my $sntab=xCAT::Table->new("servicenode", -create => 1,-autocommit => 1);
my %valuehash=();
my $value=1;
if ($delete) {$value=0;}
foreach my $sn (keys %sn_hash) {
$valuehash{$sn} = { ipforward=>$value };
}
$sntab->setNodesAttribs(\%valuehash);
#go to the service nodes to enable/disable ipforwarding
my @nodes=keys(%sn_hash);
$sub_req->({
command=>['ipforward'],
node=>\@nodes,
arg=>[$delete]},
$callback);
}
sub process_ipforward {
my $request = shift;
my $callback = shift;
my $sub_req = shift;
my $args = $request->{arg};
my $delete=0;
if ($args) {
$delete = $args->[0];
}
if ($delete) {
xCAT::NetworkUtils->setup_ip_forwarding(0);
} else {
xCAT::NetworkUtils->setup_ip_forwarding(1);
}
}
sub usage {
my $cb=shift;
my $rsp={};
$rsp->{data}->[0]= "Usage: makeroutes -h";
$rsp->{data}->[1]= " makeroutes -v";
$rsp->{data}->[2]= " makeroutes";
$rsp->{data}->[3]= " makeroutes -d";
$rsp->{data}->[4]= " makeroutes -r routename[,routename...]";
$rsp->{data}->[5]= " makeroutes -r routename[,routename...] -d";
$rsp->{data}->[6]= " makeroutes [-n network] [-m mask] [-g gateway]";
$rsp->{data}->[7]= " makeroutes [-n network] [-m mask]] [-g gateway] -d";
$cb->($rsp);
}
#check if the route exits or not from the route table
sub route_exists {
my $net = shift;
my $mask = shift;
my $gw_ip = shift;
my $gw=shift;
my $islinux=xCAT::Utils->isLinux();
my $result;
$result=`netstat -nr|grep $net`;
if ($? == 0) {
if ($result) {
my @b=split('\n', $result);
foreach my $tmp (@b) {
chomp($tmp);
my @a=split(' ', $tmp);
if ($islinux) { #Linux
if (@a >= 3) {
my $net1=$a[0];
my $mask1=$a[2];
my $gw1=$a[1];
if (($net1 eq $net) && ($mask1 eq $mask) && (($gw1 eq $gw) || ($gw1 eq $gw_ip))) {
return 1;
}
}
}
else { #AIX
if (@a >= 2) {
my $tmp1=$a[0];
my $gw1=$a[1];
#now convert $mask to bits
$net =~ /([0-9]+)\.([0-9]+)\.([0-9]+)\.([0-9]+)/;
my $netnum = ($1<<24)+($2<<16)+($3<<8)+$4;
my $bits=32;
while (($netnum % 2) == 0) {
$bits--;
$netnum=$netnum>>1;
}
my $tmp2="$net/$bits";
if (($tmp1 eq $tmp2) && (($gw1 eq $gw) || ($gw1 eq $gw_ip))) {
return 1;
}
}
}
}
}
}
return 0;
}
# sets the route with given parameters
sub set_route {
my $callback=shift;
my $net = shift;
my $mask = shift;
my $gw_ip = shift;
my $gw=shift;
my $result;
if (!route_exists($net, $mask, $gw_ip, $gw)) {
#set temporay route
my $cmd;
if (xCAT::Utils->isLinux()) {
$cmd="route add -net $net netmask $mask gw $gw_ip";
} else {
$cmd="route add -net $net -netmask $mask $gw_ip";
}
print "cmd=$cmd\n";
$result=`$cmd 2>&1`;
if ($? != 0) {
my $rsp={};
$rsp->{error}->[0]= "$cmd\nerror code=$?, result=$result\n";
$callback->($rsp);
return 1;
} else {
#TODO: set per permanent route
}
}
return 0;
}
# deletes the route with given parameters
sub delete_route {
my $callback=shift;
my $net = shift;
my $mask = shift;
my $gw_ip = shift;
my $gw=shift;
my $result;
if (route_exists($net, $mask, $gw_ip, $gw)) {
#delete route temporarily
my $cmd;
if (xCAT::Utils->isLinux()) {
$cmd="route delete -net $net netmask $mask gw $gw_ip";
} else {
$cmd="route delete -net $net -netmask $mask $gw_ip";
}
print "cmd=$cmd\n";
$result=`$cmd 2>&1`;
if ($? != 0) {
my $rsp={};
$rsp->{error}->[0]= "$cmd\nerror code=$?, result=$result\n";
$callback->($rsp);
return 1;
} else {
#TODO: delete route permanently
}
}
return 0;
}
Reference in New Issue View Git Blame Copy Permalink