sched: Make sure that affinity mask is valid

Signed-off-by: hujun5 <hujun5@xiaomi.com>
hujun260 · Sep 4, 2024 · 269f6a5 · 269f6a5
1 parent f084685
commit 269f6a5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/sched/sched/sched_setaffinity.c b/sched/sched/sched_setaffinity.c
@@ -78,6 +78,13 @@ int nxsched_set_affinity(pid_t pid, size_t cpusetsize,
 
   DEBUGASSERT(cpusetsize == sizeof(cpu_set_t) && mask != NULL);
 
+  /* Make sure that affinity mask is valid */
+
+  if (*mask == 0)
+    {
+      return -EINVAL;
+    }
+
   /* Verify that the PID corresponds to a real task */
 
   if (!pid)