1
0
mirror of https://github.com/Llewellynvdm/starship.git synced 2024-11-28 07:46:28 +00:00

fix(windows): avoid inadvertly running exes from cwd (#2885)

On Windows when running commands with their name instead of the path with Command::new, executable with that name from the current working directory will be executed.

This PR replaces all instances of Command::new with a new create_command function which will first resolve any executable paths and avoid this issue.
This commit is contained in:
David Knaack 2021-07-16 21:20:59 +02:00 committed by GitHub
parent e1fc137dc9
commit 1eaf996a36
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
19 changed files with 126 additions and 93 deletions

View File

@ -68,6 +68,8 @@ pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
} }
``` ```
If using `context.exec_cmd` isn't possible, please use `crate::utils::create_command` instead of `std::process::Command::new`.
## Logging ## Logging
Debug logging in starship is done with our custom logger implementation. Debug logging in starship is done with our custom logger implementation.

2
clippy.toml Normal file
View File

@ -0,0 +1,2 @@
# std::process::Command::new may inadvertly run executables from the current working directory
disallowed-methods = ["std::process::Command::new"]

View File

@ -265,18 +265,6 @@ mod tests {
assert!(link.contains("No+Starship+config")); assert!(link.contains("No+Starship+config"));
} }
#[test]
fn test_get_shell_info() {
env::remove_var("STARSHIP_SHELL");
let unknown_shell = get_shell_info();
assert_eq!(UNKNOWN_SHELL, &unknown_shell.name);
env::set_var("STARSHIP_SHELL", "fish");
let fish_shell = get_shell_info();
assert_eq!("fish", &fish_shell.name);
}
#[test] #[test]
#[cfg(not(windows))] #[cfg(not(windows))]
fn test_get_config_path() { fn test_get_config_path() {

View File

@ -2,7 +2,6 @@ use std::env;
use std::ffi::OsString; use std::ffi::OsString;
use std::io::ErrorKind; use std::io::ErrorKind;
use std::process; use std::process;
use std::process::Command;
use crate::config::RootModuleConfig; use crate::config::RootModuleConfig;
use crate::config::StarshipConfig; use crate::config::StarshipConfig;
@ -136,9 +135,9 @@ pub fn write_configuration(table: &mut Table) {
pub fn edit_configuration() { pub fn edit_configuration() {
let config_path = get_config_path(); let config_path = get_config_path();
let editor_cmd = shell_words::split(&get_editor()).expect("Unmatched quotes found in $EDITOR."); let editor_cmd = shell_words::split(&get_editor()).expect("Unmatched quotes found in $EDITOR.");
let editor_path = which::which(&editor_cmd[0]).expect("Unable to locate editor in $PATH.");
let command = Command::new(editor_path) let command = utils::create_command(&editor_cmd[0])
.expect("Unable to locate editor in $PATH.")
.args(&editor_cmd[1..]) .args(&editor_cmd[1..])
.arg(config_path) .arg(config_path)
.status(); .status();

View File

@ -1,3 +1,4 @@
use crate::utils::create_command;
use std::ffi::OsStr; use std::ffi::OsStr;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use std::{env, io}; use std::{env, io};
@ -51,9 +52,7 @@ impl StarshipPath {
return self.sprint(); return self.sprint();
} }
let str_path = self.str_path()?; let str_path = self.str_path()?;
let res = std::process::Command::new("cygpath.exe") let res = create_command("cygpath").and_then(|mut cmd| cmd.arg(str_path).output());
.arg(str_path)
.output();
let output = match res { let output = match res {
Ok(output) => output, Ok(output) => output,
Err(e) => { Err(e) => {

View File

@ -1,3 +1,5 @@
#![warn(clippy::disallowed_method)]
#[macro_use] #[macro_use]
extern crate shadow_rs; extern crate shadow_rs;

View File

@ -1,3 +1,5 @@
#![warn(clippy::disallowed_method)]
use clap::crate_authors; use clap::crate_authors;
use std::io; use std::io;
use std::time::SystemTime; use std::time::SystemTime;

View File

@ -5,7 +5,7 @@ use std::time::Instant;
use super::{Context, Module, RootModuleConfig}; use super::{Context, Module, RootModuleConfig};
use crate::{configs::custom::CustomConfig, formatter::StringFormatter}; use crate::{configs::custom::CustomConfig, formatter::StringFormatter, utils::create_command};
/// Creates a custom module with some configuration /// Creates a custom module with some configuration
/// ///
@ -100,7 +100,7 @@ fn get_shell<'a, 'b>(shell_args: &'b [&'a str]) -> (std::borrow::Cow<'a, str>, &
#[cfg(not(windows))] #[cfg(not(windows))]
fn shell_command(cmd: &str, shell_args: &[&str]) -> Option<Output> { fn shell_command(cmd: &str, shell_args: &[&str]) -> Option<Output> {
let (shell, shell_args) = get_shell(shell_args); let (shell, shell_args) = get_shell(shell_args);
let mut command = Command::new(shell.as_ref()); let mut command = create_command(shell.as_ref()).ok()?;
command command
.args(shell_args) .args(shell_args)
@ -118,6 +118,7 @@ fn shell_command(cmd: &str, shell_args: &[&str]) -> Option<Output> {
"Could not launch command with given shell or STARSHIP_SHELL env variable, retrying with /usr/bin/env sh" "Could not launch command with given shell or STARSHIP_SHELL env variable, retrying with /usr/bin/env sh"
); );
#[allow(clippy::disallowed_method)]
Command::new("/usr/bin/env") Command::new("/usr/bin/env")
.arg("sh") .arg("sh")
.stdin(Stdio::piped()) .stdin(Stdio::piped())
@ -141,14 +142,17 @@ fn shell_command(cmd: &str, shell_args: &[&str]) -> Option<Output> {
Some(std::borrow::Cow::Borrowed(shell_args[0])), Some(std::borrow::Cow::Borrowed(shell_args[0])),
&shell_args[1..], &shell_args[1..],
) )
} else if let Ok(env_shell) = std::env::var("STARSHIP_SHELL") { } else if let Some(env_shell) = std::env::var("STARSHIP_SHELL")
.ok()
.filter(|s| !cfg!(test) && !s.is_empty())
{
(Some(std::borrow::Cow::Owned(env_shell)), &[] as &[&str]) (Some(std::borrow::Cow::Owned(env_shell)), &[] as &[&str])
} else { } else {
(None, &[] as &[&str]) (None, &[] as &[&str])
}; };
if let Some(forced_shell) = shell { if let Some(forced_shell) = shell {
let mut command = Command::new(forced_shell.as_ref()); let mut command = create_command(forced_shell.as_ref()).ok()?;
command command
.args(shell_args) .args(shell_args)
@ -169,7 +173,8 @@ fn shell_command(cmd: &str, shell_args: &[&str]) -> Option<Output> {
); );
} }
let command = Command::new("cmd.exe") let command = create_command("cmd")
.ok()?
.arg("/C") .arg("/C")
.arg(cmd) .arg(cmd)
.stdin(Stdio::piped()) .stdin(Stdio::piped())

View File

@ -298,6 +298,7 @@ fn to_fish_style(pwd_dir_length: usize, dir_string: String, truncated_dir_string
mod tests { mod tests {
use super::*; use super::*;
use crate::test::ModuleRenderer; use crate::test::ModuleRenderer;
use crate::utils::create_command;
use crate::utils::home_dir; use crate::utils::home_dir;
use ansi_term::Color; use ansi_term::Color;
#[cfg(not(target_os = "windows"))] #[cfg(not(target_os = "windows"))]
@ -305,7 +306,6 @@ mod tests {
#[cfg(target_os = "windows")] #[cfg(target_os = "windows")]
use std::os::windows::fs::symlink_dir as symlink; use std::os::windows::fs::symlink_dir as symlink;
use std::path::Path; use std::path::Path;
use std::process::Command;
use std::{fs, io}; use std::{fs, io};
use tempfile::TempDir; use tempfile::TempDir;
@ -443,7 +443,7 @@ mod tests {
} }
fn init_repo(path: &Path) -> io::Result<()> { fn init_repo(path: &Path) -> io::Result<()> {
Command::new("git") create_command("git")?
.args(&["init"]) .args(&["init"])
.current_dir(path) .current_dir(path)
.output() .output()

View File

@ -345,10 +345,10 @@ enum FileType {
mod tests { mod tests {
use super::*; use super::*;
use crate::test::ModuleRenderer; use crate::test::ModuleRenderer;
use crate::utils::create_command;
use ansi_term::Color; use ansi_term::Color;
use std::fs::{self, OpenOptions}; use std::fs::{self, OpenOptions};
use std::io::{self, Write}; use std::io::{self, Write};
use std::process::Command;
use tempfile::{self, TempDir}; use tempfile::{self, TempDir};
#[test] #[test]
@ -551,7 +551,7 @@ mod tests {
let repo_dir = tempfile::tempdir()?; let repo_dir = tempfile::tempdir()?;
if is_repo { if is_repo {
Command::new("git") create_command("git")?
.args(&["init", "--quiet"]) .args(&["init", "--quiet"])
.current_dir(repo_dir.path()) .current_dir(repo_dir.path())
.output()?; .output()?;

View File

@ -121,9 +121,9 @@ fn get_first_grapheme(text: &str) -> &str {
mod tests { mod tests {
use ansi_term::Color; use ansi_term::Color;
use std::io; use std::io;
use std::process::Command;
use crate::test::{fixture_repo, FixtureProvider, ModuleRenderer}; use crate::test::{fixture_repo, FixtureProvider, ModuleRenderer};
use crate::utils::create_command;
#[test] #[test]
fn show_nothing_on_empty_dir() -> io::Result<()> { fn show_nothing_on_empty_dir() -> io::Result<()> {
@ -270,12 +270,12 @@ mod tests {
fn test_works_with_unborn_default_branch() -> io::Result<()> { fn test_works_with_unborn_default_branch() -> io::Result<()> {
let repo_dir = tempfile::tempdir()?; let repo_dir = tempfile::tempdir()?;
Command::new("git") create_command("git")?
.args(&["init"]) .args(&["init"])
.current_dir(&repo_dir) .current_dir(&repo_dir)
.output()?; .output()?;
Command::new("git") create_command("git")?
.args(&["symbolic-ref", "HEAD", "refs/heads/main"]) .args(&["symbolic-ref", "HEAD", "refs/heads/main"])
.current_dir(&repo_dir) .current_dir(&repo_dir)
.output()?; .output()?;
@ -297,7 +297,7 @@ mod tests {
fn test_render_branch_only_attached_on_branch() -> io::Result<()> { fn test_render_branch_only_attached_on_branch() -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
Command::new("git") create_command("git")?
.args(&["checkout", "-b", "test_branch"]) .args(&["checkout", "-b", "test_branch"])
.current_dir(repo_dir.path()) .current_dir(repo_dir.path())
.output()?; .output()?;
@ -325,7 +325,7 @@ mod tests {
fn test_render_branch_only_attached_on_detached() -> io::Result<()> { fn test_render_branch_only_attached_on_detached() -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
Command::new("git") create_command("git")?
.args(&["checkout", "@~1"]) .args(&["checkout", "@~1"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
@ -348,12 +348,12 @@ mod tests {
fn test_works_in_bare_repo() -> io::Result<()> { fn test_works_in_bare_repo() -> io::Result<()> {
let repo_dir = tempfile::tempdir()?; let repo_dir = tempfile::tempdir()?;
Command::new("git") create_command("git")?
.args(&["init", "--bare"]) .args(&["init", "--bare"])
.current_dir(&repo_dir) .current_dir(&repo_dir)
.output()?; .output()?;
Command::new("git") create_command("git")?
.args(&["symbolic-ref", "HEAD", "refs/heads/main"]) .args(&["symbolic-ref", "HEAD", "refs/heads/main"])
.current_dir(&repo_dir) .current_dir(&repo_dir)
.output()?; .output()?;
@ -379,7 +379,7 @@ mod tests {
// fn test_git_dir_env_variable() -> io::Result<()> {let repo_dir = // fn test_git_dir_env_variable() -> io::Result<()> {let repo_dir =
// tempfile::tempdir()?; // tempfile::tempdir()?;
// Command::new("git") // create_command("git")?
// .args(&["init"]) // .args(&["init"])
// .current_dir(&repo_dir) // .current_dir(&repo_dir)
// .output()?; // .output()?;
@ -424,7 +424,7 @@ mod tests {
) -> io::Result<()> { ) -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
Command::new("git") create_command("git")?
.args(&["checkout", "-b", branch_name]) .args(&["checkout", "-b", branch_name])
.current_dir(repo_dir.path()) .current_dir(repo_dir.path())
.output()?; .output()?;
@ -463,7 +463,7 @@ mod tests {
) -> io::Result<()> { ) -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
Command::new("git") create_command("git")?
.args(&["checkout", "-b", branch_name]) .args(&["checkout", "-b", branch_name])
.current_dir(repo_dir.path()) .current_dir(repo_dir.path())
.output()?; .output()?;

View File

@ -105,10 +105,10 @@ fn id_to_hex_abbrev(bytes: &[u8], len: usize) -> String {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use ansi_term::Color; use ansi_term::Color;
use std::process::Command;
use std::{io, str}; use std::{io, str};
use crate::test::{fixture_repo, FixtureProvider, ModuleRenderer}; use crate::test::{fixture_repo, FixtureProvider, ModuleRenderer};
use crate::utils::create_command;
#[test] #[test]
fn show_nothing_on_empty_dir() -> io::Result<()> { fn show_nothing_on_empty_dir() -> io::Result<()> {
@ -128,7 +128,7 @@ mod tests {
fn test_render_commit_hash() -> io::Result<()> { fn test_render_commit_hash() -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
let mut git_output = Command::new("git") let mut git_output = create_command("git")?
.args(&["rev-parse", "HEAD"]) .args(&["rev-parse", "HEAD"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()? .output()?
@ -160,7 +160,7 @@ mod tests {
fn test_render_commit_hash_len_override() -> io::Result<()> { fn test_render_commit_hash_len_override() -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
let mut git_output = Command::new("git") let mut git_output = create_command("git")?
.args(&["rev-parse", "HEAD"]) .args(&["rev-parse", "HEAD"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()? .output()?
@ -207,12 +207,12 @@ mod tests {
fn test_render_commit_hash_only_detached_on_detached() -> io::Result<()> { fn test_render_commit_hash_only_detached_on_detached() -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
Command::new("git") create_command("git")?
.args(&["checkout", "@~1"]) .args(&["checkout", "@~1"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
let mut git_output = Command::new("git") let mut git_output = create_command("git")?
.args(&["rev-parse", "HEAD"]) .args(&["rev-parse", "HEAD"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()? .output()?
@ -240,7 +240,7 @@ mod tests {
fn test_render_commit_hash_with_tag_enabled() -> io::Result<()> { fn test_render_commit_hash_with_tag_enabled() -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
let mut git_commit = Command::new("git") let mut git_commit = create_command("git")?
.args(&["rev-parse", "HEAD"]) .args(&["rev-parse", "HEAD"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()? .output()?
@ -248,7 +248,7 @@ mod tests {
git_commit.truncate(7); git_commit.truncate(7);
let commit_output = str::from_utf8(&git_commit).unwrap().trim(); let commit_output = str::from_utf8(&git_commit).unwrap().trim();
let git_tag = Command::new("git") let git_tag = create_command("git")?
.args(&["describe", "--tags", "--exact-match", "HEAD"]) .args(&["describe", "--tags", "--exact-match", "HEAD"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()? .output()?
@ -283,17 +283,17 @@ mod tests {
fn test_render_commit_hash_only_detached_on_detached_with_tag_enabled() -> io::Result<()> { fn test_render_commit_hash_only_detached_on_detached_with_tag_enabled() -> io::Result<()> {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
Command::new("git") create_command("git")?
.args(&["checkout", "@~1"]) .args(&["checkout", "@~1"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
Command::new("git") create_command("git")?
.args(&["tag", "tagOnDetached", "-m", "Testing tags on detached"]) .args(&["tag", "tagOnDetached", "-m", "Testing tags on detached"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
let mut git_commit = Command::new("git") let mut git_commit = create_command("git")?
.args(&["rev-parse", "HEAD"]) .args(&["rev-parse", "HEAD"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()? .output()?
@ -301,7 +301,7 @@ mod tests {
git_commit.truncate(7); git_commit.truncate(7);
let commit_output = str::from_utf8(&git_commit).unwrap().trim(); let commit_output = str::from_utf8(&git_commit).unwrap().trim();
let git_tag = Command::new("git") let git_tag = create_command("git")?
.args(&["describe", "--tags", "--exact-match", "HEAD"]) .args(&["describe", "--tags", "--exact-match", "HEAD"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()? .output()?
@ -337,7 +337,7 @@ mod tests {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
let mut git_commit = Command::new("git") let mut git_commit = create_command("git")?
.args(&["rev-parse", "HEAD"]) .args(&["rev-parse", "HEAD"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()? .output()?
@ -345,7 +345,7 @@ mod tests {
git_commit.truncate(7); git_commit.truncate(7);
let commit_output = str::from_utf8(&git_commit).unwrap().trim(); let commit_output = str::from_utf8(&git_commit).unwrap().trim();
Command::new("git") create_command("git")?
.args(&["tag", "v2", "-m", "Testing tags v2"]) .args(&["tag", "v2", "-m", "Testing tags v2"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
@ -353,17 +353,17 @@ mod tests {
// Wait one second between tags // Wait one second between tags
thread::sleep(time::Duration::from_millis(1000)); thread::sleep(time::Duration::from_millis(1000));
Command::new("git") create_command("git")?
.args(&["tag", "v0", "-m", "Testing tags v0", "HEAD~1"]) .args(&["tag", "v0", "-m", "Testing tags v0", "HEAD~1"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
Command::new("git") create_command("git")?
.args(&["tag", "v1", "-m", "Testing tags v1"]) .args(&["tag", "v1", "-m", "Testing tags v1"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
let git_tag = Command::new("git") let git_tag = create_command("git")?
.args(&[ .args(&[
"for-each-ref", "for-each-ref",
"--contains", "--contains",

View File

@ -93,11 +93,12 @@ impl<'a> GitDiff<'a> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use crate::utils::create_command;
use std::ffi::OsStr; use std::ffi::OsStr;
use std::fs::OpenOptions; use std::fs::OpenOptions;
use std::io::{self, Error, ErrorKind, Write}; use std::io::{self, Error, ErrorKind, Write};
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use std::process::{Command, Stdio}; use std::process::Stdio;
use ansi_term::Color; use ansi_term::Color;
@ -193,7 +194,7 @@ mod tests {
A: IntoIterator<Item = S>, A: IntoIterator<Item = S>,
S: AsRef<OsStr>, S: AsRef<OsStr>,
{ {
let mut command = Command::new("git"); let mut command = create_command("git")?;
command command
.args(args) .args(args)
.stdout(Stdio::null()) .stdout(Stdio::null())

View File

@ -177,9 +177,10 @@ mod tests {
use std::fs::OpenOptions; use std::fs::OpenOptions;
use std::io::{self, Error, ErrorKind, Write}; use std::io::{self, Error, ErrorKind, Write};
use std::path::Path; use std::path::Path;
use std::process::{Command, Stdio}; use std::process::Stdio;
use crate::test::ModuleRenderer; use crate::test::ModuleRenderer;
use crate::utils::create_command;
#[test] #[test]
fn show_nothing_on_empty_dir() -> io::Result<()> { fn show_nothing_on_empty_dir() -> io::Result<()> {
@ -278,7 +279,7 @@ mod tests {
A: IntoIterator<Item = S>, A: IntoIterator<Item = S>,
S: AsRef<OsStr>, S: AsRef<OsStr>,
{ {
let mut command = Command::new("git"); let mut command = create_command("git")?;
command command
.args(args) .args(args)
.stdout(Stdio::null()) .stdout(Stdio::null())

View File

@ -309,9 +309,9 @@ mod tests {
use std::fs::{self, File}; use std::fs::{self, File};
use std::io; use std::io;
use std::path::Path; use std::path::Path;
use std::process::Command;
use crate::test::{fixture_repo, FixtureProvider, ModuleRenderer}; use crate::test::{fixture_repo, FixtureProvider, ModuleRenderer};
use crate::utils::create_command;
/// Right after the calls to git the filesystem state may not have finished /// Right after the calls to git the filesystem state may not have finished
/// updating yet causing some of the tests to fail. These barriers are placed /// updating yet causing some of the tests to fail. These barriers are placed
@ -524,7 +524,7 @@ mod tests {
create_untracked(&repo_dir.path())?; create_untracked(&repo_dir.path())?;
Command::new("git") create_command("git")?
.args(&["config", "status.showUntrackedFiles", "no"]) .args(&["config", "status.showUntrackedFiles", "no"])
.current_dir(repo_dir.path()) .current_dir(repo_dir.path())
.output()?; .output()?;
@ -546,7 +546,7 @@ mod tests {
create_stash(&repo_dir.path())?; create_stash(&repo_dir.path())?;
Command::new("git") create_command("git")?
.args(&["reset", "--hard", "HEAD"]) .args(&["reset", "--hard", "HEAD"])
.current_dir(repo_dir.path()) .current_dir(repo_dir.path())
.output()?; .output()?;
@ -569,7 +569,7 @@ mod tests {
create_stash(&repo_dir.path())?; create_stash(&repo_dir.path())?;
barrier(); barrier();
Command::new("git") create_command("git")?
.args(&["reset", "--hard", "HEAD"]) .args(&["reset", "--hard", "HEAD"])
.current_dir(repo_dir.path()) .current_dir(repo_dir.path())
.output()?; .output()?;
@ -751,7 +751,7 @@ mod tests {
let worktree_dir = tempfile::tempdir()?; let worktree_dir = tempfile::tempdir()?;
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
Command::new("git") create_command("git")?
.args(&[ .args(&[
"config", "config",
"core.worktree", "core.worktree",
@ -781,11 +781,11 @@ mod tests {
let repo_dir = fixture_repo(FixtureProvider::Git)?; let repo_dir = fixture_repo(FixtureProvider::Git)?;
File::create(repo_dir.path().join("a"))?.sync_all()?; File::create(repo_dir.path().join("a"))?.sync_all()?;
File::create(repo_dir.path().join("b"))?.sync_all()?; File::create(repo_dir.path().join("b"))?.sync_all()?;
Command::new("git") create_command("git")?
.args(&["add", "--all"]) .args(&["add", "--all"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
Command::new("git") create_command("git")?
.args(&["commit", "-m", "add new files", "--no-gpg-sign"]) .args(&["commit", "-m", "add new files", "--no-gpg-sign"])
.current_dir(&repo_dir.path()) .current_dir(&repo_dir.path())
.output()?; .output()?;
@ -814,7 +814,7 @@ mod tests {
fn ahead(repo_dir: &Path) -> io::Result<()> { fn ahead(repo_dir: &Path) -> io::Result<()> {
File::create(repo_dir.join("readme.md"))?.sync_all()?; File::create(repo_dir.join("readme.md"))?.sync_all()?;
Command::new("git") create_command("git")?
.args(&["commit", "-am", "Update readme", "--no-gpg-sign"]) .args(&["commit", "-am", "Update readme", "--no-gpg-sign"])
.current_dir(&repo_dir) .current_dir(&repo_dir)
.output()?; .output()?;
@ -824,7 +824,7 @@ mod tests {
} }
fn behind(repo_dir: &Path) -> io::Result<()> { fn behind(repo_dir: &Path) -> io::Result<()> {
Command::new("git") create_command("git")?
.args(&["reset", "--hard", "HEAD^"]) .args(&["reset", "--hard", "HEAD^"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
@ -834,7 +834,7 @@ mod tests {
} }
fn diverge(repo_dir: &Path) -> io::Result<()> { fn diverge(repo_dir: &Path) -> io::Result<()> {
Command::new("git") create_command("git")?
.args(&["reset", "--hard", "HEAD^"]) .args(&["reset", "--hard", "HEAD^"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
@ -842,7 +842,7 @@ mod tests {
fs::write(repo_dir.join("Cargo.toml"), " ")?; fs::write(repo_dir.join("Cargo.toml"), " ")?;
Command::new("git") create_command("git")?
.args(&["commit", "-am", "Update readme", "--no-gpg-sign"]) .args(&["commit", "-am", "Update readme", "--no-gpg-sign"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
@ -852,7 +852,7 @@ mod tests {
} }
fn create_conflict(repo_dir: &Path) -> io::Result<()> { fn create_conflict(repo_dir: &Path) -> io::Result<()> {
Command::new("git") create_command("git")?
.args(&["reset", "--hard", "HEAD^"]) .args(&["reset", "--hard", "HEAD^"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
@ -860,19 +860,19 @@ mod tests {
fs::write(repo_dir.join("readme.md"), "# goodbye")?; fs::write(repo_dir.join("readme.md"), "# goodbye")?;
Command::new("git") create_command("git")?
.args(&["add", "."]) .args(&["add", "."])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
barrier(); barrier();
Command::new("git") create_command("git")?
.args(&["commit", "-m", "Change readme", "--no-gpg-sign"]) .args(&["commit", "-m", "Change readme", "--no-gpg-sign"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
barrier(); barrier();
Command::new("git") create_command("git")?
.args(&["pull", "--rebase"]) .args(&["pull", "--rebase"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
@ -885,7 +885,7 @@ mod tests {
File::create(repo_dir.join("readme.md"))?.sync_all()?; File::create(repo_dir.join("readme.md"))?.sync_all()?;
barrier(); barrier();
Command::new("git") create_command("git")?
.args(&["stash", "--all"]) .args(&["stash", "--all"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
@ -903,7 +903,7 @@ mod tests {
fn create_added(repo_dir: &Path) -> io::Result<()> { fn create_added(repo_dir: &Path) -> io::Result<()> {
File::create(repo_dir.join("license"))?.sync_all()?; File::create(repo_dir.join("license"))?.sync_all()?;
Command::new("git") create_command("git")?
.args(&["add", "-A", "-N"]) .args(&["add", "-A", "-N"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
@ -921,7 +921,7 @@ mod tests {
fn create_staged(repo_dir: &Path) -> io::Result<()> { fn create_staged(repo_dir: &Path) -> io::Result<()> {
File::create(repo_dir.join("license"))?.sync_all()?; File::create(repo_dir.join("license"))?.sync_all()?;
Command::new("git") create_command("git")?
.args(&["add", "."]) .args(&["add", "."])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
@ -931,13 +931,13 @@ mod tests {
} }
fn create_renamed(repo_dir: &Path) -> io::Result<()> { fn create_renamed(repo_dir: &Path) -> io::Result<()> {
Command::new("git") create_command("git")?
.args(&["mv", "readme.md", "readme.md.bak"]) .args(&["mv", "readme.md", "readme.md.bak"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;
barrier(); barrier();
Command::new("git") create_command("git")?
.args(&["add", "-A"]) .args(&["add", "-A"])
.current_dir(repo_dir) .current_dir(repo_dir)
.output()?; .output()?;

View File

@ -103,9 +103,9 @@ mod tests {
use std::fs; use std::fs;
use std::io; use std::io;
use std::path::Path; use std::path::Path;
use std::process::Command;
use crate::test::{fixture_repo, FixtureProvider, ModuleRenderer}; use crate::test::{fixture_repo, FixtureProvider, ModuleRenderer};
use crate::utils::create_command;
enum Expect<'a> { enum Expect<'a> {
BranchName(&'a str), BranchName(&'a str),
@ -335,7 +335,7 @@ mod tests {
} }
fn run_hg(args: &[&str], repo_dir: &Path) -> io::Result<()> { fn run_hg(args: &[&str], repo_dir: &Path) -> io::Result<()> {
Command::new("hg") create_command("hg")?
.args(args) .args(args)
.current_dir(&repo_dir) .current_dir(&repo_dir)
.output()?; .output()?;

View File

@ -1,6 +1,6 @@
use std::fs; use std::fs;
use std::path::Path; use std::path::Path;
use std::process::{Command, Output}; use std::process::Output;
use serde::Deserialize; use serde::Deserialize;
@ -8,6 +8,7 @@ use super::{Context, Module, RootModuleConfig};
use crate::configs::rust::RustConfig; use crate::configs::rust::RustConfig;
use crate::formatter::{StringFormatter, VersionFormatter}; use crate::formatter::{StringFormatter, VersionFormatter};
use crate::utils::create_command;
/// Creates a module with the current Rust version /// Creates a module with the current Rust version
pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> { pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
@ -99,7 +100,8 @@ fn env_rustup_toolchain(context: &Context) -> Option<String> {
} }
fn execute_rustup_override_list(cwd: &Path) -> Option<String> { fn execute_rustup_override_list(cwd: &Path) -> Option<String> {
let Output { stdout, .. } = Command::new("rustup") let Output { stdout, .. } = create_command("rustup")
.ok()?
.args(&["override", "list"]) .args(&["override", "list"])
.output() .output()
.ok()?; .ok()?;
@ -188,9 +190,8 @@ fn find_rust_toolchain_file(context: &Context) -> Option<String> {
} }
fn execute_rustup_run_rustc_version(toolchain: &str) -> RustupRunRustcVersionOutcome { fn execute_rustup_run_rustc_version(toolchain: &str) -> RustupRunRustcVersionOutcome {
Command::new("rustup") create_command("rustup")
.args(&["run", toolchain, "rustc", "--version"]) .and_then(|mut cmd| cmd.args(&["run", toolchain, "rustc", "--version"]).output())
.output()
.map(extract_toolchain_from_rustup_run_rustc_version) .map(extract_toolchain_from_rustup_run_rustc_version)
.unwrap_or(RustupRunRustcVersionOutcome::RustupNotWorking) .unwrap_or(RustupRunRustcVersionOutcome::RustupNotWorking)
} }
@ -212,7 +213,7 @@ fn extract_toolchain_from_rustup_run_rustc_version(output: Output) -> RustupRunR
} }
fn execute_rustc_version() -> Option<String> { fn execute_rustc_version() -> Option<String> {
match Command::new("rustc").arg("--version").output() { match create_command("rustc").ok()?.arg("--version").output() {
Ok(output) => Some(String::from_utf8(output.stdout).unwrap()), Ok(output) => Some(String::from_utf8(output.stdout).unwrap()),
Err(_) => None, Err(_) => None,
} }

View File

@ -1,11 +1,13 @@
use crate::context::{Context, Shell}; use crate::context::{Context, Shell};
use crate::logger::StarshipLogger; use crate::logger::StarshipLogger;
use crate::{config::StarshipConfig, utils::CommandOutput}; use crate::{
config::StarshipConfig,
utils::{create_command, CommandOutput},
};
use log::{Level, LevelFilter}; use log::{Level, LevelFilter};
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use std::io; use std::io;
use std::path::PathBuf; use std::path::PathBuf;
use std::process::Command;
use tempfile::TempDir; use tempfile::TempDir;
static FIXTURE_DIR: Lazy<PathBuf> = static FIXTURE_DIR: Lazy<PathBuf> =
@ -151,24 +153,24 @@ pub fn fixture_repo(provider: FixtureProvider) -> io::Result<TempDir> {
FixtureProvider::Git => { FixtureProvider::Git => {
let path = tempfile::tempdir()?; let path = tempfile::tempdir()?;
Command::new("git") create_command("git")?
.current_dir(path.path()) .current_dir(path.path())
.args(&["clone", "-b", "master"]) .args(&["clone", "-b", "master"])
.arg(GIT_FIXTURE.as_os_str()) .arg(GIT_FIXTURE.as_os_str())
.arg(&path.path()) .arg(&path.path())
.output()?; .output()?;
Command::new("git") create_command("git")?
.args(&["config", "--local", "user.email", "starship@example.com"]) .args(&["config", "--local", "user.email", "starship@example.com"])
.current_dir(&path.path()) .current_dir(&path.path())
.output()?; .output()?;
Command::new("git") create_command("git")?
.args(&["config", "--local", "user.name", "starship"]) .args(&["config", "--local", "user.name", "starship"])
.current_dir(&path.path()) .current_dir(&path.path())
.output()?; .output()?;
Command::new("git") create_command("git")?
.args(&["reset", "--hard", "HEAD"]) .args(&["reset", "--hard", "HEAD"])
.current_dir(&path.path()) .current_dir(&path.path())
.output()?; .output()?;
@ -178,7 +180,7 @@ pub fn fixture_repo(provider: FixtureProvider) -> io::Result<TempDir> {
FixtureProvider::Hg => { FixtureProvider::Hg => {
let path = tempfile::tempdir()?; let path = tempfile::tempdir()?;
Command::new("hg") create_command("hg")?
.current_dir(path.path()) .current_dir(path.path())
.arg("clone") .arg("clone")
.arg(HG_FIXTURE.as_os_str()) .arg(HG_FIXTURE.as_os_str())

View File

@ -1,7 +1,8 @@
use process_control::{ChildExt, Timeout}; use process_control::{ChildExt, Timeout};
use std::ffi::OsStr;
use std::fmt::Debug; use std::fmt::Debug;
use std::fs::read_to_string; use std::fs::read_to_string;
use std::io::Result; use std::io::{Error, ErrorKind, Result};
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use std::process::{Command, Stdio}; use std::process::{Command, Stdio};
use std::time::{Duration, Instant}; use std::time::{Duration, Instant};
@ -23,6 +24,33 @@ pub fn read_file<P: AsRef<Path> + Debug>(file_name: P) -> Result<String> {
result result
} }
/// Attempt to resolve `binary_name` from and creates a new `Command` pointing at it
/// This allows executing cmd files on Windows and prevents running executable from cwd on Windows
/// This function also initialises std{err,out,in} to protect against processes changing the console mode
pub fn create_command<T: AsRef<OsStr>>(binary_name: T) -> Result<Command> {
let binary_name = binary_name.as_ref();
log::trace!("Creating Command struct with binary name {:?}", binary_name);
let full_path = match which::which(binary_name) {
Ok(full_path) => {
log::trace!("Using {:?} as {:?}", full_path, binary_name);
full_path
}
Err(error) => {
log::trace!("Unable to find {:?} in PATH, {:?}", binary_name, error);
return Err(Error::new(ErrorKind::NotFound, error));
}
};
#[allow(clippy::disallowed_method)]
let mut cmd = Command::new(full_path);
cmd.stderr(Stdio::piped())
.stdout(Stdio::piped())
.stdin(Stdio::null());
Ok(cmd)
}
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct CommandOutput { pub struct CommandOutput {
pub stdout: String, pub stdout: String,
@ -324,6 +352,7 @@ fn internal_exec_cmd(cmd: &str, args: &[&str], time_limit: Duration) -> Option<C
let start = Instant::now(); let start = Instant::now();
#[allow(clippy::disallowed_method)]
let process = match Command::new(full_path) let process = match Command::new(full_path)
.args(args) .args(args)
.stderr(Stdio::piped()) .stderr(Stdio::piped())