1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
#include "precomp.h"
static HRESULT PathIsRemote(__in LPCWSTR pTargetFolder, __inout BOOL* fPathRemote);
static HRESULT PathIsRemovable(__in LPCWSTR pTargetFolder, __inout BOOL* fPathRemovable);
/********************************************************************
ValidatePath - Custom Action entry point
********************************************************************/
UINT __stdcall ValidatePath(MSIHANDLE hInstall)
{
HRESULT hr = S_OK;
LPWSTR pwszWixUIDir = NULL;
LPWSTR pwszInstallPath = NULL;
BOOL fInstallPathIsRemote = TRUE;
BOOL fInstallPathIsRemoveable = TRUE;
hr = WcaInitialize(hInstall, "ValidatePath");
ExitOnFailure(hr, "failed to initialize");
hr = WcaGetProperty(L"WIXUI_INSTALLDIR", &pwszWixUIDir);
ExitOnFailure(hr, "failed to get WixUI Installation Directory");
hr = WcaGetProperty(pwszWixUIDir, &pwszInstallPath);
ExitOnFailure(hr, "failed to get Installation Directory");
hr = PathIsRemote(pwszInstallPath, &fInstallPathIsRemote);
if (FAILED(hr))
{
TraceError(hr, "Unable to determine if path is remote");
//reset HR, as we need to continue and find out if is a UNC path
hr = S_OK;
}
hr = PathIsRemovable(pwszInstallPath, &fInstallPathIsRemoveable);
if (FAILED(hr))
{
TraceError(hr, "Unable to determine if path is removable");
//reset HR, as we need to continue and find out if is a UNC path
hr = S_OK;
}
// If the path does not point to a network drive, mapped drive, or removable drive,
// then set WIXUI_INSTALLDIR_VALID to "1" otherwise set it to 0
BOOL fInstallPathIsUnc = ::PathIsUNCW(pwszInstallPath);
if (!fInstallPathIsUnc && !fInstallPathIsRemote && !fInstallPathIsRemoveable)
{
// path is valid
hr = WcaSetProperty(L"WIXUI_INSTALLDIR_VALID", L"1");
ExitOnFailure(hr, "failed to set WIXUI_INSTALLDIR_VALID");
}
else
{
// path is invalid; we can't log it because we're being called from a DoAction control event
// but we can at least call WcaLog to get it to write to the debugger from a debug build
WcaLog(LOGMSG_STANDARD, "Installation path %ls is invalid: it is %s UNC path, %s remote path, or %s path on a removable drive, and must be none of these.",
pwszInstallPath, fInstallPathIsUnc ? "a" : "not a", fInstallPathIsRemote ? "a" : "not a", fInstallPathIsRemoveable ? "a" : "not a");
hr = WcaSetProperty(L"WIXUI_INSTALLDIR_VALID", L"0");
ExitOnFailure(hr, "failed to set WIXUI_INSTALLDIR_VALID");
}
LExit:
ReleaseStr(pwszInstallPath);
ReleaseStr(pwszWixUIDir);
return WcaFinalize(SUCCEEDED(hr) ? ERROR_SUCCESS : ERROR_INSTALL_FAILURE);
}
/********************************************************************
PathIsRemote - helper function for ValidatePath
********************************************************************/
static HRESULT PathIsRemote(__in LPCWSTR pTargetFolder, __inout BOOL* fPathRemote)
{
HRESULT hr = S_OK;
LPWSTR pStrippedTargetFolder = NULL;
hr = StrAllocString(&pStrippedTargetFolder, pTargetFolder, 0);
// Terminate the path at the root
if (!::PathStripToRootW(pStrippedTargetFolder))
{
hr = HRESULT_FROM_WIN32(ERROR_INVALID_DRIVE);
ExitOnFailure(hr, "failed to parse target folder");
}
UINT uResult = ::GetDriveTypeW(pStrippedTargetFolder);
*fPathRemote = (DRIVE_REMOTE == uResult);
LExit:
ReleaseStr(pStrippedTargetFolder);
return hr;
}
/********************************************************************
PathIsRemovable - helper function for ValidatePath
********************************************************************/
static HRESULT PathIsRemovable(__in LPCWSTR pTargetFolder, __inout BOOL* fPathRemovable)
{
HRESULT hr = S_OK;
LPWSTR pStrippedTargetFolder = NULL;
hr = StrAllocString(&pStrippedTargetFolder, pTargetFolder, 0);
// Terminate the path at the root
if (!::PathStripToRootW(pStrippedTargetFolder))
{
hr = HRESULT_FROM_WIN32(ERROR_INVALID_DRIVE);
ExitOnFailure(hr, "failed to parse target folder");
}
UINT uResult = ::GetDriveTypeW(pStrippedTargetFolder);
*fPathRemovable = ((DRIVE_CDROM == uResult) || (DRIVE_REMOVABLE == uResult) || (DRIVE_RAMDISK == uResult) || (DRIVE_UNKNOWN == uResult));
LExit:
ReleaseStr(pStrippedTargetFolder);
return hr;
}
|