gccrs: proc macros: Add crate type attribute check

A procedural macroa attribute can only be used on proc macro crates.

gcc/rust/ChangeLog:

	* util/rust-attributes.cc (AttributeChecker::visit): Add
	attribute check on functions.

Signed-off-by: Pierre-Emmanuel Patry <pierre-emmanuel.patry@embecosm.com>

diff --git a/gcc/rust/util/rust-attributes.cc b/gcc/rust/util/rust-attributes.cc
index 27817acf0c31d3fdcf41618d6a140df8cc485d97..a174edc625f50cd9e9baa72837c9392d6baa04ad 100644 (file)
--- a/gcc/rust/util/rust-attributes.cc
+++ b/gcc/rust/util/rust-attributes.cc
@@ -17,6 +17,7 @@
 // <http://www.gnu.org/licenses/>.
 
 #include "rust-system.h"
+#include "rust-session-manager.h"
 #include "rust-attributes.h"
 #include "rust-ast.h"
 #include "rust-ast-full.h"
@@ -44,6 +45,9 @@ static const BuiltinAttrDefinition __definitions[]
      {"path", EXPANSION},
      {"macro_use", NAME_RESOLUTION},
      {"macro_export", NAME_RESOLUTION},
+     {"proc_macro", EXPANSION},
+     {"proc_macro_derive", EXPANSION},
+     {"proc_macro_attribute", EXPANSION},
      // FIXME: This is not implemented yet, see
      // https://github.com/Rust-GCC/gccrs/issues/1475
      {"target_feature", CODE_GENERATION},
@@ -537,8 +541,35 @@ AttributeChecker::visit (AST::UseDeclaration &)
 {}
 
 void
-AttributeChecker::visit (AST::Function &)
-{}
+AttributeChecker::visit (AST::Function &fun)
+{
+  auto check_crate_type = [] (const char *name, AST::Attribute &attribute) {
+    if (!Session::get_instance ().options.is_proc_macro ())
+      rust_error_at (attribute.get_locus (),
+                    "the %<#[%s]%> attribute is only usable with crates of "
+                    "the %<proc-macro%> crate type",
+                    name);
+  };
+
+  BuiltinAttrDefinition result;
+  for (auto &attribute : fun.get_outer_attrs ())
+    {
+      if (!is_builtin (attribute, result))
+       return;
+
+      auto name = result.name.c_str ();
+
+      if (result.name == "proc_macro_derive")
+       {
+         check_crate_type (name, attribute);
+       }
+      else if (result.name == "proc_macro"
+              || result.name == "proc_macro_attribute")
+       {
+         check_crate_type (name, attribute);
+       }
+    }
+}
 
 void
 AttributeChecker::visit (AST::TypeAlias &)