mirror of https://github.com/lianthony/NT4.0
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.
1026 lines
31 KiB
1026 lines
31 KiB
//*************************************************************
|
|
//
|
|
// strtlst.c - Startup List processing code
|
|
//
|
|
// Microsoft Confidential
|
|
// Copyright (c) Microsoft Corporation 1996
|
|
// All rights reserved
|
|
//
|
|
// History:
|
|
// 23-Jan-1996 JonPa - ported from NT3.51's system.cpl
|
|
//
|
|
//*************************************************************
|
|
#include <sysdm.h>
|
|
|
|
///////////////////////////////////////////////////////////////
|
|
// Persistant vars
|
|
///////////////////////////////////////////////////////////////
|
|
static TCHAR *pszBoot = NULL;
|
|
static int nOriginalSelection;
|
|
static int nOriginalTimeout;
|
|
|
|
/*
|
|
* These functions in SETUPDLL.DLL are ANSI only!!!!
|
|
*
|
|
* Therefore any functions working with this DLL MUST remain ANSI only.
|
|
* The functions are GetRGSZEnvVar and UpdateNVRAM.
|
|
* The structure CPEnvBuf MUST also remain ANSI only.
|
|
*/
|
|
typedef int (WINAPI *GETNVRAMPROC)(CHAR **, USHORT, CHAR *, USHORT);
|
|
typedef int (WINAPI *WRITENVRAMPROC)(DWORD, PSZ *, PSZ *);
|
|
|
|
#if defined(_MIPS_) || defined(_ALPHA_) || defined(_PPC_)
|
|
#else
|
|
// X86 only vars
|
|
char szBootIniA[] = "c:\\boot.ini";
|
|
TCHAR szBootIni[] = TEXT( "c:\\boot.ini" );
|
|
TCHAR szBootLdr[] = TEXT( "boot loader" );
|
|
TCHAR szFlexBoot[] = TEXT( "flexboot" );
|
|
TCHAR szMultiBoot[] = TEXT( "multiboot" );
|
|
TCHAR szTimeout[] = TEXT( "timeout" );
|
|
TCHAR szDefault[] = TEXT( "default" );
|
|
char szOSA[] = "operating systems";
|
|
|
|
#define BUFZ 4096
|
|
|
|
#define FORMIN 0
|
|
#define FORMAX 999
|
|
#define FORDEF 30
|
|
|
|
TCHAR x86DetermineSystemPartition( IN HWND hdlg );
|
|
BOOL CheckVal( HWND hDlg, WORD wID, WORD wMin, WORD wMax, WORD wMsgID );
|
|
#endif
|
|
|
|
#if defined(_MIPS_) || defined(_ALPHA_) || defined(_PPC_)
|
|
|
|
static HMODULE hmodSetupDll; // hmod for setup - has api we need
|
|
static GETNVRAMPROC fpGetNVRAMvar; // address of function for getting nvram vars
|
|
BOOL fCanUpdateNVRAM;
|
|
|
|
typedef struct tagEnvBuf
|
|
{
|
|
int cEntries;
|
|
CHAR * pszVars[10];
|
|
} CPEnvBuf;
|
|
|
|
//*************************************************************
|
|
//
|
|
// StringToIntA
|
|
//
|
|
// Purpose: atoi
|
|
//
|
|
// Parameters: LPSTR sz - pointer of string to convert
|
|
//
|
|
// Return: void
|
|
//
|
|
// WARNING: Unlike StringToInt, this one does not skip leading
|
|
// white space
|
|
//
|
|
//*************************************************************
|
|
int StringToIntA( LPSTR sz ) {
|
|
int i = 0;
|
|
|
|
while( IsDigit( *sz ) ) {
|
|
i = i * 10 + (*sz - '0');
|
|
sz++;
|
|
}
|
|
|
|
return i;
|
|
}
|
|
|
|
//*************************************************************
|
|
//
|
|
// IntToStringA
|
|
//
|
|
// Purpose: itoa
|
|
//
|
|
// Parameters: INT i - integer to convert
|
|
// LPSTR sz - pointer where to put the result
|
|
//
|
|
// Return: void
|
|
//
|
|
//*************************************************************
|
|
#define CCH_MAX_DEC 12 // Number of chars needed to hold 2^32
|
|
|
|
void IntToStringA( INT i, LPSTR sz) {
|
|
CHAR szTemp[CCH_MAX_DEC];
|
|
int iChr;
|
|
|
|
|
|
iChr = 0;
|
|
|
|
do {
|
|
szTemp[iChr++] = '0' + (i % 10);
|
|
i = i / 10;
|
|
} while (i != 0);
|
|
|
|
do {
|
|
iChr--;
|
|
*sz++ = szTemp[iChr];
|
|
} while (iChr != 0);
|
|
|
|
*sz++ = '\0';
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
// CP_MAX_ENV assumes entire env. var. value < maxpath +
|
|
// add 20 for various quotes
|
|
// and 10 more for commas (see list description below)
|
|
////////////////////////////////////////////////////////////////////////////
|
|
#define CP_MAX_ENV (MAX_PATH + 30)
|
|
|
|
CPEnvBuf CPEBOSLoadIdentifier;
|
|
BOOL fAutoLoad;
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// This routine will query the MIPS or Alpha NVRAM for an option passed
|
|
// in szName and fill in the argv style pointer passed in.
|
|
//
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
BOOL GetRGSZEnvVar(CPEnvBuf * pEnvBuf, PCHAR pszName)
|
|
{
|
|
CHAR *pszCur, *p;
|
|
int cb, i;
|
|
CHAR *rgtmp[1];
|
|
CHAR rgchOut[CP_MAX_ENV];
|
|
|
|
// GetNVRAMVar takes an argv[] style paramater as input, so crock
|
|
// one up.
|
|
rgtmp[0] = pszName;
|
|
|
|
// GetNVRAMVar returns a 'list' of the form
|
|
// open-curly"string1","string2","string3"close-curly
|
|
//
|
|
// an empty environment string will be 5 bytes:
|
|
// open-curly""close-curly[null-terminator]
|
|
|
|
cb = fpGetNVRAMvar (rgtmp, (USHORT)1,
|
|
rgchOut, (USHORT) CP_MAX_ENV);
|
|
|
|
pEnvBuf->cEntries = 0;
|
|
|
|
// if cb was equal to 5, the string was empty (see above comment)
|
|
if (cb > 5)
|
|
{
|
|
// break the string up into array of separate strings that
|
|
// can be put into a listbox.
|
|
pszCur = rgchOut;
|
|
|
|
// skip first open-curly brace
|
|
pszCur++;
|
|
|
|
// counter for array of strings
|
|
i = 0;
|
|
while (*pszCur != '}')
|
|
{
|
|
p = pEnvBuf->pszVars[i] = MemAlloc (LPTR, MAX_PATH);
|
|
// skip first quote
|
|
pszCur++;
|
|
while (*pszCur != '"')
|
|
*p++ = *pszCur++;
|
|
|
|
// skip the close quote
|
|
pszCur++;
|
|
|
|
// null terminate destination
|
|
*p = '\0';
|
|
|
|
// skip the comma if not at end of string
|
|
if (*pszCur == ',')
|
|
{
|
|
pszCur++;
|
|
// and go to next string
|
|
}
|
|
i++;
|
|
}
|
|
pEnvBuf->cEntries = i;
|
|
}
|
|
|
|
return pEnvBuf->cEntries;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// This routine will free the memory allocated by GetRGSZEnvVar
|
|
//
|
|
// History:
|
|
// 22-Apr-1996 JonPa Created it.
|
|
//
|
|
////////////////////////////////////////////////////////////////////////////
|
|
void FreeRGSZEnvVar(CPEnvBuf * pEnvBuf) {
|
|
int i;
|
|
|
|
for( i = 0; i < pEnvBuf->cEntries; i++ ) {
|
|
MemFree( pEnvBuf->pszVars[i] );
|
|
}
|
|
}
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
// The user has made a choice among the entries.
|
|
// Now we have to arrange all the strings stored in NVRAM so
|
|
// that they have the same ordering. The selection is passed in,
|
|
// so what this function does is if selection is M, it makes the Mth item
|
|
// appear first in each of the 5 environment strings and the other items
|
|
// follow it in the list.
|
|
//
|
|
// Then if the timeout button is checked, it updates the AUTOLOAD variable
|
|
// to "yes" and set the COUNTDOWN variable to the number of seconds in the
|
|
// edit control.
|
|
//
|
|
// BUGBUG!!! - We need to add a bool parameter that indicates whether or
|
|
// not the strings should be freed. 11-Mar-1996 JonPa
|
|
//
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
BOOL UpdateNVRAM(HWND hdlg, int selection, int timeout)
|
|
{
|
|
CHAR *rgszVRAM[5] = { "SYSTEMPARTITION",
|
|
"OSLOADER",
|
|
"OSLOADPARTITION",
|
|
"OSLOADFILENAME",
|
|
"OSLOADOPTIONS"
|
|
};
|
|
CPEnvBuf rgcpeb[5];
|
|
|
|
|
|
WRITENVRAMPROC fpWriteNVRAMVar;
|
|
int iTemp, jTemp;
|
|
CHAR *pszSwap;
|
|
CHAR szTemp[10];
|
|
HMODULE hmodSetupDLL;
|
|
BOOL bChecked;
|
|
|
|
|
|
// args and charray are needed for call to SetNVRamVar() in SETUP
|
|
PSZ args[2];
|
|
CHAR chArray[CP_MAX_ENV];
|
|
PSZ pszReturn;
|
|
|
|
|
|
if ((hmodSetupDll = LoadLibrary (TEXT("setupdll"))) == NULL)
|
|
return(FALSE);
|
|
|
|
fpWriteNVRAMVar = (WRITENVRAMPROC) GetProcAddress(hmodSetupDll, "SetNVRAMVar");
|
|
if (fpWriteNVRAMVar == NULL)
|
|
return(FALSE);
|
|
|
|
// 0 is always the selection when the dialog is brought up,
|
|
// so as an optimization don't update the nvram if it's
|
|
// not necessary.
|
|
if (selection != 0)
|
|
{
|
|
// read in the strings from NVRAM. the number of strings (other than
|
|
// LOADIDENTIFIER is 5)
|
|
for (iTemp = 0; iTemp < 5; iTemp++)
|
|
{
|
|
GetRGSZEnvVar (&rgcpeb[iTemp], rgszVRAM[iTemp]);
|
|
// now re-order the strings to swap the 'selection-th' item
|
|
// string with the first string.
|
|
pszSwap = rgcpeb[iTemp].pszVars[0];
|
|
rgcpeb[iTemp].pszVars[0] = rgcpeb[iTemp].pszVars[selection];
|
|
rgcpeb[iTemp].pszVars[selection] = pszSwap;
|
|
}
|
|
// now do the same for the LOADIDENTIFIER, (this was set up earlier
|
|
// in the processing the INITDIALOG message).
|
|
pszSwap = CPEBOSLoadIdentifier.pszVars[0];
|
|
CPEBOSLoadIdentifier.pszVars[0] = CPEBOSLoadIdentifier.pszVars[selection];
|
|
CPEBOSLoadIdentifier.pszVars[selection] = pszSwap;
|
|
|
|
// now write to NVRAM: first write LOADIDENTIFIER, then the other 5
|
|
// variables.
|
|
args[0] = (PSZ)"LOADIDENTIFIER";
|
|
args[1] = chArray;
|
|
|
|
chArray[0] = '\0';
|
|
for (iTemp = 0; iTemp < CPEBOSLoadIdentifier.cEntries; iTemp++)
|
|
{
|
|
lstrcatA (chArray, CPEBOSLoadIdentifier.pszVars[iTemp]);
|
|
lstrcatA (chArray, ";");
|
|
}
|
|
// remove the last semi-colon:
|
|
chArray[lstrlenA(chArray)-1] = '\0';
|
|
|
|
fpWriteNVRAMVar ((DWORD)2, args, &pszReturn);
|
|
|
|
for (iTemp = 0; iTemp < 5; iTemp++)
|
|
{
|
|
args[0] = rgszVRAM[iTemp];
|
|
args[1] = chArray;
|
|
chArray[0] = '\0';
|
|
for (jTemp = 0; jTemp < rgcpeb[iTemp].cEntries; jTemp++)
|
|
{
|
|
lstrcatA (chArray, rgcpeb[iTemp].pszVars[jTemp]);
|
|
lstrcatA (chArray, ";");
|
|
}
|
|
chArray[lstrlenA(chArray)-1] = '\0';
|
|
|
|
fpWriteNVRAMVar ((DWORD)2, args, &pszReturn);
|
|
|
|
// We are done with this variable... Free the resources it consumes
|
|
FreeRGSZEnvVar( &rgcpeb[iTemp] );
|
|
}
|
|
|
|
}
|
|
args[0] = "AUTOLOAD";
|
|
if (bChecked = IsDlgButtonChecked (hdlg, IDC_STARTUP_SYS_ENABLECOUNTDOWN))
|
|
args[1] = "YES";
|
|
else
|
|
args[1] = "";
|
|
|
|
fpWriteNVRAMVar ((DWORD)2, args, &pszReturn);
|
|
if (bChecked)
|
|
{
|
|
args[0] = "COUNTDOWN";
|
|
IntToStringA(timeout, szTemp);
|
|
args[1] = szTemp;
|
|
fpWriteNVRAMVar ((DWORD)2, args, &pszReturn);
|
|
}
|
|
FreeLibrary (hmodSetupDll);
|
|
|
|
return TRUE;
|
|
}
|
|
#endif // _MIPS_ || _ALPHA_ || _PPC_
|
|
|
|
#if defined(_MIPS_) || defined(_ALPHA_) || defined(_PPC_)
|
|
#else
|
|
BOOL WriteableBootIni( LPTSTR szBootIni, HWND hDlg ) {
|
|
BOOL bOK;
|
|
DWORD dwFileAttr;
|
|
HANDLE hFile;
|
|
|
|
bOK = TRUE;
|
|
|
|
// Change Read-only file attrs on Boot.ini file if necessary
|
|
if ((dwFileAttr = GetFileAttributes (szBootIni)) != 0xFFFFFFFF) {
|
|
if (dwFileAttr & FILE_ATTRIBUTE_READONLY) {
|
|
if (!SetFileAttributes (szBootIni, dwFileAttr & ~FILE_ATTRIBUTE_READONLY))
|
|
{
|
|
bOK = FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (bOK)
|
|
{
|
|
|
|
hFile = CreateFile( szBootIni, GENERIC_WRITE | GENERIC_READ, FILE_SHARE_WRITE | FILE_SHARE_READ, NULL,
|
|
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL );
|
|
|
|
if (hFile != INVALID_HANDLE_VALUE) {
|
|
CloseHandle(hFile);
|
|
} else {
|
|
if (GetLastError() == ERROR_FILE_NOT_FOUND || GetLastError() == ERROR_PATH_NOT_FOUND) {
|
|
MsgBoxParam (hDlg, SYSTEM+39, INITS+1, MB_OK | MB_ICONEXCLAMATION, szBootIni);
|
|
}
|
|
bOK = FALSE;
|
|
}
|
|
|
|
// Restore read-only attr, if necessary, after writes
|
|
if (dwFileAttr != 0xFFFFFFFF && (dwFileAttr & FILE_ATTRIBUTE_READONLY)) {
|
|
SetFileAttributes (szBootIni, dwFileAttr);
|
|
}
|
|
}
|
|
|
|
return bOK;
|
|
}
|
|
#endif
|
|
|
|
void StartListInit( HWND hDlg, WPARAM wParam, LPARAM lParam ) {
|
|
HWND hwndTemp;
|
|
HMODULE hmodSetupDll;
|
|
int iTemp;
|
|
int n;
|
|
#if defined(_MIPS_) || defined(_ALPHA_) || defined(_PPC_)
|
|
#else
|
|
int i, timeout;
|
|
TCHAR szTemp2[MAX_PATH];
|
|
int selection;
|
|
TCHAR *pszKeyName;
|
|
LPTSTR pszLine;
|
|
TCHAR *pszValue;
|
|
TCHAR *pszTemp;
|
|
LPTSTR lpKey = NULL;
|
|
|
|
// ANSI string pointers
|
|
|
|
LPSTR pszSectionA;
|
|
#endif
|
|
|
|
#if defined(_MIPS_) || defined(_ALPHA_) || defined(_PPC_)
|
|
////////////////////////////////////////////////////////////////////
|
|
// Read info from NVRAM environment variables
|
|
////////////////////////////////////////////////////////////////////
|
|
|
|
// Init to 0 so we won't try to free garbage if we cant load setup.dll
|
|
CPEBOSLoadIdentifier.cEntries = 0;
|
|
|
|
fCanUpdateNVRAM = FALSE;
|
|
fAutoLoad = FALSE;
|
|
hwndTemp = GetDlgItem (hDlg, IDC_STARTUP_SYS_OS);
|
|
if (hmodSetupDll = LoadLibrary(TEXT("setupdll")))
|
|
{
|
|
if (fpGetNVRAMvar = (GETNVRAMPROC)GetProcAddress(hmodSetupDll, "GetNVRAMVar"))
|
|
{
|
|
if (fCanUpdateNVRAM = GetRGSZEnvVar (&CPEBOSLoadIdentifier, "LOADIDENTIFIER"))
|
|
{
|
|
for (iTemp = 0; iTemp < CPEBOSLoadIdentifier.cEntries; iTemp++)
|
|
n = SendMessageA (hwndTemp, CB_ADDSTRING, 0,
|
|
(LPARAM)CPEBOSLoadIdentifier.pszVars[iTemp]);
|
|
// the first one is the selection we want (offset 0)
|
|
SendMessage (hwndTemp, CB_SETCURSEL, 0, 0L);
|
|
SendDlgItemMessage (hDlg, IDC_STARTUP_SYS_SECONDS,
|
|
EM_LIMITTEXT, 3, 0L);
|
|
SendDlgItemMessage (hDlg, IDC_STARTUP_SYS_SECSCROLL,
|
|
UDM_SETRANGE, 0, (LPARAM)MAKELONG(999,0));
|
|
|
|
}
|
|
// fCanUpdateNVRAM is a global that gets set up above
|
|
if (fCanUpdateNVRAM)
|
|
{
|
|
CPEnvBuf cpebAutoLoad, cpebTimeout;
|
|
// is Autoload == YES? if so, check the checkbox
|
|
// and read setting for timeouts
|
|
// autoload == NO? disable edit control.
|
|
if (GetRGSZEnvVar (&cpebAutoLoad, "AUTOLOAD"))
|
|
{
|
|
if (!lstrcmpiA (cpebAutoLoad.pszVars[0], "yes"))
|
|
{
|
|
fAutoLoad = TRUE;
|
|
if (GetRGSZEnvVar (&cpebTimeout, "COUNTDOWN")) {
|
|
SetDlgItemInt (hDlg, IDC_STARTUP_SYS_SECONDS,
|
|
StringToIntA (cpebTimeout.pszVars[0]), FALSE);
|
|
|
|
// We are done with cpebTimeout... Free the resources it consumes
|
|
FreeRGSZEnvVar(&cpebTimeout);
|
|
}
|
|
|
|
}
|
|
|
|
// We are done with cpebTimeout... Free the resources it consumes
|
|
FreeRGSZEnvVar(&cpebAutoLoad);
|
|
}
|
|
CheckDlgButton (hDlg, IDC_STARTUP_SYS_ENABLECOUNTDOWN, fAutoLoad);
|
|
if (!fAutoLoad)
|
|
{
|
|
EnableWindow (GetDlgItem (hDlg, IDC_STARTUP_SYS_SECONDS), FALSE);
|
|
EnableWindow (GetDlgItem (hDlg, IDC_STARTUP_SYS_SECSCROLL), FALSE);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// if can't set variables (no privilege), disable controls
|
|
EnableWindow (GetDlgItem(hDlg, IDC_STARTUP_SYS_SECONDS), FALSE);
|
|
EnableWindow (GetDlgItem(hDlg, IDC_STARTUP_SYS_ENABLECOUNTDOWN), FALSE);
|
|
EnableWindow (GetDlgItem(hDlg, IDC_STARTUP_SYS_SECSCROLL), FALSE);
|
|
EnableWindow (GetDlgItem(hDlg, IDC_STARTUP_SYS_OS), FALSE);
|
|
}
|
|
}
|
|
FreeLibrary (hmodSetupDll);
|
|
}
|
|
|
|
// default to 5 seconds for now.
|
|
|
|
#else // _MIPS_ || _ALPHA_ || _PPC_
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Read info from boot.ini file and initialize OS Group box items
|
|
////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
// Get correct Boot Drive - this was added because after someone
|
|
// boots the system, they can ghost or change the drive letter
|
|
// of their boot drive from "c:" to something else.
|
|
//
|
|
|
|
// szBootIni[0] = GetBootDrive();
|
|
szBootIni[0] = x86DetermineSystemPartition (hDlg);
|
|
|
|
szBootIniA[0] = (char) szBootIni[0];
|
|
|
|
//
|
|
// Make sure we have access to BOOT.INI
|
|
//
|
|
if (!WriteableBootIni(szBootIni, hDlg)) {
|
|
// if can't set variables (no privilege), disable controls
|
|
EnableWindow (GetDlgItem(hDlg, IDC_STARTUP_SYS_SECONDS), FALSE);
|
|
EnableWindow (GetDlgItem(hDlg, IDC_STARTUP_SYS_ENABLECOUNTDOWN), FALSE);
|
|
EnableWindow (GetDlgItem(hDlg, IDC_STARTUP_SYS_SECSCROLL), FALSE);
|
|
EnableWindow (GetDlgItem(hDlg, IDC_STARTUP_SYS_OS), FALSE);
|
|
}
|
|
|
|
//
|
|
// Determine which section [boot loader]
|
|
// [flexboot]
|
|
// or [multiboot] is in file
|
|
//
|
|
n = GetPrivateProfileString (szBootLdr, NULL, NULL, szTemp2,
|
|
ARRAYSIZE(szTemp2), szBootIni);
|
|
if (n != 0)
|
|
pszBoot = szBootLdr;
|
|
else
|
|
{
|
|
n = GetPrivateProfileString (szFlexBoot, NULL, NULL, szTemp2,
|
|
ARRAYSIZE(szTemp2), szBootIni);
|
|
if (n != 0)
|
|
pszBoot = szFlexBoot;
|
|
else
|
|
{
|
|
n = GetPrivateProfileString (szMultiBoot, NULL, NULL, szTemp2,
|
|
ARRAYSIZE(szTemp2), szBootIni);
|
|
if (n != 0)
|
|
{
|
|
pszBoot = szMultiBoot;
|
|
}
|
|
else
|
|
{
|
|
//
|
|
// This final case is here because I want to DEFAULT
|
|
// to "[boot loader]" as the section name to use in
|
|
// the event that we do not find any section in the
|
|
// boot.ini file.
|
|
//
|
|
|
|
pszBoot = szBootLdr;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// Get info under [*pszBoot] section - timeout & default OS path
|
|
|
|
timeout = GetPrivateProfileInt (pszBoot, szTimeout, 0, szBootIni);
|
|
|
|
SetDlgItemInt (hDlg, IDC_STARTUP_SYS_SECONDS, timeout, FALSE);
|
|
|
|
nOriginalTimeout = timeout;
|
|
|
|
//
|
|
// Get the "Default" os selection
|
|
//
|
|
|
|
szTemp2[0] = TEXT('\0');
|
|
|
|
GetPrivateProfileString (pszBoot, szDefault, NULL, szTemp2,
|
|
ARRAYSIZE(szTemp2), szBootIni);
|
|
|
|
//
|
|
// Display all choices under [operating system] in boot.ini file
|
|
// in combobox for selection
|
|
//
|
|
|
|
hwndTemp = GetDlgItem (hDlg, IDC_STARTUP_SYS_OS);
|
|
|
|
selection = -1;
|
|
|
|
//
|
|
// ANSI Buffer!
|
|
//
|
|
|
|
pszSectionA = (LPSTR) MemAlloc (LPTR, BUFZ);
|
|
|
|
|
|
//
|
|
// Get entire section under OS to properly show user choices
|
|
//
|
|
|
|
n = GetPrivateProfileSectionA (szOSA, pszSectionA, BUFZ, szBootIniA);
|
|
|
|
if ((n >= BUFZ-2) || (n == 0))
|
|
{
|
|
ErrorReadingSection:
|
|
// Error reading data
|
|
MemFree ((LPVOID)pszSectionA);
|
|
goto ContinueSystemInit;
|
|
}
|
|
|
|
//
|
|
// Check for api errors and NoOptions
|
|
//
|
|
|
|
if ((pszSectionA == NULL) || ((*pszSectionA == '\0') && (*(pszSectionA+1) == '\0')))
|
|
goto ErrorReadingSection;
|
|
|
|
|
|
lpKey = MemAlloc (LPTR, BUFZ*sizeof(TCHAR));
|
|
pszKeyName = lpKey;
|
|
|
|
//
|
|
// Convert entire buffer from OEM to UNICODE
|
|
//
|
|
|
|
MultiByteToWideChar (CP_OEMCP, MB_PRECOMPOSED, pszSectionA, n+2, pszKeyName, BUFZ);
|
|
|
|
MemFree ((LPVOID)pszSectionA);
|
|
|
|
//
|
|
// Continue until we reach end of buffer, marked by Double '\0'
|
|
//
|
|
|
|
while (*(pszKeyName+1) != TEXT('\0'))
|
|
{
|
|
pszLine = pszKeyName;
|
|
|
|
//
|
|
// Get pointer to next line in buffer.
|
|
//
|
|
|
|
pszKeyName += lstrlen (pszKeyName) + 1;
|
|
|
|
//
|
|
// Find LHS/RHS delimiter to separate strings
|
|
//
|
|
|
|
pszValue = StrStr(pszLine, TEXT("="));
|
|
|
|
if (pszValue && (pszValue != pszLine))
|
|
{
|
|
*pszValue = '\0';
|
|
pszValue++;
|
|
}
|
|
else
|
|
{
|
|
pszValue = pszLine;
|
|
}
|
|
|
|
//
|
|
// Put it into combobox (if no descriptive name, use path)
|
|
//
|
|
|
|
n = SendMessage (hwndTemp, CB_ADDSTRING, 0, (LONG) (LPTSTR) pszValue);
|
|
|
|
//
|
|
// Find the first selection that matches the "default" selection
|
|
//
|
|
|
|
if ((selection == -1) && !lstrcmp (pszLine, szTemp2))
|
|
selection = n;
|
|
|
|
//
|
|
// Also attach pointer to KeyName (i.e. boot path) to each item
|
|
//
|
|
|
|
pszTemp = CloneString (pszLine);
|
|
|
|
SendMessage (hwndTemp, CB_SETITEMDATA, n, (LONG)pszTemp);
|
|
}
|
|
|
|
// If no selection was found up to this point, choose 0, because
|
|
// that is the default value that loader would choose.
|
|
|
|
if (selection == -1)
|
|
selection = 0;
|
|
|
|
SendDlgItemMessage (hDlg, IDC_STARTUP_SYS_SECONDS, EM_LIMITTEXT, 3, 0L);
|
|
SendDlgItemMessage (hDlg, IDC_STARTUP_SYS_SECSCROLL,
|
|
UDM_SETRANGE, 0, (LPARAM)MAKELONG(999,0));
|
|
|
|
|
|
// This call should force correct settings for the checkbox
|
|
// and "Showlist for xx seconds" controls
|
|
|
|
nOriginalSelection = selection;
|
|
|
|
SendMessage (hwndTemp, CB_SETCURSEL, selection, 0L);
|
|
|
|
ContinueSystemInit:
|
|
|
|
if (lpKey) {
|
|
MemFree (lpKey);
|
|
}
|
|
|
|
#endif // _MIPS_ || _ALPHA_ || _PPC_
|
|
}
|
|
|
|
|
|
//
|
|
// BUGBUG!!! - We need an and extra parameter that indicates
|
|
// whether we should save the variables out or not.
|
|
//
|
|
// IE. We need to handle CANCLE case.
|
|
//
|
|
int StartListExit(HWND hDlg, WPARAM wParam, LPARAM lParam ) {
|
|
HWND hwndTemp;
|
|
int selection, timeout;
|
|
#if defined(_MIPS_) || defined(_ALPHA_) || defined(_PPC_)
|
|
#else
|
|
// X86 only vars
|
|
DWORD dwFileAttr;
|
|
BOOL bOK;
|
|
HANDLE hKey;
|
|
TCHAR *pszKeyName;
|
|
int i, n;
|
|
DWORD dwBufz;
|
|
LPSTR pszSectionA;
|
|
TCHAR *pszTemp;
|
|
TCHAR szTemp[MAX_PATH];
|
|
TCHAR szTemp2[MAX_PATH];
|
|
int iTemp;
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////
|
|
// Write new info to boot.ini file
|
|
/////////////////////////////////////////////////////////////////
|
|
|
|
hwndTemp = GetDlgItem (hDlg, IDC_STARTUP_SYS_OS);
|
|
|
|
selection = SendMessage (hwndTemp, CB_GETCURSEL, 0, 0L);
|
|
|
|
if ((selection == CB_ERR) || (selection == CB_ERRSPACE))
|
|
selection = nOriginalSelection;
|
|
|
|
timeout = GetDlgItemInt (hDlg, IDC_STARTUP_SYS_SECONDS, NULL, FALSE);
|
|
|
|
#if defined(_MIPS_) || defined(_ALPHA_) || defined(_PPC_)
|
|
if (fCanUpdateNVRAM) {
|
|
TCHAR szTextNew[MAX_PATH];
|
|
TCHAR szTextTop[MAX_PATH];
|
|
|
|
UpdateNVRAM (hDlg, selection, timeout);
|
|
|
|
/*
|
|
* Now reorder list to match NVRAM
|
|
*/
|
|
// Get the current text
|
|
SendMessage( hwndTemp, CB_GETLBTEXT, selection, (LPARAM)szTextNew );
|
|
SendMessage( hwndTemp, CB_GETLBTEXT, 0, (LPARAM)szTextTop );
|
|
|
|
// Set the new text to the 0th entry in the list
|
|
SendMessage( hwndTemp, CB_DELETESTRING, 0, 0 );
|
|
SendMessage( hwndTemp, CB_INSERTSTRING, 0, (LPARAM)szTextNew);
|
|
|
|
// Set old top text selected item
|
|
SendMessage( hwndTemp, CB_DELETESTRING, selection, 0 );
|
|
SendMessage( hwndTemp, CB_INSERTSTRING, selection, (LPARAM)szTextTop);
|
|
|
|
// Now point the current selection back to the top of the list, so it matches
|
|
// what the user just chose
|
|
SendMessage( hwndTemp, CB_SETCURSEL, 0, 0);
|
|
}
|
|
#else // _MIPS_ || _ALPHA_ || _PPC_
|
|
if ((selection != nOriginalSelection) || (timeout != nOriginalTimeout))
|
|
{
|
|
bOK = TRUE;
|
|
|
|
// Change Read-only file attrs on Boot.ini file if necessary
|
|
if ((dwFileAttr = GetFileAttributes (szBootIni)) != 0xFFFFFFFF)
|
|
if (dwFileAttr & FILE_ATTRIBUTE_READONLY)
|
|
if (!SetFileAttributes (szBootIni,
|
|
dwFileAttr & ~FILE_ATTRIBUTE_READONLY))
|
|
{
|
|
BootIniWriteError:
|
|
bOK = FALSE;
|
|
MsgBoxParam (hDlg, SYSTEM+19, INITS+1, MB_OK | MB_ICONINFORMATION);
|
|
}
|
|
|
|
if (bOK)
|
|
{
|
|
//
|
|
// Write new [operating systems] section and
|
|
// set "default" selection in boot.ini file.
|
|
//
|
|
|
|
if (selection != nOriginalSelection)
|
|
{
|
|
//
|
|
// Allocate buffers for new section
|
|
//
|
|
|
|
hKey = MemAlloc (LPTR, BUFZ*sizeof(TCHAR));
|
|
|
|
pszKeyName = (TCHAR *) hKey;
|
|
|
|
//
|
|
// Total profile section buffer size
|
|
//
|
|
|
|
i = dwBufz = 0;
|
|
|
|
pszSectionA = (LPSTR) MemAlloc (LPTR, BUFZ);
|
|
|
|
//
|
|
// Get the User's selection and write it in the
|
|
// section buffer first. Then get all other items.
|
|
//
|
|
|
|
pszTemp = (LPTSTR) SendMessage (hwndTemp,
|
|
CB_GETITEMDATA,
|
|
selection, 0L);
|
|
|
|
SendMessage (hwndTemp, CB_GETLBTEXT, selection,
|
|
(LONG) (LPTSTR) szTemp);
|
|
|
|
if (pszTemp != (LPTSTR) CB_ERR)
|
|
{
|
|
lstrcpy (pszKeyName, pszTemp);
|
|
lstrcat (pszKeyName, TEXT("="));
|
|
lstrcat (pszKeyName, szTemp);
|
|
|
|
i = lstrlen (pszKeyName) + 1;
|
|
pszKeyName += i;
|
|
|
|
//
|
|
// Set "default" selection in boot.ini file
|
|
//
|
|
|
|
if (!WritePrivateProfileString (pszBoot,
|
|
szDefault,
|
|
pszTemp,
|
|
szBootIni))
|
|
{
|
|
goto BootIniWriteError;
|
|
}
|
|
}
|
|
|
|
dwBufz = (DWORD) i;
|
|
|
|
//
|
|
// Get the rest of the selections
|
|
//
|
|
|
|
n = SendMessage (hwndTemp, CB_GETCOUNT, 0, 0L);
|
|
|
|
if (n != LB_ERR)
|
|
{
|
|
for (iTemp = 0; iTemp < n; iTemp++)
|
|
{
|
|
//
|
|
// Skip the User's selection since we got it
|
|
// above.
|
|
//
|
|
|
|
if (iTemp == selection)
|
|
continue;
|
|
|
|
pszTemp = (LPTSTR) SendMessage (hwndTemp,
|
|
CB_GETITEMDATA,
|
|
iTemp, 0L);
|
|
|
|
SendMessage (hwndTemp, CB_GETLBTEXT, iTemp,
|
|
(LONG) (LPTSTR) szTemp);
|
|
|
|
if (pszTemp != (LPTSTR) CB_ERR)
|
|
{
|
|
lstrcpy (pszKeyName, pszTemp);
|
|
lstrcat (pszKeyName, TEXT("="));
|
|
lstrcat (pszKeyName, szTemp);
|
|
|
|
i = lstrlen (pszKeyName) + 1;
|
|
pszKeyName += i;
|
|
|
|
dwBufz += (DWORD) i;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// Double-Null terminate the buffer
|
|
//
|
|
|
|
*pszKeyName = TEXT('\0');
|
|
dwBufz++;
|
|
|
|
pszKeyName = (TCHAR *) hKey;
|
|
|
|
//
|
|
// Convert entire buffer from UNICODE to OEM
|
|
//
|
|
|
|
WideCharToMultiByte (CP_OEMCP, WC_COMPOSITECHECK,
|
|
pszKeyName, dwBufz,
|
|
pszSectionA, BUFZ,
|
|
NULL, NULL);
|
|
|
|
//
|
|
// Write new section under OS
|
|
//
|
|
|
|
if (!WritePrivateProfileSectionA (szOSA,
|
|
pszSectionA,
|
|
szBootIniA))
|
|
{
|
|
MemFree ((LPVOID)pszSectionA);
|
|
MemFree ((LPVOID)hKey);
|
|
goto BootIniWriteError;
|
|
}
|
|
|
|
MemFree ((LPVOID)pszSectionA);
|
|
MemFree ((LPVOID)hKey);
|
|
}
|
|
|
|
if (timeout != nOriginalTimeout)
|
|
{
|
|
GetDlgItemText (hDlg, IDC_STARTUP_SYS_SECONDS, szTemp2, sizeof(szTemp2));
|
|
|
|
if (!CheckVal (hDlg, IDC_STARTUP_SYS_SECONDS, FORMIN, FORMAX, SYSTEM+4))
|
|
return RET_BREAK;
|
|
|
|
// Write timeout value to file
|
|
|
|
if (!WritePrivateProfileString (pszBoot, szTimeout,
|
|
szTemp2, szBootIni))
|
|
goto BootIniWriteError;
|
|
}
|
|
|
|
// Restore read-only attr, if necessary, after writes
|
|
if (dwFileAttr != 0xFFFFFFFF && (dwFileAttr & FILE_ATTRIBUTE_READONLY)) {
|
|
SetFileAttributes (szBootIni, dwFileAttr);
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif // _MIPS_ || _ALPHA_ || _PPC_
|
|
|
|
return RET_CONTINUE;
|
|
|
|
}
|
|
|
|
|
|
BOOL CheckVal( HWND hDlg, WORD wID, WORD wMin, WORD wMax, WORD wMsgID )
|
|
{
|
|
WORD nVal;
|
|
BOOL bOK;
|
|
HWND hVal;
|
|
|
|
if( wMin > wMax )
|
|
{
|
|
nVal = wMin;
|
|
wMin = wMax;
|
|
wMax = nVal;
|
|
}
|
|
|
|
nVal = (WORD) GetDlgItemInt( hDlg, wID, &bOK, FALSE );
|
|
|
|
if( !bOK || ( nVal < wMin ) || ( nVal > wMax ) )
|
|
{
|
|
MsgBoxParam( hDlg, wMsgID, INITS + 1,
|
|
MB_OK | MB_ICONINFORMATION, wMin, wMax );
|
|
|
|
SendMessage( hDlg, WM_NEXTDLGCTL,
|
|
(WPARAM) ( hVal = GetDlgItem( hDlg, wID ) ), 1L );
|
|
|
|
// SendMessage(hVal, EM_SETSEL, NULL, MAKELONG(0, 32767));
|
|
|
|
SendMessage( hVal, EM_SETSEL, 0, 32767 );
|
|
|
|
return( FALSE );
|
|
}
|
|
|
|
return( TRUE );
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////
|
|
//
|
|
// Frees the data (if any) associated with the strings in the combo box
|
|
//
|
|
//////////////////////////////////////////////////////
|
|
void StartListDestroy(HWND hDlg, WPARAM wParam, LPARAM lParam) {
|
|
#if defined(_MIPS_) || defined(_ALPHA_) || defined(_PPC_)
|
|
|
|
// Reference vars to make compiler happy
|
|
FreeRGSZEnvVar(&CPEBOSLoadIdentifier);
|
|
return;
|
|
|
|
(void)hDlg;
|
|
(void)wParam;
|
|
(void)lParam;
|
|
|
|
|
|
#else
|
|
|
|
// Only X86 has data in the combo
|
|
int n;
|
|
HWND hwndTemp;
|
|
int iTemp;
|
|
TCHAR *pszTemp;
|
|
|
|
|
|
//
|
|
// Free strings stored in the listbox
|
|
//
|
|
hwndTemp = GetDlgItem (hDlg, IDC_STARTUP_SYS_OS);
|
|
|
|
n = SendMessage (hwndTemp, CB_GETCOUNT, 0, 0L);
|
|
|
|
if (n != LB_ERR)
|
|
{
|
|
for (iTemp = 0; iTemp < n; iTemp++)
|
|
{
|
|
pszTemp = (LPTSTR) SendMessage (hwndTemp,
|
|
CB_GETITEMDATA,
|
|
iTemp, 0L);
|
|
|
|
if (pszTemp != (LPTSTR) CB_ERR)
|
|
{
|
|
MemFree (pszTemp);
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif // _MIPS_ || _ALPHA_ || _PPC_
|
|
}
|