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