htsp.c

/*

pidvbip - tvheadend client for the Raspberry Pi

(C) Dave Chapman 2012-2013

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 "common.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <string.h>

#include "sha1.h"
#include "htsp.h"
#include "debug.h"

typedef struct HTSSHA1 {
    uint64_t count;
    uint8_t buffer[64];
    uint32_t state[5];
} HTSSHA1;


void htsp_init(struct htsp_t* htsp)
{
    htsp->subscriptionId = 0;
    htsp->subscriptionServer = -1;
    htsp->sync_completed = 0;
}

static int create_tcp_socket()
{
  int sock;
  if((sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0){
    perror("Can't create TCP socket");
    exit(1);
  }
  return sock;
}

static char *get_ip(char *host)
{
  struct hostent *hent;
  char *ip = calloc(INET_ADDRSTRLEN,1);

  if((hent = gethostbyname(host)) == NULL)
  {
    herror("Can't get IP");
    exit(1);
  }
  if(inet_ntop(AF_INET, (void *)hent->h_addr_list[0], ip, INET_ADDRSTRLEN) == NULL)
  {
    perror("Can't resolve host");
    exit(1);
  }
  return ip;
}

int htsp_connect(struct htsp_t* htsp, int server)
{
  int res;
  int i;

  htsp->sock[server] = create_tcp_socket();
  if (htsp->ip[server] == NULL)
    htsp->ip[server] = get_ip(htsp->host[server]);

  fprintf(stderr,"Connecting to server %d: %s (%s) port %d...\n",server, htsp->host[server],htsp->ip[server],htsp->port[server]);    

  htsp->remote[server] = (struct sockaddr_in *)malloc(sizeof(struct sockaddr_in));
  htsp->remote[server]->sin_family = AF_INET;

  res = inet_pton(AF_INET, htsp->ip[server], (void *)(&(htsp->remote[server]->sin_addr.s_addr)));

  if (res < 0) {
    perror("Can't set remote->sin_addr.s_addr");
    exit(1);
  } else if (res == 0) {
    fprintf(stderr, "%s is not a valid IP address\n", htsp->ip[server]);
    return 1;
  }
  htsp->remote[server]->sin_port = htons(htsp->port[server]);
 
  if (connect(htsp->sock[server], (struct sockaddr *)htsp->remote[server], sizeof(struct sockaddr)) < 0){
    perror("Could not connect");
    return 2;
  }
  return 0;
}

static char* hmf_labels[] = { "", "MAP","S64","STR","BIN","LIST","DBL" };

static int get_uint32_be(unsigned char* buf)
{
   return((buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3]);
}

static void put_uint32_be(unsigned char* buf, uint32_t x)
{
   buf[0] = (x >> 24) & 0xff;
   buf[1] = (x >> 16) & 0xff;
   buf[2] = (x >>  8) & 0xff;
   buf[3] = x & 0xff;
}

void htsp_dump_binary(unsigned char* buf, int len)
{
    int i;
    uint64_t x = 0;

    while (len > 0) {
       int type = buf[0]; if (type > 6) { type = 0; }
       int namelength = buf[1];
       int datalength = get_uint32_be(buf + 2);
       buf += 6; len -= 6;

       fprintf(stderr,"type=%d, datalen=%d %s: ",type,datalength,hmf_labels[type]);

       for (i=0;i<namelength;i++) { fprintf(stderr,"%c",buf[i]); }
       buf += namelength; len -= namelength;

       fprintf(stderr,"=");
       switch (type) {
           case HMF_STR:
             fprintf(stderr,"\"");
             for (i=0;i<datalength;i++) { fprintf(stderr,"%c",buf[i]); }
             fprintf(stderr,"\"");
             break;

           case HMF_BIN:
             fprintf(stderr,"\"");
             //for (i=0;i<datalength;i++) { fprintf(stderr," 0x%02x",buf[i]); }
             fprintf(stderr,"\"");
             break;

           case HMF_S64:
             x = 0;
             for (i=datalength-1;i>=0;i--) {
                x <<= 8;
                x |= buf[i];
             }
             fprintf(stderr,"%lld",x);
             fprintf(stderr," [");
             for (i=0;i<datalength;i++) { fprintf(stderr," 0x%02x",buf[i]); }
             fprintf(stderr," ]");
             break;

           case HMF_LIST:
           case HMF_MAP:
             fprintf(stderr,"\n");
	     htsp_dump_binary(buf, datalength);
             break;

           default:
             break;
       }
       fprintf(stderr,"\n");
       buf += datalength; len -= datalength;
    }

}

void htsp_dump_message(struct htsp_message_t* msg)
{
  htsp_dump_binary(msg->msg + 4, msg->msglen - 4);
}


void htsp_destroy_message(struct htsp_message_t* msg)
{
  free(msg->msg);
}

int htsp_create_message(struct htsp_message_t* msg, ...)
{
  va_list argp;
  int type;
  char* fieldname;
  int fieldnamelen;
  int datalen;
  char* s;
  int i;
  uint64_t ii;
  unsigned char* d;
  unsigned char* p;
  
  msg->msglen = 4;

  va_start(argp,msg);

  type = va_arg(argp, int);
  while (type != HMF_NULL) {
    msg->msglen++;

    fieldname = va_arg(argp,char*);
    msg->msglen += strlen(fieldname) + 1 + 4;

    switch(type) {
      case HMF_STR:
         s = va_arg(argp,char*);
         msg->msglen += strlen(s);
         break;

      case HMF_S64:
         i = va_arg(argp,int);
         ii = i;
         while (ii > 0) {
           msg->msglen++;
           ii >>= 8;
         }
         break;

      case HMF_BIN:
         i = va_arg(argp,int);
         va_arg(argp,unsigned char*);
         msg->msglen += i;
         break;

      default:
        fprintf(stderr,"FATAL ERROR: Unsupported type in htsp_create_message (%d)\n",type);
        exit(1);
        break;
    }

    type = va_arg(argp, int);
  }

  va_end(argp);

  msg->msg = malloc(msg->msglen);

  p = msg->msg;

  put_uint32_be(p,msg->msglen - 4); p += 4;
  va_start(argp,msg);

  type = va_arg(argp, int);
  while (type != HMF_NULL) {
    *p++ = type;

    fieldname = va_arg(argp,char*);
    fieldnamelen = strlen(fieldname);

    *p++ = fieldnamelen;
    
    switch(type) {
      case HMF_STR:
         s = va_arg(argp,char*);
         datalen = strlen(s);
         put_uint32_be(p,datalen); p += 4;
         memcpy(p,fieldname,fieldnamelen); p += fieldnamelen;
         memcpy(p,s,datalen); p += datalen;
         break;

      case HMF_S64:
         i = va_arg(argp,int);
         ii = i;
         datalen = 0;
         while (ii > 0) {
           datalen++;
           ii >>= 8;
         }
         put_uint32_be(p,datalen); p += 4;
         memcpy(p,fieldname,fieldnamelen); p += fieldnamelen;

         ii = i;
         while (ii > 0) {
           *p++ = (ii & 0xff);
           ii >>= 8;
         }
         break;

      case HMF_BIN:
         datalen = va_arg(argp,int);
         d = va_arg(argp,unsigned char*);
         put_uint32_be(p,datalen); p += 4;
         memcpy(p,fieldname,fieldnamelen); p += fieldnamelen;
         memcpy(p,d,datalen); p += datalen;
         break;

      default:
        fprintf(stderr,"FATAL ERROR: Unsupported type in htsp_create_message (%d)\n",type);
        exit(1);
        break;
    }

    type = va_arg(argp, int);
  }

  va_end(argp);

  return 0;
}

int htsp_send_message(struct htsp_t* htsp, int server, struct htsp_message_t* msg)
{
  int res;
  int sent = 0;

  while(sent < msg->msglen)
  {
    res = send(htsp->sock[server], msg->msg + sent, msg->msglen-sent, 0);
    if(res == -1){
      perror("Can't send query");
      return 1;
    }
    sent += res;
  }

  return 0;
}

int safe_recv(int sock, unsigned char* p, int bytesleft)
{
  int res;

  while (bytesleft) {
    res = recv(sock, p, bytesleft, 0);

    if (res < 0) {
      fprintf(stderr,"Error in recv\n");
      return res;
    }

    p += res;
    bytesleft -= res;
  }

  return 0;
}

static int do_htsp_recv_message(int fd, struct htsp_message_t* msg)
{
  unsigned char buf[4];
  int res;
  //fprintf(stderr,"Waiting for response...\n");
  res = safe_recv(fd, buf, 4);
  
  if (res < 0) {
    fprintf(stderr,"Error in recv - res=%d\n",res);
    return 2;
  }

  msg->msglen = get_uint32_be(buf);

  msg->msg = malloc(msg->msglen + 4);
  if (msg->msg == NULL) {
    fprintf(stderr,"FATAL ERROR - out of memory (tried to allocate %d bytes)\n",msg->msglen + 4);
    exit(1);
  }
  memcpy(msg->msg, buf, 4);

  res = safe_recv(fd, msg->msg+4, msg->msglen);

  return 0;
}

int htsp_recv_message(struct htsp_t* htsp, int server, struct htsp_message_t* msg, int timeout)
{
  int res;
  int max_fd;
  int i;
  fd_set fds;

  if (timeout) {
    struct timeval tv = { 0L, timeout*1000 };  /* timeout in ms */
    FD_ZERO(&fds);
    if (server >= 0) {
      FD_SET(htsp->sock[server], &fds);
      max_fd = htsp->sock[server];
    } else {
      max_fd = 0;
      for (i=0;i<htsp->numservers;i++) {
        FD_SET(htsp->sock[i], &fds);
        if (max_fd < htsp->sock[i]) 
          max_fd = htsp->sock[i];
      }
    }
    if (select(max_fd + 1, &fds, NULL, NULL, &tv)==0) {
      return 1;
    }
  }

  if (server >= 0) {
    msg->server = server;
    return do_htsp_recv_message(htsp->sock[server], msg);
  } else {
    for (i=0;i<htsp->numservers;i++) {
      if (FD_ISSET(htsp->sock[i],&fds)) {
        msg->server = i;
        return do_htsp_recv_message(htsp->sock[i], msg);
      }
    }
  }


}

int htsp_login(struct htsp_t* htsp, int server, char* tvh_user, char* tvh_pass)
{
  struct htsp_message_t msg;
  int res;
  HTSSHA1 *shactx = (HTSSHA1*)malloc(hts_sha1_size);
  uint8_t d[20];
  unsigned char* chall;
  int chall_len = 0,res_access = 0;

  htsp_create_message(&msg,HMF_STR,"method","hello",
                           HMF_STR,"clientname","pidvbip",
                           HMF_S64,"htspversion",1,
                           HMF_S64,"seq",1,
                           HMF_NULL);

  //fprintf(stderr,"Sending hello message - %d bytes\n",msg.msglen);

  if ((res = htsp_send_message(htsp,server,&msg)) > 0) {
    fprintf(stderr,"Could not send message\n");
    return 1;
  }

  htsp_destroy_message(&msg);

  res = htsp_recv_message(htsp,server,&msg,0);

  if (res > 0) {
    fprintf(stderr,"Error receiving hello response\n");
    return 1;
  } else {
    htsp_get_string(&msg, "method");
    htsp_get_bin(&msg, "challenge", &chall, &chall_len);
  }

  htsp_destroy_message(&msg);

  if ((tvh_user) && (tvh_pass)) {
    // Now authenticate
    fprintf(stderr,"Authenticating with user: %s pass: %s\n",tvh_user,tvh_pass);
    hts_sha1_init(shactx);
    hts_sha1_update(shactx, (const uint8_t *)tvh_pass, strlen(tvh_pass));
    hts_sha1_update(shactx, (const uint8_t *)chall, chall_len);
    hts_sha1_final(shactx, d);
    htsp_create_message(&msg,HMF_STR,"method","authenticate",
                             HMF_STR,"username",tvh_user,
                             HMF_BIN,"digest",20,d,
                             HMF_NULL);
    if ((res = htsp_send_message(htsp,server,&msg)) > 0) {
      fprintf(stderr,"Could not send message (authenticate)\n");
      return 1;
    };
    htsp_destroy_message(&msg);

    res = htsp_recv_message(htsp,server,&msg,0);
    if (res > 0) {
      fprintf(stderr,"Error receiving login response\n");
      return 1;
    } else {
      htsp_get_int(&msg, "noaccess",&res_access);
      fprintf(stderr,"Received htsp (login): %d - noaccess %d\n",res,res_access);
      if (res_access == 1) {
        fprintf(stderr,"Login FAILURE - No access with username and password\n");
        return 1;
      };
    };
    htsp_destroy_message(&msg);
  }
  return 0;
}

char* htsp_get_string(struct htsp_message_t* msg, char* name)
{
  unsigned char* buf = msg->msg;
  int len = msg->msglen;
  int msglen = get_uint32_be(buf); buf += 4; len -= 4;
  int matchlen = strlen(name);
  char* s;

  while (len > 0) {
    int type = buf[0]; if (type > 6) { type = 0; }
    int namelength = buf[1];
    int datalength = get_uint32_be(buf + 2);
    buf += 6; len -= 6;

    if ((type == HMF_STR) && (namelength==matchlen) && (memcmp(buf,name,matchlen)==0)) {
      s = malloc(datalength+1);
      memcpy(s,buf + namelength,datalength);
      s[datalength]=0;
      return s;
    }

    buf += namelength + datalength;
    len -= namelength + datalength;
  }

  return NULL;
}

int htsp_get_int(struct htsp_message_t* msg, char* name, int32_t* val)
{
  unsigned char* buf = msg->msg;
  int len = msg->msglen;
  int msglen = get_uint32_be(buf); buf += 4; len -= 4;
  int matchlen = strlen(name);

  while (len > 0) {
    int type = buf[0]; if (type > 6) { type = 0; }
    int namelength = buf[1];
    int datalength = get_uint32_be(buf + 2);
    buf += 6; len -= 6;

    if ((type == HMF_S64) && (namelength==matchlen) && (memcmp(buf,name,matchlen)==0)) {
      buf += namelength + datalength - 1;  // We decode backwards, it's little-endian
      *val = 0;
      while (datalength > 0) {
	*val <<= 8;
        *val |= *buf--;
        datalength--;
      }
      return 0;
    }

    buf += namelength + datalength;
    len -= namelength + datalength;
  }

  return 1;
}

int htsp_get_uint(struct htsp_message_t* msg, char* name, uint32_t* val)
{
  return htsp_get_int(msg,name,(int32_t*)val);
}

int htsp_get_int64(struct htsp_message_t* msg, char* name, int64_t* val)
{
  unsigned char* buf = msg->msg;
  int len = msg->msglen;
  int msglen = get_uint32_be(buf); buf += 4; len -= 4;
  int matchlen = strlen(name);

  while (len > 0) {
    int type = buf[0]; if (type > 6) { type = 0; }
    int namelength = buf[1];
    int datalength = get_uint32_be(buf + 2);
    buf += 6; len -= 6;

    if ((type == HMF_S64) && (namelength==matchlen) && (memcmp(buf,name,matchlen)==0)) {
      buf += namelength + datalength - 1;  // We decode backwards, it's little-endian
      *val = 0;
      while (datalength > 0) {
	*val <<= 8;
        *val |= *buf--;
        datalength--;
      }
      return 0;
    }

    buf += namelength + datalength;
    len -= namelength + datalength;
  }

  return 1;
}


int htsp_get_bin(struct htsp_message_t* msg, char* name, unsigned char** data,int* size)
{
  unsigned char* buf = msg->msg;
  int len = msg->msglen;
  int msglen = get_uint32_be(buf); buf += 4; len -= 4;
  int matchlen = strlen(name);

  while (len > 0) {
    int type = buf[0]; if (type > 6) { type = 0; }
    int namelength = buf[1];
    int datalength = get_uint32_be(buf + 2);
    buf += 6; len -= 6;

    if ((type == HMF_BIN) && (namelength==matchlen) && (memcmp(buf,name,matchlen)==0)) {
      *data = buf + namelength;
      *size = datalength;
      return 0;
    }

    buf += namelength + datalength;
    len -= namelength + datalength;
  }

  return 1;
}

int htsp_get_list(struct htsp_message_t* msg, char* name, unsigned char** data,int* size)
{
  unsigned char* buf = msg->msg;
  int len = msg->msglen;
  int msglen = get_uint32_be(buf); buf += 4; len -= 4;
  int matchlen = strlen(name);

  while (len > 0) {
    int type = buf[0]; if (type > 6) { type = 0; }
    int namelength = buf[1];
    int datalength = get_uint32_be(buf + 2);
    buf += 6; len -= 6;

    if ((type == HMF_LIST) && (namelength==matchlen) && (memcmp(buf,name,matchlen)==0)) {
      *data = buf + namelength;
      *size = datalength;
      return 0;
    }

    buf += namelength + datalength;
    len -= namelength + datalength;
  }

  return 1;
}


/* Calculate a priority value (higher the better) for a language string.
   This is currently hard-coded, but should be user-configurable. */
static int audio_lang_priority(char* lang)
{
  if (!lang) return 0;

  /* Whitelist */
  if (!strcmp(lang,"eng")) return 99;
  /* The following are the codes used on Spanish DVB-T for alternative (mostly English) language tracks */
  if (!strcmp(lang,"v.o")) return 1;
  if (!strcmp(lang,"und")) return 1;
  if (!strcmp(lang,"qaa")) return 1;
  if (!strcmp(lang,"mul")) return 1;

  /* Blacklist */
  if (!strcmp(lang,"cat")) return -1;
  if (!strcmp(lang,"spa")) return -2;

  return 0;
}

int htsp_parse_subscriptionStart(struct htsp_message_t* msg, struct htsp_subscription_t* subscription)
{
  unsigned char* list;
  int listlen;
  int i;

  if (htsp_get_list(msg,"streams",&list,&listlen) > 0)
    return 1;

  unsigned char* buf = list;

  subscription->numstreams = 0;
  subscription->numaudiostreams = 0;
  subscription->videostream = -1;
  subscription->audiostream = -1;

  int len = listlen;
  while (len > 0) {
    subscription->numstreams++;
    int type = buf[0]; if (type > 6) { type = 0; }
    int namelength = buf[1];
    int datalength = get_uint32_be(buf + 2);
    buf += 6 + namelength + datalength;
    len -= 6 + namelength + datalength;
  }

  subscription->streams = malloc(subscription->numstreams * sizeof(struct htsp_stream_t));

  len = listlen;
  buf = list;
  i = 0;
  while (len > 0) {
    int type = buf[0]; if (type > 6) { type = 0; }
    int namelength = buf[1];
    int datalength = get_uint32_be(buf + 2);
    buf += 6; len -= 6;

    struct htsp_message_t tmpmsg;
    tmpmsg.msg = buf + namelength - 4;
    tmpmsg.msglen = datalength;

    htsp_get_int(&tmpmsg,"index",&subscription->streams[i].index);

    char* typestr = htsp_get_string(&tmpmsg,"type");
    char* lang = htsp_get_string(&tmpmsg,"language");

    fprintf(stderr,"%d %s %s\n",subscription->streams[i].index, typestr, lang);

    if (typestr==NULL)
      return 1;    

    if (lang)
      strncpy(subscription->streams[i].lang,lang,4);
   
    if (strcmp(typestr,"MPEG2VIDEO")==0) {
      subscription->streams[i].type = HMF_STREAM_VIDEO;
      subscription->streams[i].codec = HMF_VIDEO_CODEC_MPEG2;
      subscription->videostream = i;
      DEBUGF("Video stream is index %d: MPEG-2\n",subscription->streams[i].index);
    } else if (strcmp(typestr,"H264")==0) {
      subscription->streams[i].type = HMF_STREAM_VIDEO;
      subscription->streams[i].codec = HMF_VIDEO_CODEC_H264;
      subscription->videostream = i;
      DEBUGF("Video stream is index %d: H264\n",subscription->streams[i].index);
    } else if (strcmp(typestr,"MPEG2AUDIO")==0) {
      subscription->streams[i].type = HMF_STREAM_AUDIO;
      subscription->streams[i].codec = HMF_AUDIO_CODEC_MPEG;
      if ((subscription->audiostream == -1) || (audio_lang_priority(lang) > audio_lang_priority(subscription->streams[subscription->audiostream].lang))) {
        subscription->audiostream = i;
        fprintf(stderr,"Audio stream is index %d: MPEG (i=%d)\n",subscription->streams[i].index,i);
      }
    } else if (strcmp(typestr,"AAC")==0) {
      subscription->streams[i].type = HMF_STREAM_AUDIO;
      subscription->streams[i].codec = HMF_AUDIO_CODEC_AAC;
      if ((subscription->audiostream == -1) || (audio_lang_priority(lang) > audio_lang_priority(subscription->streams[subscription->audiostream].lang))) {
        subscription->audiostream = i;
        fprintf(stderr,"Audio stream is index %d: AAC\n",subscription->streams[i].index);
      }
    } else if (strcmp(typestr,"AC3")==0) {
      subscription->streams[i].type = HMF_STREAM_AUDIO;
      subscription->streams[i].codec = HMF_AUDIO_CODEC_AC3;
      /* AC3 takes precedence over MPEG, if the language is the same or greater priority */
      if ((subscription->audiostream == -1) || ((subscription->streams[subscription->audiostream].codec == HMF_AUDIO_CODEC_MPEG) || (audio_lang_priority(lang) >= audio_lang_priority(subscription->streams[subscription->audiostream].lang)))) {
        subscription->audiostream = i;
        fprintf(stderr,"Audio stream is index %d: A/52\n",subscription->streams[i].index);
      }
    } else if (strcmp(typestr,"DVBSUB")==0) {
      subscription->streams[i].type = HMF_STREAM_SUB;
      subscription->streams[i].codec = HMF_SUB_CODEC_DVBSUB;
    } else {
      subscription->streams[i].type = HMF_UNKNOWN;
      subscription->streams[i].codec = HMF_UNKNOWN;
      fprintf(stderr,"Warning: Unknown stream type \"%s\"\n",typestr);
    }

    if (subscription->streams[i].type == HMF_STREAM_VIDEO) {
      htsp_get_int(&tmpmsg,"width",&subscription->streams[i].width);
      htsp_get_int(&tmpmsg,"height",&subscription->streams[i].height);
    } else if (subscription->streams[i].type == HMF_STREAM_AUDIO) {
      subscription->numaudiostreams++;
      htsp_get_int(&tmpmsg,"audio_type",&subscription->streams[i].audio_type);
    }

    free(typestr);
    if (lang != NULL) free(lang);

    buf += namelength + datalength;
    len -= namelength + datalength;
    i++;
  }

  return 0;
}

int htsp_send_skip(struct htsp_t* htsp, int server, int time)
{
  int res;
  struct htsp_message_t msg;

  res = htsp_create_message(&msg,HMF_STR,"method","subscriptionSkip",
                                 HMF_S64,"time",time*1000000,
                                 HMF_S64,"subscriptionId",htsp->subscriptionId,
                                 HMF_NULL);
  res = htsp_send_message(htsp,server,&msg);

  fprintf(stderr,"Sent subscriptionSkip(%d) message, res=%d\n",time,res);

  htsp_destroy_message(&msg);

  return res;
}