mirror of
https://github.com/rofafor/vdr-plugin-iptv.git
synced 2023-10-10 13:37:03 +02:00
420 lines
13 KiB
C
420 lines
13 KiB
C
/*
|
|
* device.c: IPTV plugin for the Video Disk Recorder
|
|
*
|
|
* See the README file for copyright information and how to reach the author.
|
|
*
|
|
*/
|
|
|
|
#include "config.h"
|
|
#include "source.h"
|
|
#include "device.h"
|
|
|
|
#define IPTV_MAX_DEVICES MAXDEVICES
|
|
|
|
static cIptvDevice * IptvDevices[IPTV_MAX_DEVICES] = { NULL };
|
|
|
|
unsigned int cIptvDevice::deviceCount = 0;
|
|
|
|
cIptvDevice::cIptvDevice(unsigned int Index)
|
|
: deviceIndex(Index),
|
|
dvrFd(-1),
|
|
isPacketDelivered(false),
|
|
isOpenDvr(false),
|
|
sidScanEnabled(false),
|
|
pidScanEnabled(false)
|
|
{
|
|
unsigned int bufsize = (unsigned int)MEGABYTE(IptvConfig.GetTsBufferSize());
|
|
bufsize -= (bufsize % TS_SIZE);
|
|
isyslog("creating IPTV device %d (CardIndex=%d)", deviceIndex, CardIndex());
|
|
tsBuffer = new cRingBufferLinear(bufsize + 1, TS_SIZE, false,
|
|
*cString::sprintf("IPTV %d", deviceIndex));
|
|
tsBuffer->SetTimeouts(10, 10);
|
|
ResetBuffering();
|
|
pUdpProtocol = new cIptvProtocolUdp();
|
|
pHttpProtocol = new cIptvProtocolHttp();
|
|
pFileProtocol = new cIptvProtocolFile();
|
|
pExtProtocol = new cIptvProtocolExt();
|
|
pIptvStreamer = new cIptvStreamer(tsBuffer, (100 * TS_SIZE));
|
|
pPidScanner = new cPidScanner;
|
|
// Initialize filter pointers
|
|
memset(secfilters, '\0', sizeof(secfilters));
|
|
// Start section handler for iptv device
|
|
StartSectionHandler();
|
|
// Sid scanner must be created after the section handler
|
|
pSidScanner = new cSidScanner;
|
|
if (pSidScanner)
|
|
AttachFilter(pSidScanner);
|
|
// Check if dvr fifo exists
|
|
struct stat sb;
|
|
cString filename = cString::sprintf(IPTV_DVR_FILENAME, deviceIndex);
|
|
stat(filename, &sb);
|
|
if (S_ISFIFO(sb.st_mode)) {
|
|
dvrFd = open(filename, O_RDWR | O_NONBLOCK);
|
|
if (dvrFd >= 0)
|
|
dsyslog("IPTV device %d redirecting input stream to '%s'", deviceIndex, *filename);
|
|
}
|
|
}
|
|
|
|
cIptvDevice::~cIptvDevice()
|
|
{
|
|
debug("cIptvDevice::~cIptvDevice(%d)\n", deviceIndex);
|
|
// Stop section handler of iptv device
|
|
StopSectionHandler();
|
|
DELETE_POINTER(pIptvStreamer);
|
|
DELETE_POINTER(pUdpProtocol);
|
|
DELETE_POINTER(pHttpProtocol);
|
|
DELETE_POINTER(pFileProtocol);
|
|
DELETE_POINTER(pExtProtocol);
|
|
DELETE_POINTER(tsBuffer);
|
|
DELETE_POINTER(pPidScanner);
|
|
// Detach and destroy sid filter
|
|
if (pSidScanner) {
|
|
Detach(pSidScanner);
|
|
DELETE_POINTER(pSidScanner);
|
|
}
|
|
// Destroy all filters
|
|
for (int i = 0; i < eMaxSecFilterCount; ++i)
|
|
DeleteFilter(i);
|
|
// Close dvr fifo
|
|
if (dvrFd >= 0) {
|
|
int fd = dvrFd;
|
|
dvrFd = -1;
|
|
close(fd);
|
|
}
|
|
}
|
|
|
|
bool cIptvDevice::Initialize(unsigned int DeviceCount)
|
|
{
|
|
debug("cIptvDevice::Initialize(): DeviceCount=%d\n", DeviceCount);
|
|
new cIptvSourceParam(IPTV_SOURCE_CHARACTER, "IPTV");
|
|
if (DeviceCount > IPTV_MAX_DEVICES)
|
|
DeviceCount = IPTV_MAX_DEVICES;
|
|
for (unsigned int i = 0; i < DeviceCount; ++i)
|
|
IptvDevices[i] = new cIptvDevice(i);
|
|
for (unsigned int i = DeviceCount; i < IPTV_MAX_DEVICES; ++i)
|
|
IptvDevices[i] = NULL;
|
|
return true;
|
|
}
|
|
|
|
unsigned int cIptvDevice::Count(void)
|
|
{
|
|
unsigned int count = 0;
|
|
debug("cIptvDevice::Count()\n");
|
|
for (unsigned int i = 0; i < IPTV_MAX_DEVICES; ++i) {
|
|
if (IptvDevices[i] != NULL)
|
|
count++;
|
|
}
|
|
return count;
|
|
}
|
|
|
|
cIptvDevice *cIptvDevice::GetIptvDevice(int CardIndex)
|
|
{
|
|
//debug("cIptvDevice::GetIptvDevice(%d)\n", CardIndex);
|
|
for (unsigned int i = 0; i < IPTV_MAX_DEVICES; ++i) {
|
|
if ((IptvDevices[i] != NULL) && (IptvDevices[i]->CardIndex() == CardIndex)) {
|
|
//debug("cIptvDevice::GetIptvDevice(%d): FOUND!\n", CardIndex);
|
|
return IptvDevices[i];
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
cString cIptvDevice::GetGeneralInformation(void)
|
|
{
|
|
//debug("cIptvDevice::GetGeneralInformation(%d)\n", deviceIndex);
|
|
return cString::sprintf("IPTV device: %d\nCardIndex: %d\n%s%s%sChannel: %s",
|
|
deviceIndex, CardIndex(),
|
|
pIptvStreamer ? *pIptvStreamer->GetInformation() : "",
|
|
pIptvStreamer ? *pIptvStreamer->GetStreamerStatistic() : "",
|
|
*GetBufferStatistic(),
|
|
*Channels.GetByNumber(cDevice::CurrentChannel())->ToText());
|
|
}
|
|
|
|
cString cIptvDevice::GetPidsInformation(void)
|
|
{
|
|
//debug("cIptvDevice::GetPidsInformation(%d)\n", deviceIndex);
|
|
return GetPidStatistic();
|
|
}
|
|
|
|
cString cIptvDevice::GetFiltersInformation(void)
|
|
{
|
|
//debug("cIptvDevice::GetFiltersInformation(%d)\n", deviceIndex);
|
|
unsigned int count = 0;
|
|
cString info("Active section filters:\n");
|
|
// loop through active section filters
|
|
for (unsigned int i = 0; i < eMaxSecFilterCount; ++i) {
|
|
if (secfilters[i]) {
|
|
info = cString::sprintf("%sFilter %d: %s Pid=0x%02X (%s)\n", *info, i,
|
|
*secfilters[i]->GetSectionStatistic(), secfilters[i]->GetPid(),
|
|
id_pid(secfilters[i]->GetPid()));
|
|
if (++count > IPTV_STATS_ACTIVE_FILTERS_COUNT)
|
|
break;
|
|
}
|
|
}
|
|
return info;
|
|
}
|
|
|
|
cString cIptvDevice::GetInformation(unsigned int Page)
|
|
{
|
|
// generate information string
|
|
cString info;
|
|
switch (Page) {
|
|
case IPTV_DEVICE_INFO_GENERAL:
|
|
info = GetGeneralInformation();
|
|
break;
|
|
case IPTV_DEVICE_INFO_PIDS:
|
|
info = GetPidsInformation();
|
|
break;
|
|
case IPTV_DEVICE_INFO_FILTERS:
|
|
info = GetFiltersInformation();
|
|
break;
|
|
default:
|
|
info = cString::sprintf("%s%s%s",
|
|
*GetGeneralInformation(),
|
|
*GetPidsInformation(),
|
|
*GetFiltersInformation());
|
|
break;
|
|
}
|
|
return info;
|
|
}
|
|
|
|
bool cIptvDevice::ProvidesSource(int Source) const
|
|
{
|
|
debug("cIptvDevice::ProvidesSource(%d)\n", deviceIndex);
|
|
return ((Source & cSource::st_Mask) == (IPTV_SOURCE_CHARACTER << 24));
|
|
}
|
|
|
|
bool cIptvDevice::ProvidesTransponder(const cChannel *Channel) const
|
|
{
|
|
debug("cIptvDevice::ProvidesTransponder(%d)\n", deviceIndex);
|
|
return (ProvidesSource(Channel->Source()));
|
|
}
|
|
|
|
bool cIptvDevice::ProvidesChannel(const cChannel *Channel, int Priority, bool *NeedsDetachReceivers) const
|
|
{
|
|
bool result = false;
|
|
bool needsDetachReceivers = false;
|
|
|
|
debug("cIptvDevice::ProvidesChannel(%d)\n", deviceIndex);
|
|
if (ProvidesTransponder(Channel))
|
|
result = true;
|
|
if (NeedsDetachReceivers)
|
|
*NeedsDetachReceivers = needsDetachReceivers;
|
|
return result;
|
|
}
|
|
|
|
int cIptvDevice::NumProvidedSystems(void) const
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
bool cIptvDevice::SetChannelDevice(const cChannel *Channel, bool LiveView)
|
|
{
|
|
cIptvProtocolIf *protocol;
|
|
cIptvTransponderParameters itp(Channel->Parameters());
|
|
|
|
debug("cIptvDevice::SetChannelDevice(%d)\n", deviceIndex);
|
|
|
|
if (isempty(itp.Address())) {
|
|
error("Unrecognized IPTV address: %s", Channel->Parameters());
|
|
return false;
|
|
}
|
|
switch (itp.Protocol()) {
|
|
case cIptvTransponderParameters::eProtocolUDP:
|
|
protocol = pUdpProtocol;
|
|
break;
|
|
case cIptvTransponderParameters::eProtocolHTTP:
|
|
protocol = pHttpProtocol;
|
|
break;
|
|
case cIptvTransponderParameters::eProtocolFILE:
|
|
protocol = pFileProtocol;
|
|
break;
|
|
case cIptvTransponderParameters::eProtocolEXT:
|
|
protocol = pExtProtocol;
|
|
break;
|
|
default:
|
|
error("Unrecognized IPTV protocol: %s", Channel->Parameters());
|
|
return false;
|
|
break;
|
|
}
|
|
sidScanEnabled = itp.SidScan() ? true : false;
|
|
pidScanEnabled = itp.PidScan() ? true : false;
|
|
if (pIptvStreamer->Set(itp.Address(), itp.Parameter(), deviceIndex, protocol)) {
|
|
if (sidScanEnabled && pSidScanner && IptvConfig.GetSectionFiltering())
|
|
pSidScanner->SetChannel(Channel);
|
|
if (pidScanEnabled && pPidScanner)
|
|
pPidScanner->SetChannel(Channel);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool cIptvDevice::SetPid(cPidHandle *Handle, int Type, bool On)
|
|
{
|
|
debug("cIptvDevice::SetPid(%d) Pid=%d Type=%d On=%d\n", deviceIndex, Handle->pid, Type, On);
|
|
return true;
|
|
}
|
|
|
|
bool cIptvDevice::DeleteFilter(unsigned int Index)
|
|
{
|
|
if ((Index < eMaxSecFilterCount) && secfilters[Index]) {
|
|
//debug("cIptvDevice::DeleteFilter(%d) Index=%d\n", deviceIndex, Index);
|
|
cIptvSectionFilter *tmp = secfilters[Index];
|
|
secfilters[Index] = NULL;
|
|
delete tmp;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool cIptvDevice::IsBlackListed(u_short Pid, u_char Tid, u_char Mask) const
|
|
{
|
|
//debug("cIptvDevice::IsBlackListed(%d) Pid=%d Tid=%02X Mask=%02X\n", deviceIndex, Pid, Tid, Mask);
|
|
// loop through section filter table
|
|
for (int i = 0; i < SECTION_FILTER_TABLE_SIZE; ++i) {
|
|
int index = IptvConfig.GetDisabledFilters(i);
|
|
// check if matches
|
|
if ((index >= 0) && (index < SECTION_FILTER_TABLE_SIZE) &&
|
|
(section_filter_table[index].pid == Pid) && (section_filter_table[index].tid == Tid) &&
|
|
(section_filter_table[index].mask == Mask)) {
|
|
//debug("cIptvDevice::IsBlackListed(%d) Found=%s\n", deviceIndex, section_filter_table[index].description);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
int cIptvDevice::OpenFilter(u_short Pid, u_char Tid, u_char Mask)
|
|
{
|
|
// Check if disabled by user
|
|
if (!IptvConfig.GetSectionFiltering())
|
|
return -1;
|
|
// Blacklist check, refuse certain filters
|
|
if (IsBlackListed(Pid, Tid, Mask))
|
|
return -1;
|
|
// Lock
|
|
cMutexLock MutexLock(&mutex);
|
|
// Search the next free filter slot
|
|
for (unsigned int i = 0; i < eMaxSecFilterCount; ++i) {
|
|
if (!secfilters[i]) {
|
|
//debug("cIptvDevice::OpenFilter(%d): Pid=%d Tid=%02X Mask=%02X Index=%d\n", deviceIndex, Pid, Tid, Mask, i);
|
|
secfilters[i] = new cIptvSectionFilter(deviceIndex, i, Pid, Tid, Mask);
|
|
return secfilters[i]->GetReadDesc();
|
|
}
|
|
}
|
|
// No free filter slot found
|
|
return -1;
|
|
}
|
|
|
|
void cIptvDevice::CloseFilter(int Handle)
|
|
{
|
|
// Lock
|
|
cMutexLock MutexLock(&mutex);
|
|
// Search the filter for deletion
|
|
for (unsigned int i = 0; i < eMaxSecFilterCount; ++i) {
|
|
if (secfilters[i] && (Handle == secfilters[i]->GetReadDesc())) {
|
|
//debug("cIptvDevice::CloseFilter(%d): %d\n", deviceIndex, Handle);
|
|
DeleteFilter(i);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool cIptvDevice::OpenDvr(void)
|
|
{
|
|
debug("cIptvDevice::OpenDvr(%d)\n", deviceIndex);
|
|
isPacketDelivered = false;
|
|
tsBuffer->Clear();
|
|
ResetBuffering();
|
|
if (pIptvStreamer)
|
|
pIptvStreamer->Open();
|
|
if (sidScanEnabled && pSidScanner && IptvConfig.GetSectionFiltering())
|
|
pSidScanner->Open();
|
|
isOpenDvr = true;
|
|
return true;
|
|
}
|
|
|
|
void cIptvDevice::CloseDvr(void)
|
|
{
|
|
debug("cIptvDevice::CloseDvr(%d)\n", deviceIndex);
|
|
if (sidScanEnabled && pSidScanner && IptvConfig.GetSectionFiltering())
|
|
pSidScanner->Close();
|
|
if (pIptvStreamer)
|
|
pIptvStreamer->Close();
|
|
isOpenDvr = false;
|
|
}
|
|
|
|
bool cIptvDevice::HasLock(int TimeoutMs)
|
|
{
|
|
//debug("cIptvDevice::HasLock(%d): %d\n", deviceIndex, TimeoutMs);
|
|
return (!IsBuffering());
|
|
}
|
|
|
|
void cIptvDevice::ResetBuffering(void)
|
|
{
|
|
debug("cIptvDevice::ResetBuffering(%d)\n", deviceIndex);
|
|
// pad prefill to multiple of TS_SIZE
|
|
tsBufferPrefill = (unsigned int)MEGABYTE(IptvConfig.GetTsBufferSize()) *
|
|
IptvConfig.GetTsBufferPrefillRatio() / 100;
|
|
tsBufferPrefill -= (tsBufferPrefill % TS_SIZE);
|
|
}
|
|
|
|
bool cIptvDevice::IsBuffering(void)
|
|
{
|
|
//debug("cIptvDevice::IsBuffering(%d): %d\n", deviceIndex);
|
|
if (tsBufferPrefill && tsBuffer->Available() < tsBufferPrefill)
|
|
return true;
|
|
else
|
|
tsBufferPrefill = 0;
|
|
return false;
|
|
}
|
|
|
|
bool cIptvDevice::GetTSPacket(uchar *&Data)
|
|
{
|
|
int Count = 0;
|
|
//debug("cIptvDevice::GetTSPacket(%d)\n", deviceIndex);
|
|
if (tsBuffer && !IsBuffering()) {
|
|
if (isPacketDelivered) {
|
|
tsBuffer->Del(TS_SIZE);
|
|
isPacketDelivered = false;
|
|
// Update buffer statistics
|
|
AddBufferStatistic(TS_SIZE, tsBuffer->Available());
|
|
}
|
|
uchar *p = tsBuffer->Get(Count);
|
|
if (p && Count >= TS_SIZE) {
|
|
if (*p != TS_SYNC_BYTE) {
|
|
for (int i = 1; i < Count; i++) {
|
|
if (p[i] == TS_SYNC_BYTE) {
|
|
Count = i;
|
|
break;
|
|
}
|
|
}
|
|
tsBuffer->Del(Count);
|
|
error("Skipped %d bytes to sync on TS packet\n", Count);
|
|
return false;
|
|
}
|
|
isPacketDelivered = true;
|
|
Data = p;
|
|
// Update pid statistics
|
|
AddPidStatistic(ts_pid(p), payload(p));
|
|
// Send data also to dvr fifo
|
|
if (dvrFd >= 0)
|
|
Count = (int)write(dvrFd, p, TS_SIZE);
|
|
// Analyze incomplete streams with built-in pid analyzer
|
|
if (pidScanEnabled && pPidScanner)
|
|
pPidScanner->Process(p);
|
|
// Lock
|
|
cMutexLock MutexLock(&mutex);
|
|
// Run the data through all filters
|
|
for (unsigned int i = 0; i < eMaxSecFilterCount; ++i) {
|
|
if (secfilters[i])
|
|
secfilters[i]->Process(p);
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
// Reduce cpu load by preventing busylooping
|
|
cCondWait::SleepMs(10);
|
|
Data = NULL;
|
|
return true;
|
|
}
|