diskpart.h File Reference

#include <stdio.h>
#include <stdlib.h>
#include <windef.h>
#include <winbase.h>
#include <winreg.h>
#include <wincon.h>
#include <conutils.h>
#include <ndk/cmfuncs.h>
#include <ndk/exfuncs.h>
#include <ndk/iofuncs.h>
#include <ndk/kefuncs.h>
#include <ndk/mmfuncs.h>
#include <ndk/obfuncs.h>
#include <ndk/psfuncs.h>
#include <ndk/rtlfuncs.h>
#include <ndk/setypes.h>
#include <ndk/umfuncs.h>
#include "resource.h"

Include dependency graph for diskpart.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	_COMMAND
struct	_PARTENTRY
struct	_BIOSDISKENTRY
struct	_DISKENTRY

Macros
#define	WIN32_NO_STATUS
#define	NTOS_MODE_USER
#define	MAX_STRING_SIZE   1024
#define	MAX_ARGS_COUNT   256

Typedefs
typedef struct _COMMAND	COMMAND
typedef struct _COMMAND *	PCOMMAND
typedef enum _FORMATSTATE	FORMATSTATE
typedef enum _FORMATSTATE *	PFORMATSTATE
typedef struct _PARTENTRY	PARTENTRY
typedef struct _PARTENTRY *	PPARTENTRY
typedef struct _BIOSDISKENTRY	BIOSDISKENTRY
typedef struct _BIOSDISKENTRY *	PBIOSDISKENTRY
typedef struct _DISKENTRY	DISKENTRY
typedef struct _DISKENTRY *	PDISKENTRY

Enumerations
enum	_FORMATSTATE {   Unformatted, UnformattedOrDamaged, UnknownFormat, Preformatted,   Formatted, Unformatted, UnformattedOrDamaged, UnknownFormat,   Preformatted, Formatted }

Functions
BOOL	active_main (INT argc, LPWSTR *argv)
BOOL	add_main (INT argc, LPWSTR *argv)
BOOL	assign_main (INT argc, LPWSTR *argv)
BOOL	attach_main (INT argc, LPWSTR *argv)
BOOL	attributes_main (INT argc, LPWSTR *argv)
BOOL	automount_main (INT argc, LPWSTR *argv)
BOOL	break_main (INT argc, LPWSTR *argv)
BOOL	clean_main (INT argc, LPWSTR *argv)
BOOL	compact_main (INT argc, LPWSTR *argv)
BOOL	convert_main (INT argc, LPWSTR *argv)
BOOL	create_main (INT argc, LPWSTR *argv)
BOOL	delete_main (INT argc, LPWSTR *argv)
BOOL	detach_main (INT argc, LPWSTR *argv)
BOOL	detail_main (INT argc, LPWSTR *argv)
BOOL	expand_main (INT argc, LPWSTR *argv)
BOOL	extend_main (INT argc, LPWSTR *argv)
BOOL	filesystems_main (INT argc, LPWSTR *argv)
BOOL	format_main (INT argc, LPWSTR *argv)
BOOL	gpt_main (INT argc, LPWSTR *argv)
BOOL	help_main (INT argc, LPWSTR *argv)
VOID	help_cmdlist (VOID)
BOOL	import_main (INT argc, LPWSTR *argv)
BOOL	inactive_main (INT argc, LPWSTR *argv)
BOOL	InterpretScript (LPWSTR line)
BOOL	InterpretCmd (INT argc, LPWSTR *argv)
VOID	InterpretMain (VOID)
BOOL	list_main (INT argc, LPWSTR *argv)
BOOL	merge_main (INT argc, LPWSTR *argv)
BOOL	offline_main (INT argc, LPWSTR *argv)
BOOL	online_main (INT argc, LPWSTR *argv)
NTSTATUS	CreatePartitionList (VOID)
VOID	DestroyPartitionList (VOID)
BOOL	recover_main (INT argc, LPWSTR *argv)
BOOL	remove_main (INT argc, LPWSTR *argv)
BOOL	repair_main (INT argc, LPWSTR *argv)
BOOL	rescan_main (INT argc, LPWSTR *argv)
BOOL	retain_main (INT argc, LPWSTR *argv)
BOOL	san_main (INT argc, LPWSTR *argv)
BOOL	select_main (INT argc, LPWSTR *argv)
BOOL	setid_main (INT argc, LPWSTR *argv)
BOOL	shrink_main (INT argc, LPWSTR *argv)
BOOL	uniqueid_main (INT argc, LPWSTR *argv)

Variables
COMMAND	cmds []
LIST_ENTRY	DiskListHead
LIST_ENTRY	BiosDiskListHead
PDISKENTRY	CurrentDisk
PPARTENTRY	CurrentPartition

Macro Definition Documentation

MAX_ARGS_COUNT

#define MAX_ARGS_COUNT   256

Definition at line 70 of file diskpart.h.

MAX_STRING_SIZE

#define MAX_STRING_SIZE   1024

Definition at line 69 of file diskpart.h.

NTOS_MODE_USER

#define NTOS_MODE_USER

Definition at line 35 of file diskpart.h.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 17 of file diskpart.h.

Typedef Documentation

BIOSDISKENTRY

typedef struct _BIOSDISKENTRY BIOSDISKENTRY

COMMAND

typedef struct _COMMAND COMMAND

DISKENTRY

typedef struct _DISKENTRY DISKENTRY

FORMATSTATE

typedef enum _FORMATSTATE FORMATSTATE

PARTENTRY

typedef struct _PARTENTRY PARTENTRY

PBIOSDISKENTRY

typedef struct _BIOSDISKENTRY * PBIOSDISKENTRY

PCOMMAND

typedef struct _COMMAND * PCOMMAND

PDISKENTRY

typedef struct _DISKENTRY * PDISKENTRY

PFORMATSTATE

typedef enum _FORMATSTATE * PFORMATSTATE

PPARTENTRY

typedef struct _PARTENTRY * PPARTENTRY

Enumeration Type Documentation

_FORMATSTATE

enum _FORMATSTATE

Enumerator
Unformatted
UnformattedOrDamaged
UnknownFormat
Preformatted
Formatted
Unformatted
UnformattedOrDamaged
UnknownFormat
Preformatted
Formatted

Definition at line 73 of file diskpart.h.

{
    Unformatted,
    UnformattedOrDamaged,
    UnknownFormat,
    Preformatted,
    Formatted
} FORMATSTATE, *PFORMATSTATE;

Function Documentation

active_main()

BOOL active_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file active.c.

{
    ConPuts(StdOut, L"\nActive\n");
    return TRUE;
}

add_main()

BOOL add_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file add.c.

{
    return TRUE;
}

assign_main()

BOOL assign_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file assign.c.

{
    return TRUE;
}

attach_main()

BOOL attach_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file attach.c.

{
    return TRUE;
}

attributes_main()

BOOL attributes_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file attributes.c.

{
    return TRUE;
}

automount_main()

BOOL automount_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file automount.c.

{
    ConPuts(StdOut, L"Automount\n");
    return TRUE;
}

break_main()

BOOL break_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file break.c.

{
    ConPuts(StdOut, L"\nTODO: Add code later since Win 7 Home Premium doesn't have this feature.\n");
    return TRUE;
}

clean_main()

BOOL clean_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file clean.c.

{
    return TRUE;
}

compact_main()

BOOL compact_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file compact.c.

{
    return 0;
}

convert_main()

BOOL convert_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file convert.c.

{
    return TRUE;
}

create_main()

BOOL create_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file create.c.

{
    return TRUE;
}

CreatePartitionList()

NTSTATUS CreatePartitionList

(

VOID

)

Definition at line 2023 of file partlist.c.

{
    PPARTLIST List;
    PDISKENTRY SystemDisk;
    OBJECT_ATTRIBUTES ObjectAttributes;
    SYSTEM_DEVICE_INFORMATION Sdi;
    IO_STATUS_BLOCK Iosb;
    ULONG ReturnSize;
    NTSTATUS Status;
    ULONG DiskNumber;
    HANDLE FileHandle;
    UNICODE_STRING Name;
    WCHAR Buffer[MAX_PATH];

    List = (PPARTLIST)RtlAllocateHeap(ProcessHeap,
                                      0,
                                      sizeof(PARTLIST));
    if (List == NULL)
        return NULL;

    List->SystemPartition = NULL;

    InitializeListHead(&List->DiskListHead);
    InitializeListHead(&List->BiosDiskListHead);

    /*
     * Enumerate the disks seen by the BIOS; this will be used later
     * to map drives seen by NTOS with their corresponding BIOS names.
     */
    EnumerateBiosDiskEntries(List);

    /* Enumerate disks seen by NTOS */
    Status = NtQuerySystemInformation(SystemDeviceInformation,
                                      &Sdi,
                                      sizeof(Sdi),
                                      &ReturnSize);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("NtQuerySystemInformation() failed, Status 0x%08lx", Status);
        RtlFreeHeap(ProcessHeap, 0, List);
        return NULL;
    }

    for (DiskNumber = 0; DiskNumber < Sdi.NumberOfDisks; DiskNumber++)
    {
        RtlStringCchPrintfW(Buffer, ARRAYSIZE(Buffer),
                            L"\\Device\\Harddisk%lu\\Partition0",
                            DiskNumber);
        RtlInitUnicodeString(&Name, Buffer);

        InitializeObjectAttributes(&ObjectAttributes,
                                   &Name,
                                   OBJ_CASE_INSENSITIVE,
                                   NULL,
                                   NULL);

        Status = NtOpenFile(&FileHandle,
                            FILE_READ_DATA | FILE_READ_ATTRIBUTES | SYNCHRONIZE,
                            &ObjectAttributes,
                            &Iosb,
                            FILE_SHARE_READ | FILE_SHARE_WRITE,
                            FILE_SYNCHRONOUS_IO_NONALERT);
        if (NT_SUCCESS(Status))
        {
            AddDiskToList(FileHandle, DiskNumber, List);
            NtClose(FileHandle);
        }
    }

    UpdateDiskSignatures(List);
    UpdateHwDiskNumbers(List);
    AssignDriveLetters(List);

    /*
     * Retrieve the system partition: the active partition on the system
     * disk (the one that will be booted by default by the hardware).
     */
    SystemDisk = GetSystemDisk(List);
    List->SystemPartition = (SystemDisk ? GetActiveDiskPartition(SystemDisk) : NULL);

    return List;
}

Referenced by LoadSetupData(), rescan_main(), SelectPartitionPage(), UpgradeRepairPage(), and wmain().

delete_main()

BOOL delete_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file delete.c.

{
    return TRUE;
}

DestroyPartitionList()

VOID DestroyPartitionList

(

VOID

)

Definition at line 1120 of file partlist.c.

{
    PDISKENTRY DiskEntry;
    PBIOSDISKENTRY BiosDiskEntry;
    PPARTENTRY PartEntry;
    PLIST_ENTRY Entry;

    CurrentDisk = NULL;
    CurrentPartition = NULL;

    /* Release disk and partition info */
    while (!IsListEmpty(&DiskListHead))
    {
        Entry = RemoveHeadList(&DiskListHead);
        DiskEntry = CONTAINING_RECORD(Entry, DISKENTRY, ListEntry);

        /* Release driver name */
        RtlFreeUnicodeString(&DiskEntry->DriverName);

        /* Release primary partition list */
        while (!IsListEmpty(&DiskEntry->PrimaryPartListHead))
        {
            Entry = RemoveHeadList(&DiskEntry->PrimaryPartListHead);
            PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);

            RtlFreeHeap(RtlGetProcessHeap(), 0, PartEntry);
        }

        /* Release logical partition list */
        while (!IsListEmpty(&DiskEntry->LogicalPartListHead))
        {
            Entry = RemoveHeadList(&DiskEntry->LogicalPartListHead);
            PartEntry = CONTAINING_RECORD(Entry, PARTENTRY, ListEntry);

            RtlFreeHeap(RtlGetProcessHeap(), 0, PartEntry);
        }

        /* Release layout buffer */
        if (DiskEntry->LayoutBuffer != NULL)
            RtlFreeHeap(RtlGetProcessHeap(), 0, DiskEntry->LayoutBuffer);


        /* Release disk entry */
        RtlFreeHeap(RtlGetProcessHeap(), 0, DiskEntry);
    }

    /* Release the bios disk info */
    while (!IsListEmpty(&BiosDiskListHead))
    {
        Entry = RemoveHeadList(&BiosDiskListHead);
        BiosDiskEntry = CONTAINING_RECORD(Entry, BIOSDISKENTRY, ListEntry);

        RtlFreeHeap(RtlGetProcessHeap(), 0, BiosDiskEntry);
    }
}

detach_main()

BOOL detach_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file detach.c.

{
    return TRUE;
}

detail_main()

BOOL detail_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file detail.c.

{
    return TRUE;
}

expand_main()

BOOL expand_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file expand.c.

{
    return TRUE;
}

extend_main()

BOOL extend_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file extend.c.

{
    return TRUE;
}

filesystems_main()

BOOL filesystems_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file filesystems.c.

{
    return TRUE;
}

format_main()

BOOL format_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file format.c.

{
    return TRUE;
}

gpt_main()

BOOL gpt_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file gpt.c.

{
    return TRUE;
}

help_cmdlist()

VOID help_cmdlist

(

VOID

)

Definition at line 15 of file help.c.

{
    PCOMMAND cmdptr;

    /* Print the header information */
    ConResPuts(StdOut, IDS_APP_HEADER);
    ConPuts(StdOut, L"\n");

    /* List all the commands and the basic descriptions */
    for (cmdptr = cmds; cmdptr->name; cmdptr++)
        ConResPuts(StdOut, cmdptr->help_desc);

    ConPuts(StdOut, L"\n");
}

Referenced by help_main(), and InterpretCmd().

help_main()

BOOL help_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 33 of file help.c.

{
    PCOMMAND cmdptr;

    if (argc == 1)
    {
        help_cmdlist();
        return TRUE;
    }

    /* Scan internal command table */
    for (cmdptr = cmds; cmdptr->name; cmdptr++)
    {
        if (_wcsicmp(argv[1], cmdptr->name) == 0)
        {
            ConResPuts(StdOut, cmdptr->help);
            return TRUE;
        }
    }

    help_cmdlist();

    return TRUE;
}

import_main()

BOOL import_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file import.c.

{
    return TRUE;
}

inactive_main()

BOOL inactive_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file inactive.c.

{
    return TRUE;
}

InterpretCmd()

BOOL InterpretCmd	(	INT	argc,
		LPWSTR *	argv
	)

InterpretMain()

VOID InterpretMain

(

VOID

)

Definition at line 145 of file interpreter.c.

{
    WCHAR input_line[MAX_STRING_SIZE];
    LPWSTR args_vector[MAX_ARGS_COUNT];
    INT args_count = 0;
    BOOL bWhiteSpace = TRUE;
    BOOL bRun = TRUE;
    LPWSTR ptr;

    while (bRun != FALSE)
    {
        args_count = 0;
        memset(args_vector, 0, sizeof(args_vector));

        /* Shown just before the input where the user places commands */
        ConResPuts(StdOut, IDS_APP_PROMPT);

        /* Get input from the user. */
        fgetws(input_line, MAX_STRING_SIZE, stdin);

        ptr = input_line;
        while (*ptr != 0)
        {
            if (iswspace(*ptr) || *ptr == L'\n')
            {
                *ptr = 0;
                bWhiteSpace = TRUE;
            }
            else
            {
                if ((bWhiteSpace != FALSE) && (args_count < MAX_ARGS_COUNT))
                {
                    args_vector[args_count] = ptr;
                    args_count++;
                }
                bWhiteSpace = FALSE;
            }
            ptr++;
        }

        /* Send the string to find the command */
        bRun = InterpretCmd(args_count, args_vector);
    }
}

Referenced by wmain().

InterpretScript()

BOOL InterpretScript

(

LPWSTR

line

)

Definition at line 102 of file interpreter.c.

{
    LPWSTR args_vector[MAX_ARGS_COUNT];
    INT args_count = 0;
    BOOL bWhiteSpace = TRUE;
    LPWSTR ptr;

    memset(args_vector, 0, sizeof(args_vector));

    ptr = input_line;
    while (*ptr != 0)
    {
        if (iswspace(*ptr) || *ptr == L'\n')
        {
            *ptr = 0;
            bWhiteSpace = TRUE;
        }
        else
        {
            if ((bWhiteSpace != FALSE) && (args_count < MAX_ARGS_COUNT))
            {
                args_vector[args_count] = ptr;
                args_count++;
            }

            bWhiteSpace = FALSE;
        }

        ptr++;
    }

    /* sends the string to find the command */
    return InterpretCmd(args_count, args_vector);
}

Referenced by RunScript().

list_main()

BOOL list_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 235 of file list.c.

{
    /* gets the first word from the string */
    if (argc == 1)
    {
        ConResPuts(StdOut, IDS_HELP_CMD_LIST);
        return TRUE;
    }

    /* determines which to list (disk, partition, etc.) */
    if (!wcsicmp(argv[1], L"disk"))
        ListDisk();
    else if (!wcsicmp(argv[1], L"partition"))
        ListPartition();
    else if (!wcsicmp(argv[1], L"volume"))
        ListVolume();
    else if (!wcsicmp(argv[1], L"vdisk"))
        ListVdisk();
    else
        ConResPuts(StdOut, IDS_HELP_CMD_LIST);

    return TRUE;
}

merge_main()

BOOL merge_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file merge.c.

{
    return TRUE;
}

offline_main()

BOOL offline_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file offline.c.

{
    return TRUE;
}

online_main()

BOOL online_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file online.c.

{
    ConResPuts(StdOut, IDS_HELP_CMD_ONLINE);
    return TRUE;
}

recover_main()

BOOL recover_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file recover.c.

{
    return TRUE;
}

remove_main()

BOOL remove_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file remove.c.

{
    return TRUE;
}

repair_main()

BOOL repair_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file repair.c.

{
    ConPuts(StdOut, L"\nTODO: Add code later since Win 7 Home Premium doesn't have this feature.\n");
    return TRUE;
}

rescan_main()

BOOL rescan_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file rescan.c.

{
    ConResPuts(StdOut, IDS_RESCAN_START);
    DestroyPartitionList();
    CreatePartitionList();
    ConResPuts(StdOut, IDS_RESCAN_END);

    return TRUE;
}

retain_main()

BOOL retain_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file retain.c.

{
    return TRUE;
}

san_main()

BOOL san_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file san.c.

{
    return TRUE;
}

select_main()

BOOL select_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 161 of file select.c.

{
    /* gets the first word from the string */
    if (argc == 1)
    {
        ConResPuts(StdOut, IDS_HELP_CMD_SELECT);
        return TRUE;
    }

    /* determines which to list (disk, partition, etc.) */
    if (!wcsicmp(argv[1], L"disk"))
        SelectDisk(argc, argv);
    else if (!wcsicmp(argv[1], L"partition"))
        SelectPartition(argc, argv);
    else
        ConResPuts(StdOut, IDS_HELP_CMD_SELECT);

    return TRUE;
}

setid_main()

BOOL setid_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file setid.c.

{
    return TRUE;
}

shrink_main()

BOOL shrink_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file shrink.c.

{
    return TRUE;
}

uniqueid_main()

BOOL uniqueid_main	(	INT	argc,
		LPWSTR *	argv
	)

Definition at line 11 of file uniqueid.c.

{
    return TRUE;
}

Variable Documentation

BiosDiskListHead

LIST_ENTRY BiosDiskListHead

Definition at line 71 of file partlist.c.

Referenced by AddDiskToList(), CreatePartitionList(), DestroyPartitionList(), and EnumerateBiosDiskEntries().

cmds

COMMAND cmds[]

Definition at line 21 of file main.c.

Referenced by DoCommand(), handle_arrow_key(), handle_edit_event(), help_cmdlist(), help_main(), InterpretCmd(), PrintCommandList(), test_ioctlsocket(), test_OleCommandTarget(), and wmain().

CurrentDisk

PDISKENTRY CurrentDisk

Definition at line 73 of file partlist.c.

Referenced by CreatePartitionList(), DestroyPartitionList(), GetNextPartition(), GetPrevPartition(), ListDisk(), ListPartition(), SelectDisk(), and SelectPartition().

CurrentPartition

PPARTENTRY CurrentPartition

Definition at line 74 of file partlist.c.

Referenced by CreatePartitionList(), and DestroyPartitionList().

DiskListHead

LIST_ENTRY DiskListHead

Definition at line 70 of file partlist.c.

Referenced by AddDiskToList(), CreatePartitionList(), DestroyPartitionList(), ListDisk(), and SelectDisk().