1 /*
2  * Copyright (C) 2010 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef __DRM_MANAGER_H__
18 #define __DRM_MANAGER_H__
19 
20 #include <utils/Errors.h>
21 #include <utils/threads.h>
22 #include <drm/drm_framework_common.h>
23 #include "IDrmEngine.h"
24 #include "PlugInManager.h"
25 #include "IDrmServiceListener.h"
26 
27 namespace android {
28 
29 class IDrmManager;
30 class DrmRegistrationInfo;
31 class DrmUnregistrationInfo;
32 class DrmRightsAcquisitionInfo;
33 class DrmConstraints;
34 class DrmMetadata;
35 class DrmRights;
36 class DrmInfo;
37 class DrmInfoStatus;
38 class DrmConvertedStatus;
39 class DrmInfoRequest;
40 class DrmSupportInfo;
41 class ActionDescription;
42 
43 /**
44  * This is implementation class for DRM Manager. This class delegates the
45  * functionality to corresponding DRM Engine.
46  *
47  * The DrmManagerService class creates an instance of this class.
48  *
49  */
50 class DrmManager : public IDrmEngine::OnInfoListener {
51 public:
52     DrmManager();
53     virtual ~DrmManager();
54 
55 public:
56     int addUniqueId(bool isNative);
57 
58     void removeUniqueId(int uniqueId);
59 
60     void addClient(int uniqueId);
61 
62     void removeClient(int uniqueId);
63 
64     status_t loadPlugIns();
65 
66     status_t loadPlugIns(const String8& plugInDirPath);
67 
68     status_t unloadPlugIns();
69 
70     status_t setDrmServiceListener(
71             int uniqueId, const sp<IDrmServiceListener>& drmServiceListener);
72 
73     DrmConstraints* getConstraints(int uniqueId, const String8* path, const int action);
74 
75     DrmMetadata* getMetadata(int uniqueId, const String8* path);
76 
77     bool canHandle(int uniqueId, const String8& path, const String8& mimeType);
78 
79     DrmInfoStatus* processDrmInfo(int uniqueId, const DrmInfo* drmInfo);
80 
81     DrmInfo* acquireDrmInfo(int uniqueId, const DrmInfoRequest* drmInfoRequest);
82 
83     status_t saveRights(int uniqueId, const DrmRights& drmRights,
84             const String8& rightsPath, const String8& contentPath);
85 
86     String8 getOriginalMimeType(int uniqueId, const String8& path, int fd);
87 
88     int getDrmObjectType(int uniqueId, const String8& path, const String8& mimeType);
89 
90     int checkRightsStatus(int uniqueId, const String8& path, int action);
91 
92     status_t consumeRights(int uniqueId, sp<DecryptHandle>& decryptHandle, int action,
93             bool reserve);
94 
95     status_t setPlaybackStatus(
96             int uniqueId, sp<DecryptHandle>& decryptHandle, int playbackStatus, int64_t position);
97 
98     bool validateAction(
99             int uniqueId, const String8& path, int action, const ActionDescription& description);
100 
101     status_t removeRights(int uniqueId, const String8& path);
102 
103     status_t removeAllRights(int uniqueId);
104 
105     int openConvertSession(int uniqueId, const String8& mimeType);
106 
107     DrmConvertedStatus* convertData(int uniqueId, int convertId, const DrmBuffer* inputData);
108 
109     DrmConvertedStatus* closeConvertSession(int uniqueId, int convertId);
110 
111     status_t getAllSupportInfo(int uniqueId, int* length, DrmSupportInfo** drmSupportInfoArray);
112 
113     sp<DecryptHandle> openDecryptSession(
114             int uniqueId, int fd, off64_t offset, off64_t length, const char* mime);
115 
116     sp<DecryptHandle> openDecryptSession(int uniqueId, const char* uri, const char* mime);
117 
118     sp<DecryptHandle> openDecryptSession(int uniqueId, const DrmBuffer& buf,
119             const String8& mimeType);
120 
121     status_t closeDecryptSession(int uniqueId, sp<DecryptHandle>& decryptHandle);
122 
123     status_t initializeDecryptUnit(int uniqueId, sp<DecryptHandle>& decryptHandle,
124             int decryptUnitId, const DrmBuffer* headerInfo);
125 
126     status_t decrypt(int uniqueId, sp<DecryptHandle>& decryptHandle, int decryptUnitId,
127             const DrmBuffer* encBuffer, DrmBuffer** decBuffer, DrmBuffer* IV);
128 
129     status_t finalizeDecryptUnit(int uniqueId, sp<DecryptHandle>& decryptHandle,
130             int decryptUnitId);
131 
132     ssize_t pread(int uniqueId, sp<DecryptHandle>& decryptHandle,
133             void* buffer, ssize_t numBytes, off64_t offset);
134 
135     void onInfo(const DrmInfoEvent& event);
136 
137 private:
138     String8 getSupportedPlugInId(int uniqueId, const String8& path, const String8& mimeType);
139 
140     String8 getSupportedPlugInId(const String8& mimeType);
141 
142     String8 getSupportedPlugInIdFromPath(int uniqueId, const String8& path);
143 
144     bool canHandle(int uniqueId, const String8& path);
145 
146 private:
147     enum {
148         kMaxNumUniqueIds = 0x1000,
149     };
150 
151     bool mUniqueIdArray[kMaxNumUniqueIds];
152     static const String8 EMPTY_STRING;
153 
154     int mDecryptSessionId;
155     int mConvertId;
156     Mutex mLock;
157     Mutex mListenerLock;
158     Mutex mDecryptLock;
159     Mutex mConvertLock;
160     TPlugInManager<IDrmEngine> mPlugInManager;
161     KeyedVector< DrmSupportInfo, String8 > mSupportInfoToPlugInIdMap;
162     KeyedVector< int, IDrmEngine*> mConvertSessionMap;
163     KeyedVector< int, sp<IDrmServiceListener> > mServiceListeners;
164     KeyedVector< int, IDrmEngine*> mDecryptSessionMap;
165 };
166 
167 };
168 
169 #endif /* __DRM_MANAGER_H__ */
170 
171