github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-09 18:00:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Public TEA

Browse Source
This commit is contained in:
Him188 2019-11-08 20:50:56 +08:00
parent 41b2948832
commit 3eca212bf6
1 changed files with 15 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/TEA.kt
View File

@ -25,11 +25,9 @@ class DecryptionFailedException : Exception()
* @param key 长度至少为 16 * @param key 长度至少为 16
* @throws DecryptionFailedException 解密错误时 * @throws DecryptionFailedException 解密错误时
*/ */
@PublishedApi fun ByteArray.encryptBy(key: ByteArray, length: Int = this.size): ByteArray = TEA.encrypt(this, key, sourceLength = length)
internal fun ByteArray.encryptBy(key: ByteArray, length: Int = this.size): ByteArray = TEA.encrypt(this, key, sourceLength = length)
@PublishedApi fun ByteArray.encryptBy(key: DecrypterByteArray, length: Int = this.size): ByteArray = TEA.encrypt(this, key.value, sourceLength = length)
internal fun ByteArray.encryptBy(key: DecrypterByteArray, length: Int = this.size): ByteArray = TEA.encrypt(this, key.value, sourceLength = length)
/** /**
* 通过 [String.hexToBytes] 将 [keyHex] 转为 [ByteArray] 后用它解密 [this]. * 通过 [String.hexToBytes] 将 [keyHex] 转为 [ByteArray] 后用它解密 [this].
@ -38,8 +36,7 @@ internal fun ByteArray.encryptBy(key: DecrypterByteArray, length: Int = this.siz
* @param keyHex 长度至少为 16 bytes * @param keyHex 长度至少为 16 bytes
* @throws DecryptionFailedException 解密错误时 * @throws DecryptionFailedException 解密错误时
*/ */
@PublishedApi fun ByteArray.encryptBy(keyHex: String, length: Int = this.size): ByteArray = encryptBy(keyHex.hexToBytes(withCache = true), length = length)
internal fun ByteArray.encryptBy(keyHex: String, length: Int = this.size): ByteArray = encryptBy(keyHex.hexToBytes(withCache = true), length = length)
/** /**
* 在 [ByteArrayPool] 缓存 [this], 然后使用 [key] 加密. * 在 [ByteArrayPool] 缓存 [this], 然后使用 [key] 加密.
@ -48,8 +45,7 @@ internal fun ByteArray.encryptBy(keyHex: String, length: Int = this.size): ByteA
* @consumer 由于缓存需要被回收, 需在方法内执行解密后明文的消耗过程 * @consumer 由于缓存需要被回收, 需在方法内执行解密后明文的消耗过程
* @throws DecryptionFailedException 解密错误时 * @throws DecryptionFailedException 解密错误时
*/ */
@PublishedApi inline fun ByteReadPacket.encryptBy(key: ByteArray, offset: Int = 0, length: Int = remaining.toInt() - offset, consumer: (ByteArray) -> Unit) {
internal inline fun ByteReadPacket.encryptBy(key: ByteArray, offset: Int = 0, length: Int = remaining.toInt() - offset, consumer: (ByteArray) -> Unit) {
ByteArrayPool.useInstance { ByteArrayPool.useInstance {
this.readFully(it, offset, length) this.readFully(it, offset, length)
consumer(it.encryptBy(key, length = length)) consumer(it.encryptBy(key, length = length))
@ -67,8 +63,7 @@ internal inline fun ByteReadPacket.encryptBy(key: ByteArray, offset: Int = 0, le
* @param key 固定长度 16 * @param key 固定长度 16
* @throws DecryptionFailedException 解密错误时 * @throws DecryptionFailedException 解密错误时
*/ */
@PublishedApi fun ByteArray.decryptBy(key: ByteArray, length: Int = this.size): ByteArray =
internal fun ByteArray.decryptBy(key: ByteArray, length: Int = this.size): ByteArray =
TEA.decrypt(checkDataLengthAndReturnSelf(length), key, sourceLength = length) TEA.decrypt(checkDataLengthAndReturnSelf(length), key, sourceLength = length)
/** /**
@ -79,8 +74,7 @@ internal fun ByteArray.decryptBy(key: ByteArray, length: Int = this.size): ByteA
* @param key 长度至少为 16 * @param key 长度至少为 16
* @throws DecryptionFailedException 解密错误时 * @throws DecryptionFailedException 解密错误时
*/ */
@PublishedApi fun ByteArray.decryptBy(key: IoBuffer, length: Int = this.size): ByteArray {
internal fun ByteArray.decryptBy(key: IoBuffer, length: Int = this.size): ByteArray {
checkDataLengthAndReturnSelf(length) checkDataLengthAndReturnSelf(length)
return ByteArrayPool.useInstance { keyBuffer -> return ByteArrayPool.useInstance { keyBuffer ->
key.readFully(keyBuffer, 0, key.readRemaining) key.readFully(keyBuffer, 0, key.readRemaining)
@ -95,8 +89,7 @@ internal fun ByteArray.decryptBy(key: IoBuffer, length: Int = this.size): ByteAr
* @param keyHex 长度至少为 16 bytes * @param keyHex 长度至少为 16 bytes
* @throws DecryptionFailedException 解密错误时 * @throws DecryptionFailedException 解密错误时
*/ */
@PublishedApi fun ByteArray.decryptBy(keyHex: String, length: Int = this.size): ByteArray = decryptBy(keyHex.hexToBytes(withCache = true), length = length)
internal fun ByteArray.decryptBy(keyHex: String, length: Int = this.size): ByteArray = decryptBy(keyHex.hexToBytes(withCache = true), length = length)
/** /**
* 在 [ByteArrayPool] 缓存 [this], 然后使用 [key] 解密. * 在 [ByteArrayPool] 缓存 [this], 然后使用 [key] 解密.
@ -104,8 +97,7 @@ internal fun ByteArray.decryptBy(keyHex: String, length: Int = this.size): ByteA
* @param key 长度至少为 16 * @param key 长度至少为 16
* @throws DecryptionFailedException 解密错误时 * @throws DecryptionFailedException 解密错误时
*/ */
@PublishedApi fun IoBuffer.decryptBy(key: ByteArray, offset: Int = 0, length: Int = readRemaining - offset): ByteArray {
internal fun IoBuffer.decryptBy(key: ByteArray, offset: Int = 0, length: Int = readRemaining - offset): ByteArray {
return ByteArrayPool.useInstance { return ByteArrayPool.useInstance {
this.readFully(it, offset, length) this.readFully(it, offset, length)
it.checkDataLengthAndReturnSelf(length) it.checkDataLengthAndReturnSelf(length)
@ -119,8 +111,7 @@ internal fun IoBuffer.decryptBy(key: ByteArray, offset: Int = 0, length: Int = r
* @param keyHex 长度至少为 16 * @param keyHex 长度至少为 16
* @throws DecryptionFailedException 解密错误时 * @throws DecryptionFailedException 解密错误时
*/ */
@PublishedApi fun IoBuffer.decryptBy(keyHex: String, offset: Int = 0, length: Int = readRemaining - offset): ByteArray =
internal fun IoBuffer.decryptBy(keyHex: String, offset: Int = 0, length: Int = readRemaining - offset): ByteArray =
decryptBy(keyHex.hexToBytes(withCache = true), offset = offset, length = length) decryptBy(keyHex.hexToBytes(withCache = true), offset = offset, length = length)
@ -128,29 +119,23 @@ internal fun IoBuffer.decryptBy(keyHex: String, offset: Int = 0, length: Int = r
// region ByteReadPacket extension // region ByteReadPacket extension
@PublishedApi fun ByteReadPacket.decryptBy(key: ByteArray): ByteReadPacket = decryptAsByteArray(key) { data -> ByteReadPacket(data, 0) }
internal fun ByteReadPacket.decryptBy(key: ByteArray): ByteReadPacket = decryptAsByteArray(key) { data -> ByteReadPacket(data, 0) }
@PublishedApi fun ByteReadPacket.decryptBy(key: IoBuffer): ByteReadPacket = decryptAsByteArray(key) { data -> ByteReadPacket(data, 0) }
internal fun ByteReadPacket.decryptBy(key: IoBuffer): ByteReadPacket = decryptAsByteArray(key) { data -> ByteReadPacket(data, 0) }
@PublishedApi fun ByteReadPacket.decryptBy(keyHex: String): ByteReadPacket = decryptBy(keyHex.hexToBytes())
internal fun ByteReadPacket.decryptBy(keyHex: String): ByteReadPacket = decryptBy(keyHex.hexToBytes())
@PublishedApi inline fun <R> ByteReadPacket.decryptAsByteArray(key: ByteArray, consumer: (ByteArray) -> R): R =
internal inline fun <R> ByteReadPacket.decryptAsByteArray(key: ByteArray, consumer: (ByteArray) -> R): R =
ByteArrayPool.useInstance { ByteArrayPool.useInstance {
val length = remaining.toInt() val length = remaining.toInt()
readFully(it, 0, length) readFully(it, 0, length)
consumer(it.decryptBy(key, length)) consumer(it.decryptBy(key, length))
}.also { close() } }.also { close() }
@PublishedApi inline fun <R> ByteReadPacket.decryptAsByteArray(keyHex: String, consumer: (ByteArray) -> R): R =
internal inline fun <R> ByteReadPacket.decryptAsByteArray(keyHex: String, consumer: (ByteArray) -> R): R =
this.decryptAsByteArray(keyHex.hexToBytes(), consumer) this.decryptAsByteArray(keyHex.hexToBytes(), consumer)
@PublishedApi inline fun <R> ByteReadPacket.decryptAsByteArray(key: IoBuffer, consumer: (ByteArray) -> R): R =
internal inline fun <R> ByteReadPacket.decryptAsByteArray(key: IoBuffer, consumer: (ByteArray) -> R): R =
ByteArrayPool.useInstance { ByteArrayPool.useInstance {
val length = remaining.toInt() val length = remaining.toInt()
readFully(it, 0, length) readFully(it, 0, length)