h/fish-shell
1
Fork 0
mirror of https://github.com/fish-shell/fish-shell.git synced 2026-04-19 06:31:13 -03:00
Code Issues Packages Projects Releases Wiki Activity

builtin ulimit: extract function

Browse Source
This commit is contained in:
Johannes Altmanninger
2025-10-25 18:08:07 +02:00
parent 0b6afbd17b
commit 3f9d8db5b7
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/builtins/ulimit.rs
View File

@@ -2,6 +2,7 @@
use libc::{RLIM_INFINITY, c_uint, rlim_t};
use nix::errno::Errno;
use nix::sys::resource::Resource as ResourceEnum;
use once_cell::sync::Lazy;
use crate::fallback::{fish_wcswidth, wcscasecmp};
@@ -85,22 +86,23 @@ macro_rules! define_on {
define_on!(NTHR, RLIMIT_NTHR; "netbsd");
}
fn convert_resource(resource: c_uint) -> ResourceEnum {
let resource: i32 = resource.try_into().unwrap();
use std::mem::{size_of, transmute};
// Resource is #[repr(i32)] so this is ok
const _: () = assert!(size_of::<c_uint>() == size_of::<ResourceEnum>());
unsafe { transmute(resource) }
}
/// Calls getrlimit.
fn getrlimit(resource: c_uint) -> Option<(rlim_t, rlim_t)> {
let resource: i32 = resource.try_into().unwrap();
// Resource is #[repr(i32)] so this is ok
let resource = unsafe { std::mem::transmute::<i32, nix::sys::resource::Resource>(resource) };
nix::sys::resource::getrlimit(resource)
nix::sys::resource::getrlimit(convert_resource(resource))
.map_err(|_| perror("getrlimit"))
.ok()
}
fn setrlimit(resource: c_uint, rlim_cur: rlim_t, rlim_max: rlim_t) -> Result<(), Errno> {
let resource: i32 = resource.try_into().unwrap();
// Resource is #[repr(i32)] so this is ok
let resource = unsafe { std::mem::transmute::<i32, nix::sys::resource::Resource>(resource) };
nix::sys::resource::setrlimit(resource, rlim_cur, rlim_max)
nix::sys::resource::setrlimit(convert_resource(resource), rlim_cur, rlim_max)
}
/// Print the value of the specified resource limit.