sysdeps/unix/sysv/linux/tst-getcpu.c

   1 #include <errno.h>
   2 #include <stdio.h>
   3 #include <sched.h>
   4 #include <unistd.h>
   5
   6
   7 static int
   8 do_test (void)
   9 {
  10   cpu_set_t cs;
  11   if (sched_getaffinity (getpid (), sizeof (cs), &cs) != 0)
  12     {
  13       printf ("getaffinity failed: %m\n");
  14       return 1;
  15     }
  16
  17   int result = 0;
  18   int cpu = 0;
  19   while (CPU_COUNT (&cs) != 0)
  20     {
  21       if (CPU_ISSET (cpu, &cs))
  22         {
  23           cpu_set_t cs2;
  24           CPU_ZERO (&cs2);
  25           CPU_SET (cpu, &cs2);
  26           if (sched_setaffinity (getpid (), sizeof (cs2), &cs2) != 0)
  27             {
  28               printf ("setaffinity(%d) failed: %m\n", cpu);
  29               result = 1;
  30             }
  31           else
  32             {
  33               int cpu2 = sched_getcpu ();
  34               if (cpu2 == -1)
  35                 {
  36                   if (errno == ENOSYS)
  37                     {
  38                       puts ("getcpu syscall not implemented");
  39                       return 0;
  40                     }
  41                   perror ("getcpu failed");
  42                   result = 1;
  43                 }
  44               if (cpu2 != cpu)
  45                 {
  46                   printf ("getcpu results %d not possible\n", cpu2);
  47                   result = 1;
  48                 }
  49             }
  50           CPU_CLR (cpu, &cs);
  51         }
  52       ++cpu;
  53     }
  54
  55   return result;
  56 }
  57
  58 #define TEST_FUNCTION do_test ()
  59 #include <test-skeleton.c>