setup_native/source/win32/customactions/shellextensions/checkpatches.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: checkpatches.cxx,v $
  10  * $Revision: 1.1.2.2 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 #define _WIN32_WINNT 0x0401
  32
  33 #ifdef _MSC_VER
  34 #pragma warning(push, 1) /* disable warnings within system headers */
  35 #endif
  36 #define WIN32_LEAN_AND_MEAN
  37 #include <windows.h>
  38 #include <msiquery.h>
  39 #ifdef _MSC_VER
  40 #pragma warning(pop)
  41 #endif
  42
  43 #include <malloc.h>
  44 #include <assert.h>
  45
  46 #ifdef UNICODE
  47 #define _UNICODE
  48 #define _tstring        wstring
  49 #else
  50 #define _tstring        string
  51 #endif
  52 #include <tchar.h>
  53 #include <string>
  54 #include <queue>
  55 #include <stdio.h>
  56 #include <strsafe.h>
  57
  58 #include <systools/win32/uwinapi.h>
  59
  60 //----------------------------------------------------------
  61 #ifdef DEBUG
  62 inline void OutputDebugStringFormat( LPCSTR pFormat, ... )
  63 {
  64     CHAR    buffer[1024];
  65     va_list args;
  66
  67     va_start( args, pFormat );
  68     StringCchVPrintfA( buffer, sizeof(buffer), pFormat, args );
  69     OutputDebugStringA( buffer );
  70 }
  71 #else
  72 static inline void OutputDebugStringFormat( LPCSTR, ... )
  73 {
  74 }
  75 #endif
  76
  77 static std::_tstring GetMsiProperty( MSIHANDLE handle, const std::_tstring& sProperty )
  78 {
  79     std::_tstring result;
  80     TCHAR szDummy[1] = TEXT("");
  81     DWORD nChars = 0;
  82
  83     if ( MsiGetProperty( handle, sProperty.c_str(), szDummy, &nChars ) == ERROR_MORE_DATA )
  84     {
  85         DWORD nBytes = ++nChars * sizeof(TCHAR);
  86         LPTSTR buffer = reinterpret_cast<LPTSTR>(_alloca(nBytes));
  87         ZeroMemory( buffer, nBytes );
  88         MsiGetProperty(handle, sProperty.c_str(), buffer, &nChars);
  89         result = buffer;
  90     }
  91
  92     return result;
  93 }
  94
  95 static void SetMsiProperty( MSIHANDLE handle, const std::_tstring& sProperty )
  96 {
  97     MsiSetProperty( handle, sProperty.c_str(), TEXT("1") );
  98 }
  99
 100 extern "C" UINT __stdcall CheckPatchList( MSIHANDLE handle )
 101 {
 102     std::_tstring sPatchList = GetMsiProperty( handle, TEXT("PATCH") );
 103     std::_tstring sRequiredPatch = GetMsiProperty( handle, TEXT("PREREQUIREDPATCH") );
 104
 105     OutputDebugStringFormat( "CheckPatchList called with PATCH=%s and PRQ= %s\n", sPatchList.c_str(), sRequiredPatch.c_str() );
 106
 107     if ( ( sPatchList.length() != 0 ) && ( sRequiredPatch.length() != 0 ) )
 108     {
 109         if ( _tcsstr( sPatchList.c_str(), sRequiredPatch.c_str() ) )
 110         {
 111             SetMsiProperty( handle, TEXT("IGNOREPREREQUIREDPATCH") );
 112             OutputDebugStringFormat( "Set Property IgnorePrerequiredPatch!\n" );
 113         }
 114     }
 115     return ERROR_SUCCESS;
 116 }