Combine RSA encrypt and decrypt methods into a single crypt method

This allows us to speed up master index signing with the CRT.

Signed-off-by: Graham <gpe@openrs2.org>
pull/132/head
Graham 3 years ago
parent 619424321f
commit 98d25539b9
  1. 7
      cache/src/main/kotlin/org/openrs2/cache/Js5MasterIndex.kt
  2. 14
      crypto/src/main/kotlin/org/openrs2/crypto/Rsa.kt
  3. 14
      crypto/src/test/kotlin/org/openrs2/crypto/RsaTest.kt

@ -7,8 +7,7 @@ import org.openrs2.buffer.crc32
import org.openrs2.buffer.use import org.openrs2.buffer.use
import org.openrs2.crypto.Rsa import org.openrs2.crypto.Rsa
import org.openrs2.crypto.Whirlpool import org.openrs2.crypto.Whirlpool
import org.openrs2.crypto.rsaDecrypt import org.openrs2.crypto.rsaCrypt
import org.openrs2.crypto.rsaEncrypt
import org.openrs2.crypto.whirlpool import org.openrs2.crypto.whirlpool
public data class Js5MasterIndex( public data class Js5MasterIndex(
@ -92,7 +91,7 @@ public data class Js5MasterIndex(
plaintext.writeByte(Rsa.MAGIC) plaintext.writeByte(Rsa.MAGIC)
plaintext.writeBytes(digest) plaintext.writeBytes(digest)
plaintext.rsaEncrypt(key).use { ciphertext -> plaintext.rsaCrypt(key).use { ciphertext ->
buf.writeBytes(ciphertext) buf.writeBytes(ciphertext)
} }
} }
@ -218,7 +217,7 @@ public data class Js5MasterIndex(
private fun decrypt(buf: ByteBuf, key: RSAKeyParameters?): ByteBuf { private fun decrypt(buf: ByteBuf, key: RSAKeyParameters?): ByteBuf {
return if (key != null) { return if (key != null) {
buf.rsaDecrypt(key) buf.rsaCrypt(key)
} else { } else {
buf.retain() buf.retain()
} }

@ -40,12 +40,8 @@ private fun BigInteger.toByteBuf(): ByteBuf {
return Unpooled.wrappedBuffer(toByteArray()) return Unpooled.wrappedBuffer(toByteArray())
} }
public fun ByteBuf.rsaEncrypt(key: RSAKeyParameters): ByteBuf { public fun ByteBuf.rsaCrypt(key: RSAKeyParameters): ByteBuf {
return Rsa.encrypt(toBigInteger(), key).toByteBuf() return Rsa.crypt(toBigInteger(), key).toByteBuf()
}
public fun ByteBuf.rsaDecrypt(key: RSAKeyParameters): ByteBuf {
return Rsa.decrypt(toBigInteger(), key).toByteBuf()
} }
public fun RSAPrivateCrtKeyParameters.toKeySpec(): KeySpec { public fun RSAPrivateCrtKeyParameters.toKeySpec(): KeySpec {
@ -87,10 +83,6 @@ public object Rsa {
return Pair(keyPair.public as RSAKeyParameters, keyPair.private as RSAPrivateCrtKeyParameters) return Pair(keyPair.public as RSAKeyParameters, keyPair.private as RSAPrivateCrtKeyParameters)
} }
public fun encrypt(plaintext: BigInteger, key: RSAKeyParameters): BigInteger {
return plaintext.modPow(key.exponent, key.modulus)
}
private fun generateBlindingFactor(m: BigInteger): Pair<BigInteger, BigInteger> { private fun generateBlindingFactor(m: BigInteger): Pair<BigInteger, BigInteger> {
val max = m - BigInteger.ONE val max = m - BigInteger.ONE
@ -105,7 +97,7 @@ public object Rsa {
} }
} }
public fun decrypt(ciphertext: BigInteger, key: RSAKeyParameters): BigInteger { public fun crypt(ciphertext: BigInteger, key: RSAKeyParameters): BigInteger {
if (key is RSAPrivateCrtKeyParameters) { if (key is RSAPrivateCrtKeyParameters) {
// blind the input // blind the input
val e = key.publicExponent val e = key.publicExponent

@ -51,34 +51,34 @@ object RsaTest {
val (public, private) = Rsa.generateKeyPair(Rsa.CLIENT_KEY_LENGTH) val (public, private) = Rsa.generateKeyPair(Rsa.CLIENT_KEY_LENGTH)
val expectedPlaintext = BigInteger("1337") val expectedPlaintext = BigInteger("1337")
val ciphertext = Rsa.encrypt(expectedPlaintext, public) val ciphertext = Rsa.crypt(expectedPlaintext, public)
val actualPlaintext = Rsa.decrypt(ciphertext, private) val actualPlaintext = Rsa.crypt(ciphertext, private)
assertEquals(expectedPlaintext, actualPlaintext) assertEquals(expectedPlaintext, actualPlaintext)
} }
@Test @Test
fun testEncrypt() { fun testEncrypt() {
val ciphertext = Rsa.encrypt(BigInteger("65"), PUBLIC_KEY) val ciphertext = Rsa.crypt(BigInteger("65"), PUBLIC_KEY)
assertEquals(BigInteger("2790"), ciphertext) assertEquals(BigInteger("2790"), ciphertext)
} }
@Test @Test
fun testDecrypt() { fun testDecrypt() {
val ciphertext = Rsa.decrypt(BigInteger("2790"), PRIVATE_KEY) val ciphertext = Rsa.crypt(BigInteger("2790"), PRIVATE_KEY)
assertEquals(BigInteger("65"), ciphertext) assertEquals(BigInteger("65"), ciphertext)
} }
@Test @Test
fun testDecryptCrt() { fun testDecryptCrt() {
val ciphertext = Rsa.decrypt(BigInteger("2790"), PRIVATE_KEY_CRT) val ciphertext = Rsa.crypt(BigInteger("2790"), PRIVATE_KEY_CRT)
assertEquals(BigInteger("65"), ciphertext) assertEquals(BigInteger("65"), ciphertext)
} }
@Test @Test
fun testEncryptByteBuf() { fun testEncryptByteBuf() {
wrappedBuffer(65).use { plaintext -> wrappedBuffer(65).use { plaintext ->
plaintext.rsaEncrypt(PUBLIC_KEY).use { ciphertext -> plaintext.rsaCrypt(PUBLIC_KEY).use { ciphertext ->
wrappedBuffer(10, 230.toByte()).use { expectedCiphertext -> wrappedBuffer(10, 230.toByte()).use { expectedCiphertext ->
assertEquals(expectedCiphertext, ciphertext) assertEquals(expectedCiphertext, ciphertext)
} }
@ -89,7 +89,7 @@ object RsaTest {
@Test @Test
fun testDecryptByteBuf() { fun testDecryptByteBuf() {
wrappedBuffer(10, 230.toByte()).use { ciphertext -> wrappedBuffer(10, 230.toByte()).use { ciphertext ->
ciphertext.rsaDecrypt(PRIVATE_KEY).use { plaintext -> ciphertext.rsaCrypt(PRIVATE_KEY).use { plaintext ->
wrappedBuffer(65).use { expectedPlaintext -> wrappedBuffer(65).use { expectedPlaintext ->
assertEquals(expectedPlaintext, plaintext) assertEquals(expectedPlaintext, plaintext)
} }

Loading…
Cancel
Save