/* FUNCTIONS ****************************************************************/
-static VOID
-GetDriverName (PDISKENTRY DiskEntry)
+static
+VOID
+GetDriverName(
+ PDISKENTRY DiskEntry)
{
- RTL_QUERY_REGISTRY_TABLE QueryTable[2];
- WCHAR KeyName[32];
- NTSTATUS Status;
+ RTL_QUERY_REGISTRY_TABLE QueryTable[2];
+ WCHAR KeyName[32];
+ NTSTATUS Status;
- RtlInitUnicodeString (&DiskEntry->DriverName,
- NULL);
+ RtlInitUnicodeString(&DiskEntry->DriverName,
+ NULL);
- swprintf (KeyName,
- L"\\Scsi\\Scsi Port %lu",
- DiskEntry->Port);
+ swprintf(KeyName,
+ L"\\Scsi\\Scsi Port %lu",
+ DiskEntry->Port);
- RtlZeroMemory (&QueryTable,
- sizeof(QueryTable));
+ RtlZeroMemory(&QueryTable,
+ sizeof(QueryTable));
- QueryTable[0].Name = L"Driver";
- QueryTable[0].Flags = RTL_QUERY_REGISTRY_DIRECT;
- QueryTable[0].EntryContext = &DiskEntry->DriverName;
+ QueryTable[0].Name = L"Driver";
+ QueryTable[0].Flags = RTL_QUERY_REGISTRY_DIRECT;
+ QueryTable[0].EntryContext = &DiskEntry->DriverName;
- Status = RtlQueryRegistryValues (RTL_REGISTRY_DEVICEMAP,
- KeyName,
- QueryTable,
- NULL,
- NULL);
- if (!NT_SUCCESS (Status))
- {
- DPRINT1 ("RtlQueryRegistryValues() failed (Status %lx)\n", Status);
- }
+ Status = RtlQueryRegistryValues(RTL_REGISTRY_DEVICEMAP,
+ KeyName,
+ QueryTable,
+ NULL,
+ NULL);
+ if (!NT_SUCCESS(Status))
+ {
+ DPRINT1("RtlQueryRegistryValues() failed (Status %lx)\n", Status);
+ }
}
-static VOID
-AssignDriverLetters (PPARTLIST List)
+static
+VOID
+AssignDriverLetters(
+ PPARTLIST List)
{
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry1;
- //PLIST_ENTRY Entry2;
- CHAR Letter;
- UCHAR i;
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry1;
+ //PLIST_ENTRY Entry2;
+ CHAR Letter;
+ UCHAR i;
- Letter = 'C';
+ Letter = 'C';
- /* Assign drive letters to primary partitions */
- Entry1 = List->DiskListHead.Flink;
- while (Entry1 != &List->DiskListHead)
- {
- DiskEntry = CONTAINING_RECORD (Entry1, DISKENTRY, ListEntry);
-
- if (!IsListEmpty (&DiskEntry->PartListHead))
+ /* Assign drive letters to primary partitions */
+ Entry1 = List->DiskListHead.Flink;
+ while (Entry1 != &List->DiskListHead)
{
- PartEntry = CONTAINING_RECORD (DiskEntry->PartListHead.Flink,
- PARTENTRY,
- ListEntry);
+ DiskEntry = CONTAINING_RECORD(Entry1, DISKENTRY, ListEntry);
- for (i=0; i<4; i++)
- PartEntry->DriveLetter[i] = 0;
-
- if (PartEntry->Unpartitioned == FALSE)
- {
- for (i=0; i<4; i++)
+ if (!IsListEmpty(&DiskEntry->PartListHead))
{
- if (IsContainerPartition (PartEntry->PartInfo[i].PartitionType))
- continue;
+ PartEntry = CONTAINING_RECORD(DiskEntry->PartListHead.Flink,
+ PARTENTRY,
+ ListEntry);
- if (IsRecognizedPartition (PartEntry->PartInfo[i].PartitionType) ||
- (PartEntry->PartInfo[i].PartitionType == PARTITION_ENTRY_UNUSED &&
- PartEntry->PartInfo[i].PartitionLength.QuadPart != 0LL))
- {
- if (Letter <= 'Z')
+ for (i = 0; i < 4; i++)
+ PartEntry->DriveLetter[i] = 0;
+
+ if (PartEntry->Unpartitioned == FALSE)
{
- PartEntry->DriveLetter[i] = Letter;
- Letter++;
+ for (i = 0; i < 4; i++)
+ {
+ if (IsContainerPartition(PartEntry->PartInfo[i].PartitionType))
+ continue;
+
+ if (IsRecognizedPartition(PartEntry->PartInfo[i].PartitionType) ||
+ (PartEntry->PartInfo[i].PartitionType == PARTITION_ENTRY_UNUSED &&
+ PartEntry->PartInfo[i].PartitionLength.QuadPart != 0LL))
+ {
+ if (Letter <= 'Z')
+ {
+ PartEntry->DriveLetter[i] = Letter;
+ Letter++;
+ }
+ }
+ }
}
- }
}
- }
- }
- Entry1 = Entry1->Flink;
- }
+ Entry1 = Entry1->Flink;
+ }
- /* Assign drive letters to logical drives */
+ /* Assign drive letters to logical drives */
#if 0
- Entry1 = List->DiskListHead.Flink;
- while (Entry1 != &List->DiskListHead)
- {
- DiskEntry = CONTAINING_RECORD (Entry1, DISKENTRY, ListEntry);
-
- Entry2 = DiskEntry->PartListHead.Flink;
- if (Entry2 != &DiskEntry->PartListHead)
- {
- Entry2 = Entry2->Flink;
- while (Entry2 != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry2,
- PARTENTRY,
- ListEntry);
-
- PartEntry->DriveLetter = 0;
+ Entry1 = List->DiskListHead.Flink;
+ while (Entry1 != &List->DiskListHead)
+ {
+ DiskEntry = CONTAINING_RECORD(Entry1, DISKENTRY, ListEntry);
- if (PartEntry->Unpartitioned == FALSE &&
- !IsContainerPartition (PartEntry->PartInfo[0].PartitionType))
+ Entry2 = DiskEntry->PartListHead.Flink;
+ if (Entry2 != &DiskEntry->PartListHead)
{
- if (IsRecognizedPartition (PartEntry->PartInfo[0].PartitionType) ||
- (PartEntry->PartInfo[0].PartitionType == PARTITION_ENTRY_UNUSED &&
- PartEntry->PartInfo[0].PartitionLength.QuadPart != 0LL))
- {
- if (Letter <= 'Z')
+ Entry2 = Entry2->Flink;
+ while (Entry2 != &DiskEntry->PartListHead)
{
- PartEntry->DriveLetter = Letter;
- Letter++;
+ PartEntry = CONTAINING_RECORD(Entry2,
+ PARTENTRY,
+ ListEntry);
+
+ PartEntry->DriveLetter = 0;
+
+ if (PartEntry->Unpartitioned == FALSE &&
+ !IsContainerPartition(PartEntry->PartInfo[0].PartitionType))
+ {
+ if (IsRecognizedPartition(PartEntry->PartInfo[0].PartitionType) ||
+ (PartEntry->PartInfo[0].PartitionType == PARTITION_ENTRY_UNUSED &&
+ PartEntry->PartInfo[0].PartitionLength.QuadPart != 0LL))
+ {
+ if (Letter <= 'Z')
+ {
+ PartEntry->DriveLetter = Letter;
+ Letter++;
+ }
+ }
+ }
+
+ Entry2 = Entry2->Flink;
}
- }
}
- Entry2 = Entry2->Flink;
- }
+ Entry1 = Entry1->Flink;
}
-
- Entry1 = Entry1->Flink;
- }
#endif
}
-static VOID
-UpdatePartitionNumbers (PDISKENTRY DiskEntry)
+static
+VOID
+UpdatePartitionNumbers(
+ PDISKENTRY DiskEntry)
{
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry;
- ULONG PartNumber;
- ULONG i;
-
- PartNumber = 1;
- Entry = DiskEntry->PartListHead.Flink;
- while (Entry != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry,
- PARTENTRY,
- ListEntry);
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry;
+ ULONG PartNumber;
+ ULONG i;
- if (PartEntry->Unpartitioned == TRUE)
- {
- for (i = 0; i < 4; i++)
- {
- PartEntry->PartInfo[i].PartitionNumber = 0;
- }
- }
- else
+ PartNumber = 1;
+ Entry = DiskEntry->PartListHead.Flink;
+ while (Entry != &DiskEntry->PartListHead)
{
- for (i = 0; i < 4; i++)
- {
- if (IsContainerPartition (PartEntry->PartInfo[i].PartitionType))
- {
- PartEntry->PartInfo[i].PartitionNumber = 0;
- }
- else if (PartEntry->PartInfo[i].PartitionType == PARTITION_ENTRY_UNUSED &&
- PartEntry->PartInfo[i].PartitionLength.QuadPart == 0ULL)
+ PartEntry = CONTAINING_RECORD(Entry,
+ PARTENTRY,
+ ListEntry);
+
+ if (PartEntry->Unpartitioned == TRUE)
{
- PartEntry->PartInfo[i].PartitionNumber = 0;
+ for (i = 0; i < 4; i++)
+ {
+ PartEntry->PartInfo[i].PartitionNumber = 0;
+ }
}
else
{
- PartEntry->PartInfo[i].PartitionNumber = PartNumber;
- PartNumber++;
+ for (i = 0; i < 4; i++)
+ {
+ if (IsContainerPartition(PartEntry->PartInfo[i].PartitionType))
+ {
+ PartEntry->PartInfo[i].PartitionNumber = 0;
+ }
+ else if (PartEntry->PartInfo[i].PartitionType == PARTITION_ENTRY_UNUSED &&
+ PartEntry->PartInfo[i].PartitionLength.QuadPart == 0ULL)
+ {
+ PartEntry->PartInfo[i].PartitionNumber = 0;
+ }
+ else
+ {
+ PartEntry->PartInfo[i].PartitionNumber = PartNumber;
+ PartNumber++;
+ }
+ }
}
- }
- }
- Entry = Entry->Flink;
- }
+ Entry = Entry->Flink;
+ }
}
-static VOID
-AddPartitionToList (ULONG DiskNumber,
- PDISKENTRY DiskEntry,
- DRIVE_LAYOUT_INFORMATION *LayoutBuffer)
+static
+VOID
+AddPartitionToList(
+ ULONG DiskNumber,
+ PDISKENTRY DiskEntry,
+ DRIVE_LAYOUT_INFORMATION *LayoutBuffer)
{
- PPARTENTRY PartEntry;
- ULONG i;
- ULONG j;
+ PPARTENTRY PartEntry;
+ ULONG i;
+ ULONG j;
- for (i = 0; i < LayoutBuffer->PartitionCount; i += 4)
- {
- for (j = 0; j < 4; j++)
+ for (i = 0; i < LayoutBuffer->PartitionCount; i += 4)
{
- if (LayoutBuffer->PartitionEntry[i+j].PartitionType != PARTITION_ENTRY_UNUSED ||
- LayoutBuffer->PartitionEntry[i+j].PartitionLength.QuadPart != 0ULL)
- {
- break;
- }
- }
- if (j >= 4)
- {
- continue;
- }
+ for (j = 0; j < 4; j++)
+ {
+ if (LayoutBuffer->PartitionEntry[i+j].PartitionType != PARTITION_ENTRY_UNUSED ||
+ LayoutBuffer->PartitionEntry[i+j].PartitionLength.QuadPart != 0ULL)
+ {
+ break;
+ }
+ }
- PartEntry = (PPARTENTRY)RtlAllocateHeap (ProcessHeap,
- 0,
- sizeof(PARTENTRY));
- if (PartEntry == NULL)
- {
- return;
- }
+ if (j >= 4)
+ {
+ continue;
+ }
- RtlZeroMemory (PartEntry,
- sizeof(PARTENTRY));
+ PartEntry = (PPARTENTRY)RtlAllocateHeap(ProcessHeap,
+ 0,
+ sizeof(PARTENTRY));
+ if (PartEntry == NULL)
+ {
+ return;
+ }
- PartEntry->Unpartitioned = FALSE;
+ RtlZeroMemory(PartEntry,
+ sizeof(PARTENTRY));
- for (j = 0; j < 4; j++)
- {
- RtlCopyMemory (&PartEntry->PartInfo[j],
- &LayoutBuffer->PartitionEntry[i+j],
- sizeof(PARTITION_INFORMATION));
- }
+ PartEntry->Unpartitioned = FALSE;
- if (IsContainerPartition(PartEntry->PartInfo[0].PartitionType))
- {
- PartEntry->FormatState = Unformatted;
- }
- else if ((PartEntry->PartInfo[0].PartitionType == PARTITION_FAT_12) ||
- (PartEntry->PartInfo[0].PartitionType == PARTITION_FAT_16) ||
- (PartEntry->PartInfo[0].PartitionType == PARTITION_HUGE) ||
- (PartEntry->PartInfo[0].PartitionType == PARTITION_XINT13) ||
- (PartEntry->PartInfo[0].PartitionType == PARTITION_FAT32) ||
- (PartEntry->PartInfo[0].PartitionType == PARTITION_FAT32_XINT13))
- {
+ for (j = 0; j < 4; j++)
+ {
+ RtlCopyMemory(&PartEntry->PartInfo[j],
+ &LayoutBuffer->PartitionEntry[i+j],
+ sizeof(PARTITION_INFORMATION));
+ }
+
+ if (IsContainerPartition(PartEntry->PartInfo[0].PartitionType))
+ {
+ PartEntry->FormatState = Unformatted;
+ }
+ else if ((PartEntry->PartInfo[0].PartitionType == PARTITION_FAT_12) ||
+ (PartEntry->PartInfo[0].PartitionType == PARTITION_FAT_16) ||
+ (PartEntry->PartInfo[0].PartitionType == PARTITION_HUGE) ||
+ (PartEntry->PartInfo[0].PartitionType == PARTITION_XINT13) ||
+ (PartEntry->PartInfo[0].PartitionType == PARTITION_FAT32) ||
+ (PartEntry->PartInfo[0].PartitionType == PARTITION_FAT32_XINT13))
+ {
#if 0
- if (CheckFatFormat())
- {
- PartEntry->FormatState = Preformatted;
- }
- else
- {
- PartEntry->FormatState = Unformatted;
- }
+ if (CheckFatFormat())
+ {
+ PartEntry->FormatState = Preformatted;
+ }
+ else
+ {
+ PartEntry->FormatState = Unformatted;
+ }
#endif
- PartEntry->FormatState = Preformatted;
- }
- else if (PartEntry->PartInfo[0].PartitionType == PARTITION_EXT2)
- {
+ PartEntry->FormatState = Preformatted;
+ }
+ else if (PartEntry->PartInfo[0].PartitionType == PARTITION_EXT2)
+ {
#if 0
- if (CheckExt2Format())
- {
- PartEntry->FormatState = Preformatted;
- }
- else
- {
- PartEntry->FormatState = Unformatted;
- }
+ if (CheckExt2Format())
+ {
+ PartEntry->FormatState = Preformatted;
+ }
+ else
+ {
+ PartEntry->FormatState = Unformatted;
+ }
#endif
- PartEntry->FormatState = Preformatted;
- }
- else if (PartEntry->PartInfo[0].PartitionType == PARTITION_IFS)
- {
+ PartEntry->FormatState = Preformatted;
+ }
+ else if (PartEntry->PartInfo[0].PartitionType == PARTITION_IFS)
+ {
#if 0
- if (CheckNtfsFormat())
- {
- PartEntry->FormatState = Preformatted;
- }
- else if (CheckHpfsFormat())
- {
- PartEntry->FormatState = Preformatted;
- }
- else
- {
- PartEntry->FormatState = Unformatted;
- }
+ if (CheckNtfsFormat())
+ {
+ PartEntry->FormatState = Preformatted;
+ }
+ else if (CheckHpfsFormat())
+ {
+ PartEntry->FormatState = Preformatted;
+ }
+ else
+ {
+ PartEntry->FormatState = Unformatted;
+ }
#endif
- PartEntry->FormatState = Preformatted;
- }
- else
- {
- PartEntry->FormatState = UnknownFormat;
- }
+ PartEntry->FormatState = Preformatted;
+ }
+ else
+ {
+ PartEntry->FormatState = UnknownFormat;
+ }
- InsertTailList (&DiskEntry->PartListHead,
- &PartEntry->ListEntry);
- }
+ InsertTailList(&DiskEntry->PartListHead,
+ &PartEntry->ListEntry);
+ }
}
-static VOID
-ScanForUnpartitionedDiskSpace (PDISKENTRY DiskEntry)
+static
+VOID
+ScanForUnpartitionedDiskSpace(
+ PDISKENTRY DiskEntry)
{
- ULONGLONG LastStartingOffset;
- ULONGLONG LastPartitionLength;
- ULONGLONG LastUnusedPartitionLength;
- PPARTENTRY PartEntry;
- PPARTENTRY NewPartEntry;
- PLIST_ENTRY Entry;
- ULONG i;
- ULONG j;
-
- if (IsListEmpty (&DiskEntry->PartListHead))
- {
- /* Create a partition table that represents the empty disk */
- PartEntry = (PPARTENTRY)RtlAllocateHeap (ProcessHeap,
- 0,
- sizeof(PARTENTRY));
- if (PartEntry == NULL)
- return;
-
- RtlZeroMemory (PartEntry,
- sizeof(PARTENTRY));
-
- PartEntry->Unpartitioned = TRUE;
- PartEntry->UnpartitionedOffset = 0ULL;
- PartEntry->UnpartitionedLength = DiskEntry->DiskSize;
-
- PartEntry->FormatState = Unformatted;
-
- InsertTailList (&DiskEntry->PartListHead,
- &PartEntry->ListEntry);
- }
- else
- {
- /* Start partition at head 1, cylinder 0 */
- LastStartingOffset = DiskEntry->TrackSize;
- LastPartitionLength = 0ULL;
- LastUnusedPartitionLength = 0ULL;
-
- i = 0;
- Entry = DiskEntry->PartListHead.Flink;
- while (Entry != &DiskEntry->PartListHead)
+ ULONGLONG LastStartingOffset;
+ ULONGLONG LastPartitionLength;
+ ULONGLONG LastUnusedPartitionLength;
+ PPARTENTRY PartEntry;
+ PPARTENTRY NewPartEntry;
+ PLIST_ENTRY Entry;
+ ULONG i;
+ ULONG j;
+
+ if (IsListEmpty (&DiskEntry->PartListHead))
{
- PartEntry = CONTAINING_RECORD (Entry, PARTENTRY, ListEntry);
+ /* Create a partition table that represents the empty disk */
+ PartEntry = (PPARTENTRY)RtlAllocateHeap(ProcessHeap,
+ 0,
+ sizeof(PARTENTRY));
+ if (PartEntry == NULL)
+ return;
- for (j = 0; j < 4; j++)
- {
- if ((!IsContainerPartition (PartEntry->PartInfo[j].PartitionType)) &&
- (PartEntry->PartInfo[j].PartitionType != PARTITION_ENTRY_UNUSED ||
- PartEntry->PartInfo[j].PartitionLength.QuadPart != 0LL))
- {
- LastUnusedPartitionLength =
- PartEntry->PartInfo[j].StartingOffset.QuadPart -
- (LastStartingOffset + LastPartitionLength);
-
- if (PartEntry->PartInfo[j].StartingOffset.QuadPart > (LastStartingOffset + LastPartitionLength) &&
- LastUnusedPartitionLength >= DiskEntry->CylinderSize)
- {
- DPRINT ("Unpartitioned disk space %I64u\n", LastUnusedPartitionLength);
-
- NewPartEntry = (PPARTENTRY)RtlAllocateHeap (ProcessHeap,
- 0,
- sizeof(PARTENTRY));
- if (NewPartEntry == NULL)
- return;
-
- RtlZeroMemory (NewPartEntry,
- sizeof(PARTENTRY));
-
- NewPartEntry->Unpartitioned = TRUE;
- NewPartEntry->UnpartitionedOffset = LastStartingOffset + LastPartitionLength;
- NewPartEntry->UnpartitionedLength = LastUnusedPartitionLength;
- if (j == 0)
- NewPartEntry->UnpartitionedLength -= DiskEntry->TrackSize;
-
- NewPartEntry->FormatState = Unformatted;
-
- /* Insert the table into the list */
- InsertTailList (&PartEntry->ListEntry,
- &NewPartEntry->ListEntry);
- }
-
- LastStartingOffset = PartEntry->PartInfo[j].StartingOffset.QuadPart;
- LastPartitionLength = PartEntry->PartInfo[j].PartitionLength.QuadPart;
- }
- }
+ RtlZeroMemory(PartEntry,
+ sizeof(PARTENTRY));
- i += 4;
- Entry = Entry->Flink;
- }
+ PartEntry->Unpartitioned = TRUE;
+ PartEntry->UnpartitionedOffset = 0ULL;
+ PartEntry->UnpartitionedLength = DiskEntry->DiskSize;
- /* Check for trailing unpartitioned disk space */
- if (DiskEntry->DiskSize > (LastStartingOffset + LastPartitionLength))
+ PartEntry->FormatState = Unformatted;
+
+ InsertTailList(&DiskEntry->PartListHead,
+ &PartEntry->ListEntry);
+ }
+ else
{
- /* Round-down to cylinder size */
- LastUnusedPartitionLength =
- (DiskEntry->DiskSize - (LastStartingOffset + LastPartitionLength))
- & ~(DiskEntry->CylinderSize - 1);
+ /* Start partition at head 1, cylinder 0 */
+ LastStartingOffset = DiskEntry->TrackSize;
+ LastPartitionLength = 0ULL;
+ LastUnusedPartitionLength = 0ULL;
+
+ i = 0;
+ Entry = DiskEntry->PartListHead.Flink;
+ while (Entry != &DiskEntry->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);
- if (LastUnusedPartitionLength >= DiskEntry->CylinderSize)
- {
- DPRINT ("Unpartitioned disk space %I64u\n", LastUnusedPartitionLength);
+ for (j = 0; j < 4; j++)
+ {
+ if ((!IsContainerPartition (PartEntry->PartInfo[j].PartitionType)) &&
+ (PartEntry->PartInfo[j].PartitionType != PARTITION_ENTRY_UNUSED ||
+ PartEntry->PartInfo[j].PartitionLength.QuadPart != 0LL))
+ {
+ LastUnusedPartitionLength =
+ PartEntry->PartInfo[j].StartingOffset.QuadPart -
+ (LastStartingOffset + LastPartitionLength);
+
+ if (PartEntry->PartInfo[j].StartingOffset.QuadPart > (LastStartingOffset + LastPartitionLength) &&
+ LastUnusedPartitionLength >= DiskEntry->CylinderSize)
+ {
+ DPRINT("Unpartitioned disk space %I64u\n", LastUnusedPartitionLength);
+
+ NewPartEntry = (PPARTENTRY)RtlAllocateHeap(ProcessHeap,
+ 0,
+ sizeof(PARTENTRY));
+ if (NewPartEntry == NULL)
+ return;
+
+ RtlZeroMemory(NewPartEntry,
+ sizeof(PARTENTRY));
+
+ NewPartEntry->Unpartitioned = TRUE;
+ NewPartEntry->UnpartitionedOffset = LastStartingOffset + LastPartitionLength;
+ NewPartEntry->UnpartitionedLength = LastUnusedPartitionLength;
+ if (j == 0)
+ NewPartEntry->UnpartitionedLength -= DiskEntry->TrackSize;
+
+ NewPartEntry->FormatState = Unformatted;
+
+ /* Insert the table into the list */
+ InsertTailList(&PartEntry->ListEntry,
+ &NewPartEntry->ListEntry);
+ }
+
+ LastStartingOffset = PartEntry->PartInfo[j].StartingOffset.QuadPart;
+ LastPartitionLength = PartEntry->PartInfo[j].PartitionLength.QuadPart;
+ }
+ }
- NewPartEntry = (PPARTENTRY)RtlAllocateHeap (ProcessHeap,
- 0,
- sizeof(PARTENTRY));
- if (NewPartEntry == NULL)
- return;
+ i += 4;
+ Entry = Entry->Flink;
+ }
+
+ /* Check for trailing unpartitioned disk space */
+ if (DiskEntry->DiskSize > (LastStartingOffset + LastPartitionLength))
+ {
+ /* Round-down to cylinder size */
+ LastUnusedPartitionLength =
+ (DiskEntry->DiskSize - (LastStartingOffset + LastPartitionLength))
+ & ~(DiskEntry->CylinderSize - 1);
+
+ if (LastUnusedPartitionLength >= DiskEntry->CylinderSize)
+ {
+ DPRINT("Unpartitioned disk space %I64u\n", LastUnusedPartitionLength);
+
+ NewPartEntry = (PPARTENTRY)RtlAllocateHeap(ProcessHeap,
+ 0,
+ sizeof(PARTENTRY));
+ if (NewPartEntry == NULL)
+ return;
- RtlZeroMemory (NewPartEntry,
- sizeof(PARTENTRY));
+ RtlZeroMemory(NewPartEntry,
+ sizeof(PARTENTRY));
- NewPartEntry->Unpartitioned = TRUE;
- NewPartEntry->UnpartitionedOffset = LastStartingOffset + LastPartitionLength;
- NewPartEntry->UnpartitionedLength = LastUnusedPartitionLength;
+ NewPartEntry->Unpartitioned = TRUE;
+ NewPartEntry->UnpartitionedOffset = LastStartingOffset + LastPartitionLength;
+ NewPartEntry->UnpartitionedLength = LastUnusedPartitionLength;
- /* Append the table to the list */
- InsertTailList (&DiskEntry->PartListHead,
- &NewPartEntry->ListEntry);
- }
+ /* Append the table to the list */
+ InsertTailList(&DiskEntry->PartListHead,
+ &NewPartEntry->ListEntry);
+ }
+ }
}
- }
}
+
NTSTATUS
NTAPI
-DiskIdentifierQueryRoutine(PWSTR ValueName,
- ULONG ValueType,
- PVOID ValueData,
- ULONG ValueLength,
- PVOID Context,
- PVOID EntryContext)
+DiskIdentifierQueryRoutine(
+ PWSTR ValueName,
+ ULONG ValueType,
+ PVOID ValueData,
+ ULONG ValueLength,
+ PVOID Context,
+ PVOID EntryContext)
{
- PBIOSDISKENTRY BiosDiskEntry = (PBIOSDISKENTRY)Context;
- UNICODE_STRING NameU;
-
- if (ValueType == REG_SZ &&
- ValueLength == 20 * sizeof(WCHAR))
- {
- NameU.Buffer = (PWCHAR)ValueData;
- NameU.Length = NameU.MaximumLength = 8 * sizeof(WCHAR);
- RtlUnicodeStringToInteger(&NameU, 16, &BiosDiskEntry->Checksum);
-
- NameU.Buffer = (PWCHAR)ValueData + 9;
- RtlUnicodeStringToInteger(&NameU, 16, &BiosDiskEntry->Signature);
-
- return STATUS_SUCCESS;
- }
- return STATUS_UNSUCCESSFUL;
+ PBIOSDISKENTRY BiosDiskEntry = (PBIOSDISKENTRY)Context;
+ UNICODE_STRING NameU;
+
+ if (ValueType == REG_SZ &&
+ ValueLength == 20 * sizeof(WCHAR))
+ {
+ NameU.Buffer = (PWCHAR)ValueData;
+ NameU.Length = NameU.MaximumLength = 8 * sizeof(WCHAR);
+ RtlUnicodeStringToInteger(&NameU, 16, &BiosDiskEntry->Checksum);
+
+ NameU.Buffer = (PWCHAR)ValueData + 9;
+ RtlUnicodeStringToInteger(&NameU, 16, &BiosDiskEntry->Signature);
+
+ return STATUS_SUCCESS;
+ }
+
+ return STATUS_UNSUCCESSFUL;
}
+
NTSTATUS
NTAPI
-DiskConfigurationDataQueryRoutine(PWSTR ValueName,
- ULONG ValueType,
- PVOID ValueData,
- ULONG ValueLength,
- PVOID Context,
- PVOID EntryContext)
+DiskConfigurationDataQueryRoutine(
+ PWSTR ValueName,
+ ULONG ValueType,
+ PVOID ValueData,
+ ULONG ValueLength,
+ PVOID Context,
+ PVOID EntryContext)
{
- PBIOSDISKENTRY BiosDiskEntry = (PBIOSDISKENTRY)Context;
- PCM_FULL_RESOURCE_DESCRIPTOR FullResourceDescriptor;
- PCM_DISK_GEOMETRY_DEVICE_DATA DiskGeometry;
- ULONG i;
+ PBIOSDISKENTRY BiosDiskEntry = (PBIOSDISKENTRY)Context;
+ PCM_FULL_RESOURCE_DESCRIPTOR FullResourceDescriptor;
+ PCM_DISK_GEOMETRY_DEVICE_DATA DiskGeometry;
+ ULONG i;
- if (ValueType != REG_FULL_RESOURCE_DESCRIPTOR ||
- ValueLength < sizeof(CM_FULL_RESOURCE_DESCRIPTOR))
- return STATUS_UNSUCCESSFUL;
+ if (ValueType != REG_FULL_RESOURCE_DESCRIPTOR ||
+ ValueLength < sizeof(CM_FULL_RESOURCE_DESCRIPTOR))
+ return STATUS_UNSUCCESSFUL;
+
+ FullResourceDescriptor = (PCM_FULL_RESOURCE_DESCRIPTOR)ValueData;
+
+ /* Hm. Version and Revision are not set on Microsoft Windows XP... */
+#if 0
+ if (FullResourceDescriptor->PartialResourceList.Version != 1 ||
+ FullResourceDescriptor->PartialResourceList.Revision != 1)
+ return STATUS_UNSUCCESSFUL;
+#endif
- FullResourceDescriptor = (PCM_FULL_RESOURCE_DESCRIPTOR)ValueData;
- /* Hm. Version and Revision are not set on Microsoft Windows XP... */
- /*if (FullResourceDescriptor->PartialResourceList.Version != 1 ||
- FullResourceDescriptor->PartialResourceList.Revision != 1)
- return STATUS_UNSUCCESSFUL;*/
+ for (i = 0; i < FullResourceDescriptor->PartialResourceList.Count; i++)
+ {
+ if (FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].Type != CmResourceTypeDeviceSpecific ||
+ FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].u.DeviceSpecificData.DataSize != sizeof(CM_DISK_GEOMETRY_DEVICE_DATA))
+ continue;
- for (i = 0; i < FullResourceDescriptor->PartialResourceList.Count; i++)
- {
- if (FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].Type != CmResourceTypeDeviceSpecific ||
- FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].u.DeviceSpecificData.DataSize != sizeof(CM_DISK_GEOMETRY_DEVICE_DATA))
- continue;
+ DiskGeometry = (PCM_DISK_GEOMETRY_DEVICE_DATA)&FullResourceDescriptor->PartialResourceList.PartialDescriptors[i + 1];
+ BiosDiskEntry->DiskGeometry = *DiskGeometry;
- DiskGeometry = (PCM_DISK_GEOMETRY_DEVICE_DATA)&FullResourceDescriptor->PartialResourceList.PartialDescriptors[i + 1];
- BiosDiskEntry->DiskGeometry = *DiskGeometry;
+ return STATUS_SUCCESS;
+ }
- return STATUS_SUCCESS;
- }
- return STATUS_UNSUCCESSFUL;
+ return STATUS_UNSUCCESSFUL;
}
+
NTSTATUS
NTAPI
-SystemConfigurationDataQueryRoutine(PWSTR ValueName,
- ULONG ValueType,
- PVOID ValueData,
- ULONG ValueLength,
- PVOID Context,
- PVOID EntryContext)
+SystemConfigurationDataQueryRoutine(
+ PWSTR ValueName,
+ ULONG ValueType,
+ PVOID ValueData,
+ ULONG ValueLength,
+ PVOID Context,
+ PVOID EntryContext)
{
- PCM_FULL_RESOURCE_DESCRIPTOR FullResourceDescriptor;
- PCM_INT13_DRIVE_PARAMETER* Int13Drives = (PCM_INT13_DRIVE_PARAMETER*)Context;
- ULONG i;
+ PCM_FULL_RESOURCE_DESCRIPTOR FullResourceDescriptor;
+ PCM_INT13_DRIVE_PARAMETER* Int13Drives = (PCM_INT13_DRIVE_PARAMETER*)Context;
+ ULONG i;
- if (ValueType != REG_FULL_RESOURCE_DESCRIPTOR ||
- ValueLength < sizeof (CM_FULL_RESOURCE_DESCRIPTOR))
- return STATUS_UNSUCCESSFUL;
+ if (ValueType != REG_FULL_RESOURCE_DESCRIPTOR ||
+ ValueLength < sizeof (CM_FULL_RESOURCE_DESCRIPTOR))
+ return STATUS_UNSUCCESSFUL;
+
+ FullResourceDescriptor = (PCM_FULL_RESOURCE_DESCRIPTOR)ValueData;
+
+ /* Hm. Version and Revision are not set on Microsoft Windows XP... */
+#if 0
+ if (FullResourceDescriptor->PartialResourceList.Version != 1 ||
+ FullResourceDescriptor->PartialResourceList.Revision != 1)
+ return STATUS_UNSUCCESSFUL;
+#endif
+
+ for (i = 0; i < FullResourceDescriptor->PartialResourceList.Count; i++)
+ {
+ if (FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].Type != CmResourceTypeDeviceSpecific ||
+ FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].u.DeviceSpecificData.DataSize % sizeof(CM_INT13_DRIVE_PARAMETER) != 0)
+ continue;
+
+ *Int13Drives = (CM_INT13_DRIVE_PARAMETER*) RtlAllocateHeap(ProcessHeap, 0, FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].u.DeviceSpecificData.DataSize);
+ if (*Int13Drives == NULL)
+ return STATUS_NO_MEMORY;
- FullResourceDescriptor = (PCM_FULL_RESOURCE_DESCRIPTOR)ValueData;
- /* Hm. Version and Revision are not set on Microsoft Windows XP... */
- /*if (FullResourceDescriptor->PartialResourceList.Version != 1 ||
- FullResourceDescriptor->PartialResourceList.Revision != 1)
- return STATUS_UNSUCCESSFUL;*/
-
- for (i = 0; i < FullResourceDescriptor->PartialResourceList.Count; i++)
- {
- if (FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].Type != CmResourceTypeDeviceSpecific ||
- FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].u.DeviceSpecificData.DataSize % sizeof(CM_INT13_DRIVE_PARAMETER) != 0)
- continue;
-
- *Int13Drives = (CM_INT13_DRIVE_PARAMETER*) RtlAllocateHeap(ProcessHeap, 0, FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].u.DeviceSpecificData.DataSize);
- if (*Int13Drives == NULL)
- return STATUS_NO_MEMORY;
- memcpy(*Int13Drives,
- &FullResourceDescriptor->PartialResourceList.PartialDescriptors[i + 1],
- FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].u.DeviceSpecificData.DataSize);
- return STATUS_SUCCESS;
- }
- return STATUS_UNSUCCESSFUL;
+ memcpy(*Int13Drives,
+ &FullResourceDescriptor->PartialResourceList.PartialDescriptors[i + 1],
+ FullResourceDescriptor->PartialResourceList.PartialDescriptors[i].u.DeviceSpecificData.DataSize);
+ return STATUS_SUCCESS;
+ }
+
+ return STATUS_UNSUCCESSFUL;
}
+
+
#define ROOT_NAME L"\\Registry\\Machine\\HARDWARE\\DESCRIPTION\\System\\MultifunctionAdapter"
static VOID
-EnumerateBiosDiskEntries(PPARTLIST PartList)
+EnumerateBiosDiskEntries(
+ PPARTLIST PartList)
{
- RTL_QUERY_REGISTRY_TABLE QueryTable[3];
- WCHAR Name[120];
- ULONG AdapterCount;
- ULONG DiskCount;
- NTSTATUS Status;
- PCM_INT13_DRIVE_PARAMETER Int13Drives;
- PBIOSDISKENTRY BiosDiskEntry;
-
- memset(QueryTable, 0, sizeof(QueryTable));
-
- QueryTable[1].Name = L"Configuration Data";
- QueryTable[1].QueryRoutine = SystemConfigurationDataQueryRoutine;
- Int13Drives = NULL;
- Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
- L"\\Registry\\Machine\\HARDWARE\\DESCRIPTION\\System",
- &QueryTable[1],
- (PVOID)&Int13Drives,
- NULL);
- if (!NT_SUCCESS(Status))
- {
- DPRINT1("Unable to query the 'Configuration Data' key in '\\Registry\\Machine\\HARDWARE\\DESCRIPTION\\System', status=%lx\n", Status);
- return;
- }
-
- AdapterCount = 0;
- while (1)
- {
- swprintf(Name, L"%s\\%lu", ROOT_NAME, AdapterCount);
+ RTL_QUERY_REGISTRY_TABLE QueryTable[3];
+ WCHAR Name[120];
+ ULONG AdapterCount;
+ ULONG DiskCount;
+ NTSTATUS Status;
+ PCM_INT13_DRIVE_PARAMETER Int13Drives;
+ PBIOSDISKENTRY BiosDiskEntry;
+
+ memset(QueryTable, 0, sizeof(QueryTable));
+
+ QueryTable[1].Name = L"Configuration Data";
+ QueryTable[1].QueryRoutine = SystemConfigurationDataQueryRoutine;
+ Int13Drives = NULL;
Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
- Name,
- &QueryTable[2],
- NULL,
+ L"\\Registry\\Machine\\HARDWARE\\DESCRIPTION\\System",
+ &QueryTable[1],
+ (PVOID)&Int13Drives,
NULL);
if (!NT_SUCCESS(Status))
{
- break;
+ DPRINT1("Unable to query the 'Configuration Data' key in '\\Registry\\Machine\\HARDWARE\\DESCRIPTION\\System', status=%lx\n", Status);
+ return;
}
- swprintf(Name, L"%s\\%lu\\DiskController", ROOT_NAME, AdapterCount);
- Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
- Name,
- &QueryTable[2],
- NULL,
- NULL);
- if (NT_SUCCESS(Status))
+ AdapterCount = 0;
+ while (1)
{
- while (1)
- {
- swprintf(Name, L"%s\\%lu\\DiskController\\0", ROOT_NAME, AdapterCount);
+ swprintf(Name, L"%s\\%lu", ROOT_NAME, AdapterCount);
Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
Name,
&QueryTable[2],
NULL);
if (!NT_SUCCESS(Status))
{
- RtlFreeHeap(ProcessHeap, 0, Int13Drives);
- return;
+ break;
}
- swprintf(Name, L"%s\\%lu\\DiskController\\0\\DiskPeripheral", ROOT_NAME, AdapterCount);
+ swprintf(Name, L"%s\\%lu\\DiskController", ROOT_NAME, AdapterCount);
Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
Name,
&QueryTable[2],
NULL);
if (NT_SUCCESS(Status))
{
- QueryTable[0].Name = L"Identifier";
- QueryTable[0].QueryRoutine = DiskIdentifierQueryRoutine;
- QueryTable[1].Name = L"Configuration Data";
- QueryTable[1].QueryRoutine = DiskConfigurationDataQueryRoutine;
- DiskCount = 0;
- while (1)
- {
- BiosDiskEntry = (BIOSDISKENTRY*) RtlAllocateHeap(ProcessHeap, HEAP_ZERO_MEMORY, sizeof(BIOSDISKENTRY));
- if (BiosDiskEntry == NULL)
- {
- break;
- }
- swprintf(Name, L"%s\\%lu\\DiskController\\0\\DiskPeripheral\\%lu", ROOT_NAME, AdapterCount, DiskCount);
- Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
- Name,
- QueryTable,
- (PVOID)BiosDiskEntry,
- NULL);
- if (!NT_SUCCESS(Status))
- {
- RtlFreeHeap(ProcessHeap, 0, BiosDiskEntry);
- break;
- }
- BiosDiskEntry->DiskNumber = DiskCount;
- BiosDiskEntry->Recognized = FALSE;
-
- if (DiskCount < Int13Drives[0].NumberDrives)
- {
- BiosDiskEntry->Int13DiskData = Int13Drives[DiskCount];
- }
- else
+ while (1)
{
- DPRINT1("Didn't find int13 drive datas for disk %u\n", DiskCount);
+ swprintf(Name, L"%s\\%lu\\DiskController\\0", ROOT_NAME, AdapterCount);
+ Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
+ Name,
+ &QueryTable[2],
+ NULL,
+ NULL);
+ if (!NT_SUCCESS(Status))
+ {
+ RtlFreeHeap(ProcessHeap, 0, Int13Drives);
+ return;
+ }
+
+ swprintf(Name, L"%s\\%lu\\DiskController\\0\\DiskPeripheral", ROOT_NAME, AdapterCount);
+ Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
+ Name,
+ &QueryTable[2],
+ NULL,
+ NULL);
+ if (NT_SUCCESS(Status))
+ {
+ QueryTable[0].Name = L"Identifier";
+ QueryTable[0].QueryRoutine = DiskIdentifierQueryRoutine;
+ QueryTable[1].Name = L"Configuration Data";
+ QueryTable[1].QueryRoutine = DiskConfigurationDataQueryRoutine;
+
+ DiskCount = 0;
+ while (1)
+ {
+ BiosDiskEntry = (BIOSDISKENTRY*) RtlAllocateHeap(ProcessHeap, HEAP_ZERO_MEMORY, sizeof(BIOSDISKENTRY));
+ if (BiosDiskEntry == NULL)
+ {
+ break;
+ }
+
+ swprintf(Name, L"%s\\%lu\\DiskController\\0\\DiskPeripheral\\%lu", ROOT_NAME, AdapterCount, DiskCount);
+ Status = RtlQueryRegistryValues(RTL_REGISTRY_ABSOLUTE,
+ Name,
+ QueryTable,
+ (PVOID)BiosDiskEntry,
+ NULL);
+ if (!NT_SUCCESS(Status))
+ {
+ RtlFreeHeap(ProcessHeap, 0, BiosDiskEntry);
+ break;
+ }
+
+ BiosDiskEntry->DiskNumber = DiskCount;
+ BiosDiskEntry->Recognized = FALSE;
+
+ if (DiskCount < Int13Drives[0].NumberDrives)
+ {
+ BiosDiskEntry->Int13DiskData = Int13Drives[DiskCount];
+ }
+ else
+ {
+ DPRINT1("Didn't find int13 drive datas for disk %u\n", DiskCount);
+ }
+
+ InsertTailList(&PartList->BiosDiskListHead, &BiosDiskEntry->ListEntry);
+
+ DPRINT("DiskNumber: %lu\n", BiosDiskEntry->DiskNumber);
+ DPRINT("Signature: %08lx\n", BiosDiskEntry->Signature);
+ DPRINT("Checksum: %08lx\n", BiosDiskEntry->Checksum);
+ DPRINT("BytesPerSector: %lu\n", BiosDiskEntry->DiskGeometry.BytesPerSector);
+ DPRINT("NumberOfCylinders: %lu\n", BiosDiskEntry->DiskGeometry.NumberOfCylinders);
+ DPRINT("NumberOfHeads: %lu\n", BiosDiskEntry->DiskGeometry.NumberOfHeads);
+ DPRINT("DriveSelect: %02x\n", BiosDiskEntry->Int13DiskData.DriveSelect);
+ DPRINT("MaxCylinders: %lu\n", BiosDiskEntry->Int13DiskData.MaxCylinders);
+ DPRINT("SectorsPerTrack: %d\n", BiosDiskEntry->Int13DiskData.SectorsPerTrack);
+ DPRINT("MaxHeads: %d\n", BiosDiskEntry->Int13DiskData.MaxHeads);
+ DPRINT("NumberDrives: %d\n", BiosDiskEntry->Int13DiskData.NumberDrives);
+
+ DiskCount++;
+ }
+ }
+
+ RtlFreeHeap(ProcessHeap, 0, Int13Drives);
+ return;
}
-
-
- InsertTailList(&PartList->BiosDiskListHead, &BiosDiskEntry->ListEntry);
-
- DPRINT("DiskNumber: %lu\n", BiosDiskEntry->DiskNumber);
- DPRINT("Signature: %08lx\n", BiosDiskEntry->Signature);
- DPRINT("Checksum: %08lx\n", BiosDiskEntry->Checksum);
- DPRINT("BytesPerSector: %lu\n", BiosDiskEntry->DiskGeometry.BytesPerSector);
- DPRINT("NumberOfCylinders: %lu\n", BiosDiskEntry->DiskGeometry.NumberOfCylinders);
- DPRINT("NumberOfHeads: %lu\n", BiosDiskEntry->DiskGeometry.NumberOfHeads);
- DPRINT("DriveSelect: %02x\n", BiosDiskEntry->Int13DiskData.DriveSelect);
- DPRINT("MaxCylinders: %lu\n", BiosDiskEntry->Int13DiskData.MaxCylinders);
- DPRINT("SectorsPerTrack: %d\n", BiosDiskEntry->Int13DiskData.SectorsPerTrack);
- DPRINT("MaxHeads: %d\n", BiosDiskEntry->Int13DiskData.MaxHeads);
- DPRINT("NumberDrives: %d\n", BiosDiskEntry->Int13DiskData.NumberDrives);
-
- DiskCount++;
- }
}
- RtlFreeHeap(ProcessHeap, 0, Int13Drives);
- return;
- }
+
+ AdapterCount++;
}
- AdapterCount++;
- }
- RtlFreeHeap(ProcessHeap, 0, Int13Drives);
-}
-static VOID
-AddDiskToList (HANDLE FileHandle,
- ULONG DiskNumber,
- PPARTLIST List)
-{
- DRIVE_LAYOUT_INFORMATION *LayoutBuffer;
- DISK_GEOMETRY DiskGeometry;
- SCSI_ADDRESS ScsiAddress;
- PDISKENTRY DiskEntry;
- IO_STATUS_BLOCK Iosb;
- NTSTATUS Status;
- PPARTITION_SECTOR Mbr;
- PULONG Buffer;
- LARGE_INTEGER FileOffset;
- WCHAR Identifier[20];
- ULONG Checksum;
- ULONG Signature;
- ULONG i;
- PLIST_ENTRY ListEntry;
- PBIOSDISKENTRY BiosDiskEntry;
- ULONG LayoutBufferSize;
-
- Status = NtDeviceIoControlFile (FileHandle,
- NULL,
- NULL,
- NULL,
- &Iosb,
- IOCTL_DISK_GET_DRIVE_GEOMETRY,
- NULL,
- 0,
- &DiskGeometry,
- sizeof(DISK_GEOMETRY));
- if (!NT_SUCCESS (Status))
- {
- return;
- }
-
- if (DiskGeometry.MediaType != FixedMedia &&
- DiskGeometry.MediaType != RemovableMedia)
- {
- return;
- }
-
- Status = NtDeviceIoControlFile (FileHandle,
- NULL,
- NULL,
- NULL,
- &Iosb,
- IOCTL_SCSI_GET_ADDRESS,
- NULL,
- 0,
- &ScsiAddress,
- sizeof(SCSI_ADDRESS));
- if (!NT_SUCCESS(Status))
- {
- return;
- }
-
- Mbr = (PARTITION_SECTOR*) RtlAllocateHeap(ProcessHeap,
- 0,
- DiskGeometry.BytesPerSector);
-
- if (Mbr == NULL)
- {
- return;
- }
-
- FileOffset.QuadPart = 0;
- Status = NtReadFile(FileHandle,
- NULL,
- NULL,
- NULL,
- &Iosb,
- (PVOID)Mbr,
- DiskGeometry.BytesPerSector,
- &FileOffset,
- NULL);
- if (!NT_SUCCESS(Status))
- {
- RtlFreeHeap(ProcessHeap,
- 0,
- Mbr);
- DPRINT1("NtReadFile failed, status=%x\n", Status);
- return;
- }
- Signature = Mbr->Signature;
-
- /* Calculate the MBR checksum */
- Checksum = 0;
- Buffer = (PULONG)Mbr;
- for (i = 0; i < 128; i++)
- {
- Checksum += Buffer[i];
- }
- Checksum = ~Checksum + 1;
-
- swprintf(Identifier, L"%08x-%08x-A", Checksum, Signature);
- DPRINT("Identifier: %S\n", Identifier);
-
- DiskEntry = (PDISKENTRY)RtlAllocateHeap (ProcessHeap,
- 0,
- sizeof(DISKENTRY));
- if (DiskEntry == NULL)
- {
- return;
- }
-
- DiskEntry->Checksum = Checksum;
- DiskEntry->Signature = Signature;
- if (Signature == 0)
- {
- /* If we have no signature, set the disk to dirty. WritePartitionsToDisk creates a new signature */
- DiskEntry->Modified = TRUE;
- }
- DiskEntry->BiosFound = FALSE;
-
- /* Check if this disk has a valid MBR */
- if (Mbr->BootCode[0] == 0 && Mbr->BootCode[1] == 0)
- DiskEntry->NoMbr = TRUE;
- else
- DiskEntry->NoMbr = FALSE;
-
- /* Free Mbr sector buffer */
- RtlFreeHeap (ProcessHeap,
- 0,
- Mbr);
-
- ListEntry = List->BiosDiskListHead.Flink;
- while(ListEntry != &List->BiosDiskListHead)
- {
- BiosDiskEntry = CONTAINING_RECORD(ListEntry, BIOSDISKENTRY, ListEntry);
- /* FIXME:
- * Compare the size from bios and the reported size from driver.
- * If we have more than one disk with a zero or with the same signatur
- * we must create new signatures and reboot. After the reboot,
- * it is possible to identify the disks.
- */
- if (BiosDiskEntry->Signature == Signature &&
- BiosDiskEntry->Checksum == Checksum &&
- !BiosDiskEntry->Recognized)
- {
- if (!DiskEntry->BiosFound)
- {
- DiskEntry->BiosDiskNumber = BiosDiskEntry->DiskNumber;
- DiskEntry->BiosFound = TRUE;
- BiosDiskEntry->Recognized = TRUE;
- }
- else
- {
- }
- }
- ListEntry = ListEntry->Flink;
- }
-
- if (!DiskEntry->BiosFound)
- {
-#if 0
- RtlFreeHeap(ProcessHeap, 0, DiskEntry);
- return;
-#else
- DPRINT1("WARNING: Setup could not find a matching BIOS disk entry. Disk %d is not be bootable by the BIOS!\n", DiskNumber);
-#endif
- }
-
- InitializeListHead (&DiskEntry->PartListHead);
-
- DiskEntry->Cylinders = DiskGeometry.Cylinders.QuadPart;
- DiskEntry->TracksPerCylinder = DiskGeometry.TracksPerCylinder;
- DiskEntry->SectorsPerTrack = DiskGeometry.SectorsPerTrack;
- DiskEntry->BytesPerSector = DiskGeometry.BytesPerSector;
-
- DPRINT ("Cylinders %I64u\n", DiskEntry->Cylinders);
- DPRINT ("TracksPerCylinder %I64u\n", DiskEntry->TracksPerCylinder);
- DPRINT ("SectorsPerTrack %I64u\n", DiskEntry->SectorsPerTrack);
- DPRINT ("BytesPerSector %I64u\n", DiskEntry->BytesPerSector);
-
- DiskEntry->TrackSize =
- (ULONGLONG)DiskGeometry.SectorsPerTrack *
- (ULONGLONG)DiskGeometry.BytesPerSector;
- DiskEntry->CylinderSize =
- (ULONGLONG)DiskGeometry.TracksPerCylinder *
- DiskEntry->TrackSize;
- DiskEntry->DiskSize =
- DiskGeometry.Cylinders.QuadPart *
- DiskEntry->CylinderSize;
-
- DiskEntry->DiskNumber = DiskNumber;
- DiskEntry->Port = ScsiAddress.PortNumber;
- DiskEntry->Bus = ScsiAddress.PathId;
- DiskEntry->Id = ScsiAddress.TargetId;
-
- GetDriverName (DiskEntry);
-
- InsertAscendingList(&List->DiskListHead, DiskEntry, DISKENTRY, ListEntry, DiskNumber);
-
- /*
- * Allocate a buffer for 26 logical drives (2 entries each == 52)
- * plus the main partiton table (4 entries). Total 56 entries.
- */
- LayoutBufferSize = sizeof(DRIVE_LAYOUT_INFORMATION) +
- ((56 - ANYSIZE_ARRAY) * sizeof(PARTITION_INFORMATION));
- LayoutBuffer = (DRIVE_LAYOUT_INFORMATION*)RtlAllocateHeap (ProcessHeap,
- 0,
- LayoutBufferSize);
- if (LayoutBuffer == NULL)
- {
- return;
- }
-
- Status = NtDeviceIoControlFile (FileHandle,
- NULL,
- NULL,
- NULL,
- &Iosb,
- IOCTL_DISK_GET_DRIVE_LAYOUT,
- NULL,
- 0,
- LayoutBuffer,
- LayoutBufferSize);
- if (NT_SUCCESS (Status))
- {
- if (LayoutBuffer->PartitionCount == 0)
- {
- DiskEntry->NewDisk = TRUE;
- }
-
- AddPartitionToList (DiskNumber,
- DiskEntry,
- LayoutBuffer);
-
- ScanForUnpartitionedDiskSpace (DiskEntry);
- }
-
- RtlFreeHeap (ProcessHeap,
- 0,
- LayoutBuffer);
+ RtlFreeHeap(ProcessHeap, 0, Int13Drives);
}
-PPARTLIST
-CreatePartitionList (SHORT Left,
- SHORT Top,
- SHORT Right,
- SHORT Bottom)
+static
+VOID
+AddDiskToList(
+ HANDLE FileHandle,
+ ULONG DiskNumber,
+ PPARTLIST List)
{
- PPARTLIST List;
- OBJECT_ATTRIBUTES ObjectAttributes;
- SYSTEM_DEVICE_INFORMATION Sdi;
- IO_STATUS_BLOCK Iosb;
- ULONG ReturnSize;
- NTSTATUS Status;
- ULONG DiskNumber;
- WCHAR Buffer[MAX_PATH];
- UNICODE_STRING Name;
- HANDLE FileHandle;
-
- List = (PPARTLIST)RtlAllocateHeap (ProcessHeap,
- 0,
- sizeof (PARTLIST));
- if (List == NULL)
- return NULL;
-
- List->Left = Left;
- List->Top = Top;
- List->Right = Right;
- List->Bottom = Bottom;
-
- List->Line = 0;
- List->Offset = 0;
-
- List->TopDisk = (ULONG)-1;
- List->TopPartition = (ULONG)-1;
-
- List->CurrentDisk = NULL;
- List->CurrentPartition = NULL;
- List->CurrentPartitionNumber = 0;
-
- InitializeListHead (&List->DiskListHead);
- InitializeListHead (&List->BiosDiskListHead);
-
- EnumerateBiosDiskEntries(List);
-
- Status = NtQuerySystemInformation (SystemDeviceInformation,
- &Sdi,
- sizeof(SYSTEM_DEVICE_INFORMATION),
- &ReturnSize);
- if (!NT_SUCCESS (Status))
- {
- RtlFreeHeap (ProcessHeap, 0, List);
- return NULL;
- }
-
- for (DiskNumber = 0; DiskNumber < Sdi.NumberOfDisks; DiskNumber++)
- {
- swprintf (Buffer,
- L"\\Device\\Harddisk%d\\Partition0",
- DiskNumber);
- RtlInitUnicodeString (&Name,
- Buffer);
-
- InitializeObjectAttributes (&ObjectAttributes,
- &Name,
- 0,
- NULL,
- NULL);
-
- Status = NtOpenFile (&FileHandle,
- FILE_READ_DATA | FILE_READ_ATTRIBUTES | SYNCHRONIZE,
- &ObjectAttributes,
- &Iosb,
- FILE_SHARE_READ,
- FILE_SYNCHRONOUS_IO_NONALERT);
- if (NT_SUCCESS(Status))
+ DRIVE_LAYOUT_INFORMATION *LayoutBuffer;
+ DISK_GEOMETRY DiskGeometry;
+ SCSI_ADDRESS ScsiAddress;
+ PDISKENTRY DiskEntry;
+ IO_STATUS_BLOCK Iosb;
+ NTSTATUS Status;
+ PPARTITION_SECTOR Mbr;
+ PULONG Buffer;
+ LARGE_INTEGER FileOffset;
+ WCHAR Identifier[20];
+ ULONG Checksum;
+ ULONG Signature;
+ ULONG i;
+ PLIST_ENTRY ListEntry;
+ PBIOSDISKENTRY BiosDiskEntry;
+ ULONG LayoutBufferSize;
+
+ Status = NtDeviceIoControlFile(FileHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ IOCTL_DISK_GET_DRIVE_GEOMETRY,
+ NULL,
+ 0,
+ &DiskGeometry,
+ sizeof(DISK_GEOMETRY));
+ if (!NT_SUCCESS(Status))
{
- AddDiskToList (FileHandle,
- DiskNumber,
- List);
-
- NtClose(FileHandle);
+ return;
}
- }
-
- AssignDriverLetters (List);
-
- List->TopDisk = 0;
- List->TopPartition = 0;
-
- /* Search for first usable disk and partition */
- if (IsListEmpty (&List->DiskListHead))
- {
- List->CurrentDisk = NULL;
- List->CurrentPartition = NULL;
- List->CurrentPartitionNumber = 0;
- }
- else
- {
- List->CurrentDisk =
- CONTAINING_RECORD (List->DiskListHead.Flink,
- DISKENTRY,
- ListEntry);
- if (IsListEmpty (&List->CurrentDisk->PartListHead))
+ if (DiskGeometry.MediaType != FixedMedia &&
+ DiskGeometry.MediaType != RemovableMedia)
{
- List->CurrentPartition = 0;
- List->CurrentPartitionNumber = 0;
+ return;
}
- else
+
+ Status = NtDeviceIoControlFile(FileHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ IOCTL_SCSI_GET_ADDRESS,
+ NULL,
+ 0,
+ &ScsiAddress,
+ sizeof(SCSI_ADDRESS));
+ if (!NT_SUCCESS(Status))
{
- List->CurrentPartition =
- CONTAINING_RECORD (List->CurrentDisk->PartListHead.Flink,
- PARTENTRY,
- ListEntry);
- List->CurrentPartitionNumber = 0;
+ return;
}
- }
-
- return List;
-}
+ Mbr = (PARTITION_SECTOR*)RtlAllocateHeap(ProcessHeap,
+ 0,
+ DiskGeometry.BytesPerSector);
+ if (Mbr == NULL)
+ {
+ return;
+ }
-VOID
-DestroyPartitionList (PPARTLIST List)
-{
- PDISKENTRY DiskEntry;
- PBIOSDISKENTRY BiosDiskEntry;
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry;
+ FileOffset.QuadPart = 0;
+ Status = NtReadFile(FileHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ (PVOID)Mbr,
+ DiskGeometry.BytesPerSector,
+ &FileOffset,
+ NULL);
+ if (!NT_SUCCESS(Status))
+ {
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ Mbr);
+ DPRINT1("NtReadFile failed, status=%x\n", Status);
+ return;
+ }
+ Signature = Mbr->Signature;
- /* Release disk and partition info */
- while (!IsListEmpty (&List->DiskListHead))
- {
- Entry = RemoveHeadList (&List->DiskListHead);
- DiskEntry = CONTAINING_RECORD (Entry, DISKENTRY, ListEntry);
+ /* Calculate the MBR checksum */
+ Checksum = 0;
+ Buffer = (PULONG)Mbr;
+ for (i = 0; i < 128; i++)
+ {
+ Checksum += Buffer[i];
+ }
+ Checksum = ~Checksum + 1;
- /* Release driver name */
- RtlFreeUnicodeString(&DiskEntry->DriverName);
+ swprintf(Identifier, L"%08x-%08x-A", Checksum, Signature);
+ DPRINT("Identifier: %S\n", Identifier);
- /* Release partition array */
- while (!IsListEmpty (&DiskEntry->PartListHead))
+ DiskEntry = (PDISKENTRY)RtlAllocateHeap(ProcessHeap,
+ 0,
+ sizeof(DISKENTRY));
+ if (DiskEntry == NULL)
{
- Entry = RemoveHeadList (&DiskEntry->PartListHead);
- PartEntry = CONTAINING_RECORD (Entry, PARTENTRY, ListEntry);
+ return;
+ }
- RtlFreeHeap (ProcessHeap,
- 0,
- PartEntry);
+ DiskEntry->Checksum = Checksum;
+ DiskEntry->Signature = Signature;
+ if (Signature == 0)
+ {
+ /* If we have no signature, set the disk to dirty. WritePartitionsToDisk creates a new signature */
+ DiskEntry->Modified = TRUE;
}
+ DiskEntry->BiosFound = FALSE;
- /* Release disk entry */
- RtlFreeHeap (ProcessHeap, 0, DiskEntry);
- }
+ /* Check if this disk has a valid MBR */
+ if (Mbr->BootCode[0] == 0 && Mbr->BootCode[1] == 0)
+ DiskEntry->NoMbr = TRUE;
+ else
+ DiskEntry->NoMbr = FALSE;
- /* release the bios disk info */
- while(!IsListEmpty(&List->BiosDiskListHead))
- {
- Entry = RemoveHeadList(&List->BiosDiskListHead);
- BiosDiskEntry = CONTAINING_RECORD(Entry, BIOSDISKENTRY, ListEntry);
+ /* Free Mbr sector buffer */
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ Mbr);
- RtlFreeHeap(ProcessHeap, 0, BiosDiskEntry);
- }
+ ListEntry = List->BiosDiskListHead.Flink;
+ while(ListEntry != &List->BiosDiskListHead)
+ {
+ BiosDiskEntry = CONTAINING_RECORD(ListEntry, BIOSDISKENTRY, ListEntry);
+ /* FIXME:
+ * Compare the size from bios and the reported size from driver.
+ * If we have more than one disk with a zero or with the same signatur
+ * we must create new signatures and reboot. After the reboot,
+ * it is possible to identify the disks.
+ */
+ if (BiosDiskEntry->Signature == Signature &&
+ BiosDiskEntry->Checksum == Checksum &&
+ !BiosDiskEntry->Recognized)
+ {
+ if (!DiskEntry->BiosFound)
+ {
+ DiskEntry->BiosDiskNumber = BiosDiskEntry->DiskNumber;
+ DiskEntry->BiosFound = TRUE;
+ BiosDiskEntry->Recognized = TRUE;
+ }
+ else
+ {
+ }
+ }
+ ListEntry = ListEntry->Flink;
+ }
+
+ if (!DiskEntry->BiosFound)
+ {
+#if 0
+ RtlFreeHeap(ProcessHeap, 0, DiskEntry);
+ return;
+#else
+ DPRINT1("WARNING: Setup could not find a matching BIOS disk entry. Disk %d is not be bootable by the BIOS!\n", DiskNumber);
+#endif
+ }
+
+ InitializeListHead(&DiskEntry->PartListHead);
+
+ DiskEntry->Cylinders = DiskGeometry.Cylinders.QuadPart;
+ DiskEntry->TracksPerCylinder = DiskGeometry.TracksPerCylinder;
+ DiskEntry->SectorsPerTrack = DiskGeometry.SectorsPerTrack;
+ DiskEntry->BytesPerSector = DiskGeometry.BytesPerSector;
+
+ DPRINT ("Cylinders %I64u\n", DiskEntry->Cylinders);
+ DPRINT ("TracksPerCylinder %I64u\n", DiskEntry->TracksPerCylinder);
+ DPRINT ("SectorsPerTrack %I64u\n", DiskEntry->SectorsPerTrack);
+ DPRINT ("BytesPerSector %I64u\n", DiskEntry->BytesPerSector);
+
+ DiskEntry->TrackSize =
+ (ULONGLONG)DiskGeometry.SectorsPerTrack *
+ (ULONGLONG)DiskGeometry.BytesPerSector;
+ DiskEntry->CylinderSize =
+ (ULONGLONG)DiskGeometry.TracksPerCylinder *
+ DiskEntry->TrackSize;
+ DiskEntry->DiskSize =
+ DiskGeometry.Cylinders.QuadPart *
+ DiskEntry->CylinderSize;
+
+ DiskEntry->DiskNumber = DiskNumber;
+ DiskEntry->Port = ScsiAddress.PortNumber;
+ DiskEntry->Bus = ScsiAddress.PathId;
+ DiskEntry->Id = ScsiAddress.TargetId;
+
+ GetDriverName(DiskEntry);
- /* Release list head */
- RtlFreeHeap (ProcessHeap, 0, List);
+ InsertAscendingList(&List->DiskListHead, DiskEntry, DISKENTRY, ListEntry, DiskNumber);
+
+ /*
+ * Allocate a buffer for 26 logical drives (2 entries each == 52)
+ * plus the main partiton table (4 entries). Total 56 entries.
+ */
+ LayoutBufferSize = sizeof(DRIVE_LAYOUT_INFORMATION) +
+ ((56 - ANYSIZE_ARRAY) * sizeof(PARTITION_INFORMATION));
+ LayoutBuffer = (DRIVE_LAYOUT_INFORMATION*)RtlAllocateHeap(ProcessHeap,
+ 0,
+ LayoutBufferSize);
+ if (LayoutBuffer == NULL)
+ {
+ return;
+ }
+
+ Status = NtDeviceIoControlFile(FileHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ IOCTL_DISK_GET_DRIVE_LAYOUT,
+ NULL,
+ 0,
+ LayoutBuffer,
+ LayoutBufferSize);
+ if (NT_SUCCESS(Status))
+ {
+ if (LayoutBuffer->PartitionCount == 0)
+ {
+ DiskEntry->NewDisk = TRUE;
+ }
+
+ AddPartitionToList(DiskNumber,
+ DiskEntry,
+ LayoutBuffer);
+
+ ScanForUnpartitionedDiskSpace(DiskEntry);
+ }
+
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ LayoutBuffer);
}
-static VOID
-PrintEmptyLine (PPARTLIST List)
+PPARTLIST
+CreatePartitionList(
+ SHORT Left,
+ SHORT Top,
+ SHORT Right,
+ SHORT Bottom)
{
- COORD coPos;
- DWORD Written;
- USHORT Width;
- USHORT Height;
+ PPARTLIST List;
+ OBJECT_ATTRIBUTES ObjectAttributes;
+ SYSTEM_DEVICE_INFORMATION Sdi;
+ IO_STATUS_BLOCK Iosb;
+ ULONG ReturnSize;
+ NTSTATUS Status;
+ ULONG DiskNumber;
+ WCHAR Buffer[MAX_PATH];
+ UNICODE_STRING Name;
+ HANDLE FileHandle;
+
+ List = (PPARTLIST)RtlAllocateHeap(ProcessHeap,
+ 0,
+ sizeof (PARTLIST));
+ if (List == NULL)
+ return NULL;
+
+ List->Left = Left;
+ List->Top = Top;
+ List->Right = Right;
+ List->Bottom = Bottom;
+
+ List->Line = 0;
+ List->Offset = 0;
+
+ List->TopDisk = (ULONG)-1;
+ List->TopPartition = (ULONG)-1;
- Width = List->Right - List->Left - 1;
- Height = List->Bottom - List->Top - 2;
+ List->CurrentDisk = NULL;
+ List->CurrentPartition = NULL;
+ List->CurrentPartitionNumber = 0;
+ InitializeListHead(&List->DiskListHead);
+ InitializeListHead(&List->BiosDiskListHead);
- coPos.X = List->Left + 1;
- coPos.Y = List->Top + 1 + List->Line;
+ EnumerateBiosDiskEntries(List);
- if (List->Line >= 0 && List->Line <= Height)
- {
- FillConsoleOutputAttribute (StdOutput,
- FOREGROUND_WHITE | BACKGROUND_BLUE,
- Width,
- coPos,
- &Written);
+ Status = NtQuerySystemInformation(SystemDeviceInformation,
+ &Sdi,
+ sizeof(SYSTEM_DEVICE_INFORMATION),
+ &ReturnSize);
+ if (!NT_SUCCESS(Status))
+ {
+ RtlFreeHeap(ProcessHeap, 0, List);
+ return NULL;
+ }
+
+ for (DiskNumber = 0; DiskNumber < Sdi.NumberOfDisks; DiskNumber++)
+ {
+ swprintf(Buffer,
+ L"\\Device\\Harddisk%d\\Partition0",
+ DiskNumber);
+ RtlInitUnicodeString(&Name,
+ Buffer);
+
+ InitializeObjectAttributes(&ObjectAttributes,
+ &Name,
+ 0,
+ NULL,
+ NULL);
+
+ Status = NtOpenFile(&FileHandle,
+ FILE_READ_DATA | FILE_READ_ATTRIBUTES | SYNCHRONIZE,
+ &ObjectAttributes,
+ &Iosb,
+ FILE_SHARE_READ,
+ FILE_SYNCHRONOUS_IO_NONALERT);
+ if (NT_SUCCESS(Status))
+ {
+ AddDiskToList(FileHandle,
+ DiskNumber,
+ List);
+
+ NtClose(FileHandle);
+ }
+ }
+
+ AssignDriverLetters(List);
+
+ List->TopDisk = 0;
+ List->TopPartition = 0;
- FillConsoleOutputCharacterA (StdOutput,
- ' ',
- Width,
- coPos,
- &Written);
- }
- List->Line++;
+ /* Search for first usable disk and partition */
+ if (IsListEmpty(&List->DiskListHead))
+ {
+ List->CurrentDisk = NULL;
+ List->CurrentPartition = NULL;
+ List->CurrentPartitionNumber = 0;
+ }
+ else
+ {
+ List->CurrentDisk = CONTAINING_RECORD(List->DiskListHead.Flink,
+ DISKENTRY,
+ ListEntry);
+
+ if (IsListEmpty(&List->CurrentDisk->PartListHead))
+ {
+ List->CurrentPartition = 0;
+ List->CurrentPartitionNumber = 0;
+ }
+ else
+ {
+ List->CurrentPartition = CONTAINING_RECORD(List->CurrentDisk->PartListHead.Flink,
+ PARTENTRY,
+ ListEntry);
+ List->CurrentPartitionNumber = 0;
+ }
+ }
+
+ return List;
}
-static VOID
-PrintPartitionData (PPARTLIST List,
- PDISKENTRY DiskEntry,
- PPARTENTRY PartEntry,
- ULONG PartNumber)
+VOID
+DestroyPartitionList(
+ PPARTLIST List)
{
- CHAR LineBuffer[128];
- COORD coPos;
- DWORD Written;
- USHORT Width;
- USHORT Height;
+ PDISKENTRY DiskEntry;
+ PBIOSDISKENTRY BiosDiskEntry;
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry;
+
+ /* Release disk and partition info */
+ while (!IsListEmpty(&List->DiskListHead))
+ {
+ Entry = RemoveHeadList(&List->DiskListHead);
+ DiskEntry = CONTAINING_RECORD(Entry, DISKENTRY, ListEntry);
- LARGE_INTEGER PartSize;
- PCHAR Unit;
- UCHAR Attribute;
- PCHAR PartType;
+ /* Release driver name */
+ RtlFreeUnicodeString(&DiskEntry->DriverName);
- Width = List->Right - List->Left - 1;
- Height = List->Bottom - List->Top - 2;
+ /* Release partition array */
+ while (!IsListEmpty(&DiskEntry->PartListHead))
+ {
+ Entry = RemoveHeadList(&DiskEntry->PartListHead);
+ PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ PartEntry);
+ }
- coPos.X = List->Left + 1;
- coPos.Y = List->Top + 1 + List->Line;
+ /* Release disk entry */
+ RtlFreeHeap(ProcessHeap, 0, DiskEntry);
+ }
- if (PartEntry->Unpartitioned == TRUE)
- {
-#if 0
- if (PartEntry->UnpartitionledLength >= 0x280000000ULL) /* 10 GB */
+ /* release the bios disk info */
+ while(!IsListEmpty(&List->BiosDiskListHead))
{
- PartSize.QuadPart = (PartEntry->UnpartitionedLength + (1 << 29)) >> 30;
- Unit = MUIGetString(STRING_GB);
+ Entry = RemoveHeadList(&List->BiosDiskListHead);
+ BiosDiskEntry = CONTAINING_RECORD(Entry, BIOSDISKENTRY, ListEntry);
+
+ RtlFreeHeap(ProcessHeap, 0, BiosDiskEntry);
+ }
+
+ /* Release list head */
+ RtlFreeHeap(ProcessHeap, 0, List);
+}
+
+
+static
+VOID
+PrintEmptyLine(
+ PPARTLIST List)
+{
+ COORD coPos;
+ DWORD Written;
+ USHORT Width;
+ USHORT Height;
+
+ Width = List->Right - List->Left - 1;
+ Height = List->Bottom - List->Top - 2;
+
+ coPos.X = List->Left + 1;
+ coPos.Y = List->Top + 1 + List->Line;
+
+ if (List->Line >= 0 && List->Line <= Height)
+ {
+ FillConsoleOutputAttribute(StdOutput,
+ FOREGROUND_WHITE | BACKGROUND_BLUE,
+ Width,
+ coPos,
+ &Written);
+
+ FillConsoleOutputCharacterA(StdOutput,
+ ' ',
+ Width,
+ coPos,
+ &Written);
+ }
+
+ List->Line++;
+}
+
+
+static
+VOID
+PrintPartitionData(
+ PPARTLIST List,
+ PDISKENTRY DiskEntry,
+ PPARTENTRY PartEntry,
+ ULONG PartNumber)
+{
+ CHAR LineBuffer[128];
+ COORD coPos;
+ DWORD Written;
+ USHORT Width;
+ USHORT Height;
+ LARGE_INTEGER PartSize;
+ PCHAR Unit;
+ UCHAR Attribute;
+ PCHAR PartType;
+
+ Width = List->Right - List->Left - 1;
+ Height = List->Bottom - List->Top - 2;
+
+ coPos.X = List->Left + 1;
+ coPos.Y = List->Top + 1 + List->Line;
+
+ if (PartEntry->Unpartitioned == TRUE)
+ {
+#if 0
+ if (PartEntry->UnpartitionledLength >= 0x280000000ULL) /* 10 GB */
+ {
+ PartSize.QuadPart = (PartEntry->UnpartitionedLength + (1 << 29)) >> 30;
+ Unit = MUIGetString(STRING_GB);
+ }
+ else
+#endif
+ if (PartEntry->UnpartitionedLength >= 0xA00000ULL) /* 10 MB */
+ {
+ PartSize.QuadPart = (PartEntry->UnpartitionedLength + (1 << 19)) >> 20;
+ Unit = MUIGetString(STRING_MB);
+ }
+ else
+ {
+ PartSize.QuadPart = (PartEntry->UnpartitionedLength + (1 << 9)) >> 10;
+ Unit = MUIGetString(STRING_KB);
+ }
+
+ sprintf(LineBuffer,
+ MUIGetString(STRING_UNPSPACE),
+ PartSize.u.LowPart,
+ Unit);
}
else
+ {
+ /* Determine partition type */
+ PartType = NULL;
+ if (PartEntry->New == TRUE)
+ {
+ PartType = MUIGetString(STRING_UNFORMATTED);
+ }
+ else if (PartEntry->Unpartitioned == FALSE)
+ {
+ if ((PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_FAT_12) ||
+ (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_FAT_16) ||
+ (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_HUGE) ||
+ (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_XINT13))
+ {
+ PartType = "FAT";
+ }
+ else if ((PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_FAT32) ||
+ (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_FAT32_XINT13))
+ {
+ PartType = "FAT32";
+ }
+ else if (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_EXT2)
+ {
+ PartType = "EXT2";
+ }
+ else if (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_IFS)
+ {
+ PartType = "NTFS"; /* FIXME: Not quite correct! */
+ }
+ }
+
+#if 0
+ if (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart >= 0x280000000LL) /* 10 GB */
+ {
+ PartSize.QuadPart = (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart + (1 << 29)) >> 30;
+ Unit = MUIGetString(STRING_GB);
+ }
+ else
#endif
- if (PartEntry->UnpartitionedLength >= 0xA00000ULL) /* 10 MB */
- {
- PartSize.QuadPart = (PartEntry->UnpartitionedLength + (1 << 19)) >> 20;
- Unit = MUIGetString(STRING_MB);
- }
- else
- {
- PartSize.QuadPart = (PartEntry->UnpartitionedLength + (1 << 9)) >> 10;
- Unit = MUIGetString(STRING_KB);
- }
-
- sprintf (LineBuffer,
- MUIGetString(STRING_UNPSPACE),
- PartSize.u.LowPart,
- Unit);
- }
- else
- {
- /* Determine partition type */
- PartType = NULL;
- if (PartEntry->New == TRUE)
- {
- PartType = MUIGetString(STRING_UNFORMATTED);
- }
- else if (PartEntry->Unpartitioned == FALSE)
- {
- if ((PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_FAT_12) ||
- (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_FAT_16) ||
- (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_HUGE) ||
- (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_XINT13))
- {
- PartType = "FAT";
- }
- else if ((PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_FAT32) ||
- (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_FAT32_XINT13))
- {
- PartType = "FAT32";
- }
- else if (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_EXT2)
- {
- PartType = "EXT2";
- }
- else if (PartEntry->PartInfo[PartNumber].PartitionType == PARTITION_IFS)
- {
- PartType = "NTFS"; /* FIXME: Not quite correct! */
- }
+ if (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart >= 0xA00000LL) /* 10 MB */
+ {
+ PartSize.QuadPart = (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart + (1 << 19)) >> 20;
+ Unit = MUIGetString(STRING_MB);
+ }
+ else
+ {
+ PartSize.QuadPart = (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart + (1 << 9)) >> 10;
+ Unit = MUIGetString(STRING_KB);
+ }
+
+ if (PartType == NULL)
+ {
+ sprintf(LineBuffer,
+ MUIGetString(STRING_HDDINFOUNK5),
+ (PartEntry->DriveLetter[PartNumber] == 0) ? '-' : PartEntry->DriveLetter[PartNumber],
+ (PartEntry->DriveLetter[PartNumber] == 0) ? '-' : ':',
+ PartEntry->PartInfo[PartNumber].PartitionType,
+ PartSize.u.LowPart,
+ Unit);
+ }
+ else
+ {
+ sprintf(LineBuffer,
+ "%c%c %-24s %6lu %s",
+ (PartEntry->DriveLetter[PartNumber] == 0) ? '-' : PartEntry->DriveLetter[PartNumber],
+ (PartEntry->DriveLetter[PartNumber] == 0) ? '-' : ':',
+ PartType,
+ PartSize.u.LowPart,
+ Unit);
+ }
}
+ Attribute = (List->CurrentDisk == DiskEntry &&
+ List->CurrentPartition == PartEntry &&
+ List->CurrentPartitionNumber == PartNumber) ?
+ FOREGROUND_BLUE | BACKGROUND_WHITE :
+ FOREGROUND_WHITE | BACKGROUND_BLUE;
+
+ if (List->Line >= 0 && List->Line <= Height)
+ {
+ FillConsoleOutputCharacterA(StdOutput,
+ ' ',
+ Width,
+ coPos,
+ &Written);
+ }
+ coPos.X += 4;
+ Width -= 8;
+ if (List->Line >= 0 && List->Line <= Height)
+ {
+ FillConsoleOutputAttribute(StdOutput,
+ Attribute,
+ Width,
+ coPos,
+ &Written);
+ }
+ coPos.X++;
+ Width -= 2;
+ if (List->Line >= 0 && List->Line <= Height)
+ {
+ WriteConsoleOutputCharacterA(StdOutput,
+ LineBuffer,
+ min(strlen(LineBuffer), Width),
+ coPos,
+ &Written);
+ }
+
+ List->Line++;
+}
+
+
+static
+VOID
+PrintDiskData(
+ PPARTLIST List,
+ PDISKENTRY DiskEntry)
+{
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry;
+ CHAR LineBuffer[128];
+ COORD coPos;
+ DWORD Written;
+ USHORT Width;
+ USHORT Height;
+ ULARGE_INTEGER DiskSize;
+ PCHAR Unit;
+ ULONG i;
+
+ Width = List->Right - List->Left - 1;
+ Height = List->Bottom - List->Top - 2;
+
+ coPos.X = List->Left + 1;
+ coPos.Y = List->Top + 1 + List->Line;
+
#if 0
- if (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart >= 0x280000000LL) /* 10 GB */
+ if (DiskEntry->DiskSize >= 0x280000000ULL) /* 10 GB */
{
- PartSize.QuadPart = (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart + (1 << 29)) >> 30;
- Unit = MUIGetString(STRING_GB);
+ DiskSize.QuadPart = (DiskEntry->DiskSize + (1 << 29)) >> 30;
+ Unit = MUIGetString(STRING_GB);
}
else
#endif
- if (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart >= 0xA00000LL) /* 10 MB */
{
- PartSize.QuadPart = (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart + (1 << 19)) >> 20;
- Unit = MUIGetString(STRING_MB);
+ DiskSize.QuadPart = (DiskEntry->DiskSize + (1 << 19)) >> 20;
+ if (DiskSize.QuadPart == 0)
+ DiskSize.QuadPart = 1;
+ Unit = MUIGetString(STRING_MB);
+ }
+
+ if (DiskEntry->DriverName.Length > 0)
+ {
+ sprintf(LineBuffer,
+ MUIGetString(STRING_HDINFOPARTSELECT),
+ DiskSize.u.LowPart,
+ Unit,
+ DiskEntry->DiskNumber,
+ DiskEntry->Port,
+ DiskEntry->Bus,
+ DiskEntry->Id,
+ DiskEntry->DriverName.Buffer);
}
else
{
- PartSize.QuadPart = (PartEntry->PartInfo[PartNumber].PartitionLength.QuadPart + (1 << 9)) >> 10;
- Unit = MUIGetString(STRING_KB);
+ sprintf(LineBuffer,
+ MUIGetString(STRING_HDDINFOUNK6),
+ DiskSize.u.LowPart,
+ Unit,
+ DiskEntry->DiskNumber,
+ DiskEntry->Port,
+ DiskEntry->Bus,
+ DiskEntry->Id);
}
- if (PartType == NULL)
+ if (List->Line >= 0 && List->Line <= Height)
{
- sprintf (LineBuffer,
- MUIGetString(STRING_HDDINFOUNK5),
- (PartEntry->DriveLetter[PartNumber] == 0) ? '-' : PartEntry->DriveLetter[PartNumber],
- (PartEntry->DriveLetter[PartNumber] == 0) ? '-' : ':',
- PartEntry->PartInfo[PartNumber].PartitionType,
- PartSize.u.LowPart,
- Unit);
+ FillConsoleOutputAttribute(StdOutput,
+ FOREGROUND_WHITE | BACKGROUND_BLUE,
+ Width,
+ coPos,
+ &Written);
+
+ FillConsoleOutputCharacterA(StdOutput,
+ ' ',
+ Width,
+ coPos,
+ &Written);
}
- else
+
+ coPos.X++;
+ if (List->Line >= 0 && List->Line <= Height)
{
- sprintf (LineBuffer,
- "%c%c %-24s %6lu %s",
- (PartEntry->DriveLetter[PartNumber] == 0) ? '-' : PartEntry->DriveLetter[PartNumber],
- (PartEntry->DriveLetter[PartNumber] == 0) ? '-' : ':',
- PartType,
- PartSize.u.LowPart,
- Unit);
- }
- }
-
- Attribute = (List->CurrentDisk == DiskEntry &&
- List->CurrentPartition == PartEntry &&
- List->CurrentPartitionNumber == PartNumber) ?
- FOREGROUND_BLUE | BACKGROUND_WHITE :
- FOREGROUND_WHITE | BACKGROUND_BLUE;
-
- if (List->Line >= 0 && List->Line <= Height)
- {
- FillConsoleOutputCharacterA (StdOutput,
- ' ',
- Width,
- coPos,
- &Written);
- }
- coPos.X += 4;
- Width -= 8;
- if (List->Line >= 0 && List->Line <= Height)
- {
- FillConsoleOutputAttribute (StdOutput,
- Attribute,
- Width,
- coPos,
- &Written);
- }
- coPos.X++;
- Width -= 2;
- if (List->Line >= 0 && List->Line <= Height)
- {
- WriteConsoleOutputCharacterA (StdOutput,
- LineBuffer,
- min (strlen (LineBuffer), Width),
- coPos,
- &Written);
- }
- List->Line++;
-}
+ WriteConsoleOutputCharacterA(StdOutput,
+ LineBuffer,
+ min((USHORT)strlen(LineBuffer), Width - 2),
+ coPos,
+ &Written);
+ }
+ List->Line++;
-static VOID
-PrintDiskData (PPARTLIST List,
- PDISKENTRY DiskEntry)
-{
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry;
- CHAR LineBuffer[128];
- COORD coPos;
- DWORD Written;
- USHORT Width;
- USHORT Height;
- ULARGE_INTEGER DiskSize;
- PCHAR Unit;
- ULONG i;
+ /* Print separator line */
+ PrintEmptyLine(List);
- Width = List->Right - List->Left - 1;
- Height = List->Bottom - List->Top - 2;
+ /* Print partition lines*/
+ Entry = DiskEntry->PartListHead.Flink;
+ while (Entry != &DiskEntry->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);
+ /* Print disk entry */
+ for (i = 0; i < 4; i++)
+ {
+ if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED ||
+ PartEntry->PartInfo[i].PartitionLength.QuadPart != 0ULL)
+ {
+ PrintPartitionData(List,
+ DiskEntry,
+ PartEntry,
+ i);
+ }
+ }
- coPos.X = List->Left + 1;
- coPos.Y = List->Top + 1 + List->Line;
+ /* Print unpartitioned entry */
+ if (PartEntry->Unpartitioned)
+ {
+ PrintPartitionData(List,
+ DiskEntry,
+ PartEntry,
+ 0);
+ }
-#if 0
- if (DiskEntry->DiskSize >= 0x280000000ULL) /* 10 GB */
- {
- DiskSize.QuadPart = (DiskEntry->DiskSize + (1 << 29)) >> 30;
- Unit = MUIGetString(STRING_GB);
- }
- else
-#endif
- {
- DiskSize.QuadPart = (DiskEntry->DiskSize + (1 << 19)) >> 20;
- if (DiskSize.QuadPart == 0)
- DiskSize.QuadPart = 1;
- Unit = MUIGetString(STRING_MB);
- }
-
- if (DiskEntry->DriverName.Length > 0)
- {
- sprintf (LineBuffer,
- MUIGetString(STRING_HDINFOPARTSELECT),
- DiskSize.u.LowPart,
- Unit,
- DiskEntry->DiskNumber,
- DiskEntry->Port,
- DiskEntry->Bus,
- DiskEntry->Id,
- DiskEntry->DriverName.Buffer);
- }
- else
- {
- sprintf (LineBuffer,
- MUIGetString(STRING_HDDINFOUNK6),
- DiskSize.u.LowPart,
- Unit,
- DiskEntry->DiskNumber,
- DiskEntry->Port,
- DiskEntry->Bus,
- DiskEntry->Id);
- }
- if (List->Line >= 0 && List->Line <= Height)
- {
- FillConsoleOutputAttribute (StdOutput,
- FOREGROUND_WHITE | BACKGROUND_BLUE,
- Width,
- coPos,
- &Written);
+ Entry = Entry->Flink;
+ }
- FillConsoleOutputCharacterA (StdOutput,
- ' ',
- Width,
- coPos,
- &Written);
- }
-
- coPos.X++;
- if (List->Line >= 0 && List->Line <= Height)
- {
- WriteConsoleOutputCharacterA (StdOutput,
- LineBuffer,
- min ((USHORT)strlen (LineBuffer), Width - 2),
- coPos,
- &Written);
- }
- List->Line++;
-
- /* Print separator line */
- PrintEmptyLine (List);
-
- /* Print partition lines*/
- Entry = DiskEntry->PartListHead.Flink;
- while (Entry != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);
-
- /* Print disk entry */
- for (i=0; i<4; i++)
- {
- if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED ||
- PartEntry->PartInfo[i].PartitionLength.QuadPart != 0ULL)
- {
- PrintPartitionData (List,
- DiskEntry,
- PartEntry,
- i);
- }
- }
-
- /* Print unpartitioned entry */
- if (PartEntry->Unpartitioned)
- {
- PrintPartitionData (List,
- DiskEntry,
- PartEntry,
- 0);
- }
-
- Entry = Entry->Flink;
- }
-
- /* Print separator line */
- PrintEmptyLine (List);
+ /* Print separator line */
+ PrintEmptyLine(List);
}
VOID
-DrawPartitionList (PPARTLIST List)
+DrawPartitionList(
+ PPARTLIST List)
{
- PLIST_ENTRY Entry, Entry2;
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry = NULL;
- COORD coPos;
- DWORD Written;
- SHORT i;
- SHORT CurrentDiskLine;
- SHORT CurrentPartLine;
- SHORT LastLine;
- BOOL CurrentPartLineFound = FALSE;
- BOOL CurrentDiskLineFound = FALSE;
-
- /* Calculate the line of the current disk and partition */
- CurrentDiskLine = 0;
- CurrentPartLine = 0;
- LastLine = 0;
- Entry = List->DiskListHead.Flink;
- while (Entry != &List->DiskListHead)
- {
- DiskEntry = CONTAINING_RECORD (Entry, DISKENTRY, ListEntry);
- LastLine += 2;
- if (CurrentPartLineFound == FALSE)
- {
- CurrentPartLine += 2;
- }
- Entry2 = DiskEntry->PartListHead.Flink;
- while (Entry2 != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry2, PARTENTRY, ListEntry);
- if (PartEntry == List->CurrentPartition)
- {
- CurrentPartLineFound = TRUE;
- }
- Entry2 = Entry2->Flink;
- if (CurrentPartLineFound == FALSE)
- {
- CurrentPartLine++;
- }
- LastLine++;
- }
- if (DiskEntry == List->CurrentDisk)
- {
- CurrentDiskLineFound = TRUE;
- }
- Entry = Entry->Flink;
- if (Entry != &List->DiskListHead)
- {
- if (CurrentDiskLineFound == FALSE)
- {
- CurrentPartLine ++;
- CurrentDiskLine = CurrentPartLine;
- }
- LastLine++;
+ PLIST_ENTRY Entry, Entry2;
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry = NULL;
+ COORD coPos;
+ DWORD Written;
+ SHORT i;
+ SHORT CurrentDiskLine;
+ SHORT CurrentPartLine;
+ SHORT LastLine;
+ BOOL CurrentPartLineFound = FALSE;
+ BOOL CurrentDiskLineFound = FALSE;
+
+ /* Calculate the line of the current disk and partition */
+ CurrentDiskLine = 0;
+ CurrentPartLine = 0;
+ LastLine = 0;
+
+ Entry = List->DiskListHead.Flink;
+ while (Entry != &List->DiskListHead)
+ {
+ DiskEntry = CONTAINING_RECORD(Entry, DISKENTRY, ListEntry);
+
+ LastLine += 2;
+ if (CurrentPartLineFound == FALSE)
+ {
+ CurrentPartLine += 2;
+ }
+
+ Entry2 = DiskEntry->PartListHead.Flink;
+ while (Entry2 != &DiskEntry->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
+ if (PartEntry == List->CurrentPartition)
+ {
+ CurrentPartLineFound = TRUE;
+ }
+
+ Entry2 = Entry2->Flink;
+ if (CurrentPartLineFound == FALSE)
+ {
+ CurrentPartLine++;
+ }
+
+ LastLine++;
+ }
+
+ if (DiskEntry == List->CurrentDisk)
+ {
+ CurrentDiskLineFound = TRUE;
+ }
+
+ Entry = Entry->Flink;
+ if (Entry != &List->DiskListHead)
+ {
+ if (CurrentDiskLineFound == FALSE)
+ {
+ CurrentPartLine ++;
+ CurrentDiskLine = CurrentPartLine;
+ }
+
+ LastLine++;
+ }
+ else
+ {
+ LastLine--;
+ }
+ }
+
+ /* If it possible, make the disk name visible */
+ if (CurrentPartLine < List->Offset)
+ {
+ List->Offset = CurrentPartLine;
+ }
+ else if (CurrentPartLine - List->Offset > List->Bottom - List->Top - 2)
+ {
+ List->Offset = CurrentPartLine - (List->Bottom - List->Top - 2);
+ }
+
+ if (CurrentDiskLine < List->Offset && CurrentPartLine - CurrentDiskLine < List->Bottom - List->Top - 2)
+ {
+ List->Offset = CurrentDiskLine;
+ }
+
+ /* draw upper left corner */
+ coPos.X = List->Left;
+ coPos.Y = List->Top;
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xDA, // '+',
+ 1,
+ coPos,
+ &Written);
+
+ /* draw upper edge */
+ coPos.X = List->Left + 1;
+ coPos.Y = List->Top;
+ if (List->Offset == 0)
+ {
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xC4, // '-',
+ List->Right - List->Left - 1,
+ coPos,
+ &Written);
}
else
{
- LastLine--;
- }
- }
-
- /* If it possible, make the disk name visible */
- if (CurrentPartLine < List->Offset)
- {
- List->Offset = CurrentPartLine;
- }
- else if (CurrentPartLine - List->Offset > List->Bottom - List->Top - 2)
- {
- List->Offset = CurrentPartLine - (List->Bottom - List->Top - 2);
- }
- if (CurrentDiskLine < List->Offset && CurrentPartLine - CurrentDiskLine < List->Bottom - List->Top - 2)
- {
- List->Offset = CurrentDiskLine;
- }
-
-
- /* draw upper left corner */
- coPos.X = List->Left;
- coPos.Y = List->Top;
- FillConsoleOutputCharacterA (StdOutput,
- 0xDA, // '+',
- 1,
- coPos,
- &Written);
-
- /* draw upper edge */
- coPos.X = List->Left + 1;
- coPos.Y = List->Top;
- if (List->Offset == 0)
- {
- FillConsoleOutputCharacterA (StdOutput,
- 0xC4, // '-',
- List->Right - List->Left - 1,
- coPos,
- &Written);
- }
- else
- {
- FillConsoleOutputCharacterA (StdOutput,
- 0xC4, // '-',
- List->Right - List->Left - 5,
- coPos,
- &Written);
- coPos.X = List->Right - 5;
- WriteConsoleOutputCharacterA (StdOutput,
- "(\x18)", // "(up)"
- 3,
- coPos,
- &Written);
- coPos.X = List->Right - 2;
- FillConsoleOutputCharacterA (StdOutput,
- 0xC4, // '-',
- 2,
- coPos,
- &Written);
- }
-
- /* draw upper right corner */
- coPos.X = List->Right;
- coPos.Y = List->Top;
- FillConsoleOutputCharacterA (StdOutput,
- 0xBF, // '+',
- 1,
- coPos,
- &Written);
-
- /* draw left and right edge */
- for (i = List->Top + 1; i < List->Bottom; i++)
- {
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xC4, // '-',
+ List->Right - List->Left - 5,
+ coPos,
+ &Written);
+ coPos.X = List->Right - 5;
+ WriteConsoleOutputCharacterA(StdOutput,
+ "(\x18)", // "(up)"
+ 3,
+ coPos,
+ &Written);
+ coPos.X = List->Right - 2;
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xC4, // '-',
+ 2,
+ coPos,
+ &Written);
+ }
+
+ /* draw upper right corner */
+ coPos.X = List->Right;
+ coPos.Y = List->Top;
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xBF, // '+',
+ 1,
+ coPos,
+ &Written);
+
+ /* draw left and right edge */
+ for (i = List->Top + 1; i < List->Bottom; i++)
+ {
+ coPos.X = List->Left;
+ coPos.Y = i;
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xB3, // '|',
+ 1,
+ coPos,
+ &Written);
+
+ coPos.X = List->Right;
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xB3, //'|',
+ 1,
+ coPos,
+ &Written);
+ }
+
+ /* draw lower left corner */
coPos.X = List->Left;
- coPos.Y = i;
- FillConsoleOutputCharacterA (StdOutput,
- 0xB3, // '|',
- 1,
- coPos,
- &Written);
+ coPos.Y = List->Bottom;
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xC0, // '+',
+ 1,
+ coPos,
+ &Written);
+
+ /* draw lower edge */
+ coPos.X = List->Left + 1;
+ coPos.Y = List->Bottom;
+ if (LastLine - List->Offset <= List->Bottom - List->Top - 2)
+ {
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xC4, // '-',
+ List->Right - List->Left - 1,
+ coPos,
+ &Written);
+ }
+ else
+ {
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xC4, // '-',
+ List->Right - List->Left - 5,
+ coPos,
+ &Written);
+ coPos.X = List->Right - 5;
+ WriteConsoleOutputCharacterA(StdOutput,
+ "(\x19)", // "(down)"
+ 3,
+ coPos,
+ &Written);
+ coPos.X = List->Right - 2;
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xC4, // '-',
+ 2,
+ coPos,
+ &Written);
+ }
+ /* draw lower right corner */
coPos.X = List->Right;
- FillConsoleOutputCharacterA (StdOutput,
- 0xB3, //'|',
- 1,
- coPos,
- &Written);
- }
-
- /* draw lower left corner */
- coPos.X = List->Left;
- coPos.Y = List->Bottom;
- FillConsoleOutputCharacterA (StdOutput,
- 0xC0, // '+',
- 1,
- coPos,
- &Written);
-
- /* draw lower edge */
- coPos.X = List->Left + 1;
- coPos.Y = List->Bottom;
- if (LastLine - List->Offset <= List->Bottom - List->Top - 2)
- {
- FillConsoleOutputCharacterA (StdOutput,
- 0xC4, // '-',
- List->Right - List->Left - 1,
- coPos,
- &Written);
- }
- else
- {
- FillConsoleOutputCharacterA (StdOutput,
- 0xC4, // '-',
- List->Right - List->Left - 5,
- coPos,
- &Written);
- coPos.X = List->Right - 5;
- WriteConsoleOutputCharacterA (StdOutput,
- "(\x19)", // "(down)"
- 3,
- coPos,
- &Written);
- coPos.X = List->Right - 2;
- FillConsoleOutputCharacterA (StdOutput,
- 0xC4, // '-',
- 2,
- coPos,
- &Written);
- }
-
- /* draw lower right corner */
- coPos.X = List->Right;
- coPos.Y = List->Bottom;
- FillConsoleOutputCharacterA (StdOutput,
- 0xD9, // '+',
- 1,
- coPos,
- &Written);
-
- /* print list entries */
- List->Line = - List->Offset;
-
- Entry = List->DiskListHead.Flink;
- while (Entry != &List->DiskListHead)
- {
- DiskEntry = CONTAINING_RECORD (Entry, DISKENTRY, ListEntry);
-
- /* Print disk entry */
- PrintDiskData (List,
- DiskEntry);
-
- Entry = Entry->Flink;
- }
+ coPos.Y = List->Bottom;
+ FillConsoleOutputCharacterA(StdOutput,
+ 0xD9, // '+',
+ 1,
+ coPos,
+ &Written);
+
+ /* print list entries */
+ List->Line = - List->Offset;
+
+ Entry = List->DiskListHead.Flink;
+ while (Entry != &List->DiskListHead)
+ {
+ DiskEntry = CONTAINING_RECORD(Entry, DISKENTRY, ListEntry);
+
+ /* Print disk entry */
+ PrintDiskData(List,
+ DiskEntry);
+
+ Entry = Entry->Flink;
+ }
}
DWORD
-SelectPartition(PPARTLIST List, ULONG DiskNumber, ULONG PartitionNumber)
+SelectPartition(
+ PPARTLIST List,
+ ULONG DiskNumber,
+ ULONG PartitionNumber)
{
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry1;
- PLIST_ENTRY Entry2;
- UCHAR i;
-
- /* Check for empty disks */
- if (IsListEmpty (&List->DiskListHead))
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry1;
+ PLIST_ENTRY Entry2;
+ UCHAR i;
+
+ /* Check for empty disks */
+ if (IsListEmpty(&List->DiskListHead))
+ return FALSE;
+
+ /* Check for first usable entry on next disk */
+ Entry1 = List->CurrentDisk->ListEntry.Flink;
+ while (Entry1 != &List->DiskListHead)
+ {
+ DiskEntry = CONTAINING_RECORD(Entry1, DISKENTRY, ListEntry);
+
+ if (DiskEntry->DiskNumber == DiskNumber)
+ {
+ Entry2 = DiskEntry->PartListHead.Flink;
+ while (Entry2 != &DiskEntry->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
+
+ for (i = 0; i < 4; i++)
+ {
+ if (PartEntry->PartInfo[i].PartitionNumber == PartitionNumber)
+ {
+ List->CurrentDisk = DiskEntry;
+ List->CurrentPartition = PartEntry;
+ List->CurrentPartitionNumber = i;
+ DrawPartitionList(List);
+ return TRUE;
+ }
+ }
+
+ Entry2 = Entry2->Flink;
+ }
+
+ return FALSE;
+ }
+
+ Entry1 = Entry1->Flink;
+ }
+
return FALSE;
+}
+
- /* Check for first usable entry on next disk */
- Entry1 = List->CurrentDisk->ListEntry.Flink;
- while (Entry1 != &List->DiskListHead)
- {
- DiskEntry = CONTAINING_RECORD (Entry1, DISKENTRY, ListEntry);
+VOID
+ScrollDownPartitionList(
+ PPARTLIST List)
+{
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry1;
+ PLIST_ENTRY Entry2;
+ UCHAR i;
+
+ /* Check for empty disks */
+ if (IsListEmpty(&List->DiskListHead))
+ return;
- if (DiskEntry->DiskNumber == DiskNumber)
+ /* Check for next usable entry on current disk */
+ if (List->CurrentPartition != NULL)
{
- Entry2 = DiskEntry->PartListHead.Flink;
- while (Entry2 != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry2, PARTENTRY, ListEntry);
+ Entry2 = &List->CurrentPartition->ListEntry;
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
- for (i = 0; i < 4; i++)
+ /* Check if we can move inside primary partitions */
+ for (i = List->CurrentPartitionNumber + 1; i < 4; i++)
{
- if (PartEntry->PartInfo[i].PartitionNumber == PartitionNumber)
- {
- List->CurrentDisk = DiskEntry;
- List->CurrentPartition = PartEntry;
+ if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED)
+ break;
+ }
+
+ if (i == 4)
+ {
+ /* We're out of partitions in the current partition table.
+ Try to move to the next one if possible. */
+ Entry2 = Entry2->Flink;
+ }
+ else
+ {
+ /* Just advance to the next partition */
List->CurrentPartitionNumber = i;
- DrawPartitionList (List);
- return TRUE;
- }
+ DrawPartitionList(List);
+ return;
+ }
+
+ while (Entry2 != &List->CurrentDisk->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
+
+// if (PartEntry->HidePartEntry == FALSE)
+ {
+ List->CurrentPartition = PartEntry;
+ List->CurrentPartitionNumber = 0;
+ DrawPartitionList(List);
+ return;
+ }
+
+ Entry2 = Entry2->Flink;
+ }
+ }
+
+ /* Check for first usable entry on next disk */
+ if (List->CurrentDisk != NULL)
+ {
+ Entry1 = List->CurrentDisk->ListEntry.Flink;
+ while (Entry1 != &List->DiskListHead)
+ {
+ DiskEntry = CONTAINING_RECORD(Entry1, DISKENTRY, ListEntry);
+
+ Entry2 = DiskEntry->PartListHead.Flink;
+ while (Entry2 != &DiskEntry->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
+
+// if (PartEntry->HidePartEntry == FALSE)
+ {
+ List->CurrentDisk = DiskEntry;
+ List->CurrentPartition = PartEntry;
+ List->CurrentPartitionNumber = 0;
+ DrawPartitionList(List);
+ return;
+ }
+
+ Entry2 = Entry2->Flink;
+ }
+
+ Entry1 = Entry1->Flink;
}
- Entry2 = Entry2->Flink;
- }
- return FALSE;
}
- Entry1 = Entry1->Flink;
- }
- return FALSE;
}
VOID
-ScrollDownPartitionList (PPARTLIST List)
+ScrollUpPartitionList(
+ PPARTLIST List)
+{
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry1;
+ PLIST_ENTRY Entry2;
+ UCHAR i;
+
+ /* Check for empty disks */
+ if (IsListEmpty(&List->DiskListHead))
+ return;
+
+ /* check for previous usable entry on current disk */
+ if (List->CurrentPartition != NULL)
+ {
+ Entry2 = &List->CurrentPartition->ListEntry;
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
+
+ /* Check if we can move inside primary partitions */
+ if (List->CurrentPartitionNumber > 0)
+ {
+ /* Find a previous partition */
+ for (i = List->CurrentPartitionNumber - 1; i > 0; i--)
+ {
+ if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED)
+ break;
+ }
+
+ /* Move to it and return */
+ List->CurrentPartitionNumber = i;
+ DrawPartitionList(List);
+ return;
+ }
+
+ /* Move to the previous entry */
+ Entry2 = Entry2->Blink;
+
+ while (Entry2 != &List->CurrentDisk->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
+
+// if (PartEntry->HidePartEntry == FALSE)
+ {
+ List->CurrentPartition = PartEntry;
+
+ /* Find last existing partition in the table */
+ for (i = 3; i > 0; i--)
+ {
+ if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED)
+ break;
+ }
+
+ /* Move to it */
+ List->CurrentPartitionNumber = i;
+
+ /* Draw partition list and return */
+ DrawPartitionList(List);
+ return;
+ }
+
+ Entry2 = Entry2->Blink;
+ }
+ }
+
+
+ /* check for last usable entry on previous disk */
+ if (List->CurrentDisk != NULL)
+ {
+ Entry1 = List->CurrentDisk->ListEntry.Blink;
+ while (Entry1 != &List->DiskListHead)
+ {
+ DiskEntry = CONTAINING_RECORD(Entry1, DISKENTRY, ListEntry);
+
+ Entry2 = DiskEntry->PartListHead.Blink;
+ while (Entry2 != &DiskEntry->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
+
+// if (PartEntry->HidePartEntry == FALSE)
+ {
+ List->CurrentDisk = DiskEntry;
+ List->CurrentPartition = PartEntry;
+
+ /* Find last existing partition in the table */
+ for (i = 3; i > 0; i--)
+ {
+ if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED)
+ break;
+ }
+
+ /* Move to it */
+ List->CurrentPartitionNumber = i;
+
+ /* Draw partition list and return */
+ DrawPartitionList(List);
+ return;
+ }
+
+ Entry2 = Entry2->Blink;
+ }
+
+ Entry1 = Entry1->Blink;
+ }
+ }
+}
+
+
+static
+PPARTENTRY
+GetPrevPartitionedEntry(
+ PDISKENTRY DiskEntry,
+ PPARTENTRY CurrentEntry)
{
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry1;
- PLIST_ENTRY Entry2;
- UCHAR i;
-
- /* Check for empty disks */
- if (IsListEmpty (&List->DiskListHead))
- return;
-
- /* Check for next usable entry on current disk */
- if (List->CurrentPartition != NULL)
- {
- Entry2 = &List->CurrentPartition->ListEntry;
- PartEntry = CONTAINING_RECORD (Entry2, PARTENTRY, ListEntry);
-
- /* Check if we can move inside primary partitions */
- for (i = List->CurrentPartitionNumber + 1; i < 4; i++)
- {
- if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED)
- break;
- }
+ PPARTENTRY PrevEntry;
+ PLIST_ENTRY Entry;
- if (i == 4)
- {
- /* We're out of partitions in the current partition table.
- Try to move to the next one if possible. */
- Entry2 = Entry2->Flink;
- }
- else
- {
- /* Just advance to the next partition */
- List->CurrentPartitionNumber = i;
- DrawPartitionList (List);
- return;
- }
+ if (CurrentEntry->ListEntry.Blink == &DiskEntry->PartListHead)
+ return NULL;
- while (Entry2 != &List->CurrentDisk->PartListHead)
+ Entry = CurrentEntry->ListEntry.Blink;
+ while (Entry != &DiskEntry->PartListHead)
{
- PartEntry = CONTAINING_RECORD (Entry2, PARTENTRY, ListEntry);
+ PrevEntry = CONTAINING_RECORD(Entry,
+ PARTENTRY,
+ ListEntry);
+ if (PrevEntry->Unpartitioned == FALSE)
+ return PrevEntry;
-// if (PartEntry->HidePartEntry == FALSE)
- {
- List->CurrentPartition = PartEntry;
- List->CurrentPartitionNumber = 0;
- DrawPartitionList (List);
- return;
- }
- Entry2 = Entry2->Flink;
+ Entry = Entry->Blink;
}
- }
- /* Check for first usable entry on next disk */
- if (List->CurrentDisk != NULL)
- {
- Entry1 = List->CurrentDisk->ListEntry.Flink;
- while (Entry1 != &List->DiskListHead)
- {
- DiskEntry = CONTAINING_RECORD (Entry1, DISKENTRY, ListEntry);
+ return NULL;
+}
- Entry2 = DiskEntry->PartListHead.Flink;
- while (Entry2 != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry2, PARTENTRY, ListEntry);
-// if (PartEntry->HidePartEntry == FALSE)
- {
- List->CurrentDisk = DiskEntry;
- List->CurrentPartition = PartEntry;
- List->CurrentPartitionNumber = 0;
- DrawPartitionList (List);
- return;
- }
+static
+PPARTENTRY
+GetNextPartitionedEntry(
+ PDISKENTRY DiskEntry,
+ PPARTENTRY CurrentEntry)
+{
+ PPARTENTRY NextEntry;
+ PLIST_ENTRY Entry;
+
+ if (CurrentEntry->ListEntry.Flink == &DiskEntry->PartListHead)
+ return NULL;
- Entry2 = Entry2->Flink;
- }
+ Entry = CurrentEntry->ListEntry.Flink;
+ while (Entry != &DiskEntry->PartListHead)
+ {
+ NextEntry = CONTAINING_RECORD(Entry,
+ PARTENTRY,
+ ListEntry);
+ if (NextEntry->Unpartitioned == FALSE)
+ return NextEntry;
- Entry1 = Entry1->Flink;
+ Entry = Entry->Flink;
}
- }
+
+ return NULL;
}
-VOID
-ScrollUpPartitionList (PPARTLIST List)
+static
+PPARTENTRY
+GetPrevUnpartitionedEntry(
+ PDISKENTRY DiskEntry,
+ PPARTENTRY PartEntry)
{
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry1;
- PLIST_ENTRY Entry2;
- UCHAR i;
-
- /* Check for empty disks */
- if (IsListEmpty (&List->DiskListHead))
- return;
-
- /* check for previous usable entry on current disk */
- if (List->CurrentPartition != NULL)
- {
- Entry2 = &List->CurrentPartition->ListEntry;
- PartEntry = CONTAINING_RECORD (Entry2, PARTENTRY, ListEntry);
-
- /* Check if we can move inside primary partitions */
- if (List->CurrentPartitionNumber > 0)
- {
- /* Find a previous partition */
- for (i = List->CurrentPartitionNumber - 1; i > 0; i--)
- {
- if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED)
- break;
- }
+ PPARTENTRY PrevPartEntry;
- /* Move to it and return */
- List->CurrentPartitionNumber = i;
- DrawPartitionList (List);
- return;
+ if (PartEntry->ListEntry.Blink != &DiskEntry->PartListHead)
+ {
+ PrevPartEntry = CONTAINING_RECORD(PartEntry->ListEntry.Blink,
+ PARTENTRY,
+ ListEntry);
+ if (PrevPartEntry->Unpartitioned == TRUE)
+ return PrevPartEntry;
}
- /* Move to the previous entry */
- Entry2 = Entry2->Blink;
+ return NULL;
+}
+
- while (Entry2 != &List->CurrentDisk->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry2, PARTENTRY, ListEntry);
+static
+PPARTENTRY
+GetNextUnpartitionedEntry(
+ PDISKENTRY DiskEntry,
+ PPARTENTRY PartEntry)
+{
+ PPARTENTRY NextPartEntry;
-// if (PartEntry->HidePartEntry == FALSE)
- {
- List->CurrentPartition = PartEntry;
+ if (PartEntry->ListEntry.Flink != &DiskEntry->PartListHead)
+ {
+ NextPartEntry = CONTAINING_RECORD(PartEntry->ListEntry.Flink,
+ PARTENTRY,
+ ListEntry);
+ if (NextPartEntry->Unpartitioned == TRUE)
+ return NextPartEntry;
+ }
- /* Find last existing partition in the table */
- for (i = 3; i > 0; i--)
- {
- if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED)
- break;
- }
+ return NULL;
+}
- /* Move to it */
- List->CurrentPartitionNumber = i;
- /* Draw partition list and return */
- DrawPartitionList (List);
+VOID
+CreateNewPartition(
+ PPARTLIST List,
+ ULONGLONG PartitionSize,
+ BOOLEAN AutoCreate)
+{
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ PPARTENTRY PrevPartEntry;
+ PPARTENTRY NextPartEntry;
+ PPARTENTRY NewPartEntry;
+
+ if (List == NULL ||
+ List->CurrentDisk == NULL ||
+ List->CurrentPartition == NULL ||
+ List->CurrentPartition->Unpartitioned == FALSE)
+ {
return;
- }
- Entry2 = Entry2->Blink;
}
- }
+ DiskEntry = List->CurrentDisk;
+ PartEntry = List->CurrentPartition;
- /* check for last usable entry on previous disk */
- if (List->CurrentDisk != NULL)
- {
- Entry1 = List->CurrentDisk->ListEntry.Blink;
- while (Entry1 != &List->DiskListHead)
+ if (AutoCreate == TRUE ||
+ PartitionSize == PartEntry->UnpartitionedLength)
{
- DiskEntry = CONTAINING_RECORD (Entry1, DISKENTRY, ListEntry);
+ /* Convert current entry to 'new (unformatted)' */
+ PartEntry->FormatState = Unformatted;
+ PartEntry->PartInfo[0].StartingOffset.QuadPart =
+ PartEntry->UnpartitionedOffset + DiskEntry->TrackSize;
+ PartEntry->PartInfo[0].HiddenSectors =
+ (ULONG)(PartEntry->PartInfo[0].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
+ PartEntry->PartInfo[0].PartitionLength.QuadPart =
+ PartEntry->UnpartitionedLength - DiskEntry->TrackSize;
+ PartEntry->PartInfo[0].PartitionType = PARTITION_ENTRY_UNUSED;
+ PartEntry->PartInfo[0].BootIndicator = FALSE; /* FIXME */
+ PartEntry->PartInfo[0].RewritePartition = TRUE;
+ PartEntry->PartInfo[1].RewritePartition = TRUE;
+ PartEntry->PartInfo[2].RewritePartition = TRUE;
+ PartEntry->PartInfo[3].RewritePartition = TRUE;
+
+ /* Get previous and next partition entries */
+ PrevPartEntry = GetPrevPartitionedEntry(DiskEntry,
+ PartEntry);
+ NextPartEntry = GetNextPartitionedEntry(DiskEntry,
+ PartEntry);
+
+ if (PrevPartEntry != NULL && NextPartEntry != NULL)
+ {
+ /* Current entry is in the middle of the list */
- Entry2 = DiskEntry->PartListHead.Blink;
- while (Entry2 != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry2, PARTENTRY, ListEntry);
+ /* Copy previous container partition data to current entry */
+ RtlCopyMemory(&PartEntry->PartInfo[1],
+ &PrevPartEntry->PartInfo[1],
+ sizeof(PARTITION_INFORMATION));
+ PartEntry->PartInfo[1].RewritePartition = TRUE;
-// if (PartEntry->HidePartEntry == FALSE)
- {
- List->CurrentDisk = DiskEntry;
- List->CurrentPartition = PartEntry;
+ /* Update previous container partition data */
- /* Find last existing partition in the table */
- for (i = 3; i > 0; i--)
- {
- if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED)
- break;
- }
+ PrevPartEntry->PartInfo[1].StartingOffset.QuadPart =
+ PartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
+ PrevPartEntry->PartInfo[1].HiddenSectors =
+ (ULONG)(PrevPartEntry->PartInfo[1].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
- /* Move to it */
- List->CurrentPartitionNumber = i;
+ if (DiskEntry->PartListHead.Flink == &PrevPartEntry->ListEntry)
+ {
+ /* Special case - previous partition is first partition */
+ PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
+ DiskEntry->DiskSize - PrevPartEntry->PartInfo[1].StartingOffset.QuadPart;
+ }
+ else
+ {
+ PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
+ PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
+ }
- /* Draw partition list and return */
- DrawPartitionList (List);
- return;
+ PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
}
+ else if (PrevPartEntry == NULL && NextPartEntry != NULL)
+ {
+ /* Current entry is the first entry */
+ return;
+ }
+ else if (PrevPartEntry != NULL && NextPartEntry == NULL)
+ {
+ /* Current entry is the last entry */
- Entry2 = Entry2->Blink;
- }
-
- Entry1 = Entry1->Blink;
- }
- }
-}
+ PrevPartEntry->PartInfo[1].StartingOffset.QuadPart =
+ PartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
+ PrevPartEntry->PartInfo[1].HiddenSectors =
+ (ULONG)(PrevPartEntry->PartInfo[1].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
+ if (DiskEntry->PartListHead.Flink == &PrevPartEntry->ListEntry)
+ {
+ /* Special case - previous partition is first partition */
+ PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
+ DiskEntry->DiskSize - PrevPartEntry->PartInfo[1].StartingOffset.QuadPart;
+ }
+ else
+ {
+ PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
+ PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
+ }
-static PPARTENTRY
-GetPrevPartitionedEntry (PDISKENTRY DiskEntry,
- PPARTENTRY CurrentEntry)
-{
- PPARTENTRY PrevEntry;
- PLIST_ENTRY Entry;
+ if ((PartEntry->PartInfo[1].StartingOffset.QuadPart +
+ PartEntry->PartInfo[1].PartitionLength.QuadPart) <
+ (1024LL * 255LL * 63LL * 512LL))
+ {
+ PrevPartEntry->PartInfo[1].PartitionType = PARTITION_EXTENDED;
+ }
+ else
+ {
+ PrevPartEntry->PartInfo[1].PartitionType = PARTITION_XINT13_EXTENDED;
+ }
- if (CurrentEntry->ListEntry.Blink == &DiskEntry->PartListHead)
- return NULL;
+ PrevPartEntry->PartInfo[1].BootIndicator = FALSE;
+ PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
+ }
- Entry = CurrentEntry->ListEntry.Blink;
- while (Entry != &DiskEntry->PartListHead)
- {
- PrevEntry = CONTAINING_RECORD (Entry,
- PARTENTRY,
- ListEntry);
- if (PrevEntry->Unpartitioned == FALSE)
- return PrevEntry;
+ PartEntry->AutoCreate = AutoCreate;
+ PartEntry->New = TRUE;
+ PartEntry->Unpartitioned = FALSE;
+ PartEntry->UnpartitionedOffset = 0ULL;
+ PartEntry->UnpartitionedLength = 0ULL;
+ }
+ else
+ {
+ /* Insert an initialize a new partition entry */
+ NewPartEntry = (PPARTENTRY)RtlAllocateHeap(ProcessHeap,
+ 0,
+ sizeof(PARTENTRY));
+ if (NewPartEntry == NULL)
+ return;
+
+ RtlZeroMemory(NewPartEntry,
+ sizeof(PARTENTRY));
+
+ /* Insert the new entry into the list */
+ InsertTailList(&PartEntry->ListEntry,
+ &NewPartEntry->ListEntry);
+
+ NewPartEntry->New = TRUE;
+
+ NewPartEntry->FormatState = Unformatted;
+ NewPartEntry->PartInfo[0].StartingOffset.QuadPart =
+ PartEntry->UnpartitionedOffset + DiskEntry->TrackSize;
+ NewPartEntry->PartInfo[0].HiddenSectors =
+ (ULONG)(NewPartEntry->PartInfo[0].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
+ NewPartEntry->PartInfo[0].PartitionLength.QuadPart =
+ PartitionSize - DiskEntry->TrackSize;
+ NewPartEntry->PartInfo[0].PartitionType = PARTITION_ENTRY_UNUSED;
+ NewPartEntry->PartInfo[0].BootIndicator = FALSE; /* FIXME */
+ NewPartEntry->PartInfo[0].RewritePartition = TRUE;
+ NewPartEntry->PartInfo[1].RewritePartition = TRUE;
+ NewPartEntry->PartInfo[2].RewritePartition = TRUE;
+ NewPartEntry->PartInfo[3].RewritePartition = TRUE;
+
+ /* Get previous and next partition entries */
+ PrevPartEntry = GetPrevPartitionedEntry(DiskEntry,
+ NewPartEntry);
+ NextPartEntry = GetNextPartitionedEntry(DiskEntry,
+ NewPartEntry);
+
+ if (PrevPartEntry != NULL && NextPartEntry != NULL)
+ {
+ /* Current entry is in the middle of the list */
- Entry = Entry->Blink;
- }
+ /* Copy previous container partition data to current entry */
+ RtlCopyMemory(&NewPartEntry->PartInfo[1],
+ &PrevPartEntry->PartInfo[1],
+ sizeof(PARTITION_INFORMATION));
+ NewPartEntry->PartInfo[1].RewritePartition = TRUE;
- return NULL;
-}
+ /* Update previous container partition data */
+ PrevPartEntry->PartInfo[1].StartingOffset.QuadPart =
+ NewPartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
+ PrevPartEntry->PartInfo[1].HiddenSectors =
+ (ULONG)(PrevPartEntry->PartInfo[1].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
+ if (DiskEntry->PartListHead.Flink == &PrevPartEntry->ListEntry)
+ {
+ /* Special case - previous partition is first partition */
+ PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
+ DiskEntry->DiskSize - PrevPartEntry->PartInfo[1].StartingOffset.QuadPart;
+ }
+ else
+ {
+ PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
+ NewPartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
+ }
-static PPARTENTRY
-GetNextPartitionedEntry (PDISKENTRY DiskEntry,
- PPARTENTRY CurrentEntry)
-{
- PPARTENTRY NextEntry;
- PLIST_ENTRY Entry;
+ PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
+ }
+ else if (PrevPartEntry == NULL && NextPartEntry != NULL)
+ {
+ /* Current entry is the first entry */
+ return;
+ }
+ else if (PrevPartEntry != NULL && NextPartEntry == NULL)
+ {
+ /* Current entry is the last entry */
- if (CurrentEntry->ListEntry.Flink == &DiskEntry->PartListHead)
- return NULL;
+ PrevPartEntry->PartInfo[1].StartingOffset.QuadPart =
+ NewPartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
+ PrevPartEntry->PartInfo[1].HiddenSectors =
+ (ULONG)(PrevPartEntry->PartInfo[1].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
- Entry = CurrentEntry->ListEntry.Flink;
- while (Entry != &DiskEntry->PartListHead)
- {
- NextEntry = CONTAINING_RECORD (Entry,
- PARTENTRY,
- ListEntry);
- if (NextEntry->Unpartitioned == FALSE)
- return NextEntry;
+ if (DiskEntry->PartListHead.Flink == &PrevPartEntry->ListEntry)
+ {
+ /* Special case - previous partition is first partition */
+ PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
+ DiskEntry->DiskSize - PrevPartEntry->PartInfo[1].StartingOffset.QuadPart;
+ }
+ else
+ {
+ PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
+ NewPartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
+ }
- Entry = Entry->Flink;
- }
+ if ((PartEntry->PartInfo[1].StartingOffset.QuadPart +
+ PartEntry->PartInfo[1].PartitionLength.QuadPart) <
+ (1024LL * 255LL * 63LL * 512LL))
+ {
+ PrevPartEntry->PartInfo[1].PartitionType = PARTITION_EXTENDED;
+ }
+ else
+ {
+ PrevPartEntry->PartInfo[1].PartitionType = PARTITION_XINT13_EXTENDED;
+ }
- return NULL;
-}
+ PrevPartEntry->PartInfo[1].BootIndicator = FALSE;
+ PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
+ }
+ /* Update offset and size of the remaining unpartitioned disk space */
+ PartEntry->UnpartitionedOffset += PartitionSize;
+ PartEntry->UnpartitionedLength -= PartitionSize;
+ }
-static PPARTENTRY
-GetPrevUnpartitionedEntry (PDISKENTRY DiskEntry,
- PPARTENTRY PartEntry)
-{
- PPARTENTRY PrevPartEntry;
+ DiskEntry->Modified = TRUE;
- if (PartEntry->ListEntry.Blink != &DiskEntry->PartListHead)
- {
- PrevPartEntry = CONTAINING_RECORD (PartEntry->ListEntry.Blink,
- PARTENTRY,
- ListEntry);
- if (PrevPartEntry->Unpartitioned == TRUE)
- return PrevPartEntry;
- }
+ UpdatePartitionNumbers(DiskEntry);
- return NULL;
+ AssignDriverLetters(List);
}
-static PPARTENTRY
-GetNextUnpartitionedEntry (PDISKENTRY DiskEntry,
- PPARTENTRY PartEntry)
+VOID
+DeleteCurrentPartition(
+ PPARTLIST List)
{
- PPARTENTRY NextPartEntry;
-
- if (PartEntry->ListEntry.Flink != &DiskEntry->PartListHead)
- {
- NextPartEntry = CONTAINING_RECORD (PartEntry->ListEntry.Flink,
- PARTENTRY,
- ListEntry);
- if (NextPartEntry->Unpartitioned == TRUE)
- return NextPartEntry;
- }
-
- return NULL;
-}
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ PPARTENTRY PrevPartEntry;
+ PPARTENTRY NextPartEntry;
+
+ if (List == NULL ||
+ List->CurrentDisk == NULL ||
+ List->CurrentPartition == NULL ||
+ List->CurrentPartition->Unpartitioned == TRUE)
+ {
+ return;
+ }
+ DiskEntry = List->CurrentDisk;
+ PartEntry = List->CurrentPartition;
-VOID
-CreateNewPartition (PPARTLIST List,
- ULONGLONG PartitionSize,
- BOOLEAN AutoCreate)
-{
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- PPARTENTRY PrevPartEntry;
- PPARTENTRY NextPartEntry;
- PPARTENTRY NewPartEntry;
-
- if (List == NULL ||
- List->CurrentDisk == NULL ||
- List->CurrentPartition == NULL ||
- List->CurrentPartition->Unpartitioned == FALSE)
- {
- return;
- }
-
- DiskEntry = List->CurrentDisk;
- PartEntry = List->CurrentPartition;
-
- if (AutoCreate == TRUE ||
- PartitionSize == PartEntry->UnpartitionedLength)
- {
- /* Convert current entry to 'new (unformatted)' */
- PartEntry->FormatState = Unformatted;
- PartEntry->PartInfo[0].StartingOffset.QuadPart =
- PartEntry->UnpartitionedOffset + DiskEntry->TrackSize;
- PartEntry->PartInfo[0].HiddenSectors =
- (ULONG)(PartEntry->PartInfo[0].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
- PartEntry->PartInfo[0].PartitionLength.QuadPart =
- PartEntry->UnpartitionedLength - DiskEntry->TrackSize;
- PartEntry->PartInfo[0].PartitionType = PARTITION_ENTRY_UNUSED;
- PartEntry->PartInfo[0].BootIndicator = FALSE; /* FIXME */
- PartEntry->PartInfo[0].RewritePartition = TRUE;
- PartEntry->PartInfo[1].RewritePartition = TRUE;
- PartEntry->PartInfo[2].RewritePartition = TRUE;
- PartEntry->PartInfo[3].RewritePartition = TRUE;
+ /* Adjust container partition entries */
/* Get previous and next partition entries */
- PrevPartEntry = GetPrevPartitionedEntry (DiskEntry,
- PartEntry);
- NextPartEntry = GetNextPartitionedEntry (DiskEntry,
- PartEntry);
+ PrevPartEntry = GetPrevPartitionedEntry(DiskEntry,
+ PartEntry);
+ NextPartEntry = GetNextPartitionedEntry(DiskEntry,
+ PartEntry);
if (PrevPartEntry != NULL && NextPartEntry != NULL)
{
- /* Current entry is in the middle of the list */
-
- /* Copy previous container partition data to current entry */
- RtlCopyMemory (&PartEntry->PartInfo[1],
- &PrevPartEntry->PartInfo[1],
- sizeof(PARTITION_INFORMATION));
- PartEntry->PartInfo[1].RewritePartition = TRUE;
-
- /* Update previous container partition data */
-
- PrevPartEntry->PartInfo[1].StartingOffset.QuadPart =
- PartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
- PrevPartEntry->PartInfo[1].HiddenSectors =
- (ULONG)(PrevPartEntry->PartInfo[1].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
-
- if (DiskEntry->PartListHead.Flink == &PrevPartEntry->ListEntry)
- {
- /* Special case - previous partition is first partition */
- PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
- DiskEntry->DiskSize - PrevPartEntry->PartInfo[1].StartingOffset.QuadPart;
- }
- else
- {
- PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
- PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
- }
-
- PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
+ /* Current entry is in the middle of the list */
+
+ /*
+ * The first extended partition can not be deleted
+ * as long as other extended partitions are present.
+ */
+ if (PrevPartEntry->ListEntry.Blink == &DiskEntry->PartListHead)
+ return;
+
+ /* Copy previous container partition data to current entry */
+ RtlCopyMemory(&PrevPartEntry->PartInfo[1],
+ &PartEntry->PartInfo[1],
+ sizeof(PARTITION_INFORMATION));
+ PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
}
else if (PrevPartEntry == NULL && NextPartEntry != NULL)
{
- /* Current entry is the first entry */
- return;
+ /*
+ * A primary partition can not be deleted as long as
+ * extended partitions are present.
+ */
+ return;
}
else if (PrevPartEntry != NULL && NextPartEntry == NULL)
{
- /* Current entry is the last entry */
-
- PrevPartEntry->PartInfo[1].StartingOffset.QuadPart =
- PartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
- PrevPartEntry->PartInfo[1].HiddenSectors =
- (ULONG)(PrevPartEntry->PartInfo[1].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
-
- if (DiskEntry->PartListHead.Flink == &PrevPartEntry->ListEntry)
- {
- /* Special case - previous partition is first partition */
- PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
- DiskEntry->DiskSize - PrevPartEntry->PartInfo[1].StartingOffset.QuadPart;
- }
- else
- {
- PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
- PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
- }
-
- if ((PartEntry->PartInfo[1].StartingOffset.QuadPart +
- PartEntry->PartInfo[1].PartitionLength.QuadPart) <
- (1024LL * 255LL * 63LL * 512LL))
- {
- PrevPartEntry->PartInfo[1].PartitionType = PARTITION_EXTENDED;
- }
- else
- {
- PrevPartEntry->PartInfo[1].PartitionType = PARTITION_XINT13_EXTENDED;
- }
-
- PrevPartEntry->PartInfo[1].BootIndicator = FALSE;
- PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
- }
-
- PartEntry->AutoCreate = AutoCreate;
- PartEntry->New = TRUE;
- PartEntry->Unpartitioned = FALSE;
- PartEntry->UnpartitionedOffset = 0ULL;
- PartEntry->UnpartitionedLength = 0ULL;
- }
- else
- {
- /* Insert an initialize a new partition entry */
- NewPartEntry = (PPARTENTRY)RtlAllocateHeap (ProcessHeap,
- 0,
- sizeof(PARTENTRY));
- if (NewPartEntry == NULL)
- return;
-
- RtlZeroMemory (NewPartEntry,
- sizeof(PARTENTRY));
-
- /* Insert the new entry into the list */
- InsertTailList (&PartEntry->ListEntry,
- &NewPartEntry->ListEntry);
-
- NewPartEntry->New = TRUE;
-
- NewPartEntry->FormatState = Unformatted;
- NewPartEntry->PartInfo[0].StartingOffset.QuadPart =
- PartEntry->UnpartitionedOffset + DiskEntry->TrackSize;
- NewPartEntry->PartInfo[0].HiddenSectors =
- (ULONG)(NewPartEntry->PartInfo[0].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
- NewPartEntry->PartInfo[0].PartitionLength.QuadPart =
- PartitionSize - DiskEntry->TrackSize;
- NewPartEntry->PartInfo[0].PartitionType = PARTITION_ENTRY_UNUSED;
- NewPartEntry->PartInfo[0].BootIndicator = FALSE; /* FIXME */
- NewPartEntry->PartInfo[0].RewritePartition = TRUE;
- NewPartEntry->PartInfo[1].RewritePartition = TRUE;
- NewPartEntry->PartInfo[2].RewritePartition = TRUE;
- NewPartEntry->PartInfo[3].RewritePartition = TRUE;
+ /* Current entry is the last entry */
+ RtlZeroMemory(&PrevPartEntry->PartInfo[1],
+ sizeof(PARTITION_INFORMATION));
+ PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
+ }
- /* Get previous and next partition entries */
- PrevPartEntry = GetPrevPartitionedEntry (DiskEntry,
- NewPartEntry);
- NextPartEntry = GetNextPartitionedEntry (DiskEntry,
- NewPartEntry);
- if (PrevPartEntry != NULL && NextPartEntry != NULL)
- {
- /* Current entry is in the middle of the list */
+ /* Adjust unpartitioned disk space entries */
- /* Copy previous container partition data to current entry */
- RtlCopyMemory (&NewPartEntry->PartInfo[1],
- &PrevPartEntry->PartInfo[1],
- sizeof(PARTITION_INFORMATION));
- NewPartEntry->PartInfo[1].RewritePartition = TRUE;
+ /* Get pointer to previous and next unpartitioned entries */
+ PrevPartEntry = GetPrevUnpartitionedEntry(DiskEntry,
+ PartEntry);
- /* Update previous container partition data */
+ NextPartEntry = GetNextUnpartitionedEntry(DiskEntry,
+ PartEntry);
+
+ if (PrevPartEntry != NULL && NextPartEntry != NULL)
+ {
+ /* Merge previous, current and next unpartitioned entry */
+
+ /* Adjust the previous entries length */
+ PrevPartEntry->UnpartitionedLength +=
+ (PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize +
+ NextPartEntry->UnpartitionedLength);
+
+ /* Remove the current entry */
+ RemoveEntryList(&PartEntry->ListEntry);
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ PartEntry);
+
+ /* Remove the next entry */
+ RemoveEntryList (&NextPartEntry->ListEntry);
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ NextPartEntry);
+
+ /* Update current partition */
+ List->CurrentPartition = PrevPartEntry;
+ }
+ else if (PrevPartEntry != NULL && NextPartEntry == NULL)
+ {
+ /* Merge current and previous unpartitioned entry */
- PrevPartEntry->PartInfo[1].StartingOffset.QuadPart =
- NewPartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
- PrevPartEntry->PartInfo[1].HiddenSectors =
- (ULONG)(PrevPartEntry->PartInfo[1].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
+ /* Adjust the previous entries length */
+ PrevPartEntry->UnpartitionedLength +=
+ (PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize);
- if (DiskEntry->PartListHead.Flink == &PrevPartEntry->ListEntry)
- {
- /* Special case - previous partition is first partition */
- PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
- DiskEntry->DiskSize - PrevPartEntry->PartInfo[1].StartingOffset.QuadPart;
- }
- else
- {
- PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
- NewPartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
- }
+ /* Remove the current entry */
+ RemoveEntryList(&PartEntry->ListEntry);
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ PartEntry);
- PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
+ /* Update current partition */
+ List->CurrentPartition = PrevPartEntry;
}
else if (PrevPartEntry == NULL && NextPartEntry != NULL)
{
- /* Current entry is the first entry */
- return;
+ /* Merge current and next unpartitioned entry */
+
+ /* Adjust the next entries offset and length */
+ NextPartEntry->UnpartitionedOffset =
+ PartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
+ NextPartEntry->UnpartitionedLength +=
+ (PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize);
+
+ /* Remove the current entry */
+ RemoveEntryList(&PartEntry->ListEntry);
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ PartEntry);
+
+ /* Update current partition */
+ List->CurrentPartition = NextPartEntry;
}
- else if (PrevPartEntry != NULL && NextPartEntry == NULL)
+ else
{
- /* Current entry is the last entry */
-
- PrevPartEntry->PartInfo[1].StartingOffset.QuadPart =
- NewPartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
- PrevPartEntry->PartInfo[1].HiddenSectors =
- (ULONG)(PrevPartEntry->PartInfo[1].StartingOffset.QuadPart / DiskEntry->BytesPerSector);
-
- if (DiskEntry->PartListHead.Flink == &PrevPartEntry->ListEntry)
- {
- /* Special case - previous partition is first partition */
- PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
- DiskEntry->DiskSize - PrevPartEntry->PartInfo[1].StartingOffset.QuadPart;
- }
- else
- {
- PrevPartEntry->PartInfo[1].PartitionLength.QuadPart =
- NewPartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
- }
-
- if ((PartEntry->PartInfo[1].StartingOffset.QuadPart +
- PartEntry->PartInfo[1].PartitionLength.QuadPart) <
- (1024LL * 255LL * 63LL * 512LL))
- {
- PrevPartEntry->PartInfo[1].PartitionType = PARTITION_EXTENDED;
- }
- else
- {
- PrevPartEntry->PartInfo[1].PartitionType = PARTITION_XINT13_EXTENDED;
- }
-
- PrevPartEntry->PartInfo[1].BootIndicator = FALSE;
- PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
- }
-
- /* Update offset and size of the remaining unpartitioned disk space */
- PartEntry->UnpartitionedOffset += PartitionSize;
- PartEntry->UnpartitionedLength -= PartitionSize;
- }
-
- DiskEntry->Modified = TRUE;
-
- UpdatePartitionNumbers (DiskEntry);
-
- AssignDriverLetters (List);
-}
+ /* Nothing to merge but change current entry */
+ PartEntry->New = FALSE;
+ PartEntry->Unpartitioned = TRUE;
+ PartEntry->UnpartitionedOffset =
+ PartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
+ PartEntry->UnpartitionedLength =
+ PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
+
+ /* Wipe the partition table */
+ RtlZeroMemory(&PartEntry->PartInfo,
+ sizeof(PartEntry->PartInfo));
+ }
+ DiskEntry->Modified = TRUE;
-VOID
-DeleteCurrentPartition (PPARTLIST List)
-{
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- PPARTENTRY PrevPartEntry;
- PPARTENTRY NextPartEntry;
-
- if (List == NULL ||
- List->CurrentDisk == NULL ||
- List->CurrentPartition == NULL ||
- List->CurrentPartition->Unpartitioned == TRUE)
- {
- return;
- }
-
- DiskEntry = List->CurrentDisk;
- PartEntry = List->CurrentPartition;
-
- /* Adjust container partition entries */
-
- /* Get previous and next partition entries */
- PrevPartEntry = GetPrevPartitionedEntry (DiskEntry,
- PartEntry);
- NextPartEntry = GetNextPartitionedEntry (DiskEntry,
- PartEntry);
-
- if (PrevPartEntry != NULL && NextPartEntry != NULL)
- {
- /* Current entry is in the middle of the list */
+ UpdatePartitionNumbers(DiskEntry);
- /*
- * The first extended partition can not be deleted
- * as long as other extended partitions are present.
- */
- if (PrevPartEntry->ListEntry.Blink == &DiskEntry->PartListHead)
- return;
-
- /* Copy previous container partition data to current entry */
- RtlCopyMemory (&PrevPartEntry->PartInfo[1],
- &PartEntry->PartInfo[1],
- sizeof(PARTITION_INFORMATION));
- PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
- }
- else if (PrevPartEntry == NULL && NextPartEntry != NULL)
- {
- /*
- * A primary partition can not be deleted as long as
- * extended partitions are present.
- */
- return;
- }
- else if (PrevPartEntry != NULL && NextPartEntry == NULL)
- {
- /* Current entry is the last entry */
- RtlZeroMemory (&PrevPartEntry->PartInfo[1],
- sizeof(PARTITION_INFORMATION));
- PrevPartEntry->PartInfo[1].RewritePartition = TRUE;
- }
-
-
- /* Adjust unpartitioned disk space entries */
-
- /* Get pointer to previous and next unpartitioned entries */
- PrevPartEntry = GetPrevUnpartitionedEntry (DiskEntry,
- PartEntry);
-
- NextPartEntry = GetNextUnpartitionedEntry (DiskEntry,
- PartEntry);
-
- if (PrevPartEntry != NULL && NextPartEntry != NULL)
- {
- /* Merge previous, current and next unpartitioned entry */
-
- /* Adjust the previous entries length */
- PrevPartEntry->UnpartitionedLength +=
- (PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize +
- NextPartEntry->UnpartitionedLength);
-
- /* Remove the current entry */
- RemoveEntryList (&PartEntry->ListEntry);
- RtlFreeHeap (ProcessHeap,
- 0,
- PartEntry);
-
- /* Remove the next entry */
- RemoveEntryList (&NextPartEntry->ListEntry);
- RtlFreeHeap (ProcessHeap,
- 0,
- NextPartEntry);
-
- /* Update current partition */
- List->CurrentPartition = PrevPartEntry;
- }
- else if (PrevPartEntry != NULL && NextPartEntry == NULL)
- {
- /* Merge current and previous unpartitioned entry */
-
- /* Adjust the previous entries length */
- PrevPartEntry->UnpartitionedLength +=
- (PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize);
-
- /* Remove the current entry */
- RemoveEntryList (&PartEntry->ListEntry);
- RtlFreeHeap (ProcessHeap,
- 0,
- PartEntry);
-
- /* Update current partition */
- List->CurrentPartition = PrevPartEntry;
- }
- else if (PrevPartEntry == NULL && NextPartEntry != NULL)
- {
- /* Merge current and next unpartitioned entry */
-
- /* Adjust the next entries offset and length */
- NextPartEntry->UnpartitionedOffset =
- PartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
- NextPartEntry->UnpartitionedLength +=
- (PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize);
-
- /* Remove the current entry */
- RemoveEntryList (&PartEntry->ListEntry);
- RtlFreeHeap (ProcessHeap,
- 0,
- PartEntry);
-
- /* Update current partition */
- List->CurrentPartition = NextPartEntry;
- }
- else
- {
- /* Nothing to merge but change current entry */
- PartEntry->New = FALSE;
- PartEntry->Unpartitioned = TRUE;
- PartEntry->UnpartitionedOffset =
- PartEntry->PartInfo[0].StartingOffset.QuadPart - DiskEntry->TrackSize;
- PartEntry->UnpartitionedLength =
- PartEntry->PartInfo[0].PartitionLength.QuadPart + DiskEntry->TrackSize;
-
- /* Wipe the partition table */
- RtlZeroMemory (&PartEntry->PartInfo,
- sizeof(PartEntry->PartInfo));
- }
-
- DiskEntry->Modified = TRUE;
-
- UpdatePartitionNumbers (DiskEntry);
-
- AssignDriverLetters (List);
+ AssignDriverLetters(List);
}
VOID
-CheckActiveBootPartition (PPARTLIST List)
+CheckActiveBootPartition(
+ PPARTLIST List)
{
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- PLIST_ENTRY ListEntry;
- UCHAR i;
-
- /* Check for empty disk list */
- if (IsListEmpty (&List->DiskListHead))
- {
- List->ActiveBootDisk = NULL;
- List->ActiveBootPartition = NULL;
- List->ActiveBootPartitionNumber = 0;
- return;
- }
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY ListEntry;
+ UCHAR i;
+
+ /* Check for empty disk list */
+ if (IsListEmpty (&List->DiskListHead))
+ {
+ List->ActiveBootDisk = NULL;
+ List->ActiveBootPartition = NULL;
+ List->ActiveBootPartitionNumber = 0;
+ return;
+ }
#if 0
- if (List->ActiveBootDisk != NULL &&
- List->ActiveBootPartition != NULL)
- {
- /* We already have an active boot partition */
- return;
- }
+ if (List->ActiveBootDisk != NULL &&
+ List->ActiveBootPartition != NULL)
+ {
+ /* We already have an active boot partition */
+ return;
+ }
#endif
- /* Choose the currently selected disk */
- DiskEntry = List->CurrentDisk;
-
- /* Check for empty partition list */
- if (IsListEmpty (&DiskEntry->PartListHead))
- {
- List->ActiveBootDisk = NULL;
- List->ActiveBootPartition = NULL;
- List->ActiveBootPartitionNumber = 0;
- return;
- }
-
- PartEntry = CONTAINING_RECORD (DiskEntry->PartListHead.Flink,
- PARTENTRY,
- ListEntry);
-
- /* Set active boot partition */
- if ((DiskEntry->NewDisk == TRUE) ||
- (PartEntry->PartInfo[0].BootIndicator == FALSE &&
- PartEntry->PartInfo[1].BootIndicator == FALSE &&
- PartEntry->PartInfo[2].BootIndicator == FALSE &&
- PartEntry->PartInfo[3].BootIndicator == FALSE))
- {
- PartEntry->PartInfo[0].BootIndicator = TRUE;
- PartEntry->PartInfo[0].RewritePartition = TRUE;
- DiskEntry->Modified = TRUE;
+ /* Choose the currently selected disk */
+ DiskEntry = List->CurrentDisk;
- /* FIXME: Might be incorrect if partitions were created by Linux FDISK */
- List->ActiveBootDisk = DiskEntry;
- List->ActiveBootPartition = PartEntry;
- List->ActiveBootPartitionNumber = 0;
-
- return;
- }
-
- /* Disk is not new, scan all partitions to find a bootable one */
- List->ActiveBootDisk = NULL;
- List->ActiveBootPartition = NULL;
- List->ActiveBootPartitionNumber = 0;
+ /* Check for empty partition list */
+ if (IsListEmpty (&DiskEntry->PartListHead))
+ {
+ List->ActiveBootDisk = NULL;
+ List->ActiveBootPartition = NULL;
+ List->ActiveBootPartitionNumber = 0;
+ return;
+ }
- ListEntry = DiskEntry->PartListHead.Flink;
- while (ListEntry != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD(ListEntry,
+ PartEntry = CONTAINING_RECORD(DiskEntry->PartListHead.Flink,
PARTENTRY,
ListEntry);
- /* Check if it's partitioned */
- if (!PartEntry->Unpartitioned)
+ /* Set active boot partition */
+ if ((DiskEntry->NewDisk == TRUE) ||
+ (PartEntry->PartInfo[0].BootIndicator == FALSE &&
+ PartEntry->PartInfo[1].BootIndicator == FALSE &&
+ PartEntry->PartInfo[2].BootIndicator == FALSE &&
+ PartEntry->PartInfo[3].BootIndicator == FALSE))
{
- /* Go through all of its 4 partitions */
- for (i=0; i<4; i++)
- {
- if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED &&
- PartEntry->PartInfo[i].BootIndicator)
- {
- /* Yes, we found it */
- List->ActiveBootDisk = DiskEntry;
- List->ActiveBootPartition = PartEntry;
- List->ActiveBootPartitionNumber = i;
+ PartEntry->PartInfo[0].BootIndicator = TRUE;
+ PartEntry->PartInfo[0].RewritePartition = TRUE;
+ DiskEntry->Modified = TRUE;
+
+ /* FIXME: Might be incorrect if partitions were created by Linux FDISK */
+ List->ActiveBootDisk = DiskEntry;
+ List->ActiveBootPartition = PartEntry;
+ List->ActiveBootPartitionNumber = 0;
+
+ return;
+ }
+
+ /* Disk is not new, scan all partitions to find a bootable one */
+ List->ActiveBootDisk = NULL;
+ List->ActiveBootPartition = NULL;
+ List->ActiveBootPartitionNumber = 0;
- DPRINT("Found bootable partition disk %d, drive letter %c\n",
- DiskEntry->DiskNumber, PartEntry->DriveLetter[i]);
+ ListEntry = DiskEntry->PartListHead.Flink;
+ while (ListEntry != &DiskEntry->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(ListEntry,
+ PARTENTRY,
+ ListEntry);
- break;
+ /* Check if it's partitioned */
+ if (!PartEntry->Unpartitioned)
+ {
+ /* Go through all of its 4 partitions */
+ for (i = 0; i < 4; i++)
+ {
+ if (PartEntry->PartInfo[i].PartitionType != PARTITION_ENTRY_UNUSED &&
+ PartEntry->PartInfo[i].BootIndicator)
+ {
+ /* Yes, we found it */
+ List->ActiveBootDisk = DiskEntry;
+ List->ActiveBootPartition = PartEntry;
+ List->ActiveBootPartitionNumber = i;
+
+ DPRINT("Found bootable partition disk %d, drive letter %c\n",
+ DiskEntry->DiskNumber, PartEntry->DriveLetter[i]);
+
+ break;
+ }
+ }
}
- }
+
+ /* Go to the next one */
+ ListEntry = ListEntry->Flink;
}
- /* Go to the next one */
- ListEntry = ListEntry->Flink;
- }
}
BOOLEAN
-CheckForLinuxFdiskPartitions (PPARTLIST List)
+CheckForLinuxFdiskPartitions(
+ PPARTLIST List)
{
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry1;
- PLIST_ENTRY Entry2;
- ULONG PartitionCount;
- ULONG i;
-
- Entry1 = List->DiskListHead.Flink;
- while (Entry1 != &List->DiskListHead)
- {
- DiskEntry = CONTAINING_RECORD (Entry1,
- DISKENTRY,
- ListEntry);
-
- Entry2 = DiskEntry->PartListHead.Flink;
- while (Entry2 != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry2,
- PARTENTRY,
- ListEntry);
-
- if (PartEntry->Unpartitioned == FALSE)
- {
- PartitionCount = 0;
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry1;
+ PLIST_ENTRY Entry2;
+ ULONG PartitionCount;
+ ULONG i;
+
+ Entry1 = List->DiskListHead.Flink;
+ while (Entry1 != &List->DiskListHead)
+ {
+ DiskEntry = CONTAINING_RECORD(Entry1,
+ DISKENTRY,
+ ListEntry);
- for (i = 0; i < 4; i++)
+ Entry2 = DiskEntry->PartListHead.Flink;
+ while (Entry2 != &DiskEntry->PartListHead)
{
- if (!IsContainerPartition (PartEntry->PartInfo[i].PartitionType) &&
- PartEntry->PartInfo[i].PartitionLength.QuadPart != 0ULL)
- {
- PartitionCount++;
- }
- }
+ PartEntry = CONTAINING_RECORD(Entry2,
+ PARTENTRY,
+ ListEntry);
- if (PartitionCount > 1)
- {
- return TRUE;
+ if (PartEntry->Unpartitioned == FALSE)
+ {
+ PartitionCount = 0;
+
+ for (i = 0; i < 4; i++)
+ {
+ if (!IsContainerPartition(PartEntry->PartInfo[i].PartitionType) &&
+ PartEntry->PartInfo[i].PartitionLength.QuadPart != 0ULL)
+ {
+ PartitionCount++;
+ }
+ }
+
+ if (PartitionCount > 1)
+ {
+ return TRUE;
+ }
+ }
+
+ Entry2 = Entry2->Flink;
}
- }
- Entry2 = Entry2->Flink;
+ Entry1 = Entry1->Flink;
}
- Entry1 = Entry1->Flink;
- }
-
- return FALSE;
+ return FALSE;
}
BOOLEAN
-WritePartitionsToDisk (PPARTLIST List)
+WritePartitionsToDisk(
+ PPARTLIST List)
{
- PDRIVE_LAYOUT_INFORMATION DriveLayout;
- OBJECT_ATTRIBUTES ObjectAttributes;
- IO_STATUS_BLOCK Iosb;
- WCHAR DstPath[MAX_PATH];
- UNICODE_STRING Name;
- HANDLE FileHandle;
- PDISKENTRY DiskEntry1;
- PDISKENTRY DiskEntry2;
- PPARTENTRY PartEntry;
- PLIST_ENTRY Entry1;
- PLIST_ENTRY Entry2;
- ULONG PartitionCount;
- ULONG DriveLayoutSize;
- ULONG Index;
- NTSTATUS Status;
-
- if (List == NULL)
- {
- return TRUE;
- }
-
- Entry1 = List->DiskListHead.Flink;
- while (Entry1 != &List->DiskListHead)
- {
- DiskEntry1 = CONTAINING_RECORD (Entry1,
- DISKENTRY,
- ListEntry);
-
- if (DiskEntry1->Modified == TRUE)
- {
- /* Count partitioned entries */
- PartitionCount = 0;
- Entry2 = DiskEntry1->PartListHead.Flink;
- while (Entry2 != &DiskEntry1->PartListHead)
- {
- PartEntry = CONTAINING_RECORD (Entry2,
- PARTENTRY,
- ListEntry);
- if (PartEntry->Unpartitioned == FALSE)
- {
- PartitionCount += 4;
- }
-
- Entry2 = Entry2->Flink;
- }
- if (PartitionCount == 0)
- {
- DriveLayoutSize = sizeof (DRIVE_LAYOUT_INFORMATION) +
- ((4 - 1) * sizeof (PARTITION_INFORMATION));
- }
- else
- {
- DriveLayoutSize = sizeof (DRIVE_LAYOUT_INFORMATION) +
- ((PartitionCount - 1) * sizeof (PARTITION_INFORMATION));
- }
- DriveLayout = (PDRIVE_LAYOUT_INFORMATION)RtlAllocateHeap (ProcessHeap,
- 0,
- DriveLayoutSize);
- if (DriveLayout == NULL)
- {
- DPRINT1 ("RtlAllocateHeap() failed\n");
- return FALSE;
- }
+ PDRIVE_LAYOUT_INFORMATION DriveLayout;
+ OBJECT_ATTRIBUTES ObjectAttributes;
+ IO_STATUS_BLOCK Iosb;
+ WCHAR DstPath[MAX_PATH];
+ UNICODE_STRING Name;
+ HANDLE FileHandle;
+ PDISKENTRY DiskEntry1;
+ PDISKENTRY DiskEntry2;
+ PPARTENTRY PartEntry;
+ PLIST_ENTRY Entry1;
+ PLIST_ENTRY Entry2;
+ ULONG PartitionCount;
+ ULONG DriveLayoutSize;
+ ULONG Index;
+ NTSTATUS Status;
+
+ if (List == NULL)
+ {
+ return TRUE;
+ }
- RtlZeroMemory (DriveLayout,
- DriveLayoutSize);
+ Entry1 = List->DiskListHead.Flink;
+ while (Entry1 != &List->DiskListHead)
+ {
+ DiskEntry1 = CONTAINING_RECORD(Entry1,
+ DISKENTRY,
+ ListEntry);
- if (PartitionCount == 0)
- {
- /* delete all partitions in the mbr */
- DriveLayout->PartitionCount = 4;
- for (Index = 0; Index < 4; Index++)
- {
- DriveLayout->PartitionEntry[Index].RewritePartition = TRUE;
- }
- }
- else
- {
- DriveLayout->PartitionCount = PartitionCount;
-
- Index = 0;
- Entry2 = DiskEntry1->PartListHead.Flink;
- while (Entry2 != &DiskEntry1->PartListHead)
+ if (DiskEntry1->Modified == TRUE)
{
- PartEntry = CONTAINING_RECORD (Entry2,
- PARTENTRY,
- ListEntry);
- if (PartEntry->Unpartitioned == FALSE)
- {
- RtlCopyMemory (&DriveLayout->PartitionEntry[Index],
- &PartEntry->PartInfo[0],
- 4 * sizeof (PARTITION_INFORMATION));
- Index += 4;
- }
-
- Entry2 = Entry2->Flink;
- }
- }
- if (DiskEntry1->Signature == 0)
- {
- LARGE_INTEGER SystemTime;
- TIME_FIELDS TimeFields;
- PUCHAR Buffer;
- Buffer = (PUCHAR)&DiskEntry1->Signature;
-
- while (1)
- {
- NtQuerySystemTime (&SystemTime);
- RtlTimeToTimeFields (&SystemTime, &TimeFields);
+ /* Count partitioned entries */
+ PartitionCount = 0;
- Buffer[0] = (UCHAR)(TimeFields.Year & 0xFF) + (UCHAR)(TimeFields.Hour & 0xFF);
- Buffer[1] = (UCHAR)(TimeFields.Year >> 8) + (UCHAR)(TimeFields.Minute & 0xFF);
- Buffer[2] = (UCHAR)(TimeFields.Month & 0xFF) + (UCHAR)(TimeFields.Second & 0xFF);
- Buffer[3] = (UCHAR)(TimeFields.Day & 0xFF) + (UCHAR)(TimeFields.Milliseconds & 0xFF);
+ Entry2 = DiskEntry1->PartListHead.Flink;
+ while (Entry2 != &DiskEntry1->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry2,
+ PARTENTRY,
+ ListEntry);
+ if (PartEntry->Unpartitioned == FALSE)
+ {
+ PartitionCount += 4;
+ }
+
+ Entry2 = Entry2->Flink;
+ }
- if (DiskEntry1->Signature == 0)
- {
- continue;
- }
-
- /* check if the signature already exist */
- /* FIXME:
- * Check also signatures from disks, which are
- * not visible (bootable) by the bios.
- */
- Entry2 = List->DiskListHead.Flink;
- while (Entry2 != &List->DiskListHead)
- {
- DiskEntry2 = CONTAINING_RECORD(Entry2, DISKENTRY, ListEntry);
- if (DiskEntry1 != DiskEntry2 &&
- DiskEntry1->Signature == DiskEntry2->Signature)
+ if (PartitionCount == 0)
{
- break;
+ DriveLayoutSize = sizeof (DRIVE_LAYOUT_INFORMATION) +
+ ((4 - 1) * sizeof (PARTITION_INFORMATION));
+ }
+ else
+ {
+ DriveLayoutSize = sizeof (DRIVE_LAYOUT_INFORMATION) +
+ ((PartitionCount - 1) * sizeof (PARTITION_INFORMATION));
}
- Entry2 = Entry2->Flink;
- }
- if (Entry2 == &List->DiskListHead)
- {
- break;
- }
- }
- /* set one partition entry to dirty, this will update the signature */
- DriveLayout->PartitionEntry[0].RewritePartition = TRUE;
+ DriveLayout = (PDRIVE_LAYOUT_INFORMATION)RtlAllocateHeap(ProcessHeap,
+ 0,
+ DriveLayoutSize);
+ if (DriveLayout == NULL)
+ {
+ DPRINT1("RtlAllocateHeap() failed\n");
+ return FALSE;
+ }
- }
+ RtlZeroMemory(DriveLayout,
+ DriveLayoutSize);
- DriveLayout->Signature = DiskEntry1->Signature;
+ if (PartitionCount == 0)
+ {
+ /* delete all partitions in the mbr */
+ DriveLayout->PartitionCount = 4;
+ for (Index = 0; Index < 4; Index++)
+ {
+ DriveLayout->PartitionEntry[Index].RewritePartition = TRUE;
+ }
+ }
+ else
+ {
+ DriveLayout->PartitionCount = PartitionCount;
+ Index = 0;
+
+ Entry2 = DiskEntry1->PartListHead.Flink;
+ while (Entry2 != &DiskEntry1->PartListHead)
+ {
+ PartEntry = CONTAINING_RECORD(Entry2,
+ PARTENTRY,
+ ListEntry);
+ if (PartEntry->Unpartitioned == FALSE)
+ {
+ RtlCopyMemory(&DriveLayout->PartitionEntry[Index],
+ &PartEntry->PartInfo[0],
+ 4 * sizeof (PARTITION_INFORMATION));
+ Index += 4;
+ }
+
+ Entry2 = Entry2->Flink;
+ }
+ }
+ if (DiskEntry1->Signature == 0)
+ {
+ LARGE_INTEGER SystemTime;
+ TIME_FIELDS TimeFields;
+ PUCHAR Buffer;
+ Buffer = (PUCHAR)&DiskEntry1->Signature;
+
+ while (1)
+ {
+ NtQuerySystemTime(&SystemTime);
+ RtlTimeToTimeFields(&SystemTime, &TimeFields);
+
+ Buffer[0] = (UCHAR)(TimeFields.Year & 0xFF) + (UCHAR)(TimeFields.Hour & 0xFF);
+ Buffer[1] = (UCHAR)(TimeFields.Year >> 8) + (UCHAR)(TimeFields.Minute & 0xFF);
+ Buffer[2] = (UCHAR)(TimeFields.Month & 0xFF) + (UCHAR)(TimeFields.Second & 0xFF);
+ Buffer[3] = (UCHAR)(TimeFields.Day & 0xFF) + (UCHAR)(TimeFields.Milliseconds & 0xFF);
+
+ if (DiskEntry1->Signature == 0)
+ {
+ continue;
+ }
+
+ /* check if the signature already exist */
+ /* FIXME:
+ * Check also signatures from disks, which are
+ * not visible (bootable) by the bios.
+ */
+ Entry2 = List->DiskListHead.Flink;
+ while (Entry2 != &List->DiskListHead)
+ {
+ DiskEntry2 = CONTAINING_RECORD(Entry2, DISKENTRY, ListEntry);
+ if (DiskEntry1 != DiskEntry2 &&
+ DiskEntry1->Signature == DiskEntry2->Signature)
+ {
+ break;
+ }
+
+ Entry2 = Entry2->Flink;
+ }
+
+ if (Entry2 == &List->DiskListHead)
+ {
+ break;
+ }
+ }
+
+ /* set one partition entry to dirty, this will update the signature */
+ DriveLayout->PartitionEntry[0].RewritePartition = TRUE;
+ }
- swprintf (DstPath,
- L"\\Device\\Harddisk%d\\Partition0",
- DiskEntry1->DiskNumber);
- RtlInitUnicodeString (&Name,
- DstPath);
- InitializeObjectAttributes (&ObjectAttributes,
- &Name,
- 0,
- NULL,
- NULL);
+ DriveLayout->Signature = DiskEntry1->Signature;
+
+ swprintf(DstPath,
+ L"\\Device\\Harddisk%d\\Partition0",
+ DiskEntry1->DiskNumber);
+ RtlInitUnicodeString(&Name,
+ DstPath);
+ InitializeObjectAttributes(&ObjectAttributes,
+ &Name,
+ 0,
+ NULL,
+ NULL);
+
+ Status = NtOpenFile(&FileHandle,
+ FILE_ALL_ACCESS,
+ &ObjectAttributes,
+ &Iosb,
+ 0,
+ FILE_SYNCHRONOUS_IO_NONALERT);
+ if (!NT_SUCCESS(Status))
+ {
+ DPRINT1("NtOpenFile() failed (Status %lx)\n", Status);
+ return FALSE;
+ }
- Status = NtOpenFile (&FileHandle,
- FILE_ALL_ACCESS,
- &ObjectAttributes,
- &Iosb,
- 0,
- FILE_SYNCHRONOUS_IO_NONALERT);
+ Status = NtDeviceIoControlFile(FileHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ IOCTL_DISK_SET_DRIVE_LAYOUT,
+ DriveLayout,
+ DriveLayoutSize,
+ NULL,
+ 0);
+ if (!NT_SUCCESS(Status))
+ {
+ DPRINT1("NtDeviceIoControlFile() failed (Status %lx)\n", Status);
+ NtClose(FileHandle);
+ return FALSE;
+ }
- if (!NT_SUCCESS (Status))
- {
- DPRINT1 ("NtOpenFile() failed (Status %lx)\n", Status);
- return FALSE;
- }
-
- Status = NtDeviceIoControlFile (FileHandle,
- NULL,
- NULL,
- NULL,
- &Iosb,
- IOCTL_DISK_SET_DRIVE_LAYOUT,
- DriveLayout,
- DriveLayoutSize,
- NULL,
- 0);
- if (!NT_SUCCESS (Status))
- {
- DPRINT1 ("NtDeviceIoControlFile() failed (Status %lx)\n", Status);
- NtClose (FileHandle);
- return FALSE;
- }
+ RtlFreeHeap(ProcessHeap,
+ 0,
+ DriveLayout);
- RtlFreeHeap (ProcessHeap,
- 0,
- DriveLayout);
+ NtClose(FileHandle);
+ }
- NtClose (FileHandle);
+ Entry1 = Entry1->Flink;
}
- Entry1 = Entry1->Flink;
- }
-
- return TRUE;
+ return TRUE;
}
-BOOL SetMountedDeviceValues(PPARTLIST List)
+
+BOOL
+SetMountedDeviceValues(
+ PPARTLIST List)
{
- PLIST_ENTRY Entry1, Entry2;
- PDISKENTRY DiskEntry;
- PPARTENTRY PartEntry;
- UCHAR i;
+ PLIST_ENTRY Entry1, Entry2;
+ PDISKENTRY DiskEntry;
+ PPARTENTRY PartEntry;
+ UCHAR i;
- if (List == NULL)
- {
- return FALSE;
- }
-
- Entry1 = List->DiskListHead.Flink;
- while (Entry1 != &List->DiskListHead)
- {
- DiskEntry = CONTAINING_RECORD (Entry1,
- DISKENTRY,
- ListEntry);
-
- Entry2 = DiskEntry->PartListHead.Flink;
- while (Entry2 != &DiskEntry->PartListHead)
- {
- PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
- if (!PartEntry->Unpartitioned)
- {
- for (i=0; i<4; i++)
+ if (List == NULL)
+ {
+ return FALSE;
+ }
+
+ Entry1 = List->DiskListHead.Flink;
+ while (Entry1 != &List->DiskListHead)
+ {
+ DiskEntry = CONTAINING_RECORD(Entry1,
+ DISKENTRY,
+ ListEntry);
+
+ Entry2 = DiskEntry->PartListHead.Flink;
+ while (Entry2 != &DiskEntry->PartListHead)
{
- if (PartEntry->DriveLetter[i])
- {
- if (!SetMountedDeviceValue(PartEntry->DriveLetter[i], DiskEntry->Signature, PartEntry->PartInfo[i].StartingOffset))
+ PartEntry = CONTAINING_RECORD(Entry2, PARTENTRY, ListEntry);
+ if (!PartEntry->Unpartitioned)
{
- return FALSE;
+ for (i = 0; i < 4; i++)
+ {
+ if (PartEntry->DriveLetter[i])
+ {
+ if (!SetMountedDeviceValue(PartEntry->DriveLetter[i],
+ DiskEntry->Signature,
+ PartEntry->PartInfo[i].StartingOffset))
+ {
+ return FALSE;
+ }
+ }
+ }
}
- }
+
+ Entry2 = Entry2->Flink;
}
- }
- Entry2 = Entry2->Flink;
- }
- Entry1 = Entry1->Flink;
- }
- return TRUE;
-}
+ Entry1 = Entry1->Flink;
+ }
+ return TRUE;
+}
/* EOF */
projects / reactos.git / commitdiff