Merge pull request llvm#98 from pcc/filc6

Implement get_mempolicy and set_mempolicy

Author: pizlonator

filc/include/pizlonated_syscalls.h

@@ -401,6 +401,10 @@ long zsys_keyctl_pkey_sign(int key_id, const char* info, const void* data, __SIZ
                            void* sig, __SIZE_TYPE__ sig_len);
 long zsys_keyctl_pkey_verify(int key_id, const char* info, const void* data, __SIZE_TYPE__ data_len,
                              const void* sig, __SIZE_TYPE__ sig_len);
+long zsys_get_mempolicy(int *mode, unsigned long *nodemask,
+                        unsigned long maxnode, void *addr, unsigned long flags);
+long zsys_set_mempolicy(int mode, const unsigned long *nodemask,
+                        unsigned long maxnode);
 
 #ifdef __cplusplus
 }

filc/src/runtime.c

@@ -511,6 +511,14 @@ long zsys_syscall(long n, ...)
         callee = zsys_getdents;
         break;
 
+    case 238 /* SYS_set_mempolicy */:
+        callee = zsys_set_mempolicy;
+        break;
+
+    case 239 /* SYS_get_mempolicy */:
+        callee = zsys_get_mempolicy;
+        break;
+
     case 444 /* SYS_landlock_create_ruleset */:
         callee = zsys_landlock_create_ruleset;
         break;

libpas/src/libpas/filc_runtime.c

@@ -10522,6 +10522,42 @@ int filc_native_zsys_sched_getaffinity(filc_thread* my_thread, int tid, size_t s
     return FILC_SYSCALL(my_thread, sched_getaffinity(tid, size, (cpu_set_t*)filc_ptr_ptr(set_ptr)));
 }
 
+long filc_native_zsys_get_mempolicy(filc_thread* my_thread, filc_ptr mode, filc_ptr nodemask,
+                                    unsigned long maxnode, filc_ptr addr, unsigned long flags)
+{
+    if (filc_ptr_ptr(mode))
+        filc_check_write(mode, sizeof(int));
+    if (filc_ptr_ptr(nodemask)) {
+        filc_check_write(
+            nodemask,
+            filc_mul_size(sizeof(unsigned long),
+                          filc_add_size(maxnode, 8 * sizeof(unsigned long) - 1) /
+                              (8 * sizeof(unsigned long))));
+    }
+    if (filc_ptr_ptr(addr))
+        filc_check_read(addr, 1);
+    return FILC_SYSCALL(my_thread,
+                        syscall(SYS_get_mempolicy, (int *)filc_ptr_ptr(mode),
+                                (unsigned long *)filc_ptr_ptr(nodemask),
+                                maxnode, filc_ptr_ptr(addr), flags));
+}
+
+long filc_native_zsys_set_mempolicy(filc_thread* my_thread, int mode, filc_ptr nodemask,
+                                    unsigned long maxnode)
+{
+    if (filc_ptr_ptr(nodemask)) {
+        filc_check_read(
+            nodemask,
+            filc_mul_size(sizeof(unsigned long),
+                          filc_add_size(maxnode, 8 * sizeof(unsigned long) - 1) /
+                              (8 * sizeof(unsigned long))));
+    }
+    return FILC_SYSCALL(my_thread,
+                        syscall(SYS_set_mempolicy, mode,
+                                (const unsigned long *)filc_ptr_ptr(nodemask),
+                                maxnode));
+}
+
 int filc_native_zsys_posix_fadvise(filc_thread* my_thread, int fd, long base, long len, int advice)
 {
     check_fd(fd);

libpas/src/libpas/generate_pizlonated_forwarders.rb

@@ -519,6 +519,8 @@ def addOutSig(name, rets, *args)
 addSig "long", "zsys_keyctl_pkey_sign", "int", "filc_ptr", "filc_ptr", "size_t", "filc_ptr", "size_t"
 addSig "long", "zsys_keyctl_pkey_verify", "int", "filc_ptr", "filc_ptr", "size_t", "filc_ptr",
        "size_t"
+addSig "long", "zsys_get_mempolicy", "filc_ptr", "filc_ptr", "unsigned long", "filc_ptr", "unsigned long"
+addSig "long", "zsys_set_mempolicy", "int", "filc_ptr", "unsigned long"
 
 addSig "filc_ptr", "zthread_self"
 addSig "unsigned", "zthread_get_id", "filc_ptr"