X-Git-Url: https://notaz.gp2x.de/cgi-bin/gitweb.cgi?p=megadrive.git;a=blobdiff_plain;f=host%2Fmain.c;h=e7026d3b3f0773efa5acc2c47fa22059cbe71221;hp=240fd96992aafd4f1b91287503c7b557706e26b4;hb=611721255b3bffb1fcb13cf978da52181660ce2d;hpb=1cb2822f8f829a2197843a60e1aa250730ae00e3

diff --git a/host/main.c b/host/main.c
index 240fd96..e7026d3 100644
--- a/host/main.c
+++ b/host/main.c
@@ -1,4 +1,5 @@
 #include <stdio.h>
+#include <stdlib.h>
 #include <string.h>
 #include <stdint.h>
 #include <sys/types.h>
@@ -8,10 +9,13 @@
 #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]))
 
@@ -232,6 +236,55 @@ out:
   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;
+  else
+    tty.c_lflag &= ~ECHO;
+
+  ret = tcsetattr(fd, TCSANOW, &tty);
+  if (ret != 0) {
+    perror("tcsetattr");
+    goto out;
+  }
+
+  retval = 0;
+out:
+  close(fd);
+
+  return retval;
+}
+
+static void signal_handler(int sig)
+{
+  enable_echo(1);
+  signal(sig, SIG_DFL);
+  raise(sig);
+}
+
 /* ?0SA 00DU, ?1CB RLDU */
 #define STATE_BYTES 2
 
@@ -341,6 +394,29 @@ int do_evdev_input(int fd)
   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 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,
@@ -348,9 +424,14 @@ enum my_urbs {
   URB_CNT
 };
 
+static void missing_arg(int a)
+{
+  fprintf(stderr, "missing arg: %d\n", a);
+  exit(1);
+}
+
 int main(int argc, char *argv[])
 {
-  char buf_dbg[64 + 1], buf_in[64], buf_out[64];
   struct teensy_dev dev;
   struct usbdevfs_urb urb[URB_CNT];
   struct usbdevfs_urb *reaped_urb;
@@ -362,22 +443,52 @@ int main(int argc, char *argv[])
   int dbg_in_sent = 0;
   int data_in_sent = 0;
   fd_set rfds, wfds;
+  struct gmv_tas *gmv = NULL;
+  const char *tasfn = NULL;
+  int tas_skip = 0;
+  int enable_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;
   int i, ret;
   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 's':
+        i++;
+        if (argv[i] == NULL)
+          missing_arg(i);
+        tas_skip = atoi(argv[i]);
+        continue;
+      default:
+        fprintf(stderr, "bad arg: %s\n", argv[i]);
+        return 1;
+      }
+    }
     if (evdev_fd_cnt >= ARRAY_SIZE(evdev_fds)) {
       fprintf(stderr, "too many evdevs\n");
       break;
     }
-		fd = open(argv[i], O_RDONLY);
+    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)),
+    ret = ioctl(fd, EVIOCGBIT(0, sizeof(evdev_support)),
                 &evdev_support);
     if (ret < 0)
       perror("EVIOCGBIT");
@@ -389,6 +500,99 @@ int main(int argc, char *argv[])
     evdev_fds[evdev_fd_cnt++] = fd;
   }
 
+  if (tasfn != NULL) {
+    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 < (long)sizeof(*gmv)) {
+      fprintf(stderr, "bad gmv size: %ld\n", size);
+      return 1;
+    }
+    gmv = malloc(size);
+    if (gmv == NULL) {
+      fprintf(stderr, "OOM?\n");
+      return 1;
+    }
+    ret = fread(gmv, 1, size, f);
+    if (ret != size) {
+      fprintf(stderr, "fread %d/%ld: ", ret, size);
+      perror("");
+      return 1;
+    }
+    fclose(f);
+    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 1;
+    }
+
+    if (strncmp(gmv->sig, "Gens Movie TEST", 15) != 0) {
+      fprintf(stderr, "bad GMV sig\n");
+      return 1;
+    }
+    if (gmv->ctrl1 != '3') {
+      fprintf(stderr, "unhandled controlled config: '%c'\n", gmv->ctrl1);
+      //return 1;
+    }
+    if (gmv->ver >= 'A') {
+      if (gmv->flags & 0x40) {
+        fprintf(stderr, "unhandled flag: movie requires a savestate\n");
+        return 1;
+      }
+      if (gmv->flags & 0x20) {
+        fprintf(stderr, "unhandled flag: 3-player movie\n");
+        return 1;
+      }
+      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);
+
+    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(&gmv->data[0], &gmv->data[tas_skip],
+          sizeof(gmv->data[0]) * frame_count);
+      }
+      else {
+        gmv = realloc(gmv, sizeof(*gmv)
+                + (frame_count - tas_skip) * sizeof(gmv->data[0]));
+        if (gmv == NULL) {
+          fprintf(stderr, "OOM?\n");
+          return 1;
+        }
+        memmove(&gmv->data[-tas_skip], &gmv->data[0],
+          sizeof(gmv->data[0]) * frame_count);
+        memset(&gmv->data[0], 0xff, sizeof(gmv->data[0]) * -tas_skip);
+        frame_count -= tas_skip;
+      }
+    }
+  }
+
+  enable_echo(0);
+  signal(SIGINT, signal_handler);
+
   dev.fd = -1;
 
   while (1)
@@ -396,7 +600,7 @@ int main(int argc, char *argv[])
     if (dev.fd == -1) {
       ret = find_device(&dev, 0x16C0, 0x0486);
       if (ret < 0)
-        return ret;
+        break;
 
       if (ret == 0) {
         if (!wait_device) {
@@ -410,34 +614,35 @@ int main(int argc, char *argv[])
       wait_device = 0;
       data_in_sent = 0;
       dbg_in_sent = 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 (!data_in_sent) {
-      memset(&urb[URB_DATA_IN], 0, sizeof(urb[URB_DATA_IN]));
-      urb[URB_DATA_IN].type = USBDEVFS_URB_TYPE_INTERRUPT;
-      urb[URB_DATA_IN].endpoint = dev.ifaces[0].ep_in;
-      urb[URB_DATA_IN].buffer = buf_in;
-      urb[URB_DATA_IN].buffer_length = sizeof(buf_in);
-
-      ret = ioctl(dev.fd, USBDEVFS_SUBMITURB, &urb[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");
-        return 1;
+        break;
       }
+
       data_in_sent = 1;
     }
     if (!dbg_in_sent) {
-      memset(&urb[URB_DBG_IN], 0, sizeof(urb[URB_DBG_IN]));
-      urb[URB_DBG_IN].type = USBDEVFS_URB_TYPE_INTERRUPT;
-      urb[URB_DBG_IN].endpoint = dev.ifaces[1].ep_in;
-      urb[URB_DBG_IN].buffer = buf_dbg;
-      urb[URB_DBG_IN].buffer_length = sizeof(buf_dbg) - 1;
-
-      ret = ioctl(dev.fd, USBDEVFS_SUBMITURB, &urb[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");
-        return 1;
+        break;
       }
+
       dbg_in_sent = 1;
     }
 
@@ -448,11 +653,12 @@ int main(int argc, char *argv[])
     FD_ZERO(&wfds);
     FD_SET(dev.fd, &wfds);
 
-    ret = select(dev.fd + 1, &rfds, &wfds, NULL, NULL);
+    ret = select(dev.fd + 1, &rfds, &wfds, NULL, timeout);
     if (ret < 0) {
       perror("select");
-      return 1;
+      break;
     }
+    timeout = NULL;
 
     /* something from input devices? */
     fixed_input_changed = 0;
@@ -464,19 +670,24 @@ int main(int argc, char *argv[])
     }
 
     /* something from USB? */
-    if (FD_ISSET(dev.fd, &wfds)) {
+    if (FD_ISSET(dev.fd, &wfds))
+    {
       reaped_urb = NULL;
       ret = ioctl(dev.fd, USBDEVFS_REAPURB, &reaped_urb);
       if (ret != 0) {
         if (errno == ENODEV)
           goto dev_close;
         perror("USBDEVFS_REAPURB");
-        return 1;
+        break;
       }
 
       if (reaped_urb != NULL && reaped_urb->status != 0) {
         errno = -reaped_urb->status;
-        perror("urb status");
+        if ((unsigned long)(reaped_urb - urb) < ARRAY_SIZE(urb))
+          fprintf(stderr, "urb #%zu: ", reaped_urb - urb);
+        else
+          fprintf(stderr, "unknown urb: ");
+        perror("");
         if (reaped_urb->status == -EILSEQ) {
           /* this is usually a sign of disconnect.. */
           usleep(250000);
@@ -485,10 +696,60 @@ int main(int argc, char *argv[])
       }
 
       if (reaped_urb == &urb[URB_DATA_IN]) {
-        printf("*data*\n");
+        /* some request from teensy */
+        int count;
+        uint8_t b;
+
+        switch (pkt_in.type) {
+        case PKT_STREAM_REQ:
+          printf("%d/%d\n", frames_sent, frame_count);
+
+          for (i = 0; i < sizeof(pkt_out.data); i++) {
+            pkt_out.data[i * 2 + 0] = 0x33;
+            pkt_out.data[i * 2 + 1] = 0x3f;
+          }
+          if (frames_sent < frame_count) {
+            pkt_out.type = PKT_STREAM_DATA;
+
+            count = frame_count - frames_sent;
+            if (count > sizeof(pkt_out.data) / 2)
+              count = sizeof(pkt_out.data) / 2;
+            for (i = 0; i < count; i++) {
+              /* SCBA RLDU */
+              b = gmv->data[frames_sent][0];
+
+              /* ?0SA 00DU, ?1CB RLDU */
+              pkt_out.data[i * 2 + 0] = (b & 0x13) | ((b >> 2) & 0x20);
+              pkt_out.data[i * 2 + 1] = (b & 0x0f) | ((b >> 1) & 0x30);
+
+              if (gmv->data[frames_sent][1] != 0xff
+                  || gmv->data[frames_sent][2] != 0xff)
+              {
+                fprintf(stderr, "f %d: unhandled byte(s) %02x %02x\n",
+                  frames_sent, gmv->data[frames_sent][1],
+                  gmv->data[frames_sent][2]);
+              }
+
+              frames_sent++;
+            }
+          }
+          else
+            pkt_out.type = PKT_STREAM_END;
+
+          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 URB_DATA_OUT PKT_STREAM_DATA");
+          break;
+
+        default:
+          printf("host: got unknown pkt type: %04x\n", pkt_in.type);
+          break;
+        }
+
         data_in_sent = 0;
       }
-      if (reaped_urb == &urb[URB_DATA_OUT]) {
+      else if (reaped_urb == &urb[URB_DATA_OUT]) {
       }
       else if (reaped_urb == &urb[URB_DBG_IN]) {
         /* debug text */
@@ -497,25 +758,33 @@ int main(int argc, char *argv[])
         dbg_in_sent = 0;
       }
       else {
-        fprintf(stderr, "reaped unknown urb? %p\n", reaped_urb);
+        fprintf(stderr, "reaped unknown urb? %p #%zu\n",
+          reaped_urb, reaped_urb - urb);
       }
     }
 
     /* something to send? */
-    if (fixed_input_changed) {
-      memset(buf_out, 0, sizeof(buf_out));
-      memcpy(buf_out, fixed_input_state, sizeof(fixed_input_state));
-
-      memset(&urb[URB_DATA_OUT], 0, sizeof(urb[URB_DATA_OUT]));
-      urb[URB_DATA_OUT].type = USBDEVFS_URB_TYPE_INTERRUPT;
-      urb[URB_DATA_OUT].endpoint = dev.ifaces[0].ep_out;
-      urb[URB_DATA_OUT].buffer = buf_out;
-      urb[URB_DATA_OUT].buffer_length = sizeof(buf_out);
+    if (gmv != NULL && !enable_sent) {
+      memset(&pkt_out, 0, sizeof(pkt_out));
+      pkt_out.type = PKT_STREAM_ENABLE;
+      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;
+      }
+      enable_sent = 1;
+    }
+    if (gmv == 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 = ioctl(dev.fd, USBDEVFS_SUBMITURB, &urb[URB_DATA_OUT]);
+      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 URB_DATA_OUT");
-        return 1;
+        break;
       }
     }
 
@@ -526,7 +795,9 @@ dev_close:
     dev.fd = -1;
   }
 
-  return 0;
+  enable_echo(1);
+
+  return ret;
 }
 
 // vim: ts=2:sw=2:expandtab