import unittest from cpuinfo import * import helpers class MockASM(ASM): is_first = False def __init__(self, restype=None, argtypes=(), machine_code=[]): super(MockASM, self).__init__(restype, argtypes, machine_code) def compile(self): self.func = self.run def run(self): machine_code = tuple(self.machine_code) # get_max_extension_support if machine_code == \ (b"\xB8\x00\x00\x00\x80" # mov ax,0x80000000 b"\x0f\xa2" # cpuid b"\xC3",): # ret return 0x8000001f # get_cache if machine_code == \ (b"\xB8\x06\x00\x00\x80" # mov ax,0x80000006 b"\x0f\xa2" # cpuid b"\x89\xC8" # mov ax,cx b"\xC3",): # ret)) return 0x2006140 # get_info if machine_code == \ (b"\xB8\x01\x00\x00\x00", # mov eax,0x1" b"\x0f\xa2" # cpuid b"\xC3",): # ret return 0x800f82 # get_processor_brand if machine_code == \ (b"\xB8\x02\x00\x00\x80", # mov ax,0x80000002 b"\x0f\xa2" # cpuid b"\x89\xC0" # mov ax,ax b"\xC3",): # ret return 0x20444d41 elif machine_code == \ (b"\xB8\x02\x00\x00\x80", # mov ax,0x80000002 b"\x0f\xa2" # cpuid b"\x89\xD8" # mov ax,bx b"\xC3",): # ret return 0x657a7952 elif machine_code == \ (b"\xB8\x02\x00\x00\x80", # mov ax,0x80000002 b"\x0f\xa2" # cpuid b"\x89\xC8" # mov ax,cx b"\xC3",): # ret return 0x2037206e elif machine_code == \ (b"\xB8\x02\x00\x00\x80", # mov ax,0x80000002 b"\x0f\xa2" # cpuid b"\x89\xD0" # mov ax,dx b"\xC3",): # ret return 0x30303732 elif machine_code == \ (b"\xB8\x03\x00\x00\x80", # mov ax,0x80000003 b"\x0f\xa2" # cpuid b"\x89\xC0" # mov ax,ax b"\xC3",): # ret return 0x69452058 elif machine_code == \ (b"\xB8\x03\x00\x00\x80", # mov ax,0x80000003 b"\x0f\xa2" # cpuid b"\x89\xD8" # mov ax,bx b"\xC3",): # ret return 0x2d746867 elif machine_code == \ (b"\xB8\x03\x00\x00\x80", # mov ax,0x80000003 b"\x0f\xa2" # cpuid b"\x89\xC8" # mov ax,cx b"\xC3",): # ret return 0x65726f43 elif machine_code == \ (b"\xB8\x03\x00\x00\x80", # mov ax,0x80000003 b"\x0f\xa2" # cpuid b"\x89\xD0" # mov ax,dx b"\xC3",): # ret return 0x6f725020 elif machine_code == \ (b"\xB8\x04\x00\x00\x80", # mov ax,0x80000004 b"\x0f\xa2" # cpuid b"\x89\xC0" # mov ax,ax b"\xC3",): # ret return 0x73736563 elif machine_code == \ (b"\xB8\x04\x00\x00\x80", # mov ax,0x80000004 b"\x0f\xa2" # cpuid b"\x89\xD8" # mov ax,bx b"\xC3",): # ret return 0x2020726f elif machine_code == \ (b"\xB8\x04\x00\x00\x80", # mov ax,0x80000004 b"\x0f\xa2" # cpuid b"\x89\xC8" # mov ax,cx b"\xC3",): # ret return 0x20202020 elif machine_code == \ (b"\xB8\x04\x00\x00\x80", # mov ax,0x80000004 b"\x0f\xa2" # cpuid b"\x89\xD0" # mov ax,dx b"\xC3",): # ret return 0x202020 # get_vendor_id if machine_code == \ (b"\x31\xC0", # xor eax,eax b"\x0F\xA2" # cpuid b"\x89\xD8" # mov ax,bx b"\xC3",): # ret return 0x68747541 elif machine_code == \ (b"\x31\xC0", # xor eax,eax b"\x0f\xa2" # cpuid b"\x89\xC8" # mov ax,cx b"\xC3",): # ret return 0x444d4163 elif machine_code == \ (b"\x31\xC0", # xor eax,eax b"\x0f\xa2" # cpuid b"\x89\xD0" # mov ax,dx b"\xC3",): # ret return 0x69746e65 # get_flags if machine_code == \ (b"\xB8\x01\x00\x00\x00", # mov eax,0x1" b"\x0f\xa2" # cpuid b"\x89\xD0" # mov ax,dx b"\xC3",): # ret return 0x178bfbff elif machine_code == \ (b"\xB8\x01\x00\x00\x00", # mov eax,0x1" b"\x0f\xa2" # cpuid b"\x89\xC8" # mov ax,cx b"\xC3",): # ret return 0x7ed8320b elif machine_code == \ (b"\x31\xC9", # xor ecx,ecx b"\xB8\x07\x00\x00\x00" # mov eax,7 b"\x0f\xa2" # cpuid b"\x89\xD8" # mov ax,bx b"\xC3",): # ret return 0x209c01a9 elif machine_code == \ (b"\x31\xC9", # xor ecx,ecx b"\xB8\x07\x00\x00\x00" # mov eax,7 b"\x0f\xa2" # cpuid b"\x89\xC8" # mov ax,cx b"\xC3",): # ret return 0x0 elif machine_code == \ (b"\xB8\x01\x00\x00\x80" # mov ax,0x80000001 b"\x0f\xa2" # cpuid b"\x89\xD8" # mov ax,bx b"\xC3",): # ret return 0x20000000 elif machine_code == \ (b"\xB8\x01\x00\x00\x80" # mov ax,0x80000001 b"\x0f\xa2" # cpuid b"\x89\xC8" # mov ax,cx b"\xC3",): # ret return 0x35c233ff # get_ticks # 32 bit if machine_code == \ (b"\x55", # push bp b"\x89\xE5", # mov bp,sp b"\x31\xC0", # xor ax,ax b"\x0F\xA2", # cpuid b"\x0F\x31", # rdtsc b"\x8B\x5D\x08", # mov bx,[di+0x8] b"\x8B\x4D\x0C", # mov cx,[di+0xc] b"\x89\x13", # mov [bp+di],dx b"\x89\x01", # mov [bx+di],ax b"\x5D", # pop bp b"\xC3",): # ret raise Exception("FIXME: Add ticks for 32bit get_ticks") # 64 bit elif machine_code == \ (b"\x48", # dec ax b"\x31\xC0", # xor ax,ax b"\x0F\xA2", # cpuid b"\x0F\x31", # rdtsc b"\x48", # dec ax b"\xC1\xE2\x20", # shl dx,byte 0x20 b"\x48", # dec ax b"\x09\xD0", # or ax,dx b"\xC3",): # ret MockASM.is_first = not MockASM.is_first if MockASM.is_first: return 19233706151817 else: return 19237434253761 raise Exception("Unexpected machine code") def free(self): self.func = None class MockDataSource: bits = '64bit' cpu_count = 1 is_windows = platform.system().lower() == 'windows' arch_string_raw = 'INVALID' uname_string_raw = 'INVALID' can_cpuid = True class TestCPUID(unittest.TestCase): def setUp(self): helpers.backup_data_source(cpuinfo) helpers.monkey_patch_data_source(cpuinfo, MockDataSource) helpers.backup_asm(cpuinfo) helpers.monkey_patch_asm(cpuinfo, MockASM) def tearDown(self): helpers.restore_data_source(cpuinfo) helpers.restore_asm(cpuinfo) # Make sure this returns {} on an invalid arch def test_return_empty(self): self.assertEqual({}, cpuinfo._get_cpu_info_from_cpuid()) def test_normal(self): cpuid = CPUID() self.assertIsNotNone(cpuid) self.assertFalse(cpuid.is_selinux_enforcing) max_extension_support = cpuid.get_max_extension_support() self.assertEqual(0x8000001f, max_extension_support) cache_info = cpuid.get_cache(max_extension_support) self.assertEqual({'size_b': 64 * 1024, 'line_size_b': 512, 'associativity': 6}, cache_info) info = cpuid.get_info() self.assertEqual({'stepping': 2, 'model': 8, 'family': 23, 'processor_type': 0}, info) processor_brand = cpuid.get_processor_brand(max_extension_support) self.assertEqual("AMD Ryzen 7 2700X Eight-Core Processor", processor_brand) hz_actual = cpuid.get_raw_hz() self.assertEqual(3728101944, hz_actual) vendor_id = cpuid.get_vendor_id() self.assertEqual('AuthenticAMD', vendor_id) flags = cpuid.get_flags(max_extension_support) self.assertEqual( ['3dnowprefetch', 'abm', 'adx', 'aes', 'apic', 'avx', 'avx2', 'bmi1', 'bmi2', 'clflush', 'clflushopt', 'cmov', 'cmp_legacy', 'cr8_legacy', 'cx16', 'cx8', 'dbx', 'de', 'extapic', 'f16c', 'fma', 'fpu', 'fxsr', 'ht', 'lahf_lm', 'lm', 'mca', 'mce', 'misalignsse', 'mmx', 'monitor', 'movbe', 'msr', 'mtrr', 'osvw', 'osxsave', 'pae', 'pat', 'pci_l2i', 'pclmulqdq', 'perfctr_core', 'perfctr_nb', 'pge', 'pni', 'popcnt', 'pse', 'pse36', 'rdrnd', 'rdseed', 'sep', 'sha', 'skinit', 'smap', 'smep', 'sse', 'sse2', 'sse4_1', 'sse4_2', 'sse4a', 'ssse3', 'svm', 'tce', 'topoext', 'tsc', 'vme', 'wdt', 'xsave' ], flags)