tests/cpusetrange.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include <stdbool.h>
   4 #include <stdlib.h>
   5
   6 /*
   7  * cpusets are in format "1,2-3,4"
   8  * iow, comma-delimited ranges
   9  */
  10 extern bool cpu_in_cpuset(int cpu, const char *cpuset);
  11
  12 void verify(bool condition) {
  13         if (condition) {
  14                 printf(" PASS\n");
  15         } else {
  16                 printf(" FAIL!\n");
  17                 exit(1);
  18         }
  19 }
  20
  21 int main() {
  22         char *a = "1,2";
  23         char *b = "1-3,5";
  24         char *c = "1,4-5";
  25         char *d = "";
  26         char *e = "\n";
  27
  28         printf("1 in %s", a);
  29         verify(cpu_in_cpuset(1, a));
  30         printf("2 in %s", a);
  31         verify(cpu_in_cpuset(2, a));
  32         printf("NOT 4 in %s", a);
  33         verify(!cpu_in_cpuset(4, a));
  34         printf("1 in %s", b);
  35         verify(cpu_in_cpuset(1, b));
  36         printf("NOT 4 in %s", b);
  37         verify(!cpu_in_cpuset(4, b));
  38         printf("5 in %s", b);
  39         verify(cpu_in_cpuset(5, b));
  40         printf("1 in %s", c);
  41         verify(cpu_in_cpuset(1, c));
  42         printf("5 in %s", c);
  43         verify(cpu_in_cpuset(5, c));
  44         printf("NOT 6 in %s", c);
  45         verify(!cpu_in_cpuset(6, c));
  46         printf("NOT 6 in empty set");
  47         verify(!cpu_in_cpuset(6, d));
  48         printf("NOT 6 in empty set(2)");
  49         verify(!cpu_in_cpuset(6, e));
  50 }