Guard for drive_letter before index

This commit is contained in:
Zhenping Zhao
2024-11-23 16:52:42 -08:00
parent 6ef5b04b25
commit 26fb863a67

View File

@ -117,8 +117,11 @@ impl DriveToPwdMap {
if let (Some(drive_letter), Some(path_str)) = if let (Some(drive_letter), Some(path_str)) =
(Self::extract_drive_letter(path), path.to_str()) (Self::extract_drive_letter(path), path.to_str())
{ {
self.map[drive_letter as usize - 'A' as usize] = Some(path_str.to_string()); if drive_letter.is_ascii_alphabetic() {
return Ok(()); let drive_letter = drive_letter.to_ascii_uppercase();
self.map[drive_letter as usize - 'A' as usize] = Some(path_str.to_string());
return Ok(());
}
} }
Err(format!("Invalid path: {}", path.display())) Err(format!("Invalid path: {}", path.display()))
} }