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