shader_jit_x64: Use CALL/RET instead of JMP for subroutines.
This commit is contained in:
parent
507e0b5989
commit
60749f2cda
|
@ -583,23 +583,15 @@ void JitCompiler::Compile_END(Instruction instr) {
|
|||
}
|
||||
|
||||
void JitCompiler::Compile_CALL(Instruction instr) {
|
||||
// Need to advance the return address past the proceeding instructions, this is the number of bytes to skip
|
||||
constexpr unsigned SKIP = 21;
|
||||
const uintptr_t start = reinterpret_cast<uintptr_t>(GetCodePtr());
|
||||
|
||||
// Push return address - not using CALL because we also want to push the offset of the return before jumping
|
||||
MOV(64, R(RAX), ImmPtr(GetCodePtr() + SKIP));
|
||||
PUSH(RAX);
|
||||
|
||||
// Push offset of the return
|
||||
PUSH(32, Imm32(instr.flow_control.dest_offset + instr.flow_control.num_instructions));
|
||||
PUSH(64, Imm32(instr.flow_control.dest_offset + instr.flow_control.num_instructions));
|
||||
|
||||
// Jump
|
||||
FixupBranch b = J(true);
|
||||
// Call the subroutine
|
||||
FixupBranch b = CALL();
|
||||
fixup_branches.push_back({ b, instr.flow_control.dest_offset });
|
||||
|
||||
// Make sure that if the above code changes, SKIP gets updated
|
||||
ASSERT(reinterpret_cast<ptrdiff_t>(GetCodePtr()) - start == SKIP);
|
||||
// Skip over the return offset that's on the stack
|
||||
ADD(64, R(RSP), Imm32(8));
|
||||
}
|
||||
|
||||
void JitCompiler::Compile_CALLC(Instruction instr) {
|
||||
|
@ -758,14 +750,12 @@ void JitCompiler::Compile_Block(unsigned end) {
|
|||
|
||||
void JitCompiler::Compile_Return() {
|
||||
// Peek return offset on the stack and check if we're at that offset
|
||||
MOV(64, R(RAX), MDisp(RSP, 0));
|
||||
MOV(64, R(RAX), MDisp(RSP, 8));
|
||||
CMP(32, R(RAX), Imm32(program_counter));
|
||||
|
||||
// If so, jump back to before CALL
|
||||
FixupBranch b = J_CC(CC_NZ, true);
|
||||
ADD(64, R(RSP), Imm32(8)); // Ignore return offset that's on the stack
|
||||
POP(RAX); // Pop off return address
|
||||
JMPptr(R(RAX));
|
||||
RET();
|
||||
SetJumpTarget(b);
|
||||
}
|
||||
|
||||
|
|
Reference in New Issue