xref: /openbmc/qemu/block.c (revision 83f6409109acd7cd13ff8a603f29c46033c4fb00)
1fc01f7e7Sbellard /*
2fc01f7e7Sbellard  * QEMU System Emulator block driver
3fc01f7e7Sbellard  *
4fc01f7e7Sbellard  * Copyright (c) 2003 Fabrice Bellard
5fc01f7e7Sbellard  *
6fc01f7e7Sbellard  * Permission is hereby granted, free of charge, to any person obtaining a copy
7fc01f7e7Sbellard  * of this software and associated documentation files (the "Software"), to deal
8fc01f7e7Sbellard  * in the Software without restriction, including without limitation the rights
9fc01f7e7Sbellard  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10fc01f7e7Sbellard  * copies of the Software, and to permit persons to whom the Software is
11fc01f7e7Sbellard  * furnished to do so, subject to the following conditions:
12fc01f7e7Sbellard  *
13fc01f7e7Sbellard  * The above copyright notice and this permission notice shall be included in
14fc01f7e7Sbellard  * all copies or substantial portions of the Software.
15fc01f7e7Sbellard  *
16fc01f7e7Sbellard  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17fc01f7e7Sbellard  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18fc01f7e7Sbellard  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19fc01f7e7Sbellard  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20fc01f7e7Sbellard  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21fc01f7e7Sbellard  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22fc01f7e7Sbellard  * THE SOFTWARE.
23fc01f7e7Sbellard  */
24fc01f7e7Sbellard #include "vl.h"
25ea2384d3Sbellard #include "block_int.h"
26fc01f7e7Sbellard 
277674e7bfSbellard #ifdef _BSD
287674e7bfSbellard #include <sys/types.h>
297674e7bfSbellard #include <sys/stat.h>
307674e7bfSbellard #include <sys/ioctl.h>
317674e7bfSbellard #include <sys/queue.h>
327674e7bfSbellard #include <sys/disk.h>
337674e7bfSbellard #endif
347674e7bfSbellard 
35*83f64091Sbellard #define SECTOR_BITS 9
36*83f64091Sbellard #define SECTOR_SIZE (1 << SECTOR_BITS)
373b0d4f61Sbellard 
38*83f64091Sbellard static int bdrv_aio_new_em(BlockDriverAIOCB *acb);
39*83f64091Sbellard static int bdrv_aio_read_em(BlockDriverAIOCB *acb, int64_t sector_num,
40*83f64091Sbellard                               uint8_t *buf, int nb_sectors);
41*83f64091Sbellard static int bdrv_aio_write_em(BlockDriverAIOCB *acb, int64_t sector_num,
42*83f64091Sbellard                                const uint8_t *buf, int nb_sectors);
43*83f64091Sbellard static void bdrv_aio_cancel_em(BlockDriverAIOCB *acb);
44*83f64091Sbellard static void bdrv_aio_delete_em(BlockDriverAIOCB *acb);
45*83f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
46*83f64091Sbellard                         uint8_t *buf, int nb_sectors);
47*83f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
48*83f64091Sbellard                          const uint8_t *buf, int nb_sectors);
49ec530c81Sbellard 
50b338082bSbellard static BlockDriverState *bdrv_first;
51ea2384d3Sbellard static BlockDriver *first_drv;
52ea2384d3Sbellard 
53*83f64091Sbellard #ifdef _WIN32
54*83f64091Sbellard #define PATH_SEP '\\'
55*83f64091Sbellard #else
56*83f64091Sbellard #define PATH_SEP '/'
573b0d4f61Sbellard #endif
583b0d4f61Sbellard 
59*83f64091Sbellard int path_is_absolute(const char *path)
60*83f64091Sbellard {
61*83f64091Sbellard     const char *p;
62*83f64091Sbellard     p = strchr(path, ':');
63*83f64091Sbellard     if (p)
64*83f64091Sbellard         p++;
65*83f64091Sbellard     else
66*83f64091Sbellard         p = path;
67*83f64091Sbellard     return (*p == PATH_SEP);
68*83f64091Sbellard }
69*83f64091Sbellard 
70*83f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
71*83f64091Sbellard    path to it by considering it is relative to base_path. URL are
72*83f64091Sbellard    supported. */
73*83f64091Sbellard void path_combine(char *dest, int dest_size,
74*83f64091Sbellard                   const char *base_path,
75*83f64091Sbellard                   const char *filename)
76*83f64091Sbellard {
77*83f64091Sbellard     const char *p, *p1;
78*83f64091Sbellard     int len;
79*83f64091Sbellard 
80*83f64091Sbellard     if (dest_size <= 0)
81*83f64091Sbellard         return;
82*83f64091Sbellard     if (path_is_absolute(filename)) {
83*83f64091Sbellard         pstrcpy(dest, dest_size, filename);
84*83f64091Sbellard     } else {
85*83f64091Sbellard         p = strchr(base_path, ':');
86*83f64091Sbellard         if (p)
87*83f64091Sbellard             p++;
88*83f64091Sbellard         else
89*83f64091Sbellard             p = base_path;
90*83f64091Sbellard         p1 = strrchr(base_path, PATH_SEP);
91*83f64091Sbellard         if (p1)
92*83f64091Sbellard             p1++;
93*83f64091Sbellard         else
94*83f64091Sbellard             p1 = base_path;
95*83f64091Sbellard         if (p1 > p)
96*83f64091Sbellard             p = p1;
97*83f64091Sbellard         len = p - base_path;
98*83f64091Sbellard         if (len > dest_size - 1)
99*83f64091Sbellard             len = dest_size - 1;
100*83f64091Sbellard         memcpy(dest, base_path, len);
101*83f64091Sbellard         dest[len] = '\0';
102*83f64091Sbellard         pstrcat(dest, dest_size, filename);
103*83f64091Sbellard     }
104*83f64091Sbellard }
105*83f64091Sbellard 
106*83f64091Sbellard 
107ea2384d3Sbellard void bdrv_register(BlockDriver *bdrv)
108ea2384d3Sbellard {
109*83f64091Sbellard     if (!bdrv->bdrv_aio_new) {
110*83f64091Sbellard         /* add AIO emulation layer */
111*83f64091Sbellard         bdrv->bdrv_aio_new = bdrv_aio_new_em;
112*83f64091Sbellard         bdrv->bdrv_aio_read = bdrv_aio_read_em;
113*83f64091Sbellard         bdrv->bdrv_aio_write = bdrv_aio_write_em;
114*83f64091Sbellard         bdrv->bdrv_aio_cancel = bdrv_aio_cancel_em;
115*83f64091Sbellard         bdrv->bdrv_aio_delete = bdrv_aio_delete_em;
116*83f64091Sbellard     } else if (!bdrv->bdrv_read && !bdrv->bdrv_pread) {
117*83f64091Sbellard         /* add synchronous IO emulation layer */
118*83f64091Sbellard         bdrv->bdrv_read = bdrv_read_em;
119*83f64091Sbellard         bdrv->bdrv_write = bdrv_write_em;
120*83f64091Sbellard     }
121ea2384d3Sbellard     bdrv->next = first_drv;
122ea2384d3Sbellard     first_drv = bdrv;
123ea2384d3Sbellard }
124b338082bSbellard 
125b338082bSbellard /* create a new block device (by default it is empty) */
126b338082bSbellard BlockDriverState *bdrv_new(const char *device_name)
127fc01f7e7Sbellard {
128b338082bSbellard     BlockDriverState **pbs, *bs;
129b338082bSbellard 
130b338082bSbellard     bs = qemu_mallocz(sizeof(BlockDriverState));
131b338082bSbellard     if(!bs)
132b338082bSbellard         return NULL;
133b338082bSbellard     pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
134ea2384d3Sbellard     if (device_name[0] != '\0') {
135b338082bSbellard         /* insert at the end */
136b338082bSbellard         pbs = &bdrv_first;
137b338082bSbellard         while (*pbs != NULL)
138b338082bSbellard             pbs = &(*pbs)->next;
139b338082bSbellard         *pbs = bs;
140ea2384d3Sbellard     }
141b338082bSbellard     return bs;
142b338082bSbellard }
143b338082bSbellard 
144ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
145ea2384d3Sbellard {
146ea2384d3Sbellard     BlockDriver *drv1;
147ea2384d3Sbellard     for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
148ea2384d3Sbellard         if (!strcmp(drv1->format_name, format_name))
149ea2384d3Sbellard             return drv1;
150ea2384d3Sbellard     }
151ea2384d3Sbellard     return NULL;
152ea2384d3Sbellard }
153ea2384d3Sbellard 
154ea2384d3Sbellard int bdrv_create(BlockDriver *drv,
155ea2384d3Sbellard                 const char *filename, int64_t size_in_sectors,
156ea2384d3Sbellard                 const char *backing_file, int flags)
157ea2384d3Sbellard {
158ea2384d3Sbellard     if (!drv->bdrv_create)
159ea2384d3Sbellard         return -ENOTSUP;
160ea2384d3Sbellard     return drv->bdrv_create(filename, size_in_sectors, backing_file, flags);
161ea2384d3Sbellard }
162ea2384d3Sbellard 
163d5249393Sbellard #ifdef _WIN32
16495389c86Sbellard void get_tmp_filename(char *filename, int size)
165d5249393Sbellard {
166*83f64091Sbellard     tmpnam(filename);
167d5249393Sbellard }
168d5249393Sbellard #else
16995389c86Sbellard void get_tmp_filename(char *filename, int size)
170ea2384d3Sbellard {
171ea2384d3Sbellard     int fd;
172d5249393Sbellard     /* XXX: race condition possible */
173ea2384d3Sbellard     pstrcpy(filename, size, "/tmp/vl.XXXXXX");
174ea2384d3Sbellard     fd = mkstemp(filename);
175ea2384d3Sbellard     close(fd);
176ea2384d3Sbellard }
177d5249393Sbellard #endif
178ea2384d3Sbellard 
179*83f64091Sbellard static BlockDriver *find_protocol(const char *filename)
180*83f64091Sbellard {
181*83f64091Sbellard     BlockDriver *drv1;
182*83f64091Sbellard     char protocol[128];
183*83f64091Sbellard     int len;
184*83f64091Sbellard     const char *p;
185*83f64091Sbellard     p = strchr(filename, ':');
186*83f64091Sbellard     if (!p)
187*83f64091Sbellard         return &bdrv_raw;
188*83f64091Sbellard     len = p - filename;
189*83f64091Sbellard     if (len > sizeof(protocol) - 1)
190*83f64091Sbellard         len = sizeof(protocol) - 1;
191*83f64091Sbellard #ifdef _WIN32
192*83f64091Sbellard     if (len == 1) {
193*83f64091Sbellard         /* specific win32 case for driver letters */
194*83f64091Sbellard         return &bdrv_raw;
195*83f64091Sbellard     }
196*83f64091Sbellard #endif
197*83f64091Sbellard     memcpy(protocol, filename, len);
198*83f64091Sbellard     protocol[len] = '\0';
199*83f64091Sbellard     for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
200*83f64091Sbellard         if (drv1->protocol_name &&
201*83f64091Sbellard             !strcmp(drv1->protocol_name, protocol))
202*83f64091Sbellard             return drv1;
203*83f64091Sbellard     }
204*83f64091Sbellard     return NULL;
205*83f64091Sbellard }
206*83f64091Sbellard 
2077674e7bfSbellard /* XXX: force raw format if block or character device ? It would
2087674e7bfSbellard    simplify the BSD case */
209ea2384d3Sbellard static BlockDriver *find_image_format(const char *filename)
210ea2384d3Sbellard {
211*83f64091Sbellard     int ret, score, score_max;
212ea2384d3Sbellard     BlockDriver *drv1, *drv;
213*83f64091Sbellard     uint8_t buf[2048];
214*83f64091Sbellard     BlockDriverState *bs;
215ea2384d3Sbellard 
216*83f64091Sbellard     drv = find_protocol(filename);
217*83f64091Sbellard     /* no need to test disk image formats for vvfat or host specific
218*83f64091Sbellard        devices */
219*83f64091Sbellard     if (drv == &bdrv_vvfat)
220*83f64091Sbellard         return drv;
221*83f64091Sbellard     if (strstart(filename, "/dev/", NULL))
222*83f64091Sbellard         return &bdrv_raw;
223*83f64091Sbellard 
224*83f64091Sbellard     ret = bdrv_file_open(&bs, filename, BDRV_O_RDONLY);
225*83f64091Sbellard     if (ret < 0)
2267674e7bfSbellard         return NULL;
227*83f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
228*83f64091Sbellard     bdrv_delete(bs);
229ea2384d3Sbellard     if (ret < 0) {
230ea2384d3Sbellard         return NULL;
231ea2384d3Sbellard     }
232ea2384d3Sbellard 
233ea2384d3Sbellard     score_max = 0;
234ea2384d3Sbellard     for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
235*83f64091Sbellard         if (drv1->bdrv_probe) {
236ea2384d3Sbellard             score = drv1->bdrv_probe(buf, ret, filename);
237ea2384d3Sbellard             if (score > score_max) {
238ea2384d3Sbellard                 score_max = score;
239ea2384d3Sbellard                 drv = drv1;
240ea2384d3Sbellard             }
241ea2384d3Sbellard         }
242*83f64091Sbellard     }
243ea2384d3Sbellard     return drv;
244ea2384d3Sbellard }
245ea2384d3Sbellard 
246*83f64091Sbellard int bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags)
247b338082bSbellard {
248*83f64091Sbellard     BlockDriverState *bs;
249*83f64091Sbellard     int ret;
2503b0d4f61Sbellard 
251*83f64091Sbellard     bs = bdrv_new("");
252*83f64091Sbellard     if (!bs)
253*83f64091Sbellard         return -ENOMEM;
254*83f64091Sbellard     ret = bdrv_open2(bs, filename, flags | BDRV_O_FILE, NULL);
255*83f64091Sbellard     if (ret < 0) {
256*83f64091Sbellard         bdrv_delete(bs);
257*83f64091Sbellard         return ret;
2583b0d4f61Sbellard     }
259*83f64091Sbellard     *pbs = bs;
260*83f64091Sbellard     return 0;
2613b0d4f61Sbellard }
2623b0d4f61Sbellard 
263*83f64091Sbellard int bdrv_open(BlockDriverState *bs, const char *filename, int flags)
264*83f64091Sbellard {
265*83f64091Sbellard     return bdrv_open2(bs, filename, flags, NULL);
266ea2384d3Sbellard }
267ea2384d3Sbellard 
268*83f64091Sbellard int bdrv_open2(BlockDriverState *bs, const char *filename, int flags,
269ea2384d3Sbellard                BlockDriver *drv)
270ea2384d3Sbellard {
271*83f64091Sbellard     int ret, open_flags;
272ea2384d3Sbellard     char tmp_filename[1024];
273*83f64091Sbellard     char backing_filename[1024];
274fc01f7e7Sbellard 
2750849bf08Sbellard     bs->read_only = 0;
276ea2384d3Sbellard     bs->is_temporary = 0;
277ea2384d3Sbellard     bs->encrypted = 0;
27833e3963eSbellard 
279*83f64091Sbellard     if (flags & BDRV_O_SNAPSHOT) {
280ea2384d3Sbellard         BlockDriverState *bs1;
281ea2384d3Sbellard         int64_t total_size;
28233e3963eSbellard 
283ea2384d3Sbellard         /* if snapshot, we create a temporary backing file and open it
284ea2384d3Sbellard            instead of opening 'filename' directly */
285ea2384d3Sbellard 
286ea2384d3Sbellard         /* if there is a backing file, use it */
287ea2384d3Sbellard         bs1 = bdrv_new("");
288ea2384d3Sbellard         if (!bs1) {
289*83f64091Sbellard             return -ENOMEM;
290ea2384d3Sbellard         }
291ea2384d3Sbellard         if (bdrv_open(bs1, filename, 0) < 0) {
292ea2384d3Sbellard             bdrv_delete(bs1);
293ea2384d3Sbellard             return -1;
294ea2384d3Sbellard         }
295*83f64091Sbellard         total_size = bdrv_getlength(bs1) >> SECTOR_BITS;
296ea2384d3Sbellard         bdrv_delete(bs1);
297ea2384d3Sbellard 
298ea2384d3Sbellard         get_tmp_filename(tmp_filename, sizeof(tmp_filename));
299ea2384d3Sbellard         if (bdrv_create(&bdrv_qcow, tmp_filename,
300ea2384d3Sbellard                         total_size, filename, 0) < 0) {
301ea2384d3Sbellard             return -1;
302ea2384d3Sbellard         }
303ea2384d3Sbellard         filename = tmp_filename;
304ea2384d3Sbellard         bs->is_temporary = 1;
305ea2384d3Sbellard     }
306ea2384d3Sbellard 
307ea2384d3Sbellard     pstrcpy(bs->filename, sizeof(bs->filename), filename);
308*83f64091Sbellard     if (flags & BDRV_O_FILE) {
309*83f64091Sbellard         drv = find_protocol(filename);
310*83f64091Sbellard         if (!drv)
311*83f64091Sbellard             return -ENOENT;
312*83f64091Sbellard     } else {
313ea2384d3Sbellard         if (!drv) {
314ea2384d3Sbellard             drv = find_image_format(filename);
315ea2384d3Sbellard             if (!drv)
316ea2384d3Sbellard                 return -1;
317ea2384d3Sbellard         }
318*83f64091Sbellard     }
319ea2384d3Sbellard     bs->drv = drv;
320ea2384d3Sbellard     bs->opaque = qemu_mallocz(drv->instance_size);
321ea2384d3Sbellard     if (bs->opaque == NULL && drv->instance_size > 0)
322ea2384d3Sbellard         return -1;
323*83f64091Sbellard     /* Note: for compatibility, we open disk image files as RDWR, and
324*83f64091Sbellard        RDONLY as fallback */
325*83f64091Sbellard     if (!(flags & BDRV_O_FILE))
326*83f64091Sbellard         open_flags = BDRV_O_RDWR;
327*83f64091Sbellard     else
328*83f64091Sbellard         open_flags = flags & ~(BDRV_O_FILE | BDRV_O_SNAPSHOT);
329*83f64091Sbellard     ret = drv->bdrv_open(bs, filename, open_flags);
330*83f64091Sbellard     if (ret == -EACCES && !(flags & BDRV_O_FILE)) {
331*83f64091Sbellard         ret = drv->bdrv_open(bs, filename, BDRV_O_RDONLY);
332*83f64091Sbellard         bs->read_only = 1;
333*83f64091Sbellard     }
334ea2384d3Sbellard     if (ret < 0) {
335ea2384d3Sbellard         qemu_free(bs->opaque);
336*83f64091Sbellard         return ret;
337ea2384d3Sbellard     }
338*83f64091Sbellard 
339ea2384d3Sbellard #ifndef _WIN32
340ea2384d3Sbellard     if (bs->is_temporary) {
341ea2384d3Sbellard         unlink(filename);
34233e3963eSbellard     }
34367b915a5Sbellard #endif
344*83f64091Sbellard     if (bs->backing_file[0] != '\0') {
345ea2384d3Sbellard         /* if there is a backing file, use it */
346ea2384d3Sbellard         bs->backing_hd = bdrv_new("");
347ea2384d3Sbellard         if (!bs->backing_hd) {
348ea2384d3Sbellard         fail:
349ea2384d3Sbellard             bdrv_close(bs);
350ea2384d3Sbellard             return -1;
351ea2384d3Sbellard         }
352*83f64091Sbellard         path_combine(backing_filename, sizeof(backing_filename),
353*83f64091Sbellard                      filename, bs->backing_file);
354*83f64091Sbellard         if (bdrv_open(bs->backing_hd, backing_filename, 0) < 0)
355ea2384d3Sbellard             goto fail;
356ea2384d3Sbellard     }
35733e3963eSbellard 
358b338082bSbellard     bs->inserted = 1;
359b338082bSbellard 
360b338082bSbellard     /* call the change callback */
361b338082bSbellard     if (bs->change_cb)
362b338082bSbellard         bs->change_cb(bs->change_opaque);
363b338082bSbellard 
364b338082bSbellard     return 0;
365fc01f7e7Sbellard }
366fc01f7e7Sbellard 
367fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
368fc01f7e7Sbellard {
369b338082bSbellard     if (bs->inserted) {
370ea2384d3Sbellard         if (bs->backing_hd)
371ea2384d3Sbellard             bdrv_delete(bs->backing_hd);
372ea2384d3Sbellard         bs->drv->bdrv_close(bs);
373ea2384d3Sbellard         qemu_free(bs->opaque);
374ea2384d3Sbellard #ifdef _WIN32
375ea2384d3Sbellard         if (bs->is_temporary) {
376ea2384d3Sbellard             unlink(bs->filename);
377ea2384d3Sbellard         }
37867b915a5Sbellard #endif
379ea2384d3Sbellard         bs->opaque = NULL;
380ea2384d3Sbellard         bs->drv = NULL;
381b338082bSbellard         bs->inserted = 0;
382b338082bSbellard 
383b338082bSbellard         /* call the change callback */
384b338082bSbellard         if (bs->change_cb)
385b338082bSbellard             bs->change_cb(bs->change_opaque);
386b338082bSbellard     }
387b338082bSbellard }
388b338082bSbellard 
389b338082bSbellard void bdrv_delete(BlockDriverState *bs)
390b338082bSbellard {
391ea2384d3Sbellard     /* XXX: remove the driver list */
392b338082bSbellard     bdrv_close(bs);
393b338082bSbellard     qemu_free(bs);
394fc01f7e7Sbellard }
395fc01f7e7Sbellard 
39633e3963eSbellard /* commit COW file into the raw image */
39733e3963eSbellard int bdrv_commit(BlockDriverState *bs)
39833e3963eSbellard {
399*83f64091Sbellard     int64_t i, total_sectors;
400ea2384d3Sbellard     int n, j;
401ea2384d3Sbellard     unsigned char sector[512];
40233e3963eSbellard 
403b338082bSbellard     if (!bs->inserted)
404ea2384d3Sbellard         return -ENOENT;
40533e3963eSbellard 
40633e3963eSbellard     if (bs->read_only) {
407ea2384d3Sbellard 	return -EACCES;
40833e3963eSbellard     }
40933e3963eSbellard 
410ea2384d3Sbellard     if (!bs->backing_hd) {
411ea2384d3Sbellard 	return -ENOTSUP;
412ea2384d3Sbellard     }
413ea2384d3Sbellard 
414*83f64091Sbellard     total_sectors = bdrv_getlength(bs) >> SECTOR_BITS;
415*83f64091Sbellard     for (i = 0; i < total_sectors;) {
416ea2384d3Sbellard         if (bs->drv->bdrv_is_allocated(bs, i, 65536, &n)) {
417ea2384d3Sbellard             for(j = 0; j < n; j++) {
41833e3963eSbellard                 if (bdrv_read(bs, i, sector, 1) != 0) {
419ea2384d3Sbellard                     return -EIO;
42033e3963eSbellard                 }
42133e3963eSbellard 
422ea2384d3Sbellard                 if (bdrv_write(bs->backing_hd, i, sector, 1) != 0) {
423ea2384d3Sbellard                     return -EIO;
42433e3963eSbellard                 }
425ea2384d3Sbellard                 i++;
426ea2384d3Sbellard 	    }
427ea2384d3Sbellard 	} else {
428ea2384d3Sbellard             i += n;
42933e3963eSbellard         }
43033e3963eSbellard     }
43195389c86Sbellard 
43295389c86Sbellard     if (bs->drv->bdrv_make_empty)
43395389c86Sbellard 	return bs->drv->bdrv_make_empty(bs);
43495389c86Sbellard 
43533e3963eSbellard     return 0;
43633e3963eSbellard }
43733e3963eSbellard 
438*83f64091Sbellard /* return < 0 if error */
439fc01f7e7Sbellard int bdrv_read(BlockDriverState *bs, int64_t sector_num,
440fc01f7e7Sbellard               uint8_t *buf, int nb_sectors)
441fc01f7e7Sbellard {
442ea2384d3Sbellard     BlockDriver *drv = bs->drv;
443fc01f7e7Sbellard 
444b338082bSbellard     if (!bs->inserted)
445b338082bSbellard         return -1;
446b338082bSbellard 
447*83f64091Sbellard     if (sector_num == 0 && bs->boot_sector_enabled && nb_sectors > 0) {
448cf98951bSbellard             memcpy(buf, bs->boot_sector_data, 512);
449*83f64091Sbellard         sector_num++;
450*83f64091Sbellard         nb_sectors--;
451*83f64091Sbellard         buf += 512;
452*83f64091Sbellard         if (nb_sectors == 0)
453fc01f7e7Sbellard             return 0;
454fc01f7e7Sbellard     }
455*83f64091Sbellard     if (drv->bdrv_pread) {
456*83f64091Sbellard         int ret, len;
457*83f64091Sbellard         len = nb_sectors * 512;
458*83f64091Sbellard         ret = drv->bdrv_pread(bs, sector_num * 512, buf, len);
459*83f64091Sbellard         if (ret < 0)
460*83f64091Sbellard             return ret;
461*83f64091Sbellard         else if (ret != len)
462*83f64091Sbellard             return -EIO;
463*83f64091Sbellard         else
464*83f64091Sbellard             return 0;
465*83f64091Sbellard     } else {
466*83f64091Sbellard         return drv->bdrv_read(bs, sector_num, buf, nb_sectors);
467*83f64091Sbellard     }
468*83f64091Sbellard }
469fc01f7e7Sbellard 
470*83f64091Sbellard /* return < 0 if error */
471fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num,
472fc01f7e7Sbellard                const uint8_t *buf, int nb_sectors)
473fc01f7e7Sbellard {
474*83f64091Sbellard     BlockDriver *drv = bs->drv;
475b338082bSbellard     if (!bs->inserted)
476b338082bSbellard         return -1;
4770849bf08Sbellard     if (bs->read_only)
4780849bf08Sbellard         return -1;
47979639d42Sbellard     if (sector_num == 0 && bs->boot_sector_enabled && nb_sectors > 0) {
48079639d42Sbellard         memcpy(bs->boot_sector_data, buf, 512);
48179639d42Sbellard     }
482*83f64091Sbellard     if (drv->bdrv_pwrite) {
483*83f64091Sbellard         int ret, len;
484*83f64091Sbellard         len = nb_sectors * 512;
485*83f64091Sbellard         ret = drv->bdrv_pwrite(bs, sector_num * 512, buf, len);
486*83f64091Sbellard         if (ret < 0)
487*83f64091Sbellard             return ret;
488*83f64091Sbellard         else if (ret != len)
489*83f64091Sbellard             return -EIO;
490*83f64091Sbellard         else
491*83f64091Sbellard             return 0;
492*83f64091Sbellard     } else {
493*83f64091Sbellard         return drv->bdrv_write(bs, sector_num, buf, nb_sectors);
494*83f64091Sbellard     }
495*83f64091Sbellard }
496*83f64091Sbellard 
497*83f64091Sbellard #if 0
498*83f64091Sbellard /* not necessary now */
499*83f64091Sbellard static int bdrv_pread_em(BlockDriverState *bs, int64_t offset,
500*83f64091Sbellard                          void *buf1, int count1)
501*83f64091Sbellard {
502*83f64091Sbellard     uint8_t *buf = buf1;
503*83f64091Sbellard     uint8_t tmp_buf[SECTOR_SIZE];
504*83f64091Sbellard     int len, nb_sectors, count;
505*83f64091Sbellard     int64_t sector_num;
506*83f64091Sbellard 
507*83f64091Sbellard     count = count1;
508*83f64091Sbellard     /* first read to align to sector start */
509*83f64091Sbellard     len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1);
510*83f64091Sbellard     if (len > count)
511*83f64091Sbellard         len = count;
512*83f64091Sbellard     sector_num = offset >> SECTOR_BITS;
513*83f64091Sbellard     if (len > 0) {
514*83f64091Sbellard         if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
515*83f64091Sbellard             return -EIO;
516*83f64091Sbellard         memcpy(buf, tmp_buf + (offset & (SECTOR_SIZE - 1)), len);
517*83f64091Sbellard         count -= len;
518*83f64091Sbellard         if (count == 0)
519*83f64091Sbellard             return count1;
520*83f64091Sbellard         sector_num++;
521*83f64091Sbellard         buf += len;
522*83f64091Sbellard     }
523*83f64091Sbellard 
524*83f64091Sbellard     /* read the sectors "in place" */
525*83f64091Sbellard     nb_sectors = count >> SECTOR_BITS;
526*83f64091Sbellard     if (nb_sectors > 0) {
527*83f64091Sbellard         if (bdrv_read(bs, sector_num, buf, nb_sectors) < 0)
528*83f64091Sbellard             return -EIO;
529*83f64091Sbellard         sector_num += nb_sectors;
530*83f64091Sbellard         len = nb_sectors << SECTOR_BITS;
531*83f64091Sbellard         buf += len;
532*83f64091Sbellard         count -= len;
533*83f64091Sbellard     }
534*83f64091Sbellard 
535*83f64091Sbellard     /* add data from the last sector */
536*83f64091Sbellard     if (count > 0) {
537*83f64091Sbellard         if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
538*83f64091Sbellard             return -EIO;
539*83f64091Sbellard         memcpy(buf, tmp_buf, count);
540*83f64091Sbellard     }
541*83f64091Sbellard     return count1;
542*83f64091Sbellard }
543*83f64091Sbellard 
544*83f64091Sbellard static int bdrv_pwrite_em(BlockDriverState *bs, int64_t offset,
545*83f64091Sbellard                           const void *buf1, int count1)
546*83f64091Sbellard {
547*83f64091Sbellard     const uint8_t *buf = buf1;
548*83f64091Sbellard     uint8_t tmp_buf[SECTOR_SIZE];
549*83f64091Sbellard     int len, nb_sectors, count;
550*83f64091Sbellard     int64_t sector_num;
551*83f64091Sbellard 
552*83f64091Sbellard     count = count1;
553*83f64091Sbellard     /* first write to align to sector start */
554*83f64091Sbellard     len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1);
555*83f64091Sbellard     if (len > count)
556*83f64091Sbellard         len = count;
557*83f64091Sbellard     sector_num = offset >> SECTOR_BITS;
558*83f64091Sbellard     if (len > 0) {
559*83f64091Sbellard         if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
560*83f64091Sbellard             return -EIO;
561*83f64091Sbellard         memcpy(tmp_buf + (offset & (SECTOR_SIZE - 1)), buf, len);
562*83f64091Sbellard         if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0)
563*83f64091Sbellard             return -EIO;
564*83f64091Sbellard         count -= len;
565*83f64091Sbellard         if (count == 0)
566*83f64091Sbellard             return count1;
567*83f64091Sbellard         sector_num++;
568*83f64091Sbellard         buf += len;
569*83f64091Sbellard     }
570*83f64091Sbellard 
571*83f64091Sbellard     /* write the sectors "in place" */
572*83f64091Sbellard     nb_sectors = count >> SECTOR_BITS;
573*83f64091Sbellard     if (nb_sectors > 0) {
574*83f64091Sbellard         if (bdrv_write(bs, sector_num, buf, nb_sectors) < 0)
575*83f64091Sbellard             return -EIO;
576*83f64091Sbellard         sector_num += nb_sectors;
577*83f64091Sbellard         len = nb_sectors << SECTOR_BITS;
578*83f64091Sbellard         buf += len;
579*83f64091Sbellard         count -= len;
580*83f64091Sbellard     }
581*83f64091Sbellard 
582*83f64091Sbellard     /* add data from the last sector */
583*83f64091Sbellard     if (count > 0) {
584*83f64091Sbellard         if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
585*83f64091Sbellard             return -EIO;
586*83f64091Sbellard         memcpy(tmp_buf, buf, count);
587*83f64091Sbellard         if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0)
588*83f64091Sbellard             return -EIO;
589*83f64091Sbellard     }
590*83f64091Sbellard     return count1;
591*83f64091Sbellard }
592*83f64091Sbellard #endif
593*83f64091Sbellard 
594*83f64091Sbellard /**
595*83f64091Sbellard  * Read with byte offsets (needed only for file protocols)
596*83f64091Sbellard  */
597*83f64091Sbellard int bdrv_pread(BlockDriverState *bs, int64_t offset,
598*83f64091Sbellard                void *buf1, int count1)
599*83f64091Sbellard {
600*83f64091Sbellard     BlockDriver *drv = bs->drv;
601*83f64091Sbellard 
602*83f64091Sbellard     if (!drv)
603*83f64091Sbellard         return -ENOENT;
604*83f64091Sbellard     if (!drv->bdrv_pread)
605*83f64091Sbellard         return -ENOTSUP;
606*83f64091Sbellard     return drv->bdrv_pread(bs, offset, buf1, count1);
607*83f64091Sbellard }
608*83f64091Sbellard 
609*83f64091Sbellard /**
610*83f64091Sbellard  * Write with byte offsets (needed only for file protocols)
611*83f64091Sbellard  */
612*83f64091Sbellard int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
613*83f64091Sbellard                 const void *buf1, int count1)
614*83f64091Sbellard {
615*83f64091Sbellard     BlockDriver *drv = bs->drv;
616*83f64091Sbellard 
617*83f64091Sbellard     if (!drv)
618*83f64091Sbellard         return -ENOENT;
619*83f64091Sbellard     if (!drv->bdrv_pwrite)
620*83f64091Sbellard         return -ENOTSUP;
621*83f64091Sbellard     return drv->bdrv_pwrite(bs, offset, buf1, count1);
622*83f64091Sbellard }
623*83f64091Sbellard 
624*83f64091Sbellard /**
625*83f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
626*83f64091Sbellard  */
627*83f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
628*83f64091Sbellard {
629*83f64091Sbellard     BlockDriver *drv = bs->drv;
630*83f64091Sbellard     if (!drv)
631*83f64091Sbellard         return -ENOENT;
632*83f64091Sbellard     if (!drv->bdrv_truncate)
633*83f64091Sbellard         return -ENOTSUP;
634*83f64091Sbellard     return drv->bdrv_truncate(bs, offset);
635*83f64091Sbellard }
636*83f64091Sbellard 
637*83f64091Sbellard /**
638*83f64091Sbellard  * Length of a file in bytes. Return < 0 if error or unknown.
639*83f64091Sbellard  */
640*83f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs)
641*83f64091Sbellard {
642*83f64091Sbellard     BlockDriver *drv = bs->drv;
643*83f64091Sbellard     if (!drv)
644*83f64091Sbellard         return -ENOENT;
645*83f64091Sbellard     if (!drv->bdrv_getlength) {
646*83f64091Sbellard         /* legacy mode */
647*83f64091Sbellard         return bs->total_sectors * SECTOR_SIZE;
648*83f64091Sbellard     }
649*83f64091Sbellard     return drv->bdrv_getlength(bs);
650fc01f7e7Sbellard }
651fc01f7e7Sbellard 
652fc01f7e7Sbellard void bdrv_get_geometry(BlockDriverState *bs, int64_t *nb_sectors_ptr)
653fc01f7e7Sbellard {
654*83f64091Sbellard     int64_t size;
655*83f64091Sbellard     size = bdrv_getlength(bs);
656*83f64091Sbellard     if (size < 0)
657*83f64091Sbellard         size = 0;
658*83f64091Sbellard     *nb_sectors_ptr = size >> SECTOR_BITS;
659fc01f7e7Sbellard }
660cf98951bSbellard 
661cf98951bSbellard /* force a given boot sector. */
662cf98951bSbellard void bdrv_set_boot_sector(BlockDriverState *bs, const uint8_t *data, int size)
663cf98951bSbellard {
664cf98951bSbellard     bs->boot_sector_enabled = 1;
665cf98951bSbellard     if (size > 512)
666cf98951bSbellard         size = 512;
667cf98951bSbellard     memcpy(bs->boot_sector_data, data, size);
668cf98951bSbellard     memset(bs->boot_sector_data + size, 0, 512 - size);
669cf98951bSbellard }
670b338082bSbellard 
671b338082bSbellard void bdrv_set_geometry_hint(BlockDriverState *bs,
672b338082bSbellard                             int cyls, int heads, int secs)
673b338082bSbellard {
674b338082bSbellard     bs->cyls = cyls;
675b338082bSbellard     bs->heads = heads;
676b338082bSbellard     bs->secs = secs;
677b338082bSbellard }
678b338082bSbellard 
679b338082bSbellard void bdrv_set_type_hint(BlockDriverState *bs, int type)
680b338082bSbellard {
681b338082bSbellard     bs->type = type;
682b338082bSbellard     bs->removable = ((type == BDRV_TYPE_CDROM ||
683b338082bSbellard                       type == BDRV_TYPE_FLOPPY));
684b338082bSbellard }
685b338082bSbellard 
68646d4767dSbellard void bdrv_set_translation_hint(BlockDriverState *bs, int translation)
68746d4767dSbellard {
68846d4767dSbellard     bs->translation = translation;
68946d4767dSbellard }
69046d4767dSbellard 
691b338082bSbellard void bdrv_get_geometry_hint(BlockDriverState *bs,
692b338082bSbellard                             int *pcyls, int *pheads, int *psecs)
693b338082bSbellard {
694b338082bSbellard     *pcyls = bs->cyls;
695b338082bSbellard     *pheads = bs->heads;
696b338082bSbellard     *psecs = bs->secs;
697b338082bSbellard }
698b338082bSbellard 
699b338082bSbellard int bdrv_get_type_hint(BlockDriverState *bs)
700b338082bSbellard {
701b338082bSbellard     return bs->type;
702b338082bSbellard }
703b338082bSbellard 
70446d4767dSbellard int bdrv_get_translation_hint(BlockDriverState *bs)
70546d4767dSbellard {
70646d4767dSbellard     return bs->translation;
70746d4767dSbellard }
70846d4767dSbellard 
709b338082bSbellard int bdrv_is_removable(BlockDriverState *bs)
710b338082bSbellard {
711b338082bSbellard     return bs->removable;
712b338082bSbellard }
713b338082bSbellard 
714b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
715b338082bSbellard {
716b338082bSbellard     return bs->read_only;
717b338082bSbellard }
718b338082bSbellard 
719b338082bSbellard int bdrv_is_inserted(BlockDriverState *bs)
720b338082bSbellard {
721b338082bSbellard     return bs->inserted;
722b338082bSbellard }
723b338082bSbellard 
724b338082bSbellard int bdrv_is_locked(BlockDriverState *bs)
725b338082bSbellard {
726b338082bSbellard     return bs->locked;
727b338082bSbellard }
728b338082bSbellard 
729b338082bSbellard void bdrv_set_locked(BlockDriverState *bs, int locked)
730b338082bSbellard {
731b338082bSbellard     bs->locked = locked;
732b338082bSbellard }
733b338082bSbellard 
734b338082bSbellard void bdrv_set_change_cb(BlockDriverState *bs,
735b338082bSbellard                         void (*change_cb)(void *opaque), void *opaque)
736b338082bSbellard {
737b338082bSbellard     bs->change_cb = change_cb;
738b338082bSbellard     bs->change_opaque = opaque;
739b338082bSbellard }
740b338082bSbellard 
741ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
742ea2384d3Sbellard {
743ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
744ea2384d3Sbellard         return 1;
745ea2384d3Sbellard     return bs->encrypted;
746ea2384d3Sbellard }
747ea2384d3Sbellard 
748ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
749ea2384d3Sbellard {
750ea2384d3Sbellard     int ret;
751ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
752ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
753ea2384d3Sbellard         if (ret < 0)
754ea2384d3Sbellard             return ret;
755ea2384d3Sbellard         if (!bs->encrypted)
756ea2384d3Sbellard             return 0;
757ea2384d3Sbellard     }
758ea2384d3Sbellard     if (!bs->encrypted || !bs->drv || !bs->drv->bdrv_set_key)
759ea2384d3Sbellard         return -1;
760ea2384d3Sbellard     return bs->drv->bdrv_set_key(bs, key);
761ea2384d3Sbellard }
762ea2384d3Sbellard 
763ea2384d3Sbellard void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size)
764ea2384d3Sbellard {
765ea2384d3Sbellard     if (!bs->inserted || !bs->drv) {
766ea2384d3Sbellard         buf[0] = '\0';
767ea2384d3Sbellard     } else {
768ea2384d3Sbellard         pstrcpy(buf, buf_size, bs->drv->format_name);
769ea2384d3Sbellard     }
770ea2384d3Sbellard }
771ea2384d3Sbellard 
772ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
773ea2384d3Sbellard                          void *opaque)
774ea2384d3Sbellard {
775ea2384d3Sbellard     BlockDriver *drv;
776ea2384d3Sbellard 
777ea2384d3Sbellard     for (drv = first_drv; drv != NULL; drv = drv->next) {
778ea2384d3Sbellard         it(opaque, drv->format_name);
779ea2384d3Sbellard     }
780ea2384d3Sbellard }
781ea2384d3Sbellard 
782b338082bSbellard BlockDriverState *bdrv_find(const char *name)
783b338082bSbellard {
784b338082bSbellard     BlockDriverState *bs;
785b338082bSbellard 
786b338082bSbellard     for (bs = bdrv_first; bs != NULL; bs = bs->next) {
787b338082bSbellard         if (!strcmp(name, bs->device_name))
788b338082bSbellard             return bs;
789b338082bSbellard     }
790b338082bSbellard     return NULL;
791b338082bSbellard }
792b338082bSbellard 
79381d0912dSbellard void bdrv_iterate(void (*it)(void *opaque, const char *name), void *opaque)
79481d0912dSbellard {
79581d0912dSbellard     BlockDriverState *bs;
79681d0912dSbellard 
79781d0912dSbellard     for (bs = bdrv_first; bs != NULL; bs = bs->next) {
79881d0912dSbellard         it(opaque, bs->device_name);
79981d0912dSbellard     }
80081d0912dSbellard }
80181d0912dSbellard 
802ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs)
803ea2384d3Sbellard {
804ea2384d3Sbellard     return bs->device_name;
805ea2384d3Sbellard }
806ea2384d3Sbellard 
8077a6cba61Spbrook void bdrv_flush(BlockDriverState *bs)
8087a6cba61Spbrook {
8097a6cba61Spbrook     if (bs->drv->bdrv_flush)
8107a6cba61Spbrook         bs->drv->bdrv_flush(bs);
8117a6cba61Spbrook     if (bs->backing_hd)
8127a6cba61Spbrook         bdrv_flush(bs->backing_hd);
8137a6cba61Spbrook }
8147a6cba61Spbrook 
815b338082bSbellard void bdrv_info(void)
816b338082bSbellard {
817b338082bSbellard     BlockDriverState *bs;
818b338082bSbellard 
819b338082bSbellard     for (bs = bdrv_first; bs != NULL; bs = bs->next) {
820b338082bSbellard         term_printf("%s:", bs->device_name);
821b338082bSbellard         term_printf(" type=");
822b338082bSbellard         switch(bs->type) {
823b338082bSbellard         case BDRV_TYPE_HD:
824b338082bSbellard             term_printf("hd");
825b338082bSbellard             break;
826b338082bSbellard         case BDRV_TYPE_CDROM:
827b338082bSbellard             term_printf("cdrom");
828b338082bSbellard             break;
829b338082bSbellard         case BDRV_TYPE_FLOPPY:
830b338082bSbellard             term_printf("floppy");
831b338082bSbellard             break;
832b338082bSbellard         }
833b338082bSbellard         term_printf(" removable=%d", bs->removable);
834b338082bSbellard         if (bs->removable) {
835b338082bSbellard             term_printf(" locked=%d", bs->locked);
836b338082bSbellard         }
837b338082bSbellard         if (bs->inserted) {
838b338082bSbellard             term_printf(" file=%s", bs->filename);
839ea2384d3Sbellard             if (bs->backing_file[0] != '\0')
840ea2384d3Sbellard                 term_printf(" backing_file=%s", bs->backing_file);
841b338082bSbellard             term_printf(" ro=%d", bs->read_only);
842ea2384d3Sbellard             term_printf(" drv=%s", bs->drv->format_name);
843ea2384d3Sbellard             if (bs->encrypted)
844ea2384d3Sbellard                 term_printf(" encrypted");
845b338082bSbellard         } else {
846b338082bSbellard             term_printf(" [not inserted]");
847b338082bSbellard         }
848b338082bSbellard         term_printf("\n");
849b338082bSbellard     }
850b338082bSbellard }
851ea2384d3Sbellard 
852*83f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
853*83f64091Sbellard                                char *filename, int filename_size)
854*83f64091Sbellard {
855*83f64091Sbellard     if (!bs->backing_hd) {
856*83f64091Sbellard         pstrcpy(filename, filename_size, "");
857*83f64091Sbellard     } else {
858*83f64091Sbellard         pstrcpy(filename, filename_size, bs->backing_file);
859*83f64091Sbellard     }
860*83f64091Sbellard }
861*83f64091Sbellard 
862*83f64091Sbellard 
863ea2384d3Sbellard /**************************************************************/
864*83f64091Sbellard /* async I/Os */
865ea2384d3Sbellard 
866*83f64091Sbellard BlockDriverAIOCB *bdrv_aio_new(BlockDriverState *bs)
867ea2384d3Sbellard {
868*83f64091Sbellard     BlockDriver *drv = bs->drv;
869*83f64091Sbellard     BlockDriverAIOCB *acb;
870*83f64091Sbellard     acb = qemu_mallocz(sizeof(BlockDriverAIOCB));
871*83f64091Sbellard     if (!acb)
872*83f64091Sbellard         return NULL;
873*83f64091Sbellard 
874*83f64091Sbellard     acb->bs = bs;
875*83f64091Sbellard     if (drv->bdrv_aio_new(acb) < 0) {
876*83f64091Sbellard         qemu_free(acb);
877*83f64091Sbellard         return NULL;
878*83f64091Sbellard     }
879*83f64091Sbellard     return acb;
880ea2384d3Sbellard }
881ea2384d3Sbellard 
882*83f64091Sbellard int bdrv_aio_read(BlockDriverAIOCB *acb, int64_t sector_num,
883*83f64091Sbellard                   uint8_t *buf, int nb_sectors,
884*83f64091Sbellard                   BlockDriverCompletionFunc *cb, void *opaque)
885ea2384d3Sbellard {
886*83f64091Sbellard     BlockDriverState *bs = acb->bs;
887*83f64091Sbellard     BlockDriver *drv = bs->drv;
888ea2384d3Sbellard 
889*83f64091Sbellard     if (!bs->inserted)
890ea2384d3Sbellard         return -1;
891*83f64091Sbellard 
892*83f64091Sbellard     /* XXX: we assume that nb_sectors == 0 is suppored by the async read */
893*83f64091Sbellard     if (sector_num == 0 && bs->boot_sector_enabled && nb_sectors > 0) {
894*83f64091Sbellard         memcpy(buf, bs->boot_sector_data, 512);
895*83f64091Sbellard         sector_num++;
896*83f64091Sbellard         nb_sectors--;
897*83f64091Sbellard         buf += 512;
898ea2384d3Sbellard     }
899*83f64091Sbellard 
900*83f64091Sbellard     acb->cb = cb;
901*83f64091Sbellard     acb->cb_opaque = opaque;
902*83f64091Sbellard     return drv->bdrv_aio_read(acb, sector_num, buf, nb_sectors);
903*83f64091Sbellard }
904*83f64091Sbellard 
905*83f64091Sbellard int bdrv_aio_write(BlockDriverAIOCB *acb, int64_t sector_num,
906*83f64091Sbellard                    const uint8_t *buf, int nb_sectors,
907*83f64091Sbellard                    BlockDriverCompletionFunc *cb, void *opaque)
9087674e7bfSbellard {
909*83f64091Sbellard     BlockDriverState *bs = acb->bs;
910*83f64091Sbellard     BlockDriver *drv = bs->drv;
911*83f64091Sbellard 
912*83f64091Sbellard     if (!bs->inserted)
913*83f64091Sbellard             return -1;
914*83f64091Sbellard     if (bs->read_only)
915*83f64091Sbellard         return -1;
916*83f64091Sbellard     if (sector_num == 0 && bs->boot_sector_enabled && nb_sectors > 0) {
917*83f64091Sbellard         memcpy(bs->boot_sector_data, buf, 512);
9187674e7bfSbellard     }
919*83f64091Sbellard 
920*83f64091Sbellard     acb->cb = cb;
921*83f64091Sbellard     acb->cb_opaque = opaque;
922*83f64091Sbellard     return drv->bdrv_aio_write(acb, sector_num, buf, nb_sectors);
923*83f64091Sbellard }
924*83f64091Sbellard 
925*83f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb)
926*83f64091Sbellard     {
927*83f64091Sbellard     BlockDriverState *bs = acb->bs;
928*83f64091Sbellard     BlockDriver *drv = bs->drv;
929*83f64091Sbellard 
930*83f64091Sbellard     drv->bdrv_aio_cancel(acb);
931*83f64091Sbellard     }
932*83f64091Sbellard 
933*83f64091Sbellard void bdrv_aio_delete(BlockDriverAIOCB *acb)
934*83f64091Sbellard {
935*83f64091Sbellard     BlockDriverState *bs = acb->bs;
936*83f64091Sbellard     BlockDriver *drv = bs->drv;
937*83f64091Sbellard 
938*83f64091Sbellard     drv->bdrv_aio_delete(acb);
939*83f64091Sbellard     qemu_free(acb);
940*83f64091Sbellard }
941*83f64091Sbellard 
942*83f64091Sbellard /**************************************************************/
943*83f64091Sbellard /* async block device emulation */
944*83f64091Sbellard 
945*83f64091Sbellard #ifdef QEMU_TOOL
946*83f64091Sbellard static int bdrv_aio_new_em(BlockDriverAIOCB *acb)
947*83f64091Sbellard {
948ea2384d3Sbellard     return 0;
949ea2384d3Sbellard }
950ea2384d3Sbellard 
951*83f64091Sbellard static int bdrv_aio_read_em(BlockDriverAIOCB *acb, int64_t sector_num,
952ea2384d3Sbellard                     uint8_t *buf, int nb_sectors)
953ea2384d3Sbellard {
954ea2384d3Sbellard     int ret;
955*83f64091Sbellard     ret = bdrv_read(acb->bs, sector_num, buf, nb_sectors);
956*83f64091Sbellard     acb->cb(acb->cb_opaque, ret);
957ea2384d3Sbellard     return 0;
958ea2384d3Sbellard }
959ea2384d3Sbellard 
960*83f64091Sbellard static int bdrv_aio_write_em(BlockDriverAIOCB *acb, int64_t sector_num,
961ea2384d3Sbellard                      const uint8_t *buf, int nb_sectors)
962ea2384d3Sbellard {
963ea2384d3Sbellard     int ret;
964*83f64091Sbellard     ret = bdrv_write(acb->bs, sector_num, buf, nb_sectors);
965*83f64091Sbellard     acb->cb(acb->cb_opaque, ret);
966ea2384d3Sbellard     return 0;
967ea2384d3Sbellard }
968ea2384d3Sbellard 
969*83f64091Sbellard static void bdrv_aio_cancel_em(BlockDriverAIOCB *acb)
970ea2384d3Sbellard {
971ea2384d3Sbellard }
972ea2384d3Sbellard 
973*83f64091Sbellard static void bdrv_aio_delete_em(BlockDriverAIOCB *acb)
974beac80cdSbellard {
975beac80cdSbellard }
976beac80cdSbellard #else
977*83f64091Sbellard typedef struct BlockDriverAIOCBSync {
978*83f64091Sbellard     QEMUBH *bh;
979*83f64091Sbellard     int ret;
980*83f64091Sbellard } BlockDriverAIOCBSync;
981*83f64091Sbellard 
982*83f64091Sbellard static void bdrv_aio_bh_cb(void *opaque)
983beac80cdSbellard {
984*83f64091Sbellard     BlockDriverAIOCB *acb = opaque;
985*83f64091Sbellard     BlockDriverAIOCBSync *acb1 = acb->opaque;
986*83f64091Sbellard     acb->cb(acb->cb_opaque, acb1->ret);
987beac80cdSbellard }
988beac80cdSbellard 
989*83f64091Sbellard static int bdrv_aio_new_em(BlockDriverAIOCB *acb)
990ea2384d3Sbellard {
991*83f64091Sbellard     BlockDriverAIOCBSync *acb1;
992ea2384d3Sbellard 
993*83f64091Sbellard     acb1 = qemu_mallocz(sizeof(BlockDriverAIOCBSync));
994*83f64091Sbellard     if (!acb1)
995*83f64091Sbellard         return -1;
996*83f64091Sbellard     acb->opaque = acb1;
997*83f64091Sbellard     acb1->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
998ea2384d3Sbellard     return 0;
999ea2384d3Sbellard }
1000ea2384d3Sbellard 
1001*83f64091Sbellard static int bdrv_aio_read_em(BlockDriverAIOCB *acb, int64_t sector_num,
1002*83f64091Sbellard                     uint8_t *buf, int nb_sectors)
10037a6cba61Spbrook {
1004*83f64091Sbellard     BlockDriverAIOCBSync *acb1 = acb->opaque;
1005*83f64091Sbellard     int ret;
1006*83f64091Sbellard 
1007*83f64091Sbellard     ret = bdrv_read(acb->bs, sector_num, buf, nb_sectors);
1008*83f64091Sbellard     acb1->ret = ret;
1009*83f64091Sbellard     qemu_bh_schedule(acb1->bh);
1010*83f64091Sbellard     return 0;
10117a6cba61Spbrook }
10127a6cba61Spbrook 
1013*83f64091Sbellard static int bdrv_aio_write_em(BlockDriverAIOCB *acb, int64_t sector_num,
1014*83f64091Sbellard                      const uint8_t *buf, int nb_sectors)
1015*83f64091Sbellard {
1016*83f64091Sbellard     BlockDriverAIOCBSync *acb1 = acb->opaque;
1017*83f64091Sbellard     int ret;
1018*83f64091Sbellard 
1019*83f64091Sbellard     ret = bdrv_write(acb->bs, sector_num, buf, nb_sectors);
1020*83f64091Sbellard     acb1->ret = ret;
1021*83f64091Sbellard     qemu_bh_schedule(acb1->bh);
1022*83f64091Sbellard     return 0;
1023*83f64091Sbellard }
1024*83f64091Sbellard 
1025*83f64091Sbellard static void bdrv_aio_cancel_em(BlockDriverAIOCB *acb)
1026*83f64091Sbellard {
1027*83f64091Sbellard     BlockDriverAIOCBSync *acb1 = acb->opaque;
1028*83f64091Sbellard     qemu_bh_cancel(acb1->bh);
1029*83f64091Sbellard }
1030*83f64091Sbellard 
1031*83f64091Sbellard static void bdrv_aio_delete_em(BlockDriverAIOCB *acb)
1032*83f64091Sbellard {
1033*83f64091Sbellard     BlockDriverAIOCBSync *acb1 = acb->opaque;
1034*83f64091Sbellard     qemu_bh_delete(acb1->bh);
1035*83f64091Sbellard }
1036*83f64091Sbellard #endif /* !QEMU_TOOL */
1037*83f64091Sbellard 
1038*83f64091Sbellard /**************************************************************/
1039*83f64091Sbellard /* sync block device emulation */
1040*83f64091Sbellard 
1041*83f64091Sbellard static void bdrv_rw_em_cb(void *opaque, int ret)
1042*83f64091Sbellard {
1043*83f64091Sbellard     *(int *)opaque = ret;
1044*83f64091Sbellard }
1045*83f64091Sbellard 
1046*83f64091Sbellard #define NOT_DONE 0x7fffffff
1047*83f64091Sbellard 
1048*83f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
1049*83f64091Sbellard                         uint8_t *buf, int nb_sectors)
1050*83f64091Sbellard {
1051*83f64091Sbellard     int async_ret, ret;
1052*83f64091Sbellard 
1053*83f64091Sbellard     if (!bs->sync_aiocb) {
1054*83f64091Sbellard         bs->sync_aiocb = bdrv_aio_new(bs);
1055*83f64091Sbellard         if (!bs->sync_aiocb)
1056*83f64091Sbellard             return -1;
1057*83f64091Sbellard     }
1058*83f64091Sbellard     async_ret = NOT_DONE;
1059*83f64091Sbellard     qemu_aio_wait_start();
1060*83f64091Sbellard     ret = bdrv_aio_read(bs->sync_aiocb, sector_num, buf, nb_sectors,
1061*83f64091Sbellard                         bdrv_rw_em_cb, &async_ret);
1062*83f64091Sbellard     if (ret < 0) {
1063*83f64091Sbellard         qemu_aio_wait_end();
1064*83f64091Sbellard         return ret;
1065*83f64091Sbellard     }
1066*83f64091Sbellard     while (async_ret == NOT_DONE) {
1067*83f64091Sbellard         qemu_aio_wait();
1068*83f64091Sbellard     }
1069*83f64091Sbellard     qemu_aio_wait_end();
1070*83f64091Sbellard     return async_ret;
1071*83f64091Sbellard }
1072*83f64091Sbellard 
1073*83f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
1074*83f64091Sbellard                          const uint8_t *buf, int nb_sectors)
1075*83f64091Sbellard {
1076*83f64091Sbellard     int async_ret, ret;
1077*83f64091Sbellard 
1078*83f64091Sbellard     if (!bs->sync_aiocb) {
1079*83f64091Sbellard         bs->sync_aiocb = bdrv_aio_new(bs);
1080*83f64091Sbellard         if (!bs->sync_aiocb)
1081*83f64091Sbellard             return -1;
1082*83f64091Sbellard     }
1083*83f64091Sbellard     async_ret = NOT_DONE;
1084*83f64091Sbellard     qemu_aio_wait_start();
1085*83f64091Sbellard     ret = bdrv_aio_write(bs->sync_aiocb, sector_num, buf, nb_sectors,
1086*83f64091Sbellard                          bdrv_rw_em_cb, &async_ret);
1087*83f64091Sbellard     if (ret < 0) {
1088*83f64091Sbellard         qemu_aio_wait_end();
1089*83f64091Sbellard         return ret;
1090*83f64091Sbellard     }
1091*83f64091Sbellard     while (async_ret == NOT_DONE) {
1092*83f64091Sbellard         qemu_aio_wait();
1093*83f64091Sbellard     }
1094*83f64091Sbellard     qemu_aio_wait_end();
1095*83f64091Sbellard     return async_ret;
1096*83f64091Sbellard }
1097ea2384d3Sbellard 
1098ea2384d3Sbellard void bdrv_init(void)
1099ea2384d3Sbellard {
1100ea2384d3Sbellard     bdrv_register(&bdrv_raw);
1101ea2384d3Sbellard #ifndef _WIN32
1102ea2384d3Sbellard     bdrv_register(&bdrv_cow);
1103ea2384d3Sbellard #endif
1104ea2384d3Sbellard     bdrv_register(&bdrv_qcow);
1105ea2384d3Sbellard     bdrv_register(&bdrv_vmdk);
11063c56521bSbellard     bdrv_register(&bdrv_cloop);
1107585d0ed9Sbellard     bdrv_register(&bdrv_dmg);
1108a8753c34Sbellard     bdrv_register(&bdrv_bochs);
11096a0f9e82Sbellard     bdrv_register(&bdrv_vpc);
1110712e7874Sbellard     bdrv_register(&bdrv_vvfat);
1111ea2384d3Sbellard }
1112