242f5d
commit a53adb79711ccfc76a4ee32b20253045cdab55c7
242f5d
Author: Mark Wielaard <mark@klomp.org>
242f5d
Date:   Mon Jul 27 16:36:17 2020 +0200
242f5d
242f5d
    Handle linux syscalls sched_getattr and sched_setattr
242f5d
    
242f5d
    The only "special" thing about these syscalls is that the given
242f5d
    struct sched_attr determines its own size for future expansion.
242f5d
    
242f5d
    Original fix by "ISHIKAWA,chiaki" <ishikawa@yk.rim.or.jp>
242f5d
    
242f5d
    https://bugs.kde.org/show_bug.cgi?id=369029
242f5d
242f5d
diff --git a/coregrind/m_syswrap/priv_syswrap-linux.h b/coregrind/m_syswrap/priv_syswrap-linux.h
242f5d
index cdc73c1e6..eb0b320ca 100644
242f5d
--- a/coregrind/m_syswrap/priv_syswrap-linux.h
242f5d
+++ b/coregrind/m_syswrap/priv_syswrap-linux.h
242f5d
@@ -227,6 +227,8 @@ DECL_TEMPLATE(linux, sys_fremovexattr);
242f5d
 // syscalls.
242f5d
 DECL_TEMPLATE(linux, sys_sched_setparam);
242f5d
 DECL_TEMPLATE(linux, sys_sched_getparam);
242f5d
+DECL_TEMPLATE(linux, sys_sched_setattr);
242f5d
+DECL_TEMPLATE(linux, sys_sched_getattr);
242f5d
 DECL_TEMPLATE(linux, sys_sched_setscheduler);
242f5d
 DECL_TEMPLATE(linux, sys_sched_getscheduler);
242f5d
 DECL_TEMPLATE(linux, sys_sched_yield);
242f5d
diff --git a/coregrind/m_syswrap/syswrap-amd64-linux.c b/coregrind/m_syswrap/syswrap-amd64-linux.c
242f5d
index 28d90135a..d6f3eb910 100644
242f5d
--- a/coregrind/m_syswrap/syswrap-amd64-linux.c
242f5d
+++ b/coregrind/m_syswrap/syswrap-amd64-linux.c
242f5d
@@ -846,9 +846,8 @@ static SyscallTableEntry syscall_table[] = {
242f5d
    LINX_(__NR_process_vm_writev, sys_process_vm_writev),// 311
242f5d
    LINX_(__NR_kcmp,              sys_kcmp),             // 312
242f5d
    LINX_(__NR_finit_module,      sys_finit_module),     // 313
242f5d
-//   LIN__(__NR_sched_setattr,     sys_ni_syscall),       // 314
242f5d
-
242f5d
-//   LIN__(__NR_sched_getattr,     sys_ni_syscall),       // 315
242f5d
+   LINX_(__NR_sched_setattr,     sys_sched_setattr),    // 314
242f5d
+   LINXY(__NR_sched_getattr,     sys_sched_getattr),    // 315
242f5d
    LINX_(__NR_renameat2,         sys_renameat2),        // 316
242f5d
 //   LIN__(__NR_seccomp,           sys_ni_syscall),       // 317
242f5d
    LINXY(__NR_getrandom,         sys_getrandom),        // 318
242f5d
diff --git a/coregrind/m_syswrap/syswrap-arm-linux.c b/coregrind/m_syswrap/syswrap-arm-linux.c
242f5d
index 579542785..70700e53f 100644
242f5d
--- a/coregrind/m_syswrap/syswrap-arm-linux.c
242f5d
+++ b/coregrind/m_syswrap/syswrap-arm-linux.c
242f5d
@@ -1009,6 +1009,8 @@ static SyscallTableEntry syscall_main_table[] = {
242f5d
    LINXY(__NR_process_vm_readv,  sys_process_vm_readv), // 376
242f5d
    LINX_(__NR_process_vm_writev, sys_process_vm_writev),// 377
242f5d
 
242f5d
+   LINX_(__NR_sched_setattr,     sys_sched_setattr),    // 380
242f5d
+   LINXY(__NR_sched_getattr,     sys_sched_getattr),    // 381
242f5d
    LINX_(__NR_renameat2,         sys_renameat2),        // 382
242f5d
 
242f5d
    LINXY(__NR_getrandom,         sys_getrandom),        // 384
242f5d
diff --git a/coregrind/m_syswrap/syswrap-arm64-linux.c b/coregrind/m_syswrap/syswrap-arm64-linux.c
242f5d
index 81e01456f..acca02442 100644
242f5d
--- a/coregrind/m_syswrap/syswrap-arm64-linux.c
242f5d
+++ b/coregrind/m_syswrap/syswrap-arm64-linux.c
242f5d
@@ -806,8 +806,8 @@ static SyscallTableEntry syscall_main_table[] = {
242f5d
    LINX_(__NR_process_vm_writev, sys_process_vm_writev), // 271
242f5d
    LINX_(__NR_kcmp,              sys_kcmp),              // 272
242f5d
    LINX_(__NR_finit_module,      sys_finit_module),      // 273
242f5d
-   //   (__NR_sched_setattr,     sys_ni_syscall),        // 274
242f5d
-   //   (__NR_sched_getattr,     sys_ni_syscall),        // 275
242f5d
+   LINX_(__NR_sched_setattr,     sys_sched_setattr),     // 274
242f5d
+   LINXY(__NR_sched_getattr,     sys_sched_getattr),     // 275
242f5d
    LINX_(__NR_renameat2,         sys_renameat2),         // 276
242f5d
    //   (__NR_seccomp,           sys_ni_syscall),        // 277
242f5d
    LINXY(__NR_getrandom,         sys_getrandom),         // 278
242f5d
diff --git a/coregrind/m_syswrap/syswrap-linux.c b/coregrind/m_syswrap/syswrap-linux.c
242f5d
index 5b5b7eee6..56be3032d 100644
242f5d
--- a/coregrind/m_syswrap/syswrap-linux.c
242f5d
+++ b/coregrind/m_syswrap/syswrap-linux.c
242f5d
@@ -3677,6 +3677,41 @@ POST(sys_sched_getparam)
242f5d
    POST_MEM_WRITE( ARG2, sizeof(struct vki_sched_param) );
242f5d
 }
242f5d
 
242f5d
+PRE(sys_sched_setattr)
242f5d
+{
242f5d
+   struct vki_sched_attr *attr;
242f5d
+   PRINT("sched_setattr ( %ld, %#" FMT_REGWORD "x, %#"
242f5d
+         FMT_REGWORD "x )", SARG1, ARG2, ARG3 );
242f5d
+   PRE_REG_READ3(long, "sched_setattr",
242f5d
+                 vki_pid_t, pid, struct sched_attr *, p, unsigned int, flags);
242f5d
+   /* We need to be able to read at least the size field.  */
242f5d
+   PRE_MEM_READ( "sched_setattr(attr->size)", ARG2, sizeof(vki_uint32_t) );
242f5d
+   attr = (struct vki_sched_attr *)(Addr)ARG2;
242f5d
+   if (ML_(safe_to_deref)(attr,sizeof(vki_uint32_t)))
242f5d
+      PRE_MEM_READ( "sched_setattr(attr)", (Addr)attr, attr->size);
242f5d
+}
242f5d
+
242f5d
+PRE(sys_sched_getattr)
242f5d
+{
242f5d
+   struct vki_sched_attr *attr;
242f5d
+   PRINT("sched_getattr ( %ld, %#" FMT_REGWORD "x, %ld, %#"
242f5d
+         FMT_REGWORD "x )", SARG1, ARG2, ARG3, ARG4 );
242f5d
+   PRE_REG_READ4(long, "sched_getattr",
242f5d
+                 vki_pid_t, pid, struct sched_attr *, p,
242f5d
+                 unsigned int, size, unsigned int, flags);
242f5d
+   /* We need to be able to read at least the size field.  */
242f5d
+   PRE_MEM_READ( "sched_setattr(attr->size)", ARG2, sizeof(vki_uint32_t) );
242f5d
+   /* And the kernel needs to be able to write to the whole struct size. */
242f5d
+   attr = (struct vki_sched_attr *)(Addr)ARG2;
242f5d
+   if (ML_(safe_to_deref)(attr,sizeof(vki_uint32_t)))
242f5d
+      PRE_MEM_WRITE( "sched_setattr(attr)", (Addr)attr, attr->size);
242f5d
+}
242f5d
+POST(sys_sched_getattr)
242f5d
+{
242f5d
+   struct vki_sched_attr *attr = (struct vki_sched_attr *)(Addr)ARG2;
242f5d
+   POST_MEM_WRITE( (Addr)attr, attr->size );
242f5d
+}
242f5d
+
242f5d
 PRE(sys_sched_getscheduler)
242f5d
 {
242f5d
    PRINT("sys_sched_getscheduler ( %ld )", SARG1);
242f5d
diff --git a/coregrind/m_syswrap/syswrap-ppc32-linux.c b/coregrind/m_syswrap/syswrap-ppc32-linux.c
242f5d
index eed12a1bc..c19cb9e0e 100644
242f5d
--- a/coregrind/m_syswrap/syswrap-ppc32-linux.c
242f5d
+++ b/coregrind/m_syswrap/syswrap-ppc32-linux.c
242f5d
@@ -1016,6 +1016,9 @@ static SyscallTableEntry syscall_table[] = {
242f5d
    LINXY(__NR_process_vm_readv,  sys_process_vm_readv), // 351
242f5d
    LINX_(__NR_process_vm_writev, sys_process_vm_writev),// 352
242f5d
 
242f5d
+   LINX_(__NR_sched_setattr,     sys_sched_setattr),    // 355
242f5d
+   LINXY(__NR_sched_getattr,     sys_sched_getattr),    // 356
242f5d
+
242f5d
    LINXY(__NR_getrandom,         sys_getrandom),        // 359
242f5d
    LINXY(__NR_memfd_create,      sys_memfd_create),     // 360
242f5d
 
242f5d
diff --git a/coregrind/m_syswrap/syswrap-ppc64-linux.c b/coregrind/m_syswrap/syswrap-ppc64-linux.c
242f5d
index d58200b49..b6422a765 100644
242f5d
--- a/coregrind/m_syswrap/syswrap-ppc64-linux.c
242f5d
+++ b/coregrind/m_syswrap/syswrap-ppc64-linux.c
242f5d
@@ -998,6 +998,8 @@ static SyscallTableEntry syscall_table[] = {
242f5d
    LINXY(__NR_process_vm_readv,  sys_process_vm_readv), // 351
242f5d
    LINX_(__NR_process_vm_writev, sys_process_vm_writev),// 352
242f5d
 
242f5d
+   LINX_(__NR_sched_setattr,     sys_sched_setattr),    // 355
242f5d
+   LINXY(__NR_sched_getattr,     sys_sched_getattr),    // 356
242f5d
    LINX_(__NR_renameat2,         sys_renameat2),        // 357
242f5d
 
242f5d
    LINXY(__NR_getrandom,         sys_getrandom),        // 359
242f5d
diff --git a/coregrind/m_syswrap/syswrap-s390x-linux.c b/coregrind/m_syswrap/syswrap-s390x-linux.c
242f5d
index a0a330aa2..3427fee16 100644
242f5d
--- a/coregrind/m_syswrap/syswrap-s390x-linux.c
242f5d
+++ b/coregrind/m_syswrap/syswrap-s390x-linux.c
242f5d
@@ -825,8 +825,8 @@ static SyscallTableEntry syscall_table[] = {
242f5d
    LINX_(__NR_kcmp, sys_kcmp),                                        // 343
242f5d
 // ?????(__NR_finit_module, ),                                        // 344
242f5d
 
242f5d
-// ?????(__NR_sched_setattr, ),                                       // 345
242f5d
-// ?????(__NR_sched_getattr, ),                                       // 346
242f5d
+   LINX_(__NR_sched_setattr, sys_sched_setattr),                      // 345
242f5d
+   LINXY(__NR_sched_getattr, sys_sched_getattr),                      // 346
242f5d
    LINX_(__NR_renameat2, sys_renameat2),                              // 347
242f5d
 // ?????(__NR_seccomp, ),                                             // 348
242f5d
    LINXY(__NR_getrandom, sys_getrandom),                              // 349
242f5d
diff --git a/coregrind/m_syswrap/syswrap-x86-linux.c b/coregrind/m_syswrap/syswrap-x86-linux.c
242f5d
index 332ed0bf2..b59d96f37 100644
242f5d
--- a/coregrind/m_syswrap/syswrap-x86-linux.c
242f5d
+++ b/coregrind/m_syswrap/syswrap-x86-linux.c
242f5d
@@ -1580,8 +1580,8 @@ static SyscallTableEntry syscall_table[] = {
242f5d
    LINX_(__NR_kcmp,              sys_kcmp),             // 349
242f5d
 
242f5d
 //   LIN__(__NR_finit_module,      sys_ni_syscall),       // 350
242f5d
-//   LIN__(__NR_sched_setattr,     sys_ni_syscall),       // 351
242f5d
-//   LIN__(__NR_sched_getattr,     sys_ni_syscall),       // 352
242f5d
+   LINX_(__NR_sched_setattr,     sys_sched_setattr),    // 351
242f5d
+   LINXY(__NR_sched_getattr,     sys_sched_getattr),    // 352
242f5d
    LINX_(__NR_renameat2,         sys_renameat2),        // 353
242f5d
 //   LIN__(__NR_seccomp,           sys_ni_syscall),       // 354
242f5d
 
242f5d
diff --git a/include/vki/vki-linux.h b/include/vki/vki-linux.h
242f5d
index 75b583165..ef93b9258 100644
242f5d
--- a/include/vki/vki-linux.h
242f5d
+++ b/include/vki/vki-linux.h
242f5d
@@ -410,6 +410,23 @@ struct vki_sched_param {
242f5d
 	int sched_priority;
242f5d
 };
242f5d
 
242f5d
+struct vki_sched_attr {
242f5d
+	vki_uint32_t size;
242f5d
+	vki_uint32_t sched_policy;
242f5d
+	vki_uint64_t sched_flags;
242f5d
+
242f5d
+	/* SCHED_NORMAL, SCHED_BATCH */
242f5d
+	vki_int32_t sched_nice;
242f5d
+
242f5d
+	/* SCHED_FIFO, SCHED_RR */
242f5d
+	vki_uint32_t sched_priority;
242f5d
+
242f5d
+	/* SCHED_DEADLINE */
242f5d
+	vki_uint64_t sched_runtime;
242f5d
+	vki_uint64_t sched_deadline;
242f5d
+	vki_uint64_t sched_period;
242f5d
+};
242f5d
+
242f5d
 #define VKI_TASK_COMM_LEN 16
242f5d
 
242f5d
 //----------------------------------------------------------------------