Statistics
| Branch: | Revision:

root / block / sheepdog.c @ 6b331efb

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