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