/* szap -- simple zapping tool for the Linux DVB API * * szap operates on VDR (http://www.cadsoft.de/people/kls/vdr/index.htm) * satellite channel lists (e.g. from http://www.dxandy.de/cgi-bin/dvbchan.pl). * szap assumes you have a "Universal LNB" (i.e. with LOFs 9750/10600 MHz). * * Compilation: `gcc -Wall -I../../ost/include -O2 szap.c -o szap` *  or, if your DVB driver is in the kernel source tree: *              `gcc -Wall -DDVB_IN_KERNEL -O2 szap.c -o szap` * * Copyright (C) 2001 Johannes Stezenbach (js@convergence.de) * for convergence integrated media * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * 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 for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */#include <stdio.h>#include <stdlib.h>#include <limits.h>#include <string.h>#include <errno.h>#include <sys/ioctl.h>#include <sys/types.h>#include <sys/stat.h>#include <sys/poll.h>#include <fcntl.h>#include <time.h>#include <unistd.h>#include <stdint.h>#include <sys/time.h>#include <linux/dvb/frontend.h>#include <linux/dvb/dmx.h>#ifndef TRUE#define TRUE (1==1)#endif#ifndef FALSE#define FALSE (1==0)#endif/* location of channel list file */#define CHANNEL_FILE "channels.conf"/* one line of the VDR channel file has the following format: * ^name:frequency_MHz:polarization:sat_no:symbolrate:vpid:apid:?:service_id$ */#define FRONTENDDEVICE "/dev/dvb/adapter%d/frontend%d"#define DEMUXDEVICE "/dev/dvb/adapter%d/demux%d"/* LNB hi/lo band switch frequency in kHz */#define SWITCHFREQ 11700000/* LNB hi/lo band local oscillator frequencies in kHz */#define LOF_HI 10600000#define LOF_LO 9750000static int exit_after_tuning;static char *usage_str =    "\nusage: szap -l\n"    "         list known channels\n"    "       szap [options] {-n channel-number|channel_name}\n"    "         zap to channel via number or full name (case insensitive)\n"    "     -a number : use given adapter (default 0)\n"    "     -f number : use given frontend (default 0)\n"    "     -d number : use given demux (default 0)\n"    "     -c file   : read channels list from 'file'\n"    "     -x        : exit after tuning\n"    "     -r        : set up /dev/dvb/adapterX/dvr0 for TS recording\n\n";static void usage(void){   fprintf(stderr, usage_str);   exit(1);}static int set_demux(int dmxfd, int pid, int audio, int dvr){   struct dmx_pes_filter_params pesfilter;   if (pid <= 0 || pid >= 0x1fff) /* ignore this pid to allow radio services */	   return TRUE;   if (dvr) {      int buffersize = 64 * 1024;      if (ioctl(dmxfd, DMX_SET_BUFFER_SIZE, buffersize) == -1)        perror("DMX_SET_BUFFER_SIZE failed");   }   pesfilter.pid = pid;   pesfilter.input = DMX_IN_FRONTEND;   pesfilter.output = dvr ? DMX_OUT_TS_TAP : DMX_OUT_DECODER;   pesfilter.pes_type = audio ? DMX_PES_AUDIO : DMX_PES_VIDEO;   pesfilter.flags = DMX_IMMEDIATE_START;   if (ioctl(dmxfd, DMX_SET_PES_FILTER, &pesfilter) == -1) {      fprintf(stderr, "DMX_SET_PES_FILTER failed "	      "(PID = 0x%04x): %d %m\n", pid, errno);      return FALSE;   }   return TRUE;}struct diseqc_cmd {   struct dvb_diseqc_master_cmd cmd;   uint32_t wait;};void diseqc_send_msg(int fd, fe_sec_voltage_t v, struct diseqc_cmd *cmd,		     fe_sec_tone_mode_t t, fe_sec_mini_cmd_t b){   if (ioctl(fd, FE_SET_TONE, SEC_TONE_OFF) == -1)      perror("FE_SET_TONE failed");   if (ioctl(fd, FE_SET_VOLTAGE, v) == -1)      perror("FE_SET_VOLTAGE failed");   usleep(15 * 1000);   if (ioctl(fd, FE_DISEQC_SEND_MASTER_CMD, &cmd->cmd) == -1)      perror("FE_DISEQC_SEND_MASTER_CMD failed");   usleep(cmd->wait * 1000);   usleep(15 * 1000);   if (ioctl(fd, FE_DISEQC_SEND_BURST, b) == -1)      perror("FE_DISEQC_SEND_BURST failed");   usleep(15 * 1000);   if (ioctl(fd, FE_SET_TONE, t) == -1)      perror("FE_SET_TONE failed");}/* digital satellite equipment control, * specification is available from http://www.eutelsat.com/ */static int diseqc(int secfd, int sat_no, int pol_vert, int hi_band){   struct diseqc_cmd cmd =       { {{0xe0, 0x10, 0x38, 0xf0, 0x00, 0x00}, 4}, 0 };   /* param: high nibble: reset bits, low nibble set bits,    * bits are: option, position, polarizaion, band    */   cmd.cmd.msg[3] =       0xf0 | (((sat_no * 4) & 0x0f) | (hi_band ? 1 : 0) | (pol_vert ? 0 : 2));   diseqc_send_msg(secfd, pol_vert ? SEC_VOLTAGE_13 : SEC_VOLTAGE_18,		   &cmd, hi_band ? SEC_TONE_ON : SEC_TONE_OFF,		   (sat_no / 4) % 2 ? SEC_MINI_B : SEC_MINI_A);   return TRUE;}static int do_tune(int fefd, unsigned int ifreq, unsigned int sr){   struct dvb_frontend_parameters tuneto;   struct dvb_frontend_event ev;   /* discard stale QPSK events */   while (1) {      if (ioctl(fefd, FE_GET_EVENT, &ev) == -1)	 break;   }   tuneto.frequency = ifreq;   tuneto.inversion = INVERSION_AUTO;   tuneto.u.qpsk.symbol_rate = sr;   tuneto.u.qpsk.fec_inner = FEC_AUTO;   if (ioctl(fefd, FE_SET_FRONTEND, &tuneto) == -1) {      perror("FE_SET_FRONTEND failed");      return FALSE;   }   return TRUE;}staticint check_frontend (int fe_fd, int dvr){   fe_status_t status;   uint16_t snr, signal;   uint32_t ber, uncorrected_blocks;   int timeout = 0;   do {      if (ioctl(fe_fd, FE_READ_STATUS, &status) == -1)         perror("FE_READ_STATUS failed");      /* some frontends might not support all these ioctls, thus we       * avoid printing errors */      if (ioctl(fe_fd, FE_READ_SIGNAL_STRENGTH, &signal) == -1)         signal = -2;      if (ioctl(fe_fd, FE_READ_SNR, &snr) == -1)         snr = -2;      if (ioctl(fe_fd, FE_READ_BER, &ber) == -1)         ber = -2;      if (ioctl(fe_fd, FE_READ_UNCORRECTED_BLOCKS, &uncorrected_blocks) == -1)         uncorrected_blocks = -2;      printf ("status %02x | signal %04x | snr %04x | ber %08x | unc %08x | ",	      status, signal, snr, ber, uncorrected_blocks);      if (status & FE_HAS_LOCK)	 printf("FE_HAS_LOCK");      printf("\n");      if (exit_after_tuning && ((status & FE_HAS_LOCK) || (++timeout >= 10)))         break;      usleep(1000000);   } while (1);   return 0;}staticint zap_to(unsigned int adapter, unsigned int frontend, unsigned int demux,      unsigned int sat_no, unsigned int freq, unsigned int pol,      unsigned int sr, unsigned int vpid, unsigned int apid, int dvr){   char fedev[128], dmxdev[128];   int fefd, videofd, audiofd;   uint32_t ifreq;   int hiband, result;   struct dvb_frontend_info fe_info;   snprintf(fedev, sizeof(fedev), FRONTENDDEVICE, adapter, frontend);   snprintf(dmxdev, sizeof(dmxdev), DEMUXDEVICE, adapter, demux);   printf("using '%s' and '%s'\n", fedev, dmxdev);   if ((fefd = open(fedev, O_RDWR | O_NONBLOCK)) < 0) {      perror("opening frontend failed");      return FALSE;   }   result = ioctl(fefd, FE_GET_INFO, &fe_info);   if (result < 0) {      perror("ioctl FE_GET_INFO failed");      close(fefd);      return FALSE;   }   if (fe_info.type != FE_QPSK) {      fprintf(stderr, "frontend device is not a QPSK (DVB-S) device!\n");      close(fefd);      return FALSE;   }   if ((videofd = open(dmxdev, O_RDWR)) < 0) {      perror("opening video demux failed");      close(fefd);      return FALSE;   }   if ((audiofd = open(dmxdev, O_RDWR)) < 0) {      perror("opening audio demux failed");      close(videofd);      close(fefd);      return FALSE;   }   hiband = (freq >= SWITCHFREQ);   if (hiband)      ifreq = freq - LOF_HI;   else      ifreq = freq - LOF_LO;   result = FALSE;   if (diseqc(fefd, sat_no, pol, hiband))      if (do_tune(fefd, ifreq, sr))	 if (set_demux(videofd, vpid, 0, dvr))	    if (set_demux(audiofd, apid, 1, dvr))	       result = TRUE;   check_frontend (fefd, dvr);      close(audiofd);   close(videofd);   close(fefd);   return result;}static int read_channels(const char *filename, int list_channels,			 uint32_t chan_no, const char *chan_name,			 unsigned int adapter, unsigned int frontend,			 unsigned int demux, int dvr){   FILE *cfp;   char buf[4096];   char *field, *tmp;   unsigned int line = 0;   unsigned int freq, pol, sat_no, sr, vpid, apid;   if (!(cfp = fopen(filename, "r"))) {      fprintf(stderr, "error opening channel list '%s': %d %m\n",	      filename, errno);      return FALSE;   }   while (!feof(cfp)) {      if (fgets(buf, sizeof(buf), cfp)) {	 line++;	 if (chan_no && chan_no != line)	    continue;	 tmp = buf;	 field = strsep(&tmp, ":");	 if (!field)	    goto syntax_err;	 if (list_channels) {	    printf("%03u %s\n", line, field);	    continue;	 }	 if (chan_name && strcasecmp(chan_name, field) != 0)	    continue;	 printf("zapping to '%s':\n", field);	 if (!(field = strsep(&tmp, ":")))	    goto syntax_err;	 freq = strtoul(field, NULL, 0);	 if (!(field = strsep(&tmp, ":")))	    goto syntax_err;	 pol = (field[0] == 'h' ? 0 : 1);	 if (!(field = strsep(&tmp, ":")))	    goto syntax_err;	 sat_no = strtoul(field, NULL, 0);	 if (!(field = strsep(&tmp, ":")))	    goto syntax_err;	 sr = strtoul(field, NULL, 0) * 1000;	 if (!(field = strsep(&tmp, ":")))	    goto syntax_err;	 vpid = strtoul(field, NULL, 0);	 if (!(field = strsep(&tmp, ":")))	    goto syntax_err;	 apid = strtoul(field, NULL, 0);	 printf("sat %u, frequency = %u MHz %c, symbolrate %u, "		"vpid = 0x%04x, apid = 0x%04x\n",		sat_no, freq, pol ? 'V' : 'H', sr, vpid, apid);	 fclose(cfp);	 if (zap_to(adapter, frontend, demux,		  sat_no, freq * 1000, pol, sr, vpid, apid, dvr))	    return TRUE;	 return FALSE;       syntax_err:	 fprintf(stderr, "syntax error in line %u: '%s'\n", line, buf);      } else if (ferror(cfp)) {	 fprintf(stderr, "error reading channel list '%s': %d %m\n",		 filename, errno);	 fclose(cfp);	 return FALSE;      } else	 break;   }   fclose(cfp);   if (!list_channels) {      fprintf(stderr, "channel not found\n");      return FALSE;   }   return TRUE;}int main(int argc, char *argv[]){   const char *home;   char chanfile[2 * PATH_MAX];   int list_channels = 0;   unsigned int chan_no = 0;   const char *chan_name = NULL;   unsigned int adapter = 0, frontend = 0, demux = 0, dvr = 0;   int opt, copt = 0;   while ((opt = getopt(argc, argv, "hlrn:a:f:d:c:x")) != -1) {      switch (opt)      {	 case '?':	 case 'h':	 default:	    usage();	 case 'l':	    list_channels = 1;	    break;	 case 'r':	    dvr = 1;	    break;	 case 'n':	    chan_no = strtoul(optarg, NULL, 0);	    break;	 case 'a':	    adapter = strtoul(optarg, NULL, 0);	    break;	 case 'f':	    frontend = strtoul(optarg, NULL, 0);	    break;	 case 'd':	    demux = strtoul(optarg, NULL, 0);	    break;	 case 'c':	    copt = 1;	    strncpy(chanfile, optarg, sizeof(chanfile));	    break;	 case 'x':	    exit_after_tuning = 1;      }   }   if (optind < argc)      chan_name = argv[optind];   if (chan_name && chan_no)      usage();   if (list_channels && (chan_name || chan_no))      usage();   if (!list_channels && !chan_name && !chan_no)      usage();   if (!copt) {       if (!(home = getenv("HOME"))) {          fprintf(stderr, "error: $HOME not set\n");          return TRUE;       }       strncpy(chanfile, home, sizeof(chanfile));       strcat(chanfile, "/.szap/" CHANNEL_FILE);   }   printf("reading channels from file '%s'\n", chanfile);   if (!read_channels(chanfile, list_channels, chan_no, chan_name,	    adapter, frontend, demux, dvr))      return TRUE;   return FALSE;}