[core] Fix temporary buffer acquiring; Fix #2309

mamoe · Nov 18, 2022 · 372c250 · 372c250
1 parent b38687a
commit 372c250
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mirai-core/src/commonMain/kotlin/utils/crypto/TEA.kt b/mirai-core/src/commonMain/kotlin/utils/crypto/TEA.kt
@@ -46,7 +46,7 @@ internal object TEA {
         length: Int = receiver.remaining.toInt() - offset,
         consumer: (ByteArray) -> Unit,
     ) {
-        ByteArrayPool.useInstance {
+        ByteArrayPool.useInstance(length) {
             receiver.readFully(it, offset, length)
             consumer(encrypt(it, key, length = length))
         }
@@ -68,7 +68,7 @@ internal object TEA {
         length: Int = (receiver.remaining - offset).toInt(),
         consumer: (ByteArray) -> R,
     ): R {
-        return ByteArrayPool.useInstance {
+        return ByteArrayPool.useInstance(length) {
             receiver.readFully(it, offset, length)
             consumer(decrypt(it, key, length))
         }.also { receiver.close() }