int Service::Open( Frontend &frontend, int pid )
{
int fd = frontend.OpenDemux( );
if( fd > 0 )
{
if( dvb_set_pesfilter( fd, pid, DMX_PES_OTHER, DMX_OUT_TSDEMUX_TAP, DMX_BUFSIZE ) != 0 )
{
frontend.LogError( "failed to set the pes filter for pid 0x%04x", pid );
frontend.CloseDemux( fd );
return -1;
}
}
return fd;
}
int main(int argc, char **argv)
{
struct arguments args;
char *homedir = getenv("HOME");
char *channel = NULL;
int lnb = -1, idx = -1;
int vpid = -1, apid = -1, sid = -1;
int pmtpid = 0;
int pat_fd = -1, pmt_fd = -1;
int audio_fd = 0, video_fd = 0;
int dvr_fd, file_fd;
struct dvb_v5_fe_parms *parms;
const struct argp argp = {
.options = options,
.parser = parse_opt,
.doc = "DVB zap utility",
.args_doc = "<initial file>",
};
memset(&args, 0, sizeof(args));
args.sat_number = -1;
argp_parse(&argp, argc, argv, 0, &idx, &args);
if (idx < argc)
channel = argv[idx];
if (!channel) {
argp_help(&argp, stderr, ARGP_HELP_STD_HELP, PROGRAM_NAME);
return -1;
}
if (args.input_format == FILE_UNKNOWN) {
fprintf(stderr, "ERROR: Please specify a valid format\n");
argp_help(&argp, stderr, ARGP_HELP_STD_HELP, PROGRAM_NAME);
return -1;
}
if (args.lnb_name) {
lnb = dvb_sat_search_lnb(args.lnb_name);
if (lnb < 0) {
printf("Please select one of the LNBf's below:\n");
print_all_lnb();
exit(1);
} else {
printf("Using LNBf ");
print_lnb(lnb);
}
}
asprintf(&args.demux_dev,
"/dev/dvb/adapter%i/demux%i", args.adapter, args.demux);
asprintf(&args.dvr_dev,
"/dev/dvb/adapter%i/dvr%i", args.adapter, args.demux);
if (args.silent < 2)
fprintf(stderr, "using demux '%s'\n", args.demux_dev);
if (!args.confname) {
if (!homedir)
ERROR("$HOME not set");
asprintf(&args.confname, "%s/.tzap/%i/%s",
homedir, args.adapter, CHANNEL_FILE);
if (access(args.confname, R_OK))
asprintf(&args.confname, "%s/.tzap/%s",
homedir, CHANNEL_FILE);
}
printf("reading channels from file '%s'\n", args.confname);
parms = dvb_fe_open(args.adapter, args.frontend, 0, args.force_dvbv3);
if (!parms)
return -1;
if (lnb)
parms->lnb = dvb_sat_get_lnb(lnb);
if (args.sat_number > 0)
parms->sat_number = args.sat_number % 3;
parms->diseqc_wait = args.diseqc_wait;
parms->freq_bpf = args.freq_bpf;
if (parse(&args, parms, channel, &vpid, &apid, &sid))
return -1;
if (setup_frontend(&args, parms) < 0)
return -1;
if (args.frontend_only) {
check_frontend(&args, parms);
dvb_fe_close(parms);
return 0;
}
if (args.rec_psi) {
if (sid < 0) {
fprintf(stderr, "Service id 0x%04x was not specified at the file\n",
sid);
return -1;
}
pmtpid = get_pmt_pid(args.demux_dev, sid);
if (pmtpid <= 0) {
fprintf(stderr, "couldn't find pmt-pid for sid %04x\n",
sid);
return -1;
}
if ((pat_fd = open(args.demux_dev, O_RDWR)) < 0) {
perror("opening pat demux failed");
return -1;
}
if (dvb_set_pesfilter(pat_fd, 0, DMX_PES_OTHER,
args.dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER,
args.dvr ? 64 * 1024 : 0) < 0)
return -1;
if ((pmt_fd = open(args.demux_dev, O_RDWR)) < 0) {
perror("opening pmt demux failed");
return -1;
}
if (dvb_set_pesfilter(pmt_fd, pmtpid, DMX_PES_OTHER,
args.dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER,
args.dvr ? 64 * 1024 : 0) < 0)
return -1;
}
if (vpid >= 0) {
if (args.silent < 2)
fprintf(stderr, "video pid %d\n", vpid);
if ((video_fd = open(args.demux_dev, O_RDWR)) < 0) {
PERROR("failed opening '%s'", args.demux_dev);
return -1;
}
printf( " dvb_set_pesfilter %d\n", vpid );
if (dvb_set_pesfilter(video_fd, vpid, DMX_PES_VIDEO,
args.dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER,
args.dvr ? 64 * 1024 : 0) < 0)
return -1;
}
if (apid >= 0) {
if (args.silent < 2)
fprintf(stderr, "audio pid %d\n", apid);
if ((audio_fd = open(args.demux_dev, O_RDWR)) < 0) {
PERROR("failed opening '%s'", args.demux_dev);
return -1;
}
if (dvb_set_pesfilter(audio_fd, apid, DMX_PES_AUDIO,
args.dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER,
args.dvr ? 64 * 1024 : 0) < 0)
return -1;
}
signal(SIGALRM, do_timeout);
if (args.timeout > 0)
alarm(args.timeout);
if (args.record) {
if (args.filename != NULL) {
if (strcmp(args.filename, "-") != 0) {
file_fd =
open(args.filename,
#ifdef O_LARGEFILE
O_LARGEFILE |
#endif
O_WRONLY | O_CREAT,
0644);
if (file_fd < 0) {
PERROR("open of '%s' failed",
args.filename);
return -1;
}
} else {
file_fd = 1;
}
} else {
PERROR("Record mode but no filename!");
return -1;
}
if ((dvr_fd = open(args.dvr_dev, O_RDONLY)) < 0) {
PERROR("failed opening '%s'", args.dvr_dev);
return -1;
}
if (args.silent < 2)
print_frontend_stats(parms, args.human_readable);
copy_to_file(dvr_fd, file_fd, args.timeout, args.silent);
if (args.silent < 2)
print_frontend_stats(parms, args.human_readable);
} else {
check_frontend(&args, parms);
}
close(pat_fd);
close(pmt_fd);
close(audio_fd);
close(video_fd);
dvb_fe_close(parms);
return 0;
}
int do_traffic_monitor(struct arguments *args,
struct dvb_v5_fe_parms *parms)
{
int fd, dvr_fd;
long long unsigned pidt[0x2001], wait;
int packets = 0;
struct timeval startt;
memset(pidt, 0, sizeof(pidt));
args->exit_after_tuning = 1;
check_frontend(args, parms);
if ((dvr_fd = open(args->dvr_dev, O_RDONLY)) < 0) {
PERROR(_("failed opening '%s'"), args->dvr_dev);
return -1;
}
if (ioctl(dvr_fd, DMX_SET_BUFFER_SIZE, 1024 * 1024) == -1)
perror(_("DMX_SET_BUFFER_SIZE failed"));
if ((fd = open(args->demux_dev, O_RDWR)) < 0) {
PERROR(_("failed opening '%s'"), args->demux_dev);
close(dvr_fd);
return -1;
}
if (args->silent < 2)
fprintf(stderr, _(" dvb_set_pesfilter to 0x2000\n"));
if (dvb_set_pesfilter(fd, 0x2000, DMX_PES_OTHER,
DMX_OUT_TS_TAP, 0) < 0) {
PERROR(_("couldn't set filter"));
return -1;
}
gettimeofday(&startt, 0);
wait = 1000;
while (1) {
unsigned char buffer[BSIZE];
struct dvb_ts_packet_header *h = (void *)buffer;
int pid, ok;
ssize_t r;
if (timeout_flag)
break;
if ((r = read(dvr_fd, buffer, BSIZE)) <= 0) {
if (errno == EOVERFLOW) {
struct timeval now;
int diff;
gettimeofday(&now, 0);
diff =
(now.tv_sec - startt.tv_sec) * 1000 +
(now.tv_usec - startt.tv_usec) / 1000;
fprintf(stderr, _("%.2fs: buffer overrun\n"), diff / 1000.);
continue;
}
perror(_("read"));
break;
}
if (r != BSIZE) {
fprintf(stderr, _("dvbtraffic: only read %zd bytes\n"), r);
break;
}
if (h->sync_byte != 0x47) {
continue;
}
bswap16(h->bitfield);
#if 0
/*
* ITU-T Rec. H.222.0 decoders shall discard Transport Stream
* packets with theadaptation_field_control field set to
* a value of '00'.
*/
if (h->adaptation_field_control == 0)
continue;
#endif
ok = 1;
pid = h->pid;
if (args->search) {
int i, sl = strlen(args->search);
ok = 0;
if (pid != 0x1fff) {
for (i = 0; i < (188 - sl); ++i) {
if (!memcmp(buffer + i, args->search, sl))
ok = 1;
}
}
}
if (ok) {
pidt[pid]++;
pidt[0x2000]++;
}
packets++;
if (!(packets & 0xFF)) {
struct timeval now;
int diff;
gettimeofday(&now, 0);
diff =
(now.tv_sec - startt.tv_sec) * 1000 +
(now.tv_usec - startt.tv_usec) / 1000;
if (diff > wait) {
if (isatty(STDOUT_FILENO))
printf("\x1b[1H\x1b[2J");
args->n_status_lines = 0;
printf(_(" PID FREQ SPEED TOTAL\n"));
int _pid = 0;
for (_pid = 0; _pid < 0x2000; _pid++) {
if (pidt[_pid]) {
if (!args->low_traffic && (pidt[_pid] * 1000. / diff) < 1)
continue;
printf("%04x %9.2f p/s %8.1f Kbps ",
_pid,
pidt[_pid] * 1000. / diff,
pidt[_pid] * 1000. / diff * 8 * 188 / 1024);
if (pidt[_pid] * 188 / 1024)
printf("%8llu KB\n", pidt[_pid] * 188 / 1024);
else
printf(" %8llu B\n", pidt[_pid] * 188);
}
}
/* 0x2000 is the total traffic */
printf("TOT %10.2f p/s %8.1f Kbps %8llu KB\n",
pidt[_pid] * 1000. / diff,
pidt[_pid] * 1000. / diff * 8 * 188 / 1024,
pidt[_pid] * 188 / 1024);
printf("\n\n");
get_show_stats(args, parms, 0);
wait += 1000;
}
}
}
close(dvr_fd);
close(fd);
return 0;
}
int main(int argc, char **argv)
{
struct arguments args;
char *homedir = getenv("HOME");
char *channel = NULL;
int lnb = -1, idx = -1;
int vpid = -1, apid = -1, sid = -1;
int pmtpid = 0;
int pat_fd = -1, pmt_fd = -1, sid_fd = -1;
int audio_fd = -1, video_fd = -1;
int dvr_fd = -1, file_fd = -1;
int err = -1;
int r;
struct dvb_v5_fe_parms *parms = NULL;
const struct argp argp = {
.options = options,
.parser = parse_opt,
.doc = N_("DVB zap utility"),
.args_doc = N_("<channel name> [or <frequency> if in monitor mode]"),
};
#ifdef ENABLE_NLS
setlocale (LC_ALL, "");
bindtextdomain (PACKAGE, LOCALEDIR);
textdomain (PACKAGE);
#endif
memset(&args, 0, sizeof(args));
args.sat_number = -1;
args.lna = LNA_AUTO;
args.input_format = FILE_DVBV5;
argp_parse(&argp, argc, argv, ARGP_NO_HELP | ARGP_NO_EXIT, &idx, &args);
if (idx < argc)
channel = argv[idx];
if (!channel) {
argp_help(&argp, stderr, ARGP_HELP_STD_HELP, PROGRAM_NAME);
return -1;
}
if (args.input_format == FILE_UNKNOWN) {
fprintf(stderr, _("ERROR: Please specify a valid format\n"));
argp_help(&argp, stderr, ARGP_HELP_STD_HELP, PROGRAM_NAME);
return -1;
}
if (!args.traffic_monitor && args.search) {
fprintf(stderr, _("ERROR: search string can be used only on monitor mode\n"));
argp_help(&argp, stderr, ARGP_HELP_STD_HELP, PROGRAM_NAME);
return -1;
}
if (args.lnb_name) {
lnb = dvb_sat_search_lnb(args.lnb_name);
if (lnb < 0) {
printf(_("Please select one of the LNBf's below:\n"));
dvb_print_all_lnb();
exit(1);
} else {
printf(_("Using LNBf "));
dvb_print_lnb(lnb);
}
}
r = asprintf(&args.demux_dev,
"/dev/dvb/adapter%i/demux%i", args.adapter, args.demux);
if (r < 0) {
fprintf(stderr, _("asprintf error\n"));
return -1;
}
r = asprintf(&args.dvr_dev,
"/dev/dvb/adapter%i/dvr%i", args.adapter, args.demux);
if (r < 0) {
fprintf(stderr, _("asprintf error\n"));
return -1;
}
if (args.silent < 2)
fprintf(stderr, _("using demux '%s'\n"), args.demux_dev);
if (!args.confname) {
if (!homedir)
ERROR(_("$HOME not set"));
r = asprintf(&args.confname, "%s/.tzap/%i/%s",
homedir, args.adapter, CHANNEL_FILE);
if (access(args.confname, R_OK)) {
free(args.confname);
r = asprintf(&args.confname, "%s/.tzap/%s",
homedir, CHANNEL_FILE);
}
}
fprintf(stderr, _("reading channels from file '%s'\n"), args.confname);
parms = dvb_fe_open(args.adapter, args.frontend, args.verbose, args.force_dvbv3);
if (!parms)
goto err;
if (lnb >= 0)
parms->lnb = dvb_sat_get_lnb(lnb);
if (args.sat_number > 0)
parms->sat_number = args.sat_number % 3;
parms->diseqc_wait = args.diseqc_wait;
parms->freq_bpf = args.freq_bpf;
parms->lna = args.lna;
r = dvb_fe_set_default_country(parms, args.cc);
if (r < 0)
fprintf(stderr, _("Failed to set the country code:%s\n"), args.cc);
if (parse(&args, parms, channel, &vpid, &apid, &sid))
goto err;
if (setup_frontend(&args, parms) < 0)
goto err;
if (args.exit_after_tuning) {
err = 0;
check_frontend(&args, parms);
goto err;
}
if (args.traffic_monitor) {
signal(SIGTERM, do_timeout);
signal(SIGINT, do_timeout);
if (args.timeout > 0) {
signal(SIGINT, do_timeout);
alarm(args.timeout);
}
err = do_traffic_monitor(&args, parms);
goto err;
}
if (args.rec_psi) {
if (sid < 0) {
fprintf(stderr, _("Service id 0x%04x was not specified at the file\n"),
sid);
goto err;
}
sid_fd = dvb_dmx_open(args.adapter, args.demux);
if (sid_fd < 0) {
perror(_("opening pat demux failed"));
return -1;
}
pmtpid = dvb_get_pmt_pid(sid_fd, sid);
dvb_dmx_close(sid_fd);
if (pmtpid <= 0) {
fprintf(stderr, _("couldn't find pmt-pid for sid %04x\n"),
sid);
goto err;
}
if ((pat_fd = open(args.demux_dev, O_RDWR)) < 0) {
perror(_("opening pat demux failed"));
goto err;
}
if (dvb_set_pesfilter(pat_fd, 0, DMX_PES_OTHER,
args.dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER,
args.dvr ? 64 * 1024 : 0) < 0)
goto err;
if ((pmt_fd = open(args.demux_dev, O_RDWR)) < 0) {
perror(_("opening pmt demux failed"));
goto err;
}
if (dvb_set_pesfilter(pmt_fd, pmtpid, DMX_PES_OTHER,
args.dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER,
args.dvr ? 64 * 1024 : 0) < 0)
goto err;
}
if (args.all_pids++) {
vpid = 0x2000;
apid = 0;
}
if (vpid >= 0) {
if (args.silent < 2) {
if (vpid == 0x2000)
fprintf(stderr, _("pass all PID's to TS\n"));
else
fprintf(stderr, _("video pid %d\n"), vpid);
}
if ((video_fd = open(args.demux_dev, O_RDWR)) < 0) {
PERROR(_("failed opening '%s'"), args.demux_dev);
goto err;
}
if (args.silent < 2)
fprintf(stderr, _(" dvb_set_pesfilter %d\n"), vpid);
if (vpid == 0x2000) {
if (ioctl(video_fd, DMX_SET_BUFFER_SIZE, 1024 * 1024) == -1)
perror(_("DMX_SET_BUFFER_SIZE failed"));
if (dvb_set_pesfilter(video_fd, vpid, DMX_PES_OTHER,
DMX_OUT_TS_TAP, 0) < 0)
goto err;
} else {
if (dvb_set_pesfilter(video_fd, vpid, DMX_PES_VIDEO,
args.dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER,
args.dvr ? 64 * 1024 : 0) < 0)
goto err;
}
}
if (apid > 0) {
if (args.silent < 2)
fprintf(stderr, _("audio pid %d\n"), apid);
if ((audio_fd = open(args.demux_dev, O_RDWR)) < 0) {
PERROR(_("failed opening '%s'"), args.demux_dev);
goto err;
}
if (args.silent < 2)
fprintf(stderr, _(" dvb_set_pesfilter %d\n"), apid);
if (dvb_set_pesfilter(audio_fd, apid, DMX_PES_AUDIO,
args.dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER,
args.dvr ? 64 * 1024 : 0) < 0)
goto err;
}
signal(SIGALRM, do_timeout);
signal(SIGTERM, do_timeout);
if (args.timeout > 0) {
signal(SIGINT, do_timeout);
alarm(args.timeout);
}
if (!check_frontend(&args, parms)) {
err = 1;
fprintf(stderr, _("frontend doesn't lock\n"));
goto err;
}
if (args.dvr) {
if (args.filename) {
file_fd = STDOUT_FILENO;
if (strcmp(args.filename, "-") != 0) {
file_fd = open(args.filename,
#ifdef O_LARGEFILE
O_LARGEFILE |
#endif
O_WRONLY | O_CREAT,
0644);
if (file_fd < 0) {
PERROR(_("open of '%s' failed"),
args.filename);
return -1;
}
}
}
if (args.silent < 2)
get_show_stats(&args, parms, 0);
if (file_fd >= 0) {
if ((dvr_fd = open(args.dvr_dev, O_RDONLY)) < 0) {
PERROR(_("failed opening '%s'"), args.dvr_dev);
goto err;
}
if (!timeout_flag)
fprintf(stderr, _("Record to file '%s' started\n"), args.filename);
copy_to_file(dvr_fd, file_fd, args.timeout, args.silent);
} else {
if (!timeout_flag)
fprintf(stderr, _("DVR interface '%s' can now be opened\n"), args.dvr_dev);
get_show_stats(&args, parms, 1);
}
if (args.silent < 2)
get_show_stats(&args, parms, 0);
}
err = 0;
err:
if (file_fd > 0)
close(file_fd);
if (dvr_fd > 0)
close(dvr_fd);
if (pat_fd > 0)
close(pat_fd);
if (pmt_fd > 0)
close(pmt_fd);
if (audio_fd > 0)
close(audio_fd);
if (video_fd > 0)
close(video_fd);
if (parms)
dvb_fe_close(parms);
if (args.confname)
free(args.confname);
if (args.demux_dev)
free(args.demux_dev);
if (args.dvr_dev)
free(args.dvr_dev);
return err;
}
int do_traffic_monitor(struct arguments *args,
struct dvb_v5_fe_parms *parms)
{
int fd, dvr_fd;
long long unsigned pidt[0x2001], wait;
int packets = 0;
struct timeval startt;
memset(pidt, 0, sizeof(pidt));
args->exit_after_tuning = 1;
check_frontend(args, parms);
if ((dvr_fd = open(args->dvr_dev, O_RDONLY)) < 0) {
PERROR("failed opening '%s'", args->dvr_dev);
return -1;
}
if (ioctl(dvr_fd, DMX_SET_BUFFER_SIZE, 1024 * 1024) == -1)
perror("DMX_SET_BUFFER_SIZE failed");
if ((fd = open(args->demux_dev, O_RDWR)) < 0) {
PERROR("failed opening '%s'", args->demux_dev);
return -1;
}
if (args->silent < 2)
fprintf(stderr, " dvb_set_pesfilter to 0x2000\n");
if (dvb_set_pesfilter(fd, 0x2000, DMX_PES_OTHER,
DMX_OUT_TS_TAP, 0) < 0) {
PERROR("couldn't set filter");
return -1;
}
gettimeofday(&startt, 0);
wait = 1000;
while (1) {
unsigned char buffer[BSIZE];
int pid, ok;
ssize_t r;
if ((r = read(dvr_fd, buffer, BSIZE)) <= 0) {
if (errno == EOVERFLOW) {
struct timeval now;
int diff;
gettimeofday(&now, 0);
diff =
(now.tv_sec - startt.tv_sec) * 1000 +
(now.tv_usec - startt.tv_usec) / 1000;
fprintf(stderr, "%.2fs: buffer overrun\n", diff / 1000.);
continue;
}
perror("read");
break;
}
if (r != BSIZE) {
fprintf(stderr, "dvbtraffic: only read %zd bytes\n", r);
break;
}
if (buffer[0] != 0x47) {
continue;
printf("desync (%x)\n", buffer[0]);
while (buffer[0] != 0x47)
read(fd, buffer, 1);
continue;
}
ok = 1;
pid = ((((unsigned) buffer[1]) << 8) |
((unsigned) buffer[2])) & 0x1FFF;
if (args->search) {
int i, sl = strlen(args->search);
ok = 0;
if (pid != 0x1fff) {
for (i = 0; i < (188 - sl); ++i) {
if (!memcmp(buffer + i, args->search, sl))
ok = 1;
}
}
}
if (ok) {
pidt[pid]++;
pidt[0x2000]++;
}
packets++;
if (!(packets & 0xFF)) {
struct timeval now;
int diff;
gettimeofday(&now, 0);
diff =
(now.tv_sec - startt.tv_sec) * 1000 +
(now.tv_usec - startt.tv_usec) / 1000;
if (diff > wait) {
if (isatty(STDOUT_FILENO))
printf("\x1b[1H\x1b[2J");
args->n_status_lines = 0;
printf(" PID FREQ SPEED TOTAL\n");
int _pid = 0;
for (_pid = 0; _pid < 0x2000; _pid++) {
if (pidt[_pid]) {
if (!args->low_traffic && (pidt[_pid] * 1000. / diff) < 1)
continue;
printf("%04x %9.2f p/s %8.1f Kbps ",
_pid,
pidt[_pid] * 1000. / diff,
pidt[_pid] * 1000. / diff * 8 * 188 / 1024);
if (pidt[_pid] * 188 / 1024)
printf("%8llu KB\n", pidt[_pid] * 188 / 1024);
else
printf(" %8llu B\n", pidt[_pid] * 188);
}
}
/* 0x2000 is the total traffic */
printf("TOT %10.2f p/s %8.1f Kbps %8llu KB\n",
pidt[_pid] * 1000. / diff,
pidt[_pid] * 1000. / diff * 8 * 188 / 1024,
pidt[_pid] * 188 / 1024);
printf("\n\n");
print_frontend_stats(stdout, args, parms);
wait += 1000;
}
}
}
close (fd);
return 0;
}
