org.sol4k.Binary.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sol4k Show documentation
Kotlin, JVM, and Android client for Solana
There is a newer version: 0.5.4
package org.sol4k

object Binary {
    @JvmStatic
    fun uint32(value: Long): ByteArray {
        val byteArray = ByteArray(4)
        byteArray[0] = (0xFFL and value).toByte()
        byteArray[1] = (0xFFL and (value shr 8)).toByte()
        byteArray[2] = (0xFFL and (value shr 16)).toByte()
        byteArray[3] = (0xFFL and (value shr 24)).toByte()
        return byteArray
    }

    @JvmStatic
    fun int64(value: Long): ByteArray {
        val byteArray = ByteArray(8)
        byteArray[0] = (0xFFL and value).toByte()
        byteArray[1] = (0xFFL and (value shr 8)).toByte()
        byteArray[2] = (0xFFL and (value shr 16)).toByte()
        byteArray[3] = (0xFFL and (value shr 24)).toByte()
        byteArray[4] = (0xFFL and (value shr 32)).toByte()
        byteArray[5] = (0xFFL and (value shr 40)).toByte()
        byteArray[6] = (0xFFL and (value shr 48)).toByte()
        byteArray[7] = (0xFFL and (value shr 56)).toByte()
        return byteArray
    }

    @JvmStatic
    fun uint16(value: Int): ByteArray {
        val byteArray = ByteArray(2)
        byteArray[0] = (0xFF and value).toByte()
        byteArray[1] = (0xFF and (value shr 8)).toByte()
        return byteArray
    }

    @JvmStatic
    fun encodeLength(len: Int): ByteArray {
        val out = ByteArray(10)
        var remLen = len
        var cursor = 0
        while (true) {
            var elem = remLen and 0x7f
            remLen = remLen shr 7
            if (remLen == 0) {
                val uint16 = uint16(elem)
                out[cursor] = uint16[0]
                out[cursor + 1] = uint16[1]
                break
            } else {
                elem = elem or 0x80
                val uint16 = uint16(elem)
                out[cursor] = uint16[0]
                out[cursor + 1] = uint16[1]
                cursor += 1
            }
        }
        val bytes = ByteArray(cursor + 1)
        System.arraycopy(out, 0, bytes, 0, cursor + 1)
        return bytes
    }

    @JvmStatic
    fun decodeLength(bytes: ByteArray): DecodedLength {
        var newBytes = bytes
        var len = 0
        var size = 0
        while (true) {
            val elem = newBytes.first().toInt().also { newBytes = newBytes.drop(1).toByteArray() }
            len = len or ((elem and 0x7f) shl (size * 7))
            size += 1
            if ((elem and 0x80) == 0) {
                break
            }
        }
        return DecodedLength(len, newBytes)
    }

    data class DecodedLength(
        val length: Int,
        val bytes: ByteArray,
    ) {
        override fun equals(other: Any?): Boolean {
            if (this === other) return true
            if (javaClass != other?.javaClass) return false

            other as DecodedLength

            if (length != other.length) return false
            if (!bytes.contentEquals(other.bytes)) return false

            return true
        }

        override fun hashCode(): Int {
            var result = length
            result = 31 * result + bytes.contentHashCode()
            return result
        }
    }
}