drm/drmserver/DrmManager.cpp - platform/frameworks/av - Git at Google

 /*
  * Copyright (C) 2010 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 //#define LOG_NDEBUG 0
 #define LOG_TAG "DrmManager(Native)"
 #include "utils/Log.h"

 #include <utils/String8.h>
 #include <drm/DrmInfo.h>
 #include <drm/DrmInfoEvent.h>
 #include <drm/DrmRights.h>
 #include <drm/DrmConstraints.h>
 #include <drm/DrmMetadata.h>
 #include <drm/DrmInfoStatus.h>
 #include <drm/DrmInfoRequest.h>
 #include <drm/DrmSupportInfo.h>
 #include <drm/DrmConvertedStatus.h>
 #include <IDrmEngine.h>

 #include "DrmManager.h"
 #include "ReadWriteUtils.h"

 #define DECRYPT_FILE_ERROR -1

 using namespace android;

 const String8 DrmManager::EMPTY_STRING("");

 DrmManager::DrmManager() :
     mDecryptSessionId(0),
     mConvertId(0) {

 }

 DrmManager::~DrmManager() {

 }

 int DrmManager::addUniqueId(bool isNative) {
     Mutex::Autolock _l(mLock);

     int temp = 0;
     bool foundUniqueId = false;
     const int size = mUniqueIdVector.size();
     const int uniqueIdRange = 0xfff;
     int maxLoopTimes = (uniqueIdRange - 1) / 2;
     srand(time(NULL));

     while (!foundUniqueId) {
         temp = rand() & uniqueIdRange;

         if (isNative) {
             // set a flag to differentiate DrmManagerClient
             // created from native side and java side
             temp |= 0x1000;
         }

         int index = 0;
         for (; index < size; ++index) {
             if (mUniqueIdVector.itemAt(index) == temp) {
                 foundUniqueId = false;
                 break;
             }
         }
         if (index == size) {
             foundUniqueId = true;
         }

         maxLoopTimes --;
         LOG_FATAL_IF(maxLoopTimes <= 0, "cannot find an unique ID for this session");
     }

     mUniqueIdVector.push(temp);
     return temp;
 }

 void DrmManager::removeUniqueId(int uniqueId) {
     Mutex::Autolock _l(mLock);
     for (unsigned int i = 0; i < mUniqueIdVector.size(); i++) {
         if (uniqueId == mUniqueIdVector.itemAt(i)) {
             mUniqueIdVector.removeAt(i);
             break;
         }
     }
 }

 status_t DrmManager::loadPlugIns() {

     String8 vendorPluginDirPath("/vendor/lib/drm");
     loadPlugIns(vendorPluginDirPath);

     String8 pluginDirPath("/system/lib/drm");
     loadPlugIns(pluginDirPath);
     return DRM_NO_ERROR;

 }

 status_t DrmManager::loadPlugIns(const String8& plugInDirPath) {
     mPlugInManager.loadPlugIns(plugInDirPath);
     Vector<String8> plugInPathList = mPlugInManager.getPlugInIdList();
     for (unsigned int i = 0; i < plugInPathList.size(); ++i) {
         String8 plugInPath = plugInPathList[i];
         DrmSupportInfo* info = mPlugInManager.getPlugIn(plugInPath).getSupportInfo(0);
         if (NULL != info) {
             if (mSupportInfoToPlugInIdMap.indexOfKey(*info) < 0) {
                 mSupportInfoToPlugInIdMap.add(*info, plugInPath);
             }
             delete info;
         }
     }
     return DRM_NO_ERROR;
 }

 status_t DrmManager::unloadPlugIns() {
     Mutex::Autolock _l(mLock);
     mConvertSessionMap.clear();
     mDecryptSessionMap.clear();
     mPlugInManager.unloadPlugIns();
     mSupportInfoToPlugInIdMap.clear();
     return DRM_NO_ERROR;
 }

 status_t DrmManager::setDrmServiceListener(
             int uniqueId, const sp<IDrmServiceListener>& drmServiceListener) {
     Mutex::Autolock _l(mListenerLock);
     if (NULL != drmServiceListener.get()) {
         mServiceListeners.add(uniqueId, drmServiceListener);
     } else {
         mServiceListeners.removeItem(uniqueId);
     }
     return DRM_NO_ERROR;
 }

 void DrmManager::addClient(int uniqueId) {
     Mutex::Autolock _l(mLock);
     if (!mSupportInfoToPlugInIdMap.isEmpty()) {
         Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();
         for (unsigned int index = 0; index < plugInIdList.size(); index++) {
             IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInIdList.itemAt(index));
             rDrmEngine.initialize(uniqueId);
             rDrmEngine.setOnInfoListener(uniqueId, this);
         }
     }
 }

 void DrmManager::removeClient(int uniqueId) {
     Mutex::Autolock _l(mLock);
     Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();
     for (unsigned int index = 0; index < plugInIdList.size(); index++) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInIdList.itemAt(index));
         rDrmEngine.terminate(uniqueId);
     }
 }

 DrmConstraints* DrmManager::getConstraints(int uniqueId, const String8* path, const int action) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, *path);
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         return rDrmEngine.getConstraints(uniqueId, path, action);
     }
     return NULL;
 }

 DrmMetadata* DrmManager::getMetadata(int uniqueId, const String8* path) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, *path);
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         return rDrmEngine.getMetadata(uniqueId, path);
     }
     return NULL;
 }

 status_t DrmManager::installDrmEngine(int uniqueId, const String8& absolutePath) {
     Mutex::Autolock _l(mLock);
     mPlugInManager.loadPlugIn(absolutePath);

     IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(absolutePath);
     rDrmEngine.initialize(uniqueId);
     rDrmEngine.setOnInfoListener(uniqueId, this);

     DrmSupportInfo* info = rDrmEngine.getSupportInfo(0);
     mSupportInfoToPlugInIdMap.add(*info, absolutePath);
     delete info;

     return DRM_NO_ERROR;
 }

 bool DrmManager::canHandle(int uniqueId, const String8& path, const String8& mimeType) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInId(mimeType);
     bool result = (EMPTY_STRING != plugInId) ? true : false;

     if (0 < path.length()) {
         if (result) {
             IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
             result = rDrmEngine.canHandle(uniqueId, path);
         } else {
             String8 extension = path.getPathExtension();
             if (String8("") != extension) {
                 result = canHandle(uniqueId, path);
             }
         }
     }
     return result;
 }

 DrmInfoStatus* DrmManager::processDrmInfo(int uniqueId, const DrmInfo* drmInfo) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInId(drmInfo->getMimeType());
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         return rDrmEngine.processDrmInfo(uniqueId, drmInfo);
     }
     return NULL;
 }

 bool DrmManager::canHandle(int uniqueId, const String8& path) {
     bool result = false;
     Vector<String8> plugInPathList = mPlugInManager.getPlugInIdList();

     for (unsigned int i = 0; i < plugInPathList.size(); ++i) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInPathList[i]);
         result = rDrmEngine.canHandle(uniqueId, path);

         if (result) {
             break;
         }
     }
     return result;
 }

 DrmInfo* DrmManager::acquireDrmInfo(int uniqueId, const DrmInfoRequest* drmInfoRequest) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInId(drmInfoRequest->getMimeType());
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         return rDrmEngine.acquireDrmInfo(uniqueId, drmInfoRequest);
     }
     return NULL;
 }

 status_t DrmManager::saveRights(int uniqueId, const DrmRights& drmRights,
             const String8& rightsPath, const String8& contentPath) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInId(drmRights.getMimeType());
     status_t result = DRM_ERROR_UNKNOWN;
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         result = rDrmEngine.saveRights(uniqueId, drmRights, rightsPath, contentPath);
     }
     return result;
 }

 String8 DrmManager::getOriginalMimeType(int uniqueId, const String8& path) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         return rDrmEngine.getOriginalMimeType(uniqueId, path);
     }
     return EMPTY_STRING;
 }

 int DrmManager::getDrmObjectType(int uniqueId, const String8& path, const String8& mimeType) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInId(uniqueId, path, mimeType);
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         return rDrmEngine.getDrmObjectType(uniqueId, path, mimeType);
     }
     return DrmObjectType::UNKNOWN;
 }

 int DrmManager::checkRightsStatus(int uniqueId, const String8& path, int action) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         return rDrmEngine.checkRightsStatus(uniqueId, path, action);
     }
     return RightsStatus::RIGHTS_INVALID;
 }

 status_t DrmManager::consumeRights(
     int uniqueId, DecryptHandle* decryptHandle, int action, bool reserve) {
     status_t result = DRM_ERROR_UNKNOWN;
     Mutex::Autolock _l(mDecryptLock);
     if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
         result = drmEngine->consumeRights(uniqueId, decryptHandle, action, reserve);
     }
     return result;
 }

 status_t DrmManager::setPlaybackStatus(
     int uniqueId, DecryptHandle* decryptHandle, int playbackStatus, int64_t position) {
     status_t result = DRM_ERROR_UNKNOWN;
     Mutex::Autolock _l(mDecryptLock);
     if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
         result = drmEngine->setPlaybackStatus(uniqueId, decryptHandle, playbackStatus, position);
     }
     return result;
 }

 bool DrmManager::validateAction(
     int uniqueId, const String8& path, int action, const ActionDescription& description) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         return rDrmEngine.validateAction(uniqueId, path, action, description);
     }
     return false;
 }

 status_t DrmManager::removeRights(int uniqueId, const String8& path) {
     Mutex::Autolock _l(mLock);
     const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
     status_t result = DRM_ERROR_UNKNOWN;
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
         result = rDrmEngine.removeRights(uniqueId, path);
     }
     return result;
 }

 status_t DrmManager::removeAllRights(int uniqueId) {
     Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();
     status_t result = DRM_ERROR_UNKNOWN;
     for (unsigned int index = 0; index < plugInIdList.size(); index++) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInIdList.itemAt(index));
         result = rDrmEngine.removeAllRights(uniqueId);
         if (DRM_NO_ERROR != result) {
             break;
         }
     }
     return result;
 }

 int DrmManager::openConvertSession(int uniqueId, const String8& mimeType) {
     Mutex::Autolock _l(mConvertLock);
     int convertId = -1;

     const String8 plugInId = getSupportedPlugInId(mimeType);
     if (EMPTY_STRING != plugInId) {
         IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);

         if (DRM_NO_ERROR == rDrmEngine.openConvertSession(uniqueId, mConvertId + 1)) {
             ++mConvertId;
             convertId = mConvertId;
             mConvertSessionMap.add(convertId, &rDrmEngine);
         }
     }
     return convertId;
 }

 DrmConvertedStatus* DrmManager::convertData(
             int uniqueId, int convertId, const DrmBuffer* inputData) {
     DrmConvertedStatus *drmConvertedStatus = NULL;

     Mutex::Autolock _l(mConvertLock);
     if (mConvertSessionMap.indexOfKey(convertId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mConvertSessionMap.valueFor(convertId);
         drmConvertedStatus = drmEngine->convertData(uniqueId, convertId, inputData);
     }
     return drmConvertedStatus;
 }

 DrmConvertedStatus* DrmManager::closeConvertSession(int uniqueId, int convertId) {
     Mutex::Autolock _l(mConvertLock);
     DrmConvertedStatus *drmConvertedStatus = NULL;

     if (mConvertSessionMap.indexOfKey(convertId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mConvertSessionMap.valueFor(convertId);
         drmConvertedStatus = drmEngine->closeConvertSession(uniqueId, convertId);
         mConvertSessionMap.removeItem(convertId);
     }
     return drmConvertedStatus;
 }

 status_t DrmManager::getAllSupportInfo(
                     int uniqueId, int* length, DrmSupportInfo** drmSupportInfoArray) {
     Mutex::Autolock _l(mLock);
     Vector<String8> plugInPathList = mPlugInManager.getPlugInIdList();
     int size = plugInPathList.size();
     int validPlugins = 0;

     if (0 < size) {
         Vector<DrmSupportInfo> drmSupportInfoList;

         for (int i = 0; i < size; ++i) {
             String8 plugInPath = plugInPathList[i];
             DrmSupportInfo* drmSupportInfo
                 = mPlugInManager.getPlugIn(plugInPath).getSupportInfo(0);
             if (NULL != drmSupportInfo) {
                 drmSupportInfoList.add(*drmSupportInfo);
                 delete drmSupportInfo; drmSupportInfo = NULL;
             }
         }

         validPlugins = drmSupportInfoList.size();
         if (0 < validPlugins) {
             *drmSupportInfoArray = new DrmSupportInfo[validPlugins];
             for (int i = 0; i < validPlugins; ++i) {
                 (*drmSupportInfoArray)[i] = drmSupportInfoList[i];
             }
         }
     }
     *length = validPlugins;
     return DRM_NO_ERROR;
 }

 DecryptHandle* DrmManager::openDecryptSession(int uniqueId, int fd, off64_t offset, off64_t length) {
     Mutex::Autolock _l(mDecryptLock);
     status_t result = DRM_ERROR_CANNOT_HANDLE;
     Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();

     DecryptHandle* handle = new DecryptHandle();
     if (NULL != handle) {
         handle->decryptId = mDecryptSessionId + 1;

         for (unsigned int index = 0; index < plugInIdList.size(); index++) {
             String8 plugInId = plugInIdList.itemAt(index);
             IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
             result = rDrmEngine.openDecryptSession(uniqueId, handle, fd, offset, length);

             if (DRM_NO_ERROR == result) {
                 ++mDecryptSessionId;
                 mDecryptSessionMap.add(mDecryptSessionId, &rDrmEngine);
                 break;
             }
         }
     }
     if (DRM_NO_ERROR != result) {
         delete handle; handle = NULL;
     }
     return handle;
 }

 DecryptHandle* DrmManager::openDecryptSession(int uniqueId, const char* uri) {
     Mutex::Autolock _l(mDecryptLock);
     status_t result = DRM_ERROR_CANNOT_HANDLE;
     Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();

     DecryptHandle* handle = new DecryptHandle();
     if (NULL != handle) {
         handle->decryptId = mDecryptSessionId + 1;

         for (unsigned int index = 0; index < plugInIdList.size(); index++) {
             String8 plugInId = plugInIdList.itemAt(index);
             IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
             result = rDrmEngine.openDecryptSession(uniqueId, handle, uri);

             if (DRM_NO_ERROR == result) {
                 ++mDecryptSessionId;
                 mDecryptSessionMap.add(mDecryptSessionId, &rDrmEngine);
                 break;
             }
         }
     }
     if (DRM_NO_ERROR != result) {
         delete handle; handle = NULL;
         ALOGV("DrmManager::openDecryptSession: no capable plug-in found");
     }
     return handle;
 }

 status_t DrmManager::closeDecryptSession(int uniqueId, DecryptHandle* decryptHandle) {
     Mutex::Autolock _l(mDecryptLock);
     status_t result = DRM_ERROR_UNKNOWN;
     if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
         result = drmEngine->closeDecryptSession(uniqueId, decryptHandle);
         if (DRM_NO_ERROR == result) {
             mDecryptSessionMap.removeItem(decryptHandle->decryptId);
         }
     }
     return result;
 }

 status_t DrmManager::initializeDecryptUnit(
     int uniqueId, DecryptHandle* decryptHandle, int decryptUnitId, const DrmBuffer* headerInfo) {
     status_t result = DRM_ERROR_UNKNOWN;
     Mutex::Autolock _l(mDecryptLock);
     if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
         result = drmEngine->initializeDecryptUnit(uniqueId, decryptHandle, decryptUnitId, headerInfo);
     }
     return result;
 }

 status_t DrmManager::decrypt(int uniqueId, DecryptHandle* decryptHandle, int decryptUnitId,
             const DrmBuffer* encBuffer, DrmBuffer** decBuffer, DrmBuffer* IV) {
     status_t result = DRM_ERROR_UNKNOWN;

     Mutex::Autolock _l(mDecryptLock);
     if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
         result = drmEngine->decrypt(
                 uniqueId, decryptHandle, decryptUnitId, encBuffer, decBuffer, IV);
     }
     return result;
 }

 status_t DrmManager::finalizeDecryptUnit(
             int uniqueId, DecryptHandle* decryptHandle, int decryptUnitId) {
     status_t result = DRM_ERROR_UNKNOWN;
     Mutex::Autolock _l(mDecryptLock);
     if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
         result = drmEngine->finalizeDecryptUnit(uniqueId, decryptHandle, decryptUnitId);
     }
     return result;
 }

 ssize_t DrmManager::pread(int uniqueId, DecryptHandle* decryptHandle,
             void* buffer, ssize_t numBytes, off64_t offset) {
     ssize_t result = DECRYPT_FILE_ERROR;

     Mutex::Autolock _l(mDecryptLock);
     if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
         IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
         result = drmEngine->pread(uniqueId, decryptHandle, buffer, numBytes, offset);
     }
     return result;
 }

 String8 DrmManager::getSupportedPlugInId(
             int uniqueId, const String8& path, const String8& mimeType) {
     String8 plugInId("");

     if (EMPTY_STRING != mimeType) {
         plugInId = getSupportedPlugInId(mimeType);
     } else {
         plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
     }
     return plugInId;
 }

 String8 DrmManager::getSupportedPlugInId(const String8& mimeType) {
     String8 plugInId("");

     if (EMPTY_STRING != mimeType) {
         for (unsigned int index = 0; index < mSupportInfoToPlugInIdMap.size(); index++) {
             const DrmSupportInfo& drmSupportInfo = mSupportInfoToPlugInIdMap.keyAt(index);

             if (drmSupportInfo.isSupportedMimeType(mimeType)) {
                 plugInId = mSupportInfoToPlugInIdMap.valueFor(drmSupportInfo);
                 break;
             }
         }
     }
     return plugInId;
 }

 String8 DrmManager::getSupportedPlugInIdFromPath(int uniqueId, const String8& path) {
     String8 plugInId("");
     const String8 fileSuffix = path.getPathExtension();

     for (unsigned int index = 0; index < mSupportInfoToPlugInIdMap.size(); index++) {
         const DrmSupportInfo& drmSupportInfo = mSupportInfoToPlugInIdMap.keyAt(index);

         if (drmSupportInfo.isSupportedFileSuffix(fileSuffix)) {
             String8 key = mSupportInfoToPlugInIdMap.valueFor(drmSupportInfo);
             IDrmEngine& drmEngine = mPlugInManager.getPlugIn(key);

             if (drmEngine.canHandle(uniqueId, path)) {
                 plugInId = key;
                 break;
             }
         }
     }
     return plugInId;
 }

 void DrmManager::onInfo(const DrmInfoEvent& event) {
     Mutex::Autolock _l(mListenerLock);
     for (unsigned int index = 0; index < mServiceListeners.size(); index++) {
         int uniqueId = mServiceListeners.keyAt(index);

         if (uniqueId == event.getUniqueId()) {
             sp<IDrmServiceListener> serviceListener = mServiceListeners.valueFor(uniqueId);
             serviceListener->notify(event);
         }
     }
 }
	/*
	* Copyright (C) 2010 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	//#define LOG_NDEBUG 0
	#define LOG_TAG "DrmManager(Native)"
	#include "utils/Log.h"

	#include <utils/String8.h>
	#include <drm/DrmInfo.h>
	#include <drm/DrmInfoEvent.h>
	#include <drm/DrmRights.h>
	#include <drm/DrmConstraints.h>
	#include <drm/DrmMetadata.h>
	#include <drm/DrmInfoStatus.h>
	#include <drm/DrmInfoRequest.h>
	#include <drm/DrmSupportInfo.h>
	#include <drm/DrmConvertedStatus.h>
	#include <IDrmEngine.h>

	#include "DrmManager.h"
	#include "ReadWriteUtils.h"

	#define DECRYPT_FILE_ERROR -1

	using namespace android;

	const String8 DrmManager::EMPTY_STRING("");

	DrmManager::DrmManager() :
	mDecryptSessionId(0),
	mConvertId(0) {

	}

	DrmManager::~DrmManager() {

	}

	int DrmManager::addUniqueId(bool isNative) {
	Mutex::Autolock _l(mLock);

	int temp = 0;
	bool foundUniqueId = false;
	const int size = mUniqueIdVector.size();
	const int uniqueIdRange = 0xfff;
	int maxLoopTimes = (uniqueIdRange - 1) / 2;
	srand(time(NULL));

	while (!foundUniqueId) {
	temp = rand() & uniqueIdRange;

	if (isNative) {
	// set a flag to differentiate DrmManagerClient
	// created from native side and java side
	temp \|= 0x1000;
	}

	int index = 0;
	for (; index < size; ++index) {
	if (mUniqueIdVector.itemAt(index) == temp) {
	foundUniqueId = false;
	break;
	}
	}
	if (index == size) {
	foundUniqueId = true;
	}

	maxLoopTimes --;
	LOG_FATAL_IF(maxLoopTimes <= 0, "cannot find an unique ID for this session");
	}

	mUniqueIdVector.push(temp);
	return temp;
	}

	void DrmManager::removeUniqueId(int uniqueId) {
	Mutex::Autolock _l(mLock);
	for (unsigned int i = 0; i < mUniqueIdVector.size(); i++) {
	if (uniqueId == mUniqueIdVector.itemAt(i)) {
	mUniqueIdVector.removeAt(i);
	break;
	}
	}
	}

	status_t DrmManager::loadPlugIns() {

	String8 vendorPluginDirPath("/vendor/lib/drm");
	loadPlugIns(vendorPluginDirPath);

	String8 pluginDirPath("/system/lib/drm");
	loadPlugIns(pluginDirPath);
	return DRM_NO_ERROR;

	}

	status_t DrmManager::loadPlugIns(const String8& plugInDirPath) {
	mPlugInManager.loadPlugIns(plugInDirPath);
	Vector<String8> plugInPathList = mPlugInManager.getPlugInIdList();
	for (unsigned int i = 0; i < plugInPathList.size(); ++i) {
	String8 plugInPath = plugInPathList[i];
	DrmSupportInfo* info = mPlugInManager.getPlugIn(plugInPath).getSupportInfo(0);
	if (NULL != info) {
	if (mSupportInfoToPlugInIdMap.indexOfKey(*info) < 0) {
	mSupportInfoToPlugInIdMap.add(*info, plugInPath);
	}
	delete info;
	}
	}
	return DRM_NO_ERROR;
	}

	status_t DrmManager::unloadPlugIns() {
	Mutex::Autolock _l(mLock);
	mConvertSessionMap.clear();
	mDecryptSessionMap.clear();
	mPlugInManager.unloadPlugIns();
	mSupportInfoToPlugInIdMap.clear();
	return DRM_NO_ERROR;
	}

	status_t DrmManager::setDrmServiceListener(
	int uniqueId, const sp<IDrmServiceListener>& drmServiceListener) {
	Mutex::Autolock _l(mListenerLock);
	if (NULL != drmServiceListener.get()) {
	mServiceListeners.add(uniqueId, drmServiceListener);
	} else {
	mServiceListeners.removeItem(uniqueId);
	}
	return DRM_NO_ERROR;
	}

	void DrmManager::addClient(int uniqueId) {
	Mutex::Autolock _l(mLock);
	if (!mSupportInfoToPlugInIdMap.isEmpty()) {
	Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();
	for (unsigned int index = 0; index < plugInIdList.size(); index++) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInIdList.itemAt(index));
	rDrmEngine.initialize(uniqueId);
	rDrmEngine.setOnInfoListener(uniqueId, this);
	}
	}
	}

	void DrmManager::removeClient(int uniqueId) {
	Mutex::Autolock _l(mLock);
	Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();
	for (unsigned int index = 0; index < plugInIdList.size(); index++) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInIdList.itemAt(index));
	rDrmEngine.terminate(uniqueId);
	}
	}

	DrmConstraints* DrmManager::getConstraints(int uniqueId, const String8* path, const int action) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, *path);
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	return rDrmEngine.getConstraints(uniqueId, path, action);
	}
	return NULL;
	}

	DrmMetadata* DrmManager::getMetadata(int uniqueId, const String8* path) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, *path);
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	return rDrmEngine.getMetadata(uniqueId, path);
	}
	return NULL;
	}

	status_t DrmManager::installDrmEngine(int uniqueId, const String8& absolutePath) {
	Mutex::Autolock _l(mLock);
	mPlugInManager.loadPlugIn(absolutePath);

	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(absolutePath);
	rDrmEngine.initialize(uniqueId);
	rDrmEngine.setOnInfoListener(uniqueId, this);

	DrmSupportInfo* info = rDrmEngine.getSupportInfo(0);
	mSupportInfoToPlugInIdMap.add(*info, absolutePath);
	delete info;

	return DRM_NO_ERROR;
	}

	bool DrmManager::canHandle(int uniqueId, const String8& path, const String8& mimeType) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInId(mimeType);
	bool result = (EMPTY_STRING != plugInId) ? true : false;

	if (0 < path.length()) {
	if (result) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	result = rDrmEngine.canHandle(uniqueId, path);
	} else {
	String8 extension = path.getPathExtension();
	if (String8("") != extension) {
	result = canHandle(uniqueId, path);
	}
	}
	}
	return result;
	}

	DrmInfoStatus* DrmManager::processDrmInfo(int uniqueId, const DrmInfo* drmInfo) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInId(drmInfo->getMimeType());
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	return rDrmEngine.processDrmInfo(uniqueId, drmInfo);
	}
	return NULL;
	}

	bool DrmManager::canHandle(int uniqueId, const String8& path) {
	bool result = false;
	Vector<String8> plugInPathList = mPlugInManager.getPlugInIdList();

	for (unsigned int i = 0; i < plugInPathList.size(); ++i) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInPathList[i]);
	result = rDrmEngine.canHandle(uniqueId, path);

	if (result) {
	break;
	}
	}
	return result;
	}

	DrmInfo* DrmManager::acquireDrmInfo(int uniqueId, const DrmInfoRequest* drmInfoRequest) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInId(drmInfoRequest->getMimeType());
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	return rDrmEngine.acquireDrmInfo(uniqueId, drmInfoRequest);
	}
	return NULL;
	}

	status_t DrmManager::saveRights(int uniqueId, const DrmRights& drmRights,
	const String8& rightsPath, const String8& contentPath) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInId(drmRights.getMimeType());
	status_t result = DRM_ERROR_UNKNOWN;
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	result = rDrmEngine.saveRights(uniqueId, drmRights, rightsPath, contentPath);
	}
	return result;
	}

	String8 DrmManager::getOriginalMimeType(int uniqueId, const String8& path) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	return rDrmEngine.getOriginalMimeType(uniqueId, path);
	}
	return EMPTY_STRING;
	}

	int DrmManager::getDrmObjectType(int uniqueId, const String8& path, const String8& mimeType) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInId(uniqueId, path, mimeType);
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	return rDrmEngine.getDrmObjectType(uniqueId, path, mimeType);
	}
	return DrmObjectType::UNKNOWN;
	}

	int DrmManager::checkRightsStatus(int uniqueId, const String8& path, int action) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	return rDrmEngine.checkRightsStatus(uniqueId, path, action);
	}
	return RightsStatus::RIGHTS_INVALID;
	}

	status_t DrmManager::consumeRights(
	int uniqueId, DecryptHandle* decryptHandle, int action, bool reserve) {
	status_t result = DRM_ERROR_UNKNOWN;
	Mutex::Autolock _l(mDecryptLock);
	if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
	result = drmEngine->consumeRights(uniqueId, decryptHandle, action, reserve);
	}
	return result;
	}

	status_t DrmManager::setPlaybackStatus(
	int uniqueId, DecryptHandle* decryptHandle, int playbackStatus, int64_t position) {
	status_t result = DRM_ERROR_UNKNOWN;
	Mutex::Autolock _l(mDecryptLock);
	if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
	result = drmEngine->setPlaybackStatus(uniqueId, decryptHandle, playbackStatus, position);
	}
	return result;
	}

	bool DrmManager::validateAction(
	int uniqueId, const String8& path, int action, const ActionDescription& description) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	return rDrmEngine.validateAction(uniqueId, path, action, description);
	}
	return false;
	}

	status_t DrmManager::removeRights(int uniqueId, const String8& path) {
	Mutex::Autolock _l(mLock);
	const String8 plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
	status_t result = DRM_ERROR_UNKNOWN;
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	result = rDrmEngine.removeRights(uniqueId, path);
	}
	return result;
	}

	status_t DrmManager::removeAllRights(int uniqueId) {
	Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();
	status_t result = DRM_ERROR_UNKNOWN;
	for (unsigned int index = 0; index < plugInIdList.size(); index++) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInIdList.itemAt(index));
	result = rDrmEngine.removeAllRights(uniqueId);
	if (DRM_NO_ERROR != result) {
	break;
	}
	}
	return result;
	}

	int DrmManager::openConvertSession(int uniqueId, const String8& mimeType) {
	Mutex::Autolock _l(mConvertLock);
	int convertId = -1;

	const String8 plugInId = getSupportedPlugInId(mimeType);
	if (EMPTY_STRING != plugInId) {
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);

	if (DRM_NO_ERROR == rDrmEngine.openConvertSession(uniqueId, mConvertId + 1)) {
	++mConvertId;
	convertId = mConvertId;
	mConvertSessionMap.add(convertId, &rDrmEngine);
	}
	}
	return convertId;
	}

	DrmConvertedStatus* DrmManager::convertData(
	int uniqueId, int convertId, const DrmBuffer* inputData) {
	DrmConvertedStatus *drmConvertedStatus = NULL;

	Mutex::Autolock _l(mConvertLock);
	if (mConvertSessionMap.indexOfKey(convertId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mConvertSessionMap.valueFor(convertId);
	drmConvertedStatus = drmEngine->convertData(uniqueId, convertId, inputData);
	}
	return drmConvertedStatus;
	}

	DrmConvertedStatus* DrmManager::closeConvertSession(int uniqueId, int convertId) {
	Mutex::Autolock _l(mConvertLock);
	DrmConvertedStatus *drmConvertedStatus = NULL;

	if (mConvertSessionMap.indexOfKey(convertId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mConvertSessionMap.valueFor(convertId);
	drmConvertedStatus = drmEngine->closeConvertSession(uniqueId, convertId);
	mConvertSessionMap.removeItem(convertId);
	}
	return drmConvertedStatus;
	}

	status_t DrmManager::getAllSupportInfo(
	int uniqueId, int* length, DrmSupportInfo** drmSupportInfoArray) {
	Mutex::Autolock _l(mLock);
	Vector<String8> plugInPathList = mPlugInManager.getPlugInIdList();
	int size = plugInPathList.size();
	int validPlugins = 0;

	if (0 < size) {
	Vector<DrmSupportInfo> drmSupportInfoList;

	for (int i = 0; i < size; ++i) {
	String8 plugInPath = plugInPathList[i];
	DrmSupportInfo* drmSupportInfo
	= mPlugInManager.getPlugIn(plugInPath).getSupportInfo(0);
	if (NULL != drmSupportInfo) {
	drmSupportInfoList.add(*drmSupportInfo);
	delete drmSupportInfo; drmSupportInfo = NULL;
	}
	}

	validPlugins = drmSupportInfoList.size();
	if (0 < validPlugins) {
	*drmSupportInfoArray = new DrmSupportInfo[validPlugins];
	for (int i = 0; i < validPlugins; ++i) {
	(*drmSupportInfoArray)[i] = drmSupportInfoList[i];
	}
	}
	}
	*length = validPlugins;
	return DRM_NO_ERROR;
	}

	DecryptHandle* DrmManager::openDecryptSession(int uniqueId, int fd, off64_t offset, off64_t length) {
	Mutex::Autolock _l(mDecryptLock);
	status_t result = DRM_ERROR_CANNOT_HANDLE;
	Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();

	DecryptHandle* handle = new DecryptHandle();
	if (NULL != handle) {
	handle->decryptId = mDecryptSessionId + 1;

	for (unsigned int index = 0; index < plugInIdList.size(); index++) {
	String8 plugInId = plugInIdList.itemAt(index);
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	result = rDrmEngine.openDecryptSession(uniqueId, handle, fd, offset, length);

	if (DRM_NO_ERROR == result) {
	++mDecryptSessionId;
	mDecryptSessionMap.add(mDecryptSessionId, &rDrmEngine);
	break;
	}
	}
	}
	if (DRM_NO_ERROR != result) {
	delete handle; handle = NULL;
	}
	return handle;
	}

	DecryptHandle* DrmManager::openDecryptSession(int uniqueId, const char* uri) {
	Mutex::Autolock _l(mDecryptLock);
	status_t result = DRM_ERROR_CANNOT_HANDLE;
	Vector<String8> plugInIdList = mPlugInManager.getPlugInIdList();

	DecryptHandle* handle = new DecryptHandle();
	if (NULL != handle) {
	handle->decryptId = mDecryptSessionId + 1;

	for (unsigned int index = 0; index < plugInIdList.size(); index++) {
	String8 plugInId = plugInIdList.itemAt(index);
	IDrmEngine& rDrmEngine = mPlugInManager.getPlugIn(plugInId);
	result = rDrmEngine.openDecryptSession(uniqueId, handle, uri);

	if (DRM_NO_ERROR == result) {
	++mDecryptSessionId;
	mDecryptSessionMap.add(mDecryptSessionId, &rDrmEngine);
	break;
	}
	}
	}
	if (DRM_NO_ERROR != result) {
	delete handle; handle = NULL;
	ALOGV("DrmManager::openDecryptSession: no capable plug-in found");
	}
	return handle;
	}

	status_t DrmManager::closeDecryptSession(int uniqueId, DecryptHandle* decryptHandle) {
	Mutex::Autolock _l(mDecryptLock);
	status_t result = DRM_ERROR_UNKNOWN;
	if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
	result = drmEngine->closeDecryptSession(uniqueId, decryptHandle);
	if (DRM_NO_ERROR == result) {
	mDecryptSessionMap.removeItem(decryptHandle->decryptId);
	}
	}
	return result;
	}

	status_t DrmManager::initializeDecryptUnit(
	int uniqueId, DecryptHandle* decryptHandle, int decryptUnitId, const DrmBuffer* headerInfo) {
	status_t result = DRM_ERROR_UNKNOWN;
	Mutex::Autolock _l(mDecryptLock);
	if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
	result = drmEngine->initializeDecryptUnit(uniqueId, decryptHandle, decryptUnitId, headerInfo);
	}
	return result;
	}

	status_t DrmManager::decrypt(int uniqueId, DecryptHandle* decryptHandle, int decryptUnitId,
	const DrmBuffer* encBuffer, DrmBuffer** decBuffer, DrmBuffer* IV) {
	status_t result = DRM_ERROR_UNKNOWN;

	Mutex::Autolock _l(mDecryptLock);
	if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
	result = drmEngine->decrypt(
	uniqueId, decryptHandle, decryptUnitId, encBuffer, decBuffer, IV);
	}
	return result;
	}

	status_t DrmManager::finalizeDecryptUnit(
	int uniqueId, DecryptHandle* decryptHandle, int decryptUnitId) {
	status_t result = DRM_ERROR_UNKNOWN;
	Mutex::Autolock _l(mDecryptLock);
	if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
	result = drmEngine->finalizeDecryptUnit(uniqueId, decryptHandle, decryptUnitId);
	}
	return result;
	}

	ssize_t DrmManager::pread(int uniqueId, DecryptHandle* decryptHandle,
	void* buffer, ssize_t numBytes, off64_t offset) {
	ssize_t result = DECRYPT_FILE_ERROR;

	Mutex::Autolock _l(mDecryptLock);
	if (mDecryptSessionMap.indexOfKey(decryptHandle->decryptId) != NAME_NOT_FOUND) {
	IDrmEngine* drmEngine = mDecryptSessionMap.valueFor(decryptHandle->decryptId);
	result = drmEngine->pread(uniqueId, decryptHandle, buffer, numBytes, offset);
	}
	return result;
	}

	String8 DrmManager::getSupportedPlugInId(
	int uniqueId, const String8& path, const String8& mimeType) {
	String8 plugInId("");

	if (EMPTY_STRING != mimeType) {
	plugInId = getSupportedPlugInId(mimeType);
	} else {
	plugInId = getSupportedPlugInIdFromPath(uniqueId, path);
	}
	return plugInId;
	}

	String8 DrmManager::getSupportedPlugInId(const String8& mimeType) {
	String8 plugInId("");

	if (EMPTY_STRING != mimeType) {
	for (unsigned int index = 0; index < mSupportInfoToPlugInIdMap.size(); index++) {
	const DrmSupportInfo& drmSupportInfo = mSupportInfoToPlugInIdMap.keyAt(index);

	if (drmSupportInfo.isSupportedMimeType(mimeType)) {
	plugInId = mSupportInfoToPlugInIdMap.valueFor(drmSupportInfo);
	break;
	}
	}
	}
	return plugInId;
	}

	String8 DrmManager::getSupportedPlugInIdFromPath(int uniqueId, const String8& path) {
	String8 plugInId("");
	const String8 fileSuffix = path.getPathExtension();

	for (unsigned int index = 0; index < mSupportInfoToPlugInIdMap.size(); index++) {
	const DrmSupportInfo& drmSupportInfo = mSupportInfoToPlugInIdMap.keyAt(index);

	if (drmSupportInfo.isSupportedFileSuffix(fileSuffix)) {
	String8 key = mSupportInfoToPlugInIdMap.valueFor(drmSupportInfo);
	IDrmEngine& drmEngine = mPlugInManager.getPlugIn(key);

	if (drmEngine.canHandle(uniqueId, path)) {
	plugInId = key;
	break;
	}
	}
	}
	return plugInId;
	}

	void DrmManager::onInfo(const DrmInfoEvent& event) {
	Mutex::Autolock _l(mListenerLock);
	for (unsigned int index = 0; index < mServiceListeners.size(); index++) {
	int uniqueId = mServiceListeners.keyAt(index);

	if (uniqueId == event.getUniqueId()) {
	sp<IDrmServiceListener> serviceListener = mServiceListeners.valueFor(uniqueId);
	serviceListener->notify(event);
	}
	}
	}