handle EINT in tokio::fs::write for io-uring

Author: vrtgs

tokio/src/fs/write.rs

@@ -73,11 +73,17 @@ async fn write_uring(path: &Path, mut buf: OwnedBuf) -> io::Result<()> {
     let mut file_offset: u64 = 0;
     while buf_offset < total {
         let (n, _buf, _fd) = Op::write_at(fd, buf, buf_offset, file_offset)?.await;
-        // TODO: handle EINT here
-        let n = n?;
-        if n == 0 {
-            return Err(io::ErrorKind::WriteZero.into());
-        }
+
+        let is_eint = |err: &io::Error| {
+            err.raw_os_error() == Some(libc::EINTR) || err.kind() == io::ErrorKind::Interrupted
+        };
+
+        let n = match n {
+            Ok(0) => return Err(io::ErrorKind::WriteZero.into()),
+            Ok(n) => n,
+            Err(e) if is_eint(&e) => 0,
+            Err(e) => return Err(e),
+        };
 
         buf = _buf;
         fd = _fd;