Merge pull request #47 from horsenit/master

Different method for retrieving process path.
juv · Jul 30, 2017 · a527f0d · a527f0d
2 parents 2c25f7e + 269061a
commit a527f0d
Showing 1 changed file with 8 additions and 13 deletions.
diff --git a/vibrance.GUI/common/ProcessExplorer.cs b/vibrance.GUI/common/ProcessExplorer.cs
@@ -7,6 +7,7 @@
 using System.IO;
 using System.Linq;
 using System.Management;
+using System.Runtime.InteropServices;
 using System.Text;
 using System.Windows.Forms;
 using vibrance.GUI.AMD;
@@ -61,6 +62,8 @@ private void GetAllProcesses()
             }
         }
 
+        [DllImport("psapi.dll")]
+        static extern uint GetModuleFileNameEx(IntPtr hProcess, IntPtr hModule, [Out] StringBuilder lpBaseName, [In] [MarshalAs(UnmanagedType.U4)] int nSize);
         /// <summary>
         /// Safely gets the executable path from the specified process.
         /// Process.MainModule.FileName crashes when called on a x64 process because vibranceGUI is running as x86 process.
@@ -69,19 +72,11 @@ private void GetAllProcesses()
         /// <returns>the fully qualified executable path</returns>
         private string GetPathFromProcessId(Process process)
         {
-            var query = "SELECT ProcessId, ExecutablePath, CommandLine FROM Win32_Process WHERE ProcessId = " + process.Id;
-
-            using (var searcher = new ManagementObjectSearcher(query))
-                foreach (ManagementObject item in searcher.Get())
-                {
-                    object id = item["ProcessID"];
-                    object path = item["ExecutablePath"];
-
-                    if (path != null && id.ToString() == process.Id.ToString())
-                    {
-                        return path.ToString();
-                    }
-                }
+            var sb = new StringBuilder(1024);
+            if (GetModuleFileNameEx(process.Handle, IntPtr.Zero, sb, sb.Capacity) > 0)
+            {
+                return sb.ToString();
+            }
             return string.Empty;
         }