74 lines
2.5 KiB
Kotlin
74 lines
2.5 KiB
Kotlin
package sushi.hardcore.aira.background_service
|
|
|
|
import java.nio.ByteBuffer
|
|
|
|
class Protocol {
|
|
companion object {
|
|
const val MESSAGE: Byte = 0x00
|
|
const val ASK_NAME: Byte = 0x01
|
|
const val TELL_NAME: Byte = 0x02
|
|
const val FILE: Byte = 0x03
|
|
const val ASK_LARGE_FILE: Byte = 0x04
|
|
const val ACCEPT_LARGE_FILE: Byte = 0x05
|
|
const val LARGE_FILE_CHUNK: Byte = 0x06
|
|
const val ACK_CHUNK: Byte = 0x07
|
|
const val ABORT_FILE_TRANSFER: Byte = 0x08
|
|
|
|
fun askName(): ByteArray {
|
|
return byteArrayOf(ASK_NAME)
|
|
}
|
|
|
|
fun tellName(name: String): ByteArray {
|
|
return byteArrayOf(TELL_NAME)+name.toByteArray()
|
|
}
|
|
|
|
fun newMessage(msg: String): ByteArray {
|
|
return byteArrayOf(MESSAGE)+msg.toByteArray()
|
|
}
|
|
|
|
fun newFile(fileName: String, buffer: ByteArray): ByteArray {
|
|
return byteArrayOf(FILE)+ByteBuffer.allocate(2).putShort(fileName.length.toShort()).array()+fileName.toByteArray()+buffer
|
|
}
|
|
|
|
fun askLargeFile(fileSize: Long, fileName: String): ByteArray {
|
|
return byteArrayOf(ASK_LARGE_FILE)+ByteBuffer.allocate(8).putLong(fileSize).array()+fileName.toByteArray()
|
|
}
|
|
|
|
fun acceptLargeFile(): ByteArray {
|
|
return byteArrayOf(ACCEPT_LARGE_FILE)
|
|
}
|
|
|
|
fun abortFileTransfer(): ByteArray {
|
|
return byteArrayOf(ABORT_FILE_TRANSFER)
|
|
}
|
|
|
|
fun ackChunk(): ByteArray {
|
|
return byteArrayOf(ACK_CHUNK)
|
|
}
|
|
|
|
class SmallFile(val rawFileName: ByteArray, val fileContent: ByteArray)
|
|
|
|
fun parseSmallFile(buffer: ByteArray): SmallFile? {
|
|
if (buffer.size > 3) {
|
|
val filenameLen = ByteBuffer.wrap(ByteArray(2) +buffer.sliceArray(1..2)).int
|
|
if (buffer.size > 3+filenameLen) {
|
|
val rawFileName = buffer.sliceArray(3 until 3+filenameLen)
|
|
return SmallFile(rawFileName, buffer.sliceArray(3+filenameLen until buffer.size))
|
|
}
|
|
}
|
|
return null
|
|
}
|
|
|
|
class FileInfo(val fileName: String, val fileSize: Long)
|
|
|
|
fun parseAskFile(buffer: ByteArray): FileInfo? {
|
|
return if (buffer.size > 9) {
|
|
val fileSize = ByteBuffer.wrap(buffer.sliceArray(1..8)).long
|
|
val fileName = buffer.sliceArray(9 until buffer.size).decodeToString()
|
|
FileInfo(fileName, fileSize)
|
|
} else {
|
|
null
|
|
}
|
|
}
|
|
}
|
|
} |