implement stream restart on r key
[megadrive.git] / host / main.c
CommitLineData
932302ef 1#include <stdio.h>
61172125 2#include <stdlib.h>
932302ef 3#include <string.h>
4#include <stdint.h>
932302ef 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>
f359b935 12#include <signal.h>
13#include <termios.h>
932302ef 14#include <errno.h>
15#include <linux/usbdevice_fs.h>
16#include <linux/usb/ch9.h>
1cb2822f 17#include <linux/input.h>
f20de073 18#include "../pkts.h"
932302ef 19
20#define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
21
932302ef 22struct 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 */
31static 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
99next:
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
112found:
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
228out:
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
f359b935 239static 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)
de3f807e 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 }
f359b935 269
270 ret = tcsetattr(fd, TCSANOW, &tty);
271 if (ret != 0) {
272 perror("tcsetattr");
273 goto out;
274 }
275
276 retval = 0;
277out:
278 close(fd);
279
280 return retval;
281}
282
283static void signal_handler(int sig)
284{
285 enable_echo(1);
286 signal(sig, SIG_DFL);
287 raise(sig);
288}
289
1cb2822f 290/* ?0SA 00DU, ?1CB RLDU */
291#define STATE_BYTES 2
292
293static uint8_t fixed_input_state[STATE_BYTES] = { 0x33, 0x3f };
294
295enum mdbtn {
296 MDBTN_UP = 1,
297 MDBTN_DOWN,
298 MDBTN_LEFT,
299 MDBTN_RIGHT,
300 MDBTN_A,
301 MDBTN_B,
302 MDBTN_C,
303 MDBTN_START,
304};
305
306static const enum mdbtn evdev_md_map[KEY_CNT] = {
307 [KEY_UP] = MDBTN_UP,
308 [KEY_DOWN] = MDBTN_DOWN,
309 [KEY_LEFT] = MDBTN_LEFT,
310 [KEY_RIGHT] = MDBTN_RIGHT,
311 [KEY_HOME] = MDBTN_A,
312 [KEY_PAGEDOWN] = MDBTN_B,
313 [KEY_END] = MDBTN_C,
314 [KEY_LEFTALT] = MDBTN_START,
315};
316
317int do_evdev_input(int fd)
318{
319 uint8_t old_state[STATE_BYTES];
320 uint8_t changed_bits[STATE_BYTES] = { 0, };
321 struct input_event ev;
322 enum mdbtn mdbtn;
323 int i, ret;
324
325 ret = read(fd, &ev, sizeof(ev));
326 if (ret != sizeof(ev)) {
327 fprintf(stderr, "evdev read %d/%zd: ", ret, sizeof(ev));
328 perror("");
329 return 0;
330 }
331
332 if (ev.type != EV_KEY)
333 return 0;
334
335 if (ev.value != 0 && ev.value != 1)
336 return 0;
337
338 if ((uint32_t)ev.code >= ARRAY_SIZE(evdev_md_map)) {
339 fprintf(stderr, "evdev read bad key: %u\n", ev.code);
340 return 0;
341 }
342
343 mdbtn = evdev_md_map[ev.code];
344 if (mdbtn == 0)
345 return 0;
346
347 memcpy(old_state, fixed_input_state, STATE_BYTES);
348
349 /* ?0SA 00DU, ?1CB RLDU */
350 switch (mdbtn) {
351 case MDBTN_UP:
352 changed_bits[0] = 0x01;
353 changed_bits[1] = 0x01;
354 break;
355 case MDBTN_DOWN:
356 changed_bits[0] = 0x02;
357 changed_bits[1] = 0x02;
358 break;
359 case MDBTN_LEFT:
360 changed_bits[0] = 0x00;
361 changed_bits[1] = 0x04;
362 break;
363 case MDBTN_RIGHT:
364 changed_bits[0] = 0x00;
365 changed_bits[1] = 0x08;
366 break;
367 case MDBTN_A:
368 changed_bits[0] = 0x10;
369 changed_bits[1] = 0x00;
370 break;
371 case MDBTN_B:
372 changed_bits[0] = 0x00;
373 changed_bits[1] = 0x10;
374 break;
375 case MDBTN_C:
376 changed_bits[0] = 0x00;
377 changed_bits[1] = 0x20;
378 break;
379 case MDBTN_START:
380 changed_bits[0] = 0x20;
381 changed_bits[1] = 0x00;
382 break;
383 }
384
385 if (ev.value) {
386 // key press
387 for (i = 0; i < STATE_BYTES; i++)
388 fixed_input_state[i] &= ~changed_bits[i];
389 }
390 else {
391 // key release
392 for (i = 0; i < STATE_BYTES; i++)
393 fixed_input_state[i] |= changed_bits[i];
394 }
395
396 return memcmp(old_state, fixed_input_state, STATE_BYTES) ? 1 : 0;
397}
398
f20de073 399struct gmv_tas {
400 char sig[15];
401 char ver;
402 uint32_t rerecord_count;
403 char ctrl1;
404 char ctrl2;
405 uint16_t flags;
406 char name[40];
407 uint8_t data[0][3];
408};
409
01f5cc9e 410static uint8_t *import_gmv(FILE *f, long size, int *frame_count)
411{
412 struct gmv_tas *gmv;
413 uint8_t *out;
414 int ret;
415 int i;
416
417 if (size < (long)sizeof(*gmv)) {
418 fprintf(stderr, "bad gmv size: %ld\n", size);
419 return NULL;
420 }
421
422 gmv = malloc(size);
423 if (gmv == NULL) {
424 fprintf(stderr, "OOM?\n");
425 return NULL;
426 }
427 ret = fread(gmv, 1, size, f);
428 if (ret != size) {
429 fprintf(stderr, "fread %d/%ld: ", ret, size);
430 perror("");
431 return NULL;
432 }
433
434 *frame_count = (size - sizeof(*gmv)) / sizeof(gmv->data[0]);
435
436 /* check the GMV.. */
437 if (*frame_count <= 0 || size != sizeof(*gmv) + *frame_count * 3) {
438 fprintf(stderr, "broken gmv? frames=%d\n", *frame_count);
439 return NULL;
440 }
441
442 if (strncmp(gmv->sig, "Gens Movie TEST", 15) != 0) {
443 fprintf(stderr, "bad GMV sig\n");
444 return NULL;
445 }
446 if (gmv->ctrl1 != '3') {
447 fprintf(stderr, "unhandled controlled config: '%c'\n", gmv->ctrl1);
448 //return NULL;
449 }
450 if (gmv->ver >= 'A') {
451 if (gmv->flags & 0x40) {
452 fprintf(stderr, "unhandled flag: movie requires a savestate\n");
453 return NULL;
454 }
455 if (gmv->flags & 0x20) {
456 fprintf(stderr, "unhandled flag: 3-player movie\n");
457 return NULL;
458 }
459 if (gmv->flags & ~0x80) {
460 //fprintf(stderr, "unhandled flag(s): %04x\n", gmv->flags);
461 //return 1;
462 }
463 }
464 gmv->name[39] = 0;
465 printf("loaded GMV: %s\n", gmv->name);
466 printf("%d frames, %u rerecords\n",
467 *frame_count, gmv->rerecord_count);
468
469 out = malloc(*frame_count * sizeof(out[0]));
470 if (out == NULL) {
471 fprintf(stderr, "OOM?\n");
472 return NULL;
473 }
474
475 for (i = 0; i < *frame_count; i++) {
476 out[i] = gmv->data[i][0];
477
478 if (gmv->data[i][1] != 0xff || gmv->data[i][2] != 0xff)
479 {
480 fprintf(stderr, "f %d: unhandled byte(s) %02x %02x\n",
481 i, gmv->data[i][1], gmv->data[i][2]);
482 }
483 }
484
485 return out;
486}
487
488static int do_bkm_char(char c, char expect, uint8_t *val, int bit)
489{
490 if (c == expect) {
491 *val &= ~(1 << bit);
492 return 0;
493 }
494 if (c == '.')
495 return 0;
496
497 fprintf(stderr, "unexpected bkm char: '%c' instead of '%c'\n",
498 c, expect);
499 return 1;
500}
501
502static uint8_t *import_bkm(FILE *f, int *frame_count)
503{
504 uint8_t *out = NULL, val;
505 int count = 0;
506 int alloc = 0;
507 int line = 0;
508 char buf[256];
509 const char *r;
510 char *p;
511 int i;
512
513 while ((p = fgets(buf, sizeof(buf), f)) != NULL) {
514 line++;
515 if (p[0] != '|')
516 continue;
517
518 if (strlen(p) < 30)
519 goto unhandled_line;
520 if (p[30] != '\r' && p[30] != '\n')
521 goto unhandled_line;
522 p[30] = 0;
523
524 if (count >= alloc) {
525 alloc = alloc * 2 + 64;
526 out = realloc(out, alloc * sizeof(out[0]));
527 if (out == NULL) {
528 fprintf(stderr, "OOM?\n");
529 return NULL;
530 }
531 }
532
533 val = 0xff;
534
535 if (strncmp(p, "|.|", 3) != 0)
536 goto unhandled_line;
537 p += 3;
538
539 const char ref[] = "UDLRABCS";
540 for (r = ref, i = 0; *r != 0; p++, r++, i++) {
541 if (do_bkm_char(*p, *r, &val, i))
542 goto unhandled_line;
543 }
544
545 if (strcmp(p, "....|............||") != 0)
546 goto unhandled_line;
547
548 out[count++] = val;
549 continue;
550
551unhandled_line:
552 fprintf(stderr, "unhandled bkm line %d: '%s'\n", line, buf);
553 return NULL;
554 }
555
556 printf("loaded bkm, %d frames\n", count);
557 *frame_count = count;
558 return out;
559}
560
f20de073 561static int submit_urb(int fd, struct usbdevfs_urb *urb, int ep,
562 void *buf, size_t buf_size)
563{
564 memset(urb, 0, sizeof(*urb));
565 urb->type = USBDEVFS_URB_TYPE_INTERRUPT;
566 urb->endpoint = ep;
567 urb->buffer = buf;
568 urb->buffer_length = buf_size;
569
570 return ioctl(fd, USBDEVFS_SUBMITURB, urb);
571}
572
932302ef 573enum my_urbs {
574 URB_DATA_IN,
575 URB_DATA_OUT,
576 URB_DBG_IN,
577 URB_CNT
578};
579
61172125 580static void missing_arg(int a)
581{
582 fprintf(stderr, "missing arg: %d\n", a);
583 exit(1);
584}
585
932302ef 586int main(int argc, char *argv[])
587{
588 struct teensy_dev dev;
589 struct usbdevfs_urb urb[URB_CNT];
590 struct usbdevfs_urb *reaped_urb;
1cb2822f 591 int fixed_input_changed;
592 int evdev_fds[16];
593 int evdev_fd_cnt = 0;
594 int evdev_support;
932302ef 595 int wait_device = 0;
596 int dbg_in_sent = 0;
597 int data_in_sent = 0;
1cb2822f 598 fd_set rfds, wfds;
61172125 599 const char *tasfn = NULL;
01f5cc9e 600 uint8_t *tas_data = NULL;
4af6d4e5 601 int use_readinc = 0; // frame increment on read
61172125 602 int tas_skip = 0;
f20de073 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;
1cb2822f 611 int i, ret;
612 int fd;
613
614 for (i = 1; i < argc; i++) {
f20de073 615 if (argv[i][0] == '-') {
61172125 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;
4af6d4e5 629 case 'r':
630 use_readinc = 1;
631 continue;
61172125 632 default:
f20de073 633 fprintf(stderr, "bad arg: %s\n", argv[i]);
634 return 1;
635 }
f20de073 636 }
1cb2822f 637 if (evdev_fd_cnt >= ARRAY_SIZE(evdev_fds)) {
638 fprintf(stderr, "too many evdevs\n");
639 break;
640 }
f20de073 641 fd = open(argv[i], O_RDONLY);
1cb2822f 642 if (fd == -1) {
643 fprintf(stderr, "open %s: ", argv[i]);
644 perror("");
645 continue;
646 }
647 evdev_support = 0;
f20de073 648 ret = ioctl(fd, EVIOCGBIT(0, sizeof(evdev_support)),
1cb2822f 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 }
932302ef 659
f20de073 660 if (tasfn != NULL) {
01f5cc9e 661 const char *ext;
f20de073 662 long size;
663 FILE *f;
01f5cc9e 664
f20de073 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);
01f5cc9e 675 if (size <= 0) {
676 fprintf(stderr, "bad size: %ld\n", size);
f20de073 677 return 1;
678 }
01f5cc9e 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);
f20de073 692 return 1;
693 }
694 fclose(f);
f20de073 695
01f5cc9e 696 if (tas_data == NULL) {
697 fprintf(stderr, "failed fo parse %s\n", tasfn);
f20de073 698 return 1;
699 }
700
61172125 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;
01f5cc9e 708 memmove(&tas_data[0], &tas_data[tas_skip],
709 sizeof(tas_data[0]) * frame_count);
61172125 710 }
711 else {
01f5cc9e 712 tas_data = realloc(tas_data,
713 (frame_count - tas_skip) * sizeof(tas_data[0]));
714 if (tas_data == NULL) {
61172125 715 fprintf(stderr, "OOM?\n");
716 return 1;
717 }
01f5cc9e 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);
61172125 721 frame_count -= tas_skip;
722 }
723 }
f20de073 724 }
725
f359b935 726 enable_echo(0);
727 signal(SIGINT, signal_handler);
728
932302ef 729 dev.fd = -1;
730
731 while (1)
732 {
733 if (dev.fd == -1) {
734 ret = find_device(&dev, 0x16C0, 0x0486);
735 if (ret < 0)
f359b935 736 break;
932302ef 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 data_in_sent = 0;
749 dbg_in_sent = 0;
f20de073 750 enable_sent = 0;
751 frames_sent = 0;
752
753 /* we wait first, then send commands, but if teensy
754 * is started already, it won't send anything */
755 tout.tv_sec = 1;
756 tout.tv_usec = 0;
757 timeout = &tout;
932302ef 758 }
759
760 if (!data_in_sent) {
f20de073 761 memset(&pkt_in, 0, sizeof(pkt_in));
762 ret = submit_urb(dev.fd, &urb[URB_DATA_IN], dev.ifaces[0].ep_in,
763 &pkt_in, sizeof(pkt_in));
932302ef 764 if (ret != 0) {
765 perror("USBDEVFS_SUBMITURB URB_DATA_IN");
f359b935 766 break;
932302ef 767 }
f20de073 768
932302ef 769 data_in_sent = 1;
770 }
771 if (!dbg_in_sent) {
f20de073 772 ret = submit_urb(dev.fd, &urb[URB_DBG_IN], dev.ifaces[1].ep_in,
773 buf_dbg, sizeof(buf_dbg) - 1);
932302ef 774 if (ret != 0) {
775 perror("USBDEVFS_SUBMITURB URB_DBG_IN");
f359b935 776 break;
932302ef 777 }
f20de073 778
932302ef 779 dbg_in_sent = 1;
780 }
781
1cb2822f 782 FD_ZERO(&rfds);
de3f807e 783 FD_SET(STDIN_FILENO, &rfds);
1cb2822f 784 for (i = 0; i < evdev_fd_cnt; i++)
785 FD_SET(evdev_fds[i], &rfds);
786
932302ef 787 FD_ZERO(&wfds);
788 FD_SET(dev.fd, &wfds);
789
f20de073 790 ret = select(dev.fd + 1, &rfds, &wfds, NULL, timeout);
932302ef 791 if (ret < 0) {
792 perror("select");
f359b935 793 break;
932302ef 794 }
f20de073 795 timeout = NULL;
932302ef 796
de3f807e 797 /* sometihng form stdin? */
1cb2822f 798 /* something from input devices? */
de3f807e 799 if (FD_ISSET(STDIN_FILENO, &rfds)) {
800 char c = 0;
801 ret = read(STDIN_FILENO, &c, 1);
802 if (ret <= 0) {
803 perror("read stdin");
804 break;
805 }
806
807 switch (c) {
808 case 'r':
809 enable_sent = 0;
810 break;
811 }
812 }
813
1cb2822f 814 fixed_input_changed = 0;
815 for (i = 0; i < evdev_fd_cnt; i++) {
816 if (FD_ISSET(evdev_fds[i], &rfds)) {
817 fixed_input_changed |=
818 do_evdev_input(evdev_fds[i]);
819 }
820 }
821
822 /* something from USB? */
f20de073 823 if (FD_ISSET(dev.fd, &wfds))
824 {
932302ef 825 reaped_urb = NULL;
826 ret = ioctl(dev.fd, USBDEVFS_REAPURB, &reaped_urb);
827 if (ret != 0) {
828 if (errno == ENODEV)
829 goto dev_close;
830 perror("USBDEVFS_REAPURB");
f359b935 831 break;
932302ef 832 }
833
834 if (reaped_urb != NULL && reaped_urb->status != 0) {
835 errno = -reaped_urb->status;
f20de073 836 if ((unsigned long)(reaped_urb - urb) < ARRAY_SIZE(urb))
837 fprintf(stderr, "urb #%zu: ", reaped_urb - urb);
838 else
839 fprintf(stderr, "unknown urb: ");
840 perror("");
932302ef 841 if (reaped_urb->status == -EILSEQ) {
842 /* this is usually a sign of disconnect.. */
843 usleep(250000);
844 goto dev_close;
845 }
846 }
847
848 if (reaped_urb == &urb[URB_DATA_IN]) {
f20de073 849 /* some request from teensy */
850 int count;
851 uint8_t b;
852
853 switch (pkt_in.type) {
854 case PKT_STREAM_REQ:
4af6d4e5 855 printf("%d/%d/%d\n", pkt_in.req.frame,
856 frames_sent, frame_count);
f20de073 857
858 for (i = 0; i < sizeof(pkt_out.data); i++) {
859 pkt_out.data[i * 2 + 0] = 0x33;
860 pkt_out.data[i * 2 + 1] = 0x3f;
861 }
862 if (frames_sent < frame_count) {
863 pkt_out.type = PKT_STREAM_DATA;
864
865 count = frame_count - frames_sent;
866 if (count > sizeof(pkt_out.data) / 2)
867 count = sizeof(pkt_out.data) / 2;
868 for (i = 0; i < count; i++) {
869 /* SCBA RLDU */
01f5cc9e 870 b = tas_data[frames_sent];
f20de073 871
872 /* ?0SA 00DU, ?1CB RLDU */
873 pkt_out.data[i * 2 + 0] = (b & 0x13) | ((b >> 2) & 0x20);
874 pkt_out.data[i * 2 + 1] = (b & 0x0f) | ((b >> 1) & 0x30);
f20de073 875 frames_sent++;
876 }
877 }
878 else
879 pkt_out.type = PKT_STREAM_END;
880
881 ret = submit_urb(dev.fd, &urb[URB_DATA_OUT],
882 dev.ifaces[0].ep_out, &pkt_out, sizeof(pkt_out));
883 if (ret != 0)
884 perror("USBDEVFS_SUBMITURB URB_DATA_OUT PKT_STREAM_DATA");
885 break;
886
887 default:
888 printf("host: got unknown pkt type: %04x\n", pkt_in.type);
889 break;
890 }
891
932302ef 892 data_in_sent = 0;
893 }
f20de073 894 else if (reaped_urb == &urb[URB_DATA_OUT]) {
1cb2822f 895 }
932302ef 896 else if (reaped_urb == &urb[URB_DBG_IN]) {
897 /* debug text */
898 buf_dbg[reaped_urb->actual_length] = 0;
899 printf("%s", buf_dbg);
900 dbg_in_sent = 0;
901 }
902 else {
f20de073 903 fprintf(stderr, "reaped unknown urb? %p #%zu\n",
904 reaped_urb, reaped_urb - urb);
932302ef 905 }
906 }
1cb2822f 907
908 /* something to send? */
01f5cc9e 909 if (tas_data != NULL && !enable_sent) {
f20de073 910 memset(&pkt_out, 0, sizeof(pkt_out));
911 pkt_out.type = PKT_STREAM_ENABLE;
4af6d4e5 912 pkt_out.start.use_readinc = use_readinc;
913
f20de073 914 ret = submit_urb(dev.fd, &urb[URB_DATA_OUT], dev.ifaces[0].ep_out,
915 &pkt_out, sizeof(pkt_out));
916 if (ret != 0) {
917 perror("USBDEVFS_SUBMITURB PKT_STREAM_ENABLE");
918 continue;
919 }
920 enable_sent = 1;
de3f807e 921 frames_sent = 0;
f20de073 922 }
01f5cc9e 923 if (tas_data == NULL && fixed_input_changed) {
f20de073 924 memset(&pkt_out, 0, sizeof(pkt_out));
925 pkt_out.type = PKT_FIXED_STATE;
926 memcpy(pkt_out.data, fixed_input_state, sizeof(fixed_input_state));
1cb2822f 927
f20de073 928 ret = submit_urb(dev.fd, &urb[URB_DATA_OUT], dev.ifaces[0].ep_out,
929 &pkt_out, sizeof(pkt_out));
1cb2822f 930 if (ret != 0) {
931 perror("USBDEVFS_SUBMITURB URB_DATA_OUT");
f359b935 932 break;
1cb2822f 933 }
934 }
935
932302ef 936 continue;
937
938dev_close:
939 close(dev.fd);
940 dev.fd = -1;
941 }
942
f359b935 943 enable_echo(1);
944
945 return ret;
932302ef 946}
947
948// vim: ts=2:sw=2:expandtab