summaryrefslogtreecommitdiffstats
path: root/dom/media/platforms/agnostic/eme/SamplesWaitingForKey.h
blob: 06d72e3aae546bad172ca33fefbb1c2188e83894 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef SamplesWaitingForKey_h_
#define SamplesWaitingForKey_h_

#include <functional>

#include "MediaInfo.h"
#include "mozilla/MozPromise.h"
#include "mozilla/Mutex.h"
#include "mozilla/RefPtr.h"

namespace mozilla {

typedef nsTArray<uint8_t> CencKeyId;

class CDMProxy;
template <typename... Es>
class MediaEventProducer;
class MediaRawData;

// Encapsulates the task of waiting for the CDMProxy to have the necessary
// keys to decrypt a given sample.
class SamplesWaitingForKey {
 public:
  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(SamplesWaitingForKey)

  typedef MozPromise<RefPtr<MediaRawData>, bool, /* IsExclusive = */ true>
      WaitForKeyPromise;

  SamplesWaitingForKey(
      CDMProxy* aProxy, TrackInfo::TrackType aType,
      const std::function<MediaEventProducer<TrackInfo::TrackType>*()>&
          aOnWaitingForKeyEvent);

  // Returns a promise that will be resolved if or when a key for decoding the
  // sample becomes usable.
  RefPtr<WaitForKeyPromise> WaitIfKeyNotUsable(MediaRawData* aSample);

  void NotifyUsable(const CencKeyId& aKeyId);

  void Flush();

  void BreakCycles();

 protected:
  ~SamplesWaitingForKey();

 private:
  Mutex mMutex MOZ_UNANNOTATED;
  RefPtr<CDMProxy> mProxy;
  struct SampleEntry {
    RefPtr<MediaRawData> mSample;
    MozPromiseHolder<WaitForKeyPromise> mPromise;
  };
  nsTArray<SampleEntry> mSamples;
  const TrackInfo::TrackType mType;
  const std::function<MediaEventProducer<TrackInfo::TrackType>*()>
      mOnWaitingForKeyEvent;
};

}  // namespace mozilla

#endif  //  SamplesWaitingForKey_h_