1 files changed, 75 insertions, 0 deletions
diff --git a/zap/source/amd64/mem/memcnt.S b/zap/source/amd64/mem/memcnt.S
new file mode 100644
index 0000000..5323301
--- /dev/null
+++ b/zap/source/amd64/mem/memcnt.S
@@ -0,0 +1,75 @@
+# Copyright 2022 Gabriel Jensen.
+# This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.
+# If a copy of the MPL was not distributed with this file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+.globl zap_memcnt
+
+zap_memcnt:
+# Address of the current element:
+#define addr     %rbx
+# Address of the element after the last input element:
+#define afterbuf %r12
+# Size of each input element:
+#define sz       %r13
+# Address of the function:
+#define fn       %r14
+# Count:
+#define cnt      %r15
+
+	# Push the callee-saved registers:
+	pushq addr
+	pushq afterbuf
+	pushq sz
+	pushq fn
+	pushq cnt
+
+	# Move registers into place:
+	movq %rdi,addr
+	movq %rsi,sz
+	movq %rdx,afterbuf
+	movq %rcx,fn
+
+	# Get the one-past-the-end address:
+	imulq sz,afterbuf  # afterbuf *= sz
+	addq addr,afterbuf # afterbuf += addr
+
+	movq $0x0,cnt # cnt = 0x0
+
+	# Iterate through the array:
+.loop:
+
+	# Check if we have reached the one-past-the-end address:
+	cmpq addr,afterbuf # if (addr == afterbuf)
+	je .done           # goto done
+
+	# Call the provided function:
+	movq addr,%rdi
+	call *fn       # fn(addr)
+
+	# Check the return value of the function:
+	testb %al,%al # if (fn(addr) == zap_false)
+	jz .skip      # goto skip
+
+	incq cnt # ++cnt
+
+	# Skip the incrementation:
+.skip:
+
+	# Continue to the next element:
+	addq sz,addr # addr += sz
+	jmp .loop    # goto loop
+
+	# Finish:
+.done:
+
+	# Move count into return register:
+	movq cnt,%rax
+	
+	# Restore the callee-saved registers:
+	popq cnt
+	popq fn
+	popq sz
+	popq afterbuf
+	popq addr
+
+	ret # return cnt