encrypting_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_ENCRYPTING_FRAGMENTER_H_
#define MEDIA_FORMATS_MP4_ENCRYPTING_FRAGMENTER_H_

#include "packager/base/memory/scoped_ptr.h"
#include "packager/media/filters/vpx_parser.h"
#include "packager/media/formats/mp4/fragmenter.h"

namespace edash_packager {
namespace media {

class AesCtrEncryptor;
struct EncryptionKey;

namespace mp4 {

class EncryptingFragmenter : public Fragmenter {
 public:
  EncryptingFragmenter(TrackFragment* traf,
                       scoped_ptr<EncryptionKey> encryption_key,
                       int64_t clear_time,
                       VideoCodec video_codec,
                       uint8_t nalu_length_size);

  ~EncryptingFragmenter() override;

  Status AddSample(scoped_refptr<MediaSample> sample) override;
  Status InitializeFragment(int64_t first_sample_dts) override;
  void FinalizeFragment() override;

 protected:
  virtual Status PrepareFragmentForEncryption(bool enable_encryption);
  virtual void FinalizeFragmentForEncryption();

  Status CreateEncryptor();

  EncryptionKey* encryption_key() { return encryption_key_.get(); }
  AesCtrEncryptor* encryptor() { return encryptor_.get(); }

  void set_encryption_key(scoped_ptr<EncryptionKey> encryption_key) {
    encryption_key_ = encryption_key.Pass();
  }

 private:
  void EncryptBytes(uint8_t* data, uint32_t size);
  Status EncryptSample(scoped_refptr<MediaSample> sample);

  // Should we enable subsample encryption?
  bool IsSubsampleEncryptionRequired() {
    return vpx_parser_ || nalu_length_size_ != 0;
  }

  scoped_ptr<EncryptionKey> encryption_key_;
  scoped_ptr<AesCtrEncryptor> encryptor_;
  // For VP8/VP9, uncompressed_header should not be encrypted; for AVC/HEVC,
  // the size and type NAL units should not be encrypted.
  VideoCodec video_codec_;
  // If this stream contains AVC, subsample encryption specifies that the size
  // and type of NAL units remain unencrypted. This field specifies the size of
  // the size field. Can be 1, 2 or 4 bytes.
  const uint8_t nalu_length_size_;
  int64_t clear_time_;

  scoped_ptr<VPxParser> vpx_parser_;

  DISALLOW_COPY_AND_ASSIGN(EncryptingFragmenter);
};

}  // namespace mp4
}  // namespace media
}  // namespace edash_packager

#endif  // MEDIA_FORMATS_MP4_ENCRYPTING_FRAGMENTER_H_