From c9829f9759327d07cb0b2776761e797a25f0dc80 Mon Sep 17 00:00:00 2001
From: Miguel Oliveira <miguel.oliveira4224@gmail.com>
Date: Sun, 10 Apr 2022 17:39:30 -0300
Subject: [PATCH] Move BLAKE3 expand function

---
 blake3.lua | 34 +++++++++++++++++-----------------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/blake3.lua b/blake3.lua
index 0f4c6d4..3e56bd5 100644
--- a/blake3.lua
+++ b/blake3.lua
@@ -118,6 +118,23 @@ local function merge(cvl, cvr)
     return cvl
 end
 
+local function expand(out, len, offset)
+    expect(1, out, "table")
+    expect(1, len, "number")
+    expect(2, offset, "nil", "number")
+    offset = offset or 0
+
+    -- Expand output.
+    local out = {}
+    for i = 0, len / 64 do
+        local n = offset + i
+        local md = compress(out.cv, out.m, n, out.n, out.f, true)
+        out[i + 1] = ("<I4I4I4I4I4I4I4I4I4I4I4I4I4I4I4I4"):pack(unpack(md))
+    end
+
+    return table.concat(out):sub(1, len)
+end
+
 local function update(state, message)
     expect(1, state, "table")
     expect(1, message, "string")
@@ -203,23 +220,6 @@ local function finalize(state)
     end
 end
 
-local function expand(out, len, offset)
-    expect(1, out, "table")
-    expect(1, len, "number")
-    expect(2, offset, "nil", "number")
-    offset = offset or 0
-
-    -- Expand output.
-    local out = {}
-    for i = 0, len / 64 do
-        local n = offset + i
-        local md = compress(out.cv, out.m, n, out.n, out.f, true)
-        out[i + 1] = ("<I4I4I4I4I4I4I4I4I4I4I4I4I4I4I4I4"):pack(unpack(md))
-    end
-
-    return table.concat(out):sub(1, len)
-end
-
 local function copy(state)
     -- Copy CV stack.
     local cvs = {}