key_rotation_fragmenter.h

// Copyright 2014 Google Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#ifndef MEDIA_FORMATS_MP4_KEY_ROTATION_FRAGMENTER_H_
#define MEDIA_FORMATS_MP4_KEY_ROTATION_FRAGMENTER_H_

#include "packager/media/base/fourccs.h"
#include "packager/media/base/key_source.h"
#include "packager/media/event/muxer_listener.h"
#include "packager/media/formats/mp4/encrypting_fragmenter.h"

namespace shaka {
namespace media {
namespace mp4 {

struct MovieFragment;

class KeyRotationFragmenter : public EncryptingFragmenter {
 public:
  KeyRotationFragmenter(MovieFragment* moof,
                        std::shared_ptr<StreamInfo> info,
                        TrackFragment* traf,
                        KeySource* encryption_key_source,
                        KeySource::TrackType track_type,
                        int64_t crypto_period_duration,
                        int64_t clear_time,
                        FourCC protection_scheme,
                        uint8_t crypt_byte_block,
                        uint8_t skip_byte_block,
                        MuxerListener* muxer_listener);
  ~KeyRotationFragmenter() override;

 protected:
  Status PrepareFragmentForEncryption(bool enable_encryption) override;

 private:
  MovieFragment* moof_;

  KeySource* encryption_key_source_;
  KeySource::TrackType track_type_;
  const int64_t crypto_period_duration_;
  int64_t prev_crypto_period_index_;

  // For notifying new pssh boxes to the event handler.
  MuxerListener* const muxer_listener_;

  DISALLOW_COPY_AND_ASSIGN(KeyRotationFragmenter);
};

}  // namespace mp4
}  // namespace media
}  // namespace shaka

#endif  // MEDIA_FORMATS_MP4_KEY_ROTATION_FRAGMENTER_H_