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" 25179a2c19Sblueswir1 #ifdef HOST_BSD 263990d09aSblueswir1 /* include native header before sys-queue.h */ 273990d09aSblueswir1 #include <sys/queue.h> 283990d09aSblueswir1 #endif 293990d09aSblueswir1 30faf07963Spbrook #include "qemu-common.h" 31376253ecSaliguori #include "monitor.h" 32ea2384d3Sbellard #include "block_int.h" 33fc01f7e7Sbellard 34179a2c19Sblueswir1 #ifdef HOST_BSD 357674e7bfSbellard #include <sys/types.h> 367674e7bfSbellard #include <sys/stat.h> 377674e7bfSbellard #include <sys/ioctl.h> 38c5e97233Sblueswir1 #ifndef __DragonFly__ 397674e7bfSbellard #include <sys/disk.h> 407674e7bfSbellard #endif 41c5e97233Sblueswir1 #endif 427674e7bfSbellard 4349dc768dSaliguori #ifdef _WIN32 4449dc768dSaliguori #include <windows.h> 4549dc768dSaliguori #endif 4649dc768dSaliguori 4783f64091Sbellard #define SECTOR_BITS 9 4883f64091Sbellard #define SECTOR_SIZE (1 << SECTOR_BITS) 493b0d4f61Sbellard 5090765429Sbellard typedef struct BlockDriverAIOCBSync { 5190765429Sbellard BlockDriverAIOCB common; 5290765429Sbellard QEMUBH *bh; 5390765429Sbellard int ret; 54f141eafeSaliguori /* vector translation state */ 55f141eafeSaliguori QEMUIOVector *qiov; 56f141eafeSaliguori uint8_t *bounce; 57f141eafeSaliguori int is_write; 5890765429Sbellard } BlockDriverAIOCBSync; 5990765429Sbellard 60f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs, 61f141eafeSaliguori int64_t sector_num, QEMUIOVector *qiov, int nb_sectors, 62c87c0672Saliguori BlockDriverCompletionFunc *cb, void *opaque); 63f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs, 64f141eafeSaliguori int64_t sector_num, QEMUIOVector *qiov, int nb_sectors, 65ce1a14dcSpbrook BlockDriverCompletionFunc *cb, void *opaque); 6683f64091Sbellard static void bdrv_aio_cancel_em(BlockDriverAIOCB *acb); 6783f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num, 6883f64091Sbellard uint8_t *buf, int nb_sectors); 6983f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num, 7083f64091Sbellard const uint8_t *buf, int nb_sectors); 71ec530c81Sbellard 727ee930d0Sblueswir1 BlockDriverState *bdrv_first; 737ee930d0Sblueswir1 74ea2384d3Sbellard static BlockDriver *first_drv; 75ea2384d3Sbellard 7683f64091Sbellard int path_is_absolute(const char *path) 7783f64091Sbellard { 7883f64091Sbellard const char *p; 7921664424Sbellard #ifdef _WIN32 8021664424Sbellard /* specific case for names like: "\\.\d:" */ 8121664424Sbellard if (*path == '/' || *path == '\\') 8221664424Sbellard return 1; 8321664424Sbellard #endif 8483f64091Sbellard p = strchr(path, ':'); 8583f64091Sbellard if (p) 8683f64091Sbellard p++; 8783f64091Sbellard else 8883f64091Sbellard p = path; 893b9f94e1Sbellard #ifdef _WIN32 903b9f94e1Sbellard return (*p == '/' || *p == '\\'); 913b9f94e1Sbellard #else 923b9f94e1Sbellard return (*p == '/'); 933b9f94e1Sbellard #endif 9483f64091Sbellard } 9583f64091Sbellard 9683f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a 9783f64091Sbellard path to it by considering it is relative to base_path. URL are 9883f64091Sbellard supported. */ 9983f64091Sbellard void path_combine(char *dest, int dest_size, 10083f64091Sbellard const char *base_path, 10183f64091Sbellard const char *filename) 10283f64091Sbellard { 10383f64091Sbellard const char *p, *p1; 10483f64091Sbellard int len; 10583f64091Sbellard 10683f64091Sbellard if (dest_size <= 0) 10783f64091Sbellard return; 10883f64091Sbellard if (path_is_absolute(filename)) { 10983f64091Sbellard pstrcpy(dest, dest_size, filename); 11083f64091Sbellard } else { 11183f64091Sbellard p = strchr(base_path, ':'); 11283f64091Sbellard if (p) 11383f64091Sbellard p++; 11483f64091Sbellard else 11583f64091Sbellard p = base_path; 1163b9f94e1Sbellard p1 = strrchr(base_path, '/'); 1173b9f94e1Sbellard #ifdef _WIN32 1183b9f94e1Sbellard { 1193b9f94e1Sbellard const char *p2; 1203b9f94e1Sbellard p2 = strrchr(base_path, '\\'); 1213b9f94e1Sbellard if (!p1 || p2 > p1) 1223b9f94e1Sbellard p1 = p2; 1233b9f94e1Sbellard } 1243b9f94e1Sbellard #endif 12583f64091Sbellard if (p1) 12683f64091Sbellard p1++; 12783f64091Sbellard else 12883f64091Sbellard p1 = base_path; 12983f64091Sbellard if (p1 > p) 13083f64091Sbellard p = p1; 13183f64091Sbellard len = p - base_path; 13283f64091Sbellard if (len > dest_size - 1) 13383f64091Sbellard len = dest_size - 1; 13483f64091Sbellard memcpy(dest, base_path, len); 13583f64091Sbellard dest[len] = '\0'; 13683f64091Sbellard pstrcat(dest, dest_size, filename); 13783f64091Sbellard } 13883f64091Sbellard } 13983f64091Sbellard 14083f64091Sbellard 1419596ebb7Spbrook static void bdrv_register(BlockDriver *bdrv) 142ea2384d3Sbellard { 143f141eafeSaliguori if (!bdrv->bdrv_aio_readv) { 14483f64091Sbellard /* add AIO emulation layer */ 145f141eafeSaliguori bdrv->bdrv_aio_readv = bdrv_aio_readv_em; 146f141eafeSaliguori bdrv->bdrv_aio_writev = bdrv_aio_writev_em; 14783f64091Sbellard bdrv->bdrv_aio_cancel = bdrv_aio_cancel_em; 14890765429Sbellard bdrv->aiocb_size = sizeof(BlockDriverAIOCBSync); 149eda578e5Saliguori } else if (!bdrv->bdrv_read) { 15083f64091Sbellard /* add synchronous IO emulation layer */ 15183f64091Sbellard bdrv->bdrv_read = bdrv_read_em; 15283f64091Sbellard bdrv->bdrv_write = bdrv_write_em; 15383f64091Sbellard } 1546bbff9a0Saliguori aio_pool_init(&bdrv->aio_pool, bdrv->aiocb_size, bdrv->bdrv_aio_cancel); 155ea2384d3Sbellard bdrv->next = first_drv; 156ea2384d3Sbellard first_drv = bdrv; 157ea2384d3Sbellard } 158b338082bSbellard 159b338082bSbellard /* create a new block device (by default it is empty) */ 160b338082bSbellard BlockDriverState *bdrv_new(const char *device_name) 161fc01f7e7Sbellard { 162b338082bSbellard BlockDriverState **pbs, *bs; 163b338082bSbellard 164b338082bSbellard bs = qemu_mallocz(sizeof(BlockDriverState)); 165b338082bSbellard pstrcpy(bs->device_name, sizeof(bs->device_name), device_name); 166ea2384d3Sbellard if (device_name[0] != '\0') { 167b338082bSbellard /* insert at the end */ 168b338082bSbellard pbs = &bdrv_first; 169b338082bSbellard while (*pbs != NULL) 170b338082bSbellard pbs = &(*pbs)->next; 171b338082bSbellard *pbs = bs; 172ea2384d3Sbellard } 173b338082bSbellard return bs; 174b338082bSbellard } 175b338082bSbellard 176ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name) 177ea2384d3Sbellard { 178ea2384d3Sbellard BlockDriver *drv1; 179ea2384d3Sbellard for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) { 180ea2384d3Sbellard if (!strcmp(drv1->format_name, format_name)) 181ea2384d3Sbellard return drv1; 182ea2384d3Sbellard } 183ea2384d3Sbellard return NULL; 184ea2384d3Sbellard } 185ea2384d3Sbellard 1865eb45639Saliguori int bdrv_create2(BlockDriver *drv, 1875eb45639Saliguori const char *filename, int64_t size_in_sectors, 1885eb45639Saliguori const char *backing_file, const char *backing_format, 1895eb45639Saliguori int flags) 1905eb45639Saliguori { 1915eb45639Saliguori if (drv->bdrv_create2) 1925eb45639Saliguori return drv->bdrv_create2(filename, size_in_sectors, backing_file, 1935eb45639Saliguori backing_format, flags); 1945eb45639Saliguori if (drv->bdrv_create) 1955eb45639Saliguori return drv->bdrv_create(filename, size_in_sectors, backing_file, 1965eb45639Saliguori flags); 1975eb45639Saliguori return -ENOTSUP; 1985eb45639Saliguori } 1995eb45639Saliguori 200ea2384d3Sbellard int bdrv_create(BlockDriver *drv, 201ea2384d3Sbellard const char *filename, int64_t size_in_sectors, 202ea2384d3Sbellard const char *backing_file, int flags) 203ea2384d3Sbellard { 204ea2384d3Sbellard if (!drv->bdrv_create) 205ea2384d3Sbellard return -ENOTSUP; 206ea2384d3Sbellard return drv->bdrv_create(filename, size_in_sectors, backing_file, flags); 207ea2384d3Sbellard } 208ea2384d3Sbellard 209d5249393Sbellard #ifdef _WIN32 21095389c86Sbellard void get_tmp_filename(char *filename, int size) 211d5249393Sbellard { 2123b9f94e1Sbellard char temp_dir[MAX_PATH]; 2133b9f94e1Sbellard 2143b9f94e1Sbellard GetTempPath(MAX_PATH, temp_dir); 2153b9f94e1Sbellard GetTempFileName(temp_dir, "qem", 0, filename); 216d5249393Sbellard } 217d5249393Sbellard #else 21895389c86Sbellard void get_tmp_filename(char *filename, int size) 219ea2384d3Sbellard { 220ea2384d3Sbellard int fd; 2217ccfb2ebSblueswir1 const char *tmpdir; 222d5249393Sbellard /* XXX: race condition possible */ 2230badc1eeSaurel32 tmpdir = getenv("TMPDIR"); 2240badc1eeSaurel32 if (!tmpdir) 2250badc1eeSaurel32 tmpdir = "/tmp"; 2260badc1eeSaurel32 snprintf(filename, size, "%s/vl.XXXXXX", tmpdir); 227ea2384d3Sbellard fd = mkstemp(filename); 228ea2384d3Sbellard close(fd); 229ea2384d3Sbellard } 230d5249393Sbellard #endif 231ea2384d3Sbellard 23219cb3738Sbellard #ifdef _WIN32 233f45512feSbellard static int is_windows_drive_prefix(const char *filename) 234f45512feSbellard { 235f45512feSbellard return (((filename[0] >= 'a' && filename[0] <= 'z') || 236f45512feSbellard (filename[0] >= 'A' && filename[0] <= 'Z')) && 237f45512feSbellard filename[1] == ':'); 238f45512feSbellard } 239f45512feSbellard 24019cb3738Sbellard static int is_windows_drive(const char *filename) 24119cb3738Sbellard { 242f45512feSbellard if (is_windows_drive_prefix(filename) && 243f45512feSbellard filename[2] == '\0') 24419cb3738Sbellard return 1; 24519cb3738Sbellard if (strstart(filename, "\\\\.\\", NULL) || 24619cb3738Sbellard strstart(filename, "//./", NULL)) 24719cb3738Sbellard return 1; 24819cb3738Sbellard return 0; 24919cb3738Sbellard } 25019cb3738Sbellard #endif 25119cb3738Sbellard 25283f64091Sbellard static BlockDriver *find_protocol(const char *filename) 25383f64091Sbellard { 25483f64091Sbellard BlockDriver *drv1; 25583f64091Sbellard char protocol[128]; 25683f64091Sbellard int len; 25783f64091Sbellard const char *p; 25819cb3738Sbellard 25919cb3738Sbellard #ifdef _WIN32 260f45512feSbellard if (is_windows_drive(filename) || 261f45512feSbellard is_windows_drive_prefix(filename)) 26219cb3738Sbellard return &bdrv_raw; 26319cb3738Sbellard #endif 26483f64091Sbellard p = strchr(filename, ':'); 26583f64091Sbellard if (!p) 26683f64091Sbellard return &bdrv_raw; 26783f64091Sbellard len = p - filename; 26883f64091Sbellard if (len > sizeof(protocol) - 1) 26983f64091Sbellard len = sizeof(protocol) - 1; 27083f64091Sbellard memcpy(protocol, filename, len); 27183f64091Sbellard protocol[len] = '\0'; 27283f64091Sbellard for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) { 27383f64091Sbellard if (drv1->protocol_name && 27483f64091Sbellard !strcmp(drv1->protocol_name, protocol)) 27583f64091Sbellard return drv1; 27683f64091Sbellard } 27783f64091Sbellard return NULL; 27883f64091Sbellard } 27983f64091Sbellard 2807674e7bfSbellard /* XXX: force raw format if block or character device ? It would 2817674e7bfSbellard simplify the BSD case */ 282ea2384d3Sbellard static BlockDriver *find_image_format(const char *filename) 283ea2384d3Sbellard { 28483f64091Sbellard int ret, score, score_max; 285ea2384d3Sbellard BlockDriver *drv1, *drv; 28683f64091Sbellard uint8_t buf[2048]; 28783f64091Sbellard BlockDriverState *bs; 288ea2384d3Sbellard 28919cb3738Sbellard /* detect host devices. By convention, /dev/cdrom[N] is always 29019cb3738Sbellard recognized as a host CDROM */ 29119cb3738Sbellard if (strstart(filename, "/dev/cdrom", NULL)) 29219cb3738Sbellard return &bdrv_host_device; 29319cb3738Sbellard #ifdef _WIN32 29419cb3738Sbellard if (is_windows_drive(filename)) 29519cb3738Sbellard return &bdrv_host_device; 29619cb3738Sbellard #else 29719cb3738Sbellard { 29819cb3738Sbellard struct stat st; 29919cb3738Sbellard if (stat(filename, &st) >= 0 && 30019cb3738Sbellard (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))) { 30119cb3738Sbellard return &bdrv_host_device; 30219cb3738Sbellard } 30319cb3738Sbellard } 30419cb3738Sbellard #endif 30519cb3738Sbellard 30683f64091Sbellard drv = find_protocol(filename); 30719cb3738Sbellard /* no need to test disk image formats for vvfat */ 30883f64091Sbellard if (drv == &bdrv_vvfat) 30983f64091Sbellard return drv; 31083f64091Sbellard 31183f64091Sbellard ret = bdrv_file_open(&bs, filename, BDRV_O_RDONLY); 31283f64091Sbellard if (ret < 0) 3137674e7bfSbellard return NULL; 31483f64091Sbellard ret = bdrv_pread(bs, 0, buf, sizeof(buf)); 31583f64091Sbellard bdrv_delete(bs); 316ea2384d3Sbellard if (ret < 0) { 317ea2384d3Sbellard return NULL; 318ea2384d3Sbellard } 319ea2384d3Sbellard 320ea2384d3Sbellard score_max = 0; 321ea2384d3Sbellard for(drv1 = first_drv; drv1 != NULL; drv1 = drv1->next) { 32283f64091Sbellard if (drv1->bdrv_probe) { 323ea2384d3Sbellard score = drv1->bdrv_probe(buf, ret, filename); 324ea2384d3Sbellard if (score > score_max) { 325ea2384d3Sbellard score_max = score; 326ea2384d3Sbellard drv = drv1; 327ea2384d3Sbellard } 328ea2384d3Sbellard } 32983f64091Sbellard } 330ea2384d3Sbellard return drv; 331ea2384d3Sbellard } 332ea2384d3Sbellard 33383f64091Sbellard int bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags) 334b338082bSbellard { 33583f64091Sbellard BlockDriverState *bs; 33683f64091Sbellard int ret; 3373b0d4f61Sbellard 33883f64091Sbellard bs = bdrv_new(""); 33983f64091Sbellard ret = bdrv_open2(bs, filename, flags | BDRV_O_FILE, NULL); 34083f64091Sbellard if (ret < 0) { 34183f64091Sbellard bdrv_delete(bs); 34283f64091Sbellard return ret; 3433b0d4f61Sbellard } 34471d0770cSaliguori bs->growable = 1; 34583f64091Sbellard *pbs = bs; 34683f64091Sbellard return 0; 3473b0d4f61Sbellard } 3483b0d4f61Sbellard 34983f64091Sbellard int bdrv_open(BlockDriverState *bs, const char *filename, int flags) 35083f64091Sbellard { 35183f64091Sbellard return bdrv_open2(bs, filename, flags, NULL); 352ea2384d3Sbellard } 353ea2384d3Sbellard 35483f64091Sbellard int bdrv_open2(BlockDriverState *bs, const char *filename, int flags, 355ea2384d3Sbellard BlockDriver *drv) 356ea2384d3Sbellard { 35783f64091Sbellard int ret, open_flags; 358eb5c851fSths char tmp_filename[PATH_MAX]; 359eb5c851fSths char backing_filename[PATH_MAX]; 360fc01f7e7Sbellard 3610849bf08Sbellard bs->read_only = 0; 362ea2384d3Sbellard bs->is_temporary = 0; 363ea2384d3Sbellard bs->encrypted = 0; 364c0f4ce77Saliguori bs->valid_key = 0; 36533e3963eSbellard 36683f64091Sbellard if (flags & BDRV_O_SNAPSHOT) { 367ea2384d3Sbellard BlockDriverState *bs1; 368ea2384d3Sbellard int64_t total_size; 3697c96d46eSaliguori int is_protocol = 0; 37033e3963eSbellard 371ea2384d3Sbellard /* if snapshot, we create a temporary backing file and open it 372ea2384d3Sbellard instead of opening 'filename' directly */ 373ea2384d3Sbellard 374ea2384d3Sbellard /* if there is a backing file, use it */ 375ea2384d3Sbellard bs1 = bdrv_new(""); 3765eb45639Saliguori ret = bdrv_open2(bs1, filename, 0, drv); 37751d7c00cSaliguori if (ret < 0) { 378ea2384d3Sbellard bdrv_delete(bs1); 37951d7c00cSaliguori return ret; 380ea2384d3Sbellard } 38183f64091Sbellard total_size = bdrv_getlength(bs1) >> SECTOR_BITS; 3827c96d46eSaliguori 3837c96d46eSaliguori if (bs1->drv && bs1->drv->protocol_name) 3847c96d46eSaliguori is_protocol = 1; 3857c96d46eSaliguori 386ea2384d3Sbellard bdrv_delete(bs1); 387ea2384d3Sbellard 388ea2384d3Sbellard get_tmp_filename(tmp_filename, sizeof(tmp_filename)); 3897c96d46eSaliguori 3907c96d46eSaliguori /* Real path is meaningless for protocols */ 3917c96d46eSaliguori if (is_protocol) 3927c96d46eSaliguori snprintf(backing_filename, sizeof(backing_filename), 3937c96d46eSaliguori "%s", filename); 3947c96d46eSaliguori else 395a817d936Sbellard realpath(filename, backing_filename); 3967c96d46eSaliguori 3975eb45639Saliguori ret = bdrv_create2(&bdrv_qcow2, tmp_filename, 3985eb45639Saliguori total_size, backing_filename, 3995eb45639Saliguori (drv ? drv->format_name : NULL), 0); 40051d7c00cSaliguori if (ret < 0) { 40151d7c00cSaliguori return ret; 402ea2384d3Sbellard } 403ea2384d3Sbellard filename = tmp_filename; 4045eb45639Saliguori drv = &bdrv_qcow2; 405ea2384d3Sbellard bs->is_temporary = 1; 406ea2384d3Sbellard } 407ea2384d3Sbellard 408ea2384d3Sbellard pstrcpy(bs->filename, sizeof(bs->filename), filename); 40983f64091Sbellard if (flags & BDRV_O_FILE) { 41083f64091Sbellard drv = find_protocol(filename); 41151d7c00cSaliguori } else if (!drv) { 412ea2384d3Sbellard drv = find_image_format(filename); 413ea2384d3Sbellard } 41451d7c00cSaliguori if (!drv) { 41551d7c00cSaliguori ret = -ENOENT; 41651d7c00cSaliguori goto unlink_and_fail; 41783f64091Sbellard } 418ea2384d3Sbellard bs->drv = drv; 419ea2384d3Sbellard bs->opaque = qemu_mallocz(drv->instance_size); 42083f64091Sbellard /* Note: for compatibility, we open disk image files as RDWR, and 42183f64091Sbellard RDONLY as fallback */ 42283f64091Sbellard if (!(flags & BDRV_O_FILE)) 4239f7965c7Saliguori open_flags = BDRV_O_RDWR | (flags & BDRV_O_CACHE_MASK); 42483f64091Sbellard else 42583f64091Sbellard open_flags = flags & ~(BDRV_O_FILE | BDRV_O_SNAPSHOT); 42683f64091Sbellard ret = drv->bdrv_open(bs, filename, open_flags); 427a0a83536Saurel32 if ((ret == -EACCES || ret == -EPERM) && !(flags & BDRV_O_FILE)) { 4289f7965c7Saliguori ret = drv->bdrv_open(bs, filename, open_flags & ~BDRV_O_RDWR); 42983f64091Sbellard bs->read_only = 1; 43083f64091Sbellard } 431ea2384d3Sbellard if (ret < 0) { 432ea2384d3Sbellard qemu_free(bs->opaque); 4336b21b973Sbellard bs->opaque = NULL; 4346b21b973Sbellard bs->drv = NULL; 43551d7c00cSaliguori unlink_and_fail: 43651d7c00cSaliguori if (bs->is_temporary) 43751d7c00cSaliguori unlink(filename); 43883f64091Sbellard return ret; 439ea2384d3Sbellard } 440d15a771dSbellard if (drv->bdrv_getlength) { 441d15a771dSbellard bs->total_sectors = bdrv_getlength(bs) >> SECTOR_BITS; 442d15a771dSbellard } 443ea2384d3Sbellard #ifndef _WIN32 444ea2384d3Sbellard if (bs->is_temporary) { 445ea2384d3Sbellard unlink(filename); 44633e3963eSbellard } 44767b915a5Sbellard #endif 44883f64091Sbellard if (bs->backing_file[0] != '\0') { 449ea2384d3Sbellard /* if there is a backing file, use it */ 4505eb45639Saliguori BlockDriver *back_drv = NULL; 451ea2384d3Sbellard bs->backing_hd = bdrv_new(""); 45283f64091Sbellard path_combine(backing_filename, sizeof(backing_filename), 45383f64091Sbellard filename, bs->backing_file); 4545eb45639Saliguori if (bs->backing_format[0] != '\0') 4555eb45639Saliguori back_drv = bdrv_find_format(bs->backing_format); 4565eb45639Saliguori ret = bdrv_open2(bs->backing_hd, backing_filename, open_flags, 4575eb45639Saliguori back_drv); 45851d7c00cSaliguori if (ret < 0) { 45951d7c00cSaliguori bdrv_close(bs); 46051d7c00cSaliguori return ret; 46151d7c00cSaliguori } 462ea2384d3Sbellard } 46333e3963eSbellard 464bb5fc20fSaliguori if (!bdrv_key_required(bs)) { 465b338082bSbellard /* call the change callback */ 46619cb3738Sbellard bs->media_changed = 1; 467b338082bSbellard if (bs->change_cb) 468b338082bSbellard bs->change_cb(bs->change_opaque); 469bb5fc20fSaliguori } 470b338082bSbellard return 0; 471fc01f7e7Sbellard } 472fc01f7e7Sbellard 473fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs) 474fc01f7e7Sbellard { 47519cb3738Sbellard if (bs->drv) { 476ea2384d3Sbellard if (bs->backing_hd) 477ea2384d3Sbellard bdrv_delete(bs->backing_hd); 478ea2384d3Sbellard bs->drv->bdrv_close(bs); 479ea2384d3Sbellard qemu_free(bs->opaque); 480ea2384d3Sbellard #ifdef _WIN32 481ea2384d3Sbellard if (bs->is_temporary) { 482ea2384d3Sbellard unlink(bs->filename); 483ea2384d3Sbellard } 48467b915a5Sbellard #endif 485ea2384d3Sbellard bs->opaque = NULL; 486ea2384d3Sbellard bs->drv = NULL; 487b338082bSbellard 488b338082bSbellard /* call the change callback */ 48919cb3738Sbellard bs->media_changed = 1; 490b338082bSbellard if (bs->change_cb) 491b338082bSbellard bs->change_cb(bs->change_opaque); 492b338082bSbellard } 493b338082bSbellard } 494b338082bSbellard 495b338082bSbellard void bdrv_delete(BlockDriverState *bs) 496b338082bSbellard { 49734c6f050Saurel32 BlockDriverState **pbs; 49834c6f050Saurel32 49934c6f050Saurel32 pbs = &bdrv_first; 50034c6f050Saurel32 while (*pbs != bs && *pbs != NULL) 50134c6f050Saurel32 pbs = &(*pbs)->next; 50234c6f050Saurel32 if (*pbs == bs) 50334c6f050Saurel32 *pbs = bs->next; 50434c6f050Saurel32 505b338082bSbellard bdrv_close(bs); 506b338082bSbellard qemu_free(bs); 507fc01f7e7Sbellard } 508fc01f7e7Sbellard 509*e97fc193Saliguori /* 510*e97fc193Saliguori * Run consistency checks on an image 511*e97fc193Saliguori * 512*e97fc193Saliguori * Returns the number of errors or -errno when an internal error occurs 513*e97fc193Saliguori */ 514*e97fc193Saliguori int bdrv_check(BlockDriverState *bs) 515*e97fc193Saliguori { 516*e97fc193Saliguori if (bs->drv->bdrv_check == NULL) { 517*e97fc193Saliguori return -ENOTSUP; 518*e97fc193Saliguori } 519*e97fc193Saliguori 520*e97fc193Saliguori return bs->drv->bdrv_check(bs); 521*e97fc193Saliguori } 522*e97fc193Saliguori 52333e3963eSbellard /* commit COW file into the raw image */ 52433e3963eSbellard int bdrv_commit(BlockDriverState *bs) 52533e3963eSbellard { 52619cb3738Sbellard BlockDriver *drv = bs->drv; 52783f64091Sbellard int64_t i, total_sectors; 528ea2384d3Sbellard int n, j; 529ea2384d3Sbellard unsigned char sector[512]; 53033e3963eSbellard 53119cb3738Sbellard if (!drv) 53219cb3738Sbellard return -ENOMEDIUM; 53333e3963eSbellard 53433e3963eSbellard if (bs->read_only) { 535ea2384d3Sbellard return -EACCES; 53633e3963eSbellard } 53733e3963eSbellard 538ea2384d3Sbellard if (!bs->backing_hd) { 539ea2384d3Sbellard return -ENOTSUP; 540ea2384d3Sbellard } 541ea2384d3Sbellard 54283f64091Sbellard total_sectors = bdrv_getlength(bs) >> SECTOR_BITS; 54383f64091Sbellard for (i = 0; i < total_sectors;) { 54419cb3738Sbellard if (drv->bdrv_is_allocated(bs, i, 65536, &n)) { 545ea2384d3Sbellard for(j = 0; j < n; j++) { 54633e3963eSbellard if (bdrv_read(bs, i, sector, 1) != 0) { 547ea2384d3Sbellard return -EIO; 54833e3963eSbellard } 54933e3963eSbellard 550ea2384d3Sbellard if (bdrv_write(bs->backing_hd, i, sector, 1) != 0) { 551ea2384d3Sbellard return -EIO; 55233e3963eSbellard } 553ea2384d3Sbellard i++; 554ea2384d3Sbellard } 555ea2384d3Sbellard } else { 556ea2384d3Sbellard i += n; 55733e3963eSbellard } 55833e3963eSbellard } 55995389c86Sbellard 56019cb3738Sbellard if (drv->bdrv_make_empty) 56119cb3738Sbellard return drv->bdrv_make_empty(bs); 56295389c86Sbellard 56333e3963eSbellard return 0; 56433e3963eSbellard } 56533e3963eSbellard 56671d0770cSaliguori static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset, 56771d0770cSaliguori size_t size) 56871d0770cSaliguori { 56971d0770cSaliguori int64_t len; 57071d0770cSaliguori 57171d0770cSaliguori if (!bdrv_is_inserted(bs)) 57271d0770cSaliguori return -ENOMEDIUM; 57371d0770cSaliguori 57471d0770cSaliguori if (bs->growable) 57571d0770cSaliguori return 0; 57671d0770cSaliguori 57771d0770cSaliguori len = bdrv_getlength(bs); 57871d0770cSaliguori 57971d0770cSaliguori if ((offset + size) > len) 58071d0770cSaliguori return -EIO; 58171d0770cSaliguori 58271d0770cSaliguori return 0; 58371d0770cSaliguori } 58471d0770cSaliguori 58571d0770cSaliguori static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num, 58671d0770cSaliguori int nb_sectors) 58771d0770cSaliguori { 588999dec57Saliguori return bdrv_check_byte_request(bs, sector_num * 512, nb_sectors * 512); 58971d0770cSaliguori } 59071d0770cSaliguori 59119cb3738Sbellard /* return < 0 if error. See bdrv_write() for the return codes */ 592fc01f7e7Sbellard int bdrv_read(BlockDriverState *bs, int64_t sector_num, 593fc01f7e7Sbellard uint8_t *buf, int nb_sectors) 594fc01f7e7Sbellard { 595ea2384d3Sbellard BlockDriver *drv = bs->drv; 596fc01f7e7Sbellard 59719cb3738Sbellard if (!drv) 59819cb3738Sbellard return -ENOMEDIUM; 59971d0770cSaliguori if (bdrv_check_request(bs, sector_num, nb_sectors)) 60071d0770cSaliguori return -EIO; 601b338082bSbellard 60283f64091Sbellard return drv->bdrv_read(bs, sector_num, buf, nb_sectors); 60383f64091Sbellard } 604fc01f7e7Sbellard 60519cb3738Sbellard /* Return < 0 if error. Important errors are: 60619cb3738Sbellard -EIO generic I/O error (may happen for all errors) 60719cb3738Sbellard -ENOMEDIUM No media inserted. 60819cb3738Sbellard -EINVAL Invalid sector number or nb_sectors 60919cb3738Sbellard -EACCES Trying to write a read-only device 61019cb3738Sbellard */ 611fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num, 612fc01f7e7Sbellard const uint8_t *buf, int nb_sectors) 613fc01f7e7Sbellard { 61483f64091Sbellard BlockDriver *drv = bs->drv; 61519cb3738Sbellard if (!bs->drv) 61619cb3738Sbellard return -ENOMEDIUM; 6170849bf08Sbellard if (bs->read_only) 61819cb3738Sbellard return -EACCES; 61971d0770cSaliguori if (bdrv_check_request(bs, sector_num, nb_sectors)) 62071d0770cSaliguori return -EIO; 62171d0770cSaliguori 62283f64091Sbellard return drv->bdrv_write(bs, sector_num, buf, nb_sectors); 62383f64091Sbellard } 62483f64091Sbellard 625eda578e5Saliguori int bdrv_pread(BlockDriverState *bs, int64_t offset, 626eda578e5Saliguori void *buf, int count1) 62783f64091Sbellard { 62883f64091Sbellard uint8_t tmp_buf[SECTOR_SIZE]; 62983f64091Sbellard int len, nb_sectors, count; 63083f64091Sbellard int64_t sector_num; 63183f64091Sbellard 63283f64091Sbellard count = count1; 63383f64091Sbellard /* first read to align to sector start */ 63483f64091Sbellard len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1); 63583f64091Sbellard if (len > count) 63683f64091Sbellard len = count; 63783f64091Sbellard sector_num = offset >> SECTOR_BITS; 63883f64091Sbellard if (len > 0) { 63983f64091Sbellard if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0) 64083f64091Sbellard return -EIO; 64183f64091Sbellard memcpy(buf, tmp_buf + (offset & (SECTOR_SIZE - 1)), len); 64283f64091Sbellard count -= len; 64383f64091Sbellard if (count == 0) 64483f64091Sbellard return count1; 64583f64091Sbellard sector_num++; 64683f64091Sbellard buf += len; 64783f64091Sbellard } 64883f64091Sbellard 64983f64091Sbellard /* read the sectors "in place" */ 65083f64091Sbellard nb_sectors = count >> SECTOR_BITS; 65183f64091Sbellard if (nb_sectors > 0) { 65283f64091Sbellard if (bdrv_read(bs, sector_num, buf, nb_sectors) < 0) 65383f64091Sbellard return -EIO; 65483f64091Sbellard sector_num += nb_sectors; 65583f64091Sbellard len = nb_sectors << SECTOR_BITS; 65683f64091Sbellard buf += len; 65783f64091Sbellard count -= len; 65883f64091Sbellard } 65983f64091Sbellard 66083f64091Sbellard /* add data from the last sector */ 66183f64091Sbellard if (count > 0) { 66283f64091Sbellard if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0) 66383f64091Sbellard return -EIO; 66483f64091Sbellard memcpy(buf, tmp_buf, count); 66583f64091Sbellard } 66683f64091Sbellard return count1; 66783f64091Sbellard } 66883f64091Sbellard 669eda578e5Saliguori int bdrv_pwrite(BlockDriverState *bs, int64_t offset, 670eda578e5Saliguori const void *buf, int count1) 67183f64091Sbellard { 67283f64091Sbellard uint8_t tmp_buf[SECTOR_SIZE]; 67383f64091Sbellard int len, nb_sectors, count; 67483f64091Sbellard int64_t sector_num; 67583f64091Sbellard 67683f64091Sbellard count = count1; 67783f64091Sbellard /* first write to align to sector start */ 67883f64091Sbellard len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1); 67983f64091Sbellard if (len > count) 68083f64091Sbellard len = count; 68183f64091Sbellard sector_num = offset >> SECTOR_BITS; 68283f64091Sbellard if (len > 0) { 68383f64091Sbellard if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0) 68483f64091Sbellard return -EIO; 68583f64091Sbellard memcpy(tmp_buf + (offset & (SECTOR_SIZE - 1)), buf, len); 68683f64091Sbellard if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0) 68783f64091Sbellard return -EIO; 68883f64091Sbellard count -= len; 68983f64091Sbellard if (count == 0) 69083f64091Sbellard return count1; 69183f64091Sbellard sector_num++; 69283f64091Sbellard buf += len; 69383f64091Sbellard } 69483f64091Sbellard 69583f64091Sbellard /* write the sectors "in place" */ 69683f64091Sbellard nb_sectors = count >> SECTOR_BITS; 69783f64091Sbellard if (nb_sectors > 0) { 69883f64091Sbellard if (bdrv_write(bs, sector_num, buf, nb_sectors) < 0) 69983f64091Sbellard return -EIO; 70083f64091Sbellard sector_num += nb_sectors; 70183f64091Sbellard len = nb_sectors << SECTOR_BITS; 70283f64091Sbellard buf += len; 70383f64091Sbellard count -= len; 70483f64091Sbellard } 70583f64091Sbellard 70683f64091Sbellard /* add data from the last sector */ 70783f64091Sbellard if (count > 0) { 70883f64091Sbellard if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0) 70983f64091Sbellard return -EIO; 71083f64091Sbellard memcpy(tmp_buf, buf, count); 71183f64091Sbellard if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0) 71283f64091Sbellard return -EIO; 71383f64091Sbellard } 71483f64091Sbellard return count1; 71583f64091Sbellard } 71683f64091Sbellard 71783f64091Sbellard /** 71883f64091Sbellard * Truncate file to 'offset' bytes (needed only for file protocols) 71983f64091Sbellard */ 72083f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset) 72183f64091Sbellard { 72283f64091Sbellard BlockDriver *drv = bs->drv; 72383f64091Sbellard if (!drv) 72419cb3738Sbellard return -ENOMEDIUM; 72583f64091Sbellard if (!drv->bdrv_truncate) 72683f64091Sbellard return -ENOTSUP; 72783f64091Sbellard return drv->bdrv_truncate(bs, offset); 72883f64091Sbellard } 72983f64091Sbellard 73083f64091Sbellard /** 73183f64091Sbellard * Length of a file in bytes. Return < 0 if error or unknown. 73283f64091Sbellard */ 73383f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs) 73483f64091Sbellard { 73583f64091Sbellard BlockDriver *drv = bs->drv; 73683f64091Sbellard if (!drv) 73719cb3738Sbellard return -ENOMEDIUM; 73883f64091Sbellard if (!drv->bdrv_getlength) { 73983f64091Sbellard /* legacy mode */ 74083f64091Sbellard return bs->total_sectors * SECTOR_SIZE; 74183f64091Sbellard } 74283f64091Sbellard return drv->bdrv_getlength(bs); 743fc01f7e7Sbellard } 744fc01f7e7Sbellard 74519cb3738Sbellard /* return 0 as number of sectors if no device present or error */ 74696b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr) 747fc01f7e7Sbellard { 74819cb3738Sbellard int64_t length; 74919cb3738Sbellard length = bdrv_getlength(bs); 75019cb3738Sbellard if (length < 0) 75119cb3738Sbellard length = 0; 75219cb3738Sbellard else 75319cb3738Sbellard length = length >> SECTOR_BITS; 75419cb3738Sbellard *nb_sectors_ptr = length; 755fc01f7e7Sbellard } 756cf98951bSbellard 757f3d54fc4Saliguori struct partition { 758f3d54fc4Saliguori uint8_t boot_ind; /* 0x80 - active */ 759f3d54fc4Saliguori uint8_t head; /* starting head */ 760f3d54fc4Saliguori uint8_t sector; /* starting sector */ 761f3d54fc4Saliguori uint8_t cyl; /* starting cylinder */ 762f3d54fc4Saliguori uint8_t sys_ind; /* What partition type */ 763f3d54fc4Saliguori uint8_t end_head; /* end head */ 764f3d54fc4Saliguori uint8_t end_sector; /* end sector */ 765f3d54fc4Saliguori uint8_t end_cyl; /* end cylinder */ 766f3d54fc4Saliguori uint32_t start_sect; /* starting sector counting from 0 */ 767f3d54fc4Saliguori uint32_t nr_sects; /* nr of sectors in partition */ 768f3d54fc4Saliguori } __attribute__((packed)); 769f3d54fc4Saliguori 770f3d54fc4Saliguori /* try to guess the disk logical geometry from the MSDOS partition table. Return 0 if OK, -1 if could not guess */ 771f3d54fc4Saliguori static int guess_disk_lchs(BlockDriverState *bs, 772f3d54fc4Saliguori int *pcylinders, int *pheads, int *psectors) 773f3d54fc4Saliguori { 774f3d54fc4Saliguori uint8_t buf[512]; 775f3d54fc4Saliguori int ret, i, heads, sectors, cylinders; 776f3d54fc4Saliguori struct partition *p; 777f3d54fc4Saliguori uint32_t nr_sects; 778a38131b6Sblueswir1 uint64_t nb_sectors; 779f3d54fc4Saliguori 780f3d54fc4Saliguori bdrv_get_geometry(bs, &nb_sectors); 781f3d54fc4Saliguori 782f3d54fc4Saliguori ret = bdrv_read(bs, 0, buf, 1); 783f3d54fc4Saliguori if (ret < 0) 784f3d54fc4Saliguori return -1; 785f3d54fc4Saliguori /* test msdos magic */ 786f3d54fc4Saliguori if (buf[510] != 0x55 || buf[511] != 0xaa) 787f3d54fc4Saliguori return -1; 788f3d54fc4Saliguori for(i = 0; i < 4; i++) { 789f3d54fc4Saliguori p = ((struct partition *)(buf + 0x1be)) + i; 790f3d54fc4Saliguori nr_sects = le32_to_cpu(p->nr_sects); 791f3d54fc4Saliguori if (nr_sects && p->end_head) { 792f3d54fc4Saliguori /* We make the assumption that the partition terminates on 793f3d54fc4Saliguori a cylinder boundary */ 794f3d54fc4Saliguori heads = p->end_head + 1; 795f3d54fc4Saliguori sectors = p->end_sector & 63; 796f3d54fc4Saliguori if (sectors == 0) 797f3d54fc4Saliguori continue; 798f3d54fc4Saliguori cylinders = nb_sectors / (heads * sectors); 799f3d54fc4Saliguori if (cylinders < 1 || cylinders > 16383) 800f3d54fc4Saliguori continue; 801f3d54fc4Saliguori *pheads = heads; 802f3d54fc4Saliguori *psectors = sectors; 803f3d54fc4Saliguori *pcylinders = cylinders; 804f3d54fc4Saliguori #if 0 805f3d54fc4Saliguori printf("guessed geometry: LCHS=%d %d %d\n", 806f3d54fc4Saliguori cylinders, heads, sectors); 807f3d54fc4Saliguori #endif 808f3d54fc4Saliguori return 0; 809f3d54fc4Saliguori } 810f3d54fc4Saliguori } 811f3d54fc4Saliguori return -1; 812f3d54fc4Saliguori } 813f3d54fc4Saliguori 814f3d54fc4Saliguori void bdrv_guess_geometry(BlockDriverState *bs, int *pcyls, int *pheads, int *psecs) 815f3d54fc4Saliguori { 816f3d54fc4Saliguori int translation, lba_detected = 0; 817f3d54fc4Saliguori int cylinders, heads, secs; 818a38131b6Sblueswir1 uint64_t nb_sectors; 819f3d54fc4Saliguori 820f3d54fc4Saliguori /* if a geometry hint is available, use it */ 821f3d54fc4Saliguori bdrv_get_geometry(bs, &nb_sectors); 822f3d54fc4Saliguori bdrv_get_geometry_hint(bs, &cylinders, &heads, &secs); 823f3d54fc4Saliguori translation = bdrv_get_translation_hint(bs); 824f3d54fc4Saliguori if (cylinders != 0) { 825f3d54fc4Saliguori *pcyls = cylinders; 826f3d54fc4Saliguori *pheads = heads; 827f3d54fc4Saliguori *psecs = secs; 828f3d54fc4Saliguori } else { 829f3d54fc4Saliguori if (guess_disk_lchs(bs, &cylinders, &heads, &secs) == 0) { 830f3d54fc4Saliguori if (heads > 16) { 831f3d54fc4Saliguori /* if heads > 16, it means that a BIOS LBA 832f3d54fc4Saliguori translation was active, so the default 833f3d54fc4Saliguori hardware geometry is OK */ 834f3d54fc4Saliguori lba_detected = 1; 835f3d54fc4Saliguori goto default_geometry; 836f3d54fc4Saliguori } else { 837f3d54fc4Saliguori *pcyls = cylinders; 838f3d54fc4Saliguori *pheads = heads; 839f3d54fc4Saliguori *psecs = secs; 840f3d54fc4Saliguori /* disable any translation to be in sync with 841f3d54fc4Saliguori the logical geometry */ 842f3d54fc4Saliguori if (translation == BIOS_ATA_TRANSLATION_AUTO) { 843f3d54fc4Saliguori bdrv_set_translation_hint(bs, 844f3d54fc4Saliguori BIOS_ATA_TRANSLATION_NONE); 845f3d54fc4Saliguori } 846f3d54fc4Saliguori } 847f3d54fc4Saliguori } else { 848f3d54fc4Saliguori default_geometry: 849f3d54fc4Saliguori /* if no geometry, use a standard physical disk geometry */ 850f3d54fc4Saliguori cylinders = nb_sectors / (16 * 63); 851f3d54fc4Saliguori 852f3d54fc4Saliguori if (cylinders > 16383) 853f3d54fc4Saliguori cylinders = 16383; 854f3d54fc4Saliguori else if (cylinders < 2) 855f3d54fc4Saliguori cylinders = 2; 856f3d54fc4Saliguori *pcyls = cylinders; 857f3d54fc4Saliguori *pheads = 16; 858f3d54fc4Saliguori *psecs = 63; 859f3d54fc4Saliguori if ((lba_detected == 1) && (translation == BIOS_ATA_TRANSLATION_AUTO)) { 860f3d54fc4Saliguori if ((*pcyls * *pheads) <= 131072) { 861f3d54fc4Saliguori bdrv_set_translation_hint(bs, 862f3d54fc4Saliguori BIOS_ATA_TRANSLATION_LARGE); 863f3d54fc4Saliguori } else { 864f3d54fc4Saliguori bdrv_set_translation_hint(bs, 865f3d54fc4Saliguori BIOS_ATA_TRANSLATION_LBA); 866f3d54fc4Saliguori } 867f3d54fc4Saliguori } 868f3d54fc4Saliguori } 869f3d54fc4Saliguori bdrv_set_geometry_hint(bs, *pcyls, *pheads, *psecs); 870f3d54fc4Saliguori } 871f3d54fc4Saliguori } 872f3d54fc4Saliguori 873b338082bSbellard void bdrv_set_geometry_hint(BlockDriverState *bs, 874b338082bSbellard int cyls, int heads, int secs) 875b338082bSbellard { 876b338082bSbellard bs->cyls = cyls; 877b338082bSbellard bs->heads = heads; 878b338082bSbellard bs->secs = secs; 879b338082bSbellard } 880b338082bSbellard 881b338082bSbellard void bdrv_set_type_hint(BlockDriverState *bs, int type) 882b338082bSbellard { 883b338082bSbellard bs->type = type; 884b338082bSbellard bs->removable = ((type == BDRV_TYPE_CDROM || 885b338082bSbellard type == BDRV_TYPE_FLOPPY)); 886b338082bSbellard } 887b338082bSbellard 88846d4767dSbellard void bdrv_set_translation_hint(BlockDriverState *bs, int translation) 88946d4767dSbellard { 89046d4767dSbellard bs->translation = translation; 89146d4767dSbellard } 89246d4767dSbellard 893b338082bSbellard void bdrv_get_geometry_hint(BlockDriverState *bs, 894b338082bSbellard int *pcyls, int *pheads, int *psecs) 895b338082bSbellard { 896b338082bSbellard *pcyls = bs->cyls; 897b338082bSbellard *pheads = bs->heads; 898b338082bSbellard *psecs = bs->secs; 899b338082bSbellard } 900b338082bSbellard 901b338082bSbellard int bdrv_get_type_hint(BlockDriverState *bs) 902b338082bSbellard { 903b338082bSbellard return bs->type; 904b338082bSbellard } 905b338082bSbellard 90646d4767dSbellard int bdrv_get_translation_hint(BlockDriverState *bs) 90746d4767dSbellard { 90846d4767dSbellard return bs->translation; 90946d4767dSbellard } 91046d4767dSbellard 911b338082bSbellard int bdrv_is_removable(BlockDriverState *bs) 912b338082bSbellard { 913b338082bSbellard return bs->removable; 914b338082bSbellard } 915b338082bSbellard 916b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs) 917b338082bSbellard { 918b338082bSbellard return bs->read_only; 919b338082bSbellard } 920b338082bSbellard 921985a03b0Sths int bdrv_is_sg(BlockDriverState *bs) 922985a03b0Sths { 923985a03b0Sths return bs->sg; 924985a03b0Sths } 925985a03b0Sths 92619cb3738Sbellard /* XXX: no longer used */ 927b338082bSbellard void bdrv_set_change_cb(BlockDriverState *bs, 928b338082bSbellard void (*change_cb)(void *opaque), void *opaque) 929b338082bSbellard { 930b338082bSbellard bs->change_cb = change_cb; 931b338082bSbellard bs->change_opaque = opaque; 932b338082bSbellard } 933b338082bSbellard 934ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs) 935ea2384d3Sbellard { 936ea2384d3Sbellard if (bs->backing_hd && bs->backing_hd->encrypted) 937ea2384d3Sbellard return 1; 938ea2384d3Sbellard return bs->encrypted; 939ea2384d3Sbellard } 940ea2384d3Sbellard 941c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs) 942c0f4ce77Saliguori { 943c0f4ce77Saliguori BlockDriverState *backing_hd = bs->backing_hd; 944c0f4ce77Saliguori 945c0f4ce77Saliguori if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key) 946c0f4ce77Saliguori return 1; 947c0f4ce77Saliguori return (bs->encrypted && !bs->valid_key); 948c0f4ce77Saliguori } 949c0f4ce77Saliguori 950ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key) 951ea2384d3Sbellard { 952ea2384d3Sbellard int ret; 953ea2384d3Sbellard if (bs->backing_hd && bs->backing_hd->encrypted) { 954ea2384d3Sbellard ret = bdrv_set_key(bs->backing_hd, key); 955ea2384d3Sbellard if (ret < 0) 956ea2384d3Sbellard return ret; 957ea2384d3Sbellard if (!bs->encrypted) 958ea2384d3Sbellard return 0; 959ea2384d3Sbellard } 960ea2384d3Sbellard if (!bs->encrypted || !bs->drv || !bs->drv->bdrv_set_key) 961ea2384d3Sbellard return -1; 962c0f4ce77Saliguori ret = bs->drv->bdrv_set_key(bs, key); 963bb5fc20fSaliguori if (ret < 0) { 964bb5fc20fSaliguori bs->valid_key = 0; 965bb5fc20fSaliguori } else if (!bs->valid_key) { 966bb5fc20fSaliguori bs->valid_key = 1; 967bb5fc20fSaliguori /* call the change callback now, we skipped it on open */ 968bb5fc20fSaliguori bs->media_changed = 1; 969bb5fc20fSaliguori if (bs->change_cb) 970bb5fc20fSaliguori bs->change_cb(bs->change_opaque); 971bb5fc20fSaliguori } 972c0f4ce77Saliguori return ret; 973ea2384d3Sbellard } 974ea2384d3Sbellard 975ea2384d3Sbellard void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size) 976ea2384d3Sbellard { 97719cb3738Sbellard if (!bs->drv) { 978ea2384d3Sbellard buf[0] = '\0'; 979ea2384d3Sbellard } else { 980ea2384d3Sbellard pstrcpy(buf, buf_size, bs->drv->format_name); 981ea2384d3Sbellard } 982ea2384d3Sbellard } 983ea2384d3Sbellard 984ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name), 985ea2384d3Sbellard void *opaque) 986ea2384d3Sbellard { 987ea2384d3Sbellard BlockDriver *drv; 988ea2384d3Sbellard 989ea2384d3Sbellard for (drv = first_drv; drv != NULL; drv = drv->next) { 990ea2384d3Sbellard it(opaque, drv->format_name); 991ea2384d3Sbellard } 992ea2384d3Sbellard } 993ea2384d3Sbellard 994b338082bSbellard BlockDriverState *bdrv_find(const char *name) 995b338082bSbellard { 996b338082bSbellard BlockDriverState *bs; 997b338082bSbellard 998b338082bSbellard for (bs = bdrv_first; bs != NULL; bs = bs->next) { 999b338082bSbellard if (!strcmp(name, bs->device_name)) 1000b338082bSbellard return bs; 1001b338082bSbellard } 1002b338082bSbellard return NULL; 1003b338082bSbellard } 1004b338082bSbellard 100551de9760Saliguori void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque) 100681d0912dSbellard { 100781d0912dSbellard BlockDriverState *bs; 100881d0912dSbellard 100981d0912dSbellard for (bs = bdrv_first; bs != NULL; bs = bs->next) { 101051de9760Saliguori it(opaque, bs); 101181d0912dSbellard } 101281d0912dSbellard } 101381d0912dSbellard 1014ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs) 1015ea2384d3Sbellard { 1016ea2384d3Sbellard return bs->device_name; 1017ea2384d3Sbellard } 1018ea2384d3Sbellard 10197a6cba61Spbrook void bdrv_flush(BlockDriverState *bs) 10207a6cba61Spbrook { 1021081501daSaliguori if (!bs->drv) 1022081501daSaliguori return; 10237a6cba61Spbrook if (bs->drv->bdrv_flush) 10247a6cba61Spbrook bs->drv->bdrv_flush(bs); 10257a6cba61Spbrook if (bs->backing_hd) 10267a6cba61Spbrook bdrv_flush(bs->backing_hd); 10277a6cba61Spbrook } 10287a6cba61Spbrook 1029c6ca28d6Saliguori void bdrv_flush_all(void) 1030c6ca28d6Saliguori { 1031c6ca28d6Saliguori BlockDriverState *bs; 1032c6ca28d6Saliguori 1033c6ca28d6Saliguori for (bs = bdrv_first; bs != NULL; bs = bs->next) 1034c6ca28d6Saliguori if (bs->drv && !bdrv_is_read_only(bs) && 1035c6ca28d6Saliguori (!bdrv_is_removable(bs) || bdrv_is_inserted(bs))) 1036c6ca28d6Saliguori bdrv_flush(bs); 1037c6ca28d6Saliguori } 1038c6ca28d6Saliguori 1039f58c7b35Sths /* 1040f58c7b35Sths * Returns true iff the specified sector is present in the disk image. Drivers 1041f58c7b35Sths * not implementing the functionality are assumed to not support backing files, 1042f58c7b35Sths * hence all their sectors are reported as allocated. 1043f58c7b35Sths * 1044f58c7b35Sths * 'pnum' is set to the number of sectors (including and immediately following 1045f58c7b35Sths * the specified sector) that are known to be in the same 1046f58c7b35Sths * allocated/unallocated state. 1047f58c7b35Sths * 1048f58c7b35Sths * 'nb_sectors' is the max value 'pnum' should be set to. 1049f58c7b35Sths */ 1050f58c7b35Sths int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors, 1051f58c7b35Sths int *pnum) 1052f58c7b35Sths { 1053f58c7b35Sths int64_t n; 1054f58c7b35Sths if (!bs->drv->bdrv_is_allocated) { 1055f58c7b35Sths if (sector_num >= bs->total_sectors) { 1056f58c7b35Sths *pnum = 0; 1057f58c7b35Sths return 0; 1058f58c7b35Sths } 1059f58c7b35Sths n = bs->total_sectors - sector_num; 1060f58c7b35Sths *pnum = (n < nb_sectors) ? (n) : (nb_sectors); 1061f58c7b35Sths return 1; 1062f58c7b35Sths } 1063f58c7b35Sths return bs->drv->bdrv_is_allocated(bs, sector_num, nb_sectors, pnum); 1064f58c7b35Sths } 1065f58c7b35Sths 1066376253ecSaliguori void bdrv_info(Monitor *mon) 1067b338082bSbellard { 1068b338082bSbellard BlockDriverState *bs; 1069b338082bSbellard 1070b338082bSbellard for (bs = bdrv_first; bs != NULL; bs = bs->next) { 1071376253ecSaliguori monitor_printf(mon, "%s:", bs->device_name); 1072376253ecSaliguori monitor_printf(mon, " type="); 1073b338082bSbellard switch(bs->type) { 1074b338082bSbellard case BDRV_TYPE_HD: 1075376253ecSaliguori monitor_printf(mon, "hd"); 1076b338082bSbellard break; 1077b338082bSbellard case BDRV_TYPE_CDROM: 1078376253ecSaliguori monitor_printf(mon, "cdrom"); 1079b338082bSbellard break; 1080b338082bSbellard case BDRV_TYPE_FLOPPY: 1081376253ecSaliguori monitor_printf(mon, "floppy"); 1082b338082bSbellard break; 1083b338082bSbellard } 1084376253ecSaliguori monitor_printf(mon, " removable=%d", bs->removable); 1085b338082bSbellard if (bs->removable) { 1086376253ecSaliguori monitor_printf(mon, " locked=%d", bs->locked); 1087b338082bSbellard } 108819cb3738Sbellard if (bs->drv) { 1089376253ecSaliguori monitor_printf(mon, " file="); 1090376253ecSaliguori monitor_print_filename(mon, bs->filename); 1091fef30743Sths if (bs->backing_file[0] != '\0') { 1092376253ecSaliguori monitor_printf(mon, " backing_file="); 1093376253ecSaliguori monitor_print_filename(mon, bs->backing_file); 1094fef30743Sths } 1095376253ecSaliguori monitor_printf(mon, " ro=%d", bs->read_only); 1096376253ecSaliguori monitor_printf(mon, " drv=%s", bs->drv->format_name); 1097376253ecSaliguori monitor_printf(mon, " encrypted=%d", bdrv_is_encrypted(bs)); 1098b338082bSbellard } else { 1099376253ecSaliguori monitor_printf(mon, " [not inserted]"); 1100b338082bSbellard } 1101376253ecSaliguori monitor_printf(mon, "\n"); 1102b338082bSbellard } 1103b338082bSbellard } 1104a36e69ddSths 1105a36e69ddSths /* The "info blockstats" command. */ 1106376253ecSaliguori void bdrv_info_stats(Monitor *mon) 1107a36e69ddSths { 1108a36e69ddSths BlockDriverState *bs; 1109a36e69ddSths 1110a36e69ddSths for (bs = bdrv_first; bs != NULL; bs = bs->next) { 1111376253ecSaliguori monitor_printf(mon, "%s:" 1112a36e69ddSths " rd_bytes=%" PRIu64 1113a36e69ddSths " wr_bytes=%" PRIu64 1114a36e69ddSths " rd_operations=%" PRIu64 1115a36e69ddSths " wr_operations=%" PRIu64 1116ebf53fcdSaliguori "\n", 1117a36e69ddSths bs->device_name, 1118a36e69ddSths bs->rd_bytes, bs->wr_bytes, 1119a36e69ddSths bs->rd_ops, bs->wr_ops); 1120a36e69ddSths } 1121a36e69ddSths } 1122ea2384d3Sbellard 1123045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs) 1124045df330Saliguori { 1125045df330Saliguori if (bs->backing_hd && bs->backing_hd->encrypted) 1126045df330Saliguori return bs->backing_file; 1127045df330Saliguori else if (bs->encrypted) 1128045df330Saliguori return bs->filename; 1129045df330Saliguori else 1130045df330Saliguori return NULL; 1131045df330Saliguori } 1132045df330Saliguori 113383f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs, 113483f64091Sbellard char *filename, int filename_size) 113583f64091Sbellard { 113683f64091Sbellard if (!bs->backing_hd) { 113783f64091Sbellard pstrcpy(filename, filename_size, ""); 113883f64091Sbellard } else { 113983f64091Sbellard pstrcpy(filename, filename_size, bs->backing_file); 114083f64091Sbellard } 114183f64091Sbellard } 114283f64091Sbellard 1143faea38e7Sbellard int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num, 1144faea38e7Sbellard const uint8_t *buf, int nb_sectors) 1145faea38e7Sbellard { 1146faea38e7Sbellard BlockDriver *drv = bs->drv; 1147faea38e7Sbellard if (!drv) 114819cb3738Sbellard return -ENOMEDIUM; 1149faea38e7Sbellard if (!drv->bdrv_write_compressed) 1150faea38e7Sbellard return -ENOTSUP; 1151faea38e7Sbellard return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors); 1152faea38e7Sbellard } 1153faea38e7Sbellard 1154faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi) 1155faea38e7Sbellard { 1156faea38e7Sbellard BlockDriver *drv = bs->drv; 1157faea38e7Sbellard if (!drv) 115819cb3738Sbellard return -ENOMEDIUM; 1159faea38e7Sbellard if (!drv->bdrv_get_info) 1160faea38e7Sbellard return -ENOTSUP; 1161faea38e7Sbellard memset(bdi, 0, sizeof(*bdi)); 1162faea38e7Sbellard return drv->bdrv_get_info(bs, bdi); 1163faea38e7Sbellard } 1164faea38e7Sbellard 1165178e08a5Saliguori int bdrv_put_buffer(BlockDriverState *bs, const uint8_t *buf, int64_t pos, int size) 1166178e08a5Saliguori { 1167178e08a5Saliguori BlockDriver *drv = bs->drv; 1168178e08a5Saliguori if (!drv) 1169178e08a5Saliguori return -ENOMEDIUM; 1170178e08a5Saliguori if (!drv->bdrv_put_buffer) 1171178e08a5Saliguori return -ENOTSUP; 1172178e08a5Saliguori return drv->bdrv_put_buffer(bs, buf, pos, size); 1173178e08a5Saliguori } 1174178e08a5Saliguori 1175178e08a5Saliguori int bdrv_get_buffer(BlockDriverState *bs, uint8_t *buf, int64_t pos, int size) 1176178e08a5Saliguori { 1177178e08a5Saliguori BlockDriver *drv = bs->drv; 1178178e08a5Saliguori if (!drv) 1179178e08a5Saliguori return -ENOMEDIUM; 1180178e08a5Saliguori if (!drv->bdrv_get_buffer) 1181178e08a5Saliguori return -ENOTSUP; 1182178e08a5Saliguori return drv->bdrv_get_buffer(bs, buf, pos, size); 1183178e08a5Saliguori } 1184178e08a5Saliguori 1185faea38e7Sbellard /**************************************************************/ 1186faea38e7Sbellard /* handling of snapshots */ 1187faea38e7Sbellard 1188faea38e7Sbellard int bdrv_snapshot_create(BlockDriverState *bs, 1189faea38e7Sbellard QEMUSnapshotInfo *sn_info) 1190faea38e7Sbellard { 1191faea38e7Sbellard BlockDriver *drv = bs->drv; 1192faea38e7Sbellard if (!drv) 119319cb3738Sbellard return -ENOMEDIUM; 1194faea38e7Sbellard if (!drv->bdrv_snapshot_create) 1195faea38e7Sbellard return -ENOTSUP; 1196faea38e7Sbellard return drv->bdrv_snapshot_create(bs, sn_info); 1197faea38e7Sbellard } 1198faea38e7Sbellard 1199faea38e7Sbellard int bdrv_snapshot_goto(BlockDriverState *bs, 1200faea38e7Sbellard const char *snapshot_id) 1201faea38e7Sbellard { 1202faea38e7Sbellard BlockDriver *drv = bs->drv; 1203faea38e7Sbellard if (!drv) 120419cb3738Sbellard return -ENOMEDIUM; 1205faea38e7Sbellard if (!drv->bdrv_snapshot_goto) 1206faea38e7Sbellard return -ENOTSUP; 1207faea38e7Sbellard return drv->bdrv_snapshot_goto(bs, snapshot_id); 1208faea38e7Sbellard } 1209faea38e7Sbellard 1210faea38e7Sbellard int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id) 1211faea38e7Sbellard { 1212faea38e7Sbellard BlockDriver *drv = bs->drv; 1213faea38e7Sbellard if (!drv) 121419cb3738Sbellard return -ENOMEDIUM; 1215faea38e7Sbellard if (!drv->bdrv_snapshot_delete) 1216faea38e7Sbellard return -ENOTSUP; 1217faea38e7Sbellard return drv->bdrv_snapshot_delete(bs, snapshot_id); 1218faea38e7Sbellard } 1219faea38e7Sbellard 1220faea38e7Sbellard int bdrv_snapshot_list(BlockDriverState *bs, 1221faea38e7Sbellard QEMUSnapshotInfo **psn_info) 1222faea38e7Sbellard { 1223faea38e7Sbellard BlockDriver *drv = bs->drv; 1224faea38e7Sbellard if (!drv) 122519cb3738Sbellard return -ENOMEDIUM; 1226faea38e7Sbellard if (!drv->bdrv_snapshot_list) 1227faea38e7Sbellard return -ENOTSUP; 1228faea38e7Sbellard return drv->bdrv_snapshot_list(bs, psn_info); 1229faea38e7Sbellard } 1230faea38e7Sbellard 1231faea38e7Sbellard #define NB_SUFFIXES 4 1232faea38e7Sbellard 1233faea38e7Sbellard char *get_human_readable_size(char *buf, int buf_size, int64_t size) 1234faea38e7Sbellard { 1235faea38e7Sbellard static const char suffixes[NB_SUFFIXES] = "KMGT"; 1236faea38e7Sbellard int64_t base; 1237faea38e7Sbellard int i; 1238faea38e7Sbellard 1239faea38e7Sbellard if (size <= 999) { 1240faea38e7Sbellard snprintf(buf, buf_size, "%" PRId64, size); 1241faea38e7Sbellard } else { 1242faea38e7Sbellard base = 1024; 1243faea38e7Sbellard for(i = 0; i < NB_SUFFIXES; i++) { 1244faea38e7Sbellard if (size < (10 * base)) { 1245faea38e7Sbellard snprintf(buf, buf_size, "%0.1f%c", 1246faea38e7Sbellard (double)size / base, 1247faea38e7Sbellard suffixes[i]); 1248faea38e7Sbellard break; 1249faea38e7Sbellard } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) { 1250faea38e7Sbellard snprintf(buf, buf_size, "%" PRId64 "%c", 1251faea38e7Sbellard ((size + (base >> 1)) / base), 1252faea38e7Sbellard suffixes[i]); 1253faea38e7Sbellard break; 1254faea38e7Sbellard } 1255faea38e7Sbellard base = base * 1024; 1256faea38e7Sbellard } 1257faea38e7Sbellard } 1258faea38e7Sbellard return buf; 1259faea38e7Sbellard } 1260faea38e7Sbellard 1261faea38e7Sbellard char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn) 1262faea38e7Sbellard { 1263faea38e7Sbellard char buf1[128], date_buf[128], clock_buf[128]; 12643b9f94e1Sbellard #ifdef _WIN32 12653b9f94e1Sbellard struct tm *ptm; 12663b9f94e1Sbellard #else 1267faea38e7Sbellard struct tm tm; 12683b9f94e1Sbellard #endif 1269faea38e7Sbellard time_t ti; 1270faea38e7Sbellard int64_t secs; 1271faea38e7Sbellard 1272faea38e7Sbellard if (!sn) { 1273faea38e7Sbellard snprintf(buf, buf_size, 1274faea38e7Sbellard "%-10s%-20s%7s%20s%15s", 1275faea38e7Sbellard "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK"); 1276faea38e7Sbellard } else { 1277faea38e7Sbellard ti = sn->date_sec; 12783b9f94e1Sbellard #ifdef _WIN32 12793b9f94e1Sbellard ptm = localtime(&ti); 12803b9f94e1Sbellard strftime(date_buf, sizeof(date_buf), 12813b9f94e1Sbellard "%Y-%m-%d %H:%M:%S", ptm); 12823b9f94e1Sbellard #else 1283faea38e7Sbellard localtime_r(&ti, &tm); 1284faea38e7Sbellard strftime(date_buf, sizeof(date_buf), 1285faea38e7Sbellard "%Y-%m-%d %H:%M:%S", &tm); 12863b9f94e1Sbellard #endif 1287faea38e7Sbellard secs = sn->vm_clock_nsec / 1000000000; 1288faea38e7Sbellard snprintf(clock_buf, sizeof(clock_buf), 1289faea38e7Sbellard "%02d:%02d:%02d.%03d", 1290faea38e7Sbellard (int)(secs / 3600), 1291faea38e7Sbellard (int)((secs / 60) % 60), 1292faea38e7Sbellard (int)(secs % 60), 1293faea38e7Sbellard (int)((sn->vm_clock_nsec / 1000000) % 1000)); 1294faea38e7Sbellard snprintf(buf, buf_size, 1295faea38e7Sbellard "%-10s%-20s%7s%20s%15s", 1296faea38e7Sbellard sn->id_str, sn->name, 1297faea38e7Sbellard get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size), 1298faea38e7Sbellard date_buf, 1299faea38e7Sbellard clock_buf); 1300faea38e7Sbellard } 1301faea38e7Sbellard return buf; 1302faea38e7Sbellard } 1303faea38e7Sbellard 130483f64091Sbellard 1305ea2384d3Sbellard /**************************************************************/ 130683f64091Sbellard /* async I/Os */ 1307ea2384d3Sbellard 13083b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num, 1309f141eafeSaliguori QEMUIOVector *qiov, int nb_sectors, 131083f64091Sbellard BlockDriverCompletionFunc *cb, void *opaque) 1311ea2384d3Sbellard { 131283f64091Sbellard BlockDriver *drv = bs->drv; 1313a36e69ddSths BlockDriverAIOCB *ret; 1314ea2384d3Sbellard 131519cb3738Sbellard if (!drv) 1316ce1a14dcSpbrook return NULL; 131771d0770cSaliguori if (bdrv_check_request(bs, sector_num, nb_sectors)) 131871d0770cSaliguori return NULL; 131983f64091Sbellard 1320f141eafeSaliguori ret = drv->bdrv_aio_readv(bs, sector_num, qiov, nb_sectors, 1321f141eafeSaliguori cb, opaque); 1322a36e69ddSths 1323a36e69ddSths if (ret) { 1324a36e69ddSths /* Update stats even though technically transfer has not happened. */ 1325a36e69ddSths bs->rd_bytes += (unsigned) nb_sectors * SECTOR_SIZE; 1326a36e69ddSths bs->rd_ops ++; 1327a36e69ddSths } 1328a36e69ddSths 1329a36e69ddSths return ret; 133083f64091Sbellard } 133183f64091Sbellard 1332f141eafeSaliguori BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num, 1333f141eafeSaliguori QEMUIOVector *qiov, int nb_sectors, 133483f64091Sbellard BlockDriverCompletionFunc *cb, void *opaque) 13357674e7bfSbellard { 133683f64091Sbellard BlockDriver *drv = bs->drv; 1337a36e69ddSths BlockDriverAIOCB *ret; 133883f64091Sbellard 133919cb3738Sbellard if (!drv) 1340ce1a14dcSpbrook return NULL; 134183f64091Sbellard if (bs->read_only) 1342ce1a14dcSpbrook return NULL; 134371d0770cSaliguori if (bdrv_check_request(bs, sector_num, nb_sectors)) 134471d0770cSaliguori return NULL; 134583f64091Sbellard 1346f141eafeSaliguori ret = drv->bdrv_aio_writev(bs, sector_num, qiov, nb_sectors, 1347f141eafeSaliguori cb, opaque); 1348a36e69ddSths 1349a36e69ddSths if (ret) { 1350a36e69ddSths /* Update stats even though technically transfer has not happened. */ 1351a36e69ddSths bs->wr_bytes += (unsigned) nb_sectors * SECTOR_SIZE; 1352a36e69ddSths bs->wr_ops ++; 1353a36e69ddSths } 1354a36e69ddSths 1355a36e69ddSths return ret; 135683f64091Sbellard } 135783f64091Sbellard 135883f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb) 135983f64091Sbellard { 13606bbff9a0Saliguori acb->pool->cancel(acb); 136183f64091Sbellard } 136283f64091Sbellard 136383f64091Sbellard 136483f64091Sbellard /**************************************************************/ 136583f64091Sbellard /* async block device emulation */ 136683f64091Sbellard 136783f64091Sbellard static void bdrv_aio_bh_cb(void *opaque) 1368beac80cdSbellard { 1369ce1a14dcSpbrook BlockDriverAIOCBSync *acb = opaque; 1370f141eafeSaliguori 1371f141eafeSaliguori if (!acb->is_write) 1372f141eafeSaliguori qemu_iovec_from_buffer(acb->qiov, acb->bounce, acb->qiov->size); 1373ceb42de8Saliguori qemu_vfree(acb->bounce); 1374ce1a14dcSpbrook acb->common.cb(acb->common.opaque, acb->ret); 1375f141eafeSaliguori 1376ce1a14dcSpbrook qemu_aio_release(acb); 1377beac80cdSbellard } 1378beac80cdSbellard 1379f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs, 1380f141eafeSaliguori int64_t sector_num, 1381f141eafeSaliguori QEMUIOVector *qiov, 1382f141eafeSaliguori int nb_sectors, 1383f141eafeSaliguori BlockDriverCompletionFunc *cb, 1384f141eafeSaliguori void *opaque, 1385f141eafeSaliguori int is_write) 1386f141eafeSaliguori 1387ea2384d3Sbellard { 1388ce1a14dcSpbrook BlockDriverAIOCBSync *acb; 138983f64091Sbellard 1390ce1a14dcSpbrook acb = qemu_aio_get(bs, cb, opaque); 1391f141eafeSaliguori acb->is_write = is_write; 1392f141eafeSaliguori acb->qiov = qiov; 1393f141eafeSaliguori acb->bounce = qemu_memalign(512, qiov->size); 1394f141eafeSaliguori 1395ce1a14dcSpbrook if (!acb->bh) 1396ce1a14dcSpbrook acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb); 1397f141eafeSaliguori 1398f141eafeSaliguori if (is_write) { 1399f141eafeSaliguori qemu_iovec_to_buffer(acb->qiov, acb->bounce); 1400f141eafeSaliguori acb->ret = bdrv_write(bs, sector_num, acb->bounce, nb_sectors); 1401f141eafeSaliguori } else { 1402f141eafeSaliguori acb->ret = bdrv_read(bs, sector_num, acb->bounce, nb_sectors); 1403f141eafeSaliguori } 1404f141eafeSaliguori 1405ce1a14dcSpbrook qemu_bh_schedule(acb->bh); 1406f141eafeSaliguori 1407ce1a14dcSpbrook return &acb->common; 14087a6cba61Spbrook } 14097a6cba61Spbrook 1410f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs, 1411f141eafeSaliguori int64_t sector_num, QEMUIOVector *qiov, int nb_sectors, 1412ce1a14dcSpbrook BlockDriverCompletionFunc *cb, void *opaque) 141383f64091Sbellard { 1414f141eafeSaliguori return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0); 141583f64091Sbellard } 141683f64091Sbellard 1417f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs, 1418f141eafeSaliguori int64_t sector_num, QEMUIOVector *qiov, int nb_sectors, 1419f141eafeSaliguori BlockDriverCompletionFunc *cb, void *opaque) 1420f141eafeSaliguori { 1421f141eafeSaliguori return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1); 1422f141eafeSaliguori } 1423f141eafeSaliguori 1424f141eafeSaliguori 1425ce1a14dcSpbrook static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb) 142683f64091Sbellard { 1427ce1a14dcSpbrook BlockDriverAIOCBSync *acb = (BlockDriverAIOCBSync *)blockacb; 1428ce1a14dcSpbrook qemu_bh_cancel(acb->bh); 1429ce1a14dcSpbrook qemu_aio_release(acb); 143083f64091Sbellard } 143183f64091Sbellard 143283f64091Sbellard /**************************************************************/ 143383f64091Sbellard /* sync block device emulation */ 143483f64091Sbellard 143583f64091Sbellard static void bdrv_rw_em_cb(void *opaque, int ret) 143683f64091Sbellard { 143783f64091Sbellard *(int *)opaque = ret; 143883f64091Sbellard } 143983f64091Sbellard 144083f64091Sbellard #define NOT_DONE 0x7fffffff 144183f64091Sbellard 144283f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num, 144383f64091Sbellard uint8_t *buf, int nb_sectors) 144483f64091Sbellard { 1445ce1a14dcSpbrook int async_ret; 1446ce1a14dcSpbrook BlockDriverAIOCB *acb; 1447f141eafeSaliguori struct iovec iov; 1448f141eafeSaliguori QEMUIOVector qiov; 144983f64091Sbellard 145083f64091Sbellard async_ret = NOT_DONE; 14513f4cb3d3Sblueswir1 iov.iov_base = (void *)buf; 1452f141eafeSaliguori iov.iov_len = nb_sectors * 512; 1453f141eafeSaliguori qemu_iovec_init_external(&qiov, &iov, 1); 1454f141eafeSaliguori acb = bdrv_aio_readv(bs, sector_num, &qiov, nb_sectors, 145583f64091Sbellard bdrv_rw_em_cb, &async_ret); 1456baf35cb9Saliguori if (acb == NULL) 1457ce1a14dcSpbrook return -1; 1458baf35cb9Saliguori 145983f64091Sbellard while (async_ret == NOT_DONE) { 146083f64091Sbellard qemu_aio_wait(); 146183f64091Sbellard } 1462baf35cb9Saliguori 146383f64091Sbellard return async_ret; 146483f64091Sbellard } 146583f64091Sbellard 146683f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num, 146783f64091Sbellard const uint8_t *buf, int nb_sectors) 146883f64091Sbellard { 1469ce1a14dcSpbrook int async_ret; 1470ce1a14dcSpbrook BlockDriverAIOCB *acb; 1471f141eafeSaliguori struct iovec iov; 1472f141eafeSaliguori QEMUIOVector qiov; 147383f64091Sbellard 147483f64091Sbellard async_ret = NOT_DONE; 1475f141eafeSaliguori iov.iov_base = (void *)buf; 1476f141eafeSaliguori iov.iov_len = nb_sectors * 512; 1477f141eafeSaliguori qemu_iovec_init_external(&qiov, &iov, 1); 1478f141eafeSaliguori acb = bdrv_aio_writev(bs, sector_num, &qiov, nb_sectors, 147983f64091Sbellard bdrv_rw_em_cb, &async_ret); 1480baf35cb9Saliguori if (acb == NULL) 1481ce1a14dcSpbrook return -1; 148283f64091Sbellard while (async_ret == NOT_DONE) { 148383f64091Sbellard qemu_aio_wait(); 148483f64091Sbellard } 148583f64091Sbellard return async_ret; 148683f64091Sbellard } 1487ea2384d3Sbellard 1488ea2384d3Sbellard void bdrv_init(void) 1489ea2384d3Sbellard { 1490ea2384d3Sbellard bdrv_register(&bdrv_raw); 149119cb3738Sbellard bdrv_register(&bdrv_host_device); 1492ea2384d3Sbellard #ifndef _WIN32 1493ea2384d3Sbellard bdrv_register(&bdrv_cow); 1494ea2384d3Sbellard #endif 1495ea2384d3Sbellard bdrv_register(&bdrv_qcow); 1496ea2384d3Sbellard bdrv_register(&bdrv_vmdk); 14973c56521bSbellard bdrv_register(&bdrv_cloop); 1498585d0ed9Sbellard bdrv_register(&bdrv_dmg); 1499a8753c34Sbellard bdrv_register(&bdrv_bochs); 15006a0f9e82Sbellard bdrv_register(&bdrv_vpc); 1501712e7874Sbellard bdrv_register(&bdrv_vvfat); 1502faea38e7Sbellard bdrv_register(&bdrv_qcow2); 15036ada7453Sths bdrv_register(&bdrv_parallels); 150475818250Sths bdrv_register(&bdrv_nbd); 1505ea2384d3Sbellard } 1506ce1a14dcSpbrook 15076bbff9a0Saliguori void aio_pool_init(AIOPool *pool, int aiocb_size, 15086bbff9a0Saliguori void (*cancel)(BlockDriverAIOCB *acb)) 1509ce1a14dcSpbrook { 15106bbff9a0Saliguori pool->aiocb_size = aiocb_size; 15116bbff9a0Saliguori pool->cancel = cancel; 15126bbff9a0Saliguori pool->free_aiocb = NULL; 15136bbff9a0Saliguori } 15146bbff9a0Saliguori 15156bbff9a0Saliguori void *qemu_aio_get_pool(AIOPool *pool, BlockDriverState *bs, 15166bbff9a0Saliguori BlockDriverCompletionFunc *cb, void *opaque) 15176bbff9a0Saliguori { 1518ce1a14dcSpbrook BlockDriverAIOCB *acb; 1519ce1a14dcSpbrook 15206bbff9a0Saliguori if (pool->free_aiocb) { 15216bbff9a0Saliguori acb = pool->free_aiocb; 15226bbff9a0Saliguori pool->free_aiocb = acb->next; 1523ce1a14dcSpbrook } else { 15246bbff9a0Saliguori acb = qemu_mallocz(pool->aiocb_size); 15256bbff9a0Saliguori acb->pool = pool; 1526ce1a14dcSpbrook } 1527ce1a14dcSpbrook acb->bs = bs; 1528ce1a14dcSpbrook acb->cb = cb; 1529ce1a14dcSpbrook acb->opaque = opaque; 1530ce1a14dcSpbrook return acb; 1531ce1a14dcSpbrook } 1532ce1a14dcSpbrook 15336bbff9a0Saliguori void *qemu_aio_get(BlockDriverState *bs, BlockDriverCompletionFunc *cb, 15346bbff9a0Saliguori void *opaque) 15356bbff9a0Saliguori { 15366bbff9a0Saliguori return qemu_aio_get_pool(&bs->drv->aio_pool, bs, cb, opaque); 15376bbff9a0Saliguori } 15386bbff9a0Saliguori 1539ce1a14dcSpbrook void qemu_aio_release(void *p) 1540ce1a14dcSpbrook { 15416bbff9a0Saliguori BlockDriverAIOCB *acb = (BlockDriverAIOCB *)p; 15426bbff9a0Saliguori AIOPool *pool = acb->pool; 15436bbff9a0Saliguori acb->next = pool->free_aiocb; 15446bbff9a0Saliguori pool->free_aiocb = acb; 1545ce1a14dcSpbrook } 154619cb3738Sbellard 154719cb3738Sbellard /**************************************************************/ 154819cb3738Sbellard /* removable device support */ 154919cb3738Sbellard 155019cb3738Sbellard /** 155119cb3738Sbellard * Return TRUE if the media is present 155219cb3738Sbellard */ 155319cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs) 155419cb3738Sbellard { 155519cb3738Sbellard BlockDriver *drv = bs->drv; 155619cb3738Sbellard int ret; 155719cb3738Sbellard if (!drv) 155819cb3738Sbellard return 0; 155919cb3738Sbellard if (!drv->bdrv_is_inserted) 156019cb3738Sbellard return 1; 156119cb3738Sbellard ret = drv->bdrv_is_inserted(bs); 156219cb3738Sbellard return ret; 156319cb3738Sbellard } 156419cb3738Sbellard 156519cb3738Sbellard /** 156619cb3738Sbellard * Return TRUE if the media changed since the last call to this 156719cb3738Sbellard * function. It is currently only used for floppy disks 156819cb3738Sbellard */ 156919cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs) 157019cb3738Sbellard { 157119cb3738Sbellard BlockDriver *drv = bs->drv; 157219cb3738Sbellard int ret; 157319cb3738Sbellard 157419cb3738Sbellard if (!drv || !drv->bdrv_media_changed) 157519cb3738Sbellard ret = -ENOTSUP; 157619cb3738Sbellard else 157719cb3738Sbellard ret = drv->bdrv_media_changed(bs); 157819cb3738Sbellard if (ret == -ENOTSUP) 157919cb3738Sbellard ret = bs->media_changed; 158019cb3738Sbellard bs->media_changed = 0; 158119cb3738Sbellard return ret; 158219cb3738Sbellard } 158319cb3738Sbellard 158419cb3738Sbellard /** 158519cb3738Sbellard * If eject_flag is TRUE, eject the media. Otherwise, close the tray 158619cb3738Sbellard */ 158719cb3738Sbellard void bdrv_eject(BlockDriverState *bs, int eject_flag) 158819cb3738Sbellard { 158919cb3738Sbellard BlockDriver *drv = bs->drv; 159019cb3738Sbellard int ret; 159119cb3738Sbellard 159219cb3738Sbellard if (!drv || !drv->bdrv_eject) { 159319cb3738Sbellard ret = -ENOTSUP; 159419cb3738Sbellard } else { 159519cb3738Sbellard ret = drv->bdrv_eject(bs, eject_flag); 159619cb3738Sbellard } 159719cb3738Sbellard if (ret == -ENOTSUP) { 159819cb3738Sbellard if (eject_flag) 159919cb3738Sbellard bdrv_close(bs); 160019cb3738Sbellard } 160119cb3738Sbellard } 160219cb3738Sbellard 160319cb3738Sbellard int bdrv_is_locked(BlockDriverState *bs) 160419cb3738Sbellard { 160519cb3738Sbellard return bs->locked; 160619cb3738Sbellard } 160719cb3738Sbellard 160819cb3738Sbellard /** 160919cb3738Sbellard * Lock or unlock the media (if it is locked, the user won't be able 161019cb3738Sbellard * to eject it manually). 161119cb3738Sbellard */ 161219cb3738Sbellard void bdrv_set_locked(BlockDriverState *bs, int locked) 161319cb3738Sbellard { 161419cb3738Sbellard BlockDriver *drv = bs->drv; 161519cb3738Sbellard 161619cb3738Sbellard bs->locked = locked; 161719cb3738Sbellard if (drv && drv->bdrv_set_locked) { 161819cb3738Sbellard drv->bdrv_set_locked(bs, locked); 161919cb3738Sbellard } 162019cb3738Sbellard } 1621985a03b0Sths 1622985a03b0Sths /* needed for generic scsi interface */ 1623985a03b0Sths 1624985a03b0Sths int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf) 1625985a03b0Sths { 1626985a03b0Sths BlockDriver *drv = bs->drv; 1627985a03b0Sths 1628985a03b0Sths if (drv && drv->bdrv_ioctl) 1629985a03b0Sths return drv->bdrv_ioctl(bs, req, buf); 1630985a03b0Sths return -ENOTSUP; 1631985a03b0Sths } 16327d780669Saliguori 1633221f715dSaliguori BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs, 1634221f715dSaliguori unsigned long int req, void *buf, 16357d780669Saliguori BlockDriverCompletionFunc *cb, void *opaque) 16367d780669Saliguori { 1637221f715dSaliguori BlockDriver *drv = bs->drv; 16387d780669Saliguori 1639221f715dSaliguori if (drv && drv->bdrv_aio_ioctl) 1640221f715dSaliguori return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque); 1641221f715dSaliguori return NULL; 16427d780669Saliguori } 1643