Fix offset into array in set_bits_r

ChillerDragon · Aug 4, 2024 · b448300 · b448300
1 parent 2b4e0c1
commit b448300
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/huffman/set_bits_r.asm b/src/huffman/set_bits_r.asm
@@ -58,7 +58,9 @@ _huff_setbits_r:
     je ._huff_setbits_r_no_leaf1_recursion
 
     ; &m_aNodes[pNode->m_aLeafs[1]]
-    lea rax, [huff_nodes + rcx]
+    mov rbx, rcx
+    imul rbx, HUFF_CNODE_SIZE
+    lea rax, [huff_nodes + rbx]
 
     ; bits | (1<<Depth)
     mov r9, 1
@@ -81,7 +83,9 @@ _huff_setbits_r:
     je ._huff_setbits_r_no_leaf0_recursion
 
     ; &m_aNodes[pNode->m_aLeafs[0]]
-    lea rax, [huff_nodes + rcx]
+    mov rbx, rcx
+    imul rbx, HUFF_CNODE_SIZE
+    lea rax, [huff_nodes + rbx]
 
     ; bits are still in rdi