Add safe rules for lseek,ftruncate,isatty,listen

Author: lucic71

rules/eventfd/tgt_refcount.rs

@@ -0,0 +1,33 @@
+// Copyright (c) 2022-present INESC-ID.
+// Distributed under the MIT license that can be found in the LICENSE file.
+
+use libcc2rs::*;
+
+#[cfg(target_os = "linux")]
+fn f1() -> i32 {
+    ::libc::EFD_CLOEXEC
+}
+
+#[cfg(target_os = "linux")]
+fn f2() -> i32 {
+    ::libc::EFD_NONBLOCK
+}
+
+#[cfg(target_os = "linux")]
+fn f3() -> i32 {
+    ::libc::EFD_SEMAPHORE
+}
+
+#[cfg(target_os = "linux")]
+fn f4(a0: u32, a1: i32) -> i32 {
+    match nix::sys::eventfd::EventFd::from_value_and_flags(
+        a0,
+        nix::sys::eventfd::EfdFlags::from_bits_truncate(a1),
+    ) {
+        Ok(__efd) => FdRegistry::register(std::os::fd::OwnedFd::from(__efd)),
+        Err(__e) => {
+            libcc2rs::cpp2rust_errno().write(__e as i32);
+            -1
+        }
+    }
+}

rules/socket/tgt_refcount.rs

@@ -202,3 +202,19 @@ fn f11(a0: i32, a1: i32, a2: i32, a3: Ptr<i32>) -> i32 {
         }
     }
 }
+
+fn f17(a0: i32, a1: i32) -> i32 {
+    match nix::sys::socket::Backlog::new(a1) {
+        Ok(__b) => match FdRegistry::with_fd(a0, |__fd| nix::sys::socket::listen(&__fd, __b)) {
+            Ok(()) => 0,
+            Err(__e) => {
+                libcc2rs::cpp2rust_errno().write(__e as i32);
+                -1
+            }
+        },
+        Err(__e) => {
+            libcc2rs::cpp2rust_errno().write(__e as i32);
+            -1
+        }
+    }
+}

rules/src/modules.rs

@@ -48,6 +48,8 @@ pub mod dirent_tgt_unsafe;
 pub mod errno_tgt_refcount;
 #[path = r#"../errno/tgt_unsafe.rs"#]
 pub mod errno_tgt_unsafe;
+#[path = r#"../eventfd/tgt_refcount.rs"#]
+pub mod eventfd_tgt_refcount;
 #[path = r#"../eventfd/tgt_unsafe.rs"#]
 pub mod eventfd_tgt_unsafe;
 #[path = r#"../fcntl/tgt_refcount.rs"#]

rules/unistd/tgt_refcount.rs

@@ -7,6 +7,22 @@ fn f1(a0: i32) -> i32 {
     FdRegistry::close(a0)
 }
 
+fn f2(a0: i32, a1: i64, a2: i32) -> i64 {
+    let __whence = match a2 {
+        0 => nix::unistd::Whence::SeekSet,
+        1 => nix::unistd::Whence::SeekCur,
+        2 => nix::unistd::Whence::SeekEnd,
+        __w => panic!("lseek: unsupported whence {__w}"),
+    };
+    match FdRegistry::with_fd(a0, |__fd| nix::unistd::lseek(__fd, a1, __whence)) {
+        Ok(__off) => __off,
+        Err(__e) => {
+            libcc2rs::cpp2rust_errno().write(__e as i32);
+            -1
+        }
+    }
+}
+
 fn f3(a0: i32, a1: AnyPtr, a2: usize) -> isize {
     match FdRegistry::with_fd(a0, |__fd| {
         a1.reinterpret_cast::<u8>()
@@ -45,6 +61,26 @@ fn f5(a0: Ptr<i32>) -> i32 {
     }
 }
 
+fn f6(a0: i32, a1: i64) -> i32 {
+    match FdRegistry::with_fd(a0, |__fd| nix::unistd::ftruncate(__fd, a1)) {
+        Ok(()) => 0,
+        Err(__e) => {
+            libcc2rs::cpp2rust_errno().write(__e as i32);
+            -1
+        }
+    }
+}
+
+fn f7(a0: i32) -> i32 {
+    match FdRegistry::with_fd(a0, |__fd| nix::unistd::isatty(__fd)) {
+        Ok(__tty) => __tty as i32,
+        Err(__e) => {
+            libcc2rs::cpp2rust_errno().write(__e as i32);
+            0
+        }
+    }
+}
+
 fn f8() -> u32 {
     nix::unistd::geteuid().as_raw()
 }