tvheadend/channels.c

/*
 *  tvheadend, channel functions
 *  Copyright (C) 2007 Andreas <EFBFBD>man
 *
 *  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 3 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, see <http://www.gnu.org/licenses/>.
 */

#include <pthread.h>

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <fcntl.h>
#include <ctype.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>

#include <libhts/htscfg.h>

#include "tvhead.h"
#include "dvb.h"
#include "v4l.h"
#include "iptv_input.h"
#include "psi.h"
#include "channels.h"
#include "transports.h"

struct th_channel_list channels;
struct th_transport_list all_transports;
int nchannels;

struct th_channel_group_queue all_channel_groups;

th_channel_group_t *defgroup;

static int dontwritesettings;

void scanner_init(void);

/**
 *
 */
#if 0
static int
dictcmp(const char *s1, const char *s2)
{
  long int da, db;
  int n1, n2;

  while(1) {
    n1 = *s1 >= '0' && *s1 <= '9' ? 1 : 0;
    n2 = *s2 >= '0' && *s2 <= '9' ? 2 : 0;

    switch(n1 | n2) {
    case 0:
      if(*s1 != *s2)
        return *(const unsigned char *)s1 - *(const unsigned char *)s2;
      if(*s1 == 0)
        return 0;
      s1++;
      s2++;
      break;
    case 1 ... 2:
      return *(const unsigned char *)s1 - *(const unsigned char *)s2;
    case 3:
      da = strtol(s1, (char **)&s1, 10);
      db = strtol(s2, (char **)&s2, 10);
      if(da != db)
        return da - db;
      break;
    }
  }
}
#endif

static int
ch_number(const char *s1)
{
  while(*s1) {
    if(*s1 >= '0' && *s1 <= '9') {
      return strtol(s1, NULL, 10);
    }
    s1++;
  }
  return INT32_MAX;
}

static int
chcmp(const char *s1, const char *s2)
{
  int n1, n2;

  n1 = ch_number(s1);
  n2 = ch_number(s2);
  
  if(n1 != n2) {
    return n1 - n2;
  }
  return strcmp(s1, s2);
}

/**
 *
 */
static int
channelcmp(th_channel_t *a, th_channel_t *b)
{
  return chcmp(a->ch_name, b->ch_name);
}


/**
 *
 */
th_channel_group_t *
channel_group_find(const char *name, int create)
{
  th_channel_group_t *tcg;

  TAILQ_FOREACH(tcg, &all_channel_groups, tcg_global_link) {
    if(!strcmp(name, tcg->tcg_name))
      return tcg;
  }
  if(!create)
    return NULL;

  tcg = calloc(1, sizeof(th_channel_group_t));
  tcg->tcg_name = strdup(name);
  tcg->tcg_tag = tag_get();
  
  TAILQ_INIT(&tcg->tcg_channels);

  TAILQ_INSERT_TAIL(&all_channel_groups, tcg, tcg_global_link);

  if(!dontwritesettings)
    channel_settings_write();

  return tcg;
}




/**
 *
 */
void
channel_set_group(th_channel_t *ch, th_channel_group_t *tcg)
{
  if(ch->ch_group != NULL)
    TAILQ_REMOVE(&ch->ch_group->tcg_channels, ch, ch_group_link);

  ch->ch_group = tcg;
  TAILQ_INSERT_SORTED(&tcg->tcg_channels, ch, ch_group_link, channelcmp);
  if(!dontwritesettings)
    channel_settings_write();
}

/**
 *
 */
void
channel_group_destroy(th_channel_group_t *tcg)
{
  th_channel_t *ch;

  if(defgroup == tcg)
    return;

  while((ch = TAILQ_FIRST(&tcg->tcg_channels)) != NULL) {
    channel_set_group(ch, defgroup);
  }

  TAILQ_REMOVE(&all_channel_groups, tcg, tcg_global_link);
  free((void *)tcg->tcg_name);
  free(tcg);
}

/**
 *
 */
th_channel_t *
channel_find(const char *name, int create, th_channel_group_t *tcg)
{
  const char *n2;
  th_channel_t *ch;
  int l, i;
  char *cp, c;

  LIST_FOREACH(ch, &channels, ch_global_link)
    if(!strcasecmp(name, ch->ch_name))
      return ch;

  if(create == 0)
    return NULL;

  ch = calloc(1, sizeof(th_channel_t));
  ch->ch_name = strdup(name);

  l = strlen(name);
  ch->ch_sname = cp = malloc(l + 1);

  n2 = utf8toprintable(name);

  for(i = 0; i < strlen(n2); i++) {
    c = tolower(n2[i]);
    if(isalnum(c))
      *cp++ = c;
    else
      *cp++ = '-';
  }
  *cp = 0;

  free((void *)n2);

  ch->ch_index = nchannels;
  TAILQ_INIT(&ch->ch_epg_events);

  LIST_INSERT_SORTED(&channels, ch, ch_global_link, channelcmp);

  channel_set_group(ch, tcg ?: defgroup);

  ch->ch_tag = tag_get();
  nchannels++;
  return ch;
}


/**
 *
 */
static int
transportcmp(th_transport_t *a, th_transport_t *b)
{
  return a->tht_prio - b->tht_prio;
}


/**
 *
 */
int 
transport_set_channel(th_transport_t *t, th_channel_t *ch)
{
  th_stream_t *st;
  char *chname;
  const char *n;

  char pid[30];
  char lang[30];
  t->tht_channel = ch;
  LIST_INSERT_SORTED(&ch->ch_transports, t, tht_channel_link, transportcmp);

  chname = utf8toprintable(ch->ch_name);

  syslog(LOG_DEBUG, "Added service \"%s\" for channel \"%s\"",
	 t->tht_name, chname);
  free(chname);

  LIST_FOREACH(st, &t->tht_streams, st_link) {
    if(st->st_caid != 0) {
      n = psi_caid2name(st->st_caid);
    } else {
      n = htstvstreamtype2txt(st->st_type);
    }
    if(st->st_pid < 8192) {
      snprintf(pid, sizeof(pid), " on pid [%d]", st->st_pid);
    } else {
      pid[0] = 0;
    }

    if(st->st_lang[0]) {
      snprintf(lang, sizeof(lang), ", language \"%s\"", st->st_lang);
    } else {
      lang[0] = 0;
    }

    syslog(LOG_DEBUG, "   Stream \"%s\"%s%s", n, lang, pid);
  }

  return 0;
}




/**
 *
 */
static void
service_load(struct config_head *head)
{
  const char *name,  *v;
  th_transport_t *t;
  int r = 1;

  if((name = config_get_str_sub(head, "channel", NULL)) == NULL)
    return;

  t = calloc(1, sizeof(th_transport_t));

  t->tht_prio = atoi(config_get_str_sub(head, "prio", ""));

  if(0) {
#ifdef ENABLE_INPUT_DVB
  } else if((v = config_get_str_sub(head, "dvbmux", NULL)) != NULL) {
    r = dvb_configure_transport(t, v, name);
#endif
#ifdef ENABLE_INPUT_IPTV
  } else if((v = config_get_str_sub(head, "iptv", NULL)) != NULL) {
    r = iptv_configure_transport(t, v, head, name);
#endif
#ifdef ENABLE_INPUT_V4L
  } else if((v = config_get_str_sub(head, "v4lmux", NULL)) != NULL) {
    r = v4l_configure_transport(t, v, name);
#endif
  }
  if(r)
    free(t);
}

/**
 *
 */
void
transport_link(th_transport_t *t, th_channel_t *ch)
{
  transport_set_channel(t, ch);
  transport_monitor_init(t);
  LIST_INSERT_HEAD(&all_transports, t, tht_global_link);
}


/**
 *
 */
void
channels_load(void)
{
  config_entry_t *ce, *ce1, *ce2;
  const char *name;
  th_channel_group_t *tcg;
  th_channel_t *ch;

  dontwritesettings = 1;

  TAILQ_INIT(&all_channel_groups);

  TAILQ_FOREACH(ce1, &settings_list, ce_link) {
    if(ce1->ce_type != CFG_SUB || strcasecmp("channel-group", ce1->ce_key))
      continue;
    
    if((name = config_get_str_sub(&ce1->ce_sub, "name", NULL)) == NULL)
      continue;

    tcg = channel_group_find(name, 1);
    TAILQ_FOREACH(ce2, &ce1->ce_sub, ce_link) {
      if(ce2->ce_type != CFG_SUB || strcasecmp("channel", ce2->ce_key))
	continue;

      if((name = config_get_str_sub(&ce2->ce_sub, "name", NULL)) == NULL)
	continue;

      ch = channel_find(name, 1, tcg);
      
      ch->ch_teletext_rundown = 
	atoi(config_get_str_sub(&ce2->ce_sub, "teletext-rundown", "0"));
    }
  }

  tcg = channel_group_find("-disabled-", 1);
  tcg->tcg_cant_delete_me = 1;
  tcg->tcg_hidden = 1;

  defgroup = channel_group_find("Uncategorized", 1);
  defgroup->tcg_cant_delete_me = 1;

  TAILQ_FOREACH(ce, &config_list, ce_link) {
    if(ce->ce_type == CFG_SUB && !strcasecmp("service", ce->ce_key)) {
      service_load(&ce->ce_sub);
    }
  }
  dontwritesettings = 0;

}


/**
 * The index stuff should go away
 */
th_channel_t *
channel_by_index(uint32_t index)
{
  th_channel_t *ch;

  LIST_FOREACH(ch, &channels, ch_global_link)
    if(ch->ch_index == index)
      return ch;

  return NULL;
}



/**
 *
 */
th_channel_t *
channel_by_tag(uint32_t tag)
{
  th_channel_t *ch;

  LIST_FOREACH(ch, &channels, ch_global_link)
    if(ch->ch_tag == tag)
      return ch;

  return NULL;
}



/**
 *
 */
th_channel_group_t *
channel_group_by_tag(uint32_t tag)
{
  th_channel_group_t *tcg;

  TAILQ_FOREACH(tcg, &all_channel_groups, tcg_global_link)
    if(tcg->tcg_tag == tag)
      return tcg;

  return NULL;
}

void
channel_group_move_next(th_channel_group_t *tcg)
{
  th_channel_group_t *n = TAILQ_NEXT(tcg, tcg_global_link);
  if(n == NULL)
    return;

  TAILQ_REMOVE(&all_channel_groups, tcg, tcg_global_link);
  TAILQ_INSERT_AFTER(&all_channel_groups, n, tcg, tcg_global_link);
  channel_settings_write();
}

void
channel_group_move_prev(th_channel_group_t *tcg)
{
  th_channel_group_t *p = TAILQ_PREV(tcg, th_channel_group_queue,
				     tcg_global_link);
  if(p == NULL)
    return;

  TAILQ_REMOVE(&all_channel_groups, tcg, tcg_global_link);
  TAILQ_INSERT_BEFORE(p, tcg, tcg_global_link);
  channel_settings_write();
}



/**
 *
 */
void
channel_settings_write(void)
{
  FILE *fp;
  th_channel_group_t *tcg;
  th_channel_t *ch;

  if(settingsfile == NULL)
    return;

  fp = fopen(settingsfile, "w+");
  if(fp == NULL)
    return;

  TAILQ_FOREACH(tcg, &all_channel_groups, tcg_global_link) {
    fprintf(fp, "channel-group {\n"
	    "\tname = %s\n", tcg->tcg_name);
    TAILQ_FOREACH(ch, &tcg->tcg_channels, ch_group_link) {
      fprintf(fp, "\tchannel {\n"
	      "\t\tname = %s\n", ch->ch_name);
      if(ch->ch_teletext_rundown)
	fprintf(fp, "\t\tteletext-rundown = %d", ch->ch_teletext_rundown);
      fprintf(fp, "\t}\n");
    }
    fprintf(fp, "}\n");
  }
  fclose(fp);
}