[Qemu-devel] [PATCH] linux-user: implement sched_{g,s}etaffinity

[Mike Frysinger <vapier@gentoo.org>]

Signed-off-by: Mike Frysinger <vapier@gentoo.org>
---
 linux-user/syscall.c |   50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 50 insertions(+), 0 deletions(-)

diff --git a/linux-user/syscall.c b/linux-user/syscall.c
index 3e285c6..f95429f 100644
--- a/linux-user/syscall.c
+++ b/linux-user/syscall.c
@@ -235,6 +235,12 @@ _syscall6(int,sys_futex,int *,uaddr,int,op,int,val,
           const struct timespec *,timeout,int *,uaddr2,int,val3)
 #endif
 #endif
+#define __NR_sys_sched_getaffinity __NR_sched_getaffinity
+_syscall3(int, sys_sched_getaffinity, pid_t, pid, unsigned int, len,
+          unsigned long *, user_mask_ptr);
+#define __NR_sys_sched_setaffinity __NR_sched_setaffinity
+_syscall3(int, sys_sched_setaffinity, pid_t, pid, unsigned int, len,
+          unsigned long *, user_mask_ptr);
 
 static bitmask_transtbl fcntl_flags_tbl[] = {
   { TARGET_O_ACCMODE,   TARGET_O_WRONLY,    O_ACCMODE,   O_WRONLY,    },
@@ -6374,6 +6380,50 @@ abi_long do_syscall(void *cpu_env, int num, abi_long arg1,
            return value. */
         ret = -TARGET_ENOTDIR;
         break;
+    case TARGET_NR_sched_getaffinity:
+        {
+            unsigned int num_sets, host_mask_size, i;
+            unsigned long *host_mask;
+            abi_ulong *target_mask;
+
+            num_sets = arg2 / sizeof(abi_ulong);
+            host_mask_size = num_sets * sizeof(*host_mask);
+            host_mask = alloca(host_mask_size);
+
+            ret = get_errno(sys_sched_getaffinity(arg1, host_mask_size, host_mask));
+
+            if (!is_error(ret)) {
+                if (!lock_user_struct(VERIFY_WRITE, target_mask, arg3, 0)) {
+                    goto efault;
+                }
+                for (i = 0; i < num_sets; ++i) {
+                    target_mask[i] = tswapl(host_mask[i]);
+                }
+                unlock_user_struct(target_mask, arg2, 1);
+            }
+        }
+        break;
+    case TARGET_NR_sched_setaffinity:
+        {
+            unsigned int num_sets, host_mask_size, i;
+            unsigned long *host_mask;
+            abi_ulong *target_mask;
+
+            num_sets = arg2 / sizeof(abi_ulong);
+            host_mask_size = num_sets * sizeof(*host_mask);
+            host_mask = alloca(host_mask_size);
+
+            if (!lock_user_struct(VERIFY_READ, target_mask, arg3, 1)) {
+                goto efault;
+            }
+            for (i = 0; i < num_sets; ++i) {
+                host_mask[i] = tswapl(target_mask[i]);
+            }
+            unlock_user_struct(target_mask, arg2, 0);
+
+            ret = get_errno(sys_sched_setaffinity(arg1, host_mask_size, host_mask));
+        }
+        break;
     case TARGET_NR_sched_setparam:
         {
             struct sched_param *target_schp;
-- 
1.7.4.rc2