From 313e439c786d68bcf374b2cb0edfe3ffd891db94 Mon Sep 17 00:00:00 2001
From: David CARLIER <devnexen@gmail.com>
Date: Sat, 15 Jul 2023 07:34:49 +0100
Subject: [PATCH] thead pool debugging support proposal using native
 capabilities if possible.

---
 src/lib/utils/os_utils.cpp                 | 28 ++++++++++++++++++----
 src/lib/utils/os_utils.h                   |  8 +++++++
 src/lib/utils/thread_utils/thread_pool.cpp |  7 +++++-
 3 files changed, 38 insertions(+), 5 deletions(-)

diff --git a/src/lib/utils/os_utils.cpp b/src/lib/utils/os_utils.cpp
index bab3c776dbf..9a6e148421c 100644
--- a/src/lib/utils/os_utils.cpp
+++ b/src/lib/utils/os_utils.cpp
@@ -16,16 +16,13 @@
 #include <chrono>
 #include <cstdlib>
 
-#if defined(BOTAN_TARGET_OS_HAS_THREADS)
-   #include <thread>
-#endif
-
 #if defined(BOTAN_TARGET_OS_HAS_EXPLICIT_BZERO)
    #include <string.h>
 #endif
 
 #if defined(BOTAN_TARGET_OS_HAS_POSIX1)
    #include <errno.h>
+   #include <pthread.h>
    #include <setjmp.h>
    #include <signal.h>
    #include <stdlib.h>
@@ -72,6 +69,10 @@ extern "C" char** environ;
    #include <sys/prctl.h>
 #endif
 
+#if defined(BOTAN_TARGET_IS_FREEBSD) || defined(BOTAN_TARGET_IS_OPENBSD)
+   #include <pthread_np.h>
+#endif
+
 namespace Botan {
 
 // Not defined in OS namespace for historical reasons
@@ -614,6 +615,25 @@ void OS::page_named(void* page, size_t size) {
 #endif
 }
 
+#if defined(BOTAN_TARGET_OS_HAS_THREADS)
+void OS::set_thread_name(std::thread& thread, const std::string& name) {
+   #if defined(BOTAN_TARGET_OS_IS_LINUX) || defined(BOTAN_TARGET_OS_IS_FREEBSD)
+   static_cast<void>(pthread_setname_np(thread.native_handle(), name.c_str()));
+   #elif defined(BOTAN_TARGET_OS_IS_OPENBSD)
+   static_cast<void>(pthread_set_name_np(thread.native_handle(), name.c_str()));
+   #elif defined(BOTAN_TARGET_OS_IS_NETBSD)
+   static_cast<void>(pthread_set_name_np(thread.native_handle(), "%s", const_cast<char*>(name.c_str())));
+   #elif defined(BOTAN_TARGET_OS_HAS_WIN32) && defined(BOTAN_BUILD_COMPILER_IS_MSVC)
+   // Using SetThreadDescription from Win10
+   BOTAN_UNUSED(thread, name);
+   #else
+   // TODO other possible oses ?
+   // macOs does not seem to allow to name threads other than the current one.
+   BOTAN_UNUSED(thread, name);
+   #endif
+}
+#endif
+
 #if defined(BOTAN_TARGET_OS_HAS_POSIX1) && !defined(BOTAN_TARGET_OS_IS_EMSCRIPTEN)
 
 namespace {
diff --git a/src/lib/utils/os_utils.h b/src/lib/utils/os_utils.h
index 85bf8407c29..cda39f10a06 100644
--- a/src/lib/utils/os_utils.h
+++ b/src/lib/utils/os_utils.h
@@ -13,6 +13,10 @@
 #include <string>
 #include <vector>
 
+#if defined(BOTAN_TARGET_OS_HAS_THREADS)
+   #include <thread>
+#endif
+
 namespace Botan::OS {
 
 /*
@@ -148,6 +152,10 @@ void page_allow_access(void* page);
 */
 void page_named(void* page, size_t size);
 
+#if defined(BOTAN_TARGET_OS_HAS_THREADS)
+void set_thread_name(std::thread& thread, const std::string& name);
+#endif
+
 /**
 * Run a probe instruction to test for support for a CPU instruction.
 * Runs in system-specific env that catches illegal instructions; this
diff --git a/src/lib/utils/thread_utils/thread_pool.cpp b/src/lib/utils/thread_utils/thread_pool.cpp
index d12eb874541..9572c730416 100644
--- a/src/lib/utils/thread_utils/thread_pool.cpp
+++ b/src/lib/utils/thread_utils/thread_pool.cpp
@@ -41,6 +41,8 @@ Thread_Pool& Thread_Pool::global_instance() {
 
 Thread_Pool::Thread_Pool(std::optional<size_t> opt_pool_size) {
    m_shutdown = false;
+   // On Linux, it is 16 length max, including terminator
+   const std::string tname = "Botan thread";
 
    if(!opt_pool_size.has_value()) {
       return;
@@ -65,8 +67,11 @@ Thread_Pool::Thread_Pool(std::optional<size_t> opt_pool_size) {
       }
    }
 
+   m_workers.resize(pool_size);
+
    for(size_t i = 0; i != pool_size; ++i) {
-      m_workers.push_back(std::thread(&Thread_Pool::worker_thread, this));
+      m_workers[i] = std::thread(&Thread_Pool::worker_thread, this);
+      OS::set_thread_name(m_workers[i], tname);
    }
 }