use std::fs;
use std::io::{Read as _, Seek, SeekFrom, Write as _};
use std::path::Path;
use std::pin::Pin;
use std::sync::Mutex;
use cfg_if::cfg_if;
use futures::future::{self, FutureExt, TryFutureExt};
use futures::io::{AsyncRead, AsyncSeek, AsyncWrite, Initializer};
use crate::future::Future;
use crate::io;
use crate::task::{blocking, Context, Poll};
#[derive(Debug)]
pub struct File {
mutex: Mutex<State>,
#[cfg(unix)]
raw_fd: std::os::unix::io::RawFd,
#[cfg(windows)]
raw_handle: UnsafeShared<std::os::windows::io::RawHandle>,
}
#[derive(Debug)]
enum State {
Idle(Option<Inner>),
Busy(blocking::JoinHandle<State>),
}
#[derive(Debug)]
struct Inner {
file: fs::File,
buf: Vec<u8>,
last_op: Option<Operation>,
}
#[derive(Debug)]
enum Operation {
Read(io::Result<usize>),
Write(io::Result<usize>),
Seek(io::Result<u64>),
Flush(io::Result<()>),
}
impl File {
pub async fn open<P: AsRef<Path>>(path: P) -> io::Result<File> {
let path = path.as_ref().to_owned();
let file = blocking::spawn(async move { fs::File::open(&path) }).await?;
#[cfg(unix)]
let file = File {
raw_fd: file.as_raw_fd(),
mutex: Mutex::new(State::Idle(Some(Inner {
file,
buf: Vec::new(),
last_op: None,
}))),
};
#[cfg(windows)]
let file = File {
raw_handle: UnsafeShared(file.as_raw_handle()),
mutex: Mutex::new(State::Idle(Some(Inner {
file,
buf: Vec::new(),
last_op: None,
}))),
};
Ok(file)
}
pub async fn create<P: AsRef<Path>>(path: P) -> io::Result<File> {
let path = path.as_ref().to_owned();
let file = blocking::spawn(async move { fs::File::create(&path) }).await?;
#[cfg(unix)]
let file = File {
raw_fd: file.as_raw_fd(),
mutex: Mutex::new(State::Idle(Some(Inner {
file,
buf: Vec::new(),
last_op: None,
}))),
};
#[cfg(windows)]
let file = File {
raw_handle: UnsafeShared(file.as_raw_handle()),
mutex: Mutex::new(State::Idle(Some(Inner {
file,
buf: Vec::new(),
last_op: None,
}))),
};
Ok(file)
}
pub async fn sync_all(&self) -> io::Result<()> {
future::poll_fn(|cx| {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => match opt.take() {
None => return Poll::Ready(None),
Some(inner) => {
let (s, r) = futures::channel::oneshot::channel();
*state = State::Busy(blocking::spawn(async move {
let res = inner.file.sync_all();
let _ = s.send(res);
State::Idle(Some(inner))
}));
return Poll::Ready(Some(r));
}
},
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
})
.map(|opt| opt.ok_or_else(|| io_error("file closed")))
.await?
.map_err(|_| io_error("blocking task failed"))
.await?
}
pub async fn sync_data(&self) -> io::Result<()> {
future::poll_fn(|cx| {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => match opt.take() {
None => return Poll::Ready(None),
Some(inner) => {
let (s, r) = futures::channel::oneshot::channel();
*state = State::Busy(blocking::spawn(async move {
let res = inner.file.sync_data();
let _ = s.send(res);
State::Idle(Some(inner))
}));
return Poll::Ready(Some(r));
}
},
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
})
.map(|opt| opt.ok_or_else(|| io_error("file closed")))
.await?
.map_err(|_| io_error("blocking task failed"))
.await?
}
pub async fn set_len(&self, size: u64) -> io::Result<()> {
future::poll_fn(|cx| {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => match opt.take() {
None => return Poll::Ready(None),
Some(inner) => {
let (s, r) = futures::channel::oneshot::channel();
*state = State::Busy(blocking::spawn(async move {
let res = inner.file.set_len(size);
let _ = s.send(res);
State::Idle(Some(inner))
}));
return Poll::Ready(Some(r));
}
},
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
})
.map(|opt| opt.ok_or_else(|| io_error("file closed")))
.await?
.map_err(|_| io_error("blocking task failed"))
.await?
}
pub async fn metadata(&self) -> io::Result<fs::Metadata> {
future::poll_fn(|cx| {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => match opt.take() {
None => return Poll::Ready(None),
Some(inner) => {
let (s, r) = futures::channel::oneshot::channel();
*state = State::Busy(blocking::spawn(async move {
let res = inner.file.metadata();
let _ = s.send(res);
State::Idle(Some(inner))
}));
return Poll::Ready(Some(r));
}
},
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
})
.map(|opt| opt.ok_or_else(|| io_error("file closed")))
.await?
.map_err(|_| io_error("blocking task failed"))
.await?
}
pub async fn set_permissions(&self, perm: fs::Permissions) -> io::Result<()> {
let mut perm = Some(perm);
future::poll_fn(|cx| {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => match opt.take() {
None => return Poll::Ready(None),
Some(inner) => {
let (s, r) = futures::channel::oneshot::channel();
let perm = perm.take().unwrap();
*state = State::Busy(blocking::spawn(async move {
let res = inner.file.set_permissions(perm);
let _ = s.send(res);
State::Idle(Some(inner))
}));
return Poll::Ready(Some(r));
}
},
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
})
.map(|opt| opt.ok_or_else(|| io_error("file closed")))
.await?
.map_err(|_| io_error("blocking task failed"))
.await?
}
}
impl AsyncRead for File {
fn poll_read(
self: Pin<&mut Self>,
cx: &mut Context<'_>,
buf: &mut [u8],
) -> Poll<io::Result<usize>> {
Pin::new(&mut &*self).poll_read(cx, buf)
}
#[inline]
unsafe fn initializer(&self) -> Initializer {
Initializer::nop()
}
}
impl AsyncRead for &File {
fn poll_read(
mut self: Pin<&mut Self>,
cx: &mut Context<'_>,
buf: &mut [u8],
) -> Poll<io::Result<usize>> {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => {
let inner = opt.as_mut().ok_or_else(|| io_error("file closed"))?;
let mut offset = 0;
if let Some(Operation::Read(res)) = inner.last_op.take() {
let n = res?;
if n <= buf.len() {
buf[..n].copy_from_slice(&inner.buf[..n]);
return Poll::Ready(Ok(n));
}
offset = n;
}
let mut inner = opt.take().unwrap();
if inner.buf.len() < buf.len() {
inner.buf.reserve(buf.len() - inner.buf.len());
}
unsafe {
inner.buf.set_len(buf.len());
}
*state = State::Busy(blocking::spawn(async move {
if offset > 0 {
let pos = SeekFrom::Current(-(offset as i64));
let _ = Seek::seek(&mut inner.file, pos);
}
let res = inner.file.read(&mut inner.buf);
inner.last_op = Some(Operation::Read(res));
State::Idle(Some(inner))
}));
}
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
}
#[inline]
unsafe fn initializer(&self) -> Initializer {
Initializer::nop()
}
}
impl AsyncWrite for File {
fn poll_write(
self: Pin<&mut Self>,
cx: &mut Context<'_>,
buf: &[u8],
) -> Poll<io::Result<usize>> {
Pin::new(&mut &*self).poll_write(cx, buf)
}
fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<()>> {
Pin::new(&mut &*self).poll_flush(cx)
}
fn poll_close(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<()>> {
Pin::new(&mut &*self).poll_close(cx)
}
}
impl AsyncWrite for &File {
fn poll_write(
mut self: Pin<&mut Self>,
cx: &mut Context<'_>,
buf: &[u8],
) -> Poll<io::Result<usize>> {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => {
let inner = opt.as_mut().ok_or_else(|| io_error("file closed"))?;
if let Some(Operation::Write(res)) = inner.last_op.take() {
let n = res?;
if n <= buf.len() {
return Poll::Ready(Ok(n));
}
} else {
let mut inner = opt.take().unwrap();
if inner.buf.len() < buf.len() {
inner.buf.reserve(buf.len() - inner.buf.len());
}
unsafe {
inner.buf.set_len(buf.len());
}
inner.buf[..buf.len()].copy_from_slice(buf);
*state = State::Busy(blocking::spawn(async move {
let res = inner.file.write(&mut inner.buf);
inner.last_op = Some(Operation::Write(res));
State::Idle(Some(inner))
}));
}
}
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
}
fn poll_flush(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<()>> {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => {
let inner = match opt.as_mut() {
None => return Poll::Ready(Ok(())),
Some(s) => s,
};
if let Some(Operation::Flush(res)) = inner.last_op.take() {
return Poll::Ready(res);
} else {
let mut inner = opt.take().unwrap();
*state = State::Busy(blocking::spawn(async move {
let res = inner.file.flush();
inner.last_op = Some(Operation::Flush(res));
State::Idle(Some(inner))
}));
}
}
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
}
fn poll_close(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<()>> {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => {
let inner = match opt.take() {
None => return Poll::Ready(Ok(())),
Some(s) => s,
};
*state = State::Busy(blocking::spawn(async move {
drop(inner);
State::Idle(None)
}));
}
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
}
}
impl AsyncSeek for File {
fn poll_seek(
self: Pin<&mut Self>,
cx: &mut Context<'_>,
pos: SeekFrom,
) -> Poll<io::Result<u64>> {
Pin::new(&mut &*self).poll_seek(cx, pos)
}
}
impl AsyncSeek for &File {
fn poll_seek(
mut self: Pin<&mut Self>,
cx: &mut Context<'_>,
pos: SeekFrom,
) -> Poll<io::Result<u64>> {
let state = &mut *self.mutex.lock().unwrap();
loop {
match state {
State::Idle(opt) => {
let inner = opt.as_mut().ok_or_else(|| io_error("file closed"))?;
if let Some(Operation::Seek(res)) = inner.last_op.take() {
return Poll::Ready(res);
} else {
let mut inner = opt.take().unwrap();
*state = State::Busy(blocking::spawn(async move {
let res = inner.file.seek(pos);
inner.last_op = Some(Operation::Seek(res));
State::Idle(Some(inner))
}));
}
}
State::Busy(task) => *state = futures::ready!(Pin::new(task).poll(cx)),
}
}
}
}
fn io_error(err: impl Into<Box<dyn std::error::Error + Send + Sync>>) -> io::Error {
io::Error::new(io::ErrorKind::Other, err)
}
impl From<std::fs::File> for File {
fn from(file: fs::File) -> File {
#[cfg(unix)]
let file = File {
raw_fd: file.as_raw_fd(),
mutex: Mutex::new(State::Idle(Some(Inner {
file,
buf: Vec::new(),
last_op: None,
}))),
};
#[cfg(windows)]
let file = File {
raw_handle: UnsafeShared(file.as_raw_handle()),
mutex: Mutex::new(State::Idle(Some(Inner {
file,
buf: Vec::new(),
last_op: None,
}))),
};
file
}
}
cfg_if! {
if #[cfg(feature = "docs")] {
use crate::os::unix::io::{AsRawFd, FromRawFd, IntoRawFd, RawFd};
use crate::os::windows::io::{AsRawHandle, FromRawHandle, IntoRawHandle, RawHandle};
} else if #[cfg(unix)] {
use std::os::unix::io::{AsRawFd, FromRawFd, IntoRawFd, RawFd};
} else if #[cfg(windows)] {
use std::os::windows::io::{AsRawHandle, FromRawHandle, IntoRawHandle, RawHandle};
}
}
#[cfg_attr(feature = "docs", doc(cfg(unix)))]
cfg_if! {
if #[cfg(any(unix, feature = "docs"))] {
impl AsRawFd for File {
fn as_raw_fd(&self) -> RawFd {
self.raw_fd
}
}
impl FromRawFd for File {
unsafe fn from_raw_fd(fd: RawFd) -> File {
fs::File::from_raw_fd(fd).into()
}
}
impl IntoRawFd for File {
fn into_raw_fd(self) -> RawFd {
self.raw_fd
}
}
}
}
#[cfg_attr(feature = "docs", doc(cfg(windows)))]
cfg_if! {
if #[cfg(any(windows, feature = "docs"))] {
impl AsRawHandle for File {
fn as_raw_handle(&self) -> RawHandle {
self.raw_handle.0
}
}
impl FromRawHandle for File {
unsafe fn from_raw_handle(handle: RawHandle) -> File {
fs::File::from_raw_handle(handle).into()
}
}
impl IntoRawHandle for File {
fn into_raw_handle(self) -> RawHandle {
self.raw_handle.0
}
}
#[derive(Debug)]
struct UnsafeShared<T>(T);
unsafe impl<T> Send for UnsafeShared<T> {}
unsafe impl<T> Sync for UnsafeShared<T> {}
}
}