rust/src/attrs.rs

/// checks for attributes

use rustc::plugin::Registry;
use rustc::lint::*;
use syntax::ast::*;
use syntax::ptr::P;
use syntax::codemap::{Span, ExpnInfo};
use syntax::parse::token::InternedString;
use utils::{in_macro, match_path, span_lint};

declare_lint! { pub INLINE_ALWAYS, Warn,
    "#[inline(always)] is usually a bad idea."}


#[derive(Copy,Clone)]
pub struct AttrPass;

impl LintPass for AttrPass {
    fn get_lints(&self) -> LintArray {
        lint_array!(INLINE_ALWAYS)
    }

    fn check_item(&mut self, cx: &Context, item: &Item) {
        if is_relevant_item(item) {
            cx.sess().codemap().with_expn_info(item.span.expn_id,
                |info| check_attrs(cx, info, &item.ident, &item.attrs))
        }
    }

    fn check_impl_item(&mut self, cx: &Context, item: &ImplItem) {
        if is_relevant_impl(item) {
            cx.sess().codemap().with_expn_info(item.span.expn_id,
                |info| check_attrs(cx, info, &item.ident, &item.attrs))
        }
    }

    fn check_trait_item(&mut self, cx: &Context, item: &TraitItem) {
        if is_relevant_trait(item) {
            cx.sess().codemap().with_expn_info(item.span.expn_id,
                |info| check_attrs(cx, info, &item.ident, &item.attrs))
        }
    }
}

fn is_relevant_item(item: &Item) -> bool {
    if let &ItemFn(_, _, _, _, _, ref block) = &item.node {
        is_relevant_block(block)
    } else { false }
}

fn is_relevant_impl(item: &ImplItem) -> bool {
    match item.node {
        MethodImplItem(_, ref block) => is_relevant_block(block),
        _ => false
    }
}

fn is_relevant_trait(item: &TraitItem) -> bool {
    match item.node {
        MethodTraitItem(_, None) => true,
        MethodTraitItem(_, Some(ref block)) => is_relevant_block(block),
        _ => false
    }
}

fn is_relevant_block(block: &Block) -> bool {
    for stmt in block.stmts.iter() {
        match stmt.node {
            StmtDecl(_, _) => return true,
            StmtExpr(ref expr, _) | StmtSemi(ref expr, _) => {
                return is_relevant_expr(expr);
            }
            _ => ()
        }
    }
    block.expr.as_ref().map_or(false, |e| is_relevant_expr(&*e))
}

fn is_relevant_expr(expr: &Expr) -> bool {
    match expr.node {
        ExprBlock(ref block) => is_relevant_block(block),
        ExprRet(Some(ref e)) | ExprParen(ref e) =>
            is_relevant_expr(&*e),
        ExprRet(None) | ExprBreak(_) | ExprMac(_) => false,
        ExprCall(ref path_expr, _) => {
            if let ExprPath(_, ref path) = path_expr.node {
                !match_path(path, &["std", "rt", "begin_unwind"])
            } else { true }
        }
        _ => true
    }
}

fn check_attrs(cx: &Context, info: Option<&ExpnInfo>, ident: &Ident,
               attrs: &[Attribute]) {
    if in_macro(cx, info) { return; }

    for attr in attrs {
        if let MetaList(ref inline, ref values) = attr.node.value.node {
            if values.len() != 1 || inline != &"inline" { continue; }
            if let MetaWord(ref always) = values[0].node {
                if always != &"always" { continue; }
                span_lint(cx, INLINE_ALWAYS, attr.span, &format!(
                    "You have declared #[inline(always)] on {}. This \
                     is usually a bad idea. Are you sure?",
                    ident.name.as_str()));
            }
        }
    }
}