Files
darling-lol/tests/test_system_profiler.py
T
34c3e4ae9f Report hardware and software overviews from IOKit and sysctl in system_profiler
system_profiler now answers SPHardwareDataType and SPSoftwareDataType (and
lists them in -listDataTypes). Model, serial number, platform UUID and
provisioning UDID come from IOKit's IOPlatformExpertDevice via ioreg; CPU,
cache and memory from sysctl; the OS version from sw_vers and uname. A
field the guest does not provide is left out.

#61 instead read a separate darling-smbios.conf written by a new
generator (gen-darling-smbios.py), with hardcoded fallbacks (an Intel Core
i9, macOS 14.7, SIP enabled). iokitd reads darling-platform.plist, so
system_profiler and IOKit could report different serials and models on the
same prefix, and the generator overwrote identities imported with
configure-darling-platform.py. Reading the guest's own APIs keeps a single
source of truth: configure-darling-platform.py remains the only way to set
or provision an identity, and it is still never run by install or launch.

tests/test_system_profiler.py runs the script under sh with stub guest
commands and runs in CI.

Split from #61.

Co-authored-by: DevWebeloper <[email protected]>
Co-Authored-By: Claude Opus 5.5 <[email protected]>
2026-10-18 17:17:05 +02:00

98 lines
4.4 KiB
Python

"""system_profiler hardware/software overviews, with stub guest commands; never starts Darling."""
import os
from pathlib import Path
import plistlib
import subprocess
import sys
import tempfile
import unittest
ROOT = Path(__file__).resolve().parents[1]
SCRIPT = ROOT / 'tools/darling-system_profiler.sh'
IOREG = '''+-o MacBookPro16,1 <class IOPlatformExpertDevice, id 0x100000110, registered, matched, active, busy 0 (0 ms), retain 30>
{
"IOPlatformSerialNumber" = "USER-SUPPLIED"
"model" = <"MacBookPro16,1">
"manufacturer" = <"Apple Inc.">
"IOPlatformUUID" = "8E1F2A3B-4C5D-4E6F-8A7B-9C0D1E2F3A4B"
"provisioning_UDID" = "8E1F2A3B-4C5D-4E6F-8A7B-9C0D1E2F3A4B"
"ROM" = <0a1b2c3d4e5f>
}
'''
SYSCTL = {
'hw.physicalcpu': '8', 'hw.packages': '1', 'hw.memsize': str(32 << 30),
'machdep.cpu.brand_string': 'AMD Ryzen 7 5800X 8-Core Processor & more',
'hw.l2cachesize': str(512 << 10), 'hw.l3cachesize': str(32 << 20), 'hw.cpufrequency': '3800000000',
}
class SystemProfilerTests(unittest.TestCase):
def setUp(self):
self.folder = tempfile.TemporaryDirectory()
self.bin = Path(self.folder.name)
self.stub('sysctl', 'import sys\nvalues = %r\nname = sys.argv[-1]\n'
'if name not in values: sys.exit(1)\nprint(values[name])' % SYSCTL)
self.stub('sw_vers', 'import sys\nprint({"-productName": "macOS", "-productVersion": "11.7.4", '
'"-buildVersion": "20G1120"}[sys.argv[1]])')
self.stub('uname', 'print("Darwin 20.6.0")')
self.stub('hostname', 'print("darling")')
self.stub('ioreg', 'import sys\nsys.stdout.write(%r)' % IOREG)
def tearDown(self):
self.folder.cleanup()
def stub(self, name, body):
path = self.bin / name
path.write_text('#!%s\n%s\n' % (sys.executable, body))
path.chmod(0o755)
def run_profiler(self, *args):
env = {'PATH': '%s:%s' % (self.bin, os.environ['PATH']), 'USER': 'player'}
return subprocess.run(['sh', str(SCRIPT), *args], env=env, check=True, capture_output=True, text=True).stdout
def xml_items(self, data_type):
report = plistlib.loads(self.run_profiler('-xml', data_type).encode())
self.assertEqual(report[0]['_dataType'], data_type)
return report[0]['_items'][0]
def test_hardware_reports_iokit_identity_and_sysctl_values(self):
item = self.xml_items('SPHardwareDataType')
self.assertEqual(item, {
'_name': 'hardware_overview', 'machine_name': 'MacBook Pro', 'machine_model': 'MacBookPro16,1',
'cpu_type': 'AMD Ryzen 7 5800X 8-Core Processor & more', 'current_processor_speed': '3.8 GHz',
'packages': 1, 'number_processors': 8, 'l2_cache_core': '512 KB', 'l3_cache': '32 MB',
'physical_memory': '32 GB', 'serial_number': 'USER-SUPPLIED',
'platform_UUID': '8E1F2A3B-4C5D-4E6F-8A7B-9C0D1E2F3A4B',
'provisioning_UDID': '8E1F2A3B-4C5D-4E6F-8A7B-9C0D1E2F3A4B',
})
text = self.run_profiler('SPHardwareDataType')
self.assertIn(' Serial Number (system): USER-SUPPLIED\n', text)
self.assertIn(' Model Name: MacBook Pro\n', text)
def test_missing_iokit_and_sysctl_values_are_omitted(self):
self.stub('ioreg', 'import sys\nsys.exit(1)')
self.stub('sysctl', 'import sys\nsys.exit(1)')
self.assertEqual(self.xml_items('SPHardwareDataType'), {'_name': 'hardware_overview'})
self.assertNotIn('Serial Number', self.run_profiler('SPHardwareDataType'))
def test_software_overview(self):
item = self.xml_items('SPSoftwareDataType')
self.assertEqual(item, {
'_name': 'os_overview', 'os_version': 'macOS 11.7.4 (20G1120)', 'kernel_version': 'Darwin 20.6.0',
'boot_volume': 'Macintosh HD', 'boot_mode': 'normal_boot', 'local_host_name': 'darling',
'user_name': 'player',
})
self.assertIn(' Boot Mode: Normal\n', self.run_profiler('SPSoftwareDataType'))
def test_listed_and_default_types(self):
listed = self.run_profiler('-listDataTypes')
self.assertIn('SPHardwareDataType', listed)
self.assertIn('SPSoftwareDataType', listed)
types = [entry['_dataType'] for entry in plistlib.loads(self.run_profiler('-xml').encode())]
self.assertEqual(types[:2], ['SPHardwareDataType', 'SPSoftwareDataType'])
if __name__ == '__main__':
unittest.main()