File:  [Qemu by Fabrice Bellard] / qemu / block-raw-win32.c
Revision 1.1.1.2 (vendor branch): download - view: text, annotated - select for diffs
Tue Apr 24 16:50:53 2018 UTC (2 years, 6 months ago) by root
Branches: qemu, MAIN
CVS tags: qemu0101, qemu0100, HEAD
qemu 0.10.0

    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_NOCACHE))
  108:         overlapped |= FILE_FLAG_NO_BUFFERING | FILE_FLAG_WRITE_THROUGH;
  109:     else if (!(flags & BDRV_O_CACHE_WB))
  110:         overlapped |= FILE_FLAG_WRITE_THROUGH;
  111:     s->hfile = CreateFile(filename, access_flags,
  112:                           FILE_SHARE_READ, NULL,
  113:                           create_flags, overlapped, NULL);
  114:     if (s->hfile == INVALID_HANDLE_VALUE) {
  115:         int err = GetLastError();
  116: 
  117:         if (err == ERROR_ACCESS_DENIED)
  118:             return -EACCES;
  119:         return -1;
  120:     }
  121:     return 0;
  122: }
  123: 
  124: static int raw_pread(BlockDriverState *bs, int64_t offset,
  125:                      uint8_t *buf, int count)
  126: {
  127:     BDRVRawState *s = bs->opaque;
  128:     OVERLAPPED ov;
  129:     DWORD ret_count;
  130:     int ret;
  131: 
  132:     memset(&ov, 0, sizeof(ov));
  133:     ov.Offset = offset;
  134:     ov.OffsetHigh = offset >> 32;
  135:     ret = ReadFile(s->hfile, buf, count, &ret_count, &ov);
  136:     if (!ret) {
  137: #ifdef WIN32_AIO
  138:         ret = GetOverlappedResult(s->hfile, &ov, &ret_count, TRUE);
  139:         if (!ret)
  140:             return -EIO;
  141:         else
  142: #endif
  143:             return ret_count;
  144:     }
  145:     return ret_count;
  146: }
  147: 
  148: static int raw_pwrite(BlockDriverState *bs, int64_t offset,
  149:                       const uint8_t *buf, int count)
  150: {
  151:     BDRVRawState *s = bs->opaque;
  152:     OVERLAPPED ov;
  153:     DWORD ret_count;
  154:     int ret;
  155: 
  156:     memset(&ov, 0, sizeof(ov));
  157:     ov.Offset = offset;
  158:     ov.OffsetHigh = offset >> 32;
  159:     ret = WriteFile(s->hfile, buf, count, &ret_count, &ov);
  160:     if (!ret) {
  161: #ifdef WIN32_AIO
  162:         ret = GetOverlappedResult(s->hfile, &ov, &ret_count, TRUE);
  163:         if (!ret)
  164:             return -EIO;
  165:         else
  166: #endif
  167:             return ret_count;
  168:     }
  169:     return ret_count;
  170: }
  171: 
  172: #ifdef WIN32_AIO
  173: static void raw_aio_cb(void *opaque)
  174: {
  175:     RawAIOCB *acb = opaque;
  176:     BlockDriverState *bs = acb->common.bs;
  177:     BDRVRawState *s = bs->opaque;
  178:     DWORD ret_count;
  179:     int ret;
  180: 
  181:     ret = GetOverlappedResult(s->hfile, &acb->ov, &ret_count, TRUE);
  182:     if (!ret || ret_count != acb->count) {
  183:         acb->common.cb(acb->common.opaque, -EIO);
  184:     } else {
  185:         acb->common.cb(acb->common.opaque, 0);
  186:     }
  187: }
  188: 
  189: static RawAIOCB *raw_aio_setup(BlockDriverState *bs,
  190:         int64_t sector_num, uint8_t *buf, int nb_sectors,
  191:         BlockDriverCompletionFunc *cb, void *opaque)
  192: {
  193:     RawAIOCB *acb;
  194:     int64_t offset;
  195: 
  196:     acb = qemu_aio_get(bs, cb, opaque);
  197:     if (acb->hEvent) {
  198:         acb->hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
  199:         if (!acb->hEvent) {
  200:             qemu_aio_release(acb);
  201:             return NULL;
  202:         }
  203:     }
  204:     memset(&acb->ov, 0, sizeof(acb->ov));
  205:     offset = sector_num * 512;
  206:     acb->ov.Offset = offset;
  207:     acb->ov.OffsetHigh = offset >> 32;
  208:     acb->ov.hEvent = acb->hEvent;
  209:     acb->count = nb_sectors * 512;
  210:     qemu_add_wait_object(acb->ov.hEvent, raw_aio_cb, acb);
  211:     return acb;
  212: }
  213: 
  214: static BlockDriverAIOCB *raw_aio_read(BlockDriverState *bs,
  215:         int64_t sector_num, uint8_t *buf, int nb_sectors,
  216:         BlockDriverCompletionFunc *cb, void *opaque)
  217: {
  218:     BDRVRawState *s = bs->opaque;
  219:     RawAIOCB *acb;
  220:     int ret;
  221: 
  222:     acb = raw_aio_setup(bs, sector_num, buf, nb_sectors, cb, opaque);
  223:     if (!acb)
  224:         return NULL;
  225:     ret = ReadFile(s->hfile, buf, acb->count, NULL, &acb->ov);
  226:     if (!ret) {
  227:         qemu_aio_release(acb);
  228:         return NULL;
  229:     }
  230:     qemu_aio_release(acb);
  231:     return (BlockDriverAIOCB *)acb;
  232: }
  233: 
  234: static BlockDriverAIOCB *raw_aio_write(BlockDriverState *bs,
  235:         int64_t sector_num, uint8_t *buf, int nb_sectors,
  236:         BlockDriverCompletionFunc *cb, void *opaque)
  237: {
  238:     BDRVRawState *s = bs->opaque;
  239:     RawAIOCB *acb;
  240:     int ret;
  241: 
  242:     acb = raw_aio_setup(bs, sector_num, buf, nb_sectors, cb, opaque);
  243:     if (!acb)
  244:         return NULL;
  245:     ret = WriteFile(s->hfile, buf, acb->count, NULL, &acb->ov);
  246:     if (!ret) {
  247:         qemu_aio_release(acb);
  248:         return NULL;
  249:     }
  250:     qemu_aio_release(acb);
  251:     return (BlockDriverAIOCB *)acb;
  252: }
  253: 
  254: static void raw_aio_cancel(BlockDriverAIOCB *blockacb)
  255: {
  256:     RawAIOCB *acb = (RawAIOCB *)blockacb;
  257:     BlockDriverState *bs = acb->common.bs;
  258:     BDRVRawState *s = bs->opaque;
  259: 
  260:     qemu_del_wait_object(acb->ov.hEvent, raw_aio_cb, acb);
  261:     /* XXX: if more than one async I/O it is not correct */
  262:     CancelIo(s->hfile);
  263:     qemu_aio_release(acb);
  264: }
  265: #endif /* #if WIN32_AIO */
  266: 
  267: static void raw_flush(BlockDriverState *bs)
  268: {
  269:     BDRVRawState *s = bs->opaque;
  270:     FlushFileBuffers(s->hfile);
  271: }
  272: 
  273: static void raw_close(BlockDriverState *bs)
  274: {
  275:     BDRVRawState *s = bs->opaque;
  276:     CloseHandle(s->hfile);
  277: }
  278: 
  279: static int raw_truncate(BlockDriverState *bs, int64_t offset)
  280: {
  281:     BDRVRawState *s = bs->opaque;
  282:     DWORD low, high;
  283: 
  284:     low = offset;
  285:     high = offset >> 32;
  286:     if (!SetFilePointer(s->hfile, low, &high, FILE_BEGIN))
  287: 	return -EIO;
  288:     if (!SetEndOfFile(s->hfile))
  289:         return -EIO;
  290:     return 0;
  291: }
  292: 
  293: static int64_t raw_getlength(BlockDriverState *bs)
  294: {
  295:     BDRVRawState *s = bs->opaque;
  296:     LARGE_INTEGER l;
  297:     ULARGE_INTEGER available, total, total_free;
  298:     DISK_GEOMETRY_EX dg;
  299:     DWORD count;
  300:     BOOL status;
  301: 
  302:     switch(s->type) {
  303:     case FTYPE_FILE:
  304:         l.LowPart = GetFileSize(s->hfile, &l.HighPart);
  305:         if (l.LowPart == 0xffffffffUL && GetLastError() != NO_ERROR)
  306:             return -EIO;
  307:         break;
  308:     case FTYPE_CD:
  309:         if (!GetDiskFreeSpaceEx(s->drive_path, &available, &total, &total_free))
  310:             return -EIO;
  311:         l.QuadPart = total.QuadPart;
  312:         break;
  313:     case FTYPE_HARDDISK:
  314:         status = DeviceIoControl(s->hfile, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX,
  315:                                  NULL, 0, &dg, sizeof(dg), &count, NULL);
  316:         if (status != 0) {
  317:             l = dg.DiskSize;
  318:         }
  319:         break;
  320:     default:
  321:         return -EIO;
  322:     }
  323:     return l.QuadPart;
  324: }
  325: 
  326: static int raw_create(const char *filename, int64_t total_size,
  327:                       const char *backing_file, int flags)
  328: {
  329:     int fd;
  330: 
  331:     if (flags || backing_file)
  332:         return -ENOTSUP;
  333: 
  334:     fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY,
  335:               0644);
  336:     if (fd < 0)
  337:         return -EIO;
  338:     set_sparse(fd);
  339:     ftruncate(fd, total_size * 512);
  340:     close(fd);
  341:     return 0;
  342: }
  343: 
  344: BlockDriver bdrv_raw = {
  345:     "raw",
  346:     sizeof(BDRVRawState),
  347:     NULL, /* no probe for protocols */
  348:     raw_open,
  349:     NULL,
  350:     NULL,
  351:     raw_close,
  352:     raw_create,
  353:     raw_flush,
  354: 
  355: #ifdef WIN32_AIO
  356:     .bdrv_aio_read = raw_aio_read,
  357:     .bdrv_aio_write = raw_aio_write,
  358:     .bdrv_aio_cancel = raw_aio_cancel,
  359:     .aiocb_size = sizeof(RawAIOCB);
  360: #endif
  361:     .bdrv_pread = raw_pread,
  362:     .bdrv_pwrite = raw_pwrite,
  363:     .bdrv_truncate = raw_truncate,
  364:     .bdrv_getlength = raw_getlength,
  365: };
  366: 
  367: /***********************************************/
  368: /* host device */
  369: 
  370: static int find_cdrom(char *cdrom_name, int cdrom_name_size)
  371: {
  372:     char drives[256], *pdrv = drives;
  373:     UINT type;
  374: 
  375:     memset(drives, 0, sizeof(drives));
  376:     GetLogicalDriveStrings(sizeof(drives), drives);
  377:     while(pdrv[0] != '\0') {
  378:         type = GetDriveType(pdrv);
  379:         switch(type) {
  380:         case DRIVE_CDROM:
  381:             snprintf(cdrom_name, cdrom_name_size, "\\\\.\\%c:", pdrv[0]);
  382:             return 0;
  383:             break;
  384:         }
  385:         pdrv += lstrlen(pdrv) + 1;
  386:     }
  387:     return -1;
  388: }
  389: 
  390: static int find_device_type(BlockDriverState *bs, const char *filename)
  391: {
  392:     BDRVRawState *s = bs->opaque;
  393:     UINT type;
  394:     const char *p;
  395: 
  396:     if (strstart(filename, "\\\\.\\", &p) ||
  397:         strstart(filename, "//./", &p)) {
  398:         if (stristart(p, "PhysicalDrive", NULL))
  399:             return FTYPE_HARDDISK;
  400:         snprintf(s->drive_path, sizeof(s->drive_path), "%c:\\", p[0]);
  401:         type = GetDriveType(s->drive_path);
  402:         if (type == DRIVE_CDROM)
  403:             return FTYPE_CD;
  404:         else
  405:             return FTYPE_FILE;
  406:     } else {
  407:         return FTYPE_FILE;
  408:     }
  409: }
  410: 
  411: static int hdev_open(BlockDriverState *bs, const char *filename, int flags)
  412: {
  413:     BDRVRawState *s = bs->opaque;
  414:     int access_flags, create_flags;
  415:     DWORD overlapped;
  416:     char device_name[64];
  417: 
  418:     if (strstart(filename, "/dev/cdrom", NULL)) {
  419:         if (find_cdrom(device_name, sizeof(device_name)) < 0)
  420:             return -ENOENT;
  421:         filename = device_name;
  422:     } else {
  423:         /* transform drive letters into device name */
  424:         if (((filename[0] >= 'a' && filename[0] <= 'z') ||
  425:              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
  426:             filename[1] == ':' && filename[2] == '\0') {
  427:             snprintf(device_name, sizeof(device_name), "\\\\.\\%c:", filename[0]);
  428:             filename = device_name;
  429:         }
  430:     }
  431:     s->type = find_device_type(bs, filename);
  432: 
  433:     if ((flags & BDRV_O_ACCESS) == O_RDWR) {
  434:         access_flags = GENERIC_READ | GENERIC_WRITE;
  435:     } else {
  436:         access_flags = GENERIC_READ;
  437:     }
  438:     create_flags = OPEN_EXISTING;
  439: 
  440: #ifdef WIN32_AIO
  441:     overlapped = FILE_FLAG_OVERLAPPED;
  442: #else
  443:     overlapped = FILE_ATTRIBUTE_NORMAL;
  444: #endif
  445:     if ((flags & BDRV_O_NOCACHE))
  446:         overlapped |= FILE_FLAG_NO_BUFFERING | FILE_FLAG_WRITE_THROUGH;
  447:     else if (!(flags & BDRV_O_CACHE_WB))
  448:         overlapped |= FILE_FLAG_WRITE_THROUGH;
  449:     s->hfile = CreateFile(filename, access_flags,
  450:                           FILE_SHARE_READ, NULL,
  451:                           create_flags, overlapped, NULL);
  452:     if (s->hfile == INVALID_HANDLE_VALUE) {
  453:         int err = GetLastError();
  454: 
  455:         if (err == ERROR_ACCESS_DENIED)
  456:             return -EACCES;
  457:         return -1;
  458:     }
  459:     return 0;
  460: }
  461: 
  462: #if 0
  463: /***********************************************/
  464: /* removable device additional commands */
  465: 
  466: static int raw_is_inserted(BlockDriverState *bs)
  467: {
  468:     return 1;
  469: }
  470: 
  471: static int raw_media_changed(BlockDriverState *bs)
  472: {
  473:     return -ENOTSUP;
  474: }
  475: 
  476: static int raw_eject(BlockDriverState *bs, int eject_flag)
  477: {
  478:     DWORD ret_count;
  479: 
  480:     if (s->type == FTYPE_FILE)
  481:         return -ENOTSUP;
  482:     if (eject_flag) {
  483:         DeviceIoControl(s->hfile, IOCTL_STORAGE_EJECT_MEDIA,
  484:                         NULL, 0, NULL, 0, &lpBytesReturned, NULL);
  485:     } else {
  486:         DeviceIoControl(s->hfile, IOCTL_STORAGE_LOAD_MEDIA,
  487:                         NULL, 0, NULL, 0, &lpBytesReturned, NULL);
  488:     }
  489: }
  490: 
  491: static int raw_set_locked(BlockDriverState *bs, int locked)
  492: {
  493:     return -ENOTSUP;
  494: }
  495: #endif
  496: 
  497: BlockDriver bdrv_host_device = {
  498:     "host_device",
  499:     sizeof(BDRVRawState),
  500:     NULL, /* no probe for protocols */
  501:     hdev_open,
  502:     NULL,
  503:     NULL,
  504:     raw_close,
  505:     NULL,
  506:     raw_flush,
  507: 
  508: #ifdef WIN32_AIO
  509:     .bdrv_aio_read = raw_aio_read,
  510:     .bdrv_aio_write = raw_aio_write,
  511:     .bdrv_aio_cancel = raw_aio_cancel,
  512:     .aiocb_size = sizeof(RawAIOCB);
  513: #endif
  514:     .bdrv_pread = raw_pread,
  515:     .bdrv_pwrite = raw_pwrite,
  516:     .bdrv_getlength = raw_getlength,
  517: };

unix.superglobalmegacorp.com