#!/usr/bin/perl -w

################################################################
#
# Dieses Tool durchsucht die Switches nach MAC-Adressen
#
# ##############
# Version 1.0
# 
#################################################################


#################################################################
# Module
#################################################################

use Net::Ping;
use DBI;
use strict;


#################################################################
# globale Variablen
#################################################################

use vars qw($MIBPATH_MAC $MIBPATH_NUMBER_OF_PORT $COMMUNITY @SWITCHS $OUTPUTFILE);

$MIBPATH_MAC = ".1.3.6.1.2.1.17.4.3.1.1";
$MIBPATH_NUMBER_OF_PORT =".1.3.6.1.2.1.17.4.3.1.2";
$COMMUNITY = "public";
#@SWITCHS = ("switch1b|1-24|n",
#	    "switch2c|1-24|n",
#	    "switch2f|1-24|n",
#	    "switch2g|1-12|n",
#	    "switch3a|1-24|switch3a;Unit1;0",
#	    "switch3b|1-24|switch3a;Unit2;25",
#	    "switch3c|1-24|switch3a;Unit3;49",
#	    "switch3d|1-24|switch3a;Unit4;73",
#	    "switch3e|1-48|n",
#	    "switch3f|1-24|n",
#	    "switch3g|1-12|n",
#	    "switch3i|1-12|n",
@SWITCHS = ("switch3l|1-12|n"
	   );
$OUTPUTFILE="/data/pcfinder/MacToPort.dat";


##################################################################
# main
##################################################################
    
for(my $i = 0; $i < scalar(@SWITCHS); $i++)
{
    my @aszSwitch = split(/\|/,$SWITCHS[$i]);
    my $szSwitchName = $aszSwitch[0];
    my $szPortsToScan = $aszSwitch[1];
    my $szStack	= $aszSwitch[2];
    my $ping = Net::Ping->new("icmp",2);
	    
    # Ueberpruefen, ob Switch verfuegbar ist	    
    if($ping->ping($szSwitchName))
    {
	&get_mac($szSwitchName, $szPortsToScan, $szStack);
    }
    else
    {
	print "Switch: ".$szSwitchName." unavailable!\n";
    } 
}



#################################################################
# Liesst die Mac-Adressen aus den Switchs aus
#################################################################
sub get_mac
{

    my $szSwitchName = $_[0];
    my @aszPortsToScan = split(/;/,$_[1]);
    
    my $szStackName = "";
    my $nPortOffset = 0;
     
    if($_[2] ne "n")
    {
	my @aszStack = split(/;/,$_[2]);
	$szStackName = "$aszStack[0]-$aszStack[1]";
	$nPortOffset = $aszStack[2];
    }
 
    my @anPort;
    my @aszMAC;
    my @aszPID;
    my @aszMID;
    my @aszPortMac;
    
    my @splitLine;
    my @temp;
    my @anMinMaxPort;
    my $nIndex = 0;
    my $nPortPos = 0;
    my $nPortMacPos=0;
    my $szSplit;
    my $bScanPort = "false";
    my $nPort;
    
 
    my $szLine;

    my @count_sw = `/usr/bin/snmpwalk -v 1 $szSwitchName -c $COMMUNITY $MIBPATH_NUMBER_OF_PORT`;
    my @snmpwalk_res = `/usr/bin/snmpwalk -v 1 $szSwitchName -c $COMMUNITY $MIBPATH_MAC`;

    foreach $szLine(@count_sw)
    {
 	chomp($szLine);
    	@splitLine = split(/=/, $szLine);
	@temp = split(/:/, $splitLine[1]);
	#$temp[1] =~ s/\s+$//g;			#Leerzeichen am Ende entfernen
	#$temp[1] =~ s/^\s+//g;			#Leerzeichen am Anfang entfernen
	$anPort[$nIndex] = $temp[1];
	    
	$szSplit = substr($MIBPATH_NUMBER_OF_PORT,12);
	@temp = split(/$szSplit/, $splitLine[0]);
	#$temp[1] =~ s/\s+$//g;			#Leerzeichen am Ende entfernen
	#$temp[1] =~ s/^\s+//g;			#Leerzeichen am Anfang entfernen
	$aszPID[$nIndex] = $temp[1];
	$nIndex++;
	
    }
    
    $nIndex = 0;

    foreach $szLine(@snmpwalk_res)
    {
	chomp($szLine);
    	@splitLine = split(/=/, $szLine);
	@temp = split(/:/, $splitLine[1]);
	$temp[1] =~ s/\s+$//g;			#Leerzeichen am Ende entfernen
	$temp[1] =~ s/^\s+//g;			#Leerzeichen am Anfang entfernen
	$temp[1] =~ tr/ /:/;
	$aszMAC[$nIndex] = $temp[1];
	
	$szSplit = substr($MIBPATH_MAC,12);
	@temp = split(/$szSplit/, $splitLine[0]);
	#$temp[1] =~ s/\s+$//g;			#Leerzeichen am Ende entfernen
	#$temp[1] =~ s/^\s+//g;			#Leerzeichen am Anfang entfernen
	$aszMID[$nIndex] = $temp[1];
	$nIndex++;
    }
    
    for($nPortPos=0; $nPortPos<scalar(@anPort);$nPortPos++)
    {
	# Ueberpuefen, ob der Port gescannt werden darf
	$bScanPort = "false";
	for($nIndex=0; $nIndex<scalar(@aszPortsToScan); $nIndex++)
	{
	    $nPort = $anPort[$nPortPos] - $nPortOffset;
	    #print "$szSwitchName - $anPort[$nPortPos] - $nPortOffset\n";
	    @anMinMaxPort = split(/-/,$aszPortsToScan[$nIndex]);
	    if(scalar(@anMinMaxPort) == 1 && $aszPortsToScan[$nIndex] == $nPort)
	    {
		$bScanPort = "true";
		last;
	    }
	    elsif(scalar(@anMinMaxPort) == 2 && ( $nPort >= $anMinMaxPort[0] && $nPort <= $anMinMaxPort[1]))
	    {
		$bScanPort = "true";
		last;
	    }	
	}
	
    
	if($bScanPort eq "true")
	{
	    for($nIndex=0; $nIndex<scalar(@aszMAC);$nIndex++)
	    {
		if($aszMID[$nIndex] eq $aszPID[$nPortPos])
		{
		    $aszPortMac[$nPortMacPos] = "$nPort;$aszMAC[$nIndex]";
		    $nPortMacPos++;
		    last;
		}
	    }
	}
    }
    # Schreiben in db
    save_file_db(\@aszPortMac, $szSwitchName, $szStackName);
}


#################################################################
sub save_file_db
{
    my ($refArray,$szSwitchName,$szStackName) = @_;
    my @aszPortToMac = @{$refArray};
    my @aszData;
    my @temp;
    my $nIndex;
    my $nLine;
    my $szDate;
    my $szInsertData;
    my $bFound = "false";
    my $szDBMac;
    my $szNewMac;
    my $nPort;
    
    # Einlesen der Daten 
    open(INPUTFILE, "<$OUTPUTFILE") or die("Die Datei ".$OUTPUTFILE." konnte nicht geoeffnet werden");
    while(<INPUTFILE>)
    {
	push(@aszData,$_);
    }
    close(INPUTFILE);
    
    for($nIndex=0; $nIndex<scalar(@aszPortToMac); $nIndex++)
    {
	$bFound = "false";
	# Daten die eingetragen werden muessen
	$szDate = get_date();
	@temp = split(/;/,$aszPortToMac[$nIndex]);
	$nPort = $temp[0];
	$szNewMac = $temp[1];
	# Format der Daten MAC;SwitchName;Port;Date
	$szInsertData =  "$szNewMac;$szStackName;$szSwitchName;$nPort;$szDate\n";
    
	# Suchen, ob der Eintrag schon vorhanden ist und geg. ersetzen
	for($nLine=0; $nLine<scalar(@aszData); $nLine++)
	{
	    @temp = split(/;/,$aszData[$nLine]);
	    $szDBMac = $temp[0];
	    # Wenn die MACs uebereinstimmen, dann ist der Eintrag vorhanden
	    if($szNewMac eq $szDBMac)
	    {
		$aszData[$nLine] = $szInsertData;
		$bFound = "true";
		last;
	    } 
	}
	
	# Falls der Eintrag noch nicht vorhanden ist
	if($bFound eq "false")
	{
	    $aszData[scalar(@aszData)] = $szInsertData;	
	}    
    }
    open(OUTPUTFILE, ">$OUTPUTFILE") or die("Die Datei konnte nicht geoeffnet werden");
    for(@aszData)
    {
	print OUTPUTFILE $_;
    }
    close(OUTPUTFILE);
}




#################################################################
# Rueckgabe des aktuellen Datums
#################################################################
sub get_date{
 (my $sec, my $min, my $hour, my $mday, my $mon, my $year, my $wday, my $yday, my $idst) = localtime(time);
 $year += 1900;
 $mon++;
 $wday = $yday = $idst = "";
 my $found = sprintf("%4d-%02d-%02d %02d:%02d:%02d", $year, $mon, $mday, $hour, $min,$sec);
 return $found;
}