use count more wisely in write

1 files changed, 2 insertions, 1 deletions

diff --git a/src/main/kotlin/Connection.kt b/src/main/kotlin/Connection.kt
index 564e8a7..6444ade 100644
--- a/src/main/kotlin/Connection.kt
+++ b/src/main/kotlin/Connection.kt
@@ -153,6 +153,7 @@ class Connection(transLay: TransportLayer) : ProtocolTranslator {
     }
 
     override fun write(fid: UInt, offset: ULong, count: UInt, data: Iterable<UByte>): Pair<String?, UInt> {
+        val data = data.take(count.toInt()).toTypedArray()
         val omsg = OutMessage(NinePMessageType.TWRITE, this.tagGen.generate(), listOf("offset", "count", "data"),
             mapOf(
                 "offset" to BigInteger(offset.toString()),
@@ -160,7 +161,7 @@ class Connection(transLay: TransportLayer) : ProtocolTranslator {
             ),
             emptyMap(),
             mapOf(
-                "data" to data.toList().toTypedArray()
+                "data" to data
             ),
             this.maxSize
         )