root / block / sheepdog.c @ a8e0fdd7
History | View | Annotate | Download (53 kB)
1 | 33b1db1c | MORITA Kazutaka | /*
|
---|---|---|---|
2 | 33b1db1c | MORITA Kazutaka | * Copyright (C) 2009-2010 Nippon Telegraph and Telephone Corporation.
|
3 | 33b1db1c | MORITA Kazutaka | *
|
4 | 33b1db1c | MORITA Kazutaka | * This program is free software; you can redistribute it and/or
|
5 | 33b1db1c | MORITA Kazutaka | * modify it under the terms of the GNU General Public License version
|
6 | 33b1db1c | MORITA Kazutaka | * 2 as published by the Free Software Foundation.
|
7 | 33b1db1c | MORITA Kazutaka | *
|
8 | 33b1db1c | MORITA Kazutaka | * You should have received a copy of the GNU General Public License
|
9 | 33b1db1c | MORITA Kazutaka | * along with this program. If not, see <http://www.gnu.org/licenses/>.
|
10 | 33b1db1c | MORITA Kazutaka | */
|
11 | 33b1db1c | MORITA Kazutaka | |
12 | 33b1db1c | MORITA Kazutaka | #include "qemu-common.h" |
13 | 33b1db1c | MORITA Kazutaka | #include "qemu-error.h" |
14 | 33b1db1c | MORITA Kazutaka | #include "qemu_socket.h" |
15 | 33b1db1c | MORITA Kazutaka | #include "block_int.h" |
16 | 8cffde73 | Michael Tokarev | #include "bitops.h" |
17 | 33b1db1c | MORITA Kazutaka | |
18 | 33b1db1c | MORITA Kazutaka | #define SD_PROTO_VER 0x01 |
19 | 33b1db1c | MORITA Kazutaka | |
20 | 33b1db1c | MORITA Kazutaka | #define SD_DEFAULT_ADDR "localhost" |
21 | 33b1db1c | MORITA Kazutaka | #define SD_DEFAULT_PORT "7000" |
22 | 33b1db1c | MORITA Kazutaka | |
23 | 33b1db1c | MORITA Kazutaka | #define SD_OP_CREATE_AND_WRITE_OBJ 0x01 |
24 | 33b1db1c | MORITA Kazutaka | #define SD_OP_READ_OBJ 0x02 |
25 | 33b1db1c | MORITA Kazutaka | #define SD_OP_WRITE_OBJ 0x03 |
26 | 33b1db1c | MORITA Kazutaka | |
27 | 33b1db1c | MORITA Kazutaka | #define SD_OP_NEW_VDI 0x11 |
28 | 33b1db1c | MORITA Kazutaka | #define SD_OP_LOCK_VDI 0x12 |
29 | 33b1db1c | MORITA Kazutaka | #define SD_OP_RELEASE_VDI 0x13 |
30 | 33b1db1c | MORITA Kazutaka | #define SD_OP_GET_VDI_INFO 0x14 |
31 | 33b1db1c | MORITA Kazutaka | #define SD_OP_READ_VDIS 0x15 |
32 | 33b1db1c | MORITA Kazutaka | |
33 | 33b1db1c | MORITA Kazutaka | #define SD_FLAG_CMD_WRITE 0x01 |
34 | 33b1db1c | MORITA Kazutaka | #define SD_FLAG_CMD_COW 0x02 |
35 | 33b1db1c | MORITA Kazutaka | |
36 | 33b1db1c | MORITA Kazutaka | #define SD_RES_SUCCESS 0x00 /* Success */ |
37 | 33b1db1c | MORITA Kazutaka | #define SD_RES_UNKNOWN 0x01 /* Unknown error */ |
38 | 33b1db1c | MORITA Kazutaka | #define SD_RES_NO_OBJ 0x02 /* No object found */ |
39 | 33b1db1c | MORITA Kazutaka | #define SD_RES_EIO 0x03 /* I/O error */ |
40 | 33b1db1c | MORITA Kazutaka | #define SD_RES_VDI_EXIST 0x04 /* Vdi exists already */ |
41 | 33b1db1c | MORITA Kazutaka | #define SD_RES_INVALID_PARMS 0x05 /* Invalid parameters */ |
42 | 33b1db1c | MORITA Kazutaka | #define SD_RES_SYSTEM_ERROR 0x06 /* System error */ |
43 | 33b1db1c | MORITA Kazutaka | #define SD_RES_VDI_LOCKED 0x07 /* Vdi is locked */ |
44 | 33b1db1c | MORITA Kazutaka | #define SD_RES_NO_VDI 0x08 /* No vdi found */ |
45 | 33b1db1c | MORITA Kazutaka | #define SD_RES_NO_BASE_VDI 0x09 /* No base vdi found */ |
46 | 33b1db1c | MORITA Kazutaka | #define SD_RES_VDI_READ 0x0A /* Cannot read requested vdi */ |
47 | 33b1db1c | MORITA Kazutaka | #define SD_RES_VDI_WRITE 0x0B /* Cannot write requested vdi */ |
48 | 33b1db1c | MORITA Kazutaka | #define SD_RES_BASE_VDI_READ 0x0C /* Cannot read base vdi */ |
49 | 33b1db1c | MORITA Kazutaka | #define SD_RES_BASE_VDI_WRITE 0x0D /* Cannot write base vdi */ |
50 | 33b1db1c | MORITA Kazutaka | #define SD_RES_NO_TAG 0x0E /* Requested tag is not found */ |
51 | 33b1db1c | MORITA Kazutaka | #define SD_RES_STARTUP 0x0F /* Sheepdog is on starting up */ |
52 | 33b1db1c | MORITA Kazutaka | #define SD_RES_VDI_NOT_LOCKED 0x10 /* Vdi is not locked */ |
53 | 33b1db1c | MORITA Kazutaka | #define SD_RES_SHUTDOWN 0x11 /* Sheepdog is shutting down */ |
54 | 33b1db1c | MORITA Kazutaka | #define SD_RES_NO_MEM 0x12 /* Cannot allocate memory */ |
55 | 33b1db1c | MORITA Kazutaka | #define SD_RES_FULL_VDI 0x13 /* we already have the maximum vdis */ |
56 | 33b1db1c | MORITA Kazutaka | #define SD_RES_VER_MISMATCH 0x14 /* Protocol version mismatch */ |
57 | 33b1db1c | MORITA Kazutaka | #define SD_RES_NO_SPACE 0x15 /* Server has no room for new objects */ |
58 | 33b1db1c | MORITA Kazutaka | #define SD_RES_WAIT_FOR_FORMAT 0x16 /* Waiting for a format operation */ |
59 | 33b1db1c | MORITA Kazutaka | #define SD_RES_WAIT_FOR_JOIN 0x17 /* Waiting for other nodes joining */ |
60 | 33b1db1c | MORITA Kazutaka | #define SD_RES_JOIN_FAILED 0x18 /* Target node had failed to join sheepdog */ |
61 | 33b1db1c | MORITA Kazutaka | |
62 | 33b1db1c | MORITA Kazutaka | /*
|
63 | 33b1db1c | MORITA Kazutaka | * Object ID rules
|
64 | 33b1db1c | MORITA Kazutaka | *
|
65 | 33b1db1c | MORITA Kazutaka | * 0 - 19 (20 bits): data object space
|
66 | 33b1db1c | MORITA Kazutaka | * 20 - 31 (12 bits): reserved data object space
|
67 | 33b1db1c | MORITA Kazutaka | * 32 - 55 (24 bits): vdi object space
|
68 | 33b1db1c | MORITA Kazutaka | * 56 - 59 ( 4 bits): reserved vdi object space
|
69 | 33b1db1c | MORITA Kazutaka | * 60 - 63 ( 4 bits): object type indentifier space
|
70 | 33b1db1c | MORITA Kazutaka | */
|
71 | 33b1db1c | MORITA Kazutaka | |
72 | 33b1db1c | MORITA Kazutaka | #define VDI_SPACE_SHIFT 32 |
73 | 33b1db1c | MORITA Kazutaka | #define VDI_BIT (UINT64_C(1) << 63) |
74 | 33b1db1c | MORITA Kazutaka | #define VMSTATE_BIT (UINT64_C(1) << 62) |
75 | 33b1db1c | MORITA Kazutaka | #define MAX_DATA_OBJS (UINT64_C(1) << 20) |
76 | 33b1db1c | MORITA Kazutaka | #define MAX_CHILDREN 1024 |
77 | 33b1db1c | MORITA Kazutaka | #define SD_MAX_VDI_LEN 256 |
78 | 33b1db1c | MORITA Kazutaka | #define SD_MAX_VDI_TAG_LEN 256 |
79 | 33b1db1c | MORITA Kazutaka | #define SD_NR_VDIS (1U << 24) |
80 | 33b1db1c | MORITA Kazutaka | #define SD_DATA_OBJ_SIZE (UINT64_C(1) << 22) |
81 | 33b1db1c | MORITA Kazutaka | #define SD_MAX_VDI_SIZE (SD_DATA_OBJ_SIZE * MAX_DATA_OBJS)
|
82 | 33b1db1c | MORITA Kazutaka | #define SECTOR_SIZE 512 |
83 | 33b1db1c | MORITA Kazutaka | |
84 | 33b1db1c | MORITA Kazutaka | #define SD_INODE_SIZE (sizeof(SheepdogInode)) |
85 | 33b1db1c | MORITA Kazutaka | #define CURRENT_VDI_ID 0 |
86 | 33b1db1c | MORITA Kazutaka | |
87 | 33b1db1c | MORITA Kazutaka | typedef struct SheepdogReq { |
88 | 33b1db1c | MORITA Kazutaka | uint8_t proto_ver; |
89 | 33b1db1c | MORITA Kazutaka | uint8_t opcode; |
90 | 33b1db1c | MORITA Kazutaka | uint16_t flags; |
91 | 33b1db1c | MORITA Kazutaka | uint32_t epoch; |
92 | 33b1db1c | MORITA Kazutaka | uint32_t id; |
93 | 33b1db1c | MORITA Kazutaka | uint32_t data_length; |
94 | 33b1db1c | MORITA Kazutaka | uint32_t opcode_specific[8];
|
95 | 33b1db1c | MORITA Kazutaka | } SheepdogReq; |
96 | 33b1db1c | MORITA Kazutaka | |
97 | 33b1db1c | MORITA Kazutaka | typedef struct SheepdogRsp { |
98 | 33b1db1c | MORITA Kazutaka | uint8_t proto_ver; |
99 | 33b1db1c | MORITA Kazutaka | uint8_t opcode; |
100 | 33b1db1c | MORITA Kazutaka | uint16_t flags; |
101 | 33b1db1c | MORITA Kazutaka | uint32_t epoch; |
102 | 33b1db1c | MORITA Kazutaka | uint32_t id; |
103 | 33b1db1c | MORITA Kazutaka | uint32_t data_length; |
104 | 33b1db1c | MORITA Kazutaka | uint32_t result; |
105 | 33b1db1c | MORITA Kazutaka | uint32_t opcode_specific[7];
|
106 | 33b1db1c | MORITA Kazutaka | } SheepdogRsp; |
107 | 33b1db1c | MORITA Kazutaka | |
108 | 33b1db1c | MORITA Kazutaka | typedef struct SheepdogObjReq { |
109 | 33b1db1c | MORITA Kazutaka | uint8_t proto_ver; |
110 | 33b1db1c | MORITA Kazutaka | uint8_t opcode; |
111 | 33b1db1c | MORITA Kazutaka | uint16_t flags; |
112 | 33b1db1c | MORITA Kazutaka | uint32_t epoch; |
113 | 33b1db1c | MORITA Kazutaka | uint32_t id; |
114 | 33b1db1c | MORITA Kazutaka | uint32_t data_length; |
115 | 33b1db1c | MORITA Kazutaka | uint64_t oid; |
116 | 33b1db1c | MORITA Kazutaka | uint64_t cow_oid; |
117 | 33b1db1c | MORITA Kazutaka | uint32_t copies; |
118 | 33b1db1c | MORITA Kazutaka | uint32_t rsvd; |
119 | 33b1db1c | MORITA Kazutaka | uint64_t offset; |
120 | 33b1db1c | MORITA Kazutaka | } SheepdogObjReq; |
121 | 33b1db1c | MORITA Kazutaka | |
122 | 33b1db1c | MORITA Kazutaka | typedef struct SheepdogObjRsp { |
123 | 33b1db1c | MORITA Kazutaka | uint8_t proto_ver; |
124 | 33b1db1c | MORITA Kazutaka | uint8_t opcode; |
125 | 33b1db1c | MORITA Kazutaka | uint16_t flags; |
126 | 33b1db1c | MORITA Kazutaka | uint32_t epoch; |
127 | 33b1db1c | MORITA Kazutaka | uint32_t id; |
128 | 33b1db1c | MORITA Kazutaka | uint32_t data_length; |
129 | 33b1db1c | MORITA Kazutaka | uint32_t result; |
130 | 33b1db1c | MORITA Kazutaka | uint32_t copies; |
131 | 33b1db1c | MORITA Kazutaka | uint32_t pad[6];
|
132 | 33b1db1c | MORITA Kazutaka | } SheepdogObjRsp; |
133 | 33b1db1c | MORITA Kazutaka | |
134 | 33b1db1c | MORITA Kazutaka | typedef struct SheepdogVdiReq { |
135 | 33b1db1c | MORITA Kazutaka | uint8_t proto_ver; |
136 | 33b1db1c | MORITA Kazutaka | uint8_t opcode; |
137 | 33b1db1c | MORITA Kazutaka | uint16_t flags; |
138 | 33b1db1c | MORITA Kazutaka | uint32_t epoch; |
139 | 33b1db1c | MORITA Kazutaka | uint32_t id; |
140 | 33b1db1c | MORITA Kazutaka | uint32_t data_length; |
141 | 33b1db1c | MORITA Kazutaka | uint64_t vdi_size; |
142 | 33b1db1c | MORITA Kazutaka | uint32_t base_vdi_id; |
143 | 33b1db1c | MORITA Kazutaka | uint32_t copies; |
144 | 33b1db1c | MORITA Kazutaka | uint32_t snapid; |
145 | 33b1db1c | MORITA Kazutaka | uint32_t pad[3];
|
146 | 33b1db1c | MORITA Kazutaka | } SheepdogVdiReq; |
147 | 33b1db1c | MORITA Kazutaka | |
148 | 33b1db1c | MORITA Kazutaka | typedef struct SheepdogVdiRsp { |
149 | 33b1db1c | MORITA Kazutaka | uint8_t proto_ver; |
150 | 33b1db1c | MORITA Kazutaka | uint8_t opcode; |
151 | 33b1db1c | MORITA Kazutaka | uint16_t flags; |
152 | 33b1db1c | MORITA Kazutaka | uint32_t epoch; |
153 | 33b1db1c | MORITA Kazutaka | uint32_t id; |
154 | 33b1db1c | MORITA Kazutaka | uint32_t data_length; |
155 | 33b1db1c | MORITA Kazutaka | uint32_t result; |
156 | 33b1db1c | MORITA Kazutaka | uint32_t rsvd; |
157 | 33b1db1c | MORITA Kazutaka | uint32_t vdi_id; |
158 | 33b1db1c | MORITA Kazutaka | uint32_t pad[5];
|
159 | 33b1db1c | MORITA Kazutaka | } SheepdogVdiRsp; |
160 | 33b1db1c | MORITA Kazutaka | |
161 | 33b1db1c | MORITA Kazutaka | typedef struct SheepdogInode { |
162 | 33b1db1c | MORITA Kazutaka | char name[SD_MAX_VDI_LEN];
|
163 | 33b1db1c | MORITA Kazutaka | char tag[SD_MAX_VDI_TAG_LEN];
|
164 | 33b1db1c | MORITA Kazutaka | uint64_t ctime; |
165 | 33b1db1c | MORITA Kazutaka | uint64_t snap_ctime; |
166 | 33b1db1c | MORITA Kazutaka | uint64_t vm_clock_nsec; |
167 | 33b1db1c | MORITA Kazutaka | uint64_t vdi_size; |
168 | 33b1db1c | MORITA Kazutaka | uint64_t vm_state_size; |
169 | 33b1db1c | MORITA Kazutaka | uint16_t copy_policy; |
170 | 33b1db1c | MORITA Kazutaka | uint8_t nr_copies; |
171 | 33b1db1c | MORITA Kazutaka | uint8_t block_size_shift; |
172 | 33b1db1c | MORITA Kazutaka | uint32_t snap_id; |
173 | 33b1db1c | MORITA Kazutaka | uint32_t vdi_id; |
174 | 33b1db1c | MORITA Kazutaka | uint32_t parent_vdi_id; |
175 | 33b1db1c | MORITA Kazutaka | uint32_t child_vdi_id[MAX_CHILDREN]; |
176 | 33b1db1c | MORITA Kazutaka | uint32_t data_vdi_id[MAX_DATA_OBJS]; |
177 | 33b1db1c | MORITA Kazutaka | } SheepdogInode; |
178 | 33b1db1c | MORITA Kazutaka | |
179 | 33b1db1c | MORITA Kazutaka | /*
|
180 | 33b1db1c | MORITA Kazutaka | * 64 bit FNV-1a non-zero initial basis
|
181 | 33b1db1c | MORITA Kazutaka | */
|
182 | 33b1db1c | MORITA Kazutaka | #define FNV1A_64_INIT ((uint64_t)0xcbf29ce484222325ULL) |
183 | 33b1db1c | MORITA Kazutaka | |
184 | 33b1db1c | MORITA Kazutaka | /*
|
185 | 33b1db1c | MORITA Kazutaka | * 64 bit Fowler/Noll/Vo FNV-1a hash code
|
186 | 33b1db1c | MORITA Kazutaka | */
|
187 | 33b1db1c | MORITA Kazutaka | static inline uint64_t fnv_64a_buf(void *buf, size_t len, uint64_t hval) |
188 | 33b1db1c | MORITA Kazutaka | { |
189 | 33b1db1c | MORITA Kazutaka | unsigned char *bp = buf; |
190 | 33b1db1c | MORITA Kazutaka | unsigned char *be = bp + len; |
191 | 33b1db1c | MORITA Kazutaka | while (bp < be) {
|
192 | 33b1db1c | MORITA Kazutaka | hval ^= (uint64_t) *bp++; |
193 | 33b1db1c | MORITA Kazutaka | hval += (hval << 1) + (hval << 4) + (hval << 5) + |
194 | 33b1db1c | MORITA Kazutaka | (hval << 7) + (hval << 8) + (hval << 40); |
195 | 33b1db1c | MORITA Kazutaka | } |
196 | 33b1db1c | MORITA Kazutaka | return hval;
|
197 | 33b1db1c | MORITA Kazutaka | } |
198 | 33b1db1c | MORITA Kazutaka | |
199 | ebabb67a | Stefan Weil | static inline int is_data_obj_writable(SheepdogInode *inode, unsigned int idx) |
200 | 33b1db1c | MORITA Kazutaka | { |
201 | 33b1db1c | MORITA Kazutaka | return inode->vdi_id == inode->data_vdi_id[idx];
|
202 | 33b1db1c | MORITA Kazutaka | } |
203 | 33b1db1c | MORITA Kazutaka | |
204 | 33b1db1c | MORITA Kazutaka | static inline int is_data_obj(uint64_t oid) |
205 | 33b1db1c | MORITA Kazutaka | { |
206 | 33b1db1c | MORITA Kazutaka | return !(VDI_BIT & oid);
|
207 | 33b1db1c | MORITA Kazutaka | } |
208 | 33b1db1c | MORITA Kazutaka | |
209 | 33b1db1c | MORITA Kazutaka | static inline uint64_t data_oid_to_idx(uint64_t oid) |
210 | 33b1db1c | MORITA Kazutaka | { |
211 | 33b1db1c | MORITA Kazutaka | return oid & (MAX_DATA_OBJS - 1); |
212 | 33b1db1c | MORITA Kazutaka | } |
213 | 33b1db1c | MORITA Kazutaka | |
214 | 33b1db1c | MORITA Kazutaka | static inline uint64_t vid_to_vdi_oid(uint32_t vid) |
215 | 33b1db1c | MORITA Kazutaka | { |
216 | 33b1db1c | MORITA Kazutaka | return VDI_BIT | ((uint64_t)vid << VDI_SPACE_SHIFT);
|
217 | 33b1db1c | MORITA Kazutaka | } |
218 | 33b1db1c | MORITA Kazutaka | |
219 | 33b1db1c | MORITA Kazutaka | static inline uint64_t vid_to_vmstate_oid(uint32_t vid, uint32_t idx) |
220 | 33b1db1c | MORITA Kazutaka | { |
221 | 33b1db1c | MORITA Kazutaka | return VMSTATE_BIT | ((uint64_t)vid << VDI_SPACE_SHIFT) | idx;
|
222 | 33b1db1c | MORITA Kazutaka | } |
223 | 33b1db1c | MORITA Kazutaka | |
224 | 33b1db1c | MORITA Kazutaka | static inline uint64_t vid_to_data_oid(uint32_t vid, uint32_t idx) |
225 | 33b1db1c | MORITA Kazutaka | { |
226 | 33b1db1c | MORITA Kazutaka | return ((uint64_t)vid << VDI_SPACE_SHIFT) | idx;
|
227 | 33b1db1c | MORITA Kazutaka | } |
228 | 33b1db1c | MORITA Kazutaka | |
229 | 33b1db1c | MORITA Kazutaka | static inline int is_snapshot(struct SheepdogInode *inode) |
230 | 33b1db1c | MORITA Kazutaka | { |
231 | 33b1db1c | MORITA Kazutaka | return !!inode->snap_ctime;
|
232 | 33b1db1c | MORITA Kazutaka | } |
233 | 33b1db1c | MORITA Kazutaka | |
234 | 33b1db1c | MORITA Kazutaka | #undef dprintf
|
235 | 33b1db1c | MORITA Kazutaka | #ifdef DEBUG_SDOG
|
236 | 33b1db1c | MORITA Kazutaka | #define dprintf(fmt, args...) \
|
237 | 33b1db1c | MORITA Kazutaka | do { \
|
238 | 33b1db1c | MORITA Kazutaka | fprintf(stdout, "%s %d: " fmt, __func__, __LINE__, ##args); \ |
239 | 33b1db1c | MORITA Kazutaka | } while (0) |
240 | 33b1db1c | MORITA Kazutaka | #else
|
241 | 33b1db1c | MORITA Kazutaka | #define dprintf(fmt, args...)
|
242 | 33b1db1c | MORITA Kazutaka | #endif
|
243 | 33b1db1c | MORITA Kazutaka | |
244 | 33b1db1c | MORITA Kazutaka | typedef struct SheepdogAIOCB SheepdogAIOCB; |
245 | 33b1db1c | MORITA Kazutaka | |
246 | 33b1db1c | MORITA Kazutaka | typedef struct AIOReq { |
247 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *aiocb; |
248 | 33b1db1c | MORITA Kazutaka | unsigned int iov_offset; |
249 | 33b1db1c | MORITA Kazutaka | |
250 | 33b1db1c | MORITA Kazutaka | uint64_t oid; |
251 | 33b1db1c | MORITA Kazutaka | uint64_t base_oid; |
252 | 33b1db1c | MORITA Kazutaka | uint64_t offset; |
253 | 33b1db1c | MORITA Kazutaka | unsigned int data_len; |
254 | 33b1db1c | MORITA Kazutaka | uint8_t flags; |
255 | 33b1db1c | MORITA Kazutaka | uint32_t id; |
256 | 33b1db1c | MORITA Kazutaka | |
257 | 33b1db1c | MORITA Kazutaka | QLIST_ENTRY(AIOReq) outstanding_aio_siblings; |
258 | 33b1db1c | MORITA Kazutaka | QLIST_ENTRY(AIOReq) aioreq_siblings; |
259 | 33b1db1c | MORITA Kazutaka | } AIOReq; |
260 | 33b1db1c | MORITA Kazutaka | |
261 | 33b1db1c | MORITA Kazutaka | enum AIOCBState {
|
262 | 33b1db1c | MORITA Kazutaka | AIOCB_WRITE_UDATA, |
263 | 33b1db1c | MORITA Kazutaka | AIOCB_READ_UDATA, |
264 | 33b1db1c | MORITA Kazutaka | }; |
265 | 33b1db1c | MORITA Kazutaka | |
266 | 33b1db1c | MORITA Kazutaka | struct SheepdogAIOCB {
|
267 | 33b1db1c | MORITA Kazutaka | BlockDriverAIOCB common; |
268 | 33b1db1c | MORITA Kazutaka | |
269 | 33b1db1c | MORITA Kazutaka | QEMUIOVector *qiov; |
270 | 33b1db1c | MORITA Kazutaka | |
271 | 33b1db1c | MORITA Kazutaka | int64_t sector_num; |
272 | 33b1db1c | MORITA Kazutaka | int nb_sectors;
|
273 | 33b1db1c | MORITA Kazutaka | |
274 | 33b1db1c | MORITA Kazutaka | int ret;
|
275 | 33b1db1c | MORITA Kazutaka | enum AIOCBState aiocb_type;
|
276 | 33b1db1c | MORITA Kazutaka | |
277 | 33b1db1c | MORITA Kazutaka | QEMUBH *bh; |
278 | 33b1db1c | MORITA Kazutaka | void (*aio_done_func)(SheepdogAIOCB *);
|
279 | 33b1db1c | MORITA Kazutaka | |
280 | 33b1db1c | MORITA Kazutaka | int canceled;
|
281 | 33b1db1c | MORITA Kazutaka | |
282 | 33b1db1c | MORITA Kazutaka | QLIST_HEAD(aioreq_head, AIOReq) aioreq_head; |
283 | 33b1db1c | MORITA Kazutaka | }; |
284 | 33b1db1c | MORITA Kazutaka | |
285 | 33b1db1c | MORITA Kazutaka | typedef struct BDRVSheepdogState { |
286 | 33b1db1c | MORITA Kazutaka | SheepdogInode inode; |
287 | 33b1db1c | MORITA Kazutaka | |
288 | 33b1db1c | MORITA Kazutaka | uint32_t min_dirty_data_idx; |
289 | 33b1db1c | MORITA Kazutaka | uint32_t max_dirty_data_idx; |
290 | 33b1db1c | MORITA Kazutaka | |
291 | 33b1db1c | MORITA Kazutaka | char name[SD_MAX_VDI_LEN];
|
292 | 33b1db1c | MORITA Kazutaka | int is_snapshot;
|
293 | 33b1db1c | MORITA Kazutaka | |
294 | 33b1db1c | MORITA Kazutaka | char *addr;
|
295 | 33b1db1c | MORITA Kazutaka | char *port;
|
296 | 33b1db1c | MORITA Kazutaka | int fd;
|
297 | 33b1db1c | MORITA Kazutaka | |
298 | 33b1db1c | MORITA Kazutaka | uint32_t aioreq_seq_num; |
299 | 33b1db1c | MORITA Kazutaka | QLIST_HEAD(outstanding_aio_head, AIOReq) outstanding_aio_head; |
300 | 33b1db1c | MORITA Kazutaka | } BDRVSheepdogState; |
301 | 33b1db1c | MORITA Kazutaka | |
302 | 33b1db1c | MORITA Kazutaka | static const char * sd_strerror(int err) |
303 | 33b1db1c | MORITA Kazutaka | { |
304 | 33b1db1c | MORITA Kazutaka | int i;
|
305 | 33b1db1c | MORITA Kazutaka | |
306 | 33b1db1c | MORITA Kazutaka | static const struct { |
307 | 33b1db1c | MORITA Kazutaka | int err;
|
308 | 33b1db1c | MORITA Kazutaka | const char *desc; |
309 | 33b1db1c | MORITA Kazutaka | } errors[] = { |
310 | 33b1db1c | MORITA Kazutaka | {SD_RES_SUCCESS, "Success"},
|
311 | 33b1db1c | MORITA Kazutaka | {SD_RES_UNKNOWN, "Unknown error"},
|
312 | 33b1db1c | MORITA Kazutaka | {SD_RES_NO_OBJ, "No object found"},
|
313 | 33b1db1c | MORITA Kazutaka | {SD_RES_EIO, "I/O error"},
|
314 | 33b1db1c | MORITA Kazutaka | {SD_RES_VDI_EXIST, "VDI exists already"},
|
315 | 33b1db1c | MORITA Kazutaka | {SD_RES_INVALID_PARMS, "Invalid parameters"},
|
316 | 33b1db1c | MORITA Kazutaka | {SD_RES_SYSTEM_ERROR, "System error"},
|
317 | 33b1db1c | MORITA Kazutaka | {SD_RES_VDI_LOCKED, "VDI is already locked"},
|
318 | 33b1db1c | MORITA Kazutaka | {SD_RES_NO_VDI, "No vdi found"},
|
319 | 33b1db1c | MORITA Kazutaka | {SD_RES_NO_BASE_VDI, "No base VDI found"},
|
320 | 33b1db1c | MORITA Kazutaka | {SD_RES_VDI_READ, "Failed read the requested VDI"},
|
321 | 33b1db1c | MORITA Kazutaka | {SD_RES_VDI_WRITE, "Failed to write the requested VDI"},
|
322 | 33b1db1c | MORITA Kazutaka | {SD_RES_BASE_VDI_READ, "Failed to read the base VDI"},
|
323 | 33b1db1c | MORITA Kazutaka | {SD_RES_BASE_VDI_WRITE, "Failed to write the base VDI"},
|
324 | 33b1db1c | MORITA Kazutaka | {SD_RES_NO_TAG, "Failed to find the requested tag"},
|
325 | 33b1db1c | MORITA Kazutaka | {SD_RES_STARTUP, "The system is still booting"},
|
326 | 33b1db1c | MORITA Kazutaka | {SD_RES_VDI_NOT_LOCKED, "VDI isn't locked"},
|
327 | 33b1db1c | MORITA Kazutaka | {SD_RES_SHUTDOWN, "The system is shutting down"},
|
328 | 33b1db1c | MORITA Kazutaka | {SD_RES_NO_MEM, "Out of memory on the server"},
|
329 | 33b1db1c | MORITA Kazutaka | {SD_RES_FULL_VDI, "We already have the maximum vdis"},
|
330 | 33b1db1c | MORITA Kazutaka | {SD_RES_VER_MISMATCH, "Protocol version mismatch"},
|
331 | 33b1db1c | MORITA Kazutaka | {SD_RES_NO_SPACE, "Server has no space for new objects"},
|
332 | 33b1db1c | MORITA Kazutaka | {SD_RES_WAIT_FOR_FORMAT, "Sheepdog is waiting for a format operation"},
|
333 | 33b1db1c | MORITA Kazutaka | {SD_RES_WAIT_FOR_JOIN, "Sheepdog is waiting for other nodes joining"},
|
334 | 33b1db1c | MORITA Kazutaka | {SD_RES_JOIN_FAILED, "Target node had failed to join sheepdog"},
|
335 | 33b1db1c | MORITA Kazutaka | }; |
336 | 33b1db1c | MORITA Kazutaka | |
337 | 33b1db1c | MORITA Kazutaka | for (i = 0; i < ARRAY_SIZE(errors); ++i) { |
338 | 33b1db1c | MORITA Kazutaka | if (errors[i].err == err) {
|
339 | 33b1db1c | MORITA Kazutaka | return errors[i].desc;
|
340 | 33b1db1c | MORITA Kazutaka | } |
341 | 33b1db1c | MORITA Kazutaka | } |
342 | 33b1db1c | MORITA Kazutaka | |
343 | 33b1db1c | MORITA Kazutaka | return "Invalid error code"; |
344 | 33b1db1c | MORITA Kazutaka | } |
345 | 33b1db1c | MORITA Kazutaka | |
346 | 33b1db1c | MORITA Kazutaka | /*
|
347 | 33b1db1c | MORITA Kazutaka | * Sheepdog I/O handling:
|
348 | 33b1db1c | MORITA Kazutaka | *
|
349 | 33b1db1c | MORITA Kazutaka | * 1. In the sd_aio_readv/writev, read/write requests are added to the
|
350 | 33b1db1c | MORITA Kazutaka | * QEMU Bottom Halves.
|
351 | 33b1db1c | MORITA Kazutaka | *
|
352 | 33b1db1c | MORITA Kazutaka | * 2. In sd_readv_writev_bh_cb, the callbacks of BHs, we send the I/O
|
353 | 33b1db1c | MORITA Kazutaka | * requests to the server and link the requests to the
|
354 | 33b1db1c | MORITA Kazutaka | * outstanding_list in the BDRVSheepdogState. we exits the
|
355 | 33b1db1c | MORITA Kazutaka | * function without waiting for receiving the response.
|
356 | 33b1db1c | MORITA Kazutaka | *
|
357 | 33b1db1c | MORITA Kazutaka | * 3. We receive the response in aio_read_response, the fd handler to
|
358 | 33b1db1c | MORITA Kazutaka | * the sheepdog connection. If metadata update is needed, we send
|
359 | 33b1db1c | MORITA Kazutaka | * the write request to the vdi object in sd_write_done, the write
|
360 | 33b1db1c | MORITA Kazutaka | * completion function. The AIOCB callback is not called until all
|
361 | 33b1db1c | MORITA Kazutaka | * the requests belonging to the AIOCB are finished.
|
362 | 33b1db1c | MORITA Kazutaka | */
|
363 | 33b1db1c | MORITA Kazutaka | |
364 | 33b1db1c | MORITA Kazutaka | static inline AIOReq *alloc_aio_req(BDRVSheepdogState *s, SheepdogAIOCB *acb, |
365 | 33b1db1c | MORITA Kazutaka | uint64_t oid, unsigned int data_len, |
366 | 33b1db1c | MORITA Kazutaka | uint64_t offset, uint8_t flags, |
367 | 33b1db1c | MORITA Kazutaka | uint64_t base_oid, unsigned int iov_offset) |
368 | 33b1db1c | MORITA Kazutaka | { |
369 | 33b1db1c | MORITA Kazutaka | AIOReq *aio_req; |
370 | 33b1db1c | MORITA Kazutaka | |
371 | 33b1db1c | MORITA Kazutaka | aio_req = qemu_malloc(sizeof(*aio_req));
|
372 | 33b1db1c | MORITA Kazutaka | aio_req->aiocb = acb; |
373 | 33b1db1c | MORITA Kazutaka | aio_req->iov_offset = iov_offset; |
374 | 33b1db1c | MORITA Kazutaka | aio_req->oid = oid; |
375 | 33b1db1c | MORITA Kazutaka | aio_req->base_oid = base_oid; |
376 | 33b1db1c | MORITA Kazutaka | aio_req->offset = offset; |
377 | 33b1db1c | MORITA Kazutaka | aio_req->data_len = data_len; |
378 | 33b1db1c | MORITA Kazutaka | aio_req->flags = flags; |
379 | 33b1db1c | MORITA Kazutaka | aio_req->id = s->aioreq_seq_num++; |
380 | 33b1db1c | MORITA Kazutaka | |
381 | 33b1db1c | MORITA Kazutaka | QLIST_INSERT_HEAD(&s->outstanding_aio_head, aio_req, |
382 | 33b1db1c | MORITA Kazutaka | outstanding_aio_siblings); |
383 | 33b1db1c | MORITA Kazutaka | QLIST_INSERT_HEAD(&acb->aioreq_head, aio_req, aioreq_siblings); |
384 | 33b1db1c | MORITA Kazutaka | |
385 | 33b1db1c | MORITA Kazutaka | return aio_req;
|
386 | 33b1db1c | MORITA Kazutaka | } |
387 | 33b1db1c | MORITA Kazutaka | |
388 | 33b1db1c | MORITA Kazutaka | static inline int free_aio_req(BDRVSheepdogState *s, AIOReq *aio_req) |
389 | 33b1db1c | MORITA Kazutaka | { |
390 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *acb = aio_req->aiocb; |
391 | 33b1db1c | MORITA Kazutaka | QLIST_REMOVE(aio_req, outstanding_aio_siblings); |
392 | 33b1db1c | MORITA Kazutaka | QLIST_REMOVE(aio_req, aioreq_siblings); |
393 | 33b1db1c | MORITA Kazutaka | qemu_free(aio_req); |
394 | 33b1db1c | MORITA Kazutaka | |
395 | 33b1db1c | MORITA Kazutaka | return !QLIST_EMPTY(&acb->aioreq_head);
|
396 | 33b1db1c | MORITA Kazutaka | } |
397 | 33b1db1c | MORITA Kazutaka | |
398 | 33b1db1c | MORITA Kazutaka | static void sd_finish_aiocb(SheepdogAIOCB *acb) |
399 | 33b1db1c | MORITA Kazutaka | { |
400 | 33b1db1c | MORITA Kazutaka | if (!acb->canceled) {
|
401 | 33b1db1c | MORITA Kazutaka | acb->common.cb(acb->common.opaque, acb->ret); |
402 | 33b1db1c | MORITA Kazutaka | } |
403 | 33b1db1c | MORITA Kazutaka | qemu_aio_release(acb); |
404 | 33b1db1c | MORITA Kazutaka | } |
405 | 33b1db1c | MORITA Kazutaka | |
406 | 33b1db1c | MORITA Kazutaka | static void sd_aio_cancel(BlockDriverAIOCB *blockacb) |
407 | 33b1db1c | MORITA Kazutaka | { |
408 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *acb = (SheepdogAIOCB *)blockacb; |
409 | 33b1db1c | MORITA Kazutaka | |
410 | 33b1db1c | MORITA Kazutaka | /*
|
411 | 33b1db1c | MORITA Kazutaka | * Sheepdog cannot cancel the requests which are already sent to
|
412 | 33b1db1c | MORITA Kazutaka | * the servers, so we just complete the request with -EIO here.
|
413 | 33b1db1c | MORITA Kazutaka | */
|
414 | 33b1db1c | MORITA Kazutaka | acb->common.cb(acb->common.opaque, -EIO); |
415 | 33b1db1c | MORITA Kazutaka | acb->canceled = 1;
|
416 | 33b1db1c | MORITA Kazutaka | } |
417 | 33b1db1c | MORITA Kazutaka | |
418 | 33b1db1c | MORITA Kazutaka | static AIOPool sd_aio_pool = {
|
419 | 33b1db1c | MORITA Kazutaka | .aiocb_size = sizeof(SheepdogAIOCB),
|
420 | 33b1db1c | MORITA Kazutaka | .cancel = sd_aio_cancel, |
421 | 33b1db1c | MORITA Kazutaka | }; |
422 | 33b1db1c | MORITA Kazutaka | |
423 | 33b1db1c | MORITA Kazutaka | static SheepdogAIOCB *sd_aio_setup(BlockDriverState *bs, QEMUIOVector *qiov,
|
424 | 33b1db1c | MORITA Kazutaka | int64_t sector_num, int nb_sectors,
|
425 | 33b1db1c | MORITA Kazutaka | BlockDriverCompletionFunc *cb, void *opaque)
|
426 | 33b1db1c | MORITA Kazutaka | { |
427 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *acb; |
428 | 33b1db1c | MORITA Kazutaka | |
429 | 33b1db1c | MORITA Kazutaka | acb = qemu_aio_get(&sd_aio_pool, bs, cb, opaque); |
430 | 33b1db1c | MORITA Kazutaka | |
431 | 33b1db1c | MORITA Kazutaka | acb->qiov = qiov; |
432 | 33b1db1c | MORITA Kazutaka | |
433 | 33b1db1c | MORITA Kazutaka | acb->sector_num = sector_num; |
434 | 33b1db1c | MORITA Kazutaka | acb->nb_sectors = nb_sectors; |
435 | 33b1db1c | MORITA Kazutaka | |
436 | 33b1db1c | MORITA Kazutaka | acb->aio_done_func = NULL;
|
437 | 33b1db1c | MORITA Kazutaka | acb->canceled = 0;
|
438 | 33b1db1c | MORITA Kazutaka | acb->bh = NULL;
|
439 | 33b1db1c | MORITA Kazutaka | acb->ret = 0;
|
440 | 33b1db1c | MORITA Kazutaka | QLIST_INIT(&acb->aioreq_head); |
441 | 33b1db1c | MORITA Kazutaka | return acb;
|
442 | 33b1db1c | MORITA Kazutaka | } |
443 | 33b1db1c | MORITA Kazutaka | |
444 | 33b1db1c | MORITA Kazutaka | static int sd_schedule_bh(QEMUBHFunc *cb, SheepdogAIOCB *acb) |
445 | 33b1db1c | MORITA Kazutaka | { |
446 | 33b1db1c | MORITA Kazutaka | if (acb->bh) {
|
447 | 6daf194d | Markus Armbruster | error_report("bug: %d %d", acb->aiocb_type, acb->aiocb_type);
|
448 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
449 | 33b1db1c | MORITA Kazutaka | } |
450 | 33b1db1c | MORITA Kazutaka | |
451 | 33b1db1c | MORITA Kazutaka | acb->bh = qemu_bh_new(cb, acb); |
452 | 33b1db1c | MORITA Kazutaka | qemu_bh_schedule(acb->bh); |
453 | 33b1db1c | MORITA Kazutaka | return 0; |
454 | 33b1db1c | MORITA Kazutaka | } |
455 | 33b1db1c | MORITA Kazutaka | |
456 | 33b1db1c | MORITA Kazutaka | #ifdef _WIN32
|
457 | 33b1db1c | MORITA Kazutaka | |
458 | 33b1db1c | MORITA Kazutaka | struct msghdr {
|
459 | 33b1db1c | MORITA Kazutaka | struct iovec *msg_iov;
|
460 | 33b1db1c | MORITA Kazutaka | size_t msg_iovlen; |
461 | 33b1db1c | MORITA Kazutaka | }; |
462 | 33b1db1c | MORITA Kazutaka | |
463 | 33b1db1c | MORITA Kazutaka | static ssize_t sendmsg(int s, const struct msghdr *msg, int flags) |
464 | 33b1db1c | MORITA Kazutaka | { |
465 | 33b1db1c | MORITA Kazutaka | size_t size = 0;
|
466 | 33b1db1c | MORITA Kazutaka | char *buf, *p;
|
467 | 33b1db1c | MORITA Kazutaka | int i, ret;
|
468 | 33b1db1c | MORITA Kazutaka | |
469 | 33b1db1c | MORITA Kazutaka | /* count the msg size */
|
470 | 33b1db1c | MORITA Kazutaka | for (i = 0; i < msg->msg_iovlen; i++) { |
471 | 33b1db1c | MORITA Kazutaka | size += msg->msg_iov[i].iov_len; |
472 | 33b1db1c | MORITA Kazutaka | } |
473 | 33b1db1c | MORITA Kazutaka | buf = qemu_malloc(size); |
474 | 33b1db1c | MORITA Kazutaka | |
475 | 33b1db1c | MORITA Kazutaka | p = buf; |
476 | 33b1db1c | MORITA Kazutaka | for (i = 0; i < msg->msg_iovlen; i++) { |
477 | 33b1db1c | MORITA Kazutaka | memcpy(p, msg->msg_iov[i].iov_base, msg->msg_iov[i].iov_len); |
478 | 33b1db1c | MORITA Kazutaka | p += msg->msg_iov[i].iov_len; |
479 | 33b1db1c | MORITA Kazutaka | } |
480 | 33b1db1c | MORITA Kazutaka | |
481 | 33b1db1c | MORITA Kazutaka | ret = send(s, buf, size, flags); |
482 | 33b1db1c | MORITA Kazutaka | |
483 | 33b1db1c | MORITA Kazutaka | qemu_free(buf); |
484 | 33b1db1c | MORITA Kazutaka | return ret;
|
485 | 33b1db1c | MORITA Kazutaka | } |
486 | 33b1db1c | MORITA Kazutaka | |
487 | 33b1db1c | MORITA Kazutaka | static ssize_t recvmsg(int s, struct msghdr *msg, int flags) |
488 | 33b1db1c | MORITA Kazutaka | { |
489 | 33b1db1c | MORITA Kazutaka | size_t size = 0;
|
490 | 33b1db1c | MORITA Kazutaka | char *buf, *p;
|
491 | 33b1db1c | MORITA Kazutaka | int i, ret;
|
492 | 33b1db1c | MORITA Kazutaka | |
493 | 33b1db1c | MORITA Kazutaka | /* count the msg size */
|
494 | 33b1db1c | MORITA Kazutaka | for (i = 0; i < msg->msg_iovlen; i++) { |
495 | 33b1db1c | MORITA Kazutaka | size += msg->msg_iov[i].iov_len; |
496 | 33b1db1c | MORITA Kazutaka | } |
497 | 33b1db1c | MORITA Kazutaka | buf = qemu_malloc(size); |
498 | 33b1db1c | MORITA Kazutaka | |
499 | 33b1db1c | MORITA Kazutaka | ret = recv(s, buf, size, flags); |
500 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
501 | 33b1db1c | MORITA Kazutaka | goto out;
|
502 | 33b1db1c | MORITA Kazutaka | } |
503 | 33b1db1c | MORITA Kazutaka | |
504 | 33b1db1c | MORITA Kazutaka | p = buf; |
505 | 33b1db1c | MORITA Kazutaka | for (i = 0; i < msg->msg_iovlen; i++) { |
506 | 33b1db1c | MORITA Kazutaka | memcpy(msg->msg_iov[i].iov_base, p, msg->msg_iov[i].iov_len); |
507 | 33b1db1c | MORITA Kazutaka | p += msg->msg_iov[i].iov_len; |
508 | 33b1db1c | MORITA Kazutaka | } |
509 | 33b1db1c | MORITA Kazutaka | out:
|
510 | 33b1db1c | MORITA Kazutaka | qemu_free(buf); |
511 | 33b1db1c | MORITA Kazutaka | return ret;
|
512 | 33b1db1c | MORITA Kazutaka | } |
513 | 33b1db1c | MORITA Kazutaka | |
514 | 33b1db1c | MORITA Kazutaka | #endif
|
515 | 33b1db1c | MORITA Kazutaka | |
516 | 33b1db1c | MORITA Kazutaka | /*
|
517 | 33b1db1c | MORITA Kazutaka | * Send/recv data with iovec buffers
|
518 | 33b1db1c | MORITA Kazutaka | *
|
519 | 33b1db1c | MORITA Kazutaka | * This function send/recv data from/to the iovec buffer directly.
|
520 | 33b1db1c | MORITA Kazutaka | * The first `offset' bytes in the iovec buffer are skipped and next
|
521 | 33b1db1c | MORITA Kazutaka | * `len' bytes are used.
|
522 | 33b1db1c | MORITA Kazutaka | *
|
523 | 33b1db1c | MORITA Kazutaka | * For example,
|
524 | 33b1db1c | MORITA Kazutaka | *
|
525 | 33b1db1c | MORITA Kazutaka | * do_send_recv(sockfd, iov, len, offset, 1);
|
526 | 33b1db1c | MORITA Kazutaka | *
|
527 | 33b1db1c | MORITA Kazutaka | * is equals to
|
528 | 33b1db1c | MORITA Kazutaka | *
|
529 | 33b1db1c | MORITA Kazutaka | * char *buf = malloc(size);
|
530 | 33b1db1c | MORITA Kazutaka | * iov_to_buf(iov, iovcnt, buf, offset, size);
|
531 | 33b1db1c | MORITA Kazutaka | * send(sockfd, buf, size, 0);
|
532 | 33b1db1c | MORITA Kazutaka | * free(buf);
|
533 | 33b1db1c | MORITA Kazutaka | */
|
534 | 33b1db1c | MORITA Kazutaka | static int do_send_recv(int sockfd, struct iovec *iov, int len, int offset, |
535 | 33b1db1c | MORITA Kazutaka | int write)
|
536 | 33b1db1c | MORITA Kazutaka | { |
537 | 33b1db1c | MORITA Kazutaka | struct msghdr msg;
|
538 | 33b1db1c | MORITA Kazutaka | int ret, diff;
|
539 | 33b1db1c | MORITA Kazutaka | |
540 | 33b1db1c | MORITA Kazutaka | memset(&msg, 0, sizeof(msg)); |
541 | 33b1db1c | MORITA Kazutaka | msg.msg_iov = iov; |
542 | 33b1db1c | MORITA Kazutaka | msg.msg_iovlen = 1;
|
543 | 33b1db1c | MORITA Kazutaka | |
544 | 33b1db1c | MORITA Kazutaka | len += offset; |
545 | 33b1db1c | MORITA Kazutaka | |
546 | 33b1db1c | MORITA Kazutaka | while (iov->iov_len < len) {
|
547 | 33b1db1c | MORITA Kazutaka | len -= iov->iov_len; |
548 | 33b1db1c | MORITA Kazutaka | |
549 | 33b1db1c | MORITA Kazutaka | iov++; |
550 | 33b1db1c | MORITA Kazutaka | msg.msg_iovlen++; |
551 | 33b1db1c | MORITA Kazutaka | } |
552 | 33b1db1c | MORITA Kazutaka | |
553 | 33b1db1c | MORITA Kazutaka | diff = iov->iov_len - len; |
554 | 33b1db1c | MORITA Kazutaka | iov->iov_len -= diff; |
555 | 33b1db1c | MORITA Kazutaka | |
556 | 33b1db1c | MORITA Kazutaka | while (msg.msg_iov->iov_len <= offset) {
|
557 | 33b1db1c | MORITA Kazutaka | offset -= msg.msg_iov->iov_len; |
558 | 33b1db1c | MORITA Kazutaka | |
559 | 33b1db1c | MORITA Kazutaka | msg.msg_iov++; |
560 | 33b1db1c | MORITA Kazutaka | msg.msg_iovlen--; |
561 | 33b1db1c | MORITA Kazutaka | } |
562 | 33b1db1c | MORITA Kazutaka | |
563 | 33b1db1c | MORITA Kazutaka | msg.msg_iov->iov_base = (char *) msg.msg_iov->iov_base + offset;
|
564 | 33b1db1c | MORITA Kazutaka | msg.msg_iov->iov_len -= offset; |
565 | 33b1db1c | MORITA Kazutaka | |
566 | 33b1db1c | MORITA Kazutaka | if (write) {
|
567 | 33b1db1c | MORITA Kazutaka | ret = sendmsg(sockfd, &msg, 0);
|
568 | 33b1db1c | MORITA Kazutaka | } else {
|
569 | 33b1db1c | MORITA Kazutaka | ret = recvmsg(sockfd, &msg, 0);
|
570 | 33b1db1c | MORITA Kazutaka | } |
571 | 33b1db1c | MORITA Kazutaka | |
572 | 33b1db1c | MORITA Kazutaka | msg.msg_iov->iov_base = (char *) msg.msg_iov->iov_base - offset;
|
573 | 33b1db1c | MORITA Kazutaka | msg.msg_iov->iov_len += offset; |
574 | 33b1db1c | MORITA Kazutaka | |
575 | 33b1db1c | MORITA Kazutaka | iov->iov_len += diff; |
576 | 33b1db1c | MORITA Kazutaka | return ret;
|
577 | 33b1db1c | MORITA Kazutaka | } |
578 | 33b1db1c | MORITA Kazutaka | |
579 | 33b1db1c | MORITA Kazutaka | static int connect_to_sdog(const char *addr, const char *port) |
580 | 33b1db1c | MORITA Kazutaka | { |
581 | 33b1db1c | MORITA Kazutaka | char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV];
|
582 | 33b1db1c | MORITA Kazutaka | int fd, ret;
|
583 | 33b1db1c | MORITA Kazutaka | struct addrinfo hints, *res, *res0;
|
584 | 33b1db1c | MORITA Kazutaka | |
585 | 33b1db1c | MORITA Kazutaka | if (!addr) {
|
586 | 33b1db1c | MORITA Kazutaka | addr = SD_DEFAULT_ADDR; |
587 | 33b1db1c | MORITA Kazutaka | port = SD_DEFAULT_PORT; |
588 | 33b1db1c | MORITA Kazutaka | } |
589 | 33b1db1c | MORITA Kazutaka | |
590 | 33b1db1c | MORITA Kazutaka | memset(&hints, 0, sizeof(hints)); |
591 | 33b1db1c | MORITA Kazutaka | hints.ai_socktype = SOCK_STREAM; |
592 | 33b1db1c | MORITA Kazutaka | |
593 | 33b1db1c | MORITA Kazutaka | ret = getaddrinfo(addr, port, &hints, &res0); |
594 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
595 | 6daf194d | Markus Armbruster | error_report("unable to get address info %s, %s",
|
596 | 33b1db1c | MORITA Kazutaka | addr, strerror(errno)); |
597 | 33b1db1c | MORITA Kazutaka | return -1; |
598 | 33b1db1c | MORITA Kazutaka | } |
599 | 33b1db1c | MORITA Kazutaka | |
600 | 33b1db1c | MORITA Kazutaka | for (res = res0; res; res = res->ai_next) {
|
601 | 33b1db1c | MORITA Kazutaka | ret = getnameinfo(res->ai_addr, res->ai_addrlen, hbuf, sizeof(hbuf),
|
602 | 33b1db1c | MORITA Kazutaka | sbuf, sizeof(sbuf), NI_NUMERICHOST | NI_NUMERICSERV);
|
603 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
604 | 33b1db1c | MORITA Kazutaka | continue;
|
605 | 33b1db1c | MORITA Kazutaka | } |
606 | 33b1db1c | MORITA Kazutaka | |
607 | 33b1db1c | MORITA Kazutaka | fd = socket(res->ai_family, res->ai_socktype, res->ai_protocol); |
608 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
609 | 33b1db1c | MORITA Kazutaka | continue;
|
610 | 33b1db1c | MORITA Kazutaka | } |
611 | 33b1db1c | MORITA Kazutaka | |
612 | 33b1db1c | MORITA Kazutaka | reconnect:
|
613 | 33b1db1c | MORITA Kazutaka | ret = connect(fd, res->ai_addr, res->ai_addrlen); |
614 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
615 | 33b1db1c | MORITA Kazutaka | if (errno == EINTR) {
|
616 | 33b1db1c | MORITA Kazutaka | goto reconnect;
|
617 | 33b1db1c | MORITA Kazutaka | } |
618 | 33b1db1c | MORITA Kazutaka | break;
|
619 | 33b1db1c | MORITA Kazutaka | } |
620 | 33b1db1c | MORITA Kazutaka | |
621 | 33b1db1c | MORITA Kazutaka | dprintf("connected to %s:%s\n", addr, port);
|
622 | 33b1db1c | MORITA Kazutaka | goto success;
|
623 | 33b1db1c | MORITA Kazutaka | } |
624 | 33b1db1c | MORITA Kazutaka | fd = -1;
|
625 | 6daf194d | Markus Armbruster | error_report("failed connect to %s:%s", addr, port);
|
626 | 33b1db1c | MORITA Kazutaka | success:
|
627 | 33b1db1c | MORITA Kazutaka | freeaddrinfo(res0); |
628 | 33b1db1c | MORITA Kazutaka | return fd;
|
629 | 33b1db1c | MORITA Kazutaka | } |
630 | 33b1db1c | MORITA Kazutaka | |
631 | 33b1db1c | MORITA Kazutaka | static int do_readv_writev(int sockfd, struct iovec *iov, int len, |
632 | 33b1db1c | MORITA Kazutaka | int iov_offset, int write) |
633 | 33b1db1c | MORITA Kazutaka | { |
634 | 33b1db1c | MORITA Kazutaka | int ret;
|
635 | 33b1db1c | MORITA Kazutaka | again:
|
636 | 33b1db1c | MORITA Kazutaka | ret = do_send_recv(sockfd, iov, len, iov_offset, write); |
637 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
638 | 33b1db1c | MORITA Kazutaka | if (errno == EINTR || errno == EAGAIN) {
|
639 | 33b1db1c | MORITA Kazutaka | goto again;
|
640 | 33b1db1c | MORITA Kazutaka | } |
641 | 6daf194d | Markus Armbruster | error_report("failed to recv a rsp, %s", strerror(errno));
|
642 | 33b1db1c | MORITA Kazutaka | return 1; |
643 | 33b1db1c | MORITA Kazutaka | } |
644 | 33b1db1c | MORITA Kazutaka | |
645 | 33b1db1c | MORITA Kazutaka | iov_offset += ret; |
646 | 33b1db1c | MORITA Kazutaka | len -= ret; |
647 | 33b1db1c | MORITA Kazutaka | if (len) {
|
648 | 33b1db1c | MORITA Kazutaka | goto again;
|
649 | 33b1db1c | MORITA Kazutaka | } |
650 | 33b1db1c | MORITA Kazutaka | |
651 | 33b1db1c | MORITA Kazutaka | return 0; |
652 | 33b1db1c | MORITA Kazutaka | } |
653 | 33b1db1c | MORITA Kazutaka | |
654 | 33b1db1c | MORITA Kazutaka | static int do_readv(int sockfd, struct iovec *iov, int len, int iov_offset) |
655 | 33b1db1c | MORITA Kazutaka | { |
656 | 33b1db1c | MORITA Kazutaka | return do_readv_writev(sockfd, iov, len, iov_offset, 0); |
657 | 33b1db1c | MORITA Kazutaka | } |
658 | 33b1db1c | MORITA Kazutaka | |
659 | 33b1db1c | MORITA Kazutaka | static int do_writev(int sockfd, struct iovec *iov, int len, int iov_offset) |
660 | 33b1db1c | MORITA Kazutaka | { |
661 | 33b1db1c | MORITA Kazutaka | return do_readv_writev(sockfd, iov, len, iov_offset, 1); |
662 | 33b1db1c | MORITA Kazutaka | } |
663 | 33b1db1c | MORITA Kazutaka | |
664 | 33b1db1c | MORITA Kazutaka | static int do_read_write(int sockfd, void *buf, int len, int write) |
665 | 33b1db1c | MORITA Kazutaka | { |
666 | 33b1db1c | MORITA Kazutaka | struct iovec iov;
|
667 | 33b1db1c | MORITA Kazutaka | |
668 | 33b1db1c | MORITA Kazutaka | iov.iov_base = buf; |
669 | 33b1db1c | MORITA Kazutaka | iov.iov_len = len; |
670 | 33b1db1c | MORITA Kazutaka | |
671 | 33b1db1c | MORITA Kazutaka | return do_readv_writev(sockfd, &iov, len, 0, write); |
672 | 33b1db1c | MORITA Kazutaka | } |
673 | 33b1db1c | MORITA Kazutaka | |
674 | 33b1db1c | MORITA Kazutaka | static int do_read(int sockfd, void *buf, int len) |
675 | 33b1db1c | MORITA Kazutaka | { |
676 | 33b1db1c | MORITA Kazutaka | return do_read_write(sockfd, buf, len, 0); |
677 | 33b1db1c | MORITA Kazutaka | } |
678 | 33b1db1c | MORITA Kazutaka | |
679 | 33b1db1c | MORITA Kazutaka | static int do_write(int sockfd, void *buf, int len) |
680 | 33b1db1c | MORITA Kazutaka | { |
681 | 33b1db1c | MORITA Kazutaka | return do_read_write(sockfd, buf, len, 1); |
682 | 33b1db1c | MORITA Kazutaka | } |
683 | 33b1db1c | MORITA Kazutaka | |
684 | 33b1db1c | MORITA Kazutaka | static int send_req(int sockfd, SheepdogReq *hdr, void *data, |
685 | 33b1db1c | MORITA Kazutaka | unsigned int *wlen) |
686 | 33b1db1c | MORITA Kazutaka | { |
687 | 33b1db1c | MORITA Kazutaka | int ret;
|
688 | 33b1db1c | MORITA Kazutaka | struct iovec iov[2]; |
689 | 33b1db1c | MORITA Kazutaka | |
690 | 33b1db1c | MORITA Kazutaka | iov[0].iov_base = hdr;
|
691 | 33b1db1c | MORITA Kazutaka | iov[0].iov_len = sizeof(*hdr); |
692 | 33b1db1c | MORITA Kazutaka | |
693 | 33b1db1c | MORITA Kazutaka | if (*wlen) {
|
694 | 33b1db1c | MORITA Kazutaka | iov[1].iov_base = data;
|
695 | 33b1db1c | MORITA Kazutaka | iov[1].iov_len = *wlen;
|
696 | 33b1db1c | MORITA Kazutaka | } |
697 | 33b1db1c | MORITA Kazutaka | |
698 | 33b1db1c | MORITA Kazutaka | ret = do_writev(sockfd, iov, sizeof(*hdr) + *wlen, 0); |
699 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
700 | 6daf194d | Markus Armbruster | error_report("failed to send a req, %s", strerror(errno));
|
701 | 33b1db1c | MORITA Kazutaka | ret = -1;
|
702 | 33b1db1c | MORITA Kazutaka | } |
703 | 33b1db1c | MORITA Kazutaka | |
704 | 33b1db1c | MORITA Kazutaka | return ret;
|
705 | 33b1db1c | MORITA Kazutaka | } |
706 | 33b1db1c | MORITA Kazutaka | |
707 | 33b1db1c | MORITA Kazutaka | static int do_req(int sockfd, SheepdogReq *hdr, void *data, |
708 | 33b1db1c | MORITA Kazutaka | unsigned int *wlen, unsigned int *rlen) |
709 | 33b1db1c | MORITA Kazutaka | { |
710 | 33b1db1c | MORITA Kazutaka | int ret;
|
711 | 33b1db1c | MORITA Kazutaka | |
712 | 33b1db1c | MORITA Kazutaka | ret = send_req(sockfd, hdr, data, wlen); |
713 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
714 | 33b1db1c | MORITA Kazutaka | ret = -1;
|
715 | 33b1db1c | MORITA Kazutaka | goto out;
|
716 | 33b1db1c | MORITA Kazutaka | } |
717 | 33b1db1c | MORITA Kazutaka | |
718 | 33b1db1c | MORITA Kazutaka | ret = do_read(sockfd, hdr, sizeof(*hdr));
|
719 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
720 | 6daf194d | Markus Armbruster | error_report("failed to get a rsp, %s", strerror(errno));
|
721 | 33b1db1c | MORITA Kazutaka | ret = -1;
|
722 | 33b1db1c | MORITA Kazutaka | goto out;
|
723 | 33b1db1c | MORITA Kazutaka | } |
724 | 33b1db1c | MORITA Kazutaka | |
725 | 33b1db1c | MORITA Kazutaka | if (*rlen > hdr->data_length) {
|
726 | 33b1db1c | MORITA Kazutaka | *rlen = hdr->data_length; |
727 | 33b1db1c | MORITA Kazutaka | } |
728 | 33b1db1c | MORITA Kazutaka | |
729 | 33b1db1c | MORITA Kazutaka | if (*rlen) {
|
730 | 33b1db1c | MORITA Kazutaka | ret = do_read(sockfd, data, *rlen); |
731 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
732 | 6daf194d | Markus Armbruster | error_report("failed to get the data, %s", strerror(errno));
|
733 | 33b1db1c | MORITA Kazutaka | ret = -1;
|
734 | 33b1db1c | MORITA Kazutaka | goto out;
|
735 | 33b1db1c | MORITA Kazutaka | } |
736 | 33b1db1c | MORITA Kazutaka | } |
737 | 33b1db1c | MORITA Kazutaka | ret = 0;
|
738 | 33b1db1c | MORITA Kazutaka | out:
|
739 | 33b1db1c | MORITA Kazutaka | return ret;
|
740 | 33b1db1c | MORITA Kazutaka | } |
741 | 33b1db1c | MORITA Kazutaka | |
742 | 33b1db1c | MORITA Kazutaka | static int add_aio_request(BDRVSheepdogState *s, AIOReq *aio_req, |
743 | 33b1db1c | MORITA Kazutaka | struct iovec *iov, int niov, int create, |
744 | 33b1db1c | MORITA Kazutaka | enum AIOCBState aiocb_type);
|
745 | 33b1db1c | MORITA Kazutaka | |
746 | 33b1db1c | MORITA Kazutaka | /*
|
747 | 33b1db1c | MORITA Kazutaka | * This function searchs pending requests to the object `oid', and
|
748 | 33b1db1c | MORITA Kazutaka | * sends them.
|
749 | 33b1db1c | MORITA Kazutaka | */
|
750 | 33b1db1c | MORITA Kazutaka | static void send_pending_req(BDRVSheepdogState *s, uint64_t oid, uint32_t id) |
751 | 33b1db1c | MORITA Kazutaka | { |
752 | 33b1db1c | MORITA Kazutaka | AIOReq *aio_req, *next; |
753 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *acb; |
754 | 33b1db1c | MORITA Kazutaka | int ret;
|
755 | 33b1db1c | MORITA Kazutaka | |
756 | 33b1db1c | MORITA Kazutaka | QLIST_FOREACH_SAFE(aio_req, &s->outstanding_aio_head, |
757 | 33b1db1c | MORITA Kazutaka | outstanding_aio_siblings, next) { |
758 | 33b1db1c | MORITA Kazutaka | if (id == aio_req->id) {
|
759 | 33b1db1c | MORITA Kazutaka | continue;
|
760 | 33b1db1c | MORITA Kazutaka | } |
761 | 33b1db1c | MORITA Kazutaka | if (aio_req->oid != oid) {
|
762 | 33b1db1c | MORITA Kazutaka | continue;
|
763 | 33b1db1c | MORITA Kazutaka | } |
764 | 33b1db1c | MORITA Kazutaka | |
765 | 33b1db1c | MORITA Kazutaka | acb = aio_req->aiocb; |
766 | 33b1db1c | MORITA Kazutaka | ret = add_aio_request(s, aio_req, acb->qiov->iov, |
767 | 33b1db1c | MORITA Kazutaka | acb->qiov->niov, 0, acb->aiocb_type);
|
768 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
769 | 6daf194d | Markus Armbruster | error_report("add_aio_request is failed");
|
770 | 33b1db1c | MORITA Kazutaka | free_aio_req(s, aio_req); |
771 | 33b1db1c | MORITA Kazutaka | if (QLIST_EMPTY(&acb->aioreq_head)) {
|
772 | 33b1db1c | MORITA Kazutaka | sd_finish_aiocb(acb); |
773 | 33b1db1c | MORITA Kazutaka | } |
774 | 33b1db1c | MORITA Kazutaka | } |
775 | 33b1db1c | MORITA Kazutaka | } |
776 | 33b1db1c | MORITA Kazutaka | } |
777 | 33b1db1c | MORITA Kazutaka | |
778 | 33b1db1c | MORITA Kazutaka | /*
|
779 | 33b1db1c | MORITA Kazutaka | * Receive responses of the I/O requests.
|
780 | 33b1db1c | MORITA Kazutaka | *
|
781 | 33b1db1c | MORITA Kazutaka | * This function is registered as a fd handler, and called from the
|
782 | 33b1db1c | MORITA Kazutaka | * main loop when s->fd is ready for reading responses.
|
783 | 33b1db1c | MORITA Kazutaka | */
|
784 | 33b1db1c | MORITA Kazutaka | static void aio_read_response(void *opaque) |
785 | 33b1db1c | MORITA Kazutaka | { |
786 | 33b1db1c | MORITA Kazutaka | SheepdogObjRsp rsp; |
787 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = opaque; |
788 | 33b1db1c | MORITA Kazutaka | int fd = s->fd;
|
789 | 33b1db1c | MORITA Kazutaka | int ret;
|
790 | 33b1db1c | MORITA Kazutaka | AIOReq *aio_req = NULL;
|
791 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *acb; |
792 | 33b1db1c | MORITA Kazutaka | int rest;
|
793 | 33b1db1c | MORITA Kazutaka | unsigned long idx; |
794 | 33b1db1c | MORITA Kazutaka | |
795 | 33b1db1c | MORITA Kazutaka | if (QLIST_EMPTY(&s->outstanding_aio_head)) {
|
796 | 33b1db1c | MORITA Kazutaka | return;
|
797 | 33b1db1c | MORITA Kazutaka | } |
798 | 33b1db1c | MORITA Kazutaka | |
799 | 33b1db1c | MORITA Kazutaka | /* read a header */
|
800 | 33b1db1c | MORITA Kazutaka | ret = do_read(fd, &rsp, sizeof(rsp));
|
801 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
802 | 6daf194d | Markus Armbruster | error_report("failed to get the header, %s", strerror(errno));
|
803 | 33b1db1c | MORITA Kazutaka | return;
|
804 | 33b1db1c | MORITA Kazutaka | } |
805 | 33b1db1c | MORITA Kazutaka | |
806 | 33b1db1c | MORITA Kazutaka | /* find the right aio_req from the outstanding_aio list */
|
807 | 33b1db1c | MORITA Kazutaka | QLIST_FOREACH(aio_req, &s->outstanding_aio_head, outstanding_aio_siblings) { |
808 | 33b1db1c | MORITA Kazutaka | if (aio_req->id == rsp.id) {
|
809 | 33b1db1c | MORITA Kazutaka | break;
|
810 | 33b1db1c | MORITA Kazutaka | } |
811 | 33b1db1c | MORITA Kazutaka | } |
812 | 33b1db1c | MORITA Kazutaka | if (!aio_req) {
|
813 | 6daf194d | Markus Armbruster | error_report("cannot find aio_req %x", rsp.id);
|
814 | 33b1db1c | MORITA Kazutaka | return;
|
815 | 33b1db1c | MORITA Kazutaka | } |
816 | 33b1db1c | MORITA Kazutaka | |
817 | 33b1db1c | MORITA Kazutaka | acb = aio_req->aiocb; |
818 | 33b1db1c | MORITA Kazutaka | |
819 | 33b1db1c | MORITA Kazutaka | switch (acb->aiocb_type) {
|
820 | 33b1db1c | MORITA Kazutaka | case AIOCB_WRITE_UDATA:
|
821 | 33b1db1c | MORITA Kazutaka | if (!is_data_obj(aio_req->oid)) {
|
822 | 33b1db1c | MORITA Kazutaka | break;
|
823 | 33b1db1c | MORITA Kazutaka | } |
824 | 33b1db1c | MORITA Kazutaka | idx = data_oid_to_idx(aio_req->oid); |
825 | 33b1db1c | MORITA Kazutaka | |
826 | 33b1db1c | MORITA Kazutaka | if (s->inode.data_vdi_id[idx] != s->inode.vdi_id) {
|
827 | 33b1db1c | MORITA Kazutaka | /*
|
828 | 33b1db1c | MORITA Kazutaka | * If the object is newly created one, we need to update
|
829 | 33b1db1c | MORITA Kazutaka | * the vdi object (metadata object). min_dirty_data_idx
|
830 | 33b1db1c | MORITA Kazutaka | * and max_dirty_data_idx are changed to include updated
|
831 | 33b1db1c | MORITA Kazutaka | * index between them.
|
832 | 33b1db1c | MORITA Kazutaka | */
|
833 | 33b1db1c | MORITA Kazutaka | s->inode.data_vdi_id[idx] = s->inode.vdi_id; |
834 | 33b1db1c | MORITA Kazutaka | s->max_dirty_data_idx = MAX(idx, s->max_dirty_data_idx); |
835 | 33b1db1c | MORITA Kazutaka | s->min_dirty_data_idx = MIN(idx, s->min_dirty_data_idx); |
836 | 33b1db1c | MORITA Kazutaka | |
837 | 33b1db1c | MORITA Kazutaka | /*
|
838 | 33b1db1c | MORITA Kazutaka | * Some requests may be blocked because simultaneous
|
839 | 33b1db1c | MORITA Kazutaka | * create requests are not allowed, so we search the
|
840 | 33b1db1c | MORITA Kazutaka | * pending requests here.
|
841 | 33b1db1c | MORITA Kazutaka | */
|
842 | 33b1db1c | MORITA Kazutaka | send_pending_req(s, vid_to_data_oid(s->inode.vdi_id, idx), rsp.id); |
843 | 33b1db1c | MORITA Kazutaka | } |
844 | 33b1db1c | MORITA Kazutaka | break;
|
845 | 33b1db1c | MORITA Kazutaka | case AIOCB_READ_UDATA:
|
846 | 33b1db1c | MORITA Kazutaka | ret = do_readv(fd, acb->qiov->iov, rsp.data_length, |
847 | 33b1db1c | MORITA Kazutaka | aio_req->iov_offset); |
848 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
849 | 6daf194d | Markus Armbruster | error_report("failed to get the data, %s", strerror(errno));
|
850 | 33b1db1c | MORITA Kazutaka | return;
|
851 | 33b1db1c | MORITA Kazutaka | } |
852 | 33b1db1c | MORITA Kazutaka | break;
|
853 | 33b1db1c | MORITA Kazutaka | } |
854 | 33b1db1c | MORITA Kazutaka | |
855 | 33b1db1c | MORITA Kazutaka | if (rsp.result != SD_RES_SUCCESS) {
|
856 | 33b1db1c | MORITA Kazutaka | acb->ret = -EIO; |
857 | 6daf194d | Markus Armbruster | error_report("%s", sd_strerror(rsp.result));
|
858 | 33b1db1c | MORITA Kazutaka | } |
859 | 33b1db1c | MORITA Kazutaka | |
860 | 33b1db1c | MORITA Kazutaka | rest = free_aio_req(s, aio_req); |
861 | 33b1db1c | MORITA Kazutaka | if (!rest) {
|
862 | 33b1db1c | MORITA Kazutaka | /*
|
863 | 33b1db1c | MORITA Kazutaka | * We've finished all requests which belong to the AIOCB, so
|
864 | 33b1db1c | MORITA Kazutaka | * we can call the callback now.
|
865 | 33b1db1c | MORITA Kazutaka | */
|
866 | 33b1db1c | MORITA Kazutaka | acb->aio_done_func(acb); |
867 | 33b1db1c | MORITA Kazutaka | } |
868 | 33b1db1c | MORITA Kazutaka | } |
869 | 33b1db1c | MORITA Kazutaka | |
870 | 33b1db1c | MORITA Kazutaka | static int aio_flush_request(void *opaque) |
871 | 33b1db1c | MORITA Kazutaka | { |
872 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = opaque; |
873 | 33b1db1c | MORITA Kazutaka | |
874 | 33b1db1c | MORITA Kazutaka | return !QLIST_EMPTY(&s->outstanding_aio_head);
|
875 | 33b1db1c | MORITA Kazutaka | } |
876 | 33b1db1c | MORITA Kazutaka | |
877 | 6defcc37 | MORITA Kazutaka | #if !defined(SOL_TCP) || !defined(TCP_CORK)
|
878 | 33b1db1c | MORITA Kazutaka | |
879 | 33b1db1c | MORITA Kazutaka | static int set_cork(int fd, int v) |
880 | 33b1db1c | MORITA Kazutaka | { |
881 | 33b1db1c | MORITA Kazutaka | return 0; |
882 | 33b1db1c | MORITA Kazutaka | } |
883 | 33b1db1c | MORITA Kazutaka | |
884 | 33b1db1c | MORITA Kazutaka | #else
|
885 | 33b1db1c | MORITA Kazutaka | |
886 | 33b1db1c | MORITA Kazutaka | static int set_cork(int fd, int v) |
887 | 33b1db1c | MORITA Kazutaka | { |
888 | 33b1db1c | MORITA Kazutaka | return setsockopt(fd, SOL_TCP, TCP_CORK, &v, sizeof(v)); |
889 | 33b1db1c | MORITA Kazutaka | } |
890 | 33b1db1c | MORITA Kazutaka | |
891 | 33b1db1c | MORITA Kazutaka | #endif
|
892 | 33b1db1c | MORITA Kazutaka | |
893 | 33b1db1c | MORITA Kazutaka | static int set_nodelay(int fd) |
894 | 33b1db1c | MORITA Kazutaka | { |
895 | 33b1db1c | MORITA Kazutaka | int ret, opt;
|
896 | 33b1db1c | MORITA Kazutaka | |
897 | 33b1db1c | MORITA Kazutaka | opt = 1;
|
898 | 33b1db1c | MORITA Kazutaka | ret = setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char *)&opt, sizeof(opt)); |
899 | 33b1db1c | MORITA Kazutaka | return ret;
|
900 | 33b1db1c | MORITA Kazutaka | } |
901 | 33b1db1c | MORITA Kazutaka | |
902 | 33b1db1c | MORITA Kazutaka | /*
|
903 | 33b1db1c | MORITA Kazutaka | * Return a socket discriptor to read/write objects.
|
904 | 33b1db1c | MORITA Kazutaka | *
|
905 | 33b1db1c | MORITA Kazutaka | * We cannot use this discriptor for other operations because
|
906 | 33b1db1c | MORITA Kazutaka | * the block driver may be on waiting response from the server.
|
907 | 33b1db1c | MORITA Kazutaka | */
|
908 | 33b1db1c | MORITA Kazutaka | static int get_sheep_fd(BDRVSheepdogState *s) |
909 | 33b1db1c | MORITA Kazutaka | { |
910 | 33b1db1c | MORITA Kazutaka | int ret, fd;
|
911 | 33b1db1c | MORITA Kazutaka | |
912 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
913 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
914 | 6daf194d | Markus Armbruster | error_report("%s", strerror(errno));
|
915 | 33b1db1c | MORITA Kazutaka | return -1; |
916 | 33b1db1c | MORITA Kazutaka | } |
917 | 33b1db1c | MORITA Kazutaka | |
918 | 33b1db1c | MORITA Kazutaka | socket_set_nonblock(fd); |
919 | 33b1db1c | MORITA Kazutaka | |
920 | 33b1db1c | MORITA Kazutaka | ret = set_nodelay(fd); |
921 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
922 | 6daf194d | Markus Armbruster | error_report("%s", strerror(errno));
|
923 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
924 | 33b1db1c | MORITA Kazutaka | return -1; |
925 | 33b1db1c | MORITA Kazutaka | } |
926 | 33b1db1c | MORITA Kazutaka | |
927 | 33b1db1c | MORITA Kazutaka | qemu_aio_set_fd_handler(fd, aio_read_response, NULL, aio_flush_request,
|
928 | 33b1db1c | MORITA Kazutaka | NULL, s);
|
929 | 33b1db1c | MORITA Kazutaka | return fd;
|
930 | 33b1db1c | MORITA Kazutaka | } |
931 | 33b1db1c | MORITA Kazutaka | |
932 | 33b1db1c | MORITA Kazutaka | /*
|
933 | 33b1db1c | MORITA Kazutaka | * Parse a filename
|
934 | 33b1db1c | MORITA Kazutaka | *
|
935 | 33b1db1c | MORITA Kazutaka | * filename must be one of the following formats:
|
936 | 33b1db1c | MORITA Kazutaka | * 1. [vdiname]
|
937 | 33b1db1c | MORITA Kazutaka | * 2. [vdiname]:[snapid]
|
938 | 33b1db1c | MORITA Kazutaka | * 3. [vdiname]:[tag]
|
939 | 33b1db1c | MORITA Kazutaka | * 4. [hostname]:[port]:[vdiname]
|
940 | 33b1db1c | MORITA Kazutaka | * 5. [hostname]:[port]:[vdiname]:[snapid]
|
941 | 33b1db1c | MORITA Kazutaka | * 6. [hostname]:[port]:[vdiname]:[tag]
|
942 | 33b1db1c | MORITA Kazutaka | *
|
943 | 33b1db1c | MORITA Kazutaka | * You can boot from the snapshot images by specifying `snapid` or
|
944 | 33b1db1c | MORITA Kazutaka | * `tag'.
|
945 | 33b1db1c | MORITA Kazutaka | *
|
946 | 33b1db1c | MORITA Kazutaka | * You can run VMs outside the Sheepdog cluster by specifying
|
947 | 33b1db1c | MORITA Kazutaka | * `hostname' and `port' (experimental).
|
948 | 33b1db1c | MORITA Kazutaka | */
|
949 | 33b1db1c | MORITA Kazutaka | static int parse_vdiname(BDRVSheepdogState *s, const char *filename, |
950 | 33b1db1c | MORITA Kazutaka | char *vdi, uint32_t *snapid, char *tag) |
951 | 33b1db1c | MORITA Kazutaka | { |
952 | 33b1db1c | MORITA Kazutaka | char *p, *q;
|
953 | 33b1db1c | MORITA Kazutaka | int nr_sep;
|
954 | 33b1db1c | MORITA Kazutaka | |
955 | 33b1db1c | MORITA Kazutaka | p = q = qemu_strdup(filename); |
956 | 33b1db1c | MORITA Kazutaka | |
957 | 33b1db1c | MORITA Kazutaka | /* count the number of separators */
|
958 | 33b1db1c | MORITA Kazutaka | nr_sep = 0;
|
959 | 33b1db1c | MORITA Kazutaka | while (*p) {
|
960 | 33b1db1c | MORITA Kazutaka | if (*p == ':') { |
961 | 33b1db1c | MORITA Kazutaka | nr_sep++; |
962 | 33b1db1c | MORITA Kazutaka | } |
963 | 33b1db1c | MORITA Kazutaka | p++; |
964 | 33b1db1c | MORITA Kazutaka | } |
965 | 33b1db1c | MORITA Kazutaka | p = q; |
966 | 33b1db1c | MORITA Kazutaka | |
967 | 33b1db1c | MORITA Kazutaka | /* use the first two tokens as hostname and port number. */
|
968 | 33b1db1c | MORITA Kazutaka | if (nr_sep >= 2) { |
969 | 33b1db1c | MORITA Kazutaka | s->addr = p; |
970 | 33b1db1c | MORITA Kazutaka | p = strchr(p, ':');
|
971 | 33b1db1c | MORITA Kazutaka | *p++ = '\0';
|
972 | 33b1db1c | MORITA Kazutaka | |
973 | 33b1db1c | MORITA Kazutaka | s->port = p; |
974 | 33b1db1c | MORITA Kazutaka | p = strchr(p, ':');
|
975 | 33b1db1c | MORITA Kazutaka | *p++ = '\0';
|
976 | 33b1db1c | MORITA Kazutaka | } else {
|
977 | 33b1db1c | MORITA Kazutaka | s->addr = NULL;
|
978 | 33b1db1c | MORITA Kazutaka | s->port = 0;
|
979 | 33b1db1c | MORITA Kazutaka | } |
980 | 33b1db1c | MORITA Kazutaka | |
981 | 33b1db1c | MORITA Kazutaka | strncpy(vdi, p, SD_MAX_VDI_LEN); |
982 | 33b1db1c | MORITA Kazutaka | |
983 | 33b1db1c | MORITA Kazutaka | p = strchr(vdi, ':');
|
984 | 33b1db1c | MORITA Kazutaka | if (p) {
|
985 | 33b1db1c | MORITA Kazutaka | *p++ = '\0';
|
986 | 33b1db1c | MORITA Kazutaka | *snapid = strtoul(p, NULL, 10); |
987 | 33b1db1c | MORITA Kazutaka | if (*snapid == 0) { |
988 | 33b1db1c | MORITA Kazutaka | strncpy(tag, p, SD_MAX_VDI_TAG_LEN); |
989 | 33b1db1c | MORITA Kazutaka | } |
990 | 33b1db1c | MORITA Kazutaka | } else {
|
991 | 33b1db1c | MORITA Kazutaka | *snapid = CURRENT_VDI_ID; /* search current vdi */
|
992 | 33b1db1c | MORITA Kazutaka | } |
993 | 33b1db1c | MORITA Kazutaka | |
994 | 33b1db1c | MORITA Kazutaka | if (s->addr == NULL) { |
995 | 33b1db1c | MORITA Kazutaka | qemu_free(q); |
996 | 33b1db1c | MORITA Kazutaka | } |
997 | 33b1db1c | MORITA Kazutaka | |
998 | 33b1db1c | MORITA Kazutaka | return 0; |
999 | 33b1db1c | MORITA Kazutaka | } |
1000 | 33b1db1c | MORITA Kazutaka | |
1001 | 33b1db1c | MORITA Kazutaka | static int find_vdi_name(BDRVSheepdogState *s, char *filename, uint32_t snapid, |
1002 | 33b1db1c | MORITA Kazutaka | char *tag, uint32_t *vid, int for_snapshot) |
1003 | 33b1db1c | MORITA Kazutaka | { |
1004 | 33b1db1c | MORITA Kazutaka | int ret, fd;
|
1005 | 33b1db1c | MORITA Kazutaka | SheepdogVdiReq hdr; |
1006 | 33b1db1c | MORITA Kazutaka | SheepdogVdiRsp *rsp = (SheepdogVdiRsp *)&hdr; |
1007 | 33b1db1c | MORITA Kazutaka | unsigned int wlen, rlen = 0; |
1008 | 33b1db1c | MORITA Kazutaka | char buf[SD_MAX_VDI_LEN + SD_MAX_VDI_TAG_LEN];
|
1009 | 33b1db1c | MORITA Kazutaka | |
1010 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1011 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1012 | 33b1db1c | MORITA Kazutaka | return -1; |
1013 | 33b1db1c | MORITA Kazutaka | } |
1014 | 33b1db1c | MORITA Kazutaka | |
1015 | 33b1db1c | MORITA Kazutaka | memset(buf, 0, sizeof(buf)); |
1016 | 33b1db1c | MORITA Kazutaka | strncpy(buf, filename, SD_MAX_VDI_LEN); |
1017 | 33b1db1c | MORITA Kazutaka | strncpy(buf + SD_MAX_VDI_LEN, tag, SD_MAX_VDI_TAG_LEN); |
1018 | 33b1db1c | MORITA Kazutaka | |
1019 | 33b1db1c | MORITA Kazutaka | memset(&hdr, 0, sizeof(hdr)); |
1020 | 33b1db1c | MORITA Kazutaka | if (for_snapshot) {
|
1021 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_GET_VDI_INFO; |
1022 | 33b1db1c | MORITA Kazutaka | } else {
|
1023 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_LOCK_VDI; |
1024 | 33b1db1c | MORITA Kazutaka | } |
1025 | 33b1db1c | MORITA Kazutaka | wlen = SD_MAX_VDI_LEN + SD_MAX_VDI_TAG_LEN; |
1026 | 33b1db1c | MORITA Kazutaka | hdr.proto_ver = SD_PROTO_VER; |
1027 | 33b1db1c | MORITA Kazutaka | hdr.data_length = wlen; |
1028 | 33b1db1c | MORITA Kazutaka | hdr.snapid = snapid; |
1029 | 33b1db1c | MORITA Kazutaka | hdr.flags = SD_FLAG_CMD_WRITE; |
1030 | 33b1db1c | MORITA Kazutaka | |
1031 | 33b1db1c | MORITA Kazutaka | ret = do_req(fd, (SheepdogReq *)&hdr, buf, &wlen, &rlen); |
1032 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1033 | 33b1db1c | MORITA Kazutaka | ret = -1;
|
1034 | 33b1db1c | MORITA Kazutaka | goto out;
|
1035 | 33b1db1c | MORITA Kazutaka | } |
1036 | 33b1db1c | MORITA Kazutaka | |
1037 | 33b1db1c | MORITA Kazutaka | if (rsp->result != SD_RES_SUCCESS) {
|
1038 | 6daf194d | Markus Armbruster | error_report("cannot get vdi info, %s, %s %d %s",
|
1039 | 33b1db1c | MORITA Kazutaka | sd_strerror(rsp->result), filename, snapid, tag); |
1040 | 33b1db1c | MORITA Kazutaka | ret = -1;
|
1041 | 33b1db1c | MORITA Kazutaka | goto out;
|
1042 | 33b1db1c | MORITA Kazutaka | } |
1043 | 33b1db1c | MORITA Kazutaka | *vid = rsp->vdi_id; |
1044 | 33b1db1c | MORITA Kazutaka | |
1045 | 33b1db1c | MORITA Kazutaka | ret = 0;
|
1046 | 33b1db1c | MORITA Kazutaka | out:
|
1047 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1048 | 33b1db1c | MORITA Kazutaka | return ret;
|
1049 | 33b1db1c | MORITA Kazutaka | } |
1050 | 33b1db1c | MORITA Kazutaka | |
1051 | 33b1db1c | MORITA Kazutaka | static int add_aio_request(BDRVSheepdogState *s, AIOReq *aio_req, |
1052 | 33b1db1c | MORITA Kazutaka | struct iovec *iov, int niov, int create, |
1053 | 33b1db1c | MORITA Kazutaka | enum AIOCBState aiocb_type)
|
1054 | 33b1db1c | MORITA Kazutaka | { |
1055 | 33b1db1c | MORITA Kazutaka | int nr_copies = s->inode.nr_copies;
|
1056 | 33b1db1c | MORITA Kazutaka | SheepdogObjReq hdr; |
1057 | 33b1db1c | MORITA Kazutaka | unsigned int wlen; |
1058 | 33b1db1c | MORITA Kazutaka | int ret;
|
1059 | 33b1db1c | MORITA Kazutaka | uint64_t oid = aio_req->oid; |
1060 | 33b1db1c | MORITA Kazutaka | unsigned int datalen = aio_req->data_len; |
1061 | 33b1db1c | MORITA Kazutaka | uint64_t offset = aio_req->offset; |
1062 | 33b1db1c | MORITA Kazutaka | uint8_t flags = aio_req->flags; |
1063 | 33b1db1c | MORITA Kazutaka | uint64_t old_oid = aio_req->base_oid; |
1064 | 33b1db1c | MORITA Kazutaka | |
1065 | 33b1db1c | MORITA Kazutaka | if (!nr_copies) {
|
1066 | 6daf194d | Markus Armbruster | error_report("bug");
|
1067 | 33b1db1c | MORITA Kazutaka | } |
1068 | 33b1db1c | MORITA Kazutaka | |
1069 | 33b1db1c | MORITA Kazutaka | memset(&hdr, 0, sizeof(hdr)); |
1070 | 33b1db1c | MORITA Kazutaka | |
1071 | 33b1db1c | MORITA Kazutaka | if (aiocb_type == AIOCB_READ_UDATA) {
|
1072 | 33b1db1c | MORITA Kazutaka | wlen = 0;
|
1073 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_READ_OBJ; |
1074 | 33b1db1c | MORITA Kazutaka | hdr.flags = flags; |
1075 | 33b1db1c | MORITA Kazutaka | } else if (create) { |
1076 | 33b1db1c | MORITA Kazutaka | wlen = datalen; |
1077 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_CREATE_AND_WRITE_OBJ; |
1078 | 33b1db1c | MORITA Kazutaka | hdr.flags = SD_FLAG_CMD_WRITE | flags; |
1079 | 33b1db1c | MORITA Kazutaka | } else {
|
1080 | 33b1db1c | MORITA Kazutaka | wlen = datalen; |
1081 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_WRITE_OBJ; |
1082 | 33b1db1c | MORITA Kazutaka | hdr.flags = SD_FLAG_CMD_WRITE | flags; |
1083 | 33b1db1c | MORITA Kazutaka | } |
1084 | 33b1db1c | MORITA Kazutaka | |
1085 | 33b1db1c | MORITA Kazutaka | hdr.oid = oid; |
1086 | 33b1db1c | MORITA Kazutaka | hdr.cow_oid = old_oid; |
1087 | 33b1db1c | MORITA Kazutaka | hdr.copies = s->inode.nr_copies; |
1088 | 33b1db1c | MORITA Kazutaka | |
1089 | 33b1db1c | MORITA Kazutaka | hdr.data_length = datalen; |
1090 | 33b1db1c | MORITA Kazutaka | hdr.offset = offset; |
1091 | 33b1db1c | MORITA Kazutaka | |
1092 | 33b1db1c | MORITA Kazutaka | hdr.id = aio_req->id; |
1093 | 33b1db1c | MORITA Kazutaka | |
1094 | 33b1db1c | MORITA Kazutaka | set_cork(s->fd, 1);
|
1095 | 33b1db1c | MORITA Kazutaka | |
1096 | 33b1db1c | MORITA Kazutaka | /* send a header */
|
1097 | 33b1db1c | MORITA Kazutaka | ret = do_write(s->fd, &hdr, sizeof(hdr));
|
1098 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1099 | 6daf194d | Markus Armbruster | error_report("failed to send a req, %s", strerror(errno));
|
1100 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
1101 | 33b1db1c | MORITA Kazutaka | } |
1102 | 33b1db1c | MORITA Kazutaka | |
1103 | 33b1db1c | MORITA Kazutaka | if (wlen) {
|
1104 | 33b1db1c | MORITA Kazutaka | ret = do_writev(s->fd, iov, wlen, aio_req->iov_offset); |
1105 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1106 | 6daf194d | Markus Armbruster | error_report("failed to send a data, %s", strerror(errno));
|
1107 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
1108 | 33b1db1c | MORITA Kazutaka | } |
1109 | 33b1db1c | MORITA Kazutaka | } |
1110 | 33b1db1c | MORITA Kazutaka | |
1111 | 33b1db1c | MORITA Kazutaka | set_cork(s->fd, 0);
|
1112 | 33b1db1c | MORITA Kazutaka | |
1113 | 33b1db1c | MORITA Kazutaka | return 0; |
1114 | 33b1db1c | MORITA Kazutaka | } |
1115 | 33b1db1c | MORITA Kazutaka | |
1116 | 33b1db1c | MORITA Kazutaka | static int read_write_object(int fd, char *buf, uint64_t oid, int copies, |
1117 | 33b1db1c | MORITA Kazutaka | unsigned int datalen, uint64_t offset, |
1118 | 33b1db1c | MORITA Kazutaka | int write, int create) |
1119 | 33b1db1c | MORITA Kazutaka | { |
1120 | 33b1db1c | MORITA Kazutaka | SheepdogObjReq hdr; |
1121 | 33b1db1c | MORITA Kazutaka | SheepdogObjRsp *rsp = (SheepdogObjRsp *)&hdr; |
1122 | 33b1db1c | MORITA Kazutaka | unsigned int wlen, rlen; |
1123 | 33b1db1c | MORITA Kazutaka | int ret;
|
1124 | 33b1db1c | MORITA Kazutaka | |
1125 | 33b1db1c | MORITA Kazutaka | memset(&hdr, 0, sizeof(hdr)); |
1126 | 33b1db1c | MORITA Kazutaka | |
1127 | 33b1db1c | MORITA Kazutaka | if (write) {
|
1128 | 33b1db1c | MORITA Kazutaka | wlen = datalen; |
1129 | 33b1db1c | MORITA Kazutaka | rlen = 0;
|
1130 | 33b1db1c | MORITA Kazutaka | hdr.flags = SD_FLAG_CMD_WRITE; |
1131 | 33b1db1c | MORITA Kazutaka | if (create) {
|
1132 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_CREATE_AND_WRITE_OBJ; |
1133 | 33b1db1c | MORITA Kazutaka | } else {
|
1134 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_WRITE_OBJ; |
1135 | 33b1db1c | MORITA Kazutaka | } |
1136 | 33b1db1c | MORITA Kazutaka | } else {
|
1137 | 33b1db1c | MORITA Kazutaka | wlen = 0;
|
1138 | 33b1db1c | MORITA Kazutaka | rlen = datalen; |
1139 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_READ_OBJ; |
1140 | 33b1db1c | MORITA Kazutaka | } |
1141 | 33b1db1c | MORITA Kazutaka | hdr.oid = oid; |
1142 | 33b1db1c | MORITA Kazutaka | hdr.data_length = datalen; |
1143 | 33b1db1c | MORITA Kazutaka | hdr.offset = offset; |
1144 | 33b1db1c | MORITA Kazutaka | hdr.copies = copies; |
1145 | 33b1db1c | MORITA Kazutaka | |
1146 | 33b1db1c | MORITA Kazutaka | ret = do_req(fd, (SheepdogReq *)&hdr, buf, &wlen, &rlen); |
1147 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1148 | 6daf194d | Markus Armbruster | error_report("failed to send a request to the sheep");
|
1149 | 33b1db1c | MORITA Kazutaka | return -1; |
1150 | 33b1db1c | MORITA Kazutaka | } |
1151 | 33b1db1c | MORITA Kazutaka | |
1152 | 33b1db1c | MORITA Kazutaka | switch (rsp->result) {
|
1153 | 33b1db1c | MORITA Kazutaka | case SD_RES_SUCCESS:
|
1154 | 33b1db1c | MORITA Kazutaka | return 0; |
1155 | 33b1db1c | MORITA Kazutaka | default:
|
1156 | 6daf194d | Markus Armbruster | error_report("%s", sd_strerror(rsp->result));
|
1157 | 33b1db1c | MORITA Kazutaka | return -1; |
1158 | 33b1db1c | MORITA Kazutaka | } |
1159 | 33b1db1c | MORITA Kazutaka | } |
1160 | 33b1db1c | MORITA Kazutaka | |
1161 | 33b1db1c | MORITA Kazutaka | static int read_object(int fd, char *buf, uint64_t oid, int copies, |
1162 | 33b1db1c | MORITA Kazutaka | unsigned int datalen, uint64_t offset) |
1163 | 33b1db1c | MORITA Kazutaka | { |
1164 | 33b1db1c | MORITA Kazutaka | return read_write_object(fd, buf, oid, copies, datalen, offset, 0, 0); |
1165 | 33b1db1c | MORITA Kazutaka | } |
1166 | 33b1db1c | MORITA Kazutaka | |
1167 | 33b1db1c | MORITA Kazutaka | static int write_object(int fd, char *buf, uint64_t oid, int copies, |
1168 | 33b1db1c | MORITA Kazutaka | unsigned int datalen, uint64_t offset, int create) |
1169 | 33b1db1c | MORITA Kazutaka | { |
1170 | 33b1db1c | MORITA Kazutaka | return read_write_object(fd, buf, oid, copies, datalen, offset, 1, create); |
1171 | 33b1db1c | MORITA Kazutaka | } |
1172 | 33b1db1c | MORITA Kazutaka | |
1173 | 33b1db1c | MORITA Kazutaka | static int sd_open(BlockDriverState *bs, const char *filename, int flags) |
1174 | 33b1db1c | MORITA Kazutaka | { |
1175 | 33b1db1c | MORITA Kazutaka | int ret, fd;
|
1176 | 33b1db1c | MORITA Kazutaka | uint32_t vid = 0;
|
1177 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
1178 | 33b1db1c | MORITA Kazutaka | char vdi[SD_MAX_VDI_LEN], tag[SD_MAX_VDI_TAG_LEN];
|
1179 | 33b1db1c | MORITA Kazutaka | uint32_t snapid; |
1180 | 33b1db1c | MORITA Kazutaka | char *buf = NULL; |
1181 | 33b1db1c | MORITA Kazutaka | |
1182 | 33b1db1c | MORITA Kazutaka | strstart(filename, "sheepdog:", (const char **)&filename); |
1183 | 33b1db1c | MORITA Kazutaka | |
1184 | 33b1db1c | MORITA Kazutaka | QLIST_INIT(&s->outstanding_aio_head); |
1185 | 33b1db1c | MORITA Kazutaka | s->fd = -1;
|
1186 | 33b1db1c | MORITA Kazutaka | |
1187 | 33b1db1c | MORITA Kazutaka | memset(vdi, 0, sizeof(vdi)); |
1188 | 33b1db1c | MORITA Kazutaka | memset(tag, 0, sizeof(tag)); |
1189 | 33b1db1c | MORITA Kazutaka | if (parse_vdiname(s, filename, vdi, &snapid, tag) < 0) { |
1190 | 33b1db1c | MORITA Kazutaka | goto out;
|
1191 | 33b1db1c | MORITA Kazutaka | } |
1192 | 33b1db1c | MORITA Kazutaka | s->fd = get_sheep_fd(s); |
1193 | 33b1db1c | MORITA Kazutaka | if (s->fd < 0) { |
1194 | 33b1db1c | MORITA Kazutaka | goto out;
|
1195 | 33b1db1c | MORITA Kazutaka | } |
1196 | 33b1db1c | MORITA Kazutaka | |
1197 | 33b1db1c | MORITA Kazutaka | ret = find_vdi_name(s, vdi, snapid, tag, &vid, 0);
|
1198 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1199 | 33b1db1c | MORITA Kazutaka | goto out;
|
1200 | 33b1db1c | MORITA Kazutaka | } |
1201 | 33b1db1c | MORITA Kazutaka | |
1202 | 33b1db1c | MORITA Kazutaka | if (snapid) {
|
1203 | 33b1db1c | MORITA Kazutaka | dprintf("%" PRIx32 " snapshot inode was open.\n", vid); |
1204 | 33b1db1c | MORITA Kazutaka | s->is_snapshot = 1;
|
1205 | 33b1db1c | MORITA Kazutaka | } |
1206 | 33b1db1c | MORITA Kazutaka | |
1207 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1208 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1209 | 6daf194d | Markus Armbruster | error_report("failed to connect");
|
1210 | 33b1db1c | MORITA Kazutaka | goto out;
|
1211 | 33b1db1c | MORITA Kazutaka | } |
1212 | 33b1db1c | MORITA Kazutaka | |
1213 | 33b1db1c | MORITA Kazutaka | buf = qemu_malloc(SD_INODE_SIZE); |
1214 | 33b1db1c | MORITA Kazutaka | ret = read_object(fd, buf, vid_to_vdi_oid(vid), 0, SD_INODE_SIZE, 0); |
1215 | 33b1db1c | MORITA Kazutaka | |
1216 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1217 | 33b1db1c | MORITA Kazutaka | |
1218 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1219 | 33b1db1c | MORITA Kazutaka | goto out;
|
1220 | 33b1db1c | MORITA Kazutaka | } |
1221 | 33b1db1c | MORITA Kazutaka | |
1222 | 33b1db1c | MORITA Kazutaka | memcpy(&s->inode, buf, sizeof(s->inode));
|
1223 | 33b1db1c | MORITA Kazutaka | s->min_dirty_data_idx = UINT32_MAX; |
1224 | 33b1db1c | MORITA Kazutaka | s->max_dirty_data_idx = 0;
|
1225 | 33b1db1c | MORITA Kazutaka | |
1226 | 33b1db1c | MORITA Kazutaka | bs->total_sectors = s->inode.vdi_size / SECTOR_SIZE; |
1227 | 33b1db1c | MORITA Kazutaka | strncpy(s->name, vdi, sizeof(s->name));
|
1228 | 33b1db1c | MORITA Kazutaka | qemu_free(buf); |
1229 | 33b1db1c | MORITA Kazutaka | return 0; |
1230 | 33b1db1c | MORITA Kazutaka | out:
|
1231 | 33b1db1c | MORITA Kazutaka | qemu_aio_set_fd_handler(s->fd, NULL, NULL, NULL, NULL, NULL); |
1232 | 33b1db1c | MORITA Kazutaka | if (s->fd >= 0) { |
1233 | 33b1db1c | MORITA Kazutaka | closesocket(s->fd); |
1234 | 33b1db1c | MORITA Kazutaka | } |
1235 | 33b1db1c | MORITA Kazutaka | qemu_free(buf); |
1236 | 33b1db1c | MORITA Kazutaka | return -1; |
1237 | 33b1db1c | MORITA Kazutaka | } |
1238 | 33b1db1c | MORITA Kazutaka | |
1239 | 33b1db1c | MORITA Kazutaka | static int do_sd_create(char *filename, int64_t vdi_size, |
1240 | 33b1db1c | MORITA Kazutaka | uint32_t base_vid, uint32_t *vdi_id, int snapshot,
|
1241 | 33b1db1c | MORITA Kazutaka | const char *addr, const char *port) |
1242 | 33b1db1c | MORITA Kazutaka | { |
1243 | 33b1db1c | MORITA Kazutaka | SheepdogVdiReq hdr; |
1244 | 33b1db1c | MORITA Kazutaka | SheepdogVdiRsp *rsp = (SheepdogVdiRsp *)&hdr; |
1245 | 33b1db1c | MORITA Kazutaka | int fd, ret;
|
1246 | 33b1db1c | MORITA Kazutaka | unsigned int wlen, rlen = 0; |
1247 | 33b1db1c | MORITA Kazutaka | char buf[SD_MAX_VDI_LEN];
|
1248 | 33b1db1c | MORITA Kazutaka | |
1249 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(addr, port); |
1250 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1251 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
1252 | 33b1db1c | MORITA Kazutaka | } |
1253 | 33b1db1c | MORITA Kazutaka | |
1254 | 33b1db1c | MORITA Kazutaka | memset(buf, 0, sizeof(buf)); |
1255 | 33b1db1c | MORITA Kazutaka | strncpy(buf, filename, SD_MAX_VDI_LEN); |
1256 | 33b1db1c | MORITA Kazutaka | |
1257 | 33b1db1c | MORITA Kazutaka | memset(&hdr, 0, sizeof(hdr)); |
1258 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_NEW_VDI; |
1259 | 33b1db1c | MORITA Kazutaka | hdr.base_vdi_id = base_vid; |
1260 | 33b1db1c | MORITA Kazutaka | |
1261 | 33b1db1c | MORITA Kazutaka | wlen = SD_MAX_VDI_LEN; |
1262 | 33b1db1c | MORITA Kazutaka | |
1263 | 33b1db1c | MORITA Kazutaka | hdr.flags = SD_FLAG_CMD_WRITE; |
1264 | 33b1db1c | MORITA Kazutaka | hdr.snapid = snapshot; |
1265 | 33b1db1c | MORITA Kazutaka | |
1266 | 33b1db1c | MORITA Kazutaka | hdr.data_length = wlen; |
1267 | 33b1db1c | MORITA Kazutaka | hdr.vdi_size = vdi_size; |
1268 | 33b1db1c | MORITA Kazutaka | |
1269 | 33b1db1c | MORITA Kazutaka | ret = do_req(fd, (SheepdogReq *)&hdr, buf, &wlen, &rlen); |
1270 | 33b1db1c | MORITA Kazutaka | |
1271 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1272 | 33b1db1c | MORITA Kazutaka | |
1273 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1274 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
1275 | 33b1db1c | MORITA Kazutaka | } |
1276 | 33b1db1c | MORITA Kazutaka | |
1277 | 33b1db1c | MORITA Kazutaka | if (rsp->result != SD_RES_SUCCESS) {
|
1278 | 6daf194d | Markus Armbruster | error_report("%s, %s", sd_strerror(rsp->result), filename);
|
1279 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
1280 | 33b1db1c | MORITA Kazutaka | } |
1281 | 33b1db1c | MORITA Kazutaka | |
1282 | 33b1db1c | MORITA Kazutaka | if (vdi_id) {
|
1283 | 33b1db1c | MORITA Kazutaka | *vdi_id = rsp->vdi_id; |
1284 | 33b1db1c | MORITA Kazutaka | } |
1285 | 33b1db1c | MORITA Kazutaka | |
1286 | 33b1db1c | MORITA Kazutaka | return 0; |
1287 | 33b1db1c | MORITA Kazutaka | } |
1288 | 33b1db1c | MORITA Kazutaka | |
1289 | a8e0fdd7 | MORITA Kazutaka | static int sd_prealloc(const char *filename) |
1290 | a8e0fdd7 | MORITA Kazutaka | { |
1291 | a8e0fdd7 | MORITA Kazutaka | BlockDriverState *bs = NULL;
|
1292 | a8e0fdd7 | MORITA Kazutaka | uint32_t idx, max_idx; |
1293 | a8e0fdd7 | MORITA Kazutaka | int64_t vdi_size; |
1294 | a8e0fdd7 | MORITA Kazutaka | void *buf = qemu_mallocz(SD_DATA_OBJ_SIZE);
|
1295 | a8e0fdd7 | MORITA Kazutaka | int ret;
|
1296 | a8e0fdd7 | MORITA Kazutaka | |
1297 | a8e0fdd7 | MORITA Kazutaka | ret = bdrv_file_open(&bs, filename, BDRV_O_RDWR); |
1298 | a8e0fdd7 | MORITA Kazutaka | if (ret < 0) { |
1299 | a8e0fdd7 | MORITA Kazutaka | goto out;
|
1300 | a8e0fdd7 | MORITA Kazutaka | } |
1301 | a8e0fdd7 | MORITA Kazutaka | |
1302 | a8e0fdd7 | MORITA Kazutaka | vdi_size = bdrv_getlength(bs); |
1303 | a8e0fdd7 | MORITA Kazutaka | if (vdi_size < 0) { |
1304 | a8e0fdd7 | MORITA Kazutaka | ret = vdi_size; |
1305 | a8e0fdd7 | MORITA Kazutaka | goto out;
|
1306 | a8e0fdd7 | MORITA Kazutaka | } |
1307 | a8e0fdd7 | MORITA Kazutaka | max_idx = DIV_ROUND_UP(vdi_size, SD_DATA_OBJ_SIZE); |
1308 | a8e0fdd7 | MORITA Kazutaka | |
1309 | a8e0fdd7 | MORITA Kazutaka | for (idx = 0; idx < max_idx; idx++) { |
1310 | a8e0fdd7 | MORITA Kazutaka | /*
|
1311 | a8e0fdd7 | MORITA Kazutaka | * The created image can be a cloned image, so we need to read
|
1312 | a8e0fdd7 | MORITA Kazutaka | * a data from the source image.
|
1313 | a8e0fdd7 | MORITA Kazutaka | */
|
1314 | a8e0fdd7 | MORITA Kazutaka | ret = bdrv_pread(bs, idx * SD_DATA_OBJ_SIZE, buf, SD_DATA_OBJ_SIZE); |
1315 | a8e0fdd7 | MORITA Kazutaka | if (ret < 0) { |
1316 | a8e0fdd7 | MORITA Kazutaka | goto out;
|
1317 | a8e0fdd7 | MORITA Kazutaka | } |
1318 | a8e0fdd7 | MORITA Kazutaka | ret = bdrv_pwrite(bs, idx * SD_DATA_OBJ_SIZE, buf, SD_DATA_OBJ_SIZE); |
1319 | a8e0fdd7 | MORITA Kazutaka | if (ret < 0) { |
1320 | a8e0fdd7 | MORITA Kazutaka | goto out;
|
1321 | a8e0fdd7 | MORITA Kazutaka | } |
1322 | a8e0fdd7 | MORITA Kazutaka | } |
1323 | a8e0fdd7 | MORITA Kazutaka | out:
|
1324 | a8e0fdd7 | MORITA Kazutaka | if (bs) {
|
1325 | a8e0fdd7 | MORITA Kazutaka | bdrv_delete(bs); |
1326 | a8e0fdd7 | MORITA Kazutaka | } |
1327 | a8e0fdd7 | MORITA Kazutaka | qemu_free(buf); |
1328 | a8e0fdd7 | MORITA Kazutaka | |
1329 | a8e0fdd7 | MORITA Kazutaka | return ret;
|
1330 | a8e0fdd7 | MORITA Kazutaka | } |
1331 | a8e0fdd7 | MORITA Kazutaka | |
1332 | 33b1db1c | MORITA Kazutaka | static int sd_create(const char *filename, QEMUOptionParameter *options) |
1333 | 33b1db1c | MORITA Kazutaka | { |
1334 | 33b1db1c | MORITA Kazutaka | int ret;
|
1335 | b4447363 | MORITA Kazutaka | uint32_t vid = 0, base_vid = 0; |
1336 | 33b1db1c | MORITA Kazutaka | int64_t vdi_size = 0;
|
1337 | 33b1db1c | MORITA Kazutaka | char *backing_file = NULL; |
1338 | b4447363 | MORITA Kazutaka | BDRVSheepdogState s; |
1339 | b4447363 | MORITA Kazutaka | char vdi[SD_MAX_VDI_LEN], tag[SD_MAX_VDI_TAG_LEN];
|
1340 | b4447363 | MORITA Kazutaka | uint32_t snapid; |
1341 | a8e0fdd7 | MORITA Kazutaka | int prealloc = 0; |
1342 | a8e0fdd7 | MORITA Kazutaka | const char *vdiname; |
1343 | 33b1db1c | MORITA Kazutaka | |
1344 | a8e0fdd7 | MORITA Kazutaka | strstart(filename, "sheepdog:", &vdiname);
|
1345 | 33b1db1c | MORITA Kazutaka | |
1346 | b4447363 | MORITA Kazutaka | memset(&s, 0, sizeof(s)); |
1347 | b4447363 | MORITA Kazutaka | memset(vdi, 0, sizeof(vdi)); |
1348 | b4447363 | MORITA Kazutaka | memset(tag, 0, sizeof(tag)); |
1349 | a8e0fdd7 | MORITA Kazutaka | if (parse_vdiname(&s, vdiname, vdi, &snapid, tag) < 0) { |
1350 | 6daf194d | Markus Armbruster | error_report("invalid filename");
|
1351 | b4447363 | MORITA Kazutaka | return -EINVAL;
|
1352 | b4447363 | MORITA Kazutaka | } |
1353 | b4447363 | MORITA Kazutaka | |
1354 | 33b1db1c | MORITA Kazutaka | while (options && options->name) {
|
1355 | 33b1db1c | MORITA Kazutaka | if (!strcmp(options->name, BLOCK_OPT_SIZE)) {
|
1356 | 33b1db1c | MORITA Kazutaka | vdi_size = options->value.n; |
1357 | 33b1db1c | MORITA Kazutaka | } else if (!strcmp(options->name, BLOCK_OPT_BACKING_FILE)) { |
1358 | 33b1db1c | MORITA Kazutaka | backing_file = options->value.s; |
1359 | a8e0fdd7 | MORITA Kazutaka | } else if (!strcmp(options->name, BLOCK_OPT_PREALLOC)) { |
1360 | a8e0fdd7 | MORITA Kazutaka | if (!options->value.s || !strcmp(options->value.s, "off")) { |
1361 | a8e0fdd7 | MORITA Kazutaka | prealloc = 0;
|
1362 | a8e0fdd7 | MORITA Kazutaka | } else if (!strcmp(options->value.s, "full")) { |
1363 | a8e0fdd7 | MORITA Kazutaka | prealloc = 1;
|
1364 | a8e0fdd7 | MORITA Kazutaka | } else {
|
1365 | a8e0fdd7 | MORITA Kazutaka | error_report("Invalid preallocation mode: '%s'",
|
1366 | a8e0fdd7 | MORITA Kazutaka | options->value.s); |
1367 | a8e0fdd7 | MORITA Kazutaka | return -EINVAL;
|
1368 | a8e0fdd7 | MORITA Kazutaka | } |
1369 | 33b1db1c | MORITA Kazutaka | } |
1370 | 33b1db1c | MORITA Kazutaka | options++; |
1371 | 33b1db1c | MORITA Kazutaka | } |
1372 | 33b1db1c | MORITA Kazutaka | |
1373 | 33b1db1c | MORITA Kazutaka | if (vdi_size > SD_MAX_VDI_SIZE) {
|
1374 | 6daf194d | Markus Armbruster | error_report("too big image size");
|
1375 | 33b1db1c | MORITA Kazutaka | return -EINVAL;
|
1376 | 33b1db1c | MORITA Kazutaka | } |
1377 | 33b1db1c | MORITA Kazutaka | |
1378 | 33b1db1c | MORITA Kazutaka | if (backing_file) {
|
1379 | 33b1db1c | MORITA Kazutaka | BlockDriverState *bs; |
1380 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s; |
1381 | 33b1db1c | MORITA Kazutaka | BlockDriver *drv; |
1382 | 33b1db1c | MORITA Kazutaka | |
1383 | 33b1db1c | MORITA Kazutaka | /* Currently, only Sheepdog backing image is supported. */
|
1384 | 33b1db1c | MORITA Kazutaka | drv = bdrv_find_protocol(backing_file); |
1385 | 33b1db1c | MORITA Kazutaka | if (!drv || strcmp(drv->protocol_name, "sheepdog") != 0) { |
1386 | 6daf194d | Markus Armbruster | error_report("backing_file must be a sheepdog image");
|
1387 | 33b1db1c | MORITA Kazutaka | return -EINVAL;
|
1388 | 33b1db1c | MORITA Kazutaka | } |
1389 | 33b1db1c | MORITA Kazutaka | |
1390 | 33b1db1c | MORITA Kazutaka | ret = bdrv_file_open(&bs, backing_file, 0);
|
1391 | 33b1db1c | MORITA Kazutaka | if (ret < 0) |
1392 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
1393 | 33b1db1c | MORITA Kazutaka | |
1394 | 33b1db1c | MORITA Kazutaka | s = bs->opaque; |
1395 | 33b1db1c | MORITA Kazutaka | |
1396 | 33b1db1c | MORITA Kazutaka | if (!is_snapshot(&s->inode)) {
|
1397 | 6daf194d | Markus Armbruster | error_report("cannot clone from a non snapshot vdi");
|
1398 | 33b1db1c | MORITA Kazutaka | bdrv_delete(bs); |
1399 | 33b1db1c | MORITA Kazutaka | return -EINVAL;
|
1400 | 33b1db1c | MORITA Kazutaka | } |
1401 | 33b1db1c | MORITA Kazutaka | |
1402 | b4447363 | MORITA Kazutaka | base_vid = s->inode.vdi_id; |
1403 | 33b1db1c | MORITA Kazutaka | bdrv_delete(bs); |
1404 | 33b1db1c | MORITA Kazutaka | } |
1405 | 33b1db1c | MORITA Kazutaka | |
1406 | a8e0fdd7 | MORITA Kazutaka | ret = do_sd_create(vdi, vdi_size, base_vid, &vid, 0, s.addr, s.port);
|
1407 | a8e0fdd7 | MORITA Kazutaka | if (!prealloc || ret) {
|
1408 | a8e0fdd7 | MORITA Kazutaka | return ret;
|
1409 | a8e0fdd7 | MORITA Kazutaka | } |
1410 | a8e0fdd7 | MORITA Kazutaka | |
1411 | a8e0fdd7 | MORITA Kazutaka | return sd_prealloc(filename);
|
1412 | 33b1db1c | MORITA Kazutaka | } |
1413 | 33b1db1c | MORITA Kazutaka | |
1414 | 33b1db1c | MORITA Kazutaka | static void sd_close(BlockDriverState *bs) |
1415 | 33b1db1c | MORITA Kazutaka | { |
1416 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
1417 | 33b1db1c | MORITA Kazutaka | SheepdogVdiReq hdr; |
1418 | 33b1db1c | MORITA Kazutaka | SheepdogVdiRsp *rsp = (SheepdogVdiRsp *)&hdr; |
1419 | 33b1db1c | MORITA Kazutaka | unsigned int wlen, rlen = 0; |
1420 | 33b1db1c | MORITA Kazutaka | int fd, ret;
|
1421 | 33b1db1c | MORITA Kazutaka | |
1422 | 33b1db1c | MORITA Kazutaka | dprintf("%s\n", s->name);
|
1423 | 33b1db1c | MORITA Kazutaka | |
1424 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1425 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1426 | 33b1db1c | MORITA Kazutaka | return;
|
1427 | 33b1db1c | MORITA Kazutaka | } |
1428 | 33b1db1c | MORITA Kazutaka | |
1429 | 33b1db1c | MORITA Kazutaka | memset(&hdr, 0, sizeof(hdr)); |
1430 | 33b1db1c | MORITA Kazutaka | |
1431 | 33b1db1c | MORITA Kazutaka | hdr.opcode = SD_OP_RELEASE_VDI; |
1432 | 33b1db1c | MORITA Kazutaka | wlen = strlen(s->name) + 1;
|
1433 | 33b1db1c | MORITA Kazutaka | hdr.data_length = wlen; |
1434 | 33b1db1c | MORITA Kazutaka | hdr.flags = SD_FLAG_CMD_WRITE; |
1435 | 33b1db1c | MORITA Kazutaka | |
1436 | 33b1db1c | MORITA Kazutaka | ret = do_req(fd, (SheepdogReq *)&hdr, s->name, &wlen, &rlen); |
1437 | 33b1db1c | MORITA Kazutaka | |
1438 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1439 | 33b1db1c | MORITA Kazutaka | |
1440 | 33b1db1c | MORITA Kazutaka | if (!ret && rsp->result != SD_RES_SUCCESS &&
|
1441 | 33b1db1c | MORITA Kazutaka | rsp->result != SD_RES_VDI_NOT_LOCKED) { |
1442 | 6daf194d | Markus Armbruster | error_report("%s, %s", sd_strerror(rsp->result), s->name);
|
1443 | 33b1db1c | MORITA Kazutaka | } |
1444 | 33b1db1c | MORITA Kazutaka | |
1445 | 33b1db1c | MORITA Kazutaka | qemu_aio_set_fd_handler(s->fd, NULL, NULL, NULL, NULL, NULL); |
1446 | 33b1db1c | MORITA Kazutaka | closesocket(s->fd); |
1447 | 33b1db1c | MORITA Kazutaka | qemu_free(s->addr); |
1448 | 33b1db1c | MORITA Kazutaka | } |
1449 | 33b1db1c | MORITA Kazutaka | |
1450 | 33b1db1c | MORITA Kazutaka | static int64_t sd_getlength(BlockDriverState *bs)
|
1451 | 33b1db1c | MORITA Kazutaka | { |
1452 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
1453 | 33b1db1c | MORITA Kazutaka | |
1454 | 33b1db1c | MORITA Kazutaka | return s->inode.vdi_size;
|
1455 | 33b1db1c | MORITA Kazutaka | } |
1456 | 33b1db1c | MORITA Kazutaka | |
1457 | 33b1db1c | MORITA Kazutaka | static int sd_truncate(BlockDriverState *bs, int64_t offset) |
1458 | 33b1db1c | MORITA Kazutaka | { |
1459 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
1460 | 33b1db1c | MORITA Kazutaka | int ret, fd;
|
1461 | 33b1db1c | MORITA Kazutaka | unsigned int datalen; |
1462 | 33b1db1c | MORITA Kazutaka | |
1463 | 33b1db1c | MORITA Kazutaka | if (offset < s->inode.vdi_size) {
|
1464 | 6daf194d | Markus Armbruster | error_report("shrinking is not supported");
|
1465 | 33b1db1c | MORITA Kazutaka | return -EINVAL;
|
1466 | 33b1db1c | MORITA Kazutaka | } else if (offset > SD_MAX_VDI_SIZE) { |
1467 | 6daf194d | Markus Armbruster | error_report("too big image size");
|
1468 | 33b1db1c | MORITA Kazutaka | return -EINVAL;
|
1469 | 33b1db1c | MORITA Kazutaka | } |
1470 | 33b1db1c | MORITA Kazutaka | |
1471 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1472 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1473 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
1474 | 33b1db1c | MORITA Kazutaka | } |
1475 | 33b1db1c | MORITA Kazutaka | |
1476 | 33b1db1c | MORITA Kazutaka | /* we don't need to update entire object */
|
1477 | 33b1db1c | MORITA Kazutaka | datalen = SD_INODE_SIZE - sizeof(s->inode.data_vdi_id);
|
1478 | 33b1db1c | MORITA Kazutaka | s->inode.vdi_size = offset; |
1479 | 33b1db1c | MORITA Kazutaka | ret = write_object(fd, (char *)&s->inode, vid_to_vdi_oid(s->inode.vdi_id),
|
1480 | 33b1db1c | MORITA Kazutaka | s->inode.nr_copies, datalen, 0, 0); |
1481 | 33b1db1c | MORITA Kazutaka | close(fd); |
1482 | 33b1db1c | MORITA Kazutaka | |
1483 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
1484 | 6daf194d | Markus Armbruster | error_report("failed to update an inode.");
|
1485 | 33b1db1c | MORITA Kazutaka | return -EIO;
|
1486 | 33b1db1c | MORITA Kazutaka | } |
1487 | 33b1db1c | MORITA Kazutaka | |
1488 | 33b1db1c | MORITA Kazutaka | return 0; |
1489 | 33b1db1c | MORITA Kazutaka | } |
1490 | 33b1db1c | MORITA Kazutaka | |
1491 | 33b1db1c | MORITA Kazutaka | /*
|
1492 | 33b1db1c | MORITA Kazutaka | * This function is called after writing data objects. If we need to
|
1493 | 33b1db1c | MORITA Kazutaka | * update metadata, this sends a write request to the vdi object.
|
1494 | 33b1db1c | MORITA Kazutaka | * Otherwise, this calls the AIOCB callback.
|
1495 | 33b1db1c | MORITA Kazutaka | */
|
1496 | 33b1db1c | MORITA Kazutaka | static void sd_write_done(SheepdogAIOCB *acb) |
1497 | 33b1db1c | MORITA Kazutaka | { |
1498 | 33b1db1c | MORITA Kazutaka | int ret;
|
1499 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = acb->common.bs->opaque; |
1500 | 33b1db1c | MORITA Kazutaka | struct iovec iov;
|
1501 | 33b1db1c | MORITA Kazutaka | AIOReq *aio_req; |
1502 | 33b1db1c | MORITA Kazutaka | uint32_t offset, data_len, mn, mx; |
1503 | 33b1db1c | MORITA Kazutaka | |
1504 | 33b1db1c | MORITA Kazutaka | mn = s->min_dirty_data_idx; |
1505 | 33b1db1c | MORITA Kazutaka | mx = s->max_dirty_data_idx; |
1506 | 33b1db1c | MORITA Kazutaka | if (mn <= mx) {
|
1507 | 33b1db1c | MORITA Kazutaka | /* we need to update the vdi object. */
|
1508 | 33b1db1c | MORITA Kazutaka | offset = sizeof(s->inode) - sizeof(s->inode.data_vdi_id) + |
1509 | 33b1db1c | MORITA Kazutaka | mn * sizeof(s->inode.data_vdi_id[0]); |
1510 | 33b1db1c | MORITA Kazutaka | data_len = (mx - mn + 1) * sizeof(s->inode.data_vdi_id[0]); |
1511 | 33b1db1c | MORITA Kazutaka | |
1512 | 33b1db1c | MORITA Kazutaka | s->min_dirty_data_idx = UINT32_MAX; |
1513 | 33b1db1c | MORITA Kazutaka | s->max_dirty_data_idx = 0;
|
1514 | 33b1db1c | MORITA Kazutaka | |
1515 | 33b1db1c | MORITA Kazutaka | iov.iov_base = &s->inode; |
1516 | 33b1db1c | MORITA Kazutaka | iov.iov_len = sizeof(s->inode);
|
1517 | 33b1db1c | MORITA Kazutaka | aio_req = alloc_aio_req(s, acb, vid_to_vdi_oid(s->inode.vdi_id), |
1518 | 33b1db1c | MORITA Kazutaka | data_len, offset, 0, 0, offset); |
1519 | 33b1db1c | MORITA Kazutaka | ret = add_aio_request(s, aio_req, &iov, 1, 0, AIOCB_WRITE_UDATA); |
1520 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1521 | 33b1db1c | MORITA Kazutaka | free_aio_req(s, aio_req); |
1522 | 33b1db1c | MORITA Kazutaka | acb->ret = -EIO; |
1523 | 33b1db1c | MORITA Kazutaka | goto out;
|
1524 | 33b1db1c | MORITA Kazutaka | } |
1525 | 33b1db1c | MORITA Kazutaka | |
1526 | 33b1db1c | MORITA Kazutaka | acb->aio_done_func = sd_finish_aiocb; |
1527 | 33b1db1c | MORITA Kazutaka | acb->aiocb_type = AIOCB_WRITE_UDATA; |
1528 | 33b1db1c | MORITA Kazutaka | return;
|
1529 | 33b1db1c | MORITA Kazutaka | } |
1530 | 33b1db1c | MORITA Kazutaka | out:
|
1531 | 33b1db1c | MORITA Kazutaka | sd_finish_aiocb(acb); |
1532 | 33b1db1c | MORITA Kazutaka | } |
1533 | 33b1db1c | MORITA Kazutaka | |
1534 | 33b1db1c | MORITA Kazutaka | /*
|
1535 | 33b1db1c | MORITA Kazutaka | * Create a writable VDI from a snapshot
|
1536 | 33b1db1c | MORITA Kazutaka | */
|
1537 | 33b1db1c | MORITA Kazutaka | static int sd_create_branch(BDRVSheepdogState *s) |
1538 | 33b1db1c | MORITA Kazutaka | { |
1539 | 33b1db1c | MORITA Kazutaka | int ret, fd;
|
1540 | 33b1db1c | MORITA Kazutaka | uint32_t vid; |
1541 | 33b1db1c | MORITA Kazutaka | char *buf;
|
1542 | 33b1db1c | MORITA Kazutaka | |
1543 | 33b1db1c | MORITA Kazutaka | dprintf("%" PRIx32 " is snapshot.\n", s->inode.vdi_id); |
1544 | 33b1db1c | MORITA Kazutaka | |
1545 | 33b1db1c | MORITA Kazutaka | buf = qemu_malloc(SD_INODE_SIZE); |
1546 | 33b1db1c | MORITA Kazutaka | |
1547 | 33b1db1c | MORITA Kazutaka | ret = do_sd_create(s->name, s->inode.vdi_size, s->inode.vdi_id, &vid, 1,
|
1548 | 33b1db1c | MORITA Kazutaka | s->addr, s->port); |
1549 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1550 | 33b1db1c | MORITA Kazutaka | goto out;
|
1551 | 33b1db1c | MORITA Kazutaka | } |
1552 | 33b1db1c | MORITA Kazutaka | |
1553 | 33b1db1c | MORITA Kazutaka | dprintf("%" PRIx32 " is created.\n", vid); |
1554 | 33b1db1c | MORITA Kazutaka | |
1555 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1556 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1557 | 6daf194d | Markus Armbruster | error_report("failed to connect");
|
1558 | 33b1db1c | MORITA Kazutaka | goto out;
|
1559 | 33b1db1c | MORITA Kazutaka | } |
1560 | 33b1db1c | MORITA Kazutaka | |
1561 | 33b1db1c | MORITA Kazutaka | ret = read_object(fd, buf, vid_to_vdi_oid(vid), s->inode.nr_copies, |
1562 | 33b1db1c | MORITA Kazutaka | SD_INODE_SIZE, 0);
|
1563 | 33b1db1c | MORITA Kazutaka | |
1564 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1565 | 33b1db1c | MORITA Kazutaka | |
1566 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
1567 | 33b1db1c | MORITA Kazutaka | goto out;
|
1568 | 33b1db1c | MORITA Kazutaka | } |
1569 | 33b1db1c | MORITA Kazutaka | |
1570 | 33b1db1c | MORITA Kazutaka | memcpy(&s->inode, buf, sizeof(s->inode));
|
1571 | 33b1db1c | MORITA Kazutaka | |
1572 | 33b1db1c | MORITA Kazutaka | s->is_snapshot = 0;
|
1573 | 33b1db1c | MORITA Kazutaka | ret = 0;
|
1574 | 33b1db1c | MORITA Kazutaka | dprintf("%" PRIx32 " was newly created.\n", s->inode.vdi_id); |
1575 | 33b1db1c | MORITA Kazutaka | |
1576 | 33b1db1c | MORITA Kazutaka | out:
|
1577 | 33b1db1c | MORITA Kazutaka | qemu_free(buf); |
1578 | 33b1db1c | MORITA Kazutaka | |
1579 | 33b1db1c | MORITA Kazutaka | return ret;
|
1580 | 33b1db1c | MORITA Kazutaka | } |
1581 | 33b1db1c | MORITA Kazutaka | |
1582 | 33b1db1c | MORITA Kazutaka | /*
|
1583 | 33b1db1c | MORITA Kazutaka | * Send I/O requests to the server.
|
1584 | 33b1db1c | MORITA Kazutaka | *
|
1585 | 33b1db1c | MORITA Kazutaka | * This function sends requests to the server, links the requests to
|
1586 | 33b1db1c | MORITA Kazutaka | * the outstanding_list in BDRVSheepdogState, and exits without
|
1587 | 33b1db1c | MORITA Kazutaka | * waiting the response. The responses are received in the
|
1588 | 33b1db1c | MORITA Kazutaka | * `aio_read_response' function which is called from the main loop as
|
1589 | 33b1db1c | MORITA Kazutaka | * a fd handler.
|
1590 | 33b1db1c | MORITA Kazutaka | */
|
1591 | 33b1db1c | MORITA Kazutaka | static void sd_readv_writev_bh_cb(void *p) |
1592 | 33b1db1c | MORITA Kazutaka | { |
1593 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *acb = p; |
1594 | 33b1db1c | MORITA Kazutaka | int ret = 0; |
1595 | 33b1db1c | MORITA Kazutaka | unsigned long len, done = 0, total = acb->nb_sectors * SECTOR_SIZE; |
1596 | 33b1db1c | MORITA Kazutaka | unsigned long idx = acb->sector_num * SECTOR_SIZE / SD_DATA_OBJ_SIZE; |
1597 | 33b1db1c | MORITA Kazutaka | uint64_t oid; |
1598 | 33b1db1c | MORITA Kazutaka | uint64_t offset = (acb->sector_num * SECTOR_SIZE) % SD_DATA_OBJ_SIZE; |
1599 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = acb->common.bs->opaque; |
1600 | 33b1db1c | MORITA Kazutaka | SheepdogInode *inode = &s->inode; |
1601 | 33b1db1c | MORITA Kazutaka | AIOReq *aio_req; |
1602 | 33b1db1c | MORITA Kazutaka | |
1603 | 33b1db1c | MORITA Kazutaka | qemu_bh_delete(acb->bh); |
1604 | 33b1db1c | MORITA Kazutaka | acb->bh = NULL;
|
1605 | 33b1db1c | MORITA Kazutaka | |
1606 | 33b1db1c | MORITA Kazutaka | if (acb->aiocb_type == AIOCB_WRITE_UDATA && s->is_snapshot) {
|
1607 | 33b1db1c | MORITA Kazutaka | /*
|
1608 | 33b1db1c | MORITA Kazutaka | * In the case we open the snapshot VDI, Sheepdog creates the
|
1609 | 33b1db1c | MORITA Kazutaka | * writable VDI when we do a write operation first.
|
1610 | 33b1db1c | MORITA Kazutaka | */
|
1611 | 33b1db1c | MORITA Kazutaka | ret = sd_create_branch(s); |
1612 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1613 | 33b1db1c | MORITA Kazutaka | acb->ret = -EIO; |
1614 | 33b1db1c | MORITA Kazutaka | goto out;
|
1615 | 33b1db1c | MORITA Kazutaka | } |
1616 | 33b1db1c | MORITA Kazutaka | } |
1617 | 33b1db1c | MORITA Kazutaka | |
1618 | 33b1db1c | MORITA Kazutaka | while (done != total) {
|
1619 | 33b1db1c | MORITA Kazutaka | uint8_t flags = 0;
|
1620 | 33b1db1c | MORITA Kazutaka | uint64_t old_oid = 0;
|
1621 | 33b1db1c | MORITA Kazutaka | int create = 0; |
1622 | 33b1db1c | MORITA Kazutaka | |
1623 | 33b1db1c | MORITA Kazutaka | oid = vid_to_data_oid(inode->data_vdi_id[idx], idx); |
1624 | 33b1db1c | MORITA Kazutaka | |
1625 | 33b1db1c | MORITA Kazutaka | len = MIN(total - done, SD_DATA_OBJ_SIZE - offset); |
1626 | 33b1db1c | MORITA Kazutaka | |
1627 | 33b1db1c | MORITA Kazutaka | if (!inode->data_vdi_id[idx]) {
|
1628 | 33b1db1c | MORITA Kazutaka | if (acb->aiocb_type == AIOCB_READ_UDATA) {
|
1629 | 33b1db1c | MORITA Kazutaka | goto done;
|
1630 | 33b1db1c | MORITA Kazutaka | } |
1631 | 33b1db1c | MORITA Kazutaka | |
1632 | 33b1db1c | MORITA Kazutaka | create = 1;
|
1633 | 33b1db1c | MORITA Kazutaka | } else if (acb->aiocb_type == AIOCB_WRITE_UDATA |
1634 | ebabb67a | Stefan Weil | && !is_data_obj_writable(inode, idx)) { |
1635 | 33b1db1c | MORITA Kazutaka | /* Copy-On-Write */
|
1636 | 33b1db1c | MORITA Kazutaka | create = 1;
|
1637 | 33b1db1c | MORITA Kazutaka | old_oid = oid; |
1638 | 33b1db1c | MORITA Kazutaka | flags = SD_FLAG_CMD_COW; |
1639 | 33b1db1c | MORITA Kazutaka | } |
1640 | 33b1db1c | MORITA Kazutaka | |
1641 | 33b1db1c | MORITA Kazutaka | if (create) {
|
1642 | 33b1db1c | MORITA Kazutaka | dprintf("update ino (%" PRIu32") %" PRIu64 " %" PRIu64 |
1643 | 33b1db1c | MORITA Kazutaka | " %" PRIu64 "\n", inode->vdi_id, oid, |
1644 | 33b1db1c | MORITA Kazutaka | vid_to_data_oid(inode->data_vdi_id[idx], idx), idx); |
1645 | 33b1db1c | MORITA Kazutaka | oid = vid_to_data_oid(inode->vdi_id, idx); |
1646 | 33b1db1c | MORITA Kazutaka | dprintf("new oid %lx\n", oid);
|
1647 | 33b1db1c | MORITA Kazutaka | } |
1648 | 33b1db1c | MORITA Kazutaka | |
1649 | 33b1db1c | MORITA Kazutaka | aio_req = alloc_aio_req(s, acb, oid, len, offset, flags, old_oid, done); |
1650 | 33b1db1c | MORITA Kazutaka | |
1651 | 33b1db1c | MORITA Kazutaka | if (create) {
|
1652 | 33b1db1c | MORITA Kazutaka | AIOReq *areq; |
1653 | 33b1db1c | MORITA Kazutaka | QLIST_FOREACH(areq, &s->outstanding_aio_head, |
1654 | 33b1db1c | MORITA Kazutaka | outstanding_aio_siblings) { |
1655 | 33b1db1c | MORITA Kazutaka | if (areq == aio_req) {
|
1656 | 33b1db1c | MORITA Kazutaka | continue;
|
1657 | 33b1db1c | MORITA Kazutaka | } |
1658 | 33b1db1c | MORITA Kazutaka | if (areq->oid == oid) {
|
1659 | 33b1db1c | MORITA Kazutaka | /*
|
1660 | 33b1db1c | MORITA Kazutaka | * Sheepdog cannot handle simultaneous create
|
1661 | 33b1db1c | MORITA Kazutaka | * requests to the same object. So we cannot send
|
1662 | 33b1db1c | MORITA Kazutaka | * the request until the previous request
|
1663 | 33b1db1c | MORITA Kazutaka | * finishes.
|
1664 | 33b1db1c | MORITA Kazutaka | */
|
1665 | 33b1db1c | MORITA Kazutaka | aio_req->flags = 0;
|
1666 | 33b1db1c | MORITA Kazutaka | aio_req->base_oid = 0;
|
1667 | 33b1db1c | MORITA Kazutaka | goto done;
|
1668 | 33b1db1c | MORITA Kazutaka | } |
1669 | 33b1db1c | MORITA Kazutaka | } |
1670 | 33b1db1c | MORITA Kazutaka | } |
1671 | 33b1db1c | MORITA Kazutaka | |
1672 | 33b1db1c | MORITA Kazutaka | ret = add_aio_request(s, aio_req, acb->qiov->iov, acb->qiov->niov, |
1673 | 33b1db1c | MORITA Kazutaka | create, acb->aiocb_type); |
1674 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
1675 | 6daf194d | Markus Armbruster | error_report("add_aio_request is failed");
|
1676 | 33b1db1c | MORITA Kazutaka | free_aio_req(s, aio_req); |
1677 | 33b1db1c | MORITA Kazutaka | acb->ret = -EIO; |
1678 | 33b1db1c | MORITA Kazutaka | goto out;
|
1679 | 33b1db1c | MORITA Kazutaka | } |
1680 | 33b1db1c | MORITA Kazutaka | done:
|
1681 | 33b1db1c | MORITA Kazutaka | offset = 0;
|
1682 | 33b1db1c | MORITA Kazutaka | idx++; |
1683 | 33b1db1c | MORITA Kazutaka | done += len; |
1684 | 33b1db1c | MORITA Kazutaka | } |
1685 | 33b1db1c | MORITA Kazutaka | out:
|
1686 | 33b1db1c | MORITA Kazutaka | if (QLIST_EMPTY(&acb->aioreq_head)) {
|
1687 | 33b1db1c | MORITA Kazutaka | sd_finish_aiocb(acb); |
1688 | 33b1db1c | MORITA Kazutaka | } |
1689 | 33b1db1c | MORITA Kazutaka | } |
1690 | 33b1db1c | MORITA Kazutaka | |
1691 | 33b1db1c | MORITA Kazutaka | static BlockDriverAIOCB *sd_aio_writev(BlockDriverState *bs, int64_t sector_num,
|
1692 | 33b1db1c | MORITA Kazutaka | QEMUIOVector *qiov, int nb_sectors,
|
1693 | 33b1db1c | MORITA Kazutaka | BlockDriverCompletionFunc *cb, |
1694 | 33b1db1c | MORITA Kazutaka | void *opaque)
|
1695 | 33b1db1c | MORITA Kazutaka | { |
1696 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *acb; |
1697 | 33b1db1c | MORITA Kazutaka | |
1698 | 33b1db1c | MORITA Kazutaka | if (bs->growable && sector_num + nb_sectors > bs->total_sectors) {
|
1699 | 33b1db1c | MORITA Kazutaka | /* TODO: shouldn't block here */
|
1700 | 33b1db1c | MORITA Kazutaka | if (sd_truncate(bs, (sector_num + nb_sectors) * SECTOR_SIZE) < 0) { |
1701 | 33b1db1c | MORITA Kazutaka | return NULL; |
1702 | 33b1db1c | MORITA Kazutaka | } |
1703 | 33b1db1c | MORITA Kazutaka | bs->total_sectors = sector_num + nb_sectors; |
1704 | 33b1db1c | MORITA Kazutaka | } |
1705 | 33b1db1c | MORITA Kazutaka | |
1706 | 33b1db1c | MORITA Kazutaka | acb = sd_aio_setup(bs, qiov, sector_num, nb_sectors, cb, opaque); |
1707 | 33b1db1c | MORITA Kazutaka | acb->aio_done_func = sd_write_done; |
1708 | 33b1db1c | MORITA Kazutaka | acb->aiocb_type = AIOCB_WRITE_UDATA; |
1709 | 33b1db1c | MORITA Kazutaka | |
1710 | 33b1db1c | MORITA Kazutaka | sd_schedule_bh(sd_readv_writev_bh_cb, acb); |
1711 | 33b1db1c | MORITA Kazutaka | return &acb->common;
|
1712 | 33b1db1c | MORITA Kazutaka | } |
1713 | 33b1db1c | MORITA Kazutaka | |
1714 | 33b1db1c | MORITA Kazutaka | static BlockDriverAIOCB *sd_aio_readv(BlockDriverState *bs, int64_t sector_num,
|
1715 | 33b1db1c | MORITA Kazutaka | QEMUIOVector *qiov, int nb_sectors,
|
1716 | 33b1db1c | MORITA Kazutaka | BlockDriverCompletionFunc *cb, |
1717 | 33b1db1c | MORITA Kazutaka | void *opaque)
|
1718 | 33b1db1c | MORITA Kazutaka | { |
1719 | 33b1db1c | MORITA Kazutaka | SheepdogAIOCB *acb; |
1720 | 33b1db1c | MORITA Kazutaka | int i;
|
1721 | 33b1db1c | MORITA Kazutaka | |
1722 | 33b1db1c | MORITA Kazutaka | acb = sd_aio_setup(bs, qiov, sector_num, nb_sectors, cb, opaque); |
1723 | 33b1db1c | MORITA Kazutaka | acb->aiocb_type = AIOCB_READ_UDATA; |
1724 | 33b1db1c | MORITA Kazutaka | acb->aio_done_func = sd_finish_aiocb; |
1725 | 33b1db1c | MORITA Kazutaka | |
1726 | 33b1db1c | MORITA Kazutaka | /*
|
1727 | 33b1db1c | MORITA Kazutaka | * TODO: we can do better; we don't need to initialize
|
1728 | 33b1db1c | MORITA Kazutaka | * blindly.
|
1729 | 33b1db1c | MORITA Kazutaka | */
|
1730 | 33b1db1c | MORITA Kazutaka | for (i = 0; i < qiov->niov; i++) { |
1731 | 33b1db1c | MORITA Kazutaka | memset(qiov->iov[i].iov_base, 0, qiov->iov[i].iov_len);
|
1732 | 33b1db1c | MORITA Kazutaka | } |
1733 | 33b1db1c | MORITA Kazutaka | |
1734 | 33b1db1c | MORITA Kazutaka | sd_schedule_bh(sd_readv_writev_bh_cb, acb); |
1735 | 33b1db1c | MORITA Kazutaka | return &acb->common;
|
1736 | 33b1db1c | MORITA Kazutaka | } |
1737 | 33b1db1c | MORITA Kazutaka | |
1738 | 33b1db1c | MORITA Kazutaka | static int sd_snapshot_create(BlockDriverState *bs, QEMUSnapshotInfo *sn_info) |
1739 | 33b1db1c | MORITA Kazutaka | { |
1740 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
1741 | 33b1db1c | MORITA Kazutaka | int ret, fd;
|
1742 | 33b1db1c | MORITA Kazutaka | uint32_t new_vid; |
1743 | 33b1db1c | MORITA Kazutaka | SheepdogInode *inode; |
1744 | 33b1db1c | MORITA Kazutaka | unsigned int datalen; |
1745 | 33b1db1c | MORITA Kazutaka | |
1746 | 33b1db1c | MORITA Kazutaka | dprintf("sn_info: name %s id_str %s s: name %s vm_state_size %d "
|
1747 | 33b1db1c | MORITA Kazutaka | "is_snapshot %d\n", sn_info->name, sn_info->id_str,
|
1748 | 33b1db1c | MORITA Kazutaka | s->name, sn_info->vm_state_size, s->is_snapshot); |
1749 | 33b1db1c | MORITA Kazutaka | |
1750 | 33b1db1c | MORITA Kazutaka | if (s->is_snapshot) {
|
1751 | 33b1db1c | MORITA Kazutaka | error_report("You can't create a snapshot of a snapshot VDI, "
|
1752 | 6daf194d | Markus Armbruster | "%s (%" PRIu32 ").", s->name, s->inode.vdi_id); |
1753 | 33b1db1c | MORITA Kazutaka | |
1754 | 33b1db1c | MORITA Kazutaka | return -EINVAL;
|
1755 | 33b1db1c | MORITA Kazutaka | } |
1756 | 33b1db1c | MORITA Kazutaka | |
1757 | 33b1db1c | MORITA Kazutaka | dprintf("%s %s\n", sn_info->name, sn_info->id_str);
|
1758 | 33b1db1c | MORITA Kazutaka | |
1759 | 33b1db1c | MORITA Kazutaka | s->inode.vm_state_size = sn_info->vm_state_size; |
1760 | 33b1db1c | MORITA Kazutaka | s->inode.vm_clock_nsec = sn_info->vm_clock_nsec; |
1761 | 33b1db1c | MORITA Kazutaka | strncpy(s->inode.tag, sn_info->name, sizeof(s->inode.tag));
|
1762 | 33b1db1c | MORITA Kazutaka | /* we don't need to update entire object */
|
1763 | 33b1db1c | MORITA Kazutaka | datalen = SD_INODE_SIZE - sizeof(s->inode.data_vdi_id);
|
1764 | 33b1db1c | MORITA Kazutaka | |
1765 | 33b1db1c | MORITA Kazutaka | /* refresh inode. */
|
1766 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1767 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1768 | 33b1db1c | MORITA Kazutaka | ret = -EIO; |
1769 | 33b1db1c | MORITA Kazutaka | goto cleanup;
|
1770 | 33b1db1c | MORITA Kazutaka | } |
1771 | 33b1db1c | MORITA Kazutaka | |
1772 | 33b1db1c | MORITA Kazutaka | ret = write_object(fd, (char *)&s->inode, vid_to_vdi_oid(s->inode.vdi_id),
|
1773 | 33b1db1c | MORITA Kazutaka | s->inode.nr_copies, datalen, 0, 0); |
1774 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
1775 | 6daf194d | Markus Armbruster | error_report("failed to write snapshot's inode.");
|
1776 | 33b1db1c | MORITA Kazutaka | ret = -EIO; |
1777 | 33b1db1c | MORITA Kazutaka | goto cleanup;
|
1778 | 33b1db1c | MORITA Kazutaka | } |
1779 | 33b1db1c | MORITA Kazutaka | |
1780 | 33b1db1c | MORITA Kazutaka | ret = do_sd_create(s->name, s->inode.vdi_size, s->inode.vdi_id, &new_vid, 1,
|
1781 | 33b1db1c | MORITA Kazutaka | s->addr, s->port); |
1782 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
1783 | 6daf194d | Markus Armbruster | error_report("failed to create inode for snapshot. %s",
|
1784 | 33b1db1c | MORITA Kazutaka | strerror(errno)); |
1785 | 33b1db1c | MORITA Kazutaka | ret = -EIO; |
1786 | 33b1db1c | MORITA Kazutaka | goto cleanup;
|
1787 | 33b1db1c | MORITA Kazutaka | } |
1788 | 33b1db1c | MORITA Kazutaka | |
1789 | 33b1db1c | MORITA Kazutaka | inode = (SheepdogInode *)qemu_malloc(datalen); |
1790 | 33b1db1c | MORITA Kazutaka | |
1791 | 33b1db1c | MORITA Kazutaka | ret = read_object(fd, (char *)inode, vid_to_vdi_oid(new_vid),
|
1792 | 33b1db1c | MORITA Kazutaka | s->inode.nr_copies, datalen, 0);
|
1793 | 33b1db1c | MORITA Kazutaka | |
1794 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
1795 | 6daf194d | Markus Armbruster | error_report("failed to read new inode info. %s", strerror(errno));
|
1796 | 33b1db1c | MORITA Kazutaka | ret = -EIO; |
1797 | 33b1db1c | MORITA Kazutaka | goto cleanup;
|
1798 | 33b1db1c | MORITA Kazutaka | } |
1799 | 33b1db1c | MORITA Kazutaka | |
1800 | 33b1db1c | MORITA Kazutaka | memcpy(&s->inode, inode, datalen); |
1801 | 33b1db1c | MORITA Kazutaka | dprintf("s->inode: name %s snap_id %x oid %x\n",
|
1802 | 33b1db1c | MORITA Kazutaka | s->inode.name, s->inode.snap_id, s->inode.vdi_id); |
1803 | 33b1db1c | MORITA Kazutaka | |
1804 | 33b1db1c | MORITA Kazutaka | cleanup:
|
1805 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1806 | 33b1db1c | MORITA Kazutaka | return ret;
|
1807 | 33b1db1c | MORITA Kazutaka | } |
1808 | 33b1db1c | MORITA Kazutaka | |
1809 | 33b1db1c | MORITA Kazutaka | static int sd_snapshot_goto(BlockDriverState *bs, const char *snapshot_id) |
1810 | 33b1db1c | MORITA Kazutaka | { |
1811 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
1812 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *old_s; |
1813 | 33b1db1c | MORITA Kazutaka | char vdi[SD_MAX_VDI_LEN], tag[SD_MAX_VDI_TAG_LEN];
|
1814 | 33b1db1c | MORITA Kazutaka | char *buf = NULL; |
1815 | 33b1db1c | MORITA Kazutaka | uint32_t vid; |
1816 | 33b1db1c | MORITA Kazutaka | uint32_t snapid = 0;
|
1817 | 33b1db1c | MORITA Kazutaka | int ret = -ENOENT, fd;
|
1818 | 33b1db1c | MORITA Kazutaka | |
1819 | 33b1db1c | MORITA Kazutaka | old_s = qemu_malloc(sizeof(BDRVSheepdogState));
|
1820 | 33b1db1c | MORITA Kazutaka | |
1821 | 33b1db1c | MORITA Kazutaka | memcpy(old_s, s, sizeof(BDRVSheepdogState));
|
1822 | 33b1db1c | MORITA Kazutaka | |
1823 | 33b1db1c | MORITA Kazutaka | memset(vdi, 0, sizeof(vdi)); |
1824 | 33b1db1c | MORITA Kazutaka | strncpy(vdi, s->name, sizeof(vdi));
|
1825 | 33b1db1c | MORITA Kazutaka | |
1826 | 33b1db1c | MORITA Kazutaka | memset(tag, 0, sizeof(tag)); |
1827 | 33b1db1c | MORITA Kazutaka | snapid = strtoul(snapshot_id, NULL, 10); |
1828 | 33b1db1c | MORITA Kazutaka | if (!snapid) {
|
1829 | 33b1db1c | MORITA Kazutaka | strncpy(tag, s->name, sizeof(tag));
|
1830 | 33b1db1c | MORITA Kazutaka | } |
1831 | 33b1db1c | MORITA Kazutaka | |
1832 | 33b1db1c | MORITA Kazutaka | ret = find_vdi_name(s, vdi, snapid, tag, &vid, 1);
|
1833 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1834 | 6daf194d | Markus Armbruster | error_report("Failed to find_vdi_name");
|
1835 | 33b1db1c | MORITA Kazutaka | ret = -ENOENT; |
1836 | 33b1db1c | MORITA Kazutaka | goto out;
|
1837 | 33b1db1c | MORITA Kazutaka | } |
1838 | 33b1db1c | MORITA Kazutaka | |
1839 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1840 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1841 | 6daf194d | Markus Armbruster | error_report("failed to connect");
|
1842 | 33b1db1c | MORITA Kazutaka | goto out;
|
1843 | 33b1db1c | MORITA Kazutaka | } |
1844 | 33b1db1c | MORITA Kazutaka | |
1845 | 33b1db1c | MORITA Kazutaka | buf = qemu_malloc(SD_INODE_SIZE); |
1846 | 33b1db1c | MORITA Kazutaka | ret = read_object(fd, buf, vid_to_vdi_oid(vid), s->inode.nr_copies, |
1847 | 33b1db1c | MORITA Kazutaka | SD_INODE_SIZE, 0);
|
1848 | 33b1db1c | MORITA Kazutaka | |
1849 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1850 | 33b1db1c | MORITA Kazutaka | |
1851 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1852 | 33b1db1c | MORITA Kazutaka | ret = -ENOENT; |
1853 | 33b1db1c | MORITA Kazutaka | goto out;
|
1854 | 33b1db1c | MORITA Kazutaka | } |
1855 | 33b1db1c | MORITA Kazutaka | |
1856 | 33b1db1c | MORITA Kazutaka | memcpy(&s->inode, buf, sizeof(s->inode));
|
1857 | 33b1db1c | MORITA Kazutaka | |
1858 | 33b1db1c | MORITA Kazutaka | if (!s->inode.vm_state_size) {
|
1859 | 6daf194d | Markus Armbruster | error_report("Invalid snapshot");
|
1860 | 33b1db1c | MORITA Kazutaka | ret = -ENOENT; |
1861 | 33b1db1c | MORITA Kazutaka | goto out;
|
1862 | 33b1db1c | MORITA Kazutaka | } |
1863 | 33b1db1c | MORITA Kazutaka | |
1864 | 33b1db1c | MORITA Kazutaka | s->is_snapshot = 1;
|
1865 | 33b1db1c | MORITA Kazutaka | |
1866 | 33b1db1c | MORITA Kazutaka | qemu_free(buf); |
1867 | 33b1db1c | MORITA Kazutaka | qemu_free(old_s); |
1868 | 33b1db1c | MORITA Kazutaka | |
1869 | 33b1db1c | MORITA Kazutaka | return 0; |
1870 | 33b1db1c | MORITA Kazutaka | out:
|
1871 | 33b1db1c | MORITA Kazutaka | /* recover bdrv_sd_state */
|
1872 | 33b1db1c | MORITA Kazutaka | memcpy(s, old_s, sizeof(BDRVSheepdogState));
|
1873 | 33b1db1c | MORITA Kazutaka | qemu_free(buf); |
1874 | 33b1db1c | MORITA Kazutaka | qemu_free(old_s); |
1875 | 33b1db1c | MORITA Kazutaka | |
1876 | 6daf194d | Markus Armbruster | error_report("failed to open. recover old bdrv_sd_state.");
|
1877 | 33b1db1c | MORITA Kazutaka | |
1878 | 33b1db1c | MORITA Kazutaka | return ret;
|
1879 | 33b1db1c | MORITA Kazutaka | } |
1880 | 33b1db1c | MORITA Kazutaka | |
1881 | 33b1db1c | MORITA Kazutaka | static int sd_snapshot_delete(BlockDriverState *bs, const char *snapshot_id) |
1882 | 33b1db1c | MORITA Kazutaka | { |
1883 | 33b1db1c | MORITA Kazutaka | /* FIXME: Delete specified snapshot id. */
|
1884 | 33b1db1c | MORITA Kazutaka | return 0; |
1885 | 33b1db1c | MORITA Kazutaka | } |
1886 | 33b1db1c | MORITA Kazutaka | |
1887 | 33b1db1c | MORITA Kazutaka | static int sd_snapshot_list(BlockDriverState *bs, QEMUSnapshotInfo **psn_tab) |
1888 | 33b1db1c | MORITA Kazutaka | { |
1889 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
1890 | 33b1db1c | MORITA Kazutaka | SheepdogReq req; |
1891 | 33b1db1c | MORITA Kazutaka | int fd, nr = 1024, ret, max = BITS_TO_LONGS(SD_NR_VDIS) * sizeof(long); |
1892 | 33b1db1c | MORITA Kazutaka | QEMUSnapshotInfo *sn_tab = NULL;
|
1893 | 33b1db1c | MORITA Kazutaka | unsigned wlen, rlen;
|
1894 | 33b1db1c | MORITA Kazutaka | int found = 0; |
1895 | 33b1db1c | MORITA Kazutaka | static SheepdogInode inode;
|
1896 | 33b1db1c | MORITA Kazutaka | unsigned long *vdi_inuse; |
1897 | 33b1db1c | MORITA Kazutaka | unsigned int start_nr; |
1898 | 33b1db1c | MORITA Kazutaka | uint64_t hval; |
1899 | 33b1db1c | MORITA Kazutaka | uint32_t vid; |
1900 | 33b1db1c | MORITA Kazutaka | |
1901 | 33b1db1c | MORITA Kazutaka | vdi_inuse = qemu_malloc(max); |
1902 | 33b1db1c | MORITA Kazutaka | |
1903 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1904 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1905 | 33b1db1c | MORITA Kazutaka | goto out;
|
1906 | 33b1db1c | MORITA Kazutaka | } |
1907 | 33b1db1c | MORITA Kazutaka | |
1908 | 33b1db1c | MORITA Kazutaka | rlen = max; |
1909 | 33b1db1c | MORITA Kazutaka | wlen = 0;
|
1910 | 33b1db1c | MORITA Kazutaka | |
1911 | 33b1db1c | MORITA Kazutaka | memset(&req, 0, sizeof(req)); |
1912 | 33b1db1c | MORITA Kazutaka | |
1913 | 33b1db1c | MORITA Kazutaka | req.opcode = SD_OP_READ_VDIS; |
1914 | 33b1db1c | MORITA Kazutaka | req.data_length = max; |
1915 | 33b1db1c | MORITA Kazutaka | |
1916 | 33b1db1c | MORITA Kazutaka | ret = do_req(fd, (SheepdogReq *)&req, vdi_inuse, &wlen, &rlen); |
1917 | 33b1db1c | MORITA Kazutaka | |
1918 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1919 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1920 | 33b1db1c | MORITA Kazutaka | goto out;
|
1921 | 33b1db1c | MORITA Kazutaka | } |
1922 | 33b1db1c | MORITA Kazutaka | |
1923 | 33b1db1c | MORITA Kazutaka | sn_tab = qemu_mallocz(nr * sizeof(*sn_tab));
|
1924 | 33b1db1c | MORITA Kazutaka | |
1925 | 33b1db1c | MORITA Kazutaka | /* calculate a vdi id with hash function */
|
1926 | 33b1db1c | MORITA Kazutaka | hval = fnv_64a_buf(s->name, strlen(s->name), FNV1A_64_INIT); |
1927 | 33b1db1c | MORITA Kazutaka | start_nr = hval & (SD_NR_VDIS - 1);
|
1928 | 33b1db1c | MORITA Kazutaka | |
1929 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1930 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1931 | 6daf194d | Markus Armbruster | error_report("failed to connect");
|
1932 | 33b1db1c | MORITA Kazutaka | goto out;
|
1933 | 33b1db1c | MORITA Kazutaka | } |
1934 | 33b1db1c | MORITA Kazutaka | |
1935 | 33b1db1c | MORITA Kazutaka | for (vid = start_nr; found < nr; vid = (vid + 1) % SD_NR_VDIS) { |
1936 | 33b1db1c | MORITA Kazutaka | if (!test_bit(vid, vdi_inuse)) {
|
1937 | 33b1db1c | MORITA Kazutaka | break;
|
1938 | 33b1db1c | MORITA Kazutaka | } |
1939 | 33b1db1c | MORITA Kazutaka | |
1940 | 33b1db1c | MORITA Kazutaka | /* we don't need to read entire object */
|
1941 | 33b1db1c | MORITA Kazutaka | ret = read_object(fd, (char *)&inode, vid_to_vdi_oid(vid),
|
1942 | 33b1db1c | MORITA Kazutaka | 0, SD_INODE_SIZE - sizeof(inode.data_vdi_id), 0); |
1943 | 33b1db1c | MORITA Kazutaka | |
1944 | 33b1db1c | MORITA Kazutaka | if (ret) {
|
1945 | 33b1db1c | MORITA Kazutaka | continue;
|
1946 | 33b1db1c | MORITA Kazutaka | } |
1947 | 33b1db1c | MORITA Kazutaka | |
1948 | 33b1db1c | MORITA Kazutaka | if (!strcmp(inode.name, s->name) && is_snapshot(&inode)) {
|
1949 | 33b1db1c | MORITA Kazutaka | sn_tab[found].date_sec = inode.snap_ctime >> 32;
|
1950 | 33b1db1c | MORITA Kazutaka | sn_tab[found].date_nsec = inode.snap_ctime & 0xffffffff;
|
1951 | 33b1db1c | MORITA Kazutaka | sn_tab[found].vm_state_size = inode.vm_state_size; |
1952 | 33b1db1c | MORITA Kazutaka | sn_tab[found].vm_clock_nsec = inode.vm_clock_nsec; |
1953 | 33b1db1c | MORITA Kazutaka | |
1954 | 33b1db1c | MORITA Kazutaka | snprintf(sn_tab[found].id_str, sizeof(sn_tab[found].id_str), "%u", |
1955 | 33b1db1c | MORITA Kazutaka | inode.snap_id); |
1956 | 33b1db1c | MORITA Kazutaka | strncpy(sn_tab[found].name, inode.tag, |
1957 | 33b1db1c | MORITA Kazutaka | MIN(sizeof(sn_tab[found].name), sizeof(inode.tag))); |
1958 | 33b1db1c | MORITA Kazutaka | found++; |
1959 | 33b1db1c | MORITA Kazutaka | } |
1960 | 33b1db1c | MORITA Kazutaka | } |
1961 | 33b1db1c | MORITA Kazutaka | |
1962 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
1963 | 33b1db1c | MORITA Kazutaka | out:
|
1964 | 33b1db1c | MORITA Kazutaka | *psn_tab = sn_tab; |
1965 | 33b1db1c | MORITA Kazutaka | |
1966 | 33b1db1c | MORITA Kazutaka | qemu_free(vdi_inuse); |
1967 | 33b1db1c | MORITA Kazutaka | |
1968 | 33b1db1c | MORITA Kazutaka | return found;
|
1969 | 33b1db1c | MORITA Kazutaka | } |
1970 | 33b1db1c | MORITA Kazutaka | |
1971 | 33b1db1c | MORITA Kazutaka | static int do_load_save_vmstate(BDRVSheepdogState *s, uint8_t *data, |
1972 | 33b1db1c | MORITA Kazutaka | int64_t pos, int size, int load) |
1973 | 33b1db1c | MORITA Kazutaka | { |
1974 | 33b1db1c | MORITA Kazutaka | int fd, create;
|
1975 | 33b1db1c | MORITA Kazutaka | int ret = 0; |
1976 | 33b1db1c | MORITA Kazutaka | unsigned int data_len; |
1977 | 33b1db1c | MORITA Kazutaka | uint64_t vmstate_oid; |
1978 | 33b1db1c | MORITA Kazutaka | uint32_t vdi_index; |
1979 | 33b1db1c | MORITA Kazutaka | uint64_t offset; |
1980 | 33b1db1c | MORITA Kazutaka | |
1981 | 33b1db1c | MORITA Kazutaka | fd = connect_to_sdog(s->addr, s->port); |
1982 | 33b1db1c | MORITA Kazutaka | if (fd < 0) { |
1983 | 33b1db1c | MORITA Kazutaka | ret = -EIO; |
1984 | 33b1db1c | MORITA Kazutaka | goto cleanup;
|
1985 | 33b1db1c | MORITA Kazutaka | } |
1986 | 33b1db1c | MORITA Kazutaka | |
1987 | 33b1db1c | MORITA Kazutaka | while (size) {
|
1988 | 33b1db1c | MORITA Kazutaka | vdi_index = pos / SD_DATA_OBJ_SIZE; |
1989 | 33b1db1c | MORITA Kazutaka | offset = pos % SD_DATA_OBJ_SIZE; |
1990 | 33b1db1c | MORITA Kazutaka | |
1991 | 33b1db1c | MORITA Kazutaka | data_len = MIN(size, SD_DATA_OBJ_SIZE); |
1992 | 33b1db1c | MORITA Kazutaka | |
1993 | 33b1db1c | MORITA Kazutaka | vmstate_oid = vid_to_vmstate_oid(s->inode.vdi_id, vdi_index); |
1994 | 33b1db1c | MORITA Kazutaka | |
1995 | 33b1db1c | MORITA Kazutaka | create = (offset == 0);
|
1996 | 33b1db1c | MORITA Kazutaka | if (load) {
|
1997 | 33b1db1c | MORITA Kazutaka | ret = read_object(fd, (char *)data, vmstate_oid,
|
1998 | 33b1db1c | MORITA Kazutaka | s->inode.nr_copies, data_len, offset); |
1999 | 33b1db1c | MORITA Kazutaka | } else {
|
2000 | 33b1db1c | MORITA Kazutaka | ret = write_object(fd, (char *)data, vmstate_oid,
|
2001 | 33b1db1c | MORITA Kazutaka | s->inode.nr_copies, data_len, offset, create); |
2002 | 33b1db1c | MORITA Kazutaka | } |
2003 | 33b1db1c | MORITA Kazutaka | |
2004 | 33b1db1c | MORITA Kazutaka | if (ret < 0) { |
2005 | 6daf194d | Markus Armbruster | error_report("failed to save vmstate %s", strerror(errno));
|
2006 | 33b1db1c | MORITA Kazutaka | ret = -EIO; |
2007 | 33b1db1c | MORITA Kazutaka | goto cleanup;
|
2008 | 33b1db1c | MORITA Kazutaka | } |
2009 | 33b1db1c | MORITA Kazutaka | |
2010 | 33b1db1c | MORITA Kazutaka | pos += data_len; |
2011 | 33b1db1c | MORITA Kazutaka | size -= data_len; |
2012 | 33b1db1c | MORITA Kazutaka | ret += data_len; |
2013 | 33b1db1c | MORITA Kazutaka | } |
2014 | 33b1db1c | MORITA Kazutaka | cleanup:
|
2015 | 33b1db1c | MORITA Kazutaka | closesocket(fd); |
2016 | 33b1db1c | MORITA Kazutaka | return ret;
|
2017 | 33b1db1c | MORITA Kazutaka | } |
2018 | 33b1db1c | MORITA Kazutaka | |
2019 | 33b1db1c | MORITA Kazutaka | static int sd_save_vmstate(BlockDriverState *bs, const uint8_t *data, |
2020 | 33b1db1c | MORITA Kazutaka | int64_t pos, int size)
|
2021 | 33b1db1c | MORITA Kazutaka | { |
2022 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
2023 | 33b1db1c | MORITA Kazutaka | |
2024 | 33b1db1c | MORITA Kazutaka | return do_load_save_vmstate(s, (uint8_t *)data, pos, size, 0); |
2025 | 33b1db1c | MORITA Kazutaka | } |
2026 | 33b1db1c | MORITA Kazutaka | |
2027 | 33b1db1c | MORITA Kazutaka | static int sd_load_vmstate(BlockDriverState *bs, uint8_t *data, |
2028 | 33b1db1c | MORITA Kazutaka | int64_t pos, int size)
|
2029 | 33b1db1c | MORITA Kazutaka | { |
2030 | 33b1db1c | MORITA Kazutaka | BDRVSheepdogState *s = bs->opaque; |
2031 | 33b1db1c | MORITA Kazutaka | |
2032 | 33b1db1c | MORITA Kazutaka | return do_load_save_vmstate(s, data, pos, size, 1); |
2033 | 33b1db1c | MORITA Kazutaka | } |
2034 | 33b1db1c | MORITA Kazutaka | |
2035 | 33b1db1c | MORITA Kazutaka | |
2036 | 33b1db1c | MORITA Kazutaka | static QEMUOptionParameter sd_create_options[] = {
|
2037 | 33b1db1c | MORITA Kazutaka | { |
2038 | 33b1db1c | MORITA Kazutaka | .name = BLOCK_OPT_SIZE, |
2039 | 33b1db1c | MORITA Kazutaka | .type = OPT_SIZE, |
2040 | 33b1db1c | MORITA Kazutaka | .help = "Virtual disk size"
|
2041 | 33b1db1c | MORITA Kazutaka | }, |
2042 | 33b1db1c | MORITA Kazutaka | { |
2043 | 33b1db1c | MORITA Kazutaka | .name = BLOCK_OPT_BACKING_FILE, |
2044 | 33b1db1c | MORITA Kazutaka | .type = OPT_STRING, |
2045 | 33b1db1c | MORITA Kazutaka | .help = "File name of a base image"
|
2046 | 33b1db1c | MORITA Kazutaka | }, |
2047 | a8e0fdd7 | MORITA Kazutaka | { |
2048 | a8e0fdd7 | MORITA Kazutaka | .name = BLOCK_OPT_PREALLOC, |
2049 | a8e0fdd7 | MORITA Kazutaka | .type = OPT_STRING, |
2050 | a8e0fdd7 | MORITA Kazutaka | .help = "Preallocation mode (allowed values: off, full)"
|
2051 | a8e0fdd7 | MORITA Kazutaka | }, |
2052 | 33b1db1c | MORITA Kazutaka | { NULL }
|
2053 | 33b1db1c | MORITA Kazutaka | }; |
2054 | 33b1db1c | MORITA Kazutaka | |
2055 | 33b1db1c | MORITA Kazutaka | BlockDriver bdrv_sheepdog = { |
2056 | 33b1db1c | MORITA Kazutaka | .format_name = "sheepdog",
|
2057 | 33b1db1c | MORITA Kazutaka | .protocol_name = "sheepdog",
|
2058 | 33b1db1c | MORITA Kazutaka | .instance_size = sizeof(BDRVSheepdogState),
|
2059 | 33b1db1c | MORITA Kazutaka | .bdrv_file_open = sd_open, |
2060 | 33b1db1c | MORITA Kazutaka | .bdrv_close = sd_close, |
2061 | 33b1db1c | MORITA Kazutaka | .bdrv_create = sd_create, |
2062 | 33b1db1c | MORITA Kazutaka | .bdrv_getlength = sd_getlength, |
2063 | 33b1db1c | MORITA Kazutaka | .bdrv_truncate = sd_truncate, |
2064 | 33b1db1c | MORITA Kazutaka | |
2065 | 33b1db1c | MORITA Kazutaka | .bdrv_aio_readv = sd_aio_readv, |
2066 | 33b1db1c | MORITA Kazutaka | .bdrv_aio_writev = sd_aio_writev, |
2067 | 33b1db1c | MORITA Kazutaka | |
2068 | 33b1db1c | MORITA Kazutaka | .bdrv_snapshot_create = sd_snapshot_create, |
2069 | 33b1db1c | MORITA Kazutaka | .bdrv_snapshot_goto = sd_snapshot_goto, |
2070 | 33b1db1c | MORITA Kazutaka | .bdrv_snapshot_delete = sd_snapshot_delete, |
2071 | 33b1db1c | MORITA Kazutaka | .bdrv_snapshot_list = sd_snapshot_list, |
2072 | 33b1db1c | MORITA Kazutaka | |
2073 | 33b1db1c | MORITA Kazutaka | .bdrv_save_vmstate = sd_save_vmstate, |
2074 | 33b1db1c | MORITA Kazutaka | .bdrv_load_vmstate = sd_load_vmstate, |
2075 | 33b1db1c | MORITA Kazutaka | |
2076 | 33b1db1c | MORITA Kazutaka | .create_options = sd_create_options, |
2077 | 33b1db1c | MORITA Kazutaka | }; |
2078 | 33b1db1c | MORITA Kazutaka | |
2079 | 33b1db1c | MORITA Kazutaka | static void bdrv_sheepdog_init(void) |
2080 | 33b1db1c | MORITA Kazutaka | { |
2081 | 33b1db1c | MORITA Kazutaka | bdrv_register(&bdrv_sheepdog); |
2082 | 33b1db1c | MORITA Kazutaka | } |
2083 | 33b1db1c | MORITA Kazutaka | block_init(bdrv_sheepdog_init); |