[Clank] Add JNI interface for media persistent license storage

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

 // Copyright 2017 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.media.MediaDrm;
 
 import org.chromium.base.Callback;
+import org.chromium.media.MediaDrmStorageBridge.PersistentInfo;
 
 import java.nio.ByteBuffer;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.List;
 import java.util.UUID;
 
 /**
  * The class manages relations among eme session ID, drm session ID and keyset
@@ skipping 131 matching lines @@
 
         // Private methods that are visible in this file only.
 
         private SessionId sessionId() {
             return mSessionId;
         }
 
         private void setKeyType(int keyType) {
             mKeyType = keyType;
         }
+
+        private PersistentInfo toPersistentInfo() {
+            assert mSessionId.keySetId() != null;
+
+            return new PersistentInfo(mSessionId.emeId(), mSessionId.keySetId(), mMimeType);
+        }
     }
 
     // Maps from DRM/EME session ID to SessionInfo. SessionInfo contains
     // SessionId, so that we can:
     //   1. Get SessionInfo with EME/DRM session ID.
     //   2. Get SessionId from EME/DRM session ID.
     //   3. Get EME/DRM session ID from DRM/EME session ID.
     // SessionId always has a valid EME session ID, so all opened session should
     // have an entry in mEmeSessionInfoMap.
     private HashMap<ByteBuffer, SessionInfo> mEmeSessionInfoMap;
     private HashMap<ByteBuffer, SessionInfo> mDrmSessionInfoMap;
 
-    public MediaDrmSessionManager() {
+    // The persistent storage to record map from EME session ID to key set ID
+    // for persistent license.
+    private MediaDrmStorageBridge mStorage;
+
+    public MediaDrmSessionManager(MediaDrmStorageBridge storage) {
         mEmeSessionInfoMap = new HashMap<>();
         mDrmSessionInfoMap = new HashMap<>();
+
+        mStorage = storage;
     }
 
     /**
      * Set key set ID. It should only be called for persistent license session.
      */
     void setKeySetId(SessionId sessionId, byte[] keySetId, Callback<Boolean> callback) {
         assert get(sessionId) != null;
         assert get(sessionId).keyType() == MediaDrm.KEY_TYPE_OFFLINE;
         assert sessionId.keySetId() == null;
 
         sessionId.setKeySetId(keySetId);
 
-        // TODO(yucliu): Write updated key set ID to persistent storage.
-        callback.onResult(true);
+        mStorage.saveInfo(get(sessionId).toPersistentInfo(), callback);
     }
 
     /**
      * Remove session and related infomration from memory, but doesn't touch
      * persistent storage.
      */
     void remove(SessionId sessionId) {
         SessionInfo info = get(sessionId);
 
         assert info != null;
@@ skipping 39 matching lines @@
 
     private SessionId getSessionIdFromMap(HashMap<ByteBuffer, SessionInfo> map, byte[] id) {
         SessionInfo info = map.get(ByteBuffer.wrap(id));
         if (info == null) {
             return null;
         }
 
         return info.sessionId();
     }
 }