forked from openrs2/openrs2
parent
eef6d31753
commit
676be4a0cb
@ -0,0 +1,45 @@ |
|||||||
|
package dev.openrs2.crypto |
||||||
|
|
||||||
|
import io.netty.buffer.ByteBuf |
||||||
|
|
||||||
|
private const val GOLDEN_RATIO = 0x9e3779b9.toInt() |
||||||
|
private const val ROUNDS = 32 |
||||||
|
|
||||||
|
fun ByteBuf.xteaEncrypt(index: Int, length: Int, key: IntArray) { |
||||||
|
require(key.size == 4) |
||||||
|
|
||||||
|
for (i in index until index + length step 8) { |
||||||
|
var sum = 0 |
||||||
|
var v0 = getInt(i) |
||||||
|
var v1 = getInt(i + 4) |
||||||
|
|
||||||
|
for (j in 0 until ROUNDS) { |
||||||
|
v0 += (((v1 shl 4) xor (v1 ushr 5)) + v1) xor (sum + key[sum and 3]) |
||||||
|
sum += GOLDEN_RATIO |
||||||
|
v1 += (((v0 shl 4) xor (v0 ushr 5)) + v0) xor (sum + key[(sum ushr 11) and 3]) |
||||||
|
} |
||||||
|
|
||||||
|
setInt(i, v0) |
||||||
|
setInt(i + 4, v1) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
fun ByteBuf.xteaDecrypt(index: Int, length: Int, key: IntArray) { |
||||||
|
require(key.size == 4) |
||||||
|
|
||||||
|
for (i in index until index + length step 8) { |
||||||
|
@Suppress("INTEGER_OVERFLOW") |
||||||
|
var sum = GOLDEN_RATIO * ROUNDS |
||||||
|
var v0 = getInt(i) |
||||||
|
var v1 = getInt(i + 4) |
||||||
|
|
||||||
|
for (j in 0 until ROUNDS) { |
||||||
|
v1 -= (((v0 shl 4) xor (v0 ushr 5)) + v0) xor (sum + key[(sum ushr 11) and 3]) |
||||||
|
sum -= GOLDEN_RATIO |
||||||
|
v0 -= (((v1 shl 4) xor (v1 ushr 5)) + v1) xor (sum + key[sum and 3]) |
||||||
|
} |
||||||
|
|
||||||
|
setInt(i, v0) |
||||||
|
setInt(i + 4, v1) |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,63 @@ |
|||||||
|
package dev.openrs2.crypto |
||||||
|
|
||||||
|
import io.netty.buffer.ByteBufUtil |
||||||
|
import io.netty.buffer.Unpooled |
||||||
|
import kotlin.test.Test |
||||||
|
import kotlin.test.assertEquals |
||||||
|
|
||||||
|
object XteaTest { |
||||||
|
private class TestVector(key: String, plaintext: String, ciphertext: String) { |
||||||
|
val key = IntArray(4) { |
||||||
|
Integer.parseUnsignedInt(key, it * 8, it * 8 + 8, 16) |
||||||
|
} |
||||||
|
val plaintext: ByteArray = ByteBufUtil.decodeHexDump(plaintext) |
||||||
|
val ciphertext: ByteArray = ByteBufUtil.decodeHexDump(ciphertext) |
||||||
|
} |
||||||
|
|
||||||
|
private val TEST_VECTORS = listOf( |
||||||
|
TestVector("000102030405060708090a0b0c0d0e0f", "4142434445464748", "497df3d072612cb5"), |
||||||
|
TestVector("000102030405060708090a0b0c0d0e0f", "4141414141414141", "e78f2d13744341d8"), |
||||||
|
TestVector("000102030405060708090a0b0c0d0e0f", "5a5b6e278948d77f", "4141414141414141"), |
||||||
|
TestVector("00000000000000000000000000000000", "4142434445464748", "a0390589f8b8efa5"), |
||||||
|
TestVector("00000000000000000000000000000000", "4141414141414141", "ed23375a821a8c2d"), |
||||||
|
TestVector("00000000000000000000000000000000", "70e1225d6e4e7655", "4141414141414141") |
||||||
|
) |
||||||
|
|
||||||
|
@Test |
||||||
|
fun testEncrypt() { |
||||||
|
for (vector in TEST_VECTORS) { |
||||||
|
val buffer = Unpooled.copiedBuffer(vector.plaintext) |
||||||
|
try { |
||||||
|
buffer.xteaEncrypt(0, buffer.readableBytes(), vector.key) |
||||||
|
|
||||||
|
val expected = Unpooled.wrappedBuffer(vector.ciphertext) |
||||||
|
try { |
||||||
|
assertEquals(expected, buffer) |
||||||
|
} finally { |
||||||
|
expected.release() |
||||||
|
} |
||||||
|
} finally { |
||||||
|
buffer.release() |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
fun testDecrypt() { |
||||||
|
for (vector in TEST_VECTORS) { |
||||||
|
val buffer = Unpooled.copiedBuffer(vector.ciphertext) |
||||||
|
try { |
||||||
|
buffer.xteaDecrypt(0, buffer.readableBytes(), vector.key) |
||||||
|
|
||||||
|
val expected = Unpooled.wrappedBuffer(vector.plaintext) |
||||||
|
try { |
||||||
|
assertEquals(expected, buffer) |
||||||
|
} finally { |
||||||
|
expected.release() |
||||||
|
} |
||||||
|
} finally { |
||||||
|
buffer.release() |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue