mirror of
https://github.com/rust-lang/rust.git
synced 2025-01-10 14:57:14 +00:00
Port middle/stack_check.rs from oldvisit to <V:Visitor> trait API.
This commit is contained in:
parent
6af6b088ad
commit
da88f69f06
@ -22,7 +22,8 @@ use syntax::ast;
|
|||||||
use syntax::ast_map;
|
use syntax::ast_map;
|
||||||
use syntax::attr;
|
use syntax::attr;
|
||||||
use syntax::codemap::span;
|
use syntax::codemap::span;
|
||||||
use visit = syntax::oldvisit;
|
use syntax::visit;
|
||||||
|
use syntax::visit::Visitor;
|
||||||
use util::ppaux::Repr;
|
use util::ppaux::Repr;
|
||||||
|
|
||||||
#[deriving(Clone)]
|
#[deriving(Clone)]
|
||||||
@ -31,44 +32,56 @@ struct Context {
|
|||||||
safe_stack: bool
|
safe_stack: bool
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct StackCheckVisitor;
|
||||||
|
|
||||||
|
impl Visitor<Context> for StackCheckVisitor {
|
||||||
|
fn visit_item(&mut self, i:@ast::item, e:Context) {
|
||||||
|
stack_check_item(*self, i, e);
|
||||||
|
}
|
||||||
|
fn visit_fn(&mut self, fk:&visit::fn_kind, fd:&ast::fn_decl,
|
||||||
|
b:&ast::Block, s:span, n:ast::NodeId, e:Context) {
|
||||||
|
stack_check_fn(*self, fk, fd, b, s, n, e);
|
||||||
|
}
|
||||||
|
fn visit_expr(&mut self, ex:@ast::expr, e:Context) {
|
||||||
|
stack_check_expr(*self, ex, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn stack_check_crate(tcx: ty::ctxt,
|
pub fn stack_check_crate(tcx: ty::ctxt,
|
||||||
crate: &ast::Crate) {
|
crate: &ast::Crate) {
|
||||||
let new_cx = Context {
|
let new_cx = Context {
|
||||||
tcx: tcx,
|
tcx: tcx,
|
||||||
safe_stack: false
|
safe_stack: false
|
||||||
};
|
};
|
||||||
let visitor = visit::mk_vt(@visit::Visitor {
|
let mut visitor = StackCheckVisitor;
|
||||||
visit_item: stack_check_item,
|
visit::walk_crate(&mut visitor, crate, new_cx);
|
||||||
visit_fn: stack_check_fn,
|
|
||||||
visit_expr: stack_check_expr,
|
|
||||||
..*visit::default_visitor()
|
|
||||||
});
|
|
||||||
visit::visit_crate(crate, (new_cx, visitor));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn stack_check_item(item: @ast::item,
|
fn stack_check_item(v: StackCheckVisitor,
|
||||||
(in_cx, v): (Context, visit::vt<Context>)) {
|
item: @ast::item,
|
||||||
|
in_cx: Context) {
|
||||||
|
let mut v = v;
|
||||||
match item.node {
|
match item.node {
|
||||||
ast::item_fn(_, ast::extern_fn, _, _, _) => {
|
ast::item_fn(_, ast::extern_fn, _, _, _) => {
|
||||||
// an extern fn is already being called from C code...
|
// an extern fn is already being called from C code...
|
||||||
let new_cx = Context {safe_stack: true, ..in_cx};
|
let new_cx = Context {safe_stack: true, ..in_cx};
|
||||||
visit::visit_item(item, (new_cx, v));
|
visit::walk_item(&mut v, item, new_cx);
|
||||||
}
|
}
|
||||||
ast::item_fn(*) => {
|
ast::item_fn(*) => {
|
||||||
let safe_stack = fixed_stack_segment(item.attrs);
|
let safe_stack = fixed_stack_segment(item.attrs);
|
||||||
let new_cx = Context {safe_stack: safe_stack, ..in_cx};
|
let new_cx = Context {safe_stack: safe_stack, ..in_cx};
|
||||||
visit::visit_item(item, (new_cx, v));
|
visit::walk_item(&mut v, item, new_cx);
|
||||||
}
|
}
|
||||||
ast::item_impl(_, _, _, ref methods) => {
|
ast::item_impl(_, _, _, ref methods) => {
|
||||||
// visit_method() would make this nicer
|
// visit_method() would make this nicer
|
||||||
for &method in methods.iter() {
|
for &method in methods.iter() {
|
||||||
let safe_stack = fixed_stack_segment(method.attrs);
|
let safe_stack = fixed_stack_segment(method.attrs);
|
||||||
let new_cx = Context {safe_stack: safe_stack, ..in_cx};
|
let new_cx = Context {safe_stack: safe_stack, ..in_cx};
|
||||||
visit::visit_method_helper(method, (new_cx, v));
|
visit::walk_method_helper(&mut v, method, new_cx);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ => {
|
_ => {
|
||||||
visit::visit_item(item, (in_cx, v));
|
visit::walk_item(&mut v, item, in_cx);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -77,12 +90,13 @@ fn stack_check_item(item: @ast::item,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn stack_check_fn<'a>(fk: &visit::fn_kind,
|
fn stack_check_fn<'a>(v: StackCheckVisitor,
|
||||||
|
fk: &visit::fn_kind,
|
||||||
decl: &ast::fn_decl,
|
decl: &ast::fn_decl,
|
||||||
body: &ast::Block,
|
body: &ast::Block,
|
||||||
sp: span,
|
sp: span,
|
||||||
id: ast::NodeId,
|
id: ast::NodeId,
|
||||||
(in_cx, v): (Context, visit::vt<Context>)) {
|
in_cx: Context) {
|
||||||
let safe_stack = match *fk {
|
let safe_stack = match *fk {
|
||||||
visit::fk_method(*) | visit::fk_item_fn(*) => {
|
visit::fk_method(*) | visit::fk_item_fn(*) => {
|
||||||
in_cx.safe_stack // see stack_check_item above
|
in_cx.safe_stack // see stack_check_item above
|
||||||
@ -102,11 +116,13 @@ fn stack_check_fn<'a>(fk: &visit::fn_kind,
|
|||||||
};
|
};
|
||||||
let new_cx = Context {safe_stack: safe_stack, ..in_cx};
|
let new_cx = Context {safe_stack: safe_stack, ..in_cx};
|
||||||
debug!("stack_check_fn(safe_stack=%b, id=%?)", safe_stack, id);
|
debug!("stack_check_fn(safe_stack=%b, id=%?)", safe_stack, id);
|
||||||
visit::visit_fn(fk, decl, body, sp, id, (new_cx, v));
|
let mut v = v;
|
||||||
|
visit::walk_fn(&mut v, fk, decl, body, sp, id, new_cx);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn stack_check_expr<'a>(expr: @ast::expr,
|
fn stack_check_expr<'a>(v: StackCheckVisitor,
|
||||||
(cx, v): (Context, visit::vt<Context>)) {
|
expr: @ast::expr,
|
||||||
|
cx: Context) {
|
||||||
debug!("stack_check_expr(safe_stack=%b, expr=%s)",
|
debug!("stack_check_expr(safe_stack=%b, expr=%s)",
|
||||||
cx.safe_stack, expr.repr(cx.tcx));
|
cx.safe_stack, expr.repr(cx.tcx));
|
||||||
if !cx.safe_stack {
|
if !cx.safe_stack {
|
||||||
@ -126,7 +142,8 @@ fn stack_check_expr<'a>(expr: @ast::expr,
|
|||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
visit::visit_expr(expr, (cx, v));
|
let mut v = v;
|
||||||
|
visit::walk_expr(&mut v, expr, cx);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn call_to_extern_fn(cx: Context, callee: @ast::expr) {
|
fn call_to_extern_fn(cx: Context, callee: @ast::expr) {
|
||||||
|
Loading…
Reference in New Issue
Block a user