Add state cleanup for delegated and pure. There doesn't seem to be a easy way to clean js state

multiplatform-crypto-delegated/src/commonMain/kotlin/com/ionspin/kotlin/crypto/Crypto.kt

@@ -189,6 +189,9 @@ class MultipartAuthenticatedEncryptor internal constructor(val key : SymmetricKe
         return EncryptedDataPart(primitive.encrypt(data, additionalData))
     }
 
+    override fun cleanup() {
+        primitive.cleanup()
+    }
 }
 
 
@@ -197,6 +200,10 @@ class MultipartAuthenticatedDecryptor internal constructor(val decryptor: XChaCh
         return DecryptedDataPart(decryptor.decrypt(data.data, additionalData))
     }
 
+    override fun cleanup() {
+        decryptor.cleanup()
+    }
+
 }
 
 expect object Initializer {

multiplatform-crypto-delegated/src/commonMain/kotlin/com/ionspin/kotlin/crypto/authenticated/DelegatedXChaCha20Poly1305.kt

@@ -16,6 +16,7 @@ expect class XChaCha20Poly1305Delegated internal constructor() {
     fun initializeForDecryption(key: UByteArray, header: UByteArray)
     fun encrypt(data: UByteArray, additionalData: UByteArray = ubyteArrayOf()) : UByteArray
     fun decrypt(data: UByteArray, additionalData: UByteArray = ubyteArrayOf()) : UByteArray
+    fun cleanup()
 
 
 }

multiplatform-crypto-delegated/src/commonTest/kotlin/com/ionspin/kotlin/crypto/highlevel/EncryptionTest.kt

@@ -40,6 +40,9 @@ class EncryptionTest {
         assertTrue {
             plaintext.contentEquals(combinedPlaintext)
         }
+        encryptor.cleanup()
+        decryptor.cleanup()
+
 
 
 

multiplatform-crypto-delegated/src/jsMain/kotlin/com/ionspin/kotlin/crypto/JsSodiumInterface.kt

@@ -52,5 +52,9 @@ interface JsSodiumInterface {
     fun crypto_secretstream_xchacha20poly1305_init_pull(header: Uint8Array, key: Uint8Array) : dynamic
     fun crypto_secretstream_xchacha20poly1305_pull(state: dynamic, ciphertext: Uint8Array, additionalData: Uint8Array) : dynamic
 
+    //util
+    fun memzero(array: Uint8Array)
+
+
 
 }

multiplatform-crypto-delegated/src/jsMain/kotlin/com/ionspin/kotlin/crypto/authenticated/XChaCha20Poly1305Delegated.kt

@@ -121,6 +121,10 @@ actual class XChaCha20Poly1305Delegated internal actual constructor() {
         return decrypted.toUByteArray()
     }
 
+    actual fun cleanup() {
+        //TODO JS cleanup
+    }
+
 
 
 }

multiplatform-crypto-delegated/src/jvmMain/kotlin/com/ionspin/kotlin/crypto/authenticated/XChaCha20Poly1305Delegated.kt

@@ -131,6 +131,11 @@ actual class XChaCha20Poly1305Delegated internal actual constructor() {
 
     }
 
+    actual fun cleanup() {
+        sodium.sodium_memzero(state.k, 32)
+        sodium.sodium_memzero(state.nonce, 12)
+    }
+
 
 
 

multiplatform-crypto-delegated/src/nativeMain/kotlin/com/ionspin/kotlin/crypto/authenticated/XChaCha20Poly1305Delegated.kt

@@ -4,7 +4,6 @@ import com.ionspin.kotlin.bignum.integer.util.hexColumsPrint
 import com.ionspin.kotlin.crypto.InvalidTagException
 import kotlinx.cinterop.*
 import libsodium.*
-import platform.posix.malloc
 
 /**
  * Created by Ugljesa Jovanovic
@@ -63,10 +62,9 @@ actual class XChaCha20Poly1305Delegated internal actual constructor() {
     }
 
     var state =
-        malloc(crypto_secretstream_xchacha20poly1305_state.size.convert())!!
+        sodium_malloc(crypto_secretstream_xchacha20poly1305_state.size.convert())!!
             .reinterpret<crypto_secretstream_xchacha20poly1305_state>()
             .pointed
-
     val header = UByteArray(crypto_secretstream_xchacha20poly1305_HEADERBYTES.toInt()) { 0U }
 
     var isInitialized = false
@@ -155,6 +153,10 @@ actual class XChaCha20Poly1305Delegated internal actual constructor() {
         return plaintext
     }
 
+    actual fun cleanup() {
+        sodium_free(state.ptr)
+    }
+
 
 
 

multiplatform-crypto-delegated/src/nativeMain/kotlin/com/ionspin/kotlin/crypto/hash/sha/Sha256Delegated.kt

@@ -18,7 +18,7 @@ actual class Sha256Delegated : Sha256 {
     val state : crypto_hash_sha256_state
 
     init {
-        val allocated = malloc(crypto_hash_sha256_state.size.convert())!!
+        val allocated = sodium_malloc(crypto_hash_sha256_state.size.convert())!!
         state = allocated.reinterpret<crypto_hash_sha256_state>().pointed
         crypto_hash_sha256_init(state.ptr)
     }
@@ -33,7 +33,7 @@ actual class Sha256Delegated : Sha256 {
         val hashResult = UByteArray(Sha256Properties.MAX_HASH_BYTES)
         val hashResultPinned = hashResult.pin()
         crypto_hash_sha256_final(state.ptr, hashResultPinned.addressOf(0))
-        free(state.ptr)
+        sodium_free(state.ptr)
         return hashResult
     }
 
@@ -51,4 +51,4 @@ actual object Sha256StatelessDelegated : StatelessSha256 {
         return hashResult
     }
 
-}
+}