mirror of
https://github.com/reactos/reactos.git
synced 2024-11-30 23:13:31 +08:00
38791299ca
So that, coverity is happy again. NFC CID 1434230, 1434233, 1434238, 1434243, 1434246, 1434272
393 lines
12 KiB
C
393 lines
12 KiB
C
/*
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
* PROJECT: ReactOS VFAT filesystem library
|
|
* FILE: fat16.c
|
|
* PURPOSE: Fat16 support
|
|
* PROGRAMMERS: Casper S. Hornstrup (chorns@users.sourceforge.net)
|
|
* Eric Kohl
|
|
*/
|
|
|
|
/* INCLUDES *******************************************************************/
|
|
|
|
#include "vfatlib.h"
|
|
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
|
|
/* FUNCTIONS ******************************************************************/
|
|
|
|
static NTSTATUS
|
|
Fat16WriteBootSector(IN HANDLE FileHandle,
|
|
IN PFAT16_BOOT_SECTOR BootSector,
|
|
IN OUT PFORMAT_CONTEXT Context)
|
|
{
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
NTSTATUS Status;
|
|
PFAT16_BOOT_SECTOR NewBootSector;
|
|
LARGE_INTEGER FileOffset;
|
|
|
|
/* Allocate buffer for new bootsector */
|
|
NewBootSector = (PFAT16_BOOT_SECTOR)RtlAllocateHeap(RtlGetProcessHeap(),
|
|
0,
|
|
BootSector->BytesPerSector);
|
|
if (NewBootSector == NULL)
|
|
return STATUS_INSUFFICIENT_RESOURCES;
|
|
|
|
/* Zero the new bootsector */
|
|
RtlZeroMemory(NewBootSector, BootSector->BytesPerSector);
|
|
|
|
/* Copy FAT16 BPB to new bootsector */
|
|
memcpy(NewBootSector, BootSector,
|
|
FIELD_OFFSET(FAT16_BOOT_SECTOR, Res2) - FIELD_OFFSET(FAT16_BOOT_SECTOR, Jump));
|
|
/* FAT16 BPB length (up to (not including) Res2) */
|
|
|
|
/* Write the boot sector signature */
|
|
NewBootSector->Signature1 = 0xAA550000;
|
|
|
|
/* Write sector 0 */
|
|
FileOffset.QuadPart = 0ULL;
|
|
Status = NtWriteFile(FileHandle,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
NewBootSector,
|
|
BootSector->BytesPerSector,
|
|
&FileOffset,
|
|
NULL);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("NtWriteFile() failed (Status %lx)\n", Status);
|
|
goto done;
|
|
}
|
|
|
|
UpdateProgress(Context, 1);
|
|
|
|
done:
|
|
/* Free the buffer */
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, NewBootSector);
|
|
return Status;
|
|
}
|
|
|
|
|
|
static NTSTATUS
|
|
Fat16WriteFAT(IN HANDLE FileHandle,
|
|
IN ULONG SectorOffset,
|
|
IN PFAT16_BOOT_SECTOR BootSector,
|
|
IN OUT PFORMAT_CONTEXT Context)
|
|
{
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
NTSTATUS Status;
|
|
PUCHAR Buffer;
|
|
LARGE_INTEGER FileOffset;
|
|
ULONG i;
|
|
ULONG Sectors;
|
|
|
|
/* Allocate buffer */
|
|
Buffer = (PUCHAR)RtlAllocateHeap(RtlGetProcessHeap(),
|
|
0,
|
|
32 * 1024);
|
|
if (Buffer == NULL)
|
|
return STATUS_INSUFFICIENT_RESOURCES;
|
|
|
|
/* Zero the buffer */
|
|
RtlZeroMemory(Buffer, 32 * 1024);
|
|
|
|
/* FAT cluster 0 */
|
|
Buffer[0] = 0xf8; /* Media type */
|
|
Buffer[1] = 0xff;
|
|
|
|
/* FAT cluster 1 */
|
|
Buffer[2] = 0xff; /* Clean shutdown, no disk read/write errors, end-of-cluster (EOC) mark */
|
|
Buffer[3] = 0xff;
|
|
|
|
/* Write first sector of the FAT */
|
|
FileOffset.QuadPart = (SectorOffset + BootSector->ReservedSectors) * BootSector->BytesPerSector;
|
|
Status = NtWriteFile(FileHandle,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
Buffer,
|
|
BootSector->BytesPerSector,
|
|
&FileOffset,
|
|
NULL);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("NtWriteFile() failed (Status %lx)\n", Status);
|
|
goto done;
|
|
}
|
|
|
|
UpdateProgress(Context, 1);
|
|
|
|
/* Zero the begin of the buffer */
|
|
RtlZeroMemory(Buffer, 4);
|
|
|
|
/* Zero the rest of the FAT */
|
|
Sectors = 32 * 1024 / BootSector->BytesPerSector;
|
|
for (i = 1; i < (ULONG)BootSector->FATSectors; i += Sectors)
|
|
{
|
|
/* Zero some sectors of the FAT */
|
|
FileOffset.QuadPart = (SectorOffset + BootSector->ReservedSectors + i) * BootSector->BytesPerSector;
|
|
|
|
if (((ULONG)BootSector->FATSectors - i) <= Sectors)
|
|
{
|
|
Sectors = (ULONG)BootSector->FATSectors - i;
|
|
}
|
|
|
|
Status = NtWriteFile(FileHandle,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
Buffer,
|
|
Sectors * BootSector->BytesPerSector,
|
|
&FileOffset,
|
|
NULL);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("NtWriteFile() failed (Status %lx)\n", Status);
|
|
goto done;
|
|
}
|
|
|
|
UpdateProgress(Context, Sectors);
|
|
}
|
|
|
|
done:
|
|
/* Free the buffer */
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer);
|
|
return Status;
|
|
}
|
|
|
|
|
|
static NTSTATUS
|
|
Fat16WriteRootDirectory(IN HANDLE FileHandle,
|
|
IN PFAT16_BOOT_SECTOR BootSector,
|
|
IN OUT PFORMAT_CONTEXT Context)
|
|
{
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
NTSTATUS Status = STATUS_SUCCESS;
|
|
PUCHAR Buffer;
|
|
LARGE_INTEGER FileOffset;
|
|
ULONG FirstRootDirSector;
|
|
ULONG RootDirSectors;
|
|
ULONG Sectors;
|
|
ULONG i;
|
|
|
|
DPRINT("BootSector->ReservedSectors = %hu\n", BootSector->ReservedSectors);
|
|
DPRINT("BootSector->FATSectors = %hu\n", BootSector->FATSectors);
|
|
DPRINT("BootSector->SectorsPerCluster = %u\n", BootSector->SectorsPerCluster);
|
|
|
|
/* Write cluster */
|
|
RootDirSectors = ((BootSector->RootEntries * 32) +
|
|
(BootSector->BytesPerSector - 1)) / BootSector->BytesPerSector;
|
|
FirstRootDirSector =
|
|
BootSector->ReservedSectors + (BootSector->FATCount * BootSector->FATSectors);
|
|
|
|
DPRINT("RootDirSectors = %lu\n", RootDirSectors);
|
|
DPRINT("FirstRootDirSector = %lu\n", FirstRootDirSector);
|
|
|
|
/* Allocate buffer for the cluster */
|
|
Buffer = (PUCHAR)RtlAllocateHeap(RtlGetProcessHeap(),
|
|
0,
|
|
32 * 1024);
|
|
if (Buffer == NULL)
|
|
return STATUS_INSUFFICIENT_RESOURCES;
|
|
|
|
/* Zero the buffer */
|
|
RtlZeroMemory(Buffer, 32 * 1024);
|
|
|
|
Sectors = 32 * 1024 / BootSector->BytesPerSector;
|
|
for (i = 0; i < RootDirSectors; i += Sectors)
|
|
{
|
|
/* Zero some sectors of the root directory */
|
|
FileOffset.QuadPart = (FirstRootDirSector + i) * BootSector->BytesPerSector;
|
|
|
|
if ((RootDirSectors - i) <= Sectors)
|
|
{
|
|
Sectors = RootDirSectors - i;
|
|
}
|
|
|
|
Status = NtWriteFile(FileHandle,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&IoStatusBlock,
|
|
Buffer,
|
|
Sectors * BootSector->BytesPerSector,
|
|
&FileOffset,
|
|
NULL);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("NtWriteFile() failed (Status %lx)\n", Status);
|
|
goto done;
|
|
}
|
|
|
|
UpdateProgress(Context, Sectors);
|
|
}
|
|
|
|
done:
|
|
/* Free the buffer */
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer);
|
|
return Status;
|
|
}
|
|
|
|
|
|
NTSTATUS
|
|
Fat16Format(IN HANDLE FileHandle,
|
|
IN PPARTITION_INFORMATION PartitionInfo,
|
|
IN PDISK_GEOMETRY DiskGeometry,
|
|
IN PUNICODE_STRING Label,
|
|
IN BOOLEAN QuickFormat,
|
|
IN ULONG ClusterSize,
|
|
IN OUT PFORMAT_CONTEXT Context)
|
|
{
|
|
FAT16_BOOT_SECTOR BootSector;
|
|
OEM_STRING VolumeLabel;
|
|
ULONG SectorCount;
|
|
ULONG RootDirSectors;
|
|
ULONG TmpVal1;
|
|
ULONG TmpVal2;
|
|
ULONG TmpVal3;
|
|
NTSTATUS Status;
|
|
|
|
/* Calculate cluster size */
|
|
if (ClusterSize == 0)
|
|
{
|
|
if (PartitionInfo->PartitionLength.QuadPart < 16LL * 1024LL * 1024LL)
|
|
{
|
|
/* Partition < 16MB ==> 1KB Cluster */
|
|
ClusterSize = 1024;
|
|
}
|
|
else if (PartitionInfo->PartitionLength.QuadPart < 128LL * 1024LL * 1024LL)
|
|
{
|
|
/* Partition < 128MB ==> 2KB Cluster */
|
|
ClusterSize = 2048;
|
|
}
|
|
else if (PartitionInfo->PartitionLength.QuadPart < 256LL * 1024LL * 1024LL)
|
|
{
|
|
/* Partition < 256MB ==> 4KB Cluster */
|
|
ClusterSize = 4096;
|
|
}
|
|
else
|
|
{
|
|
/* Partition >= 256MB (< 512MB) ==> 8KB Cluster */
|
|
ClusterSize = 8192;
|
|
}
|
|
}
|
|
|
|
SectorCount = PartitionInfo->PartitionLength.QuadPart >>
|
|
GetShiftCount(DiskGeometry->BytesPerSector); /* Use shifting to avoid 64-bit division */
|
|
|
|
RtlZeroMemory(&BootSector, sizeof(FAT16_BOOT_SECTOR));
|
|
memcpy(&BootSector.OEMName[0], "MSWIN4.1", 8);
|
|
/* FIXME: Add dummy bootloader for real */
|
|
BootSector.Jump[0] = 0xeb;
|
|
BootSector.Jump[1] = 0x3c;
|
|
BootSector.Jump[2] = 0x90;
|
|
BootSector.BytesPerSector = DiskGeometry->BytesPerSector;
|
|
BootSector.SectorsPerCluster = ClusterSize / BootSector.BytesPerSector;
|
|
BootSector.ReservedSectors = 1;
|
|
BootSector.FATCount = 2;
|
|
BootSector.RootEntries = 512;
|
|
BootSector.Sectors = (SectorCount < 0x10000) ? (unsigned short)SectorCount : 0;
|
|
BootSector.Media = 0xf8;
|
|
BootSector.FATSectors = 0; /* Set later. See below. */
|
|
BootSector.SectorsPerTrack = DiskGeometry->SectorsPerTrack;
|
|
BootSector.Heads = DiskGeometry->TracksPerCylinder;
|
|
BootSector.HiddenSectors = PartitionInfo->HiddenSectors;
|
|
BootSector.SectorsHuge = (SectorCount >= 0x10000) ? (unsigned long)SectorCount : 0;
|
|
BootSector.Drive = (DiskGeometry->MediaType == FixedMedia) ? 0x80 : 0x00;
|
|
BootSector.ExtBootSignature = 0x29;
|
|
BootSector.VolumeID = CalcVolumeSerialNumber();
|
|
if ((Label == NULL) || (Label->Buffer == NULL))
|
|
{
|
|
memcpy(&BootSector.VolumeLabel[0], "NO NAME ", 11);
|
|
}
|
|
else
|
|
{
|
|
RtlUnicodeStringToOemString(&VolumeLabel, Label, TRUE);
|
|
RtlFillMemory(&BootSector.VolumeLabel[0], 11, ' ');
|
|
memcpy(&BootSector.VolumeLabel[0], VolumeLabel.Buffer,
|
|
VolumeLabel.Length < 11 ? VolumeLabel.Length : 11);
|
|
RtlFreeOemString(&VolumeLabel);
|
|
}
|
|
|
|
memcpy(&BootSector.SysType[0], "FAT16 ", 8);
|
|
|
|
DPRINT("BootSector.SectorsHuge = %lx\n", BootSector.SectorsHuge);
|
|
|
|
RootDirSectors = ((BootSector.RootEntries * 32) +
|
|
(BootSector.BytesPerSector - 1)) / BootSector.BytesPerSector;
|
|
|
|
/* Calculate number of FAT sectors */
|
|
/* (BootSector.BytesPerSector / 2) FAT entries (16bit) fit into one sector */
|
|
TmpVal1 = SectorCount - (BootSector.ReservedSectors + RootDirSectors);
|
|
TmpVal2 = ((BootSector.BytesPerSector / 2) * BootSector.SectorsPerCluster) + BootSector.FATCount;
|
|
TmpVal3 = (TmpVal1 + (TmpVal2 - 1)) / TmpVal2;
|
|
BootSector.FATSectors = (unsigned short)(TmpVal3 & 0xffff);
|
|
DPRINT("BootSector.FATSectors = %hu\n", BootSector.FATSectors);
|
|
|
|
/* Init context data */
|
|
Context->TotalSectorCount =
|
|
1 + (BootSector.FATSectors * 2) + RootDirSectors;
|
|
|
|
if (!QuickFormat)
|
|
{
|
|
Context->TotalSectorCount += SectorCount;
|
|
|
|
Status = FatWipeSectors(FileHandle,
|
|
SectorCount,
|
|
(ULONG)BootSector.SectorsPerCluster,
|
|
(ULONG)BootSector.BytesPerSector,
|
|
Context);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("FatWipeSectors() failed with status 0x%.08x\n", Status);
|
|
return Status;
|
|
}
|
|
}
|
|
|
|
Status = Fat16WriteBootSector(FileHandle,
|
|
&BootSector,
|
|
Context);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("Fat16WriteBootSector() failed with status 0x%.08x\n", Status);
|
|
return Status;
|
|
}
|
|
|
|
/* Write first FAT copy */
|
|
Status = Fat16WriteFAT(FileHandle,
|
|
0,
|
|
&BootSector,
|
|
Context);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("Fat16WriteFAT() failed with status 0x%.08x\n", Status);
|
|
return Status;
|
|
}
|
|
|
|
/* Write second FAT copy */
|
|
Status = Fat16WriteFAT(FileHandle,
|
|
(ULONG)BootSector.FATSectors,
|
|
&BootSector,
|
|
Context);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("Fat16WriteFAT() failed with status 0x%.08x.\n", Status);
|
|
return Status;
|
|
}
|
|
|
|
Status = Fat16WriteRootDirectory(FileHandle,
|
|
&BootSector,
|
|
Context);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT("Fat16WriteRootDirectory() failed with status 0x%.08x\n", Status);
|
|
}
|
|
|
|
return Status;
|
|
}
|