xref: /openbmc/qemu/block.c (revision 71d0770c4cec9f1dc04f4dadcbf7fd6c335030a9)
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  */
243990d09aSblueswir1 #include "config-host.h"
253990d09aSblueswir1 #ifdef _BSD
263990d09aSblueswir1 /* include native header before sys-queue.h */
273990d09aSblueswir1 #include <sys/queue.h>
283990d09aSblueswir1 #endif
293990d09aSblueswir1 
30faf07963Spbrook #include "qemu-common.h"
3187ecb68bSpbrook #include "console.h"
32ea2384d3Sbellard #include "block_int.h"
33fc01f7e7Sbellard 
347674e7bfSbellard #ifdef _BSD
357674e7bfSbellard #include <sys/types.h>
367674e7bfSbellard #include <sys/stat.h>
377674e7bfSbellard #include <sys/ioctl.h>
387674e7bfSbellard #include <sys/disk.h>
397674e7bfSbellard #endif
407674e7bfSbellard 
4183f64091Sbellard #define SECTOR_BITS 9
4283f64091Sbellard #define SECTOR_SIZE (1 << SECTOR_BITS)
433b0d4f61Sbellard 
4490765429Sbellard typedef struct BlockDriverAIOCBSync {
4590765429Sbellard     BlockDriverAIOCB common;
4690765429Sbellard     QEMUBH *bh;
4790765429Sbellard     int ret;
4890765429Sbellard } BlockDriverAIOCBSync;
4990765429Sbellard 
50ce1a14dcSpbrook static BlockDriverAIOCB *bdrv_aio_read_em(BlockDriverState *bs,
51ce1a14dcSpbrook         int64_t sector_num, uint8_t *buf, int nb_sectors,
52ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque);
53ce1a14dcSpbrook static BlockDriverAIOCB *bdrv_aio_write_em(BlockDriverState *bs,
54ce1a14dcSpbrook         int64_t sector_num, const uint8_t *buf, int nb_sectors,
55ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque);
5683f64091Sbellard static void bdrv_aio_cancel_em(BlockDriverAIOCB *acb);
5783f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
5883f64091Sbellard                         uint8_t *buf, int nb_sectors);
5983f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
6083f64091Sbellard                          const uint8_t *buf, int nb_sectors);
61ec530c81Sbellard 
627ee930d0Sblueswir1 BlockDriverState *bdrv_first;
637ee930d0Sblueswir1 
64ea2384d3Sbellard static BlockDriver *first_drv;
65ea2384d3Sbellard 
6683f64091Sbellard int path_is_absolute(const char *path)
6783f64091Sbellard {
6883f64091Sbellard     const char *p;
6921664424Sbellard #ifdef _WIN32
7021664424Sbellard     /* specific case for names like: "\\.\d:" */
7121664424Sbellard     if (*path == '/' || *path == '\\')
7221664424Sbellard         return 1;
7321664424Sbellard #endif
7483f64091Sbellard     p = strchr(path, ':');
7583f64091Sbellard     if (p)
7683f64091Sbellard         p++;
7783f64091Sbellard     else
7883f64091Sbellard         p = path;
793b9f94e1Sbellard #ifdef _WIN32
803b9f94e1Sbellard     return (*p == '/' || *p == '\\');
813b9f94e1Sbellard #else
823b9f94e1Sbellard     return (*p == '/');
833b9f94e1Sbellard #endif
8483f64091Sbellard }
8583f64091Sbellard 
8683f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
8783f64091Sbellard    path to it by considering it is relative to base_path. URL are
8883f64091Sbellard    supported. */
8983f64091Sbellard void path_combine(char *dest, int dest_size,
9083f64091Sbellard                   const char *base_path,
9183f64091Sbellard                   const char *filename)
9283f64091Sbellard {
9383f64091Sbellard     const char *p, *p1;
9483f64091Sbellard     int len;
9583f64091Sbellard 
9683f64091Sbellard     if (dest_size <= 0)
9783f64091Sbellard         return;
9883f64091Sbellard     if (path_is_absolute(filename)) {
9983f64091Sbellard         pstrcpy(dest, dest_size, filename);
10083f64091Sbellard     } else {
10183f64091Sbellard         p = strchr(base_path, ':');
10283f64091Sbellard         if (p)
10383f64091Sbellard             p++;
10483f64091Sbellard         else
10583f64091Sbellard             p = base_path;
1063b9f94e1Sbellard         p1 = strrchr(base_path, '/');
1073b9f94e1Sbellard #ifdef _WIN32
1083b9f94e1Sbellard         {
1093b9f94e1Sbellard             const char *p2;
1103b9f94e1Sbellard             p2 = strrchr(base_path, '\\');
1113b9f94e1Sbellard             if (!p1 || p2 > p1)
1123b9f94e1Sbellard                 p1 = p2;
1133b9f94e1Sbellard         }
1143b9f94e1Sbellard #endif
11583f64091Sbellard         if (p1)
11683f64091Sbellard             p1++;
11783f64091Sbellard         else
11883f64091Sbellard             p1 = base_path;
11983f64091Sbellard         if (p1 > p)
12083f64091Sbellard             p = p1;
12183f64091Sbellard         len = p - base_path;
12283f64091Sbellard         if (len > dest_size - 1)
12383f64091Sbellard             len = dest_size - 1;
12483f64091Sbellard         memcpy(dest, base_path, len);
12583f64091Sbellard         dest[len] = '\0';
12683f64091Sbellard         pstrcat(dest, dest_size, filename);
12783f64091Sbellard     }
12883f64091Sbellard }
12983f64091Sbellard 
13083f64091Sbellard 
1319596ebb7Spbrook static void bdrv_register(BlockDriver *bdrv)
132ea2384d3Sbellard {
133ce1a14dcSpbrook     if (!bdrv->bdrv_aio_read) {
13483f64091Sbellard         /* add AIO emulation layer */
13583f64091Sbellard         bdrv->bdrv_aio_read = bdrv_aio_read_em;
13683f64091Sbellard         bdrv->bdrv_aio_write = bdrv_aio_write_em;
13783f64091Sbellard         bdrv->bdrv_aio_cancel = bdrv_aio_cancel_em;
13890765429Sbellard         bdrv->aiocb_size = sizeof(BlockDriverAIOCBSync);
13983f64091Sbellard     } else if (!bdrv->bdrv_read && !bdrv->bdrv_pread) {
14083f64091Sbellard         /* add synchronous IO emulation layer */
14183f64091Sbellard         bdrv->bdrv_read = bdrv_read_em;
14283f64091Sbellard         bdrv->bdrv_write = bdrv_write_em;
14383f64091Sbellard     }
144ea2384d3Sbellard     bdrv->next = first_drv;
145ea2384d3Sbellard     first_drv = bdrv;
146ea2384d3Sbellard }
147b338082bSbellard 
148b338082bSbellard /* create a new block device (by default it is empty) */
149b338082bSbellard BlockDriverState *bdrv_new(const char *device_name)
150fc01f7e7Sbellard {
151b338082bSbellard     BlockDriverState **pbs, *bs;
152b338082bSbellard 
153b338082bSbellard     bs = qemu_mallocz(sizeof(BlockDriverState));
154b338082bSbellard     pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
155ea2384d3Sbellard     if (device_name[0] != '\0') {
156b338082bSbellard         /* insert at the end */
157b338082bSbellard         pbs = &bdrv_first;
158b338082bSbellard         while (*pbs != NULL)
159b338082bSbellard             pbs = &(*pbs)->next;
160b338082bSbellard         *pbs = bs;
161ea2384d3Sbellard     }
162b338082bSbellard     return bs;
163b338082bSbellard }
164b338082bSbellard 
165ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
166ea2384d3Sbellard {
167ea2384d3Sbellard     BlockDriver *drv1;
168ea2384d3Sbellard     for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
169ea2384d3Sbellard         if (!strcmp(drv1->format_name, format_name))
170ea2384d3Sbellard             return drv1;
171ea2384d3Sbellard     }
172ea2384d3Sbellard     return NULL;
173ea2384d3Sbellard }
174ea2384d3Sbellard 
175ea2384d3Sbellard int bdrv_create(BlockDriver *drv,
176ea2384d3Sbellard                 const char *filename, int64_t size_in_sectors,
177ea2384d3Sbellard                 const char *backing_file, int flags)
178ea2384d3Sbellard {
179ea2384d3Sbellard     if (!drv->bdrv_create)
180ea2384d3Sbellard         return -ENOTSUP;
181ea2384d3Sbellard     return drv->bdrv_create(filename, size_in_sectors, backing_file, flags);
182ea2384d3Sbellard }
183ea2384d3Sbellard 
184d5249393Sbellard #ifdef _WIN32
18595389c86Sbellard void get_tmp_filename(char *filename, int size)
186d5249393Sbellard {
1873b9f94e1Sbellard     char temp_dir[MAX_PATH];
1883b9f94e1Sbellard 
1893b9f94e1Sbellard     GetTempPath(MAX_PATH, temp_dir);
1903b9f94e1Sbellard     GetTempFileName(temp_dir, "qem", 0, filename);
191d5249393Sbellard }
192d5249393Sbellard #else
19395389c86Sbellard void get_tmp_filename(char *filename, int size)
194ea2384d3Sbellard {
195ea2384d3Sbellard     int fd;
1967ccfb2ebSblueswir1     const char *tmpdir;
197d5249393Sbellard     /* XXX: race condition possible */
1980badc1eeSaurel32     tmpdir = getenv("TMPDIR");
1990badc1eeSaurel32     if (!tmpdir)
2000badc1eeSaurel32         tmpdir = "/tmp";
2010badc1eeSaurel32     snprintf(filename, size, "%s/vl.XXXXXX", tmpdir);
202ea2384d3Sbellard     fd = mkstemp(filename);
203ea2384d3Sbellard     close(fd);
204ea2384d3Sbellard }
205d5249393Sbellard #endif
206ea2384d3Sbellard 
20719cb3738Sbellard #ifdef _WIN32
208f45512feSbellard static int is_windows_drive_prefix(const char *filename)
209f45512feSbellard {
210f45512feSbellard     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
211f45512feSbellard              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
212f45512feSbellard             filename[1] == ':');
213f45512feSbellard }
214f45512feSbellard 
21519cb3738Sbellard static int is_windows_drive(const char *filename)
21619cb3738Sbellard {
217f45512feSbellard     if (is_windows_drive_prefix(filename) &&
218f45512feSbellard         filename[2] == '\0')
21919cb3738Sbellard         return 1;
22019cb3738Sbellard     if (strstart(filename, "\\\\.\\", NULL) ||
22119cb3738Sbellard         strstart(filename, "//./", NULL))
22219cb3738Sbellard         return 1;
22319cb3738Sbellard     return 0;
22419cb3738Sbellard }
22519cb3738Sbellard #endif
22619cb3738Sbellard 
22783f64091Sbellard static BlockDriver *find_protocol(const char *filename)
22883f64091Sbellard {
22983f64091Sbellard     BlockDriver *drv1;
23083f64091Sbellard     char protocol[128];
23183f64091Sbellard     int len;
23283f64091Sbellard     const char *p;
23319cb3738Sbellard 
23419cb3738Sbellard #ifdef _WIN32
235f45512feSbellard     if (is_windows_drive(filename) ||
236f45512feSbellard         is_windows_drive_prefix(filename))
23719cb3738Sbellard         return &bdrv_raw;
23819cb3738Sbellard #endif
23983f64091Sbellard     p = strchr(filename, ':');
24083f64091Sbellard     if (!p)
24183f64091Sbellard         return &bdrv_raw;
24283f64091Sbellard     len = p - filename;
24383f64091Sbellard     if (len > sizeof(protocol) - 1)
24483f64091Sbellard         len = sizeof(protocol) - 1;
24583f64091Sbellard     memcpy(protocol, filename, len);
24683f64091Sbellard     protocol[len] = '\0';
24783f64091Sbellard     for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
24883f64091Sbellard         if (drv1->protocol_name &&
24983f64091Sbellard             !strcmp(drv1->protocol_name, protocol))
25083f64091Sbellard             return drv1;
25183f64091Sbellard     }
25283f64091Sbellard     return NULL;
25383f64091Sbellard }
25483f64091Sbellard 
2557674e7bfSbellard /* XXX: force raw format if block or character device ? It would
2567674e7bfSbellard    simplify the BSD case */
257ea2384d3Sbellard static BlockDriver *find_image_format(const char *filename)
258ea2384d3Sbellard {
25983f64091Sbellard     int ret, score, score_max;
260ea2384d3Sbellard     BlockDriver *drv1, *drv;
26183f64091Sbellard     uint8_t buf[2048];
26283f64091Sbellard     BlockDriverState *bs;
263ea2384d3Sbellard 
26419cb3738Sbellard     /* detect host devices. By convention, /dev/cdrom[N] is always
26519cb3738Sbellard        recognized as a host CDROM */
26619cb3738Sbellard     if (strstart(filename, "/dev/cdrom", NULL))
26719cb3738Sbellard         return &bdrv_host_device;
26819cb3738Sbellard #ifdef _WIN32
26919cb3738Sbellard     if (is_windows_drive(filename))
27019cb3738Sbellard         return &bdrv_host_device;
27119cb3738Sbellard #else
27219cb3738Sbellard     {
27319cb3738Sbellard         struct stat st;
27419cb3738Sbellard         if (stat(filename, &st) >= 0 &&
27519cb3738Sbellard             (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))) {
27619cb3738Sbellard             return &bdrv_host_device;
27719cb3738Sbellard         }
27819cb3738Sbellard     }
27919cb3738Sbellard #endif
28019cb3738Sbellard 
28183f64091Sbellard     drv = find_protocol(filename);
28219cb3738Sbellard     /* no need to test disk image formats for vvfat */
28383f64091Sbellard     if (drv == &bdrv_vvfat)
28483f64091Sbellard         return drv;
28583f64091Sbellard 
28683f64091Sbellard     ret = bdrv_file_open(&bs, filename, BDRV_O_RDONLY);
28783f64091Sbellard     if (ret < 0)
2887674e7bfSbellard         return NULL;
28983f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
29083f64091Sbellard     bdrv_delete(bs);
291ea2384d3Sbellard     if (ret < 0) {
292ea2384d3Sbellard         return NULL;
293ea2384d3Sbellard     }
294ea2384d3Sbellard 
295ea2384d3Sbellard     score_max = 0;
296ea2384d3Sbellard     for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) {
29783f64091Sbellard         if (drv1->bdrv_probe) {
298ea2384d3Sbellard             score = drv1->bdrv_probe(buf, ret, filename);
299ea2384d3Sbellard             if (score > score_max) {
300ea2384d3Sbellard                 score_max = score;
301ea2384d3Sbellard                 drv = drv1;
302ea2384d3Sbellard             }
303ea2384d3Sbellard         }
30483f64091Sbellard     }
305ea2384d3Sbellard     return drv;
306ea2384d3Sbellard }
307ea2384d3Sbellard 
30883f64091Sbellard int bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags)
309b338082bSbellard {
31083f64091Sbellard     BlockDriverState *bs;
31183f64091Sbellard     int ret;
3123b0d4f61Sbellard 
31383f64091Sbellard     bs = bdrv_new("");
31483f64091Sbellard     if (!bs)
31583f64091Sbellard         return -ENOMEM;
31683f64091Sbellard     ret = bdrv_open2(bs, filename, flags | BDRV_O_FILE, NULL);
31783f64091Sbellard     if (ret < 0) {
31883f64091Sbellard         bdrv_delete(bs);
31983f64091Sbellard         return ret;
3203b0d4f61Sbellard     }
321*71d0770cSaliguori     bs->growable = 1;
32283f64091Sbellard     *pbs = bs;
32383f64091Sbellard     return 0;
3243b0d4f61Sbellard }
3253b0d4f61Sbellard 
32683f64091Sbellard int bdrv_open(BlockDriverState *bs, const char *filename, int flags)
32783f64091Sbellard {
32883f64091Sbellard     return bdrv_open2(bs, filename, flags, NULL);
329ea2384d3Sbellard }
330ea2384d3Sbellard 
33183f64091Sbellard int bdrv_open2(BlockDriverState *bs, const char *filename, int flags,
332ea2384d3Sbellard                BlockDriver *drv)
333ea2384d3Sbellard {
33483f64091Sbellard     int ret, open_flags;
335eb5c851fSths     char tmp_filename[PATH_MAX];
336eb5c851fSths     char backing_filename[PATH_MAX];
337fc01f7e7Sbellard 
3380849bf08Sbellard     bs->read_only = 0;
339ea2384d3Sbellard     bs->is_temporary = 0;
340ea2384d3Sbellard     bs->encrypted = 0;
34133e3963eSbellard 
34283f64091Sbellard     if (flags & BDRV_O_SNAPSHOT) {
343ea2384d3Sbellard         BlockDriverState *bs1;
344ea2384d3Sbellard         int64_t total_size;
3457c96d46eSaliguori         int is_protocol = 0;
34633e3963eSbellard 
347ea2384d3Sbellard         /* if snapshot, we create a temporary backing file and open it
348ea2384d3Sbellard            instead of opening 'filename' directly */
349ea2384d3Sbellard 
350ea2384d3Sbellard         /* if there is a backing file, use it */
351ea2384d3Sbellard         bs1 = bdrv_new("");
352ea2384d3Sbellard         if (!bs1) {
35383f64091Sbellard             return -ENOMEM;
354ea2384d3Sbellard         }
355ea2384d3Sbellard         if (bdrv_open(bs1, filename, 0) < 0) {
356ea2384d3Sbellard             bdrv_delete(bs1);
357ea2384d3Sbellard             return -1;
358ea2384d3Sbellard         }
35983f64091Sbellard         total_size = bdrv_getlength(bs1) >> SECTOR_BITS;
3607c96d46eSaliguori 
3617c96d46eSaliguori         if (bs1->drv && bs1->drv->protocol_name)
3627c96d46eSaliguori             is_protocol = 1;
3637c96d46eSaliguori 
364ea2384d3Sbellard         bdrv_delete(bs1);
365ea2384d3Sbellard 
366ea2384d3Sbellard         get_tmp_filename(tmp_filename, sizeof(tmp_filename));
3677c96d46eSaliguori 
3687c96d46eSaliguori         /* Real path is meaningless for protocols */
3697c96d46eSaliguori         if (is_protocol)
3707c96d46eSaliguori             snprintf(backing_filename, sizeof(backing_filename),
3717c96d46eSaliguori                      "%s", filename);
3727c96d46eSaliguori         else
373a817d936Sbellard             realpath(filename, backing_filename);
3747c96d46eSaliguori 
375d15a771dSbellard         if (bdrv_create(&bdrv_qcow2, tmp_filename,
376a817d936Sbellard                         total_size, backing_filename, 0) < 0) {
377ea2384d3Sbellard             return -1;
378ea2384d3Sbellard         }
379ea2384d3Sbellard         filename = tmp_filename;
380ea2384d3Sbellard         bs->is_temporary = 1;
381ea2384d3Sbellard     }
382ea2384d3Sbellard 
383ea2384d3Sbellard     pstrcpy(bs->filename, sizeof(bs->filename), filename);
38483f64091Sbellard     if (flags & BDRV_O_FILE) {
38583f64091Sbellard         drv = find_protocol(filename);
38683f64091Sbellard         if (!drv)
38783f64091Sbellard             return -ENOENT;
38883f64091Sbellard     } else {
389ea2384d3Sbellard         if (!drv) {
390ea2384d3Sbellard             drv = find_image_format(filename);
391ea2384d3Sbellard             if (!drv)
392ea2384d3Sbellard                 return -1;
393ea2384d3Sbellard         }
39483f64091Sbellard     }
395ea2384d3Sbellard     bs->drv = drv;
396ea2384d3Sbellard     bs->opaque = qemu_mallocz(drv->instance_size);
39783f64091Sbellard     /* Note: for compatibility, we open disk image files as RDWR, and
39883f64091Sbellard        RDONLY as fallback */
39983f64091Sbellard     if (!(flags & BDRV_O_FILE))
4009f7965c7Saliguori         open_flags = BDRV_O_RDWR | (flags & BDRV_O_CACHE_MASK);
40183f64091Sbellard     else
40283f64091Sbellard         open_flags = flags & ~(BDRV_O_FILE | BDRV_O_SNAPSHOT);
40383f64091Sbellard     ret = drv->bdrv_open(bs, filename, open_flags);
404a0a83536Saurel32     if ((ret == -EACCES || ret == -EPERM) && !(flags & BDRV_O_FILE)) {
4059f7965c7Saliguori         ret = drv->bdrv_open(bs, filename, open_flags & ~BDRV_O_RDWR);
40683f64091Sbellard         bs->read_only = 1;
40783f64091Sbellard     }
408ea2384d3Sbellard     if (ret < 0) {
409ea2384d3Sbellard         qemu_free(bs->opaque);
4106b21b973Sbellard         bs->opaque = NULL;
4116b21b973Sbellard         bs->drv = NULL;
41283f64091Sbellard         return ret;
413ea2384d3Sbellard     }
414d15a771dSbellard     if (drv->bdrv_getlength) {
415d15a771dSbellard         bs->total_sectors = bdrv_getlength(bs) >> SECTOR_BITS;
416d15a771dSbellard     }
417ea2384d3Sbellard #ifndef _WIN32
418ea2384d3Sbellard     if (bs->is_temporary) {
419ea2384d3Sbellard         unlink(filename);
42033e3963eSbellard     }
42167b915a5Sbellard #endif
42283f64091Sbellard     if (bs->backing_file[0] != '\0') {
423ea2384d3Sbellard         /* if there is a backing file, use it */
424ea2384d3Sbellard         bs->backing_hd = bdrv_new("");
425ea2384d3Sbellard         if (!bs->backing_hd) {
426ea2384d3Sbellard         fail:
427ea2384d3Sbellard             bdrv_close(bs);
4286b21b973Sbellard             return -ENOMEM;
429ea2384d3Sbellard         }
43083f64091Sbellard         path_combine(backing_filename, sizeof(backing_filename),
43183f64091Sbellard                      filename, bs->backing_file);
4329f7965c7Saliguori         if (bdrv_open(bs->backing_hd, backing_filename, open_flags) < 0)
433ea2384d3Sbellard             goto fail;
434ea2384d3Sbellard     }
43533e3963eSbellard 
436b338082bSbellard     /* call the change callback */
43719cb3738Sbellard     bs->media_changed = 1;
438b338082bSbellard     if (bs->change_cb)
439b338082bSbellard         bs->change_cb(bs->change_opaque);
440b338082bSbellard 
441b338082bSbellard     return 0;
442fc01f7e7Sbellard }
443fc01f7e7Sbellard 
444fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
445fc01f7e7Sbellard {
44619cb3738Sbellard     if (bs->drv) {
447ea2384d3Sbellard         if (bs->backing_hd)
448ea2384d3Sbellard             bdrv_delete(bs->backing_hd);
449ea2384d3Sbellard         bs->drv->bdrv_close(bs);
450ea2384d3Sbellard         qemu_free(bs->opaque);
451ea2384d3Sbellard #ifdef _WIN32
452ea2384d3Sbellard         if (bs->is_temporary) {
453ea2384d3Sbellard             unlink(bs->filename);
454ea2384d3Sbellard         }
45567b915a5Sbellard #endif
456ea2384d3Sbellard         bs->opaque = NULL;
457ea2384d3Sbellard         bs->drv = NULL;
458b338082bSbellard 
459b338082bSbellard         /* call the change callback */
46019cb3738Sbellard         bs->media_changed = 1;
461b338082bSbellard         if (bs->change_cb)
462b338082bSbellard             bs->change_cb(bs->change_opaque);
463b338082bSbellard     }
464b338082bSbellard }
465b338082bSbellard 
466b338082bSbellard void bdrv_delete(BlockDriverState *bs)
467b338082bSbellard {
46834c6f050Saurel32     BlockDriverState **pbs;
46934c6f050Saurel32 
47034c6f050Saurel32     pbs = &bdrv_first;
47134c6f050Saurel32     while (*pbs != bs && *pbs != NULL)
47234c6f050Saurel32         pbs = &(*pbs)->next;
47334c6f050Saurel32     if (*pbs == bs)
47434c6f050Saurel32         *pbs = bs->next;
47534c6f050Saurel32 
476b338082bSbellard     bdrv_close(bs);
477b338082bSbellard     qemu_free(bs);
478fc01f7e7Sbellard }
479fc01f7e7Sbellard 
48033e3963eSbellard /* commit COW file into the raw image */
48133e3963eSbellard int bdrv_commit(BlockDriverState *bs)
48233e3963eSbellard {
48319cb3738Sbellard     BlockDriver *drv = bs->drv;
48483f64091Sbellard     int64_t i, total_sectors;
485ea2384d3Sbellard     int n, j;
486ea2384d3Sbellard     unsigned char sector[512];
48733e3963eSbellard 
48819cb3738Sbellard     if (!drv)
48919cb3738Sbellard         return -ENOMEDIUM;
49033e3963eSbellard 
49133e3963eSbellard     if (bs->read_only) {
492ea2384d3Sbellard 	return -EACCES;
49333e3963eSbellard     }
49433e3963eSbellard 
495ea2384d3Sbellard     if (!bs->backing_hd) {
496ea2384d3Sbellard 	return -ENOTSUP;
497ea2384d3Sbellard     }
498ea2384d3Sbellard 
49983f64091Sbellard     total_sectors = bdrv_getlength(bs) >> SECTOR_BITS;
50083f64091Sbellard     for (i = 0; i < total_sectors;) {
50119cb3738Sbellard         if (drv->bdrv_is_allocated(bs, i, 65536, &n)) {
502ea2384d3Sbellard             for(j = 0; j < n; j++) {
50333e3963eSbellard                 if (bdrv_read(bs, i, sector, 1) != 0) {
504ea2384d3Sbellard                     return -EIO;
50533e3963eSbellard                 }
50633e3963eSbellard 
507ea2384d3Sbellard                 if (bdrv_write(bs->backing_hd, i, sector, 1) != 0) {
508ea2384d3Sbellard                     return -EIO;
50933e3963eSbellard                 }
510ea2384d3Sbellard                 i++;
511ea2384d3Sbellard 	    }
512ea2384d3Sbellard 	} else {
513ea2384d3Sbellard             i += n;
51433e3963eSbellard         }
51533e3963eSbellard     }
51695389c86Sbellard 
51719cb3738Sbellard     if (drv->bdrv_make_empty)
51819cb3738Sbellard 	return drv->bdrv_make_empty(bs);
51995389c86Sbellard 
52033e3963eSbellard     return 0;
52133e3963eSbellard }
52233e3963eSbellard 
523*71d0770cSaliguori static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
524*71d0770cSaliguori                                    size_t size)
525*71d0770cSaliguori {
526*71d0770cSaliguori     int64_t len;
527*71d0770cSaliguori 
528*71d0770cSaliguori     if (!bdrv_is_inserted(bs))
529*71d0770cSaliguori         return -ENOMEDIUM;
530*71d0770cSaliguori 
531*71d0770cSaliguori     if (bs->growable)
532*71d0770cSaliguori         return 0;
533*71d0770cSaliguori 
534*71d0770cSaliguori     len = bdrv_getlength(bs);
535*71d0770cSaliguori 
536*71d0770cSaliguori     if ((offset + size) > len)
537*71d0770cSaliguori         return -EIO;
538*71d0770cSaliguori 
539*71d0770cSaliguori     return 0;
540*71d0770cSaliguori }
541*71d0770cSaliguori 
542*71d0770cSaliguori static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
543*71d0770cSaliguori                               int nb_sectors)
544*71d0770cSaliguori {
545*71d0770cSaliguori     int64_t offset;
546*71d0770cSaliguori 
547*71d0770cSaliguori     /* Deal with byte accesses */
548*71d0770cSaliguori     if (sector_num < 0)
549*71d0770cSaliguori         offset = -sector_num;
550*71d0770cSaliguori     else
551*71d0770cSaliguori         offset = sector_num * 512;
552*71d0770cSaliguori 
553*71d0770cSaliguori     return bdrv_check_byte_request(bs, offset, nb_sectors * 512);
554*71d0770cSaliguori }
555*71d0770cSaliguori 
55619cb3738Sbellard /* return < 0 if error. See bdrv_write() for the return codes */
557fc01f7e7Sbellard int bdrv_read(BlockDriverState *bs, int64_t sector_num,
558fc01f7e7Sbellard               uint8_t *buf, int nb_sectors)
559fc01f7e7Sbellard {
560ea2384d3Sbellard     BlockDriver *drv = bs->drv;
561fc01f7e7Sbellard 
56219cb3738Sbellard     if (!drv)
56319cb3738Sbellard         return -ENOMEDIUM;
564*71d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
565*71d0770cSaliguori         return -EIO;
566b338082bSbellard 
56783f64091Sbellard     if (drv->bdrv_pread) {
56883f64091Sbellard         int ret, len;
56983f64091Sbellard         len = nb_sectors * 512;
57083f64091Sbellard         ret = drv->bdrv_pread(bs, sector_num * 512, buf, len);
57183f64091Sbellard         if (ret < 0)
57283f64091Sbellard             return ret;
57383f64091Sbellard         else if (ret != len)
57419cb3738Sbellard             return -EINVAL;
575a36e69ddSths         else {
576a36e69ddSths 	    bs->rd_bytes += (unsigned) len;
577a36e69ddSths 	    bs->rd_ops ++;
57883f64091Sbellard             return 0;
579a36e69ddSths 	}
58083f64091Sbellard     } else {
58183f64091Sbellard         return drv->bdrv_read(bs, sector_num, buf, nb_sectors);
58283f64091Sbellard     }
58383f64091Sbellard }
584fc01f7e7Sbellard 
58519cb3738Sbellard /* Return < 0 if error. Important errors are:
58619cb3738Sbellard   -EIO         generic I/O error (may happen for all errors)
58719cb3738Sbellard   -ENOMEDIUM   No media inserted.
58819cb3738Sbellard   -EINVAL      Invalid sector number or nb_sectors
58919cb3738Sbellard   -EACCES      Trying to write a read-only device
59019cb3738Sbellard */
591fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num,
592fc01f7e7Sbellard                const uint8_t *buf, int nb_sectors)
593fc01f7e7Sbellard {
59483f64091Sbellard     BlockDriver *drv = bs->drv;
59519cb3738Sbellard     if (!bs->drv)
59619cb3738Sbellard         return -ENOMEDIUM;
5970849bf08Sbellard     if (bs->read_only)
59819cb3738Sbellard         return -EACCES;
599*71d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
600*71d0770cSaliguori         return -EIO;
601*71d0770cSaliguori 
60283f64091Sbellard     if (drv->bdrv_pwrite) {
60342fb2807Saliguori         int ret, len, count = 0;
60483f64091Sbellard         len = nb_sectors * 512;
60542fb2807Saliguori         do {
60642fb2807Saliguori             ret = drv->bdrv_pwrite(bs, sector_num * 512, buf, len - count);
60742fb2807Saliguori             if (ret < 0) {
60842fb2807Saliguori                 printf("bdrv_write ret=%d\n", ret);
60983f64091Sbellard                 return ret;
61042fb2807Saliguori             }
61142fb2807Saliguori             count += ret;
61242fb2807Saliguori             buf += ret;
61342fb2807Saliguori         } while (count != len);
614a36e69ddSths         bs->wr_bytes += (unsigned) len;
615a36e69ddSths         bs->wr_ops ++;
61683f64091Sbellard         return 0;
617a36e69ddSths     }
61883f64091Sbellard     return drv->bdrv_write(bs, sector_num, buf, nb_sectors);
61983f64091Sbellard }
62083f64091Sbellard 
62183f64091Sbellard static int bdrv_pread_em(BlockDriverState *bs, int64_t offset,
622faea38e7Sbellard                          uint8_t *buf, int count1)
62383f64091Sbellard {
62483f64091Sbellard     uint8_t tmp_buf[SECTOR_SIZE];
62583f64091Sbellard     int len, nb_sectors, count;
62683f64091Sbellard     int64_t sector_num;
62783f64091Sbellard 
62883f64091Sbellard     count = count1;
62983f64091Sbellard     /* first read to align to sector start */
63083f64091Sbellard     len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1);
63183f64091Sbellard     if (len > count)
63283f64091Sbellard         len = count;
63383f64091Sbellard     sector_num = offset >> SECTOR_BITS;
63483f64091Sbellard     if (len > 0) {
63583f64091Sbellard         if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
63683f64091Sbellard             return -EIO;
63783f64091Sbellard         memcpy(buf, tmp_buf + (offset & (SECTOR_SIZE - 1)), len);
63883f64091Sbellard         count -= len;
63983f64091Sbellard         if (count == 0)
64083f64091Sbellard             return count1;
64183f64091Sbellard         sector_num++;
64283f64091Sbellard         buf += len;
64383f64091Sbellard     }
64483f64091Sbellard 
64583f64091Sbellard     /* read the sectors "in place" */
64683f64091Sbellard     nb_sectors = count >> SECTOR_BITS;
64783f64091Sbellard     if (nb_sectors > 0) {
64883f64091Sbellard         if (bdrv_read(bs, sector_num, buf, nb_sectors) < 0)
64983f64091Sbellard             return -EIO;
65083f64091Sbellard         sector_num += nb_sectors;
65183f64091Sbellard         len = nb_sectors << SECTOR_BITS;
65283f64091Sbellard         buf += len;
65383f64091Sbellard         count -= len;
65483f64091Sbellard     }
65583f64091Sbellard 
65683f64091Sbellard     /* add data from the last sector */
65783f64091Sbellard     if (count > 0) {
65883f64091Sbellard         if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
65983f64091Sbellard             return -EIO;
66083f64091Sbellard         memcpy(buf, tmp_buf, count);
66183f64091Sbellard     }
66283f64091Sbellard     return count1;
66383f64091Sbellard }
66483f64091Sbellard 
66583f64091Sbellard static int bdrv_pwrite_em(BlockDriverState *bs, int64_t offset,
666faea38e7Sbellard                           const uint8_t *buf, int count1)
66783f64091Sbellard {
66883f64091Sbellard     uint8_t tmp_buf[SECTOR_SIZE];
66983f64091Sbellard     int len, nb_sectors, count;
67083f64091Sbellard     int64_t sector_num;
67183f64091Sbellard 
67283f64091Sbellard     count = count1;
67383f64091Sbellard     /* first write to align to sector start */
67483f64091Sbellard     len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1);
67583f64091Sbellard     if (len > count)
67683f64091Sbellard         len = count;
67783f64091Sbellard     sector_num = offset >> SECTOR_BITS;
67883f64091Sbellard     if (len > 0) {
67983f64091Sbellard         if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
68083f64091Sbellard             return -EIO;
68183f64091Sbellard         memcpy(tmp_buf + (offset & (SECTOR_SIZE - 1)), buf, len);
68283f64091Sbellard         if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0)
68383f64091Sbellard             return -EIO;
68483f64091Sbellard         count -= len;
68583f64091Sbellard         if (count == 0)
68683f64091Sbellard             return count1;
68783f64091Sbellard         sector_num++;
68883f64091Sbellard         buf += len;
68983f64091Sbellard     }
69083f64091Sbellard 
69183f64091Sbellard     /* write the sectors "in place" */
69283f64091Sbellard     nb_sectors = count >> SECTOR_BITS;
69383f64091Sbellard     if (nb_sectors > 0) {
69483f64091Sbellard         if (bdrv_write(bs, sector_num, buf, nb_sectors) < 0)
69583f64091Sbellard             return -EIO;
69683f64091Sbellard         sector_num += nb_sectors;
69783f64091Sbellard         len = nb_sectors << SECTOR_BITS;
69883f64091Sbellard         buf += len;
69983f64091Sbellard         count -= len;
70083f64091Sbellard     }
70183f64091Sbellard 
70283f64091Sbellard     /* add data from the last sector */
70383f64091Sbellard     if (count > 0) {
70483f64091Sbellard         if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0)
70583f64091Sbellard             return -EIO;
70683f64091Sbellard         memcpy(tmp_buf, buf, count);
70783f64091Sbellard         if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0)
70883f64091Sbellard             return -EIO;
70983f64091Sbellard     }
71083f64091Sbellard     return count1;
71183f64091Sbellard }
71283f64091Sbellard 
71383f64091Sbellard /**
71483f64091Sbellard  * Read with byte offsets (needed only for file protocols)
71583f64091Sbellard  */
71683f64091Sbellard int bdrv_pread(BlockDriverState *bs, int64_t offset,
71783f64091Sbellard                void *buf1, int count1)
71883f64091Sbellard {
71983f64091Sbellard     BlockDriver *drv = bs->drv;
72083f64091Sbellard 
72183f64091Sbellard     if (!drv)
72219cb3738Sbellard         return -ENOMEDIUM;
723*71d0770cSaliguori     if (bdrv_check_byte_request(bs, offset, count1))
724*71d0770cSaliguori         return -EIO;
725*71d0770cSaliguori 
72683f64091Sbellard     if (!drv->bdrv_pread)
727faea38e7Sbellard         return bdrv_pread_em(bs, offset, buf1, count1);
72883f64091Sbellard     return drv->bdrv_pread(bs, offset, buf1, count1);
72983f64091Sbellard }
73083f64091Sbellard 
73183f64091Sbellard /**
73283f64091Sbellard  * Write with byte offsets (needed only for file protocols)
73383f64091Sbellard  */
73483f64091Sbellard int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
73583f64091Sbellard                 const void *buf1, int count1)
73683f64091Sbellard {
73783f64091Sbellard     BlockDriver *drv = bs->drv;
73883f64091Sbellard 
73983f64091Sbellard     if (!drv)
74019cb3738Sbellard         return -ENOMEDIUM;
741*71d0770cSaliguori     if (bdrv_check_byte_request(bs, offset, count1))
742*71d0770cSaliguori         return -EIO;
743*71d0770cSaliguori 
74483f64091Sbellard     if (!drv->bdrv_pwrite)
745faea38e7Sbellard         return bdrv_pwrite_em(bs, offset, buf1, count1);
74683f64091Sbellard     return drv->bdrv_pwrite(bs, offset, buf1, count1);
74783f64091Sbellard }
74883f64091Sbellard 
74983f64091Sbellard /**
75083f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
75183f64091Sbellard  */
75283f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
75383f64091Sbellard {
75483f64091Sbellard     BlockDriver *drv = bs->drv;
75583f64091Sbellard     if (!drv)
75619cb3738Sbellard         return -ENOMEDIUM;
75783f64091Sbellard     if (!drv->bdrv_truncate)
75883f64091Sbellard         return -ENOTSUP;
75983f64091Sbellard     return drv->bdrv_truncate(bs, offset);
76083f64091Sbellard }
76183f64091Sbellard 
76283f64091Sbellard /**
76383f64091Sbellard  * Length of a file in bytes. Return < 0 if error or unknown.
76483f64091Sbellard  */
76583f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs)
76683f64091Sbellard {
76783f64091Sbellard     BlockDriver *drv = bs->drv;
76883f64091Sbellard     if (!drv)
76919cb3738Sbellard         return -ENOMEDIUM;
77083f64091Sbellard     if (!drv->bdrv_getlength) {
77183f64091Sbellard         /* legacy mode */
77283f64091Sbellard         return bs->total_sectors * SECTOR_SIZE;
77383f64091Sbellard     }
77483f64091Sbellard     return drv->bdrv_getlength(bs);
775fc01f7e7Sbellard }
776fc01f7e7Sbellard 
77719cb3738Sbellard /* return 0 as number of sectors if no device present or error */
77896b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
779fc01f7e7Sbellard {
78019cb3738Sbellard     int64_t length;
78119cb3738Sbellard     length = bdrv_getlength(bs);
78219cb3738Sbellard     if (length < 0)
78319cb3738Sbellard         length = 0;
78419cb3738Sbellard     else
78519cb3738Sbellard         length = length >> SECTOR_BITS;
78619cb3738Sbellard     *nb_sectors_ptr = length;
787fc01f7e7Sbellard }
788cf98951bSbellard 
789f3d54fc4Saliguori struct partition {
790f3d54fc4Saliguori         uint8_t boot_ind;           /* 0x80 - active */
791f3d54fc4Saliguori         uint8_t head;               /* starting head */
792f3d54fc4Saliguori         uint8_t sector;             /* starting sector */
793f3d54fc4Saliguori         uint8_t cyl;                /* starting cylinder */
794f3d54fc4Saliguori         uint8_t sys_ind;            /* What partition type */
795f3d54fc4Saliguori         uint8_t end_head;           /* end head */
796f3d54fc4Saliguori         uint8_t end_sector;         /* end sector */
797f3d54fc4Saliguori         uint8_t end_cyl;            /* end cylinder */
798f3d54fc4Saliguori         uint32_t start_sect;        /* starting sector counting from 0 */
799f3d54fc4Saliguori         uint32_t nr_sects;          /* nr of sectors in partition */
800f3d54fc4Saliguori } __attribute__((packed));
801f3d54fc4Saliguori 
802f3d54fc4Saliguori /* try to guess the disk logical geometry from the MSDOS partition table. Return 0 if OK, -1 if could not guess */
803f3d54fc4Saliguori static int guess_disk_lchs(BlockDriverState *bs,
804f3d54fc4Saliguori                            int *pcylinders, int *pheads, int *psectors)
805f3d54fc4Saliguori {
806f3d54fc4Saliguori     uint8_t buf[512];
807f3d54fc4Saliguori     int ret, i, heads, sectors, cylinders;
808f3d54fc4Saliguori     struct partition *p;
809f3d54fc4Saliguori     uint32_t nr_sects;
810a38131b6Sblueswir1     uint64_t nb_sectors;
811f3d54fc4Saliguori 
812f3d54fc4Saliguori     bdrv_get_geometry(bs, &nb_sectors);
813f3d54fc4Saliguori 
814f3d54fc4Saliguori     ret = bdrv_read(bs, 0, buf, 1);
815f3d54fc4Saliguori     if (ret < 0)
816f3d54fc4Saliguori         return -1;
817f3d54fc4Saliguori     /* test msdos magic */
818f3d54fc4Saliguori     if (buf[510] != 0x55 || buf[511] != 0xaa)
819f3d54fc4Saliguori         return -1;
820f3d54fc4Saliguori     for(i = 0; i < 4; i++) {
821f3d54fc4Saliguori         p = ((struct partition *)(buf + 0x1be)) + i;
822f3d54fc4Saliguori         nr_sects = le32_to_cpu(p->nr_sects);
823f3d54fc4Saliguori         if (nr_sects && p->end_head) {
824f3d54fc4Saliguori             /* We make the assumption that the partition terminates on
825f3d54fc4Saliguori                a cylinder boundary */
826f3d54fc4Saliguori             heads = p->end_head + 1;
827f3d54fc4Saliguori             sectors = p->end_sector & 63;
828f3d54fc4Saliguori             if (sectors == 0)
829f3d54fc4Saliguori                 continue;
830f3d54fc4Saliguori             cylinders = nb_sectors / (heads * sectors);
831f3d54fc4Saliguori             if (cylinders < 1 || cylinders > 16383)
832f3d54fc4Saliguori                 continue;
833f3d54fc4Saliguori             *pheads = heads;
834f3d54fc4Saliguori             *psectors = sectors;
835f3d54fc4Saliguori             *pcylinders = cylinders;
836f3d54fc4Saliguori #if 0
837f3d54fc4Saliguori             printf("guessed geometry: LCHS=%d %d %d\n",
838f3d54fc4Saliguori                    cylinders, heads, sectors);
839f3d54fc4Saliguori #endif
840f3d54fc4Saliguori             return 0;
841f3d54fc4Saliguori         }
842f3d54fc4Saliguori     }
843f3d54fc4Saliguori     return -1;
844f3d54fc4Saliguori }
845f3d54fc4Saliguori 
846f3d54fc4Saliguori void bdrv_guess_geometry(BlockDriverState *bs, int *pcyls, int *pheads, int *psecs)
847f3d54fc4Saliguori {
848f3d54fc4Saliguori     int translation, lba_detected = 0;
849f3d54fc4Saliguori     int cylinders, heads, secs;
850a38131b6Sblueswir1     uint64_t nb_sectors;
851f3d54fc4Saliguori 
852f3d54fc4Saliguori     /* if a geometry hint is available, use it */
853f3d54fc4Saliguori     bdrv_get_geometry(bs, &nb_sectors);
854f3d54fc4Saliguori     bdrv_get_geometry_hint(bs, &cylinders, &heads, &secs);
855f3d54fc4Saliguori     translation = bdrv_get_translation_hint(bs);
856f3d54fc4Saliguori     if (cylinders != 0) {
857f3d54fc4Saliguori         *pcyls = cylinders;
858f3d54fc4Saliguori         *pheads = heads;
859f3d54fc4Saliguori         *psecs = secs;
860f3d54fc4Saliguori     } else {
861f3d54fc4Saliguori         if (guess_disk_lchs(bs, &cylinders, &heads, &secs) == 0) {
862f3d54fc4Saliguori             if (heads > 16) {
863f3d54fc4Saliguori                 /* if heads > 16, it means that a BIOS LBA
864f3d54fc4Saliguori                    translation was active, so the default
865f3d54fc4Saliguori                    hardware geometry is OK */
866f3d54fc4Saliguori                 lba_detected = 1;
867f3d54fc4Saliguori                 goto default_geometry;
868f3d54fc4Saliguori             } else {
869f3d54fc4Saliguori                 *pcyls = cylinders;
870f3d54fc4Saliguori                 *pheads = heads;
871f3d54fc4Saliguori                 *psecs = secs;
872f3d54fc4Saliguori                 /* disable any translation to be in sync with
873f3d54fc4Saliguori                    the logical geometry */
874f3d54fc4Saliguori                 if (translation == BIOS_ATA_TRANSLATION_AUTO) {
875f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
876f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_NONE);
877f3d54fc4Saliguori                 }
878f3d54fc4Saliguori             }
879f3d54fc4Saliguori         } else {
880f3d54fc4Saliguori         default_geometry:
881f3d54fc4Saliguori             /* if no geometry, use a standard physical disk geometry */
882f3d54fc4Saliguori             cylinders = nb_sectors / (16 * 63);
883f3d54fc4Saliguori 
884f3d54fc4Saliguori             if (cylinders > 16383)
885f3d54fc4Saliguori                 cylinders = 16383;
886f3d54fc4Saliguori             else if (cylinders < 2)
887f3d54fc4Saliguori                 cylinders = 2;
888f3d54fc4Saliguori             *pcyls = cylinders;
889f3d54fc4Saliguori             *pheads = 16;
890f3d54fc4Saliguori             *psecs = 63;
891f3d54fc4Saliguori             if ((lba_detected == 1) && (translation == BIOS_ATA_TRANSLATION_AUTO)) {
892f3d54fc4Saliguori                 if ((*pcyls * *pheads) <= 131072) {
893f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
894f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_LARGE);
895f3d54fc4Saliguori                 } else {
896f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
897f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_LBA);
898f3d54fc4Saliguori                 }
899f3d54fc4Saliguori             }
900f3d54fc4Saliguori         }
901f3d54fc4Saliguori         bdrv_set_geometry_hint(bs, *pcyls, *pheads, *psecs);
902f3d54fc4Saliguori     }
903f3d54fc4Saliguori }
904f3d54fc4Saliguori 
905b338082bSbellard void bdrv_set_geometry_hint(BlockDriverState *bs,
906b338082bSbellard                             int cyls, int heads, int secs)
907b338082bSbellard {
908b338082bSbellard     bs->cyls = cyls;
909b338082bSbellard     bs->heads = heads;
910b338082bSbellard     bs->secs = secs;
911b338082bSbellard }
912b338082bSbellard 
913b338082bSbellard void bdrv_set_type_hint(BlockDriverState *bs, int type)
914b338082bSbellard {
915b338082bSbellard     bs->type = type;
916b338082bSbellard     bs->removable = ((type == BDRV_TYPE_CDROM ||
917b338082bSbellard                       type == BDRV_TYPE_FLOPPY));
918b338082bSbellard }
919b338082bSbellard 
92046d4767dSbellard void bdrv_set_translation_hint(BlockDriverState *bs, int translation)
92146d4767dSbellard {
92246d4767dSbellard     bs->translation = translation;
92346d4767dSbellard }
92446d4767dSbellard 
925b338082bSbellard void bdrv_get_geometry_hint(BlockDriverState *bs,
926b338082bSbellard                             int *pcyls, int *pheads, int *psecs)
927b338082bSbellard {
928b338082bSbellard     *pcyls = bs->cyls;
929b338082bSbellard     *pheads = bs->heads;
930b338082bSbellard     *psecs = bs->secs;
931b338082bSbellard }
932b338082bSbellard 
933b338082bSbellard int bdrv_get_type_hint(BlockDriverState *bs)
934b338082bSbellard {
935b338082bSbellard     return bs->type;
936b338082bSbellard }
937b338082bSbellard 
93846d4767dSbellard int bdrv_get_translation_hint(BlockDriverState *bs)
93946d4767dSbellard {
94046d4767dSbellard     return bs->translation;
94146d4767dSbellard }
94246d4767dSbellard 
943b338082bSbellard int bdrv_is_removable(BlockDriverState *bs)
944b338082bSbellard {
945b338082bSbellard     return bs->removable;
946b338082bSbellard }
947b338082bSbellard 
948b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
949b338082bSbellard {
950b338082bSbellard     return bs->read_only;
951b338082bSbellard }
952b338082bSbellard 
953985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
954985a03b0Sths {
955985a03b0Sths     return bs->sg;
956985a03b0Sths }
957985a03b0Sths 
95819cb3738Sbellard /* XXX: no longer used */
959b338082bSbellard void bdrv_set_change_cb(BlockDriverState *bs,
960b338082bSbellard                         void (*change_cb)(void *opaque), void *opaque)
961b338082bSbellard {
962b338082bSbellard     bs->change_cb = change_cb;
963b338082bSbellard     bs->change_opaque = opaque;
964b338082bSbellard }
965b338082bSbellard 
966ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
967ea2384d3Sbellard {
968ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
969ea2384d3Sbellard         return 1;
970ea2384d3Sbellard     return bs->encrypted;
971ea2384d3Sbellard }
972ea2384d3Sbellard 
973ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
974ea2384d3Sbellard {
975ea2384d3Sbellard     int ret;
976ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
977ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
978ea2384d3Sbellard         if (ret < 0)
979ea2384d3Sbellard             return ret;
980ea2384d3Sbellard         if (!bs->encrypted)
981ea2384d3Sbellard             return 0;
982ea2384d3Sbellard     }
983ea2384d3Sbellard     if (!bs->encrypted || !bs->drv || !bs->drv->bdrv_set_key)
984ea2384d3Sbellard         return -1;
985ea2384d3Sbellard     return bs->drv->bdrv_set_key(bs, key);
986ea2384d3Sbellard }
987ea2384d3Sbellard 
988ea2384d3Sbellard void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size)
989ea2384d3Sbellard {
99019cb3738Sbellard     if (!bs->drv) {
991ea2384d3Sbellard         buf[0] = '\0';
992ea2384d3Sbellard     } else {
993ea2384d3Sbellard         pstrcpy(buf, buf_size, bs->drv->format_name);
994ea2384d3Sbellard     }
995ea2384d3Sbellard }
996ea2384d3Sbellard 
997ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
998ea2384d3Sbellard                          void *opaque)
999ea2384d3Sbellard {
1000ea2384d3Sbellard     BlockDriver *drv;
1001ea2384d3Sbellard 
1002ea2384d3Sbellard     for (drv = first_drv; drv != NULL; drv = drv->next) {
1003ea2384d3Sbellard         it(opaque, drv->format_name);
1004ea2384d3Sbellard     }
1005ea2384d3Sbellard }
1006ea2384d3Sbellard 
1007b338082bSbellard BlockDriverState *bdrv_find(const char *name)
1008b338082bSbellard {
1009b338082bSbellard     BlockDriverState *bs;
1010b338082bSbellard 
1011b338082bSbellard     for (bs = bdrv_first; bs != NULL; bs = bs->next) {
1012b338082bSbellard         if (!strcmp(name, bs->device_name))
1013b338082bSbellard             return bs;
1014b338082bSbellard     }
1015b338082bSbellard     return NULL;
1016b338082bSbellard }
1017b338082bSbellard 
101881d0912dSbellard void bdrv_iterate(void (*it)(void *opaque, const char *name), void *opaque)
101981d0912dSbellard {
102081d0912dSbellard     BlockDriverState *bs;
102181d0912dSbellard 
102281d0912dSbellard     for (bs = bdrv_first; bs != NULL; bs = bs->next) {
102381d0912dSbellard         it(opaque, bs->device_name);
102481d0912dSbellard     }
102581d0912dSbellard }
102681d0912dSbellard 
1027ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs)
1028ea2384d3Sbellard {
1029ea2384d3Sbellard     return bs->device_name;
1030ea2384d3Sbellard }
1031ea2384d3Sbellard 
10327a6cba61Spbrook void bdrv_flush(BlockDriverState *bs)
10337a6cba61Spbrook {
10347a6cba61Spbrook     if (bs->drv->bdrv_flush)
10357a6cba61Spbrook         bs->drv->bdrv_flush(bs);
10367a6cba61Spbrook     if (bs->backing_hd)
10377a6cba61Spbrook         bdrv_flush(bs->backing_hd);
10387a6cba61Spbrook }
10397a6cba61Spbrook 
1040c6ca28d6Saliguori void bdrv_flush_all(void)
1041c6ca28d6Saliguori {
1042c6ca28d6Saliguori     BlockDriverState *bs;
1043c6ca28d6Saliguori 
1044c6ca28d6Saliguori     for (bs = bdrv_first; bs != NULL; bs = bs->next)
1045c6ca28d6Saliguori         if (bs->drv && !bdrv_is_read_only(bs) &&
1046c6ca28d6Saliguori             (!bdrv_is_removable(bs) || bdrv_is_inserted(bs)))
1047c6ca28d6Saliguori             bdrv_flush(bs);
1048c6ca28d6Saliguori }
1049c6ca28d6Saliguori 
1050f58c7b35Sths /*
1051f58c7b35Sths  * Returns true iff the specified sector is present in the disk image. Drivers
1052f58c7b35Sths  * not implementing the functionality are assumed to not support backing files,
1053f58c7b35Sths  * hence all their sectors are reported as allocated.
1054f58c7b35Sths  *
1055f58c7b35Sths  * 'pnum' is set to the number of sectors (including and immediately following
1056f58c7b35Sths  * the specified sector) that are known to be in the same
1057f58c7b35Sths  * allocated/unallocated state.
1058f58c7b35Sths  *
1059f58c7b35Sths  * 'nb_sectors' is the max value 'pnum' should be set to.
1060f58c7b35Sths  */
1061f58c7b35Sths int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
1062f58c7b35Sths 	int *pnum)
1063f58c7b35Sths {
1064f58c7b35Sths     int64_t n;
1065f58c7b35Sths     if (!bs->drv->bdrv_is_allocated) {
1066f58c7b35Sths         if (sector_num >= bs->total_sectors) {
1067f58c7b35Sths             *pnum = 0;
1068f58c7b35Sths             return 0;
1069f58c7b35Sths         }
1070f58c7b35Sths         n = bs->total_sectors - sector_num;
1071f58c7b35Sths         *pnum = (n < nb_sectors) ? (n) : (nb_sectors);
1072f58c7b35Sths         return 1;
1073f58c7b35Sths     }
1074f58c7b35Sths     return bs->drv->bdrv_is_allocated(bs, sector_num, nb_sectors, pnum);
1075f58c7b35Sths }
1076f58c7b35Sths 
1077b338082bSbellard void bdrv_info(void)
1078b338082bSbellard {
1079b338082bSbellard     BlockDriverState *bs;
1080b338082bSbellard 
1081b338082bSbellard     for (bs = bdrv_first; bs != NULL; bs = bs->next) {
1082b338082bSbellard         term_printf("%s:", bs->device_name);
1083b338082bSbellard         term_printf(" type=");
1084b338082bSbellard         switch(bs->type) {
1085b338082bSbellard         case BDRV_TYPE_HD:
1086b338082bSbellard             term_printf("hd");
1087b338082bSbellard             break;
1088b338082bSbellard         case BDRV_TYPE_CDROM:
1089b338082bSbellard             term_printf("cdrom");
1090b338082bSbellard             break;
1091b338082bSbellard         case BDRV_TYPE_FLOPPY:
1092b338082bSbellard             term_printf("floppy");
1093b338082bSbellard             break;
1094b338082bSbellard         }
1095b338082bSbellard         term_printf(" removable=%d", bs->removable);
1096b338082bSbellard         if (bs->removable) {
1097b338082bSbellard             term_printf(" locked=%d", bs->locked);
1098b338082bSbellard         }
109919cb3738Sbellard         if (bs->drv) {
1100fef30743Sths             term_printf(" file=");
1101fef30743Sths 	    term_print_filename(bs->filename);
1102fef30743Sths             if (bs->backing_file[0] != '\0') {
1103fef30743Sths                 term_printf(" backing_file=");
1104fef30743Sths 		term_print_filename(bs->backing_file);
1105fef30743Sths 	    }
1106b338082bSbellard             term_printf(" ro=%d", bs->read_only);
1107ea2384d3Sbellard             term_printf(" drv=%s", bs->drv->format_name);
1108ea2384d3Sbellard             if (bs->encrypted)
1109ea2384d3Sbellard                 term_printf(" encrypted");
1110b338082bSbellard         } else {
1111b338082bSbellard             term_printf(" [not inserted]");
1112b338082bSbellard         }
1113b338082bSbellard         term_printf("\n");
1114b338082bSbellard     }
1115b338082bSbellard }
1116a36e69ddSths 
1117a36e69ddSths /* The "info blockstats" command. */
1118a36e69ddSths void bdrv_info_stats (void)
1119a36e69ddSths {
1120a36e69ddSths     BlockDriverState *bs;
1121a7cbfae0Saliguori     BlockDriverInfo bdi;
1122a36e69ddSths 
1123a36e69ddSths     for (bs = bdrv_first; bs != NULL; bs = bs->next) {
1124a36e69ddSths 	term_printf ("%s:"
1125a36e69ddSths 		     " rd_bytes=%" PRIu64
1126a36e69ddSths 		     " wr_bytes=%" PRIu64
1127a36e69ddSths 		     " rd_operations=%" PRIu64
1128a36e69ddSths 		     " wr_operations=%" PRIu64
1129a7cbfae0Saliguori                      ,
1130a36e69ddSths 		     bs->device_name,
1131a36e69ddSths 		     bs->rd_bytes, bs->wr_bytes,
1132a36e69ddSths 		     bs->rd_ops, bs->wr_ops);
1133a7cbfae0Saliguori         if (bdrv_get_info(bs, &bdi) == 0)
11341987530fSaliguori             term_printf(" high=%" PRId64
11351987530fSaliguori                         " bytes_free=%" PRId64,
11361987530fSaliguori                         bdi.highest_alloc, bdi.num_free_bytes);
1137a7cbfae0Saliguori         term_printf("\n");
1138a36e69ddSths     }
1139a36e69ddSths }
1140ea2384d3Sbellard 
114183f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
114283f64091Sbellard                                char *filename, int filename_size)
114383f64091Sbellard {
114483f64091Sbellard     if (!bs->backing_hd) {
114583f64091Sbellard         pstrcpy(filename, filename_size, "");
114683f64091Sbellard     } else {
114783f64091Sbellard         pstrcpy(filename, filename_size, bs->backing_file);
114883f64091Sbellard     }
114983f64091Sbellard }
115083f64091Sbellard 
1151faea38e7Sbellard int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
1152faea38e7Sbellard                           const uint8_t *buf, int nb_sectors)
1153faea38e7Sbellard {
1154faea38e7Sbellard     BlockDriver *drv = bs->drv;
1155faea38e7Sbellard     if (!drv)
115619cb3738Sbellard         return -ENOMEDIUM;
1157faea38e7Sbellard     if (!drv->bdrv_write_compressed)
1158faea38e7Sbellard         return -ENOTSUP;
1159faea38e7Sbellard     return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
1160faea38e7Sbellard }
1161faea38e7Sbellard 
1162faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
1163faea38e7Sbellard {
1164faea38e7Sbellard     BlockDriver *drv = bs->drv;
1165faea38e7Sbellard     if (!drv)
116619cb3738Sbellard         return -ENOMEDIUM;
1167faea38e7Sbellard     if (!drv->bdrv_get_info)
1168faea38e7Sbellard         return -ENOTSUP;
1169faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
1170faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
1171faea38e7Sbellard }
1172faea38e7Sbellard 
1173faea38e7Sbellard /**************************************************************/
1174faea38e7Sbellard /* handling of snapshots */
1175faea38e7Sbellard 
1176faea38e7Sbellard int bdrv_snapshot_create(BlockDriverState *bs,
1177faea38e7Sbellard                          QEMUSnapshotInfo *sn_info)
1178faea38e7Sbellard {
1179faea38e7Sbellard     BlockDriver *drv = bs->drv;
1180faea38e7Sbellard     if (!drv)
118119cb3738Sbellard         return -ENOMEDIUM;
1182faea38e7Sbellard     if (!drv->bdrv_snapshot_create)
1183faea38e7Sbellard         return -ENOTSUP;
1184faea38e7Sbellard     return drv->bdrv_snapshot_create(bs, sn_info);
1185faea38e7Sbellard }
1186faea38e7Sbellard 
1187faea38e7Sbellard int bdrv_snapshot_goto(BlockDriverState *bs,
1188faea38e7Sbellard                        const char *snapshot_id)
1189faea38e7Sbellard {
1190faea38e7Sbellard     BlockDriver *drv = bs->drv;
1191faea38e7Sbellard     if (!drv)
119219cb3738Sbellard         return -ENOMEDIUM;
1193faea38e7Sbellard     if (!drv->bdrv_snapshot_goto)
1194faea38e7Sbellard         return -ENOTSUP;
1195faea38e7Sbellard     return drv->bdrv_snapshot_goto(bs, snapshot_id);
1196faea38e7Sbellard }
1197faea38e7Sbellard 
1198faea38e7Sbellard int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id)
1199faea38e7Sbellard {
1200faea38e7Sbellard     BlockDriver *drv = bs->drv;
1201faea38e7Sbellard     if (!drv)
120219cb3738Sbellard         return -ENOMEDIUM;
1203faea38e7Sbellard     if (!drv->bdrv_snapshot_delete)
1204faea38e7Sbellard         return -ENOTSUP;
1205faea38e7Sbellard     return drv->bdrv_snapshot_delete(bs, snapshot_id);
1206faea38e7Sbellard }
1207faea38e7Sbellard 
1208faea38e7Sbellard int bdrv_snapshot_list(BlockDriverState *bs,
1209faea38e7Sbellard                        QEMUSnapshotInfo **psn_info)
1210faea38e7Sbellard {
1211faea38e7Sbellard     BlockDriver *drv = bs->drv;
1212faea38e7Sbellard     if (!drv)
121319cb3738Sbellard         return -ENOMEDIUM;
1214faea38e7Sbellard     if (!drv->bdrv_snapshot_list)
1215faea38e7Sbellard         return -ENOTSUP;
1216faea38e7Sbellard     return drv->bdrv_snapshot_list(bs, psn_info);
1217faea38e7Sbellard }
1218faea38e7Sbellard 
1219faea38e7Sbellard #define NB_SUFFIXES 4
1220faea38e7Sbellard 
1221faea38e7Sbellard char *get_human_readable_size(char *buf, int buf_size, int64_t size)
1222faea38e7Sbellard {
1223faea38e7Sbellard     static const char suffixes[NB_SUFFIXES] = "KMGT";
1224faea38e7Sbellard     int64_t base;
1225faea38e7Sbellard     int i;
1226faea38e7Sbellard 
1227faea38e7Sbellard     if (size <= 999) {
1228faea38e7Sbellard         snprintf(buf, buf_size, "%" PRId64, size);
1229faea38e7Sbellard     } else {
1230faea38e7Sbellard         base = 1024;
1231faea38e7Sbellard         for(i = 0; i < NB_SUFFIXES; i++) {
1232faea38e7Sbellard             if (size < (10 * base)) {
1233faea38e7Sbellard                 snprintf(buf, buf_size, "%0.1f%c",
1234faea38e7Sbellard                          (double)size / base,
1235faea38e7Sbellard                          suffixes[i]);
1236faea38e7Sbellard                 break;
1237faea38e7Sbellard             } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) {
1238faea38e7Sbellard                 snprintf(buf, buf_size, "%" PRId64 "%c",
1239faea38e7Sbellard                          ((size + (base >> 1)) / base),
1240faea38e7Sbellard                          suffixes[i]);
1241faea38e7Sbellard                 break;
1242faea38e7Sbellard             }
1243faea38e7Sbellard             base = base * 1024;
1244faea38e7Sbellard         }
1245faea38e7Sbellard     }
1246faea38e7Sbellard     return buf;
1247faea38e7Sbellard }
1248faea38e7Sbellard 
1249faea38e7Sbellard char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn)
1250faea38e7Sbellard {
1251faea38e7Sbellard     char buf1[128], date_buf[128], clock_buf[128];
12523b9f94e1Sbellard #ifdef _WIN32
12533b9f94e1Sbellard     struct tm *ptm;
12543b9f94e1Sbellard #else
1255faea38e7Sbellard     struct tm tm;
12563b9f94e1Sbellard #endif
1257faea38e7Sbellard     time_t ti;
1258faea38e7Sbellard     int64_t secs;
1259faea38e7Sbellard 
1260faea38e7Sbellard     if (!sn) {
1261faea38e7Sbellard         snprintf(buf, buf_size,
1262faea38e7Sbellard                  "%-10s%-20s%7s%20s%15s",
1263faea38e7Sbellard                  "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK");
1264faea38e7Sbellard     } else {
1265faea38e7Sbellard         ti = sn->date_sec;
12663b9f94e1Sbellard #ifdef _WIN32
12673b9f94e1Sbellard         ptm = localtime(&ti);
12683b9f94e1Sbellard         strftime(date_buf, sizeof(date_buf),
12693b9f94e1Sbellard                  "%Y-%m-%d %H:%M:%S", ptm);
12703b9f94e1Sbellard #else
1271faea38e7Sbellard         localtime_r(&ti, &tm);
1272faea38e7Sbellard         strftime(date_buf, sizeof(date_buf),
1273faea38e7Sbellard                  "%Y-%m-%d %H:%M:%S", &tm);
12743b9f94e1Sbellard #endif
1275faea38e7Sbellard         secs = sn->vm_clock_nsec / 1000000000;
1276faea38e7Sbellard         snprintf(clock_buf, sizeof(clock_buf),
1277faea38e7Sbellard                  "%02d:%02d:%02d.%03d",
1278faea38e7Sbellard                  (int)(secs / 3600),
1279faea38e7Sbellard                  (int)((secs / 60) % 60),
1280faea38e7Sbellard                  (int)(secs % 60),
1281faea38e7Sbellard                  (int)((sn->vm_clock_nsec / 1000000) % 1000));
1282faea38e7Sbellard         snprintf(buf, buf_size,
1283faea38e7Sbellard                  "%-10s%-20s%7s%20s%15s",
1284faea38e7Sbellard                  sn->id_str, sn->name,
1285faea38e7Sbellard                  get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size),
1286faea38e7Sbellard                  date_buf,
1287faea38e7Sbellard                  clock_buf);
1288faea38e7Sbellard     }
1289faea38e7Sbellard     return buf;
1290faea38e7Sbellard }
1291faea38e7Sbellard 
129283f64091Sbellard 
1293ea2384d3Sbellard /**************************************************************/
129483f64091Sbellard /* async I/Os */
1295ea2384d3Sbellard 
12963b69e4b9Saliguori typedef struct VectorTranslationState {
12973b69e4b9Saliguori     QEMUIOVector *iov;
12983b69e4b9Saliguori     uint8_t *bounce;
12993b69e4b9Saliguori     int is_write;
13003b69e4b9Saliguori     BlockDriverAIOCB *aiocb;
13013b69e4b9Saliguori     BlockDriverAIOCB *this_aiocb;
13023b69e4b9Saliguori } VectorTranslationState;
13033b69e4b9Saliguori 
13043b69e4b9Saliguori static void bdrv_aio_rw_vector_cb(void *opaque, int ret)
13053b69e4b9Saliguori {
13063b69e4b9Saliguori     VectorTranslationState *s = opaque;
13073b69e4b9Saliguori 
13083b69e4b9Saliguori     if (!s->is_write) {
1309249aa745Saliguori         qemu_iovec_from_buffer(s->iov, s->bounce, s->iov->size);
13103b69e4b9Saliguori     }
1311d905dba4Saurel32     qemu_vfree(s->bounce);
13123b69e4b9Saliguori     s->this_aiocb->cb(s->this_aiocb->opaque, ret);
13133b69e4b9Saliguori     qemu_aio_release(s->this_aiocb);
13143b69e4b9Saliguori }
13153b69e4b9Saliguori 
13163b69e4b9Saliguori static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
13173b69e4b9Saliguori                                             int64_t sector_num,
13183b69e4b9Saliguori                                             QEMUIOVector *iov,
13193b69e4b9Saliguori                                             int nb_sectors,
13203b69e4b9Saliguori                                             BlockDriverCompletionFunc *cb,
13213b69e4b9Saliguori                                             void *opaque,
13223b69e4b9Saliguori                                             int is_write)
13233b69e4b9Saliguori 
13243b69e4b9Saliguori {
13253b69e4b9Saliguori     VectorTranslationState *s = qemu_mallocz(sizeof(*s));
13263b69e4b9Saliguori     BlockDriverAIOCB *aiocb = qemu_aio_get(bs, cb, opaque);
13273b69e4b9Saliguori 
13283b69e4b9Saliguori     s->this_aiocb = aiocb;
13293b69e4b9Saliguori     s->iov = iov;
13303b69e4b9Saliguori     s->bounce = qemu_memalign(512, nb_sectors * 512);
13313b69e4b9Saliguori     s->is_write = is_write;
13323b69e4b9Saliguori     if (is_write) {
13333b69e4b9Saliguori         qemu_iovec_to_buffer(s->iov, s->bounce);
13343b69e4b9Saliguori         s->aiocb = bdrv_aio_write(bs, sector_num, s->bounce, nb_sectors,
13353b69e4b9Saliguori                                   bdrv_aio_rw_vector_cb, s);
13363b69e4b9Saliguori     } else {
13373b69e4b9Saliguori         s->aiocb = bdrv_aio_read(bs, sector_num, s->bounce, nb_sectors,
13383b69e4b9Saliguori                                  bdrv_aio_rw_vector_cb, s);
13393b69e4b9Saliguori     }
13403b69e4b9Saliguori     return aiocb;
13413b69e4b9Saliguori }
13423b69e4b9Saliguori 
13433b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
13443b69e4b9Saliguori                                  QEMUIOVector *iov, int nb_sectors,
13453b69e4b9Saliguori                                  BlockDriverCompletionFunc *cb, void *opaque)
13463b69e4b9Saliguori {
1347*71d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
1348*71d0770cSaliguori         return NULL;
1349*71d0770cSaliguori 
13503b69e4b9Saliguori     return bdrv_aio_rw_vector(bs, sector_num, iov, nb_sectors,
13513b69e4b9Saliguori                               cb, opaque, 0);
13523b69e4b9Saliguori }
13533b69e4b9Saliguori 
13543b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
13553b69e4b9Saliguori                                   QEMUIOVector *iov, int nb_sectors,
13563b69e4b9Saliguori                                   BlockDriverCompletionFunc *cb, void *opaque)
13573b69e4b9Saliguori {
1358*71d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
1359*71d0770cSaliguori         return NULL;
1360*71d0770cSaliguori 
13613b69e4b9Saliguori     return bdrv_aio_rw_vector(bs, sector_num, iov, nb_sectors,
13623b69e4b9Saliguori                               cb, opaque, 1);
13633b69e4b9Saliguori }
13643b69e4b9Saliguori 
1365ce1a14dcSpbrook BlockDriverAIOCB *bdrv_aio_read(BlockDriverState *bs, int64_t sector_num,
136683f64091Sbellard                                 uint8_t *buf, int nb_sectors,
136783f64091Sbellard                                 BlockDriverCompletionFunc *cb, void *opaque)
1368ea2384d3Sbellard {
136983f64091Sbellard     BlockDriver *drv = bs->drv;
1370a36e69ddSths     BlockDriverAIOCB *ret;
1371ea2384d3Sbellard 
137219cb3738Sbellard     if (!drv)
1373ce1a14dcSpbrook         return NULL;
1374*71d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
1375*71d0770cSaliguori         return NULL;
137683f64091Sbellard 
1377a36e69ddSths     ret = drv->bdrv_aio_read(bs, sector_num, buf, nb_sectors, cb, opaque);
1378a36e69ddSths 
1379a36e69ddSths     if (ret) {
1380a36e69ddSths 	/* Update stats even though technically transfer has not happened. */
1381a36e69ddSths 	bs->rd_bytes += (unsigned) nb_sectors * SECTOR_SIZE;
1382a36e69ddSths 	bs->rd_ops ++;
1383a36e69ddSths     }
1384a36e69ddSths 
1385a36e69ddSths     return ret;
138683f64091Sbellard }
138783f64091Sbellard 
1388ce1a14dcSpbrook BlockDriverAIOCB *bdrv_aio_write(BlockDriverState *bs, int64_t sector_num,
138983f64091Sbellard                                  const uint8_t *buf, int nb_sectors,
139083f64091Sbellard                                  BlockDriverCompletionFunc *cb, void *opaque)
13917674e7bfSbellard {
139283f64091Sbellard     BlockDriver *drv = bs->drv;
1393a36e69ddSths     BlockDriverAIOCB *ret;
139483f64091Sbellard 
139519cb3738Sbellard     if (!drv)
1396ce1a14dcSpbrook         return NULL;
139783f64091Sbellard     if (bs->read_only)
1398ce1a14dcSpbrook         return NULL;
1399*71d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
1400*71d0770cSaliguori         return NULL;
140183f64091Sbellard 
1402a36e69ddSths     ret = drv->bdrv_aio_write(bs, sector_num, buf, nb_sectors, cb, opaque);
1403a36e69ddSths 
1404a36e69ddSths     if (ret) {
1405a36e69ddSths 	/* Update stats even though technically transfer has not happened. */
1406a36e69ddSths 	bs->wr_bytes += (unsigned) nb_sectors * SECTOR_SIZE;
1407a36e69ddSths 	bs->wr_ops ++;
1408a36e69ddSths     }
1409a36e69ddSths 
1410a36e69ddSths     return ret;
141183f64091Sbellard }
141283f64091Sbellard 
141383f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb)
141483f64091Sbellard {
1415ce1a14dcSpbrook     BlockDriver *drv = acb->bs->drv;
141683f64091Sbellard 
14173b69e4b9Saliguori     if (acb->cb == bdrv_aio_rw_vector_cb) {
14183b69e4b9Saliguori         VectorTranslationState *s = acb->opaque;
14193b69e4b9Saliguori         acb = s->aiocb;
14203b69e4b9Saliguori     }
14213b69e4b9Saliguori 
142283f64091Sbellard     drv->bdrv_aio_cancel(acb);
142383f64091Sbellard }
142483f64091Sbellard 
142583f64091Sbellard 
142683f64091Sbellard /**************************************************************/
142783f64091Sbellard /* async block device emulation */
142883f64091Sbellard 
142983f64091Sbellard static void bdrv_aio_bh_cb(void *opaque)
1430beac80cdSbellard {
1431ce1a14dcSpbrook     BlockDriverAIOCBSync *acb = opaque;
1432ce1a14dcSpbrook     acb->common.cb(acb->common.opaque, acb->ret);
1433ce1a14dcSpbrook     qemu_aio_release(acb);
1434beac80cdSbellard }
1435beac80cdSbellard 
1436ce1a14dcSpbrook static BlockDriverAIOCB *bdrv_aio_read_em(BlockDriverState *bs,
1437ce1a14dcSpbrook         int64_t sector_num, uint8_t *buf, int nb_sectors,
1438ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque)
1439ea2384d3Sbellard {
1440ce1a14dcSpbrook     BlockDriverAIOCBSync *acb;
144183f64091Sbellard     int ret;
144283f64091Sbellard 
1443ce1a14dcSpbrook     acb = qemu_aio_get(bs, cb, opaque);
1444ce1a14dcSpbrook     if (!acb->bh)
1445ce1a14dcSpbrook         acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
1446ce1a14dcSpbrook     ret = bdrv_read(bs, sector_num, buf, nb_sectors);
1447ce1a14dcSpbrook     acb->ret = ret;
1448ce1a14dcSpbrook     qemu_bh_schedule(acb->bh);
1449ce1a14dcSpbrook     return &acb->common;
14507a6cba61Spbrook }
14517a6cba61Spbrook 
1452ce1a14dcSpbrook static BlockDriverAIOCB *bdrv_aio_write_em(BlockDriverState *bs,
1453ce1a14dcSpbrook         int64_t sector_num, const uint8_t *buf, int nb_sectors,
1454ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque)
145583f64091Sbellard {
1456ce1a14dcSpbrook     BlockDriverAIOCBSync *acb;
145783f64091Sbellard     int ret;
145883f64091Sbellard 
1459ce1a14dcSpbrook     acb = qemu_aio_get(bs, cb, opaque);
1460ce1a14dcSpbrook     if (!acb->bh)
1461ce1a14dcSpbrook         acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
1462ce1a14dcSpbrook     ret = bdrv_write(bs, sector_num, buf, nb_sectors);
1463ce1a14dcSpbrook     acb->ret = ret;
1464ce1a14dcSpbrook     qemu_bh_schedule(acb->bh);
1465ce1a14dcSpbrook     return &acb->common;
146683f64091Sbellard }
146783f64091Sbellard 
1468ce1a14dcSpbrook static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
146983f64091Sbellard {
1470ce1a14dcSpbrook     BlockDriverAIOCBSync *acb = (BlockDriverAIOCBSync *)blockacb;
1471ce1a14dcSpbrook     qemu_bh_cancel(acb->bh);
1472ce1a14dcSpbrook     qemu_aio_release(acb);
147383f64091Sbellard }
147483f64091Sbellard 
147583f64091Sbellard /**************************************************************/
147683f64091Sbellard /* sync block device emulation */
147783f64091Sbellard 
147883f64091Sbellard static void bdrv_rw_em_cb(void *opaque, int ret)
147983f64091Sbellard {
148083f64091Sbellard     *(int *)opaque = ret;
148183f64091Sbellard }
148283f64091Sbellard 
148383f64091Sbellard #define NOT_DONE 0x7fffffff
148483f64091Sbellard 
148583f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
148683f64091Sbellard                         uint8_t *buf, int nb_sectors)
148783f64091Sbellard {
1488ce1a14dcSpbrook     int async_ret;
1489ce1a14dcSpbrook     BlockDriverAIOCB *acb;
149083f64091Sbellard 
149183f64091Sbellard     async_ret = NOT_DONE;
1492ce1a14dcSpbrook     acb = bdrv_aio_read(bs, sector_num, buf, nb_sectors,
149383f64091Sbellard                         bdrv_rw_em_cb, &async_ret);
1494baf35cb9Saliguori     if (acb == NULL)
1495ce1a14dcSpbrook         return -1;
1496baf35cb9Saliguori 
149783f64091Sbellard     while (async_ret == NOT_DONE) {
149883f64091Sbellard         qemu_aio_wait();
149983f64091Sbellard     }
1500baf35cb9Saliguori 
150183f64091Sbellard     return async_ret;
150283f64091Sbellard }
150383f64091Sbellard 
150483f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
150583f64091Sbellard                          const uint8_t *buf, int nb_sectors)
150683f64091Sbellard {
1507ce1a14dcSpbrook     int async_ret;
1508ce1a14dcSpbrook     BlockDriverAIOCB *acb;
150983f64091Sbellard 
151083f64091Sbellard     async_ret = NOT_DONE;
1511ce1a14dcSpbrook     acb = bdrv_aio_write(bs, sector_num, buf, nb_sectors,
151283f64091Sbellard                          bdrv_rw_em_cb, &async_ret);
1513baf35cb9Saliguori     if (acb == NULL)
1514ce1a14dcSpbrook         return -1;
151583f64091Sbellard     while (async_ret == NOT_DONE) {
151683f64091Sbellard         qemu_aio_wait();
151783f64091Sbellard     }
151883f64091Sbellard     return async_ret;
151983f64091Sbellard }
1520ea2384d3Sbellard 
1521ea2384d3Sbellard void bdrv_init(void)
1522ea2384d3Sbellard {
1523ea2384d3Sbellard     bdrv_register(&bdrv_raw);
152419cb3738Sbellard     bdrv_register(&bdrv_host_device);
1525ea2384d3Sbellard #ifndef _WIN32
1526ea2384d3Sbellard     bdrv_register(&bdrv_cow);
1527ea2384d3Sbellard #endif
1528ea2384d3Sbellard     bdrv_register(&bdrv_qcow);
1529ea2384d3Sbellard     bdrv_register(&bdrv_vmdk);
15303c56521bSbellard     bdrv_register(&bdrv_cloop);
1531585d0ed9Sbellard     bdrv_register(&bdrv_dmg);
1532a8753c34Sbellard     bdrv_register(&bdrv_bochs);
15336a0f9e82Sbellard     bdrv_register(&bdrv_vpc);
1534712e7874Sbellard     bdrv_register(&bdrv_vvfat);
1535faea38e7Sbellard     bdrv_register(&bdrv_qcow2);
15366ada7453Sths     bdrv_register(&bdrv_parallels);
153775818250Sths     bdrv_register(&bdrv_nbd);
1538ea2384d3Sbellard }
1539ce1a14dcSpbrook 
1540ce1a14dcSpbrook void *qemu_aio_get(BlockDriverState *bs, BlockDriverCompletionFunc *cb,
1541ce1a14dcSpbrook                    void *opaque)
1542ce1a14dcSpbrook {
1543ce1a14dcSpbrook     BlockDriver *drv;
1544ce1a14dcSpbrook     BlockDriverAIOCB *acb;
1545ce1a14dcSpbrook 
1546ce1a14dcSpbrook     drv = bs->drv;
1547ce1a14dcSpbrook     if (drv->free_aiocb) {
1548ce1a14dcSpbrook         acb = drv->free_aiocb;
1549ce1a14dcSpbrook         drv->free_aiocb = acb->next;
1550ce1a14dcSpbrook     } else {
1551ce1a14dcSpbrook         acb = qemu_mallocz(drv->aiocb_size);
1552ce1a14dcSpbrook     }
1553ce1a14dcSpbrook     acb->bs = bs;
1554ce1a14dcSpbrook     acb->cb = cb;
1555ce1a14dcSpbrook     acb->opaque = opaque;
1556ce1a14dcSpbrook     return acb;
1557ce1a14dcSpbrook }
1558ce1a14dcSpbrook 
1559ce1a14dcSpbrook void qemu_aio_release(void *p)
1560ce1a14dcSpbrook {
1561ce1a14dcSpbrook     BlockDriverAIOCB *acb = p;
1562ce1a14dcSpbrook     BlockDriver *drv = acb->bs->drv;
1563ce1a14dcSpbrook     acb->next = drv->free_aiocb;
1564ce1a14dcSpbrook     drv->free_aiocb = acb;
1565ce1a14dcSpbrook }
156619cb3738Sbellard 
156719cb3738Sbellard /**************************************************************/
156819cb3738Sbellard /* removable device support */
156919cb3738Sbellard 
157019cb3738Sbellard /**
157119cb3738Sbellard  * Return TRUE if the media is present
157219cb3738Sbellard  */
157319cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
157419cb3738Sbellard {
157519cb3738Sbellard     BlockDriver *drv = bs->drv;
157619cb3738Sbellard     int ret;
157719cb3738Sbellard     if (!drv)
157819cb3738Sbellard         return 0;
157919cb3738Sbellard     if (!drv->bdrv_is_inserted)
158019cb3738Sbellard         return 1;
158119cb3738Sbellard     ret = drv->bdrv_is_inserted(bs);
158219cb3738Sbellard     return ret;
158319cb3738Sbellard }
158419cb3738Sbellard 
158519cb3738Sbellard /**
158619cb3738Sbellard  * Return TRUE if the media changed since the last call to this
158719cb3738Sbellard  * function. It is currently only used for floppy disks
158819cb3738Sbellard  */
158919cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
159019cb3738Sbellard {
159119cb3738Sbellard     BlockDriver *drv = bs->drv;
159219cb3738Sbellard     int ret;
159319cb3738Sbellard 
159419cb3738Sbellard     if (!drv || !drv->bdrv_media_changed)
159519cb3738Sbellard         ret = -ENOTSUP;
159619cb3738Sbellard     else
159719cb3738Sbellard         ret = drv->bdrv_media_changed(bs);
159819cb3738Sbellard     if (ret == -ENOTSUP)
159919cb3738Sbellard         ret = bs->media_changed;
160019cb3738Sbellard     bs->media_changed = 0;
160119cb3738Sbellard     return ret;
160219cb3738Sbellard }
160319cb3738Sbellard 
160419cb3738Sbellard /**
160519cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
160619cb3738Sbellard  */
160719cb3738Sbellard void bdrv_eject(BlockDriverState *bs, int eject_flag)
160819cb3738Sbellard {
160919cb3738Sbellard     BlockDriver *drv = bs->drv;
161019cb3738Sbellard     int ret;
161119cb3738Sbellard 
161219cb3738Sbellard     if (!drv || !drv->bdrv_eject) {
161319cb3738Sbellard         ret = -ENOTSUP;
161419cb3738Sbellard     } else {
161519cb3738Sbellard         ret = drv->bdrv_eject(bs, eject_flag);
161619cb3738Sbellard     }
161719cb3738Sbellard     if (ret == -ENOTSUP) {
161819cb3738Sbellard         if (eject_flag)
161919cb3738Sbellard             bdrv_close(bs);
162019cb3738Sbellard     }
162119cb3738Sbellard }
162219cb3738Sbellard 
162319cb3738Sbellard int bdrv_is_locked(BlockDriverState *bs)
162419cb3738Sbellard {
162519cb3738Sbellard     return bs->locked;
162619cb3738Sbellard }
162719cb3738Sbellard 
162819cb3738Sbellard /**
162919cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
163019cb3738Sbellard  * to eject it manually).
163119cb3738Sbellard  */
163219cb3738Sbellard void bdrv_set_locked(BlockDriverState *bs, int locked)
163319cb3738Sbellard {
163419cb3738Sbellard     BlockDriver *drv = bs->drv;
163519cb3738Sbellard 
163619cb3738Sbellard     bs->locked = locked;
163719cb3738Sbellard     if (drv && drv->bdrv_set_locked) {
163819cb3738Sbellard         drv->bdrv_set_locked(bs, locked);
163919cb3738Sbellard     }
164019cb3738Sbellard }
1641985a03b0Sths 
1642985a03b0Sths /* needed for generic scsi interface */
1643985a03b0Sths 
1644985a03b0Sths int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
1645985a03b0Sths {
1646985a03b0Sths     BlockDriver *drv = bs->drv;
1647985a03b0Sths 
1648985a03b0Sths     if (drv && drv->bdrv_ioctl)
1649985a03b0Sths         return drv->bdrv_ioctl(bs, req, buf);
1650985a03b0Sths     return -ENOTSUP;
1651985a03b0Sths }
1652