From 3857fcad8814180cbd7b9579ef47542e65ee5e4d Mon Sep 17 00:00:00 2001 From: vallard Date: Mon, 4 May 2009 20:10:48 +0000 Subject: [PATCH] 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 --- xCAT-server/share/xcat/tools/nodesw | 537 ++++++++++++++++++++++++++++ 1 file changed, 537 insertions(+) create mode 100755 xCAT-server/share/xcat/tools/nodesw diff --git a/xCAT-server/share/xcat/tools/nodesw b/xCAT-server/share/xcat/tools/nodesw new file mode 100755 index 000000000..67022d540 --- /dev/null +++ b/xCAT-server/share/xcat/tools/nodesw @@ -0,0 +1,537 @@ +#!/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] vlan \n"; + print "nodesw [-v] 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; + } + } +} +