fix an issue of lost inputs in direct control mode

[teensytas.git] / host / main.c

/*
 * TeensyTAS, TAS input player for MegaDrive
 * Copyright (c) 2014 notaz
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <sys/select.h>
#include <unistd.h>
#include <dirent.h>
#include <signal.h>
#include <termios.h>
#include <errno.h>
#include <linux/usbdevice_fs.h>
#include <linux/usb/ch9.h>
#include <linux/input.h>
#include "../pkts.h"

#define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))

struct teensy_dev {
  int fd;
  struct {
    int ep_in;
    int ep_out;
  } ifaces[2];
};

/* return 1 if founf, 0 if not, < 0 on error */
static int find_device(struct teensy_dev *dev,
  uint16_t vendor, uint16_t product)
{
  const char path_root[] = "/dev/bus/usb";
  union {
    struct usb_descriptor_header hdr;
    struct usb_device_descriptor d;
    struct usb_config_descriptor c;
    struct usb_interface_descriptor i;
    struct usb_endpoint_descriptor e;
    char space[0x100]; /* enough? */
  } desc;
  char path_bus[256], path_dev[256];
  struct dirent *ent, *ent_bus;
  DIR *dir = NULL, *dir_bus = NULL;
  int num, fd = -1;
  int iface = -1;
  int retval = -1;
  int ret;

  memset(dev, 0xff, sizeof(*dev));

  dir = opendir(path_root);
  if (dir == NULL) {
    perror("opendir");
    return -1;
  }

  for (ent = readdir(dir); ent != NULL; ent = readdir(dir)) {
    /* should be a number like 000 */
    if (sscanf(ent->d_name, "%03d", &num) != 1)
      continue;

    snprintf(path_bus, sizeof(path_bus), "%s/%s",
        path_root, ent->d_name);

    dir_bus = opendir(path_bus);
    if (dir_bus == NULL)
      continue;

    ent_bus = readdir(dir_bus);
    for (; ent_bus != NULL; ent_bus = readdir(dir_bus)) {
      if (sscanf(ent->d_name, "%03d", &num) != 1)
        continue;

      snprintf(path_dev, sizeof(path_dev), "%s/%s/%s",
          path_root, ent->d_name, ent_bus->d_name);

      fd = open(path_dev, O_RDWR);
      if (fd == -1)
        continue;

      ret = read(fd, &desc.d, sizeof(desc.d));
      if (ret != sizeof(desc.d)) {
        fprintf(stderr, "desc read: %d/%zd: ", ret, sizeof(desc.d));
        perror("");
        goto next;
      }

      if (desc.d.bDescriptorType != USB_DT_DEVICE) {
        fprintf(stderr, "%s: bad DT: 0x%02x\n",
            path_dev, desc.d.bDescriptorType);
        goto next;
      }

      if (desc.d.idVendor == vendor && desc.d.idProduct == product)
        goto found;

next:
      close(fd);
      fd = -1;
    }

    closedir(dir_bus);
    dir_bus = NULL;
  }

  /* not found */
  retval = 0;
  goto out;

found:
  if (desc.d.bNumConfigurations != 1) {
    fprintf(stderr, "unexpected bNumConfigurations: %u\n",
        desc.d.bNumConfigurations);
    goto out;
  }

  /* walk through all descriptors */
  while (1)
  {
    ret = read(fd, &desc.hdr, sizeof(desc.hdr));
    if (ret == 0)
      break;
    if (ret != sizeof(desc.hdr)) {
      fprintf(stderr, "desc.hdr read: %d/%zd: ", ret, sizeof(desc.hdr));
      perror("");
      break;
    }

    ret = (int)lseek(fd, -sizeof(desc.hdr), SEEK_CUR);
    if (ret == -1) {
      perror("lseek");
      break;
    }

    ret = read(fd, &desc, desc.hdr.bLength);
    if (ret != desc.hdr.bLength) {
      fprintf(stderr, "desc read: %d/%u: ", ret, desc.hdr.bLength);
      perror("");
      break;
    }

    switch (desc.hdr.bDescriptorType) {
      case USB_DT_CONFIG:
        if (desc.c.bNumInterfaces != 2) {
          fprintf(stderr, "unexpected bNumInterfaces: %u\n",
              desc.c.bNumInterfaces);
          goto out;
        }
        break;

      case USB_DT_INTERFACE:
        if (desc.i.bInterfaceClass != USB_CLASS_HID
            || desc.i.bInterfaceSubClass != 0
            || desc.i.bInterfaceProtocol != 0) {
          fprintf(stderr, "unexpected interface %x:%x:%x\n",
            desc.i.bInterfaceClass, desc.i.bInterfaceSubClass,
            desc.i.bInterfaceProtocol);
          goto out;
        }
        if (desc.i.bNumEndpoints != 2) {
          fprintf(stderr, "unexpected bNumEndpoints: %u\n",
            desc.i.bNumEndpoints);
          goto out;
        }
        iface++;
        break;

      case USB_DT_ENDPOINT:
        if (iface < 0 || iface >= ARRAY_SIZE(dev->ifaces)) {
          fprintf(stderr, "bad iface: %d\n", iface);
          goto out;
        }
        if (desc.e.wMaxPacketSize != 64 && desc.e.wMaxPacketSize != 32) {
          fprintf(stderr, "iface %d, EP %02x: "
            "unexpected wMaxPacketSize: %u\n",
            iface, desc.e.bEndpointAddress, desc.e.wMaxPacketSize);
          goto out;
        }
        if (desc.e.bEndpointAddress & 0x80)
          dev->ifaces[iface].ep_in = desc.e.bEndpointAddress; // & 0x7F;
        else
          dev->ifaces[iface].ep_out = desc.e.bEndpointAddress;
        break;

      case 0x21:
        /* ignore */
        break;

      default:
        fprintf(stderr, "skipping desc 0x%02x\n",
          desc.hdr.bDescriptorType);
        break;
    }
  }

  /* claim interfaces */
  for (iface = 0; iface < ARRAY_SIZE(dev->ifaces); iface++) {
    struct usbdevfs_ioctl usbio;

    if (dev->ifaces[iface].ep_in == -1) {
      fprintf(stderr, "missing ep_in, iface: %d\n", iface);
      goto out;
    }
    if (dev->ifaces[iface].ep_out == -1) {
      fprintf(stderr, "missing ep_out, iface: %d\n", iface);
      goto out;
    }

    /* disconnect default driver */
    memset(&usbio, 0, sizeof(usbio));
    usbio.ifno = iface;
    usbio.ioctl_code = USBDEVFS_DISCONNECT;
    ret = ioctl(fd, USBDEVFS_IOCTL, &usbio);
    if (ret != 0 && errno != ENODATA)
      perror("USBDEVFS_DISCONNECT");

    ret = ioctl(fd, USBDEVFS_CLAIMINTERFACE, &iface);
    if (ret != 0)
      perror("USBDEVFS_CLAIMINTERFACE");
  }

  dev->fd = fd;
  fd = -1;
  retval = 1;

out:
  if (fd != -1)
    close(fd);
  if (dir_bus != NULL)
    closedir(dir_bus);
  if (dir != NULL)
    closedir(dir);

  return retval;
}

static int enable_echo(int enable)
{
  const char *portname = "/dev/tty";
  struct termios tty;
  int retval = -1;
  int ret;
  int fd;

  memset(&tty, 0, sizeof(tty));

  fd = open(portname, O_RDWR | O_NOCTTY | O_SYNC);
  if (fd < 0) {
    fprintf(stderr, "open %s: ", portname);
    perror("");
    return 1;
  }

  ret = tcgetattr(fd, &tty);
  if (ret != 0) {
    perror("tcgetattr");
    goto out;
  }

  // printf("lflag: 0%o\n", tty.c_lflag);
  if (enable)
    tty.c_lflag |= ECHO | ICANON;
  else {
    tty.c_lflag &= ~(ECHO | ICANON);
    tty.c_cc[VMIN] = tty.c_cc[VTIME] = 0;
  }

  ret = tcsetattr(fd, TCSANOW, &tty);
  if (ret != 0) {
    perror("tcsetattr");
    goto out;
  }

  retval = 0;
out:
  close(fd);

  return retval;
}

static int g_exit;

static void signal_handler(int sig)
{
  g_exit = 1;
  signal(sig, SIG_DFL);
}

/* ?0SA 00DU, ?1CB RLDU */
#define STATE_BYTES 2

static uint8_t fixed_input_state[STATE_BYTES] = { 0x33, 0x3f };

enum mdbtn {
  MDBTN_UP = 1,
  MDBTN_DOWN,
  MDBTN_LEFT,
  MDBTN_RIGHT,
  MDBTN_A,
  MDBTN_B,
  MDBTN_C,
  MDBTN_START,
};

static const enum mdbtn evdev_md_map[KEY_CNT] = {
  [KEY_UP]       = MDBTN_UP,
  [KEY_DOWN]     = MDBTN_DOWN,
  [KEY_LEFT]     = MDBTN_LEFT,
  [KEY_RIGHT]    = MDBTN_RIGHT,
  [KEY_HOME]     = MDBTN_A,
  [KEY_PAGEDOWN] = MDBTN_B,
  [KEY_END]      = MDBTN_C,
  [KEY_LEFTALT]  = MDBTN_START,
};

int do_evdev_input(int fd)
{
  uint8_t old_state[STATE_BYTES];
  uint8_t changed_bits[STATE_BYTES] = { 0, };
  struct input_event ev;
  enum mdbtn mdbtn;
  int i, ret;

  ret = read(fd, &ev, sizeof(ev));
  if (ret != sizeof(ev)) {
    fprintf(stderr, "evdev read %d/%zd: ", ret, sizeof(ev));
    perror("");
    return 0;
  }

  if (ev.type != EV_KEY)
    return 0;

  if (ev.value != 0 && ev.value != 1)
    return 0;

  if ((uint32_t)ev.code >= ARRAY_SIZE(evdev_md_map)) {
    fprintf(stderr, "evdev read bad key: %u\n", ev.code);
    return 0;
  }

  mdbtn = evdev_md_map[ev.code];
  if (mdbtn == 0)
    return 0;

  memcpy(old_state, fixed_input_state, STATE_BYTES);

  /* ?0SA 00DU, ?1CB RLDU */
  switch (mdbtn) {
  case MDBTN_UP:
    changed_bits[0] = 0x01;
    changed_bits[1] = 0x01;
    break;
  case MDBTN_DOWN:
    changed_bits[0] = 0x02;
    changed_bits[1] = 0x02;
    break;
  case MDBTN_LEFT:
    changed_bits[0] = 0x00;
    changed_bits[1] = 0x04;
    break;
  case MDBTN_RIGHT:
    changed_bits[0] = 0x00;
    changed_bits[1] = 0x08;
    break;
  case MDBTN_A:
    changed_bits[0] = 0x10;
    changed_bits[1] = 0x00;
    break;
  case MDBTN_B:
    changed_bits[0] = 0x00;
    changed_bits[1] = 0x10;
    break;
  case MDBTN_C:
    changed_bits[0] = 0x00;
    changed_bits[1] = 0x20;
    break;
  case MDBTN_START:
    changed_bits[0] = 0x20;
    changed_bits[1] = 0x00;
    break;
  }

  if (ev.value) {
    // key press
    for (i = 0; i < STATE_BYTES; i++)
      fixed_input_state[i] &= ~changed_bits[i];
  }
  else {
    // key release
    for (i = 0; i < STATE_BYTES; i++)
      fixed_input_state[i] |=  changed_bits[i];
  }

  return memcmp(old_state, fixed_input_state, STATE_BYTES) ? 1 : 0;
}

struct gmv_tas {
  char sig[15];
  char ver;
  uint32_t rerecord_count;
  char ctrl1;
  char ctrl2;
  uint16_t flags;
  char name[40];
  uint8_t data[0][3];
};

static uint8_t *import_gmv(FILE *f, long size, int *frame_count)
{
  struct gmv_tas *gmv;
  uint8_t *out;
  int ret;
  int i;

  if (size < (long)sizeof(*gmv)) {
    fprintf(stderr, "bad gmv size: %ld\n", size);
    return NULL;
  }

  gmv = malloc(size);
  if (gmv == NULL) {
    fprintf(stderr, "OOM?\n");
    return NULL;
  }
  ret = fread(gmv, 1, size, f);
  if (ret != size) {
    fprintf(stderr, "fread %d/%ld: ", ret, size);
    perror("");
    return NULL;
  }

  *frame_count = (size - sizeof(*gmv)) / sizeof(gmv->data[0]);

  /* check the GMV.. */
  if (*frame_count <= 0 || size != sizeof(*gmv) + *frame_count * 3) {
    fprintf(stderr, "broken gmv? frames=%d\n", *frame_count);
    return NULL;
  }

  if (strncmp(gmv->sig, "Gens Movie TEST", 15) != 0) {
    fprintf(stderr, "bad GMV sig\n");
    return NULL;
  }
  if (gmv->ctrl1 != '3') {
    fprintf(stderr, "unhandled controlled config: '%c'\n", gmv->ctrl1);
    //return NULL;
  }
  if (gmv->ver >= 'A') {
    if (gmv->flags & 0x40) {
      fprintf(stderr, "unhandled flag: movie requires a savestate\n");
      return NULL;
    }
    if (gmv->flags & 0x20) {
      fprintf(stderr, "unhandled flag: 3-player movie\n");
      return NULL;
    }
    if (gmv->flags & ~0x80) {
      //fprintf(stderr, "unhandled flag(s): %04x\n", gmv->flags);
      //return 1;
    }
  }
  gmv->name[39] = 0;
  printf("loaded GMV: %s\n", gmv->name);
  printf("%d frames, %u rerecords\n",
         *frame_count, gmv->rerecord_count);

  out = malloc(*frame_count * sizeof(out[0]));
  if (out == NULL) {
    fprintf(stderr, "OOM?\n");
    return NULL;
  }

  for (i = 0; i < *frame_count; i++) {
    out[i] = gmv->data[i][0];

    if (gmv->data[i][1] != 0xff || gmv->data[i][2] != 0xff)
    {
      fprintf(stderr, "f %d: unhandled byte(s) %02x %02x\n",
        i, gmv->data[i][1], gmv->data[i][2]);
    }
  }

  return out;
}

static int do_bkm_char(char c, char expect, uint8_t *val, int bit)
{
  if (c == expect) {
    *val &= ~(1 << bit);
    return 0;
  }
  if (c == '.')
    return 0;

  fprintf(stderr, "unexpected bkm char: '%c' instead of '%c'\n",
          c, expect);
  return 1;
}

static uint8_t *import_bkm(FILE *f, int *frame_count)
{
  uint8_t *out = NULL, val;
  int count = 0;
  int alloc = 0;
  int line = 0;
  char buf[256];
  const char *r;
  char *p;
  int i;

  while ((p = fgets(buf, sizeof(buf), f)) != NULL) {
    line++;
    if (p[0] != '|')
      continue;

    if (strlen(p) < 30)
      goto unhandled_line;
    if (p[30] != '\r' && p[30] != '\n')
      goto unhandled_line;
    p[30] = 0;

    if (count >= alloc) {
      alloc = alloc * 2 + 64;
      out = realloc(out, alloc * sizeof(out[0]));
      if (out == NULL) {
        fprintf(stderr, "OOM?\n");
        return NULL;
      }
    }

    val = 0xff;

    if (strncmp(p, "|.|", 3) != 0)
      goto unhandled_line;
    p += 3;

    const char ref[] = "UDLRABCS";
    for (r = ref, i = 0; *r != 0; p++, r++, i++) {
      if (do_bkm_char(*p, *r, &val, i))
        goto unhandled_line;
    }

    if (strcmp(p, "....|............||") != 0)
      goto unhandled_line;

    out[count++] = val;
    continue;

unhandled_line:
    fprintf(stderr, "unhandled bkm line %d: '%s'\n", line, buf);
    return NULL;
  }

  printf("loaded bkm, %d frames\n", count);
  *frame_count = count;
  return out;
}

static int write_bkm_frame(FILE *f, const uint8_t *data)
{
  /* ?0SA 00DU, ?1CB RLDU */
  static const char ref[]  = "UDLRABCSXYZM";
  static const char bits[] = { 0,1,2,3, 12,4,5,13, 16,16,16,16 };
  uint32_t idata[2];
  int p, i;

  if (f == NULL) {
    fprintf(stderr, "%s called without outfile\n", __func__);
    goto out;
  }

  idata[0] = 0x10000 | (data[0] << 8) | data[1];
  idata[1] = ~0;

  fprintf(f, "|.|");
  for (p = 0; p < 2; p++) {
    for (i = 0; i < 12; i++)
      fprintf(f, "%c", (idata[p] & (1 << bits[i])) ? '.' : ref[i]);
    fprintf(f, "|");
  }
  fprintf(f, "|\n");

out:
  return 2;
}

static int tas_data_to_teensy(uint8_t b, uint8_t *data, FILE *logf)
{
  uint8_t t;

  /* SCBA RLDU */
  /*     v     */
  /* ?0SA 00DU, ?1CB RLDU */
  data[0] = (b & 0x13) | ((b >> 2) & 0x20);
  data[1] = (b & 0x0f) | ((b >> 1) & 0x30);

  if (logf != NULL) {
    fwrite(&data[0], 1, 1, logf);
    t = data[1] | 0x40; // expected TH
    fwrite(&t, 1, 1, logf);
  }

  return 2;
}

static int submit_urb(int fd, struct usbdevfs_urb *urb, int ep,
  void *buf, size_t buf_size)
{
  memset(urb, 0, sizeof(*urb));
  urb->type = USBDEVFS_URB_TYPE_INTERRUPT;
  urb->endpoint = ep;
  urb->buffer = buf;
  urb->buffer_length = buf_size;

  return ioctl(fd, USBDEVFS_SUBMITURB, urb);
}

enum my_urbs {
  URB_DATA_IN,
  URB_DATA_OUT,
  URB_DBG_IN,
  URB_CNT
};

static void missing_arg(int a)
{
  fprintf(stderr, "missing arg: %d\n", a);
  exit(1);
}

int main(int argc, char *argv[])
{
  struct teensy_dev dev;
  struct usbdevfs_urb urb[URB_CNT];
  struct usbdevfs_urb *reaped_urb;
  int fixed_input_changed = 0;
  int evdev_fds[16];
  int evdev_fd_cnt = 0;
  int evdev_support;
  int wait_device = 0;
  int pending_urbs = 0;
  fd_set rfds, wfds;
  const char *tasfn = NULL;
  const char *outfn = NULL;
  const char *logfn = NULL;
  uint8_t *tas_data = NULL;
  int use_vsync = 0; // frame increment on vsync
  int no_start_seq = 0;
  int tas_skip = 0;
  int enable_sent = 0;
  int abort_sent = 0;
  int frame_count = 0;
  int frames_sent = 0;
  char buf_dbg[64 + 1];
  struct tas_pkt pkt_in;
  struct tas_pkt pkt_out;
  struct timeval *timeout = NULL;
  struct timeval tout;
  FILE *outf = NULL;
  FILE *logf = NULL;
  int i, ret = -1;
  int fd;

  for (i = 1; i < argc; i++) {
    if (argv[i][0] == '-') {
      switch (argv[i][1] | (argv[i][2] << 8)) {
      case 'm':
        i++;
        if (argv[i] == NULL)
          missing_arg(i);
        tasfn = argv[i];
        continue;
      case 'w':
        i++;
        if (argv[i] == NULL)
          missing_arg(i);
        outfn = argv[i];
        continue;
      case 'l':
        i++;
        if (argv[i] == NULL)
          missing_arg(i);
        logfn = argv[i];
        continue;
      case 's':
        i++;
        if (argv[i] == NULL)
          missing_arg(i);
        tas_skip = atoi(argv[i]);
        continue;
      case 'v':
        use_vsync = 1;
        continue;
      case 'n':
        no_start_seq = 1;
        continue;
      default:
        fprintf(stderr, "bad arg: %s\n", argv[i]);
        return 1;
      }
    }

    /* remaining args are evdev filenames */
    if (evdev_fd_cnt >= ARRAY_SIZE(evdev_fds)) {
      fprintf(stderr, "too many evdevs\n");
      break;
    }
    fd = open(argv[i], O_RDONLY);
    if (fd == -1) {
      fprintf(stderr, "open %s: ", argv[i]);
      perror("");
      continue;
    }
    evdev_support = 0;
    ret = ioctl(fd, EVIOCGBIT(0, sizeof(evdev_support)),
                &evdev_support);
    if (ret < 0)
      perror("EVIOCGBIT");
    if (!(evdev_support & (1 << EV_KEY))) {
      fprintf(stderr, "%s doesn't have keys\n", argv[i]);
      close(fd);
      continue;
    }
    evdev_fds[evdev_fd_cnt++] = fd;
  }

  if (tasfn != NULL) {
    const char *ext;
    long size;
    FILE *f;

    f = fopen(tasfn, "rb");
    if (f == NULL) {
      fprintf(stderr, "fopen %s: ", tasfn);
      perror("");
      return 1;
    }

    fseek(f, 0, SEEK_END);
    size = ftell(f);
    fseek(f, 0, SEEK_SET);
    if (size <= 0) {
      fprintf(stderr, "bad size: %ld\n", size);
      return 1;
    }

    ext = strrchr(tasfn, '.');
    if (ext == NULL)
      ext = tasfn;
    else
      ext++;

    if (strcasecmp(ext, "gmv") == 0)
      tas_data = import_gmv(f, size, &frame_count);
    else if (strcasecmp(ext, "bkm") == 0)
      tas_data = import_bkm(f, &frame_count);
    else {
      fprintf(stderr, "unknown movie type: '%s'\n", ext);
      return 1;
    }
    fclose(f);

    if (tas_data == NULL) {
      fprintf(stderr, "failed fo parse %s\n", tasfn);
      return 1;
    }

    if (tas_skip != 0) {
      if (tas_skip >= frame_count || tas_skip <= -frame_count) {
        printf("skip out of range: %d/%d\n", tas_skip, frame_count);
        return 1;
      }
      if (tas_skip > 0) {
        frame_count -= tas_skip;
        memmove(&tas_data[0], &tas_data[tas_skip],
          sizeof(tas_data[0]) * frame_count);
      }
      else {
        tas_data = realloc(tas_data,
                     (frame_count - tas_skip) * sizeof(tas_data[0]));
        if (tas_data == NULL) {
          fprintf(stderr, "OOM?\n");
          return 1;
        }
        memmove(&tas_data[-tas_skip], &tas_data[0],
          sizeof(tas_data[0]) * frame_count);
        memset(&tas_data[0], 0xff, sizeof(tas_data[0]) * -tas_skip);
        frame_count -= tas_skip;
      }
    }
  }

  if (outfn != NULL) {
    outf = fopen(outfn, "w");
    if (outf == NULL) {
      fprintf(stderr, "fopen %s: ", outfn);
      perror("");
      return 1;
    }
  }

  if (logfn != NULL) {
    logf = fopen(logfn, "wb");
    if (logf == NULL) {
      fprintf(stderr, "fopen %s: ", logfn);
      perror("");
      return 1;
    }
  }

  enable_echo(0);
  signal(SIGINT, signal_handler);

  dev.fd = -1;

  while (!g_exit || (pending_urbs & (1 << URB_DATA_OUT)))
  {
    if (dev.fd == -1) {
      ret = find_device(&dev, 0x16C0, 0x0486);
      if (ret < 0)
        break;

      if (ret == 0) {
        if (!wait_device) {
          printf("waiting for device..\n");
          wait_device = 1;
        }
        usleep(250000);
        continue;
      }

      wait_device = 0;
      pending_urbs = 0;
      enable_sent = 0;
      frames_sent = 0;

      /* we wait first, then send commands, but if teensy
       * is started already, it won't send anything */
      tout.tv_sec = 1;
      tout.tv_usec = 0;
      timeout = &tout;
    }

    if (!(pending_urbs & (1 << URB_DATA_IN))) {
      memset(&pkt_in, 0, sizeof(pkt_in));
      ret = submit_urb(dev.fd, &urb[URB_DATA_IN], dev.ifaces[0].ep_in,
                       &pkt_in, sizeof(pkt_in));
      if (ret != 0) {
        perror("USBDEVFS_SUBMITURB URB_DATA_IN");
        break;
      }

      pending_urbs |= 1 << URB_DATA_IN;
    }
    if (!(pending_urbs & (1 << URB_DBG_IN))) {
      ret = submit_urb(dev.fd, &urb[URB_DBG_IN], dev.ifaces[1].ep_in,
                       buf_dbg, sizeof(buf_dbg) - 1);
      if (ret != 0) {
        perror("USBDEVFS_SUBMITURB URB_DBG_IN");
        break;
      }

      pending_urbs |= 1 << URB_DBG_IN;
    }

    FD_ZERO(&rfds);
    FD_SET(STDIN_FILENO, &rfds);
    for (i = 0; i < evdev_fd_cnt; i++)
      FD_SET(evdev_fds[i], &rfds);

    FD_ZERO(&wfds);
    FD_SET(dev.fd, &wfds);

    ret = select(dev.fd + 1, &rfds, &wfds, NULL, timeout);
    if (ret < 0) {
      perror("select");
      break;
    }
    timeout = NULL;

    /* sometihng form stdin? */
    if (FD_ISSET(STDIN_FILENO, &rfds)) {
      char c = 0;
      ret = read(STDIN_FILENO, &c, 1);
      if (ret <= 0) {
        perror("read stdin");
        break;
      }

      switch (c) {
      case 'r':
        enable_sent = 0;
        if (logf != NULL)
          rewind(logf);
        break;
      }
    }

    /* something from input devices? */
    for (i = 0; i < evdev_fd_cnt; i++) {
      if (FD_ISSET(evdev_fds[i], &rfds)) {
        fixed_input_changed |=
          do_evdev_input(evdev_fds[i]);
      }
    }

    /* something from USB? */
    if (FD_ISSET(dev.fd, &wfds))
    {
      unsigned int which_urb;

      reaped_urb = NULL;
      ret = ioctl(dev.fd, USBDEVFS_REAPURB, &reaped_urb);
      if (ret != 0) {
        if (errno == ENODEV)
          goto dev_close;
        perror("USBDEVFS_REAPURB");
        break;
      }
      which_urb = reaped_urb - urb;
      if (which_urb < ARRAY_SIZE(urb))
        pending_urbs &= ~(1 << which_urb);
      else {
        fprintf(stderr, "reaped unknown urb: %p #%u",
                reaped_urb, which_urb);
      }

      if (reaped_urb != NULL && reaped_urb->status != 0) {
        errno = -reaped_urb->status;
        fprintf(stderr, "urb #%u: ", which_urb);
        perror("");
        if (reaped_urb->status == -EILSEQ) {
          /* this is usually a sign of disconnect.. */
          usleep(250000);
          goto dev_close;
        }
      }
      else if (reaped_urb == &urb[URB_DATA_IN])
      {
        /* some request from teensy */
        switch (pkt_in.type) {
        case PKT_STREAM_REQ:
          printf("req: %d/%d/%d\n", pkt_in.req.frame,
            frames_sent, frame_count);

          pkt_out.size = 0;
          if (frames_sent < frame_count) {
            pkt_out.type = PKT_STREAM_DATA_TO;

            for (i = 0; i < sizeof(pkt_out.data); ) {
              i += tas_data_to_teensy(tas_data[frames_sent],
                     pkt_out.data + i, logf);

              frames_sent++;
              if (frames_sent >= frame_count)
                break;
            }
            pkt_out.size = i;
          }
          else {
            pkt_out.type = PKT_STREAM_END;
            if (logf != NULL)
              fflush(logf);
          }

          ret = submit_urb(dev.fd, &urb[URB_DATA_OUT],
                  dev.ifaces[0].ep_out, &pkt_out, sizeof(pkt_out));
          if (ret != 0)
            perror("USBDEVFS_SUBMITURB PKT_STREAM_DATA_TO");
          break;

        case PKT_STREAM_DATA_FROM:
          printf("f: %d\n", frame_count);
          if (pkt_in.size == 0 || pkt_in.size > sizeof(pkt_out.data)) {
            printf("host: got bad DATA_FROM size: %u\n", pkt_in.size);
            break;
          }
          for (i = 0; i < pkt_in.size; ) {
            i += write_bkm_frame(outf, pkt_in.data + i);
            frame_count++;
          }
          break;

        default:
          printf("host: got unknown pkt type: %04x\n", pkt_in.type);
          break;
        }
      }
      else if (reaped_urb == &urb[URB_DATA_OUT])
      {
      }
      else if (reaped_urb == &urb[URB_DBG_IN])
      {
        /* debug text */
        buf_dbg[reaped_urb->actual_length] = 0;
        printf("%s", buf_dbg);

        // continue receiving debug before sending out stuff
        tout.tv_sec = 0;
        tout.tv_usec = 1000;
        timeout = &tout;
        continue;
      }
      else {
        fprintf(stderr, "reaped unknown urb? %p #%zu\n",
          reaped_urb, reaped_urb - urb);
      }
    }

    /* something to send? */
    if (pending_urbs & (1 << URB_DATA_OUT))
      // can't do that yet
      continue;

    if ((tas_data != NULL || outf != NULL) && !enable_sent) {
      memset(&pkt_out, 0, sizeof(pkt_out));
      pkt_out.type = PKT_STREAM_ENABLE;
      pkt_out.enable.stream_to = (tas_data != NULL);
      pkt_out.enable.stream_from = (outf != NULL);
      pkt_out.enable.use_readinc = !use_vsync;
      pkt_out.enable.no_start_seq = no_start_seq;

      ret = submit_urb(dev.fd, &urb[URB_DATA_OUT], dev.ifaces[0].ep_out,
                       &pkt_out, sizeof(pkt_out));
      if (ret != 0) {
        perror("USBDEVFS_SUBMITURB PKT_STREAM_ENABLE");
        continue;
      }
      pending_urbs |= 1 << URB_DATA_OUT;
      enable_sent = 1;
      frames_sent = 0;
      continue;
    }
    if (tas_data == NULL && fixed_input_changed) {
      memset(&pkt_out, 0, sizeof(pkt_out));
      pkt_out.type = PKT_FIXED_STATE;
      memcpy(pkt_out.data, fixed_input_state, sizeof(fixed_input_state));

      ret = submit_urb(dev.fd, &urb[URB_DATA_OUT], dev.ifaces[0].ep_out,
                       &pkt_out, sizeof(pkt_out));
      if (ret != 0) {
        perror("USBDEVFS_SUBMITURB PKT_FIXED_STATE");
        break;
      }
      fixed_input_changed = 0;
      pending_urbs |= 1 << URB_DATA_OUT;
      continue;
    }
    if (g_exit && !abort_sent) {
      memset(&pkt_out, 0, sizeof(pkt_out));
      pkt_out.type = PKT_STREAM_ABORT;

      ret = submit_urb(dev.fd, &urb[URB_DATA_OUT], dev.ifaces[0].ep_out,
                       &pkt_out, sizeof(pkt_out));
      if (ret != 0) {
        perror("USBDEVFS_SUBMITURB PKT_STREAM_ABORT");
        break;
      }
      pending_urbs |= 1 << URB_DATA_OUT;
      abort_sent = 1;
      continue;
    }

    continue;

dev_close:
    close(dev.fd);
    dev.fd = -1;
  }

  enable_echo(1);

  if (outf != NULL)
    fclose(outf);
  if (logf != NULL)
    fclose(logf);

  if (dev.fd != -1) {
    /* deal with pending URBs */
    if (pending_urbs & (1 << URB_DATA_IN))
      ioctl(dev.fd, USBDEVFS_DISCARDURB, &urb[URB_DATA_IN]);
    if (pending_urbs & (1 << URB_DBG_IN))
      ioctl(dev.fd, USBDEVFS_DISCARDURB, &urb[URB_DBG_IN]);
    for (i = 0; i < URB_CNT; i++) {
      if (pending_urbs & (1 << i)) {
        ret = ioctl(dev.fd, USBDEVFS_REAPURB, &reaped_urb);
        if (ret != 0)
          perror("USBDEVFS_REAPURB");
      }
    }

    close(dev.fd);
  }

  return ret;
}

// vim: ts=2:sw=2:expandtab