86 lines
1.8 KiB
C++
86 lines
1.8 KiB
C++
|
#include "stdafx.h"
|
||
|
|
#include <iucommon.h>
|
||
|
|
#include <trust.h>
|
||
|
|
#include <logging.h>
|
||
|
|
#include <fileutil.h>
|
||
|
|
#include "iudl.h"
|
||
|
|
#include <download.h>
|
||
|
|
#include <UrlAgent.h>
|
||
|
|
|
||
|
|
extern HANDLE g_hEngineLoadQuit;
|
||
|
|
extern CIUUrlAgent *g_pIUUrlAgent;
|
||
|
|
|
||
|
|
HRESULT IUDownloadFile(LPCTSTR pszDownloadUrl,
|
||
|
|
LPCTSTR pszLocalFile,
|
||
|
|
BOOL fDecompress,
|
||
|
|
BOOL fCheckTrust,
|
||
|
|
DWORD dwFlags)
|
||
|
|
{
|
||
|
|
LOG_Block("IUDownloadFile()");
|
||
|
|
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
if (FAILED(hr = g_pIUUrlAgent->IsIdentFromPolicy()))
|
||
|
|
{
|
||
|
|
goto done;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (S_FALSE == hr)
|
||
|
|
{
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
else // S_OK
|
||
|
|
{
|
||
|
|
dwFlags |= WUDF_DONTALLOWPROXY;
|
||
|
|
LOG_Internet(_T("WUDF_DONTALLOWPROXY set"));
|
||
|
|
}
|
||
|
|
|
||
|
|
// do the download
|
||
|
|
hr = DownloadFileLite(pszDownloadUrl, pszLocalFile, g_hEngineLoadQuit, dwFlags);
|
||
|
|
if (FAILED(hr))
|
||
|
|
goto done;
|
||
|
|
|
||
|
|
// check for decompress requested
|
||
|
|
if (fCheckTrust)
|
||
|
|
{
|
||
|
|
if (FAILED(hr = VerifyFileTrust(pszLocalFile, NULL, ReadWUPolicyShowTrustUI())))
|
||
|
|
{
|
||
|
|
LOG_ErrorMsg(hr);
|
||
|
|
|
||
|
|
// The file was not trusted.. delete it.
|
||
|
|
DeleteFile(pszLocalFile);
|
||
|
|
goto done;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (fDecompress)
|
||
|
|
{
|
||
|
|
TCHAR *pszLocalDir = NULL;
|
||
|
|
DWORD cchNeed = _tcslen(pszLocalFile) + 1;
|
||
|
|
|
||
|
|
__try { pszLocalDir = (LPTSTR)_alloca(cchNeed * sizeof(TCHAR)); }
|
||
|
|
__except(EXCEPTION_EXECUTE_HANDLER) { pszLocalDir = NULL; }
|
||
|
|
if (pszLocalDir == NULL)
|
||
|
|
{
|
||
|
|
LOG_ErrorMsg(ERROR_OUTOFMEMORY);
|
||
|
|
hr = E_OUTOFMEMORY;
|
||
|
|
goto done;
|
||
|
|
}
|
||
|
|
|
||
|
|
hr=StringCchCopyEx(pszLocalDir,cchNeed,pszLocalFile,NULL,NULL,MISTSAFE_STRING_FLAGS);
|
||
|
|
if(FAILED(hr))
|
||
|
|
{
|
||
|
|
LOG_ErrorMsg(hr);
|
||
|
|
goto done;
|
||
|
|
}
|
||
|
|
|
||
|
|
PathRemoveFileSpec(pszLocalDir);
|
||
|
|
IUExtractFiles(pszLocalFile, pszLocalDir);
|
||
|
|
}
|
||
|
|
|
||
|
|
done:
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
}
|
||
|
|
|