Changeset 3505 in kBuild

Timestamp:

Dec 15, 2021 10:53:57 PM (3 years ago)

Author:

bird

Message:

kash: shthread_set_name for naming the async CloseHandle thread.

Location:

trunk/src/kash

Files:

• shfile.c (modified) (1 diff)
• shthread.c (modified) (1 diff)

trunk/src/kash/shfile.c

@@ -221 +221 @@
 {
     KBOOL decrement_pending = K_FALSE;
+    shthread_set_name("Async CloseHandle");
     while (!g_shfile_async_close.terminate_threads)
     {

trunk/src/kash/shthread.c

@@ -108 +108 @@
 }
 
+
+/**
+ * Sets the name of the current thread if supported by the OS.
+ */
+void shthread_set_name(const char *name)
+{
+#if K_OS == K_OS_WINDOWS
+    typedef BOOL (WINAPI * PFNSETTHREADDESCRIPTION)(HANDLE, WCHAR *);
+    static KBOOL volatile                   s_initialized             = K_FALSE;
+    static PFNSETTHREADDESCRIPTION volatile s_pfnSetThreadDescription = NULL;
+    PFNSETTHREADDESCRIPTION                 pfnSetThreadDescription   = s_pfnSetThreadDescription;
+    WCHAR                                   wszName[32];
+    size_t                                  i;
+
+    /* Get the function pointer, return if not available. */
+    if (pfnSetThreadDescription)
+    { }
+    else if (s_initialized)
+        return;
+    else
+    {
+        pfnSetThreadDescription = (PFNSETTHREADDESCRIPTION)GetProcAddress(GetModuleHandleW(L"KERNEL32.DLL"),
+                                                                          "SetThreadDescription");
+        s_pfnSetThreadDescription = pfnSetThreadDescription;
+        s_initialized = K_TRUE;
+        if (!pfnSetThreadDescription)
+            return;
+    }
+
+    /* Convert the name to UTF-16 and call the API. */
+    i = strlen(name);
+    kHlpAssertStmt(i < K_ELEMENTS(wszName), i = K_ELEMENTS(wszName));
+    wszName[i] = '\0';
+    while (i-- > 0)
+        wszName[i] = name[i];
+
+    pfnSetThreadDescription(GetCurrentThread(), wszName);
+#else
+    K_NOREF(name);
+#endif
+}
+