MockDrmCryptoPlugin.h revision 441a78d5e224e0d67f9b52fa9adc795c6944159b 
1/*
2 * Copyright (C) 2013 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#include <utils/Mutex.h>
18
19#include "drm/DrmAPI.h"
20#include "hardware/CryptoAPI.h"
21
22extern "C" {
23      android::DrmFactory *createDrmFactory();
24      android::CryptoFactory *createCryptoFactory();
25}
26
27namespace android {
28
29    class MockDrmFactory : public DrmFactory {
30    public:
31        MockDrmFactory() {}
32        virtual ~MockDrmFactory() {}
33
34        bool isCryptoSchemeSupported(const uint8_t uuid[16]);
35        status_t createDrmPlugin(const uint8_t uuid[16], DrmPlugin **plugin);
36    };
37
38    class MockCryptoFactory : public CryptoFactory {
39    public:
40        MockCryptoFactory() {}
41        virtual ~MockCryptoFactory() {}
42
43        bool isCryptoSchemeSupported(const uint8_t uuid[16]) const;
44        status_t createPlugin(
45            const uint8_t uuid[16], const void *data, size_t size,
46            CryptoPlugin **plugin);
47    };
48
49
50
51    class MockDrmPlugin : public DrmPlugin {
52    public:
53        MockDrmPlugin() {}
54        virtual ~MockDrmPlugin() {}
55
56        // from DrmPlugin
57        status_t openSession(Vector<uint8_t> &sessionId);
58        status_t closeSession(Vector<uint8_t> const &sessionId);
59
60        status_t
61            getLicenseRequest(Vector<uint8_t> const &sessionId,
62                              Vector<uint8_t> const &initData,
63                              String8 const &mimeType, LicenseType licenseType,
64                              KeyedVector<String8, String8> const &optionalParameters,
65                              Vector<uint8_t> &request, String8 &defaultUrl);
66
67        status_t provideLicenseResponse(Vector<uint8_t> const &sessionId,
68                                                Vector<uint8_t> const &response);
69
70        status_t removeLicense(Vector<uint8_t> const &sessionId);
71
72        status_t
73            queryLicenseStatus(Vector<uint8_t> const &sessionId,
74                               KeyedVector<String8, String8> &infoMap) const;
75
76        status_t getProvisionRequest(Vector<uint8_t> &request,
77                                             String8 &defaultUrl);
78
79        status_t provideProvisionResponse(Vector<uint8_t> const &response);
80
81        status_t getSecureStops(List<Vector<uint8_t> > &secureStops);
82        status_t releaseSecureStops(Vector<uint8_t> const &ssRelease);
83
84        status_t getPropertyString(String8 const &name, String8 &value ) const;
85        status_t getPropertyByteArray(String8 const &name,
86                                              Vector<uint8_t> &value ) const;
87
88        status_t setPropertyString(String8 const &name,
89                                   String8 const &value );
90        status_t setPropertyByteArray(String8 const &name,
91                                      Vector<uint8_t> const &value );
92
93    private:
94        String8 vectorToString(Vector<uint8_t> const &vector) const;
95        String8 arrayToString(uint8_t const *array, size_t len) const;
96        String8 stringMapToString(KeyedVector<String8, String8> map) const;
97
98        SortedVector<Vector<uint8_t> > mSessions;
99
100        static const ssize_t kNotFound = -1;
101        ssize_t findSession(Vector<uint8_t> const &sessionId) const;
102
103        Mutex mLock;
104        KeyedVector<String8, String8> mStringProperties;
105        KeyedVector<String8, Vector<uint8_t> > mByteArrayProperties;
106    };
107
108
109    class MockCryptoPlugin : public CryptoPlugin {
110
111        bool requiresSecureDecoderComponent(const char *mime) const;
112
113        ssize_t decrypt(bool secure,
114            const uint8_t key[16], const uint8_t iv[16],
115            Mode mode, const void *srcPtr,
116            const SubSample *subSamples, size_t numSubSamples,
117            void *dstPtr, AString *errorDetailMsg);
118    private:
119        String8 subSamplesToString(CryptoPlugin::SubSample const *subSamples, size_t numSubSamples) const;
120        String8 arrayToString(uint8_t const *array, size_t len) const;
121    };
122};
123