125 lines
3.1 KiB
C
125 lines
3.1 KiB
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);
|
|
}
|