mirror of
https://github.com/rofafor/vdr-plugin-iptv.git
synced 2023-10-10 13:37:03 +02:00
283 lines
7.9 KiB
C
283 lines
7.9 KiB
C
/*
|
|
* protocolhttp.c: IPTV plugin for the Video Disk Recorder
|
|
*
|
|
* See the README file for copyright information and how to reach the author.
|
|
*
|
|
* $Id: protocolhttp.c,v 1.5 2007/09/26 19:49:35 rahrenbe Exp $
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netdb.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
|
|
#include <vdr/device.h>
|
|
|
|
#include "common.h"
|
|
#include "config.h"
|
|
#include "protocolhttp.h"
|
|
|
|
cIptvProtocolHttp::cIptvProtocolHttp()
|
|
: streamPort(3000),
|
|
socketDesc(-1),
|
|
isActive(false)
|
|
{
|
|
debug("cIptvProtocolHttp::cIptvProtocolHttp(): %d/%d packets\n",
|
|
IptvConfig.GetHttpBufferSize(), IptvConfig.GetMaxBufferSize());
|
|
streamAddr = strdup("");
|
|
streamPath = strdup("/");
|
|
// Allocate receive buffer
|
|
readBuffer = MALLOC(unsigned char, (TS_SIZE * IptvConfig.GetMaxBufferSize()));
|
|
if (!readBuffer)
|
|
error("ERROR: MALLOC() failed in ProtocolHttp()");
|
|
}
|
|
|
|
cIptvProtocolHttp::~cIptvProtocolHttp()
|
|
{
|
|
debug("cIptvProtocolHttp::~cIptvProtocolHttp()\n");
|
|
// Close the socket
|
|
Close();
|
|
// Free allocated memory
|
|
free(streamPath);
|
|
free(streamAddr);
|
|
free(readBuffer);
|
|
}
|
|
|
|
bool cIptvProtocolHttp::OpenSocket(const int Port)
|
|
{
|
|
debug("cIptvProtocolHttp::OpenSocket()\n");
|
|
// If socket is there already and it is bound to a different port, it must
|
|
// be closed first
|
|
if (Port != streamPort) {
|
|
debug("cIptvProtocolHttp::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_STREAM, 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;
|
|
}
|
|
|
|
// Create default socket
|
|
memset(&sockAddr, '\0', sizeof(sockAddr));
|
|
sockAddr.sin_family = AF_INET;
|
|
sockAddr.sin_port = htons(Port);
|
|
sockAddr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
|
|
// Update stream port
|
|
streamPort = Port;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void cIptvProtocolHttp::CloseSocket(void)
|
|
{
|
|
debug("cIptvProtocolHttp::CloseSocket()\n");
|
|
// Check if socket exists
|
|
if (socketDesc >= 0) {
|
|
close(socketDesc);
|
|
socketDesc = -1;
|
|
}
|
|
}
|
|
|
|
bool cIptvProtocolHttp::Connect(void)
|
|
{
|
|
debug("cIptvProtocolHttp::Connect()\n");
|
|
// Check that stream address is valid
|
|
if (!isActive && !isempty(streamAddr) && !isempty(streamPath)) {
|
|
// Ensure that socket is valid
|
|
OpenSocket(streamPort);
|
|
|
|
// First try only the IP address
|
|
sockAddr.sin_addr.s_addr = inet_addr(streamAddr);
|
|
|
|
if (sockAddr.sin_addr.s_addr == INADDR_NONE) {
|
|
debug("Cannot convert %s directly to internet address\n", streamAddr);
|
|
|
|
// It may be a host name, get the name
|
|
struct hostent *host;
|
|
host = gethostbyname(streamAddr);
|
|
if (!host) {
|
|
error("%s is not valid address\n", streamAddr);
|
|
char tmp[64];
|
|
error("ERROR: %s", strerror_r(h_errno, tmp, sizeof(tmp)));
|
|
return false;
|
|
}
|
|
|
|
sockAddr.sin_addr.s_addr = inet_addr(*host->h_addr_list);
|
|
}
|
|
|
|
int err = connect(socketDesc, (struct sockaddr*)&sockAddr,
|
|
sizeof(sockAddr));
|
|
// Non-blocking sockets always report in-progress error when connected
|
|
if (err < 0 && errno != EINPROGRESS) {
|
|
char tmp[64];
|
|
error("ERROR: Connect(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
|
return false;
|
|
}
|
|
|
|
// Select on the socket completion
|
|
struct timeval tv;
|
|
tv.tv_sec = 1;
|
|
tv.tv_usec = 0;
|
|
// Use select to check socket writability
|
|
fd_set wfds;
|
|
FD_ZERO(&wfds);
|
|
FD_SET(socketDesc, &wfds);
|
|
int retval = select(socketDesc + 1, NULL, &wfds, NULL, &tv);
|
|
// Check if error
|
|
if (retval < 0) {
|
|
char tmp[64];
|
|
error("ERROR: select(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
|
return -1;
|
|
}
|
|
|
|
// Select has returned. Get socket errors if there are any
|
|
int socketStatus = 0;
|
|
socklen_t len = sizeof(socketStatus);
|
|
getsockopt(socketDesc, SOL_SOCKET, SO_ERROR, &socketStatus, &len);
|
|
|
|
// If not any errors, then socket must be ready and connected
|
|
if (socketStatus != 0) {
|
|
error("Cannot connect to %s\n", streamAddr);
|
|
char tmp[64];
|
|
error("ERROR: %s", strerror_r(socketStatus, tmp, sizeof(tmp)));
|
|
return false;
|
|
}
|
|
|
|
// Formulate and send HTTP request
|
|
char buffer[256];
|
|
memset(buffer, '\0', sizeof(buffer));
|
|
snprintf(buffer, sizeof(buffer),
|
|
"GET %s HTTP/1.1\r\n"
|
|
"Host: %s\r\n"
|
|
"User-Agent: vdr-iptv\r\n"
|
|
"Range: bytes=0-\r\n"
|
|
"Connection: Close\r\n"
|
|
"\r\n", streamPath, streamAddr);
|
|
|
|
//debug("Sending http request: %s\n", buffer);
|
|
err = send(socketDesc, buffer, strlen(buffer), 0);
|
|
if (err < 0) {
|
|
char tmp[64];
|
|
error("ERROR: send(): %s", strerror_r(errno, tmp, sizeof(tmp)));
|
|
return false;
|
|
}
|
|
|
|
// It would be a good idea to wait here for the reception to begin instead
|
|
// of blindly assuming that datastream is really active.
|
|
|
|
// Also parsing the reply headers should happen here to see if the
|
|
// connection should be re-located etc.
|
|
|
|
// Update active flag
|
|
isActive = true;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool cIptvProtocolHttp::Disconnect(void)
|
|
{
|
|
debug("cIptvProtocolHttp::Disconnect()\n");
|
|
// Check that stream address is valid
|
|
if (isActive) {
|
|
// Close the socket
|
|
CloseSocket();
|
|
// Update active flag
|
|
isActive = false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
int cIptvProtocolHttp::Read(unsigned char* *BufferAddr)
|
|
{
|
|
//debug("cIptvProtocolHttp::Read()\n");
|
|
socklen_t addrlen = sizeof(sockAddr);
|
|
// Set argument point to read buffer
|
|
*BufferAddr = readBuffer;
|
|
// 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, readBuffer, (TS_SIZE * IptvConfig.GetHttpBufferSize()),
|
|
MSG_DONTWAIT, (struct sockaddr *)&sockAddr, &addrlen);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
bool cIptvProtocolHttp::Open(void)
|
|
{
|
|
debug("cIptvProtocolHttp::Open(): streamAddr=%s\n", streamAddr);
|
|
// Connect the socket
|
|
return Connect();
|
|
}
|
|
|
|
bool cIptvProtocolHttp::Close(void)
|
|
{
|
|
debug("cIptvProtocolHttp::Close(): streamAddr=%s\n", streamAddr);
|
|
// Disconnect the current stream
|
|
Disconnect();
|
|
return true;
|
|
}
|
|
|
|
bool cIptvProtocolHttp::Set(const char* Address, const int Port)
|
|
{
|
|
debug("cIptvProtocolHttp::Set(): %s:%d\n", Address, Port);
|
|
if (!isempty(Address)) {
|
|
// Disconnect the current socket
|
|
Disconnect();
|
|
// Update stream address, path and port
|
|
streamAddr = strcpyrealloc(streamAddr, Address);
|
|
char *path = strstr(streamAddr, "/");
|
|
if (path) {
|
|
streamPath = strcpyrealloc(streamPath, path);
|
|
*path = 0;
|
|
}
|
|
else
|
|
streamPath = strcpyrealloc(streamPath, "/");
|
|
streamPort = Port;
|
|
debug("iptv://%s:%d%s\n", streamAddr, streamPort, streamPath);
|
|
// Re-connect the socket
|
|
Connect();
|
|
}
|
|
return true;
|
|
}
|