modules/demux/smooth/playlist/MemoryChunk.cpp

   1 /*
   2  * MemoryChunk.cpp
   3  *****************************************************************************
   4  * Copyright (C) 2015 - VideoLAN Authors
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU Lesser General Public License as published
   8  * by the Free Software Foundation; either version 2.1 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14  * GNU Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  19  *****************************************************************************/
  20 #ifdef HAVE_CONFIG_H
  21 # include "config.h"
  22 #endif
  23
  24 #include "MemoryChunk.hpp"
  25
  26 #include <vlc_block.h>
  27
  28 using namespace smooth::http;
  29
  30 MemoryChunkSource::MemoryChunkSource(block_t *block)
  31 {
  32     data = block;
  33     i_read = 0;
  34     contentLength = data->i_buffer;
  35 }
  36
  37 MemoryChunkSource::~MemoryChunkSource()
  38 {
  39     if(data)
  40         block_Release(data);
  41 }
  42
  43 bool MemoryChunkSource::hasMoreData() const
  44 {
  45     return i_read > contentLength;
  46 }
  47
  48 block_t * MemoryChunkSource::readBlock()
  49 {
  50     block_t *p_block = nullptr;
  51     if(data)
  52     {
  53         p_block = data;
  54         data = nullptr;
  55     }
  56     return p_block;
  57 }
  58
  59 block_t * MemoryChunkSource::read(size_t toread)
  60 {
  61     if(!data)
  62         return nullptr;
  63
  64     block_t * p_block = nullptr;
  65
  66     toread = __MIN(data->i_buffer - i_read, toread);
  67     if(toread > 0)
  68     {
  69         if((p_block = block_Alloc(toread)))
  70         {
  71             memcpy(p_block->p_buffer, &data->p_buffer[i_read], toread);
  72             p_block->i_buffer = toread;
  73             i_read += toread;
  74         }
  75     }
  76
  77     return p_block;
  78 }