src/dvb_utils.c

#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <assert.h>

#include "dvr_types.h"
#include "dvb_utils.h"
#include "dvr_utils.h"

#include <dmx.h>

/**
 * Set the demux's input source.
 * \param dmx_idx Demux device's index.
 * \param src The demux's input source.
 * \retval 0 On success.
 * \retval -1 On error.
 */
int dvb_set_demux_source(int dmx_idx, DVB_DemuxSource_t src)
{
    char node[32];
    struct stat st;
    int r;

    snprintf(node, sizeof(node), "/sys/class/stb/demux%d_source", dmx_idx);

    r = stat(node, &st);
    if (r == -1)
    {
        int fd, source;
        memset(node, 0, sizeof(node));
        snprintf(node, sizeof(node), "/dev/dvb0.demux%d", dmx_idx);
        fd = open(node, O_WRONLY);
        if (fd != -1)
        {
            switch (src)
            {
            case DVB_DEMUX_SOURCE_TS0:
                source = FRONTEND_TS0;
                break;
            case DVB_DEMUX_SOURCE_TS1:
                source = FRONTEND_TS1;
                break;
            case DVB_DEMUX_SOURCE_TS2:
                source = FRONTEND_TS2;
                break;
            case DVB_DEMUX_SOURCE_TS3:
                source = FRONTEND_TS3;
                break;
            case DVB_DEMUX_SOURCE_TS4:
                source = FRONTEND_TS4;
                break;
            case DVB_DEMUX_SOURCE_TS5:
                source = FRONTEND_TS5;
                break;
            case DVB_DEMUX_SOURCE_TS6:
                source = FRONTEND_TS6;
                break;
            case DVB_DEMUX_SOURCE_TS7:
                source = FRONTEND_TS7;
                break;
            case DVB_DEMUX_SOURCE_DMA0:
                source = DMA_0;
                break;
            case DVB_DEMUX_SOURCE_DMA1:
                source = DMA_1;
                break;
            case DVB_DEMUX_SOURCE_DMA2:
                source = DMA_2;
                break;
            case DVB_DEMUX_SOURCE_DMA3:
                source = DMA_3;
                break;
            case DVB_DEMUX_SOURCE_DMA4:
                source = DMA_4;
                break;
            case DVB_DEMUX_SOURCE_DMA5:
                source = DMA_5;
                break;
            case DVB_DEMUX_SOURCE_DMA6:
                source = DMA_6;
                break;
            case DVB_DEMUX_SOURCE_DMA7:
                source = DMA_7;
                break;
            default:
                assert(0);
            }
            if (ioctl(fd, DMX_SET_HW_SOURCE, &source) == -1)
            {
                DVR_DEBUG(1, "ioctl DMX_SET_HW_SOURCE:%d error:%d", source, errno);
            }
            close(fd);
        }
        else
        {
            DVR_DEBUG(1, "open \"%s\" failed, error:%d", node, errno);
        }
    }
    else
    {
        char *val;

        switch (src)
        {
        case DVB_DEMUX_SOURCE_TS0:
            val = "ts0";
            break;
        case DVB_DEMUX_SOURCE_TS1:
            val = "ts1";
            break;
        case DVB_DEMUX_SOURCE_TS2:
            val = "ts2";
            break;
        case DVB_DEMUX_SOURCE_DMA0:
            val = "hiu";
            break;
        default:
            assert(0);
        }

        r = dvr_file_echo(node, val);
    }

    return r;
}

/**
 * Get the demux's input source.
 * \param dmx_idx Demux device's index.
 * \param point src that demux's input source.
 * \retval 0 On success.
 * \retval -1 On error.
 */
int dvb_get_demux_source(int dmx_idx, DVB_DemuxSource_t *src)
{
    char node[32] = {0};
    char buf[32] = {0};
    struct stat st;
    int r, source_no;

    snprintf(node, sizeof(node), "/sys/class/stb/demux%d_source", dmx_idx);
    r = stat(node, &st);
    if (r == -1)
    {
        int fd, source;
        memset(node, 0, sizeof(node));
        snprintf(node, sizeof(node), "/dev/dvb0.demux%d", dmx_idx);
        fd = open(node, O_RDONLY);
        if (fd != -1)
        {
            if (ioctl(fd, DMX_GET_HW_SOURCE, &source) != -1)
            {
                switch (source)
                {
                case FRONTEND_TS0:
                    *src = DVB_DEMUX_SOURCE_TS0;
                    break;
                case FRONTEND_TS1:
                    *src = DVB_DEMUX_SOURCE_TS1;
                    break;
                case FRONTEND_TS2:
                    *src = DVB_DEMUX_SOURCE_TS2;
                    break;
                case FRONTEND_TS3:
                    *src = DVB_DEMUX_SOURCE_TS3;
                    break;
                case FRONTEND_TS4:
                    *src = DVB_DEMUX_SOURCE_TS4;
                    break;
                case FRONTEND_TS5:
                    *src = DVB_DEMUX_SOURCE_TS5;
                    break;
                case FRONTEND_TS6:
                    *src = DVB_DEMUX_SOURCE_TS6;
                    break;
                case FRONTEND_TS7:
                    *src = DVB_DEMUX_SOURCE_TS7;
                    break;
                case DMA_0:
                    *src = DVB_DEMUX_SOURCE_DMA0;
                    break;
                case DMA_1:
                    *src = DVB_DEMUX_SOURCE_DMA1;
                    break;
                case DMA_2:
                    *src = DVB_DEMUX_SOURCE_DMA2;
                    break;
                case DMA_3:
                    *src = DVB_DEMUX_SOURCE_DMA3;
                    break;
                case DMA_4:
                    *src = DVB_DEMUX_SOURCE_DMA4;
                    break;
                case DMA_5:
                    *src = DVB_DEMUX_SOURCE_DMA5;
                    break;
                case DMA_6:
                    *src = DVB_DEMUX_SOURCE_DMA6;
                    break;
                case DMA_7:
                    *src = DVB_DEMUX_SOURCE_DMA7;
                    break;
                default:
                    assert(0);
                }
            }
            else
            {
                DVR_DEBUG(1, "ioctl DMX_GET_HW_SOURCE:%d error:%d", source, errno);
            }
            close(fd);
        }
        else
        {
            DVR_DEBUG(1, "open \"%s\" failed, error:%d", node, errno);
        }
    }
    else
    {
        r = dvr_file_read(node, buf, sizeof(buf));
        if (r != -1)
        {
            if (strncmp(buf, "ts", 2) == 0 && strlen(buf) == 3)
            {
                sscanf(buf, "ts%d", &source_no);
                switch (source_no)
                {
                case 0:
                    *src = DVB_DEMUX_SOURCE_TS0;
                    break;
                case 1:
                    *src = DVB_DEMUX_SOURCE_TS1;
                    break;
                case 2:
                    *src = DVB_DEMUX_SOURCE_TS2;
                    break;
                default:
                    DVR_DEBUG(1, "do not support demux source:%s", buf);
                    r = -1;
                    break;
                }
            }
            else if (strncmp(buf, "hiu", 3) == 0)
            {
                *src = DVB_DEMUX_SOURCE_DMA0;
            }
            else
            {
                r = -1;
            }
            DVR_DEBUG(1, "dvb_get_demux_source \"%s\" :%s", node, buf);
        }
    }

    return r;
}