driver: Improve check for rustc arg

The rustc arg might not be exactly "rustc". It may be any path to a rustc
executable (especially if the RUSTC environment variable is set when
executing cargo). Rather check that it is a path with 'rustc' file stem.

1 files changed, 2 insertions, 1 deletions

diff --git a/src/driver.rs b/src/driver.rs
index 26ff846177b..267e460ad2e 100644
--- a/src/driver.rs
+++ b/src/driver.rs
@@ -6,6 +6,7 @@
 
 use rustc_driver::{self, driver::CompileController, Compilation};
 use rustc_plugin;
+use std::path::Path;
 use std::process::{exit, Command};
 
 #[allow(clippy::print_stdout)]
@@ -47,7 +48,7 @@ pub fn main() {
         if orig_args.len() <= 1 {
             std::process::exit(1);
         }
-        if orig_args[1] == "rustc" {
+        if Path::new(&orig_args[1]).file_stem() == Some("rustc".as_ref()) {
             // we still want to be able to invoke it normally though
             orig_args.remove(1);
         }