1 /* 2 * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 #include <Windows.h> 27 #include <msiquery.h> 28 #include <shlwapi.h> 29 30 extern "C" { 31 32 #ifdef JP_EXPORT_FUNCTION 33 #error Unexpected JP_EXPORT_FUNCTION define 34 #endif 35 #define JP_EXPORT_FUNCTION comment(linker, "/EXPORT:" __FUNCTION__ "=" __FUNCDNAME__) 36 37 BOOL WINAPI DllMain(HINSTANCE hInst, ULONG ulReason, 38 LPVOID lpvReserved) { 39 return TRUE; 40 } 41 42 BOOL DirectoryExist(TCHAR *szValue) { 43 DWORD attr = GetFileAttributes(szValue); 44 if (attr == INVALID_FILE_ATTRIBUTES) { 45 return FALSE; 46 } 47 48 if (attr & FILE_ATTRIBUTE_DIRECTORY) { 49 return TRUE; 50 } 51 52 return FALSE; 53 } 54 55 UINT __stdcall CheckInstallDir(MSIHANDLE hInstall) { 56 #pragma JP_EXPORT_FUNCTION 57 58 TCHAR *szValue = NULL; 59 DWORD cchSize = 0; 60 61 UINT result = MsiGetProperty(hInstall, TEXT("APPLICATIONFOLDER"), TEXT(""), &cchSize); 62 if (result == ERROR_MORE_DATA) { 63 cchSize = cchSize + 1; // NULL termination 64 szValue = new TCHAR[cchSize]; 65 if (szValue) { 66 result = MsiGetProperty(hInstall, TEXT("APPLICATIONFOLDER"), szValue, &cchSize); 67 } else { 68 return ERROR_INSTALL_FAILURE; 69 } 70 } 71 72 if (result != ERROR_SUCCESS) { 73 delete [] szValue; 74 return ERROR_INSTALL_FAILURE; 75 } 76 77 if (DirectoryExist(szValue)) { 78 if (PathIsDirectoryEmpty(szValue)) { 79 MsiSetProperty(hInstall, TEXT("INSTALLDIR_VALID"), TEXT("1")); 80 } else { 81 MsiSetProperty(hInstall, TEXT("INSTALLDIR_VALID"), TEXT("0")); 82 } 83 } else { 84 MsiSetProperty(hInstall, TEXT("INSTALLDIR_VALID"), TEXT("1")); 85 } 86 87 delete [] szValue; 88 89 return ERROR_SUCCESS; 90 } 91 }