nullptr/Wallenc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Проверка ключа

Browse Source
This commit is contained in:
Roman Pytkov
2025-01-05 17:28:12 +03:00
parent ccafffe398
commit ec27fc11fd
4 changed files with 35 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
domain/src/main/java/com/github/nullptroma/wallenc/domain/datatypes/EncryptKey.kt
View File

@@ -1,10 +1,15 @@
package com.github.nullptroma.wallenc.domain.datatypes
import java.security.MessageDigest
import javax.crypto.spec.SecretKeySpec
class EncryptKey(val key: String) {
fun to32Bytes(): ByteArray {
val digest = MessageDigest.getInstance("SHA-256")
return digest.digest(key.toByteArray(Charsets.UTF_8))
}
fun toAesKey() : SecretKeySpec {
return SecretKeySpec(to32Bytes(), "AES")
}
}

14
domain/src/main/java/com/github/nullptroma/wallenc/domain/encrypt/EncryptedStorageAccessor.kt
View File

@@ -5,7 +5,6 @@ import com.github.nullptroma.wallenc.domain.common.impl.CommonFile
import com.github.nullptroma.wallenc.domain.common.impl.CommonMetaInfo
import com.github.nullptroma.wallenc.domain.datatypes.DataPackage
import com.github.nullptroma.wallenc.domain.datatypes.EncryptKey
import com.github.nullptroma.wallenc.domain.datatypes.StorageEncryptionInfo
import com.github.nullptroma.wallenc.domain.interfaces.IDirectory
import com.github.nullptroma.wallenc.domain.interfaces.IFile
import com.github.nullptroma.wallenc.domain.interfaces.ILogger
@@ -23,7 +22,6 @@ import kotlinx.coroutines.flow.map
import kotlinx.coroutines.launch
import java.io.InputStream
import java.io.OutputStream
import javax.crypto.spec.SecretKeySpec
import kotlin.io.path.Path
import kotlin.io.path.pathString
@@ -46,7 +44,7 @@ class EncryptedStorageAccessor(
private val _dirsUpdates = MutableSharedFlow<DataPackage<List<IDirectory>>>()
override val dirsUpdates: SharedFlow<DataPackage<List<IDirectory>>> = _dirsUpdates
private val _encryptor = Encryptor(SecretKeySpec(key.to32Bytes(), "AES"))
private val _encryptor = Encryptor(key.toAesKey())
init {
collectSourceState()
@@ -217,14 +215,4 @@ class EncryptedStorageAccessor(
_job.cancel()
_encryptor.dispose()
}
companion object {
private const val IV_LEN = 16
private const val AES_SETTINGS = "AES/CBC/PKCS5Padding"
fun generateEncryptionInfo(key: EncryptKey): StorageEncryptionInfo {
TODO()
}
}
}

29
domain/src/main/java/com/github/nullptroma/wallenc/domain/encrypt/Encryptor.kt
View File

@@ -1,5 +1,7 @@
package com.github.nullptroma.wallenc.domain.encrypt
import com.github.nullptroma.wallenc.domain.datatypes.EncryptKey
import com.github.nullptroma.wallenc.domain.datatypes.StorageEncryptionInfo
import kotlinx.coroutines.DisposableHandle
import java.io.InputStream
import java.io.OutputStream
@@ -74,8 +76,33 @@ class Encryptor(private var _secretKey: SecretKey?) : DisposableHandle {
companion object {
private const val IV_LEN = 16
private const val TEST_DATA_LEN = 512
private const val AES_SETTINGS = "AES/CBC/PKCS5Padding"
@OptIn(ExperimentalEncodingApi::class)
fun generateEncryptionInfo(key: EncryptKey) : StorageEncryptionInfo {
val encryptor = Encryptor(key.toAesKey())
val testData = ByteArray(TEST_DATA_LEN)
val encryptedData = encryptor.encryptBytes(testData)
return StorageEncryptionInfo(
isEncrypted = true,
encryptedTestData = Base64.Default.encode(encryptedData)
)
}
@OptIn(ExperimentalEncodingApi::class)
fun checkKey(key: EncryptKey, encInfo: StorageEncryptionInfo): Boolean {
if(encInfo.encryptedTestData == null)
return false
val encryptor = Encryptor(key.toAesKey())
try {
val encData = Base64.Default.decode(encInfo.encryptedTestData)
val testData = encryptor.decryptBytes(encData)
return testData.all { it == 0.toByte() } && testData.size == TEST_DATA_LEN
}
catch (e: Exception) {
return false
}
}
}
}