From 797a4f93b208ecaf1152ba14b8f207af89052c44 Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoit@matrix.org>
Date: Tue, 10 Sep 2024 14:56:10 +0200
Subject: [PATCH] Add unit test on ScanEncryptorUtils.

---
 .../contentscanner/ScanEncryptorUtilsTest.kt  | 109 ++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 matrix-sdk-android/src/androidTest/java/org/matrix/android/sdk/internal/session/contentscanner/ScanEncryptorUtilsTest.kt

diff --git a/matrix-sdk-android/src/androidTest/java/org/matrix/android/sdk/internal/session/contentscanner/ScanEncryptorUtilsTest.kt b/matrix-sdk-android/src/androidTest/java/org/matrix/android/sdk/internal/session/contentscanner/ScanEncryptorUtilsTest.kt
new file mode 100644
index 0000000000..d2c0a6bbbc
--- /dev/null
+++ b/matrix-sdk-android/src/androidTest/java/org/matrix/android/sdk/internal/session/contentscanner/ScanEncryptorUtilsTest.kt
@@ -0,0 +1,109 @@
+/*
+ * Copyright (c) 2024 New Vector Ltd
+ *
+ * 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.
+ */
+
+package org.matrix.android.sdk.internal.session.contentscanner
+
+import okio.ByteString.Companion.decodeBase64
+import org.amshove.kluent.shouldBeEqualTo
+import org.junit.Test
+import org.matrix.android.sdk.api.session.crypto.attachments.ElementToDecrypt
+import org.matrix.android.sdk.api.session.crypto.model.EncryptedFileInfo
+import org.matrix.android.sdk.api.session.crypto.model.EncryptedFileKey
+import org.matrix.android.sdk.internal.crypto.tools.withOlmDecryption
+import org.matrix.android.sdk.internal.di.MoshiProvider
+import org.matrix.android.sdk.internal.session.contentscanner.model.DownloadBody
+import org.matrix.android.sdk.internal.session.contentscanner.model.EncryptedBody
+import org.matrix.olm.OlmPkMessage
+
+class ScanEncryptorUtilsTest {
+    private val anMxcUrl = "mxc://matrix.org/123456"
+    private val anElementToDecrypt = ElementToDecrypt(
+            k = "key",
+            iv = "iv",
+            sha256 = "sha256"
+    )
+    private val aPrivateKey = "CLYwNaeA9d0KHE0DniO1bxGgmNsPJ/pyanF4b4tcK1M="
+
+    @Test
+    fun whenNoServerKeyIsProvidedTheContentIsNotEncrypted() {
+        val result = ScanEncryptorUtils.getDownloadBodyAndEncryptIfNeeded(
+                publicServerKey = null,
+                mxcUrl = anMxcUrl,
+                elementToDecrypt = anElementToDecrypt
+        )
+        result shouldBeEqualTo DownloadBody(
+                file = EncryptedFileInfo(
+                        url = anMxcUrl,
+                        iv = anElementToDecrypt.iv,
+                        hashes = mapOf("sha256" to anElementToDecrypt.sha256),
+                        key = EncryptedFileKey(
+                                k = anElementToDecrypt.k,
+                                alg = "A256CTR",
+                                keyOps = listOf("encrypt", "decrypt"),
+                                kty = "oct",
+                                ext = true
+                        ),
+                        v = "v2"
+                ),
+                encryptedBody = null
+        )
+    }
+
+    @Test
+    fun checkThatTheCodeIsAbleToDecryptContent() {
+        System.loadLibrary("olm")
+        val clearInfo = ScanEncryptorUtils.getDownloadBodyAndEncryptIfNeeded(
+                publicServerKey = null,
+                mxcUrl = anMxcUrl,
+                elementToDecrypt = anElementToDecrypt
+        )
+        // Uncomment to get a new encrypted body
+        // val aPublicKey = "6n3l15JqsNhpM1OwRIoDCL/3c1B5idcwvy07Y5qFRyw="
+        // val encryptedBody = ScanEncryptorUtils.getDownloadBodyAndEncryptIfNeeded(
+        //         publicServerKey = aPublicKey,
+        //         mxcUrl = anMxcUrl,
+        //         elementToDecrypt = anElementToDecrypt
+        // ).encryptedBody!!
+        // println("libolmEncryptedBody: $encryptedBody")
+        val libolmEncryptedBody = EncryptedBody(
+                cipherText = "GTnDhm6xe5fPe/QCr6fyGcZXheFhZlPG" +
+                        "nJZiCK8Xwq6qTg71vSUGWtLdt3uaTmK7" +
+                        "F7fB3PBKchHu2VVv6MMgo8fpUQ7KBbmu" +
+                        "NWTrNmf3QdhXuRwUwz/q4GxsbGR2zjSX" +
+                        "/UoE5S4ymVtOVhvSfXQfssN56wVIzC6S" +
+                        "dy57y6b1IXPihlCUdvb8LMkMvViHYeNf" +
+                        "beFrAfMlsyr1+jdZEXZF5Q7iruhsH2iu" +
+                        "k7+Ayl9rdILCD5tjE9pezwe1V6uc/Agb",
+                mac = "Wk77HRg50oM",
+                ephemeral = "rMTK6/CGASinfX4USFS5qmD3r4meffxKc/jCSFIBczw"
+        )
+        // Try to decrypt the body
+        val result = withOlmDecryption { olmPkDecryption ->
+            olmPkDecryption.setPrivateKey(aPrivateKey.decodeBase64()!!.toByteArray())
+            olmPkDecryption.decrypt(
+                    OlmPkMessage().apply {
+                        mCipherText = libolmEncryptedBody.cipherText
+                        mMac = libolmEncryptedBody.mac
+                        mEphemeralKey = libolmEncryptedBody.ephemeral
+                    }
+            )
+        }
+        val parseResult = MoshiProvider.providesMoshi()
+                .adapter(DownloadBody::class.java)
+                .fromJson(result)
+        parseResult shouldBeEqualTo clearInfo
+    }
+}