mirror of
https://github.com/reactos/reactos.git
synced 2024-08-30 23:28:58 +00:00
209 lines
5.7 KiB
C
209 lines
5.7 KiB
C
/*
|
|
* PROJECT: ReactOS DiskPart
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: base/system/diskpart/delete.c
|
|
* PURPOSE: Manages all the partitions of the OS in an interactive way.
|
|
* PROGRAMMERS: Lee Schroeder
|
|
*/
|
|
|
|
#include "diskpart.h"
|
|
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
static
|
|
BOOL
|
|
IsKnownPartition(
|
|
_In_ PPARTENTRY PartEntry)
|
|
{
|
|
if (IsRecognizedPartition(PartEntry->PartitionType) ||
|
|
IsContainerPartition(PartEntry->PartitionType))
|
|
return TRUE;
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
DeleteDisk(
|
|
_In_ INT argc,
|
|
_In_ PWSTR *argv)
|
|
{
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
DeletePartition(
|
|
_In_ INT argc,
|
|
_In_ PWSTR *argv)
|
|
{
|
|
PPARTENTRY PrevPartEntry;
|
|
PPARTENTRY NextPartEntry;
|
|
PPARTENTRY LogicalPartEntry;
|
|
PLIST_ENTRY Entry;
|
|
INT i;
|
|
BOOL bOverride = FALSE;
|
|
NTSTATUS Status;
|
|
|
|
DPRINT("DeletePartition()\n");
|
|
|
|
if (CurrentDisk == NULL)
|
|
{
|
|
ConResPuts(StdOut, IDS_SELECT_NO_DISK);
|
|
return TRUE;
|
|
}
|
|
|
|
if (CurrentPartition == NULL)
|
|
{
|
|
ConResPuts(StdOut, IDS_SELECT_NO_PARTITION);
|
|
return TRUE;
|
|
}
|
|
|
|
ASSERT(CurrentPartition->PartitionType != PARTITION_ENTRY_UNUSED);
|
|
|
|
for (i = 2; i < argc; i++)
|
|
{
|
|
if (_wcsicmp(argv[i], L"noerr") == 0)
|
|
{
|
|
/* noerr */
|
|
DPRINT("NOERR\n");
|
|
ConPuts(StdOut, L"The NOERR option is not supported yet!\n");
|
|
#if 0
|
|
bNoErr = TRUE;
|
|
#endif
|
|
}
|
|
else if (_wcsicmp(argv[i], L"override") == 0)
|
|
{
|
|
/* noerr */
|
|
DPRINT("OVERRIDE\n");
|
|
bOverride = TRUE;
|
|
}
|
|
else
|
|
{
|
|
ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS);
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
|
|
/* Clear the system partition if it is being deleted */
|
|
#if 0
|
|
if (List->SystemPartition == PartEntry)
|
|
{
|
|
ASSERT(List->SystemPartition);
|
|
List->SystemPartition = NULL;
|
|
}
|
|
#endif
|
|
|
|
if ((bOverride == FALSE) && (IsKnownPartition(CurrentPartition) == FALSE))
|
|
{
|
|
ConResPuts(StdOut, IDS_DELETE_PARTITION_FAIL);
|
|
return FALSE;
|
|
}
|
|
|
|
/* Check which type of partition (primary/logical or extended) is being deleted */
|
|
if (CurrentDisk->ExtendedPartition == CurrentPartition)
|
|
{
|
|
/* An extended partition is being deleted: delete all logical partition entries */
|
|
while (!IsListEmpty(&CurrentDisk->LogicalPartListHead))
|
|
{
|
|
Entry = RemoveHeadList(&CurrentDisk->LogicalPartListHead);
|
|
LogicalPartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);
|
|
|
|
/* Dismount the logical partition */
|
|
DismountVolume(LogicalPartEntry);
|
|
|
|
/* Delete it */
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, LogicalPartEntry);
|
|
}
|
|
|
|
CurrentDisk->ExtendedPartition = NULL;
|
|
}
|
|
else
|
|
{
|
|
/* A primary partition is being deleted: dismount it */
|
|
DismountVolume(CurrentPartition);
|
|
}
|
|
|
|
/* Adjust the unpartitioned disk space entries */
|
|
|
|
/* Get pointer to previous and next unpartitioned entries */
|
|
PrevPartEntry = GetPrevUnpartitionedEntry(CurrentPartition);
|
|
NextPartEntry = GetNextUnpartitionedEntry(CurrentPartition);
|
|
|
|
if (PrevPartEntry != NULL && NextPartEntry != NULL)
|
|
{
|
|
/* Merge the previous, current and next unpartitioned entries */
|
|
|
|
/* Adjust the previous entry length */
|
|
PrevPartEntry->SectorCount.QuadPart += (CurrentPartition->SectorCount.QuadPart + NextPartEntry->SectorCount.QuadPart);
|
|
|
|
/* Remove the current and next entries */
|
|
RemoveEntryList(&CurrentPartition->ListEntry);
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, CurrentPartition);
|
|
RemoveEntryList(&NextPartEntry->ListEntry);
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, NextPartEntry);
|
|
}
|
|
else if (PrevPartEntry != NULL && NextPartEntry == NULL)
|
|
{
|
|
/* Merge the current and the previous unpartitioned entries */
|
|
|
|
/* Adjust the previous entry length */
|
|
PrevPartEntry->SectorCount.QuadPart += CurrentPartition->SectorCount.QuadPart;
|
|
|
|
/* Remove the current entry */
|
|
RemoveEntryList(&CurrentPartition->ListEntry);
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, CurrentPartition);
|
|
}
|
|
else if (PrevPartEntry == NULL && NextPartEntry != NULL)
|
|
{
|
|
/* Merge the current and the next unpartitioned entries */
|
|
|
|
/* Adjust the next entry offset and length */
|
|
NextPartEntry->StartSector.QuadPart = CurrentPartition->StartSector.QuadPart;
|
|
NextPartEntry->SectorCount.QuadPart += CurrentPartition->SectorCount.QuadPart;
|
|
|
|
/* Remove the current entry */
|
|
RemoveEntryList(&CurrentPartition->ListEntry);
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, CurrentPartition);
|
|
}
|
|
else
|
|
{
|
|
/* Nothing to merge but change the current entry */
|
|
CurrentPartition->IsPartitioned = FALSE;
|
|
CurrentPartition->OnDiskPartitionNumber = 0;
|
|
CurrentPartition->PartitionNumber = 0;
|
|
// CurrentPartition->PartitionIndex = 0;
|
|
CurrentPartition->BootIndicator = FALSE;
|
|
CurrentPartition->PartitionType = PARTITION_ENTRY_UNUSED;
|
|
CurrentPartition->FormatState = Unformatted;
|
|
CurrentPartition->FileSystemName[0] = L'\0';
|
|
CurrentPartition->DriveLetter = 0;
|
|
RtlZeroMemory(CurrentPartition->VolumeLabel, sizeof(CurrentPartition->VolumeLabel));
|
|
}
|
|
|
|
CurrentPartition = NULL;
|
|
|
|
UpdateDiskLayout(CurrentDisk);
|
|
Status = WritePartitions(CurrentDisk);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
ConResPuts(StdOut, IDS_DELETE_PARTITION_FAIL);
|
|
return TRUE;
|
|
}
|
|
|
|
ConResPuts(StdOut, IDS_DELETE_PARTITION_SUCCESS);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
DeleteVolume(
|
|
_In_ INT argc,
|
|
_In_ PWSTR *argv)
|
|
{
|
|
return TRUE;
|
|
}
|