Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 7c258c36 authored by TreeHugger Robot's avatar TreeHugger Robot Committed by Android (Google) Code Review
Browse files

Merge "Cleanup drm_hal_clearkey_test."

parents 218ff9cb 31a86d28

drm/1.0/vts/functional/drm_hal_clearkey_test.cpp

+29 −35
Original line number Original line Diff line number Diff line
@@ -89,10 +89,6 @@ static const uint8_t kInvalidUUID[16] = { 
    0x10, 0x20, 0x30, 0x40, 0x50, 0x60, 0x70, 0x80,

    0x10, 0x20, 0x30, 0x40, 0x50, 0x60, 0x70, 0x80,

    0x10, 0x20, 0x30, 0x40, 0x50, 0x60, 0x70, 0x80};

    0x10, 0x20, 0x30, 0x40, 0x50, 0x60, 0x70, 0x80};





static const uint32_t k256SubSampleByteCount = 256;

static const uint32_t k512SubSampleClearBytes = 512;

static const uint32_t k512SubSampleEncryptedBytes = 512;



class DrmHalClearkeyFactoryTest : public ::testing::VtsHalHidlTargetTestBase {

class DrmHalClearkeyFactoryTest : public ::testing::VtsHalHidlTargetTestBase {

   public:

   public:

    virtual void SetUp() override {

    virtual void SetUp() override {
@@ -349,8 +345,7 @@ SessionId DrmHalClearkeyPluginTest::openSession() { 
 * Helper method to close a session

 * Helper method to close a session

 */

 */

void DrmHalClearkeyPluginTest::closeSession(const SessionId& sessionId) {

void DrmHalClearkeyPluginTest::closeSession(const SessionId& sessionId) {

    auto result = drmPlugin->closeSession(sessionId);

    EXPECT_TRUE(drmPlugin->closeSession(sessionId).isOk());

    EXPECT_EQ(Status::OK, result);

}

}





/**

/**
@@ -793,7 +788,7 @@ TEST_F(DrmHalClearkeyPluginTest, SetMacAlgorithmNoSession) { 
 */

 */

TEST_F(DrmHalClearkeyPluginTest, GenericEncryptNotSupported) {

TEST_F(DrmHalClearkeyPluginTest, GenericEncryptNotSupported) {

    SessionId session = openSession();

    SessionId session = openSession();

    ;



    hidl_vec<uint8_t> keyId = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1};

    hidl_vec<uint8_t> keyId = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1};

    hidl_vec<uint8_t> input = {1, 2, 3, 4, 5};

    hidl_vec<uint8_t> input = {1, 2, 3, 4, 5};

    hidl_vec<uint8_t> iv = std::vector<uint8_t>(AES_BLOCK_SIZE, 0);

    hidl_vec<uint8_t> iv = std::vector<uint8_t>(AES_BLOCK_SIZE, 0);
@@ -822,7 +817,7 @@ TEST_F(DrmHalClearkeyPluginTest, GenericDecryptNotSupported) { 




TEST_F(DrmHalClearkeyPluginTest, GenericSignNotSupported) {

TEST_F(DrmHalClearkeyPluginTest, GenericSignNotSupported) {

    SessionId session = openSession();

    SessionId session = openSession();

    ;



    hidl_vec<uint8_t> keyId = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1};

    hidl_vec<uint8_t> keyId = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1};

    hidl_vec<uint8_t> message = {1, 2, 3, 4, 5};

    hidl_vec<uint8_t> message = {1, 2, 3, 4, 5};

    auto res = drmPlugin->sign(session, keyId, message,

    auto res = drmPlugin->sign(session, keyId, message,
@@ -836,7 +831,7 @@ TEST_F(DrmHalClearkeyPluginTest, GenericSignNotSupported) { 




TEST_F(DrmHalClearkeyPluginTest, GenericVerifyNotSupported) {

TEST_F(DrmHalClearkeyPluginTest, GenericVerifyNotSupported) {

    SessionId session = openSession();

    SessionId session = openSession();

    ;



    hidl_vec<uint8_t> keyId = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1};

    hidl_vec<uint8_t> keyId = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1};

    hidl_vec<uint8_t> message = {1, 2, 3, 4, 5};

    hidl_vec<uint8_t> message = {1, 2, 3, 4, 5};

    hidl_vec<uint8_t> signature = {0, 0, 0, 0, 0, 0, 0, 0,

    hidl_vec<uint8_t> signature = {0, 0, 0, 0, 0, 0, 0, 0,
@@ -926,8 +921,7 @@ sp<IMemory> DrmHalClearkeyPluginTest::getDecryptMemory(size_t size, 
 */

 */

TEST_F(DrmHalClearkeyPluginTest, SetMediaDrmSession) {

TEST_F(DrmHalClearkeyPluginTest, SetMediaDrmSession) {

    auto sessionId = openSession();

    auto sessionId = openSession();

    Status status = cryptoPlugin->setMediaDrmSession(sessionId);

    EXPECT_TRUE(cryptoPlugin->setMediaDrmSession(sessionId).isOk());

    EXPECT_EQ(Status::OK, status);

    closeSession(sessionId);

    closeSession(sessionId);

}

}
@@ -948,8 +942,7 @@ TEST_F(DrmHalClearkeyPluginTest, SetMediaDrmSessionClosedSession) { 
 */

 */

TEST_F(DrmHalClearkeyPluginTest, SetMediaDrmSessionEmptySession) {

TEST_F(DrmHalClearkeyPluginTest, SetMediaDrmSessionEmptySession) {

    SessionId sessionId;

    SessionId sessionId;

    Status status = cryptoPlugin->setMediaDrmSession(sessionId);

    EXPECT_TRUE(cryptoPlugin->setMediaDrmSession(sessionId).isOk());

    EXPECT_EQ(Status::OK, status);

}

}





/**

/**
@@ -1131,18 +1124,17 @@ TEST_F(DrmHalClearkeyDecryptTest, TestQueryKeyStatus) { 
TEST_F(DrmHalClearkeyDecryptTest, ClearSegmentTest) {

TEST_F(DrmHalClearkeyDecryptTest, ClearSegmentTest) {

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    const Pattern noPattern = {0, 0};

    const Pattern noPattern = {0, 0};

    const uint32_t kByteCount = 256;

    const vector<SubSample> subSamples = {

    const vector<SubSample> subSamples = {

        {.numBytesOfClearData = k256SubSampleByteCount,

        {.numBytesOfClearData = kByteCount,

         .numBytesOfEncryptedData = 0}};

         .numBytesOfEncryptedData = 0}};

    auto sessionId = openSession();

    auto sessionId = openSession();

    loadKeys(sessionId);

    loadKeys(sessionId);



    EXPECT_TRUE(cryptoPlugin->setMediaDrmSession(sessionId).isOk());

    Status status = cryptoPlugin->setMediaDrmSession(sessionId);

    EXPECT_EQ(Status::OK, status);





    uint32_t byteCount = decrypt(Mode::UNENCRYPTED, &iv[0], subSamples,

    uint32_t byteCount = decrypt(Mode::UNENCRYPTED, &iv[0], subSamples,

            noPattern, Status::OK);

            noPattern, Status::OK);

    EXPECT_EQ(k256SubSampleByteCount, byteCount);

    EXPECT_EQ(kByteCount, byteCount);





    closeSession(sessionId);

    closeSession(sessionId);

}

}
@@ -1154,18 +1146,18 @@ TEST_F(DrmHalClearkeyDecryptTest, ClearSegmentTest) { 
TEST_F(DrmHalClearkeyDecryptTest, EncryptedAesCtrSegmentTest) {

TEST_F(DrmHalClearkeyDecryptTest, EncryptedAesCtrSegmentTest) {

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    const Pattern noPattern = {0, 0};

    const Pattern noPattern = {0, 0};

    const uint32_t kClearBytes = 512;

    const uint32_t kEncryptedBytes = 512;

    const vector<SubSample> subSamples = {

    const vector<SubSample> subSamples = {

        {.numBytesOfClearData = k512SubSampleClearBytes,

        {.numBytesOfClearData = kClearBytes,

         .numBytesOfEncryptedData = k512SubSampleEncryptedBytes}};

         .numBytesOfEncryptedData = kEncryptedBytes}};

    auto sessionId = openSession();

    auto sessionId = openSession();

    loadKeys(sessionId);

    loadKeys(sessionId);



    EXPECT_TRUE(cryptoPlugin->setMediaDrmSession(sessionId).isOk());

    Status status = cryptoPlugin->setMediaDrmSession(sessionId);

    EXPECT_EQ(Status::OK, status);





    uint32_t byteCount = decrypt(Mode::AES_CTR, &iv[0], subSamples,

    uint32_t byteCount = decrypt(Mode::AES_CTR, &iv[0], subSamples,

            noPattern, Status::OK);

            noPattern, Status::OK);

    EXPECT_EQ(k512SubSampleClearBytes + k512SubSampleEncryptedBytes, byteCount);

    EXPECT_EQ(kClearBytes + kEncryptedBytes, byteCount);





    closeSession(sessionId);

    closeSession(sessionId);

}

}
@@ -1177,12 +1169,10 @@ TEST_F(DrmHalClearkeyDecryptTest, EncryptedAesCtrSegmentTestNoKeys) { 
    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    const Pattern noPattern = {0, 0};

    const Pattern noPattern = {0, 0};

    const vector<SubSample> subSamples = {

    const vector<SubSample> subSamples = {

        {.numBytesOfClearData = k256SubSampleByteCount,

        {.numBytesOfClearData = 256,

         .numBytesOfEncryptedData = k256SubSampleByteCount}};

         .numBytesOfEncryptedData = 256}};

    auto sessionId = openSession();

    auto sessionId = openSession();



    EXPECT_TRUE(cryptoPlugin->setMediaDrmSession(sessionId).isOk());

    Status status = cryptoPlugin->setMediaDrmSession(sessionId);

    EXPECT_EQ(Status::OK, status);





    uint32_t byteCount = decrypt(Mode::AES_CTR, &iv[0], subSamples,

    uint32_t byteCount = decrypt(Mode::AES_CTR, &iv[0], subSamples,

            noPattern, Status::ERROR_DRM_NO_LICENSE);

            noPattern, Status::ERROR_DRM_NO_LICENSE);
@@ -1207,9 +1197,9 @@ void DrmHalClearkeyDecryptTest::decryptWithInvalidKeys( 
            EXPECT_EQ(Status::OK, status);

            EXPECT_EQ(Status::OK, status);

            EXPECT_EQ(0u, myKeySetId.size());

            EXPECT_EQ(0u, myKeySetId.size());

        });

        });

    ASSERT_OK(res);

    EXPECT_OK(res);





    ASSERT_TRUE(cryptoPlugin->setMediaDrmSession(sessionId).isOk());

    EXPECT_TRUE(cryptoPlugin->setMediaDrmSession(sessionId).isOk());





    uint32_t byteCount = decrypt(Mode::AES_CTR, &iv[0], subSamples,

    uint32_t byteCount = decrypt(Mode::AES_CTR, &iv[0], subSamples,

            noPattern, Status::ERROR_DRM_NO_LICENSE);

            noPattern, Status::ERROR_DRM_NO_LICENSE);
@@ -1224,9 +1214,11 @@ void DrmHalClearkeyDecryptTest::decryptWithInvalidKeys( 
TEST_F(DrmHalClearkeyDecryptTest, DecryptWithEmptyKey) {

TEST_F(DrmHalClearkeyDecryptTest, DecryptWithEmptyKey) {

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    const Pattern noPattern = {0, 0};

    const Pattern noPattern = {0, 0};

    const uint32_t kClearBytes = 512;

    const uint32_t kEncryptedBytes = 512;

    const vector<SubSample> subSamples = {

    const vector<SubSample> subSamples = {

        {.numBytesOfClearData = k512SubSampleClearBytes,

        {.numBytesOfClearData = kClearBytes,

         .numBytesOfEncryptedData = k512SubSampleEncryptedBytes}};

         .numBytesOfEncryptedData = kEncryptedBytes}};





    // base 64 encoded JSON response string, must not contain padding character '='

    // base 64 encoded JSON response string, must not contain padding character '='

    const hidl_string emptyKeyResponse =

    const hidl_string emptyKeyResponse =
@@ -1259,9 +1251,11 @@ TEST_F(DrmHalClearkeyDecryptTest, DecryptWithEmptyKey) { 
TEST_F(DrmHalClearkeyDecryptTest, DecryptWithKeyTooLong) {

TEST_F(DrmHalClearkeyDecryptTest, DecryptWithKeyTooLong) {

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    vector<uint8_t> iv(AES_BLOCK_SIZE, 0);

    const Pattern noPattern = {0, 0};

    const Pattern noPattern = {0, 0};

    const uint32_t kClearBytes = 512;

    const uint32_t kEncryptedBytes = 512;

    const vector<SubSample> subSamples = {

    const vector<SubSample> subSamples = {

        {.numBytesOfClearData = k512SubSampleClearBytes,

        {.numBytesOfClearData = kClearBytes,

         .numBytesOfEncryptedData = k512SubSampleEncryptedBytes}};

         .numBytesOfEncryptedData = kEncryptedBytes}};





    // base 64 encoded JSON response string, must not contain padding character '='

    // base 64 encoded JSON response string, must not contain padding character '='

    const hidl_string keyTooLongResponse =

    const hidl_string keyTooLongResponse =