xref: /src/common/lib/libc/arch/x86_64/string/strchr.S

/*
 * Written by J.T. Conklin <jtc (at) acorntoolworks.com>
 * Public domain.
 */

#include <machine/asm.h>

#if defined(LIBC_SCCS)
	RCSID("$NetBSD: strchr.S,v 1.1 2005/12/20 19:28:51 christos Exp $")
#endif

#ifdef INDEX
ENTRY(index)
#else
ENTRY(strchr)
#endif
	movzbq	%sil,%rcx

	/*
	 * Align to word boundary.
	 * Consider unrolling loop?
	 */
.Lalign:
	testb	$7,%dil
	je	.Lword_aligned
	movb	(%rdi),%dl
	cmpb	%cl,%dl
	je	.Ldone
	incq	%rdi
	testb	%dl,%dl
	jne	.Lalign
	jmp	.Lzero

.Lword_aligned:
	/* copy char to all bytes in word */
	movb	%cl,%ch
	movq	%rcx,%rdx
	salq	$16,%rcx
	orq	%rdx,%rcx
	movq	%rcx,%rdx
	salq	$32,%rcx
	orq	%rdx,%rcx

	movabsq	$0x0101010101010101,%r8
	movabsq	$0x8080808080808080,%r9

	/* Check whether any byte in the word is equal to ch or 0. */
	_ALIGN_TEXT
.Lloop:
	movq	(%rdi),%rdx
	addq	$8,%rdi
	movq	%rdx,%rsi
	subq	%r8,%rdx
	xorq	%rcx,%rsi
	subq	%r8,%rsi
	orq	%rsi,%rdx
	testq	%r9,%rdx
	je	.Lloop

	/*
	 * In rare cases, the above loop may exit prematurely. We must
	 * return to the loop if none of the bytes in the word match
	 * ch or are equal to 0.
	 */

	movb	-8(%rdi),%dl
	cmpb	%cl,%dl		/* 1st byte == ch? */
	jne	1f
	subq	$8,%rdi
	jmp	.Ldone
1:	testb	%dl,%dl		/* 1st byte == 0? */
	je	.Lzero

	movb	-7(%rdi),%dl
	cmpb	%cl,%dl		/* 2nd byte == ch? */
	jne	1f
	subq	$7,%rdi
	jmp	.Ldone
1:	testb	%dl,%dl		/* 2nd byte == 0? */
	je	.Lzero

	movb	-6(%rdi),%dl
	cmpb	%cl,%dl		/* 3rd byte == ch? */
	jne	1f
	subq	$6,%rdi
	jmp	.Ldone
1:	testb	%dl,%dl		/* 3rd byte == 0? */
	je	.Lzero

	movb	-5(%rdi),%dl
	cmpb	%cl,%dl		/* 4th byte == ch? */
	jne	1f
	subq	$5,%rdi
	jmp	.Ldone
1:	testb	%dl,%dl		/* 4th byte == 0? */
	je	.Lzero

	movb	-4(%rdi),%dl
	cmpb	%cl,%dl		/* 5th byte == ch? */
	jne	1f
	subq	$4,%rdi
	jmp	.Ldone
1:	testb	%dl,%dl		/* 5th byte == 0? */
	je	.Lzero

	movb	-3(%rdi),%dl
	cmpb	%cl,%dl		/* 6th byte == ch? */
	jne	1f
	subq	$3,%rdi
	jmp	.Ldone
1:	testb	%dl,%dl		/* 6th byte == 0? */
	je	.Lzero

	movb	-2(%rdi),%dl
	cmpb	%cl,%dl		/* 7th byte == ch? */
	jne	1f
	subq	$2,%rdi
	jmp	.Ldone
1:	testb	%dl,%dl		/* 7th byte == 0? */
	je	.Lzero

	movb	-1(%rdi),%dl
	cmpb	%cl,%dl		/* 8th byte == ch? */
	jne	1f
	subq	$1,%rdi
	jmp	.Ldone
1:	testb	%dl,%dl		/* 8th byte == 0? */
	jne	.Lloop

.Lzero:
	/* If a ch wasn't found, return 0. */
	xorq	%rdi,%rdi

.Ldone:
	movq	%rdi,%rax
	ret