YGGverse/aquatic-crawler
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic-crawler.git synced 2026-03-31 17:15:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

validate info_hash in public methods only

Browse source
This commit is contained in:
yggverse 2025-08-08 02:08:48 +03:00
parent 63dbaa9dda
commit e88a3e9f50
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/preload.rs
View file

@ -39,8 +39,11 @@ impl Preload {
torrent_bytes: Vec<u8>,
persist_files: Option<HashSet<PathBuf>>,
) -> Result<()> {
if !is_info_hash(info_hash) {
bail!("Invalid info-hash `{info_hash}`")
}
// persist torrent bytes to file
let t = self.torrent(info_hash)?;
let t = self.torrent(info_hash);
fs::write(&t, torrent_bytes)?;
log::debug!("persist torrent bytes for `{}`", t.to_string_lossy());
// persist preload files
@ -114,16 +117,16 @@ impl Preload {
}
pub fn contains_torrent(&self, info_hash: &str) -> Result<bool> {
Ok(fs::exists(self.torrent(info_hash)?)?)
}
fn torrent(&self, info_hash: &str) -> Result<PathBuf> {
if !is_info_hash(info_hash) {
bail!("Invalid info-hash `{info_hash}`")
}
Ok(fs::exists(self.torrent(info_hash))?)
}
fn torrent(&self, info_hash: &str) -> PathBuf {
let mut p = PathBuf::from(&self.root);
p.push(format!("{info_hash}.torrent"));
Ok(p)
p
}
}