// Copyright (c) 2026 Petr Balvín <opensource@petrbalvin.org> (https://petrbalvin.org)
// SPDX-License-Identifier: BSD-3-Clause

//go:build riscv64

package verify

import (
	"encoding/binary"
	"fmt"
	"reflect"
	"syscall"
	"unsafe"
)

//go:nosplit
func enterJIT(fn uintptr, stack uintptr)

//go:nosplit
func leaveJIT()

var leaveJITAddr uintptr

func init() {
	leaveJITAddr = reflect.ValueOf(leaveJIT).Pointer()
}

const stackPad = 64

// Call invokes the assembled function at fnAddr with the given ABI0 argument block.
func Call(fnAddr uintptr, args []byte) ([]byte, error) {
	stackSize := stackPad + 8 + len(args) + 64
	stackMem, err := syscall.Mmap(-1, 0, stackSize,
		syscall.PROT_READ|syscall.PROT_WRITE, syscall.MAP_PRIVATE|syscall.MAP_ANON)
	if err != nil {
		return nil, fmt.Errorf("verify: stack mmap: %w", err)
	}
	defer syscall.Munmap(stackMem)

	retOff := stackPad
	binary.LittleEndian.PutUint64(stackMem[retOff:retOff+8], uint64(leaveJITAddr))
	copy(stackMem[retOff+8:], args)

	stackBase := uintptr(unsafe.Pointer(&stackMem[retOff]))
	enterJIT(fnAddr, stackBase)

	out := make([]byte, len(args))
	copy(out, stackMem[retOff+8:retOff+8+len(args)])
	return out, nil
}