usetup.c

Go to the documentation of this file.

/*
 *  ReactOS kernel
 *  Copyright (C) 2002, 2003, 2004 ReactOS Team
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along
 *  with this program; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
/*
 * COPYRIGHT:       See COPYING in the top level directory
 * PROJECT:         ReactOS text-mode setup
 * FILE:            base/setup/usetup/usetup.c
 * PURPOSE:         Text-mode setup
 * PROGRAMMERS:     Casper S. Hornstrup (chorns@users.sourceforge.net)
 *                  Hervé Poussineau (hpoussin@reactos.org)
 */
 
#include <usetup.h>
#include <math.h>
#include <ntstrsafe.h>
 
#include "cmdcons.h"
#include "devinst.h"
#include "fmtchk.h"
 
#define NDEBUG
#include <debug.h>
 
 
/* GLOBALS & LOCALS *********************************************************/
 
HANDLE ProcessHeap;
 
static USETUP_DATA USetupData;
static BOOLEAN IsUnattendedSetup;
 
/* The partition where to perform the installation */
static PPARTENTRY InstallPartition = NULL;
// static PVOLENTRY InstallVolume = NULL;
#define InstallVolume (InstallPartition->Volume)
/*
 * The system partition we will actually use. It can be different from
 * PartitionList->SystemPartition in case we don't support it, or we install
 * on a removable disk.
 * We may indeed not support the original system partition in case we do not
 * have write support on it. Please note that this situation is partly a HACK
 * and MUST NEVER happen on architectures where real system partitions are
 * mandatory (because then they are formatted in FAT FS and we support write
 * operation on them).
 */
static PPARTENTRY SystemPartition = NULL;
// static PVOLENTRY SystemVolume = NULL;
#define SystemVolume (SystemPartition->Volume)
 
 
/* OTHER Stuff *****/
 
PCWSTR SelectedLanguageId;
static WCHAR DefaultLanguage[20];   // Copy of string inside LanguageList
static WCHAR DefaultKBLayout[20];   // Copy of string inside KeyboardList
 
static BOOLEAN RepairUpdateFlag = FALSE;
 
/* Global partition list on the system */
static PPARTLIST PartitionList = NULL;
 
/* Currently selected partition entry in the list */
static PPARTENTRY CurrentPartition = NULL;
static enum {
    PartTypeData,    // On MBR-disks, primary or logical partition
    PartTypeExtended // MBR-disk container
} PartCreateType = PartTypeData;
 
/* Flag set in VOLENTRY::New when a partition/volume is created automatically */
#define VOLUME_NEW_AUTOCREATE   0x80
 
/* List of supported file systems for the partition to be formatted */
static PFILE_SYSTEM_LIST FileSystemList = NULL;
 
/*****************************************************/
 
static PNTOS_INSTALLATION CurrentInstallation = NULL;
static PGENERIC_LIST NtOsInstallsList = NULL;
 
#ifdef __REACTOS__ /* HACK */
 
/* FONT SUBSTITUTION WORKAROUND *************************************************/
 
/* For font file check */
FONTSUBSTSETTINGS s_SubstSettings = { FALSE };
 
static void
DoWatchDestFileName(LPCWSTR FileName)
{
    if (FileName[0] == 'm' || FileName[0] == 'M')
    {
        if (_wcsicmp(FileName, L"mingliu.ttc") == 0)
        {
            DPRINT("mingliu.ttc found\n");
            s_SubstSettings.bFoundFontMINGLIU = TRUE;
        }
        else if (_wcsicmp(FileName, L"msgothic.ttc") == 0)
        {
            DPRINT("msgothic.ttc found\n");
            s_SubstSettings.bFoundFontMSGOTHIC = TRUE;
        }
        else if (_wcsicmp(FileName, L"msmincho.ttc") == 0)
        {
            DPRINT("msmincho.ttc found\n");
            s_SubstSettings.bFoundFontMSMINCHO = TRUE;
        }
        else if (_wcsicmp(FileName, L"mssong.ttf") == 0)
        {
            DPRINT("mssong.ttf found\n");
            s_SubstSettings.bFoundFontMSSONG = TRUE;
        }
    }
    else
    {
        if (_wcsicmp(FileName, L"simsun.ttc") == 0)
        {
            DPRINT("simsun.ttc found\n");
            s_SubstSettings.bFoundFontSIMSUN = TRUE;
        }
        else if (_wcsicmp(FileName, L"gulim.ttc") == 0)
        {
            DPRINT("gulim.ttc found\n");
            s_SubstSettings.bFoundFontGULIM = TRUE;
        }
        else if (_wcsicmp(FileName, L"batang.ttc") == 0)
        {
            DPRINT("batang.ttc found\n");
            s_SubstSettings.bFoundFontBATANG = TRUE;
        }
    }
}
#endif  /* HACK */
 
/* FUNCTIONS ****************************************************************/
 
static VOID
PrintString(IN PCSTR fmt,...)
{
    CHAR buffer[512];
    va_list ap;
    UNICODE_STRING UnicodeString;
    ANSI_STRING AnsiString;
 
    va_start(ap, fmt);
    vsprintf(buffer, fmt, ap);
    va_end(ap);
 
    RtlInitAnsiString(&AnsiString, buffer);
    RtlAnsiStringToUnicodeString(&UnicodeString, &AnsiString, TRUE);
    NtDisplayString(&UnicodeString);
    RtlFreeUnicodeString(&UnicodeString);
}
 
 
static VOID
DrawBox(IN SHORT xLeft,
        IN SHORT yTop,
        IN SHORT Width,
        IN SHORT Height)
{
    COORD coPos;
    DWORD Written;
 
    /* Draw upper left corner */
    coPos.X = xLeft;
    coPos.Y = yTop;
    FillConsoleOutputCharacterA(StdOutput,
                                CharUpperLeftCorner, // '+',
                                1,
                                coPos,
                                &Written);
 
    /* Draw upper edge */
    coPos.X = xLeft + 1;
    coPos.Y = yTop;
    FillConsoleOutputCharacterA(StdOutput,
                                CharHorizontalLine, // '-',
                                Width - 2,
                                coPos,
                                &Written);
 
    /* Draw upper right corner */
    coPos.X = xLeft + Width - 1;
    coPos.Y = yTop;
    FillConsoleOutputCharacterA(StdOutput,
                                CharUpperRightCorner, // '+',
                                1,
                                coPos,
                                &Written);
 
    /* Draw right edge, inner space and left edge */
    for (coPos.Y = yTop + 1; coPos.Y < yTop + Height - 1; coPos.Y++)
    {
        coPos.X = xLeft;
        FillConsoleOutputCharacterA(StdOutput,
                                    CharVerticalLine, // '|',
                                    1,
                                    coPos,
                                    &Written);
 
        coPos.X = xLeft + 1;
        FillConsoleOutputCharacterA(StdOutput,
                                    ' ',
                                    Width - 2,
                                    coPos,
                                    &Written);
 
        coPos.X = xLeft + Width - 1;
        FillConsoleOutputCharacterA(StdOutput,
                                    CharVerticalLine, // '|',
                                    1,
                                    coPos,
                                    &Written);
    }
 
    /* Draw lower left corner */
    coPos.X = xLeft;
    coPos.Y = yTop + Height - 1;
    FillConsoleOutputCharacterA(StdOutput,
                                CharLowerLeftCorner, // '+',
                                1,
                                coPos,
                                &Written);
 
    /* Draw lower edge */
    coPos.X = xLeft + 1;
    coPos.Y = yTop + Height - 1;
    FillConsoleOutputCharacterA(StdOutput,
                                CharHorizontalLine, // '-',
                                Width - 2,
                                coPos,
                                &Written);
 
    /* Draw lower right corner */
    coPos.X = xLeft + Width - 1;
    coPos.Y = yTop + Height - 1;
    FillConsoleOutputCharacterA(StdOutput,
                                CharLowerRightCorner, // '+',
                                1,
                                coPos,
                                &Written);
}
 
 
VOID
PopupError(PCCH Text,
           PCCH Status,
           PINPUT_RECORD Ir,
           ULONG WaitEvent)
{
    SHORT yTop;
    SHORT xLeft;
    COORD coPos;
    DWORD Written;
    ULONG Length;
    ULONG MaxLength;
    ULONG Lines;
    PCHAR p;
    PCCH pnext;
    BOOLEAN LastLine;
    SHORT Width;
    SHORT Height;
 
    /* Count text lines and longest line */
    MaxLength = 0;
    Lines = 0;
    pnext = Text;
 
    while (TRUE)
    {
        p = strchr(pnext, '\n');
 
        if (p == NULL)
        {
            Length = strlen(pnext);
            LastLine = TRUE;
        }
        else
        {
            Length = (ULONG)(p - pnext);
            LastLine = FALSE;
        }
 
        Lines++;
 
        if (Length > MaxLength)
            MaxLength = Length;
 
        if (LastLine)
            break;
 
        pnext = p + 1;
    }
 
    /* Check length of status line */
    if (Status != NULL)
    {
        Length = strlen(Status);
 
        if (Length > MaxLength)
            MaxLength = Length;
    }
 
    Width = MaxLength + 4;
    Height = Lines + 2;
 
    if (Status != NULL)
        Height += 2;
 
    yTop = (yScreen - Height) / 2;
    xLeft = (xScreen - Width) / 2;
 
 
    /* Set screen attributes */
    coPos.X = xLeft;
    for (coPos.Y = yTop; coPos.Y < yTop + Height; coPos.Y++)
    {
        FillConsoleOutputAttribute(StdOutput,
                                   FOREGROUND_RED | BACKGROUND_WHITE,
                                   Width,
                                   coPos,
                                   &Written);
    }
 
    DrawBox(xLeft, yTop, Width, Height);
 
    /* Print message text */
    coPos.Y = yTop + 1;
    pnext = Text;
    while (TRUE)
    {
        p = strchr(pnext, '\n');
 
        if (p == NULL)
        {
            Length = strlen(pnext);
            LastLine = TRUE;
        }
        else
        {
            Length = (ULONG)(p - pnext);
            LastLine = FALSE;
        }
 
        if (Length != 0)
        {
            coPos.X = xLeft + 2;
            WriteConsoleOutputCharacterA(StdOutput,
                                         pnext,
                                         Length,
                                         coPos,
                                         &Written);
        }
 
        if (LastLine)
            break;
 
        coPos.Y++;
        pnext = p + 1;
    }
 
    /* Print separator line and status text */
    if (Status != NULL)
    {
        coPos.Y = yTop + Height - 3;
        coPos.X = xLeft;
        FillConsoleOutputCharacterA(StdOutput,
                                    CharVertLineAndRightHorizLine, // '+',
                                    1,
                                    coPos,
                                    &Written);
 
        coPos.X = xLeft + 1;
        FillConsoleOutputCharacterA(StdOutput,
                                    CharHorizontalLine, // '-',
                                    Width - 2,
                                    coPos,
                                    &Written);
 
        coPos.X = xLeft + Width - 1;
        FillConsoleOutputCharacterA(StdOutput,
                                    CharLeftHorizLineAndVertLine, // '+',
                                    1,
                                    coPos,
                                    &Written);
 
        coPos.Y++;
        coPos.X = xLeft + 2;
        WriteConsoleOutputCharacterA(StdOutput,
                                     Status,
                                     min(strlen(Status), (SIZE_T)Width - 4),
                                     coPos,
                                     &Written);
    }
 
    if (WaitEvent == POPUP_WAIT_NONE)
        return;
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if (WaitEvent == POPUP_WAIT_ANY_KEY ||
            Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D)
        {
            return;
        }
    }
}
 
 
/*
 * Confirm quit setup
 * RETURNS
 *   TRUE: Quit setup.
 *   FALSE: Don't quit setup.
 */
static BOOL
ConfirmQuit(PINPUT_RECORD Ir)
{
    BOOL Result = FALSE;
    MUIDisplayError(ERROR_NOT_INSTALLED, NULL, POPUP_WAIT_NONE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            Result = TRUE;
            break;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D)  /* ENTER */
        {
            Result = FALSE;
            break;
        }
    }
 
    return Result;
}
 
 
static VOID
UpdateKBLayout(VOID)
{
    PGENERIC_LIST_ENTRY ListEntry;
    KLID newLayout;
 
    newLayout = MUIDefaultKeyboardLayout(SelectedLanguageId);
 
    if (!USetupData.LayoutList)
    {
        USetupData.LayoutList = CreateKeyboardLayoutList(USetupData.SetupInf, SelectedLanguageId, DefaultKBLayout);
        if (!USetupData.LayoutList)
        {
            /* FIXME: Handle error! */
            return;
        }
    }
 
    /* Search for default layout (if provided) */
    if (newLayout != 0)
    {
        for (ListEntry = GetFirstListEntry(USetupData.LayoutList); ListEntry;
             ListEntry = GetNextListEntry(ListEntry))
        {
            PCWSTR pszLayoutId = ((PGENENTRY)GetListEntryData(ListEntry))->Id;
            KLID LayoutId = (KLID)(pszLayoutId ? wcstoul(pszLayoutId, NULL, 16) : 0);
            if (newLayout == LayoutId)
            {
                SetCurrentListEntry(USetupData.LayoutList, ListEntry);
                break;
            }
        }
    }
}
 
 
static NTSTATUS
NTAPI
GetSettingDescription(
    IN PGENERIC_LIST_ENTRY Entry,
    OUT PSTR Buffer,
    IN SIZE_T cchBufferSize)
{
    return RtlStringCchPrintfA(Buffer, cchBufferSize, "%S",
                               ((PGENENTRY)GetListEntryData(Entry))->Value);
}
 
static NTSTATUS
NTAPI
GetNTOSInstallationName(
    IN PGENERIC_LIST_ENTRY Entry,
    OUT PSTR Buffer,
    IN SIZE_T cchBufferSize)
{
    PNTOS_INSTALLATION NtOsInstall = (PNTOS_INSTALLATION)GetListEntryData(Entry);
    PVOLINFO VolInfo = (NtOsInstall->Volume ? &NtOsInstall->Volume->Info : NULL);
 
    if (VolInfo && VolInfo->DriveLetter)
    {
        /* We have retrieved a partition that is mounted */
        return RtlStringCchPrintfA(Buffer, cchBufferSize,
                                   "%C:%S  \"%S\"",
                                   VolInfo->DriveLetter,
                                   NtOsInstall->PathComponent,
                                   NtOsInstall->InstallationName);
    }
    else
    {
        /* We failed somewhere, just show the NT path */
        return RtlStringCchPrintfA(Buffer, cchBufferSize,
                                   "%wZ  \"%S\"",
                                   &NtOsInstall->SystemNtPath,
                                   NtOsInstall->InstallationName);
    }
}
 
 
// PSETUP_ERROR_ROUTINE
static VOID
__cdecl
USetupErrorRoutine(
    IN PUSETUP_DATA pSetupData,
    ...)
{
    INPUT_RECORD Ir;
    va_list arg_ptr;
 
    va_start(arg_ptr, pSetupData);
 
    if (pSetupData->LastErrorNumber >= ERROR_SUCCESS &&
        pSetupData->LastErrorNumber <  ERROR_LAST_ERROR_CODE)
    {
        // Note: the "POPUP_WAIT_ENTER" actually depends on the LastErrorNumber...
        MUIDisplayErrorV(pSetupData->LastErrorNumber, &Ir, POPUP_WAIT_ENTER, arg_ptr);
    }
 
    va_end(arg_ptr);
}
 
/*
 * Start page
 *
 * Next pages:
 *  LanguagePage (at once, default)
 *  InstallIntroPage (at once, if unattended)
 *  QuitPage
 *
 * SIDEEFFECTS
 *  Init Sdi
 *  Init USetupData.SourcePath
 *  Init USetupData.SourceRootPath
 *  Init USetupData.SourceRootDir
 *  Init USetupData.SetupInf
 *  Init USetupData.RequiredPartitionDiskSpace
 *  Init IsUnattendedSetup
 *  If unattended, init *List and sets the Codepage
 *  If unattended, init SelectedLanguageId
 *  If unattended, init USetupData.LanguageId
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
SetupStartPage(PINPUT_RECORD Ir)
{
    ULONG Error;
    PGENERIC_LIST_ENTRY ListEntry;
    PCWSTR LocaleId;
 
    MUIDisplayPage(SETUP_INIT_PAGE);
 
    /* Initialize Setup */
    Error = InitializeSetup(&USetupData, USetupErrorRoutine,
                            &SpFileExports, &SpInfExports);
    if (Error != ERROR_SUCCESS)
    {
        MUIDisplayError(Error, Ir, POPUP_WAIT_ENTER);
        return QUIT_PAGE;
    }
 
    /* Initialize the user-mode PnP manager */
    if (!EnableUserModePnpManager())
        DPRINT1("The user-mode PnP manager could not initialize, expect unavailable devices!\n");
 
    /* Wait for any immediate pending installations to finish */
    if (WaitNoPendingInstallEvents(NULL) != STATUS_WAIT_0)
        DPRINT1("WaitNoPendingInstallEvents() failed to wait!\n");
 
    IsUnattendedSetup = CheckUnattendedSetup(&USetupData);
    if (IsUnattendedSetup)
    {
        // TODO: Read options from inf
        /* Load the hardware, language and keyboard layout lists */
 
        USetupData.ComputerList = CreateComputerTypeList(USetupData.SetupInf);
        USetupData.DisplayList = CreateDisplayDriverList(USetupData.SetupInf);
        USetupData.KeyboardList = CreateKeyboardDriverList(USetupData.SetupInf);
 
        USetupData.LanguageList = CreateLanguageList(USetupData.SetupInf, DefaultLanguage);
 
        /* new part */
        SelectedLanguageId = DefaultLanguage;
        wcscpy(DefaultLanguage, USetupData.LocaleID);
        USetupData.LanguageId = (LANGID)(wcstol(SelectedLanguageId, NULL, 16) & 0xFFFF);
 
        USetupData.LayoutList = CreateKeyboardLayoutList(USetupData.SetupInf, SelectedLanguageId, DefaultKBLayout);
 
        /* first we hack LanguageList */
        for (ListEntry = GetFirstListEntry(USetupData.LanguageList); ListEntry;
             ListEntry = GetNextListEntry(ListEntry))
        {
            LocaleId = ((PGENENTRY)GetListEntryData(ListEntry))->Id;
            if (!_wcsicmp(USetupData.LocaleID, LocaleId))
            {
                DPRINT("found %S in LanguageList\n", LocaleId);
                SetCurrentListEntry(USetupData.LanguageList, ListEntry);
                break;
            }
        }
 
        /* now LayoutList */
        for (ListEntry = GetFirstListEntry(USetupData.LayoutList); ListEntry;
             ListEntry = GetNextListEntry(ListEntry))
        {
            LocaleId = ((PGENENTRY)GetListEntryData(ListEntry))->Id;
            if (!_wcsicmp(USetupData.LocaleID, LocaleId))
            {
                DPRINT("found %S in LayoutList\n", LocaleId);
                SetCurrentListEntry(USetupData.LayoutList, ListEntry);
                break;
            }
        }
 
        SetConsoleCodePage();
 
        return INSTALL_INTRO_PAGE;
    }
 
    return LANGUAGE_PAGE;
}
 
 
/*
 * Displays the LanguagePage.
 *
 * Next pages: WelcomePage, QuitPage
 *
 * SIDEEFFECTS
 *  Init SelectedLanguageId
 *  Init USetupData.LanguageId
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
LanguagePage(PINPUT_RECORD Ir)
{
    GENERIC_LIST_UI ListUi;
    PCWSTR NewLanguageId;
    BOOL RefreshPage = FALSE;
 
    /* Initialize the computer settings list */
    if (USetupData.LanguageList == NULL)
    {
        USetupData.LanguageList = CreateLanguageList(USetupData.SetupInf, DefaultLanguage);
        if (USetupData.LanguageList == NULL)
        {
           PopupError("Setup failed to initialize available translations", NULL, NULL, POPUP_WAIT_NONE);
           return WELCOME_PAGE;
        }
    }
 
    SelectedLanguageId = DefaultLanguage;
    USetupData.LanguageId = 0;
 
    /* Load the font */
    SetConsoleCodePage();
    UpdateKBLayout();
 
    /*
     * If there is no language or just a single one in the list,
     * skip the language selection process altogether.
     */
    if (GetNumberOfListEntries(USetupData.LanguageList) <= 1)
    {
        USetupData.LanguageId = (LANGID)(wcstol(SelectedLanguageId, NULL, 16) & 0xFFFF);
        return WELCOME_PAGE;
    }
 
    InitGenericListUi(&ListUi, USetupData.LanguageList, GetSettingDescription);
    DrawGenericList(&ListUi,
                    2, 18,
                    xScreen - 3,
                    yScreen - 3);
 
    ScrollToPositionGenericList(&ListUi, GetDefaultLanguageIndex());
 
    MUIDisplayPage(LANGUAGE_PAGE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_DOWN))  /* DOWN */
        {
            ScrollDownGenericList(&ListUi);
            RefreshPage = TRUE;
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_UP))  /* UP */
        {
            ScrollUpGenericList(&ListUi);
            RefreshPage = TRUE;
        }
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_NEXT))  /* PAGE DOWN */
        {
            ScrollPageDownGenericList(&ListUi);
            RefreshPage = TRUE;
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_PRIOR))  /* PAGE UP */
        {
            ScrollPageUpGenericList(&ListUi);
            RefreshPage = TRUE;
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            RedrawGenericList(&ListUi);
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D)  /* ENTER */
        {
            ASSERT(GetNumberOfListEntries(USetupData.LanguageList) >= 1);
 
            SelectedLanguageId =
                ((PGENENTRY)GetListEntryData(GetCurrentListEntry(USetupData.LanguageList)))->Id;
 
            USetupData.LanguageId = (LANGID)(wcstol(SelectedLanguageId, NULL, 16) & 0xFFFF);
 
            if (wcscmp(SelectedLanguageId, DefaultLanguage))
            {
                UpdateKBLayout();
            }
 
            /* Load the font */
            SetConsoleCodePage();
 
            return WELCOME_PAGE;
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar > 0x60) && (Ir->Event.KeyEvent.uChar.AsciiChar < 0x7b))
        {
            /* a-z */
            GenericListKeyPress(&ListUi, Ir->Event.KeyEvent.uChar.AsciiChar);
            RefreshPage = TRUE;
        }
 
        if (RefreshPage)
        {
            ASSERT(GetNumberOfListEntries(USetupData.LanguageList) >= 1);
 
            NewLanguageId =
                ((PGENENTRY)GetListEntryData(GetCurrentListEntry(USetupData.LanguageList)))->Id;
 
            if (wcscmp(SelectedLanguageId, NewLanguageId))
            {
                /* Clear the language page */
                MUIClearPage(LANGUAGE_PAGE);
 
                SelectedLanguageId = NewLanguageId;
 
                /* Load the font */
                SetConsoleCodePage();
 
                /* Redraw the list */
                DrawGenericList(&ListUi,
                                2, 18,
                                xScreen - 3,
                                yScreen - 3);
 
                /* Redraw language selection page in native language */
                MUIDisplayPage(LANGUAGE_PAGE);
            }
 
            RefreshPage = FALSE;
        }
    }
 
    return WELCOME_PAGE;
}
 
 
/*
 * Displays the WelcomePage.
 *
 * Next pages:
 *  InstallIntroPage (default)
 *  RepairIntroPage
 *  RecoveryPage
 *  LicensePage
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
WelcomePage(PINPUT_RECORD Ir)
{
    MUIDisplayPage(WELCOME_PAGE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
        {
            return INSTALL_INTRO_PAGE;
        }
        else if (toupper(Ir->Event.KeyEvent.uChar.AsciiChar) == 'R') /* R */
        {
            return RECOVERY_PAGE; // REPAIR_INTRO_PAGE;
        }
        else if (toupper(Ir->Event.KeyEvent.uChar.AsciiChar) == 'L') /* L */
        {
            return LICENSE_PAGE;
        }
    }
 
    return WELCOME_PAGE;
}
 
 
/*
 * Displays the License page.
 *
 * Next page:
 *  WelcomePage (default)
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
LicensePage(PINPUT_RECORD Ir)
{
    MUIDisplayPage(LICENSE_PAGE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D)  /* ENTER */
        {
            return WELCOME_PAGE;
        }
    }
 
    return LICENSE_PAGE;
}
 
 
/*
 * Displays the RepairIntroPage.
 *
 * Next pages:
 *  RebootPage (default)
 *  InstallIntroPage
 *  RecoveryPage
 *  IntroPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
RepairIntroPage(PINPUT_RECORD Ir)
{
    MUIDisplayPage(REPAIR_INTRO_PAGE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D)  /* ENTER */
        {
            return REBOOT_PAGE;
        }
        else if (toupper(Ir->Event.KeyEvent.uChar.AsciiChar) == 'U')  /* U */
        {
            RepairUpdateFlag = TRUE;
            return INSTALL_INTRO_PAGE;
        }
        else if (toupper(Ir->Event.KeyEvent.uChar.AsciiChar) == 'R')  /* R */
        {
            return RECOVERY_PAGE;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE)  /* ESC */
        {
            return WELCOME_PAGE;
        }
    }
 
    return REPAIR_INTRO_PAGE;
}
 
/*
 * Displays the UpgradeRepairPage.
 *
 * Next pages:
 *  RebootPage (default)
 *  InstallIntroPage
 *  RecoveryPage
 *  WelcomePage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
UpgradeRepairPage(PINPUT_RECORD Ir)
{
    GENERIC_LIST_UI ListUi;
 
/*** HACK!! ***/
    if (PartitionList == NULL)
    {
        PartitionList = CreatePartitionList();
        if (PartitionList == NULL)
        {
            /* FIXME: show an error dialog */
            MUIDisplayError(ERROR_DRIVE_INFORMATION, Ir, POPUP_WAIT_ENTER);
            return QUIT_PAGE;
        }
        else if (IsListEmpty(&PartitionList->DiskListHead))
        {
            MUIDisplayError(ERROR_NO_HDD, Ir, POPUP_WAIT_ENTER);
            return QUIT_PAGE;
        }
    }
/**************/
 
    NtOsInstallsList = CreateNTOSInstallationsList(PartitionList);
    if (!NtOsInstallsList)
        DPRINT1("Failed to get a list of NTOS installations; continue installation...\n");
 
    /*
     * If there is no available installation (or just a single one??) that can
     * be updated in the list, just continue with the regular installation.
     */
    if (!NtOsInstallsList || GetNumberOfListEntries(NtOsInstallsList) == 0)
    {
        RepairUpdateFlag = FALSE;
 
        // return INSTALL_INTRO_PAGE;
        return DEVICE_SETTINGS_PAGE;
        // return SCSI_CONTROLLER_PAGE;
    }
 
    MUIDisplayPage(UPGRADE_REPAIR_PAGE);
 
    InitGenericListUi(&ListUi, NtOsInstallsList, GetNTOSInstallationName);
    DrawGenericList(&ListUi,
                    2, 23,
                    xScreen - 3,
                    yScreen - 3);
 
    // return HandleGenericList(&ListUi, DEVICE_SETTINGS_PAGE, Ir);
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x00)
        {
            switch (Ir->Event.KeyEvent.wVirtualKeyCode)
            {
            case VK_DOWN:   /* DOWN */
                ScrollDownGenericList(&ListUi);
                break;
            case VK_UP:     /* UP */
                ScrollUpGenericList(&ListUi);
                break;
            case VK_NEXT:   /* PAGE DOWN */
                ScrollPageDownGenericList(&ListUi);
                break;
            case VK_PRIOR:  /* PAGE UP */
                ScrollPageUpGenericList(&ListUi);
                break;
            case VK_F3:     /* F3 */
            {
                if (ConfirmQuit(Ir))
                    return QUIT_PAGE;
                RedrawGenericList(&ListUi);
                break;
            }
#if 1
/* TODO: Temporarily kept until correct keyboard layout is in place.
 * (Actual AsciiChar of ESCAPE should be 0x1B instead of 0.)
 * Addendum to commit 8b94515b.
 */
            case VK_ESCAPE: /* ESC */
            {
                RestoreGenericListUiState(&ListUi);
                // return nextPage;    // prevPage;
 
                // return INSTALL_INTRO_PAGE;
                return DEVICE_SETTINGS_PAGE;
                // return SCSI_CONTROLLER_PAGE;
            }
 
#endif
            }
        }
#if 0
/* TODO: Restore this once correct keyboard layout is in place. */
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE) /* ESC */
        {
            RestoreGenericListUiState(&ListUi);
            // return nextPage;    // prevPage;
 
            // return INSTALL_INTRO_PAGE;
            return DEVICE_SETTINGS_PAGE;
            // return SCSI_CONTROLLER_PAGE;
        }
#endif
        else
        {
            // switch (toupper(Ir->Event.KeyEvent.uChar.AsciiChar))
            // if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
            if (toupper(Ir->Event.KeyEvent.uChar.AsciiChar) == 'U')  /* U */
            {
                /* Retrieve the current installation */
                ASSERT(GetNumberOfListEntries(NtOsInstallsList) >= 1);
 
                CurrentInstallation =
                    (PNTOS_INSTALLATION)GetListEntryData(GetCurrentListEntry(NtOsInstallsList));
 
                DPRINT1("Selected installation for repair: \"%S\" ; DiskNumber = %d , PartitionNumber = %d\n",
                        CurrentInstallation->InstallationName, CurrentInstallation->DiskNumber, CurrentInstallation->PartitionNumber);
 
                RepairUpdateFlag = TRUE;
 
                // return nextPage;
                /***/return INSTALL_INTRO_PAGE;/***/
            }
            else if ((Ir->Event.KeyEvent.uChar.AsciiChar > 0x60) &&
                     (Ir->Event.KeyEvent.uChar.AsciiChar < 0x7b))   /* a-z */
            {
                GenericListKeyPress(&ListUi, Ir->Event.KeyEvent.uChar.AsciiChar);
            }
        }
    }
 
    return UPGRADE_REPAIR_PAGE;
}
 
 
/*
 * Displays the InstallIntroPage.
 *
 * Next pages:
 *  DeviceSettingsPage  (At once if repair or update is selected)
 *  SelectPartitionPage (At once if unattended setup)
 *  DeviceSettingsPage  (default)
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
InstallIntroPage(PINPUT_RECORD Ir)
{
    if (RepairUpdateFlag)
    {
#if 1 /* Old code that looks good */
 
        // return SELECT_PARTITION_PAGE;
        return DEVICE_SETTINGS_PAGE;
 
#else /* Possible new code? */
 
        return DEVICE_SETTINGS_PAGE;
        // return SCSI_CONTROLLER_PAGE;
 
#endif
    }
 
    if (IsUnattendedSetup)
        return SELECT_PARTITION_PAGE;
 
    MUIDisplayPage(INSTALL_INTRO_PAGE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3)) /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
        {
            return UPGRADE_REPAIR_PAGE;
        }
    }
 
    return INSTALL_INTRO_PAGE;
}
 
 
#if 0
static PAGE_NUMBER
ScsiControllerPage(PINPUT_RECORD Ir)
{
    // MUIDisplayPage(SCSI_CONTROLLER_PAGE);
 
    CONSOLE_SetTextXY(6, 8, "Setup detected the following mass storage devices:");
 
    /* FIXME: print loaded mass storage driver descriptions */
#if 0
    CONSOLE_SetTextXY(8, 10, "TEST device");
#endif
 
    CONSOLE_SetStatusText("   ENTER = Continue   F3 = Quit");
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3)) /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
        {
            return DEVICE_SETTINGS_PAGE;
        }
    }
 
    return SCSI_CONTROLLER_PAGE;
}
 
static PAGE_NUMBER
OemDriverPage(PINPUT_RECORD Ir)
{
    // MUIDisplayPage(OEM_DRIVER_PAGE);
 
    CONSOLE_SetTextXY(6, 8, "This is the OEM driver page!");
 
    /* FIXME: Implement!! */
 
    CONSOLE_SetStatusText("   ENTER = Continue   F3 = Quit");
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3)) /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
        {
            return DEVICE_SETTINGS_PAGE;
        }
    }
 
    return OEM_DRIVER_PAGE;
}
#endif
 
 
/*
 * Displays the DeviceSettingsPage.
 *
 * Next pages:
 *  SelectPartitionPage (At once if repair or update is selected)
 *  ComputerSettingsPage
 *  DisplaySettingsPage
 *  KeyboardSettingsPage
 *  LayoutsettingsPage
 *  SelectPartitionPage
 *  QuitPage
 *
 * SIDEEFFECTS
 *  Init USetupData.ComputerList
 *  Init USetupData.DisplayList
 *  Init USetupData.KeyboardList
 *  Init USetupData.LayoutList
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
DeviceSettingsPage(PINPUT_RECORD Ir)
{
    static ULONG Line = 16;
 
    /* Initialize the computer settings list */
    if (USetupData.ComputerList == NULL)
    {
        USetupData.ComputerList = CreateComputerTypeList(USetupData.SetupInf);
        if (USetupData.ComputerList == NULL)
        {
            MUIDisplayError(ERROR_LOAD_COMPUTER, Ir, POPUP_WAIT_ENTER);
            return QUIT_PAGE;
        }
    }
 
    /* Initialize the display settings list */
    if (USetupData.DisplayList == NULL)
    {
        USetupData.DisplayList = CreateDisplayDriverList(USetupData.SetupInf);
        if (USetupData.DisplayList == NULL)
        {
            MUIDisplayError(ERROR_LOAD_DISPLAY, Ir, POPUP_WAIT_ENTER);
            return QUIT_PAGE;
        }
    }
 
    /* Initialize the keyboard settings list */
    if (USetupData.KeyboardList == NULL)
    {
        USetupData.KeyboardList = CreateKeyboardDriverList(USetupData.SetupInf);
        if (USetupData.KeyboardList == NULL)
        {
            MUIDisplayError(ERROR_LOAD_KEYBOARD, Ir, POPUP_WAIT_ENTER);
            return QUIT_PAGE;
        }
    }
 
    /* Initialize the keyboard layout list */
    if (!USetupData.LayoutList)
    {
        USetupData.LayoutList = CreateKeyboardLayoutList(USetupData.SetupInf, SelectedLanguageId, DefaultKBLayout);
        if (!USetupData.LayoutList)
        {
            /* FIXME: report error */
            MUIDisplayError(ERROR_LOAD_KBLAYOUT, Ir, POPUP_WAIT_ENTER);
            return QUIT_PAGE;
        }
    }
 
    if (RepairUpdateFlag)
        return SELECT_PARTITION_PAGE;
 
    // if (IsUnattendedSetup)
        // return SELECT_PARTITION_PAGE;
 
    MUIDisplayPage(DEVICE_SETTINGS_PAGE);
 
    DrawGenericListCurrentItem(USetupData.ComputerList, GetSettingDescription, 25, 11);
    DrawGenericListCurrentItem(USetupData.DisplayList , GetSettingDescription, 25, 12);
    DrawGenericListCurrentItem(USetupData.KeyboardList, GetSettingDescription, 25, 13);
    DrawGenericListCurrentItem(USetupData.LayoutList  , GetSettingDescription, 25, 14);
 
    CONSOLE_InvertTextXY(24, Line, 48, 1);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_DOWN))  /* DOWN */
        {
            CONSOLE_NormalTextXY(24, Line, 48, 1);
 
            if (Line == 14)
                Line = 16;
            else if (Line == 16)
                Line = 11;
            else
                Line++;
 
            CONSOLE_InvertTextXY(24, Line, 48, 1);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_UP))  /* UP */
        {
            CONSOLE_NormalTextXY(24, Line, 48, 1);
 
            if (Line == 11)
                Line = 16;
            else if (Line == 16)
                Line = 14;
            else
                Line--;
 
            CONSOLE_InvertTextXY(24, Line, 48, 1);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
        {
            if (Line == 11)
                return COMPUTER_SETTINGS_PAGE;
            else if (Line == 12)
                return DISPLAY_SETTINGS_PAGE;
            else if (Line == 13)
                return KEYBOARD_SETTINGS_PAGE;
            else if (Line == 14)
                return LAYOUT_SETTINGS_PAGE;
            else if (Line == 16)
                return SELECT_PARTITION_PAGE;
        }
    }
 
    return DEVICE_SETTINGS_PAGE;
}
 
 
/*
 * Handles generic selection lists.
 *
 * PARAMS
 * GenericList: The list to handle.
 * nextPage: The page it needs to jump to after this page.
 * Ir: The PINPUT_RECORD
 */
static PAGE_NUMBER
HandleGenericList(PGENERIC_LIST_UI ListUi,
                  PAGE_NUMBER nextPage,
                  PINPUT_RECORD Ir)
{
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_DOWN))  /* DOWN */
        {
            ScrollDownGenericList(ListUi);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_UP))  /* UP */
        {
            ScrollUpGenericList(ListUi);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_NEXT))  /* PAGE DOWN */
        {
            ScrollPageDownGenericList(ListUi);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_PRIOR))  /* PAGE UP */
        {
            ScrollPageUpGenericList(ListUi);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            RedrawGenericList(ListUi);
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE)  /* ESC */
        {
            RestoreGenericListUiState(ListUi);
            return nextPage;    // Use some "prevPage;" instead?
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
        {
            return nextPage;
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar > 0x60) && (Ir->Event.KeyEvent.uChar.AsciiChar < 0x7b))
        {
            /* a-z */
            GenericListKeyPress(ListUi, Ir->Event.KeyEvent.uChar.AsciiChar);
        }
    }
}
 
 
/*
 * Displays the ComputerSettingsPage.
 *
 * Next pages:
 *  DeviceSettingsPage
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
ComputerSettingsPage(PINPUT_RECORD Ir)
{
    GENERIC_LIST_UI ListUi;
    MUIDisplayPage(COMPUTER_SETTINGS_PAGE);
 
    InitGenericListUi(&ListUi, USetupData.ComputerList, GetSettingDescription);
    DrawGenericList(&ListUi,
                    2, 18,
                    xScreen - 3,
                    yScreen - 3);
 
    return HandleGenericList(&ListUi, DEVICE_SETTINGS_PAGE, Ir);
}
 
 
/*
 * Displays the DisplaySettingsPage.
 *
 * Next pages:
 *  DeviceSettingsPage
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
DisplaySettingsPage(PINPUT_RECORD Ir)
{
    GENERIC_LIST_UI ListUi;
    MUIDisplayPage(DISPLAY_SETTINGS_PAGE);
 
    InitGenericListUi(&ListUi, USetupData.DisplayList, GetSettingDescription);
    DrawGenericList(&ListUi,
                    2, 18,
                    xScreen - 3,
                    yScreen - 3);
 
    return HandleGenericList(&ListUi, DEVICE_SETTINGS_PAGE, Ir);
}
 
 
/*
 * Displays the KeyboardSettingsPage.
 *
 * Next pages:
 *  DeviceSettingsPage
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
KeyboardSettingsPage(PINPUT_RECORD Ir)
{
    GENERIC_LIST_UI ListUi;
    MUIDisplayPage(KEYBOARD_SETTINGS_PAGE);
 
    InitGenericListUi(&ListUi, USetupData.KeyboardList, GetSettingDescription);
    DrawGenericList(&ListUi,
                    2, 18,
                    xScreen - 3,
                    yScreen - 3);
 
    return HandleGenericList(&ListUi, DEVICE_SETTINGS_PAGE, Ir);
}
 
 
/*
 * Displays the LayoutSettingsPage.
 *
 * Next pages:
 *  DeviceSettingsPage
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
LayoutSettingsPage(PINPUT_RECORD Ir)
{
    GENERIC_LIST_UI ListUi;
    MUIDisplayPage(LAYOUT_SETTINGS_PAGE);
 
    InitGenericListUi(&ListUi, USetupData.LayoutList, GetSettingDescription);
    DrawGenericList(&ListUi,
                    2, 18,
                    xScreen - 3,
                    yScreen - 3);
 
    return HandleGenericList(&ListUi, DEVICE_SETTINGS_PAGE, Ir);
}
 
 
static BOOLEAN
IsMediumLargeEnough(
    _In_ ULONGLONG SizeInBytes)
{
    /* Retrieve the maximum size in MB (rounded up) */
    ULONGLONG SizeInMB = RoundingDivide(SizeInBytes, MB);
 
    /* Check the medium size */
    if (SizeInMB < USetupData.RequiredPartitionDiskSpace)
    {
        DPRINT1("Partition/Volume is too small (size: %I64u MB), required space is %lu MB\n",
                SizeInMB, USetupData.RequiredPartitionDiskSpace);
        return FALSE;
    }
    return TRUE;
}
 
 
/*
 * Displays the SelectPartitionPage.
 *
 * Next pages:
 *  SelectFileSystemPage (At once if unattended)
 *  SelectFileSystemPage (Default if free space is selected)
 *  CreatePartitionPage
 *  ConfirmDeleteSystemPartitionPage (if the selected partition is the system partition, aka with the boot flag set)
 *  DeletePartitionPage
 *  QuitPage
 *
 * SIDEEFFECTS
 *  Set InstallShortcut (only if not unattended + free space is selected)
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
SelectPartitionPage(PINPUT_RECORD Ir)
{
    PARTLIST_UI ListUi;
    ULONG Error;
    ULONGLONG MaxTargetSize;
 
    if (PartitionList == NULL)
    {
        PartitionList = CreatePartitionList();
        if (PartitionList == NULL)
        {
            MUIDisplayError(ERROR_DRIVE_INFORMATION, Ir, POPUP_WAIT_ENTER);
            return QUIT_PAGE;
        }
        else if (IsListEmpty(&PartitionList->DiskListHead))
        {
            MUIDisplayError(ERROR_NO_HDD, Ir, POPUP_WAIT_ENTER);
            return QUIT_PAGE;
        }
    }
 
    if (RepairUpdateFlag)
    {
        ASSERT(CurrentInstallation);
 
        /* Determine the selected installation disk & partition.
         * It must exist and be valid, since this is the partition
         * where the existing installation already resides. */
        InstallPartition = SelectPartition(PartitionList,
                                           CurrentInstallation->DiskNumber,
                                           CurrentInstallation->PartitionNumber);
        if (!InstallPartition)
        {
            DPRINT1("RepairUpdateFlag == TRUE, SelectPartition() returned FALSE, assert!\n");
            ASSERT(FALSE);
        }
        ASSERT(InstallPartition->IsPartitioned);
        ASSERT(InstallPartition->Volume);
 
        return START_PARTITION_OPERATIONS_PAGE;
    }
 
    MUIDisplayPage(SELECT_PARTITION_PAGE);
 
    InitPartitionListUi(&ListUi, PartitionList,
                        CurrentPartition,
                        2, 21,
                        xScreen - 3,
                        yScreen - 3);
    DrawPartitionList(&ListUi);
 
    if (IsUnattendedSetup) do
    {
        /* If DestinationDiskNumber or DestinationPartitionNumber are invalid
         * (see below), don't select the partition and show the list instead */
        if (USetupData.DestinationDiskNumber == -1 ||
            USetupData.DestinationPartitionNumber == -1)
        {
            break;
        }
 
        /* Determine the selected installation disk & partition */
        CurrentPartition = SelectPartition(PartitionList,
                                           USetupData.DestinationDiskNumber,
                                           USetupData.DestinationPartitionNumber);
 
        /* Now reset DestinationDiskNumber and DestinationPartitionNumber
         * to *invalid* values, so that if the corresponding partition is
         * determined to be invalid by the code below or in CreateInstallPartition,
         * we don't reselect it when SelectPartitionPage() is called again */
        USetupData.DestinationDiskNumber = -1;
        USetupData.DestinationPartitionNumber = -1;
 
        // FIXME: Here and in the AutoPartition case below, the CurrentPartition
        // may actually be unsuitable (MBR-extended, non-simple volume...).
        // More checks need to be made here!
        //
        // NOTE: We don't check for CurrentPartition->Volume in case
        // the partition doesn't contain a recognized volume/none exists.
        // We also don't check whether IsPartitioned is TRUE, because if
        // the partition is still empty space, we'll try to partition it.
        if (CurrentPartition && !IsContainerPartition(CurrentPartition->PartitionType))
            goto CreateInstallPartition;
 
        if (USetupData.AutoPartition)
        {
            CurrentPartition = ListUi.CurrentPartition;
            // TODO: Do more checks, and loop until we find a valid partition.
            goto CreateInstallPartition;
        }
    } while (0);
 
    while (TRUE)
    {
        ULONG uID;
 
        CurrentPartition = ListUi.CurrentPartition;
 
        /* Update status text */
        if (!CurrentPartition)
        {
            // FIXME: If we get a NULL current partition, this means that
            // the current disk is of unrecognized type. So we should display
            // instead a status string to initialize the disk with one of
            // the recognized partitioning schemes (MBR, later: GPT, etc.)
            // For the time being we don't have that, so use instead another
            // known string.
            uID = STRING_INSTALLCREATEPARTITION;
        }
        else
        {
            if (CurrentPartition->IsPartitioned)
            {
                uID = STRING_INSTALLDELETEPARTITION;
                if (!CurrentPartition->LogicalPartition &&
                    IsContainerPartition(CurrentPartition->PartitionType))
                {
                    uID = STRING_DELETEPARTITION;
                }
            }
            else
            {
                uID = STRING_INSTALLCREATEPARTITION;
                if (CurrentPartition->LogicalPartition)
                    uID = STRING_INSTALLCREATELOGICAL;
            }
        }
        CONSOLE_SetStatusText(MUIGetString(uID));
 
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
            {
                DestroyPartitionList(PartitionList);
                PartitionList = NULL;
                return QUIT_PAGE;
            }
            return SELECT_PARTITION_PAGE;
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_DOWN))  /* DOWN */
        {
            ScrollUpDownPartitionList(&ListUi, TRUE);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_UP))  /* UP */
        {
            ScrollUpDownPartitionList(&ListUi, FALSE);
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_RETURN)  /* ENTER */
        {
            ASSERT(CurrentPartition);
 
            /* Don't select an extended partition for OS installation */
            if (IsContainerPartition(CurrentPartition->PartitionType))
                continue;
 
            /*
             * Check whether the user wants to install ReactOS on a disk that
             * is not recognized by the computer's firmware and if so, display
             * a warning since such disks may not be bootable.
             */
            if (CurrentPartition->DiskEntry->MediaType == FixedMedia &&
                !CurrentPartition->DiskEntry->BiosFound)
            {
                PopupError("The disk you have selected for installing ReactOS\n"
                           "is not visible by the firmware of your computer,\n"
                           "and so may not be bootable.\n"
                           "Press ENTER to continue anyway.",
                           MUIGetString(STRING_CONTINUE),
                           Ir, POPUP_WAIT_ENTER);
                // return SELECT_PARTITION_PAGE;
            }
 
            goto CreateInstallPartition;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == 'C')  /* C */
        {
            ASSERT(CurrentPartition);
 
            Error = PartitionCreateChecks(CurrentPartition, 0ULL, 0);
            if (Error != NOT_AN_ERROR)
            {
                MUIDisplayError(Error, Ir, POPUP_WAIT_ANY_KEY);
                return SELECT_PARTITION_PAGE;
            }
 
            PartCreateType = PartTypeData;
            return CREATE_PARTITION_PAGE;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == 'E')  /* E */
        {
            ASSERT(CurrentPartition);
 
            /* Don't create an extended partition within a logical partition */
            if (CurrentPartition->LogicalPartition)
                continue;
 
            Error = PartitionCreateChecks(CurrentPartition, 0ULL, PARTITION_EXTENDED);
            if (Error != NOT_AN_ERROR)
            {
                MUIDisplayError(Error, Ir, POPUP_WAIT_ANY_KEY);
                return SELECT_PARTITION_PAGE;
            }
 
            PartCreateType = PartTypeExtended;
            return CREATE_PARTITION_PAGE;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == 'D')  /* D */
        {
            ASSERT(CurrentPartition);
 
            /* Ignore deletion in case this is not a partitioned entry */
            if (!CurrentPartition->IsPartitioned)
                continue;
 
// TODO: Do something similar before trying to format the partition?
            if (CurrentPartition->Volume && !CurrentPartition->Volume->New &&
                (CurrentPartition->Volume->FormatState != Unformatted))
            {
                UNICODE_STRING CurrentPartitionU;
                WCHAR PathBuffer[RTL_NUMBER_OF_FIELD(VOLINFO, DeviceName) + 1];
 
                ASSERT(CurrentPartition->PartitionNumber != 0);
 
                RtlStringCchPrintfW(PathBuffer, _countof(PathBuffer),
                                    L"%s\\", CurrentPartition->Volume->Info.DeviceName);
                RtlInitUnicodeString(&CurrentPartitionU, PathBuffer);
 
                /*
                 * Check whether the user attempts to delete the partition on which
                 * the installation source is present. If so, fail with an error.
                 */
                // &USetupData.SourceRootPath
                if (RtlPrefixUnicodeString(&CurrentPartitionU, &USetupData.SourcePath, TRUE))
                {
                    MUIDisplayError(ERROR_SOURCE_PATH, Ir, POPUP_WAIT_ENTER);
                    return SELECT_PARTITION_PAGE;
                }
            }
 
            if (CurrentPartition == PartitionList->SystemPartition ||
                IsPartitionActive(CurrentPartition))
            {
                return CONFIRM_DELETE_SYSTEM_PARTITION_PAGE;
            }
 
            return DELETE_PARTITION_PAGE;
        }
    }
 
CreateInstallPartition:
    ASSERT(CurrentPartition);
    ASSERT(!IsContainerPartition(CurrentPartition->PartitionType));
 
    /* Create the partition if the selected region is empty */
    if (!CurrentPartition->IsPartitioned)
    {
        Error = PartitionCreateChecks(CurrentPartition, 0ULL, 0);
        if (Error != NOT_AN_ERROR)
        {
            MUIDisplayError(Error, Ir, POPUP_WAIT_ANY_KEY);
            return SELECT_PARTITION_PAGE;
        }
 
        /* Automatically create the partition on the whole empty space;
         * it will be formatted later with default parameters */
        CreatePartition(PartitionList,
                        CurrentPartition,
                        0ULL,
                        0);
        ASSERT(CurrentPartition->IsPartitioned);
        if (CurrentPartition->Volume)
            CurrentPartition->Volume->New |= VOLUME_NEW_AUTOCREATE;
    }
 
    /* Verify the target medium size */
    MaxTargetSize = GetPartEntrySizeInBytes(CurrentPartition);
    if (!IsMediumLargeEnough(MaxTargetSize))
    {
        MUIDisplayError(ERROR_INSUFFICIENT_PARTITION_SIZE, Ir, POPUP_WAIT_ANY_KEY,
                        USetupData.RequiredPartitionDiskSpace);
        return SELECT_PARTITION_PAGE; /* Let the user select another partition */
    }
 
    InstallPartition = CurrentPartition;
    return START_PARTITION_OPERATIONS_PAGE;
}
 
 
#define PARTITION_SIZE_INPUT_FIELD_LENGTH 9
/* Restriction for MaxSize */
#define PARTITION_MAXSIZE (pow(10, (PARTITION_SIZE_INPUT_FIELD_LENGTH - 1)) - 1)
 
static VOID
ShowPartitionSizeInputBox(SHORT Left,
                          SHORT Top,
                          SHORT Right,
                          SHORT Bottom,
                          ULONG MaxSize,
                          PWSTR InputBuffer,
                          PBOOLEAN Quit,
                          PBOOLEAN Cancel)
{
    INPUT_RECORD Ir;
    COORD coPos;
    DWORD Written;
    CHAR Buffer[128];
    INT Length, Pos;
    WCHAR ch;
    SHORT iLeft;
    SHORT iTop;
 
    if (Quit != NULL)
        *Quit = FALSE;
 
    if (Cancel != NULL)
        *Cancel = FALSE;
 
    DrawBox(Left, Top, Right - Left + 1, Bottom - Top + 1);
 
    /* Print message */
    coPos.X = Left + 2;
    coPos.Y = Top + 2;
    strcpy(Buffer, MUIGetString(STRING_PARTITIONSIZE));
    iLeft = coPos.X + (USHORT)strlen(Buffer) + 1;
    iTop = coPos.Y;
 
    WriteConsoleOutputCharacterA(StdOutput,
                                 Buffer,
                                 strlen(Buffer),
                                 coPos,
                                 &Written);
 
    sprintf(Buffer, MUIGetString(STRING_MAXSIZE), MaxSize);
    coPos.X = iLeft + PARTITION_SIZE_INPUT_FIELD_LENGTH + 1;
    coPos.Y = iTop;
    WriteConsoleOutputCharacterA(StdOutput,
                                 Buffer,
                                 strlen(Buffer),
                                 coPos,
                                 &Written);
 
    swprintf(InputBuffer, L"%lu", MaxSize);
    Length = wcslen(InputBuffer);
    Pos = Length;
    CONSOLE_SetInputTextXY(iLeft,
                           iTop,
                           PARTITION_SIZE_INPUT_FIELD_LENGTH,
                           InputBuffer);
    CONSOLE_SetCursorXY(iLeft + Length, iTop);
    CONSOLE_SetCursorType(TRUE, TRUE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(&Ir);
 
        if ((Ir.Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir.Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (Quit != NULL)
                *Quit = TRUE;
 
            InputBuffer[0] = UNICODE_NULL;
            break;
        }
        else if (Ir.Event.KeyEvent.wVirtualKeyCode == VK_RETURN)    /* ENTER */
        {
            break;
        }
        else if (Ir.Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE)    /* ESC */
        {
            if (Cancel != NULL)
                *Cancel = TRUE;
 
            InputBuffer[0] = UNICODE_NULL;
            break;
        }
        else if ((Ir.Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir.Event.KeyEvent.wVirtualKeyCode == VK_HOME))  /* HOME */
        {
            Pos = 0;
            CONSOLE_SetCursorXY(iLeft + Pos, iTop);
        }
        else if ((Ir.Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir.Event.KeyEvent.wVirtualKeyCode == VK_END))  /* END */
        {
            Pos = Length;
            CONSOLE_SetCursorXY(iLeft + Pos, iTop);
        }
        else if ((Ir.Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir.Event.KeyEvent.wVirtualKeyCode == VK_LEFT))  /* LEFT */
        {
            if (Pos > 0)
            {
                Pos--;
                CONSOLE_SetCursorXY(iLeft + Pos, iTop);
            }
        }
        else if ((Ir.Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir.Event.KeyEvent.wVirtualKeyCode == VK_RIGHT))  /* RIGHT */
        {
            if (Pos < Length)
            {
                Pos++;
                CONSOLE_SetCursorXY(iLeft + Pos, iTop);
            }
        }
        else if ((Ir.Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir.Event.KeyEvent.wVirtualKeyCode == VK_DELETE))  /* DEL */
        {
            if (Pos < Length)
            {
                memmove(&InputBuffer[Pos],
                        &InputBuffer[Pos + 1],
                        (Length - Pos - 1) * sizeof(WCHAR));
                InputBuffer[Length - 1] = UNICODE_NULL;
 
                Length--;
                CONSOLE_SetInputTextXY(iLeft,
                                       iTop,
                                       PARTITION_SIZE_INPUT_FIELD_LENGTH,
                                       InputBuffer);
                CONSOLE_SetCursorXY(iLeft + Pos, iTop);
            }
        }
        else if (Ir.Event.KeyEvent.wVirtualKeyCode == VK_BACK)  /* BACKSPACE */
        {
            if (Pos > 0)
            {
                if (Pos < Length)
                    memmove(&InputBuffer[Pos - 1],
                            &InputBuffer[Pos],
                            (Length - Pos) * sizeof(WCHAR));
                InputBuffer[Length - 1] = UNICODE_NULL;
 
                Pos--;
                Length--;
                CONSOLE_SetInputTextXY(iLeft,
                                       iTop,
                                       PARTITION_SIZE_INPUT_FIELD_LENGTH,
                                       InputBuffer);
                CONSOLE_SetCursorXY(iLeft + Pos, iTop);
            }
        }
        else if (Ir.Event.KeyEvent.uChar.AsciiChar != 0x00)
        {
            if (Length < PARTITION_SIZE_INPUT_FIELD_LENGTH - 1)
            {
                ch = (WCHAR)Ir.Event.KeyEvent.uChar.AsciiChar;
 
                if ((ch >= L'0') && (ch <= L'9'))
                {
                    if (Pos < Length)
                        memmove(&InputBuffer[Pos + 1],
                                &InputBuffer[Pos],
                                (Length - Pos) * sizeof(WCHAR));
                    InputBuffer[Length + 1] = UNICODE_NULL;
                    InputBuffer[Pos] = ch;
 
                    Pos++;
                    Length++;
                    CONSOLE_SetInputTextXY(iLeft,
                                           iTop,
                                           PARTITION_SIZE_INPUT_FIELD_LENGTH,
                                           InputBuffer);
                    CONSOLE_SetCursorXY(iLeft + Pos, iTop);
                }
            }
        }
    }
 
    CONSOLE_SetCursorType(TRUE, FALSE);
}
 
 
/*
 * Displays the CreatePartitionPage.
 *
 * Next pages:
 *  SelectPartitionPage
 *  SelectFileSystemPage (default)
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
CreatePartitionPage(PINPUT_RECORD Ir)
{
    PPARTENTRY PartEntry;
    PDISKENTRY DiskEntry;
    ULONG uID;
    ULONG MaxSize;
    ULONGLONG MaxPartSize, PartSize;
    BOOLEAN Quit, Cancel;
    WCHAR InputBuffer[50];
    CHAR LineBuffer[100];
 
    if (PartitionList == NULL || CurrentPartition == NULL)
    {
        /* FIXME: show an error dialog */
        return QUIT_PAGE;
    }
 
    if (PartCreateType == PartTypeData)
    {
        uID = STRING_CHOOSE_NEW_PARTITION;
        if (CurrentPartition->LogicalPartition)
            uID = STRING_CHOOSE_NEW_LOGICAL_PARTITION;
    }
    else // if (PartCreateType == PartTypeExtended)
    {
        uID = STRING_CHOOSE_NEW_EXTENDED_PARTITION;
    }
 
    CONSOLE_SetTextXY(6, 8, MUIGetString(uID));
 
    PartEntry = CurrentPartition;
    DiskEntry = CurrentPartition->DiskEntry;
 
    DiskDescription(DiskEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_PrintTextXY(6, 10, MUIGetString(STRING_HDDISK1),
                        LineBuffer);
 
    CONSOLE_SetTextXY(6, 12, MUIGetString(STRING_HDPARTSIZE));
 
    CONSOLE_SetStatusText(MUIGetString(STRING_CREATEPARTITION));
 
    MaxPartSize = GetPartEntrySizeInBytes(PartEntry);
 
    while (TRUE)
    {
        /* Retrieve the maximum size in MB (rounded up)
         * and cap it with what the user can enter */
        MaxSize = (ULONG)RoundingDivide(MaxPartSize, MB);
        MaxSize = min(MaxSize, PARTITION_MAXSIZE);
 
        ShowPartitionSizeInputBox(12, 14, xScreen - 12, 17,
                                  MaxSize, InputBuffer, &Quit, &Cancel);
        if (Quit)
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Cancel)
        {
            return SELECT_PARTITION_PAGE;
        }
 
        PartSize = _wcstoui64(InputBuffer, NULL, 10);
 
        /* Retry if too small or too large */
        if ((PartSize < 1) || (PartSize > MaxSize))
            continue;
 
        /*
         * If the input size, given in MB, specifies the maximum partition
         * size, it may slightly under- or over-estimate the latter due to
         * rounding error. In this case, use all of the unpartitioned space.
         * Otherwise, directly convert the size to bytes.
         */
        if (PartSize == MaxSize)
            PartSize = MaxPartSize;
        else // if (PartSize < MaxSize)
            PartSize *= MB;
        DPRINT("Partition size: %I64u bytes\n", PartSize);
 
        ASSERT(PartSize <= MaxPartSize);
 
        CreatePartition(PartitionList,
                        CurrentPartition,
                        PartSize,
                        (PartCreateType == PartTypeData)
                            ? 0
                     // (PartCreateType == PartTypeExtended)
                            : PARTITION_EXTENDED);
 
        return SELECT_PARTITION_PAGE;
    }
 
    return CREATE_PARTITION_PAGE;
}
 
 
/*
 * Displays the ConfirmDeleteSystemPartitionPage.
 *
 * Next pages:
 *  DeletePartitionPage (default)
 *  SelectPartitionPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
ConfirmDeleteSystemPartitionPage(PINPUT_RECORD Ir)
{
    MUIDisplayPage(CONFIRM_DELETE_SYSTEM_PARTITION_PAGE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_RETURN) /* ENTER */
        {
            return DELETE_PARTITION_PAGE;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE)  /* ESC */
        {
            return SELECT_PARTITION_PAGE;
        }
    }
 
    return CONFIRM_DELETE_SYSTEM_PARTITION_PAGE;
}
 
 
/*
 * Displays the DeletePartitionPage.
 *
 * Next pages:
 *  SelectPartitionPage (default)
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
DeletePartitionPage(PINPUT_RECORD Ir)
{
    PPARTENTRY PartEntry;
    PDISKENTRY DiskEntry;
    CHAR LineBuffer[100];
 
    if (PartitionList == NULL || CurrentPartition == NULL)
    {
        /* FIXME: show an error dialog */
        return QUIT_PAGE;
    }
 
    PartEntry = CurrentPartition;
    DiskEntry = CurrentPartition->DiskEntry;
 
    MUIDisplayPage(DELETE_PARTITION_PAGE);
 
    PartitionDescription(PartEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_SetTextXY(6, 10, LineBuffer);
 
    DiskDescription(DiskEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_PrintTextXY(6, 12, MUIGetString(STRING_HDDISK2),
                        LineBuffer);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE)  /* ESC */
        {
            return SELECT_PARTITION_PAGE;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == 'L') /* L */
        {
            DeletePartition(PartitionList,
                            CurrentPartition,
                            &CurrentPartition);
            return SELECT_PARTITION_PAGE;
        }
    }
 
    return DELETE_PARTITION_PAGE;
}
 
 
/*
 * Displays the SelectFileSystemPage.
 *
 * Next pages:
 *  CheckFileSystemPage (At once if RepairUpdate is selected)
 *  CheckFileSystemPage (At once if Unattended and not USetupData.FormatPartition)
 *  FormatPartitionPage (Default, at once if Unattended and USetupData.FormatPartition)
 *  SelectPartitionPage (If the user aborts)
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
// PFSVOL_CALLBACK
static FSVOL_OP
CALLBACK
FsVolCallback(
    _In_opt_ PVOID Context,
    _In_ FSVOLNOTIFY FormatStatus,
    _In_ ULONG_PTR Param1,
    _In_ ULONG_PTR Param2);
 
typedef struct _FSVOL_CONTEXT
{
    PINPUT_RECORD Ir;
    PAGE_NUMBER NextPageOnAbort;
} FSVOL_CONTEXT, *PFSVOL_CONTEXT;
 
static PAGE_NUMBER
StartPartitionOperationsPage(PINPUT_RECORD Ir)
{
    FSVOL_CONTEXT FsVolContext = {Ir, QUIT_PAGE};
    BOOLEAN Success;
 
    if (PartitionList == NULL || InstallPartition == NULL)
    {
        /* FIXME: show an error dialog */
        return QUIT_PAGE;
    }
 
    /* Find or set the active system partition before starting formatting */
    Success = InitSystemPartition(PartitionList,
                                  InstallPartition,
                                  &SystemPartition,
                                  FsVolCallback,
                                  &FsVolContext);
    if (!Success)
        return FsVolContext.NextPageOnAbort;
    //
    // FIXME?? If cannot use any system partition, install FreeLdr on floppy / removable media??
    //
 
    /* Set the AUTOCREATE flag if the system partition was automatically created */
    if (SystemPartition->New && SystemVolume)
        SystemVolume->New |= VOLUME_NEW_AUTOCREATE;
 
    CONSOLE_ClearScreen();
    CONSOLE_Flush();
 
    /* Apply all pending operations on partitions: formatting and checking */
    Success = FsVolCommitOpsQueue(PartitionList,
                                  SystemVolume,
                                  InstallVolume,
                                  FsVolCallback,
                                  &FsVolContext);
    if (!Success)
        return FsVolContext.NextPageOnAbort;
    return BOOTLOADER_SELECT_PAGE;
}
 
static BOOLEAN
ChangeSystemPartitionPage(
    IN PINPUT_RECORD Ir,
    IN PPARTENTRY SystemPartition)
{
    PPARTENTRY PartEntry;
    PDISKENTRY DiskEntry;
    CHAR LineBuffer[100];
 
    // CONSOLE_ClearScreen();
    // CONSOLE_Flush();
    MUIDisplayPage(CHANGE_SYSTEM_PARTITION);
 
    PartEntry = PartitionList->SystemPartition;
    DiskEntry = PartEntry->DiskEntry;
 
    PartitionDescription(PartEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_SetTextXY(8, 10, LineBuffer);
 
    DiskDescription(DiskEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_PrintTextXY(8, 14, MUIGetString(STRING_HDDISK1),
                        LineBuffer);
 
 
    PartEntry = SystemPartition;
    DiskEntry = PartEntry->DiskEntry;
 
    PartitionDescription(PartEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_SetTextXY(8, 23, LineBuffer);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_RETURN) /* ENTER */
            break;
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE)  /* ESC */
            return FALSE;
    }
 
    return TRUE;
}
 
static VOID
ResetFileSystemList(VOID)
{
    if (!FileSystemList)
        return;
 
    DestroyFileSystemList(FileSystemList);
    FileSystemList = NULL;
}
 
static FSVOL_OP
SelectFileSystemPage(
    _In_ PFSVOL_CONTEXT FsVolContext,
    _In_ PVOLENTRY Volume)
{
    PINPUT_RECORD Ir = FsVolContext->Ir;
    PPARTENTRY PartEntry = Volume->PartEntry;
    PDISKENTRY DiskEntry = PartEntry->DiskEntry;
    PCWSTR DefaultFs;
    BOOLEAN ForceFormat;
    CHAR LineBuffer[100];
 
    DPRINT("SelectFileSystemPage()\n");
 
    ForceFormat = (Volume->New || Volume->FormatState == Unformatted);
 
Restart:
    /* Reset the file system list for each volume that is to be formatted */
    ResetFileSystemList();
 
    CONSOLE_ClearScreen();
    CONSOLE_Flush();
    MUIDisplayPage(SELECT_FILE_SYSTEM_PAGE);
 
    if (Volume->New & VOLUME_NEW_AUTOCREATE)
    {
        Volume->New &= ~VOLUME_NEW_AUTOCREATE;
 
        CONSOLE_SetTextXY(6, 8, MUIGetString(STRING_NEWPARTITION));
    }
    else if (Volume->New)
    {
        ULONG uID;
 
        if (Volume == SystemVolume)
            uID = STRING_NONFORMATTEDSYSTEMPART;
        else if (Volume == InstallVolume)
            uID = STRING_NONFORMATTEDPART;
        else
            uID = STRING_NONFORMATTEDOTHERPART;
 
        CONSOLE_SetTextXY(6, 8, MUIGetString(uID));
    }
    else
    {
        CONSOLE_SetTextXY(6, 8, MUIGetString(STRING_INSTALLONPART));
    }
 
    PartitionDescription(PartEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_SetTextXY(6, 10, LineBuffer);
 
    DiskDescription(DiskEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_PrintTextXY(6, 12, MUIGetString(STRING_HDDISK2),
                        LineBuffer);
 
    /* Show "This Partition will be formatted next" only if it is unformatted */
    if (ForceFormat)
        CONSOLE_SetTextXY(6, 14, MUIGetString(STRING_PARTFORMAT));
 
    ASSERT(!FileSystemList);
 
    if (IsUnattendedSetup)
    {
        ASSERT(USetupData.FormatPartition);
 
        switch (USetupData.FsType)
        {
            /* 1 is for BtrFS */
            case 1:
                DefaultFs = L"BTRFS";
                break;
 
            /* If we don't understand input, default to FAT */
            default:
                DefaultFs = L"FAT";
                break;
        }
    }
    else
    {
        /* By default select the "FAT" file system */
        DefaultFs = L"FAT";
    }
 
    /* Create the file system list */
    // TODO: Display only the FSes compatible with the selected volume!
    FileSystemList = CreateFileSystemList(6, 26, ForceFormat, DefaultFs);
    if (!FileSystemList)
    {
        /* FIXME: show an error dialog */
        FsVolContext->NextPageOnAbort = QUIT_PAGE;
        return FSVOL_ABORT;
    }
 
    if (IsUnattendedSetup)
    {
        ASSERT(USetupData.FormatPartition);
        return FSVOL_DOIT;
    }
 
    DrawFileSystemList(FileSystemList);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
            {
                FsVolContext->NextPageOnAbort = QUIT_PAGE;
                return FSVOL_ABORT;
            }
            break;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE)  /* ESC */
        {
            FsVolContext->NextPageOnAbort = SELECT_PARTITION_PAGE;
            return FSVOL_ABORT;
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_DOWN))  /* DOWN */
        {
            ScrollDownFileSystemList(FileSystemList);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_UP))  /* UP */
        {
            ScrollUpFileSystemList(FileSystemList);
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_RETURN) /* ENTER */
        {
            if (!FileSystemList->Selected->FileSystem)
            {
                /* The 'Keep existing filesystem' entry was chosen,
                 * the volume must be already formatted */
                ASSERT(!ForceFormat);
 
                /* Skip formatting this volume. We will also ignore
                 * file system checks on it, unless it is either the
                 * system or the installation volume. */
                if ((Volume != SystemVolume) && (Volume != InstallVolume))
                    Volume->NeedsCheck = FALSE;
 
                return FSVOL_SKIP;
            }
            else
            {
                /* Format this volume */
                return FSVOL_DOIT;
            }
        }
    }
 
    goto Restart;
}
 
static FSVOL_OP
FormatPartitionPage(
    _In_ PFSVOL_CONTEXT FsVolContext,
    _In_ PVOLENTRY Volume)
{
    PINPUT_RECORD Ir = FsVolContext->Ir;
    PPARTENTRY PartEntry = Volume->PartEntry;
    PDISKENTRY DiskEntry = PartEntry->DiskEntry;
    CHAR LineBuffer[100];
 
Restart:
    CONSOLE_ClearScreen();
    CONSOLE_Flush();
    MUIDisplayPage(FORMAT_PARTITION_PAGE);
 
    PartitionDescription(PartEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_SetTextXY(6, 10, LineBuffer);
 
    DiskDescription(DiskEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_PrintTextXY(6, 12, MUIGetString(STRING_HDDISK2),
                        LineBuffer);
 
    while (TRUE)
    {
        if (!IsUnattendedSetup)
            CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
            {
                FsVolContext->NextPageOnAbort = QUIT_PAGE;
                return FSVOL_ABORT;
            }
            goto Restart;
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_RETURN || IsUnattendedSetup) /* ENTER */
        {
            /*
             * Remove the "Press ENTER to continue" message prompt when the ENTER
             * key is pressed as the user wants to begin the partition formatting.
             */
            MUIClearStyledText(FORMAT_PARTITION_PAGE, TEXT_ID_FORMAT_PROMPT, TEXT_TYPE_REGULAR);
            CONSOLE_SetStatusText(MUIGetString(STRING_PLEASEWAIT));
 
            return FSVOL_DOIT;
        }
    }
}
 
static VOID
CheckFileSystemPage(
    _In_ PVOLENTRY Volume)
{
    PPARTENTRY PartEntry = Volume->PartEntry;
    PDISKENTRY DiskEntry = PartEntry->DiskEntry;
    CHAR LineBuffer[100];
 
    CONSOLE_ClearScreen();
    CONSOLE_Flush();
    MUIDisplayPage(CHECK_FILE_SYSTEM_PAGE);
 
    PartitionDescription(PartEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_SetTextXY(6, 10, LineBuffer);
 
    DiskDescription(DiskEntry, LineBuffer, ARRAYSIZE(LineBuffer));
    CONSOLE_PrintTextXY(6, 12, MUIGetString(STRING_HDDISK2),
                        LineBuffer);
}
 
// PFSVOL_CALLBACK
static FSVOL_OP
CALLBACK
FsVolCallback(
    _In_opt_ PVOID Context,
    _In_ FSVOLNOTIFY FormatStatus,
    _In_ ULONG_PTR Param1,
    _In_ ULONG_PTR Param2)
{
    PFSVOL_CONTEXT FsVolContext = (PFSVOL_CONTEXT)Context;
    PINPUT_RECORD Ir = FsVolContext->Ir;
 
    switch (FormatStatus)
    {
    // FIXME: Deprecate!
    case ChangeSystemPartition:
    {
        PPARTENTRY SystemPartition = (PPARTENTRY)Param1;
 
        FsVolContext->NextPageOnAbort = SELECT_PARTITION_PAGE;
        if (ChangeSystemPartitionPage(Ir, SystemPartition))
            return FSVOL_DOIT;
        return FSVOL_ABORT;
    }
 
    case FSVOLNOTIFY_PARTITIONERROR:
    {
        switch (Param1)
        {
        case STATUS_PARTITION_FAILURE:
        {
            MUIDisplayError(ERROR_WRITE_PTABLE, Ir, POPUP_WAIT_ENTER);
            FsVolContext->NextPageOnAbort = QUIT_PAGE;
            break;
        }
 
        case ERROR_SYSTEM_PARTITION_NOT_FOUND:
        {
            /* FIXME: improve the error dialog */
            //
            // Error dialog should say that we cannot find a suitable
            // system partition and create one on the system. At this point,
            // it may be nice to ask the user whether he wants to continue,
            // or use an external drive as the system drive/partition
            // (e.g. floppy, USB drive, etc...)
            //
            PopupError("The ReactOS Setup could not find a supported system partition\n"
                       "on your system or could not create a new one. Without such a partition\n"
                       "the Setup program cannot install ReactOS.\n"
                       "Press ENTER to return to the partition selection list.",
                       MUIGetString(STRING_CONTINUE),
                       Ir, POPUP_WAIT_ENTER);
 
            FsVolContext->NextPageOnAbort = SELECT_PARTITION_PAGE;
            break;
        }
 
        default:
            break;
        }
        return FSVOL_ABORT;
    }
 
    case FSVOLNOTIFY_STARTQUEUE:
    case FSVOLNOTIFY_ENDQUEUE:
        // NOTE: If needed, clear screen and flush input.
        return FSVOL_DOIT;
 
    case FSVOLNOTIFY_STARTSUBQUEUE:
    {
        if ((FSVOL_OP)Param1 == FSVOL_FORMAT)
        {
            /*
             * In case we just repair an existing installation, or make
             * an unattended setup without formatting, just go to the
             * file system check step.
             */
            if (RepairUpdateFlag)
                return FSVOL_SKIP; 
            if (IsUnattendedSetup && !USetupData.FormatPartition)
                return FSVOL_SKIP; 
        }
        return FSVOL_DOIT;
    }
 
    case FSVOLNOTIFY_ENDSUBQUEUE:
        return 0;
 
    case FSVOLNOTIFY_FORMATERROR:
    {
        PFORMAT_VOLUME_INFO FmtInfo = (PFORMAT_VOLUME_INFO)Param1;
        CHAR Buffer[MAX_PATH];
 
        // FIXME: See also FSVOLNOTIFY_PARTITIONERROR
        if (FmtInfo->ErrorStatus == STATUS_PARTITION_FAILURE)
        {
            MUIDisplayError(ERROR_WRITE_PTABLE, Ir, POPUP_WAIT_ENTER);
            FsVolContext->NextPageOnAbort = QUIT_PAGE;
            return FSVOL_ABORT;
        }
        else
        if (FmtInfo->ErrorStatus == STATUS_UNRECOGNIZED_VOLUME)
        {
            /* FIXME: show an error dialog */
            // MUIDisplayError(ERROR_FORMATTING_PARTITION, Ir, POPUP_WAIT_ANY_KEY,
            //                 FmtInfo->Volume->Info.DeviceName);
            FsVolContext->NextPageOnAbort = QUIT_PAGE;
            return FSVOL_ABORT;
        }
        else
        if (FmtInfo->ErrorStatus == STATUS_NOT_SUPPORTED)
        {
            RtlStringCbPrintfA(Buffer,
                               sizeof(Buffer),
                               "Setup is currently unable to format a partition in %S.\n"
                               "\n"
                               "  \x07  Press ENTER to continue Setup.\n"
                               "  \x07  Press F3 to quit Setup.",
                               FmtInfo->FileSystemName);
 
            PopupError(Buffer,
                       MUIGetString(STRING_QUITCONTINUE),
                       NULL, POPUP_WAIT_NONE);
 
            while (TRUE)
            {
                CONSOLE_ConInKey(Ir);
 
                if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x00 &&
                    Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3)  /* F3 */
                {
                    if (ConfirmQuit(Ir))
                    {
                        FsVolContext->NextPageOnAbort = QUIT_PAGE;
                        return FSVOL_ABORT;
                    }
                    return FSVOL_RETRY;
                }
                else if (Ir->Event.KeyEvent.uChar.AsciiChar == VK_RETURN) /* ENTER */
                {
                    return FSVOL_RETRY;
                }
            }
        }
        else if (!NT_SUCCESS(FmtInfo->ErrorStatus))
        {
            DPRINT1("FormatPartition() failed: Status 0x%08lx\n", FmtInfo->ErrorStatus);
            MUIDisplayError(ERROR_FORMATTING_PARTITION, Ir, POPUP_WAIT_ANY_KEY,
                            FmtInfo->Volume->Info.DeviceName);
            FsVolContext->NextPageOnAbort = QUIT_PAGE;
            return FSVOL_ABORT;
        }
        return FSVOL_RETRY;
    }
 
    case FSVOLNOTIFY_CHECKERROR:
    {
        PCHECK_VOLUME_INFO ChkInfo = (PCHECK_VOLUME_INFO)Param1;
        CHAR Buffer[MAX_PATH];
 
        if (ChkInfo->ErrorStatus == STATUS_NOT_SUPPORTED)
        {
            RtlStringCbPrintfA(Buffer,
                               sizeof(Buffer),
                               "Setup is currently unable to check a partition formatted in %S.\n"
                               "\n"
                               "  \x07  Press ENTER to continue Setup.\n"
                               "  \x07  Press F3 to quit Setup.",
                               ChkInfo->Volume->Info.FileSystem);
 
            PopupError(Buffer,
                       MUIGetString(STRING_QUITCONTINUE),
                       NULL, POPUP_WAIT_NONE);
 
            while (TRUE)
            {
                CONSOLE_ConInKey(Ir);
 
                if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x00 &&
                    Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3)  /* F3 */
                {
                    if (ConfirmQuit(Ir))
                    {
                        FsVolContext->NextPageOnAbort = QUIT_PAGE;
                        return FSVOL_ABORT;
                    }
                    return FSVOL_SKIP;
                }
                else if (Ir->Event.KeyEvent.uChar.AsciiChar == VK_RETURN) /* ENTER */
                {
                    return FSVOL_SKIP;
                }
            }
        }
        else if (!NT_SUCCESS(ChkInfo->ErrorStatus))
        {
            DPRINT1("ChkdskPartition() failed: Status 0x%08lx\n", ChkInfo->ErrorStatus);
 
            RtlStringCbPrintfA(Buffer,
                               sizeof(Buffer),
                               "ChkDsk detected some disk errors.\n(Status 0x%08lx).\n",
                               ChkInfo->ErrorStatus);
 
            PopupError(Buffer,
                       MUIGetString(STRING_CONTINUE),
                       Ir, POPUP_WAIT_ENTER);
            return FSVOL_SKIP;
        }
        return FSVOL_SKIP;
    }
 
    case FSVOLNOTIFY_STARTFORMAT:
    {
        PFORMAT_VOLUME_INFO FmtInfo = (PFORMAT_VOLUME_INFO)Param1;
        FSVOL_OP Result;
 
        ASSERT((FSVOL_OP)Param2 == FSVOL_FORMAT);
 
        /* Select the file system */
        Result = SelectFileSystemPage(FsVolContext, FmtInfo->Volume);
        if (Result != FSVOL_DOIT)
            return Result;
 
        /* Display the formatting page */
        Result = FormatPartitionPage(FsVolContext, FmtInfo->Volume);
        if (Result != FSVOL_DOIT)
            return Result;
 
        StartFormat(FmtInfo, FileSystemList->Selected);
        return FSVOL_DOIT;
    }
 
    case FSVOLNOTIFY_ENDFORMAT:
    {
        PFORMAT_VOLUME_INFO FmtInfo = (PFORMAT_VOLUME_INFO)Param1;
        EndFormat(FmtInfo->ErrorStatus);
 
        /* Reset the file system list */
        ResetFileSystemList();
        return 0;
    }
 
    case FSVOLNOTIFY_STARTCHECK:
    {
        PCHECK_VOLUME_INFO ChkInfo = (PCHECK_VOLUME_INFO)Param1;
 
        ASSERT((FSVOL_OP)Param2 == FSVOL_CHECK);
 
        CheckFileSystemPage(ChkInfo->Volume);
        StartCheck(ChkInfo);
        return FSVOL_DOIT;
    }
 
    case FSVOLNOTIFY_ENDCHECK:
    {
        PCHECK_VOLUME_INFO ChkInfo = (PCHECK_VOLUME_INFO)Param1;
        EndCheck(ChkInfo->ErrorStatus);
        return 0;
    }
    }
 
    return 0;
}
 
 
/*
 * Displays the InstallDirectoryPage.
 *
 * Next pages:
 *  PrepareCopyPage
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
InstallDirectoryPage(PINPUT_RECORD Ir)
{
    NTSTATUS Status;
    ULONG Length, Pos;
    WCHAR c;
    WCHAR InstallDir[MAX_PATH];
 
    if (PartitionList == NULL || InstallPartition == NULL)
    {
        /* FIXME: show an error dialog */
        return QUIT_PAGE;
    }
 
    // if (IsUnattendedSetup)
    if (RepairUpdateFlag)
        wcscpy(InstallDir, CurrentInstallation->PathComponent); // SystemNtPath
    else if (USetupData.InstallationDirectory[0])
        wcscpy(InstallDir, USetupData.InstallationDirectory);
    else
        wcscpy(InstallDir, L"\\ReactOS");
 
    /*
     * Check the validity of the predefined 'InstallDir'. If we are either
     * in unattended setup or in update/repair mode, and the installation path
     * is valid, just perform the installation. Otherwise (either in the case
     * of an invalid path, or we are in regular setup), display the UI and allow
     * the user to specify a new installation path.
     */
    if (RepairUpdateFlag || IsUnattendedSetup)
    {
        /* Check for the validity of the installation directory and pop up
         * an error if it is not the case. Then the user can fix it. */
        if (IsValidInstallDirectory(InstallDir))
            goto InitInstallDir;
 
        MUIDisplayError(ERROR_DIRECTORY_NAME, Ir, POPUP_WAIT_ENTER);
    }
 
    Length = wcslen(InstallDir);
    Pos = Length;
 
    MUIDisplayPage(INSTALL_DIRECTORY_PAGE);
    CONSOLE_SetInputTextXY(8, 11, 51, InstallDir);
    CONSOLE_SetCursorXY(8 + Pos, 11);
    CONSOLE_SetCursorType(TRUE, TRUE);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            CONSOLE_SetCursorType(TRUE, FALSE);
 
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            return INSTALL_DIRECTORY_PAGE;
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_DELETE))  /* DEL */
        {
            if (Pos < Length)
            {
                memmove(&InstallDir[Pos],
                        &InstallDir[Pos + 1],
                        (Length - Pos - 1) * sizeof(WCHAR));
                InstallDir[Length - 1] = UNICODE_NULL;
 
                Length--;
                CONSOLE_SetInputTextXY(8, 11, 51, InstallDir);
                CONSOLE_SetCursorXY(8 + Pos, 11);
            }
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_HOME))  /* HOME */
        {
            Pos = 0;
            CONSOLE_SetCursorXY(8 + Pos, 11);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_END))  /* END */
        {
            Pos = Length;
            CONSOLE_SetCursorXY(8 + Pos, 11);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_LEFT))  /* LEFT */
        {
            if (Pos > 0)
            {
                Pos--;
                CONSOLE_SetCursorXY(8 + Pos, 11);
            }
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_RIGHT))  /* RIGHT */
        {
            if (Pos < Length)
            {
                Pos++;
                CONSOLE_SetCursorXY(8 + Pos, 11);
            }
        }
        else if (Ir->Event.KeyEvent.wVirtualKeyCode == VK_ESCAPE)  /* ESC */
        {
            /* Erase the whole line */
            *InstallDir = UNICODE_NULL;
            Pos = Length = 0;
            CONSOLE_SetInputTextXY(8, 11, 51, InstallDir);
            CONSOLE_SetCursorXY(8 + Pos, 11);
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
        {
            CONSOLE_SetCursorType(TRUE, FALSE);
 
            /* Check for the validity of the installation directory and pop up
             * an error if it is not the case. Then the user can fix it. */
            if (IsValidInstallDirectory(InstallDir))
                goto InitInstallDir;
 
            MUIDisplayError(ERROR_DIRECTORY_NAME, Ir, POPUP_WAIT_ENTER);
            return INSTALL_DIRECTORY_PAGE;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x08) /* BACKSPACE */
        {
            if (Pos > 0)
            {
                if (Pos < Length)
                    memmove(&InstallDir[Pos - 1],
                            &InstallDir[Pos],
                            (Length - Pos) * sizeof(WCHAR));
                InstallDir[Length - 1] = UNICODE_NULL;
 
                Pos--;
                Length--;
                CONSOLE_SetInputTextXY(8, 11, 51, InstallDir);
                CONSOLE_SetCursorXY(8 + Pos, 11);
            }
        }
        else if (isprint(Ir->Event.KeyEvent.uChar.AsciiChar))
        {
            if (Length < 50)
            {
                /* Only accept valid characters for the installation path */
                c = (WCHAR)Ir->Event.KeyEvent.uChar.AsciiChar;
                if (IS_VALID_INSTALL_PATH_CHAR(c))
                {
                    if (Pos < Length)
                        memmove(&InstallDir[Pos + 1],
                                &InstallDir[Pos],
                                (Length - Pos) * sizeof(WCHAR));
                    InstallDir[Length + 1] = UNICODE_NULL;
                    InstallDir[Pos] = c;
 
                    Pos++;
                    Length++;
                    CONSOLE_SetInputTextXY(8, 11, 51, InstallDir);
                    CONSOLE_SetCursorXY(8 + Pos, 11);
                }
            }
        }
    }
 
InitInstallDir:
    Status = InitDestinationPaths(&USetupData, InstallDir, InstallVolume);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("InitDestinationPaths() failed: Status 0x%lx\n", Status);
        MUIDisplayError(ERROR_NO_BUILD_PATH, Ir, POPUP_WAIT_ENTER);
        return QUIT_PAGE;
    }
 
    /*
     * Check whether the user attempts to install ReactOS within the
     * installation source directory, or in a subdirectory thereof.
     * If so, fail with an error.
     */
    if (RtlPrefixUnicodeString(&USetupData.SourcePath, &USetupData.DestinationPath, TRUE))
    {
        MUIDisplayError(ERROR_SOURCE_DIR, Ir, POPUP_WAIT_ENTER);
        return INSTALL_DIRECTORY_PAGE;
    }
 
    return PREPARE_COPY_PAGE;
}
 
 
/*
 * Displays the PrepareCopyPage.
 *
 * Next pages:
 *  FileCopyPage(At once)
 *  QuitPage
 *
 * SIDEEFFECTS
 * Calls PrepareFileCopy
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
PrepareCopyPage(PINPUT_RECORD Ir)
{
    // ERROR_NUMBER ErrorNumber;
    BOOLEAN Success;
 
    MUIDisplayPage(PREPARE_COPY_PAGE);
 
    /* ErrorNumber = */ Success = PrepareFileCopy(&USetupData, NULL);
    if (/*ErrorNumber != ERROR_SUCCESS*/ !Success)
    {
        // MUIDisplayError(ErrorNumber, Ir, POPUP_WAIT_ENTER);
        return QUIT_PAGE;
    }
 
    return FILE_COPY_PAGE;
}
 
typedef struct _COPYCONTEXT
{
    ULONG TotalOperations;
    ULONG CompletedOperations;
    PPROGRESSBAR ProgressBar;
    PPROGRESSBAR MemoryBars[4];
} COPYCONTEXT, *PCOPYCONTEXT;
 
static VOID
SetupUpdateMemoryInfo(IN PCOPYCONTEXT CopyContext,
                      IN BOOLEAN First)
{
    SYSTEM_PERFORMANCE_INFORMATION PerfInfo;
 
    /* Get the memory information from the system */
    NtQuerySystemInformation(SystemPerformanceInformation,
                             &PerfInfo,
                             sizeof(PerfInfo),
                             NULL);
 
    /* Check if this is initial setup */
    if (First)
    {
        /* Set maximum limits to be total RAM pages */
        ProgressSetStepCount(CopyContext->MemoryBars[0], PerfInfo.CommitLimit);
        ProgressSetStepCount(CopyContext->MemoryBars[1], PerfInfo.CommitLimit);
        ProgressSetStepCount(CopyContext->MemoryBars[2], PerfInfo.CommitLimit);
    }
 
    /* Set current values */
    ProgressSetStep(CopyContext->MemoryBars[0], PerfInfo.PagedPoolPages + PerfInfo.NonPagedPoolPages);
    ProgressSetStep(CopyContext->MemoryBars[1], PerfInfo.ResidentSystemCachePage);
    ProgressSetStep(CopyContext->MemoryBars[2], PerfInfo.AvailablePages);
}
 
static UINT
CALLBACK
FileCopyCallback(PVOID Context,
                 UINT Notification,
                 UINT_PTR Param1,
                 UINT_PTR Param2)
{
    PCOPYCONTEXT CopyContext = (PCOPYCONTEXT)Context;
    PFILEPATHS_W FilePathInfo;
    PCWSTR SrcFileName, DstFileName;
 
    switch (Notification)
    {
        case SPFILENOTIFY_STARTSUBQUEUE:
        {
            CopyContext->TotalOperations = (ULONG)Param2;
            CopyContext->CompletedOperations = 0;
            ProgressSetStepCount(CopyContext->ProgressBar,
                                 CopyContext->TotalOperations);
            SetupUpdateMemoryInfo(CopyContext, TRUE);
            break;
        }
 
        case SPFILENOTIFY_STARTDELETE:
        case SPFILENOTIFY_STARTRENAME:
        case SPFILENOTIFY_STARTCOPY:
        {
            FilePathInfo = (PFILEPATHS_W)Param1;
 
            if (Notification == SPFILENOTIFY_STARTDELETE)
            {
                /* Display delete message */
                ASSERT(Param2 == FILEOP_DELETE);
 
                DstFileName = wcsrchr(FilePathInfo->Target, L'\\');
                if (DstFileName) ++DstFileName;
                else DstFileName = FilePathInfo->Target;
 
                CONSOLE_SetStatusText(MUIGetString(STRING_DELETING),
                                      DstFileName);
            }
            else if (Notification == SPFILENOTIFY_STARTRENAME)
            {
                /* Display move/rename message */
                ASSERT(Param2 == FILEOP_RENAME);
 
                SrcFileName = wcsrchr(FilePathInfo->Source, L'\\');
                if (SrcFileName) ++SrcFileName;
                else SrcFileName = FilePathInfo->Source;
 
                DstFileName = wcsrchr(FilePathInfo->Target, L'\\');
                if (DstFileName) ++DstFileName;
                else DstFileName = FilePathInfo->Target;
 
                if (!_wcsicmp(SrcFileName, DstFileName))
                    Param2 = STRING_MOVING;
                else
                    Param2 = STRING_RENAMING;
 
                CONSOLE_SetStatusText(MUIGetString(Param2),
                                      SrcFileName, DstFileName);
            }
            else if (Notification == SPFILENOTIFY_STARTCOPY)
            {
                static PCSTR s_pszCopying = NULL; /* Cached for speed */
 
                /* Display copy message */
                ASSERT(Param2 == FILEOP_COPY);
 
                /* NOTE: When extracting from CABs the Source is the CAB name */
                DstFileName = wcsrchr(FilePathInfo->Target, L'\\');
                if (DstFileName) ++DstFileName;
                else DstFileName = FilePathInfo->Target;
 
                if (!s_pszCopying)
                    s_pszCopying = MUIGetString(STRING_COPYING);
                CONSOLE_SetStatusText(s_pszCopying, DstFileName);
#ifdef __REACTOS__ /* HACK */
                DoWatchDestFileName(DstFileName);
#endif
            }
 
            SetupUpdateMemoryInfo(CopyContext, FALSE);
            break;
        }
 
        case SPFILENOTIFY_COPYERROR:
        {
            FilePathInfo = (PFILEPATHS_W)Param1;
 
            DPRINT1("An error happened while trying to copy file '%S' (error 0x%08lx), skipping it...\n",
                    FilePathInfo->Target, FilePathInfo->Win32Error);
            return FILEOP_SKIP;
        }
 
        case SPFILENOTIFY_ENDDELETE:
        case SPFILENOTIFY_ENDRENAME:
        case SPFILENOTIFY_ENDCOPY:
        {
            CopyContext->CompletedOperations++;
 
            /* SYSREG checkpoint */
            if (CopyContext->TotalOperations >> 1 == CopyContext->CompletedOperations)
                DPRINT1("CHECKPOINT:HALF_COPIED\n");
 
            ProgressNextStep(CopyContext->ProgressBar);
            SetupUpdateMemoryInfo(CopyContext, FALSE);
            break;
        }
    }
 
    return FILEOP_DOIT;
}
 
 
/*
 * Displays the FileCopyPage.
 *
 * Next pages:
 *  RegistryPage(At once)
 *
 * SIDEEFFECTS
 *  Calls DoFileCopy
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
FileCopyPage(PINPUT_RECORD Ir)
{
    COPYCONTEXT CopyContext;
    UINT MemBarWidth;
 
    MUIDisplayPage(FILE_COPY_PAGE);
 
    /* Create context for the copy process */
    CopyContext.TotalOperations = 0;
    CopyContext.CompletedOperations = 0;
 
    /* Create the progress bar as well */
    CopyContext.ProgressBar = CreateProgressBar(13,
                                                26,
                                                xScreen - 13,
                                                yScreen - 20,
                                                10,
                                                24,
                                                TRUE,
                                                MUIGetString(STRING_SETUPCOPYINGFILES));
 
    // fit memory bars to screen width, distribute them uniform
    MemBarWidth = (xScreen - 26) / 5;
    MemBarWidth -= MemBarWidth % 2;  // make even
    /* ATTENTION: The following progress bars are debug stuff, which should not be translated!! */
    /* Create the paged pool progress bar */
    CopyContext.MemoryBars[0] = CreateProgressBar(13,
                                                  40,
                                                  13 + MemBarWidth,
                                                  43,
                                                  13,
                                                  44,
                                                  FALSE,
                                                  "Kernel Pool");
 
    /* Create the non paged pool progress bar */
    CopyContext.MemoryBars[1] = CreateProgressBar((xScreen / 2)- (MemBarWidth / 2),
                                                  40,
                                                  (xScreen / 2) + (MemBarWidth / 2),
                                                  43,
                                                  (xScreen / 2)- (MemBarWidth / 2),
                                                  44,
                                                  FALSE,
                                                  "Kernel Cache");
 
    /* Create the global memory progress bar */
    CopyContext.MemoryBars[2] = CreateProgressBar(xScreen - 13 - MemBarWidth,
                                                  40,
                                                  xScreen - 13,
                                                  43,
                                                  xScreen - 13 - MemBarWidth,
                                                  44,
                                                  FALSE,
                                                  "Free Memory");
 
    /* Do the file copying */
    DoFileCopy(&USetupData, FileCopyCallback, &CopyContext);
 
    /* If we get here, we're done, so cleanup the progress bar */
    DestroyProgressBar(CopyContext.ProgressBar);
    DestroyProgressBar(CopyContext.MemoryBars[0]);
    DestroyProgressBar(CopyContext.MemoryBars[1]);
    DestroyProgressBar(CopyContext.MemoryBars[2]);
 
    /* Create the $winnt$.inf file */
    InstallSetupInfFile(&USetupData);
 
    /* Go display the next page */
    return REGISTRY_PAGE;
}
 
 
static VOID
__cdecl
RegistryStatus(IN REGISTRY_STATUS RegStatus, ...)
{
    /* WARNING: Please keep this lookup table in sync with the resources! */
    static const UINT StringIDs[] =
    {
        STRING_DONE,                    /* Success */
        STRING_REGHIVEUPDATE,           /* RegHiveUpdate */
        STRING_IMPORTFILE,              /* ImportRegHive */
        STRING_DISPLAYSETTINGSUPDATE,   /* DisplaySettingsUpdate */
        STRING_LOCALESETTINGSUPDATE,    /* LocaleSettingsUpdate */
        STRING_ADDKBLAYOUTS,            /* KeybLayouts */
        STRING_KEYBOARDSETTINGSUPDATE,  /* KeybSettingsUpdate */
        STRING_CODEPAGEINFOUPDATE,      /* CodePageInfoUpdate */
    };
 
    va_list args;
 
    if (RegStatus < ARRAYSIZE(StringIDs))
    {
        va_start(args, RegStatus);
        CONSOLE_SetStatusTextV(MUIGetString(StringIDs[RegStatus]), args);
        va_end(args);
    }
    else
    {
        CONSOLE_SetStatusText("Unknown status %d", RegStatus);
    }
}
 
/*
 * Displays the RegistryPage.
 *
 * Next pages:
 *  BootLoaderSelectPage
 *  QuitPage
 *
 * SIDEEFFECTS
 *  Calls UpdateRegistry
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
RegistryPage(PINPUT_RECORD Ir)
{
    ULONG Error;
 
    MUIDisplayPage(REGISTRY_PAGE);
 
    Error = UpdateRegistry(&USetupData,
                           RepairUpdateFlag,
                           PartitionList,
                           InstallVolume->Info.DriveLetter,
                           SelectedLanguageId,
                           RegistryStatus,
                           &s_SubstSettings);
    if (Error != ERROR_SUCCESS)
    {
        MUIDisplayError(Error, Ir, POPUP_WAIT_ENTER);
        return QUIT_PAGE;
    }
    else
    {
        CONSOLE_SetStatusText(MUIGetString(STRING_DONE));
        return BOOTLOADER_INSTALL_PAGE;
    }
}
 
 
/*
 * Displays the BootLoaderSelectPage.
 *
 * Next pages:
 *  SuccessPage
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
BootLoaderSelectPage(PINPUT_RECORD Ir)
{
    USHORT Line = 12;
 
    CONSOLE_SetStatusText(MUIGetString(STRING_PLEASEWAIT));
 
    /* We must have a supported system partition by now */
    ASSERT(SystemPartition && SystemPartition->IsPartitioned && SystemPartition->PartitionNumber != 0);
 
    /*
     * If we repair an existing installation and we made it up to here,
     * this means a valid bootloader and boot entry have been found.
     * Thus, there is no need to re-install it: skip its installation.
     */
    if (RepairUpdateFlag)
    {
        USetupData.BootLoaderLocation = 0;
        goto Quit;
    }
 
    /* For unattended setup, skip MBR installation or install on removable disk if needed */
    if (IsUnattendedSetup)
    {
        if ((USetupData.BootLoaderLocation == 0) ||
            (USetupData.BootLoaderLocation == 1))
        {
            goto Quit;
        }
    }
 
#if 0 // Deprecated code, whose global logic may need to be moved elsewhere...
    /*
     * We may install an MBR or VBR, but before that, check whether
     * we need to actually install the VBR on removable disk if the
     * system partition is not recognized.
     */
    if ((SystemPartition->DiskEntry->DiskStyle != PARTITION_STYLE_MBR) ||
        !IsRecognizedPartition(SystemPartition->PartitionType))
    {
        USetupData.BootLoaderLocation = 1;
        goto Quit;
    }
#endif
 
    /* Is it an unattended install on hdd? */
    if (IsUnattendedSetup)
    {
        if ((USetupData.BootLoaderLocation == 2) ||
            (USetupData.BootLoaderLocation == 3))
        {
            goto Quit;
        }
    }
 
    MUIDisplayPage(BOOTLOADER_SELECT_PAGE);
    CONSOLE_InvertTextXY(8, Line, 60, 1);
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_DOWN))  /* DOWN */
        {
            CONSOLE_NormalTextXY(8, Line, 60, 1);
 
            Line++;
            if (Line < 12)
                Line = 15;
 
            if (Line > 15)
                Line = 12;
 
            CONSOLE_InvertTextXY(8, Line, 60, 1);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_UP))  /* UP */
        {
            CONSOLE_NormalTextXY(8, Line, 60, 1);
 
            Line--;
            if (Line < 12)
                Line = 15;
 
            if (Line > 15)
                Line = 12;
 
            CONSOLE_InvertTextXY(8, Line, 60, 1);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_HOME))  /* HOME */
        {
            CONSOLE_NormalTextXY(8, Line, 60, 1);
 
            Line = 12;
 
            CONSOLE_InvertTextXY(8, Line, 60, 1);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_END))  /* END */
        {
            CONSOLE_NormalTextXY(8, Line, 60, 1);
 
            Line = 15;
 
            CONSOLE_InvertTextXY(8, Line, 60, 1);
        }
        else if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
                 (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))  /* F3 */
        {
            if (ConfirmQuit(Ir))
                return QUIT_PAGE;
            break;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D)    /* ENTER */
        {
            if (Line == 12)
            {
                /* Install on both MBR and VBR */
                USetupData.BootLoaderLocation = 2;
                break;
            }
            else if (Line == 13)
            {
                /* Install on VBR only */
                USetupData.BootLoaderLocation = 3;
                break;
            }
            else if (Line == 14)
            {
                /* Install on removable disk */
                USetupData.BootLoaderLocation = 1;
                break;
            }
            else if (Line == 15)
            {
                /* Skip installation */
                USetupData.BootLoaderLocation = 0;
                break;
            }
 
            return BOOTLOADER_SELECT_PAGE;
        }
    }
 
Quit:
    /* Continue the installation; the bootloader is installed at the end */
    return INSTALL_DIRECTORY_PAGE;
}
 
 
/*
 * Installs the bootloader on removable disk.
 */
static BOOLEAN
BootLoaderRemovableDiskPage(PINPUT_RECORD Ir)
{
    NTSTATUS Status;
 
Retry:
    CONSOLE_ClearScreen();
    CONSOLE_Flush();
    MUIDisplayPage(BOOTLOADER_REMOVABLE_DISK_PAGE);
//  CONSOLE_SetStatusText(MUIGetString(STRING_PLEASEWAIT));
 
    while (TRUE)
    {
        CONSOLE_ConInKey(Ir);
 
        if ((Ir->Event.KeyEvent.uChar.AsciiChar == 0x00) &&
            (Ir->Event.KeyEvent.wVirtualKeyCode == VK_F3))   /* F3 */
        {
            if (ConfirmQuit(Ir))
                return FALSE;
            break;
        }
        else if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
        {
            // FIXME: So far USETUP only supports the 1st floppy.
            static const UNICODE_STRING FloppyDrive = RTL_CONSTANT_STRING(L"\\Device\\Floppy0\\");
            Status = InstallBootcodeToRemovable(USetupData.ArchType,
                                                &FloppyDrive,
                                                &USetupData.SourceRootPath,
                                                &USetupData.DestinationArcPath);
            if (Status == STATUS_SUCCESS)
                return TRUE; /* Successful installation */
 
            if (Status == STATUS_DEVICE_NOT_READY)
            {
                MUIDisplayError(ERROR_NO_FLOPPY, Ir, POPUP_WAIT_ENTER);
            }
            else if (!NT_SUCCESS(Status))
            {
                /* Any other NTSTATUS failure code */
                CHAR Buffer[MAX_PATH];
 
                DPRINT1("InstallBootcodeToRemovable() failed: Status 0x%lx\n", Status);
                RtlStringCbPrintfA(Buffer, sizeof(Buffer),
                                   "Setup could not install the bootloader.\n"
                                   "(Status 0x%08lx).\n"
                                   "Press ENTER to continue anyway.",
                                   Status);
                PopupError(Buffer,
                           MUIGetString(STRING_CONTINUE),
                           Ir, POPUP_WAIT_ENTER);
            }
            goto Retry;
        }
    }
    goto Retry;
}
 
/*
 * Installs the bootloader on hard-disk.
 */
static BOOLEAN
BootLoaderHardDiskPage(PINPUT_RECORD Ir)
{
    NTSTATUS Status;
 
    /* Copy FreeLoader to the disk and save the boot entries */
    Status = InstallBootManagerAndBootEntries(
                USetupData.ArchType,
                &USetupData.SystemRootPath,
                &USetupData.SourceRootPath,
                &USetupData.DestinationArcPath,
                (USetupData.BootLoaderLocation == 2)
                   ? 1 /* Install MBR and VBR */
                   : 0 /* Install VBR only */);
    if (Status == STATUS_SUCCESS)
        return TRUE; /* Successful installation */
 
    if (Status == ERROR_WRITE_BOOT)
    {
        /* Error when writing the VBR */
        MUIDisplayError(ERROR_WRITE_BOOT, Ir, POPUP_WAIT_ENTER,
                        SystemVolume->Info.FileSystem);
    }
    else if (Status == ERROR_INSTALL_BOOTCODE)
    {
        /* Error when writing the MBR */
        MUIDisplayError(ERROR_INSTALL_BOOTCODE, Ir, POPUP_WAIT_ENTER, L"MBR");
    }
    else if (Status == STATUS_NOT_SUPPORTED)
    {
        PopupError("Setup does not currently support installing\n"
                   "the bootloader on the computer you are using.\n"
                   "Press ENTER to continue anyway.",
                   MUIGetString(STRING_CONTINUE),
                   Ir, POPUP_WAIT_ENTER);
    }
    else if (!NT_SUCCESS(Status))
    {
        /* Any other NTSTATUS failure code */
        CHAR Buffer[MAX_PATH];
 
        DPRINT1("InstallBootManagerAndBootEntries() failed: Status 0x%lx\n", Status);
        RtlStringCbPrintfA(Buffer, sizeof(Buffer),
                           "Setup could not install the bootloader.\n"
                           "(Status 0x%08lx).\n"
                           "Press ENTER to continue anyway.",
                           Status);
        PopupError(Buffer,
                   MUIGetString(STRING_CONTINUE),
                   Ir, POPUP_WAIT_ENTER);
    }
    return FALSE;
}
 
/*
 * Actually installs the bootloader at the end of the installation.
 * The bootloader installation place has already been chosen before,
 * see BootLoaderSelectPage().
 *
 * Next pages:
 *  SuccessPage (At once)
 *  QuitPage
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
BootLoaderInstallPage(PINPUT_RECORD Ir)
{
    WCHAR PathBuffer[RTL_NUMBER_OF_FIELD(PARTENTRY, DeviceName) + 1];
 
    RtlFreeUnicodeString(&USetupData.SystemRootPath);
    RtlStringCchPrintfW(PathBuffer, _countof(PathBuffer),
                        L"%s\\", SystemPartition->DeviceName);
    RtlCreateUnicodeString(&USetupData.SystemRootPath, PathBuffer);
    DPRINT1("SystemRootPath: %wZ\n", &USetupData.SystemRootPath);
 
    if (USetupData.BootLoaderLocation != 0)
        MUIDisplayPage(BOOTLOADER_INSTALL_PAGE);
 
    switch (USetupData.BootLoaderLocation)
    {
        /* Install on removable disk */
        case 1:
            return BootLoaderRemovableDiskPage(Ir) ? SUCCESS_PAGE : QUIT_PAGE;
 
        /* Install on hard-disk */
        case 2: // System partition / MBR and VBR (on BIOS-based PC)
        case 3: // VBR only (on BIOS-based PC)
            return BootLoaderHardDiskPage(Ir) ? SUCCESS_PAGE : QUIT_PAGE;
 
        /* Skip installation */
        case 0:
        default:
            return SUCCESS_PAGE;
    }
}
 
 
static
BOOLEAN NTAPI
ProgressTimeOutStringHandler(
    IN PPROGRESSBAR Bar,
    IN BOOLEAN AlwaysUpdate,
    OUT PSTR Buffer,
    IN SIZE_T cchBufferSize)
{
    ULONG OldProgress = Bar->Progress;
 
    if (Bar->StepCount == 0)
    {
        Bar->Progress = 0;
    }
    else
    {
        Bar->Progress = Bar->StepCount - Bar->CurrentStep;
    }
 
    /* Build the progress string if it has changed */
    if (Bar->ProgressFormatText &&
        (AlwaysUpdate || (Bar->Progress != OldProgress)))
    {
        RtlStringCchPrintfA(Buffer, cchBufferSize,
                            Bar->ProgressFormatText, Bar->Progress / max(1, Bar->Width) + 1);
 
        return TRUE;
    }
 
    return FALSE;
}
 
static VOID
ProgressCountdown(
    IN PINPUT_RECORD Ir,
    IN LONG TimeOut)
{
    NTSTATUS Status;
    ULONG StartTime, BarWidth, TimerDiv;
    LONG TimeElapsed;
    LONG TimerValue, OldTimerValue;
    LARGE_INTEGER Timeout;
    PPROGRESSBAR ProgressBar;
    BOOLEAN RefreshProgress = TRUE;
 
    /* Bail out if the timeout is already zero */
    if (TimeOut <= 0)
        return;
 
    /* Create the timeout progress bar and set it up */
    ProgressBar = CreateProgressBarEx(13,
                                      26,
                                      xScreen - 13,
                                      yScreen - 20,
                                      10,
                                      24,
                                      TRUE,
                                      FOREGROUND_RED | BACKGROUND_BLUE,
                                      0,
                                      NULL,
                                      MUIGetString(STRING_REBOOTPROGRESSBAR),
                                      ProgressTimeOutStringHandler);
 
    BarWidth = max(1, ProgressBar->Width);
    TimerValue = TimeOut * BarWidth;
    ProgressSetStepCount(ProgressBar, TimerValue);
 
    StartTime = NtGetTickCount();
    CONSOLE_Flush();
 
    TimerDiv = 1000 / BarWidth;
    TimerDiv = max(1, TimerDiv);
    OldTimerValue = TimerValue;
    while (TRUE)
    {
        /* Decrease the timer */
 
        /*
         * Compute how much time the previous operations took.
         * This allows us in particular to take account for any time
         * elapsed if something slowed down.
         */
        TimeElapsed = NtGetTickCount() - StartTime;
        if (TimeElapsed >= TimerDiv)
        {
            /* Increase StartTime by steps of 1 / ProgressBar->Width seconds */
            TimeElapsed /= TimerDiv;
            StartTime += (TimerDiv * TimeElapsed);
 
            if (TimeElapsed <= TimerValue)
                TimerValue -= TimeElapsed;
            else
                TimerValue = 0;
 
            RefreshProgress = TRUE;
        }
 
        if (RefreshProgress)
        {
            ProgressSetStep(ProgressBar, OldTimerValue - TimerValue);
            RefreshProgress = FALSE;
        }
 
        /* Stop when the timer reaches zero */
        if (TimerValue <= 0)
            break;
 
        /* Check for user key presses */
 
        /*
         * If the timer is used, use a passive wait of maximum 1 second
         * while monitoring for incoming console input events, so that
         * we are still able to display the timing count.
         */
 
        /* Wait a maximum of 1 second for input events */
        TimeElapsed = NtGetTickCount() - StartTime;
        if (TimeElapsed < TimerDiv)
        {
            /* Convert the time to NT format */
            Timeout.QuadPart = (TimerDiv - TimeElapsed) * -10000LL;
            Status = NtWaitForSingleObject(StdInput, FALSE, &Timeout);
        }
        else
        {
            Status = STATUS_TIMEOUT;
        }
 
        /* Check whether the input event has been signaled, or a timeout happened */
        if (Status == STATUS_TIMEOUT)
        {
            continue;
        }
        if (Status != STATUS_WAIT_0)
        {
            /* An error happened, bail out */
            DPRINT1("NtWaitForSingleObject() failed, Status 0x%08lx\n", Status);
            break;
        }
 
        /* Check for an ENTER key press */
        while (CONSOLE_ConInKeyPeek(Ir))
        {
            if (Ir->Event.KeyEvent.uChar.AsciiChar == 0x0D) /* ENTER */
            {
                /* Found it, stop waiting */
                goto Exit;
            }
        }
    }
 
Exit:
    /* Destroy the progress bar and quit */
    DestroyProgressBar(ProgressBar);
}
 
 
/*
 * Displays the QuitPage.
 *
 * Next pages:
 *  FlushPage (At once)
 *
 * SIDEEFFECTS
 *  Destroy the Lists
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
QuitPage(PINPUT_RECORD Ir)
{
    MUIDisplayPage(QUIT_PAGE);
 
    /* Destroy the NTOS installations list */
    if (NtOsInstallsList != NULL)
    {
        DestroyGenericList(NtOsInstallsList, TRUE);
        NtOsInstallsList = NULL;
    }
 
    /* Destroy the partition list */
    if (PartitionList != NULL)
    {
        DestroyPartitionList(PartitionList);
        PartitionList = NULL;
    }
 
    CONSOLE_SetStatusText(MUIGetString(STRING_REBOOTCOMPUTER2));
 
    /* Wait for maximum 15 seconds or an ENTER key before quitting */
    ProgressCountdown(Ir, 15);
    return FLUSH_PAGE;
}
 
 
/*
 * Displays the SuccessPage.
 *
 * Next pages:
 *  FlushPage (At once)
 *
 * SIDEEFFECTS
 *  Destroy the Lists
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
SuccessPage(PINPUT_RECORD Ir)
{
    MUIDisplayPage(SUCCESS_PAGE);
 
    if (IsUnattendedSetup)
        return FLUSH_PAGE;
 
    /* Wait for maximum 15 seconds or an ENTER key before quitting */
    ProgressCountdown(Ir, 15);
    return FLUSH_PAGE;
}
 
 
/*
 * Displays the FlushPage.
 *
 * Next pages:
 *  RebootPage (At once)
 *
 * RETURNS
 *   Number of the next page.
 */
static PAGE_NUMBER
FlushPage(PINPUT_RECORD Ir)
{
    MUIDisplayPage(FLUSH_PAGE);
    return REBOOT_PAGE;
}
 
 
/*
 * The start routine and page management
 */
NTSTATUS
RunUSetup(VOID)
{
    NTSTATUS Status;
    INPUT_RECORD Ir;
    PAGE_NUMBER Page;
    BOOLEAN Old;
 
    InfSetHeap(ProcessHeap);
 
    /* Tell the Cm this is a setup boot, and it has to behave accordingly */
    Status = NtInitializeRegistry(CM_BOOT_FLAG_SETUP);
    if (!NT_SUCCESS(Status))
        DPRINT1("NtInitializeRegistry() failed (Status 0x%08lx)\n", Status);
 
    /* Initialize the user-mode PnP manager */
    Status = InitializeUserModePnpManager(&USetupData.SetupInf);
    if (!NT_SUCCESS(Status))
    {
        // PrintString(??);
        DPRINT1("The user-mode PnP manager could not initialize (Status 0x%08lx), expect unavailable devices!\n", Status);
    }
 
    if (!CONSOLE_Init())
    {
        PrintString(MUIGetString(STRING_CONSOLEFAIL1));
        PrintString(MUIGetString(STRING_CONSOLEFAIL2));
        PrintString(MUIGetString(STRING_CONSOLEFAIL3));
 
        /* We failed to initialize the video, just quit the installer */
        return STATUS_APP_INIT_FAILURE;
    }
 
    /* Hide the cursor and clear the screen and keyboard buffer */
    CONSOLE_SetCursorType(TRUE, FALSE);
    CONSOLE_ClearScreen();
    CONSOLE_Flush();
 
    /* Global Initialization page */
    Page = SetupStartPage(&Ir);
 
    while (Page != REBOOT_PAGE && Page != RECOVERY_PAGE)
    {
        CONSOLE_ClearScreen();
        CONSOLE_Flush();
 
        // CONSOLE_SetUnderlinedTextXY(4, 3, " ReactOS " KERNEL_VERSION_STR " Setup ");
 
        switch (Page)
        {
            /* Language page */
            case LANGUAGE_PAGE:
                Page = LanguagePage(&Ir);
                break;
 
            /* Welcome page */
            case WELCOME_PAGE:
                Page = WelcomePage(&Ir);
                break;
 
            /* License page */
            case LICENSE_PAGE:
                Page = LicensePage(&Ir);
                break;
 
            /* Install pages */
            case INSTALL_INTRO_PAGE:
                Page = InstallIntroPage(&Ir);
                break;
 
#if 0
            case SCSI_CONTROLLER_PAGE:
                Page = ScsiControllerPage(&Ir);
                break;
 
            case OEM_DRIVER_PAGE:
                Page = OemDriverPage(&Ir);
                break;
#endif
 
            case DEVICE_SETTINGS_PAGE:
                Page = DeviceSettingsPage(&Ir);
                break;
 
            case COMPUTER_SETTINGS_PAGE:
                Page = ComputerSettingsPage(&Ir);
                break;
 
            case DISPLAY_SETTINGS_PAGE:
                Page = DisplaySettingsPage(&Ir);
                break;
 
            case KEYBOARD_SETTINGS_PAGE:
                Page = KeyboardSettingsPage(&Ir);
                break;
 
            case LAYOUT_SETTINGS_PAGE:
                Page = LayoutSettingsPage(&Ir);
                break;
 
            /* Partitioning pages */
            case SELECT_PARTITION_PAGE:
                Page = SelectPartitionPage(&Ir);
                break;
 
            case CREATE_PARTITION_PAGE:
                Page = CreatePartitionPage(&Ir);
                break;
 
            case CONFIRM_DELETE_SYSTEM_PARTITION_PAGE:
                Page = ConfirmDeleteSystemPartitionPage(&Ir);
                break;
 
            case DELETE_PARTITION_PAGE:
                Page = DeletePartitionPage(&Ir);
                break;
 
            /* File system partition operations pages */
            case START_PARTITION_OPERATIONS_PAGE:
                Page = StartPartitionOperationsPage(&Ir);
                break;
 
            /* Bootloader selection page */
            case BOOTLOADER_SELECT_PAGE:
                Page = BootLoaderSelectPage(&Ir);
                break;
 
            /* Installation pages */
            case INSTALL_DIRECTORY_PAGE:
                Page = InstallDirectoryPage(&Ir);
                break;
 
            case PREPARE_COPY_PAGE:
                Page = PrepareCopyPage(&Ir);
                break;
 
            case FILE_COPY_PAGE:
                Page = FileCopyPage(&Ir);
                break;
 
            case REGISTRY_PAGE:
                Page = RegistryPage(&Ir);
                break;
 
            /* Bootloader installation page */
            case BOOTLOADER_INSTALL_PAGE:
            // case BOOTLOADER_REMOVABLE_DISK_PAGE:
                Page = BootLoaderInstallPage(&Ir);
                break;
 
            /* Repair pages */
            case REPAIR_INTRO_PAGE:
                Page = RepairIntroPage(&Ir);
                break;
 
            case UPGRADE_REPAIR_PAGE:
                Page = UpgradeRepairPage(&Ir);
                break;
 
            case SUCCESS_PAGE:
                Page = SuccessPage(&Ir);
                break;
 
            case FLUSH_PAGE:
                Page = FlushPage(&Ir);
                break;
 
            case QUIT_PAGE:
                Page = QuitPage(&Ir);
                break;
 
            /* Virtual pages */
            case SETUP_INIT_PAGE:
            case SELECT_FILE_SYSTEM_PAGE:
            case FORMAT_PARTITION_PAGE:
            // case CHECK_FILE_SYSTEM_PAGE:
            case REBOOT_PAGE:
            case RECOVERY_PAGE:
                break;
 
            default:
                break;
        }
    }
 
    /* Terminate the user-mode PnP manager */
    TerminateUserModePnpManager();
 
    /* Setup has finished */
    FinishSetup(&USetupData);
 
    if (Page == RECOVERY_PAGE)
        RecoveryConsole();
 
    FreeConsole();
 
    /* Reboot */
    RtlAdjustPrivilege(SE_SHUTDOWN_PRIVILEGE, TRUE, FALSE, &Old);
    NtShutdownSystem(ShutdownReboot);
    RtlAdjustPrivilege(SE_SHUTDOWN_PRIVILEGE, Old, FALSE, &Old);
 
    return STATUS_SUCCESS;
}
 
 
VOID NTAPI
NtProcessStartup(PPEB Peb)
{
    NTSTATUS Status;
    LARGE_INTEGER Time;
 
    RtlNormalizeProcessParams(Peb->ProcessParameters);
 
    ProcessHeap = Peb->ProcessHeap;
 
    NtQuerySystemTime(&Time);
 
    Status = RunUSetup();
 
    if (NT_SUCCESS(Status))
    {
        /*
         * Avoid a bugcheck if RunUSetup() finishes too quickly by implementing
         * a protective waiting.
         * This wait is needed because, since we are started as SMSS.EXE,
         * the NT kernel explicitly waits 5 seconds for the initial process
         * SMSS.EXE to initialize (as a protective measure), and otherwise
         * bugchecks with the code SESSION5_INITIALIZATION_FAILED.
         */
        Time.QuadPart += 50000000;
        NtDelayExecution(FALSE, &Time);
    }
    else
    {
        /* The installer failed to start: raise a hard error (crash the system/BSOD) */
        Status = NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED,
                                  0, 0, NULL, 0, NULL);
    }
 
    NtTerminateProcess(NtCurrentProcess(), Status);
}
 
/* EOF */