Only take into account processes that have our plugin loaded

installer/src/installer-lib/process.h

 /**
  * \file process.h
  */
 
 #ifndef PROCESS_H
 #define PROCESS_H
 
 #include <vector>
 #include <set>
 #include <algorithm>
@@ skipping 62 matching lines @@
    * Copy assignment declared private and not defined.
    *
    * \par Implementation
    *   Add "= delete" for C++11.
    */
   Windows_Handle operator=( const Windows_Handle & ) ;
 
 };
 
 //-------------------------------------------------------
-// exe_name_set: case-insensitive wide-string set
+// file_name_set: case-insensitive wide-string set
 //-------------------------------------------------------
 int wcscmpi( const wchar_t * s1, const wchar_t * s2 ) ;
 
-struct exe_name
+struct file_name
 {
   /**
    * Pointer to wide-character string, which is supposed to be null-terminated.
    */
   const wchar_t * name ;
 
-  exe_name( const wchar_t * name ) : name( name ) {} ;
+  file_name( const wchar_t * name ) : name( name ) {} ;
 } ;
 
 template <>
-struct std::less< exe_name >
-  : std::binary_function< exe_name, exe_name, bool >
+struct std::less< file_name >
+  : std::binary_function< file_name, file_name, bool >
 {
-  bool operator()( const exe_name & a, const exe_name & b ) const
+  bool operator()( const file_name & a, const file_name & b ) const
   {
     return wcscmpi( a.name, b.name ) < 0 ; 
   }
 } ;
 
-struct exe_name_set
-  : public std::set< exe_name >
+struct file_name_set
+  : public std::set< file_name >
 {
-  exe_name_set( const wchar_t * exe_name_list[], size_t n_exe_names )
+  file_name_set(){};
+
+  file_name_set( const wchar_t * file_name_list[], size_t n_file_names )
   {
-    for ( unsigned int j = 0 ; j < n_exe_names ; ++ j )
+    for ( unsigned int j = 0 ; j < n_file_names ; ++ j )
     {
-      insert( exe_name( exe_name_list[ j ] ) ) ;
+      insert( file_name( file_name_list[ j ] ) ) ;
     }
   }
 } ;
 
 //-------------------------------------------------------
 //-------------------------------------------------------
 /**
- * Filter by process name. Comparison is case-insensitive.
+ * Filter by process name. Comparison is case-insensitive. With ABP module loaded
  */
-class process_by_any_exe_name_CI
-  : public std::unary_function< PROCESSENTRY32W, bool >
+class process_by_any_exe_with_any_module
+  : public std::binary_function< PROCESSENTRY32W, file_name_set, bool >
 {
-  const exe_name_set & names ;
+  const file_name_set & processNames ;
+  const file_name_set & moduleNames;
 public:
   bool operator()( const PROCESSENTRY32W & ) ;
-  process_by_any_exe_name_CI( const exe_name_set & names )
-    : names( names )
+  process_by_any_exe_with_any_module( const file_name_set & names, const file_name_set & moduleNames )
+    : processNames( names ), moduleNames( moduleNames )
   {}
 } ;
 
+
 //-------------------------------------------------------
 // Process utility functions.
 //-------------------------------------------------------
 /**
  * A promiscuous filter admits everything.
  */
 struct every_process
   : public std::unary_function< PROCESSENTRY32W, bool >
 {
   bool operator()( const PROCESSENTRY32W & ) { return true ; } ;
@@ skipping 112 matching lines @@
   Snapshot() ;
 
   /**
    * Reconstruct the current instance with a new system snapshot.
    */
   void refresh() ;
 
   /**
    * Return a pointer to the first process in the snapshot.
    */
-  PROCESSENTRY32W * begin() ;
-
-  /**
-   * The end pointer is an alias for the null pointer.
-   */
-  inline PROCESSENTRY32W * end() const { return 0 ; }
+  PROCESSENTRY32W * first() ;
 
   /**
    * Return a pointer to the next process in the snapshot.
    * begin() must have been called first.
    */
   PROCESSENTRY32W * next() ;
+} ;
+
+class ModulesSnapshot
+{
+  /**
+   * Handle to the process snapshot.
+   */
+  Windows_Handle handle;
 
   /**
-   * Type definition for pointer to underlying structure.
+   * Buffer for reading a single module entry out of the snapshot.
    */
-  typedef PROCESSENTRY32W * Pointer ;
-} ;
+  MODULEENTRY32W module;
+
+  /**
+   * Copy constructor declared private and not defined.
+   *
+   * \par Implementation
+   *   Add "= delete" for C++11.
+   */
+  ModulesSnapshot(const ModulesSnapshot&);
+
+  /**
+   * Copy assignment declared private and not defined.
+   *
+   * \par Implementation
+   *   Add "= delete" for C++11.
+   */
+  ModulesSnapshot operator=(const ModulesSnapshot&);
+
+
+public:
+  /**
+   * Default constructor takes the snapshot.
+   */
+  ModulesSnapshot(DWORD processId);
+
+  /**
+   * Return a pointer to the first process in the snapshot.
+   */
+  MODULEENTRY32W* first();
+
+  /**
+   * Return a pointer to the next process in the snapshot.
+   * begin() must have been called first.
+   */
+  MODULEENTRY32W* next();
+};
+
 
 //-------------------------------------------------------
 // initialize_process_list
 //-------------------------------------------------------
 /**
  * \tparam T The type into which a PROCESSENTRY32W struture is extracted.
  * \tparam Admittance Function type for argument 'admit'
  * \tparam Extractor Function type for argument 'extract'
  * \param admit A unary predicate function class that determines what's included
  *      A process appears in the list only if the predicate returns true. 
  *      The use of this predicate is analogous to that in std::copy_if.
  * \param convert A conversion function that takes a PROCESSENTRY32W as input argument and returns an element of type T.
  */
-template< class T, class Admittance, class Extractor >
-void initialize_process_list( std::vector< T > & v, Snapshot & snap, Admittance admit = Admittance(), Extractor extract = Extractor() )
+template<class T, class Admittance, class Extractor>
+void initialize_process_list(std::vector<T>& v, Snapshot& snap, Admittance admit = Admittance(), Extractor extract = Extractor())
 {
-  Snapshot::Pointer p = snap.begin() ;
-  while ( p != snap.end() )
+  PROCESSENTRY32W* p = snap.first();
+  while (p != NULL)
   {
-    if ( admit( * p ) )
+    if (admit(*p ))
     {
       /*
         * We don't have C++11 emplace_back, which can construct the element in place.
         * Instead, we copy the return value of the converter.
         */
-      v.push_back( extract( * p ) );
+      v.push_back(extract(*p));
     }
-    p = snap.next() ;
+    p = snap.next();
   }
-} ;
+};
 
 //-------------------------------------------------------
 // initialize_process_set
 //-------------------------------------------------------
 /**
  * \tparam T The type into which a PROCESSENTRY32W struture is extracted.
  * \tparam Admittance Function type for argument 'admit'
  * \tparam Extractor Function type for argument 'extract'
  * \param admit A unary predicate function class that determines what's included
  *      A process appears in the list only if the predicate returns true. 
  *      The use of this predicate is analogous to that in std::copy_if.
  * \param convert A conversion function that takes a PROCESSENTRY32W as input argument and returns an element of type T.
  */
-template< class T, class Admittance, class Extractor >
-void initialize_process_set( std::set< T > & set, Snapshot & snap, Admittance admit = Admittance(), Extractor extract = Extractor() )
+template<class T, class Admittance, class Extractor>
+void initialize_process_set(std::set< T > & set, Snapshot& snap, Admittance admit = Admittance(), Extractor extract = Extractor())
 {
-  Snapshot::Pointer p = snap.begin() ;
-  while ( p != snap.end() )
+  PROCESSENTRY32W* p = snap.first();
+  while (p != NULL)
   {
-    if ( admit( * p ) )
+    if (admit(*p))
     {
-      set.insert( extract( * p ) );
+      set.insert(extract(*p));
     }
-    p = snap.next() ;
+    p = snap.next();
   }
-} ;
+};
 
 //-------------------------------------------------------
 // enumerate_windows
 //-------------------------------------------------------
 
 /**
  * States of a window enumeration.
  */
 typedef enum
 {
@@ skipping 107 matching lines @@
    * Set of process identifiers matching one of the executable names.
    */
   std::set< DWORD > pid_set ;
 
   /**
    * Set of executable names by which to filter.
    *
    * The argument of the filter constructor is a set by reference.
    * Since it does not make a copy for itself, we define it as a class member to provide its allocation.
    */
-  exe_name_set exe_names ;
+  file_name_set file_names ;
+  /**
+   * Set of module (DLL) names by which to filter.
+   */
+  file_name_set module_names ;
 
-  /**
-   * Filter function object matches on any of the exe names specified in the constructor.
-   */
-  process_by_any_exe_name_CI filter ;
+  process_by_any_exe_with_any_module filter ;
 
   /**
    * Copy function object copies just the process ID.
    */
   copy_PID copy ;
 
   /** 
    * Snapshot of running processes.
    */
   Snapshot & snapshot ;
@@ skipping 30 matching lines @@
       if ( ! b )
       {
         // Assert the process that created the window is not in our pid_set
         return true ;
       }
       return f( window ) ;
     }
   } ;
 
 public:
-  Process_Closer( Snapshot & snapshot, const wchar_t * exe_name_list[], size_t n_exe_names )
-    : snapshot( snapshot ), exe_names( exe_name_list, n_exe_names ), filter( exe_names )
+  template <size_t n_file_names, size_t n_module_names>
+  Process_Closer(Snapshot & snapshot, const wchar_t* (&file_name_list)[n_file_names], const wchar_t* (&module_name_list)[n_module_names])
+    : snapshot(snapshot), file_names(file_name_list, n_file_names), module_names(module_name_list, n_module_names), filter(file_names, module_names)
   {
     update() ;
   }
+  template <size_t n_file_names>
+  Process_Closer(Snapshot & snapshot, const wchar_t * (&file_name_list)[n_file_names])
+    : snapshot(snapshot), file_names(file_name_list, n_file_names), module_names(), filter(file_names, module_names)
+  {
+    update() ;
+  }
 
   /**
    * Refresh our state to match the snapshot state.
    */
   void refresh()
   {
     pid_set.clear() ;
     update() ;
   }
 
   bool is_running() { return ! pid_set.empty() ; } ;
 
   bool contains( DWORD pid ) const { return pid_set.find( pid ) != pid_set.end() ; } ;
 
   template< class F >
   bool iterate_our_windows( F f )
   {
     only_our_processes< F > g( * this, f ) ;
     return enumerate_windows( g ) ;
   }
 
   /*
    * Shut down every process in the pid_set.
    */
   bool shut_down() ;
 
 } ;
 
 #endif