+++ /dev/null
-#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;
- 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 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;
- 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? */
- fixed_input_changed = 0;
- 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;
-
- 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;
- }
- 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