/********************************************************************
MessageQueuingExecuteInstall - CUSTOM ACTION ENTRY POINT
Input: deferred CustomActionData - MessageQueuingExecuteInstall
********************************************************************/
extern "C" UINT __stdcall MessageQueuingExecuteInstall(MSIHANDLE hInstall)
{
HRESULT hr = S_OK;
UINT er = ERROR_SUCCESS;
LPWSTR pwzCustomActionData = NULL;
LPWSTR pwzData = NULL;
// initialize
hr = WcaInitialize(hInstall, "MessageQueuingExecuteInstall");
ExitOnFailure(hr, "Failed to initialize MessageQueuingExecuteInstall");
hr = MqiInitialize();
ExitOnFailure(hr, "Failed to initialize");
// get custom action data
hr = WcaGetProperty(L"CustomActionData", &pwzCustomActionData);
ExitOnFailure(hr, "Failed to get CustomActionData");
pwzData = pwzCustomActionData;
// create message queues
hr = MqiCreateMessageQueues(&pwzData);
ExitOnFailure(hr, "Failed to create message queues");
if (S_FALSE == hr) ExitFunction();
// add message queue permissions
hr = MqiAddMessageQueuePermissions(&pwzData);
ExitOnFailure(hr, "Failed to add message queue permissions");
if (S_FALSE == hr) ExitFunction();
hr = S_OK;
LExit:
// clean up
ReleaseStr(pwzCustomActionData);
// uninitialize
MqiUninitialize();
er = SUCCEEDED(hr) ? ERROR_SUCCESS : ERROR_INSTALL_FAILURE;
return WcaFinalize(er);
}
/********************************************************************
MessageQueuingRollbackUninstall - CUSTOM ACTION ENTRY POINT
Input: deferred CustomActionData - MessageQueuingRollbackUninstall
********************************************************************/
extern "C" UINT __stdcall MessageQueuingRollbackUninstall(MSIHANDLE hInstall)
{
HRESULT hr = S_OK;
UINT er = ERROR_SUCCESS;
LPWSTR pwzCustomActionData = NULL;
LPWSTR pwzData = NULL;
// initialize
hr = WcaInitialize(hInstall, "MessageQueuingRollbackUninstall");
ExitOnFailure(hr, "Failed to initialize MessageQueuingRollbackUninstall");
hr = MqiInitialize();
ExitOnFailure(hr, "Failed to initialize");
// get custom action data
hr = WcaGetProperty(L"CustomActionData", &pwzCustomActionData);
ExitOnFailure(hr, "Failed to get CustomActionData");
pwzData = pwzCustomActionData;
// delete message queues
hr = MqiRollbackDeleteMessageQueues(&pwzData);
ExitOnFailure(hr, "Failed to delete message queues");
// remove message queue permissions
hr = MqiRollbackRemoveMessageQueuePermissions(&pwzData);
ExitOnFailure(hr, "Failed to remove message queue permissions");
hr = S_OK;
LExit:
// clean up
ReleaseStr(pwzCustomActionData);
// uninitialize
MqiUninitialize();
er = SUCCEEDED(hr) ? ERROR_SUCCESS : ERROR_INSTALL_FAILURE;
return WcaFinalize(er);
}
/********************************************************************
MessageQueuingInstall - CUSTOM ACTION ENTRY POINT for installing MSMQ message queues
********************************************************************/
extern "C" UINT __stdcall MessageQueuingInstall(MSIHANDLE hInstall)
{
HRESULT hr = S_OK;
UINT er = ERROR_SUCCESS;
MQI_MESSAGE_QUEUE_LIST lstMessageQueues;
MQI_MESSAGE_QUEUE_PERMISSION_LIST lstMessageQueuePermissions;
int iCost = 0;
LPWSTR pwzRollbackActionData = NULL;
LPWSTR pwzExecuteActionData = NULL;
::ZeroMemory(&lstMessageQueues, sizeof(lstMessageQueues));
::ZeroMemory(&lstMessageQueuePermissions, sizeof(lstMessageQueuePermissions));
// initialize
hr = WcaInitialize(hInstall, "MessageQueuingInstall");
ExitOnFailure(hr, "Failed to initialize");
do
{
hr = MqiInitialize();
if (S_FALSE == hr)
{
WcaLog(LOGMSG_STANDARD, "Failed to load mqrt.dll.");
er = WcaErrorMessage(msierrMsmqCannotConnect, hr, INSTALLMESSAGE_ERROR | MB_ABORTRETRYIGNORE, 0);
switch (er)
{
case IDABORT:
ExitFunction1(hr = E_FAIL); // bail with error
case IDRETRY:
break; // retry
case IDIGNORE: __fallthrough;
default:
ExitFunction1(hr = S_OK); // pretend everything is okay and bail
}
}
ExitOnFailure(hr, "Failed to initialize MSMQ.");
} while (S_FALSE == hr);
// read message queues
hr = MqiMessageQueueRead(&lstMessageQueues);
ExitOnFailure(hr, "Failed to read MessageQueue table");
// read message queue permissions
hr = MqiMessageQueuePermissionRead(&lstMessageQueues, &lstMessageQueuePermissions);
ExitOnFailure(hr, "Failed to read message queue permissions");
// verify message queue elementes
hr = MqiMessageQueueVerify(&lstMessageQueues);
ExitOnFailure(hr, "Failed to verify message queue elements.");
if (lstMessageQueues.iInstallCount || lstMessageQueuePermissions.iInstallCount)
{
// schedule rollback action
hr = MqiMessageQueuePermissionInstall(&lstMessageQueuePermissions, &pwzRollbackActionData);
ExitOnFailure(hr, "Failed to add message queue permissions to rollback action data");
hr = MqiMessageQueueInstall(&lstMessageQueues, TRUE, &pwzRollbackActionData);
ExitOnFailure(hr, "Failed to add message queues to rollback action data");
hr = WcaDoDeferredAction(L"MessageQueuingRollbackInstall", pwzRollbackActionData, 0);
ExitOnFailure(hr, "Failed to schedule MessageQueuingRollbackInstall");
// schedule execute action
hr = MqiMessageQueueInstall(&lstMessageQueues, FALSE, &pwzExecuteActionData);
ExitOnFailure(hr, "Failed to add message queues to execute action data");
iCost += lstMessageQueues.iInstallCount * COST_MESSAGE_QUEUE_CREATE;
hr = MqiMessageQueuePermissionInstall(&lstMessageQueuePermissions, &pwzExecuteActionData);
ExitOnFailure(hr, "Failed to add message queue permissions to execute action data");
iCost += lstMessageQueues.iInstallCount * COST_MESSAGE_QUEUE_PERMISSION_ADD;
hr = WcaDoDeferredAction(L"MessageQueuingExecuteInstall", pwzExecuteActionData, iCost);
ExitOnFailure(hr, "Failed to schedule MessageQueuingExecuteInstall");
}
hr = S_OK;
LExit:
// clean up
MqiMessageQueueFreeList(&lstMessageQueues);
MqiMessageQueuePermissionFreeList(&lstMessageQueuePermissions);
ReleaseStr(pwzRollbackActionData);
ReleaseStr(pwzExecuteActionData);
// uninitialize
MqiUninitialize();
er = SUCCEEDED(hr) ? ERROR_SUCCESS : ERROR_INSTALL_FAILURE;
return WcaFinalize(er);
}
