media: Add static MojoCdmService::GetCdm().

media/mojo/services/mojo_cdm_service.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/mojo/services/mojo_cdm_service.h"
 
+#include <map>
+
 #include "base/bind.h"
+#include "base/lazy_instance.h"
+#include "base/synchronization/lock.h"
 #include "media/base/cdm_config.h"
 #include "media/base/cdm_context.h"
 #include "media/base/cdm_factory.h"
 #include "media/base/cdm_key_information.h"
 #include "media/base/key_systems.h"
 #include "media/mojo/services/media_type_converters.h"
 #include "media/mojo/services/mojo_cdm_service_context.h"
 #include "mojo/common/common_type_converters.h"
 #include "mojo/common/url_type_converters.h"
 #include "url/gurl.h"
 
 namespace media {
 
+namespace {
+
+// Manages all CDMs created by MojoCdmService. Can only have one instance per
+// process so use a LazyInstance to ensure this.
+class CdmManager {
+ public:
+  CdmManager() {}
+  ~CdmManager() {}
+
+  // Returns the CDM associated with |cdm_id|. Can be called on any thread.
+  scoped_refptr<MediaKeys> GetCdm(int cdm_id) {
+    base::AutoLock lock(lock_);
+    auto iter = cdm_map_.find(cdm_id);
+    return iter == cdm_map_.end() ? nullptr : iter->second;
+  }
+
+  // Registers the |cdm| for |cdm_id|.
+  void RegisterCdm(int cdm_id, const scoped_refptr<MediaKeys>& cdm) {
+    base::AutoLock lock(lock_);
+    DCHECK(!cdm_map_.count(cdm_id));
+    cdm_map_[cdm_id] = cdm;
+  }
+
+  // Unregisters the CDM associated with |cdm_id|.
+  void UnregisterCdm(int cdm_id) {
+    base::AutoLock lock(lock_);
+    DCHECK(cdm_map_.count(cdm_id));
+    cdm_map_.erase(cdm_id);
+  }
+
+ private:
+  // Lock to protect |cdm_map_|.
+  base::Lock lock_;
+  std::map<int, scoped_refptr<MediaKeys>> cdm_map_;
+
+  DISALLOW_COPY_AND_ASSIGN(CdmManager);
+};
+
+base::LazyInstance<CdmManager>::Leaky g_cdm_manager = LAZY_INSTANCE_INITIALIZER;
+
+}  // namespace
+
 using SimpleMojoCdmPromise = MojoCdmPromise<>;
 using CdmIdMojoCdmPromise = MojoCdmPromise<int>;
 using NewSessionMojoCdmPromise = MojoCdmPromise<std::string>;
 
 int MojoCdmService::next_cdm_id_ = CdmContext::kInvalidCdmId + 1;
 
+// static
+scoped_refptr<MediaKeys> MojoCdmService::GetCdm(int cdm_id) {
+  DVLOG(1) << __FUNCTION__ << ": " << cdm_id;
+  return g_cdm_manager.Get().GetCdm(cdm_id);
+}
+
 MojoCdmService::MojoCdmService(
     base::WeakPtr<MojoCdmServiceContext> context,
     mojo::ServiceProvider* service_provider,
     CdmFactory* cdm_factory,
     mojo::InterfaceRequest<interfaces::ContentDecryptionModule> request)
     : binding_(this, request.Pass()),
       context_(context),
       service_provider_(service_provider),
       cdm_factory_(cdm_factory),
       cdm_id_(CdmContext::kInvalidCdmId),
       weak_factory_(this) {
   DCHECK(context_);
   DCHECK(cdm_factory_);
 }
 
 MojoCdmService::~MojoCdmService() {
+  g_cdm_manager.Get().UnregisterCdm(cdm_id_);
+
   if (cdm_id_ != CdmContext::kInvalidCdmId && context_)
     context_->UnregisterCdm(cdm_id_);
 }
 
 void MojoCdmService::SetClient(
     interfaces::ContentDecryptionModuleClientPtr client) {
   client_ = client.Pass();
 }
 
 void MojoCdmService::Initialize(
@@ skipping 91 matching lines @@
                                   const std::string& error_message) {
   // TODO(xhwang): This should not happen when KeySystemInfo is properly
   // populated. See http://crbug.com/469366
   if (!cdm || !context_) {
     promise->reject(MediaKeys::NOT_SUPPORTED_ERROR, 0, error_message);
     return;
   }
 
   cdm_ = cdm;
   cdm_id_ = next_cdm_id_++;
+
   context_->RegisterCdm(cdm_id_, this);
+  g_cdm_manager.Get().RegisterCdm(cdm_id_, cdm);
+
+  DVLOG(1) << __FUNCTION__ << ": CDM successfully created with ID " << cdm_id_;
   promise->resolve(cdm_id_);
 }
 
 void MojoCdmService::OnSessionMessage(const std::string& session_id,
                                       MediaKeys::MessageType message_type,
                                       const std::vector<uint8_t>& message,
                                       const GURL& legacy_destination_url) {
   DVLOG(2) << __FUNCTION__;
   client_->OnSessionMessage(
       session_id, static_cast<interfaces::CdmMessageType>(message_type),
@@ skipping 29 matching lines @@
                                           MediaKeys::Exception exception,
                                           uint32_t system_code,
                                           const std::string& error_message) {
   DVLOG(2) << __FUNCTION__;
   client_->OnLegacySessionError(
       session_id, static_cast<interfaces::CdmException>(exception), system_code,
       error_message);
 }
 
 }  // namespace media