From a1832770a75898bd2d46059710d4fe18046f1541 Mon Sep 17 00:00:00 2001
From: zhanghuanhuan <zhanghuanhuan@uniontech.com>
Date: Fri, 10 Nov 2023 17:12:26 +0800
Subject: [PATCH] add uidset func for variables file

---
 bash-5.1/r_variables/src/lib.rs | 37 +++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/bash-5.1/r_variables/src/lib.rs b/bash-5.1/r_variables/src/lib.rs
index 8f09f8c..267d8fa 100644
--- a/bash-5.1/r_variables/src/lib.rs
+++ b/bash-5.1/r_variables/src/lib.rs
@@ -452,3 +452,40 @@ pub unsafe extern "C" fn set_ppid() {
 
 }
 
+unsafe extern "C" fn uidset() {
+    // INT_STRLEN_BOUND(uid_t) + 1
+    let mut buff: [libc::c_char; (INT_STRLEN_BOUND!(uid_t) +1) as usize] = [0;(INT_STRLEN_BOUND!(uid_t) +1) as usize ];
+    let mut b: *mut libc::c_char = 0 as *mut libc::c_char;
+    let mut v: *mut SHELL_VAR = 0 as *mut SHELL_VAR;
+
+    b = inttostr(
+        current_user.uid as intmax_t,
+        buff.as_mut_ptr(),
+        std::mem::size_of::<[libc::c_char; (INT_STRLEN_BOUND!(uid_t) +1) as usize]>() as libc::c_ulong as size_t
+    );
+    v = find_variable(b"UID\0" as *const u8 as *const libc::c_char);
+    if v.is_null() {
+        v = bind_variable(
+            b"UID\0" as *const u8 as *const libc::c_char,
+            b,
+            0 as libc::c_int,
+        );
+        VSETATTR!(v,(att_readonly | att_integer) as libc::c_int);
+    }
+    if current_user.euid != current_user.uid {
+        b = inttostr(
+            current_user.euid as intmax_t,
+            buff.as_mut_ptr(),
+            std::mem::size_of::< [libc::c_char; (INT_STRLEN_BOUND!(uid_t) +1) as usize]>() as libc::c_ulong  as size_t
+        );
+    }
+    v = find_variable(b"EUID\0" as *const u8 as *const libc::c_char);
+    if v.is_null() {
+        v = bind_variable(
+            b"EUID\0" as *const u8 as *const libc::c_char,
+            b,
+            0 as libc::c_int,
+        );
+        VSETATTR!(v,(att_readonly | att_integer));
+    }
+}
-- 
Gitee