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; 365e268ca52Saliguori /* buffer_alignment defaulted to 512, drivers can change this value */ 366e268ca52Saliguori bs->buffer_alignment = 512; 36733e3963eSbellard 36883f64091Sbellard if (flags & BDRV_O_SNAPSHOT) { 369ea2384d3Sbellard BlockDriverState *bs1; 370ea2384d3Sbellard int64_t total_size; 3717c96d46eSaliguori int is_protocol = 0; 37233e3963eSbellard 373ea2384d3Sbellard /* if snapshot, we create a temporary backing file and open it 374ea2384d3Sbellard instead of opening 'filename' directly */ 375ea2384d3Sbellard 376ea2384d3Sbellard /* if there is a backing file, use it */ 377ea2384d3Sbellard bs1 = bdrv_new(""); 3785eb45639Saliguori ret = bdrv_open2(bs1, filename, 0, drv); 37951d7c00cSaliguori if (ret < 0) { 380ea2384d3Sbellard bdrv_delete(bs1); 38151d7c00cSaliguori return ret; 382ea2384d3Sbellard } 38383f64091Sbellard total_size = bdrv_getlength(bs1) >> SECTOR_BITS; 3847c96d46eSaliguori 3857c96d46eSaliguori if (bs1->drv && bs1->drv->protocol_name) 3867c96d46eSaliguori is_protocol = 1; 3877c96d46eSaliguori 388ea2384d3Sbellard bdrv_delete(bs1); 389ea2384d3Sbellard 390ea2384d3Sbellard get_tmp_filename(tmp_filename, sizeof(tmp_filename)); 3917c96d46eSaliguori 3927c96d46eSaliguori /* Real path is meaningless for protocols */ 3937c96d46eSaliguori if (is_protocol) 3947c96d46eSaliguori snprintf(backing_filename, sizeof(backing_filename), 3957c96d46eSaliguori "%s", filename); 3967c96d46eSaliguori else 397a817d936Sbellard realpath(filename, backing_filename); 3987c96d46eSaliguori 3995eb45639Saliguori ret = bdrv_create2(&bdrv_qcow2, tmp_filename, 4005eb45639Saliguori total_size, backing_filename, 4015eb45639Saliguori (drv ? drv->format_name : NULL), 0); 40251d7c00cSaliguori if (ret < 0) { 40351d7c00cSaliguori return ret; 404ea2384d3Sbellard } 405ea2384d3Sbellard filename = tmp_filename; 4065eb45639Saliguori drv = &bdrv_qcow2; 407ea2384d3Sbellard bs->is_temporary = 1; 408ea2384d3Sbellard } 409ea2384d3Sbellard 410ea2384d3Sbellard pstrcpy(bs->filename, sizeof(bs->filename), filename); 41183f64091Sbellard if (flags & BDRV_O_FILE) { 41283f64091Sbellard drv = find_protocol(filename); 41351d7c00cSaliguori } else if (!drv) { 414ea2384d3Sbellard drv = find_image_format(filename); 415ea2384d3Sbellard } 41651d7c00cSaliguori if (!drv) { 41751d7c00cSaliguori ret = -ENOENT; 41851d7c00cSaliguori goto unlink_and_fail; 41983f64091Sbellard } 420ea2384d3Sbellard bs->drv = drv; 421ea2384d3Sbellard bs->opaque = qemu_mallocz(drv->instance_size); 42283f64091Sbellard /* Note: for compatibility, we open disk image files as RDWR, and 42383f64091Sbellard RDONLY as fallback */ 42483f64091Sbellard if (!(flags & BDRV_O_FILE)) 4259f7965c7Saliguori open_flags = BDRV_O_RDWR | (flags & BDRV_O_CACHE_MASK); 42683f64091Sbellard else 42783f64091Sbellard open_flags = flags & ~(BDRV_O_FILE | BDRV_O_SNAPSHOT); 42883f64091Sbellard ret = drv->bdrv_open(bs, filename, open_flags); 429a0a83536Saurel32 if ((ret == -EACCES || ret == -EPERM) && !(flags & BDRV_O_FILE)) { 4309f7965c7Saliguori ret = drv->bdrv_open(bs, filename, open_flags & ~BDRV_O_RDWR); 43183f64091Sbellard bs->read_only = 1; 43283f64091Sbellard } 433ea2384d3Sbellard if (ret < 0) { 434ea2384d3Sbellard qemu_free(bs->opaque); 4356b21b973Sbellard bs->opaque = NULL; 4366b21b973Sbellard bs->drv = NULL; 43751d7c00cSaliguori unlink_and_fail: 43851d7c00cSaliguori if (bs->is_temporary) 43951d7c00cSaliguori unlink(filename); 44083f64091Sbellard return ret; 441ea2384d3Sbellard } 442d15a771dSbellard if (drv->bdrv_getlength) { 443d15a771dSbellard bs->total_sectors = bdrv_getlength(bs) >> SECTOR_BITS; 444d15a771dSbellard } 445ea2384d3Sbellard #ifndef _WIN32 446ea2384d3Sbellard if (bs->is_temporary) { 447ea2384d3Sbellard unlink(filename); 44833e3963eSbellard } 44967b915a5Sbellard #endif 45083f64091Sbellard if (bs->backing_file[0] != '\0') { 451ea2384d3Sbellard /* if there is a backing file, use it */ 4525eb45639Saliguori BlockDriver *back_drv = NULL; 453ea2384d3Sbellard bs->backing_hd = bdrv_new(""); 45483f64091Sbellard path_combine(backing_filename, sizeof(backing_filename), 45583f64091Sbellard filename, bs->backing_file); 4565eb45639Saliguori if (bs->backing_format[0] != '\0') 4575eb45639Saliguori back_drv = bdrv_find_format(bs->backing_format); 4585eb45639Saliguori ret = bdrv_open2(bs->backing_hd, backing_filename, open_flags, 4595eb45639Saliguori back_drv); 46051d7c00cSaliguori if (ret < 0) { 46151d7c00cSaliguori bdrv_close(bs); 46251d7c00cSaliguori return ret; 46351d7c00cSaliguori } 464ea2384d3Sbellard } 46533e3963eSbellard 466bb5fc20fSaliguori if (!bdrv_key_required(bs)) { 467b338082bSbellard /* call the change callback */ 46819cb3738Sbellard bs->media_changed = 1; 469b338082bSbellard if (bs->change_cb) 470b338082bSbellard bs->change_cb(bs->change_opaque); 471bb5fc20fSaliguori } 472b338082bSbellard return 0; 473fc01f7e7Sbellard } 474fc01f7e7Sbellard 475fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs) 476fc01f7e7Sbellard { 47719cb3738Sbellard if (bs->drv) { 478ea2384d3Sbellard if (bs->backing_hd) 479ea2384d3Sbellard bdrv_delete(bs->backing_hd); 480ea2384d3Sbellard bs->drv->bdrv_close(bs); 481ea2384d3Sbellard qemu_free(bs->opaque); 482ea2384d3Sbellard #ifdef _WIN32 483ea2384d3Sbellard if (bs->is_temporary) { 484ea2384d3Sbellard unlink(bs->filename); 485ea2384d3Sbellard } 48667b915a5Sbellard #endif 487ea2384d3Sbellard bs->opaque = NULL; 488ea2384d3Sbellard bs->drv = NULL; 489b338082bSbellard 490b338082bSbellard /* call the change callback */ 49119cb3738Sbellard bs->media_changed = 1; 492b338082bSbellard if (bs->change_cb) 493b338082bSbellard bs->change_cb(bs->change_opaque); 494b338082bSbellard } 495b338082bSbellard } 496b338082bSbellard 497b338082bSbellard void bdrv_delete(BlockDriverState *bs) 498b338082bSbellard { 49934c6f050Saurel32 BlockDriverState **pbs; 50034c6f050Saurel32 50134c6f050Saurel32 pbs = &bdrv_first; 50234c6f050Saurel32 while (*pbs != bs && *pbs != NULL) 50334c6f050Saurel32 pbs = &(*pbs)->next; 50434c6f050Saurel32 if (*pbs == bs) 50534c6f050Saurel32 *pbs = bs->next; 50634c6f050Saurel32 507b338082bSbellard bdrv_close(bs); 508b338082bSbellard qemu_free(bs); 509fc01f7e7Sbellard } 510fc01f7e7Sbellard 511e97fc193Saliguori /* 512e97fc193Saliguori * Run consistency checks on an image 513e97fc193Saliguori * 514e97fc193Saliguori * Returns the number of errors or -errno when an internal error occurs 515e97fc193Saliguori */ 516e97fc193Saliguori int bdrv_check(BlockDriverState *bs) 517e97fc193Saliguori { 518e97fc193Saliguori if (bs->drv->bdrv_check == NULL) { 519e97fc193Saliguori return -ENOTSUP; 520e97fc193Saliguori } 521e97fc193Saliguori 522e97fc193Saliguori return bs->drv->bdrv_check(bs); 523e97fc193Saliguori } 524e97fc193Saliguori 52533e3963eSbellard /* commit COW file into the raw image */ 52633e3963eSbellard int bdrv_commit(BlockDriverState *bs) 52733e3963eSbellard { 52819cb3738Sbellard BlockDriver *drv = bs->drv; 52983f64091Sbellard int64_t i, total_sectors; 530ea2384d3Sbellard int n, j; 531ea2384d3Sbellard unsigned char sector[512]; 53233e3963eSbellard 53319cb3738Sbellard if (!drv) 53419cb3738Sbellard return -ENOMEDIUM; 53533e3963eSbellard 53633e3963eSbellard if (bs->read_only) { 537ea2384d3Sbellard return -EACCES; 53833e3963eSbellard } 53933e3963eSbellard 540ea2384d3Sbellard if (!bs->backing_hd) { 541ea2384d3Sbellard return -ENOTSUP; 542ea2384d3Sbellard } 543ea2384d3Sbellard 54483f64091Sbellard total_sectors = bdrv_getlength(bs) >> SECTOR_BITS; 54583f64091Sbellard for (i = 0; i < total_sectors;) { 54619cb3738Sbellard if (drv->bdrv_is_allocated(bs, i, 65536, &n)) { 547ea2384d3Sbellard for(j = 0; j < n; j++) { 54833e3963eSbellard if (bdrv_read(bs, i, sector, 1) != 0) { 549ea2384d3Sbellard return -EIO; 55033e3963eSbellard } 55133e3963eSbellard 552ea2384d3Sbellard if (bdrv_write(bs->backing_hd, i, sector, 1) != 0) { 553ea2384d3Sbellard return -EIO; 55433e3963eSbellard } 555ea2384d3Sbellard i++; 556ea2384d3Sbellard } 557ea2384d3Sbellard } else { 558ea2384d3Sbellard i += n; 55933e3963eSbellard } 56033e3963eSbellard } 56195389c86Sbellard 56219cb3738Sbellard if (drv->bdrv_make_empty) 56319cb3738Sbellard return drv->bdrv_make_empty(bs); 56495389c86Sbellard 56533e3963eSbellard return 0; 56633e3963eSbellard } 56733e3963eSbellard 56871d0770cSaliguori static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset, 56971d0770cSaliguori size_t size) 57071d0770cSaliguori { 57171d0770cSaliguori int64_t len; 57271d0770cSaliguori 57371d0770cSaliguori if (!bdrv_is_inserted(bs)) 57471d0770cSaliguori return -ENOMEDIUM; 57571d0770cSaliguori 57671d0770cSaliguori if (bs->growable) 57771d0770cSaliguori return 0; 57871d0770cSaliguori 57971d0770cSaliguori len = bdrv_getlength(bs); 58071d0770cSaliguori 581*fbb7b4e0SKevin Wolf if (offset < 0) 582*fbb7b4e0SKevin Wolf return -EIO; 583*fbb7b4e0SKevin Wolf 584*fbb7b4e0SKevin Wolf if ((offset > len) || (len - offset < size)) 58571d0770cSaliguori return -EIO; 58671d0770cSaliguori 58771d0770cSaliguori return 0; 58871d0770cSaliguori } 58971d0770cSaliguori 59071d0770cSaliguori static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num, 59171d0770cSaliguori int nb_sectors) 59271d0770cSaliguori { 593999dec57Saliguori return bdrv_check_byte_request(bs, sector_num * 512, nb_sectors * 512); 59471d0770cSaliguori } 59571d0770cSaliguori 59619cb3738Sbellard /* return < 0 if error. See bdrv_write() for the return codes */ 597fc01f7e7Sbellard int bdrv_read(BlockDriverState *bs, int64_t sector_num, 598fc01f7e7Sbellard uint8_t *buf, int nb_sectors) 599fc01f7e7Sbellard { 600ea2384d3Sbellard BlockDriver *drv = bs->drv; 601fc01f7e7Sbellard 60219cb3738Sbellard if (!drv) 60319cb3738Sbellard return -ENOMEDIUM; 60471d0770cSaliguori if (bdrv_check_request(bs, sector_num, nb_sectors)) 60571d0770cSaliguori return -EIO; 606b338082bSbellard 60783f64091Sbellard return drv->bdrv_read(bs, sector_num, buf, nb_sectors); 60883f64091Sbellard } 609fc01f7e7Sbellard 61019cb3738Sbellard /* Return < 0 if error. Important errors are: 61119cb3738Sbellard -EIO generic I/O error (may happen for all errors) 61219cb3738Sbellard -ENOMEDIUM No media inserted. 61319cb3738Sbellard -EINVAL Invalid sector number or nb_sectors 61419cb3738Sbellard -EACCES Trying to write a read-only device 61519cb3738Sbellard */ 616fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num, 617fc01f7e7Sbellard const uint8_t *buf, int nb_sectors) 618fc01f7e7Sbellard { 61983f64091Sbellard BlockDriver *drv = bs->drv; 62019cb3738Sbellard if (!bs->drv) 62119cb3738Sbellard return -ENOMEDIUM; 6220849bf08Sbellard if (bs->read_only) 62319cb3738Sbellard return -EACCES; 62471d0770cSaliguori if (bdrv_check_request(bs, sector_num, nb_sectors)) 62571d0770cSaliguori return -EIO; 62671d0770cSaliguori 62783f64091Sbellard return drv->bdrv_write(bs, sector_num, buf, nb_sectors); 62883f64091Sbellard } 62983f64091Sbellard 630eda578e5Saliguori int bdrv_pread(BlockDriverState *bs, int64_t offset, 631eda578e5Saliguori void *buf, int count1) 63283f64091Sbellard { 63383f64091Sbellard uint8_t tmp_buf[SECTOR_SIZE]; 63483f64091Sbellard int len, nb_sectors, count; 63583f64091Sbellard int64_t sector_num; 63683f64091Sbellard 63783f64091Sbellard count = count1; 63883f64091Sbellard /* first read to align to sector start */ 63983f64091Sbellard len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1); 64083f64091Sbellard if (len > count) 64183f64091Sbellard len = count; 64283f64091Sbellard sector_num = offset >> SECTOR_BITS; 64383f64091Sbellard if (len > 0) { 64483f64091Sbellard if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0) 64583f64091Sbellard return -EIO; 64683f64091Sbellard memcpy(buf, tmp_buf + (offset & (SECTOR_SIZE - 1)), len); 64783f64091Sbellard count -= len; 64883f64091Sbellard if (count == 0) 64983f64091Sbellard return count1; 65083f64091Sbellard sector_num++; 65183f64091Sbellard buf += len; 65283f64091Sbellard } 65383f64091Sbellard 65483f64091Sbellard /* read the sectors "in place" */ 65583f64091Sbellard nb_sectors = count >> SECTOR_BITS; 65683f64091Sbellard if (nb_sectors > 0) { 65783f64091Sbellard if (bdrv_read(bs, sector_num, buf, nb_sectors) < 0) 65883f64091Sbellard return -EIO; 65983f64091Sbellard sector_num += nb_sectors; 66083f64091Sbellard len = nb_sectors << SECTOR_BITS; 66183f64091Sbellard buf += len; 66283f64091Sbellard count -= len; 66383f64091Sbellard } 66483f64091Sbellard 66583f64091Sbellard /* add data from the last sector */ 66683f64091Sbellard if (count > 0) { 66783f64091Sbellard if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0) 66883f64091Sbellard return -EIO; 66983f64091Sbellard memcpy(buf, tmp_buf, count); 67083f64091Sbellard } 67183f64091Sbellard return count1; 67283f64091Sbellard } 67383f64091Sbellard 674eda578e5Saliguori int bdrv_pwrite(BlockDriverState *bs, int64_t offset, 675eda578e5Saliguori const void *buf, int count1) 67683f64091Sbellard { 67783f64091Sbellard uint8_t tmp_buf[SECTOR_SIZE]; 67883f64091Sbellard int len, nb_sectors, count; 67983f64091Sbellard int64_t sector_num; 68083f64091Sbellard 68183f64091Sbellard count = count1; 68283f64091Sbellard /* first write to align to sector start */ 68383f64091Sbellard len = (SECTOR_SIZE - offset) & (SECTOR_SIZE - 1); 68483f64091Sbellard if (len > count) 68583f64091Sbellard len = count; 68683f64091Sbellard sector_num = offset >> SECTOR_BITS; 68783f64091Sbellard if (len > 0) { 68883f64091Sbellard if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0) 68983f64091Sbellard return -EIO; 69083f64091Sbellard memcpy(tmp_buf + (offset & (SECTOR_SIZE - 1)), buf, len); 69183f64091Sbellard if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0) 69283f64091Sbellard return -EIO; 69383f64091Sbellard count -= len; 69483f64091Sbellard if (count == 0) 69583f64091Sbellard return count1; 69683f64091Sbellard sector_num++; 69783f64091Sbellard buf += len; 69883f64091Sbellard } 69983f64091Sbellard 70083f64091Sbellard /* write the sectors "in place" */ 70183f64091Sbellard nb_sectors = count >> SECTOR_BITS; 70283f64091Sbellard if (nb_sectors > 0) { 70383f64091Sbellard if (bdrv_write(bs, sector_num, buf, nb_sectors) < 0) 70483f64091Sbellard return -EIO; 70583f64091Sbellard sector_num += nb_sectors; 70683f64091Sbellard len = nb_sectors << SECTOR_BITS; 70783f64091Sbellard buf += len; 70883f64091Sbellard count -= len; 70983f64091Sbellard } 71083f64091Sbellard 71183f64091Sbellard /* add data from the last sector */ 71283f64091Sbellard if (count > 0) { 71383f64091Sbellard if (bdrv_read(bs, sector_num, tmp_buf, 1) < 0) 71483f64091Sbellard return -EIO; 71583f64091Sbellard memcpy(tmp_buf, buf, count); 71683f64091Sbellard if (bdrv_write(bs, sector_num, tmp_buf, 1) < 0) 71783f64091Sbellard return -EIO; 71883f64091Sbellard } 71983f64091Sbellard return count1; 72083f64091Sbellard } 72183f64091Sbellard 72283f64091Sbellard /** 72383f64091Sbellard * Truncate file to 'offset' bytes (needed only for file protocols) 72483f64091Sbellard */ 72583f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset) 72683f64091Sbellard { 72783f64091Sbellard BlockDriver *drv = bs->drv; 72883f64091Sbellard if (!drv) 72919cb3738Sbellard return -ENOMEDIUM; 73083f64091Sbellard if (!drv->bdrv_truncate) 73183f64091Sbellard return -ENOTSUP; 73283f64091Sbellard return drv->bdrv_truncate(bs, offset); 73383f64091Sbellard } 73483f64091Sbellard 73583f64091Sbellard /** 73683f64091Sbellard * Length of a file in bytes. Return < 0 if error or unknown. 73783f64091Sbellard */ 73883f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs) 73983f64091Sbellard { 74083f64091Sbellard BlockDriver *drv = bs->drv; 74183f64091Sbellard if (!drv) 74219cb3738Sbellard return -ENOMEDIUM; 74383f64091Sbellard if (!drv->bdrv_getlength) { 74483f64091Sbellard /* legacy mode */ 74583f64091Sbellard return bs->total_sectors * SECTOR_SIZE; 74683f64091Sbellard } 74783f64091Sbellard return drv->bdrv_getlength(bs); 748fc01f7e7Sbellard } 749fc01f7e7Sbellard 75019cb3738Sbellard /* return 0 as number of sectors if no device present or error */ 75196b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr) 752fc01f7e7Sbellard { 75319cb3738Sbellard int64_t length; 75419cb3738Sbellard length = bdrv_getlength(bs); 75519cb3738Sbellard if (length < 0) 75619cb3738Sbellard length = 0; 75719cb3738Sbellard else 75819cb3738Sbellard length = length >> SECTOR_BITS; 75919cb3738Sbellard *nb_sectors_ptr = length; 760fc01f7e7Sbellard } 761cf98951bSbellard 762f3d54fc4Saliguori struct partition { 763f3d54fc4Saliguori uint8_t boot_ind; /* 0x80 - active */ 764f3d54fc4Saliguori uint8_t head; /* starting head */ 765f3d54fc4Saliguori uint8_t sector; /* starting sector */ 766f3d54fc4Saliguori uint8_t cyl; /* starting cylinder */ 767f3d54fc4Saliguori uint8_t sys_ind; /* What partition type */ 768f3d54fc4Saliguori uint8_t end_head; /* end head */ 769f3d54fc4Saliguori uint8_t end_sector; /* end sector */ 770f3d54fc4Saliguori uint8_t end_cyl; /* end cylinder */ 771f3d54fc4Saliguori uint32_t start_sect; /* starting sector counting from 0 */ 772f3d54fc4Saliguori uint32_t nr_sects; /* nr of sectors in partition */ 773f3d54fc4Saliguori } __attribute__((packed)); 774f3d54fc4Saliguori 775f3d54fc4Saliguori /* try to guess the disk logical geometry from the MSDOS partition table. Return 0 if OK, -1 if could not guess */ 776f3d54fc4Saliguori static int guess_disk_lchs(BlockDriverState *bs, 777f3d54fc4Saliguori int *pcylinders, int *pheads, int *psectors) 778f3d54fc4Saliguori { 779f3d54fc4Saliguori uint8_t buf[512]; 780f3d54fc4Saliguori int ret, i, heads, sectors, cylinders; 781f3d54fc4Saliguori struct partition *p; 782f3d54fc4Saliguori uint32_t nr_sects; 783a38131b6Sblueswir1 uint64_t nb_sectors; 784f3d54fc4Saliguori 785f3d54fc4Saliguori bdrv_get_geometry(bs, &nb_sectors); 786f3d54fc4Saliguori 787f3d54fc4Saliguori ret = bdrv_read(bs, 0, buf, 1); 788f3d54fc4Saliguori if (ret < 0) 789f3d54fc4Saliguori return -1; 790f3d54fc4Saliguori /* test msdos magic */ 791f3d54fc4Saliguori if (buf[510] != 0x55 || buf[511] != 0xaa) 792f3d54fc4Saliguori return -1; 793f3d54fc4Saliguori for(i = 0; i < 4; i++) { 794f3d54fc4Saliguori p = ((struct partition *)(buf + 0x1be)) + i; 795f3d54fc4Saliguori nr_sects = le32_to_cpu(p->nr_sects); 796f3d54fc4Saliguori if (nr_sects && p->end_head) { 797f3d54fc4Saliguori /* We make the assumption that the partition terminates on 798f3d54fc4Saliguori a cylinder boundary */ 799f3d54fc4Saliguori heads = p->end_head + 1; 800f3d54fc4Saliguori sectors = p->end_sector & 63; 801f3d54fc4Saliguori if (sectors == 0) 802f3d54fc4Saliguori continue; 803f3d54fc4Saliguori cylinders = nb_sectors / (heads * sectors); 804f3d54fc4Saliguori if (cylinders < 1 || cylinders > 16383) 805f3d54fc4Saliguori continue; 806f3d54fc4Saliguori *pheads = heads; 807f3d54fc4Saliguori *psectors = sectors; 808f3d54fc4Saliguori *pcylinders = cylinders; 809f3d54fc4Saliguori #if 0 810f3d54fc4Saliguori printf("guessed geometry: LCHS=%d %d %d\n", 811f3d54fc4Saliguori cylinders, heads, sectors); 812f3d54fc4Saliguori #endif 813f3d54fc4Saliguori return 0; 814f3d54fc4Saliguori } 815f3d54fc4Saliguori } 816f3d54fc4Saliguori return -1; 817f3d54fc4Saliguori } 818f3d54fc4Saliguori 819f3d54fc4Saliguori void bdrv_guess_geometry(BlockDriverState *bs, int *pcyls, int *pheads, int *psecs) 820f3d54fc4Saliguori { 821f3d54fc4Saliguori int translation, lba_detected = 0; 822f3d54fc4Saliguori int cylinders, heads, secs; 823a38131b6Sblueswir1 uint64_t nb_sectors; 824f3d54fc4Saliguori 825f3d54fc4Saliguori /* if a geometry hint is available, use it */ 826f3d54fc4Saliguori bdrv_get_geometry(bs, &nb_sectors); 827f3d54fc4Saliguori bdrv_get_geometry_hint(bs, &cylinders, &heads, &secs); 828f3d54fc4Saliguori translation = bdrv_get_translation_hint(bs); 829f3d54fc4Saliguori if (cylinders != 0) { 830f3d54fc4Saliguori *pcyls = cylinders; 831f3d54fc4Saliguori *pheads = heads; 832f3d54fc4Saliguori *psecs = secs; 833f3d54fc4Saliguori } else { 834f3d54fc4Saliguori if (guess_disk_lchs(bs, &cylinders, &heads, &secs) == 0) { 835f3d54fc4Saliguori if (heads > 16) { 836f3d54fc4Saliguori /* if heads > 16, it means that a BIOS LBA 837f3d54fc4Saliguori translation was active, so the default 838f3d54fc4Saliguori hardware geometry is OK */ 839f3d54fc4Saliguori lba_detected = 1; 840f3d54fc4Saliguori goto default_geometry; 841f3d54fc4Saliguori } else { 842f3d54fc4Saliguori *pcyls = cylinders; 843f3d54fc4Saliguori *pheads = heads; 844f3d54fc4Saliguori *psecs = secs; 845f3d54fc4Saliguori /* disable any translation to be in sync with 846f3d54fc4Saliguori the logical geometry */ 847f3d54fc4Saliguori if (translation == BIOS_ATA_TRANSLATION_AUTO) { 848f3d54fc4Saliguori bdrv_set_translation_hint(bs, 849f3d54fc4Saliguori BIOS_ATA_TRANSLATION_NONE); 850f3d54fc4Saliguori } 851f3d54fc4Saliguori } 852f3d54fc4Saliguori } else { 853f3d54fc4Saliguori default_geometry: 854f3d54fc4Saliguori /* if no geometry, use a standard physical disk geometry */ 855f3d54fc4Saliguori cylinders = nb_sectors / (16 * 63); 856f3d54fc4Saliguori 857f3d54fc4Saliguori if (cylinders > 16383) 858f3d54fc4Saliguori cylinders = 16383; 859f3d54fc4Saliguori else if (cylinders < 2) 860f3d54fc4Saliguori cylinders = 2; 861f3d54fc4Saliguori *pcyls = cylinders; 862f3d54fc4Saliguori *pheads = 16; 863f3d54fc4Saliguori *psecs = 63; 864f3d54fc4Saliguori if ((lba_detected == 1) && (translation == BIOS_ATA_TRANSLATION_AUTO)) { 865f3d54fc4Saliguori if ((*pcyls * *pheads) <= 131072) { 866f3d54fc4Saliguori bdrv_set_translation_hint(bs, 867f3d54fc4Saliguori BIOS_ATA_TRANSLATION_LARGE); 868f3d54fc4Saliguori } else { 869f3d54fc4Saliguori bdrv_set_translation_hint(bs, 870f3d54fc4Saliguori BIOS_ATA_TRANSLATION_LBA); 871f3d54fc4Saliguori } 872f3d54fc4Saliguori } 873f3d54fc4Saliguori } 874f3d54fc4Saliguori bdrv_set_geometry_hint(bs, *pcyls, *pheads, *psecs); 875f3d54fc4Saliguori } 876f3d54fc4Saliguori } 877f3d54fc4Saliguori 878b338082bSbellard void bdrv_set_geometry_hint(BlockDriverState *bs, 879b338082bSbellard int cyls, int heads, int secs) 880b338082bSbellard { 881b338082bSbellard bs->cyls = cyls; 882b338082bSbellard bs->heads = heads; 883b338082bSbellard bs->secs = secs; 884b338082bSbellard } 885b338082bSbellard 886b338082bSbellard void bdrv_set_type_hint(BlockDriverState *bs, int type) 887b338082bSbellard { 888b338082bSbellard bs->type = type; 889b338082bSbellard bs->removable = ((type == BDRV_TYPE_CDROM || 890b338082bSbellard type == BDRV_TYPE_FLOPPY)); 891b338082bSbellard } 892b338082bSbellard 89346d4767dSbellard void bdrv_set_translation_hint(BlockDriverState *bs, int translation) 89446d4767dSbellard { 89546d4767dSbellard bs->translation = translation; 89646d4767dSbellard } 89746d4767dSbellard 898b338082bSbellard void bdrv_get_geometry_hint(BlockDriverState *bs, 899b338082bSbellard int *pcyls, int *pheads, int *psecs) 900b338082bSbellard { 901b338082bSbellard *pcyls = bs->cyls; 902b338082bSbellard *pheads = bs->heads; 903b338082bSbellard *psecs = bs->secs; 904b338082bSbellard } 905b338082bSbellard 906b338082bSbellard int bdrv_get_type_hint(BlockDriverState *bs) 907b338082bSbellard { 908b338082bSbellard return bs->type; 909b338082bSbellard } 910b338082bSbellard 91146d4767dSbellard int bdrv_get_translation_hint(BlockDriverState *bs) 91246d4767dSbellard { 91346d4767dSbellard return bs->translation; 91446d4767dSbellard } 91546d4767dSbellard 916b338082bSbellard int bdrv_is_removable(BlockDriverState *bs) 917b338082bSbellard { 918b338082bSbellard return bs->removable; 919b338082bSbellard } 920b338082bSbellard 921b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs) 922b338082bSbellard { 923b338082bSbellard return bs->read_only; 924b338082bSbellard } 925b338082bSbellard 926985a03b0Sths int bdrv_is_sg(BlockDriverState *bs) 927985a03b0Sths { 928985a03b0Sths return bs->sg; 929985a03b0Sths } 930985a03b0Sths 93119cb3738Sbellard /* XXX: no longer used */ 932b338082bSbellard void bdrv_set_change_cb(BlockDriverState *bs, 933b338082bSbellard void (*change_cb)(void *opaque), void *opaque) 934b338082bSbellard { 935b338082bSbellard bs->change_cb = change_cb; 936b338082bSbellard bs->change_opaque = opaque; 937b338082bSbellard } 938b338082bSbellard 939ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs) 940ea2384d3Sbellard { 941ea2384d3Sbellard if (bs->backing_hd && bs->backing_hd->encrypted) 942ea2384d3Sbellard return 1; 943ea2384d3Sbellard return bs->encrypted; 944ea2384d3Sbellard } 945ea2384d3Sbellard 946c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs) 947c0f4ce77Saliguori { 948c0f4ce77Saliguori BlockDriverState *backing_hd = bs->backing_hd; 949c0f4ce77Saliguori 950c0f4ce77Saliguori if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key) 951c0f4ce77Saliguori return 1; 952c0f4ce77Saliguori return (bs->encrypted && !bs->valid_key); 953c0f4ce77Saliguori } 954c0f4ce77Saliguori 955ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key) 956ea2384d3Sbellard { 957ea2384d3Sbellard int ret; 958ea2384d3Sbellard if (bs->backing_hd && bs->backing_hd->encrypted) { 959ea2384d3Sbellard ret = bdrv_set_key(bs->backing_hd, key); 960ea2384d3Sbellard if (ret < 0) 961ea2384d3Sbellard return ret; 962ea2384d3Sbellard if (!bs->encrypted) 963ea2384d3Sbellard return 0; 964ea2384d3Sbellard } 965ea2384d3Sbellard if (!bs->encrypted || !bs->drv || !bs->drv->bdrv_set_key) 966ea2384d3Sbellard return -1; 967c0f4ce77Saliguori ret = bs->drv->bdrv_set_key(bs, key); 968bb5fc20fSaliguori if (ret < 0) { 969bb5fc20fSaliguori bs->valid_key = 0; 970bb5fc20fSaliguori } else if (!bs->valid_key) { 971bb5fc20fSaliguori bs->valid_key = 1; 972bb5fc20fSaliguori /* call the change callback now, we skipped it on open */ 973bb5fc20fSaliguori bs->media_changed = 1; 974bb5fc20fSaliguori if (bs->change_cb) 975bb5fc20fSaliguori bs->change_cb(bs->change_opaque); 976bb5fc20fSaliguori } 977c0f4ce77Saliguori return ret; 978ea2384d3Sbellard } 979ea2384d3Sbellard 980ea2384d3Sbellard void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size) 981ea2384d3Sbellard { 98219cb3738Sbellard if (!bs->drv) { 983ea2384d3Sbellard buf[0] = '\0'; 984ea2384d3Sbellard } else { 985ea2384d3Sbellard pstrcpy(buf, buf_size, bs->drv->format_name); 986ea2384d3Sbellard } 987ea2384d3Sbellard } 988ea2384d3Sbellard 989ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name), 990ea2384d3Sbellard void *opaque) 991ea2384d3Sbellard { 992ea2384d3Sbellard BlockDriver *drv; 993ea2384d3Sbellard 994ea2384d3Sbellard for (drv = first_drv; drv != NULL; drv = drv->next) { 995ea2384d3Sbellard it(opaque, drv->format_name); 996ea2384d3Sbellard } 997ea2384d3Sbellard } 998ea2384d3Sbellard 999b338082bSbellard BlockDriverState *bdrv_find(const char *name) 1000b338082bSbellard { 1001b338082bSbellard BlockDriverState *bs; 1002b338082bSbellard 1003b338082bSbellard for (bs = bdrv_first; bs != NULL; bs = bs->next) { 1004b338082bSbellard if (!strcmp(name, bs->device_name)) 1005b338082bSbellard return bs; 1006b338082bSbellard } 1007b338082bSbellard return NULL; 1008b338082bSbellard } 1009b338082bSbellard 101051de9760Saliguori void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque) 101181d0912dSbellard { 101281d0912dSbellard BlockDriverState *bs; 101381d0912dSbellard 101481d0912dSbellard for (bs = bdrv_first; bs != NULL; bs = bs->next) { 101551de9760Saliguori it(opaque, bs); 101681d0912dSbellard } 101781d0912dSbellard } 101881d0912dSbellard 1019ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs) 1020ea2384d3Sbellard { 1021ea2384d3Sbellard return bs->device_name; 1022ea2384d3Sbellard } 1023ea2384d3Sbellard 10247a6cba61Spbrook void bdrv_flush(BlockDriverState *bs) 10257a6cba61Spbrook { 1026081501daSaliguori if (!bs->drv) 1027081501daSaliguori return; 10287a6cba61Spbrook if (bs->drv->bdrv_flush) 10297a6cba61Spbrook bs->drv->bdrv_flush(bs); 10307a6cba61Spbrook if (bs->backing_hd) 10317a6cba61Spbrook bdrv_flush(bs->backing_hd); 10327a6cba61Spbrook } 10337a6cba61Spbrook 1034c6ca28d6Saliguori void bdrv_flush_all(void) 1035c6ca28d6Saliguori { 1036c6ca28d6Saliguori BlockDriverState *bs; 1037c6ca28d6Saliguori 1038c6ca28d6Saliguori for (bs = bdrv_first; bs != NULL; bs = bs->next) 1039c6ca28d6Saliguori if (bs->drv && !bdrv_is_read_only(bs) && 1040c6ca28d6Saliguori (!bdrv_is_removable(bs) || bdrv_is_inserted(bs))) 1041c6ca28d6Saliguori bdrv_flush(bs); 1042c6ca28d6Saliguori } 1043c6ca28d6Saliguori 1044f58c7b35Sths /* 1045f58c7b35Sths * Returns true iff the specified sector is present in the disk image. Drivers 1046f58c7b35Sths * not implementing the functionality are assumed to not support backing files, 1047f58c7b35Sths * hence all their sectors are reported as allocated. 1048f58c7b35Sths * 1049f58c7b35Sths * 'pnum' is set to the number of sectors (including and immediately following 1050f58c7b35Sths * the specified sector) that are known to be in the same 1051f58c7b35Sths * allocated/unallocated state. 1052f58c7b35Sths * 1053f58c7b35Sths * 'nb_sectors' is the max value 'pnum' should be set to. 1054f58c7b35Sths */ 1055f58c7b35Sths int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors, 1056f58c7b35Sths int *pnum) 1057f58c7b35Sths { 1058f58c7b35Sths int64_t n; 1059f58c7b35Sths if (!bs->drv->bdrv_is_allocated) { 1060f58c7b35Sths if (sector_num >= bs->total_sectors) { 1061f58c7b35Sths *pnum = 0; 1062f58c7b35Sths return 0; 1063f58c7b35Sths } 1064f58c7b35Sths n = bs->total_sectors - sector_num; 1065f58c7b35Sths *pnum = (n < nb_sectors) ? (n) : (nb_sectors); 1066f58c7b35Sths return 1; 1067f58c7b35Sths } 1068f58c7b35Sths return bs->drv->bdrv_is_allocated(bs, sector_num, nb_sectors, pnum); 1069f58c7b35Sths } 1070f58c7b35Sths 1071376253ecSaliguori void bdrv_info(Monitor *mon) 1072b338082bSbellard { 1073b338082bSbellard BlockDriverState *bs; 1074b338082bSbellard 1075b338082bSbellard for (bs = bdrv_first; bs != NULL; bs = bs->next) { 1076376253ecSaliguori monitor_printf(mon, "%s:", bs->device_name); 1077376253ecSaliguori monitor_printf(mon, " type="); 1078b338082bSbellard switch(bs->type) { 1079b338082bSbellard case BDRV_TYPE_HD: 1080376253ecSaliguori monitor_printf(mon, "hd"); 1081b338082bSbellard break; 1082b338082bSbellard case BDRV_TYPE_CDROM: 1083376253ecSaliguori monitor_printf(mon, "cdrom"); 1084b338082bSbellard break; 1085b338082bSbellard case BDRV_TYPE_FLOPPY: 1086376253ecSaliguori monitor_printf(mon, "floppy"); 1087b338082bSbellard break; 1088b338082bSbellard } 1089376253ecSaliguori monitor_printf(mon, " removable=%d", bs->removable); 1090b338082bSbellard if (bs->removable) { 1091376253ecSaliguori monitor_printf(mon, " locked=%d", bs->locked); 1092b338082bSbellard } 109319cb3738Sbellard if (bs->drv) { 1094376253ecSaliguori monitor_printf(mon, " file="); 1095376253ecSaliguori monitor_print_filename(mon, bs->filename); 1096fef30743Sths if (bs->backing_file[0] != '\0') { 1097376253ecSaliguori monitor_printf(mon, " backing_file="); 1098376253ecSaliguori monitor_print_filename(mon, bs->backing_file); 1099fef30743Sths } 1100376253ecSaliguori monitor_printf(mon, " ro=%d", bs->read_only); 1101376253ecSaliguori monitor_printf(mon, " drv=%s", bs->drv->format_name); 1102376253ecSaliguori monitor_printf(mon, " encrypted=%d", bdrv_is_encrypted(bs)); 1103b338082bSbellard } else { 1104376253ecSaliguori monitor_printf(mon, " [not inserted]"); 1105b338082bSbellard } 1106376253ecSaliguori monitor_printf(mon, "\n"); 1107b338082bSbellard } 1108b338082bSbellard } 1109a36e69ddSths 1110a36e69ddSths /* The "info blockstats" command. */ 1111376253ecSaliguori void bdrv_info_stats(Monitor *mon) 1112a36e69ddSths { 1113a36e69ddSths BlockDriverState *bs; 1114a36e69ddSths 1115a36e69ddSths for (bs = bdrv_first; bs != NULL; bs = bs->next) { 1116376253ecSaliguori monitor_printf(mon, "%s:" 1117a36e69ddSths " rd_bytes=%" PRIu64 1118a36e69ddSths " wr_bytes=%" PRIu64 1119a36e69ddSths " rd_operations=%" PRIu64 1120a36e69ddSths " wr_operations=%" PRIu64 1121ebf53fcdSaliguori "\n", 1122a36e69ddSths bs->device_name, 1123a36e69ddSths bs->rd_bytes, bs->wr_bytes, 1124a36e69ddSths bs->rd_ops, bs->wr_ops); 1125a36e69ddSths } 1126a36e69ddSths } 1127ea2384d3Sbellard 1128045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs) 1129045df330Saliguori { 1130045df330Saliguori if (bs->backing_hd && bs->backing_hd->encrypted) 1131045df330Saliguori return bs->backing_file; 1132045df330Saliguori else if (bs->encrypted) 1133045df330Saliguori return bs->filename; 1134045df330Saliguori else 1135045df330Saliguori return NULL; 1136045df330Saliguori } 1137045df330Saliguori 113883f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs, 113983f64091Sbellard char *filename, int filename_size) 114083f64091Sbellard { 114183f64091Sbellard if (!bs->backing_hd) { 114283f64091Sbellard pstrcpy(filename, filename_size, ""); 114383f64091Sbellard } else { 114483f64091Sbellard pstrcpy(filename, filename_size, bs->backing_file); 114583f64091Sbellard } 114683f64091Sbellard } 114783f64091Sbellard 1148faea38e7Sbellard int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num, 1149faea38e7Sbellard const uint8_t *buf, int nb_sectors) 1150faea38e7Sbellard { 1151faea38e7Sbellard BlockDriver *drv = bs->drv; 1152faea38e7Sbellard if (!drv) 115319cb3738Sbellard return -ENOMEDIUM; 1154faea38e7Sbellard if (!drv->bdrv_write_compressed) 1155faea38e7Sbellard return -ENOTSUP; 1156*fbb7b4e0SKevin Wolf if (bdrv_check_request(bs, sector_num, nb_sectors)) 1157*fbb7b4e0SKevin Wolf return -EIO; 1158faea38e7Sbellard return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors); 1159faea38e7Sbellard } 1160faea38e7Sbellard 1161faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi) 1162faea38e7Sbellard { 1163faea38e7Sbellard BlockDriver *drv = bs->drv; 1164faea38e7Sbellard if (!drv) 116519cb3738Sbellard return -ENOMEDIUM; 1166faea38e7Sbellard if (!drv->bdrv_get_info) 1167faea38e7Sbellard return -ENOTSUP; 1168faea38e7Sbellard memset(bdi, 0, sizeof(*bdi)); 1169faea38e7Sbellard return drv->bdrv_get_info(bs, bdi); 1170faea38e7Sbellard } 1171faea38e7Sbellard 1172178e08a5Saliguori int bdrv_put_buffer(BlockDriverState *bs, const uint8_t *buf, int64_t pos, int size) 1173178e08a5Saliguori { 1174178e08a5Saliguori BlockDriver *drv = bs->drv; 1175178e08a5Saliguori if (!drv) 1176178e08a5Saliguori return -ENOMEDIUM; 1177178e08a5Saliguori if (!drv->bdrv_put_buffer) 1178178e08a5Saliguori return -ENOTSUP; 1179178e08a5Saliguori return drv->bdrv_put_buffer(bs, buf, pos, size); 1180178e08a5Saliguori } 1181178e08a5Saliguori 1182178e08a5Saliguori int bdrv_get_buffer(BlockDriverState *bs, uint8_t *buf, int64_t pos, int size) 1183178e08a5Saliguori { 1184178e08a5Saliguori BlockDriver *drv = bs->drv; 1185178e08a5Saliguori if (!drv) 1186178e08a5Saliguori return -ENOMEDIUM; 1187178e08a5Saliguori if (!drv->bdrv_get_buffer) 1188178e08a5Saliguori return -ENOTSUP; 1189178e08a5Saliguori return drv->bdrv_get_buffer(bs, buf, pos, size); 1190178e08a5Saliguori } 1191178e08a5Saliguori 1192faea38e7Sbellard /**************************************************************/ 1193faea38e7Sbellard /* handling of snapshots */ 1194faea38e7Sbellard 1195faea38e7Sbellard int bdrv_snapshot_create(BlockDriverState *bs, 1196faea38e7Sbellard QEMUSnapshotInfo *sn_info) 1197faea38e7Sbellard { 1198faea38e7Sbellard BlockDriver *drv = bs->drv; 1199faea38e7Sbellard if (!drv) 120019cb3738Sbellard return -ENOMEDIUM; 1201faea38e7Sbellard if (!drv->bdrv_snapshot_create) 1202faea38e7Sbellard return -ENOTSUP; 1203faea38e7Sbellard return drv->bdrv_snapshot_create(bs, sn_info); 1204faea38e7Sbellard } 1205faea38e7Sbellard 1206faea38e7Sbellard int bdrv_snapshot_goto(BlockDriverState *bs, 1207faea38e7Sbellard const char *snapshot_id) 1208faea38e7Sbellard { 1209faea38e7Sbellard BlockDriver *drv = bs->drv; 1210faea38e7Sbellard if (!drv) 121119cb3738Sbellard return -ENOMEDIUM; 1212faea38e7Sbellard if (!drv->bdrv_snapshot_goto) 1213faea38e7Sbellard return -ENOTSUP; 1214faea38e7Sbellard return drv->bdrv_snapshot_goto(bs, snapshot_id); 1215faea38e7Sbellard } 1216faea38e7Sbellard 1217faea38e7Sbellard int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id) 1218faea38e7Sbellard { 1219faea38e7Sbellard BlockDriver *drv = bs->drv; 1220faea38e7Sbellard if (!drv) 122119cb3738Sbellard return -ENOMEDIUM; 1222faea38e7Sbellard if (!drv->bdrv_snapshot_delete) 1223faea38e7Sbellard return -ENOTSUP; 1224faea38e7Sbellard return drv->bdrv_snapshot_delete(bs, snapshot_id); 1225faea38e7Sbellard } 1226faea38e7Sbellard 1227faea38e7Sbellard int bdrv_snapshot_list(BlockDriverState *bs, 1228faea38e7Sbellard QEMUSnapshotInfo **psn_info) 1229faea38e7Sbellard { 1230faea38e7Sbellard BlockDriver *drv = bs->drv; 1231faea38e7Sbellard if (!drv) 123219cb3738Sbellard return -ENOMEDIUM; 1233faea38e7Sbellard if (!drv->bdrv_snapshot_list) 1234faea38e7Sbellard return -ENOTSUP; 1235faea38e7Sbellard return drv->bdrv_snapshot_list(bs, psn_info); 1236faea38e7Sbellard } 1237faea38e7Sbellard 1238faea38e7Sbellard #define NB_SUFFIXES 4 1239faea38e7Sbellard 1240faea38e7Sbellard char *get_human_readable_size(char *buf, int buf_size, int64_t size) 1241faea38e7Sbellard { 1242faea38e7Sbellard static const char suffixes[NB_SUFFIXES] = "KMGT"; 1243faea38e7Sbellard int64_t base; 1244faea38e7Sbellard int i; 1245faea38e7Sbellard 1246faea38e7Sbellard if (size <= 999) { 1247faea38e7Sbellard snprintf(buf, buf_size, "%" PRId64, size); 1248faea38e7Sbellard } else { 1249faea38e7Sbellard base = 1024; 1250faea38e7Sbellard for(i = 0; i < NB_SUFFIXES; i++) { 1251faea38e7Sbellard if (size < (10 * base)) { 1252faea38e7Sbellard snprintf(buf, buf_size, "%0.1f%c", 1253faea38e7Sbellard (double)size / base, 1254faea38e7Sbellard suffixes[i]); 1255faea38e7Sbellard break; 1256faea38e7Sbellard } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) { 1257faea38e7Sbellard snprintf(buf, buf_size, "%" PRId64 "%c", 1258faea38e7Sbellard ((size + (base >> 1)) / base), 1259faea38e7Sbellard suffixes[i]); 1260faea38e7Sbellard break; 1261faea38e7Sbellard } 1262faea38e7Sbellard base = base * 1024; 1263faea38e7Sbellard } 1264faea38e7Sbellard } 1265faea38e7Sbellard return buf; 1266faea38e7Sbellard } 1267faea38e7Sbellard 1268faea38e7Sbellard char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn) 1269faea38e7Sbellard { 1270faea38e7Sbellard char buf1[128], date_buf[128], clock_buf[128]; 12713b9f94e1Sbellard #ifdef _WIN32 12723b9f94e1Sbellard struct tm *ptm; 12733b9f94e1Sbellard #else 1274faea38e7Sbellard struct tm tm; 12753b9f94e1Sbellard #endif 1276faea38e7Sbellard time_t ti; 1277faea38e7Sbellard int64_t secs; 1278faea38e7Sbellard 1279faea38e7Sbellard if (!sn) { 1280faea38e7Sbellard snprintf(buf, buf_size, 1281faea38e7Sbellard "%-10s%-20s%7s%20s%15s", 1282faea38e7Sbellard "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK"); 1283faea38e7Sbellard } else { 1284faea38e7Sbellard ti = sn->date_sec; 12853b9f94e1Sbellard #ifdef _WIN32 12863b9f94e1Sbellard ptm = localtime(&ti); 12873b9f94e1Sbellard strftime(date_buf, sizeof(date_buf), 12883b9f94e1Sbellard "%Y-%m-%d %H:%M:%S", ptm); 12893b9f94e1Sbellard #else 1290faea38e7Sbellard localtime_r(&ti, &tm); 1291faea38e7Sbellard strftime(date_buf, sizeof(date_buf), 1292faea38e7Sbellard "%Y-%m-%d %H:%M:%S", &tm); 12933b9f94e1Sbellard #endif 1294faea38e7Sbellard secs = sn->vm_clock_nsec / 1000000000; 1295faea38e7Sbellard snprintf(clock_buf, sizeof(clock_buf), 1296faea38e7Sbellard "%02d:%02d:%02d.%03d", 1297faea38e7Sbellard (int)(secs / 3600), 1298faea38e7Sbellard (int)((secs / 60) % 60), 1299faea38e7Sbellard (int)(secs % 60), 1300faea38e7Sbellard (int)((sn->vm_clock_nsec / 1000000) % 1000)); 1301faea38e7Sbellard snprintf(buf, buf_size, 1302faea38e7Sbellard "%-10s%-20s%7s%20s%15s", 1303faea38e7Sbellard sn->id_str, sn->name, 1304faea38e7Sbellard get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size), 1305faea38e7Sbellard date_buf, 1306faea38e7Sbellard clock_buf); 1307faea38e7Sbellard } 1308faea38e7Sbellard return buf; 1309faea38e7Sbellard } 1310faea38e7Sbellard 131183f64091Sbellard 1312ea2384d3Sbellard /**************************************************************/ 131383f64091Sbellard /* async I/Os */ 1314ea2384d3Sbellard 13153b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num, 1316f141eafeSaliguori QEMUIOVector *qiov, int nb_sectors, 131783f64091Sbellard BlockDriverCompletionFunc *cb, void *opaque) 1318ea2384d3Sbellard { 131983f64091Sbellard BlockDriver *drv = bs->drv; 1320a36e69ddSths BlockDriverAIOCB *ret; 1321ea2384d3Sbellard 132219cb3738Sbellard if (!drv) 1323ce1a14dcSpbrook return NULL; 132471d0770cSaliguori if (bdrv_check_request(bs, sector_num, nb_sectors)) 132571d0770cSaliguori return NULL; 132683f64091Sbellard 1327f141eafeSaliguori ret = drv->bdrv_aio_readv(bs, sector_num, qiov, nb_sectors, 1328f141eafeSaliguori cb, opaque); 1329a36e69ddSths 1330a36e69ddSths if (ret) { 1331a36e69ddSths /* Update stats even though technically transfer has not happened. */ 1332a36e69ddSths bs->rd_bytes += (unsigned) nb_sectors * SECTOR_SIZE; 1333a36e69ddSths bs->rd_ops ++; 1334a36e69ddSths } 1335a36e69ddSths 1336a36e69ddSths return ret; 133783f64091Sbellard } 133883f64091Sbellard 1339f141eafeSaliguori BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num, 1340f141eafeSaliguori QEMUIOVector *qiov, int nb_sectors, 134183f64091Sbellard BlockDriverCompletionFunc *cb, void *opaque) 13427674e7bfSbellard { 134383f64091Sbellard BlockDriver *drv = bs->drv; 1344a36e69ddSths BlockDriverAIOCB *ret; 134583f64091Sbellard 134619cb3738Sbellard if (!drv) 1347ce1a14dcSpbrook return NULL; 134883f64091Sbellard if (bs->read_only) 1349ce1a14dcSpbrook return NULL; 135071d0770cSaliguori if (bdrv_check_request(bs, sector_num, nb_sectors)) 135171d0770cSaliguori return NULL; 135283f64091Sbellard 1353f141eafeSaliguori ret = drv->bdrv_aio_writev(bs, sector_num, qiov, nb_sectors, 1354f141eafeSaliguori cb, opaque); 1355a36e69ddSths 1356a36e69ddSths if (ret) { 1357a36e69ddSths /* Update stats even though technically transfer has not happened. */ 1358a36e69ddSths bs->wr_bytes += (unsigned) nb_sectors * SECTOR_SIZE; 1359a36e69ddSths bs->wr_ops ++; 1360a36e69ddSths } 1361a36e69ddSths 1362a36e69ddSths return ret; 136383f64091Sbellard } 136483f64091Sbellard 136583f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb) 136683f64091Sbellard { 13676bbff9a0Saliguori acb->pool->cancel(acb); 136883f64091Sbellard } 136983f64091Sbellard 137083f64091Sbellard 137183f64091Sbellard /**************************************************************/ 137283f64091Sbellard /* async block device emulation */ 137383f64091Sbellard 137483f64091Sbellard static void bdrv_aio_bh_cb(void *opaque) 1375beac80cdSbellard { 1376ce1a14dcSpbrook BlockDriverAIOCBSync *acb = opaque; 1377f141eafeSaliguori 1378f141eafeSaliguori if (!acb->is_write) 1379f141eafeSaliguori qemu_iovec_from_buffer(acb->qiov, acb->bounce, acb->qiov->size); 1380ceb42de8Saliguori qemu_vfree(acb->bounce); 1381ce1a14dcSpbrook acb->common.cb(acb->common.opaque, acb->ret); 1382f141eafeSaliguori 1383ce1a14dcSpbrook qemu_aio_release(acb); 1384beac80cdSbellard } 1385beac80cdSbellard 1386f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs, 1387f141eafeSaliguori int64_t sector_num, 1388f141eafeSaliguori QEMUIOVector *qiov, 1389f141eafeSaliguori int nb_sectors, 1390f141eafeSaliguori BlockDriverCompletionFunc *cb, 1391f141eafeSaliguori void *opaque, 1392f141eafeSaliguori int is_write) 1393f141eafeSaliguori 1394ea2384d3Sbellard { 1395ce1a14dcSpbrook BlockDriverAIOCBSync *acb; 139683f64091Sbellard 1397ce1a14dcSpbrook acb = qemu_aio_get(bs, cb, opaque); 1398f141eafeSaliguori acb->is_write = is_write; 1399f141eafeSaliguori acb->qiov = qiov; 1400e268ca52Saliguori acb->bounce = qemu_blockalign(bs, qiov->size); 1401f141eafeSaliguori 1402ce1a14dcSpbrook if (!acb->bh) 1403ce1a14dcSpbrook acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb); 1404f141eafeSaliguori 1405f141eafeSaliguori if (is_write) { 1406f141eafeSaliguori qemu_iovec_to_buffer(acb->qiov, acb->bounce); 1407f141eafeSaliguori acb->ret = bdrv_write(bs, sector_num, acb->bounce, nb_sectors); 1408f141eafeSaliguori } else { 1409f141eafeSaliguori acb->ret = bdrv_read(bs, sector_num, acb->bounce, nb_sectors); 1410f141eafeSaliguori } 1411f141eafeSaliguori 1412ce1a14dcSpbrook qemu_bh_schedule(acb->bh); 1413f141eafeSaliguori 1414ce1a14dcSpbrook return &acb->common; 14157a6cba61Spbrook } 14167a6cba61Spbrook 1417f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs, 1418f141eafeSaliguori int64_t sector_num, QEMUIOVector *qiov, int nb_sectors, 1419ce1a14dcSpbrook BlockDriverCompletionFunc *cb, void *opaque) 142083f64091Sbellard { 1421f141eafeSaliguori return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0); 142283f64091Sbellard } 142383f64091Sbellard 1424f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs, 1425f141eafeSaliguori int64_t sector_num, QEMUIOVector *qiov, int nb_sectors, 1426f141eafeSaliguori BlockDriverCompletionFunc *cb, void *opaque) 1427f141eafeSaliguori { 1428f141eafeSaliguori return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1); 1429f141eafeSaliguori } 1430f141eafeSaliguori 1431f141eafeSaliguori 1432ce1a14dcSpbrook static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb) 143383f64091Sbellard { 1434ce1a14dcSpbrook BlockDriverAIOCBSync *acb = (BlockDriverAIOCBSync *)blockacb; 1435ce1a14dcSpbrook qemu_bh_cancel(acb->bh); 1436ce1a14dcSpbrook qemu_aio_release(acb); 143783f64091Sbellard } 143883f64091Sbellard 143983f64091Sbellard /**************************************************************/ 144083f64091Sbellard /* sync block device emulation */ 144183f64091Sbellard 144283f64091Sbellard static void bdrv_rw_em_cb(void *opaque, int ret) 144383f64091Sbellard { 144483f64091Sbellard *(int *)opaque = ret; 144583f64091Sbellard } 144683f64091Sbellard 144783f64091Sbellard #define NOT_DONE 0x7fffffff 144883f64091Sbellard 144983f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num, 145083f64091Sbellard uint8_t *buf, int nb_sectors) 145183f64091Sbellard { 1452ce1a14dcSpbrook int async_ret; 1453ce1a14dcSpbrook BlockDriverAIOCB *acb; 1454f141eafeSaliguori struct iovec iov; 1455f141eafeSaliguori QEMUIOVector qiov; 145683f64091Sbellard 145783f64091Sbellard async_ret = NOT_DONE; 14583f4cb3d3Sblueswir1 iov.iov_base = (void *)buf; 1459f141eafeSaliguori iov.iov_len = nb_sectors * 512; 1460f141eafeSaliguori qemu_iovec_init_external(&qiov, &iov, 1); 1461f141eafeSaliguori acb = bdrv_aio_readv(bs, sector_num, &qiov, nb_sectors, 146283f64091Sbellard bdrv_rw_em_cb, &async_ret); 1463baf35cb9Saliguori if (acb == NULL) 1464ce1a14dcSpbrook return -1; 1465baf35cb9Saliguori 146683f64091Sbellard while (async_ret == NOT_DONE) { 146783f64091Sbellard qemu_aio_wait(); 146883f64091Sbellard } 1469baf35cb9Saliguori 147083f64091Sbellard return async_ret; 147183f64091Sbellard } 147283f64091Sbellard 147383f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num, 147483f64091Sbellard const uint8_t *buf, int nb_sectors) 147583f64091Sbellard { 1476ce1a14dcSpbrook int async_ret; 1477ce1a14dcSpbrook BlockDriverAIOCB *acb; 1478f141eafeSaliguori struct iovec iov; 1479f141eafeSaliguori QEMUIOVector qiov; 148083f64091Sbellard 148183f64091Sbellard async_ret = NOT_DONE; 1482f141eafeSaliguori iov.iov_base = (void *)buf; 1483f141eafeSaliguori iov.iov_len = nb_sectors * 512; 1484f141eafeSaliguori qemu_iovec_init_external(&qiov, &iov, 1); 1485f141eafeSaliguori acb = bdrv_aio_writev(bs, sector_num, &qiov, nb_sectors, 148683f64091Sbellard bdrv_rw_em_cb, &async_ret); 1487baf35cb9Saliguori if (acb == NULL) 1488ce1a14dcSpbrook return -1; 148983f64091Sbellard while (async_ret == NOT_DONE) { 149083f64091Sbellard qemu_aio_wait(); 149183f64091Sbellard } 149283f64091Sbellard return async_ret; 149383f64091Sbellard } 1494ea2384d3Sbellard 1495ea2384d3Sbellard void bdrv_init(void) 1496ea2384d3Sbellard { 1497ea2384d3Sbellard bdrv_register(&bdrv_raw); 149819cb3738Sbellard bdrv_register(&bdrv_host_device); 1499ea2384d3Sbellard #ifndef _WIN32 1500ea2384d3Sbellard bdrv_register(&bdrv_cow); 1501ea2384d3Sbellard #endif 1502ea2384d3Sbellard bdrv_register(&bdrv_qcow); 1503ea2384d3Sbellard bdrv_register(&bdrv_vmdk); 15043c56521bSbellard bdrv_register(&bdrv_cloop); 1505585d0ed9Sbellard bdrv_register(&bdrv_dmg); 1506a8753c34Sbellard bdrv_register(&bdrv_bochs); 15076a0f9e82Sbellard bdrv_register(&bdrv_vpc); 1508712e7874Sbellard bdrv_register(&bdrv_vvfat); 1509faea38e7Sbellard bdrv_register(&bdrv_qcow2); 15106ada7453Sths bdrv_register(&bdrv_parallels); 151175818250Sths bdrv_register(&bdrv_nbd); 1512ea2384d3Sbellard } 1513ce1a14dcSpbrook 15146bbff9a0Saliguori void aio_pool_init(AIOPool *pool, int aiocb_size, 15156bbff9a0Saliguori void (*cancel)(BlockDriverAIOCB *acb)) 1516ce1a14dcSpbrook { 15176bbff9a0Saliguori pool->aiocb_size = aiocb_size; 15186bbff9a0Saliguori pool->cancel = cancel; 15196bbff9a0Saliguori pool->free_aiocb = NULL; 15206bbff9a0Saliguori } 15216bbff9a0Saliguori 15226bbff9a0Saliguori void *qemu_aio_get_pool(AIOPool *pool, BlockDriverState *bs, 15236bbff9a0Saliguori BlockDriverCompletionFunc *cb, void *opaque) 15246bbff9a0Saliguori { 1525ce1a14dcSpbrook BlockDriverAIOCB *acb; 1526ce1a14dcSpbrook 15276bbff9a0Saliguori if (pool->free_aiocb) { 15286bbff9a0Saliguori acb = pool->free_aiocb; 15296bbff9a0Saliguori pool->free_aiocb = acb->next; 1530ce1a14dcSpbrook } else { 15316bbff9a0Saliguori acb = qemu_mallocz(pool->aiocb_size); 15326bbff9a0Saliguori acb->pool = pool; 1533ce1a14dcSpbrook } 1534ce1a14dcSpbrook acb->bs = bs; 1535ce1a14dcSpbrook acb->cb = cb; 1536ce1a14dcSpbrook acb->opaque = opaque; 1537ce1a14dcSpbrook return acb; 1538ce1a14dcSpbrook } 1539ce1a14dcSpbrook 15406bbff9a0Saliguori void *qemu_aio_get(BlockDriverState *bs, BlockDriverCompletionFunc *cb, 15416bbff9a0Saliguori void *opaque) 15426bbff9a0Saliguori { 15436bbff9a0Saliguori return qemu_aio_get_pool(&bs->drv->aio_pool, bs, cb, opaque); 15446bbff9a0Saliguori } 15456bbff9a0Saliguori 1546ce1a14dcSpbrook void qemu_aio_release(void *p) 1547ce1a14dcSpbrook { 15486bbff9a0Saliguori BlockDriverAIOCB *acb = (BlockDriverAIOCB *)p; 15496bbff9a0Saliguori AIOPool *pool = acb->pool; 15506bbff9a0Saliguori acb->next = pool->free_aiocb; 15516bbff9a0Saliguori pool->free_aiocb = acb; 1552ce1a14dcSpbrook } 155319cb3738Sbellard 155419cb3738Sbellard /**************************************************************/ 155519cb3738Sbellard /* removable device support */ 155619cb3738Sbellard 155719cb3738Sbellard /** 155819cb3738Sbellard * Return TRUE if the media is present 155919cb3738Sbellard */ 156019cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs) 156119cb3738Sbellard { 156219cb3738Sbellard BlockDriver *drv = bs->drv; 156319cb3738Sbellard int ret; 156419cb3738Sbellard if (!drv) 156519cb3738Sbellard return 0; 156619cb3738Sbellard if (!drv->bdrv_is_inserted) 156719cb3738Sbellard return 1; 156819cb3738Sbellard ret = drv->bdrv_is_inserted(bs); 156919cb3738Sbellard return ret; 157019cb3738Sbellard } 157119cb3738Sbellard 157219cb3738Sbellard /** 157319cb3738Sbellard * Return TRUE if the media changed since the last call to this 157419cb3738Sbellard * function. It is currently only used for floppy disks 157519cb3738Sbellard */ 157619cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs) 157719cb3738Sbellard { 157819cb3738Sbellard BlockDriver *drv = bs->drv; 157919cb3738Sbellard int ret; 158019cb3738Sbellard 158119cb3738Sbellard if (!drv || !drv->bdrv_media_changed) 158219cb3738Sbellard ret = -ENOTSUP; 158319cb3738Sbellard else 158419cb3738Sbellard ret = drv->bdrv_media_changed(bs); 158519cb3738Sbellard if (ret == -ENOTSUP) 158619cb3738Sbellard ret = bs->media_changed; 158719cb3738Sbellard bs->media_changed = 0; 158819cb3738Sbellard return ret; 158919cb3738Sbellard } 159019cb3738Sbellard 159119cb3738Sbellard /** 159219cb3738Sbellard * If eject_flag is TRUE, eject the media. Otherwise, close the tray 159319cb3738Sbellard */ 159419cb3738Sbellard void bdrv_eject(BlockDriverState *bs, int eject_flag) 159519cb3738Sbellard { 159619cb3738Sbellard BlockDriver *drv = bs->drv; 159719cb3738Sbellard int ret; 159819cb3738Sbellard 159919cb3738Sbellard if (!drv || !drv->bdrv_eject) { 160019cb3738Sbellard ret = -ENOTSUP; 160119cb3738Sbellard } else { 160219cb3738Sbellard ret = drv->bdrv_eject(bs, eject_flag); 160319cb3738Sbellard } 160419cb3738Sbellard if (ret == -ENOTSUP) { 160519cb3738Sbellard if (eject_flag) 160619cb3738Sbellard bdrv_close(bs); 160719cb3738Sbellard } 160819cb3738Sbellard } 160919cb3738Sbellard 161019cb3738Sbellard int bdrv_is_locked(BlockDriverState *bs) 161119cb3738Sbellard { 161219cb3738Sbellard return bs->locked; 161319cb3738Sbellard } 161419cb3738Sbellard 161519cb3738Sbellard /** 161619cb3738Sbellard * Lock or unlock the media (if it is locked, the user won't be able 161719cb3738Sbellard * to eject it manually). 161819cb3738Sbellard */ 161919cb3738Sbellard void bdrv_set_locked(BlockDriverState *bs, int locked) 162019cb3738Sbellard { 162119cb3738Sbellard BlockDriver *drv = bs->drv; 162219cb3738Sbellard 162319cb3738Sbellard bs->locked = locked; 162419cb3738Sbellard if (drv && drv->bdrv_set_locked) { 162519cb3738Sbellard drv->bdrv_set_locked(bs, locked); 162619cb3738Sbellard } 162719cb3738Sbellard } 1628985a03b0Sths 1629985a03b0Sths /* needed for generic scsi interface */ 1630985a03b0Sths 1631985a03b0Sths int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf) 1632985a03b0Sths { 1633985a03b0Sths BlockDriver *drv = bs->drv; 1634985a03b0Sths 1635985a03b0Sths if (drv && drv->bdrv_ioctl) 1636985a03b0Sths return drv->bdrv_ioctl(bs, req, buf); 1637985a03b0Sths return -ENOTSUP; 1638985a03b0Sths } 16397d780669Saliguori 1640221f715dSaliguori BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs, 1641221f715dSaliguori unsigned long int req, void *buf, 16427d780669Saliguori BlockDriverCompletionFunc *cb, void *opaque) 16437d780669Saliguori { 1644221f715dSaliguori BlockDriver *drv = bs->drv; 16457d780669Saliguori 1646221f715dSaliguori if (drv && drv->bdrv_aio_ioctl) 1647221f715dSaliguori return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque); 1648221f715dSaliguori return NULL; 16497d780669Saliguori } 1650e268ca52Saliguori 1651e268ca52Saliguori void *qemu_blockalign(BlockDriverState *bs, size_t size) 1652e268ca52Saliguori { 1653e268ca52Saliguori return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size); 1654e268ca52Saliguori } 1655