From 4d51d90809da34022991e4a8c17caeb28c31bde0 Mon Sep 17 00:00:00 2001
From: Simon Rettberg
Date: Wed, 3 Apr 2019 18:10:28 +0200
Subject: [statistics/locations] Add locationid constraints; update on delete

We didn't update subnetlocationid when deleting a location,
leading to machines pointing to invalid locations.
---
 .../locations/inc/autolocation.inc.php             | 27 ++++++++++++++++------
 1 file changed, 20 insertions(+), 7 deletions(-)

(limited to 'modules-available/locations/inc')

diff --git a/modules-available/locations/inc/autolocation.inc.php b/modules-available/locations/inc/autolocation.inc.php
index cccfe9b1..61507ebf 100644
--- a/modules-available/locations/inc/autolocation.inc.php
+++ b/modules-available/locations/inc/autolocation.inc.php
@@ -3,11 +3,23 @@
 class AutoLocation
 {
 
-	public static function rebuildAll()
+	/**
+	 * Rebuild the assigned subnetlocationid for clients that currently map to
+	 * the given locationids (if given), or all clients, if locationid is false
+	 *
+	 * @param int[]|false $locations Locations to rebuild, or false for everything
+	 */
+	public static function rebuildAll($locations = false)
 	{
 		if (Module::get('statistics') === false)
 			return; // Nothing to do
-		$res = Database::simpleQuery("SELECT machineuuid, clientip FROM machine");
+		if ($locations === false) {
+			// All
+			$res = Database::simpleQuery("SELECT machineuuid, clientip FROM machine");
+		} else {
+			$res = Database::simpleQuery("SELECT machineuuid, clientip FROM machine
+					WHERE fixedlocationid IN(:lid) OR subnetlocationid IN(:lid)", ['lid' => $locations]);
+		}
 		$updates = array();
 		$nulls = array();
 		while ($row = $res->fetch(PDO::FETCH_ASSOC)) {
@@ -22,13 +34,14 @@ class AutoLocation
 			}
 		}
 		if (!empty($nulls)) {
-			$qs = '?' . str_repeat(',?', count($nulls) - 1);
-			Database::exec("UPDATE machine SET subnetlocationid = NULL WHERE machineuuid IN ($qs)", $nulls);
+			Database::exec("UPDATE machine SET subnetlocationid = NULL WHERE machineuuid IN (:nulls)",
+				['nulls' => $nulls]);
 		}
 		foreach ($updates as $lid => $machines) {
-			$qs = '?' . str_repeat(',?', count($machines) - 1);
-			$lid = (int)$lid;
-			Database::exec("UPDATE machine SET subnetlocationid = $lid WHERE machineuuid IN ($qs)", $machines);
+			if (empty($machines))
+				continue;
+			Database::exec("UPDATE machine SET subnetlocationid = :lid WHERE machineuuid IN (:machines)",
+				['lid' => $lid, 'machines' => $machines]);
 		}
 	}
 
-- 
cgit v1.2.3-55-g7522