granicus.if.org Git - strace/blob - tests/sched_xetaffinity.c

   1 /*
   2  * This file is part of sched_xetaffinity strace test.
   3  *
   4  * Copyright (c) 2016 Dmitry V. Levin <ldv@altlinux.org>
   5  * All rights reserved.
   6  *
   7  * SPDX-License-Identifier: GPL-2.0-or-later
   8  */
   9
  10 #include "tests.h"
  11 #include <asm/unistd.h>
  12 #include <sched.h>
  13
  14 #if defined __NR_sched_getaffinity && defined __NR_sched_setaffinity \
  15  && defined CPU_ISSET_S && defined CPU_ZERO_S && defined CPU_SET_S
  16
  17 # include <assert.h>
  18 # include <errno.h>
  19 # include <stdio.h>
  20 # include <unistd.h>
  21
  22 static const char *errstr;
  23
  24 static int
  25 getaffinity(unsigned long pid, unsigned long size, void *set)
  26 {
  27         int rc = syscall(__NR_sched_getaffinity, pid, size, set);
  28         errstr = sprintrc(rc);
  29         return rc;
  30 }
  31
  32 static int
  33 setaffinity(unsigned long pid, unsigned long size, void *set)
  34 {
  35         int rc = syscall(__NR_sched_setaffinity, pid, size, set);
  36         errstr = sprintrc(rc);
  37         return rc;
  38 }
  39
  40 int
  41 main(void)
  42 {
  43         unsigned int cpuset_size = 1;
  44         const pid_t pid = getpid();
  45
  46         while (cpuset_size) {
  47                 assert(getaffinity(pid, cpuset_size, NULL) == -1);
  48                 if (EFAULT == errno)
  49                         break;
  50                 if (EINVAL != errno)
  51                         perror_msg_and_skip("sched_getaffinity");
  52                 printf("sched_getaffinity(%d, %u, NULL) = %s\n",
  53                        pid, cpuset_size, errstr);
  54                 cpuset_size <<= 1;
  55         }
  56         assert(cpuset_size);
  57         printf("sched_getaffinity(%d, %u, NULL) = %s\n",
  58                pid, cpuset_size, errstr);
  59
  60         cpu_set_t *cpuset = tail_alloc(cpuset_size);
  61         getaffinity(pid, cpuset_size, cpuset + 1);
  62         printf("sched_getaffinity(%d, %u, %p) = %s\n",
  63                pid, cpuset_size, cpuset + 1, errstr);
  64
  65         int ret_size = getaffinity(pid, cpuset_size, cpuset);
  66         if (ret_size < 0)
  67                 perror_msg_and_fail("sched_getaffinity(%d, %u, %p) = %s\n",
  68                                     pid, (unsigned) cpuset_size, cpuset, errstr);
  69         assert(ret_size <= (int) cpuset_size);
  70
  71         printf("sched_getaffinity(%d, %u, [", pid, cpuset_size);
  72         const char *sep;
  73         unsigned int i, cpu;
  74         for (i = 0, cpu = 0, sep = ""; i < (unsigned) ret_size * 8; ++i) {
  75                 if (CPU_ISSET_S(i, (unsigned) ret_size, cpuset)) {
  76                         printf("%s%u", sep, i);
  77                         sep = ", ";
  78                         cpu = i;
  79                 }
  80         }
  81         printf("]) = %s\n", errstr);
  82
  83         CPU_ZERO_S(cpuset_size, cpuset);
  84         CPU_SET_S(cpu, cpuset_size, cpuset);
  85         if (setaffinity(pid, cpuset_size, cpuset))
  86                 perror_msg_and_skip("sched_setaffinity");
  87         printf("sched_setaffinity(%d, %u, [%u]) = 0\n",
  88                pid, cpuset_size, cpu);
  89
  90         const unsigned int big_size = cpuset_size < 128 ? 128 : cpuset_size * 2;
  91         cpuset = tail_alloc(big_size);
  92         ret_size = getaffinity(pid, big_size, cpuset);
  93         if (ret_size < 0)
  94                 perror_msg_and_fail("sched_getaffinity(%d, %u, %p) = %s\n",
  95                                     pid, big_size, cpuset, errstr);
  96         assert(ret_size <= (int) big_size);
  97         printf("sched_getaffinity(%d, %u, [", pid, big_size);
  98         for (i = 0, sep = ""; i < (unsigned) ret_size * 8; ++i) {
  99                 if (CPU_ISSET_S(i, (unsigned) ret_size, cpuset)) {
 100                         printf("%s%u", sep, i);
 101                         sep = ", ";
 102                 }
 103         }
 104         printf("]) = %s\n", errstr);
 105
 106         puts("+++ exited with 0 +++");
 107         return 0;
 108 }
 109
 110 #else
 111
 112 SKIP_MAIN_UNDEFINED("__NR_sched_getaffinity && __NR_sched_setaffinity"
 113                     " && CPU_ISSET_S && CPU_ZERO_S && CPU_SET_S")
 114
 115 #endif