Changeset 2187 for trunk/eraser/Eraser.Util/PhysicalDriveInfo.cs

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)

File:

• trunk/eraser/Eraser.Util/PhysicalDriveInfo.cs (modified) (5 diffs)

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>