diff --git a/src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt b/src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt index 32f48c4..7348988 100644 --- a/src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt +++ b/src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt @@ -48,7 +48,7 @@ class KyberAgreement(kemKeyPair: KyberKEMKeyPair) { * @param randomness Random 32 Bytes * @return KyberCipherText The Cipher Text to send to the second party. */ - private fun toCipherText(encryptionKey: KyberEncryptionKey, plainText: ByteArray, randomness: ByteArray): KyberCipherText { + internal fun toCipherText(encryptionKey: KyberEncryptionKey, plainText: ByteArray, randomness: ByteArray): KyberCipherText { val parameter = encryptionKey.parameter val decodedKey = KyberMath.byteDecode(encryptionKey.keyBytes, 12) val nttKeyVector = Array(parameter.K) { ShortArray(KyberConstants.N) } @@ -135,7 +135,7 @@ class KyberAgreement(kemKeyPair: KyberKEMKeyPair) { * @param cipherText Cipher Text from the second party. * @return ByteArray The original Plain Text. */ - private fun fromCipherText(decapsulationKey: KyberDecapsulationKey, cipherText: KyberCipherText): ByteArray { + internal fun fromCipherText(decapsulationKey: KyberDecapsulationKey, cipherText: KyberCipherText): ByteArray { val parameter = cipherText.parameter val coefficients = Array(cipherText.parameter.K) { ShortArray(KyberConstants.N) } diff --git a/src/commonTest/kotlin/asia/hombre/kyber/Tests.kt b/src/commonTest/kotlin/asia/hombre/kyber/Tests.kt index 3700d6c..f08db18 100644 --- a/src/commonTest/kotlin/asia/hombre/kyber/Tests.kt +++ b/src/commonTest/kotlin/asia/hombre/kyber/Tests.kt @@ -263,7 +263,7 @@ class Tests { val original = SecureRandom().nextBytesOf(32) val cipher = KyberAgreement.encapsulate(keyPairBob.encapsulationKey, original).cipherText - val recovered = KyberAgreement.decapsulate(keyPairBob.decapsulationKey, cipher) + val recovered = KyberAgreement.fromCipherText(keyPairBob.decapsulationKey, cipher) assertContentEquals(original, recovered, "PKE Encryption and Decryption for 512 failed at attempt $i!") } @@ -276,7 +276,7 @@ class Tests { val original = SecureRandom().nextBytesOf(32) val cipher = KyberAgreement.encapsulate(keyPairBob.encapsulationKey, original).cipherText - val recovered = KyberAgreement.decapsulate(keyPairBob.decapsulationKey, cipher) + val recovered = KyberAgreement.fromCipherText(keyPairBob.decapsulationKey, cipher) assertContentEquals(original, recovered, "PKE Encryption and Decryption for 768 failed at attempt $i!") } @@ -289,7 +289,7 @@ class Tests { val original = SecureRandom().nextBytesOf(32) val cipher = KyberAgreement.encapsulate(keyPairBob.encapsulationKey, original).cipherText - val recovered = KyberAgreement.decapsulate(keyPairBob.decapsulationKey, cipher) + val recovered = KyberAgreement.fromCipherText(keyPairBob.decapsulationKey, cipher) assertContentEquals(original, recovered, "PKE Encryption and Decryption for 1024 failed at attempt $i!") }