120 lines
3.1 KiB
C
120 lines
3.1 KiB
C
/*
|
|
* protocoludp.c: IPTV plugin for the Video Disk Recorder
|
|
*
|
|
* See the README file for copyright information and how to reach the author.
|
|
*
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netdb.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
|
|
#include <vdr/device.h>
|
|
|
|
#include "common.h"
|
|
#include "config.h"
|
|
#include "log.h"
|
|
#include "socket.h"
|
|
#include "protocoludp.h"
|
|
|
|
cIptvProtocolUdp::cIptvProtocolUdp()
|
|
: isIGMPv3M(false),
|
|
sourceAddrM(strdup("")),
|
|
streamAddrM(strdup("")),
|
|
streamPortM(0)
|
|
{
|
|
debug1("%s", __PRETTY_FUNCTION__);
|
|
}
|
|
|
|
cIptvProtocolUdp::~cIptvProtocolUdp()
|
|
{
|
|
debug1("%s", __PRETTY_FUNCTION__);
|
|
// Drop the multicast group and close the socket
|
|
cIptvProtocolUdp::Close();
|
|
// Free allocated memory
|
|
free(streamAddrM);
|
|
free(sourceAddrM);
|
|
}
|
|
|
|
bool cIptvProtocolUdp::Open(void)
|
|
{
|
|
debug1("%s streamAddr='%s'", __PRETTY_FUNCTION__, streamAddrM);
|
|
OpenSocket(streamPortM, streamAddrM, sourceAddrM, isIGMPv3M);
|
|
if (!isempty(streamAddrM)) {
|
|
// Join a new multicast group
|
|
JoinMulticast();
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool cIptvProtocolUdp::Close(void)
|
|
{
|
|
debug1("%s streamAddr='%s'", __PRETTY_FUNCTION__, streamAddrM);
|
|
if (!isempty(streamAddrM)) {
|
|
// Drop the multicast group
|
|
OpenSocket(streamPortM, streamAddrM, sourceAddrM, isIGMPv3M);
|
|
DropMulticast();
|
|
}
|
|
// Close the socket
|
|
CloseSocket();
|
|
// Do NOT reset stream and source addresses
|
|
//sourceAddrM = strcpyrealloc(sourceAddrM, "");
|
|
//streamAddrM = strcpyrealloc(streamAddrM, "");
|
|
//streamPortM = 0;
|
|
return true;
|
|
}
|
|
|
|
int cIptvProtocolUdp::Read(unsigned char* bufferAddrP, unsigned int bufferLenP)
|
|
{
|
|
return cIptvUdpSocket::Read(bufferAddrP, bufferLenP);
|
|
}
|
|
|
|
bool cIptvProtocolUdp::SetSource(const char* locationP, const int parameterP, const int indexP)
|
|
{
|
|
debug1("%s (%s, %d, %d)", __PRETTY_FUNCTION__, locationP, parameterP, indexP);
|
|
if (!isempty(locationP)) {
|
|
// Drop the multicast group
|
|
if (!isempty(streamAddrM)) {
|
|
OpenSocket(streamPortM, streamAddrM, sourceAddrM, isIGMPv3M);
|
|
DropMulticast();
|
|
}
|
|
// Update stream address and port
|
|
streamAddrM = strcpyrealloc(streamAddrM, locationP);
|
|
// <group address> or <source address>@<group address>
|
|
char *p = strstr(streamAddrM, "@");
|
|
if (p) {
|
|
*p = 0;
|
|
sourceAddrM = strcpyrealloc(sourceAddrM, streamAddrM);
|
|
streamAddrM = strcpyrealloc(streamAddrM, p + 1);
|
|
isIGMPv3M = true;
|
|
}
|
|
else {
|
|
sourceAddrM = strcpyrealloc(sourceAddrM, streamAddrM);
|
|
isIGMPv3M = false;
|
|
}
|
|
streamPortM = parameterP;
|
|
// Join a new multicast group
|
|
if (!isempty(streamAddrM)) {
|
|
OpenSocket(streamPortM, streamAddrM, sourceAddrM, isIGMPv3M);
|
|
JoinMulticast();
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool cIptvProtocolUdp::SetPid(int pidP, int typeP, bool onP)
|
|
{
|
|
debug16("%s (%d, %d, %d)", __PRETTY_FUNCTION__, pidP, typeP, onP);
|
|
return true;
|
|
}
|
|
|
|
cString cIptvProtocolUdp::GetInformation(void)
|
|
{
|
|
debug16("%s", __PRETTY_FUNCTION__);
|
|
if (isIGMPv3M)
|
|
return cString::sprintf("udp://%s@%s:%d", sourceAddrM, streamAddrM, streamPortM);
|
|
return cString::sprintf("udp://%s:%d", streamAddrM, streamPortM);
|
|
}
|