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, DecryptHandle* decryptHandle, int action, bool reserve);
93 
94     status_t setPlaybackStatus(
95             int uniqueId, DecryptHandle* decryptHandle, int playbackStatus, int64_t position);
96 
97     bool validateAction(
98             int uniqueId, const String8& path, int action, const ActionDescription& description);
99 
100     status_t removeRights(int uniqueId, const String8& path);
101 
102     status_t removeAllRights(int uniqueId);
103 
104     int openConvertSession(int uniqueId, const String8& mimeType);
105 
106     DrmConvertedStatus* convertData(int uniqueId, int convertId, const DrmBuffer* inputData);
107 
108     DrmConvertedStatus* closeConvertSession(int uniqueId, int convertId);
109 
110     status_t getAllSupportInfo(int uniqueId, int* length, DrmSupportInfo** drmSupportInfoArray);
111 
112     DecryptHandle* openDecryptSession(
113             int uniqueId, int fd, off64_t offset, off64_t length, const char* mime);
114 
115     DecryptHandle* openDecryptSession(int uniqueId, const char* uri, const char* mime);
116 
117     DecryptHandle* openDecryptSession(int uniqueId, const DrmBuffer& buf,
118             const String8& mimeType);
119 
120     status_t closeDecryptSession(int uniqueId, DecryptHandle* decryptHandle);
121 
122     status_t initializeDecryptUnit(int uniqueId, DecryptHandle* decryptHandle,
123             int decryptUnitId, const DrmBuffer* headerInfo);
124 
125     status_t decrypt(int uniqueId, DecryptHandle* decryptHandle, int decryptUnitId,
126             const DrmBuffer* encBuffer, DrmBuffer** decBuffer, DrmBuffer* IV);
127 
128     status_t finalizeDecryptUnit(int uniqueId, DecryptHandle* decryptHandle, int decryptUnitId);
129 
130     ssize_t pread(int uniqueId, DecryptHandle* decryptHandle,
131             void* buffer, ssize_t numBytes, off64_t offset);
132 
133     void onInfo(const DrmInfoEvent& event);
134 
135 private:
136     String8 getSupportedPlugInId(int uniqueId, const String8& path, const String8& mimeType);
137 
138     String8 getSupportedPlugInId(const String8& mimeType);
139 
140     String8 getSupportedPlugInIdFromPath(int uniqueId, const String8& path);
141 
142     bool canHandle(int uniqueId, const String8& path);
143 
144 private:
145     enum {
146         kMaxNumUniqueIds = 0x1000,
147     };
148 
149     bool mUniqueIdArray[kMaxNumUniqueIds];
150     static const String8 EMPTY_STRING;
151 
152     int mDecryptSessionId;
153     int mConvertId;
154     Mutex mLock;
155     Mutex mListenerLock;
156     Mutex mDecryptLock;
157     Mutex mConvertLock;
158     TPlugInManager<IDrmEngine> mPlugInManager;
159     KeyedVector< DrmSupportInfo, String8 > mSupportInfoToPlugInIdMap;
160     KeyedVector< int, IDrmEngine*> mConvertSessionMap;
161     KeyedVector< int, sp<IDrmServiceListener> > mServiceListeners;
162     KeyedVector< int, IDrmEngine*> mDecryptSessionMap;
163 };
164 
165 };
166 
167 #endif /* __DRM_MANAGER_H__ */
168 
169