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