Add size return to verify packet length

82f1fda6 · Him188 · 6b93ce3c · 82f1fda6
Commit 82f1fda6 authored Jan 01, 2020 by Him188
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/io/OutputUtils.kt ...commonMain/kotlin/net.mamoe.mirai/utils/io/OutputUtils.kt +2 -1

No files found.
--- a/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/io/OutputUtils.kt
+++ b/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/io/OutputUtils.kt
@@ -38,9 +38,10 @@ fun BytePacketBuilder.writeShortLVByteArrayLimitedLength(array: ByteArray, maxLe
    }
 }

-fun BytePacketBuilder.writeShortLVByteArray(byteArray: ByteArray) {
+fun BytePacketBuilder.writeShortLVByteArray(byteArray: ByteArray): Int {
    this.writeShort(byteArray.size.toShort())
    this.writeFully(byteArray)
+    return byteArray.size
 }

 fun BytePacketBuilder.writeShortLVPacket(tag: UByte? = null, lengthOffset: ((Long) -> Long)? = null, builder: BytePacketBuilder.() -> Unit): Int =