HLSTree.h

/*
*      Copyright (C) 2017 peak3d
*      http://www.peak3d.de
*
*  This Program is free software; you can redistribute it and/or modify
*  it under the terms of the GNU General Public License as published by
*  the Free Software Foundation; either version 2, or (at your option)
*  any later version.
*
*  This Program is distributed in the hope that it will be useful,
*  but WITHOUT ANY WARRANTY; without even the implied warranty of
*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
*  GNU General Public License for more details.
*
*  <http://www.gnu.org/licenses/>.
*
*/

#pragma once

#include "../common/AdaptiveTree.h"
#include <map>

class AESDecrypter;

namespace adaptive
{

  class HLSTree : public AdaptiveTree
  {
  public:
    enum
    {
      ENCRYPTIONTYPE_INVALID = 0,
      ENCRYPTIONTYPE_CLEAR = 1,
      ENCRYPTIONTYPE_AES128 = 2,
      ENCRYPTIONTYPE_WIDEVINE = 3,
      ENCRYPTIONTYPE_UNKNOWN = 4,
    };
    HLSTree(AESDecrypter *decrypter) : AdaptiveTree(), m_decrypter(decrypter) {};
    virtual ~HLSTree();

    virtual bool open(const std::string &url, const std::string &manifestUpdateParam) override;
    virtual PREPARE_RESULT prepareRepresentation(Period* period,
                                                 AdaptationSet* adp,
                                                 Representation* rep,
                                                 bool update = false) override;
    virtual bool write_data(void *buffer, size_t buffer_size, void *opaque) override;
    virtual void OnDataArrived(unsigned int segNum, uint16_t psshSet, uint8_t iv[16], const uint8_t *src, uint8_t *dst, size_t dstOffset, size_t dataSize) override;
    virtual void RefreshSegments(Period* period,
                                 AdaptationSet* adp,
                                 Representation* rep,
                                 StreamType type) override;
    virtual bool processManifest(std::stringstream& stream, const std::string &url);

  protected:
    virtual void RefreshLiveSegments() override;

  private:
    int processEncryption(std::string baseUrl, std::map<std::string, std::string>& map);
    std::string m_audioCodec;

    struct EXTGROUP
    {
      std::string m_codec;
      std::vector<AdaptationSet*> m_sets;

      void setCodec(const std::string &codec)
      {
        if (m_codec.empty())
        {
          m_codec = codec;
          for (auto &set : m_sets)
            set->representations_[0]->codecs_ = codec;
        }
      }
    };

    std::map<std::string, EXTGROUP> m_extGroups;
    bool m_refreshPlayList = true;
    uint8_t m_segmentIntervalSec = 4;
    AESDecrypter *m_decrypter;
    std::stringstream manifest_stream;
    bool m_hasDiscontSeq = false;
    uint32_t m_discontSeq = 0;
  };

} // namespace