0

我想用 tcp 客户端将文件传输到服务器，但图像文件错误。

我的客户代码是

import com.sun.xml.internal.messaging.saaj.util.ByteOutputStream
import org.msgpack.core.MessageBufferPacker
import org.msgpack.core.MessagePack
import org.msgpack.core.MessageUnpacker
import java.io.*
import java.net.Socket
import java.util.*

fun main(args: Array<String>) {
    fileClient("localhost",1988,"fruit.jpg")
}

class fileClient (host:String, port:Int, file:String){
    var s : Socket ?= null
    var out = ByteArrayOutputStream()
    var msg : MessageBufferPacker = MessagePack.newDefaultBufferPacker()

    init {
        try {
            s = Socket(host,port)
            sendFile(file)
        }catch (e:Exception){
            e.printStackTrace()
        }
    }

    @Throws(IOException::class)
    fun sendFile(file: String) {
        val dos = DataOutputStream(s!!.getOutputStream())
        val buffer = ByteArray(4096)
        val filebytes = File(file).readBytes()
        var msgdata = ByteOutputStream()

        msg.packString(file)
        msg.packBinaryHeader(filebytes.size)
        msg.writePayload(filebytes)

        msg.close()



        val data = msg.toByteArray()
        val datasize = data.size
        val ins = ByteArrayInputStream(data)
        dos.writeInt(datasize)
        while (ins.read(buffer) > 0) {
            dos.write(buffer)
        }
        dos.close()
    }
}

我的服务器代码是

import com.sun.xml.internal.messaging.saaj.util.ByteOutputStream
import org.msgpack.core.MessagePack
import org.msgpack.core.MessageUnpacker
import java.awt.List
import java.io.*
import java.net.ServerSocket
import java.net.Socket
import java.text.SimpleDateFormat
import java.util.*

fun main(args: Array<String>) {
    var fs = FileServer(1988)
    fs.start()
}

class FileServer(port: Int) : Thread() {

    private var ss: ServerSocket? = null

    var fileRealName : String ?= null

    init {
        try {
            ss = ServerSocket(port)
        } catch (e: IOException) {
            e.printStackTrace()
        }

    }

    override fun run() {
        while (true) {
            try {
                val clientSock = ss!!.accept()
                saveFile(clientSock)
            } catch (e: IOException) {
                e.printStackTrace()
            }

        }
    }

    @Throws(IOException::class)
    private fun saveFile(clientSock: Socket) {
        var msgList = ArrayList<Any>()
        val dis = DataInputStream(clientSock.inputStream)
        val msgdata = ByteOutputStream()
        val buffer = ByteArray(4096)
        var read = 0


        while (true) {
            val datalen = dis.readInt() // data length

            if(datalen!= null && datalen >0){
                var finaldata = ByteArray(datalen)
                var process = 0;
                while (process <= datalen) {
                    read = dis.read(buffer)
                    if (read < 0) {
                        return
                    }
                    msgdata.write(buffer)
                    process += 4096
                }
                println(process.toString() + " "+ datalen.toString())
                var allData = msgdata.toByteArray().slice(0..datalen).toByteArray()
                unpackByte(allData)

            }

        }

        msgdata.close()
        dis.close()
    }

    private fun unpackByte(data:ByteArray){

        var unpacker : MessageUnpacker = MessagePack.newDefaultUnpacker(data)

        var fileName = unpacker.unpackString().toString()

        var filesize = unpacker.unpackBinaryHeader()
        var buffer = ByteArray(filesize)
        unpacker.readPayload(buffer)

        var fos = FileOutputStream(fileName)
        fos.write(buffer)
        fos.close()

        unpacker.close()

    }

}

我要传输的文件是 在此处输入图像描述

但是传输后，服务器上的图像是这样的。 在此处输入图像描述

如何正确传输此文件？

4

1 回答 1

1

我没有成功重现您的问题。但是我想我已经找到了可能导致文件传输损坏的错误。

在您的服务器代码中有一个无限循环，该循环会立即从方法中返回，使其余方法无法访问。这是关闭连接和流的清理代码。很可能OutputStream没有正确关闭，这是文件写入损坏的原因。

服务器代码应该是这样的：

val datalen = dis.readInt() // data length

if (datalen > 0) {
    var finaldata = ByteArray(datalen)
     var process = 0;
     while (process <= datalen) {
        read = dis.read(buffer)
        if (read < 0) {
            break
        }
        msgdata.write(buffer)
        process += 4096
    }
    println(process.toString() + " " + datalen.toString())
    val allData = msgdata.toByteArray().slice(0..datalen).toByteArray()
    unpackByte(allData)

}

msgdata.close()
dis.close()

while循环是不必要的。此外，您可能应该只是break循环，而不是return函数。

PS 你有没有考虑过使用IOUtils来处理所有的 IO 读/写？使用这个库只需几行代码就可以替换一半的代码。

于 2016-08-23T08:57:49.840 回答