MockDrmCryptoPlugin.h revision 68d9d71a792deed75d32fe13febc07c9c12c8449
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 bool isContentTypeSupported(const String8 &mimeType); 36 status_t createDrmPlugin(const uint8_t uuid[16], DrmPlugin **plugin); 37 }; 38 39 class MockCryptoFactory : public CryptoFactory { 40 public: 41 MockCryptoFactory() {} 42 virtual ~MockCryptoFactory() {} 43 44 bool isCryptoSchemeSupported(const uint8_t uuid[16]) const; 45 status_t createPlugin( 46 const uint8_t uuid[16], const void *data, size_t size, 47 CryptoPlugin **plugin); 48 }; 49 50 51 52 class MockDrmPlugin : public DrmPlugin { 53 public: 54 MockDrmPlugin() {} 55 virtual ~MockDrmPlugin() {} 56 57 // from DrmPlugin 58 status_t openSession(Vector<uint8_t> &sessionId); 59 status_t closeSession(Vector<uint8_t> const &sessionId); 60 61 status_t getKeyRequest(Vector<uint8_t> const &sessionId, 62 Vector<uint8_t> const &initData, 63 String8 const &mimeType, KeyType keyType, 64 KeyedVector<String8, String8> const &optionalParameters, 65 Vector<uint8_t> &request, String8 &defaultUrl); 66 67 status_t provideKeyResponse(Vector<uint8_t> const &sessionId, 68 Vector<uint8_t> const &response, 69 Vector<uint8_t> &keySetId); 70 71 status_t removeKeys(Vector<uint8_t> const &keySetId); 72 73 status_t restoreKeys(Vector<uint8_t> const &sessionId, 74 Vector<uint8_t> const &keySetId); 75 76 status_t queryKeyStatus(Vector<uint8_t> const &sessionId, 77 KeyedVector<String8, String8> &infoMap) const; 78 79 status_t getProvisionRequest(String8 const &certType, 80 String8 const &certAuthority, 81 Vector<uint8_t> &request, 82 String8 &defaultUrl); 83 84 status_t provideProvisionResponse(Vector<uint8_t> const &response, 85 Vector<uint8_t> &certificate, 86 Vector<uint8_t> &wrappedKey); 87 88 status_t getSecureStops(List<Vector<uint8_t> > &secureStops); 89 status_t releaseSecureStops(Vector<uint8_t> const &ssRelease); 90 91 status_t getPropertyString(String8 const &name, String8 &value ) const; 92 status_t getPropertyByteArray(String8 const &name, 93 Vector<uint8_t> &value ) const; 94 95 status_t setPropertyString(String8 const &name, 96 String8 const &value ); 97 status_t setPropertyByteArray(String8 const &name, 98 Vector<uint8_t> const &value ); 99 100 status_t setCipherAlgorithm(Vector<uint8_t> const &sessionId, 101 String8 const &algorithm); 102 103 status_t setMacAlgorithm(Vector<uint8_t> const &sessionId, 104 String8 const &algorithm); 105 106 status_t encrypt(Vector<uint8_t> const &sessionId, 107 Vector<uint8_t> const &keyId, 108 Vector<uint8_t> const &input, 109 Vector<uint8_t> const &iv, 110 Vector<uint8_t> &output); 111 112 status_t decrypt(Vector<uint8_t> const &sessionId, 113 Vector<uint8_t> const &keyId, 114 Vector<uint8_t> const &input, 115 Vector<uint8_t> const &iv, 116 Vector<uint8_t> &output); 117 118 status_t sign(Vector<uint8_t> const &sessionId, 119 Vector<uint8_t> const &keyId, 120 Vector<uint8_t> const &message, 121 Vector<uint8_t> &signature); 122 123 status_t verify(Vector<uint8_t> const &sessionId, 124 Vector<uint8_t> const &keyId, 125 Vector<uint8_t> const &message, 126 Vector<uint8_t> const &signature, 127 bool &match); 128 129 status_t signRSA(Vector<uint8_t> const &sessionId, 130 String8 const &algorithm, 131 Vector<uint8_t> const &message, 132 Vector<uint8_t> const &wrappedKey, 133 Vector<uint8_t> &signature); 134 135 private: 136 String8 vectorToString(Vector<uint8_t> const &vector) const; 137 String8 arrayToString(uint8_t const *array, size_t len) const; 138 String8 stringMapToString(KeyedVector<String8, String8> map) const; 139 140 SortedVector<Vector<uint8_t> > mSessions; 141 SortedVector<Vector<uint8_t> > mKeySets; 142 143 static const ssize_t kNotFound = -1; 144 ssize_t findSession(Vector<uint8_t> const &sessionId) const; 145 ssize_t findKeySet(Vector<uint8_t> const &keySetId) const; 146 147 Mutex mLock; 148 KeyedVector<String8, String8> mStringProperties; 149 KeyedVector<String8, Vector<uint8_t> > mByteArrayProperties; 150 }; 151 152 153 class MockCryptoPlugin : public CryptoPlugin { 154 155 bool requiresSecureDecoderComponent(const char *mime) const; 156 157 ssize_t decrypt(bool secure, 158 const uint8_t key[16], const uint8_t iv[16], 159 Mode mode, const void *srcPtr, 160 const SubSample *subSamples, size_t numSubSamples, 161 void *dstPtr, AString *errorDetailMsg); 162 private: 163 String8 subSamplesToString(CryptoPlugin::SubSample const *subSamples, size_t numSubSamples) const; 164 String8 arrayToString(uint8_t const *array, size_t len) const; 165 }; 166}; 167