diff options
Diffstat (limited to 'src/ca')
| -rw-r--r-- | src/ca/CustomMsiErrors.h | 130 | ||||
| -rw-r--r-- | src/ca/cost.h | 5 | ||||
| -rw-r--r-- | src/ca/dllmain.cpp | 26 | ||||
| -rw-r--r-- | src/ca/firewall.cpp | 1067 | ||||
| -rw-r--r-- | src/ca/fwca.def | 9 | ||||
| -rw-r--r-- | src/ca/fwca.vcxproj | 72 | ||||
| -rw-r--r-- | src/ca/packages.config | 6 | ||||
| -rw-r--r-- | src/ca/precomp.h | 17 |
8 files changed, 1332 insertions, 0 deletions
diff --git a/src/ca/CustomMsiErrors.h b/src/ca/CustomMsiErrors.h new file mode 100644 index 00000000..f149fb31 --- /dev/null +++ b/src/ca/CustomMsiErrors.h | |||
| @@ -0,0 +1,130 @@ | |||
| 1 | #pragma once | ||
| 2 | // 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. | ||
| 3 | |||
| 4 | |||
| 5 | #define GLOBAL_ERROR_BASE 25501 | ||
| 6 | |||
| 7 | #define msierrSecureObjectsFailedCreateSD 25520 | ||
| 8 | #define msierrSecureObjectsFailedSet 25521 | ||
| 9 | #define msierrSecureObjectsUnknownType 25522 | ||
| 10 | |||
| 11 | #define msierrXmlFileFailedRead 25530 | ||
| 12 | #define msierrXmlFileFailedOpen 25531 | ||
| 13 | #define msierrXmlFileFailedSelect 25532 | ||
| 14 | #define msierrXmlFileFailedSave 25533 | ||
| 15 | |||
| 16 | #define msierrXmlConfigFailedRead 25540 | ||
| 17 | #define msierrXmlConfigFailedOpen 25541 | ||
| 18 | #define msierrXmlConfigFailedSelect 25542 | ||
| 19 | #define msierrXmlConfigFailedSave 25543 | ||
| 20 | |||
| 21 | #define msierrFirewallCannotConnect 25580 | ||
| 22 | |||
| 23 | //--------------------------------------------------------------------------- | ||
| 24 | // Server CustomAction Errors | ||
| 25 | // SERVER range: 26001-26100 | ||
| 26 | #define SERVER_ERROR_BASE 26000 | ||
| 27 | |||
| 28 | #define msierrIISCannotConnect 26001 | ||
| 29 | #define msierrIISFailedReadWebSite 26002 | ||
| 30 | #define msierrIISFailedReadWebDirs 26003 | ||
| 31 | #define msierrIISFailedReadVDirs 26004 | ||
| 32 | #define msierrIISFailedReadFilters 26005 | ||
| 33 | #define msierrIISFailedReadAppPool 26006 | ||
| 34 | #define msierrIISFailedReadMimeMap 26007 | ||
| 35 | #define msierrIISFailedReadProp 26008 | ||
| 36 | #define msierrIISFailedReadWebSvcExt 26009 | ||
| 37 | #define msierrIISFailedReadWebError 26010 | ||
| 38 | #define msierrIISFailedReadHttpHeader 26011 | ||
| 39 | |||
| 40 | #define msierrIISFailedSchedTransaction 26031 | ||
| 41 | #define msierrIISFailedSchedInstallWebs 26032 | ||
| 42 | #define msierrIISFailedSchedInstallWebDirs 26033 | ||
| 43 | #define msierrIISFailedSchedInstallVDirs 26034 | ||
| 44 | #define msierrIISFailedSchedInstallFilters 26035 | ||
| 45 | #define msierrIISFailedSchedInstallAppPool 26036 | ||
| 46 | #define msierrIISFailedSchedInstallProp 26037 | ||
| 47 | #define msierrIISFailedSchedInstallWebSvcExt 26038 | ||
| 48 | |||
| 49 | #define msierrIISFailedSchedUninstallWebs 26051 | ||
| 50 | #define msierrIISFailedSchedUninstallWebDirs 26052 | ||
| 51 | #define msierrIISFailedSchedUninstallVDirs 26053 | ||
| 52 | #define msierrIISFailedSchedUninstallFilters 26054 | ||
| 53 | #define msierrIISFailedSchedUninstallAppPool 26055 | ||
| 54 | #define msierrIISFailedSchedUninstallProp 26056 | ||
| 55 | #define msierrIISFailedSchedUninstallWebSvcExt 26057 | ||
| 56 | |||
| 57 | #define msierrIISFailedStartTransaction 26101 | ||
| 58 | #define msierrIISFailedOpenKey 26102 | ||
| 59 | #define msierrIISFailedCreateKey 26103 | ||
| 60 | #define msierrIISFailedWriteData 26104 | ||
| 61 | #define msierrIISFailedCreateApp 26105 | ||
| 62 | #define msierrIISFailedDeleteKey 26106 | ||
| 63 | #define msierrIISFailedDeleteApp 26107 | ||
| 64 | #define msierrIISFailedDeleteValue 26108 | ||
| 65 | #define msierrIISFailedCommitInUse 26109 | ||
| 66 | |||
| 67 | #define msierrSQLFailedCreateDatabase 26201 | ||
| 68 | #define msierrSQLFailedDropDatabase 26202 | ||
| 69 | #define msierrSQLFailedConnectDatabase 26203 | ||
| 70 | #define msierrSQLFailedExecString 26204 | ||
| 71 | #define msierrSQLDatabaseAlreadyExists 26205 | ||
| 72 | |||
| 73 | #define msierrPERFMONFailedRegisterDLL 26251 | ||
| 74 | #define msierrPERFMONFailedUnregisterDLL 26252 | ||
| 75 | #define msierrInstallPerfCounterData 26253 | ||
| 76 | #define msierrUninstallPerfCounterData 26254 | ||
| 77 | |||
| 78 | #define msierrSMBFailedCreate 26301 | ||
| 79 | #define msierrSMBFailedDrop 26302 | ||
| 80 | |||
| 81 | #define msierrCERTFailedOpen 26351 | ||
| 82 | #define msierrCERTFailedAdd 26352 | ||
| 83 | |||
| 84 | #define msierrUSRFailedUserCreate 26401 | ||
| 85 | #define msierrUSRFailedUserCreatePswd 26402 | ||
| 86 | #define msierrUSRFailedUserGroupAdd 26403 | ||
| 87 | #define msierrUSRFailedUserCreateExists 26404 | ||
| 88 | #define msierrUSRFailedGrantLogonAsService 26405 | ||
| 89 | |||
| 90 | #define msierrDependencyMissingDependencies 26451 | ||
| 91 | #define msierrDependencyHasDependents 26452 | ||
| 92 | |||
| 93 | //-------------------------------------------------------------------------- | ||
| 94 | // Managed code CustomAction Errors | ||
| 95 | // MANAGED range: 27000-27100 | ||
| 96 | #define MANAGED_ERROR_BASE 27000 | ||
| 97 | |||
| 98 | #define msierrDotNetRuntimeRequired 27000 | ||
| 99 | //--------------------------------------------------------------------------- | ||
| 100 | // Public CustomAction Errors | ||
| 101 | // PUBLIC range: 28001-28100 | ||
| 102 | #define PUBLIC_ERROR_BASE 28000 | ||
| 103 | |||
| 104 | #define msierrComPlusCannotConnect 28001 | ||
| 105 | #define msierrComPlusPartitionReadFailed 28002 | ||
| 106 | #define msierrComPlusPartitionRoleReadFailed 28003 | ||
| 107 | #define msierrComPlusUserInPartitionRoleReadFailed 28004 | ||
| 108 | #define msierrComPlusPartitionUserReadFailed 28005 | ||
| 109 | #define msierrComPlusApplicationReadFailed 28006 | ||
| 110 | #define msierrComPlusApplicationRoleReadFailed 28007 | ||
| 111 | #define msierrComPlusUserInApplicationRoleReadFailed 28008 | ||
| 112 | #define msierrComPlusAssembliesReadFailed 28009 | ||
| 113 | #define msierrComPlusSubscriptionReadFailed 28010 | ||
| 114 | #define msierrComPlusPartitionDependency 28011 | ||
| 115 | #define msierrComPlusPartitionNotFound 28012 | ||
| 116 | #define msierrComPlusPartitionIdConflict 28013 | ||
| 117 | #define msierrComPlusPartitionNameConflict 28014 | ||
| 118 | #define msierrComPlusApplicationDependency 28015 | ||
| 119 | #define msierrComPlusApplicationNotFound 28016 | ||
| 120 | #define msierrComPlusApplicationIdConflict 28017 | ||
| 121 | #define msierrComPlusApplicationNameConflict 28018 | ||
| 122 | #define msierrComPlusApplicationRoleDependency 28019 | ||
| 123 | #define msierrComPlusApplicationRoleNotFound 28020 | ||
| 124 | #define msierrComPlusApplicationRoleConflict 28021 | ||
| 125 | #define msierrComPlusAssemblyDependency 28022 | ||
| 126 | #define msierrComPlusSubscriptionIdConflict 28023 | ||
| 127 | #define msierrComPlusSubscriptionNameConflict 28024 | ||
| 128 | #define msierrComPlusFailedLookupNames 28025 | ||
| 129 | |||
| 130 | #define msierrMsmqCannotConnect 28101 | ||
diff --git a/src/ca/cost.h b/src/ca/cost.h new file mode 100644 index 00000000..da68c667 --- /dev/null +++ b/src/ca/cost.h | |||
| @@ -0,0 +1,5 @@ | |||
| 1 | #pragma once | ||
| 2 | // 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. | ||
| 3 | |||
| 4 | |||
| 5 | const UINT COST_FIREWALL_EXCEPTION = 2000; | ||
diff --git a/src/ca/dllmain.cpp b/src/ca/dllmain.cpp new file mode 100644 index 00000000..df53f872 --- /dev/null +++ b/src/ca/dllmain.cpp | |||
| @@ -0,0 +1,26 @@ | |||
| 1 | // 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. | ||
| 2 | |||
| 3 | #include "precomp.h" | ||
| 4 | |||
| 5 | /******************************************************************** | ||
| 6 | DllMain - standard entry point for all WiX CustomActions | ||
| 7 | |||
| 8 | ********************************************************************/ | ||
| 9 | extern "C" BOOL WINAPI DllMain( | ||
| 10 | IN HINSTANCE hInst, | ||
| 11 | IN ULONG ulReason, | ||
| 12 | IN LPVOID) | ||
| 13 | { | ||
| 14 | switch(ulReason) | ||
| 15 | { | ||
| 16 | case DLL_PROCESS_ATTACH: | ||
| 17 | WcaGlobalInitialize(hInst); | ||
| 18 | break; | ||
| 19 | |||
| 20 | case DLL_PROCESS_DETACH: | ||
| 21 | WcaGlobalFinalize(); | ||
| 22 | break; | ||
| 23 | } | ||
| 24 | |||
| 25 | return TRUE; | ||
| 26 | } | ||
diff --git a/src/ca/firewall.cpp b/src/ca/firewall.cpp new file mode 100644 index 00000000..62a5b454 --- /dev/null +++ b/src/ca/firewall.cpp | |||
| @@ -0,0 +1,1067 @@ | |||
| 1 | // 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. | ||
| 2 | |||
| 3 | #include "precomp.h" | ||
| 4 | |||
| 5 | LPCWSTR vcsFirewallExceptionQuery = | ||
| 6 | L"SELECT `Name`, `RemoteAddresses`, `Port`, `Protocol`, `Program`, `Attributes`, `Profile`, `Component_`, `Description` FROM `WixFirewallException`"; | ||
| 7 | enum eFirewallExceptionQuery { feqName = 1, feqRemoteAddresses, feqPort, feqProtocol, feqProgram, feqAttributes, feqProfile, feqComponent, feqDescription }; | ||
| 8 | enum eFirewallExceptionTarget { fetPort = 1, fetApplication, fetUnknown }; | ||
| 9 | enum eFirewallExceptionAttributes { feaIgnoreFailures = 1 }; | ||
| 10 | |||
| 11 | /****************************************************************** | ||
| 12 | SchedFirewallExceptions - immediate custom action worker to | ||
| 13 | register and remove firewall exceptions. | ||
| 14 | |||
| 15 | ********************************************************************/ | ||
| 16 | static UINT SchedFirewallExceptions( | ||
| 17 | __in MSIHANDLE hInstall, | ||
| 18 | WCA_TODO todoSched | ||
| 19 | ) | ||
| 20 | { | ||
| 21 | HRESULT hr = S_OK; | ||
| 22 | UINT er = ERROR_SUCCESS; | ||
| 23 | int cFirewallExceptions = 0; | ||
| 24 | |||
| 25 | PMSIHANDLE hView = NULL; | ||
| 26 | PMSIHANDLE hRec = NULL; | ||
| 27 | |||
| 28 | LPWSTR pwzCustomActionData = NULL; | ||
| 29 | LPWSTR pwzName = NULL; | ||
| 30 | LPWSTR pwzRemoteAddresses = NULL; | ||
| 31 | LPWSTR pwzPort = NULL; | ||
| 32 | int iProtocol = 0; | ||
| 33 | int iAttributes = 0; | ||
| 34 | int iProfile = 0; | ||
| 35 | LPWSTR pwzProgram = NULL; | ||
| 36 | LPWSTR pwzComponent = NULL; | ||
| 37 | LPWSTR pwzFormattedFile = NULL; | ||
| 38 | LPWSTR pwzDescription = NULL; | ||
| 39 | |||
| 40 | // initialize | ||
| 41 | hr = WcaInitialize(hInstall, "SchedFirewallExceptions"); | ||
| 42 | ExitOnFailure(hr, "failed to initialize"); | ||
| 43 | |||
| 44 | // anything to do? | ||
| 45 | if (S_OK != WcaTableExists(L"WixFirewallException")) | ||
| 46 | { | ||
| 47 | WcaLog(LOGMSG_STANDARD, "WixFirewallException table doesn't exist, so there are no firewall exceptions to configure."); | ||
| 48 | ExitFunction(); | ||
| 49 | } | ||
| 50 | |||
| 51 | // query and loop through all the firewall exceptions | ||
| 52 | hr = WcaOpenExecuteView(vcsFirewallExceptionQuery, &hView); | ||
| 53 | ExitOnFailure(hr, "failed to open view on WixFirewallException table"); | ||
| 54 | |||
| 55 | while (S_OK == (hr = WcaFetchRecord(hView, &hRec))) | ||
| 56 | { | ||
| 57 | hr = WcaGetRecordFormattedString(hRec, feqName, &pwzName); | ||
| 58 | ExitOnFailure(hr, "failed to get firewall exception name"); | ||
| 59 | |||
| 60 | hr = WcaGetRecordFormattedString(hRec, feqRemoteAddresses, &pwzRemoteAddresses); | ||
| 61 | ExitOnFailure(hr, "failed to get firewall exception remote addresses"); | ||
| 62 | |||
| 63 | hr = WcaGetRecordFormattedString(hRec, feqPort, &pwzPort); | ||
| 64 | ExitOnFailure(hr, "failed to get firewall exception port"); | ||
| 65 | |||
| 66 | hr = WcaGetRecordInteger(hRec, feqProtocol, &iProtocol); | ||
| 67 | ExitOnFailure(hr, "failed to get firewall exception protocol"); | ||
| 68 | |||
| 69 | hr = WcaGetRecordFormattedString(hRec, feqProgram, &pwzProgram); | ||
| 70 | ExitOnFailure(hr, "failed to get firewall exception program"); | ||
| 71 | |||
| 72 | hr = WcaGetRecordInteger(hRec, feqAttributes, &iAttributes); | ||
| 73 | ExitOnFailure(hr, "failed to get firewall exception attributes"); | ||
| 74 | |||
| 75 | hr = WcaGetRecordInteger(hRec, feqProfile, &iProfile); | ||
| 76 | ExitOnFailure(hr, "failed to get firewall exception profile"); | ||
| 77 | |||
| 78 | hr = WcaGetRecordString(hRec, feqComponent, &pwzComponent); | ||
| 79 | ExitOnFailure(hr, "failed to get firewall exception component"); | ||
| 80 | |||
| 81 | hr = WcaGetRecordString(hRec, feqDescription, &pwzDescription); | ||
| 82 | ExitOnFailure(hr, "failed to get firewall description"); | ||
| 83 | |||
| 84 | // figure out what we're doing for this exception, treating reinstall the same as install | ||
| 85 | WCA_TODO todoComponent = WcaGetComponentToDo(pwzComponent); | ||
| 86 | if ((WCA_TODO_REINSTALL == todoComponent ? WCA_TODO_INSTALL : todoComponent) != todoSched) | ||
| 87 | { | ||
| 88 | WcaLog(LOGMSG_STANDARD, "Component '%ls' action state (%d) doesn't match request (%d)", pwzComponent, todoComponent, todoSched); | ||
| 89 | continue; | ||
| 90 | } | ||
| 91 | |||
| 92 | // action :: name :: profile :: remoteaddresses :: attributes :: target :: {port::protocol | path} | ||
| 93 | ++cFirewallExceptions; | ||
| 94 | hr = WcaWriteIntegerToCaData(todoComponent, &pwzCustomActionData); | ||
| 95 | ExitOnFailure(hr, "failed to write exception action to custom action data"); | ||
| 96 | |||
| 97 | hr = WcaWriteStringToCaData(pwzName, &pwzCustomActionData); | ||
| 98 | ExitOnFailure(hr, "failed to write exception name to custom action data"); | ||
| 99 | |||
| 100 | hr = WcaWriteIntegerToCaData(iProfile, &pwzCustomActionData); | ||
| 101 | ExitOnFailure(hr, "failed to write exception profile to custom action data"); | ||
| 102 | |||
| 103 | hr = WcaWriteStringToCaData(pwzRemoteAddresses, &pwzCustomActionData); | ||
| 104 | ExitOnFailure(hr, "failed to write exception remote addresses to custom action data"); | ||
| 105 | |||
| 106 | hr = WcaWriteIntegerToCaData(iAttributes, &pwzCustomActionData); | ||
| 107 | ExitOnFailure(hr, "failed to write exception attributes to custom action data"); | ||
| 108 | |||
| 109 | if (*pwzProgram) | ||
| 110 | { | ||
| 111 | // If program is defined, we have an application exception. | ||
| 112 | hr = WcaWriteIntegerToCaData(fetApplication, &pwzCustomActionData); | ||
| 113 | ExitOnFailure(hr, "failed to write exception target (application) to custom action data"); | ||
| 114 | |||
| 115 | hr = WcaWriteStringToCaData(pwzProgram, &pwzCustomActionData); | ||
| 116 | ExitOnFailure(hr, "failed to write application path to custom action data"); | ||
| 117 | } | ||
| 118 | else | ||
| 119 | { | ||
| 120 | // we have a port-only exception | ||
| 121 | hr = WcaWriteIntegerToCaData(fetPort, &pwzCustomActionData); | ||
| 122 | ExitOnFailure(hr, "failed to write exception target (port) to custom action data"); | ||
| 123 | } | ||
| 124 | |||
| 125 | hr = WcaWriteStringToCaData(pwzPort, &pwzCustomActionData); | ||
| 126 | ExitOnFailure(hr, "failed to write application path to custom action data"); | ||
| 127 | |||
| 128 | hr = WcaWriteIntegerToCaData(iProtocol, &pwzCustomActionData); | ||
| 129 | ExitOnFailure(hr, "failed to write exception protocol to custom action data"); | ||
| 130 | |||
| 131 | hr = WcaWriteStringToCaData(pwzDescription, &pwzCustomActionData); | ||
| 132 | ExitOnFailure(hr, "failed to write firewall rule description to custom action data"); | ||
| 133 | } | ||
| 134 | |||
| 135 | // reaching the end of the list is actually a good thing, not an error | ||
| 136 | if (E_NOMOREITEMS == hr) | ||
| 137 | { | ||
| 138 | hr = S_OK; | ||
| 139 | } | ||
| 140 | ExitOnFailure(hr, "failure occured while processing WixFirewallException table"); | ||
| 141 | |||
| 142 | // schedule ExecFirewallExceptions if there's anything to do | ||
| 143 | if (pwzCustomActionData && *pwzCustomActionData) | ||
| 144 | { | ||
| 145 | WcaLog(LOGMSG_STANDARD, "Scheduling firewall exception (%ls)", pwzCustomActionData); | ||
| 146 | |||
| 147 | if (WCA_TODO_INSTALL == todoSched) | ||
| 148 | { | ||
| 149 | hr = WcaDoDeferredAction(L"WixRollbackFirewallExceptionsInstall", pwzCustomActionData, cFirewallExceptions * COST_FIREWALL_EXCEPTION); | ||
| 150 | ExitOnFailure(hr, "failed to schedule firewall install exceptions rollback"); | ||
| 151 | hr = WcaDoDeferredAction(L"WixExecFirewallExceptionsInstall", pwzCustomActionData, cFirewallExceptions * COST_FIREWALL_EXCEPTION); | ||
| 152 | ExitOnFailure(hr, "failed to schedule firewall install exceptions execution"); | ||
| 153 | } | ||
| 154 | else | ||
| 155 | { | ||
| 156 | hr = WcaDoDeferredAction(L"WixRollbackFirewallExceptionsUninstall", pwzCustomActionData, cFirewallExceptions * COST_FIREWALL_EXCEPTION); | ||
| 157 | ExitOnFailure(hr, "failed to schedule firewall uninstall exceptions rollback"); | ||
| 158 | hr = WcaDoDeferredAction(L"WixExecFirewallExceptionsUninstall", pwzCustomActionData, cFirewallExceptions * COST_FIREWALL_EXCEPTION); | ||
| 159 | ExitOnFailure(hr, "failed to schedule firewall uninstall exceptions execution"); | ||
| 160 | } | ||
| 161 | } | ||
| 162 | else | ||
| 163 | { | ||
| 164 | WcaLog(LOGMSG_STANDARD, "No firewall exceptions scheduled"); | ||
| 165 | } | ||
| 166 | |||
| 167 | LExit: | ||
| 168 | ReleaseStr(pwzCustomActionData); | ||
| 169 | ReleaseStr(pwzName); | ||
| 170 | ReleaseStr(pwzRemoteAddresses); | ||
| 171 | ReleaseStr(pwzPort); | ||
| 172 | ReleaseStr(pwzProgram); | ||
| 173 | ReleaseStr(pwzComponent); | ||
| 174 | ReleaseStr(pwzDescription); | ||
| 175 | ReleaseStr(pwzFormattedFile); | ||
| 176 | |||
| 177 | return WcaFinalize(er = FAILED(hr) ? ERROR_INSTALL_FAILURE : er); | ||
| 178 | } | ||
| 179 | |||
| 180 | /****************************************************************** | ||
| 181 | SchedFirewallExceptionsInstall - immediate custom action entry | ||
| 182 | point to register firewall exceptions. | ||
| 183 | |||
| 184 | ********************************************************************/ | ||
| 185 | extern "C" UINT __stdcall SchedFirewallExceptionsInstall( | ||
| 186 | __in MSIHANDLE hInstall | ||
| 187 | ) | ||
| 188 | { | ||
| 189 | return SchedFirewallExceptions(hInstall, WCA_TODO_INSTALL); | ||
| 190 | } | ||
| 191 | |||
| 192 | /****************************************************************** | ||
| 193 | SchedFirewallExceptionsUninstall - immediate custom action entry | ||
| 194 | point to remove firewall exceptions. | ||
| 195 | |||
| 196 | ********************************************************************/ | ||
| 197 | extern "C" UINT __stdcall SchedFirewallExceptionsUninstall( | ||
| 198 | __in MSIHANDLE hInstall | ||
| 199 | ) | ||
| 200 | { | ||
| 201 | return SchedFirewallExceptions(hInstall, WCA_TODO_UNINSTALL); | ||
| 202 | } | ||
| 203 | |||
| 204 | /****************************************************************** | ||
| 205 | GetFirewallRules - Get the collection of firewall rules. | ||
| 206 | |||
| 207 | ********************************************************************/ | ||
| 208 | static HRESULT GetFirewallRules( | ||
| 209 | __in BOOL fIgnoreFailures, | ||
| 210 | __out INetFwRules** ppNetFwRules | ||
| 211 | ) | ||
| 212 | { | ||
| 213 | HRESULT hr = S_OK; | ||
| 214 | INetFwPolicy2* pNetFwPolicy2 = NULL; | ||
| 215 | INetFwRules* pNetFwRules = NULL; | ||
| 216 | *ppNetFwRules = NULL; | ||
| 217 | |||
| 218 | do | ||
| 219 | { | ||
| 220 | ReleaseNullObject(pNetFwPolicy2); | ||
| 221 | ReleaseNullObject(pNetFwRules); | ||
| 222 | |||
| 223 | if (SUCCEEDED(hr = ::CoCreateInstance(__uuidof(NetFwPolicy2), NULL, CLSCTX_ALL, __uuidof(INetFwPolicy2), (void**)&pNetFwPolicy2)) && | ||
| 224 | SUCCEEDED(hr = pNetFwPolicy2->get_Rules(&pNetFwRules))) | ||
| 225 | { | ||
| 226 | break; | ||
| 227 | } | ||
| 228 | else if (fIgnoreFailures) | ||
| 229 | { | ||
| 230 | ExitFunction1(hr = S_FALSE); | ||
| 231 | } | ||
| 232 | else | ||
| 233 | { | ||
| 234 | WcaLog(LOGMSG_STANDARD, "Failed to connect to Windows Firewall"); | ||
| 235 | UINT er = WcaErrorMessage(msierrFirewallCannotConnect, hr, INSTALLMESSAGE_ERROR | MB_ABORTRETRYIGNORE, 0); | ||
| 236 | switch (er) | ||
| 237 | { | ||
| 238 | case IDABORT: // exit with the current HRESULT | ||
| 239 | ExitFunction(); | ||
| 240 | case IDRETRY: // clean up and retry the loop | ||
| 241 | hr = S_FALSE; | ||
| 242 | break; | ||
| 243 | case IDIGNORE: // pass S_FALSE back to the caller, who knows how to ignore the failure | ||
| 244 | ExitFunction1(hr = S_FALSE); | ||
| 245 | default: // No UI, so default is to fail. | ||
| 246 | ExitFunction(); | ||
| 247 | } | ||
| 248 | } | ||
| 249 | } while (S_FALSE == hr); | ||
| 250 | |||
| 251 | *ppNetFwRules = pNetFwRules; | ||
| 252 | pNetFwRules = NULL; | ||
| 253 | |||
| 254 | LExit: | ||
| 255 | ReleaseObject(pNetFwPolicy2); | ||
| 256 | ReleaseObject(pNetFwRules); | ||
| 257 | |||
| 258 | return hr; | ||
| 259 | } | ||
| 260 | |||
| 261 | /****************************************************************** | ||
| 262 | CreateFwRuleObject - CoCreate a firewall rule, and set the common set of properties which are shared | ||
| 263 | between port and application firewall rules | ||
| 264 | |||
| 265 | ********************************************************************/ | ||
| 266 | static HRESULT CreateFwRuleObject( | ||
| 267 | __in BSTR bstrName, | ||
| 268 | __in int iProfile, | ||
| 269 | __in_opt LPCWSTR wzRemoteAddresses, | ||
| 270 | __in LPCWSTR wzPort, | ||
| 271 | __in int iProtocol, | ||
| 272 | __in LPCWSTR wzDescription, | ||
| 273 | __out INetFwRule** ppNetFwRule | ||
| 274 | ) | ||
| 275 | { | ||
| 276 | HRESULT hr = S_OK; | ||
| 277 | BSTR bstrRemoteAddresses = NULL; | ||
| 278 | BSTR bstrPort = NULL; | ||
| 279 | BSTR bstrDescription = NULL; | ||
| 280 | INetFwRule* pNetFwRule = NULL; | ||
| 281 | *ppNetFwRule = NULL; | ||
| 282 | |||
| 283 | // convert to BSTRs to make COM happy | ||
| 284 | bstrRemoteAddresses = ::SysAllocString(wzRemoteAddresses); | ||
| 285 | ExitOnNull(bstrRemoteAddresses, hr, E_OUTOFMEMORY, "failed SysAllocString for remote addresses"); | ||
| 286 | bstrPort = ::SysAllocString(wzPort); | ||
| 287 | ExitOnNull(bstrPort, hr, E_OUTOFMEMORY, "failed SysAllocString for port"); | ||
| 288 | bstrDescription = ::SysAllocString(wzDescription); | ||
| 289 | ExitOnNull(bstrDescription, hr, E_OUTOFMEMORY, "failed SysAllocString for description"); | ||
| 290 | |||
| 291 | hr = ::CoCreateInstance(__uuidof(NetFwRule), NULL, CLSCTX_ALL, __uuidof(INetFwRule), (void**)&pNetFwRule); | ||
| 292 | ExitOnFailure(hr, "failed to create NetFwRule object"); | ||
| 293 | |||
| 294 | hr = pNetFwRule->put_Name(bstrName); | ||
| 295 | ExitOnFailure(hr, "failed to set exception name"); | ||
| 296 | |||
| 297 | hr = pNetFwRule->put_Profiles(static_cast<NET_FW_PROFILE_TYPE2>(iProfile)); | ||
| 298 | ExitOnFailure(hr, "failed to set exception profile"); | ||
| 299 | |||
| 300 | if (MSI_NULL_INTEGER != iProtocol) | ||
| 301 | { | ||
| 302 | hr = pNetFwRule->put_Protocol(static_cast<NET_FW_IP_PROTOCOL>(iProtocol)); | ||
| 303 | ExitOnFailure(hr, "failed to set exception protocol"); | ||
| 304 | } | ||
| 305 | |||
| 306 | if (bstrPort && *bstrPort) | ||
| 307 | { | ||
| 308 | hr = pNetFwRule->put_LocalPorts(bstrPort); | ||
| 309 | ExitOnFailure(hr, "failed to set exception port"); | ||
| 310 | } | ||
| 311 | |||
| 312 | if (bstrRemoteAddresses && *bstrRemoteAddresses) | ||
| 313 | { | ||
| 314 | hr = pNetFwRule->put_RemoteAddresses(bstrRemoteAddresses); | ||
| 315 | ExitOnFailure(hr, "failed to set exception remote addresses '%ls'", bstrRemoteAddresses); | ||
| 316 | } | ||
| 317 | |||
| 318 | if (bstrDescription && *bstrDescription) | ||
| 319 | { | ||
| 320 | hr = pNetFwRule->put_Description(bstrDescription); | ||
| 321 | ExitOnFailure(hr, "failed to set exception description '%ls'", bstrDescription); | ||
| 322 | } | ||
| 323 | |||
| 324 | *ppNetFwRule = pNetFwRule; | ||
| 325 | pNetFwRule = NULL; | ||
| 326 | |||
| 327 | LExit: | ||
| 328 | ReleaseBSTR(bstrRemoteAddresses); | ||
| 329 | ReleaseBSTR(bstrPort); | ||
| 330 | ReleaseBSTR(bstrDescription); | ||
| 331 | ReleaseObject(pNetFwRule); | ||
| 332 | |||
| 333 | return hr; | ||
| 334 | } | ||
| 335 | |||
| 336 | /****************************************************************** | ||
| 337 | FSupportProfiles - Returns true if we support profiles on this machine. | ||
| 338 | (Only on Vista or later) | ||
| 339 | |||
| 340 | ********************************************************************/ | ||
| 341 | static BOOL FSupportProfiles() | ||
| 342 | { | ||
| 343 | BOOL fSupportProfiles = FALSE; | ||
| 344 | INetFwRules* pNetFwRules = NULL; | ||
| 345 | |||
| 346 | // We only support profiles if we can co-create an instance of NetFwPolicy2. | ||
| 347 | // This will not work on pre-vista machines. | ||
| 348 | if (SUCCEEDED(GetFirewallRules(TRUE, &pNetFwRules)) && pNetFwRules != NULL) | ||
| 349 | { | ||
| 350 | fSupportProfiles = TRUE; | ||
| 351 | ReleaseObject(pNetFwRules); | ||
| 352 | } | ||
| 353 | |||
| 354 | return fSupportProfiles; | ||
| 355 | } | ||
| 356 | |||
| 357 | /****************************************************************** | ||
| 358 | GetCurrentFirewallProfile - get the active firewall profile as an | ||
| 359 | INetFwProfile, which owns the lists of exceptions we're | ||
| 360 | updating. | ||
| 361 | |||
| 362 | ********************************************************************/ | ||
| 363 | static HRESULT GetCurrentFirewallProfile( | ||
| 364 | __in BOOL fIgnoreFailures, | ||
| 365 | __out INetFwProfile** ppfwProfile | ||
| 366 | ) | ||
| 367 | { | ||
| 368 | HRESULT hr = S_OK; | ||
| 369 | INetFwMgr* pfwMgr = NULL; | ||
| 370 | INetFwPolicy* pfwPolicy = NULL; | ||
| 371 | INetFwProfile* pfwProfile = NULL; | ||
| 372 | *ppfwProfile = NULL; | ||
| 373 | |||
| 374 | do | ||
| 375 | { | ||
| 376 | ReleaseNullObject(pfwPolicy); | ||
| 377 | ReleaseNullObject(pfwMgr); | ||
| 378 | ReleaseNullObject(pfwProfile); | ||
| 379 | |||
| 380 | if (SUCCEEDED(hr = ::CoCreateInstance(__uuidof(NetFwMgr), NULL, CLSCTX_INPROC_SERVER, __uuidof(INetFwMgr), (void**)&pfwMgr)) && | ||
| 381 | SUCCEEDED(hr = pfwMgr->get_LocalPolicy(&pfwPolicy)) && | ||
| 382 | SUCCEEDED(hr = pfwPolicy->get_CurrentProfile(&pfwProfile))) | ||
| 383 | { | ||
| 384 | break; | ||
| 385 | } | ||
| 386 | else if (fIgnoreFailures) | ||
| 387 | { | ||
| 388 | ExitFunction1(hr = S_FALSE); | ||
| 389 | } | ||
| 390 | else | ||
| 391 | { | ||
| 392 | WcaLog(LOGMSG_STANDARD, "Failed to connect to Windows Firewall"); | ||
| 393 | UINT er = WcaErrorMessage(msierrFirewallCannotConnect, hr, INSTALLMESSAGE_ERROR | MB_ABORTRETRYIGNORE, 0); | ||
| 394 | switch (er) | ||
| 395 | { | ||
| 396 | case IDABORT: // exit with the current HRESULT | ||
| 397 | ExitFunction(); | ||
| 398 | case IDRETRY: // clean up and retry the loop | ||
| 399 | hr = S_FALSE; | ||
| 400 | break; | ||
| 401 | case IDIGNORE: // pass S_FALSE back to the caller, who knows how to ignore the failure | ||
| 402 | ExitFunction1(hr = S_FALSE); | ||
| 403 | default: // No UI, so default is to fail. | ||
| 404 | ExitFunction(); | ||
| 405 | } | ||
| 406 | } | ||
| 407 | } while (S_FALSE == hr); | ||
| 408 | |||
| 409 | *ppfwProfile = pfwProfile; | ||
| 410 | pfwProfile = NULL; | ||
| 411 | |||
| 412 | LExit: | ||
| 413 | ReleaseObject(pfwPolicy); | ||
| 414 | ReleaseObject(pfwMgr); | ||
| 415 | ReleaseObject(pfwProfile); | ||
| 416 | |||
| 417 | return hr; | ||
| 418 | } | ||
| 419 | |||
| 420 | /****************************************************************** | ||
| 421 | AddApplicationException | ||
| 422 | |||
| 423 | ********************************************************************/ | ||
| 424 | static HRESULT AddApplicationException( | ||
| 425 | __in LPCWSTR wzFile, | ||
| 426 | __in LPCWSTR wzName, | ||
| 427 | __in int iProfile, | ||
| 428 | __in_opt LPCWSTR wzRemoteAddresses, | ||
| 429 | __in BOOL fIgnoreFailures, | ||
| 430 | __in LPCWSTR wzPort, | ||
| 431 | __in int iProtocol, | ||
| 432 | __in LPCWSTR wzDescription | ||
| 433 | ) | ||
| 434 | { | ||
| 435 | HRESULT hr = S_OK; | ||
| 436 | BSTR bstrFile = NULL; | ||
| 437 | BSTR bstrName = NULL; | ||
| 438 | INetFwRules* pNetFwRules = NULL; | ||
| 439 | INetFwRule* pNetFwRule = NULL; | ||
| 440 | |||
| 441 | // convert to BSTRs to make COM happy | ||
| 442 | bstrFile = ::SysAllocString(wzFile); | ||
| 443 | ExitOnNull(bstrFile, hr, E_OUTOFMEMORY, "failed SysAllocString for path"); | ||
| 444 | bstrName = ::SysAllocString(wzName); | ||
| 445 | ExitOnNull(bstrName, hr, E_OUTOFMEMORY, "failed SysAllocString for name"); | ||
| 446 | |||
| 447 | // get the collection of firewall rules | ||
| 448 | hr = GetFirewallRules(fIgnoreFailures, &pNetFwRules); | ||
| 449 | ExitOnFailure(hr, "failed to get firewall rules object"); | ||
| 450 | if (S_FALSE == hr) // user or package author chose to ignore missing firewall | ||
| 451 | { | ||
| 452 | ExitFunction(); | ||
| 453 | } | ||
| 454 | |||
| 455 | // try to find it (i.e., support reinstall) | ||
| 456 | hr = pNetFwRules->Item(bstrName, &pNetFwRule); | ||
| 457 | if (HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) == hr) | ||
| 458 | { | ||
| 459 | hr = CreateFwRuleObject(bstrName, iProfile, wzRemoteAddresses, wzPort, iProtocol, wzDescription, &pNetFwRule); | ||
| 460 | ExitOnFailure(hr, "failed to create FwRule object"); | ||
| 461 | |||
| 462 | // set edge traversal to true | ||
| 463 | hr = pNetFwRule->put_EdgeTraversal(VARIANT_TRUE); | ||
| 464 | ExitOnFailure(hr, "failed to set application exception edgetraversal property"); | ||
| 465 | |||
| 466 | // set path | ||
| 467 | hr = pNetFwRule->put_ApplicationName(bstrFile); | ||
| 468 | ExitOnFailure(hr, "failed to set application name"); | ||
| 469 | |||
| 470 | // enable it | ||
| 471 | hr = pNetFwRule->put_Enabled(VARIANT_TRUE); | ||
| 472 | ExitOnFailure(hr, "failed to to enable application exception"); | ||
| 473 | |||
| 474 | // add it to the list of authorized apps | ||
| 475 | hr = pNetFwRules->Add(pNetFwRule); | ||
| 476 | ExitOnFailure(hr, "failed to add app to the authorized apps list"); | ||
| 477 | } | ||
| 478 | else | ||
| 479 | { | ||
| 480 | // we found an existing app exception (if we succeeded, that is) | ||
| 481 | ExitOnFailure(hr, "failed trying to find existing app"); | ||
| 482 | |||
| 483 | // enable it (just in case it was disabled) | ||
| 484 | pNetFwRule->put_Enabled(VARIANT_TRUE); | ||
| 485 | } | ||
| 486 | |||
| 487 | LExit: | ||
| 488 | ReleaseBSTR(bstrName); | ||
| 489 | ReleaseBSTR(bstrFile); | ||
| 490 | ReleaseObject(pNetFwRules); | ||
| 491 | ReleaseObject(pNetFwRule); | ||
| 492 | |||
| 493 | return fIgnoreFailures ? S_OK : hr; | ||
| 494 | } | ||
| 495 | |||
| 496 | /****************************************************************** | ||
| 497 | AddApplicationExceptionOnCurrentProfile | ||
| 498 | |||
| 499 | ********************************************************************/ | ||
| 500 | static HRESULT AddApplicationExceptionOnCurrentProfile( | ||
| 501 | __in LPCWSTR wzFile, | ||
| 502 | __in LPCWSTR wzName, | ||
| 503 | __in_opt LPCWSTR wzRemoteAddresses, | ||
| 504 | __in BOOL fIgnoreFailures | ||
| 505 | ) | ||
| 506 | { | ||
| 507 | HRESULT hr = S_OK; | ||
| 508 | BSTR bstrFile = NULL; | ||
| 509 | BSTR bstrName = NULL; | ||
| 510 | BSTR bstrRemoteAddresses = NULL; | ||
| 511 | INetFwProfile* pfwProfile = NULL; | ||
| 512 | INetFwAuthorizedApplications* pfwApps = NULL; | ||
| 513 | INetFwAuthorizedApplication* pfwApp = NULL; | ||
| 514 | |||
| 515 | // convert to BSTRs to make COM happy | ||
| 516 | bstrFile = ::SysAllocString(wzFile); | ||
| 517 | ExitOnNull(bstrFile, hr, E_OUTOFMEMORY, "failed SysAllocString for path"); | ||
| 518 | bstrName = ::SysAllocString(wzName); | ||
| 519 | ExitOnNull(bstrName, hr, E_OUTOFMEMORY, "failed SysAllocString for name"); | ||
| 520 | bstrRemoteAddresses = ::SysAllocString(wzRemoteAddresses); | ||
| 521 | ExitOnNull(bstrRemoteAddresses, hr, E_OUTOFMEMORY, "failed SysAllocString for remote addresses"); | ||
| 522 | |||
| 523 | // get the firewall profile, which is our entry point for adding exceptions | ||
| 524 | hr = GetCurrentFirewallProfile(fIgnoreFailures, &pfwProfile); | ||
| 525 | ExitOnFailure(hr, "failed to get firewall profile"); | ||
| 526 | if (S_FALSE == hr) // user or package author chose to ignore missing firewall | ||
| 527 | { | ||
| 528 | ExitFunction(); | ||
| 529 | } | ||
| 530 | |||
| 531 | // first, let's see if the app is already on the exception list | ||
| 532 | hr = pfwProfile->get_AuthorizedApplications(&pfwApps); | ||
| 533 | ExitOnFailure(hr, "failed to get list of authorized apps"); | ||
| 534 | |||
| 535 | // try to find it (i.e., support reinstall) | ||
| 536 | hr = pfwApps->Item(bstrFile, &pfwApp); | ||
| 537 | if (HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) == hr) | ||
| 538 | { | ||
| 539 | // not found, so we get to add it | ||
| 540 | hr = ::CoCreateInstance(__uuidof(NetFwAuthorizedApplication), NULL, CLSCTX_INPROC_SERVER, __uuidof(INetFwAuthorizedApplication), reinterpret_cast<void**>(&pfwApp)); | ||
| 541 | ExitOnFailure(hr, "failed to create authorized app"); | ||
| 542 | |||
| 543 | // set the display name | ||
| 544 | hr = pfwApp->put_Name(bstrName); | ||
| 545 | ExitOnFailure(hr, "failed to set authorized app name"); | ||
| 546 | |||
| 547 | // set path | ||
| 548 | hr = pfwApp->put_ProcessImageFileName(bstrFile); | ||
| 549 | ExitOnFailure(hr, "failed to set authorized app path"); | ||
| 550 | |||
| 551 | // set the allowed remote addresses | ||
| 552 | if (bstrRemoteAddresses && *bstrRemoteAddresses) | ||
| 553 | { | ||
| 554 | hr = pfwApp->put_RemoteAddresses(bstrRemoteAddresses); | ||
| 555 | ExitOnFailure(hr, "failed to set authorized app remote addresses"); | ||
| 556 | } | ||
| 557 | |||
| 558 | // add it to the list of authorized apps | ||
| 559 | hr = pfwApps->Add(pfwApp); | ||
| 560 | ExitOnFailure(hr, "failed to add app to the authorized apps list"); | ||
| 561 | } | ||
| 562 | else | ||
| 563 | { | ||
| 564 | // we found an existing app exception (if we succeeded, that is) | ||
| 565 | ExitOnFailure(hr, "failed trying to find existing app"); | ||
| 566 | |||
| 567 | // enable it (just in case it was disabled) | ||
| 568 | pfwApp->put_Enabled(VARIANT_TRUE); | ||
| 569 | } | ||
| 570 | |||
| 571 | LExit: | ||
| 572 | ReleaseBSTR(bstrRemoteAddresses); | ||
| 573 | ReleaseBSTR(bstrName); | ||
| 574 | ReleaseBSTR(bstrFile); | ||
| 575 | ReleaseObject(pfwApp); | ||
| 576 | ReleaseObject(pfwApps); | ||
| 577 | ReleaseObject(pfwProfile); | ||
| 578 | |||
| 579 | return fIgnoreFailures ? S_OK : hr; | ||
| 580 | } | ||
| 581 | |||
| 582 | /****************************************************************** | ||
| 583 | AddPortException | ||
| 584 | |||
| 585 | ********************************************************************/ | ||
| 586 | static HRESULT AddPortException( | ||
| 587 | __in LPCWSTR wzName, | ||
| 588 | __in int iProfile, | ||
| 589 | __in_opt LPCWSTR wzRemoteAddresses, | ||
| 590 | __in BOOL fIgnoreFailures, | ||
| 591 | __in LPCWSTR wzPort, | ||
| 592 | __in int iProtocol, | ||
| 593 | __in LPCWSTR wzDescription | ||
| 594 | ) | ||
| 595 | { | ||
| 596 | HRESULT hr = S_OK; | ||
| 597 | BSTR bstrName = NULL; | ||
| 598 | INetFwRules* pNetFwRules = NULL; | ||
| 599 | INetFwRule* pNetFwRule = NULL; | ||
| 600 | |||
| 601 | // convert to BSTRs to make COM happy | ||
| 602 | bstrName = ::SysAllocString(wzName); | ||
| 603 | ExitOnNull(bstrName, hr, E_OUTOFMEMORY, "failed SysAllocString for name"); | ||
| 604 | |||
| 605 | // get the collection of firewall rules | ||
| 606 | hr = GetFirewallRules(fIgnoreFailures, &pNetFwRules); | ||
| 607 | ExitOnFailure(hr, "failed to get firewall rules object"); | ||
| 608 | if (S_FALSE == hr) // user or package author chose to ignore missing firewall | ||
| 609 | { | ||
| 610 | ExitFunction(); | ||
| 611 | } | ||
| 612 | |||
| 613 | // try to find it (i.e., support reinstall) | ||
| 614 | hr = pNetFwRules->Item(bstrName, &pNetFwRule); | ||
| 615 | if (HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) == hr) | ||
| 616 | { | ||
| 617 | hr = CreateFwRuleObject(bstrName, iProfile, wzRemoteAddresses, wzPort, iProtocol, wzDescription, &pNetFwRule); | ||
| 618 | ExitOnFailure(hr, "failed to create FwRule object"); | ||
| 619 | |||
| 620 | // enable it | ||
| 621 | hr = pNetFwRule->put_Enabled(VARIANT_TRUE); | ||
| 622 | ExitOnFailure(hr, "failed to to enable port exception"); | ||
| 623 | |||
| 624 | // add it to the list of authorized ports | ||
| 625 | hr = pNetFwRules->Add(pNetFwRule); | ||
| 626 | ExitOnFailure(hr, "failed to add app to the authorized ports list"); | ||
| 627 | } | ||
| 628 | else | ||
| 629 | { | ||
| 630 | // we found an existing port exception (if we succeeded, that is) | ||
| 631 | ExitOnFailure(hr, "failed trying to find existing port rule"); | ||
| 632 | |||
| 633 | // enable it (just in case it was disabled) | ||
| 634 | pNetFwRule->put_Enabled(VARIANT_TRUE); | ||
| 635 | } | ||
| 636 | |||
| 637 | LExit: | ||
| 638 | ReleaseBSTR(bstrName); | ||
| 639 | ReleaseObject(pNetFwRules); | ||
| 640 | ReleaseObject(pNetFwRule); | ||
| 641 | |||
| 642 | return fIgnoreFailures ? S_OK : hr; | ||
| 643 | } | ||
| 644 | |||
| 645 | /****************************************************************** | ||
| 646 | AddPortExceptionOnCurrentProfile | ||
| 647 | |||
| 648 | ********************************************************************/ | ||
| 649 | static HRESULT AddPortExceptionOnCurrentProfile( | ||
| 650 | __in LPCWSTR wzName, | ||
| 651 | __in_opt LPCWSTR wzRemoteAddresses, | ||
| 652 | __in BOOL fIgnoreFailures, | ||
| 653 | __in int iPort, | ||
| 654 | __in int iProtocol | ||
| 655 | ) | ||
| 656 | { | ||
| 657 | HRESULT hr = S_OK; | ||
| 658 | BSTR bstrName = NULL; | ||
| 659 | BSTR bstrRemoteAddresses = NULL; | ||
| 660 | INetFwProfile* pfwProfile = NULL; | ||
| 661 | INetFwOpenPorts* pfwPorts = NULL; | ||
| 662 | INetFwOpenPort* pfwPort = NULL; | ||
| 663 | |||
| 664 | // convert to BSTRs to make COM happy | ||
| 665 | bstrName = ::SysAllocString(wzName); | ||
| 666 | ExitOnNull(bstrName, hr, E_OUTOFMEMORY, "failed SysAllocString for name"); | ||
| 667 | bstrRemoteAddresses = ::SysAllocString(wzRemoteAddresses); | ||
| 668 | ExitOnNull(bstrRemoteAddresses, hr, E_OUTOFMEMORY, "failed SysAllocString for remote addresses"); | ||
| 669 | |||
| 670 | // create and initialize a new open port object | ||
| 671 | hr = ::CoCreateInstance(__uuidof(NetFwOpenPort), NULL, CLSCTX_INPROC_SERVER, __uuidof(INetFwOpenPort), reinterpret_cast<void**>(&pfwPort)); | ||
| 672 | ExitOnFailure(hr, "failed to create new open port"); | ||
| 673 | |||
| 674 | hr = pfwPort->put_Port(iPort); | ||
| 675 | ExitOnFailure(hr, "failed to set exception port"); | ||
| 676 | |||
| 677 | hr = pfwPort->put_Protocol(static_cast<NET_FW_IP_PROTOCOL>(iProtocol)); | ||
| 678 | ExitOnFailure(hr, "failed to set exception protocol"); | ||
| 679 | |||
| 680 | if (bstrRemoteAddresses && *bstrRemoteAddresses) | ||
| 681 | { | ||
| 682 | hr = pfwPort->put_RemoteAddresses(bstrRemoteAddresses); | ||
| 683 | ExitOnFailure(hr, "failed to set exception remote addresses '%ls'", bstrRemoteAddresses); | ||
| 684 | } | ||
| 685 | |||
| 686 | hr = pfwPort->put_Name(bstrName); | ||
| 687 | ExitOnFailure(hr, "failed to set exception name"); | ||
| 688 | |||
| 689 | // get the firewall profile, its current list of open ports, and add ours | ||
| 690 | hr = GetCurrentFirewallProfile(fIgnoreFailures, &pfwProfile); | ||
| 691 | ExitOnFailure(hr, "failed to get firewall profile"); | ||
| 692 | if (S_FALSE == hr) // user or package author chose to ignore missing firewall | ||
| 693 | { | ||
| 694 | ExitFunction(); | ||
| 695 | } | ||
| 696 | |||
| 697 | hr = pfwProfile->get_GloballyOpenPorts(&pfwPorts); | ||
| 698 | ExitOnFailure(hr, "failed to get open ports"); | ||
| 699 | |||
| 700 | hr = pfwPorts->Add(pfwPort); | ||
| 701 | ExitOnFailure(hr, "failed to add exception to global list"); | ||
| 702 | |||
| 703 | LExit: | ||
| 704 | ReleaseBSTR(bstrRemoteAddresses); | ||
| 705 | ReleaseBSTR(bstrName); | ||
| 706 | ReleaseObject(pfwProfile); | ||
| 707 | ReleaseObject(pfwPorts); | ||
| 708 | ReleaseObject(pfwPort); | ||
| 709 | |||
| 710 | return fIgnoreFailures ? S_OK : hr; | ||
| 711 | } | ||
| 712 | |||
| 713 | /****************************************************************** | ||
| 714 | RemoveException - Removes the exception rule with the given name. | ||
| 715 | |||
| 716 | ********************************************************************/ | ||
| 717 | static HRESULT RemoveException( | ||
| 718 | __in LPCWSTR wzName, | ||
| 719 | __in BOOL fIgnoreFailures | ||
| 720 | ) | ||
| 721 | { | ||
| 722 | HRESULT hr = S_OK;; | ||
| 723 | INetFwRules* pNetFwRules = NULL; | ||
| 724 | |||
| 725 | // convert to BSTRs to make COM happy | ||
| 726 | BSTR bstrName = ::SysAllocString(wzName); | ||
| 727 | ExitOnNull(bstrName, hr, E_OUTOFMEMORY, "failed SysAllocString for path"); | ||
| 728 | |||
| 729 | // get the collection of firewall rules | ||
| 730 | hr = GetFirewallRules(fIgnoreFailures, &pNetFwRules); | ||
| 731 | ExitOnFailure(hr, "failed to get firewall rules object"); | ||
| 732 | if (S_FALSE == hr) // user or package author chose to ignore missing firewall | ||
| 733 | { | ||
| 734 | ExitFunction(); | ||
| 735 | } | ||
| 736 | |||
| 737 | hr = pNetFwRules->Remove(bstrName); | ||
| 738 | ExitOnFailure(hr, "failed to remove authorized app"); | ||
| 739 | |||
| 740 | LExit: | ||
| 741 | ReleaseBSTR(bstrName); | ||
| 742 | ReleaseObject(pNetFwRules); | ||
| 743 | |||
| 744 | return fIgnoreFailures ? S_OK : hr; | ||
| 745 | } | ||
| 746 | |||
| 747 | /****************************************************************** | ||
| 748 | RemoveApplicationExceptionFromCurrentProfile | ||
| 749 | |||
| 750 | ********************************************************************/ | ||
| 751 | static HRESULT RemoveApplicationExceptionFromCurrentProfile( | ||
| 752 | __in LPCWSTR wzFile, | ||
| 753 | __in BOOL fIgnoreFailures | ||
| 754 | ) | ||
| 755 | { | ||
| 756 | HRESULT hr = S_OK; | ||
| 757 | INetFwProfile* pfwProfile = NULL; | ||
| 758 | INetFwAuthorizedApplications* pfwApps = NULL; | ||
| 759 | |||
| 760 | // convert to BSTRs to make COM happy | ||
| 761 | BSTR bstrFile = ::SysAllocString(wzFile); | ||
| 762 | ExitOnNull(bstrFile, hr, E_OUTOFMEMORY, "failed SysAllocString for path"); | ||
| 763 | |||
| 764 | // get the firewall profile, which is our entry point for removing exceptions | ||
| 765 | hr = GetCurrentFirewallProfile(fIgnoreFailures, &pfwProfile); | ||
| 766 | ExitOnFailure(hr, "failed to get firewall profile"); | ||
| 767 | if (S_FALSE == hr) // user or package author chose to ignore missing firewall | ||
| 768 | { | ||
| 769 | ExitFunction(); | ||
| 770 | } | ||
| 771 | |||
| 772 | // now get the list of app exceptions and remove the one | ||
| 773 | hr = pfwProfile->get_AuthorizedApplications(&pfwApps); | ||
| 774 | ExitOnFailure(hr, "failed to get list of authorized apps"); | ||
| 775 | |||
| 776 | hr = pfwApps->Remove(bstrFile); | ||
| 777 | ExitOnFailure(hr, "failed to remove authorized app"); | ||
| 778 | |||
| 779 | LExit: | ||
| 780 | ReleaseBSTR(bstrFile); | ||
| 781 | ReleaseObject(pfwApps); | ||
| 782 | ReleaseObject(pfwProfile); | ||
| 783 | |||
| 784 | return fIgnoreFailures ? S_OK : hr; | ||
| 785 | } | ||
| 786 | |||
| 787 | /****************************************************************** | ||
| 788 | RemovePortExceptionFromCurrentProfile | ||
| 789 | |||
| 790 | ********************************************************************/ | ||
| 791 | static HRESULT RemovePortExceptionFromCurrentProfile( | ||
| 792 | __in int iPort, | ||
| 793 | __in int iProtocol, | ||
| 794 | __in BOOL fIgnoreFailures | ||
| 795 | ) | ||
| 796 | { | ||
| 797 | HRESULT hr = S_OK; | ||
| 798 | INetFwProfile* pfwProfile = NULL; | ||
| 799 | INetFwOpenPorts* pfwPorts = NULL; | ||
| 800 | |||
| 801 | // get the firewall profile, which is our entry point for adding exceptions | ||
| 802 | hr = GetCurrentFirewallProfile(fIgnoreFailures, &pfwProfile); | ||
| 803 | ExitOnFailure(hr, "failed to get firewall profile"); | ||
| 804 | if (S_FALSE == hr) // user or package author chose to ignore missing firewall | ||
| 805 | { | ||
| 806 | ExitFunction(); | ||
| 807 | } | ||
| 808 | |||
| 809 | hr = pfwProfile->get_GloballyOpenPorts(&pfwPorts); | ||
| 810 | ExitOnFailure(hr, "failed to get open ports"); | ||
| 811 | |||
| 812 | hr = pfwPorts->Remove(iPort, static_cast<NET_FW_IP_PROTOCOL>(iProtocol)); | ||
| 813 | ExitOnFailure(hr, "failed to remove open port %d, protocol %d", iPort, iProtocol); | ||
| 814 | |||
| 815 | LExit: | ||
| 816 | return fIgnoreFailures ? S_OK : hr; | ||
| 817 | } | ||
| 818 | |||
| 819 | static HRESULT AddApplicationException( | ||
| 820 | __in BOOL fSupportProfiles, | ||
| 821 | __in LPCWSTR wzFile, | ||
| 822 | __in LPCWSTR wzName, | ||
| 823 | __in int iProfile, | ||
| 824 | __in_opt LPCWSTR wzRemoteAddresses, | ||
| 825 | __in BOOL fIgnoreFailures, | ||
| 826 | __in LPCWSTR wzPort, | ||
| 827 | __in int iProtocol, | ||
| 828 | __in LPCWSTR wzDescription | ||
| 829 | ) | ||
| 830 | { | ||
| 831 | HRESULT hr = S_OK; | ||
| 832 | |||
| 833 | if (fSupportProfiles) | ||
| 834 | { | ||
| 835 | hr = AddApplicationException(wzFile, wzName, iProfile, wzRemoteAddresses, fIgnoreFailures, wzPort, iProtocol, wzDescription); | ||
| 836 | } | ||
| 837 | else | ||
| 838 | { | ||
| 839 | if (0 != *wzPort || MSI_NULL_INTEGER != iProtocol) | ||
| 840 | { | ||
| 841 | // NOTE: This is treated as an error rather than either creating a rule based on just the application (no port), or | ||
| 842 | // just the port because it is unclear what is the proper fall back. For example, suppose that you have code that | ||
| 843 | // runs in dllhost.exe. Clearly falling back to opening all of dllhost is wrong. Because the firewall is a security | ||
| 844 | // feature, it seems better to require the MSI author to indicate the behavior that they want. | ||
| 845 | WcaLog(LOGMSG_STANDARD, "FirewallExtension: Cannot add firewall rule '%ls', which defines both an application and a port or protocol. Such a rule requires Microsoft Windows Vista or later.", wzName); | ||
| 846 | return fIgnoreFailures ? S_OK : E_NOTIMPL; | ||
| 847 | } | ||
| 848 | |||
| 849 | hr = AddApplicationExceptionOnCurrentProfile(wzFile, wzName, wzRemoteAddresses, fIgnoreFailures); | ||
| 850 | } | ||
| 851 | |||
| 852 | return hr; | ||
| 853 | } | ||
| 854 | |||
| 855 | static HRESULT AddPortException( | ||
| 856 | __in BOOL fSupportProfiles, | ||
| 857 | __in LPCWSTR wzName, | ||
| 858 | __in int iProfile, | ||
| 859 | __in_opt LPCWSTR wzRemoteAddresses, | ||
| 860 | __in BOOL fIgnoreFailures, | ||
| 861 | __in LPCWSTR wzPort, | ||
| 862 | __in int iProtocol, | ||
| 863 | __in LPCWSTR wzDescription | ||
| 864 | ) | ||
| 865 | { | ||
| 866 | HRESULT hr = S_OK; | ||
| 867 | |||
| 868 | if (fSupportProfiles) | ||
| 869 | { | ||
| 870 | hr = AddPortException(wzName, iProfile, wzRemoteAddresses, fIgnoreFailures, wzPort, iProtocol, wzDescription); | ||
| 871 | } | ||
| 872 | else | ||
| 873 | { | ||
| 874 | hr = AddPortExceptionOnCurrentProfile(wzName, wzRemoteAddresses, fIgnoreFailures, wcstol(wzPort, NULL, 10), iProtocol); | ||
| 875 | } | ||
| 876 | |||
| 877 | return hr; | ||
| 878 | } | ||
| 879 | |||
| 880 | static HRESULT RemoveApplicationException( | ||
| 881 | __in BOOL fSupportProfiles, | ||
| 882 | __in LPCWSTR wzName, | ||
| 883 | __in LPCWSTR wzFile, | ||
| 884 | __in BOOL fIgnoreFailures, | ||
| 885 | __in LPCWSTR wzPort, | ||
| 886 | __in int iProtocol | ||
| 887 | ) | ||
| 888 | { | ||
| 889 | HRESULT hr = S_OK; | ||
| 890 | |||
| 891 | if (fSupportProfiles) | ||
| 892 | { | ||
| 893 | hr = RemoveException(wzName, fIgnoreFailures); | ||
| 894 | } | ||
| 895 | else | ||
| 896 | { | ||
| 897 | if (0 != *wzPort || MSI_NULL_INTEGER != iProtocol) | ||
| 898 | { | ||
| 899 | WcaLog(LOGMSG_STANDARD, "FirewallExtension: Cannot remove firewall rule '%ls', which defines both an application and a port or protocol. Such a rule requires Microsoft Windows Vista or later.", wzName); | ||
| 900 | return S_OK; | ||
| 901 | } | ||
| 902 | |||
| 903 | hr = RemoveApplicationExceptionFromCurrentProfile(wzFile, fIgnoreFailures); | ||
| 904 | } | ||
| 905 | |||
| 906 | return hr; | ||
| 907 | } | ||
| 908 | |||
| 909 | static HRESULT RemovePortException( | ||
| 910 | __in BOOL fSupportProfiles, | ||
| 911 | __in LPCWSTR wzName, | ||
| 912 | __in LPCWSTR wzPort, | ||
| 913 | __in int iProtocol, | ||
| 914 | __in BOOL fIgnoreFailures | ||
| 915 | ) | ||
| 916 | { | ||
| 917 | HRESULT hr = S_OK; | ||
| 918 | |||
| 919 | if (fSupportProfiles) | ||
| 920 | { | ||
| 921 | hr = RemoveException(wzName, fIgnoreFailures); | ||
| 922 | } | ||
| 923 | else | ||
| 924 | { | ||
| 925 | hr = RemovePortExceptionFromCurrentProfile(wcstol(wzPort, NULL, 10), iProtocol, fIgnoreFailures); | ||
| 926 | } | ||
| 927 | |||
| 928 | return hr; | ||
| 929 | } | ||
| 930 | |||
| 931 | /****************************************************************** | ||
| 932 | ExecFirewallExceptions - deferred custom action entry point to | ||
| 933 | register and remove firewall exceptions. | ||
| 934 | |||
| 935 | ********************************************************************/ | ||
| 936 | extern "C" UINT __stdcall ExecFirewallExceptions( | ||
| 937 | __in MSIHANDLE hInstall | ||
| 938 | ) | ||
| 939 | { | ||
| 940 | HRESULT hr = S_OK; | ||
| 941 | BOOL fSupportProfiles = FALSE; | ||
| 942 | LPWSTR pwz = NULL; | ||
| 943 | LPWSTR pwzCustomActionData = NULL; | ||
| 944 | int iTodo = WCA_TODO_UNKNOWN; | ||
| 945 | LPWSTR pwzName = NULL; | ||
| 946 | LPWSTR pwzRemoteAddresses = NULL; | ||
| 947 | int iAttributes = 0; | ||
| 948 | int iTarget = fetUnknown; | ||
| 949 | LPWSTR pwzFile = NULL; | ||
| 950 | LPWSTR pwzPort = NULL; | ||
| 951 | LPWSTR pwzDescription = NULL; | ||
| 952 | int iProtocol = 0; | ||
| 953 | int iProfile = 0; | ||
| 954 | |||
| 955 | // initialize | ||
| 956 | hr = WcaInitialize(hInstall, "ExecFirewallExceptions"); | ||
| 957 | ExitOnFailure(hr, "failed to initialize"); | ||
| 958 | |||
| 959 | hr = WcaGetProperty( L"CustomActionData", &pwzCustomActionData); | ||
| 960 | ExitOnFailure(hr, "failed to get CustomActionData"); | ||
| 961 | WcaLog(LOGMSG_TRACEONLY, "CustomActionData: %ls", pwzCustomActionData); | ||
| 962 | |||
| 963 | hr = ::CoInitialize(NULL); | ||
| 964 | ExitOnFailure(hr, "failed to initialize COM"); | ||
| 965 | |||
| 966 | // Find out if we support profiles (only on Vista or later) | ||
| 967 | fSupportProfiles = FSupportProfiles(); | ||
| 968 | |||
| 969 | // loop through all the passed in data | ||
| 970 | pwz = pwzCustomActionData; | ||
| 971 | while (pwz && *pwz) | ||
| 972 | { | ||
| 973 | // extract the custom action data and if rolling back, swap INSTALL and UNINSTALL | ||
| 974 | hr = WcaReadIntegerFromCaData(&pwz, &iTodo); | ||
| 975 | ExitOnFailure(hr, "failed to read todo from custom action data"); | ||
| 976 | if (::MsiGetMode(hInstall, MSIRUNMODE_ROLLBACK)) | ||
| 977 | { | ||
| 978 | if (WCA_TODO_INSTALL == iTodo) | ||
| 979 | { | ||
| 980 | iTodo = WCA_TODO_UNINSTALL; | ||
| 981 | } | ||
| 982 | else if (WCA_TODO_UNINSTALL == iTodo) | ||
| 983 | { | ||
| 984 | iTodo = WCA_TODO_INSTALL; | ||
| 985 | } | ||
| 986 | } | ||
| 987 | |||
| 988 | hr = WcaReadStringFromCaData(&pwz, &pwzName); | ||
| 989 | ExitOnFailure(hr, "failed to read name from custom action data"); | ||
| 990 | |||
| 991 | hr = WcaReadIntegerFromCaData(&pwz, &iProfile); | ||
| 992 | ExitOnFailure(hr, "failed to read profile from custom action data"); | ||
| 993 | |||
| 994 | hr = WcaReadStringFromCaData(&pwz, &pwzRemoteAddresses); | ||
| 995 | ExitOnFailure(hr, "failed to read remote addresses from custom action data"); | ||
| 996 | |||
| 997 | hr = WcaReadIntegerFromCaData(&pwz, &iAttributes); | ||
| 998 | ExitOnFailure(hr, "failed to read attributes from custom action data"); | ||
| 999 | BOOL fIgnoreFailures = feaIgnoreFailures == (iAttributes & feaIgnoreFailures); | ||
| 1000 | |||
| 1001 | hr = WcaReadIntegerFromCaData(&pwz, &iTarget); | ||
| 1002 | ExitOnFailure(hr, "failed to read target from custom action data"); | ||
| 1003 | |||
| 1004 | if (iTarget == fetApplication) | ||
| 1005 | { | ||
| 1006 | hr = WcaReadStringFromCaData(&pwz, &pwzFile); | ||
| 1007 | ExitOnFailure(hr, "failed to read file path from custom action data"); | ||
| 1008 | } | ||
| 1009 | |||
| 1010 | hr = WcaReadStringFromCaData(&pwz, &pwzPort); | ||
| 1011 | ExitOnFailure(hr, "failed to read port from custom action data"); | ||
| 1012 | hr = WcaReadIntegerFromCaData(&pwz, &iProtocol); | ||
| 1013 | ExitOnFailure(hr, "failed to read protocol from custom action data"); | ||
| 1014 | hr = WcaReadStringFromCaData(&pwz, &pwzDescription); | ||
| 1015 | ExitOnFailure(hr, "failed to read protocol from custom action data"); | ||
| 1016 | |||
| 1017 | switch (iTarget) | ||
| 1018 | { | ||
| 1019 | case fetPort: | ||
| 1020 | switch (iTodo) | ||
| 1021 | { | ||
| 1022 | case WCA_TODO_INSTALL: | ||
| 1023 | case WCA_TODO_REINSTALL: | ||
| 1024 | WcaLog(LOGMSG_STANDARD, "Installing firewall exception2 %ls on port %ls, protocol %d", pwzName, pwzPort, iProtocol); | ||
| 1025 | hr = AddPortException(fSupportProfiles, pwzName, iProfile, pwzRemoteAddresses, fIgnoreFailures, pwzPort, iProtocol, pwzDescription); | ||
| 1026 | ExitOnFailure(hr, "failed to add/update port exception for name '%ls' on port %ls, protocol %d", pwzName, pwzPort, iProtocol); | ||
| 1027 | break; | ||
| 1028 | |||
| 1029 | case WCA_TODO_UNINSTALL: | ||
| 1030 | WcaLog(LOGMSG_STANDARD, "Uninstalling firewall exception2 %ls on port %ls, protocol %d", pwzName, pwzPort, iProtocol); | ||
| 1031 | hr = RemovePortException(fSupportProfiles, pwzName, pwzPort, iProtocol, fIgnoreFailures); | ||
| 1032 | ExitOnFailure(hr, "failed to remove port exception for name '%ls' on port %ls, protocol %d", pwzName, pwzPort, iProtocol); | ||
| 1033 | break; | ||
| 1034 | } | ||
| 1035 | break; | ||
| 1036 | |||
| 1037 | case fetApplication: | ||
| 1038 | switch (iTodo) | ||
| 1039 | { | ||
| 1040 | case WCA_TODO_INSTALL: | ||
| 1041 | case WCA_TODO_REINSTALL: | ||
| 1042 | WcaLog(LOGMSG_STANDARD, "Installing firewall exception2 %ls (%ls)", pwzName, pwzFile); | ||
| 1043 | hr = AddApplicationException(fSupportProfiles, pwzFile, pwzName, iProfile, pwzRemoteAddresses, fIgnoreFailures, pwzPort, iProtocol, pwzDescription); | ||
| 1044 | ExitOnFailure(hr, "failed to add/update application exception for name '%ls', file '%ls'", pwzName, pwzFile); | ||
| 1045 | break; | ||
| 1046 | |||
| 1047 | case WCA_TODO_UNINSTALL: | ||
| 1048 | WcaLog(LOGMSG_STANDARD, "Uninstalling firewall exception2 %ls (%ls)", pwzName, pwzFile); | ||
| 1049 | hr = RemoveApplicationException(fSupportProfiles, pwzName, pwzFile, fIgnoreFailures, pwzPort, iProtocol); | ||
| 1050 | ExitOnFailure(hr, "failed to remove application exception for name '%ls', file '%ls'", pwzName, pwzFile); | ||
| 1051 | break; | ||
| 1052 | } | ||
| 1053 | break; | ||
| 1054 | } | ||
| 1055 | } | ||
| 1056 | |||
| 1057 | LExit: | ||
| 1058 | ReleaseStr(pwzCustomActionData); | ||
| 1059 | ReleaseStr(pwzName); | ||
| 1060 | ReleaseStr(pwzRemoteAddresses); | ||
| 1061 | ReleaseStr(pwzFile); | ||
| 1062 | ReleaseStr(pwzPort); | ||
| 1063 | ReleaseStr(pwzDescription); | ||
| 1064 | ::CoUninitialize(); | ||
| 1065 | |||
| 1066 | return WcaFinalize(FAILED(hr) ? ERROR_INSTALL_FAILURE : ERROR_SUCCESS); | ||
| 1067 | } | ||
diff --git a/src/ca/fwca.def b/src/ca/fwca.def new file mode 100644 index 00000000..d32c5379 --- /dev/null +++ b/src/ca/fwca.def | |||
| @@ -0,0 +1,9 @@ | |||
| 1 | ; 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. | ||
| 2 | |||
| 3 | |||
| 4 | LIBRARY "fwca" | ||
| 5 | |||
| 6 | EXPORTS | ||
| 7 | SchedFirewallExceptionsInstall | ||
| 8 | SchedFirewallExceptionsUninstall | ||
| 9 | ExecFirewallExceptions | ||
diff --git a/src/ca/fwca.vcxproj b/src/ca/fwca.vcxproj new file mode 100644 index 00000000..b6075790 --- /dev/null +++ b/src/ca/fwca.vcxproj | |||
| @@ -0,0 +1,72 @@ | |||
| 1 | <?xml version="1.0" encoding="utf-8"?> | ||
| 2 | <!-- 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. --> | ||
| 3 | |||
| 4 | <Project DefaultTargets="Build" ToolsVersion="15.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003"> | ||
| 5 | <Import Project="..\..\packages\WixToolset.DUtil.4.0.6\build\WixToolset.DUtil.props" Condition="Exists('..\..\packages\WixToolset.DUtil.4.0.6\build\WixToolset.DUtil.props')" /> | ||
| 6 | <Import Project="..\..\packages\WixToolset.WcaUtil.4.0.2\build\WixToolset.WcaUtil.props" Condition="Exists('..\..\packages\WixToolset.WcaUtil.4.0.2\build\WixToolset.WcaUtil.props')" /> | ||
| 7 | |||
| 8 | <ItemGroup Label="ProjectConfigurations"> | ||
| 9 | <ProjectConfiguration Include="Debug|Win32"> | ||
| 10 | <Configuration>Debug</Configuration> | ||
| 11 | <Platform>Win32</Platform> | ||
| 12 | </ProjectConfiguration> | ||
| 13 | <ProjectConfiguration Include="Release|Win32"> | ||
| 14 | <Configuration>Release</Configuration> | ||
| 15 | <Platform>Win32</Platform> | ||
| 16 | </ProjectConfiguration> | ||
| 17 | </ItemGroup> | ||
| 18 | |||
| 19 | <PropertyGroup Label="Globals"> | ||
| 20 | <ProjectGuid>{F72D34CA-48DA-4DFD-91A9-A0C78BEF6981}</ProjectGuid> | ||
| 21 | <ConfigurationType>DynamicLibrary</ConfigurationType> | ||
| 22 | <TargetName>fwca</TargetName> | ||
| 23 | <PlatformToolset>v141</PlatformToolset> | ||
| 24 | <CharacterSet>Unicode</CharacterSet> | ||
| 25 | <ProjectModuleDefinitionFile>fwca.def</ProjectModuleDefinitionFile> | ||
| 26 | <Description>WiX Toolset Firewall CustomAction</Description> | ||
| 27 | </PropertyGroup> | ||
| 28 | |||
| 29 | <Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" /> | ||
| 30 | <Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" /> | ||
| 31 | |||
| 32 | <ImportGroup Label="ExtensionSettings"> | ||
| 33 | </ImportGroup> | ||
| 34 | |||
| 35 | <ImportGroup Label="Shared"> | ||
| 36 | <Import Project="..\..\packages\Microsoft.VisualStudio.Setup.Configuration.Native.1.14.114\build\native\Microsoft.VisualStudio.Setup.Configuration.Native.targets" Condition="Exists('..\..\packages\Microsoft.VisualStudio.Setup.Configuration.Native.1.14.114\build\native\Microsoft.VisualStudio.Setup.Configuration.Native.targets')" /> | ||
| 37 | </ImportGroup> | ||
| 38 | |||
| 39 | <PropertyGroup> | ||
| 40 | <ProjectAdditionalLinkLibraries>msi.lib</ProjectAdditionalLinkLibraries> | ||
| 41 | </PropertyGroup> | ||
| 42 | |||
| 43 | <ItemGroup> | ||
| 44 | <ClCompile Include="dllmain.cpp"> | ||
| 45 | <PrecompiledHeader>Create</PrecompiledHeader> | ||
| 46 | </ClCompile> | ||
| 47 | <ClCompile Include="firewall.cpp" /> | ||
| 48 | </ItemGroup> | ||
| 49 | |||
| 50 | <ItemGroup> | ||
| 51 | <ClInclude Include="cost.h" /> | ||
| 52 | <ClInclude Include="CustomMsiErrors.h" /> | ||
| 53 | <ClInclude Include="precomp.h" /> | ||
| 54 | </ItemGroup> | ||
| 55 | |||
| 56 | <ItemGroup> | ||
| 57 | <None Include="packages.config" /> | ||
| 58 | <None Include="fwca.def" /> | ||
| 59 | </ItemGroup> | ||
| 60 | |||
| 61 | <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" /> | ||
| 62 | |||
| 63 | <Target Name="EnsureNuGetPackageBuildImports" BeforeTargets="PrepareForBuild"> | ||
| 64 | <PropertyGroup> | ||
| 65 | <ErrorText>This project references NuGet package(s) that are missing on this computer. Use NuGet Package Restore to download them. For more information, see http://go.microsoft.com/fwlink/?LinkID=322105. The missing file is {0}.</ErrorText> | ||
| 66 | </PropertyGroup> | ||
| 67 | <Error Condition="!Exists('..\..\packages\Microsoft.VisualStudio.Setup.Configuration.Native.1.14.114\build\native\Microsoft.VisualStudio.Setup.Configuration.Native.targets')" Text="$([System.String]::Format('$(ErrorText)', '..\..\packages\Microsoft.VisualStudio.Setup.Configuration.Native.1.14.114\build\native\Microsoft.VisualStudio.Setup.Configuration.Native.targets'))" /> | ||
| 68 | <Error Condition="!Exists('..\..\packages\WixToolset.DUtil.4.0.6\build\WixToolset.DUtil.props')" Text="$([System.String]::Format('$(ErrorText)', '..\..\packages\WixToolset.DUtil.4.0.6\build\WixToolset.DUtil.props'))" /> | ||
| 69 | <Error Condition="!Exists('..\..\packages\WixToolset.WcaUtil.4.0.2\build\WixToolset.WcaUtil.props')" Text="$([System.String]::Format('$(ErrorText)', '..\..\packages\WixToolset.WcaUtil.4.0.2\build\WixToolset.WcaUtil.props'))" /> | ||
| 70 | </Target> | ||
| 71 | </Project> | ||
| 72 | |||
diff --git a/src/ca/packages.config b/src/ca/packages.config new file mode 100644 index 00000000..b74ff5d0 --- /dev/null +++ b/src/ca/packages.config | |||
| @@ -0,0 +1,6 @@ | |||
| 1 | <?xml version="1.0" encoding="utf-8"?> | ||
| 2 | <packages> | ||
| 3 | <package id="Microsoft.VisualStudio.Setup.Configuration.Native" version="1.14.114" targetFramework="native" developmentDependency="true" /> | ||
| 4 | <package id="WixToolset.DUtil" version="4.0.6" targetFramework="native" /> | ||
| 5 | <package id="WixToolset.WcaUtil" version="4.0.2" targetFramework="native" /> | ||
| 6 | </packages> \ No newline at end of file | ||
diff --git a/src/ca/precomp.h b/src/ca/precomp.h new file mode 100644 index 00000000..a84bacff --- /dev/null +++ b/src/ca/precomp.h | |||
| @@ -0,0 +1,17 @@ | |||
| 1 | #pragma once | ||
| 2 | // 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. | ||
| 3 | |||
| 4 | |||
| 5 | #include <windows.h> | ||
| 6 | #include <msidefs.h> | ||
| 7 | #include <msiquery.h> | ||
| 8 | #include <strsafe.h> | ||
| 9 | #include <netfw.h> | ||
| 10 | |||
| 11 | #include "wcautil.h" | ||
| 12 | #include "fileutil.h" | ||
| 13 | #include "pathutil.h" | ||
| 14 | #include "strutil.h" | ||
| 15 | |||
| 16 | #include "CustomMsiErrors.h" | ||
| 17 | #include "cost.h" | ||
