diff --git a/block/src/raw_async.rs b/block/src/raw_async.rs index b5ae8f50a..7dcda5bbd 100644 --- a/block/src/raw_async.rs +++ b/block/src/raw_async.rs @@ -14,8 +14,10 @@ use vmm_sys_util::eventfd::EventFd; use crate::async_io::{ AsyncIo, AsyncIoError, AsyncIoResult, BorrowedDiskFd, DiskFile, DiskFileError, DiskFileResult, }; +use crate::error::{BlockError, BlockErrorKind, BlockResult}; use crate::{ - BatchRequest, DiskTopology, RequestType, SECTOR_SIZE, probe_sparse_support, query_device_size, + BatchRequest, DiskTopology, RequestType, SECTOR_SIZE, disk_file, probe_sparse_support, + query_device_size, }; #[derive(Debug)] @@ -72,6 +74,14 @@ impl DiskFile for RawFileDisk { } } +impl disk_file::DiskSize for RawFileDisk { + fn logical_size(&self) -> BlockResult { + query_device_size(&self.file) + .map(|(logical_size, _)| logical_size) + .map_err(|e| BlockError::new(BlockErrorKind::Io, DiskFileError::Size(e))) + } +} + pub struct RawFileAsync { fd: RawFd, io_uring: IoUring,