librenms/includes/polling/unix-agent.inc.php

<?php

global $debug;

if ($device['os_group'] == "unix")
{
  echo($config['project_name']." UNIX Agent: ");

  // FIXME - this should be in config and overridable in database
  $agent_port='6556';

  $agent_start = utime();
  $agent = fsockopen($device['hostname'], $agent_port, $errno, $errstr, 10);

  if (!$agent)
  {
    echo "Connection to UNIX agent failed on port ".$port.".";
  } else {
    while (!feof($agent))
    {
      $agent_raw .= fgets($agent, 128);
    }
  }
  
  $agent_end = utime(); $agent_time = round(($agent_end - $agent_start) * 1000);

  if (!empty($agent_raw))
  {
    echo("execution time: ".$agent_time."ms");
    $agent_rrd = $config['rrd_dir'] . "/" . $device['hostname'] . "/agent.rrd";
    if (!is_file($agent_rrd))
    {
      rrdtool_create ($agent_rrd, "DS:time:GAUGE:600:0:U ".$config['rrd_rra']);
    }
    rrdtool_update($agent_rrd, "N:".$agent_time);
    $graphs['agent'] = TRUE;

    foreach (explode("<<<", $agent_raw) as $section)
    {
      list($section, $data) = explode(">>>", $section);
      list($sa, $sb) = explode("-", $section, 2);

      if ($section == "apache") { $sa = "app"; $sb = "apache"; }
      if ($section == "mysql")  { $sa = "app"; $sb = "mysql"; }
      if ($section == "nginx")  { $sa = "app"; $sb = "nginx"; }
      if ($section == "bind")  { $sa = "app"; $sb = "bind"; }
      if ($section == "tinydns")  { $sa = "app"; $sb = "tinydns"; }
#      if ($section == "drbd")   { $sa = "app"; $sb = "drbd"; }

      if (!empty($sa) && !empty($sb))
      {
        $agent_data[$sa][$sb] = trim($data);
      } else {
        $agent_data[$section] = trim($data);
      }
    }

    if ($debug) { print_r($agent_data); }

    include("unix-agent/packages.inc.php");
    include("unix-agent/munin-plugins.inc.php");

    foreach (array_keys($agent_data) as $key)
    {
      if (file_exists("includes/polling/unix-agent/$key.inc.php"))
      {
        if ($debug) { echo("Including: unix-agent/$key.inc.php"); }

        include("unix-agent/$key.inc.php");
      }
    }

    foreach (array_keys($agent_data['app']) as $key)
    {
      if (file_exists("includes/polling/applications/$key.inc.php"))
      {
        if ($debug) { echo("Including: applications/$key.inc.php"); }

        include("applications/$key.inc.php");
      }
    }

    // Processes
    if (!empty($agent_data['ps']))
    {
      echo("Processes: ");
      dbDelete('processes', 'device_id = ?', array($device['device_id']));
      foreach (explode("\n", $agent_data['ps']) as $process)
      {
        $process = preg_replace("/\((.*),([0-9]*),([0-9]*),([0-9\:]*),([0-9]*)\)\ (.*)/", "\\1|\\2|\\3|\\4|\\5|\\6", $process);
        list($user, $vsz, $rss, $cputime, $pid, $command) = explode("|", $process, 6);
        if( !empty($command) ) {
          dbInsert(array('device_id' => $device['device_id'], 'pid' => $pid, 'user' => $user, 'vsz' => $vsz, 'rss' => $rss, 'cputime' => $cputime, 'command' => $command), 'processes');
        }
      }
      echo("\n");
    }

    // Apache
    if (!empty($agent_data['app']['apache']))
    {
      $app_found['apache'] = TRUE;
      if (dbFetchCell("SELECT COUNT(*) FROM `applications` WHERE `device_id` = ? AND `app_type` = ?", array($device['device_id'], 'apache')) == "0")
      {
        echo("Found new application 'Apache'\n");
        dbInsert(array('device_id' => $device['device_id'], 'app_type' => 'apache'), 'applications');
      }
    }

    // memcached
    if (!empty($agent_data['app']['memcached']))
    {
      $agent_data['app']['memcached'] = unserialize($agent_data['app']['memcached']);
      foreach ($agent_data['app']['memcached'] as $memcached_host => $memcached_data)
      {
        if (dbFetchCell("SELECT COUNT(*) FROM `applications` WHERE `device_id` = ? AND `app_type` = ? AND `app_instance` = ?", array($device['device_id'], 'memcached', $memcached_host)) == "0")
        {
          echo("Found new application 'Memcached' $memcached_host\n");
          dbInsert(array('device_id' => $device['device_id'], 'app_type' => 'memcached', 'app_instance' => $memcached_host), 'applications');
        }
      }
    }

    // MySQL
    if (!empty($agent_data['app']['mysql']))
    {
      $app_found['mysql'] = TRUE;
      if (dbFetchCell("SELECT COUNT(*) FROM `applications` WHERE `device_id` = ? AND `app_type` = ?", array($device['device_id'], 'mysql')) == "0")
      {
        echo("Found new application 'MySQL'\n");
        dbInsert(array('device_id' => $device['device_id'], 'app_type' => 'mysql'), 'applications');
      }
    }

    // DRBD
    if (!empty($agent_data['drbd']))
    {
      $agent_data['app']['drbd'] = array();
      foreach (explode("\n", $agent_data['drbd']) as $drbd_entry)
      {
        list($drbd_dev, $drbd_data) = explode(":", $drbd_entry);
        if (preg_match("/^drbd/", $drbd_dev))
        {
          $agent_data['app']['drbd'][$drbd_dev] = $drbd_data;
          if (dbFetchCell("SELECT COUNT(*) FROM `applications` WHERE `device_id` = ? AND `app_type` = ? AND `app_instance` = ?", array($device['device_id'], 'drbd', $drbd_dev)) == "0")
          {
            echo("Found new application 'DRBd' $drbd_dev\n");
            dbInsert(array('device_id' => $device['device_id'], 'app_type' => 'drbd', 'app_instance' => $drbd_dev), 'applications');
          }
        }
      }
    }
  }

  if (!empty($agent_sensors))
  {
    echo("Sensors: ");
    check_valid_sensors($device, 'temperature', $valid['sensor'], 'agent');
    echo("\n");
  }

  echo("\n");
}

?>