Merge pull request #2716 from hi-rustin/rustin-patch-arg0

report error when rustup is invoked with unsupported arg0

Author: rbtcollins

src/bin/rustup-init.rs

@@ -28,6 +28,7 @@ use rustup::cli::setup_mode;
 use rustup::currentprocess::{process, with, OSProcess};
 use rustup::env_var::RUST_RECURSION_COUNT_MAX;
 use rustup::utils::utils;
+use rustup::TOOLS;
 
 fn main() {
     let process = OSProcess::default();
@@ -92,7 +93,13 @@ fn run_rustup_inner() -> Result<utils::ExitCode> {
                 }
             }
         }
-        Some(_) => proxy_mode::main(),
+        Some(n) => {
+            if TOOLS.iter().find(|&&name| name == n).is_some() {
+                proxy_mode::main()
+            } else {
+                Err(ErrorKind::UnknownProxyName(n.to_string()).into())
+            }
+        }
         None => {
             // Weird case. No arg0, or it's unparsable.
             Err(ErrorKind::NoExeName.into())

src/cli/errors.rs

@@ -13,6 +13,7 @@ use strsim::damerau_levenshtein;
 
 use super::rustup_mode::CompletionCommand;
 use crate::dist::temp;
+use crate::TOOLS;
 
 error_chain! {
     links {
@@ -47,6 +48,14 @@ error_chain! {
         NoExeName {
             description("couldn't determine self executable name")
         }
+        UnknownProxyName(n: String) {
+            description("unknown proxy name")
+            display(
+                "unknown proxy name: '{}'; valid proxy names are {}",
+                n,
+                valid_proxy_names(),
+            )
+        }
         NotSelfInstalled(p: PathBuf) {
             description("rustup is not installed")
             display("rustup is not installed at '{}'", p.display())
@@ -74,6 +83,14 @@ error_chain! {
     }
 }
 
+fn valid_proxy_names() -> String {
+    TOOLS
+        .iter()
+        .map(|s| format!("'{}'", s))
+        .collect::<Vec<_>>()
+        .join(", ")
+}
+
 fn maybe_suggest_toolchain(bad_name: &str) -> String {
     let bad_name = &bad_name.to_ascii_lowercase();
     static VALID_CHANNELS: &[&str] = &["stable", "beta", "nightly"];

tests/cli-misc.rs

@@ -325,10 +325,7 @@ fn rustup_failed_path_search() {
         expect_err(
             config,
             broken,
-            &format!(
-                "'fake_proxy{}' is not installed for the toolchain 'custom'",
-                EXE_SUFFIX
-            ),
+            &format!("unknown proxy name: 'fake_proxy'; valid proxy names are 'rustc', 'rustdoc', 'cargo', 'rust-lldb', 'rust-gdb', 'rls', 'cargo-clippy', 'clippy-driver', 'cargo-miri'",),
         );
 
         // Hardlink will be automatically cleaned up by test setup code