/*
	Copyright 2022 Gabriel Jensen

	This file is part of rgo.

	rgo is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

	rgo is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.

	You should have received a copy of the GNU Lesser General Public License along with rgo. If not, see <https://www.gnu.org/licenses/>. 
*/

#include <rgo.h>

.global rgo_fndchr

rgo_fndchr:
	/*
		char const * str
		char         chr
	*/
#if defined(__i386__)
	/* eax: Address of the current character. */
	movl 0x4(%esp),%eax
	/* ecx: Character. */
	movb 0x8(%esp),%cl
	/* edx: Current character. */
.loop:
	movb (%eax),%dl
	cmpb %dl,%cl
	je .fnd /* Exit loop if we have found the character. */
	testb %dl,%dl
	je .nfnd /* We encountered the null-terminator but not the specified character. */
	incl %eax
	jmp .loop
.fnd:
	subl 0x4(%esp),%eax
	ret
.nfnd:
	movl $0xFFFFFFFF,%eax
	ret
#elif defined(__x86_64__)
	/* rax: Address of the current character. */
	movq %rdi,%rax
	/* rdx: Current character. */
.loop:
	movb (%rax),%dl
	cmpb %dl,%sil
	je .fnd /* Exit loop if we have found the character. */
	testb %dl,%dl
	je .nfnd /* We encountered the null-terminator but not the specified character. */
	incq %rax
	jmp .loop
.fnd:
	subq %rdi,%rax
	ret
.nfnd:
	movq $0xFFFFFFFFFFFFFFFF,%rax
	ret
#endif