windows-nt/Source/XPSP1/NT/base/ntsetup/textmode/kernel/spfatfmt.c

858 lines
23 KiB
C
Raw Normal View History

2020-09-26 03:20:57 -05:00
#include "spprecmp.h"
#pragma hdrstop
//
// This variable is needed since it contains a buffer that can
// be used in kernel mode. The buffer is used by NtFsControlFile,
// since the Zw API is not exported
//
extern PSETUP_COMMUNICATION CommunicationParams;
#define VERIFY_SIZE 65536
typedef struct {
UCHAR IntelNearJumpCommand[1]; // Intel Jump command
UCHAR BootStrapJumpOffset[2]; // offset of boot strap code
UCHAR OemData[8]; // OEM data
UCHAR BytesPerSector[2]; // BPB
UCHAR SectorsPerCluster[1]; //
UCHAR ReservedSectors[2]; //
UCHAR Fats[1]; //
UCHAR RootEntries[2]; //
UCHAR Sectors[2]; //
UCHAR Media[1]; //
UCHAR SectorsPerFat[2]; //
UCHAR SectorsPerTrack[2]; //
UCHAR Heads[2]; //
UCHAR HiddenSectors[4]; //
UCHAR LargeSectors[4]; //
UCHAR PhysicalDrive[1]; // 0 = removable, 80h = fixed
UCHAR CurrentHead[1]; // not used by fs utils
UCHAR Signature[1]; // boot signature
UCHAR SerialNumber[4]; // serial number
UCHAR Label[11]; // volume label, aligned padded
UCHAR SystemIdText[8]; // system ID, FAT for example
} UNALIGNED_SECTOR_ZERO, *PUNALIGNED_SECTOR_ZERO;
#define CSEC_FAT32MEG 65536
#define CSEC_FAT16BIT 32680
#define MIN_CLUS_BIG 4085 // Minimum clusters for a big FAT.
#define MAX_CLUS_BIG 65525 // Maximum + 1 clusters for big FAT.
USHORT
ComputeSecPerCluster(
IN ULONG NumSectors,
IN BOOLEAN SmallFat
)
/*++
Routine Description:
This routine computes the number of sectors per cluster.
Arguments:
NumSectors - Supplies the number of sectors on the disk.
SmallFat - Supplies whether or not the FAT should be small.
Return Value:
The number of sectors per cluster necessary.
--*/
{
ULONG threshold;
USHORT sec_per_clus;
USHORT min_sec_per_clus;
threshold = SmallFat ? MIN_CLUS_BIG : MAX_CLUS_BIG;
sec_per_clus = 1;
while (NumSectors >= threshold) {
sec_per_clus *= 2;
threshold *= 2;
}
if (SmallFat) {
min_sec_per_clus = 8;
} else {
min_sec_per_clus = 4;
}
return max(sec_per_clus, min_sec_per_clus);
}
ULONG
SpComputeSerialNumber(
VOID
)
/*++
Routine Description:
This routine computes a new serial number for a volume.
Arguments:
Seed - Supplies a seed for the serial number.
Return Value:
A new volume serial number.
--*/
{
PUCHAR p;
ULONG i;
TIME_FIELDS time_fields;
static ULONG Seed = 0;
ULONG SerialNumber;
BOOLEAN b;
//
// If this is the first time we've entered this routine,
// generate a seed value based on the real time clock.
//
if(!Seed) {
b = HalQueryRealTimeClock(&time_fields);
ASSERT(b);
Seed = ((time_fields.Year - 1970) *366*24*60*60) +
(time_fields.Month *31*24*60*60) +
(time_fields.Day *24*60*60) +
(time_fields.Hour *60*60) +
(time_fields.Minute *60) +
time_fields.Second +
((ULONG)time_fields.Milliseconds << 10);
ASSERT(Seed);
if(!Seed) {
Seed = 1;
}
}
SerialNumber = Seed;
p = (PUCHAR)&SerialNumber;
for(i=0; i<sizeof(ULONG); i++) {
SerialNumber += p[i];
SerialNumber = (SerialNumber >> 2) + (SerialNumber << 30);
}
if(++Seed == 0) { // unlikely, but possible.
Seed++;
}
return SerialNumber;
}
VOID
EditFat(
IN USHORT ClusterNumber,
IN USHORT ClusterEntry,
IN OUT PUCHAR Fat,
IN BOOLEAN SmallFat
)
/*++
Routine Description:
This routine edits the FAT entry 'ClusterNumber' with 'ClusterEntry'.
Arguments:
ClusterNumber - Supplies the number of the cluster to edit.
ClusterEntry - Supplies the new value for that cluster number.
Fat - Supplies the FAT to edit.
SmallFat - Supplies whether or not the FAT is small.
Return Value:
None.
--*/
{
ULONG n;
if (SmallFat) {
n = ClusterNumber*3;
if (n%2) {
Fat[n/2] = (UCHAR) ((Fat[n/2]&0x0F) | ((ClusterEntry&0x000F)<<4));
Fat[n/2 + 1] = (UCHAR) ((ClusterEntry&0x0FF0)>>4);
} else {
Fat[n/2] = (UCHAR) (ClusterEntry&0x00FF);
Fat[n/2 + 1] = (UCHAR) ((Fat[n/2 + 1]&0xF0) |
((ClusterEntry&0x0F00)>>8));
}
} else {
((PUSHORT) Fat)[ClusterNumber] = ClusterEntry;
}
}
NTSTATUS
FmtFillFormatBuffer(
IN ULONGLONG NumberOfSectors,
IN ULONG SectorSize,
IN ULONG SectorsPerTrack,
IN ULONG NumberOfHeads,
IN ULONGLONG NumberOfHiddenSectors,
OUT PVOID FormatBuffer,
IN ULONG FormatBufferSize,
OUT PULONGLONG SuperAreaSize,
IN PULONG BadSectorsList,
IN ULONG NumberOfBadSectors,
OUT PUCHAR SystemId
)
/*++
Routine Description:
This routine computes a FAT super area based on the disk size,
disk geometry, and bad sectors of the volume.
Arguments:
NumberOfSectors - Supplies the number of sectors on the volume.
SectorSize - Supplies the number of bytes per sector.
SectorsPerTrack - Supplies the number of sectors per track.
NumberOfHeads - Supplies the number of heads.
NumberOfHiddenSectors - Supplies the number of hidden sectors.
FormatBuffer - Returns the super area for the volume.
FormatBufferSize - Supplies the number of bytes in the supplied
buffer.
SuperAreaSize - Returns the number of bytes in the super area.
BadSectorsList - Supplies the list of bad sectors on the volume.
NumberOfBadSectors - Supplies the number of bad sectors in the list.
Return Value:
ENOMEM - The buffer wasn't big enough.
E2BIG - The disk is too large to be formatted.
EIO - There is a bad sector in the super area.
EINVAL - There is a bad sector off the end of the disk.
ESUCCESS
--*/
{
PUNALIGNED_SECTOR_ZERO psecz;
PUCHAR puchar;
USHORT tmp_ushort;
ULONG tmp_ulong;
BOOLEAN small_fat;
ULONG num_sectors;
UCHAR partition_id;
ULONG sec_per_fat;
ULONG sec_per_root;
ULONG sec_per_clus;
ULONG i;
ULONG sec_per_sa;
RtlZeroMemory(FormatBuffer,FormatBufferSize);
// Make sure that there's enough room for the BPB.
if(!FormatBuffer || FormatBufferSize < SectorSize) {
return(STATUS_BUFFER_TOO_SMALL);
}
// Compute the number of sectors on disk.
num_sectors = (ULONG)NumberOfSectors;
// Compute the partition identifier.
partition_id = num_sectors < CSEC_FAT16BIT ? PARTITION_FAT_12 :
num_sectors < CSEC_FAT32MEG ? PARTITION_FAT_16 :
PARTITION_HUGE;
// Compute whether or not to have a big or small FAT.
small_fat = (BOOLEAN) (partition_id == PARTITION_FAT_12);
psecz = (PUNALIGNED_SECTOR_ZERO) FormatBuffer;
puchar = (PUCHAR) FormatBuffer;
//
// Copy the fat boot code into the format buffer.
//
if (!IsNEC_98) { //NEC98
ASSERT(sizeof(FatBootCode) == 512);
RtlMoveMemory(psecz,FatBootCode,sizeof(FatBootCode));
// Set up the jump instruction.
psecz->IntelNearJumpCommand[0] = 0xeb;
psecz->IntelNearJumpCommand[1] = 0x3c;
psecz->IntelNearJumpCommand[2] = 0x90;
} else {
ASSERT(sizeof(PC98FatBootCode) == 512);
RtlMoveMemory(psecz,PC98FatBootCode,sizeof(PC98FatBootCode));
//
// Already written jump instruction to bootcode.
// So,do not reset jump code.
//
} //NEC98
// Set up the OEM data.
memcpy(psecz->OemData, "MSDOS5.0", 8);
// Set up the bytes per sector.
U_USHORT(psecz->BytesPerSector) = (USHORT)SectorSize;
// Set up the number of sectors per cluster.
sec_per_clus = ComputeSecPerCluster(num_sectors, small_fat);
if (sec_per_clus > 128) {
// The disk is too large to be formatted.
return(STATUS_INVALID_PARAMETER);
}
psecz->SectorsPerCluster[0] = (UCHAR) sec_per_clus;
// Set up the number of reserved sectors.
U_USHORT(psecz->ReservedSectors) = (USHORT)max(1,512/SectorSize);
// Set up the number of FATs.
psecz->Fats[0] = 2;
// Set up the number of root entries and number of sectors for the root.
U_USHORT(psecz->RootEntries) = 512;
sec_per_root = (512*32 - 1)/SectorSize + 1;
// Set up the number of sectors.
if (num_sectors >= 1<<16) {
tmp_ushort = 0;
tmp_ulong = num_sectors;
} else {
tmp_ushort = (USHORT) num_sectors;
tmp_ulong = 0;
}
U_USHORT(psecz->Sectors) = tmp_ushort;
U_ULONG(psecz->LargeSectors) = tmp_ulong;
// Set up the media byte.
psecz->Media[0] = 0xF8;
// Set up the number of sectors per FAT.
if (small_fat) {
sec_per_fat = num_sectors/(2 + SectorSize*sec_per_clus*2/3);
} else {
sec_per_fat = num_sectors/(2 + SectorSize*sec_per_clus/2);
}
sec_per_fat++;
U_USHORT(psecz->SectorsPerFat) = (USHORT)sec_per_fat;
// Set up the number of sectors per track.
U_USHORT(psecz->SectorsPerTrack) = (USHORT)SectorsPerTrack;
// Set up the number of heads.
U_USHORT(psecz->Heads) = (USHORT)NumberOfHeads;
// Set up the number of hidden sectors.
U_ULONG(psecz->HiddenSectors) = (ULONG)NumberOfHiddenSectors;
// Set up the physical drive number.
psecz->PhysicalDrive[0] = 0x80;
psecz->CurrentHead[0] = 0;
// Set up the BPB signature.
psecz->Signature[0] = 0x29;
// Set up the serial number.
U_ULONG(psecz->SerialNumber) = SpComputeSerialNumber();
// Set up the volume label.
memcpy(psecz->Label, "NO NAME ",11);
// Set up the system id.
memcpy(psecz->SystemIdText, small_fat ? "FAT12 " : "FAT16 ", 8);
// Set up the boot signature.
puchar[510] = 0x55;
puchar[511] = 0xAA;
// Now make sure that the buffer has enough room for both of the
// FATs and the root directory.
sec_per_sa = 1 + 2*sec_per_fat + sec_per_root;
*SuperAreaSize = SectorSize*sec_per_sa;
if (*SuperAreaSize > FormatBufferSize) {
return(STATUS_BUFFER_TOO_SMALL);
}
// Set up the first FAT.
puchar[SectorSize] = 0xF8;
puchar[SectorSize + 1] = 0xFF;
puchar[SectorSize + 2] = 0xFF;
if (!small_fat) {
puchar[SectorSize + 3] = 0xFF;
}
for (i = 0; i < NumberOfBadSectors; i++) {
if (BadSectorsList[i] < sec_per_sa) {
// There's a bad sector in the super area.
return(STATUS_UNSUCCESSFUL);
}
if (BadSectorsList[i] >= num_sectors) {
// Bad sector out of range.
return(STATUS_NONEXISTENT_SECTOR);
}
// Compute the bad cluster number;
tmp_ushort = (USHORT)
((BadSectorsList[i] - sec_per_sa)/sec_per_clus + 2);
EditFat(tmp_ushort, (USHORT) 0xFFF7, &puchar[SectorSize], small_fat);
}
// Copy the first FAT onto the second.
memcpy(&puchar[SectorSize*(1 + sec_per_fat)],
&puchar[SectorSize],
(unsigned int) SectorSize*sec_per_fat);
*SystemId = partition_id;
return(STATUS_SUCCESS);
}
VOID
FmtVerifySectors(
IN HANDLE Handle,
IN ULONG NumberOfSectors,
IN ULONG SectorSize,
OUT PULONG* BadSectorsList,
OUT PULONG NumberOfBadSectors
)
/*++
Routine Description:
This routine verifies all of the sectors on the volume.
It returns a pointer to a list of bad sectors. The pointer
will be NULL if there was an error detected.
Arguments:
Handle - Supplies a handle to the partition for verifying.
NumberOfSectors - Supplies the number of partition sectors.
SectorSize - Supplies the number of bytes per sector.
BadSectorsList - Returns the list of bad sectors.
NumberOfBadSectors - Returns the number of bad sectors in the list.
Return Value:
None.
--*/
{
ULONG num_read_sec;
ULONG i, j;
PULONG bad_sec_buf;
ULONG max_num_bad;
PVOID Gauge;
IO_STATUS_BLOCK IoStatusBlock;
NTSTATUS Status;
VERIFY_INFORMATION VerifyInfo;
max_num_bad = 100;
bad_sec_buf = SpMemAlloc(max_num_bad*sizeof(ULONG));
ASSERT(bad_sec_buf);
*NumberOfBadSectors = 0;
num_read_sec = VERIFY_SIZE/SectorSize;
//
// Initialize the Gas Gauge
//
SpFormatMessage(
TemporaryBuffer,
sizeof(TemporaryBuffer),
SP_TEXT_SETUP_IS_FORMATTING
);
Gauge = SpCreateAndDisplayGauge(
NumberOfSectors/num_read_sec,
0,
VideoVars.ScreenHeight - STATUS_HEIGHT - (3*GAUGE_HEIGHT/2),
TemporaryBuffer,
NULL,
GF_PERCENTAGE,
0
);
VerifyInfo.StartingOffset.QuadPart = 0;
for (i = 0; i < NumberOfSectors; i += num_read_sec) {
if (i + num_read_sec > NumberOfSectors) {
num_read_sec = NumberOfSectors - i;
}
//
// Verify this many sectors at the current offset.
//
VerifyInfo.Length = num_read_sec * SectorSize;
Status = ZwDeviceIoControlFile(
Handle,
NULL,
NULL,
NULL,
&IoStatusBlock,
IOCTL_DISK_VERIFY,
&VerifyInfo,
sizeof(VerifyInfo),
NULL,
0
);
//
// I/O should be synchronous.
//
ASSERT(Status != STATUS_PENDING);
if(!NT_SUCCESS(Status)) {
//
// Range is bad -- verify individual sectors.
//
VerifyInfo.Length = SectorSize;
for (j = 0; j < num_read_sec; j++) {
Status = ZwDeviceIoControlFile(
Handle,
NULL,
NULL,
NULL,
&IoStatusBlock,
IOCTL_DISK_VERIFY,
&VerifyInfo,
sizeof(VerifyInfo),
NULL,
0
);
ASSERT(Status != STATUS_PENDING);
if(!NT_SUCCESS(Status)) {
if (*NumberOfBadSectors == max_num_bad) {
max_num_bad += 100;
bad_sec_buf = SpMemRealloc(
bad_sec_buf,
max_num_bad*sizeof(ULONG)
);
ASSERT(bad_sec_buf);
}
bad_sec_buf[(*NumberOfBadSectors)++] = i + j;
}
//
// Advance to next sector.
//
VerifyInfo.StartingOffset.QuadPart += SectorSize;
}
} else {
//
// Advance to next range of sectors.
//
VerifyInfo.StartingOffset.QuadPart += VerifyInfo.Length;
}
if(Gauge) {
SpTickGauge(Gauge);
}
}
if(Gauge) {
SpTickGauge(Gauge);
}
*BadSectorsList = bad_sec_buf;
//return(STATUS_SUCCESS);
}
#if 0
//
// Code not used, we call autoformat
//
NTSTATUS
SpFatFormat(
IN PDISK_REGION Region
)
/*++
Routine Description:
This routine does a FAT format on the given partition.
The caller should have cleared the screen and displayed
any message in the upper portion; this routine will
maintain the gas gauge in the lower portion of the screen.
Arguments:
Region - supplies the disk region descriptor for the
partition to be formatted.
Return Value:
--*/
{
ULONG hidden_sectors;
PULONG bad_sectors;
ULONG num_bad_sectors;
PVOID format_buffer;
PVOID unaligned_format_buffer;
ULONG max_sec_per_sa;
ULONG super_area_size;
PHARD_DISK pHardDisk;
ULONG PartitionOrdinal;
NTSTATUS Status;
HANDLE Handle;
ULONG BytesPerSector;
IO_STATUS_BLOCK IoStatusBlock;
LARGE_INTEGER LargeZero;
UCHAR SysId;
ULONG ActualSectorCount;
SET_PARTITION_INFORMATION PartitionInfo;
ASSERT(Region->PartitionedSpace);
ASSERT(Region->TablePosition < PTABLE_DIMENSION);
ASSERT(Region->Filesystem != FilesystemDoubleSpace);
pHardDisk = &HardDisks[Region->DiskNumber];
BytesPerSector = pHardDisk->Geometry.BytesPerSector;
PartitionOrdinal = SpPtGetOrdinal(Region,PartitionOrdinalCurrent);
//
// Make SURE it's not partition0! The results of formatting partition0
// are so disasterous that theis warrants a special check.
//
if(!PartitionOrdinal) {
SpBugCheck(
SETUP_BUGCHECK_PARTITION,
PARTITIONBUG_B,
Region->DiskNumber,
0
);
}
#ifdef _X86_
//
// If we're going to format C:, then clear the previous OS entry
// in boot.ini.
//
if(Region == SpPtValidSystemPartition()) {
*OldSystemLine = '\0';
}
#endif
//
// Query the number of hidden sectors and the actual number
// of sectors in the volume.
//
SpPtGetSectorLayoutInformation(Region,&hidden_sectors,&ActualSectorCount);
//
// Open the partition for read/write access.
// This shouldn't lock the volume so we need to lock it below.
//
Status = SpOpenPartition(
pHardDisk->DevicePath,
PartitionOrdinal,
&Handle,
TRUE
);
if(!NT_SUCCESS(Status)) {
KdPrintEx((DPFLTR_SETUP_ID, DPFLTR_ERROR_LEVEL,
"SETUP: SpFatFormat: unable to open %ws partition %u (%lx)\n",
pHardDisk->DevicePath,
PartitionOrdinal,
Status
));
return(Status);
}
//
// Lock the drive
//
Status = SpLockUnlockVolume( Handle, TRUE );
//
// We shouldn't have any file opened that would cause this volume
// to already be locked, so if we get failure (ie, STATUS_ACCESS_DENIED)
// something is really wrong. This typically indicates something is
// wrong with the hard disk that won't allow us to access it.
//
if(!NT_SUCCESS(Status)) {
KdPrintEx((DPFLTR_SETUP_ID, DPFLTR_ERROR_LEVEL, "SETUP: SpFatFormat: status %lx, unable to lock drive \n",Status));
ZwClose(Handle);
return(Status);
}
bad_sectors = NULL;
FmtVerifySectors(
Handle,
ActualSectorCount,
BytesPerSector,
&bad_sectors,
&num_bad_sectors
);
max_sec_per_sa = 1 +
2*((2*65536 - 1)/BytesPerSector + 1) +
((512*32 - 1)/BytesPerSector + 1);
unaligned_format_buffer = SpMemAlloc(max_sec_per_sa*BytesPerSector);
ASSERT(unaligned_format_buffer);
format_buffer = ALIGN(unaligned_format_buffer,BytesPerSector);
Status = FmtFillFormatBuffer(
ActualSectorCount,
BytesPerSector,
pHardDisk->Geometry.SectorsPerTrack,
pHardDisk->Geometry.TracksPerCylinder,
hidden_sectors,
format_buffer,
max_sec_per_sa*BytesPerSector,
&super_area_size,
bad_sectors,
num_bad_sectors,
&SysId
);
if(bad_sectors) {
SpMemFree(bad_sectors);
}
if(!NT_SUCCESS(Status)) {
KdPrintEx((DPFLTR_SETUP_ID, DPFLTR_ERROR_LEVEL, "SETUP: SpFatFormat: status %lx from FmtFillFormatBuffer\n",Status));
SpLockUnlockVolume( Handle, FALSE );
ZwClose(Handle);
SpMemFree(unaligned_format_buffer);
return(Status);
}
//
// Write the super area.
//
LargeZero.QuadPart = 0;
Status = ZwWriteFile(
Handle,
NULL,
NULL,
NULL,
&IoStatusBlock,
format_buffer,
super_area_size,
&LargeZero,
NULL
);
//
// I/O should be synchronous.
//
ASSERT(Status != STATUS_PENDING);
SpMemFree(unaligned_format_buffer);
if(!NT_SUCCESS(Status)) {
KdPrintEx((DPFLTR_SETUP_ID, DPFLTR_ERROR_LEVEL, "SETUP: SpFatFormat: status %lx from ZwWriteFile\n",Status));
SpLockUnlockVolume( Handle, FALSE );
ZwClose(Handle);
return(Status);
}
//
// If we wrote the super area then the drive is now FAT!
// If we don't change, say, a type of ntfs to fat, then code
// that lays down the x86 boot code (i386\bootini.c) will
// come along and write 16 sectors of NTFS boot code into
// sector 0 of our nice FAT volume -- very bad!
// Preserve the filesystem type of FilesystemNewlyCreated
// since other code later in setup relies on this.
//
if(Region->Filesystem >= FilesystemFirstKnown) {
Region->Filesystem = FilesystemFat;
}
//
// Set the partition type.
//
PartitionInfo.PartitionType = SysId;
Status = ZwDeviceIoControlFile(
Handle,
NULL,
NULL,
NULL,
&IoStatusBlock,
IOCTL_DISK_SET_PARTITION_INFO,
&PartitionInfo,
sizeof(PartitionInfo),
NULL,
0
);
if(!NT_SUCCESS(Status)) {
KdPrintEx((DPFLTR_SETUP_ID, DPFLTR_ERROR_LEVEL, "SETUP: unable to set partition type (status = %lx)\n",Status));
}
//
// Dismount the drive
//
Status = SpDismountVolume( Handle );
if(!NT_SUCCESS(Status)) {
KdPrintEx((DPFLTR_SETUP_ID, DPFLTR_ERROR_LEVEL, "SETUP: SpFatFormat: status %lx, unable to dismount drive\n",Status));
SpLockUnlockVolume( Handle, FALSE );
ZwClose(Handle);
return(Status);
}
//
// Unlock the drive
//
Status = SpLockUnlockVolume( Handle, FALSE );
if(!NT_SUCCESS(Status)) {
KdPrintEx((DPFLTR_SETUP_ID, DPFLTR_ERROR_LEVEL, "SETUP: SpFatFormat: status %lx, unable to unlock drive\n",Status));
}
ZwClose(Handle);
return(Status);
}
#endif