arif/xcat-core
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
5ae480d3eb
xcat-core/xCAT-server/share/xcat/tools/nodesw
vallard 3857fcad88 added nodesw to change vlans for nodes 
git-svn-id: https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@3291 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd
2009-05-04 20:10:48 +00:00

538 lines
13 KiB
Perl
Executable File
Raw Blame History

#!/usr/bin/perl
use strict;
use SNMP;
use Socket;
use Data::Dumper;
use Getopt::Long;
$SNMP::debugging = 1;
$SNMP::verbose = 1;
$SNMP::best_guess = 1;
# each 0 is four bits: 0000 0000
# thus its broken down:
# - ports 1-8 are in the first hex number
# - ports 9-16 are in the second hex number
# - ports 17-24 are in the third hex number
# - ports 25-32 are in the fourth hex number
# - ports 33-40 are in the fifth hex number
# - ports 41-48 are in the sixth hex number
# - ports 49-56 are in the seventh hex number
my @bitmap = (0x01, 0x80, 0x40, 0x20, 0x10, 0x08, 0x04, 0x02);
sub getVlanMask{
my $session = shift;
my $vlan = shift;
my @xs;
my @hs;
my $oid = ".1.3.6.1.2.1.17.7.1.4.3.1.4.$vlan";
my $hs = $session->get($oid);
if($session->{ErrorNum}){
die "Couldn't get OID!" . $session->{ErrorStr} . "\n";
}
$hs =~ s/\"//g; # get rid of quotes!
@hs = split(" ",$hs);
#foreach(@hs){ print "$_\n"; }
@xs = map(hex, @hs);
#foreach (@xs){
# print $_ . "\n";
#}
unless(scalar(@xs) eq 7){
print "Could not get vlan mask for $vlan\n";
}
#print "Switch mask:\n";
#printf("%02x %02x %02x %02x %02x %02x %02x\n", @xs);
return(@xs);
}
# add port 1
# returns a 7 digit hex string to logically or with existing
# to add this port to the subnet
sub getPortMask {
my $port = shift;
my @portMask;
my $hex;
my $xIndex;
if($port < 9){ $xIndex = 0;
}elsif($port < 17){ $xIndex = 1;
}elsif($port < 25){ $xIndex = 2;
}elsif($port < 33){ $xIndex = 3;
}elsif($port < 41){ $xIndex = 4;
}elsif($port < 49){ $xIndex = 5;
}elsif($port < 57){ $xIndex = 6;
}else{
print "I don't know how to handle this port...\n";
exit;
}
$port = $port % 8;
#print "$port\n";
$hex = $bitmap[$port];
#printf("%02x\n", $hex);
for(0 .. 6){
if($_ eq $xIndex){
push @portMask, $hex;
}else{
push @portMask, hex(0);
}
}
#print "Port mask:\n";
#printf("%02x %02x %02x %02x %02x %02x %02x\n", @portMask);
return @portMask;
}
sub getSwitchInfo{
my $switch = shift;
my ($version,$community,@junk);
# bunch of xCAT tables stuff here to connect to switch
my $info = `tabdump switches | grep smc001 | sed 's/"//g'`;
if($?){
# if not in here, we try the defaults.
$version = 1;
$community = "public";
}else{
(undef,$version,undef,$community,@junk) = split(/,/,$info);
}
if($::DEBUG){
print "switch parameters for $switch:\n";
print "\tSNMP Version: $version\n";
print "\tSNMP Community: $community\n";
}
return($community, $version);
}
sub connectToSwitch{
my $switch = shift;
my $session;
my ($community, $snmpver) = getSwitchInfo($switch);
$session = new SNMP::Session(
DestHost => $switch,
Version => $snmpver,
Community => $community,
UseSprintValue => 1,
);
unless($session) {
#ERROR:
print "Failed to communicate with $switch\n";
print $SNMP::Session::ErrorStr . "\n";
#xCAT::MsgUtils->message("S","Failed to communicate with $switch");
}
return $session;
}
sub xorMasks{
my $sm = shift;
my $pm = shift;
my @nm;
foreach(0 .. 6){
my $foo = @$sm[$_] ^ @$pm[$_];
##printf("%02x\n", $foo);
$nm[$_] = $foo;
}
return @nm;
}
sub orMasks{
my $sm = shift;
my $pm = shift;
my @nm;
foreach(0 .. 6){
my $foo = @$sm[$_] | @$pm[$_];
##printf("%02x\n", $foo);
$nm[$_] = $foo;
}
return @nm;
}
sub andMasks{
my $sm = shift;
my $pm = shift;
my @nm;
foreach(0 .. 6){
my $foo = @$sm[$_] & @$pm[$_];
##printf("%02x\n", $foo);
$nm[$_] = $foo;
}
return @nm;
}
# KLUDGE function because I can't figure out how to do this with
# SNMP.pm and I give up after a week of trying.
sub snmpset {
my $sess = shift;
my $oid = shift;
my $type = shift;
my $val = shift;
my $snmpset = "/usr/bin/snmpset";
my ($cmd, $comm, $vers, $switch);
unless(-r "/usr/bin/snmpset"){
print "/usr/bin/snmpset command not found! Please install net-snmp-utils\n";
exit 1;
}
$comm = $sess->{Community};
$vers = $sess->{Version};
$switch = $sess->{DestHost};
$cmd = "$snmpset -c $comm -v $vers $switch $oid $type $val";
#print "$cmd\n";
system("$cmd >/dev/null");
#print "ec: $?\n";
if($? > 0){
print "Failed to execute command $cmd\n";
exit 1;
}
return $?;
}
sub addNodeToVlan{
# to run: switchport allowed vlan add $port (use bitmap)
# snmpset .. $oid1.$vlan x 00 00 00 00 00 00 00
my $oid1 = '.1.3.6.1.2.1.17.7.1.4.3.1.4';
# to run: switchport native vlan $vlan
# snmpset .. $oid2.$port u $vlan
my $oid2 = '.1.3.6.1.2.1.17.7.1.4.5.1.1';
my $session = shift;
my $vlan = shift;
my $port = shift;
#$oid1 .= ".$vlan";
#print "$oid1 \n";
my @xs; # netmask for current switch
my @pm; # netmask for port
my @jm; # the joined netmask
@xs = getVlanMask($session, $vlan);
@pm = getPortMask($port);
@jm = orMasks(\@xs, \@pm);
#print "Join mask:\n";
printf("%02x %02x %02x %02x %02x %02x %02x\n", @jm) if $::DEBUG;
my $mask = sprintf("%02x %02x %02x %02x %02x %02x %02x ", @jm);
#################################
# PART 1: add the switchport allowed capability
#################################
# TODO: This part I can't get working so I'm just going to do an
# snmpset command here instead.
#my $v1 = new SNMP::Varbind([$oid1,$vlan,$mask, 'OCTETSTR']);
#print Dumper($v1);
#$session->set($v1);
#if($session->{ErrorStr}) {
# print "Error! " . $session->{ErrorStr} . "\n";
#}
snmpset($session, "$oid1.$vlan", "x", "\'$mask\'");
#######################################
# PART 2: add the switchport native
#######################################
# first get the current one for part 3:
my $currNativeVlan = $session->get("$oid2.$port");
if($session->{ErrorNum}){
die "Couldn't get OID!" . $session->{ErrorStr} . "\n";
}
print "currNativeVLAN: $currNativeVlan\n" if $::DEBUG;
print "$currNativeVlan -> ";
# set the new switchport native vlan
my $v2 = new SNMP::Varbind([$oid2,$port,$vlan,'GAUGE32']);
#print Dumper($v2);
$session->set($v2);
if($session->{ErrorStr}) {
print "Error! " . $session->{ErrorStr} . "\n";
}
#######################################
# PART 3: take it off the other one it is on
#######################################
delPortFromVlan($session, $port, $currNativeVlan);
}
# returns the vlan number
sub getVlans{
my $session = shift;
my %vlans;
# IF-MIB:ifName: .1.3.6.1.2.1.31.1.1.1.1
my $ifName = '.1.3.6.1.2.1.31.1.1.1.1';
my $varbind = new SNMP::Varbind([$ifName, '']);
$session->getnext($varbind);
if($session->{ErrorStr}) {
print "Error! " . $session->{ErrorStr} . "\n";
}
# varbind: name: ifName, 1, Port1, OCTETSTR
while($varbind->[2]){
#print $varbind->[2] . "\n";
my $name = $varbind->[2];
if($name =~ /VLAN/){
#print "Found $name on " . $session->{DestHost} . "\n";
$name =~ s/VLAN//g;
# we subtract 1000 off the name to give us the actual VLAN.
$vlans{$name} = $varbind->[1] - 1000;
#foreach(@$varbind){
# print "\t$_\n";
#}
}
$session->getnext($varbind);
}
#print Dumper(%vlans);
#foreach(keys %vlans){
# print "VLAN: $_ has value: ". $vlans{$_} ."\n";
#}
return \%vlans;
}
# return 1 if port is on vlan return 0 if not on vlan
sub checkPortVlan{
my $rc = 0;
my $session = shift;
my $p = shift;
my $v = shift;
my @xs; # netmask for current switch
my @pm; # netmask for port
my @jm; # the joined netmask
@xs = getVlanMask($session, $v);
@pm = getPortMask($p);
@jm = andMasks(\@xs, \@pm);
#print "And Mask:\n";
#printf("%02x %02x %02x %02x %02x %02x %02x\n", @jm);
my $m = sprintf("%x%x%x%x%x%x%x", @jm);
#print "m: $m\n";
if($m > 0){
#print "port $p is on VLAN $v on switch " . $session->{DestHost} . "\n";
$rc = 1;
}
return $rc;
}
sub delPortFromVlan {
my $session = shift;
my $port = shift;
my $vlan = shift;
print "removing port $port from vlan $vlan\n" if $::DEBUG;
# first check and see if its on there:
unless(checkPortVlan($session, $port, $vlan)){
print "Port $port is not on VLAN $vlan\n";
}
my @vm = getVlanMask($session, $vlan);
my @pm = getPortMask($port);
my @xm = xorMasks(\@vm, \@pm);
#print "portmask:\n";
#printf("%02x %02x %02x %02x %02x %02x %02x\n", @pm);
#print "vlanmask:\n";
#printf("%02x %02x %02x %02x %02x %02x %02x\n", @vm);
#print "xormask:\n";
#printf("%02x %02x %02x %02x %02x %02x %02x\n", @xm);
# this is the untagged remove
my $oid1 = '.1.3.6.1.2.1.17.7.1.4.3.1.4';
my $mask = sprintf("%02x %02x %02x %02x %02x %02x %02x ", @xm);
snmpset($session, "$oid1.$vlan", "x", "\'$mask\'");
# this is the tagged remove
my $oid2 = '.1.3.6.1.2.1.17.7.1.4.3.1.2';
#my $mask = sprintf("%02x %02x %02x %02x %02x %02x %02x ", @xm);
#print "tagged mask: $mask\n";
snmpset($session, "$oid2.$vlan", "x", "\'$mask\'");
}
sub displayHelp{
my $ec = shift;
print "nodesw changes the vlan of a node to a specified vlan\n";
print "requires xCAT 2.0, Switch configured with SNMP sets, and only tested on SMC8648T\n";
print "nodesw -h|--help\n";
print "nodesw [-v] <noderange> vlan <vlan number>\n";
print "nodesw [-v] <noderange> show\n";
exit $ec;
}
sub getNodeRange{
my $nr = shift;
my @nr = `/opt/xcat/bin/nodels $nr switch.switch switch.port`;
my $nh;
chomp(@nr);
if($?){
print $nr[0];
exit 1;
}
foreach(@nr){
my($n,$char,$val) = split(/:/, $_);
$char = (split(/\./,$char))[1];
$val =~ s/ //g;
$nh->{$n}{$char} = $val;
}
if($::DEBUG){
foreach(keys %$nh){
print $_ .":";
print " switch:" . $nh->{$_}{'switch'} ;
print " port:" . $nh->{$_}{'port'};
print "\n";
}
}
# make sure all fields are defined
my $e = 0;
foreach my $node (keys %$nh){
unless($nh->{$node}{'switch'}){
print "$node does not have a defined switch in xCAT! (nodels $node switch.switch)\n";
$e++;
}
unless($nh->{$node}{'port'}){
print "$node does not have a defined port in xCAT! (nodels $node switch.port)\n";
$e++;
}
}
if($e > 0){
exit 1;
}
# return the node hash
return $nh;
}
sub show{
my $oid2 = '.1.3.6.1.2.1.17.7.1.4.5.1.1';
my $nh = shift;
foreach my $node (keys %$nh){
my $port = $nh->{$node}{'port'};
my $switch = $nh->{$node}{'switch'};
my $session = connectToSwitch($switch);
my $currNativeVlan = $session->get("$oid2.$port");
if($session->{ErrorNum}){
die "Couldn't get OID!" . $session->{ErrorStr} . "\n";
}
print "$node: $currNativeVlan\n";
}
}
##### commands:
# get VLANS
# check VLANs that contain port X
# we want to put this port on this new vlan, here is how we do it:
# connect to switch to see:
my $help =0;
$::DEBUG = 0;
GetOptions(
'h|help' => \$help,
'v|verbose' => \$::DEBUG
);
if($help){
displayHelp(0);
}
if($::DEBUG){
print "verbose is set to on!\n";
}
my $nr = "";
my $nodeRange = shift;
my $cmd = shift;
my $vlan = shift;
unless ($nodeRange){
print "missing noderange!\n\n";
displayHelp(1);
}
unless($cmd) {
print "missing operation! [show | vlan ]\n\n";
displayHelp(1);
}
if($cmd eq 'vlan'){
unless($vlan){
print "missing vlan number!\n\n";
displayHelp(1);
}
$nr = getNodeRange($nodeRange);
chVlan($nr, $vlan);
}elsif($cmd eq 'show'){
print "showing $nodeRange vlan settings\n" if $::DEBUG;
$nr = getNodeRange($nodeRange);
show($nr);
}else{
print "unrecognized operation requested: $cmd\n\n";
displayHelp(1);
}
################################################################################
# getVlans
# find all vlans of a switch
################################################################################
#my $vlans = getVlans($session);
################################################################################
# checkPort Vlan
################################################################################
#checkPortVlan($session, $port, $vlans->{$_};
################################################################################
# addPortToVLAN
################################################################################
# get all VLANs
sub chVlan{
my $nh = shift;
my $currSwitch = '';
my ($session, $vlans, $exists);
foreach my $node (keys %$nh){
my $port = $nh->{$node}{'port'};
my $switch = $nh->{$node}{'switch'};
unless($switch eq $currSwitch){
$session = connectToSwitch($switch);
$vlans = getVlans($session);
$exists = 0;
$currSwitch = $switch;
}
foreach(keys %$vlans){
# see if the requested VLAN actually exists
if( $_ eq $vlan){
print "VLAN $_ exists on $switch\n" if $::DEBUG;
# if it exists see if its already on it
$exists = 1;
if(checkPortVlan($session, $port, $vlans->{$_})){
print "$node: port $port already exists on $switch VLAN $vlan\n";
exit 1;
}else{
print "Adding Port: $port to VLAN: $vlan\n" if $::DEBUG;
print $node . ": ";
if(addNodeToVlan($session, $vlan, $port) eq 0){
print "Added port: $port to VLAN: $vlan\n" if $::DEBUG;
print "$vlan\n";
}
}
}
}
unless($exists){
print "VLAN $vlan does not exist on " . $session->{DestHost} . "\n";
exit 1;
}
}
}
Reference in New Issue View Git Blame Copy Permalink