Native Client Visual Studio Add-in

visual_studio/NativeClientVSAddIn/NativeClientVSAddIn/ProcessSearcher.cs

+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+namespace NativeClientVSAddIn
+{
+  using System;
+  using System.Collections.Generic;
+  using System.Linq;
+  using System.Management;
+
+  /// <summary>
+  /// Queries the system for the list of running processes.
+  /// </summary>
+  public class ProcessSearcher
+  {
+    /// <summary>
+    /// Returns results of a process search subject to given constraints.
+    /// </summary>
+    /// <param name="constraints">
+    /// A function taking a ProcessInfo object and returning true if the
+    /// ProcessInfo object satisfies the constraints.
+    /// </param>
+    /// <returns>List of matching processes.</returns>
+    public List<ProcessInfo> GetResults(Func<ProcessInfo, bool> constraints)
+    {
+      return GetSystemProcesses().Where(constraints).ToList();
+    }
+
+    /// <summary>
+    /// Searches the system for all processes of a given name.
+    /// </summary>
+    /// <param name="name">Name to search for.</param>
+    /// <returns>List of matching processes.</returns>
+    public List<ProcessInfo> GetResultsByName(string name)
+    {
+      return GetResults(p => name.Equals(p.Name, StringComparison.OrdinalIgnoreCase));
+    }
+
+    /// <summary>
+    /// Searches the system for all processes of a given process ID.
+    /// </summary>
+    /// <param name="procID">ID to search for.</param>
+    /// <returns>List of matching processes.</returns>
+    public List<ProcessInfo> GetResultsByID(uint procID)
+    {
+      return GetResults(p => procID == p.ID);
+    }
+
+    /// <summary>
+    /// Queries the system for the full list of processes.
+    /// </summary>
+    /// <returns>List of processes on the system.</returns>
+    protected virtual List<ProcessInfo> GetSystemProcesses()
+    {
+      var processList = new List<ProcessInfo>();
+      string query = "select * from Win32_Process";
+      using (ManagementObjectSearcher searcher = new ManagementObjectSearcher(query))
+      {
+        using (ManagementObjectCollection results = searcher.Get())
+        {
+          foreach (ManagementObject process in results)
+          {
+            processList.Add((ProcessInfo)process);
+          }
+        }
+      }
+
+      return processList;
+    }
+  }
+}