зеркало из https://github.com/mozilla/gecko-dev.git
104 строки
2.7 KiB
C++
104 строки
2.7 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* 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 WidevineUtils_h_
|
|
#define WidevineUtils_h_
|
|
|
|
#include "stddef.h"
|
|
#include "content_decryption_module.h"
|
|
#include "gmp-api/gmp-decryption.h"
|
|
#include "gmp-api/gmp-platform.h"
|
|
#include "nsISupportsImpl.h"
|
|
#include "nsTArray.h"
|
|
#include "mozilla/Logging.h"
|
|
|
|
namespace mozilla {
|
|
|
|
namespace detail {
|
|
LogModule* GetCDMLog();
|
|
} // namespace detail
|
|
|
|
#define CDM_LOG(...) MOZ_LOG(detail::GetCDMLog(), mozilla::LogLevel::Debug, (__VA_ARGS__))
|
|
|
|
#define ENSURE_TRUE(condition, rv) { \
|
|
if (!(condition)) {\
|
|
CDM_LOG("ENSURE_TRUE FAILED %s:%d", __FILE__, __LINE__); \
|
|
return rv; \
|
|
} \
|
|
} \
|
|
|
|
#define ENSURE_GMP_SUCCESS(err, rv) { \
|
|
if (GMP_FAILED(err)) {\
|
|
CDM_LOG("ENSURE_GMP_SUCCESS FAILED %s:%d", __FILE__, __LINE__); \
|
|
return rv; \
|
|
} \
|
|
} \
|
|
|
|
GMPErr
|
|
ToGMPErr(cdm::Status aStatus);
|
|
|
|
class WidevineDecryptor;
|
|
|
|
class CDMWrapper {
|
|
public:
|
|
NS_INLINE_DECL_THREADSAFE_REFCOUNTING(CDMWrapper)
|
|
|
|
explicit CDMWrapper(cdm::ContentDecryptionModule_8* aCDM,
|
|
WidevineDecryptor* aDecryptor);
|
|
cdm::ContentDecryptionModule_8* GetCDM() const { return mCDM; }
|
|
private:
|
|
~CDMWrapper();
|
|
cdm::ContentDecryptionModule_8* mCDM;
|
|
RefPtr<WidevineDecryptor> mDecryptor;
|
|
};
|
|
|
|
void InitInputBuffer(const GMPEncryptedBufferMetadata* aCrypto,
|
|
int64_t aTimestamp,
|
|
const uint8_t* aData,
|
|
size_t aDataSize,
|
|
cdm::InputBuffer &aInputBuffer,
|
|
nsTArray<cdm::SubsampleEntry> &aSubsamples);
|
|
|
|
class WidevineBuffer : public cdm::Buffer
|
|
{
|
|
public:
|
|
explicit WidevineBuffer(size_t aSize);
|
|
~WidevineBuffer() override;
|
|
void Destroy() override;
|
|
uint32_t Capacity() const override;
|
|
uint8_t* Data() override;
|
|
void SetSize(uint32_t aSize) override;
|
|
uint32_t Size() const override;
|
|
|
|
// Moves contents of buffer out into temporary.
|
|
// Note: This empties the buffer.
|
|
nsTArray<uint8_t> ExtractBuffer();
|
|
|
|
private:
|
|
nsTArray<uint8_t> mBuffer;
|
|
WidevineBuffer(const WidevineBuffer&);
|
|
void operator=(const WidevineBuffer&);
|
|
};
|
|
|
|
class WidevineDecryptedBlock : public cdm::DecryptedBlock
|
|
{
|
|
public:
|
|
|
|
WidevineDecryptedBlock();
|
|
~WidevineDecryptedBlock() override;
|
|
void SetDecryptedBuffer(cdm::Buffer* aBuffer) override;
|
|
cdm::Buffer* DecryptedBuffer() override;
|
|
void SetTimestamp(int64_t aTimestamp) override;
|
|
int64_t Timestamp() const override;
|
|
|
|
private:
|
|
cdm::Buffer* mBuffer;
|
|
int64_t mTimestamp;
|
|
};
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif // WidevineUtils_h_
|