From 67c915aa7094a32b0dfdd4ef60382bb1d3eeb506 Mon Sep 17 00:00:00 2001 From: Simon Rettberg Date: Thu, 21 Feb 2019 11:09:17 +0100 Subject: Fix UUID handling; refactor constraints for machineuuid --- modules-available/statistics/api.inc.php | 29 +++++++++--------- modules-available/statistics/inc/parser.inc.php | 2 +- modules-available/statistics/install.inc.php | 39 ++++++++++++------------- 3 files changed, 32 insertions(+), 38 deletions(-) (limited to 'modules-available/statistics') diff --git a/modules-available/statistics/api.inc.php b/modules-available/statistics/api.inc.php index d4b8f346..19ae3cb6 100644 --- a/modules-available/statistics/api.inc.php +++ b/modules-available/statistics/api.inc.php @@ -14,23 +14,20 @@ if (substr($ip, 0, 7) === '::ffff:') $ip = substr($ip, 7); 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) === '000000000000001-') { - // 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; + $macaddr = Request::post('macaddr', false, 'string'); + if ($macaddr !== false) { + $macaddr = strtolower(str_replace(':', '-', $macaddr)); + if (strlen($macaddr) !== 17 || $macaddr{2} !== '-') { + $macaddr = false; } } + if ($macaddr !== false && $uuid{8} !== '-' && substr($uuid, 0, 16) === '000000000000001-') { + $uuid = 'baad1d00-9491-4716-b98b-' . str_replace('-', '', $macaddr); + } + if (strlen($uuid) !== 36 || !preg_match('/^[0-9A-F]{8}-[0-9A-F]{4}-[0-9A-F]{4}-[0-9A-F]{4}-[0-9A-F]{12}$/i', $uuid)) { + die("Invalid UUID.\n"); + } + $uuid = strtoupper($uuid); // External mode of operation? $mode = Request::post('mode', false, 'string'); $NOW = time(); @@ -45,7 +42,7 @@ if ($type{0} === '~') { if ($mode === false && $type === '~poweron') { // Poweron & hw stats $uptime = Request::post('uptime', 0, 'integer'); - if (strlen($macaddr) > 17) die("Invalid MAC.\n"); + if ($macaddr === false) die("No/Invalid MAC address.\n"); if ($uptime < 0 || $uptime > 4000000) die("Implausible uptime.\n"); $realcores = Request::post('realcores', 0, 'integer'); if ($realcores < 0 || $realcores > 512) $realcores = 0; diff --git a/modules-available/statistics/inc/parser.inc.php b/modules-available/statistics/inc/parser.inc.php index 0d39079d..8ac3fcf3 100644 --- a/modules-available/statistics/inc/parser.inc.php +++ b/modules-available/statistics/inc/parser.inc.php @@ -90,7 +90,7 @@ class Parser { } } if (empty($row['ramslotcount'])) { - $row['ramslotcount'] = count($row['ramslot']); + $row['ramslotcount'] = isset($row['ramslot']) ? count($row['ramslot']) : 0; } } diff --git a/modules-available/statistics/install.inc.php b/modules-available/statistics/install.inc.php index 84e038a4..a9e8fb7b 100644 --- a/modules-available/statistics/install.inc.php +++ b/modules-available/statistics/install.inc.php @@ -196,32 +196,15 @@ if ($addTrigger) { if (Module::isAvailable('locations')) { if (tableExists('subnet')) { AutoLocation::rebuildAll(); - } else { - finalResponse(UPDATE_RETRY, 'Locations module not installed yet, retry later'); } } $res[] = UPDATE_DONE; } -if ($machineHwCreate === UPDATE_DONE) { - $ret = Database::exec('ALTER TABLE `machine_x_hw` - ADD CONSTRAINT `machine_x_hw_ibfk_1` FOREIGN KEY (`hwid`) REFERENCES `statistic_hw` (`hwid`) ON DELETE CASCADE, - ADD CONSTRAINT `machine_x_hw_ibfk_2` FOREIGN KEY (`machineuuid`) REFERENCES `machine` (`machineuuid`) ON DELETE CASCADE'); - if ($ret === false) { - finalResponse(UPDATE_FAILED, 'Adding constraints to machine_x_hw failed: ' . Database::lastError()); - } - $ret = Database::exec('ALTER TABLE `machine_x_hw_prop` - ADD CONSTRAINT `machine_x_hw_prop_ibfk_1` FOREIGN KEY (`machinehwid`) REFERENCES `machine_x_hw` (`machinehwid`) ON DELETE CASCADE'); - if ($ret === false) { - finalResponse(UPDATE_FAILED, 'Adding constraint to machine_x_hw_prop failed: ' . Database::lastError()); - } - $ret = Database::exec('ALTER TABLE `statistic_hw_prop` - ADD CONSTRAINT `statistic_hw_prop_ibfk_1` FOREIGN KEY (`hwid`) REFERENCES `statistic_hw` (`hwid`) ON DELETE CASCADE'); - if ($ret === false) { - finalResponse(UPDATE_FAILED, 'Adding constraint to statistic_hw_prop failed: ' . Database::lastError()); - } - $res[] = UPDATE_DONE; -} +$res[] = tableAddConstraint('machine_x_hw', 'hwid', 'statistic_hw', 'hwid', 'ON DELETE CASCADE'); +$res[] = tableAddConstraint('machine_x_hw', 'machineuuid', 'machine', 'machineuuid', 'ON DELETE CASCADE ON UPDATE CASCADE'); +$res[] = tableAddConstraint('machine_x_hw_prop', 'machinehwid', 'machine_x_hw', 'machinehwid', 'ON DELETE CASCADE'); +$res[] = tableAddConstraint('statistic_hw_prop', 'hwid', 'statistic_hw', 'hwid', 'ON DELETE CASCADE'); // 2017-11-27: Add state column if (!tableHasColumn('machine', 'state')) { @@ -251,5 +234,19 @@ if (!tableHasColumn('machine', 'live_tmpsize')) { $res[] = UPDATE_DONE; } +// 2019-02-20: Convert bogus UUIDs +$res2 = Database::simpleQuery("SELECT machineuuid, macaddr FROM machine WHERE machineuuid LIKE '000000000000001-%'"); +while ($row = $res2->fetch(PDO::FETCH_ASSOC)) { + $new = strtoupper('baad1d00-9491-4716-b98b-' . preg_replace('/[^0-9a-f]/i', '', $row['macaddr'])); + error_log('Replacing ' . $row['machineuuid'] . ' with ' . $new); + if (strlen($new) === 36) { + if (Database::exec("UPDATE machine SET machineuuid = :new WHERE machineuuid = :old", + ['old' => $row['machineuuid'], 'new' => $new]) === false) { + error_log('Result: ' . Database::lastError()); + Database::exec("DELETE FROM machine WHERE machineuuid = :old", ['old' => $row['machineuuid']]); + } + } +} + // Create response responseFromArray($res); -- cgit v1.2.3-55-g7522