From 8d0bc4a8ea2a5d0d6aba036f15b185d60e18778f Mon Sep 17 00:00:00 2001 From: "John \"Juce\" Bruce" Date: Mon, 23 Jun 2014 12:11:56 -0700 Subject: Add Unit Tests for ClearKey Plugin Change-Id: I0008a01d29ade169df9a0fd30ca0303adf9ecad3 --- .../clearkey/tests/InitDataParserUnittest.cpp | 235 +++++++++++++++++++++ 1 file changed, 235 insertions(+) create mode 100644 drm/mediadrm/plugins/clearkey/tests/InitDataParserUnittest.cpp (limited to 'drm/mediadrm/plugins/clearkey/tests/InitDataParserUnittest.cpp') 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 +#include + +#include +#include +#include +#include + +#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& initData, + const String8& initDataType, + Vector* licenseRequest) { + InitDataParser parser; + return parser.parse(initData, initDataType, licenseRequest); + } + + void attemptParseExpectingSuccess(const Vector& initData, + const String8& initDataType, + const Vector& expectedKeys) { + const String8 kRequestPrefix("{\"kids\":["); + const String8 kRequestSuffix("],\"type\":\"temporary\"}"); + Vector request; + ASSERT_EQ(android::OK, attemptParse(initData, initDataType, &request)); + + String8 requestString(reinterpret_cast(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& initData, + const String8& initDataType) { + Vector 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 initData; + initData.appendArray(pssh, 52); + + Vector 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 initData; + initData.appendArray(pssh, 84); + + Vector 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 initData; + initData.appendArray(initDataRaw, 16); + + Vector 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 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 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 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 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 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 initData; + initData.appendArray(pssh, 52); + + attemptParseExpectingFailure(initData, kCencType); +} + +} // namespace clearkeydrm -- cgit v1.1