Support encrypted audio stream in demuxer.

media/webm/webm_stream_parser.cc

 // Copyright (c) 2012 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/webm/webm_stream_parser.h"
 
 #include <string>
 
 #include "base/callback.h"
 #include "base/logging.h"
 #include "media/ffmpeg/ffmpeg_common.h"
 #include "media/filters/ffmpeg_glue.h"
 #include "media/filters/in_memory_url_protocol.h"
 #include "media/webm/webm_cluster_parser.h"
 #include "media/webm/webm_constants.h"
 #include "media/webm/webm_content_encodings.h"
 #include "media/webm/webm_info_parser.h"
 #include "media/webm/webm_tracks_parser.h"
 
 namespace media {
 
+// Fire needkey event through the |need_key_cb|.
+static void FireNeedKey(const StreamParser::NeedKeyCB& need_key_cb,
+                        const std::string& key_id) {
+  int key_id_size = key_id.size();
+  DCHECK_GT(key_id_size, 0);
+  scoped_array<uint8> key_id_array(new uint8[key_id_size]);
+  memcpy(key_id_array.get(), key_id.data(), key_id_size);
+  need_key_cb.Run(key_id_array.Pass(), key_id_size);
+}
+
 // Helper class that uses FFmpeg to create AudioDecoderConfig &
 // VideoDecoderConfig objects.
 //
 // This dependency on FFmpeg can be removed once we update WebMTracksParser
 // to parse the necessary data to construct AudioDecoderConfig &
 // VideoDecoderConfig objects. http://crbug.com/108756
 class FFmpegConfigHelper {
  public:
   FFmpegConfigHelper();
   ~FFmpegConfigHelper();
@@ skipping 312 matching lines @@
     int64 duration_in_us = info_parser.duration() * mult;
     duration = base::TimeDelta::FromMicroseconds(duration_in_us);
   }
 
   FFmpegConfigHelper config_helper;
   if (!config_helper.Parse(data, bytes_parsed)) {
     DVLOG(1) << "Failed to parse config data.";
     return -1;
   }
 
+  bool is_audio_encrypted = !tracks_parser.audio_encryption_key_id().empty();
+  AudioDecoderConfig audio_config;
+  if (is_audio_encrypted) {
+    const AudioDecoderConfig& original_audio_config =
+        config_helper.audio_config();
+
+    audio_config.Initialize(original_audio_config.codec(),
+                            original_audio_config.bits_per_channel(),
+                            original_audio_config.channel_layout(),
+                            original_audio_config.samples_per_second(),
+                            original_audio_config.extra_data(),
+                            original_audio_config.extra_data_size(),
+                            is_audio_encrypted, false);
+
+    FireNeedKey(need_key_cb_, tracks_parser.audio_encryption_key_id());

[acolwell GONE FROM CHROMIUM, 2012/10/11 01:11:24]

nit: Why are you using a static function here instead of a member function? 

[xhwang, 2012/10/11 16:36:10]

Done.

+  } else {
+    audio_config.CopyFrom(config_helper.audio_config());
+  }
+
   // TODO(xhwang): Support decryption of audio (see http://crbug.com/123421).
   bool is_video_encrypted = !tracks_parser.video_encryption_key_id().empty();
 
   VideoDecoderConfig video_config;
   if (is_video_encrypted) {
     const VideoDecoderConfig& original_video_config =
         config_helper.video_config();
     video_config.Initialize(original_video_config.codec(),
                             original_video_config.profile(),
                             original_video_config.format(),
                             original_video_config.coded_size(),
                             original_video_config.visible_rect(),
                             original_video_config.natural_size(),
                             original_video_config.extra_data(),
                             original_video_config.extra_data_size(),
                             is_video_encrypted, false);
 
-    // Fire needkey event.
-    std::string key_id = tracks_parser.video_encryption_key_id();
-    int key_id_size = key_id.size();
-    DCHECK_GT(key_id_size, 0);
-    scoped_array<uint8> key_id_array(new uint8[key_id_size]);
-    memcpy(key_id_array.get(), key_id.data(), key_id_size);
-    need_key_cb_.Run(key_id_array.Pass(), key_id_size);
+    FireNeedKey(need_key_cb_, tracks_parser.video_encryption_key_id());
   } else {
     video_config.CopyFrom(config_helper.video_config());
   }
 
-  if (!config_cb_.Run(config_helper.audio_config(), video_config)) {
+  if (!config_cb_.Run(audio_config, video_config)) {
     DVLOG(1) << "New config data isn't allowed.";
     return -1;
   }
 
   cluster_parser_.reset(new WebMClusterParser(
       info_parser.timecode_scale(),
       tracks_parser.audio_track_num(),
       tracks_parser.video_track_num(),
+      tracks_parser.audio_encryption_key_id(),
       tracks_parser.video_encryption_key_id()));
 
   ChangeState(kParsingClusters);
 
   if (!init_cb_.is_null()) {
     init_cb_.Run(true, duration);
     init_cb_.Reset();
   }
 
   return bytes_parsed;
@@ skipping 48 matching lines @@
   if (!video_buffers.empty() && !video_cb_.Run(video_buffers))
     return -1;
 
   if (cluster_ended)
     end_of_segment_cb_.Run();
 
   return bytes_parsed;
 }
 
 }  // namespace media