primary obfuscation working

Author: efraespada

build.gradle

@@ -24,4 +24,9 @@ compileKotlin {
 }
 compileTestKotlin {
     kotlinOptions.jvmTarget = "1.8"
-}
+}
+
+processResources {
+    from 'src/main/kotlin/components/jni'
+    include '*.dylib', '*.dll'
+}

src/main/kotlin/components/Extensions.kt

@@ -95,3 +95,15 @@ fun File.restore(projectPath: String): File {
     return restore
 }
 
+fun ByteArray.toReadableString(): String {
+    val builder = StringBuilder()
+    this.forEachIndexed { index, byte ->
+        if (index == this.size - 1) {
+            builder.append(byte)
+        } else {
+            builder.append("$byte, ")
+        }
+    }
+    return builder.toString()
+}
+

src/main/kotlin/components/Stark.kt

@@ -0,0 +1,40 @@
+package components
+
+import java.io.File
+import java.io.FileOutputStream
+
+class Stark {
+
+    companion object {
+        init {
+            when (getOs()) {
+                Os.WINDOWS -> loadLib("..\\$winLib")
+                Os.OSX -> loadLib("../$osxLib")
+            }
+        }
+
+        private fun loadLib(name: String) {
+            // val path = (Stark::class.java.protectionDomain.codeSource.location.toURI()).path
+            val inputStream = Stark::class.java.getResourceAsStream(name)
+            val buffer = ByteArray(1024)
+            var read = -1
+            val temp = File.createTempFile(name, "")
+            val fos = FileOutputStream(temp)
+
+            read = inputStream.read(buffer)
+            while (read != -1) {
+                fos.write(buffer, 0, read)
+                read = inputStream.read(buffer)
+            }
+            fos.close()
+            inputStream.close()
+
+            System.load(temp.absolutePath)
+        }
+
+        @JvmStatic external fun obfuscate(mainModule: String, key: String, value: ByteArray): ByteArray
+
+        @JvmStatic external fun reveal(mainModule: String, key: String, value: ByteArray): ByteArray
+    }
+
+}

src/main/kotlin/components/Vars.kt

@@ -2,6 +2,8 @@ package components
 
 internal const val resourceBackup = "resbackup"
 internal const val extensionName = "stringcare"
+internal const val winLib = "libsignKey.dll"
+internal const val osxLib = "libsignKey.dylib"
 internal const val wrapperOsX = "./gradlew"
 internal const val wrapperWindows = "gradlew.bat"
 internal const val emptyChar = ""

src/main/kotlin/components/XParser.kt

@@ -4,7 +4,6 @@ import models.Configuration
 import models.ResourceFile
 import models.SAttribute
 import models.StringEntity
-import org.xml.sax.InputSource
 import java.io.File
 import javax.xml.parsers.DocumentBuilderFactory
 
@@ -61,3 +60,14 @@ fun parseXML(file: File, module: String, debug: Boolean): List<StringEntity> {
     return entities
 }
 
+fun obfuscate(mainModule: String, key: String, entity: StringEntity): StringEntity {
+    val obfuscation = Stark.obfuscate(mainModule, key, entity.value.toByteArray()).toReadableString()
+    return StringEntity(entity.name, entity.attributes, obfuscation)
+}
+
+fun reveal(mainModule: String, key: String, entity: StringEntity): StringEntity {
+    val arr: ByteArray = entity.value.split(", ").map { it.toInt().toByte() }.toByteArray()
+    val original = String(Stark.reveal(mainModule, key, arr))
+    return StringEntity(entity.name, entity.attributes, original)
+}
+

src/main/kotlin/components/jni/libsignKey.dll

@@ -1,4 +1,6 @@
 package models
 
-data class StringEntity(val name: String, val attributes: List<SAttribute>, val value: String)
+class StringEntity(var name: String, var attributes: List<SAttribute>, var value: String) {
+
+}
 data class SAttribute(val name: String, val value: String)

src/main/kotlin/components/jni/libsignKey.dylib

@@ -1,12 +1,21 @@
 import components.*
+import org.junit.Before
 import org.junit.Test
+import java.io.File
 
 
 class SCTest {
 
     private val logger by logger()
 
     private val projectName = "KotlinSample"
+    private val mainModuleTest = "app"
+
+    private val librarySetupTask = """
+            cp src/main/kotlin/components/jni/$osxLib out/production/classes/$osxLib
+            cp src/main/kotlin/components/jni/$winLib out/production/classes/$winLib
+        """.trimIndent()
+
     private val prepareTask = """
             rm -rf $projectName &&
             git clone https://github.com/StringCare/$projectName.git &&
@@ -20,6 +29,11 @@ class SCTest {
             ${signingReportTask()}
         """.trimIndent()
 
+    @Before
+    fun setup() {
+        librarySetupTask.runCommand()
+    }
+
     @Test
     fun `terminal verification`() {
         "echo $extensionName".runCommand { command, result ->
@@ -68,19 +82,33 @@ class SCTest {
         prepareTask.runCommand { _, _ ->
             val files = backupFiles(projectName, defaultConfig())
             files.forEach {
-                assert(parseXML(it, "app", true).isNotEmpty())
+                assert(parseXML(it, mainModuleTest, true).isNotEmpty())
             }
         }
     }
 
 
     @Test
     fun `obfuscate string files`() {
-        prepareTask.runCommand { _, _ ->
+        signingReportTask.runCommand { _, report ->
             val files = backupFiles(projectName, defaultConfig())
-            files.forEach {
-                val entities = parseXML(it, "app", true)
-
+            files.forEach { file ->
+                val entities = parseXML(file, mainModuleTest, true)
+                entities.forEach { entity ->
+                    val obfuscated = obfuscate(
+                        "$projectName${File.separator}$mainModuleTest",
+                        report.extractFingerprint(),
+                        entity)
+                    assert(obfuscated.value != entity.value)
+
+                    val original = reveal(
+                        "$projectName${File.separator}$mainModuleTest",
+                        report.extractFingerprint(),
+                        obfuscated)
+
+                    assert(original.value == entity.value)
+
+                }
             }
         }
     }