mirror of
https://github.com/pjh456/OxideJS.git
synced 2026-10-23 18:21:13 +08:00
pjh123
6a5ed5c78e
异常边只从 TRY 标记所在 BB 发出，分支/循环 BB 内调用点的 liveness 不含仅 catch/finally 存活的寄存器；按存活上界截断窗口会使 unwind 恢复后 handler 读到 callee 残留垃圾值（可为悬垂对象指针）。含 TRY_BEGIN/TRY_FINALLY_BEGIN 的函数 整体跳过窗口编码（回退全量窗口），恢复基线保守行为，普通函数不受影响。 补分支内调用 + catch 读 try 前变量的 Red/Green 测试与 finally 回归锚点。
174 lines
8.2 KiB
Rust
174 lines
8.2 KiB
Rust
//! 调用点存活上界编码：RegAlloc 改写后按物理存活集为调用指令回填 ext 高 8 位。
|
||
//!
|
||
//! 调用指令 `ext[0] = nargs(低 8 位) | 存活上界(高 8 位)`。上界 = 调用点存活
|
||
//! 物理寄存器最大槽号 + 1（即压帧窗口 `regs[0..上界]`）；上界 0 表示未编码
|
||
//! （运行时回退到调用方 `active_reg_limit` 全量窗口）。`regs[254]/[255]`
|
||
//! （this/new.target）由帧单独保存不占窗口，计算时排除。生成器 / 异步函数体
|
||
//! 内部调用点不编码——挂起恢复按全量寄存器快照搬移，保持保守语义。含 TRY
|
||
//! 指令的函数同样不编码——异常 handler 的存活集不经分支/循环内调用点传播，
|
||
//! 截断窗口会丢仅 handler 存活的槽（见 `encode_call_window` 内 has_try 说明）。
|
||
|
||
use oxide_bytecode::opcode::OpCode;
|
||
use oxide_ir::IRFunction;
|
||
use oxide_liveness::LiveInfo;
|
||
|
||
/// 为每个字节码调用指令（CALL/NEW_EXPRESSION/SUPER_CALL）编码存活上界到 ext 高 8 位。
|
||
/// 嵌套子函数递归处理：每个子函数独立重跑 CFG + liveness（与 `alloc` 同纪律）。
|
||
///
|
||
/// # 边界与前提
|
||
/// - `live` 与 `f.insts` 维度不符时内部重算（防过期 LiveInfo 下标错位）
|
||
/// - 生成器 / 异步函数体内部调用点不编码（保持全量窗口）
|
||
/// - 只改写 `ext[0]` 高 8 位，nargs 低 8 位不动；无存活时保持 0（全量）
|
||
pub fn encode_call_window(f: &mut IRFunction, live: &LiveInfo) {
|
||
if f.insts.is_empty() && f.nested.is_empty() {
|
||
return;
|
||
}
|
||
// 含 try/catch/finally 的函数跳过编码：异常边只从 TRY 标记所在 BB 发出，
|
||
// 分支/循环 BB 内调用点的 liveness 不含仅 catch/finally 存活的寄存器，
|
||
// 截断窗口会丢槽（unwind 恢复后 handler 读到 callee 残留）。整体回退全量窗口。
|
||
let has_try = f
|
||
.insts
|
||
.iter()
|
||
.any(|i| matches!(i.op, OpCode::TRY_BEGIN | OpCode::TRY_FINALLY_BEGIN));
|
||
// LiveInfo 维度守卫：与 alloc 相同纪律，过期则重算。
|
||
let live = if live.inst_live_before.len() == f.insts.len() && !f.insts.is_empty() {
|
||
live.clone()
|
||
} else {
|
||
let cfg = oxide_cfg::build_cfg(f);
|
||
oxide_liveness::liveness(f, &cfg)
|
||
};
|
||
// 生成器 / 异步函数体 + 含 try 的函数：调用点保持全量窗口（挂起恢复按全量
|
||
// 寄存器快照搬移；异常 handler 存活集经截断窗口会丢值）。
|
||
if !has_try && !f.is_generator && !f.is_async {
|
||
for (i, inst) in f.insts.iter_mut().enumerate() {
|
||
if matches!(inst.op, OpCode::CALL | OpCode::NEW_EXPRESSION | OpCode::SUPER_CALL) {
|
||
let after = &live.inst_live_after[i];
|
||
// 存活上界 = 最大存活槽号 + 1；254/255 由帧单独保存，不进窗口。
|
||
let mut upper = 0u32;
|
||
for (reg, &alive) in after.iter().enumerate().take(254) {
|
||
if alive {
|
||
upper = (reg + 1) as u32;
|
||
}
|
||
}
|
||
if upper > 0 {
|
||
inst.ext[0] = (inst.ext[0] & 0xFF) | (upper << 8);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
for child in &mut f.nested {
|
||
let cfg = oxide_cfg::build_cfg(child);
|
||
let l = oxide_liveness::liveness(child, &cfg);
|
||
encode_call_window(child, &l);
|
||
}
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
use oxide_ir::inst::Inst;
|
||
use oxide_ir::operand::Operand;
|
||
|
||
#[test]
|
||
fn call_window_packs_upper_bound_from_live_after() {
|
||
// 0: CALL（r1=callee, r2=this, r3=首参, 1 参） def {0} use {1,2,3}
|
||
// 1: ADD r5 = r4 + r4 r4 跨调用存活
|
||
// 2: RETURN r5
|
||
let mut f = IRFunction::new();
|
||
f.insts.push(Inst::call(Operand::Reg(1), Operand::Reg(2), Operand::Reg(3), 1));
|
||
f.insts
|
||
.push(Inst::new(OpCode::ADD, Operand::Reg(5), Operand::Reg(4), Operand::Reg(4)));
|
||
f.insts
|
||
.push(Inst::new(OpCode::RETURN, Operand::Reg(5), Operand::None, Operand::None));
|
||
let cfg = oxide_cfg::build_cfg(&f);
|
||
let live = oxide_liveness::liveness(&f, &cfg);
|
||
encode_call_window(&mut f, &live);
|
||
// 存活集 {4} → 上界 5，ext = 1 | (5 << 8)
|
||
assert_eq!(f.insts[0].ext[0], 1 | (5 << 8));
|
||
}
|
||
|
||
#[test]
|
||
fn call_window_zero_when_nothing_survives() {
|
||
// 0: CALL（结果未用） 1: LOAD_CONST r6 2: RETURN r6
|
||
let mut f = IRFunction::new();
|
||
f.insts.push(Inst::call(Operand::Reg(1), Operand::Reg(2), Operand::Reg(3), 1));
|
||
f.insts.push(Inst::load_const(Operand::Reg(6), 0));
|
||
f.insts
|
||
.push(Inst::new(OpCode::RETURN, Operand::Reg(6), Operand::None, Operand::None));
|
||
let cfg = oxide_cfg::build_cfg(&f);
|
||
let live = oxide_liveness::liveness(&f, &cfg);
|
||
encode_call_window(&mut f, &live);
|
||
// 无跨调用存活 → 保持全量（ext 高 8 位 = 0）
|
||
assert_eq!(f.insts[0].ext[0], 1);
|
||
}
|
||
|
||
#[test]
|
||
fn call_window_excludes_this_and_new_target() {
|
||
// 0: LOAD_CONST r6 1: CALL 2: LOAD_VAR r8, This（254 跨调用存活）
|
||
// 3: ADD r7=r6+r8 4: RETURN r7
|
||
let mut f = IRFunction::new();
|
||
f.insts.push(Inst::load_const(Operand::Reg(6), 0));
|
||
f.insts.push(Inst::call(Operand::Reg(1), Operand::Reg(2), Operand::Reg(3), 1));
|
||
f.insts
|
||
.push(Inst::new(OpCode::LOAD_VAR, Operand::Reg(8), Operand::This, Operand::None));
|
||
f.insts
|
||
.push(Inst::new(OpCode::ADD, Operand::Reg(7), Operand::Reg(6), Operand::Reg(8)));
|
||
f.insts
|
||
.push(Inst::new(OpCode::RETURN, Operand::Reg(7), Operand::None, Operand::None));
|
||
let cfg = oxide_cfg::build_cfg(&f);
|
||
let live = oxide_liveness::liveness(&f, &cfg);
|
||
encode_call_window(&mut f, &live);
|
||
// 存活 {6, 254} → 上界 7（254 由帧单独保存，不放大窗口）
|
||
assert_eq!(f.insts[1].ext[0], 1 | (7 << 8));
|
||
}
|
||
|
||
#[test]
|
||
fn call_window_skips_generator_body() {
|
||
let mut f = IRFunction::new();
|
||
f.is_generator = true;
|
||
f.insts.push(Inst::call(Operand::Reg(1), Operand::Reg(2), Operand::Reg(3), 1));
|
||
f.insts
|
||
.push(Inst::new(OpCode::ADD, Operand::Reg(5), Operand::Reg(4), Operand::Reg(4)));
|
||
f.insts
|
||
.push(Inst::new(OpCode::RETURN, Operand::Reg(5), Operand::None, Operand::None));
|
||
let cfg = oxide_cfg::build_cfg(&f);
|
||
let live = oxide_liveness::liveness(&f, &cfg);
|
||
encode_call_window(&mut f, &live);
|
||
// 生成器体不编码 → ext 保持纯 nargs
|
||
assert_eq!(f.insts[0].ext[0], 1);
|
||
}
|
||
|
||
#[test]
|
||
fn call_window_stale_live_reruns() {
|
||
// 空 LiveInfo（维度不符）→ 内部重跑，仍正确编码。
|
||
let mut f = IRFunction::new();
|
||
f.insts.push(Inst::call(Operand::Reg(1), Operand::Reg(2), Operand::Reg(3), 1));
|
||
f.insts
|
||
.push(Inst::new(OpCode::ADD, Operand::Reg(5), Operand::Reg(4), Operand::Reg(4)));
|
||
f.insts
|
||
.push(Inst::new(OpCode::RETURN, Operand::Reg(5), Operand::None, Operand::None));
|
||
encode_call_window(&mut f, &LiveInfo::new());
|
||
assert_eq!(f.insts[0].ext[0], 1 | (5 << 8));
|
||
}
|
||
|
||
#[test]
|
||
fn call_window_skips_function_with_try() {
|
||
// 含 TRY_BEGIN 的函数整体跳过编码（全量窗口）：异常边只从 TRY 所在 BB 出发，
|
||
// 分支/循环内调用点的 liveness 不含仅 handler 存活的寄存器，截断窗口会丢槽。
|
||
let mut f = IRFunction::new();
|
||
f.insts.push(Inst::try_begin(0)); // 0: TRY_BEGIN → L0
|
||
f.insts.push(Inst::call(Operand::Reg(1), Operand::Reg(2), Operand::Reg(3), 1));
|
||
f.insts
|
||
.push(Inst::new(OpCode::ADD, Operand::Reg(5), Operand::Reg(4), Operand::Reg(4)));
|
||
f.insts
|
||
.push(Inst::new(OpCode::RETURN, Operand::Reg(5), Operand::None, Operand::None));
|
||
f.label_pos = vec![Some(2)];
|
||
f.label_count = 1;
|
||
let cfg = oxide_cfg::build_cfg(&f);
|
||
let live = oxide_liveness::liveness(&f, &cfg);
|
||
encode_call_window(&mut f, &live);
|
||
// 存活集 {4} 本可编码上界 5，但含 try 保持全量（ext 高 8 位 = 0）
|
||
assert_eq!(f.insts[1].ext[0], 1);
|
||
}
|
||
}
|