git.gag.com Git - fw/altos/blob - ao-bringup/test-igniter

   1 #!/usr/bin/nickle
   2
   3 import File;
   4
   5 string timed_read(file f, int timeout) {
   6         thread reader = fork func() {
   7                 try {
   8                         return fgets(f);
   9                 } catch Thread::signal(int i) {
  10                         return "";
  11                 }
  12         }();
  13
  14         thread killer = fork func() {
  15                 try {
  16                         sleep (timeout);
  17                         Thread::send_signal(reader, 1);
  18                 } catch Thread::signal(int i) {
  19                         return;
  20                 }
  21         }();
  22
  23         poly v = Thread::join(reader);
  24         Thread::send_signal(killer, 1);
  25         Thread::join(killer);
  26         if (is_string(v))
  27                 return v;
  28         return "";
  29 }
  30
  31 void flush_input(file f) {
  32         for (;;) {
  33                 string s = timed_read(f, 200);
  34                 if (s == "")
  35                         break;
  36         }
  37 }
  38
  39 string[*] settings(file f) {
  40         string[...] x = {};
  41
  42         flush_input(f);
  43         fprintf (f, "c s\nv\n");
  44         flush(f);
  45         for (;;) {
  46                 string l = File::fgets(f);
  47                 x[dim(x)] = l;
  48                 if (String::index(l, "software-version") == 0)
  49                         break;
  50         }
  51         return x;
  52 }
  53
  54 string[*] find_setting(string[*] s, string match) {
  55         for (int i = 0; i < dim(s); i++)
  56                 if (String::index(s[i], match) == 0)
  57                         return String::wordsplit(s[i], " ");
  58         return (string[*]) {};
  59 }
  60
  61 string[*] igniters(file f) {
  62         string[...] x = {};
  63
  64         flush_input(f);
  65         fprintf (f, "t\nv\n");
  66         flush(f);
  67         for (;;) {
  68                 string l = File::fgets(f);
  69                 x[dim(x)] = l;
  70                 if (String::index(l, "software-version") == 0)
  71                         break;
  72         }
  73         return x;
  74 }
  75
  76 string[*] find_igniter(string[*] s, string match) {
  77         for (int i = 0; i < dim(s); i++)
  78                 if (String::index(s[i], match) >= 0)
  79                         return String::wordsplit(s[i], " ");
  80         return (string[*]) {};
  81 }
  82
  83 bool
  84 do_igniter(file f, string igniter) {
  85         string[*] i = igniters(f);
  86         string[*] status = find_igniter(i, igniter);
  87         if (dim(status) < 4) {
  88                 printf ("no igniter %s found in %v\n", igniter, i);
  89                 return false;
  90         }
  91         if (String::index(status[3], "ready") < 0) {
  92                 printf("igniter %s status is \"%s\"\n", igniter, status[3]);
  93                 return false;
  94         }
  95         fprintf(f, "i DoIt %s\n", igniter);
  96         flush(f);
  97         flush_input(f);
  98         return true;
  99 }
 100
 101 void main () {
 102         string  name = argv[1];
 103         string[dim(argv)-2]     igniters = { [i] = argv[i+2] };
 104         file    f = open(name, "r+");
 105         bool ret = true;
 106
 107         for (int i = 0; i < dim(igniters); i++) {
 108                 if (!do_igniter(f, igniters[i]))
 109                         ret = false;
 110         }
 111         exit (ret? 0 : 1);
 112 }
 113
 114 main();