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 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 65 static inline void OutputDebugStringFormat( LPCSTR, ... ) 66 { 67 } 68 #endif 69 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 88 static void SetMsiProperty( MSIHANDLE handle, const std::_tstring& sProperty ) 89 { 90 MsiSetProperty( handle, sProperty.c_str(), TEXT("1") ); 91 } 92 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