#include "precomp.h"
#pragma hdrstop
#include <spidgen.h>
#include <pencrypt.h>
#include "digpid.h"
#define SETUP_TYPE_BUFFER_LEN 8
#define MAX_PID30_SITE 3
#define MAX_PID30_RPC 5
TCHAR Pid30Rpc[7] = TEXT("00000");
TCHAR Pid30Site[4];
LONG SourceInstallType = RetailInstall;
BOOL EulaComplete = TRUE;
#ifdef PRERELEASE
BOOL NoPid = FALSE;
#endif
//
// global variable used for subclassing.
//
WNDPROC OldPidEditProc[5];
WNDPROC OldEulaEditProc;
static BOOL PidMatchesMedia;
BOOL
ValidatePid30(
LPTSTR Edit1,
LPTSTR Edit2,
LPTSTR Edit3,
LPTSTR Edit4,
LPTSTR Edit5
);
void GetPID();
VOID ShowPidBox();
VOID
GetSourceInstallType(
OUT OPTIONAL LPDWORD InstallVariation
)
/*++
Routine Description:
Determines the installation type (by looking in setupp.ini in the source directory)
Arguments:
Installvaration - one of the install variations defined in compliance.h
Returns:
none. sets SourceInstallType global variable.
--*/
{
TCHAR TypeBuffer[256];
TCHAR FilePath[MAX_PATH];
DWORD InstallVar = COMPLIANCE_INSTALLVAR_UNKNOWN;
TCHAR MPCode[6] = { -1 };
//
// SourcePaths is guaranteed to be valid at this point, so just use it
//
lstrcpy(FilePath,NativeSourcePaths[0]);
ConcatenatePaths (FilePath, SETUPP_INI, MAX_PATH );
GetPrivateProfileString(PID_SECTION,
PID_KEY,
TEXT(""),
TypeBuffer,
sizeof(TypeBuffer)/sizeof(TCHAR),
FilePath);
if (lstrlen(TypeBuffer)==SETUP_TYPE_BUFFER_LEN) {
if (lstrcmp(&TypeBuffer[5], OEM_INSTALL_RPC) == 0) {
SourceInstallType = OEMInstall;
InstallVar = COMPLIANCE_INSTALLVAR_OEM;
} else if (lstrcmp(&TypeBuffer[5], SELECT_INSTALL_RPC) == 0) {
SourceInstallType = SelectInstall;
InstallVar = COMPLIANCE_INSTALLVAR_SELECT;
// Since Select also requires a PID, don't zero the PID and call.
/* // get/set the pid.
{
TCHAR Temp[5][ MAX_PID30_EDIT + 1 ];
Temp[0][0] = TEXT('\0');
ValidatePid30(Temp[0],Temp[1],Temp[2],Temp[3],Temp[4]);
}*/
} else if (lstrcmp(&TypeBuffer[5], MSDN_INSTALL_RPC) == 0) {
SourceInstallType = RetailInstall;
InstallVar = COMPLIANCE_INSTALLVAR_MSDN;
} else {
// defaulting
SourceInstallType = RetailInstall;
InstallVar = COMPLIANCE_INSTALLVAR_CDRETAIL;
}
lstrcpy(Pid30Site,&TypeBuffer[5]);
lstrcpyn(Pid30Rpc, TypeBuffer, 6 );
Pid30Rpc[MAX_PID30_RPC] = (TCHAR)0;
} else {
//
// the retail install doesn't have an RPC code in the PID, so it's shorter in length
//
SourceInstallType = RetailInstall;
InstallVar = COMPLIANCE_INSTALLVAR_CDRETAIL;
}
if (lstrlen(TypeBuffer) >= 5) {
lstrcpyn(MPCode, TypeBuffer, 6);
if (lstrcmp(MPCode, EVAL_MPC) == 0) {
InstallVar = COMPLIANCE_INSTALLVAR_EVAL;
} else if ((lstrcmp(MPCode, SRV_NFR_MPC) == 0) ||
(lstrcmp(MPCode, ASRV_NFR_MPC) == 0)) {
InstallVar = COMPLIANCE_INSTALLVAR_NFR;
}
}
if (InstallVariation){
*InstallVariation = InstallVar;
}
}
BOOL
SetPid30(
HWND hdlg,
LONG ExpectedPidType,
LPTSTR pProductId
)
/*++
Routine Description:
sets the pid in the wizard page to the data specified in the answer file.
Arguments:
hdlg - window handle to pid dialog
ExpectedPidType - InstallType enum identifying what sort of pid we're looking for.
pProductId - string passed in from unattend file
Returns:
true on successfully setting the data, false means the data was missing or invalid.
may set some the dialog text in the specified dialog
--*/
{
TCHAR *ptr;
TCHAR Temp[5][ MAX_PID30_EDIT + 1 ];
UINT i;
//
// make sure we were provided with a product ID
//
if (!pProductId || !*pProductId) {
return(FALSE);
}
if ( (ExpectedPidType != RetailInstall) &&
(ExpectedPidType != OEMInstall) &&
(ExpectedPidType != SelectInstall)
){
return(FALSE);
}
//
// OEM and cd retail are the same case
// Check that the string specified on the unattended script file
// represents a valid 25 digit product id:
//
// 1 2 3 4 5 - 1 2 3 4 5 - 1 2 3 4 5 - 1 2 3 4 5 - 1 2 3 4 5
// 0 1 2 3 4 5 6 7 8 9 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2
// 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8
//
// As a first validation test, we verify that the length is correct,
// then we check if the "-" characters are in the correct place
//
//
if( ( lstrlen( pProductId ) != (4+ MAX_PID30_EDIT*5)) ||
( pProductId[5] != (TCHAR)TEXT('-') ) ||
( pProductId[11] != (TCHAR)TEXT('-') ) ||
( pProductId[17] != (TCHAR)TEXT('-') ) ||
( pProductId[23] != (TCHAR)TEXT('-') )
) {
//
// The Pid in the unattended script file is invalid.
//
return(FALSE);
}
for (i = 0;i<5;i++) {
//
// quintet i
//
ptr = &pProductId[i*(MAX_PID30_EDIT+1)];
lstrcpyn(Temp[i], ptr, MAX_PID30_EDIT+1 );
Temp[i][MAX_PID30_EDIT] = (TCHAR)'\0';
}
//
// check with pid30 to make sure it's valid
//
if (!ValidatePid30(Temp[0],Temp[1],Temp[2],Temp[3],Temp[4])) {
return(FALSE);
}
//
// all of the specified pid items are valid, set the dialog text and return.
//
SetDlgItemText( hdlg,IDT_EDIT_PID1, Temp[0] );
SetDlgItemText( hdlg,IDT_EDIT_PID2, Temp[1] );
SetDlgItemText( hdlg,IDT_EDIT_PID3, Temp[2] );
SetDlgItemText( hdlg,IDT_EDIT_PID4, Temp[3] );
SetDlgItemText( hdlg,IDT_EDIT_PID5, Temp[4] );
return(TRUE);
}
BOOL
pGetCdKey (
OUT PBYTE CdKey
)
{
DIGITALPID dpid;
DWORD type;
DWORD rc;
HKEY key;
DWORD size = sizeof (dpid);
BOOL b = FALSE;
rc = RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion"), &key);
if (rc == ERROR_SUCCESS) {
rc = RegQueryValueEx (key, TEXT("DigitalProductId"), NULL, &type, (LPBYTE)&dpid, &size);
if (rc == ERROR_SUCCESS && type == REG_BINARY) {
CopyMemory (CdKey, &dpid.abCdKey, sizeof (dpid.abCdKey));
b = TRUE;
}
RegCloseKey (key);
}
return b;
}
const unsigned int iBase = 24;
//
// obtained from Jim Harkins 11/27/2000
//
void EncodePid3g(
TCHAR *pchCDKey3Chars, // [OUT] pointer to 29+1 character Secure Product key
LPBYTE pbCDKey3) // [IN] pointer to 15-byte binary Secure Product Key
{
// Given the binary PID 3.0 we need to encode
// it into ASCII characters. We're only allowed to
// use 24 characters so we need to do a base 2 to
// base 24 conversion. It's just like any other
// base conversion execpt the numbers are bigger
// so we have to do the long division ourselves.
const TCHAR achDigits[] = TEXT("BCDFGHJKMPQRTVWXY2346789");
int iCDKey3Chars = 29;
int cGroup = 0;
pchCDKey3Chars[iCDKey3Chars--] = TEXT('\0');
while (0 <= iCDKey3Chars)
{
unsigned int i = 0; // accumulator
int iCDKey3;
for (iCDKey3 = 15-1; 0 <= iCDKey3; --iCDKey3)
{
i = (i * 256) + pbCDKey3[iCDKey3];
pbCDKey3[iCDKey3] = (BYTE)(i / iBase);
i %= iBase;
}
// i now contains the remainder, which is the current digit
pchCDKey3Chars[iCDKey3Chars--] = achDigits[i];
// add '-' between groups of 5 chars
if (++cGroup % 5 == 0 && iCDKey3Chars > 0)
{
pchCDKey3Chars[iCDKey3Chars--] = TEXT('-');
}
}
return;
}
LRESULT
CALLBACK
PidEditSubProc(
IN HWND hwnd,
IN UINT msg,
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
Edit control subclass routine, sets the focus to the correct edit box when the user enters text.
This routine assumes that the pid controls ids are in sequential order.
Arguments:
Standard window proc arguments.
Returns:
Message-dependent value.
--*/
{
DWORD len, id;
//
// eat spaces
//
if ((msg == WM_CHAR) && (wParam == VK_SPACE)) {
return(0);
}
if ((msg == WM_CHAR)) {
//
// First override: if we have the max characters in the current edit
// box, let's post the character to the next box and set focus to that
// control.
//
if ( ( (len = (DWORD)SendMessage(hwnd, WM_GETTEXTLENGTH, 0, 0)) == MAX_PID30_EDIT) &&
((wParam != VK_DELETE) && (wParam != VK_BACK)) ) {
//
// set the focus to the next edit control and post the character
// to that edit control
//
if ((id = GetDlgCtrlID(hwnd)) < IDT_EDIT_PID5 ) {
DWORD start, end;
SendMessage(hwnd, EM_GETSEL, (WPARAM)&start,(LPARAM)&end);
if (start == end) {
HWND hNext = GetDlgItem(GetParent(hwnd),id+1);
SetFocus(hNext);
SendMessage(hNext, EM_SETSEL, (WPARAM)-1,(LPARAM)-1);
PostMessage( GetDlgItem(GetParent(hwnd),id+1), WM_CHAR, wParam, lParam );
return(0);
}
}
//
// Second override: if the user hit's a delete key and they are at the
// the start of an edit box, then post the delete to the previous edit
// box.
//
} else if ( (len == 0) &&
((id = GetDlgCtrlID(hwnd)) > IDT_EDIT_PID1) &&
((wParam == VK_DELETE) || (wParam == VK_BACK) )) {
//
// set the focus to the previous edit control and post the command
// to that edit control
//
HWND hPrev = GetDlgItem(GetParent(hwnd),id-1);
SetFocus(hPrev);
SendMessage(hPrev, EM_SETSEL, (WPARAM)MAX_PID30_EDIT-1,(LPARAM)MAX_PID30_EDIT);
PostMessage( hPrev, WM_CHAR, wParam, lParam );
return(0);
//
// Third override: if posting this message will give us the maximum
// characters in our in the current edit box, let's post the character
// to the next box and set focus to that control.
//
} else if ( (len == MAX_PID30_EDIT-1) &&
((wParam != VK_DELETE) && (wParam != VK_BACK)) &&
((id = GetDlgCtrlID(hwnd)) < IDT_EDIT_PID5) ) {
DWORD start, end;
SendMessage(hwnd, EM_GETSEL, (WPARAM)&start,(LPARAM)&end);
if (start == end) {
HWND hNext = GetDlgItem(GetParent(hwnd),id+1);
//
// post the message to the edit box
//
CallWindowProc(OldPidEditProc[GetDlgCtrlID(hwnd)-IDT_EDIT_PID1],hwnd,msg,wParam,lParam);
//
// now set the focus to the next edit control
//
SetFocus(hNext);
SendMessage(hNext, EM_SETSEL, (WPARAM)-1,(LPARAM)-1);
return(0);
}
}
}
return(CallWindowProc(OldPidEditProc[GetDlgCtrlID(hwnd)-IDT_EDIT_PID1],hwnd,msg,wParam,lParam));
}
LRESULT
CALLBACK
EulaEditSubProc(
IN HWND hwnd,
IN UINT msg,
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
Edit control subclass routine, to avoid highlighting text when user
tabs to the edit control.
Arguments:
Standard window proc arguments.
Returns:
Message-dependent value.
--*/
{
static BOOL firstTime = TRUE;
//
// For setsel messages, make start and end the same.
//
if((msg == EM_SETSEL) && ((LPARAM)wParam != lParam)) {
lParam = wParam;
}
//
// also, if the user hits tab, set focus to the correct radio button
// after the first time, the tab starts working like we want it to
//
if ((msg == WM_KEYDOWN) && (wParam == VK_TAB) && firstTime) {
firstTime = FALSE;
if (! ((IsDlgButtonChecked(GetParent(hwnd), IDYES) == BST_CHECKED) ||
(IsDlgButtonChecked(GetParent(hwnd), IDNO ) == BST_CHECKED))) {
CheckDlgButton( GetParent(hwnd),IDYES, BST_CHECKED );
PostMessage(GetParent(hwnd),WM_COMMAND,IDYES,0);
SetFocus(GetDlgItem(GetParent(hwnd),IDYES));
}
}
return(CallWindowProc(OldEulaEditProc,hwnd,msg,wParam,lParam));
}
LPTSTR
DoInitializeEulaText(
HWND EditControl,
PBOOL TranslationProblem
)
/*++
Routine Description:
retrieves the text out of eula.txt, and sets up the eula subclass routine
Arguments:
EditControl - window handle for edit control
TranslationProblem - if we fail, was is because we couldn't translate the
text?
Returns:
pointer to eula text so that it can be freed, NULL on failure
--*/
{
TCHAR EulaPath[MAX_PATH];
DWORD err;
HANDLE hFile, hFileMapping;
DWORD FileSize;
BYTE *pbFile;
LPTSTR EulaText = NULL;
int i;
// HFONT hFont;
// accoding to MSDN LOCALE_IDEFAULTANSICODEPAGE and LOCALE_IDEFAULTCODEPAGE
// are a max of 6 characters. Unsure of whether or not this includes ending
// null character, and we need a leading _T('.')
TCHAR cpName[8];
if (TranslationProblem) {
*TranslationProblem = FALSE;
}
//
// Map the file containing the licensing agreement.
//
lstrcpy(EulaPath, NativeSourcePaths[0]);
ConcatenatePaths (EulaPath, TEXT("eula.txt"), MAX_PATH );
//
// Open and map the inf file.
//
err = MapFileForRead(EulaPath,&FileSize,&hFile,&hFileMapping,&pbFile);
if(err != NO_ERROR) {
goto c0;
}
if(FileSize == 0xFFFFFFFF) {
goto c1;
}
EulaText = MALLOC((FileSize+1) * sizeof(TCHAR));
if(EulaText == NULL) {
goto c1;
}
#ifdef UNICODE
// the Eula will be in the language of the build, so we should set out locale
// to use the codepage that the source wants.
if(!GetLocaleInfo(SourceNativeLangID,LOCALE_IDEFAULTANSICODEPAGE,&cpName[1],sizeof(cpName)/sizeof(TCHAR))){
if(!GetLocaleInfo(SourceNativeLangID,LOCALE_IDEFAULTCODEPAGE,&cpName[1],sizeof(cpName)/sizeof(TCHAR))){
FREE(EulaText);
EulaText = NULL;
if (TranslationProblem){
*TranslationProblem = TRUE;
}
goto c1;
}
}
cpName[0] = _T('.');
_tsetlocale(LC_ALL,cpName);
//
// Translate the text from ANSI to Unicode.
//
if(!mbstowcs(EulaText,pbFile,FileSize)){
FREE(EulaText);
EulaText = NULL;
if (TranslationProblem) {
*TranslationProblem = TRUE;
}
goto c1;
}
_tsetlocale(LC_ALL,_T(""));
/*
// we use mbstowcs instead of MultiByteToWideChar because mbstowcs will
// take into account the code page of the locale, which we've just set
// to be in relation to the language of the build
if (!MultiByteToWideChar (
CP_ACP,
MB_ERR_INVALID_CHARS,
pbFile,
FileSize,
EulaText,
(FileSize+1) * sizeof(WCHAR)
) ) {
FREE( EulaText );
EulaText = NULL;
if (TranslationProblem) {
*TranslationProblem = TRUE;
}
goto c1;
}
*/
#else
CopyMemory(EulaText, pbFile, FileSize);
#endif
//
// add the trailing NULL character
//
EulaText[FileSize] = 0;
//
// setup the eula subclass
//
OldEulaEditProc = (WNDPROC)GetWindowLongPtr(EditControl,GWLP_WNDPROC);
SetWindowLongPtr(EditControl,GWLP_WNDPROC,(LONG_PTR)EulaEditSubProc);
#if 0
//
// need a fixed width font for the EULA so it's formatted correctly for all resolutions
//
hFont = GetStockObject(SYSTEM_FIXED_FONT);
if (hFont) {
SendMessage( EditControl, WM_SETFONT, hFont, TRUE );
}
#endif
//
// set the actual text
//
SetWindowText(EditControl,(LPCTSTR)EulaText);
c1:
UnmapFile( hFileMapping, pbFile );
CloseHandle( hFile );
c0:
return EulaText;
}
BOOL
EulaWizPage(
IN HWND hdlg,
IN UINT msg,
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
Eula wizard page.
Arguments:
Standard window proc arguments.
Returns:
Message-dependent value.
--*/
{
BOOL b;
PPAGE_RUNTIME_DATA WizPage = (PPAGE_RUNTIME_DATA)GetWindowLongPtr(hdlg,DWLP_USER);
static LPTSTR EulaText = NULL;
static BOOL ShowEula = TRUE;
BOOL TranslationProblem;
CHECKUPGRADEONLY();
switch(msg) {
case WM_INITDIALOG:
//
// check if the current language and target language match
//
ShowEula = IsLanguageMatched;
//
// Set eula radio buttons.
//
CheckDlgButton( hdlg,IDYES, BST_UNCHECKED );
CheckDlgButton( hdlg,IDNO, BST_UNCHECKED );
//
// setup the eula
//
EulaText = DoInitializeEulaText(
GetDlgItem( hdlg, IDT_EULA_LIC_TEXT ),
&TranslationProblem );
//
// if we couldn't read the eula, check if it was because of a translation problem,
// in which case we defer to textmode setup
//
if (!EulaText && TranslationProblem == TRUE) {
ShowEula = FALSE;
}
//
// if this fails, only bail out if we were going to show the EULA in the first place.
//
if (!EulaText && ShowEula) {
MessageBoxFromMessage(
hdlg,
MSG_EULA_FAILED,
FALSE,
AppTitleStringId,
MB_OK | MB_ICONERROR | MB_TASKMODAL
);
Cancelled = TRUE;
PropSheet_PressButton(GetParent(hdlg),PSBTN_CANCEL);
b = FALSE;
}
PropSheet_SetWizButtons( GetParent(hdlg),
WizPage->CommonData.Buttons & (~PSWIZB_NEXT)
);
//
// Set focus to radio button
//
SetFocus(GetDlgItem(hdlg,IDYES));
b = FALSE;
break;
case WM_COMMAND:
if (wParam == IDYES) {
PropSheet_SetWizButtons( GetParent(hdlg),
WizPage->CommonData.Buttons | PSWIZB_NEXT
);
b = TRUE;
} else if (wParam == IDNO) {
PropSheet_SetWizButtons( GetParent(hdlg),
WizPage->CommonData.Buttons | PSWIZB_NEXT
);
b = TRUE;
}
else
b = FALSE;
break;
case WMX_ACTIVATEPAGE:
b = TRUE;
if(wParam) {
//
// don't activate the page in restart mode
//
if (Winnt32RestartedWithAF ()) {
EulaComplete = TRUE;
return FALSE;
}
//
// activation
//
if (!ShowEula) {
//
// the target install language and the source language do not match up
// since this means that we might not have fonts installed for the current
// language, we'll just defer this to textmode setup where we know we have
// the correct fonts
//
EulaComplete = FALSE;
if (IsDlgButtonChecked(hdlg, IDYES) == BST_CHECKED) {
PropSheet_PressButton(GetParent(hdlg),
(lParam == PSN_WIZBACK) ? PSBTN_BACK : PSBTN_NEXT);
} else {
CheckDlgButton( hdlg,IDYES, BST_CHECKED );
PropSheet_PressButton(GetParent(hdlg),PSBTN_NEXT);
}
return(b);
}
//
// set state of next button if user has backed up and reentered this dialog
//
if ( (IsDlgButtonChecked(hdlg, IDYES) == BST_CHECKED) ||
(IsDlgButtonChecked(hdlg, IDNO ) == BST_CHECKED) ) {
PropSheet_SetWizButtons( GetParent(hdlg),
WizPage->CommonData.Buttons | PSWIZB_NEXT
);
} else {
SendMessage(GetParent(hdlg),
PSM_SETWIZBUTTONS,
0, (LPARAM)WizPage->CommonData.Buttons & (~PSWIZB_NEXT));
}
//
// Advance page in unattended case.
//
UNATTENDED(PSBTN_NEXT);
} else {
//
// deactivation
//
if (EulaText) FREE( EulaText );
EulaText = NULL;
if (IsDlgButtonChecked(hdlg, IDNO ) == BST_CHECKED) {
Cancelled = TRUE;
PropSheet_PressButton(GetParent(hdlg),PSBTN_CANCEL);
}
}
break;
case WMX_UNATTENDED:
//
// necessary?
//
if (EulaText) FREE( EulaText );
EulaText = NULL;
b = FALSE;
break;
case WMX_I_AM_VISIBLE:
//
// Force repainting first to make sure the page is visible.
//
InvalidateRect(hdlg,NULL,FALSE);
UpdateWindow(hdlg);
b = TRUE;
break;
default:
b = FALSE;
break;
}
return(b);
}
BOOL
SelectPid30WizPage(
IN HWND hdlg,
IN UINT msg,
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
OEM Pid wizard page. depends on SourceInstallType being set correctly.
Arguments:
Standard window proc arguments.
Returns:
Message-dependent value.
--*/
{
BOOL b;
PPAGE_RUNTIME_DATA WizPage = (PPAGE_RUNTIME_DATA)GetWindowLongPtr(hdlg,DWLP_USER);
static BOOL bUnattendPid = FALSE;
DWORD i;
switch(msg) {
case WM_INITDIALOG:
GetPID();
// Disable the IME on the PID edit controls
for (i = 0; i < 5;i++)
{
ImmAssociateContext(GetDlgItem(hdlg, IDT_EDIT_PID1+i), (HIMC)NULL);
}
//
// subclass the edit controls and limit the number of characters
//
for (i = 0; i < 5;i++) {
SendDlgItemMessage(hdlg,IDT_EDIT_PID1+i,EM_LIMITTEXT,MAX_PID30_EDIT,0);
OldPidEditProc[i] = (WNDPROC)GetWindowLongPtr(GetDlgItem(hdlg, IDT_EDIT_PID1+i),GWLP_WNDPROC);
SetWindowLongPtr(GetDlgItem(hdlg, IDT_EDIT_PID1+i),GWLP_WNDPROC,(LONG_PTR)PidEditSubProc);
}
//
// set focus to first pid entry
//
SetFocus(GetDlgItem(hdlg,IDT_EDIT_PID1));
b = FALSE;
break;
case WMX_ACTIVATEPAGE:
CHECKUPGRADEONLY();
if (BuildCmdcons) {
return(FALSE);
}
// If we have an ecrypted PID and don't have the right crypto installed
// defer the PID validation until GUI mode
if (g_bDeferPIDValidation)
{
return FALSE;
}
b = TRUE;
if(wParam) {
//
// activation
//
#ifdef PRERELEASE
if (NoPid) {
//
// don't show the page in this case
//
b = FALSE;
break;
}
#endif
if (SourceInstallType != SelectInstall) {
//
// don't show the page in this case
//
b = FALSE;
break;
}
//
// don't activate the page in restart mode
//
if (Winnt32RestartedWithAF ()) {
if (GetPrivateProfileString (
WINNT_USERDATA,
WINNT_US_PRODUCTKEY,
TEXT(""),
ProductId,
sizeof (ProductId) / sizeof (ProductId[0]),
g_DynUpdtStatus->RestartAnswerFile
)) {
return FALSE;
}
}
if (UnattendedOperation) {
//
// make sure the pid is specified in the unattend file else we should stop
//
ShowPidBox(hdlg, SW_HIDE);
if (SetPid30(hdlg,SourceInstallType, (LPTSTR)&ProductId) ) {
UNATTENDED(PSBTN_NEXT);
} else {
//
// a hack so that the correct wizard page is active when we put up our message box
//
bUnattendPid = TRUE;
ShowPidBox(hdlg, SW_SHOW);
PostMessage(hdlg,WMX_I_AM_VISIBLE,0,0);
}
}
} else {
//
// deactivation. don't verify anything if they are backing up
//
if (!Cancelled && lParam != PSN_WIZBACK) {
TCHAR tmpBuffer1[6];
TCHAR tmpBuffer2[6];
TCHAR tmpBuffer3[6];
TCHAR tmpBuffer4[6];
TCHAR tmpBuffer5[6];
GetDlgItemText(hdlg,IDT_EDIT_PID1,tmpBuffer1,sizeof(tmpBuffer1)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID2,tmpBuffer2,sizeof(tmpBuffer2)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID3,tmpBuffer3,sizeof(tmpBuffer3)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID4,tmpBuffer4,sizeof(tmpBuffer4)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID5,tmpBuffer5,sizeof(tmpBuffer5)/sizeof(TCHAR));
b = ValidatePid30( tmpBuffer1,
tmpBuffer2,
tmpBuffer3,
tmpBuffer4,
tmpBuffer5
);
if (!b) {
if (UnattendedOperation) {
// We should not fail ValidatePid30 if we succeeded
// in the SetPid30 above. If we failed in SetPid30 above
// we should already be showing the pid boxes.
ShowPidBox(hdlg, SW_SHOW);
}
MessageBoxFromMessage(hdlg,
bUnattendPid ? MSG_UNATTEND_OEM_PID_IS_INVALID : MSG_OEM_PID_IS_INVALID,
FALSE,AppTitleStringId,MB_OK|MB_ICONSTOP);
SetFocus(GetDlgItem(hdlg,IDT_EDIT_PID1));
b = FALSE;
} else {
//
// user entered a valid PID, save it for later on.
//
wsprintf( ProductId,
TEXT("%s-%s-%s-%s-%s"),
tmpBuffer1,
tmpBuffer2,
tmpBuffer3,
tmpBuffer4,
tmpBuffer5
);
}
}
}
break;
case WMX_I_AM_VISIBLE:
//
// Force repainting first to make sure the page is visible.
//
InvalidateRect(hdlg,NULL,FALSE);
UpdateWindow(hdlg);
if (bUnattendPid) {
MessageBoxFromMessage(hdlg,MSG_UNATTEND_OEM_PID_IS_INVALID,FALSE,AppTitleStringId,MB_OK|MB_ICONSTOP);
bUnattendPid = FALSE;
}
b = TRUE;
break;
default:
b = FALSE;
break;
}
return(b);
}
BOOL
OemPid30WizPage(
IN HWND hdlg,
IN UINT msg,
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
OEM Pid wizard page. depends on SourceInstallType being set correctly.
Arguments:
Standard window proc arguments.
Returns:
Message-dependent value.
--*/
{
BOOL b;
PPAGE_RUNTIME_DATA WizPage = (PPAGE_RUNTIME_DATA)GetWindowLongPtr(hdlg,DWLP_USER);
static BOOL bUnattendPid = FALSE;
DWORD i;
// HFONT hFont;
switch(msg) {
case WM_INITDIALOG:
GetPID();
// hFont = GetStockObject(SYSTEM_FIXED_FONT);
// Disable the IME on the PID edit controls
for (i = 0; i < 5;i++)
{
ImmAssociateContext(GetDlgItem(hdlg, IDT_EDIT_PID1+i), (HIMC)NULL);
}
//
// subclass the edit controls and limit the number of characters
//
for (i = 0; i < 5;i++) {
SendDlgItemMessage(hdlg,IDT_EDIT_PID1+i,EM_LIMITTEXT,MAX_PID30_EDIT,0);
// SendDlgItemMessage(hdlg, IDT_EDIT_PID1+i, WM_SETFONT, hFont, TRUE );
OldPidEditProc[i] = (WNDPROC)GetWindowLongPtr(GetDlgItem(hdlg, IDT_EDIT_PID1+i),GWLP_WNDPROC);
SetWindowLongPtr(GetDlgItem(hdlg, IDT_EDIT_PID1+i),GWLP_WNDPROC,(LONG_PTR)PidEditSubProc);
}
//
// set focus to first pid entry
//
SetFocus(GetDlgItem(hdlg,IDT_EDIT_PID1));
b = FALSE;
break;
case WM_COMMAND:
b = FALSE;
break;
case WMX_ACTIVATEPAGE:
CHECKUPGRADEONLY();
if (BuildCmdcons) {
return(FALSE);
}
// If we have an ecrypted PID and don't have the right crypto installed
// defer the PID validation until GUI mode
if (g_bDeferPIDValidation)
{
return FALSE;
}
b = TRUE;
if(wParam) {
//
// activation
//
#ifdef PRERELEASE
if (NoPid) {
//
// don't show the page in this case
//
b = FALSE;
break;
}
#endif
if (SourceInstallType != OEMInstall) {
//
// don't show the page in this case
//
b = FALSE;
break;
}
//
// don't activate the page in restart mode
//
if (Winnt32RestartedWithAF ()) {
if (GetPrivateProfileString (
WINNT_USERDATA,
WINNT_US_PRODUCTKEY,
TEXT(""),
ProductId,
sizeof (ProductId) / sizeof (ProductId[0]),
g_DynUpdtStatus->RestartAnswerFile
)) {
return FALSE;
}
}
if (UnattendedOperation) {
//
// make sure the pid is specified in the unattend file else we should stop
//
if (SetPid30(hdlg,SourceInstallType, (LPTSTR)&ProductId) ) {
UNATTENDED(PSBTN_NEXT);
} else {
//
// a hack so that the correct wizard page is active when we put up our message box
//
bUnattendPid = TRUE;
PostMessage(hdlg,WMX_I_AM_VISIBLE,0,0);
}
}
#if 0
if (!Upgrade || (SourceInstallType != OEMInstall)) {
//
// don't show the page in this case
//
b = FALSE;
break;
} else {
NOTHING;
}
#endif
} else {
//
// deactivation. don't verify anything if they are backing up
//
if (!Cancelled && lParam != PSN_WIZBACK) {
TCHAR tmpBuffer1[6];
TCHAR tmpBuffer2[6];
TCHAR tmpBuffer3[6];
TCHAR tmpBuffer4[6];
TCHAR tmpBuffer5[6];
GetDlgItemText(hdlg,IDT_EDIT_PID1,tmpBuffer1,sizeof(tmpBuffer1)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID2,tmpBuffer2,sizeof(tmpBuffer2)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID3,tmpBuffer3,sizeof(tmpBuffer3)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID4,tmpBuffer4,sizeof(tmpBuffer4)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID5,tmpBuffer5,sizeof(tmpBuffer5)/sizeof(TCHAR));
b = ValidatePid30( tmpBuffer1,
tmpBuffer2,
tmpBuffer3,
tmpBuffer4,
tmpBuffer5
);
if (!b) {
MessageBoxFromMessage(hdlg,
bUnattendPid ? MSG_UNATTEND_OEM_PID_IS_INVALID : MSG_OEM_PID_IS_INVALID,
FALSE,AppTitleStringId,MB_OK|MB_ICONSTOP);
SetFocus(GetDlgItem(hdlg,IDT_EDIT_PID1));
b = FALSE;
} else {
//
// user entered a valid PID, save it for later on.
//
wsprintf( ProductId,
TEXT("%s-%s-%s-%s-%s"),
tmpBuffer1,
tmpBuffer2,
tmpBuffer3,
tmpBuffer4,
tmpBuffer5
);
}
}
}
break;
case WMX_I_AM_VISIBLE:
//
// Force repainting first to make sure the page is visible.
//
InvalidateRect(hdlg,NULL,FALSE);
UpdateWindow(hdlg);
if (bUnattendPid) {
MessageBoxFromMessage(hdlg,MSG_UNATTEND_OEM_PID_IS_INVALID,FALSE,AppTitleStringId,MB_OK|MB_ICONSTOP);
bUnattendPid = FALSE;
}
b = TRUE;
break;
default:
b = FALSE;
break;
}
return(b);
}
BOOL
CdPid30WizPage(
IN HWND hdlg,
IN UINT msg,
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
CD Retail pid wizard page. depends on SourceInstallType being set correctly.
Arguments:
Standard window proc arguments.
Returns:
Message-dependent value.
--*/
{
BOOL b;
PPAGE_RUNTIME_DATA WizPage = (PPAGE_RUNTIME_DATA)GetWindowLongPtr(hdlg,DWLP_USER);
static BOOL bUnattendPid = FALSE;
DWORD i;
// HFONT hFont;
switch(msg) {
case WM_INITDIALOG:
GetPID();
// hFont = GetStockObject(SYSTEM_FIXED_FONT);
// Disable the IME on the PID edit controls
for (i = 0; i < 5;i++)
{
ImmAssociateContext(GetDlgItem(hdlg, IDT_EDIT_PID1+i), (HIMC)NULL);
}
//
// subclass the edit controls and limit the number of characters
//
for (i = 0; i < 5;i++) {
SendDlgItemMessage(hdlg,IDT_EDIT_PID1+i,EM_LIMITTEXT,MAX_PID30_EDIT,0);
// SendDlgItemMessage(hdlg, IDT_EDIT_PID1+i, WM_SETFONT, hFont, TRUE );
OldPidEditProc[i] = (WNDPROC)GetWindowLongPtr(GetDlgItem(hdlg, IDT_EDIT_PID1+i),GWLP_WNDPROC);
SetWindowLongPtr(GetDlgItem(hdlg, IDT_EDIT_PID1+i),GWLP_WNDPROC,(LONG_PTR)PidEditSubProc);
}
//
// set focus to first pid entry
//
SetFocus(GetDlgItem(hdlg,IDT_EDIT_PID1));
b = FALSE;
break;
case WM_COMMAND:
//
// nothing to do here
//
b = FALSE;
break;
case WMX_ACTIVATEPAGE:
CHECKUPGRADEONLY();
if (BuildCmdcons) {
return(FALSE);
}
// If we have an ecrypted PID and don't have the right crypto installed
// defer the PID validation until GUI mode
if (g_bDeferPIDValidation)
{
return FALSE;
}
b = TRUE;
if(wParam) {
//
// activation
//
#ifdef PRERELEASE
if (NoPid) {
//
// don't show the page in this case
//
b = FALSE;
break;
}
#endif
if (SourceInstallType != RetailInstall) {
//
// don't show the page in this case
//
b = FALSE;
break;
}
//
// don't activate the page in restart mode
//
if (Winnt32RestartedWithAF ()) {
if (GetPrivateProfileString (
WINNT_USERDATA,
WINNT_US_PRODUCTKEY,
TEXT(""),
ProductId,
sizeof (ProductId) / sizeof (ProductId[0]),
g_DynUpdtStatus->RestartAnswerFile
)) {
return FALSE;
}
}
if (UnattendedOperation) {
//
// make sure the pid is specified in the unattend file else we should stop
//
if (SetPid30(hdlg,SourceInstallType, (LPTSTR)&ProductId)) {
UNATTENDED(PSBTN_NEXT);
} else {
//
// a hack so that the correct wizard page is active when we put up our message box
//
bUnattendPid = TRUE;
PostMessage(hdlg,WMX_I_AM_VISIBLE,0,0);
}
}
} else {
//
// deactivation. don't verify anything if they are backing up
//
if ( !Cancelled && lParam != PSN_WIZBACK ) {
TCHAR tmpBuffer1[6];
TCHAR tmpBuffer2[6];
TCHAR tmpBuffer3[6];
TCHAR tmpBuffer4[6];
TCHAR tmpBuffer5[6];
GetDlgItemText(hdlg,IDT_EDIT_PID1,tmpBuffer1,sizeof(tmpBuffer1)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID2,tmpBuffer2,sizeof(tmpBuffer2)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID3,tmpBuffer3,sizeof(tmpBuffer3)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID4,tmpBuffer4,sizeof(tmpBuffer4)/sizeof(TCHAR));
GetDlgItemText(hdlg,IDT_EDIT_PID5,tmpBuffer5,sizeof(tmpBuffer5)/sizeof(TCHAR));
b = ValidatePid30( tmpBuffer1,
tmpBuffer2,
tmpBuffer3,
tmpBuffer4,
tmpBuffer5
);
if (!b) {
if (PidMatchesMedia){
MessageBoxFromMessage(hdlg,
bUnattendPid ? MSG_UNATTEND_CD_PID_IS_INVALID :MSG_CD_PID_IS_INVALID,
FALSE,AppTitleStringId,MB_OK|MB_ICONSTOP);
} else {
MessageBoxFromMessage(hdlg,
UpgradeOnly ? MSG_CCP_MEDIA_FPP_PID : MSG_FPP_MEDIA_CCP_PID,
FALSE,AppTitleStringId,MB_OK|MB_ICONSTOP);
}
SetFocus(GetDlgItem(hdlg,IDT_EDIT_PID1));
b = FALSE;
} else {
//
// user entered a valid PID, save it for later on.
//
wsprintf( ProductId,
TEXT("%s-%s-%s-%s-%s"),
tmpBuffer1,
tmpBuffer2,
tmpBuffer3,
tmpBuffer4,
tmpBuffer5
);
}
}
}
break;
case WMX_I_AM_VISIBLE:
//
// Force repainting first to make sure the page is visible.
//
InvalidateRect(hdlg,NULL,FALSE);
UpdateWindow(hdlg);
if (bUnattendPid) {
MessageBoxFromMessage(hdlg,MSG_UNATTEND_CD_PID_IS_INVALID,FALSE,AppTitleStringId,MB_OK|MB_ICONSTOP);
bUnattendPid = FALSE;
}
b = TRUE;
break;
default:
b = FALSE;
break;
}
return(b);
}
LPTSTR pSelectGpID[] = { TEXT("23"),
TEXT("26"),
NULL };
LPTSTR pSelectChID[] = { TEXT("080"),
TEXT("041"),
NULL };
BOOL IsSelect(LPTSTR Pid20)
{
LPTSTR pTmp;
BOOL bSelect = FALSE;
#if defined(UNICODE)
pTmp = wcsrchr(Pid20, L'-');
#else
pTmp = strrchr(Pid20, '-');
#endif
if (pTmp)
{
int i = 0;
TCHAR GpID[3];
pTmp++;
lstrcpyn(GpID, pTmp, 3);
while (!bSelect && pSelectGpID[i])
{
bSelect = (lstrcmp(GpID, pSelectGpID[i]) == 0);
i++;
}
}
if (!bSelect)
{
// If we did not determine that we are select. This could be beta VL key
#if defined(UNICODE)
pTmp = wcschr(Pid20, L'-');
#else
pTmp = strchr(Pid20, '-');
#endif
if (pTmp)
{
int i = 0;
// The channel ID is 3 character + termination
TCHAR ChID[4];
pTmp++;
lstrcpyn(ChID, pTmp, 4); // Copy 4 characters, includingt he NULL termination.
while (!bSelect && pSelectChID[i])
{
bSelect = (lstrcmp(ChID, pSelectChID[i]) == 0);
i++;
}
}
}
return bSelect;
}
BOOL
ValidatePidEx(LPTSTR PID, BOOL *pbStepup, BOOL *bSelect)
{
TCHAR Pid20Id[MAX_PATH];
BYTE Pid30[1024]={0};
TCHAR pszSkuCode[10];
BOOL fStepUp;
// it seems that sku code really doesn't matter in winnt32, only syssetup
lstrcpy(pszSkuCode,TEXT("1797XYZZY"));
#if 0
wsprintf(DebugBuffer,
TEXT("cd-key: %s\nRPC: %s\nOEM Key: %d"),
PID,
Pid30Rpc,
(SourceInstallType == OEMInstall)
);
OutputDebugString(DebugBuffer);
#endif
*(LPDWORD)Pid30 = sizeof(Pid30);
if (!SetupPIDGen(
PID, // [IN] 25-character Secure CD-Key (gets U-Cased)
Pid30Rpc, // [IN] 5-character Release Product Code
// note sku code is not kept around in winnt32, only syssetup.
pszSkuCode, // [IN] Stock Keeping Unit (formatted like 123-12345)
(SourceInstallType == OEMInstall), // [IN] is this an OEM install?
Pid20Id, // [OUT] PID 2.0, pass in ptr to 24 character array
Pid30, // [OUT] pointer to binary PID3 buffer. First DWORD is the length
pbStepup // [OUT] optional ptr to Compliance Checking flag (can be NULL)
)) {
if (g_EncryptedPID)
{
GlobalFree(g_EncryptedPID);
g_EncryptedPID = NULL;
}
return(FALSE);
}
// PID is valid,
// if the caller wants, return if this is a Volume License PID
if(bSelect)
{
*bSelect = IsSelect(Pid20Id);
}
return TRUE;
}
BOOL
ValidatePid30(
LPTSTR Edit1,
LPTSTR Edit2,
LPTSTR Edit3,
LPTSTR Edit4,
LPTSTR Edit5
)
{
TCHAR tmpProductId[MAX_PATH]={0};
TCHAR Pid20Id[MAX_PATH];
BYTE Pid30[1024]={0};
TCHAR pszSkuCode[10];
BOOL fStepUp;
//TCHAR DebugBuffer[1024];
// until we know better, assume the Pid matches the media type
PidMatchesMedia = TRUE;
if (!Edit1 || !Edit2 || !Edit3 || !Edit4 || !Edit5) {
return(FALSE);
}
// Since we now need a PID in the select case too, fill in the string.
/*
if (SourceInstallType == SelectInstall){
tmpProductId[0] = TEXT('\0');
}
else
*/
{
wsprintf( tmpProductId,
TEXT("%s-%s-%s-%s-%s"),
Edit1,
Edit2,
Edit3,
Edit4,
Edit5 );
}
if (!ValidatePidEx(tmpProductId, &fStepUp, NULL))
{
return(FALSE);
}
if (SourceInstallType != OEMInstall){
// we want OEM FPP and CCP keys to be accepted by either media. It seems like
// there will be OEM CCP media, but only FPP keys, which is why we aren't
// checking to make sure they match, as it's broken by design.
if (UpgradeOnly != fStepUp){
// user is trying to do a clean install with upgrade only media. Bad user, bad.
PidMatchesMedia = FALSE;
return FALSE;
}
}
return(TRUE);
}
void GetPID()
{
if (!ProductId[0] && UnattendedOperation && !g_bDeferPIDValidation){
//
// On Whistler upgrades, reuse existing DPID
//
BYTE abCdKey[16];
BOOL bDontCare, bSelect;
if (Upgrade &&
ISNT() &&
OsVersion.dwMajorVersion == 5 && OsVersion.dwMinorVersion == 1 &&
pGetCdKey (abCdKey)
) {
EncodePid3g (ProductId, abCdKey);
if (ValidatePidEx(ProductId, &bDontCare, &bSelect) && bSelect)
{
HRESULT hr;
if (g_EncryptedPID)
{
GlobalFree(g_EncryptedPID);
g_EncryptedPID = NULL;
}
// Prepare the encrypted PID so that we can write it to winnt.sif
hr = PrepareEncryptedPID(ProductId, 1, &g_EncryptedPID);
if (hr != S_OK)
{
DebugLog (Winnt32LogInformation, TEXT("PrepareEncryptedPID failed: <hr=0x%1!lX!>"), 0, hr);
}
}
}
}
}
VOID ShowPidBox(
IN HWND hdlg,
IN int nCmdShow
)
/*++
Routine Description:
shows or hide the pid boxes in the wizard page
Arguments:
hdlg - window handle to pid dialog
nCmdShow - SW_SHOW or SW_HIDE
--*/
{
int i;
for (i = 0; i<5; i++) {
ShowWindow(GetDlgItem(hdlg,IDT_EDIT_PID1+i), nCmdShow);
}
}