internal static unsafe byte *ReadSectorDataPointerFromHandle(IntPtr handle, ulong startSector, ulong numSectors, ulong sectorSize) { uint bytesRead; fixed(byte *data = new byte[sectorSize *numSectors]) { LARGE_INTEGER li; li.LowPart = 0; li.QuadPart = (long)(startSector * sectorSize); NativeDisk.SetFilePointer(handle, li.LowPart, out li.HighPart, EMoveMethod.Begin); if (!NativeDisk.ReadFile(handle, new IntPtr(data), (uint)(sectorSize * numSectors), out bytesRead, IntPtr.Zero)) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to read sector data pointer from handle.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } if (bytesRead < (sectorSize * numSectors)) { for (uint i = bytesRead; i < (sectorSize * numSectors); i++) { data[i] = 0; } } return(data); } }
internal static Task <int> ReadSectorDataFromHandleAsync(IntPtr handle, byte[] data, ulong startSector, ulong numSectors, ulong sectorSize) { return(Task.Run(() => { uint bytesRead; LARGE_INTEGER li; li.LowPart = 0; li.QuadPart = (long)(startSector * sectorSize); NativeDisk.SetFilePointer(handle, li.LowPart, out li.HighPart, EMoveMethod.Begin); if (!NativeDisk.ReadFile(handle, data, (uint)(sectorSize * numSectors), out bytesRead, IntPtr.Zero)) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to read sector data from handle async.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } if (bytesRead < (sectorSize * numSectors)) { for (uint i = bytesRead; i < (sectorSize * numSectors); i++) { data[i] = 0; } } return (int)bytesRead; })); }
public void Dispose() { for (int i = 0; i < volumeHandles.Length; i++) { if (volumeHandles[i] != IntPtr.Zero) { try { NativeDiskWrapper.RemoveLockOnVolume(volumeHandles[i]); } catch { } NativeDisk.CloseHandle(volumeHandles[i]); volumeHandles[i] = IntPtr.Zero; } } if (fileHandle != IntPtr.Zero) { NativeDisk.CloseHandle(fileHandle); fileHandle = IntPtr.Zero; } for (int i = 0; i < deviceHandles.Length; i++) { if (deviceHandles[i] != IntPtr.Zero) { NativeDisk.CloseHandle(deviceHandles[i]); deviceHandles[i] = IntPtr.Zero; } } Utils.AllowComputerSleep(); }
internal static bool IsVolumeMounted(IntPtr handle) { int bytesReturned = 0; bool result; result = NativeDisk.DeviceIoControl(handle, NativeDisk.FSCTL_IS_VOLUME_MOUNTED, IntPtr.Zero, 0, IntPtr.Zero, 0, ref bytesReturned, IntPtr.Zero); return(!result); }
internal static string GetDriveLabel(string drive) { StringBuilder sb = new StringBuilder(261); if (!NativeDisk.GetVolumeInformation(drive, sb, 261, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero, 0)) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get drive label.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } return(sb.ToString()); }
internal static void DiskUpdateProperties(IntPtr handle) { int bytesReturned = 0; var result = NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_DISK_UPDATE_PROPERTIES, IntPtr.Zero, 0, IntPtr.Zero, 0, ref bytesReturned, IntPtr.Zero); if (!result) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to update disk properties.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } }
internal static IntPtr GetHandleOnFile(string filePath, uint access) { IntPtr _handle; _handle = NativeDisk.CreateFileW(filePath, access, (access == NativeDisk.GENERIC_READ) ? NativeDisk.FILE_SHARE_READ : 0U, IntPtr.Zero, access == NativeDisk.GENERIC_READ ? NativeDisk.OPEN_EXISTING : NativeDisk.CREATE_ALWAYS, 0, IntPtr.Zero); if (_handle == NativeDisk.INVALID_HANDLE_VALUE) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get handle on file.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } return(_handle); }
public static ulong GetDeviceLength(IntPtr handle) { ulong length = 0; unsafe { int returnLength = 0; IntPtr lengthPtr = new IntPtr(&length); NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_DISK_GET_LENGTH_INFO, IntPtr.Zero, 0, lengthPtr, sizeof(ulong), ref returnLength, IntPtr.Zero); } return(length); }
internal static bool CheckReadOnly(string drive) { StringBuilder sb = new StringBuilder(261); FileSystemFeature flags; if (!NativeDisk.GetVolumeInformation(drive, sb, 261, IntPtr.Zero, IntPtr.Zero, out flags, IntPtr.Zero, 0)) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to check if device is read only.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } return((flags & FileSystemFeature.ReadOnlyVolume) > 0); }
internal static bool SpaceAvailible(string location, ulong spaceNeeded, out ulong freeSpace) { ulong foo = 0; bool result; result = NativeDisk.GetDiskFreeSpaceEx(location, out foo, out foo, out freeSpace); if (!result) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get space availible.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } return(spaceNeeded <= freeSpace); }
internal static IntPtr GetHandleOnVolume(int volume, uint access) { IntPtr _handle; _handle = NativeDisk.CreateFile(string.Format(@"\\.\{0}:", (char)(volume + 'A')), access, NativeDisk.FILE_SHARE_READ | NativeDisk.FILE_SHARE_WRITE, IntPtr.Zero, NativeDisk.OPEN_EXISTING, 0, IntPtr.Zero); if (_handle == NativeDisk.INVALID_HANDLE_VALUE) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get handle on volume.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } return(_handle); }
internal static STORAGE_DEVICE_NUMBER GetDiskProperty(IntPtr handle, ref STORAGE_DEVICE_DESCRIPTOR devDescriptor) { STORAGE_PROPERTY_QUERY query = new STORAGE_PROPERTY_QUERY(); STORAGE_DEVICE_NUMBER devInfo = new STORAGE_DEVICE_NUMBER(); int outBytes = 0; bool result; IntPtr queryStructPtr; IntPtr devDescriptorStructPtr; IntPtr devInfoStructPtr; query.PropertyId = STORAGE_PROPERTY_ID.StorageDeviceProperty; query.QueryType = STORAGE_QUERY_TYPE.PropertyStandardQuery; queryStructPtr = NativeDisk.StructToPtr(query); devDescriptorStructPtr = NativeDisk.StructToPtr(devDescriptor); devInfoStructPtr = NativeDisk.StructToPtr(devInfo); result = NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_STORAGE_QUERY_PROPERTY, queryStructPtr, Marshal.SizeOf(query), devDescriptorStructPtr, Marshal.SizeOf(devDescriptor), ref outBytes, IntPtr.Zero); Marshal.FreeHGlobal(queryStructPtr); Marshal.FreeHGlobal(devDescriptorStructPtr); if (result) { result = NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_STORAGE_GET_DEVICE_NUMBER, IntPtr.Zero, 0, devInfoStructPtr, Marshal.SizeOf(devInfo), ref outBytes, IntPtr.Zero); if (!result) { Marshal.FreeHGlobal(devInfoStructPtr); var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get disk property IOCTL_STORAGE_GET_DEVICE_NUMBER.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } } else { if (NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_STORAGE_CHECK_VERIFY2, IntPtr.Zero, 0, IntPtr.Zero, 0, ref outBytes, IntPtr.Zero)) { Marshal.FreeHGlobal(devInfoStructPtr); var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get disk property IOCTL_STORAGE_CHECK_VERIFY2.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } } var resultStruct = Marshal.PtrToStructure <STORAGE_DEVICE_NUMBER>(devInfoStructPtr); Marshal.FreeHGlobal(devInfoStructPtr); return(resultStruct); }
internal static bool UnmountVolume(IntPtr handle) { int bytesReturned = 0; bool result; result = NativeDisk.DeviceIoControl(handle, NativeDisk.FSCTL_DISMOUNT_VOLUME, IntPtr.Zero, 0, IntPtr.Zero, 0, ref bytesReturned, IntPtr.Zero); if (!result) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to unmount volume.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } return(result); }
internal static void DiskSetDriveLayoutEx(IntPtr handle, DRIVE_LAYOUT_INFORMATION_EX layout) { var buffSize = Marshal.SizeOf(layout); var buffer = Marshal.AllocHGlobal(buffSize); int bytesReturned = 0; Marshal.StructureToPtr(layout, buffer, false); var result = NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_DISK_SET_DRIVE_LAYOUT_EX, buffer, buffSize, IntPtr.Zero, 0, ref bytesReturned, IntPtr.Zero); Marshal.FreeHGlobal(buffer); if (!result) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to set drive layout.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } }
internal static ulong GetFilesizeInSectors(IntPtr handle, ulong sectorSize, int offset = 0) { ulong retVal = 0; long fileSize = 0; if (!NativeDisk.GetFileSizeEx(handle, out fileSize)) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get file size in sectors.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } fileSize -= offset; retVal = (ulong)((fileSize / (long)sectorSize) + ((fileSize % (long)sectorSize) > 0 ? 1 : 0)); return(retVal); }
internal static void WriteSectorDataToHandle(IntPtr handle, byte[] data, ulong startSector, ulong numSectors, ulong sectorSize, int length) { uint bytesWritten; bool result; LARGE_INTEGER li; li.LowPart = 0; li.QuadPart = (long)(startSector * sectorSize); NativeDisk.SetFilePointer(handle, li.LowPart, out li.HighPart, EMoveMethod.Begin); result = NativeDisk.WriteFile(handle, data, (uint)(length), out bytesWritten, IntPtr.Zero); if (!result) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to write sector data to handle.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } }
public static ulong GetDeviceLength(int deviceID) { ulong length = 0; IntPtr deviceHandle = NativeDiskWrapper.GetHandleOnDevice(deviceID, NativeDisk.GENERIC_READ); unsafe { int returnLength = 0; IntPtr lengthPtr = new IntPtr(&length); NativeDisk.DeviceIoControl(deviceHandle, NativeDisk.IOCTL_DISK_GET_LENGTH_INFO, IntPtr.Zero, 0, lengthPtr, sizeof(ulong), ref returnLength, IntPtr.Zero); } NativeDisk.CloseHandle(deviceHandle); return(length); }
internal static DRIVE_LAYOUT_INFORMATION_EX DiskGetDriveLayoutEx(IntPtr handle) { var buffSize = Marshal.SizeOf(typeof(DRIVE_LAYOUT_INFORMATION_EX)); var buffer = Marshal.AllocHGlobal(buffSize); int outBytes = 0; var result = NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_DISK_GET_DRIVE_LAYOUT_EX, IntPtr.Zero, 0, buffer, buffSize, ref outBytes, IntPtr.Zero); if (!result) { Marshal.FreeHGlobal(buffer); var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get drive layout.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } else { var resultStruct = Marshal.PtrToStructure <DRIVE_LAYOUT_INFORMATION_EX>(buffer); Marshal.FreeHGlobal(buffer); return(resultStruct); } }
internal static PARTITION_INFORMATION_EX GetDiskPartitionInfo(IntPtr handle) { bool result; int outBytes = 0; PARTITION_INFORMATION_EX partitionInfo = new PARTITION_INFORMATION_EX(); IntPtr partitionInfoStrPtr; partitionInfoStrPtr = NativeDisk.StructToPtr(partitionInfo); result = NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_DISK_GET_PARTITION_INFO_EX, IntPtr.Zero, 0, partitionInfoStrPtr, Marshal.SizeOf(partitionInfo), ref outBytes, IntPtr.Zero); if (!result) { Marshal.FreeHGlobal(partitionInfoStrPtr); var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get disk partition info.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } var resultStruct = partitionInfo = Marshal.PtrToStructure <PARTITION_INFORMATION_EX>(partitionInfoStrPtr); Marshal.FreeHGlobal(partitionInfoStrPtr); return(resultStruct); }
internal static ulong GetNumberOfSectors(IntPtr handle, ref ulong sectorSize) { int foo = 0; DISK_GEOMETRY_EX diskGeometry = new DISK_GEOMETRY_EX(); bool result; IntPtr diskGeometryPtr = NativeDisk.StructToPtr(diskGeometry); result = NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_DISK_GET_DRIVE_GEOMETRY_EX, IntPtr.Zero, 0, diskGeometryPtr, Marshal.SizeOf(diskGeometry), ref foo, IntPtr.Zero); if (!result) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to get number of sectors.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } diskGeometry = Marshal.PtrToStructure <DISK_GEOMETRY_EX>(diskGeometryPtr); Marshal.FreeHGlobal(diskGeometryPtr); sectorSize = (ulong)diskGeometry.Geometry.BytesPerSector; return((ulong)(diskGeometry.DiskSize / diskGeometry.Geometry.BytesPerSector)); }
internal static void DiskCreateDiskMBR(IntPtr handle, uint Signature) { CREATE_DISK disk = new CREATE_DISK(); disk.PartitionStyle = PARTITION_STYLE.MasterBootRecord; // MBR disk.Mbr.Signature = Signature; var buffSize = Marshal.SizeOf(disk); var buffer = Marshal.AllocHGlobal(buffSize); Marshal.StructureToPtr(disk, buffer, false); int bytesReturned = 0; var result = NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_DISK_CREATE_DISK, buffer, buffSize, IntPtr.Zero, 0, ref bytesReturned, IntPtr.Zero); Marshal.FreeHGlobal(buffer); if (!result) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to create disk MBR.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } }
public static DiskPartitionInfo GetDiskPartitionInfo(char driveLetter) { int deviceID = NativeDiskWrapper.CheckDriveType(string.Format(@"\\.\{0}:\", driveLetter)); IntPtr deviceHandle = NativeDiskWrapper.GetHandleOnDevice(deviceID, NativeDisk.GENERIC_READ); uint partitionTableSize = 0; DiskPartitionInfo partitionsInfo = new DiskPartitionInfo(); var partitionInfo = NativeDiskWrapper.GetDiskPartitionInfo(deviceHandle); partitionsInfo.DiskTotalSize = GetDeviceLength(deviceID); if (partitionInfo.PartitionStyle == PARTITION_STYLE.MasterBootRecord) { partitionsInfo.PartitionType = PartitionType.MBR; partitionTableSize = 512; unsafe { byte *data = NativeDiskWrapper.ReadSectorDataPointerFromHandle(deviceHandle, 0, 1, 512); for (int i = 0; i < 4; i++) { ulong partitionStartSector = (uint)Marshal.ReadInt32(new IntPtr(data), 0x1BE + 8 + 16 * i); ulong partitionNumSectors = (uint)Marshal.ReadInt32(new IntPtr(data), 0x1BE + 12 + 16 * i); if (partitionStartSector + partitionNumSectors > 0) { partitionsInfo.PartitionSizes.Add(partitionNumSectors * 512); } } } } else if (partitionInfo.PartitionStyle == PARTITION_STYLE.GuidPartitionTable) { partitionsInfo.PartitionType = PartitionType.GPT; partitionTableSize = 17408; unsafe { byte *data = NativeDiskWrapper.ReadSectorDataPointerFromHandle(deviceHandle, 0, 1, 512); uint gptHeaderOffset = (uint)Marshal.ReadInt32(new IntPtr(data), 0x1C6); data = NativeDiskWrapper.ReadSectorDataPointerFromHandle(deviceHandle, gptHeaderOffset, 1, 512); ulong partitionTableSector = (ulong)Marshal.ReadInt64(new IntPtr(data), 0x48); uint noOfPartitionEntries = (uint)Marshal.ReadInt32(new IntPtr(data), 0x50); uint sizeOfPartitionEntry = (uint)Marshal.ReadInt32(new IntPtr(data), 0x54); data = NativeDiskWrapper.ReadSectorDataPointerFromHandle(deviceHandle, partitionTableSector, (512 / sizeOfPartitionEntry) * noOfPartitionEntries, 512); for (int i = 0; i < noOfPartitionEntries; i++) { ulong partitionStartSector = (ulong)Marshal.ReadInt64(new IntPtr(data), (int)(0x20 + sizeOfPartitionEntry * i)); ulong partitionNumSectors = (ulong)Marshal.ReadInt64(new IntPtr(data), (int)(0x28 + sizeOfPartitionEntry * i)); if (partitionStartSector + partitionNumSectors > 0) { partitionsInfo.PartitionSizes.Add(partitionNumSectors * 512); } } } } else { partitionsInfo.PartitionType = PartitionType.RAW; } NativeDisk.CloseHandle(deviceHandle); partitionsInfo.UnallocatedSize = partitionsInfo.DiskTotalSize - (partitionsInfo.PartitionSizes.Sum() + partitionTableSize); return(partitionsInfo); }
internal static int CheckDriveType(string name) { int retVal = 0; int bytesReaded = 0; IntPtr handle; string nameNoSlash = name.Substring(0, name.Length - 1); var driveType = NativeDisk.GetDriveType(name); switch (driveType) { case DriveType.Fixed: case DriveType.Removable: handle = NativeDisk.CreateFile(nameNoSlash, NativeDisk.FILE_READ_ATTRIBUTES, NativeDisk.FILE_SHARE_READ | NativeDisk.FILE_SHARE_WRITE, IntPtr.Zero, NativeDisk.OPEN_EXISTING, 0, IntPtr.Zero); if (handle == NativeDisk.INVALID_HANDLE_VALUE) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to check drive type 1.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } else { STORAGE_DEVICE_DESCRIPTOR devDescriptor = new STORAGE_DEVICE_DESCRIPTOR(); STORAGE_DEVICE_NUMBER devNumber = GetDiskProperty(handle, ref devDescriptor); if ((driveType == DriveType.Removable && devDescriptor.BusType != STORAGE_BUS_TYPE.BusTypeSata) || (driveType == DriveType.Fixed && devDescriptor.BusType == STORAGE_BUS_TYPE.BusTypeUsb) || (devDescriptor.BusType == STORAGE_BUS_TYPE.BusTypeSd) || (devDescriptor.BusType == STORAGE_BUS_TYPE.BusTypeMmc)) { if (NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_STORAGE_CHECK_VERIFY2, IntPtr.Zero, 0, IntPtr.Zero, 0, ref bytesReaded, IntPtr.Zero)) { retVal = devNumber.DeviceNumber; } else { NativeDisk.CloseHandle(handle); Utils.SetErrorMode(Utils.ErrorModes.SEM_FAILCRITICALERRORS); try { handle = NativeDisk.CreateFile(name, NativeDisk.FILE_READ_DATA, NativeDisk.FILE_SHARE_READ | NativeDisk.FILE_SHARE_WRITE, IntPtr.Zero, NativeDisk.OPEN_EXISTING, 0, IntPtr.Zero); } finally { Utils.SetErrorMode(Utils.ErrorModes.SYSTEM_DEFAULT); } if (handle == NativeDisk.INVALID_HANDLE_VALUE) { var exception = new Win32Exception(Marshal.GetLastWin32Error()); throw new Exception(string.Format("Error occured when trying to check drive type 2.\nError code: {0}\nMessage: {1}", exception.NativeErrorCode, exception.Message)); } if (NativeDisk.DeviceIoControl(handle, NativeDisk.IOCTL_STORAGE_CHECK_VERIFY, IntPtr.Zero, 0, IntPtr.Zero, 0, ref bytesReaded, IntPtr.Zero)) { retVal = devNumber.DeviceNumber; } } } NativeDisk.CloseHandle(handle); } break; default: throw new Win32Exception("Invalid device type"); } return(retVal); }
internal static bool ByteArrayCompare(byte[] b1, byte[] b2) { // Validate buffers are the same length. // This also ensures that the count does not exceed the length of either buffer. return(NativeDisk.memcmp(b1, b2, b1.Length > b2.Length ? b2.Length : b1.Length) == 0); }
internal static uint GetLogicalDrives() { return(NativeDisk.GetLogicalDrives()); }