블럭 방지.

ZCube · Apr 17, 2017 · ea6c0cd · ea6c0cd
1 parent dd97a18
commit ea6c0cd
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/ACTWebSocket.Core/Native.cs b/ACTWebSocket.Core/Native.cs
@@ -195,6 +195,15 @@ public static List<string> SearchForWindow(string title)
 
         public static bool EnumProc(IntPtr hWnd, ref SearchData data)
         {
+            // 종료 과정 중 자신의 프로세스 체크시 정지되는 경우 있음.
+            // 만악의 근원
+            uint id = 0;
+            GetWindowThreadProcessId(hWnd, out id);
+            if(GetCurrentProcessId() == id)
+            {
+                return true;
+            }
+
             StringBuilder sb = new StringBuilder(1024);
             int ret = GetClassName(hWnd, sb, sb.Capacity);
             if(ret == 0)
@@ -221,6 +230,12 @@ public class SearchData
 
         private delegate bool EnumWindowsProc(IntPtr hWnd, ref SearchData data);
 
+        [DllImport("user32.dll", SetLastError = true)]
+        static extern uint GetWindowThreadProcessId(IntPtr hWnd, out uint lpdwProcessId);
+
+        [DllImport("kernel32.dll")]
+        static extern uint GetCurrentProcessId();
+
         [DllImport("kernel32.dll", SetLastError = true)]
         [ReliabilityContract(Consistency.WillNotCorruptState, Cer.Success)]
         [SuppressUnmanagedCodeSecurity]