path: root/modules-available/passthrough/page.inc.php

<?php

class Page_Passthrough extends Page
{

	protected function doPreprocess()
	{
		User::load();
		User::assertPermission('view');
		$action = Request::post('action');
		if ($action === 'save-hwlist') {
			$this->saveHwList();
		} elseif ($action === 'save-location') {
			$this->saveLocation();
		}
		if (Request::isPost()) {
			Util::redirect('?do=passthrough');
		}
	}

	private function saveHwList()
	{
		User::assertPermission('edit.group');
		$newgroups = Request::post('newgroup', [], 'array');
		foreach ($newgroups as $id => $title) {
			$id = strtoupper(preg_replace('/[^a-z0-9_\-]/i', '', $id));
			if (empty($id))
				continue;
			Database::exec("INSERT IGNORE INTO passthrough_group (groupid, title)
				VALUES (:group, :title)",
				['group' => $id, 'title' => $title]);
		}
		$groups = Request::post('ptgroup', Request::REQUIRED, 'array');
		$insert = [];
		$delete = [];
		foreach ($groups as $hwid => $group) {
			if (empty($group)) {
				$delete[] = $hwid;
			} else {
				$insert[] = [$hwid, '@PASSTHROUGH', $group];
			}
		}
		if (!empty($insert)) {
			Database::exec("INSERT INTO statistic_hw_prop (hwid, prop, `value`) VALUES :list
				ON DUPLICATE KEY UPDATE `value` = VALUES(`value`)",
				['list' => $insert]);
		}
		if (!empty($delete)) {
			Database::exec("DELETE FROM statistic_hw_prop WHERE hwid IN (:list) AND prop = '@PASSTHROUGH'", ['list' => $delete]);
		}
		Message::addSuccess('list-updated');
		Util::redirect('?do=passthrough&show=hwlist');
	}

	private function saveLocation()
	{
		$locationId = Request::post('locationid', Request::REQUIRED, 'int');
		User::assertPermission('edit.location', $locationId);
		$list = [];
		$groups = [];
		foreach (Request::post('enabled', [], 'array') as $groupId) {
			$groupId = (string)$groupId;
			$list[] = [$groupId, $locationId];
			$groups[] = $groupId;
		}
		if (!empty($list)) {
			Database::exec("INSERT IGNORE INTO passthrough_group_x_location (groupid, locationid)
					VALUES :list", ['list' => $list]);
			Database::exec("DELETE FROM passthrough_group_x_location
					WHERE locationid = :lid AND groupid NOT IN (:groups)", ['lid' => $locationId, 'groups' => $groups]);
		} else {
			Database::exec("DELETE FROM passthrough_group_x_location
					WHERE locationid = :lid", ['lid' => $locationId]);
		}
		Message::addSuccess('location-updated', Location::getName($locationId));
		Util::redirect('?do=passthrough&show=assignlocation&locationid=' . $locationId);
	}

	/*
	 *
	 */

	protected function doRender()
	{
		$show = Request::get('show');
		if ($show === 'hwlist') {
			$this->showHardwareList();
		} elseif ($show === 'assignlocation') {
			$this->showLocationMapping();
		} else {
			Util::redirect('?do=passthrough&show=hwlist');
		}
	}

	/**
	 * Show all the hardware that is known. Start with video adapters.
	 * @return void
	 */
	private function showHardwareList()
	{
		$q = new HardwareQuery(HardwareInfo::PCI_DEVICE, null, false);
		$q->addGlobalColumn('vendor');
		$q->addGlobalColumn('device');
		$q->addGlobalColumn('rev');
		$q->addGlobalColumn('class');
		$q->addGlobalColumn('@PASSTHROUGH');
		$rows = [];
		foreach ($q->query('`shw`.`hwid`') as $row) {
			$row['ptlist'] = Passthrough::getGroupDropdown($row);
			$rows[] = $row;
		}
		// Sort Graphics Cards first, rest by class, vendor, device
		usort($rows, function ($row1, $row2) {
			$a = $row1['class'];
			$b = $row2['class'];
			if ($a === $b)
				return hexdec($row1['vendor'].$row1['device']) - hexdec($row2['vendor'] . $row2['device']);
			if ($a === '0300')
				return -1;
			if ($b === '0300')
				return 1;
			return hexdec($a) - hexdec($b);
		});
		$finalRows = [];
		$missing = [];
		$lastClass = '';
		foreach ($rows as $row) {
			if ($row['class'] !== $lastClass) {
				// Add class row header
				$lastClass = $row['class'];
				$finalRows[$lastClass] = [
					'collapse' => $row['class'] !== '0300',
					'class' => $row['class'],
					'class_name' => PciId::getPciId(PciId::DEVCLASS, $row['class'], true) ?: 'Unknown',
					'devlist' => [],
				];
			}
			$row['vendor_name'] = PciId::getPciId(PciId::VENDOR, $row['vendor'] ?? '');
			$row['device_name'] = PciId::getPciId(PciId::DEVICE, $row['vendor'] . ':' . $row['device']);
			$finalRows[$lastClass]['devlist'][] = $row;
			// Build up query
			if ($row['vendor_name'] === false) {
				$missing[$row['vendor']] = true;
			}
			if ($row['device_name'] === false) {
				$missing[$row['vendor'] . ':' . $row['device']] = true;
			}
		}
		Render::addTemplate('hardware-list', ['classlist' => array_values($finalRows)]);
		if (!empty($missing)) {
			Render::addTemplate('js-pciquery',
				['missing_ids' => json_encode(array_keys($missing))], 'statistics');
		}
	}

	/**
	 * Show mapping between specific location and passthrough groups.
	 * @return void
	 */
	private function showLocationMapping()
	{
		$locationId = Request::get('locationid', Request::REQUIRED, 'int');
		$locationIds = Location::getLocationRootChain($locationId);
		$res = Database::queryAll("SELECT g.groupid, g.title, GROUP_CONCAT(gxl.locationid) AS lids FROM passthrough_group g
				LEFT JOIN passthrough_group_x_location gxl ON (g.groupid = gxl.groupid AND gxl.locationid IN (:lids))
				GROUP BY groupid, title 
				ORDER BY lids ASC",
			['lids' => $locationIds]);
		foreach ($res as &$item) {
			if ($item['lids'] === null)
				continue;
			$item['checked'] = 'checked';
			$list = explode(',', $item['lids']);
			if (!in_array($locationId, $list)) {
				$item['disabled'] = true;
				$item['parent_location'] = Location::getName($list[0]);
			}
		}
		Render::addTemplate('location-assign', [
			'list' => array_reverse($res),
			'locationid' => $locationId,
			'locationname' => Location::getName($locationId),
		]);
	}

	/*
	 *
	 */

	protected function doAjax()
	{
		//
	}

}