mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-17 12:56:39 +00:00
ea76b2d8d8
Add RISC-V support for XRay. The RV64 implementation has been tested in both QEMU and in our hardware environment. Currently this requires D and C extensions, but since both RV64GC and RVA22/RVA23 are becoming mainstream, I don't think this requirement will be a big problem. Based on the previous work by @a-poduval : https://reviews.llvm.org/D117929 --------- Co-authored-by: Ashwin Poduval <ashwin.poduval@gmail.com>
97 lines
2.4 KiB
ArmAsm
97 lines
2.4 KiB
ArmAsm
//===-- xray_trampoline_riscv_common.s --------------------------*- ASM -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file is a part of XRay, a dynamic runtime instrumentation system.
|
|
//
|
|
// This implements the trampolines code shared between riscv32 and riscv64.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "../builtins/assembly.h"
|
|
|
|
.text
|
|
.p2align 2
|
|
.global ASM_SYMBOL(__xray_FunctionEntry)
|
|
ASM_TYPE_FUNCTION(__xray_FunctionEntry)
|
|
ASM_SYMBOL(__xray_FunctionEntry):
|
|
CFI_STARTPROC
|
|
SAVE_ARG_REGISTERS
|
|
|
|
// Load the handler function pointer into a2
|
|
la a2, ASM_SYMBOL(_ZN6__xray19XRayPatchedFunctionE)
|
|
LOAD_XLEN a2, 0(a2)
|
|
|
|
// Handler address will be null if it is not set
|
|
beq a2, x0, 1f
|
|
|
|
// If we reach here, we are tracing an event
|
|
// a0 already contains function id
|
|
// a1 = 0 means we are tracing an entry event
|
|
li a1, 0
|
|
jalr a2
|
|
|
|
1:
|
|
RESTORE_ARG_REGISTERS
|
|
jr ra
|
|
ASM_SIZE(__xray_FunctionEntry)
|
|
CFI_ENDPROC
|
|
|
|
.text
|
|
.p2align 2
|
|
.global ASM_SYMBOL(__xray_FunctionExit)
|
|
ASM_TYPE_FUNCTION(__xray_FunctionExit)
|
|
ASM_SYMBOL(__xray_FunctionExit):
|
|
CFI_STARTPROC
|
|
SAVE_RET_REGISTERS
|
|
|
|
// Load the handler function pointer into a2
|
|
la a2, ASM_SYMBOL(_ZN6__xray19XRayPatchedFunctionE)
|
|
LOAD_XLEN a2, 0(a2)
|
|
|
|
// Handler address will be null if it is not set
|
|
beq a2, x0, 1f
|
|
|
|
// If we reach here, we are tracing an event
|
|
// a0 already contains function id
|
|
// a1 = 1 means we are tracing an exit event
|
|
li a1, 1
|
|
jalr a2
|
|
|
|
1:
|
|
RESTORE_RET_REGISTERS
|
|
jr ra
|
|
ASM_SIZE(__xray_FunctionExit)
|
|
CFI_ENDPROC
|
|
|
|
.text
|
|
.p2align 2
|
|
.global ASM_SYMBOL(__xray_FunctionTailExit)
|
|
ASM_TYPE_FUNCTION(__xray_FunctionTailExit)
|
|
ASM_SYMBOL(__xray_FunctionTailExit):
|
|
CFI_STARTPROC
|
|
SAVE_ARG_REGISTERS
|
|
|
|
// Load the handler function pointer into a2
|
|
la a2, ASM_SYMBOL(_ZN6__xray19XRayPatchedFunctionE)
|
|
LOAD_XLEN a2, 0(a2)
|
|
|
|
// Handler address will be null if it is not set
|
|
beq a2, x0, 1f
|
|
|
|
// If we reach here, we are tracing an event
|
|
// a0 already contains function id
|
|
// a1 = 2 means we are tracing a tail exit event
|
|
li a1, 2
|
|
jalr a2
|
|
|
|
1:
|
|
RESTORE_ARG_REGISTERS
|
|
jr ra
|
|
ASM_SIZE(__xray_FunctionTailExit)
|
|
CFI_ENDPROC
|