Add ulimit command (#11324)

# Description Add `ulimit` command to Nushell. Closes #9563 Closes #3976 Related pr #11246 Reference: https://github.com/fish-shell/fish-shell/blob/master/fish-rust/src/builtins/ulimit.rs https://github.com/mirror/busybox/blob/master/shell/shell_common.c#L529 # User-Facing Changes ``` nushell on  ulimit is 📦 v0.88.2 via 🦀 v1.72.1 [3/246] ❯ ulimit -a ╭────┬──────────────────────────────────────────────────────────────────────────┬───────────┬───────────╮ │ # │ description │ soft │ hard │ ├────┼──────────────────────────────────────────────────────────────────────────┼───────────┼───────────┤ │ 0 │ Maximum size of core files created (kB, -c) │ unlimited │ unlimited │ │ 1 │ Maximum size of a process's data segment (kB, -d) │ unlimited │ unlimited │ │ 2 │ Controls of maximum nice priority (-e) │ 0 │ 0 │ │ 3 │ Maximum size of files created by the shell (kB, -f) │ unlimited │ unlimited │ │ 4 │ Maximum number of pending signals (-i) │ 55273 │ 55273 │ │ 5 │ Maximum size that may be locked into memory (kB, -l) │ 8192 │ 8192 │ │ 6 │ Maximum resident set size (kB, -m) │ unlimited │ unlimited │ │ 7 │ Maximum number of open file descriptors (-n) │ 1024 │ 524288 │ │ 8 │ Maximum bytes in POSIX message queues (kB, -q) │ 800 │ 800 │ │ 9 │ Maximum realtime scheduling priority (-r) │ 0 │ 0 │ │ 10 │ Maximum stack size (kB, -s) │ 8192 │ unlimited │ │ 11 │ Maximum amount of CPU time in seconds (seconds, -t) │ unlimited │ unlimited │ │ 12 │ Maximum number of processes available to the current user (-u) │ 55273 │ 55273 │ │ 13 │ Maximum amount of virtual memory available to each process (kB, -v) │ unlimited │ unlimited │ │ 14 │ Maximum number of file locks (-x) │ unlimited │ unlimited │ │ 15 │ Maximum contiguous realtime CPU time (-y) │ unlimited │ unlimited │ ╰────┴──────────────────────────────────────────────────────────────────────────┴───────────┴───────────╯ nushell on  ulimit is 📦 v0.88.2 via 🦀 v1.72.1 ❯ ulimit -s ╭───┬─────────────────────────────┬──────┬───────────╮ │ # │ description │ soft │ hard │ ├───┼─────────────────────────────┼──────┼───────────┤ │ 0 │ Maximum stack size (kB, -s) │ 8192 │ unlimited │ ╰───┴─────────────────────────────┴──────┴───────────╯ nushell on  ulimit is 📦 v0.88.2 via 🦀 v1.72.1 ❯ ulimit -s 100 nushell on  ulimit is 📦 v0.88.2 via 🦀 v1.72.1 ❯ ulimit -s ╭───┬─────────────────────────────┬──────┬──────╮ │ # │ description │ soft │ hard │ ├───┼─────────────────────────────┼──────┼──────┤ │ 0 │ Maximum stack size (kB, -s) │ 100 │ 100 │ ╰───┴─────────────────────────────┴──────┴──────╯ nushell on  ulimit is 📦 v0.88.2 via 🦀 v1.72.1 ``` # Tests + Formatting - [x] add commands::ulimit::limit_set_soft1 - [x] add commands::ulimit::limit_set_soft2 - [x] add commands::ulimit::limit_set_hard1 - [x] add commands::ulimit::limit_set_hard2 - [x] add commands::ulimit::limit_set_invalid1 - [x] add commands::ulimit::limit_set_invalid2 - [x] `cargo fmt --all -- --check` to check standard code formatting (`cargo fmt --all` applies these changes) - [x] `cargo clippy --workspace -- -D warnings -D clippy::unwrap_used` to check that you're using the standard code style - [x] `cargo test --workspace` to check that all tests pass (on Windows make sure to [enable developer mode](https://learn.microsoft.com/en-us/windows/apps/get-started/developer-mode-features-and-debugging)) - [x] `cargo run -- -c "use std testing; testing run-tests --path crates/nu-std"` to run the tests for the standard library
2024-10-05 09:52:40 +02:00 · 2023-12-15 21:11:17 +08:00 · 2023-12-15 21:11:17 +08:00 · 84742275a1
commit 84742275a1
parent 92d968b8c8
6 changed files with 692 additions and 1 deletions
--- a/crates/nu-command/Cargo.toml
+++ b/crates/nu-command/Cargo.toml
@ -104,7 +104,7 @@ winreg = "0.52"
 [target.'cfg(unix)'.dependencies]
 libc = "0.2"
 umask = "2.1"
-nix = { version = "0.27", default-features = false, features = ["user"] }
+nix = { version = "0.27", default-features = false, features = ["user", "resource"] }

 [target.'cfg(all(unix, not(target_os = "macos"), not(target_os = "android"), not(target_os = "ios")))'.dependencies]
 procfs = "0.16.0"
--- a/crates/nu-command/src/default_context.rs
+++ b/crates/nu-command/src/default_context.rs
@ -233,6 +233,9 @@ pub fn add_shell_command_context(mut engine_state: EngineState) -> EngineState {
            Whoami,
        };

+        #[cfg(unix)]
+        bind_command! { ULimit };
+
        // Date
        bind_command! {
            Date,
--- a/crates/nu-command/src/platform/mod.rs
+++ b/crates/nu-command/src/platform/mod.rs
@ -7,6 +7,8 @@ mod is_terminal;
 mod kill;
 mod sleep;
 mod term_size;
+#[cfg(unix)]
+mod ulimit;
 mod whoami;

 pub use ansi::{Ansi, AnsiLink, AnsiStrip};
@ -20,4 +22,6 @@ pub use is_terminal::IsTerminal;
 pub use kill::Kill;
 pub use sleep::Sleep;
 pub use term_size::TermSize;
+#[cfg(unix)]
+pub use ulimit::ULimit;
 pub use whoami::Whoami;
--- a/crates/nu-command/src/platform/ulimit.rs
+++ b/crates/nu-command/src/platform/ulimit.rs
@ -0,0 +1,574 @@
+use nix::sys::resource::{rlim_t, Resource, RLIM_INFINITY};
+use nu_engine::CallExt;
+use nu_protocol::{
+    ast::Call,
+    engine::{Command, EngineState, Stack},
+    Category, Example, IntoPipelineData, PipelineData, Record, ShellError, Signature, Span,
+    Spanned, SyntaxShape, Type, Value,
+};
+use once_cell::sync::Lazy;
+
+/// An object contains resource related parameters
+struct ResourceInfo<'a> {
+    name: &'a str,
+    desc: &'a str,
+    flag: char,
+    multiplier: rlim_t,
+    resource: Resource,
+}
+
+impl<'a> ResourceInfo<'a> {
+    /// Create a `ResourceInfo` object
+    fn new(
+        name: &'a str,
+        desc: &'a str,
+        flag: char,
+        multiplier: rlim_t,
+        resource: Resource,
+    ) -> Self {
+        Self {
+            name,
+            desc,
+            flag,
+            multiplier,
+            resource,
+        }
+    }
+
+    /// Get unit
+    fn get_unit(&self) -> &str {
+        if self.resource == Resource::RLIMIT_CPU {
+            "(seconds, "
+        } else if self.multiplier == 1 {
+            "("
+        } else {
+            "(kB, "
+        }
+    }
+}
+
+impl<'a> Default for ResourceInfo<'a> {
+    fn default() -> Self {
+        Self {
+            name: "file-size",
+            desc: "Maximum size of files created by the shell",
+            flag: 'f',
+            multiplier: 1024,
+            resource: Resource::RLIMIT_FSIZE,
+        }
+    }
+}
+
+static RESOURCE_ARRAY: Lazy<Vec<ResourceInfo>> = Lazy::new(|| {
+    let resources = [
+        #[cfg(any(target_os = "freebsd", target_os = "dragonfly"))]
+        (
+            "socket-buffers",
+            "Maximum size of socket buffers",
+            'b',
+            1024,
+            Resource::RLIMIT_SBSIZE,
+        ),
+        (
+            "core-size",
+            "Maximum size of core files created",
+            'c',
+            1024,
+            Resource::RLIMIT_CORE,
+        ),
+        (
+            "data-size",
+            "Maximum size of a process's data segment",
+            'd',
+            1024,
+            Resource::RLIMIT_DATA,
+        ),
+        #[cfg(any(target_os = "android", target_os = "linux"))]
+        (
+            "nice",
+            "Controls of maximum nice priority",
+            'e',
+            1,
+            Resource::RLIMIT_NICE,
+        ),
+        (
+            "file-size",
+            "Maximum size of files created by the shell",
+            'f',
+            1024,
+            Resource::RLIMIT_FSIZE,
+        ),
+        #[cfg(any(target_os = "android", target_os = "linux"))]
+        (
+            "pending-signals",
+            "Maximum number of pending signals",
+            'i',
+            1,
+            Resource::RLIMIT_SIGPENDING,
+        ),
+        #[cfg(any(
+            target_os = "android",
+            target_os = "freebsd",
+            target_os = "openbsd",
+            target_os = "linux",
+            target_os = "netbsd"
+        ))]
+        (
+            "lock-size",
+            "Maximum size that may be locked into memory",
+            'l',
+            1024,
+            Resource::RLIMIT_MEMLOCK,
+        ),
+        #[cfg(any(
+            target_os = "android",
+            target_os = "freebsd",
+            target_os = "netbsd",
+            target_os = "openbsd",
+            target_os = "linux",
+            target_os = "aix",
+        ))]
+        (
+            "resident-set-size",
+            "Maximum resident set size",
+            'm',
+            1024,
+            Resource::RLIMIT_RSS,
+        ),
+        (
+            "file-descriptor-count",
+            "Maximum number of open file descriptors",
+            'n',
+            1,
+            Resource::RLIMIT_NOFILE,
+        ),
+        #[cfg(any(target_os = "android", target_os = "linux"))]
+        (
+            "queue-size",
+            "Maximum bytes in POSIX message queues",
+            'q',
+            1024,
+            Resource::RLIMIT_MSGQUEUE,
+        ),
+        #[cfg(any(target_os = "android", target_os = "linux"))]
+        (
+            "realtime-priority",
+            "Maximum realtime scheduling priority",
+            'r',
+            1,
+            Resource::RLIMIT_RTPRIO,
+        ),
+        (
+            "stack-size",
+            "Maximum stack size",
+            's',
+            1024,
+            Resource::RLIMIT_STACK,
+        ),
+        (
+            "cpu-time",
+            "Maximum amount of CPU time in seconds",
+            't',
+            1,
+            Resource::RLIMIT_CPU,
+        ),
+        #[cfg(any(
+            target_os = "android",
+            target_os = "freebsd",
+            target_os = "netbsd",
+            target_os = "openbsd",
+            target_os = "linux",
+            target_os = "aix",
+        ))]
+        (
+            "process-count",
+            "Maximum number of processes available to the current user",
+            'u',
+            1,
+            Resource::RLIMIT_NPROC,
+        ),
+        #[cfg(not(any(target_os = "freebsd", target_os = "netbsd", target_os = "openbsd")))]
+        (
+            "virtual-memory-size",
+            "Maximum amount of virtual memory available to each process",
+            'v',
+            1024,
+            Resource::RLIMIT_AS,
+        ),
+        #[cfg(target_os = "freebsd")]
+        (
+            "swap-size",
+            "Maximum swap space",
+            'w',
+            1024,
+            Resource::RLIMIT_SWAP,
+        ),
+        #[cfg(any(target_os = "android", target_os = "linux"))]
+        (
+            "file-locks",
+            "Maximum number of file locks",
+            'x',
+            1,
+            Resource::RLIMIT_LOCKS,
+        ),
+        #[cfg(target_os = "linux")]
+        (
+            "realtime-maxtime",
+            "Maximum contiguous realtime CPU time",
+            'y',
+            1,
+            Resource::RLIMIT_RTTIME,
+        ),
+        #[cfg(target_os = "freebsd")]
+        (
+            "kernel-queues",
+            "Maximum number of kqueues",
+            'K',
+            1,
+            Resource::RLIMIT_KQUEUES,
+        ),
+        #[cfg(target_os = "freebsd")]
+        (
+            "ptys",
+            "Maximum number of pseudo-terminals",
+            'P',
+            1,
+            Resource::RLIMIT_NPTY,
+        ),
+    ];
+
+    let mut resource_array = Vec::new();
+    for (name, desc, flag, multiplier, res) in resources {
+        resource_array.push(ResourceInfo::new(name, desc, flag, multiplier, res));
+    }
+
+    resource_array
+});
+
+/// Convert `rlim_t` to `Value` representation
+fn limit_to_value(limit: rlim_t, multiplier: rlim_t, span: Span) -> Result<Value, ShellError> {
+    if limit == RLIM_INFINITY {
+        return Ok(Value::string("unlimited", span));
+    }
+
+    let val = match i64::try_from(limit / multiplier) {
+        Ok(v) => v,
+        Err(e) => {
+            return Err(ShellError::CantConvert {
+                to_type: "i64".into(),
+                from_type: "rlim_t".into(),
+                span,
+                help: Some(e.to_string()),
+            });
+        }
+    };
+
+    Ok(Value::int(val, span))
+}
+
+/// Get maximum length of all flag descriptions
+fn max_desc_len(call: &Call, print_all: bool) -> usize {
+    let mut desc_len = 0;
+    let mut unit_len = 0;
+
+    for res in RESOURCE_ARRAY.iter() {
+        if !print_all && !call.has_flag(res.name) {
+            continue;
+        }
+
+        desc_len = res.desc.len().max(desc_len);
+        unit_len = res.get_unit().len().max(unit_len);
+    }
+
+    // Use `RLIMIT_FSIZE` limit if no resource flag provided.
+    if desc_len == 0 {
+        let res = ResourceInfo::default();
+        desc_len = res.desc.len().max(desc_len);
+        unit_len = res.get_unit().len().max(unit_len);
+    }
+
+    // desc.len() + unit.len() + '-X)'.len()
+    desc_len + unit_len + 3
+}
+
+/// Fill `ResourceInfo` to the record entry
+fn fill_record(
+    res: &ResourceInfo,
+    max_len: usize,
+    soft: bool,
+    hard: bool,
+    span: Span,
+) -> Result<Record, ShellError> {
+    let mut record = Record::new();
+    let mut desc = String::new();
+
+    desc.push_str(res.desc);
+
+    debug_assert!(res.desc.len() + res.get_unit().len() + 3 <= max_len);
+    let width = max_len - res.desc.len() - res.get_unit().len() - 3;
+    if width == 0 {
+        desc.push_str(format!(" {}-{})", res.get_unit(), res.flag).as_str());
+    } else {
+        desc.push_str(format!("{:>width$} {}-{})", ' ', res.get_unit(), res.flag).as_str());
+    }
+
+    record.push("description", Value::string(desc, span));
+
+    let (soft_limit, hard_limit) = getrlimit(res.resource)?;
+
+    if soft {
+        let soft_limit = limit_to_value(soft_limit, res.multiplier, span)?;
+        record.push("soft", soft_limit);
+    }
+
+    if hard {
+        let hard_limit = limit_to_value(hard_limit, res.multiplier, span)?;
+        record.push("hard", hard_limit);
+    }
+
+    Ok(record)
+}
+
+/// Set limits
+fn set_limits(
+    spanned_limit: &Spanned<String>,
+    res: &ResourceInfo,
+    soft: bool,
+    hard: bool,
+) -> Result<(), ShellError> {
+    let (mut soft_limit, mut hard_limit) = getrlimit(res.resource)?;
+    let new_limit = parse_limit(spanned_limit, res.multiplier, soft, soft_limit, hard_limit)?;
+
+    if hard {
+        hard_limit = new_limit;
+    }
+
+    if soft {
+        soft_limit = new_limit;
+
+        // Do not attempt to set the soft limit higher than the hard limit.
+        if (new_limit > hard_limit || new_limit == RLIM_INFINITY) && hard_limit != RLIM_INFINITY {
+            soft_limit = hard_limit;
+        }
+    }
+
+    setrlimit(res.resource, soft_limit, hard_limit)
+}
+
+/// Print limits
+fn print_limits(
+    call: &Call,
+    print_all: bool,
+    soft: bool,
+    hard: bool,
+) -> Result<PipelineData, ShellError> {
+    let mut output = Vec::new();
+    let mut print_default_limit = true;
+    let max_len = max_desc_len(call, print_all);
+
+    for res in RESOURCE_ARRAY.iter() {
+        if !print_all {
+            // Print specified limit.
+            if !call.has_flag(res.name) {
+                continue;
+            }
+        }
+
+        let record = fill_record(res, max_len, soft, hard, call.head)?;
+        output.push(Value::record(record, call.head));
+
+        if print_default_limit {
+            print_default_limit = false;
+        }
+    }
+
+    // Print `RLIMIT_FSIZE` limit if no resource flag provided.
+    if print_default_limit {
+        let res = ResourceInfo::default();
+        let record = fill_record(&res, max_len, soft, hard, call.head)?;
+        output.push(Value::record(record, call.head));
+    }
+
+    Ok(Value::list(output, call.head).into_pipeline_data())
+}
+
+/// Wrap `nix::sys::resource::getrlimit`
+fn setrlimit(res: Resource, soft_limit: rlim_t, hard_limit: rlim_t) -> Result<(), ShellError> {
+    nix::sys::resource::setrlimit(res, soft_limit, hard_limit).map_err(|e| {
+        ShellError::GenericError {
+            error: e.to_string(),
+            msg: String::new(),
+            span: None,
+            help: None,
+            inner: vec![],
+        }
+    })
+}
+
+/// Wrap `nix::sys::resource::setrlimit`
+fn getrlimit(res: Resource) -> Result<(rlim_t, rlim_t), ShellError> {
+    nix::sys::resource::getrlimit(res).map_err(|e| ShellError::GenericError {
+        error: e.to_string(),
+        msg: String::new(),
+        span: None,
+        help: None,
+        inner: vec![],
+    })
+}
+
+/// Parse user input
+fn parse_limit(
+    spanned_limit: &Spanned<String>,
+    multiplier: rlim_t,
+    soft: bool,
+    soft_limit: rlim_t,
+    hard_limit: rlim_t,
+) -> Result<rlim_t, ShellError> {
+    let limit = &spanned_limit.item;
+    let span = spanned_limit.span;
+
+    if limit.eq("unlimited") {
+        Ok(RLIM_INFINITY)
+    } else if limit.eq("soft") {
+        if soft {
+            Ok(hard_limit)
+        } else {
+            Ok(soft_limit)
+        }
+    } else if limit.eq("hard") {
+        Ok(hard_limit)
+    } else {
+        let v = limit
+            .parse::<rlim_t>()
+            .map_err(|e| ShellError::CantConvert {
+                to_type: "rlim_t".into(),
+                from_type: "String".into(),
+                span,
+                help: Some(e.to_string()),
+            })?;
+
+        let (value, overflow) = v.overflowing_mul(multiplier);
+        if overflow {
+            return Err(ShellError::OperatorOverflow {
+                msg: "Multiple overflow".into(),
+                span,
+                help: String::new(),
+            });
+        } else {
+            Ok(value)
+        }
+    }
+}
+
+#[derive(Clone)]
+pub struct ULimit;
+
+impl Command for ULimit {
+    fn name(&self) -> &str {
+        "ulimit"
+    }
+
+    fn usage(&self) -> &str {
+        "Set or get resource usage limits."
+    }
+
+    fn signature(&self) -> Signature {
+        let mut sig = Signature::build("ulimit")
+            .input_output_types(vec![
+                (Type::Nothing, Type::Table(vec![])),
+                (Type::Nothing, Type::Nothing),
+            ])
+            .switch("soft", "Sets soft resource limit", Some('S'))
+            .switch("hard", "Sets hard resource limit", Some('H'))
+            .switch("all", "Prints all current limits", Some('a'))
+            .optional("limit", SyntaxShape::String, "Limit value.")
+            .category(Category::Platform);
+
+        for res in RESOURCE_ARRAY.iter() {
+            sig = sig.switch(res.name, res.desc, Some(res.flag));
+        }
+
+        sig
+    }
+
+    fn run(
+        &self,
+        engine_state: &EngineState,
+        stack: &mut Stack,
+        call: &Call,
+        _input: PipelineData,
+    ) -> Result<PipelineData, ShellError> {
+        let mut soft = call.has_flag("soft");
+        let mut hard = call.has_flag("hard");
+        let all = call.has_flag("all");
+
+        if !hard && !soft {
+            // Set both hard and soft limits if neither was specified.
+            hard = true;
+            soft = true;
+        }
+
+        if let Some(spanned_limit) = call.opt::<Spanned<String>>(engine_state, stack, 0)? {
+            let mut set_default_limit = true;
+
+            for res in RESOURCE_ARRAY.iter() {
+                if call.has_flag(res.name) {
+                    set_limits(&spanned_limit, res, soft, hard)?;
+
+                    if set_default_limit {
+                        set_default_limit = false;
+                    }
+                }
+            }
+
+            // Set `RLIMIT_FSIZE` limit if no resource flag provided.
+            if set_default_limit {
+                let res = ResourceInfo::default();
+                set_limits(&spanned_limit, &res, hard, soft)?;
+            }
+
+            Ok(PipelineData::Empty)
+        } else {
+            print_limits(call, all, soft, hard)
+        }
+    }
+
+    fn examples(&self) -> Vec<Example> {
+        vec![
+            Example {
+                description: "Print all current limits",
+                example: "ulimit -a",
+                result: None,
+            },
+            Example {
+                description: "Print specified limits",
+                example: "ulimit --core-size --data-size --file-size",
+                result: None,
+            },
+            Example {
+                description: "Set limit",
+                example: "ulimit --core-size 102400",
+                result: None,
+            },
+            Example {
+                description: "Set stack size soft limit",
+                example: "ulimit -s -S 10240",
+                result: None,
+            },
+            Example {
+                description: "Set virtual memory size hard limit",
+                example: "ulimit -v -H 10240",
+                result: None,
+            },
+            Example {
+                description: "Set core size limit to unlimited",
+                example: "ulimit -c unlimited",
+                result: None,
+            },
+        ]
+    }
+
+    fn search_terms(&self) -> Vec<&str> {
+        vec!["resource", "limits"]
+    }
+}
--- a/crates/nu-command/tests/commands/mod.rs
+++ b/crates/nu-command/tests/commands/mod.rs
@ -106,6 +106,8 @@ mod touch;
 mod transpose;
 mod try_;
 mod ucp;
+#[cfg(unix)]
+mod ulimit;
 mod umkdir;
 mod uniq;
 mod uniq_by;
--- a/crates/nu-command/tests/commands/ulimit.rs
+++ b/crates/nu-command/tests/commands/ulimit.rs
@ -0,0 +1,108 @@
+use nu_test_support::playground::Playground;
+use nu_test_support::{nu, pipeline};
+
+#[test]
+fn limit_set_soft1() {
+    Playground::setup("limit_set_soft1", |dirs, _sandbox| {
+        let actual = nu!(
+            cwd: dirs.test(), pipeline(
+            "
+                let soft = (ulimit -s | first | get soft | into string);
+                ulimit -s -H $soft;
+                let hard = (ulimit -s | first | get hard | into string);
+                $soft == $hard
+            "
+        ));
+
+        assert!(actual.out.contains("true"));
+    });
+}
+
+#[test]
+fn limit_set_soft2() {
+    Playground::setup("limit_set_soft2", |dirs, _sandbox| {
+        let actual = nu!(
+            cwd: dirs.test(), pipeline(
+            "
+                let soft = (ulimit -s | first | get soft | into string);
+                ulimit -s -H soft;
+                let hard = (ulimit -s | first | get hard | into string);
+                $soft == $hard
+            "
+        ));
+
+        assert!(actual.out.contains("true"));
+    });
+}
+
+#[test]
+fn limit_set_hard1() {
+    Playground::setup("limit_set_hard1", |dirs, _sandbox| {
+        let actual = nu!(
+            cwd: dirs.test(), pipeline(
+            "
+                let hard = (ulimit -s | first | get hard | into string);
+                ulimit -s $hard;
+                let soft = (ulimit -s | first | get soft | into string);
+                $soft == $hard
+           "
+        ));
+
+        assert!(actual.out.contains("true"));
+    });
+}
+
+#[test]
+fn limit_set_hard2() {
+    Playground::setup("limit_set_hard2", |dirs, _sandbox| {
+        let actual = nu!(
+            cwd: dirs.test(), pipeline(
+            "
+                let hard = (ulimit -s | first | get hard | into string);
+                ulimit -s hard;
+                let soft = (ulimit -s | first | get soft | into string);
+                $soft == $hard
+            "
+        ));
+
+        assert!(actual.out.contains("true"));
+    });
+}
+
+#[test]
+fn limit_set_invalid1() {
+    Playground::setup("limit_set_invalid1", |dirs, _sandbox| {
+        let actual = nu!(
+        cwd: dirs.test(), pipeline(
+        "
+            let hard = (ulimit -s | first | get hard);
+            match $hard {
+                \"unlimited\" => { echo \"unlimited\" },
+                $x => {
+                    let new = ($x + 1 | into string);
+                    ulimit -s $new
+                }
+            }
+        "
+        ));
+
+        assert!(
+            actual.out.contains("unlimited")
+                || actual.err.contains("EPERM: Operation not permitted")
+        );
+    });
+}
+
+#[test]
+fn limit_set_invalid2() {
+    Playground::setup("limit_set_invalid2", |dirs, _sandbox| {
+        let actual = nu!(
+            cwd: dirs.test(),
+            "
+                ulimit -c abcd
+            "
+        );
+
+        assert!(actual.err.contains("Can't convert to rlim_t."));
+    });
+}