Warn if include macro fails to include entire file

Mark-Simulacrum · Mark-Simulacrum · commit e068cec13e2a · 2019-10-07T19:47:53.000-04:00
diff --git a/src/librustc/lint/builtin.rs b/src/librustc/lint/builtin.rs
@@ -368,6 +368,12 @@ pub mod parser {
         Allow,
         "possible meta-variable misuse at macro definition"
     }
+
+    declare_lint! {
+        pub INCOMPLETE_INCLUDE,
+        Deny,
+        "trailing content in included file"
+    }
 }
 
 declare_lint! {
diff --git a/src/librustc/lint/mod.rs b/src/librustc/lint/mod.rs
@@ -28,6 +28,7 @@ use crate::hir::intravisit;
 use crate::hir;
 use crate::lint::builtin::BuiltinLintDiagnostics;
 use crate::lint::builtin::parser::{ILL_FORMED_ATTRIBUTE_INPUT, META_VARIABLE_MISUSE};
+use crate::lint::builtin::parser::INCOMPLETE_INCLUDE;
 use crate::session::{Session, DiagnosticMessageId};
 use crate::ty::TyCtxt;
 use crate::ty::query::Providers;
@@ -83,6 +84,7 @@ impl Lint {
         match lint_id {
             BufferedEarlyLintId::IllFormedAttributeInput => ILL_FORMED_ATTRIBUTE_INPUT,
             BufferedEarlyLintId::MetaVariableMisuse => META_VARIABLE_MISUSE,
+            BufferedEarlyLintId::IncompleteInclude => INCOMPLETE_INCLUDE,
         }
     }
 
diff --git a/src/libsyntax/early_buffered_lints.rs b/src/libsyntax/early_buffered_lints.rs
@@ -11,6 +11,7 @@ use syntax_pos::MultiSpan;
 pub enum BufferedEarlyLintId {
     IllFormedAttributeInput,
     MetaVariableMisuse,
+    IncompleteInclude,
 }
 
 /// Stores buffered lint info which can later be passed to `librustc`.
diff --git a/src/libsyntax_ext/source_util.rs b/src/libsyntax_ext/source_util.rs
@@ -5,6 +5,7 @@ use syntax::print::pprust;
 use syntax::ptr::P;
 use syntax::symbol::Symbol;
 use syntax::tokenstream::TokenStream;
+use syntax::early_buffered_lints::BufferedEarlyLintId;
 
 use smallvec::SmallVec;
 use syntax_pos::{self, Pos, Span};
@@ -83,7 +84,16 @@ pub fn expand_include<'cx>(cx: &'cx mut ExtCtxt<'_>, sp: Span, tts: TokenStream)
     }
     impl<'a> base::MacResult for ExpandResult<'a> {
         fn make_expr(mut self: Box<ExpandResult<'a>>) -> Option<P<ast::Expr>> {
-            Some(panictry!(self.p.parse_expr()))
+            let r = panictry!(self.p.parse_expr());
+            if self.p.token != token::Eof {
+                self.p.sess.buffer_lint(
+                    BufferedEarlyLintId::IncompleteInclude,
+                    self.p.token.span,
+                    ast::CRATE_NODE_ID,
+                    "include macro expected single expression in source",
+                );
+            }
+            Some(r)
         }
 
         fn make_items(mut self: Box<ExpandResult<'a>>) -> Option<SmallVec<[P<ast::Item>; 1]>> {
diff --git a/src/test/ui/include-single-expr-helper-1.rs b/src/test/ui/include-single-expr-helper-1.rs
@@ -0,0 +1,5 @@
+// ignore-test auxiliary file for include-single-expr.rs
+
+0
+
+// trailing comment permitted
diff --git a/src/test/ui/include-single-expr-helper.rs b/src/test/ui/include-single-expr-helper.rs
@@ -0,0 +1,5 @@
+// ignore-test auxiliary file for include-single-expr.rs
+
+0
+10
+100
diff --git a/src/test/ui/include-single-expr.rs b/src/test/ui/include-single-expr.rs
@@ -0,0 +1,6 @@
+// error-pattern include macro expected single expression
+
+fn main() {
+    include!("include-single-expr-helper.rs");
+    include!("include-single-expr-helper-1.rs");
+}
diff --git a/src/test/ui/include-single-expr.stderr b/src/test/ui/include-single-expr.stderr
@@ -0,0 +1,10 @@
+error: include macro expected single expression in source
+  --> $DIR/include-single-expr-helper.rs:4:1
+   |
+LL | 10
+   | ^^
+   |
+   = note: `#[deny(incomplete_include)]` on by default
+
+error: aborting due to previous error
+
diff --git a/src/tools/error_index_generator/build.rs b/src/tools/error_index_generator/build.rs
@@ -15,7 +15,7 @@ fn main() {
             println!("cargo:rerun-if-changed={}", entry.path().to_str().unwrap());
             let file = fs::read_to_string(entry.path()).unwrap()
                 .replace("syntax::register_diagnostics!", "register_diagnostics!");
-            let contents = format!("(|| {{\n{}\n}})();", file);
+            let contents = format!("(|| {{\n{}\n}})()", file);
 
             fs::write(&out_dir.join(&format!("error_{}.rs", idx)), &contents).unwrap();
 

Original file line number	Diff line number	Diff line change
`@@ -368,6 +368,12 @@ pub mod parser {`
`368`	`368`	`Allow,`
`369`	`369`	`"possible meta-variable misuse at macro definition"`
`370`	`370`	`}`
	`371`	`+`
	`372`	`+ declare_lint! {`
	`373`	`+ pub INCOMPLETE_INCLUDE,`
	`374`	`+ Deny,`
	`375`	`+ "trailing content in included file"`
	`376`	`+ }`
`371`	`377`	`}`
`372`	`378`
`373`	`379`	`declare_lint! {`
Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,7 @@ use crate::hir::intravisit;`
`28`	`28`	`use crate::hir;`
`29`	`29`	`use crate::lint::builtin::BuiltinLintDiagnostics;`
`30`	`30`	`use crate::lint::builtin::parser::{ILL_FORMED_ATTRIBUTE_INPUT, META_VARIABLE_MISUSE};`
	`31`	`+use crate::lint::builtin::parser::INCOMPLETE_INCLUDE;`
`31`	`32`	`use crate::session::{Session, DiagnosticMessageId};`
`32`	`33`	`use crate::ty::TyCtxt;`
`33`	`34`	`use crate::ty::query::Providers;`
`@@ -83,6 +84,7 @@ impl Lint {`
`83`	`84`	`match lint_id {`
`84`	`85`	`BufferedEarlyLintId::IllFormedAttributeInput => ILL_FORMED_ATTRIBUTE_INPUT,`
`85`	`86`	`BufferedEarlyLintId::MetaVariableMisuse => META_VARIABLE_MISUSE,`
	`87`	`+ BufferedEarlyLintId::IncompleteInclude => INCOMPLETE_INCLUDE,`
`86`	`88`	`}`
`87`	`89`	`}`
`88`	`90`
Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,7 @@ use syntax_pos::MultiSpan;`
`11`	`11`	`pub enum BufferedEarlyLintId {`
`12`	`12`	`IllFormedAttributeInput,`
`13`	`13`	`MetaVariableMisuse,`
	`14`	`+ IncompleteInclude,`
`14`	`15`	`}`
`15`	`16`
`16`	`17`	/// Stores buffered lint info which can later be passed to `librustc`.
-Original file line number
+Diff line change
 +// ignore-test auxiliary file for include-single-expr.rs
++
 +0
 +10
 +100