Move MediaDrmBridge provision communication to native side.

media/base/android/java/src/org/chromium/media/MediaDrmBridge.java

 // Copyright 2013 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.
 
 package org.chromium.media;
 
 import android.annotation.TargetApi;
 import android.media.MediaCrypto;
 import android.media.MediaDrm;
-import android.os.AsyncTask;
 import android.os.Build;
 
 import org.chromium.base.Log;
 import org.chromium.base.annotations.CalledByNative;
 import org.chromium.base.annotations.JNINamespace;
 
-import java.io.BufferedInputStream;
-import java.io.ByteArrayOutputStream;
-import java.io.IOException;
-import java.net.HttpURLConnection;
-import java.net.MalformedURLException;
-import java.net.URL;
 import java.nio.ByteBuffer;
 import java.util.ArrayDeque;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.List;
 import java.util.UUID;
 
 /**
  * A wrapper of the android MediaDrm class. Each MediaDrmBridge manages multiple
@@ skipping 17 matching lines @@
     //   b) Finish createSession() if previous createSession() was interrupted
     //      by a NotProvisionedException.
     // - Whenever an unexpected error occurred, we'll call release() to release
     //   all resources and clear all states. In that case all calls to this
     //   object will be no-op. All public APIs and callbacks should check
     //   mMediaBridge to make sure release() hasn't been called. Also, we call
     //   release() immediately after the error happens (e.g. after mMediaDrm)
     //   calls. Indirect calls should not call release() again to avoid
     //   duplication (even though it doesn't hurt to call release() twice).
 
-    private static final String TAG = "cr.media";
+    private static final String TAG = "cr_media";
     private static final String SECURITY_LEVEL = "securityLevel";
     private static final String SERVER_CERTIFICATE = "serviceCertificate";
     private static final String PRIVACY_MODE = "privacyMode";
     private static final String SESSION_SHARING = "sessionSharing";
     private static final String ENABLE = "enable";
     private static final int INVALID_SESSION_ID = 0;
     private static final char[] HEX_CHAR_LOOKUP = "0123456789ABCDEF".toCharArray();
     private static final long INVALID_NATIVE_MEDIA_DRM_BRIDGE = 0;
 
     // On Android L and before, MediaDrm doesn't support KeyStatus. Use a dummy
@@ skipping 335 matching lines @@
     private MediaCrypto getMediaCrypto() {
         return mMediaCrypto;
     }
 
     /**
      * Reset the device DRM credentials.
      */
     @CalledByNative
     private void resetDeviceCredentials() {
         mResetDeviceCredentialsPending = true;
-        MediaDrm.ProvisionRequest request = mMediaDrm.getProvisionRequest();
-        PostRequestTask postTask = new PostRequestTask(request.getData());
-        postTask.executeOnExecutor(AsyncTask.THREAD_POOL_EXECUTOR, request.getDefaultUrl());
+        startProvisioning();
     }
 
     /**
      * Destroy the MediaDrmBridge object.
      */
     @CalledByNative
     private void destroy() {
         mNativeMediaDrmBridge = INVALID_NATIVE_MEDIA_DRM_BRIDGE;
         if (mMediaDrm != null) {
             release();
@@ skipping 304 matching lines @@
     @CalledByNative
     private String getSecurityLevel() {
         if (mMediaDrm == null) {
             Log.e(TAG, "getSecurityLevel() called when MediaDrm is null.");
             return null;
         }
         return mMediaDrm.getPropertyString("securityLevel");
     }
 
     private void startProvisioning() {
+        if (mProvisioningPending) {
+            Log.d(TAG, "startProvisioning: another provisioning is in progress, returning");
+            return;
+        }
+
         Log.d(TAG, "startProvisioning");
+        mProvisioningPending = true;
         assert mMediaDrm != null;
-        assert !mProvisioningPending;
-        mProvisioningPending = true;
         MediaDrm.ProvisionRequest request = mMediaDrm.getProvisionRequest();
-        PostRequestTask postTask = new PostRequestTask(request.getData());
-        postTask.executeOnExecutor(AsyncTask.THREAD_POOL_EXECUTOR, request.getDefaultUrl());
+
+        if (isNativeMediaDrmBridgeValid()) {
+            nativeOnStartProvisioning(
+                    mNativeMediaDrmBridge, request.getDefaultUrl(), request.getData());
+        }
     }
 
     /**
      * Called when the provision response is received.
      *
      * @param response Response data from the provision server.
      */
-    private void onProvisionResponse(byte[] response) {
-        Log.d(TAG, "onProvisionResponse()");
+    @CalledByNative
+    private void processProvisionResponse(boolean success, byte[] response) {
+        Log.d(TAG, "processProvisionResponse()");
         assert mProvisioningPending;
         mProvisioningPending = false;
 
         // If |mMediaDrm| is released, there is no need to callback native.
         if (mMediaDrm == null) {
             return;
         }
 
-        boolean success = provideProvisionResponse(response);
+        if (success) {
+            success = provideProvisionResponse(response);

[xhwang, 2015/11/13 05:15:42]

nit: usually we don't change the input parameter. can we have a different
variable?

[Tima Vaisburd, 2015/11/13 21:04:44]

Done.

+        }
 
         if (mResetDeviceCredentialsPending) {
             onResetDeviceCredentialsCompleted(success);
             mResetDeviceCredentialsPending = false;
         }
 
         if (success) {
             processPendingCreateSessionData();
         }
     }
 
     /**
-     * Provide the provisioning response to MediaDrm.
+     * Provides the provision response to MediaDrm.
      *
      * @returns false if the response is invalid or on error, true otherwise.
      */
     boolean provideProvisionResponse(byte[] response) {
         if (response == null || response.length == 0) {
             Log.e(TAG, "Invalid provision response.");
             return false;
         }
 
         try {
@@ skipping 168 matching lines @@
 
     @TargetApi(Build.VERSION_CODES.M)
     private class ExpirationUpdateListener implements MediaDrm.OnExpirationUpdateListener {
         @Override
         public void onExpirationUpdate(MediaDrm md, byte[] sessionId, long expirationTime) {
             Log.d(TAG, "ExpirationUpdate: " + bytesToHexString(sessionId) + ", " + expirationTime);
             onSessionExpirationUpdate(sessionId, expirationTime);
         }
     }
 
-    private class PostRequestTask extends AsyncTask<String, Void, Void> {
-        private static final String TAG = "PostRequestTask";
-
-        private byte[] mDrmRequest;
-        private byte[] mResponseBody;
-
-        public PostRequestTask(byte[] drmRequest) {
-            mDrmRequest = drmRequest;
-        }
-
-        @Override
-        protected Void doInBackground(String... urls) {
-            mResponseBody = postRequest(urls[0], mDrmRequest);
-            if (mResponseBody != null) {
-                Log.d(TAG, "response length=%d", mResponseBody.length);
-            }
-            return null;
-        }
-
-        private byte[] postRequest(String url, byte[] drmRequest) {
-            HttpURLConnection urlConnection = null;
-            try {
-                URL request = new URL(url + "&signedRequest=" + new String(drmRequest));
-                urlConnection = (HttpURLConnection) request.openConnection();
-                urlConnection.setDoOutput(true);
-                urlConnection.setDoInput(true);
-                urlConnection.setUseCaches(false);
-                urlConnection.setRequestMethod("POST");
-                urlConnection.setRequestProperty("User-Agent", "Widevine CDM v1.0");
-                urlConnection.setRequestProperty("Content-Type", "application/json");
-
-                int responseCode = urlConnection.getResponseCode();
-                if (responseCode == 200) {
-                    BufferedInputStream bis =
-                            new BufferedInputStream(urlConnection.getInputStream());
-                    ByteArrayOutputStream bos = new ByteArrayOutputStream();
-                    int read = 0;
-                    int bufferSize = 512;
-                    byte[] buffer = new byte[bufferSize];
-                    try {
-                        while (true) {
-                            read = bis.read(buffer);
-                            if (read == -1) break;
-                            bos.write(buffer, 0, read);
-                        }
-                    } finally {
-                        bis.close();
-                    }
-                    return bos.toByteArray();
-                } else {
-                    Log.d(TAG, "Server returned HTTP error code %d", responseCode);
-                    return null;
-                }
-            } catch (MalformedURLException e) {
-                e.printStackTrace();
-            } catch (IOException e) {
-                e.printStackTrace();
-            } catch (IllegalStateException e) {
-                e.printStackTrace();
-            } finally {
-                if (urlConnection != null) urlConnection.disconnect();
-            }
-            return null;
-        }
-
-        @Override
-        protected void onPostExecute(Void v) {
-            onProvisionResponse(mResponseBody);
-        }
-    }
-
     // Native functions. At the native side, must post the task immediately to
     // avoid reentrancy issues.
     private native void nativeOnMediaCryptoReady(long nativeMediaDrmBridge);
 
+    private native void nativeOnStartProvisioning(
+            long nativeMediaDrmBridge, String defaultUrl, byte[] requestData);
+
     private native void nativeOnPromiseResolved(long nativeMediaDrmBridge, long promiseId);
     private native void nativeOnPromiseResolvedWithSession(
             long nativeMediaDrmBridge, long promiseId, byte[] sessionId);
     private native void nativeOnPromiseRejected(
             long nativeMediaDrmBridge, long promiseId, String errorMessage);
 
     private native void nativeOnSessionMessage(long nativeMediaDrmBridge, byte[] sessionId,
             int requestType, byte[] message, String destinationUrl);
     private native void nativeOnSessionClosed(long nativeMediaDrmBridge, byte[] sessionId);
     private native void nativeOnSessionKeysChange(long nativeMediaDrmBridge, byte[] sessionId,
             Object[] keysInfo, boolean hasAdditionalUsableKey);
     private native void nativeOnSessionExpirationUpdate(
             long nativeMediaDrmBridge, byte[] sessionId, long expirationTime);
     private native void nativeOnLegacySessionError(
             long nativeMediaDrmBridge, byte[] sessionId, String errorMessage);
 
     private native void nativeOnResetDeviceCredentialsCompleted(
             long nativeMediaDrmBridge, boolean success);
 }