1 /* 2 * Copyright (C) 2018 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 CLEARKEY_SESSION_H_ 18 #define CLEARKEY_SESSION_H_ 19 20 #include <utils/Mutex.h> 21 #include <utils/RefBase.h> 22 #include <vector> 23 24 #include "ClearKeyTypes.h" 25 26 27 namespace android { 28 namespace hardware { 29 namespace drm { 30 namespace V1_4 { 31 namespace clearkey { 32 33 namespace drm = ::android::hardware::drm; 34 using drm::V1_0::Status; 35 using drm::V1_0::SubSample; 36 37 typedef drm::V1_2::Status Status_V1_2; 38 39 class Session : public RefBase { 40 public: Session(const std::vector<uint8_t> & sessionId)41 explicit Session(const std::vector<uint8_t>& sessionId) 42 : mSessionId(sessionId), mMockError(Status_V1_2::OK) {} ~Session()43 virtual ~Session() {} 44 sessionId()45 const std::vector<uint8_t>& sessionId() const { return mSessionId; } 46 47 Status getKeyRequest( 48 const std::vector<uint8_t>& initDataType, 49 const std::string& mimeType, 50 V1_0::KeyType keyType, 51 std::vector<uint8_t>* keyRequest) const; 52 53 Status provideKeyResponse( 54 const std::vector<uint8_t>& response); 55 56 Status_V1_2 decrypt( 57 const KeyId keyId, const Iv iv, const uint8_t* srcPtr, 58 uint8_t* dstPtr, const std::vector<SubSample> subSamples, 59 size_t* bytesDecryptedOut); 60 setMockError(Status_V1_2 error)61 void setMockError(Status_V1_2 error) {mMockError = error;} getMockError()62 Status_V1_2 getMockError() const {return mMockError;} 63 64 private: 65 CLEARKEY_DISALLOW_COPY_AND_ASSIGN(Session); 66 67 const std::vector<uint8_t> mSessionId; 68 KeyMap mKeyMap; 69 Mutex mMapLock; 70 71 // For mocking error return scenarios 72 Status_V1_2 mMockError; 73 }; 74 75 } // namespace clearkey 76 } // namespace V1_4 77 } // namespace drm 78 } // namespace hardware 79 } // namespace android 80 81 #endif // CLEARKEY_SESSION_H_ 82