windows-nt/Source/XPSP1/NT/sdktools/debuggers/setup/msisetup.c
2020-09-26 16:20:57 +08:00

752 lines
22 KiB
C

//depot/Lab01_N/sdktools/cabs/symbolcd/tools/setup/msisetup.c#16 - edit change 2216 (text)
#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include "tchar.h"
#include "shlwapi.h"
#define MSI_BUILD_VER_ALPHA 816 // MSI version for Win2K Alpha WIN2KMINBUILD
#define MSI_BUILD_VER_X86 1029 // Latest MSI version for Win2K x86
#define WIN2K_MIN_BUILD_X86 2183 // Win2K RC3
#define WIN2K_MIN_BUILD_ALPHA 2128 // Last supported Win2K Alpha build
typedef struct _CommandArgs {
BOOL QuietInstall;
BOOL StressInstall;
BOOL UIStressInstall;
TCHAR szInstDir[ _MAX_PATH*sizeof(TCHAR) ];
TCHAR szMsiName[ _MAX_PATH*sizeof(TCHAR) ];
TCHAR szProductRegKey[ _MAX_PATH*sizeof(TCHAR) ];
} COMMAND_ARGS, *PCOMMAND_ARGS;
// Function prototypes
BOOL
RunCommand(
PTCHAR szCommandLine,
HINSTANCE hInst
);
BOOL
GetCommandLineArgs(
LPTSTR szCmdLine,
PCOMMAND_ARGS pComArgs
);
TCHAR szMSIInstFile[_MAX_PATH*sizeof(TCHAR)];
TCHAR szPkgInstFile[_MAX_PATH*sizeof(TCHAR)];
TCHAR szPkgInstCommand[_MAX_PATH*2*sizeof(TCHAR)];
// For stress installs, this command will be used to
// remove the current package but don't remove its
// files, if the current package with the same
// product ID is already installed.
TCHAR szPkgRemoveCommand[_MAX_PATH*2*sizeof(TCHAR)];
TCHAR szPkgRemoveCommand2[_MAX_PATH*2*sizeof(TCHAR)];
// If the first install fails, stress tries again without
// the quiet switch before giving a pop-up
TCHAR szPkgInstCommandNoQuiet[_MAX_PATH*2*sizeof(TCHAR)];
TCHAR szCommandFullPath[_MAX_PATH*sizeof(TCHAR)];
int WINAPI WinMain(
HINSTANCE hInstance,
HINSTANCE hPrevInstance,
LPTSTR lpszCmdLine,
int nCmdShow
)
{
OSVERSIONINFO VersionInfo;
SYSTEM_INFO SystemInfo;
BOOL rc;
BOOL MSIIsInstalled;
PTCHAR ch;
TCHAR szBuf[1000];
TCHAR szSystemDirectory[_MAX_PATH];
COMMAND_ARGS ComArgs;
HKEY hKey;
DWORD dwrc;
DWORD dwSizeValue;
DWORD dwType;
HANDLE hFile;
WIN32_FIND_DATA FindFileData;
MSIIsInstalled=FALSE;
// Get this info for later use
VersionInfo.dwOSVersionInfoSize = sizeof( OSVERSIONINFO );
GetVersionEx( &VersionInfo );
GetSystemInfo( &SystemInfo );
// Parse through the command line for the various arguments
rc = GetCommandLineArgs(lpszCmdLine, &ComArgs );
if (!rc) {
_stprintf( szBuf, _T("%s%s%s%s%s"),
_T(" Usage: \n\n"),
_T(" setup.exe [ /q [ /i <InstDir> ] ]\n\n"),
_T(" /q\tGive pop-ups only for errors\n\n"),
_T(" /i\tInstall to <Instdir>\n\n"),
_T(" /n\tInstall <msi package Name>\n\n")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"), 0 );
return (1);
}
//
// Set the full path to this setup.exe
//
if (GetModuleFileName( NULL, szCommandFullPath, MAX_PATH ) == 0) {
return(1);
}
// Put an end of string after the directory that this was
// started from
ch = szCommandFullPath + _tcslen(szCommandFullPath);
while ( *ch != _T('\\') && ( ch > szCommandFullPath ) ) ch--;
*ch=_T('\0');
// This will become the full path and name of the MSI file to install
_tcscpy( szMSIInstFile, szCommandFullPath);
// Set the full path and name of the msi package
_tcscpy( szPkgInstFile, szCommandFullPath);
_tcscat( szPkgInstFile, _T("\\") );
_tcscat( szPkgInstFile, ComArgs.szMsiName );
// See if the package exists
hFile = FindFirstFile( szPkgInstFile, &FindFileData );
if ( hFile == INVALID_HANDLE_VALUE ) {
_stprintf( szBuf, _T("%s%s%s%s"),
_T(" The Microsoft Debugging Tools package "),
szPkgInstFile,
_T(" does not exist.\n Setup cannot contine"),
_T(" for this platform.")
);
MessageBox( NULL,
szBuf,
_T("Microsoft Debugging Tools"),
0
);
return(1);
}
FindClose(hFile);
// Set the command for installing the package
_tcscpy( szPkgInstCommand, _T("msiexec /i ") );
_tcscat( szPkgInstCommand, szPkgInstFile );
// Set the command for removing the current package
// that is installed.
_tcscpy( szBuf, _T("") );
dwrc = RegOpenKeyEx( HKEY_CURRENT_USER,
ComArgs.szProductRegKey,
0,
KEY_QUERY_VALUE,
&hKey
);
if ( dwrc == ERROR_SUCCESS ) {
_tcscpy( szBuf, _T("") );
dwSizeValue=sizeof(szBuf);
RegQueryValueEx ( hKey,
_T("ProductCode"),
0,
&dwType,
(PBYTE)szBuf,
&dwSizeValue
);
RegCloseKey(hKey);
}
// Set the command to remove the current package
// that has an Add/Remove link in the start menu
_tcscpy(szPkgRemoveCommand2, _T("") );
if ( _tcslen(szBuf) > 0 ) {
_tcscpy(szPkgRemoveCommand2, _T("msiexec /x ") );
_tcscat(szPkgRemoveCommand2, szBuf);
_tcscat(szPkgRemoveCommand2, _T(" REMOVETHEFILES=0 /qn") );
}
// Set the command to remove the current package so that
// this program works like it used to.
_tcscpy(szPkgRemoveCommand, _T("msiexec /x ") );
_tcscat(szPkgRemoveCommand, szPkgInstFile );
_tcscat(szPkgRemoveCommand, _T(" REMOVETHEFILES=0 /qn") );
// Add a user override installation directory
if ( _tcslen(ComArgs.szInstDir) > 0 ) {
_tcscat( szPkgInstCommand, _T(" INSTDIR=") );
_tcscat( szPkgInstCommand, ComArgs.szInstDir );
} else if ( ComArgs.UIStressInstall ) {
GetSystemDirectory( szSystemDirectory, _MAX_PATH );
_tcscat( szPkgInstCommand, _T(" INSTDIR=") );
_tcscat( szPkgInstCommand, szSystemDirectory );
}
// If this is an "undocumented" stress install
// don't remove the files of the previous install
// when you upgrade
// FEATURESTOREMOVE should never actually need to be used, unless
// the user has something screwed up on his system where the registry
// key and products installed don't agree, or MSI thinks there's more
// products installed than the registry key we look at.
if ( ComArgs.StressInstall ) {
_tcscat( szPkgInstCommand, _T(" FEATURESTOREMOVE=\"\"") );
}
// If this is an "undocumented" UI stress install
// only install the private extensions
if ( ComArgs.UIStressInstall ) {
_tcscat( szPkgInstCommand,
_T(" ADDLOCAL=DBG.DbgExts.Internal,DBG.NtsdFix.Internal") );
}
// Add the quiet switch
// Save the command without a quiet switch
_tcscpy( szPkgInstCommandNoQuiet, szPkgInstCommand);
if ( ComArgs.QuietInstall ) {
_tcscat( szPkgInstCommand, _T(" /qn") );
}
// Do version checks for whether msi is already installed
//
// If this is Windows 2000 and build number is >=
// WIN2K_MIN_BUILD_X86 then MSI is installed
// Don't try to run instmsi.exe on Windows 2000 because
// you will get file system protection pop-ups.
//
if ( (VersionInfo.dwPlatformId == VER_PLATFORM_WIN32_NT) &&
(VersionInfo.dwMajorVersion >= 5.0 ) ) {
switch (SystemInfo.wProcessorArchitecture) {
case PROCESSOR_ARCHITECTURE_ALPHA:
if (VersionInfo.dwBuildNumber < WIN2K_MIN_BUILD_ALPHA) {
// The version of MSI that is on early builds of Windows
// 2000 shouldn't be trusted for installs.
_stprintf( szBuf, "%s",
_T("The Debugging Tools does not install on ")
_T("this version of Alpha Windows 2000. Please upgrade ")
_T("your system to Windows 2000 Beta 3 ")
_T("before trying to install this package.")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"),0);
return(1);
}
break;
case PROCESSOR_ARCHITECTURE_INTEL:
if (VersionInfo.dwBuildNumber < WIN2K_MIN_BUILD_X86 ) {
// The version of MSI that is on early builds of Windows
// 2000 shouldn't be trusted for installs.
_stprintf( szBuf, "%s%s%s%s",
_T("The Debugging Tools does not install on "),
_T("this version of Windows 2000. Please upgrade "),
_T("your system to a retail version of Windows 2000 "),
_T("before trying to install this package.")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"),0);
return(1);
}
break;
case PROCESSOR_ARCHITECTURE_IA64:
break;
default:
_stprintf( szBuf, "%s",
_T("Unknown computer architecture.")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"),0);
return(1);
}
MSIIsInstalled = TRUE;
} else if ( SystemInfo.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_INTEL) {
//
// For Intel OS's prior to Windows 2000, run instmsi.exe
//
//
// NT4 X86
//
if ( VersionInfo.dwPlatformId == VER_PLATFORM_WIN32_NT ) {
_tcscat( szMSIInstFile,
_T("\\setup\\winnt\\i386\\instmsi.exe /q") );
}
//
// Win9x
//
else if ( VersionInfo.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS ) {
_tcscat( szMSIInstFile,
_T("\\setup\\win9x\\instmsi.exe /q") );
} else {
_stprintf( szBuf, _T("%s %s"),
_T("The Microsoft Debugging Tools does not install"),
_T("on this system.")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"),0);
return(1);
}
} else if ( SystemInfo.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_ALPHA ) {
if ( VersionInfo.dwBuildNumber >= WIN2K_MIN_BUILD_ALPHA ) {
MSIIsInstalled = TRUE;
} else {
_tcscat( szMSIInstFile,
_T("\\setup\\winnt\\alpha\\instmsi.exe /q"));
}
} else {
MessageBox( NULL,
_T("The Microsoft Debugging Tools cannot be installed on this system."),
_T("Microsoft Debugging Tools"),
0 );
return(1);
}
// Install MSI if it is not already installed
if ( !MSIIsInstalled ) {
if ( RunCommand( szMSIInstFile, hInstance ) ) {
MSIIsInstalled = TRUE;
}
if (!MSIIsInstalled) {
_stprintf( szBuf, _T("%s %s %s %s"),
_T("The Windows Installer could not be installed"),
_T("on this system. This is required before"),
_T("installing the Microsoft Debugging Tools package."),
_T("Try logging in as an administrator and try again.")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"),0);
return(1);
}
}
//
// Now, if this is a stress install,
// Try to remove the current package in case it is installed
//
if ( ComArgs.StressInstall ) {
if ( _tcslen(szPkgRemoveCommand2) > 0 ) {
RunCommand( szPkgRemoveCommand2, hInstance);
}
RunCommand( szPkgRemoveCommand, hInstance );
if ( !RunCommand( szPkgInstCommand, hInstance ) ) {
// Try again without the quiet switch, so that the user will get
// a pop-up from dbg.msi and quit calling us
_stprintf( szBuf, _T("%s %s %s %s"),
_T("There were errors when trying to install the"),
_T("debuggers.\nClick OK to attempt an install of the"),
_T("debuggers with\n the GUI and you will see the"),
_T("correct error message.")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"), 0);
if ( !RunCommand( szPkgInstCommandNoQuiet, hInstance ) ) {
_stprintf( szBuf, _T("%s %s %s"),
_T("There were still errors in the install.\n"),
_T("Please see http://dbg/triage/top10.html #2 "),
_T("for more help.")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"), 0);
return(1);
}
}
return(0);
}
//
// Now, install the package dbg.msi
//
if ( !RunCommand( szPkgInstCommand, hInstance ) ) {
if (ComArgs.QuietInstall) {
_stprintf( szBuf, _T("%s %s %s %s"),
_T("There were errors in the Debugging Tools install."),
_T(" Please run "),
szPkgInstFile,
_T("to receive more detailed error information.")
);
MessageBox( NULL, szBuf, _T("Microsoft Debugging Tools"),0);
}
return(1);
}
return(0);
}
//
// RunCommand
//
// Purpose: Install MSI
//
// Return Values:
// 0 error
// 1 successful
BOOL
RunCommand( PTCHAR szCommandLine,
HINSTANCE hInst)
{
BOOL rc;
DWORD dwRet;
PROCESS_INFORMATION ProcInfo = {0};
STARTUPINFO SI= {0};
// Spawn the command line specified by szCommandLine
rc = CreateProcess(NULL,
szCommandLine,
NULL,
NULL,
FALSE,
CREATE_DEFAULT_ERROR_MODE | NORMAL_PRIORITY_CLASS,
NULL,
NULL,
&SI,
&ProcInfo );
if ( (!rc) || (!ProcInfo.hProcess) ) {
goto cleanup;
}
//
// Wait for command to complete ... Give it 20 minutes
//
dwRet = WaitForSingleObject(ProcInfo.hProcess, 1200000);
if (dwRet != WAIT_OBJECT_0) {
rc = FALSE;
goto cleanup;
}
// Get the process exit code
rc = GetExitCodeProcess( ProcInfo.hProcess, &dwRet);
if (dwRet == ERROR_SUCCESS ) {
rc = 1;
} else {
rc = 0;
}
cleanup:
if (ProcInfo.hProcess)
CloseHandle(ProcInfo.hProcess);
return (rc);
}
BOOL
GetCommandLineArgs(
LPTSTR szCmdLine,
PCOMMAND_ARGS pComArgs
)
{
ULONG length;
ULONG i,cur;
BOOL SkippingSpaces=FALSE;
BOOL QuotedString=FALSE;
BOOL NeedSecond=FALSE;
BOOL rc=TRUE;
LPTSTR *argv;
ULONG argc=0;
LPTSTR szCmdLineTmp;
TCHAR c;
ZeroMemory(pComArgs, sizeof(COMMAND_ARGS));
// Create a line to use for temporary marking
length=_tcslen(szCmdLine);
szCmdLineTmp= (LPTSTR)malloc( (_tcslen(szCmdLine) + 1) * sizeof(TCHAR) );
if (szCmdLineTmp==NULL)
{
return FALSE;
}
_tcscpy(szCmdLineTmp, szCmdLine);
// Count the number of arguments
// Create a argv and argc
SkippingSpaces=TRUE;
QuotedString=FALSE;
argc=0;
for ( i=0; i<length; i++ )
{
c=szCmdLineTmp[i];
switch (szCmdLineTmp[i]) {
case _T(' '):
case _T('\t'): if (QuotedString)
{
break;
}
if (!SkippingSpaces)
{
SkippingSpaces=TRUE;
}
break;
case _T('\"'): if (QuotedString)
{
// This is the end of a quoted string
// The next character to read in is a space
QuotedString=FALSE;
SkippingSpaces=TRUE;
if ( i < (length-1) &&
szCmdLineTmp[i+1] != _T(' ') &&
szCmdLineTmp[i+1] != _T('\t') )
{
// This is the end of a quote and its not
// followed by a space
rc=FALSE;
goto CommandLineFinish;
}
break;
}
if (SkippingSpaces) {
// This is the beginning of a quoted string
// Its a new argument and it follows spaces
argc++;
SkippingSpaces=FALSE;
QuotedString=TRUE;
break;
}
// This is an error -- This is a quote in the middle of a string
rc=FALSE;
goto CommandLineFinish;
break;
default: if (QuotedString) {
break;
}
if (SkippingSpaces) {
argc++;
SkippingSpaces=FALSE;
}
break;
}
}
if (QuotedString)
{
// Make sure that all the quotes got a finished pair
rc=FALSE;
goto CommandLineFinish;
}
// Now, create argv with the correct number of entries
argv=(LPTSTR*)malloc(argc * sizeof(LPTSTR) );
if (argv==NULL)
{
free(szCmdLineTmp);
return FALSE;
}
// Set argv to point to the correct place on szCmdLineTmp
// and put '\0' after each token.
SkippingSpaces=TRUE;
QuotedString=FALSE;
argc=0;
for ( i=0; i<length; i++ )
{
c=szCmdLineTmp[i];
switch (szCmdLineTmp[i]) {
case _T(' '):
case _T('\t'): if (QuotedString)
{
break;
}
if (!SkippingSpaces)
{
szCmdLineTmp[i]='\0';
SkippingSpaces=TRUE;
}
break;
case _T('\"'): if (QuotedString)
{
// This is the end of a quoted string
// The next character to read in is a space
QuotedString=FALSE;
SkippingSpaces=TRUE;
szCmdLineTmp[i+1]=_T('\0');
break;
}
if (SkippingSpaces) {
// This is the beginning of a quoted string
// Its a new argument and it follows spaces
argv[argc]=szCmdLineTmp+i;
argc++;
SkippingSpaces=FALSE;
QuotedString=TRUE;
break;
}
// This is an error -- This is a quote in the middle of a string
rc=FALSE;
goto CommandLineFinish;
break;
default: if (QuotedString)
{
break;
}
if (SkippingSpaces) {
argv[argc]=szCmdLineTmp+i;
argc++;
SkippingSpaces=FALSE;
}
break;
}
}
// Now, parse the arguments
NeedSecond=FALSE;
for (i=0; i<argc; i++) {
if (!NeedSecond)
{
if ( (argv[i][0] != '/') && (argv[i][0] != '-') )
{
rc=FALSE;
goto CommandLineFinish;
}
if ( _tcslen(argv[i]) != 2 )
{
rc=FALSE;
goto CommandLineFinish;
}
c=argv[i][1];
switch ( c )
{
case 'q':
case 'Q': pComArgs->QuietInstall=TRUE;
break;
case 'i':
case 'I': NeedSecond=TRUE;;
break;
case 'n':
case 'N': NeedSecond=TRUE;
break;
case 'z':
case 'Z': pComArgs->StressInstall=TRUE;
break;
case 'u':
case 'U': pComArgs->UIStressInstall=TRUE;
pComArgs->StressInstall=TRUE;
break;
default: {
rc=FALSE;
goto CommandLineFinish;
}
}
} else {
NeedSecond = FALSE;
switch ( c )
{
case 'i':
case 'I': _tcscpy(pComArgs->szInstDir,argv[i]);
break;
case 'n':
case 'N': _tcscpy(pComArgs->szMsiName,argv[i]);
break;
default: {
rc=FALSE;
goto CommandLineFinish;
}
}
}
}
if (pComArgs->szMsiName[0] == 0)
{
#ifdef BUILD_X86
_tcscpy(pComArgs->szMsiName, _T("dbg_x86.msi") );
_tcscpy(pComArgs->szProductRegKey, _T("Software\\Microsoft\\DebuggingTools\\AddRemove") );
#elif defined(BUILD_IA64)
_tcscpy(pComArgs->szMsiName, _T("dbg_ia64.msi") );
_tcscpy(pComArgs->szProductRegKey, _T("Software\\Microsoft\\DebuggingTools64\\AddRemove") );
#endif
}
CommandLineFinish:
free(szCmdLineTmp);
free(argv);
return (rc);
}