From 3b7cd74bfdf5632be8d4af1fe28c774f96f1a4fd Mon Sep 17 00:00:00 2001
From: Corinna Vinschen <corinna@vinschen.de>
Date: Thu, 11 Aug 2011 07:28:42 +0000
Subject: 	* (winpids::add): Make sure to store always a Windows PID in 
 pidlist, even if pid is a Cygwin PID. 	(winpids::enum_processes): Fetch
 Cygwin processes from listing of 	shared cygwin object dir in the native
 NT namespace.  Only if winpid 	is true, fetch Windows processes using an
 additional call to 	NtQuerySystemInformation.

---
 winsup/cygwin/ChangeLog |  9 +++++
 winsup/cygwin/pinfo.cc  | 93 ++++++++++++++++++++++++++++++++++++-------------
 2 files changed, 77 insertions(+), 25 deletions(-)

diff --git a/winsup/cygwin/ChangeLog b/winsup/cygwin/ChangeLog
index 7696cbe21..04c80fe9a 100644
--- a/winsup/cygwin/ChangeLog
+++ b/winsup/cygwin/ChangeLog
@@ -1,3 +1,12 @@
+2011-08-11  Corinna Vinschen  <corinna@vinschen.de>
+
+	* (winpids::add): Make sure to store always a Windows PID in
+	pidlist, even if pid is a Cygwin PID.
+	(winpids::enum_processes): Fetch Cygwin processes from listing of
+	shared cygwin object dir in the native NT namespace.  Only if winpid
+	is true, fetch Windows processes using an additional call to
+	NtQuerySystemInformation.
+
 2011-08-10  Corinna Vinschen  <corinna@vinschen.de>
 
 	* fhandler_process.cc (format_process_status): Always print process name
diff --git a/winsup/cygwin/pinfo.cc b/winsup/cygwin/pinfo.cc
index a526770f9..89a96ef16 100644
--- a/winsup/cygwin/pinfo.cc
+++ b/winsup/cygwin/pinfo.cc
@@ -1212,44 +1212,87 @@ out:
 	}
     }
   if (p || winpid)
-    pidlist[nelem++] = pid;
+    pidlist[nelem++] = !p ? pid : p->dwProcessId;
 }
 
 DWORD
 winpids::enum_processes (bool winpid)
 {
-  static DWORD szprocs;
-  static SYSTEM_PROCESSES *procs;
-
   DWORD nelem = 0;
-  if (!szprocs)
-    procs = (SYSTEM_PROCESSES *) malloc (sizeof (*procs) + (szprocs = 200 * sizeof (*procs)));
-
-  NTSTATUS res;
-  for (;;)
+  DWORD cygwin_pid_nelem = 0;
+  NTSTATUS status;
+  ULONG context;
+  struct fdbi
     {
-      res = NtQuerySystemInformation (SystemProcessesAndThreadsInformation,
-				      procs, szprocs, NULL);
-      if (res == 0)
-	break;
+      DIRECTORY_BASIC_INFORMATION dbi;
+      WCHAR buf[2][NAME_MAX + 1];
+    } f;
+  HANDLE dir = get_shared_parent_dir ();
+  BOOLEAN restart = TRUE;
 
-      if (res == STATUS_INFO_LENGTH_MISMATCH)
-	procs =  (SYSTEM_PROCESSES *) realloc (procs, szprocs += 200 * sizeof (*procs));
-      else
+  do
+    {
+      status = NtQueryDirectoryObject (dir, &f, sizeof f, TRUE, restart,
+				       &context, NULL);
+      if (NT_SUCCESS (status))
 	{
-	  system_printf ("error %p reading system process information", res);
-	  return 0;
+	  restart = FALSE;
+	  f.dbi.ObjectName.Buffer[f.dbi.ObjectName.Length / sizeof (WCHAR)]
+	    = L'\0';
+	  if (wcsncmp (f.dbi.ObjectName.Buffer, L"cygpid.", 7) == 0)
+	    {
+	      DWORD pid = wcstoul (f.dbi.ObjectName.Buffer + 7, NULL, 10);
+	      add (nelem, false, pid);
+	    }
 	}
     }
+  while (NT_SUCCESS (status));
+  cygwin_pid_nelem = nelem;
 
-  SYSTEM_PROCESSES *px = procs;
-  for (;;)
+  if (winpid)
     {
-      if (px->ProcessId)
-	add (nelem, winpid, px->ProcessId);
-      if (!px->NextEntryDelta)
-	break;
-      px = (SYSTEM_PROCESSES *) ((char *) px + px->NextEntryDelta);
+      static DWORD szprocs;
+      static SYSTEM_PROCESSES *procs;
+
+      if (!szprocs)
+	procs = (SYSTEM_PROCESSES *) malloc (sizeof (*procs) + (szprocs = 200 * sizeof (*procs)));
+
+      NTSTATUS res;
+      for (;;)
+	{
+	  res = NtQuerySystemInformation (SystemProcessesAndThreadsInformation,
+					  procs, szprocs, NULL);
+	  if (res == 0)
+	    break;
+
+	  if (res == STATUS_INFO_LENGTH_MISMATCH)
+	    procs =  (SYSTEM_PROCESSES *) realloc (procs, szprocs += 200 * sizeof (*procs));
+	  else
+	    {
+	      system_printf ("error %p reading system process information", res);
+	      return 0;
+	    }
+	}
+
+      SYSTEM_PROCESSES *px = procs;
+      for (;;)
+	{
+	  if (px->ProcessId)
+	    {
+	      bool do_add = true;
+	      for (unsigned i = 0; i < cygwin_pid_nelem; ++i)
+		if (pidlist[i] == px->ProcessId)
+		  {
+		    do_add = false;
+		    break;
+		  }
+	      if (do_add)
+		add (nelem, true, px->ProcessId);
+	    }
+	  if (!px->NextEntryDelta)
+	    break;
+	  px = (SYSTEM_PROCESSES *) ((char *) px + px->NextEntryDelta);
+	}
     }
 
   return nelem;
-- 
cgit v1.2.3