<?php
if (empty($_POST['type'])) die('Missing options.');
$type = mb_strtolower($_POST['type']);
$ip = $_SERVER['REMOTE_ADDR'];
if (substr($ip, 0, 7) === '::ffff:') $ip = substr($ip, 7);
/*
* Section 1/2
* Power/hw/usage stats
*/
if ($type{0} === '~') {
// UUID is mandatory
$uuid = Request::post('uuid', '', 'string');
if (strlen($uuid) !== 36) die("Invalid UUID.\n");
$macaddr = Request::post('macaddr', '', 'string');
if (!empty($macaddr) && substr($uuid, 0, 16) === '000000000000000-') {
// Override uuid if the mac is known and unique
$res = Database::simpleQuery('SELECT machineuuid FROM machine WHERE macaddr = :macaddr AND machineuuid <> :uuid', compact('macaddr', 'uuid'));
$override = false;
while ($row = $res->fetch(PDO::FETCH_ASSOC)) {
if ($override !== false) {
$override = false;
break;
}
$override = $row['machineuuid'];
}
if ($override !== false) {
$uuid = $override;
}
}
$NOW = time();
$old = Database::queryFirst('SELECT clientip, logintime, lastseen FROM machine WHERE machineuuid = :uuid', array('uuid' => $uuid));
if ($old !== false) {
settype($old['logintime'], 'integer');
settype($old['lastseen'], 'integer');
}
// Handle event type
if ($type === '~poweron') {
// Poweron & hw stats
$uptime = Request::post('uptime', '', 'integer');
if (strlen($macaddr) > 17) die("Invalid MAC.\n");
if ($uptime < 0 || $uptime > 4000000) die("Implausible uptime.\n");
$realcores = Request::post('realcores', 0, 'integer');
if ($realcores < 0 || $realcores > 512) $realcores = 0;
$mbram = Request::post('mbram', 0, 'integer');
if ($mbram < 0 || $mbram > 2048000) $mbram = 0;
$kvmstate = Request::post('kvmstate', 'UNKNOWN', 'string');
$valid = array('UNKNOWN', 'UNSUPPORTED', 'DISABLED', 'ENABLED');
if (!in_array($kvmstate, $valid)) $kvmstate = 'UNKNOWN';
$cpumodel = Request::post('cpumodel', '', 'string');
$systemmodel = Request::post('systemmodel', '', 'string');
$id44mb = Request::post('id44mb', 0, 'integer');
if ($id44mb < 0 || $id44mb > 10240000) $id44mb = 0;
$badsectors = Request::post('badsectors', 0, 'integer');
if ($badsectors < 0 || $badsectors > 100000) $badsectors = 0;
$hostname = gethostbyaddr($ip);
if (!is_string($hostname) || $hostname === $ip) {
$hostname = '';
}
$data = Request::post('data', '', 'string');
if ($uptime < 120) {
// See if we have a lingering session, create statistic entry if so
if ($old !== false && $old['logintime'] !== 0) {
$sessionLength = $old['lastseen'] - $old['logintime'];
if ($sessionLength > 30 && $sessionLength < 86400*2) {
Database::exec('INSERT INTO statistic (dateline, typeid, machineuuid, clientip, username, data)'
. " VALUES (:start, '~session-length', :uuid, :clientip, '', :length)", array(
'start' => $old['logintime'],
'uuid' => $uuid,
'clientip' => $ip,
'length' => $sessionLength
));
}
}
// Write poweroff period length to statistic table
if ($old !== false && $old['lastseen'] !== 0) {
$lastSeen = $old['lastseen'];
$offtime = ($NOW - $uptime) - $lastSeen;
if ($offtime > 300 && $offtime < 86400 * 90) {
Database::exec('INSERT INTO statistic (dateline, typeid, machineuuid, clientip, username, data)'
. " VALUES (:shutdown, '~offline-length', :uuid, :clientip, '', :length)", array(
'shutdown' => $lastSeen,
'uuid' => $uuid,
'clientip' => $ip,
'length' => $offtime
));
}
}
}
// Create/update machine entry
Database::exec('INSERT INTO machine '
. '(machineuuid, macaddr, clientip, firstseen, lastseen, logintime, position, lastboot, realcores, mbram,'
. ' kvmstate, cpumodel, systemmodel, id44mb, badsectors, data, hostname) VALUES '
. "(:uuid, :macaddr, :clientip, :firstseen, :lastseen, 0, '', :lastboot, :realcores, :mbram,"
. ' :kvmstate, :cpumodel, :systemmodel, :id44mb, :badsectors, :data, :hostname)'
. ' ON DUPLICATE KEY UPDATE'
. ' macaddr = VALUES(macaddr),'
. ' clientip = VALUES(clientip),'
. ' lastseen = VALUES(lastseen),'
. ($uptime < 180 ? ' logintime = 0, currentuser = NULL, currentsession = NULL,' : '')
. ' lastboot = VALUES(lastboot),'
. ' realcores = VALUES(realcores),'
. ' mbram = VALUES(mbram),'
. ' kvmstate = VALUES(kvmstate),'
. ' cpumodel = VALUES(cpumodel),'
. ' systemmodel = VALUES(systemmodel),'
. ' id44mb = VALUES(id44mb),'
. ' badsectors = VALUES(badsectors),'
. ' data = VALUES(data),'
. " hostname = If(VALUES(hostname) = '', hostname, VALUES(hostname))", array(
'uuid' => $uuid,
'macaddr' => $macaddr,
'clientip' => $ip,
'firstseen' => $NOW,
'lastseen' => $NOW,
'lastboot' => $NOW - $uptime,
'realcores' => $realcores,
'mbram' => $mbram,
'kvmstate' => $kvmstate,
'cpumodel' => $cpumodel,
'systemmodel'=> $systemmodel,
'id44mb' => $id44mb,
'badsectors' => $badsectors,
'data' => $data,
'hostname' => $hostname,
));
if (($old === false || $old['clientip'] !== $ip) && Module::isAvailable('locations')) {
// New, or ip changed (dynamic pool?), update subnetlicationid
Location::updateMapIpToLocation($uuid, $ip);
}
// Write statistics data
} else if ($type === '~runstate') {
// Usage (occupied/free)
$sessionLength = 0;
$used = Request::post('used', 0, 'integer');
if ($old === false) die("Unknown machine.\n");
if ($old['clientip'] !== $ip) {
EventLog::warning("[runstate] IP address of client $uuid seems to have changed ({$old['clientip']} -> $ip)");
die("Address changed.\n");
}
// Figure out what's happening
if ($used === 0) {
// Is not in use
if ($old['logintime'] !== 0) {
// Was in use, is free now
// 1) Log last session length
if ($NOW - $old['lastseen'] > 610) {
// Old session timed out - might be caused by hard reboot
$sessionLength = $old['lastseen'] - $old['logintime'];
} else {
$sessionLength = $NOW - $old['logintime'];
}
}
Database::exec('UPDATE machine SET lastseen = UNIX_TIMESTAMP(), logintime = 0 WHERE machineuuid = :uuid', array('uuid' => $uuid));
} else {
// Machine is in use
if ($old['logintime'] !== 0 && $NOW - $old['lastseen'] > 610) {
// Old session timed out - might be caused by hard reboot
$sessionLength = $old['lastseen'] - $old['logintime'];
}
if ($sessionLength !== 0 || $old['logintime'] === 0) {
// This event is a start of a new session, rather than an update
Database::exec('UPDATE machine SET lastseen = UNIX_TIMESTAMP(), logintime = UNIX_TIMESTAMP() WHERE machineuuid = :uuid', array('uuid' => $uuid));
} else {
// Nothing changed, simple lastseen update
Database::exec('UPDATE machine SET lastseen = UNIX_TIMESTAMP() WHERE machineuuid = :uuid', array('uuid' => $uuid));
}
}
// 9) Log last session length if applicable
if ($sessionLength > 0 && $sessionLength < 86400*2 && $old['logintime'] !== 0) {
Database::exec('INSERT INTO statistic (dateline, typeid, machineuuid, clientip, username, data)'
. " VALUES (:start, '~session-length', :uuid, :clientip, '', :length)", array(
'start' => $old['logintime'],
'uuid' => $uuid,
'clientip' => $ip,
'length' => $sessionLength
));
}
} elseif ($type === '~poweroff') {
if ($old === false) die("Unknown machine.\n");
if ($old['clientip'] !== $ip) {
EventLog::warning("[poweroff] IP address of client $uuid seems to have changed ({$old['clientip']} -> $ip)");
die("Address changed.\n");
}
if ($old['logintime'] !== 0) {
$sessionLength = $old['lastseen'] - $old['logintime'];
if ($sessionLength > 0 && $sessionLength < 86400*2) {
Database::exec('INSERT INTO statistic (dateline, typeid, machineuuid, clientip, username, data)'
. " VALUES (:start, '~session-length', :uuid, :clientip, '', :length)", array(
'start' => $old['logintime'],
'uuid' => $uuid,
'clientip' => $ip,
'length' => $sessionLength
));
}
}
Database::exec('UPDATE machine SET logintime = 0, lastseen = UNIX_TIMESTAMP(), lastboot = 0 WHERE machineuuid = :uuid', array('uuid' => $uuid));
} elseif ($type === '~screens') {
$screens = Request::post('screen', false, 'array');
if (is_array($screens)) {
// `devicetype`, `devicename`, `subid`, `machineuuid`
// Make sure all screens are in the general hardware table
$hwids = array();
foreach ($screens as $port => $screen) {
if (!array_key_exists('name', $screen))
continue;
if (array_key_exists($screen['name'], $hwids)) {
$hwid = $hwids[$screen['name']];
} else {
$hwid = (int)Database::insertIgnore('statistic_hw', 'hwid',
array('hwtype' => DeviceType::SCREEN, 'hwname' => $screen['name']));
$hwids[$screen['name']] = $hwid;
}
// Now add new entries
$machinehwid = Database::insertIgnore('machine_x_hw', 'machinehwid', array(
'hwid' => $hwid,
'machineuuid' => $uuid,
'devpath' => $port,
), array('disconnecttime' => 0));
$validProps = array();
if (count($screen) > 1) {
// Screen has additional properties (resolution, size, etc.)
unset($screen['name']);
foreach ($screen as $key => $value) {
if (!preg_match('/^[a-zA-Z0-9][\x21-\x7e]{0,15}$/', $key)) {
echo "No matsch '$key'\n";
continue; // Ignore evil key names
}
$validProps[] = $key;
Database::exec("INSERT INTO machine_x_hw_prop (machinehwid, prop, value)"
. " VALUES (:id, :key, :value) ON DUPLICATE KEY UPDATE value = VALUES(value)", array(
'id' => $machinehwid,
'key' => $key,
'value' => $value,
));
}
}
// Purge properties that might have existed in the past
if (empty($validProps)) {
Database::exec("DELETE FROM machine_x_hw_prop WHERE machinehwid = :machinehwid AND prop NOT LIKE '@%'",
array('machinehwid' => $machinehwid));
} else {
$qs = '?' . str_repeat(',?', count($validProps) - 1);
array_unshift($validProps, $machinehwid);
Database::exec("DELETE FROM machine_x_hw_prop"
. " WHERE machinehwid = ? AND prop NOT LIKE '@%' AND prop NOT IN ($qs)",
$validProps);
}
}
// Remove/disable stale entries
if (empty($hwids)) {
// No screens connected at all, purge all screen entries for this machine
Database::exec("UPDATE machine_x_hw x, statistic_hw h"
. " SET x.disconnecttime = UNIX_TIMESTAMP()"
. " WHERE x.machineuuid = :uuid AND x.hwid = h.hwid AND h.hwtype = :type AND x.disconnecttime = 0",
array('uuid' => $uuid, 'type' => DeviceType::SCREEN));
} else {
// Some screens connected, make sure old entries get removed
$params = array_values($hwids);
array_unshift($params, $uuid);
array_unshift($params, DeviceType::SCREEN);
$qs = '?' . str_repeat(',?', count($hwids) - 1);
Database::exec("UPDATE machine_x_hw x, statistic_hw h"
. " SET x.disconnecttime = UNIX_TIMESTAMP()"
. " WHERE h.hwid = x.hwid AND x.disconnecttime = 0 AND h.hwtype = ? AND x.machineuuid = ? AND x.hwid NOT IN ($qs)", $params);
}
}
} else {
die("INVALID ACTION '$type'");
}
die("OK. (RESULT=0)\n");
}
/*
* Section 2/2
* Session information
*/
function writeStatisticLog($type, $username, $data)
{
global $ip;
// Spam from IP
$row = Database::queryFirst('SELECT Count(*) AS cnt FROM statistic WHERE clientip = :client AND dateline + 300 > UNIX_TIMESTAMP()', array(':client' => $ip));
if ($row !== false && $row['cnt'] > 8) {
return;
}
Database::exec('INSERT INTO statistic (dateline, typeid, clientip, username, data) VALUES (UNIX_TIMESTAMP(), :type, :client, :username, :data)', array(
'type' => $type,
'client' => $ip,
'username' => $username,
'data' => $data,
));
}
// For backwards compat, we require the . prefix
if ($type{0} === '.') {
if ($type === '.vmchooser-session') {
$user = Request::post('user', 'unknown', 'string');
$loguser = Request::post('loguser', 0, 'int') !== 0;
$sessionName = Request::post('name', 'unknown', 'string');
$sessionUuid = Request::post('uuid', '', 'string');
$session = strlen($sessionUuid) === 36 ? $sessionUuid : $sessionName;
Database::exec("UPDATE machine SET currentuser = :user, currentsession = :session WHERE clientip = :ip",
compact('user', 'session', 'ip'));
writeStatisticLog('.vmchooser-session-name', ($loguser ? $user : 'anonymous'), $sessionName);
} else {
if (!isset($_POST['description'])) die('Missing options..');
$description = $_POST['description'];
// and username embedded in message
if (preg_match('#^\[([^\]]+)\]\s*(.*)$#m', $description, $out)) {
writeStatisticLog($type, $out[1], $out[2]);
}
}
}
echo "OK.\n";
