// SPDX-License-Identifier: GPL-2.0-or-later

#include <linux/regset.h>

#include <asm/switch_to.h>

#include "ptrace-decl.h"

int ptrace_get_fpr(struct task_struct *child, int index, unsigned long *data)
{
#ifdef CONFIG_PPC_FPU_REGS
        unsigned int fpidx = index - PT_FPR0;
#endif

        if (index > PT_FPSCR)
                return -EIO;

#ifdef CONFIG_PPC_FPU_REGS
        flush_fp_to_thread(child);
        if (fpidx < (PT_FPSCR - PT_FPR0)) {
                if (IS_ENABLED(CONFIG_PPC32))
                        // On 32-bit the index we are passed refers to 32-bit words
                        *data = ((u32 *)child->thread.fp_state.fpr)[fpidx];
                else
                        memcpy(data, &child->thread.TS_FPR(fpidx), sizeof(long));
        } else
                *data = child->thread.fp_state.fpscr;
#else
        *data = 0;
#endif

        return 0;
}

int ptrace_put_fpr(struct task_struct *child, int index, unsigned long data)
{
#ifdef CONFIG_PPC_FPU_REGS
        unsigned int fpidx = index - PT_FPR0;
#endif

        if (index > PT_FPSCR)
                return -EIO;

#ifdef CONFIG_PPC_FPU_REGS
        flush_fp_to_thread(child);
        if (fpidx < (PT_FPSCR - PT_FPR0)) {
                if (IS_ENABLED(CONFIG_PPC32))
                        // On 32-bit the index we are passed refers to 32-bit words
                        ((u32 *)child->thread.fp_state.fpr)[fpidx] = data;
                else
                        memcpy(&child->thread.TS_FPR(fpidx), &data, sizeof(long));
        } else
                child->thread.fp_state.fpscr = data;
#endif

        return 0;
}