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 }