diff options
Diffstat (limited to 'rgo/src/memfill.c')
-rw-r--r-- | rgo/src/memfill.c | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/rgo/src/memfill.c b/rgo/src/memfill.c new file mode 100644 index 0000000..ec5e1fa --- /dev/null +++ b/rgo/src/memfill.c @@ -0,0 +1,55 @@ +/* + 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 <rgo-priv.h> + +#if defined(rgo_priv_fastimpl) +__asm__ ( + ".global rgo_memfill\n" + + "rgo_memfill:\n" + /* + void const * ptr + sus_typ_usz num + sus_typ_u8 val + */ +#if defined(sus_arch_amd64) + /* rdi: Address of the current element. */ + /* rsi: Address of the element after the last element. */ + "addq %rdi,%rsi\n" + ".loop:\n" + "cmpq %rsi,%rdi\n" + "je .done\n" /* Exit loop if we have reached the final element. */ + "movb %dl,(%rdi)\n" + "incq %rdi\n" + "jmp .loop\n" /* Continue to next element. */ + ".done:\n" + "ret\n" +#elif defined(sus_arch_ia32) + /* eax: Address of the current element. */ + "movl 0x4(%esp),%eax\n" + /* ecx: Address of the element after the last element. */ + "movl 0x4(%esp),%ecx\n" + "addl 0x8(%esp),%ecx\n" + /* edx: Byte value. */ + "movb 0xC(%esp),%dl\n" + ".loop:\n" + "cmpl %eax,%ecx\n" + "je .done\n" /* Exit loop if we have reached the final element. */ + "movb %dl,(%eax)\n" + "incl %eax\n" + "jmp .loop\n" /* Continue to next element. */ + ".done:\n" + "ret\n" +#endif +); +#else +void rgo_memfill(void * const sus_restr _ptr,sus_typ_usz const _num,sus_typ_u8 const _byte) { + sus_typ_u8 * pos = (sus_typ_u8 *)_ptr; + sus_typ_u8 * const afterbuf = pos + _num; + for (;pos != afterbuf;++pos) {*pos = _byte;} +} +#endif |