Implement streaming API

2026-03-11 18:53:00 +01:00
parent cb3ff7ec6c
commit 328d4b957e
34 changed files with 936 additions and 593 deletions
--- a/src/commonMain/kotlin/com/infendro/hash/md/MD5.kt
+++ b/src/commonMain/kotlin/com/infendro/hash/md/MD5.kt
@@ -1,79 +1,44 @@
 package com.infendro.hash.md

 import com.infendro.bytes.ByteOrder.LITTLE_ENDIAN
-import com.infendro.bytes.bytearray.toIntArray
+import com.infendro.bytes.bytearray.copyInto
 import com.infendro.bytes.intarray.copyInto
-import com.infendro.bytes.long.copyInto
 import com.infendro.hash.HashFunction
-import com.infendro.hash.util.align

-object MD5 : HashFunction {
-    override val bytes: Int
-        get() = 16
+class MD5 : HashFunction(16, 64) {
+    private val h = initial.copyOf()

-    override val block: Int
-        get() = 64
+    private val w = IntArray(16)
+    private val v = IntArray(4)

-    private val initial = intArrayOf(
-        +0x67452301, -0x10325477,
-        -0x67452302, +0x10325476,
-    )
+    override fun reset() {
+        super.reset()
+        initial.copyInto(h)
+    }

-    private val c = intArrayOf(
-        -0x28955b88, -0x173848aa, +0x242070db, -0x3e423112,
-        -0x0a83f051, +0x4787c62a, -0x57cfb9ed, -0x02b96aff,
-        +0x698098d8, -0x74bb0851, -0x0000a44f, -0x76a32842,
-        +0x6b901122, -0x02678e6d, -0x5986bc72, +0x49b40821,
-        -0x09e1da9e, -0x3fbf4cc0, +0x265e5a51, -0x16493856,
-        -0x29d0efa3, +0x02441453, -0x275e197f, -0x182c0438,
-        +0x21e1cde6, -0x3cc8f82a, -0x0b2af279, +0x455a14ed,
-        -0x561c16fb, -0x03105c08, +0x676f02d9, -0x72d5b376,
-        -0x0005c6be, -0x788e097f, +0x6d9d6122, -0x021ac7f4,
-        -0x5b4115bc, +0x4bdecfa9, -0x0944b4a0, -0x41404390,
-        +0x289b7ec6, -0x155ed806, -0x2b10cf7b, +0x04881d05,
-        -0x262b2fc7, -0x1924661b, +0x1fa27cf8, -0x3b53a99b,
-        -0x0bd6ddbc, +0x432aff97, -0x546bdc59, -0x036c5fc7,
-        +0x655b59c3, -0x70f3336e, -0x00100b83, -0x7a7ba22f,
-        +0x6fa87e4f, -0x01d31920, -0x5cfebcec, +0x4e0811a1,
-        -0x08ac817e, -0x42c50dcb, +0x2ad7d2bb, -0x14792c6f,
-    )
+    override fun process() = process(buffer.value)

-    private val shifts = intArrayOf(
-        7, 12, 17, 22, 7, 12, 17, 22,
-        7, 12, 17, 22, 7, 12, 17, 22,
-        5, 9, 14, 20, 5, 9, 14, 20,
-        5, 9, 14, 20, 5, 9, 14, 20,
-        4, 11, 16, 23, 4, 11, 16, 23,
-        4, 11, 16, 23, 4, 11, 16, 23,
-        6, 10, 15, 21, 6, 10, 15, 21,
-        6, 10, 15, 21, 6, 10, 15, 21,
-    )
-
-    override fun hashInto(value: ByteArray, destination: ByteArray, destinationOffset: Int) {
-        val input = pad(value)
-        val h = initial.copyOf()
-
-        for (i in input.indices step block) {
-            val block = input.sliceArray(i..<(i + block))
-            process(h, block)
+    private fun finalize() {
+        buffer.append(0x80.toByte())
+        if (buffer.free < 8) {
+            process()
+            buffer.reset()
        }
+        buffer.jumpTo(56)
+        buffer.append(length * 8, LITTLE_ENDIAN)
+
+        process()
+    }
+
+    override fun digestInto(destination: ByteArray, destinationOffset: Int) {
+        finalize()
        h.copyInto(destination, destinationOffset, order = LITTLE_ENDIAN)
+        reset()
    }

-    private fun pad(value: ByteArray): ByteArray {
-        val length = value.size * 8L
-
-        val size = (value.size + 9).align(block)
-        return ByteArray(size).also {
-            value.copyInto(it)
-            it[value.size] = -0x80
-            length.copyInto(it, size - 8, order = LITTLE_ENDIAN)
-        }
-    }
-
-    private fun process(h: IntArray, block: ByteArray) {
-        val w = block.toIntArray(order = LITTLE_ENDIAN)
-        val v = h.copyOf()
+    private fun process(block: ByteArray) {
+        block.copyInto(w, order = LITTLE_ENDIAN)
+        h.copyInto(v)

        repeat(64) { i ->
            var f = when (i) {
@@ -99,4 +64,49 @@ object MD5 : HashFunction {
            h[i] += v[i]
        }
    }
+
+    companion object {
+        private val initial = intArrayOf(
+            +0x67452301, -0x10325477,
+            -0x67452302, +0x10325476,
+        )
+
+        private val c = intArrayOf(
+            -0x28955b88, -0x173848aa, +0x242070db, -0x3e423112,
+            -0x0a83f051, +0x4787c62a, -0x57cfb9ed, -0x02b96aff,
+            +0x698098d8, -0x74bb0851, -0x0000a44f, -0x76a32842,
+            +0x6b901122, -0x02678e6d, -0x5986bc72, +0x49b40821,
+            -0x09e1da9e, -0x3fbf4cc0, +0x265e5a51, -0x16493856,
+            -0x29d0efa3, +0x02441453, -0x275e197f, -0x182c0438,
+            +0x21e1cde6, -0x3cc8f82a, -0x0b2af279, +0x455a14ed,
+            -0x561c16fb, -0x03105c08, +0x676f02d9, -0x72d5b376,
+            -0x0005c6be, -0x788e097f, +0x6d9d6122, -0x021ac7f4,
+            -0x5b4115bc, +0x4bdecfa9, -0x0944b4a0, -0x41404390,
+            +0x289b7ec6, -0x155ed806, -0x2b10cf7b, +0x04881d05,
+            -0x262b2fc7, -0x1924661b, +0x1fa27cf8, -0x3b53a99b,
+            -0x0bd6ddbc, +0x432aff97, -0x546bdc59, -0x036c5fc7,
+            +0x655b59c3, -0x70f3336e, -0x00100b83, -0x7a7ba22f,
+            +0x6fa87e4f, -0x01d31920, -0x5cfebcec, +0x4e0811a1,
+            -0x08ac817e, -0x42c50dcb, +0x2ad7d2bb, -0x14792c6f,
+        )
+
+        private val shifts = intArrayOf(
+            7, 12, 17, 22,
+            7, 12, 17, 22,
+            7, 12, 17, 22,
+            7, 12, 17, 22,
+            5, 9, 14, 20,
+            5, 9, 14, 20,
+            5, 9, 14, 20,
+            5, 9, 14, 20,
+            4, 11, 16, 23,
+            4, 11, 16, 23,
+            4, 11, 16, 23,
+            4, 11, 16, 23,
+            6, 10, 15, 21,
+            6, 10, 15, 21,
+            6, 10, 15, 21,
+            6, 10, 15, 21,
+        )
+    }
 }