[virt/cpu] Use better return value for SYS_sched_getaffinity.

s5z · Feb 22, 2020 · 775d84b · 775d84b
1 parent 4879ffc
commit 775d84b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/virt/cpu.cpp b/src/virt/cpu.cpp
@@ -80,7 +80,9 @@ PostPatchFn PatchSchedGetaffinity(PrePatchArgs args) {
             return PPA_NOTHING;
         }
         // On success, the syscall returns the size of cpumask_t in bytes.
-        const int maxSize = MAX(1024, (1 << (ilog2(zinfo->numCores) + 1))) / 8;
+        int maxSize = -err;
+        // When the simulated number of cores is greater, extend cpumask_t.
+        maxSize = MAX(maxSize, (1 << (ilog2(zinfo->numCores - 1) + 1)) / 8);
         PIN_SetSyscallNumber(args.ctxt, args.std, maxSize);
         uint32_t linuxTid = PIN_GetSyscallArgument(args.ctxt, args.std, 0);
         uint32_t tid = (linuxTid == 0 ? args.tid : zinfo->sched->getTidFromLinuxTid(linuxTid));