Statistics
| Branch: | Revision:

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);