host/main.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <stdint.h>
   5 #include <sys/types.h>
   6 #include <sys/stat.h>
   7 #include <fcntl.h>
   8 #include <sys/ioctl.h>
   9 #include <sys/select.h>
  10 #include <unistd.h>
  11 #include <dirent.h>
  12 #include <signal.h>
  13 #include <termios.h>
  14 #include <errno.h>
  15 #include <linux/usbdevice_fs.h>
  16 #include <linux/usb/ch9.h>
  17 #include <linux/input.h>
  18 #include "../pkts.h"
  19
  20 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
  21
  22 struct teensy_dev {
  23   int fd;
  24   struct {
  25     int ep_in;
  26     int ep_out;
  27   } ifaces[2];
  28 };
  29
  30 /* return 1 if founf, 0 if not, < 0 on error */
  31 static int find_device(struct teensy_dev *dev,
  32   uint16_t vendor, uint16_t product)
  33 {
  34   const char path_root[] = "/dev/bus/usb";
  35   union {
  36     struct usb_descriptor_header hdr;
  37     struct usb_device_descriptor d;
  38     struct usb_config_descriptor c;
  39     struct usb_interface_descriptor i;
  40     struct usb_endpoint_descriptor e;
  41     char space[0x100]; /* enough? */
  42   } desc;
  43   char path_bus[256], path_dev[256];
  44   struct dirent *ent, *ent_bus;
  45   DIR *dir = NULL, *dir_bus = NULL;
  46   int num, fd = -1;
  47   int iface = -1;
  48   int retval = -1;
  49   int ret;
  50
  51   memset(dev, 0xff, sizeof(*dev));
  52
  53   dir = opendir(path_root);
  54   if (dir == NULL) {
  55     perror("opendir");
  56     return -1;
  57   }
  58
  59   for (ent = readdir(dir); ent != NULL; ent = readdir(dir)) {
  60     /* should be a number like 000 */
  61     if (sscanf(ent->d_name, "%03d", &num) != 1)
  62       continue;
  63
  64     snprintf(path_bus, sizeof(path_bus), "%s/%s",
  65         path_root, ent->d_name);
  66
  67     dir_bus = opendir(path_bus);
  68     if (dir_bus == NULL)
  69       continue;
  70
  71     ent_bus = readdir(dir_bus);
  72     for (; ent_bus != NULL; ent_bus = readdir(dir_bus)) {
  73       if (sscanf(ent->d_name, "%03d", &num) != 1)
  74         continue;
  75
  76       snprintf(path_dev, sizeof(path_dev), "%s/%s/%s",
  77           path_root, ent->d_name, ent_bus->d_name);
  78
  79       fd = open(path_dev, O_RDWR);
  80       if (fd == -1)
  81         continue;
  82
  83       ret = read(fd, &desc.d, sizeof(desc.d));
  84       if (ret != sizeof(desc.d)) {
  85         fprintf(stderr, "desc read: %d/%zd: ", ret, sizeof(desc.d));
  86         perror("");
  87         goto next;
  88       }
  89
  90       if (desc.d.bDescriptorType != USB_DT_DEVICE) {
  91         fprintf(stderr, "%s: bad DT: 0x%02x\n",
  92             path_dev, desc.d.bDescriptorType);
  93         goto next;
  94       }
  95
  96       if (desc.d.idVendor == vendor && desc.d.idProduct == product)
  97         goto found;
  98
  99 next:
 100       close(fd);
 101       fd = -1;
 102     }
 103
 104     closedir(dir_bus);
 105     dir_bus = NULL;
 106   }
 107
 108   /* not found */
 109   retval = 0;
 110   goto out;
 111
 112 found:
 113   if (desc.d.bNumConfigurations != 1) {
 114     fprintf(stderr, "unexpected bNumConfigurations: %u\n",
 115         desc.d.bNumConfigurations);
 116     goto out;
 117   }
 118
 119   /* walk through all descriptors */
 120   while (1)
 121   {
 122     ret = read(fd, &desc.hdr, sizeof(desc.hdr));
 123     if (ret == 0)
 124       break;
 125     if (ret != sizeof(desc.hdr)) {
 126       fprintf(stderr, "desc.hdr read: %d/%zd: ", ret, sizeof(desc.hdr));
 127       perror("");
 128       break;
 129     }
 130
 131     ret = (int)lseek(fd, -sizeof(desc.hdr), SEEK_CUR);
 132     if (ret == -1) {
 133       perror("lseek");
 134       break;
 135     }
 136
 137     ret = read(fd, &desc, desc.hdr.bLength);
 138     if (ret != desc.hdr.bLength) {
 139       fprintf(stderr, "desc read: %d/%u: ", ret, desc.hdr.bLength);
 140       perror("");
 141       break;
 142     }
 143
 144     switch (desc.hdr.bDescriptorType) {
 145       case USB_DT_CONFIG:
 146         if (desc.c.bNumInterfaces != 2) {
 147           fprintf(stderr, "unexpected bNumInterfaces: %u\n",
 148               desc.c.bNumInterfaces);
 149           goto out;
 150         }
 151         break;
 152
 153       case USB_DT_INTERFACE:
 154         if (desc.i.bInterfaceClass != USB_CLASS_HID
 155             || desc.i.bInterfaceSubClass != 0
 156             || desc.i.bInterfaceProtocol != 0) {
 157           fprintf(stderr, "unexpected interface %x:%x:%x\n",
 158             desc.i.bInterfaceClass, desc.i.bInterfaceSubClass,
 159             desc.i.bInterfaceProtocol);
 160           goto out;
 161         }
 162         if (desc.i.bNumEndpoints != 2) {
 163           fprintf(stderr, "unexpected bNumEndpoints: %u\n",
 164             desc.i.bNumEndpoints);
 165           goto out;
 166         }
 167         iface++;
 168         break;
 169
 170       case USB_DT_ENDPOINT:
 171         if (iface < 0 || iface >= ARRAY_SIZE(dev->ifaces)) {
 172           fprintf(stderr, "bad iface: %d\n", iface);
 173           goto out;
 174         }
 175         if (desc.e.wMaxPacketSize != 64 && desc.e.wMaxPacketSize != 32) {
 176           fprintf(stderr, "iface %d, EP %02x: "
 177             "unexpected wMaxPacketSize: %u\n",
 178             iface, desc.e.bEndpointAddress, desc.e.wMaxPacketSize);
 179           goto out;
 180         }
 181         if (desc.e.bEndpointAddress & 0x80)
 182           dev->ifaces[iface].ep_in = desc.e.bEndpointAddress; // & 0x7F;
 183         else
 184           dev->ifaces[iface].ep_out = desc.e.bEndpointAddress;
 185         break;
 186
 187       case 0x21:
 188         /* ignore */
 189         break;
 190
 191       default:
 192         fprintf(stderr, "skipping desc 0x%02x\n",
 193           desc.hdr.bDescriptorType);
 194         break;
 195     }
 196   }
 197
 198   /* claim interfaces */
 199   for (iface = 0; iface < ARRAY_SIZE(dev->ifaces); iface++) {
 200     struct usbdevfs_ioctl usbio;
 201
 202     if (dev->ifaces[iface].ep_in == -1) {
 203       fprintf(stderr, "missing ep_in, iface: %d\n", iface);
 204       goto out;
 205     }
 206     if (dev->ifaces[iface].ep_out == -1) {
 207       fprintf(stderr, "missing ep_out, iface: %d\n", iface);
 208       goto out;
 209     }
 210
 211     /* disconnect default driver */
 212     memset(&usbio, 0, sizeof(usbio));
 213     usbio.ifno = iface;
 214     usbio.ioctl_code = USBDEVFS_DISCONNECT;
 215     ret = ioctl(fd, USBDEVFS_IOCTL, &usbio);
 216     if (ret != 0 && errno != ENODATA)
 217       perror("USBDEVFS_DISCONNECT");
 218
 219     ret = ioctl(fd, USBDEVFS_CLAIMINTERFACE, &iface);
 220     if (ret != 0)
 221       perror("USBDEVFS_CLAIMINTERFACE");
 222   }
 223
 224   dev->fd = fd;
 225   fd = -1;
 226   retval = 1;
 227
 228 out:
 229   if (fd != -1)
 230     close(fd);
 231   if (dir_bus != NULL)
 232     closedir(dir_bus);
 233   if (dir != NULL)
 234     closedir(dir);
 235
 236   return retval;
 237 }
 238
 239 static int enable_echo(int enable)
 240 {
 241   const char *portname = "/dev/tty";
 242   struct termios tty;
 243   int retval = -1;
 244   int ret;
 245   int fd;
 246
 247   memset(&tty, 0, sizeof(tty));
 248
 249   fd = open(portname, O_RDWR | O_NOCTTY | O_SYNC);
 250   if (fd < 0) {
 251     fprintf(stderr, "open %s: ", portname);
 252     perror("");
 253     return 1;
 254   }
 255
 256   ret = tcgetattr(fd, &tty);
 257   if (ret != 0) {
 258     perror("tcgetattr");
 259     goto out;
 260   }
 261
 262   // printf("lflag: 0%o\n", tty.c_lflag);
 263   if (enable)
 264     tty.c_lflag |= ECHO | ICANON;
 265   else {
 266     tty.c_lflag &= ~(ECHO | ICANON);
 267     tty.c_cc[VMIN] = tty.c_cc[VTIME] = 0;
 268   }
 269
 270   ret = tcsetattr(fd, TCSANOW, &tty);
 271   if (ret != 0) {
 272     perror("tcsetattr");
 273     goto out;
 274   }
 275
 276   retval = 0;
 277 out:
 278   close(fd);
 279
 280   return retval;
 281 }
 282
 283 static int g_exit;
 284
 285 static void signal_handler(int sig)
 286 {
 287   g_exit = 1;
 288   signal(sig, SIG_DFL);
 289 }
 290
 291 /* ?0SA 00DU, ?1CB RLDU */
 292 #define STATE_BYTES 2
 293
 294 static uint8_t fixed_input_state[STATE_BYTES] = { 0x33, 0x3f };
 295
 296 enum mdbtn {
 297   MDBTN_UP = 1,
 298   MDBTN_DOWN,
 299   MDBTN_LEFT,
 300   MDBTN_RIGHT,
 301   MDBTN_A,
 302   MDBTN_B,
 303   MDBTN_C,
 304   MDBTN_START,
 305 };
 306
 307 static const enum mdbtn evdev_md_map[KEY_CNT] = {
 308   [KEY_UP]       = MDBTN_UP,
 309   [KEY_DOWN]     = MDBTN_DOWN,
 310   [KEY_LEFT]     = MDBTN_LEFT,
 311   [KEY_RIGHT]    = MDBTN_RIGHT,
 312   [KEY_HOME]     = MDBTN_A,
 313   [KEY_PAGEDOWN] = MDBTN_B,
 314   [KEY_END]      = MDBTN_C,
 315   [KEY_LEFTALT]  = MDBTN_START,
 316 };
 317
 318 int do_evdev_input(int fd)
 319 {
 320   uint8_t old_state[STATE_BYTES];
 321   uint8_t changed_bits[STATE_BYTES] = { 0, };
 322   struct input_event ev;
 323   enum mdbtn mdbtn;
 324   int i, ret;
 325
 326   ret = read(fd, &ev, sizeof(ev));
 327   if (ret != sizeof(ev)) {
 328     fprintf(stderr, "evdev read %d/%zd: ", ret, sizeof(ev));
 329     perror("");
 330     return 0;
 331   }
 332
 333   if (ev.type != EV_KEY)
 334     return 0;
 335
 336   if (ev.value != 0 && ev.value != 1)
 337     return 0;
 338
 339   if ((uint32_t)ev.code >= ARRAY_SIZE(evdev_md_map)) {
 340     fprintf(stderr, "evdev read bad key: %u\n", ev.code);
 341     return 0;
 342   }
 343
 344   mdbtn = evdev_md_map[ev.code];
 345   if (mdbtn == 0)
 346     return 0;
 347
 348   memcpy(old_state, fixed_input_state, STATE_BYTES);
 349
 350   /* ?0SA 00DU, ?1CB RLDU */
 351   switch (mdbtn) {
 352   case MDBTN_UP:
 353     changed_bits[0] = 0x01;
 354     changed_bits[1] = 0x01;
 355     break;
 356   case MDBTN_DOWN:
 357     changed_bits[0] = 0x02;
 358     changed_bits[1] = 0x02;
 359     break;
 360   case MDBTN_LEFT:
 361     changed_bits[0] = 0x00;
 362     changed_bits[1] = 0x04;
 363     break;
 364   case MDBTN_RIGHT:
 365     changed_bits[0] = 0x00;
 366     changed_bits[1] = 0x08;
 367     break;
 368   case MDBTN_A:
 369     changed_bits[0] = 0x10;
 370     changed_bits[1] = 0x00;
 371     break;
 372   case MDBTN_B:
 373     changed_bits[0] = 0x00;
 374     changed_bits[1] = 0x10;
 375     break;
 376   case MDBTN_C:
 377     changed_bits[0] = 0x00;
 378     changed_bits[1] = 0x20;
 379     break;
 380   case MDBTN_START:
 381     changed_bits[0] = 0x20;
 382     changed_bits[1] = 0x00;
 383     break;
 384   }
 385
 386   if (ev.value) {
 387     // key press
 388     for (i = 0; i < STATE_BYTES; i++)
 389       fixed_input_state[i] &= ~changed_bits[i];
 390   }
 391   else {
 392     // key release
 393     for (i = 0; i < STATE_BYTES; i++)
 394       fixed_input_state[i] |=  changed_bits[i];
 395   }
 396
 397   return memcmp(old_state, fixed_input_state, STATE_BYTES) ? 1 : 0;
 398 }
 399
 400 struct gmv_tas {
 401   char sig[15];
 402   char ver;
 403   uint32_t rerecord_count;
 404   char ctrl1;
 405   char ctrl2;
 406   uint16_t flags;
 407   char name[40];
 408   uint8_t data[0][3];
 409 };
 410
 411 static uint8_t *import_gmv(FILE *f, long size, int *frame_count)
 412 {
 413   struct gmv_tas *gmv;
 414   uint8_t *out;
 415   int ret;
 416   int i;
 417
 418   if (size < (long)sizeof(*gmv)) {
 419     fprintf(stderr, "bad gmv size: %ld\n", size);
 420     return NULL;
 421   }
 422
 423   gmv = malloc(size);
 424   if (gmv == NULL) {
 425     fprintf(stderr, "OOM?\n");
 426     return NULL;
 427   }
 428   ret = fread(gmv, 1, size, f);
 429   if (ret != size) {
 430     fprintf(stderr, "fread %d/%ld: ", ret, size);
 431     perror("");
 432     return NULL;
 433   }
 434
 435   *frame_count = (size - sizeof(*gmv)) / sizeof(gmv->data[0]);
 436
 437   /* check the GMV.. */
 438   if (*frame_count <= 0 || size != sizeof(*gmv) + *frame_count * 3) {
 439     fprintf(stderr, "broken gmv? frames=%d\n", *frame_count);
 440     return NULL;
 441   }
 442
 443   if (strncmp(gmv->sig, "Gens Movie TEST", 15) != 0) {
 444     fprintf(stderr, "bad GMV sig\n");
 445     return NULL;
 446   }
 447   if (gmv->ctrl1 != '3') {
 448     fprintf(stderr, "unhandled controlled config: '%c'\n", gmv->ctrl1);
 449     //return NULL;
 450   }
 451   if (gmv->ver >= 'A') {
 452     if (gmv->flags & 0x40) {
 453       fprintf(stderr, "unhandled flag: movie requires a savestate\n");
 454       return NULL;
 455     }
 456     if (gmv->flags & 0x20) {
 457       fprintf(stderr, "unhandled flag: 3-player movie\n");
 458       return NULL;
 459     }
 460     if (gmv->flags & ~0x80) {
 461       //fprintf(stderr, "unhandled flag(s): %04x\n", gmv->flags);
 462       //return 1;
 463     }
 464   }
 465   gmv->name[39] = 0;
 466   printf("loaded GMV: %s\n", gmv->name);
 467   printf("%d frames, %u rerecords\n",
 468          *frame_count, gmv->rerecord_count);
 469
 470   out = malloc(*frame_count * sizeof(out[0]));
 471   if (out == NULL) {
 472     fprintf(stderr, "OOM?\n");
 473     return NULL;
 474   }
 475
 476   for (i = 0; i < *frame_count; i++) {
 477     out[i] = gmv->data[i][0];
 478
 479     if (gmv->data[i][1] != 0xff || gmv->data[i][2] != 0xff)
 480     {
 481       fprintf(stderr, "f %d: unhandled byte(s) %02x %02x\n",
 482         i, gmv->data[i][1], gmv->data[i][2]);
 483     }
 484   }
 485
 486   return out;
 487 }
 488
 489 static int do_bkm_char(char c, char expect, uint8_t *val, int bit)
 490 {
 491   if (c == expect) {
 492     *val &= ~(1 << bit);
 493     return 0;
 494   }
 495   if (c == '.')
 496     return 0;
 497
 498   fprintf(stderr, "unexpected bkm char: '%c' instead of '%c'\n",
 499           c, expect);
 500   return 1;
 501 }
 502
 503 static uint8_t *import_bkm(FILE *f, int *frame_count)
 504 {
 505   uint8_t *out = NULL, val;
 506   int count = 0;
 507   int alloc = 0;
 508   int line = 0;
 509   char buf[256];
 510   const char *r;
 511   char *p;
 512   int i;
 513
 514   while ((p = fgets(buf, sizeof(buf), f)) != NULL) {
 515     line++;
 516     if (p[0] != '|')
 517       continue;
 518
 519     if (strlen(p) < 30)
 520       goto unhandled_line;
 521     if (p[30] != '\r' && p[30] != '\n')
 522       goto unhandled_line;
 523     p[30] = 0;
 524
 525     if (count >= alloc) {
 526       alloc = alloc * 2 + 64;
 527       out = realloc(out, alloc * sizeof(out[0]));
 528       if (out == NULL) {
 529         fprintf(stderr, "OOM?\n");
 530         return NULL;
 531       }
 532     }
 533
 534     val = 0xff;
 535
 536     if (strncmp(p, "|.|", 3) != 0)
 537       goto unhandled_line;
 538     p += 3;
 539
 540     const char ref[] = "UDLRABCS";
 541     for (r = ref, i = 0; *r != 0; p++, r++, i++) {
 542       if (do_bkm_char(*p, *r, &val, i))
 543         goto unhandled_line;
 544     }
 545
 546     if (strcmp(p, "....|............||") != 0)
 547       goto unhandled_line;
 548
 549     out[count++] = val;
 550     continue;
 551
 552 unhandled_line:
 553     fprintf(stderr, "unhandled bkm line %d: '%s'\n", line, buf);
 554     return NULL;
 555   }
 556
 557   printf("loaded bkm, %d frames\n", count);
 558   *frame_count = count;
 559   return out;
 560 }
 561
 562 static int submit_urb(int fd, struct usbdevfs_urb *urb, int ep,
 563   void *buf, size_t buf_size)
 564 {
 565   memset(urb, 0, sizeof(*urb));
 566   urb->type = USBDEVFS_URB_TYPE_INTERRUPT;
 567   urb->endpoint = ep;
 568   urb->buffer = buf;
 569   urb->buffer_length = buf_size;
 570
 571   return ioctl(fd, USBDEVFS_SUBMITURB, urb);
 572 }
 573
 574 enum my_urbs {
 575   URB_DATA_IN,
 576   URB_DATA_OUT,
 577   URB_DBG_IN,
 578   URB_CNT
 579 };
 580
 581 static void missing_arg(int a)
 582 {
 583   fprintf(stderr, "missing arg: %d\n", a);
 584   exit(1);
 585 }
 586
 587 int main(int argc, char *argv[])
 588 {
 589   struct teensy_dev dev;
 590   struct usbdevfs_urb urb[URB_CNT];
 591   struct usbdevfs_urb *reaped_urb;
 592   int fixed_input_changed;
 593   int evdev_fds[16];
 594   int evdev_fd_cnt = 0;
 595   int evdev_support;
 596   int wait_device = 0;
 597   int pending_urbs = 0;
 598   fd_set rfds, wfds;
 599   const char *tasfn = NULL;
 600   uint8_t *tas_data = NULL;
 601   int use_readinc = 0; // frame increment on read
 602   int tas_skip = 0;
 603   int enable_sent = 0;
 604   int frame_count = 0;
 605   int frames_sent = 0;
 606   char buf_dbg[64 + 1];
 607   struct tas_pkt pkt_in;
 608   struct tas_pkt pkt_out;
 609   struct timeval *timeout = NULL;
 610   struct timeval tout;
 611   int i, ret = -1;
 612   int fd;
 613
 614   for (i = 1; i < argc; i++) {
 615     if (argv[i][0] == '-') {
 616       switch (argv[i][1] | (argv[i][2] << 8)) {
 617       case 'm':
 618         i++;
 619         if (argv[i] == NULL)
 620           missing_arg(i);
 621         tasfn = argv[i];
 622         continue;
 623       case 's':
 624         i++;
 625         if (argv[i] == NULL)
 626           missing_arg(i);
 627         tas_skip = atoi(argv[i]);
 628         continue;
 629       case 'r':
 630         use_readinc = 1;
 631         continue;
 632       default:
 633         fprintf(stderr, "bad arg: %s\n", argv[i]);
 634         return 1;
 635       }
 636     }
 637     if (evdev_fd_cnt >= ARRAY_SIZE(evdev_fds)) {
 638       fprintf(stderr, "too many evdevs\n");
 639       break;
 640     }
 641     fd = open(argv[i], O_RDONLY);
 642     if (fd == -1) {
 643       fprintf(stderr, "open %s: ", argv[i]);
 644       perror("");
 645       continue;
 646     }
 647     evdev_support = 0;
 648     ret = ioctl(fd, EVIOCGBIT(0, sizeof(evdev_support)),
 649                 &evdev_support);
 650     if (ret < 0)
 651       perror("EVIOCGBIT");
 652     if (!(evdev_support & (1 << EV_KEY))) {
 653       fprintf(stderr, "%s doesn't have keys\n", argv[i]);
 654       close(fd);
 655       continue;
 656     }
 657     evdev_fds[evdev_fd_cnt++] = fd;
 658   }
 659
 660   if (tasfn != NULL) {
 661     const char *ext;
 662     long size;
 663     FILE *f;
 664
 665     f = fopen(tasfn, "rb");
 666     if (f == NULL) {
 667       fprintf(stderr, "fopen %s: ", tasfn);
 668       perror("");
 669       return 1;
 670     }
 671
 672     fseek(f, 0, SEEK_END);
 673     size = ftell(f);
 674     fseek(f, 0, SEEK_SET);
 675     if (size <= 0) {
 676       fprintf(stderr, "bad size: %ld\n", size);
 677       return 1;
 678     }
 679
 680     ext = strrchr(tasfn, '.');
 681     if (ext == NULL)
 682       ext = tasfn;
 683     else
 684       ext++;
 685
 686     if (strcasecmp(ext, "gmv") == 0)
 687       tas_data = import_gmv(f, size, &frame_count);
 688     else if (strcasecmp(ext, "bkm") == 0)
 689       tas_data = import_bkm(f, &frame_count);
 690     else {
 691       fprintf(stderr, "unknown movie type: '%s'\n", ext);
 692       return 1;
 693     }
 694     fclose(f);
 695
 696     if (tas_data == NULL) {
 697       fprintf(stderr, "failed fo parse %s\n", tasfn);
 698       return 1;
 699     }
 700
 701     if (tas_skip != 0) {
 702       if (tas_skip >= frame_count || tas_skip <= -frame_count) {
 703         printf("skip out of range: %d/%d\n", tas_skip, frame_count);
 704         return 1;
 705       }
 706       if (tas_skip > 0) {
 707         frame_count -= tas_skip;
 708         memmove(&tas_data[0], &tas_data[tas_skip],
 709           sizeof(tas_data[0]) * frame_count);
 710       }
 711       else {
 712         tas_data = realloc(tas_data,
 713                      (frame_count - tas_skip) * sizeof(tas_data[0]));
 714         if (tas_data == NULL) {
 715           fprintf(stderr, "OOM?\n");
 716           return 1;
 717         }
 718         memmove(&tas_data[-tas_skip], &tas_data[0],
 719           sizeof(tas_data[0]) * frame_count);
 720         memset(&tas_data[0], 0xff, sizeof(tas_data[0]) * -tas_skip);
 721         frame_count -= tas_skip;
 722       }
 723     }
 724   }
 725
 726   enable_echo(0);
 727   signal(SIGINT, signal_handler);
 728
 729   dev.fd = -1;
 730
 731   while (!g_exit)
 732   {
 733     if (dev.fd == -1) {
 734       ret = find_device(&dev, 0x16C0, 0x0486);
 735       if (ret < 0)
 736         break;
 737
 738       if (ret == 0) {
 739         if (!wait_device) {
 740           printf("waiting for device..\n");
 741           wait_device = 1;
 742         }
 743         usleep(250000);
 744         continue;
 745       }
 746
 747       wait_device = 0;
 748       pending_urbs = 0;
 749       enable_sent = 0;
 750       frames_sent = 0;
 751
 752       /* we wait first, then send commands, but if teensy
 753        * is started already, it won't send anything */
 754       tout.tv_sec = 1;
 755       tout.tv_usec = 0;
 756       timeout = &tout;
 757     }
 758
 759     if (!(pending_urbs & (1 << URB_DATA_IN))) {
 760       memset(&pkt_in, 0, sizeof(pkt_in));
 761       ret = submit_urb(dev.fd, &urb[URB_DATA_IN], dev.ifaces[0].ep_in,
 762                        &pkt_in, sizeof(pkt_in));
 763       if (ret != 0) {
 764         perror("USBDEVFS_SUBMITURB URB_DATA_IN");
 765         break;
 766       }
 767
 768       pending_urbs |= 1 << URB_DATA_IN;
 769     }
 770     if (!(pending_urbs & (1 << URB_DBG_IN))) {
 771       ret = submit_urb(dev.fd, &urb[URB_DBG_IN], dev.ifaces[1].ep_in,
 772                        buf_dbg, sizeof(buf_dbg) - 1);
 773       if (ret != 0) {
 774         perror("USBDEVFS_SUBMITURB URB_DBG_IN");
 775         break;
 776       }
 777
 778       pending_urbs |= 1 << URB_DBG_IN;
 779     }
 780
 781     FD_ZERO(&rfds);
 782     FD_SET(STDIN_FILENO, &rfds);
 783     for (i = 0; i < evdev_fd_cnt; i++)
 784       FD_SET(evdev_fds[i], &rfds);
 785
 786     FD_ZERO(&wfds);
 787     FD_SET(dev.fd, &wfds);
 788
 789     ret = select(dev.fd + 1, &rfds, &wfds, NULL, timeout);
 790     if (ret < 0) {
 791       perror("select");
 792       break;
 793     }
 794     timeout = NULL;
 795
 796     /* sometihng form stdin? */
 797     /* something from input devices? */
 798     if (FD_ISSET(STDIN_FILENO, &rfds)) {
 799       char c = 0;
 800       ret = read(STDIN_FILENO, &c, 1);
 801       if (ret <= 0) {
 802         perror("read stdin");
 803         break;
 804       }
 805
 806       switch (c) {
 807       case 'r':
 808         enable_sent = 0;
 809         break;
 810       }
 811     }
 812
 813     fixed_input_changed = 0;
 814     for (i = 0; i < evdev_fd_cnt; i++) {
 815       if (FD_ISSET(evdev_fds[i], &rfds)) {
 816         fixed_input_changed |=
 817           do_evdev_input(evdev_fds[i]);
 818       }
 819     }
 820
 821     /* something from USB? */
 822     if (FD_ISSET(dev.fd, &wfds))
 823     {
 824       unsigned int which_urb;
 825
 826       reaped_urb = NULL;
 827       ret = ioctl(dev.fd, USBDEVFS_REAPURB, &reaped_urb);
 828       if (ret != 0) {
 829         if (errno == ENODEV)
 830           goto dev_close;
 831         perror("USBDEVFS_REAPURB");
 832         break;
 833       }
 834       which_urb = reaped_urb - urb;
 835       if (which_urb < ARRAY_SIZE(urb))
 836         pending_urbs &= ~(1 << which_urb);
 837       else {
 838         fprintf(stderr, "reaped unknown urb: %p #%u",
 839                 reaped_urb, which_urb);
 840       }
 841
 842       if (reaped_urb != NULL && reaped_urb->status != 0) {
 843         errno = -reaped_urb->status;
 844         fprintf(stderr, "urb #%u: ", which_urb);
 845         perror("");
 846         if (reaped_urb->status == -EILSEQ) {
 847           /* this is usually a sign of disconnect.. */
 848           usleep(250000);
 849           goto dev_close;
 850         }
 851       }
 852       else if (reaped_urb == &urb[URB_DATA_IN])
 853       {
 854         /* some request from teensy */
 855         int count;
 856         uint8_t b;
 857
 858         switch (pkt_in.type) {
 859         case PKT_STREAM_REQ:
 860           printf("req: %d/%d/%d\n", pkt_in.req.frame,
 861             frames_sent, frame_count);
 862
 863           for (i = 0; i < sizeof(pkt_out.data) / 2; i++) {
 864             pkt_out.data[i * 2 + 0] = 0x33;
 865             pkt_out.data[i * 2 + 1] = 0x3f;
 866           }
 867           if (frames_sent < frame_count) {
 868             pkt_out.type = PKT_STREAM_DATA;
 869
 870             count = frame_count - frames_sent;
 871             if (count > sizeof(pkt_out.data) / 2)
 872               count = sizeof(pkt_out.data) / 2;
 873             for (i = 0; i < count; i++) {
 874               /* SCBA RLDU */
 875               b = tas_data[frames_sent];
 876
 877               /* ?0SA 00DU, ?1CB RLDU */
 878               pkt_out.data[i * 2 + 0] = (b & 0x13) | ((b >> 2) & 0x20);
 879               pkt_out.data[i * 2 + 1] = (b & 0x0f) | ((b >> 1) & 0x30);
 880               frames_sent++;
 881             }
 882           }
 883           else
 884             pkt_out.type = PKT_STREAM_END;
 885
 886           ret = submit_urb(dev.fd, &urb[URB_DATA_OUT],
 887                   dev.ifaces[0].ep_out, &pkt_out, sizeof(pkt_out));
 888           if (ret != 0)
 889             perror("USBDEVFS_SUBMITURB URB_DATA_OUT PKT_STREAM_DATA");
 890           break;
 891
 892         default:
 893           printf("host: got unknown pkt type: %04x\n", pkt_in.type);
 894           break;
 895         }
 896       }
 897       else if (reaped_urb == &urb[URB_DATA_OUT])
 898       {
 899       }
 900       else if (reaped_urb == &urb[URB_DBG_IN])
 901       {
 902         /* debug text */
 903         buf_dbg[reaped_urb->actual_length] = 0;
 904         printf("%s", buf_dbg);
 905       }
 906       else {
 907         fprintf(stderr, "reaped unknown urb? %p #%zu\n",
 908           reaped_urb, reaped_urb - urb);
 909       }
 910     }
 911
 912     /* something to send? */
 913     if (pending_urbs & (1 << URB_DATA_OUT))
 914       // can't do that yet
 915       continue;
 916
 917     if (tas_data != NULL && !enable_sent) {
 918       memset(&pkt_out, 0, sizeof(pkt_out));
 919       pkt_out.type = PKT_STREAM_ENABLE;
 920       pkt_out.start.use_readinc = use_readinc;
 921
 922       ret = submit_urb(dev.fd, &urb[URB_DATA_OUT], dev.ifaces[0].ep_out,
 923                        &pkt_out, sizeof(pkt_out));
 924       if (ret != 0) {
 925         perror("USBDEVFS_SUBMITURB PKT_STREAM_ENABLE");
 926         continue;
 927       }
 928       enable_sent = 1;
 929       frames_sent = 0;
 930     }
 931     if (tas_data == NULL && fixed_input_changed) {
 932       memset(&pkt_out, 0, sizeof(pkt_out));
 933       pkt_out.type = PKT_FIXED_STATE;
 934       memcpy(pkt_out.data, fixed_input_state, sizeof(fixed_input_state));
 935
 936       ret = submit_urb(dev.fd, &urb[URB_DATA_OUT], dev.ifaces[0].ep_out,
 937                        &pkt_out, sizeof(pkt_out));
 938       if (ret != 0) {
 939         perror("USBDEVFS_SUBMITURB URB_DATA_OUT");
 940         break;
 941       }
 942       pending_urbs |= 1 << URB_DATA_OUT;
 943       continue;
 944     }
 945
 946     continue;
 947
 948 dev_close:
 949     close(dev.fd);
 950     dev.fd = -1;
 951   }
 952
 953   enable_echo(1);
 954
 955   if (dev.fd != -1) {
 956     /* deal with pending URBs */
 957     if (pending_urbs & (1 << URB_DATA_IN))
 958       ioctl(dev.fd, USBDEVFS_DISCARDURB, &urb[URB_DATA_IN]);
 959     if (pending_urbs & (1 << URB_DBG_IN))
 960       ioctl(dev.fd, USBDEVFS_DISCARDURB, &urb[URB_DBG_IN]);
 961     for (i = 0; i < URB_CNT; i++) {
 962       if (pending_urbs & (1 << i)) {
 963         ret = ioctl(dev.fd, USBDEVFS_REAPURB, &reaped_urb);
 964         if (ret != 0)
 965           perror("USBDEVFS_REAPURB");
 966       }
 967     }
 968
 969     close(dev.fd);
 970   }
 971
 972   return ret;
 973 }
 974
 975 // vim: ts=2:sw=2:expandtab