Changeset 30185 in vbox

Timestamp:

Jun 14, 2010 8:38:54 PM (15 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

62642

Message:

AsyncCompletion/File: Quick fix for raw disks on Windows

File:

• trunk/src/VBox/VMM/PDMAsyncCompletionFile.cpp (modified) (4 diffs)

trunk/src/VBox/VMM/PDMAsyncCompletionFile.cpp

@@ -38 +38 @@
 #include <iprt/thread.h>
 #include <iprt/path.h>
+
+#ifdef RT_OS_WINDOWS
+# define _WIN32_WINNT 0x0500
+# include <windows.h>
+# include <winioctl.h>
+#endif
 
 #include "PDMAsyncCompletionFileInternal.h"
@@ -598 +604 @@
 }
 
+/**
+ * Get the size of the given file.
+ * Works for block devices too.
+ *
+ * @returns VBox status code.
+ * @param   hFile    The file handle.
+ * @param   pcbSize  Where to store the size of the file on success.
+ */
+static int pdmacFileEpNativeGetSize(RTFILE hFile, uint64_t *pcbSize)
+{
+    int rc = VINF_SUCCESS;
+    uint64_t cbSize = 0;
+
+    rc = RTFileGetSize(hFile, &cbSize);
+    if (RT_SUCCESS(rc) && (cbSize != 0))
+        *pcbSize = cbSize;
+    else
+    {
+#ifdef RT_OS_WINDOWS
+        DISK_GEOMETRY DriveGeo;
+        DWORD cbDriveGeo;
+        if (DeviceIoControl((HANDLE)hFile,
+                            IOCTL_DISK_GET_DRIVE_GEOMETRY, NULL, 0,
+                            &DriveGeo, sizeof(DriveGeo), &cbDriveGeo, NULL))
+        {
+            if (   DriveGeo.MediaType == FixedMedia
+                || DriveGeo.MediaType == RemovableMedia)
+            {
+                cbSize =     DriveGeo.Cylinders.QuadPart
+                         *   DriveGeo.TracksPerCylinder
+                         *   DriveGeo.SectorsPerTrack
+                         *   DriveGeo.BytesPerSector;
+
+                GET_LENGTH_INFORMATION DiskLenInfo;
+                DWORD junk;
+                if (DeviceIoControl((HANDLE)hFile,
+                                    IOCTL_DISK_GET_LENGTH_INFO, NULL, 0,
+                                    &DiskLenInfo, sizeof(DiskLenInfo), &junk, (LPOVERLAPPED)NULL))
+                {
+                    /* IOCTL_DISK_GET_LENGTH_INFO is supported -- override cbSize. */
+                    cbSize = DiskLenInfo.Length.QuadPart;
+                }
+
+                rc = VINF_SUCCESS;
+            }
+            else
+            {
+                rc = VERR_INVALID_PARAMETER;
+            }
+        }
+        else
+        {
+            rc = RTErrConvertFromWin32(GetLastError());
+        }
+#else
+        /* Could be a block device */
+        rc = RTFileSeek(hFile, 0, RTFILE_SEEK_END, &cbSize);
+
+#endif
+        if (RT_SUCCESS(rc) && (cbSize != 0))
+            *pcbSize = cbSize;
+        else if (RT_SUCCESS(rc))
+            rc = VERR_NOT_SUPPORTED;
+    }
+
+    return rc;
+}
+
 static int pdmacFileInitialize(PPDMASYNCCOMPLETIONEPCLASS pClassGlobals, PCFGMNODE pCfgNode)
 {
@@ -748 +822 @@
             uint64_t cbSize;
 
-            rc = RTFileGetSize(File, &cbSize);
+            rc = pdmacFileEpNativeGetSize(File, &cbSize);
+            Assert(RT_FAILURE(rc) || cbSize != 0);
+
             if (RT_SUCCESS(rc) && ((cbSize % 512) == 0))
                 fFileFlags |= RTFILE_O_NO_CACHE;
@@ -804 +880 @@
         pEpFile->fFlags = fFileFlags;
 
-        rc = RTFileGetSize(pEpFile->File, (uint64_t *)&pEpFile->cbFile);
-        if (RT_SUCCESS(rc) && (pEpFile->cbFile == 0))
-        {
-            /* Could be a block device */
-            rc = RTFileSeek(pEpFile->File, 0, RTFILE_SEEK_END, (uint64_t *)&pEpFile->cbFile);
-        }
+        rc = pdmacFileEpNativeGetSize(pEpFile->File, (uint64_t *)&pEpFile->cbFile);
+        Assert(RT_FAILURE(rc) || pEpFile->cbFile != 0);
 
         if (RT_SUCCESS(rc))