]> git.lizzy.rs Git - rust.git/blob - src/librustc/lint/context.rs
Changes to tests
[rust.git] / src / librustc / lint / context.rs
1 // Copyright 2012-2015 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
4 //
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
10
11 //! Implementation of lint checking.
12 //!
13 //! The lint checking is mostly consolidated into one pass which runs just
14 //! before translation to LLVM bytecode. Throughout compilation, lint warnings
15 //! can be added via the `add_lint` method on the Session structure. This
16 //! requires a span and an id of the node that the lint is being added to. The
17 //! lint isn't actually emitted at that time because it is unknown what the
18 //! actual lint level at that location is.
19 //!
20 //! To actually emit lint warnings/errors, a separate pass is used just before
21 //! translation. A context keeps track of the current state of all lint levels.
22 //! Upon entering a node of the ast which can modify the lint settings, the
23 //! previous lint state is pushed onto a stack and the ast is then recursed
24 //! upon.  As the ast is traversed, this keeps track of the current lint level
25 //! for all lint attributes.
26 use self::TargetLint::*;
27
28 use middle::privacy::ExportedItems;
29 use middle::ty::{self, Ty};
30 use session::{early_error, Session};
31 use lint::{Level, LevelSource, Lint, LintId, LintArray, LintPass, LintPassObject};
32 use lint::{Default, CommandLine, Node, Allow, Warn, Deny, Forbid};
33 use lint::builtin;
34 use util::nodemap::FnvHashMap;
35
36 use std::cell::RefCell;
37 use std::cmp;
38 use std::mem;
39 use syntax::ast_util::{self, IdVisitingOperation};
40 use syntax::attr::{self, AttrMetaMethods};
41 use syntax::codemap::Span;
42 use syntax::parse::token::InternedString;
43 use syntax::ast;
44 use rustc_front::hir;
45 use rustc_front::util;
46 use rustc_front::visit as hir_visit;
47 use syntax::visit as ast_visit;
48 use syntax::diagnostic;
49
50 /// Information about the registered lints.
51 ///
52 /// This is basically the subset of `Context` that we can
53 /// build early in the compile pipeline.
54 pub struct LintStore {
55     /// Registered lints. The bool is true if the lint was
56     /// added by a plugin.
57     lints: Vec<(&'static Lint, bool)>,
58
59     /// Trait objects for each lint pass.
60     /// This is only `None` while iterating over the objects. See the definition
61     /// of run_lints.
62     passes: Option<Vec<LintPassObject>>,
63
64     /// Lints indexed by name.
65     by_name: FnvHashMap<String, TargetLint>,
66
67     /// Current levels of each lint, and where they were set.
68     levels: FnvHashMap<LintId, LevelSource>,
69
70     /// Map of registered lint groups to what lints they expand to. The bool
71     /// is true if the lint group was added by a plugin.
72     lint_groups: FnvHashMap<&'static str, (Vec<LintId>, bool)>,
73
74     /// Maximum level a lint can be
75     lint_cap: Option<Level>,
76 }
77
78 /// The targed of the `by_name` map, which accounts for renaming/deprecation.
79 enum TargetLint {
80     /// A direct lint target
81     Id(LintId),
82
83     /// Temporary renaming, used for easing migration pain; see #16545
84     Renamed(String, LintId),
85
86     /// Lint with this name existed previously, but has been removed/deprecated.
87     /// The string argument is the reason for removal.
88     Removed(String),
89 }
90
91 enum FindLintError {
92     NotFound,
93     Removed
94 }
95
96 impl LintStore {
97     fn get_level_source(&self, lint: LintId) -> LevelSource {
98         match self.levels.get(&lint) {
99             Some(&s) => s,
100             None => (Allow, Default),
101         }
102     }
103
104     fn set_level(&mut self, lint: LintId, mut lvlsrc: LevelSource) {
105         if let Some(cap) = self.lint_cap {
106             lvlsrc.0 = cmp::min(lvlsrc.0, cap);
107         }
108         if lvlsrc.0 == Allow {
109             self.levels.remove(&lint);
110         } else {
111             self.levels.insert(lint, lvlsrc);
112         }
113     }
114
115     pub fn new() -> LintStore {
116         LintStore {
117             lints: vec!(),
118             passes: Some(vec!()),
119             by_name: FnvHashMap(),
120             levels: FnvHashMap(),
121             lint_groups: FnvHashMap(),
122             lint_cap: None,
123         }
124     }
125
126     pub fn get_lints<'t>(&'t self) -> &'t [(&'static Lint, bool)] {
127         &self.lints
128     }
129
130     pub fn get_lint_groups<'t>(&'t self) -> Vec<(&'static str, Vec<LintId>, bool)> {
131         self.lint_groups.iter().map(|(k, v)| (*k,
132                                               v.0.clone(),
133                                               v.1)).collect()
134     }
135
136     pub fn register_pass(&mut self, sess: Option<&Session>,
137                          from_plugin: bool, pass: LintPassObject) {
138         for &lint in pass.get_lints() {
139             self.lints.push((*lint, from_plugin));
140
141             let id = LintId::of(*lint);
142             if self.by_name.insert(lint.name_lower(), Id(id)).is_some() {
143                 let msg = format!("duplicate specification of lint {}", lint.name_lower());
144                 match (sess, from_plugin) {
145                     // We load builtin lints first, so a duplicate is a compiler bug.
146                     // Use early_error when handling -W help with no crate.
147                     (None, _) => early_error(diagnostic::Auto, &msg[..]),
148                     (Some(sess), false) => sess.bug(&msg[..]),
149
150                     // A duplicate name from a plugin is a user error.
151                     (Some(sess), true)  => sess.err(&msg[..]),
152                 }
153             }
154
155             if lint.default_level != Allow {
156                 self.levels.insert(id, (lint.default_level, Default));
157             }
158         }
159         self.passes.as_mut().unwrap().push(pass);
160     }
161
162     pub fn register_group(&mut self, sess: Option<&Session>,
163                           from_plugin: bool, name: &'static str,
164                           to: Vec<LintId>) {
165         let new = self.lint_groups.insert(name, (to, from_plugin)).is_none();
166
167         if !new {
168             let msg = format!("duplicate specification of lint group {}", name);
169             match (sess, from_plugin) {
170                 // We load builtin lints first, so a duplicate is a compiler bug.
171                 // Use early_error when handling -W help with no crate.
172                 (None, _) => early_error(diagnostic::Auto, &msg[..]),
173                 (Some(sess), false) => sess.bug(&msg[..]),
174
175                 // A duplicate name from a plugin is a user error.
176                 (Some(sess), true)  => sess.err(&msg[..]),
177             }
178         }
179     }
180
181     pub fn register_renamed(&mut self, old_name: &str, new_name: &str) {
182         let target = match self.by_name.get(new_name) {
183             Some(&Id(lint_id)) => lint_id.clone(),
184             _ => panic!("invalid lint renaming of {} to {}", old_name, new_name)
185         };
186         self.by_name.insert(old_name.to_string(), Renamed(new_name.to_string(), target));
187     }
188
189     pub fn register_removed(&mut self, name: &str, reason: &str) {
190         self.by_name.insert(name.into(), Removed(reason.into()));
191     }
192
193     #[allow(unused_variables)]
194     fn find_lint(&self, lint_name: &str, sess: &Session, span: Option<Span>)
195                  -> Result<LintId, FindLintError>
196     {
197         match self.by_name.get(lint_name) {
198             Some(&Id(lint_id)) => Ok(lint_id),
199             Some(&Renamed(ref new_name, lint_id)) => {
200                 let warning = format!("lint {} has been renamed to {}",
201                                       lint_name, new_name);
202                 match span {
203                     Some(span) => sess.span_warn(span, &warning[..]),
204                     None => sess.warn(&warning[..]),
205                 };
206                 Ok(lint_id)
207             },
208             Some(&Removed(ref reason)) => {
209                 let warning = format!("lint {} has been removed: {}", lint_name, reason);
210                 match span {
211                     Some(span) => sess.span_warn(span, &warning[..]),
212                     None => sess.warn(&warning[..])
213                 }
214                 Err(FindLintError::Removed)
215             },
216             None => Err(FindLintError::NotFound)
217         }
218     }
219
220     pub fn process_command_line(&mut self, sess: &Session) {
221         for &(ref lint_name, level) in &sess.opts.lint_opts {
222             match self.find_lint(&lint_name[..], sess, None) {
223                 Ok(lint_id) => self.set_level(lint_id, (level, CommandLine)),
224                 Err(_) => {
225                     match self.lint_groups.iter().map(|(&x, pair)| (x, pair.0.clone()))
226                                                  .collect::<FnvHashMap<&'static str,
227                                                                        Vec<LintId>>>()
228                                                  .get(&lint_name[..]) {
229                         Some(v) => {
230                             v.iter()
231                              .map(|lint_id: &LintId|
232                                      self.set_level(*lint_id, (level, CommandLine)))
233                              .collect::<Vec<()>>();
234                         }
235                         None => sess.err(&format!("unknown {} flag: {}",
236                                                  level.as_str(), lint_name)),
237                     }
238                 }
239             }
240         }
241
242         self.lint_cap = sess.opts.lint_cap;
243         if let Some(cap) = self.lint_cap {
244             for level in self.levels.iter_mut().map(|p| &mut (p.1).0) {
245                 *level = cmp::min(*level, cap);
246             }
247         }
248     }
249 }
250
251 /// Context for lint checking after type checking.
252 pub struct LateContext<'a, 'tcx: 'a> {
253     /// Type context we're checking in.
254     pub tcx: &'a ty::ctxt<'tcx>,
255
256     /// The crate being checked.
257     pub krate: &'a hir::Crate,
258
259     /// Items exported from the crate being checked.
260     pub exported_items: &'a ExportedItems,
261
262     /// The store of registered lints.
263     lints: LintStore,
264
265     /// When recursing into an attributed node of the ast which modifies lint
266     /// levels, this stack keeps track of the previous lint levels of whatever
267     /// was modified.
268     level_stack: Vec<(LintId, LevelSource)>,
269
270     /// Level of lints for certain NodeIds, stored here because the body of
271     /// the lint needs to run in trans.
272     node_levels: RefCell<FnvHashMap<(ast::NodeId, LintId), LevelSource>>,
273 }
274
275 pub type Context<'a, 'tcx: 'a> = LateContext<'a, 'tcx>;
276
277 /// Context for lint checking of the AST, after expansion, before lowering to
278 /// HIR.
279 pub struct EarlyContext<'a> {
280     /// Type context we're checking in.
281     pub sess: &'a Session,
282
283     /// The crate being checked.
284     pub krate: &'a ast::Crate,
285
286     /// The store of registered lints.
287     lints: LintStore,
288
289     /// When recursing into an attributed node of the ast which modifies lint
290     /// levels, this stack keeps track of the previous lint levels of whatever
291     /// was modified.
292     level_stack: Vec<(LintId, LevelSource)>,
293 }
294
295 /// Convenience macro for calling a `LintPass` method on every pass in the context.
296 macro_rules! run_lints { ($cx:expr, $f:ident, $($args:expr),*) => ({
297     // Move the vector of passes out of `$cx` so that we can
298     // iterate over it mutably while passing `$cx` to the methods.
299     let mut passes = $cx.mut_lints().passes.take().unwrap();
300     for obj in &mut passes {
301         obj.$f($cx, $($args),*);
302     }
303     $cx.mut_lints().passes = Some(passes);
304 }) }
305
306 /// Parse the lint attributes into a vector, with `Err`s for malformed lint
307 /// attributes. Writing this as an iterator is an enormous mess.
308 // See also the hir version just below.
309 pub fn gather_attrs(attrs: &[ast::Attribute])
310                     -> Vec<Result<(InternedString, Level, Span), Span>> {
311     let mut out = vec!();
312     for attr in attrs {
313         let level = match Level::from_str(&attr.name()) {
314             None => continue,
315             Some(lvl) => lvl,
316         };
317
318         attr::mark_used(attr);
319
320         let meta = &attr.node.value;
321         let metas = match meta.node {
322             ast::MetaList(_, ref metas) => metas,
323             _ => {
324                 out.push(Err(meta.span));
325                 continue;
326             }
327         };
328
329         for meta in metas {
330             out.push(match meta.node {
331                 ast::MetaWord(ref lint_name) => Ok((lint_name.clone(), level, meta.span)),
332                 _ => Err(meta.span),
333             });
334         }
335     }
336     out
337 }
338
339 /// Emit a lint as a warning or an error (or not at all)
340 /// according to `level`.
341 ///
342 /// This lives outside of `Context` so it can be used by checks
343 /// in trans that run after the main lint pass is finished. Most
344 /// lints elsewhere in the compiler should call
345 /// `Session::add_lint()` instead.
346 pub fn raw_emit_lint(sess: &Session, lint: &'static Lint,
347                      lvlsrc: LevelSource, span: Option<Span>, msg: &str) {
348     let (mut level, source) = lvlsrc;
349     if level == Allow { return }
350
351     let name = lint.name_lower();
352     let mut def = None;
353     let msg = match source {
354         Default => {
355             format!("{}, #[{}({})] on by default", msg,
356                     level.as_str(), name)
357         },
358         CommandLine => {
359             format!("{} [-{} {}]", msg,
360                     match level {
361                         Warn => 'W', Deny => 'D', Forbid => 'F',
362                         Allow => panic!()
363                     }, name.replace("_", "-"))
364         },
365         Node(src) => {
366             def = Some(src);
367             msg.to_string()
368         }
369     };
370
371     // For purposes of printing, we can treat forbid as deny.
372     if level == Forbid { level = Deny; }
373
374     match (level, span) {
375         (Warn, Some(sp)) => sess.span_warn(sp, &msg[..]),
376         (Warn, None)     => sess.warn(&msg[..]),
377         (Deny, Some(sp)) => sess.span_err(sp, &msg[..]),
378         (Deny, None)     => sess.err(&msg[..]),
379         _ => sess.bug("impossible level in raw_emit_lint"),
380     }
381
382     if let Some(span) = def {
383         sess.span_note(span, "lint level defined here");
384     }
385 }
386
387 pub trait LintContext: Sized {
388     fn sess(&self) -> &Session;
389     fn lints(&self) -> &LintStore;
390     fn mut_lints(&mut self) -> &mut LintStore;
391     fn level_stack(&mut self) -> &mut Vec<(LintId, LevelSource)>;
392     fn enter_attrs(&mut self, attrs: &[hir::Attribute]);
393     fn exit_attrs(&mut self, attrs: &[hir::Attribute]);
394
395     /// Get the level of `lint` at the current position of the lint
396     /// traversal.
397     fn current_level(&self, lint: &'static Lint) -> Level {
398         self.lints().levels.get(&LintId::of(lint)).map_or(Allow, |&(lvl, _)| lvl)
399     }
400
401     fn lookup_and_emit(&self, lint: &'static Lint, span: Option<Span>, msg: &str) {
402         let (level, src) = match self.lints().levels.get(&LintId::of(lint)) {
403             None => return,
404             Some(&(Warn, src)) => {
405                 let lint_id = LintId::of(builtin::WARNINGS);
406                 (self.lints().get_level_source(lint_id).0, src)
407             }
408             Some(&pair) => pair,
409         };
410
411         raw_emit_lint(&self.sess(), lint, (level, src), span, msg);
412     }
413
414     /// Emit a lint at the appropriate level, for a particular span.
415     fn span_lint(&self, lint: &'static Lint, span: Span, msg: &str) {
416         self.lookup_and_emit(lint, Some(span), msg);
417     }
418
419     /// Emit a lint at the appropriate level, with no associated span.
420     fn lint(&self, lint: &'static Lint, msg: &str) {
421         self.lookup_and_emit(lint, None, msg);
422     }
423
424     /// Merge the lints specified by any lint attributes into the
425     /// current lint context, call the provided function, then reset the
426     /// lints in effect to their previous state.
427     fn with_lint_attrs<F>(&mut self,
428                           attrs: &[ast::Attribute],
429                           f: F)
430         where F: FnOnce(&mut Self),
431     {
432         // Parse all of the lint attributes, and then add them all to the
433         // current dictionary of lint information. Along the way, keep a history
434         // of what we changed so we can roll everything back after invoking the
435         // specified closure
436         let mut pushed = 0;
437
438         for result in gather_attrs(attrs) {
439             let v = match result {
440                 Err(span) => {
441                     span_err!(self.sess(), span, E0452,
442                               "malformed lint attribute");
443                     continue;
444                 }
445                 Ok((lint_name, level, span)) => {
446                     match self.lints().find_lint(&lint_name, &self.sess(), Some(span)) {
447                         Ok(lint_id) => vec![(lint_id, level, span)],
448                         Err(FindLintError::NotFound) => {
449                             match self.lints().lint_groups.get(&lint_name[..]) {
450                                 Some(&(ref v, _)) => v.iter()
451                                                       .map(|lint_id: &LintId|
452                                                            (*lint_id, level, span))
453                                                       .collect(),
454                                 None => {
455                                     self.span_lint(builtin::UNKNOWN_LINTS, span,
456                                                    &format!("unknown `{}` attribute: `{}`",
457                                                             level.as_str(), lint_name));
458                                     continue;
459                                 }
460                             }
461                         },
462                         Err(FindLintError::Removed) => { continue; }
463                     }
464                 }
465             };
466
467             for (lint_id, level, span) in v {
468                 let now = self.lints().get_level_source(lint_id).0;
469                 if now == Forbid && level != Forbid {
470                     let lint_name = lint_id.as_str();
471                     span_err!(self.sess(), span, E0453,
472                               "{}({}) overruled by outer forbid({})",
473                               level.as_str(), lint_name,
474                               lint_name);
475                 } else if now != level {
476                     let src = self.lints().get_level_source(lint_id).1;
477                     self.level_stack().push((lint_id, (now, src)));
478                     pushed += 1;
479                     self.mut_lints().set_level(lint_id, (level, Node(span)));
480                 }
481             }
482         }
483
484         self.enter_attrs(attrs);
485         f(self);
486         self.exit_attrs(attrs);
487
488         // rollback
489         for _ in 0..pushed {
490             let (lint, lvlsrc) = self.level_stack().pop().unwrap();
491             self.mut_lints().set_level(lint, lvlsrc);
492         }
493     }
494
495     fn with_ast_lint_attrs<F>(&mut self,
496                           attrs: &[ast::Attribute],
497                           f: F)
498         where F: FnOnce(&mut Self),
499     {
500         self.with_lint_attrs(&lower_attrs(attrs), f)
501     }
502 }
503
504
505 impl<'a> EarlyContext<'a> {
506     fn new(sess: &'a Session,
507            krate: &'a ast::Crate) -> EarlyContext<'a> {
508         // We want to own the lint store, so move it out of the session. Remember
509         // to put it back later...
510         let lint_store = mem::replace(&mut *sess.lint_store.borrow_mut(),
511                                       LintStore::new());
512         EarlyContext {
513             sess: sess,
514             krate: krate,
515             lints: lint_store,
516             level_stack: vec![],
517         }
518     }
519
520     fn visit_ids<F>(&mut self, f: F)
521         where F: FnOnce(&mut ast_util::IdVisitor<EarlyContext>)
522     {
523         let mut v = ast_util::IdVisitor {
524             operation: self,
525             pass_through_items: false,
526             visited_outermost: false,
527         };
528         f(&mut v);
529     }
530 }
531
532 impl<'a, 'tcx> LateContext<'a, 'tcx> {
533     fn new(tcx: &'a ty::ctxt<'tcx>,
534            krate: &'a hir::Crate,
535            exported_items: &'a ExportedItems) -> LateContext<'a, 'tcx> {
536         // We want to own the lint store, so move it out of the session.
537         let lint_store = mem::replace(&mut *tcx.sess.lint_store.borrow_mut(),
538                                       LintStore::new());
539
540         LateContext {
541             tcx: tcx,
542             krate: krate,
543             exported_items: exported_items,
544             lints: lint_store,
545             level_stack: vec![],
546             node_levels: RefCell::new(FnvHashMap()),
547         }
548     }
549
550     fn visit_ids<F>(&mut self, f: F)
551         where F: FnOnce(&mut util::IdVisitor<LateContext>)
552     {
553         let mut v = util::IdVisitor {
554             operation: self,
555             pass_through_items: false,
556             visited_outermost: false,
557         };
558         f(&mut v);
559     }
560 }
561
562 impl<'a, 'tcx> LintContext for LateContext<'a, 'tcx> {
563     /// Get the overall compiler `Session` object.
564     fn sess(&self) -> &Session {
565         &self.tcx.sess
566     }
567
568     fn lints(&self) -> &LintStore {
569         &self.lints
570     }
571
572     fn mut_lints(&mut self) -> &mut LintStore {
573         &mut self.lints
574     }
575
576     fn level_stack(&mut self) -> &mut Vec<(LintId, LevelSource)> {
577         &mut self.level_stack
578     }
579
580     fn enter_attrs(&mut self, attrs: &[hir::Attribute]) {
581         run_lints!(self, enter_lint_attrs, attrs);
582     }
583
584     fn exit_attrs(&mut self, attrs: &[hir::Attribute]) {
585         run_lints!(self, exit_lint_attrs, attrs);
586     }
587 }
588
589 impl<'a> LintContext for EarlyContext<'a> {
590     /// Get the overall compiler `Session` object.
591     fn sess(&self) -> &Session {
592         &self.sess
593     }
594
595     fn lints(&self) -> &LintStore {
596         &self.lints
597     }
598
599     fn mut_lints(&mut self) -> &mut LintStore {
600         &mut self.lints
601     }
602
603     fn level_stack(&mut self) -> &mut Vec<(LintId, LevelSource)> {
604         &mut self.level_stack
605     }
606
607     fn enter_attrs(&mut self, attrs: &[hir::Attribute]) {
608         run_lints!(self, ast_enter_lint_attrs, attrs);
609     }
610
611     fn exit_attrs(&mut self, attrs: &[hir::Attribute]) {
612         run_lints!(self, ast_exit_lint_attrs, attrs);
613     }
614 }
615
616 impl<'a, 'tcx, 'v> hir_visit::Visitor<'v> for LateContext<'a, 'tcx> {
617     fn visit_item(&mut self, it: &hir::Item) {
618         self.with_lint_attrs(&it.attrs, |cx| {
619             run_lints!(cx, check_item, it);
620             cx.visit_ids(|v| v.visit_item(it));
621             hir_visit::walk_item(cx, it);
622         })
623     }
624
625     fn visit_foreign_item(&mut self, it: &hir::ForeignItem) {
626         self.with_lint_attrs(&it.attrs, |cx| {
627             run_lints!(cx, check_foreign_item, it);
628             hir_visit::walk_foreign_item(cx, it);
629         })
630     }
631
632     fn visit_pat(&mut self, p: &hir::Pat) {
633         run_lints!(self, check_pat, p);
634         hir_visit::walk_pat(self, p);
635     }
636
637     fn visit_expr(&mut self, e: &hir::Expr) {
638         run_lints!(self, check_expr, e);
639         hir_visit::walk_expr(self, e);
640     }
641
642     fn visit_stmt(&mut self, s: &hir::Stmt) {
643         run_lints!(self, check_stmt, s);
644         hir_visit::walk_stmt(self, s);
645     }
646
647     fn visit_fn(&mut self, fk: hir_visit::FnKind<'v>, decl: &'v hir::FnDecl,
648                 body: &'v hir::Block, span: Span, id: ast::NodeId) {
649         run_lints!(self, check_fn, fk, decl, body, span, id);
650         hir_visit::walk_fn(self, fk, decl, body, span);
651     }
652
653     fn visit_struct_def(&mut self,
654                         s: &hir::StructDef,
655                         ident: ast::Ident,
656                         g: &hir::Generics,
657                         id: ast::NodeId) {
658         run_lints!(self, check_struct_def, s, ident, g, id);
659         hir_visit::walk_struct_def(self, s);
660         run_lints!(self, check_struct_def_post, s, ident, g, id);
661     }
662
663     fn visit_struct_field(&mut self, s: &hir::StructField) {
664         self.with_lint_attrs(&s.node.attrs, |cx| {
665             run_lints!(cx, check_struct_field, s);
666             hir_visit::walk_struct_field(cx, s);
667         })
668     }
669
670     fn visit_variant(&mut self, v: &hir::Variant, g: &hir::Generics) {
671         self.with_lint_attrs(&v.node.attrs, |cx| {
672             run_lints!(cx, check_variant, v, g);
673             hir_visit::walk_variant(cx, v, g);
674             run_lints!(cx, check_variant_post, v, g);
675         })
676     }
677
678     fn visit_ty(&mut self, t: &hir::Ty) {
679         run_lints!(self, check_ty, t);
680         hir_visit::walk_ty(self, t);
681     }
682
683     fn visit_ident(&mut self, sp: Span, id: ast::Ident) {
684         run_lints!(self, check_ident, sp, id);
685     }
686
687     fn visit_mod(&mut self, m: &hir::Mod, s: Span, n: ast::NodeId) {
688         run_lints!(self, check_mod, m, s, n);
689         hir_visit::walk_mod(self, m);
690     }
691
692     fn visit_local(&mut self, l: &hir::Local) {
693         run_lints!(self, check_local, l);
694         hir_visit::walk_local(self, l);
695     }
696
697     fn visit_block(&mut self, b: &hir::Block) {
698         run_lints!(self, check_block, b);
699         hir_visit::walk_block(self, b);
700     }
701
702     fn visit_arm(&mut self, a: &hir::Arm) {
703         run_lints!(self, check_arm, a);
704         hir_visit::walk_arm(self, a);
705     }
706
707     fn visit_decl(&mut self, d: &hir::Decl) {
708         run_lints!(self, check_decl, d);
709         hir_visit::walk_decl(self, d);
710     }
711
712     fn visit_expr_post(&mut self, e: &hir::Expr) {
713         run_lints!(self, check_expr_post, e);
714     }
715
716     fn visit_generics(&mut self, g: &hir::Generics) {
717         run_lints!(self, check_generics, g);
718         hir_visit::walk_generics(self, g);
719     }
720
721     fn visit_trait_item(&mut self, trait_item: &hir::TraitItem) {
722         self.with_lint_attrs(&trait_item.attrs, |cx| {
723             run_lints!(cx, check_trait_item, trait_item);
724             cx.visit_ids(|v| v.visit_trait_item(trait_item));
725             hir_visit::walk_trait_item(cx, trait_item);
726         });
727     }
728
729     fn visit_impl_item(&mut self, impl_item: &hir::ImplItem) {
730         self.with_lint_attrs(&impl_item.attrs, |cx| {
731             run_lints!(cx, check_impl_item, impl_item);
732             cx.visit_ids(|v| v.visit_impl_item(impl_item));
733             hir_visit::walk_impl_item(cx, impl_item);
734         });
735     }
736
737     fn visit_opt_lifetime_ref(&mut self, sp: Span, lt: &Option<hir::Lifetime>) {
738         run_lints!(self, check_opt_lifetime_ref, sp, lt);
739     }
740
741     fn visit_lifetime_ref(&mut self, lt: &hir::Lifetime) {
742         run_lints!(self, check_lifetime_ref, lt);
743     }
744
745     fn visit_lifetime_def(&mut self, lt: &hir::LifetimeDef) {
746         run_lints!(self, check_lifetime_def, lt);
747     }
748
749     fn visit_explicit_self(&mut self, es: &hir::ExplicitSelf) {
750         run_lints!(self, check_explicit_self, es);
751         hir_visit::walk_explicit_self(self, es);
752     }
753
754     fn visit_path(&mut self, p: &hir::Path, id: ast::NodeId) {
755         run_lints!(self, check_path, p, id);
756         hir_visit::walk_path(self, p);
757     }
758
759     fn visit_attribute(&mut self, attr: &ast::Attribute) {
760         run_lints!(self, check_attribute, attr);
761     }
762 }
763
764 impl<'a, 'v> ast_visit::Visitor<'v> for EarlyContext<'a> {
765     fn visit_item(&mut self, it: &ast::Item) {
766         self.with_ast_lint_attrs(&it.attrs, |cx| {
767             run_lints!(cx, check_ast_item, it);
768             cx.visit_ids(|v| v.visit_item(it));
769             ast_visit::walk_item(cx, it);
770         })
771     }
772
773     fn visit_foreign_item(&mut self, it: &ast::ForeignItem) {
774         self.with_ast_lint_attrs(&it.attrs, |cx| {
775             run_lints!(cx, check_ast_foreign_item, it);
776             ast_visit::walk_foreign_item(cx, it);
777         })
778     }
779
780     fn visit_pat(&mut self, p: &ast::Pat) {
781         run_lints!(self, check_ast_pat, p);
782         ast_visit::walk_pat(self, p);
783     }
784
785     fn visit_expr(&mut self, e: &ast::Expr) {
786         run_lints!(self, check_ast_expr, e);
787         ast_visit::walk_expr(self, e);
788     }
789
790     fn visit_stmt(&mut self, s: &ast::Stmt) {
791         run_lints!(self, check_ast_stmt, s);
792         ast_visit::walk_stmt(self, s);
793     }
794
795     fn visit_fn(&mut self, fk: ast_visit::FnKind<'v>, decl: &'v ast::FnDecl,
796                 body: &'v ast::Block, span: Span, id: ast::NodeId) {
797         run_lints!(self, check_ast_fn, fk, decl, body, span, id);
798         ast_visit::walk_fn(self, fk, decl, body, span);
799     }
800
801     fn visit_struct_def(&mut self,
802                         s: &ast::StructDef,
803                         ident: ast::Ident,
804                         g: &ast::Generics,
805                         id: ast::NodeId) {
806         run_lints!(self, check_ast_struct_def, s, ident, g, id);
807         ast_visit::walk_struct_def(self, s);
808         run_lints!(self, check_ast_struct_def_post, s, ident, g, id);
809     }
810
811     fn visit_struct_field(&mut self, s: &ast::StructField) {
812         self.with_ast_lint_attrs(&s.node.attrs, |cx| {
813             run_lints!(cx, check_ast_struct_field, s);
814             ast_visit::walk_struct_field(cx, s);
815         })
816     }
817
818     fn visit_variant(&mut self, v: &ast::Variant, g: &ast::Generics) {
819         self.with_ast_lint_attrs(&v.node.attrs, |cx| {
820             run_lints!(cx, check_ast_variant, v, g);
821             ast_visit::walk_variant(cx, v, g);
822             run_lints!(cx, check_ast_variant_post, v, g);
823         })
824     }
825
826     fn visit_ty(&mut self, t: &ast::Ty) {
827         run_lints!(self, check_ast_ty, t);
828         ast_visit::walk_ty(self, t);
829     }
830
831     fn visit_ident(&mut self, sp: Span, id: ast::Ident) {
832         run_lints!(self, check_ast_ident, sp, id);
833     }
834
835     fn visit_mod(&mut self, m: &ast::Mod, s: Span, n: ast::NodeId) {
836         run_lints!(self, check_ast_mod, m, s, n);
837         ast_visit::walk_mod(self, m);
838     }
839
840     fn visit_local(&mut self, l: &ast::Local) {
841         run_lints!(self, check_ast_local, l);
842         ast_visit::walk_local(self, l);
843     }
844
845     fn visit_block(&mut self, b: &ast::Block) {
846         run_lints!(self, check_ast_block, b);
847         ast_visit::walk_block(self, b);
848     }
849
850     fn visit_arm(&mut self, a: &ast::Arm) {
851         run_lints!(self, check_ast_arm, a);
852         ast_visit::walk_arm(self, a);
853     }
854
855     fn visit_decl(&mut self, d: &ast::Decl) {
856         run_lints!(self, check_ast_decl, d);
857         ast_visit::walk_decl(self, d);
858     }
859
860     fn visit_expr_post(&mut self, e: &ast::Expr) {
861         run_lints!(self, check_ast_expr_post, e);
862     }
863
864     fn visit_generics(&mut self, g: &ast::Generics) {
865         run_lints!(self, check_ast_generics, g);
866         ast_visit::walk_generics(self, g);
867     }
868
869     fn visit_trait_item(&mut self, trait_item: &ast::TraitItem) {
870         self.with_ast_lint_attrs(&trait_item.attrs, |cx| {
871             run_lints!(cx, check_ast_trait_item, trait_item);
872             cx.visit_ids(|v| v.visit_trait_item(trait_item));
873             ast_visit::walk_trait_item(cx, trait_item);
874         });
875     }
876
877     fn visit_impl_item(&mut self, impl_item: &ast::ImplItem) {
878         self.with_ast_lint_attrs(&impl_item.attrs, |cx| {
879             run_lints!(cx, check_ast_impl_item, impl_item);
880             cx.visit_ids(|v| v.visit_impl_item(impl_item));
881             ast_visit::walk_impl_item(cx, impl_item);
882         });
883     }
884
885     fn visit_opt_lifetime_ref(&mut self, sp: Span, lt: &Option<ast::Lifetime>) {
886         run_lints!(self, check_ast_opt_lifetime_ref, sp, lt);
887     }
888
889     fn visit_lifetime_ref(&mut self, lt: &ast::Lifetime) {
890         run_lints!(self, check_ast_lifetime_ref, lt);
891     }
892
893     fn visit_lifetime_def(&mut self, lt: &ast::LifetimeDef) {
894         run_lints!(self, check_ast_lifetime_def, lt);
895     }
896
897     fn visit_explicit_self(&mut self, es: &ast::ExplicitSelf) {
898         run_lints!(self, check_ast_explicit_self, es);
899         ast_visit::walk_explicit_self(self, es);
900     }
901
902     fn visit_path(&mut self, p: &ast::Path, id: ast::NodeId) {
903         run_lints!(self, check_ast_path, p, id);
904         ast_visit::walk_path(self, p);
905     }
906
907     fn visit_attribute(&mut self, attr: &ast::Attribute) {
908         run_lints!(self, check_ast_attribute, attr);
909     }
910 }
911
912 // Output any lints that were previously added to the session.
913 impl<'a, 'tcx> IdVisitingOperation for LateContext<'a, 'tcx> {
914     fn visit_id(&mut self, id: ast::NodeId) {
915         match self.sess().lints.borrow_mut().remove(&id) {
916             None => {}
917             Some(lints) => {
918                 for (lint_id, span, msg) in lints {
919                     self.span_lint(lint_id.lint, span, &msg[..])
920                 }
921             }
922         }
923     }
924 }
925 impl<'a> IdVisitingOperation for EarlyContext<'a> {
926     fn visit_id(&mut self, id: ast::NodeId) {
927         match self.sess.lints.borrow_mut().remove(&id) {
928             None => {}
929             Some(lints) => {
930                 for (lint_id, span, msg) in lints {
931                     self.span_lint(lint_id.lint, span, &msg[..])
932                 }
933             }
934         }
935     }
936 }
937
938 // This lint pass is defined here because it touches parts of the `LateContext`
939 // that we don't want to expose. It records the lint level at certain AST
940 // nodes, so that the variant size difference check in trans can call
941 // `raw_emit_lint`.
942
943 pub struct GatherNodeLevels;
944
945 impl LintPass for GatherNodeLevels {
946     fn get_lints(&self) -> LintArray {
947         lint_array!()
948     }
949
950     fn check_item(&mut self, cx: &LateContext, it: &hir::Item) {
951         match it.node {
952             hir::ItemEnum(..) => {
953                 let lint_id = LintId::of(builtin::VARIANT_SIZE_DIFFERENCES);
954                 let lvlsrc = cx.lints.get_level_source(lint_id);
955                 match lvlsrc {
956                     (lvl, _) if lvl != Allow => {
957                         cx.node_levels.borrow_mut()
958                             .insert((it.id, lint_id), lvlsrc);
959                     },
960                     _ => { }
961                 }
962             },
963             _ => { }
964         }
965     }
966 }
967
968 /// Perform lint checking on a crate.
969 ///
970 /// Consumes the `lint_store` field of the `Session`.
971 pub fn check_crate(tcx: &ty::ctxt,
972                    krate: &hir::Crate,
973                    exported_items: &ExportedItems) {
974
975     let mut cx = LateContext::new(tcx, krate, exported_items);
976
977     // Visit the whole crate.
978     cx.with_lint_attrs(&krate.attrs, |cx| {
979         cx.visit_id(ast::CRATE_NODE_ID);
980         cx.visit_ids(|v| {
981             v.visited_outermost = true;
982             hir_visit::walk_crate(v, krate);
983         });
984
985         // since the root module isn't visited as an item (because it isn't an
986         // item), warn for it here.
987         run_lints!(cx, check_crate, krate);
988
989         hir_visit::walk_crate(cx, krate);
990     });
991
992     // If we missed any lints added to the session, then there's a bug somewhere
993     // in the iteration code.
994     for (id, v) in tcx.sess.lints.borrow().iter() {
995         for &(lint, span, ref msg) in v {
996             tcx.sess.span_bug(span,
997                               &format!("unprocessed lint {} at {}: {}",
998                                        lint.as_str(), tcx.map.node_to_string(*id), *msg))
999         }
1000     }
1001
1002     *tcx.node_lint_levels.borrow_mut() = cx.node_levels.into_inner();
1003 }
1004
1005 pub fn check_ast_crate(sess: &Session, krate: &ast::Crate) {
1006     let mut cx = EarlyContext::new(sess, krate);
1007
1008     // Visit the whole crate.
1009     cx.with_ast_lint_attrs(&krate.attrs, |cx| {
1010         cx.visit_id(ast::CRATE_NODE_ID);
1011         cx.visit_ids(|v| {
1012             v.visited_outermost = true;
1013             ast_visit::walk_crate(v, krate);
1014         });
1015
1016         // since the root module isn't visited as an item (because it isn't an
1017         // item), warn for it here.
1018         run_lints!(cx, check_ast_crate, krate);
1019
1020         ast_visit::walk_crate(cx, krate);
1021     });
1022
1023     // Put the lint store back in the session.
1024     mem::replace(&mut *sess.lint_store.borrow_mut(), cx.lints);
1025
1026     // If we missed any lints added to the session, then there's a bug somewhere
1027     // in the iteration code.
1028     for (_, v) in sess.lints.borrow().iter() {
1029         for &(lint, span, ref msg) in v {
1030             sess.span_bug(span,
1031                           &format!("unprocessed lint {}: {}",
1032                                    lint.as_str(), *msg))
1033         }
1034     }
1035 }