dsmccobjcarousel.cpp

Go to the documentation of this file.

/*
 *  Copyright (C) David C.J. Matthews 2005, 2006
 *     Derived from dsmcc by Richard Palmer
 */
#include <cstdio>
#include <cstdlib>
#include <zlib.h>
#undef Z_NULL
#define Z_NULL nullptr
 
#include "libmythbase/mythlogging.h"
 
#include "dsmccobjcarousel.h"
#include "dsmccreceiver.h"
#include "dsmcccache.h"
#include "dsmcc.h"
 
// Construct the data for a new module from a DII message.
DSMCCCacheModuleData::DSMCCCacheModuleData(DsmccDii *dii,
                                           DsmccModuleInfo *info,
                                           unsigned short streamTag)
    : m_carouselId(dii->m_downloadId), m_moduleId(info->m_moduleId),
      m_streamId(streamTag),           m_version(info->m_moduleVersion),
      m_moduleSize(info->m_moduleSize),
      // Copy the descriptor information.
      m_descriptorData(info->m_modInfo.m_descriptorData)
{
    // The number of blocks needed to hold this module.
    int num_blocks = (m_moduleSize + dii->m_blockSize - 1) / dii->m_blockSize;
    m_blocks.resize(num_blocks, nullptr); // Set it to all zeros
}
 
DSMCCCacheModuleData::~DSMCCCacheModuleData()
{
    for (auto & block : m_blocks)
        delete block;
    m_blocks.clear();
}
 
std::vector<uint8_t> DSMCCCacheModuleData::AddModuleData(DsmccDb *ddb,
                                                   const unsigned char *data)
{
    if (m_version != ddb->m_moduleVersion)
    {
        LOG(VB_DSMCC, LOG_WARNING, QString("[dsmcc] Module %1 my version %2 != %3")
            .arg(ddb->m_moduleId).arg(m_version).arg(ddb->m_moduleVersion));
        return {}; // Wrong version
    }
 
    if (m_completed)
        return {}; // Already got it.
 
    if (ddb->m_blockNumber >= m_blocks.size())
    {
        LOG(VB_DSMCC, LOG_WARNING, QString("[dsmcc] Module %1 block number %2 "
                                           "is larger than %3")
            .arg(ddb->m_moduleId).arg(ddb->m_blockNumber)
            .arg(m_blocks.size()));
        return {};
    }
 
    // Check if we have this block already or not. If not append to list
    if (m_blocks[ddb->m_blockNumber])
    {
        QString s;
        for (auto & block : m_blocks)
            s += block ? '+' : 'X';
 
        LOG(VB_DSMCC, LOG_INFO, QString("[dsmcc] Module %1 block %2 dup: %3")
            .arg(ddb->m_moduleId).arg(ddb->m_blockNumber +1).arg(s));
 
        return {}; // We have seen this block before.
    }
 
    // Add this to our set of blocks.
    m_blocks[ddb->m_blockNumber] = new QByteArray((char*) data, ddb->m_len);
    if (m_blocks[ddb->m_blockNumber])
        m_receivedData += ddb->m_len;
 
    LOG(VB_DSMCC, LOG_INFO, QString("[dsmcc] Module %1 block %2/%3 bytes %4/%5")
        .arg(ddb->m_moduleId)
        .arg(ddb->m_blockNumber +1).arg(m_blocks.size())
        .arg(m_receivedData).arg(m_moduleSize));
 
    if (m_receivedData < m_moduleSize)
        return {}; // Not yet complete
 
    LOG(VB_DSMCC, LOG_INFO,
        QString("[dsmcc] Reconstructing module %1 from blocks")
            .arg(m_moduleId));
 
    // Re-assemble the blocks into the complete module.
    std::vector<uint8_t> tmp_data;
    try {
        tmp_data.reserve(m_receivedData);
    } catch (const std::bad_alloc& e) {
        return {};
    }
    for (auto & block : m_blocks)
    {
        if (block == nullptr)
        {
            LOG(VB_DSMCC, LOG_INFO,
                QString("[dsmcc] Null data found, aborting reconstruction"));
            return {};
        }
        uint size = block->size();
        tmp_data.insert(tmp_data.end(), block->data(), block->data() + size);
        delete block;
        block = nullptr;
    }
 
    /* Uncompress....  */
    if (m_descriptorData.m_isCompressed)
    {
        unsigned long dataLen = m_descriptorData.m_originalSize;
        LOG(VB_DSMCC, LOG_INFO, QString("[dsmcc] uncompressing: "
                                        "compressed size %1, final size %2")
            .arg(m_moduleSize).arg(dataLen));
 
        std::vector<uint8_t> uncompressed;
        uncompressed.resize(dataLen);
        int ret = uncompress(uncompressed.data(), &dataLen, tmp_data.data(), m_moduleSize);
        if (ret != Z_OK)
        {
            LOG(VB_DSMCC, LOG_ERR, "[dsmcc] compression error, skipping");
            return {};
        }
 
        tmp_data = uncompressed;
    }
 
    m_completed = true;
    m_blocks.clear(); // No longer required: free the space.
    return tmp_data;
}
 
 
ObjCarousel::~ObjCarousel()
{
    for (const auto & cache : m_Cache)
        delete cache;
    m_Cache.clear();
}
 
void ObjCarousel::AddModuleInfo(DsmccDii *dii, Dsmcc *status,
                                unsigned short streamTag)
{
    for (int i = 0; i < dii->m_numberModules; i++)
    {
        DsmccModuleInfo *info = &(dii->m_modules[i]);
        bool bFound = false;
        // Do we already know this module?
        // If so and it is the same version we don't need to do anything.
        // If the version has changed we have to replace it.
        for (auto it = m_Cache.begin(); it != m_Cache.end(); ++it)
        {
            DSMCCCacheModuleData *cachep = *it;
            if (cachep->CarouselId() == dii->m_downloadId &&
                    cachep->ModuleId() == info->m_moduleId)
            {
                /* already known */
                if (cachep->Version() == info->m_moduleVersion)
                {
                    LOG(VB_DSMCC, LOG_DEBUG, QString("[dsmcc] Already Know Module %1")
                            .arg(info->m_moduleId));
 
                    if (cachep->ModuleSize() == info->m_moduleSize)
                    {
                        bFound = true;
                        break;
                    }
                    // It seems that when ITV4 starts broadcasting it
                    // updates the contents of a file but doesn't
                    // update the version.  This is a work-around.
                    LOG(VB_DSMCC, LOG_ERR,
                        QString("[dsmcc] Module %1 size has changed (%2 to %3) "
                                "but version has not!!")
                             .arg(info->m_moduleId)
                             .arg(info->m_moduleSize)
                             .arg(cachep->DataSize()));
                }
                // Version has changed - Drop old data.
                LOG(VB_DSMCC, LOG_INFO, QString("[dsmcc] Updated Module %1")
                        .arg(info->m_moduleId));
 
                // Remove and delete the cache object.
                m_Cache.erase(it);
                delete cachep;
                break;
            }
        }
 
        if (bFound)
            continue;
 
        LOG(VB_DSMCC, LOG_INFO, QString("[dsmcc] Saving info for module %1")
                .arg(dii->m_modules[i].m_moduleId));
 
        // Create a new cache module data object.
        auto *cachep = new DSMCCCacheModuleData(dii, info, streamTag);
        int tag = info->m_modInfo.m_tap.m_assocTag;
        LOG(VB_DSMCC, LOG_DEBUG, QString("[dsmcc] Module info tap identifies "
                                         "tag %1 with carousel %2")
            .arg(tag).arg(cachep->CarouselId()));
 
        // If a carousel with this id does not exist create it.
        status->AddTap(tag, cachep->CarouselId());
 
        // Add this module to the cache.
        m_Cache.push_back(cachep);
    }
}
 
void ObjCarousel::AddModuleData(DsmccDb *ddb, const unsigned char *data)
{
    LOG(VB_DSMCC, LOG_DEBUG, QString("[dsmcc] Data block on carousel %1").arg(m_id));
 
    // Search the saved module info for this module
    for (auto *cachep : m_Cache)
    {
        if (cachep->CarouselId() == m_id &&
            (cachep->ModuleId() == ddb->m_moduleId))
        {
            // Add the block to the module
            std::vector<uint8_t> tmp_data = cachep->AddModuleData(ddb, data);
            if (!tmp_data.empty())
            {
                // It is complete and we have the data
                unsigned int len   = cachep->DataSize();
                unsigned long curp = 0;
                LOG(VB_DSMCC, LOG_DEBUG, QString("[biop] Module size (uncompressed) = %1").arg(len));
 
                // Now process the BIOP tables in this module.
                // Tables may be file contents or the descriptions of
                // directories or service gateways (root directories).
                while (curp < len)
                {
                    BiopMessage bm;
                    if (!bm.Process(cachep, &m_fileCache, tmp_data.data(), &curp))
                        break;
                }
            }
            return;
        }
    }
    LOG(VB_DSMCC, LOG_INFO, QString("[dsmcc] Data block module %1 not on carousel %2")
        .arg(ddb->m_moduleId).arg(m_id));
}