From c9d0967383d67e2b51c222b17eb5502a8746aad0 Mon Sep 17 00:00:00 2001 From: Simonas Kazlauskas Date: Wed, 18 Feb 2015 20:30:17 +0200 Subject: Validate inline attribute arguments --- src/libsyntax/attr.rs | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'src/libsyntax') diff --git a/src/libsyntax/attr.rs b/src/libsyntax/attr.rs index 62e676891a0..4fc08c0c2b2 100644 --- a/src/libsyntax/attr.rs +++ b/src/libsyntax/attr.rs @@ -292,7 +292,7 @@ pub enum InlineAttr { } /// Determine what `#[inline]` attribute is present in `attrs`, if any. -pub fn find_inline_attr(attrs: &[Attribute]) -> InlineAttr { +pub fn find_inline_attr(diagnostic: Option<&SpanHandler>, attrs: &[Attribute]) -> InlineAttr { // FIXME (#2809)---validate the usage of #[inline] and #[inline] attrs.iter().fold(InlineNone, |ia,attr| { match attr.node.value.node { @@ -302,12 +302,16 @@ pub fn find_inline_attr(attrs: &[Attribute]) -> InlineAttr { } MetaList(ref n, ref items) if *n == "inline" => { mark_used(attr); - if contains_name(&items[..], "always") { + if items.len() != 1 { + diagnostic.map(|d|{ d.span_err(attr.span, "expected one argument"); }); + InlineNone + } else if contains_name(&items[..], "always") { InlineAlways } else if contains_name(&items[..], "never") { InlineNever } else { - InlineHint + diagnostic.map(|d|{ d.span_err((*items[0]).span, "invalid argument"); }); + InlineNone } } _ => ia @@ -317,7 +321,7 @@ pub fn find_inline_attr(attrs: &[Attribute]) -> InlineAttr { /// True if `#[inline]` or `#[inline(always)]` is present in `attrs`. pub fn requests_inline(attrs: &[Attribute]) -> bool { - match find_inline_attr(attrs) { + match find_inline_attr(None, attrs) { InlineHint | InlineAlways => true, InlineNone | InlineNever => false, } -- cgit 1.4.1-3-g733a5