forge_lint/sol/analysis/

access_control.rs

//! Access-control guard detection: whether a function dominates its body with a check comparing
//! the caller against contract state, and which state that check depends on.

use super::{
    branch_always_exits, is_require_or_assert, is_sender_member, lhs_local_var, loop_stmts,
    stmt_expr, underlying_var, visit_stmts,
};
use solar::sema::{
    Gcx,
    hir::{self, BinOpKind, Expr, ExprKind, FunctionId, Stmt, StmtKind, UnOpKind, VariableId},
};
use std::{collections::HashSet, iter, ops::ControlFlow};

/// True when the function or one of its modifiers contains a dominating access check.
pub fn is_protected<'gcx>(gcx: Gcx<'gcx>, func_id: FunctionId) -> bool {
    modifiers_and_self(gcx, func_id).any(|id| has_access_guard(gcx, id, &mut HashSet::new()))
}

/// The modifiers of `func_id` that resolve to functions, followed by `func_id` itself.
pub fn modifiers_and_self<'gcx>(
    gcx: Gcx<'gcx>,
    func_id: FunctionId,
) -> impl Iterator<Item = FunctionId> + 'gcx {
    gcx.hir
        .function(func_id)
        .modifiers
        .iter()
        .filter_map(move |modifier| {
            gcx.hir.function(func_id).contract.map_or_else(
                || modifier.id.as_function(),
                |contract| gcx.resolve_modifier_target(contract, modifier),
            )
        })
        .chain(iter::once(func_id))
}

/// Whether `func_id` checks the caller before its `_` placeholder (anywhere for functions): a
/// guarding `if`, a `require`/`assert` on an access check, or a call into a function that does.
/// Bodyless declarations (interface functions, virtual modifiers) fall back to a name heuristic.
pub fn has_access_guard<'gcx>(
    gcx: Gcx<'gcx>,
    func_id: FunctionId,
    seen: &mut HashSet<FunctionId>,
) -> bool {
    if !seen.insert(func_id) {
        return false;
    }
    let func = gcx.hir.function(func_id);
    match func.body {
        Some(body) => for_each_guard(gcx, body, seen, &mut |_| ControlFlow::Break(())).is_break(),
        None => looks_like_access_control(func),
    }
}

/// State variables the access checks of `func_id` and its modifiers (up to `_`) depend on.
pub fn guard_vars<'gcx>(gcx: Gcx<'gcx>, func_id: FunctionId) -> HashSet<VariableId> {
    let mut out = HashSet::new();
    for id in modifiers_and_self(gcx, func_id) {
        let Some(body) = gcx.hir.function(id).body else { continue };
        let mut seen = HashSet::from([id]);
        let _ = for_each_guard(gcx, body, &mut HashSet::from([id]), &mut |guard| {
            match guard {
                Guard::Check(cond) => expr_state_vars(gcx, cond, &mut seen, &mut out),
                Guard::Call(callee_id) => function_state_vars(gcx, callee_id, &mut seen, &mut out),
            }
            ControlFlow::Continue(())
        });
    }
    out
}

/// A function whose name marks it as an access check (`auth`, `onlyOwner`, `_checkRole`, ...)
/// and that returns nothing, so calling it for its effect is meaningful.
pub fn looks_like_access_control(func: &hir::Function<'_>) -> bool {
    let Some(name) = func.name else { return false };
    if !func.returns.is_empty() {
        return false;
    }
    let lower = name.as_str().to_ascii_lowercase();
    matches!(lower.as_str(), "auth" | "requiresauth" | "restricted")
        || ["only", "check", "_check"].iter().any(|prefix| {
            ["admin", "guardian", "manager", "owner", "role"]
                .iter()
                .any(|role| lower.starts_with(&format!("{prefix}{role}")))
        })
}

/// `Some(true)` when `expr` holding means the caller is authorized, `Some(false)` when it means
/// the caller is *not* authorized, `None` when `expr` is not an access check. An access check
/// reads `msg.sender`/`tx.origin` (directly, through `aliases` or through a helper) and state
/// (directly or through a helper).
pub fn access_check_polarity<'gcx>(
    gcx: Gcx<'gcx>,
    expr: &Expr<'_>,
    aliases: &HashSet<VariableId>,
) -> Option<bool> {
    let is_check = |sender: &Expr<'_>, authority: &Expr<'_>| {
        expr_reads_sender(gcx, sender, &mut HashSet::new(), aliases)
            && expr_reads_state(gcx, authority)
    };
    match &expr.peel_parens().kind {
        ExprKind::Unary(op, inner) if op.kind == UnOpKind::Not => {
            access_check_polarity(gcx, inner, aliases).map(|polarity| !polarity)
        }
        ExprKind::Binary(lhs, op, rhs) if matches!(op.kind, BinOpKind::And | BinOpKind::Or) => {
            // `a && b` is authorized as soon as one side is; `a || b` is unauthorized as soon as
            // one side is. The opposite polarity needs both sides.
            let dominant = op.kind == BinOpKind::And;
            let lhs = access_check_polarity(gcx, lhs, aliases);
            let rhs = access_check_polarity(gcx, rhs, aliases);
            if lhs == Some(dominant) || rhs == Some(dominant) {
                Some(dominant)
            } else if lhs == Some(!dominant) && rhs == Some(!dominant) {
                Some(!dominant)
            } else {
                None
            }
        }
        ExprKind::Binary(lhs, op, rhs)
            if matches!(op.kind, BinOpKind::Eq | BinOpKind::Ne)
                && (is_check(lhs, rhs) || is_check(rhs, lhs)) =>
        {
            Some(op.kind == BinOpKind::Eq)
        }
        _ => is_check(expr, expr).then_some(true),
    }
}

/// Applies `stmt` to the set of locals holding a `msg.sender`-derived value: a local initialized
/// or assigned from a value that reads the sender becomes an alias, and one reassigned from
/// anything else stops being one.
fn update_sender_aliases<'gcx>(
    gcx: Gcx<'gcx>,
    stmt: &Stmt<'gcx>,
    aliases: &mut HashSet<VariableId>,
) {
    let (var_id, value) = match stmt.kind {
        StmtKind::DeclSingle(var_id) => (Some(var_id), gcx.hir.variable(var_id).initializer),
        StmtKind::Expr(expr) => match &expr.peel_parens().kind {
            ExprKind::Assign(lhs, _, rhs) => (lhs_local_var(gcx, lhs), Some(*rhs)),
            _ => (None, None),
        },
        _ => (None, None),
    };
    if let Some(var_id) = var_id
        && let Some(value) = value
    {
        if expr_reads_sender(gcx, value, &mut HashSet::new(), aliases) {
            aliases.insert(var_id);
        } else {
            aliases.remove(&var_id);
        }
    }
}

/// Whether `expr` reads `msg.sender`/`tx.origin`, one of `aliases`, or calls a user function that
/// reads the sender.
pub fn expr_reads_sender<'gcx>(
    gcx: Gcx<'gcx>,
    expr: &Expr<'_>,
    seen: &mut HashSet<FunctionId>,
    aliases: &HashSet<VariableId>,
) -> bool {
    expr.visit(&mut |e| {
        let reads = is_sender_member(gcx, e)
            || underlying_var(gcx, e).is_some_and(|v| aliases.contains(&v))
            || matches!(&e.kind, ExprKind::Call(callee, ..)
                if matches!(callee.peel_parens().kind, ExprKind::Ident(_))
                    && gcx.resolved_function(callee).is_some_and(|id| function_reads_sender(gcx, id, seen)));
        if reads { ControlFlow::Break(()) } else { ControlFlow::Continue(()) }
    })
    .is_break()
}

/// Whether the body of `func_id` reads `msg.sender`/`tx.origin`, following calls.
pub fn function_reads_sender<'gcx>(
    gcx: Gcx<'gcx>,
    func_id: FunctionId,
    seen: &mut HashSet<FunctionId>,
) -> bool {
    seen.insert(func_id)
        && gcx.hir.function(func_id).body.is_some_and(|body| {
            visit_stmts(&gcx.hir, body.stmts, |stmt| {
                let reads = stmt_expr(&gcx.hir, stmt)
                    .is_some_and(|expr| expr_reads_sender(gcx, expr, seen, &HashSet::new()));
                if reads { ControlFlow::Break(()) } else { ControlFlow::Continue(()) }
            })
            .is_break()
        })
}

/// State variables read by `expr`, following calls into user functions.
pub fn expr_state_vars<'gcx>(
    gcx: Gcx<'gcx>,
    expr: &Expr<'_>,
    seen: &mut HashSet<FunctionId>,
    out: &mut HashSet<VariableId>,
) {
    let _ = expr.visit(&mut |e| {
        if let Some(var_id) = underlying_var(gcx, e)
            && gcx.hir.variable(var_id).kind.is_state()
        {
            out.insert(var_id);
        }
        if let ExprKind::Call(callee, ..) = &e.kind
            && matches!(callee.peel_parens().kind, ExprKind::Ident(_))
            && let Some(callee_id) = gcx.resolved_function(callee)
        {
            function_state_vars(gcx, callee_id, seen, out);
        }
        ControlFlow::<()>::Continue(())
    });
}

/// State variables read by the body of `func_id`, following calls into user functions.
pub fn function_state_vars<'gcx>(
    gcx: Gcx<'gcx>,
    func_id: FunctionId,
    seen: &mut HashSet<FunctionId>,
    out: &mut HashSet<VariableId>,
) {
    if seen.insert(func_id)
        && let Some(body) = gcx.hir.function(func_id).body
    {
        let _ = visit_stmts(&gcx.hir, body.stmts, |stmt| {
            if let Some(expr) = stmt_expr(&gcx.hir, stmt) {
                expr_state_vars(gcx, expr, seen, out);
            }
            ControlFlow::Continue(())
        });
    }
}

fn expr_reads_state<'gcx>(gcx: Gcx<'gcx>, expr: &Expr<'_>) -> bool {
    let mut vars = HashSet::new();
    expr_state_vars(gcx, expr, &mut HashSet::new(), &mut vars);
    !vars.is_empty()
}

/// An access check among the dominating statements of a function body.
enum Guard<'a> {
    /// The condition of a guarding `if` or of a `require`/`assert`.
    Check(&'a Expr<'a>),
    /// A call into a function that itself checks the caller.
    Call(FunctionId),
}

/// Calls `f` for every access check that dominates `body` (runs unconditionally before the `_`
/// placeholder) until it breaks. `seen` guards the recursion into called functions.
fn for_each_guard<'gcx>(
    gcx: Gcx<'gcx>,
    body: hir::Block<'gcx>,
    seen: &mut HashSet<FunctionId>,
    f: &mut impl FnMut(Guard<'_>) -> ControlFlow<()>,
) -> ControlFlow<()> {
    let mut stmts = Vec::new();
    let _ = dominating_stmts(body.stmts, &mut stmts);
    // Aliases as of each statement: a check is evaluated against the locals that read the sender
    // at that point, so a reassignment neither validates a later check nor invalidates an earlier
    // one.
    let mut aliases = HashSet::new();
    for stmt in stmts {
        if let StmtKind::If(cond, then_stmt, else_stmt) = stmt.kind {
            let exits = match access_check_polarity(gcx, cond, &aliases) {
                Some(false) => branch_always_exits(gcx, then_stmt),
                Some(true) => else_stmt.is_some_and(|expr| branch_always_exits(gcx, expr)),
                None => false,
            };
            if exits {
                f(Guard::Check(cond))?;
            }
            continue;
        }
        update_sender_aliases(gcx, stmt, &mut aliases);
        let Some(expr) = stmt_expr(&gcx.hir, stmt) else { continue };
        expr.visit(&mut |e| {
            match &e.kind {
                ExprKind::Call(callee, args, _) if is_require_or_assert(gcx, callee) => {
                    if let Some(cond) = args.exprs().next()
                        && access_check_polarity(gcx, cond, &aliases) == Some(true)
                    {
                        f(Guard::Check(cond))?;
                    }
                }
                ExprKind::Call(callee, ..)
                    if matches!(callee.peel_parens().kind, ExprKind::Ident(_)) =>
                {
                    if let Some(callee_id) = gcx.resolved_function(callee)
                        && (looks_like_access_control(gcx.hir.function(callee_id))
                            || has_access_guard(gcx, callee_id, seen))
                    {
                        f(Guard::Call(callee_id))?;
                    }
                }
                _ => {}
            }
            ControlFlow::Continue(())
        })?;
    }
    ControlFlow::Continue(())
}

/// Collects into `out` the statements that run unconditionally before the `_` placeholder (all of
/// them for functions), descending into blocks and loops. Breaks when the placeholder is reached.
fn dominating_stmts<'gcx>(
    stmts: impl IntoIterator<Item = &'gcx Stmt<'gcx>>,
    out: &mut Vec<&'gcx Stmt<'gcx>>,
) -> ControlFlow<()> {
    for stmt in stmts {
        match stmt.kind {
            StmtKind::Placeholder => return ControlFlow::Break(()),
            StmtKind::Block(block) | StmtKind::UncheckedBlock(block) => {
                dominating_stmts(block.stmts, out)?;
            }
            StmtKind::Loop(block, source) => dominating_stmts(loop_stmts(block, source), out)?,
            _ => out.push(stmt),
        }
    }
    ControlFlow::Continue(())
}