/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * test helper assembly functions
 *
 * Copyright (C) 2016 Simon Guo, IBM Corporation.
 * Copyright 2022 Michael Ellerman, IBM Corporation.
 */
#include "basic_asm.h"

#define GPR_SIZE	__SIZEOF_LONG__
#define FIRST_GPR	14
#define NUM_GPRS	(32 - FIRST_GPR)
#define STACK_SIZE	(NUM_GPRS * GPR_SIZE)

// gpr_child_loop(int *read_flag, int *write_flag,
//		  unsigned long *gpr_buf, double *fpr_buf);
FUNC_START(gpr_child_loop)
	// r3 = read_flag
	// r4 = write_flag
	// r5 = gpr_buf
	// r6 = fpr_buf
	PUSH_BASIC_STACK(STACK_SIZE)

	// Save non-volatile GPRs
	OP_REGS PPC_STL, GPR_SIZE, FIRST_GPR, 31, %r1, STACK_FRAME_LOCAL(0, 0), FIRST_GPR

	// Load GPRs with expected values
	OP_REGS PPC_LL, GPR_SIZE, FIRST_GPR, 31, r5, 0, FIRST_GPR

	// Load FPRs with expected values
	OP_REGS lfd, 8, 0, 31, r6

	// Signal to parent that we're ready
	li	r0, 1
	stw	r0, 0(r4)

	// Wait for parent to finish
1:	lwz	r0, 0(r3)
	cmpwi	r0, 0
	beq	1b	// Loop while flag is zero

	// Save GPRs back to caller buffer
	OP_REGS PPC_STL, GPR_SIZE, FIRST_GPR, 31, r5, 0, FIRST_GPR

	// Save FPRs
	OP_REGS stfd, 8, 0, 31, r6

	// Reload non-volatile GPRs
	OP_REGS PPC_LL, GPR_SIZE, FIRST_GPR, 31, %r1, STACK_FRAME_LOCAL(0, 0), FIRST_GPR

	POP_BASIC_STACK(STACK_SIZE)
	blr