118 lines
4.7 KiB
C
118 lines
4.7 KiB
C
/**********************************************************************************************/
|
|
/* The MIT License */
|
|
/* */
|
|
/* Copyright 2016-2016 Twitch Interactive, Inc. or its affiliates. All Rights Reserved. */
|
|
/* */
|
|
/* 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 <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include <linux/input.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include "caption.h"
|
|
#include "flv.h"
|
|
|
|
char charcode[] = {
|
|
0, 0, '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '-', '=', 0, 0,
|
|
'Q', 'W', 'E', 'R', 'T', 'Y', 'U', 'I', 'O', 'P', '[', ']','\n', 0, 'A', 'S',
|
|
'D', 'F', 'G', 'H', 'J', 'K', 'L', ';', '\'', '`', 0, '\\', 'Z', 'X', 'C', 'V',
|
|
'B', 'N', 'M', ',', '.', '/', 0, '*', 0, ' ', 0, 0, 0, 0, 0, 0,
|
|
};
|
|
|
|
int data_ready (int fd)
|
|
{
|
|
fd_set set;
|
|
struct timeval timeout = {0,0};
|
|
FD_ZERO (&set);
|
|
FD_SET (fd,&set);
|
|
int cnt = select (fd+1, &set, 0, 0, &timeout);
|
|
FD_ZERO (&set);
|
|
return (0 < cnt);
|
|
}
|
|
|
|
#define MAX_CAP_LENGTH (SCREEN_ROWS*SCREEN_COLS)
|
|
int main (int argc, char** argv)
|
|
{
|
|
int fd;
|
|
ssize_t n;
|
|
flvtag_t tag;
|
|
struct input_event ev;
|
|
int has_audio, has_video;
|
|
const char* dev = argv[1];
|
|
char text[MAX_CAP_LENGTH+1];
|
|
memset (text,0,MAX_CAP_LENGTH+1);
|
|
|
|
FILE* flv = flv_open_read ("-");
|
|
FILE* out = flv_open_write ("-");
|
|
fd = open (dev, O_RDONLY);
|
|
|
|
if (fd == -1) {
|
|
fprintf (stderr, "Cannot open %s: %s.\n", dev, strerror (errno));
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (!flv_read_header (flv,&has_audio,&has_video)) {
|
|
fprintf (stderr,"%s is not an flv file\n", argv[1]);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (!flv_write_header (out,has_audio,has_video)) {
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
flvtag_init (&tag);
|
|
|
|
while (flv_read_tag (flv,&tag)) {
|
|
if (flvtag_avcpackettype_nalu == flvtag_avcpackettype (&tag)) {
|
|
if (data_ready (fd)) {
|
|
n = read (fd, &ev, sizeof ev);
|
|
|
|
if (n == (ssize_t)-1) {
|
|
if (errno == EINTR) {
|
|
continue;
|
|
} else {
|
|
break;
|
|
}
|
|
} else if (n != sizeof ev) {
|
|
errno = EIO;
|
|
break;
|
|
}
|
|
|
|
int len = strlen (text);
|
|
char c = (EV_KEY == ev.type && 1 == ev.value && ev.code < 64) ? charcode[ev.code] : 0;
|
|
|
|
if (0 < len && '\n' == c) {
|
|
fprintf (stderr,"='%s'\n", text);
|
|
flvtag_addcaption (&tag, text);
|
|
memset (text,0,MAX_CAP_LENGTH+1);
|
|
} else if (0 != c && len < MAX_CAP_LENGTH) {
|
|
text[len] = c;
|
|
}
|
|
}
|
|
}
|
|
|
|
flv_write_tag (out,&tag);
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|