class RubyVM::RJIT::CodeBlock
Public Class Methods
new(mem_block:, mem_size:, outlined: false)
click to toggle source
@param mem_block [Integer] JIT buffer address @param mem_size [Integer] JIT buffer size @param outliend [TrueClass,FalseClass] true for outlined CodeBlock
# File ruby_vm/rjit/code_block.rb, line 6 def initialize(mem_block:, mem_size:, outlined: false) @comments = Hash.new { |h, k| h[k] = [] } @mem_block = mem_block @mem_size = mem_size @write_pos = 0 @outlined = outlined end
Public Instance Methods
dump_disasm(from, to, io: STDOUT, color: true, test: false)
click to toggle source
# File ruby_vm/rjit/code_block.rb, line 61 def dump_disasm(from, to, io: STDOUT, color: true, test: false) C.dump_disasm(from, to, test:).each do |address, mnemonic, op_str| @comments.fetch(address, []).each do |comment| io.puts colorize(" # #{comment}", bold: true, color:) end io.puts colorize(" 0x#{format("%x", address)}: #{mnemonic} #{op_str}", color:) end io.puts end
include?(addr)
click to toggle source
# File ruby_vm/rjit/code_block.rb, line 57 def include?(addr) (@mem_block...(@mem_block + @mem_size)).include?(addr) end
set_write_addr(addr)
click to toggle source
# File ruby_vm/rjit/code_block.rb, line 40 def set_write_addr(addr) @write_pos = addr - @mem_block @comments.delete(addr) # TODO: clean up old comments for all the overwritten range? end
with_write_addr(addr) { || ... }
click to toggle source
# File ruby_vm/rjit/code_block.rb, line 45 def with_write_addr(addr) old_write_pos = @write_pos set_write_addr(addr) yield ensure @write_pos = old_write_pos end
write(asm)
click to toggle source
@param asm [RubyVM::RJIT::Assembler]
# File ruby_vm/rjit/code_block.rb, line 15 def write(asm) return 0 if @write_pos + asm.size >= @mem_size start_addr = write_addr # Write machine code C.mprotect_write(@mem_block, @mem_size) @write_pos += asm.assemble(start_addr) C.mprotect_exec(@mem_block, @mem_size) end_addr = write_addr # Convert comment indexes to addresses asm.comments.each do |index, comments| @comments[start_addr + index] += comments end asm.comments.clear # Dump disasm if --rjit-dump-disasm if C.rjit_opts.dump_disasm && start_addr < end_addr dump_disasm(start_addr, end_addr) end start_addr end
write_addr()
click to toggle source
# File ruby_vm/rjit/code_block.rb, line 53 def write_addr @mem_block + @write_pos end
Private Instance Methods
bold(text)
click to toggle source
# File ruby_vm/rjit/code_block.rb, line 83 def bold(text) "\e[1m#{text}\e[0m" end
colorize(text, bold: false, color:)
click to toggle source
# File ruby_vm/rjit/code_block.rb, line 73 def colorize(text, bold: false, color:) return text unless color buf = +'' buf << "\e[1m" if bold buf << "\e[34m" if @outlined buf << text buf << "\e[0m" buf end