mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00

This change fixes some cases where YJIT fails to fire tracing events. Most of the situations YJIT did not handle correctly involves enabling tracing while running inside generated code. A new operation to invalidate all generated code is added, which uses patching to make generated code exit at the next VM instruction boundary. A new routine called `jit_prepare_routine_call()` is introduced to facilitate this and should be used when generating code that could allocate, or could otherwise use `RB_VM_LOCK_ENTER()`. The `c_return` event is fired in the middle of an instruction as opposed to at an instruction boundary, so it requires special handling. C method call return points are patched to go to a fucntion which does everything the interpreter does, including firing the `c_return` event. The generated code for C method calls normally does not fire the event. Invalided code should not change after patching so the exits are not clobbered. A new variable is introduced to track the region of code that should not change.
55 lines
1.3 KiB
C
55 lines
1.3 KiB
C
#ifndef YJIT_CODEGEN_H
|
|
#define YJIT_CODEGEN_H 1
|
|
|
|
#include "stddef.h"
|
|
#include "yjit_core.h"
|
|
|
|
// Code blocks we generate code into
|
|
extern codeblock_t *cb;
|
|
extern codeblock_t *ocb;
|
|
extern uint32_t yjit_codepage_frozen_bytes;
|
|
|
|
// Code generation state
|
|
typedef struct JITState
|
|
{
|
|
// Block version being compiled
|
|
block_t* block;
|
|
|
|
// Instruction sequence this is associated with
|
|
const rb_iseq_t *iseq;
|
|
|
|
// Index of the current instruction being compiled
|
|
uint32_t insn_idx;
|
|
|
|
// Opcode for the instruction being compiled
|
|
int opcode;
|
|
|
|
// PC of the instruction being compiled
|
|
VALUE *pc;
|
|
|
|
// Execution context when compilation started
|
|
// This allows us to peek at run-time values
|
|
rb_execution_context_t* ec;
|
|
|
|
// Whether we need to record the code address at
|
|
// the end of this bytecode instruction for tracing suppoert
|
|
bool record_boundary_patch_point;
|
|
|
|
} jitstate_t;
|
|
|
|
typedef enum codegen_status {
|
|
YJIT_END_BLOCK,
|
|
YJIT_KEEP_COMPILING,
|
|
YJIT_CANT_COMPILE
|
|
} codegen_status_t;
|
|
|
|
// Code generation function signature
|
|
typedef codegen_status_t (*codegen_fn)(jitstate_t* jit, ctx_t* ctx);
|
|
|
|
uint8_t* yjit_entry_prologue(const rb_iseq_t* iseq);
|
|
|
|
void yjit_gen_block(block_t* block, rb_execution_context_t* ec);
|
|
|
|
void yjit_init_codegen(void);
|
|
|
|
#endif // #ifndef YJIT_CODEGEN_H
|