Add support for RTP output.

ChangeLog

@@ -2,6 +2,8 @@ next : Version next
  * Add --output-tos (-g) option used to set output TOS value.
  * Add --input-dump (-W) option used save input stream in file.
  * Add --pid-report (-j) option used to turn on PID reporting.
+ * Add RTP output enabled by --output-rtp (-r) option.
+ * Add --output-rtp-ssrc (-k) option used to set RTP SSRC.
 
 2011-12-23 : Version 5.0
  * Add --bench (-b) option that benchmarks libdvbcsa decryption.

Makefile

@@ -45,7 +45,7 @@ tsdecrypt_SRC = data.c \
  tables.c \
  notify.c \
  tsdecrypt.c
-tsdecrypt_LIBS = -lcrypto -ldvbcsa -lpthread
+tsdecrypt_LIBS = -lcrypto -ldvbcsa -lpthread -lrt
 tsdecrypt_OBJS = $(FUNCS_LIB) $(TS_LIB) $(tsdecrypt_SRC:.c=.o)
 
 ifeq "$(shell uname -s)" "Linux"

README

@@ -47,6 +47,8 @@ Output options:
  -o --output-intf <addr>    | Set multicast output interface. Default: 0.0.0.0
  -t --output-ttl <ttl>      | Set multicast ttl. Default: 1
  -g --output-tos <tos>      | Set TOS value of output packets. Default: none
+ -r --output-rtp            | Enable RTP output.
+ -k --output-rtp-ssrc <id>  | Set RTP SSRC. Default: 0
  -p --no-output-filter      | Disable output filtering. Default: enabled
  -y --output-nit-pass       | Pass through NIT.
  -w --output-eit-pass       | Pass through EIT (EPG).

data.h

@@ -244,6 +244,9 @@ struct ts {
 	int					ecm_cw_log;
 
 	int					rtp_input;
+	int					rtp_output;
+	uint32_t			rtp_ssrc;
+	uint16_t			rtp_seqnum;
 
 	struct io			input;
 	struct io			output;

process.c

@@ -204,7 +204,37 @@ void *decode_thread(void *_ts) {
 static inline void output_write(struct ts *ts, uint8_t *data, unsigned int data_size) {
 	if (!data)
 		return;
-	write(ts->output.fd, data, data_size);
+	if (!ts->rtp_output) {
+		write(ts->output.fd, data, data_size);
+	} else {
+		struct iovec iov[2];
+		uint8_t rtp_header[12];
+		uint32_t rtime = get_time() * 9 / 100;
+
+		ts->rtp_seqnum++;
+
+		rtp_header[ 0] = 0x80;
+		rtp_header[ 1] = 33; // MPEG TS rtp payload type
+		rtp_header[ 2] = ts->rtp_seqnum >> 8;
+		rtp_header[ 3] = ts->rtp_seqnum & 0xff;
+		rtp_header[ 4] = (rtime >> 24) & 0xff;
+		rtp_header[ 5] = (rtime >> 16) & 0xff;
+		rtp_header[ 6] = (rtime >>  8) & 0xff;
+		rtp_header[ 7] =  rtime        & 0xff;
+
+		rtp_header[ 8] = (ts->rtp_ssrc >> 24) & 0xff;
+		rtp_header[ 9] = (ts->rtp_ssrc >> 16) & 0xff;
+		rtp_header[10] = (ts->rtp_ssrc >>  8) & 0xff;
+		rtp_header[11] =  ts->rtp_ssrc        & 0xff;
+
+		iov[0].iov_base = rtp_header;
+		iov[0].iov_len  = sizeof(rtp_header);
+
+		iov[1].iov_base = data;
+		iov[1].iov_len  = data_size;
+
+		writev(ts->output.fd, iov, 2);
+	}
 }
 
 void *write_thread(void *_ts) {

tsdecrypt.1

@@ -106,6 +106,13 @@ Set multicast ttl. The default value is \fB1\fR.
 \fB\-g\fR, \fB\-\-output\-tos\fR
 Set TOS value of output packets. The default is not to set any specific TOS.
 .TP
+\fB\-r\fR, \fB\-\-output\-rtp\fR
+Enable RTP output. The default output is standart MPEG TS over UDP, this
+option enables tsdecrypt to output RTP packets.
+.TP
+\fB\-k\fR, \fB\-\-output\-rtp\-ssrc\fR <ssrc>
+Set RTP SSRC field to <ssrc>. By default is is set to \fB0\fR.
+.TP
 \fB\-p\fR, \fB\-\-no\-output\-filter\fR
 Disable output filtering. By default the output filter is enabled and only
 PAT/PMT/SDT and data packets are left in the output. Everything else not

tsdecrypt.c

@@ -27,6 +27,7 @@
 #include <syslog.h>
 
 #include <dvbcsa/dvbcsa.h>
+#include <openssl/rand.h>
 
 #include "libfuncs/libfuncs.h"
 
@@ -103,9 +104,9 @@ void run_benchmark(void) {
 	puts("* Done *");
 }
 
-static const char short_options[] = "i:d:N:Sl:L:I:RzM:W:O:o:t:g:pwxyc:C:A:s:U:P:B:eZ:Ef:X:H:G:KJ:D:jbhV";
+static const char short_options[] = "i:d:N:Sl:L:I:RzM:W:O:o:t:rk:g:pwxyc:C:A:s:U:P:B:eZ:Ef:X:H:G:KJ:D:jbhV";
 
-// Unused short options: FQTYakmnqruv0123456789
+// Unused short options: FQTYamnquv0123456789
 static const struct option long_options[] = {
 	{ "ident",				required_argument, NULL, 'i' },
 	{ "daemon",				required_argument, NULL, 'd' },
@@ -123,6 +124,8 @@ static const struct option long_options[] = {
 	{ "output",				required_argument, NULL, 'O' },
 	{ "output-intf",		required_argument, NULL, 'o' },
 	{ "output-ttl",			required_argument, NULL, 't' },
+	{ "output-rtp",			no_argument,       NULL, 'r' },
+	{ "output-rtp-ssrc",	required_argument, NULL, 'k' },
 	{ "output-tos",			required_argument, NULL, 'g' },
 	{ "output-filter",		no_argument,       NULL, 'p' },
 	{ "no-output-filter",	no_argument,       NULL, 'p' },
@@ -191,6 +194,8 @@ static void show_help(struct ts *ts) {
 	printf("                            .    -O -              (write to stdout) (default)\n");
 	printf(" -o --output-intf <addr>    | Set multicast output interface. Default: %s\n", inet_ntoa(ts->output.intf));
 	printf(" -t --output-ttl <ttl>      | Set multicast ttl. Default: %d\n", ts->output.ttl);
+	printf(" -r --output-rtp            | Enable RTP output.\n");
+	printf(" -k --output-rtp-ssrc <id>  | Set RTP SSRC. Default: %u\n", ts->rtp_ssrc);
 	printf(" -g --output-tos <tos>      | Set TOS value of output packets. Default: none\n");
 	printf(" -p --no-output-filter      | Disable output filtering. Default: %s\n", ts->pid_filter ? "enabled" : "disabled");
 	printf(" -y --output-nit-pass       | Pass through NIT.\n");
@@ -331,6 +336,12 @@ static void parse_options(struct ts *ts, int argc, char **argv) {
 			case 't':
 				ts->output.ttl = atoi(optarg);
 				break;
+			case 'r':
+				ts->rtp_output = 1;
+				break;
+			case 'k':
+				ts->rtp_ssrc = strtoul(optarg, NULL, 0);
+				break;
 			case 'g':
 				ts->output.tos = (uint8_t)strtol(optarg, NULL, 0);
 				break;
@@ -539,11 +550,19 @@ static void parse_options(struct ts *ts, int argc, char **argv) {
 	if (!ts->emm_only)
 	{
 		if (ts->output.type == NET_IO) {
-			ts_LOGf("Output addr: udp://%s:%u/\n", inet_ntoa(ts->output.addr), ts->output.port);
+			ts_LOGf("Output addr: %s://%s:%u/\n",
+				ts->rtp_output ? "rtp" : "udp",
+				inet_ntoa(ts->output.addr), ts->output.port);
 			ts_LOGf("Output intf: %s\n", inet_ntoa(ts->output.intf));
 			ts_LOGf("Output ttl : %d\n", ts->output.ttl);
 			if (ts->output.tos > -1)
 				ts_LOGf("Output TOS : %u (0x%02x)\n", ts->output.tos, ts->output.tos);
+			if (ts->rtp_output) {
+				ts_LOGf("RTP SSRC   : %u (0x%04x)\n",
+					ts->rtp_ssrc, ts->rtp_ssrc);
+				// It is recommended that RTP seqnum starts with random number
+				RAND_bytes((unsigned char *)&(ts->rtp_seqnum), 2);
+			}
 		} else if (ts->output.type == FILE_IO) {
 			ts_LOGf("Output file: %s\n", ts->output.fd == 1 ? "STDOUT" : ts->output.fname);
 		}

util.c

@@ -16,7 +16,9 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  */
 
+#include <time.h>
 #include <string.h>
+#include <errno.h>
 
 #include "util.h"
 
@@ -154,3 +156,10 @@ int decode_hex_string(char *hex, uint8_t *bin, int asc_len) {
 	}
 	return asc_len / 2;
 }
+
+int64_t get_time(void) {
+	struct timespec ts;
+	if (clock_gettime(CLOCK_MONOTONIC, &ts) == EINVAL) // Shouldn't happen on modern Linux
+		clock_gettime(CLOCK_REALTIME, &ts);
+	return ts.tv_sec * 1000000ll + ts.tv_nsec / 1000;
+}

util.h

@@ -27,5 +27,6 @@ uint8_t *init_4l(uint32_t val, uint8_t *b);
 uint8_t *init_2b(uint32_t val, uint8_t *b);
 void set_thread_name(char *thread_name);
 int decode_hex_string(char *hex, uint8_t *bin, int asc_len);
+int64_t get_time(void);
 
 #endif