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 #include "kvm/parse-options.h"
7
8 #include <linux/types.h>
9 #include <linux/fs.h> /* for BLKGETSIZE64 */
10 #include <sys/ioctl.h>
11 #include <sys/types.h>
12 #include <sys/mman.h>
13 #include <sys/stat.h>
14 #include <stdbool.h>
15 #include <sys/uio.h>
16 #include <stdbool.h>
17 #include <stddef.h>
18 #include <stdlib.h>
19 #include <unistd.h>
20 #include <fcntl.h>
21
22 #ifdef CONFIG_HAS_AIO
23 #include <libaio.h>
24 #endif
25
26 #define SECTOR_SHIFT 9
27 #define SECTOR_SIZE (1UL << SECTOR_SHIFT)
28
29 enum {
30 DISK_IMAGE_REGULAR,
31 DISK_IMAGE_MMAP,
32 };
33
34 #define MAX_DISK_IMAGES 4
35
36 struct disk_image;
37
38 struct disk_image_operations {
39 ssize_t (*read)(struct disk_image *disk, u64 sector, const struct iovec *iov,
40 int iovcount, void *param);
41 ssize_t (*write)(struct disk_image *disk, u64 sector, const struct iovec *iov,
42 int iovcount, void *param);
43 int (*flush)(struct disk_image *disk);
44 int (*wait)(struct disk_image *disk);
45 int (*close)(struct disk_image *disk);
46 bool async;
47 };
48
49 struct disk_image_params {
50 const char *filename;
51 /* wwpn == World Wide Port Number */
52 const char *wwpn;
53 bool readonly;
54 bool direct;
55 };
56
57 struct disk_image {
58 int fd;
59 u64 size;
60 struct disk_image_operations *ops;
61 void *priv;
62 void *disk_req_cb_param;
63 void (*disk_req_cb)(void *param, long len);
64 bool readonly;
65 bool async;
66 #ifdef CONFIG_HAS_AIO
67 io_context_t ctx;
68 int evt;
69 pthread_t thread;
70 u64 aio_inflight;
71 #endif /* CONFIG_HAS_AIO */
72 const char *wwpn;
73 int debug_iodelay;
74 };
75
76 int disk_img_name_parser(const struct option *opt, const char *arg, int unset);
77 int disk_image__init(struct kvm *kvm);
78 int disk_image__exit(struct kvm *kvm);
79 struct disk_image *disk_image__new(int fd, u64 size, struct disk_image_operations *ops, int mmap);
80 int disk_image__flush(struct disk_image *disk);
81 int disk_image__wait(struct disk_image *disk);
82 ssize_t disk_image__read(struct disk_image *disk, u64 sector, const struct iovec *iov,
83 int iovcount, void *param);
84 ssize_t disk_image__write(struct disk_image *disk, u64 sector, const struct iovec *iov,
85 int iovcount, void *param);
86 ssize_t disk_image__get_serial(struct disk_image *disk, struct iovec *iov,
87 int iovcount, ssize_t len);
88
89 struct disk_image *raw_image__probe(int fd, struct stat *st, bool readonly);
90 struct disk_image *blkdev__probe(const char *filename, int flags, struct stat *st);
91
92 ssize_t raw_image__read_sync(struct disk_image *disk, u64 sector,
93 const struct iovec *iov, int iovcount, void *param);
94 ssize_t raw_image__write_sync(struct disk_image *disk, u64 sector,
95 const struct iovec *iov, int iovcount, void *param);
96 ssize_t raw_image__read_mmap(struct disk_image *disk, u64 sector,
97 const struct iovec *iov, int iovcount, void *param);
98 ssize_t raw_image__write_mmap(struct disk_image *disk, u64 sector,
99 const struct iovec *iov, int iovcount, void *param);
100 int raw_image__close(struct disk_image *disk);
101 void disk_image__set_callback(struct disk_image *disk, void (*disk_req_cb)(void *param, long len));
102
103 #ifdef CONFIG_HAS_AIO
104 int disk_aio_setup(struct disk_image *disk);
105 void disk_aio_destroy(struct disk_image *disk);
106 ssize_t raw_image__read_async(struct disk_image *disk, u64 sector,
107 const struct iovec *iov, int iovcount, void *param);
108 ssize_t raw_image__write_async(struct disk_image *disk, u64 sector,
109 const struct iovec *iov, int iovcount, void *param);
110 int raw_image__wait(struct disk_image *disk);
111
112 #define raw_image__read raw_image__read_async
113 #define raw_image__write raw_image__write_async
114
115 #else /* !CONFIG_HAS_AIO */
disk_aio_setup(struct disk_image * disk)116 static inline int disk_aio_setup(struct disk_image *disk)
117 {
118 /* No-op */
119 return 0;
120 }
disk_aio_destroy(struct disk_image * disk)121 static inline void disk_aio_destroy(struct disk_image *disk)
122 {
123 }
124
raw_image__wait(struct disk_image * disk)125 static inline int raw_image__wait(struct disk_image *disk)
126 {
127 return 0;
128 }
129 #define raw_image__read raw_image__read_sync
130 #define raw_image__write raw_image__write_sync
131 #endif /* CONFIG_HAS_AIO */
132
133 #endif /* KVM__DISK_IMAGE_H */
134