librenms-librenms/includes/polling/applications/proxmox.inc.php

<?php

/**
 * Check if a port on a Proxmox VM exists
 * @param string  $p Port name
 * @param string  $c Clustername
 * @param integer $i VM ID
 * @return integer|boolean The port-ID if the port exists, false if it doesn't exist
 */
function proxmox_port_exists($i, $c, $p) {
    if ($row = dbFetchRow("SELECT pmp.id FROM proxmox_ports pmp, proxmox pm WHERE pm.id = pmp.vm_id AND pmp.port = ? AND pm.cluster = ? AND pm.vmid = ?", array($p, $c, $i))) {
        return $row['id'];
    }

    return false;
}

/**
 * Check if a Proxmox VM exists
 * @param integer $i VM ID
 * @param string  $c Clustername
 * @param array   $pmxcache Reference to the Proxmox VM Cache
 * @return boolean true if the VM exists, false if it doesn't
 */
function proxmox_vm_exists($i, $c, &$pmxcache) {

    if (isset($pmxcache[$c][$i]) && $pmxcache[$c][$i] > 0) {
        return true;
    }
    if ($row = dbFetchRow("SELECT id FROM proxmox WHERE vmid = ? AND cluster = ?", array($i, $c))) {
        $pmxcache[$c][$i] = (integer) $row['id'];
        return true;
    }

    return false;
}

$name = 'proxmox';
$app_id = $app['app_id'];

if (isset($config['enable_proxmox']) && $config['enable_proxmox'] && !empty($agent_data['app'][$name])) {
    $proxmox = $agent_data['app'][$name];
    $pmxlines = explode("\n", $proxmox);
    $pmxcluster = array_shift($pmxlines);
    dbUpdate(array('device_id' => $device['device_id'], 'app_type' => $name, 'app_instance' => $pmxcluster),
        'applications', '`device_id` = ? AND `app_type` = ?', array($device['device_id'], $name));

    if (count($pmxlines) > 0) {
        $pmxcache = array();

        foreach ($pmxlines as $vm) {
            list($vmid, $vmport, $vmpin, $vmpout, $vmdesc) = explode('/', $vm, 5);
            print "Proxmox ($pmxcluster): $vmdesc: $vmpin/$vmpout/$vmport\n";

            $rrd_proxmox_name = array(
                'pmxcluster' => $pmxcluster,
                'vmid' => $vmid,
                'vmport' => $vmport
            );
            $rrd_def = array(
                'DS:INOCTETS:DERIVE:600:0:12500000000',
                'DS:OUTOCTETS:DERIVE:600:0:12500000000'
            );
            $fields = array(
                'INOCTETS' => $vmpin,
                'OUTOCTETS' => $vmpout
            );

            $tags = compact('name', 'app_id', 'pmxcluster', 'vmid', 'vmport', 'rrd_proxmox_name', 'rrd_def');
            data_update($device, 'app', $tags, $fields);

            if (proxmox_vm_exists($vmid, $pmxcluster, $pmxcache) === true) {
                dbUpdate(array(
                    'device_id' => $device['device_id'],
                    'last_seen' => array('NOW()'),
                    'description' => $vmdesc
                ), $name, '`vmid` = ? AND `cluster` = ?', array($vmid, $pmxcluster));
            } else {
                $pmxcache[$pmxcluster][$vmid] = dbInsert(array(
                    'cluster' => $pmxcluster,
                    'vmid' => $vmid,
                    'description' => $vmdesc,
                    'device_id' => $device['device_id']
                ), $name);
            }

            if ($portid = proxmox_port_exists($vmid, $pmxcluster, $vmport) !== false) {
                dbUpdate(array('last_seen' => array('NOW()')), 'proxmox_ports', '`vm_id` = ? AND `port` = ?',
                    array($pmxcache[$pmxcluster][$vmid], $vmport));
            } else {
                dbInsert(array('vm_id' => $pmxcache[$pmxcluster][$vmid], 'port' => $vmport), 'proxmox_ports');
            }
        }
    }
}

unset($pmxlines, $pmxcluster, $pmxcdir, $proxmox, $pmxcache);