leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Disable upload: do not expose in default builds, move under a feature flag.

Browse source
This commit is contained in:
Igor Katson 2024-10-03 09:01:38 +01:00
parent b99a1e2b73
commit 990c5f866b
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
11 changed files with 45 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

8
crates/librqbit/src/torrent_state/live/mod.rs
View file

@ -525,7 +525,7 @@ impl TorrentStateLive {
let state = self;
loop {
let addr = peer_queue_rx.recv().await.context("torrent closed")?;
if state.torrent.options.disable_upload && state.is_finished_and_no_active_streams() {
if state.torrent.options.disable_upload() && state.is_finished_and_no_active_streams() {
debug!("ignoring peer {} as we are finished", addr);
state.peers.mark_peer_not_needed(addr);
continue;
@ -939,7 +939,7 @@ impl<'a> PeerConnectionHandler for &'a PeerHandler {
}
fn should_send_bitfield(&self) -> bool {
if self.state.torrent().options.disable_upload {
if self.state.torrent().options.disable_upload() {
return false;
}
@ -947,7 +947,7 @@ impl<'a> PeerConnectionHandler for &'a PeerHandler {
}
fn should_transmit_have(&self, id: ValidPieceIndex) -> bool {
if self.state.torrent.options.disable_upload {
if self.state.torrent.options.disable_upload() {
return false;
}
let have = self
@ -1187,7 +1187,7 @@ impl PeerHandler {
}
fn on_download_request(&self, request: Request) -> anyhow::Result<()> {
if self.state.torrent().options.disable_upload {
if self.state.torrent().options.disable_upload() {
anyhow::bail!("upload disabled, but peer requested a piece")
}

15
crates/librqbit/src/torrent_state/mod.rs
View file

@ -103,7 +103,20 @@ pub(crate) struct ManagedTorrentOptions {
pub allow_overwrite: bool,
pub output_folder: PathBuf,
pub disk_write_queue: Option<DiskWorkQueueSender>,
pub disable_upload: bool,
#[cfg(feature = "disable-upload")]
pub _disable_upload: bool,
}
impl ManagedTorrentOptions {
#[cfg(feature = "disable-upload")]
pub fn disable_upload(&self) -> bool {
self._disable_upload
}
#[cfg(not(feature = "disable-upload"))]
pub const fn disable_upload(&self) -> bool {
false
}
}
/// Common information about torrent shared among all possible states.