xref: /AOO41X/main/desktop/win32/source/setup/setup.cpp (revision 13ac66067bb747a4eb70bcaea767c3d0442347d5)
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