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

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