Initial implementation of Encrypted Media Extensions in Chrome.

webkit/media/webmediaplayer_impl.cc

Index: webkit/media/webmediaplayer_impl.cc
diff --git a/webkit/media/webmediaplayer_impl.cc b/webkit/media/webmediaplayer_impl.cc
index e5afb5d04cf465be009f49053a6fa792c62447df..55c48190a162f610418911ee368b53141e80b7c0 100644
--- a/webkit/media/webmediaplayer_impl.cc
+++ b/webkit/media/webmediaplayer_impl.cc
@@ -12,6 +12,7 @@
 #include "base/command_line.h"
 #include "base/message_loop_proxy.h"
 #include "base/metrics/histogram.h"
+#include "base/string_number_conversions.h"
 #include "base/synchronization/waitable_event.h"
 #include "media/audio/null_audio_sink.h"
 #include "media/base/filter_collection.h"
@@ -26,10 +27,12 @@
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebView.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebRect.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebSize.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebURL.h"
 #include "v8/include/v8.h"
 #include "webkit/media/buffered_data_source.h"
 #include "webkit/media/filter_helpers.h"
+#include "webkit/media/mime_util.h"
 #include "webkit/media/webmediaplayer_delegate.h"
 #include "webkit/media/webmediaplayer_proxy.h"
 #include "webkit/media/webvideoframe_impl.h"
@@ -668,6 +671,102 @@ void WebMediaPlayerImpl::sourceEndOfStream(
   proxy_->DemuxerEndOfStream(pipeline_status);
 }
 
+WebKit::WebMediaPlayer::MediaKeyException
+WebMediaPlayerImpl::generateKeyRequest(const WebKit::WebString& key_system,
+                                       const unsigned char* init_data,
+                                       unsigned init_data_length) {
+  if (!mime_util::isKeySystemSupported(key_system))
+    return WebKit::WebMediaPlayer::MediaKeyExceptionKeySystemNotSupported;
+
+  // Every request call creates a unique ID.
+  // TODO(ddorwin): Move this to the CDM implementations since the CDMs may
+  // create their own IDs and since CDMs supporting multiple renderer processes
+  // need globally unique IDs.
+  // Everything from here until the return should probably be handled by
+  // the decrypter - see http://crbug.com/123260.
+  static uint32_t next_available_session_id = 1;
+  uint32_t session_id = next_available_session_id++;
+
+  WebKit::WebString session_id_string(base::UintToString16(session_id));
+
+  DVLOG(1) << "generateKeyRequest: " << key_system.utf8().data() << ": "
+           << std::string(reinterpret_cast<const char*>(init_data),
+                          static_cast<size_t>(init_data_length))
+           << " [" << session_id_string.utf8().data() << "]";
+
+  // TODO(ddorwin): Generate a key request in the decrypter and fire
+  // keyMessage when it completes.
+  // For now, just fire the event with the init_data as the request.
+  const unsigned char* message = init_data;
+  unsigned message_length = init_data_length;
+
+  MessageLoop::current()->PostTask(FROM_HERE, base::Bind(
+      &WebKit::WebMediaPlayerClient::keyMessage,
+      base::Unretained(GetClient()),
+      key_system,
+      session_id_string,
+      message,
+      message_length));
+
+  return WebKit::WebMediaPlayer::MediaKeyExceptionNoError;
+}
+
+WebKit::WebMediaPlayer::MediaKeyException WebMediaPlayerImpl::addKey(
+    const WebKit::WebString& key_system,

[darin (slow to review), 2012/04/17 22:41:54]

nit: consider adding using directives for WebKit API types.  at least it seems
like a 'using WebKit::WebString' would be helpful.

[ddorwin, 2012/04/17 23:25:37]

Done.

+    const unsigned char* key,
+    unsigned key_length,
+    const unsigned char* init_data,
+    unsigned init_data_length,
+    const WebKit::WebString& session_id) {
+  if (!mime_util::isKeySystemSupported(key_system))
+    return WebKit::WebMediaPlayer::MediaKeyExceptionKeySystemNotSupported;
+
+
+  DVLOG(1) << "addKey: " << key_system.utf8().data() << ": "
+           << std::string(reinterpret_cast<const char*>(key),
+                          static_cast<size_t>(key_length)) << ", "
+           << std::string(reinterpret_cast<const char*>(init_data),
+                          static_cast<size_t>(init_data_length))
+           << " [" << session_id.utf8().data() << "]";
+
+  // TODO(ddorwin): Add the key to the decrypter and fire keyAdded when it
+  // completes. Check the key length there.
+  // Everything from here until the return should probably be handled by
+  // the decrypter - see http://crbug.com/123260.
+  // Temporarily, fire an error for invalid key length so we can test the error
+  // event and fire the keyAdded event in all other cases.
+  const unsigned kSupportedKeyLength = 16;  // 128-bit key.
+  if (key_length != kSupportedKeyLength) {
+    DLOG(ERROR) << "addKey: invalid key length: " << key_length;
+    MessageLoop::current()->PostTask(FROM_HERE, base::Bind(
+        &WebKit::WebMediaPlayerClient::keyError,
+        base::Unretained(GetClient()),
+        key_system,
+        session_id,
+        WebKit::WebMediaPlayerClient::MediaKeyErrorCodeUnknown,
+        0));
+  } else {
+    MessageLoop::current()->PostTask(FROM_HERE, base::Bind(
+        &WebKit::WebMediaPlayerClient::keyAdded,
+        base::Unretained(GetClient()),
+        key_system,
+        session_id));
+  }
+
+  return WebKit::WebMediaPlayer::MediaKeyExceptionNoError;
+}
+
+WebKit::WebMediaPlayer::MediaKeyException WebMediaPlayerImpl::cancelKeyRequest(
+    const WebKit::WebString& key_system,
+    const WebKit::WebString& session_id) {
+  if (!mime_util::isKeySystemSupported(key_system))
+    return WebKit::WebMediaPlayer::MediaKeyExceptionKeySystemNotSupported;
+
+  // TODO(ddorwin): Cancel the key request in the decrypter.
+
+  return WebKit::WebMediaPlayer::MediaKeyExceptionNoError;
+}
+
 void WebMediaPlayerImpl::WillDestroyCurrentMessageLoop() {
   Destroy();
   main_loop_ = NULL;