Cleaning residue plus minor jmp advantage

helper.s

@@ -18,14 +18,13 @@ clear_heap_block_loop:
   mov byte ptr [rdi+rcx], 0
   inc rcx
   cmp rcx, rsi
-  jle clear_heap_block_loop
+  jl clear_heap_block_loop
   ret 
 
 # rdi = size in octets
 brk_alloc:
   push rbx
   push r12
-  inc rdi      # so you get all of the requested space
   mov rbx, rdi
 
   mov rax, 12
@@ -65,6 +64,7 @@ read:
 # rsi = char*
 write:
   push r15
+  push r11
   mov r15, rdi
   mov rdi, rsi # no need to touch rsi
   call count
@@ -74,6 +74,7 @@ write:
   mov rdi, r15
 
   syscall
+  pop r11
   pop r15
   ret
 
@@ -81,9 +82,9 @@ write:
 count:
   mov rax, rdi
 count_loop:
-  mov r8b, [rax]
+  mov r9b, [rax]
   inc rax
-  cmp r8b, 0
+  cmp r9b, 0
   jnz count_loop
   dec rax
   sub rax, rdi

main.s

@@ -165,9 +165,22 @@ process_prnt:
   ret
 
 process_inpt:
+  # Gotta flush the buffer first
+  mov rdi, 1
+  mov rsi, r13
+  call write
+
+  # Clear Text buffer
+  mov rdi, r10
+  sub rdi, r13
+  mov rsi, rdi
+  mov rdi, r13
+  call clear_heap_block
+  mov r10, r13
+  
+  # Ask character
   mov rdi, r14
   add rdi, r15
-
   mov rsi, rdi # char*
   xor rdi, rdi # fd
   mov rdx, 1   # size
@@ -176,32 +189,24 @@ process_inpt:
 
 # In stack it is 
 # A = Address to loop back
-# C = Cell to check
-# AC AC AC AC AC AC AC <- pop here
 process_loop_start:
   pop rdi
   push r11 # Address to loopback
-  push r15 # Cell to check
   push rdi
   ret 
 
 process_loop_end:
   # Gotta get ret value
   pop rsi
-  pop rdi # Cell
   pop rax # Address
   
   # if cell is 0 we are done with loop
-  cmp byte ptr [r14+rdi], 0
+  cmp byte ptr [r14+r15], 0
   je 1f
   mov r11, rax
 
   # Make sure stack isn't caught on fire
   push rax
-  push rdi
-  push rsi
-  ret
 1:
-  push rsi
-  ret # Discard
+  jmp rsi
 

tests/test1.bf

@@ -0,0 +1,2 @@
++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++.>
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++.

tests/test3.bf

@@ -0,0 +1 @@
+++++[>++++[>++++<-]<-]>>++.[<<+>>-]<<.

tests/test4.bf

@@ -0,0 +1 @@
+++++++++[>++++[>++>+++>+++>+<<<<-]>+>+>->>+[<]<-]>>.>---.+++++++..+++.>>.<-.<.+++.------.--------.>>+.>++.

tests/test5.bf

@@ -0,0 +1 @@
+++++++++[>++++++++<-]>+.,+.,,

welcome.bf

@@ -0,0 +1 @@
++++++++++[>++++++++>+++++++++++>++++>++++++++++>+++++++++++>++++>+++++++++++<<<<<<<-]>.>++.+++++++..+++.>---.-.>---.>++.<<<---.>>>--.<<<+++.--.>>>++.<<.<+++++++.-----.>.>>--.--.<<<+++++.>>>>+++.<<<<-.>.>>+.<<<-.>>>-.++++++++.+++++.>>+++.<<<<<+++.>>>>>---.<<---.<<.>>--.+++++.<<<-.>>>>>++.<<++++.--.++.>>.<<++.>>.<<--.>+++++.<<<.<+++.>>>.>>++++.<<++..>>----.<<------.<<.>>-----.+++++.<<.<<-------.>>>----.------.<+.>>>>>++++++++++.