XFESystem

"""Small verifier controls; fixtures are never labeled as native GPU evidence."""
import unittest
from GpuDiscoveryEvidence import validate_discovery


BEGIN = 'XFE_GPU_DISCOVERY_BEGIN|Version|1|ReadOnly|1|BarAccess|0|GpuAcceleration|0|Cuda|0'
FIRMWARE = 'XFE_GPU_FIRMWARE_HANDOFF|Valid|1|Count|0|ResourceOnly|1|Cuda|0|GpuRasterization|0'
PLATFORM = 'XFE_GPU_DISCOVERY_PLATFORM|McfgStatus|1|McfgEntries|1|Segment|0|RecordLimit|8'
END = 'XFE_GPU_DISCOVERY_END|State|2|Devices|0|MatchedDevices|0|Omitted|0|ConfigReads|8202|EcamPageMaps|0|UnmapFailures|0|BarMaps|0|BarReads|0|ConfigWrites|0|GpuAcceleration|0|Cuda|0'
ABSENT = '\n'.join((FIRMWARE, BEGIN, PLATFORM, END, 'XFE_NATIVE_SELFTEST_OK independent_cpu_memory_interrupt_framebuffer'))


class GpuDiscoveryEvidenceTests(unittest.TestCase):
    def test_no_device_is_valid_probe_but_never_gpu_success(self):
        result = validate_discovery(ABSENT, True, True)
        self.assertTrue(result['Passed']); self.assertTrue(result['NativeProbeObserved'])
        self.assertFalse(result['Exact5090CandidateObserved']); self.assertFalse(result['GpuExecutionVerified'])
        self.assertFalse(result['CudaExecutionVerified']); self.assertFalse(result['GpuRasterizationVerified'])

    def test_missing_partial_duplicate_or_failed_scan_rejected(self):
        for value in ('', BEGIN, ABSENT.replace(END, ''), ABSENT + '\n' + BEGIN,
                      ABSENT.replace('State|2', 'State|3'), ABSENT.replace('ConfigReads|8202', 'ConfigReads|1'),
                      ABSENT.replace('ConfigReads|8202', 'ConfigReads|900000'), ABSENT.replace('ReadOnly|1', 'ReadOnly|0')):
            with self.subTest(value=value), self.assertRaises(AssertionError): validate_discovery(value)

    def test_any_acceleration_mutation_mapping_or_resource_leak_claim_rejected(self):
        for field in ('GpuAcceleration', 'Cuda', 'BarMaps', 'BarReads', 'ConfigWrites', 'UnmapFailures'):
            with self.subTest(field=field), self.assertRaises(AssertionError):
                validate_discovery(ABSENT.replace(field + '|0', field + '|1'))

    def test_actual_firmware_handoff_required_not_just_bools(self):
        for value in (ABSENT.replace(FIRMWARE, ''), ABSENT.replace('Valid|1', 'Valid|0'),
                      ABSENT.replace('Count|0', 'Count|1'), ABSENT + '\n' + FIRMWARE):
            with self.subTest(value=value), self.assertRaises(AssertionError): validate_discovery(value, True, True)

    def device_fixture(self):
        device = 'XFE_GPU_DEVICE|Index|0|Bus|1|Device|0|Function|0|Vendor|0x10de|DeviceId|0x2b85|Class|0x030000|Sm120Candidate|1|PcieStatus|1|ExtendedStatus|0|RebarStatus|0|Ecam|0x0'
        bars = [f'XFE_GPU_BAR|DeviceIndex|0|Index|{i}|Address|0x0|Bytes|0|Flags|0|SizeSource|0|SizeKnown|0|MappingDenied|1|Reason|DiscoveryOnly' for i in range(6)]
        summary = END.replace('Devices|0', 'Devices|1').replace('MatchedDevices|0', 'MatchedDevices|1')
        return '\n'.join((BEGIN, PLATFORM, device, *bars, summary))

    def test_exact_device_is_discovery_only_and_absent_profile_rejects_it(self):
        serial = self.device_fixture(); result = validate_discovery(serial)
        self.assertTrue(result['Exact5090CandidateObserved']); self.assertFalse(result['CudaExecutionVerified'])
        with self.assertRaises(AssertionError): validate_discovery(serial, True)

    def test_missing_bar_false_candidate_guessed_size_and_bad_geometry_rejected(self):
        serial = self.device_fixture()
        for value in (serial.replace('Index|5|Address', 'Index|4|Address'),
                      serial.replace('DeviceId|0x2b85', 'DeviceId|0x2b87'),
                      serial.replace('PcieStatus|1', 'PcieStatus|0'),
                      serial.replace('Bytes|0', 'Bytes|16777216'),
                      serial.replace('MappingDenied|1', 'MappingDenied|0'),
                      serial.replace('Address|0x0|Bytes|0|Flags|0|SizeSource|0|SizeKnown|0',
                                     'Address|0x90000001|Bytes|16777216|Flags|129|SizeSource|2|SizeKnown|1')):
            with self.subTest(value=value), self.assertRaises(AssertionError): validate_discovery(value)


if __name__ == '__main__':
    unittest.main()