Test Case: rust-autocomplete-1263

  1let root = paths::root();  2        let dir = root.join(format!("{}_{}", name, id));  3        Workdir {  4            root,  5            dir,  6            flexible: false,  7        }  8    }  9​ 10    pub fn flexible(mut self, flexible: bool) -> Workdir { 11        self.flexible = flexible; 12        self 13    } 14​ 15    pub fn path(&self) -> &Path { 16        &self.dir 17    } 18​ 19    pub fn mkdir(&self) -> &Workdir { 20        fs::create_dir_all(&self.dir).unwrap(); 21        self 22    } 23​ 24    pub fn read_file(&self, name: &str) -> String { 25        let contents = fs::read_to_string(self.dir.join(name)).unwrap(); 26        contents 27    } 28​ 29    pub fn read_file_binary(&self, name: &str) -> Vec<u8> { 30        let contents = fs::read(self.dir.join(name)).unwrap(); 31        contents 32    } 33​ 34    pub fn write_file(&self, name: &str, contents: &str) -> &Workdir { 35        self.write_file_binary(name, contents.as_bytes()) 36    } 37​ 38    pub fn write_file_binary(&self, name: &str, contents: &[u8]) -> &Workdir { 39        let dst = self.dir.join(name); 40        fs::write(&dst, contents).unwrap(); 41        self 42    } 43​ 44    pub fn add_file(&self, name: &str, contents: &str) -> &Workdir { 45        self.add_file_binary(name, contents.as_bytes()) 46    } 47​ 48    pub fn add_file_binary(&self, name: &str, contents: &[u8]) -> &Workdir { 49        self.add_file_with_mode(name, contents, 0o644) 50    } 51​ 52    pub fn add_file_with_mode(&self, name: &str, contents: &[u8], mode: u32) -> &Workdir { 53        let dst = self.dir.join(name); 54        fs::write(&dst, contents).unwrap(); 55        fs::set_permissions(&dst, PermissionsExt::from_mode(mode)).unwrap(); 56        self 57    } 58​ 59    pub fn add_template(&self, name: &str) -> &Workdir { 60        self.add_template_with_name(name, name) 61    } 62​ 63    pub fn add_template_with_name(&self, name: &str, template: &str) -> &Workdir { 64        let template = paths::root().join("templates").join(template); 65        let contents = fs::read_to_string(template).unwrap(); 66        self.write_file(name, &contents); 67        self 68    } 69​ 70    pub fn add_shell(&self, name: &str, command: &str) -> &Workdir { 71        let n = if cfg!(windows) { 72            format!("{}.bat", name) 73        } else { 74            name.to_string() 75        }; 76        if cfg!(windows) { 77            self.write_file(&n, command); 78        } else { 79            self.write_file(&n, &format!("#!/bin/sh\n{}", command)); 80            self.chmod(&n, 0o755); 81        } 82        self 83    } 84​ 85    pub fn chmod(&self, name: &str, mode: u32) -> &Workdir { 86        let dst = self.dir.join(name); 87        fs::set_permissions(&dst, PermissionsExt::from_mode(mode)).unwrap(); 88        self 89    } 90​ 91    pub fn rm(&self) -> &Workdir { 92        if !self.flexible { 93            panic!( 94                "Workdir::rm called, but `flexible` wasn't enabled. \ 95                 This is typically a bug in a test, as the work directory \ 96                 is needed for inspection after the test fails. To fix this, \ 97                 call `flexible(true)` on the `Workdir` builder." 98            ); 99        }100        drop(fs::remove_dir_all(&self.dir));101        self102    }103​104    pub fn command<S: AsRef<OsStr>>(&self, program: S) -> Command {105        let mut cmd = Command::new(program);106        cmd.current_dir(&self.dir);107        cmd108    }109}110​111impl Drop for Workdir {112    fn drop(&mut self) {113        if thread::panicking() {114            return;115        }116        if !self.flexible {117            return;118        }119        drop(fs::remove_dir_all(&self.dir));120    }121}

1. warningLine 105

   Potential for command injection if input is not sanitized.