<?php
class HardwareInfo
{
// Never change these!
const RAM_MODULE = 'RAM';
const MAINBOARD = 'MAINBOARD';
const DMI_SYSTEM = 'DMI_SYSTEM';
const POWER_SUPPLY = 'POWER_SUPPLY';
const SYSTEM_SLOT = 'SYSTEM_SLOT';
const PCI_DEVICE = 'PCI_DEVICE';
const HDD = 'HDD';
const CPU = 'CPU';
/**
* Get a KCL modification string for the given machine, enabling GVT, PCI passthrough etc.
* You can provide a UUID and/or MAC, or nothing. If nothing is provided,
* the "uuid" and "mac" GET parameters will be used. If both are provided,
* the resulting machine that has the greatest "lastseen" value will be used.
* @param ?string $uuid UUID of machine
* @param ?string $mac MAC of machine
* @return string
*/
public static function getKclModifications($uuid = null, $mac = null): string
{
if ($uuid === null && $mac === null) {
$uuid = Request::get('uuid', '', 'string');
$mac = Request::get('mac', '', 'string');
$mac = str_replace(':', '-', $mac);
}
$res = Database::simpleQuery("SELECT machineuuid, lastseen FROM machine
WHERE machineuuid = :uuid OR macaddr = :mac", ['uuid' => $uuid, 'mac' => $mac]);
$best = null;
foreach ($res as $row) {
if ($best === null || $best['lastseen'] < $row['lastseen']) {
$best = $row;
}
}
if ($best === null)
return '';
$hw = new HardwareQuery(self::PCI_DEVICE, $best['machineuuid'], true);
// TODO: Get list of enabled pass through groups for this client's location
$hw->addWhere(true, '@PASSTHROUGH', 'IN', ['GPU', 'GVT']);
$hw->addGlobalColumn('vendor');
$hw->addGlobalColumn('device');
$hw->addLocalColumn('slot');
$res = $hw->query();
$passthrough = [];
$slots = [];
$gvt = false;
foreach ($res as $row) {
if ($row['@PASSTHROUGH'] === 'GVT') {
$gvt = true;
} else {
$passthrough[$row['vendor'] . ':' . $row['device']] = 1;
error_log('Passthouorgh: ' . $row['vendor'] . ':' . $row['device']);
$slots[preg_replace('/\.[0-9]+$/', '', $row['slot'])] = 1;
}
}
$kcl = '';
if ($gvt || !empty($passthrough)) {
$kcl = '-iommu -intel_iommu iommu=pt intel_iommu=on'; // TODO AMD
}
if (!empty($passthrough)) {
foreach (array_keys($slots) as $slot) {
error_log('Querying slot ' . $slot);
$hw = new HardwareQuery(self::PCI_DEVICE, $best['machineuuid'], true);
$hw->addWhere(false, 'slot', 'LIKE', $slot . '.%');
$hw->addGlobalColumn('vendor');
$hw->addGlobalColumn('device');
foreach ($hw->query() as $row) {
$passthrough[$row['vendor'] . ':' . $row['device']] = 1;
error_log('Extra PT: ' . $row['vendor'] . ':' . $row['device']);
}
}
$kcl .= ' vfio-pci.ids=' . implode(',', array_keys($passthrough));
}
if ($gvt) {
$kcl .= ' i915.enable_gvt=1';
}
return $kcl;
}
}
