Gecko [ arkitchencabinet.com ]

Name	Size	Permission	Date
assembler.rb	32.81 KB	-rw-r--r--	2026-01-27 16:00
block.rb	568 B	-rw-r--r--	2026-01-27 16:00
branch_stub.rb	879 B	-rw-r--r--	2026-01-27 16:00
c_pointer.rb	10.68 KB	-rw-r--r--	2026-01-27 16:00
c_type.rb	2.66 KB	-rw-r--r--	2026-01-27 16:00
code_block.rb	2.28 KB	-rw-r--r--	2026-01-27 16:00
compiler.rb	16.72 KB	-rw-r--r--	2026-01-27 16:00
context.rb	11.14 KB	-rw-r--r--	2026-01-27 16:00
entry_stub.rb	226 B	-rw-r--r--	2026-01-27 16:00
exit_compiler.rb	4.83 KB	-rw-r--r--	2026-01-27 16:00
hooks.rb	949 B	-rw-r--r--	2026-01-27 16:00
insn_compiler.rb	204.81 KB	-rw-r--r--	2026-01-27 16:00
instruction.rb	38.66 KB	-rw-r--r--	2026-01-27 16:00
invariants.rb	4.72 KB	-rw-r--r--	2026-01-27 16:00
jit_state.rb	1.96 KB	-rw-r--r--	2026-01-27 16:00
stats.rb	7.4 KB	-rw-r--r--	2026-01-27 16:00
type.rb	5.91 KB	-rw-r--r--	2026-01-27 16:00

Rename

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] = [] } if dump_disasm?
      @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 if dump_disasm?
      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) if dump_disasm?
    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

def dump_disasm?
      C.rjit_opts.dump_disasm
    end
  end
end