refactor: 退役三族挂起快照克隆重写与指针重写死函数

This commit is contained in:
2026-10-02 23:01:13 +00:00
parent ccd6b06b6e
commit 26a46f802b
4 changed files with 1 additions and 406 deletions
-39
View File
@@ -764,45 +764,6 @@ pub(crate) fn async_native_cell_edges(obj: &JsObject) -> Vec<*mut Cell> {
out
}
/// 用转发函数重写状态快照中的所有 JsValue(session GC 移动式清扫 / promote 用)。
#[allow(dead_code)]
pub(crate) fn rewrite_async_native(obj: &JsObject, mut rewrite: impl FnMut(JsValue) -> JsValue) {
let Some(state) = async_state_mut(obj) else {
return;
};
state.callee = rewrite(state.callee);
for v in &mut state.args {
*v = rewrite(*v);
}
state.result = rewrite(state.result);
state.promise = rewrite(state.promise);
state.resolve = rewrite(state.resolve);
state.reject = rewrite(state.reject);
state.suspended.rewrite_values(rewrite);
}
/// 深拷贝状态盒到新对象(promote / sweep 用):新对象持独立 Box,源盒可安全释放。
#[allow(dead_code)]
pub(crate) fn clone_async_native_with_rewrite(
old: &JsObject, new: &mut JsObject, mut rewrite: impl FnMut(JsValue) -> JsValue,
) {
let Some(state) = async_state_mut(old) else {
return;
};
let cloned = AsyncState {
phase: state.phase,
callee: rewrite(state.callee),
args: state.args.iter().copied().map(&mut rewrite).collect(),
this_value: state.this_value,
result: rewrite(state.result),
promise: rewrite(state.promise),
resolve: rewrite(state.resolve),
reject: rewrite(state.reject),
suspended: state.suspended.clone_with_rewrite(rewrite),
};
new.set_native_data(Box::into_raw(Box::new(cloned)) as *mut u8);
}
/// 只读核算异步状态盒字节(不释放),供 GC 账目核算。
pub(crate) fn async_native_size(obj: &JsObject) -> u64 {
if !obj.is_async_obj() {
-60
View File
@@ -1752,66 +1752,6 @@ pub(crate) fn async_generator_native_cell_edges(obj: &JsObject) -> Vec<*mut Cell
out
}
/// 用转发函数重写状态快照中的所有 JsValue(session GC 移动式清扫 / promote 用)。
#[allow(dead_code)]
pub(crate) fn rewrite_async_generator_native(obj: &JsObject, mut rewrite: impl FnMut(JsValue) -> JsValue) {
let Some(state) = async_gen_state_mut(obj) else {
return;
};
state.callee = rewrite(state.callee);
for v in &mut state.args {
*v = rewrite(*v);
}
state.result = rewrite(state.result);
for req in state.queue.iter_mut() {
req.promise = rewrite(req.promise);
req.resolve = rewrite(req.resolve);
req.reject = rewrite(req.reject);
}
if let Some(req) = &mut state.current {
req.promise = rewrite(req.promise);
req.resolve = rewrite(req.resolve);
req.reject = rewrite(req.reject);
}
state.suspended.rewrite_values(rewrite);
}
/// 深拷贝状态盒到新对象(promote / sweep 用):新对象持独立 Box,源盒可安全释放。
#[allow(dead_code)]
pub(crate) fn clone_async_generator_native_with_rewrite(
old: &JsObject, new: &mut JsObject, mut rewrite: impl FnMut(JsValue) -> JsValue,
) {
let Some(state) = async_gen_state_mut(old) else {
return;
};
let cloned = AsyncGeneratorState {
phase: state.phase,
callee: rewrite(state.callee),
args: state.args.iter().copied().map(&mut rewrite).collect(),
this_value: state.this_value,
result: rewrite(state.result),
queue: state
.queue
.iter()
.map(|req| AsyncGenRequest {
mode: req.mode,
promise: rewrite(req.promise),
resolve: rewrite(req.resolve),
reject: rewrite(req.reject),
})
.collect(),
current: state.current.as_ref().map(|req| AsyncGenRequest {
mode: req.mode,
promise: rewrite(req.promise),
resolve: rewrite(req.resolve),
reject: rewrite(req.reject),
}),
delegate_pending: state.delegate_pending,
suspended: state.suspended.clone_with_rewrite(rewrite),
};
new.set_native_data(Box::into_raw(Box::new(cloned)) as *mut u8);
}
/// 只读核算异步生成器状态盒字节(不释放),供 GC 账目核算。
pub(crate) fn async_generator_native_size(obj: &JsObject) -> u64 {
if !obj.is_async_generator_obj() {
-33
View File
@@ -1022,39 +1022,6 @@ pub(crate) fn generator_native_cell_edges(obj: &JsObject) -> Vec<*mut Cell> {
out
}
/// 用转发函数重写状态快照中的所有 JsValue(promote 用)。
#[allow(dead_code)]
pub(crate) fn rewrite_generator_native(obj: &JsObject, mut rewrite: impl FnMut(JsValue) -> JsValue) {
let Some(state) = generator_state_mut(obj) else {
return;
};
state.callee = rewrite(state.callee);
for v in &mut state.args {
*v = rewrite(*v);
}
state.result = rewrite(state.result);
state.suspended.rewrite_values(rewrite);
}
/// 深拷贝状态盒到新对象(promote / sweep 用):新对象持独立 Box,源盒可安全释放。
#[allow(dead_code)]
pub(crate) fn clone_generator_native_with_rewrite(
old: &JsObject, new: &mut JsObject, mut rewrite: impl FnMut(JsValue) -> JsValue,
) {
let Some(state) = generator_state_mut(old) else {
return;
};
let cloned = GeneratorState {
phase: state.phase,
callee: rewrite(state.callee),
args: state.args.iter().copied().map(&mut rewrite).collect(),
this_value: state.this_value,
result: rewrite(state.result),
suspended: state.suspended.clone_with_rewrite(rewrite),
};
new.set_native_data(Box::into_raw(Box::new(cloned)) as *mut u8);
}
/// 只读核算生成器状态盒字节(不释放),供 GC 账目核算。
pub(crate) fn generator_native_size(obj: &JsObject) -> u64 {
if !obj.is_generator_obj() {
+1 -274
View File
@@ -1,7 +1,7 @@
//! 挂起执行上下文快照:生成器 / 异步函数 / 异步生成器 三份状态结构体共享的执行核心。
//!
//! 挂起时把 VM 的执行核心(regs/pc/bytecode/各栈段/在途异常与完成)整体搬入,
//! 恢复时搬回;GC 侧对快照的根收集与指针重写也统一走本结构的方法。
//! 恢复时搬回;GC 侧对快照的根收集统一走 for_each_value。
//! 约定:本结构是"值"的容器,不含调度标志(generator_dispatch/async_*_dispatch/
//! async_context 等仍由各恢复包装函数管理,避免跨状态机污染)。
@@ -177,7 +177,6 @@ impl SuspendedFrame {
}
/// GC 根遍历:产出全部 JsValue(对象与字符串都产出),含 cell/for_in 解引用。
/// 与 `rewrite_values` 字段一一对应。
pub fn for_each_value(&self, mut f: impl FnMut(JsValue)) {
for v in self.regs.iter() {
f(*v);
@@ -242,197 +241,6 @@ impl SuspendedFrame {
}
}
/// GC 指针重写:与 `for_each_value` 字段一一对应。
#[allow(dead_code)]
pub fn rewrite_values(&mut self, mut rewrite: impl FnMut(JsValue) -> JsValue) {
for v in self.regs.iter_mut() {
*v = rewrite(*v);
}
if let Some(frame) = &mut self.frame {
frame.saved_this = rewrite(frame.saved_this);
frame.saved_new_target = rewrite(frame.saved_new_target);
frame.callee = rewrite(frame.callee);
frame.constructed_this = frame.constructed_this.map(&mut rewrite);
}
for v in &mut self.spill_stack {
*v = rewrite(*v);
}
for v in &mut self.save_stack {
*v = rewrite(*v);
}
for cells in &mut self.cell_stack {
for &mut p in cells.iter_mut() {
if p.is_null() {
continue;
}
// SAFETY: cell 经 alloc_cell 独立堆分配,本 session 内指针有效。
let cell = unsafe { &mut *p };
cell.value = rewrite(cell.value);
}
}
for entry in &mut self.for_of_iters {
entry.iterator = rewrite(entry.iterator);
entry.last_result = rewrite(entry.last_result);
entry.fast_value = rewrite(entry.fast_value);
}
self.delegated_iterator = self.delegated_iterator.map(&mut rewrite);
self.exception_value = self.exception_value.map(&mut rewrite);
self.pending_exception = self.pending_exception.map(&mut rewrite);
self.pending_completion = self.pending_completion.map(|completion| match completion {
Completion::Return {
value,
remaining_finally,
for_of_count,
for_in_count,
} => Completion::Return {
value: rewrite(value),
remaining_finally,
for_of_count,
for_in_count,
},
other => other,
});
self.pending_async_escape = self.pending_async_escape.as_mut().map(|pend| {
let completion = match pend.completion {
Completion::Return {
value,
remaining_finally,
for_of_count,
for_in_count,
} => Completion::Return {
value: rewrite(value),
remaining_finally,
for_of_count,
for_in_count,
},
other => other,
};
PendingAsyncEscape {
close_promise: rewrite(pend.close_promise),
completion,
remaining: pend.remaining.iter().copied().map(&mut rewrite).collect(),
}
});
for iter in &mut self.for_in_iters {
if iter.is_null() {
continue;
}
// SAFETY: for_in_iters 存放堆上迭代器体,状态盒独占持有。
for (v, _si) in unsafe { (*(*iter)).keys.iter_mut() } {
*v = rewrite(*v);
}
}
}
/// 深拷贝(promote / sweep 搬移用,替代 async/asyncgen 各自的 clone_*_with_rewrite)。
#[allow(dead_code)]
pub fn clone_with_rewrite(&self, mut rewrite: impl FnMut(JsValue) -> JsValue) -> Self {
SuspendedFrame {
regs: Box::new({
let mut regs = [JsValue::undefined(); 256];
for (i, v) in self.regs.iter().enumerate() {
regs[i] = rewrite(*v);
}
regs
}),
pc: self.pc,
bytecode: self.bytecode.clone(),
sub_idx: self.sub_idx,
active_reg_limit: self.active_reg_limit,
root_reg_limit: self.root_reg_limit,
frame: self.frame.as_ref().map(|f| CallFrame {
return_addr: f.return_addr,
function_name: f.function_name,
caller_reg_limit: f.caller_reg_limit,
caller_active_reg_limit: f.caller_active_reg_limit,
saved_reg_offset: f.saved_reg_offset,
spill_offset: f.spill_offset,
arguments_base: f.arguments_base,
arguments_count: f.arguments_count,
saved_this: rewrite(f.saved_this),
saved_new_target: rewrite(f.saved_new_target),
callee: rewrite(f.callee),
construct_result_reg: f.construct_result_reg,
constructed_this: f.constructed_this.map(&mut rewrite),
is_derived_constructor: f.is_derived_constructor,
super_called: f.super_called,
strict: f.strict,
continuation: f.continuation,
upvalues: f.upvalues,
}),
spill_stack: self.spill_stack.iter().copied().map(&mut rewrite).collect(),
save_stack: self.save_stack.iter().copied().map(&mut rewrite).collect(),
cell_stack: self.cell_stack.clone(),
try_stack: self.try_stack.clone(),
for_in_iters: self
.for_in_iters
.iter()
.map(|&iter| {
if iter.is_null() {
return iter;
}
// SAFETY: 指针是堆上迭代器体,原件状态盒独占持有;克隆新建
// 独立体,原件与克隆各持自有体,恰好各释放一次。
let body = unsafe { &*iter };
Box::into_raw(Box::new(ForInIter {
keys: body.keys.iter().map(|(v, si)| (rewrite(*v), *si)).collect(),
index: body.index,
}))
})
.collect(),
for_of_iters: self
.for_of_iters
.iter()
.copied()
.map(|entry| ForOfEntry {
iterator: rewrite(entry.iterator),
last_result: rewrite(entry.last_result),
is_async: entry.is_async,
fast: entry.fast,
fast_value: rewrite(entry.fast_value),
})
.collect(),
delegated_iterator: self.delegated_iterator.map(&mut rewrite),
saved_bytecode_stack: self.saved_bytecode_stack.clone(),
saved_immutables_stack: self.saved_immutables_stack.clone(),
exception_value: self.exception_value.map(&mut rewrite),
pending_exception: self.pending_exception.map(&mut rewrite),
pending_error_kind: self.pending_error_kind,
pending_completion: self.pending_completion.map(|completion| match completion {
Completion::Return {
value,
remaining_finally,
for_of_count,
for_in_count,
} => Completion::Return {
value: rewrite(value),
remaining_finally,
for_of_count,
for_in_count,
},
other => other,
}),
pending_async_escape: self.pending_async_escape.as_ref().map(|p| PendingAsyncEscape {
close_promise: rewrite(p.close_promise),
completion: match p.completion {
Completion::Return {
value,
remaining_finally,
for_of_count,
for_in_count,
} => Completion::Return {
value: rewrite(value),
remaining_finally,
for_of_count,
for_in_count,
},
other => other,
},
remaining: p.remaining.iter().copied().map(&mut rewrite).collect(),
}),
}
}
/// 堆字节数(drop 记账用:bytecode + spill/save/for_of 容量)。
pub fn heap_bytes(&self) -> u64 {
self.bytecode.len() as u64 * std::mem::size_of::<opcode::Instr>() as u64
@@ -463,8 +271,6 @@ impl Drop for SuspendedFrame {
mod tests {
use super::*;
use crate::vm::{FrameContinuation, Vm};
use std::collections::HashSet;
fn vm() -> Vm {
Vm::new()
}
@@ -492,85 +298,6 @@ mod tests {
}
}
fn fill_frame(frame: &mut SuspendedFrame) {
frame.regs[0] = JsValue::float(1.0);
frame.regs[7] = JsValue::float(2.0);
frame.pc = 11;
frame.bytecode = Arc::from(vec![0u32]);
frame.sub_idx = 3;
frame.active_reg_limit = 4;
frame.root_reg_limit = 5;
frame.frame = Some(sentinel_frame(JsValue::float(3.0)));
frame.spill_stack.push(JsValue::float(4.0));
frame.save_stack.push(JsValue::float(5.0));
frame.cell_stack.push(Vec::new());
frame.try_stack.push(TryHandler {
catch_pc: None,
finally_pc: Some(2),
finally_active: false,
frame_depth: 0,
for_of_depth: 0,
});
frame.for_of_iters.push(ForOfEntry {
iterator: JsValue::float(6.0),
last_result: JsValue::float(7.0),
is_async: false,
fast: None,
fast_value: JsValue::undefined(),
});
frame.delegated_iterator = Some(JsValue::float(8.0));
frame.saved_bytecode_stack.push(Arc::from(vec![0u32]));
frame
.saved_immutables_stack
.push(std::ptr::slice_from_raw_parts(std::ptr::null(), 0));
frame.exception_value = Some(JsValue::float(9.0));
frame.pending_exception = Some(JsValue::float(10.0));
frame.pending_error_kind = Some("Error");
frame.pending_completion = Some(Completion::Return {
value: JsValue::float(11.0),
remaining_finally: 0,
for_of_count: 0,
for_in_count: 0,
});
frame.pending_async_escape = Some(PendingAsyncEscape {
close_promise: JsValue::float(12.0),
completion: Completion::Return {
value: JsValue::float(13.0),
remaining_finally: 0,
for_of_count: 0,
for_in_count: 0,
},
remaining: vec![JsValue::float(14.0)],
});
}
#[test]
fn for_each_rewrite_cover_same_fields() {
let mut frame = SuspendedFrame::new_empty();
fill_frame(&mut frame);
// for_each_value 与 rewrite_values 必须覆盖同一组字段:先统计 for_each 访问
// 的位点数,再把 rewrite 访问的每个位点改写为唯一值,最后 for_each 收集应
// 恰好产出全部唯一值——若两侧字段不对称则计数不等。
let mut for_each_count = 0usize;
frame.for_each_value(|_| for_each_count += 1);
let mut rewrite_count = 0usize;
let mut n = 1000.0f64;
frame.rewrite_values(|_v| {
rewrite_count += 1;
n += 1.0;
JsValue::float(n)
});
assert_eq!(rewrite_count, for_each_count);
let mut collected = HashSet::new();
frame.for_each_value(|v| {
collected.insert(v.to_bits());
});
assert_eq!(collected.len(), rewrite_count);
}
#[test]
fn save_restore_round_trip() {
let mut machine = vm();