Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292/* * PROJECT: ReactOS DiskPart * LICENSE: GPL - See COPYING in the top level directory * FILE: base/system/diskpart/select.c * PURPOSE: Manages all the partitions of the OS in an interactive way. * PROGRAMMERS: Lee Schroeder */
#include "diskpart.h"
#define NDEBUG#include <debug.h>
/* FUNCTIONS ******************************************************************/
EXIT_CODESelectDisk( _In_ INT argc, _In_ PWSTR *argv){ PLIST_ENTRY Entry; PDISKENTRY DiskEntry; ULONG ulValue;
DPRINT("Select Disk()\n");
if (argc > 3) { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
if (argc == 2) { if (CurrentDisk == NULL) ConResPuts(StdOut, IDS_SELECT_NO_DISK); else ConResPrintf(StdOut, IDS_SELECT_DISK, CurrentDisk->DiskNumber); return EXIT_SUCCESS; }
if (!_wcsicmp(argv[2], L"system")) { CurrentDisk = NULL;
Entry = DiskListHead.Flink; DiskEntry = CONTAINING_RECORD(Entry, DISKENTRY, ListEntry);
CurrentDisk = DiskEntry; CurrentPartition = NULL; ConResPrintf(StdOut, IDS_SELECT_DISK, CurrentDisk->DiskNumber); return EXIT_SUCCESS; } else if (!_wcsicmp(argv[2], L"next")) { if (CurrentDisk == NULL) { CurrentPartition = NULL; ConResPuts(StdErr, IDS_SELECT_DISK_ENUM_NO_START); return EXIT_SUCCESS; }
if (CurrentDisk->ListEntry.Flink == &DiskListHead) { CurrentDisk = NULL; CurrentPartition = NULL; ConResPuts(StdErr, IDS_SELECT_DISK_ENUM_FINISHED); return EXIT_SUCCESS; }
Entry = CurrentDisk->ListEntry.Flink;
DiskEntry = CONTAINING_RECORD(Entry, DISKENTRY, ListEntry);
CurrentDisk = DiskEntry; CurrentPartition = NULL; ConResPrintf(StdOut, IDS_SELECT_DISK, CurrentDisk->DiskNumber); return EXIT_SUCCESS; } else if (IsDecString(argv[2])) { ulValue = wcstoul(argv[2], NULL, 10); if ((ulValue == 0) && (errno == ERANGE)) { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
CurrentDisk = NULL;
Entry = DiskListHead.Flink; while (Entry != &DiskListHead) { DiskEntry = CONTAINING_RECORD(Entry, DISKENTRY, ListEntry);
if (DiskEntry->DiskNumber == ulValue) { CurrentDisk = DiskEntry; CurrentPartition = NULL; ConResPrintf(StdOut, IDS_SELECT_DISK, CurrentDisk->DiskNumber); return EXIT_SUCCESS; }
Entry = Entry->Flink; } } else { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
ConResPuts(StdErr, IDS_SELECT_DISK_INVALID); return EXIT_SUCCESS;}
EXIT_CODESelectPartition( _In_ INT argc, _In_ PWSTR *argv){ PLIST_ENTRY Entry; PPARTENTRY PartEntry; ULONG ulValue; ULONG PartNumber = 1;
DPRINT("Select Partition()\n");
if (argc > 3) { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
if (CurrentDisk == NULL) { ConResPuts(StdOut, IDS_SELECT_PARTITION_NO_DISK); return EXIT_SUCCESS; }
if (argc == 2) { if (CurrentPartition == NULL) ConResPuts(StdOut, IDS_SELECT_NO_PARTITION); else ConResPrintf(StdOut, IDS_SELECT_PARTITION, CurrentPartition); return EXIT_SUCCESS; }
if (!IsDecString(argv[2])) { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
ulValue = wcstoul(argv[2], NULL, 10); if ((ulValue == 0) && (errno == ERANGE)) { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
if (CurrentDisk->PartitionStyle == PARTITION_STYLE_MBR) { Entry = CurrentDisk->PrimaryPartListHead.Flink; while (Entry != &CurrentDisk->PrimaryPartListHead) { PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);
if (PartEntry->Mbr.PartitionType != PARTITION_ENTRY_UNUSED) { if (PartNumber == ulValue) { CurrentPartition = PartEntry; ConResPrintf(StdOut, IDS_SELECT_PARTITION, PartNumber); return EXIT_SUCCESS; }
PartNumber++; }
Entry = Entry->Flink; }
Entry = CurrentDisk->LogicalPartListHead.Flink; while (Entry != &CurrentDisk->LogicalPartListHead) { PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);
if (PartEntry->Mbr.PartitionType != PARTITION_ENTRY_UNUSED) { if (PartNumber == ulValue) { CurrentPartition = PartEntry; ConResPrintf(StdOut, IDS_SELECT_PARTITION, PartNumber); return EXIT_SUCCESS; }
PartNumber++; } Entry = Entry->Flink; } } else if (CurrentDisk->PartitionStyle == PARTITION_STYLE_GPT) { Entry = CurrentDisk->PrimaryPartListHead.Flink; while (Entry != &CurrentDisk->PrimaryPartListHead) { PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);
if (!IsEqualGUID(&PartEntry->Gpt.PartitionType, &PARTITION_ENTRY_UNUSED_GUID)) { if (PartNumber == ulValue) { CurrentPartition = PartEntry; ConResPrintf(StdOut, IDS_SELECT_PARTITION, PartNumber); return EXIT_SUCCESS; }
PartNumber++; }
Entry = Entry->Flink; } }
ConResPuts(StdErr, IDS_SELECT_PARTITION_INVALID); return EXIT_SUCCESS;}
EXIT_CODESelectVolume( _In_ INT argc, _In_ PWSTR *argv){ PLIST_ENTRY Entry; PVOLENTRY VolumeEntry; ULONG ulValue;
DPRINT("SelectVolume()\n");
if (argc > 3) { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
if (argc == 2) { if (CurrentDisk == NULL) ConResPuts(StdOut, IDS_SELECT_NO_VOLUME); else ConResPrintf(StdOut, IDS_SELECT_VOLUME, CurrentVolume->VolumeNumber); return EXIT_SUCCESS; }
if (!IsDecString(argv[2])) { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
ulValue = wcstoul(argv[2], NULL, 10); if ((ulValue == 0) && (errno == ERANGE)) { ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS); return EXIT_SUCCESS; }
CurrentVolume = NULL;
Entry = VolumeListHead.Flink; while (Entry != &VolumeListHead) { VolumeEntry = CONTAINING_RECORD(Entry, VOLENTRY, ListEntry);
if (VolumeEntry->VolumeNumber == ulValue) { CurrentVolume = VolumeEntry; ConResPrintf(StdOut, IDS_SELECT_VOLUME, CurrentVolume->VolumeNumber); return TRUE; }
Entry = Entry->Flink; }
ConResPuts(StdErr, IDS_SELECT_VOLUME_INVALID); return EXIT_SUCCESS;}