ReactOS 0.4.15-dev-5669-g09dde2c
fsutil.h File Reference
#include <fmifs/fmifs.h>
Include dependency graph for fsutil.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros

#define FAT_BOOTSECTOR_SIZE   (1 * SECTORSIZE)
 
#define FAT32_BOOTSECTOR_SIZE   (1 * SECTORSIZE)
 
#define BTRFS_BOOTSECTOR_SIZE   (3 * SECTORSIZE)
 
#define NTFS_BOOTSECTOR_SIZE   (16 * SECTORSIZE)
 
#define InstallFat12BootCode   InstallFatBootCode
 
#define InstallFat16BootCode   InstallFatBootCode
 

Typedefs

typedef IN HANDLE DstPath
 
typedef IN HANDLE IN HANDLE RootPartition
 

Functions

BOOLEAN GetRegisteredFileSystems (IN ULONG Index, OUT PCWSTR *FileSystemName)
 
NTSTATUS ChkdskFileSystem_UStr (IN PUNICODE_STRING DriveRoot, IN PCWSTR FileSystemName, IN BOOLEAN FixErrors, IN BOOLEAN Verbose, IN BOOLEAN CheckOnlyIfDirty, IN BOOLEAN ScanDrive, IN PFMIFSCALLBACK Callback)
 
NTSTATUS ChkdskFileSystem (IN PCWSTR DriveRoot, IN PCWSTR FileSystemName, IN BOOLEAN FixErrors, IN BOOLEAN Verbose, IN BOOLEAN CheckOnlyIfDirty, IN BOOLEAN ScanDrive, IN PFMIFSCALLBACK Callback)
 
NTSTATUS FormatFileSystem_UStr (IN PUNICODE_STRING DriveRoot, IN PCWSTR FileSystemName, IN FMIFS_MEDIA_FLAG MediaFlag, IN PUNICODE_STRING Label, IN BOOLEAN QuickFormat, IN ULONG ClusterSize, IN PFMIFSCALLBACK Callback)
 
NTSTATUS FormatFileSystem (IN PCWSTR DriveRoot, IN PCWSTR FileSystemName, IN FMIFS_MEDIA_FLAG MediaFlag, IN PCWSTR Label, IN BOOLEAN QuickFormat, IN ULONG ClusterSize, IN PFMIFSCALLBACK Callback)
 
typedef NTSTATUS (*PFS_INSTALL_BOOTCODE)(IN PCWSTR SrcPath
 
NTSTATUS InstallFatBootCode (IN PCWSTR SrcPath, IN HANDLE DstPath, IN HANDLE RootPartition)
 
NTSTATUS InstallFat32BootCode (IN PCWSTR SrcPath, IN HANDLE DstPath, IN HANDLE RootPartition)
 
NTSTATUS InstallBtrfsBootCode (IN PCWSTR SrcPath, IN HANDLE DstPath, IN HANDLE RootPartition)
 
NTSTATUS InstallNtfsBootCode (IN PCWSTR SrcPath, IN HANDLE DstPath, IN HANDLE RootPartition)
 
NTSTATUS ChkdskPartition (IN PPARTENTRY PartEntry, IN BOOLEAN FixErrors, IN BOOLEAN Verbose, IN BOOLEAN CheckOnlyIfDirty, IN BOOLEAN ScanDrive, IN PFMIFSCALLBACK Callback)
 
NTSTATUS FormatPartition (IN PPARTENTRY PartEntry, IN PCWSTR FileSystemName, IN FMIFS_MEDIA_FLAG MediaFlag, IN PCWSTR Label, IN BOOLEAN QuickFormat, IN ULONG ClusterSize, IN PFMIFSCALLBACK Callback)
 

Macro Definition Documentation

◆ BTRFS_BOOTSECTOR_SIZE

#define BTRFS_BOOTSECTOR_SIZE   (3 * SECTORSIZE)

Definition at line 70 of file fsutil.h.

◆ FAT32_BOOTSECTOR_SIZE

#define FAT32_BOOTSECTOR_SIZE   (1 * SECTORSIZE)

Definition at line 69 of file fsutil.h.

◆ FAT_BOOTSECTOR_SIZE

#define FAT_BOOTSECTOR_SIZE   (1 * SECTORSIZE)

Definition at line 68 of file fsutil.h.

◆ InstallFat12BootCode

#define InstallFat12BootCode   InstallFatBootCode

Definition at line 85 of file fsutil.h.

◆ InstallFat16BootCode

#define InstallFat16BootCode   InstallFatBootCode

Definition at line 86 of file fsutil.h.

◆ NTFS_BOOTSECTOR_SIZE

#define NTFS_BOOTSECTOR_SIZE   (16 * SECTORSIZE)

Definition at line 71 of file fsutil.h.

Typedef Documentation

◆ DstPath

typedef IN HANDLE DstPath

Definition at line 76 of file fsutil.h.

◆ RootPartition

Definition at line 77 of file fsutil.h.

Function Documentation

◆ ChkdskFileSystem()

NTSTATUS ChkdskFileSystem ( IN PCWSTR  DriveRoot,
IN PCWSTR  FileSystemName,
IN BOOLEAN  FixErrors,
IN BOOLEAN  Verbose,
IN BOOLEAN  CheckOnlyIfDirty,
IN BOOLEAN  ScanDrive,
IN PFMIFSCALLBACK  Callback 
)

Definition at line 287 of file fsutil.c.

295{
296 UNICODE_STRING DriveRootU;
297
298 RtlInitUnicodeString(&DriveRootU, DriveRoot);
299 return ChkdskFileSystem_UStr(&DriveRootU,
300 FileSystemName,
301 FixErrors,
302 Verbose,
303 CheckOnlyIfDirty,
304 ScanDrive,
305 Callback);
306}
BOOL FixErrors
Definition: chkdsk.c:69
BOOL Verbose
Definition: chkdsk.c:72
NTSYSAPI VOID NTAPI RtlInitUnicodeString(PUNICODE_STRING DestinationString, PCWSTR SourceString)
NTSTATUS ChkdskFileSystem_UStr(IN PUNICODE_STRING DriveRoot, IN PCWSTR FileSystemName, IN BOOLEAN FixErrors, IN BOOLEAN Verbose, IN BOOLEAN CheckOnlyIfDirty, IN BOOLEAN ScanDrive, IN PFMIFSCALLBACK Callback)
Definition: fsutil.c:244
_In_ WDFINTERRUPT _In_ PFN_WDF_INTERRUPT_SYNCHRONIZE Callback
Definition: wdfinterrupt.h:458

Referenced by ChkdskPartition().

◆ ChkdskFileSystem_UStr()

NTSTATUS ChkdskFileSystem_UStr ( IN PUNICODE_STRING  DriveRoot,
IN PCWSTR  FileSystemName,
IN BOOLEAN  FixErrors,
IN BOOLEAN  Verbose,
IN BOOLEAN  CheckOnlyIfDirty,
IN BOOLEAN  ScanDrive,
IN PFMIFSCALLBACK  Callback 
)

ChkdskEx()

Definition at line 244 of file fsutil.c.

252{
256
257 FileSystem = GetFileSystemByName(FileSystemName);
258
259 if (!FileSystem || !FileSystem->ChkdskFunc)
260 {
261 // Success = FALSE;
262 // Callback(DONE, 0, &Success);
264 }
265
267 Success = FileSystem->ChkdskFunc(DriveRoot,
268 Callback,
269 FixErrors,
270 Verbose,
271 CheckOnlyIfDirty,
272 ScanDrive,
273 NULL,
274 NULL,
275 NULL,
276 NULL,
277 (PULONG)&Status);
278 if (!Success)
279 DPRINT1("ChkdskFunc() failed with Status 0x%lx\n", Status);
280
281 // Callback(DONE, 0, &Success);
282
283 return Status;
284}
unsigned char BOOLEAN
LONG NTSTATUS
Definition: precomp.h:26
#define DPRINT1
Definition: precomp.h:8
PWCHAR FileSystem
Definition: format.c:72
#define NULL
Definition: types.h:112
@ Success
Definition: eventcreate.c:712
Status
Definition: gdiplustypes.h:25
#define STATUS_NOT_SUPPORTED
Definition: ntstatus.h:423
static PFILE_SYSTEM GetFileSystemByName(IN PCWSTR FileSystemName)
Definition: fsutil.c:197
#define STATUS_SUCCESS
Definition: shellext.h:65
uint32_t * PULONG
Definition: typedefs.h:59

Referenced by ChkdskFileSystem().

◆ ChkdskPartition()

NTSTATUS ChkdskPartition ( IN PPARTENTRY  PartEntry,
IN BOOLEAN  FixErrors,
IN BOOLEAN  Verbose,
IN BOOLEAN  CheckOnlyIfDirty,
IN BOOLEAN  ScanDrive,
IN PFMIFSCALLBACK  Callback 
)

Definition at line 757 of file fsutil.c.

764{
766 PDISKENTRY DiskEntry = PartEntry->DiskEntry;
767 // UNICODE_STRING PartitionRootPath;
768 WCHAR PartitionRootPath[MAX_PATH]; // PathBuffer
769
770 ASSERT(PartEntry->IsPartitioned && PartEntry->PartitionNumber != 0);
771
772 /* HACK: Do not try to check a partition with an unknown filesystem */
773 if (!*PartEntry->FileSystem)
774 {
775 PartEntry->NeedsCheck = FALSE;
776 return STATUS_SUCCESS;
777 }
778
779 /* Set PartitionRootPath */
780 RtlStringCchPrintfW(PartitionRootPath, ARRAYSIZE(PartitionRootPath),
781 L"\\Device\\Harddisk%lu\\Partition%lu",
782 DiskEntry->DiskNumber,
783 PartEntry->PartitionNumber);
784 DPRINT("PartitionRootPath: %S\n", PartitionRootPath);
785
786 /* Check the partition */
787 Status = ChkdskFileSystem(PartitionRootPath,
788 PartEntry->FileSystem,
789 FixErrors,
790 Verbose,
791 CheckOnlyIfDirty,
792 ScanDrive,
793 Callback);
794 if (!NT_SUCCESS(Status))
795 return Status;
796
797 PartEntry->NeedsCheck = FALSE;
798 return STATUS_SUCCESS;
799}
#define FALSE
Definition: types.h:117
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
#define ARRAYSIZE(array)
Definition: filtermapper.c:47
#define MAX_PATH
Definition: compat.h:34
#define ASSERT(a)
Definition: mode.c:44
NTSTRSAFEVAPI RtlStringCchPrintfW(_Out_writes_(cchDest) _Always_(_Post_z_) NTSTRSAFE_PWSTR pszDest, _In_ size_t cchDest, _In_ _Printf_format_string_ NTSTRSAFE_PCWSTR pszFormat,...)
Definition: ntstrsafe.h:1110
#define L(x)
Definition: ntvdm.h:50
NTSTATUS ChkdskFileSystem(IN PCWSTR DriveRoot, IN PCWSTR FileSystemName, IN BOOLEAN FixErrors, IN BOOLEAN Verbose, IN BOOLEAN CheckOnlyIfDirty, IN BOOLEAN ScanDrive, IN PFMIFSCALLBACK Callback)
Definition: fsutil.c:287
#define DPRINT
Definition: sndvol32.h:71
ULONG DiskNumber
Definition: partlist.h:111
__wchar_t WCHAR
Definition: xmlstorage.h:180

Referenced by DoChkdsk().

◆ FormatFileSystem()

NTSTATUS FormatFileSystem ( IN PCWSTR  DriveRoot,
IN PCWSTR  FileSystemName,
IN FMIFS_MEDIA_FLAG  MediaFlag,
IN PCWSTR  Label,
IN BOOLEAN  QuickFormat,
IN ULONG  ClusterSize,
IN PFMIFSCALLBACK  Callback 
)

Definition at line 376 of file fsutil.c.

384{
385 UNICODE_STRING DriveRootU;
386 UNICODE_STRING LabelU;
387
388 RtlInitUnicodeString(&DriveRootU, DriveRoot);
389 RtlInitUnicodeString(&LabelU, Label);
390
391 return FormatFileSystem_UStr(&DriveRootU,
392 FileSystemName,
393 MediaFlag,
394 &LabelU,
397 Callback);
398}
BOOL QuickFormat
Definition: format.c:66
PWCHAR Label
Definition: format.c:70
DWORD ClusterSize
Definition: format.c:67
NTSTATUS FormatFileSystem_UStr(IN PUNICODE_STRING DriveRoot, IN PCWSTR FileSystemName, IN FMIFS_MEDIA_FLAG MediaFlag, IN PUNICODE_STRING Label, IN BOOLEAN QuickFormat, IN ULONG ClusterSize, IN PFMIFSCALLBACK Callback)
Definition: fsutil.c:311

Referenced by FormatPartition(), and InstallFatBootcodeToFloppy().

◆ FormatFileSystem_UStr()

NTSTATUS FormatFileSystem_UStr ( IN PUNICODE_STRING  DriveRoot,
IN PCWSTR  FileSystemName,
IN FMIFS_MEDIA_FLAG  MediaFlag,
IN PUNICODE_STRING  Label,
IN BOOLEAN  QuickFormat,
IN ULONG  ClusterSize,
IN PFMIFSCALLBACK  Callback 
)

FormatEx()

Definition at line 311 of file fsutil.c.

319{
322 BOOLEAN BackwardCompatible = FALSE; // Default to latest FS versions.
323 MEDIA_TYPE MediaType;
324
325 FileSystem = GetFileSystemByName(FileSystemName);
326
327 if (!FileSystem || !FileSystem->FormatFunc)
328 {
329 // Success = FALSE;
330 // Callback(DONE, 0, &Success);
332 }
333
334 /* Set the BackwardCompatible flag in case we format with older FAT12/16 */
335 if (wcsicmp(FileSystemName, L"FAT") == 0)
336 BackwardCompatible = TRUE;
337 // else if (wcsicmp(FileSystemName, L"FAT32") == 0)
338 // BackwardCompatible = FALSE;
339
340 /* Convert the FMIFS MediaFlag to a NT MediaType */
341 // FIXME: Actually covert all the possible flags.
342 switch (MediaFlag)
343 {
344 case FMIFS_FLOPPY:
345 MediaType = F5_320_1024; // FIXME: This is hardfixed!
346 break;
347 case FMIFS_REMOVABLE:
348 MediaType = RemovableMedia;
349 break;
350 case FMIFS_HARDDISK:
351 MediaType = FixedMedia;
352 break;
353 default:
354 DPRINT1("Unknown FMIFS MediaFlag %d, converting 1-to-1 to NT MediaType\n",
355 MediaFlag);
356 MediaType = (MEDIA_TYPE)MediaFlag;
357 break;
358 }
359
360 Success = FileSystem->FormatFunc(DriveRoot,
361 Callback,
363 BackwardCompatible,
364 MediaType,
365 Label,
367 if (!Success)
368 DPRINT1("FormatFunc() failed\n");
369
370 // Callback(DONE, 0, &Success);
371
373}
#define TRUE
Definition: types.h:120
#define wcsicmp
Definition: compat.h:15
@ FMIFS_FLOPPY
Definition: fmifs.h:47
@ FMIFS_REMOVABLE
Definition: fmifs.h:50
@ FMIFS_HARDDISK
Definition: fmifs.h:51
enum _MEDIA_TYPE MEDIA_TYPE
@ RemovableMedia
Definition: ntdddisk.h:387
@ FixedMedia
Definition: ntdddisk.h:388
@ F5_320_1024
Definition: ntdddisk.h:384
#define STATUS_UNSUCCESSFUL
Definition: udferr_usr.h:132

Referenced by FormatFileSystem().

◆ FormatPartition()

NTSTATUS FormatPartition ( IN PPARTENTRY  PartEntry,
IN PCWSTR  FileSystemName,
IN FMIFS_MEDIA_FLAG  MediaFlag,
IN PCWSTR  Label,
IN BOOLEAN  QuickFormat,
IN ULONG  ClusterSize,
IN PFMIFSCALLBACK  Callback 
)

Definition at line 802 of file fsutil.c.

810{
812 PDISKENTRY DiskEntry = PartEntry->DiskEntry;
814 // UNICODE_STRING PartitionRootPath;
815 WCHAR PartitionRootPath[MAX_PATH]; // PathBuffer
816
817 ASSERT(PartEntry->IsPartitioned && PartEntry->PartitionNumber != 0);
818
819 if (!FileSystemName || !*FileSystemName)
820 {
821 DPRINT1("No file system specified?\n");
823 }
824
825 /*
826 * Prepare the partition for formatting (for MBR disks, reset the
827 * partition type), and adjust the filesystem name in case of FAT
828 * vs. FAT32, depending on the geometry of the partition.
829 */
830
831// FIXME: Do this only if QuickFormat == FALSE? What about FAT handling?
832
833 /*
834 * Retrieve a partition type as a hint only. It will be used to determine
835 * whether to actually use FAT12/16 or FAT32 filesystem, depending on the
836 * geometry of the partition. If the partition resides on an MBR disk,
837 * the partition style will be reset to this value as well, unless the
838 * partition is OEM.
839 */
841 PartEntry->StartSector.QuadPart,
842 PartEntry->SectorCount.QuadPart);
844 {
845 /* Unknown file system */
846 DPRINT1("Unknown file system '%S'\n", FileSystemName);
848 }
849
850 /* Reset the MBR partition type, unless this is an OEM partition */
851 if (DiskEntry->DiskStyle == PARTITION_STYLE_MBR)
852 {
853 if (!IsOEMPartition(PartEntry->PartitionType))
855 }
856
857 /*
858 * Adjust the filesystem name in case of FAT vs. FAT32, according to
859 * the type of partition returned by FileSystemToMBRPartitionType().
860 */
861 if (wcsicmp(FileSystemName, L"FAT") == 0)
862 {
865 {
866 FileSystemName = L"FAT32";
867 }
868 }
869
870 /* Commit the partition changes to the disk */
871 Status = WritePartitions(DiskEntry);
872 if (!NT_SUCCESS(Status))
873 {
874 DPRINT1("WritePartitions(disk %lu) failed, Status 0x%08lx\n",
875 DiskEntry->DiskNumber, Status);
877 }
878
879 /* Set PartitionRootPath */
880 RtlStringCchPrintfW(PartitionRootPath, ARRAYSIZE(PartitionRootPath),
881 L"\\Device\\Harddisk%lu\\Partition%lu",
882 DiskEntry->DiskNumber,
883 PartEntry->PartitionNumber);
884 DPRINT("PartitionRootPath: %S\n", PartitionRootPath);
885
886 /* Format the partition */
887 Status = FormatFileSystem(PartitionRootPath,
888 FileSystemName,
889 MediaFlag,
890 Label,
893 Callback);
894 if (!NT_SUCCESS(Status))
895 return Status;
896
897//
898// TODO: Here, call a partlist.c function that update the actual
899// FS name and the label fields of the volume.
900//
901 PartEntry->FormatState = Formatted;
902
903 /* Set the new partition's file system proper */
904 RtlStringCbCopyW(PartEntry->FileSystem,
905 sizeof(PartEntry->FileSystem),
906 FileSystemName);
907
908 PartEntry->New = FALSE;
909
910 return STATUS_SUCCESS;
911}
#define PARTITION_ENTRY_UNUSED
Definition: disk.h:86
#define PARTITION_FAT32
Definition: disk.h:95
#define PARTITION_FAT32_XINT13
Definition: disk.h:96
UCHAR FileSystemToMBRPartitionType(IN PCWSTR FileSystem, IN ULONGLONG StartSector, IN ULONGLONG SectorCount)
Definition: fsrec.c:333
#define IsOEMPartition(PartitionType)
Definition: partlist.h:22
@ Formatted
Definition: partlist.h:38
@ PARTITION_STYLE_MBR
Definition: imports.h:201
#define STATUS_PARTITION_FAILURE
Definition: ntstatus.h:604
NTSTRSAFEAPI RtlStringCbCopyW(_Out_writes_bytes_(cbDest) _Always_(_Post_z_) NTSTRSAFE_PWSTR pszDest, _In_ size_t cbDest, _In_ NTSTRSAFE_PCWSTR pszSrc)
Definition: ntstrsafe.h:174
CHAR PartitionType
Definition: part_xbox.c:32
NTSTATUS FormatFileSystem(IN PCWSTR DriveRoot, IN PCWSTR FileSystemName, IN FMIFS_MEDIA_FLAG MediaFlag, IN PCWSTR Label, IN BOOLEAN QuickFormat, IN ULONG ClusterSize, IN PFMIFSCALLBACK Callback)
Definition: fsutil.c:376
VOID SetMBRPartitionType(IN PPARTENTRY PartEntry, IN UCHAR PartitionType)
Definition: partlist.c:3937
NTSTATUS WritePartitions(IN PDISKENTRY DiskEntry)
Definition: partlist.c:3619
PARTITION_STYLE DiskStyle
Definition: partlist.h:121
#define STATUS_UNRECOGNIZED_VOLUME
Definition: udferr_usr.h:173
unsigned char UCHAR
Definition: xmlstorage.h:181

Referenced by DoFormat().

◆ GetRegisteredFileSystems()

BOOLEAN GetRegisteredFileSystems ( IN ULONG  Index,
OUT PCWSTR FileSystemName 
)

QueryAvailableFileSystemFormat()

Definition at line 182 of file fsutil.c.

185{
187 return FALSE;
188
189 *FileSystemName = RegisteredFileSystems[Index].FileSystemName;
190
191 return TRUE;
192}
static FILE_SYSTEM RegisteredFileSystems[]
Definition: fsutil.c:159
PCWSTR FileSystemName
Definition: fsutil.c:153
_In_ WDFCOLLECTION _In_ ULONG Index

Referenced by InitializeFileSystemList().

◆ InstallBtrfsBootCode()

NTSTATUS InstallBtrfsBootCode ( IN PCWSTR  SrcPath,
IN HANDLE  DstPath,
IN HANDLE  RootPartition 
)

Definition at line 585 of file fsutil.c.

589{
591 NTSTATUS LockStatus;
596 BOOTCODE NewBootSector = {0};
597
598 /* Allocate and read the new bootsector from SrcPath */
599 RtlInitUnicodeString(&Name, SrcPath);
600 Status = ReadBootCodeFromFile(&NewBootSector,
601 &Name,
603 if (!NT_SUCCESS(Status))
604 return Status;
605
606 /*
607 * The BTRFS driver requires the volume to be locked in order to modify
608 * the first sectors of the partition, even though they are outside the
609 * file-system space / in the reserved area (they are situated before
610 * the super-block at 0x1000) and is in principle allowed by the NT
611 * storage stack.
612 * So we lock here in order to write the bootsector at sector 0.
613 * If locking fails, we ignore and continue nonetheless.
614 */
615 LockStatus = NtFsControlFile(DstPath,
616 NULL,
617 NULL,
618 NULL,
621 NULL,
622 0,
623 NULL,
624 0);
625 if (!NT_SUCCESS(LockStatus))
626 {
627 DPRINT1("WARNING: Failed to lock BTRFS volume for writing bootsector! Operations may fail! (Status 0x%lx)\n", LockStatus);
628 }
629
630 /* Obtain partition info and write it to the bootsector */
632 NULL,
633 NULL,
634 NULL,
637 NULL,
638 0,
639 &PartInfo,
640 sizeof(PartInfo));
641 if (!NT_SUCCESS(Status))
642 {
643 DPRINT1("IOCTL_DISK_GET_PARTITION_INFO_EX failed (Status %lx)\n", Status);
644 goto Quit;
645 }
646
647 /* Write new bootsector to RootPath */
648 ((PBTRFS_BOOTSECTOR)NewBootSector.BootCode)->PartitionStartLBA =
650
651 /* Write sector 0 */
652 FileOffset.QuadPart = 0ULL;
654 NULL,
655 NULL,
656 NULL,
658 NewBootSector.BootCode,
659 NewBootSector.Length,
660 &FileOffset,
661 NULL);
662 if (!NT_SUCCESS(Status))
663 {
664 DPRINT1("NtWriteFile() failed (Status %lx)\n", Status);
665 goto Quit;
666 }
667
668Quit:
669 /* Unlock the volume */
670 LockStatus = NtFsControlFile(DstPath,
671 NULL,
672 NULL,
673 NULL,
676 NULL,
677 0,
678 NULL,
679 0);
680 if (!NT_SUCCESS(LockStatus))
681 {
682 DPRINT1("Failed to unlock BTRFS volume (Status 0x%lx)\n", LockStatus);
683 }
684
685 /* Free the new bootsector */
686 FreeBootCode(&NewBootSector);
687
688 return Status;
689}
struct NameRec_ * Name
Definition: cdprocs.h:460
VOID FreeBootCode(IN OUT PBOOTCODE BootCodeInfo)
Definition: bootcode.c:104
NTSTATUS ReadBootCodeFromFile(IN OUT PBOOTCODE BootCodeInfo, IN PUNICODE_STRING FilePath, IN ULONG Length OPTIONAL)
Definition: bootcode.c:69
#define SECTORSIZE
Definition: bootcode.h:13
_In_ PFCB _In_ LONGLONG FileOffset
Definition: cdprocs.h:160
#define IOCTL_DISK_GET_PARTITION_INFO_EX
Definition: ntddk_ex.h:206
#define ULL(a, b)
Definition: format_msg.c:27
static OUT PIO_STATUS_BLOCK IoStatusBlock
Definition: pipe.c:75
#define FSCTL_LOCK_VOLUME
Definition: nt_native.h:832
NTSYSAPI NTSTATUS NTAPI NtWriteFile(IN HANDLE hFile, IN HANDLE hEvent OPTIONAL, IN PIO_APC_ROUTINE IoApcRoutine OPTIONAL, IN PVOID IoApcContext OPTIONAL, OUT PIO_STATUS_BLOCK pIoStatusBlock, IN PVOID WriteBuffer, IN ULONG WriteBufferLength, IN PLARGE_INTEGER FileOffset OPTIONAL, IN PULONG LockOperationKey OPTIONAL)
NTSYSAPI NTSTATUS NTAPI NtDeviceIoControlFile(IN HANDLE hFile, IN HANDLE hEvent OPTIONAL, IN PIO_APC_ROUTINE IoApcRoutine OPTIONAL, IN PVOID IoApcContext OPTIONAL, OUT PIO_STATUS_BLOCK pIoStatusBlock, IN ULONG DeviceIoControlCode, IN PVOID InBuffer OPTIONAL, IN ULONG InBufferLength, OUT PVOID OutBuffer OPTIONAL, IN ULONG OutBufferLength)
#define FSCTL_UNLOCK_VOLUME
Definition: nt_native.h:833
NTSYSAPI NTSTATUS NTAPI NtFsControlFile(IN HANDLE hFile, IN HANDLE hEvent OPTIONAL, IN PIO_APC_ROUTINE IoApcRoutine OPTIONAL, IN PVOID IoApcContext OPTIONAL, OUT PIO_STATUS_BLOCK pIoStatusBlock, IN ULONG DeviceIoControlCode, IN PVOID InBuffer OPTIONAL, IN ULONG InBufferLength, OUT PVOID OutBuffer OPTIONAL, IN ULONG OutBufferLength)
struct _BTRFS_BOOTSECTOR * PBTRFS_BOOTSECTOR
IN HANDLE IN HANDLE RootPartition
Definition: fsutil.h:77
#define BTRFS_BOOTSECTOR_SIZE
Definition: fsutil.h:70
IN HANDLE DstPath
Definition: fsutil.h:76
ULONG Length
Definition: bootcode.h:18
PVOID BootCode
Definition: bootcode.h:17
LARGE_INTEGER StartingOffset
Definition: imports.h:221
LONGLONG QuadPart
Definition: typedefs.h:114

Referenced by InstallBtrfsBootcodeToPartition().

◆ InstallFat32BootCode()

NTSTATUS InstallFat32BootCode ( IN PCWSTR  SrcPath,
IN HANDLE  DstPath,
IN HANDLE  RootPartition 
)

Definition at line 464 of file fsutil.c.

468{
474 BOOTCODE OrigBootSector = {0};
475 BOOTCODE NewBootSector = {0};
476
477 /* Allocate and read the current original partition bootsector */
478 Status = ReadBootCodeByHandle(&OrigBootSector,
481 if (!NT_SUCCESS(Status))
482 return Status;
483
484 /* Allocate and read the new bootsector (2 sectors) from SrcPath */
485 RtlInitUnicodeString(&Name, SrcPath);
486 Status = ReadBootCodeFromFile(&NewBootSector,
487 &Name,
489 if (!NT_SUCCESS(Status))
490 {
491 FreeBootCode(&OrigBootSector);
492 return Status;
493 }
494
495 /* Adjust the bootsector (copy a part of the FAT32 BPB) */
496 RtlCopyMemory(&((PFAT32_BOOTSECTOR)NewBootSector.BootCode)->OemName,
497 &((PFAT32_BOOTSECTOR)OrigBootSector.BootCode)->OemName,
498 FIELD_OFFSET(FAT32_BOOTSECTOR, BootCodeAndData) -
500
501 /*
502 * We know we copy the boot code to a file only when DstPath != RootPartition,
503 * otherwise the boot code is copied to the specified root partition.
504 */
505 if (DstPath != RootPartition)
506 {
507 /* Copy to a file: Disable the backup bootsector */
508 ((PFAT32_BOOTSECTOR)NewBootSector.BootCode)->BackupBootSector = 0;
509 }
510 else
511 {
512 /* Copy to a disk: Get the location of the backup bootsector */
513 BackupBootSector = ((PFAT32_BOOTSECTOR)OrigBootSector.BootCode)->BackupBootSector;
514 }
515
516 /* Free the original bootsector */
517 FreeBootCode(&OrigBootSector);
518
519 /* Write the first sector of the new bootcode to DstPath sector 0 */
520 FileOffset.QuadPart = 0ULL;
522 NULL,
523 NULL,
524 NULL,
526 NewBootSector.BootCode,
528 &FileOffset,
529 NULL);
530 if (!NT_SUCCESS(Status))
531 {
532 DPRINT1("NtWriteFile() failed (Status %lx)\n", Status);
533 FreeBootCode(&NewBootSector);
534 return Status;
535 }
536
537 if (DstPath == RootPartition)
538 {
539 /* Copy to a disk: Write the backup bootsector */
540 if ((BackupBootSector != 0x0000) && (BackupBootSector != 0xFFFF))
541 {
544 NULL,
545 NULL,
546 NULL,
548 NewBootSector.BootCode,
550 &FileOffset,
551 NULL);
552 if (!NT_SUCCESS(Status))
553 {
554 DPRINT1("NtWriteFile() failed (Status %lx)\n", Status);
555 FreeBootCode(&NewBootSector);
556 return Status;
557 }
558 }
559 }
560
561 /* Write the second sector of the new bootcode to boot disk sector 14 */
562 // FileOffset.QuadPart = (ULONGLONG)(14 * FAT32_BOOTSECTOR_SIZE);
563 FileOffset.QuadPart = 14 * FAT32_BOOTSECTOR_SIZE;
564 Status = NtWriteFile(DstPath, // or really RootPartition ???
565 NULL,
566 NULL,
567 NULL,
569 ((PUCHAR)NewBootSector.BootCode + FAT32_BOOTSECTOR_SIZE),
571 &FileOffset,
572 NULL);
573 if (!NT_SUCCESS(Status))
574 {
575 DPRINT1("NtWriteFile() failed (Status %lx)\n", Status);
576 }
577
578 /* Free the new bootsector */
579 FreeBootCode(&NewBootSector);
580
581 return Status;
582}
BOOL BackupBootSector(LPCTSTR lpszVolumeName)
Definition: install.c:61
NTSTATUS ReadBootCodeByHandle(IN OUT PBOOTCODE BootCodeInfo, IN HANDLE FileHandle, IN ULONG Length OPTIONAL)
Definition: bootcode.c:21
IN PDCB IN POEM_STRING OemName
Definition: fatprocs.h:1304
unsigned short USHORT
Definition: pedump.c:61
struct _FAT32_BOOTSECTOR * PFAT32_BOOTSECTOR
#define FAT32_BOOTSECTOR_SIZE
Definition: fsutil.h:69
#define FIELD_OFFSET(t, f)
Definition: typedefs.h:255
#define RtlCopyMemory(Destination, Source, Length)
Definition: typedefs.h:263
unsigned char * PUCHAR
Definition: typedefs.h:53
uint32_t ULONG
Definition: typedefs.h:59
uint64_t ULONGLONG
Definition: typedefs.h:67

Referenced by InstallFatBootcodeToPartition().

◆ InstallFatBootCode()

NTSTATUS InstallFatBootCode ( IN PCWSTR  SrcPath,
IN HANDLE  DstPath,
IN HANDLE  RootPartition 
)

Definition at line 406 of file fsutil.c.

410{
415 BOOTCODE OrigBootSector = {0};
416 BOOTCODE NewBootSector = {0};
417
418 /* Allocate and read the current original partition bootsector */
419 Status = ReadBootCodeByHandle(&OrigBootSector,
422 if (!NT_SUCCESS(Status))
423 return Status;
424
425 /* Allocate and read the new bootsector from SrcPath */
426 RtlInitUnicodeString(&Name, SrcPath);
427 Status = ReadBootCodeFromFile(&NewBootSector,
428 &Name,
430 if (!NT_SUCCESS(Status))
431 {
432 FreeBootCode(&OrigBootSector);
433 return Status;
434 }
435
436 /* Adjust the bootsector (copy a part of the FAT12/16 BPB) */
437 RtlCopyMemory(&((PFAT_BOOTSECTOR)NewBootSector.BootCode)->OemName,
438 &((PFAT_BOOTSECTOR)OrigBootSector.BootCode)->OemName,
439 FIELD_OFFSET(FAT_BOOTSECTOR, BootCodeAndData) -
441
442 /* Free the original bootsector */
443 FreeBootCode(&OrigBootSector);
444
445 /* Write the new bootsector to DstPath */
446 FileOffset.QuadPart = 0ULL;
448 NULL,
449 NULL,
450 NULL,
452 NewBootSector.BootCode,
453 NewBootSector.Length,
454 &FileOffset,
455 NULL);
456
457 /* Free the new bootsector */
458 FreeBootCode(&NewBootSector);
459
460 return Status;
461}
#define FAT_BOOTSECTOR_SIZE
Definition: fsutil.h:68

◆ InstallNtfsBootCode()

NTSTATUS InstallNtfsBootCode ( IN PCWSTR  SrcPath,
IN HANDLE  DstPath,
IN HANDLE  RootPartition 
)

Definition at line 692 of file fsutil.c.

696{
701 BOOTCODE OrigBootSector = {0};
702 BOOTCODE NewBootSector = {0};
703
704 /* Allocate and read the current original partition bootsector */
706 if (!NT_SUCCESS(Status))
707 {
708 DPRINT1("InstallNtfsBootCode: Status %lx\n", Status);
709 return Status;
710 }
711
712 /* Allocate and read the new bootsector (16 sectors) from SrcPath */
713 RtlInitUnicodeString(&Name, SrcPath);
715 if (!NT_SUCCESS(Status))
716 {
717 DPRINT1("InstallNtfsBootCode: Status %lx\n", Status);
718 FreeBootCode(&OrigBootSector);
719 return Status;
720 }
721
722 /* Adjust the bootsector (copy a part of the NTFS BPB) */
723 RtlCopyMemory(&((PNTFS_BOOTSECTOR)NewBootSector.BootCode)->OEMID,
724 &((PNTFS_BOOTSECTOR)OrigBootSector.BootCode)->OEMID,
726
727 /* Write sector 0 */
728 FileOffset.QuadPart = 0ULL;
730 NULL,
731 NULL,
732 NULL,
734 NewBootSector.BootCode,
735 NewBootSector.Length,
736 &FileOffset,
737 NULL);
738 if (!NT_SUCCESS(Status))
739 {
740 DPRINT1("NtWriteFile() failed (Status %lx)\n", Status);
741 goto Quit;
742 }
743
744Quit:
745 /* Free the new bootsector */
746 FreeBootCode(&NewBootSector);
747
748 return Status;
749}
#define NTFS_BOOTSECTOR_SIZE
Definition: fsutil.h:71

Referenced by InstallNtfsBootcodeToPartition().

◆ NTSTATUS()

typedef NTSTATUS ( PFS_INSTALL_BOOTCODE)