#include "config.h"
#if ENABLE(ENCRYPTED_MEDIA_V2)
#include "CDM.h"
#include "CDMPrivateClearKey.h"
#include "CDMPrivateMediaPlayer.h"
#include "CDMSession.h"
#include "MediaKeyError.h"
#include "MediaKeys.h"
#include "MediaPlayer.h"
#include <wtf/NeverDestroyed.h>
#include <wtf/text/WTFString.h>
#if PLATFORM(MAC) && ENABLE(MEDIA_SOURCE)
#include "CDMPrivateMediaSourceAVFObjC.h"
#endif
namespace WebCore {
struct CDMFactory {
WTF_MAKE_NONCOPYABLE(CDMFactory); WTF_MAKE_FAST_ALLOCATED;
public:
CDMFactory(CreateCDM constructor, CDMSupportsKeySystem supportsKeySystem, CDMSupportsKeySystemAndMimeType supportsKeySystemAndMimeType)
: constructor(constructor)
, supportsKeySystem(supportsKeySystem)
, supportsKeySystemAndMimeType(supportsKeySystemAndMimeType)
{
}
CreateCDM constructor;
CDMSupportsKeySystem supportsKeySystem;
CDMSupportsKeySystemAndMimeType supportsKeySystemAndMimeType;
};
static Vector<CDMFactory*>& installedCDMFactories()
{
static NeverDestroyed<Vector<CDMFactory*>> cdms;
static bool queriedCDMs = false;
if (!queriedCDMs) {
queriedCDMs = true;
cdms.get().append(new CDMFactory([](CDM* cdm) { return std::make_unique<CDMPrivateClearKey>(cdm); },
CDMPrivateClearKey::supportsKeySystem, CDMPrivateClearKey::supportsKeySystemAndMimeType));
cdms.get().append(new CDMFactory([](CDM* cdm) { return std::make_unique<CDMPrivateMediaPlayer>(cdm); },
CDMPrivateMediaPlayer::supportsKeySystem, CDMPrivateMediaPlayer::supportsKeySystemAndMimeType));
#if PLATFORM(MAC) && ENABLE(MEDIA_SOURCE)
cdms.get().append(new CDMFactory([](CDM* cdm) { return std::make_unique<CDMPrivateMediaSourceAVFObjC>(cdm); },
CDMPrivateMediaSourceAVFObjC::supportsKeySystem, CDMPrivateMediaSourceAVFObjC::supportsKeySystemAndMimeType));
#endif
}
return cdms;
}
void CDM::registerCDMFactory(CreateCDM constructor, CDMSupportsKeySystem supportsKeySystem, CDMSupportsKeySystemAndMimeType supportsKeySystemAndMimeType)
{
installedCDMFactories().append(new CDMFactory(constructor, supportsKeySystem, supportsKeySystemAndMimeType));
}
static CDMFactory* CDMFactoryForKeySystem(const String& keySystem)
{
for (auto& factory : installedCDMFactories()) {
if (factory->supportsKeySystem(keySystem))
return factory;
}
return 0;
}
bool CDM::supportsKeySystem(const String& keySystem)
{
return CDMFactoryForKeySystem(keySystem);
}
bool CDM::keySystemSupportsMimeType(const String& keySystem, const String& mimeType)
{
if (CDMFactory* factory = CDMFactoryForKeySystem(keySystem))
return factory->supportsKeySystemAndMimeType(keySystem, mimeType);
return false;
}
std::unique_ptr<CDM> CDM::create(const String& keySystem)
{
if (!supportsKeySystem(keySystem))
return nullptr;
return std::make_unique<CDM>(keySystem);
}
CDM::CDM(const String& keySystem)
: m_keySystem(keySystem)
, m_client(nullptr)
{
m_private = CDMFactoryForKeySystem(keySystem)->constructor(this);
}
CDM::~CDM()
{
}
bool CDM::supportsMIMEType(const String& mimeType) const
{
return m_private->supportsMIMEType(mimeType);
}
std::unique_ptr<CDMSession> CDM::createSession(CDMSessionClient* client)
{
std::unique_ptr<CDMSession> session = m_private->createSession(client);
if (mediaPlayer())
mediaPlayer()->setCDMSession(session.get());
return session;
}
MediaPlayer* CDM::mediaPlayer() const
{
if (!m_client)
return 0;
return m_client->cdmMediaPlayer(this);
}
}
#endif // ENABLE(ENCRYPTED_MEDIA_V2)