/*
- * Copyright (c) 2015 Dmitry V. Levin <ldv@altlinux.org>
+ * Copyright (c) 2015-2017 Dmitry V. Levin <ldv@altlinux.org>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
-#ifdef HAVE_CONFIG_H
-# include "config.h"
-#endif
-
-#include <stdio.h>
-#include <inttypes.h>
-#include <unistd.h>
-#include <sys/syscall.h>
+#include "tests.h"
+#include <asm/unistd.h>
#if defined __NR_sched_getattr && defined __NR_sched_setattr
+# include <inttypes.h>
+# include <stdio.h>
+# include <sched.h>
+# include <unistd.h>
+# include "sched_attr.h"
+# include "xlat.h"
+# include "xlat/schedulers.h"
+
+static const char *errstr;
+
+static long
+sys_sched_getattr(kernel_ulong_t pid, kernel_ulong_t attr,
+ kernel_ulong_t size, kernel_ulong_t flags)
+{
+ long rc = syscall(__NR_sched_getattr, pid, attr, size, flags);
+ errstr = sprintrc(rc);
+ return rc;
+}
+
+static long
+sys_sched_setattr(kernel_ulong_t pid, kernel_ulong_t attr, kernel_ulong_t flags)
+{
+ long rc = syscall(__NR_sched_setattr, pid, attr, flags);
+ errstr = sprintrc(rc);
+ return rc;
+}
+
int
main(void)
{
- static union {
- 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;
- char buf[256];
- } sched;
-
- if (syscall(__NR_sched_getattr, 0, &sched, sizeof(sched), 0))
- return 77;
-
- printf("sched_getattr\\(0, \\{size=%u, sched_policy=SCHED_[A-Z]+, sched_flags=%s, sched_nice=%u, sched_priority=%u, sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64 ", sched_period=%" PRIu64 "\\}, 256, 0\\) += 0\n",
- sched.attr.size,
- sched.attr.sched_flags ? "SCHED_FLAG_RESET_ON_FORK" : "0",
- sched.attr.sched_nice,
- sched.attr.sched_priority,
- sched.attr.sched_runtime,
- sched.attr.sched_deadline,
- sched.attr.sched_period);
-
- sched.attr.sched_flags |= 1;
- if (syscall(__NR_sched_setattr, 0, &sched, 0))
- return 77;
-
- printf("sched_setattr\\(0, \\{size=%u, sched_policy=SCHED_[A-Z]+, sched_flags=%s, sched_nice=%u, sched_priority=%u, sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64 ", sched_period=%" PRIu64 "\\}, 0\\) += 0\n",
- sched.attr.size,
- "SCHED_FLAG_RESET_ON_FORK",
- sched.attr.sched_nice,
- sched.attr.sched_priority,
- sched.attr.sched_runtime,
- sched.attr.sched_deadline,
- sched.attr.sched_period);
+ static const kernel_ulong_t bogus_pid =
+ (kernel_ulong_t) 0xdefacedfacefeedULL;
+ static const kernel_ulong_t bogus_size =
+ (kernel_ulong_t) 0xdefacedcafef00dULL;
+ static const kernel_ulong_t bogus_flags =
+ (kernel_ulong_t) 0xdefaceddeadc0deULL;
+
+ TAIL_ALLOC_OBJECT_CONST_PTR(struct sched_attr, attr);
+ TAIL_ALLOC_OBJECT_CONST_PTR(unsigned int, psize);
+ void *const efault = attr + 1;
+
+ sys_sched_getattr(0, 0, 0, 0);
+ printf("sched_getattr(0, NULL, 0, 0) = %s\n", errstr);
+
+ sys_sched_getattr(0, (unsigned long) attr, 0, 0);
+ printf("sched_getattr(0, %p, 0, 0) = %s\n", attr, errstr);
+
+ sys_sched_getattr(bogus_pid, 0, 0, 0);
+ printf("sched_getattr(%d, NULL, 0, 0) = %s\n", (int) bogus_pid, errstr);
+
+ sys_sched_getattr(-1U, (unsigned long) attr, bogus_size, bogus_flags);
+ printf("sched_getattr(-1, %p, %s%u, %u) = %s\n",
+ attr,
+# if defined __arm64__ || defined __aarch64__
+ "0xdefaced<<32|",
+# else
+ "",
+# endif
+ (unsigned) bogus_size, (unsigned) bogus_flags, errstr);
+
+ sys_sched_getattr(0, (unsigned long) efault, sizeof(*attr), 0);
+ printf("sched_getattr(0, %p, %u, 0) = %s\n",
+ efault, (unsigned) sizeof(*attr), errstr);
+
+ if (sys_sched_getattr(0, (unsigned long) attr, sizeof(*attr), 0))
+ perror_msg_and_skip("sched_getattr");
+ printf("sched_getattr(0, {size=%u, sched_policy=", attr->size);
+ printxval(schedulers, attr->sched_policy, NULL);
+ printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
+ ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
+ ", sched_period=%" PRIu64 "}, %u, 0) = 0\n",
+ attr->sched_flags ? "SCHED_FLAG_RESET_ON_FORK" : "0",
+ attr->sched_nice,
+ attr->sched_priority,
+ attr->sched_runtime,
+ attr->sched_deadline,
+ attr->sched_period,
+ (unsigned) sizeof(*attr));
+
+# if defined __arm64__ || defined __aarch64__
+ long rc =
+# endif
+ sys_sched_getattr(F8ILL_KULONG_MASK, (unsigned long) attr,
+ F8ILL_KULONG_MASK | sizeof(*attr), F8ILL_KULONG_MASK);
+# if defined __arm64__ || defined __aarch64__
+ if (rc) {
+ printf("sched_getattr(0, %p, 0xffffffff<<32|%u, 0) = %s\n",
+ attr, (unsigned) sizeof(*attr), errstr);
+ } else
+# endif
+ {
+ printf("sched_getattr(0, {size=%u, sched_policy=", attr->size);
+ printxval(schedulers, attr->sched_policy, NULL);
+ printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
+ ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
+ ", sched_period=%" PRIu64 "}, %u, 0) = 0\n",
+ attr->sched_flags ? "SCHED_FLAG_RESET_ON_FORK" : "0",
+ attr->sched_nice,
+ attr->sched_priority,
+ attr->sched_runtime,
+ attr->sched_deadline,
+ attr->sched_period,
+ (unsigned) sizeof(*attr));
+ }
+
+ sys_sched_setattr(bogus_pid, 0, 0);
+ printf("sched_setattr(%d, NULL, 0) = %s\n", (int) bogus_pid, errstr);
+
+ attr->sched_flags |= 1;
+ if (sys_sched_setattr(0, (unsigned long) attr, 0))
+ perror_msg_and_skip("sched_setattr");
+ printf("sched_setattr(0, {size=%u, sched_policy=", attr->size);
+ printxval(schedulers, attr->sched_policy, NULL);
+ printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
+ ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
+ ", sched_period=%" PRIu64 "}, 0) = 0\n",
+ "SCHED_FLAG_RESET_ON_FORK",
+ attr->sched_nice,
+ attr->sched_priority,
+ attr->sched_runtime,
+ attr->sched_deadline,
+ attr->sched_period);
+
+ sys_sched_setattr(F8ILL_KULONG_MASK, (unsigned long) attr,
+ F8ILL_KULONG_MASK);
+ printf("sched_setattr(0, {size=%u, sched_policy=", attr->size);
+ printxval(schedulers, attr->sched_policy, NULL);
+ printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
+ ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
+ ", sched_period=%" PRIu64 "}, 0) = 0\n",
+ "SCHED_FLAG_RESET_ON_FORK",
+ attr->sched_nice,
+ attr->sched_priority,
+ attr->sched_runtime,
+ attr->sched_deadline,
+ attr->sched_period);
+
+ *psize = attr->size;
+
+ sys_sched_setattr(0, (unsigned long) psize, 0);
+ printf("sched_setattr(0, %p, 0) = %s\n", psize, errstr);
+
+ attr->size = 0;
+
+ sys_sched_setattr(0, (unsigned long) attr, 0);
+ printf("sched_setattr(0, {size=%u, sched_policy=", attr->size);
+ printxval(schedulers, attr->sched_policy, NULL);
+ printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
+ ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
+ ", sched_period=%" PRIu64 "}, 0) = 0\n",
+ "SCHED_FLAG_RESET_ON_FORK",
+ attr->sched_nice,
+ attr->sched_priority,
+ attr->sched_runtime,
+ attr->sched_deadline,
+ attr->sched_period);
+
+ attr->size = 1;
+
+ sys_sched_setattr(0, (unsigned long) attr, 0);
+ printf("sched_setattr(0, {size=%u} => {size=%u}, 0) = %s\n",
+ 1, attr->size, errstr);
+
+ attr->size = SCHED_ATTR_MIN_SIZE - 1;
+
+ sys_sched_setattr(0, (unsigned long) attr, 0);
+ printf("sched_setattr(0, {size=%u} => {size=%u}, 0) = %s\n",
+ SCHED_ATTR_MIN_SIZE - 1, attr->size, errstr);
+
+ attr->size = 0x90807060;
+ attr->sched_policy = 0xca7faced;
+ attr->sched_flags = 0xbadc0ded1057da7aULL;
+ attr->sched_nice = 0xafbfcfdf;
+ attr->sched_priority = 0xb8c8d8e8;
+ attr->sched_runtime = 0xbadcaffedeadf157ULL;
+ attr->sched_deadline = 0xc0de70a57badac75ULL;
+ attr->sched_period = 0xded1ca7edda7aca7ULL;
+
+ sys_sched_setattr(bogus_pid, (unsigned long) attr, bogus_flags);
+ printf("sched_setattr(%d, {size=%u, sched_policy=%#x /* SCHED_??? */, "
+ "sched_flags=%#" PRIx64 " /* SCHED_FLAG_??? */, "
+ "sched_nice=%d, sched_priority=%u, sched_runtime=%" PRIu64 ", "
+ "sched_deadline=%" PRIu64 ", sched_period=%" PRIu64 ", ...}, %u)"
+ " = %s\n",
+ (int) bogus_pid,
+ attr->size,
+ attr->sched_policy,
+ attr->sched_flags,
+ attr->sched_nice,
+ attr->sched_priority,
+ attr->sched_runtime,
+ attr->sched_deadline,
+ attr->sched_period,
+ (unsigned) bogus_flags, errstr);
+
+ if (F8ILL_KULONG_SUPPORTED) {
+ const kernel_ulong_t ill = f8ill_ptr_to_kulong(attr);
+
+ sys_sched_getattr(0, ill, sizeof(*attr), 0);
+ printf("sched_getattr(0, %#llx, %u, 0) = %s\n",
+ (unsigned long long) ill, (unsigned) sizeof(*attr),
+ errstr);
+
+ sys_sched_setattr(0, ill, 0);
+ printf("sched_setattr(0, %#llx, 0) = %s\n",
+ (unsigned long long) ill, errstr);
+ }
+
+ puts("+++ exited with 0 +++");
return 0;
}
#else
-int
-main(void)
-{
- return 77;
-}
+SKIP_MAIN_UNDEFINED("__NR_sched_getattr && __NR_sched_setattr")
#endif