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 && errno == ENOSYS)
  35                 {
  36                   puts ("getcpu syscall not implemented");
  37                   return 0;
  38                 }
  39               if (cpu2 != cpu)
  40                 {
  41                   printf ("getcpu results %d not possible\n", cpu2);
  42                   result = 1;
  43                 }
  44             }
  45           CPU_CLR (cpu, &cs);
  46         }
  47       ++cpu;
  48     }
  49
  50   return result;
  51 }
  52
  53 #define TEST_FUNCTION do_test ()
  54 #include <test-skeleton.c>