Statistics
| Branch: | Revision:

root / block / sheepdog.c @ 80fa3341

History | View | Annotate | Download (51.5 kB)

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