|
|
#include "precomp.h"
extern TCHAR g_szTitle[];extern TCHAR g_szBuildTemp[];extern TCHAR g_szBaseURL[];extern TCHAR g_szWizRoot[];extern HWND g_hWait;extern BOOL g_fLocalMode, g_fOCW;extern int s_iType;
extern DWORD GetRootFree(LPCTSTR pcszPath);extern HRESULT DownloadCab(HWND hDlg, LPTSTR szUrl, LPTSTR szFilename, LPCTSTR pcszDisplayname, int sComponent, BOOL &fIgnore);
void UpdateIni(LPCTSTR pcszCurrentIni, CCifComponent_t * pCifComponent_t){ TCHAR szNew[INTERNET_MAX_URL_LENGTH];
if (SUCCEEDED(pCifComponent_t->GetCustomData(TEXT("Win32"), szNew, countof(szNew)))) WritePrivateProfileString(TEXT("IEAK"), TEXT("Win32"), szNew, pcszCurrentIni); if (SUCCEEDED(pCifComponent_t->GetCustomData(TEXT("Ini"), szNew, countof(szNew)))) WritePrivateProfileString(VERSION, IK_INI, szNew, pcszCurrentIni);
}
BOOL CheckIniVersion(LPTSTR pszId, LPCTSTR pcszCurrentIni, CCifComponent_t * pCifComponent_t){ TCHAR szCurrentVer[64]; TCHAR szNewVer[64]; BOOL fRet = TRUE;
GetPrivateProfileString(VERSION, pszId, TEXT(""), szCurrentVer, countof(szCurrentVer), pcszCurrentIni);
fRet = SUCCEEDED(pCifComponent_t->GetCustomData(pszId, szNewVer, countof(szNewVer)));
return (fRet && (CheckVer(szCurrentVer, szNewVer) < 0));}
DWORD DownloadUpdateThreadProc(LPVOID lpvUrl){ TCHAR szDest[MAX_PATH]; TCHAR szMsg[MAX_PATH]; LPTSTR pFile; DWORD dwRet = 0;
GetTempPath(countof(szDest), szDest);
pFile = StrRChr((LPTSTR)lpvUrl, NULL, TEXT('/'));
if (pFile != NULL) pFile++;
if (StrCmpI(StrRChr((LPTSTR)lpvUrl, NULL, TEXT('.')), TEXT(".EXE")) == 0) { BOOL fIgnore = FALSE;
PathAppend(szDest, pFile);
if (DownloadCab(g_hWait, (LPTSTR)lpvUrl, szDest, NULL, 0, fIgnore) != NOERROR) { LoadString(g_rvInfo.hInst, IDS_UPDATEERROR, szMsg, countof(szMsg)); MessageBox(g_hWizard, szMsg, g_szTitle, MB_OK); return (DWORD)-1; } } else { BOOL fSuccess = TRUE; BOOL fIgnore = FALSE;
PathCombine(szDest, g_szBuildTemp, pFile);
if (DownloadCab(g_hWait, (LPTSTR)lpvUrl, szDest, NULL, 0, fIgnore) == NOERROR) { TCHAR szTempDir[MAX_PATH];
PathCombine(szTempDir, g_szBuildTemp, TEXT("update")); PathCreatePath(szTempDir);
if (ExtractFilesWrap(szDest, szTempDir, 0, NULL, NULL, 0) == ERROR_SUCCESS) { TCHAR szInf[MAX_PATH];
PathCombine(szInf, szTempDir, TEXT("ieak6.inf")); dwRet = RunSetupCommandWrap(g_hWizard, szInf, NULL, szTempDir, NULL, NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL);
PathRemovePath(szTempDir); DeleteFile(szDest); } else fSuccess = FALSE; } else fSuccess = FALSE;
SendMessage(g_hWait, WM_CLOSE, 0, 0);
if (!fSuccess) { LoadString(g_rvInfo.hInst, IDS_UPDATEERROR, szMsg, countof(szMsg)); MessageBox(g_hWizard, szMsg, g_szTitle, MB_OK); return (DWORD)-1; } }
return dwRet;}
void UpdateIEAK(HWND hDlg){ TCHAR szCurrentIni[MAX_PATH]; TCHAR szMsg[MAX_PATH]; TCHAR szUpdateUrl[INTERNET_MAX_URL_LENGTH]; TCHAR szCifLang[8]; TCHAR szWizLang[8]; ICifComponent * pCifComponent; CCifComponent_t * pCifComponent_t; DWORD dwTid; DWORD dwRet = ERROR_SUCCESS; BOOL fNo = FALSE; HWND hWait;
if (!g_lpCifFileNew || !(SUCCEEDED(g_lpCifFileNew->FindComponent(TEXT("ieak6OPT"), &pCifComponent)))) return;
pCifComponent_t = new CCifComponent_t((ICifRWComponent *)pCifComponent);
pCifComponent_t->GetCustomData(TEXT("Lang"), szCifLang, countof(szCifLang));
StrCpy(szUpdateUrl, g_szBaseURL); if (*(CharPrev(szUpdateUrl, szUpdateUrl + StrLen(szUpdateUrl))) != TEXT('/')) StrCat(szUpdateUrl, TEXT("/"));
PathCombine(szCurrentIni, g_szWizRoot, TEXT("ieak.ini"));
GetPrivateProfileString(VERSION, TEXT("Lang"), TEXT(""), szWizLang, countof(szWizLang), szCurrentIni);
// only allow self-update for same language as the current wizard
if (StrCmpI(szCifLang, szWizLang) != 0) goto exit;
if (CheckIniVersion(IK_FULL, szCurrentIni, pCifComponent_t)) { pCifComponent_t->GetCustomData(TEXT("FullText"), szMsg, countof(szMsg));
if (MessageBox(g_hWizard, szMsg, g_szTitle, MB_YESNO | MB_DEFBUTTON2) == IDYES) { HANDLE hThread; TCHAR szExeName[MAX_PATH]; DWORD dwFlags; ICifComponent * pCifExeComponent; CCifComponent_t * pCifExeComponent_t;
if (SUCCEEDED(g_lpCifFileNew->FindComponent(TEXT("ieak6EXE"), &pCifExeComponent))) { pCifExeComponent_t = new CCifComponent_t((ICifRWComponent *)pCifExeComponent);
if (SUCCEEDED(pCifExeComponent_t->GetUrl(0, szExeName, countof(szExeName), &dwFlags))) { // do not allow self-update to continue if not enough size (size of Exe +
// extract size approximated by twice the size of the exe)
if (((dwFlags = pCifExeComponent_t->GetDownloadSize() * 3) > GetRootFree(g_szBuildTemp)) && dwFlags) { TCHAR szMsg[MAX_PATH]; TCHAR szMsgTemplate[MAX_PATH];
LoadString(g_rvInfo.hInst, IDS_ERROR_UPDATESPACE, szMsgTemplate, countof(szMsgTemplate)); wnsprintf(szMsg, countof(szMsg), szMsgTemplate, pCifExeComponent_t->GetDownloadSize() * 3); dwRet = (DWORD)-1; MessageBox(hDlg, szMsg, g_szTitle, MB_OK | MB_ICONEXCLAMATION); } else { StrCat(szUpdateUrl, szExeName);
hWait = CreateDialog(g_rvInfo.hInst, MAKEINTRESOURCE(IDD_UPDATE), hDlg, DownloadStatusDlgProc); ShowWindow( hWait, SW_SHOWNORMAL );
hThread = CreateThread(NULL, 4096, DownloadUpdateThreadProc, (LPVOID)szUpdateUrl, 0, &dwTid);
while (MsgWaitForMultipleObjects(1, &hThread, FALSE, INFINITE, QS_ALLINPUT) != WAIT_OBJECT_0) { MSG msg;
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) { TranslateMessage(&msg); DispatchMessage(&msg); } } GetExitCodeThread(hThread, &dwRet); CloseHandle(hThread); SendMessage(hWait, WM_CLOSE, 0, 0); } } else dwRet = (DWORD)-1;
delete pCifExeComponent_t; } else dwRet = (DWORD)-1;
if (dwRet == -1) { g_fLocalMode = TRUE; goto exit; } else { SHELLEXECUTEINFO shInfo; DWORD dwPID; TCHAR szTempBuf[MAX_PATH + 32]; TCHAR szTempPath[MAX_PATH];
PathCombine(szTempBuf, g_szWizRoot, TEXT("update.exe")); GetTempPath(countof(szTempPath), szTempPath); CopyFileToDir(szTempBuf, szTempPath);
ZeroMemory(&shInfo, sizeof(shInfo)); shInfo.cbSize = sizeof(shInfo); shInfo.hwnd = GetDesktopWindow(); shInfo.fMask = SEE_MASK_NOCLOSEPROCESS; shInfo.lpVerb = TEXT("open"); shInfo.lpFile = TEXT("update.exe"); dwPID = GetCurrentProcessId(); WCHAR wcType; switch (s_iType) { case BRANDED: wcType = TEXT('B'); break; case REDIST: wcType = TEXT('R'); break;
case INTRANET: default: wcType = TEXT('I'); break; } if (g_fOCW) wnsprintf(szTempBuf, countof(szTempBuf), TEXT("/o /p:%lu /m:%c"), dwPID,wcType); else wnsprintf(szTempBuf, countof(szTempBuf), TEXT("/p:%lu /m:%c"), dwPID,wcType);
shInfo.lpParameters = szTempBuf; shInfo.lpDirectory = szTempPath; shInfo.nShow = SW_SHOWNORMAL;
ShellExecuteEx(&shInfo); CloseHandle(shInfo.hProcess); DoCancel(); goto exit; } } else fNo = TRUE; }
if (!fNo && (CheckIniVersion(IK_CAB, szCurrentIni, pCifComponent_t))) { pCifComponent_t->GetCustomData(TEXT("CabText"), szMsg, countof(szMsg));
if (MessageBox(g_hWizard, szMsg, g_szTitle, MB_YESNO | MB_DEFBUTTON2) == IDYES) { TCHAR szNewVer[MAX_PATH]; TCHAR szCabName[32]; DWORD dwFlags; HANDLE hThread; ICifComponent * pCifCabComponent; CCifComponent_t * pCifCabComponent_t;
if (SUCCEEDED(g_lpCifFileNew->FindComponent(TEXT("ieak6CAB"), &pCifCabComponent))) { pCifCabComponent_t = new CCifComponent_t((ICifRWComponent *)pCifCabComponent);
if (SUCCEEDED(pCifCabComponent_t->GetUrl(0, szCabName, countof(szCabName), &dwFlags))) { // do not allow self-update to continue if not enough size (size of Cab +
// extract size approximated by twice the size of the cab)
if (((dwFlags = pCifCabComponent_t->GetDownloadSize() * 3) > GetRootFree(g_szBuildTemp)) && dwFlags) { TCHAR szMsg[MAX_PATH]; TCHAR szMsgTemplate[MAX_PATH];
LoadString(g_rvInfo.hInst, IDS_ERROR_UPDATESPACE, szMsgTemplate, countof(szMsgTemplate)); wnsprintf(szMsg, countof(szMsg), szMsgTemplate, pCifCabComponent_t->GetDownloadSize() * 3); dwRet = (DWORD)-1; MessageBox(hDlg, szMsg, g_szTitle, MB_OK | MB_ICONEXCLAMATION); } else { StrCat(szUpdateUrl, szCabName);
hWait = CreateDialog(g_rvInfo.hInst, MAKEINTRESOURCE(IDD_UPDATE), hDlg, DownloadStatusDlgProc); ShowWindow( hWait, SW_SHOWNORMAL );
hThread = CreateThread(NULL, 4096, DownloadUpdateThreadProc, (LPVOID)szUpdateUrl, 0, &dwTid);
while (MsgWaitForMultipleObjects(1, &hThread, FALSE, INFINITE, QS_ALLINPUT) != WAIT_OBJECT_0) { MSG msg;
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) { TranslateMessage(&msg); DispatchMessage(&msg); } } GetExitCodeThread(hThread, &dwRet); CloseHandle(hThread); SendMessage(hWait, WM_CLOSE, 0, 0); } } else dwRet = (DWORD)-1;
delete pCifCabComponent_t; } else dwRet = (DWORD)-1;
if (dwRet == (DWORD)-1) { g_fLocalMode = TRUE; goto exit; }
if (CheckIniVersion(IK_INI, szCurrentIni, pCifComponent_t)) UpdateIni(szCurrentIni, pCifComponent_t);
if (SUCCEEDED(pCifComponent_t->GetCustomData(TEXT("Cab"), szNewVer, countof(szNewVer)))) WritePrivateProfileString(VERSION, IK_CAB, szNewVer, szCurrentIni); } else fNo = TRUE; }
if (!fNo && (CheckIniVersion(IK_INI, szCurrentIni, pCifComponent_t))) { pCifComponent_t->GetCustomData(TEXT("IniText"), szMsg, countof(szMsg));
if (MessageBox(g_hWizard, szMsg, g_szTitle, MB_YESNO | MB_DEFBUTTON2) == IDYES) { UpdateIni(szCurrentIni, pCifComponent_t); } else fNo = TRUE; }
if (dwRet == ERROR_SUCCESS_REBOOT_REQUIRED) DoReboot(g_hWizard);
if (fNo) g_fLocalMode = TRUE;
exit: delete pCifComponent_t;}
|
