Matthew Jackson
061f68594a
Library-wide review-and-fix pass: tightened AACS keydb/handshake/variant handling and trailing-partial-unit policy, corrected MPLS mark offset and added UDF allocation bounds, hardened the mux/codec framing and M2TS paths, guarded SCSI READ CAPACITY short transfers and unified error mapping, added overflow guards on untrusted disc input, and made prefetch shutdown deterministic. Release profile now builds with thin LTO + single codegen unit.
100 lines
3.4 KiB
Rust
100 lines
3.4 KiB
Rust
//! Windows drive discovery and device resolution.
|
|
|
|
use crate::drive::DeviceResolution;
|
|
use crate::error::Result;
|
|
use crate::identity::DriveId;
|
|
use std::path::Path;
|
|
|
|
/// SCSI peripheral device type 5 = MMC / optical, in the low 5 bits of
|
|
/// INQUIRY byte 0.
|
|
const SCSI_PERIPHERAL_TYPE_OPTICAL: u8 = 0x05;
|
|
|
|
/// Discover optical drives. Probes `\\.\CdRom0..15` first; only if none
|
|
/// are found does it fall back to scanning drive letters `D..Z`. Each
|
|
/// candidate is opened, INQUIRY'd, and kept only if its peripheral device
|
|
/// type is optical (MMC, type 0x05). Returns normalized `\\.\` paths.
|
|
pub fn find_drives() -> Vec<(String, DriveId)> {
|
|
let mut drives = Vec::new();
|
|
|
|
// Try CdRom0..CdRom15
|
|
for i in 0..16 {
|
|
let path = format!("\\\\.\\CdRom{}", i);
|
|
if let Ok(mut transport) = crate::scsi::open(Path::new(&path)) {
|
|
if let Ok(id) = DriveId::from_drive(transport.as_mut()) {
|
|
if !id.raw_inquiry.is_empty()
|
|
&& (id.raw_inquiry[0] & 0x1F) == SCSI_PERIPHERAL_TYPE_OPTICAL
|
|
{
|
|
drives.push((path, id));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Also try drive letters if CdRom didn't find anything
|
|
if drives.is_empty() {
|
|
for letter in b'D'..=b'Z' {
|
|
let path = format!("{}:", letter as char);
|
|
if let Ok(mut transport) = crate::scsi::open(Path::new(&path)) {
|
|
if let Ok(id) = DriveId::from_drive(transport.as_mut()) {
|
|
if !id.raw_inquiry.is_empty()
|
|
&& (id.raw_inquiry[0] & 0x1F) == SCSI_PERIPHERAL_TYPE_OPTICAL
|
|
{
|
|
// Normalize so returned paths are consistently in
|
|
// \\.\ form regardless of which loop matched.
|
|
drives.push((normalize_path(&path), id));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
drives
|
|
}
|
|
|
|
/// Resolve a device path to its normalized Windows `\\.\` form. Windows
|
|
/// has no `sr`→`sg` symlink-target indirection, so resolution is purely a
|
|
/// path normalization and always reports [`DeviceResolution::Direct`].
|
|
pub fn resolve_device(path: &str) -> Result<(String, DeviceResolution)> {
|
|
Ok((normalize_path(path), DeviceResolution::Direct))
|
|
}
|
|
|
|
/// Normalize a device path to Windows \\.\X: format.
|
|
///
|
|
/// Accepts: "D:", "D:\\", "\\.\D:", "\\.\CdRom0"
|
|
///
|
|
/// NOTE: A near-identical `normalize_device_path` exists in `scsi::windows`.
|
|
/// Both are kept because they live in separate `cfg(windows)` modules that
|
|
/// cannot easily share a helper without introducing cross-module coupling.
|
|
fn normalize_path(path: &str) -> String {
|
|
if path.starts_with("\\\\.\\") {
|
|
return path.to_string();
|
|
}
|
|
let trimmed = path.trim_end_matches('\\');
|
|
if trimmed.len() == 2 && trimmed.as_bytes()[1] == b':' {
|
|
return format!("\\\\.\\{}", trimmed);
|
|
}
|
|
format!("\\\\.\\{}", path)
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn normalize_drive_letter() {
|
|
assert_eq!(normalize_path("D:"), "\\\\.\\D:");
|
|
assert_eq!(normalize_path("E:\\"), "\\\\.\\E:");
|
|
}
|
|
|
|
#[test]
|
|
fn normalize_already_prefixed() {
|
|
assert_eq!(normalize_path("\\\\.\\D:"), "\\\\.\\D:");
|
|
assert_eq!(normalize_path("\\\\.\\CdRom0"), "\\\\.\\CdRom0");
|
|
}
|
|
|
|
#[test]
|
|
fn normalize_cdrom() {
|
|
assert_eq!(normalize_path("CdRom0"), "\\\\.\\CdRom0");
|
|
}
|
|
}
|