archive/ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2025-08-26 06:25:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Rewrite the stub if it's the last code

Browse source
This commit is contained in:
Takashi Kokubun 2023-01-01 23:13:43 -08:00
parent c3d99d0f12
commit 4b6c738180
5 changed files with 45 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

26
lib/ruby_vm/mjit/compiler.rb
View file

@ -63,22 +63,28 @@ module RubyVM::MJIT
# Update cfp->pc for `jit.at_current_insn?`
cfp.pc = stub.pc
# Compile the jump target
new_addr = Assembler.new.then do |asm|
# Prepare the jump target
new_asm = Assembler.new.tap do |asm|
jit = JITState.new(iseq: stub.iseq, cfp:)
index = (stub.pc - stub.iseq.body.iseq_encoded.to_i) / C.VALUE.size
compile_block(asm, jit:, index:, ctx: stub.ctx)
@cb.write(asm)
end
# Re-generate the jump source
@cb.with_addr(stub.addr) do
asm = Assembler.new
asm.comment('regenerate block stub')
asm.jmp(new_addr)
@cb.write(asm)
# Rewrite the stub
if @cb.write_addr == stub.end_addr
# If the stub jump is the last code, overwrite the jump with the new code.
@cb.set_write_addr(stub.start_addr)
@cb.write(new_asm)
else
# If the stub jump is old code, change the jump target to the new code.
new_addr = @cb.write(new_asm)
@cb.with_write_addr(stub.start_addr) do
asm = Assembler.new
asm.comment('regenerate block stub')
asm.jmp(new_addr)
@cb.write(asm)
end
end
new_addr
end
private