<?php

### Discover BGP peers

  echo("BGP Sessions : ");

  $as_cmd  = $config['snmpwalk'] . " -M " . $config['mibdir'] . " -m BGP4-MIB -CI -Oqvn -" . $device['snmpver'] . " -c" . $device['community'] . " " . $device['hostname'].":".$device['port'] . " ";
  $as_cmd .= ".1.3.6.1.2.1.15.2";
  $bgpLocalAs = trim(shell_exec($as_cmd));
  
  if(is_numeric($bgpLocalAs))
  {
    echo("AS$bgpLocalAs ");

    if($bgpLocalAs != $device['bgpLocalAs']) 
    {
      mysql_query("UPDATE devices SET bgpLocalAs = '$bgpLocalAs' WHERE device_id = '".$device['device_id']."'"); echo("Updated AS ");
    }

    $peers_cmd  = $config['snmpwalk'] . " -M " . $config['mibdir'] . " -m BGP4-MIB -CI -Oq -" . $device['snmpver'] . " -c" . $device['community'] . " " . $device['hostname'].":".$device['port'] . " ";
    $peers_cmd .= "BGP4-MIB::bgpPeerRemoteAs"; 
    $peers_data = shell_exec($peers_cmd);
    if($debug) { echo("Peers : $peers_cmd --> \n$peers_data \n"); }
    $peers = trim(str_replace("BGP4-MIB::bgpPeerRemoteAs.", "", $peers_data));  

    foreach (explode("\n", $peers) as $peer)  
    {
      list($peer_ip, $peer_as) = split(" ",  $peer);
      
      if($peer && $peer_ip != "0.0.0.0") 
      {
        if ($debug) echo "Found peer $peer_ip (AS$peer_as)\n";
	$peerlist[] = array('ip' => $peer_ip, 'as' => $peer_as);
      } 
    } # Foreach  

    if ($device['os'] == "junos")
    {
      ## Juniper BGP4-V2 MIB, ipv6 only for now, because v4 should be covered in BGP4-MIB above
      $peers_cmd  = $config['snmpwalk'] . " -M " . $config['mibdir'] . " -M +".$config['install_dir']."/mibs/junos -m BGP4-V2-MIB-JUNIPER -CI -Onq -" . $device['snmpver'] . " -c" . $device['community'] . " " . $device['hostname'].":".$device['port'] . " ";
      $peers_cmd .= "jnxBgpM2PeerRemoteAs.0.ipv6";  # FIXME: is .0 the only possible value here?
      $peers = trim(str_replace(".1.3.6.1.4.1.2636.5.1.1.2.1.1.1.13.0.","", `$peers_cmd`));  
      foreach (explode("\n", $peers) as $peer)  
      {
        list($peer_ip_snmp, $peer_as) = split(" ",  $peer);

        # Magic! Basically, takes SNMP form and finds peer IPs from the walk OIDs.
        $peer_ip = Net_IPv6::compress(snmp2ipv6(implode('.',array_slice(explode('.',$peer_ip_snmp),count(explode('.',$peer_ip_snmp))-16))));
      
        if($peer) 
        {
          if ($debug) echo "Found peer $peer_ip (AS$peer_as)\n";
	  $peerlist[] = array('ip' => $peer_ip, 'as' => $peer_as);
        }
      } # Foreach  
    } # OS junos
  } else { 
    echo("No BGP on host");
    if($device['bgpLocalAs']) {
     mysql_query("UPDATE devices SET bgpLocalAs = NULL WHERE device_id = '".$device['device_id']."'"); echo(" (Removed ASN) "); 
    } # End if
  } # End if

## Process disovered peers

if (isset($peerlist))
{
  foreach ($peerlist as $peer)
  {
    $astext = get_astext($peer['as']);
        
    if(mysql_result(mysql_query("SELECT COUNT(*) FROM `bgpPeers` WHERE `device_id` = '".$device['device_id']."' AND bgpPeerIdentifier = '".$peer['ip']."'"),0) < '1') 
    {
      $add = mysql_query("INSERT INTO bgpPeers (`device_id`, `bgpPeerIdentifier`, `bgpPeerRemoteAS`) VALUES ('".$device['device_id']."','".$peer['ip']."','".$peer['as']."')");
      echo("+"); 
    } else { 
      $update = mysql_query("UPDATE `bgpPeers` SET bgpPeerRemoteAs = " . $peer['as'] . ", astext = '" . mres($astext) . "' WHERE `device_id` = '".$device['device_id']."' AND bgpPeerIdentifier = '".$peer['ip']."'");
      echo(".");
    }

    ## Get afi/safi and populate cbgp on cisco ios (xe/xr)
    if ($device['os'] == "ios") 
    {
      unset($af_list);
      $af_cmd  = $config['snmpwalk'] . " -M " . $config['mibdir'] . " -CI -m CISCO-BGP4-MIB -OsQ -" . $device['snmpver'] . " -c" . $device['community'] . " " . $device['hostname'].":".$device['port'] . " ";
      $af_cmd .= "cbgpPeerAddrFamilyName." . $peer['ip'];
      $af_data = shell_exec($af_cmd);
      if($debug) { echo("afi data :: $af_cmd --> \n $af_data \n"); }
      $afs = trim(str_replace("cbgpPeerAddrFamilyName.".$peer['ip'].".", "", $af_data));
      foreach (explode("\n", $afs) as $af)  
      {
        if($debug) { echo("AFISAFI = $af\n"); }
        list($afisafi, $text) = explode(" = ", $af);
        list($afi, $safi) = explode(".", $afisafi);
        if($afi && $safi) 
        {
          $af_list['$afi']['$safi'] = 1;
          if(mysql_result(mysql_query("SELECT COUNT(*) FROM `bgpPeers_cbgp` WHERE `device_id` = '".$device['device_id']."' AND bgpPeerIdentifier = '".$peer['ip']."' AND afi = '$afi' AND safi = '$safi'"),0) == 0) 
          {
            mysql_query("INSERT INTO `bgpPeers_cbgp` (device_id,bgpPeerIdentifier, afi, safi) VALUES ('".$device['device_id']."','".$peer['ip']."','$afi','$safi')");
          }
        }
      }
    
      $af_query = mysql_query("SELECT * FROM bgpPeers_cbgp WHERE `device_id` = '".$device['device_id']."' AND bgpPeerIdentifier = '".$peer['ip']."'");
      while ($entry = mysql_fetch_array($af_query)) 
      {
        $afi = $entry['afi'];
        $afi = $entry['safi'];
        if (!$af_list['$afi']['$safi']) 
        {
          mysql_query("DELETE FROM `bgpPeers_cbgp` WHERE `device_id` = '".$device['device_id']."' AND bgpPeerIdentifier = '".$peer['ip']."' AND afi = '$afi' AND safi = '$safi'");
        }
      } # AF list
    } # if os = ios 
  } # Foreach
} # isset

## Delete removed peers

$sql = "SELECT * FROM bgpPeers AS B, devices AS D WHERE B.device_id = D.device_id AND D.device_id = '".$device['device_id']."'";
$query = mysql_query($sql);

while ($entry = mysql_fetch_array($query)) {
        unset($exists);
        $i = 0;
        while ($i < count($peerlist) && !isset($exists)) 
        {
          if ($peerlist[$i]['ip'] == $entry['bgpPeerIdentifier']) { $exists = 1; }
          $i++;
        }
        if(!isset($exists)) { 
          mysql_query("DELETE FROM bgpPeers WHERE bgpPeer_id = '" . $entry['bgpPeer_id'] . "'"); 
	  mysql_query("DELETE FROM bgpPeers_cbgp WHERE bgpPeer_id = '" . $entry['bgpPeer_id'] . "'");
          echo("-");
        }
}

unset($peerlist);

echo("\n");

?>
