commit 283eb4fc79bdbb47529e97ea9e3626134413304f
author Kyle Zhang <kelzhan@google.com> Sat Mar 30 05:55:04 2024 +0000
committer Kyle Zhang <kelzhan@google.com> Tue Apr 02 05:04:30 2024 +0000
tree 73af0c3e32722b67377ef2f3a6afe4fd2088390b
parent e2420d26e06318a6c1d212416ece5869401da199

Add crypto scheme check when create drm plugin

Bug: 330819085
Change-Id: Ica19e221e7ad5e939396cacafeb53f970855ca2f

drm/libmediadrm/DrmHalAidl.cpp

diff --git a/drm/libmediadrm/DrmHalAidl.cpp b/drm/libmediadrm/DrmHalAidl.cpp
index 7106d66..650a589 100644
--- a/drm/libmediadrm/DrmHalAidl.cpp
+++ b/drm/libmediadrm/DrmHalAidl.cpp
@@ -466,6 +466,12 @@
     mMetrics->SetAppPackageName(appPackageName);
     mMetrics->SetAppUid(AIBinder_getCallingUid());
     for (ssize_t i = mFactories.size() - 1; i >= 0; i--) {
+        CryptoSchemes schemes{};
+        auto err = mFactories[i]->getSupportedCryptoSchemes(&schemes);
+        if (!err.isOk() || !std::count(schemes.uuids.begin(), schemes.uuids.end(), uuidAidl)) {
+            continue;
+        }
+
         ::ndk::ScopedAStatus status =
                 mFactories[i]->createDrmPlugin(uuidAidl, appPackageNameAidl, &pluginAidl);
         if (status.isOk()) {