Changeset 1812 for trunk/eraser6/Eraser.Util.Native/OpenHandle.NameResolver.cpp

Timestamp:

2/11/2010 2:15:09 AM (2 years ago)

Author:

lowjoel

Message:

• Move the contents of the OpenHandle::ResolveHandlePath? function to the ResolveHandleName? function in native code (we would have p/invoked quite a bit in that function)
• Move the global name resolution thread and it's task queue to native code and in a anonymous namespace

File:

• trunk/eraser6/Eraser.Util.Native/OpenHandle.NameResolver.cpp (modified) (2 diffs)

trunk/eraser6/Eraser.Util.Native/OpenHandle.NameResolver.cpp

@@ -23 +23 @@
 #include "OpenHandle.h"
 
-#pragma managed(push)
 #pragma unmanaged
 
 namespace {
+    HANDLE NameResolutionThread;
+    NameResolutionThreadParams NameResolutionThreadParam;
+
     DWORD __stdcall nameResolutionThread(void* data)
     {
@@ -86 +88 @@
         return 0;
     }
+
+    void CreateNameThread(HANDLE& handle, NameResolutionThreadParams& params)
+    {
+        //If the handle is valid terminate the thread
+        if (handle)
+        {
+            TerminateThread(handle, 1);
+            CloseHandle(handle);
+        }
+
+        //Create the thread
+        handle = CreateThread(NULL, 0, nameResolutionThread, &params, 0, NULL);
+    }
+
 }
 
-void CreateNameThread(HANDLE& handle, NameResolutionThreadParams& params)
+std::wstring ResolveHandleName(HANDLE handle, int pid)
 {
-    //If the handle is valid terminate the thread
-    if (handle)
+    //Start a name resolution thread (in case one entry hangs)
+    if (NameResolutionThread == NULL)
+        CreateNameThread(NameResolutionThread, NameResolutionThreadParam);
+
+    //Create a duplicate handle
+    HANDLE localHandle;
+    HANDLE processHandle = OpenProcess(PROCESS_DUP_HANDLE, false, pid);
+    DuplicateHandle(processHandle, static_cast<void*>(handle), GetCurrentProcess(),
+        &localHandle, 0, false, DUPLICATE_SAME_ACCESS);
+    CloseHandle(processHandle);
+
+    //We need a handle
+    if (!localHandle)
+        return std::wstring();
+
+    //Send the handle to the secondary thread for name resolution
+    NameResult result(localHandle);
+    NameResolutionThreadParam.Input.push_back(&result);
+    ReleaseSemaphore(NameResolutionThreadParam.Semaphore, 1, NULL);
+
+    //Wait for the result
+    if (WaitForSingleObject(result.Event, 50) != WAIT_OBJECT_0)
     {
-        TerminateThread(handle, 1);
-        CloseHandle(handle);
+        //The wait failed. Terminate the thread and recreate another.
+        CreateNameThread(NameResolutionThread, NameResolutionThreadParam);
     }
 
-    //Create the thread
-    handle = CreateThread(NULL, 0, nameResolutionThread, &params, 0, NULL);
+    //Close the handle which we duplicated
+    CloseHandle(localHandle);
+
+    //Return the result
+    return result.Name;
 }