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("INSTALLDIR"), 62 TEXT(""), &cchSize); 63 if (result == ERROR_MORE_DATA) { 64 cchSize = cchSize + 1; // NULL termination 65 szValue = new TCHAR[cchSize]; 66 if (szValue) { 67 result = MsiGetProperty(hInstall, TEXT("INSTALLDIR"), 68 szValue, &cchSize); 69 } else { 70 return ERROR_INSTALL_FAILURE; 71 } 72 } 73 74 if (result != ERROR_SUCCESS) { 75 delete [] szValue; 76 return ERROR_INSTALL_FAILURE; 77 } 78 79 if (DirectoryExist(szValue)) { 80 if (PathIsDirectoryEmpty(szValue)) { 81 MsiSetProperty(hInstall, TEXT("INSTALLDIR_VALID"), TEXT("1")); 82 } else { 83 MsiSetProperty(hInstall, TEXT("INSTALLDIR_VALID"), TEXT("0")); 84 } 85 } else { 86 MsiSetProperty(hInstall, TEXT("INSTALLDIR_VALID"), TEXT("1")); 87 } 88 89 delete [] szValue; 90 91 return ERROR_SUCCESS; 92 } 93 }