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
|
