block-raw-win32.c

   1 /*
   2  * Block driver for RAW files (win32)
   3  *
   4  * Copyright (c) 2006 Fabrice Bellard
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  */
  24 #include "qemu-common.h"
  25 #include "qemu-timer.h"
  26 #include "block_int.h"
  27 #include <assert.h>
  28 #include <winioctl.h>
  29
  30 //#define WIN32_AIO
  31
  32 #define FTYPE_FILE 0
  33 #define FTYPE_CD     1
  34 #define FTYPE_HARDDISK 2
  35
  36 typedef struct BDRVRawState {
  37     HANDLE hfile;
  38     int type;
  39     char drive_path[16]; /* format: "d:\" */
  40 } BDRVRawState;
  41
  42 typedef struct RawAIOCB {
  43     BlockDriverAIOCB common;
  44     HANDLE hEvent;
  45     OVERLAPPED ov;
  46     int count;
  47 } RawAIOCB;
  48
  49 int qemu_ftruncate64(int fd, int64_t length)
  50 {
  51     LARGE_INTEGER li;
  52     LONG high;
  53     HANDLE h;
  54     BOOL res;
  55
  56     if ((GetVersion() & 0x80000000UL) && (length >> 32) != 0)
  57         return -1;
  58
  59     h = (HANDLE)_get_osfhandle(fd);
  60
  61     /* get current position, ftruncate do not change position */
  62     li.HighPart = 0;
  63     li.LowPart = SetFilePointer (h, 0, &li.HighPart, FILE_CURRENT);
  64     if (li.LowPart == 0xffffffffUL && GetLastError() != NO_ERROR)
  65         return -1;
  66
  67     high = length >> 32;
  68     if (!SetFilePointer(h, (DWORD) length, &high, FILE_BEGIN))
  69         return -1;
  70     res = SetEndOfFile(h);
  71
  72     /* back to old position */
  73     SetFilePointer(h, li.LowPart, &li.HighPart, FILE_BEGIN);
  74     return res ? 0 : -1;
  75 }
  76
  77 static int set_sparse(int fd)
  78 {
  79     DWORD returned;
  80     return (int) DeviceIoControl((HANDLE)_get_osfhandle(fd), FSCTL_SET_SPARSE,
  81                                  NULL, 0, NULL, 0, &returned, NULL);
  82 }
  83
  84 static int raw_open(BlockDriverState *bs, const char *filename, int flags)
  85 {
  86     BDRVRawState *s = bs->opaque;
  87     int access_flags, create_flags;
  88     DWORD overlapped;
  89
  90     s->type = FTYPE_FILE;
  91
  92     if ((flags & BDRV_O_ACCESS) == O_RDWR) {
  93         access_flags = GENERIC_READ | GENERIC_WRITE;
  94     } else {
  95         access_flags = GENERIC_READ;
  96     }
  97     if (flags & BDRV_O_CREAT) {
  98         create_flags = CREATE_ALWAYS;
  99     } else {
 100         create_flags = OPEN_EXISTING;
 101     }
 102 #ifdef WIN32_AIO
 103     overlapped = FILE_FLAG_OVERLAPPED;
 104 #else
 105     overlapped = FILE_ATTRIBUTE_NORMAL;
 106 #endif
 107     if (flags & BDRV_O_DIRECT)
 108         overlapped |= FILE_FLAG_NO_BUFFERING | FILE_FLAG_WRITE_THROUGH;
 109     s->hfile = CreateFile(filename, access_flags,
 110                           FILE_SHARE_READ, NULL,
 111                           create_flags, overlapped, NULL);
 112     if (s->hfile == INVALID_HANDLE_VALUE) {
 113         int err = GetLastError();
 114
 115         if (err == ERROR_ACCESS_DENIED)
 116             return -EACCES;
 117         return -1;
 118     }
 119     return 0;
 120 }
 121
 122 static int raw_pread(BlockDriverState *bs, int64_t offset,
 123                      uint8_t *buf, int count)
 124 {
 125     BDRVRawState *s = bs->opaque;
 126     OVERLAPPED ov;
 127     DWORD ret_count;
 128     int ret;
 129
 130     memset(&ov, 0, sizeof(ov));
 131     ov.Offset = offset;
 132     ov.OffsetHigh = offset >> 32;
 133     ret = ReadFile(s->hfile, buf, count, &ret_count, &ov);
 134     if (!ret) {
 135 #ifdef WIN32_AIO
 136         ret = GetOverlappedResult(s->hfile, &ov, &ret_count, TRUE);
 137         if (!ret)
 138             return -EIO;
 139         else
 140 #endif
 141             return ret_count;
 142     }
 143     return ret_count;
 144 }
 145
 146 static int raw_pwrite(BlockDriverState *bs, int64_t offset,
 147                       const uint8_t *buf, int count)
 148 {
 149     BDRVRawState *s = bs->opaque;
 150     OVERLAPPED ov;
 151     DWORD ret_count;
 152     int ret;
 153
 154     memset(&ov, 0, sizeof(ov));
 155     ov.Offset = offset;
 156     ov.OffsetHigh = offset >> 32;
 157     ret = WriteFile(s->hfile, buf, count, &ret_count, &ov);
 158     if (!ret) {
 159 #ifdef WIN32_AIO
 160         ret = GetOverlappedResult(s->hfile, &ov, &ret_count, TRUE);
 161         if (!ret)
 162             return -EIO;
 163         else
 164 #endif
 165             return ret_count;
 166     }
 167     return ret_count;
 168 }
 169
 170 #ifdef WIN32_AIO
 171 static void raw_aio_cb(void *opaque)
 172 {
 173     RawAIOCB *acb = opaque;
 174     BlockDriverState *bs = acb->common.bs;
 175     BDRVRawState *s = bs->opaque;
 176     DWORD ret_count;
 177     int ret;
 178
 179     ret = GetOverlappedResult(s->hfile, &acb->ov, &ret_count, TRUE);
 180     if (!ret || ret_count != acb->count) {
 181         acb->common.cb(acb->common.opaque, -EIO);
 182     } else {
 183         acb->common.cb(acb->common.opaque, 0);
 184     }
 185 }
 186
 187 static RawAIOCB *raw_aio_setup(BlockDriverState *bs,
 188         int64_t sector_num, uint8_t *buf, int nb_sectors,
 189         BlockDriverCompletionFunc *cb, void *opaque)
 190 {
 191     RawAIOCB *acb;
 192     int64_t offset;
 193
 194     acb = qemu_aio_get(bs, cb, opaque);
 195     if (acb->hEvent) {
 196         acb->hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
 197         if (!acb->hEvent) {
 198             qemu_aio_release(acb);
 199             return NULL;
 200         }
 201     }
 202     memset(&acb->ov, 0, sizeof(acb->ov));
 203     offset = sector_num * 512;
 204     acb->ov.Offset = offset;
 205     acb->ov.OffsetHigh = offset >> 32;
 206     acb->ov.hEvent = acb->hEvent;
 207     acb->count = nb_sectors * 512;
 208     qemu_add_wait_object(acb->ov.hEvent, raw_aio_cb, acb);
 209     return acb;
 210 }
 211
 212 static BlockDriverAIOCB *raw_aio_read(BlockDriverState *bs,
 213         int64_t sector_num, uint8_t *buf, int nb_sectors,
 214         BlockDriverCompletionFunc *cb, void *opaque)
 215 {
 216     BDRVRawState *s = bs->opaque;
 217     RawAIOCB *acb;
 218     int ret;
 219
 220     acb = raw_aio_setup(bs, sector_num, buf, nb_sectors, cb, opaque);
 221     if (!acb)
 222         return NULL;
 223     ret = ReadFile(s->hfile, buf, acb->count, NULL, &acb->ov);
 224     if (!ret) {
 225         qemu_aio_release(acb);
 226         return NULL;
 227     }
 228     qemu_aio_release(acb);
 229     return (BlockDriverAIOCB *)acb;
 230 }
 231
 232 static BlockDriverAIOCB *raw_aio_write(BlockDriverState *bs,
 233         int64_t sector_num, uint8_t *buf, int nb_sectors,
 234         BlockDriverCompletionFunc *cb, void *opaque)
 235 {
 236     BDRVRawState *s = bs->opaque;
 237     RawAIOCB *acb;
 238     int ret;
 239
 240     acb = raw_aio_setup(bs, sector_num, buf, nb_sectors, cb, opaque);
 241     if (!acb)
 242         return NULL;
 243     ret = WriteFile(s->hfile, buf, acb->count, NULL, &acb->ov);
 244     if (!ret) {
 245         qemu_aio_release(acb);
 246         return NULL;
 247     }
 248     qemu_aio_release(acb);
 249     return (BlockDriverAIOCB *)acb;
 250 }
 251
 252 static void raw_aio_cancel(BlockDriverAIOCB *blockacb)
 253 {
 254     RawAIOCB *acb = (RawAIOCB *)blockacb;
 255     BlockDriverState *bs = acb->common.bs;
 256     BDRVRawState *s = bs->opaque;
 257
 258     qemu_del_wait_object(acb->ov.hEvent, raw_aio_cb, acb);
 259     /* XXX: if more than one async I/O it is not correct */
 260     CancelIo(s->hfile);
 261     qemu_aio_release(acb);
 262 }
 263 #endif /* #if WIN32_AIO */
 264
 265 static void raw_flush(BlockDriverState *bs)
 266 {
 267     BDRVRawState *s = bs->opaque;
 268     FlushFileBuffers(s->hfile);
 269 }
 270
 271 static void raw_close(BlockDriverState *bs)
 272 {
 273     BDRVRawState *s = bs->opaque;
 274     CloseHandle(s->hfile);
 275 }
 276
 277 static int raw_truncate(BlockDriverState *bs, int64_t offset)
 278 {
 279     BDRVRawState *s = bs->opaque;
 280     DWORD low, high;
 281
 282     low = offset;
 283     high = offset >> 32;
 284     if (!SetFilePointer(s->hfile, low, &high, FILE_BEGIN))
 285         return -EIO;
 286     if (!SetEndOfFile(s->hfile))
 287         return -EIO;
 288     return 0;
 289 }
 290
 291 static int64_t raw_getlength(BlockDriverState *bs)
 292 {
 293     BDRVRawState *s = bs->opaque;
 294     LARGE_INTEGER l;
 295     ULARGE_INTEGER available, total, total_free;
 296     DISK_GEOMETRY_EX dg;
 297     DWORD count;
 298     BOOL status;
 299
 300     switch(s->type) {
 301     case FTYPE_FILE:
 302         l.LowPart = GetFileSize(s->hfile, &l.HighPart);
 303         if (l.LowPart == 0xffffffffUL && GetLastError() != NO_ERROR)
 304             return -EIO;
 305         break;
 306     case FTYPE_CD:
 307         if (!GetDiskFreeSpaceEx(s->drive_path, &available, &total, &total_free))
 308             return -EIO;
 309         l.QuadPart = total.QuadPart;
 310         break;
 311     case FTYPE_HARDDISK:
 312         status = DeviceIoControl(s->hfile, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX,
 313                                  NULL, 0, &dg, sizeof(dg), &count, NULL);
 314         if (status != 0) {
 315             l = dg.DiskSize;
 316         }
 317         break;
 318     default:
 319         return -EIO;
 320     }
 321     return l.QuadPart;
 322 }
 323
 324 static int raw_create(const char *filename, int64_t total_size,
 325                       const char *backing_file, int flags)
 326 {
 327     int fd;
 328
 329     if (flags || backing_file)
 330         return -ENOTSUP;
 331
 332     fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY,
 333               0644);
 334     if (fd < 0)
 335         return -EIO;
 336     set_sparse(fd);
 337     ftruncate(fd, total_size * 512);
 338     close(fd);
 339     return 0;
 340 }
 341
 342 void qemu_aio_init(void)
 343 {
 344 }
 345
 346 void qemu_aio_flush(void)
 347 {
 348 }
 349
 350 void qemu_aio_wait(void)
 351 {
 352     qemu_bh_poll();
 353 }
 354
 355 BlockDriver bdrv_raw = {
 356     "raw",
 357     sizeof(BDRVRawState),
 358     NULL, /* no probe for protocols */
 359     raw_open,
 360     NULL,
 361     NULL,
 362     raw_close,
 363     raw_create,
 364     raw_flush,
 365
 366 #ifdef WIN32_AIO
 367     .bdrv_aio_read = raw_aio_read,
 368     .bdrv_aio_write = raw_aio_write,
 369     .bdrv_aio_cancel = raw_aio_cancel,
 370     .aiocb_size = sizeof(RawAIOCB);
 371 #endif
 372     .protocol_name = "file",
 373     .bdrv_pread = raw_pread,
 374     .bdrv_pwrite = raw_pwrite,
 375     .bdrv_truncate = raw_truncate,
 376     .bdrv_getlength = raw_getlength,
 377 };
 378
 379 /***********************************************/
 380 /* host device */
 381
 382 static int find_cdrom(char *cdrom_name, int cdrom_name_size)
 383 {
 384     char drives[256], *pdrv = drives;
 385     UINT type;
 386
 387     memset(drives, 0, sizeof(drives));
 388     GetLogicalDriveStrings(sizeof(drives), drives);
 389     while(pdrv[0] != '\0') {
 390         type = GetDriveType(pdrv);
 391         switch(type) {
 392         case DRIVE_CDROM:
 393             snprintf(cdrom_name, cdrom_name_size, "\\\\.\\%c:", pdrv[0]);
 394             return 0;
 395             break;
 396         }
 397         pdrv += lstrlen(pdrv) + 1;
 398     }
 399     return -1;
 400 }
 401
 402 static int find_device_type(BlockDriverState *bs, const char *filename)
 403 {
 404     BDRVRawState *s = bs->opaque;
 405     UINT type;
 406     const char *p;
 407
 408     if (strstart(filename, "\\\\.\\", &p) ||
 409         strstart(filename, "//./", &p)) {
 410         if (stristart(p, "PhysicalDrive", NULL))
 411             return FTYPE_HARDDISK;
 412         snprintf(s->drive_path, sizeof(s->drive_path), "%c:\\", p[0]);
 413         type = GetDriveType(s->drive_path);
 414         if (type == DRIVE_CDROM)
 415             return FTYPE_CD;
 416         else
 417             return FTYPE_FILE;
 418     } else {
 419         return FTYPE_FILE;
 420     }
 421 }
 422
 423 static int hdev_open(BlockDriverState *bs, const char *filename, int flags)
 424 {
 425     BDRVRawState *s = bs->opaque;
 426     int access_flags, create_flags;
 427     DWORD overlapped;
 428     char device_name[64];
 429
 430     if (strstart(filename, "/dev/cdrom", NULL)) {
 431         if (find_cdrom(device_name, sizeof(device_name)) < 0)
 432             return -ENOENT;
 433         filename = device_name;
 434     } else {
 435         /* transform drive letters into device name */
 436         if (((filename[0] >= 'a' && filename[0] <= 'z') ||
 437              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
 438             filename[1] == ':' && filename[2] == '\0') {
 439             snprintf(device_name, sizeof(device_name), "\\\\.\\%c:", filename[0]);
 440             filename = device_name;
 441         }
 442     }
 443     s->type = find_device_type(bs, filename);
 444
 445     if ((flags & BDRV_O_ACCESS) == O_RDWR) {
 446         access_flags = GENERIC_READ | GENERIC_WRITE;
 447     } else {
 448         access_flags = GENERIC_READ;
 449     }
 450     create_flags = OPEN_EXISTING;
 451
 452 #ifdef WIN32_AIO
 453     overlapped = FILE_FLAG_OVERLAPPED;
 454 #else
 455     overlapped = FILE_ATTRIBUTE_NORMAL;
 456 #endif
 457     if (flags & BDRV_O_DIRECT)
 458         overlapped |= FILE_FLAG_NO_BUFFERING | FILE_FLAG_WRITE_THROUGH;
 459     s->hfile = CreateFile(filename, access_flags,
 460                           FILE_SHARE_READ, NULL,
 461                           create_flags, overlapped, NULL);
 462     if (s->hfile == INVALID_HANDLE_VALUE) {
 463         int err = GetLastError();
 464
 465         if (err == ERROR_ACCESS_DENIED)
 466             return -EACCES;
 467         return -1;
 468     }
 469     return 0;
 470 }
 471
 472 #if 0
 473 /***********************************************/
 474 /* removable device additional commands */
 475
 476 static int raw_is_inserted(BlockDriverState *bs)
 477 {
 478     return 1;
 479 }
 480
 481 static int raw_media_changed(BlockDriverState *bs)
 482 {
 483     return -ENOTSUP;
 484 }
 485
 486 static int raw_eject(BlockDriverState *bs, int eject_flag)
 487 {
 488     DWORD ret_count;
 489
 490     if (s->type == FTYPE_FILE)
 491         return -ENOTSUP;
 492     if (eject_flag) {
 493         DeviceIoControl(s->hfile, IOCTL_STORAGE_EJECT_MEDIA,
 494                         NULL, 0, NULL, 0, &lpBytesReturned, NULL);
 495     } else {
 496         DeviceIoControl(s->hfile, IOCTL_STORAGE_LOAD_MEDIA,
 497                         NULL, 0, NULL, 0, &lpBytesReturned, NULL);
 498     }
 499 }
 500
 501 static int raw_set_locked(BlockDriverState *bs, int locked)
 502 {
 503     return -ENOTSUP;
 504 }
 505 #endif
 506
 507 BlockDriver bdrv_host_device = {
 508     "host_device",
 509     sizeof(BDRVRawState),
 510     NULL, /* no probe for protocols */
 511     hdev_open,
 512     NULL,
 513     NULL,
 514     raw_close,
 515     NULL,
 516     raw_flush,
 517
 518 #ifdef WIN32_AIO
 519     .bdrv_aio_read = raw_aio_read,
 520     .bdrv_aio_write = raw_aio_write,
 521     .bdrv_aio_cancel = raw_aio_cancel,
 522     .aiocb_size = sizeof(RawAIOCB);
 523 #endif
 524     .bdrv_pread = raw_pread,
 525     .bdrv_pwrite = raw_pwrite,
 526     .bdrv_getlength = raw_getlength,
 527 };