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
27namespace android {
28
29class IDrmManager;
30class DrmRegistrationInfo;
31class DrmUnregistrationInfo;
32class DrmRightsAcquisitionInfo;
33class DrmConstraints;
34class DrmMetadata;
35class DrmRights;
36class DrmInfo;
37class DrmInfoStatus;
38class DrmConvertedStatus;
39class DrmInfoRequest;
40class DrmSupportInfo;
41class 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 */
50class DrmManager : public IDrmEngine::OnInfoListener {
51public:
52    DrmManager();
53    virtual ~DrmManager();
54
55public:
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
135private:
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
144private:
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