blob: 145f3e6c8025ebd51e71db408b0c8e02c289d974 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
package space.anity
import java.io.*
import java.security.*
import javax.crypto.*
import javax.crypto.spec.*
class CryptoHandler @Throws(NoSuchPaddingException::class, NoSuchAlgorithmException::class)
internal constructor(private val secretKey: SecretKey, cipher: String) {
private val cipher: Cipher = Cipher.getInstance(cipher)
@Throws(InvalidKeyException::class, IOException::class)
internal fun encrypt(content: String, fileName: String) {
cipher.init(Cipher.ENCRYPT_MODE, secretKey)
val iv = cipher.iv
FileOutputStream(fileName).use { fileOut ->
fileOut.write(iv)
CipherOutputStream(fileOut, cipher).use { cipherOut ->
cipherOut.write(content.toByteArray())
}
}
}
@Throws(InvalidAlgorithmParameterException::class, InvalidKeyException::class, IOException::class)
internal fun decrypt(fileName: String): String {
var content = ""
FileInputStream(fileName).use { fileIn ->
val iv = ByteArray(16)
fileIn.read(iv)
cipher.init(Cipher.DECRYPT_MODE, secretKey, IvParameterSpec(iv))
CipherInputStream(fileIn, cipher).use { cipherIn ->
InputStreamReader(cipherIn).use { inputReader ->
BufferedReader(inputReader).use { reader ->
val sb = StringBuilder()
var line: String? = reader.readLine()
while (line != null) {
sb.append(line)
line = reader.readLine()
}
content = sb.toString()
}
}
}
}
return content // TODO: Fix char handling as 1 byte in decryption
}
}
|
