Add RLIMIT_AS and RLIMIT_DATA support to prlimit64 (#984)

* add prlimit resource

* add prlimit resource

* add prlimit resource

* update test

* update constants

Author: rennergade

src/rawposix/src/sys_calls.rs

@@ -18,11 +18,12 @@ use std::time::Duration;
 use sysdefs::constants::err_const::{get_errno, handle_errno, syscall_error, Errno, VERBOSE};
 use sysdefs::constants::fs_const::{STDERR_FILENO, STDIN_FILENO, STDOUT_FILENO};
 use sysdefs::constants::lind_platform_const::{
-    RAWPOSIX_CAGEID, UNUSED_ARG, UNUSED_ID, UNUSED_NAME, WASMTIME_CAGEID,
+    MAX_LINEAR_MEMORY_SIZE, RAWPOSIX_CAGEID, UNUSED_ARG, UNUSED_ID, UNUSED_NAME, WASMTIME_CAGEID,
 };
 use sysdefs::constants::sys_const::{
-    DEFAULT_GID, DEFAULT_UID, EXIT_SUCCESS, ITIMER_REAL, SIGCHLD, SIGKILL, SIGSTOP, SIG_BLOCK,
-    SIG_SETMASK, SIG_UNBLOCK, WNOHANG,
+    DEFAULT_GID, DEFAULT_UID, EXIT_SUCCESS, ITIMER_REAL, RLIMIT_AS, RLIMIT_DATA, RLIMIT_NOFILE,
+    RLIMIT_RSS, RLIMIT_STACK, SIGCHLD, SIGKILL, SIGSTOP, SIG_BLOCK, SIG_SETMASK, SIG_UNBLOCK,
+    WNOHANG,
 };
 use sysdefs::data::fs_struct::{ITimerVal, Rlimit, SigactionStruct};
 use sysdefs::logging::lind_debug_panic;
@@ -1196,16 +1197,18 @@ pub extern "C" fn prlimit64_syscall(
             Err(e) => return syscall_error(e, "prlimit64", "bad address"),
         };
         match resource {
-            3 => {
-                // RLIMIT_STACK: 8MiB
+            RLIMIT_STACK => {
                 old_limit.rlim_cur = 8 * 1024 * 1024;
                 old_limit.rlim_max = 8 * 1024 * 1024;
             }
-            7 => {
-                // RLIMIT_NOFILE: 1024
+            RLIMIT_NOFILE => {
                 old_limit.rlim_cur = 1024;
                 old_limit.rlim_max = 1024;
             }
+            RLIMIT_DATA | RLIMIT_RSS | RLIMIT_AS => {
+                old_limit.rlim_cur = MAX_LINEAR_MEMORY_SIZE as u32;
+                old_limit.rlim_max = MAX_LINEAR_MEMORY_SIZE as u32;
+            }
             _ => {
                 lind_debug_panic(&format!("prlimit64: unsupported resource {}", resource));
                 old_limit.rlim_cur = 0;

src/sysdefs/src/constants/sys_const.rs

@@ -23,9 +23,17 @@ pub const NOFILE_MAX: u64 = 4 * 1024; // Hard limit for number of open files
 pub const STACK_CUR: u64 = 8192 * 1024; // Soft limit for stack size (8MB)
 pub const STACK_MAX: u64 = 1 << 32; // Hard limit for stack size (4GB)
 
-// Resource identifiers
-pub const RLIMIT_STACK: u64 = 0; // Limit type for stack size
-pub const RLIMIT_NOFILE: u64 = 1; // Limit type for number of files
+// Resource identifiers (must match glibc bits/resource.h)
+pub const RLIMIT_CPU: u32 = 0;
+pub const RLIMIT_FSIZE: u32 = 1;
+pub const RLIMIT_DATA: u32 = 2;
+pub const RLIMIT_STACK: u32 = 3;
+pub const RLIMIT_CORE: u32 = 4;
+pub const RLIMIT_RSS: u32 = 5;
+pub const RLIMIT_NPROC: u32 = 6;
+pub const RLIMIT_NOFILE: u32 = 7;
+pub const RLIMIT_MEMLOCK: u32 = 8;
+pub const RLIMIT_AS: u32 = 9;
 
 // ===== Process Exit Status =====
 // Source: <stdlib.h> and POSIX standard

tests/unit-tests/file_tests/deterministic/prlimit64.c

@@ -16,6 +16,18 @@ int main(void) {
         return 1;
     }
 
+    ret = getrlimit(RLIMIT_AS, &rl);
+    if (ret != 0 || rl.rlim_cur == 0) {
+        printf("FAIL\n");
+        return 1;
+    }
+
+    ret = getrlimit(RLIMIT_DATA, &rl);
+    if (ret != 0 || rl.rlim_cur == 0) {
+        printf("FAIL\n");
+        return 1;
+    }
+
     printf("All prlimit64 tests passed\n");
     return 0;
 }