108 lines
3.7 KiB
C++
108 lines
3.7 KiB
C++
/*
|
|
Copyright 1991-2015 Amebis
|
|
|
|
This file is part of MSICA.
|
|
|
|
MSICA is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
MSICA is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with MSICA. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
|
|
|
|
namespace MSICA {
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
// COpFileDelete
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
COpFileDelete::COpFileDelete(LPCWSTR pszFileName, int iTicks) :
|
|
COpTypeSingleString(pszFileName, iTicks)
|
|
{
|
|
}
|
|
|
|
|
|
HRESULT COpFileDelete::Execute(CSession *pSession)
|
|
{
|
|
DWORD dwError;
|
|
|
|
if (pSession->m_bRollbackEnabled) {
|
|
ATL::CAtlStringW sBackupName;
|
|
UINT uiCount = 0;
|
|
|
|
do {
|
|
// Rename the file to make a backup.
|
|
sBackupName.Format(L"%ls (orig %u)", (LPCWSTR)m_sValue, ++uiCount);
|
|
dwError = ::MoveFileW(m_sValue, sBackupName) ? NO_ERROR : ::GetLastError();
|
|
} while (dwError == ERROR_ALREADY_EXISTS);
|
|
if (dwError == NO_ERROR) {
|
|
// Order rollback action to restore from backup copy.
|
|
pSession->m_olRollback.AddHead(new COpFileMove(sBackupName, m_sValue));
|
|
|
|
// Order commit action to delete backup copy.
|
|
pSession->m_olCommit.AddTail(new COpFileDelete(sBackupName));
|
|
}
|
|
} else {
|
|
// Delete the file.
|
|
dwError = ::DeleteFileW(m_sValue) ? NO_ERROR : ::GetLastError();
|
|
}
|
|
|
|
if (dwError == NO_ERROR || dwError == ERROR_FILE_NOT_FOUND)
|
|
return S_OK;
|
|
else {
|
|
PMSIHANDLE hRecordProg = ::MsiCreateRecord(3);
|
|
::MsiRecordSetInteger(hRecordProg, 1, ERROR_INSTALL_FILE_DELETE);
|
|
::MsiRecordSetStringW(hRecordProg, 2, m_sValue );
|
|
::MsiRecordSetInteger(hRecordProg, 3, dwError );
|
|
::MsiProcessMessage(pSession->m_hInstall, INSTALLMESSAGE_ERROR, hRecordProg);
|
|
return AtlHresultFromWin32(dwError);
|
|
}
|
|
}
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
// COpFileMove
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
COpFileMove::COpFileMove(LPCWSTR pszFileSrc, LPCWSTR pszFileDst, int iTicks) :
|
|
COpTypeSrcDstString(pszFileSrc, pszFileDst, iTicks)
|
|
{
|
|
}
|
|
|
|
|
|
HRESULT COpFileMove::Execute(CSession *pSession)
|
|
{
|
|
DWORD dwError;
|
|
|
|
// Move the file.
|
|
dwError = ::MoveFileW(m_sValue1, m_sValue2) ? NO_ERROR : ::GetLastError();
|
|
if (dwError == NO_ERROR) {
|
|
if (pSession->m_bRollbackEnabled) {
|
|
// Order rollback action to move it back.
|
|
pSession->m_olRollback.AddHead(new COpFileMove(m_sValue2, m_sValue1));
|
|
}
|
|
|
|
return S_OK;
|
|
} else {
|
|
PMSIHANDLE hRecordProg = ::MsiCreateRecord(4);
|
|
::MsiRecordSetInteger(hRecordProg, 1, ERROR_INSTALL_FILE_MOVE);
|
|
::MsiRecordSetStringW(hRecordProg, 2, m_sValue1 );
|
|
::MsiRecordSetStringW(hRecordProg, 3, m_sValue2 );
|
|
::MsiRecordSetInteger(hRecordProg, 4, dwError );
|
|
::MsiProcessMessage(pSession->m_hInstall, INSTALLMESSAGE_ERROR, hRecordProg);
|
|
return AtlHresultFromWin32(dwError);
|
|
}
|
|
}
|
|
|
|
} // namespace MSICA
|