Support SEEK_DATA and SEEK_HOLE for lseek

src/backend/libc/fs/syscalls.rs

@@ -900,6 +900,8 @@ pub(crate) fn seek(fd: BorrowedFd<'_>, pos: SeekFrom) -> io::Result<u64> {
         }
         SeekFrom::End(offset) => (c::SEEK_END, offset),
         SeekFrom::Current(offset) => (c::SEEK_CUR, offset),
+        SeekFrom::Data(offset) => (c::SEEK_DATA, offset),
+        SeekFrom::Hole(offset) => (c::SEEK_HOLE, offset),
     };
     let offset = unsafe { ret_off_t(libc_lseek(borrowed_fd(fd), offset, whence))? };
     Ok(offset as u64)

src/backend/linux_raw/fs/syscalls.rs

@@ -37,7 +37,8 @@ use linux_raw_sys::general::stat as linux_stat64;
 use linux_raw_sys::general::{
     __kernel_fsid_t, __kernel_timespec, open_how, statx, AT_EACCESS, AT_FDCWD, AT_REMOVEDIR,
     AT_SYMLINK_NOFOLLOW, F_ADD_SEALS, F_GETFL, F_GETLEASE, F_GETOWN, F_GETPIPE_SZ, F_GETSIG,
-    F_GET_SEALS, F_SETFL, F_SETPIPE_SZ, SEEK_CUR, SEEK_END, SEEK_SET, STATX__RESERVED,
+    F_GET_SEALS, F_SETFL, F_SETPIPE_SZ, SEEK_CUR, SEEK_DATA, SEEK_END, SEEK_HOLE, SEEK_SET,
+    STATX__RESERVED,
 };
 #[cfg(target_pointer_width = "32")]
 use {
@@ -214,6 +215,8 @@ pub(crate) fn seek(fd: BorrowedFd<'_>, pos: SeekFrom) -> io::Result<u64> {
         }
         SeekFrom::End(offset) => (SEEK_END, offset),
         SeekFrom::Current(offset) => (SEEK_CUR, offset),
+        SeekFrom::Data(offset) => (SEEK_DATA, offset),
+        SeekFrom::Hole(offset) => (SEEK_HOLE, offset),
     };
     _seek(fd, offset, whence)
 }

src/io/mod.rs

@@ -22,7 +22,7 @@ mod poll;
 mod procfs;
 #[cfg(not(windows))]
 mod read_write;
-#[cfg(not(feature = "std"))]
+#[cfg(not(windows))]
 mod seek_from;
 #[cfg(not(windows))]
 mod stdio;
@@ -89,10 +89,7 @@ pub use read_write::{pread, pwrite, read, readv, write, writev, IoSlice, IoSlice
 pub use read_write::{preadv, pwritev};
 #[cfg(any(target_os = "android", target_os = "linux"))]
 pub use read_write::{preadv2, pwritev2, ReadWriteFlags};
-#[cfg(not(feature = "std"))]
 pub use seek_from::SeekFrom;
-#[cfg(feature = "std")]
-pub use std::io::SeekFrom;
 #[cfg(not(windows))]
 pub use stdio::{
     raw_stderr, raw_stdin, raw_stdout, stderr, stdin, stdout, take_stderr, take_stdin, take_stdout,

src/io/seek_from.rs

@@ -27,4 +27,22 @@ pub enum SeekFrom {
     /// to seek before byte 0.
     #[cfg_attr(staged_api, stable(feature = "rust1", since = "1.0.0"))]
     Current(#[cfg_attr(staged_api, stable(feature = "rust1", since = "1.0.0"))] i64),
+
+    /// Sets the offset to next location in the file greater than or equal to the
+    /// specified number of bytes that contains data.
+    ///
+    /// If the offset points to data, the file offset is set to it.
+    #[cfg_attr(staged_api, stable(feature = "rust1", since = "1.0.0"))]
+    Data(#[cfg_attr(staged_api, stable(feature = "rust1", since = "1.0.0"))] i64),
+
+    /// Sets the offset to the next hole in the file greater than or equal to the
+    /// specified number of bytes.
+    ///
+    /// If offset points into the middle of a hole, then the file offset is set to
+    /// offset.
+    ///
+    /// If there is no hole past offset, then the file offset is adjusted to the end
+    /// of the file (i.e., there is an implicit hole at the end of any file).
+    #[cfg_attr(staged_api, stable(feature = "rust1", since = "1.0.0"))]
+    Hole(#[cfg_attr(staged_api, stable(feature = "rust1", since = "1.0.0"))] i64),
 }

tests/io/read_write.rs

@@ -97,7 +97,7 @@ fn test_readwrite_v() {
 fn test_readwrite() {
     use rustix::fs::{cwd, openat, seek, Mode, OFlags};
     use rustix::io::{read, write};
-    use std::io::SeekFrom;
+    use rustix::io::SeekFrom;
 
     let tmp = tempfile::tempdir().unwrap();
     let dir = openat(cwd(), tmp.path(), OFlags::RDONLY, Mode::empty()).unwrap();