/******************************************************************
WixSchedInternetShortcuts - entry point
********************************************************************/
extern "C" UINT __stdcall WixSchedInternetShortcuts(
__in MSIHANDLE hInstall
)
{
HRESULT hr = S_OK;
UINT er = ERROR_SUCCESS;
UINT uiCost = 0;
PMSIHANDLE hView = NULL;
PMSIHANDLE hRec = NULL;
MSIHANDLE hCreateFolderTable = NULL;
MSIHANDLE hCreateFolderColumns = NULL;
LPWSTR pwzCustomActionData = NULL;
LPWSTR pwzComponent = NULL;
LPWSTR pwzDirectory = NULL;
LPWSTR pwzFilename = NULL;
LPWSTR pwzTarget = NULL;
LPWSTR pwzShortcutPath = NULL;
int iAttr = 0;
LPWSTR pwzIconFile = NULL;
int iIconIndex = 0;
IUniformResourceLocatorW* piURL = NULL;
IShellLinkW* piShellLink = NULL;
BOOL fInitializedCom = FALSE;
hr = WcaInitialize(hInstall, "WixSchedInternetShortcuts");
ExitOnFailure(hr, "failed to initialize WixSchedInternetShortcuts.");
// anything to do?
if (S_OK != WcaTableExists(L"WixInternetShortcut"))
{
WcaLog(LOGMSG_STANDARD, "WixInternetShortcut table doesn't exist, so there are no Internet shortcuts to process");
goto LExit;
}
// check to see if we can create a shortcut - Server Core and others may not have a shell registered.
hr = ::CoInitialize(NULL);
ExitOnFailure(hr, "failed to initialize COM");
fInitializedCom = TRUE;
hr = ::CoCreateInstance(CLSID_InternetShortcut, NULL, CLSCTX_ALL, IID_IUniformResourceLocatorW, (void**)&piURL);
if (S_OK != hr)
{
WcaLog(LOGMSG_STANDARD, "failed to create an instance of IUniformResourceLocatorW, skipping shortcut creation");
ExitFunction1(hr = S_OK);
}
hr = ::CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_ALL, IID_IShellLinkW, (void**)&piShellLink);
if (S_OK != hr)
{
WcaLog(LOGMSG_STANDARD, "failed to create an instance of IShellLinkW, skipping shortcut creation");
ExitFunction1(hr = S_OK);
}
// query and loop through all the shortcuts
hr = WcaOpenExecuteView(vcsShortcutsQuery, &hView);
ExitOnFailure(hr, "failed to open view on WixInternetShortcut table");
while (S_OK == (hr = WcaFetchRecord(hView, &hRec)))
{
// read column values
hr = WcaGetRecordString(hRec, esqComponent, &pwzComponent);
ExitOnFailure(hr, "failed to get shortcut component");
hr = WcaGetRecordString(hRec, esqDirectory, &pwzDirectory);
ExitOnFailure(hr, "failed to get shortcut directory");
hr = WcaGetRecordString(hRec, esqFilename, &pwzFilename);
ExitOnFailure(hr, "failed to get shortcut filename");
hr = WcaGetRecordFormattedString(hRec, esqTarget, &pwzTarget);
ExitOnFailure(hr, "failed to get shortcut target");
hr = WcaGetRecordInteger(hRec, esqAttributes, &iAttr);
ExitOnFailure(hr, "failed to get shortcut attributes");
hr = WcaGetRecordFormattedString(hRec, esqIconFile, &pwzIconFile);
ExitOnFailure(hr, "failed to get shortcut icon file");
hr = WcaGetRecordInteger(hRec, esqIconIndex, &iIconIndex);
ExitOnFailure(hr, "failed to get shortcut icon index");
// skip processing this WixInternetShortcut row if the component isn't being configured
WCA_TODO todo = WcaGetComponentToDo(pwzComponent);
if (WCA_TODO_UNKNOWN == todo)
{
WcaLog(LOGMSG_VERBOSE, "Skipping shortcut for null-action component '%ls'", pwzComponent);
continue;
}
// we need to create the directory where the shortcut is supposed to live; rather
// than doing so in our deferred custom action, use the CreateFolder table to have MSI
// make (and remove) them on our behalf (including the correct cleanup of parent directories).
MSIDBERROR dbError = MSIDBERROR_NOERROR;
WcaLog(LOGMSG_STANDARD, "Adding folder '%ls', component '%ls' to the CreateFolder table", pwzDirectory, pwzComponent);
hr = WcaAddTempRecord(&hCreateFolderTable, &hCreateFolderColumns, L"CreateFolder", &dbError, 0, 2, pwzDirectory, pwzComponent);
if (MSIDBERROR_DUPLICATEKEY == dbError)
{
WcaLog(LOGMSG_STANDARD, "Folder '%ls' already exists in the CreateFolder table; the above error is harmless", pwzDirectory);
hr = S_OK;
}
ExitOnFailure(hr, "Couldn't add temporary CreateFolder row");
// only if we're installing/reinstalling do we need to schedule the deferred CA
// (uninstallation is handled via permanent RemoveFile rows and temporary CreateFolder rows)
if (WCA_TODO_INSTALL == todo || WCA_TODO_REINSTALL == todo)
{
// turn the Directory_ id into a path
hr = WcaGetTargetPath(pwzDirectory, &pwzShortcutPath);
ExitOnFailure(hr, "failed to allocate string for shortcut directory");
// append the shortcut filename
hr = StrAllocConcat(&pwzShortcutPath, pwzFilename, 0);
ExitOnFailure(hr, "failed to allocate string for shortcut filename");
// write the shortcut path and target to custom action data for deferred CAs
hr = WcaWriteStringToCaData(pwzShortcutPath, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write shortcut path to custom action data");
hr = WcaWriteStringToCaData(pwzTarget, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write shortcut target to custom action data");
hr = WcaWriteIntegerToCaData(iAttr, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write shortcut attributes to custom action data");
hr = WcaWriteStringToCaData(pwzIconFile, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write icon file to custom action data");
hr = WcaWriteIntegerToCaData(iIconIndex, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write icon index to custom action data");
uiCost += COST_INTERNETSHORTCUT;
}
}
if (E_NOMOREITEMS == hr)
{
hr = S_OK;
}
ExitOnFailure(hr, "Failure occured while processing WixInternetShortcut table");
// if we have any shortcuts to install
if (pwzCustomActionData && *pwzCustomActionData)
{
// add cost to progress bar
hr = WcaProgressMessage(uiCost, TRUE);
ExitOnFailure(hr, "failed to extend progress bar for InternetShortcuts");
// provide custom action data to deferred and rollback CAs
hr = WcaSetProperty(PLATFORM_DECORATION(L"WixRollbackInternetShortcuts"), pwzCustomActionData);
ExitOnFailure(hr, "failed to set WixRollbackInternetShortcuts rollback custom action data");
hr = WcaSetProperty(PLATFORM_DECORATION(L"WixCreateInternetShortcuts"), pwzCustomActionData);
ExitOnFailure(hr, "failed to set WixCreateInternetShortcuts custom action data");
}
LExit:
if (hCreateFolderTable)
{
::MsiCloseHandle(hCreateFolderTable);
}
if (hCreateFolderColumns)
{
::MsiCloseHandle(hCreateFolderColumns);
}
ReleaseStr(pwzCustomActionData);
ReleaseStr(pwzComponent);
ReleaseStr(pwzDirectory);
ReleaseStr(pwzFilename);
ReleaseStr(pwzTarget);
ReleaseStr(pwzShortcutPath);
ReleaseObject(piShellLink);
ReleaseObject(piURL);
if (fInitializedCom)
{
::CoUninitialize();
}
er = SUCCEEDED(hr) ? ERROR_SUCCESS : ERROR_INSTALL_FAILURE;
return WcaFinalize(er);
}
/********************************************************************
WixRegisterRestartResources - Immediate CA to register resources with RM.
Enumerates components before InstallValidate and registers resources
to be restarted by Restart Manager if the component action
is anything other than None.
Do not disable file system redirection.
********************************************************************/
extern "C" UINT __stdcall WixRegisterRestartResources(
__in MSIHANDLE hInstall
)
{
HRESULT hr = S_OK;
UINT er = ERROR_SUCCESS;
PMSIHANDLE hView = NULL;
PMSIHANDLE hRec = NULL;
LPWSTR wzSessionKey = NULL;
size_t cchSessionKey = 0;
PRMU_SESSION pSession = NULL;
LPWSTR wzRestartResource = NULL;
LPWSTR wzComponent = NULL;
LPWSTR wzResource = NULL;
int iAttributes = NULL;
BOOL fIsComponentNull = FALSE;
WCA_TODO todo = WCA_TODO_UNKNOWN;
int iType = etInvalid;
hr = WcaInitialize(hInstall, "WixRegisterRestartResources");
ExitOnFailure(hr, "Failed to initialize.");
// Skip if the table doesn't exist.
if (S_OK != WcaTableExists(L"WixRestartResource"))
{
WcaLog(LOGMSG_STANDARD, "The RestartResource table does not exist; there are no resources to register with Restart Manager.");
ExitFunction();
}
// Get the existing Restart Manager session if available.
hr = WcaGetProperty(L"MsiRestartManagerSessionKey", &wzSessionKey);
ExitOnFailure(hr, "Failed to get the MsiRestartManagerSessionKey property.");
hr = ::StringCchLengthW(wzSessionKey, CCH_SESSION_KEY, &cchSessionKey);
ExitOnFailure(hr, "Failed to get the MsiRestartManagerSessionKey string length.");
// Skip if the property doesn't exist.
if (0 == cchSessionKey)
{
WcaLog(LOGMSG_STANDARD, "The MsiRestartManagerSessionKey property is not available to join.");
ExitFunction();
}
// Join the existing Restart Manager session if supported.
hr = RmuJoinSession(&pSession, wzSessionKey);
if (E_MODNOTFOUND == hr)
{
WcaLog(LOGMSG_STANDARD, "The Restart Manager is not supported on this platform. Skipping.");
ExitFunction1(hr = S_OK);
}
else if (FAILED(hr))
{
WcaLog(LOGMSG_STANDARD, "Failed to join the existing Restart Manager session %ls.", wzSessionKey);
ExitFunction1(hr = S_OK);
}
// Loop through each record in the table.
hr = WcaOpenExecuteView(vcsRestartResourceQuery, &hView);
ExitOnFailure(hr, "Failed to open a view on the RestartResource table.");
while (S_OK == (hr = WcaFetchRecord(hView, &hRec)))
{
hr = WcaGetRecordString(hRec, rrqRestartResource, &wzRestartResource);
ExitOnFailure(hr, "Failed to get the RestartResource field value.");
hr = WcaGetRecordString(hRec, rrqComponent, &wzComponent);
ExitOnFailure(hr, "Failed to get the Component_ field value.");
hr = WcaGetRecordFormattedString(hRec, rrqResource, &wzResource);
ExitOnFailure(hr, "Failed to get the Resource formatted field value.");
hr = WcaGetRecordInteger(hRec, rrqAttributes, &iAttributes);
ExitOnFailure(hr, "Failed to get the Attributes field value.");
fIsComponentNull = ::MsiRecordIsNull(hRec, rrqComponent);
todo = WcaGetComponentToDo(wzComponent);
// Only register resources for components that are null, or being installed, reinstalled, or uninstalled.
if (!fIsComponentNull && WCA_TODO_UNKNOWN == todo)
{
WcaLog(LOGMSG_VERBOSE, "Skipping resource %ls.", wzRestartResource);
continue;
}
// Get the type from Attributes and add to the Restart Manager.
iType = iAttributes & etTypeMask;
switch (iType)
{
case etFilename:
WcaLog(LOGMSG_VERBOSE, "Registering file name %ls with the Restart Manager.", wzResource);
hr = RmuAddFile(pSession, wzResource);
ExitOnFailure(hr, "Failed to register the file name with the Restart Manager session.");
break;
case etApplication:
WcaLog(LOGMSG_VERBOSE, "Registering process name %ls with the Restart Manager.", wzResource);
hr = RmuAddProcessesByName(pSession, wzResource);
ExitOnFailure(hr, "Failed to register the process name with the Restart Manager session.");
break;
case etServiceName:
WcaLog(LOGMSG_VERBOSE, "Registering service name %ls with the Restart Manager.", wzResource);
hr = RmuAddService(pSession, wzResource);
ExitOnFailure(hr, "Failed to register the service name with the Restart Manager session.");
break;
default:
WcaLog(LOGMSG_VERBOSE, "The resource type %d for %ls is not supported and will not be registered.", iType, wzRestartResource);
break;
}
}
if (E_NOMOREITEMS == hr)
{
hr = S_OK;
}
ExitOnFailure(hr, "Failed while looping through all rows to register resources.");
// Register the resources and unjoin the session.
hr = RmuEndSession(pSession);
if (FAILED(hr))
{
WcaLog(LOGMSG_VERBOSE, "Failed to register the resources with the Restart Manager.");
ExitFunction1(hr = S_OK);
}
LExit:
ReleaseStr(wzRestartResource);
ReleaseStr(wzComponent);
ReleaseStr(wzResource);
if (FAILED(hr))
{
er = ERROR_INSTALL_FAILURE;
}
return WcaFinalize(er);
}
/******************************************************************
SchedAddinRegistration - entry point for AddinRegistration Custom Action
********************************************************************/
HRESULT SchedAddinRegistration(MSIHANDLE hInstall, BOOL fInstall)
{
// AssertSz(FALSE, "debug SchedRegisterAddins");
HRESULT hr = S_OK;
LPWSTR pwzCustomActionData = NULL;
PMSIHANDLE hView = NULL;
PMSIHANDLE hRec = NULL;
LPWSTR pwzData = NULL;
LPWSTR pwzTemp = NULL;
LPWSTR pwzComponent = NULL;
LPWSTR pwzId = NULL;
LPWSTR pwzFile = NULL;
LPWSTR pwzFriendlyName = NULL;
LPWSTR pwzDescription = NULL;
int iBitness = 0;
int iCommandLineSafe = 1;
int iLoadBehavior = 0;
LPWSTR pwzAllUsers = NULL;
int nAddins = 0;
hr = WcaGetProperty(L"ALLUSERS", &pwzAllUsers);
ExitOnFailure(hr, "failed to read value of ALLUSERS property");
// loop through all the RegisterAddin records
hr = WcaOpenExecuteView(vcsRegisterAddinsQuery, &hView);
ExitOnFailure(hr, "failed to open view on AddinRegistration table");
while (S_OK == (hr = WcaFetchRecord(hView, &hRec)))
{
++nAddins;
// Get Id
hr = WcaGetRecordString(hRec, arqId, &pwzId);
ExitOnFailure(hr, "failed to get AddinRegistration.AddinRegistration");
// Get File
hr = WcaGetRecordString(hRec, arqFile, &pwzData);
ExitOnFailure1(hr, "failed to get AddinRegistration.File_ for record: %ls", pwzId);
hr = StrAllocFormatted(&pwzTemp, L"[#%ls]", pwzData);
ExitOnFailure1(hr, "failed to format file string for file: %ls", pwzData);
hr = WcaGetFormattedString(pwzTemp, &pwzFile);
ExitOnFailure1(hr, "failed to get formatted string for file: %ls", pwzData);
// Get name
hr = WcaGetRecordFormattedString(hRec, arqName, &pwzFriendlyName);
ExitOnFailure1(hr, "failed to get AddinRegistration.Name for record: %ls", pwzId);
// Get description
hr = WcaGetRecordFormattedString(hRec, arqDescription, &pwzDescription);
ExitOnFailure1(hr, "failed to get AddinRegistration.Description for record: %ls", pwzId);
// Get description
hr = WcaGetRecordInteger(hRec, arqBitness, &iBitness);
ExitOnFailure1(hr, "failed to get AddinRegistration.Bitnesss for record: %ls", pwzId);
// Get description
hr = WcaGetRecordInteger(hRec, arqCommandLineSafe, &iCommandLineSafe);
ExitOnFailure1(hr, "failed to get AddinRegistration.CommandLineSafe for record: %ls", pwzId);
// Get description
hr = WcaGetRecordInteger(hRec, arqLoadBehavior, &iLoadBehavior);
ExitOnFailure1(hr, "failed to get AddinRegistration.LoadBehavior for record: %ls", pwzId);
// get component and its install/action states
hr = WcaGetRecordString(hRec, arqComponent, &pwzComponent);
ExitOnFailure(hr, "failed to get addin component id");
// we need to know if the component's being installed, uninstalled, or reinstalled
WCA_TODO todo = WcaGetComponentToDo(pwzComponent);
// skip this entry if this is the install CA and we are uninstalling the component
if (fInstall && WCA_TODO_UNINSTALL == todo)
{
continue;
}
// skip this entry if this is an uninstall CA and we are not uninstalling the component
if (!fInstall && WCA_TODO_UNINSTALL != todo)
{
continue;
}
// write custom action data: operation, instance guid, path, directory
hr = WcaWriteIntegerToCaData(todo, &pwzCustomActionData);
ExitOnFailure1(hr, "failed to write operation to custom action data for instance id: %ls", pwzId);
hr = WcaWriteStringToCaData(pwzId, &pwzCustomActionData);
ExitOnFailure1(hr, "failed to write id to custom action data for instance id: %ls", pwzId);
hr = WcaWriteStringToCaData(pwzFile, &pwzCustomActionData);
ExitOnFailure1(hr, "failed to write custom action data for instance id: %ls", pwzId);
hr = WcaWriteStringToCaData(pwzFriendlyName, &pwzCustomActionData);
ExitOnFailure1(hr, "failed to write addin name to custom action data for instance id: %ls", pwzId);
hr = WcaWriteStringToCaData(pwzDescription, &pwzCustomActionData);
ExitOnFailure1(hr, "failed to write addin description to custom action data for instance id: %ls", pwzId);
hr = WcaWriteIntegerToCaData(iBitness, &pwzCustomActionData);
ExitOnFailure1(hr, "failed to write Bitness to custom action data for instance id: %ls", pwzId);
hr = WcaWriteIntegerToCaData(iCommandLineSafe, &pwzCustomActionData);
ExitOnFailure1(hr, "failed to write CommandLineSafe to custom action data for instance id: %ls", pwzId);
hr = WcaWriteIntegerToCaData(iLoadBehavior, &pwzCustomActionData);
ExitOnFailure1(hr, "failed to write LoadBehavior to custom action data for instance id: %ls", pwzId);
hr = WcaWriteStringToCaData(pwzAllUsers, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write allusers property to custom action data for instance id: %ls", pwzId);
}
if (E_NOMOREITEMS == hr)
hr = S_OK;
ExitOnFailure(hr, "failed while looping through all files to create native images for");
// Schedule the install custom action
if (pwzCustomActionData && *pwzCustomActionData)
{
WcaLog(LOGMSG_STANDARD, "Scheduling Addin Registration (%ls)", pwzCustomActionData);
hr = WcaDoDeferredAction(L"RollbackAddinRegistration", pwzCustomActionData, nAddins * COST_REGISTER_ADDIN);
ExitOnFailure(hr, "Failed to schedule addin registration rollback");
hr = WcaDoDeferredAction(L"ExecAddinRegistration", pwzCustomActionData, nAddins * COST_REGISTER_ADDIN);
ExitOnFailure(hr, "Failed to schedule addin registration execution");
}
LExit:
ReleaseStr(pwzAllUsers);
ReleaseStr(pwzCustomActionData);
ReleaseStr(pwzId);
ReleaseStr(pwzData);
ReleaseStr(pwzData);
ReleaseStr(pwzTemp);
ReleaseStr(pwzComponent);
ReleaseStr(pwzFile);
ReleaseStr(pwzFriendlyName);
ReleaseStr(pwzDescription);
return hr;
}
/******************************************************************
SchedFirewallExceptions - immediate custom action worker to
register and remove firewall exceptions.
********************************************************************/
static UINT SchedFirewallExceptions(
__in MSIHANDLE hInstall,
WCA_TODO todoSched
)
{
HRESULT hr = S_OK;
UINT er = ERROR_SUCCESS;
int cFirewallExceptions = 0;
PMSIHANDLE hView = NULL;
PMSIHANDLE hRec = NULL;
LPWSTR pwzCustomActionData = NULL;
LPWSTR pwzName = NULL;
LPWSTR pwzRemoteAddresses = NULL;
LPWSTR pwzPort = NULL;
int iProtocol = 0;
int iAttributes = 0;
int iProfile = 0;
LPWSTR pwzProgram = NULL;
LPWSTR pwzComponent = NULL;
LPWSTR pwzFormattedFile = NULL;
LPWSTR pwzDescription = NULL;
// initialize
hr = WcaInitialize(hInstall, "SchedFirewallExceptions");
ExitOnFailure(hr, "failed to initialize");
// anything to do?
if (S_OK != WcaTableExists(L"WixFirewallException"))
{
WcaLog(LOGMSG_STANDARD, "WixFirewallException table doesn't exist, so there are no firewall exceptions to configure.");
ExitFunction();
}
// query and loop through all the firewall exceptions
hr = WcaOpenExecuteView(vcsFirewallExceptionQuery, &hView);
ExitOnFailure(hr, "failed to open view on WixFirewallException table");
while (S_OK == (hr = WcaFetchRecord(hView, &hRec)))
{
hr = WcaGetRecordFormattedString(hRec, feqName, &pwzName);
ExitOnFailure(hr, "failed to get firewall exception name");
hr = WcaGetRecordFormattedString(hRec, feqRemoteAddresses, &pwzRemoteAddresses);
ExitOnFailure(hr, "failed to get firewall exception remote addresses");
hr = WcaGetRecordFormattedString(hRec, feqPort, &pwzPort);
ExitOnFailure(hr, "failed to get firewall exception port");
hr = WcaGetRecordInteger(hRec, feqProtocol, &iProtocol);
ExitOnFailure(hr, "failed to get firewall exception protocol");
hr = WcaGetRecordFormattedString(hRec, feqProgram, &pwzProgram);
ExitOnFailure(hr, "failed to get firewall exception program");
hr = WcaGetRecordInteger(hRec, feqAttributes, &iAttributes);
ExitOnFailure(hr, "failed to get firewall exception attributes");
hr = WcaGetRecordInteger(hRec, feqProfile, &iProfile);
ExitOnFailure(hr, "failed to get firewall exception profile");
hr = WcaGetRecordString(hRec, feqComponent, &pwzComponent);
ExitOnFailure(hr, "failed to get firewall exception component");
hr = WcaGetRecordString(hRec, feqDescription, &pwzDescription);
ExitOnFailure(hr, "failed to get firewall description");
// figure out what we're doing for this exception, treating reinstall the same as install
WCA_TODO todoComponent = WcaGetComponentToDo(pwzComponent);
if ((WCA_TODO_REINSTALL == todoComponent ? WCA_TODO_INSTALL : todoComponent) != todoSched)
{
WcaLog(LOGMSG_STANDARD, "Component '%ls' action state (%d) doesn't match request (%d)", pwzComponent, todoComponent, todoSched);
continue;
}
// action :: name :: profile :: remoteaddresses :: attributes :: target :: {port::protocol | path}
++cFirewallExceptions;
hr = WcaWriteIntegerToCaData(todoComponent, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write exception action to custom action data");
hr = WcaWriteStringToCaData(pwzName, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write exception name to custom action data");
hr = WcaWriteIntegerToCaData(iProfile, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write exception profile to custom action data");
hr = WcaWriteStringToCaData(pwzRemoteAddresses, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write exception remote addresses to custom action data");
hr = WcaWriteIntegerToCaData(iAttributes, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write exception attributes to custom action data");
if (*pwzProgram)
{
// If program is defined, we have an application exception.
hr = WcaWriteIntegerToCaData(fetApplication, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write exception target (application) to custom action data");
hr = WcaWriteStringToCaData(pwzProgram, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write application path to custom action data");
}
else
{
// we have a port-only exception
hr = WcaWriteIntegerToCaData(fetPort, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write exception target (port) to custom action data");
}
hr = WcaWriteStringToCaData(pwzPort, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write application path to custom action data");
hr = WcaWriteIntegerToCaData(iProtocol, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write exception protocol to custom action data");
hr = WcaWriteStringToCaData(pwzDescription, &pwzCustomActionData);
ExitOnFailure(hr, "failed to write firewall rule description to custom action data");
}
// reaching the end of the list is actually a good thing, not an error
if (E_NOMOREITEMS == hr)
{
hr = S_OK;
}
ExitOnFailure(hr, "failure occured while processing WixFirewallException table");
// schedule ExecFirewallExceptions if there's anything to do
if (pwzCustomActionData && *pwzCustomActionData)
{
WcaLog(LOGMSG_STANDARD, "Scheduling firewall exception (%ls)", pwzCustomActionData);
if (WCA_TODO_INSTALL == todoSched)
{
hr = WcaDoDeferredAction(L"WixRollbackFirewallExceptionsInstall", pwzCustomActionData, cFirewallExceptions * COST_FIREWALL_EXCEPTION);
ExitOnFailure(hr, "failed to schedule firewall install exceptions rollback");
hr = WcaDoDeferredAction(L"WixExecFirewallExceptionsInstall", pwzCustomActionData, cFirewallExceptions * COST_FIREWALL_EXCEPTION);
ExitOnFailure(hr, "failed to schedule firewall install exceptions execution");
}
else
{
hr = WcaDoDeferredAction(L"WixRollbackFirewallExceptionsUninstall", pwzCustomActionData, cFirewallExceptions * COST_FIREWALL_EXCEPTION);
ExitOnFailure(hr, "failed to schedule firewall uninstall exceptions rollback");
hr = WcaDoDeferredAction(L"WixExecFirewallExceptionsUninstall", pwzCustomActionData, cFirewallExceptions * COST_FIREWALL_EXCEPTION);
ExitOnFailure(hr, "failed to schedule firewall uninstall exceptions execution");
}
}
else
{
WcaLog(LOGMSG_STANDARD, "No firewall exceptions scheduled");
}
LExit:
ReleaseStr(pwzCustomActionData);
ReleaseStr(pwzName);
ReleaseStr(pwzRemoteAddresses);
ReleaseStr(pwzPort);
ReleaseStr(pwzProgram);
ReleaseStr(pwzComponent);
ReleaseStr(pwzDescription);
ReleaseStr(pwzFormattedFile);
return WcaFinalize(er = FAILED(hr) ? ERROR_INSTALL_FAILURE : er);
}
