Changeset 7318 for trunk/src/kernel32/thread.cpp

Timestamp:

Nov 10, 2001, 1:47:48 PM (24 years ago)

Author:

sandervl

Message:

Implemented Get/SetProcessAffinityMask & SetThreadAffinityMask

File:

• trunk/src/kernel32/thread.cpp (modified) (4 diffs)

trunk/src/kernel32/thread.cpp

@@ -1 @@
-/* $Id: thread.cpp,v 1.32 2001-09-26 15:29:39 phaller Exp $ */
+/* $Id: thread.cpp,v 1.33 2001-11-10 12:47:48 sandervl Exp $ */
 
 /*
@@ -31 +31 @@
 #include "exceptutil.h"
 #include "oslibmisc.h"
+#include "oslibdos.h"
 #include <handlemanager.h>
 
@@ -126 +127 @@
   O32_ExitThread(exitcode);
 }
+/*****************************************************************************
+ * Name      : DWORD SetThreadAffinityMask
+ * Purpose   : The SetThreadAffinityMask function sets a processor affinity
+ *             mask for a specified thread.
+ *             A thread affinity mask is a bit vector in which each bit
+ *             represents the processors that a thread is allowed to run on.
+ *             A thread affinity mask must be a proper subset of the process
+ *             affinity mask for the containing process of a thread. A thread
+ *             is only allowed to run on the processors its process is allowed to run on.
+ * Parameters: HANDLE hThread              handle to the thread of interest
+ *             DWORD  dwThreadAffinityMask a thread affinity mask
+ * Variables :
+ * Result    : TRUE / FALSE
+ * Remark    :
+ * Status    : UNTESTED STUB
+ *
+ * Author    : Patrick Haller [Mon, 1998/06/15 08:00]
+ *****************************************************************************/
+
+DWORD WIN32API SetThreadAffinityMask(HANDLE hThread,
+                                     DWORD  dwThreadAffinityMask)
+{
+  dprintf(("KERNEL32: SetThreadAffinityMask(%08xh,%08xh)",
+           hThread, dwThreadAffinityMask));
+
+  if(hThread != GetCurrentThread()) {
+      dprintf(("WARNING: Setting the affinity mask for another thread than the current one is not supported!!"));
+      return FALSE;
+  }
+  return OSLibDosSetThreadAffinity(dwThreadAffinityMask);
+}
 //******************************************************************************
 //******************************************************************************
@@ -171 +203 @@
 
   SetWin32TIB();
+
+  DWORD dwProcessAffinityMask, dwSystemAffinityMask;
+
+  //Change the affinity mask of this thread to the mask for the whole process
+  if(GetProcessAffinityMask(GetCurrentProcess(), &dwProcessAffinityMask, &dwSystemAffinityMask) == TRUE) {
+      SetThreadAffinityMask(GetCurrentThread(), dwProcessAffinityMask);
+  }
+
   WinExe->tlsAttachThread();              //setup TLS structure of main exe
   Win32DllBase::tlsAttachThreadToAllDlls(); //setup TLS structures of all dlls