Robustify mpers.awk against invalid input

[strace] / sched.c

diff --git a/sched.c b/sched.c
index 965013c611acf2fabef2fe396bc1478bc4315be9..e4730469a7f6d92ff4dbf150cfde645dc8fd6fcd 100644 (file)
--- a/sched.c
+++ b/sched.c
@@ -3,6 +3,7 @@
 #include <sched.h>
 
 #include "xlat/schedulers.h"
+#include "xlat/sched_flags.h"
 
 SYS_FUNC(sched_getscheduler)
 {
@@ -18,13 +19,12 @@ SYS_FUNC(sched_getscheduler)
 
 SYS_FUNC(sched_setscheduler)
 {
-       if (entering(tcp)) {
-               tprintf("%d, ", (int) tcp->u_arg[0]);
-               printxval(schedulers, tcp->u_arg[1], "SCHED_???");
-               tprints(", ");
-               printnum_int(tcp, tcp->u_arg[2], "%d");
-       }
-       return 0;
+       tprintf("%d, ", (int) tcp->u_arg[0]);
+       printxval(schedulers, tcp->u_arg[1], "SCHED_???");
+       tprints(", ");
+       printnum_int(tcp, tcp->u_arg[2], "%d");
+
+       return RVAL_DECODED;
 }
 
 SYS_FUNC(sched_getparam)
@@ -38,19 +38,17 @@ SYS_FUNC(sched_getparam)
 
 SYS_FUNC(sched_setparam)
 {
-       if (entering(tcp)) {
-               tprintf("%d, ", (int) tcp->u_arg[0]);
-               printnum_int(tcp, tcp->u_arg[1], "%d");
-       }
-       return 0;
+       tprintf("%d, ", (int) tcp->u_arg[0]);
+       printnum_int(tcp, tcp->u_arg[1], "%d");
+
+       return RVAL_DECODED;
 }
 
 SYS_FUNC(sched_get_priority_min)
 {
-       if (entering(tcp)) {
-               printxval(schedulers, tcp->u_arg[0], "SCHED_???");
-       }
-       return 0;
+       printxval(schedulers, tcp->u_arg[0], "SCHED_???");
+
+       return RVAL_DECODED;
 }
 
 SYS_FUNC(sched_rr_get_interval)
@@ -65,3 +63,56 @@ SYS_FUNC(sched_rr_get_interval)
        }
        return 0;
 }
+
+static void
+print_sched_attr(struct tcb *tcp, const long addr, unsigned int size)
+{
+       struct {
+               uint32_t size;
+               uint32_t sched_policy;
+               uint64_t sched_flags;
+               uint32_t sched_nice;
+               uint32_t sched_priority;
+               uint64_t sched_runtime;
+               uint64_t sched_deadline;
+               uint64_t sched_period;
+       } attr = {};
+
+       if (size > sizeof(attr))
+               size = sizeof(attr);
+       if (umoven_or_printaddr(tcp, addr, size, &attr))
+               return;
+
+       tprintf("{size=%u, sched_policy=", attr.size);
+       printxval(schedulers, attr.sched_policy, "SCHED_???");
+       tprints(", sched_flags=");
+       printflags(sched_flags, attr.sched_flags, "SCHED_FLAG_???");
+       tprintf(", sched_nice=%d", attr.sched_nice);
+       tprintf(", sched_priority=%u", attr.sched_priority);
+       tprintf(", sched_runtime=%" PRIu64, attr.sched_runtime);
+       tprintf(", sched_deadline=%" PRIu64, attr.sched_deadline);
+       tprintf(", sched_period=%" PRIu64 "}", attr.sched_period);
+}
+
+SYS_FUNC(sched_setattr)
+{
+       tprintf("%d, ", (int) tcp->u_arg[0]);
+       print_sched_attr(tcp, tcp->u_arg[1], 0x100);
+       tprintf(", %u", (unsigned int) tcp->u_arg[2]);
+
+       return RVAL_DECODED;
+}
+
+SYS_FUNC(sched_getattr)
+{
+       if (entering(tcp)) {
+               tprintf("%d, ", (int) tcp->u_arg[0]);
+       } else {
+               print_sched_attr(tcp, tcp->u_arg[1], tcp->u_arg[2]);
+               tprintf(", %u, %u",
+                       (unsigned int) tcp->u_arg[2],
+                       (unsigned int) tcp->u_arg[3]);
+       }
+
+       return 0;
+}