java - get process id in a way that works cross-platform

Jetski5822 · Sep 26, 2016 · b3c53bf · b3c53bf
1 parent 1c52dc8
commit b3c53bf
Showing 1 changed file with 22 additions and 9 deletions.
diff --git a/java/client/src/org/openqa/selenium/os/ProcessUtils.java b/java/client/src/org/openqa/selenium/os/ProcessUtils.java
@@ -184,15 +184,28 @@ private static void closeAllStreamsAndDestroyProcess(Process process) {
   }
 
   static int getProcessId(Process p) {
-    if (Platform.getCurrent().is(WINDOWS)) {
-      throw new IllegalStateException("UnixUtils may not be used on Windows");
-    }
-    try {
-      Field f = p.getClass().getDeclaredField("pid");
-      f.setAccessible(true);
-      return (Integer) f.get(p);
-    } catch (Exception e) {
-      throw new RuntimeException("Couldn't detect pid", e);
+    if (thisIsWindows()) {
+      try {
+        Field f = p.getClass().getDeclaredField("handle");
+        f.setAccessible(true);
+        long hndl = f.getLong(p);
+
+        Kernel32 kernel = Kernel32.INSTANCE;
+        WinNT.HANDLE handle = new WinNT.HANDLE();
+        handle.setPointer(Pointer.createConstant(hndl));
+        int pid = kernel.GetProcessId(handle);
+        return pid;
+      } catch (Exception e) {
+        throw new RuntimeException("Couldn't detect pid", e);
+      }
+    } else {
+      try {
+        Field f = p.getClass().getDeclaredField("pid");
+        f.setAccessible(true);
+        return (Integer) f.get(p);
+      } catch (Exception e) {
+        throw new RuntimeException("Couldn't detect pid", e);
+      }
     }
   }