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

Timestamp:

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

Author:

sandervl

Message:

Implemented Get/SetProcessAffinityMask & SetThreadAffinityMask

File:

• trunk/src/kernel32/oslibdos.cpp (modified) (2 diffs)

trunk/src/kernel32/oslibdos.cpp

@@ -1 @@
-/* $Id: oslibdos.cpp,v 1.83 2001-11-05 15:00:36 sandervl Exp $ */
+/* $Id: oslibdos.cpp,v 1.84 2001-11-10 12:47:46 sandervl Exp $ */
 /*
  * Wrappers for OS/2 Dos* API
@@ -2815 +2815 @@
 //******************************************************************************
 //******************************************************************************
+BOOL OSLibDosSetThreadAffinity(DWORD dwThreadAffinityMask)
+{
+  static PROC_DosSetThreadAffinity pfnDosSetThreadAffinity = NULL;
+  static BOOL fInit = FALSE;
+  MPAFFINITY mask;
+
+    if(fInit == FALSE && pfnDosSetThreadAffinity == NULL) {
+        HMODULE hDoscalls;
+        if(DosQueryModuleHandle("DOSCALLS", &hDoscalls) == NO_ERROR) {
+            DosQueryProcAddr(hDoscalls, 564, NULL, (PFN *)&pfnDosSetThreadAffinity);
+        }
+        fInit = TRUE;
+    }
+    if(fInit && pfnDosSetThreadAffinity == NULL) {
+        SetLastError(ERROR_SUCCESS_W);
+        return TRUE;
+    }
+    APIRET rc;
+    USHORT sel = RestoreOS2FS();
+
+    mask.mask[0] = dwThreadAffinityMask;
+    mask.mask[1] = 0; //TODO: this might not be a good idea, but then again, not many people have > 32 cpus
+
+    rc = pfnDosSetThreadAffinity(&mask);
+    SetFS(sel);
+
+    SetLastError(error2WinError(rc,ERROR_INVALID_PARAMETER));
+    return (rc == NO_ERROR);
+}
+//******************************************************************************
+//******************************************************************************
+BOOL OSLibDosQueryAffinity(DWORD fMaskType, DWORD *pdwThreadAffinityMask)
+{
+  static PROC_DosQueryThreadAffinity pfnDosQueryThreadAffinity = NULL;
+  static BOOL fInit = FALSE;
+  MPAFFINITY mask;
+
+    if(fInit == FALSE && pfnDosQueryThreadAffinity == NULL) {
+        HMODULE hDoscalls;
+        if(DosQueryModuleHandle("DOSCALLS", &hDoscalls) == NO_ERROR) {
+            DosQueryProcAddr(hDoscalls, 563, NULL, (PFN *)&pfnDosQueryThreadAffinity);
+        }
+        fInit = TRUE;
+    }
+    if(fInit && pfnDosQueryThreadAffinity == NULL) {
+        *pdwThreadAffinityMask = 1;
+        SetLastError(ERROR_SUCCESS_W);
+        return TRUE;
+    }
+
+    ULONG scope = (fMaskType == MASK_SYSTEM) ? AFNTY_SYSTEM : AFNTY_THREAD;
+
+    APIRET rc;
+    USHORT sel = RestoreOS2FS();
+
+    rc = pfnDosQueryThreadAffinity(scope, &mask);
+    SetFS(sel);
+
+    if(rc == NO_ERROR) {
+        *pdwThreadAffinityMask = mask.mask[0];
+    }
+    SetLastError(error2WinError(rc,ERROR_INVALID_PARAMETER));
+    return (rc == NO_ERROR);
+}
+//******************************************************************************
+//******************************************************************************