summaryrefslogtreecommitdiffstats
path: root/private/utils/fmifs/src/format.cxx
diff options
context:
space:
mode:
Diffstat (limited to 'private/utils/fmifs/src/format.cxx')
-rw-r--r--private/utils/fmifs/src/format.cxx497
1 files changed, 497 insertions, 0 deletions
diff --git a/private/utils/fmifs/src/format.cxx b/private/utils/fmifs/src/format.cxx
new file mode 100644
index 000000000..ee28d457b
--- /dev/null
+++ b/private/utils/fmifs/src/format.cxx
@@ -0,0 +1,497 @@
+#define _NTAPI_ULIB_
+
+#include "ulib.hxx"
+extern "C" {
+#include "fmifs.h"
+};
+#include "fmifsmsg.hxx"
+#include "ifssys.hxx"
+#include "wstring.hxx"
+#include "ifsentry.hxx"
+#include "system.hxx"
+#include "drive.hxx"
+
+
+MEDIA_TYPE
+ComputeNtMediaType(
+ IN FMIFS_MEDIA_TYPE MediaType
+ )
+/*++
+
+Routine Description:
+
+ This routine translates the FMIFS media type to the NT media type.
+
+Arguments:
+
+ MediaType - Supplies the FMIFS media type.
+
+Return Value:
+
+ The NT media type corresponding to the input.
+
+--*/
+{
+ MEDIA_TYPE media_type;
+
+ switch (MediaType) {
+ case FmMediaFixed:
+ media_type = FixedMedia;
+ break;
+ case FmMediaRemovable:
+ media_type = RemovableMedia;
+ break;
+ case FmMediaF5_1Pt2_512:
+ media_type = F5_1Pt2_512;
+ break;
+ case FmMediaF5_360_512:
+ media_type = F5_360_512;
+ break;
+ case FmMediaF5_320_512:
+ media_type = F5_320_512;
+ break;
+ case FmMediaF5_320_1024:
+ media_type = F5_320_1024;
+ break;
+ case FmMediaF5_180_512:
+ media_type = F5_180_512;
+ break;
+ case FmMediaF5_160_512:
+ media_type = F5_160_512;
+ break;
+ case FmMediaF3_2Pt88_512:
+ media_type = F3_2Pt88_512;
+ break;
+ case FmMediaF3_1Pt44_512:
+ media_type = F3_1Pt44_512;
+ break;
+ case FmMediaF3_720_512:
+ media_type = F3_720_512;
+ break;
+ case FmMediaF3_20Pt8_512:
+ media_type = F3_20Pt8_512;
+ break;
+ case FmMediaF3_120M_512:
+ media_type = F3_120M_512;
+ break;
+ case FmMediaUnknown:
+ default:
+ media_type = Unknown;
+ break;
+ }
+
+ return media_type;
+}
+
+
+FMIFS_MEDIA_TYPE
+ComputeFmMediaType(
+ IN MEDIA_TYPE MediaType
+ )
+/*++
+
+Routine Description:
+
+ This routine translates the NT media type to the FMIFS media type.
+
+Arguments:
+
+ MediaType - Supplies the NT media type.
+
+Return Value:
+
+ The FMIFS media type corresponding to the input.
+
+--*/
+{
+ FMIFS_MEDIA_TYPE media_type;
+
+ switch (MediaType) {
+ case FixedMedia:
+ media_type = FmMediaFixed;
+ break;
+ case RemovableMedia:
+ media_type = FmMediaRemovable;
+ break;
+ case F5_1Pt2_512:
+ media_type = FmMediaF5_1Pt2_512;
+ break;
+ case F5_360_512:
+ media_type = FmMediaF5_360_512;
+ break;
+ case F5_320_512:
+ media_type = FmMediaF5_320_512;
+ break;
+ case F5_320_1024:
+ media_type = FmMediaF5_320_1024;
+ break;
+ case F5_180_512:
+ media_type = FmMediaF5_180_512;
+ break;
+ case F5_160_512:
+ media_type = FmMediaF5_160_512;
+ break;
+ case F3_2Pt88_512:
+ media_type = FmMediaF3_2Pt88_512;
+ break;
+ case F3_1Pt44_512:
+ media_type = FmMediaF3_1Pt44_512;
+ break;
+ case F3_720_512:
+ media_type = FmMediaF3_720_512;
+ break;
+ case F3_20Pt8_512:
+ media_type = FmMediaF3_20Pt8_512;
+ break;
+ case F3_120M_512:
+ media_type = FmMediaF3_120M_512;
+ break;
+ case Unknown:
+ default:
+ media_type = FmMediaUnknown;
+ break;
+ }
+
+ return media_type;
+}
+
+
+VOID
+Format(
+ IN PWSTR DriveName,
+ IN FMIFS_MEDIA_TYPE MediaType,
+ IN PWSTR FileSystemName,
+ IN PWSTR Label,
+ IN BOOLEAN Quick,
+ IN FMIFS_CALLBACK Callback
+ )
+/*++
+
+Routine Description:
+
+ This routine loads and calls the correct DLL to format the
+ given volume.
+
+Arguments:
+
+ DriveName - Supplies the DOS style drive name.
+ MediaType - Supplies the media type.
+ FileSystemName - Supplies the file system type to format to.
+ Label - Supplies a new label for the volume.
+ Quick - Supplies whether or not to perform a quick
+ format.
+ Callback - Supplies the necessary call back for
+ communication with the file manager.
+
+Return Value:
+
+ FALSE - Failure.
+ TRUE - Success.
+
+--*/
+{
+ FMIFS_MESSAGE message;
+ DSTRING format_string;
+ DSTRING library_name;
+ DSTRING file_system_name;
+ FORMAT_FN format_function;
+ HANDLE dll_handle;
+ DSTRING ntdrivename;
+ BOOLEAN result;
+ DSTRING label_string;
+ DSTRING dosdrivename;
+ FMIFS_FINISHED_INFORMATION finished_info;
+ DWORD OldErrorMode;
+
+ // Initialize the message object with the callback function.
+ // Load the file system DLL.
+ // Compute the NT style drive name.
+
+ if (!message.Initialize(Callback) ||
+ !format_string.Initialize("Format") ||
+ !library_name.Initialize("U") ||
+ !file_system_name.Initialize(FileSystemName) ||
+ !library_name.Strcat(&file_system_name) ||
+ !(format_function = (FORMAT_FN)
+ SYSTEM::QueryLibraryEntryPoint(&library_name,
+ &format_string,
+ &dll_handle)) ||
+ !dosdrivename.Initialize(DriveName) ||
+ !label_string.Initialize(Label) ||
+ !IFS_SYSTEM::DosDriveNameToNtDriveName(&dosdrivename, &ntdrivename)) {
+
+ finished_info.Success = FALSE;
+ Callback(FmIfsFinished,
+ sizeof(FMIFS_FINISHED_INFORMATION),
+ &finished_info);
+ return;
+ }
+
+ // Disable hard-error popups.
+ OldErrorMode = SetErrorMode( SEM_FAILCRITICALERRORS );
+
+ // Call format.
+
+ result = format_function(&ntdrivename,
+ &message,
+ Quick,
+ ComputeNtMediaType(MediaType),
+ &label_string, 0);
+
+ // Enable hard-error popups.
+ SetErrorMode( OldErrorMode );
+
+ SYSTEM::FreeLibraryHandle(dll_handle);
+
+ finished_info.Success = result;
+ Callback(FmIfsFinished,
+ sizeof(FMIFS_FINISHED_INFORMATION),
+ &finished_info);
+}
+
+
+VOID
+FormatEx(
+ IN PWSTR DriveName,
+ IN FMIFS_MEDIA_TYPE MediaType,
+ IN PWSTR FileSystemName,
+ IN PWSTR Label,
+ IN BOOLEAN Quick,
+ IN DWORD ClusterSize,
+ IN FMIFS_CALLBACK Callback
+ )
+/*++
+
+Routine Description:
+
+ This routine loads and calls the correct DLL to format the
+ given volume.
+
+Arguments:
+
+ DriveName - Supplies the DOS style drive name.
+ MediaType - Supplies the media type.
+ FileSystemName - Supplies the file system type to format to.
+ Label - Supplies a new label for the volume.
+ Quick - Supplies whether or not to perform a quick
+ format.
+ ClusterSize - Size of volume cluster in bytes.
+ Callback - Supplies the necessary call back for
+ communication with the file manager.
+
+Return Value:
+
+ FALSE - Failure.
+ TRUE - Success.
+
+--*/
+{
+ FMIFS_MESSAGE message;
+ DSTRING format_string;
+ DSTRING library_name;
+ DSTRING file_system_name;
+ FORMAT_FN format_function;
+ HANDLE dll_handle;
+ DSTRING ntdrivename;
+ BOOLEAN result;
+ DSTRING label_string;
+ DSTRING dosdrivename;
+ FMIFS_FINISHED_INFORMATION finished_info;
+ DWORD OldErrorMode;
+
+ // Initialize the message object with the callback function.
+ // Load the file system DLL.
+ // Compute the NT style drive name.
+
+ if (!message.Initialize(Callback) ||
+ !format_string.Initialize("Format") ||
+ !library_name.Initialize("U") ||
+ !file_system_name.Initialize(FileSystemName) ||
+ !library_name.Strcat(&file_system_name) ||
+ !(format_function = (FORMAT_FN)
+ SYSTEM::QueryLibraryEntryPoint(&library_name,
+ &format_string,
+ &dll_handle)) ||
+ !dosdrivename.Initialize(DriveName) ||
+ !label_string.Initialize(Label) ||
+ !IFS_SYSTEM::DosDriveNameToNtDriveName(&dosdrivename, &ntdrivename)) {
+
+ finished_info.Success = FALSE;
+ Callback(FmIfsFinished,
+ sizeof(FMIFS_FINISHED_INFORMATION),
+ &finished_info);
+ return;
+ }
+
+ // Disable hard-error popups.
+ OldErrorMode = SetErrorMode( SEM_FAILCRITICALERRORS );
+
+ // Call format.
+
+ result = format_function(&ntdrivename,
+ &message,
+ Quick,
+ ComputeNtMediaType(MediaType),
+ &label_string, ClusterSize);
+
+ // Enable hard-error popups.
+ SetErrorMode( OldErrorMode );
+
+ SYSTEM::FreeLibraryHandle(dll_handle);
+
+ finished_info.Success = result;
+ Callback(FmIfsFinished,
+ sizeof(FMIFS_FINISHED_INFORMATION),
+ &finished_info);
+}
+
+
+BOOLEAN
+EnableVolumeCompression(
+ IN PWSTR DriveName,
+ IN USHORT CompressionFormat
+ )
+/*++
+
+Routine Description:
+
+ This sets the compression attribute on the root directory of an NTFS volume.
+ Note that the compression state of any files already contained on the
+ volume is not affected.
+
+Arguments:
+
+ DriveName - Supplies the drive name.
+ Expects a string like "C:\".
+
+ CompressionFormat - COMPRESSION_FORMAT_NONE = Uncompressed.
+ COMPRESSION_FORMAT_DEFAULT = Default compression.
+ COMPRESSION_FORMAT_LZNT1 = Use LZNT1 compression format.
+ (as defined in NTRTL.H)
+
+Return Value:
+
+ FALSE - Failure.
+ TRUE - Success.
+
+--*/
+{
+ HANDLE hFile;
+ BOOLEAN bStatus = FALSE;
+
+ //
+ // Don't even try if no drive name provided.
+ //
+ if (DriveName[0])
+ {
+ //
+ // Try to open the root directory - READ_DATA | WRITE_DATA.
+ //
+ if ((hFile = CreateFile(DriveName,
+ FILE_READ_DATA | FILE_WRITE_DATA,
+ FILE_SHARE_READ | FILE_SHARE_WRITE,
+ NULL,
+ OPEN_EXISTING,
+ FILE_FLAG_BACKUP_SEMANTICS,
+ NULL )) != INVALID_HANDLE_VALUE)
+ {
+ ULONG Length = 0;
+
+ if (DeviceIoControl( hFile,
+ FSCTL_SET_COMPRESSION,
+ &CompressionFormat,
+ sizeof(CompressionFormat),
+ NULL,
+ 0,
+ &Length,
+ NULL))
+ {
+ //
+ // Successfully set compression on root directory.
+ //
+ bStatus = TRUE;
+ }
+ CloseHandle(hFile);
+ }
+ }
+ return bStatus;
+}
+
+BOOLEAN
+QuerySupportedMedia(
+ IN PWSTR DriveName,
+ OUT PFMIFS_MEDIA_TYPE MediaTypeArray,
+ IN ULONG NumberOfArrayEntries,
+ OUT PULONG NumberOfMediaTypes
+ )
+/*++
+
+Routine Description:
+
+ This routine computes a list of the supported media types for
+ the given drive.
+
+ If NULL is passed for the array then 'NumberOfMediaTypes'
+ is filled in with the correct number.
+
+Arguments:
+
+ DriveName - Supplies the drive name.
+ MediaTypeArray - Returns the supported media types.
+ NumberOfArrayEntries - Supplies the number of entries in
+ 'MediaTypeArray'.
+ NumberOfMediaTypes - Returns the number of media types
+ returned in the media type array.
+
+Return Value:
+
+ FALSE - Failure.
+ TRUE - Success.
+
+--*/
+{
+ DSTRING dosdrivename, ntdrivename;
+ DP_DRIVE dpdrive;
+ PCDRTYPE nt_media_types;
+ INT num_types;
+ ULONG i, j;
+ FMIFS_MEDIA_TYPE tmp;
+
+ if (!dosdrivename.Initialize(DriveName) ||
+ !IFS_SYSTEM::DosDriveNameToNtDriveName(&dosdrivename, &ntdrivename)) {
+
+ return FALSE;
+ }
+
+ if (!dpdrive.Initialize(&ntdrivename)) {
+ SetLastError(RtlNtStatusToDosError(dpdrive.QueryLastNtStatus()));
+ return FALSE;
+ }
+
+ if (!(nt_media_types = dpdrive.GetSupportedList(&num_types))) {
+ return FALSE;
+ }
+
+ if (!MediaTypeArray) {
+ *NumberOfMediaTypes = num_types;
+ return TRUE;
+ }
+
+ *NumberOfMediaTypes = min(NumberOfArrayEntries, (ULONG)num_types);
+
+ for (i = 0; i < *NumberOfMediaTypes; i++) {
+ MediaTypeArray[i] = ComputeFmMediaType(nt_media_types[i].MediaType);
+ }
+
+ for (i = 0; i < *NumberOfMediaTypes; i++) {
+ for (j = 0; j < *NumberOfMediaTypes - 1; j++) {
+ if (MediaTypeArray[j] < MediaTypeArray[j + 1]) {
+ tmp = MediaTypeArray[j];
+ MediaTypeArray[j] = MediaTypeArray[j + 1];
+ MediaTypeArray[j + 1] = tmp;
+ }
+ }
+ }
+
+ return TRUE;
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-10-18 19:13:40 +0200