summaryrefslogtreecommitdiff
path: root/zap/src/mem/foreach.S
blob: fe965386a892cc5657ae2d8ddba170c6140b4fc7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/*
	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/.
*/

#include <zap/priv.h>

.globl zap_foreach

zap_foreach:

	/*
		void *  ptr
		zap_sz  sz
		zap_sz  num
		void (* fn)(void *)
	*/

#if defined(__amd64__)

	# rbx: Address of the current element.
	# r12: Address of the element after the last input element.
	# r13: Size of each input element.
	# r14: Address of the function.

	# We're gonna use callee-saved registers for storing values so they don't get overwritten with each function call.

	# Push the callee-saved registers:
	pushq %rbx
	pushq %r12
	pushq %r13
	pushq %r14

	# Move registers into place:
	movq %rdi,%rbx
	movq %rsi,%r13
	movq %rcx,%r14

	# Get the one-past-the-end address:
	movq %rdx,%r12
	imulq %r13,%r12 # Calculate the array size in bytes (sz * num). We're using signed multiply because the equivalent using the unsigned instruction would use more instructions.
	addq %rbx,%r12

	# Iterate through the array:
.loop:

	# Check if we have reached the one-past-the-end address:
	cmpq %rbx,%r12
	je .done

	# Call the provided function:
	movq %rbx,%rdi # Provide the current address to the function.
	call *%r14 # We don't need to save any registers for this as we only use callee-saved registers.

	# Continue to the next element:
	addq %r13,%rbx
	jmp .loop

	# Finish:
.done:
	
	# Restore the callee-saved registers:
	popq %r14
	popq %r13
	popq %r12
	popq %rbx

	ret

#endif