protocoludp.cpp

/*
 * protocoludp.c: IPTV plugin for the Video Disk Recorder
 *
 * See the README file for copyright information and how to reach the author.
 *
 */

#include <vdr/device.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() {
    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() {
    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(SourceParameter parameter) {
    debug1("%s (%s, %d, %d)", __PRETTY_FUNCTION__, parameter.locationP, parameter.parameterP, parameter.indexP);

    if (!isempty(parameter.locationP)) {
        // Drop the multicast group
        if (!isempty(streamAddrM)) {
            OpenSocket(streamPortM, streamAddrM, sourceAddrM, isIGMPv3M);
            DropMulticast();
        }

        // Update stream address and port
        streamAddrM = strcpyrealloc(streamAddrM, parameter.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 = parameter.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() {
    debug16("%s", __PRETTY_FUNCTION__);

    if (isIGMPv3M) {
        return cString::sprintf("udp://%s@%s:%d", sourceAddrM, streamAddrM, streamPortM);
    }

    return cString::sprintf("udp://%s:%d", streamAddrM, streamPortM);
}