113ac6606SAndrew Rist /************************************************************** 2cdf0e10cSrcweir * 313ac6606SAndrew Rist * Licensed to the Apache Software Foundation (ASF) under one 413ac6606SAndrew Rist * or more contributor license agreements. See the NOTICE file 513ac6606SAndrew Rist * distributed with this work for additional information 613ac6606SAndrew Rist * regarding copyright ownership. The ASF licenses this file 713ac6606SAndrew Rist * to you under the Apache License, Version 2.0 (the 813ac6606SAndrew Rist * "License"); you may not use this file except in compliance 913ac6606SAndrew Rist * with the License. You may obtain a copy of the License at 1013ac6606SAndrew Rist * 1113ac6606SAndrew Rist * http://www.apache.org/licenses/LICENSE-2.0 1213ac6606SAndrew Rist * 1313ac6606SAndrew Rist * Unless required by applicable law or agreed to in writing, 1413ac6606SAndrew Rist * software distributed under the License is distributed on an 1513ac6606SAndrew Rist * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 1613ac6606SAndrew Rist * KIND, either express or implied. See the License for the 1713ac6606SAndrew Rist * specific language governing permissions and limitations 1813ac6606SAndrew Rist * under the License. 1913ac6606SAndrew Rist * 2013ac6606SAndrew Rist *************************************************************/ 2113ac6606SAndrew Rist 2213ac6606SAndrew Rist 23cdf0e10cSrcweir 24cdf0e10cSrcweir #define WIN // scope W32 API 25cdf0e10cSrcweir 26cdf0e10cSrcweir #if defined _MSC_VER 27cdf0e10cSrcweir #pragma warning(push, 1) 28cdf0e10cSrcweir #endif 29cdf0e10cSrcweir #include <windows.h> 30cdf0e10cSrcweir #if defined _MSC_VER 31cdf0e10cSrcweir #pragma warning(pop) 32cdf0e10cSrcweir #endif 33cdf0e10cSrcweir #include <tchar.h> 34cdf0e10cSrcweir #include <assert.h> 35cdf0e10cSrcweir #include <shlwapi.h> 36cdf0e10cSrcweir #include <new> 37cdf0e10cSrcweir #include <time.h> 38cdf0e10cSrcweir #include <mbctype.h> 39cdf0e10cSrcweir #include <locale.h> 40cdf0e10cSrcweir #include <Msiquery.h> 41cdf0e10cSrcweir #include <MsiDefs.h> 42cdf0e10cSrcweir #include "strsafe.h" 43cdf0e10cSrcweir 44cdf0e10cSrcweir #include "setup.hxx" 45cdf0e10cSrcweir 46cdf0e10cSrcweir #include "resource.h" 47cdf0e10cSrcweir 48cdf0e10cSrcweir //-------------------------------------------------------------------------- 49cdf0e10cSrcweir 50cdf0e10cSrcweir #define MAX_STR_LENGTH 32000 51cdf0e10cSrcweir #define MAX_TEXT_LENGTH 1024 52cdf0e10cSrcweir #define MAX_LANGUAGE_LEN 80 53cdf0e10cSrcweir #define MAX_STR_CAPTION 256 54cdf0e10cSrcweir #define VERSION_SIZE 80 55cdf0e10cSrcweir #define SECTION_SETUP TEXT( "Setup" ) 56cdf0e10cSrcweir #define SECTION_LANGUAGE TEXT( "Languages" ) 57cdf0e10cSrcweir #define PRODUCT_NAME_VAR TEXT( "%PRODUCTNAME" ) 58cdf0e10cSrcweir #define PRODUCT_VERSION TEXT( "ProductVersion" ) 59cdf0e10cSrcweir #define ERROR_SHOW_USAGE -2 60cdf0e10cSrcweir #define ERROR_SETUP_TO_OLD -3 61cdf0e10cSrcweir #define ERROR_SETUP_NOT_FOUND -4 62cdf0e10cSrcweir 63cdf0e10cSrcweir #define PARAM_SETUP_USED TEXT( " SETUP_USED=1 " ) 64cdf0e10cSrcweir #define PARAM_PACKAGE TEXT( "/I " ) 65cdf0e10cSrcweir #define PARAM_MINOR_UPGRADE TEXT( "/FVOMUS " ) 66cdf0e10cSrcweir #define PARAM_ADMIN TEXT( "/A " ) 67cdf0e10cSrcweir #define PARAM_TRANSFORM TEXT( " TRANSFORMS=" ) 68cdf0e10cSrcweir #define PARAM_REBOOT TEXT( " REBOOT=Force" ) 69cdf0e10cSrcweir #define PARAM_PATCH TEXT( " /update " ) 70cdf0e10cSrcweir #define PARAM_REG_ALL_MSO_TYPES TEXT( "REGISTER_ALL_MSO_TYPES=1 " ) 71cdf0e10cSrcweir #define PARAM_REG_NO_MSO_TYPES TEXT( "REGISTER_NO_MSO_TYPES=1 " ) 72cdf0e10cSrcweir #define PARAM_SILENTINSTALL TEXT( " /QB" ) 73cdf0e10cSrcweir 74cdf0e10cSrcweir #define PARAM_RUNNING TEXT( "ignore_running" ) 75cdf0e10cSrcweir #define CMDLN_REG_ALL_MSO_TYPES TEXT( "msoreg=1" ) 76cdf0e10cSrcweir #define CMDLN_REG_NO_MSO_TYPES TEXT( "msoreg=0" ) 77cdf0e10cSrcweir 78cdf0e10cSrcweir #define MSI_DLL TEXT( "msi.dll" ) 79cdf0e10cSrcweir #define ADVAPI32_DLL TEXT( "advapi32.dll" ) 80cdf0e10cSrcweir #define PROFILE_NAME TEXT( "setup.ini" ) 81cdf0e10cSrcweir 82cdf0e10cSrcweir #define RUNTIME_X64_NAME TEXT( "redist\\vcredist_x64.exe" ) 83cdf0e10cSrcweir #define RUNTIME_X86_NAME TEXT( "redist\\vcredist_x86.exe" ) 84*d45c90b4Sarielch // Microsoft Visual C++ 2008 Redistributable - x86 9.0.30729.6161 85*d45c90b4Sarielch #define PRODUCTCODE_X86 TEXT( "{9BE518E6-ECC6-35A9-88E4-87755C07200F}" ) 86*d45c90b4Sarielch // Microsoft Visual C++ 2008 Redistributable - x64 9.0.30729.6161 87*d45c90b4Sarielch #define PRODUCTCODE_X64 TEXT( "{5FCE6D76-F5DC-37AB-B2B8-22AB8CEDB1D4}" ) 88cdf0e10cSrcweir 89cdf0e10cSrcweir #define MSIAPI_DllGetVersion "DllGetVersion" 90cdf0e10cSrcweir #define ADVAPI32API_CheckTokenMembership "CheckTokenMembership" 91cdf0e10cSrcweir 92cdf0e10cSrcweir typedef HRESULT (CALLBACK* PFnDllGetVersion)( DLLVERSIONINFO *pdvi); 93cdf0e10cSrcweir typedef BOOL (WINAPI* PFnCheckTokenMembership)(HANDLE TokenHandle, PSID SidToCheck, PBOOL IsMember); 94cdf0e10cSrcweir 95cdf0e10cSrcweir #ifdef DEBUG 96cdf0e10cSrcweir inline void OutputDebugStringFormat( LPCTSTR pFormat, ... ) 97cdf0e10cSrcweir { 98cdf0e10cSrcweir TCHAR buffer[1024]; 99cdf0e10cSrcweir va_list args; 100cdf0e10cSrcweir 101cdf0e10cSrcweir va_start( args, pFormat ); 102cdf0e10cSrcweir StringCchVPrintf( buffer, sizeof(buffer), pFormat, args ); 103cdf0e10cSrcweir OutputDebugString( buffer ); 104cdf0e10cSrcweir } 105cdf0e10cSrcweir #else 106cdf0e10cSrcweir static inline void OutputDebugStringFormat( LPCTSTR, ... ) 107cdf0e10cSrcweir { 108cdf0e10cSrcweir } 109cdf0e10cSrcweir #endif 110cdf0e10cSrcweir 111cdf0e10cSrcweir //-------------------------------------------------------------------------- 112cdf0e10cSrcweir 113cdf0e10cSrcweir const TCHAR sInstKey[] = TEXT( "Software\\Microsoft\\Windows\\CurrentVersion\\Installer" ); 114cdf0e10cSrcweir const TCHAR sInstLocValue[] = TEXT( "InstallerLocation" ); 115cdf0e10cSrcweir const TCHAR sMsiDll[] = TEXT( "\\msi.dll" ); 116cdf0e10cSrcweir const TCHAR sMsiExe[] = TEXT( "\\msiexec.exe" ); 117cdf0e10cSrcweir const TCHAR sDelayReboot[] = TEXT( " /c:\"msiinst /delayreboot\"" ); 118cdf0e10cSrcweir const TCHAR sMsiQuiet[] = TEXT( " /q" ); 119cdf0e10cSrcweir const TCHAR sMemMapName[] = TEXT( "Global\\MsiErrorObject" ); 120cdf0e10cSrcweir 121cdf0e10cSrcweir //-------------------------------------------------------------------------- 122cdf0e10cSrcweir SetupAppX::SetupAppX() 123cdf0e10cSrcweir { 124cdf0e10cSrcweir m_hInst = NULL; 125cdf0e10cSrcweir m_hMapFile = NULL; 126cdf0e10cSrcweir m_pAppTitle = NULL; 127cdf0e10cSrcweir m_pCmdLine = NULL; 128cdf0e10cSrcweir 129cdf0e10cSrcweir m_pDatabase = NULL; 130cdf0e10cSrcweir m_pReqVersion = NULL; 131cdf0e10cSrcweir m_pProductName = NULL; 132cdf0e10cSrcweir m_pAdvertise = NULL; 133cdf0e10cSrcweir m_pTmpName = NULL; 134cdf0e10cSrcweir m_pLogFile = NULL; 135cdf0e10cSrcweir m_pModuleFile = NULL; 136cdf0e10cSrcweir m_pPatchFiles = NULL; 137cdf0e10cSrcweir m_pMSIErrorCode = NULL; 138cdf0e10cSrcweir m_pUpgradeKey = NULL; 139cdf0e10cSrcweir m_pProductVersion = NULL; 140cdf0e10cSrcweir 141cdf0e10cSrcweir m_pErrorText = new TCHAR[ MAX_TEXT_LENGTH ]; 142cdf0e10cSrcweir m_pErrorText[0] = '\0'; 143cdf0e10cSrcweir 144cdf0e10cSrcweir m_nLanguageID = 0; 145cdf0e10cSrcweir m_nLanguageCount = 0; 146cdf0e10cSrcweir m_ppLanguageList = NULL; 147cdf0e10cSrcweir 148cdf0e10cSrcweir m_bQuiet = false; 149cdf0e10cSrcweir m_bRegNoMsoTypes = false; 150cdf0e10cSrcweir m_bRegAllMsoTypes = false; 151cdf0e10cSrcweir m_bIsMinorUpgrade = false; 152cdf0e10cSrcweir m_bSupportsPatch = false; 153cdf0e10cSrcweir 154cdf0e10cSrcweir m_bIgnoreAlreadyRunning = false; 155cdf0e10cSrcweir } 156cdf0e10cSrcweir 157cdf0e10cSrcweir //-------------------------------------------------------------------------- 158cdf0e10cSrcweir SetupAppX::~SetupAppX() 159cdf0e10cSrcweir { 160cdf0e10cSrcweir if ( m_ppLanguageList ) 161cdf0e10cSrcweir { 162cdf0e10cSrcweir for ( int i = 0; i < m_nLanguageCount; i++ ) 163cdf0e10cSrcweir if ( m_ppLanguageList[i] ) 164cdf0e10cSrcweir delete m_ppLanguageList[ i ]; 165cdf0e10cSrcweir delete [] m_ppLanguageList; 166cdf0e10cSrcweir } 167cdf0e10cSrcweir 168cdf0e10cSrcweir time_t aTime; 169cdf0e10cSrcweir time( &aTime ); 170cdf0e10cSrcweir tm *pTime = localtime( &aTime ); // Convert time to struct tm form 171cdf0e10cSrcweir 172cdf0e10cSrcweir Log( TEXT( "End: %s\n\r\n\r\n" ), _tasctime( pTime ) ); 173cdf0e10cSrcweir 174cdf0e10cSrcweir if ( m_pLogFile ) fclose( m_pLogFile ); 175cdf0e10cSrcweir 176cdf0e10cSrcweir if ( m_pTmpName ) 177cdf0e10cSrcweir { 178cdf0e10cSrcweir _tremove( m_pTmpName ); 179cdf0e10cSrcweir free( m_pTmpName ); 180cdf0e10cSrcweir } 181cdf0e10cSrcweir 182cdf0e10cSrcweir if ( m_pMSIErrorCode ) UnmapViewOfFile( m_pMSIErrorCode ); 183cdf0e10cSrcweir if ( m_hMapFile ) CloseHandle( m_hMapFile ); 184cdf0e10cSrcweir 185cdf0e10cSrcweir if ( m_pAppTitle ) delete [] m_pAppTitle; 186cdf0e10cSrcweir if ( m_pDatabase ) delete [] m_pDatabase; 187cdf0e10cSrcweir if ( m_pReqVersion ) delete [] m_pReqVersion; 188cdf0e10cSrcweir if ( m_pProductName ) delete [] m_pProductName; 189cdf0e10cSrcweir if ( m_pAdvertise ) delete [] m_pAdvertise; 190cdf0e10cSrcweir if ( m_pLogFile ) delete [] m_pLogFile; 191cdf0e10cSrcweir if ( m_pErrorText ) delete [] m_pErrorText; 192cdf0e10cSrcweir if ( m_pModuleFile ) delete [] m_pModuleFile; 193cdf0e10cSrcweir if ( m_pPatchFiles ) delete [] m_pPatchFiles; 194cdf0e10cSrcweir if ( m_pUpgradeKey ) delete [] m_pUpgradeKey; 195cdf0e10cSrcweir if ( m_pProductVersion ) delete [] m_pProductVersion; 196cdf0e10cSrcweir } 197cdf0e10cSrcweir 198cdf0e10cSrcweir //-------------------------------------------------------------------------- 199cdf0e10cSrcweir boolean SetupAppX::Initialize( HINSTANCE hInst ) 200cdf0e10cSrcweir { 201cdf0e10cSrcweir m_pCmdLine = WIN::GetCommandLine(); 202cdf0e10cSrcweir m_hInst = hInst; 203cdf0e10cSrcweir 204cdf0e10cSrcweir // Load our AppTitle (caption) 205cdf0e10cSrcweir m_pAppTitle = new TCHAR[ MAX_STR_CAPTION ]; 206cdf0e10cSrcweir m_pAppTitle[0] = '\0'; 207cdf0e10cSrcweir WIN::LoadString( hInst, IDS_APP_TITLE, m_pAppTitle, MAX_STR_CAPTION ); 208cdf0e10cSrcweir 209cdf0e10cSrcweir // Obtain path we are running from 210cdf0e10cSrcweir m_pModuleFile = new TCHAR[ MAX_PATH ]; 211cdf0e10cSrcweir m_pModuleFile[ 0 ] = '\0'; 212cdf0e10cSrcweir 213cdf0e10cSrcweir if ( 0 == WIN::GetModuleFileName( hInst, m_pModuleFile, MAX_PATH ) ) 214cdf0e10cSrcweir { 215cdf0e10cSrcweir SetError( WIN::GetLastError() ); 216cdf0e10cSrcweir return false; 217cdf0e10cSrcweir } 218cdf0e10cSrcweir 219cdf0e10cSrcweir if ( ! GetCmdLineParameters( &m_pCmdLine ) ) 220cdf0e10cSrcweir return false; 221cdf0e10cSrcweir 222cdf0e10cSrcweir m_hMapFile = CreateFileMapping( 223cdf0e10cSrcweir INVALID_HANDLE_VALUE, // use paging file 224cdf0e10cSrcweir NULL, // default security 225cdf0e10cSrcweir PAGE_READWRITE, // read/write access 226cdf0e10cSrcweir 0, // max. object size 227cdf0e10cSrcweir sizeof( int ), // buffer size 228cdf0e10cSrcweir sMemMapName ); 229cdf0e10cSrcweir if ( m_hMapFile ) 230cdf0e10cSrcweir { 231cdf0e10cSrcweir m_pMSIErrorCode = (int*) MapViewOfFile( m_hMapFile, // handle to map object 232cdf0e10cSrcweir FILE_MAP_ALL_ACCESS, // read/write permission 233cdf0e10cSrcweir 0, 234cdf0e10cSrcweir 0, 235cdf0e10cSrcweir sizeof( int ) ); 236cdf0e10cSrcweir if ( m_pMSIErrorCode ) 237cdf0e10cSrcweir *m_pMSIErrorCode = 0; 238cdf0e10cSrcweir else 239cdf0e10cSrcweir OutputDebugStringFormat( TEXT("Could not map view of file (%d).\n"), GetLastError() ); 240cdf0e10cSrcweir } 241cdf0e10cSrcweir else 242cdf0e10cSrcweir OutputDebugStringFormat( TEXT("Could not create file mapping object (%d).\n"), GetLastError() ); 243cdf0e10cSrcweir 244cdf0e10cSrcweir Log( TEXT("Starting: %s\r\n"), m_pModuleFile ); 245cdf0e10cSrcweir Log( TEXT(" CommandLine=<%s>\r\n"), m_pCmdLine ); 246cdf0e10cSrcweir 247cdf0e10cSrcweir if ( m_bQuiet ) 248cdf0e10cSrcweir Log( TEXT(" Using quiet install mode\r\n") ); 249cdf0e10cSrcweir 250cdf0e10cSrcweir time_t aTime; 251cdf0e10cSrcweir time( &aTime ); 252cdf0e10cSrcweir tm* pTime = localtime( &aTime ); 253cdf0e10cSrcweir Log( TEXT(" Begin: %s\n"), _tasctime( pTime ) ); 254cdf0e10cSrcweir 255cdf0e10cSrcweir return true; 256cdf0e10cSrcweir } 257cdf0e10cSrcweir 258cdf0e10cSrcweir //-------------------------------------------------------------------------- 259cdf0e10cSrcweir boolean SetupAppX::GetProfileSection( LPCTSTR pFileName, LPCTSTR pSection, 260cdf0e10cSrcweir DWORD& rSize, LPTSTR *pRetBuf ) 261cdf0e10cSrcweir { 262cdf0e10cSrcweir if ( !rSize || !*pRetBuf ) 263cdf0e10cSrcweir { 264cdf0e10cSrcweir rSize = 512; 265cdf0e10cSrcweir *pRetBuf = new TCHAR[ rSize ]; 266cdf0e10cSrcweir } 267cdf0e10cSrcweir 268cdf0e10cSrcweir DWORD nRet = GetPrivateProfileSection( pSection, *pRetBuf, rSize, pFileName ); 269cdf0e10cSrcweir 270cdf0e10cSrcweir if ( nRet && ( nRet + 2 > rSize ) ) // buffer was too small, retry with bigger one 271cdf0e10cSrcweir { 272cdf0e10cSrcweir if ( nRet < 32767 - 2 ) 273cdf0e10cSrcweir { 274cdf0e10cSrcweir delete [] (*pRetBuf); 275cdf0e10cSrcweir rSize = nRet + 2; 276cdf0e10cSrcweir *pRetBuf = new TCHAR[ rSize ]; 277cdf0e10cSrcweir 278cdf0e10cSrcweir nRet = GetPrivateProfileSection( pSection, *pRetBuf, rSize, pFileName ); 279cdf0e10cSrcweir } 280cdf0e10cSrcweir } 281cdf0e10cSrcweir 282cdf0e10cSrcweir if ( !nRet ) 283cdf0e10cSrcweir { 284cdf0e10cSrcweir SetError( WIN::GetLastError() ); 285cdf0e10cSrcweir 286cdf0e10cSrcweir TCHAR sBuf[80]; 287cdf0e10cSrcweir StringCchPrintf( sBuf, 80, TEXT("ERROR: GetPrivateProfileSection(): GetLastError returned %u\r\n"), GetError() ); 288cdf0e10cSrcweir Log( sBuf ); 289cdf0e10cSrcweir return false; 290cdf0e10cSrcweir } 291cdf0e10cSrcweir else if ( nRet + 2 > rSize ) 292cdf0e10cSrcweir { 293cdf0e10cSrcweir SetError( ERROR_OUTOFMEMORY ); 294cdf0e10cSrcweir Log( TEXT( "ERROR: GetPrivateProfileSection() out of memory\r\n" ) ); 295cdf0e10cSrcweir return false; 296cdf0e10cSrcweir } 297cdf0e10cSrcweir 298cdf0e10cSrcweir Log( TEXT( " GetProfileSection read %s\r\n" ), pSection ); 299cdf0e10cSrcweir 300cdf0e10cSrcweir return true; 301cdf0e10cSrcweir } 302cdf0e10cSrcweir 303cdf0e10cSrcweir //-------------------------------------------------------------------------- 304cdf0e10cSrcweir boolean SetupAppX::ReadProfile() 305cdf0e10cSrcweir { 306cdf0e10cSrcweir boolean bRet = false; 307cdf0e10cSrcweir TCHAR *sProfilePath = 0; 308cdf0e10cSrcweir 309cdf0e10cSrcweir if ( GetPathToFile( PROFILE_NAME, &sProfilePath ) ) 310cdf0e10cSrcweir { 311cdf0e10cSrcweir DWORD nSize = 0; 312cdf0e10cSrcweir LPTSTR pRetBuf = NULL; 313cdf0e10cSrcweir 314cdf0e10cSrcweir Log( TEXT( " Open ini file: <%s>\r\n" ), sProfilePath ); 315cdf0e10cSrcweir 316cdf0e10cSrcweir bRet = GetProfileSection( sProfilePath, SECTION_SETUP, nSize, &pRetBuf ); 317cdf0e10cSrcweir 318cdf0e10cSrcweir if ( !bRet ) 319cdf0e10cSrcweir { 320cdf0e10cSrcweir LPTSTR pTmpFile = CopyIniFile( sProfilePath ); 321cdf0e10cSrcweir delete [] sProfilePath; 322cdf0e10cSrcweir sProfilePath = pTmpFile; 323cdf0e10cSrcweir 324cdf0e10cSrcweir if ( sProfilePath ) 325cdf0e10cSrcweir { 326cdf0e10cSrcweir SetError( ERROR_SUCCESS ); 327cdf0e10cSrcweir 328cdf0e10cSrcweir Log( TEXT( " Could not open inifile, copied ini file to: <%s>\r\n" ), sProfilePath ); 329cdf0e10cSrcweir bRet = GetProfileSection( sProfilePath, SECTION_SETUP, nSize, &pRetBuf ); 330cdf0e10cSrcweir } 331cdf0e10cSrcweir } 332cdf0e10cSrcweir 333cdf0e10cSrcweir if ( bRet ) 334cdf0e10cSrcweir { 335cdf0e10cSrcweir LPTSTR pCurLine = pRetBuf; 336cdf0e10cSrcweir while ( *pCurLine ) 337cdf0e10cSrcweir { 338cdf0e10cSrcweir LPTSTR pName = 0; 339cdf0e10cSrcweir LPTSTR pValue = 0; 340cdf0e10cSrcweir 341cdf0e10cSrcweir pCurLine += GetNameValue( pCurLine, &pName, &pValue ); 342cdf0e10cSrcweir 343cdf0e10cSrcweir if ( lstrcmpi( TEXT( "database" ), pName ) == 0 ) 344cdf0e10cSrcweir { 345cdf0e10cSrcweir m_pDatabase = pValue; 346cdf0e10cSrcweir Log( TEXT( " Database = %s\r\n" ), pValue ); 347cdf0e10cSrcweir } 348cdf0e10cSrcweir else if ( lstrcmpi( TEXT( "msiversion" ), pName ) == 0 ) 349cdf0e10cSrcweir { 350cdf0e10cSrcweir m_pReqVersion = pValue; 351cdf0e10cSrcweir Log( TEXT( " msiversion = %s\r\n" ), pValue ); 352cdf0e10cSrcweir } 353cdf0e10cSrcweir else if ( lstrcmpi( TEXT( "productname" ), pName ) == 0 ) 354cdf0e10cSrcweir { 355cdf0e10cSrcweir m_pProductName = pValue; 356cdf0e10cSrcweir Log( TEXT( " productname = %s\r\n" ), pValue ); 357cdf0e10cSrcweir m_pAppTitle = SetProdToAppTitle( m_pProductName ); 358cdf0e10cSrcweir } 359cdf0e10cSrcweir else if ( lstrcmpi( TEXT( "upgradekey" ), pName ) == 0 ) 360cdf0e10cSrcweir { 361cdf0e10cSrcweir m_pUpgradeKey = pValue; 362cdf0e10cSrcweir Log( TEXT( " upgradekey = %s\r\n" ), pValue ); 363cdf0e10cSrcweir } 364cdf0e10cSrcweir else if ( lstrcmpi( TEXT( "productversion" ), pName ) == 0 ) 365cdf0e10cSrcweir { 366cdf0e10cSrcweir m_pProductVersion = pValue; 367cdf0e10cSrcweir Log( TEXT( " productversion = %s\r\n" ), pValue ); 368cdf0e10cSrcweir } 369cdf0e10cSrcweir else if ( lstrcmpi( TEXT( "productcode" ), pName ) == 0 ) 370cdf0e10cSrcweir { 371cdf0e10cSrcweir delete [] pValue; 372cdf0e10cSrcweir } 373cdf0e10cSrcweir else 374cdf0e10cSrcweir { 375cdf0e10cSrcweir Log( TEXT( "Warning: unknown entry in profile <%s>\r\n" ), pName ); 376cdf0e10cSrcweir delete [] pValue; 377cdf0e10cSrcweir } 378cdf0e10cSrcweir } 379cdf0e10cSrcweir } 380cdf0e10cSrcweir 381cdf0e10cSrcweir if ( bRet && ( !m_pDatabase || !m_pReqVersion || !m_pProductName ) ) 382cdf0e10cSrcweir { 383cdf0e10cSrcweir Log( TEXT( "ERROR: incomplete 'Setup' section in profile\r\n" ) ); 384cdf0e10cSrcweir SetError( ERROR_INVALID_DATA ); 385cdf0e10cSrcweir bRet = false; 386cdf0e10cSrcweir } 387cdf0e10cSrcweir 388cdf0e10cSrcweir if ( bRet ) 389cdf0e10cSrcweir bRet = GetProfileSection( sProfilePath, SECTION_LANGUAGE, nSize, &pRetBuf ); 390cdf0e10cSrcweir 391cdf0e10cSrcweir if ( bRet ) 392cdf0e10cSrcweir { 393cdf0e10cSrcweir LPTSTR pName = 0; 394cdf0e10cSrcweir LPTSTR pValue = 0; 395cdf0e10cSrcweir LPTSTR pCurLine = pRetBuf; 396cdf0e10cSrcweir LPTSTR pLastChar; 397cdf0e10cSrcweir int nNext = 0; 398cdf0e10cSrcweir 399cdf0e10cSrcweir // first line in this section should be the language count 400cdf0e10cSrcweir nNext = GetNameValue( pCurLine, &pName, &pValue ); 401cdf0e10cSrcweir if ( lstrcmpi( TEXT( "count" ), pName ) == 0 ) 402cdf0e10cSrcweir { 403cdf0e10cSrcweir Log( TEXT( " Languages = %s\r\n" ), pValue ); 404cdf0e10cSrcweir m_nLanguageCount = _tcstol( pValue, &pLastChar, 10 ); 405cdf0e10cSrcweir pCurLine += nNext; 406cdf0e10cSrcweir delete [] pValue; 407cdf0e10cSrcweir } 408cdf0e10cSrcweir 409cdf0e10cSrcweir m_ppLanguageList = new LanguageDataX*[ m_nLanguageCount ]; 410cdf0e10cSrcweir 411cdf0e10cSrcweir for ( int i=0; i < m_nLanguageCount; i++ ) 412cdf0e10cSrcweir { 413cdf0e10cSrcweir if ( !*pCurLine ) 414cdf0e10cSrcweir { 415cdf0e10cSrcweir m_nLanguageCount = i; 416cdf0e10cSrcweir break; 417cdf0e10cSrcweir } 418cdf0e10cSrcweir 419cdf0e10cSrcweir pCurLine += GetNameValue( pCurLine, &pName, &pValue ); 420cdf0e10cSrcweir m_ppLanguageList[ i ] = new LanguageDataX( pValue ); 421cdf0e10cSrcweir Log( TEXT( " Language = %s\r\n" ), pValue ); 422cdf0e10cSrcweir 423cdf0e10cSrcweir if ( m_ppLanguageList[ i ]->m_pTransform ) 424cdf0e10cSrcweir Log( TEXT( " Transform = %s\r\n" ), m_ppLanguageList[ i ]->m_pTransform ); 425cdf0e10cSrcweir 426cdf0e10cSrcweir delete [] pValue; 427cdf0e10cSrcweir } 428cdf0e10cSrcweir } 429cdf0e10cSrcweir 430cdf0e10cSrcweir if ( pRetBuf ) 431cdf0e10cSrcweir delete [] pRetBuf; 432cdf0e10cSrcweir } 433cdf0e10cSrcweir 434cdf0e10cSrcweir if ( sProfilePath && ! m_pTmpName ) 435cdf0e10cSrcweir delete [] sProfilePath; 436cdf0e10cSrcweir 437cdf0e10cSrcweir return bRet; 438cdf0e10cSrcweir } 439cdf0e10cSrcweir 440cdf0e10cSrcweir //-------------------------------------------------------------------------- 441cdf0e10cSrcweir void SetupAppX::AddFileToPatchList( TCHAR* pPath, TCHAR* pFile ) 442cdf0e10cSrcweir { 443cdf0e10cSrcweir if ( m_pPatchFiles == NULL ) 444cdf0e10cSrcweir { 445cdf0e10cSrcweir m_pPatchFiles = new TCHAR[ MAX_STR_LENGTH ]; 446cdf0e10cSrcweir StringCchCopy( m_pPatchFiles, MAX_STR_LENGTH, TEXT("\"") ); 447cdf0e10cSrcweir } 448cdf0e10cSrcweir else 449cdf0e10cSrcweir StringCchCat( m_pPatchFiles, MAX_STR_LENGTH, TEXT(";") ); 450cdf0e10cSrcweir 451cdf0e10cSrcweir StringCchCat( m_pPatchFiles, MAX_STR_LENGTH, pPath ); 452cdf0e10cSrcweir StringCchCat( m_pPatchFiles, MAX_STR_LENGTH, pFile ); 453cdf0e10cSrcweir } 454cdf0e10cSrcweir 455cdf0e10cSrcweir //-------------------------------------------------------------------------- 456cdf0e10cSrcweir boolean SetupAppX::GetPatches() 457cdf0e10cSrcweir { 458cdf0e10cSrcweir boolean bRet = true; 459cdf0e10cSrcweir 460cdf0e10cSrcweir int nPatternLen = lstrlen( m_pModuleFile ) + 7; // 1 for null terminator, 1 for back slash, 5 for extensions 461cdf0e10cSrcweir TCHAR* pPattern = new TCHAR[ nPatternLen ]; 462cdf0e10cSrcweir TCHAR* pBaseDir = new TCHAR[ nPatternLen ]; 463cdf0e10cSrcweir 464cdf0e10cSrcweir // find 'setup.exe' in the path so we can remove it 465cdf0e10cSrcweir TCHAR *pFilePart = 0; 466cdf0e10cSrcweir if ( 0 == GetFullPathName( m_pModuleFile, nPatternLen, pPattern, &pFilePart ) ) 467cdf0e10cSrcweir { 468cdf0e10cSrcweir SetError( WIN::GetLastError() ); 469cdf0e10cSrcweir bRet = false; 470cdf0e10cSrcweir } 471cdf0e10cSrcweir else 472cdf0e10cSrcweir { 473cdf0e10cSrcweir if ( pFilePart ) 474cdf0e10cSrcweir *pFilePart = '\0'; 475cdf0e10cSrcweir StringCchCopy( pBaseDir, nPatternLen, pPattern ); 476cdf0e10cSrcweir StringCchCat( pPattern, nPatternLen, TEXT("*.msp") ); 477cdf0e10cSrcweir 478cdf0e10cSrcweir WIN32_FIND_DATA aFindFileData; 479cdf0e10cSrcweir 480cdf0e10cSrcweir HANDLE hFindPatches = FindFirstFile( pPattern, &aFindFileData ); 481cdf0e10cSrcweir 482cdf0e10cSrcweir if ( hFindPatches != INVALID_HANDLE_VALUE ) 483cdf0e10cSrcweir { 484cdf0e10cSrcweir if ( ! IsPatchInstalled( pBaseDir, aFindFileData.cFileName ) ) 485cdf0e10cSrcweir AddFileToPatchList( pBaseDir, aFindFileData.cFileName ); 486cdf0e10cSrcweir 487cdf0e10cSrcweir while ( FindNextFile( hFindPatches, &aFindFileData ) ) 488cdf0e10cSrcweir { 489cdf0e10cSrcweir if ( ! IsPatchInstalled( pBaseDir, aFindFileData.cFileName ) ) 490cdf0e10cSrcweir AddFileToPatchList( pBaseDir, aFindFileData.cFileName ); 491cdf0e10cSrcweir } 492cdf0e10cSrcweir 493cdf0e10cSrcweir if ( m_pPatchFiles != NULL ) 494cdf0e10cSrcweir StringCchCat( m_pPatchFiles, MAX_STR_LENGTH, TEXT("\"") ); 495cdf0e10cSrcweir 496cdf0e10cSrcweir FindClose( hFindPatches ); 497cdf0e10cSrcweir } 498cdf0e10cSrcweir } 499cdf0e10cSrcweir 500cdf0e10cSrcweir delete [] pPattern; 501cdf0e10cSrcweir delete [] pBaseDir; 502cdf0e10cSrcweir 503cdf0e10cSrcweir return bRet; 504cdf0e10cSrcweir } 505cdf0e10cSrcweir 506cdf0e10cSrcweir //-------------------------------------------------------------------------- 507cdf0e10cSrcweir boolean SetupAppX::GetPathToFile( TCHAR* pFileName, TCHAR** pPath ) 508cdf0e10cSrcweir { 509cdf0e10cSrcweir // generate the path to the file = szModuleFile + FileName 510cdf0e10cSrcweir // note: FileName is a relative path 511cdf0e10cSrcweir 512cdf0e10cSrcweir boolean bRet = true; 513cdf0e10cSrcweir 514cdf0e10cSrcweir int nTempPath = lstrlen( m_pModuleFile ) + lstrlen( pFileName ) + 2; // 1 for null terminator, 1 for back slash 515cdf0e10cSrcweir TCHAR* pTempPath = new TCHAR[ nTempPath ]; 516cdf0e10cSrcweir 517cdf0e10cSrcweir // find 'setup.exe' in the path so we can remove it 518cdf0e10cSrcweir TCHAR *pFilePart = 0; 519cdf0e10cSrcweir if ( 0 == GetFullPathName( m_pModuleFile, nTempPath, pTempPath, &pFilePart ) ) 520cdf0e10cSrcweir { 521cdf0e10cSrcweir SetError( WIN::GetLastError() ); 522cdf0e10cSrcweir bRet = false; 523cdf0e10cSrcweir } 524cdf0e10cSrcweir else 525cdf0e10cSrcweir { 526cdf0e10cSrcweir if ( pFilePart ) 527cdf0e10cSrcweir *pFilePart = '\0'; 528cdf0e10cSrcweir 529cdf0e10cSrcweir StringCchCat( pTempPath, nTempPath, pFileName ); 530cdf0e10cSrcweir 531cdf0e10cSrcweir int nPath = 2 * nTempPath; 532cdf0e10cSrcweir *pPath = new TCHAR[ nPath ]; 533cdf0e10cSrcweir 534cdf0e10cSrcweir // normalize the path 535cdf0e10cSrcweir int nReturn = GetFullPathName( pTempPath, nPath, *pPath, &pFilePart ); 536cdf0e10cSrcweir 537cdf0e10cSrcweir if ( nReturn > nPath ) 538cdf0e10cSrcweir { 539cdf0e10cSrcweir // try again, with larger buffer 540cdf0e10cSrcweir delete [] (*pPath); 541cdf0e10cSrcweir nPath = nReturn; 542cdf0e10cSrcweir *pPath = new TCHAR[ nPath ]; 543cdf0e10cSrcweir 544cdf0e10cSrcweir nReturn = GetFullPathName( pTempPath, nPath, *pPath, &pFilePart ); 545cdf0e10cSrcweir } 546cdf0e10cSrcweir 547cdf0e10cSrcweir if ( 0 == nReturn ) 548cdf0e10cSrcweir { 549cdf0e10cSrcweir // error -- invalid path 550cdf0e10cSrcweir SetError( WIN::GetLastError() ); 551cdf0e10cSrcweir bRet = false; 552cdf0e10cSrcweir } 553cdf0e10cSrcweir } 554cdf0e10cSrcweir 555cdf0e10cSrcweir if ( bRet ) // check for the file's existence 556cdf0e10cSrcweir { 557cdf0e10cSrcweir DWORD dwFileAttrib = GetFileAttributes( *pPath ); 558cdf0e10cSrcweir 559cdf0e10cSrcweir if (0xFFFFFFFF == dwFileAttrib) 560cdf0e10cSrcweir { 561cdf0e10cSrcweir StringCchCopy( m_pErrorText, MAX_TEXT_LENGTH, pFileName ); 562cdf0e10cSrcweir SetError( ERROR_FILE_NOT_FOUND ); 563cdf0e10cSrcweir bRet = false; 564cdf0e10cSrcweir } 565cdf0e10cSrcweir } 566cdf0e10cSrcweir 567cdf0e10cSrcweir delete [] pTempPath; 568cdf0e10cSrcweir return bRet; 569cdf0e10cSrcweir } 570cdf0e10cSrcweir 571cdf0e10cSrcweir //-------------------------------------------------------------------------- 572cdf0e10cSrcweir int SetupAppX::GetNameValue( TCHAR* pLine, TCHAR** pName, TCHAR** pValue ) 573cdf0e10cSrcweir { 574cdf0e10cSrcweir int nRet = lstrlen( pLine ) + 1; 575cdf0e10cSrcweir *pValue = 0; 576cdf0e10cSrcweir 577cdf0e10cSrcweir if ( nRet == 1 ) 578cdf0e10cSrcweir return nRet; 579cdf0e10cSrcweir 580cdf0e10cSrcweir LPTSTR pChar = pLine; 581cdf0e10cSrcweir LPTSTR pLast = NULL; 582cdf0e10cSrcweir 583cdf0e10cSrcweir // Skip leading spaces. 584cdf0e10cSrcweir while (' ' == *pChar || '\t' == *pChar) 585cdf0e10cSrcweir pChar = CharNext( pChar ); 586cdf0e10cSrcweir 587cdf0e10cSrcweir *pName = pChar; 588cdf0e10cSrcweir 589cdf0e10cSrcweir // look for the end of the name 590cdf0e10cSrcweir while( *pChar && (' ' != *pChar) && 591cdf0e10cSrcweir ( '\t' != *pChar ) && ( '=' != *pChar ) ) 592cdf0e10cSrcweir pChar = CharNext( pChar ); 593cdf0e10cSrcweir 594cdf0e10cSrcweir if ( ! *pChar ) 595cdf0e10cSrcweir return nRet; 596cdf0e10cSrcweir 597cdf0e10cSrcweir pLast = pChar; 598cdf0e10cSrcweir pChar = CharNext( pChar ); 599cdf0e10cSrcweir *pLast = '\0'; 600cdf0e10cSrcweir 601cdf0e10cSrcweir // look for the start of the value 602cdf0e10cSrcweir while( ( ' ' == *pChar ) || ( '\t' == *pChar ) || 603cdf0e10cSrcweir ( '=' == *pChar ) ) 604cdf0e10cSrcweir pChar = CharNext( pChar ); 605cdf0e10cSrcweir 606cdf0e10cSrcweir int nValueLen = lstrlen( pChar ) + 1; 607cdf0e10cSrcweir *pValue = new TCHAR[ nValueLen ]; 608cdf0e10cSrcweir 609cdf0e10cSrcweir if ( *pValue ) 610cdf0e10cSrcweir StringCchCopy( *pValue, nValueLen, pChar ); 611cdf0e10cSrcweir 612cdf0e10cSrcweir return nRet; 613cdf0e10cSrcweir } 614cdf0e10cSrcweir 615cdf0e10cSrcweir //-------------------------------------------------------------------------- 616cdf0e10cSrcweir boolean SetupAppX::ChooseLanguage( long& rLanguage ) 617cdf0e10cSrcweir { 618cdf0e10cSrcweir rLanguage = 0; 619cdf0e10cSrcweir 620cdf0e10cSrcweir if ( m_bQuiet ) 621cdf0e10cSrcweir return true; 622cdf0e10cSrcweir 623cdf0e10cSrcweir // When there are none or only one language, there is nothing 624cdf0e10cSrcweir // to do here 625cdf0e10cSrcweir if ( m_nLanguageCount > 1 ) 626cdf0e10cSrcweir { 627cdf0e10cSrcweir TCHAR *sString = new TCHAR[ MAX_LANGUAGE_LEN ]; 628cdf0e10cSrcweir 629cdf0e10cSrcweir LANGID nUserDefLang = GetUserDefaultLangID(); 630cdf0e10cSrcweir LANGID nSysDefLang = GetSystemDefaultLangID(); 631cdf0e10cSrcweir 632cdf0e10cSrcweir int nUserPrimary = PRIMARYLANGID( nUserDefLang ); 633cdf0e10cSrcweir int nSysPrimary = PRIMARYLANGID( nSysDefLang ); 634cdf0e10cSrcweir 635cdf0e10cSrcweir long nUserIndex = -1; 636cdf0e10cSrcweir long nUserPrimIndex = -1; 637cdf0e10cSrcweir long nSystemIndex = -1; 638cdf0e10cSrcweir long nSystemPrimIndex = -1; 639cdf0e10cSrcweir long nParamIndex = -1; 640cdf0e10cSrcweir 641cdf0e10cSrcweir for ( long i=0; i<GetLanguageCount(); i++ ) 642cdf0e10cSrcweir { 643cdf0e10cSrcweir long nLanguage = GetLanguageID( i ); 644cdf0e10cSrcweir int nPrimary = PRIMARYLANGID( nLanguage ); 645cdf0e10cSrcweir GetLanguageName( nLanguage, sString ); 646cdf0e10cSrcweir Log( TEXT( " Info: found Language: %s\r\n" ), sString ); 647cdf0e10cSrcweir 648cdf0e10cSrcweir if ( nLanguage == nUserDefLang ) 649cdf0e10cSrcweir nUserIndex = i; 650cdf0e10cSrcweir if ( nPrimary == nUserPrimary ) 651cdf0e10cSrcweir nUserPrimIndex = i; 652cdf0e10cSrcweir if ( nLanguage == nSysDefLang ) 653cdf0e10cSrcweir nSystemIndex = i; 654cdf0e10cSrcweir if ( nPrimary == nSysPrimary ) 655cdf0e10cSrcweir nSystemPrimIndex = i; 656cdf0e10cSrcweir if ( m_nLanguageID && ( nLanguage == m_nLanguageID ) ) 657cdf0e10cSrcweir nParamIndex = i; 658cdf0e10cSrcweir } 659cdf0e10cSrcweir 660cdf0e10cSrcweir if ( m_nLanguageID && ( nParamIndex == -1 ) ) 661cdf0e10cSrcweir { 662cdf0e10cSrcweir Log( TEXT( "Warning: Language chosen with parameter -lang not found.\r\n" ) ); 663cdf0e10cSrcweir } 664cdf0e10cSrcweir 665cdf0e10cSrcweir if ( nParamIndex != -1 ) 666cdf0e10cSrcweir { 667cdf0e10cSrcweir Log( TEXT( "Info: Found language chosen with parameter -lang.\r\n" ) ); 668cdf0e10cSrcweir rLanguage = GetLanguageID( nParamIndex ); 669cdf0e10cSrcweir } 670cdf0e10cSrcweir else if ( nUserIndex != -1 ) 671cdf0e10cSrcweir { 672cdf0e10cSrcweir Log( TEXT( "Info: Found user default language.\r\n" ) ); 673cdf0e10cSrcweir rLanguage = GetLanguageID( nUserIndex ); 674cdf0e10cSrcweir } 675cdf0e10cSrcweir else if ( nUserPrimIndex != -1 ) 676cdf0e10cSrcweir { 677cdf0e10cSrcweir Log( TEXT( "Info: Found user default primary language.\r\n" ) ); 678cdf0e10cSrcweir rLanguage = GetLanguageID( nUserPrimIndex ); 679cdf0e10cSrcweir } 680cdf0e10cSrcweir else if ( nSystemIndex != -1 ) 681cdf0e10cSrcweir { 682cdf0e10cSrcweir Log( TEXT( "Info: Found system default language.\r\n" ) ); 683cdf0e10cSrcweir rLanguage = GetLanguageID( nSystemIndex ); 684cdf0e10cSrcweir } 685cdf0e10cSrcweir else if ( nSystemPrimIndex != -1 ) 686cdf0e10cSrcweir { 687cdf0e10cSrcweir Log( TEXT( "Info: Found system default primary language.\r\n" ) ); 688cdf0e10cSrcweir rLanguage = GetLanguageID( nSystemPrimIndex ); 689cdf0e10cSrcweir } 690cdf0e10cSrcweir else 691cdf0e10cSrcweir { 692cdf0e10cSrcweir Log( TEXT( "Info: Use default language from ini file.\r\n" ) ); 693cdf0e10cSrcweir rLanguage = GetLanguageID( 0 ); 694cdf0e10cSrcweir } 695cdf0e10cSrcweir delete [] sString; 696cdf0e10cSrcweir } 697cdf0e10cSrcweir 698cdf0e10cSrcweir return true; 699cdf0e10cSrcweir } 700cdf0e10cSrcweir 701cdf0e10cSrcweir //-------------------------------------------------------------------------- 702cdf0e10cSrcweir HMODULE SetupAppX::LoadMsiLibrary() 703cdf0e10cSrcweir { 704cdf0e10cSrcweir HMODULE hMsi = NULL; 705cdf0e10cSrcweir HKEY hInstKey = NULL; 706cdf0e10cSrcweir 707cdf0e10cSrcweir // find registered location of Msi.dll 708cdf0e10cSrcweir if ( ERROR_SUCCESS == RegOpenKeyEx( HKEY_LOCAL_MACHINE, sInstKey, 0, KEY_READ, &hInstKey ) ) 709cdf0e10cSrcweir { 710cdf0e10cSrcweir long nRet = ERROR_SUCCESS; 711cdf0e10cSrcweir TCHAR *sMsiFolder = new TCHAR[ MAX_PATH + 1 ]; 712cdf0e10cSrcweir DWORD dwMsiFolderSize = MAX_PATH + 1; 713cdf0e10cSrcweir DWORD dwType = 0; 714cdf0e10cSrcweir 715cdf0e10cSrcweir if ( ERROR_MORE_DATA == ( nRet = RegQueryValueEx( hInstKey, sInstLocValue, NULL, 716cdf0e10cSrcweir &dwType, (BYTE*)sMsiFolder, &dwMsiFolderSize ) ) ) 717cdf0e10cSrcweir { 718cdf0e10cSrcweir // try again with larger buffer 719cdf0e10cSrcweir delete [] sMsiFolder; 720cdf0e10cSrcweir sMsiFolder = new TCHAR[ dwMsiFolderSize ]; 721cdf0e10cSrcweir 722cdf0e10cSrcweir nRet = RegQueryValueEx( hInstKey, sInstLocValue, NULL, &dwType, 723cdf0e10cSrcweir (BYTE*)sMsiFolder, &dwMsiFolderSize ); 724cdf0e10cSrcweir } 725cdf0e10cSrcweir 726cdf0e10cSrcweir if ( ERROR_SUCCESS == nRet && dwType == REG_SZ && dwMsiFolderSize > 0 ) 727cdf0e10cSrcweir { 728cdf0e10cSrcweir // load Msi.dll from registered location 729cdf0e10cSrcweir int nLength = lstrlen( sMsiDll ) + dwMsiFolderSize + 1; // use StringCchLength ? 730cdf0e10cSrcweir TCHAR *pMsiLocation = new TCHAR[ nLength ]; 731cdf0e10cSrcweir 732cdf0e10cSrcweir if ( SUCCEEDED( StringCchCopy( pMsiLocation, nLength, sMsiFolder ) ) && 733cdf0e10cSrcweir SUCCEEDED( StringCchCat( pMsiLocation, nLength, sMsiDll ) ) ) 734cdf0e10cSrcweir { 735cdf0e10cSrcweir hMsi = LoadLibrary( pMsiLocation ); 736cdf0e10cSrcweir } 737cdf0e10cSrcweir } 738cdf0e10cSrcweir } 739cdf0e10cSrcweir 740cdf0e10cSrcweir if ( !hMsi ) // use the default location 741cdf0e10cSrcweir { 742cdf0e10cSrcweir hMsi = LoadLibrary( sMsiDll ); 743cdf0e10cSrcweir } 744cdf0e10cSrcweir 745cdf0e10cSrcweir return hMsi; 746cdf0e10cSrcweir } 747cdf0e10cSrcweir 748cdf0e10cSrcweir //-------------------------------------------------------------------------- 749cdf0e10cSrcweir LPCTSTR SetupAppX::GetPathToMSI() 750cdf0e10cSrcweir { 751cdf0e10cSrcweir LPTSTR sMsiPath = NULL; 752cdf0e10cSrcweir HKEY hInstKey = NULL; 753cdf0e10cSrcweir TCHAR *sMsiFolder = new TCHAR[ MAX_PATH + 1 ]; 754cdf0e10cSrcweir DWORD nMsiFolderSize = MAX_PATH + 1; 755cdf0e10cSrcweir 756cdf0e10cSrcweir sMsiFolder[0] = '\0'; 757cdf0e10cSrcweir 758cdf0e10cSrcweir // find registered location of Msi.dll 759cdf0e10cSrcweir if ( ERROR_SUCCESS == RegOpenKeyEx( HKEY_LOCAL_MACHINE, sInstKey, 0, KEY_READ, &hInstKey ) ) 760cdf0e10cSrcweir { 761cdf0e10cSrcweir LONG nRet = ERROR_SUCCESS; 762cdf0e10cSrcweir DWORD dwType = 0; 763cdf0e10cSrcweir 764cdf0e10cSrcweir if ( ERROR_MORE_DATA == ( nRet = RegQueryValueEx( hInstKey, sInstLocValue, NULL, 765cdf0e10cSrcweir &dwType, (BYTE*)sMsiFolder, &nMsiFolderSize ) ) ) 766cdf0e10cSrcweir { 767cdf0e10cSrcweir // try again with larger buffer 768cdf0e10cSrcweir delete [] sMsiFolder; 769cdf0e10cSrcweir sMsiFolder = new TCHAR[ nMsiFolderSize ]; 770cdf0e10cSrcweir 771cdf0e10cSrcweir nRet = RegQueryValueEx( hInstKey, sInstLocValue, NULL, &dwType, 772cdf0e10cSrcweir (BYTE*)sMsiFolder, &nMsiFolderSize ); 773cdf0e10cSrcweir } 774cdf0e10cSrcweir 775cdf0e10cSrcweir if ( ERROR_SUCCESS != nRet || dwType != REG_SZ || nMsiFolderSize == 0 ) 776cdf0e10cSrcweir sMsiFolder[0] = '\0'; 777cdf0e10cSrcweir } 778cdf0e10cSrcweir 779cdf0e10cSrcweir if ( sMsiFolder[0] == '\0' ) // use the default location 780cdf0e10cSrcweir { 781cdf0e10cSrcweir Log( TEXT( " Could not find path to msiexec.exe in registry" ) ); 782cdf0e10cSrcweir 783cdf0e10cSrcweir DWORD nRet = WIN::GetSystemDirectory( sMsiFolder, nMsiFolderSize ); 784cdf0e10cSrcweir if ( nRet > nMsiFolderSize ) 785cdf0e10cSrcweir { 786cdf0e10cSrcweir delete [] sMsiFolder; 787cdf0e10cSrcweir sMsiFolder = new TCHAR[ nRet ]; 788cdf0e10cSrcweir nMsiFolderSize = nRet; 789cdf0e10cSrcweir 790cdf0e10cSrcweir nRet = WIN::GetSystemDirectory( sMsiFolder, nMsiFolderSize ); 791cdf0e10cSrcweir } 792cdf0e10cSrcweir if ( 0 == nRet ) 793cdf0e10cSrcweir { 794cdf0e10cSrcweir sMsiFolder[0] = '\0'; 795cdf0e10cSrcweir SetError( WIN::GetLastError() ); 796cdf0e10cSrcweir } 797cdf0e10cSrcweir nMsiFolderSize = nRet; 798cdf0e10cSrcweir } 799cdf0e10cSrcweir 800cdf0e10cSrcweir if ( sMsiFolder[0] != '\0' ) 801cdf0e10cSrcweir { 802cdf0e10cSrcweir int nLength = lstrlen( sMsiExe ) + lstrlen( sMsiFolder ) + 1; 803cdf0e10cSrcweir sMsiPath = new TCHAR[ nLength ]; 804cdf0e10cSrcweir 805cdf0e10cSrcweir if ( FAILED( StringCchCopy( sMsiPath, nLength, sMsiFolder ) ) || 806cdf0e10cSrcweir FAILED( StringCchCat( sMsiPath, nLength, sMsiExe ) ) ) 807cdf0e10cSrcweir { 808cdf0e10cSrcweir delete [] sMsiPath; 809cdf0e10cSrcweir sMsiPath = NULL; 810cdf0e10cSrcweir } 811cdf0e10cSrcweir } 812cdf0e10cSrcweir 813cdf0e10cSrcweir if ( ! sMsiPath ) 814cdf0e10cSrcweir Log( TEXT( "ERROR: Can't build path to msiexec.exe!" ) ); 815cdf0e10cSrcweir 816cdf0e10cSrcweir return sMsiPath; 817cdf0e10cSrcweir } 818cdf0e10cSrcweir 819cdf0e10cSrcweir //-------------------------------------------------------------------------- 820cdf0e10cSrcweir boolean SetupAppX::LaunchInstaller( LPCTSTR pParam ) 821cdf0e10cSrcweir { 822cdf0e10cSrcweir LPCTSTR sMsiPath = GetPathToMSI(); 823cdf0e10cSrcweir 824cdf0e10cSrcweir if ( !sMsiPath ) 825cdf0e10cSrcweir { 826cdf0e10cSrcweir Log( TEXT( "ERROR: msiexec not found!" ) ); 827cdf0e10cSrcweir SetError( ERROR_FILE_NOT_FOUND ); 828cdf0e10cSrcweir return false; 829cdf0e10cSrcweir } 830cdf0e10cSrcweir 831cdf0e10cSrcweir STARTUPINFO aSUI; 832cdf0e10cSrcweir PROCESS_INFORMATION aPI; 833cdf0e10cSrcweir 834cdf0e10cSrcweir Log( TEXT( " Will install using <%s>\r\n" ), sMsiPath ); 835cdf0e10cSrcweir Log( TEXT( " Prameters are: %s\r\n" ), pParam ); 836cdf0e10cSrcweir 837cdf0e10cSrcweir OutputDebugStringFormat( TEXT( " Will install using <%s>\r\n" ), sMsiPath ); 838cdf0e10cSrcweir OutputDebugStringFormat( TEXT( " Prameters are: %s\r\n" ), pParam ); 839cdf0e10cSrcweir 840cdf0e10cSrcweir ZeroMemory( (void*)&aPI, sizeof( PROCESS_INFORMATION ) ); 841cdf0e10cSrcweir ZeroMemory( (void*)&aSUI, sizeof( STARTUPINFO ) ); 842cdf0e10cSrcweir 843cdf0e10cSrcweir aSUI.cb = sizeof(STARTUPINFO); 844cdf0e10cSrcweir aSUI.dwFlags = STARTF_USESHOWWINDOW; 845cdf0e10cSrcweir aSUI.wShowWindow = SW_SHOW; 846cdf0e10cSrcweir 847cdf0e10cSrcweir DWORD nCmdLineLength = lstrlen( sMsiPath ) + lstrlen( pParam ) + 2; 848cdf0e10cSrcweir TCHAR *sCmdLine = new TCHAR[ nCmdLineLength ]; 849cdf0e10cSrcweir 850cdf0e10cSrcweir if ( FAILED( StringCchCopy( sCmdLine, nCmdLineLength, sMsiPath ) ) || 851cdf0e10cSrcweir FAILED( StringCchCat( sCmdLine, nCmdLineLength, TEXT( " " ) ) ) || 852cdf0e10cSrcweir FAILED( StringCchCat( sCmdLine, nCmdLineLength, pParam ) ) ) 853cdf0e10cSrcweir { 854cdf0e10cSrcweir delete [] sCmdLine; 855cdf0e10cSrcweir SetError( ERROR_INSTALL_FAILURE ); 856cdf0e10cSrcweir return false; 857cdf0e10cSrcweir } 858cdf0e10cSrcweir 859cdf0e10cSrcweir if ( !WIN::CreateProcess( NULL, sCmdLine, NULL, NULL, FALSE, 860cdf0e10cSrcweir CREATE_DEFAULT_ERROR_MODE, NULL, NULL, 861cdf0e10cSrcweir &aSUI, &aPI ) ) 862cdf0e10cSrcweir { 863cdf0e10cSrcweir Log( TEXT( "ERROR: Could not create process %s.\r\n" ), sCmdLine ); 864cdf0e10cSrcweir SetError( WIN::GetLastError() ); 865cdf0e10cSrcweir delete [] sCmdLine; 866cdf0e10cSrcweir return false; 867cdf0e10cSrcweir } 868cdf0e10cSrcweir 869cdf0e10cSrcweir DWORD nResult = WaitForProcess( aPI.hProcess ); 870cdf0e10cSrcweir bool bRet = true; 871cdf0e10cSrcweir 872cdf0e10cSrcweir if( ERROR_SUCCESS != nResult ) 873cdf0e10cSrcweir { 874cdf0e10cSrcweir Log( TEXT( "ERROR: While waiting for %s.\r\n" ), sCmdLine ); 875cdf0e10cSrcweir SetError( nResult ); 876cdf0e10cSrcweir bRet = false; 877cdf0e10cSrcweir } 878cdf0e10cSrcweir else 879cdf0e10cSrcweir { 880cdf0e10cSrcweir GetExitCodeProcess( aPI.hProcess, &nResult ); 881cdf0e10cSrcweir SetError( nResult ); 882cdf0e10cSrcweir 883cdf0e10cSrcweir if ( nResult != ERROR_SUCCESS ) 884cdf0e10cSrcweir { 885cdf0e10cSrcweir TCHAR sBuf[80]; 886cdf0e10cSrcweir StringCchPrintf( sBuf, 80, TEXT("Warning: msiexec returned %u.\r\n"), nResult ); 887cdf0e10cSrcweir Log( sBuf ); 888cdf0e10cSrcweir } 889cdf0e10cSrcweir else 890cdf0e10cSrcweir Log( TEXT( " Installation completed successfully.\r\n" ) ); 891cdf0e10cSrcweir } 892cdf0e10cSrcweir 893cdf0e10cSrcweir CloseHandle( aPI.hProcess ); 894cdf0e10cSrcweir 895cdf0e10cSrcweir delete [] sCmdLine; 896cdf0e10cSrcweir 897cdf0e10cSrcweir return bRet; 898cdf0e10cSrcweir } 899cdf0e10cSrcweir 900cdf0e10cSrcweir //-------------------------------------------------------------------------- 901cdf0e10cSrcweir boolean SetupAppX::Install( long nLanguage ) 902cdf0e10cSrcweir { 903cdf0e10cSrcweir LPTSTR pTransform = NULL; 904cdf0e10cSrcweir 905cdf0e10cSrcweir if ( nLanguage ) // look for transformation 906cdf0e10cSrcweir { 907cdf0e10cSrcweir for ( int i = 0; i < m_nLanguageCount; i++ ) 908cdf0e10cSrcweir { 909cdf0e10cSrcweir if ( m_ppLanguageList[i]->m_nLanguageID == nLanguage ) 910cdf0e10cSrcweir { 911cdf0e10cSrcweir if ( m_ppLanguageList[i]->m_pTransform ) 912cdf0e10cSrcweir { 913cdf0e10cSrcweir if ( !GetPathToFile( m_ppLanguageList[i]->m_pTransform, 914cdf0e10cSrcweir &pTransform ) ) 915cdf0e10cSrcweir { 916cdf0e10cSrcweir Log( TEXT( "ERROR: Could not find transform <%s\r\n" ), m_ppLanguageList[i]->m_pTransform ); 917cdf0e10cSrcweir return false; 918cdf0e10cSrcweir } 919cdf0e10cSrcweir } 920cdf0e10cSrcweir break; 921cdf0e10cSrcweir } 922cdf0e10cSrcweir } 923cdf0e10cSrcweir } 924cdf0e10cSrcweir 925cdf0e10cSrcweir TCHAR *pDataBasePath = NULL; 926cdf0e10cSrcweir 927cdf0e10cSrcweir if ( ! GetPathToFile( m_pDatabase, &pDataBasePath ) ) 928cdf0e10cSrcweir { 929cdf0e10cSrcweir Log( TEXT( "ERROR: Could not find database <%s\r\n" ), m_pDatabase ); 930cdf0e10cSrcweir SetError( ERROR_INSTALL_SOURCE_ABSENT ); 931cdf0e10cSrcweir return false; 932cdf0e10cSrcweir } 933cdf0e10cSrcweir 934cdf0e10cSrcweir // we will always use the parameter setup used 935cdf0e10cSrcweir int nParLen = lstrlen( PARAM_SETUP_USED ); 936cdf0e10cSrcweir 937cdf0e10cSrcweir if ( m_bRegNoMsoTypes ) 938cdf0e10cSrcweir nParLen += lstrlen( PARAM_REG_NO_MSO_TYPES ); 939cdf0e10cSrcweir else if ( m_bRegAllMsoTypes ) 940cdf0e10cSrcweir nParLen += lstrlen( PARAM_REG_ALL_MSO_TYPES ); 941cdf0e10cSrcweir 942cdf0e10cSrcweir if ( m_pAdvertise ) 943cdf0e10cSrcweir nParLen += lstrlen( m_pAdvertise ) + 1; // one for the space 944cdf0e10cSrcweir else if ( m_bIsMinorUpgrade ) 945cdf0e10cSrcweir nParLen += lstrlen( PARAM_MINOR_UPGRADE ); 946cdf0e10cSrcweir else 947cdf0e10cSrcweir nParLen += lstrlen( PARAM_PACKAGE ); 948cdf0e10cSrcweir 949cdf0e10cSrcweir nParLen += lstrlen( pDataBasePath ) + 3; // two quotes, one null 950cdf0e10cSrcweir 951cdf0e10cSrcweir if ( NeedReboot() ) 952cdf0e10cSrcweir nParLen += lstrlen( PARAM_REBOOT ); 953cdf0e10cSrcweir 954cdf0e10cSrcweir if ( m_pPatchFiles ) 955cdf0e10cSrcweir { 956cdf0e10cSrcweir nParLen += lstrlen( PARAM_PATCH ); 957cdf0e10cSrcweir nParLen += lstrlen( m_pPatchFiles ); 958cdf0e10cSrcweir } 959cdf0e10cSrcweir 960cdf0e10cSrcweir if ( pTransform ) 961cdf0e10cSrcweir { 962cdf0e10cSrcweir nParLen += lstrlen( PARAM_TRANSFORM ); 963cdf0e10cSrcweir nParLen += lstrlen( pTransform ) + 2; // two quotes 964cdf0e10cSrcweir } 965cdf0e10cSrcweir 966cdf0e10cSrcweir if ( m_pCmdLine ) 967cdf0e10cSrcweir nParLen += lstrlen( m_pCmdLine ) + 1; // one for the space; 968cdf0e10cSrcweir 969cdf0e10cSrcweir TCHAR *pParams = new TCHAR[ nParLen ]; 970cdf0e10cSrcweir 971cdf0e10cSrcweir StringCchCopy( pParams, nParLen, PARAM_SETUP_USED ); 972cdf0e10cSrcweir 973cdf0e10cSrcweir if ( m_bRegNoMsoTypes ) 974cdf0e10cSrcweir StringCchCat( pParams, nParLen, PARAM_REG_NO_MSO_TYPES ); 975cdf0e10cSrcweir else if ( m_bRegAllMsoTypes ) 976cdf0e10cSrcweir StringCchCat( pParams, nParLen, PARAM_REG_ALL_MSO_TYPES ); 977cdf0e10cSrcweir 978cdf0e10cSrcweir if ( m_pAdvertise ) 979cdf0e10cSrcweir StringCchCat( pParams, nParLen, m_pAdvertise ); 980cdf0e10cSrcweir else if ( IsAdminInstall() ) 981cdf0e10cSrcweir StringCchCat( pParams, nParLen, PARAM_ADMIN ); 982cdf0e10cSrcweir else if ( m_bIsMinorUpgrade ) 983cdf0e10cSrcweir StringCchCat( pParams, nParLen, PARAM_MINOR_UPGRADE ); 984cdf0e10cSrcweir else 985cdf0e10cSrcweir StringCchCat( pParams, nParLen, PARAM_PACKAGE ); 986cdf0e10cSrcweir 987cdf0e10cSrcweir StringCchCat( pParams, nParLen, TEXT( "\"" ) ); 988cdf0e10cSrcweir StringCchCat( pParams, nParLen, pDataBasePath ); 989cdf0e10cSrcweir StringCchCat( pParams, nParLen, TEXT( "\"" ) ); 990cdf0e10cSrcweir 991cdf0e10cSrcweir if ( NeedReboot() ) 992cdf0e10cSrcweir StringCchCat( pParams, nParLen, PARAM_REBOOT ); 993cdf0e10cSrcweir 994cdf0e10cSrcweir if ( m_pPatchFiles ) 995cdf0e10cSrcweir { 996cdf0e10cSrcweir StringCchCat( pParams, nParLen, PARAM_PATCH ); 997cdf0e10cSrcweir StringCchCat( pParams, nParLen, m_pPatchFiles ); 998cdf0e10cSrcweir } 999cdf0e10cSrcweir 1000cdf0e10cSrcweir if ( pTransform ) 1001cdf0e10cSrcweir { 1002cdf0e10cSrcweir StringCchCat( pParams, nParLen, PARAM_TRANSFORM ); 1003cdf0e10cSrcweir StringCchCat( pParams, nParLen, TEXT( "\"" ) ); 1004cdf0e10cSrcweir StringCchCat( pParams, nParLen, pTransform ); 1005cdf0e10cSrcweir StringCchCat( pParams, nParLen, TEXT( "\"" ) ); 1006cdf0e10cSrcweir } 1007cdf0e10cSrcweir 1008cdf0e10cSrcweir if ( m_pCmdLine ) 1009cdf0e10cSrcweir { 1010cdf0e10cSrcweir StringCchCat( pParams, nParLen, TEXT( " " ) ); 1011cdf0e10cSrcweir StringCchCat( pParams, nParLen, m_pCmdLine ); 1012cdf0e10cSrcweir } 1013cdf0e10cSrcweir 1014cdf0e10cSrcweir return LaunchInstaller( pParams ); 1015cdf0e10cSrcweir } 1016cdf0e10cSrcweir 1017cdf0e10cSrcweir //-------------------------------------------------------------------------- 1018cdf0e10cSrcweir UINT SetupAppX::GetError() const 1019cdf0e10cSrcweir { 1020cdf0e10cSrcweir UINT nErr = 0; 1021cdf0e10cSrcweir 1022cdf0e10cSrcweir if ( m_pMSIErrorCode ) 1023cdf0e10cSrcweir nErr = (UINT) *m_pMSIErrorCode; 1024cdf0e10cSrcweir 1025cdf0e10cSrcweir if ( nErr == 0 ) 1026cdf0e10cSrcweir nErr = m_uiRet; 1027cdf0e10cSrcweir 1028cdf0e10cSrcweir if ( nErr != 0 ) 1029cdf0e10cSrcweir OutputDebugStringFormat( TEXT("Setup will return error (%d).\n"), nErr ); 1030cdf0e10cSrcweir return nErr; 1031cdf0e10cSrcweir } 1032cdf0e10cSrcweir 1033cdf0e10cSrcweir //-------------------------------------------------------------------------- 1034cdf0e10cSrcweir void SetupAppX::DisplayError( UINT nErr ) const 1035cdf0e10cSrcweir { 1036cdf0e10cSrcweir TCHAR sError[ MAX_TEXT_LENGTH ] = {0}; 1037cdf0e10cSrcweir TCHAR sTmp[ MAX_TEXT_LENGTH ] = {0}; 1038cdf0e10cSrcweir 1039cdf0e10cSrcweir UINT nMsgType = MB_OK | MB_ICONERROR; 1040cdf0e10cSrcweir 1041cdf0e10cSrcweir switch ( nErr ) 1042cdf0e10cSrcweir { 1043cdf0e10cSrcweir case ERROR_SUCCESS: break; // 0 1044cdf0e10cSrcweir 1045cdf0e10cSrcweir case ERROR_FILE_NOT_FOUND: // 2 1046cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_FILE_NOT_FOUND, sTmp, MAX_TEXT_LENGTH ); 1047cdf0e10cSrcweir StringCchPrintf( sError, MAX_TEXT_LENGTH, sTmp, m_pErrorText ); 1048cdf0e10cSrcweir break; 1049cdf0e10cSrcweir case ERROR_INVALID_DATA: // 13 1050cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_INVALID_PROFILE, sError, MAX_TEXT_LENGTH ); 1051cdf0e10cSrcweir break; 1052cdf0e10cSrcweir case ERROR_OUTOFMEMORY: WIN::LoadString( m_hInst, IDS_OUTOFMEM, sError, MAX_TEXT_LENGTH ); 1053cdf0e10cSrcweir break; 1054cdf0e10cSrcweir case ERROR_INSTALL_USEREXIT: 1055cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_USER_CANCELLED, sError, MAX_TEXT_LENGTH ); 1056cdf0e10cSrcweir break; 1057cdf0e10cSrcweir case ERROR_INSTALL_ALREADY_RUNNING: // 1618 1058cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_ALREADY_RUNNING, sError, MAX_TEXT_LENGTH ); 1059cdf0e10cSrcweir break; 1060cdf0e10cSrcweir case ERROR_INSTALL_SOURCE_ABSENT: 1061cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_NOMSI, sError, MAX_TEXT_LENGTH ); 1062cdf0e10cSrcweir break; 1063cdf0e10cSrcweir case ERROR_DS_INSUFF_ACCESS_RIGHTS: // 8344 1064cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_REQUIRES_ADMIN_PRIV, sError, MAX_TEXT_LENGTH ); 1065cdf0e10cSrcweir break; 1066cdf0e10cSrcweir case E_ABORT: WIN::LoadString( m_hInst, IDS_UNKNOWN_ERROR, sError, MAX_TEXT_LENGTH ); 1067cdf0e10cSrcweir break; 1068cdf0e10cSrcweir case ERROR_INVALID_PARAMETER: // 87 1069cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_INVALID_PARAM, sTmp, MAX_TEXT_LENGTH ); 1070cdf0e10cSrcweir StringCchPrintf( sError, MAX_TEXT_LENGTH, sTmp, m_pErrorText ); 1071cdf0e10cSrcweir break; 1072cdf0e10cSrcweir 1073cdf0e10cSrcweir case ERROR_SETUP_TO_OLD: // - 3 1074cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_SETUP_TO_OLD, sTmp, MAX_TEXT_LENGTH ); 1075cdf0e10cSrcweir StringCchPrintf( sError, MAX_TEXT_LENGTH, sTmp, m_pReqVersion, m_pErrorText ); 1076cdf0e10cSrcweir break; 1077cdf0e10cSrcweir case ERROR_SETUP_NOT_FOUND: // - 4 1078cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_SETUP_NOT_FOUND, sTmp, MAX_TEXT_LENGTH ); 1079cdf0e10cSrcweir StringCchPrintf( sError, MAX_TEXT_LENGTH, sTmp, m_pReqVersion ); 1080cdf0e10cSrcweir break; 1081cdf0e10cSrcweir case ERROR_SHOW_USAGE: // - 2 1082cdf0e10cSrcweir nMsgType = MB_OK | MB_ICONINFORMATION; 1083cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_USAGE, sError, MAX_TEXT_LENGTH ); 1084cdf0e10cSrcweir break; 1085cdf0e10cSrcweir 1086cdf0e10cSrcweir default: WIN::LoadString( m_hInst, IDS_UNKNOWN_ERROR, sError, MAX_TEXT_LENGTH ); 1087cdf0e10cSrcweir break; 1088cdf0e10cSrcweir } 1089cdf0e10cSrcweir 1090cdf0e10cSrcweir if ( sError[0] ) 1091cdf0e10cSrcweir { 1092cdf0e10cSrcweir if ( !m_bQuiet ) 1093cdf0e10cSrcweir { 1094cdf0e10cSrcweir ConvertNewline( sError ); 1095cdf0e10cSrcweir WIN::MessageBox( NULL, sError, m_pAppTitle, nMsgType ); 1096cdf0e10cSrcweir } 1097cdf0e10cSrcweir 1098cdf0e10cSrcweir Log( TEXT( "ERROR: %s\r\n" ), sError ); 1099cdf0e10cSrcweir } 1100cdf0e10cSrcweir } 1101cdf0e10cSrcweir 1102cdf0e10cSrcweir //-------------------------------------------------------------------------- 1103cdf0e10cSrcweir long SetupAppX::GetLanguageID( long nIndex ) const 1104cdf0e10cSrcweir { 1105cdf0e10cSrcweir if ( nIndex >=0 && nIndex < m_nLanguageCount ) 1106cdf0e10cSrcweir return m_ppLanguageList[ nIndex ]->m_nLanguageID; 1107cdf0e10cSrcweir else 1108cdf0e10cSrcweir return 0; 1109cdf0e10cSrcweir } 1110cdf0e10cSrcweir 1111cdf0e10cSrcweir //-------------------------------------------------------------------------- 1112cdf0e10cSrcweir void SetupAppX::GetLanguageName( long nLanguage, LPTSTR sName ) const 1113cdf0e10cSrcweir { 1114cdf0e10cSrcweir switch ( nLanguage ) 1115cdf0e10cSrcweir { 1116cdf0e10cSrcweir case 1028: WIN::LoadString( m_hInst, IDS_LANGUAGE_ZH_TW, sName, MAX_LANGUAGE_LEN ); break; 1117cdf0e10cSrcweir case 1029: WIN::LoadString( m_hInst, IDS_LANGUAGE_CS, sName, MAX_LANGUAGE_LEN ); break; 1118cdf0e10cSrcweir case 1030: WIN::LoadString( m_hInst, IDS_LANGUAGE_DA, sName, MAX_LANGUAGE_LEN ); break; 1119cdf0e10cSrcweir case 1031: WIN::LoadString( m_hInst, IDS_LANGUAGE_DE_DE, sName, MAX_LANGUAGE_LEN ); break; 1120cdf0e10cSrcweir case 1032: WIN::LoadString( m_hInst, IDS_LANGUAGE_EL, sName, MAX_LANGUAGE_LEN ); break; 1121cdf0e10cSrcweir case 1033: WIN::LoadString( m_hInst, IDS_LANGUAGE_EN_US, sName, MAX_LANGUAGE_LEN ); break; 1122cdf0e10cSrcweir case 1034: WIN::LoadString( m_hInst, IDS_LANGUAGE_ES, sName, MAX_LANGUAGE_LEN ); break; 1123cdf0e10cSrcweir case 1035: WIN::LoadString( m_hInst, IDS_LANGUAGE_FI, sName, MAX_LANGUAGE_LEN ); break; 1124cdf0e10cSrcweir case 1036: WIN::LoadString( m_hInst, IDS_LANGUAGE_FR_FR, sName, MAX_LANGUAGE_LEN ); break; 1125cdf0e10cSrcweir case 1037: WIN::LoadString( m_hInst, IDS_LANGUAGE_HE, sName, MAX_LANGUAGE_LEN ); break; 1126cdf0e10cSrcweir case 1038: WIN::LoadString( m_hInst, IDS_LANGUAGE_HU, sName, MAX_LANGUAGE_LEN ); break; 1127cdf0e10cSrcweir case 1040: WIN::LoadString( m_hInst, IDS_LANGUAGE_IT_IT, sName, MAX_LANGUAGE_LEN ); break; 1128cdf0e10cSrcweir case 1041: WIN::LoadString( m_hInst, IDS_LANGUAGE_JA, sName, MAX_LANGUAGE_LEN ); break; 1129cdf0e10cSrcweir case 1042: WIN::LoadString( m_hInst, IDS_LANGUAGE_KO, sName, MAX_LANGUAGE_LEN ); break; 1130cdf0e10cSrcweir case 1043: WIN::LoadString( m_hInst, IDS_LANGUAGE_NL_NL, sName, MAX_LANGUAGE_LEN ); break; 1131cdf0e10cSrcweir case 1044: WIN::LoadString( m_hInst, IDS_LANGUAGE_NO_NO, sName, MAX_LANGUAGE_LEN ); break; 1132cdf0e10cSrcweir case 1045: WIN::LoadString( m_hInst, IDS_LANGUAGE_PL, sName, MAX_LANGUAGE_LEN ); break; 1133cdf0e10cSrcweir case 1046: WIN::LoadString( m_hInst, IDS_LANGUAGE_PT_BR, sName, MAX_LANGUAGE_LEN ); break; 1134cdf0e10cSrcweir case 1049: WIN::LoadString( m_hInst, IDS_LANGUAGE_RU, sName, MAX_LANGUAGE_LEN ); break; 1135cdf0e10cSrcweir case 1051: WIN::LoadString( m_hInst, IDS_LANGUAGE_SK, sName, MAX_LANGUAGE_LEN ); break; 1136cdf0e10cSrcweir case 1053: WIN::LoadString( m_hInst, IDS_LANGUAGE_SV_SE, sName, MAX_LANGUAGE_LEN ); break; 1137cdf0e10cSrcweir case 1054: WIN::LoadString( m_hInst, IDS_LANGUAGE_TH, sName, MAX_LANGUAGE_LEN ); break; 1138cdf0e10cSrcweir case 1055: WIN::LoadString( m_hInst, IDS_LANGUAGE_TR, sName, MAX_LANGUAGE_LEN ); break; 1139cdf0e10cSrcweir case 1061: WIN::LoadString( m_hInst, IDS_LANGUAGE_ET, sName, MAX_LANGUAGE_LEN ); break; 1140cdf0e10cSrcweir case 2052: WIN::LoadString( m_hInst, IDS_LANGUAGE_ZH_CN, sName, MAX_LANGUAGE_LEN ); break; 1141cdf0e10cSrcweir case 2070: WIN::LoadString( m_hInst, IDS_LANGUAGE_PT_PT, sName, MAX_LANGUAGE_LEN ); break; 1142cdf0e10cSrcweir 1143cdf0e10cSrcweir default: 1144cdf0e10cSrcweir { 1145cdf0e10cSrcweir TCHAR sTmp[ MAX_LANGUAGE_LEN ] = {0}; 1146cdf0e10cSrcweir 1147cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_UNKNOWN_LANG, sTmp, MAX_LANGUAGE_LEN ); 1148cdf0e10cSrcweir StringCchPrintf( sName, MAX_LANGUAGE_LEN, sTmp, nLanguage ); 1149cdf0e10cSrcweir } 1150cdf0e10cSrcweir } 1151cdf0e10cSrcweir } 1152cdf0e10cSrcweir 1153cdf0e10cSrcweir //-------------------------------------------------------------------------- 1154cdf0e10cSrcweir boolean SetupAppX::CheckVersion() 1155cdf0e10cSrcweir { 1156cdf0e10cSrcweir boolean bRet = false; 1157cdf0e10cSrcweir HMODULE hMsi = LoadMsiLibrary(); 1158cdf0e10cSrcweir 1159cdf0e10cSrcweir Log( TEXT( " Looking for installed MSI with version >= %s\r\n" ), m_pReqVersion ); 1160cdf0e10cSrcweir 1161cdf0e10cSrcweir if ( !hMsi ) 1162cdf0e10cSrcweir { 1163cdf0e10cSrcweir Log( TEXT( "Error: No MSI found!\r\n" ) ); 1164cdf0e10cSrcweir SetError( (UINT) ERROR_SETUP_NOT_FOUND ); 1165cdf0e10cSrcweir } 1166cdf0e10cSrcweir else 1167cdf0e10cSrcweir { 1168cdf0e10cSrcweir PFnDllGetVersion pDllGetVersion = (PFnDllGetVersion) GetProcAddress( hMsi, MSIAPI_DllGetVersion ); 1169cdf0e10cSrcweir 1170cdf0e10cSrcweir if ( pDllGetVersion ) 1171cdf0e10cSrcweir { 1172cdf0e10cSrcweir DLLVERSIONINFO aInfo; 1173cdf0e10cSrcweir 1174cdf0e10cSrcweir aInfo.cbSize = sizeof( DLLVERSIONINFO ); 1175cdf0e10cSrcweir if ( NOERROR == pDllGetVersion( &aInfo ) ) 1176cdf0e10cSrcweir { 1177cdf0e10cSrcweir TCHAR pMsiVersion[ VERSION_SIZE ]; 1178cdf0e10cSrcweir StringCchPrintf( pMsiVersion, VERSION_SIZE, TEXT("%d.%d.%4d"), 1179cdf0e10cSrcweir aInfo.dwMajorVersion, 1180cdf0e10cSrcweir aInfo.dwMinorVersion, 1181cdf0e10cSrcweir aInfo.dwBuildNumber ); 1182cdf0e10cSrcweir if ( _tcsncmp( pMsiVersion, m_pReqVersion, _tcslen( pMsiVersion ) ) < 0 ) 1183cdf0e10cSrcweir { 1184cdf0e10cSrcweir StringCchCopy( m_pErrorText, MAX_TEXT_LENGTH, pMsiVersion ); 1185cdf0e10cSrcweir SetError( (UINT) ERROR_SETUP_TO_OLD ); 1186cdf0e10cSrcweir Log( TEXT( "Warning: Old MSI version found <%s>, update needed!\r\n" ), pMsiVersion ); 1187cdf0e10cSrcweir } 1188cdf0e10cSrcweir else 1189cdf0e10cSrcweir { 1190cdf0e10cSrcweir Log( TEXT( " Found MSI version <%s>, no update needed\r\n" ), pMsiVersion ); 1191cdf0e10cSrcweir bRet = true; 1192cdf0e10cSrcweir } 1193cdf0e10cSrcweir if ( aInfo.dwMajorVersion >= 3 ) 1194cdf0e10cSrcweir m_bSupportsPatch = true; 1195cdf0e10cSrcweir else 1196cdf0e10cSrcweir Log( TEXT("Warning: Patching not supported! MSI-Version <%s>\r\n"), pMsiVersion ); 1197cdf0e10cSrcweir } 1198cdf0e10cSrcweir } 1199cdf0e10cSrcweir 1200cdf0e10cSrcweir FreeLibrary( hMsi ); 1201cdf0e10cSrcweir } 1202cdf0e10cSrcweir 1203cdf0e10cSrcweir return bRet; 1204cdf0e10cSrcweir } 1205cdf0e10cSrcweir 1206cdf0e10cSrcweir //-------------------------------------------------------------------------- 1207cdf0e10cSrcweir boolean SetupAppX::CheckForUpgrade() 1208cdf0e10cSrcweir { 1209cdf0e10cSrcweir // When we have patch files we will never try an Minor upgrade 1210cdf0e10cSrcweir if ( m_pPatchFiles ) return true; 1211cdf0e10cSrcweir 1212cdf0e10cSrcweir if ( !m_pUpgradeKey || ( _tcslen( m_pUpgradeKey ) == 0 ) ) 1213cdf0e10cSrcweir { 1214cdf0e10cSrcweir Log( TEXT( " No Upgrade Key Found -> continue with standard installation!\r\n" ) ); 1215cdf0e10cSrcweir return true; 1216cdf0e10cSrcweir } 1217cdf0e10cSrcweir 1218cdf0e10cSrcweir HKEY hInstKey = NULL; 1219cdf0e10cSrcweir 1220cdf0e10cSrcweir if ( ERROR_SUCCESS == RegOpenKeyEx( HKEY_LOCAL_MACHINE, m_pUpgradeKey, 0, KEY_READ, &hInstKey ) ) 1221cdf0e10cSrcweir { 1222cdf0e10cSrcweir Log( TEXT( " Found Upgrade Key in Registry (HKLM) -> will try minor upgrade!\r\n" ) ); 1223cdf0e10cSrcweir m_bIsMinorUpgrade = true; 1224cdf0e10cSrcweir } 1225cdf0e10cSrcweir else if ( ERROR_SUCCESS == RegOpenKeyEx( HKEY_CURRENT_USER, m_pUpgradeKey, 0, KEY_READ, &hInstKey ) ) 1226cdf0e10cSrcweir { 1227cdf0e10cSrcweir Log( TEXT( " Found Upgrade Key in Registry (HKCU) -> will try minor upgrade!\r\n" ) ); 1228cdf0e10cSrcweir m_bIsMinorUpgrade = true; 1229cdf0e10cSrcweir } 1230cdf0e10cSrcweir else 1231cdf0e10cSrcweir { 1232cdf0e10cSrcweir Log( TEXT( " Didn't Find Upgrade Key in Registry -> continue with standard installation!\r\n" ) ); 1233cdf0e10cSrcweir return true; 1234cdf0e10cSrcweir } 1235cdf0e10cSrcweir 1236cdf0e10cSrcweir if ( m_pProductVersion && ( _tcslen( m_pProductVersion ) > 0 ) ) 1237cdf0e10cSrcweir { 1238cdf0e10cSrcweir TCHAR *sProductVersion = new TCHAR[ MAX_PATH + 1 ]; 1239cdf0e10cSrcweir DWORD nSize = MAX_PATH + 1; 1240cdf0e10cSrcweir 1241cdf0e10cSrcweir sProductVersion[0] = '\0'; 1242cdf0e10cSrcweir 1243cdf0e10cSrcweir // get product version 1244cdf0e10cSrcweir if ( ERROR_SUCCESS == RegQueryValueEx( hInstKey, PRODUCT_VERSION, NULL, NULL, (LPBYTE)sProductVersion, &nSize ) ) 1245cdf0e10cSrcweir { 1246cdf0e10cSrcweir if ( lstrcmpi( sProductVersion, m_pProductVersion ) == 0 ) 1247cdf0e10cSrcweir { 1248cdf0e10cSrcweir Log( TEXT( " Same Product Version already installed, no minor upgrade!\r\n" ) ); 1249cdf0e10cSrcweir m_bIsMinorUpgrade = false; 1250cdf0e10cSrcweir } 1251cdf0e10cSrcweir } 1252cdf0e10cSrcweir 1253cdf0e10cSrcweir delete [] sProductVersion; 1254cdf0e10cSrcweir } 1255cdf0e10cSrcweir 1256cdf0e10cSrcweir return true; 1257cdf0e10cSrcweir } 1258cdf0e10cSrcweir 1259cdf0e10cSrcweir //-------------------------------------------------------------------------- 1260cdf0e10cSrcweir boolean SetupAppX::IsTerminalServerInstalled() const 1261cdf0e10cSrcweir { 1262cdf0e10cSrcweir boolean bIsTerminalServer = false; 1263cdf0e10cSrcweir 1264cdf0e10cSrcweir const TCHAR sSearchStr[] = TEXT("Terminal Server"); 1265cdf0e10cSrcweir const TCHAR sKey[] = TEXT("System\\CurrentControlSet\\Control\\ProductOptions"); 1266cdf0e10cSrcweir const TCHAR sValue[] = TEXT("ProductSuite"); 1267cdf0e10cSrcweir 1268cdf0e10cSrcweir DWORD dwSize = 0; 1269cdf0e10cSrcweir HKEY hKey = 0; 1270cdf0e10cSrcweir DWORD dwType = 0; 1271cdf0e10cSrcweir 1272cdf0e10cSrcweir if ( ERROR_SUCCESS == RegOpenKeyEx( HKEY_LOCAL_MACHINE, sKey, 0, KEY_READ, &hKey ) && 1273cdf0e10cSrcweir ERROR_SUCCESS == RegQueryValueEx( hKey, sValue, NULL, &dwType, NULL, &dwSize ) && 1274cdf0e10cSrcweir dwSize > 0 && 1275cdf0e10cSrcweir REG_MULTI_SZ == dwType ) 1276cdf0e10cSrcweir { 1277cdf0e10cSrcweir TCHAR* sSuiteList = new TCHAR[ (dwSize*sizeof(byte)/sizeof(TCHAR)) + 1 ]; 1278cdf0e10cSrcweir 1279cdf0e10cSrcweir ZeroMemory(sSuiteList, dwSize); 1280cdf0e10cSrcweir 1281cdf0e10cSrcweir if ( ERROR_SUCCESS == RegQueryValueEx( hKey, sValue, NULL, &dwType, (LPBYTE)sSuiteList, &dwSize) ) 1282cdf0e10cSrcweir { 1283cdf0e10cSrcweir DWORD nMulti = 0; 1284cdf0e10cSrcweir DWORD nSrch = lstrlen( sSearchStr ); 1285cdf0e10cSrcweir const TCHAR *sSubString = sSuiteList; 1286cdf0e10cSrcweir 1287cdf0e10cSrcweir while (*sSubString) 1288cdf0e10cSrcweir { 1289cdf0e10cSrcweir nMulti = lstrlen( sSubString ); 1290cdf0e10cSrcweir if ( nMulti == nSrch && 0 == lstrcmp( sSearchStr, sSubString ) ) 1291cdf0e10cSrcweir { 1292cdf0e10cSrcweir bIsTerminalServer = true; 1293cdf0e10cSrcweir break; 1294cdf0e10cSrcweir } 1295cdf0e10cSrcweir 1296cdf0e10cSrcweir sSubString += (nMulti + 1); 1297cdf0e10cSrcweir } 1298cdf0e10cSrcweir } 1299cdf0e10cSrcweir delete [] sSuiteList; 1300cdf0e10cSrcweir } 1301cdf0e10cSrcweir 1302cdf0e10cSrcweir if ( hKey ) 1303cdf0e10cSrcweir RegCloseKey( hKey ); 1304cdf0e10cSrcweir 1305cdf0e10cSrcweir return bIsTerminalServer; 1306cdf0e10cSrcweir } 1307cdf0e10cSrcweir 1308cdf0e10cSrcweir //-------------------------------------------------------------------------- 1309cdf0e10cSrcweir boolean SetupAppX::AlreadyRunning() const 1310cdf0e10cSrcweir { 1311cdf0e10cSrcweir if ( m_bIgnoreAlreadyRunning ) 1312cdf0e10cSrcweir { 1313cdf0e10cSrcweir Log( TEXT("Ignoring already running MSI instance!\r\n") ); 1314cdf0e10cSrcweir return false; 1315cdf0e10cSrcweir } 1316cdf0e10cSrcweir 1317cdf0e10cSrcweir const TCHAR *sMutexName = NULL; 1318cdf0e10cSrcweir const TCHAR sGUniqueName[] = TEXT( "Global\\_MSISETUP_{EA8130C1-8D3D-4338-9309-1A52D530D846}" ); 1319cdf0e10cSrcweir const TCHAR sUniqueName[] = TEXT( "_MSISETUP_{EA8130C1-8D3D-4338-9309-1A52D530D846}" ); 1320cdf0e10cSrcweir 1321cdf0e10cSrcweir if ( IsWin9x() ) 1322cdf0e10cSrcweir sMutexName = sUniqueName; 1323cdf0e10cSrcweir else if ( ( GetOSVersion() < 5 ) && ! IsTerminalServerInstalled() ) 1324cdf0e10cSrcweir sMutexName = sUniqueName; 1325cdf0e10cSrcweir else 1326cdf0e10cSrcweir sMutexName = sGUniqueName; 1327cdf0e10cSrcweir 1328cdf0e10cSrcweir HANDLE hMutex = 0; 1329cdf0e10cSrcweir 1330cdf0e10cSrcweir hMutex = WIN::CreateMutex( NULL, FALSE, sMutexName ); 1331cdf0e10cSrcweir 1332cdf0e10cSrcweir if ( !hMutex || ERROR_ALREADY_EXISTS == WIN::GetLastError() ) 1333cdf0e10cSrcweir { 1334cdf0e10cSrcweir if ( !hMutex ) 1335cdf0e10cSrcweir Log( TEXT( "ERROR: AlreadyRunning() could not create mutex!\r\n" ) ); 1336cdf0e10cSrcweir else 1337cdf0e10cSrcweir Log( TEXT( "ERROR: There's already a setup running!\r\n" ) ); 1338cdf0e10cSrcweir 1339cdf0e10cSrcweir return true; 1340cdf0e10cSrcweir } 1341cdf0e10cSrcweir Log( TEXT( " No running Setup found\r\n" ) ); 1342cdf0e10cSrcweir 1343cdf0e10cSrcweir return false; 1344cdf0e10cSrcweir } 1345cdf0e10cSrcweir 1346cdf0e10cSrcweir //-------------------------------------------------------------------------- 1347cdf0e10cSrcweir DWORD SetupAppX::WaitForProcess( HANDLE hHandle ) 1348cdf0e10cSrcweir { 1349cdf0e10cSrcweir DWORD nResult = NOERROR; 1350cdf0e10cSrcweir boolean bLoop = true; 1351cdf0e10cSrcweir 1352cdf0e10cSrcweir MSG aMsg; 1353cdf0e10cSrcweir ZeroMemory( (void*) &aMsg, sizeof(MSG) ); 1354cdf0e10cSrcweir 1355cdf0e10cSrcweir while ( bLoop ) 1356cdf0e10cSrcweir { 1357cdf0e10cSrcweir switch ( WIN::MsgWaitForMultipleObjects( 1, &hHandle, false, 1358cdf0e10cSrcweir INFINITE, QS_ALLINPUT ) ) 1359cdf0e10cSrcweir { 1360cdf0e10cSrcweir case WAIT_OBJECT_0: bLoop = false; 1361cdf0e10cSrcweir break; 1362cdf0e10cSrcweir 1363cdf0e10cSrcweir case (WAIT_OBJECT_0 + 1): 1364cdf0e10cSrcweir { 1365cdf0e10cSrcweir if ( WIN::PeekMessage( &aMsg, NULL, NULL, NULL, PM_REMOVE ) ) 1366cdf0e10cSrcweir { 1367cdf0e10cSrcweir WIN::TranslateMessage( &aMsg ); 1368cdf0e10cSrcweir WIN::DispatchMessage( &aMsg ); 1369cdf0e10cSrcweir } 1370cdf0e10cSrcweir break; 1371cdf0e10cSrcweir } 1372cdf0e10cSrcweir 1373cdf0e10cSrcweir default: 1374cdf0e10cSrcweir { 1375cdf0e10cSrcweir nResult = WIN::GetLastError(); 1376cdf0e10cSrcweir bLoop = false; 1377cdf0e10cSrcweir } 1378cdf0e10cSrcweir } 1379cdf0e10cSrcweir } 1380cdf0e10cSrcweir 1381cdf0e10cSrcweir return nResult; 1382cdf0e10cSrcweir } 1383cdf0e10cSrcweir 1384cdf0e10cSrcweir //-------------------------------------------------------------------------- 1385cdf0e10cSrcweir void SetupAppX::Log( LPCTSTR pMessage, LPCTSTR pText ) const 1386cdf0e10cSrcweir { 1387cdf0e10cSrcweir if ( m_pLogFile ) 1388cdf0e10cSrcweir { 1389cdf0e10cSrcweir static boolean bInit = false; 1390cdf0e10cSrcweir 1391cdf0e10cSrcweir if ( !bInit ) 1392cdf0e10cSrcweir { 1393cdf0e10cSrcweir bInit = true; 1394cdf0e10cSrcweir if ( ! IsWin9x() ) 1395cdf0e10cSrcweir _ftprintf( m_pLogFile, TEXT("%c"), 0xfeff ); 1396cdf0e10cSrcweir 1397cdf0e10cSrcweir _tsetlocale( LC_ALL, TEXT("") ); 1398cdf0e10cSrcweir _ftprintf( m_pLogFile, TEXT("\nCodepage=%s\nMultiByte Codepage=[%d]\n"), 1399cdf0e10cSrcweir _tsetlocale( LC_ALL, NULL ), _getmbcp() ); 1400cdf0e10cSrcweir } 1401cdf0e10cSrcweir if ( pText ) 1402cdf0e10cSrcweir { 1403cdf0e10cSrcweir _ftprintf( m_pLogFile, pMessage, pText ); 1404cdf0e10cSrcweir OutputDebugStringFormat( pMessage, pText ); 1405cdf0e10cSrcweir } 1406cdf0e10cSrcweir else 1407cdf0e10cSrcweir { 1408cdf0e10cSrcweir _ftprintf( m_pLogFile, pMessage ); 1409cdf0e10cSrcweir OutputDebugStringFormat( pMessage ); 1410cdf0e10cSrcweir } 1411cdf0e10cSrcweir 1412cdf0e10cSrcweir fflush( m_pLogFile ); 1413cdf0e10cSrcweir } 1414cdf0e10cSrcweir } 1415cdf0e10cSrcweir 1416cdf0e10cSrcweir //-------------------------------------------------------------------------- 1417cdf0e10cSrcweir DWORD SetupAppX::GetNextArgument( LPCTSTR pStr, LPTSTR *pArg, LPTSTR *pNext, 1418cdf0e10cSrcweir boolean bStripQuotes ) 1419cdf0e10cSrcweir { 1420cdf0e10cSrcweir boolean bInQuotes = false; 1421cdf0e10cSrcweir boolean bFoundArgEnd = false; 1422cdf0e10cSrcweir LPCTSTR pChar = pStr; 1423cdf0e10cSrcweir LPCTSTR pFirst = NULL; 1424cdf0e10cSrcweir 1425cdf0e10cSrcweir if ( NULL == pChar ) 1426cdf0e10cSrcweir return ERROR_NO_MORE_ITEMS; 1427cdf0e10cSrcweir 1428cdf0e10cSrcweir while ( ' ' == (*pChar) || '\t' == (*pChar) ) 1429cdf0e10cSrcweir pChar = CharNext( pChar ); 1430cdf0e10cSrcweir 1431cdf0e10cSrcweir if ( '\0' == (*pChar) ) 1432cdf0e10cSrcweir return ERROR_NO_MORE_ITEMS; 1433cdf0e10cSrcweir 1434cdf0e10cSrcweir int nCount = 1; 1435cdf0e10cSrcweir pFirst = pChar; 1436cdf0e10cSrcweir 1437cdf0e10cSrcweir while ( ! bFoundArgEnd ) 1438cdf0e10cSrcweir { 1439cdf0e10cSrcweir if ( '\0' == (*pChar) ) 1440cdf0e10cSrcweir bFoundArgEnd = true; 1441cdf0e10cSrcweir else if ( !bInQuotes && ' ' == (*pChar) ) 1442cdf0e10cSrcweir bFoundArgEnd = true; 1443cdf0e10cSrcweir else if ( !bInQuotes && '\t' == (*pChar) ) 1444cdf0e10cSrcweir bFoundArgEnd = true; 1445cdf0e10cSrcweir else 1446cdf0e10cSrcweir { 1447cdf0e10cSrcweir if ( '\"' == (*pChar) ) 1448cdf0e10cSrcweir { 1449cdf0e10cSrcweir bInQuotes = !bInQuotes; 1450cdf0e10cSrcweir if ( bStripQuotes ) 1451cdf0e10cSrcweir { 1452cdf0e10cSrcweir if ( pChar == pFirst ) 1453cdf0e10cSrcweir pFirst = CharNext( pFirst ); 1454cdf0e10cSrcweir nCount -= 1; 1455cdf0e10cSrcweir } 1456cdf0e10cSrcweir } 1457cdf0e10cSrcweir 1458cdf0e10cSrcweir pChar = CharNext( pChar ); 1459cdf0e10cSrcweir nCount += 1; 1460cdf0e10cSrcweir } 1461cdf0e10cSrcweir } 1462cdf0e10cSrcweir 1463cdf0e10cSrcweir if ( pArg ) 1464cdf0e10cSrcweir { 1465cdf0e10cSrcweir *pArg = new TCHAR[ nCount ]; 1466cdf0e10cSrcweir StringCchCopyN ( *pArg, nCount, pFirst, nCount-1 ); 1467cdf0e10cSrcweir } 1468cdf0e10cSrcweir 1469cdf0e10cSrcweir if ( pNext ) 1470cdf0e10cSrcweir *pNext = CharNext( pChar ); 1471cdf0e10cSrcweir 1472cdf0e10cSrcweir return ERROR_SUCCESS; 1473cdf0e10cSrcweir } 1474cdf0e10cSrcweir 1475cdf0e10cSrcweir //-------------------------------------------------------------------------- 1476cdf0e10cSrcweir boolean SetupAppX::GetCmdLineParameters( LPTSTR *pCmdLine ) 1477cdf0e10cSrcweir { 1478cdf0e10cSrcweir int nRet = ERROR_SUCCESS; 1479cdf0e10cSrcweir LPTSTR pStart = NULL; 1480cdf0e10cSrcweir LPTSTR pNext = NULL; 1481cdf0e10cSrcweir 1482cdf0e10cSrcweir if ( GetNextArgument( *pCmdLine, NULL, &pNext ) != ERROR_SUCCESS ) 1483cdf0e10cSrcweir { 1484cdf0e10cSrcweir SetError( ERROR_NO_MORE_ITEMS ); 1485cdf0e10cSrcweir return false; 1486cdf0e10cSrcweir } 1487cdf0e10cSrcweir 1488cdf0e10cSrcweir int nSize = lstrlen( *pCmdLine ) + 2; 1489cdf0e10cSrcweir TCHAR *pNewCmdLine = new TCHAR[ nSize ]; 1490cdf0e10cSrcweir pNewCmdLine[0] = '\0'; 1491cdf0e10cSrcweir 1492cdf0e10cSrcweir while ( GetNextArgument( pNext, &pStart, &pNext ) == ERROR_SUCCESS ) 1493cdf0e10cSrcweir { 1494cdf0e10cSrcweir boolean bDeleteStart = true; 1495cdf0e10cSrcweir 1496cdf0e10cSrcweir if ( (*pStart) == '/' || (*pStart) == '-' ) 1497cdf0e10cSrcweir { 1498cdf0e10cSrcweir LPTSTR pSub = CharNext( pStart ); 1499cdf0e10cSrcweir if ( (*pSub) == 'l' || (*pSub) == 'L' ) 1500cdf0e10cSrcweir { 1501cdf0e10cSrcweir pSub = CharNext( pSub ); 1502cdf0e10cSrcweir if ( (*pSub) == 'a' || (*pSub) == 'A' ) 1503cdf0e10cSrcweir { // --- handle the lang parameter --- 1504cdf0e10cSrcweir LPTSTR pLanguage = NULL; 1505cdf0e10cSrcweir LPTSTR pLastChar; 1506cdf0e10cSrcweir if ( GetNextArgument( pNext, &pLanguage, &pNext, true ) != ERROR_SUCCESS ) 1507cdf0e10cSrcweir { 1508cdf0e10cSrcweir StringCchCopy( m_pErrorText, MAX_TEXT_LENGTH, pStart ); 1509cdf0e10cSrcweir nRet = ERROR_INVALID_PARAMETER; 1510cdf0e10cSrcweir break; 1511cdf0e10cSrcweir } 1512cdf0e10cSrcweir 1513cdf0e10cSrcweir m_nLanguageID = _tcstol( pLanguage, &pLastChar, 10 ); 1514cdf0e10cSrcweir delete [] pLanguage; 1515cdf0e10cSrcweir } 1516cdf0e10cSrcweir else 1517cdf0e10cSrcweir { // --- handle the l(og) parameter --- 1518cdf0e10cSrcweir boolean bAppend = false; 1519cdf0e10cSrcweir LPTSTR pFileName = NULL; 1520cdf0e10cSrcweir 1521cdf0e10cSrcweir while ( *pSub ) 1522cdf0e10cSrcweir { 1523cdf0e10cSrcweir if ( *pSub == '+' ) 1524cdf0e10cSrcweir { 1525cdf0e10cSrcweir bAppend = true; 1526cdf0e10cSrcweir break; 1527cdf0e10cSrcweir } 1528cdf0e10cSrcweir pSub = CharNext( pSub ); 1529cdf0e10cSrcweir } 1530cdf0e10cSrcweir 1531cdf0e10cSrcweir if ( GetNextArgument( pNext, &pFileName, &pNext, true ) != ERROR_SUCCESS ) 1532cdf0e10cSrcweir { 1533cdf0e10cSrcweir StringCchCopy( m_pErrorText, MAX_TEXT_LENGTH, pStart ); 1534cdf0e10cSrcweir nRet = ERROR_INVALID_PARAMETER; 1535cdf0e10cSrcweir break; 1536cdf0e10cSrcweir } 1537cdf0e10cSrcweir 1538cdf0e10cSrcweir if ( FAILED( StringCchCat( pNewCmdLine, nSize, pStart ) ) ) 1539cdf0e10cSrcweir { 1540cdf0e10cSrcweir nRet = ERROR_OUTOFMEMORY; 1541cdf0e10cSrcweir break; 1542cdf0e10cSrcweir } 1543cdf0e10cSrcweir // we need to append a '+' otherwise msiexec would overwrite our log file 1544cdf0e10cSrcweir if ( !bAppend && FAILED( StringCchCat( pNewCmdLine, nSize, TEXT( "+" ) ) ) ) 1545cdf0e10cSrcweir { 1546cdf0e10cSrcweir nRet = ERROR_OUTOFMEMORY; 1547cdf0e10cSrcweir break; 1548cdf0e10cSrcweir } 1549cdf0e10cSrcweir if ( FAILED( StringCchCat( pNewCmdLine, nSize, TEXT( " \"" ) ) ) || 1550cdf0e10cSrcweir FAILED( StringCchCat( pNewCmdLine, nSize, pFileName ) ) || 1551cdf0e10cSrcweir FAILED( StringCchCat( pNewCmdLine, nSize, TEXT( "\" " ) ) ) ) 1552cdf0e10cSrcweir { 1553cdf0e10cSrcweir nRet = ERROR_OUTOFMEMORY; 1554cdf0e10cSrcweir break; 1555cdf0e10cSrcweir } 1556cdf0e10cSrcweir 1557cdf0e10cSrcweir if ( bAppend ) 1558cdf0e10cSrcweir m_pLogFile = _tfopen( pFileName, TEXT( "ab" ) ); 1559cdf0e10cSrcweir else 1560cdf0e10cSrcweir m_pLogFile = _tfopen( pFileName, TEXT( "wb" ) ); 1561cdf0e10cSrcweir 1562cdf0e10cSrcweir delete [] pFileName; 1563cdf0e10cSrcweir } 1564cdf0e10cSrcweir } 1565cdf0e10cSrcweir else if ( (*pSub) == 'q' || (*pSub) == 'Q' ) 1566cdf0e10cSrcweir { // --- Handle quiet file parameter --- 1567cdf0e10cSrcweir pSub = CharNext( pSub ); 1568cdf0e10cSrcweir if ( ! (*pSub) || (*pSub) == 'n' || (*pSub) == 'N' ) 1569cdf0e10cSrcweir m_bQuiet = true; 1570cdf0e10cSrcweir 1571cdf0e10cSrcweir if ( FAILED( StringCchCat( pNewCmdLine, nSize, pStart ) ) || 1572cdf0e10cSrcweir FAILED( StringCchCat( pNewCmdLine, nSize, TEXT( " " ) ) ) ) 1573cdf0e10cSrcweir { 1574cdf0e10cSrcweir nRet = ERROR_OUTOFMEMORY; 1575cdf0e10cSrcweir break; 1576cdf0e10cSrcweir } 1577cdf0e10cSrcweir } 1578cdf0e10cSrcweir else if ( _tcsnicmp( pSub, PARAM_RUNNING, _tcslen( PARAM_RUNNING ) ) == 0 ) 1579cdf0e10cSrcweir { 1580cdf0e10cSrcweir m_bIgnoreAlreadyRunning = true; 1581cdf0e10cSrcweir } 1582cdf0e10cSrcweir else if ( _tcsnicmp( pSub, CMDLN_REG_ALL_MSO_TYPES, _tcslen( CMDLN_REG_ALL_MSO_TYPES ) ) == 0 ) 1583cdf0e10cSrcweir { 1584cdf0e10cSrcweir m_bRegAllMsoTypes = true; 1585cdf0e10cSrcweir } 1586cdf0e10cSrcweir else if ( _tcsnicmp( pSub, CMDLN_REG_NO_MSO_TYPES, _tcslen( CMDLN_REG_NO_MSO_TYPES ) ) == 0 ) 1587cdf0e10cSrcweir { 1588cdf0e10cSrcweir m_bRegNoMsoTypes = true; 1589cdf0e10cSrcweir } 1590cdf0e10cSrcweir else if ( (*pSub) == 'i' || (*pSub) == 'I' || (*pSub) == 'f' || (*pSub) == 'F' || 1591cdf0e10cSrcweir (*pSub) == 'p' || (*pSub) == 'P' || (*pSub) == 'x' || (*pSub) == 'X' || 1592cdf0e10cSrcweir (*pSub) == 'y' || (*pSub) == 'Y' || (*pSub) == 'z' || (*pSub) == 'Z' ) 1593cdf0e10cSrcweir { 1594cdf0e10cSrcweir StringCchCopy( m_pErrorText, MAX_TEXT_LENGTH, pStart ); 1595cdf0e10cSrcweir nRet = ERROR_INVALID_PARAMETER; 1596cdf0e10cSrcweir break; 1597cdf0e10cSrcweir } 1598cdf0e10cSrcweir else if ( (*pSub) == 'a' || (*pSub) == 'A' ) 1599cdf0e10cSrcweir { // --- Handle Adminstrative Installation --- 1600cdf0e10cSrcweir SetAdminInstall( true ); 1601cdf0e10cSrcweir } 1602cdf0e10cSrcweir else if ( (*pSub) == 'j' || (*pSub) == 'J' ) 1603cdf0e10cSrcweir { // --- Handle Adminstrative Installation --- 1604cdf0e10cSrcweir m_pAdvertise = pStart; 1605cdf0e10cSrcweir m_bQuiet = true; 1606cdf0e10cSrcweir bDeleteStart = false; 1607cdf0e10cSrcweir } 1608cdf0e10cSrcweir else if ( (*pSub) == '?' || (*pSub) == 'h' || (*pSub) == 'H' ) 1609cdf0e10cSrcweir { // --- Handle Show Usage --- 1610cdf0e10cSrcweir nRet = ERROR_SHOW_USAGE; 1611cdf0e10cSrcweir break; 1612cdf0e10cSrcweir } 1613cdf0e10cSrcweir else 1614cdf0e10cSrcweir { 1615cdf0e10cSrcweir if ( FAILED( StringCchCat( pNewCmdLine, nSize, pStart ) ) || 1616cdf0e10cSrcweir FAILED( StringCchCat( pNewCmdLine, nSize, TEXT( " " ) ) ) ) 1617cdf0e10cSrcweir { 1618cdf0e10cSrcweir nRet = ERROR_OUTOFMEMORY; 1619cdf0e10cSrcweir break; 1620cdf0e10cSrcweir } 1621cdf0e10cSrcweir } 1622cdf0e10cSrcweir } 1623cdf0e10cSrcweir else 1624cdf0e10cSrcweir { 1625cdf0e10cSrcweir if ( FAILED( StringCchCat( pNewCmdLine, nSize, pStart ) ) || 1626cdf0e10cSrcweir FAILED( StringCchCat( pNewCmdLine, nSize, TEXT( " " ) ) ) ) 1627cdf0e10cSrcweir { 1628cdf0e10cSrcweir nRet = ERROR_OUTOFMEMORY; 1629cdf0e10cSrcweir break; 1630cdf0e10cSrcweir } 1631cdf0e10cSrcweir } 1632cdf0e10cSrcweir 1633cdf0e10cSrcweir if ( bDeleteStart ) delete [] pStart; 1634cdf0e10cSrcweir pStart = NULL; 1635cdf0e10cSrcweir } 1636cdf0e10cSrcweir 1637cdf0e10cSrcweir if ( pStart ) delete [] pStart; 1638cdf0e10cSrcweir 1639cdf0e10cSrcweir *pCmdLine = pNewCmdLine; 1640cdf0e10cSrcweir 1641cdf0e10cSrcweir if ( nRet != ERROR_SUCCESS ) 1642cdf0e10cSrcweir { 1643cdf0e10cSrcweir SetError( nRet ); 1644cdf0e10cSrcweir return false; 1645cdf0e10cSrcweir } 1646cdf0e10cSrcweir else 1647cdf0e10cSrcweir return true;; 1648cdf0e10cSrcweir } 1649cdf0e10cSrcweir 1650cdf0e10cSrcweir //-------------------------------------------------------------------------- 1651cdf0e10cSrcweir boolean SetupAppX::IsAdmin() 1652cdf0e10cSrcweir { 1653cdf0e10cSrcweir if ( IsWin9x() ) 1654cdf0e10cSrcweir return true; 1655cdf0e10cSrcweir 1656cdf0e10cSrcweir PSID aPsidAdmin; 1657cdf0e10cSrcweir SID_IDENTIFIER_AUTHORITY aAuthority = SECURITY_NT_AUTHORITY; 1658cdf0e10cSrcweir 1659cdf0e10cSrcweir if ( !AllocateAndInitializeSid( &aAuthority, 2, SECURITY_BUILTIN_DOMAIN_RID, 1660cdf0e10cSrcweir DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, 1661cdf0e10cSrcweir &aPsidAdmin ) ) 1662cdf0e10cSrcweir return false; 1663cdf0e10cSrcweir 1664cdf0e10cSrcweir BOOL bIsAdmin = FALSE; 1665cdf0e10cSrcweir 1666cdf0e10cSrcweir if ( GetOSVersion() >= 5 ) 1667cdf0e10cSrcweir { 1668cdf0e10cSrcweir HMODULE hAdvapi32 = LoadLibrary( ADVAPI32_DLL ); 1669cdf0e10cSrcweir 1670cdf0e10cSrcweir if ( !hAdvapi32 ) 1671cdf0e10cSrcweir bIsAdmin = FALSE; 1672cdf0e10cSrcweir else 1673cdf0e10cSrcweir { 1674cdf0e10cSrcweir PFnCheckTokenMembership pfnCheckTokenMembership = (PFnCheckTokenMembership) GetProcAddress( hAdvapi32, ADVAPI32API_CheckTokenMembership); 1675cdf0e10cSrcweir if ( !pfnCheckTokenMembership || !pfnCheckTokenMembership( NULL, aPsidAdmin, &bIsAdmin ) ) 1676cdf0e10cSrcweir bIsAdmin = FALSE; 1677cdf0e10cSrcweir } 1678cdf0e10cSrcweir FreeLibrary( hAdvapi32 ); 1679cdf0e10cSrcweir } 1680cdf0e10cSrcweir else 1681cdf0e10cSrcweir { 1682cdf0e10cSrcweir // NT4, check groups of user 1683cdf0e10cSrcweir HANDLE hAccessToken = 0; 1684cdf0e10cSrcweir UCHAR *szInfoBuffer = new UCHAR[ 1024 ]; // may need to resize if TokenInfo too big 1685cdf0e10cSrcweir DWORD dwInfoBufferSize = 1024; 1686cdf0e10cSrcweir DWORD dwRetInfoBufferSize = 0; 1687cdf0e10cSrcweir UINT i=0; 1688cdf0e10cSrcweir 1689cdf0e10cSrcweir if ( WIN::OpenProcessToken( WIN::GetCurrentProcess(), TOKEN_READ, &hAccessToken ) ) 1690cdf0e10cSrcweir { 1691cdf0e10cSrcweir bool bSuccess = false; 1692cdf0e10cSrcweir bSuccess = WIN::GetTokenInformation( hAccessToken, TokenGroups, 1693cdf0e10cSrcweir szInfoBuffer, dwInfoBufferSize, 1694cdf0e10cSrcweir &dwRetInfoBufferSize ) == TRUE; 1695cdf0e10cSrcweir 1696cdf0e10cSrcweir if( dwRetInfoBufferSize > dwInfoBufferSize ) 1697cdf0e10cSrcweir { 1698cdf0e10cSrcweir delete [] szInfoBuffer; 1699cdf0e10cSrcweir szInfoBuffer = new UCHAR[ dwRetInfoBufferSize ]; 1700cdf0e10cSrcweir dwInfoBufferSize = dwRetInfoBufferSize; 1701cdf0e10cSrcweir bSuccess = WIN::GetTokenInformation( hAccessToken, TokenGroups, 1702cdf0e10cSrcweir szInfoBuffer, dwInfoBufferSize, 1703cdf0e10cSrcweir &dwRetInfoBufferSize ) == TRUE; 1704cdf0e10cSrcweir } 1705cdf0e10cSrcweir 1706cdf0e10cSrcweir WIN::CloseHandle( hAccessToken ); 1707cdf0e10cSrcweir 1708cdf0e10cSrcweir if ( bSuccess ) 1709cdf0e10cSrcweir { 1710cdf0e10cSrcweir PTOKEN_GROUPS pGroups = (PTOKEN_GROUPS)(UCHAR*) szInfoBuffer; 1711cdf0e10cSrcweir for( i=0; i<pGroups->GroupCount; i++ ) 1712cdf0e10cSrcweir { 1713cdf0e10cSrcweir if( WIN::EqualSid( aPsidAdmin, pGroups->Groups[i].Sid ) ) 1714cdf0e10cSrcweir { 1715cdf0e10cSrcweir bIsAdmin = TRUE; 1716cdf0e10cSrcweir break; 1717cdf0e10cSrcweir } 1718cdf0e10cSrcweir } 1719cdf0e10cSrcweir } 1720cdf0e10cSrcweir 1721cdf0e10cSrcweir delete [] szInfoBuffer; 1722cdf0e10cSrcweir } 1723cdf0e10cSrcweir } 1724cdf0e10cSrcweir 1725cdf0e10cSrcweir WIN::FreeSid( aPsidAdmin ); 1726cdf0e10cSrcweir 1727cdf0e10cSrcweir return bIsAdmin ? true : false; 1728cdf0e10cSrcweir } 1729cdf0e10cSrcweir 1730cdf0e10cSrcweir //-------------------------------------------------------------------------- 1731cdf0e10cSrcweir LPTSTR SetupAppX::CopyIniFile( LPCTSTR pIniFile ) 1732cdf0e10cSrcweir { 1733cdf0e10cSrcweir m_pTmpName = _ttempnam( TEXT( "C:\\" ), TEXT( "Setup" ) ); 1734cdf0e10cSrcweir 1735cdf0e10cSrcweir if ( !m_pTmpName ) 1736cdf0e10cSrcweir { 1737cdf0e10cSrcweir Log( TEXT( "ERROR: Could not create temp file\n" ) ); 1738cdf0e10cSrcweir return NULL; 1739cdf0e10cSrcweir } 1740cdf0e10cSrcweir 1741cdf0e10cSrcweir FILE *pOut = _tfopen( m_pTmpName, TEXT( "wb" ) ); 1742cdf0e10cSrcweir FILE *pIn = _tfopen( pIniFile, TEXT( "rb" ) ); 1743cdf0e10cSrcweir 1744cdf0e10cSrcweir if ( pOut && pIn ) 1745cdf0e10cSrcweir { 1746cdf0e10cSrcweir size_t nRead, nWritten; 1747cdf0e10cSrcweir BYTE pBuf[1024]; 1748cdf0e10cSrcweir 1749cdf0e10cSrcweir nRead = fread( pBuf, sizeof( BYTE ), 1024, pIn ); 1750cdf0e10cSrcweir while ( nRead && !ferror( pIn ) ) 1751cdf0e10cSrcweir { 1752cdf0e10cSrcweir nWritten = fwrite( pBuf, sizeof( BYTE ), nRead, pOut ); 1753cdf0e10cSrcweir if ( nWritten != nRead ) 1754cdf0e10cSrcweir { 1755cdf0e10cSrcweir Log( TEXT( "ERROR: Could not write all bytes to temp file\n" ) ); 1756cdf0e10cSrcweir break; 1757cdf0e10cSrcweir } 1758cdf0e10cSrcweir nRead = fread( pBuf, sizeof( BYTE ), 1024, pIn ); 1759cdf0e10cSrcweir } 1760cdf0e10cSrcweir } 1761cdf0e10cSrcweir 1762cdf0e10cSrcweir if ( pOut ) fclose( pOut ); 1763cdf0e10cSrcweir if ( pIn ) fclose( pIn ); 1764cdf0e10cSrcweir 1765cdf0e10cSrcweir return m_pTmpName; 1766cdf0e10cSrcweir } 1767cdf0e10cSrcweir 1768cdf0e10cSrcweir //-------------------------------------------------------------------------- 1769cdf0e10cSrcweir void SetupAppX::ConvertNewline( LPTSTR pText ) const 1770cdf0e10cSrcweir { 1771cdf0e10cSrcweir int i=0; 1772cdf0e10cSrcweir 1773cdf0e10cSrcweir while ( pText[i] != 0 ) 1774cdf0e10cSrcweir { 1775cdf0e10cSrcweir if ( ( pText[i] == '\\' ) && ( pText[i+1] == 'n' ) ) 1776cdf0e10cSrcweir { 1777cdf0e10cSrcweir pText[i] = 0x0d; 1778cdf0e10cSrcweir pText[i+1] = 0x0a; 1779cdf0e10cSrcweir i+=2; 1780cdf0e10cSrcweir } 1781cdf0e10cSrcweir else 1782cdf0e10cSrcweir i+=1; 1783cdf0e10cSrcweir } 1784cdf0e10cSrcweir } 1785cdf0e10cSrcweir 1786cdf0e10cSrcweir //-------------------------------------------------------------------------- 1787cdf0e10cSrcweir LPTSTR SetupAppX::SetProdToAppTitle( LPCTSTR pProdName ) 1788cdf0e10cSrcweir { 1789cdf0e10cSrcweir if ( !pProdName ) return m_pAppTitle; 1790cdf0e10cSrcweir 1791cdf0e10cSrcweir LPTSTR pAppProdTitle = new TCHAR[ MAX_STR_CAPTION ]; 1792cdf0e10cSrcweir pAppProdTitle[0] = '\0'; 1793cdf0e10cSrcweir 1794cdf0e10cSrcweir WIN::LoadString( m_hInst, IDS_APP_PROD_TITLE, pAppProdTitle, MAX_STR_CAPTION ); 1795cdf0e10cSrcweir 1796cdf0e10cSrcweir int nAppLen = lstrlen( pAppProdTitle ); 1797cdf0e10cSrcweir int nProdLen = lstrlen( pProdName ); 1798cdf0e10cSrcweir 1799cdf0e10cSrcweir if ( ( nAppLen == 0 ) || ( nProdLen == 0 ) ) 1800cdf0e10cSrcweir { 1801cdf0e10cSrcweir delete [] pAppProdTitle; 1802cdf0e10cSrcweir return m_pAppTitle; 1803cdf0e10cSrcweir } 1804cdf0e10cSrcweir 1805cdf0e10cSrcweir int nLen = nAppLen + nProdLen + 3; 1806cdf0e10cSrcweir 1807cdf0e10cSrcweir if ( nLen > STRSAFE_MAX_CCH ) return m_pAppTitle; 1808cdf0e10cSrcweir 1809cdf0e10cSrcweir LPTSTR pIndex = _tcsstr( pAppProdTitle, PRODUCT_NAME_VAR ); 1810cdf0e10cSrcweir 1811cdf0e10cSrcweir if ( pIndex ) 1812cdf0e10cSrcweir { 1813cdf0e10cSrcweir int nOffset = pIndex - pAppProdTitle; 1814cdf0e10cSrcweir int nVarLen = lstrlen( PRODUCT_NAME_VAR ); 1815cdf0e10cSrcweir 1816cdf0e10cSrcweir LPTSTR pNewTitle = new TCHAR[ nLen ]; 1817cdf0e10cSrcweir pNewTitle[0] = '\0'; 1818cdf0e10cSrcweir 1819cdf0e10cSrcweir if ( nOffset > 0 ) 1820cdf0e10cSrcweir { 1821cdf0e10cSrcweir StringCchCopyN( pNewTitle, nLen, pAppProdTitle, nOffset ); 1822cdf0e10cSrcweir } 1823cdf0e10cSrcweir 1824cdf0e10cSrcweir StringCchCat( pNewTitle, nLen, pProdName ); 1825cdf0e10cSrcweir 1826cdf0e10cSrcweir if ( nOffset + nVarLen < nAppLen ) 1827cdf0e10cSrcweir { 1828cdf0e10cSrcweir StringCchCat( pNewTitle, nLen, pIndex + nVarLen ); 1829cdf0e10cSrcweir } 1830cdf0e10cSrcweir 1831cdf0e10cSrcweir delete [] m_pAppTitle; 1832cdf0e10cSrcweir m_pAppTitle = pNewTitle; 1833cdf0e10cSrcweir } 1834cdf0e10cSrcweir 1835cdf0e10cSrcweir delete [] pAppProdTitle; 1836cdf0e10cSrcweir 1837cdf0e10cSrcweir return m_pAppTitle; 1838cdf0e10cSrcweir } 1839cdf0e10cSrcweir 1840cdf0e10cSrcweir 1841cdf0e10cSrcweir //-------------------------------------------------------------------------- 1842cdf0e10cSrcweir boolean SetupAppX::IsPatchInstalled( TCHAR* pBaseDir, TCHAR* pFileName ) 1843cdf0e10cSrcweir { 1844cdf0e10cSrcweir if ( !m_bSupportsPatch ) 1845cdf0e10cSrcweir return false; 1846cdf0e10cSrcweir 1847cdf0e10cSrcweir PMSIHANDLE hSummaryInfo; 1848cdf0e10cSrcweir int nLen = lstrlen( pBaseDir ) + lstrlen( pFileName ) + 1; 1849cdf0e10cSrcweir TCHAR *szDatabasePath = new TCHAR [ nLen ]; 1850cdf0e10cSrcweir TCHAR sBuf[80]; 1851cdf0e10cSrcweir 1852cdf0e10cSrcweir StringCchCopy( szDatabasePath, nLen, pBaseDir ); 1853cdf0e10cSrcweir StringCchCat( szDatabasePath, nLen, pFileName ); 1854cdf0e10cSrcweir 1855cdf0e10cSrcweir UINT nRet = MsiGetSummaryInformation( NULL, szDatabasePath, 0, &hSummaryInfo ); 1856cdf0e10cSrcweir 1857cdf0e10cSrcweir if ( nRet != ERROR_SUCCESS ) 1858cdf0e10cSrcweir { 1859cdf0e10cSrcweir StringCchPrintf( sBuf, 80, TEXT("ERROR: IsPatchInstalled: MsiGetSummaryInformation returned %u.\r\n"), nRet ); 1860cdf0e10cSrcweir Log( sBuf ); 1861cdf0e10cSrcweir return false; 1862cdf0e10cSrcweir } 1863cdf0e10cSrcweir 1864cdf0e10cSrcweir UINT uiDataType; 1865cdf0e10cSrcweir LPTSTR szPatchID = new TCHAR[ 64 ]; 1866cdf0e10cSrcweir DWORD cchValueBuf = 64; 1867cdf0e10cSrcweir nRet = MsiSummaryInfoGetProperty( hSummaryInfo, PID_REVNUMBER, &uiDataType, NULL, NULL, szPatchID, &cchValueBuf ); 1868cdf0e10cSrcweir 1869cdf0e10cSrcweir if ( nRet != ERROR_SUCCESS ) 1870cdf0e10cSrcweir { 1871cdf0e10cSrcweir StringCchPrintf( sBuf, 80, TEXT("ERROR: IsPatchInstalled: MsiSummaryInfoGetProperty returned %u.\r\n"), nRet ); 1872cdf0e10cSrcweir Log( sBuf ); 1873cdf0e10cSrcweir return false; 1874cdf0e10cSrcweir } 1875cdf0e10cSrcweir 1876cdf0e10cSrcweir nRet = MsiGetPatchInfo( szPatchID, INSTALLPROPERTY_LOCALPACKAGE, NULL, NULL ); 1877cdf0e10cSrcweir 1878cdf0e10cSrcweir StringCchPrintf( sBuf, 80, TEXT(" GetPatchInfo for (%s) returned (%u)\r\n"), szPatchID, nRet ); 1879cdf0e10cSrcweir Log( sBuf ); 1880cdf0e10cSrcweir 1881cdf0e10cSrcweir delete []szPatchID; 1882cdf0e10cSrcweir 1883cdf0e10cSrcweir if ( nRet == ERROR_BAD_CONFIGURATION ) 1884cdf0e10cSrcweir return false; 1885cdf0e10cSrcweir else if ( nRet == ERROR_INVALID_PARAMETER ) 1886cdf0e10cSrcweir return false; 1887cdf0e10cSrcweir else if ( nRet == ERROR_MORE_DATA ) 1888cdf0e10cSrcweir return true; 1889cdf0e10cSrcweir else if ( nRet == ERROR_SUCCESS ) 1890cdf0e10cSrcweir return true; 1891cdf0e10cSrcweir else if ( nRet == ERROR_UNKNOWN_PRODUCT ) 1892cdf0e10cSrcweir return false; 1893cdf0e10cSrcweir else if ( nRet == ERROR_UNKNOWN_PROPERTY ) 1894cdf0e10cSrcweir return false; 1895cdf0e10cSrcweir else return false; 1896cdf0e10cSrcweir 1897cdf0e10cSrcweir return false; 1898cdf0e10cSrcweir } 1899cdf0e10cSrcweir 1900cdf0e10cSrcweir //-------------------------------------------------------------------------- 1901cdf0e10cSrcweir boolean SetupAppX::InstallRuntimes( TCHAR *sProductCode, TCHAR *sRuntimePath ) 1902cdf0e10cSrcweir { 1903cdf0e10cSrcweir INSTALLSTATE nRet = MsiQueryProductState( sProductCode ); 1904cdf0e10cSrcweir OutputDebugStringFormat( TEXT( "MsiQueryProductState returned <%d>\r\n" ), nRet ); 1905cdf0e10cSrcweir if ( nRet == INSTALLSTATE_DEFAULT ) 1906cdf0e10cSrcweir return true; 1907cdf0e10cSrcweir 1908cdf0e10cSrcweir Log( TEXT( " Will install runtime <%s>\r\n" ), sRuntimePath ); 1909cdf0e10cSrcweir OutputDebugStringFormat( TEXT( " Will install runtime <%s>\r\n" ), sRuntimePath ); 1910cdf0e10cSrcweir 1911cdf0e10cSrcweir STARTUPINFO aSUI; 1912cdf0e10cSrcweir PROCESS_INFORMATION aPI; 1913cdf0e10cSrcweir 1914cdf0e10cSrcweir ZeroMemory( (void*)&aPI, sizeof( PROCESS_INFORMATION ) ); 1915cdf0e10cSrcweir ZeroMemory( (void*)&aSUI, sizeof( STARTUPINFO ) ); 1916cdf0e10cSrcweir 1917cdf0e10cSrcweir aSUI.cb = sizeof(STARTUPINFO); 1918cdf0e10cSrcweir aSUI.dwFlags = STARTF_USESHOWWINDOW; 1919cdf0e10cSrcweir aSUI.wShowWindow = SW_SHOW; 1920cdf0e10cSrcweir 1921cdf0e10cSrcweir DWORD nCmdLineLength = lstrlen( sRuntimePath ) + lstrlen( PARAM_SILENTINSTALL ) + 2; 1922cdf0e10cSrcweir TCHAR *sCmdLine = new TCHAR[ nCmdLineLength ]; 1923cdf0e10cSrcweir 1924cdf0e10cSrcweir if ( FAILED( StringCchCopy( sCmdLine, nCmdLineLength, sRuntimePath ) ) || 1925cdf0e10cSrcweir FAILED( StringCchCat( sCmdLine, nCmdLineLength, PARAM_SILENTINSTALL ) ) ) 1926cdf0e10cSrcweir { 1927cdf0e10cSrcweir delete [] sCmdLine; 1928cdf0e10cSrcweir SetError( ERROR_INSTALL_FAILURE ); 1929cdf0e10cSrcweir return false; 1930cdf0e10cSrcweir } 1931cdf0e10cSrcweir 1932cdf0e10cSrcweir if ( !WIN::CreateProcess( NULL, sCmdLine, NULL, NULL, FALSE, 1933cdf0e10cSrcweir CREATE_DEFAULT_ERROR_MODE, NULL, NULL, 1934cdf0e10cSrcweir &aSUI, &aPI ) ) 1935cdf0e10cSrcweir { 1936cdf0e10cSrcweir Log( TEXT( "ERROR: Could not create process %s.\r\n" ), sCmdLine ); 1937cdf0e10cSrcweir SetError( WIN::GetLastError() ); 1938cdf0e10cSrcweir delete [] sCmdLine; 1939cdf0e10cSrcweir return false; 1940cdf0e10cSrcweir } 1941cdf0e10cSrcweir 1942cdf0e10cSrcweir DWORD nResult = WaitForProcess( aPI.hProcess ); 1943cdf0e10cSrcweir bool bRet = true; 1944cdf0e10cSrcweir 1945cdf0e10cSrcweir if( ERROR_SUCCESS != nResult ) 1946cdf0e10cSrcweir { 1947cdf0e10cSrcweir Log( TEXT( "ERROR: While waiting for %s.\r\n" ), sCmdLine ); 1948cdf0e10cSrcweir SetError( nResult ); 1949cdf0e10cSrcweir bRet = false; 1950cdf0e10cSrcweir } 1951cdf0e10cSrcweir else 1952cdf0e10cSrcweir { 1953cdf0e10cSrcweir GetExitCodeProcess( aPI.hProcess, &nResult ); 1954cdf0e10cSrcweir SetError( nResult ); 1955cdf0e10cSrcweir 1956cdf0e10cSrcweir if ( nResult != ERROR_SUCCESS ) 1957cdf0e10cSrcweir { 1958cdf0e10cSrcweir TCHAR sBuf[80]; 1959cdf0e10cSrcweir StringCchPrintf( sBuf, 80, TEXT("Warning: install runtime returned %u.\r\n"), nResult ); 1960cdf0e10cSrcweir Log( sBuf ); 1961cdf0e10cSrcweir } 1962cdf0e10cSrcweir else 1963cdf0e10cSrcweir Log( TEXT( " Installation of runtime completed successfully.\r\n" ) ); 1964cdf0e10cSrcweir } 1965cdf0e10cSrcweir 1966cdf0e10cSrcweir CloseHandle( aPI.hProcess ); 1967cdf0e10cSrcweir 1968cdf0e10cSrcweir delete [] sCmdLine; 1969cdf0e10cSrcweir 1970cdf0e10cSrcweir return bRet; 1971cdf0e10cSrcweir } 1972cdf0e10cSrcweir 1973cdf0e10cSrcweir //-------------------------------------------------------------------------- 1974cdf0e10cSrcweir boolean SetupAppX::InstallRuntimes() 1975cdf0e10cSrcweir { 1976cdf0e10cSrcweir TCHAR *sRuntimePath = 0; 1977cdf0e10cSrcweir SYSTEM_INFO siSysInfo; 1978cdf0e10cSrcweir 1979cdf0e10cSrcweir HMODULE hKernel32 = ::LoadLibrary(_T("Kernel32.dll")); 1980cdf0e10cSrcweir if ( hKernel32 != NULL ) 1981cdf0e10cSrcweir { 1982cdf0e10cSrcweir typedef void (CALLBACK* pfnGetNativeSystemInfo_t)(LPSYSTEM_INFO); 1983cdf0e10cSrcweir pfnGetNativeSystemInfo_t pfnGetNativeSystemInfo; 1984cdf0e10cSrcweir pfnGetNativeSystemInfo = (pfnGetNativeSystemInfo_t)::GetProcAddress(hKernel32, "GetNativeSystemInfo"); 1985cdf0e10cSrcweir if ( pfnGetNativeSystemInfo != NULL ) 1986cdf0e10cSrcweir { 1987cdf0e10cSrcweir pfnGetNativeSystemInfo(&siSysInfo); 1988cdf0e10cSrcweir } 1989cdf0e10cSrcweir else 1990cdf0e10cSrcweir { 1991cdf0e10cSrcweir // GetNativeSystemInfo does not exist. Maybe the code is running under Windows 2000. 1992cdf0e10cSrcweir // Use GetSystemInfo instead. 1993cdf0e10cSrcweir GetSystemInfo(&siSysInfo); 1994cdf0e10cSrcweir } 1995cdf0e10cSrcweir FreeLibrary(hKernel32); 1996cdf0e10cSrcweir } 1997cdf0e10cSrcweir else 1998cdf0e10cSrcweir { 1999cdf0e10cSrcweir // Failed to check Kernel32.dll. There may be something wrong. 2000cdf0e10cSrcweir // Use GetSystemInfo instead anyway. 2001cdf0e10cSrcweir GetSystemInfo(&siSysInfo); 2002cdf0e10cSrcweir } 2003cdf0e10cSrcweir 2004cdf0e10cSrcweir OutputDebugStringFormat( TEXT( "found architecture<%d>\r\n" ), siSysInfo.wProcessorArchitecture ); 2005cdf0e10cSrcweir 2006cdf0e10cSrcweir if ( siSysInfo.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64 ) 2007cdf0e10cSrcweir { 2008cdf0e10cSrcweir if ( GetPathToFile( RUNTIME_X64_NAME, &sRuntimePath ) ) 2009cdf0e10cSrcweir InstallRuntimes( PRODUCTCODE_X64, sRuntimePath ); 2010cdf0e10cSrcweir else 2011cdf0e10cSrcweir Log( TEXT( "ERROR: no installer for x64 runtime libraries found!" ) ); 2012cdf0e10cSrcweir 2013cdf0e10cSrcweir if ( sRuntimePath ) 2014cdf0e10cSrcweir { 2015cdf0e10cSrcweir delete [] sRuntimePath; 2016cdf0e10cSrcweir sRuntimePath = 0; 2017cdf0e10cSrcweir } 2018cdf0e10cSrcweir } 2019cdf0e10cSrcweir 2020cdf0e10cSrcweir if ( GetPathToFile( RUNTIME_X86_NAME, &sRuntimePath ) ) 2021cdf0e10cSrcweir InstallRuntimes( PRODUCTCODE_X86, sRuntimePath ); 2022cdf0e10cSrcweir else 2023cdf0e10cSrcweir Log( TEXT( "ERROR: no installer for x86 runtime libraries found!" ) ); 2024cdf0e10cSrcweir 2025cdf0e10cSrcweir if ( sRuntimePath ) 2026cdf0e10cSrcweir delete [] sRuntimePath; 2027cdf0e10cSrcweir 2028cdf0e10cSrcweir return true; 2029cdf0e10cSrcweir } 2030cdf0e10cSrcweir 2031cdf0e10cSrcweir //-------------------------------------------------------------------------- 2032cdf0e10cSrcweir //-------------------------------------------------------------------------- 2033cdf0e10cSrcweir LanguageDataX::LanguageDataX( LPTSTR pData ) 2034cdf0e10cSrcweir { 2035cdf0e10cSrcweir m_nLanguageID = 0; 2036cdf0e10cSrcweir m_pTransform = NULL; 2037cdf0e10cSrcweir 2038cdf0e10cSrcweir LPTSTR pLastChar; 2039cdf0e10cSrcweir 2040cdf0e10cSrcweir m_nLanguageID = _tcstol( pData, &pLastChar, 10 ); 2041cdf0e10cSrcweir 2042cdf0e10cSrcweir if ( *pLastChar == ',' ) 2043cdf0e10cSrcweir { 2044cdf0e10cSrcweir pLastChar += 1; 2045cdf0e10cSrcweir int nLen = lstrlen( pLastChar ) + 1; 2046cdf0e10cSrcweir m_pTransform = new TCHAR [ nLen ]; 2047cdf0e10cSrcweir StringCchCopy( m_pTransform, nLen, pLastChar ); 2048cdf0e10cSrcweir } 2049cdf0e10cSrcweir } 2050cdf0e10cSrcweir 2051cdf0e10cSrcweir //-------------------------------------------------------------------------- 2052cdf0e10cSrcweir LanguageDataX::~LanguageDataX() 2053cdf0e10cSrcweir { 2054cdf0e10cSrcweir if ( m_pTransform ) delete [] m_pTransform; 2055cdf0e10cSrcweir } 2056cdf0e10cSrcweir 2057cdf0e10cSrcweir //-------------------------------------------------------------------------- 2058cdf0e10cSrcweir //-------------------------------------------------------------------------- 2059cdf0e10cSrcweir SetupApp* Create_SetupAppX() 2060cdf0e10cSrcweir { 2061cdf0e10cSrcweir return new SetupAppX; 2062cdf0e10cSrcweir } 2063cdf0e10cSrcweir 2064cdf0e10cSrcweir //-------------------------------------------------------------------------- 2065