vdr-plugin-iptv/protocolfile.c

/*
 * protocolfile.c: IPTV plugin for the Video Disk Recorder
 *
 * See the README file for copyright information and how to reach the author.
 *
 */

#include <fcntl.h>
#include <unistd.h>

#include <vdr/device.h>

#include "common.h"
#include "config.h"
#include "log.h"
#include "protocolfile.h"

cIptvProtocolFile::cIptvProtocolFile()
: fileLocationM(strdup("")),
  fileDelayM(0),
  fileStreamM(NULL),
  isActiveM(false)
{
  debug1("%s", __PRETTY_FUNCTION__);
}

cIptvProtocolFile::~cIptvProtocolFile()
{
  debug1("%s", __PRETTY_FUNCTION__);
  // Drop open handles
  cIptvProtocolFile::Close();
  // Free allocated memory
  free(fileLocationM);
}

bool cIptvProtocolFile::OpenFile(void)
{
  debug1("%s", __PRETTY_FUNCTION__);
  // Check that stream address is valid
  if (!isActiveM && !isempty(fileLocationM)) {
     fileStreamM = fopen(fileLocationM, "rb");
     ERROR_IF_RET(!fileStreamM || ferror(fileStreamM), "fopen()", return false);
     // Update active flag
     isActiveM = true;
     }
  return true;
}

void cIptvProtocolFile::CloseFile(void)
{
  debug1("%s", __PRETTY_FUNCTION__);
  // Check that file stream is valid
  if (isActiveM && !isempty(fileLocationM)) {
     fclose(fileStreamM);
     // Update active flag
     isActiveM = false;
     }
}

int cIptvProtocolFile::Read(unsigned char* bufferAddrP, unsigned int bufferLenP)
{
   debug16("%s (, %u)", __PRETTY_FUNCTION__, bufferLenP);
   // Check errors
   if (!fileStreamM || ferror(fileStreamM)) {
      debug1("%s (, %d) Stream error", __PRETTY_FUNCTION__, bufferLenP);
      return -1;
      }
   // Rewind if EOF
   if (feof(fileStreamM))
      rewind(fileStreamM);
   // Sleep before reading the file stream to prevent aggressive busy looping
   // and prevent transfer ringbuffer overflows
   if (fileDelayM)
      cCondWait::SleepMs(fileDelayM);
   // This check is to prevent a race condition where file may be switched off
   // during the sleep and buffers are disposed. Check here that the plugin is
   // still active before accessing the buffers
   if (isActiveM)
      return (int)fread(bufferAddrP, sizeof(unsigned char), bufferLenP, fileStreamM);
   return -1;
}

bool cIptvProtocolFile::Open(void)
{
  debug1("%s", __PRETTY_FUNCTION__);
  // Open the file stream
  OpenFile();
  return true;
}

bool cIptvProtocolFile::Close(void)
{
  debug1("%s", __PRETTY_FUNCTION__);
  // Close the file stream
  CloseFile();
  return true;
}

bool cIptvProtocolFile::SetSource(const char* locationP, const int parameterP, const int indexP)
{
  debug1("%s (%s, %d, %d)", __PRETTY_FUNCTION__, locationP, parameterP, indexP);
  if (!isempty(locationP)) {
     // Close the file stream
     CloseFile();
     // Update stream address and port
     fileLocationM = strcpyrealloc(fileLocationM, locationP);
     fileDelayM = parameterP;
     // Open the file for input
     OpenFile();
     }
  return true;
}

bool cIptvProtocolFile::SetPid(int pidP, int typeP, bool onP)
{
  debug16("%s (%d, %d, %d)", __PRETTY_FUNCTION__, pidP, typeP, onP);
  return true;
}

cString cIptvProtocolFile::GetInformation(void)
{
  debug16("%s", __PRETTY_FUNCTION__);
  return cString::sprintf("file://%s:%d", fileLocationM, fileDelayM);
}