1 #ifndef KVM__DISK_IMAGE_H 2 #define KVM__DISK_IMAGE_H 3 4 #include "kvm/read-write.h" 5 #include "kvm/util.h" 6 7 #include <linux/types.h> 8 #include <linux/fs.h> /* for BLKGETSIZE64 */ 9 #include <sys/ioctl.h> 10 #include <sys/types.h> 11 #include <sys/mman.h> 12 #include <sys/stat.h> 13 #include <stdbool.h> 14 #include <sys/uio.h> 15 #include <stdbool.h> 16 #include <stddef.h> 17 #include <stdlib.h> 18 #include <unistd.h> 19 #include <fcntl.h> 20 21 #define SECTOR_SHIFT 9 22 #define SECTOR_SIZE (1UL << SECTOR_SHIFT) 23 24 enum { 25 DISK_IMAGE_REGULAR, 26 DISK_IMAGE_MMAP, 27 }; 28 29 #define MAX_DISK_IMAGES 4 30 31 struct disk_image; 32 33 struct disk_image_operations { 34 ssize_t (*read)(struct disk_image *disk, u64 sector, const struct iovec *iov, 35 int iovcount, void *param); 36 ssize_t (*write)(struct disk_image *disk, u64 sector, const struct iovec *iov, 37 int iovcount, void *param); 38 int (*flush)(struct disk_image *disk); 39 int (*close)(struct disk_image *disk); 40 }; 41 42 struct disk_image_params { 43 const char *filename; 44 bool readonly; 45 bool direct; 46 }; 47 48 struct disk_image { 49 int fd; 50 u64 size; 51 struct disk_image_operations *ops; 52 void *priv; 53 void *disk_req_cb_param; 54 void (*disk_req_cb)(void *param, long len); 55 bool async; 56 int evt; 57 #ifdef CONFIG_HAS_AIO 58 io_context_t ctx; 59 #endif 60 }; 61 62 struct disk_image *disk_image__open(const char *filename, bool readonly, bool direct); 63 struct disk_image **disk_image__open_all(struct disk_image_params *params, int count); 64 struct disk_image *disk_image__new(int fd, u64 size, struct disk_image_operations *ops, int mmap); 65 int disk_image__close(struct disk_image *disk); 66 int disk_image__close_all(struct disk_image **disks, int count); 67 int disk_image__flush(struct disk_image *disk); 68 ssize_t disk_image__read(struct disk_image *disk, u64 sector, const struct iovec *iov, 69 int iovcount, void *param); 70 ssize_t disk_image__write(struct disk_image *disk, u64 sector, const struct iovec *iov, 71 int iovcount, void *param); 72 ssize_t disk_image__get_serial(struct disk_image *disk, void *buffer, ssize_t *len); 73 74 struct disk_image *raw_image__probe(int fd, struct stat *st, bool readonly); 75 struct disk_image *blkdev__probe(const char *filename, int flags, struct stat *st); 76 77 ssize_t raw_image__read(struct disk_image *disk, u64 sector, 78 const struct iovec *iov, int iovcount, void *param); 79 ssize_t raw_image__write(struct disk_image *disk, u64 sector, 80 const struct iovec *iov, int iovcount, void *param); 81 ssize_t raw_image__read_mmap(struct disk_image *disk, u64 sector, 82 const struct iovec *iov, int iovcount, void *param); 83 ssize_t raw_image__write_mmap(struct disk_image *disk, u64 sector, 84 const struct iovec *iov, int iovcount, void *param); 85 int raw_image__close(struct disk_image *disk); 86 void disk_image__set_callback(struct disk_image *disk, void (*disk_req_cb)(void *param, long len)); 87 #endif /* KVM__DISK_IMAGE_H */ 88