strchr.S

/* SPDX-License-Identifier: GPL-2.0-only */

/*
 * Copyright (C) 2025 Feng Jiang <[email protected]>
 */

#include <linux/linkage.h>
#include <asm/asm.h>

/* char *strchr(const char *s, int c) */
SYM_FUNC_START(strchr)
	/*
	 * Parameters
	 *   a0 - The string to be searched
	 *   a1 - The character to search for
	 *
	 * Returns
	 *   a0 - Address of first occurrence of 'c' or 0
	 *
	 * Clobbers
	 *   t0
	 */
	andi	a1, a1, 0xff
1:
	lbu	t0, 0(a0)
	beq	t0, a1, 2f
	addi	a0, a0, 1
	bnez	t0, 1b
	li	a0, 0
2:
	ret
SYM_FUNC_END(strchr)

SYM_FUNC_ALIAS_WEAK(__pi_strchr, strchr)
EXPORT_SYMBOL(strchr)