| /************************************************************** |
| * |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| * |
| *************************************************************/ |
| |
| |
| |
| #define UNICODE |
| |
| #ifdef _MSC_VER |
| #pragma warning(push,1) // disable warnings within system headers |
| #endif |
| #include <windows.h> |
| #include <msiquery.h> |
| #ifdef _MSC_VER |
| #pragma warning(pop) |
| #endif |
| |
| #include <string.h> |
| #include <malloc.h> |
| #include <stdio.h> |
| |
| //---------------------------------------------------------- |
| BOOL UnicodeEquals( wchar_t* pStr1, wchar_t* pStr2 ) |
| { |
| if ( pStr1 == NULL && pStr2 == NULL ) |
| return TRUE; |
| else if ( pStr1 == NULL || pStr2 == NULL ) |
| return FALSE; |
| |
| while( *pStr1 == *pStr2 && *pStr1 && *pStr2 ) |
| pStr1++, pStr2++; |
| |
| return ( *pStr1 == 0 && *pStr2 == 0 ); |
| } |
| |
| //---------------------------------------------------------- |
| BOOL GetMsiProp( MSIHANDLE hMSI, const wchar_t* pPropName, wchar_t** ppValue ) |
| { |
| DWORD sz = 0; |
| if ( MsiGetProperty( hMSI, pPropName, L"", &sz ) == ERROR_MORE_DATA ) |
| { |
| sz++; |
| DWORD nbytes = sz * sizeof( wchar_t ); |
| wchar_t* buff = reinterpret_cast<wchar_t*>( malloc( nbytes ) ); |
| ZeroMemory( buff, nbytes ); |
| MsiGetProperty( hMSI, pPropName, buff, &sz ); |
| *ppValue = buff; |
| |
| return TRUE; |
| } |
| |
| return FALSE; |
| } |
| |
| //---------------------------------------------------------- |
| BOOL MakeInstallForAllUsers( MSIHANDLE hMSI ) |
| { |
| BOOL bResult = FALSE; |
| wchar_t* pVal = NULL; |
| if ( GetMsiProp( hMSI, L"ALLUSERS", &pVal ) && pVal ) |
| { |
| bResult = UnicodeEquals( pVal , L"1" ); |
| free( pVal ); |
| } |
| |
| return bResult; |
| } |
| |
| //---------------------------------------------------------- |
| extern "C" UINT __stdcall PatchActiveXControl( MSIHANDLE hMSI ) |
| { |
| // MessageBox(NULL, L"PatchActiveXControl", L"Information", MB_OK | MB_ICONINFORMATION); |
| |
| INSTALLSTATE current_state; |
| INSTALLSTATE future_state; |
| |
| if ( ERROR_SUCCESS == MsiGetFeatureState( hMSI, L"gm_o_Activexcontrol", ¤t_state, &future_state ) ) |
| { |
| BOOL bInstallForAllUsers = MakeInstallForAllUsers( hMSI ); |
| |
| if ( future_state == INSTALLSTATE_LOCAL |
| || ( current_state == INSTALLSTATE_LOCAL && future_state == INSTALLSTATE_UNKNOWN ) ) |
| { |
| HKEY hkey = NULL; |
| char* aSubKey = "Software\\Classes\\MIME\\DataBase\\Content Type\\application/vnd.sun.xml.base"; |
| if ( ERROR_SUCCESS == RegCreateKeyA(bInstallForAllUsers ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER, aSubKey, &hkey) ) |
| { |
| RegDeleteValueA( hkey, "CLSID" ); |
| RegCloseKey( hkey ), hkey = NULL; |
| } |
| } |
| } |
| else |
| { |
| // assert( FALSE ); |
| } |
| |
| return ERROR_SUCCESS; |
| } |
| |
| |