#!/opt/gnu/bin/perl

# -*- mode: Perl -*-

##################################################################

# Config file creator

##################################################################

# Cloned form cfgmaker Mark Trettin <mkt@lucent.com>

# Created by Tobias Oetiker <oetiker@ee.ethz.ch>

# this produces a config file for one router, by bulling info

# off the router via snmp

#################################################################

#

# Distributed under the GNU copyleft

#

# $Id: atmmaker,v 1.1.1.1 2002/02/26 10:16:31 oetiker Exp $

#

use SNMP_Session;

use BER;

use Socket;

use strict;

%snmpget::OIDS = (  'sysDescr' => '1.3.6.1.2.1.1.1.0',

		    'sysContact' => '1.3.6.1.2.1.1.4.0',

		    'sysName' => '1.3.6.1.2.1.1.5.0',

		    'sysLocation' => '1.3.6.1.2.1.1.6.0',

		    'sysUptime' => '1.3.6.1.2.1.1.3.0',

		    'ifNumber' =>  '1.3.6.1.2.1.2.1.0',

		    ###################################

		    # add the ifNumber ....

		    'ifDescr' => '1.3.6.1.2.1.2.2.1.2',

		    'ifType' => '1.3.6.1.2.1.2.2.1.3',

		    'ifIndex' => '1.3.6.1.2.1.2.2.1.1',

		    'ifSpeed' => '1.3.6.1.2.1.2.2.1.5', 

		    'ifOperStatus' => '1.3.6.1.2.1.2.2.1.8',		 

		    'ifAdminStatus' => '1.3.6.1.2.1.2.2.1.7',		 

		    # up 1, down 2, testing 3

		    'ipAdEntAddr' => '1.3.6.1.2.1.4.20.1.1',

		    'ipAdEntIfIndex' => '1.3.6.1.2.1.4.20.1.2',

		    'sysObjectID' => '1.3.6.1.2.1.1.2.0',

		    'CiscolocIfDescr' => '1.3.6.1.4.1.9.2.2.1.1.28',

		    # asx port group info

		    'asxPortAdmnStat' => '1.3.6.1.4.1.326.2.2.2.1.2.2.1.2',

		    'asxPortOperStat' => '1.3.6.1.4.1.326.2.2.2.1.2.2.1.3',

		    'asxPortRemoteIpAddress'=> '1.3.6.1.4.1.326.2.2.2.1.2.2.1.6',

		    'asxPortILMIRemoteIpAddress' => '1.3.6.1.4.1.326.2.2.2.1.2.2.1.22',

		    'asxPortMaxBandwidthIn' => '1.3.6.1.4.1.326.2.2.2.1.2.2.1.9',

		    'asxhwPortName' => '1.3.6.1.4.1.326.2.2.1.1.2.3.1.13',

		    'asxPortNumber' => '1.3.6.1.4.1.326.2.2.2.1.2.2.1.1',

		 );

sub main {

  my(%ifType_d)=('1'=>'Other',

		 '2'=>'regular1822',

		 '3'=>'hdh1822',

		 '4'=>'ddnX25',

		 '5'=>'rfc877x25',

		 '6'=>'ethernetCsmacd',

		 '7'=>'iso88023Csmacd',

		 '8'=>'iso88024TokenBus',

		 '9'=>'iso88025TokenRing',

		 '10'=>'iso88026Man',

		 '11'=>'starLan',

		 '12'=>'proteon10Mbit',

		 '13'=>'proteon80Mbit',

		 '14'=>'hyperchannel',

		 '15'=>'fddi',

		 '16'=>'lapb',

		 '17'=>'sdlc',

		 '18'=>'ds1',

		 '19'=>'e1',

		 '20'=>'basicISDN',

		 '21'=>'primaryISDN',

		 '22'=>'propPointToPointSerial',

		 '23'=>'ppp',

		 '24'=>'softwareLoopback',

		 '25'=>'eon',

		 '26'=>'ethernet-3Mbit',

		 '27'=>'nsip',

		 '28'=>'slip',

		 '29'=>'ultra',

		 '30'=>'ds3',

		 '31'=>'sip',

		 '32'=>'frame-relay',

		 '33'=>'rs232',

		 '34'=>'para',

		 '35'=>'arcnet',

		 '36'=>'arcnetPlus',

		 '37'=>'atm',

		 '38'=>'miox25',

		 '39'=>'sonet',

		 '40'=>'x25ple',

		 '41'=>'iso88022llc',

		 '42'=>'localTalk',

		 '43'=>'smdsDxi',

		 '44'=>'frameRelayService',

		 '45'=>'v35',

		 '46'=>'hssi',

		 '47'=>'hippi',

		 '48'=>'modem',

		 '49'=>'aal5',

		 '50'=>'sonetPath',

		 '51'=>'sonetVT',

		 '52'=>'smdsIcip',

		 '53'=>'propVirtual',

		 '54'=>'propMultiplexor',

		 '55'=>'100BaseVG'

		 );

  my($community,$router) = split /\@/, $ARGV[0];

  die <

USAGE: cfgmaker 'community'\@'router'

EXAMPLE:  cfgmaker public\@ezwf7.ethz.ch >>mrtg.cfg

USAGE

  my($sysDescr,$sysContact,$sysName,$sysLocation,$ifNumber,$sysObjectID) =

    snmpget($router,$community,

	    'sysDescr','sysContact','sysName',	'sysLocation', 'ifNumber',

	    'sysObjectID');

   $sysDescr =~ s/\r/
/g;  # Change returns to 

   my($cisco_router_sysid) = '1\.3\.6\.1\.4\.1\.9';

   my($ciscobox) = ($sysObjectID =~ /^$cisco_router_sysid/);

   my($foreasx_switch_sysid) = '1\.3\.6\.1\.4\.1\.326';

   my($foreasx) = ($sysObjectID =~ /^$foreasx_switch_sysid/);

    print <

######################################################################

# Description: $sysDescr

#     Contact: $sysContact

# System Name: $sysName

#    Location: $sysLocation

#.....................................................................

ECHO

  my @ipadent = snmpgettable($router,$community, 'ipAdEntAddr');

  my @ipadentif = snmpgettable($router,$community, 'ipAdEntIfIndex');

  my(%ipaddr, %iphost,$index);

  while (scalar @ipadentif){

    $index = shift @ipadentif;

    $ipaddr{$index} = shift @ipadent;

    $iphost{$index} = 

      gethostbyaddr(pack('C4',split(/\./,$ipaddr{$index})), AF_INET);

    if ($iphost{$index} eq ''){

	 $iphost{$index} = 'No hostname defined for IP address';

    }

  }

  my(@ifdescr) = snmpgettable($router,$community, 'ifDescr');

  my(@iftype) = snmpgettable($router,$community, 'ifType');

  my(@ifspeed) = snmpgettable($router,$community, 'ifSpeed');

  my(@ifadminstatus) = snmpgettable($router,$community, 'ifAdminStatus');

  my(@ifoperstatus) = snmpgettable($router,$community, 'ifOperStatus');

  my(@ifindex) = snmpgettable($router,$community, 'ifIndex');

  # Get the user configurable interface description entered in the config 

  # if it's a cisco-box

  my @ciscodescr;

  if ($ciscobox) {

      @ciscodescr = snmpgettable($router,$community, 'CiscolocIfDescr');

  }

  if ($foreasx) {

      # Get the FORE ATM Switch port configurations

      # if it's a ASX Switch

      my @asxipaddress;

      my @asxilmiaddress;

      my @asxifadminstatus;

      my @asxifoperstatus;

      my @asxifilmiaddress;

      my @asxifipaddress;

      my @asxmaxbandwidthin;

      my @asxhwportname;

      my @asxportnumber;

      @asxportnumber = snmpgettable($router,$community, 'asxPortNumber');

      @asxhwportname = snmpgettable($router,$community, 'asxhwPortName');

      @asxipaddress = snmpgettable($router,$community, 'asxPortRemoteIpAddress');

      @asxilmiaddress = snmpgettable($router,$community, 'asxPortILMIRemoteIpAddress');

      @asxmaxbandwidthin = snmpgettable($router,$community, 'asxPortMaxBandwidthIn');

      @asxifadminstatus = snmpgettable($router,$community, 'asxPortAdmnStat');

      @asxifoperstatus = snmpgettable($router,$community, 'asxPortOperStat');

      my(%ipaddr, %iphost,%ipiladdr, %ipilhost);

      my(%sportnumber, %sportname, %sifadminstatus, %sifoperstatus, %smaxbandwidth);

      while (scalar @asxportnumber) {

	    $index = shift @asxportnumber;

	    $sportnumber{$index} = $index;

	    $sportname{$index} = shift @asxhwportname;

	    $smaxbandwidth{$index} = shift @asxmaxbandwidthin;

	    $sifadminstatus{$index} = shift @asxifadminstatus;

	    $sifoperstatus{$index} = shift @asxifoperstatus;

	    $ipaddr{$index} = shift @asxipaddress;

	    $ipiladdr{$index} = shift @asxilmiaddress;

	    $iphost{$index} = 

	      gethostbyaddr(pack('C4',split(/\./,$ipaddr{$index})), AF_INET);

	    if ($iphost{$index} eq ''){

		 $iphost{$index} = $ipaddr{$index};

	    }

	    $ipilhost{$index} = 

	      gethostbyaddr(pack('C4',split(/\./,$ipiladdr{$index})), AF_INET);

	    if ($ipilhost{$index} eq ''){

		 $ipilhost{$index} = $ipiladdr{$index};

	    }

      }

      foreach $index (keys %sportnumber) {

 	my $c;

	my $speed = int($smaxbandwidth{$index}*53); # cells to bytes

 	my $speed_str=&fmi($speed);

	my $name="$router.$index";

	if (($sifoperstatus{$index} != 1)

		&& ($sifadminstatus{$index} == 3)) {

	    print <

########

######## This Interface is one of the following

######## - administratively not UP

######## - it is in test mode

######## - it is a softwareLoopback interface

######## - has a unrealistic speed setting

######## It is commented out for this reason.

########

ECHO

	    $c="# ";

	} else {

	    $c = '';

	}

	print <

${c}

${c}Target[$name]: CellsRcvd.$index&CellsTrans.$index:$community\@$router \* 53

${c}MaxBytes[$name]: $speed

${c}Options[$name]: growright

${c}WithPeak[$name]: dwmy

${c}Title[$name]: $sysName ($iphost{$index}): $sportname{$index}

${c}PageTop[$name]: 
Traffic Analysis for $sportname{$index}

${c} 

${c} 

${c}   System:$sysName in $sysLocation

${c}   Maintainer:$sysContact

${c}   Interface:$sportname{$index} ($index)

ECHO

	print <

${c}   Remote IP:$iphost{$index} ($ipaddr{$index})

ECHO

	print <

${c}   Remote ILMI IP:$ipilhost{$index} ($ipiladdr{$index})

ECHO

	print <

${c}   Max Speed:

${c}       $speed_str (Sonet)

${c}  

${c}

#---------------------------------------------------------------

ECHO

    }

  }

  else {

	  my(%sifdesc,%siftype,%sifspeed,%sifadminstatus,%sifoperstatus,%sciscodescr);

	  while (scalar @ifindex) {

	  # as these arrays get filled from the bottom, 

	  # we need to empty them from the botom as well ...

	  # fifo

	    $index = shift @ifindex;

	    $sifdesc{$index} = shift @ifdescr;

	    $siftype{$index} = shift @iftype;

	    print STDERR $siftype{$index};

	    $sifspeed{$index} = shift @ifspeed;

	    $sifadminstatus{$index} = shift @ifadminstatus;

	    $sifoperstatus{$index} = shift @ifoperstatus;

	    # especially since cisco does not return a if

	    # descr for each interface it has ...

	    $sciscodescr{$index} = "
" . (shift @ciscodescr) if @ciscodescr;

	}

	foreach $index (keys %sifdesc) {

	my $c;

	my $speed = int($sifspeed{$index} / 8); # bits to byte

 	my $speed_str=&fmi($speed);

	my $name="$router.$index";

	if (($sifadminstatus{$index} != 1)

		|| ($siftype{$index} == 24)

		|| ($speed == 0 ) 

		|| ($speed > 400 * 10**6)  #speeds of 400 MByte/s are not realistic

		|| ($sifoperstatus{$index} == 3)) {

	print <

########

######## This Interface is one of the following

######## - administratively not UP

######## - it is in test mode

######## - it is a softwareLoopback interface

######## - has a unrealistic speed setting

######## It is commented out for this reason.

########

ECHO

  $c="# ";

  }else {

    $c = '';

  }

  print <

${c}

${c}Target[$name]: $index:$community\@$router

${c}MaxBytes[$name]: $speed

${c}Options[$name]: growright

${c}WithPeak[$name]: dwmy

${c}Title[$name]: $sysName ($iphost{$index}): $sifdesc{$index}

${c}PageTop[$name]: 
Traffic Analysis for $sifdesc{$index}

${c} $sciscodescr{$index}

${c} 

${c}   System:$sysName in $sysLocation

${c}   Maintainer:$sysContact

${c}   Interface:$sifdesc{$index} ($index)

${c}   IP:$iphost{$index} ($ipaddr{$index})

${c}   Max Speed:

${c}       $speed_str ($ifType_d{$siftype{$index}})

${c}  

${c}

#---------------------------------------------------------------

ECHO

  }

 }

}

main;

exit(0);

sub snmpget{  

  my($host,$community,@vars) = @_;

  my(@enoid, $var,$response, $bindings, $binding, $value, $inoid,$outoid,

     $upoid,$oid,@retvals);

  foreach $var (@vars) {

    die "Unknown SNMP var $var\n" 

      unless $snmpget::OIDS{$var};

    push @enoid,  encode_oid((split /\./, $snmpget::OIDS{$var}));

  }

  srand();

  my $session = SNMP_Session->open ($host ,

                                 $community, 

                                 161);

  if ($session->get_request_response(@enoid)) {

    $response = $session->pdu_buffer;

    ($bindings) = $session->decode_get_response ($response);

    $session->close ();    

    while ($bindings) {

      ($binding,$bindings) = decode_sequence ($bindings);

      ($oid,$value) = decode_by_template ($binding, "%O%@");

      my $tempo = pretty_print($value);

      $tempo=~s/\t/ /g;

      $tempo=~s/\n/ /g;

      $tempo=~s/^\s+//;

      $tempo=~s/\s+$//;

      push @retvals,  $tempo;

    }

    return (@retvals);

  } else {

    die "No answer from $ARGV[0]. You may be using the wrong community\n";

  }

}

sub snmpgettable{

  my($host,$community,$var) = @_;

  my($next_oid,$enoid,$orig_oid, 

     $response, $bindings, $binding, $value, $inoid,$outoid,

     $upoid,$oid,@table,$tempo);

  die "Unknown SNMP var $var\n" 

    unless $snmpget::OIDS{$var};

  $orig_oid = encode_oid(split /\./, $snmpget::OIDS{$var});

  $enoid=$orig_oid;

  srand();

  my $session = SNMP_Session->open ($host ,

                                 $community, 

                                 161);

  for(;;)  {

    if ($session->getnext_request_response(($enoid))) {

      $response = $session->pdu_buffer;

      ($bindings) = $session->decode_get_response ($response);

      ($binding,$bindings) = decode_sequence ($bindings);

      ($next_oid,$value) = decode_by_template ($binding, "%O%@");

      # quit once we are outside the table

      last unless BER::encoded_oid_prefix_p($orig_oid,$next_oid);

      $tempo = pretty_print($value);

      #print "$var: '$tempo'\n";

      $tempo=~s/\t/ /g;

      $tempo=~s/\n/ /g;

      $tempo=~s/^\s+//;

      $tempo=~s/\s+$//;

      push @table, $tempo;

    } else {

      die "No answer from $ARGV[0]\n";

    }

    $enoid=$next_oid;

  }

  $session->close ();    

  return (@table);

}

sub fmi {

  my($number) = $_[0];

  my(@short);

  @short = ("Bytes/s","kBytes/s","MBytes/s","GBytes/s","TBytes/s");

  my $digits=length("".$number);

  my $divm=0;

  while ($digits-$divm*3 > 4) { $divm++; }

  my $divnum = $number/10**($divm*3);

  return sprintf("%1.1f %s",$divnum,$short[$divm]);

}