Add os.setreuid and os.setresuid methods

Toan Phan · web-flow · commit 6266081d9a87 · 2020-04-28T13:35:57.000+09:00
diff --git a/tests/snippets/stdlib_os.py b/tests/snippets/stdlib_os.py
@@ -293,6 +293,8 @@ def __exit__(self, exc_type, exc_val, exc_tb):
         assert_raises(PermissionError, lambda: os.setpgid(os.getpid(), 42))
         assert_raises(PermissionError, lambda: os.setuid(42))
         assert_raises(PermissionError, lambda: os.seteuid(42))
+        assert_raises(PermissionError, lambda: os.setreuid(42, 42))
+        assert_raises(PermissionError, lambda: os.setresuid(42, 42, 42))
 
     # pty
     a, b = os.openpty()
diff --git a/vm/src/stdlib/os.rs b/vm/src/stdlib/os.rs
@@ -1247,6 +1247,28 @@ fn os_seteuid(euid: u32, vm: &VirtualMachine) -> PyResult<()> {
     unistd::seteuid(Uid::from_raw(euid)).map_err(|err| convert_nix_error(vm, err))
 }
 
+#[cfg(all(unix, not(target_os = "redox")))]
+fn os_setreuid(ruid: u32, euid: u32, vm: &VirtualMachine) -> PyResult<()> {
+    unistd::setuid(Uid::from_raw(ruid)).map_err(|err| convert_nix_error(vm, err))?;
+    unistd::seteuid(Uid::from_raw(euid)).map_err(|err| convert_nix_error(vm, err))
+}
+
+// cfg from nix
+#[cfg(any(
+    target_os = "android",
+    target_os = "freebsd",
+    target_os = "linux",
+    target_os = "openbsd"
+))]
+fn os_setresuid(ruid: u32, euid: u32, suid: u32, vm: &VirtualMachine) -> PyResult<()> {
+    unistd::setresuid(
+        Uid::from_raw(ruid),
+        Uid::from_raw(euid),
+        Uid::from_raw(suid),
+    )
+    .map_err(|err| convert_nix_error(vm, err))
+}
+
 #[cfg(all(unix, not(target_os = "redox")))]
 pub fn os_openpty(vm: &VirtualMachine) -> PyResult {
     match openpty(None, None) {
@@ -1622,12 +1644,24 @@ fn extend_module_platform_specific(vm: &VirtualMachine, module: &PyObjectRef) {
         "setsid" => ctx.new_function(os_setsid),
         "setegid" => ctx.new_function(os_setegid),
         "seteuid" => ctx.new_function(os_seteuid),
+        "setreuid" => ctx.new_function(os_setreuid),
         "openpty" => ctx.new_function(os_openpty),
         "O_DSYNC" => ctx.new_int(libc::O_DSYNC),
         "O_NDELAY" => ctx.new_int(libc::O_NDELAY),
         "O_NOCTTY" => ctx.new_int(libc::O_NOCTTY),
     });
 
+    // cfg taken from nix
+    #[cfg(any(
+        target_os = "android",
+        target_os = "freebsd",
+        target_os = "linux",
+        target_os = "openbsd"
+    ))]
+    extend_module!(vm, module, {
+        "setresuid" => ctx.new_function(os_setresuid),
+    });
+
     // cfg taken from nix
     #[cfg(any(
         target_os = "dragonfly",
