Merge pull request #887 from isanae/spawning-and-waiting

Spawning, waiting and locking

1 files changed, 38 insertions, 2 deletions

diff --git a/src/envmodule.h b/src/envmodule.h
index deb7520f..d152b840 100644
--- a/src/envmodule.h
+++ b/src/envmodule.h
@@ -7,6 +7,23 @@
 namespace env
 {
 
+// used by HandlePtr, calls CloseHandle() as the deleter
+//
+struct HandleCloser
+{
+  using pointer = HANDLE;
+
+  void operator()(HANDLE h)
+  {
+    if (h != INVALID_HANDLE_VALUE) {
+      ::CloseHandle(h);
+    }
+  }
+};
+
+using HandlePtr = std::unique_ptr<HANDLE, HandleCloser>;
+
+
 // represents one module
 //
 class Module
@@ -101,25 +118,44 @@ private:
 class Process
 {
 public:
-  Process(DWORD pid, QString name);
+  Process();
+  explicit Process(HANDLE h);
+  Process(DWORD pid, DWORD ppid, QString name);
 
+  bool isValid() const;
   DWORD pid() const;
+  DWORD ppid() const;
   const QString& name() const;
 
+  HandlePtr openHandleForWait() const;
+
   // whether this process can be accessed; fails if the current process doesn't
   // have the proper permissions
   //
   bool canAccess() const;
 
+  void addChild(Process p);
+  std::vector<Process>& children();
+
 private:
   DWORD m_pid;
-  QString m_name;
+  mutable std::optional<DWORD> m_ppid;
+  mutable std::optional<QString> m_name;
+  std::vector<Process> m_children;
 };
 
 
 std::vector<Process> getRunningProcesses();
 std::vector<Module> getLoadedModules();
 
+Process getProcessTree(HANDLE parent);
+
+QString getProcessName(DWORD pid);
+QString getProcessName(HANDLE process);
+
+DWORD getProcessParentID(DWORD pid);
+DWORD getProcessParentID(HANDLE handle);
+
 } // namespace env
 
 #endif // ENV_MODULE_H