diff options
Diffstat (limited to 'drm/mediadrm')
5 files changed, 665 insertions, 2 deletions
diff --git a/drm/mediadrm/plugins/clearkey/Android.mk b/drm/mediadrm/plugins/clearkey/Android.mk index c0492e7..716b945 100644 --- a/drm/mediadrm/plugins/clearkey/Android.mk +++ b/drm/mediadrm/plugins/clearkey/Android.mk @@ -18,8 +18,6 @@ include $(CLEAR_VARS) LOCAL_SRC_FILES := \ AesCtrDecryptor.cpp \ - InitDataParser.cpp \ - JsonWebKey.cpp \ ClearKeyUUID.cpp \ CreatePluginFactories.cpp \ CryptoFactory.cpp \ diff --git a/drm/mediadrm/plugins/clearkey/tests/AesCtrDecryptorUnittest.cpp b/drm/mediadrm/plugins/clearkey/tests/AesCtrDecryptorUnittest.cpp new file mode 100644 index 0000000..039e402 --- /dev/null +++ b/drm/mediadrm/plugins/clearkey/tests/AesCtrDecryptorUnittest.cpp @@ -0,0 +1,421 @@ +/* + * Copyright (C) 2014 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include <gtest/gtest.h> +#include <string.h> + +#include <utils/String8.h> +#include <utils/Vector.h> + +#include "AesCtrDecryptor.h" + +namespace clearkeydrm { + +using namespace android; + +class AesCtrDecryptorTest : public ::testing::Test { + protected: + typedef uint8_t Key[kBlockSize]; + + status_t attemptDecrypt(const Key& key, const Iv& iv, const uint8_t* source, + uint8_t* destination, const SubSample* subSamples, + size_t numSubSamples, size_t* bytesDecryptedOut) { + Vector<uint8_t> keyVector; + keyVector.appendArray(key, kBlockSize); + + AesCtrDecryptor decryptor; + return decryptor.decrypt(keyVector, iv, source, destination, subSamples, + numSubSamples, bytesDecryptedOut); + } + + template <size_t totalSize> + void attemptDecryptExpectingSuccess(const Key& key, const Iv& iv, + const uint8_t* encrypted, + const uint8_t* decrypted, + const SubSample* subSamples, + size_t numSubSamples) { + uint8_t outputBuffer[totalSize] = {}; + size_t bytesDecrypted = 0; + ASSERT_EQ(android::OK, attemptDecrypt(key, iv, encrypted, outputBuffer, + subSamples, numSubSamples, + &bytesDecrypted)); + EXPECT_EQ(totalSize, bytesDecrypted); + EXPECT_EQ(0, memcmp(outputBuffer, decrypted, totalSize)); + } +}; + +TEST_F(AesCtrDecryptorTest, DecryptsContiguousEncryptedBlock) { + const size_t kTotalSize = 64; + const size_t kNumSubsamples = 1; + + // Test vectors from NIST-800-38A + Key key = { + 0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, + 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c + }; + + Iv iv = { + 0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, + 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff + }; + + uint8_t encrypted[kTotalSize] = { + 0x87, 0x4d, 0x61, 0x91, 0xb6, 0x20, 0xe3, 0x26, + 0x1b, 0xef, 0x68, 0x64, 0x99, 0x0d, 0xb6, 0xce, + 0x98, 0x06, 0xf6, 0x6b, 0x79, 0x70, 0xfd, 0xff, + 0x86, 0x17, 0x18, 0x7b, 0xb9, 0xff, 0xfd, 0xff, + 0x5a, 0xe4, 0xdf, 0x3e, 0xdb, 0xd5, 0xd3, 0x5e, + 0x5b, 0x4f, 0x09, 0x02, 0x0d, 0xb0, 0x3e, 0xab, + 0x1e, 0x03, 0x1d, 0xda, 0x2f, 0xbe, 0x03, 0xd1, + 0x79, 0x21, 0x70, 0xa0, 0xf3, 0x00, 0x9c, 0xee + }; + + uint8_t decrypted[kTotalSize] = { + 0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96, + 0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a, + 0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c, + 0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e, 0x51, + 0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11, + 0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52, 0xef, + 0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17, + 0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37, 0x10 + }; + + SubSample subSamples[kNumSubsamples] = { + {0, 64} + }; + + attemptDecryptExpectingSuccess<kTotalSize>(key, iv, encrypted, decrypted, + subSamples, kNumSubsamples); +} + +TEST_F(AesCtrDecryptorTest, DecryptsAlignedBifurcatedEncryptedBlock) { + const size_t kTotalSize = 64; + const size_t kNumSubsamples = 2; + + // Test vectors from NIST-800-38A + Key key = { + 0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, + 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c + }; + + Iv iv = { + 0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, + 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff + }; + + uint8_t encrypted[kTotalSize] = { + 0x87, 0x4d, 0x61, 0x91, 0xb6, 0x20, 0xe3, 0x26, + 0x1b, 0xef, 0x68, 0x64, 0x99, 0x0d, 0xb6, 0xce, + 0x98, 0x06, 0xf6, 0x6b, 0x79, 0x70, 0xfd, 0xff, + 0x86, 0x17, 0x18, 0x7b, 0xb9, 0xff, 0xfd, 0xff, + 0x5a, 0xe4, 0xdf, 0x3e, 0xdb, 0xd5, 0xd3, 0x5e, + 0x5b, 0x4f, 0x09, 0x02, 0x0d, 0xb0, 0x3e, 0xab, + 0x1e, 0x03, 0x1d, 0xda, 0x2f, 0xbe, 0x03, 0xd1, + 0x79, 0x21, 0x70, 0xa0, 0xf3, 0x00, 0x9c, 0xee + }; + + uint8_t decrypted[kTotalSize] = { + 0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96, + 0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a, + 0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c, + 0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e, 0x51, + 0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11, + 0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52, 0xef, + 0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17, + 0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37, 0x10 + }; + + SubSample subSamples[kNumSubsamples] = { + {0, 32}, + {0, 32} + }; + + attemptDecryptExpectingSuccess<kTotalSize>(key, iv, encrypted, decrypted, + subSamples, kNumSubsamples); +} + +TEST_F(AesCtrDecryptorTest, DecryptsUnalignedBifurcatedEncryptedBlock) { + const size_t kTotalSize = 64; + const size_t kNumSubsamples = 2; + + // Test vectors from NIST-800-38A + Key key = { + 0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, + 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c + }; + + Iv iv = { + 0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, + 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff + }; + + uint8_t encrypted[kTotalSize] = { + 0x87, 0x4d, 0x61, 0x91, 0xb6, 0x20, 0xe3, 0x26, + 0x1b, 0xef, 0x68, 0x64, 0x99, 0x0d, 0xb6, 0xce, + 0x98, 0x06, 0xf6, 0x6b, 0x79, 0x70, 0xfd, 0xff, + 0x86, 0x17, 0x18, 0x7b, 0xb9, 0xff, 0xfd, 0xff, + 0x5a, 0xe4, 0xdf, 0x3e, 0xdb, 0xd5, 0xd3, 0x5e, + 0x5b, 0x4f, 0x09, 0x02, 0x0d, 0xb0, 0x3e, 0xab, + 0x1e, 0x03, 0x1d, 0xda, 0x2f, 0xbe, 0x03, 0xd1, + 0x79, 0x21, 0x70, 0xa0, 0xf3, 0x00, 0x9c, 0xee + }; + + uint8_t decrypted[kTotalSize] = { + 0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96, + 0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a, + 0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c, + 0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e, 0x51, + 0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11, + 0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52, 0xef, + 0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17, + 0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37, 0x10 + }; + + SubSample subSamples[kNumSubsamples] = { + {0, 29}, + {0, 35} + }; + + attemptDecryptExpectingSuccess<kTotalSize>(key, iv, encrypted, decrypted, + subSamples, kNumSubsamples); +} + +TEST_F(AesCtrDecryptorTest, DecryptsOneMixedSubSample) { + const size_t kTotalSize = 72; + const size_t kNumSubsamples = 1; + + // Based on test vectors from NIST-800-38A + Key key = { + 0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, + 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c + }; + + Iv iv = { + 0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, + 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff + }; + + uint8_t encrypted[kTotalSize] = { + // 8 clear bytes + 0xf0, 0x13, 0xca, 0xc7, 0x00, 0x64, 0x0b, 0xbb, + // 64 encrypted bytes + 0x87, 0x4d, 0x61, 0x91, 0xb6, 0x20, 0xe3, 0x26, + 0x1b, 0xef, 0x68, 0x64, 0x99, 0x0d, 0xb6, 0xce, + 0x98, 0x06, 0xf6, 0x6b, 0x79, 0x70, 0xfd, 0xff, + 0x86, 0x17, 0x18, 0x7b, 0xb9, 0xff, 0xfd, 0xff, + 0x5a, 0xe4, 0xdf, 0x3e, 0xdb, 0xd5, 0xd3, 0x5e, + 0x5b, 0x4f, 0x09, 0x02, 0x0d, 0xb0, 0x3e, 0xab, + 0x1e, 0x03, 0x1d, 0xda, 0x2f, 0xbe, 0x03, 0xd1, + 0x79, 0x21, 0x70, 0xa0, 0xf3, 0x00, 0x9c, 0xee + }; + + uint8_t decrypted[kTotalSize] = { + 0xf0, 0x13, 0xca, 0xc7, 0x00, 0x64, 0x0b, 0xbb, + 0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96, + 0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a, + 0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c, + 0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e, 0x51, + 0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11, + 0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52, 0xef, + 0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17, + 0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37, 0x10 + }; + + SubSample subSamples[kNumSubsamples] = { + {8, 64} + }; + + attemptDecryptExpectingSuccess<kTotalSize>(key, iv, encrypted, decrypted, + subSamples, kNumSubsamples); +} + +TEST_F(AesCtrDecryptorTest, DecryptsAlignedMixedSubSamples) { + const size_t kTotalSize = 80; + const size_t kNumSubsamples = 2; + + // Based on test vectors from NIST-800-38A + Key key = { + 0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, + 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c + }; + + Iv iv = { + 0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, + 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff + }; + + uint8_t encrypted[kTotalSize] = { + // 8 clear bytes + 0xf0, 0x13, 0xca, 0xc7, 0x00, 0x64, 0x0b, 0xbb, + // 32 encrypted bytes + 0x87, 0x4d, 0x61, 0x91, 0xb6, 0x20, 0xe3, 0x26, + 0x1b, 0xef, 0x68, 0x64, 0x99, 0x0d, 0xb6, 0xce, + 0x98, 0x06, 0xf6, 0x6b, 0x79, 0x70, 0xfd, 0xff, + 0x86, 0x17, 0x18, 0x7b, 0xb9, 0xff, 0xfd, 0xff, + // 8 clear bytes + 0x94, 0xba, 0x88, 0x2e, 0x0e, 0x12, 0x11, 0x55, + // 32 encrypted bytes + 0x5a, 0xe4, 0xdf, 0x3e, 0xdb, 0xd5, 0xd3, 0x5e, + 0x5b, 0x4f, 0x09, 0x02, 0x0d, 0xb0, 0x3e, 0xab, + 0x1e, 0x03, 0x1d, 0xda, 0x2f, 0xbe, 0x03, 0xd1, + 0x79, 0x21, 0x70, 0xa0, 0xf3, 0x00, 0x9c, 0xee + }; + + uint8_t decrypted[kTotalSize] = { + 0xf0, 0x13, 0xca, 0xc7, 0x00, 0x64, 0x0b, 0xbb, + 0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96, + 0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a, + 0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c, + 0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e, 0x51, + 0x94, 0xba, 0x88, 0x2e, 0x0e, 0x12, 0x11, 0x55, + 0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11, + 0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52, 0xef, + 0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17, + 0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37, 0x10 + }; + + SubSample subSamples[kNumSubsamples] = { + {8, 32}, + {8, 32} + }; + + attemptDecryptExpectingSuccess<kTotalSize>(key, iv, encrypted, decrypted, + subSamples, kNumSubsamples); +} + +TEST_F(AesCtrDecryptorTest, DecryptsUnalignedMixedSubSamples) { + const size_t kTotalSize = 80; + const size_t kNumSubsamples = 2; + + // Based on test vectors from NIST-800-38A + Key key = { + 0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, + 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c + }; + + Iv iv = { + 0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, + 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff + }; + + uint8_t encrypted[kTotalSize] = { + // 8 clear bytes + 0xf0, 0x13, 0xca, 0xc7, 0x00, 0x64, 0x0b, 0xbb, + // 30 encrypted bytes + 0x87, 0x4d, 0x61, 0x91, 0xb6, 0x20, 0xe3, 0x26, + 0x1b, 0xef, 0x68, 0x64, 0x99, 0x0d, 0xb6, 0xce, + 0x98, 0x06, 0xf6, 0x6b, 0x79, 0x70, 0xfd, 0xff, + 0x86, 0x17, 0x18, 0x7b, 0xb9, 0xff, + // 8 clear bytes + 0x94, 0xba, 0x88, 0x2e, 0x0e, 0x12, 0x11, 0x55, + // 34 encrypted bytes + 0xfd, 0xff, 0x5a, 0xe4, 0xdf, 0x3e, 0xdb, 0xd5, + 0xd3, 0x5e, 0x5b, 0x4f, 0x09, 0x02, 0x0d, 0xb0, + 0x3e, 0xab, 0x1e, 0x03, 0x1d, 0xda, 0x2f, 0xbe, + 0x03, 0xd1, 0x79, 0x21, 0x70, 0xa0, 0xf3, 0x00, + 0x9c, 0xee + }; + + uint8_t decrypted[kTotalSize] = { + 0xf0, 0x13, 0xca, 0xc7, 0x00, 0x64, 0x0b, 0xbb, + 0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96, + 0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a, + 0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c, + 0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x94, 0xba, + 0x88, 0x2e, 0x0e, 0x12, 0x11, 0x55, 0x8e, 0x51, + 0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11, + 0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52, 0xef, + 0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17, + 0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37, 0x10 + }; + + SubSample subSamples[kNumSubsamples] = { + {8, 30}, + {8, 34} + }; + + attemptDecryptExpectingSuccess<kTotalSize>(key, iv, encrypted, decrypted, + subSamples, kNumSubsamples); +} + +TEST_F(AesCtrDecryptorTest, DecryptsComplexMixedSubSamples) { + const size_t kTotalSize = 72; + const size_t kNumSubsamples = 6; + + // Based on test vectors from NIST-800-38A + Key key = { + 0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6, + 0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c + }; + + Iv iv = { + 0xf0, 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, + 0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff + }; + + uint8_t encrypted[kTotalSize] = { + // 4 clear bytes + 0xf0, 0x13, 0xca, 0xc7, + // 1 encrypted bytes + 0x87, + // 9 encrypted bytes + 0x4d, 0x61, 0x91, 0xb6, 0x20, 0xe3, 0x26, 0x1b, + 0xef, + // 11 clear bytes + 0x81, 0x4f, 0x24, 0x87, 0x0e, 0xde, 0xba, 0xad, + 0x11, 0x9b, 0x46, + // 20 encrypted bytes + 0x68, 0x64, 0x99, 0x0d, 0xb6, 0xce, + 0x98, 0x06, 0xf6, 0x6b, 0x79, 0x70, 0xfd, 0xff, + 0x86, 0x17, 0x18, 0x7b, 0xb9, 0xff, + // 8 clear bytes + 0x94, 0xba, 0x88, 0x2e, 0x0e, 0x12, 0x11, 0x55, + // 3 clear bytes + 0x10, 0xf5, 0x22, + // 14 encrypted bytes + 0xfd, 0xff, 0x5a, 0xe4, 0xdf, 0x3e, 0xdb, 0xd5, + 0xd3, 0x5e, 0x5b, 0x4f, 0x09, 0x02, + // 2 clear bytes + 0x02, 0x01 + }; + + uint8_t decrypted[kTotalSize] = { + 0xf0, 0x13, 0xca, 0xc7, 0x6b, 0xc1, 0xbe, 0xe2, + 0x2e, 0x40, 0x9f, 0x96, 0xe9, 0x3d, 0x81, 0x4f, + 0x24, 0x87, 0x0e, 0xde, 0xba, 0xad, 0x11, 0x9b, + 0x46, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a, 0xae, + 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c, 0x9e, + 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x94, 0xba, 0x88, + 0x2e, 0x0e, 0x12, 0x11, 0x55, 0x10, 0xf5, 0x22, + 0x8e, 0x51, 0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, + 0xe4, 0x11, 0xe5, 0xfb, 0xc1, 0x19, 0x02, 0x01 + }; + + SubSample subSamples[kNumSubsamples] = { + {4, 1}, + {0, 9}, + {11, 20}, + {8, 0}, + {3, 14}, + {2, 0} + }; + + attemptDecryptExpectingSuccess<kTotalSize>(key, iv, encrypted, decrypted, + subSamples, kNumSubsamples); +} + +} // namespace clearkeydrm diff --git a/drm/mediadrm/plugins/clearkey/tests/Android.mk b/drm/mediadrm/plugins/clearkey/tests/Android.mk index c12cd9f..0c895ea 100644 --- a/drm/mediadrm/plugins/clearkey/tests/Android.mk +++ b/drm/mediadrm/plugins/clearkey/tests/Android.mk @@ -22,5 +22,13 @@ test_name := JsonWebKeyUnittest test_src_dir := . include $(LOCAL_PATH)/unit-test.mk +test_name := AesCtrDecryptorUnittest +test_src_dir := . +include $(LOCAL_PATH)/unit-test.mk + +test_name := InitDataParserUnittest +test_src_dir := . +include $(LOCAL_PATH)/unit-test.mk + test_name := test_src_dir := diff --git a/drm/mediadrm/plugins/clearkey/tests/InitDataParserUnittest.cpp b/drm/mediadrm/plugins/clearkey/tests/InitDataParserUnittest.cpp new file mode 100644 index 0000000..4ba65ed --- /dev/null +++ b/drm/mediadrm/plugins/clearkey/tests/InitDataParserUnittest.cpp @@ -0,0 +1,235 @@ +/* + * Copyright (C) 2014 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include <gtest/gtest.h> +#include <string.h> + +#include <media/stagefright/foundation/AString.h> +#include <media/stagefright/foundation/base64.h> +#include <utils/String8.h> +#include <utils/Vector.h> + +#include "InitDataParser.h" + +namespace clearkeydrm { + +using namespace android; + +namespace { + const size_t kKeyIdSize = 16; + const String8 kCencType("cenc"); + const String8 kWebMType("webm"); + const String8 kBase64Padding("="); +} + +class InitDataParserTest : public ::testing::Test { + protected: + status_t attemptParse(const Vector<uint8_t>& initData, + const String8& initDataType, + Vector<uint8_t>* licenseRequest) { + InitDataParser parser; + return parser.parse(initData, initDataType, licenseRequest); + } + + void attemptParseExpectingSuccess(const Vector<uint8_t>& initData, + const String8& initDataType, + const Vector<String8>& expectedKeys) { + const String8 kRequestPrefix("{\"kids\":["); + const String8 kRequestSuffix("],\"type\":\"temporary\"}"); + Vector<uint8_t> request; + ASSERT_EQ(android::OK, attemptParse(initData, initDataType, &request)); + + String8 requestString(reinterpret_cast<const char*>(request.array()), + request.size()); + EXPECT_EQ(0, requestString.find(kRequestPrefix)); + EXPECT_EQ(requestString.size() - kRequestSuffix.size(), + requestString.find(kRequestSuffix)); + for (size_t i = 0; i < expectedKeys.size(); ++i) { + AString encodedIdAString; + android::encodeBase64(expectedKeys[i], kKeyIdSize, + &encodedIdAString); + String8 encodedId(encodedIdAString.c_str()); + encodedId.removeAll(kBase64Padding); + EXPECT_TRUE(requestString.contains(encodedId)); + } + } + + void attemptParseExpectingFailure(const Vector<uint8_t>& initData, + const String8& initDataType) { + Vector<uint8_t> request; + ASSERT_NE(android::OK, attemptParse(initData, initDataType, &request)); + EXPECT_EQ(0, request.size()); + } +}; + +TEST_F(InitDataParserTest, ParsesSingleKeyPssh) { + uint8_t pssh[52] = { + 0, 0, 0, 52, // Total Size + 'p', 's', 's', 'h', // PSSH + 1, 0, 0, 0, // Version + 0x10, 0x77, 0xef, 0xec, 0xc0, 0xb2, 0x4d, 0x02, // System ID + 0xac, 0xe3, 0x3c, 0x1e, 0x52, 0xe2, 0xfb, 0x4b, + 0, 0, 0, 1, // Key Count + 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, // Key ID #1 + 0x38, 0x39, 0x30, 0x41, 0x42, 0x43, 0x44, 0x45, // "01234567890ABCDE" + 0, 0, 0, 0 // Data Size (always 0) + }; + Vector<uint8_t> initData; + initData.appendArray(pssh, 52); + + Vector<String8> expectedKeys; + expectedKeys.push(String8("01234567890ABCDE")); + + attemptParseExpectingSuccess(initData, kCencType, expectedKeys); +} + +TEST_F(InitDataParserTest, ParsesMultipleKeyPssh) { + uint8_t pssh[84] = { + 0, 0, 0, 84, // Total Size + 'p', 's', 's', 'h', // PSSH + 1, 0, 0, 0, // Version + 0x10, 0x77, 0xef, 0xec, 0xc0, 0xb2, 0x4d, 0x02, // System ID + 0xac, 0xe3, 0x3c, 0x1e, 0x52, 0xe2, 0xfb, 0x4b, + 0, 0, 0, 3, // Key Count + 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, // Key ID #1 + 0x38, 0x39, 0x30, 0x41, 0x42, 0x43, 0x44, 0x45, // "01234567890ABCDE" + 0x43, 0x6c, 0x65, 0x61, 0x72, 0x4b, 0x65, 0x79, // Key ID #2 + 0x43, 0x6c, 0x65, 0x61, 0x72, 0x4b, 0x65, 0x79, // "ClearKeyClearKey" + 0x20, 0x47, 0x4f, 0x4f, 0x47, 0x4c, 0x45, 0x20, // Key ID #3 + 0x20, 0x47, 0x4f, 0x4f, 0x47, 0x4c, 0x45, 0x20, // " GOOGLE GOOGLE " + 0, 0, 0, 0 // Data Size (always 0) + }; + Vector<uint8_t> initData; + initData.appendArray(pssh, 84); + + Vector<String8> expectedKeys; + expectedKeys.push(String8("01234567890ABCDE")); + expectedKeys.push(String8("ClearKeyClearKey")); + expectedKeys.push(String8(" GOOGLE GOOGLE ")); + + attemptParseExpectingSuccess(initData, kCencType, expectedKeys); +} + +TEST_F(InitDataParserTest, ParsesWebM) { + uint8_t initDataRaw[16] = { + 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, // Key ID + 0x38, 0x39, 0x30, 0x41, 0x42, 0x43, 0x44, 0x45, // "01234567890ABCDE" + }; + Vector<uint8_t> initData; + initData.appendArray(initDataRaw, 16); + + Vector<String8> expectedKeys; + expectedKeys.push(String8("01234567890ABCDE")); + + attemptParseExpectingSuccess(initData, kWebMType, expectedKeys); +} + +TEST_F(InitDataParserTest, FailsForPsshTooSmall) { + uint8_t pssh[16] = { + 0, 0, 0, 52, + 'p', 's', 's', 'h', + 1, 0, 0, 0, + 0x10, 0x77, 0xef, 0xec + }; + Vector<uint8_t> initData; + initData.appendArray(pssh, 16); + + attemptParseExpectingFailure(initData, kCencType); +} + +TEST_F(InitDataParserTest, FailsForWebMTooSmall) { + uint8_t initDataRaw[8] = { + 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37 + }; + Vector<uint8_t> initData; + initData.appendArray(initDataRaw, 8); + + attemptParseExpectingFailure(initData, kWebMType); +} + +TEST_F(InitDataParserTest, FailsForPsshBadSystemId) { + uint8_t pssh[52] = { + 0, 0, 0, 52, // Total Size + 'p', 's', 's', 'h', // PSSH + 1, 0, 0, 0, // Version + 0xac, 0xe3, 0x3c, 0x1e, 0x52, 0xe2, 0xfb, 0x4b, // System ID + 0x10, 0x77, 0xef, 0xec, 0xc0, 0xb2, 0x4d, 0x02, + 0, 0, 0, 1, // Key Count + 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, // Key ID #1 + 0x38, 0x39, 0x30, 0x41, 0x42, 0x43, 0x44, 0x45, // "01234567890ABCDE" + 0, 0, 0, 0 // Data Size (always 0) + }; + Vector<uint8_t> initData; + initData.appendArray(pssh, 52); + + attemptParseExpectingFailure(initData, kCencType); +} + +TEST_F(InitDataParserTest, FailsForPsshBadSize) { + uint8_t pssh[52] = { + 0, 0, 70, 200, // Total Size + 'p', 's', 's', 'h', // PSSH + 1, 0, 0, 0, // Version + 0x10, 0x77, 0xef, 0xec, 0xc0, 0xb2, 0x4d, 0x02, // System ID + 0xac, 0xe3, 0x3c, 0x1e, 0x52, 0xe2, 0xfb, 0x4b, + 0, 0, 0, 1, // Key Count + 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, // Key ID #1 + 0x38, 0x39, 0x30, 0x41, 0x42, 0x43, 0x44, 0x45, // "01234567890ABCDE" + 0, 0, 0, 0 // Data Size (always 0) + }; + Vector<uint8_t> initData; + initData.appendArray(pssh, 52); + + attemptParseExpectingFailure(initData, kCencType); +} + +TEST_F(InitDataParserTest, FailsForPsshWrongVersion) { + uint8_t pssh[52] = { + 0, 0, 0, 52, // Total Size + 'p', 's', 's', 'h', // PSSH + 0, 0, 0, 0, // Version + 0x10, 0x77, 0xef, 0xec, 0xc0, 0xb2, 0x4d, 0x02, // System ID + 0xac, 0xe3, 0x3c, 0x1e, 0x52, 0xe2, 0xfb, 0x4b, + 0, 0, 0, 1, // Key Count + 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, // Key ID #1 + 0x38, 0x39, 0x30, 0x41, 0x42, 0x43, 0x44, 0x45, // "01234567890ABCDE" + 0, 0, 0, 0 // Data Size (always 0) + }; + Vector<uint8_t> initData; + initData.appendArray(pssh, 52); + + attemptParseExpectingFailure(initData, kCencType); +} + +TEST_F(InitDataParserTest, FailsForPsshBadKeyCount) { + uint8_t pssh[52] = { + 0, 0, 0, 52, // Total Size + 'p', 's', 's', 'h', // PSSH + 1, 0, 0, 0, // Version + 0x10, 0x77, 0xef, 0xec, 0xc0, 0xb2, 0x4d, 0x02, // System ID + 0xac, 0xe3, 0x3c, 0x1e, 0x52, 0xe2, 0xfb, 0x4b, + 0, 0, 0, 7, // Key Count + 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, // Key ID #1 + 0x38, 0x39, 0x30, 0x41, 0x42, 0x43, 0x44, 0x45, // "01234567890ABCDE" + 0, 0, 0, 0 // Data Size (always 0) + }; + Vector<uint8_t> initData; + initData.appendArray(pssh, 52); + + attemptParseExpectingFailure(initData, kCencType); +} + +} // namespace clearkeydrm diff --git a/drm/mediadrm/plugins/clearkey/tests/unit-test.mk b/drm/mediadrm/plugins/clearkey/tests/unit-test.mk index 6afb5a3..7504f34 100644 --- a/drm/mediadrm/plugins/clearkey/tests/unit-test.mk +++ b/drm/mediadrm/plugins/clearkey/tests/unit-test.mk @@ -45,6 +45,7 @@ LOCAL_SHARED_LIBRARIES := \ libcrypto \ libdrmclearkeyplugin \ liblog \ + libstagefright_foundation \ libstlport \ libutils \ |