|
6 | 6 | // just peeks and looks for that attribute.
|
7 | 7 |
|
8 | 8 | use crate::session::Session;
|
| 9 | +use core::num::IntErrorKind; |
| 10 | +use rustc::bug; |
9 | 11 | use rustc_span::symbol::{sym, Symbol};
|
10 | 12 | use syntax::ast;
|
11 | 13 |
|
12 | 14 | use rustc_data_structures::sync::Once;
|
13 | 15 |
|
14 | 16 | pub fn update_limits(sess: &Session, krate: &ast::Crate) {
|
15 |
| - update_limit(krate, &sess.recursion_limit, sym::recursion_limit, 128); |
16 |
| - update_limit(krate, &sess.type_length_limit, sym::type_length_limit, 1048576); |
| 17 | + update_limit(sess, krate, &sess.recursion_limit, sym::recursion_limit, 128); |
| 18 | + update_limit(sess, krate, &sess.type_length_limit, sym::type_length_limit, 1048576); |
17 | 19 | }
|
18 | 20 |
|
19 |
| -fn update_limit(krate: &ast::Crate, limit: &Once<usize>, name: Symbol, default: usize) { |
| 21 | +fn update_limit( |
| 22 | + sess: &Session, |
| 23 | + krate: &ast::Crate, |
| 24 | + limit: &Once<usize>, |
| 25 | + name: Symbol, |
| 26 | + default: usize, |
| 27 | +) { |
20 | 28 | for attr in &krate.attrs {
|
21 | 29 | if !attr.check_name(name) {
|
22 | 30 | continue;
|
23 | 31 | }
|
24 | 32 |
|
25 | 33 | if let Some(s) = attr.value_str() {
|
26 |
| - if let Some(n) = s.as_str().parse().ok() { |
27 |
| - limit.set(n); |
28 |
| - return; |
| 34 | + match s.as_str().parse() { |
| 35 | + Ok(n) => { |
| 36 | + limit.set(n); |
| 37 | + return; |
| 38 | + } |
| 39 | + Err(e) => { |
| 40 | + let mut err = sess.struct_span_err( |
| 41 | + attr.span, |
| 42 | + "`recursion_limit` must be a non-negative integer", |
| 43 | + ); |
| 44 | + |
| 45 | + let value_span = attr |
| 46 | + .meta() |
| 47 | + .and_then(|meta| meta.name_value_literal().cloned()) |
| 48 | + .map(|lit| lit.span) |
| 49 | + .unwrap_or(attr.span); |
| 50 | + |
| 51 | + let error_str = match e.kind() { |
| 52 | + IntErrorKind::Overflow => "`recursion_limit` is too large", |
| 53 | + IntErrorKind::Empty => "`recursion_limit` must be a non-negative integer", |
| 54 | + IntErrorKind::InvalidDigit => "not a valid integer", |
| 55 | + IntErrorKind::Underflow => bug!("`recursion_limit` should never underflow"), |
| 56 | + IntErrorKind::Zero => bug!("zero is a valid `recursion_limit`"), |
| 57 | + kind => bug!("unimplemented IntErrorKind variant: {:?}", kind), |
| 58 | + }; |
| 59 | + |
| 60 | + err.span_label(value_span, error_str); |
| 61 | + err.emit(); |
| 62 | + } |
29 | 63 | }
|
30 | 64 | }
|
31 | 65 | }
|
|
0 commit comments