mirror of
https://github.com/reactos/reactos.git
synced 2024-10-31 03:48:17 +00:00
a2aaa28436
Select a disk and run "dump disk 0" to dump the MBR of the disk.
279 lines
6.3 KiB
C
279 lines
6.3 KiB
C
/*
|
|
* PROJECT: ReactOS DiskPart
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: base/system/diskpart/dump.c
|
|
* PURPOSE: Manages all the partitions of the OS in an interactive way.
|
|
* PROGRAMMERS: Eric Kohl
|
|
*/
|
|
|
|
#include "diskpart.h"
|
|
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
/* FUNCTIONS ******************************************************************/
|
|
|
|
static
|
|
VOID
|
|
HexDump(
|
|
_In_ UCHAR *addr,
|
|
_In_ int len)
|
|
{
|
|
WCHAR Buffer[17];
|
|
UCHAR *pc;
|
|
int i;
|
|
|
|
Buffer[16] = L'\0';
|
|
|
|
pc = addr;
|
|
for (i = 0; i < len; i++)
|
|
{
|
|
if ((i % 16) == 0)
|
|
ConPrintf(StdOut, L" %04x ", i);
|
|
|
|
ConPrintf(StdOut, L" %02x", pc[i]);
|
|
|
|
if ((pc[i] < 0x20) || (pc[i] > 0x7e))
|
|
Buffer[i % 16] = L'.';
|
|
else
|
|
Buffer[i % 16] = (WCHAR)(USHORT)pc[i];
|
|
|
|
if ((i % 16) == (16 - 1))
|
|
ConPrintf(StdOut, L" %s\n", Buffer);
|
|
}
|
|
}
|
|
|
|
|
|
static
|
|
VOID
|
|
DumpDisk(
|
|
_In_ INT argc,
|
|
_In_ LPWSTR *argv)
|
|
{
|
|
OBJECT_ATTRIBUTES ObjectAttributes;
|
|
IO_STATUS_BLOCK Iosb;
|
|
NTSTATUS Status;
|
|
WCHAR Buffer[MAX_PATH];
|
|
UNICODE_STRING Name;
|
|
HANDLE FileHandle = NULL;
|
|
PUCHAR pSectorBuffer = NULL;
|
|
LARGE_INTEGER FileOffset;
|
|
LONGLONG Sector;
|
|
LPWSTR endptr = NULL;
|
|
|
|
#if 0
|
|
if (argc == 2)
|
|
{
|
|
ConResPuts(StdOut, IDS_HELP_CMD_DUMP_DISK);
|
|
return TRUE;
|
|
}
|
|
#endif
|
|
|
|
if (CurrentDisk == NULL)
|
|
{
|
|
ConResPuts(StdOut, IDS_SELECT_NO_DISK);
|
|
return;
|
|
}
|
|
|
|
Sector = _wcstoi64(argv[2], &endptr, 0);
|
|
if (((Sector == 0) && (endptr == argv[2])) ||
|
|
(Sector < 0))
|
|
{
|
|
ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS);
|
|
return;
|
|
}
|
|
|
|
pSectorBuffer = RtlAllocateHeap(RtlGetProcessHeap(), HEAP_ZERO_MEMORY, CurrentDisk->BytesPerSector);
|
|
if (pSectorBuffer == NULL)
|
|
{
|
|
DPRINT1("\n");
|
|
/* Error message */
|
|
goto done;
|
|
}
|
|
|
|
swprintf(Buffer,
|
|
L"\\Device\\Harddisk%d\\Partition0",
|
|
CurrentDisk->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))
|
|
{
|
|
DPRINT1("\n");
|
|
goto done;
|
|
}
|
|
|
|
FileOffset.QuadPart = Sector * CurrentDisk->BytesPerSector;
|
|
Status = NtReadFile(FileHandle,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&Iosb,
|
|
(PVOID)pSectorBuffer,
|
|
CurrentDisk->BytesPerSector,
|
|
&FileOffset,
|
|
NULL);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT1("NtReadFile failed, status=%x\n", Status);
|
|
goto done;
|
|
}
|
|
|
|
HexDump(pSectorBuffer, CurrentDisk->BytesPerSector);
|
|
|
|
done:
|
|
if (FileHandle)
|
|
NtClose(FileHandle);
|
|
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, pSectorBuffer);
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
static
|
|
VOID
|
|
DumpPartition(
|
|
_In_ INT argc,
|
|
_In_ LPWSTR *argv)
|
|
{
|
|
OBJECT_ATTRIBUTES ObjectAttributes;
|
|
IO_STATUS_BLOCK Iosb;
|
|
NTSTATUS Status;
|
|
WCHAR Buffer[MAX_PATH];
|
|
UNICODE_STRING Name;
|
|
HANDLE FileHandle = NULL;
|
|
PUCHAR pSectorBuffer = NULL;
|
|
LARGE_INTEGER FileOffset;
|
|
LONGLONG Sector;
|
|
LPWSTR endptr = NULL;
|
|
|
|
|
|
#if 0
|
|
if (argc == 2)
|
|
{
|
|
ConResPuts(StdOut, IDS_HELP_CMD_DUMP_DISK);
|
|
return TRUE;
|
|
}
|
|
#endif
|
|
|
|
if (CurrentDisk == NULL)
|
|
{
|
|
ConResPuts(StdOut, IDS_SELECT_NO_DISK);
|
|
return;
|
|
}
|
|
|
|
if (CurrentPartition == NULL)
|
|
{
|
|
ConResPuts(StdOut, IDS_SELECT_NO_PARTITION);
|
|
return;
|
|
}
|
|
|
|
Sector = _wcstoi64(argv[2], &endptr, 0);
|
|
if (((Sector == 0) && (endptr == argv[2])) ||
|
|
(Sector < 0))
|
|
{
|
|
ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS);
|
|
return;
|
|
}
|
|
|
|
pSectorBuffer = RtlAllocateHeap(RtlGetProcessHeap(), HEAP_ZERO_MEMORY, CurrentDisk->BytesPerSector);
|
|
if (pSectorBuffer == NULL)
|
|
{
|
|
DPRINT1("\n");
|
|
/* Error message */
|
|
goto done;
|
|
}
|
|
|
|
swprintf(Buffer,
|
|
L"\\Device\\Harddisk%d\\Partition%d",
|
|
CurrentDisk->DiskNumber,
|
|
CurrentPartition->PartitionNumber);
|
|
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))
|
|
{
|
|
DPRINT1("\n");
|
|
goto done;
|
|
}
|
|
|
|
FileOffset.QuadPart = Sector * CurrentDisk->BytesPerSector;
|
|
Status = NtReadFile(FileHandle,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
&Iosb,
|
|
(PVOID)pSectorBuffer,
|
|
CurrentDisk->BytesPerSector,
|
|
&FileOffset,
|
|
NULL);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT1("NtReadFile failed, status=%x\n", Status);
|
|
goto done;
|
|
}
|
|
|
|
HexDump(pSectorBuffer, CurrentDisk->BytesPerSector);
|
|
|
|
done:
|
|
if (FileHandle)
|
|
NtClose(FileHandle);
|
|
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, pSectorBuffer);
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
BOOL
|
|
dump_main(
|
|
_In_ INT argc,
|
|
_In_ LPWSTR *argv)
|
|
{
|
|
/* gets the first word from the string */
|
|
#if 0
|
|
if (argc == 1)
|
|
{
|
|
ConResPuts(StdOut, IDS_HELP_CMD_LIST);
|
|
return TRUE;
|
|
}
|
|
#endif
|
|
|
|
/* determines which to list (disk, partition, etc.) */
|
|
if (!wcsicmp(argv[1], L"disk"))
|
|
DumpDisk(argc, argv);
|
|
else if (!wcsicmp(argv[1], L"partition"))
|
|
DumpPartition(argc, argv);
|
|
#if 0
|
|
else
|
|
ConResPuts(StdOut, IDS_HELP_CMD_LIST);
|
|
#endif
|
|
|
|
return TRUE;
|
|
}
|