1/*
2 * Copyright (C) 2017 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//#define LOG_NDEBUG 0
18#define LOG_TAG "ClearKeyFetcherTest"
19#include <utils/Log.h>
20#include <gtest/gtest.h>
21#include <stddef.h>
22#include <algorithm>
23#include <string>
24
25#include "ClearKeyFetcher.h"
26#include "ClearKeyLicenseFetcher.h"
27#include "protos/license_protos.pb.h"
28
29namespace android {
30namespace clearkeycas {
31
32const char *kTestAssetInJson =
33        "{                                                   "
34        "  \"id\": 21140844,                                 "
35        "  \"name\": \"Test Title\",                         "
36        "  \"lowercase_organization_name\": \"Android\",     "
37        "  \"asset_key\": {                                  "
38        "  \"encryption_key\": \"nezAr3CHFrmBR9R8Tedotw==\"  "
39        "  },                                                "
40        "  \"cas_type\": 1,                                  "
41        "  \"track_types\": [ ]                              "
42        "}                                                   " ;
43
44const uint8_t kTestEcmContainer[] = {
45        0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
46        0x00, 0x00, 0x27, 0x10, 0x02, 0x00, 0x01, 0x77,
47        0x01, 0x42, 0x95, 0x6c, 0x0e, 0xe3, 0x91, 0xbc,
48        0xfd, 0x05, 0xb1, 0x60, 0x4f, 0x17, 0x82, 0xa4,
49        0x86, 0x9b, 0x23, 0x56, 0x00, 0x01, 0x00, 0x00,
50        0x00, 0x01, 0x00, 0x00, 0x27, 0x10, 0x02, 0x00,
51        0x01, 0x77, 0x01, 0x42, 0x95, 0x6c, 0xd7, 0x43,
52        0x62, 0xf8, 0x1c, 0x62, 0x19, 0x05, 0xc7, 0x3a,
53        0x42, 0xcd, 0xfd, 0xd9, 0x13, 0x48,
54};
55
56const uint8_t kTestContentKey0[] = {
57        0x0e, 0xe3, 0x91, 0xbc, 0xfd, 0x05, 0xb1, 0x60,
58        0x4f, 0x17, 0x82, 0xa4, 0x86, 0x9b, 0x23, 0x56};
59
60const uint8_t kTestContentKey1[] = {
61        0xd7, 0x43, 0x62, 0xf8, 0x1c, 0x62, 0x19, 0x05,
62        0xc7, 0x3a, 0x42, 0xcd, 0xfd, 0xd9, 0x13, 0x48};
63
64constexpr uint32_t kTestEcmCount = 2;
65
66class ClearKeyFetcherTest : public testing::Test {
67protected:
68    virtual void SetUp();
69
70protected:
71    std::unique_ptr<ClearKeyLicenseFetcher> license_fetcher_;
72    sp<ABuffer> ecm_;
73    sp<ABuffer> content_key_[kTestEcmCount];
74};
75
76void ClearKeyFetcherTest::SetUp() {
77    license_fetcher_.reset(new ClearKeyLicenseFetcher());
78    EXPECT_EQ(OK, license_fetcher_->Init(kTestAssetInJson));
79    ecm_ = new ABuffer((void*) (kTestEcmContainer), sizeof(kTestEcmContainer));
80    content_key_[0] = new ABuffer(
81            (void*)kTestContentKey0, sizeof(kTestContentKey0));
82    content_key_[1] = new ABuffer(
83            (void*)kTestContentKey1, sizeof(kTestContentKey1));
84}
85
86TEST_F(ClearKeyFetcherTest, Ctor) {
87    ClearKeyFetcher fetcher(std::move(license_fetcher_));
88}
89
90TEST_F(ClearKeyFetcherTest, Success) {
91    ClearKeyFetcher fetcher(std::move(license_fetcher_));
92    EXPECT_EQ(OK, fetcher.Init());
93    uint64_t asset_id;
94    std::vector<KeyFetcher::KeyInfo> keys;
95    EXPECT_EQ(OK, fetcher.ObtainKey(ecm_, &asset_id, &keys));
96    EXPECT_EQ(2, keys.size());
97    EXPECT_EQ(0, keys[0].key_id);
98    EXPECT_EQ(content_key_[0]->size(), keys[0].key_bytes->size());
99    EXPECT_EQ(0, memcmp(content_key_[0]->data(),
100            keys[0].key_bytes->data(), content_key_[0]->size()));
101    EXPECT_EQ(1, keys[1].key_id);
102    EXPECT_EQ(content_key_[1]->size(), keys[1].key_bytes->size());
103    EXPECT_EQ(0, memcmp(content_key_[1]->data(),
104            keys[1].key_bytes->data(), content_key_[1]->size()));
105}
106
107} // namespace clearkeycas
108} // namespace android
109