Implementation working

Author: Carterpersall

src/windows/mod.rs

@@ -10,9 +10,11 @@ use windows::{
     core::PSTR, Win32::System::Power::GetSystemPowerStatus,
     Win32::System::Power::SYSTEM_POWER_STATUS,
     Win32::System::SystemInformation::GetComputerNameExA,
+    Win32::System::SystemInformation::GetLogicalProcessorInformation,
     Win32::System::SystemInformation::GetTickCount64,
     Win32::System::SystemInformation::GlobalMemoryStatusEx,
     Win32::System::SystemInformation::MEMORYSTATUSEX,
+    Win32::System::SystemInformation::RelationProcessorCore,
     Win32::System::WindowsProgramming::GetUserNameA,
 };
 
@@ -295,11 +297,76 @@ impl GeneralReadout for WindowsGeneralReadout {
     }
 
     fn cpu_physical_cores(&self) -> Result<usize, ReadoutError> {
-        Err(ReadoutError::NotImplemented)
+        // Source: https://github.com/seanmonstar/num_cpus/blob/master/src/lib.rs#L129
+        #[allow(non_camel_case_types, dead_code)]
+        struct SYSTEM_LOGICAL_PROCESSOR_INFORMATION {
+            mask: usize,
+            relationship: u32,
+            _unused: [u64; 2]
+        }
+
+        // The required size of the buffer, in bytes
+        let mut needed_size = 0;
+
+        // Get the required size of the buffer
+        unsafe {
+            GetLogicalProcessorInformation(std::ptr::null_mut(), &mut needed_size);
+        }
+
+        let struct_size = std::mem::size_of::<SYSTEM_LOGICAL_PROCESSOR_INFORMATION>() as u32;
+
+        // Could be 0, or some other bogus size
+        if needed_size == 0 || needed_size < struct_size || needed_size % struct_size != 0 {
+            return Err(ReadoutError::Other(
+                "Call to \"GetLogicalProcessorInformation\" returned an invalid size.".to_string()
+            ));
+        }
+
+        let count = needed_size / struct_size;
+
+        // Allocate some memory where we will store the processor info
+        let mut buf = Vec::with_capacity(count as usize);
+
+        let result;
+
+        // Populate the buffer with processor information
+        unsafe {
+            result = GetLogicalProcessorInformation(buf.as_mut_ptr(), &mut needed_size);
+        }
+
+        // If failed for some reason
+        if result.as_bool() == false {
+            return Err(ReadoutError::Other(
+                "Call to \"GetLogicalProcessorInformation\" failed.".to_string()
+            ))?;
+        }
+
+        // Number of logical processor entries
+        let count = needed_size / struct_size;
+
+        unsafe {
+            buf.set_len(count as usize);
+        }
+
+        let phys_proc_count = buf.iter()
+            // Only interested in processor packages (physical processors.)
+            .filter(|proc_info| proc_info.Relationship == RelationProcessorCore)
+            .count();
+
+        return if phys_proc_count == 0 {
+            Err(ReadoutError::Other(
+                "No physical processor cores found.".to_string()
+            ))?
+        } else {
+            Ok(phys_proc_count)
+        }
     }
 
     fn cpu_cores(&self) -> Result<usize, ReadoutError> {
-        Err(ReadoutError::NotImplemented)
+        // Open the registry key containing the CPUs information.
+        let cpu_info = RegKey::predef(HKEY_LOCAL_MACHINE).open_subkey("HARDWARE\\DESCRIPTION\\System\\CentralProcessor")?;
+        // Count the number of subkeys, which is the number of CPU logical processors.
+        Ok(cpu_info.enum_keys().count())
     }
 
     fn uptime(&self) -> Result<usize, ReadoutError> {