[dpdk-dev] [PATCH] eal/windows: fix invalid thread handle

[Tasnim Bashar]

Casting thread ID to handle is not accurate way to get thread handle.
Need to use OpenThread function to get thread handle from thread ID.

pthread_setaffinity_np and pthread_getaffinity_np functions
for Windows are affected because of it.

Signed-off-by: Tasnim Bashar <tbas...@mellanox.com>
---
 lib/librte_eal/windows/include/pthread.h | 40 +++++++++++++++++++++---
 1 file changed, 35 insertions(+), 5 deletions(-)

diff --git a/lib/librte_eal/windows/include/pthread.h 
b/lib/librte_eal/windows/include/pthread.h
index 0bbed5c3b8..d2a986f8fd 100644
--- a/lib/librte_eal/windows/include/pthread.h
+++ b/lib/librte_eal/windows/include/pthread.h
@@ -18,6 +18,7 @@ extern "C" {
 
 #include <windows.h>
 #include <rte_common.h>
+#include <rte_windows.h>
 
 #define PTHREAD_BARRIER_SERIAL_THREAD TRUE
 
@@ -50,7 +51,19 @@ typedef SYNCHRONIZATION_BARRIER pthread_barrier_t;
 static inline int
 eal_set_thread_affinity_mask(pthread_t threadid, unsigned long *cpuset)
 {
-       SetThreadAffinityMask((HANDLE) threadid, *cpuset);
+       HANDLE thread_handle = OpenThread(THREAD_ALL_ACCESS, FALSE, threadid);
+       if (thread_handle == NULL) {
+               RTE_LOG_WIN32_ERR("OpenThread()");
+               return -1;
+       }
+
+       DWORD ret = SetThreadAffinityMask(thread_handle, *cpuset);
+       if (ret == 0) {
+               RTE_LOG_WIN32_ERR("SetThreadAffinityMask()");
+               CloseHandle(thread_handle);
+               return -1;
+       }
+       CloseHandle(thread_handle);
        return 0;
 }
 
@@ -60,12 +73,29 @@ eal_get_thread_affinity_mask(pthread_t threadid, unsigned 
long *cpuset)
        /* Workaround for the lack of a GetThreadAffinityMask()
         *API in Windows
         */
-               /* obtain previous mask by setting dummy mask */
-       DWORD dwprevaffinitymask =
-               SetThreadAffinityMask((HANDLE) threadid, 0x1);
+       HANDLE thread_handle = OpenThread(THREAD_ALL_ACCESS, FALSE, threadid);
+       if (thread_handle == NULL) {
+               RTE_LOG_WIN32_ERR("OpenThread()");
+               return -1;
+       }
+
+       /* obtain previous mask by setting dummy mask */
+       DWORD dwprevaffinitymask = SetThreadAffinityMask(thread_handle, 0x1);
+       if (dwprevaffinitymask == 0) {
+               RTE_LOG_WIN32_ERR("SetThreadAffinityMask()");
+               CloseHandle(thread_handle);
+               return -1;
+       }
+
        /* set it back! */
-       SetThreadAffinityMask((HANDLE) threadid, dwprevaffinitymask);
+       DWORD ret = SetThreadAffinityMask(thread_handle, dwprevaffinitymask);
+       if (ret == 0) {
+               RTE_LOG_WIN32_ERR("SetThreadAffinityMask()");
+               CloseHandle(thread_handle);
+               return -1;
+       }
        *cpuset = dwprevaffinitymask;
+       CloseHandle(thread_handle);
        return 0;
 }
 
-- 
2.19.1.windows.1