Buffer UDF sector reads — eliminates scan hang on USB drives

This commit is contained in:
Matt Jackson
2026-04-17 18:24:29 +00:00
parent bc62cb35f4
commit c6b08381ce
3 changed files with 160 additions and 4 deletions
+22
View File
@@ -12,8 +12,19 @@ impl Disc {
reader: &mut dyn SectorReader,
udf_fs: &udf::UdfFs,
) -> Vec<DiscTitle> {
let t0 = std::time::Instant::now();
let mut titles = Vec::new();
if let Some(playlist_dir) = udf_fs.find_dir("/BDMV/PLAYLIST") {
let mpls_count = playlist_dir
.entries
.iter()
.filter(|e| !e.is_dir && e.name.to_lowercase().ends_with(".mpls"))
.count();
eprintln!(
"[titles {:.1}s] {} MPLS files",
t0.elapsed().as_secs_f64(),
mpls_count
);
for entry in &playlist_dir.entries {
if !entry.is_dir && entry.name.to_lowercase().ends_with(".mpls") {
let path = format!("/BDMV/PLAYLIST/{}", entry.name);
@@ -21,12 +32,23 @@ impl Disc {
if let Some(title) =
Self::parse_playlist(reader, udf_fs, &entry.name, &mpls_data)
{
eprintln!(
"[titles {:.1}s] {} -> title ({:.0}s)",
t0.elapsed().as_secs_f64(),
entry.name,
title.duration_secs
);
titles.push(title);
}
}
}
}
}
eprintln!(
"[titles {:.1}s] done: {} titles",
t0.elapsed().as_secs_f64(),
titles.len()
);
titles
}
+43 -4
View File
@@ -940,17 +940,37 @@ impl Disc {
/// The session must be open and unlocked (Drive::open handles this).
/// All disc reads use standard READ(10) via UDF -- no vendor SCSI commands.
pub fn scan(session: &mut Drive, opts: &ScanOptions) -> Result<Self> {
let t0 = std::time::Instant::now();
// READ CAPACITY may fail in LibreDrive mode — proceed with 0 and estimate later
let capacity = Self::read_capacity(session).unwrap_or(0);
eprintln!(
"[scan {:.1}s] capacity={}",
t0.elapsed().as_secs_f64(),
capacity
);
// AACS handshake (Blu-ray/UHD)
let handshake = Self::do_handshake(session, opts);
eprintln!("[scan {:.1}s] handshake done", t0.elapsed().as_secs_f64());
// Request max read speed — removes riplock on DVD
// (BD/UHD speed is set by firmware init, but DVD needs explicit SET CD SPEED)
session.set_speed(0xFFFF);
eprintln!("[scan {:.1}s] speed set", t0.elapsed().as_secs_f64());
let mut disc = Self::scan_with(session, capacity, handshake, opts)?;
let batch = detect_max_batch_sectors(session.device_path());
let mut buffered = udf::BufferedSectorReader::new(session, batch);
let udf_fs = udf::read_filesystem(&mut buffered)?;
eprintln!(
"[scan {:.1}s] UDF parsed, prefetching metadata ({} sectors)...",
t0.elapsed().as_secs_f64(),
udf_fs.metadata_sectors()
);
buffered.prefetch(udf_fs.metadata_start(), udf_fs.metadata_sectors());
eprintln!("[scan {:.1}s] metadata cached", t0.elapsed().as_secs_f64());
let mut disc = Self::scan_with(&mut buffered, capacity, handshake, opts, udf_fs)?;
eprintln!("[scan {:.1}s] scan_with done", t0.elapsed().as_secs_f64());
// CSS key extraction for DVDs (bus auth → disc key → title key).
// Must be a single auth session — can't call authenticate() separately.
@@ -988,7 +1008,8 @@ impl Disc {
capacity: u32,
opts: &ScanOptions,
) -> Result<Self> {
Self::scan_with(reader, capacity, None, opts)
let udf_fs = udf::read_filesystem(reader)?;
Self::scan_with(reader, capacity, None, opts, udf_fs)
}
/// Core scan pipeline — works with any SectorReader.
@@ -997,15 +1018,19 @@ impl Disc {
capacity: u32,
handshake: Option<HandshakeResult>,
opts: &ScanOptions,
udf_fs: udf::UdfFs,
) -> Result<Self> {
// 1. UDF filesystem
let udf_fs = udf::read_filesystem(reader)?;
let t0 = std::time::Instant::now();
// 2. Resolve encryption (AACS, CSS, or none)
let encrypted =
udf_fs.find_dir("/AACS").is_some() || udf_fs.find_dir("/BDMV/AACS").is_some();
let aacs = if encrypted {
eprintln!(
"[scan_with {:.1}s] resolving encryption...",
t0.elapsed().as_secs_f64()
);
if let Some(keydb_path) = opts.resolve_keydb() {
Self::resolve_encryption(&udf_fs, reader, &keydb_path, handshake.as_ref()).ok()
} else {
@@ -1014,6 +1039,10 @@ impl Disc {
} else {
None
};
eprintln!(
"[scan_with {:.1}s] encryption done",
t0.elapsed().as_secs_f64()
);
// 3. Titles — BD (MPLS playlists) or DVD (IFO title sets)
let (mut titles, content_format) = if udf_fs.find_dir("/BDMV").is_some() {
@@ -1029,6 +1058,11 @@ impl Disc {
} else {
(Vec::new(), ContentFormat::BdTs)
};
eprintln!(
"[scan_with {:.1}s] titles done: {}",
t0.elapsed().as_secs_f64(),
titles.len()
);
titles.sort_by(|a, b| {
b.duration_secs
.partial_cmp(&a.duration_secs)
@@ -1037,7 +1071,12 @@ impl Disc {
// 4. Metadata + labels
let meta_title = Self::read_meta_title(reader, &udf_fs);
eprintln!(
"[scan_with {:.1}s] meta_title done",
t0.elapsed().as_secs_f64()
);
crate::labels::apply(reader, &udf_fs, &mut titles);
eprintln!("[scan_with {:.1}s] labels done", t0.elapsed().as_secs_f64());
// 5. Derive format, layers, region
let format = Self::detect_format(&titles);