Changeset 2187

Timestamp:

6/18/2010 11:33:59 AM (2 years ago)

Author:

lowjoel

Message:

Allow us to query the size of non-ready partitions and disks (for more informative UIs)

Location:

trunk/eraser/Eraser.Util

Files:

• NativeMethods/Kernel.cs (modified) (1 diff)
• PhysicalDriveInfo.cs (modified) (5 diffs)
• VolumeInfo.cs (modified) (2 diffs)

trunk/eraser/Eraser.Util/NativeMethods/Kernel.cs

@@ -675 +675 @@
         }
 
+        [DllImport("Kernel32.dll", SetLastError = true)]
+        [return: MarshalAs(UnmanagedType.Bool)]
+        private extern static bool DeviceIoControl(SafeFileHandle hDevice,
+            uint dwIoControlCode, IntPtr lpInBuffer, uint nInBufferSize,
+            out long lpOutBuffer, uint nOutBufferSize, out uint lpBytesReturned,
+            IntPtr lpOverlapped);
+
+        public static bool DeviceIoControl(SafeFileHandle hDevice,
+            uint dwIoControlCode, IntPtr lpInBuffer, uint nInBufferSize,
+            out long lpOutBuffer, out uint lpBytesReturned, IntPtr lpOverlapped)
+        {
+            return DeviceIoControl(hDevice, dwIoControlCode, lpInBuffer, nInBufferSize,
+                out lpOutBuffer, sizeof(long), out lpBytesReturned, lpOverlapped);
+        }
+
+        /// <summary>
+        /// Retrieves the length of the specified disk, volume, or partition.
+        /// </summary>
+        public const int IOCTL_DISK_GET_LENGTH_INFO =
+            (0x00000007 << 16) | (0x0001 << 14) | (0x0017 << 2);
+
         [DllImport("Kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true)]
         [return: MarshalAs(UnmanagedType.Bool)]

trunk/eraser/Eraser.Util/PhysicalDriveInfo.cs

@@ -65 +65 @@
                 for (int i = 0; ; ++i)
                 {
-                    using (SafeFileHandle handle = OpenWin32Device(GetDiskPath(i)))
+                    using (SafeFileHandle handle = OpenWin32Device(GetDiskPath(i),
+                        NativeMethods.FILE_READ_ATTRIBUTES))
                     {
                         if (handle.IsInvalid)
@@ -82 +83 @@
         /// </summary>
         /// <param name="deviceName">The name of the device to open.</param>
+        /// <param name="access">The access needed for the handle.</param>
         /// <returns>A <see cref="SafeFileHandle"/> to the device.</returns>
-        private static SafeFileHandle OpenWin32Device(string deviceName)
+        private static SafeFileHandle OpenWin32Device(string deviceName, uint access)
         {
             //Define the DOS device name for access
@@ -100 +102 @@
                 //Open the device handle.
                 return NativeMethods.CreateFile(string.Format(CultureInfo.InvariantCulture,
-                    "\\\\.\\{0}", dosDeviceName), NativeMethods.FILE_READ_ATTRIBUTES,
+                    "\\\\.\\{0}", dosDeviceName), access,
                     NativeMethods.FILE_SHARE_READ | NativeMethods.FILE_SHARE_WRITE, IntPtr.Zero,
                     (int)FileMode.Open, (uint)FileAttributes.ReadOnly, IntPtr.Zero);
@@ -131 +133 @@
                 {
                     string path = GetPartitionPath(i);
-                    using (SafeFileHandle handle = OpenWin32Device(path))
+                    using (SafeFileHandle handle = OpenWin32Device(path,
+                        NativeMethods.FILE_READ_ATTRIBUTES))
                     {
                         if (handle.IsInvalid)
@@ -161 +164 @@
 
         /// <summary>
+        /// Gets the size of the disk, in bytes.
+        /// </summary>
+        public long Size
+        {
+            get
+            {
+                using (SafeFileHandle handle = OpenWin32Device(GetDiskPath(),
+                    NativeMethods.GENERIC_READ))
+                {
+                    if (handle.IsInvalid)
+                        throw Win32ErrorCode.GetExceptionForWin32Error(Marshal.GetLastWin32Error());
+
+                    long result = 0;
+                    uint returned = 0;
+                    if (NativeMethods.DeviceIoControl(handle, NativeMethods.IOCTL_DISK_GET_LENGTH_INFO,
+                        IntPtr.Zero, 0, out result, out returned, IntPtr.Zero))
+                    {
+                        return result;
+                    }
+
+                    throw Win32ErrorCode.GetExceptionForWin32Error(Marshal.GetLastWin32Error());
+                }
+            }
+        }
+
+        /// <summary>
         /// The format string for accessing partitions.
         /// </summary>

trunk/eraser/Eraser.Util/VolumeInfo.cs

@@ -463 +463 @@
             get
             {
-                if (!IsReady)
-                    throw new InvalidOperationException("The volume has not been mounted or is not " +
-                        "currently ready.");
-
                 ulong result, dummy;
                 if (NativeMethods.GetDiskFreeSpaceEx(VolumeId, out dummy, out result, out dummy))
@@ -473 +469 @@
                 }
 
+                //Try the alternative method
+                using (SafeFileHandle handle = OpenHandle(NativeMethods.GENERIC_READ,
+                    FileShare.ReadWrite, FileOptions.None))
+                {
+                    if (handle.IsInvalid)
+                        throw Win32ErrorCode.GetExceptionForWin32Error(Marshal.GetLastWin32Error());
+
+                    long result2;
+                    uint returned = 0;
+                    if (NativeMethods.DeviceIoControl(handle,
+                        NativeMethods.IOCTL_DISK_GET_LENGTH_INFO, IntPtr.Zero, 0, out result2,
+                        out returned, IntPtr.Zero))
+                    {
+                        return result2;
+                    }
+                }
+
+                //Otherwise, throw
                 throw Win32ErrorCode.GetExceptionForWin32Error(Marshal.GetLastWin32Error());
             }