xyz.capybara.clamav.commands.scan.InStream.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of clamav-client Show documentation

Java Client for the ClamAV antivirus service.

There is a newer version: 2.1.2

package xyz.capybara.clamav.commands.scan

import xyz.capybara.clamav.commands.Command
import xyz.capybara.clamav.commands.scan.result.ScanResult
import xyz.capybara.clamav.CommunicationException

import java.io.IOException
import java.io.InputStream
import java.net.InetSocketAddress
import java.nio.ByteBuffer
import java.nio.ByteOrder
import java.nio.channels.SocketChannel

class InStream(private val inputStream: InputStream) : ScanCommand() {
    private val CHUNK_SIZE = 2048

    override val commandString: String
        get() = "INSTREAM"

    override val format: Command.CommandFormat
        get() = Command.CommandFormat.NULL_CHAR

    override fun send(server: InetSocketAddress): ScanResult {
        try {
            SocketChannel.open(server).use {
                it.write(rawCommand)

                // ByteBuffer order must be big-endian ( == network byte order)
                // It is, by default, but it doesn't hurt to set it anyway
                val length = ByteBuffer.allocate(4).order(ByteOrder.BIG_ENDIAN)
                val data = ByteArray(CHUNK_SIZE)
                var chunkSize = CHUNK_SIZE
                while (chunkSize == CHUNK_SIZE) {
                    chunkSize = inputStream.read(data)
                    if (chunkSize > 0) {
                        length.clear()
                        length.putInt(chunkSize).flip()
                        // The format of the chunk is: ''
                        it.write(length)
                        it.write(ByteBuffer.wrap(data, 0, chunkSize))
                    }
                }
                length.clear()
                // Terminate the stream by sending a zero-length chunk
                length.putInt(0).flip()
                it.write(length)

                return readResponse(it)
            }
        } catch (e: IOException) {
            throw CommunicationException(e)
        }

    }
}