19f532d00SPekka Enberg #ifndef KVM__DISK_IMAGE_H 29f532d00SPekka Enberg #define KVM__DISK_IMAGE_H 39f532d00SPekka Enberg 4*c9310ac4SAsias He #include "kvm/read-write.h" 5*c9310ac4SAsias He #include "kvm/util.h" 6*c9310ac4SAsias He 73fdf659dSSasha Levin #include <linux/types.h> 8*c9310ac4SAsias He #include <linux/fs.h> /* for BLKGETSIZE64 */ 9*c9310ac4SAsias He #include <sys/ioctl.h> 10*c9310ac4SAsias He #include <sys/types.h> 11*c9310ac4SAsias He #include <sys/mman.h> 12*c9310ac4SAsias He #include <sys/stat.h> 139ac38fe1SSasha Levin #include <stdbool.h> 1470b53f25SSasha Levin #include <sys/uio.h> 15*c9310ac4SAsias He #include <stdbool.h> 16*c9310ac4SAsias He #include <stddef.h> 17*c9310ac4SAsias He #include <stdlib.h> 1829084a74SPrasad Joshi #include <unistd.h> 19*c9310ac4SAsias He #include <fcntl.h> 209f532d00SPekka Enberg 212924f993SAsias He #define SECTOR_SHIFT 9 222924f993SAsias He #define SECTOR_SIZE (1UL << SECTOR_SHIFT) 232924f993SAsias He 24499f3bedSPekka Enberg struct disk_image; 25499f3bedSPekka Enberg 26499f3bedSPekka Enberg struct disk_image_operations { 2743835ac9SSasha Levin int (*read_sector)(struct disk_image *disk, u64 sector, void *dst, u32 dst_len); 2843835ac9SSasha Levin int (*write_sector)(struct disk_image *disk, u64 sector, void *src, u32 src_len); 2943835ac9SSasha Levin ssize_t (*read_sector_iov)(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 3043835ac9SSasha Levin ssize_t (*write_sector_iov)(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 3129084a74SPrasad Joshi int (*flush)(struct disk_image *disk); 3243835ac9SSasha Levin void (*close)(struct disk_image *disk); 33499f3bedSPekka Enberg }; 34499f3bedSPekka Enberg 359f532d00SPekka Enberg struct disk_image { 369f532d00SPekka Enberg int fd; 373fdf659dSSasha Levin u64 size; 38499f3bedSPekka Enberg struct disk_image_operations *ops; 39499f3bedSPekka Enberg void *priv; 409f532d00SPekka Enberg }; 419f532d00SPekka Enberg 429ac38fe1SSasha Levin struct disk_image *disk_image__open(const char *filename, bool readonly); 433fdf659dSSasha Levin struct disk_image *disk_image__new(int fd, u64 size, struct disk_image_operations *ops); 443fdf659dSSasha Levin struct disk_image *disk_image__new_readonly(int fd, u64 size, struct disk_image_operations *ops); 4543835ac9SSasha Levin void disk_image__close(struct disk_image *disk); 46499f3bedSPekka Enberg 4743835ac9SSasha Levin static inline int disk_image__read_sector(struct disk_image *disk, u64 sector, void *dst, u32 dst_len) 48499f3bedSPekka Enberg { 4943835ac9SSasha Levin return disk->ops->read_sector(disk, sector, dst, dst_len); 50499f3bedSPekka Enberg } 51499f3bedSPekka Enberg 5243835ac9SSasha Levin static inline int disk_image__write_sector(struct disk_image *disk, u64 sector, void *src, u32 src_len) 53499f3bedSPekka Enberg { 5443835ac9SSasha Levin return disk->ops->write_sector(disk, sector, src, src_len); 55499f3bedSPekka Enberg } 569f532d00SPekka Enberg 5770b0d7b0SSasha Levin ssize_t disk_image__read_sector_iov(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 5870b0d7b0SSasha Levin ssize_t disk_image__write_sector_iov(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 5970b53f25SSasha Levin 6029084a74SPrasad Joshi static inline int disk_image__flush(struct disk_image *disk) 6129084a74SPrasad Joshi { 6229084a74SPrasad Joshi if (disk->ops->flush) 6329084a74SPrasad Joshi return disk->ops->flush(disk); 6429084a74SPrasad Joshi return fsync(disk->fd); 6529084a74SPrasad Joshi } 6629084a74SPrasad Joshi 67*c9310ac4SAsias He struct disk_image *raw_image__probe(int fd, struct stat *st, bool readonly); 68*c9310ac4SAsias He struct disk_image *blkdev__probe(const char *filename, struct stat *st); 69*c9310ac4SAsias He 709f532d00SPekka Enberg #endif /* KVM__DISK_IMAGE_H */ 71