mirror of
https://github.com/rofafor/vdr-plugin-iptv.git
synced 2023-10-10 13:37:03 +02:00
207 lines
5.6 KiB
C
207 lines
5.6 KiB
C
|
/*
|
||
|
* streamer.c: IPTV plugin for the Video Disk Recorder
|
||
|
*
|
||
|
* See the README file for copyright information and how to reach the author.
|
||
|
*
|
||
|
* $Id: protocoludp.c,v 1.1 2007/09/14 15:44:25 rahrenbe Exp $
|
||
|
*/
|
||
|
|
||
|
#include <sys/types.h>
|
||
|
#include <sys/socket.h>
|
||
|
#include <netdb.h>
|
||
|
#include <fcntl.h>
|
||
|
#include <unistd.h>
|
||
|
|
||
|
#include "common.h"
|
||
|
#include "protocoludp.h"
|
||
|
|
||
|
cIptvProtocolUdp::cIptvProtocolUdp()
|
||
|
: streamPort(1234),
|
||
|
socketDesc(-1),
|
||
|
mcastActive(false)
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::cIptvProtocolUdp()\n");
|
||
|
streamAddr = strdup("");
|
||
|
}
|
||
|
|
||
|
cIptvProtocolUdp::~cIptvProtocolUdp()
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::~cIptvProtocolUdp()\n");
|
||
|
// Drop the multicast group and close the socket
|
||
|
Close();
|
||
|
// Free allocated memory
|
||
|
free(streamAddr);
|
||
|
}
|
||
|
|
||
|
bool cIptvProtocolUdp::OpenSocket(const int Port)
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::OpenSocket()\n");
|
||
|
// If socket is there already and it is bound to a different port, it must
|
||
|
// be closed first
|
||
|
if (Port != streamPort) {
|
||
|
debug("cIptvProtocolUdp::OpenSocket(): Socket tear-down\n");
|
||
|
CloseSocket();
|
||
|
}
|
||
|
// Bind to the socket if it is not active already
|
||
|
if (socketDesc < 0) {
|
||
|
int yes = 1;
|
||
|
// Create socket
|
||
|
socketDesc = socket(PF_INET, SOCK_DGRAM, 0);
|
||
|
if (socketDesc < 0) {
|
||
|
char tmp[64];
|
||
|
error("ERROR: socket(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
||
|
return false;
|
||
|
}
|
||
|
// Make it use non-blocking I/O to avoid stuck read calls
|
||
|
if (fcntl(socketDesc, F_SETFL, O_NONBLOCK)) {
|
||
|
char tmp[64];
|
||
|
error("ERROR: fcntl(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
||
|
CloseSocket();
|
||
|
return false;
|
||
|
}
|
||
|
// Allow multiple sockets to use the same PORT number
|
||
|
if (setsockopt(socketDesc, SOL_SOCKET, SO_REUSEADDR, &yes,
|
||
|
sizeof(yes)) < 0) {
|
||
|
char tmp[64];
|
||
|
error("ERROR: setsockopt(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
||
|
CloseSocket();
|
||
|
return false;
|
||
|
}
|
||
|
// Bind socket
|
||
|
memset(&sockAddr, '\0', sizeof(sockAddr));
|
||
|
sockAddr.sin_family = AF_INET;
|
||
|
sockAddr.sin_port = htons(Port);
|
||
|
sockAddr.sin_addr.s_addr = htonl(INADDR_ANY);
|
||
|
int err = bind(socketDesc, (struct sockaddr *)&sockAddr, sizeof(sockAddr));
|
||
|
if (err < 0) {
|
||
|
char tmp[64];
|
||
|
error("ERROR: bind(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
||
|
CloseSocket();
|
||
|
return false;
|
||
|
}
|
||
|
// Update stream port
|
||
|
streamPort = Port;
|
||
|
}
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
void cIptvProtocolUdp::CloseSocket(void)
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::CloseSocket()\n");
|
||
|
// Check if socket exists
|
||
|
if (socketDesc >= 0) {
|
||
|
close(socketDesc);
|
||
|
socketDesc = -1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
bool cIptvProtocolUdp::JoinMulticast(void)
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::JoinMulticast()\n");
|
||
|
// Check that stream address is valid
|
||
|
if (!mcastActive && !isempty(streamAddr)) {
|
||
|
// Ensure that socket is valid
|
||
|
OpenSocket(streamPort);
|
||
|
// Join a new multicast group
|
||
|
struct ip_mreq mreq;
|
||
|
mreq.imr_multiaddr.s_addr = inet_addr(streamAddr);
|
||
|
mreq.imr_interface.s_addr = htonl(INADDR_ANY);
|
||
|
int err = setsockopt(socketDesc, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq,
|
||
|
sizeof(mreq));
|
||
|
if (err < 0) {
|
||
|
char tmp[64];
|
||
|
error("ERROR: setsockopt(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
||
|
return false;
|
||
|
}
|
||
|
// Update multicasting flag
|
||
|
mcastActive = true;
|
||
|
}
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
bool cIptvProtocolUdp::DropMulticast(void)
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::DropMulticast()\n");
|
||
|
// Check that stream address is valid
|
||
|
if (mcastActive && !isempty(streamAddr)) {
|
||
|
// Ensure that socket is valid
|
||
|
OpenSocket(streamPort);
|
||
|
// Drop the multicast group
|
||
|
struct ip_mreq mreq;
|
||
|
mreq.imr_multiaddr.s_addr = inet_addr(streamAddr);
|
||
|
mreq.imr_interface.s_addr = htonl(INADDR_ANY);
|
||
|
int err = setsockopt(socketDesc, IPPROTO_IP, IP_DROP_MEMBERSHIP, &mreq,
|
||
|
sizeof(mreq));
|
||
|
if (err < 0) {
|
||
|
char tmp[64];
|
||
|
error("ERROR: setsockopt(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
||
|
return false;
|
||
|
}
|
||
|
// Update multicasting flag
|
||
|
mcastActive = false;
|
||
|
}
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
int cIptvProtocolUdp::Read(unsigned char *Buffer, int Len)
|
||
|
{
|
||
|
//debug("cIptvProtocolUdp::Read()\n");
|
||
|
socklen_t addrlen = sizeof(sockAddr);
|
||
|
// Wait for data
|
||
|
struct timeval tv;
|
||
|
tv.tv_sec = 0;
|
||
|
tv.tv_usec = 500000;
|
||
|
// Use select
|
||
|
fd_set rfds;
|
||
|
FD_ZERO(&rfds);
|
||
|
FD_SET(socketDesc, &rfds);
|
||
|
int retval = select(socketDesc + 1, &rfds, NULL, NULL, &tv);
|
||
|
// Check if error
|
||
|
if (retval < 0) {
|
||
|
char tmp[64];
|
||
|
error("ERROR: select(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
||
|
return -1;
|
||
|
}
|
||
|
// Check if data available
|
||
|
else if (retval) {
|
||
|
// Read data from socket
|
||
|
return recvfrom(socketDesc, Buffer, Len, MSG_DONTWAIT,
|
||
|
(struct sockaddr *)&sockAddr, &addrlen);
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
bool cIptvProtocolUdp::Open(void)
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::Open(): streamAddr = %s\n", streamAddr);
|
||
|
// Join a new multicast group
|
||
|
JoinMulticast();
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
bool cIptvProtocolUdp::Close(void)
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::Close(): streamAddr = %s\n", streamAddr);
|
||
|
// Drop the multicast group
|
||
|
DropMulticast();
|
||
|
// Close the socket
|
||
|
CloseSocket();
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
bool cIptvProtocolUdp::Set(const char* Address, const int Port)
|
||
|
{
|
||
|
debug("cIptvProtocolUdp::Set(): %s:%d\n", Address, Port);
|
||
|
if (!isempty(Address)) {
|
||
|
// Drop the multicast group
|
||
|
DropMulticast();
|
||
|
// Update stream address and port
|
||
|
streamAddr = strcpyrealloc(streamAddr, Address);
|
||
|
streamPort = Port;
|
||
|
// Join a new multicast group
|
||
|
JoinMulticast();
|
||
|
}
|
||
|
return true;
|
||
|
}
|
||
|
|