19f532d00SPekka Enberg #ifndef KVM__DISK_IMAGE_H 29f532d00SPekka Enberg #define KVM__DISK_IMAGE_H 39f532d00SPekka Enberg 4c9310ac4SAsias He #include "kvm/read-write.h" 5c9310ac4SAsias He #include "kvm/util.h" 6c9310ac4SAsias He 73fdf659dSSasha Levin #include <linux/types.h> 8c9310ac4SAsias He #include <linux/fs.h> /* for BLKGETSIZE64 */ 9c9310ac4SAsias He #include <sys/ioctl.h> 10c9310ac4SAsias He #include <sys/types.h> 11c9310ac4SAsias He #include <sys/mman.h> 12c9310ac4SAsias He #include <sys/stat.h> 139ac38fe1SSasha Levin #include <stdbool.h> 1470b53f25SSasha Levin #include <sys/uio.h> 15c9310ac4SAsias He #include <stdbool.h> 16c9310ac4SAsias He #include <stddef.h> 17c9310ac4SAsias He #include <stdlib.h> 1829084a74SPrasad Joshi #include <unistd.h> 19c9310ac4SAsias He #include <fcntl.h> 209f532d00SPekka Enberg 212924f993SAsias He #define SECTOR_SHIFT 9 222924f993SAsias He #define SECTOR_SIZE (1UL << SECTOR_SHIFT) 232924f993SAsias He 247d22135fSAsias He #define DISK_IMAGE_MMAP 0 257d22135fSAsias He #define DISK_IMAGE_NOMMAP 1 267d22135fSAsias He 27499f3bedSPekka Enberg struct disk_image; 28499f3bedSPekka Enberg 29499f3bedSPekka Enberg struct disk_image_operations { 30*72133dd2SAsias He /* 31*72133dd2SAsias He * The following two are used for reading or writing with a single buffer. 32*72133dd2SAsias He * The implentation can use readv/writev or memcpy. 33*72133dd2SAsias He */ 34*72133dd2SAsias He ssize_t (*read_sector)(struct disk_image *disk, u64 sector, void *dst, u32 dst_len); 35*72133dd2SAsias He ssize_t (*write_sector)(struct disk_image *disk, u64 sector, void *src, u32 src_len); 36*72133dd2SAsias He /* 37*72133dd2SAsias He * The following two are used for reading or writing with multiple buffers. 38*72133dd2SAsias He * The implentation can use readv/writev or memcpy. 39*72133dd2SAsias He */ 4043835ac9SSasha Levin ssize_t (*read_sector_iov)(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 4143835ac9SSasha Levin ssize_t (*write_sector_iov)(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 4229084a74SPrasad Joshi int (*flush)(struct disk_image *disk); 43*72133dd2SAsias He int (*close)(struct disk_image *disk); 44499f3bedSPekka Enberg }; 45499f3bedSPekka Enberg 469f532d00SPekka Enberg struct disk_image { 479f532d00SPekka Enberg int fd; 483fdf659dSSasha Levin u64 size; 49499f3bedSPekka Enberg struct disk_image_operations *ops; 50499f3bedSPekka Enberg void *priv; 519f532d00SPekka Enberg }; 529f532d00SPekka Enberg 539ac38fe1SSasha Levin struct disk_image *disk_image__open(const char *filename, bool readonly); 547d22135fSAsias He struct disk_image *disk_image__new(int fd, u64 size, struct disk_image_operations *ops, int mmap); 55*72133dd2SAsias He int disk_image__close(struct disk_image *disk); 56499f3bedSPekka Enberg 57b8861977SAsias He ssize_t disk_image__read(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 58b8861977SAsias He ssize_t disk_image__write(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 67c9310ac4SAsias He struct disk_image *raw_image__probe(int fd, struct stat *st, bool readonly); 68c9310ac4SAsias He struct disk_image *blkdev__probe(const char *filename, struct stat *st); 69c9310ac4SAsias He 70708dc6cbSAsias He ssize_t raw_image__read_sector_iov(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 71708dc6cbSAsias He ssize_t raw_image__write_sector_iov(struct disk_image *disk, u64 sector, const struct iovec *iov, int iovcount); 72*72133dd2SAsias He ssize_t raw_image__read_sector_ro_mmap(struct disk_image *disk, u64 sector, void *dst, u32 dst_len); 73*72133dd2SAsias He ssize_t raw_image__write_sector_ro_mmap(struct disk_image *disk, u64 sector, void *src, u32 src_len); 74*72133dd2SAsias He int raw_image__close_ro_mmap(struct disk_image *disk); 75708dc6cbSAsias He 769f532d00SPekka Enberg #endif /* KVM__DISK_IMAGE_H */ 77