summaryrefslogtreecommitdiffstats
path: root/modules-available/remoteaccess/inc/remoteaccess.inc.php
blob: 1dbe02c3b798f43372599d75de5aeb5cd2f27ec0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
<?php

class RemoteAccess
{

	const PROP_ALLOWED_VNC_NET = 'remoteaccess.allowedvncaccess';

	const PROP_TRY_VIRT_HANDOVER = 'remoteaccess.virthandover';

	public static function getEnabledLocations($group = 0)
	{
		if ($group === 0) {
			return Database::queryColumnArray("SELECT DISTINCT rxl.locationid FROM remoteaccess_x_location rxl
				INNER JOIN remoteaccess_group g ON (g.groupid = rxl.groupid AND g.active = 1)");
		}
		return Database::queryColumnArray("SELECT DISTINCT locationid FROM remoteaccess_x_location
				WHERE groupid = :gid", ['gid' => $group]);
	}

	public static function ensureMachinesRunning()
	{
		$res = Database::simpleQuery("SELECT rg.groupid, rg.wolcount, GROUP_CONCAT(rxl.locationid) AS locs FROM remoteaccess_group rg
	INNER JOIN remoteaccess_x_location rxl USING (groupid)
	WHERE rg.active = 1
	GROUP BY groupid");

		while ($row = $res->fetch(PDO::FETCH_ASSOC)) {
			if ($row['wolcount'] <= 0)
				continue;
			// This can't really be anything but a CSV list, but better be safe
			$locs = preg_replace('/[^0-9,]/', '', $row['locs']);
			if (empty($locs))
				continue;
			$active = Database::queryFirst("SELECT Count(*) AS cnt FROM machine m
					INNER JOIN remoteaccess_machine rm USING (machineuuid)
					WHERE m.locationid IN ($locs) AND m.state = 'IDLE'");
			$wantNum = $row['wolcount'] - (isset($active['cnt']) ? $active['cnt'] : 0);
			if ($wantNum <= 0)
				continue;
			self::tryWakeMachines($locs, $wantNum);
		}
	}

	private static function tryWakeMachines($locs, $num)
	{
		$res = Database::simpleQuery("SELECT m.machineuuid, m.macaddr, m.clientip FROM machine m
				LEFT JOIN remoteaccess_machine rm USING (machineuuid)
				WHERE m.locationid IN ($locs) AND m.state IN ('OFFLINE', 'STANDBY')
				ORDER BY rm.woltime ASC");
		$NOW = time();
		while ($num > 0) {
			$list = [];
			for ($i = 0; $i < $num && $row = $res->fetch(PDO::FETCH_ASSOC); ++$i) {
				$list[] = $row;
				Database::exec("INSERT INTO remoteaccess_machine (machineuuid, password, woltime)
						VALUES (:uuid, NULL, :now)
						ON DUPLICATE KEY UPDATE woltime = VALUES(woltime)",
					['uuid' => $row['machineuuid'], 'now' => $NOW]);
			}
			if (empty($list))
				break; // No more clients in this location
			error_log('Trying to wake ' . count($list) . " in $locs");
			RebootControl::wakeMachines($list, $fails);
			$num -= count($list) - count($fails);
			if (!empty($fails)) {
				error_log(count($fails) . ' failed');
				$failIds = ArrayUtil::flattenByKey($fails, 'machineuuid');
				// Reduce time so they won't be marked as wol_in_progress
				Database::exec('UPDATE remoteaccess_machine SET woltime = :faketime WHERE machineuuid IN (:fails)',
					['faketime' => $NOW - 95, 'fails' => $failIds]);
			}
		}
		if ($num > 0) {
			error_log("Could not wake $num clients...");
		}
	}

}