@@ -840,6 +840,8 @@ public:
// Returns the kind of the module
enum ModuleKind get_kind () const { return kind; }
+ Unsafety get_unsafety () const { return safety; }
+
// TODO: think of better way to do this - mutable getter seems dodgy
const std::vector<Attribute> &get_inner_attrs () const { return inner_attrs; }
std::vector<Attribute> &get_inner_attrs () { return inner_attrs; }
@@ -17,6 +17,7 @@
// <http://www.gnu.org/licenses/>.
#include "rust-ast-validation.h"
+#include "rust-common.h"
#include "rust-diagnostics.h"
#include "rust-item.h"
#include "rust-keyword-values.h"
@@ -136,4 +137,13 @@ ASTValidation::visit (AST::Trait &trait)
AST::ContextualASTVisitor::visit (trait);
}
+void
+ASTValidation::visit (AST::Module &module)
+{
+ if (module.get_unsafety () == Unsafety::Unsafe)
+ rust_error_at (module.get_locus (), "module cannot be declared unsafe");
+
+ AST::ContextualASTVisitor::visit (module);
+}
+
} // namespace Rust
@@ -34,6 +34,7 @@ public:
void check (AST::Crate &crate) { AST::ContextualASTVisitor::visit (crate); }
+ virtual void visit (AST::Module &module);
virtual void visit (AST::ConstantItem &const_item);
virtual void visit (AST::Lifetime &lifetime);
virtual void visit (AST::LoopLabel &label);
@@ -2429,6 +2429,8 @@ std::unique_ptr<AST::Module>
Parser<ManagedTokenSource>::parse_module (AST::Visibility vis,
AST::AttrVec outer_attrs)
{
+ location_t locus = lexer.peek_token ()->get_locus ();
+
Unsafety safety = Unsafety::Normal;
if (lexer.peek_token ()->get_id () == UNSAFE)
{
@@ -2436,7 +2438,6 @@ Parser<ManagedTokenSource>::parse_module (AST::Visibility vis,
skip_token (UNSAFE);
}
- location_t locus = lexer.peek_token ()->get_locus ();
skip_token (MOD);
const_TokenPtr module_name = expect_token (IDENTIFIER);