crates/radicle/src/io.rs — heartwood

rad:z3gqcJUoA1n9HaHKufZs5FCSGazv5

Radicle Heartwood Protocol & Stack

.. canonical cob git identity logger node profile storage test canonical.rs cli.rs cob.rs collections.rs explorer.rs git.rs identity.rs io.rs lib.rs logger.rs node.rs profile.rs rad.rs schemars_ext.rs serde_ext.rs sql.rs storage.rs test.rs version.rs web.rs

use std::io;

#[cfg(unix)]
use libc::{RLIMIT_NOFILE, getrlimit, rlim_t, rlimit, setrlimit};

#[cfg(unix)]
type Int = rlim_t;

#[cfg(not(unix))]
#[inline]
pub fn set_file_limit<T>(_: T) -> io::Result<()> {
    Err(io::Error::new(
        io::ErrorKind::Unsupported,
        "setting file limits is not supported on this platform",
    ))
}

/// Sets the open file limit to the given value, or the maximum allowed value.
#[cfg(unix)]
pub fn set_file_limit<N>(n: N) -> io::Result<Int>
where
    N: Copy + std::fmt::Display,
    Int: TryFrom<N>,
{
    let Ok(n) = Int::try_from(n) else {
        return Err(io::Error::new(
            io::ErrorKind::InvalidInput,
            format!("invalid file limit '{n}'"),
        ));
    };
    let mut rlim = rlimit {
        rlim_cur: 0, // Initial soft limit value
        rlim_max: 0, // Initial hard limit value
    };
    // Get the current limits.
    unsafe {
        if getrlimit(RLIMIT_NOFILE, &mut rlim) != 0 {
            return Err(io::Error::last_os_error());
        }
    }
    if rlim.rlim_cur >= n {
        return Ok(rlim.rlim_cur);
    }
    // Set the soft limit to the given value, up to the hard limit.
    rlim.rlim_cur = n.min(rlim.rlim_max);
    unsafe {
        if setrlimit(RLIMIT_NOFILE, &rlim as *const rlimit) != 0 {
            return Err(io::Error::last_os_error());
        }
    }
    Ok(rlim.rlim_cur)
}