Changeset 37154 in vbox for trunk/src/VBox/Runtime/r3

Timestamp:

May 19, 2011 12:54:32 PM (14 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

71834

Message:

RTThread[SG]etAffinity cleanup

Location:

trunk/src/VBox/Runtime/r3

Files:

• os2/thread-os2.cpp (modified) (3 diffs)
• posix/thread-posix.cpp (modified) (1 diff)
• solaris/thread-affinity-solaris.cpp (added)
• win/thread-win.cpp (modified) (3 diffs)

trunk/src/VBox/Runtime/r3/os2/thread-os2.cpp

@@ -199 +199 @@
 
 
-RTDECL(uint64_t) RTThreadGetAffinity(void)
+RTR3DECL(int) RTThreadGetAffinity(PRTCPUSET pCpuSet)
+{
+    return VINF_SUCCESS;
+}
+
+RTR3DECL(int) RTThreadGetAffinity(PRTCPUSET pCpuSet)
 {
     union
@@ -207 +212 @@
     } u;
 
-    int rc = DosQueryThreadAffinity(AFNTY_THREAD, &u.mpaff);
-    if (rc)
-        u.u64 = 1;
-    return u.u64;
-}
-
-
-RTDECL(int) RTThreadSetAffinity(uint64_t u64Mask)
+    APIRET rc = DosQueryThreadAffinity(AFNTY_THREAD, &u.mpaff);
+    if (!rc)
+    {
+        RTCpuSetFromU64(pCpuSet, u.u64);
+        return VINF_SUCCESS;
+    }
+    return RTErrConvertFromOS2(rc);
+}
+
+
+RTR3DECL(int) RTThreadSetAffinity(PCRTCPUSET pCpuSet)
 {
     union
@@ -221 +229 @@
         MPAFFINITY mpaff;
     } u;
-    u.u64 = u64Mask;
+    u.u64 = pCpuSet ? RTCpuSetToU64(pCpuSet) : UINT64_MAX;
     int rc = DosSetThreadAffinity(&u.mpaff);
     if (!rc)

trunk/src/VBox/Runtime/r3/posix/thread-posix.cpp

@@ -384 +384 @@
 
 
-RTR3DECL(uint64_t) RTThreadGetAffinity(void)
-{
-    return 1;
-}
-
-
-RTR3DECL(int) RTThreadSetAffinity(uint64_t u64Mask)
-{
-    if (u64Mask != 1)
-        return VERR_INVALID_PARAMETER;
-    return VINF_SUCCESS;
-}
-
-
 #ifdef RTTHREAD_POSIX_WITH_POKE
 RTDECL(int) RTThreadPoke(RTTHREAD hThread)

trunk/src/VBox/Runtime/r3/win/thread-win.cpp

@@ -281 +281 @@
 
 
-RTR3DECL(int) RTThreadSetAffinity(uint64_t u64Mask)
-{
-    Assert((DWORD_PTR)u64Mask == u64Mask || u64Mask == ~(uint64_t)0);
-    DWORD_PTR dwRet = SetThreadAffinityMask(GetCurrentThread(), (DWORD_PTR)u64Mask);
+RTR3DECL(int) RTThreadSetAffinity(PCRTCPUSET pCpuSet)
+{
+    DWORD_PTR fNewMask = pCpuSet ? RTCpuSetToU64(pCpuSet) : ~(DWORD_PTR)0;
+    DWORD_PTR dwRet = SetThreadAffinityMask(GetCurrentThread(), fNewMask);
     if (dwRet)
         return VINF_SUCCESS;
@@ -294 +294 @@
 
 
-RTR3DECL(uint64_t) RTThreadGetAffinity(void)
+RTR3DECL(int) RTThreadGetAffinity(PRTCPUSET pCpuSet)
 {
     /*
@@ -309 +309 @@
             DWORD_PTR dwSet = SetThreadAffinityMask(hThread, dwRet);
             Assert(dwSet == dwProcAff); NOREF(dwRet);
-            return dwRet;
+
+            RTCpuSetFromU64(pCpuSet, (uint64_t)dwSet);
+            return VINF_SUCCESS;
         }
     }