1 /**************************************************************
2 *
3 * Licensed to the Apache Software Foundation (ASF) under one
4 * or more contributor license agreements. See the NOTICE file
5 * distributed with this work for additional information
6 * regarding copyright ownership. The ASF licenses this file
7 * to you under the Apache License, Version 2.0 (the
8 * "License"); you may not use this file except in compliance
9 * with the License. You may obtain a copy of the License at
10 *
11 * http://www.apache.org/licenses/LICENSE-2.0
12 *
13 * Unless required by applicable law or agreed to in writing,
14 * software distributed under the License is distributed on an
15 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16 * KIND, either express or implied. See the License for the
17 * specific language governing permissions and limitations
18 * under the License.
19 *
20 *************************************************************/
21
22
23
24 #define _WIN32_WINNT 0x0401
25
26 #ifdef _MSC_VER
27 #pragma warning(push, 1) /* disable warnings within system headers */
28 #endif
29 #define WIN32_LEAN_AND_MEAN
30 #include <windows.h>
31 #include <msiquery.h>
32 #ifdef _MSC_VER
33 #pragma warning(pop)
34 #endif
35
36 #include <malloc.h>
37 #include <assert.h>
38
39 #ifdef UNICODE
40 #define _UNICODE
41 #define _tstring wstring
42 #else
43 #define _tstring string
44 #endif
45 #include <tchar.h>
46 #include <string>
47 #include <queue>
48 #include <stdio.h>
49 #include <strsafe.h>
50
51 #include <systools/win32/uwinapi.h>
52
53 //----------------------------------------------------------
54 #ifdef DEBUG
OutputDebugStringFormat(LPCSTR pFormat,...)55 inline void OutputDebugStringFormat( LPCSTR pFormat, ... )
56 {
57 CHAR buffer[1024];
58 va_list args;
59
60 va_start( args, pFormat );
61 StringCchVPrintfA( buffer, sizeof(buffer), pFormat, args );
62 OutputDebugStringA( buffer );
63 }
64 #else
OutputDebugStringFormat(LPCSTR,...)65 static inline void OutputDebugStringFormat( LPCSTR, ... )
66 {
67 }
68 #endif
69
GetMsiProperty(MSIHANDLE handle,const std::_tstring & sProperty)70 static std::_tstring GetMsiProperty( MSIHANDLE handle, const std::_tstring& sProperty )
71 {
72 std::_tstring result;
73 TCHAR szDummy[1] = TEXT("");
74 DWORD nChars = 0;
75
76 if ( MsiGetProperty( handle, sProperty.c_str(), szDummy, &nChars ) == ERROR_MORE_DATA )
77 {
78 DWORD nBytes = ++nChars * sizeof(TCHAR);
79 LPTSTR buffer = reinterpret_cast<LPTSTR>(_alloca(nBytes));
80 ZeroMemory( buffer, nBytes );
81 MsiGetProperty(handle, sProperty.c_str(), buffer, &nChars);
82 result = buffer;
83 }
84
85 return result;
86 }
87
SetMsiProperty(MSIHANDLE handle,const std::_tstring & sProperty)88 static void SetMsiProperty( MSIHANDLE handle, const std::_tstring& sProperty )
89 {
90 MsiSetProperty( handle, sProperty.c_str(), TEXT("1") );
91 }
92
CheckPatchList(MSIHANDLE handle)93 extern "C" UINT __stdcall CheckPatchList( MSIHANDLE handle )
94 {
95 std::_tstring sPatchList = GetMsiProperty( handle, TEXT("PATCH") );
96 std::_tstring sRequiredPatch = GetMsiProperty( handle, TEXT("PREREQUIREDPATCH") );
97
98 OutputDebugStringFormat( "CheckPatchList called with PATCH=%s and PRQ= %s\n", sPatchList.c_str(), sRequiredPatch.c_str() );
99
100 if ( ( sPatchList.length() != 0 ) && ( sRequiredPatch.length() != 0 ) )
101 {
102 if ( _tcsstr( sPatchList.c_str(), sRequiredPatch.c_str() ) )
103 {
104 SetMsiProperty( handle, TEXT("IGNOREPREREQUIREDPATCH") );
105 OutputDebugStringFormat( "Set Property IgnorePrerequiredPatch!\n" );
106 }
107 }
108 return ERROR_SUCCESS;
109 }
110