convert.c File Reference

#include "diskpart.h"
#include <debug.h>

Include dependency graph for convert.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
NTSTATUS	CreateDisk (_In_ ULONG DiskNumber, _In_ PCREATE_DISK DiskInfo)
BOOL	ConvertGPT (_In_ INT argc, _In_ PWSTR *argv)
BOOL	ConvertMBR (_In_ INT argc, _In_ PWSTR *argv)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 11 of file convert.c.

Function Documentation

ConvertGPT()

BOOL ConvertGPT	(	_In_ INT	argc,
		_In_ PWSTR *	argv
	)

Definition at line 88 of file convert.c.

{
    CREATE_DISK DiskInfo;
    NTSTATUS Status;
 
    DPRINT("ConvertGPT()\n");
 
    if (CurrentDisk == NULL)
    {
        ConResPuts(StdOut, IDS_SELECT_NO_DISK);
        return TRUE;
    }
 
    if (CurrentDisk->PartitionStyle == PARTITION_STYLE_GPT)
    {
        ConResPuts(StdOut, IDS_CONVERT_GPT_ALREADY);
        return TRUE;
    }
 
    if (GetPrimaryPartitionCount(CurrentDisk) != 0)
    {
        ConResPuts(StdOut, IDS_CONVERT_GPT_NOT_EMPTY);
        return TRUE;
    }
 
#if 0
    /* Fail if disk size is less than 128MB */
    if (CurrentDisk->SectorCount.QuadPart * CurrentDisk->BytesPerSector < 128ULL * 1024ULL * 1024ULL)
    {
        ConResPuts(StdOut, IDS_CONVERT_GPT_TOO_SMALL);
        return TRUE;
    }
#endif
 
    DiskInfo.PartitionStyle = PARTITION_STYLE_GPT;
    CreateGUID(&DiskInfo.Gpt.DiskId);
    DiskInfo.Gpt.MaxPartitionCount = 128;
 
    Status = CreateDisk(CurrentDisk->DiskNumber, &DiskInfo);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("CreateDisk() failed!\n");
        return TRUE;
    }
 
    CurrentDisk->StartSector.QuadPart = AlignDown(CurrentDisk->LayoutBuffer->Gpt.StartingUsableOffset.QuadPart / CurrentDisk->BytesPerSector,
                                                  CurrentDisk->SectorAlignment) + (ULONGLONG)CurrentDisk->SectorAlignment;
    CurrentDisk->EndSector.QuadPart = AlignDown(CurrentDisk->StartSector.QuadPart + (CurrentDisk->LayoutBuffer->Gpt.UsableLength.QuadPart / CurrentDisk->BytesPerSector) - 1,
                                                CurrentDisk->SectorAlignment);
 
    ScanForUnpartitionedGptDiskSpace(CurrentDisk);
    ConResPuts(StdOut, IDS_CONVERT_GPT_SUCCESS);
 
    return TRUE;
}

ConvertMBR()

BOOL ConvertMBR	(	_In_ INT	argc,
		_In_ PWSTR *	argv
	)

Definition at line 148 of file convert.c.

{
    CREATE_DISK DiskInfo;
    NTSTATUS Status;
 
    DPRINT("ConvertMBR()\n");
 
    if (CurrentDisk == NULL)
    {
        ConResPuts(StdOut, IDS_SELECT_NO_DISK);
        return TRUE;
    }
 
    if ((CurrentDisk->PartitionStyle == PARTITION_STYLE_MBR) ||
        (CurrentDisk->PartitionStyle == PARTITION_STYLE_RAW))
    {
        ConResPuts(StdOut, IDS_CONVERT_MBR_ALREADY);
        return TRUE;
    }
 
    if (GetPrimaryPartitionCount(CurrentDisk) != 0)
    {
        ConResPuts(StdOut, IDS_CONVERT_MBR_NOT_EMPTY);
        return TRUE;
    }
 
    DiskInfo.PartitionStyle = PARTITION_STYLE_MBR;
    CreateSignature(&DiskInfo.Mbr.Signature);
 
    Status = CreateDisk(CurrentDisk->DiskNumber, &DiskInfo);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("CreateDisk() failed!\n");
        return TRUE;
    }
 
    CurrentDisk->StartSector.QuadPart = (ULONGLONG)CurrentDisk->SectorAlignment;
    CurrentDisk->EndSector.QuadPart = min(CurrentDisk->SectorCount.QuadPart, 0x100000000) - 1;
 
    ScanForUnpartitionedMbrDiskSpace(CurrentDisk);
    ConResPuts(StdOut, IDS_CONVERT_MBR_SUCCESS);
 
    return TRUE;
}

CreateDisk()

NTSTATUS CreateDisk	(	_In_ ULONG	DiskNumber,
		_In_ PCREATE_DISK	DiskInfo
	)

Definition at line 15 of file convert.c.

{
    WCHAR DstPath[MAX_PATH];
    OBJECT_ATTRIBUTES ObjectAttributes;
    UNICODE_STRING Name;
    HANDLE FileHandle = NULL;
    IO_STATUS_BLOCK Iosb;
    NTSTATUS Status;
 
    DPRINT("CreateDisk(%lu %p)\n", DiskNumber, DiskInfo);
 
    StringCchPrintfW(DstPath, ARRAYSIZE(DstPath),
                     L"\\Device\\Harddisk%lu\\Partition0",
                     CurrentDisk->DiskNumber);
    RtlInitUnicodeString(&Name, DstPath);
 
    InitializeObjectAttributes(&ObjectAttributes,
                               &Name,
                               OBJ_CASE_INSENSITIVE,
                               NULL,
                               NULL);
 
    Status = NtOpenFile(&FileHandle,
                        GENERIC_READ | GENERIC_WRITE | SYNCHRONIZE,
                        &ObjectAttributes,
                        &Iosb,
                        0,
                        FILE_SYNCHRONOUS_IO_NONALERT);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("NtOpenFile() failed (Status %lx)\n", Status);
        goto done;
    }
 
    Status = NtDeviceIoControlFile(FileHandle,
                                   NULL,
                                   NULL,
                                   NULL,
                                   &Iosb,
                                   IOCTL_DISK_CREATE_DISK,
                                   DiskInfo,
                                   sizeof(*DiskInfo),
                                   NULL,
                                   0);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("NtDeviceIoControlFile() failed (Status %lx)\n", Status);
        goto done;
    }
 
    /* Free the layout buffer */
    if (CurrentDisk->LayoutBuffer)
        RtlFreeHeap(RtlGetProcessHeap(), 0, CurrentDisk->LayoutBuffer);
 
    CurrentDisk->LayoutBuffer = NULL;
    CurrentDisk->ExtendedPartition = NULL;
    CurrentDisk->Dirty = FALSE;
    CurrentDisk->NewDisk = TRUE;
    CurrentDisk->PartitionStyle = DiskInfo->PartitionStyle;
 
    ReadLayoutBuffer(FileHandle, CurrentDisk);
 
done:
    if (FileHandle)
        NtClose(FileHandle);
 
    return Status;
}

Referenced by ConvertGPT(), ConvertMBR(), CreateExtendedPartition(), and CreatePrimaryPartition().