mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
616 lines
15 KiB
616 lines
15 KiB
//----------------------------------------------------------------------------
|
|
//
|
|
// Copyright (c) 1997-1999 Microsoft Corporation
|
|
// All rights reserved.
|
|
//
|
|
// File Name:
|
|
// cmdlines.c
|
|
//
|
|
// Description:
|
|
// Dialog proc for the cmdlines.txt page
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
#include "pch.h"
|
|
#include "resource.h"
|
|
|
|
#define MAX_CMDLINE 1024
|
|
#define CMD_FILENAME _T("cmdlines.txt")
|
|
|
|
static NAMELIST CmdLinesList = { 0 };
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: SetPathToCmdlines
|
|
//
|
|
// Purpose: Determines the path to the cmdlines.txt and set PathBuffer to it.
|
|
// PathBuffer is assumed to be MAX_PATH long
|
|
//
|
|
// Arguments: OUT TCHAR *PathBuffer - path to cmdlines.txt
|
|
//
|
|
// Returns: VOID
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
static VOID
|
|
SetPathToCmdlines( OUT TCHAR *PathBuffer, DWORD cbPath )
|
|
{
|
|
|
|
if( WizGlobals.iProductInstall == PRODUCT_SYSPREP )
|
|
{
|
|
TCHAR szDrive[MAX_PATH];
|
|
TCHAR szSysprepPath[MAX_PATH] = _T("");
|
|
|
|
ExpandEnvironmentStrings( _T("%SystemDrive%"),
|
|
szDrive,
|
|
MAX_PATH );
|
|
|
|
ConcatenatePaths( szSysprepPath,
|
|
szDrive,
|
|
_T("\\sysprep\\i386\\$oem$"),
|
|
NULL );
|
|
|
|
EnsureDirExists( szSysprepPath );
|
|
|
|
lstrcpyn( PathBuffer, szSysprepPath, cbPath );
|
|
}
|
|
else
|
|
{
|
|
lstrcpyn( PathBuffer, WizGlobals.OemFilesPath, cbPath );
|
|
}
|
|
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: LoadCmdLinesFile
|
|
//
|
|
// Purpose: Loads the contents of cmdlines.txt into memory.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
VOID LoadCmdLinesFile(HWND hwnd)
|
|
{
|
|
FILE *fp;
|
|
TCHAR CmdLineBuffer[MAX_CMDLINE + 1];
|
|
TCHAR PathBuffer[MAX_PATH];
|
|
|
|
ResetNameList(&CmdLinesList);
|
|
|
|
SetPathToCmdlines( PathBuffer, AS(PathBuffer) );
|
|
|
|
ConcatenatePaths(PathBuffer, CMD_FILENAME, NULL);
|
|
|
|
if ( (fp = My_fopen(PathBuffer, _T("r") )) == NULL )
|
|
return;
|
|
|
|
//
|
|
// Add all the entries to the namelist except for the [Commands] line
|
|
//
|
|
|
|
while( My_fgets(CmdLineBuffer, MAX_CMDLINE, fp) != NULL ) {
|
|
|
|
if( _tcsstr( CmdLineBuffer, _T("[Commands]") ) == NULL ) {
|
|
|
|
AddNameToNameList( &CmdLinesList,
|
|
CleanSpaceAndQuotes( CmdLineBuffer ) );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
My_fclose(fp);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: WriteCmdLinesFile
|
|
//
|
|
// Purpose: Writes the contents of our in-memory cmdlines to disk.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
VOID WriteCmdLinesFile(HWND hwnd)
|
|
{
|
|
UINT i, nNames;
|
|
UINT iNumCmdLinesEntries;
|
|
FILE *fp;
|
|
TCHAR PathBuffer[MAX_PATH], *pCommand;
|
|
|
|
//
|
|
// If there are no command lines to write then don't create the
|
|
// cmdlines.txt file.
|
|
//
|
|
iNumCmdLinesEntries = GetNameListSize( &CmdLinesList );
|
|
|
|
if( iNumCmdLinesEntries == 0 ) {
|
|
return;
|
|
}
|
|
|
|
//
|
|
// Keep trying to open cmdlines.txt until it's open or until the
|
|
// user gives up
|
|
//
|
|
|
|
SetPathToCmdlines( PathBuffer, AS(PathBuffer) );
|
|
|
|
ConcatenatePaths(PathBuffer, CMD_FILENAME, NULL);
|
|
|
|
do {
|
|
if ( (fp = My_fopen(PathBuffer, _T("w") )) == NULL ) {
|
|
UINT iRet = ReportErrorId(
|
|
hwnd,
|
|
MSGTYPE_RETRYCANCEL | MSGTYPE_WIN32,
|
|
IDS_OPEN_CMDLINES_FAILED,
|
|
PathBuffer);
|
|
if ( iRet != IDRETRY )
|
|
return;
|
|
} else
|
|
break;
|
|
|
|
} while ( TRUE );
|
|
|
|
//
|
|
// ISSUE-2002/02/28-stelo- Check return value from fputs
|
|
//
|
|
|
|
My_fputs( _T("[Commands]\n"), fp );
|
|
|
|
//
|
|
// Write out each command in CmdLinesList
|
|
//
|
|
|
|
for ( i = 0, nNames = GetNameListSize(&CmdLinesList);
|
|
i < nNames;
|
|
i++ ) {
|
|
|
|
pCommand = GetNameListName(&CmdLinesList, i);
|
|
My_fputs( _T("\""), fp );
|
|
My_fputs(pCommand, fp);
|
|
My_fputs( _T("\"\n"), fp );
|
|
}
|
|
|
|
My_fclose(fp);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: GreyCmdLinesPage
|
|
//
|
|
// Purpose: Greys out the remove button if nothing selected
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
VOID GreyCmdLinesPage(HWND hwnd)
|
|
{
|
|
INT_PTR idx;
|
|
HWND hCtrl = GetDlgItem(hwnd, IDC_REMOVECMD);
|
|
|
|
idx = SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_GETCURSEL,
|
|
(WPARAM) 0,
|
|
(LPARAM) 0);
|
|
|
|
//
|
|
// Grey the remove button unless something is selected
|
|
//
|
|
|
|
EnableWindow(hCtrl, idx != LB_ERR);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: OnSelChangeCmdLines
|
|
//
|
|
// Purpose: Called when user selects an item on the cmd list. We need
|
|
// to ungrey the remove button
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
VOID OnCmdLinesSelChange(HWND hwnd)
|
|
{
|
|
|
|
SetArrows( hwnd,
|
|
IDC_CMDLIST,
|
|
IDC_BUT_MOVE_UP,
|
|
IDC_BUT_MOVE_DOWN );
|
|
|
|
GreyCmdLinesPage(hwnd);
|
|
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: OnSetActiveCmdLines
|
|
//
|
|
// Purpose: Called at SETACTIVE time. We load the current contents
|
|
// of cmdline.txt into memory and display it. We read the
|
|
// file at SETACTIVE time so that the user has a way of
|
|
// refreshing the display.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
VOID OnSetActiveCmdLines(HWND hwnd)
|
|
{
|
|
UINT i, nNames;
|
|
LPTSTR pNextName;
|
|
|
|
//
|
|
// Load cmdlines.txt into memory
|
|
//
|
|
|
|
LoadCmdLinesFile(hwnd);
|
|
|
|
//
|
|
// Reset the display to match what is in memory
|
|
//
|
|
|
|
SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_RESETCONTENT,
|
|
(WPARAM) 0,
|
|
(LPARAM) 0);
|
|
|
|
|
|
for ( i = 0, nNames = GetNameListSize(&CmdLinesList);
|
|
i < nNames;
|
|
i++ ) {
|
|
|
|
pNextName = GetNameListName(&CmdLinesList, i);
|
|
|
|
SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_ADDSTRING,
|
|
(WPARAM) 0,
|
|
(LPARAM) pNextName);
|
|
}
|
|
|
|
GreyCmdLinesPage(hwnd);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: OnAddCmdLine
|
|
//
|
|
// Purpose: Called when user pushes ADD button. Get command from
|
|
// edit field and add it to the in-memory list.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
VOID OnAddCmdLine(HWND hwnd)
|
|
{
|
|
TCHAR CmdBuffer[MAX_CMDLINE + 1];
|
|
|
|
//
|
|
// get the command the user typed in
|
|
//
|
|
|
|
GetDlgItemText(hwnd, IDT_CMDLINE, CmdBuffer, MAX_CMDLINE);
|
|
|
|
//
|
|
// Don't add a blank command
|
|
//
|
|
|
|
if( CmdBuffer[0] == _T('\0') )
|
|
{
|
|
return;
|
|
}
|
|
|
|
//
|
|
// display what the user typed-in in the listbox
|
|
// and clear out the name the user typed
|
|
//
|
|
|
|
SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_ADDSTRING,
|
|
(WPARAM) 0,
|
|
(LPARAM) CmdBuffer);
|
|
|
|
SetArrows( hwnd,
|
|
IDC_CMDLIST,
|
|
IDC_BUT_MOVE_UP,
|
|
IDC_BUT_MOVE_DOWN );
|
|
|
|
SetDlgItemText( hwnd, IDT_CMDLINE, _T("") );
|
|
|
|
SetFocus(GetDlgItem(hwnd, IDT_CMDLINE));
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: OnRemoveCmdLine
|
|
//
|
|
// Purpose: Called when user pushes REMOVE button. Get selected command
|
|
// and remove it from display and memory.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
VOID OnRemoveCmdLine(HWND hwnd)
|
|
{
|
|
TCHAR CmdBuffer[MAX_CMDLINE + 1];
|
|
INT_PTR idx, Count;
|
|
|
|
//
|
|
// Get users selection of the command to remove
|
|
//
|
|
|
|
idx = SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_GETCURSEL,
|
|
(WPARAM) 0,
|
|
(LPARAM) 0);
|
|
|
|
if ( idx == LB_ERR )
|
|
return;
|
|
|
|
//
|
|
// Retrieve the name to remove from listbox
|
|
//
|
|
|
|
SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_GETTEXT,
|
|
(WPARAM) idx,
|
|
(LPARAM) CmdBuffer);
|
|
|
|
//
|
|
// Remove it from the listbox display
|
|
//
|
|
|
|
SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_DELETESTRING,
|
|
(WPARAM) idx,
|
|
(LPARAM) 0);
|
|
|
|
//
|
|
// Have to set a new selection.
|
|
//
|
|
|
|
Count = SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_GETCOUNT,
|
|
(WPARAM) 0,
|
|
(LPARAM) 0);
|
|
if ( Count ) {
|
|
if ( idx >= Count )
|
|
idx--;
|
|
SendDlgItemMessage(hwnd,
|
|
IDC_CMDLIST,
|
|
LB_SETCURSEL,
|
|
(WPARAM) idx,
|
|
(LPARAM) 0);
|
|
}
|
|
|
|
SetArrows( hwnd,
|
|
IDC_CMDLIST,
|
|
IDC_BUT_MOVE_UP,
|
|
IDC_BUT_MOVE_DOWN );
|
|
|
|
//
|
|
// There might be nothing selected now
|
|
//
|
|
|
|
GreyCmdLinesPage(hwnd);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: OnCommandLinesInitDialog
|
|
//
|
|
// Purpose:
|
|
//
|
|
// Arguments: IN HWND hwnd - handle to the dialog
|
|
//
|
|
// Returns: VOID
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
VOID
|
|
OnCommandLinesInitDialog( IN HWND hwnd )
|
|
{
|
|
|
|
//
|
|
// Set text limit
|
|
//
|
|
|
|
SendDlgItemMessage(hwnd,
|
|
IDT_CMDLINE,
|
|
EM_LIMITTEXT,
|
|
(WPARAM) MAX_CMDLINE,
|
|
(LPARAM) 0);
|
|
|
|
SetArrows( hwnd,
|
|
IDC_CMDLIST,
|
|
IDC_BUT_MOVE_UP,
|
|
IDC_BUT_MOVE_DOWN );
|
|
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: OnWizNextCmdLines
|
|
//
|
|
// Purpose: Called when user pushes NEXT button. Write the file out.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
VOID OnWizNextCmdLines(HWND hwnd)
|
|
{
|
|
INT_PTR i;
|
|
INT_PTR iRetVal;
|
|
INT_PTR iNumItems;
|
|
TCHAR CmdBuffer[MAX_CMDLINE + 1];
|
|
BOOL bStayHere = FALSE;
|
|
|
|
//
|
|
// If the user typed something into the command field but failed to
|
|
// ADD it, auto-add it
|
|
//
|
|
|
|
GetDlgItemText(hwnd, IDT_CMDLINE, CmdBuffer, MAX_CMDLINE + 1);
|
|
|
|
if ( CmdBuffer[0] != _T('\0') )
|
|
OnAddCmdLine(hwnd);
|
|
|
|
|
|
//
|
|
// Store all the entries in the list box into the Command Lines namelist
|
|
//
|
|
|
|
iNumItems = SendDlgItemMessage( hwnd,
|
|
IDC_CMDLIST,
|
|
LB_GETCOUNT,
|
|
(WPARAM) 0,
|
|
(LPARAM) 0 );
|
|
|
|
ResetNameList( &CmdLinesList );
|
|
|
|
for( i = 0; i < iNumItems; i++ )
|
|
{
|
|
|
|
iRetVal = SendDlgItemMessage( hwnd,
|
|
IDC_CMDLIST,
|
|
LB_GETTEXT,
|
|
(WPARAM) i,
|
|
(LPARAM) CmdBuffer );
|
|
|
|
if( iRetVal == LB_ERR )
|
|
{
|
|
AssertMsg( FALSE,
|
|
"Error adding items to namelist." );
|
|
|
|
break;
|
|
}
|
|
|
|
AddNameToNameList(&CmdLinesList, CmdBuffer);
|
|
|
|
}
|
|
|
|
//
|
|
// Write the cmd lines file and move the wizard on
|
|
//
|
|
|
|
WriteCmdLinesFile(hwnd);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Function: DlgCommandLinesPage
|
|
//
|
|
// Purpose: Dlg proc.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
INT_PTR CALLBACK DlgCommandLinesPage(
|
|
IN HWND hwnd,
|
|
IN UINT uMsg,
|
|
IN WPARAM wParam,
|
|
IN LPARAM lParam)
|
|
{
|
|
BOOL bStatus = TRUE;
|
|
|
|
switch (uMsg) {
|
|
|
|
case WM_INITDIALOG:
|
|
OnCommandLinesInitDialog( hwnd );
|
|
break;
|
|
|
|
case WM_COMMAND:
|
|
{
|
|
int nButtonId;
|
|
|
|
switch ( nButtonId = LOWORD(wParam) ) {
|
|
|
|
case IDC_ADDCMD:
|
|
if ( HIWORD(wParam) == BN_CLICKED )
|
|
OnAddCmdLine(hwnd);
|
|
break;
|
|
|
|
case IDC_REMOVECMD:
|
|
if ( HIWORD(wParam) == BN_CLICKED )
|
|
OnRemoveCmdLine(hwnd);
|
|
break;
|
|
|
|
case IDC_CMDLIST:
|
|
if ( HIWORD(wParam) == LBN_SELCHANGE )
|
|
OnCmdLinesSelChange(hwnd);
|
|
break;
|
|
|
|
case IDC_BUT_MOVE_UP:
|
|
|
|
OnUpButtonPressed( hwnd, IDC_CMDLIST );
|
|
|
|
SetArrows( hwnd,
|
|
IDC_CMDLIST,
|
|
IDC_BUT_MOVE_UP,
|
|
IDC_BUT_MOVE_DOWN );
|
|
|
|
break;
|
|
|
|
|
|
case IDC_BUT_MOVE_DOWN:
|
|
|
|
OnDownButtonPressed( hwnd, IDC_CMDLIST );
|
|
|
|
SetArrows( hwnd,
|
|
IDC_CMDLIST,
|
|
IDC_BUT_MOVE_UP,
|
|
IDC_BUT_MOVE_DOWN );
|
|
break;
|
|
|
|
default:
|
|
bStatus = FALSE;
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case WM_NOTIFY:
|
|
{
|
|
LPNMHDR pnmh = (LPNMHDR)lParam;
|
|
|
|
switch( pnmh->code ) {
|
|
|
|
case PSN_QUERYCANCEL:
|
|
WIZ_CANCEL(hwnd);
|
|
break;
|
|
|
|
case PSN_SETACTIVE:
|
|
|
|
g_App.dwCurrentHelp = IDH_ADDL_CMND;
|
|
|
|
if ( WizGlobals.iProductInstall == PRODUCT_UNATTENDED_INSTALL )
|
|
WIZ_BUTTONS(hwnd, PSWIZB_BACK | PSWIZB_FINISH);
|
|
else
|
|
WIZ_BUTTONS(hwnd, PSWIZB_BACK | PSWIZB_NEXT);
|
|
|
|
OnSetActiveCmdLines(hwnd);
|
|
break;
|
|
|
|
case PSN_WIZBACK:
|
|
bStatus = FALSE;
|
|
break;
|
|
|
|
case PSN_WIZNEXT:
|
|
OnWizNextCmdLines(hwnd);
|
|
bStatus = FALSE;
|
|
break;
|
|
|
|
case PSN_HELP:
|
|
WIZ_HELP();
|
|
break;
|
|
|
|
default:
|
|
bStatus = FALSE;
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
bStatus = FALSE;
|
|
break;
|
|
}
|
|
return bStatus;
|
|
}
|