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

disable umask on windows

Browse source
This commit is contained in:
Igor Katson 2024-08-26 14:04:34 +01:00
parent 016d759512
commit 10e8db4c9a
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
crates/rqbit/src/main.rs
View file

@ -28,6 +28,7 @@ enum LogLevel {
Error, Error,
} }
#[cfg(not(target_os = "windows"))]
fn parse_umask(value: &str) -> anyhow::Result<libc::mode_t> { fn parse_umask(value: &str) -> anyhow::Result<libc::mode_t> {
fn parse_oct_digit(d: u8) -> Option<libc::mode_t> { fn parse_oct_digit(d: u8) -> Option<libc::mode_t> {
Some(match d { Some(match d {
@ -194,6 +195,7 @@ struct Opts {
/// This will affect the file mode of created files. /// This will affect the file mode of created files.
/// ///
/// Read more at https://man7.org/linux/man-pages/man2/umask.2.html /// Read more at https://man7.org/linux/man-pages/man2/umask.2.html
#[cfg(not(target_os = "windows"))]
#[arg(long, env = "RQBIT_UMASK", value_parser=parse_umask)] #[arg(long, env = "RQBIT_UMASK", value_parser=parse_umask)]
umask: Option<libc::mode_t>, umask: Option<libc::mode_t>,
} }
@ -328,6 +330,7 @@ fn _start_deadlock_detector_thread() {
fn main() -> anyhow::Result<()> { fn main() -> anyhow::Result<()> {
let opts = Opts::parse(); let opts = Opts::parse();
#[cfg(not(target_os = "windows"))]
if let Some(umask) = opts.umask { if let Some(umask) = opts.umask {
unsafe { libc::umask(umask) }; unsafe { libc::umask(umask) };
} }
@ -755,10 +758,10 @@ async fn async_main(opts: Opts) -> anyhow::Result<()> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use crate::parse_umask; #[cfg(not(target_os = "windows"))]
#[test] #[test]
fn test_parse_umask() { fn test_parse_umask() {
use crate::parse_umask;
let range = b'0'..=b'7'; let range = b'0'..=b'7';
for d0 in range.clone() { for d0 in range.clone() {
for d1 in range.clone() { for d1 in range.clone() {