Add allocator interface for use by cdm_adapter

media/cdm/cdm_adapter.h

 // Copyright 2015 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.
 
 #ifndef MEDIA_CDM_CDM_ADAPTER_H_
 #define MEDIA_CDM_CDM_ADAPTER_H_
 
 #include <stdint.h>
 
 #include <string>
@@ skipping 13 matching lines @@
 #include "media/base/cdm_promise_adapter.h"
 #include "media/base/decryptor.h"
 #include "media/base/media_export.h"
 #include "media/base/media_keys.h"
 #include "media/cdm/api/content_decryption_module.h"
 #include "ui/gfx/geometry/size.h"
 
 namespace media {
 
 class AudioFramesImpl;
+class CdmBufferAllocator;
 class CdmWrapper;
 
 class MEDIA_EXPORT CdmAdapter : public MediaKeys,
                                 public CdmContext,
                                 public Decryptor,
                                 NON_EXPORTED_BASE(public cdm::Host_7),
                                 NON_EXPORTED_BASE(public cdm::Host_8) {
  public:
   // Create the CDM using |cdm_path| and initialize it using |key_system| and
   // |cdm_config|. Callbacks will be used for events generated by the CDM.
   // |cdm_created_cb| will be called when the CDM is loaded and initialized.
   static void Create(
       const std::string& key_system,
       const base::FilePath& cdm_path,
       const CdmConfig& cdm_config,
       const SessionMessageCB& session_message_cb,
       const SessionClosedCB& session_closed_cb,
       const LegacySessionErrorCB& legacy_session_error_cb,
       const SessionKeysChangeCB& session_keys_change_cb,
       const SessionExpirationUpdateCB& session_expiration_update_cb,
-      const CdmCreatedCB& cdm_created_cb);
+      const CdmCreatedCB& cdm_created_cb,
+      CdmBufferAllocator* allocator);
 
   // MediaKeys implementation.
   void SetServerCertificate(const std::vector<uint8_t>& certificate,
                             scoped_ptr<SimpleCdmPromise> promise) final;
   void CreateSessionAndGenerateRequest(
       SessionType session_type,
       EmeInitDataType init_data_type,
       const std::vector<uint8_t>& init_data,
       scoped_ptr<NewSessionCdmPromise> promise) final;
   void LoadSession(SessionType session_type,
@@ skipping 76 matching lines @@
                                     cdm::Status decoder_status) override;
   cdm::FileIO* CreateFileIO(cdm::FileIOClient* client) override;
 
  private:
   CdmAdapter(const std::string& key_system,
              const CdmConfig& cdm_config,
              const SessionMessageCB& session_message_cb,
              const SessionClosedCB& session_closed_cb,
              const LegacySessionErrorCB& legacy_session_error_cb,
              const SessionKeysChangeCB& session_keys_change_cb,
-             const SessionExpirationUpdateCB& session_expiration_update_cb);
+             const SessionExpirationUpdateCB& session_expiration_update_cb,
+             CdmBufferAllocator* allocator);
   ~CdmAdapter() final;
 
   // Load the CDM using |cdm_path| and initialize it. |promise| is resolved if
   // the CDM is successfully loaded and initialized, rejected otherwise.
   void Initialize(const base::FilePath& cdm_path,
                   scoped_ptr<media::SimpleCdmPromise> promise);
 
   // Create an instance of the |key_system| CDM contained in |cdm_path|.
   // Caller owns the returned pointer. On error (unable to load, does not
   // support |key_system|, does not support an supported interface, etc.)
@@ skipping 35 matching lines @@
   NewKeyCB new_video_key_cb_;
 
   // Keep track of video frame natural size from the latest configuration
   // as the CDM doesn't provide it.
   gfx::Size natural_size_;
 
   // Keep track of audio parameters.
   int audio_samples_per_second_;
   ChannelLayout audio_channel_layout_;
 
+  CdmBufferAllocator* allocator_;
+
   scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
 
   // NOTE: Weak pointers must be invalidated before all other member variables.
   base::WeakPtrFactory<CdmAdapter> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(CdmAdapter);
 };
 
 }  // namespace media
 
 #endif  // MEDIA_CDM_CDM_ADAPTER_H_