mirror of
https://github.com/cloud-hypervisor/cloud-hypervisor.git
synced 2024-09-19 13:30:56 +00:00
163 lines
4.5 KiB
Rust
163 lines
4.5 KiB
Rust
|
// Copyright © 2021 Intel Corporation
|
||
|
//
|
||
|
// SPDX-License-Identifier: Apache-2.0 AND BSD-3-Clause
|
||
|
|
||
|
use crate::async_io::{
|
||
|
AsyncIo, AsyncIoError, AsyncIoResult, DiskFile, DiskFileError, DiskFileResult,
|
||
|
};
|
||
|
use qcow::RawFile;
|
||
|
use std::fs::File;
|
||
|
use std::io::{IoSlice, IoSliceMut, Read, Seek, SeekFrom, Write};
|
||
|
use std::sync::{Arc, Mutex};
|
||
|
use vmm_sys_util::eventfd::EventFd;
|
||
|
|
||
|
pub struct RawFileDiskSync {
|
||
|
raw_file: RawFile,
|
||
|
semaphore: Arc<Mutex<()>>,
|
||
|
}
|
||
|
|
||
|
impl RawFileDiskSync {
|
||
|
pub fn new(file: File, direct_io: bool) -> Self {
|
||
|
RawFileDiskSync {
|
||
|
raw_file: RawFile::new(file, direct_io),
|
||
|
semaphore: Arc::new(Mutex::new(())),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
impl DiskFile for RawFileDiskSync {
|
||
|
fn size(&mut self) -> DiskFileResult<u64> {
|
||
|
// Take the semaphore to ensure other threads are not interacting with
|
||
|
// the underlying file.
|
||
|
let _lock = self.semaphore.lock().unwrap();
|
||
|
|
||
|
Ok(self
|
||
|
.raw_file
|
||
|
.seek(SeekFrom::End(0))
|
||
|
.map_err(DiskFileError::Size)? as u64)
|
||
|
}
|
||
|
|
||
|
fn new_async_io(&self, _ring_depth: u32) -> DiskFileResult<Box<dyn AsyncIo>> {
|
||
|
Ok(Box::new(RawFileSync::new(
|
||
|
self.raw_file.clone(),
|
||
|
self.semaphore.clone(),
|
||
|
)) as Box<dyn AsyncIo>)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub struct RawFileSync {
|
||
|
raw_file: RawFile,
|
||
|
eventfd: EventFd,
|
||
|
completion_list: Vec<(u64, i32)>,
|
||
|
semaphore: Arc<Mutex<()>>,
|
||
|
}
|
||
|
|
||
|
impl RawFileSync {
|
||
|
pub fn new(raw_file: RawFile, semaphore: Arc<Mutex<()>>) -> Self {
|
||
|
RawFileSync {
|
||
|
raw_file,
|
||
|
eventfd: EventFd::new(libc::EFD_NONBLOCK).expect("Failed creating EventFd for RawFile"),
|
||
|
completion_list: Vec::new(),
|
||
|
semaphore,
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
impl AsyncIo for RawFileSync {
|
||
|
fn notifier(&self) -> &EventFd {
|
||
|
&self.eventfd
|
||
|
}
|
||
|
|
||
|
fn read_vectored(
|
||
|
&mut self,
|
||
|
offset: libc::off_t,
|
||
|
iovecs: Vec<libc::iovec>,
|
||
|
user_data: u64,
|
||
|
) -> AsyncIoResult<()> {
|
||
|
// Convert libc::iovec into IoSliceMut
|
||
|
let mut slices = Vec::new();
|
||
|
for iovec in iovecs.iter() {
|
||
|
slices.push(IoSliceMut::new(unsafe { std::mem::transmute(*iovec) }));
|
||
|
}
|
||
|
|
||
|
let result = {
|
||
|
// Take the semaphore to ensure other threads are not interacting
|
||
|
// with the underlying file.
|
||
|
let _lock = self.semaphore.lock().unwrap();
|
||
|
|
||
|
// Move the cursor to the right offset
|
||
|
self.raw_file
|
||
|
.seek(SeekFrom::Start(offset as u64))
|
||
|
.map_err(AsyncIoError::ReadVectored)?;
|
||
|
|
||
|
// Read vectored
|
||
|
self.raw_file
|
||
|
.read_vectored(slices.as_mut_slice())
|
||
|
.map_err(AsyncIoError::ReadVectored)?
|
||
|
};
|
||
|
|
||
|
self.completion_list.push((user_data, result as i32));
|
||
|
self.eventfd.write(1).unwrap();
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
|
||
|
fn write_vectored(
|
||
|
&mut self,
|
||
|
offset: libc::off_t,
|
||
|
iovecs: Vec<libc::iovec>,
|
||
|
user_data: u64,
|
||
|
) -> AsyncIoResult<()> {
|
||
|
// Convert libc::iovec into IoSlice
|
||
|
let mut slices = Vec::new();
|
||
|
for iovec in iovecs.iter() {
|
||
|
slices.push(IoSlice::new(unsafe { std::mem::transmute(*iovec) }));
|
||
|
}
|
||
|
|
||
|
let result = {
|
||
|
// Take the semaphore to ensure other threads are not interacting
|
||
|
// with the underlying file.
|
||
|
let _lock = self.semaphore.lock().unwrap();
|
||
|
|
||
|
// Move the cursor to the right offset
|
||
|
self.raw_file
|
||
|
.seek(SeekFrom::Start(offset as u64))
|
||
|
.map_err(AsyncIoError::WriteVectored)?;
|
||
|
|
||
|
// Write vectored
|
||
|
self.raw_file
|
||
|
.write_vectored(slices.as_slice())
|
||
|
.map_err(AsyncIoError::WriteVectored)?
|
||
|
};
|
||
|
|
||
|
self.completion_list.push((user_data, result as i32));
|
||
|
self.eventfd.write(1).unwrap();
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
|
||
|
fn fsync(&mut self, user_data: Option<u64>) -> AsyncIoResult<()> {
|
||
|
let result: i32 = {
|
||
|
// Take the semaphore to ensure other threads are not interacting
|
||
|
// with the underlying file.
|
||
|
let _lock = self.semaphore.lock().unwrap();
|
||
|
|
||
|
// Flush
|
||
|
self.raw_file.flush().map_err(AsyncIoError::Fsync)?;
|
||
|
|
||
|
0
|
||
|
};
|
||
|
|
||
|
if let Some(user_data) = user_data {
|
||
|
self.completion_list.push((user_data, result));
|
||
|
self.eventfd.write(1).unwrap();
|
||
|
}
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
|
||
|
fn complete(&mut self) -> Vec<(u64, i32)> {
|
||
|
self.completion_list.drain(..).collect()
|
||
|
}
|
||
|
}
|