123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930 |
- /*
- * tsdecrypt
- * Copyright (C) 2011-2012 Unix Solutions Ltd.
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License version 2
- * as published by the Free Software Foundation.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License (COPYING file) for more details.
- *
- */
- #include <stdlib.h>
- #include <unistd.h>
- #include <getopt.h>
- #include <string.h>
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <signal.h>
- #include <fcntl.h>
- #include <errno.h>
- #include <syslog.h>
- #include <sys/resource.h>
-
- #include <openssl/rand.h>
-
- #include "libfuncs/libfuncs.h"
-
- #include "data.h"
- #include "util.h"
- #include "csa.h"
- #include "camd.h"
- #include "process.h"
- #include "udp.h"
- #include "notify.h"
-
- #define FIRST_REPORT_SEC 3
-
- #define PROGRAM_NAME "tsdecrypt"
- static const char *program_id = PROGRAM_NAME " v" VERSION " (" GIT_VER ", build " BUILD_ID " " DLIB ")";
-
- static int keep_running = 1;
- static FILE *log_file = NULL;
- static char *log_filename = NULL;
- static int local_syslog = 0;
- static int remote_syslog = 0;
-
- static void do_log(FILE *f, time_t now, const char *msg) {
- char date[64];
- struct tm tm;
- localtime_r(&now, &tm);
- strftime(date, sizeof(date), "%F %H:%M:%S", localtime_r(&now, &tm));
- fprintf(f, "%s | %s", date, msg);
- }
-
- static void LOG_func(const char *msg) {
- time_t now = time(NULL);
- do_log(stderr, now, msg);
- if (log_file)
- do_log(log_file, now, msg);
- if (local_syslog)
- syslog(LOG_INFO, msg, strlen(msg));
- if (remote_syslog)
- LOG(msg);
- }
-
- static const char short_options[] = "i:d:N:Sl:L:F:I:RzM:T:W:O:o:t:rk:g:upwxyc:C:Y:Q:A:s:U:P:B:eZ:Ef:X:H:G:KJ:D:jbhV";
-
- // Unused short options: amnqv0123456789
- static const struct option long_options[] = {
- { "ident", required_argument, NULL, 'i' },
- { "daemon", required_argument, NULL, 'd' },
- { "syslog", no_argument, NULL, 'S' },
- { "syslog-host", required_argument, NULL, 'l' },
- { "syslog-port", required_argument, NULL, 'L' },
- { "log-file", required_argument, NULL, 'F' },
- { "notify-program", required_argument, NULL, 'N' },
-
- { "input", required_argument, NULL, 'I' },
- { "input-rtp", no_argument, NULL, 'R' },
- { "input-ignore-disc", no_argument, NULL, 'z' },
- { "input-service", required_argument, NULL, 'M' },
- { "input-buffer", required_argument, NULL, 'T' },
- { "input-dump", required_argument, NULL, 'W' },
-
- { "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' },
- { "no-output-on-error", no_argument, NULL, 'u' },
- { "no-output-filter", no_argument, NULL, 'p' },
- { "output-nit-pass", no_argument, NULL, 'y' },
- { "output-eit-pass", no_argument, NULL, 'w' },
- { "output-tdt-pass", no_argument, NULL, 'x' },
-
- { "ca-system", required_argument, NULL, 'c' },
- { "caid", required_argument, NULL, 'C' },
- { "const-cw", required_argument, NULL, 'Y' },
- { "biss-key", required_argument, NULL, 'Q' },
-
- { "camd-proto", required_argument, NULL, 'A' },
- { "camd-server", required_argument, NULL, 's' },
- { "camd-user", required_argument, NULL, 'U' },
- { "camd-pass", required_argument, NULL, 'P' },
- { "camd-des-key", required_argument, NULL, 'B' },
-
- { "emm", no_argument, NULL, 'e' },
- { "emm-pid", required_argument, NULL, 'Z' },
- { "emm-only", no_argument, NULL, 'E' },
- { "emm-report-time", required_argument, NULL, 'f' },
-
- { "ecm-pid", required_argument, NULL, 'X' },
- { "ecm-report-time", required_argument, NULL, 'H' },
- { "ecm-irdeto-type", required_argument, NULL, 'G' },
- { "ecm-no-log", no_argument , NULL, 'K' },
- { "cw-warn-time", required_argument, NULL, 'J' },
-
- { "debug", required_argument, NULL, 'D' },
- { "pid-report", no_argument, NULL, 'j' },
- { "bench", no_argument, NULL, 'b' },
- { "help", no_argument, NULL, 'h' },
- { "version", no_argument, NULL, 'V' },
-
- { 0, 0, 0, 0 }
- };
-
- static void show_help(struct ts *ts) {
- printf("%s\n", program_id);
- printf("Copyright (C) 2011-2012 Unix Solutions Ltd.\n");
- printf("\n");
- printf(" Usage: " PROGRAM_NAME " [opts]\n");
- printf("\n");
- printf("Main options:\n");
- printf(" -i --ident <server> | Format PROVIDER/CHANNEL. Default: empty\n");
- printf(" -d --daemon <pidfile> | Daemonize program and write pid file.\n");
- printf(" -N --notify-program <prg> | Execute <prg> to report events. Default: empty\n");
- printf("\n");
- printf("Input options:\n");
- printf(" -I --input <source> | Where to read from. File or multicast address.\n");
- printf(" . -I 224.0.0.1:5000 (multicast receive)\n");
- printf(" . -I file.ts (read from file)\n");
- printf(" . -I - (read from stdin) (default)\n");
- printf(" -R --input-rtp | Enable RTP input\n");
- printf(" -z --input-ignore-disc | Do not report discontinuty errors in input.\n");
- printf(" -M --input-service <srvid> | Choose service id when input is MPTS.\n");
- printf(" -T --input-buffer <ms> | Set input buffer time in ms. Default: %u\n", ts->input_buffer_time);
- printf(" -W --input-dump <filename> | Save input stream in file.\n");
- printf("\n");
- printf("Output options:\n");
- printf(" -O --output <dest> | Where to send output. File or multicast address.\n");
- printf(" . -O 239.0.0.1:5000 (multicast send)\n");
- printf(" . -O file.ts (write to file)\n");
- 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(" -u --no-output-on-error | Do not output data when the code word is missing.\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");
- printf(" -w --output-eit-pass | Pass through EIT (EPG).\n");
- printf(" -x --output-tdt-pass | Pass through TDT/TOT.\n");
- printf("\n");
- printf("CA options:\n");
- printf(" -c --ca-system <ca_sys> | Process input EMM/ECM from <ca_sys>.\n");
- printf(" | Valid systems are: CONAX (default), CRYPTOWORKS,\n");
- printf(" . IRDETO, SECA (MEDIAGUARD), VIACCESS,\n");
- printf(" . VIDEOGUARD (NDS), NAGRA, DRECRYPT and BULCRYPT.\n");
- printf(" -C --caid <caid> | Set CAID. Default: Taken from --ca-system.\n");
- printf(" -Y --const-cw <codeword> | Set constant code word for decryption.\n");
- printf(" . Example cw: a1a2a3a4a5a6a7a8b1b2b3b4b5b6b7b8\n");
- printf(" -Q --biss-key <biss-key> | Set BISS key for decryption.\n");
- printf(" . Example key: 112233445566\n");
- printf("\n");
- printf("CAMD server options:\n");
- printf(" -A --camd-proto <proto> | Set CAMD network protocol.\n");
- printf(" . Valid protocols are: CS378X (default) and NEWCAMD\n");
- printf(" -s --camd-server <addr> | Set CAMD server ip_address:port (1.2.3.4:2233).\n");
- printf(" -U --camd-user <user> | Set CAMD server user. Default: %s\n", ts->camd.user);
- printf(" -P --camd-pass <pass> | Set CAMD server password. Default: %s\n", ts->camd.pass);
- printf(" -B --camd-des-key <key> | Set DES key for newcamd protocol.\n");
- printf(" . Default: %s\n", ts->camd.newcamd.hex_des_key);
- printf("\n");
- printf("EMM options:\n");
- printf(" -e --emm | Enable sending EMM's to CAMD. Default: %s\n", ts->emm_send ? "enabled" : "disabled");
- printf(" -E --emm-only | Send only EMMs to CAMD, skipping ECMs and without\n");
- printf(" . decoding the input stream.\n");
- printf(" -Z --emm-pid <pid> | Force EMM pid. Default: none\n");
- printf(" -f --emm-report-time <sec> | Report each <sec> seconds how much EMMs have been\n");
- printf(" . received/processed. Set <sec> to 0 to disable\n");
- printf(" . the reports. Default: %d sec\n", ts->emm_report_interval);
- printf("\n");
- printf("ECM options:\n");
- printf(" -X --ecm-pid <pid> | Force ECM pid. Default: none\n");
- printf(" -H --ecm-report-time <sec> | Report each <sec> how much ECMs and CWs have been\n");
- printf(" . processed/skipped. Set <sec> to 0 to disable\n");
- printf(" . the reports. Default: %d sec\n", ts->ecm_report_interval);
- printf(" -G --ecm-irdeto-type <int> | Process IRDETO ECMs with type X /0-3/. Default: %d\n", ts->irdeto_ecm);
- printf(" -K --ecm-no-log | Disable ECM and code words logging.\n");
- printf(" -J --cw-warn-time <sec> | Warn if no valid code word has been received.\n");
- printf(" . Set <sec> to 0 to disable. Default: %d sec\n", ts->cw_warn_sec);
- printf("\n");
- printf("Logging options:\n");
- printf(" -S --syslog | Log messages using syslog.\n");
- printf(" -l --syslog-host <host> | Syslog server address. Default: disabled\n");
- printf(" -L --syslog-port <port> | Syslog server port. Default: %d\n", ts->syslog_port);
- printf(" -F --log-file <filename> | Log to file <filename>.\n");
- printf(" -D --debug <level> | Message debug level.\n");
- printf(" . 0 = default messages\n");
- printf(" . 1 = show PSI tables\n");
- printf(" . 2 = show EMMs\n");
- printf(" . 3 = show duplicate ECMs\n");
- printf(" . 4 = packet debug\n");
- printf(" . 5 = packet debug + packet dump\n");
- printf("\n");
- printf("Misc options:\n");
- printf(" -j --pid-report | Report how much packets were received.\n");
- printf(" -b --bench | Benchmark decrypton.\n");
- printf(" -h --help | Show help screen.\n");
- printf(" -V --version | Show program version.\n");
- printf("\n");
- }
-
- static int parse_io_param(struct io *io, char *opt, int open_flags, mode_t open_mode) {
- io->type = WTF_IO;
- char *p = strrchr(opt, ':');
- if (!p) {
- io->type = FILE_IO;
- if (strcmp(opt, "-") != 0) {
- io->fd = open(opt, open_flags, open_mode);
- if (io->fd < 0) {
- fprintf(stderr, "ERROR: Can not open file (%s): %s\n", opt, strerror(errno));
- exit(EXIT_FAILURE);
- }
- }
- io->fname = strdup(opt);
- return 0;
- }
- *p = 0x00;
- io->type = NET_IO;
- io->port = atoi(p + 1);
- if (inet_aton(opt, &io->addr) == 0)
- return 1;
- return 0;
- }
-
- static void parse_options(struct ts *ts, int argc, char **argv) {
- int j, i, ca_err = 0, server_err = 1, input_addr_err = 0, output_addr_err = 0, output_intf_err = 0, ident_err = 0, port_set = 0;
- while ((j = getopt_long(argc, argv, short_options, long_options, NULL)) != -1) {
- char *p = NULL;
- switch (j) {
- case 'i': // -- ident
- ts->ident = optarg;
- break;
- case 'd': // --daemon
- ts->pidfile = optarg;
- break;
- case 'N': // --notify-program
- ts->notify_program = optarg;
- break;
-
- case 'S': // --syslog
- ts->syslog_active = 1;
- ts->syslog_remote = 0;
- break;
- case 'l': // --syslog-host
- ts->syslog_host = optarg;
- ts->syslog_active = 1;
- ts->syslog_remote = 1;
- break;
- case 'L': // --syslog-port
- ts->syslog_port = atoi(optarg);
- break;
- case 'F': // --log-file
- log_filename = optarg;
- break;
-
- case 'I': // --input
- input_addr_err = parse_io_param(&ts->input, optarg, O_RDONLY, 0);
- break;
- case 'R': // --input-rtp
- ts->rtp_input = !ts->rtp_input;
- break;
- case 'z': // --input-ignore-disc
- ts->ts_discont = !ts->ts_discont;
- break;
- case 'M': // --input-service
- ts->forced_service_id = strtoul(optarg, NULL, 0) & 0xffff;
- break;
- case 'T': // --input-buffer
- ts->input_buffer_time = strtoul(optarg, NULL, 0);
- break;
- case 'W': // --input-dump
- ts->input_dump_filename = optarg;
- break;
-
- case 'O': // --output
- output_addr_err = parse_io_param(&ts->output, optarg,
- O_CREAT | O_WRONLY | O_TRUNC,
- S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
- break;
- case 'o': // --output-intf
- if (inet_aton(optarg, &ts->output.intf) == 0)
- output_intf_err = 1;
- break;
- case 't': // --output-ttl
- ts->output.ttl = atoi(optarg);
- break;
- case 'r': // --output-rtp
- ts->rtp_output = 1;
- break;
- case 'k': // --output-rtp-ssrc
- ts->rtp_ssrc = strtoul(optarg, NULL, 0);
- break;
- case 'g': // --output-tos
- ts->output.tos = (uint8_t)strtol(optarg, NULL, 0);
- break;
- case 'u': // --no-output-on-error
- ts->no_output_on_error = !ts->no_output_on_error;
- break;
- case 'p': // --no-output-filter
- ts->pid_filter = !ts->pid_filter;
- break;
- case 'y': // --output-nit-pass
- ts->nit_passthrough = !ts->nit_passthrough;
- break;
- case 'w': // --output-eit-pass
- ts->eit_passthrough = !ts->eit_passthrough;
- break;
- case 'x': // --output-tdt-pass
- ts->tdt_passthrough = !ts->tdt_passthrough;
- break;
- case 'c': // --ca-system
- if (strcasecmp("IRDETO", optarg) == 0)
- ts->req_CA_sys = CA_IRDETO;
- else if (strcasecmp("CONNAX", optarg) == 0 || strcasecmp("CONAX", optarg) == 0)
- ts->req_CA_sys = CA_CONAX;
- else if (strcasecmp("CRYPTOWORKS", optarg) == 0)
- ts->req_CA_sys = CA_CRYPTOWORKS;
- else if (strcasecmp("SECA", optarg) == 0 || strcasecmp("MEDIAGUARD", optarg) == 0)
- ts->req_CA_sys = CA_SECA;
- else if (strcasecmp("VIACCESS", optarg) == 0)
- ts->req_CA_sys = CA_VIACCESS;
- else if (strcasecmp("VIDEOGUARD", optarg) == 0 || strcasecmp("NDS", optarg) == 0)
- ts->req_CA_sys = CA_VIDEOGUARD;
- else if (strcasecmp("NAGRA", optarg) == 0)
- ts->req_CA_sys = CA_NAGRA;
- else if (strcasecmp("DRE-CRYPT", optarg) == 0 || strcasecmp("DRECRYPT", optarg) == 0)
- ts->req_CA_sys = CA_DRECRYPT;
- else if (strcasecmp("BULCRYPT", optarg) == 0)
- ts->req_CA_sys = CA_BULCRYPT;
- else
- ca_err = 1;
- break;
- case 'C': // --caid
- ts->forced_caid = strtoul(optarg, NULL, 0) & 0xffff;
- break;
- case 'Y': // --const-cw
- ts->camd.constant_codeword = 1;
- if (strlen(optarg) > 2 && optarg[0] == '0' && optarg[1] == 'x')
- optarg += 2;
- if (strlen(optarg) != CODEWORD_LENGTH * 2) {
- fprintf(stderr, "ERROR: Constant code word should be %u characters long.\n", CODEWORD_LENGTH * 2);
- exit(EXIT_FAILURE);
- }
- if (decode_hex_string(optarg, ts->camd.key->cw, strlen(optarg)) < 0) {
- fprintf(stderr, "ERROR: Invalid hex string for constant code word: %s\n", optarg);
- exit(EXIT_FAILURE);
- }
- camd_set_cw(ts, ts->camd.key->cw, 0);
- ts->camd.key->is_valid_cw = 1;
- break;
- case 'Q': // --biss-key
- ts->camd.constant_codeword = 1;
- if (strlen(optarg) > 2 && optarg[0] == '0' && optarg[1] == 'x')
- optarg += 2;
- uint8_t *key = ts->camd.key->cw;
- // Sometimes the BISS keys are entered with their checksums already calculated (16 symbols, 8 bytes)
- // This is the same as constant cw with the same key for even and odd
- if (strlen(optarg) == (BISSKEY_LENGTH + 2) * 2) {
- if (decode_hex_string(optarg, key, strlen(optarg)) < 0) {
- fprintf(stderr, "ERROR: Invalid hex string for BISS key: %s\n", optarg);
- exit(EXIT_FAILURE);
- }
- } else {
- // BISS key without checksum (12 symbols, 6 bytes)
- if (strlen(optarg) != BISSKEY_LENGTH * 2) {
- fprintf(stderr, "ERROR: BISS key should be %u characters long.\n", BISSKEY_LENGTH * 2);
- exit(EXIT_FAILURE);
- }
- if (decode_hex_string(optarg, key, strlen(optarg)) < 0) {
- fprintf(stderr, "ERROR: Invalid hex string for BISS key: %s\n", optarg);
- exit(EXIT_FAILURE);
- }
- // Calculate BISS KEY crc
- memmove(key + 4, key + 3, 3);
- key[3] = (uint8_t)(key[0] + key[1] + key[2]);
- key[7] = (uint8_t)(key[4] + key[5] + key[6]);
- }
- // Even and odd keys are the same
- memcpy(key + 8, key, 8);
- camd_set_cw(ts, ts->camd.key->cw, 0);
- ts->camd.key->is_valid_cw = 1;
- break;
-
- case 'A': // --camd-proto
- if (strcasecmp(optarg, "cs378x") == 0) {
- camd_proto_cs378x(&ts->camd.ops);
- } else if (strcasecmp(optarg, "newcamd") == 0) {
- camd_proto_newcamd(&ts->camd.ops);
- } else {
- fprintf(stderr, "Unknown CAMD protocol: %s\n", optarg);
- exit(EXIT_FAILURE);
- }
- break;
- case 's': // --camd-server
- p = strrchr(optarg, ':');
- if (p) {
- *p = 0x00;
- ts->camd.server_port = atoi(p + 1);
- port_set = 1;
- }
- if (inet_aton(optarg, &ts->camd.server_addr) == 0)
- server_err = 1;
- else
- server_err = 0;
- break;
- case 'U': // --camd-user
- if (strlen(optarg) < 64)
- ts->camd.user = optarg;
- break;
- case 'P': // --camd-pass
- ts->camd.pass = optarg;
- break;
- case 'B': // --camd-des-key
- if (strlen(optarg) > 2 && optarg[0] == '0' && optarg[1] == 'x')
- optarg += 2;
- if (strlen(optarg) != DESKEY_LENGTH) {
- fprintf(stderr, "ERROR: des key should be %u characters long.\n", DESKEY_LENGTH);
- exit(EXIT_FAILURE);
- }
- strncpy(ts->camd.newcamd.hex_des_key, optarg, sizeof(ts->camd.newcamd.hex_des_key) - 1);
- ts->camd.newcamd.hex_des_key[sizeof(ts->camd.newcamd.hex_des_key) - 1] = 0;
- break;
-
- case 'e': // --emm
- ts->emm_send = !ts->emm_send;
- break;
- case 'Z': // --emm-pid
- ts->forced_emm_pid = strtoul(optarg, NULL, 0) & 0x1fff;
- break;
- case 'E': // --emm-only
- ts->emm_only = 1;
- ts->emm_send = 1;
- break;
- case 'f': // --emm-report-time
- ts->emm_report_interval = strtoul(optarg, NULL, 10);
- if (ts->emm_report_interval > 86400)
- ts->emm_report_interval = 86400;
- break;
-
- case 'X': // --ecm-pid
- ts->forced_ecm_pid = strtoul(optarg, NULL, 0) & 0x1fff;
- break;
- case 'H': // --ecm-report-time
- ts->ecm_report_interval = strtoul(optarg, NULL, 10);
- if (ts->ecm_report_interval > 86400)
- ts->ecm_report_interval = 86400;
- break;
- case 'G': // --ecm-irdeto-type
- ts->irdeto_ecm = atoi(optarg);
- break;
- case 'K': // --ecm-no-log
- ts->ecm_cw_log = !ts->ecm_cw_log;
- break;
- case 'J': // --cw-warn-time
- ts->cw_warn_sec = strtoul(optarg, NULL, 10);
- if (ts->cw_warn_sec > 86400)
- ts->cw_warn_sec = 86400;
- ts->cw_last_warn= ts->cw_last_warn + ts->cw_warn_sec;
- break;
-
- case 'D': // --debug
- ts->debug_level = atoi(optarg);
- if (ts->debug_level > 0)
- ts->pid_report = 1;
- break;
- case 'j': // --pid-report
- ts->pid_report = 1;
- break;
- case 'b': // --bench
- csa_benchmark();
- exit(EXIT_SUCCESS);
-
- case 'h': // --help
- show_help(ts);
- exit(EXIT_SUCCESS);
-
- case 'V': // --version
- printf("%s\n", program_id);
- exit(EXIT_SUCCESS);
- }
- }
- if (!ts->ident) {
- if (ts->syslog_active || ts->notify_program)
- ident_err = 1;
- }
-
- // Constant codeword is special. Disable conflicting options
- if (ts->camd.constant_codeword) {
- server_err = 0; // No server settings are required
- ts->forced_caid = 0;
- ts->forced_ecm_pid = 0;
- ts->forced_emm_pid = 0;
- ts->emm_send = 0;
- ts->emm_report_interval = 0;
- ts->ecm_report_interval = 0;
- ts->cw_warn_sec = 0;
- }
-
- if (ident_err || ca_err || server_err || input_addr_err || output_addr_err || ts->input.type == WTF_IO || ts->output.type == WTF_IO) {
- show_help(ts);
- if (ident_err)
- fprintf(stderr, "ERROR: Ident is not set, please use --ident option.\n");
- if (ca_err)
- fprintf(stderr, "ERROR: Requested CA system is unsupported.\n");
- if (server_err)
- fprintf(stderr, "ERROR: CAMD server IP address is not set or it is invalid.\n");
- if (input_addr_err)
- fprintf(stderr, "ERROR: Input IP address is invalid.\n");
- if (output_addr_err)
- fprintf(stderr, "ERROR: Output IP address is invalid.\n");
- if (output_intf_err)
- fprintf(stderr, "ERROR: Output interface address is invalid.\n");
- exit(EXIT_FAILURE);
- }
- if (decode_hex_string(ts->camd.newcamd.hex_des_key, ts->camd.newcamd.bin_des_key, DESKEY_LENGTH) < 0) {
- fprintf(stderr, "ERROR: Invalid hex string for des key: %s\n", ts->camd.newcamd.hex_des_key);
- exit(EXIT_FAILURE);
- }
- if (ts->camd.ops.proto == CAMD_NEWCAMD && !port_set) {
- fprintf(stderr, "ERROR: CAMD server port is not set. Use --camd-server %s:xxxx to set the port.\n", inet_ntoa(ts->camd.server_addr));
- exit(EXIT_FAILURE);
- }
-
- if (log_filename) {
- log_file = fopen(log_filename, "a");
- if (!log_file) {
- fprintf(stderr, "ERROR: Can't open log file %s: %s\n", log_filename, strerror(errno));
- exit(EXIT_FAILURE);
- }
- }
-
- if (ts->ident)
- ts_LOGf("Ident : %s\n", ts->ident);
- if (ts->notify_program)
- ts_LOGf("Notify prg : %s\n", ts->notify_program);
- if (ts->pidfile)
- ts_LOGf("Daemonize : %s pid file.\n", ts->pidfile);
- if (ts->syslog_active) {
- if (ts->syslog_remote)
- ts_LOGf("Syslog : %s:%d\n", ts->syslog_host, ts->syslog_port);
- else
- ts_LOGf("Syslog : enabled\n");
- } else
- ts_LOGf("Syslog : disabled\n");
-
- if (!ts->camd.constant_codeword) {
- if (ts->forced_caid)
- ts->req_CA_sys = ts_get_CA_sys(ts->forced_caid);
- if (!ts->forced_caid)
- ts_LOGf("CA System : %s\n", ts_get_CA_sys_txt(ts->req_CA_sys));
- else
- ts_LOGf("CA System : %s | CAID: 0x%04x (%d)\n",
- ts_get_CA_sys_txt(ts->req_CA_sys),
- ts->forced_caid, ts->forced_caid);
- } else {
- char cw_even[64], cw_odd[64];
- ts_hex_dump_buf(cw_even, sizeof(cw_even), ts->key.cw , 8, 0);
- ts_hex_dump_buf(cw_odd , sizeof(cw_odd ), ts->key.cw + 8, 8, 0);
- ts_LOGf("Constant CW: even = %s\n", cw_even);
- ts_LOGf("Constant CW: odd = %s\n", cw_odd);
- }
-
- if (ts->input.type == NET_IO) {
- ts_LOGf("Input addr : %s://%s:%u/\n",
- ts->rtp_input ? "rtp" : "udp",
- inet_ntoa(ts->input.addr), ts->input.port);
- if (ts->input_buffer_time) {
- ts_LOGf("Input buff : %u ms\n", ts->input_buffer_time);
- }
- } else if (ts->input.type == FILE_IO) {
- ts_LOGf("Input file : %s\n", ts->input.fd == 0 ? "STDIN" : ts->input.fname);
- }
- if (ts->input_dump_filename) {
- ts->input_dump_file = fopen(ts->input_dump_filename, "w");
- if (ts->input_dump_file)
- ts_LOGf("Input dump : %s\n", ts->input_dump_filename);
- else
- ts_LOGf("Input dump : %s | ERROR: %s\n", ts->input_dump_filename, strerror(errno));
- }
- if (ts->forced_service_id)
- ts_LOGf("Service id : 0x%04x (%d)\n",
- ts->forced_service_id, ts->forced_service_id);
- if (ts->req_CA_sys == CA_IRDETO)
- ts_LOGf("Irdeto ECM : %d\n", ts->irdeto_ecm);
-
- if (!ts->emm_only)
- {
- if (ts->output.type == NET_IO) {
- 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);
- }
- ts_LOGf("Out filter : %s (%s)%s\n",
- ts->pid_filter ? "enabled" : "disabled",
- ts->pid_filter ? "output only service related PIDs" : "output everything",
- ts->no_output_on_error ? " (No output on CW error)" : ""
- );
- if (ts->pid_filter) {
- if (ts->nit_passthrough)
- ts_LOGf("Out filter : Pass through NIT.\n");
- if (ts->eit_passthrough)
- ts_LOGf("Out filter : Pass through EIT (EPG).\n");
- if (ts->tdt_passthrough)
- ts_LOGf("Out filter : Pass through TDT/TOT.\n");
- }
- }
- if (!ts->camd.constant_codeword) {
- ts_LOGf("CAMD proto : %s\n", ts->camd.ops.ident);
- ts_LOGf("CAMD addr : tcp://%s:%u/\n", inet_ntoa(ts->camd.server_addr), ts->camd.server_port);
- ts_LOGf("CAMD user : %s\n", ts->camd.user);
- ts_LOGf("CAMD pass : %s\n", ts->camd.pass);
- if (ts->camd.ops.proto == CAMD_NEWCAMD)
- ts_LOGf("CAMD deskey: %s\n", ts->camd.newcamd.hex_des_key);
- }
-
- ts_LOGf("TS discont : %s\n", ts->ts_discont ? "report" : "ignore");
- ts->threaded = !(ts->input.type == FILE_IO && ts->input.fd != 0);
- if (ts->emm_send && ts->emm_report_interval)
- ts_LOGf("EMM report : %d sec\n", ts->emm_report_interval);
- if (ts->emm_send && ts->emm_report_interval == 0)
- ts_LOGf("EMM report : disabled\n");
- if (ts->forced_emm_pid)
- ts_LOGf("EMM pid : 0x%04x (%d)\n", ts->forced_emm_pid, ts->forced_emm_pid);
- if (ts->emm_only) {
- ts_LOGf("EMM only : %s\n", ts->emm_only ? "yes" : "no");
- } else {
- if (!ts->camd.constant_codeword)
- ts_LOGf("EMM send : %s\n", ts->emm_send ? "enabled" : "disabled");
- ts_LOGf("Decoding : %s\n", ts->threaded ? "threaded" : "single thread");
- }
-
- if (!ts->emm_only && ts->ecm_report_interval)
- ts_LOGf("ECM report : %d sec\n", ts->emm_report_interval);
- if (!ts->emm_only && ts->ecm_report_interval == 0 && !ts->camd.constant_codeword)
- ts_LOGf("ECM report : disabled\n");
- if (ts->forced_ecm_pid)
- ts_LOGf("ECM pid : 0x%04x (%d)\n", ts->forced_ecm_pid, ts->forced_ecm_pid);
-
- if (!ts->emm_only && ts->cw_warn_sec)
- ts_LOGf("CW warning : %d sec\n", ts->cw_warn_sec);
- if (!ts->emm_only && ts->cw_warn_sec && !ts->camd.constant_codeword)
- ts_LOGf("CW warning : disabled\n");
-
- if (!ts->ecm_cw_log && !ts->camd.constant_codeword)
- ts_LOGf("ECM/CW log : disabled\n");
-
- if (ts->ident) {
- int len = strlen(ts->ident);
- for (i = 0; i < len; i++) {
- if (ts->ident[i] == '/')
- ts->ident[i] = '-';
- }
- }
- }
-
- static void report_emms(struct ts *ts, time_t now) {
- ts_LOGf("EMM | Received %u and processed %u in %lu seconds.\n",
- ts->emm_seen_count,
- ts->emm_processed_count,
- now - ts->emm_last_report);
- if (ts->emm_seen_count == 0) {
- notify(ts, "NO_EMM_RECEIVED", "No EMMs were received in last %lu seconds.",
- now - ts->emm_last_report);
- }
- ts->emm_last_report = now;
- ts->emm_seen_count = 0;
- ts->emm_processed_count = 0;
- }
-
- static void report_ecms(struct ts *ts, time_t now) {
- ts_LOGf("ECM | Received %u (%u dup) and processed %u in %lu seconds.\n",
- ts->ecm_seen_count,
- ts->ecm_duplicate_count,
- ts->ecm_processed_count,
- now - ts->ecm_last_report);
- ts->ecm_last_report = now;
- ts->ecm_seen_count = 0;
- ts->ecm_duplicate_count = 0;
- ts->ecm_processed_count = 0;
- }
-
- static void report_cw_warn(struct ts *ts, time_t now) {
- if (now - ts->key.ts > 1) {
- notify(ts, "NO_CODE_WORD", "No valid code word was received in %ld sec.",
- now - ts->key.ts);
- ts_LOGf("CW | *ERR* No valid code word was received for %ld seconds!\n",
- now - ts->key.ts);
- }
- ts->cw_last_warn = now;
- ts->cw_next_warn = now + ts->cw_warn_sec;
- }
-
- static void do_reports(struct ts *ts) {
- static int first_emm_report = 1;
- static int first_ecm_report = 1;
- time_t now = time(NULL);
- if (ts->emm_send && ts->emm_report_interval) {
- if (first_emm_report && now >= ts->emm_last_report) {
- first_emm_report = 0;
- ts->emm_last_report -= FIRST_REPORT_SEC;
- report_emms(ts, now);
- } else if ((time_t)(ts->emm_last_report + ts->emm_report_interval) <= now) {
- report_emms(ts, now);
- }
- }
- if (!ts->emm_only && ts->ecm_report_interval) {
- if (first_ecm_report && now >= ts->ecm_last_report) {
- first_ecm_report = 0;
- ts->ecm_last_report -= FIRST_REPORT_SEC;
- report_ecms(ts, now);
- } else if ((time_t)(ts->ecm_last_report + ts->ecm_report_interval) <= now) {
- report_ecms(ts, now);
- }
- }
-
- if (!ts->emm_only && !ts->key.is_valid_cw) {
- if (ts->cw_warn_sec && now >= ts->cw_next_warn) {
- report_cw_warn(ts, now);
- }
- }
- }
-
- void signal_quit(int sig) {
- if (!keep_running)
- raise(sig);
- keep_running = 0;
- ts_LOGf("Killed %s with signal %d\n", program_id, sig);
- signal(sig, SIG_DFL);
- }
-
- #define RTP_HDR_SZ 12
-
- static uint8_t ts_packet[FRAME_SIZE + RTP_HDR_SZ];
- static uint8_t rtp_hdr[2][RTP_HDR_SZ];
- static struct ts ts;
-
- int main(int argc, char **argv) {
- ssize_t readen;
- int have_data = 1;
- int ntimeouts = 0;
- time_t timeout_start = time(NULL);
- int rtp_hdr_pos = 0, num_packets = 0;
- struct rlimit rl;
-
- if (getrlimit(RLIMIT_STACK, &rl) == 0) {
- if (rl.rlim_cur > THREAD_STACK_SIZE) {
- rl.rlim_cur = THREAD_STACK_SIZE;
- setrlimit(RLIMIT_STACK, &rl);
- }
- }
-
- memset(rtp_hdr[0], 0, RTP_HDR_SZ);
- memset(rtp_hdr[1], 0, RTP_HDR_SZ);
-
- data_init(&ts);
-
- ts_set_log_func(LOG_func);
-
- parse_options(&ts, argc, argv);
-
- if (ts.pidfile)
- daemonize(ts.pidfile);
-
- if (ts.syslog_active) {
- if (ts.syslog_remote) {
- log_init(ts.ident, 1, 1, ts.syslog_host, ts.syslog_port);
- remote_syslog = 1;
- } else {
- openlog(ts.ident, LOG_NDELAY | LOG_PID, LOG_USER);
- local_syslog = 1;
- }
- }
-
- ts.notify = notify_alloc(&ts);
-
- ts_LOGf("Start %s\n", program_id);
- notify(&ts, "START", "Starting %s", program_id);
-
- if (ts.input.type == NET_IO && udp_connect_input(&ts.input) < 1)
- goto EXIT;
- if (ts.output.type == NET_IO && udp_connect_output(&ts.output) < 1)
- goto EXIT;
-
- signal(SIGCHLD, SIG_IGN);
- signal(SIGPIPE, SIG_IGN);
-
- signal(SIGINT , signal_quit);
- signal(SIGTERM, signal_quit);
-
- if (ts.threaded) {
- pthread_create(&ts.decode_thread, &ts.thread_attr, &decode_thread, &ts);
- pthread_create(&ts.write_thread , &ts.thread_attr, &write_thread , &ts);
- }
-
- ts.emm_last_report = time(NULL) + FIRST_REPORT_SEC;
- ts.ecm_last_report = time(NULL) + FIRST_REPORT_SEC;
- camd_start(&ts);
- do {
- if (!ts.camd.constant_codeword)
- do_reports(&ts);
-
- if (ts.input.type == NET_IO) {
- set_log_io_errors(0);
- if (!ts.rtp_input) {
- readen = fdread_ex(ts.input.fd, (char *)ts_packet, FRAME_SIZE, 250, 4, 1);
- } else {
- readen = fdread_ex(ts.input.fd, (char *)ts_packet, FRAME_SIZE + RTP_HDR_SZ, 250, 4, 1);
- if (readen > RTP_HDR_SZ) {
- memcpy(rtp_hdr[rtp_hdr_pos], ts_packet, RTP_HDR_SZ);
- memmove(ts_packet, ts_packet + RTP_HDR_SZ, FRAME_SIZE);
- readen -= RTP_HDR_SZ;
- uint16_t ssrc = (rtp_hdr[rtp_hdr_pos][2] << 8) | rtp_hdr[rtp_hdr_pos][3];
- uint16_t pssrc = (rtp_hdr[!rtp_hdr_pos][2] << 8) | rtp_hdr[!rtp_hdr_pos][3];
- rtp_hdr_pos = !rtp_hdr_pos;
- if (pssrc + 1 != ssrc && (ssrc != 0 && pssrc != 0xffff) && num_packets > 2)
- if (ts.ts_discont)
- ts_LOGf("--- | RTP discontinuity last_ssrc %5d, curr_ssrc %5d, lost %d packet\n",
- pssrc, ssrc, ((ssrc - pssrc)-1) & 0xffff);
- num_packets++;
- }
- }
- set_log_io_errors(1);
- if (readen < 0) {
- ts_LOGf("--- | Input read timeout.\n");
- if (!ntimeouts) {
- timeout_start = time(NULL);
- notify(&ts, "INPUT_TIMEOUT", "Read timeout on input %s://%s:%u/",
- ts.rtp_input ? "rtp" : "udp",
- inet_ntoa(ts.input.addr), ts.input.port);
- }
- ntimeouts++;
- } else {
- if (ntimeouts && readen > 0) {
- notify(&ts, "INPUT_OK", "Data is available on input %s://%s:%u/ after %ld seconds timeout.",
- ts.rtp_input ? "rtp" : "udp",
- inet_ntoa(ts.input.addr), ts.input.port,
- (time(NULL) - timeout_start) + 2); // Timeout is detected when ~2 seconds there is no incoming data
- ntimeouts = 0;
- }
- }
- } else {
- readen = read(ts.input.fd, ts_packet, FRAME_SIZE);
- have_data = !(readen <= 0);
- }
- if (readen > 0) {
- if (ts.input_dump_file)
- fwrite(ts_packet, readen, 1, ts.input_dump_file);
- process_packets(&ts, ts_packet, readen);
- }
- if (!keep_running)
- break;
- } while (have_data);
- EXIT:
- camd_stop(&ts);
-
- if (ts.threaded) {
- ts.decode_stop = 1;
- ts.write_stop = 1;
-
- if (ts.decode_thread)
- pthread_join(ts.decode_thread, NULL);
- if (ts.write_thread)
- pthread_join(ts.write_thread, NULL);
- }
-
- show_pid_report(&ts);
-
- notify_sync(&ts, "STOP", "Stopping %s", program_id);
- ts_LOGf("Stop %s\n", program_id);
-
- if (ts.syslog_active) {
- if (ts.syslog_remote)
- log_close();
- else
- closelog();
- }
-
- if (ts.input_dump_file)
- fclose(ts.input_dump_file);
-
- if (ts.pidfile)
- unlink(ts.pidfile);
-
- if (log_file)
- fclose(log_file);
-
- notify_free(&ts.notify);
- data_free(&ts);
-
- exit(EXIT_SUCCESS);
- }
|