YGGverse/aquatic-crawler
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic-crawler.git synced 2026-04-02 18:15:31 +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>, torrent_bytes: Vec<u8>,
persist_files: Option<HashSet<PathBuf>>, persist_files: Option<HashSet<PathBuf>>,
) -> Result<()> { ) -> Result<()> {
if !is_info_hash(info_hash) {
bail!("Invalid info-hash `{info_hash}`")
}
// persist torrent bytes to file // persist torrent bytes to file
let t = self.torrent(info_hash)?; let t = self.torrent(info_hash);
fs::write(&t, torrent_bytes)?; fs::write(&t, torrent_bytes)?;
log::debug!("persist torrent bytes for `{}`", t.to_string_lossy()); log::debug!("persist torrent bytes for `{}`", t.to_string_lossy());
// persist preload files // persist preload files
@ -114,16 +117,16 @@ impl Preload {
} }
pub fn contains_torrent(&self, info_hash: &str) -> Result<bool> { 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) { if !is_info_hash(info_hash) {
bail!("Invalid 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); let mut p = PathBuf::from(&self.root);
p.push(format!("{info_hash}.torrent")); p.push(format!("{info_hash}.torrent"));
Ok(p) p
} }
} }