зеркало из https://github.com/github/ruby.git
88 строки
2.2 KiB
Ruby
88 строки
2.2 KiB
Ruby
module RubyVM::RJIT
|
|
class CodeBlock
|
|
# @param mem_block [Integer] JIT buffer address
|
|
# @param mem_size [Integer] JIT buffer size
|
|
# @param outliend [TrueClass,FalseClass] true for outlined CodeBlock
|
|
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
|
|
|
|
# @param asm [RubyVM::RJIT::Assembler]
|
|
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
|
|
|
|
def set_write_addr(addr)
|
|
@write_pos = addr - @mem_block
|
|
@comments.delete(addr) # TODO: clean up old comments for all the overwritten range?
|
|
end
|
|
|
|
def with_write_addr(addr)
|
|
old_write_pos = @write_pos
|
|
set_write_addr(addr)
|
|
yield
|
|
ensure
|
|
@write_pos = old_write_pos
|
|
end
|
|
|
|
def write_addr
|
|
@mem_block + @write_pos
|
|
end
|
|
|
|
def include?(addr)
|
|
(@mem_block...(@mem_block + @mem_size)).include?(addr)
|
|
end
|
|
|
|
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
|
|
|
|
private
|
|
|
|
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
|
|
|
|
def bold(text)
|
|
"\e[1m#{text}\e[0m"
|
|
end
|
|
end
|
|
end
|